elf_reader_test.cc

// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// -*- mode: C++ -*-
//
// Copyright 2023 Google LLC
//
// Licensed under the Apache License v2.0 with LLVM Exceptions (the
// "License"); you may not use this file except in compliance with the
// License.  You may obtain a copy of the License at
//
//     https://llvm.org/LICENSE.txt
//
// 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.
//
// Author: Aleksei Vetrov

#include <string_view>

#include <catch2/catch.hpp>
#include "elf_loader.h"
#include "elf_reader.h"
#include "graph.h"

namespace Test {

using SymbolTable = stg::elf::internal::SymbolTable;
using SymbolTableEntry = stg::elf::SymbolTableEntry;

SymbolTableEntry MakeSymbol(std::string_view name) {
  return {
    .name = name,
    .value = 0,
    .size = 0,
    .symbol_type = SymbolTableEntry::SymbolType::OBJECT,
    .binding = SymbolTableEntry::Binding::GLOBAL,
    .visibility = SymbolTableEntry::Visibility::DEFAULT,
    .section_index = 0,
    .value_type = SymbolTableEntry::ValueType::RELATIVE_TO_SECTION,
  };
}


TEST_CASE("GetKsymtabSymbols") {
  const SymbolTable all_symbols = {
    MakeSymbol("foo"),
    MakeSymbol("__ksymtab_foo"),
    MakeSymbol("bar"),
  };
  const auto ksymtab = stg::elf::internal::GetKsymtabSymbols(all_symbols);
  REQUIRE(ksymtab.size() == 1);
  CHECK(*ksymtab.begin() == "foo");
}

}  // namespace Test