def __init__(self):
super().__init__()
super(QtWidgets.QMainWindow, self).__init__()
self.status_message = ['● Idle', '● Idle', '● Idle', '']
config_file = Path('config.json')
# config_file = open('config.json', 'w+')
if config_file.exists():
self.config = json.load(open('config.json', 'r'))
else:
self.config = dict()
json.dump(self.config, open('config.json', 'w+'))
"""Load UI"""
self.ui = uic.loadUi('mainwindow.ui', self)
self.init_ui()
self.ui.comboBox_Interface.currentIndexChanged.connect(
self.on_interface_selection_change)
self.ui.button_Refresh.clicked.connect(self.on_refresh_button_clicked)
self.ui.button_TcpClient.clicked.connect(
self.on_tcp_client_connect_button_clicked
)
self.ui.button_TcpClientSend.clicked.connect(
self.on_tcp_client_message_send
)
self.ui.lineEdit_TcpClientSend.returnPressed.connect(
self.on_tcp_client_message_send
)
self.ui.button_TcpServer.clicked.connect(
self.on_tcp_server_start_stop_button_clicked)
self.ui.button_TcpServerSend.clicked.connect(
self.on_tcp_server_message_send
)
self.ui.lineEdit_TcpServerSend.returnPressed.connect(
self.on_tcp_server_message_send
)
self.ui.button_Udp.clicked.connect(
self.on_udp_server_start_stop_button_clicked
)
self.ui.button_UdpSend.clicked.connect(
self.on_udp_message_send
)
self.ui.lineEdit_UdpSend.returnPressed.connect(
self.on_udp_message_send
)
self.udp_send = UDPServer(
'0.0.0.0',
1234)
self.ui.tabWidget.currentChanged.connect(
self.on_tab_changed
)
def on_udp_server_start_stop_button_clicked(self):
if self.ui.button_Udp.text() == 'Start':
self.ui.button_Udp.setEnabled(False)
self.ui.lineEdit_UdpListenPort.setEnabled(False)
self.udp_thread = QThread()
self.udp_server = UDPServer(
self.ui.label_LocalIP.text(),
int(self.ui.lineEdit_UdpListenPort.text()))
self.udp_thread.started.connect(self.udp_server.start)
self.udp_server.status.connect(self.on_udp_server_status_update)
self.udp_server.message.connect(self.on_udp_server_message_ready)
self.udp_server.moveToThread(self.udp_thread)
self.udp_thread.start()
self.config['UDP_Listen_Port'] = self.ui.lineEdit_UdpListenPort.text()
self.save_config()
elif self.ui.button_Udp.text() == 'Stop':
self.ui.button_Udp.setEnabled(False)
self.udp_server.close()
def __init__(self, address):
self.msgQ = Queue.Queue()
self.server = UDPServer(address, self.msgQ)
thread.start_new_thread(self.server.serve_forever, ())
thread.start_new_thread(self.server.keepAlive, ())
self.sim = ServerSimulation()
self.sim.verbose = True
thread.start_new_thread(self.sim.simulate, ())
self.players = {}
self.alive=True
self.lastPing = 0
thread.start_new_thread(self.serve, ())
class Server():
MSGS_PER_SECOND = 20
META_PERIOD = 1000
DROP_PERIOD = 300
def __init__(self, address):
self.msgQ = Queue.Queue()
self.server = UDPServer(address, self.msgQ)
thread.start_new_thread(self.server.serve_forever, ())
thread.start_new_thread(self.server.keepAlive, ())
self.sim = ServerSimulation()
self.sim.verbose = True
thread.start_new_thread(self.sim.simulate, ())
self.players = {}
self.alive=True
self.lastPing = 0
thread.start_new_thread(self.serve, ())
def kill(self):
self.server.shutdown()
self.sim.alive = False
self.alive= False
def broadcastState(self):
clock = pygame.time.Clock()
timer = 0
while self.alive:
timer += clock.tick(self.MSGS_PER_SECOND)
self.server.sendToAll(self.sim.getWorldState().toString())
if timer > self.META_PERIOD:
timer -= self.META_PERIOD
self.server.sendToAll(self.sim.getMeta().toString())
msg = PingMessage()
self.lastPing = pygame.time.get_ticks()
msg.timestamp = self.lastPing
self.server.sendToAll(msg.toString())
def serve(self):
while self.alive:
udpMsg = self.msgQ.get()
if udpMsg.data == '':
self.sim.removePlayer(self.players[udpMsg.connID])
continue
msg = Message.getMessage(udpMsg.data)
if msg.type == Message.CONNECT:
self.players[udpMsg.connID] = self.sim.addPlayer(msg)
self.server.send(ConnectMessage(str(self.players[udpMsg.connID])).toString(),
udpMsg.connID)
elif (pygame.time.get_ticks() - msg.timestamp > self.DROP_PERIOD):
continue
elif msg.type == Message.CHAT:
self.server.sendToAll(udpMsg.data)
elif msg.type == Message.INPUT:
if (udpMsg.connID in self.players):
self.sim.receiveInput(self.players[udpMsg.connID], msg)
elif msg.type == Message.PING:
if (udpMsg.connID in self.players):
self.sim.updateLatency(self.players[udpMsg.connID],
(pygame.time.get_ticks() - self.lastPing)/2)
def server():
TCPServer()
UDPServer()
class MyApp(QtWidgets.QMainWindow):
def __init__(self):
super().__init__()
super(QtWidgets.QMainWindow, self).__init__()
self.status_message = ['● Idle', '● Idle', '● Idle', '']
config_file = Path('config.json')
# config_file = open('config.json', 'w+')
if config_file.exists():
self.config = json.load(open('config.json', 'r'))
else:
self.config = dict()
json.dump(self.config, open('config.json', 'w+'))
"""Load UI"""
self.ui = uic.loadUi('mainwindow.ui', self)
self.init_ui()
self.ui.comboBox_Interface.currentIndexChanged.connect(
self.on_interface_selection_change)
self.ui.button_Refresh.clicked.connect(self.on_refresh_button_clicked)
self.ui.button_TcpClient.clicked.connect(
self.on_tcp_client_connect_button_clicked
)
self.ui.button_TcpClientSend.clicked.connect(
self.on_tcp_client_message_send
)
self.ui.lineEdit_TcpClientSend.returnPressed.connect(
self.on_tcp_client_message_send
)
self.ui.button_TcpServer.clicked.connect(
self.on_tcp_server_start_stop_button_clicked)
self.ui.button_TcpServerSend.clicked.connect(
self.on_tcp_server_message_send
)
self.ui.lineEdit_TcpServerSend.returnPressed.connect(
self.on_tcp_server_message_send
)
self.ui.button_Udp.clicked.connect(
self.on_udp_server_start_stop_button_clicked
)
self.ui.button_UdpSend.clicked.connect(
self.on_udp_message_send
)
self.ui.lineEdit_UdpSend.returnPressed.connect(
self.on_udp_message_send
)
self.udp_send = UDPServer(
'0.0.0.0',
1234)
self.ui.tabWidget.currentChanged.connect(
self.on_tab_changed
)
def save_config(self):
try:
json.dump(self.config, open('config.json', 'w+'))
except PermissionError as err:
pass
def init_ui(self):
# Interface
self.update_network_interfaces()
self.ui.tabWidget.setCurrentIndex(self.config.get('Tab_Index', 0))
self.on_tab_changed(self.config.get('Tab_Index', 0))
# TCP Client
self.ui.textBrowser_TcpClientMessage.setEnabled(False)
self.ui.lineEdit_TcpClientSend.setEnabled(False)
self.ui.button_TcpClientSend.setEnabled(False)
tcp_client_ip = self.config.get('TCP_Client_IP', '127.0.0.1')
tcp_client_port = self.config.get('TCP_Client_Port', '1234')
self.ui.lineEdit_TcpClientTargetIP.setText(tcp_client_ip)
self.ui.lineEdit_TcpClientTargetPort.setText(tcp_client_port)
# TCP Server
self.ui.textBrowser_TcpServerMessage.setEnabled(False)
self.ui.lineEdit_TcpServerSend.setEnabled(False)
self.ui.button_TcpServerSend.setEnabled(False)
tcp_server_port = self.config.get('TCP_Server_Port', '1234')
self.ui.lineEdit_TcpServerListenPort.setText(tcp_server_port)
# UDP
udp_listen_port = self.config.get('UDP_Listen_Port', '1234')
udp_target_ip = self.config.get('UDP_Target_IP', '127.0.0.1')
udp_target_port = self.config.get('UDP_Target_Port', '1234')
self.ui.lineEdit_UdpListenPort.setText(udp_listen_port)
self.ui.lineEdit_UdpTargetIP.setText(udp_target_ip)
self.ui.lineEdit_UdpTargetPort.setText(udp_target_port)
def update_network_interfaces(self):
self.ui.comboBox_Interface.clear()
self.net_if = psutil.net_if_addrs()
net_if_stats = psutil.net_if_stats()
net_names = list(self.net_if.keys())
for if_name in net_names:
if not net_if_stats[if_name].isup:
self.net_if.pop(if_name, None)
else:
self.ui.comboBox_Interface.addItem(if_name)
interface_idx = self.config.get('Interface', 0)
if interface_idx >= self.ui.comboBox_Interface.count():
self.ui.comboBox_Interface.setCurrentIndex(0)
else:
self.ui.comboBox_Interface.setCurrentIndex(interface_idx)
current_interface = self.ui.comboBox_Interface.currentText()
self.config['Interface'] = self.ui.comboBox_Interface.currentIndex()
for snicaddr in self.net_if[current_interface]:
if snicaddr.family == socket.AF_INET:
ipv4_add = snicaddr.address
break
else:
ipv4_add = '0.0.0.0'
self.ui.label_LocalIP.setText(ipv4_add)
self.save_config()
def on_interface_selection_change(self):
current_interface = self.ui.comboBox_Interface.currentText()
if current_interface in self.net_if:
for snicaddr in self.net_if[current_interface]:
if snicaddr.family == socket.AF_INET:
ipv4_add = snicaddr.address
break
else:
ipv4_add = '0.0.0.0'
else:
return
self.ui.label_LocalIP.setText(ipv4_add)
self.config['Interface'] = self.ui.comboBox_Interface.currentIndex()
self.save_config()
def on_refresh_button_clicked(self):
self.update_network_interfaces()
# TCP Client
def on_tcp_client_connect_button_clicked(self):
if self.ui.button_TcpClient.text() == 'Connect':
self.ui.button_TcpClient.setEnabled(False)
self.ui.lineEdit_TcpClientTargetIP.setEnabled(False)
self.ui.lineEdit_TcpClientTargetPort.setEnabled(False)
self.tcp_client_thread = QThread()
self.tcp_client = TCPClient(
self.ui.label_LocalIP.text(),
int(self.ui.lineEdit_TcpClientTargetPort.text()))
self.tcp_client_thread.started.connect(self.tcp_client.start)
self.tcp_client.status.connect(self.on_tcp_client_status_update)
self.tcp_client.message.connect(self.on_tcp_client_message_ready)
self.tcp_client.moveToThread(self.tcp_client_thread)
self.tcp_client_thread.start()
self.config['TCP_Client_IP'] = self.ui.label_LocalIP.text()
self.config['TCP_Client_Port'] = self.ui.lineEdit_TcpClientTargetPort.text()
self.save_config()
elif self.ui.button_TcpClient.text() == 'Disconnect':
self.ui.button_TcpClient.setEnabled(False)
self.tcp_client.close()
def on_tcp_client_status_update(self, status, addr):
if status == TCPClient.STOP:
self.tcp_client.status.disconnect()
self.tcp_client.message.disconnect()
self.ui.button_TcpClient.setText('Connect')
self.tcp_client_thread.quit()
self.ui.lineEdit_TcpClientTargetIP.setEnabled(True)
self.ui.lineEdit_TcpClientTargetPort.setEnabled(True)
self.ui.textBrowser_TcpClientMessage.setEnabled(False)
self.ui.lineEdit_TcpClientSend.setEnabled(False)
self.ui.button_TcpClientSend.setEnabled(False)
self.status_message[0] = '● Idle'
if self.ui.tabWidget.currentIndex() == 0:
self.on_tab_changed(0)
elif status == TCPClient.CONNECTED:
self.ui.button_TcpClient.setText('Disconnect')
self.ui.textBrowser_TcpClientMessage.setEnabled(True)
self.ui.lineEdit_TcpClientSend.setEnabled(True)
self.ui.button_TcpClientSend.setEnabled(True)
self.status_message[0] = '● Connected to ' +\
self.ui.label_LocalIP.text() +\
':'+self.ui.lineEdit_TcpClientTargetPort.text()
if self.ui.tabWidget.currentIndex() == 0:
self.on_tab_changed(0)
self.ui.button_TcpClient.setEnabled(True)
def on_tcp_client_message_ready(self, source, msg):
self.ui.textBrowser_TcpClientMessage.append(
'<p style="text-align: center;"><span style="color: #2196F3;"><strong>----- ' +
source +
' -----</strong></span></p>')
self.ui.textBrowser_TcpClientMessage.append(
'<p style="text-align: center;"><span style="color: #2196F3;">' +
msg +
'</span></p>')
def on_tcp_client_message_send(self):
self.tcp_client.send(self.ui.lineEdit_TcpClientSend.text())
self.ui.textBrowser_TcpClientMessage.append(
'<p style="text-align: center;"><strong>----- ' +
'this' +
' -----</strong></p>')
self.ui.textBrowser_TcpClientMessage.append(
'<p style="text-align: center;">' +
self.ui.lineEdit_TcpClientSend.text() +
'</p>')
self.ui.lineEdit_TcpClientSend.clear()
# TCP Server
def on_tcp_server_start_stop_button_clicked(self):
if self.ui.button_TcpServer.text() == 'Start':
self.ui.button_TcpServer.setEnabled(False)
self.ui.lineEdit_TcpServerListenPort.setEnabled(False)
self.tcp_server_thread = QThread()
self.tcp_server = TCPServer(
self.ui.label_LocalIP.text(),
int(self.ui.lineEdit_TcpServerListenPort.text()))
self.tcp_server_thread.started.connect(self.tcp_server.start)
self.tcp_server.status.connect(self.on_tcp_server_status_update)
self.tcp_server.message.connect(self.on_tcp_server_message_ready)
self.tcp_server.moveToThread(self.tcp_server_thread)
self.tcp_server_thread.start()
self.config['TCP_Server_Port'] = self.ui.lineEdit_TcpServerListenPort.text()
self.save_config()
elif self.ui.button_TcpServer.text() == 'Stop':
self.ui.button_TcpServer.setEnabled(False)
self.tcp_server.close()
elif self.ui.button_TcpServer.text() == 'Disconnect':
self.ui.button_TcpServer.setEnabled(False)
self.tcp_server.disconnect()
def on_tcp_server_status_update(self, status, addr):
if status == TCPServer.STOP:
self.tcp_server.status.disconnect()
self.tcp_server.message.disconnect()
self.ui.button_TcpServer.setText('Start')
self.tcp_server_thread.quit()
self.ui.textBrowser_TcpServerMessage.setEnabled(False)
self.ui.lineEdit_TcpServerSend.setEnabled(False)
self.ui.button_TcpServerSend.setEnabled(False)
self.ui.lineEdit_TcpServerListenPort.setEnabled(True)
self.status_message[1] = '● Idle'
if self.ui.tabWidget.currentIndex() == 1:
self.on_tab_changed(1)
elif status == TCPServer.LISTEN:
self.ui.button_TcpServer.setText('Stop')
self.ui.textBrowser_TcpServerMessage.setEnabled(False)
self.ui.lineEdit_TcpServerSend.setEnabled(False)
self.ui.button_TcpServerSend.setEnabled(False)
self.status_message[1] = '● Listen on ' +\
self.ui.label_LocalIP.text()+':' +\
self.ui.lineEdit_TcpServerListenPort.text()
if self.ui.tabWidget.currentIndex() == 1:
self.on_tab_changed(1)
elif status == TCPServer.CONNECTED:
self.ui.button_TcpServer.setText('Disconnect')
self.ui.textBrowser_TcpServerMessage.setEnabled(True)
self.ui.lineEdit_TcpServerSend.setEnabled(True)
self.ui.button_TcpServerSend.setEnabled(True)
self.status_message[1] = '● Connected to '+addr
if self.ui.tabWidget.currentIndex() == 1:
self.on_tab_changed(1)
# self.tcp_server.send('Hello World')
self.ui.button_TcpServer.setEnabled(True)
def on_tcp_server_message_ready(self, source, msg):
self.ui.textBrowser_TcpServerMessage.append(
'<p style="text-align: center;"><span style="color: #2196F3;"><strong>----- ' +
source +
' -----</strong></span></p>')
self.ui.textBrowser_TcpServerMessage.append(
'<p style="text-align: center;"><span style="color: #2196F3;">' +
msg +
'</span></p>')
def on_tcp_server_message_send(self):
self.tcp_server.send(self.ui.lineEdit_TcpServerSend.text())
self.ui.textBrowser_TcpServerMessage.append(
'<p style="text-align: center;"><strong>----- ' +
'this' +
' -----</strong></p>')
self.ui.textBrowser_TcpServerMessage.append(
'<p style="text-align: center;">' +
self.ui.lineEdit_TcpServerSend.text() +
'</p>')
self.ui.lineEdit_TcpServerSend.clear()
# UDP
def on_udp_server_start_stop_button_clicked(self):
if self.ui.button_Udp.text() == 'Start':
self.ui.button_Udp.setEnabled(False)
self.ui.lineEdit_UdpListenPort.setEnabled(False)
self.udp_thread = QThread()
self.udp_server = UDPServer(
self.ui.label_LocalIP.text(),
int(self.ui.lineEdit_UdpListenPort.text()))
self.udp_thread.started.connect(self.udp_server.start)
self.udp_server.status.connect(self.on_udp_server_status_update)
self.udp_server.message.connect(self.on_udp_server_message_ready)
self.udp_server.moveToThread(self.udp_thread)
self.udp_thread.start()
self.config['UDP_Listen_Port'] = self.ui.lineEdit_UdpListenPort.text()
self.save_config()
elif self.ui.button_Udp.text() == 'Stop':
self.ui.button_Udp.setEnabled(False)
self.udp_server.close()
def on_udp_server_status_update(self, status, addr):
if status == UDPServer.STOP:
self.udp_server.status.disconnect()
self.udp_server.message.disconnect()
self.ui.button_Udp.setText('Start')
# self.tcp_server_thread.terminate()
self.udp_thread.quit()
self.ui.lineEdit_UdpListenPort.setEnabled(True)
self.status_message[2] = '● Idle'
if self.ui.tabWidget.currentIndex() == 2:
self.on_tab_changed(2)
elif status == UDPServer.LISTEN:
self.ui.button_Udp.setText('Stop')
self.status_message[2] = '● Listen on ' +\
self.ui.label_LocalIP.text()+':' +\
self.ui.lineEdit_TcpServerListenPort.text()
if self.ui.tabWidget.currentIndex() == 2:
self.on_tab_changed(2)
self.ui.button_Udp.setEnabled(True)
def on_udp_server_message_ready(self, source, msg):
self.ui.textBrowser_UdpMessage.append(
'<p style="text-align: center;"><span style="color: #2196F3;"><strong>----- ' +
source +
' -----</strong></span></p>')
self.ui.textBrowser_UdpMessage.append(
'<p style="text-align: center;"><span style="color: #2196F3;">' +
msg +
'</span></p>')
def on_udp_message_send(self):
self.udp_send.send(
self.ui.lineEdit_UdpSend.text(),
self.ui.lineEdit_UdpTargetIP.text(),
int(self.ui.lineEdit_UdpTargetPort.text())
)
self.ui.textBrowser_UdpMessage.append(
'<p style="text-align: center;"><strong>----- ' +
'this' +
' -----</strong></p>')
self.ui.textBrowser_UdpMessage.append(
'<p style="text-align: center;">' +
self.ui.lineEdit_UdpSend.text() +
'</p>')
self.ui.lineEdit_UdpSend.clear()
self.config['UDP_Target_IP'] = self.ui.lineEdit_UdpTargetIP.text()
self.config['UDP_Target_Port'] = self.ui.lineEdit_UdpTargetPort.text()
self.save_config()
def on_tab_changed(self, index):
self.ui.statusBar.clearMessage()
self.ui.statusBar.setStyleSheet('color: green')
self.ui.statusBar.showMessage(self.status_message[index])
self.config['Tab_Index'] = self.ui.tabWidget.currentIndex()
self.save_config()
from udp import UDPServer
import Queue, sys, thread
def printMsgs(queue, server):
while True:
msg = queue.get()
print msg.data
server.send('OK', msg.connID)
if __name__ == "__main__":
if len(sys.argv) < 2:
print 'Usage:script.py ip port'
sys.exit(0)
queue = Queue.Queue()
try:
server = UDPServer((sys.argv[1], int(sys.argv[2])), queue)
thread.start_new_thread(printMsgs, (queue,server,))
thread.start_new_thread(server.keepAlive, ())
server.serve_forever()
except KeyboardInterrupt:
server.shutdown()
def run_tcp_server(tcp_obj):
server.register_instance(tcp_obj)
server.serve_forever()
if __name__ == '__main__':
logging.basicConfig(filename='mp3.log',level=logging.INFO, filemode='w')
os.system("rm sdfs/*")
console = logging.StreamHandler()
console.setLevel(logging.DEBUG)
logging.getLogger('').addHandler(console)
sdfs_master = SDFS_Master()
slave = Slave(logging, sdfs_master)
udpserver = UDPServer(slave)
udpserver.run_server()
slave.run()
cli = CLI(slave, logging)
cli.run()
slave.init_join()
tcpserver = TCPServer(slave, sdfs_master, logging)
run_tcp_server(tcpserver)
def initialize_udp(slave):
udpserver = UDPServer(slave)
return udpserver