113 lines
4.0 KiB
JavaScript
113 lines
4.0 KiB
JavaScript
const http = require("http");
|
|
const socketIo = require("socket.io");
|
|
const SSHClient = require("ssh2").Client;
|
|
|
|
const server = http.createServer();
|
|
const io = socketIo(server, {
|
|
cors: {
|
|
origin: "*",
|
|
methods: ["GET", "POST"],
|
|
credentials: true
|
|
},
|
|
allowEIO3: true
|
|
});
|
|
|
|
io.on("connection", (socket) => {
|
|
console.log("New socket connection established");
|
|
|
|
let stream = null;
|
|
|
|
socket.on("connectToHost", (cols, rows, hostConfig) => {
|
|
if (!hostConfig || !hostConfig.ip || !hostConfig.user || (!hostConfig.password && !hostConfig.rsaKey) || !hostConfig.port) {
|
|
console.error("Invalid hostConfig received:", hostConfig);
|
|
return;
|
|
}
|
|
|
|
// Redact only sensitive info for logging
|
|
const safeHostConfig = {
|
|
ip: hostConfig.ip,
|
|
port: hostConfig.port,
|
|
user: hostConfig.user,
|
|
password: hostConfig.password ? '***REDACTED***' : undefined,
|
|
rsaKey: hostConfig.rsaKey ? '***REDACTED***' : undefined,
|
|
};
|
|
|
|
console.log("Received hostConfig:", safeHostConfig);
|
|
const { ip, port, user, password, rsaKey } = hostConfig;
|
|
|
|
const conn = new SSHClient();
|
|
conn
|
|
.on("ready", function () {
|
|
console.log("SSH connection established");
|
|
socket.emit("data", "\r\n*** SSH CONNECTION ESTABLISHED ***\r\n");
|
|
|
|
conn.shell({ term: "xterm-256color" }, function (err, newStream) {
|
|
if (err) {
|
|
console.error("Error opening SSH shell:", err);
|
|
return socket.emit(
|
|
"data",
|
|
"\r\n*** SSH SHELL ERROR: " + err.message + " ***\r\n"
|
|
);
|
|
}
|
|
stream = newStream;
|
|
|
|
// Set initial terminal size
|
|
stream.setWindow(rows, cols, rows * 100, cols * 100);
|
|
console.log(`Initial terminal size: cols=${cols}, rows=${rows}`);
|
|
|
|
// Pipe SSH output to client
|
|
stream.on("data", function (data) {
|
|
socket.emit("data", data.toString("binary"));
|
|
});
|
|
|
|
stream.on("close", function () {
|
|
console.log("SSH stream closed");
|
|
conn.end();
|
|
});
|
|
|
|
// Send keystrokes from terminal to SSH
|
|
socket.on("data", function (data) {
|
|
stream.write(data);
|
|
});
|
|
|
|
// Resize SSH terminal when client resizes
|
|
socket.on("resize", ({ cols, rows }) => {
|
|
if (stream && stream.setWindow) {
|
|
stream.setWindow(rows, cols, rows * 100, cols * 100);
|
|
console.log(`Terminal resized: cols=${cols}, rows=${rows}`);
|
|
}
|
|
});
|
|
|
|
// Auto-send initial terminal size to backend
|
|
socket.emit("resize", { cols, rows });
|
|
});
|
|
})
|
|
.on("close", function () {
|
|
console.log("SSH connection closed");
|
|
socket.emit("data", "\r\n*** SSH CONNECTION CLOSED ***\r\n");
|
|
})
|
|
.on("error", function (err) {
|
|
console.error("SSH connection error:", err);
|
|
socket.emit(
|
|
"data",
|
|
"\r\n*** SSH CONNECTION ERROR: " + err.message + " ***\r\n"
|
|
);
|
|
})
|
|
.connect({
|
|
host: ip,
|
|
port: port,
|
|
username: user,
|
|
password: password,
|
|
privateKey: rsaKey ? Buffer.from(rsaKey) : undefined,
|
|
});
|
|
});
|
|
|
|
socket.on("disconnect", () => {
|
|
console.log("Client disconnected");
|
|
});
|
|
});
|
|
|
|
server.listen(8081, '0.0.0.0', () => {
|
|
console.log("Server is running on port 8081");
|
|
});
|