modules_s/xlog/xlog.c

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <unistd.h>
#include <fcntl.h>
#include <time.h>

#include "../../sr_module.h"
#include "../../dprint.h"
#include "../../error.h"
#include "../../mem/mem.h"

#include "xl_lib.h"

#define NO_SCRIPT -1

MODULE_VERSION

char *log_buf = NULL;

int buf_size=4096;

static int mod_init(void);
static int child_init(int);

static int xlog(struct sip_msg*, char*, char*);
static int xdbg(struct sip_msg*, char*, char*);

static int xlog_fixup(void** param, int param_no);
static int xdbg_fixup(void** param, int param_no);

static void destroy(void);

static cmd_export_t cmds[]={
        {"xlog",  xlog,  2, xlog_fixup, REQUEST_ROUTE | FAILURE_ROUTE |
                 ONREPLY_ROUTE | BRANCH_ROUTE},
        {"xdbg",  xdbg,  1, xdbg_fixup, REQUEST_ROUTE | FAILURE_ROUTE |
                ONREPLY_ROUTE | BRANCH_ROUTE},
        {"xbind",       (cmd_function)xl_bind, NO_SCRIPT, 0, 0},
        {"xprint",      (cmd_function)xl_print_log, NO_SCRIPT, 0, 0},
        {"xparse",      (cmd_function)xl_parse_format, NO_SCRIPT, 0, 0},
        {"shm_xparse",  (cmd_function)xl_shm_parse_format, NO_SCRIPT, 0, 0},
        {"xparse2",     (cmd_function)xl_parse_format2, NO_SCRIPT, 0, 0},
        {"shm_xparse2", (cmd_function)xl_shm_parse_format2, NO_SCRIPT, 0, 0},
        {"xfree",       (cmd_function)xl_elog_free_all, NO_SCRIPT, 0, 0},
        {"shm_xfree",   (cmd_function)xl_elog_shm_free_all, NO_SCRIPT, 0, 0},
        {"xnulstr",     (cmd_function)xl_get_nulstr, NO_SCRIPT, 0, 0},
        {0,0,0,0,0}
};


static param_export_t params[]={
        {"buf_size",  PARAM_INT, &buf_size},
        {0,0,0}
};


struct module_exports exports= {
        "xlog",
        cmds,
        0,        /* RPC methods */
        params,

        mod_init,   /* module initialization function */
        (response_function) 0,
        (destroy_function) destroy,
        0,
        child_init  /* per-child init function */
};

static int mod_init(void)
{
        DBG("XLOG: initializing ...\n");
        log_buf = (char*)pkg_malloc((buf_size+1)*sizeof(char));
        if(log_buf==NULL)
        {
                LOG(L_ERR, "XLOG:mod_init: ERROR: no more memory\n");
                return -1;
        }

        return xl_mod_init();
}

static int child_init(int rank)
{
        DBG("XLOG: init_child [%d]  pid [%d]\n", rank, getpid());
        return xl_child_init(rank);
}

static int xlog(struct sip_msg* msg, char* lev, char* frm)
{
        int log_len, level;

        if (get_int_fparam(&level, msg, (fparam_t *)lev)) {
                LOG(L_ERR, "XLOG:xlog: cannot get log level\n");
                return -1;
        }
        if (level < L_ALERT)
                level = L_ALERT;
        else if (level > L_DBG)
                level = L_DBG;

        log_len = buf_size;

        if(xl_print_log(msg, (xl_elog_t*)frm, log_buf, &log_len)<0)
                return -1;

        /* log_buf[log_len] = '\0'; */
        LOG_(DEFAULT_FACILITY, level, "<script>: ", "%.*s", log_len, log_buf);

        return 1;
}

static int xdbg(struct sip_msg* msg, char* frm, char* str2)
{
        int log_len;

        log_len = buf_size;

        if(xl_print_log(msg, (xl_elog_t*)frm, log_buf, &log_len)<0)
                return -1;

        /* log_buf[log_len] = '\0'; */
        LOG_(DEFAULT_FACILITY, L_DBG, "<script>: ", "%.*s", log_len, log_buf);

        return 1;
}

static void destroy(void)
{
        DBG("XLOG: destroy module ...\n");
        if(log_buf)
                pkg_free(log_buf);
}

static int xlog_fixup(void** param, int param_no)
{
        int level;
        fparam_t        *p;

        if(param_no==1)
        {
                if (*param == NULL) {
                        LOG(L_ERR, "XLOG:xlog_fixup: NULL parameter\n");
                        return -1;
                }

                if ((((char*)(*param))[0] == '$')
                        || (((char*)(*param))[0] == '@')
                ) {
                        /* avp or select parameter */
                        return fixup_var_int_1(param, 1);
                }

                if(strlen((char*)(*param))<3)
                {
                        LOG(L_ERR, "XLOG:xlog_fixup: wrong log level\n");
                        return E_UNSPEC;
                }
                switch(((char*)(*param))[2])
                {
                case 'A': level = L_ALERT; break;
                case 'C': level = L_CRIT; break;
                case 'E': level = L_ERR; break;
                case 'W': level = L_WARN; break;
                case 'N': level = L_NOTICE; break;
                case 'I': level = L_INFO; break;
                case 'D': level = L_DBG; break;
                default:
                        LOG(L_ERR, "XLOG:xlog_fixup: unknown log level\n");
                        return E_UNSPEC;
                }
                /* constant parameter, but the fparam structure
                 * needs to be created */
                p = (fparam_t*)pkg_malloc(sizeof(fparam_t));
                if (!p) {
                        LOG(L_ERR, "XLOG:xlog_fixup: not enough memory\n");
                        return -1;
                }
                p->v.i = level;
                p->type = FPARAM_INT;
                p->orig = *param;

                *param = p;
                return 0;
        }

        if(param_no==2)
                return xdbg_fixup(param, 1);

        return 0;
}

static int xdbg_fixup(void** param, int param_no)
{
        xl_elog_t *model;

        if(param_no==1)
        {
                if(*param)
                {
                        if(xl_parse_format((char*)(*param), &model)<0)
                        {
                                LOG(L_ERR, "XLOG:xdbg_fixup: ERROR: wrong format[%s]\n",
                                        (char*)(*param));
                                return E_UNSPEC;
                        }

                        *param = (void*)model;
                        return 0;
                }
                else
                {
                        LOG(L_ERR, "XLOG:xdbg_fixup: ERROR: null format\n");
                        return E_UNSPEC;
                }
        }

        return 0;

}