我如何(可以吗?)从命令中获取退出状态?

时间:2013-04-05 15:00:49

标签: python ssh twisted twisted.conch

使用twisted.conch,我正在使用以下SSHChannel实现,该实现提供了一个在实例化时运行的命令:

class CommandChannel(channel.SSHChannel):
    name = 'session'

    def __init__(self, command, *args, **kwargs):
        channel.SSHChannel.__init__(self, *args, **kwargs)
        self.command = command

    def channelOpen(self, data):
        d = self.conn.sendRequest(
            self, 'exec', common.NS(self.command), wantReply=True)
        d.addCallback(self._gotResponse)

    def _gotResponse(self, _):
        self.conn.sendEOF(self)
        self.loseConnection()

    def dataReceived(self, data):
        log.msg('dataReceived: data=%r' % data)

    def extReceived(self, dataType, data):
        log.msg('extReceived: dataType=%r, data=%r' % (dataType, data))

    def closed(self):
        reactor.stop()

为此,我正在提供一个应该返回某种退出状态的shell命令。我可以阅读退出状态是什么吗?我曾希望extReceived可能是我的方式,但它看起来不像。

在我的日志中,我看到了这一点,这似乎应该是一个提示在哪里看,但我不确定如何遵循上述提示:

2013-04-05 10:39:37-0400 [SSHChannel session (0) on SSHService ssh-connection
    on CommandTransport,client] unhandled request for exit-status

1 个答案:

答案 0 :(得分:4)

您可能需要为此实现两个额外的回调。这些是频道上的回调,所以你已经非常接近了。

第一个是request_exit_status。实现此功能后,Conch将不再记录exit-status请求未处理。这是一个示例实现:

def request_exit_status(self, data):
    """
    When the server sends the command's exit status, record it for later
    delivery to the protocol.

    @param data: The network-order four byte representation of the exit
        status of the command.
    @type data: L{bytes}
    """
    (status,) = unpack('>L', data)
    if status != 0:
        self._reason = ProcessTerminated(status, None, None)

第二个是request_exit_signal。如果命令被信号终止,则发送exit-signal而不是exit-status。这是一个示例实现:

def request_exit_signal(self, data):
    """
    When the server sends the command's exit status, record it for later
    delivery to the protocol.

    @param data: The network-order four byte representation of the exit
        signal of the command.
    @type data: L{bytes}
    """
    (signal,) = unpack('>L', data)
    self._reason = ProcessTerminated(None, signal, None)

这里的一般模式是当SSHChannel收到 foo 的请求时,它会尝试使用构成请求的字节来调用request_foo