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程序设计有所帮助。