76 lines
1.5 KiB
Python
76 lines
1.5 KiB
Python
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
import socket
|
|
|
|
|
|
import prot_codec as pc
|
|
|
|
|
|
|
|
# 定义服务器和客户端
|
|
# device:
|
|
# server
|
|
# client
|
|
# proxy
|
|
# 定义操作类型
|
|
# option:
|
|
# connect
|
|
# disconnect
|
|
# data
|
|
# login
|
|
# close
|
|
|
|
|
|
_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)
|
|
# 只要有一个断了所有的都断开
|
|
def close(self):
|
|
cmd={'device':'proxy','option':'close'}
|
|
data=pc.encode(json.dumps(cmd).encode('utf-8'),b'default')
|
|
# 这里有可能连接已经断开，无法发送
|
|
if(_tcp_client is not None):
|
|
try:
|
|
_tcp_client.send(data)
|
|
except Exception as e:
|
|
print("target_close:",str(e))
|
|
if(_tcp_server is not None):
|
|
try:
|
|
_tcp_server.send(data)
|
|
except Exception as e:
|
|
print("target_close:",str(e))
|
|
|
|
|
|
|