foo*sum: report I/O errors, don't merely exit with 1.

[oweals/busybox.git] / coreutils / seq.c

diff --git a/coreutils/seq.c b/coreutils/seq.c
index 8006be83dc5227481906dd08ce75ac4ef9bab194..8986192930e61025f22ec06840c18add4928217c 100644 (file)
--- a/coreutils/seq.c
+++ b/coreutils/seq.c
@@ -2,50 +2,108 @@
 /*
  * seq implementation for busybox
  *
- *  This program is free software; you can redistribute it and/or modify
- *  it under the terms of version 2 of the GNU General Public License as
- *  published by the Free Software Foundation.
+ * Copyright (C) 2004, Glenn McGrath
  *
- *  This program is distributed in the hope that it will be useful,
- *  but WITHOUT ANY WARRANTY; without even the implied warranty of
- *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- *  GNU Library General Public License for more details.
- *
- *  You should have received a copy of the GNU General Public License
- *  along with this program; if not, write to the Free Software
- *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ * Licensed under GPLv2, see file LICENSE in this source tree.
  */
 
-#include <stdio.h>
-#include <stdlib.h>
-#include "busybox.h"
+//usage:#define seq_trivial_usage
+//usage:       "[-w] [-s SEP] [FIRST [INC]] LAST"
+//usage:#define seq_full_usage "\n\n"
+//usage:       "Print numbers from FIRST to LAST, in steps of INC.\n"
+//usage:       "FIRST, INC default to 1.\n"
+//usage:     "\n       -w      Pad to last with leading zeros"
+//usage:     "\n       -s SEP  String separator"
+
+#include "libbb.h"
 
-extern int seq_main(int argc, char **argv)
+/* This is a NOFORK applet. Be very careful! */
+
+int seq_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int seq_main(int argc, char **argv)
 {
-       double last;
-       double first = 1;
-       double increment = 1;
-       double i;
-
-       if (argc == 4) {
-               first = atof(argv[1]);
-               increment = atof(argv[2]);
-       } else if (argc == 3) {
-               first = atof(argv[1]);
-       } else if (argc != 2) {
-               bb_show_usage();
+       enum {
+               OPT_w = (1 << 0),
+               OPT_s = (1 << 1),
+       };
+       double first, last, increment, v;
+       unsigned n;
+       unsigned width;
+       unsigned frac_part;
+       const char *sep, *opt_s = "\n";
+       unsigned opt;
+
+#if ENABLE_LOCALE_SUPPORT
+       /* Undo busybox.c: on input, we want to use dot
+        * as fractional separator, regardless of current locale */
+       setlocale(LC_NUMERIC, "C");
+#endif
+
+       opt = getopt32(argv, "+ws:", &opt_s);
+       argc -= optind;
+       argv += optind;
+       first = increment = 1;
+       errno = 0;
+       switch (argc) {
+                       char *pp;
+               case 3:
+                       increment = strtod(argv[1], &pp);
+                       errno |= *pp;
+               case 2:
+                       first = strtod(argv[0], &pp);
+                       errno |= *pp;
+               case 1:
+                       last = strtod(argv[argc-1], &pp);
+                       if (!errno && *pp == '\0')
+                               break;
+               default:
+                       bb_show_usage();
        }
-       last = atof(argv[argc - 1]);
 
-       /* You should note that this is pos-5.0.91 semantics, -- FK. */
-       if ((first > last) && (increment > 0)) {
-               return EXIT_SUCCESS;
+#if ENABLE_LOCALE_SUPPORT
+       setlocale(LC_NUMERIC, "");
+#endif
+
+       /* Last checked to be compatible with: coreutils-6.10 */
+       width = 0;
+       frac_part = 0;
+       while (1) {
+               char *dot = strchrnul(*argv, '.');
+               int w = (dot - *argv);
+               int f = strlen(dot);
+               if (width < w)
+                       width = w;
+               argv++;
+               if (!*argv)
+                       break;
+               /* Why do the above _before_ frac check below?
+                * Try "seq 1 2.0" and "seq 1.0 2.0":
+                * coreutils never pay attention to the number
+                * of fractional digits in last arg. */
+               if (frac_part < f)
+                       frac_part = f;
+       }
+       if (frac_part) {
+               frac_part--;
+               if (frac_part)
+                       width += frac_part + 1;
        }
+       if (!(opt & OPT_w))
+               width = 0;
 
-       for (i = first; ((first <= last) ? (i <= last) : (i >= last));
-            i += increment) {
-               printf("%g\n", i);
+       sep = "";
+       v = first;
+       n = 0;
+       while (increment >= 0 ? v <= last : v >= last) {
+               if (printf("%s%0*.*f", sep, width, frac_part, v) < 0)
+                       break; /* I/O error, bail out (yes, this really happens) */
+               sep = opt_s;
+               /* v += increment; - would accumulate floating point errors */
+               n++;
+               v = first + n * increment;
        }
+       if (n) /* if while loop executed at least once */
+               bb_putchar('\n');
 
-       return EXIT_SUCCESS;
+       return fflush_all();
 }