Fix a few other goroutine leaks with contexts

1 files changed, 4 insertions, 0 deletions

diff --git a/forged/internal/incoming/lmtp/config.go b/forged/internal/incoming/lmtp/config.go
index c622648..ce32f3d 100644
--- a/forged/internal/incoming/lmtp/config.go
+++ b/forged/internal/incoming/lmtp/config.go
@@ -2,6 +2,7 @@ package lmtp
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"net"
 
@@ -52,6 +53,9 @@ func (server *Server) Run(ctx context.Context) error {
 	for {
 		conn, err := listener.Accept()
 		if err != nil {
+			if errors.Is(err, net.ErrClosed) {
+				return nil
+			}
 			return fmt.Errorf("accept conn: %w", err)
 		}