日韩久久久精品,亚洲精品久久久久久久久久久,亚洲欧美一区二区三区国产精品 ,一区二区福利

python實現(xiàn)簡單的TCP代理服務器

系統(tǒng) 1995 0

本文實例講述了python實現(xiàn)簡單的TCP代理服務器的方法,分享給大家供大家參考。

具體實現(xiàn)代碼如下:

            
# -*- coding: utf-8 -*-

'''
filename:rtcp.py
@desc:
利用python的socket端口轉(zhuǎn)發(fā),用于遠程維護
如果連接不到遠程,會sleep 36s,最多嘗試200(即兩小時)

@usage:
./rtcp.py stream1 stream2
stream為:l:port或c:host:port
l:port表示監(jiān)聽指定的本地端口
c:host:port表示監(jiān)聽遠程指定的端口

@author: watercloud, zd, knownsec team
@web: www.knownsec.com, blog.knownsec.com
@date: 2009-7
'''

import socket
import sys
import threading
import time

streams = [None, None] # 存放需要進行數(shù)據(jù)轉(zhuǎn)發(fā)的兩個數(shù)據(jù)流(都是SocketObj對象)
debug = 1 # 調(diào)試狀態(tài) 0 or 1

def _usage():
 print 'Usage: ./rtcp.py stream1 stream2\nstream : l:port or c:host:port'

def _get_another_stream(num):
 '''
 從streams獲取另外一個流對象,如果當前為空,則等待
 '''
 if num == 0:
 num = 1
 elif num == 1:
 num = 0
 else:
 raise "ERROR"

 while True:
 if streams[num] == 'quit':
  print("can't connect to the target, quit now!")
  sys.exit(1)

 if streams[num] != None:
  return streams[num]
 else:
  time.sleep(1)

def _xstream(num, s1, s2):
 '''
 交換兩個流的數(shù)據(jù)
 num為當前流編號,主要用于調(diào)試目的,區(qū)分兩個回路狀態(tài)用。
 '''
 try:
 while True:
  #注意,recv函數(shù)會阻塞,直到對端完全關閉(close后還需要一定時間才能關閉,最快關閉方法是shutdow)
  buff = s1.recv(1024)
  if debug > 0:
  print num,"recv"
  if len(buff) == 0: #對端關閉連接,讀不到數(shù)據(jù)
  print num,"one closed"
  break
  s2.sendall(buff)
  if debug > 0:
  print num,"sendall"
 except :
 print num,"one connect closed."

 try:
 s1.shutdown(socket.SHUT_RDWR)
 s1.close()
 except:
 pass

 try:
 s2.shutdown(socket.SHUT_RDWR)
 s2.close()
 except:
 pass

 streams[0] = None
 streams[1] = None
 print num, "CLOSED"

def _server(port, num):
 '''
 處理服務情況,num為流編號(第0號還是第1號)
 '''
 srv = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 srv.bind(('0.0.0.0', port))
 srv.listen(1)
 while True:
 conn, addr = srv.accept()
 print "connected from:", addr
 streams[num] = conn # 放入本端流對象
 s2 = _get_another_stream(num) # 獲取另一端流對象
 _xstream(num, conn, s2)

def _connect(host, port, num):
 ''' 處理連接,num為流編號(第0號還是第1號)

 @note: 如果連接不到遠程,會sleep 36s,最多嘗試200(即兩小時)
 '''
 not_connet_time = 0
 wait_time = 36
 try_cnt = 199
 while True:
 if not_connet_time > try_cnt:
  streams[num] = 'quit'
  print('not connected')
  return None

 conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 try:
  conn.connect((host, port))
 except Exception, e:
  print ('can not connect %s:%s!' % (host, port))
  not_connet_time += 1
  time.sleep(wait_time)
  continue

 print "connected to %s:%i" % (host, port)
 streams[num] = conn #放入本端流對象
 s2 = _get_another_stream(num) #獲取另一端流對象
 _xstream(num, conn, s2)


if __name__ == '__main__':
 if len(sys.argv) != 3:
 _usage()
 sys.exit(1)
 tlist = [] # 線程列表,最終存放兩個線程對象
 targv = [sys.argv[1], sys.argv[2] ]
 for i in [0, 1]:
 s = targv[i] # stream描述 c:ip:port 或 l:port
 sl = s.split(':')
 if len(sl) == 2 and (sl[0] == 'l' or sl[0] == 'L'): # l:port
  t = threading.Thread(target=_server, args=(int(sl[1]), i))
  tlist.append(t)
 elif len(sl) == 3 and (sl[0] == 'c' or sl[0] == 'C'): # c:host:port
  t = threading.Thread(target=_connect, args=(sl[1], int(sl[2]), i))
  tlist.append(t)
 else:
  _usage()
  sys.exit(1)

 for t in tlist:
 t.start()
 for t in tlist:
 t.join()
 sys.exit(0)


          

完整實例代碼點擊此處本站下載。

希望本文所述對大家的Python程序設計有所幫助。


更多文章、技術交流、商務合作、聯(lián)系博主

微信掃碼或搜索:z360901061

微信掃一掃加我為好友

QQ號聯(lián)系: 360901061

您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。

【本文對您有幫助就好】

您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描上面二維碼支持博主2元、5元、10元、自定義金額等您想捐的金額吧,站長會非常 感謝您的哦!!!

發(fā)表我的評論
最新評論 總共0條評論
主站蜘蛛池模板: 丹寨县| 冷水江市| 都匀市| 金湖县| 伊宁市| 肃北| 峡江县| 汉阴县| 宁都县| 格尔木市| 丽江市| 昆明市| 石城县| 朝阳市| 开阳县| 钦州市| 东港市| 浦城县| 理塘县| 桦川县| 车致| 当涂县| 永和县| 津市市| 阿荣旗| 祁东县| 莒南县| 滁州市| 定边县| 龙川县| 库尔勒市| 阿图什市| 芦山县| 清河县| 绥宁县| 山阳县| 清新县| 榆社县| 顺义区| 大足县| 天气|