我有一个使用套接字的 Python 基本客户端-服务器脚本。服务器绑定(bind)到特定端口并等待客户端连接。当客户端连接时,会出现 raw_input 提示,该提示将输入的命令发送到服务器上的子进程,并将输出通过管道返回给客户端。 有时,当我从客户端执行命令时,输出将挂起并且不会向我显示 raw_input 提示,直到我按 [enter] 键。 起初我以为这可能是缓冲区问题,但当我使用输出较小的命令(例如“clear”或“ls”等)时,就会发生这种情况。
客户端代码:
import os, sys
import socket
from base64 import *
import time
try:
HOST = sys.argv[1]
PORT = int(sys.argv[2])
except IndexError:
print("You must specify a host IP address and port number!")
print("usage: ./handler_client.py 192.168.1.4 4444")
sys.exit()
socksize = 4096
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
server.connect((HOST, PORT))
print("[+] Connection established!")
print("[+] Type ':help' to view commands.")
except:
print("[!] Connection error!")
sys.exit(2)
while True:
data = server.recv(socksize)
cmd = raw_input(data)
server.sendall(str(cmd))
server.close()
服务器代码:
import os,sys
import socket
import time
from subprocess import Popen,PIPE,STDOUT,call
HOST = ''
PORT = 4444
socksize = 4096
activePID = []
conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
conn.bind((HOST, PORT))
conn.listen(5)
print("Listening on TCP port %s" % PORT)
def reaper():
while activePID:
pid,stat = os.waitpid(0, os.WNOHANG)
if not pid: break
activePID.remove(pid)
def handler(connection):
time.sleep(3)
while True:
cmd = connection.recv(socksize)
proc = Popen(cmd,
shell=True,
stdout=PIPE,
stderr=PIPE,
stdin=PIPE,
)
stdout, stderr = proc.communicate()
if cmd == ":killme":
connection.close()
sys.exit(0)
elif proc:
connection.send( stdout )
connection.send("\nshell => ")
connection.close()
os._exit(0)
def accept():
while 1:
global connection
connection, address = conn.accept()
print "[!] New connection!"
connection.send("\nshell => ")
reaper()
childPid = os.fork() # forks the incoming connection and sends to conn handler
if childPid == 0:
handler(connection)
else:
activePID.append(childPid)
accept()
最佳答案
我看到的问题是客户端中的最后一个循环仅执行一次 server.recv(socksize)
,然后调用 raw_input()
。如果该 recv()
调用未获取服务器在该单个调用中发送的所有数据,那么它也不会收集命令输出后面的提示,因此不会显示下一个提示。未收集的输入将保留在套接字中,直到您输入下一个命令,然后它将被收集并显示。 (原则上,可能需要多次 recv()
调用来耗尽套接字并到达附加提示,而不仅仅是两次调用。)
如果发生这种情况,那么如果该命令发送回超过一个缓冲区的数据 (4KB),或者它以及时间隔的小块形式生成输出,以便服务器端可以传播该数据,那么您就会遇到问题多次发送的数据合并速度不够快,客户端无法在单个 recv()
中收集所有数据。
要解决此问题,您需要让客户端执行尽可能多的 recv()
调用,以完全耗尽套接字。因此,您需要想出一种方法,让客户端知道套接字已耗尽服务器在此交互中要发送的所有内容。
最简单的方法是让服务器将边界标记添加到数据流中,然后让客户端检查这些标记以发现当前交互的最终数据何时被收集。有多种方法可以做到这一点,但我可能会让服务器在发送的每个 block 之前插入一个“这是以下数据 block 的长度”标记,并在最后一个 block 之后发送一个长度为零的标记 block 。
客户端主循环变为:
forever:
read a marker;
if the length carried in the marker is zero then
break;
else
read exactly that many bytes;.
请注意,客户端在对其进行操作之前必须确保 recv()
完整的标记;内容可以以任意大小的 block 形式从流套接字中出来,与将这些内容发送到发送方一侧的套接字的写入大小完全无关。
您可以决定是将标记作为可变长度文本(带有独特的分隔符)还是作为固定长度二进制文件发送(在这种情况下,如果客户端和服务器可以位于不同的系统上,则必须担心字节序问题)。您还可以决定客户端是否应该在每个 block 到达时显示它(显然您不能使用 raw_input()
来做到这一点),或者是否应该收集所有 block 并显示整个内容在收集完最后一 block 之后的一次爆炸中。
关于Python 客户端-服务器脚本挂起,直到我按 [enter],我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/10156095/