sbin/fsck_msdosfs/main.c

   1 /*
   2  * Copyright (C) 1995 Wolfgang Solfrank
   3  * Copyright (c) 1995 Martin Husemann
   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 Martin Husemann
  16  *      and Wolfgang Solfrank.
  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 AUTHORS ``AS IS'' AND ANY EXPRESS OR
  22  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  23  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  24  * IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  25  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  26  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  27  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  28  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  29  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  30  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31  *
  32  * $NetBSD: main.c,v 1.10 1997/10/01 02:18:14 enami Exp $
  33  * $FreeBSD: src/sbin/fsck_msdosfs/main.c,v 1.4.2.1 2001/08/01 05:47:56 obrien Exp $
  34  */
  35
  36 #include <stdlib.h>
  37 #include <string.h>
  38 #include <ctype.h>
  39 #include <stdio.h>
  40 #include <unistd.h>
  41 #include <errno.h>
  42 #include <stdarg.h>
  43
  44 #include "fsutil.h"
  45 #include "ext.h"
  46
  47 int alwaysno;           /* assume "no" for all questions */
  48 int alwaysyes;          /* assume "yes" for all questions */
  49 int preen;              /* set when preening */
  50 int rdonly;             /* device is opened read only (supersedes above) */
  51
  52 static void usage(void);
  53 int main(int, char **);
  54
  55 static void
  56 usage(void)
  57 {
  58         errexit("Usage: fsck_msdos [-fnpy] filesystem ... \n");
  59 }
  60
  61 int
  62 main(int argc, char **argv)
  63 {
  64         int ret = 0, erg;
  65         int ch;
  66
  67         while ((ch = getopt(argc, argv, "fFnpy")) != -1) {
  68                 switch (ch) {
  69                 case 'f':
  70                         /*
  71                          * We are always forced, since we don't
  72                          * have a clean flag
  73                          */
  74                         break;
  75                 case 'F':
  76                         /* We can never run in background */
  77                         exit(5);
  78                         break;
  79                 case 'n':
  80                         alwaysno = 1;
  81                         alwaysyes = preen = 0;
  82                         break;
  83                 case 'y':
  84                         alwaysyes = 1;
  85                         alwaysno = preen = 0;
  86                         break;
  87
  88                 case 'p':
  89                         preen = 1;
  90                         alwaysyes = alwaysno = 0;
  91                         break;
  92
  93                 default:
  94                         usage();
  95                         break;
  96                 }
  97         }
  98         argc -= optind;
  99         argv += optind;
 100
 101         if (!argc)
 102                 usage();
 103
 104         while (--argc >= 0) {
 105                 setcdevname(*argv, preen);
 106                 erg = checkfilesys(*argv++);
 107                 if (erg > ret)
 108                         ret = erg;
 109         }
 110
 111         return ret;
 112 }
 113
 114
 115 /*VARARGS*/
 116 int
 117 ask(int def, const char *fmt, ...)
 118 {
 119         va_list ap;
 120
 121         char prompt[256];
 122         int c;
 123
 124         if (preen) {
 125                 if (rdonly)
 126                         def = 0;
 127                 if (def)
 128                         printf("FIXED\n");
 129                 return def;
 130         }
 131
 132         va_start(ap, fmt);
 133         vsnprintf(prompt, sizeof(prompt), fmt, ap);
 134         if (alwaysyes || rdonly) {
 135                 printf("%s? %s\n", prompt, rdonly ? "no" : "yes");
 136                 return !rdonly;
 137         }
 138         do {
 139                 printf("%s? [yn] ", prompt);
 140                 fflush(stdout);
 141                 c = getchar();
 142                 while (c != '\n' && getchar() != '\n')
 143                         if (feof(stdin))
 144                                 return 0;
 145         } while (c != 'y' && c != 'Y' && c != 'n' && c != 'N');
 146         return c == 'y' || c == 'Y';
 147 }