srt: support standard streamID syntax (#2469)

internal/servers/srt/conn.go

@@ -6,7 +6,6 @@
 	"errors"
 	"fmt"
 	"net"
-	"strings"
 	"sync"
 	"time"
 
@@ -147,50 +146,30 @@
 }
 
 func (c *conn) runInner2(req srtNewConnReq) (bool, error) {
-	parts := strings.Split(req.connReq.StreamId(), ":")
-	if (len(parts) < 2 || len(parts) > 5) || (parts[0] != "read" && parts[0] != "publish") {
-		return false, fmt.Errorf("invalid streamid '%s':"+
-			" it must be 'action:pathname[:query]' or 'action:pathname:user:pass[:query]', "+
-			"where action is either read or publish, pathname is the path name, user and pass are the credentials, "+
-			"query is an optional token containing additional information",
-			req.connReq.StreamId())
-	}
-
-	pathName := parts[1]
-	user := ""
-	pass := ""
-	query := ""
-
-	if len(parts) == 4 || len(parts) == 5 {
-		user, pass = parts[2], parts[3]
-	}
-
-	if len(parts) == 3 {
-		query = parts[2]
-	}
-
-	if len(parts) == 5 {
-		query = parts[4]
+	var streamID streamID
+	err := streamID.unmarshal(req.connReq.StreamId())
+	if err != nil {
+		return false, fmt.Errorf("invalid stream ID '%s': %w", req.connReq.StreamId(), err)
 	}
 
-	if parts[0] == "publish" {
-		return c.runPublish(req, pathName, user, pass, query)
+	if streamID.mode == streamIDModePublish {
+		return c.runPublish(req, &streamID)
 	}
-	return c.runRead(req, pathName, user, pass, query)
+	return c.runRead(req, &streamID)
 }
 
-func (c *conn) runPublish(req srtNewConnReq, pathName string, user string, pass string, query string) (bool, error) {
+func (c *conn) runPublish(req srtNewConnReq, streamID *streamID) (bool, error) {
 	res := c.pathManager.AddPublisher(defs.PathAddPublisherReq{
 		Author: c,
 		AccessRequest: defs.PathAccessRequest{
-			Name:    pathName,
+			Name:    streamID.path,
 			IP:      c.ip(),
 			Publish: true,
-			User:    user,
-			Pass:    pass,
+			User:    streamID.user,
+			Pass:    streamID.pass,
 			Proto:   defs.AuthProtocolSRT,
 			ID:      &c.uuid,
-			Query:   query,
+			Query:   streamID.query,
 		},
 	})
 
@@ -219,8 +198,8 @@
 
 	c.mutex.Lock()
 	c.state = connStatePublish
-	c.pathName = pathName
-	c.query = query
+	c.pathName = streamID.path
+	c.query = streamID.query
 	c.sconn = sconn
 	c.mutex.Unlock()
 
@@ -280,17 +259,17 @@
 	}
 }
 
-func (c *conn) runRead(req srtNewConnReq, pathName string, user string, pass string, query string) (bool, error) {
+func (c *conn) runRead(req srtNewConnReq, streamID *streamID) (bool, error) {
 	res := c.pathManager.AddReader(defs.PathAddReaderReq{
 		Author: c,
 		AccessRequest: defs.PathAccessRequest{
-			Name:  pathName,
+			Name:  streamID.path,
 			IP:    c.ip(),
-			User:  user,
-			Pass:  pass,
+			User:  streamID.user,
+			Pass:  streamID.pass,
 			Proto: defs.AuthProtocolSRT,
 			ID:    &c.uuid,
-			Query: query,
+			Query: streamID.query,
 		},
 	})
 
@@ -320,8 +299,8 @@
 
 	c.mutex.Lock()
 	c.state = connStateRead
-	c.pathName = pathName
-	c.query = query
+	c.pathName = streamID.path
+	c.query = streamID.query
 	c.sconn = sconn
 	c.mutex.Unlock()
 
@@ -345,7 +324,7 @@
 		Conf:            res.Path.SafeConf(),
 		ExternalCmdEnv:  res.Path.ExternalCmdEnv(),
 		Reader:          c.APIReaderDescribe(),
-		Query:           query,
+		Query:           streamID.query,
 	})
 	defer onUnreadHook()
 

internal/servers/srt/streamid.go

@@ -0,0 +1,100 @@
+package srt
+
+import (
+	"fmt"
+	"strings"
+)
+
+type streamIDMode int
+
+const (
+	streamIDModeRead streamIDMode = iota
+	streamIDModePublish
+)
+
+type streamID struct {
+	mode  streamIDMode
+	path  string
+	query string
+	user  string
+	pass  string
+}
+
+func (s *streamID) unmarshal(raw string) error {
+	// standard syntax
+	// https://github.com/Haivision/srt/blob/master/docs/features/access-control.md
+	if strings.HasPrefix(raw, "#!::") {
+		for _, kv := range strings.Split(raw[len("#!::"):], ",") {
+			kv2 := strings.SplitN(kv, "=", 2)
+			if len(kv2) != 2 {
+				return fmt.Errorf("invalid value")
+			}
+
+			key, value := kv2[0], kv2[1]
+
+			switch key {
+			case "u":
+				s.user = value
+
+			case "r":
+				s.path = value
+
+			case "h":
+
+			case "s":
+				s.pass = value
+
+			case "t":
+
+			case "m":
+				switch value {
+				case "request":
+					s.mode = streamIDModeRead
+
+				case "publish":
+					s.mode = streamIDModePublish
+
+				default:
+					return fmt.Errorf("unsupported mode '%s'", value)
+				}
+
+			default:
+				return fmt.Errorf("unsupported key '%s'", key)
+			}
+		}
+	} else {
+		parts := strings.Split(raw, ":")
+		if len(parts) < 2 || len(parts) > 5 {
+			return fmt.Errorf("stream ID must be 'action:pathname[:query]' or 'action:pathname:user:pass[:query]', " +
+				"where action is either read or publish, pathname is the path name, user and pass are the credentials, " +
+				"query is an optional token containing additional information")
+		}
+
+		switch parts[0] {
+		case "read":
+			s.mode = streamIDModeRead
+
+		case "publish":
+			s.mode = streamIDModePublish
+
+		default:
+			return fmt.Errorf("stream ID must be 'action:pathname[:query]' or 'action:pathname:user:pass[:query]', " +
+				"where action is either read or publish, pathname is the path name, user and pass are the credentials, " +
+				"query is an optional token containing additional information")
+		}
+
+		s.path = parts[1]
+
+		if len(parts) == 4 || len(parts) == 5 {
+			s.user, s.pass = parts[2], parts[3]
+		}
+
+		if len(parts) == 3 {
+			s.query = parts[2]
+		} else if len(parts) == 5 {
+			s.query = parts[4]
+		}
+	}
+
+	return nil
+}

internal/servers/srt/streamid_test.go

@@ -0,0 +1,61 @@
+package srt
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestStreamIDUnmarshal(t *testing.T) {
+	for _, ca := range []struct {
+		name string
+		raw  string
+		dec  streamID
+	}{
+		{
+			"mediamtx syntax 1",
+			"read:mypath",
+			streamID{
+				mode: streamIDModeRead,
+				path: "mypath",
+			},
+		},
+		{
+			"mediamtx syntax 2",
+			"publish:mypath:myquery",
+			streamID{
+				mode:  streamIDModePublish,
+				path:  "mypath",
+				query: "myquery",
+			},
+		},
+		{
+			"mediamtx syntax 3",
+			"read:mypath:myuser:mypass:myquery",
+			streamID{
+				mode:  streamIDModeRead,
+				path:  "mypath",
+				user:  "myuser",
+				pass:  "mypass",
+				query: "myquery",
+			},
+		},
+		{
+			"standard syntax",
+			"#!::u=johnny,t=file,m=publish,r=results.csv,s=mypass,h=myhost.com",
+			streamID{
+				mode: streamIDModePublish,
+				path: "results.csv",
+				user: "johnny",
+				pass: "mypass",
+			},
+		},
+	} {
+		t.Run(ca.name, func(t *testing.T) {
+			var streamID streamID
+			err := streamID.unmarshal(ca.raw)
+			require.NoError(t, err)
+			require.Equal(t, ca.dec, streamID)
+		})
+	}
+}