mirror of
https://github.com/pion/webrtc.git
synced 2025-12-24 11:51:03 +08:00
128 lines
2.9 KiB
Go
128 lines
2.9 KiB
Go
// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package webrtc
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/pion/ice/v4"
|
|
)
|
|
|
|
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 (
|
|
// NetworkTypeUnknown is the enum's zero-value.
|
|
NetworkTypeUnknown NetworkType = iota
|
|
|
|
// NetworkTypeUDP4 indicates UDP over IPv4.
|
|
NetworkTypeUDP4
|
|
|
|
// 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 { //nolint:staticcheck
|
|
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 NetworkTypeUnknown, 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 NetworkTypeUnknown, fmt.Errorf("%w: %s", errNetworkTypeUnknown, iceNetworkType.String())
|
|
}
|
|
}
|
|
|
|
func toICENetworkTypes(networkTypes []NetworkType) []ice.NetworkType {
|
|
if len(networkTypes) == 0 {
|
|
return nil
|
|
}
|
|
|
|
converted := make([]ice.NetworkType, 0, len(networkTypes))
|
|
for _, networkType := range networkTypes {
|
|
converted = append(converted, networkType.toICE())
|
|
}
|
|
|
|
return converted
|
|
}
|
|
|
|
func (networkType NetworkType) toICE() ice.NetworkType {
|
|
return ice.NetworkType(networkType)
|
|
}
|