Source: lib/util/pssh.js

/**
 * @license
 * Copyright 2016 Google Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

goog.provide('shaka.util.Pssh');

goog.require('shaka.log');
goog.require('shaka.util.DataViewReader');
goog.require('shaka.util.Mp4Parser');
goog.require('shaka.util.Uint8ArrayUtils');



/**
 * Parse a PSSH box and extract the system IDs.
 *
 * @param {!Uint8Array} psshBox
 * @constructor
 * @struct
 * @throws {shaka.util.Error} if a PSSH box is truncated or contains a size
 *   field over 53 bits.
 */
shaka.util.Pssh = function(psshBox) {
  /**
   * In hex.
   * @type {!Array.<string>}
   */
  this.systemIds = [];

  /**
   * In hex.
   * @type {!Array.<string>}
   */
  this.cencKeyIds = [];

  /*
  * Array of tuples that define the startIndex + size for each
  * discrete pssh within |psshBox|
  * */
  this.dataBoundaries = [];

  // Parse the PSSH box.
  var reader = new shaka.util.DataViewReader(
      new DataView(psshBox.buffer),
      shaka.util.DataViewReader.Endianness.BIG_ENDIAN);

  var psshBoxFound = false;

  // There could be multiple boxes concatenated together.
  while (reader.hasMoreData()) {
    var size = shaka.util.Mp4Parser.findBox(shaka.util.Pssh.BOX_TYPE, reader);
    if (size == shaka.util.Mp4Parser.BOX_NOT_FOUND)
      break;

    // a file can have several pssh boxes, mark that at least
    // one was found
    psshBoxFound = true;
    var startPosition = reader.getPosition() - 8;
    var version = reader.readUint8();
    if (version > 1) {
      shaka.log.warning('Unrecognized PSSH version found!');
      reader.skip(size - (reader.getPosition() - startPosition));
      continue;
    }

    reader.skip(3);  // Skip flags.

    var systemId = shaka.util.Uint8ArrayUtils.toHex(reader.readBytes(16));
    var keyIds = [];
    if (version > 0) {
      var numKeyIds = reader.readUint32();
      for (var i = 0; i < numKeyIds; ++i) {
        var keyId = shaka.util.Uint8ArrayUtils.toHex(reader.readBytes(16));
        keyIds.push(keyId);
      }
    }

    var dataSize = reader.readUint32();
    reader.skip(dataSize);  // Ignore the data section.

    // Now that everything has been succesfully parsed from this box,
    // update member variables.
    this.cencKeyIds.push.apply(this.cencKeyIds, keyIds);
    this.systemIds.push(systemId);
    this.dataBoundaries.push({
      start: startPosition,
      end: reader.getPosition() - 1 }
    );

    if (reader.getPosition() != startPosition + size) {
      shaka.log.warning('Mismatch between box size and data size!');
      reader.skip(size - (reader.getPosition() - startPosition));
    }
  }

  if (!psshBoxFound)
    shaka.log.warning('No pssh box found!');
};


/** @const {number} */
shaka.util.Pssh.BOX_TYPE = 0x70737368;