sbin/newfs_msdos/newfs_msdos.c

   1 /*
   2  * Copyright (c) 1998 Robert Nordier
   3  * 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
  12  *    the documentation and/or other materials provided with the
  13  *    distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS
  16  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  17  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY
  19  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
  21  * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  22  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
  23  * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  24  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
  25  * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #ifndef lint
  29 static const char rcsid[] =
  30   "$FreeBSD$";
  31 #endif /* not lint */
  32
  33 #include <sys/param.h>
  34 #include <sys/stat.h>
  35 #include <err.h>
  36 #include <errno.h>
  37 #include <paths.h>
  38 #include <stdio.h>
  39 #include <stdlib.h>
  40 #include <string.h>
  41 #include <unistd.h>
  42
  43 #include "mkfs_msdos.h"
  44
  45 #define argto1(arg, lo, msg)  argtou(arg, lo, 0xff, msg)
  46 #define argto2(arg, lo, msg)  argtou(arg, lo, 0xffff, msg)
  47 #define argto4(arg, lo, msg)  argtou(arg, lo, 0xffffffff, msg)
  48 #define argtox(arg, lo, msg)  argtou(arg, lo, UINT_MAX, msg)
  49
  50 static u_int argtou(const char *, u_int, u_int, const char *);
  51 static off_t argtooff(const char *, const char *);
  52 static void usage(void);
  53
  54 static time_t
  55 get_tstamp(const char *b)
  56 {
  57     struct stat st;
  58     char *eb;
  59     long long l;
  60
  61     if (stat(b, &st) != -1)
  62         return (time_t)st.st_mtime;
  63
  64     errno = 0;
  65     l = strtoll(b, &eb, 0);
  66     if (b == eb || *eb || errno)
  67         errx(EXIT_FAILURE, "Can't parse timestamp '%s'", b);
  68     return (time_t)l;
  69 }
  70
  71 /*
  72  * Construct a FAT12, FAT16, or FAT32 file system.
  73  */
  74 int
  75 main(int argc, char *argv[])
  76 {
  77     static const char opts[] = "@:NB:C:F:I:L:O:S:a:b:c:e:f:h:i:k:m:n:o:r:s:T:u:";
  78     struct msdos_options o;
  79     const char *fname, *dtype;
  80     char buf[MAXPATHLEN];
  81     int ch;
  82
  83     memset(&o, 0, sizeof(o));
  84
  85     while ((ch = getopt(argc, argv, opts)) != -1)
  86         switch (ch) {
  87         case '@':
  88             o.offset = argtooff(optarg, "offset");
  89             break;
  90         case 'N':
  91             o.no_create = 1;
  92             break;
  93         case 'B':
  94             o.bootstrap = optarg;
  95             break;
  96         case 'C':
  97             o.create_size = argtooff(optarg, "create size");
  98             break;
  99         case 'F':
 100             if (strcmp(optarg, "12") &&
 101                 strcmp(optarg, "16") &&
 102                 strcmp(optarg, "32"))
 103                 errx(1, "%s: bad FAT type", optarg);
 104             o.fat_type = atoi(optarg);
 105             break;
 106         case 'I':
 107             o.volume_id = argto4(optarg, 0, "volume ID");
 108             o.volume_id_set = 1;
 109             break;
 110         case 'L':
 111             o.volume_label = optarg;
 112             break;
 113         case 'O':
 114             o.OEM_string = optarg;
 115             break;
 116         case 'S':
 117             o.bytes_per_sector = argto2(optarg, 1, "bytes/sector");
 118             break;
 119         case 'a':
 120             o.sectors_per_fat = argto4(optarg, 1, "sectors/FAT");
 121             break;
 122         case 'b':
 123             o.block_size = argtox(optarg, 1, "block size");
 124             o.sectors_per_cluster = 0;
 125             break;
 126         case 'c':
 127             o.sectors_per_cluster = argto1(optarg, 1, "sectors/cluster");
 128             o.block_size = 0;
 129             break;
 130         case 'e':
 131             o.directory_entries = argto2(optarg, 1, "directory entries");
 132             break;
 133         case 'f':
 134             o.floppy = optarg;
 135             break;
 136         case 'h':
 137             o.drive_heads = argto2(optarg, 1, "drive heads");
 138             break;
 139         case 'i':
 140             o.info_sector = argto2(optarg, 1, "info sector");
 141             break;
 142         case 'k':
 143             o.backup_sector = argto2(optarg, 1, "backup sector");
 144             break;
 145         case 'm':
 146             o.media_descriptor = argto1(optarg, 0, "media descriptor");
 147             o.media_descriptor_set = 1;
 148             break;
 149         case 'n':
 150             o.num_FAT = argto1(optarg, 1, "number of FATs");
 151             break;
 152         case 'o':
 153             o.hidden_sectors = argto4(optarg, 0, "hidden sectors");
 154             o.hidden_sectors_set = 1;
 155             break;
 156         case 'r':
 157             o.reserved_sectors = argto2(optarg, 1, "reserved sectors");
 158             break;
 159         case 's':
 160             o.size = argto4(optarg, 1, "file system size");
 161             break;
 162         case 'T':
 163             o.timestamp_set = 1;
 164             o.timestamp = get_tstamp(optarg);
 165             break;
 166         case 'u':
 167             o.sectors_per_track = argto2(optarg, 1, "sectors/track");
 168             break;
 169         default:
 170             usage();
 171         }
 172     argc -= optind;
 173     argv += optind;
 174     if (argc < 1 || argc > 2)
 175         usage();
 176     fname = *argv++;
 177     if (!o.create_size && !strchr(fname, '/')) {
 178         snprintf(buf, sizeof(buf), "%s%s", _PATH_DEV, fname);
 179         if (!(fname = strdup(buf)))
 180             err(1, NULL);
 181     }
 182     dtype = *argv;
 183     return !!mkfs_msdos(fname, dtype, &o);
 184 }
 185
 186 /*
 187  * Convert and check a numeric option argument.
 188  */
 189 static u_int
 190 argtou(const char *arg, u_int lo, u_int hi, const char *msg)
 191 {
 192     char *s;
 193     u_long x;
 194
 195     errno = 0;
 196     x = strtoul(arg, &s, 0);
 197     if (errno || !*arg || *s || x < lo || x > hi)
 198         errx(1, "%s: bad %s", arg, msg);
 199     return x;
 200 }
 201
 202 /*
 203  * Same for off_t, with optional skmgpP suffix
 204  */
 205 static off_t
 206 argtooff(const char *arg, const char *msg)
 207 {
 208     char *s;
 209     off_t x;
 210
 211     errno = 0;
 212     x = strtoll(arg, &s, 0);
 213     /* allow at most one extra char */
 214     if (errno || x < 0 || (s[0] && s[1]) )
 215         errx(1, "%s: bad %s", arg, msg);
 216     if (*s) {   /* the extra char is the multiplier */
 217         switch (*s) {
 218         default:
 219             errx(1, "%s: bad %s", arg, msg);
 220             /* notreached */
 221
 222         case 's':               /* sector */
 223         case 'S':
 224             x <<= 9;            /* times 512 */
 225             break;
 226
 227         case 'k':               /* kilobyte */
 228         case 'K':
 229             x <<= 10;           /* times 1024 */
 230             break;
 231
 232         case 'm':               /* megabyte */
 233         case 'M':
 234             x <<= 20;           /* times 1024*1024 */
 235             break;
 236
 237         case 'g':               /* gigabyte */
 238         case 'G':
 239             x <<= 30;           /* times 1024*1024*1024 */
 240             break;
 241
 242         case 'p':               /* partition start */
 243         case 'P':
 244         case 'l':               /* partition length */
 245         case 'L':
 246             errx(1, "%s: not supported yet %s", arg, msg);
 247             /* notreached */
 248         }
 249     }
 250     return x;
 251 }
 252
 253 /*
 254  * Print usage message.
 255  */
 256 static void
 257 usage(void)
 258 {
 259     fprintf(stderr,
 260             "usage: %s [ -options ] special [disktype]\n", getprogname());
 261     fprintf(stderr, "where the options are:\n");
 262 static struct {
 263     char o;
 264     const char *h;
 265 } opts[] = {
 266 #define AOPT(_opt, _type, _name, _min, _desc) { _opt, _desc },
 267 ALLOPTS
 268 #undef AOPT
 269     };
 270     for (size_t i = 0; i < nitems(opts); i++)
 271         fprintf(stderr, "\t-%c %s\n", opts[i].o, opts[i].h);
 272     exit(1);
 273 }