hush: get rid of charmap[]

[oweals/busybox.git] / util-linux / dmesg.c

diff --git a/util-linux/dmesg.c b/util-linux/dmesg.c
index c8b3aeeab4b074d4ec5fb100a44811d7ddff358f..f52026c2fdee5649739c4c2a6083849d8d625b7b 100644 (file)
--- a/util-linux/dmesg.c
+++ b/util-linux/dmesg.c
@@ -4,50 +4,64 @@
  * dmesg - display/control kernel ring buffer.
  *
  * Copyright 2006 Rob Landley <rob@landley.net>
- * Copyright 2006 Bernhard Fischer <rep.nop@aon.at>
+ * Copyright 2006 Bernhard Reutner-Fischer <rep.nop@aon.at>
  *
- * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
+ * Licensed under GPLv2, see file LICENSE in this tarball for details.
  */
 
-#include "busybox.h"
-#include <unistd.h>
 #include <sys/klog.h>
+#include "libbb.h"
 
-int dmesg_main(int argc, char *argv[])
+int dmesg_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int dmesg_main(int argc UNUSED_PARAM, char **argv)
 {
+       int len;
+       char *buf;
        char *size, *level;
-       int flags = bb_getopt_ulflags(argc, argv, "cs:n:", &size, &level);
+       unsigned flags = getopt32(argv, "cs:n:", &size, &level);
+       enum {
+               OPT_c = 1<<0,
+               OPT_s = 1<<1,
+               OPT_n = 1<<2
+       };
 
-       if (flags & 4) {
-               if (klogctl(8, NULL, bb_xgetlarg(level, 10, 0, 10)))
+       if (flags & OPT_n) {
+               if (klogctl(8, NULL, xatoul_range(level, 0, 10)))
                        bb_perror_msg_and_die("klogctl");
-       } else {
-               int len;
-               char *buf;
+               return EXIT_SUCCESS;
+       }
 
-               len = (flags & 2) ? bb_xgetlarg(size, 10, 2, INT_MAX) : 16384;
-               buf = xmalloc(len);
-               if (0 > (len = klogctl(3 + (flags & 1), buf, len)))
-                       bb_perror_msg_and_die("klogctl");
+       len = (flags & OPT_s) ? xatoul_range(size, 2, INT_MAX) : 16384;
+       buf = xmalloc(len);
+       len = klogctl(3 + (flags & OPT_c), buf, len);
+       if (len < 0)
+               bb_perror_msg_and_die("klogctl");
+       if (len == 0)
+               return EXIT_SUCCESS;
 
-               // Skip <#> at the start of lines, and make sure we end with a newline.
+       /* Skip <#> at the start of lines, and make sure we end with a newline. */
 
-               if (ENABLE_FEATURE_DMESG_PRETTY) {
-                       int last = '\n';
-                       int in;
+       if (ENABLE_FEATURE_DMESG_PRETTY) {
+               int last = '\n';
+               int in = 0;
 
-                       for (in = 0; in<len;) {
-                               if (last == '\n' && buf[in] == '<') in += 3;
-                               else putchar(last = buf[in++]);
+               do {
+                       if (last == '\n' && buf[in] == '<')
+                               in += 3;
+                       else {
+                               last = buf[in++];
+                               bb_putchar(last);
                        }
-                       if (last != '\n') putchar('\n');
-               } else {
-                       write(1,buf,len);
-                       if (len && buf[len-1]!='\n') putchar('\n');
-               }
-
-               if (ENABLE_FEATURE_CLEAN_UP) free(buf);
+               } while (in < len);
+               if (last != '\n')
+                       bb_putchar('\n');
+       } else {
+               full_write(STDOUT_FILENO, buf, len);
+               if (buf[len-1] != '\n')
+                       bb_putchar('\n');
        }
 
-       return 0;
+       if (ENABLE_FEATURE_CLEAN_UP) free(buf);
+
+       return EXIT_SUCCESS;
 }