diff --git a/src/main/java/net/schmizz/sshj/transport/DisconnectListener.java b/src/main/java/net/schmizz/sshj/transport/DisconnectListener.java index 09013761..f2a4c299 100644 --- a/src/main/java/net/schmizz/sshj/transport/DisconnectListener.java +++ b/src/main/java/net/schmizz/sshj/transport/DisconnectListener.java @@ -19,6 +19,6 @@ import net.schmizz.sshj.common.DisconnectReason; public interface DisconnectListener { - void notifyDisconnect(DisconnectReason reason); + void notifyDisconnect(DisconnectReason reason, String message); } diff --git a/src/main/java/net/schmizz/sshj/transport/TransportImpl.java b/src/main/java/net/schmizz/sshj/transport/TransportImpl.java index 22821397..75a0ed24 100644 --- a/src/main/java/net/schmizz/sshj/transport/TransportImpl.java +++ b/src/main/java/net/schmizz/sshj/transport/TransportImpl.java @@ -89,7 +89,7 @@ public final class TransportImpl private final DisconnectListener nullDisconnectListener = new DisconnectListener() { @Override - public void notifyDisconnect(DisconnectReason reason) { + public void notifyDisconnect(DisconnectReason reason, String message) { log.info("Disconnected - {}", reason); } }; @@ -383,7 +383,7 @@ public final class TransportImpl close.lock(); try { if (isRunning()) { - disconnectListener.notifyDisconnect(reason); + disconnectListener.notifyDisconnect(reason, message); getService().notifyError(new TransportException(reason, "Disconnected")); sendDisconnect(reason, message); finishOff(); @@ -576,7 +576,7 @@ public final class TransportImpl final SSHException causeOfDeath = SSHException.chainer.chain(ex); - disconnectListener.notifyDisconnect(causeOfDeath.getDisconnectReason()); + disconnectListener.notifyDisconnect(causeOfDeath.getDisconnectReason(), causeOfDeath.getMessage()); ErrorDeliveryUtil.alertEvents(causeOfDeath, close, serviceAccept); kexer.notifyError(causeOfDeath); diff --git a/src/test/java/net/schmizz/sshj/transport/Disconnection.java b/src/test/java/net/schmizz/sshj/transport/Disconnection.java index 1b5586e5..d0efa6ce 100644 --- a/src/test/java/net/schmizz/sshj/transport/Disconnection.java +++ b/src/test/java/net/schmizz/sshj/transport/Disconnection.java @@ -42,7 +42,7 @@ public class Disconnection { fixture.getClient().getTransport().setDisconnectListener(new DisconnectListener() { @Override - public void notifyDisconnect(DisconnectReason reason) { + public void notifyDisconnect(DisconnectReason reason, String message) { notified = true; } });