99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package ice
|
|
|
|
import (
|
|
"net"
|
|
)
|
|
|
|
// CandidateRelay ...
|
|
type CandidateRelay struct {
|
|
candidateBase
|
|
|
|
relayProtocol string
|
|
onClose func() error
|
|
}
|
|
|
|
// CandidateRelayConfig is the config required to create a new CandidateRelay
|
|
type CandidateRelayConfig struct {
|
|
CandidateID string
|
|
Network string
|
|
Address string
|
|
Port int
|
|
Component uint16
|
|
Priority uint32
|
|
Foundation string
|
|
RelAddr string
|
|
RelPort int
|
|
RelayProtocol string
|
|
OnClose func() error
|
|
}
|
|
|
|
// NewCandidateRelay creates a new relay candidate
|
|
func NewCandidateRelay(config *CandidateRelayConfig) (*CandidateRelay, error) {
|
|
candidateID := config.CandidateID
|
|
|
|
if candidateID == "" {
|
|
candidateID = globalCandidateIDGenerator.Generate()
|
|
}
|
|
|
|
ip := net.ParseIP(config.Address)
|
|
if ip == nil {
|
|
return nil, ErrAddressParseFailed
|
|
}
|
|
|
|
networkType, err := determineNetworkType(config.Network, ip)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &CandidateRelay{
|
|
candidateBase: candidateBase{
|
|
id: candidateID,
|
|
networkType: networkType,
|
|
candidateType: CandidateTypeRelay,
|
|
address: config.Address,
|
|
port: config.Port,
|
|
resolvedAddr: &net.UDPAddr{IP: ip, Port: config.Port},
|
|
component: config.Component,
|
|
foundationOverride: config.Foundation,
|
|
priorityOverride: config.Priority,
|
|
relatedAddress: &CandidateRelatedAddress{
|
|
Address: config.RelAddr,
|
|
Port: config.RelPort,
|
|
},
|
|
remoteCandidateCaches: map[AddrPort]Candidate{},
|
|
},
|
|
relayProtocol: config.RelayProtocol,
|
|
onClose: config.OnClose,
|
|
}, nil
|
|
}
|
|
|
|
// RelayProtocol returns the protocol used between the endpoint and the relay server.
|
|
func (c *CandidateRelay) RelayProtocol() string {
|
|
return c.relayProtocol
|
|
}
|
|
|
|
func (c *CandidateRelay) close() error {
|
|
err := c.candidateBase.close()
|
|
if c.onClose != nil {
|
|
err = c.onClose()
|
|
c.onClose = nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (c *CandidateRelay) copy() (Candidate, error) {
|
|
cc, err := c.candidateBase.copy()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if ccr, ok := cc.(*CandidateRelay); ok {
|
|
ccr.relayProtocol = c.relayProtocol
|
|
}
|
|
|
|
return cc, nil
|
|
}
|