python实现简单的TCP代理服务器


Posted in Python onOctober 08, 2014

本文实例讲述了python实现简单的TCP代理服务器的方法,分享给大家供大家参考。

具体实现代码如下:

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

'''
filename:rtcp.py
@desc:
利用python的socket端口转发,用于远程维护
如果连接不到远程,会sleep 36s,最多尝试200(即两小时)

@usage:
./rtcp.py stream1 stream2
stream为:l:port或c:host:port
l:port表示监听指定的本地端口
c:host:port表示监听远程指定的端口

@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] # 存放需要进行数据转发的两个数据流(都是SocketObj对象)
debug = 1 # 调试状态 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):
 '''
 交换两个流的数据
 num为当前流编号,主要用于调试目的,区分两个回路状态用。
 '''
 try:
 while True:
  #注意,recv函数会阻塞,直到对端完全关闭(close后还需要一定时间才能关闭,最快关闭方法是shutdow)
  buff = s1.recv(1024)
  if debug > 0:
  print num,"recv"
  if len(buff) == 0: #对端关闭连接,读不到数据
  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程序设计有所帮助。

Python 相关文章推荐
Python多线程编程(四):使用Lock互斥锁
Apr 05 Python
pygame学习笔记(1):矩形、圆型画图实例
Apr 15 Python
python写日志封装类实例
Jun 28 Python
Django查找网站项目根目录和对正则表达式的支持
Jul 15 Python
详解tensorflow载入数据的三种方式
Apr 24 Python
在windows下使用python进行串口通讯的方法
Jul 02 Python
Python通过递归获取目录下指定文件代码实例
Nov 07 Python
python安装和pycharm环境搭建设置方法
May 27 Python
Python-openCV开运算实例
Jul 05 Python
Python 抓取数据存储到Redis中的操作
Jul 16 Python
Python中logger日志模块详解
Aug 04 Python
Pytorch实现WGAN用于动漫头像生成
Mar 04 Python
python操作CouchDB的方法
Oct 08 #Python
python基于queue和threading实现多线程下载实例
Oct 08 #Python
python实现封装得到virustotal扫描结果
Oct 05 #Python
python解析xml文件操作实例
Oct 05 #Python
python写xml文件的操作实例
Oct 05 #Python
python实现上传样本到virustotal并查询扫描信息的方法
Oct 05 #Python
python实现计算资源图标crc值的方法
Oct 05 #Python
You might like
改德生G88 - 加装等响度低音提升电路
2021/03/02 无线电
php5编程中的异常处理详细方法介绍
2008/07/29 PHP
PHP版本如何选择?应该使用哪个版本?
2015/05/13 PHP
PHP如何通过AJAX方式实现登录功能
2015/11/23 PHP
PHP微信开发之有道翻译
2016/06/23 PHP
PHP基于socket实现客户端和服务端通讯功能
2017/07/13 PHP
Laravel中unique和exists验证规则的优化详解
2018/01/28 PHP
PHP中的Iterator迭代对象属性详解
2019/04/12 PHP
jquery调用wcf并展示出数据的方法
2011/07/07 Javascript
各浏览器中querySelector和querySelectorAll的实现差异分析
2012/05/23 Javascript
javascript实现网页字符定位的方法
2015/07/14 Javascript
Node.js与Sails ~项目结构与Mvc实现及日志机制
2015/10/14 Javascript
JS中JSON对象和String之间的互转及处理技巧
2016/04/06 Javascript
jQuery实现的表头固定效果实例【附完整demo源码下载】
2016/08/01 Javascript
鼠标拖动改变DIV等网页元素的大小的实现方法
2017/07/06 Javascript
如何编写一个d.ts文件的步骤详解
2018/04/13 Javascript
Js中将Long转换成日期格式的实现方法
2018/06/05 Javascript
vue 音乐App QQ音乐搜索列表最新接口跨域设置方法
2018/09/25 Javascript
Python新手在作用域方面经常容易碰到的问题
2015/04/03 Python
利用python获取Ping结果示例代码
2017/07/06 Python
pytorch多进程加速及代码优化方法
2019/08/19 Python
python中get和post有什么区别
2020/06/19 Python
HTML5 中新的全局属性(整理)
2013/07/31 HTML / CSS
利用三角函数在canvas上画虚线的方法
2018/01/11 HTML / CSS
Oasis服装官网:时尚女装在线
2020/07/09 全球购物
计算机求职信
2013/12/01 职场文书
诉讼代理人授权委托书
2014/04/08 职场文书
感恩寄语大全
2014/04/11 职场文书
银行先进个人事迹材料
2014/05/11 职场文书
入党综合考察材料
2014/06/02 职场文书
2014年社区妇联工作总结
2014/12/02 职场文书
2015年初三班主任工作总结
2015/05/21 职场文书
乡镇团代会开幕词
2016/03/04 职场文书
导游词之峨眉山
2019/12/16 职场文书
详细聊聊浏览器是如何看闭包的
2021/11/11 Javascript
Mysql Innodb存储引擎之索引与算法
2022/02/15 MySQL