python用socket实现协议TCP长连接框架
作者:多姿多彩 发布时间:2022-05-08 00:22:47
“ 使用python实现协议中常见的TCP长连接框架。”
分析多了协议就会发现,很多的应用,特别是游戏类和IM类应用,它们的协议会使用长连接的方式,来保持客户端与服务器的联系,这些长连接,通常是TCP承载的。
如果我们要模拟这个客户端的行为,根据不同应用服务器的实现情况,有些长连接不是必须的,但有些长连接,就必须去实现它。例如最近分析的某应用,虽然它主要使用HTTP协议进行交互,但它在TCP长连接中传输了一些必须的信息,如果不实现长连接,就会有很多信息无法处理。
在python中,很容易实现HTTP协议,当然,也容易实现TCP协议,它的TCP实现,使用socket库就可以了,只是需要注意,TCP长连接中通常传输的是十六进制数据,协议非标准的,需要自行根据协议分析结果来封装数据格式。
这里以一个使用到TCP长连接的协议为样例,来给出协议的TCP长连接框架,大家有需要可以参考实现,当然,代码也是从样例中摘出来的,并不是完整的。
我的TCP长连接框架,首先是外部的包装,初始化一些参数,例如长连接使用到的ip端口及socket套接字等:
self.longip='im.langren001.com'
self.longport= 6656
self.threadLock = threading.Lock()
self.sockmain = socket.socket(socket.AF_INET, socket.SOCK_STREAM);
self.longlinktcpstart2()
tlonglink = threading.Thread(target=lrsuser.longlinktcpth2,name='mainlink_'+ self.playinfo['uid'], args=(self,))
tlonglink.start()
self.threadinfo.append(tlonglink)
这个里面调用了两个函数,一个是longlinktcpstart2函数,作用是建立socket连接,并对一些连接建立初始时的交互进行实现,另一个是longlinktcpth2函数,是一个线程,实现对连接内的数据进行收发处理。一般来说,这两个可以在一起实现,但为了方便socket异常断开的处理,分成了两个函数。
longlinktcpstart2的实现如下:
def longlinktcpstart2(self):
server_address = (self.longip, int(self.longport))
self.savelogs('longlinktcpstart2', 'Connecting to %s:%d.' % server_address)
self.sockmain.connect(server_address)
self.databuf = b''
message = genbaseinfo.genalive()
self.sockmain.sendall(message)
message = genbaseinfo.genfirstdata()
if len(message)==0:
self.savelogs('longlinktcpstart2', 'genfirstdata error ')
return False
self.sockmain.sendall(message)
self.longlinkcnt=2
cnt = 0
while (cnt < 2):
try:
buf = self.sockmain.recv(2048)
sz = len(buf)
self.savelogs('longlinktcpstart2', "recv data len "+str(sz) )
if sz > 0:
self.databuf +=buf
self.dealdatabuf()
if cnt == 0:
alivemsg = genbaseinfo.genalive()
self.sockmain.sendall(alivemsg)
self.savelogs('longlinktcpstart2', "sendalive")
regtime=int(round(time.time() * 1000))-random.randint(14400000,25200000)
regtime=regtime*1000
pcode = self.versionstr + '.0'
message = genbaseinfo.genseconddata()
if len(message) == 0:
self.savelogs('longlinktcpstart2', 'genseconddata error ')
return False
self.sockmain.sendall(message)
self.longlinkcnt = self.longlinkcnt + 1
elif cnt == 1:
pcode = self.versionstr + '.0'
message = genbaseinfo.genotherdata()
if len(message) == 0:
self.savelogs('longlinktcpstart2', 'genthirddata error ')
return False
self.sockmain.sendall(message)
self.longlinkcnt = self.longlinkcnt + 1
cnt = cnt + 1
else:
self.savelogs('longlinktcpstart2', 'recv data alive')
except: # socket.error
self.savelogs('longlinktcpstart2', 'socket error,do connect fail')
return False
return True
这里面的genbaseinfo 相关的函数可以忽略,是用来生成发送的消息数据的实现,用自己的函数去替换即可。dealdatabuf函数是用来处理收到的消息数据实现,这两个都要根据具体的协议分析情况去实现,注意,生成的用来发送的数据和接收到的需要处理的数据,都需要按十六进制处理,这里不做详述。
线程longlinktcpth2是一个循环,协议不退出,循环不结束,实现如下:
def longlinktcpth2(self):
tmalive = 0;
r_inputs = set()
r_inputs.add(self.sockmain)
w_inputs = set()
w_inputs.add(self.sockmain)
e_inputs = set()
e_inputs.add(self.sockmain)
tm=int(round(time.time()))
self.savelogs('longlinktcpth2', 'enter' )
while (self.quitflag==0):
try:
r_list, w_list, e_list = select.select(r_inputs, w_inputs, e_inputs, 1)
for event in r_list:
try:
buf = event.recv(2048)
sz = len(buf)
self.savelogs('longlinktcpth2', "loop recv data len:"+ str(sz) )
if sz > 0:
self.databuf += buf
self.dealdatabuf()
alivemsg = genbaseinfo.genalive()
self.sockmain.sendall(alivemsg)
self.savelogs('longlinktcpth2', "sendalive")
else:
self.savelogs('longlinktcpth2', "远程断开连接,do reconnect")
r_inputs.clear()
time.sleep(3)
self.sockmain = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.longlinktcpstart2()
r_inputs = set()
r_inputs.add(self.sockmain)
w_inputs = set()
w_inputs.add(self.sockmain)
e_inputs = set()
e_inputs.add(self.sockmain)
except Exception as e:
self.savelogs('longlinktcpth2', str(e))
self.threadLock.acquire()
if (len(self.msglist) > 0):
msg = self.msglist.pop(0)
self.threadLock.release()
self.sockmain.sendall(msg)
self.savelogs('longlinktcpth2',"send a msg")
else:
self.threadLock.release()
tmnow=int(round(time.time()))
if tmnow-tm>30:
message = genbaseinfo.genotherdata()
if len(message) == 0:
self.savelogs('longlinktcpth2', 'genalivedata error ')
return False
self.sockmain.sendall(message)
self.savelogs('longlinktcpth2', "send alivemsg"+str(self.longlinkcnt))
self.longlinkcnt = self.longlinkcnt + 1 #这个要一条连接统一,不能乱,回头加锁
tm=tmnow
if len(w_list) > 0: # 产生了可写的事件,即连接完成
self.savelogs('longlinktcpth2',str(w_list))
w_inputs.clear() # 当连接完成之后,清除掉完成连接的socket
if len(e_list) > 0: # 产生了错误的事件,即连接错误
self.savelogs('longlinktcpth2', str(e_list))
e_inputs.clear() # 当连接有错误发生时,清除掉发生错误的socket
except OSError as e:
self.savelogs('longlinktcpth2', 'socket error,do reconnect')
time.sleep(3)
self.sockmain = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.longlinktcpstart2()
r_inputs = set()
r_inputs.add(self.sockmain)
w_inputs = set()
w_inputs.add(self.sockmain)
e_inputs = set()
e_inputs.add(self.sockmain)
self.savelogs('longlinktcpth2', 'leave')
由于这个代码主要是在windows上使用,因此,longlinktcpth2线程采用了select来实现,而没有使用epoll。在循环中,对异常进行了处理,如果发生异常,连接被断开,则调用longlinktcpstart2重新连接,而不退出循环,其余的和longlinktcpstart2里面一致。
由于TCP连接是流的概念,因此,需要对数据进行缓存拼接,这就是上面代码中databuf的作用,防止每次收到的数据不完整或者太多,方便后续的处理,这才是一个合格的码农的信仰的自我升华。
来源:https://blog.csdn.net/yeyiqun/article/details/122852505
猜你喜欢
- DW2004的中文乱码情况你遇到过么?乱码一般是怎么出现的呢?也许很多时候用其他软件(比如Editplus)写程序的时候,忘了meta标签里
- <?php /** * Created by JetBrains Ph
- 一、项目工程目录:二、具体工程文件代码:1、新建一个包名:common(用于存放基本函数封装)(1)在common包下新建一个base.py
- 本文实例为大家分享了Python实现滑雪小游戏的具体代码,供大家参考,具体内容如下源码分享:import sysimport cfgimpo
- Django 分页功能的实现,供大家参考,具体内容如下创建项目创建APP,添加APP这些就不在多说我们这次重点来看到视图函数下面是路由设置视
- urllib3是一款Python 3的HTTP客户端。Python标准库提供了urllib。在Python 2中,另外提供了urllib2;
- try { int readByte = 0;  
- 写在前面:前一段时间 kejun 给我们培训JavaScript的时候,在幻灯片上推荐了很多特别经典的文章,其中就有这一篇。读过之后感觉很不
- 本文实例讲述了python有证书的加密解密实现方法。分享给大家供大家参考。具体实现方法如下:最近在做python的加解密工作,同时加完密的串
- 关于list的insert函数list#insert(ind,value)在ind元素前面插入value首先对ind进行预处理:如果ind&
- Python 模块安装一. 打开命令提示符win + R 输入 cmd 点击确定或者win + S 搜索输入 cmd二. 环境变量没有问题的
- 0. 学习目标在顺序存储方式中,根据数据元素的序号就可随机存取表中任何一个元素,但同时在插入和删除运算需要移动大量的元素,造成算法效率较低。
- 问题描述如下:解决方案如下:下图中字体调整为18及以上效果如下:来源:https://blog.csdn.net/appleyuchi/ar
- unittest是python的一个单元测试框架关于断言它是用于对一个确定结果和预测结果的一种判断,如果结果正确无任何返回效果,如果结果错误
- 今天 大白 问了一个关于CSS权重的问题:关于选择器权重的问题 。class的权重是10 标签权重是 1 。比如说 p span{} 权重是
- 抓取“xmly”鬼故事音频import json # 在这个url,音频链接为JSON动态生成,所以用到了json模块impor
- 以下为引用的内容:DROP PROCEDURE test_insert ;DELIMITER ;;CREATE PROCEDURE test
- 这个效果前些日子有人在论坛问起,今天有空研究了下。我打了原创标记,因为我在写这个方法时的确没有参考过别人的代码。我的方法其实就是层遮罩加AL
- 本文实例为大家分享了python实现简易五子棋游戏的具体代码,供大家参考,具体内容如下运行效果: 完整代码+注释: fi
- 本节笔者主将要介绍Active Server Page的另一对象Response,Response对象的作用是在向浏览器发送数据。将Resp