bam/auto-boot-ollama-host
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add debug logging for SSH command execution

Browse Source 
Introduce a function to check the DEBUG environment variable and adjust logging behavior accordingly. When DEBUG is enabled, full SSH command details are logged; otherwise, only the command itself is logged to prevent exposure of sensitive information.
This commit is contained in:
Bastian (BaM)
2025-09-14 20:31:24 +02:00
parent e8b11d43da
commit 76ddedbf14
1 changed files with 26 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
scripts/ssh.lua
View File

@@ -5,6 +5,12 @@ local utils = require("utils")
local ssh_module = {} local ssh_module = {}
-- Check if DEBUG environment variable is set to "true"
local function is_debug()
local debug_env = os.getenv("DEBUG")
return debug_env and (string.lower(debug_env) == "true" or debug_env == "1")
end
-- Execute a remote command over SSH -- Execute a remote command over SSH
-- Signature: ssh(command, user, host, port, identity_file) -- Signature: ssh(command, user, host, port, identity_file)
function ssh_module.execute(command, user, host, port, identity_file) function ssh_module.execute(command, user, host, port, identity_file)
@@ -35,7 +41,7 @@ function ssh_module.execute(command, user, host, port, identity_file)
"-o", "UserKnownHostsFile=/root/.ssh/known_hosts", "-o", "UserKnownHostsFile=/root/.ssh/known_hosts",
"-o", "StrictHostKeyChecking=yes", "-o", "StrictHostKeyChecking=yes",
} }
if identity_file ~= "" then if identity_file ~= "" then
table.insert(pieces, "-i") table.insert(pieces, "-i")
table.insert(pieces, identity_file) table.insert(pieces, identity_file)
@@ -54,8 +60,14 @@ function ssh_module.execute(command, user, host, port, identity_file)
end end
local full = join(pieces) local full = join(pieces)
utils.log("SSH exec: " .. full)
-- Log based on DEBUG environment variable
if is_debug() then
utils.log("SSH exec: " .. full)
else
utils.log("SSH exec: " .. "'" .. command:gsub("'", "'\\''") .. "'")
end
local ok, reason, code = os.execute(full) local ok, reason, code = os.execute(full)
if ok == true or ok == 0 then if ok == true or ok == 0 then
utils.log("SSH command completed successfully") utils.log("SSH command completed successfully")
@@ -89,7 +101,7 @@ function ssh_module.execute_with_output(command, user, host, port, identity_file
"-o", "UserKnownHostsFile=/root/.ssh/known_hosts", "-o", "UserKnownHostsFile=/root/.ssh/known_hosts",
"-o", "StrictHostKeyChecking=yes", "-o", "StrictHostKeyChecking=yes",
} }
if identity_file ~= "" then if identity_file ~= "" then
table.insert(pieces, "-i") table.insert(pieces, "-i")
table.insert(pieces, identity_file) table.insert(pieces, identity_file)
@@ -107,17 +119,23 @@ function ssh_module.execute_with_output(command, user, host, port, identity_file
end end
local full = join(pieces) local full = join(pieces)
utils.log("SSH exec (with output): " .. full)
-- Log based on DEBUG environment variable
if is_debug() then
utils.log("SSH exec (with output): " .. full)
else
utils.log("SSH exec (with output): " .. "'" .. command:gsub("'", "'\\''") .. "'")
end
-- Use io.popen to capture output -- Use io.popen to capture output
local fh = io.popen(full, "r") local fh = io.popen(full, "r")
if not fh then if not fh then
return false, "", "Failed to open SSH command" return false, "", "Failed to open SSH command"
end end
local output = fh:read("*a") local output = fh:read("*a")
local success, reason, code = fh:close() local success, reason, code = fh:close()
if success then if success then
utils.log("SSH command completed successfully with output") utils.log("SSH command completed successfully with output")
return true, output, nil return true, output, nil