Browse Source

Move logger code to a separate module.

tags/0.9.0
Vsevolod Stakhov 9 years ago
parent
commit
5d376e9b1a
4 changed files with 148 additions and 120 deletions
  1. 1
    0
      src/lua/CMakeLists.txt
  2. 0
    120
      src/lua/lua_common.c
  3. 1
    0
      src/lua/lua_common.h
  4. 146
    0
      src/lua/lua_logger.c

+ 1
- 0
src/lua/CMakeLists.txt View File

@@ -1,5 +1,6 @@
# Lua support makefile
SET(LUASRC ${CMAKE_CURRENT_SOURCE_DIR}/lua_common.c
${CMAKE_CURRENT_SOURCE_DIR}/lua_logger.c
${CMAKE_CURRENT_SOURCE_DIR}/lua_task.c
${CMAKE_CURRENT_SOURCE_DIR}/lua_config.c
${CMAKE_CURRENT_SOURCE_DIR}/lua_classifier.c

+ 0
- 120
src/lua/lua_common.c View File

@@ -32,21 +32,6 @@ const luaL_reg null_reg[] = {
{NULL, NULL}
};

/* Logger methods */
LUA_FUNCTION_DEF (logger, err);
LUA_FUNCTION_DEF (logger, warn);
LUA_FUNCTION_DEF (logger, info);
LUA_FUNCTION_DEF (logger, debug);

static const struct luaL_reg loggerlib_f[] = {
LUA_INTERFACE_DEF (logger, err),
LUA_INTERFACE_DEF (logger, warn),
LUA_INTERFACE_DEF (logger, info),
LUA_INTERFACE_DEF (logger, debug),
{"__tostring", rspamd_lua_class_tostring},
{NULL, NULL}
};


/* Util functions */
/**
@@ -159,111 +144,6 @@ rspamd_lua_table_get (lua_State *L, const gchar *index)
return result;
}

static void
lua_common_log (GLogLevelFlags level, lua_State *L, const gchar *msg)
{
lua_Debug d;
gchar func_buf[128], *p;

if (lua_getstack (L, 1, &d) == 1) {
(void)lua_getinfo (L, "Sl", &d);
if ((p = strrchr (d.short_src, '/')) == NULL) {
p = d.short_src;
}
else {
p++;
}
rspamd_snprintf (func_buf, sizeof (func_buf), "%s:%d", p,
d.currentline);
if (level == G_LOG_LEVEL_DEBUG) {
rspamd_conditional_debug (rspamd_main->logger,
NULL,
func_buf,
"%s",
msg);
}
else {
rspamd_common_log_function (rspamd_main->logger,
level,
func_buf,
"%s",
msg);
}
}
else {
if (level == G_LOG_LEVEL_DEBUG) {
rspamd_conditional_debug (rspamd_main->logger,
NULL,
__FUNCTION__,
"%s",
msg);
}
else {
rspamd_common_log_function (rspamd_main->logger,
level,
__FUNCTION__,
"%s",
msg);
}
}
}

/*** Logger interface ***/
static gint
lua_logger_err (lua_State * L)
{
const gchar *msg;
msg = luaL_checkstring (L, 1);
lua_common_log (G_LOG_LEVEL_CRITICAL, L, msg);
return 1;
}

static gint
lua_logger_warn (lua_State * L)
{
const gchar *msg;
msg = luaL_checkstring (L, 1);
lua_common_log (G_LOG_LEVEL_WARNING, L, msg);
return 1;
}

static gint
lua_logger_info (lua_State * L)
{
const gchar *msg;
msg = luaL_checkstring (L, 1);
lua_common_log (G_LOG_LEVEL_INFO, L, msg);
return 1;
}

static gint
lua_logger_debug (lua_State * L)
{
const gchar *msg;
msg = luaL_checkstring (L, 1);
lua_common_log (G_LOG_LEVEL_DEBUG, L, msg);
return 1;
}


/*** Init functions ***/

static gint
lua_load_logger (lua_State *L)
{
lua_newtable (L);
luaL_register (L, NULL, loggerlib_f);

return 1;
}

static void
luaopen_logger (lua_State * L)
{
rspamd_lua_add_preload (L, "rspamd_logger", lua_load_logger);
}


static void
lua_add_actions_global (lua_State *L)
{

+ 1
- 0
src/lua/lua_common.h View File

@@ -188,6 +188,7 @@ void luaopen_dns_resolver (lua_State * L);
void luaopen_rsa (lua_State * L);
void luaopen_ip (lua_State * L);
void luaopen_expression (lua_State * L);
void luaopen_logger (lua_State * L);

gint rspamd_lua_call_filter (const gchar *function, struct rspamd_task *task);
gint rspamd_lua_call_chain_filter (const gchar *function,

+ 146
- 0
src/lua/lua_logger.c View File

@@ -0,0 +1,146 @@
/*
* Copyright (c) 2015, Vsevolod Stakhov
*
* 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.
*
* THIS SOFTWARE IS PROVIDED BY AUTHOR ''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 AUTHOR 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 "lua_common.h"


/* Logger methods */
LUA_FUNCTION_DEF (logger, err);
LUA_FUNCTION_DEF (logger, warn);
LUA_FUNCTION_DEF (logger, info);
LUA_FUNCTION_DEF (logger, debug);

static const struct luaL_reg loggerlib_f[] = {
LUA_INTERFACE_DEF (logger, err),
LUA_INTERFACE_DEF (logger, warn),
LUA_INTERFACE_DEF (logger, info),
LUA_INTERFACE_DEF (logger, debug),
{"__tostring", rspamd_lua_class_tostring},
{NULL, NULL}
};

static void
lua_common_log (GLogLevelFlags level, lua_State *L, const gchar *msg)
{
lua_Debug d;
gchar func_buf[128], *p;

if (lua_getstack (L, 1, &d) == 1) {
(void)lua_getinfo (L, "Sl", &d);
if ((p = strrchr (d.short_src, '/')) == NULL) {
p = d.short_src;
}
else {
p++;
}
rspamd_snprintf (func_buf, sizeof (func_buf), "%s:%d", p,
d.currentline);
if (level == G_LOG_LEVEL_DEBUG) {
rspamd_conditional_debug (rspamd_main->logger,
NULL,
func_buf,
"%s",
msg);
}
else {
rspamd_common_log_function (rspamd_main->logger,
level,
func_buf,
"%s",
msg);
}
}
else {
if (level == G_LOG_LEVEL_DEBUG) {
rspamd_conditional_debug (rspamd_main->logger,
NULL,
__FUNCTION__,
"%s",
msg);
}
else {
rspamd_common_log_function (rspamd_main->logger,
level,
__FUNCTION__,
"%s",
msg);
}
}
}

/*** Logger interface ***/
static gint
lua_logger_err (lua_State * L)
{
const gchar *msg;
msg = luaL_checkstring (L, 1);
lua_common_log (G_LOG_LEVEL_CRITICAL, L, msg);
return 1;
}

static gint
lua_logger_warn (lua_State * L)
{
const gchar *msg;
msg = luaL_checkstring (L, 1);
lua_common_log (G_LOG_LEVEL_WARNING, L, msg);
return 1;
}

static gint
lua_logger_info (lua_State * L)
{
const gchar *msg;
msg = luaL_checkstring (L, 1);
lua_common_log (G_LOG_LEVEL_INFO, L, msg);
return 1;
}

static gint
lua_logger_debug (lua_State * L)
{
const gchar *msg;
msg = luaL_checkstring (L, 1);
lua_common_log (G_LOG_LEVEL_DEBUG, L, msg);
return 1;
}


/*** Init functions ***/

static gint
lua_load_logger (lua_State *L)
{
lua_newtable (L);
luaL_register (L, NULL, loggerlib_f);

return 1;
}

void
luaopen_logger (lua_State * L)
{
rspamd_lua_add_preload (L, "rspamd_logger", lua_load_logger);
}

Loading…
Cancel
Save