src/tbm_log.c

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