182 lines
4.2 KiB
C++
182 lines
4.2 KiB
C++
#include <dlfcn.h> // for dladdr
|
|
#include <cxxabi.h> // for __cxa_demangle
|
|
#include <cstdarg>
|
|
|
|
#include <string>
|
|
#include <iostream>
|
|
|
|
#include <log.h>
|
|
#include <pthread.h>
|
|
|
|
#if defined(__ANDROID__)
|
|
#include <android/log.h>
|
|
#endif
|
|
|
|
static int enabled = 0;
|
|
|
|
using namespace std;
|
|
extern "C" {
|
|
|
|
#define COLORIZE "\033[0;32m"
|
|
#define COLOR_RESET "\033[0m"
|
|
|
|
extern void *blacklist[];
|
|
#define skip_blacklisted(f) for (int z=0; blacklist[z]!=NULL; z++) if (blacklist[z]==f) return;
|
|
|
|
static bool lock_needed =
|
|
#ifndef __ANDROID__
|
|
true;
|
|
#else
|
|
false;
|
|
#endif
|
|
|
|
static pthread_mutex_t lock;
|
|
|
|
// see https://github.com/littlstar/asprintf.c/blob/master/asprintf.c
|
|
static int
|
|
_vasprintf_ (char **str, const char *fmt, va_list args) {
|
|
char *dest = NULL;
|
|
int size = 0;
|
|
va_list tmpa;
|
|
|
|
va_copy(tmpa, args);
|
|
size = vsnprintf(dest, size, fmt, tmpa);
|
|
va_end(tmpa);
|
|
|
|
if (size < 0) { return -1; }
|
|
*str = (char *) malloc(size + 1);
|
|
|
|
if (NULL == *str) { return -1; }
|
|
|
|
size = vsprintf(*str, fmt, args);
|
|
return size;
|
|
}
|
|
|
|
static int
|
|
_asprintf_ (char **str, const char *fmt, ...) {
|
|
int size = 0;
|
|
va_list args;
|
|
|
|
va_start(args, fmt);
|
|
size = _vasprintf_(str, fmt, args);
|
|
va_end(args);
|
|
|
|
return size;
|
|
}
|
|
|
|
void defaultLogFn(int priority, const char *format, va_list ap) {
|
|
char *buffer;
|
|
char *prio = NULL;
|
|
switch (priority) {
|
|
case LOG_VERBOSE: prio = (char*)"V: "; break;
|
|
case LOG_DEBUG: prio = (char*)"D: "; break;
|
|
case LOG_INFO: prio = (char*)"I: "; break;
|
|
case LOG_WARN: prio = (char*)"W: "; break;
|
|
case LOG_ERROR: prio = (char*)"E: "; break;
|
|
case LOG_FATAL: prio = (char*)"F: "; break;
|
|
case LOG_PROFILER: prio = (char*)"P: "; break;
|
|
default: prio = (char*)"D: ";
|
|
}
|
|
_asprintf_(&buffer, "%s%s\n", prio, format);
|
|
vprintf(buffer, ap);
|
|
free(buffer);
|
|
if (priority == LOG_FATAL) exit(1);
|
|
}
|
|
|
|
#if defined(__ANDROID__)
|
|
void androidLogFn(int prio, const char *format, va_list ap) {
|
|
__android_log_vprint(prio, "LorieNative", format, ap);
|
|
if (prio == LOG_FATAL) exit(1);
|
|
}
|
|
#endif
|
|
|
|
static void (*logFunction)(int, const char*, va_list) =
|
|
#if !defined(__ANDROID__)
|
|
defaultLogFn;
|
|
#else
|
|
androidLogFn;
|
|
#endif
|
|
|
|
static void LogMessageInternal(int priority, const char *format, ...) {
|
|
if (logFunction == NULL) return;
|
|
|
|
va_list argptr;
|
|
va_start(argptr, format);
|
|
logFunction(priority, format, argptr);
|
|
va_end(argptr);
|
|
}
|
|
|
|
void LogMessage(int priority, const char *format, ...) {
|
|
if (logFunction == NULL) return;
|
|
|
|
if (lock_needed) pthread_mutex_lock(&::lock);
|
|
va_list argptr;
|
|
va_start(argptr, format);
|
|
logFunction(priority, format, argptr);
|
|
va_end(argptr);
|
|
if (lock_needed) pthread_mutex_unlock(&::lock);
|
|
}
|
|
|
|
void LogInit(void) {
|
|
if (lock_needed && pthread_mutex_init(&::lock, NULL) != 0) {
|
|
LogMessageInternal(LOG_ERROR, "Logger mutex init failed\n");
|
|
return;
|
|
}
|
|
|
|
LOGD("Logging initialized");
|
|
}
|
|
|
|
static int level = -1;
|
|
|
|
void print_func(void *func, int enter);
|
|
|
|
void __attribute__((no_instrument_function)) __cyg_profile_func_enter (void *this_fn, void *call_site);
|
|
void __attribute__((no_instrument_function)) __cyg_profile_func_exit (void *this_fn, void *call_site);
|
|
|
|
void __attribute__((no_instrument_function))
|
|
__cyg_profile_func_enter (void *func, void *caller) {
|
|
if (!enabled) return;
|
|
skip_blacklisted(func);
|
|
level++;
|
|
print_func(func, 1);
|
|
}
|
|
|
|
void __attribute__((no_instrument_function))
|
|
__cyg_profile_func_exit (void *func, void *caller) {
|
|
if (!enabled) return;
|
|
skip_blacklisted(func);
|
|
print_func(func, 0);
|
|
level--;
|
|
}
|
|
|
|
void print_func(void *func, int enter) {
|
|
for (int i=0; i<level; i++) printf(" ");
|
|
Dl_info info;
|
|
if (dladdr(func, &info) && info.dli_sname != NULL) {
|
|
char *demangled = NULL;
|
|
int status;
|
|
demangled = abi::__cxa_demangle(info.dli_sname, NULL, 0, &status);
|
|
LOGP("%s%s %s%s", (level==2)?COLORIZE:"", enter ? ">" : "<", status == 0 ? demangled : info.dli_sname, (level==2)?COLOR_RESET:"");
|
|
free(demangled);
|
|
} else {
|
|
LOGP("%s%p%s", (level==2)?COLORIZE:"", func, (level==2)?COLOR_RESET:"");
|
|
}
|
|
}
|
|
|
|
void *blacklist[] = {
|
|
(void*) _vasprintf_,
|
|
(void*) _asprintf_,
|
|
(void*) defaultLogFn,
|
|
#if defined(__ANDROID__)
|
|
(void*) androidLogFn,
|
|
#endif
|
|
(void*) LogInit,
|
|
(void*) LogMessage,
|
|
(void*) LogMessageInternal,
|
|
(void*) __cyg_profile_func_enter,
|
|
(void*) __cyg_profile_func_exit,
|
|
(void*) print_func
|
|
};
|
|
|
|
} // extern "C"
|