Merge pull request #33 from cloudradar-monitoring/test-hub

-t flag to test the HUB connection & creds

cmd/cagent/main.go

@@ -77,6 +77,7 @@ func main() {
 
 	serviceUninstallPtr := flag.Bool("u", false, fmt.Sprintf("stop and uninstall the system service(%s)", systemManager.String()))
 	printConfigPtr := flag.Bool("p", false, "print the active config")
+	testConfigPtr := flag.Bool("t", false, "test the HUB config")
 	versionPtr := flag.Bool("version", false, "show the cagent version")
 
 	flag.Parse()
@@ -108,6 +109,18 @@ func main() {
 		return
 	}
 
+	if *testConfigPtr {
+		err := ca.TestHub()
+		if err != nil {
+			fmt.Printf("Cagent HUB test failed: %s\n", err.Error())
+			os.Exit(1)
+			return
+		}
+
+		fmt.Printf("HUB connection test succeed and credentials are correct!\n")
+		return
+	}
+
 	var osNotice string
 
 	/*if runtime.GOOS == "windows" && !cagent.CheckIfRawICMPAvailable() {
@@ -141,7 +154,7 @@ func main() {
 		if serviceInstallPtr != nil && *serviceInstallPtr || serviceInstallUserPtr != nil && *serviceInstallUserPtr != "" {
 			fmt.Println(" ****** Before start you need to set 'hub_url' config param at ", *cfgPathPtr)
 		} else {
-			fmt.Println("Missing output file flag(-o) hub_url in config")
+			fmt.Println("Missing output file flag(-o) or hub_url in the config")
 			flag.PrintDefaults()
 			return
 		}

handler.go

@@ -6,6 +6,7 @@ import (
 	"crypto/tls"
 	"encoding/json"
 	"errors"
+	"fmt"
 	"io/ioutil"
 	"net/http"
 	"net/url"
@@ -15,7 +16,6 @@ import (
 	"strings"
 	"time"
 
-	"github.com/shirou/gopsutil/load"
 	log "github.com/sirupsen/logrus"
 )
 
@@ -51,10 +51,41 @@ func (ca *Cagent) initHubHttpClient() {
 	}
 }
 
+func (ca *Cagent) TestHub() error {
+	if ca.HubURL == "" {
+		return fmt.Errorf("please set the hub_url config param")
+	}
+
+	ca.initHubHttpClient()
+	req, err := http.NewRequest("HEAD", ca.HubURL, nil)
+	if err != nil {
+		return err
+	}
+
+	req.Header.Add("User-Agent", ca.userAgent())
+	if ca.HubUser != "" {
+		req.SetBasicAuth(ca.HubUser, ca.HubPassword)
+	}
+
+	resp, err := ca.hubHttpClient.Do(req)
+	if err != nil {
+		return fmt.Errorf("unable to connect. %s. If you have a proxy or firewall, it may be blocking the connection", err.Error())
+	}
+
+	if resp.StatusCode == 401 && ca.HubUser == "" {
+		return fmt.Errorf("unable to authorise without credentials. Please set hub_user & hub_password in the config")
+	} else if resp.StatusCode == 401 && ca.HubUser != "" {
+		return fmt.Errorf("unable to authorise with the provided credentials. Please correct the hub_user & hub_password in the config")
+	} else if resp.StatusCode < 200 || resp.StatusCode >= 400 {
+		return fmt.Errorf("got bad response status: %d, %s. If you have a proxy or firewall it may be blocking the connection", resp.StatusCode, resp.Status)
+	}
+
+	return nil
+}
+
 func (ca *Cagent) PostResultsToHub(result Result) error {
 	ca.initHubHttpClient()
 
-	load.Avg()
 	b, err := json.Marshal(result)
 	if err != nil {
 		return err

pkg-scripts/postinstall.sh

@@ -1,2 +1,3 @@
 #!/bin/sh
-/usr/bin/cagent -s cagent -c /etc/cagent/cagent.conf
+/usr/bin/cagent -s cagent -c /etc/cagent/cagent.conf
+/usr/bin/cagent -t || true