084c76d36cd5ee31c612e75b2fb9ac344d110023
[oweals/busybox.git] / util-linux / fsck_minix.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * fsck.c - a file system consistency checker for Linux.
4  *
5  * (C) 1991, 1992 Linus Torvalds. This file may be redistributed
6  * as per the GNU copyleft.
7  */
8
9 /*
10  * 09.11.91  -  made the first rudimetary functions
11  *
12  * 10.11.91  -  updated, does checking, no repairs yet.
13  *              Sent out to the mailing-list for testing.
14  *
15  * 14.11.91  -  Testing seems to have gone well. Added some
16  *              correction-code, and changed some functions.
17  *
18  * 15.11.91  -  More correction code. Hopefully it notices most
19  *              cases now, and tries to do something about them.
20  *
21  * 16.11.91  -  More corrections (thanks to Mika Jalava). Most
22  *              things seem to work now. Yeah, sure.
23  *
24  *
25  * 19.04.92  -  Had to start over again from this old version, as a
26  *              kernel bug ate my enhanced fsck in february.
27  *
28  * 28.02.93  -  added support for different directory entry sizes..
29  *
30  * Sat Mar  6 18:59:42 1993, faith@cs.unc.edu: Output namelen with
31  *                           super-block information
32  *
33  * Sat Oct  9 11:17:11 1993, faith@cs.unc.edu: make exit status conform
34  *                           to that required by fsutil
35  *
36  * Mon Jan  3 11:06:52 1994 - Dr. Wettstein (greg%wind.uucp@plains.nodak.edu)
37  *                            Added support for file system valid flag.  Also
38  *                            added program_version variable and output of
39  *                            program name and version number when program
40  *                            is executed.
41  *
42  * 30.10.94 - added support for v2 filesystem
43  *            (Andreas Schwab, schwab@issan.informatik.uni-dortmund.de)
44  *
45  * 10.12.94  -  added test to prevent checking of mounted fs adapted
46  *              from Theodore Ts'o's (tytso@athena.mit.edu) e2fsck
47  *              program.  (Daniel Quinlan, quinlan@yggdrasil.com)
48  *
49  * 01.07.96  - Fixed the v2 fs stuff to use the right #defines and such
50  *             for modern libcs (janl@math.uio.no, Nicolai Langfeldt)
51  *
52  * 02.07.96  - Added C bit fiddling routines from rmk@ecs.soton.ac.uk 
53  *             (Russell King).  He made them for ARM.  It would seem
54  *             that the ARM is powerful enough to do this in C whereas
55  *             i386 and m64k must use assembly to get it fast >:-)
56  *             This should make minix fsck systemindependent.
57  *             (janl@math.uio.no, Nicolai Langfeldt)
58  *
59  * 04.11.96  - Added minor fixes from Andreas Schwab to avoid compiler
60  *             warnings.  Added mc68k bitops from 
61  *             Joerg Dorchain <dorchain@mpi-sb.mpg.de>.
62  *
63  * 06.11.96  - Added v2 code submitted by Joerg Dorchain, but written by
64  *             Andreas Schwab.
65  *
66  * 1999-02-22 Arkadiusz Mi¶kiewicz <misiek@misiek.eu.org>
67  * - added Native Language Support
68  *
69  *
70  * I've had no time to add comments - hopefully the function names
71  * are comments enough. As with all file system checkers, this assumes
72  * the file system is quiescent - don't use it on a mounted device
73  * unless you can be sure nobody is writing to it (and remember that the
74  * kernel can write to it when it searches for files).
75  *
76  * Usuage: fsck [-larvsm] device
77  *      -l for a listing of all the filenames
78  *      -a for automatic repairs (not implemented)
79  *      -r for repairs (interactive) (not implemented)
80  *      -v for verbose (tells how many files)
81  *      -s for super-block info
82  *      -m for minix-like "mode not cleared" warnings
83  *      -f force filesystem check even if filesystem marked as valid
84  *
85  * The device may be a block device or a image of one, but this isn't
86  * enforced (but it's not much fun on a character device :-). 
87  */
88
89 #include "internal.h"
90 #include <stdio.h>
91 #include <errno.h>
92 #include <unistd.h>
93 #include <string.h>
94 #include <fcntl.h>
95 #include <ctype.h>
96 #include <stdlib.h>
97 #include <termios.h>
98 #include <mntent.h>
99 #include <sys/stat.h>
100 #include <sys/param.h>                  /* for PATH_MAX */
101
102 #include <linux/fs.h>
103 #include <linux/minix_fs.h>
104
105 #ifdef MINIX2_SUPER_MAGIC2
106 #define HAVE_MINIX2 1
107 #endif
108
109 #ifndef __linux__
110 #define volatile
111 #endif
112
113 #define ROOT_INO 1
114
115 #define UPPER(size,n) ((size+((n)-1))/(n))
116 #define INODE_SIZE (sizeof(struct minix_inode))
117 #ifdef HAVE_MINIX2
118 #define INODE_SIZE2 (sizeof(struct minix2_inode))
119 #define INODE_BLOCKS UPPER(INODES, (version2 ? MINIX2_INODES_PER_BLOCK \
120                                     : MINIX_INODES_PER_BLOCK))
121 #else
122 #define INODE_BLOCKS UPPER(INODES, (MINIX_INODES_PER_BLOCK))
123 #endif
124 #define INODE_BUFFER_SIZE (INODE_BLOCKS * BLOCK_SIZE)
125
126 #define BITS_PER_BLOCK (BLOCK_SIZE<<3)
127
128 static char *program_name = "fsck.minix";
129 static char *program_version = "1.2 - 11/11/96";
130 static char *device_name = NULL;
131 static int IN;
132 static int repair = 0, automatic = 0, verbose = 0, list = 0, show =
133         0, warn_mode = 0, force = 0;
134 static int directory = 0, regular = 0, blockdev = 0, chardev = 0, links =
135         0, symlinks = 0, total = 0;
136
137 static int changed = 0;                 /* flags if the filesystem has been changed */
138 static int errors_uncorrected = 0;      /* flag if some error was not corrected */
139 static int dirsize = 16;
140 static int namelen = 14;
141 static int version2 = 0;
142 static struct termios termios;
143 static int termios_set = 0;
144
145 /* File-name data */
146 #define MAX_DEPTH 50
147 static int name_depth = 0;
148 static char name_list[MAX_DEPTH][PATH_MAX + 1];
149
150 static char *inode_buffer = NULL;
151
152 #define Inode (((struct minix_inode *) inode_buffer)-1)
153 #define Inode2 (((struct minix2_inode *) inode_buffer)-1)
154 static char super_block_buffer[BLOCK_SIZE];
155
156 #define Super (*(struct minix_super_block *)super_block_buffer)
157 #define INODES ((unsigned long)Super.s_ninodes)
158 #ifdef HAVE_MINIX2
159 #define ZONES ((unsigned long)(version2 ? Super.s_zones : Super.s_nzones))
160 #else
161 #define ZONES ((unsigned long)(Super.s_nzones))
162 #endif
163 #define IMAPS ((unsigned long)Super.s_imap_blocks)
164 #define ZMAPS ((unsigned long)Super.s_zmap_blocks)
165 #define FIRSTZONE ((unsigned long)Super.s_firstdatazone)
166 #define ZONESIZE ((unsigned long)Super.s_log_zone_size)
167 #define MAXSIZE ((unsigned long)Super.s_max_size)
168 #define MAGIC (Super.s_magic)
169 #define NORM_FIRSTZONE (2+IMAPS+ZMAPS+INODE_BLOCKS)
170
171 static char *inode_map;
172 static char *zone_map;
173
174 static unsigned char *inode_count = NULL;
175 static unsigned char *zone_count = NULL;
176
177 static void recursive_check(unsigned int ino);
178 static void recursive_check2(unsigned int ino);
179
180 #define inode_in_use(x) (bit(inode_map,(x)))
181 #define zone_in_use(x) (bit(zone_map,(x)-FIRSTZONE+1))
182
183 #define mark_inode(x) (setbit(inode_map,(x)),changed=1)
184 #define unmark_inode(x) (clrbit(inode_map,(x)),changed=1)
185
186 #define mark_zone(x) (setbit(zone_map,(x)-FIRSTZONE+1),changed=1)
187 #define unmark_zone(x) (clrbit(zone_map,(x)-FIRSTZONE+1),changed=1)
188
189 static void leave(int) __attribute__ ((noreturn));
190 static void leave(int status)
191 {
192         if (termios_set)
193                 tcsetattr(0, TCSANOW, &termios);
194         exit(status);
195 }
196
197 static void show_usage(void)
198 {
199         fprintf(stderr, "BusyBox v%s (%s) multi-call binary -- GPL2\n\n",
200                         BB_VER, BB_BT);
201         fprintf(stderr, "Usage: %s [-larvsmf] /dev/name\n\n", program_name);
202         fprintf(stderr,
203                         "Performs a consistency check for MINIX filesystems.\n\n");
204         fprintf(stderr, "OPTIONS:\n");
205         fprintf(stderr, "\t-l\tLists all filenames\n");
206         fprintf(stderr, "\t-r\tPerform interactive repairs\n");
207         fprintf(stderr, "\t-a\tPerform automatic repairs\n");
208         fprintf(stderr, "\t-v\tverbose\n");
209         fprintf(stderr, "\t-s\tOutputs super-block information\n");
210         fprintf(stderr,
211                         "\t-m\tActivates MINIX-like \"mode not cleared\" warnings\n");
212         fprintf(stderr, "\t-f\tForce file system check.\n\n");
213         leave(16);
214 }
215
216 static void die(const char *str)
217 {
218         fprintf(stderr, "%s: %s\n", program_name, str);
219         leave(8);
220 }
221
222 /*
223  * This simply goes through the file-name data and prints out the
224  * current file.
225  */
226 static void print_current_name(void)
227 {
228         int i = 0;
229
230         while (i < name_depth)
231                 printf("/%.*s", namelen, name_list[i++]);
232         if (i == 0)
233                 printf("/");
234 }
235
236 static int ask(const char *string, int def)
237 {
238         int c;
239
240         if (!repair) {
241                 printf("\n");
242                 errors_uncorrected = 1;
243                 return 0;
244         }
245         if (automatic) {
246                 printf("\n");
247                 if (!def)
248                         errors_uncorrected = 1;
249                 return def;
250         }
251         printf(def ? "%s (y/n)? " : "%s (n/y)? ", string);
252         for (;;) {
253                 fflush(stdout);
254                 if ((c = getchar()) == EOF) {
255                         if (!def)
256                                 errors_uncorrected = 1;
257                         return def;
258                 }
259                 c = toupper(c);
260                 if (c == 'Y') {
261                         def = 1;
262                         break;
263                 } else if (c == 'N') {
264                         def = 0;
265                         break;
266                 } else if (c == ' ' || c == '\n')
267                         break;
268         }
269         if (def)
270                 printf("y\n");
271         else {
272                 printf("n\n");
273                 errors_uncorrected = 1;
274         }
275         return def;
276 }
277
278 /*
279  * Make certain that we aren't checking a filesystem that is on a
280  * mounted partition.  Code adapted from e2fsck, Copyright (C) 1993,
281  * 1994 Theodore Ts'o.  Also licensed under GPL.
282  */
283 static void check_mount(void)
284 {
285         FILE *f;
286         struct mntent *mnt;
287         int cont;
288         int fd;
289
290         if ((f = setmntent(MOUNTED, "r")) == NULL)
291                 return;
292         while ((mnt = getmntent(f)) != NULL)
293                 if (strcmp(device_name, mnt->mnt_fsname) == 0)
294                         break;
295         endmntent(f);
296         if (!mnt)
297                 return;
298
299         /*
300          * If the root is mounted read-only, then /etc/mtab is
301          * probably not correct; so we won't issue a warning based on
302          * it.
303          */
304         fd = open(MOUNTED, O_RDWR);
305         if (fd < 0 && errno == EROFS)
306                 return;
307         else
308                 close(fd);
309
310         printf("%s is mounted.   ", device_name);
311         if (isatty(0) && isatty(1))
312                 cont = ask("Do you really want to continue", 0);
313         else
314                 cont = 0;
315         if (!cont) {
316                 printf("check aborted.\n");
317                 exit(0);
318         }
319         return;
320 }
321
322 /*
323  * check_zone_nr checks to see that *nr is a valid zone nr. If it
324  * isn't, it will possibly be repaired. Check_zone_nr sets *corrected
325  * if an error was corrected, and returns the zone (0 for no zone
326  * or a bad zone-number).
327  */
328 static int check_zone_nr(unsigned short *nr, int *corrected)
329 {
330         if (!*nr)
331                 return 0;
332         if (*nr < FIRSTZONE)
333                 printf("Zone nr < FIRSTZONE in file `");
334         else if (*nr >= ZONES)
335                 printf("Zone nr >= ZONES in file `");
336         else
337                 return *nr;
338         print_current_name();
339         printf("'.");
340         if (ask("Remove block", 1)) {
341                 *nr = 0;
342                 *corrected = 1;
343         }
344         return 0;
345 }
346
347 #ifdef HAVE_MINIX2
348 static int check_zone_nr2(unsigned int *nr, int *corrected)
349 {
350         if (!*nr)
351                 return 0;
352         if (*nr < FIRSTZONE)
353                 printf("Zone nr < FIRSTZONE in file `");
354         else if (*nr >= ZONES)
355                 printf("Zone nr >= ZONES in file `");
356         else
357                 return *nr;
358         print_current_name();
359         printf("'.");
360         if (ask("Remove block", 1)) {
361                 *nr = 0;
362                 *corrected = 1;
363         }
364         return 0;
365 }
366 #endif
367
368 /*
369  * read-block reads block nr into the buffer at addr.
370  */
371 static void read_block(unsigned int nr, char *addr)
372 {
373         if (!nr) {
374                 memset(addr, 0, BLOCK_SIZE);
375                 return;
376         }
377         if (BLOCK_SIZE * nr != lseek(IN, BLOCK_SIZE * nr, SEEK_SET)) {
378                 printf("Read error: unable to seek to block in file '");
379                 print_current_name();
380                 printf("'\n");
381                 memset(addr, 0, BLOCK_SIZE);
382                 errors_uncorrected = 1;
383         } else if (BLOCK_SIZE != read(IN, addr, BLOCK_SIZE)) {
384                 printf("Read error: bad block in file '");
385                 print_current_name();
386                 printf("'\n");
387                 memset(addr, 0, BLOCK_SIZE);
388                 errors_uncorrected = 1;
389         }
390 }
391
392 /*
393  * write_block writes block nr to disk.
394  */
395 static void write_block(unsigned int nr, char *addr)
396 {
397         if (!nr)
398                 return;
399         if (nr < FIRSTZONE || nr >= ZONES) {
400                 printf("Internal error: trying to write bad block\n"
401                            "Write request ignored\n");
402                 errors_uncorrected = 1;
403                 return;
404         }
405         if (BLOCK_SIZE * nr != lseek(IN, BLOCK_SIZE * nr, SEEK_SET))
406                 die("seek failed in write_block");
407         if (BLOCK_SIZE != write(IN, addr, BLOCK_SIZE)) {
408                 printf("Write error: bad block in file '");
409                 print_current_name();
410                 printf("'\n");
411                 errors_uncorrected = 1;
412         }
413 }
414
415 /*
416  * map-block calculates the absolute block nr of a block in a file.
417  * It sets 'changed' if the inode has needed changing, and re-writes
418  * any indirect blocks with errors.
419  */
420 static int map_block(struct minix_inode *inode, unsigned int blknr)
421 {
422         unsigned short ind[BLOCK_SIZE >> 1];
423         unsigned short dind[BLOCK_SIZE >> 1];
424         int blk_chg, block, result;
425
426         if (blknr < 7)
427                 return check_zone_nr(inode->i_zone + blknr, &changed);
428         blknr -= 7;
429         if (blknr < 512) {
430                 block = check_zone_nr(inode->i_zone + 7, &changed);
431                 read_block(block, (char *) ind);
432                 blk_chg = 0;
433                 result = check_zone_nr(blknr + ind, &blk_chg);
434                 if (blk_chg)
435                         write_block(block, (char *) ind);
436                 return result;
437         }
438         blknr -= 512;
439         block = check_zone_nr(inode->i_zone + 8, &changed);
440         read_block(block, (char *) dind);
441         blk_chg = 0;
442         result = check_zone_nr(dind + (blknr / 512), &blk_chg);
443         if (blk_chg)
444                 write_block(block, (char *) dind);
445         block = result;
446         read_block(block, (char *) ind);
447         blk_chg = 0;
448         result = check_zone_nr(ind + (blknr % 512), &blk_chg);
449         if (blk_chg)
450                 write_block(block, (char *) ind);
451         return result;
452 }
453
454 #ifdef HAVE_MINIX2
455 static int map_block2(struct minix2_inode *inode, unsigned int blknr)
456 {
457         unsigned int ind[BLOCK_SIZE >> 2];
458         unsigned int dind[BLOCK_SIZE >> 2];
459         unsigned int tind[BLOCK_SIZE >> 2];
460         int blk_chg, block, result;
461
462         if (blknr < 7)
463                 return check_zone_nr2(inode->i_zone + blknr, &changed);
464         blknr -= 7;
465         if (blknr < 256) {
466                 block = check_zone_nr2(inode->i_zone + 7, &changed);
467                 read_block(block, (char *) ind);
468                 blk_chg = 0;
469                 result = check_zone_nr2(blknr + ind, &blk_chg);
470                 if (blk_chg)
471                         write_block(block, (char *) ind);
472                 return result;
473         }
474         blknr -= 256;
475         if (blknr >= 256 * 256) {
476                 block = check_zone_nr2(inode->i_zone + 8, &changed);
477                 read_block(block, (char *) dind);
478                 blk_chg = 0;
479                 result = check_zone_nr2(dind + blknr / 256, &blk_chg);
480                 if (blk_chg)
481                         write_block(block, (char *) dind);
482                 block = result;
483                 read_block(block, (char *) ind);
484                 blk_chg = 0;
485                 result = check_zone_nr2(ind + blknr % 256, &blk_chg);
486                 if (blk_chg)
487                         write_block(block, (char *) ind);
488                 return result;
489         }
490         blknr -= 256 * 256;
491         block = check_zone_nr2(inode->i_zone + 9, &changed);
492         read_block(block, (char *) tind);
493         blk_chg = 0;
494         result = check_zone_nr2(tind + blknr / (256 * 256), &blk_chg);
495         if (blk_chg)
496                 write_block(block, (char *) tind);
497         block = result;
498         read_block(block, (char *) dind);
499         blk_chg = 0;
500         result = check_zone_nr2(dind + (blknr / 256) % 256, &blk_chg);
501         if (blk_chg)
502                 write_block(block, (char *) dind);
503         block = result;
504         read_block(block, (char *) ind);
505         blk_chg = 0;
506         result = check_zone_nr2(ind + blknr % 256, &blk_chg);
507         if (blk_chg)
508                 write_block(block, (char *) ind);
509         return result;
510 }
511 #endif
512
513 static void write_super_block(void)
514 {
515         /*
516          * Set the state of the filesystem based on whether or not there
517          * are uncorrected errors.  The filesystem valid flag is
518          * unconditionally set if we get this far.
519          */
520         Super.s_state |= MINIX_VALID_FS;
521         if (errors_uncorrected)
522                 Super.s_state |= MINIX_ERROR_FS;
523         else
524                 Super.s_state &= ~MINIX_ERROR_FS;
525
526         if (BLOCK_SIZE != lseek(IN, BLOCK_SIZE, SEEK_SET))
527                 die("seek failed in write_super_block");
528         if (BLOCK_SIZE != write(IN, super_block_buffer, BLOCK_SIZE))
529                 die("unable to write super-block");
530
531         return;
532 }
533
534 static void write_tables(void)
535 {
536         write_super_block();
537
538         if (IMAPS * BLOCK_SIZE != write(IN, inode_map, IMAPS * BLOCK_SIZE))
539                 die("Unable to write inode map");
540         if (ZMAPS * BLOCK_SIZE != write(IN, zone_map, ZMAPS * BLOCK_SIZE))
541                 die("Unable to write zone map");
542         if (INODE_BUFFER_SIZE != write(IN, inode_buffer, INODE_BUFFER_SIZE))
543                 die("Unable to write inodes");
544 }
545
546 static void get_dirsize(void)
547 {
548         int block;
549         char blk[BLOCK_SIZE];
550         int size;
551
552 #if HAVE_MINIX2
553         if (version2)
554                 block = Inode2[ROOT_INO].i_zone[0];
555         else
556 #endif
557                 block = Inode[ROOT_INO].i_zone[0];
558         read_block(block, blk);
559         for (size = 16; size < BLOCK_SIZE; size <<= 1) {
560                 if (strcmp(blk + size + 2, "..") == 0) {
561                         dirsize = size;
562                         namelen = size - 2;
563                         return;
564                 }
565         }
566         /* use defaults */
567 }
568
569 static void read_superblock(void)
570 {
571         if (BLOCK_SIZE != lseek(IN, BLOCK_SIZE, SEEK_SET))
572                 die("seek failed");
573         if (BLOCK_SIZE != read(IN, super_block_buffer, BLOCK_SIZE))
574                 die("unable to read super block");
575         if (MAGIC == MINIX_SUPER_MAGIC) {
576                 namelen = 14;
577                 dirsize = 16;
578                 version2 = 0;
579         } else if (MAGIC == MINIX_SUPER_MAGIC2) {
580                 namelen = 30;
581                 dirsize = 32;
582                 version2 = 0;
583 #ifdef HAVE_MINIX2
584         } else if (MAGIC == MINIX2_SUPER_MAGIC) {
585                 namelen = 14;
586                 dirsize = 16;
587                 version2 = 1;
588         } else if (MAGIC == MINIX2_SUPER_MAGIC2) {
589                 namelen = 30;
590                 dirsize = 32;
591                 version2 = 1;
592 #endif
593         } else
594                 die("bad magic number in super-block");
595         if (ZONESIZE != 0 || BLOCK_SIZE != 1024)
596                 die("Only 1k blocks/zones supported");
597         if (IMAPS * BLOCK_SIZE * 8 < INODES + 1)
598                 die("bad s_imap_blocks field in super-block");
599         if (ZMAPS * BLOCK_SIZE * 8 < ZONES - FIRSTZONE + 1)
600                 die("bad s_zmap_blocks field in super-block");
601 }
602
603 static void read_tables(void)
604 {
605         inode_map = malloc(IMAPS * BLOCK_SIZE);
606         if (!inode_map)
607                 die("Unable to allocate buffer for inode map");
608         zone_map = malloc(ZMAPS * BLOCK_SIZE);
609         if (!inode_map)
610                 die("Unable to allocate buffer for zone map");
611         memset(inode_map, 0, sizeof(inode_map));
612         memset(zone_map, 0, sizeof(zone_map));
613         inode_buffer = malloc(INODE_BUFFER_SIZE);
614         if (!inode_buffer)
615                 die("Unable to allocate buffer for inodes");
616         inode_count = malloc(INODES + 1);
617         if (!inode_count)
618                 die("Unable to allocate buffer for inode count");
619         zone_count = malloc(ZONES);
620         if (!zone_count)
621                 die("Unable to allocate buffer for zone count");
622         if (IMAPS * BLOCK_SIZE != read(IN, inode_map, IMAPS * BLOCK_SIZE))
623                 die("Unable to read inode map");
624         if (ZMAPS * BLOCK_SIZE != read(IN, zone_map, ZMAPS * BLOCK_SIZE))
625                 die("Unable to read zone map");
626         if (INODE_BUFFER_SIZE != read(IN, inode_buffer, INODE_BUFFER_SIZE))
627                 die("Unable to read inodes");
628         if (NORM_FIRSTZONE != FIRSTZONE) {
629                 printf("Warning: Firstzone != Norm_firstzone\n");
630                 errors_uncorrected = 1;
631         }
632         get_dirsize();
633         if (show) {
634                 printf("%ld inodes\n", INODES);
635                 printf("%ld blocks\n", ZONES);
636                 printf("Firstdatazone=%ld (%ld)\n", FIRSTZONE, NORM_FIRSTZONE);
637                 printf("Zonesize=%d\n", BLOCK_SIZE << ZONESIZE);
638                 printf("Maxsize=%ld\n", MAXSIZE);
639                 printf("Filesystem state=%d\n", Super.s_state);
640                 printf("namelen=%d\n\n", namelen);
641         }
642 }
643
644 struct minix_inode *get_inode(unsigned int nr)
645 {
646         struct minix_inode *inode;
647
648         if (!nr || nr > INODES)
649                 return NULL;
650         total++;
651         inode = Inode + nr;
652         if (!inode_count[nr]) {
653                 if (!inode_in_use(nr)) {
654                         printf("Inode %d marked not used, but used for file '", nr);
655                         print_current_name();
656                         printf("'\n");
657                         if (repair) {
658                                 if (ask("Mark in use", 1))
659                                         mark_inode(nr);
660                         } else {
661                                 errors_uncorrected = 1;
662                         }
663                 }
664                 if (S_ISDIR(inode->i_mode))
665                         directory++;
666                 else if (S_ISREG(inode->i_mode))
667                         regular++;
668                 else if (S_ISCHR(inode->i_mode))
669                         chardev++;
670                 else if (S_ISBLK(inode->i_mode))
671                         blockdev++;
672                 else if (S_ISLNK(inode->i_mode))
673                         symlinks++;
674                 else if (S_ISSOCK(inode->i_mode));
675                 else if (S_ISFIFO(inode->i_mode));
676                 else {
677                         print_current_name();
678                         printf(" has mode %05o\n", inode->i_mode);
679                 }
680
681         } else
682                 links++;
683         if (!++inode_count[nr]) {
684                 printf("Warning: inode count too big.\n");
685                 inode_count[nr]--;
686                 errors_uncorrected = 1;
687         }
688         return inode;
689 }
690
691 #ifdef HAVE_MINIX2
692 struct minix2_inode *get_inode2(unsigned int nr)
693 {
694         struct minix2_inode *inode;
695
696         if (!nr || nr > INODES)
697                 return NULL;
698         total++;
699         inode = Inode2 + nr;
700         if (!inode_count[nr]) {
701                 if (!inode_in_use(nr)) {
702                         printf("Inode %d marked not used, but used for file '", nr);
703                         print_current_name();
704                         printf("'\n");
705                         if (repair) {
706                                 if (ask("Mark in use", 1))
707                                         mark_inode(nr);
708                                 else
709                                         errors_uncorrected = 1;
710                         }
711                 }
712                 if (S_ISDIR(inode->i_mode))
713                         directory++;
714                 else if (S_ISREG(inode->i_mode))
715                         regular++;
716                 else if (S_ISCHR(inode->i_mode))
717                         chardev++;
718                 else if (S_ISBLK(inode->i_mode))
719                         blockdev++;
720                 else if (S_ISLNK(inode->i_mode))
721                         symlinks++;
722                 else if (S_ISSOCK(inode->i_mode));
723                 else if (S_ISFIFO(inode->i_mode));
724                 else {
725                         print_current_name();
726                         printf(" has mode %05o\n", inode->i_mode);
727                 }
728         } else
729                 links++;
730         if (!++inode_count[nr]) {
731                 printf("Warning: inode count too big.\n");
732                 inode_count[nr]--;
733                 errors_uncorrected = 1;
734         }
735         return inode;
736 }
737 #endif
738
739 static void check_root(void)
740 {
741         struct minix_inode *inode = Inode + ROOT_INO;
742
743         if (!inode || !S_ISDIR(inode->i_mode))
744                 die("root inode isn't a directory");
745 }
746
747 #ifdef HAVE_MINIX2
748 static void check_root2(void)
749 {
750         struct minix2_inode *inode = Inode2 + ROOT_INO;
751
752         if (!inode || !S_ISDIR(inode->i_mode))
753                 die("root inode isn't a directory");
754 }
755 #endif
756
757 static int add_zone(unsigned short *znr, int *corrected)
758 {
759         int result;
760         int block;
761
762         result = 0;
763         block = check_zone_nr(znr, corrected);
764         if (!block)
765                 return 0;
766         if (zone_count[block]) {
767                 printf("Block has been used before. Now in file `");
768                 print_current_name();
769                 printf("'.");
770                 if (ask("Clear", 1)) {
771                         *znr = 0;
772                         block = 0;
773                         *corrected = 1;
774                 }
775         }
776         if (!block)
777                 return 0;
778         if (!zone_in_use(block)) {
779                 printf("Block %d in file `", block);
780                 print_current_name();
781                 printf("' is marked not in use.");
782                 if (ask("Correct", 1))
783                         mark_zone(block);
784         }
785         if (!++zone_count[block])
786                 zone_count[block]--;
787         return block;
788 }
789
790 #ifdef HAVE_MINIX2
791 static int add_zone2(unsigned int *znr, int *corrected)
792 {
793         int result;
794         int block;
795
796         result = 0;
797         block = check_zone_nr2(znr, corrected);
798         if (!block)
799                 return 0;
800         if (zone_count[block]) {
801                 printf("Block has been used before. Now in file `");
802                 print_current_name();
803                 printf("'.");
804                 if (ask("Clear", 1)) {
805                         *znr = 0;
806                         block = 0;
807                         *corrected = 1;
808                 }
809         }
810         if (!block)
811                 return 0;
812         if (!zone_in_use(block)) {
813                 printf("Block %d in file `", block);
814                 print_current_name();
815                 printf("' is marked not in use.");
816                 if (ask("Correct", 1))
817                         mark_zone(block);
818         }
819         if (!++zone_count[block])
820                 zone_count[block]--;
821         return block;
822 }
823 #endif
824
825 static void add_zone_ind(unsigned short *znr, int *corrected)
826 {
827         static char blk[BLOCK_SIZE];
828         int i, chg_blk = 0;
829         int block;
830
831         block = add_zone(znr, corrected);
832         if (!block)
833                 return;
834         read_block(block, blk);
835         for (i = 0; i < (BLOCK_SIZE >> 1); i++)
836                 add_zone(i + (unsigned short *) blk, &chg_blk);
837         if (chg_blk)
838                 write_block(block, blk);
839 }
840
841 #ifdef HAVE_MINIX2
842 static void add_zone_ind2(unsigned int *znr, int *corrected)
843 {
844         static char blk[BLOCK_SIZE];
845         int i, chg_blk = 0;
846         int block;
847
848         block = add_zone2(znr, corrected);
849         if (!block)
850                 return;
851         read_block(block, blk);
852         for (i = 0; i < BLOCK_SIZE >> 2; i++)
853                 add_zone2(i + (unsigned int *) blk, &chg_blk);
854         if (chg_blk)
855                 write_block(block, blk);
856 }
857 #endif
858
859 static void add_zone_dind(unsigned short *znr, int *corrected)
860 {
861         static char blk[BLOCK_SIZE];
862         int i, blk_chg = 0;
863         int block;
864
865         block = add_zone(znr, corrected);
866         if (!block)
867                 return;
868         read_block(block, blk);
869         for (i = 0; i < (BLOCK_SIZE >> 1); i++)
870                 add_zone_ind(i + (unsigned short *) blk, &blk_chg);
871         if (blk_chg)
872                 write_block(block, blk);
873 }
874
875 #ifdef HAVE_MINIX2
876 static void add_zone_dind2(unsigned int *znr, int *corrected)
877 {
878         static char blk[BLOCK_SIZE];
879         int i, blk_chg = 0;
880         int block;
881
882         block = add_zone2(znr, corrected);
883         if (!block)
884                 return;
885         read_block(block, blk);
886         for (i = 0; i < BLOCK_SIZE >> 2; i++)
887                 add_zone_ind2(i + (unsigned int *) blk, &blk_chg);
888         if (blk_chg)
889                 write_block(block, blk);
890 }
891
892 static void add_zone_tind2(unsigned int *znr, int *corrected)
893 {
894         static char blk[BLOCK_SIZE];
895         int i, blk_chg = 0;
896         int block;
897
898         block = add_zone2(znr, corrected);
899         if (!block)
900                 return;
901         read_block(block, blk);
902         for (i = 0; i < BLOCK_SIZE >> 2; i++)
903                 add_zone_dind2(i + (unsigned int *) blk, &blk_chg);
904         if (blk_chg)
905                 write_block(block, blk);
906 }
907 #endif
908
909 static void check_zones(unsigned int i)
910 {
911         struct minix_inode *inode;
912
913         if (!i || i > INODES)
914                 return;
915         if (inode_count[i] > 1)         /* have we counted this file already? */
916                 return;
917         inode = Inode + i;
918         if (!S_ISDIR(inode->i_mode) && !S_ISREG(inode->i_mode) &&
919                 !S_ISLNK(inode->i_mode)) return;
920         for (i = 0; i < 7; i++)
921                 add_zone(i + inode->i_zone, &changed);
922         add_zone_ind(7 + inode->i_zone, &changed);
923         add_zone_dind(8 + inode->i_zone, &changed);
924 }
925
926 #ifdef HAVE_MINIX2
927 static void check_zones2(unsigned int i)
928 {
929         struct minix2_inode *inode;
930
931         if (!i || i > INODES)
932                 return;
933         if (inode_count[i] > 1)         /* have we counted this file already? */
934                 return;
935         inode = Inode2 + i;
936         if (!S_ISDIR(inode->i_mode) && !S_ISREG(inode->i_mode)
937                 && !S_ISLNK(inode->i_mode))
938                 return;
939         for (i = 0; i < 7; i++)
940                 add_zone2(i + inode->i_zone, &changed);
941         add_zone_ind2(7 + inode->i_zone, &changed);
942         add_zone_dind2(8 + inode->i_zone, &changed);
943         add_zone_tind2(9 + inode->i_zone, &changed);
944 }
945 #endif
946
947 static void check_file(struct minix_inode *dir, unsigned int offset)
948 {
949         static char blk[BLOCK_SIZE];
950         struct minix_inode *inode;
951         int ino;
952         char *name;
953         int block;
954
955         block = map_block(dir, offset / BLOCK_SIZE);
956         read_block(block, blk);
957         name = blk + (offset % BLOCK_SIZE) + 2;
958         ino = *(unsigned short *) (name - 2);
959         if (ino > INODES) {
960                 print_current_name();
961                 printf(" contains a bad inode number for file '");
962                 printf("%.*s'.", namelen, name);
963                 if (ask(" Remove", 1)) {
964                         *(unsigned short *) (name - 2) = 0;
965                         write_block(block, blk);
966                 }
967                 ino = 0;
968         }
969         if (name_depth < MAX_DEPTH)
970                 strncpy(name_list[name_depth], name, namelen);
971         name_depth++;
972         inode = get_inode(ino);
973         name_depth--;
974         if (!offset) {
975                 if (!inode || strcmp(".", name)) {
976                         print_current_name();
977                         printf(": bad directory: '.' isn't first\n");
978                         errors_uncorrected = 1;
979                 } else
980                         return;
981         }
982         if (offset == dirsize) {
983                 if (!inode || strcmp("..", name)) {
984                         print_current_name();
985                         printf(": bad directory: '..' isn't second\n");
986                         errors_uncorrected = 1;
987                 } else
988                         return;
989         }
990         if (!inode)
991                 return;
992         if (name_depth < MAX_DEPTH)
993                 strncpy(name_list[name_depth], name, namelen);
994         name_depth++;
995         if (list) {
996                 if (verbose)
997                         printf("%6d %07o %3d ", ino, inode->i_mode, inode->i_nlinks);
998                 print_current_name();
999                 if (S_ISDIR(inode->i_mode))
1000                         printf(":\n");
1001                 else
1002                         printf("\n");
1003         }
1004         check_zones(ino);
1005         if (inode && S_ISDIR(inode->i_mode))
1006                 recursive_check(ino);
1007         name_depth--;
1008         return;
1009 }
1010
1011 #ifdef HAVE_MINIX2
1012 static void check_file2(struct minix2_inode *dir, unsigned int offset)
1013 {
1014         static char blk[BLOCK_SIZE];
1015         struct minix2_inode *inode;
1016         int ino;
1017         char *name;
1018         int block;
1019
1020         block = map_block2(dir, offset / BLOCK_SIZE);
1021         read_block(block, blk);
1022         name = blk + (offset % BLOCK_SIZE) + 2;
1023         ino = *(unsigned short *) (name - 2);
1024         if (ino > INODES) {
1025                 print_current_name();
1026                 printf(" contains a bad inode number for file '");
1027                 printf("%.*s'.", namelen, name);
1028                 if (ask(" Remove", 1)) {
1029                         *(unsigned short *) (name - 2) = 0;
1030                         write_block(block, blk);
1031                 }
1032                 ino = 0;
1033         }
1034         if (name_depth < MAX_DEPTH)
1035                 strncpy(name_list[name_depth], name, namelen);
1036         name_depth++;
1037         inode = get_inode2(ino);
1038         name_depth--;
1039         if (!offset) {
1040                 if (!inode || strcmp(".", name)) {
1041                         print_current_name();
1042                         printf(": bad directory: '.' isn't first\n");
1043                         errors_uncorrected = 1;
1044                 } else
1045                         return;
1046         }
1047         if (offset == dirsize) {
1048                 if (!inode || strcmp("..", name)) {
1049                         print_current_name();
1050                         printf(": bad directory: '..' isn't second\n");
1051                         errors_uncorrected = 1;
1052                 } else
1053                         return;
1054         }
1055         if (!inode)
1056                 return;
1057         name_depth++;
1058         if (list) {
1059                 if (verbose)
1060                         printf("%6d %07o %3d ", ino, inode->i_mode, inode->i_nlinks);
1061                 print_current_name();
1062                 if (S_ISDIR(inode->i_mode))
1063                         printf(":\n");
1064                 else
1065                         printf("\n");
1066         }
1067         check_zones2(ino);
1068         if (inode && S_ISDIR(inode->i_mode))
1069                 recursive_check2(ino);
1070         name_depth--;
1071         return;
1072 }
1073 #endif
1074
1075 static void recursive_check(unsigned int ino)
1076 {
1077         struct minix_inode *dir;
1078         unsigned int offset;
1079
1080         dir = Inode + ino;
1081         if (!S_ISDIR(dir->i_mode))
1082                 die("internal error");
1083         if (dir->i_size < 2 * dirsize) {
1084                 print_current_name();
1085                 printf(": bad directory: size<32");
1086                 errors_uncorrected = 1;
1087         }
1088         for (offset = 0; offset < dir->i_size; offset += dirsize)
1089                 check_file(dir, offset);
1090 }
1091
1092 #ifdef HAVE_MINIX2
1093 static void recursive_check2(unsigned int ino)
1094 {
1095         struct minix2_inode *dir;
1096         unsigned int offset;
1097
1098         dir = Inode2 + ino;
1099         if (!S_ISDIR(dir->i_mode))
1100                 die("internal error");
1101         if (dir->i_size < 2 * dirsize) {
1102                 print_current_name();
1103                 printf(": bad directory: size < 32");
1104                 errors_uncorrected = 1;
1105         }
1106         for (offset = 0; offset < dir->i_size; offset += dirsize)
1107                 check_file2(dir, offset);
1108 }
1109 #endif
1110
1111 static int bad_zone(int i)
1112 {
1113         char buffer[1024];
1114
1115         if (BLOCK_SIZE * i != lseek(IN, BLOCK_SIZE * i, SEEK_SET))
1116                 die("seek failed in bad_zone");
1117         return (BLOCK_SIZE != read(IN, buffer, BLOCK_SIZE));
1118 }
1119
1120 static void check_counts(void)
1121 {
1122         int i;
1123
1124         for (i = 1; i <= INODES; i++) {
1125                 if (!inode_in_use(i) && Inode[i].i_mode && warn_mode) {
1126                         printf("Inode %d mode not cleared.", i);
1127                         if (ask("Clear", 1)) {
1128                                 Inode[i].i_mode = 0;
1129                                 changed = 1;
1130                         }
1131                 }
1132                 if (!inode_count[i]) {
1133                         if (!inode_in_use(i))
1134                                 continue;
1135                         printf("Inode %d not used, marked used in the bitmap.", i);
1136                         if (ask("Clear", 1))
1137                                 unmark_inode(i);
1138                         continue;
1139                 }
1140                 if (!inode_in_use(i)) {
1141                         printf("Inode %d used, marked unused in the bitmap.", i);
1142                         if (ask("Set", 1))
1143                                 mark_inode(i);
1144                 }
1145                 if (Inode[i].i_nlinks != inode_count[i]) {
1146                         printf("Inode %d (mode = %07o), i_nlinks=%d, counted=%d.",
1147                                    i, Inode[i].i_mode, Inode[i].i_nlinks, inode_count[i]);
1148                         if (ask("Set i_nlinks to count", 1)) {
1149                                 Inode[i].i_nlinks = inode_count[i];
1150                                 changed = 1;
1151                         }
1152                 }
1153         }
1154         for (i = FIRSTZONE; i < ZONES; i++) {
1155                 if (zone_in_use(i) == zone_count[i])
1156                         continue;
1157                 if (!zone_count[i]) {
1158                         if (bad_zone(i))
1159                                 continue;
1160                         printf("Zone %d: marked in use, no file uses it.", i);
1161                         if (ask("Unmark", 1))
1162                                 unmark_zone(i);
1163                         continue;
1164                 }
1165                 printf("Zone %d: %sin use, counted=%d\n",
1166                            i, zone_in_use(i) ? "" : "not ", zone_count[i]);
1167         }
1168 }
1169
1170 #ifdef HAVE_MINIX2
1171 static void check_counts2(void)
1172 {
1173         int i;
1174
1175         for (i = 1; i <= INODES; i++) {
1176                 if (!inode_in_use(i) && Inode2[i].i_mode && warn_mode) {
1177                         printf("Inode %d mode not cleared.", i);
1178                         if (ask("Clear", 1)) {
1179                                 Inode2[i].i_mode = 0;
1180                                 changed = 1;
1181                         }
1182                 }
1183                 if (!inode_count[i]) {
1184                         if (!inode_in_use(i))
1185                                 continue;
1186                         printf("Inode %d not used, marked used in the bitmap.", i);
1187                         if (ask("Clear", 1))
1188                                 unmark_inode(i);
1189                         continue;
1190                 }
1191                 if (!inode_in_use(i)) {
1192                         printf("Inode %d used, marked unused in the bitmap.", i);
1193                         if (ask("Set", 1))
1194                                 mark_inode(i);
1195                 }
1196                 if (Inode2[i].i_nlinks != inode_count[i]) {
1197                         printf("Inode %d (mode = %07o), i_nlinks=%d, counted=%d.",
1198                                    i, Inode2[i].i_mode, Inode2[i].i_nlinks,
1199                                    inode_count[i]);
1200                         if (ask("Set i_nlinks to count", 1)) {
1201                                 Inode2[i].i_nlinks = inode_count[i];
1202                                 changed = 1;
1203                         }
1204                 }
1205         }
1206         for (i = FIRSTZONE; i < ZONES; i++) {
1207                 if (zone_in_use(i) == zone_count[i])
1208                         continue;
1209                 if (!zone_count[i]) {
1210                         if (bad_zone(i))
1211                                 continue;
1212                         printf("Zone %d: marked in use, no file uses it.", i);
1213                         if (ask("Unmark", 1))
1214                                 unmark_zone(i);
1215                         continue;
1216                 }
1217                 printf("Zone %d: %sin use, counted=%d\n",
1218                            i, zone_in_use(i) ? "" : "not ", zone_count[i]);
1219         }
1220 }
1221 #endif
1222
1223 static void check(void)
1224 {
1225         memset(inode_count, 0, (INODES + 1) * sizeof(*inode_count));
1226         memset(zone_count, 0, ZONES * sizeof(*zone_count));
1227         check_zones(ROOT_INO);
1228         recursive_check(ROOT_INO);
1229         check_counts();
1230 }
1231
1232 #ifdef HAVE_MINIX2
1233 static void check2(void)
1234 {
1235         memset(inode_count, 0, (INODES + 1) * sizeof(*inode_count));
1236         memset(zone_count, 0, ZONES * sizeof(*zone_count));
1237         check_zones2(ROOT_INO);
1238         recursive_check2(ROOT_INO);
1239         check_counts2();
1240 }
1241 #endif
1242
1243 extern int fsck_minix_main(int argc, char **argv)
1244 {
1245         struct termios tmp;
1246         int count;
1247         int retcode = 0;
1248
1249         if (argc && *argv)
1250                 program_name = *argv;
1251         if (INODE_SIZE * MINIX_INODES_PER_BLOCK != BLOCK_SIZE)
1252                 die("bad inode size");
1253 #ifdef HAVE_MINIX2
1254         if (INODE_SIZE2 * MINIX2_INODES_PER_BLOCK != BLOCK_SIZE)
1255                 die("bad v2 inode size");
1256 #endif
1257         while (argc-- > 1) {
1258                 argv++;
1259                 if (argv[0][0] != '-') {
1260                         if (device_name)
1261                                 show_usage();
1262                         else
1263                                 device_name = argv[0];
1264                 } else
1265                         while (*++argv[0])
1266                                 switch (argv[0][0]) {
1267                                 case 'l':
1268                                         list = 1;
1269                                         break;
1270                                 case 'a':
1271                                         automatic = 1;
1272                                         repair = 1;
1273                                         break;
1274                                 case 'r':
1275                                         automatic = 0;
1276                                         repair = 1;
1277                                         break;
1278                                 case 'v':
1279                                         verbose = 1;
1280                                         break;
1281                                 case 's':
1282                                         show = 1;
1283                                         break;
1284                                 case 'm':
1285                                         warn_mode = 1;
1286                                         break;
1287                                 case 'f':
1288                                         force = 1;
1289                                         break;
1290                                 default:
1291                                         show_usage();
1292                                 }
1293         }
1294         if (!device_name)
1295                 show_usage();
1296         check_mount();                          /* trying to check a mounted filesystem? */
1297         if (repair && !automatic) {
1298                 if (!isatty(0) || !isatty(1))
1299                         die("need terminal for interactive repairs");
1300         }
1301         IN = open(device_name, repair ? O_RDWR : O_RDONLY);
1302         if (IN < 0)
1303                 die("unable to open '%s'");
1304         for (count = 0; count < 3; count++)
1305                 sync();
1306         read_superblock();
1307
1308         /*
1309          * Determine whether or not we should continue with the checking.
1310          * This is based on the status of the filesystem valid and error
1311          * flags and whether or not the -f switch was specified on the 
1312          * command line.
1313          */
1314         printf("%s, %s\n", program_name, program_version);
1315         if (!(Super.s_state & MINIX_ERROR_FS) &&
1316                 (Super.s_state & MINIX_VALID_FS) && !force) {
1317                 if (repair)
1318                         printf("%s is clean, no check.\n", device_name);
1319                 return retcode;
1320         } else if (force)
1321                 printf("Forcing filesystem check on %s.\n", device_name);
1322         else if (repair)
1323                 printf("Filesystem on %s is dirty, needs checking.\n",
1324                            device_name);
1325
1326         read_tables();
1327
1328         if (repair && !automatic) {
1329                 tcgetattr(0, &termios);
1330                 tmp = termios;
1331                 tmp.c_lflag &= ~(ICANON | ECHO);
1332                 tcsetattr(0, TCSANOW, &tmp);
1333                 termios_set = 1;
1334         }
1335 #if HAVE_MINIX2
1336         if (version2) {
1337                 check_root2();
1338                 check2();
1339         } else
1340 #endif
1341         {
1342                 check_root();
1343                 check();
1344         }
1345         if (verbose) {
1346                 int i, free;
1347
1348                 for (i = 1, free = 0; i <= INODES; i++)
1349                         if (!inode_in_use(i))
1350                                 free++;
1351                 printf("\n%6ld inodes used (%ld%%)\n", (INODES - free),
1352                            100 * (INODES - free) / INODES);
1353                 for (i = FIRSTZONE, free = 0; i < ZONES; i++)
1354                         if (!zone_in_use(i))
1355                                 free++;
1356                 printf("%6ld zones used (%ld%%)\n", (ZONES - free),
1357                            100 * (ZONES - free) / ZONES);
1358                 printf("\n%6d regular files\n"
1359                            "%6d directories\n"
1360                            "%6d character device files\n"
1361                            "%6d block device files\n"
1362                            "%6d links\n"
1363                            "%6d symbolic links\n"
1364                            "------\n"
1365                            "%6d files\n",
1366                            regular, directory, chardev, blockdev,
1367                            links - 2 * directory + 1, symlinks,
1368                            total - 2 * directory + 1);
1369         }
1370         if (changed) {
1371                 write_tables();
1372                 printf("----------------------------\n"
1373                            "FILE SYSTEM HAS BEEN CHANGED\n"
1374                            "----------------------------\n");
1375                 for (count = 0; count < 3; count++)
1376                         sync();
1377         } else if (repair)
1378                 write_super_block();
1379
1380         if (repair && !automatic)
1381                 tcsetattr(0, TCSANOW, &termios);
1382
1383         if (changed)
1384                 retcode += 3;
1385         if (errors_uncorrected)
1386                 retcode += 4;
1387         return retcode;
1388 }