From ac55bd843589e2a5f74b601475152114384b733a Mon Sep 17 00:00:00 2001 From: "insanity@loafle.com" Date: Wed, 9 Aug 2017 11:45:34 +0900 Subject: [PATCH] test --- collector/collector.go | 4 +- matcher/cassandra/cassandra_test.go | 143 ++++++++++++++-------------- matcher/mongodb/mongodb_test.go | 105 ++++++++++---------- matcher/mssql/mssql_test.go | 113 +++++++++++----------- matcher/redis/redis_test.go | 71 +++++++------- 5 files changed, 216 insertions(+), 220 deletions(-) diff --git a/collector/collector.go b/collector/collector.go index 056bec9..98b6963 100644 --- a/collector/collector.go +++ b/collector/collector.go @@ -81,12 +81,12 @@ func (c *Collector) collect(id string) { out, err := dc.Get(context.Background(), in) finishedAt := time.Now().Unix() if err != nil { - log.Error("Cannot collect [ID: %s] [Crawler : %s] - [Err : %s]\n", conf.Id, conf.Crawler.Name, err.Error()) + log.Errorf("Cannot collect [ID: %s] [Crawler : %s] - [Err : %s]\n", conf.Id, conf.Crawler.Name, err.Error()) return } out.StartDate = startAt out.EndDate = finishedAt - log.Error("COLLECTED. [ID: %s] [Crawler : %s] [Result : %s]\n", conf.Id, conf.Crawler.Name, out.GetData()) + log.Infof("COLLECTED. [ID: %s] [Crawler : %s] [Result : %s]\n", conf.Id, conf.Crawler.Name, out.GetData()) c.dataCh <- out }() } diff --git a/matcher/cassandra/cassandra_test.go b/matcher/cassandra/cassandra_test.go index 3e82e8d..085a71d 100644 --- a/matcher/cassandra/cassandra_test.go +++ b/matcher/cassandra/cassandra_test.go @@ -1,74 +1,73 @@ package cassandra -// -//import ( -// "crypto/tls" -// "git.loafle.net/overflow/overflow_probe/collector/core/scan/service/matcher/packet" -// "net" -// "testing" -//) -// -//func TestCassandra(t *testing.T) { -// -// m := NewCassandraMatcher() -// -// conn, err := net.Dial("tcp", "192.168.1.104:9042") -// 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("Cassandra found") -// return -// } -// -// t.Error("Cassandra not found") -// } -// -//} -// -//func TestCassandraTLS(t *testing.T) { -// -// m := NewCassandraMatcher() -// -// conn, err := tls.Dial( -// "tcp", -// "192.168.1.104:9042", -// &tls.Config{ -// InsecureSkipVerify: true, -// ServerName: "192.168.1.104", -// }, -// ) -// 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("Cassandra found") -// return -// } -// -// t.Error("Cassandra not found") -// } -// -//} +import ( + "crypto/tls" + "git.loafle.net/overflow/overflow_probe/matcher/packet" + "net" + "testing" +) + +func TestCassandra(t *testing.T) { + + m := NewCassandraMatcher() + + conn, err := net.Dial("tcp", "192.168.1.16:19041") + 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("Cassandra found") + return + } + + t.Error("Cassandra not found") + } + +} + +func TestCassandraTLS(t *testing.T) { + + m := NewCassandraMatcher() + + conn, err := tls.Dial( + "tcp", + "192.168.1.16:19042", + &tls.Config{ + InsecureSkipVerify: true, + ServerName: "192.168.1.16", + }, + ) + 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("Cassandra found") + return + } + + t.Error("Cassandra not found") + } + +} diff --git a/matcher/mongodb/mongodb_test.go b/matcher/mongodb/mongodb_test.go index 560edb6..8176b72 100644 --- a/matcher/mongodb/mongodb_test.go +++ b/matcher/mongodb/mongodb_test.go @@ -1,55 +1,54 @@ package mongodb -// -//import ( -// "crypto/tls" -// "git.loafle.net/overflow/overflow_probe/collector/core/scan/service/matcher/packet" -// "net" -// "testing" -//) -// -//func TestMongoNor(t *testing.T) { -// -// conn, _ := net.Dial("tcp", "192.168.1.105:27017") -// -// defer conn.Close() -// -// MongoRun(conn, t) -// -//} -//func TestMongoTLS(t *testing.T) { -// conn, _ := tls.Dial( -// "tcp", -// "192.168.1.105:27017", -// &tls.Config{ -// InsecureSkipVerify: true, -// ServerName: "192.168.1.105", -// }, -// ) -// -// defer conn.Close() -// -// MongoRun(conn, t) -//} -// -//func MongoRun(conn net.Conn, t *testing.T) { -// -// m := NewMongoDBMatcher() -// -// 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("MongoDB found") -// return -// } -// -// t.Error("MongoDB not found") -// } -// -//} +import ( + "crypto/tls" + "git.loafle.net/overflow/overflow_probe/matcher/packet" + "net" + "testing" +) + +func TestMongoNor(t *testing.T) { + + conn, _ := net.Dial("tcp", "192.168.1.16:37017") + + defer conn.Close() + + MongoRun(conn, t) + +} +func TestMongoTLS(t *testing.T) { + conn, _ := tls.Dial( + "tcp", + "192.168.1.16:47017", + &tls.Config{ + InsecureSkipVerify: true, + ServerName: "192.168.1.16", + }, + ) + + defer conn.Close() + + MongoRun(conn, t) +} + +func MongoRun(conn net.Conn, t *testing.T) { + + m := NewMongoDBMatcher() + + 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("MongoDB found") + return + } + + t.Error("MongoDB not found") + } + +} diff --git a/matcher/mssql/mssql_test.go b/matcher/mssql/mssql_test.go index 3120f8e..458fc85 100644 --- a/matcher/mssql/mssql_test.go +++ b/matcher/mssql/mssql_test.go @@ -1,64 +1,63 @@ package mssql +import ( + "git.loafle.net/overflow/overflow_probe/matcher/packet" + "net" + "testing" +) + +/* +192.168.1.106:1433 - normal +192.168.1.103:1433 - ssl +*/ +func TestSqlNor(t *testing.T) { + + conn, _ := net.Dial("tcp", "192.168.1.16:11433") + + defer conn.Close() + + sqlServerRun(conn, t) + +} + +//func TestSqlTLS(t *testing.T) { +// conn, err := tls.Dial( +// "tcp", +// "192.168.1.103:7680", +// &tls.Config{ +// InsecureSkipVerify: true, +// ServerName: "192.168.1.103", +// }, +// ) // -//import ( -// "git.loafle.net/overflow/overflow_probe/collector/discovery/scan/matcher/packet" -// "net" -// "testing" -//) -// -///* -//192.168.1.106:1433 - normal -//192.168.1.103:1433 - ssl -//*/ -//func TestSqlNor(t *testing.T) { -// -// conn, _ := net.Dial("tcp", "192.168.1.103:1433") +// if err != nil { +// t.Log(err) +// return +// } // // defer conn.Close() // -// sqlServerRun(conn, t) -// -//} -// -////func TestSqlTLS(t *testing.T) { -//// conn, err := tls.Dial( -//// "tcp", -//// "192.168.1.103:7680", -//// &tls.Config{ -//// InsecureSkipVerify: true, -//// ServerName: "192.168.1.103", -//// }, -//// ) -//// -//// if err != nil { -//// t.Log(err) -//// return -//// } -//// -//// defer conn.Close() -//// -//// sqlServerRun(conn, t) -////} -// -//func sqlServerRun(conn net.Conn, t *testing.T) { -// -// m := NewMSSqlMatcher() -// -// 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("MSSQL not found") -// } -// +// sqlServerRun(conn, t) //} + +func sqlServerRun(conn net.Conn, t *testing.T) { + + m := NewMSSqlMatcher() + + 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("MSSQL not found") + } + +} diff --git a/matcher/redis/redis_test.go b/matcher/redis/redis_test.go index f461eaf..d7348d0 100644 --- a/matcher/redis/redis_test.go +++ b/matcher/redis/redis_test.go @@ -1,38 +1,37 @@ package redis -// -//import ( -// "git.loafle.net/overflow/overflow_probe/collector/core/scan/service/matcher/packet" -// "net" -// "testing" -//) -// -//const ( -// ADDR string = "192.168.1.215:6379" -//) -// -//func TestRedisMatcher(t *testing.T) { -// -// m := NewRedisMatcher() -// -// conn, _ := net.Dial("tcp", ADDR) -// 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("Redis found.") -// return -// } -// -// t.Error("Redis not found") -// } -// -//} +import ( + "git.loafle.net/overflow/overflow_probe/matcher/packet" + "net" + "testing" +) + +const ( + ADDR string = "192.168.1.16:26379" +) + +func TestRedisMatcher(t *testing.T) { + + m := NewRedisMatcher() + + conn, _ := net.Dial("tcp", ADDR) + 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("Redis found.") + return + } + + t.Error("Redis not found") + } + +}