sbin/newfs_msdos/newfs_msdos.c

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