/* * Copyright (c) 2010 The WebM project authors. All Rights Reserved. * * Use of this source code is governed by a BSD-style license * that can be found in the LICENSE file in the root of the source * tree. An additional intellectual property rights grant can be found * in the file PATENTS. All contributing project authors may * be found in the AUTHORS file in the root of the source tree. */ #include "EbmlWriter.h" #include <stdlib.h> #include <wchar.h> #include <string.h> #include <limits.h> #include "EbmlBufferWriter.h" #if defined(_MSC_VER) #define LITERALU64(n) n #else #define LITERALU64(n) n##LLU #endif void Ebml_WriteLen(EbmlGlobal *glob, int64_t val) { /* TODO check and make sure we are not > than 0x0100000000000000LLU */ unsigned char size = 8; /* size in bytes to output */ /* mask to compare for byte size */ int64_t minVal = 0xff; for (size = 1; size < 8; size ++) { if (val < minVal) break; minVal = (minVal << 7); } val |= (((uint64_t)0x80) << ((size - 1) * 7)); Ebml_Serialize(glob, (void *) &val, sizeof(val), size); } void Ebml_WriteString(EbmlGlobal *glob, const char *str) { const size_t size_ = strlen(str); const uint64_t size = size_; Ebml_WriteLen(glob, size); /* TODO: it's not clear from the spec whether the nul terminator * should be serialized too. For now we omit the null terminator. */ Ebml_Write(glob, str, (unsigned long)size); } void Ebml_WriteUTF8(EbmlGlobal *glob, const wchar_t *wstr) { const size_t strlen = wcslen(wstr); /* TODO: it's not clear from the spec whether the nul terminator * should be serialized too. For now we include it. */ const uint64_t size = strlen; Ebml_WriteLen(glob, size); Ebml_Write(glob, wstr, (unsigned long)size); } void Ebml_WriteID(EbmlGlobal *glob, unsigned long class_id) { int len; if (class_id >= 0x01000000) len = 4; else if (class_id >= 0x00010000) len = 3; else if (class_id >= 0x00000100) len = 2; else len = 1; Ebml_Serialize(glob, (void *)&class_id, sizeof(class_id), len); } void Ebml_SerializeUnsigned32(EbmlGlobal *glob, unsigned long class_id, uint32_t ui) { unsigned char sizeSerialized = 8 | 0x80; Ebml_WriteID(glob, class_id); Ebml_Serialize(glob, &sizeSerialized, sizeof(sizeSerialized), 1); Ebml_Serialize(glob, &ui, sizeof(ui), 4); } void Ebml_SerializeUnsigned64(EbmlGlobal *glob, unsigned long class_id, uint64_t ui) { unsigned char sizeSerialized = 8 | 0x80; Ebml_WriteID(glob, class_id); Ebml_Serialize(glob, &sizeSerialized, sizeof(sizeSerialized), 1); Ebml_Serialize(glob, &ui, sizeof(ui), 8); } void Ebml_SerializeUnsigned(EbmlGlobal *glob, unsigned long class_id, unsigned long ui) { unsigned char size = 8; /* size in bytes to output */ unsigned char sizeSerialized = 0; unsigned long minVal; Ebml_WriteID(glob, class_id); minVal = 0x7fLU; /* mask to compare for byte size */ for (size = 1; size < 4; size ++) { if (ui < minVal) { break; } minVal <<= 7; } sizeSerialized = 0x80 | size; Ebml_Serialize(glob, &sizeSerialized, sizeof(sizeSerialized), 1); Ebml_Serialize(glob, &ui, sizeof(ui), size); } /* TODO: perhaps this is a poor name for this id serializer helper function */ void Ebml_SerializeBinary(EbmlGlobal *glob, unsigned long class_id, unsigned long bin) { int size; for (size = 4; size > 1; size--) { if (bin & (unsigned int)0x000000ff << ((size - 1) * 8)) break; } Ebml_WriteID(glob, class_id); Ebml_WriteLen(glob, size); Ebml_WriteID(glob, bin); } void Ebml_SerializeFloat(EbmlGlobal *glob, unsigned long class_id, double d) { unsigned char len = 0x88; Ebml_WriteID(glob, class_id); Ebml_Serialize(glob, &len, sizeof(len), 1); Ebml_Serialize(glob, &d, sizeof(d), 8); } void Ebml_WriteSigned16(EbmlGlobal *glob, short val) { signed long out = ((val & 0x003FFFFF) | 0x00200000) << 8; Ebml_Serialize(glob, &out, sizeof(out), 3); } void Ebml_SerializeString(EbmlGlobal *glob, unsigned long class_id, const char *s) { Ebml_WriteID(glob, class_id); Ebml_WriteString(glob, s); } void Ebml_SerializeUTF8(EbmlGlobal *glob, unsigned long class_id, wchar_t *s) { Ebml_WriteID(glob, class_id); Ebml_WriteUTF8(glob, s); } void Ebml_SerializeData(EbmlGlobal *glob, unsigned long class_id, unsigned char *data, unsigned long data_length) { Ebml_WriteID(glob, class_id); Ebml_WriteLen(glob, data_length); Ebml_Write(glob, data, data_length); } void Ebml_WriteVoid(EbmlGlobal *glob, unsigned long vSize) { unsigned char tmp = 0; unsigned long i = 0; Ebml_WriteID(glob, 0xEC); Ebml_WriteLen(glob, vSize); for (i = 0; i < vSize; i++) { Ebml_Write(glob, &tmp, 1); } } /* TODO Serialize Date */