]> git.tuebingen.mpg.de Git - paraslash.git/blobdiff - server.c
Implement colored logging for para_server.
[paraslash.git] / server.c
index 244c1ecd25da7959cf3ce6b9dcff3d9e814b2ad0..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;
@@ -107,15 +110,17 @@ struct server_args_info conf;
 /** A random value used in child context for authentication. */
 uint32_t afs_socket_cookie;
 
+/** The mutex protecting the shared memory area containing the mmd struct. */
+int mmd_mutex;
+
 /* global variables for server-internal use */
 static FILE *logfile;
 /** The file containing user information (public key, permissions). */
 static char *user_list_file = NULL;
-static int mmd_mutex, mmd_shm_id;
-static pid_t afs_pid;
+static int mmd_shm_id;
 
 
-/** The task resposible for server command handling. */
+/** The task responsible for server command handling. */
 struct server_command_task {
        /** TCP port on which para_server listens for connections. */
        int listen_fd;
@@ -127,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));
@@ -195,33 +282,22 @@ err_out:
 }
 
 /**
- * Lock the shared memory area containing the mmd struct.
+ * (Re-)read the server configuration files.
  *
- * \sa semop(2), struct misc_meta_data.
- */
-void mmd_lock(void)
-{
-       mutex_lock(mmd_mutex);
-}
-
-/**
- * Unlock the shared memory area containing the mmd struct.
+ * \param override Passed to gengetopt to activate the override feature.
  *
- * \sa semop(2), struct misc_meta_data.
+ * This function also re-opens the logfile and sets the global \a
+ * user_list_file variable.
  */
-
-void mmd_unlock(void)
-{
-       mutex_unlock(mmd_mutex);
-}
-
-static void parse_config(int override)
+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
@@ -251,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);
@@ -274,12 +352,10 @@ 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 (afs_pid)
-               kill(afs_pid, SIGHUP);
+       if (mmd->afs_pid)
+               kill(mmd->afs_pid, SIGHUP);
 }
 
 static void signal_post_select(struct sched *s, struct task *t)
@@ -300,7 +376,7 @@ static void signal_post_select(struct sched *s, struct task *t)
                        int ret = para_reap_child(&pid);
                        if (ret <= 0)
                                break;
-                       if (pid != afs_pid)
+                       if (pid != mmd->afs_pid)
                                continue;
                        PARA_EMERG_LOG("fatal: afs died\n");
                        goto genocide;
@@ -466,10 +542,10 @@ static int init_afs(void)
        if (ret < 0)
                exit(EXIT_FAILURE);
        afs_socket_cookie = para_random((uint32_t)-1);
-       afs_pid = fork();
-       if (afs_pid < 0)
+       mmd->afs_pid = fork();
+       if (mmd->afs_pid < 0)
                exit(EXIT_FAILURE);
-       if (!afs_pid) { /* child (afs) */
+       if (!mmd->afs_pid) { /* child (afs) */
                close(afs_server_socket[0]);
                afs_init(afs_socket_cookie, afs_server_socket[1]);
        }
@@ -499,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 */
@@ -511,10 +589,9 @@ static void server_init(int argc, char **argv)
                daemon_init();
        PARA_NOTICE_LOG("initializing audio format handlers\n");
        afh_init();
-       mmd->server_pid = getpid();
-       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);
@@ -552,9 +629,9 @@ static int server_select(int max_fileno, fd_set *readfds, fd_set *writefds,
        int ret;
 
        status_refresh();
-       mmd_unlock();
+       mutex_unlock(mmd_mutex);
        ret = para_select(max_fileno + 1, readfds, writefds, timeout_tv);
-       mmd_lock();
+       mutex_lock(mmd_mutex);
        return ret;
 }
 
@@ -577,7 +654,7 @@ int main(int argc, char *argv[])
                .select_function = server_select
        };
        server_init(argc, argv);
-       mmd_lock();
+       mutex_lock(mmd_mutex);
        ret = schedule(&s);
        if (ret < 0) {
                PARA_EMERG_LOG("%s\n", para_strerror(-ret));