68 lines
1.5 KiB
C
68 lines
1.5 KiB
C
/*
|
|
* Copyright (C) 2010 Martin Willi
|
|
* Copyright (C) 2010 revosec AG
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License as published by the
|
|
* Free Software Foundation; either version 2 of the License, or (at your
|
|
* option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
|
|
*
|
|
* This program 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 General Public License
|
|
* for more details.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
|
|
#include <library.h>
|
|
#include <utils/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;
|
|
|
|
library_init(NULL, "fetch");
|
|
atexit(library_deinit);
|
|
lib->plugins->load(lib->plugins, PLUGINS);
|
|
|
|
if (argc != 3 || (!streq(argv[1], "a") && !streq(argv[1], "s")))
|
|
{
|
|
fprintf(stderr, "usage: %s a|s <url>\n", argv[0]);
|
|
return 1;
|
|
}
|
|
if (streq(argv[1], "a"))
|
|
{
|
|
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;
|
|
}
|