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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
/*
* Worldvisions Weaver Software:
* Copyright (C) 1997-2002 Net Integration Technologies, Inc.
*
* WvHttpPool test. Downloads the urls given on stdin.
*/
#include "wvhttppool.h"
#include "wvfile.h"
#include "strutils.h"
#include <signal.h>
static WvLog mylog("http2test", WvLog::Info);
static bool want_to_die = false;
static void sighandler_die(int signum)
{
fprintf(stderr, "Caught signal %d; cleaning up and terminating.\n",
signum);
want_to_die = true;
signal(signum, SIG_DFL);
}
static void close_callback(WvStream *s)
{
if (!s->isok())
mylog(WvLog::Error, "%s\n", s->geterr());
}
int main(int argc, char **argv)
{
WvIStreamList l;
WvHttpPool p;
WvString headers("");
char *line;
#ifndef _WIN32
signal(SIGPIPE, SIG_IGN);
signal(SIGINT, sighandler_die);
#endif
l.append(wvcon, false, "wvcon");
l.append(&p, false, "WvHttpPool");
while (!want_to_die && p.isok() && (wvcon->isok() || !p.idle()))
{
if (l.select(1000))
{
l.callback();
line = wvcon->getline();
if (line)
{
line = trim_string(line);
if (!line[0])
continue;
else if (strstr(line, ": "))
{
// an extra http header
headers = WvString("%s%s\n", headers, line);
continue;
}
WvStream *s = p.addurl(line, "GET", headers);
if (s)
{
static int num = 0;
WvFile *f = new WvFile(WvString("/tmp/url_%s", ++num),
O_CREAT|O_WRONLY|O_TRUNC);
assert(!f->readable);
s->autoforward(*f);
s->setclosecallback(wv::bind(close_callback, s));
l.append(s, true, "url");
l.append(f, true, "file");
}
}
}
}
if (!p.isok() && p.geterr())
mylog("HttpPool: %s\n", p.errstr());
return 0;
}
|
