extern/sshuttle
1
1
Fork 0
mirror of https://github.com/sshuttle/sshuttle.git synced 2024-11-22 16:03:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

When DNS response received, MUX channel no longer required. Delete it.

Browse Source
This commit is contained in:
Brian May 2011-06-06 10:58:15 +10:00
parent 50c2b86f15
commit 6500067905
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
client.py
View File

@ -220,10 +220,11 @@ def onaccept_tcp(listener, mux, handlers):
expire_connections(time.time(), mux)
def dns_done(chan, data):
def dns_done(chan, mux, data):
peer,sock,timeout = dnsreqs.get(chan) or (None,None,None)
debug3('dns_done: channel=%r peer=%r\n' % (chan, peer))
if peer:
del mux.channels[chan]
del dnsreqs[chan]
debug3('doing sendto %r\n' % (peer,))
sock.sendto(data, peer)
@ -237,7 +238,7 @@ def ondns(listener, mux, handlers):
chan = mux.next_channel()
dnsreqs[chan] = peer,listener,now+30
mux.send(chan, ssnet.CMD_DNS_REQ, pkt)
mux.channels[chan] = lambda cmd,data: dns_done(chan,data)
mux.channels[chan] = lambda cmd,data: dns_done(chan, mux, data)
expire_connections(now, mux)