8 import twisted
.internet
9 from twisted
.web
.server
import NOT_DONE_YET
11 #import twisted.web.server import Site
12 #from twisted.web.resource import Resource
20 max_batch_down = 65536
23 target_requests_outstanding = 3
24 ipif = userv root ipif %(local)s,%(peer)s,%(mtu)s,slip %(rnets)s
37 max_batch_down = 262144
39 max_request_time = 121
40 target_requests_outstanding = 10
43 #---------- "router" ----------
45 def route(packet
, saddr
, daddr
):
46 print('TRACE ', saddr
, daddr
, packet
)
47 try: client
= clients
[daddr
]
48 except KeyError: dclient
= None
49 if dclient
is not None:
50 dclient
.queue_outbound(packet
)
51 elif saddr
.is_link_local
or daddr
.is_link_local
:
52 log_discard(packet
, saddr
, daddr
, 'link-local')
53 elif daddr
== c
.server
or daddr
not in c
.network
:
54 print('TRACE INBOUND ', saddr
, daddr
, packet
)
57 log_discard(packet
, saddr
, daddr
, 'relay')
59 log_discard(packet
, saddr
, daddr
, 'no client')
61 def log_discard(packet
, saddr
, daddr
, why
):
62 print('DROP ', saddr
, daddr
, why
)
63 # syslog.syslog(syslog.LOG_DEBUG,
64 # 'discarded packet %s -> %s (%s)' % (saddr, daddr, why))
66 #---------- client ----------
69 def __init__(self
, ip
, cs
, pw
):
70 # instance data members
74 self
._rq
= collections
.deque() # requests
75 # self._pq = PacketQueue(...)
80 # .target_requests_outstanding
82 if ip
not in c
.network
:
83 raise ValueError('client %s not in network' % ip
)
85 for k
in ('max_batch_down','max_queue_time','max_request_time',
86 'target_requests_outstanding'):
87 req
= cfg
.getint(cs
, k
)
88 limit
= cfg
.getint('limits',k
)
89 self
.__dict__
[k
] = min(req
, limit
)
90 self
._pq
= PacketQueue(self
.max_queue_time
)
93 raise ValueError('multiple client cfg sections for %s' % ip
)
96 def process_arriving_data(self
, d
):
97 for packet
in slip
.decode(d
):
98 (saddr
, daddr
) = packet_addrs(packet
)
100 raise ValueError('wrong source address %s' % saddr
)
101 route(packet
, saddr
, daddr
)
103 def _req_cancel(self
, request
):
106 def _req_error(self
, err
, request
):
107 self
._req_cancel(request
)
109 def queue_outbound(self
, packet
):
110 self
._pq
.append(packet
)
112 def http_request(self
, request
):
113 request
.setHeader('Content-Type','application/octet-stream')
114 reactor
.callLater(self
.max_request_time
, self
._req_cancel
, request
)
115 request
.notifyFinish().addErrback(self
._req_error
, request
)
116 self
._rq
.append(request
)
117 self
._check_outbound()
119 def _check_outbound(self
):
121 try: request
= self
._rq
[0]
122 except IndexError: request
= None
123 if request
and request
.finished
:
127 if not self
._pq
.nonempty():
128 # no packets, oh well
135 # request, and also some non-expired packets
137 packet
= self
.pq
.popleft()
138 if packet
is None: break
140 encoded
= slip
.encode(packet
)
142 if request
.sentLength
> 0:
143 if (request
.sentLength
+ len(slip
.delimiter
)
144 + len(encoded
) > self
.max_batch_down
):
146 request
.write(slip
.delimiter
)
148 request
.write(encoded
)
151 assert(request
.sentLength
)
154 # round again, looking for more to do
156 while len(self
._rq
) > self
.target_requests_outstanding
:
157 request
= self
._rq
.popleft()
160 class IphttpResource(twisted
.web
.resource
.Resource
):
162 def render_POST(self
, request
):
163 # find client, update config, etc.
164 ci
= ipaddr(request
.args
['i'])
166 pw
= request
.args
['pw']
167 if pw
!= c
.pw
: raise ValueError('bad password')
170 for r
, w
in (('mbd', 'max_batch_down'),
171 ('mqt', 'max_queue_time'),
172 ('mrt', 'max_request_time'),
173 ('tro', 'target_requests_outstanding')):
174 try: v
= request
.args
[r
]
175 except KeyError: continue
179 try: d
= request
.args
['d']
180 except KeyError: d
= ''
182 c
.process_arriving_data(d
)
183 c
.new_request(request
)
185 def render_GET(self
, request
):
186 return b
'<html><body>hippotat</body></html>'
189 resource
= IphttpResource()
190 site
= twisted
.web
.server
.Site(resource
)
192 ep
= sa
.make_endpoint()
193 crash_on_defer(ep
.listen(site
))
195 #---------- config and setup ----------
198 process_cfg_common_always()
200 process_cfg_network()
203 c
.relay
= cfg
.get('virtual','relay')
204 except NoOptionError
:
205 for search
in c
.network
.hosts():
206 if search
== c
.server
: continue
211 process_cfg_clients(Client
)
213 process_cfg_ipif('server',
216 ('rnets','network')))
218 common_startup(defcfg
)
220 start_ipif(c
.ipif_command
, route
)