101 lines
3.8 KiB
JavaScript
101 lines
3.8 KiB
JavaScript
const { Server } = require('socket.io')
|
||
const { Client: SSHClient } = require('ssh2')
|
||
const { readHostList, readSSHRecord, verifyAuthSync, RSADecryptSync, AESDecryptSync } = require('../utils')
|
||
|
||
function createTerminal(socket, sshClient) {
|
||
sshClient.shell({ term: 'xterm-color' }, (err, stream) => {
|
||
if (err) return socket.emit('output', err.toString())
|
||
// 终端输出
|
||
stream
|
||
.on('data', (data) => {
|
||
socket.emit('output', data.toString())
|
||
})
|
||
.on('close', () => {
|
||
consola.info('关闭终端')
|
||
sshClient.end()
|
||
})
|
||
// web端输入
|
||
socket.on('input', key => {
|
||
if (sshClient._sock.writable === false) return consola.info('终端连接已关闭')
|
||
stream.write(key)
|
||
})
|
||
socket.emit('connect_terminal') // 已连接终端，web端可以执行指令了
|
||
|
||
// 监听按键重置终端大小
|
||
socket.on('resize', ({ rows, cols }) => {
|
||
consola.info('更改tty终端行&列: ', { rows, cols })
|
||
stream.setWindow(rows, cols)
|
||
})
|
||
})
|
||
}
|
||
|
||
module.exports = (httpServer) => {
|
||
const serverIo = new Server(httpServer, {
|
||
path: '/terminal',
|
||
cors: {
|
||
origin: '*' // 'http://localhost:8080'
|
||
}
|
||
})
|
||
serverIo.on('connection', (socket) => {
|
||
// 前者兼容nginx反代, 后者兼容nodejs自身服务
|
||
let clientIp = socket.handshake.headers['x-forwarded-for'] || socket.handshake.address
|
||
let sshClient = new SSHClient()
|
||
consola.success('terminal websocket 已连接')
|
||
|
||
socket.on('create', async ({ host: ip, token }) => {
|
||
const { code } = await verifyAuthSync(token, clientIp)
|
||
if (code !== 1) {
|
||
socket.emit('token_verify_fail')
|
||
socket.disconnect()
|
||
return
|
||
}
|
||
const hostList = await readHostList()
|
||
const targetHostInfo = hostList.find(item => item.host === ip) || {}
|
||
let { authType, host, port, username } = targetHostInfo
|
||
if (!host) return socket.emit('create_fail', `查找【${ ip }】凭证信息失败`)
|
||
let authInfo = { host, port, username }
|
||
// 统一使用commonKey解密
|
||
try {
|
||
// 解密放到try里面，防止报错【commonKey必须配对, 否则需要重新添加服务器密钥】
|
||
if (authType === 'credential') {
|
||
let credentialId = await AESDecryptSync(targetHostInfo[authType])
|
||
const sshRecordList = await readSSHRecord()
|
||
const sshRecord = sshRecordList.find(item => item._id === credentialId)
|
||
authInfo.authType = sshRecord.authType
|
||
authInfo[authInfo.authType] = await AESDecryptSync(sshRecord[authInfo.authType])
|
||
}
|
||
consola.info('准备连接终端：', host)
|
||
targetHostInfo[targetHostInfo.authType] = await AESDecryptSync(targetHostInfo[targetHostInfo.authType])
|
||
// :TODO: 初始化后连接失败...
|
||
// console.log('targetHostInfo:', targetHostInfo)
|
||
consola.log('连接信息', { username, port, authType })
|
||
sshClient
|
||
.on('ready', () => {
|
||
consola.success('已连接到终端：', host)
|
||
socket.emit('connect_success', `已连接到终端：${ host }`)
|
||
createTerminal(socket, sshClient)
|
||
})
|
||
.on('error', (err) => {
|
||
console.log(err)
|
||
consola.error('连接终端失败:', err.level)
|
||
socket.emit('connect_fail', err.message)
|
||
})
|
||
.connect({
|
||
...authInfo
|
||
// debug: (info) => console.log(info)
|
||
})
|
||
} catch (err) {
|
||
consola.error('创建终端失败:', err.message)
|
||
socket.emit('create_fail', err.message)
|
||
}
|
||
})
|
||
|
||
socket.on('disconnect', (reason) => {
|
||
consola.info('终端连接断开:', reason)
|
||
sshClient.end()
|
||
sshClient.destroy()
|
||
sshClient = null
|
||
})
|
||
})
|
||
}
|