1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
//===-- RaiseAsm.cpp ------------------------------------------------------===//
//
// The KLEE Symbolic Virtual Machine
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "Passes.h"
#include "klee/Config/Version.h"
#include "klee/Support/ErrorHandling.h"
#include "llvm/CodeGen/TargetLowering.h"
#include "llvm/CodeGen/TargetSubtargetInfo.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/IR/InlineAsm.h"
#include "llvm/IR/Instructions.h"
#include "llvm/IR/LLVMContext.h"
#include "llvm/Support/Host.h"
#include "llvm/Support/TargetRegistry.h"
#include "llvm/Target/TargetMachine.h"
using namespace llvm;
using namespace klee;
char RaiseAsmPass::ID = 0;
Function *RaiseAsmPass::getIntrinsic(llvm::Module &M, unsigned IID, Type **Tys,
unsigned NumTys) {
return Intrinsic::getDeclaration(&M, (llvm::Intrinsic::ID) IID,
llvm::ArrayRef<llvm::Type*>(Tys, NumTys));
}
// FIXME: This should just be implemented as a patch to
// X86TargetAsmInfo.cpp, then everyone will benefit.
bool RaiseAsmPass::runOnInstruction(Module &M, Instruction *I) {
// We can just raise inline assembler using calls
CallInst *ci = dyn_cast<CallInst>(I);
if (!ci)
return false;
#if LLVM_VERSION_CODE >= LLVM_VERSION(8, 0)
InlineAsm *ia = dyn_cast<InlineAsm>(ci->getCalledOperand());
#else
InlineAsm *ia = dyn_cast<InlineAsm>(ci->getCalledValue());
#endif
if (!ia)
return false;
// Try to use existing infrastructure
if (!TLI)
return false;
if (TLI->ExpandInlineAsm(ci))
return true;
if ((triple.getArch() == llvm::Triple::x86 ||
triple.getArch() == llvm::Triple::x86_64) &&
(triple.isOSLinux() || triple.isMacOSX() || triple.isOSFreeBSD())) {
if (ia->getAsmString() == "" && ia->hasSideEffects() &&
ia->getFunctionType()->getReturnType()->isVoidTy()) {
IRBuilder<> Builder(I);
Builder.CreateFence(llvm::AtomicOrdering::SequentiallyConsistent);
I->eraseFromParent();
return true;
}
}
return false;
}
bool RaiseAsmPass::runOnModule(Module &M) {
bool changed = false;
std::string Err;
// Use target triple from the module if possible.
std::string TargetTriple = M.getTargetTriple();
if (TargetTriple.empty())
TargetTriple = llvm::sys::getDefaultTargetTriple();
const Target *Target = TargetRegistry::lookupTarget(TargetTriple, Err);
TargetMachine * TM = 0;
if (Target == 0) {
klee_warning("Warning: unable to select target: %s", Err.c_str());
TLI = 0;
} else {
TM = Target->createTargetMachine(TargetTriple, "", "", TargetOptions(),
None);
TLI = TM->getSubtargetImpl(*(M.begin()))->getTargetLowering();
triple = llvm::Triple(TargetTriple);
}
for (Module::iterator fi = M.begin(), fe = M.end(); fi != fe; ++fi) {
for (Function::iterator bi = fi->begin(), be = fi->end(); bi != be; ++bi) {
for (BasicBlock::iterator ii = bi->begin(), ie = bi->end(); ii != ie;) {
Instruction *i = &*ii;
++ii;
changed |= runOnInstruction(M, i);
}
}
}
delete TM;
return changed;
}
|
