ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/root-tail/root-tail.c
Revision: 1.19
Committed: Sat Mar 27 16:25:11 2004 UTC (22 years, 5 months ago) by pcg
Content type: text/plain
Branch: MAIN
Changes since 1.18: +3 -4 lines
Log Message:
*** empty log message ***

File Contents

# User Rev Content
1 pcg 1.1 /*
2     * Copyright 2001 by Marco d'Itri <md@linux.it>
3     *
4     * Original version by Mike Baker, then maintained by pcg@goof.com.
5     *
6     * This program is free software; you can redistribute it and/or modify
7     * it under the terms of the GNU General Public License as published by
8     * the Free Software Foundation; either version 2 of the License, or
9     * (at your option) any later version.
10     *
11     * This program is distributed in the hope that it will be useful,
12     * but WITHOUT ANY WARRANTY; without even the implied warranty of
13     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14     * GNU General Public License for more details.
15     *
16     * You should have received a copy of the GNU General Public License
17     * along with this program; if not, write to the Free Software
18     * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
19     */
20    
21     #include "config.h"
22     #include <stdlib.h>
23     #include <stdio.h>
24     #include <unistd.h>
25     #include <string.h>
26     #include <signal.h>
27     #include <time.h>
28     #include <fcntl.h>
29     #include <errno.h>
30     #include <sys/time.h>
31     #include <sys/stat.h>
32     #include <sys/types.h>
33 pcg 1.7 #include <locale.h>
34 pcg 1.11 #include <ctype.h>
35     #include <stdarg.h>
36 pcg 1.1 #if HAS_REGEX
37     #include <regex.h>
38     #endif
39     #include <X11/Xlib.h>
40 pcg 1.7 #include <X11/Xatom.h>
41 pcg 1.1 #include <X11/Xutil.h>
42    
43     /* data structures */
44 pcg 1.11 struct logfile_entry
45     {
46     struct logfile_entry *next;
47    
48     char *fname; /* name of file */
49     char *desc; /* alternative description */
50     char *buf; /* text read but not yet displayed */
51     FILE *fp; /* FILE struct associated with file */
52     ino_t inode; /* inode of the file opened */
53     off_t last_size; /* file size at the last check */
54     unsigned long color; /* color to be used for printing */
55     int partial; /* true if the last line isn't complete */
56     int lastpartial; /* true if the previous output wasn't complete */
57     int index; /* index into linematrix of a partial line */
58 pcg 1.1 };
59    
60 pcg 1.11 struct linematrix
61     {
62     char *line;
63 pcg 1.12 int len;
64 pcg 1.11 unsigned long color;
65 pcg 1.1 };
66    
67     /* global variables */
68 pcg 1.12 struct linematrix *lines;
69 pcg 1.11 int width = STD_WIDTH, height = STD_HEIGHT, listlen;
70 pcg 1.1 int win_x = LOC_X, win_y = LOC_Y;
71 pcg 1.11 int font_descent, font_height;
72 pcg 1.1 int do_reopen;
73 pcg 1.11 struct timeval interval = { 3, 0 };
74 pcg 1.7 XFontSet fontset;
75 pcg 1.1
76     /* command line options */
77 pcg 1.5 int opt_noinitial, opt_shade, opt_frame, opt_reverse, opt_nofilename,
78 pcg 1.11 opt_whole, opt_update, geom_mask, reload = 0;
79 pcg 1.1 const char *command = NULL,
80 pcg 1.11 *fontname = USE_FONT, *dispname = NULL, *def_color = DEF_COLOR,
81     *continuation = "[+]";
82 pcg 1.1
83     struct logfile_entry *loglist = NULL, *loglist_tail = NULL;
84    
85     Display *disp;
86     Window root;
87     GC WinGC;
88    
89     #if HAS_REGEX
90 pcg 1.11 struct re_list
91     {
92     regex_t from;
93     const char *to;
94     struct re_list *next;
95 pcg 1.1 };
96     struct re_list *re_head, *re_tail;
97     #endif
98    
99    
100     /* prototypes */
101 pcg 1.11 void list_files (int);
102     void force_reopen (int);
103     void force_refresh (int);
104     void blank_window (int);
105    
106     void InitWindow (void);
107     unsigned long GetColor (const char *);
108     void redraw (void);
109 pcg 1.12 void refresh (int, int);
110 pcg 1.11
111     void transform_line (char *s);
112     int lineinput (struct logfile_entry *);
113     void reopen (void);
114     void check_open_files (void);
115     FILE *openlog (struct logfile_entry *);
116 pcg 1.12 static void main_loop (void);
117 pcg 1.11
118     void display_version (void);
119     void display_help (char *);
120     void install_signal (int, void (*)(int));
121     void *xstrdup (const char *);
122     void *xmalloc (size_t);
123     int daemonize (void);
124 pcg 1.1
125     /* signal handlers */
126 pcg 1.11 void
127     list_files (int dummy)
128 pcg 1.1 {
129 pcg 1.11 struct logfile_entry *e;
130 pcg 1.1
131 pcg 1.11 fprintf (stderr, "Files opened:\n");
132     for (e = loglist; e; e = e->next)
133     fprintf (stderr, "\t%s (%s)\n", e->fname, e->desc);
134 pcg 1.1 }
135    
136 pcg 1.11 void
137     force_reopen (int dummy)
138 pcg 1.1 {
139 pcg 1.11 do_reopen = 1;
140 pcg 1.1 }
141    
142 pcg 1.11 void
143     force_refresh (int dummy)
144 pcg 1.1 {
145 pcg 1.11 redraw ();
146 pcg 1.1 }
147    
148 pcg 1.11 void
149     blank_window (int dummy)
150 pcg 1.1 {
151 pcg 1.19 XClearArea (disp, root, win_x - 2, win_y - 2, width + 5, height + 5, False);
152 pcg 1.11 XFlush (disp);
153     exit (0);
154 pcg 1.1 }
155    
156     /* X related functions */
157 pcg 1.11 unsigned long
158     GetColor (const char *ColorName)
159 pcg 1.1 {
160 pcg 1.11 XColor Color;
161     XWindowAttributes Attributes;
162 pcg 1.1
163 pcg 1.11 XGetWindowAttributes (disp, root, &Attributes);
164     Color.pixel = 0;
165 pcg 1.8
166 pcg 1.11 if (!XParseColor (disp, Attributes.colormap, ColorName, &Color))
167     fprintf (stderr, "can't parse %s\n", ColorName);
168     else if (!XAllocColor (disp, Attributes.colormap, &Color))
169     fprintf (stderr, "can't allocate %s\n", ColorName);
170 pcg 1.8
171 pcg 1.11 return Color.pixel;
172 pcg 1.1 }
173    
174 pcg 1.11 static Window
175     root_window (Display * display, int screen_number)
176 pcg 1.7 {
177 pcg 1.11 Atom SWM_VROOT = XInternAtom (display, "__SWM_VROOT", False);
178 pcg 1.7 Window real_root_window = RootWindow (display, screen_number);
179    
180 pcg 1.11 if (root) /* root window set via option */
181 pcg 1.8 return root;
182    
183 pcg 1.11 if (SWM_VROOT != None)
184 pcg 1.7 {
185     Window unused, *windows;
186     unsigned int count;
187    
188     if (XQueryTree (display, real_root_window, &unused, &unused, &windows,
189 pcg 1.11 &count))
190     {
191     int i;
192    
193     for (i = 0; i < count; i++)
194     {
195     Atom type;
196     int format;
197     unsigned long nitems, bytes_after_return;
198     unsigned char *virtual_root_window;
199    
200     if (XGetWindowProperty (display, windows[i], SWM_VROOT,
201     0, 1, False, XA_WINDOW, &type, &format,
202     &nitems, &bytes_after_return,
203     &virtual_root_window) == Success)
204     {
205     if (type != None)
206     {
207     if (type == XA_WINDOW)
208     {
209     XFree (windows);
210     return (Window) virtual_root_window;
211     }
212     else
213     fprintf (stderr,
214     "__SWM_VROOT property type mismatch");
215     }
216     }
217     else
218     fprintf (stderr,
219     "failed to get __SWM_VROOT property on window 0x%lx",
220     windows[i]);
221     }
222    
223     if (count)
224     XFree (windows);
225     }
226 pcg 1.7 else
227 pcg 1.11 fprintf (stderr, "Can't query tree on root window 0x%lx",
228     real_root_window);
229 pcg 1.7 }
230     else
231     /* This shouldn't happen. The Xlib documentation is wrong BTW. */
232     fprintf (stderr, "Can't intern atom __SWM_VROOT");
233    
234     return real_root_window;
235     }
236    
237 pcg 1.11 void
238     InitWindow (void)
239 pcg 1.1 {
240 pcg 1.11 XGCValues gcv;
241     unsigned long gcm;
242     int screen, ScreenWidth, ScreenHeight;
243    
244     if (!(disp = XOpenDisplay (dispname)))
245     {
246     fprintf (stderr, "Can't open display %s.\n", dispname);
247     exit (1);
248     }
249    
250     screen = DefaultScreen (disp);
251     ScreenHeight = DisplayHeight (disp, screen);
252     ScreenWidth = DisplayWidth (disp, screen);
253    
254     root = root_window (disp, screen);
255    
256     gcm = GCBackground;
257     gcv.graphics_exposures = True;
258     WinGC = XCreateGC (disp, root, gcm, &gcv);
259     XMapWindow (disp, root);
260     XSetForeground (disp, WinGC, GetColor (DEF_COLOR));
261    
262     {
263     char **missing_charset_list;
264     int missing_charset_count;
265     char *def_string;
266    
267     fontset = XCreateFontSet (disp, fontname,
268     &missing_charset_list, &missing_charset_count,
269     &def_string);
270    
271     if (missing_charset_count)
272     {
273     fprintf (stderr,
274     "Missing charsets in String to FontSet conversion (%s)\n",
275     missing_charset_list[0]);
276     XFreeStringList (missing_charset_list);
277     }
278     }
279    
280     if (!fontset)
281     {
282     fprintf (stderr, "unable to create fontset, exiting.\n");
283     exit (1);
284     }
285    
286     {
287     XFontSetExtents *xfe = XExtentsOfFontSet (fontset);
288    
289     font_height = xfe->max_logical_extent.height;
290     font_descent = xfe->max_logical_extent.y;
291     }
292    
293     if (geom_mask & XNegative)
294     win_x = win_x + ScreenWidth - width;
295     if (geom_mask & YNegative)
296     win_y = win_y + ScreenHeight - height;
297    
298     listlen = height / font_height;
299 pcg 1.1
300 pcg 1.11 if (!listlen)
301     {
302     fprintf (stderr, "height too small for a single line, setting to %d\n",
303     font_height);
304     listlen = 1;
305     }
306    
307     height = listlen * font_height;
308    
309     XSelectInput (disp, root, ExposureMask | FocusChangeMask);
310 pcg 1.1 }
311    
312     /*
313     * redraw does a complete redraw, rather than an update (i.e. the area
314     * gets cleared first)
315     * the rest is handled by regular refresh()'es
316     */
317 pcg 1.11 void
318     redraw (void)
319 pcg 1.1 {
320 pcg 1.19 XClearArea (disp, root, win_x - 2, win_y - 2, width + 5, height + 5, False);
321 pcg 1.12 refresh (0, 32768);
322 pcg 1.1 }
323    
324     /* Just redraw everything without clearing (i.e. after an EXPOSE event) */
325 pcg 1.11 void
326 pcg 1.12 refresh (int miny, int maxy)
327 pcg 1.1 {
328 pcg 1.11 int lin;
329     int offset = (listlen + 1) * font_height;
330     unsigned long black_color = GetColor ("black");
331 pcg 1.1
332 pcg 1.11 miny -= win_y + font_height;
333     maxy -= win_y - font_height;
334 pcg 1.1
335 pcg 1.12
336 pcg 1.11 for (lin = listlen; lin--;)
337     {
338 pcg 1.12 struct linematrix *line = lines + (opt_reverse ? listlen - lin - 1 : lin);
339 pcg 1.7
340 pcg 1.11 offset -= font_height;
341 pcg 1.7
342 pcg 1.11 if (offset < miny || offset > maxy)
343     continue;
344 jebusbfb 1.4
345 pcg 1.11 if (opt_shade)
346     {
347     XSetForeground (disp, WinGC, black_color);
348     XmbDrawString (disp, root, fontset, WinGC, win_x + 2,
349 pcg 1.12 win_y + offset + 2, line->line, line->len);
350 pcg 1.11 }
351    
352     XSetForeground (disp, WinGC, line->color);
353     XmbDrawString (disp, root, fontset, WinGC, win_x, win_y + offset,
354 pcg 1.12 line->line, line->len);
355 pcg 1.1 }
356    
357 pcg 1.11 if (opt_frame)
358 pcg 1.15 {
359     XSetForeground (disp, WinGC, GetColor (def_color));
360     XDrawRectangle (disp, root, WinGC, win_x - 2, win_y - 2, width + 4, height + 4);
361     }
362 pcg 1.1 }
363    
364     #if HAS_REGEX
365 pcg 1.11 void
366     transform_line (char *s)
367 pcg 1.1 {
368     #ifdef I_AM_Md
369 pcg 1.11 int i;
370     if (1)
371     {
372     for (i = 16; s[i]; i++)
373     s[i] = s[i + 11];
374 pcg 1.1 }
375 pcg 1.11 s[i + 1] = '\0';
376 pcg 1.1 #endif
377    
378 pcg 1.11 if (transformre)
379     {
380     int i;
381     regmatch_t matched[16];
382    
383     i = regexec (&transformre, string, 16, matched, 0);
384     if (i == 0)
385     { /* matched */
386     }
387 pcg 1.1 }
388     }
389     #endif
390    
391 pcg 1.12 char *
392     concat_line (const char *p1, const char *p2)
393     {
394     int l1 = p1 ? strlen (p1) : 0;
395     int l2 = strlen (p2);
396     char *r = xmalloc (l1 + l2 + 1);
397    
398     memcpy (r, p1, l1);
399     memcpy (r + l1, p2, l2);
400     r[l1 + l2] = 0;
401    
402     return r;
403     }
404 pcg 1.1
405     /*
406 pcg 1.12 * This routine should read a single line, no matter how long.
407 pcg 1.1 */
408 pcg 1.11 int
409     lineinput (struct logfile_entry *logfile)
410 pcg 1.1 {
411 pcg 1.12 char buff[1024], *p = buff;
412     int ch;
413     int ofs = logfile->buf ? strlen (logfile->buf) : 0;
414 pcg 1.1
415 pcg 1.11 do
416     {
417 pcg 1.12 ch = fgetc (logfile->fp);
418 pcg 1.1
419 pcg 1.18 if (ch == '\n' || ch == EOF)
420     break;
421 pcg 1.17 else if (ch == '\r')
422 pcg 1.18 continue; /* skip */
423 pcg 1.12 else if (ch == '\t')
424     {
425     do
426     {
427     *p++ = ' ';
428     ofs++;
429     }
430     while (ofs & 7);
431     }
432     else
433     {
434     *p++ = ch;
435     ofs++;
436     }
437 pcg 1.11 }
438 pcg 1.12 while (p < buff + (sizeof buff) - 8 - 1);
439 pcg 1.1
440 pcg 1.19 if (p == buff && ch == EOF)
441 pcg 1.12 return 0;
442    
443     *p = 0;
444    
445     p = concat_line (logfile->buf, buff);
446     free (logfile->buf); logfile->buf = p;
447 chris_moore 1.9
448 pcg 1.12 logfile->lastpartial = logfile->partial;
449     logfile->partial = ch == EOF;
450    
451     if (logfile->partial && opt_whole)
452 pcg 1.11 return 0;
453 pcg 1.1
454     #if HAS_REGEX
455 pcg 1.12 transform_line (logfile->buf);
456 pcg 1.1 #endif
457 pcg 1.12 return 1;
458 pcg 1.1 }
459    
460     /* input: reads file->fname
461     * output: fills file->fp, file->inode
462     * returns file->fp
463     * in case of error, file->fp is NULL
464     */
465 pcg 1.11 FILE *
466     openlog (struct logfile_entry * file)
467     {
468     struct stat stats;
469    
470     if ((file->fp = fopen (file->fname, "r")) == NULL)
471     {
472     file->fp = NULL;
473     return NULL;
474     }
475    
476     fstat (fileno (file->fp), &stats);
477     if (S_ISFIFO (stats.st_mode))
478     {
479     if (fcntl (fileno (file->fp), F_SETFL, O_NONBLOCK) < 0)
480     perror ("fcntl"), exit (1);
481     file->inode = 0;
482     }
483     else
484     file->inode = stats.st_ino;
485    
486     if (opt_noinitial)
487     fseek (file->fp, 0, SEEK_END);
488     else if (stats.st_size > (listlen + 1) * width)
489     fseek (file->fp, -((listlen + 2) * width), SEEK_END);
490    
491     file->last_size = stats.st_size;
492     return file->fp;
493     }
494    
495     void
496     reopen (void)
497 pcg 1.1 {
498 pcg 1.11 struct logfile_entry *e;
499    
500     for (e = loglist; e; e = e->next)
501     {
502     if (!e->inode)
503     continue; /* skip stdin */
504 pcg 1.1
505 pcg 1.11 if (e->fp)
506     fclose (e->fp);
507     /* if fp is NULL we will try again later */
508     openlog (e);
509     }
510    
511     do_reopen = 0;
512     }
513 pcg 1.1
514 pcg 1.11 void
515     check_open_files (void)
516     {
517     struct logfile_entry *e;
518     struct stat stats;
519 pcg 1.1
520 pcg 1.11 for (e = loglist; e; e = e->next)
521     {
522     if (!e->inode)
523     continue; /* skip stdin */
524 pcg 1.1
525 pcg 1.11 if (stat (e->fname, &stats) < 0)
526     { /* file missing? */
527     sleep (1);
528     if (e->fp)
529     fclose (e->fp);
530     if (openlog (e) == NULL)
531     break;
532     }
533    
534     if (stats.st_ino != e->inode)
535     { /* file renamed? */
536     if (e->fp)
537     fclose (e->fp);
538     if (openlog (e) == NULL)
539     break;
540     }
541    
542     if (stats.st_size < e->last_size)
543     { /* file truncated? */
544     fseek (e->fp, 0, SEEK_SET);
545     e->last_size = stats.st_size;
546     }
547 pcg 1.1 }
548     }
549    
550 pcg 1.14 /*
551     * insert a single physical line (that must be short enough to fit)
552     * at position "idx" by pushing up lines above it. the caller
553     * MUST then fill in lines[idx] with valid data.
554     */
555 pcg 1.12 static void
556     insert_line (int idx)
557     {
558     int cur_line;
559     struct logfile_entry *current;
560    
561     free (lines[0].line);
562    
563     for (cur_line = 0; cur_line < idx; cur_line++)
564     lines[cur_line] = lines[cur_line + 1];
565    
566     for (current = loglist; current; current = current->next)
567 pcg 1.16 if (current->index <= idx)
568 pcg 1.12 current->index--;
569 pcg 1.11 }
570    
571 pcg 1.14 /*
572     * remove a single physical line at position "idx" by moving the lines above it
573     * down and inserting a "~" line at the top.
574     */
575 pcg 1.12 static void
576     delete_line (int idx)
577     {
578     int cur_line;
579     struct logfile_entry *current;
580    
581     for (cur_line = idx; cur_line > 0; cur_line--)
582     lines[cur_line] = lines[cur_line - 1];
583    
584     lines[0].line = strdup ("~");
585    
586     for (current = loglist; current; current = current->next)
587 pcg 1.16 if (current->index >= 0 && current->index <= idx)
588 pcg 1.12 current->index++;
589     }
590    
591 pcg 1.14 /*
592     * takes a logical log file line and split it into multiple physical
593     * screen lines by splitting it whenever a part becomes too long.
594     * lal lines will be inserted at position "idx".
595     */
596 pcg 1.12 static void
597     split_line (int idx, const char *str, unsigned long color)
598     {
599     int l = strlen (str);
600     const char *p = str;
601    
602     do
603     {
604     const char *beg = p;
605     int w = 0;
606    
607     while (*p)
608     {
609     int len = mblen (p, l);
610     if (len <= 0)
611     len = 1; /* ignore (don't skip) ilegal character sequences */
612    
613     int cw = XmbTextEscapement (fontset, p, len);
614     if (cw + w >= width)
615     break;
616    
617     w += cw;
618     p += len;
619     l -= len;
620     }
621    
622     {
623     char *s = xmalloc (p - beg + 1);
624     memcpy (s, beg, p - beg);
625     s[p - beg] = 0;
626     insert_line (idx);
627     lines[idx].line = s;
628     lines[idx].len = p - beg;
629     lines[idx].color = color;
630     }
631     }
632     while (l);
633     }
634    
635 pcg 1.14 /*
636     * append something to an existing physical line. this is done
637     * by deleting the file on-screen, concatenating the new data to it
638     * and splitting it again.
639     */
640 pcg 1.12 static void
641     append_line (int idx, const char *str)
642     {
643     unsigned long color = lines[idx].color;
644     char *old = lines[idx].line;
645     char *new = concat_line (old, str);
646    
647     free (old);
648    
649     delete_line (idx);
650     split_line (idx, new, color);
651     }
652    
653     static void
654 pcg 1.11 main_loop (void)
655     {
656 pcg 1.12 lines = xmalloc (sizeof (struct linematrix) * listlen);
657 pcg 1.11 int lin, miny, maxy;
658     time_t lastreload;
659     Region region = XCreateRegion ();
660     XEvent xev;
661     struct logfile_entry *lastprinted = NULL;
662     struct logfile_entry *current;
663 pcg 1.18 int need_update = 1;
664 pcg 1.11
665     maxy = 0;
666     miny = win_y + height;
667     lastreload = time (NULL);
668    
669     /* Initialize linematrix */
670     for (lin = 0; lin < listlen; lin++)
671     {
672 pcg 1.12 lines[lin].line = strdup ("~");
673     lines[lin].len = 1;
674 pcg 1.11 lines[lin].color = GetColor (def_color);
675     }
676    
677     for (;;)
678     {
679     /* read logs */
680     for (current = loglist; current; current = current->next)
681     {
682     if (!current->fp)
683     continue; /* skip missing files */
684    
685     clearerr (current->fp);
686    
687 pcg 1.12 while (lineinput (current))
688 pcg 1.11 {
689 pcg 1.12 need_update = 1;
690 pcg 1.11 /* if we're trying to update old partial lines in
691     * place, and the last time this file was updated the
692     * output was partial, and that partial line is not
693     * too close to the top of the screen, then update
694     * that partial line */
695 pcg 1.16 if (opt_update && current->lastpartial && current->index >= 0)
696 pcg 1.11 {
697 pcg 1.16 int idx = current->index;
698     append_line (idx, current->buf);
699     current->index = idx;
700 pcg 1.13 free (current->buf), current->buf = 0;
701 pcg 1.12 continue;
702 pcg 1.11 }
703    
704     /* print filename if any, and if last line was from
705     * different file */
706 pcg 1.13 if (!opt_nofilename && lastprinted != current && current->desc[0])
707 pcg 1.11 {
708 pcg 1.12 char buf[1024]; /* HACK */
709     snprintf (buf, sizeof (buf), "[%s]", current->desc);
710     split_line (listlen - 1, buf, current->color);
711 pcg 1.11 }
712    
713     /* if this is the same file we showed last, and the
714     * last time we showed it, it wasn't finished, then
715     * append to the last line shown */
716     if (lastprinted == current && current->lastpartial)
717     {
718 pcg 1.12 append_line (listlen - 1, current->buf);
719 pcg 1.13 free (current->buf), current->buf = 0;
720 pcg 1.12 continue;
721 pcg 1.11 }
722     else
723     {
724 pcg 1.12 split_line (listlen - 1, current->buf, current->color);
725 pcg 1.13 free (current->buf), current->buf = 0;
726 pcg 1.11 }
727    
728     current->index = listlen - 1;
729     lastprinted = current;
730     }
731     }
732    
733     if (need_update)
734 pcg 1.18 {
735     redraw ();
736     need_update = 0;
737     }
738 pcg 1.11 else
739     {
740     XFlush (disp);
741    
742     if (!XPending (disp))
743     {
744     fd_set fdr;
745     struct timeval to = interval;
746    
747     FD_ZERO (&fdr);
748     FD_SET (ConnectionNumber (disp), &fdr);
749     select (ConnectionNumber (disp) + 1, &fdr, 0, 0, &to);
750     }
751     }
752    
753     check_open_files ();
754    
755     if (do_reopen)
756     reopen ();
757    
758     /* we ignore possible errors due to window resizing &c */
759     while (XPending (disp))
760     {
761     XNextEvent (disp, &xev);
762    
763     switch (xev.type)
764     {
765     case Expose:
766     {
767     XRectangle r;
768    
769     r.x = xev.xexpose.x;
770     r.y = xev.xexpose.y;
771     r.width = xev.xexpose.width;
772     r.height = xev.xexpose.height;
773     XUnionRectWithRegion (&r, region, region);
774     if (miny > r.y)
775     miny = r.y;
776     if (maxy < r.y + r.height)
777     maxy = r.y + r.height;
778     }
779     break;
780     default:
781     #ifdef DEBUGMODE
782     fprintf (stderr, "PANIC! Unknown event %d\n", xev.type);
783     #endif
784     break;
785     }
786     }
787    
788     /* reload if requested */
789     if (reload && lastreload + reload < time (NULL))
790     {
791     if (command && command[0])
792     system (command);
793    
794     reopen ();
795     lastreload = time (NULL);
796     }
797    
798     if (!XEmptyRegion (region))
799     {
800     XSetRegion (disp, WinGC, region);
801    
802 pcg 1.12 refresh (miny, maxy);
803 pcg 1.11 XDestroyRegion (region);
804     region = XCreateRegion ();
805     maxy = 0;
806 pcg 1.15 miny = 32768;
807 pcg 1.11 }
808     }
809     }
810 pcg 1.1
811 pcg 1.11
812     int
813     main (int argc, char *argv[])
814 pcg 1.1 {
815 pcg 1.11 int i;
816     int opt_daemonize = 0;
817     int opt_partial = 0, file_count = 0;
818 pcg 1.1 #if HAS_REGEX
819 pcg 1.11 char *transform = NULL;
820 pcg 1.1 #endif
821    
822 pcg 1.11 setlocale (LC_CTYPE, ""); /* try to initialize the locale. */
823    
824     /* window needs to be initialized before colorlookups can be done */
825     /* just a dummy to get the color lookups right */
826     geom_mask = NoValue;
827     InitWindow ();
828    
829     for (i = 1; i < argc; i++)
830     {
831     const char *arg = argv[i];
832 pcg 1.7
833 pcg 1.11 if (arg[0] == '-' && arg[1] != '\0' && arg[1] != ',')
834     {
835     if (arg[1] == '-')
836     arg++;
837    
838     if (!strcmp (arg, "-?") ||
839     !strcmp (arg, "-help") || !strcmp (arg, "-h"))
840     display_help (argv[0]);
841     else if (!strcmp (arg, "-V"))
842     display_version ();
843     else if (!strcmp (arg, "-g") || !strcmp (arg, "-geometry"))
844     geom_mask =
845     XParseGeometry (argv[++i], &win_x, &win_y, &width, &height);
846     else if (!strcmp (arg, "-display"))
847     dispname = argv[++i];
848     else if (!strcmp (arg, "-cont"))
849     continuation = argv[++i];
850     else if (!strcmp (arg, "-font") || !strcmp (arg, "-fn"))
851     fontname = argv[++i];
852 pcg 1.1 #if HAS_REGEX
853 pcg 1.11 else if (!strcmp (arg, "-t"))
854     transform = argv[++i];
855 pcg 1.1 #endif
856 pcg 1.11 else if (!strcmp (arg, "-fork") || !strcmp (arg, "-f"))
857     opt_daemonize = 1;
858     else if (!strcmp (arg, "-reload"))
859     {
860     reload = atoi (argv[++i]);
861     command = argv[++i];
862     }
863     else if (!strcmp (arg, "-shade"))
864     opt_shade = 1;
865     else if (!strcmp (arg, "-frame"))
866     opt_frame = 1;
867     else if (!strcmp (arg, "-no-filename"))
868     opt_nofilename = 1;
869     else if (!strcmp (arg, "-reverse"))
870     opt_reverse = 1;
871     else if (!strcmp (arg, "-whole"))
872     opt_whole = 1;
873     else if (!strcmp (arg, "-partial"))
874     opt_partial = 1;
875     else if (!strcmp (arg, "-update"))
876     opt_update = opt_partial = 1;
877     else if (!strcmp (arg, "-color"))
878     def_color = argv[++i];
879     else if (!strcmp (arg, "-noinitial"))
880     opt_noinitial = 1;
881     else if (!strcmp (arg, "-id"))
882     root = atoi (argv[++i]);
883     else if (!strcmp (arg, "-interval") || !strcmp (arg, "-i"))
884     {
885     double iv = atof (argv[++i]);
886    
887     interval.tv_sec = (int) iv;
888     interval.tv_usec = (iv - interval.tv_sec) * 1e6;
889     }
890     else
891     {
892     fprintf (stderr, "Unknown option '%s'.\n"
893     "Try --help for more information.\n", arg);
894     exit (1);
895     }
896     }
897     else
898     { /* it must be a filename */
899     struct logfile_entry *e;
900     const char *fname, *desc, *fcolor = def_color;
901     char *p;
902    
903     file_count++;
904    
905     /* this is not foolproof yet (',' in filenames are not allowed) */
906     fname = desc = arg;
907     if ((p = strchr (arg, ',')))
908     {
909     *p = '\0';
910     fcolor = p + 1;
911    
912     if ((p = strchr (fcolor, ',')))
913     {
914     *p = '\0';
915     desc = p + 1;
916     }
917     }
918    
919     e = xmalloc (sizeof (struct logfile_entry));
920 pcg 1.12 e->partial = 0;
921     e->buf = 0;
922 pcg 1.16 e->index = -1;
923 pcg 1.12
924 pcg 1.11 if (arg[0] == '-' && arg[1] == '\0')
925     {
926     if ((e->fp = fdopen (0, "r")) == NULL)
927     perror ("fdopen"), exit (1);
928     if (fcntl (0, F_SETFL, O_NONBLOCK) < 0)
929     perror ("fcntl"), exit (1);
930     e->fname = NULL;
931     e->inode = 0;
932     e->desc = xstrdup ("stdin");
933     }
934     else
935     {
936     int l;
937    
938     e->fname = xstrdup (fname);
939     if (openlog (e) == NULL)
940     perror (fname), exit (1);
941    
942     l = strlen (desc);
943     if (l > width - 2) /* must account for [ ] */
944     l = width - 2;
945     e->desc = xmalloc (l + 1);
946     memcpy (e->desc, desc, l);
947     *(e->desc + l) = '\0';
948     }
949    
950     e->color = GetColor (fcolor);
951     e->partial = 0;
952     e->next = NULL;
953    
954     if (!loglist)
955     loglist = e;
956     if (loglist_tail)
957     loglist_tail->next = e;
958     loglist_tail = e;
959     }
960     }
961    
962     if (!loglist)
963     {
964     fprintf (stderr, "You did not specify any files to tail\n"
965     "use %s --help for help\n", argv[0]);
966     exit (1);
967     }
968    
969     if (opt_partial && opt_whole)
970     {
971     fprintf (stderr, "Specify at most one of -partial and -whole\n");
972     exit (1);
973 chris_moore 1.9 }
974    
975 pcg 1.11 if (opt_partial)
976 chris_moore 1.9 /* if we specifically requested to see partial lines then don't insist on whole lines */
977 pcg 1.11 opt_whole = 0;
978     else if (file_count > 1)
979 chris_moore 1.9 /* otherwise, if we've viewing multiple files, default to showing whole lines */
980 pcg 1.11 opt_whole = 1;
981 chris_moore 1.9
982 pcg 1.1 #if HAS_REGEX
983 pcg 1.11 if (transform)
984     {
985     int i;
986 pcg 1.1
987 pcg 1.11 transformre = xmalloc (sizeof (transformre));
988     i = regcomp (&transformre, transform, REG_EXTENDED);
989     if (i != 0)
990     {
991     char buf[512];
992    
993     regerror (i, &transformre, buf, sizeof (buf));
994     fprintf (stderr, "Cannot compile regular expression: %s\n", buf);
995     }
996 pcg 1.1 }
997     #endif
998    
999 pcg 1.11 InitWindow ();
1000 pcg 1.1
1001 pcg 1.11 install_signal (SIGINT, blank_window);
1002     install_signal (SIGQUIT, blank_window);
1003     install_signal (SIGTERM, blank_window);
1004     install_signal (SIGHUP, force_reopen);
1005     install_signal (SIGUSR1, list_files);
1006     install_signal (SIGUSR2, force_refresh);
1007 pcg 1.1
1008 pcg 1.11 if (opt_daemonize)
1009     daemonize ();
1010 pcg 1.1
1011 pcg 1.11 main_loop ();
1012 pcg 1.1
1013 pcg 1.11 exit (1); /* to make gcc -Wall stop complaining */
1014 pcg 1.1 }
1015    
1016 pcg 1.11 void
1017     install_signal (int sig, void (*handler) (int))
1018 pcg 1.1 {
1019 pcg 1.11 struct sigaction action;
1020    
1021     action.sa_handler = handler;
1022     sigemptyset (&action.sa_mask);
1023     action.sa_flags = SA_RESTART;
1024 pcg 1.1
1025 pcg 1.11 if (sigaction (sig, &action, NULL) < 0)
1026     fprintf (stderr, "sigaction(%d): %s\n", sig, strerror (errno)), exit (1);
1027 pcg 1.1 }
1028    
1029 pcg 1.11 void *
1030     xstrdup (const char *string)
1031 pcg 1.1 {
1032 pcg 1.11 void *p;
1033 pcg 1.1
1034 pcg 1.11 while ((p = strdup (string)) == NULL)
1035     {
1036     fprintf (stderr, "Memory exausted.");
1037     sleep (10);
1038 pcg 1.1 }
1039 pcg 1.11
1040     return p;
1041 pcg 1.1 }
1042    
1043 pcg 1.11 void *
1044     xmalloc (size_t size)
1045 pcg 1.1 {
1046 pcg 1.11 void *p;
1047 pcg 1.1
1048 pcg 1.11 while ((p = malloc (size)) == NULL)
1049     {
1050     fprintf (stderr, "Memory exausted.");
1051     sleep (10);
1052 pcg 1.1 }
1053 pcg 1.12
1054 pcg 1.11 return p;
1055 pcg 1.1 }
1056    
1057 pcg 1.11 void
1058     display_help (char *myname)
1059     {
1060     printf ("Usage: %s [options] file1[,color[,desc]] "
1061     "[file2[,color[,desc]] ...]\n", myname);
1062     printf (" -g | -geometry geometry -g WIDTHxHEIGHT+X+Y\n"
1063     " -color color use color $color as default\n"
1064     " -reload sec command reload after $sec and run command\n"
1065     " -id id window id to use instead of the root window\n"
1066     " -font FONTSPEC (-fn) font to use\n"
1067     " -f | -fork fork into background\n"
1068     " -reverse print new lines at the top\n"
1069     " -whole wait for \\n before showing a line\n"
1070     " -partial show lines even if they don't end with a \\n\n"
1071     " -update allow updates to old partial lines\n"
1072     " -cont string to prefix continued partial lines with\n"
1073     " defaults to \"[+]\"\n"
1074     " -shade add shading to font\n"
1075     " -noinitial don't display the last file lines on\n"
1076     " startup\n"
1077     " -i | -interval seconds interval between checks (fractional\n"
1078     " values o.k.). Default 3 seconds\n"
1079     " -V display version information and exit\n"
1080     "\n");
1081     printf ("Example:\n%s -g 80x25+100+50 -font fixed /var/log/messages,green "
1082     "/var/log/secure,red,'ALERT'\n", myname);
1083     exit (0);
1084     }
1085    
1086     void
1087     display_version (void)
1088 pcg 1.1 {
1089 pcg 1.11 printf ("root-tail version " VERSION "\n");
1090     exit (0);
1091 pcg 1.1 }
1092    
1093 pcg 1.11 int
1094     daemonize (void)
1095     {
1096     switch (fork ())
1097     {
1098 pcg 1.1 case -1:
1099 pcg 1.11 return -1;
1100 pcg 1.1 case 0:
1101 pcg 1.11 break;
1102 pcg 1.1 default:
1103 pcg 1.11 _exit (0);
1104 pcg 1.1 }
1105    
1106 pcg 1.11 if (setsid () == -1)
1107     return -1;
1108 pcg 1.1
1109 pcg 1.11 return 0;
1110 pcg 1.1 }