ing

auth/info/info.go

@@ -0,0 +1,122 @@
+package info
+
+import (
+	"bytes"
+	"encoding/base64"
+	"encoding/json"
+	"errors"
+	"net"
+
+	"git.loafle.net/commons_go/util/net/gateway"
+	noauthprobeM "git.loafle.net/overflow/overflow_commons_go/modules/noauthprobe/model"
+	"git.loafle.net/overflow/overflow_probes/config"
+	"github.com/shirou/gopsutil/host"
+)
+
+func GetRegistHeader() (string, error) {
+	var err error
+	nap := noauthprobeM.NoAuthProbe{
+		APIKey: config.Config.Central.APIKey,
+	}
+
+	var napd *noauthprobeM.NoAuthProbeDescription
+	if napd, err = getDescription(); nil != err {
+		return "", err
+	}
+
+	var buf []byte
+	if buf, err = json.Marshal(napd); nil != err {
+		return "", err
+	}
+	nap.Description = string(buf)
+
+	if buf, err = json.Marshal(nap); nil != err {
+		return "", err
+	}
+
+	enc := base64.StdEncoding.EncodeToString(buf)
+
+	return enc, nil
+}
+
+func getDescription() (*noauthprobeM.NoAuthProbeDescription, error) {
+	var err error
+	napd := &noauthprobeM.NoAuthProbeDescription{}
+
+	if napd.Host, err = getHost(); nil != err {
+		return nil, err
+	}
+
+	if napd.Network, err = getNetwork(); nil != err {
+		return nil, err
+	}
+
+	return napd, nil
+}
+
+func getHost() (*noauthprobeM.NoAuthProbeDescriptionHost, error) {
+	if i, err := host.Info(); nil == err {
+		h := &noauthprobeM.NoAuthProbeDescriptionHost{}
+
+		h.Name = i.Hostname
+		h.OS = i.OS
+		h.Platform = i.Platform
+		h.PlatformFamily = i.PlatformFamily
+		h.KernelVersion = i.KernelVersion
+		h.HostID = i.HostID
+
+		return h, nil
+	} else {
+		return nil, err
+	}
+}
+
+func getNetwork() (*noauthprobeM.NoAuthProbeDescriptionNetwork, error) {
+	var ip net.IP
+	var iface string
+	var err error
+	if ip, iface, err = gateway.DiscoverGateway(); nil != err {
+		return nil, err
+	}
+
+	interfaces, err := net.Interfaces()
+	if err != nil {
+		return nil, err
+	}
+
+	idx := -1
+
+	for _idx, i := range interfaces {
+		if i.Name == iface {
+			idx = _idx
+			break
+		}
+	}
+
+	if -1 == idx {
+		return nil, errors.New("Interface of gateway is not exist")
+	}
+
+	n := &noauthprobeM.NoAuthProbeDescriptionNetwork{}
+
+	i := interfaces[idx]
+
+	n.Name = i.Name
+	n.MacAddress = i.HardwareAddr.String()
+	n.Gateway = ip.String()
+
+	if addrs, err := i.Addrs(); nil == err {
+		var buffer bytes.Buffer
+		for _idx, a := range addrs {
+			if 0 < _idx {
+				buffer.WriteString("|")
+			}
+			buffer.WriteString(a.String())
+		}
+		n.Address = buffer.String()
+	} else {
+		return nil, err
+	}
+
+	return n, nil
+}