dennii/Termix
1
0
Fork 0
Code Issues Pull Requests 2 Actions Packages Projects Releases 22 Wiki Activity

Update ssh.ts

Browse Source
This commit is contained in:
Karmaa
2025-07-31 15:20:27 -05:00
committed by GitHub
parent 4b53139c20
commit 38f836b94f
1 changed files with 30 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

94
src/backend/ssh/ssh.ts
View File

@@ -10,19 +10,13 @@ const formatMessage = (level: string, colorFn: chalk.Chalk, message: string): st
return `${getTimeStamp()} ${colorFn(`[${level.toUpperCase()}]`)} ${chalk.hex('#1e3a8a')(`[${sshIconSymbol}]`)} ${message}`;
};
const logger = {
info: (msg: string): void => {
console.log(formatMessage('info', chalk.cyan, msg));
},
warn: (msg: string): void => {
console.warn(formatMessage('warn', chalk.yellow, msg));
},
info: (msg: string): void => console.log(formatMessage('info', chalk.cyan, msg)),
warn: (msg: string): void => console.warn(formatMessage('warn', chalk.yellow, msg)),
error: (msg: string, err?: unknown): void => {
console.error(formatMessage('error', chalk.redBright, msg));
if (err) console.error(err);
},
success: (msg: string): void => {
console.log(formatMessage('success', chalk.greenBright, msg));
},
success: (msg: string): void => console.log(formatMessage('success', chalk.greenBright, msg)),
debug: (msg: string): void => {
if (process.env.NODE_ENV !== 'production') {
console.debug(formatMessage('debug', chalk.magenta, msg));
@@ -33,6 +27,7 @@ const logger = {
wss.on('connection', (ws: WebSocket) => {
let sshConn: Client | null = null;
let sshStream: ClientChannel | null = null;
let keepAliveTimer: NodeJS.Timeout | null = null;
ws.on('close', () => {
cleanupSSH();
@@ -54,38 +49,21 @@ wss.on('connection', (ws: WebSocket) => {
case 'connectToHost':
handleConnectToHost(data);
break;
case 'resize':
handleResize(data);
break;
case 'disconnect':
cleanupSSH();
break;
case 'input':
if (sshStream) sshStream.write(data);
break;
default:
logger.warn('Unknown message type: ' + type);
}
});
function handleConnectToHost(data: {
cols: number;
rows: number;
hostConfig: {
ip: string;
port: number;
username: string;
password?: string;
key?: string;
keyPassword?: string;
keyType?: string;
authType?: string;
};
}) {
function handleConnectToHost(data: any) {
const { cols, rows, hostConfig } = data;
const { ip, port, username, password, key, keyPassword, keyType, authType } = hostConfig;
@@ -94,13 +72,11 @@ wss.on('connection', (ws: WebSocket) => {
ws.send(JSON.stringify({ type: 'error', message: 'Invalid username provided' }));
return;
}
if (!ip || typeof ip !== 'string' || ip.trim() === '') {
logger.error('Invalid IP provided');
ws.send(JSON.stringify({ type: 'error', message: 'Invalid IP provided' }));
return;
}
if (!port || typeof port !== 'number' || port <= 0) {
logger.error('Invalid port provided');
ws.send(JSON.stringify({ type: 'error', message: 'Invalid port provided' }));
@@ -130,17 +106,12 @@ wss.on('connection', (ws: WebSocket) => {
sshStream = stream;
const keepaliveTimer = setInterval(() => {
if (sshStream && sshStream.writable) {
sshStream.write('\x00');
}
}, 30000);
stream.on('data', (chunk: Buffer) => {
ws.send(JSON.stringify({ type: 'data', data: chunk.toString() }));
});
stream.on('close', () => {
ws.send(JSON.stringify({ type: 'disconnected', message: 'SSH session closed' }));
cleanupSSH();
});
@@ -150,32 +121,37 @@ wss.on('connection', (ws: WebSocket) => {
});
ws.send(JSON.stringify({ type: 'connected', message: 'SSH connected' }));
keepAliveTimer = setInterval(() => {
if (sshStream && sshStream.writable) {
sshStream.write(''); // keepalive
}
}, 30000);
});
});
sshConn.on('error', (err: Error) => {
logger.error('SSH connection error: ' + err.message);
let errorMessage = 'SSH error: ' + err.message;
if (err.message.includes('No matching key exchange algorithm')) {
errorMessage = 'SSH error: No compatible key exchange algorithm found. This may be due to an older SSH server or network device.';
errorMessage = 'SSH error: No compatible key exchange algorithm found.';
} else if (err.message.includes('No matching cipher')) {
errorMessage = 'SSH error: No compatible cipher found. This may be due to an older SSH server or network device.';
} else if (err.message.includes('No matching MAC')) {
errorMessage = 'SSH error: No compatible MAC algorithm found. This may be due to an older SSH server or network device.';
errorMessage = 'SSH error: No compatible cipher found.';
} else if (err.message.includes('ENOTFOUND') || err.message.includes('ENOENT')) {
errorMessage = 'SSH error: Could not resolve hostname or connect to server.';
} else if (err.message.includes('ECONNREFUSED')) {
errorMessage = 'SSH error: Connection refused. The server may not be running or the port may be incorrect.';
errorMessage = 'SSH error: Connection refused.';
} else if (err.message.includes('ETIMEDOUT')) {
errorMessage = 'SSH error: Connection timed out. Check your network connection and server availability.';
errorMessage = 'SSH error: Connection timed out.';
}
ws.send(JSON.stringify({ type: 'error', message: errorMessage }));
cleanupSSH();
});
sshConn.on('close', () => {
ws.send(JSON.stringify({ type: 'disconnected', message: 'SSH connection closed' }));
cleanupSSH();
});
@@ -186,7 +162,6 @@ wss.on('connection', (ws: WebSocket) => {
keepaliveInterval: 10000,
keepaliveCountMax: 60,
readyTimeout: 10000,
algorithms: {
kex: [
'diffie-hellman-group14-sha256',
@@ -209,30 +184,18 @@ wss.on('connection', (ws: WebSocket) => {
'aes256-cbc',
'3des-cbc'
],
hmac: [
'hmac-sha2-256',
'hmac-sha2-512',
'hmac-sha1',
'hmac-md5'
],
compress: [
'none',
'zlib@openssh.com',
'zlib'
]
hmac: ['hmac-sha2-256', 'hmac-sha2-512', 'hmac-sha1', 'hmac-md5'],
compress: ['none', 'zlib@openssh.com', 'zlib']
}
};
if (authType === 'key' && key) {
connectConfig.privateKey = key;
if (keyPassword) {
connectConfig.passphrase = keyPassword;
}
if (keyType && keyType !== 'auto') {
connectConfig.privateKeyType = keyType;
}
if (keyPassword) connectConfig.passphrase = keyPassword;
if (keyType && keyType !== 'auto') connectConfig.privateKeyType = keyType;
} else if (authType === 'key') {
logger.error('SSH key authentication requested but no key provided');
ws.send(JSON.stringify({ type: 'error', message: 'SSH key authentication requested but no key provided' }));
logger.error('SSH key auth requested but no key provided');
ws.send(JSON.stringify({ type: 'error', message: 'SSH key auth requested but no key provided' }));
return;
} else {
connectConfig.password = password;
@@ -249,6 +212,10 @@ wss.on('connection', (ws: WebSocket) => {
}
function cleanupSSH() {
if (keepAliveTimer) {
clearInterval(keepAliveTimer);
keepAliveTimer = null;
}
if (sshStream) {
try {
sshStream.end();
@@ -257,7 +224,6 @@ wss.on('connection', (ws: WebSocket) => {
}
sshStream = null;
}
if (sshConn) {
try {
sshConn.end();