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
Ebml_WriteLen(EbmlGlobal * glob,long long val)20 void 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
Ebml_WriteString(EbmlGlobal * glob,const char * str)39 void 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
Ebml_WriteUTF8(EbmlGlobal * glob,const wchar_t * wstr)49 void 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
Ebml_WriteID(EbmlGlobal * glob,unsigned long class_id)61 void 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 }
Ebml_SerializeUnsigned64(EbmlGlobal * glob,unsigned long class_id,uint64_t ui)72 void 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
Ebml_SerializeUnsigned(EbmlGlobal * glob,unsigned long class_id,unsigned long ui)80 void 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
Ebml_SerializeBinary(EbmlGlobal * glob,unsigned long class_id,unsigned long bin)104 void 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
Ebml_SerializeFloat(EbmlGlobal * glob,unsigned long class_id,double d)117 void 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
Ebml_WriteSigned16(EbmlGlobal * glob,short val)126 void Ebml_WriteSigned16(EbmlGlobal *glob, short val)
127 {
128 signed long out = ((val & 0x003FFFFF) | 0x00200000) << 8;
129 Ebml_Serialize(glob, &out, 3);
130 }
131
Ebml_SerializeString(EbmlGlobal * glob,unsigned long class_id,const char * s)132 void 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
Ebml_SerializeUTF8(EbmlGlobal * glob,unsigned long class_id,wchar_t * s)138 void 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
Ebml_SerializeData(EbmlGlobal * glob,unsigned long class_id,unsigned char * data,unsigned long data_length)144 void 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
Ebml_WriteVoid(EbmlGlobal * glob,unsigned long vSize)152 void 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