/*
**  L2 - OSSP Logging Library
**  Copyright (c) 2001 The OSSP Project (http://www.ossp.org/)
**  Copyright (c) 2001 Cable & Wireless Deutschland (http://www.cw.com/de/)
**
**  This file is part of OSSP L2, a flexible logging library which
**  can be found at http://www.ossp.org/pkg/l2/.
**
**  Permission to use, copy, modify, and distribute this software for
**  any purpose with or without fee is hereby granted, provided that
**  the above copyright notice and this permission notice appear in all
**  copies.
**
**  THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED 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 THE AUTHORS AND COPYRIGHT HOLDERS AND THEIR
**  CONTRIBUTORS 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.
**
**  l2_ch_buffer.c: buffering channel implementation
*/

#include "l2.h"
#include "l2_p.h"

#include <string.h>

typedef struct {
    char  *buf;
    int    bufpos;
    int    bufsize;
} l2_ch_buffer_t;

static int hook_create(l2_context_t *ctx)
{
    l2_ch_buffer_t *cfg;

    if ((cfg = (l2_ch_buffer_t *)malloc(sizeof(l2_ch_buffer_t))) == NULL)
        return L2_ERROR;
    cfg->buf     = NULL;
    cfg->bufpos  = 0;
    cfg->bufsize = 4096;
    ctx->vp = cfg;
    return L2_OK;
}

static int hook_configure(l2_context_t *ctx, const char *fmt, va_list ap)
{
    l2_ch_buffer_t *cfg;
    l2_param_t pa[3]; 
    l2_result_t rv;

    if ((cfg = (l2_ch_buffer_t *)ctx->vp) == NULL)
        return L2_ERROR;
    L2_PARAM_SET(pa[0], size, INT, &cfg->bufsize);
    L2_PARAM_END(pa[2]);
    rv = l2_channel_setparams(pa, fmt, ap);
    if (cfg->bufsize < 0) 
        return L2_ERROR;
    return rv;
}

static int hook_open(l2_context_t *ctx, l2_channel_t *downstream)
{
    l2_ch_buffer_t *cfg;

    if ((cfg = (l2_ch_buffer_t *)ctx->vp) == NULL)
        return L2_ERROR;
    if (cfg->bufsize > 0) {
        if ((cfg->buf = malloc(cfg->bufsize)) == NULL)
            return L2_ERROR;
        cfg->bufpos = 0;
    }
    if (downstream != NULL)
        if (l2_channel_open(downstream) == L2_ERROR)
            return L2_ERROR;
    return L2_OK;
}

static int hook_write(l2_context_t *ctx, l2_channel_t *downstream, 
                      const char *buf, size_t bufsize)
{
    l2_ch_buffer_t *cfg;

    if ((cfg = (l2_ch_buffer_t *)ctx->vp) == NULL)
        return L2_ERROR;

    if (bufsize > (cfg->bufsize - cfg->bufpos))
        return L2_ERROR; /* The buffer is too small */

    memcpy(cfg->buf+cfg->bufpos, buf, bufsize);
    cfg->bufpos += bufsize;

    return L2_OK;
}

static int hook_flush(l2_context_t *ctx, l2_channel_t *downstream)
{
    l2_ch_buffer_t *cfg;

    if ((cfg = (l2_ch_buffer_t *)ctx->vp) == NULL)
        return L2_ERROR;

    if (cfg->bufpos > 0) {
        if (downstream != NULL)
            if (l2_channel_write(downstream, cfg->buf, cfg->bufpos) == L2_ERROR)
                return L2_ERROR;
        cfg->bufpos = 0;
        bzero(cfg->buf, cfg->bufsize); /* Guard against improper usage */
    }

    if (downstream != NULL)
        if (l2_channel_flush(downstream) == L2_ERROR)
            return L2_ERROR;
    return L2_OK;
}

static int hook_close(l2_context_t *ctx, l2_channel_t *downstream)
{
    l2_ch_buffer_t *cfg;

    if (downstream != NULL)
        if (l2_channel_close(downstream) == L2_ERROR)
            return L2_ERROR;
    if ((cfg = (l2_ch_buffer_t *)ctx->vp) == NULL)
        return L2_ERROR;
    if (cfg->buf != NULL) {
        free(cfg->buf);
        cfg->buf = NULL;
    }
    return L2_OK;
}

static int hook_destroy(l2_context_t *ctx)
{
    l2_ch_buffer_t *cfg;

    if ((cfg = (l2_ch_buffer_t *)ctx->vp) == NULL)
        return L2_ERROR;
    if (cfg->buf != NULL) 
        free(cfg->buf);
    free(cfg);
    return L2_OK;
}

l2_handler_t l2_handler_buffer = {
    hook_create,
    hook_configure,
    hook_open,
    hook_write,
    hook_flush,
    hook_close,
    hook_destroy
};