Add SQLite persistence and write-through gRPC mutations
Database (internal/db) stores listeners, routes, and firewall rules with WAL mode, foreign keys, and idempotent migrations. First run seeds from TOML config; subsequent runs load from DB as source of truth. gRPC admin API now writes to the database before updating in-memory state (write-through cache pattern). Adds snapshot command for VACUUM INTO backups. Refactors firewall.New to accept raw rule slices instead of config struct for flexibility. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -12,6 +12,7 @@ func rootCmd() *cobra.Command {
|
|||||||
}
|
}
|
||||||
|
|
||||||
cmd.AddCommand(serverCmd())
|
cmd.AddCommand(serverCmd())
|
||||||
|
cmd.AddCommand(snapshotCmd())
|
||||||
|
|
||||||
return cmd
|
return cmd
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,14 +2,18 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"fmt"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"os"
|
"os"
|
||||||
"os/signal"
|
"os/signal"
|
||||||
|
"strings"
|
||||||
"syscall"
|
"syscall"
|
||||||
|
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
|
|
||||||
"git.wntrmute.dev/kyle/mc-proxy/internal/config"
|
"git.wntrmute.dev/kyle/mc-proxy/internal/config"
|
||||||
|
"git.wntrmute.dev/kyle/mc-proxy/internal/db"
|
||||||
|
"git.wntrmute.dev/kyle/mc-proxy/internal/firewall"
|
||||||
"git.wntrmute.dev/kyle/mc-proxy/internal/grpcserver"
|
"git.wntrmute.dev/kyle/mc-proxy/internal/grpcserver"
|
||||||
"git.wntrmute.dev/kyle/mc-proxy/internal/server"
|
"git.wntrmute.dev/kyle/mc-proxy/internal/server"
|
||||||
)
|
)
|
||||||
@@ -30,17 +34,53 @@ func serverCmd() *cobra.Command {
|
|||||||
Level: parseLogLevel(cfg.Log.Level),
|
Level: parseLogLevel(cfg.Log.Level),
|
||||||
}))
|
}))
|
||||||
|
|
||||||
srv, err := server.New(cfg, logger, version)
|
// Open and migrate the database.
|
||||||
|
store, err := db.Open(cfg.Database.Path)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("opening database: %w", err)
|
||||||
|
}
|
||||||
|
defer store.Close()
|
||||||
|
|
||||||
|
if err := store.Migrate(); err != nil {
|
||||||
|
return fmt.Errorf("running migrations: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Seed from config on first run, or load from DB.
|
||||||
|
empty, err := store.IsEmpty()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("checking database: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if empty {
|
||||||
|
if len(cfg.Listeners) == 0 {
|
||||||
|
return fmt.Errorf("database is empty and no listeners defined in config for seeding")
|
||||||
|
}
|
||||||
|
logger.Info("seeding database from config")
|
||||||
|
if err := store.Seed(cfg.Listeners, cfg.Firewall); err != nil {
|
||||||
|
return fmt.Errorf("seeding database: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load listeners and routes from DB.
|
||||||
|
listenerData, err := loadListenersFromDB(store)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Load firewall rules from DB.
|
||||||
|
fw, err := loadFirewallFromDB(store, cfg.Firewall.GeoIPDB)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
srv := server.New(cfg, fw, listenerData, logger, version)
|
||||||
|
|
||||||
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
|
|
||||||
// Start gRPC admin API if configured.
|
// Start gRPC admin API if configured.
|
||||||
if cfg.GRPC.Addr != "" {
|
if cfg.GRPC.Addr != "" {
|
||||||
grpcSrv, ln, err := grpcserver.New(cfg.GRPC, srv, logger)
|
grpcSrv, ln, err := grpcserver.New(cfg.GRPC, srv, store, logger)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -75,6 +115,56 @@ func serverCmd() *cobra.Command {
|
|||||||
return cmd
|
return cmd
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func loadListenersFromDB(store *db.Store) ([]server.ListenerData, error) {
|
||||||
|
dbListeners, err := store.ListListeners()
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("loading listeners: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var result []server.ListenerData
|
||||||
|
for _, l := range dbListeners {
|
||||||
|
dbRoutes, err := store.ListRoutes(l.ID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("loading routes for listener %q: %w", l.Addr, err)
|
||||||
|
}
|
||||||
|
routes := make(map[string]string, len(dbRoutes))
|
||||||
|
for _, r := range dbRoutes {
|
||||||
|
routes[strings.ToLower(r.Hostname)] = r.Backend
|
||||||
|
}
|
||||||
|
result = append(result, server.ListenerData{
|
||||||
|
ID: l.ID,
|
||||||
|
Addr: l.Addr,
|
||||||
|
Routes: routes,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func loadFirewallFromDB(store *db.Store, geoIPPath string) (*firewall.Firewall, error) {
|
||||||
|
rules, err := store.ListFirewallRules()
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("loading firewall rules: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var ips, cidrs, countries []string
|
||||||
|
for _, r := range rules {
|
||||||
|
switch r.Type {
|
||||||
|
case "ip":
|
||||||
|
ips = append(ips, r.Value)
|
||||||
|
case "cidr":
|
||||||
|
cidrs = append(cidrs, r.Value)
|
||||||
|
case "country":
|
||||||
|
countries = append(countries, r.Value)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fw, err := firewall.New(geoIPPath, ips, cidrs, countries)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("initializing firewall: %w", err)
|
||||||
|
}
|
||||||
|
return fw, nil
|
||||||
|
}
|
||||||
|
|
||||||
func parseLogLevel(s string) slog.Level {
|
func parseLogLevel(s string) slog.Level {
|
||||||
switch s {
|
switch s {
|
||||||
case "debug":
|
case "debug":
|
||||||
|
|||||||
54
cmd/mc-proxy/snapshot.go
Normal file
54
cmd/mc-proxy/snapshot.go
Normal file
@@ -0,0 +1,54 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"path/filepath"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
|
||||||
|
"git.wntrmute.dev/kyle/mc-proxy/internal/config"
|
||||||
|
"git.wntrmute.dev/kyle/mc-proxy/internal/db"
|
||||||
|
)
|
||||||
|
|
||||||
|
func snapshotCmd() *cobra.Command {
|
||||||
|
var (
|
||||||
|
configPath string
|
||||||
|
outputPath string
|
||||||
|
)
|
||||||
|
|
||||||
|
cmd := &cobra.Command{
|
||||||
|
Use: "snapshot",
|
||||||
|
Short: "Create a database backup",
|
||||||
|
RunE: func(cmd *cobra.Command, args []string) error {
|
||||||
|
cfg, err := config.Load(configPath)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
store, err := db.Open(cfg.Database.Path)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("opening database: %w", err)
|
||||||
|
}
|
||||||
|
defer store.Close()
|
||||||
|
|
||||||
|
if outputPath == "" {
|
||||||
|
dir := filepath.Dir(cfg.Database.Path)
|
||||||
|
ts := time.Now().UTC().Format("20060102T150405Z")
|
||||||
|
outputPath = filepath.Join(dir, "backups", fmt.Sprintf("mc-proxy-%s.db", ts))
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := store.Snapshot(outputPath); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("snapshot written to %s\n", outputPath)
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
cmd.Flags().StringVarP(&configPath, "config", "c", "mc-proxy.toml", "path to configuration file")
|
||||||
|
cmd.Flags().StringVarP(&outputPath, "output", "o", "", "output path (default: backups/mc-proxy-<timestamp>.db)")
|
||||||
|
|
||||||
|
return cmd
|
||||||
|
}
|
||||||
13
go.mod
13
go.mod
@@ -1,6 +1,6 @@
|
|||||||
module git.wntrmute.dev/kyle/mc-proxy
|
module git.wntrmute.dev/kyle/mc-proxy
|
||||||
|
|
||||||
go 1.24.0
|
go 1.25.0
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/oschwald/maxminddb-golang v1.13.1
|
github.com/oschwald/maxminddb-golang v1.13.1
|
||||||
@@ -8,13 +8,22 @@ require (
|
|||||||
github.com/spf13/cobra v1.10.2
|
github.com/spf13/cobra v1.10.2
|
||||||
google.golang.org/grpc v1.79.2
|
google.golang.org/grpc v1.79.2
|
||||||
google.golang.org/protobuf v1.36.11
|
google.golang.org/protobuf v1.36.11
|
||||||
|
modernc.org/sqlite v1.46.2
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
|
github.com/dustin/go-humanize v1.0.1 // indirect
|
||||||
|
github.com/google/uuid v1.6.0 // indirect
|
||||||
github.com/inconshreveable/mousetrap v1.1.0 // indirect
|
github.com/inconshreveable/mousetrap v1.1.0 // indirect
|
||||||
|
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||||
|
github.com/ncruces/go-strftime v1.0.0 // indirect
|
||||||
|
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
|
||||||
github.com/spf13/pflag v1.0.9 // indirect
|
github.com/spf13/pflag v1.0.9 // indirect
|
||||||
golang.org/x/net v0.48.0 // indirect
|
golang.org/x/net v0.48.0 // indirect
|
||||||
golang.org/x/sys v0.39.0 // indirect
|
golang.org/x/sys v0.42.0 // indirect
|
||||||
golang.org/x/text v0.32.0 // indirect
|
golang.org/x/text v0.32.0 // indirect
|
||||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20251202230838-ff82c1b0f217 // indirect
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20251202230838-ff82c1b0f217 // indirect
|
||||||
|
modernc.org/libc v1.70.0 // indirect
|
||||||
|
modernc.org/mathutil v1.7.1 // indirect
|
||||||
|
modernc.org/memory v1.11.0 // indirect
|
||||||
)
|
)
|
||||||
|
|||||||
51
go.sum
51
go.sum
@@ -3,6 +3,8 @@ github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XL
|
|||||||
github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
|
github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
|
||||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
|
||||||
|
github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
|
||||||
github.com/go-logr/logr v1.4.3 h1:CjnDlHq8ikf6E492q6eKboGOC0T8CDaOvkHCIg8idEI=
|
github.com/go-logr/logr v1.4.3 h1:CjnDlHq8ikf6E492q6eKboGOC0T8CDaOvkHCIg8idEI=
|
||||||
github.com/go-logr/logr v1.4.3/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=
|
github.com/go-logr/logr v1.4.3/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=
|
||||||
github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag=
|
github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag=
|
||||||
@@ -11,16 +13,26 @@ github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek
|
|||||||
github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
|
github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
|
||||||
github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
|
github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
|
||||||
github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
|
github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
|
||||||
|
github.com/google/pprof v0.0.0-20250317173921-a4b03ec1a45e h1:ijClszYn+mADRFY17kjQEVQ1XRhq2/JR1M3sGqeJoxs=
|
||||||
|
github.com/google/pprof v0.0.0-20250317173921-a4b03ec1a45e/go.mod h1:boTsfXsheKC2y+lKOCMpSfarhxDeIzfZG1jqGcPl3cA=
|
||||||
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||||
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||||
|
github.com/hashicorp/golang-lru/v2 v2.0.7 h1:a+bsQ5rvGLjzHuww6tVxozPZFVghXaHOwFs4luLUK2k=
|
||||||
|
github.com/hashicorp/golang-lru/v2 v2.0.7/go.mod h1:QeFd9opnmA6QUJc5vARoKUSoFhyfM2/ZepoAG6RGpeM=
|
||||||
github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
|
github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
|
||||||
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
|
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
|
||||||
|
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||||
|
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||||
|
github.com/ncruces/go-strftime v1.0.0 h1:HMFp8mLCTPp341M/ZnA4qaf7ZlsbTc+miZjCLOFAw7w=
|
||||||
|
github.com/ncruces/go-strftime v1.0.0/go.mod h1:Fwc5htZGVVkseilnfgOVb9mKy6w1naJmn9CehxcKcls=
|
||||||
github.com/oschwald/maxminddb-golang v1.13.1 h1:G3wwjdN9JmIK2o/ermkHM+98oX5fS+k5MbwsmL4MRQE=
|
github.com/oschwald/maxminddb-golang v1.13.1 h1:G3wwjdN9JmIK2o/ermkHM+98oX5fS+k5MbwsmL4MRQE=
|
||||||
github.com/oschwald/maxminddb-golang v1.13.1/go.mod h1:K4pgV9N/GcK694KSTmVSDTODk4IsCNThNdTmnaBZ/F8=
|
github.com/oschwald/maxminddb-golang v1.13.1/go.mod h1:K4pgV9N/GcK694KSTmVSDTODk4IsCNThNdTmnaBZ/F8=
|
||||||
github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0t5Ec4=
|
github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0t5Ec4=
|
||||||
github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
|
github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
|
||||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
|
||||||
|
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
|
||||||
github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
|
github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
|
||||||
github.com/spf13/cobra v1.10.2 h1:DMTTonx5m65Ic0GOoRY2c16WCbHxOOw6xxezuLaBpcU=
|
github.com/spf13/cobra v1.10.2 h1:DMTTonx5m65Ic0GOoRY2c16WCbHxOOw6xxezuLaBpcU=
|
||||||
github.com/spf13/cobra v1.10.2/go.mod h1:7C1pvHqHw5A4vrJfjNwvOdzYu0Gml16OCs2GRiTUUS4=
|
github.com/spf13/cobra v1.10.2/go.mod h1:7C1pvHqHw5A4vrJfjNwvOdzYu0Gml16OCs2GRiTUUS4=
|
||||||
@@ -41,12 +53,19 @@ go.opentelemetry.io/otel/sdk/metric v1.39.0/go.mod h1:xq9HEVH7qeX69/JnwEfp6fVq5w
|
|||||||
go.opentelemetry.io/otel/trace v1.39.0 h1:2d2vfpEDmCJ5zVYz7ijaJdOF59xLomrvj7bjt6/qCJI=
|
go.opentelemetry.io/otel/trace v1.39.0 h1:2d2vfpEDmCJ5zVYz7ijaJdOF59xLomrvj7bjt6/qCJI=
|
||||||
go.opentelemetry.io/otel/trace v1.39.0/go.mod h1:88w4/PnZSazkGzz/w84VHpQafiU4EtqqlVdxWy+rNOA=
|
go.opentelemetry.io/otel/trace v1.39.0/go.mod h1:88w4/PnZSazkGzz/w84VHpQafiU4EtqqlVdxWy+rNOA=
|
||||||
go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
|
go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
|
||||||
|
golang.org/x/mod v0.33.0 h1:tHFzIWbBifEmbwtGz65eaWyGiGZatSrT9prnU8DbVL8=
|
||||||
|
golang.org/x/mod v0.33.0/go.mod h1:swjeQEj+6r7fODbD2cqrnje9PnziFuw4bmLbBZFrQ5w=
|
||||||
golang.org/x/net v0.48.0 h1:zyQRTTrjc33Lhh0fBgT/H3oZq9WuvRR5gPC70xpDiQU=
|
golang.org/x/net v0.48.0 h1:zyQRTTrjc33Lhh0fBgT/H3oZq9WuvRR5gPC70xpDiQU=
|
||||||
golang.org/x/net v0.48.0/go.mod h1:+ndRgGjkh8FGtu1w1FGbEC31if4VrNVMuKTgcAAnQRY=
|
golang.org/x/net v0.48.0/go.mod h1:+ndRgGjkh8FGtu1w1FGbEC31if4VrNVMuKTgcAAnQRY=
|
||||||
golang.org/x/sys v0.39.0 h1:CvCKL8MeisomCi6qNZ+wbb0DN9E5AATixKsvNtMoMFk=
|
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
|
||||||
golang.org/x/sys v0.39.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
||||||
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.42.0 h1:omrd2nAlyT5ESRdCLYdm3+fMfNFE/+Rf4bDIQImRJeo=
|
||||||
|
golang.org/x/sys v0.42.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
|
||||||
golang.org/x/text v0.32.0 h1:ZD01bjUt1FQ9WJ0ClOL5vxgxOI/sVCNgX1YtKwcY0mU=
|
golang.org/x/text v0.32.0 h1:ZD01bjUt1FQ9WJ0ClOL5vxgxOI/sVCNgX1YtKwcY0mU=
|
||||||
golang.org/x/text v0.32.0/go.mod h1:o/rUWzghvpD5TXrTIBuJU77MTaN0ljMWE47kxGJQ7jY=
|
golang.org/x/text v0.32.0/go.mod h1:o/rUWzghvpD5TXrTIBuJU77MTaN0ljMWE47kxGJQ7jY=
|
||||||
|
golang.org/x/tools v0.42.0 h1:uNgphsn75Tdz5Ji2q36v/nsFSfR/9BRFvqhGBaJGd5k=
|
||||||
|
golang.org/x/tools v0.42.0/go.mod h1:Ma6lCIwGZvHK6XtgbswSoWroEkhugApmsXyrUmBhfr0=
|
||||||
gonum.org/v1/gonum v0.16.0 h1:5+ul4Swaf3ESvrOnidPp4GZbzf0mxVQpDCYUQE7OJfk=
|
gonum.org/v1/gonum v0.16.0 h1:5+ul4Swaf3ESvrOnidPp4GZbzf0mxVQpDCYUQE7OJfk=
|
||||||
gonum.org/v1/gonum v0.16.0/go.mod h1:fef3am4MQ93R2HHpKnLk4/Tbh/s0+wqD5nfa6Pnwy4E=
|
gonum.org/v1/gonum v0.16.0/go.mod h1:fef3am4MQ93R2HHpKnLk4/Tbh/s0+wqD5nfa6Pnwy4E=
|
||||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20251202230838-ff82c1b0f217 h1:gRkg/vSppuSQoDjxyiGfN4Upv/h/DQmIR10ZU8dh4Ww=
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20251202230838-ff82c1b0f217 h1:gRkg/vSppuSQoDjxyiGfN4Upv/h/DQmIR10ZU8dh4Ww=
|
||||||
@@ -58,3 +77,31 @@ google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j
|
|||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
modernc.org/cc/v4 v4.27.1 h1:9W30zRlYrefrDV2JE2O8VDtJ1yPGownxciz5rrbQZis=
|
||||||
|
modernc.org/cc/v4 v4.27.1/go.mod h1:uVtb5OGqUKpoLWhqwNQo/8LwvoiEBLvZXIQ/SmO6mL0=
|
||||||
|
modernc.org/ccgo/v4 v4.32.0 h1:hjG66bI/kqIPX1b2yT6fr/jt+QedtP2fqojG2VrFuVw=
|
||||||
|
modernc.org/ccgo/v4 v4.32.0/go.mod h1:6F08EBCx5uQc38kMGl+0Nm0oWczoo1c7cgpzEry7Uc0=
|
||||||
|
modernc.org/fileutil v1.4.0 h1:j6ZzNTftVS054gi281TyLjHPp6CPHr2KCxEXjEbD6SM=
|
||||||
|
modernc.org/fileutil v1.4.0/go.mod h1:EqdKFDxiByqxLk8ozOxObDSfcVOv/54xDs/DUHdvCUU=
|
||||||
|
modernc.org/gc/v2 v2.6.5 h1:nyqdV8q46KvTpZlsw66kWqwXRHdjIlJOhG6kxiV/9xI=
|
||||||
|
modernc.org/gc/v2 v2.6.5/go.mod h1:YgIahr1ypgfe7chRuJi2gD7DBQiKSLMPgBQe9oIiito=
|
||||||
|
modernc.org/gc/v3 v3.1.2 h1:ZtDCnhonXSZexk/AYsegNRV1lJGgaNZJuKjJSWKyEqo=
|
||||||
|
modernc.org/gc/v3 v3.1.2/go.mod h1:HFK/6AGESC7Ex+EZJhJ2Gni6cTaYpSMmU/cT9RmlfYY=
|
||||||
|
modernc.org/goabi0 v0.2.0 h1:HvEowk7LxcPd0eq6mVOAEMai46V+i7Jrj13t4AzuNks=
|
||||||
|
modernc.org/goabi0 v0.2.0/go.mod h1:CEFRnnJhKvWT1c1JTI3Avm+tgOWbkOu5oPA8eH8LnMI=
|
||||||
|
modernc.org/libc v1.70.0 h1:U58NawXqXbgpZ/dcdS9kMshu08aiA6b7gusEusqzNkw=
|
||||||
|
modernc.org/libc v1.70.0/go.mod h1:OVmxFGP1CI/Z4L3E0Q3Mf1PDE0BucwMkcXjjLntvHJo=
|
||||||
|
modernc.org/mathutil v1.7.1 h1:GCZVGXdaN8gTqB1Mf/usp1Y/hSqgI2vAGGP4jZMCxOU=
|
||||||
|
modernc.org/mathutil v1.7.1/go.mod h1:4p5IwJITfppl0G4sUEDtCr4DthTaT47/N3aT6MhfgJg=
|
||||||
|
modernc.org/memory v1.11.0 h1:o4QC8aMQzmcwCK3t3Ux/ZHmwFPzE6hf2Y5LbkRs+hbI=
|
||||||
|
modernc.org/memory v1.11.0/go.mod h1:/JP4VbVC+K5sU2wZi9bHoq2MAkCnrt2r98UGeSK7Mjw=
|
||||||
|
modernc.org/opt v0.1.4 h1:2kNGMRiUjrp4LcaPuLY2PzUfqM/w9N23quVwhKt5Qm8=
|
||||||
|
modernc.org/opt v0.1.4/go.mod h1:03fq9lsNfvkYSfxrfUhZCWPk1lm4cq4N+Bh//bEtgns=
|
||||||
|
modernc.org/sortutil v1.2.1 h1:+xyoGf15mM3NMlPDnFqrteY07klSFxLElE2PVuWIJ7w=
|
||||||
|
modernc.org/sortutil v1.2.1/go.mod h1:7ZI3a3REbai7gzCLcotuw9AC4VZVpYMjDzETGsSMqJE=
|
||||||
|
modernc.org/sqlite v1.46.2 h1:gkXQ6R0+AjxFC/fTDaeIVLbNLNrRoOK7YYVz5BKhTcE=
|
||||||
|
modernc.org/sqlite v1.46.2/go.mod h1:hWjRO6Tj/5Ik8ieqxQybiEOUXy0NJFNp2tpvVpKlvig=
|
||||||
|
modernc.org/strutil v1.2.1 h1:UneZBkQA+DX2Rp35KcM69cSsNES9ly8mQWD71HKlOA0=
|
||||||
|
modernc.org/strutil v1.2.1/go.mod h1:EHkiggD70koQxjVdSBM3JKM7k6L0FbGE5eymy9i3B9A=
|
||||||
|
modernc.org/token v1.1.0 h1:Xl7Ap9dKaEs5kLoOQeQmPWevfnk/DM5qcLcYlA8ys6Y=
|
||||||
|
modernc.org/token v1.1.0/go.mod h1:UGzOrNV1mAFSEB63lOFHIpNRUVMvYTc6yu1SMY/XTDM=
|
||||||
|
|||||||
@@ -10,12 +10,17 @@ import (
|
|||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
Listeners []Listener `toml:"listeners"`
|
Listeners []Listener `toml:"listeners"`
|
||||||
|
Database Database `toml:"database"`
|
||||||
GRPC GRPC `toml:"grpc"`
|
GRPC GRPC `toml:"grpc"`
|
||||||
Firewall Firewall `toml:"firewall"`
|
Firewall Firewall `toml:"firewall"`
|
||||||
Proxy Proxy `toml:"proxy"`
|
Proxy Proxy `toml:"proxy"`
|
||||||
Log Log `toml:"log"`
|
Log Log `toml:"log"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type Database struct {
|
||||||
|
Path string `toml:"path"`
|
||||||
|
}
|
||||||
|
|
||||||
type GRPC struct {
|
type GRPC struct {
|
||||||
Addr string `toml:"addr"`
|
Addr string `toml:"addr"`
|
||||||
TLSCert string `toml:"tls_cert"`
|
TLSCert string `toml:"tls_cert"`
|
||||||
@@ -80,18 +85,15 @@ func Load(path string) (*Config, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) validate() error {
|
func (c *Config) validate() error {
|
||||||
if len(c.Listeners) == 0 {
|
if c.Database.Path == "" {
|
||||||
return fmt.Errorf("at least one listener is required")
|
return fmt.Errorf("database.path is required")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Validate listeners if provided (used for seeding on first run).
|
||||||
for i, l := range c.Listeners {
|
for i, l := range c.Listeners {
|
||||||
if l.Addr == "" {
|
if l.Addr == "" {
|
||||||
return fmt.Errorf("listener %d: addr is required", i)
|
return fmt.Errorf("listener %d: addr is required", i)
|
||||||
}
|
}
|
||||||
if len(l.Routes) == 0 {
|
|
||||||
return fmt.Errorf("listener %d (%s): at least one route is required", i, l.Addr)
|
|
||||||
}
|
|
||||||
|
|
||||||
seen := make(map[string]bool)
|
seen := make(map[string]bool)
|
||||||
for j, r := range l.Routes {
|
for j, r := range l.Routes {
|
||||||
if r.Hostname == "" {
|
if r.Hostname == "" {
|
||||||
|
|||||||
@@ -11,6 +11,9 @@ func TestLoadValid(t *testing.T) {
|
|||||||
path := filepath.Join(dir, "test.toml")
|
path := filepath.Join(dir, "test.toml")
|
||||||
|
|
||||||
data := `
|
data := `
|
||||||
|
[database]
|
||||||
|
path = "/tmp/test.db"
|
||||||
|
|
||||||
[[listeners]]
|
[[listeners]]
|
||||||
addr = ":443"
|
addr = ":443"
|
||||||
|
|
||||||
@@ -49,11 +52,37 @@ level = "info"
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLoadNoListeners(t *testing.T) {
|
func TestLoadNoDatabasePath(t *testing.T) {
|
||||||
dir := t.TempDir()
|
dir := t.TempDir()
|
||||||
path := filepath.Join(dir, "test.toml")
|
path := filepath.Join(dir, "test.toml")
|
||||||
|
|
||||||
data := `
|
data := `
|
||||||
|
[[listeners]]
|
||||||
|
addr = ":443"
|
||||||
|
|
||||||
|
[[listeners.routes]]
|
||||||
|
hostname = "example.com"
|
||||||
|
backend = "127.0.0.1:8443"
|
||||||
|
`
|
||||||
|
if err := os.WriteFile(path, []byte(data), 0600); err != nil {
|
||||||
|
t.Fatalf("write config: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err := Load(path)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("expected error for missing database path")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadNoListenersValid(t *testing.T) {
|
||||||
|
dir := t.TempDir()
|
||||||
|
path := filepath.Join(dir, "test.toml")
|
||||||
|
|
||||||
|
// No listeners is valid — DB may already have them.
|
||||||
|
data := `
|
||||||
|
[database]
|
||||||
|
path = "/tmp/test.db"
|
||||||
|
|
||||||
[log]
|
[log]
|
||||||
level = "info"
|
level = "info"
|
||||||
`
|
`
|
||||||
@@ -62,26 +91,8 @@ level = "info"
|
|||||||
}
|
}
|
||||||
|
|
||||||
_, err := Load(path)
|
_, err := Load(path)
|
||||||
if err == nil {
|
if err != nil {
|
||||||
t.Fatal("expected error for missing listeners")
|
t.Fatalf("unexpected error: %v", err)
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoadNoRoutes(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
path := filepath.Join(dir, "test.toml")
|
|
||||||
|
|
||||||
data := `
|
|
||||||
[[listeners]]
|
|
||||||
addr = ":443"
|
|
||||||
`
|
|
||||||
if err := os.WriteFile(path, []byte(data), 0600); err != nil {
|
|
||||||
t.Fatalf("write config: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err := Load(path)
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("expected error for missing routes")
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -90,6 +101,9 @@ func TestLoadDuplicateHostnames(t *testing.T) {
|
|||||||
path := filepath.Join(dir, "test.toml")
|
path := filepath.Join(dir, "test.toml")
|
||||||
|
|
||||||
data := `
|
data := `
|
||||||
|
[database]
|
||||||
|
path = "/tmp/test.db"
|
||||||
|
|
||||||
[[listeners]]
|
[[listeners]]
|
||||||
addr = ":443"
|
addr = ":443"
|
||||||
|
|
||||||
@@ -116,6 +130,9 @@ func TestLoadGeoIPRequiredWithCountries(t *testing.T) {
|
|||||||
path := filepath.Join(dir, "test.toml")
|
path := filepath.Join(dir, "test.toml")
|
||||||
|
|
||||||
data := `
|
data := `
|
||||||
|
[database]
|
||||||
|
path = "/tmp/test.db"
|
||||||
|
|
||||||
[[listeners]]
|
[[listeners]]
|
||||||
addr = ":443"
|
addr = ":443"
|
||||||
|
|
||||||
@@ -141,6 +158,9 @@ func TestLoadMultipleListeners(t *testing.T) {
|
|||||||
path := filepath.Join(dir, "test.toml")
|
path := filepath.Join(dir, "test.toml")
|
||||||
|
|
||||||
data := `
|
data := `
|
||||||
|
[database]
|
||||||
|
path = "/tmp/test.db"
|
||||||
|
|
||||||
[[listeners]]
|
[[listeners]]
|
||||||
addr = ":443"
|
addr = ":443"
|
||||||
|
|
||||||
|
|||||||
63
internal/db/db.go
Normal file
63
internal/db/db.go
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
package db
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
_ "modernc.org/sqlite"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Store wraps a SQLite database connection for mc-proxy persistence.
|
||||||
|
type Store struct {
|
||||||
|
db *sql.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open opens (or creates) the SQLite database at path with WAL mode,
|
||||||
|
// foreign keys, and a busy timeout. The file is created with 0600 permissions.
|
||||||
|
func Open(path string) (*Store, error) {
|
||||||
|
// Ensure the file has restrictive permissions if it doesn't exist.
|
||||||
|
if _, err := os.Stat(path); os.IsNotExist(err) {
|
||||||
|
f, err := os.OpenFile(path, os.O_CREATE|os.O_RDWR, 0600)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("creating database file: %w", err)
|
||||||
|
}
|
||||||
|
f.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
db, err := sql.Open("sqlite", path)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("opening database: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Apply connection pragmas.
|
||||||
|
pragmas := []string{
|
||||||
|
"PRAGMA journal_mode = WAL",
|
||||||
|
"PRAGMA foreign_keys = ON",
|
||||||
|
"PRAGMA busy_timeout = 5000",
|
||||||
|
}
|
||||||
|
for _, p := range pragmas {
|
||||||
|
if _, err := db.Exec(p); err != nil {
|
||||||
|
db.Close()
|
||||||
|
return nil, fmt.Errorf("setting pragma %q: %w", p, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return &Store{db: db}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close closes the database connection.
|
||||||
|
func (s *Store) Close() error {
|
||||||
|
return s.db.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsEmpty returns true if the listeners table has no rows.
|
||||||
|
// Used to determine if the database needs seeding from config.
|
||||||
|
func (s *Store) IsEmpty() (bool, error) {
|
||||||
|
var count int
|
||||||
|
err := s.db.QueryRow("SELECT COUNT(*) FROM listeners").Scan(&count)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
return count == 0, nil
|
||||||
|
}
|
||||||
331
internal/db/db_test.go
Normal file
331
internal/db/db_test.go
Normal file
@@ -0,0 +1,331 @@
|
|||||||
|
package db
|
||||||
|
|
||||||
|
import (
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"git.wntrmute.dev/kyle/mc-proxy/internal/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
func openTestDB(t *testing.T) *Store {
|
||||||
|
t.Helper()
|
||||||
|
dir := t.TempDir()
|
||||||
|
store, err := Open(filepath.Join(dir, "test.db"))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("open: %v", err)
|
||||||
|
}
|
||||||
|
if err := store.Migrate(); err != nil {
|
||||||
|
t.Fatalf("migrate: %v", err)
|
||||||
|
}
|
||||||
|
t.Cleanup(func() { store.Close() })
|
||||||
|
return store
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMigrate(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
// Running migrate again should be idempotent.
|
||||||
|
if err := store.Migrate(); err != nil {
|
||||||
|
t.Fatalf("second migrate: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIsEmpty(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
empty, err := store.IsEmpty()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("is empty: %v", err)
|
||||||
|
}
|
||||||
|
if !empty {
|
||||||
|
t.Fatal("expected empty database")
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := store.CreateListener(":443"); err != nil {
|
||||||
|
t.Fatalf("create listener: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
empty, err = store.IsEmpty()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("is empty: %v", err)
|
||||||
|
}
|
||||||
|
if empty {
|
||||||
|
t.Fatal("expected non-empty database")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestListenerCRUD(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
id, err := store.CreateListener(":443")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("create: %v", err)
|
||||||
|
}
|
||||||
|
if id == 0 {
|
||||||
|
t.Fatal("expected non-zero ID")
|
||||||
|
}
|
||||||
|
|
||||||
|
listeners, err := store.ListListeners()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("list: %v", err)
|
||||||
|
}
|
||||||
|
if len(listeners) != 1 {
|
||||||
|
t.Fatalf("got %d listeners, want 1", len(listeners))
|
||||||
|
}
|
||||||
|
if listeners[0].Addr != ":443" {
|
||||||
|
t.Fatalf("got addr %q, want %q", listeners[0].Addr, ":443")
|
||||||
|
}
|
||||||
|
|
||||||
|
l, err := store.GetListenerByAddr(":443")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("get by addr: %v", err)
|
||||||
|
}
|
||||||
|
if l.ID != id {
|
||||||
|
t.Fatalf("got ID %d, want %d", l.ID, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := store.DeleteListener(id); err != nil {
|
||||||
|
t.Fatalf("delete: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
listeners, err = store.ListListeners()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("list after delete: %v", err)
|
||||||
|
}
|
||||||
|
if len(listeners) != 0 {
|
||||||
|
t.Fatalf("got %d listeners after delete, want 0", len(listeners))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestListenerDuplicateAddr(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
if _, err := store.CreateListener(":443"); err != nil {
|
||||||
|
t.Fatalf("first create: %v", err)
|
||||||
|
}
|
||||||
|
if _, err := store.CreateListener(":443"); err == nil {
|
||||||
|
t.Fatal("expected error for duplicate addr")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRouteCRUD(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
listenerID, err := store.CreateListener(":443")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("create listener: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
routeID, err := store.CreateRoute(listenerID, "example.com", "127.0.0.1:8443")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("create route: %v", err)
|
||||||
|
}
|
||||||
|
if routeID == 0 {
|
||||||
|
t.Fatal("expected non-zero route ID")
|
||||||
|
}
|
||||||
|
|
||||||
|
routes, err := store.ListRoutes(listenerID)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("list routes: %v", err)
|
||||||
|
}
|
||||||
|
if len(routes) != 1 {
|
||||||
|
t.Fatalf("got %d routes, want 1", len(routes))
|
||||||
|
}
|
||||||
|
if routes[0].Hostname != "example.com" {
|
||||||
|
t.Fatalf("got hostname %q, want %q", routes[0].Hostname, "example.com")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := store.DeleteRoute(listenerID, "example.com"); err != nil {
|
||||||
|
t.Fatalf("delete route: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
routes, err = store.ListRoutes(listenerID)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("list after delete: %v", err)
|
||||||
|
}
|
||||||
|
if len(routes) != 0 {
|
||||||
|
t.Fatalf("got %d routes after delete, want 0", len(routes))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRouteDuplicateHostname(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
listenerID, _ := store.CreateListener(":443")
|
||||||
|
if _, err := store.CreateRoute(listenerID, "example.com", "127.0.0.1:8443"); err != nil {
|
||||||
|
t.Fatalf("first create: %v", err)
|
||||||
|
}
|
||||||
|
if _, err := store.CreateRoute(listenerID, "example.com", "127.0.0.1:9443"); err == nil {
|
||||||
|
t.Fatal("expected error for duplicate hostname on same listener")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRouteCascadeDelete(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
listenerID, _ := store.CreateListener(":443")
|
||||||
|
store.CreateRoute(listenerID, "a.example.com", "127.0.0.1:8443")
|
||||||
|
store.CreateRoute(listenerID, "b.example.com", "127.0.0.1:9443")
|
||||||
|
|
||||||
|
if err := store.DeleteListener(listenerID); err != nil {
|
||||||
|
t.Fatalf("delete listener: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
routes, err := store.ListRoutes(listenerID)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("list routes: %v", err)
|
||||||
|
}
|
||||||
|
if len(routes) != 0 {
|
||||||
|
t.Fatalf("got %d routes after cascade delete, want 0", len(routes))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFirewallRuleCRUD(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
id, err := store.CreateFirewallRule("ip", "192.0.2.1")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("create: %v", err)
|
||||||
|
}
|
||||||
|
if id == 0 {
|
||||||
|
t.Fatal("expected non-zero ID")
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := store.CreateFirewallRule("cidr", "198.51.100.0/24"); err != nil {
|
||||||
|
t.Fatalf("create cidr: %v", err)
|
||||||
|
}
|
||||||
|
if _, err := store.CreateFirewallRule("country", "CN"); err != nil {
|
||||||
|
t.Fatalf("create country: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
rules, err := store.ListFirewallRules()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("list: %v", err)
|
||||||
|
}
|
||||||
|
if len(rules) != 3 {
|
||||||
|
t.Fatalf("got %d rules, want 3", len(rules))
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := store.DeleteFirewallRule("ip", "192.0.2.1"); err != nil {
|
||||||
|
t.Fatalf("delete: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
rules, err = store.ListFirewallRules()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("list after delete: %v", err)
|
||||||
|
}
|
||||||
|
if len(rules) != 2 {
|
||||||
|
t.Fatalf("got %d rules after delete, want 2", len(rules))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFirewallRuleDuplicate(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
if _, err := store.CreateFirewallRule("ip", "192.0.2.1"); err != nil {
|
||||||
|
t.Fatalf("first create: %v", err)
|
||||||
|
}
|
||||||
|
if _, err := store.CreateFirewallRule("ip", "192.0.2.1"); err == nil {
|
||||||
|
t.Fatal("expected error for duplicate rule")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSeed(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
listeners := []config.Listener{
|
||||||
|
{
|
||||||
|
Addr: ":443",
|
||||||
|
Routes: []config.Route{
|
||||||
|
{Hostname: "a.example.com", Backend: "127.0.0.1:8443"},
|
||||||
|
{Hostname: "b.example.com", Backend: "127.0.0.1:9443"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Addr: ":8443",
|
||||||
|
Routes: []config.Route{
|
||||||
|
{Hostname: "c.example.com", Backend: "127.0.0.1:18443"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
fw := config.Firewall{
|
||||||
|
BlockedIPs: []string{"192.0.2.1"},
|
||||||
|
BlockedCIDRs: []string{"198.51.100.0/24"},
|
||||||
|
BlockedCountries: []string{"cn", "KP"},
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := store.Seed(listeners, fw); err != nil {
|
||||||
|
t.Fatalf("seed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
dbListeners, err := store.ListListeners()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("list listeners: %v", err)
|
||||||
|
}
|
||||||
|
if len(dbListeners) != 2 {
|
||||||
|
t.Fatalf("got %d listeners, want 2", len(dbListeners))
|
||||||
|
}
|
||||||
|
|
||||||
|
routes, err := store.ListRoutes(dbListeners[0].ID)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("list routes: %v", err)
|
||||||
|
}
|
||||||
|
if len(routes) != 2 {
|
||||||
|
t.Fatalf("got %d routes for listener 0, want 2", len(routes))
|
||||||
|
}
|
||||||
|
|
||||||
|
rules, err := store.ListFirewallRules()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("list firewall rules: %v", err)
|
||||||
|
}
|
||||||
|
if len(rules) != 4 {
|
||||||
|
t.Fatalf("got %d firewall rules, want 4", len(rules))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSnapshot(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
store.CreateListener(":443")
|
||||||
|
|
||||||
|
dest := filepath.Join(t.TempDir(), "backup.db")
|
||||||
|
if err := store.Snapshot(dest); err != nil {
|
||||||
|
t.Fatalf("snapshot: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open the snapshot and verify.
|
||||||
|
backup, err := Open(dest)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("open backup: %v", err)
|
||||||
|
}
|
||||||
|
defer backup.Close()
|
||||||
|
|
||||||
|
if err := backup.Migrate(); err != nil {
|
||||||
|
t.Fatalf("migrate backup: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
listeners, err := backup.ListListeners()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("list from backup: %v", err)
|
||||||
|
}
|
||||||
|
if len(listeners) != 1 {
|
||||||
|
t.Fatalf("backup has %d listeners, want 1", len(listeners))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDeleteNonexistent(t *testing.T) {
|
||||||
|
store := openTestDB(t)
|
||||||
|
|
||||||
|
if err := store.DeleteListener(999); err == nil {
|
||||||
|
t.Fatal("expected error deleting nonexistent listener")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := store.DeleteRoute(999, "example.com"); err == nil {
|
||||||
|
t.Fatal("expected error deleting nonexistent route")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := store.DeleteFirewallRule("ip", "1.2.3.4"); err == nil {
|
||||||
|
t.Fatal("expected error deleting nonexistent firewall rule")
|
||||||
|
}
|
||||||
|
}
|
||||||
57
internal/db/firewall.go
Normal file
57
internal/db/firewall.go
Normal file
@@ -0,0 +1,57 @@
|
|||||||
|
package db
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
// FirewallRule is a database firewall rule record.
|
||||||
|
type FirewallRule struct {
|
||||||
|
ID int64
|
||||||
|
Type string // "ip", "cidr", "country"
|
||||||
|
Value string
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListFirewallRules returns all firewall rules.
|
||||||
|
func (s *Store) ListFirewallRules() ([]FirewallRule, error) {
|
||||||
|
rows, err := s.db.Query("SELECT id, type, value FROM firewall_rules ORDER BY type, value")
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("querying firewall rules: %w", err)
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
|
||||||
|
var rules []FirewallRule
|
||||||
|
for rows.Next() {
|
||||||
|
var r FirewallRule
|
||||||
|
if err := rows.Scan(&r.ID, &r.Type, &r.Value); err != nil {
|
||||||
|
return nil, fmt.Errorf("scanning firewall rule: %w", err)
|
||||||
|
}
|
||||||
|
rules = append(rules, r)
|
||||||
|
}
|
||||||
|
return rules, rows.Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateFirewallRule inserts a firewall rule and returns its ID.
|
||||||
|
func (s *Store) CreateFirewallRule(ruleType, value string) (int64, error) {
|
||||||
|
result, err := s.db.Exec(
|
||||||
|
"INSERT INTO firewall_rules (type, value) VALUES (?, ?)",
|
||||||
|
ruleType, value,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return 0, fmt.Errorf("inserting firewall rule: %w", err)
|
||||||
|
}
|
||||||
|
return result.LastInsertId()
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteFirewallRule deletes a firewall rule by type and value.
|
||||||
|
func (s *Store) DeleteFirewallRule(ruleType, value string) error {
|
||||||
|
result, err := s.db.Exec(
|
||||||
|
"DELETE FROM firewall_rules WHERE type = ? AND value = ?",
|
||||||
|
ruleType, value,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("deleting firewall rule: %w", err)
|
||||||
|
}
|
||||||
|
n, _ := result.RowsAffected()
|
||||||
|
if n == 0 {
|
||||||
|
return fmt.Errorf("firewall rule (%s, %s) not found", ruleType, value)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
61
internal/db/listeners.go
Normal file
61
internal/db/listeners.go
Normal file
@@ -0,0 +1,61 @@
|
|||||||
|
package db
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
// Listener is a database listener record.
|
||||||
|
type Listener struct {
|
||||||
|
ID int64
|
||||||
|
Addr string
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListListeners returns all listeners.
|
||||||
|
func (s *Store) ListListeners() ([]Listener, error) {
|
||||||
|
rows, err := s.db.Query("SELECT id, addr FROM listeners ORDER BY id")
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("querying listeners: %w", err)
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
|
||||||
|
var listeners []Listener
|
||||||
|
for rows.Next() {
|
||||||
|
var l Listener
|
||||||
|
if err := rows.Scan(&l.ID, &l.Addr); err != nil {
|
||||||
|
return nil, fmt.Errorf("scanning listener: %w", err)
|
||||||
|
}
|
||||||
|
listeners = append(listeners, l)
|
||||||
|
}
|
||||||
|
return listeners, rows.Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateListener inserts a listener and returns its ID.
|
||||||
|
func (s *Store) CreateListener(addr string) (int64, error) {
|
||||||
|
result, err := s.db.Exec("INSERT INTO listeners (addr) VALUES (?)", addr)
|
||||||
|
if err != nil {
|
||||||
|
return 0, fmt.Errorf("inserting listener: %w", err)
|
||||||
|
}
|
||||||
|
return result.LastInsertId()
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteListener deletes a listener by ID. Routes are cascade-deleted.
|
||||||
|
func (s *Store) DeleteListener(id int64) error {
|
||||||
|
result, err := s.db.Exec("DELETE FROM listeners WHERE id = ?", id)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("deleting listener: %w", err)
|
||||||
|
}
|
||||||
|
n, _ := result.RowsAffected()
|
||||||
|
if n == 0 {
|
||||||
|
return fmt.Errorf("listener %d not found", id)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetListenerByAddr returns a listener by its address.
|
||||||
|
func (s *Store) GetListenerByAddr(addr string) (Listener, error) {
|
||||||
|
var l Listener
|
||||||
|
err := s.db.QueryRow("SELECT id, addr FROM listeners WHERE addr = ?", addr).
|
||||||
|
Scan(&l.ID, &l.Addr)
|
||||||
|
if err != nil {
|
||||||
|
return Listener{}, fmt.Errorf("querying listener by addr %q: %w", addr, err)
|
||||||
|
}
|
||||||
|
return l, nil
|
||||||
|
}
|
||||||
93
internal/db/migrations.go
Normal file
93
internal/db/migrations.go
Normal file
@@ -0,0 +1,93 @@
|
|||||||
|
package db
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
type migration struct {
|
||||||
|
version int
|
||||||
|
name string
|
||||||
|
fn func(tx *sql.Tx) error
|
||||||
|
}
|
||||||
|
|
||||||
|
var migrations = []migration{
|
||||||
|
{1, "create_core_tables", migrate001CreateCoreTables},
|
||||||
|
}
|
||||||
|
|
||||||
|
// Migrate runs all unapplied migrations sequentially.
|
||||||
|
func (s *Store) Migrate() error {
|
||||||
|
// Ensure the migration tracking table exists.
|
||||||
|
_, err := s.db.Exec(`
|
||||||
|
CREATE TABLE IF NOT EXISTS schema_migrations (
|
||||||
|
version INTEGER PRIMARY KEY,
|
||||||
|
applied TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%SZ', 'now'))
|
||||||
|
)
|
||||||
|
`)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("creating schema_migrations table: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var current int
|
||||||
|
err = s.db.QueryRow("SELECT COALESCE(MAX(version), 0) FROM schema_migrations").Scan(¤t)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("querying current migration version: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, m := range migrations {
|
||||||
|
if m.version <= current {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
tx, err := s.db.Begin()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("beginning migration %d (%s): %w", m.version, m.name, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := m.fn(tx); err != nil {
|
||||||
|
tx.Rollback()
|
||||||
|
return fmt.Errorf("running migration %d (%s): %w", m.version, m.name, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := tx.Exec("INSERT INTO schema_migrations (version) VALUES (?)", m.version); err != nil {
|
||||||
|
tx.Rollback()
|
||||||
|
return fmt.Errorf("recording migration %d (%s): %w", m.version, m.name, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := tx.Commit(); err != nil {
|
||||||
|
return fmt.Errorf("committing migration %d (%s): %w", m.version, m.name, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func migrate001CreateCoreTables(tx *sql.Tx) error {
|
||||||
|
stmts := []string{
|
||||||
|
`CREATE TABLE IF NOT EXISTS listeners (
|
||||||
|
id INTEGER PRIMARY KEY,
|
||||||
|
addr TEXT NOT NULL UNIQUE
|
||||||
|
)`,
|
||||||
|
`CREATE TABLE IF NOT EXISTS routes (
|
||||||
|
id INTEGER PRIMARY KEY,
|
||||||
|
listener_id INTEGER NOT NULL REFERENCES listeners(id) ON DELETE CASCADE,
|
||||||
|
hostname TEXT NOT NULL,
|
||||||
|
backend TEXT NOT NULL,
|
||||||
|
UNIQUE(listener_id, hostname)
|
||||||
|
)`,
|
||||||
|
`CREATE INDEX IF NOT EXISTS idx_routes_listener ON routes(listener_id)`,
|
||||||
|
`CREATE TABLE IF NOT EXISTS firewall_rules (
|
||||||
|
id INTEGER PRIMARY KEY,
|
||||||
|
type TEXT NOT NULL CHECK(type IN ('ip', 'cidr', 'country')),
|
||||||
|
value TEXT NOT NULL,
|
||||||
|
UNIQUE(type, value)
|
||||||
|
)`,
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, stmt := range stmts {
|
||||||
|
if _, err := tx.Exec(stmt); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
61
internal/db/routes.go
Normal file
61
internal/db/routes.go
Normal file
@@ -0,0 +1,61 @@
|
|||||||
|
package db
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
// Route is a database route record.
|
||||||
|
type Route struct {
|
||||||
|
ID int64
|
||||||
|
ListenerID int64
|
||||||
|
Hostname string
|
||||||
|
Backend string
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListRoutes returns all routes for a listener.
|
||||||
|
func (s *Store) ListRoutes(listenerID int64) ([]Route, error) {
|
||||||
|
rows, err := s.db.Query(
|
||||||
|
"SELECT id, listener_id, hostname, backend FROM routes WHERE listener_id = ? ORDER BY hostname",
|
||||||
|
listenerID,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("querying routes: %w", err)
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
|
||||||
|
var routes []Route
|
||||||
|
for rows.Next() {
|
||||||
|
var r Route
|
||||||
|
if err := rows.Scan(&r.ID, &r.ListenerID, &r.Hostname, &r.Backend); err != nil {
|
||||||
|
return nil, fmt.Errorf("scanning route: %w", err)
|
||||||
|
}
|
||||||
|
routes = append(routes, r)
|
||||||
|
}
|
||||||
|
return routes, rows.Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateRoute inserts a route and returns its ID.
|
||||||
|
func (s *Store) CreateRoute(listenerID int64, hostname, backend string) (int64, error) {
|
||||||
|
result, err := s.db.Exec(
|
||||||
|
"INSERT INTO routes (listener_id, hostname, backend) VALUES (?, ?, ?)",
|
||||||
|
listenerID, hostname, backend,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return 0, fmt.Errorf("inserting route: %w", err)
|
||||||
|
}
|
||||||
|
return result.LastInsertId()
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteRoute deletes a route by listener ID and hostname.
|
||||||
|
func (s *Store) DeleteRoute(listenerID int64, hostname string) error {
|
||||||
|
result, err := s.db.Exec(
|
||||||
|
"DELETE FROM routes WHERE listener_id = ? AND hostname = ?",
|
||||||
|
listenerID, hostname,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("deleting route: %w", err)
|
||||||
|
}
|
||||||
|
n, _ := result.RowsAffected()
|
||||||
|
if n == 0 {
|
||||||
|
return fmt.Errorf("route %q not found on listener %d", hostname, listenerID)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
56
internal/db/seed.go
Normal file
56
internal/db/seed.go
Normal file
@@ -0,0 +1,56 @@
|
|||||||
|
package db
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"git.wntrmute.dev/kyle/mc-proxy/internal/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Seed populates the database from TOML config data. Only called when the
|
||||||
|
// database is empty (first run).
|
||||||
|
func (s *Store) Seed(listeners []config.Listener, fw config.Firewall) error {
|
||||||
|
tx, err := s.db.Begin()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("beginning seed transaction: %w", err)
|
||||||
|
}
|
||||||
|
defer tx.Rollback()
|
||||||
|
|
||||||
|
for _, l := range listeners {
|
||||||
|
result, err := tx.Exec("INSERT INTO listeners (addr) VALUES (?)", l.Addr)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("seeding listener %q: %w", l.Addr, err)
|
||||||
|
}
|
||||||
|
listenerID, _ := result.LastInsertId()
|
||||||
|
|
||||||
|
for _, r := range l.Routes {
|
||||||
|
_, err := tx.Exec(
|
||||||
|
"INSERT INTO routes (listener_id, hostname, backend) VALUES (?, ?, ?)",
|
||||||
|
listenerID, strings.ToLower(r.Hostname), r.Backend,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("seeding route %q on listener %q: %w", r.Hostname, l.Addr, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, ip := range fw.BlockedIPs {
|
||||||
|
if _, err := tx.Exec("INSERT INTO firewall_rules (type, value) VALUES ('ip', ?)", ip); err != nil {
|
||||||
|
return fmt.Errorf("seeding blocked IP %q: %w", ip, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, cidr := range fw.BlockedCIDRs {
|
||||||
|
if _, err := tx.Exec("INSERT INTO firewall_rules (type, value) VALUES ('cidr', ?)", cidr); err != nil {
|
||||||
|
return fmt.Errorf("seeding blocked CIDR %q: %w", cidr, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, code := range fw.BlockedCountries {
|
||||||
|
if _, err := tx.Exec("INSERT INTO firewall_rules (type, value) VALUES ('country', ?)", strings.ToUpper(code)); err != nil {
|
||||||
|
return fmt.Errorf("seeding blocked country %q: %w", code, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return tx.Commit()
|
||||||
|
}
|
||||||
12
internal/db/snapshot.go
Normal file
12
internal/db/snapshot.go
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
package db
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
// Snapshot creates a consistent backup of the database using VACUUM INTO.
|
||||||
|
func (s *Store) Snapshot(destPath string) error {
|
||||||
|
_, err := s.db.Exec("VACUUM INTO ?", destPath)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("snapshot to %q: %w", destPath, err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -8,7 +8,6 @@ import (
|
|||||||
|
|
||||||
"github.com/oschwald/maxminddb-golang"
|
"github.com/oschwald/maxminddb-golang"
|
||||||
|
|
||||||
"git.wntrmute.dev/kyle/mc-proxy/internal/config"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type geoIPRecord struct {
|
type geoIPRecord struct {
|
||||||
@@ -27,15 +26,15 @@ type Firewall struct {
|
|||||||
mu sync.RWMutex // protects all mutable state
|
mu sync.RWMutex // protects all mutable state
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a Firewall from the given configuration.
|
// New creates a Firewall from raw rule lists and an optional GeoIP database path.
|
||||||
func New(cfg config.Firewall) (*Firewall, error) {
|
func New(geoIPPath string, ips, cidrs, countries []string) (*Firewall, error) {
|
||||||
f := &Firewall{
|
f := &Firewall{
|
||||||
blockedIPs: make(map[netip.Addr]struct{}),
|
blockedIPs: make(map[netip.Addr]struct{}),
|
||||||
blockedCountries: make(map[string]struct{}),
|
blockedCountries: make(map[string]struct{}),
|
||||||
geoDBPath: cfg.GeoIPDB,
|
geoDBPath: geoIPPath,
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, ip := range cfg.BlockedIPs {
|
for _, ip := range ips {
|
||||||
addr, err := netip.ParseAddr(ip)
|
addr, err := netip.ParseAddr(ip)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("invalid blocked IP %q: %w", ip, err)
|
return nil, fmt.Errorf("invalid blocked IP %q: %w", ip, err)
|
||||||
@@ -43,7 +42,7 @@ func New(cfg config.Firewall) (*Firewall, error) {
|
|||||||
f.blockedIPs[addr] = struct{}{}
|
f.blockedIPs[addr] = struct{}{}
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, cidr := range cfg.BlockedCIDRs {
|
for _, cidr := range cidrs {
|
||||||
prefix, err := netip.ParsePrefix(cidr)
|
prefix, err := netip.ParsePrefix(cidr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("invalid blocked CIDR %q: %w", cidr, err)
|
return nil, fmt.Errorf("invalid blocked CIDR %q: %w", cidr, err)
|
||||||
@@ -51,12 +50,12 @@ func New(cfg config.Firewall) (*Firewall, error) {
|
|||||||
f.blockedCIDRs = append(f.blockedCIDRs, prefix)
|
f.blockedCIDRs = append(f.blockedCIDRs, prefix)
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, code := range cfg.BlockedCountries {
|
for _, code := range countries {
|
||||||
f.blockedCountries[strings.ToUpper(code)] = struct{}{}
|
f.blockedCountries[strings.ToUpper(code)] = struct{}{}
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(f.blockedCountries) > 0 {
|
if len(f.blockedCountries) > 0 && geoIPPath != "" {
|
||||||
if err := f.loadGeoDB(cfg.GeoIPDB); err != nil {
|
if err := f.loadGeoDB(geoIPPath); err != nil {
|
||||||
return nil, fmt.Errorf("loading GeoIP database: %w", err)
|
return nil, fmt.Errorf("loading GeoIP database: %w", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,12 +3,10 @@ package firewall
|
|||||||
import (
|
import (
|
||||||
"net/netip"
|
"net/netip"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"git.wntrmute.dev/kyle/mc-proxy/internal/config"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestEmptyFirewall(t *testing.T) {
|
func TestEmptyFirewall(t *testing.T) {
|
||||||
fw, err := New(config.Firewall{})
|
fw, err := New("", nil, nil, nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("unexpected error: %v", err)
|
t.Fatalf("unexpected error: %v", err)
|
||||||
}
|
}
|
||||||
@@ -24,9 +22,7 @@ func TestEmptyFirewall(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestIPBlocking(t *testing.T) {
|
func TestIPBlocking(t *testing.T) {
|
||||||
fw, err := New(config.Firewall{
|
fw, err := New("", []string{"192.0.2.1", "2001:db8::dead"}, nil, nil)
|
||||||
BlockedIPs: []string{"192.0.2.1", "2001:db8::dead"},
|
|
||||||
})
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("unexpected error: %v", err)
|
t.Fatalf("unexpected error: %v", err)
|
||||||
}
|
}
|
||||||
@@ -51,9 +47,7 @@ func TestIPBlocking(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestCIDRBlocking(t *testing.T) {
|
func TestCIDRBlocking(t *testing.T) {
|
||||||
fw, err := New(config.Firewall{
|
fw, err := New("", nil, []string{"198.51.100.0/24", "2001:db8::/32"}, nil)
|
||||||
BlockedCIDRs: []string{"198.51.100.0/24", "2001:db8::/32"},
|
|
||||||
})
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("unexpected error: %v", err)
|
t.Fatalf("unexpected error: %v", err)
|
||||||
}
|
}
|
||||||
@@ -79,15 +73,12 @@ func TestCIDRBlocking(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestIPv4MappedIPv6(t *testing.T) {
|
func TestIPv4MappedIPv6(t *testing.T) {
|
||||||
fw, err := New(config.Firewall{
|
fw, err := New("", []string{"192.0.2.1"}, nil, nil)
|
||||||
BlockedIPs: []string{"192.0.2.1"},
|
|
||||||
})
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("unexpected error: %v", err)
|
t.Fatalf("unexpected error: %v", err)
|
||||||
}
|
}
|
||||||
defer fw.Close()
|
defer fw.Close()
|
||||||
|
|
||||||
// IPv4-mapped IPv6 representation of 192.0.2.1.
|
|
||||||
addr := netip.MustParseAddr("::ffff:192.0.2.1")
|
addr := netip.MustParseAddr("::ffff:192.0.2.1")
|
||||||
if !fw.Blocked(addr) {
|
if !fw.Blocked(addr) {
|
||||||
t.Fatal("expected IPv4-mapped IPv6 address to be blocked")
|
t.Fatal("expected IPv4-mapped IPv6 address to be blocked")
|
||||||
@@ -95,28 +86,21 @@ func TestIPv4MappedIPv6(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestInvalidIP(t *testing.T) {
|
func TestInvalidIP(t *testing.T) {
|
||||||
_, err := New(config.Firewall{
|
_, err := New("", []string{"not-an-ip"}, nil, nil)
|
||||||
BlockedIPs: []string{"not-an-ip"},
|
|
||||||
})
|
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Fatal("expected error for invalid IP")
|
t.Fatal("expected error for invalid IP")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestInvalidCIDR(t *testing.T) {
|
func TestInvalidCIDR(t *testing.T) {
|
||||||
_, err := New(config.Firewall{
|
_, err := New("", nil, []string{"not-a-cidr"}, nil)
|
||||||
BlockedCIDRs: []string{"not-a-cidr"},
|
|
||||||
})
|
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Fatal("expected error for invalid CIDR")
|
t.Fatal("expected error for invalid CIDR")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestCombinedRules(t *testing.T) {
|
func TestCombinedRules(t *testing.T) {
|
||||||
fw, err := New(config.Firewall{
|
fw, err := New("", []string{"10.0.0.1"}, []string{"192.168.0.0/16"}, nil)
|
||||||
BlockedIPs: []string{"10.0.0.1"},
|
|
||||||
BlockedCIDRs: []string{"192.168.0.0/16"},
|
|
||||||
})
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("unexpected error: %v", err)
|
t.Fatalf("unexpected error: %v", err)
|
||||||
}
|
}
|
||||||
@@ -126,10 +110,10 @@ func TestCombinedRules(t *testing.T) {
|
|||||||
addr string
|
addr string
|
||||||
blocked bool
|
blocked bool
|
||||||
}{
|
}{
|
||||||
{"10.0.0.1", true}, // IP match
|
{"10.0.0.1", true},
|
||||||
{"10.0.0.2", false}, // no match
|
{"10.0.0.2", false},
|
||||||
{"192.168.1.1", true}, // CIDR match
|
{"192.168.1.1", true},
|
||||||
{"172.16.0.1", false}, // no match
|
{"172.16.0.1", false},
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
@@ -139,3 +123,30 @@ func TestCombinedRules(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestRuntimeMutation(t *testing.T) {
|
||||||
|
fw, err := New("", nil, nil, nil)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
defer fw.Close()
|
||||||
|
|
||||||
|
addr := netip.MustParseAddr("10.0.0.1")
|
||||||
|
if fw.Blocked(addr) {
|
||||||
|
t.Fatal("should not be blocked initially")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := fw.AddIP("10.0.0.1"); err != nil {
|
||||||
|
t.Fatalf("add IP: %v", err)
|
||||||
|
}
|
||||||
|
if !fw.Blocked(addr) {
|
||||||
|
t.Fatal("should be blocked after AddIP")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := fw.RemoveIP("10.0.0.1"); err != nil {
|
||||||
|
t.Fatalf("remove IP: %v", err)
|
||||||
|
}
|
||||||
|
if fw.Blocked(addr) {
|
||||||
|
t.Fatal("should not be blocked after RemoveIP")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ import (
|
|||||||
"log/slog"
|
"log/slog"
|
||||||
"net"
|
"net"
|
||||||
"os"
|
"os"
|
||||||
|
"strings"
|
||||||
|
|
||||||
"google.golang.org/grpc"
|
"google.golang.org/grpc"
|
||||||
"google.golang.org/grpc/codes"
|
"google.golang.org/grpc/codes"
|
||||||
@@ -17,6 +18,7 @@ import (
|
|||||||
|
|
||||||
pb "git.wntrmute.dev/kyle/mc-proxy/gen/mc-proxy/v1"
|
pb "git.wntrmute.dev/kyle/mc-proxy/gen/mc-proxy/v1"
|
||||||
"git.wntrmute.dev/kyle/mc-proxy/internal/config"
|
"git.wntrmute.dev/kyle/mc-proxy/internal/config"
|
||||||
|
"git.wntrmute.dev/kyle/mc-proxy/internal/db"
|
||||||
"git.wntrmute.dev/kyle/mc-proxy/internal/server"
|
"git.wntrmute.dev/kyle/mc-proxy/internal/server"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -24,11 +26,12 @@ import (
|
|||||||
type AdminServer struct {
|
type AdminServer struct {
|
||||||
pb.UnimplementedProxyAdminServer
|
pb.UnimplementedProxyAdminServer
|
||||||
srv *server.Server
|
srv *server.Server
|
||||||
|
store *db.Store
|
||||||
logger *slog.Logger
|
logger *slog.Logger
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a gRPC server with TLS and optional mTLS.
|
// New creates a gRPC server with TLS and optional mTLS.
|
||||||
func New(cfg config.GRPC, srv *server.Server, logger *slog.Logger) (*grpc.Server, net.Listener, error) {
|
func New(cfg config.GRPC, srv *server.Server, store *db.Store, logger *slog.Logger) (*grpc.Server, net.Listener, error) {
|
||||||
cert, err := tls.LoadX509KeyPair(cfg.TLSCert, cfg.TLSKey)
|
cert, err := tls.LoadX509KeyPair(cfg.TLSCert, cfg.TLSKey)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, nil, fmt.Errorf("loading TLS keypair: %w", err)
|
return nil, nil, fmt.Errorf("loading TLS keypair: %w", err)
|
||||||
@@ -39,7 +42,6 @@ func New(cfg config.GRPC, srv *server.Server, logger *slog.Logger) (*grpc.Server
|
|||||||
MinVersion: tls.VersionTLS13,
|
MinVersion: tls.VersionTLS13,
|
||||||
}
|
}
|
||||||
|
|
||||||
// mTLS: require and verify client certificates.
|
|
||||||
if cfg.ClientCA != "" {
|
if cfg.ClientCA != "" {
|
||||||
caCert, err := os.ReadFile(cfg.ClientCA)
|
caCert, err := os.ReadFile(cfg.ClientCA)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -58,6 +60,7 @@ func New(cfg config.GRPC, srv *server.Server, logger *slog.Logger) (*grpc.Server
|
|||||||
|
|
||||||
admin := &AdminServer{
|
admin := &AdminServer{
|
||||||
srv: srv,
|
srv: srv,
|
||||||
|
store: store,
|
||||||
logger: logger,
|
logger: logger,
|
||||||
}
|
}
|
||||||
pb.RegisterProxyAdminServer(grpcServer, admin)
|
pb.RegisterProxyAdminServer(grpcServer, admin)
|
||||||
@@ -90,7 +93,7 @@ func (a *AdminServer) ListRoutes(_ context.Context, req *pb.ListRoutesRequest) (
|
|||||||
return resp, nil
|
return resp, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// AddRoute adds a route to a listener's route table.
|
// AddRoute writes to the database first, then updates in-memory state.
|
||||||
func (a *AdminServer) AddRoute(_ context.Context, req *pb.AddRouteRequest) (*pb.AddRouteResponse, error) {
|
func (a *AdminServer) AddRoute(_ context.Context, req *pb.AddRouteRequest) (*pb.AddRouteResponse, error) {
|
||||||
if req.Route == nil {
|
if req.Route == nil {
|
||||||
return nil, status.Error(codes.InvalidArgument, "route is required")
|
return nil, status.Error(codes.InvalidArgument, "route is required")
|
||||||
@@ -104,15 +107,23 @@ func (a *AdminServer) AddRoute(_ context.Context, req *pb.AddRouteRequest) (*pb.
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ls.AddRoute(req.Route.Hostname, req.Route.Backend); err != nil {
|
hostname := strings.ToLower(req.Route.Hostname)
|
||||||
|
|
||||||
|
// Write-through: DB first, then memory.
|
||||||
|
if _, err := a.store.CreateRoute(ls.ID, hostname, req.Route.Backend); err != nil {
|
||||||
return nil, status.Errorf(codes.AlreadyExists, "%v", err)
|
return nil, status.Errorf(codes.AlreadyExists, "%v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
a.logger.Info("route added", "listener", ls.Addr, "hostname", req.Route.Hostname, "backend", req.Route.Backend)
|
if err := ls.AddRoute(hostname, req.Route.Backend); err != nil {
|
||||||
|
// DB succeeded but memory failed (should not happen since DB enforces uniqueness).
|
||||||
|
a.logger.Error("inconsistency: DB write succeeded but memory update failed", "error", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
a.logger.Info("route added", "listener", ls.Addr, "hostname", hostname, "backend", req.Route.Backend)
|
||||||
return &pb.AddRouteResponse{}, nil
|
return &pb.AddRouteResponse{}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// RemoveRoute removes a route from a listener's route table.
|
// RemoveRoute writes to the database first, then updates in-memory state.
|
||||||
func (a *AdminServer) RemoveRoute(_ context.Context, req *pb.RemoveRouteRequest) (*pb.RemoveRouteResponse, error) {
|
func (a *AdminServer) RemoveRoute(_ context.Context, req *pb.RemoveRouteRequest) (*pb.RemoveRouteResponse, error) {
|
||||||
if req.Hostname == "" {
|
if req.Hostname == "" {
|
||||||
return nil, status.Error(codes.InvalidArgument, "hostname is required")
|
return nil, status.Error(codes.InvalidArgument, "hostname is required")
|
||||||
@@ -123,11 +134,17 @@ func (a *AdminServer) RemoveRoute(_ context.Context, req *pb.RemoveRouteRequest)
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ls.RemoveRoute(req.Hostname); err != nil {
|
hostname := strings.ToLower(req.Hostname)
|
||||||
|
|
||||||
|
if err := a.store.DeleteRoute(ls.ID, hostname); err != nil {
|
||||||
return nil, status.Errorf(codes.NotFound, "%v", err)
|
return nil, status.Errorf(codes.NotFound, "%v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
a.logger.Info("route removed", "listener", ls.Addr, "hostname", req.Hostname)
|
if err := ls.RemoveRoute(hostname); err != nil {
|
||||||
|
a.logger.Error("inconsistency: DB delete succeeded but memory update failed", "error", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
a.logger.Info("route removed", "listener", ls.Addr, "hostname", hostname)
|
||||||
return &pb.RemoveRouteResponse{}, nil
|
return &pb.RemoveRouteResponse{}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -158,61 +175,74 @@ func (a *AdminServer) GetFirewallRules(_ context.Context, _ *pb.GetFirewallRules
|
|||||||
return &pb.GetFirewallRulesResponse{Rules: rules}, nil
|
return &pb.GetFirewallRulesResponse{Rules: rules}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// AddFirewallRule adds a firewall rule.
|
// AddFirewallRule writes to the database first, then updates in-memory state.
|
||||||
func (a *AdminServer) AddFirewallRule(_ context.Context, req *pb.AddFirewallRuleRequest) (*pb.AddFirewallRuleResponse, error) {
|
func (a *AdminServer) AddFirewallRule(_ context.Context, req *pb.AddFirewallRuleRequest) (*pb.AddFirewallRuleResponse, error) {
|
||||||
if req.Rule == nil {
|
if req.Rule == nil {
|
||||||
return nil, status.Error(codes.InvalidArgument, "rule is required")
|
return nil, status.Error(codes.InvalidArgument, "rule is required")
|
||||||
}
|
}
|
||||||
|
|
||||||
fw := a.srv.Firewall()
|
ruleType, err := protoRuleTypeToString(req.Rule.Type)
|
||||||
switch req.Rule.Type {
|
if err != nil {
|
||||||
case pb.FirewallRuleType_FIREWALL_RULE_TYPE_IP:
|
return nil, err
|
||||||
if err := fw.AddIP(req.Rule.Value); err != nil {
|
|
||||||
return nil, status.Errorf(codes.InvalidArgument, "%v", err)
|
|
||||||
}
|
|
||||||
case pb.FirewallRuleType_FIREWALL_RULE_TYPE_CIDR:
|
|
||||||
if err := fw.AddCIDR(req.Rule.Value); err != nil {
|
|
||||||
return nil, status.Errorf(codes.InvalidArgument, "%v", err)
|
|
||||||
}
|
|
||||||
case pb.FirewallRuleType_FIREWALL_RULE_TYPE_COUNTRY:
|
|
||||||
if req.Rule.Value == "" {
|
|
||||||
return nil, status.Error(codes.InvalidArgument, "country code is required")
|
|
||||||
}
|
|
||||||
fw.AddCountry(req.Rule.Value)
|
|
||||||
default:
|
|
||||||
return nil, status.Error(codes.InvalidArgument, "unknown rule type")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
a.logger.Info("firewall rule added", "type", req.Rule.Type, "value", req.Rule.Value)
|
if req.Rule.Value == "" {
|
||||||
|
return nil, status.Error(codes.InvalidArgument, "value is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write-through: DB first, then memory.
|
||||||
|
if _, err := a.store.CreateFirewallRule(ruleType, req.Rule.Value); err != nil {
|
||||||
|
return nil, status.Errorf(codes.AlreadyExists, "%v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fw := a.srv.Firewall()
|
||||||
|
switch ruleType {
|
||||||
|
case "ip":
|
||||||
|
if err := fw.AddIP(req.Rule.Value); err != nil {
|
||||||
|
a.logger.Error("inconsistency: DB write succeeded but memory update failed", "error", err)
|
||||||
|
}
|
||||||
|
case "cidr":
|
||||||
|
if err := fw.AddCIDR(req.Rule.Value); err != nil {
|
||||||
|
a.logger.Error("inconsistency: DB write succeeded but memory update failed", "error", err)
|
||||||
|
}
|
||||||
|
case "country":
|
||||||
|
fw.AddCountry(req.Rule.Value)
|
||||||
|
}
|
||||||
|
|
||||||
|
a.logger.Info("firewall rule added", "type", ruleType, "value", req.Rule.Value)
|
||||||
return &pb.AddFirewallRuleResponse{}, nil
|
return &pb.AddFirewallRuleResponse{}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// RemoveFirewallRule removes a firewall rule.
|
// RemoveFirewallRule writes to the database first, then updates in-memory state.
|
||||||
func (a *AdminServer) RemoveFirewallRule(_ context.Context, req *pb.RemoveFirewallRuleRequest) (*pb.RemoveFirewallRuleResponse, error) {
|
func (a *AdminServer) RemoveFirewallRule(_ context.Context, req *pb.RemoveFirewallRuleRequest) (*pb.RemoveFirewallRuleResponse, error) {
|
||||||
if req.Rule == nil {
|
if req.Rule == nil {
|
||||||
return nil, status.Error(codes.InvalidArgument, "rule is required")
|
return nil, status.Error(codes.InvalidArgument, "rule is required")
|
||||||
}
|
}
|
||||||
|
|
||||||
fw := a.srv.Firewall()
|
ruleType, err := protoRuleTypeToString(req.Rule.Type)
|
||||||
switch req.Rule.Type {
|
if err != nil {
|
||||||
case pb.FirewallRuleType_FIREWALL_RULE_TYPE_IP:
|
return nil, err
|
||||||
if err := fw.RemoveIP(req.Rule.Value); err != nil {
|
|
||||||
return nil, status.Errorf(codes.InvalidArgument, "%v", err)
|
|
||||||
}
|
|
||||||
case pb.FirewallRuleType_FIREWALL_RULE_TYPE_CIDR:
|
|
||||||
if err := fw.RemoveCIDR(req.Rule.Value); err != nil {
|
|
||||||
return nil, status.Errorf(codes.InvalidArgument, "%v", err)
|
|
||||||
}
|
|
||||||
case pb.FirewallRuleType_FIREWALL_RULE_TYPE_COUNTRY:
|
|
||||||
if req.Rule.Value == "" {
|
|
||||||
return nil, status.Error(codes.InvalidArgument, "country code is required")
|
|
||||||
}
|
|
||||||
fw.RemoveCountry(req.Rule.Value)
|
|
||||||
default:
|
|
||||||
return nil, status.Error(codes.InvalidArgument, "unknown rule type")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
a.logger.Info("firewall rule removed", "type", req.Rule.Type, "value", req.Rule.Value)
|
if err := a.store.DeleteFirewallRule(ruleType, req.Rule.Value); err != nil {
|
||||||
|
return nil, status.Errorf(codes.NotFound, "%v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fw := a.srv.Firewall()
|
||||||
|
switch ruleType {
|
||||||
|
case "ip":
|
||||||
|
if err := fw.RemoveIP(req.Rule.Value); err != nil {
|
||||||
|
a.logger.Error("inconsistency: DB delete succeeded but memory update failed", "error", err)
|
||||||
|
}
|
||||||
|
case "cidr":
|
||||||
|
if err := fw.RemoveCIDR(req.Rule.Value); err != nil {
|
||||||
|
a.logger.Error("inconsistency: DB delete succeeded but memory update failed", "error", err)
|
||||||
|
}
|
||||||
|
case "country":
|
||||||
|
fw.RemoveCountry(req.Rule.Value)
|
||||||
|
}
|
||||||
|
|
||||||
|
a.logger.Info("firewall rule removed", "type", ruleType, "value", req.Rule.Value)
|
||||||
return &pb.RemoveFirewallRuleResponse{}, nil
|
return &pb.RemoveFirewallRuleResponse{}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -244,3 +274,16 @@ func (a *AdminServer) findListener(addr string) (*server.ListenerState, error) {
|
|||||||
}
|
}
|
||||||
return nil, status.Errorf(codes.NotFound, "listener %q not found", addr)
|
return nil, status.Errorf(codes.NotFound, "listener %q not found", addr)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func protoRuleTypeToString(t pb.FirewallRuleType) (string, error) {
|
||||||
|
switch t {
|
||||||
|
case pb.FirewallRuleType_FIREWALL_RULE_TYPE_IP:
|
||||||
|
return "ip", nil
|
||||||
|
case pb.FirewallRuleType_FIREWALL_RULE_TYPE_CIDR:
|
||||||
|
return "cidr", nil
|
||||||
|
case pb.FirewallRuleType_FIREWALL_RULE_TYPE_COUNTRY:
|
||||||
|
return "country", nil
|
||||||
|
default:
|
||||||
|
return "", status.Error(codes.InvalidArgument, "unknown rule type")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ import (
|
|||||||
|
|
||||||
// ListenerState holds the mutable state for a single proxy listener.
|
// ListenerState holds the mutable state for a single proxy listener.
|
||||||
type ListenerState struct {
|
type ListenerState struct {
|
||||||
|
ID int64 // database primary key
|
||||||
Addr string
|
Addr string
|
||||||
routes map[string]string // lowercase hostname → backend addr
|
routes map[string]string // lowercase hostname → backend addr
|
||||||
mu sync.RWMutex
|
mu sync.RWMutex
|
||||||
@@ -75,6 +76,13 @@ func (ls *ListenerState) lookupRoute(hostname string) (string, bool) {
|
|||||||
return backend, ok
|
return backend, ok
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ListenerData holds the data needed to construct a ListenerState.
|
||||||
|
type ListenerData struct {
|
||||||
|
ID int64
|
||||||
|
Addr string
|
||||||
|
Routes map[string]string // lowercase hostname → backend
|
||||||
|
}
|
||||||
|
|
||||||
// Server is the mc-proxy server. It manages listeners, firewall evaluation,
|
// Server is the mc-proxy server. It manages listeners, firewall evaluation,
|
||||||
// SNI-based routing, and bidirectional proxying.
|
// SNI-based routing, and bidirectional proxying.
|
||||||
type Server struct {
|
type Server struct {
|
||||||
@@ -82,27 +90,19 @@ type Server struct {
|
|||||||
fw *firewall.Firewall
|
fw *firewall.Firewall
|
||||||
listeners []*ListenerState
|
listeners []*ListenerState
|
||||||
logger *slog.Logger
|
logger *slog.Logger
|
||||||
wg sync.WaitGroup // tracks active connections
|
wg sync.WaitGroup
|
||||||
startedAt time.Time
|
startedAt time.Time
|
||||||
version string
|
version string
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a Server from the given configuration.
|
// New creates a Server from pre-loaded data.
|
||||||
func New(cfg *config.Config, logger *slog.Logger, version string) (*Server, error) {
|
func New(cfg *config.Config, fw *firewall.Firewall, listenerData []ListenerData, logger *slog.Logger, version string) *Server {
|
||||||
fw, err := firewall.New(cfg.Firewall)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("initializing firewall: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var listeners []*ListenerState
|
var listeners []*ListenerState
|
||||||
for _, lcfg := range cfg.Listeners {
|
for _, ld := range listenerData {
|
||||||
routes := make(map[string]string, len(lcfg.Routes))
|
|
||||||
for _, r := range lcfg.Routes {
|
|
||||||
routes[strings.ToLower(r.Hostname)] = r.Backend
|
|
||||||
}
|
|
||||||
listeners = append(listeners, &ListenerState{
|
listeners = append(listeners, &ListenerState{
|
||||||
Addr: lcfg.Addr,
|
ID: ld.ID,
|
||||||
routes: routes,
|
Addr: ld.Addr,
|
||||||
|
routes: ld.Routes,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -112,7 +112,7 @@ func New(cfg *config.Config, logger *slog.Logger, version string) (*Server, erro
|
|||||||
listeners: listeners,
|
listeners: listeners,
|
||||||
logger: logger,
|
logger: logger,
|
||||||
version: version,
|
version: version,
|
||||||
}, nil
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Firewall returns the server's firewall for use by the gRPC admin API.
|
// Firewall returns the server's firewall for use by the gRPC admin API.
|
||||||
@@ -162,23 +162,19 @@ func (s *Server) Run(ctx context.Context) error {
|
|||||||
netListeners = append(netListeners, ln)
|
netListeners = append(netListeners, ln)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Start accept loops.
|
|
||||||
for i, ln := range netListeners {
|
for i, ln := range netListeners {
|
||||||
ln := ln
|
ln := ln
|
||||||
ls := s.listeners[i]
|
ls := s.listeners[i]
|
||||||
go s.serve(ctx, ln, ls)
|
go s.serve(ctx, ln, ls)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Block until shutdown signal.
|
|
||||||
<-ctx.Done()
|
<-ctx.Done()
|
||||||
s.logger.Info("shutting down")
|
s.logger.Info("shutting down")
|
||||||
|
|
||||||
// Stop accepting new connections.
|
|
||||||
for _, ln := range netListeners {
|
for _, ln := range netListeners {
|
||||||
ln.Close()
|
ln.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
// Wait for in-flight connections with a timeout.
|
|
||||||
done := make(chan struct{})
|
done := make(chan struct{})
|
||||||
go func() {
|
go func() {
|
||||||
s.wg.Wait()
|
s.wg.Wait()
|
||||||
|
|||||||
@@ -1,6 +1,13 @@
|
|||||||
# mc-proxy configuration
|
# mc-proxy configuration
|
||||||
|
|
||||||
|
# Database. Required. Listeners, routes, and firewall rules are persisted here.
|
||||||
|
# On first run, the database is seeded from the config below.
|
||||||
|
# On subsequent runs, the database is the source of truth.
|
||||||
|
[database]
|
||||||
|
path = "/srv/mc-proxy/mc-proxy.db"
|
||||||
|
|
||||||
# Listeners. Each listener binds a TCP port and has its own route table.
|
# Listeners. Each listener binds a TCP port and has its own route table.
|
||||||
|
# These are used to seed the database on first run only.
|
||||||
[[listeners]]
|
[[listeners]]
|
||||||
addr = ":443"
|
addr = ":443"
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user