lldb/unittests/Symbol/MangledTest.cpp

   1 //===-- SymbolTest.cpp ----------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "lldb/Core/Mangled.h"
  10 #include "lldb/Core/DataFileCache.h"
  11 #include "lldb/Utility/DataEncoder.h"
  12 #include "lldb/Utility/DataExtractor.h"
  13
  14 #include "gtest/gtest.h"
  15
  16 using namespace lldb;
  17 using namespace lldb_private;
  18
  19 static void EncodeDecode(const Mangled &object, ByteOrder byte_order) {
  20   const uint8_t addr_size = 8;
  21   DataEncoder file(byte_order, addr_size);
  22   DataEncoder strtab_encoder(byte_order, addr_size);
  23   ConstStringTable const_strtab;
  24
  25   object.Encode(file, const_strtab);
  26
  27   llvm::ArrayRef<uint8_t> bytes = file.GetData();
  28   DataExtractor data(bytes.data(), bytes.size(), byte_order, addr_size);
  29
  30   const_strtab.Encode(strtab_encoder);
  31   llvm::ArrayRef<uint8_t> strtab_bytes = strtab_encoder.GetData();
  32   DataExtractor strtab_data(strtab_bytes.data(), strtab_bytes.size(),
  33                             byte_order, addr_size);
  34   StringTableReader strtab_reader;
  35   offset_t strtab_data_offset = 0;
  36   ASSERT_EQ(strtab_reader.Decode(strtab_data, &strtab_data_offset), true);
  37
  38   Mangled decoded_object;
  39   offset_t data_offset = 0;
  40   decoded_object.Decode(data, &data_offset, strtab_reader);
  41   EXPECT_EQ(object, decoded_object);
  42 }
  43
  44 static void EncodeDecode(const Mangled &object) {
  45   EncodeDecode(object, eByteOrderLittle);
  46   EncodeDecode(object, eByteOrderBig);
  47 }
  48
  49 TEST(MangledTest, EncodeDecodeMangled) {
  50   Mangled mangled;
  51   // Test encoding and decoding an empty mangled object.
  52   EncodeDecode(mangled);
  53
  54   // Test encoding a mangled object that hasn't demangled its name yet.
  55   mangled.SetMangledName(ConstString("_Z3fooi"));
  56   EncodeDecode(mangled);
  57
  58   // Test encoding a mangled object that has demangled its name by computing it.
  59   mangled.GetDemangledName();
  60   // EncodeDecode(mangled);
  61
  62   // Test encoding a mangled object that has just a demangled name
  63   mangled.SetMangledName(ConstString());
  64   mangled.SetDemangledName(ConstString("hello"));
  65   EncodeDecode(mangled);
  66
  67   // Test encoding a mangled name that has both a mangled and demangled name
  68   // that are not mangled/demangled counterparts of each other.
  69   mangled.SetMangledName(ConstString("world"));
  70   EncodeDecode(mangled);
  71 }