improvement(hlsstream): protect the segment packet which is encrypted

Signed-off-by: pingkai <pingkai010@gmail.com>
alibaba · Jun 11, 2020 · bd2ca4b · bd2ca4b
1 parent 8ff9bf0
commit bd2ca4b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/framework/demuxer/play_list/HLSStream.cpp b/framework/demuxer/play_list/HLSStream.cpp
@@ -807,6 +807,7 @@ namespace Cicada {
     int HLSStream::updateDecrypter()
     {
         int ret = 0;
+        mProtectedBuffer = mCurSeg->encryption.method != SegmentEncryption::NONE;
 
         if (mCurSeg->encryption.method == SegmentEncryption::AES_128 ||
                 mCurSeg->encryption.method == SegmentEncryption::AES_PRIVATE) {
@@ -973,6 +974,10 @@ namespace Cicada {
 
         if (packet != nullptr) {
             //  AF_LOGD("read a frame \n");
+
+            if (mProtectedBuffer) {
+                packet->setProtected();
+            }
             if (mPTracker->getStreamType() != STREAM_TYPE_MIXED) {
                 packet->getInfo().streamIndex = 0;
             }

diff --git a/framework/demuxer/play_list/HLSStream.h b/framework/demuxer/play_list/HLSStream.h
@@ -213,6 +213,7 @@ namespace Cicada {
 
         std::unique_ptr<DemuxerMetaInfo> mDemuxerMeta = nullptr;
         int OpenedStreamIndex = 0;
+        bool mProtectedBuffer{false};
     };
 }