From 354f334cc312c5b0d72e310188b3337d7e18aa10 Mon Sep 17 00:00:00 2001 From: snoop Date: Tue, 8 Aug 2017 18:40:28 +0900 Subject: [PATCH] fixed test --- matcher/mysql/mysql_test.go | 138 ++++++++++++++++++------------------ 1 file changed, 69 insertions(+), 69 deletions(-) diff --git a/matcher/mysql/mysql_test.go b/matcher/mysql/mysql_test.go index a2befba..cd0d798 100644 --- a/matcher/mysql/mysql_test.go +++ b/matcher/mysql/mysql_test.go @@ -1,71 +1,71 @@ package mysql -// -//import ( -// "crypto/tls" -// "git.loafle.net/overflow/overflow_probe/collector/core/scan/service/matcher/packet" -// "net" -// "testing" -//) -// -//func TestMySql(t *testing.T) { -// -// m := NewMySqlMatcher() -// -// /* -// 192.168.1.103:3306 - normal -// 192.168.1.105:8306 - ssl -// 192.168.1.203:3306 - mysql with error code -// */ -// conn, _ := net.Dial("tcp", "192.168.1.215:3306") -// -// defer conn.Close() -// -// bytes := make([]byte, 1024) -// n, _ := conn.Read(bytes) -// p := packet.NewPacket(bytes, n) -// -// if m.Match(0, p, nil) { -// t.Log(m.ServiceName()) -// return -// } -// -// t.Error("MySQL not found") -// -//} -// -//func TestCassandraTLS(t *testing.T) { -// -// m := NewMySqlMatcher() -// -// conn, err := tls.Dial( -// "tcp", -// "192.168.1.105:8306", -// &tls.Config{ -// InsecureSkipVerify: true, -// ServerName: "192.168.1.105", -// }, -// ) -// if err != nil { -// t.Fatal(err) -// } -// -// 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(m.ServiceName()) -// return -// } -// -// t.Error("MySQL not found") -// } -// -//} + +import ( + "crypto/tls" + "git.loafle.net/overflow/overflow_probe/matcher/packet" + "net" + "testing" +) + +func TestMySql(t *testing.T) { + + m := NewMySqlMatcher() + + /* + 192.168.1.103:3306 - normal + 192.168.1.105:8306 - ssl + 192.168.1.203:3306 - mysql with error code + */ + conn, _ := net.Dial("tcp", "192.168.1.15:3306") + + defer conn.Close() + + bytes := make([]byte, 1024) + n, _ := conn.Read(bytes) + p := packet.NewPacket(bytes, n) + + if m.Match(0, p, nil) { + t.Log(m.ServiceName()) + return + } + + t.Error("MySQL not found") + +} + +func TestCassandraTLS(t *testing.T) { + + m := NewMySqlMatcher() + + conn, err := tls.Dial( + "tcp", + "192.168.1.105:8306", + &tls.Config{ + InsecureSkipVerify: true, + ServerName: "192.168.1.105", + }, + ) + if err != nil { + t.Fatal(err) + } + + 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(m.ServiceName()) + return + } + + t.Error("MySQL not found") + } + +}