usr.bin/jot/jot.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-3-Clause
   3  *
   4  * Copyright (c) 1993
   5  *      The Regents of the University of California.  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 the
  14  *    documentation and/or other materials provided with the distribution.
  15  * 3. Neither the name of the University nor the names of its contributors
  16  *    may be used to endorse or promote products derived from this software
  17  *    without specific prior written permission.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  29  * SUCH DAMAGE.
  30  */
  31
  32 #ifndef lint
  33 static const char copyright[] =
  34 "@(#) Copyright (c) 1993\n\
  35         The Regents of the University of California.  All rights reserved.\n";
  36 #endif /* not lint */
  37
  38 #ifndef lint
  39 #if 0
  40 static char sccsid[] = "@(#)jot.c       8.1 (Berkeley) 6/6/93";
  41 #endif
  42 #endif
  43 #include <sys/cdefs.h>
  44 __FBSDID("$FreeBSD$");
  45
  46 /*
  47  * jot - print sequential or random data
  48  *
  49  * Author:  John Kunze, Office of Comp. Affairs, UCB
  50  */
  51
  52 #include <sys/capsicum.h>
  53 #include <capsicum_helpers.h>
  54 #include <ctype.h>
  55 #include <err.h>
  56 #include <errno.h>
  57 #include <limits.h>
  58 #include <stdio.h>
  59 #include <stdint.h>
  60 #include <stdlib.h>
  61 #include <stdbool.h>
  62 #include <string.h>
  63 #include <time.h>
  64 #include <unistd.h>
  65
  66 /* Defaults */
  67 #define REPS_DEF        100
  68 #define BEGIN_DEF       1
  69 #define ENDER_DEF       100
  70 #define STEP_DEF        1
  71
  72 /* Flags of options that have been set */
  73 #define HAVE_STEP       1
  74 #define HAVE_ENDER      2
  75 #define HAVE_BEGIN      4
  76 #define HAVE_REPS       8
  77
  78 #define is_default(s)   (*(s) == 0 || strcmp((s), "-") == 0)
  79
  80 static bool     boring;
  81 static int      prec = -1;
  82 static bool     longdata;
  83 static bool     intdata;
  84 static bool     chardata;
  85 static bool     nosign;
  86 static const    char *sepstring = "\n";
  87 static char     format[BUFSIZ];
  88
  89 static void     getformat(void);
  90 static int      getprec(const char *);
  91 static int      putdata(double, bool);
  92 static void     usage(void);
  93
  94 int
  95 main(int argc, char **argv)
  96 {
  97         cap_rights_t rights;
  98         bool    have_format = false;
  99         bool    infinity = false;
 100         bool    nofinalnl = false;
 101         bool    randomize = false;
 102         bool    use_random = false;
 103         int     ch;
 104         int     mask = 0;
 105         int     n = 0;
 106         double  begin = BEGIN_DEF;
 107         double  divisor;
 108         double  ender = ENDER_DEF;
 109         double  s = STEP_DEF;
 110         double  x, y;
 111         long    i;
 112         long    reps = REPS_DEF;
 113
 114         if (caph_limit_stdio() < 0)
 115                 err(1, "unable to limit rights for stdio");
 116         cap_rights_init(&rights);
 117         if (caph_rights_limit(STDIN_FILENO, &rights) < 0)
 118                 err(1, "unable to limit rights for stdin");
 119
 120         /*
 121          * Cache NLS data, for strerror, for err(3), before entering capability
 122          * mode.
 123          */
 124         caph_cache_catpages();
 125
 126         if (caph_enter() < 0)
 127                 err(1, "unable to enter capability mode");
 128
 129         while ((ch = getopt(argc, argv, "b:cnp:rs:w:")) != -1)
 130                 switch (ch) {
 131                 case 'b':
 132                         boring = true;
 133                         /* FALLTHROUGH */
 134                 case 'w':
 135                         if (strlcpy(format, optarg, sizeof(format)) >=
 136                             sizeof(format))
 137                                 errx(1, "-%c word too long", ch);
 138                         have_format = true;
 139                         break;
 140                 case 'c':
 141                         chardata = true;
 142                         break;
 143                 case 'n':
 144                         nofinalnl = true;
 145                         break;
 146                 case 'p':
 147                         prec = atoi(optarg);
 148                         if (prec < 0)
 149                                 errx(1, "bad precision value");
 150                         have_format = true;
 151                         break;
 152                 case 'r':
 153                         randomize = true;
 154                         break;
 155                 case 's':
 156                         sepstring = optarg;
 157                         break;
 158                 default:
 159                         usage();
 160                 }
 161         argc -= optind;
 162         argv += optind;
 163
 164         switch (argc) { /* examine args right to left, falling thru cases */
 165         case 4:
 166                 if (!is_default(argv[3])) {
 167                         if (!sscanf(argv[3], "%lf", &s))
 168                                 errx(1, "bad s value: %s", argv[3]);
 169                         mask |= HAVE_STEP;
 170                         if (randomize)
 171                                 use_random = true;
 172                 }
 173                 /* FALLTHROUGH */
 174         case 3:
 175                 if (!is_default(argv[2])) {
 176                         if (!sscanf(argv[2], "%lf", &ender))
 177                                 ender = argv[2][strlen(argv[2])-1];
 178                         mask |= HAVE_ENDER;
 179                         if (prec < 0)
 180                                 n = getprec(argv[2]);
 181                 }
 182                 /* FALLTHROUGH */
 183         case 2:
 184                 if (!is_default(argv[1])) {
 185                         if (!sscanf(argv[1], "%lf", &begin))
 186                                 begin = argv[1][strlen(argv[1])-1];
 187                         mask |= HAVE_BEGIN;
 188                         if (prec < 0)
 189                                 prec = getprec(argv[1]);
 190                         if (n > prec)           /* maximum precision */
 191                                 prec = n;
 192                 }
 193                 /* FALLTHROUGH */
 194         case 1:
 195                 if (!is_default(argv[0])) {
 196                         if (!sscanf(argv[0], "%ld", &reps))
 197                                 errx(1, "bad reps value: %s", argv[0]);
 198                         mask |= HAVE_REPS;
 199                 }
 200                 break;
 201         case 0:
 202                 usage();
 203         default:
 204                 errx(1, "too many arguments.  What do you mean by %s?",
 205                     argv[4]);
 206         }
 207         getformat();
 208
 209         if (prec == -1)
 210                 prec = 0;
 211
 212         while (mask)    /* 4 bit mask has 1's where last 4 args were given */
 213                 switch (mask) { /* fill in the 0's by default or computation */
 214                 case HAVE_STEP:
 215                 case HAVE_ENDER:
 216                 case HAVE_ENDER | HAVE_STEP:
 217                 case HAVE_BEGIN:
 218                 case HAVE_BEGIN | HAVE_STEP:
 219                         reps = REPS_DEF;
 220                         mask |= HAVE_REPS;
 221                         break;
 222                 case HAVE_BEGIN | HAVE_ENDER:
 223                         s = ender > begin ? 1 : -1;
 224                         mask |= HAVE_STEP;
 225                         break;
 226                 case HAVE_BEGIN | HAVE_ENDER | HAVE_STEP:
 227                         if (randomize)
 228                                 reps = REPS_DEF;
 229                         else if (s == 0.0)
 230                                 reps = 0;
 231                         else
 232                                 reps = (ender - begin + s) / s;
 233                         if (reps <= 0)
 234                                 errx(1, "impossible stepsize");
 235                         mask = 0;
 236                         break;
 237                 case HAVE_REPS:
 238                 case HAVE_REPS | HAVE_STEP:
 239                         begin = BEGIN_DEF;
 240                         mask |= HAVE_BEGIN;
 241                         break;
 242                 case HAVE_REPS | HAVE_ENDER:
 243                         s = STEP_DEF;
 244                         mask = HAVE_REPS | HAVE_ENDER | HAVE_STEP;
 245                         break;
 246                 case HAVE_REPS | HAVE_ENDER | HAVE_STEP:
 247                         if (randomize)
 248                                 begin = BEGIN_DEF;
 249                         else if (reps == 0)
 250                                 errx(1, "must specify begin if reps == 0");
 251                         begin = ender - reps * s + s;
 252                         mask = 0;
 253                         break;
 254                 case HAVE_REPS | HAVE_BEGIN:
 255                         s = STEP_DEF;
 256                         mask = HAVE_REPS | HAVE_BEGIN | HAVE_STEP;
 257                         break;
 258                 case HAVE_REPS | HAVE_BEGIN | HAVE_STEP:
 259                         if (randomize)
 260                                 ender = ENDER_DEF;
 261                         else
 262                                 ender = begin + reps * s - s;
 263                         mask = 0;
 264                         break;
 265                 case HAVE_REPS | HAVE_BEGIN | HAVE_ENDER:
 266                         if (!randomize) {
 267                                 if (reps == 0)
 268                                         errx(1, "infinite sequences cannot "
 269                                             "be bounded");
 270                                 else if (reps == 1)
 271                                         s = 0.0;
 272                                 else
 273                                         s = (ender - begin) / (reps - 1);
 274                         }
 275                         mask = 0;
 276                         break;
 277                 case HAVE_REPS | HAVE_BEGIN | HAVE_ENDER | HAVE_STEP:
 278                         /* if reps given and implied, */
 279                         if (!randomize && s != 0.0) {
 280                                 long t = (ender - begin + s) / s;
 281                                 if (t <= 0)
 282                                         errx(1, "impossible stepsize");
 283                                 if (t < reps)           /* take lesser */
 284                                         reps = t;
 285                         }
 286                         mask = 0;
 287                         break;
 288                 default:
 289                         errx(1, "bad mask");
 290                 }
 291         if (reps == 0)
 292                 infinity = true;
 293         if (randomize) {
 294                 if (use_random) {
 295                         srandom((unsigned long)s);
 296                         divisor = (double)INT32_MAX + 1;
 297                 } else
 298                         divisor = (double)UINT32_MAX + 1;
 299
 300                 /*
 301                  * Attempt to DWIM when the user has specified an
 302                  * integer range within that of the random number
 303                  * generator: distribute the numbers equally in
 304                  * the range [begin .. ender].  Jot's default %.0f
 305                  * format would make the appearance of the first and
 306                  * last specified value half as likely as the rest.
 307                  */
 308                 if (!have_format && prec == 0 &&
 309                     begin >= 0 && begin < divisor &&
 310                     ender >= 0 && ender < divisor) {
 311                         if (begin <= ender)
 312                                 ender += 1;
 313                         else
 314                                 begin += 1;
 315                         nosign = true;
 316                         intdata = true;
 317                         (void)strlcpy(format,
 318                             chardata ? "%c" : "%u", sizeof(format));
 319                 }
 320                 x = ender - begin;
 321                 for (i = 1; i <= reps || infinity; i++) {
 322                         if (use_random)
 323                                 y = random() / divisor;
 324                         else
 325                                 y = arc4random() / divisor;
 326                         if (putdata(y * x + begin, !(reps - i)))
 327                                 errx(1, "range error in conversion");
 328                 }
 329         } else
 330                 for (i = 1, x = begin; i <= reps || infinity; i++, x += s)
 331                         if (putdata(x, !(reps - i)))
 332                                 errx(1, "range error in conversion");
 333         if (!nofinalnl)
 334                 putchar('\n');
 335         exit(0);
 336 }
 337
 338 /*
 339  * Send x to stdout using the specified format.
 340  * Last is  true if this is the set's last value.
 341  * Return 0 if OK, or a positive number if the number passed was
 342  * outside the range specified by the various flags.
 343  */
 344 static int
 345 putdata(double x, bool last)
 346 {
 347
 348         if (boring)
 349                 printf("%s", format);
 350         else if (longdata && nosign) {
 351                 if (x <= (double)ULONG_MAX && x >= (double)0)
 352                         printf(format, (unsigned long)x);
 353                 else
 354                         return (1);
 355         } else if (longdata) {
 356                 if (x <= (double)LONG_MAX && x >= (double)LONG_MIN)
 357                         printf(format, (long)x);
 358                 else
 359                         return (1);
 360         } else if (chardata || (intdata && !nosign)) {
 361                 if (x <= (double)INT_MAX && x >= (double)INT_MIN)
 362                         printf(format, (int)x);
 363                 else
 364                         return (1);
 365         } else if (intdata) {
 366                 if (x <= (double)UINT_MAX && x >= (double)0)
 367                         printf(format, (unsigned int)x);
 368                 else
 369                         return (1);
 370
 371         } else
 372                 printf(format, x);
 373         if (!last)
 374                 fputs(sepstring, stdout);
 375
 376         return (0);
 377 }
 378
 379 static void
 380 usage(void)
 381 {
 382         fprintf(stderr, "%s\n%s\n",
 383         "usage: jot [-cnr] [-b word] [-w word] [-s string] [-p precision]",
 384         "           [reps [begin [end [s]]]]");
 385         exit(1);
 386 }
 387
 388 /*
 389  * Return the number of digits following the number's decimal point.
 390  * Return 0 if no decimal point is found.
 391  */
 392 static int
 393 getprec(const char *str)
 394 {
 395         const char      *p;
 396         const char      *q;
 397
 398         for (p = str; *p; p++)
 399                 if (*p == '.')
 400                         break;
 401         if (!*p)
 402                 return (0);
 403         for (q = ++p; *p; p++)
 404                 if (!isdigit((unsigned char)*p))
 405                         break;
 406         return (p - q);
 407 }
 408
 409 /*
 410  * Set format, intdata, chardata, longdata, and nosign
 411  * based on the command line arguments.
 412  */
 413 static void
 414 getformat(void)
 415 {
 416         char    *p, *p2;
 417         int dot, hash, space, sign, numbers = 0;
 418         size_t sz;
 419
 420         if (boring)                             /* no need to bother */
 421                 return;
 422         for (p = format; *p; p++)               /* look for '%' */
 423                 if (*p == '%') {
 424                         if (p[1] == '%')
 425                                 p++;            /* leave %% alone */
 426                         else
 427                                 break;
 428                 }
 429         sz = sizeof(format) - strlen(format) - 1;
 430         if (!*p && !chardata) {
 431                 if (snprintf(p, sz, "%%.%df", prec) >= (int)sz)
 432                         errx(1, "-w word too long");
 433         } else if (!*p && chardata) {
 434                 if (strlcpy(p, "%c", sz) >= sz)
 435                         errx(1, "-w word too long");
 436                 intdata = true;
 437         } else if (!*(p+1)) {
 438                 if (sz <= 0)
 439                         errx(1, "-w word too long");
 440                 strcat(format, "%");            /* cannot end in single '%' */
 441         } else {
 442                 /*
 443                  * Allow conversion format specifiers of the form
 444                  * %[#][ ][{+,-}][0-9]*[.[0-9]*]? where ? must be one of
 445                  * [l]{d,i,o,u,x} or {f,e,g,E,G,d,o,x,D,O,U,X,c,u}
 446                  */
 447                 p2 = p++;
 448                 dot = hash = space = sign = numbers = 0;
 449                 while (!isalpha((unsigned char)*p)) {
 450                         if (isdigit((unsigned char)*p)) {
 451                                 numbers++;
 452                                 p++;
 453                         } else if ((*p == '#' && !(numbers|dot|sign|space|
 454                             hash++)) ||
 455                             (*p == ' ' && !(numbers|dot|space++)) ||
 456                             ((*p == '+' || *p == '-') && !(numbers|dot|sign++))
 457                             || (*p == '.' && !(dot++)))
 458                                 p++;
 459                         else
 460                                 goto fmt_broken;
 461                 }
 462                 if (*p == 'l') {
 463                         longdata = true;
 464                         if (*++p == 'l') {
 465                                 if (p[1] != '\0')
 466                                         p++;
 467                                 goto fmt_broken;
 468                         }
 469                 }
 470                 switch (*p) {
 471                 case 'o': case 'u': case 'x': case 'X':
 472                         intdata = nosign = true;
 473                         break;
 474                 case 'd': case 'i':
 475                         intdata = true;
 476                         break;
 477                 case 'D':
 478                         if (!longdata) {
 479                                 intdata = true;
 480                                 break;
 481                         }
 482                 case 'O': case 'U':
 483                         if (!longdata) {
 484                                 intdata = nosign = true;
 485                                 break;
 486                         }
 487                 case 'c':
 488                         if (!(intdata | longdata)) {
 489                                 chardata = true;
 490                                 break;
 491                         }
 492                 case 'h': case 'n': case 'p': case 'q': case 's': case 'L':
 493                 case '$': case '*':
 494                         goto fmt_broken;
 495                 case 'f': case 'e': case 'g': case 'E': case 'G':
 496                         if (!longdata)
 497                                 break;
 498                         /* FALLTHROUGH */
 499                 default:
 500 fmt_broken:
 501                         *++p = '\0';
 502                         errx(1, "illegal or unsupported format '%s'", p2);
 503                         /* NOTREACHED */
 504                 }
 505                 while (*++p)
 506                         if (*p == '%' && *(p+1) && *(p+1) != '%')
 507                                 errx(1, "too many conversions");
 508                         else if (*p == '%' && *(p+1) == '%')
 509                                 p++;
 510                         else if (*p == '%' && !*(p+1)) {
 511                                 if (strlcat(format, "%", sizeof(format)) >=
 512                                     sizeof(format))
 513                                         errx(1, "-w word too long");
 514                                 break;
 515                         }
 516         }
 517 }