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