1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
/*
* 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;
}
|
