From 58cbd072c8463b92810726b2ba051fd3d0f31448 Mon Sep 17 00:00:00 2001
From: aler9 <46489434+aler9@users.noreply.github.com>
Date: Sat, 11 Jun 2022 00:31:05 +0200
Subject: [PATCH] rtmp: support ping requests

---
 internal/rtmp/message/readwriter.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/internal/rtmp/message/readwriter.go b/internal/rtmp/message/readwriter.go
index ccede8c4..8ff19d7d 100644
--- a/internal/rtmp/message/readwriter.go
+++ b/internal/rtmp/message/readwriter.go
@@ -16,7 +16,7 @@ func NewReadWriter(bc *bytecounter.ReadWriter) *ReadWriter {
 
 	r := NewReader(bc.Reader, func(count uint32) error {
 		return w.Write(&MsgAcknowledge{
-			Value: (count),
+			Value: count,
 		})
 	})
 
@@ -33,8 +33,14 @@ func (rw *ReadWriter) Read() (Message, error) {
 		return nil, err
 	}
 
-	if tmsg, ok := msg.(*MsgAcknowledge); ok {
+	switch tmsg := msg.(type) {
+	case *MsgAcknowledge:
 		rw.w.SetAcknowledgeValue(tmsg.Value)
+
+	case *MsgUserControlPingRequest:
+		rw.w.Write(&MsgUserControlPingRequest{
+			ServerTime: tmsg.ServerTime,
+		})
 	}
 
 	return msg, nil