- add libbb function str_tolower to convert a string to lowercase.

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

diff --git a/coreutils/head.c b/coreutils/head.c
index 17d3ef890166aabd36dcaa4deb66a06e1eb68e5f..ed1d01d52a4e430cbf84c8d56877c4a7117256b9 100644 (file)
--- a/coreutils/head.c
+++ b/coreutils/head.c
@@ -11,11 +11,6 @@
 /* BB_AUDIT GNU compatible -c, -q, and -v options in 'fancy' configuration. */
 /* http://www.opengroup.org/onlinepubs/007904975/utilities/head.html */
 
-#include <stdio.h>
-#include <stdlib.h>
-#include <limits.h>
-#include <ctype.h>
-#include <unistd.h>
 #include "busybox.h"
 
 static const char head_opts[] =
@@ -25,8 +20,18 @@ static const char head_opts[] =
 #endif
        ;
 
+#if ENABLE_FEATURE_FANCY_HEAD
+static const struct suffix_mult head_suffixes[] = {
+       { "b", 512 },
+       { "k", 1024 },
+       { "m", 1024*1024 },
+       { NULL, 0 }
+};
+#endif
+
 static const char header_fmt_str[] = "\n==> %s <==\n";
 
+int head_main(int argc, char **argv);
 int head_main(int argc, char **argv)
 {
        unsigned long count = 10;
@@ -43,11 +48,10 @@ int head_main(int argc, char **argv)
        int c;
        int retval = EXIT_SUCCESS;
 
-#if !ENABLE_DEBUG_YANK_SUSv2 || ENABLE_FEATURE_FANCY_HEAD
+#if ENABLE_INCLUDE_SUSv2 || ENABLE_FEATURE_FANCY_HEAD
        /* Allow legacy syntax of an initial numeric option without -n. */
-       if ((argc > 1) && (argv[1][0] == '-')
-               /* && (isdigit)(argv[1][1]) */
-               && (((unsigned int)(argv[1][1] - '0')) <= 9)
+       if (argc > 1 && argv[1][0] == '-'
+        && isdigit(argv[1][1])
        ) {
                --argc;
                ++argv;
@@ -56,35 +60,40 @@ int head_main(int argc, char **argv)
        }
 #endif
 
-       /* No size benefit in converting this to bb_getopt_ulflags */
+       /* No size benefit in converting this to getopt32 */
        while ((opt = getopt(argc, argv, head_opts)) > 0) {
-               switch(opt) {
+               switch (opt) {
 #if ENABLE_FEATURE_FANCY_HEAD
-                       case 'q':
-                               header_threshhold = INT_MAX;
-                               break;
-                       case 'v':
-                               header_threshhold = -1;
-                               break;
-                       case 'c':
-                               count_bytes = 1;
-                               /* fall through */
+               case 'q':
+                       header_threshhold = INT_MAX;
+                       break;
+               case 'v':
+                       header_threshhold = -1;
+                       break;
+               case 'c':
+                       count_bytes = 1;
+                       /* fall through */
 #endif
-                       case 'n':
-                               p = optarg;
-#if !ENABLE_DEBUG_YANK_SUSv2 || ENABLE_FEATURE_FANCY_HEAD
-                       GET_COUNT:
+               case 'n':
+                       p = optarg;
+#if ENABLE_INCLUDE_SUSv2 || ENABLE_FEATURE_FANCY_HEAD
+ GET_COUNT:
+#endif
+
+#if !ENABLE_FEATURE_FANCY_HEAD
+                       count = xatoul(p);
+#else
+                       count = xatoul_sfx(p, head_suffixes);
 #endif
-                               count = bb_xgetularg10(p);
-                               break;
-                       default:
-                               bb_show_usage();
+                       break;
+               default:
+                       bb_show_usage();
                }
        }
 
        argv += optind;
        if (!*argv) {
-               *--argv = "-";
+               *--argv = (char*)"-";
        }
 
        fmt = header_fmt_str + 1;
@@ -103,12 +112,13 @@ int head_main(int argc, char **argv)
 #endif
 
        do {
-               if ((fp = bb_wfopen_input(*argv)) != NULL) {
+               fp = fopen_or_warn_stdin(*argv);
+               if (fp) {
                        if (fp == stdin) {
                                *argv = (char *) bb_msg_standard_input;
                        }
                        if (header_threshhold) {
-                               bb_printf(fmt, *argv);
+                               printf(fmt, *argv);
                        }
                        i = count;
                        while (i && ((c = getc(fp)) != EOF)) {
@@ -117,14 +127,14 @@ int head_main(int argc, char **argv)
                                }
                                putchar(c);
                        }
-                       if (bb_fclose_nonstdin(fp)) {
+                       if (fclose_if_not_stdin(fp)) {
                                bb_perror_msg("%s", *argv);     /* Avoid multibyte problems. */
                                retval = EXIT_FAILURE;
                        }
-                       bb_xferror_stdout();
+                       die_if_ferror_stdout();
                }
                fmt = header_fmt_str;
        } while (*++argv);
 
-       bb_fflush_stdout_and_exit(retval);
+       fflush_stdout_and_exit(retval);
 }