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-exception
6//
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 allocation
21// function (e.g. that can be optimized out/...); it just tells us how many
22// 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(C&: Context);
27
28 Function *AllocSizeFn = Function::Create(
29 Ty: FunctionType::get(Result: PointerType::getUnqual(C&: Context), Params: {ArgTy}, isVarArg: false),
30 Linkage: GlobalValue::ExternalLinkage, N: "F", M: &M);
31
32 AllocSizeFn->addFnAttr(
33 Attr: Attribute::getWithAllocSizeArgs(Context, ElemSizeArg: 1, NumElemsArg: std::nullopt));
34
35 // 100 is arbitrary.
36 std::unique_ptr<CallInst> Caller(
37 CallInst::Create(Func: AllocSizeFn, Args: {ConstantInt::get(Ty: ArgTy, V: 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 allocation
43 // functions.
44 EXPECT_FALSE(isAllocationFn(Caller.get(), TLI));
45}
46}
47

source code of llvm/unittests/Analysis/MemoryBuiltinsTest.cpp