本文实例讲述了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:
fwd.connect((self.newhost,self.newport))
except Exception ,ex:
log("connet newhost error:"+str(ex))
break
p1=pipethread(newsock,fwd,self.protocol)
p1.start()
p2=pipethread(fwd,newsock,self.protocol)
p2.start()
class pipethreadUDP(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('new thread for new connction')
def run(self):
while True:
try:
data,addr=self.connection['socket'].recvfrom(4096)
#log('recv from addr"%s' % str(addr))
except Exception ,ex:
log("recvfrom error:"+str(ex))
break
try:
self.connection['lock'].acquire()
self.connection['Serversocket'].sendto(data,self.connection['address'])
#log('sendto address:%s' % str(self.connection['address']))
except Exception ,ex:
log("sendto error:"+str(ex))
break
finally:self.connection['lock'].release()
self.connection['time']=time.time()
self.connection['socket'].close()
log("thread exit for: %s" % str(self.connection['address']))
self.table_lock.acquire()
self.connectionTable.pop(self.connection['address'])
self.table_lock.release()
log('Release udp connection for timeout:%s' % str(self.connection['address']))
class portmapUDP(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('udp port redirect run->local_ip:%s,local_port:%d,remote_ip:%s,remote_port:%d' % (local_ip,port,newhost,newport))
def run(self):
while True:
data,addr=self.sock.recvfrom(4096)
connection=None
newsock=None
self.table_lock.acquire()
connection=self.connetcTable.get(addr)
newconn=False
if connection is None:
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('new connection:%s' % str(addr))
self.table_lock.release()
try:
connection['socket'].sendto(data,(self.newhost,self.newport))
except Exception ,ex:
log("sendto error:"+str(ex))
#break
if newconn:
self.connetcTable[addr]=connection
t1=pipethreadUDP(connection,self.connetcTable,self.table_lock)
t1.start()
log('main thread exit')
for key in self.connetcTable.keys():
self.connetcTable[key]['socket'].close()
if __name__=='__main__':
myp=portmapUDP(10061,'10.0.1.29',161)
myp.start()
#myp.__stop()
希望本文所述对大家的Python程序设计有所帮助。
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
P70系列延期,华为新旗舰将在下月发布
3月20日消息,近期博主@数码闲聊站 透露,原定三月份发布的华为新旗舰P70系列延期发布,预计4月份上市。
而博主@定焦数码 爆料,华为的P70系列在定位上已经超过了Mate60,成为了重要的旗舰系列之一。它肩负着重返影像领域顶尖的使命。那么这次P70会带来哪些令人惊艳的创新呢?
根据目前爆料的消息来看,华为P70系列将推出三个版本,其中P70和P70 Pro采用了三角形的摄像头模组设计,而P70 Art则采用了与上一代P60 Art相似的不规则形状设计。这样的外观是否好看见仁见智,但辨识度绝对拉满。
更新日志
- 《原神》「星路拾忆」网页活动壁纸展示
- 群星.2024-祝你幸福!电影原声带【阿里巴巴】【FLAC分轨】
- 于台烟.2004-执着【乾坤唱片】【WAV+CUE】
- 黄品源.2004-感谢·情人(引进版)【滚石】【WAV+CUE】
- 《最终幻想16》评测:游、戏两难全的系列转折
- 《纸嫁衣》五部打包¥104,《鬼泣5》仅需¥44
- 《崩坏:星穹铁道》造物大赛卷出新高度
- 《王菲 菲常传奇 2CD》[WAV+CUE][950MB]
- 《王菲 空灵独味唱腔 非常传奇 紫水晶2CD》[WAV+CUE][860MB]
- 《车载专业测试王1号 极品音质》[WAV/分轨][980MB]
- 炉石传说的国服回归送什么奖励 炉石传说9.25回归最新消息介绍
- 云顶之弈最强阵容汇总2024 云顶之弈最强阵容搭配最新版本
- S14lol全球总决赛在哪观看 lol全球总决赛观赛直播地址汇总
- 黄思婷2011-感谢天[豪记唱片][WAV+CUE]
- 孙淑媚2013-太阳[台湾首版][WAV+CUE]