109 lines · cpp
1//===- MachineBasicBlockTest.cpp ------------------------------------------===//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-exception6//7//===----------------------------------------------------------------------===//8 9#include "llvm/CodeGen/MachineBasicBlock.h"10#include "llvm/CodeGen/CodeGenTargetMachineImpl.h"11#include "llvm/CodeGen/MachineFunction.h"12#include "llvm/CodeGen/MachineInstr.h"13#include "llvm/CodeGen/MachineInstrBuilder.h"14#include "llvm/CodeGen/MachineModuleInfo.h"15#include "llvm/CodeGen/TargetFrameLowering.h"16#include "llvm/CodeGen/TargetInstrInfo.h"17#include "llvm/CodeGen/TargetLowering.h"18#include "llvm/CodeGen/TargetSubtargetInfo.h"19#include "llvm/IR/DIBuilder.h"20#include "llvm/IR/DebugInfoMetadata.h"21#include "llvm/IR/IRBuilder.h"22#include "llvm/IR/Module.h"23#include "llvm/MC/TargetRegistry.h"24#include "gmock/gmock.h"25#include "gtest/gtest.h"26 27using namespace llvm;28 29namespace {30// Include helper functions to ease the manipulation of MachineFunctions.31#include "MFCommon.inc"32 33TEST(FindDebugLocTest, DifferentIterators) {34 LLVMContext Ctx;35 Module Mod("Module", Ctx);36 auto MF = createMachineFunction(Ctx, Mod);37 auto &MBB = *MF->CreateMachineBasicBlock();38 39 // Create metadata: CU, subprogram, some blocks and an inline function40 // scope.41 DIBuilder DIB(Mod);42 DIFile *OurFile = DIB.createFile("foo.c", "/bar");43 DICompileUnit *OurCU = DIB.createCompileUnit(44 DISourceLanguageName(dwarf::DW_LANG_C99), OurFile, "", false, "", 0);45 auto OurSubT = DIB.createSubroutineType(DIB.getOrCreateTypeArray({}));46 DISubprogram *OurFunc =47 DIB.createFunction(OurCU, "bees", "", OurFile, 1, OurSubT, 1,48 DINode::FlagZero, DISubprogram::SPFlagDefinition);49 50 DebugLoc DL0;51 DebugLoc DL1 = DILocation::get(Ctx, 1, 0, OurFunc);52 DebugLoc DL2 = DILocation::get(Ctx, 2, 0, OurFunc);53 DebugLoc DL3 = DILocation::get(Ctx, 3, 0, OurFunc);54 55 // Test using and empty MBB.56 EXPECT_EQ(DL0, MBB.findDebugLoc(MBB.instr_begin()));57 EXPECT_EQ(DL0, MBB.findDebugLoc(MBB.instr_end()));58 59 EXPECT_EQ(DL0, MBB.rfindDebugLoc(MBB.instr_rbegin()));60 EXPECT_EQ(DL0, MBB.rfindDebugLoc(MBB.instr_rend()));61 62 EXPECT_EQ(DL0, MBB.findPrevDebugLoc(MBB.instr_begin()));63 EXPECT_EQ(DL0, MBB.findPrevDebugLoc(MBB.instr_end()));64 65 EXPECT_EQ(DL0, MBB.rfindPrevDebugLoc(MBB.instr_rbegin()));66 EXPECT_EQ(DL0, MBB.rfindPrevDebugLoc(MBB.instr_rend()));67 68 // Insert two MIs with DebugLoc DL1 and DL3.69 // Also add a DBG_VALUE with a different DebugLoc in between.70 MCInstrDesc COPY = {TargetOpcode::COPY, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};71 MCInstrDesc DBG = {TargetOpcode::DBG_VALUE, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};72 auto MI3 = MF->CreateMachineInstr(COPY, DL3);73 MBB.insert(MBB.begin(), MI3);74 auto MI2 = MF->CreateMachineInstr(DBG, DL2);75 MBB.insert(MBB.begin(), MI2);76 auto MI1 = MF->CreateMachineInstr(COPY, DL1);77 MBB.insert(MBB.begin(), MI1);78 79 // Test using two MIs with a debug instruction in between.80 EXPECT_EQ(DL1, MBB.findDebugLoc(MBB.instr_begin()));81 EXPECT_EQ(DL1, MBB.findDebugLoc(MI1));82 EXPECT_EQ(DL3, MBB.findDebugLoc(MI2));83 EXPECT_EQ(DL3, MBB.findDebugLoc(MI3));84 EXPECT_EQ(DL0, MBB.findDebugLoc(MBB.instr_end()));85 86 EXPECT_EQ(DL1, MBB.rfindDebugLoc(MBB.instr_rend()));87 EXPECT_EQ(DL1, MBB.rfindDebugLoc(MI1));88 EXPECT_EQ(DL3, MBB.rfindDebugLoc(MI2));89 EXPECT_EQ(DL3, MBB.rfindDebugLoc(MI3));90 EXPECT_EQ(DL3, MBB.rfindDebugLoc(MBB.instr_rbegin()));91 92 EXPECT_EQ(DL0, MBB.findPrevDebugLoc(MBB.instr_begin()));93 EXPECT_EQ(DL0, MBB.findPrevDebugLoc(MI1));94 EXPECT_EQ(DL1, MBB.findPrevDebugLoc(MI2));95 EXPECT_EQ(DL1, MBB.findPrevDebugLoc(MI3));96 EXPECT_EQ(DL3, MBB.findPrevDebugLoc(MBB.instr_end()));97 98 EXPECT_EQ(DL0, MBB.rfindPrevDebugLoc(MBB.instr_rend()));99 EXPECT_EQ(DL0, MBB.rfindPrevDebugLoc(MI1));100 EXPECT_EQ(DL1, MBB.rfindPrevDebugLoc(MI2));101 EXPECT_EQ(DL1, MBB.rfindPrevDebugLoc(MI3));102 EXPECT_EQ(DL1, MBB.rfindPrevDebugLoc(MBB.instr_rbegin()));103 104 // Finalize DIBuilder to avoid memory leaks.105 DIB.finalize();106}107 108} // end namespace109