feat: v0.3.0 版本 - SSH终端、SFTP、端口转发功能实现
核心功能: - SSH终端: 密码/密钥认证、XTerm.js终端、多主题配色、IP高亮 - SFTP管理: 双栏文件管理、上传下载、传输队列 - 端口转发: 本地/远程转发、SOCKS5动态代理、规则管理 架构改进: - 拆分大型组件 (SFTPView/TabBar/Sidebar) - 添加统一日志和错误处理 - 修复React严格模式下的终端黑屏问题 UI更新: - 新增端口转发视图 - 更新UI预览文档 - 左侧栏右键菜单支持SFTP快捷打开
This commit is contained in:
Logan.Li committed
1 parent
cf1c054381
commit
484d850abd
105 files changed
+25330
-64
No files matched your search
@@ -0,0 +1,685 @@
|
||||
import { Client, ClientChannel } from 'ssh2'
|
||||
import { SFTPWrapper } from 'ssh2'
|
||||
import { v4 as uuidv4 } from 'uuid'
|
||||
import { createLogger } from '../utils/logger'
|
||||
import * as net from 'net'
|
||||
|
||||
const logger = createLogger('SSHService')
|
||||
|
||||
// ===== 彩色调试日志 =====
|
||||
const DEBUG = true
|
||||
function debug(...args: any[]) {
|
||||
if (DEBUG) console.log('[SSHService]', ...args)
|
||||
}
|
||||
function logHeader(msg: string) {
|
||||
console.log(`%c${msg}`, 'background: #e74c3c; color: #fff; padding: 4px 8px; border-radius: 4px; font-weight: bold;')
|
||||
}
|
||||
function logInfo(label: string, value: any) {
|
||||
console.log(`%c${label}:%c ${value}`, 'background: #2c3e50; color: #fff; padding: 2px 6px; border-radius: 3px;', 'color: #3498db; font-weight: bold;')
|
||||
}
|
||||
function logSuccess(msg: string) {
|
||||
console.log(`%c✓ ${msg}`, 'color: #27ae60; font-weight: bold;')
|
||||
}
|
||||
function logError(msg: string) {
|
||||
console.log(`%c✗ ${msg}`, 'color: #e74c3c; font-weight: bold;')
|
||||
}
|
||||
function logWarning(msg: string) {
|
||||
console.log(`%c⚠ ${msg}`, 'color: #f39c12; font-weight: bold;')
|
||||
}
|
||||
|
||||
export interface SSHConfig {
|
||||
host: string
|
||||
port?: number
|
||||
username: string
|
||||
password?: string
|
||||
privateKey?: string
|
||||
passphrase?: string
|
||||
agent?: string
|
||||
readyTimeout?: number
|
||||
keepaliveInterval?: number
|
||||
}
|
||||
|
||||
export interface Connection {
|
||||
id: string
|
||||
client: Client
|
||||
config: SSHConfig
|
||||
status: 'connecting' | 'connected' | 'disconnected' | 'error'
|
||||
}
|
||||
|
||||
export interface ShellSession {
|
||||
id: string
|
||||
connectionId: string
|
||||
channel: ClientChannel
|
||||
}
|
||||
|
||||
export interface SFTPSession {
|
||||
id: string
|
||||
connectionId: string
|
||||
sftp: SFTPWrapper
|
||||
}
|
||||
|
||||
export class SSHService {
|
||||
private connections: Map<string, Connection> = new Map()
|
||||
private shells: Map<string, ShellSession> = new Map()
|
||||
private sftps: Map<string, SFTPSession> = new Map()
|
||||
|
||||
async connect(config: SSHConfig): Promise<Connection> {
|
||||
console.log('=== SSHService connect ===')
|
||||
console.log(' host:', config.host)
|
||||
console.log(' port:', config.port || 22)
|
||||
console.log(' username:', config.username)
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
const client = new Client()
|
||||
const connectionId = uuidv4()
|
||||
debug('>>> Generated connectionId:', connectionId)
|
||||
|
||||
const connection: Connection = {
|
||||
id: connectionId,
|
||||
client,
|
||||
config,
|
||||
status: 'connecting'
|
||||
}
|
||||
|
||||
this.connections.set(connectionId, connection)
|
||||
debug('>>> Connection stored, total connections:', this.connections.size)
|
||||
|
||||
client.on('ready', () => {
|
||||
connection.status = 'connected'
|
||||
logger.info(`SSH connection ${connectionId} established`)
|
||||
debug('>>> SSH client ready, connectionId:', connectionId)
|
||||
resolve(connection)
|
||||
})
|
||||
|
||||
client.on('error', (err) => {
|
||||
connection.status = 'error'
|
||||
logger.error(`SSH connection ${connectionId} error:`, err)
|
||||
debug('>>> SSH client ERROR:', err.message)
|
||||
reject(err)
|
||||
})
|
||||
|
||||
client.on('close', () => {
|
||||
connection.status = 'disconnected'
|
||||
logger.info(`SSH connection ${connectionId} closed`)
|
||||
debug('>>> SSH client closed, connectionId:', connectionId)
|
||||
this.connections.delete(connectionId)
|
||||
})
|
||||
|
||||
debug('>>> Starting SSH client.connect...')
|
||||
client.connect({
|
||||
host: config.host,
|
||||
port: config.port || 22,
|
||||
username: config.username,
|
||||
password: config.password,
|
||||
privateKey: config.privateKey,
|
||||
passphrase: config.passphrase,
|
||||
agent: config.agent,
|
||||
readyTimeout: config.readyTimeout || 20000,
|
||||
keepaliveInterval: config.keepaliveInterval || 60000,
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
async disconnect(connectionId: string): Promise<void> {
|
||||
const connection = this.connections.get(connectionId)
|
||||
if (connection) {
|
||||
connection.client.end()
|
||||
this.connections.delete(connectionId)
|
||||
|
||||
// Clean up associated shells
|
||||
for (const [shellId, shell] of this.shells.entries()) {
|
||||
if (shell.connectionId === connectionId) {
|
||||
shell.channel.close()
|
||||
this.shells.delete(shellId)
|
||||
}
|
||||
}
|
||||
|
||||
// Clean up associated SFTP sessions
|
||||
for (const [sftpId, sftp] of this.sftps.entries()) {
|
||||
if (sftp.connectionId === connectionId) {
|
||||
this.sftps.delete(sftpId)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
disconnectAll(): void {
|
||||
for (const connectionId of this.connections.keys()) {
|
||||
this.disconnect(connectionId)
|
||||
}
|
||||
}
|
||||
|
||||
async exec(connectionId: string, command: string): Promise<string> {
|
||||
const connection = this.connections.get(connectionId)
|
||||
if (!connection || connection.status !== 'connected') {
|
||||
throw new Error('Connection not established')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
let stdout = ''
|
||||
let stderr = ''
|
||||
|
||||
connection.client.exec(command, (err, stream) => {
|
||||
if (err) {
|
||||
reject(err)
|
||||
return
|
||||
}
|
||||
|
||||
stream.on('close', (code: number) => {
|
||||
if (code !== 0) {
|
||||
reject(new Error(`Command failed with code ${code}: ${stderr}`))
|
||||
} else {
|
||||
resolve(stdout)
|
||||
}
|
||||
})
|
||||
|
||||
stream.on('data', (data: Buffer) => {
|
||||
stdout += data.toString()
|
||||
})
|
||||
|
||||
stream.stderr.on('data', (data: Buffer) => {
|
||||
stderr += data.toString()
|
||||
})
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
async createShell(
|
||||
connectionId: string,
|
||||
options: { cols?: number; rows?: number; term?: string } = {},
|
||||
onData?: (shellId: string, data: Buffer) => void,
|
||||
preferredShellId?: string
|
||||
): Promise<ShellSession> {
|
||||
console.log('=== SSHService createShell ===')
|
||||
console.log(' connectionId:', connectionId)
|
||||
console.log(' options:', options)
|
||||
console.log(' preferredShellId:', preferredShellId)
|
||||
|
||||
const connection = this.connections.get(connectionId)
|
||||
console.log(' connection found:', !!connection)
|
||||
console.log(' connection status:', connection?.status)
|
||||
|
||||
if (!connection || connection.status !== 'connected') {
|
||||
throw new Error('Connection not established')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
debug('>>> Calling client.shell...')
|
||||
connection.client.shell({
|
||||
cols: options.cols || 80,
|
||||
rows: options.rows || 24,
|
||||
term: options.term || 'xterm-256color'
|
||||
}, (err, channel) => {
|
||||
if (err) {
|
||||
debug('>>> client.shell ERROR:', err.message)
|
||||
reject(err)
|
||||
return
|
||||
}
|
||||
|
||||
debug('>>> client.shell success, channel created')
|
||||
|
||||
// 使用前端提供的 shellId 或生成新的
|
||||
const shellId = preferredShellId || uuidv4()
|
||||
debug('>>> Using shellId:', shellId)
|
||||
|
||||
const shell: ShellSession = {
|
||||
id: shellId,
|
||||
connectionId,
|
||||
channel
|
||||
}
|
||||
|
||||
// Store shell BEFORE setting up callbacks
|
||||
this.shells.set(shellId, shell)
|
||||
debug('>>> Shell stored, total shells:', this.shells.size)
|
||||
|
||||
// Set up data callback AFTER shell is stored
|
||||
if (onData) {
|
||||
debug('>>> Setting up channel.on(data) handler')
|
||||
channel.on('data', (data: Buffer) => {
|
||||
// Log first few chunks for debugging
|
||||
const text = data.toString('utf-8')
|
||||
console.log('=== SSHService channel.on(data) ===')
|
||||
console.log(' shellId:', shellId)
|
||||
console.log(' data length:', data.length)
|
||||
console.log(' data preview:', text.substring(0, 100).replace(/\n/g, '\\n'))
|
||||
|
||||
if (text.includes('Last login') || text.includes('Welcome')) {
|
||||
logger.info('[SSHService] Shell data (first 200 chars):', text.substring(0, 200))
|
||||
}
|
||||
debug('>>> Calling onData callback...')
|
||||
onData(shellId, data)
|
||||
debug('>>> onData callback returned')
|
||||
})
|
||||
} else {
|
||||
debug('>>> WARNING: onData callback is undefined!')
|
||||
}
|
||||
|
||||
channel.on('close', () => {
|
||||
debug('>>> Channel closed, shellId:', shellId)
|
||||
this.shells.delete(shellId)
|
||||
})
|
||||
|
||||
debug('>>> Resolving shell:', shellId)
|
||||
resolve(shell)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
shellWrite(shellId: string, data: string | Uint8Array): void {
|
||||
const shell = this.shells.get(shellId)
|
||||
if (shell) {
|
||||
shell.channel.write(data)
|
||||
}
|
||||
}
|
||||
|
||||
shellResize(shellId: string, cols: number, rows: number): void {
|
||||
const shell = this.shells.get(shellId)
|
||||
if (shell) {
|
||||
shell.channel.setWindow(rows, cols, 0, 0)
|
||||
}
|
||||
}
|
||||
|
||||
getShell(shellId: string): ShellSession | undefined {
|
||||
return this.shells.get(shellId)
|
||||
}
|
||||
|
||||
getConnection(connectionId: string): Connection | undefined {
|
||||
return this.connections.get(connectionId)
|
||||
}
|
||||
|
||||
// ===== SFTP 操作 =====
|
||||
async createSFTP(connectionId: string): Promise<SFTPSession> {
|
||||
logHeader('📁 SFTP CREATE')
|
||||
logInfo('connectionId', connectionId)
|
||||
|
||||
const connection = this.connections.get(connectionId)
|
||||
if (!connection || connection.status !== 'connected') {
|
||||
logError('Connection not established')
|
||||
throw new Error('Connection not established')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
debug('>>> Calling client.sftp()...')
|
||||
connection.client.sftp((err, sftp) => {
|
||||
if (err) {
|
||||
logError(`SFTP creation failed: ${err.message}`)
|
||||
reject(err)
|
||||
return
|
||||
}
|
||||
|
||||
const sftpId = uuidv4()
|
||||
const session: SFTPSession = {
|
||||
id: sftpId,
|
||||
connectionId,
|
||||
sftp
|
||||
}
|
||||
|
||||
this.sftps.set(sftpId, session)
|
||||
logSuccess(`SFTP session created: ${sftpId}`)
|
||||
resolve(session)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
async sftpListDirectory(sftpId: string, path: string): Promise<any[]> {
|
||||
logHeader('📁 SFTP LIST DIRECTORY')
|
||||
logInfo('sftpId', sftpId)
|
||||
logInfo('path', path)
|
||||
|
||||
const session = this.sftps.get(sftpId)
|
||||
if (!session) {
|
||||
logError('SFTP session not found')
|
||||
throw new Error('SFTP session not found')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
session.sftp.readdir(path, (err, list) => {
|
||||
if (err) {
|
||||
logError(`List directory failed: ${err.message}`)
|
||||
reject(err)
|
||||
return
|
||||
}
|
||||
logSuccess(`Listed ${list.length} items in ${path}`)
|
||||
resolve(list)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
async sftpUpload(
|
||||
sftpId: string,
|
||||
localPath: string,
|
||||
remotePath: string,
|
||||
onProgress?: (transferred: number, total: number) => void
|
||||
): Promise<void> {
|
||||
logHeader('📤 SFTP UPLOAD')
|
||||
logInfo('sftpId', sftpId)
|
||||
logInfo('localPath', localPath)
|
||||
logInfo('remotePath', remotePath)
|
||||
|
||||
const session = this.sftps.get(sftpId)
|
||||
if (!session) {
|
||||
logError('SFTP session not found')
|
||||
throw new Error('SFTP session not found')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
session.sftp.fastPut(localPath, remotePath, {
|
||||
step: (transferred: number, _chunk: number, total: number) => {
|
||||
debug(`Upload progress: ${transferred}/${total}`)
|
||||
if (onProgress) {
|
||||
onProgress(transferred, total)
|
||||
}
|
||||
}
|
||||
}, (err) => {
|
||||
if (err) {
|
||||
logError(`Upload failed: ${err.message}`)
|
||||
reject(err)
|
||||
return
|
||||
}
|
||||
logSuccess(`Upload complete: ${remotePath}`)
|
||||
resolve()
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
async sftpDownload(
|
||||
sftpId: string,
|
||||
remotePath: string,
|
||||
localPath: string,
|
||||
onProgress?: (transferred: number, total: number) => void
|
||||
): Promise<void> {
|
||||
logHeader('📥 SFTP DOWNLOAD')
|
||||
logInfo('sftpId', sftpId)
|
||||
logInfo('remotePath', remotePath)
|
||||
logInfo('localPath', localPath)
|
||||
|
||||
const session = this.sftps.get(sftpId)
|
||||
if (!session) {
|
||||
logError('SFTP session not found')
|
||||
throw new Error('SFTP session not found')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
session.sftp.fastGet(remotePath, localPath, {
|
||||
step: (transferred: number, _chunk: number, total: number) => {
|
||||
debug(`Download progress: ${transferred}/${total}`)
|
||||
if (onProgress) {
|
||||
onProgress(transferred, total)
|
||||
}
|
||||
}
|
||||
}, (err) => {
|
||||
if (err) {
|
||||
logError(`Download failed: ${err.message}`)
|
||||
reject(err)
|
||||
return
|
||||
}
|
||||
logSuccess(`Download complete: ${remotePath}`)
|
||||
resolve()
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
async sftpRename(sftpId: string, oldPath: string, newPath: string): Promise<void> {
|
||||
const session = this.sftps.get(sftpId)
|
||||
if (!session) {
|
||||
throw new Error('SFTP session not found')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
session.sftp.rename(oldPath, newPath, (err) => {
|
||||
if (err) {
|
||||
reject(err)
|
||||
return
|
||||
}
|
||||
resolve()
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
async sftpDelete(sftpId: string, path: string, isDirectory: boolean): Promise<void> {
|
||||
const session = this.sftps.get(sftpId)
|
||||
if (!session) {
|
||||
throw new Error('SFTP session not found')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
if (isDirectory) {
|
||||
session.sftp.rmdir(path, (err) => {
|
||||
if (err) {
|
||||
reject(err)
|
||||
return
|
||||
}
|
||||
resolve()
|
||||
})
|
||||
} else {
|
||||
session.sftp.unlink(path, (err) => {
|
||||
if (err) {
|
||||
reject(err)
|
||||
return
|
||||
}
|
||||
resolve()
|
||||
})
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
async sftpMkdir(sftpId: string, path: string): Promise<void> {
|
||||
const session = this.sftps.get(sftpId)
|
||||
if (!session) {
|
||||
throw new Error('SFTP session not found')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
session.sftp.mkdir(path, (err) => {
|
||||
if (err) {
|
||||
reject(err)
|
||||
return
|
||||
}
|
||||
resolve()
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
// 端口转发
|
||||
// ===== 本地端口转发 =====
|
||||
async forwardLocal(
|
||||
connectionId: string,
|
||||
localPort: number,
|
||||
remoteHost: string,
|
||||
remotePort: number
|
||||
): Promise<{ success: boolean; boundPort: number }> {
|
||||
logHeader('🔀 LOCAL PORT FORWARD')
|
||||
logInfo('connectionId', connectionId)
|
||||
logInfo('localPort', localPort)
|
||||
logInfo('remoteHost', remoteHost)
|
||||
logInfo('remotePort', remotePort)
|
||||
|
||||
const connection = this.connections.get(connectionId)
|
||||
if (!connection || connection.status !== 'connected') {
|
||||
logError('Connection not established')
|
||||
throw new Error('Connection not established')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
debug('>>> Creating local server on port', localPort)
|
||||
const server = net.createServer((socket) => {
|
||||
debug('>>> Local connection received, forwarding to', remoteHost, remotePort)
|
||||
connection.client.forwardOut('127.0.0.1', 0, remoteHost, remotePort, (err, stream) => {
|
||||
if (err) {
|
||||
logError(`ForwardOut failed: ${err.message}`)
|
||||
socket.end()
|
||||
return
|
||||
}
|
||||
socket.pipe(stream).pipe(socket)
|
||||
debug('>>> Pipeline established')
|
||||
})
|
||||
})
|
||||
|
||||
server.listen(localPort, '127.0.0.1', () => {
|
||||
const actualPort = (server.address() as net.AddressInfo).port
|
||||
logSuccess(`Local forward established: 127.0.0.1:${actualPort} -> ${remoteHost}:${remotePort}`)
|
||||
resolve({ success: true, boundPort: actualPort })
|
||||
})
|
||||
|
||||
server.on('error', (err) => {
|
||||
logError(`Server error: ${err.message}`)
|
||||
reject(err)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
// ===== 远程端口转发 =====
|
||||
async forwardRemote(
|
||||
connectionId: string,
|
||||
remotePort: number,
|
||||
localHost: string,
|
||||
localPort: number
|
||||
): Promise<{ success: boolean }> {
|
||||
logHeader('🔀 REMOTE PORT FORWARD')
|
||||
logInfo('connectionId', connectionId)
|
||||
logInfo('remotePort', remotePort)
|
||||
logInfo('localHost', localHost)
|
||||
logInfo('localPort', localPort)
|
||||
|
||||
const connection = this.connections.get(connectionId)
|
||||
if (!connection || connection.status !== 'connected') {
|
||||
logError('Connection not established')
|
||||
throw new Error('Connection not established')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
connection.client.forwardIn('127.0.0.1', remotePort, (err) => {
|
||||
if (err) {
|
||||
logError(`forwardIn failed: ${err.message}`)
|
||||
reject(err)
|
||||
return
|
||||
}
|
||||
logSuccess(`Remote forward established: remote:${remotePort} -> ${localHost}:${localPort}`)
|
||||
resolve({ success: true })
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
// ===== 动态端口转发 (SOCKS5) =====
|
||||
private socksServers: Map<string, net.Server> = new Map()
|
||||
|
||||
async forwardDynamic(
|
||||
connectionId: string,
|
||||
localPort: number
|
||||
): Promise<{ success: boolean; boundPort: number }> {
|
||||
logHeader('🧦 SOCKS5 DYNAMIC FORWARD')
|
||||
logInfo('connectionId', connectionId)
|
||||
logInfo('localPort', localPort)
|
||||
|
||||
const connection = this.connections.get(connectionId)
|
||||
if (!connection || connection.status !== 'connected') {
|
||||
logError('Connection not established')
|
||||
throw new Error('Connection not established')
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
const server = net.createServer((socket) => {
|
||||
debug('>>> SOCKS5 client connected')
|
||||
|
||||
socket.once('data', (data) => {
|
||||
// 解析 SOCKS5 握手
|
||||
if (data.length < 3 || data[0] !== 0x05) {
|
||||
logError('Invalid SOCKS5 handshake')
|
||||
socket.end()
|
||||
return
|
||||
}
|
||||
|
||||
// 发送握手响应 (无认证)
|
||||
socket.write(Buffer.from([0x05, 0x00]))
|
||||
|
||||
socket.once('data', (data) => {
|
||||
// 解析 SOCKS5 请求
|
||||
const req = Buffer.from(data)
|
||||
if (req.length < 10 || req[0] !== 0x05) {
|
||||
logError('Invalid SOCKS5 request')
|
||||
socket.end()
|
||||
return
|
||||
}
|
||||
|
||||
const cmd = req[1]
|
||||
const atyp = req[3]
|
||||
|
||||
if (cmd !== 0x01) { // CONNECT only
|
||||
socket.write(Buffer.from([0x05, 0x07, 0x00, 0x01, 0, 0, 0, 0, 0, 0]))
|
||||
socket.end()
|
||||
return
|
||||
}
|
||||
|
||||
let targetHost: string
|
||||
let targetPort: number
|
||||
let headerLen: number
|
||||
|
||||
if (atyp === 0x01) { // IPv4
|
||||
targetHost = `${req[4]}.${req[5]}.${req[6]}.${req[7]}`
|
||||
targetPort = req.readUInt16BE(8)
|
||||
headerLen = 10
|
||||
} else if (atyp === 0x03) { // Domain
|
||||
const domainLen = req[4]
|
||||
targetHost = req.slice(5, 5 + domainLen).toString('ascii')
|
||||
targetPort = req.readUInt16BE(5 + domainLen)
|
||||
headerLen = 5 + domainLen + 2
|
||||
} else {
|
||||
socket.write(Buffer.from([0x05, 0x08, 0x00, 0x01, 0, 0, 0, 0, 0, 0]))
|
||||
socket.end()
|
||||
return
|
||||
}
|
||||
|
||||
logInfo('SOCKS5 target', `${targetHost}:${targetPort}`)
|
||||
|
||||
// 通过 SSH 连接目标
|
||||
connection.client.forwardOut('127.0.0.1', 0, targetHost, targetPort, (err, stream) => {
|
||||
if (err) {
|
||||
logError(`SOCKS5 forwardOut failed: ${err.message}`)
|
||||
socket.write(Buffer.from([0x05, 0x04, 0x00, 0x01, 0, 0, 0, 0, 0, 0]))
|
||||
socket.end()
|
||||
return
|
||||
}
|
||||
|
||||
// 发送成功响应
|
||||
socket.write(Buffer.from([0x05, 0x00, 0x00, 0x01, 0, 0, 0, 0, 0, 0]))
|
||||
|
||||
// 建立双向管道
|
||||
stream.pipe(socket)
|
||||
socket.pipe(stream)
|
||||
|
||||
// 如果有残留数据，发送给 stream
|
||||
if (req.length > headerLen) {
|
||||
stream.write(req.slice(headerLen))
|
||||
}
|
||||
|
||||
logSuccess(`SOCKS5 tunnel established to ${targetHost}:${targetPort}`)
|
||||
})
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
server.listen(localPort, '127.0.0.1', () => {
|
||||
const actualPort = (server.address() as net.AddressInfo).port
|
||||
this.socksServers.set(`${connectionId}:${actualPort}`, server)
|
||||
logSuccess(`SOCKS5 proxy started on 127.0.0.1:${actualPort}`)
|
||||
resolve({ success: true, boundPort: actualPort })
|
||||
})
|
||||
|
||||
server.on('error', (err) => {
|
||||
logError(`SOCKS5 server error: ${err.message}`)
|
||||
reject(err)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
// 停止 SOCKS5 代理
|
||||
stopDynamicForward(connectionId: string, localPort: number): void {
|
||||
const key = `${connectionId}:${localPort}`
|
||||
const server = this.socksServers.get(key)
|
||||
if (server) {
|
||||
server.close()
|
||||
this.socksServers.delete(key)
|
||||
logSuccess(`SOCKS5 proxy stopped on port ${localPort}`)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user