From 524837627fbc7208078f32b0591c801a20303f78 Mon Sep 17 00:00:00 2001 From: wh1te909 Date: Sat, 3 Dec 2022 01:58:42 -0800 Subject: [PATCH] add env for script checks --- agent/checks.go | 2 +- shared/types.go | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/agent/checks.go b/agent/checks.go index ec44b04..00925d7 100644 --- a/agent/checks.go +++ b/agent/checks.go @@ -169,7 +169,7 @@ type ScriptCheckResult struct { // ScriptCheck runs either bat, powershell or python script func (a *Agent) ScriptCheck(data rmm.Check, r *resty.Client) { start := time.Now() - stdout, stderr, retcode, _ := a.RunScript(data.Script.Code, data.Script.Shell, data.ScriptArgs, data.Timeout, data.Script.RunAsUser, data.Script.EnvVars) + stdout, stderr, retcode, _ := a.RunScript(data.Script.Code, data.Script.Shell, data.ScriptArgs, data.Timeout, data.Script.RunAsUser, data.EnvVars) payload := ScriptCheckResult{ ID: data.CheckPK, diff --git a/shared/types.go b/shared/types.go index f857ccb..5ff983d 100644 --- a/shared/types.go +++ b/shared/types.go @@ -163,6 +163,7 @@ type Check struct { Disk string `json:"disk"` IP string `json:"ip"` ScriptArgs []string `json:"script_args"` + EnvVars []string `json:"env_vars"` Timeout int `json:"timeout"` ServiceName string `json:"svc_name"` PassStartPending bool `json:"pass_if_start_pending"`