基于tcp实现远程执行命令

命令执行服务器：

# Author : Kelvin
# Date : 2019/1/30 20:10
from socket import *
import subprocess

ip_conf = ("127.0.0.1", 8888)
buffer_capacity = 1024
tcp_server = socket(AF_INET, SOCK_STREAM)
tcp_server.bind(ip_conf)
tcp_server.listen(5)
while True:
    conn, addr = tcp_server.accept()
    while True:
        try:
            cmd = conn.recv(buffer_capacity) #如果强制断开连接会触发try，try正是解决强制中断连接的问题
            print("收到的cmd：%s"%cmd)
            if not cmd: #如果使用quit断开连接，服务器会死循环收到空，该判断正是解决此问题
                break
            res = subprocess.Popen(cmd.decode("utf8"), shell=True, stdout=subprocess.PIPE, stdin=subprocess.PIPE,
                                   stderr=subprocess.PIPE)
            err = res.stderr.read()
            if err:
                back_msg = err
            else:
                back_msg = res.stdout.read()
            conn.send(back_msg)
        except Exception as e:
            print(e)
            break
tcp_server.close()

客户端：

# Author : Kelvin
# Date : 2019/1/30 20:10
from socket import *

ip_conf = ("127.0.0.1", 8888)
buffer_capacity = 1024
tcp_client = socket(AF_INET, SOCK_STREAM)
tcp_client.connect(ip_conf)
while True:
    cmd = input("Please input cmd : ")
    if not cmd:
        continue
    if cmd == "quit":
        break
    tcp_client.send(cmd.encode("utf8"))
    back_msg = tcp_client.recv(buffer_capacity).decode("gbk")
    print(back_msg)
tcp_client.close()

执行结果：