1 files changed, 48 insertions, 4 deletions
diff --git a/cmd/mitmuxd/main.go b/cmd/mitmuxd/main.go
index f79f2cf..0d7f214 100644
--- a/cmd/mitmuxd/main.go
+++ b/cmd/mitmuxd/main.go
@@ -1,24 +1,31 @@
 // Command mitmuxd is the mitmux headless proxy daemon. It owns the
-// listening socket and (in later build steps) the traffic database; a
-// TUI or other client attaches separately without interrupting capture.
+// listening socket and the history database; a TUI or other client
+// attaches separately, over a Unix control socket, without interrupting
+// capture.
 package main
 
 import (
 	"context"
 	"flag"
 	"log"
+	"net"
 	"os"
 	"os/signal"
+	"path/filepath"
 	"syscall"
 	"time"
 
 	"mitmux/internal/ca"
+	"mitmux/internal/ipc"
 	"mitmux/internal/proxy"
+	"mitmux/internal/store"
 )
 
 func main() {
 	listen := flag.String("listen", "127.0.0.1:8080", "proxy listen address")
-	caDir := flag.String("ca-dir", "", "directory for CA cert/key (default: XDG config dir)")
+	caDir := flag.String("ca-dir", "", "directory for CA cert/key and history db (default: XDG config dir)")
+	dbPath := flag.String("db", "", "path to history database (default: <ca-dir>/history.db)")
+	socketPath := flag.String("socket", "", "control socket path (default: $XDG_RUNTIME_DIR/mitmux.sock, else <ca-dir>/mitmux.sock)")
 	flag.Parse()
 
 	dir := *caDir
@@ -36,7 +43,43 @@ func main() {
 	}
 	log.Printf("CA ready: %s (install %s/ca.pem in your client's trust store to avoid TLS warnings)", root.Cert.Subject.CommonName, dir)
 
-	srv := proxy.New(*listen, root)
+	dbFile := *dbPath
+	if dbFile == "" {
+		dbFile = filepath.Join(dir, "history.db")
+	}
+	db, err := store.Open(dbFile)
+	if err != nil {
+		log.Fatalf("open history db: %v", err)
+	}
+	defer db.Close()
+	log.Printf("history db: %s", dbFile)
+
+	sockFile := *socketPath
+	if sockFile == "" {
+		if rt := os.Getenv("XDG_RUNTIME_DIR"); rt != "" {
+			sockFile = filepath.Join(rt, "mitmux.sock")
+		} else {
+			sockFile = filepath.Join(dir, "mitmux.sock")
+		}
+	}
+	os.Remove(sockFile) // stale socket from an unclean shutdown
+	sockLn, err := net.Listen("unix", sockFile)
+	if err != nil {
+		log.Fatalf("listen on control socket %s: %v", sockFile, err)
+	}
+	defer os.Remove(sockFile)
+	log.Printf("control socket: %s", sockFile)
+
+	hub := ipc.NewHub()
+	ipcSrv := ipc.NewServer(db, hub)
+	go func() {
+		if err := ipcSrv.Serve(sockLn); err != nil {
+			log.Printf("control socket: %v", err)
+		}
+	}()
+
+	srv := proxy.New(*listen, root, db)
+	srv.OnEntry = ipc.LogAndBroadcast(hub)
 
 	errCh := make(chan error, 1)
 	go func() {
@@ -58,5 +101,6 @@ func main() {
 		if err := srv.Shutdown(ctx); err != nil {
 			log.Fatalf("shutdown: %v", err)
 		}
+		sockLn.Close()
 	}
 }