我有一个正在使用的python反向外壳,该外壳利用使用TCP的客户端-服务器连接。我现在正在本地Windows机器上测试它们,并且正在利用子进程库来处理命令。客户端应该向服务器发送命令,服务器将使用输出回复。
服务器:
import socket
import subprocess
import os
# Server
# creates TCP socket
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# port and server ip(localhost)
LOCAL_HOST = '127.0.0.1'
PORT = 5565
BUFFER_SIZE = 5000 # size of message
no_char_message = "-1: Please enter a command"
# test connection
print("Server starting up on %s with port number %s" % (LOCAL_HOST, PORT))
# bind socket to ip and port
sock.bind((LOCAL_HOST, PORT))
# listen to socket
sock.listen(1)
# socket will accept connection and client address
print("Waiting for connection") # waiting for connection
connection, address = sock.accept() # accept connection with client address
print("Connected to", address) # connected by address
while True:
command = connection.recv(BUFFER_SIZE) # receive message from client
if not command:
break
if len(command) == 0:
connection.send(str.encode(no_char_message))
if len(command) > 0:
terminal = subprocess.Popen(command[:].decode("utf-8"), shell=True, stdout=subprocess.PIPE,
stderr=subprocess.PIPE, stdin=subprocess.PIPE)
output = terminal.stdout.read() + terminal.stderr.read()
output_as_string = str(output, "utf-8")
connection.send(str.encode(output_as_string))
print(output_as_string)
print("Closing Server")
sock.close()
connection.close()
客户
import socket
# Client
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # creates TCP Socket
# local host and port
LOCAL_HOST = '127.0.0.1'
PORT = 5565
BUFFER_SIZE = 5000 # size of message
# connect socket to ip and port
sock.connect((LOCAL_HOST, PORT))
print("Connected to server\n")
while True:
message = input("Please enter a command:\n") # ask user to input message
if message == 'quit':
break
print("Sending %s" % message)
sock.send(str.encode(message)) # send message
command = str(sock.recv(BUFFER_SIZE), "utf-8") # receive message
print("received %s" % command)
print("closing connection with server")
sock.close()
问题是,当我向服务器发送空消息时,它挂起,只是说在终端中发送,而服务器却什么也没收到。我不确定是什么原因造成的,但是我假设管道被阻塞或者我没有正确处理。
我希望服务器将错误消息返回给客户端,而不是在客户端本身中处理消息错误。
我尝试检查命令的长度是否为0的情况,并通过错误消息对其进行处理,但该命令不起作用并仍然挂起。
例如,当我尝试使用date命令时,该程序似乎也挂起。
通常,如果无法识别,为空或未成功执行命令,该如何处理条件?
TCP没有空消息的概念。 TCP根本没有消息的概念,它只知道字节。因此,如果您使用空字符串调用send,它将仅向服务器发送任何内容(不是空数据包,而是根本没有数据包),这意味着服务器没有任何内容可接收-在等待数据时它仍会阻塞。换句话说:没有空命令,根本就没有注释。