Add EnableLoopbackCandidate flag

pion · Nov 23, 2022 · c75da54 · c75da54
1 parent 5faad1e
commit c75da54
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 10 deletions.
diff --git a/icegatherer.go b/icegatherer.go
@@ -108,6 +108,7 @@ func (g *ICEGatherer) createAgent() error {
 		IPFilter:               g.api.settingEngine.candidates.IPFilter,
 		NAT1To1IPs:             g.api.settingEngine.candidates.NAT1To1IPs,
 		NAT1To1IPCandidateType: nat1To1CandiTyp,
+		IncludeLoopback:        g.api.settingEngine.candidates.IncludeLoopbackCandidate,
 		Net:                    g.api.settingEngine.vnet,
 		MulticastDNSMode:       mDNSMode,
 		MulticastDNSHostName:   g.api.settingEngine.candidates.MulticastDNSHostName,

diff --git a/settingengine.go b/settingengine.go
@@ -37,16 +37,17 @@ type SettingEngine struct {
 		ICERelayAcceptanceMinWait *time.Duration
 	}
 	candidates struct {
-		ICELite                bool
-		ICENetworkTypes        []NetworkType
-		InterfaceFilter        func(string) bool
-		IPFilter               func(net.IP) bool
-		NAT1To1IPs             []string
-		NAT1To1IPCandidateType ICECandidateType
-		MulticastDNSMode       ice.MulticastDNSMode
-		MulticastDNSHostName   string
-		UsernameFragment       string
-		Password               string
+		ICELite                  bool
+		ICENetworkTypes          []NetworkType
+		InterfaceFilter          func(string) bool
+		IPFilter                 func(net.IP) bool
+		NAT1To1IPs               []string
+		NAT1To1IPCandidateType   ICECandidateType
+		MulticastDNSMode         ice.MulticastDNSMode
+		MulticastDNSHostName     string
+		UsernameFragment         string
+		Password                 string
+		IncludeLoopbackCandidate bool
 	}
 	replayProtection struct {
 		DTLS  *uint
@@ -195,6 +196,12 @@ func (e *SettingEngine) SetNAT1To1IPs(ips []string, candidateType ICECandidateTy
 	e.candidates.NAT1To1IPCandidateType = candidateType
 }
 
+// SetIncludeLoopbackCandidate enable pion to gather loopback candidates, it is useful
+// for some VM have public IP mapped to loopback interface
+func (e *SettingEngine) SetIncludeLoopbackCandidate(include bool) {
+	e.candidates.IncludeLoopbackCandidate = include
+}
+
 // SetAnsweringDTLSRole sets the DTLS role that is selected when offering
 // The DTLS role controls if the WebRTC Client as a client or server. This
 // may be useful when interacting with non-compliant clients or debugging issues.