#ifdef HAVE_CONFIG_H
# include <config.h>
#endif
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include "ntpd.h"
#include "ntp_io.h"
#include "ntp_string.h"
#include "ntp_calendar.h"
#include "ntp_filegen.h"
#include "ntp_stdlib.h"
#define SUFFIX_SEP '.'
static void filegen_open (FILEGEN *, u_long);
static int valid_fileref (const char *, const char *);
static void filegen_init (const char *, const char *, FILEGEN *);
#ifdef DEBUG
static void filegen_uninit (FILEGEN *);
#endif
static void
filegen_init(
const char * prefix,
const char * basename,
FILEGEN * fgp
)
{
fgp->fp = NULL;
fgp->prefix = prefix;
fgp->basename = estrdup(basename);
fgp->id = 0;
fgp->type = FILEGEN_DAY;
fgp->flag = FGEN_FLAG_LINK;
}
#ifdef DEBUG
static void
filegen_uninit(
FILEGEN * fgp
)
{
free(fgp->basename);
}
#endif
static void
filegen_open(
FILEGEN * gen,
u_long newid
)
{
char *filename;
char *basename;
u_int len;
FILE *fp;
struct calendar cal;
len = strlen(gen->prefix) + strlen(gen->basename) + 1;
basename = emalloc(len);
snprintf(basename, len, "%s%s", gen->prefix, gen->basename);
switch(gen->type) {
default:
msyslog(LOG_ERR,
"unsupported file generations type %d for "
"\"%s\" - reverting to FILEGEN_NONE",
gen->type, basename);
gen->type = FILEGEN_NONE;
case FILEGEN_NONE:
filename = estrdup(basename);
break;
case FILEGEN_PID:
filename = emalloc(len + 1 + 1 + 10);
snprintf(filename, len + 1 + 1 + 10,
"%s%c#%ld",
basename, SUFFIX_SEP, newid);
break;
case FILEGEN_DAY:
caljulian(newid, &cal);
filename = emalloc(len + 1 + 4 + 2 + 2);
snprintf(filename, len + 1 + 4 + 2 + 2,
"%s%c%04d%02d%02d",
basename, SUFFIX_SEP,
cal.year, cal.month, cal.monthday);
break;
case FILEGEN_WEEK:
caljulian(newid, &cal);
filename = emalloc(len + 1 + 4 + 1 + 2);
snprintf(filename, len + 1 + 4 + 1 + 2,
"%s%c%04dw%02d",
basename, SUFFIX_SEP,
cal.year, cal.yearday / 7);
break;
case FILEGEN_MONTH:
caljulian(newid, &cal);
filename = emalloc(len + 1 + 4 + 2);
snprintf(filename, len + 1 + 4 + 2,
"%s%c%04d%02d",
basename, SUFFIX_SEP, cal.year, cal.month);
break;
case FILEGEN_YEAR:
caljulian(newid, &cal);
filename = emalloc(len + 1 + 4);
snprintf(filename, len + 1 + 4,
"%s%c%04d",
basename, SUFFIX_SEP, cal.year);
break;
case FILEGEN_AGE:
filename = emalloc(len + 1 + 2 + 10);
snprintf(filename, len + 1 + 2 + 10,
"%s%ca%08ld",
basename, SUFFIX_SEP, newid);
}
if (FILEGEN_NONE != gen->type) {
struct stat stats;
static u_long conflicts = 0;
#ifndef S_ISREG
#define S_ISREG(mode) (((mode) & S_IFREG) == S_IFREG)
#endif
if (stat(basename, &stats) == 0) {
if (S_ISREG(stats.st_mode)) {
if (stats.st_nlink <= 1) {
char *savename;
savename = emalloc(len + 1 + 1 + 10 + 10);
snprintf(savename, len + 1 + 1 + 10 + 10,
"%s%c%dC%lu",
basename, SUFFIX_SEP,
(int)getpid(), conflicts++);
if (rename(basename, savename) != 0)
msyslog(LOG_ERR,
"couldn't save %s: %m",
basename);
free(savename);
} else {
if (
#if !defined(VMS)
unlink(basename) != 0
#else
delete(basename) != 0
#endif
)
msyslog(LOG_ERR,
"couldn't unlink %s: %m",
basename);
}
} else {
msyslog(LOG_ERR,
"expected regular file for %s "
"(found mode 0%lo)",
basename,
(unsigned long)stats.st_mode);
}
} else {
if (ENOENT != errno)
msyslog(LOG_ERR, "stat(%s) failed: %m",
basename);
}
}
fp = fopen(filename, "a");
DPRINTF(4, ("opening filegen (type=%d/id=%lu) \"%s\"\n",
gen->type, newid, filename));
if (NULL == fp) {
if (ENOENT != errno)
msyslog(LOG_ERR, "can't open %s: %m", filename);
} else {
if (NULL != gen->fp) {
fclose(gen->fp);
gen->fp = NULL;
}
gen->fp = fp;
gen->id = newid;
if (gen->flag & FGEN_FLAG_LINK) {
#if defined SYS_WINNT || defined SYS_VXWORKS
SetLastError(0);
#elif defined(VMS)
errno = 0;
#else
if (link(filename, basename) != 0)
if (EEXIST != errno)
msyslog(LOG_ERR,
"can't link(%s, %s): %m",
filename, basename);
#endif
}
}
free(basename);
free(filename);
return;
}
void
filegen_setup(
FILEGEN * gen,
u_long now
)
{
u_long new_gen = ~ (u_long) 0;
struct calendar cal;
if (!(gen->flag & FGEN_FLAG_ENABLED)) {
if (NULL != gen->fp) {
fclose(gen->fp);
gen->fp = NULL;
}
return;
}
switch (gen->type) {
case FILEGEN_NONE:
if (NULL != gen->fp)
return;
break;
case FILEGEN_PID:
new_gen = getpid();
break;
case FILEGEN_DAY:
caljulian(now, &cal);
cal.hour = cal.minute = cal.second = 0;
new_gen = caltontp(&cal);
break;
case FILEGEN_WEEK:
new_gen = now - (now - calyearstart(now)) % TIMES7(SECSPERDAY)
+ 60;
caljulian(new_gen, &cal);
cal.hour = cal.minute = cal.second = 0;
new_gen = caltontp(&cal);
break;
case FILEGEN_MONTH:
caljulian(now, &cal);
cal.yearday = (u_short) (cal.yearday - cal.monthday + 1);
cal.monthday = 1;
cal.hour = cal.minute = cal.second = 0;
new_gen = caltontp(&cal);
break;
case FILEGEN_YEAR:
new_gen = calyearstart(now);
break;
case FILEGEN_AGE:
new_gen = current_time - (current_time % SECSPERDAY);
break;
}
if (NULL == gen->fp || gen->id != new_gen) {
DPRINTF(1, ("filegen %0x %lu %lu %lu\n",
gen->type, now, gen->id, new_gen));
filegen_open(gen, new_gen);
}
}
void
filegen_config(
FILEGEN * gen,
const char * basename,
u_int type,
u_int flag
)
{
int file_existed = 0;
size_t octets;
if ((strcmp(basename, gen->basename) == 0) && type == gen->type
&& flag == gen->flag)
return;
if (!valid_fileref(gen->prefix, basename))
return;
if (NULL != gen->fp) {
fclose(gen->fp);
gen->fp = NULL;
file_existed = 1;
}
DPRINTF(3, ("configuring filegen:\n"
"\tprefix:\t%s\n"
"\tbasename:\t%s -> %s\n"
"\ttype:\t%d -> %d\n"
"\tflag: %x -> %x\n",
gen->prefix,
gen->basename, basename,
gen->type, type,
gen->flag, flag));
if (strcmp(gen->basename, basename) != 0) {
octets = strlen(basename) + 1;
gen->basename = erealloc(gen->basename, octets);
memcpy(gen->basename, basename, octets);
}
gen->type = (u_char) type;
gen->flag = (u_char) flag;
if (file_existed) {
l_fp now;
get_systime(&now);
filegen_setup(gen, now.l_ui);
}
}
static int
valid_fileref(
const char * prefix,
const char * basename
)
{
register const char *p = basename;
if (NULL == prefix || '\0' == *prefix)
return 0;
if (NULL == basename)
return 0;
for (p = basename; p; p = strchr(p, DIR_SEP)) {
if ('.' == p[0] && '.' == p[1]
&& ('\0' == p[2] || DIR_SEP == p[2]))
return 0;
}
return 1;
}
static struct filegen_entry {
char * name;
FILEGEN * filegen;
struct filegen_entry * next;
} *filegen_registry = NULL;
FILEGEN *
filegen_get(
const char * name
)
{
struct filegen_entry *f = filegen_registry;
while (f) {
if (f->name == name || strcmp(name, f->name) == 0) {
DPRINTF(4, ("filegen_get(%s) = %p\n",
name, f->filegen));
return f->filegen;
}
f = f->next;
}
DPRINTF(4, ("filegen_get(%s) = NULL\n", name));
return NULL;
}
void
filegen_register(
const char * prefix,
const char * name,
FILEGEN * filegen
)
{
struct filegen_entry **ppfe;
DPRINTF(4, ("filegen_register(%s, %p)\n", name, filegen));
filegen_init(prefix, name, filegen);
ppfe = &filegen_registry;
while (NULL != *ppfe) {
if ((*ppfe)->name == name
|| !strcmp((*ppfe)->name, name)) {
DPRINTF(5, ("replacing filegen %p\n",
(*ppfe)->filegen));
(*ppfe)->filegen = filegen;
return;
}
ppfe = &((*ppfe)->next);
}
*ppfe = emalloc(sizeof **ppfe);
(*ppfe)->next = NULL;
(*ppfe)->name = estrdup(name);
(*ppfe)->filegen = filegen;
DPRINTF(6, ("adding new filegen\n"));
return;
}
#ifdef DEBUG
void
filegen_unregister(
char *name
)
{
struct filegen_entry ** ppfe;
struct filegen_entry * pfe;
FILEGEN * fg;
DPRINTF(4, ("filegen_unregister(%s)\n", name));
ppfe = &filegen_registry;
while (NULL != *ppfe) {
if ((*ppfe)->name == name
|| !strcmp((*ppfe)->name, name)) {
pfe = *ppfe;
*ppfe = (*ppfe)->next;
fg = pfe->filegen;
free(pfe->name);
free(pfe);
filegen_uninit(fg);
break;
}
ppfe = &((*ppfe)->next);
}
}
#endif