Merge pull request #9 from wwllww/master

add sctp map
notedit · May 12, 2022 · 504dbb3 · 504dbb3
2 parents e4233c6 + cff6a6a
commit 504dbb3
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/sdpinfo.go b/sdpinfo.go
@@ -451,6 +451,11 @@ func (s *SDPInfo) String() string {
 		}
 
 		if strings.ToLower(media.mtype) == "application" {
+			mediaMap.SctpMap = &transform.SctpMapStuct{
+				5000,
+				"webrtc-datachannel",
+				1024,
+			}
 			mediaMap.Payloads = media.Payloads
 			mediaMap.SctpPort = 5000
 			mediaMap.SctpMaxSize = 256 * 1024

diff --git a/transform/grammer.go b/transform/grammer.go
@@ -529,9 +529,9 @@ var rulesMap map[byte][]*Rule = map[byte][]*Rule{
 			Format: "",
 			FormatFunc: func(obj *gabs.Container) string {
 				if hasValue(obj, "maxMessageSize") {
-					return "sctpmap:%s %s %s"
+					return "sctpmap:%d %s %d"
 				} else {
-					return "sctpmap:%s %s"
+					return "sctpmap:%d %s"
 				}
 			},
 		},

diff --git a/transform/sdpstruct.go b/transform/sdpstruct.go
@@ -145,6 +145,7 @@ type MediaStruct struct {
 	Simulcast   *SimulcastStruct   `json:"simulcast,omitempty"`
 	SctpPort    int                `json:"sctp-port,omitempty"`
 	SctpMaxSize int                `json:"sctp-max-message-size,omitempty"`
+	SctpMap     *SctpMapStuct      `json:"sctpmap,omitempty"`
 }
 
 type SdpStruct struct {
@@ -159,3 +160,9 @@ type SdpStruct struct {
 	Connection   *ConnectionStruct   `json:"connection,omitempty"`
 	Icelite      string              `json:"icelite,omitempty"`
 }
+
+type SctpMapStuct struct {
+	Port           int       `json:"sctpmapNumber"`
+	App            string    `json:"app"`
+	MessageSize    int       `json:"maxMessageSize"`
+}