kumi/danicoin
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity
danicoin/contrib/epee/include/misc_log_ex.cpp
Antonio Juarez 4363a9f100 Multi-signatures added
2014-08-13 15:07:19 +01:00

1029 lines
34 KiB
C++
Raw Blame History

// Copyright (c) 2006-2013, Andrey N. Sabelnikov, www.sabelnikov.net
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
// * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
// * Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
// * Neither the name of the Andrey N. Sabelnikov nor the
// names of its contributors may be used to endorse or promote products
// derived from this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
// WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER BE LIABLE FOR ANY
// DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
// ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
#include "misc_log_ex.h"
#include <atomic>
#include <string>
#include <iostream>
#include <sstream>
#include <iomanip>
#include <fstream>
#include <algorithm>
#include <list>
#include <map>
#include <time.h>
#include <boost/cstdint.hpp>
#include <boost/filesystem.hpp>
#if defined(WIN32)
#include <io.h>
#else
#include <unistd.h>
#endif
#include "static_initializer.h"
#include "string_tools.h"
#include "time_helper.h"
#include "misc_os_dependent.h"
#include "syncobj.h"
#define LOG_LEVEL_SILENT -1
#define LOG_LEVEL_0 0
#define LOG_LEVEL_1 1
#define LOG_LEVEL_2 2
#define LOG_LEVEL_3 3
#define LOG_LEVEL_4 4
#define LOG_LEVEL_MIN LOG_LEVEL_SILENT
#define LOG_LEVEL_MAX LOG_LEVEL_4
#define LOGGER_NULL 0
#define LOGGER_FILE 1
#define LOGGER_DEBUGGER 2
#define LOGGER_CONSOLE 3
#define LOGGER_DUMP 4
#ifndef LOCAL_ASSERT
#include <assert.h>
#if (defined _MSC_VER)
#define LOCAL_ASSERT(expr) {if(epee::debug::get_set_enable_assert()){_ASSERTE(expr);}}
#else
#define LOCAL_ASSERT(expr)
#endif
#endif
namespace epee {
namespace log_space {
//----------------------------------------------------------------------------
bool is_stdout_a_tty()
{
static std::atomic<bool> initialized(false);
static std::atomic<bool> is_a_tty(false);
if (!initialized.load(std::memory_order_acquire))
{
#if defined(WIN32)
is_a_tty.store(0 != _isatty(_fileno(stdout)), std::memory_order_relaxed);
#else
is_a_tty.store(0 != isatty(fileno(stdout)), std::memory_order_relaxed);
#endif
initialized.store(true, std::memory_order_release);
}
return is_a_tty.load(std::memory_order_relaxed);
}
//----------------------------------------------------------------------------
void set_console_color(int color, bool bright)
{
if (!is_stdout_a_tty())
return;
switch(color)
{
case console_color_default:
{
#ifdef WIN32
HANDLE h_stdout = GetStdHandle(STD_OUTPUT_HANDLE);
SetConsoleTextAttribute(h_stdout, FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE| (bright ? FOREGROUND_INTENSITY:0));
#else
if(bright)
std::cout << "\033[1;37m";
else
std::cout << "\033[0m";
#endif
}
break;
case console_color_white:
{
#ifdef WIN32
HANDLE h_stdout = GetStdHandle(STD_OUTPUT_HANDLE);
SetConsoleTextAttribute(h_stdout, FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | (bright ? FOREGROUND_INTENSITY:0));
#else
if(bright)
std::cout << "\033[1;37m";
else
std::cout << "\033[0;37m";
#endif
}
break;
case console_color_red:
{
#ifdef WIN32
HANDLE h_stdout = GetStdHandle(STD_OUTPUT_HANDLE);
SetConsoleTextAttribute(h_stdout, FOREGROUND_RED | (bright ? FOREGROUND_INTENSITY:0));
#else
if(bright)
std::cout << "\033[1;31m";
else
std::cout << "\033[0;31m";
#endif
}
break;
case console_color_green:
{
#ifdef WIN32
HANDLE h_stdout = GetStdHandle(STD_OUTPUT_HANDLE);
SetConsoleTextAttribute(h_stdout, FOREGROUND_GREEN | (bright ? FOREGROUND_INTENSITY:0));
#else
if(bright)
std::cout << "\033[1;32m";
else
std::cout << "\033[0;32m";
#endif
}
break;
case console_color_blue:
{
#ifdef WIN32
HANDLE h_stdout = GetStdHandle(STD_OUTPUT_HANDLE);
SetConsoleTextAttribute(h_stdout, FOREGROUND_BLUE | FOREGROUND_INTENSITY);//(bright ? FOREGROUND_INTENSITY:0));
#else
if(bright)
std::cout << "\033[1;34m";
else
std::cout << "\033[0;34m";
#endif
}
break;
case console_color_cyan:
{
#ifdef WIN32
HANDLE h_stdout = GetStdHandle(STD_OUTPUT_HANDLE);
SetConsoleTextAttribute(h_stdout, FOREGROUND_GREEN | FOREGROUND_BLUE | (bright ? FOREGROUND_INTENSITY:0));
#else
if(bright)
std::cout << "\033[1;36m";
else
std::cout << "\033[0;36m";
#endif
}
break;
case console_color_magenta:
{
#ifdef WIN32
HANDLE h_stdout = GetStdHandle(STD_OUTPUT_HANDLE);
SetConsoleTextAttribute(h_stdout, FOREGROUND_BLUE | FOREGROUND_RED | (bright ? FOREGROUND_INTENSITY:0));
#else
if(bright)
std::cout << "\033[1;35m";
else
std::cout << "\033[0;35m";
#endif
}
break;
case console_color_yellow:
{
#ifdef WIN32
HANDLE h_stdout = GetStdHandle(STD_OUTPUT_HANDLE);
SetConsoleTextAttribute(h_stdout, FOREGROUND_RED | FOREGROUND_GREEN | (bright ? FOREGROUND_INTENSITY:0));
#else
if(bright)
std::cout << "\033[1;33m";
else
std::cout << "\033[0;33m";
#endif
}
break;
}
}
//----------------------------------------------------------------------------
void reset_console_color() {
if (!is_stdout_a_tty())
return;
#ifdef WIN32
HANDLE h_stdout = GetStdHandle(STD_OUTPUT_HANDLE);
SetConsoleTextAttribute(h_stdout, FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE);
#else
std::cout << "\033[0m";
std::cout.flush();
#endif
}
//----------------------------------------------------------------------------
bool rotate_log_file(const char* pfile_path)
{
#ifdef _MSC_VER
if(!pfile_path)
return false;
std::string file_path = pfile_path;
std::string::size_type a = file_path .rfind('.');
if ( a != std::string::npos )
file_path .erase( a, file_path .size());
::DeleteFileA( (file_path + ".0").c_str() );
::MoveFileA( (file_path + ".log").c_str(), (file_path + ".0").c_str() );
#else
return false;//not implemented yet
#endif
return true;
}
//----------------------------------------------------------------------------
#ifdef _MSC_VER
bool debug_output_stream::out_buffer( const char* buffer, int buffer_len , int log_level, int color, const char* plog_name/* = NULL*/)
{
for ( int i = 0; i < buffer_len; i = i + max_dbg_str_len )
{
std::string s( buffer + i, buffer_len- i < max_dbg_str_len ?
buffer_len - i : max_dbg_str_len );
::OutputDebugStringA( s.c_str() );
}
return true;
}
#endif
//----------------------------------------------------------------------------
console_output_stream::console_output_stream()
{
#ifdef _MSC_VER
if(!::GetStdHandle(STD_OUTPUT_HANDLE))
m_have_to_kill_console = true;
else
m_have_to_kill_console = false;
::AllocConsole();
#endif
}
//----------------------------------------------------------------------------
console_output_stream::~console_output_stream()
{
#ifdef _MSC_VER
if(m_have_to_kill_console)
::FreeConsole();
#endif
}
//----------------------------------------------------------------------------
bool console_output_stream::out_buffer( const char* buffer, int buffer_len , int log_level, int color, const char* plog_name/* = NULL*/)
{
if(plog_name)
return true; //skip alternative logs from console
set_console_color(color, log_level < 1);
#ifdef _MSC_VER
const char* ptarget_buf = NULL;
char* pallocated_buf = NULL;
//
int i = 0;
for(; i < buffer_len; i++)
if(buffer[i] == '\a') break;
if(i == buffer_len)
ptarget_buf = buffer;
else
{
pallocated_buf = new char[buffer_len];
ptarget_buf = pallocated_buf;
for(i = 0; i < buffer_len; i++)
{
if(buffer[i] == '\a')
pallocated_buf[i] = '^';
else
pallocated_buf[i] = buffer[i];
}
}
//uint32_t b = 0;
//::WriteConsoleA(::GetStdHandle(STD_OUTPUT_HANDLE), ptarget_buf, buffer_len, (DWORD*)&b, 0);
std::cout << ptarget_buf;
if(pallocated_buf) delete [] pallocated_buf;
#else
std::string buf(buffer, buffer_len);
for(size_t i = 0; i!= buf.size(); i++)
{
if(buf[i] == 7 || buf[i] == -107)
buf[i] = '^';
}
std::cout << buf;
#endif
reset_console_color();
return true;
}
//----------------------------------------------------------------------------
file_output_stream::file_output_stream(std::string default_log_file_name, std::string log_path)
{
m_default_log_filename = default_log_file_name;
m_max_logfile_size = 0;
m_default_log_path = log_path;
m_pdefault_file_stream = add_new_stream_and_open(default_log_file_name.c_str());
}
//----------------------------------------------------------------------------
file_output_stream::~file_output_stream()
{
for(named_log_streams::iterator it = m_log_file_names.begin(); it!=m_log_file_names.end(); it++)
{
if ( it->second->is_open() )
{
it->second->flush();
it->second->close();
}
delete it->second;
}
}
//----------------------------------------------------------------------------
std::ofstream* file_output_stream::add_new_stream_and_open(const char* pstream_name)
{
//log_space::rotate_log_file((m_default_log_path + "\\" + pstream_name).c_str());
std::ofstream* pstream = (m_log_file_names[pstream_name] = new std::ofstream);
std::string target_path = m_default_log_path + "/" + pstream_name;
pstream->open( target_path.c_str(), std::ios_base::out | std::ios::app /*ios_base::trunc */);
if(pstream->fail())
return NULL;
return pstream;
}
//----------------------------------------------------------------------------
bool file_output_stream::set_max_logfile_size(uint64_t max_size)
{
m_max_logfile_size = max_size;
return true;
}
//----------------------------------------------------------------------------
bool file_output_stream::set_log_rotate_cmd(const std::string& cmd)
{
m_log_rotate_cmd = cmd;
return true;
}
//----------------------------------------------------------------------------
bool file_output_stream::out_buffer(const char* buffer, int buffer_len, int log_level, int color, const char* plog_name/* = NULL*/)
{
std::ofstream* m_target_file_stream = m_pdefault_file_stream;
if(plog_name)
{ //find named stream
named_log_streams::iterator it = m_log_file_names.find(plog_name);
if(it == m_log_file_names.end())
m_target_file_stream = add_new_stream_and_open(plog_name);
else
m_target_file_stream = it->second;
}
if(!m_target_file_stream || !m_target_file_stream->is_open())
return false;//TODO: add assert here
m_target_file_stream->write(buffer, buffer_len );
m_target_file_stream->flush();
if(m_max_logfile_size)
{
std::ofstream::pos_type pt = m_target_file_stream->tellp();
uint64_t current_sz = pt;
if(current_sz > m_max_logfile_size)
{
std::cout << "current_sz= " << current_sz << " m_max_logfile_size= " << m_max_logfile_size << std::endl;
std::string log_file_name;
if(!plog_name)
log_file_name = m_default_log_filename;
else
log_file_name = plog_name;
m_target_file_stream->close();
std::string new_log_file_name = log_file_name;
time_t tm = 0;
time(&tm);
int err_count = 0;
boost::system::error_code ec;
do
{
new_log_file_name = string_tools::cut_off_extension(log_file_name);
if(err_count)
new_log_file_name += misc_utils::get_time_str_v2(tm) + "(" + boost::lexical_cast<std::string>(err_count) + ")" + ".log";
else
new_log_file_name += misc_utils::get_time_str_v2(tm) + ".log";
err_count++;
}while(boost::filesystem::exists(m_default_log_path + "/" + new_log_file_name, ec));
std::string new_log_file_path = m_default_log_path + "/" + new_log_file_name;
boost::filesystem::rename(m_default_log_path + "/" + log_file_name, new_log_file_path, ec);
if(ec)
{
std::cout << "Filed to rename, ec = " << ec.message() << std::endl;
}
if(m_log_rotate_cmd.size())
{
std::string m_log_rotate_cmd_local_copy = m_log_rotate_cmd;
//boost::replace_all(m_log_rotate_cmd, "[*SOURCE*]", new_log_file_path);
boost::replace_all(m_log_rotate_cmd_local_copy, "[*TARGET*]", new_log_file_path);
misc_utils::call_sys_cmd(m_log_rotate_cmd_local_copy);
}
m_target_file_stream->open( (m_default_log_path + "/" + log_file_name).c_str(), std::ios_base::out | std::ios::app /*ios_base::trunc */);
if(m_target_file_stream->fail())
return false;
}
}
return true;
}
//----------------------------------------------------------------------------
log_stream_splitter::~log_stream_splitter()
{
//free pointers
std::for_each(m_log_streams.begin(), m_log_streams.end(), delete_ptr());
}
//----------------------------------------------------------------------------
bool log_stream_splitter::set_max_logfile_size(uint64_t max_size)
{
for(streams_container::iterator it = m_log_streams.begin(); it!=m_log_streams.end();it++)
it->first->set_max_logfile_size(max_size);
return true;
}
//----------------------------------------------------------------------------
bool log_stream_splitter::set_log_rotate_cmd(const std::string& cmd)
{
for(streams_container::iterator it = m_log_streams.begin(); it!=m_log_streams.end();it++)
it->first->set_log_rotate_cmd(cmd);
return true;
}
//----------------------------------------------------------------------------
bool log_stream_splitter::do_log_message(const std::string& rlog_mes, int log_level, int color, const char* plog_name/* = NULL*/)
{
std::string str_mess = rlog_mes;
size_t str_len = str_mess.size();
const char* pstr = str_mess.c_str();
for(streams_container::iterator it = m_log_streams.begin(); it!=m_log_streams.end();it++)
if(it->second >= log_level)
it->first->out_buffer(pstr, (int)str_len, log_level, color, plog_name);
return true;
}
//----------------------------------------------------------------------------
bool log_stream_splitter::add_logger(int type, const char* pdefault_file_name, const char* pdefault_log_folder, int log_level_limit/* = LOG_LEVEL_4*/)
{
ibase_log_stream* ls = NULL;
switch( type )
{
case LOGGER_FILE:
ls = new file_output_stream( pdefault_file_name, pdefault_log_folder );
break;
case LOGGER_DEBUGGER:
#ifdef _MSC_VER
ls = new debug_output_stream( );
#else
return false;//not implemented yet
#endif
break;
case LOGGER_CONSOLE:
ls = new console_output_stream( );
break;
}
if ( ls ) {
m_log_streams.push_back(streams_container::value_type(ls, log_level_limit));
return true;
}
return ls ? true:false;
}
//----------------------------------------------------------------------------
bool log_stream_splitter::add_logger(ibase_log_stream* pstream, int log_level_limit/* = LOG_LEVEL_4*/)
{
m_log_streams.push_back(streams_container::value_type(pstream, log_level_limit) );
return true;
}
//----------------------------------------------------------------------------
bool log_stream_splitter::remove_logger(int type)
{
streams_container::iterator it = m_log_streams.begin();
for(;it!=m_log_streams.end(); it++)
{
if(it->first->get_type() == type)
{
delete it->first;
m_log_streams.erase(it);
return true;
}
}
return false;
}
//----------------------------------------------------------------------------
std::string get_daytime_string2()
{
boost::posix_time::ptime p = boost::posix_time::microsec_clock::local_time();
return misc_utils::get_time_str_v3(p);
}
//----------------------------------------------------------------------------
std::string get_day_time_string()
{
return get_daytime_string2();
//time_t tm = 0;
//time(&tm);
//return misc_utils::get_time_str(tm);
}
//----------------------------------------------------------------------------
std::string get_time_string()
{
return get_daytime_string2();
}
//----------------------------------------------------------------------------
#ifdef _MSC_VER
std::string get_time_string_adv(SYSTEMTIME* pst/* = NULL*/)
{
SYSTEMTIME st = {0};
if(!pst)
{
pst = &st;
GetSystemTime(&st);
}
std::stringstream str_str;
str_str.fill('0');
str_str << std::setw(2) << pst->wHour << "_"
<< std::setw(2) << pst->wMinute << "_"
<< std::setw(2) << pst->wSecond << "_"
<< std::setw(3) << pst->wMilliseconds;
return str_str.str();
}
#endif
//----------------------------------------------------------------------------
logger::logger()
{
CRITICAL_REGION_BEGIN(m_critical_sec);
init();
CRITICAL_REGION_END();
}
//----------------------------------------------------------------------------
bool logger::set_max_logfile_size(uint64_t max_size)
{
CRITICAL_REGION_BEGIN(m_critical_sec);
m_log_target.set_max_logfile_size(max_size);
CRITICAL_REGION_END();
return true;
}
//----------------------------------------------------------------------------
bool logger::set_log_rotate_cmd(const std::string& cmd)
{
CRITICAL_REGION_BEGIN(m_critical_sec);
m_log_target.set_log_rotate_cmd(cmd);
CRITICAL_REGION_END();
return true;
}
//----------------------------------------------------------------------------
bool logger::take_away_journal(std::list<std::string>& journal)
{
CRITICAL_REGION_BEGIN(m_critical_sec);
m_journal.swap(journal);
CRITICAL_REGION_END();
return true;
}
//----------------------------------------------------------------------------
bool logger::do_log_message(const std::string& rlog_mes, int log_level, int color, bool add_to_journal/* = false*/, const char* plog_name/* = NULL*/)
{
CRITICAL_REGION_BEGIN(m_critical_sec);
m_log_target.do_log_message(rlog_mes, log_level, color, plog_name);
if(add_to_journal)
m_journal.push_back(rlog_mes);
return true;
CRITICAL_REGION_END();
}
//----------------------------------------------------------------------------
bool logger::add_logger( int type, const char* pdefault_file_name, const char* pdefault_log_folder, int log_level_limit/* = LOG_LEVEL_4*/)
{
CRITICAL_REGION_BEGIN(m_critical_sec);
return m_log_target.add_logger( type, pdefault_file_name, pdefault_log_folder, log_level_limit);
CRITICAL_REGION_END();
}
//----------------------------------------------------------------------------
bool logger::add_logger( ibase_log_stream* pstream, int log_level_limit/* = LOG_LEVEL_4*/)
{
CRITICAL_REGION_BEGIN(m_critical_sec);
return m_log_target.add_logger(pstream, log_level_limit);
CRITICAL_REGION_END();
}
//----------------------------------------------------------------------------
bool logger::remove_logger(int type)
{
CRITICAL_REGION_BEGIN(m_critical_sec);
return m_log_target.remove_logger(type);
CRITICAL_REGION_END();
}
//----------------------------------------------------------------------------
bool logger::set_thread_prefix(const std::string& prefix)
{
CRITICAL_REGION_BEGIN(m_critical_sec);
m_thr_prefix_strings[misc_utils::get_thread_string_id()] = prefix;
CRITICAL_REGION_END();
return true;
}
//----------------------------------------------------------------------------
bool logger::init()
{
m_process_name = string_tools::get_current_module_name();
init_log_path_by_default();
//init default set of loggers
init_default_loggers();
std::stringstream ss;
ss << get_time_string() << " Init logging. Level=" << get_set_log_detalisation_level()
<< " Log path=" << m_default_log_folder << std::endl;
this->do_log_message(ss.str(), console_color_white, LOG_LEVEL_0);
return true;
}
//----------------------------------------------------------------------------
bool logger::init_default_loggers()
{
return true;
}
//----------------------------------------------------------------------------
bool logger::init_log_path_by_default()
{
//load process name
m_default_log_folder = string_tools::get_current_module_folder();
m_default_log_file = m_process_name;
std::string::size_type a = m_default_log_file.rfind('.');
if ( a != std::string::npos )
m_default_log_file.erase( a, m_default_log_file.size());
m_default_log_file += ".log";
return true;
}
//----------------------------------------------------------------------------
int log_singletone::get_log_detalisation_level()
{
get_or_create_instance();//to initialize logger, if it not initialized
return get_set_log_detalisation_level();
}
//----------------------------------------------------------------------------
bool log_singletone::is_filter_error(int error_code)
{
return false;
}
//----------------------------------------------------------------------------
bool log_singletone::do_log_message(const std::string& rlog_mes, int log_level, int color, bool keep_in_journal, const char* plog_name/* = NULL*/)
{
logger* plogger = get_or_create_instance();
bool res = false;
if(plogger)
res = plogger->do_log_message(rlog_mes, log_level, color, keep_in_journal, plog_name);
else
{ //globally uninitialized, create new logger for each call of do_log_message() and then delete it
plogger = new logger();
//TODO: some extra initialization
res = plogger->do_log_message(rlog_mes, log_level, color, keep_in_journal, plog_name);
delete plogger;
plogger = NULL;
}
return res;
}
//----------------------------------------------------------------------------
bool log_singletone::take_away_journal(std::list<std::string>& journal)
{
logger* plogger = get_or_create_instance();
bool res = false;
if(plogger)
res = plogger->take_away_journal(journal);
return res;
}
//----------------------------------------------------------------------------
bool log_singletone::set_max_logfile_size(uint64_t file_size)
{
logger* plogger = get_or_create_instance();
if(!plogger) return false;
return plogger->set_max_logfile_size(file_size);
}
//----------------------------------------------------------------------------
bool log_singletone::set_log_rotate_cmd(const std::string& cmd)
{
logger* plogger = get_or_create_instance();
if(!plogger) return false;
return plogger->set_log_rotate_cmd(cmd);
}
//----------------------------------------------------------------------------
bool log_singletone::add_logger( int type, const char* pdefault_file_name, const char* pdefault_log_folder, int log_level_limit/* = LOG_LEVEL_4*/)
{
logger* plogger = get_or_create_instance();
if(!plogger) return false;
return plogger->add_logger(type, pdefault_file_name, pdefault_log_folder, log_level_limit);
}
//----------------------------------------------------------------------------
std::string log_singletone::get_default_log_file()
{
logger* plogger = get_or_create_instance();
if(plogger)
return plogger->get_default_log_file();
return "";
}
//----------------------------------------------------------------------------
std::string log_singletone::get_default_log_folder()
{
logger* plogger = get_or_create_instance();
if(plogger)
return plogger->get_default_log_folder();
return "";
}
//----------------------------------------------------------------------------
bool log_singletone::add_logger(ibase_log_stream* pstream, int log_level_limit/* = LOG_LEVEL_4*/)
{
logger* plogger = get_or_create_instance();
if(!plogger) return false;
return plogger->add_logger(pstream, log_level_limit);
}
//----------------------------------------------------------------------------
bool log_singletone::remove_logger(int type)
{
logger* plogger = get_or_create_instance();
if(!plogger) return false;
return plogger->remove_logger(type);
}
//----------------------------------------------------------------------------
PUSH_WARNINGS
DISABLE_GCC_WARNING(maybe-uninitialized)
int log_singletone::get_set_log_detalisation_level(bool is_need_set/* = false*/, int log_level_to_set/* = LOG_LEVEL_1*/)
{
static int log_detalisation_level = LOG_LEVEL_1;
if(is_need_set)
log_detalisation_level = log_level_to_set;
return log_detalisation_level;
}
POP_WARNINGS
//----------------------------------------------------------------------------
int log_singletone::get_set_time_level(bool is_need_set/* = false*/, int time_log_level/* = LOG_LEVEL_0*/)
{
static int val_time_log_level = LOG_LEVEL_0;
if(is_need_set)
val_time_log_level = time_log_level;
return val_time_log_level;
}
//----------------------------------------------------------------------------
int log_singletone::get_set_process_level(bool is_need_set/* = false*/, int process_log_level/* = LOG_LEVEL_0*/)
{
static int val_process_log_level = LOG_LEVEL_0;
if(is_need_set)
val_process_log_level = process_log_level;
return val_process_log_level;
}
//----------------------------------------------------------------------------
bool log_singletone::get_set_need_thread_id(bool is_need_set/* = false*/, bool is_need_val/* = false*/)
{
static bool is_need = false;
if(is_need_set)
is_need = is_need_val;
return is_need;
}
//----------------------------------------------------------------------------
bool log_singletone::get_set_need_proc_name(bool is_need_set/* = false*/, bool is_need_val/* = false*/)
{
static bool is_need = true;
if(is_need_set)
is_need = is_need_val;
return is_need;
}
//----------------------------------------------------------------------------
uint64_t log_singletone::get_set_err_count(bool is_need_set/* = false*/, uint64_t err_val/* = false*/)
{
static uint64_t err_count = 0;
if(is_need_set)
err_count = err_val;
return err_count;
}
//----------------------------------------------------------------------------
#ifdef _MSC_VER
void log_singletone::SetThreadName( DWORD dwThreadID, const char* threadName)
{
#define MS_VC_EXCEPTION 0x406D1388
#pragma pack(push,8)
typedef struct tagTHREADNAME_INFO
{
DWORD dwType; // Must be 0x1000.
LPCSTR szName; // Pointer to name (in user addr space).
DWORD dwThreadID; // Thread ID (-1=caller thread).
DWORD dwFlags; // Reserved for future use, must be zero.
} THREADNAME_INFO;
#pragma pack(pop)
Sleep(10);
THREADNAME_INFO info;
info.dwType = 0x1000;
info.szName = (char*)threadName;
info.dwThreadID = dwThreadID;
info.dwFlags = 0;
__try
{
RaiseException( MS_VC_EXCEPTION, 0, sizeof(info)/sizeof(ULONG_PTR), (ULONG_PTR*)&info );
}
__except(EXCEPTION_EXECUTE_HANDLER)
{
}
}
#endif
//----------------------------------------------------------------------------
bool log_singletone::set_thread_log_prefix(const std::string& prefix)
{
#ifdef _MSC_VER
SetThreadName(-1, prefix.c_str());
#endif
logger* plogger = get_or_create_instance();
if(!plogger) return false;
return plogger->set_thread_prefix(prefix);
}
//----------------------------------------------------------------------------
std::string log_singletone::get_prefix_entry()
{
std::stringstream str_prefix;
//write time entry
if ( get_set_time_level() <= get_set_log_detalisation_level() )
str_prefix << get_day_time_string() << " ";
//write process info
logger* plogger = get_or_create_instance();
//bool res = false;
if(!plogger)
{ //globally uninitialized, create new logger for each call of get_prefix_entry() and then delete it
plogger = new logger();
}
//if ( get_set_need_proc_name() && get_set_process_level() <= get_set_log_detalisation_level() )
// str_prefix << "[" << plogger->m_process_name << " (id=" << GetCurrentProcessId() << ")] ";
//#ifdef _MSC_VER_EX
if ( get_set_need_thread_id() /*&& get_set_tid_level() <= get_set_log_detalisation_level()*/ )
str_prefix << "tid:" << misc_utils::get_thread_string_id() << " ";
//#endif
if(plogger->m_thr_prefix_strings.size())
{
CRITICAL_REGION_LOCAL(plogger->m_critical_sec);
std::string thr_str = misc_utils::get_thread_string_id();
std::map<std::string, std::string>::iterator it = plogger->m_thr_prefix_strings.find(thr_str);
if(it!=plogger->m_thr_prefix_strings.end())
{
str_prefix << it->second;
}
}
if(get_set_is_uninitialized())
delete plogger;
return str_prefix.str();
}
//----------------------------------------------------------------------------
bool log_singletone::init()
{
return true;/*do nothing here*/
}
//----------------------------------------------------------------------------
bool log_singletone::un_init()
{
//delete object
logger* plogger = get_set_instance_internal();
if(plogger) delete plogger;
//set uninitialized
get_set_is_uninitialized(true, true);
get_set_instance_internal(true, NULL);
return true;
}
//----------------------------------------------------------------------------
logger* log_singletone::get_or_create_instance()
{
logger* plogger = get_set_instance_internal();
if(!plogger)
if(!get_set_is_uninitialized())
get_set_instance_internal(true, plogger = new logger);
return plogger;
}
//----------------------------------------------------------------------------
logger* log_singletone::get_set_instance_internal(bool is_need_set/* = false*/, logger* pnew_logger_val/* = NULL*/)
{
static logger* val_plogger = NULL;
if(is_need_set)
val_plogger = pnew_logger_val;
return val_plogger;
}
//----------------------------------------------------------------------------
bool log_singletone::get_set_is_uninitialized(bool is_need_set/* = false*/, bool is_uninitialized/* = false*/)
{
static bool val_is_uninitialized = false;
if(is_need_set)
val_is_uninitialized = is_uninitialized;
return val_is_uninitialized;
}
//----------------------------------------------------------------------------
log_frame::log_frame(const std::string& name, int dlevel/* = LOG_LEVEL_2*/, const char* plog_name/* = NULL*/)
{
#ifdef _MSC_VER
int lasterr=::GetLastError();
#endif
m_plog_name = plog_name;
if ( dlevel <= log_singletone::get_log_detalisation_level() )
{
m_name = name;
std::stringstream ss;
ss << log_space::log_singletone::get_prefix_entry() << "-->>" << m_name << std::endl;
log_singletone::do_log_message(ss.str(), dlevel, console_color_default, false, m_plog_name);
}
m_level = dlevel;
#ifdef _MSC_VER
::SetLastError(lasterr);
#endif
}
//----------------------------------------------------------------------------
log_frame::~log_frame()
{
#ifdef _MSC_VER
int lasterr=::GetLastError();
#endif
if (m_level <= log_singletone::get_log_detalisation_level() )
{
std::stringstream ss;
ss << log_space::log_singletone::get_prefix_entry() << "<<--" << m_name << std::endl;
log_singletone::do_log_message(ss.str(), m_level, console_color_default, false,m_plog_name);
}
#ifdef _MSC_VER
::SetLastError(lasterr);
#endif
}
//----------------------------------------------------------------------------
std::string get_win32_err_descr(int err_no)
{
#ifdef _MSC_VER
LPVOID lpMsgBuf;
FormatMessageA(
FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_SYSTEM,
NULL,
err_no,
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(char*) &lpMsgBuf,
0, NULL );
std::string fix_sys_message = "(null)";
if(lpMsgBuf) fix_sys_message = (char*)lpMsgBuf;
std::string::size_type a;
if ( (a = fix_sys_message.rfind( '\n' )) != std::string::npos )
fix_sys_message.erase(a);
if ( (a = fix_sys_message.rfind( '\r' )) != std::string::npos )
fix_sys_message.erase(a);
LocalFree(lpMsgBuf);
return fix_sys_message;
#else
return "Not implemented yet";
#endif
}
//----------------------------------------------------------------------------
bool getwin32_err_text(std::stringstream& ref_message, int error_no)
{
ref_message << "win32 error:" << get_win32_err_descr(error_no);
return true;
}
}
}
Reference in a new issue View git blame Copy permalink