mirror of
https://github.com/netbirdio/netbird.git
synced 2026-08-24 16:41:30 +02:00
e4b8bf39d2
* Fix staticcheck findings reported by the updated golangci-lint * Skip the receive error log when the local context is done
60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"os/signal"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/netbirdio/netbird/client/iface"
|
|
"github.com/netbirdio/netbird/sharedsock"
|
|
)
|
|
|
|
func main() {
|
|
|
|
port := 51820
|
|
rawSock, err := sharedsock.Listen(port, sharedsock.NewIncomingSTUNFilter(), iface.DefaultMTU) //nolint:staticcheck
|
|
if err != nil { //nolint:staticcheck // always errors on non-Linux builds
|
|
panic(err)
|
|
}
|
|
|
|
log.Infof("attached to the raw socket on port %d", port)
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
// read packets
|
|
go func() {
|
|
buf := make([]byte, 1500)
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
log.Debugf("stopped reading from the shared socket")
|
|
return
|
|
default:
|
|
size, addr, err := rawSock.ReadFrom(buf)
|
|
if err != nil {
|
|
log.Errorf("error while reading packet from the shared socket: %s", err)
|
|
continue
|
|
}
|
|
log.Infof("read a STUN packet of size %d from %s", size, addr.String())
|
|
}
|
|
}
|
|
}()
|
|
|
|
// terminate the program on ^C
|
|
c := make(chan os.Signal, 1)
|
|
signal.Notify(c, os.Interrupt)
|
|
go func() {
|
|
for range c {
|
|
log.Infof("received ^C signal, stopping the program")
|
|
cancel()
|
|
err = rawSock.Close()
|
|
if err != nil {
|
|
log.Errorf("failed closing raw socket")
|
|
}
|
|
}
|
|
}()
|
|
|
|
<-ctx.Done()
|
|
}
|