WIP: backend udp

cmd/netbounce/main.go

@@ -17,23 +17,30 @@ package main
 */
 
 import (
+	"context"
+	"os/signal"
+
 	"gitea.suyono.dev/suyono/netbounce"
 	"gitea.suyono.dev/suyono/netbounce/config"
 	"gitea.suyono.dev/suyono/netbounce/flag"
 	"github.com/rs/zerolog"
 	"github.com/rs/zerolog/log"
+	"golang.org/x/sys/unix"
 )
 
 func main() {
 	zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
 
+	ctx, stop := signal.NotifyContext(context.Background(), unix.SIGINT, unix.SIGTERM)
+	defer stop()
+
 	flag.Parse()
 	config.CollectEnv()
 	if err := config.ReadConfig(); err != nil {
 		log.Fatal().Err(err).Msg("error reading config")
 	}
 
-	if err := netbounce.Bounce(); err != nil {
+	if err := netbounce.Bounce(ctx); err != nil {
 		log.Fatal().Err(err).Msg("starting netbounce")
 	}
 }