output.c

Go to the documentation of this file.

#include "config.h"
#include "output.h"

#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <string.h>

#include "len.h"

#define CHECK_MODE(i, c) (((i) & (c)) == (c))
#define APPEND_MODE(v, m, c) if(CHECK_MODE(m, c)) \
                v |= c;

#define CASE(s, v) case LOG_##s: \
      v = strdup(LOG_##s##_NAME); \
      break;

static short _output_lvl, _output_mode;
static FILE *_output_file_stream;

char *levelname(int lvl);

int set_output_level(int lvl)
{
        if(lvl == _OUTPUT_GET_CURRENT_VALUE)
                return _output_lvl;
        if(lvl >= LOG_EMERG && lvl <= LOG_DEBUG)
                _output_lvl = lvl;
        return _output_lvl;
}

int set_output_mode(int mode)
{
        if(mode == _OUTPUT_GET_CURRENT_VALUE)
                return _output_mode;

        _output_mode = 0;
        APPEND_MODE(_output_mode, mode, _OUTPUT_MODE_PRINT)
        APPEND_MODE(_output_mode, mode, _OUTPUT_MODE_SYSLOG)
        APPEND_MODE(_output_mode, mode, _OUTPUT_MODE_FILE)
        return _output_mode;
}

int set_output_file(FILE *out)
{
        _output_file_stream = out;
        return 0;
}

char *levelname(int lvl)
{
        char *ret = NULL;
        switch(lvl)
        {
                CASE(EMERG, ret)
                CASE(ALERT, ret)
                CASE(CRIT, ret)
                CASE(ERR, ret)
                CASE(WARNING, ret)
                CASE(NOTICE, ret)
                CASE(INFO, ret)
                CASE(DEBUG, ret)
                default: break;
        }
        return ret;
}

void output1(int lvl, const char *msg, ...)
{
        if(lvl <= _output_lvl)
        {
                va_list ap;
                va_start(ap, msg);
                char *msg1 = NULL;
                char *buf = NULL;
                int buf_len;
                if(lvl <= LOG_WARNING || lvl == LOG_DEBUG)
                {
                        char *lvlname = levelname(lvl);
                        asprintf(&msg1, "%s: %s", lvlname, msg);
                        free(lvlname);
                        lvlname = NULL;
                        buf_len = vasprintf(&buf, msg1, ap);
                }
                else
                        buf_len = vasprintf(&buf, msg, ap);

                va_end(ap);

                if(CHECK_MODE(_output_mode , _OUTPUT_MODE_SYSLOG))
                        syslog(LOG_DAEMON | lvl, "%s", buf);

                if(CHECK_MODE(_output_mode , _OUTPUT_MODE_PRINT))
                        puts(buf);

                if(CHECK_MODE(_output_mode , _OUTPUT_MODE_FILE))
                {
                        buf[buf_len] = '\n';
                        fwrite(buf, buf_len + 1, 1, _output_file_stream);
                        fflush(_output_file_stream);
                }

                free(buf);
                buf = NULL;
                free(msg1);
                msg1 = NULL;
        }
        return;
}