add additionalHosts

aler9 · Jun 10, 2024 · 3bbc7c1 · 3bbc7c1
1 parent 952fbe0
commit 3bbc7c1
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/icegatherer.go b/icegatherer.go
@@ -112,6 +112,7 @@ func (g *ICEGatherer) createAgent() error {
 		IPFilter:               g.api.settingEngine.candidates.IPFilter,
 		NAT1To1IPs:             g.api.settingEngine.candidates.NAT1To1IPs,
 		NAT1To1IPCandidateType: nat1To1CandiTyp,
+		AdditionalHosts:        g.api.settingEngine.candidates.AdditionalHosts,
 		IncludeLoopback:        g.api.settingEngine.candidates.IncludeLoopbackCandidate,
 		Net:                    g.api.settingEngine.net,
 		MulticastDNSMode:       mDNSMode,
@@ -120,6 +121,7 @@ func (g *ICEGatherer) createAgent() error {
 		LocalPwd:               g.api.settingEngine.candidates.Password,
 		TCPMux:                 g.api.settingEngine.iceTCPMux,
 		UDPMux:                 g.api.settingEngine.iceUDPMux,
+		UDPRandom:              g.api.settingEngine.iceUDPRandom,
 		ProxyDialer:            g.api.settingEngine.iceProxyDialer,
 		DisableActiveTCP:       g.api.settingEngine.iceDisableActiveTCP,
 		BindingRequestHandler:  g.api.settingEngine.iceBindingRequestHandler,

diff --git a/settingengine.go b/settingengine.go
@@ -51,6 +51,7 @@ type SettingEngine struct {
 		IPFilter                 func(net.IP) bool
 		NAT1To1IPs               []string
 		NAT1To1IPCandidateType   ICECandidateType
+		AdditionalHosts          []string
 		MulticastDNSMode         ice.MulticastDNSMode
 		MulticastDNSHostName     string
 		UsernameFragment         string
@@ -88,6 +89,7 @@ type SettingEngine struct {
 	LoggerFactory                             logging.LoggerFactory
 	iceTCPMux                                 ice.TCPMux
 	iceUDPMux                                 ice.UDPMux
+	iceUDPRandom                              bool
 	iceProxyDialer                            proxy.Dialer
 	iceDisableActiveTCP                       bool
 	iceBindingRequestHandler                  func(m *stun.Message, local, remote ice.Candidate, pair *ice.CandidatePair) bool
@@ -230,6 +232,10 @@ func (e *SettingEngine) SetNAT1To1IPs(ips []string, candidateType ICECandidateTy
 	e.candidates.NAT1To1IPCandidateType = candidateType
 }
 
+func (e *SettingEngine) SetAdditionalHosts(hosts []string) {
+	e.candidates.AdditionalHosts = hosts
+}
+
 // 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) {
@@ -348,6 +354,10 @@ func (e *SettingEngine) SetICEUDPMux(udpMux ice.UDPMux) {
 	e.iceUDPMux = udpMux
 }
 
+func (e *SettingEngine) SetICEUDPRandom(v bool) {
+	e.iceUDPRandom = v
+}
+
 // SetICEProxyDialer sets the proxy dialer interface based on golang.org/x/net/proxy.
 func (e *SettingEngine) SetICEProxyDialer(d proxy.Dialer) {
 	e.iceProxyDialer = d