Merge pull request #55 from ddosify/develop

Develop

datastore/backend.go

@@ -11,6 +11,7 @@ import (
 	"os"
 	"sort"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/ddosify/alaz/config"
@@ -36,6 +37,8 @@ var NodeID string
 
 // set from ldflags
 var tag string
+var kernelVersion string
+var cloudProvider CloudProvider
 
 func init() {
 	MonitoringID = os.Getenv("MONITORING_ID")
@@ -52,6 +55,67 @@ func init() {
 		log.Logger.Fatal().Msg("tag is not set")
 	}
 	log.Logger.Info().Str("tag", tag).Msg("alaz tag")
+
+	kernelVersion = extractKernelVersion()
+	cloudProvider = getCloudProvider()
+}
+
+func extractKernelVersion() string {
+	// Path to the /proc/version file
+	filePath := "/proc/version"
+	file, err := os.Open(filePath)
+	if err != nil {
+		log.Logger.Fatal().AnErr("error", err).Msgf("Unable to open file %s", filePath)
+	}
+
+	// Read the content of the file
+	content, err := io.ReadAll(file)
+	if err != nil {
+		log.Logger.Fatal().AnErr("error", err).Msgf("Unable to read file %s", filePath)
+	}
+
+	// Convert the content to a string
+	versionInfo := string(content)
+
+	// Split the versionInfo string into lines
+	lines := strings.Split(versionInfo, "\n")
+
+	// Extract the kernel version from the first line
+	// Assuming the kernel version is the first word in the first line
+	if len(lines) > 0 {
+		fields := strings.Fields(lines[0])
+		if len(fields) > 2 {
+			return fields[2]
+		}
+	}
+
+	return "Unable to extract kernel version"
+}
+
+type CloudProvider string
+
+const (
+	CloudProviderAWS          CloudProvider = "AWS"
+	CloudProviderGCP          CloudProvider = "GCP"
+	CloudProviderAzure        CloudProvider = "Azure"
+	CloudProviderDigitalOcean CloudProvider = "DigitalOcean"
+	CloudProviderUnknown      CloudProvider = ""
+)
+
+func getCloudProvider() CloudProvider {
+	if vendor, err := os.ReadFile("/sys/class/dmi/id/board_vendor"); err == nil {
+		switch strings.TrimSpace(string(vendor)) {
+		case "Amazon EC2":
+			return CloudProviderAWS
+		case "Google":
+			return CloudProviderGCP
+		case "Microsoft Corporation":
+			return CloudProviderAzure
+		case "DigitalOcean":
+			return CloudProviderDigitalOcean
+		}
+	}
+	return CloudProviderUnknown
 }
 
 var resourceBatchSize int64 = 50
@@ -527,7 +591,7 @@ func (b *BackendDS) PersistContainer(c Container, eventType string) error {
 	return nil
 }
 
-func (b *BackendDS) SendHealthCheck(ebpf bool, metrics bool) {
+func (b *BackendDS) SendHealthCheck(ebpf bool, metrics bool, k8sVersion string) {
 	t := time.NewTicker(10 * time.Second)
 	defer t.Stop()
 
@@ -546,6 +610,15 @@ func (b *BackendDS) SendHealthCheck(ebpf bool, metrics bool) {
 				EbpfEnabled:    ebpf,
 				MetricsEnabled: metrics,
 			},
+			Telemetry: struct {
+				KernelVersion string `json:"kernel_version"`
+				K8sVersion    string `json:"k8s_version"`
+				CloudProvider string `json:"cloud_provider"`
+			}{
+				KernelVersion: kernelVersion,
+				K8sVersion:    k8sVersion,
+				CloudProvider: string(cloudProvider),
+			},
 		}
 	}
 

datastore/payload.go

@@ -13,6 +13,11 @@ type HealthCheckPayload struct {
 		EbpfEnabled    bool `json:"ebpf"`
 		MetricsEnabled bool `json:"metrics"`
 	} `json:"alaz_info"`
+	Telemetry struct {
+		KernelVersion string `json:"kernel_version"`
+		K8sVersion    string `json:"k8s_version"`
+		CloudProvider string `json:"cloud_provider"`
+	} `json:"telemetry"`
 }
 
 type EventPayload struct {

k8s/informer.go

@@ -41,6 +41,8 @@ const (
 	DELETE = "Delete"
 )
 
+var k8sVersion string
+
 type K8sCollector struct {
 	ctx              context.Context
 	informersFactory informers.SharedInformerFactory
@@ -178,6 +180,13 @@ func NewK8sCollector(parentCtx context.Context) (*K8sCollector, error) {
 		return nil, fmt.Errorf("unable to create clientset: %w", err)
 	}
 
+	version, err := clientset.ServerVersion()
+	if err != nil {
+		return nil, fmt.Errorf("unable to get k8s server version: %w", err)
+	}
+
+	k8sVersion = version.String()
+
 	factory := informers.NewSharedInformerFactory(clientset, 0)
 
 	collector := &K8sCollector{
@@ -196,6 +205,10 @@ func NewK8sCollector(parentCtx context.Context) (*K8sCollector, error) {
 	return collector, nil
 }
 
+func (k *K8sCollector) GetK8sVersion() string {
+	return k8sVersion
+}
+
 func (k *K8sCollector) close() {
 	log.Logger.Info().Msg("k8sCollector closing...")
 	close(k.stopper) // stop informers

main.go

@@ -35,13 +35,15 @@ func main() {
 
 	var k8sCollector *k8s.K8sCollector
 	kubeEvents := make(chan interface{}, 1000)
+	var k8sVersion string
 	if os.Getenv("K8S_COLLECTOR_ENABLED") != "false" {
 		// k8s collector
 		var err error
 		k8sCollector, err = k8s.NewK8sCollector(ctx)
 		if err != nil {
 			panic(err)
 		}
+		k8sVersion = k8sCollector.GetK8sVersion()
 		go k8sCollector.Init(kubeEvents)
 	}
 
@@ -55,7 +57,7 @@ func main() {
 		MetricsExport:         metricsEnabled,
 		MetricsExportInterval: 10,
 	})
-	go dsBackend.SendHealthCheck(ebpfEnabled, metricsEnabled)
+	go dsBackend.SendHealthCheck(ebpfEnabled, metricsEnabled, k8sVersion)
 
 	// deploy ebpf programs
 	var ec *ebpf.EbpfCollector