16#include "llvm/IR/Constants.h"
17#include "llvm/IR/DataLayout.h"
18#include "llvm/IR/InstrTypes.h"
19#include "llvm/IR/Instructions.h"
20#include "llvm/IR/Intrinsics.h"
21#include "llvm/IR/LegacyPassManager.h"
22#include "llvm/IR/Metadata.h"
23#include "llvm/IR/Module.h"
24#include "llvm/IR/Operator.h"
25#include "llvm/IR/ValueSymbolTable.h"
26#include "llvm/Support/ErrorExtras.h"
27#include "llvm/Support/raw_ostream.h"
28#include "llvm/Transforms/IPO.h"
30#include "clang/AST/ASTContext.h"
60 llvm::Value *ret =
m_maker(function);
68 if (function->empty())
71 return &*function->getEntryBlock().getFirstNonPHIOrDbg();
79 const char *func_name)
91 raw_string_ostream rso(s);
96static std::string
PrintType(
const llvm::Type *type) {
100 raw_string_ostream rso(s);
106 llvm_function.setLinkage(GlobalValue::ExternalLinkage);
113 NamedMDNode *named_metadata =
114 module->getNamedMetadata("clang.global.decl.ptrs");
119 unsigned num_nodes = named_metadata->getNumOperands();
122 for (node_index = 0; node_index < num_nodes; ++node_index) {
123 llvm::MDNode *metadata_node =
124 dyn_cast<llvm::MDNode>(named_metadata->getOperand(node_index));
128 if (metadata_node->getNumOperands() != 2)
131 if (mdconst::dyn_extract_or_null<GlobalValue>(
132 metadata_node->getOperand(0)) != global_val)
135 ConstantInt *constant_int =
136 mdconst::dyn_extract<ConstantInt>(metadata_node->getOperand(1));
141 uintptr_t ptr = constant_int->getZExtValue();
143 return reinterpret_cast<clang::NamedDecl *
>(ptr);
155 bool check_ms_abi =
true) {
157 mangled_symbol.starts_with(
"_ZGV");
159 result |= mangled_symbol.ends_with(
"@4IA");
171 ValueSymbolTable &value_symbol_table =
m_module->getValueSymbolTable();
173 llvm::StringRef result_name;
174 bool found_result =
false;
176 for (StringMapEntry<llvm::Value *> &value_symbol : value_symbol_table) {
177 result_name = value_symbol.first();
183 if (result_name.contains(
"$__lldb_expr_result_ptr") && !is_guard_var) {
189 if (result_name.contains(
"$__lldb_expr_result") && !is_guard_var) {
197 LLDB_LOG(log,
"Couldn't find result variable");
202 LLDB_LOG(log,
"Result name: \"{0}\"", result_name);
204 Value *result_value =
m_module->getNamedValue(result_name);
207 LLDB_LOG(log,
"Result variable had no data");
209 m_error_stream.Format(
"Internal error [IRForTarget]: Result variable's "
210 "name ({0}) exists, but not its definition\n",
218 GlobalVariable *result_global = dyn_cast<GlobalVariable>(result_value);
220 if (!result_global) {
221 LLDB_LOG(log,
"Result variable isn't a GlobalVariable");
223 m_error_stream.Format(
"Internal error [IRForTarget]: Result variable ({0}) "
224 "is defined, but is not a global variable\n",
230 clang::NamedDecl *result_decl =
DeclForGlobal(result_global);
232 LLDB_LOG(log,
"Result variable doesn't have a corresponding Decl");
234 m_error_stream.Format(
"Internal error [IRForTarget]: Result variable ({0}) "
235 "does not have a corresponding Clang entity\n",
242 std::string decl_desc_str;
243 raw_string_ostream decl_desc_stream(decl_desc_str);
244 result_decl->print(decl_desc_stream);
246 LLDB_LOG(log,
"Found result decl: \"{0}\"", decl_desc_str);
249 clang::VarDecl *result_var = dyn_cast<clang::VarDecl>(result_decl);
251 LLDB_LOG(log,
"Result variable Decl isn't a VarDecl");
253 m_error_stream.Format(
"Internal error [IRForTarget]: Result variable "
254 "({0})'s corresponding Clang entity isn't a "
267 clang::QualType pointer_qual_type = result_var->getType();
268 const clang::Type *pointer_type = pointer_qual_type.getTypePtr();
270 const clang::PointerType *pointer_pointertype =
271 pointer_type->getAs<clang::PointerType>();
272 const clang::ObjCObjectPointerType *pointer_objcobjpointertype =
273 pointer_type->getAs<clang::ObjCObjectPointerType>();
275 if (pointer_pointertype) {
276 clang::QualType element_qual_type = pointer_pointertype->getPointeeType();
279 m_decl_map->GetTypeSystem()->GetType(element_qual_type));
280 }
else if (pointer_objcobjpointertype) {
281 clang::QualType element_qual_type =
282 clang::QualType(pointer_objcobjpointertype->getObjectType(), 0);
285 m_decl_map->GetTypeSystem()->GetType(element_qual_type));
287 LLDB_LOG(log,
"Expected result to have pointer type, but it did not");
289 m_error_stream.Format(
"Internal error [IRForTarget]: Lvalue result ({0}) "
290 "is not a pointer variable\n",
297 m_decl_map->GetTypeSystem()->GetType(result_var->getType()));
301 auto bit_size_or_err =
m_result_type.GetBitSize(target_sp.get());
302 if (!bit_size_or_err) {
306 LLDB_LOG(log,
"Result type has unknown size");
308 m_error_stream.Printf(
"Error [IRForTarget]: Size of result type '%s' "
309 "couldn't be determined\n%s",
311 llvm::toString(bit_size_or_err.takeError()).c_str());
319 LLDB_LOG(log,
"Result decl type: \"{0}\"", type_desc_stream.
GetData());
324 LLDB_LOG(log,
"Creating a new result global: \"{0}\" with size {1}",
326 llvm::expectedToOptional(
m_result_type.GetByteSize(target_sp.get()))
331 GlobalVariable *new_result_global =
new GlobalVariable(
332 (*
m_module), result_global->getValueType(),
false,
333 GlobalValue::ExternalLinkage,
nullptr,
343 ConstantInt *new_constant_int =
344 ConstantInt::get(llvm::Type::getInt64Ty(
m_module->getContext()),
345 reinterpret_cast<uintptr_t
>(result_decl),
false);
347 llvm::Metadata *values[2];
348 values[0] = ConstantAsMetadata::get(new_result_global);
349 values[1] = ConstantAsMetadata::get(new_constant_int);
351 ArrayRef<Metadata *> value_ref(values, 2);
353 MDNode *persistent_global_md = MDNode::get(
m_module->getContext(), value_ref);
354 NamedMDNode *named_metadata =
355 m_module->getNamedMetadata(
"clang.global.decl.ptrs");
356 named_metadata->addOperand(persistent_global_md);
361 if (result_global->use_empty()) {
365 BasicBlock &entry_block(llvm_function.getEntryBlock());
366 Instruction *first_entry_instruction(&*entry_block.getFirstNonPHIOrDbg());
368 if (!first_entry_instruction)
371 if (!result_global->hasInitializer()) {
372 LLDB_LOG(log,
"Couldn't find initializer for unused variable");
374 m_error_stream.Format(
"Internal error [IRForTarget]: Result variable "
375 "({0}) has no writes and no initializer\n",
381 Constant *initializer = result_global->getInitializer();
383 StoreInst *synthesized_store =
new StoreInst(
384 initializer, new_result_global, first_entry_instruction->getIterator());
386 LLDB_LOG(log,
"Synthesized result store \"{0}\"\n",
389 result_global->replaceAllUsesWith(new_result_global);
396 result_global->eraseFromParent();
402 llvm::GlobalVariable *cstr) {
405 Type *ns_str_ty = ns_str->getType();
407 Type *i8_ptr_ty = PointerType::getUnqual(
m_module->getContext());
408 Type *i32_ty = Type::getInt32Ty(
m_module->getContext());
409 Type *i8_ty = Type::getInt8Ty(
m_module->getContext());
415 "CFStringCreateWithBytes");
417 bool missing_weak =
false;
419 g_CFStringCreateWithBytes_str, missing_weak);
421 LLDB_LOG(log,
"Couldn't find CFStringCreateWithBytes in the target");
423 m_error_stream.Printf(
"Error [IRForTarget]: Rewriting an Objective-C "
424 "constant string requires "
425 "CFStringCreateWithBytes\n");
430 LLDB_LOG(log,
"Found CFStringCreateWithBytes at {0}",
431 CFStringCreateWithBytes_addr);
451 Type *arg_type_array[5];
453 arg_type_array[0] = i8_ptr_ty;
454 arg_type_array[1] = i8_ptr_ty;
456 arg_type_array[3] = i32_ty;
457 arg_type_array[4] = i8_ty;
459 ArrayRef<Type *> CFSCWB_arg_types(arg_type_array, 5);
461 llvm::FunctionType *CFSCWB_ty =
462 FunctionType::get(ns_str_ty, CFSCWB_arg_types,
false);
465 PointerType *CFSCWB_ptr_ty = PointerType::getUnqual(
m_module->getContext());
466 Constant *CFSCWB_addr_int =
467 ConstantInt::get(
m_intptr_ty, CFStringCreateWithBytes_addr,
false);
469 CFSCWB_ty, ConstantExpr::getIntToPtr(CFSCWB_addr_int, CFSCWB_ptr_ty)};
472 ConstantDataSequential *string_array =
nullptr;
475 string_array = dyn_cast<ConstantDataSequential>(cstr->getInitializer());
477 Constant *alloc_arg = Constant::getNullValue(i8_ptr_ty);
478 Constant *bytes_arg = cstr ? cstr : Constant::getNullValue(i8_ptr_ty);
479 Constant *numBytes_arg = ConstantInt::get(
480 m_intptr_ty, cstr ? (string_array->getNumElements() - 1) * string_array->getElementByteSize() : 0,
false);
481 int encoding_flags = 0;
482 switch (cstr ? string_array->getElementByteSize() : 1) {
484 encoding_flags = 0x08000100;
487 encoding_flags = 0x0100;
490 encoding_flags = 0x0c000100;
493 encoding_flags = 0x0600;
494 LLDB_LOG(log,
"Encountered an Objective-C constant string with unusual "
496 string_array->getElementByteSize());
498 Constant *encoding_arg = ConstantInt::get(i32_ty, encoding_flags,
false);
499 Constant *isExternal_arg =
500 ConstantInt::get(i8_ty, 0x0,
false);
502 Value *argument_array[5];
504 argument_array[0] = alloc_arg;
505 argument_array[1] = bytes_arg;
506 argument_array[2] = numBytes_arg;
507 argument_array[3] = encoding_arg;
508 argument_array[4] = isExternal_arg;
510 ArrayRef<Value *> CFSCWB_arguments(argument_array, 5);
513 [
this, &CFSCWB_arguments](llvm::Function *function) -> llvm::Value * {
514 return CallInst::Create(
516 "CFStringCreateWithBytes",
517 llvm::cast<Instruction>(
524 std::string error_msg = llvm::toString(std::move(err));
526 "Couldn't replace the NSString with the result of the call: {0}",
529 m_error_stream.Format(
"error [IRForTarget internal]: Couldn't replace an "
530 "Objective-C constant string with a dynamic "
537 ns_str->eraseFromParent();
545 ValueSymbolTable &value_symbol_table =
m_module->getValueSymbolTable();
547 std::vector<std::pair<GlobalVariable *, GlobalVariable *>>
548 nsstring_to_cstr_list;
550 for (StringMapEntry<llvm::Value *> &value_symbol : value_symbol_table) {
551 llvm::StringRef value_name = value_symbol.first();
553 if (value_name.contains(
"_unnamed_cfstring_")) {
554 Value *nsstring_value = value_symbol.second;
556 GlobalVariable *nsstring_global =
557 dyn_cast<GlobalVariable>(nsstring_value);
559 if (!nsstring_global) {
560 LLDB_LOG(log,
"NSString variable is not a GlobalVariable");
562 m_error_stream.Printf(
"Internal error [IRForTarget]: An Objective-C "
563 "constant string is not a global variable\n");
568 if (!nsstring_global->hasInitializer()) {
569 LLDB_LOG(log,
"NSString variable does not have an initializer");
571 m_error_stream.Printf(
"Internal error [IRForTarget]: An Objective-C "
572 "constant string does not have an initializer\n");
577 ConstantStruct *nsstring_struct =
578 dyn_cast<ConstantStruct>(nsstring_global->getInitializer());
580 if (!nsstring_struct) {
582 "NSString variable's initializer is not a ConstantStruct");
584 m_error_stream.Printf(
"Internal error [IRForTarget]: An Objective-C "
585 "constant string is not a structure constant\n");
599 if (nsstring_struct->getNumOperands() != 4) {
602 "NSString variable's initializer structure has an "
603 "unexpected number of members. Should be 4, is {0}",
604 nsstring_struct->getNumOperands());
606 m_error_stream.Printf(
"Internal error [IRForTarget]: The struct for an "
607 "Objective-C constant string is not as "
613 Constant *nsstring_member = nsstring_struct->getOperand(2);
615 if (!nsstring_member) {
616 LLDB_LOG(log,
"NSString initializer's str element was empty");
618 m_error_stream.Printf(
"Internal error [IRForTarget]: An Objective-C "
619 "constant string does not have a string "
625 auto *cstr_global = dyn_cast<GlobalVariable>(nsstring_member);
628 "NSString initializer's str element is not a GlobalVariable");
631 "constant string initializer\n");
636 if (!cstr_global->hasInitializer()) {
637 LLDB_LOG(log,
"NSString initializer's str element does not have an "
640 m_error_stream.Printf(
"Internal error [IRForTarget]: An Objective-C "
641 "constant string's string initializer doesn't "
642 "point to initialized data\n");
677 ConstantDataArray *cstr_array =
678 dyn_cast<ConstantDataArray>(cstr_global->getInitializer());
681 LLDB_LOG(log,
"Found NSString constant {0}, which contains \"{1}\"",
682 value_name, cstr_array->getAsString());
684 LLDB_LOG(log,
"Found NSString constant {0}, which contains \"\"",
688 cstr_global =
nullptr;
692 nsstring_to_cstr_list.emplace_back(nsstring_global, cstr_global);
696 for (
auto [nsstring_global, cstr_global] : nsstring_to_cstr_list) {
698 LLDB_LOG(log,
"Error rewriting the constant string");
703 for (StringMapEntry<llvm::Value *> &value_symbol : value_symbol_table) {
704 llvm::StringRef value_name = value_symbol.first();
706 if (value_name ==
"__CFConstantStringClassReference") {
707 GlobalVariable *gv = dyn_cast<GlobalVariable>(value_symbol.second);
711 "__CFConstantStringClassReference is not a global variable");
714 "CFConstantStringClassReference, but it is not a "
720 gv->eraseFromParent();
730 GlobalVariable *global_variable = dyn_cast<GlobalVariable>(value);
733 !global_variable || !global_variable->hasName() ||
734 !global_variable->getName().starts_with(
"OBJC_SELECTOR_REFERENCES_"));
741 LoadInst *load = dyn_cast<LoadInst>(selector_load);
760 GlobalVariable *_objc_selector_references_ =
761 dyn_cast<GlobalVariable>(load->getPointerOperand());
763 if (!_objc_selector_references_ ||
764 !_objc_selector_references_->hasInitializer())
767 Constant *osr_initializer = _objc_selector_references_->getInitializer();
768 if (!osr_initializer)
773 GlobalVariable *_objc_meth_var_name_ =
774 dyn_cast<GlobalVariable>(osr_initializer);
776 if (!_objc_meth_var_name_ || !_objc_meth_var_name_->hasInitializer())
779 Constant *omvn_initializer = _objc_meth_var_name_->getInitializer();
781 ConstantDataArray *omvn_initializer_array =
782 dyn_cast<ConstantDataArray>(omvn_initializer);
784 if (!omvn_initializer_array->isString())
787 std::string omvn_initializer_string =
788 std::string(omvn_initializer_array->getAsString());
790 LLDB_LOG(log,
"Found Objective-C selector reference \"{0}\"",
791 omvn_initializer_string);
798 bool missing_weak =
false;
805 LLDB_LOG(log,
"Found sel_registerName at {0}", sel_registerName_addr);
814 Type *sel_ptr_type = PointerType::getUnqual(
m_module->getContext());
818 type_array[0] = llvm::PointerType::getUnqual(
m_module->getContext());
820 ArrayRef<Type *> srN_arg_types(type_array, 1);
822 llvm::FunctionType *srN_type =
823 FunctionType::get(sel_ptr_type, srN_arg_types,
false);
826 PointerType *srN_ptr_ty = PointerType::getUnqual(
m_module->getContext());
827 Constant *srN_addr_int =
828 ConstantInt::get(
m_intptr_ty, sel_registerName_addr,
false);
830 ConstantExpr::getIntToPtr(srN_addr_int, srN_ptr_ty)};
835 "sel_registerName", selector_load->getIterator());
839 selector_load->replaceAllUsesWith(srN_call);
841 selector_load->eraseFromParent();
851 for (Instruction &inst : basic_block) {
852 if (LoadInst *load = dyn_cast<LoadInst>(&inst))
854 selector_loads.push_back(&inst);
857 for (Instruction *inst : selector_loads) {
859 m_error_stream.Printf(
"Internal error [IRForTarget]: Couldn't change a "
860 "static reference to an Objective-C selector to a "
861 "dynamic reference\n");
863 LLDB_LOG(log,
"Couldn't rewrite a reference to an Objective-C selector");
876 AllocaInst *alloc = dyn_cast<AllocaInst>(persistent_alloc);
878 MDNode *alloc_md = alloc->getMetadata(
"clang.decl.ptr");
880 if (!alloc_md || !alloc_md->getNumOperands())
883 ConstantInt *constant_int =
884 mdconst::dyn_extract<ConstantInt>(alloc_md->getOperand(0));
891 uintptr_t ptr = constant_int->getZExtValue();
893 clang::VarDecl *decl =
reinterpret_cast<clang::VarDecl *
>(ptr);
896 m_decl_map->GetTypeSystem()->GetType(decl->getType()));
898 StringRef decl_name(decl->getName());
900 if (!
m_decl_map->AddPersistentVariable(decl, persistent_variable_name,
901 result_decl_type,
false,
false))
904 GlobalVariable *persistent_global =
new GlobalVariable(
905 (*
m_module), alloc->getType(),
false,
906 GlobalValue::ExternalLinkage,
nullptr,
907 alloc->getName().str());
912 NamedMDNode *named_metadata =
913 m_module->getOrInsertNamedMetadata(
"clang.global.decl.ptrs");
915 llvm::Metadata *values[2];
916 values[0] = ConstantAsMetadata::get(persistent_global);
917 values[1] = ConstantAsMetadata::get(constant_int);
919 ArrayRef<llvm::Metadata *> value_ref(values, 2);
921 MDNode *persistent_global_md = MDNode::get(
m_module->getContext(), value_ref);
922 named_metadata->addOperand(persistent_global_md);
927 LoadInst *persistent_load =
928 new LoadInst(persistent_global->getValueType(), persistent_global,
"",
929 alloc->getIterator());
934 alloc->replaceAllUsesWith(persistent_load);
935 alloc->eraseFromParent();
948 for (Instruction &inst : basic_block) {
950 if (AllocaInst *alloc = dyn_cast<AllocaInst>(&inst)) {
951 llvm::StringRef alloc_name = alloc->getName();
953 if (alloc_name.starts_with(
"$") && !alloc_name.starts_with(
"$__lldb")) {
954 if (alloc_name.find_first_of(
"0123456789") == 1) {
955 LLDB_LOG(log,
"Rejecting a numeric persistent variable.");
957 m_error_stream.Printf(
"Error [IRForTarget]: Names starting with $0, "
958 "$1, ... are reserved for use as result "
964 pvar_allocs.push_back(alloc);
969 for (Instruction *inst : pvar_allocs) {
971 m_error_stream.Printf(
"Internal error [IRForTarget]: Couldn't rewrite "
972 "the creation of a persistent variable\n");
974 LLDB_LOG(log,
"Couldn't rewrite the creation of a persistent variable");
989 if (ConstantExpr *constant_expr = dyn_cast<ConstantExpr>(llvm_value_ptr)) {
990 switch (constant_expr->getOpcode()) {
993 case Instruction::GetElementPtr:
994 case Instruction::BitCast:
995 Value *s = constant_expr->getOperand(0);
999 }
else if (GlobalVariable *global_variable =
1000 dyn_cast<GlobalVariable>(llvm_value_ptr)) {
1001 if (!GlobalValue::isExternalLinkage(global_variable->getLinkage()))
1004 clang::NamedDecl *named_decl =
DeclForGlobal(global_variable);
1010 if (!global_variable->hasExternalLinkage())
1013 LLDB_LOG(log,
"Found global variable \"{0}\" without metadata",
1014 global_variable->getName());
1019 llvm::StringRef name(named_decl->getName());
1021 clang::ValueDecl *value_decl = dyn_cast<clang::ValueDecl>(named_decl);
1022 if (value_decl ==
nullptr)
1026 m_decl_map->GetTypeSystem()->GetType(value_decl->getType());
1028 const Type *value_type =
nullptr;
1030 if (name.starts_with(
"$")) {
1042 value_type = PointerType::getUnqual(global_variable->getContext());
1044 value_type = global_variable->getType();
1048 std::optional<uint64_t> value_size =
1049 llvm::expectedToOptional(compiler_type.
GetByteSize(target));
1052 std::optional<size_t> opt_alignment = compiler_type.
GetTypeBitAlign(target);
1058 "Type of \"{0}\" is [clang \"{1}\", llvm \"{2}\"] [size {3}, "
1062 PrintType(value_type), *value_size, value_alignment);
1066 llvm_value_ptr, *value_size,
1068 }
else if (isa<llvm::Function>(llvm_value_ptr)) {
1069 LLDB_LOG(log,
"Function pointers aren't handled right now");
1087 LLDB_LOG(log,
"Symbol \"{0}\" had no address", name);
1092 LLDB_LOG(log,
"Found \"{0}\" at {1}", name, symbol_addr);
1094 Type *symbol_type = symbol->getType();
1096 Constant *symbol_addr_int = ConstantInt::get(
m_intptr_ty, symbol_addr,
false);
1098 Value *symbol_addr_ptr =
1099 ConstantExpr::getIntToPtr(symbol_addr_int, symbol_type);
1104 symbol->replaceAllUsesWith(symbol_addr_ptr);
1114 for (
unsigned op_index = 0, num_ops = Old->arg_size();
1115 op_index < num_ops; ++op_index)
1118 m_error_stream.Printf(
"Internal error [IRForTarget]: Couldn't rewrite "
1119 "one of the arguments of a function call.\n");
1130 GlobalVariable *global_variable =
1131 dyn_cast<GlobalVariable>(classlist_reference);
1133 if (!global_variable)
1136 Constant *initializer = global_variable->getInitializer();
1141 if (!initializer->hasName())
1144 StringRef name(initializer->getName());
1149 LLDB_LOG(log,
"Found reference to Objective-C class {0} ({1})", name,
1150 (
unsigned long long)class_ptr);
1155 if (global_variable->use_empty())
1158 SmallVector<LoadInst *, 2> load_instructions;
1160 for (llvm::User *u : global_variable->users()) {
1161 if (LoadInst *load_instruction = dyn_cast<LoadInst>(u))
1162 load_instructions.push_back(load_instruction);
1165 if (load_instructions.empty())
1168 Constant *class_addr = ConstantInt::get(
m_intptr_ty, (uint64_t)class_ptr);
1170 for (LoadInst *load_instruction : load_instructions) {
1171 Constant *class_bitcast =
1172 ConstantExpr::getIntToPtr(class_addr, load_instruction->getType());
1174 load_instruction->replaceAllUsesWith(class_bitcast);
1176 load_instruction->eraseFromParent();
1183 std::vector<CallInst *> calls_to_remove;
1185 for (Instruction &inst : basic_block) {
1186 CallInst *call = dyn_cast<CallInst>(&inst);
1192 bool remove =
false;
1194 llvm::Function *func = call->getCalledFunction();
1196 if (func && func->getName() ==
"__cxa_atexit")
1199 llvm::Value *val = call->getCalledOperand();
1201 if (val && val->getName() ==
"__cxa_atexit")
1205 calls_to_remove.push_back(call);
1208 for (CallInst *ci : calls_to_remove)
1209 ci->eraseFromParent();
1217 for (Instruction &inst : basic_block) {
1218 CallInst *call = dyn_cast<CallInst>(&inst);
1231 for (GlobalVariable &global_var :
m_module->globals()) {
1232 llvm::StringRef global_name = global_var.getName();
1234 LLDB_LOG(log,
"Examining {0}, DeclForGlobalValue returns {1}", global_name,
1237 if (global_name.starts_with(
"OBJC_IVAR")) {
1239 m_error_stream.Format(
"Error [IRForTarget]: Couldn't find Objective-C "
1240 "indirect ivar symbol {0}\n",
1245 }
else if (global_name.contains(
"OBJC_CLASSLIST_REFERENCES_$")) {
1247 m_error_stream.Printf(
"Error [IRForTarget]: Couldn't resolve the class "
1248 "for an Objective-C static method call\n");
1252 }
else if (global_name.contains(
"OBJC_CLASSLIST_SUP_REFS_$")) {
1254 m_error_stream.Printf(
"Error [IRForTarget]: Couldn't resolve the class "
1255 "for an Objective-C static method call\n");
1261 m_error_stream.Format(
"Internal error [IRForTarget]: Couldn't rewrite "
1262 "external variable {0}\n",
1274 GlobalVariable *GV = dyn_cast<GlobalVariable>(V);
1283 Constant *zero(Constant::getNullValue(guard_load->getType()));
1284 guard_load->replaceAllUsesWith(zero);
1285 guard_load->eraseFromParent();
1289 guard_store->eraseFromParent();
1298 for (Instruction &inst : basic_block) {
1300 if (LoadInst *load = dyn_cast<LoadInst>(&inst))
1302 guard_loads.push_back(&inst);
1304 if (StoreInst *store = dyn_cast<StoreInst>(&inst))
1306 guard_stores.push_back(&inst);
1309 for (Instruction *inst : guard_loads)
1312 for (Instruction *inst : guard_stores)
1320 llvm::Function *llvm_function,
1324 SmallVector<User *, 16> users;
1328 for (llvm::User *u : old_constant->users())
1331 for (User *user : users) {
1332 if (Constant *constant = dyn_cast<Constant>(user)) {
1335 if (ConstantExpr *constant_expr = dyn_cast<ConstantExpr>(constant)) {
1336 switch (constant_expr->getOpcode()) {
1338 return llvm::createStringErrorV(
1339 "unhandled constant expression type: \"{0}\".",
1342 case Instruction::BitCast: {
1344 [&value_maker, &entry_instruction_finder, old_constant,
1345 constant_expr](llvm::Function *function) -> llvm::Value * {
1349 if (constant_expr->getOperand(0) != old_constant)
1350 return constant_expr;
1352 return new BitCastInst(
1353 value_maker.
GetValue(function), constant_expr->getType(),
1355 llvm::cast<Instruction>(
1356 entry_instruction_finder.
GetValue(function))
1362 entry_instruction_finder, error_stream))
1365 case Instruction::GetElementPtr: {
1371 [&value_maker, &entry_instruction_finder, old_constant,
1372 constant_expr](llvm::Function *function) -> llvm::Value * {
1373 auto *gep = cast<llvm::GEPOperator>(constant_expr);
1374 Value *ptr = gep->getPointerOperand();
1376 if (ptr == old_constant)
1377 ptr = value_maker.
GetValue(function);
1379 std::vector<Value *> index_vector;
1380 for (Value *operand : gep->indices()) {
1381 if (operand == old_constant)
1382 operand = value_maker.
GetValue(function);
1384 index_vector.push_back(operand);
1387 ArrayRef<Value *> indices(index_vector);
1389 return GetElementPtrInst::Create(
1390 gep->getSourceElementType(), ptr, indices,
"",
1391 llvm::cast<Instruction>(
1392 entry_instruction_finder.
GetValue(function))
1397 get_element_pointer_maker,
1398 entry_instruction_finder, error_stream))
1402 }
else if (ConstantPtrAuth *constant_ptr_auth =
1403 dyn_cast<ConstantPtrAuth>(constant)) {
1406 if (constant_ptr_auth->hasAddressDiscriminator() &&
1407 constant_ptr_auth->getAddrDiscriminator() == old_constant)
1410 return llvm::createStringErrorV(
"unhandled constant type \"{0}\".",
1413 return llvm::createStringErrorV(
"unhandled constant type \"{0}\".",
1416 }
else if (Instruction *inst = llvm::dyn_cast<Instruction>(user)) {
1417 if (llvm_function && inst->getParent()->getParent() != llvm_function)
1418 return llvm::createStringError(
1419 "capturing non-local variables in expressions is unsupported.");
1421 inst->replaceUsesOfWith(
1422 old_constant, value_maker.
GetValue(inst->getParent()->getParent()));
1424 return llvm::createStringErrorV(
"unhandled non-constant type: \"{0}\".",
1429 if (!isa<GlobalValue>(old_constant)) {
1430 old_constant->destroyConstant();
1433 return llvm::Error::success();
1444 LLDB_LOG(log,
"Element arrangement:");
1446 uint32_t num_elements;
1447 uint32_t element_index;
1452 if (!
m_decl_map->GetStructInfo(num_elements, size, alignment))
1455 Function::arg_iterator iter(llvm_function.arg_begin());
1457 if (iter == llvm_function.arg_end()) {
1458 m_error_stream.Printf(
"Internal error [IRForTarget]: Wrapper takes no "
1459 "arguments (should take at least a struct pointer)");
1464 Argument *argument = &*iter;
1466 if (argument->getName() ==
"this") {
1469 if (iter == llvm_function.arg_end()) {
1470 m_error_stream.Printf(
"Internal error [IRForTarget]: Wrapper takes only "
1471 "'this' argument (should take a struct pointer "
1478 }
else if (argument->getName() ==
"self") {
1481 if (iter == llvm_function.arg_end()) {
1482 m_error_stream.Printf(
"Internal error [IRForTarget]: Wrapper takes only "
1483 "'self' argument (should take '_cmd' and a struct "
1489 if (iter->getName() !=
"_cmd") {
1490 m_error_stream.Format(
"Internal error [IRForTarget]: Wrapper takes '{0}' "
1491 "after 'self' argument (should take '_cmd')",
1499 if (iter == llvm_function.arg_end()) {
1500 m_error_stream.Printf(
"Internal error [IRForTarget]: Wrapper takes only "
1501 "'self' and '_cmd' arguments (should take a struct "
1510 if (argument->getName() !=
"$__lldb_arg") {
1511 m_error_stream.Format(
"Internal error [IRForTarget]: Wrapper takes an "
1512 "argument named '{0}' instead of the struct pointer",
1513 argument->getName());
1520 BasicBlock &entry_block(llvm_function.getEntryBlock());
1521 Instruction *FirstEntryInstruction(&*entry_block.getFirstNonPHIOrDbg());
1523 if (!FirstEntryInstruction) {
1524 m_error_stream.Printf(
"Internal error [IRForTarget]: Couldn't find the "
1525 "first instruction in the wrapper for use in "
1531 LLVMContext &context(
m_module->getContext());
1532 IntegerType *offset_type(Type::getInt32Ty(context));
1536 "Internal error [IRForTarget]: Couldn't produce an offset type");
1541 for (element_index = 0; element_index < num_elements; ++element_index) {
1542 const clang::NamedDecl *decl =
nullptr;
1543 Value *value =
nullptr;
1547 if (!
m_decl_map->GetStructElement(decl, value, offset, name,
1550 "Internal error [IRForTarget]: Structure information is incomplete");
1555 LLDB_LOG(log,
" \"{0}\" (\"{1}\") placed at {2}", name,
1556 decl->getNameAsString(), offset);
1562 [
this, name, offset_type, offset, argument,
1563 value](llvm::Function *function) -> llvm::Value * {
1570 llvm::Instruction *entry_instruction = llvm::cast<Instruction>(
1573 Type *int8Ty = Type::getInt8Ty(function->getContext());
1574 ConstantInt *offset_int(
1575 ConstantInt::get(offset_type, offset,
true));
1576 GetElementPtrInst *get_element_ptr =
1577 GetElementPtrInst::Create(int8Ty, argument, offset_int,
"",
1578 entry_instruction->getIterator());
1582 new LoadInst(value->getType(), get_element_ptr,
"",
1583 entry_instruction->getIterator());
1587 return get_element_ptr;
1591 if (Constant *constant = dyn_cast<Constant>(value)) {
1598 }
else if (Instruction *instruction = dyn_cast<Instruction>(value)) {
1599 if (instruction->getParent()->getParent() != &llvm_function) {
1600 m_error_stream.PutCString(
"error: Capturing non-local variables in "
1601 "expressions is unsupported.\n");
1604 value->replaceAllUsesWith(
1605 body_result_maker.GetValue(instruction->getParent()->getParent()));
1607 LLDB_LOG(log,
"Unhandled non-constant type: \"{0}\"",
1612 if (GlobalVariable *var = dyn_cast<GlobalVariable>(value))
1613 var->eraseFromParent();
1617 LLDB_LOG(log,
"Total structure [align {0}, size {1}]", (int64_t)alignment,
1633 raw_string_ostream oss(s);
1637 LLDB_LOG(log,
"Module as passed in to IRForTarget: \n\"{0}\"", s);
1640 Function *
const main_function =
1647 m_error_stream.Format(
"Internal error [IRForTarget]: Couldn't find wrapper "
1648 "'{0}' in the module",
1654 if (main_function) {
1656 LLDB_LOG(log,
"Couldn't fix the linkage for the function");
1663 if (main_function) {
1665 LLDB_LOG(log,
"CreateResultVariable() failed");
1675 raw_string_ostream oss(s);
1679 LLDB_LOG(log,
"Module after creating the result variable: \n\"{0}\"", s);
1682 for (llvm::Function &function : *
m_module) {
1683 for (BasicBlock &bb : function) {
1685 LLDB_LOG(log,
"RemoveGuards() failed");
1693 LLDB_LOG(log,
"RewritePersistentAllocs() failed");
1702 LLDB_LOG(log,
"RemoveCXAAtExit() failed");
1713 LLDB_LOG(log,
"RewriteObjCConstStrings() failed");
1720 for (llvm::Function &function : *
m_module) {
1721 for (llvm::BasicBlock &bb : function) {
1723 LLDB_LOG(log,
"RewriteObjCSelectors() failed");
1733 for (llvm::Function &function : *
m_module) {
1734 for (BasicBlock &bb : function) {
1736 LLDB_LOG(log,
"ResolveCalls() failed");
1746 if (main_function) {
1748 LLDB_LOG(log,
"ResolveExternals() failed");
1756 LLDB_LOG(log,
"ReplaceVariables() failed");
1765 if (llvm::Error
error =
1768 "InsertPointerSigningFixups() failed: {0}");
1774 raw_string_ostream oss(s);
1778 LLDB_LOG(log,
"Module after preparing for execution: \n\"{0}\"", s);
static llvm::raw_ostream & error(Stream &strm)
static std::string PrintValue(const Value *value)
static bool isGuardVariableSymbol(llvm::StringRef mangled_symbol, bool check_ms_abi=true)
Returns true iff the mangled symbol is for a static guard variable.
static void ExciseGuardStore(Instruction *guard_store)
static llvm::Value * FindEntryInstruction(llvm::Function *function)
static bool IsObjCSelectorRef(Value *value)
SmallVector< Instruction *, 2 > InstrList
static bool isGuardVariableRef(Value *V)
static std::string PrintType(const llvm::Type *type)
static std::string PrintValue(const Value *value, bool truncate=false)
static std::string PrintType(const Type *type, bool truncate=false)
#define LLDB_LOG(log,...)
The LLDB_LOG* macros defined below are the way to emit log messages.
#define LLDB_LOG_ERROR(log, error,...)
FunctionValueCache(Maker const &maker)
llvm::Value * GetValue(llvm::Function *function)
std::function< llvm::Value *(llvm::Function *)> Maker
FunctionValueMap m_values
static llvm::Error UnfoldConstant(llvm::Constant *old_constant, llvm::Function *llvm_function, FunctionValueCache &value_maker, FunctionValueCache &entry_instruction_finder, lldb_private::Stream &error_stream)
UnfoldConstant operates on a constant [Old] which has just been replaced with a value [New].
lldb_private::ConstString m_func_name
The name of the function to translate.
IRForTarget(lldb_private::ClangExpressionDeclMap *decl_map, bool resolve_vars, lldb_private::IRExecutionUnit &execution_unit, lldb_private::Stream &error_stream, lldb_private::ExecutionPolicy execution_policy, const char *func_name="$__lldb_expr")
Constructor.
bool MaybeHandleVariable(llvm::Value *value)
A function-level pass to find all external variables and functions used in the IR.
bool FixFunctionLinkage(llvm::Function &llvm_function)
Ensures that the current function's linkage is set to external.
lldb_private::IRExecutionUnit & m_execution_unit
The execution unit containing the IR being created.
bool CreateResultVariable(llvm::Function &llvm_function)
The top-level pass implementation.
llvm::Module * m_module
The module being processed, or NULL if that has not been determined yet.
bool HandleSymbol(llvm::Value *symbol)
Handle a single externally-defined symbol.
bool RewriteObjCConstStrings()
The top-level pass implementation.
bool ResolveCalls(llvm::BasicBlock &basic_block)
Resolve variable references in calls to external functions.
bool m_result_is_pointer
True if the function's result in the AST is a pointer (see comments in ASTResultSynthesizer::Synthesi...
bool RewriteObjCConstString(llvm::GlobalVariable *NSStr, llvm::GlobalVariable *CStr)
A module-level pass to find Objective-C constant strings and transform them to calls to CFStringCreat...
bool RemoveGuards(llvm::BasicBlock &basic_block)
The top-level pass implementation.
lldb_private::Stream & m_error_stream
The stream on which errors should be printed.
bool HandleObjCClass(llvm::Value *classlist_reference)
Handle a single externally-defined Objective-C class.
bool m_resolve_vars
True if external variable references and persistent variable references should be resolved.
static clang::NamedDecl * DeclForGlobal(const llvm::GlobalValue *global_val, llvm::Module *module)
A function-level pass to take the generated global value $__lldb_expr_result and make it into a persi...
lldb_private::TypeFromParser m_result_type
The type of the result variable.
llvm::FunctionCallee m_CFStringCreateWithBytes
The address of the function CFStringCreateWithBytes, cast to the appropriate function pointer type.
bool MaybeHandleCallArguments(llvm::CallInst *call_inst)
Handle all the arguments to a function call.
lldb_private::ExecutionPolicy m_policy
bool runOnModule(llvm::Module &llvm_module)
Run this IR transformer on a single module.
bool RewriteObjCSelectors(llvm::BasicBlock &basic_block)
The top-level pass implementation.
lldb_private::ConstString m_result_name
The name of the result variable ($0, $1, ...)
lldb_private::ClangExpressionDeclMap * m_decl_map
The DeclMap containing the Decls.
bool RemoveCXAAtExit(llvm::BasicBlock &basic_block)
Remove calls to __cxa_atexit, which should never be generated by expressions.
bool RewritePersistentAllocs(llvm::BasicBlock &basic_block)
The top-level pass implementation.
bool RewritePersistentAlloc(llvm::Instruction *persistent_alloc)
A basic block-level pass to find all newly-declared persistent variables and register them with the C...
void TurnGuardLoadIntoZero(llvm::Instruction *guard_load)
A basic block-level pass to excise guard variables from the code.
FunctionValueCache m_entry_instruction_finder
const llvm::DataLayout * m_target_data
The target data for the module being processed, or nullptr if there is no module.
llvm::FunctionCallee m_sel_registerName
The address of the function sel_registerName, cast to the appropriate function pointer type.
llvm::IntegerType * m_intptr_ty
The type of an integer large enough to hold a pointer.
bool RewriteObjCSelector(llvm::Instruction *selector_load)
A basic block-level pass to find all Objective-C method calls and rewrite them to use sel_registerNam...
bool ResolveExternals(llvm::Function &llvm_function)
The top-level pass implementation.
bool ReplaceVariables(llvm::Function &llvm_function)
A function-level pass to make all external variable references point at the correct offsets from the ...
"lldb/Expression/ClangExpressionDeclMap.h" Manages named entities that are defined in LLDB's debug in...
Generic representation of a type in a programming language.
CompilerType GetPointerType() const
Return a new CompilerType that is a pointer to this type.
llvm::Expected< uint64_t > GetByteSize(ExecutionContextScope *exe_scope) const
Return the size of the type in bytes.
std::optional< size_t > GetTypeBitAlign(ExecutionContextScope *exe_scope) const
A uniqued constant string class.
"lldb/Expression/IRExecutionUnit.h" Contains the IR and, optionally, JIT- compiled code for a module.
const char * GetData() const
A stream class that can stream formatted output to a file.
#define LLDB_INVALID_ADDRESS
Error InjectPointerSigningFixupCode(llvm::Module &M, ExecutionPolicy execution_policy)
TaggedASTType< 0 > TypeFromParser
ExecutionPolicy
Expression execution policies.
std::shared_ptr< lldb_private::Target > TargetSP
static clang::QualType GetQualType(const CompilerType &ct)