feature: Add support for getting inbound connection.

Also fixes any bugs I found.
This commit is contained in:
greysoh 2024-05-07 18:35:04 -04:00
parent b99d6a89dc
commit a9fcdea037
Signed by: imterah
GPG key ID: 8FA7DD57BA6CEA37
8 changed files with 146 additions and 17 deletions

View file

@ -5,14 +5,13 @@ meta {
}
post {
url: http://127.0.0.1:3000/api/v1/backends/remove
url: http://127.0.0.1:3000/api/v1/backends/lookup
body: json
auth: none
}
body:json {
{
"token": "f1b89cc337073476289ade17ffbe7a6419b4bd52aa7ede26114bffd76fa263b5cb1bcaf389462e1d9e7acb7f4b6a7c28152a9cc9af83e3ec862f1892b1",
"id": "2"
"token": "7d69814cdada551dd22521ad97b23b22a106278826a2b4e87dd76246594b56f973894e8265437a5d520ed7258d7c856d0d294e89b1de1a98db7fa4a"
}
}

View file

@ -1,4 +1,4 @@
import type { BackendBaseClass } from "./base.js";
import { BackendBaseClass } from "./base.js";
import { PassyFireBackendProvider } from "./passyfire-reimpl/index.js";
import { SSHBackendProvider } from "./ssh.js";
@ -7,3 +7,7 @@ export const backendProviders: Record<string, typeof BackendBaseClass> = {
ssh: SSHBackendProvider,
passyfire: PassyFireBackendProvider,
};
if (process.env.NODE_ENV != "production") {
backendProviders["dummy"] = BackendBaseClass;
}

View file

@ -179,7 +179,7 @@ export class SSHBackendProvider implements BackendBaseClass {
srcConn.write(chunk);
});
destConn.addListener("close", () => {
destConn.addListener("end", () => {
this.clients.splice(this.clients.indexOf(client), 1);
srcConn.end();
});
@ -190,7 +190,7 @@ export class SSHBackendProvider implements BackendBaseClass {
srcConn.on("end", () => {
this.clients.splice(this.clients.indexOf(client), 1);
destConn.close();
destConn.end();
});
},
);

View file

@ -45,19 +45,23 @@ export function route(routeOptions: RouteOptions) {
},
});
if (!forward)
if (!forward) {
return res.status(400).send({
error: "Could not find forward entry",
});
}
if (!backends[forward.destProviderID])
if (!backends[forward.destProviderID]) {
return res.status(400).send({
error: "Backend not found",
});
}
return {
success: true,
data: backends[forward.destProviderID].getAllConnections(),
data: backends[forward.destProviderID].getAllConnections().filter((i) => {
return i.connectionDetails.sourceIP == forward.sourceIP && i.connectionDetails.sourcePort == forward.sourcePort && i.connectionDetails.destPort == forward.destPort;
}),
};
},
);