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

Converted TCL list style and added hint parser.

Browse source
This commit is contained in:
James Wells 2021-11-08 06:58:55 -08:00
parent de41740bac
commit abba1972d1
Signed by: jwells
GPG key ID: 73196D10B8E65666
6 changed files with 100 additions and 63 deletions
Download patch file Download diff file Expand all files Collapse all files

2
README.md
View file

@ -18,7 +18,7 @@ Imagine, for a moment, that you are on Mars and need to perform a data search in
1 : ###################
2 : ### Hello Earth ###
3 : ###################
4 : array set GHINT ["thermal measurements" "gravity measurements" "gravity fluctuations"]
4 : set GHINT [split "thermal measurements, gravity measurements, gravity fluctuations" ,]
5 : proc hello_earth {} {
6 : puts "Hello Earth, does localized tempurature variations alter specific gravity?"
7 : }

2
assets/examples/hello-earth.tcl
View file

@ -1,7 +1,7 @@
###################
### Hello Earth ###
###################
array set GHINT ["thermal measurements" "gravity measurements" "gravity fluctuations"]
set GHINT [split "thermal measurements, gravity measurements, gravity fluctuations" ,]
proc hello_earth {} {
puts "Hello Earth, does localized tempurature variations alter specific gravity?"
}

82
cmd/gagent/main.go
View file

@ -30,28 +30,31 @@ import (
)
var (
semVER = "0.0.2"
semVER = "0.0.3"
)
var (
wg sync.WaitGroup
)
var exitCodes = struct {
m map[string]int
}{m: map[string]int{}}
/*
* Exit Codes
* 0 Success
* 1 Configuration file is missing or unreadable
* 2 Setup failed
* 3 Invalid mode of operation
* 4 Agent file is missing or unreadable
* 5 Agent is missing tags
* 6 No routers defined
* 7 No workers defined
* 8 Agent not defined
* 9 Agent hints / tags not defined
* 10 Router not connected
*/
var config gstructs.GagentConfig
var agent gstructs.AgentDetails
func main() {
filter := &logutils.LevelFilter{
Levels: []logutils.LogLevel{"DEBUG", "INFO", "WARN", "ERROR"},
MinLevel: logutils.LogLevel("DEBUG"),
Writer: os.Stderr,
}
log.SetOutput(filter)
log.Printf("[DEBUG] Configuration is %v\n", config)
switch config.Mode {
@ -60,7 +63,7 @@ func main() {
if len(config.Routers) == 0 {
log.Printf("[ERROR] No routers defined.\n")
os.Exit(exitCodes.m["NO_ROUTERS_DEFINED"])
os.Exit(6)
}
wg.Add(1)
@ -71,7 +74,7 @@ func main() {
if len(config.Workers) == 0 {
log.Printf("[ERROR] No workers defined.\n")
os.Exit(exitCodes.m["NO_WORKERS_DEFINED"])
os.Exit(7)
}
wg.Add(1)
@ -82,7 +85,7 @@ func main() {
if len(config.Routers) == 0 {
log.Printf("[ERROR] No routers defined.\n")
os.Exit(exitCodes.m["NO_ROUTERS_DEFINED"])
os.Exit(6)
}
wg.Add(1)
@ -96,38 +99,27 @@ func main() {
default:
log.Printf("[ERROR] Unknown operating mode, exiting.\n")
os.Exit(exitCodes.m["INVALID_MODE"])
os.Exit(3)
}
wg.Wait()
os.Exit(exitCodes.m["SUCCESS"])
os.Exit(0)
}
func init() {
var err error
// var err error
autorestart.StartWatcher()
filter := &logutils.LevelFilter{
Levels: []logutils.LogLevel{"DEBUG", "INFO", "WARN", "ERROR"},
MinLevel: logutils.LogLevel("DEBUG"),
Writer: os.Stderr,
}
log.SetOutput(filter)
http.Handle("/metrics", promhttp.Handler())
/*
* Start Prometheus metrics exporter
*/
go http.ListenAndServe(fmt.Sprintf("%s:%d", config.ListenAddr, config.ClientPort), nil)
/*
* Initialize the exit codes
*/
exitCodes.m["SUCCESS"] = 0
exitCodes.m["INVALID_MODE"] = 1
exitCodes.m["CONFIG_FILE_MISSING"] = 2
exitCodes.m["NO_ROUTERS_DEFINED"] = 3
exitCodes.m["NO_WORKERS_DEFINED"] = 4
exitCodes.m["AGENT_NOT_DEFINED"] = 5
exitCodes.m["AGENT_LOAD_FAILED"] = 6
exitCodes.m["AGENT_MISSING_TAGS"] = 7
exitCodes.m["AGENT_NOT_DEFINED"] = 8
/*
* Initialize the configuration
*/
@ -152,6 +144,12 @@ func init() {
*/
config.ListenAddr = "0.0.0.0"
/*
* By default, G'Agent will use port 9101 or monitoring via prometheus.
* It can be overridden in the configuration file by setting clientport
*/
config.MonitorPort = 9101
/*
* By default, G'Agent client will use port 35571 to communicate with the
* routers, but you can override it by setting the clientport in the
@ -223,10 +221,10 @@ func init() {
config.File = opts["--config"].(string)
}
err = hclsimple.DecodeFile(config.File, nil, &config)
err := hclsimple.DecodeFile(config.File, nil, &config)
if err != nil && opts["setup"] == false {
log.Printf("[ERROR] Failed to load configuration file: %s.\n", config.File)
os.Exit(exitCodes.m["CONFIG_FILE_MISSING"])
os.Exit(1)
}
/*
@ -239,7 +237,7 @@ func init() {
config.Mode = "client"
if opts["--agent"] == nil {
log.Printf("[ERROR] Agent file not specified")
os.Exit(exitCodes.m["AGENT_NOT_DEFINED"])
os.Exit(8)
} else {
config.File = opts["--agent"].(string)
}
@ -263,4 +261,10 @@ func init() {
}
log.Printf("[DEBUG] Config is %v\n", config)
/*
* Start Prometheus metrics exporter
*/
go http.ListenAndServe(fmt.Sprintf("%s:%d", config.ListenAddr, config.MonitorPort), nil)
}

46
internal/client/client.go
View file

@ -6,6 +6,8 @@ import (
ioutil "io/ioutil"
log "log"
os "os"
regexp "regexp"
strings "strings"
sync "sync"
time "time"
@ -32,12 +34,14 @@ func Main(wg *sync.WaitGroup, config gstructs.GagentConfig) {
agent.ScriptCode, err = ioutil.ReadFile(config.File)
if err != nil {
log.Printf("[ERROR] No such file or directory: %s", config.File)
os.Exit(6)
os.Exit(4)
}
agent.Shasum = fmt.Sprintf("%x", sha.Sum256(agent.ScriptCode))
agent.Status = "loaded"
log.Printf("[DEBUG] SHA256 of Agent file: %s", agent.Shasum)
log.Printf("[INFO] SHA256 of Agent file: %s", agent.Shasum)
log.Printf("[DEBUG] Agent file contents: \n%s\n", agent.ScriptCode)
}
agent.Hints = getTagsFromHints(agent)
for key := range config.Routers {
/*
@ -51,12 +55,27 @@ func Main(wg *sync.WaitGroup, config gstructs.GagentConfig) {
wg.Add(1)
go sendAgent(wg, config.UUID, connectString, agent.ScriptCode)
time.Sleep(10 * time.Millisecond)
}
}
/*
* Parse Agent file for GHINT data to populate the G'Agent hints
*/
func getTagsFromHints(agent gstructs.AgentDetails) []string {
var tags []string
re := regexp.MustCompile(`\s*set\s+GHINT\s*\[\s*split\s*"(?P<Hints>.+)"\s*\,\s*\]`)
res := re.FindStringSubmatch(string(agent.ScriptCode))
if len(res) < 1 {
log.Printf("[ERROR] Agent is missing GHINT tags")
os.Exit(4)
}
tags = strings.Split(res[1], ",")
log.Printf("[DEBUG] G'Agent hints: %v\n", tags)
return tags
}
func sendAgent(wg *sync.WaitGroup, uuid string, connectString string, agent []byte) {
log.Printf("[DEBUG] Attempting to connect to %s\n", connectString)
defer wg.Done()
var mu sync.Mutex
@ -66,10 +85,23 @@ func sendAgent(wg *sync.WaitGroup, uuid string, connectString string, agent []by
defer sock.Close()
sock.SetIdentity(uuid)
sock.Connect(connectString)
log.Printf("[DEBUG] Attempting to connect to %s\n", connectString)
err := sock.Connect(connectString)
if err != nil {
log.Printf("[ERROR] Failed to connect to %s\n", connectString)
os.Exit(10)
}
log.Printf("[DEBUG] Start sending agent...\n")
sock.SendMessage(agent)
log.Printf("[DEBUG] End sending agent...\n")
status, err := sock.SendMessage(agent)
if err != nil {
log.Printf("[ERROR] Failed to send agent to router\n")
// os.Exit(11)
return
}
log.Printf("[DEBUG] Agent send status: %d\n", status)
mu.Unlock()
time.Sleep(10 * time.Second)
}

29
internal/gstructs/gstructs.go
View file

@ -2,19 +2,20 @@ package gstructs
// GagentConfig is the primary construct used by all modes
type GagentConfig struct {
Name string `hcl:"name,optional"`
Mode string `hcl:"mode,attr"`
UUID string `hcl:"uuid,optional"`
ListenAddr string `hcl:"listenaddr,optional"`
ClientPort int64 `hcl:"clientport,optional"`
RouterPort int64 `hcl:"routerport,optional"`
WorkerPort int64 `hcl:"workerport,optional"`
Clients []*ClientDetails `hcl:"client,block"`
Routers []*RouterDetails `hcl:"router,block"`
Workers []*WorkerDetails `hcl:"worker,block"`
Version string
File string
CMode bool
Name string `hcl:"name,optional"`
Mode string `hcl:"mode,attr"`
UUID string `hcl:"uuid,optional"`
MonitorPort int `hcl:"monitorport,optional"`
ListenAddr string `hcl:"listenaddr,optional"`
ClientPort int64 `hcl:"clientport,optional"`
RouterPort int64 `hcl:"routerport,optional"`
WorkerPort int64 `hcl:"workerport,optional"`
Clients []*ClientDetails `hcl:"client,block"`
Routers []*RouterDetails `hcl:"router,block"`
Workers []*WorkerDetails `hcl:"worker,block"`
Version string
File string
CMode bool
}
/*
@ -127,5 +128,5 @@ type AgentDetails struct {
Shasum string `hcl:"shasum"`
Status string `hcl:"status"`
ScriptCode []byte
Hints []*string
Hints []string
}

2
internal/setup/setup.go
View file

@ -12,7 +12,7 @@ import (
)
func Main(wg *sync.WaitGroup, config gs.GagentConfig) {
log.Printf("[INFO] setup mode\n")
log.Printf("[INFO] Starting setup\n")
defer wg.Done()
f := hclwrite.NewEmptyFile()