From d46efaae3cdc62e6c56dcf3e9b3e21138d7eba34 Mon Sep 17 00:00:00 2001
From: Alexandre Iooss <erdnaxe@crans.org>
Date: Tue, 29 Sep 2020 14:11:56 +0200
Subject: [PATCH] Limit WebRTC UDP ports

---
 stream/stream.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/stream/stream.go b/stream/stream.go
index df9866e..6a4c204 100644
--- a/stream/stream.go
+++ b/stream/stream.go
@@ -45,7 +45,15 @@ func newPeerHandler(remoteSdp webrtc.SessionDescription) webrtc.SessionDescripti
 	}
 
 	// Create a new PeerConnection
-	api := webrtc.NewAPI(webrtc.WithMediaEngine(mediaEngine))
+	settingsEngine := webrtc.SettingEngine{}
+	if err := settingsEngine.SetEphemeralUDPPortRange(10000, 11000); err != nil {
+		log.Println("Failed to set min/max UDP ports", err)
+		return webrtc.SessionDescription{}
+	}
+	api := webrtc.NewAPI(
+		webrtc.WithMediaEngine(mediaEngine),
+		webrtc.WithSettingEngine(settingsEngine),
+	)
 	peerConnection, err := api.NewPeerConnection(webrtc.Configuration{
 		ICEServers: []webrtc.ICEServer{
 			{