1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
package main
import (
"bytes"
"fmt"
"log"
"os"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/ses"
"github.com/mcrute/go-smtpd/smtpd"
)
const (
SesSizeLimit = 10000000
DefaultAddr = ":2500"
)
var sesClient *ses.SES
type Envelope struct {
from string
rcpts []*string
b bytes.Buffer
}
func (e *Envelope) AddRecipient(rcpt smtpd.MailAddress) error {
email := rcpt.Email()
e.rcpts = append(e.rcpts, &email)
return nil
}
func (e *Envelope) BeginData() error {
if len(e.rcpts) == 0 {
return smtpd.SMTPError("554 5.5.1 Error: no valid recipients")
}
return nil
}
func (e *Envelope) Write(line []byte) error {
e.b.Write(line)
if e.b.Len() > SesSizeLimit { // SES limitation
log.Printf("message size %d exceeds SES limit of %d\n", e.b.Len(), SesSizeLimit)
return smtpd.SMTPError("554 5.5.1 Error: maximum message size exceeded")
}
return nil
}
func (e *Envelope) logMessageSend() {
dr := make([]string, len(e.rcpts))
for i := range e.rcpts {
dr[i] = *e.rcpts[i]
}
log.Printf("sending message from %+v to %+v", e.from, dr)
}
func (e *Envelope) Close() error {
e.logMessageSend()
r := &ses.SendRawEmailInput{
Source: &e.from,
Destinations: e.rcpts,
RawMessage: &ses.RawMessage{Data: e.b.Bytes()},
}
_, err := sesClient.SendRawEmail(r)
if err != nil {
log.Printf("ERROR: ses: %v", err)
return smtpd.SMTPError(fmt.Sprintf("451 4.5.1 Temporary server error. Please try again later: %v", err))
}
return err
}
func main() {
sesClient = ses.New(session.Must(session.NewSession()))
addr := DefaultAddr
if len(os.Args) == 2 {
addr = os.Args[1]
} else if len(os.Args) > 2 {
log.Fatalf("usage: %s [listen_host:port]", os.Args[0])
}
s := &smtpd.Server{
Addr: addr,
OnNewMail: func(c smtpd.Connection, from smtpd.MailAddress) (smtpd.Envelope, error) {
return &Envelope{from: from.Email()}, nil
},
}
log.Printf("ListenAndServe on %s", addr)
err := s.ListenAndServe()
if err != nil {
log.Fatalf("ListenAndServe: %v", err)
}
}
|
