//===-- TestObjectFileELF.cpp -----------------------------------*- C++ -*-===//
//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "Plugins/ObjectFile/ELF/ObjectFileELF.h"
#include "Plugins/SymbolFile/Symtab/SymbolFileSymtab.h"
#include "TestingSupport/TestUtilities.h"
#include "lldb/Core/Module.h"
#include "lldb/Core/ModuleSpec.h"
#include "lldb/Core/Section.h"
#include "lldb/Host/FileSystem.h"
#include "lldb/Host/HostInfo.h"
#include "lldb/Utility/DataBufferHeap.h"
#include "llvm/ADT/Optional.h"
#include "llvm/Support/Compression.h"
#include "llvm/Support/FileUtilities.h"
#include "llvm/Support/Path.h"
#include "llvm/Support/Program.h"
#include "llvm/Support/raw_ostream.h"
#include "llvm/Testing/Support/Error.h"
#include "gtest/gtest.h"

using namespace lldb_private;
using namespace lldb;

class ObjectFileELFTest : public testing::Test {
public:
  void SetUp() override {
    FileSystem::Initialize();
    HostInfo::Initialize();
    ObjectFileELF::Initialize();
    SymbolFileSymtab::Initialize();
  }

  void TearDown() override {
    SymbolFileSymtab::Terminate();
    ObjectFileELF::Terminate();
    HostInfo::Terminate();
    FileSystem::Terminate();
  }

protected:
};

TEST_F(ObjectFileELFTest, SectionsResolveConsistently) {
  llvm::SmallString<128> obj;
  ASSERT_NO_ERROR(llvm::sys::fs::createTemporaryFile(
      "sections-resolve-consistently-%%%%%%", "obj", obj));
  llvm::FileRemover remover(obj);
  ASSERT_THAT_ERROR(
      ReadYAMLObjectFile("sections-resolve-consistently.yaml", obj),
      llvm::Succeeded());

  ModuleSpec spec{FileSpec(obj)};
  spec.GetSymbolFileSpec().SetFile(obj, FileSpec::Style::native);
  auto module_sp = std::make_shared<Module>(spec);
  SectionList *list = module_sp->GetSectionList();
  ASSERT_NE(nullptr, list);

  auto bss_sp = list->FindSectionByName(ConstString(".bss"));
  ASSERT_NE(nullptr, bss_sp);
  auto data_sp = list->FindSectionByName(ConstString(".data"));
  ASSERT_NE(nullptr, data_sp);
  auto text_sp = list->FindSectionByName(ConstString(".text"));
  ASSERT_NE(nullptr, text_sp);

  const Symbol *X = module_sp->FindFirstSymbolWithNameAndType(ConstString("X"),
                                                              eSymbolTypeAny);
  ASSERT_NE(nullptr, X);
  EXPECT_EQ(bss_sp, X->GetAddress().GetSection());

  const Symbol *Y = module_sp->FindFirstSymbolWithNameAndType(ConstString("Y"),
                                                              eSymbolTypeAny);
  ASSERT_NE(nullptr, Y);
  EXPECT_EQ(data_sp, Y->GetAddress().GetSection());

  const Symbol *start = module_sp->FindFirstSymbolWithNameAndType(
      ConstString("_start"), eSymbolTypeAny);
  ASSERT_NE(nullptr, start);
  EXPECT_EQ(text_sp, start->GetAddress().GetSection());
}

// Test that GetModuleSpecifications works on an "atypical" object file which
// has section headers right after the ELF header (instead of the more common
// layout where the section headers are at the very end of the object file).
//
// Test file generated with yaml2obj (@svn rev 324254) from the following input:
/*
--- !ELF
FileHeader:
  Class:           ELFCLASS64
  Data:            ELFDATA2LSB
  Type:            ET_EXEC
  Machine:         EM_X86_64
  Entry:           0x00000000004003D0
Sections:
  - Name:            .note.gnu.build-id
    Type:            SHT_NOTE
    Flags:           [ SHF_ALLOC ]
    Address:         0x0000000000400274
    AddressAlign:    0x0000000000000004
    Content:         040000001400000003000000474E55001B8A73AC238390E32A7FF4AC8EBE4D6A41ECF5C9
  - Name:            .text
    Type:            SHT_PROGBITS
    Flags:           [ SHF_ALLOC, SHF_EXECINSTR ]
    Address:         0x00000000004003D0
    AddressAlign:    0x0000000000000010
    Content:         DEADBEEFBAADF00D
...
*/
TEST_F(ObjectFileELFTest, GetModuleSpecifications_EarlySectionHeaders) {
  std::string SO = GetInputFilePath("early-section-headers.so");
  ModuleSpecList Specs;
  ASSERT_EQ(1u, ObjectFile::GetModuleSpecifications(FileSpec(SO), 0, 0, Specs));
  ModuleSpec Spec;
  ASSERT_TRUE(Specs.GetModuleSpecAtIndex(0, Spec)) ;
  UUID Uuid;
  Uuid.SetFromStringRef("1b8a73ac238390e32a7ff4ac8ebe4d6a41ecf5c9", 20);
  EXPECT_EQ(Spec.GetUUID(), Uuid);
}

static void CHECK_ABS32(uint8_t *bytes, uint32_t offset, uint32_t addend) {
  uint32_t res;
  memcpy(&res, reinterpret_cast<uint32_t *>(bytes + offset), sizeof(uint32_t));
  ASSERT_EQ(addend, res);
}

static void CHECK_ABS64(uint8_t *bytes, uint64_t offset, uint64_t addend) {
  uint64_t res;
  memcpy(&res, reinterpret_cast<uint64_t *>(bytes + offset), sizeof(uint64_t));
  ASSERT_EQ(addend, res);
}

TEST_F(ObjectFileELFTest, TestAARCH64Relocations) {
  llvm::SmallString<128> obj;
  ASSERT_NO_ERROR(llvm::sys::fs::createTemporaryFile(
      "debug-info-relocations-%%%%%%", "obj", obj));
  llvm::FileRemover remover(obj);
  ASSERT_THAT_ERROR(ReadYAMLObjectFile("debug-info-relocations.pcm.yaml", obj),
                    llvm::Succeeded());

  ModuleSpec spec{FileSpec(obj)};
  spec.GetSymbolFileSpec().SetFile(obj, FileSpec::Style::native);
  auto module_sp = std::make_shared<Module>(spec);

  auto objfile = static_cast<ObjectFileELF *>(module_sp->GetObjectFile());
  SectionList *section_list = objfile->GetSectionList();
  ASSERT_NE(nullptr, section_list);

  auto debug_info_sp =
      section_list->FindSectionByName(ConstString(".debug_info"));
  ASSERT_NE(nullptr, debug_info_sp);
  objfile->RelocateSection(debug_info_sp.get());

  DataExtractor data;
  // length of 0x10 is not needed but length 0x0 crashes
  objfile->GetData(0x00, 0x10, data);
  DataBufferSP &data_buffer_sp = data.GetSharedDataBuffer();
  uint8_t *bytes = data_buffer_sp->GetBytes();

  addr_t debug_info_offset = debug_info_sp->GetFileOffset();
  bytes += debug_info_offset;

  // Sanity check - The first byte from the yaml file is 0x47
  ASSERT_EQ(0x47, *bytes);

  // .rela.debug_info contains 9 relocations:
  // 7 R_AARCH64_ABS32 - 2 R_AARCH64_ABS64
  // None have a value. Four have addends.
  CHECK_ABS32(bytes, 0x6, 0);
  CHECK_ABS32(bytes, 0xC, 0);
  CHECK_ABS32(bytes, 0x12, 45);
  CHECK_ABS32(bytes, 0x16, 0);
  CHECK_ABS32(bytes, 0x1A, 55);
  CHECK_ABS64(bytes, 0x1E, 0);
  CHECK_ABS64(bytes, 0x2B, 0);
  CHECK_ABS32(bytes, 0x39, 73);
  CHECK_ABS32(bytes, 0x44, 75);
}