2018-08-13 07:48:32 +00:00
|
|
|
package smtp
|
|
|
|
|
|
|
|
import (
|
|
|
|
"strings"
|
|
|
|
|
2018-08-15 07:17:18 +00:00
|
|
|
osm "git.loafle.net/overflow/service_matcher-go"
|
2018-08-13 07:48:32 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type SmtpMatcher struct {
|
2018-08-15 07:17:18 +00:00
|
|
|
osm.Matchers
|
2018-08-13 07:48:32 +00:00
|
|
|
}
|
|
|
|
|
2018-09-03 13:36:57 +00:00
|
|
|
func (m *SmtpMatcher) Key() string {
|
2018-08-13 07:48:32 +00:00
|
|
|
return "SMTP"
|
|
|
|
}
|
|
|
|
|
2018-09-12 04:26:27 +00:00
|
|
|
func (m *SmtpMatcher) Type() string {
|
|
|
|
return "MAIL"
|
|
|
|
}
|
|
|
|
|
2018-09-03 13:36:57 +00:00
|
|
|
func (m *SmtpMatcher) Name(matchCtx *osm.MatchCtx) string {
|
2018-08-13 07:48:32 +00:00
|
|
|
return "SMTP"
|
|
|
|
}
|
|
|
|
|
2018-09-03 13:41:28 +00:00
|
|
|
func (m *SmtpMatcher) IsPrePacket() bool {
|
2018-08-13 07:48:32 +00:00
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
2018-09-03 13:36:57 +00:00
|
|
|
func (m *SmtpMatcher) HasResponse(matchCtx *osm.MatchCtx, index int) bool {
|
2018-08-13 07:48:32 +00:00
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
2018-09-03 13:36:57 +00:00
|
|
|
func (m *SmtpMatcher) IsError(matchCtx *osm.MatchCtx, index int, packet *osm.Packet) bool {
|
2018-08-13 07:48:32 +00:00
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
2018-09-03 13:36:57 +00:00
|
|
|
func (m *SmtpMatcher) Match(matchCtx *osm.MatchCtx, index int, packet *osm.Packet) error {
|
2018-08-13 07:48:32 +00:00
|
|
|
if packet == nil {
|
2018-09-03 06:42:56 +00:00
|
|
|
return osm.NoPacketReceivedError()
|
2018-08-13 07:48:32 +00:00
|
|
|
}
|
2018-09-03 06:42:56 +00:00
|
|
|
|
|
|
|
buf := string(packet.Bytes())
|
2018-08-13 07:48:32 +00:00
|
|
|
if len(buf) == 0 || len(buf) < 5 {
|
2018-09-03 06:42:56 +00:00
|
|
|
return osm.NotMatchedError()
|
2018-08-13 07:48:32 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
splits := strings.Split(buf, "\r\n")
|
|
|
|
splits = strings.Split(buf, " ")
|
|
|
|
if index == 0 {
|
|
|
|
if splits[0] == "220" {
|
2018-09-03 06:42:56 +00:00
|
|
|
return nil
|
2018-08-13 07:48:32 +00:00
|
|
|
}
|
|
|
|
} else if index == 1 {
|
|
|
|
if splits[0] == "250" {
|
2018-09-03 06:42:56 +00:00
|
|
|
return nil
|
2018-08-13 07:48:32 +00:00
|
|
|
}
|
|
|
|
} else if index == 2 {
|
|
|
|
if splits[0] == "221" {
|
2018-09-03 06:42:56 +00:00
|
|
|
return nil
|
2018-08-13 07:48:32 +00:00
|
|
|
}
|
|
|
|
}
|
2018-09-03 06:42:56 +00:00
|
|
|
return osm.NotMatchedError()
|
2018-08-13 07:48:32 +00:00
|
|
|
}
|
|
|
|
|
2018-08-15 07:17:18 +00:00
|
|
|
func NewMatcher() osm.Matcher {
|
2018-08-13 07:48:32 +00:00
|
|
|
|
|
|
|
m := &SmtpMatcher{}
|
|
|
|
|
|
|
|
b := []byte("helo test\r\n")
|
2018-08-15 07:17:18 +00:00
|
|
|
m.AddPacket(osm.NewPacket(b, len(b)))
|
2018-08-13 07:48:32 +00:00
|
|
|
|
|
|
|
b = []byte("quit\r\n")
|
2018-08-15 07:17:18 +00:00
|
|
|
m.AddPacket(osm.NewPacket(b, len(b)))
|
2018-08-13 07:48:32 +00:00
|
|
|
|
|
|
|
return m
|
|
|
|
}
|