webrtc/rtcicecandidate.go

package webrtc

import (
	"errors"
	"fmt"
	"net"

	"github.com/pions/sdp"
	"github.com/pions/webrtc/pkg/ice"
)

// RTCIceCandidate represents a ice candidate
type RTCIceCandidate struct {
	Foundation     string              `json:"foundation"`
	Priority       uint32              `json:"priority"`
	IP             string              `json:"ip"`
	Protocol       RTCIceProtocol      `json:"protocol"`
	Port           uint16              `json:"port"`
	Typ            RTCIceCandidateType `json:"type"`
	Component      uint16              `json:"component"`
	RelatedAddress string              `json:"relatedAddress"`
	RelatedPort    uint16              `json:"relatedPort"`
}

// Conversion for package sdp

func newRTCIceCandidateFromSDP(c sdp.ICECandidate) (RTCIceCandidate, error) {
	typ, err := newRTCIceCandidateType(c.Typ)
	if err != nil {
		return RTCIceCandidate{}, err
	}
	protocol, err := newRTCIceProtocol(c.Protocol)
	if err != nil {
		return RTCIceCandidate{}, err
	}
	return RTCIceCandidate{
		Foundation:     c.Foundation,
		Priority:       c.Priority,
		IP:             c.IP,
		Protocol:       protocol,
		Port:           c.Port,
		Component:      c.Component,
		Typ:            typ,
		RelatedAddress: c.RelatedAddress,
		RelatedPort:    c.RelatedPort,
	}, nil
}

func (c RTCIceCandidate) toSDP() sdp.ICECandidate {
	return sdp.ICECandidate{
		Foundation:     c.Foundation,
		Priority:       c.Priority,
		IP:             c.IP,
		Protocol:       c.Protocol.String(),
		Port:           c.Port,
		Component:      c.Component,
		Typ:            c.Typ.String(),
		RelatedAddress: c.RelatedAddress,
		RelatedPort:    c.RelatedPort,
	}
}

// Conversion for package ice

func newRTCIceCandidatesFromICE(iceCandidates []*ice.Candidate) ([]RTCIceCandidate, error) {
	candidates := []RTCIceCandidate{}

	for _, i := range iceCandidates {
		c, err := newRTCIceCandidateFromICE(i)
		if err != nil {
			return nil, err
		}
		candidates = append(candidates, c)
	}

	return candidates, nil
}

func newRTCIceCandidateFromICE(i *ice.Candidate) (RTCIceCandidate, error) {
	typ, err := convertTypeFromICE(i.Type)
	if err != nil {
		return RTCIceCandidate{}, err
	}
	protocol, err := newRTCIceProtocol(i.NetworkType.NetworkShort())
	if err != nil {
		return RTCIceCandidate{}, err
	}

	c := RTCIceCandidate{
		Foundation: "foundation",
		Priority:   uint32(i.Priority()),
		IP:         i.IP.String(),
		Protocol:   protocol,
		Port:       uint16(i.Port),
		Component:  i.Component,
		Typ:        typ,
	}

	if i.RelatedAddress != nil {
		c.RelatedAddress = i.RelatedAddress.Address
		c.RelatedPort = uint16(i.RelatedAddress.Port)
	}

	return c, nil
}

func (c RTCIceCandidate) toICE() (*ice.Candidate, error) {
	ip := net.ParseIP(c.IP)
	if ip == nil {
		return nil, errors.New("Failed to parse IP address")
	}

	switch c.Typ {
	case RTCIceCandidateTypeHost:
		return ice.NewCandidateHost(c.Protocol.String(), ip, int(c.Port), c.Component)
	case RTCIceCandidateTypeSrflx:
		return ice.NewCandidateServerReflexive(c.Protocol.String(), ip, int(c.Port), c.Component,
			c.RelatedAddress, int(c.RelatedPort))
	case RTCIceCandidateTypePrflx:
		return ice.NewCandidatePeerReflexive(c.Protocol.String(), ip, int(c.Port), c.Component,
			c.RelatedAddress, int(c.RelatedPort))
	case RTCIceCandidateTypeRelay:
		return ice.NewCandidateRelay(c.Protocol.String(), ip, int(c.Port), c.Component,
			c.RelatedAddress, int(c.RelatedPort))
	default:
		return nil, fmt.Errorf("Unknown candidate type: %s", c.Typ)
	}
}

func convertTypeFromICE(t ice.CandidateType) (RTCIceCandidateType, error) {
	switch t {
	case ice.CandidateTypeHost:
		return RTCIceCandidateTypeHost, nil
	case ice.CandidateTypeServerReflexive:
		return RTCIceCandidateTypeSrflx, nil
	case ice.CandidateTypePeerReflexive:
		return RTCIceCandidateTypePrflx, nil
	case ice.CandidateTypeRelay:
		return RTCIceCandidateTypeRelay, nil
	default:
		return RTCIceCandidateType(t), fmt.Errorf("Unknown ICE candidate type: %s", t)
	}
}