From 99026c0e48f6e1ebba5eae43bb14b9ed06a17101 Mon Sep 17 00:00:00 2001
From: Juliusz Chroboczek <jch@irif.fr>
Date: Thu, 29 Apr 2021 22:02:36 +0200
Subject: [PATCH] Implement option -udp-range.

---
 galene.go      | 19 +++++++++++++++++++
 group/group.go |  6 ++++++
 2 files changed, 25 insertions(+)

diff --git a/galene.go b/galene.go
index 842c520..8a25d85 100644
--- a/galene.go
+++ b/galene.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"flag"
+	"fmt"
 	"log"
 	"os"
 	"os/signal"
@@ -20,6 +21,7 @@ import (
 
 func main() {
 	var cpuprofile, memprofile, mutexprofile, httpAddr, dataDir string
+	var udpRange string
 
 	flag.StringVar(&httpAddr, "http", ":8443", "web server `address`")
 	flag.StringVar(&webserver.StaticRoot, "static", "./static/",
@@ -40,6 +42,8 @@ func main() {
 		"store memory profile in `file`")
 	flag.StringVar(&mutexprofile, "mutexprofile", "",
 		"store mutex profile in `file`")
+	flag.StringVar(&udpRange, "udp-range", "",
+		"UDP port `range`")
 	flag.BoolVar(&group.UseMDNS, "mdns", false, "gather mDNS addresses")
 	flag.BoolVar(&ice.ICERelayOnly, "relay-only", false,
 		"require use of TURN relays for all media traffic")
@@ -47,6 +51,21 @@ func main() {
 		"built-in TURN server `address` (\"\" to disable)")
 	flag.Parse()
 
+	if udpRange != "" {
+		var min, max uint16
+		n, err := fmt.Sscanf(udpRange, "%v-%v", &min, &max)
+		if err != nil {
+			log.Printf("UDP range: %v", err)
+			os.Exit(1)
+		}
+		if n != 2 || min <= 0 || max <= 0 || min > max {
+			log.Printf("UDP range: bad range")
+			os.Exit(1)
+		}
+		group.UDPMin = min
+		group.UDPMax = max
+	}
+
 	if cpuprofile != "" {
 		f, err := os.Create(cpuprofile)
 		if err != nil {
diff --git a/group/group.go b/group/group.go
index 54e55ce..e6cbb74 100644
--- a/group/group.go
+++ b/group/group.go
@@ -18,6 +18,7 @@ import (
 
 var Directory string
 var UseMDNS bool
+var UDPMin, UDPMax uint16
 
 var ErrNotAuthorised = errors.New("not authorised")
 
@@ -247,6 +248,11 @@ func APIFromCodecs(codecs []webrtc.RTPCodecCapability) (*webrtc.API, error) {
 			continue
 		}
 	}
+
+	if UDPMin > 0 && UDPMax > 0 {
+		s.SetEphemeralUDPPortRange(UDPMin, UDPMax)
+	}
+
 	return webrtc.NewAPI(
 		webrtc.WithSettingEngine(s),
 		webrtc.WithMediaEngine(&m),