# HG changeset patch
# User Rob Landley <rob@landley.net>
# Date 1416973518 21600
# Node ID d26f235323db3dcfbaf7462cd90324872ad757a5
# Parent  991f4d61338888e3ea62a7c10a0df6148284c1da
Update dmesg, loosely based on a patch from Elliott Hughes.

Probe the default buffer size, replace the constants with FLAG_x macros, add -r,
replace the byte at a time output with a single xwrite(), more comments.

diff -r 991f4d613388 -r d26f235323db toys/lsb/dmesg.c
--- a/toys/lsb/dmesg.c	Tue Nov 25 18:49:34 2014 -0600
+++ b/toys/lsb/dmesg.c	Tue Nov 25 21:45:18 2014 -0600
@@ -4,19 +4,21 @@
  *
  * http://refspecs.linuxfoundation.org/LSB_4.1.0/LSB-Core-generic/LSB-Core-generic/dmesg.html
 
-USE_DMESG(NEWTOY(dmesg, "s#n#c", TOYFLAG_BIN))
+// We care that FLAG_c is 1, so keep c at the end.
+USE_DMESG(NEWTOY(dmesg, "rs#<1n#c", TOYFLAG_BIN))
 
 config DMESG
   bool "dmesg"
   default y
   help
-    usage: dmesg [-n level] [-s bufsize] | -c
+    usage: dmesg [-n LEVEL] [-s SIZE] | -c
 
     Print or control the kernel ring buffer.
 
-    -n	Set kernel logging level (1-9).
-    -s	Size of buffer to read (in bytes), default 16384.
-    -c	Clear the ring buffer after printing.
+    -c	Clear the ring buffer after printing
+    -n	Set kernel logging LEVEL (1-9)
+    -r	Raw output (with <level markers>)
+    -s	Show the last SIZE many bytes
 */
 
 #define FOR_dmesg
@@ -31,25 +33,36 @@
 void dmesg_main(void)
 {
   // For -n just tell kernel to which messages to keep.
-  if (toys.optflags & 2) {
-    if (klogctl(8, NULL, TT.level)) error_exit("klogctl");
+  if (toys.optflags & FLAG_n) {
+    if (klogctl(8, NULL, TT.level)) perror_exit("klogctl");
   } else {
-    int size, i, last = '\n';
-    char *data;
+    char *data, *to, *from;
+    int size;
 
     // Figure out how much data we need, and fetch it.
     size = TT.size;
-    if (size<2) size = 16384;
-    data = xmalloc(size);
-    size = klogctl(3 + (toys.optflags&1), data, size);
+    if (!size && 1>(size = klogctl(10, 0, 0))) perror_exit("klogctl");;
+    data = to = from = xmalloc(size+1);
+    size = klogctl(3 + (toys.optflags & FLAG_c), data, size);
     if (size < 0) error_exit("klogctl");
+    data[size] = 0;
+
+    // Filter out level markers.
+    if (!(toys.optflags & FLAG_r)) while ((from - data) < size) {
+      if ((from == data || from[-1] == '\n') && *from == '<') {
+        int i = stridx(from, '>');
 
-    // Display data, filtering out level markers.
-    for (i=0; i<size; ) {
-      if (last=='\n' && data[i]=='<') i += 3;
-      else xputc(last = data[i++]);
+        if (i>0) from += i+1;
+      }
+      *(to++) = *(from++);
+    } else to = data+size;
+
+    // Write result. The odds of somebody requesting a buffer of size 3 and
+    // getting "<1>" are remote, but don't segfault if they do.
+    if (to != data) {
+      xwrite(1, data, to-data);
+      if (to[-1] != '\n') xputc('\n');
     }
-    if (last!='\n') xputc('\n');
     if (CFG_TOYBOX_FREE) free(data);
   }
 }