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