1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
/* klogd.c - Klogd, The kernel log Dameon.
*
* Copyright 2013 Sandeep Sharma <sandeep.jack2756@gmail.com>
*
* No standard
USE_KLOGD(NEWTOY(klogd, "c#<1>8n", TOYFLAG_SBIN))
config KLOGD
bool "klogd"
default n
help
usage: klogd [-n] [-c N]
-c N Print to console messages more urgent than prio N (1-8)"
-n Run in foreground.
config KLOGD_SOURCE_RING_BUFFER
bool "enable kernel ring buffer as log source."
default n
depends on KLOGD
*/
#define FOR_klogd
#include "toys.h"
#include <signal.h>
GLOBALS(
long level;
int fd;
)
#if CFG_KLOGD_SOURCE_RING_BUFFER
#include <sys/klog.h>
static void open_klogd(void)
{
syslog(LOG_NOTICE, "KLOGD: started with Kernel ring buffer as log source\n");
klogctl(1, NULL, 0);
}
static int read_klogd(char *bufptr, int len)
{
return klogctl(2, bufptr, len);
}
static void set_log_level(int level)
{
klogctl(8, NULL, level);
}
static void close_klogd(void)
{
klogctl(7, NULL, 0);
klogctl(0, NULL, 0);
}
#else
static void open_klogd(void)
{
TT.fd = xopen("/proc/kmsg", O_RDONLY); //_PATH_KLOG in paths.h
syslog(LOG_NOTICE, "KLOGD: started with /proc/kmsg as log source\n");
}
static int read_klogd(char *bufptr, int len)
{
return xread(TT.fd, bufptr, len);
}
static void set_log_level(int level)
{
FILE *fptr = xfopen("/proc/sys/kernel/printk", "w");
fprintf(fptr, "%u\n", level);
fclose(fptr);
fptr = NULL;
}
static void close_klogd(void)
{
set_log_level(7);
xclose(TT.fd);
}
#endif
static void handle_signal(int sig)
{
close_klogd();
syslog(LOG_NOTICE,"KLOGD: Daemon exiting......");
exit(1);
}
static int daemonize(void)
{
pid_t pid;
int fd = open("/dev/null", O_RDWR);
if (fd < 0) fd = open("/", O_RDONLY, 0666);
if((pid = fork()) < 0) {
perror_msg("DAEMON: fail to fork");
return -1;
}
if (pid) exit(EXIT_SUCCESS);
setsid();
dup2(fd, 0);
dup2(fd, 1);
dup2(fd, 2);
if (fd > 2) close(fd);
return 0;
}
/*
* Read kernel ring buffer in local buff and keep track of
* "used" amount to track next read to start.
*/
void klogd_main(void)
{
int prio, size, used = 0;
char *start, *line_start, msg_buffer[16348]; //LOG_LINE_LENGTH - Ring buffer size
sigatexit(handle_signal);
if (toys.optflags & FLAG_c) set_log_level(TT.level); //set log level
if (!(toys.optflags & FLAG_n)) daemonize(); //Make it daemon
open_klogd();
openlog("Kernel", 0, LOG_KERN); //open connection to system logger..
while(1) {
start = msg_buffer + used; //start updated for re-read.
size = read_klogd(start, sizeof(msg_buffer) - used - 1);
if (size < 0) perror_exit("error reading file:");
start[size] = '\0'; //Ensure last line to be NUL terminated.
if (used) start = msg_buffer;
while(start) {
if ((line_start = strsep(&start, "\n")) != NULL && start != NULL) used = 0;
else { //Incomplete line, copy it to start of buff.
used = strlen(line_start);
strcpy(msg_buffer, line_start);
if (used < (sizeof(msg_buffer) - 1)) break;
used = 0; //we have buffer full, log it as it is.
}
prio = LOG_INFO; //we dont know priority, mark it INFO
if (*line_start == '<') { //we have new line to syslog
line_start++;
if (line_start) prio = (int)strtoul(line_start, &line_start, 10);
if (*line_start == '>') line_start++;
}
if (*line_start) syslog(prio, "%s", line_start);
}
}
}
|
