X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=daemon.c;h=cae63ea7f48488124085d79360458ee44b5620a8;hp=f7a19221aa78fddd860415e6d4ab8b368c146992;hb=8b0ab0837ecdd5d28c7b6f31a605552f6942ca64;hpb=6bdac07456cb5872f824028912d1049883a9c21f diff --git a/daemon.c b/daemon.c index f7a19221..cae63ea7 100644 --- a/daemon.c +++ b/daemon.c @@ -8,12 +8,141 @@ #include "para.h" #include "daemon.h" #include - -/* getgrnam() */ -#include +#include /* getgrnam() */ #include #include "string.h" +#include "color.h" + +/** The internal state of the daemon. */ +struct daemon { + /** See \ref daemon_flags. */ + unsigned flags; + /** Set by \ref daemon_set_logfile(). */ + char *logfile_name; + /** Current loglevel, see \ref daemon_set_loglevel(). */ + int loglevel; + + /** Used by \ref server_uptime() and \ref uptime_str(). */ + time_t startuptime; + /** The file pointer if the logfile is open. */ + FILE *logfile; + /** Used by para_log() if \p DF_LOG_HOSTNAME is set. */ + char *hostname; + /** Used for colored log messages. */ + char log_colors[NUM_LOGLEVELS][COLOR_MAXLEN]; +}; + +static struct daemon the_daemon, *me = &the_daemon; + +/** + * Activate default log colors. + * + * This should be called early if color support is wanted. + */ +void daemon_set_default_log_colors(void) +{ + int i; + static const char *default_log_colors[NUM_LOGLEVELS] = { + [LL_DEBUG] = "normal", + [LL_INFO] = "normal", + [LL_NOTICE] = "normal", + [LL_WARNING] = "yellow", + [LL_ERROR] = "red", + [LL_CRIT] = "magenta bold", + [LL_EMERG] = "red bold", + }; + for (i = 0; i < NUM_LOGLEVELS; i++) { + int ret = color_parse(default_log_colors[i], me->log_colors[i]); + assert(ret >= 0); + } +} + +/** + * Set the color for one loglevel. + * + * \param arg The loglevel/color specifier. + * + * \a arg must be of the form "ll:[fg [bg]] [attr]". + * + * \return 1 On success, -1 on errors. + */ +int daemon_set_log_color(char const *arg) +{ + char *p = strchr(arg, ':'); + int ret, ll; + + if (!p) + goto err; + ret = get_loglevel_by_name(arg); + if (ret < 0) + goto err; + ll = ret; + p++; + ret = color_parse(p, me->log_colors[ll]); + if (ret < 0) + goto err; + return 1; +err: + PARA_ERROR_LOG("%s: color syntax error\n", arg); + return -1; + +} + +/** + * Init or change the name of the log file. + * + * \param logfile_name The full path of the logfile. + */ +void daemon_set_logfile(char *logfile_name) +{ + free(me->logfile_name); + me->logfile_name = NULL; + if (logfile_name) + me->logfile_name = para_strdup(logfile_name); +} + +/** + * Supress log messages with severity lower than the given loglevel. + * + * \param loglevel The smallest level that should be logged. + */ +void daemon_set_loglevel(char *loglevel) +{ + int ret = get_loglevel_by_name(loglevel); + + assert(ret >= 0); + me->loglevel = ret; +} + +/** + * Set one of the daemon config flags. + * + * \param flag The flag to set. + * + * \sa \ref daemon_flags. + */ +void daemon_set_flag(unsigned flag) +{ + me->flags |= flag; +} + +/** + * Clear one of the daemon config flags. + * + * \param flag The flag to clear. + * + * \sa \ref daemon_flags. + */ +void daemon_clear_flag(unsigned flag) +{ + me->flags &= ~flag; +} + +static unsigned daemon_test_flag(unsigned flag) +{ + return me->flags & flag; +} /** * Do the usual stuff to become a daemon. @@ -22,7 +151,7 @@ * * \sa fork(2), setsid(2), dup(2). */ -void daemon_init(void) +void daemonize(void) { pid_t pid; int null; @@ -56,50 +185,43 @@ err: } /** - * fopen() the given file in append mode. - * - * \param logfile_name The name of the file to open. - * - * \return Either calls exit() or returns a valid file handle. + * Close the log file of the daemon. */ -FILE *open_log(const char *logfile_name) +void daemon_close_log(void) { - FILE *logfile; - - assert(logfile_name); - logfile = fopen(logfile_name, "a"); - if (!logfile) { - PARA_EMERG_LOG("can not open %s: %s\n", logfile_name, - strerror(errno)); - exit(EXIT_FAILURE); - } - setlinebuf(logfile); - return logfile; + if (!me->logfile) + return; + PARA_INFO_LOG("closing logfile\n"); + fclose(me->logfile); + me->logfile = NULL; } /** - * Close the log file of the daemon. - * - * \param logfile The log file handle. + * fopen() the logfile in append mode. * - * It's OK to call this with logfile == \p NULL. + * \return Either succeeds or exits. */ -void close_log(FILE* logfile) +void daemon_open_log_or_die(void) { - if (!logfile) + daemon_close_log(); + if (!me->logfile_name) return; - PARA_INFO_LOG("closing logfile\n"); - fclose(logfile); + me->logfile = fopen(me->logfile_name, "a"); + if (!me->logfile) { + PARA_EMERG_LOG("can not open %s: %s\n", me->logfile_name, + strerror(errno)); + exit(EXIT_FAILURE); + } + setlinebuf(me->logfile); } /** * Log the startup message containing the paraslash version. */ -void log_welcome(const char *whoami, int loglevel) +void log_welcome(const char *whoami) { PARA_INFO_LOG("welcome to %s " PACKAGE_VERSION " ("BUILD_DATE")\n", whoami); - PARA_DEBUG_LOG("using loglevel %d\n", loglevel); } /** @@ -116,7 +238,7 @@ void log_welcome(const char *whoami, int loglevel) * * \sa getpwnam(3), getuid(2), setuid(2), getgrnam(2), setgid(2) */ -void para_drop_privileges(const char *username, const char *groupname) +void drop_privileges_or_die(const char *username, const char *groupname) { struct passwd *p; char *tmp; @@ -168,16 +290,15 @@ void para_drop_privileges(const char *username, const char *groupname) */ time_t server_uptime(enum uptime set_or_get) { - static time_t startuptime; time_t now; double diff; if (set_or_get == UPTIME_SET) { - time(&startuptime); + time(&me->startuptime); return 0; } time(&now); - diff = difftime(now, startuptime); + diff = difftime(now, me->startuptime); return (time_t) diff; } @@ -194,3 +315,51 @@ __malloc char *uptime_str(void) return make_message("%li:%02li:%02li", t / 86400, (t / 3600) % 24, (t / 60) % 60); } + +/** + * The log function for para_server and para_audiod. + * + * \param ll The log level. + * \param fmt The format string describing the log message. + */ +__printf_2_3 void para_log(int ll, const char* fmt,...) +{ + va_list argp; + FILE *fp; + struct tm *tm; + time_t t1; + char *color, str[MAXLINE] = ""; + + ll = PARA_MIN(ll, NUM_LOGLEVELS - 1); + ll = PARA_MAX(ll, LL_DEBUG); + if (ll < me->loglevel) + return; + + fp = me->logfile? me->logfile : stderr; + color = daemon_test_flag(DF_COLOR_LOG)? me->log_colors[ll] : NULL; + if (color) + fprintf(fp, "%s", color); + if (daemon_test_flag(DF_LOG_TIME)) { + /* date and time */ + time(&t1); + tm = localtime(&t1); + strftime(str, MAXLINE, "%b %d %H:%M:%S", tm); + fprintf(fp, "%s ", str); + } + if (daemon_test_flag(DF_LOG_HOSTNAME)) { + if (!me->hostname) + me->hostname = para_hostname(); + fprintf(fp, "%s ", me->hostname); + } + if (daemon_test_flag(DF_LOG_LL)) /* log loglevel */ + fprintf(fp, "(%d) ", ll); + if (daemon_test_flag(DF_LOG_PID)) { /* log pid */ + pid_t mypid = getpid(); + fprintf(fp, "(%d) ", (int)mypid); + } + va_start(argp, fmt); + vfprintf(fp, fmt, argp); + va_end(argp); + if (color) + fprintf(fp, "%s", COLOR_RESET); +}