|
|
|
@ -51,15 +51,23 @@ typedef struct OutputFileLogger_ {
|
|
|
|
|
struct OutputFileLogger_ *next;
|
|
|
|
|
const char *name;
|
|
|
|
|
TmmId module_id;
|
|
|
|
|
ThreadInitFunc ThreadInit;
|
|
|
|
|
ThreadDeinitFunc ThreadDeinit;
|
|
|
|
|
ThreadExitPrintStatsFunc ThreadExitPrintStats;
|
|
|
|
|
} OutputFileLogger;
|
|
|
|
|
|
|
|
|
|
static OutputFileLogger *list = NULL;
|
|
|
|
|
|
|
|
|
|
int OutputRegisterFileLogger(const char *name, FileLogger LogFunc, OutputCtx *output_ctx)
|
|
|
|
|
int OutputRegisterFileLogger(const char *name, FileLogger LogFunc,
|
|
|
|
|
OutputCtx *output_ctx, ThreadInitFunc ThreadInit,
|
|
|
|
|
ThreadDeinitFunc ThreadDeinit,
|
|
|
|
|
ThreadExitPrintStatsFunc ThreadExitPrintStats)
|
|
|
|
|
{
|
|
|
|
|
#if 0
|
|
|
|
|
int module_id = TmModuleGetIdByName(name);
|
|
|
|
|
if (module_id < 0)
|
|
|
|
|
return -1;
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
OutputFileLogger *op = SCMalloc(sizeof(*op));
|
|
|
|
|
if (op == NULL)
|
|
|
|
@ -69,7 +77,12 @@ int OutputRegisterFileLogger(const char *name, FileLogger LogFunc, OutputCtx *ou
|
|
|
|
|
op->LogFunc = LogFunc;
|
|
|
|
|
op->output_ctx = output_ctx;
|
|
|
|
|
op->name = name;
|
|
|
|
|
#if 0
|
|
|
|
|
op->module_id = (TmmId) module_id;
|
|
|
|
|
#endif
|
|
|
|
|
op->ThreadInit = ThreadInit;
|
|
|
|
|
op->ThreadDeinit = ThreadDeinit;
|
|
|
|
|
op->ThreadExitPrintStats = ThreadExitPrintStats;
|
|
|
|
|
|
|
|
|
|
if (list == NULL)
|
|
|
|
|
list = op;
|
|
|
|
@ -190,16 +203,9 @@ static TmEcode OutputFileLogThreadInit(ThreadVars *tv, void *initdata, void **da
|
|
|
|
|
|
|
|
|
|
OutputFileLogger *logger = list;
|
|
|
|
|
while (logger) {
|
|
|
|
|
TmModule *tm_module = TmModuleGetByName((char *)logger->name);
|
|
|
|
|
if (tm_module == NULL) {
|
|
|
|
|
SCLogError(SC_ERR_INVALID_ARGUMENT,
|
|
|
|
|
"TmModuleGetByName for %s failed", logger->name);
|
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (tm_module->ThreadInit) {
|
|
|
|
|
if (logger->ThreadInit) {
|
|
|
|
|
void *retptr = NULL;
|
|
|
|
|
if (tm_module->ThreadInit(tv, (void *)logger->output_ctx, &retptr) == TM_ECODE_OK) {
|
|
|
|
|
if (logger->ThreadInit(tv, (void *)logger->output_ctx, &retptr) == TM_ECODE_OK) {
|
|
|
|
|
OutputLoggerThreadStore *ts = SCMalloc(sizeof(*ts));
|
|
|
|
|
/* todo */ BUG_ON(ts == NULL);
|
|
|
|
|
memset(ts, 0x00, sizeof(*ts));
|
|
|
|
@ -233,15 +239,8 @@ static TmEcode OutputFileLogThreadDeinit(ThreadVars *tv, void *thread_data)
|
|
|
|
|
OutputFileLogger *logger = list;
|
|
|
|
|
|
|
|
|
|
while (logger && store) {
|
|
|
|
|
TmModule *tm_module = TmModuleGetByName((char *)logger->name);
|
|
|
|
|
if (tm_module == NULL) {
|
|
|
|
|
SCLogError(SC_ERR_INVALID_ARGUMENT,
|
|
|
|
|
"TmModuleGetByName for %s failed", logger->name);
|
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (tm_module->ThreadDeinit) {
|
|
|
|
|
tm_module->ThreadDeinit(tv, store->thread_data);
|
|
|
|
|
if (logger->ThreadDeinit) {
|
|
|
|
|
logger->ThreadDeinit(tv, store->thread_data);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
OutputLoggerThreadStore *next_store = store->next;
|
|
|
|
@ -261,15 +260,8 @@ static void OutputFileLogExitPrintStats(ThreadVars *tv, void *thread_data)
|
|
|
|
|
OutputFileLogger *logger = list;
|
|
|
|
|
|
|
|
|
|
while (logger && store) {
|
|
|
|
|
TmModule *tm_module = TmModuleGetByName((char *)logger->name);
|
|
|
|
|
if (tm_module == NULL) {
|
|
|
|
|
SCLogError(SC_ERR_INVALID_ARGUMENT,
|
|
|
|
|
"TmModuleGetByName for %s failed", logger->name);
|
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (tm_module->ThreadExitPrintStats) {
|
|
|
|
|
tm_module->ThreadExitPrintStats(tv, store->thread_data);
|
|
|
|
|
if (logger->ThreadExitPrintStats) {
|
|
|
|
|
logger->ThreadExitPrintStats(tv, store->thread_data);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
logger = logger->next;
|
|
|
|
|