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