Support HLS MPEG2 TS with SAMPLE-AES encryption.

media/formats/mp2t/ts_section_pmt.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "media/formats/mp2t/ts_section_pmt.h"
 
 #include <map>
+#include <string>

[ddorwin, 2016/04/12 00:40:48]

unused?

[dougsteed, 2016/05/08 23:18:45]

Done.

 
 #include "base/logging.h"
 #include "media/base/bit_reader.h"
 #include "media/formats/mp2t/mp2t_common.h"
 
 namespace media {
 namespace mp2t {
 
 TsSectionPmt::TsSectionPmt(const RegisterPesCb& register_pes_cb)
     : register_pes_cb_(register_pes_cb) {
@@ skipping 54 matching lines @@
 
   // Read the program info descriptor.
   // TODO(damienv): check wether any of the descriptors could be useful.
   // Defined in section 2.6 of ISO-13818.
   RCHECK(bit_reader->SkipBits(8 * program_info_length));
 
   // Read the ES description table.
   // The end of the PID map if 4 bytes away from the end of the section
   // (4 bytes = size of the CRC).
   int pid_map_end_marker = section_start_marker - section_length + 4;
-  std::map<int, int> pid_map;
+  using PidMapValue = std::pair<int, Descriptors>;
+  using PidMapElement = std::pair<int, PidMapValue>;
+  std::map<int, PidMapValue> pid_map;
   while (bit_reader->bits_available() > 8 * pid_map_end_marker) {
     int stream_type;
     int reserved;
     int pid_es;
     int es_info_length;
     RCHECK(bit_reader->ReadBits(8, &stream_type));
     RCHECK(bit_reader->ReadBits(3, &reserved));
     RCHECK(bit_reader->ReadBits(13, &pid_es));
     RCHECK(bit_reader->ReadBits(4, &reserved));
     RCHECK(bit_reader->ReadBits(12, &es_info_length));
 
+    Descriptors descriptors;
+    RCHECK(descriptors.Read(bit_reader, es_info_length));
+
     // Do not register the PID right away.
     // Wait for the end of the section to be fully parsed
     // to make sure there is no error.
-    pid_map.insert(std::pair<int, int>(pid_es, stream_type));
-
-    // Read the ES info descriptors.
-    // TODO(damienv): check wether any of the descriptors could be useful.
-    // Defined in section 2.6 of ISO-13818.
-    RCHECK(bit_reader->SkipBits(8 * es_info_length));
+    PidMapValue stream_info(stream_type, descriptors);
+    pid_map.insert(PidMapElement(pid_es, stream_info));
   }
 
   // Read the CRC.
   int crc32;
   RCHECK(bit_reader->ReadBits(32, &crc32));
 
   // Once the PMT has been proved to be correct, register the PIDs.
-  for (std::map<int, int>::iterator it = pid_map.begin();
-       it != pid_map.end(); ++it)
-    register_pes_cb_.Run(it->first, it->second);
+  for (const auto& it : pid_map)
+    register_pes_cb_.Run(it.first, it.second.first, it.second.second);
 
   return true;
 }
 
 void TsSectionPmt::ResetPsiSection() {
 }
 
 }  // namespace mp2t
 }  // namespace media