tools/relocation_packer/src/leb128_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "leb128.h"
   6
   7 #include <vector>
   8 #include "testing/gtest/include/gtest/gtest.h"
   9
  10 namespace relocation_packer {
  11
  12 TEST(Leb128, Encoder) {
  13   std::vector<ELF::Xword> values;
  14   values.push_back(624485);
  15   values.push_back(0);
  16   values.push_back(1);
  17   values.push_back(127);
  18   values.push_back(128);
  19
  20   Leb128Encoder encoder;
  21   encoder.EnqueueAll(values);
  22
  23   encoder.Enqueue(4294967295);
  24   encoder.Enqueue(18446744073709551615ul);
  25
  26   std::vector<uint8_t> encoding;
  27   encoder.GetEncoding(&encoding);
  28
  29   EXPECT_EQ(23, encoding.size());
  30   // 624485
  31   EXPECT_EQ(0xe5, encoding[0]);
  32   EXPECT_EQ(0x8e, encoding[1]);
  33   EXPECT_EQ(0x26, encoding[2]);
  34   // 0
  35   EXPECT_EQ(0x00, encoding[3]);
  36   // 1
  37   EXPECT_EQ(0x01, encoding[4]);
  38   // 127
  39   EXPECT_EQ(0x7f, encoding[5]);
  40   // 128
  41   EXPECT_EQ(0x80, encoding[6]);
  42   EXPECT_EQ(0x01, encoding[7]);
  43   // 4294967295
  44   EXPECT_EQ(0xff, encoding[8]);
  45   EXPECT_EQ(0xff, encoding[9]);
  46   EXPECT_EQ(0xff, encoding[10]);
  47   EXPECT_EQ(0xff, encoding[11]);
  48   EXPECT_EQ(0x0f, encoding[12]);
  49   // 18446744073709551615
  50   EXPECT_EQ(0xff, encoding[13]);
  51   EXPECT_EQ(0xff, encoding[14]);
  52   EXPECT_EQ(0xff, encoding[15]);
  53   EXPECT_EQ(0xff, encoding[16]);
  54   EXPECT_EQ(0xff, encoding[17]);
  55   EXPECT_EQ(0xff, encoding[18]);
  56   EXPECT_EQ(0xff, encoding[19]);
  57   EXPECT_EQ(0xff, encoding[20]);
  58   EXPECT_EQ(0xff, encoding[21]);
  59   EXPECT_EQ(0x01, encoding[22]);
  60 }
  61
  62 TEST(Leb128, Decoder) {
  63   std::vector<uint8_t> encoding;
  64   // 624485
  65   encoding.push_back(0xe5);
  66   encoding.push_back(0x8e);
  67   encoding.push_back(0x26);
  68   // 0
  69   encoding.push_back(0x00);
  70   // 1
  71   encoding.push_back(0x01);
  72   // 127
  73   encoding.push_back(0x7f);
  74   // 128
  75   encoding.push_back(0x80);
  76   encoding.push_back(0x01);
  77   // 4294967295
  78   encoding.push_back(0xff);
  79   encoding.push_back(0xff);
  80   encoding.push_back(0xff);
  81   encoding.push_back(0xff);
  82   encoding.push_back(0x0f);
  83   // 18446744073709551615
  84   encoding.push_back(0xff);
  85   encoding.push_back(0xff);
  86   encoding.push_back(0xff);
  87   encoding.push_back(0xff);
  88   encoding.push_back(0xff);
  89   encoding.push_back(0xff);
  90   encoding.push_back(0xff);
  91   encoding.push_back(0xff);
  92   encoding.push_back(0xff);
  93   encoding.push_back(0x01);
  94
  95   Leb128Decoder decoder(encoding);
  96
  97   EXPECT_EQ(624485, decoder.Dequeue());
  98
  99   std::vector<ELF::Xword> dequeued;
 100   decoder.DequeueAll(&dequeued);
 101
 102   EXPECT_EQ(6, dequeued.size());
 103   EXPECT_EQ(0, dequeued[0]);
 104   EXPECT_EQ(1, dequeued[1]);
 105   EXPECT_EQ(127, dequeued[2]);
 106   EXPECT_EQ(128, dequeued[3]);
 107   EXPECT_EQ(4294967295, dequeued[4]);
 108   EXPECT_EQ(18446744073709551615ul, dequeued[5]);
 109 }
 110
 111 }  // namespace relocation_packer