58 lines
1.0 KiB
Python
58 lines
1.0 KiB
Python
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
import socket
|
|
|
|
|
|
import prot_codec as pc
|
|
|
|
|
|
|
|
# 定义服务器和客户端
|
|
# device:
|
|
# server
|
|
# client
|
|
# 定义操作类型
|
|
# option:
|
|
# connect
|
|
# disconnect
|
|
# data
|
|
# login
|
|
|
|
|
|
_tcp_server=None
|
|
_tcp_client=None
|
|
|
|
|
|
|
|
class tcp_target(object):
|
|
def __init__(self,tcp_handler:socket):
|
|
self.handler=tcp_handler
|
|
self.state="idle"
|
|
def recv_handler(self,recv_data:bytearray):
|
|
global _tcp_client
|
|
global _tcp_server
|
|
cmd,data=pc.decode(recv_data)
|
|
print(cmd.decode('utf-8'))
|
|
j=json.loads(cmd)
|
|
if(j["device"]=="client"):
|
|
if(j["option"]=="login"):
|
|
if(_tcp_client is not None):
|
|
_tcp_client.close()
|
|
_tcp_client=self.handler
|
|
else:
|
|
if(_tcp_server is not None):
|
|
_tcp_server.send(recv_data)
|
|
if(j["device"]=="server"):
|
|
if(j["option"]=="login"):
|
|
if(_tcp_server is not None):
|
|
_tcp_server.close()
|
|
_tcp_server=self.handler
|
|
else:
|
|
if(_tcp_client is not None):
|
|
_tcp_client.send(recv_data)
|
|
|
|
|