1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
#include "logger.h"
#include "myutils.h"
#include "platform.h"
#include <stdarg.h>
#include <stdio.h>
#ifdef WIN32
#include <Windows.h>
#include <time.h>
static void flockfile(FILE* fp) { _lock_file(fp); }
static void funlockfile(FILE* fp) { _unlock_file(fp); }
static const void* current_thread_id(void)
{
return reinterpret_cast<const void*>(static_cast<uintptr_t>(GetCurrentThreadId()));
}
#else
#include <pthread.h>
static const void* current_thread_id(void) { return (void*)(pthread_self()); }
#endif
namespace securefs
{
void Logger::vlog(
LoggingLevel level, const char* funcsig, int lineno, const char* format, va_list args) noexcept
{
if (!m_fp || level < this->get_level())
return;
struct tm now;
int now_ns = 0;
OSService::get_current_time_in_tm(&now, &now_ns);
flockfile(m_fp);
DEFER(funlockfile(m_fp));
if (m_console_color)
{
switch (level)
{
case kLogWarning:
m_console_color->use(Colour::Warning);
break;
case kLogError:
m_console_color->use(Colour::Error);
break;
default:
break;
}
}
fprintf(m_fp,
"[%s] [%p] [%d-%02d-%02d %02d:%02d:%02d.%09d UTC] [%s:%d] ",
stringify(level),
current_thread_id(),
now.tm_year + 1900,
now.tm_mon + 1,
now.tm_mday,
now.tm_hour,
now.tm_min,
now.tm_sec,
now_ns,
funcsig,
lineno);
vfprintf(m_fp, format, args);
if (m_console_color && (level == kLogWarning || level == kLogError))
{
m_console_color->use(Colour::Default);
}
putc('\n', m_fp);
fflush(m_fp);
}
void Logger::log(
LoggingLevel level, const char* funcsig, int lineno, const char* format, ...) noexcept
{
if (!m_fp || level < this->get_level())
return;
va_list args;
va_start(args, format);
vlog(level, funcsig, lineno, format, args);
va_end(args);
}
Logger::Logger(FILE* fp, bool close_on_exit)
: m_level(kLogInfo), m_fp(fp), m_close_on_exit(close_on_exit)
{
m_console_color = ConsoleColourSetter::create_setter(m_fp);
}
Logger::~Logger()
{
if (m_close_on_exit)
fclose(m_fp);
}
Logger* Logger::create_stderr_logger() { return new Logger(stderr, false); }
Logger* Logger::create_file_logger(const std::string& path)
{
#ifdef WIN32
FILE* fp = _wfopen(widen_string(path).c_str(), L"a");
#else
FILE* fp = fopen(path.c_str(), "a");
#endif
if (!fp)
THROW_POSIX_EXCEPTION(errno, path);
return new Logger(fp, true);
}
Logger* global_logger = Logger::create_stderr_logger();
} // namespace securefs
|
