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
|
// Call handlers: flexible map of call targets to arbitrary handling code
//
// Each handler needs DEF_CALL_HANDLER and SETUP_CALL_HANDLER
typedef std::string (CppWriter::*CallHandler)(const CallInst*, std::string Name, unsigned NumArgs);
typedef std::map<std::string, CallHandler> CallHandlerMap;
CallHandlerMap *CallHandlers;
// Definitions
#define DEF_CALL_HANDLER(Ident, Code) \
std::string CH_##Ident(const CallInst *CI, std::string Name, unsigned NumArgs) { Code }
DEF_CALL_HANDLER(__default__, {
Type *RT = CI->getType();
std::string text = Name + "(";
for (unsigned i = 0; i < NumArgs; i++) {
text += getValueAsCastStr(CI->getArgOperand(i)); // FIXME: differentiate ffi calls
if (i < NumArgs - 1) text += ", ";
}
text += ")";
if (!RT->isVoidTy()) {
text = getAssign(getCppName(CI), RT) + getCast(text, RT);
}
return text;
})
DEF_CALL_HANDLER(llvm_nacl_atomic_store_i32, {
return "HEAP32[" + getValueAsStr(CI->getArgOperand(0)) + ">>2]=" + getValueAsStr(CI->getArgOperand(1));
})
DEF_CALL_HANDLER(llvm_memcpy_p0i8_p0i8_i32, {
Declares.insert("memcpy");
return CH___default__(CI, "_memcpy", 3) + "|0";
})
DEF_CALL_HANDLER(llvm_memset_p0i8_i32, {
Declares.insert("memmset");
return CH___default__(CI, "_memset", 3) + "|0";
})
DEF_CALL_HANDLER(llvm_memmove_p0i8_p0i8_i32, {
Declares.insert("memmove");
return CH___default__(CI, "_memmove", 3) + "|0";
})
// Setups
void setupCallHandlers() {
CallHandlers = new CallHandlerMap;
#define SETUP_CALL_HANDLER(Ident) \
(*CallHandlers)[std::string("_") + #Ident] = &CppWriter::CH_##Ident;
SETUP_CALL_HANDLER(__default__);
SETUP_CALL_HANDLER(llvm_nacl_atomic_store_i32);
SETUP_CALL_HANDLER(llvm_memcpy_p0i8_p0i8_i32);
SETUP_CALL_HANDLER(llvm_memset_p0i8_i32);
SETUP_CALL_HANDLER(llvm_memmove_p0i8_p0i8_i32);
}
std::string handleCall(const CallInst *CI) {
std::string Name = getCppName(CI->getCalledValue());
unsigned NumArgs = CI->getNumArgOperands();
CallHandlerMap::iterator CH = CallHandlers->find(Name);
if (CH == CallHandlers->end()) CH = CallHandlers->find("___default__");
return (this->*(CH->second))(CI, Name, NumArgs);
}
|