#include "llvm/LLVMContext.h"
#include "llvm/DerivedTypes.h"
#include "llvm/ExecutionEngine/ExecutionEngine.h"
#include "llvm/ExecutionEngine/JIT.h"
#include "llvm/Module.h"
#include "llvm/PassManager.h"
#include "llvm/Intrinsics.h"
#include "llvm/Analysis/Verifier.h"
#include "llvm/Target/TargetData.h"
#include "llvm/Target/TargetSelect.h"
#include "llvm/Target/TargetOptions.h"
#include "llvm/Transforms/Scalar.h"
#include "llvm/Support/IRBuilder.h"
#include "llvm/Support/Dwarf.h"
#include <cstdio>
#include <string>
#include <sstream>
#include <map>
#include <vector>
#include <stdexcept>
#ifndef USE_GLOBAL_STR_CONSTS
#define USE_GLOBAL_STR_CONSTS true
#endif
extern "C" {
typedef enum {
_URC_NO_REASON = 0,
_URC_FOREIGN_EXCEPTION_CAUGHT = 1,
_URC_FATAL_PHASE2_ERROR = 2,
_URC_FATAL_PHASE1_ERROR = 3,
_URC_NORMAL_STOP = 4,
_URC_END_OF_STACK = 5,
_URC_HANDLER_FOUND = 6,
_URC_INSTALL_CONTEXT = 7,
_URC_CONTINUE_UNWIND = 8
} _Unwind_Reason_Code;
typedef enum {
_UA_SEARCH_PHASE = 1,
_UA_CLEANUP_PHASE = 2,
_UA_HANDLER_FRAME = 4,
_UA_FORCE_UNWIND = 8,
_UA_END_OF_STACK = 16
} _Unwind_Action;
struct _Unwind_Exception;
typedef void (*_Unwind_Exception_Cleanup_Fn) (_Unwind_Reason_Code,
struct _Unwind_Exception *);
struct _Unwind_Exception {
uint64_t exception_class;
_Unwind_Exception_Cleanup_Fn exception_cleanup;
uintptr_t private_1;
uintptr_t private_2;
} __attribute__((__aligned__));
struct _Unwind_Context;
typedef struct _Unwind_Context* _Unwind_Context_t;
extern const uint8_t* _Unwind_GetLanguageSpecificData (_Unwind_Context_t c);
extern uintptr_t _Unwind_GetGR (_Unwind_Context_t c, int i);
extern void _Unwind_SetGR (_Unwind_Context_t c, int i, uintptr_t n);
extern void _Unwind_SetIP (_Unwind_Context_t, uintptr_t new_value);
extern uintptr_t _Unwind_GetIP (_Unwind_Context_t context);
extern uintptr_t _Unwind_GetRegionStart (_Unwind_Context_t context);
}
struct OurExceptionType_t {
int type;
};
struct OurBaseException_t {
struct OurExceptionType_t type;
struct _Unwind_Exception unwindException;
};
typedef struct OurBaseException_t OurException;
typedef struct _Unwind_Exception OurUnwindException;
static std::map<std::string, llvm::Value*> namedValues;
int64_t ourBaseFromUnwindOffset;
const unsigned char ourBaseExcpClassChars[] =
{'o', 'b', 'j', '\0', 'b', 'a', 's', '\0'};
static uint64_t ourBaseExceptionClass = 0;
static std::vector<std::string> ourTypeInfoNames;
static std::map<int, std::string> ourTypeInfoNamesIndex;
static llvm::StructType* ourTypeInfoType;
static llvm::StructType* ourExceptionType;
static llvm::StructType* ourUnwindExceptionType;
static llvm::ConstantInt* ourExceptionNotThrownState;
static llvm::ConstantInt* ourExceptionThrownState;
static llvm::ConstantInt* ourExceptionCaughtState;
typedef std::vector<std::string> ArgNames;
typedef std::vector<const llvm::Type*> ArgTypes;
llvm::Function *createFunction(llvm::Module& module,
const llvm::Type* retType,
const ArgTypes& theArgTypes,
const ArgNames& theArgNames,
const std::string& functName,
llvm::GlobalValue::LinkageTypes linkage,
bool declarationOnly,
bool isVarArg) {
llvm::FunctionType* functType = llvm::FunctionType::get(retType,
theArgTypes,
isVarArg);
llvm::Function* ret = llvm::Function::Create(functType,
linkage,
functName,
&module);
if (!ret || declarationOnly)
return(ret);
namedValues.clear();
unsigned i = 0;
for (llvm::Function::arg_iterator argIndex = ret->arg_begin();
i != theArgNames.size();
++argIndex, ++i) {
argIndex->setName(theArgNames[i]);
namedValues[theArgNames[i]] = argIndex;
}
return(ret);
}
static llvm::AllocaInst *createEntryBlockAlloca(llvm::Function& function,
const std::string &varName,
const llvm::Type* type,
llvm::Constant* initWith = NULL) {
llvm::BasicBlock& block = function.getEntryBlock();
llvm::IRBuilder<> tmp(&block, block.begin());
llvm::AllocaInst* ret = tmp.CreateAlloca(type, 0, varName.c_str());
if (initWith)
tmp.CreateStore(initWith, ret);
return(ret);
}
extern "C" {
void print32Int(int intToPrint, const char* format) {
if (format) {
fprintf(stderr, format, intToPrint);
}
else {
fprintf(stderr, "::print32Int(...):NULL arg.\n");
}
}
void print64Int(long int intToPrint, const char* format) {
if (format) {
fprintf(stderr, format, intToPrint);
}
else {
fprintf(stderr, "::print64Int(...):NULL arg.\n");
}
}
void printStr(char* toPrint) {
if (toPrint) {
fprintf(stderr, "%s", toPrint);
}
else {
fprintf(stderr, "::printStr(...):NULL arg.\n");
}
}
void deleteOurException(OurUnwindException* expToDelete) {
#ifdef DEBUG
fprintf(stderr,
"deleteOurException(...).\n");
#endif
if (expToDelete &&
(expToDelete->exception_class == ourBaseExceptionClass)) {
free(((char*) expToDelete) + ourBaseFromUnwindOffset);
}
}
void deleteFromUnwindOurException(_Unwind_Reason_Code reason,
OurUnwindException* expToDelete) {
#ifdef DEBUG
fprintf(stderr,
"deleteFromUnwindOurException(...).\n");
#endif
deleteOurException(expToDelete);
}
OurUnwindException* createOurException(int type) {
size_t size = sizeof(OurException);
OurException* ret = (OurException*) memset(malloc(size), 0, size);
(ret->type).type = type;
(ret->unwindException).exception_class = ourBaseExceptionClass;
(ret->unwindException).exception_cleanup = deleteFromUnwindOurException;
return(&(ret->unwindException));
}
static uintptr_t readULEB128(const uint8_t** data) {
uintptr_t result = 0;
uintptr_t shift = 0;
unsigned char byte;
const uint8_t* p = *data;
do {
byte = *p++;
result |= (byte & 0x7f) << shift;
shift += 7;
}
while (byte & 0x80);
*data = p;
return result;
}
static uintptr_t readSLEB128(const uint8_t** data) {
uintptr_t result = 0;
uintptr_t shift = 0;
unsigned char byte;
const uint8_t* p = *data;
do {
byte = *p++;
result |= (byte & 0x7f) << shift;
shift += 7;
}
while (byte & 0x80);
*data = p;
if ((byte & 0x40) && (shift < (sizeof(result) << 3))) {
result |= (~0 << shift);
}
return result;
}
static uintptr_t readEncodedPointer(const uint8_t** data, uint8_t encoding) {
uintptr_t result = 0;
const uint8_t* p = *data;
if (encoding == llvm::dwarf::DW_EH_PE_omit)
return(result);
switch (encoding & 0x0F) {
case llvm::dwarf::DW_EH_PE_absptr:
result = *((uintptr_t*)p);
p += sizeof(uintptr_t);
break;
case llvm::dwarf::DW_EH_PE_uleb128:
result = readULEB128(&p);
break;
case llvm::dwarf::DW_EH_PE_sleb128:
result = readSLEB128(&p);
break;
case llvm::dwarf::DW_EH_PE_udata2:
result = *((uint16_t*)p);
p += sizeof(uint16_t);
break;
case llvm::dwarf::DW_EH_PE_udata4:
result = *((uint32_t*)p);
p += sizeof(uint32_t);
break;
case llvm::dwarf::DW_EH_PE_udata8:
result = *((uint64_t*)p);
p += sizeof(uint64_t);
break;
case llvm::dwarf::DW_EH_PE_sdata2:
result = *((int16_t*)p);
p += sizeof(int16_t);
break;
case llvm::dwarf::DW_EH_PE_sdata4:
result = *((int32_t*)p);
p += sizeof(int32_t);
break;
case llvm::dwarf::DW_EH_PE_sdata8:
result = *((int64_t*)p);
p += sizeof(int64_t);
break;
default:
abort();
break;
}
switch (encoding & 0x70) {
case llvm::dwarf::DW_EH_PE_absptr:
break;
case llvm::dwarf::DW_EH_PE_pcrel:
result += (uintptr_t)(*data);
break;
case llvm::dwarf::DW_EH_PE_textrel:
case llvm::dwarf::DW_EH_PE_datarel:
case llvm::dwarf::DW_EH_PE_funcrel:
case llvm::dwarf::DW_EH_PE_aligned:
default:
abort();
break;
}
if (encoding & llvm::dwarf::DW_EH_PE_indirect) {
result = *((uintptr_t*)result);
}
*data = p;
return result;
}
static bool handleActionValue(int64_t *resultAction,
struct OurExceptionType_t **classInfo,
uintptr_t actionEntry,
uint64_t exceptionClass,
struct _Unwind_Exception *exceptionObject) {
bool ret = false;
if (!resultAction ||
!exceptionObject ||
(exceptionClass != ourBaseExceptionClass))
return(ret);
struct OurBaseException_t* excp = (struct OurBaseException_t*)
(((char*) exceptionObject) + ourBaseFromUnwindOffset);
struct OurExceptionType_t *excpType = &(excp->type);
int type = excpType->type;
#ifdef DEBUG
fprintf(stderr,
"handleActionValue(...): exceptionObject = <%p>, "
"excp = <%p>.\n",
exceptionObject,
excp);
#endif
const uint8_t *actionPos = (uint8_t*) actionEntry,
*tempActionPos;
int64_t typeOffset = 0,
actionOffset;
for (int i = 0; true; ++i) {
typeOffset = readSLEB128(&actionPos);
tempActionPos = actionPos;
actionOffset = readSLEB128(&tempActionPos);
#ifdef DEBUG
fprintf(stderr,
"handleActionValue(...):typeOffset: <%lld>, "
"actionOffset: <%lld>.\n",
typeOffset,
actionOffset);
#endif
assert((typeOffset >= 0) &&
"handleActionValue(...):filters are not supported.");
if ((typeOffset > 0) &&
(type == (classInfo[-typeOffset])->type)) {
#ifdef DEBUG
fprintf(stderr,
"handleActionValue(...):actionValue <%d> found.\n",
i);
#endif
*resultAction = i + 1;
ret = true;
break;
}
#ifdef DEBUG
fprintf(stderr,
"handleActionValue(...):actionValue not found.\n");
#endif
if (!actionOffset)
break;
actionPos += actionOffset;
}
return(ret);
}
static _Unwind_Reason_Code handleLsda(int version,
const uint8_t* lsda,
_Unwind_Action actions,
uint64_t exceptionClass,
struct _Unwind_Exception* exceptionObject,
_Unwind_Context_t context) {
_Unwind_Reason_Code ret = _URC_CONTINUE_UNWIND;
if (!lsda)
return(ret);
#ifdef DEBUG
fprintf(stderr,
"handleLsda(...):lsda is non-zero.\n");
#endif
uintptr_t pc = _Unwind_GetIP(context)-1;
uintptr_t funcStart = _Unwind_GetRegionStart(context);
uintptr_t pcOffset = pc - funcStart;
struct OurExceptionType_t** classInfo = NULL;
uint8_t lpStartEncoding = *lsda++;
if (lpStartEncoding != llvm::dwarf::DW_EH_PE_omit) {
readEncodedPointer(&lsda, lpStartEncoding);
}
uint8_t ttypeEncoding = *lsda++;
uintptr_t classInfoOffset;
if (ttypeEncoding != llvm::dwarf::DW_EH_PE_omit) {
classInfoOffset = readULEB128(&lsda);
classInfo = (struct OurExceptionType_t**) (lsda + classInfoOffset);
}
uint8_t callSiteEncoding = *lsda++;
uint32_t callSiteTableLength = readULEB128(&lsda);
const uint8_t* callSiteTableStart = lsda;
const uint8_t* callSiteTableEnd = callSiteTableStart +
callSiteTableLength;
const uint8_t* actionTableStart = callSiteTableEnd;
const uint8_t* callSitePtr = callSiteTableStart;
bool foreignException = false;
while (callSitePtr < callSiteTableEnd) {
uintptr_t start = readEncodedPointer(&callSitePtr,
callSiteEncoding);
uintptr_t length = readEncodedPointer(&callSitePtr,
callSiteEncoding);
uintptr_t landingPad = readEncodedPointer(&callSitePtr,
callSiteEncoding);
uintptr_t actionEntry = readULEB128(&callSitePtr);
if (exceptionClass != ourBaseExceptionClass) {
actionEntry = 0;
foreignException = true;
}
if (landingPad == 0) {
#ifdef DEBUG
fprintf(stderr,
"handleLsda(...): No landing pad found.\n");
#endif
continue; }
if (actionEntry) {
actionEntry += ((uintptr_t) actionTableStart) - 1;
}
else {
#ifdef DEBUG
fprintf(stderr,
"handleLsda(...):No action table found.\n");
#endif
}
bool exceptionMatched = false;
if ((start <= pcOffset) && (pcOffset < (start + length))) {
#ifdef DEBUG
fprintf(stderr,
"handleLsda(...): Landing pad found.\n");
#endif
int64_t actionValue = 0;
if (actionEntry) {
exceptionMatched = handleActionValue
(
&actionValue,
classInfo,
actionEntry,
exceptionClass,
exceptionObject
);
}
if (!(actions & _UA_SEARCH_PHASE)) {
#ifdef DEBUG
fprintf(stderr,
"handleLsda(...): installed landing pad "
"context.\n");
#endif
_Unwind_SetGR(context,
__builtin_eh_return_data_regno(0),
(uintptr_t)exceptionObject);
if (!actionEntry || !exceptionMatched) {
_Unwind_SetGR(context,
__builtin_eh_return_data_regno(1),
0);
}
else {
_Unwind_SetGR(context,
__builtin_eh_return_data_regno(1),
actionValue);
}
_Unwind_SetIP(context, funcStart + landingPad);
ret = _URC_INSTALL_CONTEXT;
}
else if (exceptionMatched) {
#ifdef DEBUG
fprintf(stderr,
"handleLsda(...): setting handler found.\n");
#endif
ret = _URC_HANDLER_FOUND;
}
else {
#ifdef DEBUG
fprintf(stderr,
"handleLsda(...): cleanup handler found.\n");
#endif
}
break;
}
}
return(ret);
}
_Unwind_Reason_Code ourPersonality(int version,
_Unwind_Action actions,
uint64_t exceptionClass,
struct _Unwind_Exception* exceptionObject,
_Unwind_Context_t context) {
#ifdef DEBUG
fprintf(stderr,
"We are in ourPersonality(...):actions is <%d>.\n",
actions);
if (actions & _UA_SEARCH_PHASE) {
fprintf(stderr, "ourPersonality(...):In search phase.\n");
}
else {
fprintf(stderr, "ourPersonality(...):In non-search phase.\n");
}
#endif
const uint8_t* lsda = (uint8_t*)
_Unwind_GetLanguageSpecificData(context);
#ifdef DEBUG
fprintf(stderr,
"ourPersonality(...):lsda = <%p>.\n",
lsda);
#endif
return(handleLsda(version,
lsda,
actions,
exceptionClass,
exceptionObject,
context));
}
uint64_t genClass(const unsigned char classChars[], size_t classCharsSize)
{
uint64_t ret = classChars[0];
for (unsigned i = 1; i < classCharsSize; ++i) {
ret <<= 8;
ret += classChars[i];
}
return(ret);
}
}
void generateStringPrint(llvm::LLVMContext& context,
llvm::Module& module,
llvm::IRBuilder<>& builder,
std::string toPrint,
bool useGlobal = true) {
llvm::Function *printFunct = module.getFunction("printStr");
llvm::Value *stringVar;
llvm::Constant* stringConstant =
llvm::ConstantArray::get(context, toPrint);
if (useGlobal) {
stringVar =
new llvm::GlobalVariable(module,
stringConstant->getType(),
true,
llvm::GlobalValue::LinkerPrivateLinkage,
stringConstant,
"");
}
else {
stringVar = builder.CreateAlloca(stringConstant->getType());
builder.CreateStore(stringConstant, stringVar);
}
llvm::Value* cast =
builder.CreatePointerCast(stringVar,
builder.getInt8Ty()->getPointerTo());
builder.CreateCall(printFunct, cast);
}
void generateIntegerPrint(llvm::LLVMContext& context,
llvm::Module& module,
llvm::IRBuilder<>& builder,
llvm::Function& printFunct,
llvm::Value& toPrint,
std::string format,
bool useGlobal = true) {
llvm::Constant *stringConstant = llvm::ConstantArray::get(context, format);
llvm::Value *stringVar;
if (useGlobal) {
stringVar =
new llvm::GlobalVariable(module,
stringConstant->getType(),
true,
llvm::GlobalValue::LinkerPrivateLinkage,
stringConstant,
"");
}
else {
stringVar = builder.CreateAlloca(stringConstant->getType());
builder.CreateStore(stringConstant, stringVar);
}
llvm::Value* cast =
builder.CreateBitCast(stringVar,
builder.getInt8Ty()->getPointerTo());
builder.CreateCall2(&printFunct, &toPrint, cast);
}
static llvm::BasicBlock* createFinallyBlock(llvm::LLVMContext& context,
llvm::Module& module,
llvm::IRBuilder<>& builder,
llvm::Function& toAddTo,
std::string& blockName,
std::string& functionId,
llvm::BasicBlock& terminatorBlock,
llvm::BasicBlock& unwindResumeBlock,
llvm::Value** exceptionCaughtFlag,
llvm::Value** exceptionStorage) {
assert(exceptionCaughtFlag &&
"ExceptionDemo::createFinallyBlock(...):exceptionCaughtFlag "
"is NULL");
assert(exceptionStorage &&
"ExceptionDemo::createFinallyBlock(...):exceptionStorage "
"is NULL");
*exceptionCaughtFlag =
createEntryBlockAlloca(toAddTo,
"exceptionCaught",
ourExceptionNotThrownState->getType(),
ourExceptionNotThrownState);
const llvm::PointerType* exceptionStorageType =
builder.getInt8Ty()->getPointerTo();
*exceptionStorage =
createEntryBlockAlloca(toAddTo,
"exceptionStorage",
exceptionStorageType,
llvm::ConstantPointerNull::get(
exceptionStorageType));
llvm::BasicBlock *ret = llvm::BasicBlock::Create(context,
blockName,
&toAddTo);
builder.SetInsertPoint(ret);
std::ostringstream bufferToPrint;
bufferToPrint << "Gen: Executing finally block "
<< blockName
<< " in "
<< functionId
<< std::endl;
generateStringPrint(context,
module,
builder,
bufferToPrint.str(),
USE_GLOBAL_STR_CONSTS);
llvm::SwitchInst* theSwitch =
builder.CreateSwitch(builder.CreateLoad(*exceptionCaughtFlag),
&terminatorBlock,
2);
theSwitch->addCase(ourExceptionCaughtState, &terminatorBlock);
theSwitch->addCase(ourExceptionThrownState, &unwindResumeBlock);
return(ret);
}
static llvm::BasicBlock* createCatchBlock(llvm::LLVMContext& context,
llvm::Module& module,
llvm::IRBuilder<>& builder,
llvm::Function& toAddTo,
std::string& blockName,
std::string& functionId,
llvm::BasicBlock& terminatorBlock,
llvm::Value& exceptionCaughtFlag) {
llvm::BasicBlock *ret = llvm::BasicBlock::Create(context,
blockName,
&toAddTo);
builder.SetInsertPoint(ret);
std::ostringstream bufferToPrint;
bufferToPrint << "Gen: Executing catch block "
<< blockName
<< " in "
<< functionId
<< std::endl;
generateStringPrint(context,
module,
builder,
bufferToPrint.str(),
USE_GLOBAL_STR_CONSTS);
builder.CreateStore(ourExceptionCaughtState, &exceptionCaughtFlag);
builder.CreateBr(&terminatorBlock);
return(ret);
}
static
llvm::Function* createCatchWrappedInvokeFunction(llvm::Module& module,
llvm::IRBuilder<>& builder,
llvm::FunctionPassManager& fpm,
llvm::Function& toInvoke,
std::string ourId,
unsigned numExceptionsToCatch,
unsigned exceptionTypesToCatch[]) {
llvm::LLVMContext& context = module.getContext();
llvm::Function *toPrint32Int = module.getFunction("print32Int");
ArgTypes argTypes;
argTypes.push_back(builder.getInt32Ty());
ArgNames argNames;
argNames.push_back("exceptTypeToThrow");
llvm::Function* ret = createFunction(module,
builder.getVoidTy(),
argTypes,
argNames,
ourId,
llvm::Function::ExternalLinkage,
false,
false);
llvm::BasicBlock *entryBlock = llvm::BasicBlock::Create(context,
"entry",
ret);
llvm::BasicBlock *normalBlock = llvm::BasicBlock::Create(context,
"normal",
ret);
llvm::BasicBlock *exceptionBlock =
llvm::BasicBlock::Create(context, "exception", ret);
llvm::BasicBlock *exceptionRouteBlock =
llvm::BasicBlock::Create(context, "exceptionRoute", ret);
llvm::BasicBlock *externalExceptionBlock =
llvm::BasicBlock::Create(context, "externalException", ret);
llvm::BasicBlock *unwindResumeBlock =
llvm::BasicBlock::Create(context, "unwindResume", ret);
llvm::BasicBlock *endBlock =
llvm::BasicBlock::Create(context, "end", ret);
std::string nextName;
std::vector<llvm::BasicBlock*> catchBlocks(numExceptionsToCatch);
llvm::Value* exceptionCaughtFlag = NULL;
llvm::Value* exceptionStorage = NULL;
llvm::BasicBlock* finallyBlock = createFinallyBlock(context,
module,
builder,
*ret,
nextName = "finally",
ourId,
*endBlock,
*unwindResumeBlock,
&exceptionCaughtFlag,
&exceptionStorage);
for (unsigned i = 0; i < numExceptionsToCatch; ++i) {
nextName = ourTypeInfoNames[exceptionTypesToCatch[i]];
catchBlocks[i] = createCatchBlock(context,
module,
builder,
*ret,
nextName,
ourId,
*finallyBlock,
*exceptionCaughtFlag);
}
builder.SetInsertPoint(entryBlock);
std::vector<llvm::Value*> args;
args.push_back(namedValues["exceptTypeToThrow"]);
builder.CreateInvoke(&toInvoke,
normalBlock,
exceptionBlock,
args.begin(),
args.end());
builder.SetInsertPoint(endBlock);
generateStringPrint(context,
module,
builder,
"Gen: In end block: exiting in " + ourId + ".\n",
USE_GLOBAL_STR_CONSTS);
llvm::Function *deleteOurException =
module.getFunction("deleteOurException");
builder.CreateCall(deleteOurException,
builder.CreateLoad(exceptionStorage));
builder.CreateRetVoid();
builder.SetInsertPoint(normalBlock);
generateStringPrint(context,
module,
builder,
"Gen: No exception in " + ourId + "!\n",
USE_GLOBAL_STR_CONSTS);
builder.CreateBr(finallyBlock);
builder.SetInsertPoint(unwindResumeBlock);
llvm::Function *resumeOurException =
module.getFunction("_Unwind_Resume");
builder.CreateCall(resumeOurException,
builder.CreateLoad(exceptionStorage));
builder.CreateUnreachable();
builder.SetInsertPoint(exceptionBlock);
llvm::Function *ehException = module.getFunction("llvm.eh.exception");
llvm::Value* unwindException = builder.CreateCall(ehException);
builder.CreateStore(unwindException, exceptionStorage);
builder.CreateStore(ourExceptionThrownState, exceptionCaughtFlag);
llvm::Function *personality = module.getFunction("ourPersonality");
llvm::Value* functPtr =
builder.CreatePointerCast(personality,
builder.getInt8Ty()->getPointerTo());
args.clear();
args.push_back(unwindException);
args.push_back(functPtr);
for (unsigned i = 0; i < numExceptionsToCatch; ++i) {
args.push_back(
module.getGlobalVariable(
ourTypeInfoNames[exceptionTypesToCatch[i]]));
}
args.push_back(llvm::ConstantInt::get(builder.getInt32Ty(), 0));
llvm::Function *ehSelector = module.getFunction("llvm.eh.selector");
llvm::Value* retTypeInfoIndex = builder.CreateCall(ehSelector,
args.begin(),
args.end());
llvm::Value* unwindExceptionClass =
builder.CreateLoad(
builder.CreateStructGEP(
builder.CreatePointerCast(
unwindException,
ourUnwindExceptionType->getPointerTo()),
0));
builder.CreateCondBr(
builder.CreateICmpEQ(unwindExceptionClass,
llvm::ConstantInt::get(builder.getInt64Ty(),
ourBaseExceptionClass)),
exceptionRouteBlock,
externalExceptionBlock);
builder.SetInsertPoint(externalExceptionBlock);
generateStringPrint(context,
module,
builder,
"Gen: Foreign exception received.\n",
USE_GLOBAL_STR_CONSTS);
builder.CreateBr(finallyBlock);
builder.SetInsertPoint(exceptionRouteBlock);
llvm::Value* typeInfoThrown =
builder.CreatePointerCast(
builder.CreateConstGEP1_64(unwindException,
ourBaseFromUnwindOffset),
ourExceptionType->getPointerTo());
typeInfoThrown = builder.CreateStructGEP(typeInfoThrown, 0);
llvm::Value* typeInfoThrownType =
builder.CreateStructGEP(typeInfoThrown, 0);
generateIntegerPrint(context,
module,
builder,
*toPrint32Int,
*(builder.CreateLoad(typeInfoThrownType)),
"Gen: Exception type <%d> received (stack unwound) "
" in " +
ourId +
".\n",
USE_GLOBAL_STR_CONSTS);
llvm::SwitchInst* switchToCatchBlock =
builder.CreateSwitch(retTypeInfoIndex,
finallyBlock,
numExceptionsToCatch);
unsigned nextTypeToCatch;
for (unsigned i = 1; i <= numExceptionsToCatch; ++i) {
nextTypeToCatch = i - 1;
switchToCatchBlock->addCase(llvm::ConstantInt::get(
llvm::Type::getInt32Ty(context),
i),
catchBlocks[nextTypeToCatch]);
}
llvm::verifyFunction(*ret);
fpm.run(*ret);
return(ret);
}
static
llvm::Function* createThrowExceptionFunction(llvm::Module& module,
llvm::IRBuilder<>& builder,
llvm::FunctionPassManager& fpm,
std::string ourId,
int32_t nativeThrowType,
llvm::Function& nativeThrowFunct) {
llvm::LLVMContext& context = module.getContext();
namedValues.clear();
ArgTypes unwindArgTypes;
unwindArgTypes.push_back(builder.getInt32Ty());
ArgNames unwindArgNames;
unwindArgNames.push_back("exceptTypeToThrow");
llvm::Function *ret = createFunction(module,
builder.getVoidTy(),
unwindArgTypes,
unwindArgNames,
ourId,
llvm::Function::ExternalLinkage,
false,
false);
llvm::BasicBlock *entryBlock = llvm::BasicBlock::Create(context,
"entry",
ret);
llvm::BasicBlock *nativeThrowBlock =
llvm::BasicBlock::Create(context,
"nativeThrow",
ret);
llvm::BasicBlock *generatedThrowBlock =
llvm::BasicBlock::Create(context,
"generatedThrow",
ret);
llvm::Value* exceptionType = namedValues["exceptTypeToThrow"];
builder.SetInsertPoint(nativeThrowBlock);
builder.CreateCall(&nativeThrowFunct, exceptionType);
builder.CreateUnreachable();
builder.SetInsertPoint(entryBlock);
llvm::Function *toPrint32Int = module.getFunction("print32Int");
generateIntegerPrint(context,
module,
builder,
*toPrint32Int,
*exceptionType,
"\nGen: About to throw exception type <%d> in " +
ourId +
".\n",
USE_GLOBAL_STR_CONSTS);
llvm::SwitchInst* theSwitch = builder.CreateSwitch(exceptionType,
generatedThrowBlock,
1);
theSwitch->addCase(llvm::ConstantInt::get(llvm::Type::getInt32Ty(context),
nativeThrowType),
nativeThrowBlock);
builder.SetInsertPoint(generatedThrowBlock);
llvm::Function *createOurException =
module.getFunction("createOurException");
llvm::Function *raiseOurException =
module.getFunction("_Unwind_RaiseException");
llvm::Value* exception =
builder.CreateCall(createOurException,
namedValues["exceptTypeToThrow"]);
builder.CreateCall(raiseOurException, exception);
builder.CreateUnreachable();
llvm::verifyFunction(*ret);
fpm.run(*ret);
return(ret);
}
static void createStandardUtilityFunctions(unsigned numTypeInfos,
llvm::Module& module,
llvm::IRBuilder<>& builder);
llvm::Function* createUnwindExceptionTest(llvm::Module& module,
llvm::IRBuilder<>& builder,
llvm::FunctionPassManager& fpm,
std::string nativeThrowFunctName) {
unsigned numTypeInfos = 6;
createStandardUtilityFunctions(numTypeInfos,
module,
builder);
llvm::Function *nativeThrowFunct =
module.getFunction(nativeThrowFunctName);
llvm::Function* throwFunct =
createThrowExceptionFunction(module,
builder,
fpm,
"throwFunct",
~0,
*nativeThrowFunct);
unsigned innerExceptionTypesToCatch[] = {6, 2, 4};
size_t numExceptionTypesToCatch = sizeof(innerExceptionTypesToCatch) /
sizeof(unsigned);
llvm::Function* innerCatchFunct =
createCatchWrappedInvokeFunction(module,
builder,
fpm,
*throwFunct,
"innerCatchFunct",
numExceptionTypesToCatch,
innerExceptionTypesToCatch);
unsigned outerExceptionTypesToCatch[] = {3, 1, 5};
numExceptionTypesToCatch = sizeof(outerExceptionTypesToCatch) /
sizeof(unsigned);
llvm::Function* outerCatchFunct =
createCatchWrappedInvokeFunction(module,
builder,
fpm,
*innerCatchFunct,
"outerCatchFunct",
numExceptionTypesToCatch,
outerExceptionTypesToCatch);
return(outerCatchFunct);
}
class OurCppRunException : public std::runtime_error {
public:
OurCppRunException(const std::string reason) :
std::runtime_error(reason) {}
OurCppRunException (const OurCppRunException& toCopy) :
std::runtime_error(toCopy) {}
OurCppRunException& operator = (const OurCppRunException& toCopy) {
return(reinterpret_cast<OurCppRunException&>(
std::runtime_error::operator = (toCopy)
));
}
~OurCppRunException (void) throw () {};
};
extern "C"
void throwCppException (int32_t ignoreIt) {
throw(OurCppRunException("thrown by throwCppException(...)"));
}
typedef void (*OurExceptionThrowFunctType) (int32_t typeToThrow);
static
void runExceptionThrow(llvm::ExecutionEngine* engine,
llvm::Function* function,
int32_t typeToThrow) {
OurExceptionThrowFunctType functPtr =
reinterpret_cast<OurExceptionThrowFunctType>(
reinterpret_cast<intptr_t>(
engine->getPointerToFunction(function)
)
);
try {
(*functPtr)(typeToThrow);
}
catch (OurCppRunException exc) {
fprintf(stderr,
"\nrunExceptionThrow(...):In C++ catch OurCppRunException "
"with reason: %s.\n",
exc.what());
}
catch (...) {
fprintf(stderr,
"\nrunExceptionThrow(...):In C++ catch all.\n");
}
}
static void createStandardUtilityFunctions(unsigned numTypeInfos,
llvm::Module& module,
llvm::IRBuilder<>& builder) {
llvm::LLVMContext& context = module.getContext();
ourExceptionNotThrownState =
llvm::ConstantInt::get(llvm::Type::getInt8Ty(context), 0),
ourExceptionThrownState =
llvm::ConstantInt::get(llvm::Type::getInt8Ty(context), 1),
ourExceptionCaughtState =
llvm::ConstantInt::get(llvm::Type::getInt8Ty(context), 2),
ourTypeInfoType = llvm::StructType::get(context,
builder.getInt32Ty(),
NULL);
ourExceptionType = llvm::StructType::get(context,
ourTypeInfoType,
NULL);
ourUnwindExceptionType = llvm::StructType::get(context,
builder.getInt64Ty(),
NULL);
struct OurBaseException_t dummyException;
ourBaseFromUnwindOffset = ((uintptr_t) &dummyException) -
((uintptr_t) &(dummyException.unwindException));
#ifdef DEBUG
fprintf(stderr,
"createStandardUtilityFunctions(...):ourBaseFromUnwindOffset "
"= %lld, sizeof(struct OurBaseException_t) - "
"sizeof(struct _Unwind_Exception) = %lu.\n",
ourBaseFromUnwindOffset,
sizeof(struct OurBaseException_t) -
sizeof(struct _Unwind_Exception));
#endif
size_t numChars = sizeof(ourBaseExcpClassChars) / sizeof(char);
ourBaseExceptionClass = genClass(ourBaseExcpClassChars, numChars);
std::string baseStr = "typeInfo", typeInfoName;
std::ostringstream typeInfoNameBuilder;
std::vector<llvm::Constant*> structVals;
llvm::Constant *nextStruct;
llvm::GlobalVariable* nextGlobal = NULL;
for (unsigned i = 0; i <= numTypeInfos; ++i) {
structVals.clear();
structVals.push_back(llvm::ConstantInt::get(builder.getInt32Ty(), i));
nextStruct = llvm::ConstantStruct::get(ourTypeInfoType, structVals);
typeInfoNameBuilder.str("");
typeInfoNameBuilder << baseStr << i;
typeInfoName = typeInfoNameBuilder.str();
nextGlobal =
new llvm::GlobalVariable(module,
ourTypeInfoType,
true,
llvm::GlobalValue::ExternalLinkage,
nextStruct,
typeInfoName);
ourTypeInfoNames.push_back(typeInfoName);
ourTypeInfoNamesIndex[i] = typeInfoName;
}
ArgNames argNames;
ArgTypes argTypes;
llvm::Function* funct = NULL;
const llvm::Type* retType = builder.getVoidTy();
argTypes.clear();
argTypes.push_back(builder.getInt32Ty());
argTypes.push_back(builder.getInt8Ty()->getPointerTo());
argNames.clear();
createFunction(module,
retType,
argTypes,
argNames,
"print32Int",
llvm::Function::ExternalLinkage,
true,
false);
retType = builder.getVoidTy();
argTypes.clear();
argTypes.push_back(builder.getInt64Ty());
argTypes.push_back(builder.getInt8Ty()->getPointerTo());
argNames.clear();
createFunction(module,
retType,
argTypes,
argNames,
"print64Int",
llvm::Function::ExternalLinkage,
true,
false);
retType = builder.getVoidTy();
argTypes.clear();
argTypes.push_back(builder.getInt8Ty()->getPointerTo());
argNames.clear();
createFunction(module,
retType,
argTypes,
argNames,
"printStr",
llvm::Function::ExternalLinkage,
true,
false);
retType = builder.getVoidTy();
argTypes.clear();
argTypes.push_back(builder.getInt32Ty());
argNames.clear();
createFunction(module,
retType,
argTypes,
argNames,
"throwCppException",
llvm::Function::ExternalLinkage,
true,
false);
retType = builder.getVoidTy();
argTypes.clear();
argTypes.push_back(builder.getInt8Ty()->getPointerTo());
argNames.clear();
createFunction(module,
retType,
argTypes,
argNames,
"deleteOurException",
llvm::Function::ExternalLinkage,
true,
false);
retType = builder.getInt8Ty()->getPointerTo();
argTypes.clear();
argTypes.push_back(builder.getInt32Ty());
argNames.clear();
createFunction(module,
retType,
argTypes,
argNames,
"createOurException",
llvm::Function::ExternalLinkage,
true,
false);
retType = builder.getInt32Ty();
argTypes.clear();
argTypes.push_back(builder.getInt8Ty()->getPointerTo());
argNames.clear();
funct = createFunction(module,
retType,
argTypes,
argNames,
"_Unwind_RaiseException",
llvm::Function::ExternalLinkage,
true,
false);
funct->addFnAttr(llvm::Attribute::NoReturn);
retType = builder.getInt32Ty();
argTypes.clear();
argTypes.push_back(builder.getInt8Ty()->getPointerTo());
argNames.clear();
funct = createFunction(module,
retType,
argTypes,
argNames,
"_Unwind_Resume",
llvm::Function::ExternalLinkage,
true,
false);
funct->addFnAttr(llvm::Attribute::NoReturn);
retType = builder.getInt32Ty();
argTypes.clear();
argTypes.push_back(builder.getInt32Ty());
argTypes.push_back(builder.getInt32Ty());
argTypes.push_back(builder.getInt64Ty());
argTypes.push_back(builder.getInt8Ty()->getPointerTo());
argTypes.push_back(builder.getInt8Ty()->getPointerTo());
argNames.clear();
createFunction(module,
retType,
argTypes,
argNames,
"ourPersonality",
llvm::Function::ExternalLinkage,
true,
false);
getDeclaration(&module, llvm::Intrinsic::eh_selector);
getDeclaration(&module, llvm::Intrinsic::eh_exception);
getDeclaration(&module, llvm::Intrinsic::eh_typeid_for);
}
int main(int argc, char* argv[]) {
if (argc == 1) {
fprintf(stderr,
"\nUsage: ExceptionDemo <exception type to throw> "
"[<type 2>...<type n>].\n"
" Each type must have the value of 1 - 6 for "
"generated exceptions to be caught;\n"
" the value -1 for foreign C++ exceptions to be "
"generated and thrown;\n"
" or the values > 6 for exceptions to be ignored.\n"
"\nTry: ExceptionDemo 2 3 7 -1\n"
" for a full test.\n\n");
return(0);
}
llvm::DwarfExceptionHandling = true;
llvm::InitializeNativeTarget();
llvm::LLVMContext& context = llvm::getGlobalContext();
llvm::IRBuilder<> theBuilder(context);
llvm::Module* module = new llvm::Module("my cool jit", context);
llvm::EngineBuilder factory(module);
factory.setEngineKind(llvm::EngineKind::JIT);
factory.setAllocateGVsWithCode(false);
llvm::ExecutionEngine* executionEngine = factory.create();
{
llvm::FunctionPassManager fpm(module);
fpm.add(new llvm::TargetData(*executionEngine->getTargetData()));
#ifdef ADD_OPT_PASSES
fpm.add(llvm::createPromoteMemoryToRegisterPass());
fpm.add(llvm::createInstructionCombiningPass());
fpm.add(llvm::createReassociatePass());
fpm.add(llvm::createGVNPass());
fpm.add(llvm::createCFGSimplificationPass());
#endif // ADD_OPT_PASSES
fpm.doInitialization();
llvm::Function* toRun =
createUnwindExceptionTest(*module,
theBuilder,
fpm,
"throwCppException");
fprintf(stderr, "\nBegin module dump:\n\n");
module->dump();
fprintf(stderr, "\nEnd module dump:\n");
fprintf(stderr, "\n\nBegin Test:\n");
for (int i = 1; i < argc; ++i) {
runExceptionThrow(executionEngine,
toRun,
(unsigned) strtoul(argv[i], NULL, 10));
}
fprintf(stderr, "\nEnd Test:\n\n");
}
delete executionEngine;
return 0;
}