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 } 274