-
Notifications
You must be signed in to change notification settings - Fork 50
/
Copy pathudp_Logic.py
154 lines (141 loc) · 5.38 KB
/
udp_Logic.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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2019/1/11 15:48
# @Author : SeniorZhu1994
# @Site :
# @File : udp_Logic.py
# @Software: PyCharm
from PyQt5.QtWidgets import QMessageBox
from tcp_udp_ui import Tcp_ucpUi
import socket
import threading
import stopThreading
class UdpLogic(Tcp_ucpUi):
def __init__(self):
super(UdpLogic, self).__init__()
self.us = None # us代表udp socket
self.us_th = None
self.link = False # 初始化连接状态为False
self.working = False # 初始化工作状态为False
def socket_open_udp(self):
"""
功能函数，UDPClient开启的方法
:return: None
"""
self.open_btn.setEnabled(False)
self.working = True
lo_ip = self.Localip_lineedit.text()
lo_port = self.Localport_lineedit.text()
lo_ip_port = (lo_ip, int(lo_port))
self.BUFSIZE = 1024
self.us = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
try:
self.us.bind(lo_ip_port) # 绑定地址
except Exception as ret:
print('Error:', ret)
QMessageBox.critical(self, '错误', '端口已被占用')
# 关闭udp socket
self.socket_close_u()
else:
print('UDP establishing...')
self.us_th = threading.Thread(target=self.udp_client_concurrency)
self.us_th.setDaemon(True)
self.us_th.start()
self.link = True
def udp_client_concurrency(self):
"""
创建新线程以供UDPClient持续监听Server的消息
:return:
"""
show_client_info = True
while True:
try:
recv_msg, self.addr = self.us.recvfrom(self.BUFSIZE)
except Exception as ret:
pass
else:
statusbar_client_info = '%s:%d' % (self.addr[0], self.addr[1])
connect_info = '[Remote IP %s Port: %s ]' % (
self.addr[0], self.addr[1])
print(self.addr, type(self.addr))
if show_client_info is True:
# 状态栏显示客户端连接成功信息
self.signal_status_connected.emit(statusbar_client_info)
self.signal_add_clientstatus_info.emit(connect_info)
show_client_info = False
# 判断是否以16进制显示并处理
self.if_hex_show_tcpc_udp(recv_msg)
# 将接收到的数据字节数显示在状态栏的计数区域
self.rx_count += len(recv_msg)
self.statusbar_dict['rx'].setText('接收计数：%s' % self.rx_count)
def socket_close_u(self):
'''
关闭udpsocket以及其相关线程
:return:
'''
try:
self.us.close()
self.working = False
self.open_btn.setEnabled(True)
print('UDP closed...')
self.open_btn.setEnabled(True)
except Exception as ret:
pass
try:
stopThreading.stop_thread(self.us_th)
except Exception as ret:
pass
def data_send_u(self):
"""
用于UDP发送消息
:return:
"""
remote_ip = self.remoteip_text.text()
remote_port = self.remoteport_text.text()
try:
self.remote_ip_port = (remote_ip, int(remote_port))
except Exception as e:
QMessageBox.critical(self, '错误', '请填写正确的远程主机IP和端口号')
else:
if self.working is False:
QMessageBox.critical(self, '警告', '请先设置UDP网络')
else:
if self.link:
get_msg = self.DataSendtext.toPlainText() # 从发送区获取数据
# 判断附加为功能是否勾选并进行后续处理
get_msg = self.is_sendcheck_send(get_msg)
# 判断是否是16进制发送
send_msg = self.if_hex_send(get_msg)
print(send_msg)
if get_msg:
try:
self.us.sendto(send_msg, self.remote_ip_port)
self.tx_count += len(send_msg)
self.statusbar_dict['tx'].setText(
'发送计数：%s' % self.tx_count)
except Exception as ret:
pass
else:
QMessageBox.critical(self, '警告', '发送不可为空')
else:
QMessageBox.critical(self, '警告', '当前无任何连接')
def file_send_u(self):
"""
用于UDP发送文件
:return:
"""
if self.working is False:
QMessageBox.critical(self, '警告', '请先设置UDP网络')
else:
if self.link:
if self.file_load.isChecked():
send_msg = self.f_data
else:
send_msg = b''
# print(send_msg)
if send_msg != b'':
self.us.sendto(send_msg, self.remote_ip_port)
else:
QMessageBox.critical(self, '警告', '发送不可为空')
else:
QMessageBox.critical(self, '警告', '当前无任何连接')