/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "mozilla/Unused.h"

#include "SinfParser.h"

#include "AtomType.h"

#include "Box.h"

#include "ByteStream.h"

namespace mozilla {

Sinf::Sinf(Box& aBox)

  : mDefaultIVSize(0)

  , mDefaultEncryptionType()

{

  SinfParser parser(aBox);

  if (parser.GetSinf().IsValid()) {

    *this = parser.GetSinf();

  }

}

SinfParser::SinfParser(Box& aBox)

{

  for (Box box = aBox.FirstChild(); box.IsAvailable(); box = box.Next()) {

    if (box.IsType("schm")) {

      mozilla::Unused << ParseSchm(box);

    } else if (box.IsType("schi")) {

      mozilla::Unused << ParseSchi(box);

    }

  }

}

Result<Ok, nsresult>

SinfParser::ParseSchm(Box& aBox)

{

  BoxReader reader(aBox);

  if (reader->Remaining() < 8) {

    return Err(NS_ERROR_FAILURE);

  }

  MOZ_TRY(reader->ReadU32()); // flags -- ignore

  MOZ_TRY_VAR(mSinf.mDefaultEncryptionType, reader->ReadU32());

  return Ok();

}

Result<Ok, nsresult>

SinfParser::ParseSchi(Box& aBox)

{

  for (Box box = aBox.FirstChild(); box.IsAvailable(); box = box.Next()) {

    if (box.IsType("tenc") && ParseTenc(box).isErr()) {

      return Err(NS_ERROR_FAILURE);

    }

  }

  return Ok();

}

Result<Ok, nsresult>

SinfParser::ParseTenc(Box& aBox)

{

  BoxReader reader(aBox);

  if (reader->Remaining() < 24) {

    return Err(NS_ERROR_FAILURE);

  }

  MOZ_TRY(reader->ReadU32()); // flags -- ignore

  uint32_t isEncrypted;

  MOZ_TRY_VAR(isEncrypted, reader->ReadU24());

  MOZ_TRY_VAR(mSinf.mDefaultIVSize, reader->ReadU8());

  memcpy(mSinf.mDefaultKeyID, reader->Read(16), 16);

  return Ok();

}

}