Annotation of embedaddon/ntp/ntpd/ntp_filegen.c, revision 1.1
1.1 ! misho 1: /*
! 2: * ntp_filegen.c,v 3.12 1994/01/25 19:06:11 kardel Exp
! 3: *
! 4: * implements file generations support for NTP
! 5: * logfiles and statistic files
! 6: *
! 7: *
! 8: * Copyright (C) 1992, 1996 by Rainer Pruy
! 9: * Friedrich-Alexander Universität Erlangen-Nürnberg, Germany
! 10: *
! 11: * This code may be modified and used freely
! 12: * provided credits remain intact.
! 13: */
! 14:
! 15: #ifdef HAVE_CONFIG_H
! 16: # include <config.h>
! 17: #endif
! 18:
! 19: #include <stdio.h>
! 20: #include <sys/types.h>
! 21: #include <sys/stat.h>
! 22:
! 23: #include "ntpd.h"
! 24: #include "ntp_io.h"
! 25: #include "ntp_string.h"
! 26: #include "ntp_calendar.h"
! 27: #include "ntp_filegen.h"
! 28: #include "ntp_stdlib.h"
! 29:
! 30: /*
! 31: * NTP is intended to run long periods of time without restart.
! 32: * Thus log and statistic files generated by NTP will grow large.
! 33: *
! 34: * this set of routines provides a central interface
! 35: * to generating files using file generations
! 36: *
! 37: * the generation of a file is changed according to file generation type
! 38: */
! 39:
! 40:
! 41: /*
! 42: * redefine this if your system dislikes filename suffixes like
! 43: * X.19910101 or X.1992W50 or ....
! 44: */
! 45: #define SUFFIX_SEP '.'
! 46:
! 47: static void filegen_open (FILEGEN *, u_long);
! 48: static int valid_fileref (const char *, const char *);
! 49: static void filegen_init (const char *, const char *, FILEGEN *);
! 50: #ifdef DEBUG
! 51: static void filegen_uninit (FILEGEN *);
! 52: #endif /* DEBUG */
! 53:
! 54:
! 55: /*
! 56: * filegen_init
! 57: */
! 58:
! 59: static void
! 60: filegen_init(
! 61: const char * prefix,
! 62: const char * basename,
! 63: FILEGEN * fgp
! 64: )
! 65: {
! 66: fgp->fp = NULL;
! 67: fgp->prefix = prefix; /* Yes, this is TOTALLY lame! */
! 68: fgp->basename = estrdup(basename);
! 69: fgp->id = 0;
! 70: fgp->type = FILEGEN_DAY;
! 71: fgp->flag = FGEN_FLAG_LINK; /* not yet enabled !!*/
! 72: }
! 73:
! 74:
! 75: /*
! 76: * filegen_uninit - free memory allocated by filegen_init
! 77: */
! 78: #ifdef DEBUG
! 79: static void
! 80: filegen_uninit(
! 81: FILEGEN * fgp
! 82: )
! 83: {
! 84: free(fgp->basename);
! 85: }
! 86: #endif
! 87:
! 88:
! 89: /*
! 90: * open a file generation according to the current settings of gen
! 91: * will also provide a link to basename if requested to do so
! 92: */
! 93:
! 94: static void
! 95: filegen_open(
! 96: FILEGEN * gen,
! 97: u_long newid
! 98: )
! 99: {
! 100: char *filename;
! 101: char *basename;
! 102: u_int len;
! 103: FILE *fp;
! 104: struct calendar cal;
! 105:
! 106: len = strlen(gen->prefix) + strlen(gen->basename) + 1;
! 107: basename = emalloc(len);
! 108: snprintf(basename, len, "%s%s", gen->prefix, gen->basename);
! 109:
! 110: switch (gen->type) {
! 111:
! 112: default:
! 113: msyslog(LOG_ERR,
! 114: "unsupported file generations type %d for "
! 115: "\"%s\" - reverting to FILEGEN_NONE",
! 116: gen->type, basename);
! 117: gen->type = FILEGEN_NONE;
! 118: /* fall through to FILEGEN_NONE */
! 119:
! 120: case FILEGEN_NONE:
! 121: filename = estrdup(basename);
! 122: break;
! 123:
! 124: case FILEGEN_PID:
! 125: filename = emalloc(len + 1 + 1 + 10);
! 126: snprintf(filename, len + 1 + 1 + 10,
! 127: "%s%c#%ld",
! 128: basename, SUFFIX_SEP, newid);
! 129: break;
! 130:
! 131: case FILEGEN_DAY:
! 132: /*
! 133: * You can argue here in favor of using MJD, but I
! 134: * would assume it to be easier for humans to interpret
! 135: * dates in a format they are used to in everyday life.
! 136: */
! 137: caljulian(newid, &cal);
! 138: filename = emalloc(len + 1 + 4 + 2 + 2);
! 139: snprintf(filename, len + 1 + 4 + 2 + 2,
! 140: "%s%c%04d%02d%02d",
! 141: basename, SUFFIX_SEP,
! 142: cal.year, cal.month, cal.monthday);
! 143: break;
! 144:
! 145: case FILEGEN_WEEK:
! 146: /*
! 147: * This is still a hack
! 148: * - the term week is not correlated to week as it is used
! 149: * normally - it just refers to a period of 7 days
! 150: * starting at Jan 1 - 'weeks' are counted starting from zero
! 151: */
! 152: caljulian(newid, &cal);
! 153: filename = emalloc(len + 1 + 4 + 1 + 2);
! 154: snprintf(filename, len + 1 + 4 + 1 + 2,
! 155: "%s%c%04dw%02d",
! 156: basename, SUFFIX_SEP,
! 157: cal.year, cal.yearday / 7);
! 158: break;
! 159:
! 160: case FILEGEN_MONTH:
! 161: caljulian(newid, &cal);
! 162: filename = emalloc(len + 1 + 4 + 2);
! 163: snprintf(filename, len + 1 + 4 + 2,
! 164: "%s%c%04d%02d",
! 165: basename, SUFFIX_SEP, cal.year, cal.month);
! 166: break;
! 167:
! 168: case FILEGEN_YEAR:
! 169: caljulian(newid, &cal);
! 170: filename = emalloc(len + 1 + 4);
! 171: snprintf(filename, len + 1 + 4,
! 172: "%s%c%04d",
! 173: basename, SUFFIX_SEP, cal.year);
! 174: break;
! 175:
! 176: case FILEGEN_AGE:
! 177: filename = emalloc(len + 1 + 2 + 10);
! 178: snprintf(filename, len + 1 + 2 + 10,
! 179: "%s%ca%08ld",
! 180: basename, SUFFIX_SEP, newid);
! 181: }
! 182:
! 183: if (FILEGEN_NONE != gen->type) {
! 184: /*
! 185: * check for existence of a file with name 'basename'
! 186: * as we disallow such a file
! 187: * if FGEN_FLAG_LINK is set create a link
! 188: */
! 189: struct stat stats;
! 190: /*
! 191: * try to resolve name collisions
! 192: */
! 193: static u_long conflicts = 0;
! 194:
! 195: #ifndef S_ISREG
! 196: #define S_ISREG(mode) (((mode) & S_IFREG) == S_IFREG)
! 197: #endif
! 198: if (stat(basename, &stats) == 0) {
! 199: /* Hm, file exists... */
! 200: if (S_ISREG(stats.st_mode)) {
! 201: if (stats.st_nlink <= 1) {
! 202: /*
! 203: * Oh, it is not linked - try to save it
! 204: */
! 205: char *savename;
! 206:
! 207: savename = emalloc(len + 1 + 1 + 10 + 10);
! 208: snprintf(savename, len + 1 + 1 + 10 + 10,
! 209: "%s%c%dC%lu",
! 210: basename, SUFFIX_SEP,
! 211: (int)getpid(), conflicts++);
! 212:
! 213: if (rename(basename, savename) != 0)
! 214: msyslog(LOG_ERR,
! 215: "couldn't save %s: %m",
! 216: basename);
! 217: free(savename);
! 218: } else {
! 219: /*
! 220: * there is at least a second link to
! 221: * this file.
! 222: * just remove the conflicting one
! 223: */
! 224: if (
! 225: #if !defined(VMS)
! 226: unlink(basename) != 0
! 227: #else
! 228: delete(basename) != 0
! 229: #endif
! 230: )
! 231: msyslog(LOG_ERR,
! 232: "couldn't unlink %s: %m",
! 233: basename);
! 234: }
! 235: } else {
! 236: /*
! 237: * Ehh? Not a regular file ?? strange !!!!
! 238: */
! 239: msyslog(LOG_ERR,
! 240: "expected regular file for %s "
! 241: "(found mode 0%lo)",
! 242: basename,
! 243: (unsigned long)stats.st_mode);
! 244: }
! 245: } else {
! 246: /*
! 247: * stat(..) failed, but it is absolutely correct for
! 248: * 'basename' not to exist
! 249: */
! 250: if (ENOENT != errno)
! 251: msyslog(LOG_ERR, "stat(%s) failed: %m",
! 252: basename);
! 253: }
! 254: }
! 255:
! 256: /*
! 257: * now, try to open new file generation...
! 258: */
! 259: fp = fopen(filename, "a");
! 260:
! 261: DPRINTF(4, ("opening filegen (type=%d/id=%lu) \"%s\"\n",
! 262: gen->type, newid, filename));
! 263:
! 264: if (NULL == fp) {
! 265: /* open failed -- keep previous state
! 266: *
! 267: * If the file was open before keep the previous generation.
! 268: * This will cause output to end up in the 'wrong' file,
! 269: * but I think this is still better than losing output
! 270: *
! 271: * ignore errors due to missing directories
! 272: */
! 273:
! 274: if (ENOENT != errno)
! 275: msyslog(LOG_ERR, "can't open %s: %m", filename);
! 276: } else {
! 277: if (NULL != gen->fp) {
! 278: fclose(gen->fp);
! 279: gen->fp = NULL;
! 280: }
! 281: gen->fp = fp;
! 282: gen->id = newid;
! 283:
! 284: if (gen->flag & FGEN_FLAG_LINK) {
! 285: /*
! 286: * need to link file to basename
! 287: * have to use hardlink for now as I want to allow
! 288: * gen->basename spanning directory levels
! 289: * this would make it more complex to get the correct
! 290: * filename for symlink
! 291: *
! 292: * Ok, it would just mean taking the part following
! 293: * the last '/' in the name.... Should add it later....
! 294: */
! 295:
! 296: /* Windows NT does not support file links -Greg Schueman 1/18/97 */
! 297:
! 298: #if defined SYS_WINNT || defined SYS_VXWORKS
! 299: SetLastError(0); /* On WinNT, don't support FGEN_FLAG_LINK */
! 300: #elif defined(VMS)
! 301: errno = 0; /* On VMS, don't support FGEN_FLAG_LINK */
! 302: #else /* not (VMS) / VXWORKS / WINNT ; DO THE LINK) */
! 303: if (link(filename, basename) != 0)
! 304: if (EEXIST != errno)
! 305: msyslog(LOG_ERR,
! 306: "can't link(%s, %s): %m",
! 307: filename, basename);
! 308: #endif /* SYS_WINNT || VXWORKS */
! 309: } /* flags & FGEN_FLAG_LINK */
! 310: } /* else fp == NULL */
! 311:
! 312: free(basename);
! 313: free(filename);
! 314: return;
! 315: }
! 316:
! 317: /*
! 318: * this function sets up gen->fp to point to the correct
! 319: * generation of the file for the time specified by 'now'
! 320: *
! 321: * 'now' usually is interpreted as second part of a l_fp as is in the cal...
! 322: * library routines
! 323: */
! 324:
! 325: void
! 326: filegen_setup(
! 327: FILEGEN * gen,
! 328: u_long now
! 329: )
! 330: {
! 331: u_long new_gen = ~ (u_long) 0;
! 332: struct calendar cal;
! 333:
! 334: if (!(gen->flag & FGEN_FLAG_ENABLED)) {
! 335: if (NULL != gen->fp) {
! 336: fclose(gen->fp);
! 337: gen->fp = NULL;
! 338: }
! 339: return;
! 340: }
! 341:
! 342: switch (gen->type) {
! 343:
! 344: case FILEGEN_NONE:
! 345: if (NULL != gen->fp)
! 346: return; /* file already open */
! 347: break;
! 348:
! 349: case FILEGEN_PID:
! 350: new_gen = getpid();
! 351: break;
! 352:
! 353: case FILEGEN_DAY:
! 354: caljulian(now, &cal);
! 355: cal.hour = cal.minute = cal.second = 0;
! 356: new_gen = caltontp(&cal);
! 357: break;
! 358:
! 359: case FILEGEN_WEEK:
! 360: /* Would be nice to have a calweekstart() routine */
! 361: /* so just use a hack ... */
! 362: /* just round time to integral 7 day period for actual year */
! 363: new_gen = now - (now - calyearstart(now)) % TIMES7(SECSPERDAY)
! 364: + 60;
! 365: /*
! 366: * just to be sure -
! 367: * the computation above would fail in the presence of leap seconds
! 368: * so at least carry the date to the next day (+60 (seconds))
! 369: * and go back to the start of the day via calendar computations
! 370: */
! 371: caljulian(new_gen, &cal);
! 372: cal.hour = cal.minute = cal.second = 0;
! 373: new_gen = caltontp(&cal);
! 374: break;
! 375:
! 376: case FILEGEN_MONTH:
! 377: caljulian(now, &cal);
! 378: cal.yearday = (u_short) (cal.yearday - cal.monthday + 1);
! 379: cal.monthday = 1;
! 380: cal.hour = cal.minute = cal.second = 0;
! 381: new_gen = caltontp(&cal);
! 382: break;
! 383:
! 384: case FILEGEN_YEAR:
! 385: new_gen = calyearstart(now);
! 386: break;
! 387:
! 388: case FILEGEN_AGE:
! 389: new_gen = current_time - (current_time % SECSPERDAY);
! 390: break;
! 391: }
! 392: /*
! 393: * try to open file if not yet open
! 394: * reopen new file generation file on change of generation id
! 395: */
! 396: if (NULL == gen->fp || gen->id != new_gen) {
! 397:
! 398: DPRINTF(1, ("filegen %0x %lu %lu %lu\n",
! 399: gen->type, now, gen->id, new_gen));
! 400:
! 401: filegen_open(gen, new_gen);
! 402: }
! 403: }
! 404:
! 405:
! 406: /*
! 407: * change settings for filegen files
! 408: */
! 409: void
! 410: filegen_config(
! 411: FILEGEN * gen,
! 412: const char * basename,
! 413: u_int type,
! 414: u_int flag
! 415: )
! 416: {
! 417: int file_existed = 0;
! 418:
! 419: /*
! 420: * if nothing would be changed...
! 421: */
! 422: if ((strcmp(basename, gen->basename) == 0) && type == gen->type
! 423: && flag == gen->flag)
! 424: return;
! 425:
! 426: /*
! 427: * validate parameters
! 428: */
! 429: if (!valid_fileref(gen->prefix, basename))
! 430: return;
! 431:
! 432: if (NULL != gen->fp) {
! 433: fclose(gen->fp);
! 434: gen->fp = NULL;
! 435: file_existed = 1;
! 436: }
! 437:
! 438: DPRINTF(3, ("configuring filegen:\n"
! 439: "\tprefix:\t%s\n"
! 440: "\tbasename:\t%s -> %s\n"
! 441: "\ttype:\t%d -> %d\n"
! 442: "\tflag: %x -> %x\n",
! 443: gen->prefix,
! 444: gen->basename, basename,
! 445: gen->type, type,
! 446: gen->flag, flag));
! 447:
! 448: if (strcmp(gen->basename, basename) != 0) {
! 449: free(gen->basename);
! 450: gen->basename = estrdup(basename);
! 451: }
! 452: gen->type = (u_char)type;
! 453: gen->flag = (u_char)flag;
! 454:
! 455: /*
! 456: * make filegen use the new settings
! 457: * special action is only required when a generation file
! 458: * is currently open
! 459: * otherwise the new settings will be used anyway at the next open
! 460: */
! 461: if (file_existed) {
! 462: l_fp now;
! 463:
! 464: get_systime(&now);
! 465: filegen_setup(gen, now.l_ui);
! 466: }
! 467: }
! 468:
! 469:
! 470: /*
! 471: * check whether concatenating prefix and basename
! 472: * yields a legal filename
! 473: */
! 474: static int
! 475: valid_fileref(
! 476: const char * prefix,
! 477: const char * basename
! 478: )
! 479: {
! 480: /*
! 481: * prefix cannot be changed dynamically
! 482: * (within the context of filegen)
! 483: * so just reject basenames containing '..'
! 484: *
! 485: * ASSUMPTION:
! 486: * file system parts 'below' prefix may be
! 487: * specified without infringement of security
! 488: *
! 489: * restricting prefix to legal values
! 490: * has to be ensured by other means
! 491: * (however, it would be possible to perform some checks here...)
! 492: */
! 493: register const char *p = basename;
! 494:
! 495: /*
! 496: * Just to catch, dumb errors opening up the world...
! 497: */
! 498: if (NULL == prefix || '\0' == *prefix)
! 499: return 0;
! 500:
! 501: if (NULL == basename)
! 502: return 0;
! 503:
! 504: for (p = basename; p; p = strchr(p, DIR_SEP)) {
! 505: if ('.' == p[0] && '.' == p[1]
! 506: && ('\0' == p[2] || DIR_SEP == p[2]))
! 507: return 0;
! 508: }
! 509:
! 510: return 1;
! 511: }
! 512:
! 513:
! 514: /*
! 515: * filegen registry
! 516: */
! 517:
! 518: static struct filegen_entry {
! 519: char * name;
! 520: FILEGEN * filegen;
! 521: struct filegen_entry * next;
! 522: } *filegen_registry = NULL;
! 523:
! 524:
! 525: FILEGEN *
! 526: filegen_get(
! 527: const char * name
! 528: )
! 529: {
! 530: struct filegen_entry *f = filegen_registry;
! 531:
! 532: while (f) {
! 533: if (f->name == name || strcmp(name, f->name) == 0) {
! 534: DPRINTF(4, ("filegen_get(%s) = %p\n",
! 535: name, f->filegen));
! 536: return f->filegen;
! 537: }
! 538: f = f->next;
! 539: }
! 540: DPRINTF(4, ("filegen_get(%s) = NULL\n", name));
! 541: return NULL;
! 542: }
! 543:
! 544:
! 545: void
! 546: filegen_register(
! 547: const char * prefix,
! 548: const char * name,
! 549: FILEGEN * filegen
! 550: )
! 551: {
! 552: struct filegen_entry **ppfe;
! 553:
! 554: DPRINTF(4, ("filegen_register(%s, %p)\n", name, filegen));
! 555:
! 556: filegen_init(prefix, name, filegen);
! 557:
! 558: ppfe = &filegen_registry;
! 559: while (NULL != *ppfe) {
! 560: if ((*ppfe)->name == name
! 561: || !strcmp((*ppfe)->name, name)) {
! 562:
! 563: DPRINTF(5, ("replacing filegen %p\n",
! 564: (*ppfe)->filegen));
! 565:
! 566: (*ppfe)->filegen = filegen;
! 567: return;
! 568: }
! 569: ppfe = &((*ppfe)->next);
! 570: }
! 571:
! 572: *ppfe = emalloc(sizeof **ppfe);
! 573:
! 574: (*ppfe)->next = NULL;
! 575: (*ppfe)->name = estrdup(name);
! 576: (*ppfe)->filegen = filegen;
! 577:
! 578: DPRINTF(6, ("adding new filegen\n"));
! 579:
! 580: return;
! 581: }
! 582:
! 583:
! 584: /*
! 585: * filegen_unregister frees memory allocated by filegen_register for
! 586: * name.
! 587: */
! 588: #ifdef DEBUG
! 589: void
! 590: filegen_unregister(
! 591: char *name
! 592: )
! 593: {
! 594: struct filegen_entry ** ppfe;
! 595: struct filegen_entry * pfe;
! 596: FILEGEN * fg;
! 597:
! 598: DPRINTF(4, ("filegen_unregister(%s)\n", name));
! 599:
! 600: ppfe = &filegen_registry;
! 601:
! 602: while (NULL != *ppfe) {
! 603: if ((*ppfe)->name == name
! 604: || !strcmp((*ppfe)->name, name)) {
! 605: pfe = *ppfe;
! 606: *ppfe = (*ppfe)->next;
! 607: fg = pfe->filegen;
! 608: free(pfe->name);
! 609: free(pfe);
! 610: filegen_uninit(fg);
! 611: break;
! 612: }
! 613: ppfe = &((*ppfe)->next);
! 614: }
! 615: }
! 616: #endif /* DEBUG */
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>