-
Notifications
You must be signed in to change notification settings - Fork 139
/
TcpServer.py
117 lines (97 loc) · 4.08 KB
/
TcpServer.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
import socket
import threading
import json
class TcpServer:
def __init__(self):
self.port=2233#设置端口
self.HEAD_LEN=8
self.tcpServerSocket=socket.socket()#创建socket对象
hostname= socket.gethostname()#获取本地主机名
sysinfo = socket.gethostbyname_ex(hostname)
hostip=sysinfo[2][2]
self.tcpServerSocket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)#让端口可以复用
self.tcpServerSocket.bind((hostip,self.port))#将地址与套接字绑定,且套接字要求是从未被绑定过的
self.tcpServerSocket.listen(5)#代办事件中排队等待connect的最大数目
def set_receive_listener(self,receive_listener):
self.receive_listener = receive_listener
def server(self):
while True:
print("等待连接")
self.clientSocket, addr = self.tcpServerSocket.accept()
print ('连接地址:', addr)
if self.connected_listener:
self.connected_listener()
while True:
try:
head_data=self.clientSocket.recv(self.HEAD_LEN)
if not len(head_data)==8:
print("bad package!head_data len:",head_data)
self.restart()
return
body_len = self.get_length_from_head_data(head_data)
body_data = self.clientSocket.recv(body_len)
if not body_len==len(body_data):
print("bad package!body_len:",body_len)
self.restart()
return
data_type = self.get_type_from_head_data(head_data)
if data_type == 1:#test/json data
text = body_data.decode()
print(text)
if not text:
break
if self.receive_listener:
self.receive_listener(text)
elif data_type == 2:#image data
pass
except ConnectionResetError:
print("ConnectionResetError!")
self.restart()
return
self.clientSocket.close() # 关闭连接
self.tcpServerSocket.close()
def get_length_from_head_data(self,head_data):
if(not len(head_data)==8):
return
ch1 = head_data[4] & 0x00FF;
ch2 = head_data[5] & 0x00FF;
ch3 = head_data[6] & 0x00FF;
ch4 = head_data[7] & 0x00FF;
return ((ch1 << 24) + (ch2 << 16) + (ch3 << 8) + (ch4 << 0));
def get_type_from_head_data(self,head_data):
if(not len(head_data)==8):
return
ch1 = head_data[0] & 0x00FF;
ch2 = head_data[1] & 0x00FF;
ch3 = head_data[2] & 0x00FF;
ch4 = head_data[3] & 0x00FF;
return ((ch1 << 24) + (ch2 << 16) + (ch3 << 8) + (ch4 << 0));
def start(self):
self.server_threading = threading.Thread(target=self.server, args=())
self.server_threading.start()
def restart(self):
self.start()
def send_data(self, data):
try:
self.clientSocket.send(data)
except ConnectionResetError:
print("ConnectionResetError!")
self.restart()
def send_img(self, bytes_data):
data = self.wrapper_data(2,bytes_data)
self.send_data(data)
def send_text(self, text):
data = self.wrapper_data(1,text.encode())
self.send_data(data)
def wrapper_data(self,data_type,body_data):
print("data_type:",data_type)
print("body_data len:",len(body_data))
type_bytes=data_type.to_bytes(4,'big')
print("type_bytes:",type_bytes)
body_len = len(body_data)
body_len_bytes = body_len.to_bytes(4,'big')
print("body_len_bytes:",body_len_bytes)
head_data = type_bytes + body_len_bytes
print("head_data:",head_data)
data = head_data+body_data
return data