fix generating media URL with leading slash in control attribute (blu…

…environ/mediamtx#3517) (#625)
bluenviron · Sep 26, 2024 · 9a9e982 · 9a9e982
1 parent 6ed93de
commit 9a9e982
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/pkg/description/media.go b/pkg/description/media.go
@@ -268,7 +268,7 @@ func (m Media) URL(contentBase *base.URL) (*base.URL, error) {
 	// if there's a query, insert it after the query
 	// otherwise insert it after the path
 	strURL := contentBase.String()
-	if m.Control[0] != '?' && !strings.HasSuffix(strURL, "/") {
+	if m.Control[0] != '?' && m.Control[0] != '/' && !strings.HasSuffix(strURL, "/") {
 		strURL += "/"
 	}
 

diff --git a/pkg/description/media_test.go b/pkg/description/media_test.go
@@ -54,6 +54,16 @@ func TestMediaURL(t *testing.T) {
 			mustParseURL("rtsps://myuser:[email protected]:554/"),
 			mustParseURL("rtsps://myuser:[email protected]:554/path/trackID=7"),
 		},
+		{
+			"absolute control, base url no trailing slash",
+			[]byte("v=0\r\n" +
+				"s= \r\n" +
+				"m=video 0 RTP/AVP 96\r\n" +
+				"a=rtpmap:96 H264/90000\r\n" +
+				"a=control:/test"),
+			mustParseURL("rtsp://myuser:[email protected]:554"),
+			mustParseURL("rtsp://myuser:[email protected]:554/test"),
+		},
 		{
 			"relative control",
 			[]byte("v=0\r\n" +