llvm/unittests/IR/AsmWriterTest.cpp

   1 //===- llvm/unittest/IR/AsmWriter.cpp - AsmWriter tests -------------------===//
   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 #include "llvm/BinaryFormat/Dwarf.h"
   9 #include "llvm/IR/DebugInfoMetadata.h"
  10 #include "llvm/IR/Function.h"
  11 #include "llvm/IR/IRBuilder.h"
  12 #include "llvm/IR/LLVMContext.h"
  13 #include "llvm/IR/MDBuilder.h"
  14 #include "llvm/IR/Module.h"
  15 #include "gmock/gmock.h"
  16 #include "gtest/gtest.h"
  17
  18 using namespace llvm;
  19 using ::testing::HasSubstr;
  20
  21 namespace {
  22
  23 TEST(AsmWriterTest, DebugPrintDetachedInstruction) {
  24
  25   // PR24852: Ensure that an instruction can be printed even when it
  26   // has metadata attached but no parent.
  27   LLVMContext Ctx;
  28   auto Ty = Type::getInt32Ty(Ctx);
  29   auto Poison = PoisonValue::get(Ty);
  30   std::unique_ptr<BinaryOperator> Add(BinaryOperator::CreateAdd(Poison, Poison));
  31   Add->setMetadata(
  32       "", MDNode::get(Ctx, {ConstantAsMetadata::get(ConstantInt::get(Ty, 1))}));
  33   std::string S;
  34   raw_string_ostream OS(S);
  35   Add->print(OS);
  36   EXPECT_THAT(S, HasSubstr("<badref> = add i32 poison, poison, !<empty"));
  37 }
  38
  39 TEST(AsmWriterTest, DebugPrintDetachedArgument) {
  40   LLVMContext Ctx;
  41   auto Ty = Type::getInt32Ty(Ctx);
  42   auto Arg = new Argument(Ty);
  43
  44   std::string S;
  45   raw_string_ostream OS(S);
  46   Arg->print(OS);
  47   EXPECT_EQ(S, "i32 <badref>");
  48   delete Arg;
  49 }
  50
  51 TEST(AsmWriterTest, DumpDIExpression) {
  52   LLVMContext Ctx;
  53   uint64_t Ops[] = {
  54     dwarf::DW_OP_constu, 4,
  55     dwarf::DW_OP_minus,
  56     dwarf::DW_OP_deref,
  57   };
  58   DIExpression *Expr = DIExpression::get(Ctx, Ops);
  59   std::string S;
  60   raw_string_ostream OS(S);
  61   Expr->print(OS);
  62   EXPECT_EQ("!DIExpression(DW_OP_constu, 4, DW_OP_minus, DW_OP_deref)", S);
  63 }
  64
  65 TEST(AsmWriterTest, PrintAddrspaceWithNullOperand) {
  66   LLVMContext Ctx;
  67   Module M("test module", Ctx);
  68   SmallVector<Type *, 3> FArgTypes;
  69   FArgTypes.push_back(Type::getInt64Ty(Ctx));
  70   FunctionType *FTy = FunctionType::get(Type::getVoidTy(Ctx), FArgTypes, false);
  71   Function *F = Function::Create(FTy, Function::ExternalLinkage, "", &M);
  72   Argument *Arg0 = F->getArg(0);
  73   Value *Args[] = {Arg0};
  74   std::unique_ptr<CallInst> Call(CallInst::Create(F, Args));
  75   // This will make Call's operand null.
  76   Call->dropAllReferences();
  77
  78   std::string S;
  79   raw_string_ostream OS(S);
  80   Call->print(OS);
  81   EXPECT_THAT(S, HasSubstr("<cannot get addrspace!>"));
  82 }
  83
  84 TEST(AsmWriterTest, PrintNullOperandBundle) {
  85   LLVMContext C;
  86   Type *Int32Ty = Type::getInt32Ty(C);
  87   FunctionType *FnTy = FunctionType::get(Int32Ty, Int32Ty, /*isVarArg=*/false);
  88   Value *Callee = Constant::getNullValue(PointerType::getUnqual(C));
  89   Value *Args[] = {ConstantInt::get(Int32Ty, 42)};
  90   std::unique_ptr<BasicBlock> NormalDest(BasicBlock::Create(C));
  91   std::unique_ptr<BasicBlock> UnwindDest(BasicBlock::Create(C));
  92   OperandBundleDef Bundle("bundle", UndefValue::get(Int32Ty));
  93   std::unique_ptr<InvokeInst> Invoke(
  94       InvokeInst::Create(FnTy, Callee, NormalDest.get(), UnwindDest.get(), Args,
  95                          Bundle, "result"));
  96   // Makes the operand bundle null.
  97   Invoke->dropAllReferences();
  98   Invoke->setNormalDest(NormalDest.get());
  99   Invoke->setUnwindDest(UnwindDest.get());
 100
 101   std::string S;
 102   raw_string_ostream OS(S);
 103   Invoke->print(OS);
 104   EXPECT_THAT(S, HasSubstr("<null operand bundle!>"));
 105 }
 106 }