media/libmkv/EbmlWriter.c

   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
  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
  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
  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
  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
  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
  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
  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 */
 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
 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
 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
 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
 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
 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
 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 */