47 lines · cpp
1//===- MemoryBuiltinsTest.cpp - Tests for utilities in MemoryBuiltins.h ---===//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/Analysis/MemoryBuiltins.h"10#include "llvm/IR/Attributes.h"11#include "llvm/IR/Constants.h"12#include "llvm/IR/Function.h"13#include "llvm/IR/LLVMContext.h"14#include "llvm/IR/Module.h"15#include "gtest/gtest.h"16 17using namespace llvm;18 19namespace {20// allocsize should not imply that a function is a traditional allocation21// function (e.g. that can be optimized out/...); it just tells us how many22// bytes exist at the pointer handed back by the function.23TEST(AllocSize, AllocationBuiltinsTest) {24 LLVMContext Context;25 Module M("", Context);26 IntegerType *ArgTy = Type::getInt32Ty(Context);27 28 Function *AllocSizeFn = Function::Create(29 FunctionType::get(PointerType::getUnqual(Context), {ArgTy}, false),30 GlobalValue::ExternalLinkage, "F", &M);31 32 AllocSizeFn->addFnAttr(33 Attribute::getWithAllocSizeArgs(Context, 1, std::nullopt));34 35 // 100 is arbitrary.36 std::unique_ptr<CallInst> Caller(37 CallInst::Create(AllocSizeFn, {ConstantInt::get(ArgTy, 100)}));38 39 const TargetLibraryInfo *TLI = nullptr;40 EXPECT_FALSE(isAllocLikeFn(Caller.get(), TLI));41 42 // FIXME: We might be able to treat allocsize functions as general allocation43 // functions.44 EXPECT_FALSE(isAllocationFn(Caller.get(), TLI));45}46}47