src/server/util/io.h

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2013 Olaf Wintermann. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *   1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *   2. 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 THE COPYRIGHT HOLDERS AND CONTRIBUTORS "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 THE COPYRIGHT HOLDER OR 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.
 */

#ifndef IOSTREAM_H
#define	IOSTREAM_H

#include "../public/nsapi.h"

#ifdef	__cplusplus
extern "C" {
#endif

typedef struct io_stream IOStream;

typedef ssize_t(*io_write_f)(IOStream *, void *, size_t);
typedef ssize_t(*io_writev_f)(IOStream *, struct iovec *, int);
typedef ssize_t(*io_read_f)(IOStream *, void *, size_t);
typedef ssize_t(*io_sendfile_f)(IOStream *, sendfiledata *);
typedef void(*io_close_f)(IOStream *);
typedef void(*io_finish_f)(IOStream *);

struct io_stream {
    io_write_f    write;
    io_writev_f   writev;
    io_read_f     read;
    io_sendfile_f sendfile;
    io_close_f    close;
    io_finish_f   finish;
};

typedef struct SystemIOStream {
    IOStream st;
    int      fd;
} SystemIOStream;

typedef struct NetIOStream {
    IOStream st;
    int      fd;
    size_t   max_read;
    size_t   read;
    int      chunked_enc;
    int      buffered;
} NetIOStream;


/* net_ functions */
ssize_t net_read(SYS_NETFD fd, void *buf, size_t nbytes);
ssize_t net_write(SYS_NETFD fd, void *buf, size_t nbytes);
ssize_t net_printf(SYS_NETFD fd, char *format, ...);
ssize_t net_sendfile(SYS_NETFD fd, sendfiledata *sfd);
void    net_close(SYS_NETFD fd);
void    net_finish(SYS_NETFD fd);

/* iovec buffer */
typedef struct iovec_buf{
    struct iovec   *iov;
    int            iovctn;
    int            maxiovec;
    pool_handle_t  *pool;
} iovec_buf_t;


/* system stream */
IOStream* stream_new_from_fd(pool_handle_t *pool, int fd);

ssize_t system_write(IOStream *st, void *buf, size_t nbytes);
ssize_t system_writev(IOStream *st, struct iovec *iovec, int iovcnt);
ssize_t system_read(IOStream *st, void *buf, size_t nbytes);

/* net stream */
IOStream* net_stream_from_fd(pool_handle_t *pool, int fd);

ssize_t net_stream_write(NetIOStream *st, void *buf, size_t nbytes);
ssize_t net_stream_writev(NetIOStream *st, struct iovec *iovec, int iovcnt);
ssize_t net_stream_read(NetIOStream *st, void *buf, size_t nbytes);
ssize_t net_stream_sendfile(NetIOStream *st, sendfiledata *sfd);
void    net_stream_close(NetIOStream *st);
void    net_stream_finish(NetIOStream *st);

/* iovec buffer */
iovec_buf_t *iovec_buf_create(pool_handle_t *pool);
void iovec_buf_write(iovec_buf_t *io, void *buf, size_t nbyte);
ssize_t iovec_buf_flush(iovec_buf_t *io, int fd);


#ifdef __SunOS_5_10

int vasprintf(char **str, const char *format, va_list ap);
int asprintf(char **str, const char *format, ...);

#endif

#ifdef	__cplusplus
}
#endif

#endif	/* IOSTREAM_H */