1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
|
package webrtc
import (
"fmt"
"github.com/pion/ice/v2"
)
func supportedNetworkTypes() []NetworkType {
return []NetworkType{
NetworkTypeUDP4,
NetworkTypeUDP6,
// NetworkTypeTCP4, // Not supported yet
// NetworkTypeTCP6, // Not supported yet
}
}
// NetworkType represents the type of network
type NetworkType int
const (
// NetworkTypeUDP4 indicates UDP over IPv4.
NetworkTypeUDP4 NetworkType = iota + 1
// NetworkTypeUDP6 indicates UDP over IPv6.
NetworkTypeUDP6
// NetworkTypeTCP4 indicates TCP over IPv4.
NetworkTypeTCP4
// NetworkTypeTCP6 indicates TCP over IPv6.
NetworkTypeTCP6
)
// This is done this way because of a linter.
const (
networkTypeUDP4Str = "udp4"
networkTypeUDP6Str = "udp6"
networkTypeTCP4Str = "tcp4"
networkTypeTCP6Str = "tcp6"
)
func (t NetworkType) String() string {
switch t {
case NetworkTypeUDP4:
return networkTypeUDP4Str
case NetworkTypeUDP6:
return networkTypeUDP6Str
case NetworkTypeTCP4:
return networkTypeTCP4Str
case NetworkTypeTCP6:
return networkTypeTCP6Str
default:
return ErrUnknownType.Error()
}
}
// Protocol returns udp or tcp
func (t NetworkType) Protocol() string {
switch t {
case NetworkTypeUDP4:
return "udp"
case NetworkTypeUDP6:
return "udp"
case NetworkTypeTCP4:
return "tcp"
case NetworkTypeTCP6:
return "tcp"
default:
return ErrUnknownType.Error()
}
}
// NewNetworkType allows create network type from string
// It will be useful for getting custom network types from external config.
func NewNetworkType(raw string) (NetworkType, error) {
switch raw {
case networkTypeUDP4Str:
return NetworkTypeUDP4, nil
case networkTypeUDP6Str:
return NetworkTypeUDP6, nil
case networkTypeTCP4Str:
return NetworkTypeTCP4, nil
case networkTypeTCP6Str:
return NetworkTypeTCP6, nil
default:
return NetworkType(Unknown), fmt.Errorf("%w: %s", errNetworkTypeUnknown, raw)
}
}
func getNetworkType(iceNetworkType ice.NetworkType) (NetworkType, error) {
switch iceNetworkType {
case ice.NetworkTypeUDP4:
return NetworkTypeUDP4, nil
case ice.NetworkTypeUDP6:
return NetworkTypeUDP6, nil
case ice.NetworkTypeTCP4:
return NetworkTypeTCP4, nil
case ice.NetworkTypeTCP6:
return NetworkTypeTCP6, nil
default:
return NetworkType(Unknown), fmt.Errorf("%w: %s", errNetworkTypeUnknown, iceNetworkType.String())
}
}
|