54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
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()
|