service_matcher-go/imap/imap.go
crusader bce329e44e ing
2018-10-23 13:31:25 +09:00

104 lines
1.7 KiB
Go

package imap
import (
osm "git.loafle.net/overflow/service_matcher-go"
)
const (
PRE_COMPARE_STR = "* OK"
SEND_COMPARE_STR = "* BYE"
)
type IMAPMatcher struct {
osm.Matchers
}
func (m *IMAPMatcher) Key(matchCtx *osm.MatchCtx) string {
return "IMAP"
}
func (m *IMAPMatcher) Type(matchCtx *osm.MatchCtx) string {
return "MAIL"
}
func (m *IMAPMatcher) Vendor(matchCtx *osm.MatchCtx) string {
return "UNKNOWN"
}
func (m *IMAPMatcher) Version(matchCtx *osm.MatchCtx) string {
return "UNKNOWN"
}
func (m *IMAPMatcher) OsType(matchCtx *osm.MatchCtx) string {
return "UNKNOWN"
}
func (m *IMAPMatcher) OsVersion(matchCtx *osm.MatchCtx) string {
return "UNKNOWN"
}
func (m *IMAPMatcher) Name(matchCtx *osm.MatchCtx) string {
return "IMAP"
}
func (m *IMAPMatcher) IsPrePacket() bool {
return true
}
func (m *IMAPMatcher) HasResponse(matchCtx *osm.MatchCtx, index int) bool {
return true
}
func (m *IMAPMatcher) IsError(matchCtx *osm.MatchCtx, index int, packet *osm.Packet) bool {
return false
}
func (m *IMAPMatcher) Match(matchCtx *osm.MatchCtx, index int, packet *osm.Packet) error {
switch index {
case 0:
recvStr := string(packet.Bytes())
if len(recvStr) < 3 {
return osm.NotMatchedError()
}
compareStr := recvStr[0:4]
if compareStr == PRE_COMPARE_STR {
return nil
}
case 1:
recvStr := string(packet.Bytes())
if len(recvStr) < 5 {
return osm.NotMatchedError()
}
compareStr := recvStr[0:5]
if compareStr == SEND_COMPARE_STR {
return nil
}
}
return osm.NotMatchedError()
}
func NewMatcher() osm.Matcher {
m := &IMAPMatcher{}
reqStr := "A0001 LOGOUT\r\n"
byte := make([]byte, len(reqStr))
copy(byte[:], reqStr)
m.AddPacket(osm.NewPacket(byte, len(reqStr)))
return m
}