Commit | Line | Data |
---|---|---|
c55f394e IJ |
1 | #!/usr/bin/python3 |
2 | ||
3 | from hippotat import * | |
4 | ||
034284c3 | 5 | client_cs = None |
88487243 IJ |
6 | |
7 | def set_client(ci,cs,pw): | |
034284c3 | 8 | global client_cs |
88487243 | 9 | global password |
034284c3 IJ |
10 | assert(client_cs is None) |
11 | client_cs = cs | |
12 | c.client = ci | |
88487243 | 13 | c.max_outstanding = cfg.getint(cs, 'max_requests_outstanding') |
7b07f0b5 | 14 | c.target_outstanding = cfg.getint(cs, 'target_requests_outstanding') |
88487243 IJ |
15 | password = pw |
16 | ||
87a7c0c7 IJ |
17 | def process_cfg(): |
18 | global url | |
19 | global max_requests_outstanding | |
c55f394e | 20 | |
87a7c0c7 | 21 | process_cfg_common_always() |
88487243 IJ |
22 | process_cfg_server() |
23 | ||
24 | try: | |
25 | c.url = cfg.get('server','url') | |
26 | except NoOptionError: | |
27 | process_cfg_saddrs() | |
1d023c89 | 28 | sa = c.saddrs[0].url() |
88487243 IJ |
29 | |
30 | process_cfg_clients(set_client) | |
87a7c0c7 | 31 | |
ca732796 | 32 | c.routes = cfg.get('virtual','routes') |
7b07f0b5 IJ |
33 | c.max_queue_time = cfg.getint(client_cs, 'max_queue_time') |
34 | c.max_batch_up = cfg.getint(client_cs, 'max_batch_up') | |
034284c3 IJ |
35 | |
36 | process_cfg_ipif(client_cs, | |
37 | (('local', 'client'), | |
38 | ('peer', 'server'), | |
39 | ('rnets', 'routes'))) | |
40 | ||
ca732796 IJ |
41 | outstanding = 0 |
42 | ||
43 | def start_client(): | |
44 | global queue | |
7b07f0b5 | 45 | global agent |
ca732796 | 46 | queue = PacketQueue(c.max_queue_time) |
7b07f0b5 | 47 | agent = twisted.web.client.Agent(reactor, connectTimeout = c.http_timeout) |
ca732796 | 48 | |
034284c3 | 49 | def outbound(packet, saddr, daddr): |
ca732796 IJ |
50 | #print('OUT ', saddr, daddr, repr(packet)) |
51 | queue.append(packet) | |
52 | check_outbound() | |
53 | ||
3dbadade IJ |
54 | def req_ok(resp): |
55 | ||
7b07f0b5 IJ |
56 | |
57 | def req_err(err): | |
58 | print(err, >>sys.stderr) | |
7b07f0b5 IJ |
59 | |
60 | def req_fin(*args): | |
3dbadade | 61 | outstanding-- |
7b07f0b5 | 62 | |
ca732796 IJ |
63 | def check_outbound(): |
64 | while True: | |
7b07f0b5 IJ |
65 | if outstanding >= c.max_outstanding : break |
66 | if not queue.nonempty() && outstanding >= c.target_outstanding: break | |
67 | ||
68 | d = b'' | |
69 | queue.process((lambda: len(d)), | |
70 | (lambda s: d += s), | |
71 | c.max_batch_up) | |
72 | assert(len(d)) | |
73 | ||
74 | crlf = b'\r\n' | |
75 | mime = (b'--b' + crlf + | |
76 | b'Content-Disposition: form-data; name="m"' + crlf + | |
77 | password + crlf + | |
78 | c.client + crlf + | |
79 | c.target_outstanding + crlf + | |
80 | b'--b' + crlf + | |
81 | b'Content-Disposition: form-data; name="d"' + crlf + | |
82 | mime_translate(d) + crlf + | |
83 | b'--b--' + crlf) | |
ca732796 | 84 | |
7b07f0b5 IJ |
85 | hh = { 'User-Agent': ['hippotat'], |
86 | 'Content-Type': ['multipart/form-data; boundary="b"'] } | |
3dbadade | 87 | req = agent.request(b'POST', |
7b07f0b5 IJ |
88 | c.url, |
89 | twisted.web.client.Headers(hh)) | |
90 | req.addTimeout(c.http_timeout, | |
91 | req.addCallbacks(req_ok, req_err) | |
92 | req.addBoth(req_fin) | |
93 | outstanding++ | |
034284c3 | 94 | |
1321ad5f | 95 | common_startup() |
87a7c0c7 | 96 | process_cfg() |
7b07f0b5 | 97 | start_client() |
034284c3 IJ |
98 | start_ipif(c.ipif_command, outbound) |
99 | common_run() |