Implement colored logging for para_server.
[paraslash.git] / server.c
index 7da05898b69b528563fc0e902f0568c0728752a8..1fadf7279306b6fab8d195199b15d83ac5646607 100644 (file)
--- a/server.c
+++ b/server.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 1997-2008 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 1997-2009 Andre Noll <maan@systemlinux.org>
  *
  * Licensed under the GPL v2. For licencing details see COPYING.
  */
@@ -12,8 +12,9 @@
  *
  * Starting points for getting an overview:
  *
+ *
  *     - The main programs: \ref server.c, \ref audiod.c, \ref client.c,
- *       \ref audioc.c, \ref fsck.c,
+ *       \ref audioc.c, \ref fsck.c, \ref afh.c
  *     - Server: \ref server_command, \ref sender,
  *     - Audio file selector: \ref audio_format_handler, \ref mood, \ref afs_table,
  *     - Client: \ref receiver, \ref receiver_node, \ref filter, \ref filter_node.
  *
  * The gory details, listed by topic:
  *
- *     - Audio format handlers: \ref mp3_afh.c, \ref ogg_afh.c, \ref aac_afh.c,
- *     - Decoders: \ref mp3dec.c, \ref oggdec.c, \ref aacdec.c,
- *     - Volume normalizer: \ref compress.c,
+ *     - Audio format handlers: \ref send_common.c \ref mp3_afh.c, \ref ogg_afh.c, \ref aac_afh.c,
+ *     - Decoders: \ref mp3dec_filter.c, \ref oggdec_filter.c, \ref aacdec_filter.c,
+ *     - Volume normalizer: \ref compress_filter.c,
  *     - Output: \ref alsa_write.c, \ref osx_write.c,
  *     - http: \ref http_recv.c, \ref http_send.c,
- *     - ortp: \ref ortp_recv.c, \ref ortp_send.c,
+ *     - udp: \ref udp_recv.c, \ref udp_send.c,
  *     - dccp: \ref dccp_recv.c, \ref dccp_send.c,
  *     - Audio file selector: \ref afs.c, \ref aft.c, \ref mood.c,
  *     - Afs structures: \ref afs_table, \ref audio_file_data,
@@ -63,6 +64,7 @@
 
 #include <signal.h>
 #include <dirent.h>
+#include <sys/time.h>
 
 #include "para.h"
 #include "error.h"
@@ -83,6 +85,7 @@
 #include "sched.h"
 #include "signal.h"
 #include "user_list.h"
+#include "color.h"
 
 /** Define the array of error lists needed by para_server. */
 INIT_SERVER_ERRLISTS;
@@ -129,42 +132,124 @@ struct server_command_task {
        struct task task;
 };
 
+static int want_colors(void)
+{
+       if (conf.color_arg == color_arg_no)
+               return 0;
+       if (conf.color_arg == color_arg_yes)
+               return 1;
+       if (logfile)
+               return 0;
+       return isatty(STDERR_FILENO);
+}
+
+static int get_loglevel_by_name(const char *txt, size_t n)
+{
+       if (!strncasecmp(txt, "debug", n))
+               return LL_DEBUG;
+       if (!strncasecmp(txt, "info", n))
+               return LL_INFO;
+       if (!strncasecmp(txt, "notice", n))
+               return LL_NOTICE;
+       if (!strncasecmp(txt, "warning", n))
+               return LL_WARNING;
+       if (!strncasecmp(txt, "error", n))
+               return LL_ERROR;
+       if (!strncasecmp(txt, "crit", n))
+               return LL_CRIT;
+       if (!strncasecmp(txt, "emerg", n))
+               return LL_EMERG;
+       return -1;
+}
+
+static char log_colors[NUM_LOGLEVELS][COLOR_MAXLEN];
+
+static void init_colors_or_die(void)
+{
+       int ret, i;
+       static const char *default_log_colors[NUM_LOGLEVELS] = {
+                       [LL_DEBUG] = "normal",
+                       [LL_INFO] = "white bold",
+                       [LL_NOTICE] = "cyan bold",
+                       [LL_WARNING] = "green bold",
+                       [LL_ERROR] = "yellow bold",
+                       [LL_CRIT] = "magenta bold",
+                       [LL_EMERG] = "red bold",
+       };
+
+       for (i = 0; i < NUM_LOGLEVELS; i++) {
+               ret = color_parse(default_log_colors[i], log_colors[i]);
+               assert(ret >= 0);
+       }
+
+       for (i = 0; i < conf.log_color_given; i++) {
+               char *arg = conf.log_color_arg[i], *p = strchr(arg, ':');
+               int ll;
+               if (!p)
+                       goto err;
+               ret = get_loglevel_by_name(arg, p - arg);
+               if (ret < 0)
+                       goto err;
+               ll = ret;
+               p++;
+               ret = color_parse(p, log_colors[ll]);
+               if (ret < 0)
+                       goto err;
+       }
+       return;
+err:
+       PARA_EMERG_LOG("color syntax error, arg %d (%s)\n", i,
+               conf.log_color_arg[i]);
+       exit(EXIT_FAILURE);
+}
+
 /**
  * Para_server's log function.
  *
  * \param ll The log level.
  * \param fmt The format string describing the log message.
  */
-void para_log(int ll, const char* fmt,...)
+__printf_2_3 void para_log(int ll, const char* fmt,...)
 {
        va_list argp;
-       FILE *outfd;
+       FILE *fp;
        struct tm *tm;
        time_t t1;
-       char str[MAXLINE] = "";
-       pid_t mypid;
+       char *color, str[MAXLINE] = "";
 
+       ll = PARA_MIN(ll, NUM_LOGLEVELS - 1);
+       ll = PARA_MAX(ll, LL_DEBUG);
        if (ll < conf.loglevel_arg)
                return;
-       outfd = logfile? logfile : stderr;
+
+       fp = logfile? logfile : stderr;
+       color = want_colors()? log_colors[ll] : NULL;
+
+       if (color)
+               fprintf(fp, "%s", color);
+       /* date and time */
        time(&t1);
        tm = localtime(&t1);
        strftime(str, MAXLINE, "%b %d %H:%M:%S", tm);
-       fprintf(outfd, "%s ", str);
-       if (conf.loglevel_arg <= INFO)
-               fprintf(outfd, "%i: ", ll);
-       mypid = getpid();
-       if (conf.loglevel_arg <= INFO)
-               fprintf(outfd, "(%d) ", (int)mypid);
+       fprintf(fp, "%s ", str);
+       /* loglevel */
+       if (conf.loglevel_arg <= LL_INFO)
+               fprintf(fp, "%i: ", ll);
+       if (conf.loglevel_arg <= LL_INFO) { /* log pid */
+               pid_t mypid = getpid();
+               fprintf(fp, "(%d) ", (int)mypid);
+       }
        va_start(argp, fmt);
-       vfprintf(outfd, fmt, argp);
+       vfprintf(fp, fmt, argp);
        va_end(argp);
+       if (color)
+               fprintf(fp, "%s", COLOR_RESET);
 }
 
 /*
  * setup shared memory area and get mutex for locking
  */
-static void shm_init(void)
+static void init_ipc_or_die(void)
 {
        void *shm;
        int ret = shm_new(sizeof(struct misc_meta_data));
@@ -196,13 +281,23 @@ err_out:
        exit(EXIT_FAILURE);
 }
 
-static void parse_config(int override)
+/**
+ * (Re-)read the server configuration files.
+ *
+ * \param override Passed to gengetopt to activate the override feature.
+ *
+ * This function also re-opens the logfile and sets the global \a
+ * user_list_file variable.
+ */
+void parse_config_or_die(int override)
 {
        char *home = para_homedir();
        struct stat statbuf;
        int ret;
        char *cf;
 
+       close_log(logfile);
+       logfile = NULL;
        if (conf.config_file_given)
                cf = para_strdup(conf.config_file_arg);
        else
@@ -232,6 +327,8 @@ static void parse_config(int override)
        }
        if (conf.logfile_given)
                logfile = open_log(conf.logfile_arg);
+       if (want_colors())
+               init_colors_or_die();
        ret = 1;
 out:
        free(cf);
@@ -255,9 +352,7 @@ static void signal_pre_select(struct sched *s, struct task *t)
 static void handle_sighup(void)
 {
        PARA_NOTICE_LOG("SIGHUP\n");
-       close_log(logfile); /* gets reopened if necessary by parse_config */
-       logfile = NULL;
-       parse_config(1); /* reopens log */
+       parse_config_or_die(1); /* reopens log */
        init_user_list(user_list_file); /* reload user list */
        if (mmd->afs_pid)
                kill(mmd->afs_pid, SIGHUP);
@@ -480,11 +575,13 @@ static void server_init(int argc, char **argv)
        /* parse command line options */
        server_cmdline_parser_ext(argc, argv, &conf, &params);
        HANDLE_VERSION_FLAG("server", conf);
-       para_drop_privileges(conf.user_arg, conf.group_arg);
+       drop_privileges_or_die(conf.user_arg, conf.group_arg);
        /* parse config file, open log and set defaults */
-       parse_config(0);
+       parse_config_or_die(0);
        log_welcome("para_server", conf.loglevel_arg);
-       shm_init(); /* init mmd struct */
+       init_ipc_or_die(); /* init mmd struct and mmd->lock */
+       /* make sure, the global now pointer is uptodate */
+       gettimeofday(now, NULL);
        server_uptime(UPTIME_SET); /* reset server uptime */
        init_user_list(user_list_file);
        /* become daemon */
@@ -492,9 +589,9 @@ static void server_init(int argc, char **argv)
                daemon_init();
        PARA_NOTICE_LOG("initializing audio format handlers\n");
        afh_init();
-       init_signal_task();
        PARA_NOTICE_LOG("initializing the audio file selector\n");
        afs_socket = init_afs();
+       init_signal_task();
        PARA_NOTICE_LOG("initializing virtual streaming system\n");
        init_vss_task(afs_socket);
        init_server_command_task(argc, argv);