SSHJ: exec挂起或丢失数据

Question

我使用Trilead和Jsch作为Hg的SSH客户端，现在我正在使用SSHJ，因为它似乎提供了更现代的密钥支持。草图代码如下所示

SSHClient client = createClientAndAuthenticate();
Session session = client.startSession();
Command cmd = session.exec(command); // command usually is "hg -R /path/to/hg/repository serve --stdio"

startThreadToCopyInputStream(cmd.getOutputStream());

startThreadToCopyOutputStream(cmd.getInputStream());
startThreadToCopyOutputStream(cmd.getErrorStream());

cmd.join(); // here it hangs endlessly

startThreadToCopyInputStream方法将本地Hg进程中的所有字节复制到cmd.getOutputStream()中，然后完成输入流。但与Trilead和JSch相比，cmd.getInputStream()和cmd.getErrorStream()仍无休止地开放。

我现在已经将代码更改为如下内容：

SSHClient client = createClientAndAuthenticate();
Session session = client.startSession();
Command cmd = session.exec(command); // command usually is "hg -R /path/to/hg/repository serve --stdio"

startThreadToCopyInputStream(cmd.getOutputStream());

startThreadToCopyOutputStream(cmd.getInputStream());
startThreadToCopyOutputStream(cmd.getErrorStream());

waitUntilInputStreamIsClosed();
cmd.close();

这在90%的情况下都能正常工作，但有时Hg会抱怨服务器提前关闭了连接。由于服务器进程已经完成，我如何知道何时能够调用cmd.close()？

Answer 1

看起来这是SSHJ中的一个bug (参见机票143)。解决办法不是直接使用cmd.getOutputStream()，而是：

OutputStream cmdOutputStream = new FilterOutputStream(cmd.getOutputStream()) {
    @Override
    public void close() throws IOException {
        try {
            super.close();
        }
        finally {
            client.getTransport().write(new SSHPacket(Message.CHANNEL_EOF).putUInt32(session.getRecipient()));
        }
    }
};

在传递到startThreadToCopyInputStream(cmdOutputStream);之前。