fu rtmp

Author: ThibaultBee

README.md

@@ -6,11 +6,11 @@ A RTMP client and server (soon) library for Kotlin Multiplatform.
 
 Features:
 
-- [x] RTMP publish client
-- [ ] RTMP play client
-- [ ] RTMP play2 client
-- [ ] RTMP server
+- [x] RTMP client
+- [x] RTMP server
 - [ ] Statistics
+- [x] Support for legacy RTMP
+- [x] Support for enhanced RTMP v2: AV1, HEVC, VP8, VP9
 
 Supported protocols:
 
@@ -43,7 +43,7 @@ Then prepare your live by sending these messages to the server:
 ```kotlin
 client.connect()
 client.createStream()
-client.publish(Command.Publish.Type.LIVE)
+client.publish(StreamPublishType.LIVE)
 ```
 
 If you already have FLV data, write your video/audio data:

rtmp/src/commonMain/kotlin/io/github/thibaultbee/krtmp/rtmp/messages/command/ConnectObject.kt

@@ -49,6 +49,9 @@ enum class ObjectEncoding(val value: Int) {
  * @param videoFunction The supported (by the client) video functions
  * @param pageUrl  The URL of the web page in which the media was embedded
  * @param objectEncoding The AMF encoding version
+ * @param audioTrackIdInfoMap The audio track ID information map (enhanced RTMP v2)
+ * @param videoFourCcInfoMap The video FourCC information map (enhanced RTMP v2)
+ * @param capsEx The extended capabilities of the client (enhanced RTMP v2)
  */
 class ConnectObjectBuilder(
     var app: String,
@@ -61,7 +64,10 @@ class ConnectObjectBuilder(
     var videoCodecs: List<VideoMediaType>? = DEFAULT_VIDEO_CODECS,
     var videoFunction: List<ConnectObject.VideoFunction> = DEFAULT_VIDEO_FUNCTION,
     var pageUrl: String? = null,
-    var objectEncoding: ObjectEncoding = ObjectEncoding.AMF0
+    var objectEncoding: ObjectEncoding = ObjectEncoding.AMF0,
+    var audioTrackIdInfoMap: Map<AudioMediaType, List<ConnectObject.FourCCInfo>>? = null,
+    var videoFourCcInfoMap: Map<VideoMediaType, List<ConnectObject.FourCCInfo>>? = null,
+    var capsEx: List<ConnectObject.CapsEx>? = null,
 ) {
     fun build() = ConnectObject(
         app,
@@ -87,7 +93,22 @@ class ConnectObjectBuilder(
             acc or vFunction.value
         }.toDouble(),
         pageUrl,
-        objectEncoding.value.toDouble()
+        objectEncoding.value.toDouble(),
+        audioTrackIdInfoMap?.entries?.filter { it.key.fourCCs != null }?.associate {
+            it.key.fourCCs!!.value.toString() to
+                    it.value.fold(0) { acc, fourCCInfo ->
+                        acc or fourCCInfo.value
+                    }.toDouble()
+        },
+        videoFourCcInfoMap?.entries?.filter { it.key.fourCCs != null }?.associate {
+            it.key.fourCCs!!.value.toString() to
+                    it.value.fold(0) { acc, fourCCInfo ->
+                        acc or fourCCInfo.value
+                    }.toDouble()
+        },
+        capsEx?.fold(0) { acc, capEx ->
+            acc or capEx.value
+        }?.toDouble()
     )
 }
 
@@ -101,9 +122,12 @@ class ConnectObjectBuilder(
  * @param fpad True if proxy is used
  * @param audioCodecs The supported (by the client) audio codecs
  * @param videoCodecs The supported (by the client) video codecs
- * @param fourCcList The supported (by the client) video codecs (extended RTMP)
+ * @param fourCcList The supported (by the client) video codecs (enhanced RTMP v1)
  * @param pageUrl  The URL of the web page in which the media was embedded
  * @param objectEncoding The AMF encoding version
+ * @param audioTrackIdInfoMap The audio track ID information map (enhanced RTMP v2)
+ * @param videoTrackIdInfoMap The video track ID information map (enhanced RTMP v2)
+ * @param capsEx The extended capabilities of the client (enhanced RTMP v2)
  */
 @Serializable
 class ConnectObject
@@ -119,7 +143,10 @@ internal constructor(
     val fourCcList: List<String>?,
     val videoFunction: Double = 0.0,
     val pageUrl: String?,
-    val objectEncoding: Double = ObjectEncoding.AMF0.value.toDouble()
+    val objectEncoding: Double = ObjectEncoding.AMF0.value.toDouble(),
+    val audioTrackIdInfoMap: Map<String, Double>? = null,
+    val videoTrackIdInfoMap: Map<String, Double>? = null,
+    val capsEx: Double? = null
 ) {
     override fun toString(): String {
         return "ConnectObject(app='$app', flashVer='$flashVer', tcUrl='$tcUrl', swfUrl=$swfUrl, fpad=$fpad, capabilities=$capabilities, audioCodecs=$audioCodecs, videoCodecs=$videoCodecs, fourCcList=$fourCcList, videoFunction=$videoFunction, pageUrl=$pageUrl, objectEncoding=$objectEncoding)"
@@ -136,7 +163,7 @@ internal constructor(
             VideoMediaType.SORENSON_H263, VideoMediaType.AVC
         )
     }
-    
+
     enum class AudioCodec(val value: Int, val mediaType: AudioMediaType?) {
         NONE(0x0001, null),
         ADPCM(0x0002, AudioMediaType.ADPCM),
@@ -214,4 +241,17 @@ internal constructor(
         CLIENT_PACKET_TYPE_METADATA(0x4),
         CLIENT_LARGE_SCALE_TILE(0x8),
     }
+
+    enum class CapsEx(val value: Int) {
+        RECONNECT(0x01),
+        MULTITRACK(0x02),
+        MODEX(0x04),
+        TIMESTAMP_NANO_OFFSET(0x08),
+    }
+
+    enum class FourCCInfo(val value: Int) {
+        CAN_DECODE(0x01),
+        CAN_ENCODE(0x02),
+        CAN_FORWARD(0x04),
+    }
 }

samples/rtmpclient-cli/src/main/kotlin/Main.kt

@@ -59,15 +59,10 @@ class RTMPClientCli : SuspendingCliktCommand() {
 
             val header = parser.decodeFlvHeader()
             echo("Parsed FLV header: $header")
-
-            /*
+            
             parser.decodeAllRaw { tag ->
                 echo("Sending: $tag")
                 client.write(tag)
-            }*/
-            parser.decodeAll { tag ->
-                echo("Sending: $tag")
-                client.write(tag)
             }
         } catch (t: Throwable) {
             echo("Error reading FLV file: ${t.message}")