args.c revision 30230
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. All advertising materials mentioning features or use of this software 18 * must display the following acknowledgement: 19 * This product includes software developed by the University of 20 * California, Berkeley and its contributors. 21 * 4. Neither the name of the University nor the names of its contributors 22 * may be used to endorse or promote products derived from this software 23 * without specific prior written permission. 24 * 25 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 26 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 27 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 28 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 29 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 30 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 31 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 32 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 33 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 34 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 35 * SUCH DAMAGE. 36 * 37 * $Id: args.c,v 1.8 1997/02/22 14:02:41 peter Exp $ 38 */ 39 40#ifndef lint 41static char const sccsid[] = "@(#)args.c 8.3 (Berkeley) 4/2/94"; 42#endif /* not lint */ 43 44#include <sys/types.h> 45 46#include <err.h> 47#include <errno.h> 48#include <limits.h> 49#include <stdio.h> 50#include <stdlib.h> 51#include <string.h> 52 53#include "dd.h" 54#include "extern.h" 55 56static int c_arg __P((const void *, const void *)); 57static int c_conv __P((const void *, const void *)); 58static void f_bs __P((char *)); 59static void f_cbs __P((char *)); 60static void f_conv __P((char *)); 61static void f_count __P((char *)); 62static void f_files __P((char *)); 63static void f_ibs __P((char *)); 64static void f_if __P((char *)); 65static void f_obs __P((char *)); 66static void f_of __P((char *)); 67static void f_seek __P((char *)); 68static void f_skip __P((char *)); 69static u_long get_bsz __P((char *)); 70 71static struct arg { 72 char *name; 73 void (*f) __P((char *)); 74 u_int set, noset; 75} args[] = { 76 { "bs", f_bs, C_BS, C_BS|C_IBS|C_OBS|C_OSYNC }, 77 { "cbs", f_cbs, C_CBS, C_CBS }, 78 { "conv", f_conv, 0, 0 }, 79 { "count", f_count, C_COUNT, C_COUNT }, 80 { "files", f_files, C_FILES, C_FILES }, 81 { "ibs", f_ibs, C_IBS, C_BS|C_IBS }, 82 { "if", f_if, C_IF, C_IF }, 83 { "obs", f_obs, C_OBS, C_BS|C_OBS }, 84 { "of", f_of, C_OF, C_OF }, 85 { "seek", f_seek, C_SEEK, C_SEEK }, 86 { "skip", f_skip, C_SKIP, C_SKIP }, 87}; 88 89static char *oper; 90 91/* 92 * args -- parse JCL syntax of dd. 93 */ 94void 95jcl(argv) 96 char **argv; 97{ 98 struct arg *ap, tmp; 99 char *arg; 100 101 in.dbsz = out.dbsz = 512; 102 103 while ((oper = *++argv) != NULL) { 104 if ((oper = strdup(oper)) == NULL) 105 errx(1, "unable to allocate space for the argument \"%s\"", *argv); 106 if ((arg = strchr(oper, '=')) == NULL) 107 errx(1, "unknown operand %s", oper); 108 *arg++ = '\0'; 109 if (!*arg) 110 errx(1, "no value specified for %s", oper); 111 tmp.name = oper; 112 if (!(ap = (struct arg *)bsearch(&tmp, args, 113 sizeof(args)/sizeof(struct arg), sizeof(struct arg), 114 c_arg))) 115 errx(1, "unknown operand %s", tmp.name); 116 if (ddflags & ap->noset) 117 errx(1, "%s: illegal argument combination or already set", 118 tmp.name); 119 ddflags |= ap->set; 120 ap->f(arg); 121 } 122 123 /* Final sanity checks. */ 124 125 if (ddflags & C_BS) { 126 /* 127 * Bs is turned off by any conversion -- we assume the user 128 * just wanted to set both the input and output block sizes 129 * and didn't want the bs semantics, so we don't warn. 130 */ 131 if (ddflags & (C_BLOCK|C_LCASE|C_SWAB|C_UCASE|C_UNBLOCK)) 132 ddflags &= ~C_BS; 133 134 /* Bs supersedes ibs and obs. */ 135 if (ddflags & C_BS && ddflags & (C_IBS|C_OBS)) 136 warnx("bs supersedes ibs and obs"); 137 } 138 139 /* 140 * Ascii/ebcdic and cbs implies block/unblock. 141 * Block/unblock requires cbs and vice-versa. 142 */ 143 if (ddflags & (C_BLOCK|C_UNBLOCK)) { 144 if (!(ddflags & C_CBS)) 145 errx(1, "record operations require cbs"); 146 if (cbsz == 0) 147 errx(1, "cbs cannot be zero"); 148 cfunc = ddflags & C_BLOCK ? block : unblock; 149 } else if (ddflags & C_CBS) { 150 if (ddflags & (C_ASCII|C_EBCDIC)) { 151 if (ddflags & C_ASCII) { 152 ddflags |= C_UNBLOCK; 153 cfunc = unblock; 154 } else { 155 ddflags |= C_BLOCK; 156 cfunc = block; 157 } 158 } else 159 errx(1, "cbs meaningless if not doing record operations"); 160 if (cbsz == 0) 161 errx(1, "cbs cannot be zero"); 162 } else 163 cfunc = def; 164 165 if (in.dbsz == 0 || out.dbsz == 0) 166 errx(1, "buffer sizes cannot be zero"); 167 168 /* 169 * Read, write and seek calls take ints as arguments. Seek sizes 170 * could be larger if we wanted to do it in stages or check only 171 * regular files, but it's probably not worth it. 172 */ 173 if (in.dbsz > INT_MAX || out.dbsz > INT_MAX) 174 errx(1, "buffer sizes cannot be greater than %d", INT_MAX); 175 if (in.offset > INT_MAX / in.dbsz || out.offset > INT_MAX / out.dbsz) 176 errx(1, "seek offsets cannot be larger than %d", INT_MAX); 177} 178 179static int 180c_arg(a, b) 181 const void *a, *b; 182{ 183 184 return (strcmp(((struct arg *)a)->name, ((struct arg *)b)->name)); 185} 186 187static void 188f_bs(arg) 189 char *arg; 190{ 191 192 in.dbsz = out.dbsz = (int)get_bsz(arg); 193} 194 195static void 196f_cbs(arg) 197 char *arg; 198{ 199 200 cbsz = (int)get_bsz(arg); 201} 202 203static void 204f_count(arg) 205 char *arg; 206{ 207 208 cpy_cnt = (u_int)get_bsz(arg); 209 if (!cpy_cnt) 210 terminate(0); 211} 212 213static void 214f_files(arg) 215 char *arg; 216{ 217 218 files_cnt = (int)get_bsz(arg); 219} 220 221static void 222f_ibs(arg) 223 char *arg; 224{ 225 226 if (!(ddflags & C_BS)) 227 in.dbsz = (int)get_bsz(arg); 228} 229 230static void 231f_if(arg) 232 char *arg; 233{ 234 235 in.name = arg; 236} 237 238static void 239f_obs(arg) 240 char *arg; 241{ 242 243 if (!(ddflags & C_BS)) 244 out.dbsz = (int)get_bsz(arg); 245} 246 247static void 248f_of(arg) 249 char *arg; 250{ 251 252 out.name = arg; 253} 254 255static void 256f_seek(arg) 257 char *arg; 258{ 259 260 out.offset = (u_int)get_bsz(arg); 261} 262 263static void 264f_skip(arg) 265 char *arg; 266{ 267 268 in.offset = (u_int)get_bsz(arg); 269} 270 271static struct conv { 272 char *name; 273 u_int set, noset; 274 u_char *ctab; 275} clist[] = { 276 { "ascii", C_ASCII, C_EBCDIC, e2a_POSIX }, 277 { "block", C_BLOCK, C_UNBLOCK, NULL }, 278 { "ebcdic", C_EBCDIC, C_ASCII, a2e_POSIX }, 279 { "ibm", C_EBCDIC, C_ASCII, a2ibm_POSIX }, 280 { "lcase", C_LCASE, C_UCASE, NULL }, 281 { "noerror", C_NOERROR, 0, NULL }, 282 { "notrunc", C_NOTRUNC, 0, NULL }, 283 { "oldascii", C_ASCII, C_EBCDIC, e2a_32V }, 284 { "oldebcdic", C_EBCDIC, C_ASCII, a2e_32V }, 285 { "oldibm", C_EBCDIC, C_ASCII, a2ibm_32V }, 286 { "osync", C_OSYNC, C_BS, NULL }, 287 { "swab", C_SWAB, 0, NULL }, 288 { "sync", C_SYNC, 0, NULL }, 289 { "ucase", C_UCASE, C_LCASE, NULL }, 290 { "unblock", C_UNBLOCK, C_BLOCK, NULL }, 291}; 292 293static void 294f_conv(arg) 295 char *arg; 296{ 297 struct conv *cp, tmp; 298 299 while (arg != NULL) { 300 tmp.name = strsep(&arg, ","); 301 if (!(cp = (struct conv *)bsearch(&tmp, clist, 302 sizeof(clist)/sizeof(struct conv), sizeof(struct conv), 303 c_conv))) 304 errx(1, "unknown conversion %s", tmp.name); 305 if (ddflags & cp->noset) 306 errx(1, "%s: illegal conversion combination", tmp.name); 307 ddflags |= cp->set; 308 if (cp->ctab) 309 ctab = cp->ctab; 310 } 311} 312 313static int 314c_conv(a, b) 315 const void *a, *b; 316{ 317 318 return (strcmp(((struct conv *)a)->name, ((struct conv *)b)->name)); 319} 320 321/* 322 * Convert an expression of the following forms to an unsigned long. 323 * 1) A positive decimal number. 324 * 2) A positive decimal number followed by a b (mult by 512). 325 * 3) A positive decimal number followed by a k (mult by 1024). 326 * 4) A positive decimal number followed by a m (mult by 512). 327 * 5) A positive decimal number followed by a w (mult by sizeof int) 328 * 6) Two or more positive decimal numbers (with/without k,b or w). 329 * seperated by x (also * for backwards compatibility), specifying 330 * the product of the indicated values. 331 */ 332static u_long 333get_bsz(val) 334 char *val; 335{ 336 u_long num, t; 337 char *expr; 338 339 num = strtoul(val, &expr, 0); 340 if (num == ULONG_MAX) /* Overflow. */ 341 err(1, "%s", oper); 342 if (expr == val) /* No digits. */ 343 errx(1, "%s: illegal numeric value", oper); 344 345 switch(*expr) { 346 case 'b': 347 t = num; 348 num *= 512; 349 if (t > num) 350 goto erange; 351 ++expr; 352 break; 353 case 'k': 354 t = num; 355 num *= 1024; 356 if (t > num) 357 goto erange; 358 ++expr; 359 break; 360 case 'm': 361 t = num; 362 num *= 1048576; 363 if (t > num) 364 goto erange; 365 ++expr; 366 break; 367 case 'w': 368 t = num; 369 num *= sizeof(int); 370 if (t > num) 371 goto erange; 372 ++expr; 373 break; 374 } 375 376 switch(*expr) { 377 case '\0': 378 break; 379 case '*': /* Backward compatible. */ 380 case 'x': 381 t = num; 382 num *= get_bsz(expr + 1); 383 if (t > num) 384erange: errx(1, "%s: %s", oper, strerror(ERANGE)); 385 break; 386 default: 387 errx(1, "%s: illegal numeric value", oper); 388 } 389 return (num); 390} 391