implement sink IDRRequest handling

libwds/public/media_manager.h

@@ -155,6 +155,8 @@ class SinkMediaManager : public MediaManager {
    * @return connector type. @see ConnectorType
    */
   virtual ConnectorType GetConnectorType() const = 0;
+
+  virtual void IDRRequestDone(bool success) = 0;
 };
 
 /**

libwds/public/sink.h

@@ -42,6 +42,8 @@ class Sink : public Peer {
    * @return newly created Sink instance
    */
   static Sink* Create(Peer::Delegate* delegate, SinkMediaManager* mng);
+
+  virtual bool IDRRequest() = 0;
 };
 
 }

libwds/sink/sink.cpp

@@ -28,6 +28,8 @@
 #include "libwds/rtsp/play.h"
 #include "libwds/rtsp/teardown.h"
 #include "libwds/rtsp/triggermethod.h"
+#include "libwds/rtsp/idrrequest.h"
+#include "libwds/rtsp/setparameter.h"
 #include "libwds/public/media_manager.h"
 #include "libwds/sink/cap_negotiation_state.h"
 #include "libwds/sink/init_state.h"
@@ -109,6 +111,7 @@ class SinkImpl final : public Sink, public RTSPInputHandler, public MessageHandl
   bool Teardown() override;
   bool Play() override;
   bool Pause() override;
+  bool IDRRequest() override;
 
   // RTSPInputHandler
   void MessageParsed(std::unique_ptr<Message> message) override;
@@ -180,6 +183,19 @@ bool SinkImpl::Pause() {
   return HandleCommand(CreateCommand<rtsp::Pause, Request::M9>());
 }
 
+bool SinkImpl::IDRRequest() {
+  auto request = new rtsp::SetParameter(manager_->GetPresentationUrl());
+  request->header().set_session(manager_->GetSessionId());
+  request->header().set_cseq(state_machine_->GetNextCSeq());
+
+  auto request_payload = new rtsp::PropertyMapPayload();
+  request_payload->AddProperty(std::shared_ptr<rtsp::Property>(new rtsp::IDRRequest()));
+  request->set_payload(std::unique_ptr<rtsp::Payload>(request_payload));
+
+  request->set_id(Request::M13);
+  return HandleCommand(std::unique_ptr<Message>(request));
+}
+
 void SinkImpl::MessageParsed(std::unique_ptr<Message> message) {
   if (message->is_request() && !InitializeRequestId(ToRequest(message.get()))) {
     WDS_ERROR("Cannot identify the received message");

libwds/sink/streaming_state.cpp

@@ -211,6 +211,19 @@ class M9SenderOptional final : public OptionalMessageSender<Request::M9> {
   }
 };
 
+class M13SenderOptional final : public OptionalMessageSender<Request::M13> {
+ public:
+  M13SenderOptional(const InitParams& init_params)
+    : OptionalMessageSender<Request::M13>(init_params) {
+  }
+ private:
+  bool HandleReply(Reply* reply) override {
+    bool ok = reply->response_code() == rtsp::STATUS_OK;
+    ToSinkMediaManager(manager_)->IDRRequestDone(ok);
+    return ok;
+  }
+};
+
 StreamingState::StreamingState(const InitParams& init_params, MessageHandlerPtr m16_handler)
   : MessageSequenceWithOptionalSetHandler(init_params) {
   AddSequencedHandler(make_ptr(new TeardownHandler(init_params)));
@@ -219,10 +232,11 @@ StreamingState::StreamingState(const InitParams& init_params, MessageHandlerPtr
   AddOptionalHandler(make_ptr(new M3Handler(init_params)));
   AddOptionalHandler(make_ptr(new M4Handler(init_params)));
 
-  // optional senders that handle sending play, pause and teardown
+  // optional senders that handle sending play, pause, teardown and idr-request
   AddOptionalHandler(make_ptr(new M7SenderOptional(init_params)));
   AddOptionalHandler(make_ptr(new M8SenderOptional(init_params)));
   AddOptionalHandler(make_ptr(new M9SenderOptional(init_params)));
+  AddOptionalHandler(make_ptr(new M13SenderOptional(init_params)));
   AddOptionalHandler(m16_handler);
 }
 

sink/gst_sink_media_manager.cpp

@@ -92,3 +92,6 @@ bool GstSinkMediaManager::SetOptimalVideoFormat(const wds::H264VideoFormat& opti
 wds::ConnectorType GstSinkMediaManager::GetConnectorType() const {
   return wds::ConnectorTypeNone;
 }
+
+void GstSinkMediaManager::IDRRequestDone(bool success) {
+}

sink/gst_sink_media_manager.h

@@ -45,6 +45,7 @@ class GstSinkMediaManager : public wds::SinkMediaManager {
   wds::NativeVideoFormat GetNativeVideoFormat() const override;
   bool SetOptimalVideoFormat(const wds::H264VideoFormat& optimal_format) override;
   wds::ConnectorType GetConnectorType() const override;
+  void IDRRequestDone(bool success) override;
 
  private:
   std::string hostname_;