Revert
[yaz-moved-to-github.git] / src / log.c
index 06540ce..05a92b3 100644 (file)
--- a/src/log.c
+++ b/src/log.c
@@ -1,13 +1,13 @@
 /*
- * Copyright (C) 1995-2005, Index Data ApS
+ * Copyright (C) 1995-2006, Index Data ApS
  * See the file LICENSE for details.
  *
- * $Id: log.c,v 1.26 2005-09-09 10:33:45 adam Exp $
+ * $Id: log.c,v 1.35 2006-05-07 18:26:25 adam Exp $
  */
 
 /**
  * \file log.c
- * \brief Implements logging utility
+ * \brief Logging utility
  */
 
 #if HAVE_CONFIG_H
@@ -42,6 +42,7 @@ static int mutex_init_flag = 0; /* not yet initialized */
 
 #define HAS_STRERROR 1
 
+
 #if HAS_STRERROR
 
 #else
@@ -54,13 +55,35 @@ char *strerror(int n)
 #endif
 
 
+static int default_log_level() {
+    char *env = getenv("YAZ_LOG");
+    if (env != 0)
+        return yaz_log_mask_str_x(env, YLOG_DEFAULT_LEVEL);
+    else
+        return YLOG_DEFAULT_LEVEL;
+}
+
+
+static int l_level = -1;        /* will be set from default_log_level() */
+
+enum l_file_type {  use_stderr, use_none, use_file };
+static enum l_file_type yaz_file_type = use_stderr;
+static FILE *yaz_global_log_file = NULL;
+
+static void (*start_hook_func)(int, const char *, void *) = NULL;
+static void *start_hook_info;
+
+static void (*end_hook_func)(int, const char *, void *) = NULL;
+static void *end_hook_info;
+
+static void (*hook_func)(int, const char *, void *) = NULL;
+static void *hook_info;
 
-static int l_level = YLOG_DEFAULT_LEVEL;
-static FILE *l_file = NULL;
 static char l_prefix[512] = "";
 static char l_prefix2[512] = "";
 static char l_fname[512] = "";
 
+
 static char l_old_default_format[] = "%H:%M:%S-%d/%m";
 static char l_new_default_format[] = "%Y%m%d-%H%M%S";
 #define TIMEFORMAT_LEN 50
@@ -94,85 +117,80 @@ static struct {
     { 0, NULL }
     /* the rest will be filled in if the user defines dynamic modules*/
 };  
+
 static unsigned int next_log_bit = YLOG_LAST_BIT<<1; /* first dynamic bit */
 
 static void init_mutex()
 {
     if (mutex_init_flag)
         return;
-    nmem_mutex_create (&log_mutex);
+    nmem_mutex_create(&log_mutex);
     mutex_init_flag = 1;
 }
 
-
 FILE *yaz_log_file(void)
 {
-    if (!l_file)
-        l_file = stderr;
-    return l_file;
+    FILE *f = 0;
+    switch(yaz_file_type)
+    {
+        case use_stderr: f = stderr; break;
+        case use_none: f = 0; break;
+        case use_file: f = yaz_global_log_file; break;
+    }
+    return f;
 }
 
-void yaz_log_init_file (const char *fname)
+void yaz_log_close()
 {
-    init_mutex();
-    if (fname)
+    if (yaz_file_type == use_file && yaz_global_log_file)
     {
-        strncpy(l_fname, fname, sizeof(l_fname)-1);
-        l_fname[sizeof(l_fname)-1] = '\0';
+        fclose(yaz_global_log_file);
+        yaz_global_log_file = 0;
     }
-    else
-        l_fname[0] = '\0';
-    yaz_log_reopen();
 }
 
-void yaz_log_reopen(void)
+void yaz_log_init_file(const char *fname)
 {
-    FILE *new_file;
     init_mutex();
-    if (!l_file)
-        l_file = stderr;
-    if (!*l_fname)
-        new_file = stderr;
-    else if (!(new_file = fopen(l_fname, "a")))
+
+    yaz_log_close();
+    if (fname)
     {
-        new_file=l_file;
-        l_file=stderr;  /* just to be sure we don't rotate logs and recurse */
-        yaz_log(YLOG_WARN|YLOG_ERRNO,"Could not open log file '%s'",l_fname);
-        l_file=new_file; /* restore to old value, probably stderr as well */
-        return;
+        if (*fname == '\0')
+            yaz_file_type = use_stderr; /* empty name; use stderr */
+        else
+            yaz_file_type = use_file;
+        strncpy(l_fname, fname, sizeof(l_fname)-1);
+        l_fname[sizeof(l_fname)-1] = '\0';
     }
-    if (l_file != stderr)
+    else
     {
-        fclose (l_file);
+        yaz_file_type = use_none;  /* NULL name; use no file at all */
+        l_fname[0] = '\0'; 
     }
-    if (l_level & YLOG_FLUSH)
-        setvbuf(new_file, 0, _IONBF, 0);
-    l_file = new_file;
+    yaz_log_reopen();
 }
 
-static void rotate_log()
+static void rotate_log(const char *cur_fname)
 {
     char newname[512];
-    if (l_file==stderr)
-        return; /* can't rotate that */
-    if (!*l_fname)
-        return; /* hmm, no name, can't rotate */
-    strncpy(newname, l_fname, 509);
+    strncpy(newname, cur_fname, 509);
     newname[509] = '\0'; /* make sure it is terminated */
-    strcat(newname,".1");
+    strcat(newname, ".1");
 #ifdef WIN32
     /* windows can't rename a file if it is open */
-    fclose(l_file);
-    l_file = stderr;
+    fclose(yaz_global_log_file);
+    yaz_global_log_file = 0;
+    MoveFileEx(cur_fname, newname, MOVEFILE_REPLACE_EXISTING);
+#else
+    rename(cur_fname, newname);
 #endif
-    rename(l_fname, newname);
-    yaz_log_reopen();
 }
 
-
-void yaz_log_init_level (int level)
+void yaz_log_init_level(int level)
 {
     init_mutex();
+    if (l_level < 0) l_level = default_log_level();
     if ( (l_level & YLOG_FLUSH) != (level & YLOG_FLUSH) )
     {
         l_level = level;
@@ -205,7 +223,7 @@ void yaz_log_init_level (int level)
     }
 }
 
-void yaz_log_init_prefix (const char *prefix)
+void yaz_log_init_prefix(const char *prefix)
 {
     if (prefix && *prefix)
         sprintf(l_prefix, "%.511s ", prefix);
@@ -213,7 +231,7 @@ void yaz_log_init_prefix (const char *prefix)
         *l_prefix = 0;
 }
 
-void yaz_log_init_prefix2 (const char *prefix)
+void yaz_log_init_prefix2(const char *prefix)
 {
     if (prefix && *prefix)
         sprintf(l_prefix2, "%.511s ", prefix);
@@ -224,10 +242,10 @@ void yaz_log_init_prefix2 (const char *prefix)
 void yaz_log_init(int level, const char *prefix, const char *fname)
 {
     init_mutex();
-    yaz_log_init_level (level);
-    yaz_log_init_prefix (prefix);
+    yaz_log_init_level(level);
+    yaz_log_init_prefix(prefix);
     if (fname && *fname)
-        yaz_log_init_file (fname);
+        yaz_log_init_file(fname);
 }
 
 void yaz_log_init_max_size(int mx)
@@ -238,60 +256,165 @@ void yaz_log_init_max_size(int mx)
         l_max_size = mx;
 }
 
-static void (*start_hook_func)(int, const char *, void *) = NULL;
-static void *start_hook_info;
-static void (*end_hook_func)(int, const char *, void *) = NULL;
-static void *end_hook_info;
+void yaz_log_set_handler(void (*func)(int, const char *, void *), void *info)
+{
+    hook_func = func;
+    hook_info = info;
+}
 
-void log_event_start (void (*func)(int, const char *, void *), void *info)
+void log_event_start(void (*func)(int, const char *, void *), void *info)
 {
-    start_hook_func = func;
-    start_hook_info = info;
+     start_hook_func = func;
+     start_hook_info = info;
 }
 
-void log_event_end (void (*func)(int, const char *, void *), void *info)
+void log_event_end(void (*func)(int, const char *, void *), void *info)
 {
-    end_hook_func = func;
-    end_hook_info = info;
+     end_hook_func = func;
+     end_hook_info = info;
 }
 
-void yaz_log(int level, const char *fmt, ...)
+static void yaz_log_open_check(struct tm *tm, int force)
 {
-    va_list ap;
-    char buf[4096], flags[1024];
+    char new_filename[512];
+    static char cur_filename[512] = "";
+
+    if (l_fname && *l_fname)
+    {
+        strftime(new_filename, sizeof(new_filename)-1, l_fname, tm);
+        if (strcmp(new_filename, cur_filename))
+        {
+            strcpy(cur_filename, new_filename);
+            force = 1;
+        }
+    }
+
+    if (l_max_size > 0 && yaz_global_log_file && yaz_file_type == use_file)
+    {
+        long flen = ftell(yaz_global_log_file);
+        if (flen > l_max_size)
+        {
+            rotate_log(cur_filename);
+            force = 1;
+        }
+    }
+    if (force && yaz_file_type == use_file && *cur_filename)
+    {
+        yaz_log_close();
+        yaz_global_log_file = fopen(cur_filename, "a");
+        if (l_level < 0) l_level = default_log_level();
+        if (l_level & YLOG_FLUSH)
+            setvbuf(yaz_global_log_file, 0, _IONBF, 0);
+    }
+}
+
+void yaz_log_reopen()
+{
+    time_t cur_time = time(0);
+#if HAVE_LOCALTIME_R
+    struct tm tm0, *tm = &tm0;
+#else
+    struct tm *tm;
+#endif
+
+    nmem_mutex_enter(log_mutex);
+#if HAVE_LOCALTIME_R
+    localtime_r(&cur_time, tm);
+#else
+    tm = localtime(&cur_time);
+#endif
+    yaz_log_open_check(tm, 1);
+    nmem_mutex_leave(log_mutex);
+}
+
+static void yaz_strftime(char *dst, size_t sz,
+                         const char *fmt, const struct tm *tm)
+{
+    const char *cp = strstr(fmt, "%!");
+    if (cp && strlen(fmt) < 60)
+    {
+        char fmt2[80];
+        char tpidstr[20];
+#ifdef WIN32
+        DWORD tid = GetCurrentThreadId();
+#else
+        long tid = 0;
+#endif
+        memcpy(fmt2, fmt, cp-fmt);
+        fmt2[cp-fmt] = '\0';
+        sprintf(tpidstr, "%08lx", (long) tid);
+        strcat(fmt2, tpidstr);
+        strcat(fmt2, cp+2);
+        strftime(dst, sz, fmt2, tm);     
+    }
+    else
+        strftime(dst, sz, fmt, tm);
+}
+                            
+static void yaz_log_to_file(int level, FILE *file, const char *buf)
+{
+    time_t ti = time(0);
     int i;
-    time_t ti;
-    struct tm *tim;
-    char tbuf[TIMEFORMAT_LEN] = "";
-    int o_level = level;
-    int flen; 
+#if HAVE_LOCALTIME_R
+    struct tm tm0, *tm = &tm0;
+#else
+    struct tm *tm;
+#endif
 
-    if (!(level & l_level))
-        return;
     init_mutex();
+
     nmem_mutex_enter(log_mutex);
-    if (!l_file)
-        l_file = stderr;
+
+    #if HAVE_LOCALTIME_R
+    localtime_r(&ti, tm);
+#else
+    tm = localtime(&ti);
+#endif
     
-    if ((l_file != stderr) && (l_max_size>0))
-    {
-        flen = ftell(l_file);
-        if (flen>l_max_size) 
-            rotate_log();
-    }
+    yaz_log_open_check(tm, 0);  
+    file = yaz_log_file(); /* file may change in yaz_log_open_check */
 
-    *flags = '\0';
-    for (i = 0; level && mask_names[i].name; i++)
-        if ( mask_names[i].mask & level)
-        {
-            if (*mask_names[i].name && mask_names[i].mask && 
-                 mask_names[i].mask != YLOG_ALL)
+    if (file)
+    {
+        char tbuf[TIMEFORMAT_LEN];
+        char flags[1024];
+        
+        *flags = '\0';
+        for (i = 0; level && mask_names[i].name; i++)
+            if ( mask_names[i].mask & level)
             {
-                sprintf(flags + strlen(flags), "[%s]", mask_names[i].name);
-                level &= ~mask_names[i].mask;
+                if (*mask_names[i].name && mask_names[i].mask && 
+                    mask_names[i].mask != YLOG_ALL)
+                {
+                    sprintf(flags + strlen(flags), "[%s]", mask_names[i].name);
+                    level &= ~mask_names[i].mask;
+                }
             }
-        }
+        
+        if (l_level & YLOG_NOTIME)
+            tbuf[0] = '\0';
+        else
+            yaz_strftime(tbuf, TIMEFORMAT_LEN-1, l_actual_format, tm);
+        tbuf[TIMEFORMAT_LEN-1] = '\0';
+        
+        fprintf(file, "%s %s%s %s%s\n", tbuf, l_prefix, flags, l_prefix2, buf);
+        if (l_level & (YLOG_FLUSH|YLOG_DEBUG) )
+            fflush(file);
+    }
+    nmem_mutex_leave(log_mutex);
+}
 
+void yaz_log(int level, const char *fmt, ...)
+{
+    va_list ap;
+    char buf[4096];
+    FILE *file;
+    int o_level = level;
+
+    if (l_level < 0)
+        l_level = default_log_level();
+    if (!(level & l_level))
+        return;
     va_start(ap, fmt);
 #ifdef WIN32
     _vsnprintf(buf, sizeof(buf)-1, fmt, ap);
@@ -313,20 +436,13 @@ void yaz_log(int level, const char *fmt, ...)
     va_end (ap);
     if (start_hook_func)
         (*start_hook_func)(o_level, buf, start_hook_info);
-    ti = time(0);
-    tim = localtime(&ti);
-    if (l_level & YLOG_NOTIME)
-        tbuf[0] = '\0';
-    else
-        strftime(tbuf, TIMEFORMAT_LEN-1, l_actual_format, tim);
-    tbuf[TIMEFORMAT_LEN-1] = '\0';
-    fprintf(l_file, "%s %s%s %s%s\n", tbuf, l_prefix, flags,
-            l_prefix2, buf);
-    if (l_level & (YLOG_FLUSH|YLOG_DEBUG) )
-        fflush(l_file);
+    if (hook_func)
+        (*hook_func)(o_level, buf, hook_info);
+    file = yaz_log_file();
+    if (file)
+        yaz_log_to_file(level, file, buf);
     if (end_hook_func)
         (*end_hook_func)(o_level, buf, end_hook_info);
-    nmem_mutex_leave(log_mutex);
 }
 
 void yaz_log_time_format(const char *fmt)
@@ -336,7 +452,7 @@ void yaz_log_time_format(const char *fmt)
         l_actual_format = l_new_default_format;
         return; 
     }
-    if (0==strcmp(fmt,"old"))
+    if (0==strcmp(fmt, "old"))
     { /* force the old format */
         l_actual_format = l_old_default_format;
         return; 
@@ -356,9 +472,9 @@ static char *clean_name(const char *name, int len, char *namebuf, int buflen)
         len = buflen-1; 
     strncpy(namebuf, name, len);
     namebuf[len] = '\0';
-    while ((p = strchr(start,'/')))
+    while ((p = strchr(start, '/')))
         start = p+1;
-    if ((p = strrchr(start,'.')))
+    if ((p = strrchr(start, '.')))
         *p = '\0';
     return start;
 }
@@ -382,7 +498,7 @@ static int define_module_bit(const char *name)
     }
     mask_names[i].mask = next_log_bit;
     next_log_bit = next_log_bit<<1;
-    mask_names[i].name = malloc (strlen(name)+1);
+    mask_names[i].name = malloc(strlen(name)+1);
     strcpy(mask_names[i].name, name);
     mask_names[i+1].name = NULL;
     mask_names[i+1].mask = 0;
@@ -413,12 +529,12 @@ int yaz_log_module_level(const char *name)
     return 0;
 }
 
-int yaz_log_mask_str (const char *str)
+int yaz_log_mask_str(const char *str)
 {
-    return yaz_log_mask_str_x (str, YLOG_DEFAULT_LEVEL);
+    return yaz_log_mask_str_x(str, default_log_level());
 }
 
-int yaz_log_mask_str_x (const char *str, int level)
+int yaz_log_mask_str_x(const char *str, int level)
 {
     const char *p;
 
@@ -461,4 +577,3 @@ int yaz_log_mask_str_x (const char *str, int level)
  * End:
  * vim: shiftwidth=4 tabstop=8 expandtab
  */
-