extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2025-07-01 15:12:05 +02:00
Code Issues Packages Projects Releases Wiki Activity

Check for stun packet with a fixed size

Browse Source
This commit is contained in:
braginini
2022-09-06 21:07:21 +02:00
parent 73b5f8d63b
commit f49c299d77
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
iface/bind.go
View File

@ -114,7 +114,7 @@ func (b *ICEBind) makeReceiveIPv4(c net.PacketConn, bindMux BindMux) conn.Receiv
if err != nil { if err != nil {
return 0, nil, err return 0, nil, err
} }
if !stun.IsMessage(buff[:n]) { if !stun.IsMessage(buff[:20]) {
// WireGuard traffic // WireGuard traffic
return n, (*conn.StdNetEndpoint)(&net.UDPAddr{ return n, (*conn.StdNetEndpoint)(&net.UDPAddr{
IP: e.Addr().AsSlice(), IP: e.Addr().AsSlice(),

2
iface/udp_mux.go
View File

@ -85,7 +85,7 @@ func (m *UDPMuxDefault) HandlePacket(p []byte, n int, addr net.Addr) error {
m.addressMapMu.Unlock() m.addressMapMu.Unlock()
// If we haven't seen this address before but is a STUN packet lookup by ufrag // If we haven't seen this address before but is a STUN packet lookup by ufrag
if destinationConn == nil && stun.IsMessage(p[:n]) { if destinationConn == nil && stun.IsMessage(p[:20]) {
msg := &stun.Message{ msg := &stun.Message{
Raw: append([]byte{}, p[:n]...), Raw: append([]byte{}, p[:n]...),
} }

2
iface/udp_mux_universal.go
View File

@ -80,7 +80,7 @@ func (m *UniversalUDPMuxDefault) Type() string {
} }
func (m *UniversalUDPMuxDefault) HandlePacket(p []byte, n int, addr net.Addr) error { func (m *UniversalUDPMuxDefault) HandlePacket(p []byte, n int, addr net.Addr) error {
if stun.IsMessage(p[:n]) { if stun.IsMessage(p[:20]) {
msg := &stun.Message{ msg := &stun.Message{
Raw: append([]byte{}, p[:n]...), Raw: append([]byte{}, p[:n]...),
} }