add hotspot proxy

This commit is contained in:
Simon 2020-11-28 20:52:22 +00:00
parent d8c1f40bc3
commit 6afdcf9b87

53
hotspot_proxy.py Normal file
View File

@ -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()