system_cmds-433.8.tar.gz

[apple/system_cmds.git] / dmesg.tproj / dmesg.c

diff --git a/dmesg.tproj/dmesg.c b/dmesg.tproj/dmesg.c
index 10d5caeea1a180b6ea732c800d40bd1bd6357ccc..373653dcebf45422d890323b43c5e670046b95b7 100644 (file)
--- a/dmesg.tproj/dmesg.c
+++ b/dmesg.tproj/dmesg.c
@@ -54,117 +54,34 @@
  * SUCH DAMAGE.
  */
 
-#include <sys/cdefs.h>
-#include <sys/msgbuf.h>
-
-#include <fcntl.h>
-#include <kvm.h>
-#include <limits.h>
-#include <nlist.h>
 #include <stdio.h>
 #include <stdlib.h>
-#include <time.h>
-#include <unistd.h>
 #include <vis.h>
+#include <libproc.h>
 
-struct nlist nl[] = {
-#define        X_MSGBUF        0
-       { "_msgbufp" },
-       { NULL },
-};
-
-void usage __P((void));
-
-#define        KREAD(addr, var) \
-       kvm_read(kd, addr, &var, sizeof(var)) != sizeof(var)
+void
+usage() {
+       (void)fprintf(stderr, "usage: sudo dmesg\n");
+       exit(1);
+}
 
 int
-main(argc, argv)
-       int argc;
-       char *argv[];
-{
-       register int ch, newl, skip;
-       register char *p, *ep;
-       struct msgbuf *bufp, cur;
-       char *memf, *nlistf;
-       kvm_t *kd;
-       char buf[5];
-
-       memf = nlistf = NULL;
-       while ((ch = getopt(argc, argv, "M:N:")) != EOF)
-               switch(ch) {
-               case 'M':
-                       memf = optarg;
-                       break;
-               case 'N':
-                       nlistf = optarg;
-                       break;
-               case '?':
-               default:
-                       usage();
-               }
-       argc -= optind;
-       argv += optind;
+main(int argc, char **argv) {
+       char msgbuf[16*1024], *visbuf;
+       long data_size;
 
-       /*
-        * Discard setgid privileges if not the running kernel so that bad
-        * guys can't print interesting stuff from kernel memory.
-        */
-       if (memf != NULL || nlistf != NULL)
-               setgid(getgid());
-
-       /* Read in kernel message buffer, do sanity checks. */
-       if ((kd = kvm_open(nlistf, memf, NULL, O_RDONLY, "dmesg")) == NULL)
-               exit (1);
-       if (kvm_nlist(kd, nl) == -1)
-               errx(1, "kvm_nlist: %s", kvm_geterr(kd));
-       if (nl[X_MSGBUF].n_type == 0)
-               errx(1, "%s: msgbufp not found", nlistf ? nlistf : "namelist");
-       if (KREAD(nl[X_MSGBUF].n_value, bufp) || KREAD((long)bufp, cur))
-               errx(1, "kvm_read: %s", kvm_geterr(kd));
-       kvm_close(kd);
-       if (cur.msg_magic != MSG_MAGIC)
-               errx(1, "magic number incorrect");
-       if (cur.msg_bufx >= MSG_BSIZE)
-               cur.msg_bufx = 0;
-
-       /*
-        * The message buffer is circular; start at the read pointer, and
-        * go to the write pointer - 1.
-        */
-       p = cur.msg_bufc + cur.msg_bufx;
-       ep = cur.msg_bufc + cur.msg_bufx - 1;
-       for (newl = skip = 0; p != ep; ++p) {
-               if (p == cur.msg_bufc + MSG_BSIZE)
-                       p = cur.msg_bufc;
-               ch = *p;
-               /* Skip "\n<.*>" syslog sequences. */
-               if (skip) {
-                       if (ch == '>')
-                               newl = skip = 0;
-                       continue;
-               }
-               if (newl && ch == '<') {
-                       skip = 1;
-                       continue;
-               }
-               if (ch == '\0')
-                       continue;
-               newl = ch == '\n';
-               (void)vis(buf, ch, 0, 0);
-               if (buf[1] == 0)
-                       (void)putchar(buf[0]);
-               else
-                       (void)printf("%s", buf);
+       if (argc > 1)
+               usage();
+       
+       if ((data_size = proc_kmsgbuf(msgbuf, sizeof(msgbuf))) == 0){
+               perror("Unable to obtain kernel buffer");
+               usage();
        }
-       if (!newl)
-               (void)putchar('\n');
+
+       visbuf = malloc(data_size*4);
+       strvis(visbuf, msgbuf, 0);
+       printf("%s", visbuf);
+       free(visbuf);
        exit(0);
 }
 
-void
-usage()
-{
-       (void)fprintf(stderr, "usage: dmesg [-M core] [-N system]\n");
-       exit(1);
-}