1 /*
2 * logsave.c --- A program which saves the output of a program until
3 * /var/log is mounted.
4 *
5 * Copyright (C) 2003 Theodore Ts'o.
6 *
7 * %Begin-Header%
8 * This file may be redistributed under the terms of the GNU Public
9 * License.
10 * %End-Header%
11 */
12
13 #define _XOPEN_SOURCE 600 /* for inclusion of sa_handler in Solaris */
14
15 #include <stdio.h>
16 #include <stdlib.h>
17 #include <unistd.h>
18 #include <string.h>
19 #include <sys/types.h>
20 #include <sys/wait.h>
21 #include <fcntl.h>
22 #include <time.h>
23 #include <errno.h>
24 #ifdef HAVE_SIGNAL_H
25 #include <signal.h>
26 #endif
27 #ifdef HAVE_GETOPT_H
28 #include <getopt.h>
29 #else
30 extern char *optarg;
31 extern int optind;
32 #endif
33
34 int outfd = -1;
35 int outbufsize = 0;
36 void *outbuf = 0;
37 int verbose = 0;
38 int do_skip = 0;
39 int skip_mode = 0;
40 pid_t child_pid = -1;
41
usage(char * progname)42 static void usage(char *progname)
43 {
44 printf("Usage: %s [-v] [-d dir] logfile program\n", progname);
45 exit(1);
46 }
47
48 #define SEND_LOG 0x01
49 #define SEND_CONSOLE 0x02
50 #define SEND_BOTH 0x03
51
52 /*
53 * Helper function that does the right thing if write returns a
54 * partial write, or an EGAIN/EINTR error.
55 */
write_all(int fd,const char * buf,size_t count)56 static int write_all(int fd, const char *buf, size_t count)
57 {
58 ssize_t ret;
59 int c = 0;
60
61 while (count > 0) {
62 ret = write(fd, buf, count);
63 if (ret < 0) {
64 if ((errno == EAGAIN) || (errno == EINTR))
65 continue;
66 return -1;
67 }
68 count -= ret;
69 buf += ret;
70 c += ret;
71 }
72 return c;
73 }
74
send_output(const char * buffer,int c,int flag)75 static void send_output(const char *buffer, int c, int flag)
76 {
77 const char *cp;
78 char *n;
79 int cnt, d, del;
80
81 if (c == 0)
82 c = strlen(buffer);
83
84 if (flag & SEND_CONSOLE) {
85 cnt = c;
86 cp = buffer;
87 while (cnt) {
88 del = 0;
89 for (d=0; d < cnt; d++) {
90 if (skip_mode &&
91 (cp[d] == '\001' || cp[d] == '\002')) {
92 del = 1;
93 break;
94 }
95 }
96 write_all(1, cp, d);
97 if (del)
98 d++;
99 cnt -= d;
100 cp += d;
101 }
102 }
103 if (!(flag & SEND_LOG))
104 return;
105 if (outfd > 0)
106 write_all(outfd, buffer, c);
107 else {
108 n = realloc(outbuf, outbufsize + c);
109 if (n) {
110 outbuf = n;
111 memcpy(((char *)outbuf)+outbufsize, buffer, c);
112 outbufsize += c;
113 }
114 }
115 }
116
do_read(int fd)117 static int do_read(int fd)
118 {
119 int c;
120 char buffer[4096], *cp, *sep;
121
122 c = read(fd, buffer, sizeof(buffer)-1);
123 if (c <= 0)
124 return c;
125 if (do_skip) {
126 send_output(buffer, c, SEND_CONSOLE);
127 buffer[c] = 0;
128 cp = buffer;
129 while (*cp) {
130 if (skip_mode) {
131 cp = strchr(cp, '\002');
132 if (!cp)
133 return 0;
134 cp++;
135 skip_mode = 0;
136 continue;
137 }
138 sep = strchr(cp, '\001');
139 if (sep)
140 *sep = 0;
141 send_output(cp, 0, SEND_LOG);
142 if (sep) {
143 cp = sep + 1;
144 skip_mode = 1;
145 } else
146 break;
147 }
148 } else
149 send_output(buffer, c, SEND_BOTH);
150 return c;
151 }
152
signal_term(int sig)153 static void signal_term(int sig)
154 {
155 if (child_pid > 0)
156 kill(child_pid, sig);
157 }
158
run_program(char ** argv)159 static int run_program(char **argv)
160 {
161 int fds[2];
162 int status, rc, pid;
163 char buffer[80];
164 #ifdef HAVE_SIGNAL_H
165 struct sigaction sa;
166 #endif
167
168 if (pipe(fds) < 0) {
169 perror("pipe");
170 exit(1);
171 }
172
173 #ifdef HAVE_SIGNAL_H
174 memset(&sa, 0, sizeof(struct sigaction));
175 sa.sa_handler = signal_term;
176 sigaction(SIGINT, &sa, 0);
177 sigaction(SIGTERM, &sa, 0);
178 #ifdef SA_RESTART
179 sa.sa_flags = SA_RESTART;
180 #endif
181 #endif
182
183 pid = fork();
184 if (pid < 0) {
185 perror("vfork");
186 exit(1);
187 }
188 if (pid == 0) {
189 dup2(fds[1],1); /* fds[1] replaces stdout */
190 dup2(fds[1],2); /* fds[1] replaces stderr */
191 close(fds[0]); /* don't need this here */
192
193 execvp(argv[0], argv);
194 perror(argv[0]);
195 exit(1);
196 }
197 child_pid = pid;
198 close(fds[1]);
199
200 while (!(waitpid(pid, &status, WNOHANG ))) {
201 do_read(fds[0]);
202 }
203 child_pid = -1;
204 do_read(fds[0]);
205 close(fds[0]);
206
207 if ( WIFEXITED(status) ) {
208 rc = WEXITSTATUS(status);
209 if (rc) {
210 send_output(argv[0], 0, SEND_BOTH);
211 sprintf(buffer, " died with exit status %d\n", rc);
212 send_output(buffer, 0, SEND_BOTH);
213 }
214 } else {
215 if (WIFSIGNALED(status)) {
216 send_output(argv[0], 0, SEND_BOTH);
217 sprintf(buffer, "died with signal %d\n",
218 WTERMSIG(status));
219 send_output(buffer, 0, SEND_BOTH);
220 rc = 1;
221 }
222 rc = 0;
223 }
224 return rc;
225 }
226
copy_from_stdin(void)227 static int copy_from_stdin(void)
228 {
229 int c, bad_read = 0;
230
231 while (1) {
232 c = do_read(0);
233 if ((c == 0 ) ||
234 ((c < 0) && ((errno == EAGAIN) || (errno == EINTR)))) {
235 if (bad_read++ > 3)
236 break;
237 continue;
238 }
239 if (c < 0) {
240 perror("read");
241 exit(1);
242 }
243 bad_read = 0;
244 }
245 return 0;
246 }
247
248
249
main(int argc,char ** argv)250 int main(int argc, char **argv)
251 {
252 int c, pid, rc;
253 char *outfn, **cpp;
254 int openflags = O_CREAT|O_WRONLY|O_TRUNC;
255 int send_flag = SEND_LOG;
256 int do_stdin;
257 time_t t;
258
259 while ((c = getopt(argc, argv, "+asv")) != EOF) {
260 switch (c) {
261 case 'a':
262 openflags &= ~O_TRUNC;
263 openflags |= O_APPEND;
264 break;
265 case 's':
266 do_skip = 1;
267 break;
268 case 'v':
269 verbose++;
270 send_flag |= SEND_CONSOLE;
271 break;
272 }
273 }
274 if (optind == argc || optind+1 == argc)
275 usage(argv[0]);
276 outfn = argv[optind];
277 optind++;
278 argv += optind;
279 argc -= optind;
280
281 outfd = open(outfn, openflags, 0644);
282 do_stdin = !strcmp(argv[0], "-");
283
284 send_output("Log of ", 0, send_flag);
285 if (do_stdin)
286 send_output("stdin", 0, send_flag);
287 else {
288 for (cpp = argv; *cpp; cpp++) {
289 send_output(*cpp, 0, send_flag);
290 send_output(" ", 0, send_flag);
291 }
292 }
293 send_output("\n", 0, send_flag);
294 t = time(0);
295 send_output(ctime(&t), 0, send_flag);
296 send_output("\n", 0, send_flag);
297
298 if (do_stdin)
299 rc = copy_from_stdin();
300 else
301 rc = run_program(argv);
302
303 send_output("\n", 0, send_flag);
304 t = time(0);
305 send_output(ctime(&t), 0, send_flag);
306 send_output("----------------\n", 0, send_flag);
307
308 if (outbuf) {
309 pid = fork();
310 if (pid < 0) {
311 perror("fork");
312 exit(1);
313 }
314 if (pid) {
315 if (verbose)
316 printf("Backgrounding to save %s later\n",
317 outfn);
318 exit(rc);
319 }
320 setsid(); /* To avoid getting killed by init */
321 while (outfd < 0) {
322 outfd = open(outfn, openflags, 0644);
323 sleep(1);
324 }
325 write_all(outfd, outbuf, outbufsize);
326 free(outbuf);
327 }
328 close(outfd);
329
330 exit(rc);
331 }
332