Python 通过套接字发送命令

Question

我遇到了一点麻烦。我想创建一个简单的程序，连接到服务器并使用子进程执行命令，然后将结果返回给客户端。这很简单，但我无法让它发挥作用。现在这就是我所拥有的： 客户端：

import sys, socket, subprocess
conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
host = sys.argv[1]
port = int(sys.argv[2])
socksize = 1024
conn.connect((host, port))
while True:
    shell = raw_input("$ ")
    conn.send(shell)
    data = conn.recv(socksize)
    #msglen = len(data)
    output = data
    iotype = subprocess.PIPE
    cmd = ['/bin/sh', '-c', shell]
    proc = subprocess.Popen(cmd, stdout=iotype).wait()
    stdout,stderr = proc.communicate()
    conn.send(stdout)
    print(output)
    if proc.returncode != 0:
        print("Error")

服务器：

import sys, socket, subprocess
host = ''               
port = 50106
socksize = 1024

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind((host, port))
print("Server started on port: %s" %port)
s.listen(1)
print("Now listening...\n")
conn, addr = s.accept()
while True:
    print 'New connection from %s:%d' % (addr[0], addr[1])
    data = conn.recv(socksize)
    cmd = ['/bin/sh', '-c', data]
    proc = subprocess.Popen(cmd, stdout=subprocess.PIPE).wait()
    stdout,stderr = cmd.communicate()
    if not data:
        break
    elif data == 'killsrv':
        sys.exit()

Answer 1

危险，威尔·罗宾逊！！！

您确实想在不通过网络进行身份验证的情况下以明文形式发送命令吗？这是非常非常危险的。

使用 paramiko 通过 SSH 完成此操作。

好吧，这个答案我听过太多次了。我不想使用 SSH，我只是构建它来了解有关套接字的更多信息。如果我想向系统发送命令，我不会实际使用它。 – 奥斯汀M

我无法从你的问题中推断出这个崇高的追求。 :-)

套接字模块是 posix 库上的一个薄层；普通的插座既乏味又难以正确使用。截至今天（2014 年），异步 I/O 和并发性并不是 Python 最强的特性 - 3.4 开始改变这一点，但库将落后一段时间。我的建议是花时间学习一些更高级别的 API，例如 Twisted (twistedmatrix.com/trac)。如果您确实对底层内容感兴趣，请深入研究项目源代码。

好吧。关于如何使用扭曲来处理这种类型的事情有什么想法吗？ – 奥斯汀M

看看twistedmatrix.com/documents/current/core/examples/#auto2

Answer 2

好吧，我可以理解你的沮丧奥斯汀；我也在同一条船上。然而，反复试验最终成功了。希望您正在寻找这个：

print "Command is:",command
op = subprocess.Popen(command, shell=True, stderr=subprocess.PIPE, stdout=subprocess.PIPE)
if op:
    output=str(op.stdout.read())
    print "Output:",output
    conn.sendall(output)

else:
    error=str(op.stderr.read())
    print "Error:",error
    conn.sendall(error)

Answer 3

目前还不清楚为什么在客户端和服务器中使用 subprocess.Popen() 来执行相同的命令。这是我会尝试做的事情的概述（伪代码）：

客户端

while True:
    read command from user
    send command to server
    wait for and then read response from server
    print response to user

服务器

while True:
    wait for and then read command from client
    if command is "killsrv", exit
    execute command and capture output
    send output to client

Answer 4

您的代码的问题在于这一行（在客户端和服务器中）：

proc = subprocess.Popen(cmd, stdout=iotype).wait()
stdout,stderr = proc.communicate()

您正在对 Popen 对象调用 wait，这意味着变量 proc 正在获取一个 int （由 wait 返回）而不是 Popen 对象。您可以摆脱 wait - 因为 communicate 在返回之前等待进程结束，并且无论如何您都不会检查退出代码，因此您不会需要调用它。

然后，在您的客户端中，我认为您甚至不需要子进程调用，除非您正在运行服务器发回的某些命令。