Add the DragonFly cvs id and perform general cleanups on cvs/rcs/sccs ids. Most
[dragonfly.git] / usr.sbin / cron / lib / misc.c
1 /* Copyright 1988,1990,1993,1994 by Paul Vixie
2  * All rights reserved
3  *
4  * Distribute freely, except: don't remove my name from the source or
5  * documentation (don't take credit for my work), mark your changes (don't
6  * get me blamed for your possible bugs), don't alter or remove this
7  * notice.  May be sold if buildable source is provided to buyer.  No
8  * warrantee of any kind, express or implied, is included with this
9  * software; use at your own risk, responsibility for damages (if any) to
10  * anyone resulting from the use of this software rests entirely with the
11  * user.
12  *
13  * Send bug reports, bug fixes, enhancements, requests, flames, etc., and
14  * I'll try to keep a version up to date.  I can be reached as follows:
15  * Paul Vixie          <paul@vix.com>          uunet!decwrl!vixie!paul
16  *
17  * $FreeBSD: src/usr.sbin/cron/lib/misc.c,v 1.8.2.2 2002/04/28 22:45:53 dwmalone Exp $
18  * $DragonFly: src/usr.sbin/cron/lib/misc.c,v 1.2 2003/06/17 04:29:53 dillon Exp $
19  */
20
21 /* vix 26jan87 [RCS has the rest of the log]
22  * vix 30dec86 [written]
23  */
24
25
26 #include "cron.h"
27 #if SYS_TIME_H
28 # include <sys/time.h>
29 #else
30 # include <time.h>
31 #endif
32 #include <sys/file.h>
33 #include <sys/stat.h>
34 #include <err.h>
35 #include <errno.h>
36 #include <string.h>
37 #include <fcntl.h>
38 #if defined(SYSLOG)
39 # include <syslog.h>
40 #endif
41
42
43 #if defined(LOG_DAEMON) && !defined(LOG_CRON)
44 #define LOG_CRON LOG_DAEMON
45 #endif
46
47
48 static int              LogFD = ERR;
49
50
51 int
52 strcmp_until(left, right, until)
53         char    *left;
54         char    *right;
55         int     until;
56 {
57         register int    diff;
58
59         while (*left && *left != until && *left == *right) {
60                 left++;
61                 right++;
62         }
63
64         if ((*left=='\0' || *left == until) &&
65             (*right=='\0' || *right == until)) {
66                 diff = 0;
67         } else {
68                 diff = *left - *right;
69         }
70
71         return diff;
72 }
73
74
75 /* strdtb(s) - delete trailing blanks in string 's' and return new length
76  */
77 int
78 strdtb(s)
79         char    *s;
80 {
81         char    *x = s;
82
83         /* scan forward to the null
84          */
85         while (*x)
86                 x++;
87
88         /* scan backward to either the first character before the string,
89          * or the last non-blank in the string, whichever comes first.
90          */
91         do      {x--;}
92         while (x >= s && isspace(*x));
93
94         /* one character beyond where we stopped above is where the null
95          * goes.
96          */
97         *++x = '\0';
98
99         /* the difference between the position of the null character and
100          * the position of the first character of the string is the length.
101          */
102         return x - s;
103 }
104
105
106 int
107 set_debug_flags(flags)
108         char    *flags;
109 {
110         /* debug flags are of the form    flag[,flag ...]
111          *
112          * if an error occurs, print a message to stdout and return FALSE.
113          * otherwise return TRUE after setting ERROR_FLAGS.
114          */
115
116 #if !DEBUGGING
117
118         printf("this program was compiled without debugging enabled\n");
119         return FALSE;
120
121 #else /* DEBUGGING */
122
123         char    *pc = flags;
124
125         DebugFlags = 0;
126
127         while (*pc) {
128                 char    **test;
129                 int     mask;
130
131                 /* try to find debug flag name in our list.
132                  */
133                 for (   test = DebugFlagNames, mask = 1;
134                         *test && strcmp_until(*test, pc, ',');
135                         test++, mask <<= 1
136                     )
137                         ;
138
139                 if (!*test) {
140                         fprintf(stderr,
141                                 "unrecognized debug flag <%s> <%s>\n",
142                                 flags, pc);
143                         return FALSE;
144                 }
145
146                 DebugFlags |= mask;
147
148                 /* skip to the next flag
149                  */
150                 while (*pc && *pc != ',')
151                         pc++;
152                 if (*pc == ',')
153                         pc++;
154         }
155
156         if (DebugFlags) {
157                 int     flag;
158
159                 fprintf(stderr, "debug flags enabled:");
160
161                 for (flag = 0;  DebugFlagNames[flag];  flag++)
162                         if (DebugFlags & (1 << flag))
163                                 fprintf(stderr, " %s", DebugFlagNames[flag]);
164                 fprintf(stderr, "\n");
165         }
166
167         return TRUE;
168
169 #endif /* DEBUGGING */
170 }
171
172
173 void
174 set_cron_uid()
175 {
176 #if defined(BSD) || defined(POSIX)
177         if (seteuid(ROOT_UID) < OK)
178                 err(ERROR_EXIT, "seteuid");
179 #else
180         if (setuid(ROOT_UID) < OK)
181                 err(ERROR_EXIT, "setuid");
182 #endif
183 }
184
185
186 void
187 set_cron_cwd()
188 {
189         struct stat     sb;
190
191         /* first check for CRONDIR ("/var/cron" or some such)
192          */
193         if (stat(CRONDIR, &sb) < OK && errno == ENOENT) {
194                 warn("%s", CRONDIR);
195                 if (OK == mkdir(CRONDIR, 0700)) {
196                         warnx("%s: created", CRONDIR);
197                         stat(CRONDIR, &sb);
198                 } else {
199                         err(ERROR_EXIT, "%s: mkdir", CRONDIR);
200                 }
201         }
202         if (!(sb.st_mode & S_IFDIR))
203                 err(ERROR_EXIT, "'%s' is not a directory, bailing out", CRONDIR);
204         if (chdir(CRONDIR) < OK)
205                 err(ERROR_EXIT, "cannot chdir(%s), bailing out", CRONDIR);
206
207         /* CRONDIR okay (now==CWD), now look at SPOOL_DIR ("tabs" or some such)
208          */
209         if (stat(SPOOL_DIR, &sb) < OK && errno == ENOENT) {
210                 warn("%s", SPOOL_DIR);
211                 if (OK == mkdir(SPOOL_DIR, 0700)) {
212                         warnx("%s: created", SPOOL_DIR);
213                         stat(SPOOL_DIR, &sb);
214                 } else {
215                         err(ERROR_EXIT, "%s: mkdir", SPOOL_DIR);
216                 }
217         }
218         if (!(sb.st_mode & S_IFDIR))
219                 err(ERROR_EXIT, "'%s' is not a directory, bailing out", SPOOL_DIR);
220 }
221
222
223 /* acquire_daemonlock() - write our PID into /etc/cron.pid, unless
224  *      another daemon is already running, which we detect here.
225  *
226  * note: main() calls us twice; once before forking, once after.
227  *      we maintain static storage of the file pointer so that we
228  *      can rewrite our PID into the PIDFILE after the fork.
229  *
230  * it would be great if fflush() disassociated the file buffer.
231  */
232 void
233 acquire_daemonlock(closeflag)
234         int closeflag;
235 {
236         static  FILE    *fp = NULL;
237
238         if (closeflag && fp) {
239                 fclose(fp);
240                 fp = NULL;
241                 return;
242         }
243
244         if (!fp) {
245                 char    pidfile[MAX_FNAME];
246                 char    buf[MAX_TEMPSTR];
247                 int     fd, otherpid;
248
249                 (void) sprintf(pidfile, PIDFILE, PIDDIR);
250                 if ((-1 == (fd = open(pidfile, O_RDWR|O_CREAT, 0644)))
251                     || (NULL == (fp = fdopen(fd, "r+")))
252                     ) {
253                         sprintf(buf, "can't open or create %s: %s",
254                                 pidfile, strerror(errno));
255                         log_it("CRON", getpid(), "DEATH", buf);
256                         errx(ERROR_EXIT, "%s", buf);
257                 }
258
259                 if (flock(fd, LOCK_EX|LOCK_NB) < OK) {
260                         int save_errno = errno;
261
262                         fscanf(fp, "%d", &otherpid);
263                         sprintf(buf, "can't lock %s, otherpid may be %d: %s",
264                                 pidfile, otherpid, strerror(save_errno));
265                         log_it("CRON", getpid(), "DEATH", buf);
266                         errx(ERROR_EXIT, "%s", buf);
267                 }
268
269                 (void) fcntl(fd, F_SETFD, 1);
270         }
271
272         rewind(fp);
273         fprintf(fp, "%d\n", getpid());
274         fflush(fp);
275         (void) ftruncate(fileno(fp), ftell(fp));
276
277         /* abandon fd and fp even though the file is open. we need to
278          * keep it open and locked, but we don't need the handles elsewhere.
279          */
280 }
281
282 /* get_char(file) : like getc() but increment LineNumber on newlines
283  */
284 int
285 get_char(file)
286         FILE    *file;
287 {
288         int     ch;
289
290         ch = getc(file);
291         if (ch == '\n')
292                 Set_LineNum(LineNumber + 1)
293         return ch;
294 }
295
296
297 /* unget_char(ch, file) : like ungetc but do LineNumber processing
298  */
299 void
300 unget_char(ch, file)
301         int     ch;
302         FILE    *file;
303 {
304         ungetc(ch, file);
305         if (ch == '\n')
306                 Set_LineNum(LineNumber - 1)
307 }
308
309
310 /* get_string(str, max, file, termstr) : like fgets() but
311  *              (1) has terminator string which should include \n
312  *              (2) will always leave room for the null
313  *              (3) uses get_char() so LineNumber will be accurate
314  *              (4) returns EOF or terminating character, whichever
315  */
316 int
317 get_string(string, size, file, terms)
318         char    *string;
319         int     size;
320         FILE    *file;
321         char    *terms;
322 {
323         int     ch;
324
325         while (EOF != (ch = get_char(file)) && !strchr(terms, ch)) {
326                 if (size > 1) {
327                         *string++ = (char) ch;
328                         size--;
329                 }
330         }
331
332         if (size > 0)
333                 *string = '\0';
334
335         return ch;
336 }
337
338
339 /* skip_comments(file) : read past comment (if any)
340  */
341 void
342 skip_comments(file)
343         FILE    *file;
344 {
345         int     ch;
346
347         while (EOF != (ch = get_char(file))) {
348                 /* ch is now the first character of a line.
349                  */
350
351                 while (ch == ' ' || ch == '\t')
352                         ch = get_char(file);
353
354                 if (ch == EOF)
355                         break;
356
357                 /* ch is now the first non-blank character of a line.
358                  */
359
360                 if (ch != '\n' && ch != '#')
361                         break;
362
363                 /* ch must be a newline or comment as first non-blank
364                  * character on a line.
365                  */
366
367                 while (ch != '\n' && ch != EOF)
368                         ch = get_char(file);
369
370                 /* ch is now the newline of a line which we're going to
371                  * ignore.
372                  */
373         }
374         if (ch != EOF)
375                 unget_char(ch, file);
376 }
377
378
379 /* int in_file(char *string, FILE *file)
380  *      return TRUE if one of the lines in file matches string exactly,
381  *      FALSE otherwise.
382  */
383 static int
384 in_file(string, file)
385         char *string;
386         FILE *file;
387 {
388         char line[MAX_TEMPSTR];
389
390         rewind(file);
391         while (fgets(line, MAX_TEMPSTR, file)) {
392                 if (line[0] != '\0')
393                         if (line[strlen(line)-1] == '\n')
394                                 line[strlen(line)-1] = '\0';
395                 if (0 == strcmp(line, string))
396                         return TRUE;
397         }
398         return FALSE;
399 }
400
401
402 /* int allowed(char *username)
403  *      returns TRUE if (ALLOW_FILE exists and user is listed)
404  *      or (DENY_FILE exists and user is NOT listed)
405  *      or (neither file exists but user=="root" so it's okay)
406  */
407 int
408 allowed(username)
409         char *username;
410 {
411         static int      init = FALSE;
412         static FILE     *allow, *deny;
413
414         if (!init) {
415                 init = TRUE;
416 #if defined(ALLOW_FILE) && defined(DENY_FILE)
417                 allow = fopen(ALLOW_FILE, "r");
418                 deny = fopen(DENY_FILE, "r");
419                 Debug(DMISC, ("allow/deny enabled, %d/%d\n", !!allow, !!deny))
420 #else
421                 allow = NULL;
422                 deny = NULL;
423 #endif
424         }
425
426         if (allow)
427                 return (in_file(username, allow));
428         if (deny)
429                 return (!in_file(username, deny));
430
431 #if defined(ALLOW_ONLY_ROOT)
432         return (strcmp(username, ROOT_USER) == 0);
433 #else
434         return TRUE;
435 #endif
436 }
437
438
439 void
440 log_it(username, xpid, event, detail)
441         char    *username;
442         int     xpid;
443         char    *event;
444         char    *detail;
445 {
446         PID_T                   pid = xpid;
447 #if defined(LOG_FILE)
448         char                    *msg;
449         TIME_T                  now = time((TIME_T) 0);
450         register struct tm      *t = localtime(&now);
451 #endif /*LOG_FILE*/
452
453 #if defined(SYSLOG)
454         static int              syslog_open = 0;
455 #endif
456
457 #if defined(LOG_FILE)
458         /* we assume that MAX_TEMPSTR will hold the date, time, &punctuation.
459          */
460         msg = malloc(strlen(username)
461                      + strlen(event)
462                      + strlen(detail)
463                      + MAX_TEMPSTR);
464
465         if (msg == NULL)
466                 warnx("failed to allocate memory for log message");
467         else {
468                 if (LogFD < OK) {
469                         LogFD = open(LOG_FILE, O_WRONLY|O_APPEND|O_CREAT, 0600);
470                         if (LogFD < OK) {
471                                 warn("can't open log file %s", LOG_FILE);
472                         } else {
473                                 (void) fcntl(LogFD, F_SETFD, 1);
474                         }
475                 }
476
477                 /* we have to sprintf() it because fprintf() doesn't always
478                  * write everything out in one chunk and this has to be
479                  * atomically appended to the log file.
480                  */
481                 sprintf(msg, "%s (%02d/%02d-%02d:%02d:%02d-%d) %s (%s)\n",
482                         username,
483                         t->tm_mon+1, t->tm_mday, t->tm_hour, t->tm_min,
484                         t->tm_sec, pid, event, detail);
485
486                 /* we have to run strlen() because sprintf() returns (char*)
487                  * on old BSD.
488                  */
489                 if (LogFD < OK || write(LogFD, msg, strlen(msg)) < OK) {
490                         if (LogFD >= OK)
491                                 warn("%s", LOG_FILE);
492                         warnx("can't write to log file");
493                         write(STDERR, msg, strlen(msg));
494                 }
495
496                 free(msg);
497         }
498 #endif /*LOG_FILE*/
499
500 #if defined(SYSLOG)
501         if (!syslog_open) {
502                 /* we don't use LOG_PID since the pid passed to us by
503                  * our client may not be our own.  therefore we want to
504                  * print the pid ourselves.
505                  */
506 # ifdef LOG_DAEMON
507                 openlog(ProgramName, LOG_PID, LOG_CRON);
508 # else
509                 openlog(ProgramName, LOG_PID);
510 # endif
511                 syslog_open = TRUE;             /* assume openlog success */
512         }
513
514         syslog(LOG_INFO, "(%s) %s (%s)\n", username, event, detail);
515
516 #endif /*SYSLOG*/
517
518 #if DEBUGGING
519         if (DebugFlags) {
520                 fprintf(stderr, "log_it: (%s %d) %s (%s)\n",
521                         username, pid, event, detail);
522         }
523 #endif
524 }
525
526
527 void
528 log_close() {
529         if (LogFD != ERR) {
530                 close(LogFD);
531                 LogFD = ERR;
532         }
533 }
534
535
536 /* two warnings:
537  *      (1) this routine is fairly slow
538  *      (2) it returns a pointer to static storage
539  */
540 char *
541 first_word(s, t)
542         register char *s;       /* string we want the first word of */
543         register char *t;       /* terminators, implicitly including \0 */
544 {
545         static char retbuf[2][MAX_TEMPSTR + 1]; /* sure wish C had GC */
546         static int retsel = 0;
547         register char *rb, *rp;
548
549         /* select a return buffer */
550         retsel = 1-retsel;
551         rb = &retbuf[retsel][0];
552         rp = rb;
553
554         /* skip any leading terminators */
555         while (*s && (NULL != strchr(t, *s))) {
556                 s++;
557         }
558
559         /* copy until next terminator or full buffer */
560         while (*s && (NULL == strchr(t, *s)) && (rp < &rb[MAX_TEMPSTR])) {
561                 *rp++ = *s++;
562         }
563
564         /* finish the return-string and return it */
565         *rp = '\0';
566         return rb;
567 }
568
569
570 /* warning:
571  *      heavily ascii-dependent.
572  */
573 void
574 mkprint(dst, src, len)
575         register char *dst;
576         register unsigned char *src;
577         register int len;
578 {
579         while (len-- > 0)
580         {
581                 register unsigned char ch = *src++;
582
583                 if (ch < ' ') {                 /* control character */
584                         *dst++ = '^';
585                         *dst++ = ch + '@';
586                 } else if (ch < 0177) {         /* printable */
587                         *dst++ = ch;
588                 } else if (ch == 0177) {        /* delete/rubout */
589                         *dst++ = '^';
590                         *dst++ = '?';
591                 } else {                        /* parity character */
592                         sprintf(dst, "\\%03o", ch);
593                         dst += 4;
594                 }
595         }
596         *dst = '\0';
597 }
598
599
600 /* warning:
601  *      returns a pointer to malloc'd storage, you must call free yourself.
602  */
603 char *
604 mkprints(src, len)
605         register unsigned char *src;
606         register unsigned int len;
607 {
608         register char *dst = malloc(len*4 + 1);
609
610         if (dst != NULL)
611                 mkprint(dst, src, len);
612
613         return dst;
614 }
615
616
617 #ifdef MAIL_DATE
618 /* Sat, 27 Feb 93 11:44:51 CST
619  * 123456789012345678901234567
620  */
621 char *
622 arpadate(clock)
623         time_t *clock;
624 {
625         time_t t = clock ?*clock :time(0L);
626         struct tm *tm = localtime(&t);
627         static char ret[32];    /* zone name might be >3 chars */
628
629         if (tm->tm_year >= 100)
630                 tm->tm_year += 1900;
631
632         (void) snprintf(ret, sizeof(ret), "%s, %2d %s %d %02d:%02d:%02d %s",
633                        DowNames[tm->tm_wday],
634                        tm->tm_mday,
635                        MonthNames[tm->tm_mon],
636                        tm->tm_year,
637                        tm->tm_hour,
638                        tm->tm_min,
639                        tm->tm_sec,
640                        TZONE(*tm));
641         return ret;
642 }
643 #endif /*MAIL_DATE*/
644
645
646 #ifdef HAVE_SAVED_UIDS
647 static int save_euid;
648 int swap_uids() { save_euid = geteuid(); return seteuid(getuid()); }
649 int swap_uids_back() { return seteuid(save_euid); }
650 #else /*HAVE_SAVED_UIDS*/
651 int swap_uids() { return setreuid(geteuid(), getuid()); }
652 int swap_uids_back() { return swap_uids(); }
653 #endif /*HAVE_SAVED_UIDS*/