ing
This commit is contained in:
80
service/matcher/imap/imap.go
Normal file
80
service/matcher/imap/imap.go
Normal file
@@ -0,0 +1,80 @@
|
||||
package imap
|
||||
|
||||
import (
|
||||
"git.loafle.net/overflow/overflow_discovery/service/matcher"
|
||||
)
|
||||
|
||||
const (
|
||||
PRE_COMPARE_STR = "* OK"
|
||||
SEND_COMPARE_STR = "* BYE"
|
||||
)
|
||||
|
||||
type IMAPMatcher struct {
|
||||
matcher.Matchers
|
||||
}
|
||||
|
||||
func (i *IMAPMatcher) ServiceName() string {
|
||||
return "IMAP"
|
||||
}
|
||||
|
||||
func (i *IMAPMatcher) IsPrePacket() bool {
|
||||
return true
|
||||
}
|
||||
|
||||
func (i *IMAPMatcher) HasResponse(index int) bool {
|
||||
return true
|
||||
}
|
||||
|
||||
func (i *IMAPMatcher) IsError(info matcher.MatchInfo, index int, packet *matcher.Packet) bool {
|
||||
return false
|
||||
}
|
||||
|
||||
func (i *IMAPMatcher) Match(info matcher.MatchInfo, index int, packet *matcher.Packet) bool {
|
||||
|
||||
switch index {
|
||||
case 0:
|
||||
|
||||
recvStr := string(packet.Buffer)
|
||||
|
||||
if len(recvStr) < 3 {
|
||||
return false
|
||||
}
|
||||
|
||||
compareStr := recvStr[0:4]
|
||||
|
||||
if compareStr == PRE_COMPARE_STR {
|
||||
return true
|
||||
}
|
||||
|
||||
case 1:
|
||||
|
||||
recvStr := string(packet.Buffer)
|
||||
|
||||
if len(recvStr) < 5 {
|
||||
return false
|
||||
}
|
||||
|
||||
compareStr := recvStr[0:5]
|
||||
|
||||
if compareStr == SEND_COMPARE_STR {
|
||||
return true
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func NewMatcher() matcher.Matcher {
|
||||
|
||||
m := &IMAPMatcher{}
|
||||
|
||||
reqStr := "A0001 LOGOUT\r\n"
|
||||
byte := make([]byte, len(reqStr))
|
||||
copy(byte[:], reqStr)
|
||||
|
||||
m.AddPacket(matcher.NewPacket(byte, len(reqStr)))
|
||||
|
||||
return m
|
||||
|
||||
}
|
||||
151
service/matcher/imap/imap_test.go
Normal file
151
service/matcher/imap/imap_test.go
Normal file
@@ -0,0 +1,151 @@
|
||||
package imap
|
||||
|
||||
import (
|
||||
"crypto/tls"
|
||||
"fmt"
|
||||
//"git.loafle.net/overflow/overflow_discovery/collector/core/scan/port"
|
||||
"git.loafle.net/overflow/overflow_discovery/service/matcher"
|
||||
//"git.loafle.net/overflow/overflow_discovery/collector/core/scan/service/matcher/scaninfo"
|
||||
//"git.loafle.net/overflow/overflow_discovery/collector/discovery/types"
|
||||
"net"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func ImapRun(client net.Conn, t *testing.T) {
|
||||
|
||||
lm := NewMatcher()
|
||||
|
||||
//port := types.NewPort("143", types.NewHost("192.168.1.215"), types.TYPE_TCP)
|
||||
//
|
||||
//scanInfo := types.NewServiceScanInfo(port)
|
||||
//
|
||||
//var ipport string
|
||||
//ipport = port.Host.Ip + ":" + string(port.Port)
|
||||
//
|
||||
//fmt.Println(ipport)
|
||||
//client, _ := net.Dial("tcp", ipport)
|
||||
|
||||
//defer client.Close()
|
||||
|
||||
bytett := make([]byte, 1024)
|
||||
|
||||
rr, _ := client.Read(bytett)
|
||||
|
||||
bb := lm.Match(nil, 0, matcher.NewPacket(bytett, rr))
|
||||
|
||||
if bb {
|
||||
t.Log("good!")
|
||||
}
|
||||
|
||||
fmt.Println(lm.PacketCount())
|
||||
|
||||
for ii := 0; ii < lm.PacketCount(); ii++ {
|
||||
|
||||
pack := lm.Packet(ii)
|
||||
|
||||
//fmt.Println(pack)
|
||||
|
||||
client.Write(pack.Buffer)
|
||||
|
||||
bytes := make([]byte, 1024)
|
||||
|
||||
read, _ := client.Read(bytes)
|
||||
|
||||
fmt.Println(cap(bytes))
|
||||
|
||||
//fmt.Println(bytes)
|
||||
|
||||
b := lm.Match(nil, ii+1, matcher.NewPacket(bytes, read))
|
||||
|
||||
if b {
|
||||
t.Log("send Good!")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestIMapTls(t *testing.T) {
|
||||
|
||||
conn, _ := tls.Dial(
|
||||
"tcp",
|
||||
"192.168.1.15:993",
|
||||
&tls.Config{
|
||||
InsecureSkipVerify: true,
|
||||
ServerName: "192.168.1.15",
|
||||
},
|
||||
)
|
||||
|
||||
defer conn.Close()
|
||||
|
||||
ImapRun(conn, t)
|
||||
|
||||
}
|
||||
|
||||
func TestIMapNormal(t *testing.T) {
|
||||
|
||||
client, err := net.Dial("tcp", "192.168.1.15:143")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
defer client.Close()
|
||||
|
||||
ImapRun(client, t)
|
||||
|
||||
}
|
||||
|
||||
func TestImap(t *testing.T) {
|
||||
|
||||
lm := NewMatcher()
|
||||
|
||||
//port := types.NewPort("143", types.NewHost("192.168.1.215"), types.TYPE_TCP)
|
||||
|
||||
//scanInfo := scaninfo.NewServiceScanInfo(port)
|
||||
|
||||
var ipport string
|
||||
//ipport = port.Host.Ip + ":" + port.Port_
|
||||
|
||||
fmt.Println(ipport)
|
||||
client, _ := net.Dial("tcp", ipport)
|
||||
|
||||
defer client.Close()
|
||||
|
||||
bytett := make([]byte, 1024)
|
||||
|
||||
rr, _ := client.Read(bytett)
|
||||
|
||||
//bb := lm.Match(0, matcher.NewPacket(bytett, rr), scanInfo)
|
||||
bb := lm.Match(nil, 0, matcher.NewPacket(bytett, rr))
|
||||
|
||||
if bb {
|
||||
t.Log("good!")
|
||||
}
|
||||
|
||||
fmt.Println(lm.PacketCount())
|
||||
|
||||
for ii := 0; ii < lm.PacketCount(); ii++ {
|
||||
|
||||
pack := lm.Packet(ii)
|
||||
|
||||
//fmt.Println(pack)
|
||||
|
||||
client.Write(pack.Buffer)
|
||||
|
||||
bytes := make([]byte, 1024)
|
||||
|
||||
read, _ := client.Read(bytes)
|
||||
|
||||
fmt.Println(cap(bytes))
|
||||
|
||||
//fmt.Println(bytes)
|
||||
|
||||
b := lm.Match(nil, ii+1, matcher.NewPacket(bytes, read))
|
||||
|
||||
if b {
|
||||
t.Log("send Good!")
|
||||
}
|
||||
|
||||
}
|
||||
//t.Log(scanInfo)
|
||||
}
|
||||
Reference in New Issue
Block a user