Skip to content

Commit c7373f0

Browse files
committed
Merge pull request #186 from bluekeyes/fix/read-end-of-stream
Detect end-of-stream in TransportImpl#init
2 parents 5fc08a3 + 3ebd2eb commit c7373f0

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/main/java/net/schmizz/sshj/transport/TransportImpl.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,10 @@ public void init(String remoteHost, int remotePort, InputStream in, OutputStream
158158
// Read server's ID
159159
final Buffer.PlainBuffer buf = new Buffer.PlainBuffer();
160160
while ((serverID = readIdentification(buf)).isEmpty()) {
161-
buf.putByte((byte) connInfo.in.read());
161+
int b = connInfo.in.read();
162+
if (b == -1)
163+
throw new TransportException("Server closed connection during identification exchange");
164+
buf.putByte((byte) b);
162165
}
163166

164167
log.info("Server identity string: {}", serverID);

0 commit comments

Comments
 (0)