Function executeSSHCommand

apps/sim/app/api/tools/ssh/utils.ts:182–234

(client: Client, command: string)

 * Execute a command on the SSH connection
 */
export function executeSSHCommand(client: Client, command: string): Promise<SSHCommandResult> {
  return new Promise((resolve, reject) => {
    client.exec(command, (err, stream) => {
      if (err) {
        reject(err)
        return
      }

      let stdout = ''
      let stderr = ''
      let stdoutBytes = 0
      let stderrBytes = 0
      let stdoutTruncated = false
      let stderrTruncated = false

      stream.on('close', (code: number) => {
        resolve({
          stdout: stdoutTruncated
            ? `${stdout.trim()}\n[output truncated: exceeded 16MB limit]`
            : stdout.trim(),
          stderr: stderrTruncated
            ? `${stderr.trim()}\n[stderr truncated: exceeded 16MB limit]`
            : stderr.trim(),
          exitCode: code ?? -1,
        })
      })

      stream.on('data', (data: Buffer) => {
        const remaining = MAX_OUTPUT_BYTES - stdoutBytes
        if (remaining <= 0) {
          stdoutTruncated = true
          return
        }
        const chunk = data.subarray(0, remaining)
        stdout += chunk.toString()
        stdoutBytes += chunk.length
        if (data.length > remaining) stdoutTruncated = true
      })

      stream.stderr.on('data', (data: Buffer) => {
        const remaining = MAX_OUTPUT_BYTES - stderrBytes
        if (remaining <= 0) {
          stderrTruncated = true
          return
        }
        const chunk = data.subarray(0, remaining)
        stderr += chunk.toString()
        stderrBytes += chunk.length
        if (data.length > remaining) stderrTruncated = true
      })
    })
  })
}

/**
 * Sanitize command input to prevent command injection