| //===- ModuleUtilsTest.cpp - Unit tests for Module utility ----===// |
| // |
| // 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 "llvm/Transforms/Utils/ModuleUtils.h" |
| #include "llvm/ADT/StringRef.h" |
| #include "llvm/AsmParser/Parser.h" |
| #include "llvm/IR/LLVMContext.h" |
| #include "llvm/IR/Module.h" |
| #include "llvm/Support/SourceMgr.h" |
| #include "gtest/gtest.h" |
| |
| using namespace llvm; |
| |
| static std::unique_ptr<Module> parseIR(LLVMContext &C, const char *IR) { |
| SMDiagnostic Err; |
| std::unique_ptr<Module> Mod = parseAssemblyString(IR, Err, C); |
| if (!Mod) |
| Err.print("ModuleUtilsTest", errs()); |
| return Mod; |
| } |
| |
| static int getUsedListSize(Module &M, StringRef Name) { |
| auto *UsedList = M.getGlobalVariable(Name); |
| if (!UsedList) |
| return 0; |
| auto *UsedListBaseArrayType = cast<ArrayType>(UsedList->getValueType()); |
| return UsedListBaseArrayType->getNumElements(); |
| } |
| |
| TEST(ModuleUtils, AppendToUsedList1) { |
| LLVMContext C; |
| |
| std::unique_ptr<Module> M = parseIR( |
| C, R"(@x = addrspace(4) global [2 x i32] zeroinitializer, align 4)"); |
| SmallVector<GlobalValue *, 2> Globals; |
| for (auto &G : M->globals()) { |
| Globals.push_back(&G); |
| } |
| EXPECT_EQ(0, getUsedListSize(*M, "llvm.compiler.used")); |
| appendToCompilerUsed(*M, Globals); |
| EXPECT_EQ(1, getUsedListSize(*M, "llvm.compiler.used")); |
| |
| EXPECT_EQ(0, getUsedListSize(*M, "llvm.used")); |
| appendToUsed(*M, Globals); |
| EXPECT_EQ(1, getUsedListSize(*M, "llvm.used")); |
| } |
| |
| TEST(ModuleUtils, AppendToUsedList2) { |
| LLVMContext C; |
| |
| std::unique_ptr<Module> M = |
| parseIR(C, R"(@x = global [2 x i32] zeroinitializer, align 4)"); |
| SmallVector<GlobalValue *, 2> Globals; |
| for (auto &G : M->globals()) { |
| Globals.push_back(&G); |
| } |
| EXPECT_EQ(0, getUsedListSize(*M, "llvm.compiler.used")); |
| appendToCompilerUsed(*M, Globals); |
| EXPECT_EQ(1, getUsedListSize(*M, "llvm.compiler.used")); |
| |
| EXPECT_EQ(0, getUsedListSize(*M, "llvm.used")); |
| appendToUsed(*M, Globals); |
| EXPECT_EQ(1, getUsedListSize(*M, "llvm.used")); |
| } |