/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "read_elf.h"

#include <gtest/gtest.h>

#include <map>

#include <android-base/file.h>

#include "get_test_data.h"
#include "read_apk.h"
#include "test_util.h"
#include "utils.h"

#define ELF_NOTE_GNU "GNU"
#define NT_GNU_BUILD_ID 3

using namespace simpleperf;

TEST(read_elf, GetBuildIdFromNoteSection) {
  BuildId build_id;
  std::vector<char> data;
  // Fail to read build id for no data.
  ASSERT_FALSE(GetBuildIdFromNoteSection(data.data(), 0, &build_id));

  // Read build id from data starting from different alignment addresses.
  char build_id_data[20];
  for (int i = 0; i < 20; ++i) {
    build_id_data[i] = i;
  }
  BuildId expected_build_id(build_id_data, 20);
  data.resize(100, '\0');

  for (size_t alignment = 0; alignment <= 3; ++alignment) {
    char* start = data.data() + alignment;
    char* p = start;
    uint32_t type = NT_GNU_BUILD_ID;
    uint32_t namesz = 4;
    uint32_t descsz = 20;
    MoveToBinaryFormat(namesz, p);
    MoveToBinaryFormat(descsz, p);
    MoveToBinaryFormat(type, p);
    MoveToBinaryFormat(ELF_NOTE_GNU, 4, p);
    MoveToBinaryFormat(build_id_data, 20, p);
    ASSERT_TRUE(GetBuildIdFromNoteSection(start, p - start, &build_id));
    ASSERT_TRUE(build_id == expected_build_id);
  }
}

TEST(read_elf, GetBuildIdFromElfFile) {
  BuildId build_id;
  ElfStatus status;
  auto elf = ElfFile::Open(GetTestData(ELF_FILE), &status);
  ASSERT_EQ(status, ElfStatus::NO_ERROR);
  ASSERT_EQ(ElfStatus::NO_ERROR, elf->GetBuildId(&build_id));
  ASSERT_EQ(build_id, BuildId(elf_file_build_id));
}

TEST(read_elf, GetBuildIdFromEmbeddedElfFile) {
  BuildId build_id;
  ElfStatus status;
  std::string path = GetUrlInApk(APK_FILE, NATIVELIB_IN_APK);
  auto elf = ElfFile::Open(GetTestData(path), &status);
  ASSERT_EQ(status, ElfStatus::NO_ERROR);
  ASSERT_EQ(ElfStatus::NO_ERROR, elf->GetBuildId(&build_id));
  ASSERT_EQ(build_id, native_lib_build_id);
}

void ParseSymbol(const ElfFileSymbol& symbol, std::map<std::string, ElfFileSymbol>* symbols) {
  (*symbols)[symbol.name] = symbol;
}

static void CheckGlobalVariableSymbols(const std::map<std::string, ElfFileSymbol>& symbols) {
  auto pos = symbols.find("GlobalVar");
  ASSERT_NE(pos, symbols.end());
  ASSERT_FALSE(pos->second.is_func);
}

static void CheckFunctionSymbols(const std::map<std::string, ElfFileSymbol>& symbols) {
  auto pos = symbols.find("GlobalFunc");
  ASSERT_NE(pos, symbols.end());
  ASSERT_TRUE(pos->second.is_func);
  ASSERT_TRUE(pos->second.is_in_text_section);
}

void CheckElfFileSymbols(const std::map<std::string, ElfFileSymbol>& symbols) {
  CheckGlobalVariableSymbols(symbols);
  CheckFunctionSymbols(symbols);
}

TEST(read_elf, parse_symbols_from_elf_file_with_correct_build_id) {
  std::map<std::string, ElfFileSymbol> symbols;
  ElfStatus status;
  auto elf = ElfFile::Open(GetTestData(ELF_FILE), &elf_file_build_id, &status);
  ASSERT_EQ(ElfStatus::NO_ERROR, status);
  ASSERT_EQ(ElfStatus::NO_ERROR,
            elf->ParseSymbols(std::bind(ParseSymbol, std::placeholders::_1, &symbols)));
  CheckElfFileSymbols(symbols);
}

TEST(read_elf, parse_symbols_from_elf_file_without_build_id) {
  std::map<std::string, ElfFileSymbol> symbols;
  ElfStatus status;
  // Test no build_id.
  auto elf = ElfFile::Open(GetTestData(ELF_FILE), &status);
  ASSERT_EQ(ElfStatus::NO_ERROR, status);
  ASSERT_EQ(ElfStatus::NO_ERROR,
            elf->ParseSymbols(std::bind(ParseSymbol, std::placeholders::_1, &symbols)));
  CheckElfFileSymbols(symbols);

  // Test empty build id.
  symbols.clear();
  BuildId build_id;
  elf = ElfFile::Open(GetTestData(ELF_FILE), &build_id, &status);
  ASSERT_EQ(ElfStatus::NO_ERROR, status);
  ASSERT_EQ(ElfStatus::NO_ERROR,
            elf->ParseSymbols(std::bind(ParseSymbol, std::placeholders::_1, &symbols)));
  CheckElfFileSymbols(symbols);
}

TEST(read_elf, parse_symbols_from_elf_file_with_wrong_build_id) {
  BuildId build_id("01010101010101010101");
  std::map<std::string, ElfFileSymbol> symbols;
  ElfStatus status;
  auto elf = ElfFile::Open(GetTestData(ELF_FILE), &build_id, &status);
  ASSERT_EQ(ElfStatus::BUILD_ID_MISMATCH, status);
}

TEST(read_elf, ParseSymbolsFromEmbeddedElfFile) {
  std::map<std::string, ElfFileSymbol> symbols;
  ElfStatus status;
  std::string path = GetUrlInApk(APK_FILE, NATIVELIB_IN_APK);
  auto elf = ElfFile::Open(GetTestData(path), &native_lib_build_id, &status);
  ASSERT_EQ(status, ElfStatus::NO_ERROR);
  ASSERT_EQ(ElfStatus::NO_SYMBOL_TABLE,
            elf->ParseSymbols(std::bind(ParseSymbol, std::placeholders::_1, &symbols)));
  CheckElfFileSymbols(symbols);
}

TEST(read_elf, ParseSymbolFromMiniDebugInfoElfFile) {
  std::map<std::string, ElfFileSymbol> symbols;
  ElfStatus status;
  auto elf = ElfFile::Open(GetTestData(ELF_FILE_WITH_MINI_DEBUG_INFO), &status);
  ASSERT_EQ(ElfStatus::NO_ERROR, status);
  ASSERT_EQ(ElfStatus::NO_ERROR,
            elf->ParseSymbols(std::bind(ParseSymbol, std::placeholders::_1, &symbols)));
  CheckFunctionSymbols(symbols);
}

TEST(read_elf, arm_mapping_symbol) {
  ASSERT_TRUE(IsArmMappingSymbol("$a"));
  ASSERT_FALSE(IsArmMappingSymbol("$b"));
  ASSERT_TRUE(IsArmMappingSymbol("$a.anything"));
  ASSERT_FALSE(IsArmMappingSymbol("$a_no_dot"));
}

TEST(read_elf, ElfFile_Open) {
  auto IsValidElfPath = [](const std::string& path) {
    ElfStatus status;
    ElfFile::Open(path, &status);
    return status;
  };
  ASSERT_NE(ElfStatus::NO_ERROR, IsValidElfPath("/dev/zero"));
  TemporaryFile tmp_file;
  ASSERT_EQ(ElfStatus::READ_FAILED, IsValidElfPath(tmp_file.path));
  ASSERT_TRUE(android::base::WriteStringToFile("wrong format for elf", tmp_file.path));
  ASSERT_EQ(ElfStatus::FILE_MALFORMED, IsValidElfPath(tmp_file.path));
  ASSERT_EQ(ElfStatus::NO_ERROR, IsValidElfPath(GetTestData(ELF_FILE)));
}

TEST(read_elf, check_symbol_for_plt_section) {
  std::map<std::string, ElfFileSymbol> symbols;
  ElfStatus status;
  auto elf = ElfFile::Open(GetTestData(ELF_FILE), &status);
  ASSERT_EQ(ElfStatus::NO_ERROR, status);
  ASSERT_EQ(ElfStatus::NO_ERROR,
            elf->ParseSymbols(std::bind(ParseSymbol, std::placeholders::_1, &symbols)));
  ASSERT_NE(symbols.find("@plt"), symbols.end());
}

TEST(read_elf, read_elf_with_broken_section_table) {
  std::string elf_path = GetTestData("libsgmainso-6.4.36.so");
  std::map<std::string, ElfFileSymbol> symbols;
  ElfStatus status;
  auto elf = ElfFile::Open(elf_path, &status);
  ASSERT_EQ(ElfStatus::NO_ERROR, status);
  ASSERT_EQ(ElfStatus::NO_SYMBOL_TABLE,
            elf->ParseSymbols(std::bind(ParseSymbol, std::placeholders::_1, &symbols)));

  BuildId build_id;
  ASSERT_EQ(ElfStatus::NO_BUILD_ID, elf->GetBuildId(&build_id));

  uint64_t file_offset_of_min_vaddr;
  uint64_t min_vaddr = elf->ReadMinExecutableVaddr(&file_offset_of_min_vaddr);
  ASSERT_EQ(min_vaddr, 0u);
  ASSERT_EQ(file_offset_of_min_vaddr, 0u);
}

TEST(read_elf, ReadMinExecutableVaddr) {
  ElfStatus status;
  auto elf = ElfFile::Open(GetTestData("libc.so"), &status);
  ASSERT_EQ(status, ElfStatus::NO_ERROR);
  uint64_t file_offset_of_min_vaddr;
  uint64_t min_vaddr = elf->ReadMinExecutableVaddr(&file_offset_of_min_vaddr);
  ASSERT_EQ(min_vaddr, 0x29000u);
  ASSERT_EQ(file_offset_of_min_vaddr, 0x29000u);
}

TEST(read_elf, NoUndefinedSymbol) {
  // Check if we read undefined symbols (like dlerror) from libc.so.
  bool has_dlerror = false;
  auto parse_symbol = [&](const ElfFileSymbol& symbol) {
    if (symbol.name == "dlerror") {
      has_dlerror = true;
    }
  };

  ElfStatus status;
  auto elf = ElfFile::Open(GetTestData("libc.so"), &status);
  ASSERT_EQ(status, ElfStatus::NO_ERROR);
  ASSERT_EQ(ElfStatus::NO_ERROR, elf->ParseSymbols(parse_symbol));
  ASSERT_FALSE(has_dlerror);
}

TEST(read_elf, VaddrToOff) {
  auto elf = ElfFile::Open(GetTestData(ELF_FILE));
  ASSERT_TRUE(elf != nullptr);
  uint64_t off;
  ASSERT_TRUE(elf->VaddrToOff(0x400200, &off));
  ASSERT_EQ(off, 0x200);
  ASSERT_FALSE(elf->VaddrToOff(0x300200, &off));
  ASSERT_FALSE(elf->VaddrToOff(0x420000, &off));
}

TEST(read_elf, GetSectionHeader) {
  auto elf = ElfFile::Open(GetTestData(ELF_FILE));
  ASSERT_TRUE(elf != nullptr);
  std::vector<ElfSection> sections = elf->GetSectionHeader();
  ASSERT_EQ(sections.size(), 30);
  ASSERT_EQ(sections[13].name, ".text");
  ASSERT_EQ(sections[13].vaddr, 0x400400);
  ASSERT_EQ(sections[13].file_offset, 0x400);
  ASSERT_EQ(sections[13].size, 0x1b2);
}