llvm-capstone/llvm/unittests/CodeGen/MachineBasicBlockTest.cpp
Bjorn Pettersson 28d418b16c [unittest][codegen] Avoid memory leak in recently added FindDebugLocTest unittest
Need to finalize the DIBuilder to avoid leak sanitizer errors
like this:

Direct leak of 48 byte(s) in 1 object(s) allocated from:
    #0 0x55c99ea1761d in operator new(unsigned long)
    #1 0x55c9a518ae49 in operator new
    #2 0x55c9a518ae49 in llvm::MDTuple::getImpl(...)
    #3 0x55c9a4f1b1ec in getTemporary
    #4 0x55c9a4f1b1ec in llvm::DIBuilder::createFunction(...)
2023-05-25 18:31:36 +02:00

109 lines
4.0 KiB
C++

//===- MachineBasicBlockTest.cpp ------------------------------------------===//
//
// 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/CodeGen/MachineBasicBlock.h"
#include "llvm/CodeGen/MachineFunction.h"
#include "llvm/CodeGen/MachineInstr.h"
#include "llvm/CodeGen/MachineInstrBuilder.h"
#include "llvm/CodeGen/MachineModuleInfo.h"
#include "llvm/CodeGen/TargetFrameLowering.h"
#include "llvm/CodeGen/TargetInstrInfo.h"
#include "llvm/CodeGen/TargetLowering.h"
#include "llvm/CodeGen/TargetSubtargetInfo.h"
#include "llvm/IR/DIBuilder.h"
#include "llvm/IR/DebugInfoMetadata.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/MC/TargetRegistry.h"
#include "llvm/Target/TargetMachine.h"
#include "gmock/gmock.h"
#include "gtest/gtest.h"
using namespace llvm;
namespace {
// Include helper functions to ease the manipulation of MachineFunctions.
#include "MFCommon.inc"
TEST(FindDebugLocTest, DifferentIterators) {
LLVMContext Ctx;
Module Mod("Module", Ctx);
auto MF = createMachineFunction(Ctx, Mod);
auto &MBB = *MF->CreateMachineBasicBlock();
// Create metadata: CU, subprogram, some blocks and an inline function
// scope.
DIBuilder DIB(Mod);
DIFile *OurFile = DIB.createFile("foo.c", "/bar");
DICompileUnit *OurCU =
DIB.createCompileUnit(dwarf::DW_LANG_C99, OurFile, "", false, "", 0);
auto OurSubT =
DIB.createSubroutineType(DIB.getOrCreateTypeArray(std::nullopt));
DISubprogram *OurFunc =
DIB.createFunction(OurCU, "bees", "", OurFile, 1, OurSubT, 1,
DINode::FlagZero, DISubprogram::SPFlagDefinition);
DebugLoc DL0;
DebugLoc DL1 = DILocation::get(Ctx, 1, 0, OurFunc);
DebugLoc DL2 = DILocation::get(Ctx, 2, 0, OurFunc);
DebugLoc DL3 = DILocation::get(Ctx, 3, 0, OurFunc);
// Test using and empty MBB.
EXPECT_EQ(DL0, MBB.findDebugLoc(MBB.instr_begin()));
EXPECT_EQ(DL0, MBB.findDebugLoc(MBB.instr_end()));
EXPECT_EQ(DL0, MBB.rfindDebugLoc(MBB.instr_rbegin()));
EXPECT_EQ(DL0, MBB.rfindDebugLoc(MBB.instr_rend()));
EXPECT_EQ(DL0, MBB.findPrevDebugLoc(MBB.instr_begin()));
EXPECT_EQ(DL0, MBB.findPrevDebugLoc(MBB.instr_end()));
EXPECT_EQ(DL0, MBB.rfindPrevDebugLoc(MBB.instr_rbegin()));
EXPECT_EQ(DL0, MBB.rfindPrevDebugLoc(MBB.instr_rend()));
// Insert two MIs with DebugLoc DL1 and DL3.
// Also add a DBG_VALUE with a different DebugLoc in between.
MCInstrDesc COPY = {TargetOpcode::COPY, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
MCInstrDesc DBG = {TargetOpcode::DBG_VALUE, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
auto MI3 = MF->CreateMachineInstr(COPY, DL3);
MBB.insert(MBB.begin(), MI3);
auto MI2 = MF->CreateMachineInstr(DBG, DL2);
MBB.insert(MBB.begin(), MI2);
auto MI1 = MF->CreateMachineInstr(COPY, DL1);
MBB.insert(MBB.begin(), MI1);
// Test using two MIs with a debug instruction in between.
EXPECT_EQ(DL1, MBB.findDebugLoc(MBB.instr_begin()));
EXPECT_EQ(DL1, MBB.findDebugLoc(MI1));
EXPECT_EQ(DL3, MBB.findDebugLoc(MI2));
EXPECT_EQ(DL3, MBB.findDebugLoc(MI3));
EXPECT_EQ(DL0, MBB.findDebugLoc(MBB.instr_end()));
EXPECT_EQ(DL1, MBB.rfindDebugLoc(MBB.instr_rend()));
EXPECT_EQ(DL1, MBB.rfindDebugLoc(MI1));
EXPECT_EQ(DL3, MBB.rfindDebugLoc(MI2));
EXPECT_EQ(DL3, MBB.rfindDebugLoc(MI3));
EXPECT_EQ(DL3, MBB.rfindDebugLoc(MBB.instr_rbegin()));
EXPECT_EQ(DL0, MBB.findPrevDebugLoc(MBB.instr_begin()));
EXPECT_EQ(DL0, MBB.findPrevDebugLoc(MI1));
EXPECT_EQ(DL1, MBB.findPrevDebugLoc(MI2));
EXPECT_EQ(DL1, MBB.findPrevDebugLoc(MI3));
EXPECT_EQ(DL3, MBB.findPrevDebugLoc(MBB.instr_end()));
EXPECT_EQ(DL0, MBB.rfindPrevDebugLoc(MBB.instr_rend()));
EXPECT_EQ(DL0, MBB.rfindPrevDebugLoc(MI1));
EXPECT_EQ(DL1, MBB.rfindPrevDebugLoc(MI2));
EXPECT_EQ(DL1, MBB.rfindPrevDebugLoc(MI3));
EXPECT_EQ(DL1, MBB.rfindPrevDebugLoc(MBB.instr_rbegin()));
// Finalize DIBuilder to avoid memory leaks.
DIB.finalize();
}
} // end namespace