stream.h

/* FreeTDS - Library of routines accessing Sybase and Microsoft databases
 * Copyright (C) 2013  Frediano Ziglio
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#ifndef _tdsguard_a5iqP6Ed7kDGP9L1OvOL7W_
#define _tdsguard_a5iqP6Ed7kDGP9L1OvOL7W_

#ifndef _tdsguard_hfOrWb5znoUCWdBPoNQvqN_
#error Include tds.h first
#endif

#include <stdio.h>      /* FILE * */
#include <freetds/tds/iconv.h>  /* TDS_ICONV_DIRECTION */
#include <freetds/pushvis.h>

 /* Support 64-bit file seek if possible */
#ifdef HAVE_FSEEKO
typedef off_t offset_type;
#elif defined(_WIN32) || defined(_WIN64)
/* win32 version */
typedef __int64 offset_type;

#if defined(HAVE__FSEEKI64) && defined(HAVE__FTELLI64)
#define fseeko(f,o,w) _fseeki64((f),o,w)
#define ftello(f) _ftelli64((f))
#else
#define fseeko(f,o,w) (_lseeki64(fileno(f),o,w) == -1 ? -1 : 0)
#define ftello(f) _telli64(fileno(f))
#endif
#else
/* use old version */
#define fseeko(f,o,w) fseek(f,o,w)
#define ftello(f) ftell(f)
typedef long offset_type;
#endif


typedef struct tds_input_stream
{
    int (*read)(struct tds_input_stream *stream, void *ptr, size_t len);
} TDSINSTREAM;

typedef struct tds_output_stream {
    int (*write)(struct tds_output_stream *stream, size_t len);
    char *buffer;
    size_t buf_len;
} TDSOUTSTREAM;

typedef struct tds_fileout_stream
{
    TDSOUTSTREAM stream;
    FILE *fp;
    char block[4096];
    int buff_mode;
} TDSFILEOUTSTREAM;

void tds_fileout_stream_init(TDSFILEOUTSTREAM * s, FILE * fp, int buff_mode);

TDSRET tds_fileout_stream_flush(TDSFILEOUTSTREAM * s);
TDSRET tds_fileout_stream_put(TDSFILEOUTSTREAM * s, const void *src, size_t n);

TDSRET tds_convert_stream(TDSSOCKET * tds, TDSICONV * char_conv, TDS_ICONV_DIRECTION direction,
    TDSINSTREAM * istream, TDSOUTSTREAM *ostream);
TDSRET tds_copy_stream(TDSINSTREAM * istream, TDSOUTSTREAM * ostream);

/* Additional streams */

typedef struct tds_datain_stream {
    TDSINSTREAM stream;
    size_t wire_size;   
    TDSSOCKET *tds;
} TDSDATAINSTREAM;

void tds_datain_stream_init(TDSDATAINSTREAM * stream, TDSSOCKET * tds, size_t wire_size);

typedef struct tds_dataout_stream {
    TDSOUTSTREAM stream;
    TDSSOCKET *tds;
    size_t written;
} TDSDATAOUTSTREAM;

void tds_dataout_stream_init(TDSDATAOUTSTREAM * stream, TDSSOCKET * tds);

typedef struct tds_staticin_stream {
    TDSINSTREAM stream;
    const char *buffer;
    size_t buf_left;
} TDSSTATICINSTREAM;

void tds_staticin_stream_init(TDSSTATICINSTREAM * stream, const void *ptr, size_t len);

typedef struct tds_staticout_stream {
    TDSOUTSTREAM stream;
} TDSSTATICOUTSTREAM;

void tds_staticout_stream_init(TDSSTATICOUTSTREAM * stream, void *ptr, size_t len);

typedef struct tds_dynamic_stream {
    TDSOUTSTREAM stream;
    void **buf;
    size_t allocated;
    size_t size;
} TDSDYNAMICSTREAM;

TDSRET tds_dynamic_stream_init(TDSDYNAMICSTREAM * stream, void **ptr, size_t allocated);

/* input stream to read a file with option to terminate the read
 * when a delimiter sequence is read.
 *
 * This structure may read more byte from the file than are consumed by this
 * operation. This structure expects no other seek, read or write operations
 * are performed on the file pointer while it's active in this structure.
 */
enum
{ TDSFILESTREAM_BLOCKSIZE = 512 };

typedef struct tds_file_stream
{
    TDSINSTREAM stream;

    FILE *f;
    offset_type offset;

    char cbuf[50];
    size_t cpos;

    const char *terminator;
    size_t term_len;

    /* Read buffering */
    char inbuf[TDSFILESTREAM_BLOCKSIZE];
    size_t inpos;
    size_t inlen;

} TDSFILESTREAM;

TDSRET tds_file_stream_init(TDSFILESTREAM * stream, FILE * f);

/* Read raw data from stream (no terminator or iconv). Return number of bytes read. */
size_t tds_file_stream_read_raw(TDSFILESTREAM * stream, void *ptr, size_t n);
TDSRET tds_file_stream_seek_set(TDSFILESTREAM * stream, offset_type seek_to);
offset_type tds_file_stream_tell(TDSFILESTREAM * stream);
TDSRET tds_file_stream_close(TDSFILESTREAM * stream);

#include <freetds/popvis.h>

#endif