netbird/sharedsock/example/main.go
Yury Gargay 9e8725618e
Extend linter rules (#1300)
- dupword checks for duplicate words in the source code
- durationcheck checks for two durations multiplied together
- forbidigo forbids identifiers
- mirror reports wrong mirror patterns of bytes/strings usage
- misspell finds commonly misspelled English words in comments
- predeclared finds code that shadows one of Go's predeclared identifiers
- thelper detects Go test helpers without t.Helper() call and checks the consistency of test helpers
2023-11-10 16:33:13 +01:00

58 lines
1.1 KiB
Go

package main
import (
"context"
"os"
"os/signal"
"github.com/netbirdio/netbird/sharedsock"
log "github.com/sirupsen/logrus"
)
func main() {
port := 51820
rawSock, err := sharedsock.Listen(port, sharedsock.NewIncomingSTUNFilter())
if err != nil {
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()
}