cabele/rmmagent
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Error Handling for NewAgentConfig

Browse source
This commit is contained in:
redanthrax 2022-08-19 10:56:56 -07:00
parent 89d7ec8242
commit 87c035cd39
3 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

6
agent/agent_unix.go
View file

@ -131,7 +131,7 @@ func (a *Agent) osString() string {
return fmt.Sprintf("%s %s %s %s", strings.Title(h.Platform), h.PlatformVersion, h.KernelArch, h.KernelVersion)
}
func NewAgentConfig() *rmm.AgentConfig {
func NewAgentConfig() (*rmm.AgentConfig, error) {
viper.SetConfigName("tacticalagent")
viper.SetConfigType("json")
viper.AddConfigPath("/etc/")
@ -139,7 +139,7 @@ func NewAgentConfig() *rmm.AgentConfig {
err := viper.ReadInConfig()
if err != nil {
return &rmm.AgentConfig{}
return &rmm.AgentConfig{}, err
}
agentpk := viper.GetString("agentpk")
@ -159,7 +159,7 @@ func NewAgentConfig() *rmm.AgentConfig {
NatsProxyPort: viper.GetString("natsproxyport"),
NatsStandardPort: viper.GetString("natsstandardport"),
}
return ret
return ret, nil
}
func (a *Agent) RunScript(code string, shell string, args []string, timeout int, runasuser bool) (stdout, stderr string, exitcode int, e error) {