bts.c

libbt-1.04

#include "config.h"

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#ifdef HAVE_STRINGS_H
#include <strings.h>
#endif
#include <stdarg.h>
#include <assert.h>
#include "bts.h"
#include "util.h"
#if WIN32
#	define vsnprintf _vsnprintf
#endif

/*
 * btStream subclasses are used to de/serialize the btObject hierarchy
 * from/into bencoded streams of bytes.
 */
typedef struct btFileStream {
#define BTFILESTREAM 1
    struct btStream bts;
    FILE *fp;
} btFileStream;

typedef struct btStrStream {
#define BTSTRSTREAM 2
    struct btStream bts;
    int cur;
    unsigned char *buf;
    int len;
    int max;
} btStrStream;

int bts_strstream_read(btStream*bts, char *buf, int len) {
    btStrStream *ss = (btStrStream *)bts;
    if (ss->bts.type != BTSTRSTREAM) return -1;
    if (ss->cur+len > ss->len) {
        return 1;
    }
    memcpy(buf, ss->buf + ss->cur, len);
    ss->cur += len;
    return 0;
}

int bts_strstream_peek(btStream*bts) {
    btStrStream *ss = (btStrStream *)bts;
    if (ss->bts.type != BTSTRSTREAM) return -1;
    if (ss->cur >= ss->len) {
        return -1;
    }
    return ss->buf[ss->cur];
}

int bts_strstream_write(btStream*bts, char *buf, int len) {
    btStrStream *ss = (btStrStream *)bts;
    if (ss->bts.type != BTSTRSTREAM) return -1;
    if (ss->len + len > ss->max) {
        /* resize buffer */
        ss->max = ss->len + len + 100;
        ss->buf = btrealloc(ss->buf, ss->max);
    }
    memcpy(ss->buf + ss->len, buf, len);
    ss->len += len;
    return 0;
}

void bts_strstream_destroy( btStream *bts) {
    btStrStream *ss = (btStrStream *)bts;
    if (!ss) return;
    if (ss->bts.type != BTSTRSTREAM) return;
    if (ss->buf) {
        btfree(ss->buf);
	ss->buf=NULL;
	ss->len=0;
	ss->max=0;
    }
    btfree( ss);
}

int bts_strstream_rewind( btStream *bts, btsIo iodir) {
    btStrStream *ss = (btStrStream *)bts;
    if (ss->bts.type != BTSTRSTREAM) return 1;
    ss->bts.iodir = iodir;
    if (iodir == BTS_INPUT) {
	ss->cur = 0;
    } else if (iodir == BTS_OUTPUT) {
	ss->len = 0;
    } else {
	return 1;
    }
    return 0;
}

btStream* bts_create_strstream( btsIo iodir) {
    btStrStream *bts = btcalloc( 1, sizeof(btStrStream));
    bts->bts.type = BTSTRSTREAM;
    bts->bts.iodir = iodir;
    bts->bts.read = bts_strstream_read;
    bts->bts.write = bts_strstream_write;
    bts->bts.peek = bts_strstream_peek;
    bts->bts.destroy = bts_strstream_destroy;
    bts->bts.rewind = bts_strstream_rewind;
    return &bts->bts;
}

int bts_filestream_read(btStream*bts, char *buf, int len) {
    btFileStream *fs = (btFileStream *)bts;
    int res;
    if (fs->bts.type != BTFILESTREAM) return -1;
    if (fs->fp == NULL) return -1;
    res = fread( buf, 1, len, fs->fp);
    if (res < len) return -1;
    return 0;
}

int bts_filestream_write(btStream*bts, char *buf, int len) {
    btFileStream *fs = (btFileStream *)bts;
    int res;
    if (fs->bts.type != BTFILESTREAM) return -1;
    if (fs->fp == NULL) return -1;
    res = fwrite( buf, 1, len, fs->fp);
    if (res < len) return -1;
    return 0;
}

size_t writebts( void *buf, size_t size, size_t nmemb, void* stream ) {
    btStream *bts=stream;
    int len=size*nmemb;
    if (bts_write(bts, buf, len) != 0) {
	return 0;
    }
    return len;
}

int bts_filestream_peek(btStream*bts) {
    btFileStream *fs = (btFileStream *)bts;
    int c;
    int err;
    if (fs->bts.type != BTFILESTREAM) return -1;
    if (fs->fp == NULL) return -1;
    c = getc( fs->fp);
    if (c == EOF) return -1;
    err = ungetc( c, fs->fp);
    if (err == EOF) return -1;
    return c;
}

void bts_filestream_destroy( btStream *bts) {
    btFileStream *fs = (btFileStream *)bts;
    if (!fs) return;
    if (fs->bts.type != BTFILESTREAM) return;
    if (fs->fp == NULL) return;
    fclose(fs->fp);
    fs->fp=NULL;
    btfree( fs);
}

int bts_filestream_rewind( btStream *bts, btsIo iodir) {
    btFileStream *fs = (btFileStream *)bts;
    if (fs->bts.type != BTFILESTREAM) return 1;
    printf("unimplemented bts_filestream_rewind()\n");
    return 1;
}

btStream* bts_create_filestream_fp( FILE *fp, btsIo iodir) {
    btFileStream *bts = btcalloc( 1, sizeof(btFileStream));
    bts->bts.type = BTFILESTREAM;

#if 0
    if (fp->flags & _IO_NO_READS) {
        iodir&=~BTS_OUTPUT;
    }
    if (fp->flags & _IO_NO_WRITES) { 
        iodir&=~BTS_INPUT;
    }
#endif 

    bts->bts.iodir = iodir;
    bts->bts.read = bts_filestream_read;
    bts->bts.write = bts_filestream_write;
    bts->bts.peek = bts_filestream_peek;
    bts->bts.destroy = bts_filestream_destroy;
    bts->bts.rewind = bts_filestream_rewind;
    bts->fp = fp;
    return &bts->bts;
}


btStream* bts_create_filestream( char *fname, btsIo iodir) {
    FILE *fp;
  
    if (iodir == BTS_INPUT) {
        fp = fopen( fname, "r");
    } else if (iodir == BTS_OUTPUT) {
        fp = fopen( fname, "w");
    } else {
        fp = NULL;
    }
    return bts_create_filestream_fp( fp, iodir);
}

void bts_destroy( btStream *bts) {
    bts->destroy(bts);
}

int bts_rewind( btStream* bts, btsIo iodir) {
    return bts->rewind(bts, iodir);
}


struct btstrbuf bts_get_buf( btStream *bts) {
    struct btstrbuf buf = { NULL, 0 };
    btStrStream *bss = (btStrStream*)bts;
    if (bts->type != BTSTRSTREAM) return buf;
    buf.buf = bss->buf;
    buf.len = bss->len;
    return buf;
}

int bts_read( btStream* bts, char *buf, int len) {
    if (len<0) return -1;
    if (len == 0) { 
    	*buf = 0; 
	return 0; 
    }
    return bts->read(bts,buf,len);
}

int bts_write( btStream* bts, char *buf, int len) {
    assert(len >= 0);
    return bts->write(bts,buf,len);
}

int bts_peek( btStream* bts) {
    return bts->peek(bts);
}

int bts_printf( btStream* bts, char *fmt, ...) {
    static char buf[8192];
    int res;
    int osize;
    va_list va;
    va_start(va, fmt);
    osize=vsnprintf(buf, sizeof(buf), fmt, va);
    DIE_UNLESS(osize < sizeof(buf));
    res = bts_write( bts, buf, osize);
    va_end(va);
    return res;
}

/* 
 * bts_scanbreak
 *   Read from an input stream and copy characters into 'buf' until hitting
 *   a break character.  If there are any characters that are not listed 
 *   in 'cset' then an error value is returned.  Scanbreak is used to read
 *   bEncoded integers.
 *
 *   bts   - I/O stream, must be an input stream
 *   cset  - List of allowed characters
 *   bset  - List of break characters
 *   buf   - Return value of data read from the stream
 *   bufsz - Available space in the buffer
 *
 * return 1 on error, 0 on success 
 */
int bts_scanbreak( btStream* bts, char *cset, char *bset, char *buf, int bufsz) 
{
    char s[1];
    int res;
    int len=0;

    while (len < bufsz) {
	res = bts_read(bts, s, 1);
	if (res) return res;
	if ( strchr( bset, *s)) break;
	if ( !strchr( cset, *s)) return 1; 
	buf[len++]=*s;
    }
    if (len == bufsz) return 1;
    buf[len]=0; 
    return 0;
}