Add features from influxdata#546

plugins/inputs/snmp/snmp.go

@@ -1,7 +1,6 @@
 package snmp
 
 import (
-	"fmt"
 	"io/ioutil"
 	"log"
 	"net"
@@ -41,14 +40,18 @@ type Host struct {
 
 type Data struct {
 	Name string
-	// OIds
+	// OID (could be numbers or name)
 	Oid string
 	// Unit
 	Unit string
 	//  SNMP getbulk max repetition
 	MaxRepetition uint8 `toml:"max_repetition"`
-	// We want to resolve oid to names oids text file
-	Snmptranslate bool
+	// SNMP Instance (default 0)
+	// (only used with  GET request and if
+	//  OID is a name from snmptranslate file)
+	Instance string
+	// OID (only number) (used for computation)
+	rawOid string
 }
 
 type Node struct {
@@ -63,6 +66,8 @@ var initNode = Node{
 	subnodes: make(map[string]Node),
 }
 
+var NameToOid = make(map[string]string)
+
 var sampleConfig = `
   # Use 'oids.txt' file to translate oids to names
   # To generate 'oids.txt' you need to run:
@@ -93,25 +98,28 @@ var sampleConfig = `
     collect = ["mybulk"]
 
   [[inputs.snmp.get]]
-    name = "sysservices"
-    oid = ".1.3.6.1.2.1.1.7.0"
+    name = "ifnumber"
+    oid = "ifNumber"
 
   [[inputs.snmp.get]]
-    name = "sysdescr"
-    oid = ".1.3.6.1.2.1.1.1.0"
-    snmptranslate = true
+    name = "interface_speed"
+    oid = "ifSpeed"
+    instance = 0
 
   [[inputs.snmp.get]]
     name = "sysuptime"
     oid = ".1.3.6.1.2.1.1.3.0"
-    unit = "seconds"
-    snmptranslate = true
+    unit = "second"
 
   [[inputs.snmp.bulk]]
     name = "mybulk"
     max_repetition = 127
     oid = ".1.3.6.1.2.1.1"
-    snmptranslate = true
+
+  [[inputs.snmp.bulk]]
+    name = "ifoutoctets"
+    max_repetition = 127
+    oid = "ifOutOctets"
 `
 
 // SampleConfig returns sample configuration message
@@ -144,10 +152,10 @@ func fillnode(parentNode Node, oid_name string, ids []string) {
 	}
 }
 
-func findnodename(node Node, ids []string) string {
+func findnodename(node Node, ids []string) (string, string) {
 	// ids = ["1", "3", "6", ...]
 	if len(ids) == 1 {
-		return node.name
+		return node.name, ids[0]
 	}
 	id, ids := ids[0], ids[1:]
 	// Get node
@@ -159,24 +167,25 @@ func findnodename(node Node, ids []string) string {
 	// Get node name
 	if node.name != "" && len(ids) == 0 && id == "0" {
 		// node with instance 0
-		return node.name
+		return node.name, "0"
 	} else if node.name != "" && len(ids) == 0 && id != "0" {
 		// node with an instance
-		return node.name + "." + string(id)
+		return node.name, string(id)
 	} else if node.name != "" && len(ids) > 0 {
 		// node with subinstances
-		return node.name + "." + strings.Join(ids, ".")
+		return node.name, strings.Join(ids, ".")
 	}
 	// return an empty node name
-	return node.name
+	return node.name, ""
 }
 
 func (s *Snmp) Gather(acc inputs.Accumulator) error {
 	// Create oid tree
 	if s.SnmptranslateFile != "" {
 		data, err := ioutil.ReadFile(s.SnmptranslateFile)
 		if err != nil {
-			fmt.Print(err)
+			log.Printf("Reading SNMPtranslate file error: %s", err)
+			return err
 		} else {
 			for _, line := range strings.Split(string(data), "\n") {
 				oidsRegEx := regexp.MustCompile(`([^\t]*)\t*([^\t]*)`)
@@ -185,6 +194,7 @@ func (s *Snmp) Gather(acc inputs.Accumulator) error {
 					oid_name := oids[1]
 					oid := oids[2]
 					fillnode(initNode, oid_name, strings.Split(string(oid), "."))
+					NameToOid[oid_name] = oid
 				}
 			}
 		}
@@ -209,12 +219,27 @@ func (s *Snmp) Gather(acc inputs.Accumulator) error {
 			// Get GET oids
 			for _, oid := range s.Get {
 				if oid.Name == oid_name {
+					if val, ok := NameToOid[oid.Oid]; ok {
+						// TODO should we add the 0 instance ?
+						if oid.Instance != "" {
+							oid.rawOid = "." + val + "." + oid.Instance
+						} else {
+							oid.rawOid = "." + val + ".0"
+						}
+					} else {
+						oid.rawOid = oid.Oid
+					}
 					host.getOids = append(host.getOids, oid)
 				}
 			}
 			// Get GETBULK oids
 			for _, oid := range s.Bulk {
 				if oid.Name == oid_name {
+					if val, ok := NameToOid[oid.Oid]; ok {
+						oid.rawOid = "." + val
+					} else {
+						oid.rawOid = oid.Oid
+					}
 					host.bulkOids = append(host.bulkOids, oid)
 				}
 			}
@@ -239,11 +264,11 @@ func (h *Host) SNMPGet(acc inputs.Accumulator) error {
 	// Prepare OIDs
 	oidsList := make(map[string]Data)
 	for _, oid := range h.getOids {
-		oidsList[oid.Oid] = oid
+		oidsList[oid.rawOid] = oid
 	}
 	oidsNameList := make([]string, 0, len(oidsList))
 	for _, oid := range oidsList {
-		oidsNameList = append(oidsNameList, oid.Oid)
+		oidsNameList = append(oidsNameList, oid.rawOid)
 	}
 
 	// gosnmp.MAX_OIDS == 60
@@ -281,11 +306,11 @@ func (h *Host) SNMPBulk(acc inputs.Accumulator) error {
 	// Prepare OIDs
 	oidsList := make(map[string]Data)
 	for _, oid := range h.bulkOids {
-		oidsList[oid.Oid] = oid
+		oidsList[oid.rawOid] = oid
 	}
 	oidsNameList := make([]string, 0, len(oidsList))
 	for _, oid := range oidsList {
-		oidsNameList = append(oidsNameList, oid.Oid)
+		oidsNameList = append(oidsNameList, oid.rawOid)
 	}
 	// TODO Trying to make requests with more than one OID
 	// to reduce the number of requests
@@ -379,31 +404,28 @@ func (h *Host) HandleResponse(oids map[string]Data, result *gosnmp.SnmpPacket, a
 					// Get name and instance
 					var oid_name string
 					var instance string
-					if oid.Snmptranslate {
-						// Get oidname and instannce from translate file
-						oidfullname := findnodename(initNode,
-							strings.Split(string(variable.Name[1:]), "."))
-						res := strings.SplitN(string(oidfullname), ".", 2)
-						oid_name = res[0]
-						if len(res) > 1 {
-							instance = res[1]
-						}
-					}
+					// Get oidname and instannce from translate file
+					oid_name, instance = findnodename(initNode,
+						strings.Split(string(variable.Name[1:]), "."))
+
 					if instance != "" {
 						tags["instance"] = instance
 					}
+
 					// Set name
 					var field_name string
 					if oid_name != "" {
+						// Set fieldname as oid name from translate file
 						field_name = oid_name
-					} else if oid.Oid == variable.Name {
-						field_name = oid.Name
 					} else {
-						field_name = variable.Name
+						// Set fieldname as oid name from inputs.snmp.get section
+						// Because the result oid is equal to inputs.snmp.get section
+						field_name = oid.Name
 					}
 					tags["host"], _, _ = net.SplitHostPort(h.Address)
 					fields := make(map[string]interface{})
 					fields[string(field_name)] = variable.Value
+
 					acc.AddFields(field_name, fields, tags)
 				case gosnmp.NoSuchObject, gosnmp.NoSuchInstance:
 					// Oid not found