diff --git a/hotspot_proxy.py b/hotspot_proxy.py new file mode 100644 index 0000000..4c47715 --- /dev/null +++ b/hotspot_proxy.py @@ -0,0 +1,53 @@ +from twisted.internet.protocol import DatagramProtocol +from twisted.internet import reactor +from time import time + + +class Proxy(DatagramProtocol): + + def __init__(self,ListenPort,connTrack,Timeout): + self.connTrack = connTrack + self.timeout = Timeout + + def datagramReceived(self, data, addr): + host,port = addr + + if host == '127.0.0.1' and port in self.connTrack: + if int(self.connTrack[port]['time'])+self.timeout > time(): + self.transport.write(data,(self.connTrack[port]['host'],self.connTrack[port]['sport'])) + return + + for dport in self.connTrack: + if self.connTrack[dport]['host'] == host and self.connTrack[dport]['sport'] == port and (int(self.connTrack[dport]['time'])+self.timeout > time()): + self.connTrack[dport]['time'] = time() + self.connTrack[dport]['host'] = host + self.connTrack[dport]['sport'] = port + self.transport.write(data, ('127.0.0.1',dport)) + self.connTrack[dport]['time'] = time() + return + + for dport in self.connTrack: + if (self.connTrack[dport]['time'] == False or (int(self.connTrack[dport]['time'])+self.timeout < time())): + self.connTrack[dport]['sport'] = port + self.connTrack[dport]['host'] = host + self.connTrack[dport]['time'] = time() + self.transport.write(data, ('127.0.0.1',dport)) + return + + + +if __name__ == '__main__': + + ListenPort = 62031 + DestportStart = 50000 + DestPortEnd = 50300 + Timeout = 60 + + CONNTRACK = {} + + for port in range(DestportStart,DestPortEnd,1): + CONNTRACK[port] = {'host': False,'time': False,'sport':False} + + reactor.listenUDP(ListenPort,Proxy(ListenPort,CONNTRACK,Timeout)) + + reactor.run()