libmist backport from new metadata branch with improved typing and styling
This commit is contained in:
Erik Zandvliet
Thulinma
parent
7a03d3e96c
commit
10fa4b7e7b
88 changed files with 5957 additions and 5757 deletions
|
|
@ -1,11 +1,309 @@
|
|||
#include "mp4_encryption.h"
|
||||
#include "defines.h"
|
||||
|
||||
namespace MP4 {
|
||||
|
||||
PSSH::PSSH() {
|
||||
memcpy(data + 4, "pssh", 4);
|
||||
}
|
||||
|
||||
std::string PSSH::toPrettyString(uint32_t indent) {
|
||||
std::stringstream r;
|
||||
r << std::string(indent, ' ') << "[pssh] Protection System Specific Header Box (" << boxedSize() << ")" << std::endl;
|
||||
r << fullBox::toPrettyString(indent);
|
||||
r << std::string(indent + 1, ' ') << "SystemID: " << getSystemIDHex() << std::endl;
|
||||
if (getVersion()){
|
||||
r << std::string(indent + 1, ' ') << "KID_count: " << getKIDCount() << std::endl;
|
||||
}
|
||||
r << std::string(indent + 1, ' ') << "DataSize: " << getDataSize() << std::endl;
|
||||
r << std::string(indent + 1, ' ') << "Data: ";
|
||||
size_t dataSize = getDataSize();
|
||||
char * data = getData();
|
||||
for (size_t i = 0; i < dataSize; ++i){
|
||||
r << std::hex << std::setw(2) << std::setfill('0') << (int)data[i] << std::dec << "";
|
||||
}
|
||||
r << std::endl;
|
||||
return r.str();
|
||||
}
|
||||
|
||||
std::string PSSH::getSystemIDHex() {
|
||||
char * systemID = getString(4);
|
||||
std::stringstream res;
|
||||
for (size_t i = 0; i < 16; ++i){
|
||||
res << std::hex << std::setw(2) << std::setfill('0') << (int)systemID[i] << std::dec;
|
||||
}
|
||||
return "0x" + res.str();
|
||||
}
|
||||
|
||||
void PSSH::setSystemIDHex(const std::string & systemID) {
|
||||
setString(systemID, 4);
|
||||
}
|
||||
|
||||
size_t PSSH::getKIDCount(){
|
||||
return getVersion() ? getInt32(20) : 0;
|
||||
}
|
||||
|
||||
size_t PSSH::getDataSize(){
|
||||
if (getVersion()){
|
||||
size_t kidCount = getInt32(20);
|
||||
return getInt32(24 + (kidCount * 16));
|
||||
}
|
||||
return getInt32(20);
|
||||
}
|
||||
|
||||
char * PSSH::getData(){
|
||||
if (getVersion()){
|
||||
size_t kidCount = getInt32(20);
|
||||
return getString(24 + (kidCount * 16) + 4);
|
||||
}
|
||||
return getString(24);
|
||||
}
|
||||
|
||||
void PSSH::setData(const std::string & data){
|
||||
if (getVersion()){
|
||||
WARN_MSG("Not implemented yet!");
|
||||
return;
|
||||
}
|
||||
for (int i = 0; i < data.size(); i++){
|
||||
setInt8(data[i], 24 + i);
|
||||
}
|
||||
setInt32(data.size(), 20);
|
||||
}
|
||||
|
||||
std::string TENC::toPrettyString(uint32_t indent) {
|
||||
std::stringstream r;
|
||||
r << std::string(indent, ' ') << "[tenc] Track Encryption Box (" << boxedSize() << ")" << std::endl;
|
||||
r << fullBox::toPrettyString(indent);
|
||||
r << std::string(indent + 1, ' ') << "default_isEncrypted: " << getDefaultIsEncrypted() << std::endl;
|
||||
r << std::string(indent + 1, ' ') << "default_IV_size: " << getDefaultIVSize() << std::endl;
|
||||
r << std::string(indent + 1, ' ') << "default_KID: ";
|
||||
std::string defaultKID = getDefaultKID();
|
||||
for (int i = 0; i < 16; i++){
|
||||
r << std::hex << std::setw(2) << std::setfill('0') << (int)defaultKID[i] << std::dec << " ";
|
||||
}
|
||||
r << std::endl;
|
||||
return r.str();
|
||||
}
|
||||
|
||||
TENC::TENC() {
|
||||
memcpy(data + 4, "tenc", 4);
|
||||
setDefaultIsEncrypted(1);
|
||||
setDefaultIVSize(8);
|
||||
}
|
||||
size_t TENC::getDefaultIsEncrypted(){
|
||||
return getInt24(4);
|
||||
}
|
||||
void TENC::setDefaultIsEncrypted(size_t isEncrypted){
|
||||
setInt24(isEncrypted, 4);
|
||||
}
|
||||
size_t TENC::getDefaultIVSize(){
|
||||
return getInt8(7);
|
||||
}
|
||||
void TENC::setDefaultIVSize(uint8_t ivSize){
|
||||
setInt8(ivSize, 7);
|
||||
}
|
||||
std::string TENC::getDefaultKID(){
|
||||
std::string result;
|
||||
for (int i = 8; i < 24; i++){
|
||||
result += getInt8(i);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
void TENC::setDefaultKID(const std::string & kid){
|
||||
for (int i = 0; i < 16; i++){
|
||||
if (i < kid.size()){
|
||||
setInt8(kid[i], i + 8);
|
||||
}else{
|
||||
setInt8(0, i + 8);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
SENC::SENC() {
|
||||
memcpy(data + 4, "senc", 4);
|
||||
setFlags(2);
|
||||
}
|
||||
uint32_t SENC::getSampleCount() const {
|
||||
return getInt32(4);
|
||||
}
|
||||
|
||||
#define IV_SIZE 8
|
||||
void SENC::setSample(UUID_SampleEncryption_Sample newSample, size_t index) {
|
||||
int myOffset = 8;
|
||||
for (unsigned int i = 0; i < std::min(index, (size_t)getSampleCount()); i++) {
|
||||
myOffset += IV_SIZE;
|
||||
if (getFlags() & 0x02) {
|
||||
int entryCount = getInt16(myOffset);
|
||||
myOffset += 2 + (entryCount * 6);
|
||||
}
|
||||
}
|
||||
if (index > getSampleCount()) {
|
||||
ERROR_MSG("First fill intermediate entries!");
|
||||
return;
|
||||
/*
|
||||
//we are now at the end of currently reserved space, reserve more and adapt offset accordingly.
|
||||
int reserveSize = ((index - getSampleCount())) * (IV_SIZE + (getFlags() & 0x02));
|
||||
reserveSize += IV_SIZE;
|
||||
if (getFlags() & 0x02) {
|
||||
reserveSize += 2 + newSample.Entries.size();
|
||||
}
|
||||
if (!reserve(myOffset, 0, reserveSize)) {
|
||||
return;//Memory errors...
|
||||
}
|
||||
myOffset += (index - getSampleCount()) * (IV_SIZE + (getFlags() & 0x02));
|
||||
*/
|
||||
}
|
||||
//write box.
|
||||
for (int i = 0; i < IV_SIZE; i++) {
|
||||
setInt8(newSample.InitializationVector[i], myOffset ++);//set and skip
|
||||
}
|
||||
if (getFlags() & 0x02) {
|
||||
setInt16(newSample.Entries.size(), myOffset);
|
||||
myOffset += 2;
|
||||
for (std::vector<UUID_SampleEncryption_Sample_Entry>::iterator it = newSample.Entries.begin(); it != newSample.Entries.end(); it++) {
|
||||
setInt16(it->BytesClear, myOffset);
|
||||
myOffset += 2;
|
||||
setInt32(it->BytesEncrypted, myOffset);
|
||||
myOffset += 4;
|
||||
}
|
||||
}
|
||||
if (index >= getSampleCount()) {
|
||||
setInt32(index + 1, 4);
|
||||
}
|
||||
}
|
||||
|
||||
UUID_SampleEncryption_Sample SENC::getSample(size_t index) const {
|
||||
if (index >= getSampleCount()) {
|
||||
return UUID_SampleEncryption_Sample();
|
||||
}
|
||||
int myOffset = 8;
|
||||
for (unsigned int i = 0; i < index; i++) {
|
||||
myOffset += IV_SIZE;
|
||||
if (getFlags() & 0x02) {
|
||||
int entryCount = getInt16(myOffset);
|
||||
myOffset += 2;//skip over entrycount
|
||||
myOffset += entryCount * 6;//skip entryCount sample entries
|
||||
}
|
||||
}
|
||||
UUID_SampleEncryption_Sample result;
|
||||
for (int i = 0; i < IV_SIZE; i++) {
|
||||
result.InitializationVector += (char)getInt8(myOffset++);//read and skip
|
||||
}
|
||||
if (getFlags() & 0x02) {
|
||||
result.NumberOfEntries = getInt16(myOffset);
|
||||
myOffset += 2;
|
||||
for (unsigned int i = 0; i < result.NumberOfEntries; i++) {
|
||||
result.Entries.push_back(UUID_SampleEncryption_Sample_Entry());
|
||||
result.Entries[i].BytesClear = getInt16(myOffset);
|
||||
myOffset += 2;
|
||||
result.Entries[i].BytesEncrypted = getInt32(myOffset);
|
||||
myOffset += 4;
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
std::string SENC::toPrettyString(uint32_t indent) const {
|
||||
std::stringstream r;
|
||||
r << std::string(indent, ' ') << "[senc] Sample Encryption Box (" << boxedSize() << ")" << std::endl;
|
||||
r << fullBox::toPrettyString(indent);
|
||||
r << std::string(indent + 1, ' ') << "Sample Count: " << getSampleCount() << std::endl;
|
||||
for (unsigned int i = 0; i < getSampleCount(); i++) {
|
||||
UUID_SampleEncryption_Sample tmpSample = getSample(i);
|
||||
r << std::string(indent + 1, ' ') << "[" << i << "]" << std::endl;
|
||||
r << std::string(indent + 3, ' ') << "Initialization Vector: 0x";
|
||||
for (unsigned int j = 0; j < tmpSample.InitializationVector.size(); j++) {
|
||||
r << std::hex << std::setw(2) << std::setfill('0') << (int)tmpSample.InitializationVector[j] << std::dec;
|
||||
}
|
||||
r << std::endl;
|
||||
if (getFlags() & 0x02) {
|
||||
r << std::string(indent + 3, ' ') << "Number of entries: " << tmpSample.NumberOfEntries << std::endl;
|
||||
for (unsigned int j = 0; j < tmpSample.NumberOfEntries; j++) {
|
||||
r << std::string(indent + 3, ' ') << "[" << j << "]" << std::endl;
|
||||
r << std::string(indent + 5, ' ') << "Bytes clear: " << tmpSample.Entries[j].BytesClear << std::endl;
|
||||
r << std::string(indent + 5, ' ') << "Bytes encrypted: " << tmpSample.Entries[j].BytesEncrypted << std::endl;
|
||||
}
|
||||
}
|
||||
}
|
||||
return r.str();
|
||||
}
|
||||
|
||||
SAIZ::SAIZ(size_t entryCount){
|
||||
memcpy(data + 4, "saiz", 4);
|
||||
setFlags(0);
|
||||
setVersion(0);
|
||||
setInt24(0, 4); //Default sample size
|
||||
setInt16(entryCount, 7);//EntryCount Samples
|
||||
for (int i = 0; i < entryCount; i++){
|
||||
setInt8(16, i+9);//16 bytes IV's
|
||||
}
|
||||
}
|
||||
|
||||
size_t SAIZ::getDefaultSampleSize(){
|
||||
return getInt24(4);
|
||||
}
|
||||
|
||||
size_t SAIZ::getEntryCount(){
|
||||
return getInt16(7);
|
||||
}
|
||||
|
||||
size_t SAIZ::getEntrySize(size_t entryNo){
|
||||
if (entryNo >= getEntryCount()){return -1;}
|
||||
return getInt8(9 + entryNo);
|
||||
}
|
||||
|
||||
std::string SAIZ::toPrettyString(uint32_t indent) {
|
||||
std::stringstream r;
|
||||
r << std::string(indent, ' ') << "[saiz] Sample Auxiliary Information Size Box (" << boxedSize() << ")" << std::endl;
|
||||
r << fullBox::toPrettyString(indent);
|
||||
r << std::string(indent + 1, ' ') << "Default Sample Size: " << getDefaultSampleSize() << std::endl;
|
||||
r << std::string(indent + 1, ' ') << "Entry Count: " << getEntryCount() << std::endl;
|
||||
for (size_t i = 0; i < getEntryCount(); ++i){
|
||||
r << std::string(indent + 2, ' ') << "[" << i << "]: " << getEntrySize(i) << std::endl;
|
||||
}
|
||||
return r.str();
|
||||
}
|
||||
|
||||
SAIO::SAIO(size_t offset){
|
||||
memcpy(data + 4, "saio", 4);
|
||||
setInt32(1, 4);
|
||||
setInt32(offset, 8);
|
||||
}
|
||||
|
||||
size_t SAIO::getEntryCount(){
|
||||
return getInt32(4);
|
||||
}
|
||||
|
||||
size_t SAIO::getEntrySize(size_t entryNo){
|
||||
if (entryNo >= getEntryCount()){return -1;}
|
||||
return getInt32(8 + (entryNo * 4));
|
||||
}
|
||||
|
||||
std::string SAIO::toPrettyString(uint32_t indent) {
|
||||
std::stringstream r;
|
||||
r << std::string(indent, ' ') << "[saio] Sample Auxiliary Information Offset Box (" << boxedSize() << ")" << std::endl;
|
||||
r << fullBox::toPrettyString(indent);
|
||||
r << std::string(indent + 1, ' ') << "Entry Count: " << getEntryCount() << std::endl;
|
||||
for (size_t i = 0; i < getEntryCount(); ++i){
|
||||
r << std::string(indent + 2, ' ') << "[" << i << "]: " << getEntrySize(i) << std::endl;
|
||||
}
|
||||
return r.str();
|
||||
}
|
||||
|
||||
UUID_SampleEncryption::UUID_SampleEncryption() {
|
||||
setUUID((std::string)"a2394f52-5a9b-4f14-a244-6c427c648df4");
|
||||
}
|
||||
|
||||
UUID_SampleEncryption::UUID_SampleEncryption(const SENC & senc){
|
||||
setUUID((std::string)"a2394f52-5a9b-4f14-a244-6c427c648df4");
|
||||
setVersion(0);
|
||||
setFlags(2);
|
||||
size_t sampleCount = senc.getSampleCount();
|
||||
for (size_t i = 0; i < sampleCount; ++i){
|
||||
setSample(senc.getSample(i), i);
|
||||
}
|
||||
}
|
||||
|
||||
void UUID_SampleEncryption::setVersion(uint32_t newVersion) {
|
||||
setInt8(newVersion, 16);
|
||||
}
|
||||
|
|
@ -93,17 +391,9 @@ namespace MP4 {
|
|||
myOffset += 2 + (entryCount * 6);
|
||||
}
|
||||
}
|
||||
if (index >= getSampleCount()) {
|
||||
//we are now at the end of currently reserved space, reserve more and adapt offset accordingly.
|
||||
int reserveSize = ((index - getSampleCount())) * (IV_SIZE + (getFlags() & 0x02));
|
||||
reserveSize += IV_SIZE;
|
||||
if (getFlags() & 0x02) {
|
||||
reserveSize += 2 + newSample.Entries.size();
|
||||
}
|
||||
if (!reserve(myOffset, 0, reserveSize)) {
|
||||
return;//Memory errors...
|
||||
}
|
||||
myOffset += (index - getSampleCount()) * (IV_SIZE + (getFlags() & 0x02));
|
||||
if (index > getSampleCount()) {
|
||||
ERROR_MSG("First fill intermediate entries!");
|
||||
return;
|
||||
}
|
||||
//write box.
|
||||
for (int i = 0; i < IV_SIZE; i++) {
|
||||
|
|
@ -339,15 +629,13 @@ namespace MP4 {
|
|||
Box & SINF::getEntry(uint32_t no) {
|
||||
static Box ret = Box((char *)"\000\000\000\010erro", false);
|
||||
if (no > 4) {
|
||||
ret = Box((char *)"\000\000\000\010erro", false);
|
||||
return ret;
|
||||
}
|
||||
int tempLoc = 0;
|
||||
for (unsigned int i = 0; i < no; i++) {
|
||||
tempLoc += Box(getBox(tempLoc).asBox(), false).boxedSize();
|
||||
tempLoc += getBoxLen(tempLoc);
|
||||
}
|
||||
ret = Box(getBox(tempLoc).asBox(), false);
|
||||
return ret;
|
||||
return getBox(tempLoc);
|
||||
}
|
||||
|
||||
std::string SINF::toPrettyString(uint32_t indent) {
|
||||
|
|
@ -363,11 +651,12 @@ namespace MP4 {
|
|||
return r.str();
|
||||
}
|
||||
|
||||
FRMA::FRMA() {
|
||||
FRMA::FRMA(const std::string & originalFormat) {
|
||||
memcpy(data + 4, "frma", 4);
|
||||
setOriginalFormat(originalFormat);
|
||||
}
|
||||
|
||||
void FRMA::setOriginalFormat(std::string newFormat) {
|
||||
void FRMA::setOriginalFormat(const std::string & newFormat) {
|
||||
for (unsigned int i = 0; i < 4; i++) {
|
||||
if (i < newFormat.size()) {
|
||||
setInt8(newFormat[i], i);
|
||||
|
|
@ -388,8 +677,10 @@ namespace MP4 {
|
|||
return r.str();
|
||||
}
|
||||
|
||||
SCHM::SCHM() {
|
||||
SCHM::SCHM(uint32_t schemeType, uint32_t schemeVersion){
|
||||
memcpy(data + 4, "schm", 4);
|
||||
setSchemeType(schemeType);
|
||||
setSchemeVersion(schemeVersion);
|
||||
}
|
||||
|
||||
void SCHM::setSchemeType(uint32_t newType) {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue