Fixes snmp sender

telenornms · Aug 10, 2023 · 8ad7e7c · 8ad7e7c
1 parent c124124
commit 8ad7e7c
Show file tree

Hide file tree

Showing 2 changed files with 80 additions and 92 deletions.
diff --git a/sender/snmp.go b/sender/snmp.go
@@ -22,6 +22,9 @@ type SNMP struct {
 	g         *gosnmp.GoSNMP
 }
 
+/*
+ * SNMP trap sender
+ */
 func (x *SNMP) init() {
 	var version gosnmp.SnmpVersion
 	if x.Version == "2c" {
@@ -60,52 +63,56 @@ func (x *SNMP) Send(c *skogul.Container) error {
 		return x.err
 	}
 
-	pdutypes := []gosnmp.SnmpPDU{}
-
-	for _, m := range c.Metrics {
-		for j, i := range m.Data {
-			var pdutype gosnmp.SnmpPDU
-
-			pduName := fmt.Sprintf("%s", x.Oidmap[j])
-
-			switch reflect.TypeOf(i).Kind() {
-			case reflect.String:
-				pdutype = gosnmp.SnmpPDU{
-					Value: i,
-					Name:  pduName,
-					Type:  gosnmp.OctetString,
-				}
-			case reflect.Bool:
-				pdutype = gosnmp.SnmpPDU{
-					Value: i,
-					Name:  pduName,
-					Type:  gosnmp.Boolean,
-				}
-			case reflect.Int:
-				pdutype = gosnmp.SnmpPDU{
-					Value: i,
-					Name:  pduName,
-					Type:  gosnmp.Integer,
-				}
-			default:
+	pdutypes := []gosnmp.SnmpPDU{
+		{
+			Value: ".1.3.6.1.4.1.12748.2023.0.888",
+			Name:  ".1.3.6.1.6.3.1.1.4.1.0",
+			Type:  gosnmp.ObjectIdentifier,
+		},
+	}
+
+	m := c.Metrics[0]
+
+	for j, i := range m.Data {
+		var pdutype gosnmp.SnmpPDU
+
+		pduName := fmt.Sprintf("%s", x.Oidmap[j])
+
+		switch reflect.TypeOf(i).Kind() {
+		case reflect.String:
+			pdutype = gosnmp.SnmpPDU{
+				Value: i,
+				Name:  pduName,
+				Type:  gosnmp.OctetString,
+			}
+		case reflect.Bool:
+			pdutype = gosnmp.SnmpPDU{
+				Value: i,
+				Name:  pduName,
+				Type:  gosnmp.Boolean,
 			}
-			pdutypes = append(pdutypes, pdutype)
-
-			trap := gosnmp.SnmpTrap{}
-			trap.Variables = pdutypes
-			trap.IsInform = false
-			trap.Enterprise = "no"
-			trap.AgentAddress = "localhost"
-			trap.GenericTrap = 1
-			trap.SpecificTrap = 1
-			_, err := x.g.SendTrap(trap)
-
-			if err != nil {
-				x.err = err
+		case reflect.Float64:
+			k := int(i.(float64))
+			pdutype = gosnmp.SnmpPDU{
+				Value: k,
+				Name:  pduName,
+				Type:  gosnmp.Integer,
 			}
+		default:
 		}
+		pdutypes = append(pdutypes, pdutype)
+	}
+
+	trap := gosnmp.SnmpTrap{}
+	trap.Variables = pdutypes
+	trap.IsInform = false
+	trap.Enterprise = "no"
+	trap.AgentAddress = "localhost"
+	_, err := x.g.SendTrap(trap)
 
+	if err != nil {
+		x.err = err
 	}
 
-	return nil
+	return x.err
 }
diff --git a/sender/snmp_test.go b/sender/snmp_test.go
@@ -16,60 +16,41 @@ import (
 
 func TestSnmpSenderTest(t *testing.T) {
 	t.Skip()
-	c, _ := loadJsonFile(t, "ble ble")
+	c, _ := loadJsonFile(t, "")
 
 	sconf := fmt.Sprintln(`
-{
-  "receivers": {
-    "http": {
-      "type": "http",
-      "address": "localhost:1337",
-      "handlers": {
-				"/": "h"
+	{
+		"receivers": {
+			"http": {
+				"type": "http",
+				"address": ":11111",
+				"handlers": {
+					"/": "h"
+				}
 			}
-    }
-  },
-	"transformers": {
-		"flatten_data": {
-			"type": "data",
-      "flattenSeparator": "drop",
-      "flatten": [
-        [
-          "kek"
-        ]
-      ]
 		},
-    "remove_data": {
-      "type": "data",
-      "remove": [
-        "kek"
-      ]
-    }
-  },
-  "handlers": {
-    "h": {
-      "parser": "skogul",
-      "transformers": [
-        "now",
-				"flatten_data",
-				"remove_data"
-      ],
-      "sender": "snmp"
-    }
-  },
-  "senders": {
-    "snmp": {
-      "type": "snmp",
-			"port": 7331,
-			"community": "xxx",
-			"version": "2c",
-			"target": "localhost",
-			"oidmap": {
-				"kek": "1.3.3.7",
+		"transformers": {},
+		"handlers": {
+			"h": {
+				"parser": "skogul",
+				"transformers": [
+					"now"
+				],
+				"sender": "print"
 			}
-    }
-  }
-}`)
+		},
+		"senders": {
+			"snmp": {
+				"type": "snmp",
+				"port": 1337,
+				"community": "public",
+				"version": "2c",
+				"target": "localhost",
+				"oidmap": {}
+			}
+		}
+	}
+	`)
 
 	conf, err := config.Bytes([]byte(sconf))
 
@@ -91,7 +72,7 @@ func TestSnmpSenderTest(t *testing.T) {
 	}
 
 	go httpRcv.Start()
-	time.Sleep(time.Duration(20 * time.Millisecond))
+	time.Sleep(time.Duration(5 * time.Second))
 
 	err = snmpSender.Send(c)