python实现端口转发器的方法实现端口转发器的方法
主要介绍了python实现端口转发器的方法,涉及Python实现端口转发的技巧,支持TCP和UDP协议,需要的朋友可
以参考下
本文实例讲述了python实现端口转发器的方法。分享给大家供大家参考。具体如下：
下面的python代码实现端口转发器，支持udp端口转发
由于工作需要用到一个端口转发器，并且要求支持TCP和UDP协议。在网上找了蛮久，但没有中意的。于是就自己写了一
个。这个转发器是基于python cookbook的一个示例改写的，原先的这个示例只支持TCP协议，我这里增加了UDP协议的支
持，程序写的不怎么好，不过它确实能用！
portmap.py代码如下：
复制代码 代码如下:
#-* -coding: UTF-8 -* -
'''
Created on 2012-5-8
@author: qh
'''
import time,socket,threading
def log(strLog):
strs=time.strftime("%Y-%m-%d %H:%M:%S")
print strs+"->"+strLog
class pipethread(threading.Thread):
'''
classdocs
'''
def __init__(self,source,sink):
'''
Constructor
'''
threading.Thread.__init__(self)
self.source=source
self.sink=sink
log("New Pipe create:%s->%s" % (self.source.getpeername(),self.sink.getpeername()))
def run(self):
while True:
try:
data=self.source.recv(1024)
if not data: break
self.sink.send(data)
except Exception ,ex:
log("redirect error:"+str(ex))
break
self.source.close()
self.sink.close()
class portmap(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("start listen protocol:%s,port:%d " % ('tcp',port))
def run(self):
while True:
fwd=None
newsock=None
newsock,address=self.sock.accept()
log("new connection->protocol:%s,local port:%d,remote address:%s" % ('tcp',self.port,address[0]))
fwd=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
try:
评论0