service_matcher-go/pop/pop.go
crusader d5bc04d5ee ing
2018-09-03 22:41:28 +09:00

70 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) 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
}