Added alloc/stream options to fetcher test utility

This commit is contained in:
Martin Willi 2011-04-01 11:40:18 +02:00
parent f27705cea1
commit 4ceb31f941
1 changed files with 32 additions and 6 deletions

View File

@ -19,6 +19,18 @@
#include <library.h>
#include <debug.h>
static int count = 0;
static bool cb(void *userdata, chunk_t chunk)
{
if (write(1, chunk.ptr, chunk.len) == chunk.len)
{
count++;
return TRUE;
}
return FALSE;
}
int main(int argc, char *argv[])
{
chunk_t res;
@ -27,15 +39,29 @@ int main(int argc, char *argv[])
atexit(library_deinit);
lib->plugins->load(lib->plugins, NULL, PLUGINS);
if (argc != 2)
if (argc != 3 || (!streq(argv[1], "a") && !streq(argv[1], "s")))
{
fprintf(stderr, "usage: %s <url>\n", argv[0]);
fprintf(stderr, "usage: %s a|s <url>\n", argv[0]);
return 1;
}
if (lib->fetcher->fetch(lib->fetcher, argv[1], &res, FETCH_END) == SUCCESS)
if (streq(argv[1], "a"))
{
ignore_result(write(1, res.ptr, res.len));
free(res.ptr);
return 0;
if (lib->fetcher->fetch(lib->fetcher, argv[2], &res,
FETCH_END) == SUCCESS)
{
ignore_result(write(1, res.ptr, res.len));
free(res.ptr);
return 0;
}
}
else
{
if (lib->fetcher->fetch(lib->fetcher, argv[2], NULL,
FETCH_CALLBACK, cb, FETCH_END) == SUCCESS)
{
fprintf(stderr, "received %d chunks\n", count);
return 0;
}
}
return 1;
}