More work on proxy

This commit is contained in:
Simon 2020-12-28 14:33:11 +00:00
parent 870208de8e
commit 00681cd9db

View File

@ -24,12 +24,13 @@ class Proxy(DatagramProtocol):
self.transport.write(data,(self.connTrack[port]['host'],self.connTrack[port]['sport'])) self.transport.write(data,(self.connTrack[port]['host'],self.connTrack[port]['sport']))
#if master refuses login, remove tracking #if master refuses login, remove tracking
if data[0:6] == b'MSTNAK': if data[0:6] == b'MSTNAK':
del self.sourceTrack[host+":"+str(port)] if self.sourceTrack[self.connTrack[port]['host']+":"+self.connTrack[port]['sport']]:
del self.sourceTrack[self.connTrack[port]['host']+":"+self.connTrack[port]['sport']]
if Debug: if Debug:
print("return path match") print("return path match")
print(data) print(data)
elif host+":"+str(port) in self.sourceTrack: elif host+":"+str(port) in self.sourceTrack:
del self.sourceTrack[host+":"+str(port)] del self.sourceTrack[self.connTrack[port]['host']+":"+self.connTrack[port]['sport']]
return return
#If we have a sourcetrack for this connect and thenowtimeout has not expired, forward to tracked port #If we have a sourcetrack for this connect and thenowtimeout has not expired, forward to tracked port