From c75da54795729fb2c18eafb3c49fc8a5dc872b21 Mon Sep 17 00:00:00 2001
From: cnderrauber <zengjie9004@gmail.com>
Date: Wed, 23 Nov 2022 10:15:46 +0800
Subject: [PATCH] Add EnableLoopbackCandidate flag

Add EnableLoopbackCandidate flag
---
 icegatherer.go   |  1 +
 settingengine.go | 27 +++++++++++++++++----------
 2 files changed, 18 insertions(+), 10 deletions(-)

diff --git a/icegatherer.go b/icegatherer.go
index ab7e76db98..a38f08b747 100644
--- 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
index f55bf821eb..0c1e0c3a6f 100644
--- 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.