service_matcher-go/wmi/wmi_test.go

40 lines
587 B
Go
Raw Normal View History

2018-08-13 07:48:32 +00:00
package wmi
import (
"net"
"testing"
2018-08-15 07:17:18 +00:00
osm "git.loafle.net/overflow/service_matcher-go"
2018-08-13 07:48:32 +00:00
)
2018-09-03 03:43:12 +00:00
// dcerpc
2018-08-13 07:48:32 +00:00
func TestWMI(t *testing.T) {
2018-08-27 01:53:32 +00:00
m := NewMatcher()
2018-08-13 07:48:32 +00:00
2018-09-03 03:43:12 +00:00
conn, err := net.Dial("tcp", "192.168.1.200:135")
2018-08-27 01:53:32 +00:00
if err != nil {
t.Error(err)
}
2018-08-13 07:48:32 +00:00
2018-08-27 01:53:32 +00:00
defer conn.Close()
2018-08-13 07:48:32 +00:00
2018-08-27 01:53:32 +00:00
for i := 0; i < m.PacketCount(); i++ {
_, err := conn.Write(m.Packet(i).Buffer)
if err != nil {
t.Error(err)
}
2018-08-13 07:48:32 +00:00
bytes := make([]byte, 1024)
2018-08-27 01:53:32 +00:00
n, _ := conn.Read(bytes)
p := osm.NewPacket(bytes, n)
2018-08-13 07:48:32 +00:00
2018-08-27 01:53:32 +00:00
if err := m.Match(nil, i, p); err != nil {
t.Error(err)
return
2018-08-13 07:48:32 +00:00
}
}
2018-08-27 01:53:32 +00:00
t.Log(m.Name())
t.Log(m.Meta())
2018-08-13 07:48:32 +00:00
}