examples/main.go

@@ -1,56 +0,0 @@
-package main
-
-import (
-	"fmt"
-	"net"
-
-	"net/textproto"
-
-	"git.ailur.dev/ailur/smtp"
-)
-
-// DatabaseBackend is a smtp.DatabaseBackend implementation that always returns true for CheckUser and prints the mail data to stdout.
-var DatabaseBackend = smtp.DatabaseBackend{
-	CheckUser: func(address *smtp.Address) (bool, error) {
-		return true, nil
-	},
-	WriteMail: func(mail *smtp.Mail) error {
-		fmt.Println(string(mail.Data))
-		return nil
-	},
-}
-
-// AuthenticationBackend is a smtp.AuthenticationBackend implementation that always returns a fixed address for Authenticate.
-var AuthenticationBackend = smtp.AuthenticationBackend{
-	Authenticate: func(initial string, conn *textproto.Conn) (smtp.CheckAddress, error) {
-		return func(address *smtp.Address) (bool, error) {
-			return true, nil
-		}, nil
-	},
-}
-
-func main() {
-	go func() {
-		// Serve on the server-to-server port
-		listener, err := net.Listen("tcp", ":25")
-		if err != nil {
-			panic(err)
-		}
-		receiver := smtp.NewReceiver(listener, "localhost", []string{"localhost", "127.0.0.1", "0.0.0.0", "example.org", "192.168.1.253"}, false, DatabaseBackend, AuthenticationBackend, nil)
-		err = receiver.Serve()
-		panic(err)
-	}()
-	go func() {
-		// Serve on the submission port
-		listener, err := net.Listen("tcp", ":587")
-		if err != nil {
-			panic(err)
-		}
-		receiver := smtp.NewReceiver(listener, "localhost", []string{"localhost", "127.0.0.1", "0.0.0.0", "cta.social"}, false, DatabaseBackend, AuthenticationBackend, nil)
-		err = receiver.Serve()
-		panic(err)
-	}()
-
-	// Block forever
-	select {}
-}

go.mod

@@ -1,4 +1,4 @@
-module git.ailur.dev/ailur/smtp
+module smtp
 
 go 1.23
 

smtp.go

@@ -7,12 +7,12 @@ import (
 	"net"
 	"strconv"
 	"strings"
-	"time"
 
 	"crypto/tls"
 	"net/textproto"
 
 	"git.ailur.dev/ailur/spf"
+	"github.com/google/uuid"
 )
 
 var (
@@ -22,7 +22,7 @@ var (
 		"250-SMTPUTF8",
 		"250 BINARYMIME",
 	}
-	queue = make(map[time.Time]*MailQueueItem)
+	queue = make(map[uuid.UUID]*MailQueueItem)
 )
 
 // MailQueueItem is a struct that represents an item in the mail queue
@@ -33,7 +33,7 @@ type MailQueueItem struct {
 }
 
 // ViewMailQueue returns the current mail queue
-func ViewMailQueue() map[time.Time]*MailQueueItem {
+func ViewMailQueue() map[uuid.UUID]*MailQueueItem {
 	return queue
 }
 
@@ -53,17 +53,14 @@ type Mail struct {
 // DatabaseBackend is a struct that represents a database backend
 type DatabaseBackend struct {
 	CheckUser func(*Address) (bool, error)
-	WriteMail func(*Mail) error
+	WriteMail func(*Mail) (uuid.UUID, error)
 }
 
 // AuthenticationBackend is a struct that represents an authentication backend
 type AuthenticationBackend struct {
-	Authenticate        func(initial string, conn *textproto.Conn) (CheckAddress, error)
+	Authenticate func(string) (*Address, error)
-	SupportedMechanisms []string
 }
 
-type CheckAddress func(*Address) (bool, error)
-
 func readMultilineCodeResponse(conn *textproto.Conn) (int, string, error) {
 	var lines strings.Builder
 	for {
@@ -86,7 +83,7 @@ func readMultilineCodeResponse(conn *textproto.Conn) (int, string, error) {
 }
 
 func systemError(err error, receiver *Address, database DatabaseBackend) {
-	_ = database.WriteMail(&Mail{
+	_, _ = database.WriteMail(&Mail{
 		From: &Address{
 			Name:    "EMail System",
 			Address: "system",
@@ -96,7 +93,7 @@ func systemError(err error, receiver *Address, database DatabaseBackend) {
 	})
 }
 
-func sendEmail(args SenderArgs, mail *Mail, database DatabaseBackend, queueID time.Time) {
+func sendEmail(args SenderArgs, mail *Mail, database DatabaseBackend, queueID uuid.UUID) {
 	mxs, err := net.LookupMX(mail.To[0].Address)
 	if err != nil {
 		systemError(err, queue[queueID].From, database)
@@ -150,7 +147,7 @@ func speakMultiLine(conn *textproto.Conn, lines []string) error {
 type Receiver struct {
 	underlyingListener net.Listener
 	hostname           string
-	ownedDomains       map[string]struct{}
+	ownedDomains       map[string]any
 	enforceTLS         bool
 	tlsConfig          *tls.Config
 	database           DatabaseBackend
@@ -159,9 +156,9 @@ type Receiver struct {
 
 // NewReceiver creates a new Receiver
 func NewReceiver(conn net.Listener, hostname string, ownedDomains []string, enforceTLS bool, database DatabaseBackend, authentication AuthenticationBackend, tlsConfig *tls.Config) *Receiver {
-	var ownedDomainsMap = make(map[string]struct{})
+	var ownedDomainsMap = make(map[string]any)
 	for _, domain := range ownedDomains {
-		ownedDomainsMap[domain] = struct{}{}
+		ownedDomainsMap[domain] = nil
 	}
 	return &Receiver{
 		underlyingListener: conn,
@@ -194,7 +191,7 @@ func (fr *Receiver) Serve() error {
 func (fr *Receiver) handleConnection(conn net.Conn) {
 	var state struct {
 		HELO bool
-		AUTH CheckAddress
+		AUTH *Address
 		TLS  bool
 		FROM *Address
 		RCPT []*Address
@@ -212,6 +209,8 @@ func (fr *Receiver) handleConnection(conn net.Conn) {
 		return
 	}
 
+	fmt.Println("Connection from", conn.RemoteAddr().String())
+
 	for {
 		line, err := textProto.ReadLine()
 		if err != nil {
@@ -282,9 +281,6 @@ func (fr *Receiver) handleConnection(conn net.Conn) {
 				if fr.enforceTLS {
 					capabilities = append(capabilities, "250-REQUIRETLS")
 				}
-				if fr.auth.SupportedMechanisms != nil {
-					capabilities = append(capabilities, "250-AUTH "+strings.Join(fr.auth.SupportedMechanisms, " "))
-				}
 				capabilities = append(capabilities, defaultCapabilities...)
 				state.HELO = true
 				err = speakMultiLine(textProto, capabilities)
@@ -324,14 +320,14 @@ func (fr *Receiver) handleConnection(conn net.Conn) {
 				}
 				continue
 			} else {
-				checkAddress, err := fr.auth.Authenticate(strings.TrimPrefix(line, "AUTH "), textProto)
+				address, err := fr.auth.Authenticate(line)
 				if err != nil {
-					_ = textProto.PrintfLine(err.Error())
+					_ = textProto.PrintfLine("421 4.7.0 Temporary server error")
 					_ = conn.Close()
 					return
 				}
 
-				if checkAddress == nil {
+				if address == nil {
 					err = textProto.PrintfLine("535 5.7.8 Authentication failed")
 					if err != nil {
 						_ = textProto.PrintfLine("421 4.7.0 Temporary server error")
@@ -339,7 +335,7 @@ func (fr *Receiver) handleConnection(conn net.Conn) {
 						return
 					}
 				} else {
-					state.AUTH = checkAddress
+					state.AUTH = address
 					err = textProto.PrintfLine("235 2.7.0 Authentication successful")
 					if err != nil {
 						_ = textProto.PrintfLine("421 4.7.0 Temporary server error")
@@ -414,14 +410,7 @@ func (fr *Receiver) handleConnection(conn net.Conn) {
 					continue
 				}
 
-				ok, err := state.AUTH(address)
+				if *address != *state.AUTH {
-				if err != nil {
-					_ = textProto.PrintfLine("421 4.7.0 Temporary server error")
-					_ = conn.Close()
-					return
-				}
-
-				if !ok {
 					err = textProto.PrintfLine("535 5.7.8 Authenticated wrong user")
 					if err != nil {
 						_ = textProto.PrintfLine("421 4.7.0 Temporary server error")
@@ -568,7 +557,7 @@ func (fr *Receiver) handleConnection(conn net.Conn) {
 			}
 
 			if !isSubmission {
-				err := fr.database.WriteMail(mail)
+				_, err := fr.database.WriteMail(mail)
 				if err != nil {
 					_ = textProto.PrintfLine(err.Error())
 					_ = conn.Close()
@@ -582,7 +571,7 @@ func (fr *Receiver) handleConnection(conn net.Conn) {
 					return
 				}
 			} else {
-				queueID := time.Now()
+				queueID := uuid.New()
 
 				queue[queueID] = &MailQueueItem{
 					From: state.FROM,
@@ -590,6 +579,7 @@ func (fr *Receiver) handleConnection(conn net.Conn) {
 					Host: strings.Split(conn.RemoteAddr().String(), ":")[0],
 				}
 				go sendEmail(SenderArgs{
+					Hostname:   fr.hostname,
 					EnforceTLS: fr.enforceTLS,
 				}, mail, fr.database, queueID)
 
@@ -618,6 +608,7 @@ func (fr *Receiver) handleConnection(conn net.Conn) {
 
 // SenderArgs is a struct that represents the arguments for the Sender
 type SenderArgs struct {
+	Hostname   string
 	EnforceTLS bool
 }
 
@@ -648,7 +639,7 @@ func Send(args SenderArgs, mail *Mail, conn net.Conn, mxHost string) (err error)
 		return errors.New("unexpected greeting - " + line)
 	}
 
-	err = textConn.PrintfLine("EHLO %s", mxHost)
+	err = textConn.PrintfLine("EHLO %s", args.Hostname)
 	if err != nil {
 		return err
 	}
@@ -682,11 +673,16 @@ func Send(args SenderArgs, mail *Mail, conn net.Conn, mxHost string) (err error)
 			InsecureSkipVerify: false,
 		})
 
+		err = tlsConn.Handshake()
+		if err != nil {
+			return err
+		}
+
 		textConn = textproto.NewConn(tlsConn)
 
 		// Just use HELO, no point using EHLO when we already have all the capabilities
 		// This also gets us out of using readMultilineCodeResponse
-		err = textConn.PrintfLine("HELO %s", mxHost)
+		err = textConn.PrintfLine("HELO %s", args.Hostname)
 		if err != nil {
 			return err
 		}
@@ -709,7 +705,10 @@ func Send(args SenderArgs, mail *Mail, conn net.Conn, mxHost string) (err error)
 	}
 
 	code, line, err = textConn.ReadCodeLine(0)
+	fmt.Println(code, line, err)
 	if err != nil {
+		// For some reason the EHLO stuff ends up here
+		fmt.Println("5")
 		return err
 	}
 
@@ -724,7 +723,9 @@ func Send(args SenderArgs, mail *Mail, conn net.Conn, mxHost string) (err error)
 		}
 
 		code, line, err = textConn.ReadCodeLine(0)
+		fmt.Println(code, line, err)
 		if err != nil {
+			fmt.Println("6")
 			return err
 		}
 
@@ -739,7 +740,9 @@ func Send(args SenderArgs, mail *Mail, conn net.Conn, mxHost string) (err error)
 	}
 
 	code, line, err = textConn.ReadCodeLine(0)
+	fmt.Println(code, line, err)
 	if err != nil {
+		fmt.Println("7")
 		return err
 	}
 
@@ -759,7 +762,9 @@ func Send(args SenderArgs, mail *Mail, conn net.Conn, mxHost string) (err error)
 	}
 
 	code, line, err = textConn.ReadCodeLine(0)
+	fmt.Println(code, line, err)
 	if err != nil {
+		fmt.Println("8")
 		return err
 	}
 
@@ -773,7 +778,9 @@ func Send(args SenderArgs, mail *Mail, conn net.Conn, mxHost string) (err error)
 	}
 
 	code, line, err = textConn.ReadCodeLine(0)
+	fmt.Println(code, line, err)
 	if err != nil {
+		fmt.Println("9")
 		return err
 	}