1// Copyright (c) 2010 The WebM project authors. All Rights Reserved. 2// 3// Use of this source code is governed by a BSD-style license 4// that can be found in the LICENSE file in the root of the source 5// tree. An additional intellectual property rights grant can be found 6// in the file PATENTS. All contributing project authors may 7// be found in the AUTHORS file in the root of the source tree. 8 9 10#include "EbmlWriter.h" 11#include <stdlib.h> 12#include <wchar.h> 13#include <string.h> 14#if defined(_MSC_VER) 15#define LITERALU64(n) n 16#else 17#define LITERALU64(n) n##LLU 18#endif 19 20void Ebml_WriteLen(EbmlGlobal *glob, long long val) 21{ 22 //TODO check and make sure we are not > than 0x0100000000000000LLU 23 unsigned char size = 8; //size in bytes to output 24 unsigned long long minVal = LITERALU64(0x00000000000000ff); //mask to compare for byte size 25 26 for (size = 1; size < 8; size ++) 27 { 28 if (val < minVal) 29 break; 30 31 minVal = (minVal << 7); 32 } 33 34 val |= (LITERALU64(0x000000000000080) << ((size - 1) * 7)); 35 36 Ebml_Serialize(glob, (void *) &val, size); 37} 38 39void Ebml_WriteString(EbmlGlobal *glob, const char *str) 40{ 41 const size_t size_ = strlen(str); 42 const unsigned long long size = size_; 43 Ebml_WriteLen(glob, size); 44 //TODO: it's not clear from the spec whether the nul terminator 45 //should be serialized too. For now we omit the null terminator. 46 Ebml_Write(glob, str, size); 47} 48 49void Ebml_WriteUTF8(EbmlGlobal *glob, const wchar_t *wstr) 50{ 51 const size_t strlen = wcslen(wstr); 52 53 //TODO: it's not clear from the spec whether the nul terminator 54 //should be serialized too. For now we include it. 55 const unsigned long long size = strlen; 56 57 Ebml_WriteLen(glob, size); 58 Ebml_Write(glob, wstr, size); 59} 60 61void Ebml_WriteID(EbmlGlobal *glob, unsigned long class_id) 62{ 63 if (class_id >= 0x01000000) 64 Ebml_Serialize(glob, (void *)&class_id, 4); 65 else if (class_id >= 0x00010000) 66 Ebml_Serialize(glob, (void *)&class_id, 3); 67 else if (class_id >= 0x00000100) 68 Ebml_Serialize(glob, (void *)&class_id, 2); 69 else 70 Ebml_Serialize(glob, (void *)&class_id, 1); 71} 72void Ebml_SerializeUnsigned64(EbmlGlobal *glob, unsigned long class_id, uint64_t ui) 73{ 74 unsigned char sizeSerialized = 8 | 0x80; 75 Ebml_WriteID(glob, class_id); 76 Ebml_Serialize(glob, &sizeSerialized, 1); 77 Ebml_Serialize(glob, &ui, 8); 78} 79 80void Ebml_SerializeUnsigned(EbmlGlobal *glob, unsigned long class_id, unsigned long ui) 81{ 82 unsigned char size = 8; //size in bytes to output 83 unsigned char sizeSerialized = 0; 84 unsigned long minVal; 85 86 Ebml_WriteID(glob, class_id); 87 minVal = 0x7fLU; //mask to compare for byte size 88 89 for (size = 1; size < 4; size ++) 90 { 91 if (ui < minVal) 92 { 93 break; 94 } 95 96 minVal <<= 7; 97 } 98 99 sizeSerialized = 0x80 | size; 100 Ebml_Serialize(glob, &sizeSerialized, 1); 101 Ebml_Serialize(glob, &ui, size); 102} 103//TODO: perhaps this is a poor name for this id serializer helper function 104void Ebml_SerializeBinary(EbmlGlobal *glob, unsigned long class_id, unsigned long bin) 105{ 106 int size; 107 for (size=4; size > 1; size--) 108 { 109 if (bin & 0x000000ff << ((size-1) * 8)) 110 break; 111 } 112 Ebml_WriteID(glob, class_id); 113 Ebml_WriteLen(glob, size); 114 Ebml_WriteID(glob, bin); 115} 116 117void Ebml_SerializeFloat(EbmlGlobal *glob, unsigned long class_id, double d) 118{ 119 unsigned char len = 0x88; 120 121 Ebml_WriteID(glob, class_id); 122 Ebml_Serialize(glob, &len, 1); 123 Ebml_Serialize(glob, &d, 8); 124} 125 126void Ebml_WriteSigned16(EbmlGlobal *glob, short val) 127{ 128 signed long out = ((val & 0x003FFFFF) | 0x00200000) << 8; 129 Ebml_Serialize(glob, &out, 3); 130} 131 132void Ebml_SerializeString(EbmlGlobal *glob, unsigned long class_id, const char *s) 133{ 134 Ebml_WriteID(glob, class_id); 135 Ebml_WriteString(glob, s); 136} 137 138void Ebml_SerializeUTF8(EbmlGlobal *glob, unsigned long class_id, wchar_t *s) 139{ 140 Ebml_WriteID(glob, class_id); 141 Ebml_WriteUTF8(glob, s); 142} 143 144void Ebml_SerializeData(EbmlGlobal *glob, unsigned long class_id, unsigned char *data, unsigned long data_length) 145{ 146 unsigned char size = 4; 147 Ebml_WriteID(glob, class_id); 148 Ebml_WriteLen(glob, data_length); 149 Ebml_Write(glob, data, data_length); 150} 151 152void Ebml_WriteVoid(EbmlGlobal *glob, unsigned long vSize) 153{ 154 unsigned char tmp = 0; 155 unsigned long i = 0; 156 157 Ebml_WriteID(glob, 0xEC); 158 Ebml_WriteLen(glob, vSize); 159 160 for (i = 0; i < vSize; i++) 161 { 162 Ebml_Write(glob, &tmp, 1); 163 } 164} 165 166//TODO Serialize Date 167