| //===-- IRDynamicChecks.cpp -------------------------------------------*- C++ -*-===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "lldb/Expression/IRDynamicChecks.h" |
| |
| #include "lldb/Core/ConstString.h" |
| #include "lldb/Core/Log.h" |
| #include "lldb/Expression/ClangUtilityFunction.h" |
| #include "lldb/Target/ExecutionContext.h" |
| #include "lldb/Target/ObjCLanguageRuntime.h" |
| #include "lldb/Target/Process.h" |
| #include "lldb/Target/StackFrame.h" |
| |
| #include "llvm/Support/raw_ostream.h" |
| #include "llvm/Constants.h" |
| #include "llvm/Function.h" |
| #include "llvm/Instructions.h" |
| #include "llvm/Module.h" |
| #include "llvm/Value.h" |
| |
| using namespace llvm; |
| using namespace lldb_private; |
| |
| static char ID; |
| |
| #define VALID_POINTER_CHECK_NAME "$__lldb_valid_pointer_check" |
| #define VALID_OBJC_OBJECT_CHECK_NAME "$__lldb_objc_object_check" |
| |
| static const char g_valid_pointer_check_text[] = |
| "extern \"C\" void\n" |
| "$__lldb_valid_pointer_check (unsigned char *$__lldb_arg_ptr)\n" |
| "{\n" |
| " unsigned char $__lldb_local_val = *$__lldb_arg_ptr;\n" |
| "}"; |
| |
| DynamicCheckerFunctions::DynamicCheckerFunctions () |
| { |
| } |
| |
| DynamicCheckerFunctions::~DynamicCheckerFunctions () |
| { |
| } |
| |
| bool |
| DynamicCheckerFunctions::Install(Stream &error_stream, |
| ExecutionContext &exe_ctx) |
| { |
| m_valid_pointer_check.reset(new ClangUtilityFunction(g_valid_pointer_check_text, |
| VALID_POINTER_CHECK_NAME)); |
| if (!m_valid_pointer_check->Install(error_stream, exe_ctx)) |
| return false; |
| |
| if (exe_ctx.process) |
| { |
| ObjCLanguageRuntime *objc_language_runtime = exe_ctx.process->GetObjCLanguageRuntime(); |
| |
| if (objc_language_runtime) |
| { |
| m_objc_object_check.reset(objc_language_runtime->CreateObjectChecker(VALID_OBJC_OBJECT_CHECK_NAME)); |
| |
| if (!m_objc_object_check->Install(error_stream, exe_ctx)) |
| return false; |
| } |
| } |
| |
| return true; |
| } |
| |
| static std::string |
| PrintValue(llvm::Value *V, bool truncate = false) |
| { |
| std::string s; |
| raw_string_ostream rso(s); |
| V->print(rso); |
| rso.flush(); |
| if (truncate) |
| s.resize(s.length() - 1); |
| return s; |
| } |
| |
| //---------------------------------------------------------------------- |
| /// @class Instrumenter IRDynamicChecks.cpp |
| /// @brief Finds and instruments individual LLVM IR instructions |
| /// |
| /// When instrumenting LLVM IR, it is frequently desirable to first search |
| /// for instructions, and then later modify them. This way iterators |
| /// remain intact, and multiple passes can look at the same code base without |
| /// treading on each other's toes. |
| /// |
| /// The Instrumenter class implements this functionality. A client first |
| /// calls Inspect on a function, which populates a list of instructions to |
| /// be instrumented. Then, later, when all passes' Inspect functions have |
| /// been called, the client calls Instrument, which adds the desired |
| /// instrumentation. |
| /// |
| /// A subclass of Instrumenter must override InstrumentInstruction, which |
| /// is responsible for adding whatever instrumentation is necessary. |
| /// |
| /// A subclass of Instrumenter may override: |
| /// |
| /// - InspectInstruction [default: does nothing] |
| /// |
| /// - InspectBasicBlock [default: iterates through the instructions in a |
| /// basic block calling InspectInstruction] |
| /// |
| /// - InspectFunction [default: iterates through the basic blocks in a |
| /// function calling InspectBasicBlock] |
| //---------------------------------------------------------------------- |
| class Instrumenter { |
| public: |
| //------------------------------------------------------------------ |
| /// Constructor |
| /// |
| /// @param[in] module |
| /// The module being instrumented. |
| //------------------------------------------------------------------ |
| Instrumenter (llvm::Module &module, |
| DynamicCheckerFunctions &checker_functions) : |
| m_module(module), |
| m_checker_functions(checker_functions), |
| m_i8ptr_ty(NULL) |
| { |
| } |
| |
| virtual~Instrumenter () |
| { |
| } |
| |
| //------------------------------------------------------------------ |
| /// Inspect a function to find instructions to instrument |
| /// |
| /// @param[in] function |
| /// The function to inspect. |
| /// |
| /// @return |
| /// True on success; false on error. |
| //------------------------------------------------------------------ |
| bool Inspect (llvm::Function &function) |
| { |
| return InspectFunction(function); |
| } |
| |
| //------------------------------------------------------------------ |
| /// Instrument all the instructions found by Inspect() |
| /// |
| /// @return |
| /// True on success; false on error. |
| //------------------------------------------------------------------ |
| bool Instrument () |
| { |
| for (InstIterator ii = m_to_instrument.begin(), last_ii = m_to_instrument.end(); |
| ii != last_ii; |
| ++ii) |
| { |
| if (!InstrumentInstruction(*ii)) |
| return false; |
| } |
| |
| return true; |
| } |
| protected: |
| //------------------------------------------------------------------ |
| /// Add instrumentation to a single instruction |
| /// |
| /// @param[in] inst |
| /// The instruction to be instrumented. |
| /// |
| /// @return |
| /// True on success; false otherwise. |
| //------------------------------------------------------------------ |
| virtual bool InstrumentInstruction(llvm::Instruction *inst) = 0; |
| |
| //------------------------------------------------------------------ |
| /// Register a single instruction to be instrumented |
| /// |
| /// @param[in] inst |
| /// The instruction to be instrumented. |
| //------------------------------------------------------------------ |
| void RegisterInstruction(llvm::Instruction &i) |
| { |
| m_to_instrument.push_back(&i); |
| } |
| |
| //------------------------------------------------------------------ |
| /// Determine whether a single instruction is interesting to |
| /// instrument, and, if so, call RegisterInstruction |
| /// |
| /// @param[in] i |
| /// The instruction to be inspected. |
| /// |
| /// @return |
| /// False if there was an error scanning; true otherwise. |
| //------------------------------------------------------------------ |
| virtual bool InspectInstruction(llvm::Instruction &i) |
| { |
| return true; |
| } |
| |
| //------------------------------------------------------------------ |
| /// Scan a basic block to see if any instructions are interesting |
| /// |
| /// @param[in] bb |
| /// The basic block to be inspected. |
| /// |
| /// @return |
| /// False if there was an error scanning; true otherwise. |
| //------------------------------------------------------------------ |
| virtual bool InspectBasicBlock(llvm::BasicBlock &bb) |
| { |
| for (llvm::BasicBlock::iterator ii = bb.begin(), last_ii = bb.end(); |
| ii != last_ii; |
| ++ii) |
| { |
| if (!InspectInstruction(*ii)) |
| return false; |
| } |
| |
| return true; |
| } |
| |
| //------------------------------------------------------------------ |
| /// Scan a function to see if any instructions are interesting |
| /// |
| /// @param[in] f |
| /// The function to be inspected. |
| /// |
| /// @return |
| /// False if there was an error scanning; true otherwise. |
| //------------------------------------------------------------------ |
| virtual bool InspectFunction(llvm::Function &f) |
| { |
| for (llvm::Function::iterator bbi = f.begin(), last_bbi = f.end(); |
| bbi != last_bbi; |
| ++bbi) |
| { |
| if (!InspectBasicBlock(*bbi)) |
| return false; |
| } |
| |
| return true; |
| } |
| |
| //------------------------------------------------------------------ |
| /// Build a function pointer for a function with signature |
| /// void (*)(uint8_t*) with a given address |
| /// |
| /// @param[in] start_address |
| /// The address of the function. |
| /// |
| /// @return |
| /// The function pointer, for use in a CallInst. |
| //------------------------------------------------------------------ |
| llvm::Value *BuildPointerValidatorFunc(lldb::addr_t start_address) |
| { |
| std::vector<const llvm::Type*> params; |
| |
| const IntegerType *intptr_ty = llvm::Type::getIntNTy(m_module.getContext(), |
| (m_module.getPointerSize() == llvm::Module::Pointer64) ? 64 : 32); |
| |
| params.push_back(GetI8PtrTy()); |
| |
| FunctionType *fun_ty = FunctionType::get(llvm::Type::getVoidTy(m_module.getContext()), params, true); |
| PointerType *fun_ptr_ty = PointerType::getUnqual(fun_ty); |
| Constant *fun_addr_int = ConstantInt::get(intptr_ty, start_address, false); |
| return ConstantExpr::getIntToPtr(fun_addr_int, fun_ptr_ty); |
| } |
| |
| const PointerType *GetI8PtrTy() |
| { |
| if (!m_i8ptr_ty) |
| m_i8ptr_ty = llvm::Type::getInt8PtrTy(m_module.getContext()); |
| |
| return m_i8ptr_ty; |
| } |
| |
| typedef std::vector <llvm::Instruction *> InstVector; |
| typedef InstVector::iterator InstIterator; |
| |
| InstVector m_to_instrument; ///< List of instructions the inspector found |
| llvm::Module &m_module; ///< The module which is being instrumented |
| DynamicCheckerFunctions &m_checker_functions; ///< The dynamic checker functions for the process |
| private: |
| const PointerType *m_i8ptr_ty; |
| }; |
| |
| class ValidPointerChecker : public Instrumenter |
| { |
| public: |
| ValidPointerChecker (llvm::Module &module, |
| DynamicCheckerFunctions &checker_functions) : |
| Instrumenter(module, checker_functions), |
| m_valid_pointer_check_func(NULL) |
| { |
| } |
| |
| virtual ~ValidPointerChecker () |
| { |
| } |
| private: |
| bool InstrumentInstruction(llvm::Instruction *inst) |
| { |
| lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS)); |
| |
| if(log) |
| log->Printf("Instrumenting load/store instruction: %s\n", |
| PrintValue(inst).c_str()); |
| |
| if (!m_valid_pointer_check_func) |
| m_valid_pointer_check_func = BuildPointerValidatorFunc(m_checker_functions.m_valid_pointer_check->StartAddress()); |
| |
| llvm::Value *dereferenced_ptr; |
| |
| if (llvm::LoadInst *li = dyn_cast<llvm::LoadInst> (inst)) |
| dereferenced_ptr = li->getPointerOperand(); |
| else if (llvm::StoreInst *si = dyn_cast<llvm::StoreInst> (inst)) |
| dereferenced_ptr = si->getPointerOperand(); |
| else |
| return false; |
| |
| // Insert an instruction to cast the loaded value to int8_t* |
| |
| BitCastInst *bit_cast = new BitCastInst(dereferenced_ptr, |
| GetI8PtrTy(), |
| "", |
| inst); |
| |
| // Insert an instruction to call the helper with the result |
| |
| SmallVector <llvm::Value*, 1> args; |
| args.push_back(bit_cast); |
| |
| CallInst::Create(m_valid_pointer_check_func, |
| args.begin(), |
| args.end(), |
| "", |
| inst); |
| |
| return true; |
| } |
| |
| bool InspectInstruction(llvm::Instruction &i) |
| { |
| if (dyn_cast<llvm::LoadInst> (&i) || |
| dyn_cast<llvm::StoreInst> (&i)) |
| RegisterInstruction(i); |
| |
| return true; |
| } |
| |
| llvm::Value *m_valid_pointer_check_func; |
| }; |
| |
| class ObjcObjectChecker : public Instrumenter |
| { |
| public: |
| ObjcObjectChecker(llvm::Module &module, |
| DynamicCheckerFunctions &checker_functions) : |
| Instrumenter(module, checker_functions), |
| m_objc_object_check_func(NULL) |
| { |
| } |
| |
| virtual |
| ~ObjcObjectChecker () |
| { |
| } |
| |
| private: |
| bool InstrumentInstruction(llvm::Instruction *inst) |
| { |
| CallInst *call_inst = dyn_cast<CallInst>(inst); |
| |
| if (!call_inst) |
| return false; // call_inst really shouldn't be NULL, because otherwise InspectInstruction wouldn't have registered it |
| |
| if (!m_objc_object_check_func) |
| m_objc_object_check_func = BuildPointerValidatorFunc(m_checker_functions.m_objc_object_check->StartAddress()); |
| |
| llvm::Value *target_object; |
| |
| // id objc_msgSend(id theReceiver, SEL theSelector, ...) |
| |
| target_object = call_inst->getArgOperand(0); |
| |
| // Insert an instruction to cast the receiver id to int8_t* |
| |
| BitCastInst *bit_cast = new BitCastInst(target_object, |
| GetI8PtrTy(), |
| "", |
| inst); |
| |
| // Insert an instruction to call the helper with the result |
| |
| SmallVector <llvm::Value*, 1> args; |
| args.push_back(bit_cast); |
| |
| CallInst::Create(m_objc_object_check_func, |
| args.begin(), |
| args.end(), |
| "", |
| inst); |
| |
| return true; |
| } |
| |
| bool InspectInstruction(llvm::Instruction &i) |
| { |
| lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS)); |
| |
| CallInst *call_inst = dyn_cast<CallInst>(&i); |
| |
| if (call_inst) |
| { |
| // This metadata is set by IRForTarget::MaybeHandleCall(). |
| |
| MDNode *metadata = call_inst->getMetadata("lldb.call.realName"); |
| |
| if (!metadata) |
| return true; |
| |
| if (metadata->getNumOperands() != 1) |
| { |
| if (log) |
| log->Printf("Function call metadata has %d operands for [%p] %s", metadata->getNumOperands(), call_inst, PrintValue(call_inst).c_str()); |
| return false; |
| } |
| |
| ConstantArray *real_name = dyn_cast<ConstantArray>(metadata->getOperand(0)); |
| |
| if (!real_name) |
| { |
| if (log) |
| log->Printf("Function call metadata is not a ConstantArray for [%p] %s", call_inst, PrintValue(call_inst).c_str()); |
| return false; |
| } |
| |
| if (!real_name->isString()) |
| { |
| if (log) |
| log->Printf("Function call metadata is not a string for [%p] %s", call_inst, PrintValue(call_inst).c_str()); |
| return false; |
| } |
| |
| if (log) |
| log->Printf("Found call to %s: %s\n", real_name->getAsString().c_str(), PrintValue(call_inst).c_str()); |
| |
| if (real_name->getAsString().find("objc_msgSend") != std::string::npos) |
| RegisterInstruction(i); |
| } |
| |
| return true; |
| } |
| |
| llvm::Value *m_objc_object_check_func; |
| }; |
| |
| IRDynamicChecks::IRDynamicChecks(DynamicCheckerFunctions &checker_functions, |
| const char *func_name) : |
| ModulePass(ID), |
| m_func_name(func_name), |
| m_checker_functions(checker_functions) |
| { |
| } |
| |
| IRDynamicChecks::~IRDynamicChecks() |
| { |
| } |
| |
| bool |
| IRDynamicChecks::runOnModule(llvm::Module &M) |
| { |
| lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS)); |
| |
| llvm::Function* function = M.getFunction(StringRef(m_func_name.c_str())); |
| |
| if (!function) |
| { |
| if (log) |
| log->Printf("Couldn't find %s() in the module", m_func_name.c_str()); |
| |
| return false; |
| } |
| |
| ValidPointerChecker vpc(M, m_checker_functions); |
| |
| if (!vpc.Inspect(*function)) |
| return false; |
| |
| if (!vpc.Instrument()) |
| return false; |
| |
| ObjcObjectChecker ooc(M, m_checker_functions); |
| |
| if (!ooc.Inspect(*function)) |
| return false; |
| |
| if (!ooc.Instrument()) |
| return false; |
| |
| if (log && log->GetVerbose()) |
| { |
| std::string s; |
| raw_string_ostream oss(s); |
| |
| M.print(oss, NULL); |
| |
| oss.flush(); |
| |
| log->Printf ("Module after dynamic checks: \n%s", s.c_str()); |
| } |
| |
| return true; |
| } |
| |
| void |
| IRDynamicChecks::assignPassManager(PMStack &PMS, |
| PassManagerType T) |
| { |
| } |
| |
| PassManagerType |
| IRDynamicChecks::getPotentialPassManagerType() const |
| { |
| return PMT_ModulePassManager; |
| } |