diff options
Diffstat (limited to 'test/inputstream.c')
-rw-r--r-- | test/inputstream.c | 97 |
1 files changed, 97 insertions, 0 deletions
diff --git a/test/inputstream.c b/test/inputstream.c new file mode 100644 index 0000000..bad3127 --- /dev/null +++ b/test/inputstream.c @@ -0,0 +1,97 @@ +#include <inttypes.h> +#include <stdio.h> + +#include <parserutils/parserutils.h> +#include <parserutils/charset/utf8.h> +#include <parserutils/input/inputstream.h> + +#include "utils/utils.h" + +#include "testutils.h" + +static void *myrealloc(void *ptr, size_t len, void *pw) +{ + UNUSED(pw); + + return realloc(ptr, len); +} + +int main(int argc, char **argv) +{ + parserutils_inputstream *stream; + FILE *fp; + size_t len, origlen; +#define CHUNK_SIZE (4096) + uint8_t buf[CHUNK_SIZE]; + uintptr_t c; + size_t clen; + + if (argc != 3) { + printf("Usage: %s <aliases_file> <filename>\n", argv[0]); + return 1; + } + + /* Initialise library */ + assert(parserutils_initialise(argv[1], myrealloc, NULL) == + PARSERUTILS_OK); + + stream = parserutils_inputstream_create("UTF-8", 1, NULL, + myrealloc, NULL); + assert(stream != NULL); + + fp = fopen(argv[2], "rb"); + if (fp == NULL) { + printf("Failed opening %s\n", argv[2]); + return 1; + } + + fseek(fp, 0, SEEK_END); + origlen = len = ftell(fp); + fseek(fp, 0, SEEK_SET); + + while (len >= CHUNK_SIZE) { + fread(buf, 1, CHUNK_SIZE, fp); + + assert(parserutils_inputstream_append(stream, + buf, CHUNK_SIZE) == PARSERUTILS_OK); + + len -= CHUNK_SIZE; + + while ((c = parserutils_inputstream_peek(stream, 0, &clen)) != + PARSERUTILS_INPUTSTREAM_OOD) { + parserutils_inputstream_advance(stream, clen); + } + } + + if (len > 0) { + fread(buf, 1, len, fp); + + assert(parserutils_inputstream_append(stream, + buf, len) == PARSERUTILS_OK); + + len = 0; + } + + fclose(fp); + + assert(parserutils_inputstream_insert(stream, + (const uint8_t *) "hello!!!", + SLEN("hello!!!")) == PARSERUTILS_OK); + + assert(parserutils_inputstream_append(stream, NULL, 0) == + PARSERUTILS_OK); + + while ((c = parserutils_inputstream_peek(stream, 0, &clen)) != + PARSERUTILS_INPUTSTREAM_EOF) { + parserutils_inputstream_advance(stream, clen); + } + + parserutils_inputstream_destroy(stream); + + assert(parserutils_finalise(myrealloc, NULL) == PARSERUTILS_OK); + + printf("PASS\n"); + + return 0; +} + |