Merge branch 'vendor/OPENSSL'
[dragonfly.git] / bin / dd / args.c
CommitLineData
984263bc
MD
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.
1902d433 17 * 3. Neither the name of the University nor the names of its contributors
984263bc
MD
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.
1de703da
MD
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 $
1902d433 35 * $DragonFly: src/bin/dd/args.c,v 1.6 2008/01/28 16:08:02 matthias Exp $
984263bc
MD
36 */
37
984263bc
MD
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
9dbf638f
DR
49static int c_arg (const void *, const void *);
50static int c_conv (const void *, const void *);
51static void f_bs (char *);
52static void f_cbs (char *);
53static void f_conv (char *);
54static void f_count (char *);
55static void f_files (char *);
56static void f_ibs (char *);
57static void f_if (char *);
58static void f_obs(char *);
59static void f_of (char *);
60static void f_seek (char *);
61static void f_skip (char *);
62static quad_t get_num (char *);
63static off_t get_offset (char *);
984263bc
MD
64
65static const struct arg {
66 const char *name;
9dbf638f 67 void (*f) (char *);
984263bc
MD
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
85static char *oper;
86
87/*
88 * args -- parse JCL syntax of dd.
89 */
90void
b5744197 91jcl(char **argv)
984263bc
MD
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 */
a276dc6b
MD
162 if (in.offset > QUAD_MAX / (ssize_t)in.dbsz ||
163 out.offset > QUAD_MAX / (ssize_t)out.dbsz) {
a78cd756 164 errx(1, "seek offsets cannot be larger than %lld", QUAD_MAX);
a276dc6b 165 }
984263bc
MD
166}
167
168static int
b5744197 169c_arg(const void *a, const void *b)
984263bc
MD
170{
171
172 return (strcmp(((const struct arg *)a)->name,
173 ((const struct arg *)b)->name));
174}
175
176static void
b5744197 177f_bs(char *arg)
984263bc
MD
178{
179 quad_t res;
180
181 res = get_num(arg);
182 if (res < 1 || res > SSIZE_MAX)
a276dc6b 183 errx(1, "bs must be between 1 and %zd", SSIZE_MAX);
984263bc
MD
184 in.dbsz = out.dbsz = (size_t)res;
185}
186
187static void
b5744197 188f_cbs(char *arg)
984263bc
MD
189{
190 quad_t res;
191
192 res = get_num(arg);
193 if (res < 1 || res > SSIZE_MAX)
a276dc6b 194 errx(1, "cbs must be between 1 and %zd", SSIZE_MAX);
984263bc
MD
195 cbsz = (size_t)res;
196}
197
198static void
b5744197 199f_count(char *arg)
984263bc
MD
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
209static void
b5744197 210f_files(char *arg)
984263bc
MD
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
218static void
b5744197 219f_ibs(char *arg)
984263bc
MD
220{
221 quad_t res;
222
223 if (!(ddflags & C_BS)) {
224 res = get_num(arg);
225 if (res < 1 || res > SSIZE_MAX)
a276dc6b 226 errx(1, "ibs must be between 1 and %zd", SSIZE_MAX);
984263bc
MD
227 in.dbsz = (size_t)res;
228 }
229}
230
231static void
b5744197 232f_if(char *arg)
984263bc
MD
233{
234
235 in.name = arg;
236}
237
238static void
b5744197 239f_obs(char *arg)
984263bc
MD
240{
241 quad_t res;
242
243 if (!(ddflags & C_BS)) {
244 res = get_num(arg);
245 if (res < 1 || res > SSIZE_MAX)
a276dc6b 246 errx(1, "obs must be between 1 and %zd", SSIZE_MAX);
984263bc
MD
247 out.dbsz = (size_t)res;
248 }
249}
250
251static void
b5744197 252f_of(char *arg)
984263bc
MD
253{
254
255 out.name = arg;
256}
257
258static void
b5744197 259f_seek(char *arg)
984263bc
MD
260{
261
262 out.offset = get_offset(arg);
263}
264
265static void
b5744197 266f_skip(char *arg)
984263bc
MD
267{
268
269 in.offset = get_offset(arg);
270}
271
272static 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
295static void
b5744197 296f_conv(char *arg)
984263bc
MD
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
314static int
b5744197 315c_conv(const void *a, const void *b)
984263bc
MD
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 */
334static quad_t
b5744197 335get_num(char *val)
984263bc
MD
336{
337 quad_t num, t;
338 char *expr;
339
340 errno = 0;
a78cd756 341 num = strtoll(val, &expr, 0);
984263bc
MD
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':
1902d433 350 case 'B':
984263bc
MD
351 t = num;
352 num *= 512;
353 if (t > num)
354 goto erange;
355 ++expr;
356 break;
357 case 'k':
1902d433 358 case 'K':
984263bc
MD
359 t = num;
360 num *= 1 << 10;
361 if (t > num)
362 goto erange;
363 ++expr;
364 break;
365 case 'm':
1902d433 366 case 'M':
984263bc
MD
367 t = num;
368 num *= 1 << 20;
369 if (t > num)
370 goto erange;
371 ++expr;
372 break;
373 case 'g':
1902d433 374 case 'G':
984263bc
MD
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;
399erange:
400 errx(1, "%s: %s", oper, strerror(ERANGE));
401 default:
402 errx(1, "%s: illegal numeric value", oper);
403 }
404 return (num);
405}
406
407static off_t
b5744197 408get_offset(char *val)
984263bc
MD
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}