From b0364107dfe572ca5def1f54e04204751ca354ec Mon Sep 17 00:00:00 2001 From: snoop Date: Wed, 9 Aug 2017 12:26:20 +0900 Subject: [PATCH] fixed test --- matcher/netbios/netbios_test.go | 64 ++++++++++++++++----------------- matcher/smb/smb_test.go | 2 +- 2 files changed, 33 insertions(+), 33 deletions(-) diff --git a/matcher/netbios/netbios_test.go b/matcher/netbios/netbios_test.go index fdc2838..7c9911c 100644 --- a/matcher/netbios/netbios_test.go +++ b/matcher/netbios/netbios_test.go @@ -1,34 +1,34 @@ package netbios -// -//import ( -// "git.loafle.net/overflow/overflow_probe/collector/core/scan/service/matcher/packet" -// "net" -// "testing" -//) -// -//func TestNBSS(t *testing.T) { -// -// m := NewNetBiosMatcher() -// -// conn, _ := net.Dial("tcp", "192.168.1.106:139") -// -// 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("NBSS found") -// return -// } -// -// t.Error("NBSS not found") -// } -// -//} + +import ( + "git.loafle.net/overflow/overflow_probe/matcher/packet" + "net" + "testing" +) + +func TestNBSS(t *testing.T) { + + m := NewNetBiosMatcher() + + conn, _ := net.Dial("tcp", "192.168.1.106:139") + + 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("NBSS found") + return + } + + t.Error("NBSS not found") + } + +} diff --git a/matcher/smb/smb_test.go b/matcher/smb/smb_test.go index df394df..720e9a3 100644 --- a/matcher/smb/smb_test.go +++ b/matcher/smb/smb_test.go @@ -7,7 +7,7 @@ import ( ) const ( - ADDR string = "192.168.1.106:139" + ADDR string = "192.168.1.106:445" ) func TestSMBMatcher(t *testing.T) {