From 0c5340da272f183f58bc4b566c3a1587fe0b8569 Mon Sep 17 00:00:00 2001
From: langhuihui <178529795@qq.com>
Date: Sat, 14 Nov 2020 11:53:35 +0800
Subject: [PATCH] =?UTF-8?q?=E5=85=BC=E5=AE=B9=E6=9C=AC=E5=9C=B0=E8=AE=BF?=
 =?UTF-8?q?=E9=97=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index ddf16a9..a12cf86 100644
--- a/main.go
+++ b/main.go
@@ -173,7 +173,9 @@ func (rtc *WebRTC) Publish(streamPath string) bool {
 		DefaultPayloadTypeH264,
 		new(avformat.H264)))
 	//m.RegisterCodec(NewRTPPCMUCodec(DefaultPayloadTypePCMU, 8000))
-	rtc.s.SetNAT1To1IPs(config.PublicIP, ICECandidateTypeHost)
+	if !strings.HasPrefix(rtc.RemoteAddr, "127.0.0.1") && !strings.HasPrefix(rtc.RemoteAddr, "[::1]") {
+		rtc.s.SetNAT1To1IPs(config.PublicIP, ICECandidateTypeHost)
+	}
 	rtc.api = NewAPI(WithMediaEngine(rtc.m), WithSettingEngine(rtc.s))
 	peerConnection, err := rtc.api.NewPeerConnection(Configuration{
 		ICEServers: []ICEServer{
@@ -300,7 +302,9 @@ func run() {
 			DefaultPayloadTypeH264,
 			&rtc.payloader))
 		rtc.m.RegisterCodec(NewRTPPCMACodec(DefaultPayloadTypePCMA, 8000))
-		rtc.s.SetNAT1To1IPs(config.PublicIP, ICECandidateTypeHost)
+		if !strings.HasPrefix(r.RemoteAddr, "127.0.0.1") && !strings.HasPrefix(r.RemoteAddr, "[::1]") {
+			rtc.s.SetNAT1To1IPs(config.PublicIP, ICECandidateTypeHost)
+		}
 		rtc.api = NewAPI(WithMediaEngine(rtc.m), WithSettingEngine(rtc.s))
 		peerConnection, err := rtc.api.NewPeerConnection(Configuration{
 			// ICEServers: []ICEServer{