Add IP whitelist configuration, enhance message handling with envelope identification, and log remote addresses for improved connection control

app/config/config.go

@@ -16,6 +16,7 @@ import (
 	"go-simpler.org/env"
 	lol "lol.mleku.dev"
 	"lol.mleku.dev/chk"
+	"lol.mleku.dev/log"
 	"next.orly.dev/pkg/version"
 )
 
@@ -29,6 +30,7 @@ type C struct {
 	Port        int      `env:"ORLY_PORT" default:"3334" usage:"port to listen on"`
 	LogLevel    string   `env:"ORLY_LOG_LEVEL" default:"info" usage:"debug level: fatal error warn info debug trace"`
 	Pprof       string   `env:"ORLY_PPROF" usage:"enable pprof in modes: cpu,memory,allocation"`
+	IPWhitelist []string `env:"ORLY_IP_WHITELIST" usage:"comma-separated list of IP addresses to allow access from, matches on prefixes to allow private subnets, eg 10.0.0 = 10.0.0.0/8"`
 }
 
 // New creates and initializes a new configuration object for the relay
@@ -69,6 +71,7 @@ func New() (cfg *C, err error) {
 		os.Exit(0)
 	}
 	lol.SetLogLevel(cfg.LogLevel)
+	log.I.S(cfg.IPWhitelist)
 	return
 }
 

app/handle-message.go

@@ -1,9 +1,55 @@
 package app
 
 import (
+	"fmt"
+
+	"lol.mleku.dev/chk"
 	"lol.mleku.dev/log"
+	"next.orly.dev/pkg/encoders/envelopes"
+	"next.orly.dev/pkg/encoders/envelopes/authenvelope"
+	"next.orly.dev/pkg/encoders/envelopes/closeenvelope"
+	"next.orly.dev/pkg/encoders/envelopes/eventenvelope"
+	"next.orly.dev/pkg/encoders/envelopes/reqenvelope"
 )
 
-func (s *Server) HandleMessage(msg []byte) {
-	log.I.F("received message:\n%s\n", msg)
+func (s *Server) HandleMessage(msg []byte, remote string) {
+	log.D.C(
+		func() string {
+			return fmt.Sprintf(
+				"%s received message:\n%s", remote, msg,
+			)
+		},
+	)
+	var notice []byte
+	var err error
+	var t string
+	var rem []byte
+	if t, rem, err = envelopes.Identify(msg); chk.E(err) {
+		notice = []byte(err.Error())
+	}
+	switch t {
+	case eventenvelope.L:
+		log.D.F("eventenvelope: %s", rem)
+	case reqenvelope.L:
+		log.D.F("reqenvelope: %s", rem)
+	case closeenvelope.L:
+		log.D.F("closeenvelope: %s", rem)
+	case authenvelope.L:
+		log.D.F("authenvelope: %s", rem)
+	default:
+		notice = []byte(fmt.Sprintf("unknown envelope type %s\n%s", t, rem))
+	}
+	if len(notice) > 0 {
+		log.D.C(
+			func() string {
+				return fmt.Sprintf(
+					"notice->%s %s", remote, notice,
+				)
+			},
+		)
+		// if err = noticeenvelope.NewFrom(notice).Write(a.Listener); chk.E(err) {
+		// 	return
+		// }
+	}
+
 }

app/handle-websocket.go

@@ -28,6 +28,19 @@ const (
 
 func (s *Server) HandleWebsocket(w http.ResponseWriter, r *http.Request) {
 	remote := GetRemoteFromReq(r)
+	log.D.F("handling websocket connection from %s", remote)
+	if len(s.Config.IPWhitelist) > 0 {
+		for _, ip := range s.Config.IPWhitelist {
+			log.T.F("checking IP whitelist: %s", ip)
+			if strings.HasPrefix(remote, ip) {
+				log.T.F("IP whitelisted %s", remote)
+				goto whitelist
+			}
+		}
+		log.T.F("IP not whitelisted: %s", remote)
+		return
+	}
+whitelist:
 	var cancel context.CancelFunc
 	s.Ctx, cancel = context.WithCancel(s.Ctx)
 	defer cancel()
@@ -39,7 +52,6 @@ func (s *Server) HandleWebsocket(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 	defer conn.CloseNow()
-
 	go s.Pinger(s.Ctx, conn, time.NewTicker(time.Second*10), cancel)
 	for {
 		select {
@@ -73,7 +85,7 @@ func (s *Server) HandleWebsocket(w http.ResponseWriter, r *http.Request) {
 			}
 			continue
 		}
-		go s.HandleMessage(msg)
+		go s.HandleMessage(msg, remote)
 	}
 }