tools/relocation_packer/src/run_length_encoder_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 "run_length_encoder.h"
   6
   7 #include <vector>
   8 #include "elf.h"
   9 #include "elf_traits.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 namespace {
  13
  14 void AddRelocation(ELF::Addr addr, std::vector<ELF::Rel>* relocations) {
  15   ELF::Rel relocation;
  16   relocation.r_offset = addr;
  17   relocation.r_info = ELF_R_INFO(0, ELF::kRelativeRelocationCode);
  18   relocations->push_back(relocation);
  19 }
  20
  21 bool CheckRelocation(ELF::Addr addr, const ELF::Rel& relocation) {
  22   return relocation.r_offset == addr &&
  23       ELF_R_SYM(relocation.r_info) == 0 &&
  24       ELF_R_TYPE(relocation.r_info) == ELF::kRelativeRelocationCode;
  25 }
  26
  27 }  // namespace
  28
  29 namespace relocation_packer {
  30
  31 TEST(RunLength, Encode) {
  32   std::vector<ELF::Rel> relocations;
  33   std::vector<ELF::Xword> packed;
  34
  35   RelocationRunLengthCodec codec;
  36
  37   packed.clear();
  38   codec.Encode(relocations, &packed);
  39
  40   EXPECT_EQ(0, packed.size());
  41
  42   // Add one relocation (insufficient data to encode).
  43   AddRelocation(0xf00d0000, &relocations);
  44
  45   packed.clear();
  46   codec.Encode(relocations, &packed);
  47
  48   EXPECT_EQ(0, packed.size());
  49
  50   // Add a second relocation, 4 byte delta (minimum data to encode).
  51   AddRelocation(0xf00d0004, &relocations);
  52
  53   packed.clear();
  54   codec.Encode(relocations, &packed);
  55
  56   EXPECT_EQ(4, packed.size());
  57   // One count-delta pair present.
  58   EXPECT_EQ(1, packed[0]);
  59   // Initial relocation.
  60   EXPECT_EQ(0xf00d0000, packed[1]);
  61   // Run of a single relocation, 4 byte delta.
  62   EXPECT_EQ(1, packed[2]);
  63   EXPECT_EQ(4, packed[3]);
  64
  65   // Add a third relocation, 4 byte delta.
  66   AddRelocation(0xf00d0008, &relocations);
  67
  68   // Add three more relocations, 8 byte deltas.
  69   AddRelocation(0xf00d0010, &relocations);
  70   AddRelocation(0xf00d0018, &relocations);
  71   AddRelocation(0xf00d0020, &relocations);
  72
  73   packed.clear();
  74   codec.Encode(relocations, &packed);
  75
  76   EXPECT_EQ(6, packed.size());
  77   // Two count-delta pairs present.
  78   EXPECT_EQ(2, packed[0]);
  79   // Initial relocation.
  80   EXPECT_EQ(0xf00d0000, packed[1]);
  81   // Run of two relocations, 4 byte deltas.
  82   EXPECT_EQ(2, packed[2]);
  83   EXPECT_EQ(4, packed[3]);
  84   // Run of three relocations, 8 byte deltas.
  85   EXPECT_EQ(3, packed[4]);
  86   EXPECT_EQ(8, packed[5]);
  87 }
  88
  89 TEST(RunLength, Decode) {
  90   std::vector<ELF::Xword> packed;
  91   std::vector<ELF::Rel> relocations;
  92
  93   RelocationRunLengthCodec codec;
  94   codec.Decode(packed, &relocations);
  95
  96   EXPECT_EQ(0, relocations.size());
  97
  98   // Two count-delta pairs.
  99   packed.push_back(2);
 100   // Initial relocation.
 101   packed.push_back(0xc0de0000);
 102   // Run of two relocations, 4 byte deltas.
 103   packed.push_back(2);
 104   packed.push_back(4);
 105   // Run of three relocations, 8 byte deltas.
 106   packed.push_back(3);
 107   packed.push_back(8);
 108
 109   relocations.clear();
 110   codec.Decode(packed, &relocations);
 111
 112   EXPECT_EQ(6, relocations.size());
 113   // Initial relocation.
 114   EXPECT_TRUE(CheckRelocation(0xc0de0000, relocations[0]));
 115   // Two relocations, 4 byte deltas.
 116   EXPECT_TRUE(CheckRelocation(0xc0de0004, relocations[1]));
 117   EXPECT_TRUE(CheckRelocation(0xc0de0008, relocations[2]));
 118   // Three relocations, 8 byte deltas.
 119   EXPECT_TRUE(CheckRelocation(0xc0de0010, relocations[3]));
 120   EXPECT_TRUE(CheckRelocation(0xc0de0018, relocations[4]));
 121   EXPECT_TRUE(CheckRelocation(0xc0de0020, relocations[5]));
 122 }
 123
 124 }  // namespace relocation_packer