diff --git a/matcher/snmp/snmpv2_test.go b/matcher/snmp/snmpv2_test.go index 515d4f6..3326a00 100644 --- a/matcher/snmp/snmpv2_test.go +++ b/matcher/snmp/snmpv2_test.go @@ -1,37 +1,37 @@ package snmp -// -//import ( -// "git.loafle.net/overflow/overflow_probe/collector/core/scan/service/matcher/packet" -// "net" -// "testing" -//) -// -//func TestSNMP2(t *testing.T) { -// -// m := NewSNMPv2Matcher() -// -// conn, err := net.Dial("udp", "192.168.1.215:161") -// if err != nil { -// t.Error(err) -// return -// } -// defer conn.Close() -// -// for i := 0; i < m.PacketCount(); i++ { -// -// pack := m.Packet(i) -// conn.Write(pack.Buffer) -// bytes := make([]byte, 1024) -// n, _ := conn.Read(bytes) -// p := packet.NewPacket(bytes, n) -// -// if m.Match(i, p, nil) { -// t.Log("SNMP found") -// return -// } -// -// t.Error("SNMP not found") -// } -// -//} + +import ( + "git.loafle.net/overflow/overflow_probe/matcher/packet" + "net" + "testing" +) + +func TestSNMP2(t *testing.T) { + + m := NewSNMPv2Matcher() + + conn, err := net.Dial("udp", "192.168.1.15:161") + if err != nil { + t.Error(err) + return + } + defer conn.Close() + + for i := 0; i < m.PacketCount(); i++ { + + pack := m.Packet(i) + conn.Write(pack.Buffer) + bytes := make([]byte, 1024) + n, _ := conn.Read(bytes) + p := packet.NewPacket(bytes, n) + + if m.Match(i, p, nil) { + t.Log("SNMP found") + return + } + + t.Error("SNMP not found") + } + +} diff --git a/matcher/snmp/snmpv3_test.go b/matcher/snmp/snmpv3_test.go index b9d6a97..97772b1 100644 --- a/matcher/snmp/snmpv3_test.go +++ b/matcher/snmp/snmpv3_test.go @@ -1,37 +1,37 @@ package snmp -// -//import ( -// "git.loafle.net/overflow/overflow_probe/collector/core/scan/service/matcher/packet" -// "net" -// "testing" -//) -// -//func TestSNMP3(t *testing.T) { -// -// m := NewSNMPv3Matcher() -// -// conn, err := net.Dial("udp", "192.168.1.254:161") -// if err != nil { -// t.Error(err) -// return -// } -// defer conn.Close() -// -// for i := 0; i < m.PacketCount(); i++ { -// -// pack := m.Packet(i) -// conn.Write(pack.Buffer) -// bytes := make([]byte, 1024) -// n, _ := conn.Read(bytes) -// p := packet.NewPacket(bytes, n) -// -// if m.Match(i, p, nil) { -// t.Log("SNMP found") -// return -// } -// -// t.Error("SNMP not found") -// } -// -//} + +import ( + "git.loafle.net/overflow/overflow_probe/matcher/packet" + "net" + "testing" +) + +func TestSNMP3(t *testing.T) { + + m := NewSNMPv3Matcher() + + conn, err := net.Dial("udp", "192.168.1.15:161") + if err != nil { + t.Error(err) + return + } + defer conn.Close() + + for i := 0; i < m.PacketCount(); i++ { + + pack := m.Packet(i) + conn.Write(pack.Buffer) + bytes := make([]byte, 1024) + n, _ := conn.Read(bytes) + p := packet.NewPacket(bytes, n) + + if m.Match(i, p, nil) { + t.Log("SNMP found") + return + } + + t.Error("SNMP not found") + } + +}