mirror of
https://github.com/neilalexander/yggmail.git
synced 2026-05-16 00:26:29 +03:00
Initial commit
This commit is contained in:
Neil Alexander
191
cmd/yggmail/main.go
Normal file
191
cmd/yggmail/main.go
Normal file
@@ -0,0 +1,191 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/ed25519"
|
||||
"encoding/hex"
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"strings"
|
||||
"sync"
|
||||
|
||||
"github.com/emersion/go-imap/server"
|
||||
"github.com/emersion/go-sasl"
|
||||
"github.com/emersion/go-smtp"
|
||||
"golang.org/x/term"
|
||||
|
||||
"github.com/neilalexander/yggmail/internal/config"
|
||||
"github.com/neilalexander/yggmail/internal/imapserver"
|
||||
"github.com/neilalexander/yggmail/internal/smtpsender"
|
||||
"github.com/neilalexander/yggmail/internal/smtpserver"
|
||||
"github.com/neilalexander/yggmail/internal/storage/sqlite3"
|
||||
"github.com/neilalexander/yggmail/internal/transport"
|
||||
)
|
||||
|
||||
var database = flag.String("database", "yggmail.db", "SQLite database file")
|
||||
var smtpaddr = flag.String("smtp", "localhost:1025", "SMTP listen address")
|
||||
var imapaddr = flag.String("imap", "localhost:1026", "IMAP listen address")
|
||||
var peeraddr = flag.String("peer", "", "Yggdrasil static peer")
|
||||
var createuser = flag.String("createuser", "", "Create a user")
|
||||
|
||||
func main() {
|
||||
flag.Parse()
|
||||
|
||||
rawlog := log.New(os.Stdout, "", 0)
|
||||
log := log.New(rawlog.Writer(), "[ \033[32mYggmail\033[0m ] ", 0)
|
||||
|
||||
storage, err := sqlite3.NewSQLite3StorageStorage(*database)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
log.Printf("Using database file %q\n", *database)
|
||||
|
||||
skStr, err := storage.ConfigGet("private_key")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
sk := make(ed25519.PrivateKey, ed25519.PrivateKeySize)
|
||||
if skStr == "" {
|
||||
if _, sk, err = ed25519.GenerateKey(nil); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
if err := storage.ConfigSet("private_key", hex.EncodeToString(sk)); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
log.Printf("Generated new server identity")
|
||||
} else {
|
||||
skBytes, err := hex.DecodeString(skStr)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
copy(sk, skBytes)
|
||||
}
|
||||
pk := sk.Public().(ed25519.PublicKey)
|
||||
log.Println("Mail domain:", hex.EncodeToString(pk))
|
||||
|
||||
switch {
|
||||
case createuser != nil && *createuser != "":
|
||||
fmt.Printf("New password: ")
|
||||
password1, err := term.ReadPassword(0)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
fmt.Println()
|
||||
fmt.Printf("Confirm password: ")
|
||||
password2, err := term.ReadPassword(0)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
fmt.Println()
|
||||
if !bytes.Equal(password1, password2) {
|
||||
fmt.Println("The supplied passwords do not match")
|
||||
os.Exit(1)
|
||||
}
|
||||
if err := storage.CreateUser(*createuser, strings.TrimSpace(string(password1))); err != nil {
|
||||
fmt.Printf("Failed to create user %q\n", *createuser)
|
||||
os.Exit(1)
|
||||
}
|
||||
if err := storage.MailboxCreate(*createuser, "INBOX"); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
fmt.Printf("Created user %q\n", *createuser)
|
||||
fmt.Printf("Email address will be %s@%s\n", *createuser, hex.EncodeToString(pk))
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
cfg := &config.Config{
|
||||
PublicKey: pk,
|
||||
PrivateKey: sk,
|
||||
}
|
||||
wg := &sync.WaitGroup{}
|
||||
wg.Add(2)
|
||||
|
||||
transport, err := transport.NewYggdrasilTransport(rawlog, sk, pk, *peeraddr)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
queues := smtpsender.NewQueues(cfg, log, transport)
|
||||
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
|
||||
imapBackend := &imapserver.Backend{
|
||||
Log: log,
|
||||
Config: cfg,
|
||||
Storage: storage,
|
||||
}
|
||||
|
||||
imapServer := server.New(imapBackend)
|
||||
imapServer.Addr = *imapaddr
|
||||
imapServer.AllowInsecureAuth = true
|
||||
imapServer.EnableAuth(sasl.Login, func(conn server.Conn) sasl.Server {
|
||||
return sasl.NewLoginServer(func(username, password string) error {
|
||||
_, err := imapBackend.Login(nil, username, password)
|
||||
return err
|
||||
})
|
||||
})
|
||||
|
||||
log.Println("Listening for IMAP on:", imapServer.Addr)
|
||||
if err := imapServer.ListenAndServe(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
|
||||
localBackend := &smtpserver.Backend{
|
||||
Log: log,
|
||||
Mode: smtpserver.BackendModeInternal,
|
||||
Config: cfg,
|
||||
Storage: storage,
|
||||
Queues: queues,
|
||||
}
|
||||
|
||||
localServer := smtp.NewServer(localBackend)
|
||||
localServer.Addr = *smtpaddr
|
||||
localServer.Domain = hex.EncodeToString(pk)
|
||||
localServer.MaxMessageBytes = 1024 * 1024
|
||||
localServer.MaxRecipients = 50
|
||||
localServer.AllowInsecureAuth = true
|
||||
localServer.EnableAuth(sasl.Login, func(conn *smtp.Conn) sasl.Server {
|
||||
return sasl.NewLoginServer(func(username, password string) error {
|
||||
_, err := localBackend.Login(nil, username, password)
|
||||
return err
|
||||
})
|
||||
})
|
||||
|
||||
log.Println("Listening for SMTP on:", localServer.Addr)
|
||||
if err := localServer.ListenAndServe(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
|
||||
overlayBackend := &smtpserver.Backend{
|
||||
Log: log,
|
||||
Mode: smtpserver.BackendModeExternal,
|
||||
Config: cfg,
|
||||
Storage: storage,
|
||||
Queues: queues,
|
||||
}
|
||||
|
||||
overlayServer := smtp.NewServer(overlayBackend)
|
||||
overlayServer.Domain = hex.EncodeToString(pk)
|
||||
overlayServer.MaxMessageBytes = 1024 * 1024
|
||||
overlayServer.MaxRecipients = 50
|
||||
overlayServer.AuthDisabled = true
|
||||
|
||||
if err := overlayServer.Serve(transport.Listener()); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
wg.Wait()
|
||||
}
|
||||
101
cmd/yggproxy/main.go
Normal file
101
cmd/yggproxy/main.go
Normal file
@@ -0,0 +1,101 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"crypto/ed25519"
|
||||
"encoding/hex"
|
||||
"flag"
|
||||
"log"
|
||||
"net"
|
||||
"os"
|
||||
|
||||
"github.com/neilalexander/yggmail/internal/transport"
|
||||
)
|
||||
|
||||
var dst = flag.String("dst", "", "Destination public key to proxy to")
|
||||
var peeraddr = flag.String("peer", "", "Yggdrasil static peer")
|
||||
|
||||
func main() {
|
||||
flag.Parse()
|
||||
|
||||
/*
|
||||
pk, sk, err := ed25519.GenerateKey(nil)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
*/
|
||||
|
||||
sk := make(ed25519.PrivateKey, ed25519.PrivateKeySize)
|
||||
if sks, err := hex.DecodeString("f50a6a4688ca602307dcf282304583b1746093a558e934d3e9a817bb1e7be77b7c824efcd702e80a3a6912e15ebc4e13454022947ce8ee46ddb871e8b9a9147f"); err != nil {
|
||||
panic(err)
|
||||
} else {
|
||||
copy(sk, sks)
|
||||
}
|
||||
pk := sk.Public().(ed25519.PublicKey)
|
||||
|
||||
log.Println("Private key:", hex.EncodeToString(sk))
|
||||
|
||||
log := log.New(os.Stdout, "", 0)
|
||||
transport, err := transport.NewYggdrasilTransport(log, sk, pk, *peeraddr)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
listener, err := net.Listen("tcp", "localhost:1026")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
log.Println("Proxying", listener.Addr(), "to", *dst)
|
||||
|
||||
for {
|
||||
conn, err := listener.Accept()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
log.Println("Accepted connection from", conn.RemoteAddr())
|
||||
|
||||
upstream, err := transport.Dial(*dst)
|
||||
if err != nil {
|
||||
log.Println("Failed to dial upstream:", err)
|
||||
conn.Close()
|
||||
continue
|
||||
}
|
||||
|
||||
go func(conn, upstream net.Conn) {
|
||||
defer conn.Close()
|
||||
defer upstream.Close()
|
||||
var b [1024]byte
|
||||
for {
|
||||
n, err := conn.Read(b[:])
|
||||
if err != nil {
|
||||
log.Println("conn.Read:", err)
|
||||
return
|
||||
}
|
||||
_, err = upstream.Write(b[:n])
|
||||
if err != nil {
|
||||
log.Println("upstream.Write:", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
}(conn, upstream)
|
||||
|
||||
go func(conn, upstream net.Conn) {
|
||||
defer conn.Close()
|
||||
defer upstream.Close()
|
||||
var b [1024]byte
|
||||
for {
|
||||
n, err := upstream.Read(b[:])
|
||||
if err != nil {
|
||||
log.Println("upstream.Read:", err)
|
||||
return
|
||||
}
|
||||
_, err = conn.Write(b[:n])
|
||||
if err != nil {
|
||||
log.Println("conn.Write:", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
}(conn, upstream)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user