unittests/Support/EndianStreamTest.cpp

   1 //===- unittests/Support/EndianStreamTest.cpp - EndianStream.h tests ------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "llvm/Support/EndianStream.h"
  11 #include "llvm/ADT/SmallString.h"
  12 #include "llvm/Support/DataTypes.h"
  13 #include "gtest/gtest.h"
  14 using namespace llvm;
  15 using namespace support;
  16
  17 namespace {
  18
  19 TEST(EndianStream, WriteInt32LE) {
  20   SmallString<16> data;
  21
  22   {
  23     raw_svector_ostream OS(data);
  24     endian::Writer<little> LE(OS);
  25     LE.write(static_cast<int32_t>(-1362446643));
  26   }
  27
  28   EXPECT_EQ(static_cast<uint8_t>(data[0]), 0xCD);
  29   EXPECT_EQ(static_cast<uint8_t>(data[1]), 0xB6);
  30   EXPECT_EQ(static_cast<uint8_t>(data[2]), 0xCA);
  31   EXPECT_EQ(static_cast<uint8_t>(data[3]), 0xAE);
  32 }
  33
  34 TEST(EndianStream, WriteInt32BE) {
  35   SmallVector<char, 16> data;
  36
  37   {
  38     raw_svector_ostream OS(data);
  39     endian::Writer<big> BE(OS);
  40     BE.write(static_cast<int32_t>(-1362446643));
  41   }
  42
  43   EXPECT_EQ(static_cast<uint8_t>(data[0]), 0xAE);
  44   EXPECT_EQ(static_cast<uint8_t>(data[1]), 0xCA);
  45   EXPECT_EQ(static_cast<uint8_t>(data[2]), 0xB6);
  46   EXPECT_EQ(static_cast<uint8_t>(data[3]), 0xCD);
  47 }
  48
  49
  50 TEST(EndianStream, WriteFloatLE) {
  51   SmallString<16> data;
  52
  53   {
  54     raw_svector_ostream OS(data);
  55     endian::Writer<little> LE(OS);
  56     LE.write(12345.0f);
  57   }
  58
  59   EXPECT_EQ(static_cast<uint8_t>(data[0]), 0x00);
  60   EXPECT_EQ(static_cast<uint8_t>(data[1]), 0xE4);
  61   EXPECT_EQ(static_cast<uint8_t>(data[2]), 0x40);
  62   EXPECT_EQ(static_cast<uint8_t>(data[3]), 0x46);
  63 }
  64
  65 TEST(EndianStream, WriteFloatBE) {
  66   SmallVector<char, 16> data;
  67
  68   {
  69     raw_svector_ostream OS(data);
  70     endian::Writer<big> BE(OS);
  71     BE.write(12345.0f);
  72   }
  73
  74   EXPECT_EQ(static_cast<uint8_t>(data[0]), 0x46);
  75   EXPECT_EQ(static_cast<uint8_t>(data[1]), 0x40);
  76   EXPECT_EQ(static_cast<uint8_t>(data[2]), 0xE4);
  77   EXPECT_EQ(static_cast<uint8_t>(data[3]), 0x00);
  78 }
  79
  80 TEST(EndianStream, WriteInt64LE) {
  81   SmallString<16> data;
  82
  83   {
  84     raw_svector_ostream OS(data);
  85     endian::Writer<little> LE(OS);
  86     LE.write(static_cast<int64_t>(-136244664332342323));
  87   }
  88
  89   EXPECT_EQ(static_cast<uint8_t>(data[0]), 0xCD);
  90   EXPECT_EQ(static_cast<uint8_t>(data[1]), 0xAB);
  91   EXPECT_EQ(static_cast<uint8_t>(data[2]), 0xED);
  92   EXPECT_EQ(static_cast<uint8_t>(data[3]), 0x1B);
  93   EXPECT_EQ(static_cast<uint8_t>(data[4]), 0x33);
  94   EXPECT_EQ(static_cast<uint8_t>(data[5]), 0xF6);
  95   EXPECT_EQ(static_cast<uint8_t>(data[6]), 0x1B);
  96   EXPECT_EQ(static_cast<uint8_t>(data[7]), 0xFE);
  97 }
  98
  99 TEST(EndianStream, WriteInt64BE) {
 100   SmallVector<char, 16> data;
 101
 102   {
 103     raw_svector_ostream OS(data);
 104     endian::Writer<big> BE(OS);
 105     BE.write(static_cast<int64_t>(-136244664332342323));
 106   }
 107
 108   EXPECT_EQ(static_cast<uint8_t>(data[0]), 0xFE);
 109   EXPECT_EQ(static_cast<uint8_t>(data[1]), 0x1B);
 110   EXPECT_EQ(static_cast<uint8_t>(data[2]), 0xF6);
 111   EXPECT_EQ(static_cast<uint8_t>(data[3]), 0x33);
 112   EXPECT_EQ(static_cast<uint8_t>(data[4]), 0x1B);
 113   EXPECT_EQ(static_cast<uint8_t>(data[5]), 0xED);
 114   EXPECT_EQ(static_cast<uint8_t>(data[6]), 0xAB);
 115   EXPECT_EQ(static_cast<uint8_t>(data[7]), 0xCD);
 116 }
 117
 118 TEST(EndianStream, WriteDoubleLE) {
 119   SmallString<16> data;
 120
 121   {
 122     raw_svector_ostream OS(data);
 123     endian::Writer<little> LE(OS);
 124     LE.write(-2349214918.58107);
 125   }
 126
 127   EXPECT_EQ(static_cast<uint8_t>(data[0]), 0x20);
 128   EXPECT_EQ(static_cast<uint8_t>(data[1]), 0x98);
 129   EXPECT_EQ(static_cast<uint8_t>(data[2]), 0xD2);
 130   EXPECT_EQ(static_cast<uint8_t>(data[3]), 0x98);
 131   EXPECT_EQ(static_cast<uint8_t>(data[4]), 0xC5);
 132   EXPECT_EQ(static_cast<uint8_t>(data[5]), 0x80);
 133   EXPECT_EQ(static_cast<uint8_t>(data[6]), 0xE1);
 134   EXPECT_EQ(static_cast<uint8_t>(data[7]), 0xC1);
 135 }
 136
 137 TEST(EndianStream, WriteDoubleBE) {
 138   SmallVector<char, 16> data;
 139
 140   {
 141     raw_svector_ostream OS(data);
 142     endian::Writer<big> BE(OS);
 143     BE.write(-2349214918.58107);
 144   }
 145
 146   EXPECT_EQ(static_cast<uint8_t>(data[0]), 0xC1);
 147   EXPECT_EQ(static_cast<uint8_t>(data[1]), 0xE1);
 148   EXPECT_EQ(static_cast<uint8_t>(data[2]), 0x80);
 149   EXPECT_EQ(static_cast<uint8_t>(data[3]), 0xC5);
 150   EXPECT_EQ(static_cast<uint8_t>(data[4]), 0x98);
 151   EXPECT_EQ(static_cast<uint8_t>(data[5]), 0xD2);
 152   EXPECT_EQ(static_cast<uint8_t>(data[6]), 0x98);
 153   EXPECT_EQ(static_cast<uint8_t>(data[7]), 0x20);
 154 }
 155
 156 TEST(EndianStream, WriteArrayLE) {
 157   SmallString<16> Data;
 158
 159   {
 160     raw_svector_ostream OS(Data);
 161     endian::Writer<little> LE(OS);
 162     LE.write<uint16_t>({0x1234, 0x5678});
 163   }
 164
 165   EXPECT_EQ(static_cast<uint8_t>(Data[0]), 0x34);
 166   EXPECT_EQ(static_cast<uint8_t>(Data[1]), 0x12);
 167   EXPECT_EQ(static_cast<uint8_t>(Data[2]), 0x78);
 168   EXPECT_EQ(static_cast<uint8_t>(Data[3]), 0x56);
 169 }
 170
 171 TEST(EndianStream, WriteVectorLE) {
 172   SmallString<16> Data;
 173
 174   {
 175     raw_svector_ostream OS(Data);
 176     endian::Writer<little> LE(OS);
 177     std::vector<uint16_t> Vec{0x1234, 0x5678};
 178     LE.write<uint16_t>(Vec);
 179   }
 180
 181   EXPECT_EQ(static_cast<uint8_t>(Data[0]), 0x34);
 182   EXPECT_EQ(static_cast<uint8_t>(Data[1]), 0x12);
 183   EXPECT_EQ(static_cast<uint8_t>(Data[2]), 0x78);
 184   EXPECT_EQ(static_cast<uint8_t>(Data[3]), 0x56);
 185 }
 186
 187 TEST(EndianStream, WriteFloatArrayLE) {
 188   SmallString<16> Data;
 189
 190   {
 191     raw_svector_ostream OS(Data);
 192     endian::Writer<little> LE(OS);
 193     LE.write<float>({12345.0f, 12346.0f});
 194   }
 195
 196   EXPECT_EQ(static_cast<uint8_t>(Data[0]), 0x00);
 197   EXPECT_EQ(static_cast<uint8_t>(Data[1]), 0xE4);
 198   EXPECT_EQ(static_cast<uint8_t>(Data[2]), 0x40);
 199   EXPECT_EQ(static_cast<uint8_t>(Data[3]), 0x46);
 200
 201   EXPECT_EQ(static_cast<uint8_t>(Data[4]), 0x00);
 202   EXPECT_EQ(static_cast<uint8_t>(Data[5]), 0xE8);
 203   EXPECT_EQ(static_cast<uint8_t>(Data[6]), 0x40);
 204   EXPECT_EQ(static_cast<uint8_t>(Data[7]), 0x46);
 205 }
 206
 207
 208 } // end anon namespace