initial: ZeroTier-like P2P mesh VPN server with multi-tenant Web UI

cmd/zeromesh-agent/main.go

@@ -0,0 +1,46 @@
+package main
+
+import (
+	"flag"
+	"log/slog"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"zeromesh/internal/agent"
+	"zeromesh/internal/config"
+)
+
+func main() {
+	cfgPath := flag.String("config", "config.yaml", "config file path")
+	flag.Parse()
+
+	cfg, err := config.Load(*cfgPath)
+	if err != nil {
+		slog.Error("load config", "err", err)
+		os.Exit(1)
+	}
+
+	logger := slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{Level: slog.LevelInfo}))
+	slog.SetDefault(logger)
+
+	ag, err := agent.New(cfg.Agent, logger)
+	if err != nil {
+		slog.Error("init agent", "err", err)
+		os.Exit(1)
+	}
+
+	if err := ag.Start(); err != nil {
+		slog.Error("start agent", "err", err)
+		os.Exit(1)
+	}
+
+	slog.Info("agent started", "address", ag.Identity().Address)
+
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+
+	slog.Info("stopping agent...")
+	ag.Stop()
+}

cmd/zeromesh-cli/main.go

@@ -0,0 +1,36 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+
+	"zeromesh/internal/identity"
+)
+
+func main() {
+	var cmd string
+	flag.StringVar(&cmd, "cmd", "", "command: gen-identity | show-identity")
+	flag.Parse()
+
+	switch cmd {
+	case "gen-identity":
+		id := identity.Generate()
+		data, _ := id.Serialize()
+		fmt.Println(data)
+	case "show-identity":
+		idPath := flag.Arg(0)
+		if idPath == "" {
+			idPath = "./data/agent.identity"
+		}
+		id, err := identity.LoadOrGenerate(idPath)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "error: %v\n", err)
+			os.Exit(1)
+		}
+		fmt.Printf("Address:  %s\n", id.Address)
+		fmt.Printf("Public:   %s\n", id.PublicKeyHex())
+	default:
+		fmt.Println("Usage: zeromesh-cli -cmd=gen-identity")
+	}
+}

cmd/zeromesh-controller/main.go

@@ -0,0 +1,94 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"log/slog"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/gin-gonic/gin"
+
+	"zeromesh/internal/config"
+	"zeromesh/internal/controller"
+	"zeromesh/internal/database"
+	"zeromesh/internal/router"
+)
+
+func main() {
+	cfgPath := flag.String("config", "config.yaml", "config file path")
+	flag.Parse()
+
+	cfg, err := config.Load(*cfgPath)
+	if err != nil {
+		slog.Error("load config", "err", err)
+		os.Exit(1)
+	}
+
+	logLevel := new(slog.LevelVar)
+	switch cfg.Logging.Level {
+	case "debug":
+		logLevel.Set(slog.LevelDebug)
+	case "warn":
+		logLevel.Set(slog.LevelWarn)
+	case "error":
+		logLevel.Set(slog.LevelError)
+	default:
+		logLevel.Set(slog.LevelInfo)
+	}
+	opts := &slog.HandlerOptions{Level: logLevel}
+	var handler slog.Handler
+	if cfg.Logging.Format == "json" {
+		handler = slog.NewJSONHandler(os.Stderr, opts)
+	} else {
+		handler = slog.NewTextHandler(os.Stderr, opts)
+	}
+	logger := slog.New(handler)
+	slog.SetDefault(logger)
+
+	db, err := database.Init(&cfg.Database)
+	if err != nil {
+		slog.Error("init database", "err", err)
+		os.Exit(1)
+	}
+
+	ctr, err := controller.New(cfg, db, logger)
+	if err != nil {
+		slog.Error("init controller", "err", err)
+		os.Exit(1)
+	}
+	if err := ctr.Start(); err != nil {
+		slog.Error("start controller", "err", err)
+		os.Exit(1)
+	}
+
+	gin.SetMode(gin.ReleaseMode)
+	r := router.Setup(cfg, db)
+
+	srv := &http.Server{
+		Addr:    cfg.Server.Addr(),
+		Handler: r,
+	}
+
+	go func() {
+		slog.Info("http server listening", "addr", srv.Addr)
+		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			slog.Error("http serve", "err", err)
+		}
+	}()
+
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+
+	slog.Info("shutting down...")
+	ctr.Stop()
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+	if err := srv.Shutdown(ctx); err != nil {
+		slog.Error("server shutdown", "err", err)
+	}
+}

cmd/zeromesh-relay/main.go

@@ -0,0 +1,62 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"log/slog"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"zeromesh/internal/config"
+	"zeromesh/internal/vl1"
+)
+
+func main() {
+	cfgPath := flag.String("config", "config.yaml", "config file path")
+	flag.Parse()
+
+	cfg, err := config.Load(*cfgPath)
+	if err != nil {
+		slog.Error("load config", "err", err)
+		os.Exit(1)
+	}
+
+	logger := slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{Level: slog.LevelInfo}))
+	slog.SetDefault(logger)
+
+	transport, err := vl1.NewTransport(cfg.Relay.ListenPort, logger)
+	if err != nil {
+		slog.Error("init relay transport", "err", err)
+		os.Exit(1)
+	}
+
+	slog.Info("relay started",
+		"realm", cfg.Relay.Realm,
+		"port", transport.Port(),
+	)
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	go func() {
+		buf := make([]byte, vl1.MaxPacketSize)
+		for {
+			n, addr, err := transport.ReadFrom(buf)
+			if err != nil {
+				if ctx.Err() != nil {
+					return
+				}
+				slog.Error("relay read", "err", err)
+				continue
+			}
+			logger.Debug("relay packet", "from", addr, "len", n)
+		}
+	}()
+
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+	cancel()
+	transport.Close()
+}