如何在远程 shell 的上下文中对 pdb 进行 stdin/stdout 重定向?
How to make stdin/stdout redirection of pdb, in the context of a remote shell?
我正在写一个 web shell,在服务器端使用 ptpython、gevent 和 Flask,在客户端使用 xtermjs 和 websockets。
我希望 Python 的正常 breakpoint()
功能可以使用 pdb
。
我很高兴发现 Pdb
class 采用自定义 stdin
和 stdout
,但是我无法使它正常工作 :(
在 web shell 中,用户输入的内容通过 websocket 进入服务器进程:有一个监听 greenlet,它写入由 ptpython 读取的管道。
然后,ptpython 输出通过 websocket 发送。到目前为止一切顺利,效果很好。
现在使用 pdb
:感谢 sys.breakpointhook
我创建了一个自定义 Pdb
实例,stdin 是一个管道,stdout 是另一个管道。我小心地使用 gevent.fileobject.FileObjectPosix
来获得非阻塞、合作的 I/O 流。我负责让用户输入写入
输入管道,我希望输出到另一个管道,所以我可以将它重定向到我的代码中适当的“标准输出”处理例程。
但是,在收到与 pdb
提示相对应的第一条消息后,我卡住了,一切似乎都被阻止了。
我使用以下代码重现了我的行为,如有任何帮助,我们将不胜感激:
from pdb import Pdb as _Pdb
import gevent
from gevent.fileobject import FileObjectPosix as File
import io
import os
import sys
import socket
debugger = None
class GeventPdb(_Pdb):
def __init__(self, own_stdout):
self._stdout = own_stdout
r1, w1 = os.pipe()
r2, w2 = os.pipe()
self.r = File(r1)
self.w = File(w1, "w")
self.r2 = File(r2)
self.w2 = File(w2, "w")
super().__init__(stdin=self.r, stdout=self.w2)
self.stdout_handling = gevent.spawn(self.handle_stdout)
def send_text(self, txt):
# this should write to the 'stdin' pipe,
# thus provoking input reading from Pdb
print("WRITING TO PDB's STDIN")
self.w.write(txt.decode())
def handle_stdout(self):
# this should read what is written by Pdb
# to Pdb's stdout, to redirect it to our
# own stdout
while True:
print("BEFORE READING PDB's STDOUT")
char = self.r2.read(1)
print(f"WRITING TO CUSTOM STDOUT, stdout len is {len(self._stdout.getvalue())}")
self._stdout.write(char)
def test_func():
debugger.set_trace()
if __name__ == '__main__':
custom_stdout = io.StringIO() # this simulates a custom stdout pipe
debugger = GeventPdb(custom_stdout)
# the next socket is the server socket,
# for 'remote' operation
s = socket.socket()
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.bind(('', 4444))
s.listen(1)
print("Connect client to continue... The client is used to send commands to the debugger")
client, client_addr = s.accept()
def handle_client(client):
while True:
cmd = client.recv(1024)
if not cmd:
break
print("SENDING TEXT TO DEBUGGER...")
debugger.send_text(cmd)
gevent.spawn(handle_client, client)
print("now start a function which starts the debugger...")
print("we should see the pdb prompt")
test_func()
我运行以下代码,然后通过telnet localhost 4444
连接然后我可以看到我收到pdb
提示,然后我可以输入命令并按[enter]:似乎没有收到
pdb
的输入管道。它会阻塞。
如有任何帮助,我们将不胜感激!
你必须刷新你的写入(除了猴子补丁):
def send_text(self, txt):
# this should write to the 'stdin' pipe,
# thus provoking input reading from Pdb
print("WRITING TO PDB's STDIN")
self.w.write(txt.decode())
self.w.flush() # !!!
我正在写一个 web shell,在服务器端使用 ptpython、gevent 和 Flask,在客户端使用 xtermjs 和 websockets。
我希望 Python 的正常 breakpoint()
功能可以使用 pdb
。
我很高兴发现 Pdb
class 采用自定义 stdin
和 stdout
,但是我无法使它正常工作 :(
在 web shell 中,用户输入的内容通过 websocket 进入服务器进程:有一个监听 greenlet,它写入由 ptpython 读取的管道。 然后,ptpython 输出通过 websocket 发送。到目前为止一切顺利,效果很好。
现在使用 pdb
:感谢 sys.breakpointhook
我创建了一个自定义 Pdb
实例,stdin 是一个管道,stdout 是另一个管道。我小心地使用 gevent.fileobject.FileObjectPosix
来获得非阻塞、合作的 I/O 流。我负责让用户输入写入
输入管道,我希望输出到另一个管道,所以我可以将它重定向到我的代码中适当的“标准输出”处理例程。
但是,在收到与 pdb
提示相对应的第一条消息后,我卡住了,一切似乎都被阻止了。
我使用以下代码重现了我的行为,如有任何帮助,我们将不胜感激:
from pdb import Pdb as _Pdb
import gevent
from gevent.fileobject import FileObjectPosix as File
import io
import os
import sys
import socket
debugger = None
class GeventPdb(_Pdb):
def __init__(self, own_stdout):
self._stdout = own_stdout
r1, w1 = os.pipe()
r2, w2 = os.pipe()
self.r = File(r1)
self.w = File(w1, "w")
self.r2 = File(r2)
self.w2 = File(w2, "w")
super().__init__(stdin=self.r, stdout=self.w2)
self.stdout_handling = gevent.spawn(self.handle_stdout)
def send_text(self, txt):
# this should write to the 'stdin' pipe,
# thus provoking input reading from Pdb
print("WRITING TO PDB's STDIN")
self.w.write(txt.decode())
def handle_stdout(self):
# this should read what is written by Pdb
# to Pdb's stdout, to redirect it to our
# own stdout
while True:
print("BEFORE READING PDB's STDOUT")
char = self.r2.read(1)
print(f"WRITING TO CUSTOM STDOUT, stdout len is {len(self._stdout.getvalue())}")
self._stdout.write(char)
def test_func():
debugger.set_trace()
if __name__ == '__main__':
custom_stdout = io.StringIO() # this simulates a custom stdout pipe
debugger = GeventPdb(custom_stdout)
# the next socket is the server socket,
# for 'remote' operation
s = socket.socket()
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.bind(('', 4444))
s.listen(1)
print("Connect client to continue... The client is used to send commands to the debugger")
client, client_addr = s.accept()
def handle_client(client):
while True:
cmd = client.recv(1024)
if not cmd:
break
print("SENDING TEXT TO DEBUGGER...")
debugger.send_text(cmd)
gevent.spawn(handle_client, client)
print("now start a function which starts the debugger...")
print("we should see the pdb prompt")
test_func()
我运行以下代码,然后通过telnet localhost 4444
连接然后我可以看到我收到pdb
提示,然后我可以输入命令并按[enter]:似乎没有收到
pdb
的输入管道。它会阻塞。
如有任何帮助,我们将不胜感激!
你必须刷新你的写入(除了猴子补丁):
def send_text(self, txt):
# this should write to the 'stdin' pipe,
# thus provoking input reading from Pdb
print("WRITING TO PDB's STDIN")
self.w.write(txt.decode())
self.w.flush() # !!!