replica
/
hanchenye-llvm-project
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add CodeGen support for the alloc_size attribute 

llvm-svn: 157483
This commit is contained in:
Nuno Lopes 2012-05-25 17:04:42 +00:00
parent debb881079
commit 147dcd8c33
2 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
clang/lib/CodeGen/CGCall.cpp
View File

@ -2087,6 +2087,23 @@ RValue CodeGenFunction::EmitCall(const CGFunctionInfo &CallInfo,
CS.setAttributes(Attrs);
CS.setCallingConv(static_cast<llvm::CallingConv::ID>(CallingConv));
// add metadata for __attribute__((alloc_size(foo)))
if (TargetDecl) {
if (const AllocSizeAttr* Attr = TargetDecl->getAttr<AllocSizeAttr>()) {
std::vector<llvm::Value*> Args;
llvm::IntegerType *Ty = llvm::IntegerType::getInt32Ty(getLLVMContext());
bool isMethod = isa<CXXMethodDecl>(TargetDecl);
for (AllocSizeAttr::args_iterator I = Attr->args_begin(),
E = Attr->args_end(); I != E; ++I) {
Args.push_back(llvm::ConstantInt::get(Ty, *I + isMethod));
}
llvm::MDNode *MD = llvm::MDNode::get(getLLVMContext(), Args);
CS.getInstruction()->setMetadata("alloc_size", MD);
}
}
// In ObjC ARC mode with no ObjC ARC exception safety, tell the ARC
// optimizer it can aggressively ignore unwind edges.
if (CGM.getLangOpts().ObjCAutoRefCount)

11
clang/test/CodeGen/alloc_size.c Normal file
View File

@ -0,0 +1,11 @@
// RUN: %clang_cc1 %s -emit-llvm -o - | FileCheck %s
void *my_recalloc(void *, unsigned, unsigned) __attribute__((alloc_size(2,3)));
// CHECK: @f
void* f() {
// CHECK: call i8* @my_recalloc{{.*}}, !alloc_size !0
return my_recalloc(0, 11, 27);
}
// CHECK: !0 = metadata !{i32 1, i32 2}