add hotspot proxy
This commit is contained in:
parent
d8c1f40bc3
commit
6afdcf9b87
53
hotspot_proxy.py
Normal file
53
hotspot_proxy.py
Normal 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()
|
Loading…
Reference in New Issue
Block a user