common/tdm_log.c

   1 /**************************************************************************
   2  *
   3  * libtdm
   4  *
   5  * Copyright 2015 Samsung Electronics co., Ltd. All Rights Reserved.
   6  *
   7  * Contact: Eunchul Kim <chulspro.kim@samsung.com>,
   8  *          JinYoung Jeon <jy0.jeon@samsung.com>,
   9  *          Taeheon Kim <th908.kim@samsung.com>,
  10  *          YoungJun Cho <yj44.cho@samsung.com>,
  11  *          SooChan Lim <sc1.lim@samsung.com>,
  12  *          Boram Park <sc1.lim@samsung.com>
  13  *
  14  * Permission is hereby granted, free of charge, to any person obtaining a
  15  * copy of this software and associated documentation files (the
  16  * "Software"), to deal in the Software without restriction, including
  17  * without limitation the rights to use, copy, modify, merge, publish,
  18  * distribute, sub license, and/or sell copies of the Software, and to
  19  * permit persons to whom the Software is furnished to do so, subject to
  20  * the following conditions:
  21  *
  22  * The above copyright notice and this permission notice (including the
  23  * next paragraph) shall be included in all copies or substantial portions
  24  * of the Software.
  25  *
  26  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  27  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  28  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  29  * IN NO EVENT SHALL PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR
  30  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  31  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  32  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  33  *
  34 **************************************************************************/
  35
  36 #ifdef HAVE_CONFIG_H
  37 #include "config.h"
  38 #endif
  39
  40 #include <stdio.h>
  41 #include <unistd.h>
  42 #include <time.h>
  43 #include <assert.h>
  44 #include <sys/syscall.h>
  45 #include <dlog.h>
  46 #include <pthread.h>
  47
  48 #include "tdm.h"
  49 #include "tdm_log.h"
  50 #include "tdm_macro.h"
  51
  52 #define LOG_MAX_LEN    4076
  53
  54 #define COLOR_RED "\x1b[31m"      /* for error */
  55 #define COLOR_YELLOW "\x1b[33m"   /* for warning */
  56 #define COLOR_GREEN "\x1b[32m"    /* for info */
  57 #define COLOR_RESET "\x1b[0m"
  58
  59 #undef LOG_TAG
  60 #define LOG_TAG "TDM"
  61
  62 static unsigned int dlog_enable = 1;
  63 static unsigned int color_enable = 1;
  64
  65 static unsigned int assert_level = TDM_LOG_LEVEL_NONE;
  66
  67 static unsigned int log_lock_init;
  68 static pthread_mutex_t log_lock;
  69
  70 unsigned int tdm_log_debug_level = TDM_LOG_LEVEL_INFO;
  71
  72 static int stdout_fd = -1;
  73
  74 EXTERN void
  75 tdm_log_enable_color(unsigned int enable)
  76 {
  77         color_enable = enable;
  78 }
  79
  80 EXTERN void
  81 tdm_log_enable_dlog(unsigned int enable)
  82 {
  83         const char *str = getenv("TDM_DLOG");
  84         if (str)
  85                 enable = (str[0] == '1') ? 1 : 0;
  86         dlog_enable = enable;
  87 }
  88
  89 EXTERN void
  90 tdm_log_set_debug_level(int level)
  91 {
  92         const char *str = getenv("TDM_DEBUG_LEVEL");
  93         if (str)
  94                 level = str[0] - '0';
  95         tdm_log_debug_level = level;
  96 }
  97
  98 EXTERN void
  99 tdm_log_set_assert_level(int level)
 100 {
 101         const char *str = getenv("TDM_ASSERT_LEVEL");
 102         if (str)
 103                 level = str[0] - '0';
 104         assert_level = level;
 105 }
 106
 107 EXTERN void
 108 tdm_log_set_path(const char *path)
 109 {
 110         if (!path) {
 111                 if (stdout_fd != -1) {
 112                         fflush(stdout);
 113                         close(STDOUT_FILENO);
 114                         dup2(stdout_fd, STDOUT_FILENO);
 115                         close(stdout_fd);
 116                         stdout_fd = -1;
 117                 }
 118         } else {
 119                 char fd_name[TDM_PATH_LEN];
 120                 int  log_fd = -1;
 121                 FILE *log_fl;
 122
 123                 snprintf(fd_name, TDM_PATH_LEN, "%s", path);
 124
 125                 log_fl = fopen(fd_name, "a");
 126                 if (!log_fl) {
 127                         TDM_ERR("failed: open file(%s)\n", fd_name);
 128                         return;
 129                 }
 130
 131                 if (stdout_fd == -1) {
 132                         fflush(stdout);
 133                         stdout_fd = dup(STDOUT_FILENO);
 134                         if (stdout_fd < 0) {
 135                                 TDM_ERR("dup failed: %m\n");
 136                                 fclose(log_fl);
 137                                 return;
 138                         }
 139                 }
 140
 141                 setvbuf(log_fl, NULL, _IOLBF, 512);
 142                 log_fd = fileno(log_fl);
 143
 144                 close(STDOUT_FILENO);
 145                 dup2(log_fd, STDOUT_FILENO);
 146                 fclose(log_fl);
 147         }
 148 }
 149
 150 static void
 151 _tdm_log_vprint_stdout(int level, const char *fmt, va_list ap)
 152 {
 153         if (!log_lock_init) {
 154                 log_lock_init = 1;
 155                 pthread_mutex_init(&log_lock, NULL);
 156         }
 157
 158         if (level > tdm_log_debug_level)
 159                 return;
 160
 161         char *lvl_str[] = {"TDM_NON", "TDM_ERR", "TDM_WRN", "TDM_INF", "TDM_DBG"};
 162         char *color[] = {COLOR_RESET, COLOR_RED, COLOR_YELLOW, COLOR_GREEN, COLOR_RESET};
 163
 164         pthread_mutex_lock(&log_lock);
 165
 166         if (color_enable)
 167                 printf("%s", color[level]);
 168         printf("[%s]", lvl_str[level]);
 169         if (color_enable)
 170                 printf(COLOR_RESET);
 171         vprintf(fmt, ap);
 172         printf("\n");
 173         pthread_mutex_unlock(&log_lock);
 174 }
 175
 176
 177 EXTERN void
 178 tdm_log_print_stdout(int level, const char *fmt, ...)
 179 {
 180         va_list arg;
 181         va_start(arg, fmt);
 182         _tdm_log_vprint_stdout(level, fmt, arg);
 183         va_end(arg);
 184 }
 185
 186 EXTERN void
 187 tdm_log_print(int level, const char *fmt, ...)
 188 {
 189         va_list arg;
 190
 191         if (level > tdm_log_debug_level)
 192                 return;
 193
 194         if (dlog_enable) {
 195                 log_priority dlog_prio;
 196                 switch (level) {
 197                 case TDM_LOG_LEVEL_ERR:
 198                         dlog_prio = DLOG_ERROR;
 199                         break;
 200                 case TDM_LOG_LEVEL_WRN:
 201                         dlog_prio = DLOG_WARN;
 202                         break;
 203                 case TDM_LOG_LEVEL_INFO:
 204                         dlog_prio = DLOG_INFO;
 205                         break;
 206                 case TDM_LOG_LEVEL_DBG:
 207                         dlog_prio = DLOG_DEBUG;
 208                         break;
 209                 default:
 210                         return;
 211                 }
 212                 va_start(arg, fmt);
 213                 __dlog_vprint(LOG_ID_SYSTEM, dlog_prio, LOG_TAG, fmt, arg);
 214                 va_end(arg);
 215         } else {
 216                 va_start(arg, fmt);
 217                 _tdm_log_vprint_stdout(level, fmt, arg);
 218                 va_end(arg);
 219         }
 220
 221         assert(level > assert_level);
 222 }
 223
 224 EXTERN void
 225 tdm_log_reset(void)
 226 {
 227         pthread_mutex_unlock(&log_lock);
 228 }