40
vendor/github.com/pion/turn/v2/internal/ipnet/util.go
generated
vendored
Normal file
40
vendor/github.com/pion/turn/v2/internal/ipnet/util.go
generated
vendored
Normal file
@@ -0,0 +1,40 @@
|
||||
// Package ipnet contains helper functions around net and IP
|
||||
package ipnet
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net"
|
||||
)
|
||||
|
||||
var errFailedToCastAddr = errors.New("failed to cast net.Addr to *net.UDPAddr or *net.TCPAddr")
|
||||
|
||||
// AddrIPPort extracts the IP and Port from a net.Addr
|
||||
func AddrIPPort(a net.Addr) (net.IP, int, error) {
|
||||
aUDP, ok := a.(*net.UDPAddr)
|
||||
if ok {
|
||||
return aUDP.IP, aUDP.Port, nil
|
||||
}
|
||||
|
||||
aTCP, ok := a.(*net.TCPAddr)
|
||||
if ok {
|
||||
return aTCP.IP, aTCP.Port, nil
|
||||
}
|
||||
|
||||
return nil, 0, errFailedToCastAddr
|
||||
}
|
||||
|
||||
// AddrEqual asserts that two net.Addrs are equal
|
||||
// Currently only supprots UDP but will be extended in the future to support others
|
||||
func AddrEqual(a, b net.Addr) bool {
|
||||
aUDP, ok := a.(*net.UDPAddr)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
|
||||
bUDP, ok := b.(*net.UDPAddr)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
|
||||
return aUDP.IP.Equal(bUDP.IP) && aUDP.Port == bUDP.Port
|
||||
}
|
||||
Reference in New Issue
Block a user