Add disconnect message from server.

This commit is contained in:
David Kocher
2014-05-16 15:30:04 +02:00
parent f34667521d
commit a8d2ea2028
3 changed files with 5 additions and 5 deletions

View File

@@ -19,6 +19,6 @@ import net.schmizz.sshj.common.DisconnectReason;
public interface DisconnectListener { public interface DisconnectListener {
void notifyDisconnect(DisconnectReason reason); void notifyDisconnect(DisconnectReason reason, String message);
} }

View File

@@ -89,7 +89,7 @@ public final class TransportImpl
private final DisconnectListener nullDisconnectListener = new DisconnectListener() { private final DisconnectListener nullDisconnectListener = new DisconnectListener() {
@Override @Override
public void notifyDisconnect(DisconnectReason reason) { public void notifyDisconnect(DisconnectReason reason, String message) {
log.info("Disconnected - {}", reason); log.info("Disconnected - {}", reason);
} }
}; };
@@ -383,7 +383,7 @@ public final class TransportImpl
close.lock(); close.lock();
try { try {
if (isRunning()) { if (isRunning()) {
disconnectListener.notifyDisconnect(reason); disconnectListener.notifyDisconnect(reason, message);
getService().notifyError(new TransportException(reason, "Disconnected")); getService().notifyError(new TransportException(reason, "Disconnected"));
sendDisconnect(reason, message); sendDisconnect(reason, message);
finishOff(); finishOff();
@@ -576,7 +576,7 @@ public final class TransportImpl
final SSHException causeOfDeath = SSHException.chainer.chain(ex); final SSHException causeOfDeath = SSHException.chainer.chain(ex);
disconnectListener.notifyDisconnect(causeOfDeath.getDisconnectReason()); disconnectListener.notifyDisconnect(causeOfDeath.getDisconnectReason(), causeOfDeath.getMessage());
ErrorDeliveryUtil.alertEvents(causeOfDeath, close, serviceAccept); ErrorDeliveryUtil.alertEvents(causeOfDeath, close, serviceAccept);
kexer.notifyError(causeOfDeath); kexer.notifyError(causeOfDeath);

View File

@@ -42,7 +42,7 @@ public class Disconnection {
fixture.getClient().getTransport().setDisconnectListener(new DisconnectListener() { fixture.getClient().getTransport().setDisconnectListener(new DisconnectListener() {
@Override @Override
public void notifyDisconnect(DisconnectReason reason) { public void notifyDisconnect(DisconnectReason reason, String message) {
notified = true; notified = true;
} }
}); });