wangtao
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[msan] Instrument bswap intrinsic. 

llvm-svn: 169383
This commit is contained in:
Evgeniy Stepanov 2012-12-05 14:39:55 +00:00
parent bc09a7ea85
commit 8b51bab495
2 changed files with 37 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
llvm/lib/Transforms/Instrumentation/MemorySanitizer.cpp
View File

@ -1101,6 +1101,25 @@ struct MemorySanitizerVisitor : public InstVisitor<MemorySanitizerVisitor> {
VAHelper->visitVACopyInst(I); VAHelper->visitVACopyInst(I);
} }
void handleBswap(IntrinsicInst &I) {
IRBuilder<> IRB(&I);
Value *Op = I.getArgOperand(0);
Type *OpType = Op->getType();
Function *BswapFunc = Intrinsic::getDeclaration(
F.getParent(), Intrinsic::bswap, ArrayRef<Type*>(&OpType, 1));
setShadow(&I, IRB.CreateCall(BswapFunc, getShadow(Op)));
setOrigin(&I, getOrigin(Op));
}
void visitIntrinsicInst(IntrinsicInst &I) {
switch (I.getIntrinsicID()) {
case llvm::Intrinsic::bswap:
handleBswap(I); break;
default:
visitInstruction(I); break;
}
}
void visitCallSite(CallSite CS) { void visitCallSite(CallSite CS) {
Instruction &I = *CS.getInstruction(); Instruction &I = *CS.getInstruction();
assert((CS.isCall() || CS.isInvoke()) && "Unknown type of CallSite"); assert((CS.isCall() || CS.isInvoke()) && "Unknown type of CallSite");
@ -1120,12 +1139,8 @@ struct MemorySanitizerVisitor : public InstVisitor<MemorySanitizerVisitor> {
// will get propagated to a void RetVal. // will get propagated to a void RetVal.
if (Call->isTailCall() && Call->getType() != Call->getParent()->getType()) if (Call->isTailCall() && Call->getType() != Call->getParent()->getType())
Call->setTailCall(false); Call->setTailCall(false);
if (isa<IntrinsicInst>(&I)) {
// All intrinsics we care about are handled in corresponding visit* assert(!isa<IntrinsicInst>(&I) && "intrinsics are handled elsewhere");
// methods. Add checks for the arguments, mark retval as clean.
visitInstruction(I);
return;
}
} }
IRBuilder<> IRB(&I); IRBuilder<> IRB(&I);
unsigned ArgOffset = 0; unsigned ArgOffset = 0;

16
llvm/test/Instrumentation/MemorySanitizer/msan_basic.ll
View File

@ -351,3 +351,19 @@ define <4 x i32> @ShuffleVector(<4 x i32> %vec, <4 x i32> %vec1) {
; CHECK-NOT: call void @__msan_warning ; CHECK-NOT: call void @__msan_warning
; CHECK: shufflevector ; CHECK: shufflevector
; CHECK: ret <4 x i32> ; CHECK: ret <4 x i32>
; Test bswap intrinsic instrumentation
define i32 @BSwap(i32 %x) nounwind uwtable readnone {
%y = tail call i32 @llvm.bswap.i32(i32 %x)
ret i32 %y
}
declare i32 @llvm.bswap.i32(i32) nounwind readnone
; CHECK: @BSwap
; CHECK-NOT: call void @__msan_warning
; CHECK: @llvm.bswap.i32
; CHECK-NOT: call void @__msan_warning
; CHECK: @llvm.bswap.i32
; CHECK-NOT: call void @__msan_warning
; CHECK: ret i32