imterah/hermes
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases 1 Packages Wiki Activity Actions

feature: Adds code to reinitialize SSH after disconnection.

Browse source
This commit is contained in:
greysoh 2024-05-30 16:39:17 -04:00
parent 59a1eed141
commit 431acc505e
Signed by: imterah
GPG key ID: 8FA7DD57BA6CEA37
1 changed files with 23 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

23
api/src/backendimpl/ssh.ts
View file

@ -103,6 +103,29 @@ export class SSHBackendProvider implements BackendBaseClass {
return false;
}
if (this.sshInstance.connection) {
this.sshInstance.connection.on("end", async() => {
if (this.state != "started") return;
this.logs.push("We disconnected from the SSH server. Restarting...");
// Create a new array from the existing list of proxies, so we have a backup of the proxy list before
// we wipe the list of all proxies and clients (as we're disconnected anyways)
const proxies = Array.from(this.proxies);
this.proxies.splice(0, this.proxies.length);
this.clients.splice(0, this.clients.length);
await this.start();
if (this.state != "started") return;
for (const proxy of proxies) {
if (!proxy.enabled) continue;
this.addConnection(proxy.sourceIP, proxy.sourcePort, proxy.destPort, "tcp");
}
});
};
this.state = "started";
this.logs.push("Successfully started SSHBackendProvider.");