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. 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
  38 #ifndef lint
  39 #if 0
  40 static char sccsid[] = "@(#)args.c      8.3 (Berkeley) 4/2/94";
  41 #endif
  42 static const char rcsid[] =
  43         "$Id: args.c,v 1.12 1998/05/06 06:51:29 charnier Exp $";
  44 #endif /* not lint */
  45
  46 #include <sys/types.h>
  47
  48 #include <err.h>
  49 #include <errno.h>
  50 #include <limits.h>
  51 #include <stdlib.h>
  52 #include <string.h>
  53
  54 #include "dd.h"
  55 #include "extern.h"
  56
  57 static int      c_arg __P((const void *, const void *));
  58 static int      c_conv __P((const void *, const void *));
  59 static void     f_bs __P((char *));
  60 static void     f_cbs __P((char *));
  61 static void     f_conv __P((char *));
  62 static void     f_count __P((char *));
  63 static void     f_files __P((char *));
  64 static void     f_ibs __P((char *));
  65 static void     f_if __P((char *));
  66 static void     f_obs __P((char *));
  67 static void     f_of __P((char *));
  68 static void     f_seek __P((char *));
  69 static void     f_skip __P((char *));
  70 static u_long   get_bsz __P((char *));
  71
  72 static struct arg {
  73         char *name;
  74         void (*f) __P((char *));
  75         u_int set, noset;
  76 } args[] = {
  77         { "bs",         f_bs,           C_BS,    C_BS|C_IBS|C_OBS|C_OSYNC },
  78         { "cbs",        f_cbs,          C_CBS,   C_CBS },
  79         { "conv",       f_conv,         0,       0 },
  80         { "count",      f_count,        C_COUNT, C_COUNT },
  81         { "files",      f_files,        C_FILES, C_FILES },
  82         { "ibs",        f_ibs,          C_IBS,   C_BS|C_IBS },
  83         { "if",         f_if,           C_IF,    C_IF },
  84         { "obs",        f_obs,          C_OBS,   C_BS|C_OBS },
  85         { "of",         f_of,           C_OF,    C_OF },
  86         { "seek",       f_seek,         C_SEEK,  C_SEEK },
  87         { "skip",       f_skip,         C_SKIP,  C_SKIP },
  88 };
  89
  90 static char *oper;
  91
  92 /*
  93  * args -- parse JCL syntax of dd.
  94  */
  95 void
  96 jcl(argv)
  97         char **argv;
  98 {
  99         struct arg *ap, tmp;
 100         char *arg;
 101
 102         in.dbsz = out.dbsz = 512;
 103
 104         while ((oper = *++argv) != NULL) {
 105                 if ((oper = strdup(oper)) == NULL)
 106                         errx(1, "unable to allocate space for the argument \"%s\"", *argv);
 107                 if ((arg = strchr(oper, '=')) == NULL)
 108                         errx(1, "unknown operand %s", oper);
 109                 *arg++ = '\0';
 110                 if (!*arg)
 111                         errx(1, "no value specified for %s", oper);
 112                 tmp.name = oper;
 113                 if (!(ap = (struct arg *)bsearch(&tmp, args,
 114                     sizeof(args)/sizeof(struct arg), sizeof(struct arg),
 115                     c_arg)))
 116                         errx(1, "unknown operand %s", tmp.name);
 117                 if (ddflags & ap->noset)
 118                         errx(1, "%s: illegal argument combination or already set",
 119                             tmp.name);
 120                 ddflags |= ap->set;
 121                 ap->f(arg);
 122         }
 123
 124         /* Final sanity checks. */
 125
 126         if (ddflags & C_BS) {
 127                 /*
 128                  * Bs is turned off by any conversion -- we assume the user
 129                  * just wanted to set both the input and output block sizes
 130                  * and didn't want the bs semantics, so we don't warn.
 131                  */
 132                 if (ddflags & (C_BLOCK|C_LCASE|C_SWAB|C_UCASE|C_UNBLOCK))
 133                         ddflags &= ~C_BS;
 134
 135                 /* Bs supersedes ibs and obs. */
 136                 if (ddflags & C_BS && ddflags & (C_IBS|C_OBS))
 137                         warnx("bs supersedes ibs and obs");
 138         }
 139
 140         /*
 141          * Ascii/ebcdic and cbs implies block/unblock.
 142          * Block/unblock requires cbs and vice-versa.
 143          */
 144         if (ddflags & (C_BLOCK|C_UNBLOCK)) {
 145                 if (!(ddflags & C_CBS))
 146                         errx(1, "record operations require cbs");
 147                 if (cbsz == 0)
 148                         errx(1, "cbs cannot be zero");
 149                 cfunc = ddflags & C_BLOCK ? block : unblock;
 150         } else if (ddflags & C_CBS) {
 151                 if (ddflags & (C_ASCII|C_EBCDIC)) {
 152                         if (ddflags & C_ASCII) {
 153                                 ddflags |= C_UNBLOCK;
 154                                 cfunc = unblock;
 155                         } else {
 156                                 ddflags |= C_BLOCK;
 157                                 cfunc = block;
 158                         }
 159                 } else
 160                         errx(1, "cbs meaningless if not doing record operations");
 161                 if (cbsz == 0)
 162                         errx(1, "cbs cannot be zero");
 163         } else
 164                 cfunc = def;
 165
 166         if (in.dbsz == 0 || out.dbsz == 0)
 167                 errx(1, "buffer sizes cannot be zero");
 168
 169         /*
 170          * Read, write and seek calls take ints as arguments.  Seek sizes
 171          * could be larger if we wanted to do it in stages or check only
 172          * regular files, but it's probably not worth it.
 173          */
 174         if (in.dbsz > INT_MAX || out.dbsz > INT_MAX)
 175                 errx(1, "buffer sizes cannot be greater than %d", INT_MAX);
 176         if (in.offset > INT_MAX / in.dbsz || out.offset > INT_MAX / out.dbsz)
 177                 errx(1, "seek offsets cannot be larger than %d", INT_MAX);
 178 }
 179
 180 static int
 181 c_arg(a, b)
 182         const void *a, *b;
 183 {
 184
 185         return (strcmp(((struct arg *)a)->name, ((struct arg *)b)->name));
 186 }
 187
 188 static void
 189 f_bs(arg)
 190         char *arg;
 191 {
 192
 193         in.dbsz = out.dbsz = (int)get_bsz(arg);
 194 }
 195
 196 static void
 197 f_cbs(arg)
 198         char *arg;
 199 {
 200
 201         cbsz = (int)get_bsz(arg);
 202 }
 203
 204 static void
 205 f_count(arg)
 206         char *arg;
 207 {
 208
 209         cpy_cnt = (u_int)get_bsz(arg);
 210         if (!cpy_cnt)
 211                 terminate(0);
 212 }
 213
 214 static void
 215 f_files(arg)
 216         char *arg;
 217 {
 218
 219         files_cnt = (int)get_bsz(arg);
 220 }
 221
 222 static void
 223 f_ibs(arg)
 224         char *arg;
 225 {
 226
 227         if (!(ddflags & C_BS))
 228                 in.dbsz = (int)get_bsz(arg);
 229 }
 230
 231 static void
 232 f_if(arg)
 233         char *arg;
 234 {
 235
 236         in.name = arg;
 237 }
 238
 239 static void
 240 f_obs(arg)
 241         char *arg;
 242 {
 243
 244         if (!(ddflags & C_BS))
 245                 out.dbsz = (int)get_bsz(arg);
 246 }
 247
 248 static void
 249 f_of(arg)
 250         char *arg;
 251 {
 252
 253         out.name = arg;
 254 }
 255
 256 static void
 257 f_seek(arg)
 258         char *arg;
 259 {
 260
 261         out.offset = (u_int)get_bsz(arg);
 262 }
 263
 264 static void
 265 f_skip(arg)
 266         char *arg;
 267 {
 268
 269         in.offset = (u_int)get_bsz(arg);
 270 }
 271
 272 static struct conv {
 273         char *name;
 274         u_int set, noset;
 275         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(arg)
 297         char *arg;
 298 {
 299         struct conv *cp, tmp;
 300
 301         while (arg != NULL) {
 302                 tmp.name = strsep(&arg, ",");
 303                 if (!(cp = (struct conv *)bsearch(&tmp, clist,
 304                     sizeof(clist)/sizeof(struct conv), sizeof(struct conv),
 305                     c_conv)))
 306                         errx(1, "unknown conversion %s", tmp.name);
 307                 if (ddflags & cp->noset)
 308                         errx(1, "%s: illegal conversion combination", tmp.name);
 309                 ddflags |= cp->set;
 310                 if (cp->ctab)
 311                         ctab = cp->ctab;
 312         }
 313 }
 314
 315 static int
 316 c_conv(a, b)
 317         const void *a, *b;
 318 {
 319
 320         return (strcmp(((struct conv *)a)->name, ((struct conv *)b)->name));
 321 }
 322
 323 /*
 324  * Convert an expression of the following forms to an unsigned long.
 325  *      1) A positive decimal number.
 326  *      2) A positive decimal number followed by a b (mult by 512).
 327  *      3) A positive decimal number followed by a k (mult by 1024).
 328  *      4) A positive decimal number followed by a m (mult by 512).
 329  *      5) A positive decimal number followed by a w (mult by sizeof int)
 330  *      6) Two or more positive decimal numbers (with/without k,b or w).
 331  *         seperated by x (also * for backwards compatibility), specifying
 332  *         the product of the indicated values.
 333  */
 334 static u_long
 335 get_bsz(val)
 336         char *val;
 337 {
 338         u_long num, t;
 339         char *expr;
 340
 341         num = strtoul(val, &expr, 0);
 342         if (num == ULONG_MAX)                   /* Overflow. */
 343                 err(1, "%s", oper);
 344         if (expr == val)                        /* No digits. */
 345                 errx(1, "%s: illegal numeric value", oper);
 346
 347         switch(*expr) {
 348         case 'b':
 349                 t = num;
 350                 num *= 512;
 351                 if (t > num)
 352                         goto erange;
 353                 ++expr;
 354                 break;
 355         case 'k':
 356                 t = num;
 357                 num *= 1024;
 358                 if (t > num)
 359                         goto erange;
 360                 ++expr;
 361                 break;
 362         case 'm':
 363                 t = num;
 364                 num *= 1048576;
 365                 if (t > num)
 366                         goto erange;
 367                 ++expr;
 368                 break;
 369         case 'w':
 370                 t = num;
 371                 num *= sizeof(int);
 372                 if (t > num)
 373                         goto erange;
 374                 ++expr;
 375                 break;
 376         }
 377
 378         switch(*expr) {
 379                 case '\0':
 380                         break;
 381                 case '*':                       /* Backward compatible. */
 382                 case 'x':
 383                         t = num;
 384                         num *= get_bsz(expr + 1);
 385                         if (t > num)
 386 erange:                         errx(1, "%s: %s", oper, strerror(ERANGE));
 387                         break;
 388                 default:
 389                         errx(1, "%s: illegal numeric value", oper);
 390         }
 391         return (num);
 392 }