service_matcher-go/pop/pop.go
2018-09-12 13:26:27 +09:00

74 lines
1.1 KiB
Go

package pop
import (
osm "git.loafle.net/overflow/service_matcher-go"
)
const (
COMPARE_STR = "+OK"
)
type POPMatcher struct {
osm.Matchers
}
func (m *POPMatcher) Key() string {
return "POP3"
}
func (m *POPMatcher) Type() string {
return "MAIL"
}
func (m *POPMatcher) Name(matchCtx *osm.MatchCtx) string {
return "POP3"
}
func (m *POPMatcher) IsPrePacket() bool {
return true
}
func (m *POPMatcher) HasResponse(matchCtx *osm.MatchCtx, index int) bool {
return true
}
func (m *POPMatcher) IsError(matchCtx *osm.MatchCtx, index int, packet *osm.Packet) bool {
return false
}
func (m *POPMatcher) Match(matchCtx *osm.MatchCtx, index int, packet *osm.Packet) error {
switch index {
case 0:
fallthrough
case 1:
recvStr := string(packet.Bytes())
if len(recvStr) < 3 {
return osm.NotMatchedError()
}
compareStr := recvStr[0:3]
if compareStr == COMPARE_STR {
return nil
}
}
return osm.NotMatchedError()
}
func NewMatcher() osm.Matcher {
m := &POPMatcher{}
reqStr := "QUIT\r\n"
byte := make([]byte, len(reqStr))
copy(byte[:], reqStr)
m.AddPacket(osm.NewPacket(byte, len(reqStr)))
return m
}