murmur/tt-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

使用black格式化

Browse Source
This commit is contained in:
murmur 2023-03-11 21:25:34 +08:00
parent 0212f884c6
commit ac02bc50db
1 changed files with 79 additions and 76 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

149
server.py
View File

@ -1,50 +1,52 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
#coding:utf-8 # coding:utf-8
import socketserver import socketserver
import subprocess import subprocess
import pickle import pickle
offs = 0 # 偏移量
offs = 0 #偏移量 p = {}
p ={} # 保存终端信息，ccid与地址对
#保存终端信息，ccid与地址对 # p={ccid=addr,}
#p={ccid=addr,} # 缓存未能正常发送消息的客户端
#缓存未能正常发送消息的客户端 pp = {}
pp={}
# cache={} # cache={}
from collections import defaultdict from collections import defaultdict
cache=defaultdict(list)
cache = defaultdict(list)
def save_cache(dic): def save_cache(dic):
'''缓存未发送成功的消息 """缓存未发送成功的消息
仅在未发送成功时缓存''' 仅在未发送成功时缓存"""
# f=open("./cache.txt",mode='w+') # f=open("./cache.txt",mode='w+')
# f.write(str(dic)) # f.write(str(dic))
# f.close # f.close
pickle.dump(dic,open("./cache.txt",'wb')) pickle.dump(dic, open("./cache.txt", "wb"))
def load_cache(): def load_cache():
'''程序启动时加载缓存''' """程序启动时加载缓存"""
try: try:
# f=open("./cache.txt") # f=open("./cache.txt")
# c = eval(f.read()) # c = eval(f.read())
# f.close() # f.close()
c=pickle.load(open("./cache.txt",'rb')) c = pickle.load(open("./cache.txt", "rb"))
return c return c
except: except:
return None return None
def update_pairs(addr,ccid,clear=0):
'''更新终端库 def update_pairs(addr, ccid, clear=0):
"""更新终端库
addr:绑定tcp地址 addr:绑定tcp地址
ccid:终端号 ccid:终端号
clear:清空掉线终端''' clear:清空掉线终端"""
if ccid and not clear: if ccid and not clear:
#添加或更新客户端 # 添加或更新客户端
p[ccid] = addr p[ccid] = addr
pp[addr] = ccid pp[addr] = ccid
else: else:
@ -53,29 +55,32 @@ def update_pairs(addr,ccid,clear=0):
# 打印在线终端 # 打印在线终端
for i in p: for i in p:
print(i,"<--->",p[i]) print(i, "<--->", p[i])
# for i in pp: # for i in pp:
# print(i,"-",pp[i]) # print(i,"-",pp[i])
def get_addr(ccid): def get_addr(ccid):
'''获取ccid对应的地址''' """获取ccid对应的地址"""
try: try:
# 如果有直接返回 # 如果有直接返回
return p[ccid] return p[ccid]
except Exception: except Exception:
#没有就返回None # 没有就返回None
return None return None
def get_ccid(addr): def get_ccid(addr):
try: try:
# 如果有直接返回 # 如果有直接返回
return pp[addr] return pp[addr]
except Exception: except Exception:
#没有就返回None # 没有就返回None
return None return None
'''
"""
+--------+--------------+----------+----------+--------------+------------------------+--------+ +--------+--------------+----------+----------+--------------+------------------------+--------+
| ff | 长度 | 地址信息 | 信令 | 内容 | 校验位 | | ff | 长度 | 地址信息 | 信令 | 内容 | 校验位 |
+--------+--------------+----------+----------+--------------+------------------------+--------+ +--------+--------------+----------+----------+--------------+------------------------+--------+
@ -83,21 +88,23 @@ def get_ccid(addr):
+ $TTMS +--------------+----------+----------+--------------+------------------------+--------+ + $TTMS +--------------+----------+----------+--------------+------------------------+--------+
| | 2Bytes | 4Bytes | 传输数据 | 0xaa | 载荷 | 2Bytes | | | 2Bytes | 4Bytes | 传输数据 | 0xaa | 载荷 | 2Bytes |
+--------+--------------+----------+----------+--------------+------------------------+--------+ +--------+--------------+----------+----------+--------------+------------------------+--------+
''' """
def send(fd,data):
def send(fd, data):
try: try:
fd.send(data) fd.send(data)
return 0 return 0
except: except:
return -1 return -1
def tt_hh(addr,data):
'''处理心跳''' def tt_hh(addr, data):
"""处理心跳"""
# 更新pairs # 更新pairs
ccid = data[7+offs:11+offs] ccid = data[7 + offs : 11 + offs]
update_pairs(addr,ccid) update_pairs(addr, ccid)
# 组帧再返回 # 组帧再返回
# 从缓存中匹配数据并重发 # 从缓存中匹配数据并重发
@ -112,42 +119,41 @@ def tt_hh(addr,data):
# 异常时退出循环，不用再尝试发送后续msgs # 异常时退出循环，不用再尝试发送后续msgs
# 退出前清空已发送成功的缓存数据 # 退出前清空已发送成功的缓存数据
# del方法会改变list大小 # del方法会改变list大小
while(i): while i:
cache[ccid].pop(0) cache[ccid].pop(0)
i = i-1 i = i - 1
break break
return 0 return 0
def tt_trans(addr,data):
'''发送数据到指定ccid'''
tccid = data[7+offs:11+offs] def tt_trans(addr, data):
"""发送数据到指定ccid"""
tccid = data[7 + offs : 11 + offs]
taddr = get_addr(tccid) taddr = get_addr(tccid)
# 组帧 # 组帧
msg = bytearray(data[:]) msg = bytearray(data[:])
msg[:5]=b'$TTMX' msg[:5] = b"$TTMX"
msg[7+offs:11+offs] = get_ccid(addr) msg[7 + offs : 11 + offs] = get_ccid(addr)
# 计算校验和 # 计算校验和
crc = 0 crc = 0
for i in msg[:-1]: for i in msg[:-1]:
crc = crc ^ i crc = crc ^ i
msg[-1]=crc msg[-1] = crc
if not taddr: if not taddr:
# 未找到匹配的在线终端 # 未找到匹配的在线终端
#更新缓存 # 更新缓存
#{tccid1=[msg1,msg2,...],...} # {tccid1=[msg1,msg2,...],...}
cache[tccid].append(msg) cache[tccid].append(msg)
save_cache(cache) save_cache(cache)
print("终端未在线或ccid错。") print("终端未在线或ccid错。")
return 0 return 0
print("匹配终端：",tccid,"<--->",taddr) print("匹配终端：", tccid, "<--->", taddr)
# 发送 # 发送
try: try:
@ -156,52 +162,50 @@ def tt_trans(addr,data):
# 发送失败 # 发送失败
taddr.close() taddr.close()
# 更新pairs，清空对应终端 # 更新pairs，清空对应终端
update_pairs(taddr,tccid,1) update_pairs(taddr, tccid, 1)
#更新缓存 # 更新缓存
#{tccid1=[msg1,msg2,...],...} # {tccid1=[msg1,msg2,...],...}
cache[tccid].append(msg) cache[tccid].append(msg)
save_cache(cache) save_cache(cache)
print("发送失败，终端可能已掉线。",taddr) print("发送失败，终端可能已掉线。", taddr)
return 0 return 0
def err_handle(flag,addr):
'''错误处理 def err_handle(flag, addr):
"""错误处理
默认打印错误信息 默认打印错误信息
1-非法连接，关闭连接 1-非法连接，关闭连接
''' """
return 0 return 0
def tt_decode(addr, data):
def tt_decode(addr,data): """处理收到的数据
'''处理收到的数据
数据正常返回0 数据正常返回0
数据异常返回-1''' 数据异常返回-1"""
if data[:5] != b"$TTMS" or len(data) < 12 or len(data) > 200:
if data[:5] != b'$TTMS' or len(data) <12 or len(data)>200:
# 数据长度不可能小于12，考虑非法接入的可能，主动断开连接 # 数据长度不可能小于12，考虑非法接入的可能，主动断开连接
return -1 return -1
cmd = data[11 + offs]
cmd = data[11+offs] if cmd == 0x01:
if cmd==0x01:
# 处理心跳 # 处理心跳
tt_hh(addr,data) tt_hh(addr, data)
elif cmd==0xaa: elif cmd == 0xAA:
# 处理数据 # 处理数据
tt_trans(addr,data) tt_trans(addr, data)
else: else:
return -1 return -1
class MyServer(socketserver.BaseRequestHandler): class MyServer(socketserver.BaseRequestHandler):
def handle(self):#回调 def handle(self): # 回调
print ("终端已上线:",self.client_address) print("终端已上线:", self.client_address)
conn = self.request conn = self.request
# print(type(conn),conn.fd) # print(type(conn),conn.fd)
@ -210,23 +214,22 @@ class MyServer(socketserver.BaseRequestHandler):
# update_pairs(conn,data[7+offs:11+offs]) # update_pairs(conn,data[7+offs:11+offs])
if not data: if not data:
break break
print ("接收到新数据",self.client_address,"，长度",len(data),"：\r\n", data.hex(" ")) print(
#ack_msg = "got from "+ str(ip) + " to " + str(self.client_address) + data "接收到新数据", self.client_address, "，长度", len(data), "：\r\n", data.hex(" ")
)
# ack_msg = "got from "+ str(ip) + " to " + str(self.client_address) + data
# conn.send(data[:6]) # conn.send(data[:6])
if tt_decode(conn,data) : if tt_decode(conn, data):
conn.close() conn.close()
print(self.client_address,"疑是非法连接，已切断。") print(self.client_address, "疑是非法连接，已切断。")
break break
if __name__ == '__main__': if __name__ == "__main__":
# load_cache # load_cache
cache = load_cache() cache = load_cache()
server = socketserver.ThreadingTCPServer(('localhost',8005), MyServer) server = socketserver.ThreadingTCPServer(("localhost", 8005), MyServer)
ip, port = server.server_address ip, port = server.server_address
print ("服务端已建立：",ip, port) print("服务端已建立：", ip, port)
server.serve_forever() server.serve_forever()