wsstream_file.c File Reference

#include "wsint.h"

Go to the source code of this file.

Data Structures
struct	WsStreamFileCtxRec

Typedefs
typedef struct WsStreamFileCtxRec	WsStreamFileCtx

Functions
static size_t	file_input (void *context, WsUInt32 *buf, size_t buflen)
static size_t	file_output (void *context, WsUInt32 *buf, size_t buflen)
static WsBool	file_flush (void *context)
static void	file_close (void *context)
WsStream *	ws_stream_new_file (FILE *fp, WsBool output, WsBool close)

Typedef Documentation

WsStreamFileCtx

typedef struct WsStreamFileCtxRec WsStreamFileCtx

Definition at line 92 of file wsstream_file.c.

Function Documentation

file_close()

static void file_close ( void *  context )

static

Definition at line 194 of file wsstream_file.c.

References WsStreamFileCtxRec::close_fp, WsStreamFileCtxRec::fp, and ws_free().

Referenced by ws_stream_new_file().

{
    WsStreamFileCtx *ctx = (WsStreamFileCtx *) context;

    if (ctx->close_fp)
        fclose(ctx->fp);

    ws_free(ctx);
}

file_flush()

static WsBool file_flush ( void *  context )

static

Definition at line 171 of file wsstream_file.c.

References WsStreamFileCtxRec::buf, WsStreamFileCtxRec::data_in_buf, WsStreamFileCtxRec::fp, and WS_FALSE.

Referenced by ws_stream_new_file().

{
    WsStreamFileCtx *ctx = (WsStreamFileCtx *) context;

    /* If the internal buffer has any data, then this stream must be an
       output stream.  The variable `data_in_buf' is not updated on
       input streams. */
    if (ctx->data_in_buf) {
        if (fwrite(ctx->buf, 1, ctx->data_in_buf, ctx->fp) != ctx->data_in_buf) {
            /* The write failed. */
            ctx->data_in_buf = 0;
            return WS_FALSE;
        }

        /* The temporary buffer is not empty. */
        ctx->data_in_buf = 0;
    }

    /* Flush the underlying file stream. */
    return fflush(ctx->fp) == 0;
}

file_input()

static size_t file_input ( void *  context, WsUInt32 *  buf, size_t  buflen  )

static

Definition at line 96 of file wsstream_file.c.

References WsStreamFileCtxRec::buf, and WsStreamFileCtxRec::fp.

Referenced by ws_stream_new_file().

{
    WsStreamFileCtx *ctx = (WsStreamFileCtx *) context;
    size_t read = 0;

    while (buflen > 0) {
        size_t toread = buflen < sizeof(ctx->buf) ? buflen : sizeof(ctx->buf);
        size_t got, i;

        got = fread(ctx->buf, 1, toread, ctx->fp);

        /* Convert the data to the stream's IO buffer. */
        for (i = 0; i < got; i++)
            buf[i] = ctx->buf[i];

        buflen -= got;
        buf += got;
        read += got;

        if (got < toread)
            /* EOF seen. */
            break;
    }

    return read;
}

file_output()

static size_t file_output ( void *  context, WsUInt32 *  buf, size_t  buflen  )

static

Definition at line 124 of file wsstream_file.c.

References WsStreamFileCtxRec::buf, WsStreamFileCtxRec::data_in_buf, WsStreamFileCtxRec::fp, and WS_STREAM_BUFFER_SIZE.

Referenced by ws_stream_new_file().

{
    WsStreamFileCtx *ctx = (WsStreamFileCtx *) context;
    size_t wrote = 0;
    unsigned char ch;

    while (buflen) {
        /* Do we have any space in the stream's internal IO buffer? */
        if (ctx->data_in_buf >= WS_STREAM_BUFFER_SIZE) {
            size_t w;

            /* No, flush something to our file stream. */
            w = fwrite(ctx->buf, 1, ctx->data_in_buf, ctx->fp);
            if (w < ctx->data_in_buf) {
                /* Write failed.  As a result code we return the number
                          of characters written from our current write
                          request. */
                ctx->data_in_buf = 0;
                return wrote;
            }

            ctx->data_in_buf = 0;
        }
        /* Now we have space in the internal buffer. */

        /* Here we could perform some sort of conversions from ISO 10646
           to the output character set.  Currently we just support
           ISO-8859/1 and all unknown characters are replaced with
           '?'. */

        if (*buf > 0xff)
            ch = '?';
        else
            ch = (unsigned char) * buf;

        ctx->buf[ctx->data_in_buf++] = ch;

        /* Move forward. */
        buf++;
        buflen--;
        wrote++;
    }

    return wrote;
}

ws_stream_new_file()

WsStream* ws_stream_new_file	(	FILE *	fp,
		WsBool	output,
		WsBool	close
	)

Definition at line 206 of file wsstream_file.c.

References WsStreamFileCtxRec::close_fp, file_close(), file_flush(), file_input(), file_output(), WsStreamFileCtxRec::fp, ws_calloc(), and ws_stream_new().

Referenced by ws_compile_file().

{
    WsStreamFileCtx *ctx = ws_calloc(1, sizeof(*ctx));
    WsStream *stream;

    if (ctx == NULL)
        return NULL;

    ctx->fp = fp;
    ctx->close_fp = close;

    if (output)
        stream = ws_stream_new(ctx, file_output, file_flush, file_close);
    else
        stream = ws_stream_new(ctx, file_input, file_flush, file_close);

    if (stream == NULL)
        /* The stream creation failed.  Close the stream context. */
        file_close(ctx);

    return stream;
}