我正在用python编写一个与远程PowerShell代理交互的工具(基本上是PowerShell帝国,但用于psv1)。我想使用HTTP(S)作为代理和用户界面之间的命令和输出的传输媒介。你知道吗
为此,我让Flask作为子进程运行,然后在主线程中向用户显示一个简单的命令行。这在Linux上运行良好,但在Windows下,我在Process
上调用start时遇到以下错误:
TypeError("can't pickle _thread.lock objects")
我相信这与Windows如何处理Process
内的线程有关。有人能建议一种方法让它在Windows上工作吗?我的代码如下:
from flask import Flask, request
from multiprocessing import Process, Queue
import logging
port = 8080 # TODO: set this from cmdline args
cmds = Queue()
quitting = Queue()
app = Flask(__name__)
@app.route('/', methods=['GET'])
def get_command():
if cmds.empty():
return ''
else:
return cmds.get()
@app.route('/', methods=['POST'])
def get_output():
if request.data.decode() == "quitting":
quitting.put(True)
print("{} ".format(request.data.decode()),end='\n> ')
return ''
if __name__ == "__main__":
log = logging.getLogger('werkzeug')
log.setLevel(logging.ERROR)
t = Process(target=app.run, kwargs={'host': '0.0.0.0'})
t.start()
running = True
while running:
cmd = input("> ")
if cmd == "kill" or cmd =="quit":
print("Waiting for agent to close...")
cmds.put("quit")
while quitting.empty():
pass
t.terminate()
t.join()
running = False
else:
cmds.put(cmd)
目前没有回答
相关问题 更多 >
编程相关推荐