Gnarwhal/diodemail
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages 1 Wiki Activity Actions

Slight reworking of config file. Connect to MTAs via port 25 and use certificate for StartTLS.

Browse source
This commit is contained in:
Gnarwhal 2024-10-04 20:41:21 +00:00
parent e8979dc08e
commit 1750872280
Signed by: Gnarwhal
GPG key ID: 0989A73D8C421174
4 changed files with 67 additions and 77 deletions
Download patch file Download diff file Expand all files Collapse all files

54
smtp/server.go
View file

@ -27,23 +27,14 @@ import (
"github.com/rs/zerolog/log"
)
type PlainConfig struct {
Port string
}
type TLSConfig struct {
Port string
TlsConfig tls.Config
}
func handle(connection net.Conn, host string, password_hash string) {
func handle(connection net.Conn, host string, password_hash string, tls_config tls.Config) {
log.Info().Msgf(
"New connection %v. Starting session.",
connection.RemoteAddr(),
)
defer connection.Close()
session := MakeSMTPSession(connection, host, password_hash)
session := MakeSMTPSession(connection, host, password_hash, tls_config)
err := session.Run()
if err != nil {
log.Error().Msgf(
@ -59,36 +50,51 @@ func handle(connection net.Conn, host string, password_hash string) {
}
}
func Run(host string, password_hash string, plain_config *PlainConfig, tls_config *TLSConfig) error {
func Run(
host string,
password_hash string,
plain_port string,
tls_port string,
tls_config tls.Config,
) error {
var wait_group sync.WaitGroup
if plain_config != nil {
listener, err := net.Listen("tcp", fmt.Sprintf(":%v", plain_config.Port))
if plain_port != "disabled" {
listener, err := net.Listen("tcp", fmt.Sprintf(":%v", plain_port))
if err != nil {
return err
}
log.Info().Msgf("Plain text server started on port %v for host %v", plain_config.Port, host)
log.Info().Msgf("Plain text server started on port %v for host %v", plain_port, host)
wait_group.Add(1)
go Listen(wait_group, host, password_hash, listener)
go Listen(wait_group, host, password_hash, tls_config, listener)
}
if tls_config != nil {
listener, err := tls.Listen("tcp", fmt.Sprintf(":%v", tls_config.Port), &tls_config.TlsConfig)
if tls_port != "disabled" {
listener, err := tls.Listen("tcp", fmt.Sprintf(":%v", tls_port), &tls_config)
if err != nil {
return err
}
log.Info().Msgf("TLS server started on port %v for host %v", tls_config.Port, host)
log.Info().Msgf("TLS server started on port %v for host %v", tls_port, host)
wait_group.Add(1)
go Listen(wait_group, host, password_hash, listener)
go Listen(wait_group, host, password_hash, tls_config, listener)
}
wait_group.Wait()
return nil
}
func Listen(wait_group sync.WaitGroup, host string, password_hash string, listener net.Listener) {
func Listen(
wait_group sync.WaitGroup,
host string,
password_hash string,
tls_config tls.Config,
listener net.Listener,
) {
defer wait_group.Done()
for {
connection, _ := listener.Accept()
go handle(connection, host, password_hash)
connection, err := listener.Accept()
if err != nil {
log.Error().Msgf("Failed to accept client: %v", err)
} else {
go handle(connection, host, password_hash, tls_config)
}
}
}

15
smtp/session.go
View file

@ -33,13 +33,14 @@ type SMTPSession struct {
connection net.Conn
host string
password_hash string
tls_config tls.Config
buffer [4096]byte
ReversePathBuffer *string
ForwardPathBuffer []string
}
func MakeSMTPSession(connection net.Conn, host string, password_hash string) SMTPSession {
func MakeSMTPSession(connection net.Conn, host string, password_hash string, tls_config tls.Config) SMTPSession {
return SMTPSession{
connection: connection,
host: host,
@ -161,11 +162,7 @@ func (self *SMTPSession) SendMail(data string) error {
} else {
smtp_hostname = mx[0].Host
}
tls_connection, err := tls.Dial("tcp", fmt.Sprintf("%v:465", smtp_hostname), nil)
if err != nil {
return err
}
smtp_client, err := smtp.NewClient(tls_connection, "")
smtp_client, err := smtp.Dial(fmt.Sprintf("%v:25", smtp_hostname))
if err != nil {
return err
}
@ -175,6 +172,12 @@ func (self *SMTPSession) SendMail(data string) error {
if err != nil {
return err
}
tls_config := self.tls_config
tls_config.ServerName = smtp_hostname
err = smtp_client.StartTLS(&tls_config)
if err != nil {
return err
}
err = smtp_client.Mail(*self.ReversePathBuffer)
if err != nil {
return err