Merge from vendor branch BINUTILS:
[dragonfly.git] / sbin / badsect / badsect.c
1 /*
2  * Copyright (c) 1981, 1983, 1993
3  *      The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *      This product includes software developed by the University of
16  *      California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * @(#) Copyright (c) 1981, 1983, 1993 The Regents of the University of California.  All rights reserved.
34  * @(#)badsect.c        8.1 (Berkeley) 6/5/93
35  * $FreeBSD: src/sbin/badsect/badsect.c,v 1.7.2.2 2001/07/30 10:30:04 dd Exp $
36  * $DragonFly: src/sbin/badsect/badsect.c,v 1.5 2003/11/01 17:15:57 drhodus Exp $
37  */
38
39 /*
40  * badsect
41  *
42  * Badsect takes a list of file-system relative sector numbers
43  * and makes files containing the blocks of which these sectors are a part.
44  * It can be used to contain sectors which have problems if these sectors
45  * are not part of the bad file for the pack (see bad144).  For instance,
46  * this program can be used if the driver for the file system in question
47  * does not support bad block forwarding.
48  */
49 #include <sys/param.h>
50 #include <sys/stat.h>
51
52 #include <vfs/ufs/fs.h>
53 #include <vfs/ufs/dinode.h>
54
55 #include <err.h>
56 #include <dirent.h>
57 #include <fcntl.h>
58 #include <paths.h>
59 #include <stdio.h>
60 #include <stdlib.h>
61 #include <string.h>
62 #include <unistd.h>
63
64 union {
65         struct  fs fs;
66         char    fsx[SBSIZE];
67 } ufs;
68 #define sblock  ufs.fs
69 union {
70         struct  cg cg;
71         char    cgx[MAXBSIZE];
72 } ucg;
73 #define acg     ucg.cg
74 struct  fs *fs;
75 int     fso, fsi;
76 int     errs;
77 long    dev_bsize = 1;
78
79 char buf[MAXBSIZE];
80
81 void    rdfs(daddr_t, int, char *);
82 int     chkuse(daddr_t, int);
83
84 static void
85 usage(void)
86 {
87         fprintf(stderr, "usage: badsect bbdir blkno ...\n");
88         exit(1);
89 }
90
91 int
92 main(int argc, char **argv)
93 {
94         daddr_t diskbn;
95         daddr_t number;
96         struct stat stbuf, devstat;
97         register struct dirent *dp;
98         DIR *dirp;
99         char name[2 * MAXPATHLEN];
100         char *name_dir_end;
101
102         if (argc < 3)
103                 usage();
104         if (chdir(argv[1]) < 0 || stat(".", &stbuf) < 0)
105                 err(2, "%s", argv[1]);
106         strcpy(name, _PATH_DEV);
107         if ((dirp = opendir(name)) == NULL)
108                 err(3, "%s", name);
109         name_dir_end = name + strlen(name);
110         while ((dp = readdir(dirp)) != NULL) {
111                 strcpy(name_dir_end, dp->d_name);
112                 if (lstat(name, &devstat) < 0)
113                         err(4, "%s", name);
114                 if (stbuf.st_dev == devstat.st_rdev &&
115                     (devstat.st_mode & IFMT) == IFCHR)
116                         break;
117         }
118         closedir(dirp);
119         if (dp == NULL) {
120                 printf("Cannot find dev 0%lo corresponding to %s\n",
121                     (u_long)stbuf.st_rdev, argv[1]);
122                 exit(5);
123         }
124         if ((fsi = open(name, O_RDONLY)) < 0)
125                 err(6, "%s", name);
126         fs = &sblock;
127         rdfs(SBOFF, SBSIZE, (char *)fs);
128         dev_bsize = fs->fs_fsize / fsbtodb(fs, 1);
129         for (argc -= 2, argv += 2; argc > 0; argc--, argv++) {
130                 number = atol(*argv);
131                 if (chkuse(number, 1))
132                         continue;
133                 /*
134                  * Print a warning if converting the block number to a dev_t
135                  * will truncate it.  badsect was not very useful in versions
136                  * of BSD before 4.4 because dev_t was 16 bits and another
137                  * bit was lost by bogus sign extensions.
138                  */
139                 diskbn = dbtofsb(fs, number);
140                 if ((dev_t)diskbn != diskbn) {
141                         printf("sector %ld cannot be represented as a dev_t\n",
142                             (long)number);
143                         errs++;
144                 }
145                 else if (mknod(*argv, IFMT|0600, (dev_t)diskbn) < 0) {
146                         warn("%s", *argv);
147                         errs++;
148                 }
149         }
150         printf("Don't forget to run ``fsck %s''\n", name);
151         exit(errs);
152 }
153
154 int
155 chkuse(daddr_t blkno, int cnt)
156 {
157         int cg;
158         daddr_t fsbn, bn;
159
160         fsbn = dbtofsb(fs, blkno);
161         if ((unsigned)(fsbn+cnt) > fs->fs_size) {
162                 printf("block %ld out of range of file system\n", (long)blkno);
163                 return (1);
164         }
165         cg = dtog(fs, fsbn);
166         if (fsbn < cgdmin(fs, cg)) {
167                 if (cg == 0 || (fsbn+cnt) > cgsblock(fs, cg)) {
168                         printf("block %ld in non-data area: cannot attach\n",
169                             (long)blkno);
170                         return (1);
171                 }
172         } else {
173                 if ((fsbn+cnt) > cgbase(fs, cg+1)) {
174                         printf("block %ld in non-data area: cannot attach\n",
175                             (long)blkno);
176                         return (1);
177                 }
178         }
179         rdfs(fsbtodb(fs, cgtod(fs, cg)), (int)sblock.fs_cgsize,
180             (char *)&acg);
181         if (!cg_chkmagic(&acg)) {
182                 fprintf(stderr, "cg %d: bad magic number\n", cg);
183                 errs++;
184                 return (1);
185         }
186         bn = dtogd(fs, fsbn);
187         if (isclr(cg_blksfree(&acg), bn))
188                 printf("Warning: sector %ld is in use\n", (long)blkno);
189         return (0);
190 }
191
192 /*
193  * read a block from the file system
194  */
195 void
196 rdfs(daddr_t bno, int size, char *bf)
197 {
198         int n;
199
200         if (lseek(fsi, (off_t)bno * dev_bsize, SEEK_SET) < 0) {
201                 printf("seek error: %ld\n", (long)bno);
202                 err(1, "rdfs");
203         }
204         n = read(fsi, bf, size);
205         if (n != size) {
206                 printf("read error: %ld\n", (long)bno);
207                 err(1, "rdfs");
208         }
209 }