ric/MailHog
1
0
Fork 0
mirror of https://gitlab.com/ric_harvey/MailHog.git synced 2024-11-27 16:24:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add hook for EXTERNAL authentication mechanism

Browse source
This commit is contained in:
Ian Kent 2014-11-22 19:26:11 +00:00
parent 256849f2c6
commit 2b58b571cb
2 changed files with 16 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
mailhog/smtp/protocol/protocol.go
View file

@ -23,10 +23,11 @@ type Protocol struct {
message *data.SMTPMessage
hostname string
LogHandler func(message string, args ...interface{})
MessageReceivedHandler func(*data.Message) (string, error)
ValidateSenderHandler func(from string) bool
ValidateRecipientHandler func(to string) bool
LogHandler func(message string, args ...interface{})
MessageReceivedHandler func(*data.Message) (string, error)
ValidateSenderHandler func(from string) bool
ValidateRecipientHandler func(to string) bool
ValidateAuthenticationHandler func(mechanism string, args ...string) bool
}
// NewProtocol returns a new SMTP state machine in INVALID state
@ -182,6 +183,11 @@ func (proto *Protocol) Command(command *Command) (reply *Reply) {
return ReplyAuthResponse("PDQxOTI5NDIzNDEuMTI4Mjg0NzJAc291cmNlZm91ci5hbmRyZXcuY211LmVkdT4=")
case strings.HasPrefix(command.args, "EXTERNAL "):
proto.logf("Got EXTERNAL authentication: %s", strings.TrimPrefix(command.args, "EXTERNAL "))
if proto.ValidateAuthenticationHandler != nil {
if !proto.ValidateAuthenticationHandler("EXTERNAL", command.args) {
// TODO error reply
}
}
return ReplyAuthOk()
default:
return ReplyUnsupportedAuth()

8
mailhog/smtp/server/session.go
View file

@ -28,9 +28,10 @@ func Accept(conn *net.TCPConn, conf *config.Config) {
proto := protocol.NewProtocol()
session := &Session{conn, proto, conf, false, ""}
proto.LogHandler = session.logf
proto.MessageReceivedHandler = session.acceptMessageHandler
proto.MessageReceivedHandler = session.acceptMessage
proto.ValidateSenderHandler = session.validateSender
proto.ValidateRecipientHandler = session.validateRecipient
proto.ValidateAuthenticationHandler = session.validateAuthentication
session.logf("Starting session")
session.Write(proto.Start(conf.Hostname))
@ -39,6 +40,9 @@ func Accept(conn *net.TCPConn, conf *config.Config) {
session.logf("Session ended")
}
func (c *Session) validateAuthentication(mechanism string, args ...string) bool {
return true
}
func (c *Session) validateRecipient(to string) bool {
return true
}
@ -47,7 +51,7 @@ func (c *Session) validateSender(from string) bool {
return true
}
func (c *Session) acceptMessageHandler(msg *data.Message) (id string, err error) {
func (c *Session) acceptMessage(msg *data.Message) (id string, err error) {
switch c.conf.Storage.(type) {
case *storage.MongoDB:
c.logf("Storing message using MongoDB")