Merge branch 'vendor/MPFR'
[dragonfly.git] / bin / dd / args.c
1 /*-
2  * Copyright (c) 1991, 1993, 1994
3  *      The Regents of the University of California.  All rights reserved.
4  *
5  * This code is derived from software contributed to Berkeley by
6  * Keith Muller of the University of California, San Diego and Lance
7  * Visser of Convex Computer Corporation.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  * 2. Redistributions in binary form must reproduce the above copyright
15  *    notice, this list of conditions and the following disclaimer in the
16  *    documentation and/or other materials provided with the distribution.
17  * 3. 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 REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * @(#)args.c   8.3 (Berkeley) 4/2/94
34  * $FreeBSD: src/bin/dd/args.c,v 1.25.2.2 2001/01/23 14:20:03 asmodai Exp $
35  * $DragonFly: src/bin/dd/args.c,v 1.6 2008/01/28 16:08:02 matthias Exp $
36  */
37
38 #include <sys/types.h>
39
40 #include <err.h>
41 #include <errno.h>
42 #include <limits.h>
43 #include <stdlib.h>
44 #include <string.h>
45
46 #include "dd.h"
47 #include "extern.h"
48
49 static int      c_arg (const void *, const void *);
50 static int      c_conv (const void *, const void *);
51 static void     f_bs (char *);
52 static void     f_cbs (char *);
53 static void     f_conv (char *);
54 static void     f_count (char *);
55 static void     f_files (char *);
56 static void     f_ibs (char *);
57 static void     f_if (char *);
58 static void     f_obs(char *);
59 static void     f_of (char *);
60 static void     f_seek (char *);
61 static void     f_skip (char *);
62 static quad_t   get_num (char *);
63 static off_t    get_offset (char *);
64
65 static const struct arg {
66         const char *name;
67         void (*f) (char *);
68         u_int set, noset;
69 } args[] = {
70         { "bs",         f_bs,           C_BS,    C_BS|C_IBS|C_OBS|C_OSYNC },
71         { "cbs",        f_cbs,          C_CBS,   C_CBS },
72         { "conv",       f_conv,         0,       0 },
73         { "count",      f_count,        C_COUNT, C_COUNT },
74         { "files",      f_files,        C_FILES, C_FILES },
75         { "ibs",        f_ibs,          C_IBS,   C_BS|C_IBS },
76         { "if",         f_if,           C_IF,    C_IF },
77         { "iseek",      f_skip,         C_SKIP,  C_SKIP },
78         { "obs",        f_obs,          C_OBS,   C_BS|C_OBS },
79         { "of",         f_of,           C_OF,    C_OF },
80         { "oseek",      f_seek,         C_SEEK,  C_SEEK },
81         { "seek",       f_seek,         C_SEEK,  C_SEEK },
82         { "skip",       f_skip,         C_SKIP,  C_SKIP },
83 };
84
85 static char *oper;
86
87 /*
88  * args -- parse JCL syntax of dd.
89  */
90 void
91 jcl(char **argv)
92 {
93         struct arg *ap, tmp;
94         char *arg;
95
96         in.dbsz = out.dbsz = 512;
97
98         while ((oper = *++argv) != NULL) {
99                 if ((oper = strdup(oper)) == NULL)
100                         errx(1, "unable to allocate space for the argument \"%s\"", *argv);
101                 if ((arg = strchr(oper, '=')) == NULL)
102                         errx(1, "unknown operand %s", oper);
103                 *arg++ = '\0';
104                 if (!*arg)
105                         errx(1, "no value specified for %s", oper);
106                 tmp.name = oper;
107                 if (!(ap = (struct arg *)bsearch(&tmp, args,
108                     sizeof(args)/sizeof(struct arg), sizeof(struct arg),
109                     c_arg)))
110                         errx(1, "unknown operand %s", tmp.name);
111                 if (ddflags & ap->noset)
112                         errx(1, "%s: illegal argument combination or already set",
113                             tmp.name);
114                 ddflags |= ap->set;
115                 ap->f(arg);
116         }
117
118         /* Final sanity checks. */
119
120         if (ddflags & C_BS) {
121                 /*
122                  * Bs is turned off by any conversion -- we assume the user
123                  * just wanted to set both the input and output block sizes
124                  * and didn't want the bs semantics, so we don't warn.
125                  */
126                 if (ddflags & (C_BLOCK | C_LCASE | C_SWAB | C_UCASE |
127                     C_UNBLOCK))
128                         ddflags &= ~C_BS;
129
130                 /* Bs supersedes ibs and obs. */
131                 if (ddflags & C_BS && ddflags & (C_IBS | C_OBS))
132                         warnx("bs supersedes ibs and obs");
133         }
134
135         /*
136          * Ascii/ebcdic and cbs implies block/unblock.
137          * Block/unblock requires cbs and vice-versa.
138          */
139         if (ddflags & (C_BLOCK | C_UNBLOCK)) {
140                 if (!(ddflags & C_CBS))
141                         errx(1, "record operations require cbs");
142                 if (cbsz == 0)
143                         errx(1, "cbs cannot be zero");
144                 cfunc = ddflags & C_BLOCK ? block : unblock;
145         } else if (ddflags & C_CBS) {
146                 if (ddflags & (C_ASCII | C_EBCDIC)) {
147                         if (ddflags & C_ASCII) {
148                                 ddflags |= C_UNBLOCK;
149                                 cfunc = unblock;
150                         } else {
151                                 ddflags |= C_BLOCK;
152                                 cfunc = block;
153                         }
154                 } else
155                         errx(1, "cbs meaningless if not doing record operations");
156         } else
157                 cfunc = def;
158
159         /*
160          * Bail out if the calculation of a file offset would overflow.
161          */
162         if (in.offset > QUAD_MAX / (ssize_t)in.dbsz ||
163             out.offset > QUAD_MAX / (ssize_t)out.dbsz) {
164                 errx(1, "seek offsets cannot be larger than %lld", QUAD_MAX);
165         }
166 }
167
168 static int
169 c_arg(const void *a, const void *b)
170 {
171
172         return (strcmp(((const struct arg *)a)->name,
173             ((const struct arg *)b)->name));
174 }
175
176 static void
177 f_bs(char *arg)
178 {
179         quad_t res;
180
181         res = get_num(arg);
182         if (res < 1 || res > SSIZE_MAX)
183                 errx(1, "bs must be between 1 and %zd", SSIZE_MAX);
184         in.dbsz = out.dbsz = (size_t)res;
185 }
186
187 static void
188 f_cbs(char *arg)
189 {
190         quad_t res;
191
192         res = get_num(arg);
193         if (res < 1 || res > SSIZE_MAX)
194                 errx(1, "cbs must be between 1 and %zd", SSIZE_MAX);
195         cbsz = (size_t)res;
196 }
197
198 static void
199 f_count(char *arg)
200 {
201
202         cpy_cnt = get_num(arg);
203         if (cpy_cnt < 0)
204                 errx(1, "count cannot be negative");
205         if (cpy_cnt == 0)
206                 cpy_cnt = -1;
207 }
208
209 static void
210 f_files(char *arg)
211 {
212
213         files_cnt = get_num(arg);
214         if (files_cnt < 1)
215                 errx(1, "files must be between 1 and %qd", QUAD_MAX);
216 }
217
218 static void
219 f_ibs(char *arg)
220 {
221         quad_t res;
222
223         if (!(ddflags & C_BS)) {
224                 res = get_num(arg);
225                 if (res < 1 || res > SSIZE_MAX)
226                         errx(1, "ibs must be between 1 and %zd", SSIZE_MAX);
227                 in.dbsz = (size_t)res;
228         }
229 }
230
231 static void
232 f_if(char *arg)
233 {
234
235         in.name = arg;
236 }
237
238 static void
239 f_obs(char *arg)
240 {
241         quad_t res;
242
243         if (!(ddflags & C_BS)) {
244                 res = get_num(arg);
245                 if (res < 1 || res > SSIZE_MAX)
246                         errx(1, "obs must be between 1 and %zd", SSIZE_MAX);
247                 out.dbsz = (size_t)res;
248         }
249 }
250
251 static void
252 f_of(char *arg)
253 {
254
255         out.name = arg;
256 }
257
258 static void
259 f_seek(char *arg)
260 {
261
262         out.offset = get_offset(arg);
263 }
264
265 static void
266 f_skip(char *arg)
267 {
268
269         in.offset = get_offset(arg);
270 }
271
272 static const struct conv {
273         const char *name;
274         u_int set, noset;
275         const u_char *ctab;
276 } clist[] = {
277         { "ascii",      C_ASCII,        C_EBCDIC,       e2a_POSIX },
278         { "block",      C_BLOCK,        C_UNBLOCK,      NULL },
279         { "ebcdic",     C_EBCDIC,       C_ASCII,        a2e_POSIX },
280         { "ibm",        C_EBCDIC,       C_ASCII,        a2ibm_POSIX },
281         { "lcase",      C_LCASE,        C_UCASE,        NULL },
282         { "noerror",    C_NOERROR,      0,              NULL },
283         { "notrunc",    C_NOTRUNC,      0,              NULL },
284         { "oldascii",   C_ASCII,        C_EBCDIC,       e2a_32V },
285         { "oldebcdic",  C_EBCDIC,       C_ASCII,        a2e_32V },
286         { "oldibm",     C_EBCDIC,       C_ASCII,        a2ibm_32V },
287         { "osync",      C_OSYNC,        C_BS,           NULL },
288         { "sparse",     C_SPARSE,       0,              NULL },
289         { "swab",       C_SWAB,         0,              NULL },
290         { "sync",       C_SYNC,         0,              NULL },
291         { "ucase",      C_UCASE,        C_LCASE,        NULL },
292         { "unblock",    C_UNBLOCK,      C_BLOCK,        NULL },
293 };
294
295 static void
296 f_conv(char *arg)
297 {
298         struct conv *cp, tmp;
299
300         while (arg != NULL) {
301                 tmp.name = strsep(&arg, ",");
302                 cp = bsearch(&tmp, clist, sizeof(clist) / sizeof(struct conv),
303                     sizeof(struct conv), c_conv);
304                 if (cp == NULL)
305                         errx(1, "unknown conversion %s", tmp.name);
306                 if (ddflags & cp->noset)
307                         errx(1, "%s: illegal conversion combination", tmp.name);
308                 ddflags |= cp->set;
309                 if (cp->ctab)
310                         ctab = cp->ctab;
311         }
312 }
313
314 static int
315 c_conv(const void *a, const void *b)
316 {
317
318         return (strcmp(((const struct conv *)a)->name,
319             ((const struct conv *)b)->name));
320 }
321
322 /*
323  * Convert an expression of the following forms to a quad_t.
324  *      1) A positive decimal number.
325  *      2) A positive decimal number followed by a b (mult by 512.)
326  *      3) A positive decimal number followed by a k (mult by 1 << 10.)
327  *      4) A positive decimal number followed by a m (mult by 1 << 20.)
328  *      5) A positive decimal number followed by a g (mult by 1 << 30.)
329  *      5) A positive decimal number followed by a w (mult by sizeof int.)
330  *      6) Two or more positive decimal numbers (with/without [bkmgw])
331  *         separated by x (also * for backwards compatibility), specifying
332  *         the product of the indicated values.
333  */
334 static quad_t
335 get_num(char *val)
336 {
337         quad_t num, t;
338         char *expr;
339
340         errno = 0;
341         num = strtoll(val, &expr, 0);
342         if (errno != 0)                         /* Overflow or underflow. */
343                 err(1, "%s", oper);
344         
345         if (expr == val)                        /* No valid digits. */
346                 errx(1, "%s: illegal numeric value", oper);
347
348         switch (*expr) {
349         case 'b':
350         case 'B':
351                 t = num;
352                 num *= 512;
353                 if (t > num)
354                         goto erange;
355                 ++expr;
356                 break;
357         case 'k':
358         case 'K':
359                 t = num;
360                 num *= 1 << 10;
361                 if (t > num)
362                         goto erange;
363                 ++expr;
364                 break;
365         case 'm':
366         case 'M':
367                 t = num;
368                 num *= 1 << 20;
369                 if (t > num)
370                         goto erange;
371                 ++expr;
372                 break;
373         case 'g':
374         case 'G':
375                 t = num;
376                 num *= 1 << 30;
377                 if (t > num)
378                         goto erange;
379                 ++expr;
380                 break;
381         case 'w':
382                 t = num;
383                 num *= sizeof(int);
384                 if (t > num)
385                         goto erange;
386                 ++expr;
387                 break;
388         }
389
390         switch (*expr) {
391                 case '\0':
392                         break;
393                 case '*':                       /* Backward compatible. */
394                 case 'x':
395                         t = num;
396                         num *= get_num(expr + 1);
397                         if (t <= num)
398                                 break;
399 erange:
400                         errx(1, "%s: %s", oper, strerror(ERANGE));
401                 default:
402                         errx(1, "%s: illegal numeric value", oper);
403         }
404         return (num);
405 }
406
407 static off_t
408 get_offset(char *val)
409 {
410         quad_t num;
411
412         num = get_num(val);
413         if (num > QUAD_MAX)     /* XXX can't happen && quad_t != off_t */
414                 errx(1, "%s: illegal offset", oper);    /* Too big. */
415         return ((off_t)num);
416 }