coreutils/seq.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * seq implementation for busybox
   4  *
   5  * Copyright (C) 2004, Glenn McGrath
   6  *
   7  * Licensed under GPLv2, see file LICENSE in this source tree.
   8  */
   9 //config:config SEQ
  10 //config:       bool "seq (3.6 kb)"
  11 //config:       default y
  12 //config:       help
  13 //config:       print a sequence of numbers
  14
  15 //applet:IF_SEQ(APPLET_NOFORK(seq, seq, BB_DIR_USR_BIN, BB_SUID_DROP, seq))
  16
  17 //kbuild:lib-$(CONFIG_SEQ) += seq.o
  18
  19 //usage:#define seq_trivial_usage
  20 //usage:       "[-w] [-s SEP] [FIRST [INC]] LAST"
  21 //usage:#define seq_full_usage "\n\n"
  22 //usage:       "Print numbers from FIRST to LAST, in steps of INC.\n"
  23 //usage:       "FIRST, INC default to 1.\n"
  24 //usage:     "\n        -w      Pad to last with leading zeros"
  25 //usage:     "\n        -s SEP  String separator"
  26
  27 #include "libbb.h"
  28
  29 /* This is a NOFORK applet. Be very careful! */
  30
  31 int seq_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  32 int seq_main(int argc, char **argv)
  33 {
  34         enum {
  35                 OPT_w = (1 << 0),
  36                 OPT_s = (1 << 1),
  37         };
  38         double first, last, increment, v;
  39         unsigned n;
  40         unsigned width;
  41         unsigned frac_part;
  42         const char *sep, *opt_s = "\n";
  43         unsigned opt;
  44
  45 #if ENABLE_LOCALE_SUPPORT
  46         /* Undo busybox.c: on input, we want to use dot
  47          * as fractional separator, regardless of current locale */
  48         setlocale(LC_NUMERIC, "C");
  49 #endif
  50
  51         opt = getopt32(argv, "+ws:", &opt_s);
  52         argc -= optind;
  53         argv += optind;
  54         first = increment = 1;
  55         errno = 0;
  56         switch (argc) {
  57                         char *pp;
  58                 case 3:
  59                         increment = strtod(argv[1], &pp);
  60                         errno |= *pp;
  61                 case 2:
  62                         first = strtod(argv[0], &pp);
  63                         errno |= *pp;
  64                 case 1:
  65                         last = strtod(argv[argc-1], &pp);
  66                         if (!errno && *pp == '\0')
  67                                 break;
  68                 default:
  69                         bb_show_usage();
  70         }
  71
  72 #if ENABLE_LOCALE_SUPPORT
  73         setlocale(LC_NUMERIC, "");
  74 #endif
  75
  76         /* Last checked to be compatible with: coreutils-6.10 */
  77         width = 0;
  78         frac_part = 0;
  79         while (1) {
  80                 char *dot = strchrnul(*argv, '.');
  81                 int w = (dot - *argv);
  82                 int f = strlen(dot);
  83                 if (width < w)
  84                         width = w;
  85                 argv++;
  86                 if (!*argv)
  87                         break;
  88                 /* Why do the above _before_ frac check below?
  89                  * Try "seq 1 2.0" and "seq 1.0 2.0":
  90                  * coreutils never pay attention to the number
  91                  * of fractional digits in last arg. */
  92                 if (frac_part < f)
  93                         frac_part = f;
  94         }
  95         if (frac_part) {
  96                 frac_part--;
  97                 if (frac_part)
  98                         width += frac_part + 1;
  99         }
 100         if (!(opt & OPT_w))
 101                 width = 0;
 102
 103         sep = "";
 104         v = first;
 105         n = 0;
 106         while (increment >= 0 ? v <= last : v >= last) {
 107                 if (printf("%s%0*.*f", sep, width, frac_part, v) < 0)
 108                         break; /* I/O error, bail out (yes, this really happens) */
 109                 sep = opt_s;
 110                 /* v += increment; - would accumulate floating point errors */
 111                 n++;
 112                 v = first + n * increment;
 113         }
 114         if (n) /* if while loop executed at least once */
 115                 bb_putchar('\n');
 116
 117         return fflush_all();
 118 }