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