dragonheim/gagent
dragonheim/gagent
1
0
Fork 0
mirror of https://github.com/dragonheim/gagent.git synced 2025-04-26 02:18:59 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: was using struct member for two different things.

Browse source
This commit is contained in:
James Wells 2023-03-25 09:15:27 -07:00
parent 7a9225ce86
commit db351f3892
Signed by: jwells
GPG key ID: 73196D10B8E65666
5 changed files with 18 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

7
internal/client/client.go
View file

@ -8,6 +8,7 @@ import (
log "log"
os "os"
regexp "regexp"
strconv "strconv"
strings "strings"
sync "sync"
time "time"
@ -33,9 +34,9 @@ func Main(wg *sync.WaitGroup, config gstructs.GagentConfig) {
var err error
if config.CMode {
agent.ScriptCode, err = ioutil.ReadFile(config.File)
agent.ScriptCode, err = ioutil.ReadFile(config.Agent)
if err != nil {
log.Printf("[ERROR] No such file or directory: %s", config.File)
log.Printf("[ERROR] No such file or directory: %s", config.Agent)
os.Exit(4)
}
log.Printf("[DEBUG] Agent file contents: \n----- -----\n%s\n----- -----\n", agent.ScriptCode)
@ -56,7 +57,7 @@ func Main(wg *sync.WaitGroup, config gstructs.GagentConfig) {
if config.Routers[key].ClientPort != 0 {
rport = config.Routers[key].ClientPort
}
connectString := fmt.Sprintf("tcp://%s:%d", config.Routers[key].RouterAddr, rport)
connectString := "tcp://" + config.Routers[key].RouterAddr + ":" + strconv.Itoa(rport)
wg.Add(1)
go sendAgent(wg, config.UUID, connectString, agent)