python实现端口转发器的方法
本文实例讲述了python实现端口转发器的方法。分享给大家供大家参考。具体如下：
下面的python代码实现端口转发器，支持udp端口转发
由于工作需要用到一个端口转发器，并且要求支持TCP和UDP协议。在网上找了蛮久，但没有中意的。于是就自己写了一个。这个转发器是基于pythoncookbook的一个示例改写的，原先的这个示例只支持TCP协议，我这里增加了UDP协议的支持，程序写的不怎么好，不过它确实能用！
portmap.py代码如下：
#-*-coding:UTF-8-*-
'''
Createdon2012-5-8
@author:qh
'''
importtime,socket,threading
deflog(strLog):
strs=time.strftime("%Y-%m-%d%H:%M:%S")
printstrs+"->"+strLog
classpipethread(threading.Thread):
'''
classdocs
'''
def__init__(self,source,sink):
'''
Constructor
'''
threading.Thread.__init__(self)
self.source=source
self.sink=sink
log("NewPipecreate:%s->%s"%(self.source.getpeername(),self.sink.getpeername()))
defrun(self):
whileTrue:
try:
data=self.source.recv(1024)
ifnotdata:break
self.sink.send(data)
exceptException,ex:
log("redirecterror:"+str(ex))
break
self.source.close()
self.sink.close()
classportmap(threading.Thread):
def__init__(self,port,newhost,newport,local_ip=''):
threading.Thread.__init__(self)
self.newhost=newhost
self.newport=newport
self.port=port
self.local_ip=local_ip
self.sock=None
self.sock=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
self.sock.bind((self.local_ip,port))
self.sock.listen(5)
log("startlistenprotocol:%s,port:%d"%('tcp',port))
defrun(self):
whileTrue:
fwd=None
newsock=None
newsock,address=self.sock.accept()
log("newconnection->protocol:%s,localport:%d,remoteaddress:%s"%('tcp',self.port,address[0]))
fwd=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
try:
fwd.connect((self.newhost,self.newport))
exceptException,ex:
log("connetnewhosterror:"+str(ex))
break
p1=pipethread(newsock,fwd,self.protocol)
p1.start()
p2=pipethread(fwd,newsock,self.protocol)
p2.start()
classpipethreadUDP(threading.Thread):
def__init__(self,connection,connectionTable,table_lock):
threading.Thread.__init__(self)
self.connection=connection
self.connectionTable=connectionTable
self.table_lock=table_lock
log('newthreadfornewconnction')
defrun(self):
whileTrue:
try:
data,addr=self.connection['socket'].recvfrom(4096)
#log('recvfromaddr"%s'%str(addr))
exceptException,ex:
log("recvfromerror:"+str(ex))
break
try:
self.connection['lock'].acquire()
self.connection['Serversocket'].sendto(data,self.connection['address'])
#log('sendtoaddress:%s'%str(self.connection['address']))
exceptException,ex:
log("sendtoerror:"+str(ex))
break
finally:self.connection['lock'].release()
self.connection['time']=time.time()
self.connection['socket'].close()
log("threadexitfor:%s"%str(self.connection['address']))
self.table_lock.acquire()
self.connectionTable.pop(self.connection['address'])
self.table_lock.release()
log('Releaseudpconnectionfortimeout:%s'%str(self.connection['address']))
classportmapUDP(threading.Thread):
def__init__(self,port,newhost,newport,local_ip=''):
threading.Thread.__init__(self)
self.newhost=newhost
self.newport=newport
self.port=port
self.local_ip=local_ip
self.sock=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
self.sock.bind((self.local_ip,port))
self.connetcTable={}
self.port_lock=threading.Lock()
self.table_lock=threading.Lock()
self.timeout=300
#ScanUDP(self.connetcTable,self.table_lock).start()
log('udpportredirectrun->local_ip:%s,local_port:%d,remote_ip:%s,remote_port:%d'%(local_ip,port,newhost,newport))
defrun(self):
whileTrue:
data,addr=self.sock.recvfrom(4096)
connection=None
newsock=None
self.table_lock.acquire()
connection=self.connetcTable.get(addr)
newconn=False
ifconnectionisNone:
connection={}
connection['address']=addr
newsock=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
newsock.settimeout(self.timeout)
connection['socket']=newsock
connection['lock']=self.port_lock
connection['Serversocket']=self.sock
connection['time']=time.time()
newconn=True
log('newconnection:%s'%str(addr))
self.table_lock.release()
try:
connection['socket'].sendto(data,(self.newhost,self.newport))
exceptException,ex:
log("sendtoerror:"+str(ex))
#break
ifnewconn:
self.connetcTable[addr]=connection
t1=pipethreadUDP(connection,self.connetcTable,self.table_lock)
t1.start()
log('mainthreadexit')
forkeyinself.connetcTable.keys():
self.connetcTable[key]['socket'].close()
if__name__=='__main__':
myp=portmapUDP(10061,'10.0.1.29',161)
myp.start()
#myp.__stop()
希望本文所述对大家的Python程序设计有所帮助。