smtp.go

   1 package main
   2
   3 import (
   4         "crypto/tls"
   5         "fmt"
   6         "net"
   7         "net/mail"
   8         "os"
   9         "path"
  10
  11         "github.com/uber-go/zap"
  12
  13         "src.bluestatic.org/mailpopbox/smtp"
  14 )
  15
  16 func runSMTPServer(config Config, log zap.Logger) <-chan ServerControlMessage {
  17         server := smtpServer{
  18                 config:      config,
  19                 controlChan: make(chan ServerControlMessage),
  20                 log:         log.With(zap.String("server", "smtp")),
  21         }
  22         go server.run()
  23         return server.controlChan
  24 }
  25
  26 type smtpServer struct {
  27         config    Config
  28         tlsConfig *tls.Config
  29
  30         log zap.Logger
  31
  32         controlChan chan ServerControlMessage
  33 }
  34
  35 func (server *smtpServer) run() {
  36         if !server.loadTLSConfig() {
  37                 return
  38         }
  39
  40         addr := fmt.Sprintf(":%d", server.config.SMTPPort)
  41         server.log.Info("starting server", zap.String("address", addr))
  42
  43         l, err := net.Listen("tcp", addr)
  44         if err != nil {
  45                 server.log.Error("listen", zap.Error(err))
  46                 server.controlChan <- ServerControlFatalError
  47                 return
  48         }
  49
  50         connChan := make(chan net.Conn)
  51         go RunAcceptLoop(l, connChan, server.log)
  52
  53         reloadChan := CreateReloadSignal()
  54
  55         for {
  56                 select {
  57                 case <-reloadChan:
  58                         if !server.loadTLSConfig() {
  59                                 return
  60                         }
  61                 case conn, ok := <-connChan:
  62                         if ok {
  63                                 go smtp.AcceptConnection(conn, server, server.log)
  64                         } else {
  65                                 break
  66                         }
  67                 }
  68         }
  69 }
  70
  71 func (server *smtpServer) loadTLSConfig() bool {
  72         var err error
  73         server.tlsConfig, err = server.config.GetTLSConfig()
  74         if err != nil {
  75                 server.log.Error("failed to configure TLS", zap.Error(err))
  76                 server.controlChan <- ServerControlFatalError
  77                 return false
  78         }
  79         server.log.Info("loaded TLS config")
  80         return true
  81 }
  82
  83 func (server *smtpServer) Name() string {
  84         return server.config.Hostname
  85 }
  86
  87 func (server *smtpServer) TLSConfig() *tls.Config {
  88         return server.tlsConfig
  89 }
  90
  91 func (server *smtpServer) VerifyAddress(addr mail.Address) smtp.ReplyLine {
  92         if server.maildropForAddress(addr) == "" {
  93                 return smtp.ReplyBadMailbox
  94         }
  95         return smtp.ReplyOK
  96 }
  97
  98 func (server *smtpServer) OnMessageDelivered(en smtp.Envelope) *smtp.ReplyLine {
  99         maildrop := server.maildropForAddress(en.RcptTo[0])
 100         if maildrop == "" {
 101                 // TODO: log error
 102                 return &smtp.ReplyBadMailbox
 103         }
 104
 105         f, err := os.Create(path.Join(maildrop, en.ID+".msg"))
 106         if err != nil {
 107                 // TODO: log error
 108                 return &smtp.ReplyBadMailbox
 109         }
 110
 111         smtp.WriteEnvelopeForDelivery(f, en)
 112         f.Close()
 113         return nil
 114 }
 115
 116 func (server *smtpServer) maildropForAddress(addr mail.Address) string {
 117         domain := smtp.DomainForAddress(addr)
 118         for _, s := range server.config.Servers {
 119                 if domain == s.Domain {
 120                         return s.MaildropPath
 121                 }
 122         }
 123
 124         return ""
 125 }