/
trap.go
80 lines (68 loc) · 1.64 KB
/
trap.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package mibs
import (
"os"
"strings"
"github.com/kentik/ktranslate/pkg/eggs/logger"
"github.com/kentik/ktranslate/pkg/kt"
"gopkg.in/yaml.v2"
)
type Trap struct {
Oid string `yaml:"trap_oid"`
Name string `yaml:"trap_name"`
DropUndefined bool `yaml:"drop_undefined"`
Events []OID `yaml:"events"`
}
type TrapBase struct {
logger.ContextL `yaml:"-"`
Traps []Trap `yaml:"traps"`
From string `yaml:"from,omitempty"`
}
func (t *Trap) DropUndefinedVars() bool {
if t == nil {
return false
}
return t.DropUndefined
}
func normalizeOid(oid string) string {
if strings.HasPrefix(oid, ".") {
return oid
}
return "." + oid
}
func (mdb *MibDB) parseTrapsFromYml(fname string, file os.DirEntry, extends map[string]*Profile) error {
t := TrapBase{ContextL: mdb.log, From: file.Name()}
data, err := os.ReadFile(fname)
if err != nil {
return err
}
err = yaml.Unmarshal(data, &t)
if err != nil {
return err
}
added := 0
for _, trap := range t.Traps {
for _, event := range trap.Events {
mib := &kt.Mib{
Oid: event.Oid,
Name: event.Name,
Enum: event.Enum,
Tag: event.Tag,
Conversion: event.Conversion,
Extra: trap.Name,
Mib: trap.Oid,
}
if len(mib.Enum) > 0 {
mib.EnumRev = make(map[int64]string)
}
for k, v := range mib.Enum {
mib.Enum[strings.ToLower(k)] = v
mib.EnumRev[v] = k
}
added++
mdb.trapMibs[normalizeOid(mib.Oid)] = mib
}
mdb.traps[normalizeOid(trap.Oid)] = trap
}
mdb.log.Infof("Loading %d snmp trap data points and %d traps from %s.", added, len(mdb.traps), fname)
return nil
}