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

This commit is contained in:
2025-08-31 10:00:50 +01:00
parent 576475e3dc
commit 94383f29e9
3 changed files with 71 additions and 10 deletions

View File

@@ -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
// }
}
}