ubase/dmesg.c

109 lines
1.9 KiB
C
Raw Normal View History

2013-08-06 19:08:41 +00:00
/* See LICENSE file for copyright and license details. */
#include <sys/klog.h>
2013-08-06 19:08:41 +00:00
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
2013-08-06 19:08:41 +00:00
#include "util.h"
static int dmesg_show(int fd, const void *buf, size_t n);
enum {
SYSLOG_ACTION_READ_ALL = 3,
2013-08-14 13:34:57 +00:00
SYSLOG_ACTION_CLEAR = 5,
SYSLOG_ACTION_CONSOLE_LEVEL = 8,
SYSLOG_ACTION_SIZE_BUFFER = 10
};
2013-08-06 19:08:41 +00:00
static void
usage(void)
{
2014-04-12 12:53:32 +00:00
eprintf("usage: [-Ccr] [-n level] %s\n", argv0);
2013-08-06 19:08:41 +00:00
}
int
main(int argc, char *argv[])
{
int n;
char *buf;
2013-08-16 13:59:53 +00:00
int cflag = 0;
2013-08-17 16:22:30 +00:00
int rflag = 0;
long level;
2013-08-06 19:08:41 +00:00
ARGBEGIN {
2013-08-14 13:34:57 +00:00
case 'C':
if (klogctl(SYSLOG_ACTION_CLEAR, NULL, 0) < 0)
eprintf("klogctl:");
return EXIT_SUCCESS;
2013-08-16 13:59:53 +00:00
case 'c':
cflag = 1;
break;
2013-08-17 16:22:30 +00:00
case 'r':
rflag = 1;
break;
case 'n':
level = estrtol(EARGF(usage()), 10);
if (klogctl(SYSLOG_ACTION_CONSOLE_LEVEL, NULL, level) < 0)
eprintf("klogctl:");
return EXIT_SUCCESS;
2013-08-06 19:08:41 +00:00
default:
usage();
} ARGEND;
n = klogctl(SYSLOG_ACTION_SIZE_BUFFER, NULL, 0);
2013-08-06 19:08:41 +00:00
if (n < 0)
eprintf("klogctl:");
2013-08-12 10:29:21 +00:00
2013-08-06 19:08:41 +00:00
buf = malloc(n);
if (!buf)
eprintf("malloc:");
2013-08-12 10:29:21 +00:00
n = klogctl(SYSLOG_ACTION_READ_ALL, buf, n);
2013-08-06 19:08:41 +00:00
if (n < 0)
eprintf("klogctl:");
2013-08-12 10:29:21 +00:00
2013-08-17 16:22:30 +00:00
if (rflag) {
if (write(STDOUT_FILENO, buf, n) != n)
eprintf("write:");
} else {
n = dmesg_show(STDOUT_FILENO, buf, n);
if (n < 0)
eprintf("dmesg_show:");
}
2013-08-12 10:29:21 +00:00
2013-08-16 13:59:53 +00:00
if (cflag && klogctl(SYSLOG_ACTION_CLEAR, NULL, 0) < 0)
eprintf("klogctl:");
2013-08-06 19:08:41 +00:00
free(buf);
2013-10-07 18:11:40 +00:00
return EXIT_SUCCESS;
2013-08-06 19:08:41 +00:00
}
static int
dmesg_show(int fd, const void *buf, size_t n)
{
int last = '\n';
char newbuf[n], *q = newbuf;
const char *p = buf;
ssize_t r;
size_t i;
memset(newbuf, 0, n);
for (i = 0; i < n; ) {
if (last == '\n' && p[i] == '<') {
i += 2;
if (i + 1 < n && p[i + 1] == '>')
i++;
} else {
*q++ = p[i];
}
last = p[i++];
}
r = write(fd, newbuf, n);
if(r < 0 || (size_t)r != n)
return -1;
if (last != '\n')
if (write(fd, "\n", 1) != 1)
return -1;
return 0;
}