blob: 9ac4a934a3db7809cf08300d4b98dacf6a9fb896 [file] [log] [blame]
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001//===-- IRForTarget.cpp -------------------------------------------*- C++ -*-===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10#include "lldb/Expression/IRForTarget.h"
11
12#include "llvm/Support/raw_ostream.h"
Sean Callanan2a8c3382011-04-14 02:01:31 +000013#include "llvm/Constants.h"
Sean Callanan5cf4a1c2010-07-03 01:35:46 +000014#include "llvm/InstrTypes.h"
Sean Callanan8bce6652010-07-13 21:41:46 +000015#include "llvm/Instructions.h"
Sean Callanan3cb1fd82010-09-28 23:55:00 +000016#include "llvm/Intrinsics.h"
Sean Callanan5cf4a1c2010-07-03 01:35:46 +000017#include "llvm/Module.h"
Sean Callanan8bce6652010-07-13 21:41:46 +000018#include "llvm/Target/TargetData.h"
Sean Callanan82b74c82010-08-12 01:56:52 +000019#include "llvm/ValueSymbolTable.h"
Sean Callanan8bce6652010-07-13 21:41:46 +000020
21#include "clang/AST/ASTContext.h"
Sean Callanan5cf4a1c2010-07-03 01:35:46 +000022
Greg Clayton8de27c72010-10-15 22:48:33 +000023#include "lldb/Core/ConstString.h"
Sean Callanan5cf4a1c2010-07-03 01:35:46 +000024#include "lldb/Core/dwarf.h"
25#include "lldb/Core/Log.h"
26#include "lldb/Core/Scalar.h"
27#include "lldb/Core/StreamString.h"
28#include "lldb/Expression/ClangExpressionDeclMap.h"
Sean Callanan696cf5f2011-05-07 01:06:41 +000029#include "lldb/Symbol/ClangASTContext.h"
Sean Callanan5cf4a1c2010-07-03 01:35:46 +000030
31#include <map>
32
33using namespace llvm;
34
Sean Callanan3351dac2010-08-18 18:50:51 +000035static char ID;
36
Greg Clayton3c7feb42010-11-19 01:05:25 +000037IRForTarget::IRForTarget (lldb_private::ClangExpressionDeclMap *decl_map,
38 bool resolve_vars,
Sean Callanan696cf5f2011-05-07 01:06:41 +000039 lldb::ClangExpressionVariableSP &const_result,
Sean Callanan97c924e2011-01-27 01:07:04 +000040 lldb_private::Stream *error_stream,
Greg Clayton3c7feb42010-11-19 01:05:25 +000041 const char *func_name) :
Sean Callanan47a5c4c2010-09-23 03:01:22 +000042 ModulePass(ID),
Stephen Wilsondbeb3e12011-04-11 19:41:40 +000043 m_resolve_vars(resolve_vars),
44 m_func_name(func_name),
Sean Callanan8bce6652010-07-13 21:41:46 +000045 m_decl_map(decl_map),
Sean Callanan6ba533e2010-11-17 23:00:36 +000046 m_CFStringCreateWithBytes(NULL),
Sean Callanan65dafa82010-08-27 01:01:44 +000047 m_sel_registerName(NULL),
Stephen Wilsondbeb3e12011-04-11 19:41:40 +000048 m_error_stream(error_stream),
Sean Callanan6a925532011-01-13 08:53:35 +000049 m_has_side_effects(false),
Sean Callanan696cf5f2011-05-07 01:06:41 +000050 m_result_store(NULL),
51 m_result_is_pointer(false),
52 m_const_result(const_result)
Sean Callanan5cf4a1c2010-07-03 01:35:46 +000053{
54}
55
Sean Callanan771131d2010-09-30 21:18:25 +000056/* Handy utility functions used at several places in the code */
Sean Callanana48fe162010-08-11 03:57:18 +000057
58static std::string
Greg Clayton3c7feb42010-11-19 01:05:25 +000059PrintValue(const Value *value, bool truncate = false)
Sean Callanana48fe162010-08-11 03:57:18 +000060{
61 std::string s;
62 raw_string_ostream rso(s);
Greg Clayton3c7feb42010-11-19 01:05:25 +000063 value->print(rso);
Sean Callanana48fe162010-08-11 03:57:18 +000064 rso.flush();
65 if (truncate)
66 s.resize(s.length() - 1);
67 return s;
68}
69
Sean Callanan771131d2010-09-30 21:18:25 +000070static std::string
Greg Clayton3c7feb42010-11-19 01:05:25 +000071PrintType(const Type *type, bool truncate = false)
Sean Callanan771131d2010-09-30 21:18:25 +000072{
73 std::string s;
74 raw_string_ostream rso(s);
Greg Clayton3c7feb42010-11-19 01:05:25 +000075 type->print(rso);
Sean Callanan771131d2010-09-30 21:18:25 +000076 rso.flush();
77 if (truncate)
78 s.resize(s.length() - 1);
79 return s;
80}
81
Sean Callanan5cf4a1c2010-07-03 01:35:46 +000082IRForTarget::~IRForTarget()
83{
84}
85
Sean Callanan82b74c82010-08-12 01:56:52 +000086bool
Sean Callanan05a5a1b2010-12-16 03:17:46 +000087IRForTarget::HasSideEffects (llvm::Module &llvm_module,
88 llvm::Function &llvm_function)
89{
90 llvm::Function::iterator bbi;
91 BasicBlock::iterator ii;
Sean Callanan696cf5f2011-05-07 01:06:41 +000092
Sean Callanan05a5a1b2010-12-16 03:17:46 +000093 for (bbi = llvm_function.begin();
94 bbi != llvm_function.end();
95 ++bbi)
96 {
97 BasicBlock &basic_block = *bbi;
98
99 for (ii = basic_block.begin();
100 ii != basic_block.end();
101 ++ii)
102 {
103 switch (ii->getOpcode())
104 {
105 default:
106 return true;
107 case Instruction::Store:
108 {
109 StoreInst *store_inst = dyn_cast<StoreInst>(ii);
110
111 Value *store_ptr = store_inst->getPointerOperand();
112
Sean Callanan696cf5f2011-05-07 01:06:41 +0000113 std::string ptr_name;
114
115 if (store_ptr->hasName())
116 ptr_name = store_ptr->getNameStr();
117
118 if (isa <AllocaInst> (store_ptr))
Sean Callanan05a5a1b2010-12-16 03:17:46 +0000119 break;
Sean Callanan696cf5f2011-05-07 01:06:41 +0000120
121 if (ptr_name.find("$__lldb_expr_result") != std::string::npos)
122 {
123 if (ptr_name.find("GV") == std::string::npos)
124 m_result_store = store_inst;
125 }
126 else
127 {
128 return true;
129 }
130
131 break;
Sean Callanan05a5a1b2010-12-16 03:17:46 +0000132 }
133 case Instruction::Load:
134 case Instruction::Alloca:
135 case Instruction::GetElementPtr:
Sean Callanan696cf5f2011-05-07 01:06:41 +0000136 case Instruction::BitCast:
Sean Callanan05a5a1b2010-12-16 03:17:46 +0000137 case Instruction::Ret:
Sean Callanan696cf5f2011-05-07 01:06:41 +0000138 case Instruction::ICmp:
139 case Instruction::Br:
Sean Callanan05a5a1b2010-12-16 03:17:46 +0000140 break;
141 }
142 }
143 }
144
145 return false;
146}
147
Sean Callanan696cf5f2011-05-07 01:06:41 +0000148clang::NamedDecl *
149IRForTarget::DeclForGlobal (llvm::Module &module, GlobalValue *global_val)
150{
151 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
152
153 NamedMDNode *named_metadata = module.getNamedMetadata("clang.global.decl.ptrs");
154
155 if (!named_metadata)
156 return NULL;
157
158 unsigned num_nodes = named_metadata->getNumOperands();
159 unsigned node_index;
160
161 for (node_index = 0;
162 node_index < num_nodes;
163 ++node_index)
164 {
165 MDNode *metadata_node = named_metadata->getOperand(node_index);
166
167 if (!metadata_node)
168 return NULL;
169
170 if (metadata_node->getNumOperands() != 2)
171 continue;
172
173 if (metadata_node->getOperand(0) != global_val)
174 continue;
175
176 ConstantInt *constant_int = dyn_cast<ConstantInt>(metadata_node->getOperand(1));
177
178 if (!constant_int)
179 return NULL;
180
181 uintptr_t ptr = constant_int->getZExtValue();
182
183 return reinterpret_cast<clang::NamedDecl *>(ptr);
184 }
185
186 return NULL;
187}
188
Sean Callanan05a5a1b2010-12-16 03:17:46 +0000189void
190IRForTarget::MaybeSetConstantResult (llvm::Constant *initializer,
191 const lldb_private::ConstString &name,
192 lldb_private::TypeFromParser type)
193{
Sean Callanan696cf5f2011-05-07 01:06:41 +0000194 if (llvm::ConstantExpr *init_expr = dyn_cast<llvm::ConstantExpr>(initializer))
195 {
196 switch (init_expr->getOpcode())
197 {
198 default:
199 return;
200 case Instruction::IntToPtr:
201 MaybeSetConstantResult (init_expr->getOperand(0), name, type);
202 return;
203 }
204 }
205 else if (llvm::ConstantInt *init_int = dyn_cast<llvm::ConstantInt>(initializer))
206 {
207 m_const_result = m_decl_map->BuildIntegerVariable(name, type, init_int->getValue());
208 }
209}
210
211void
212IRForTarget::MaybeSetCastResult (llvm::Module &llvm_module, lldb_private::TypeFromParser type)
213{
214 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
215
216 if (!m_result_store)
Sean Callanan05a5a1b2010-12-16 03:17:46 +0000217 return;
218
Sean Callanan696cf5f2011-05-07 01:06:41 +0000219 LoadInst *original_load = NULL;
220
221 for (llvm::Value *current_value = m_result_store->getValueOperand(), *next_value;
222 current_value != NULL;
223 current_value = next_value)
Sean Callanan05a5a1b2010-12-16 03:17:46 +0000224 {
Sean Callanan696cf5f2011-05-07 01:06:41 +0000225 CastInst *cast_inst = dyn_cast<CastInst>(current_value);
226 LoadInst *load_inst = dyn_cast<LoadInst>(current_value);
227
228 if (cast_inst)
229 {
230 next_value = cast_inst->getOperand(0);
231 }
232 else if(load_inst)
233 {
234 if (isa<LoadInst>(load_inst->getPointerOperand()))
235 {
236 next_value = load_inst->getPointerOperand();
237 }
238 else
239 {
240 original_load = load_inst;
241 break;
242 }
243 }
244 else
245 {
246 return;
247 }
Sean Callanan05a5a1b2010-12-16 03:17:46 +0000248 }
Sean Callanan696cf5f2011-05-07 01:06:41 +0000249
250 Value *loaded_value = original_load->getPointerOperand();
251 GlobalVariable *loaded_global = dyn_cast<GlobalVariable>(loaded_value);
252
253 if (!loaded_global)
254 return;
255
256 clang::NamedDecl *loaded_decl = DeclForGlobal(llvm_module, loaded_global);
257
258 if (!loaded_decl)
259 return;
260
261 clang::VarDecl *loaded_var = dyn_cast<clang::VarDecl>(loaded_decl);
262
263 if (!loaded_var)
264 return;
265
266 if (log)
267 {
268 lldb_private::StreamString type_desc_stream;
269 type.DumpTypeDescription(&type_desc_stream);
270
271 log->Printf("Type to cast variable to: \"%s\"", type_desc_stream.GetString().c_str());
272 }
273
274 m_const_result = m_decl_map->BuildCastVariable(m_result_name, loaded_var, type);
Sean Callanan05a5a1b2010-12-16 03:17:46 +0000275}
276
277bool
Greg Clayton3c7feb42010-11-19 01:05:25 +0000278IRForTarget::CreateResultVariable (llvm::Module &llvm_module, llvm::Function &llvm_function)
Sean Callanan82b74c82010-08-12 01:56:52 +0000279{
Greg Claytone005f2c2010-11-06 01:53:30 +0000280 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
Sean Callanan82b74c82010-08-12 01:56:52 +0000281
Sean Callanane8a59a82010-09-13 21:34:21 +0000282 if (!m_resolve_vars)
283 return true;
284
285 // Find the result variable. If it doesn't exist, we can give up right here.
Sean Callanan3c9c5eb2010-09-21 00:44:12 +0000286
Greg Clayton8de27c72010-10-15 22:48:33 +0000287 ValueSymbolTable& value_symbol_table = llvm_module.getValueSymbolTable();
Sean Callanan3c9c5eb2010-09-21 00:44:12 +0000288
289 const char *result_name = NULL;
290
291 for (ValueSymbolTable::iterator vi = value_symbol_table.begin(), ve = value_symbol_table.end();
292 vi != ve;
293 ++vi)
294 {
Sean Callanan6a925532011-01-13 08:53:35 +0000295 if (strstr(vi->first(), "$__lldb_expr_result_ptr") &&
296 !strstr(vi->first(), "GV"))
297 {
298 result_name = vi->first();
299 m_result_is_pointer = true;
300 break;
301 }
302
Greg Clayton8de27c72010-10-15 22:48:33 +0000303 if (strstr(vi->first(), "$__lldb_expr_result") &&
Sean Callananc04743d2010-09-28 21:13:03 +0000304 !strstr(vi->first(), "GV"))
305 {
Sean Callanan3c9c5eb2010-09-21 00:44:12 +0000306 result_name = vi->first();
Sean Callanan6a925532011-01-13 08:53:35 +0000307 m_result_is_pointer = false;
Sean Callananc04743d2010-09-28 21:13:03 +0000308 break;
309 }
Sean Callanan3c9c5eb2010-09-21 00:44:12 +0000310 }
311
312 if (!result_name)
313 {
314 if (log)
315 log->PutCString("Couldn't find result variable");
316
317 return true;
318 }
319
Sean Callananc04743d2010-09-28 21:13:03 +0000320 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +0000321 log->Printf("Result name: \"%s\"", result_name);
Sean Callananc04743d2010-09-28 21:13:03 +0000322
Greg Clayton8de27c72010-10-15 22:48:33 +0000323 Value *result_value = llvm_module.getNamedValue(result_name);
Sean Callanan82b74c82010-08-12 01:56:52 +0000324
325 if (!result_value)
326 {
327 if (log)
Sean Callanan3c9c5eb2010-09-21 00:44:12 +0000328 log->PutCString("Result variable had no data");
Sean Callanan6a925532011-01-13 08:53:35 +0000329
Sean Callanan97c924e2011-01-27 01:07:04 +0000330 if (m_error_stream)
331 m_error_stream->Printf("Internal error [IRForTarget]: Result variable's name (%s) exists, but not its definition\n", result_name);
332
Sean Callanan3c9c5eb2010-09-21 00:44:12 +0000333 return false;
Sean Callanan82b74c82010-08-12 01:56:52 +0000334 }
Sean Callanane8a59a82010-09-13 21:34:21 +0000335
Sean Callanan82b74c82010-08-12 01:56:52 +0000336 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +0000337 log->Printf("Found result in the IR: \"%s\"", PrintValue(result_value, false).c_str());
Sean Callanan82b74c82010-08-12 01:56:52 +0000338
339 GlobalVariable *result_global = dyn_cast<GlobalVariable>(result_value);
340
341 if (!result_global)
342 {
343 if (log)
344 log->PutCString("Result variable isn't a GlobalVariable");
Sean Callanan97c924e2011-01-27 01:07:04 +0000345
346 if (m_error_stream)
347 m_error_stream->Printf("Internal error [IRForTarget]: Result variable (%s) is defined, but is not a global variable\n", result_name);
348
Sean Callanan82b74c82010-08-12 01:56:52 +0000349 return false;
350 }
351
Sean Callanan696cf5f2011-05-07 01:06:41 +0000352 clang::NamedDecl *result_decl = DeclForGlobal (llvm_module, result_global);
353 if (!result_decl)
Sean Callanan82b74c82010-08-12 01:56:52 +0000354 {
355 if (log)
Sean Callanan696cf5f2011-05-07 01:06:41 +0000356 log->PutCString("Result variable doesn't have a corresponding Decl");
Sean Callanan82b74c82010-08-12 01:56:52 +0000357
Sean Callanan97c924e2011-01-27 01:07:04 +0000358 if (m_error_stream)
Sean Callanan696cf5f2011-05-07 01:06:41 +0000359 m_error_stream->Printf("Internal error [IRForTarget]: Result variable (%s) does not have a corresponding Clang entity\n", result_name);
Sean Callanan97c924e2011-01-27 01:07:04 +0000360
Sean Callanan82b74c82010-08-12 01:56:52 +0000361 return false;
362 }
Sean Callanan82b74c82010-08-12 01:56:52 +0000363
Sean Callanan696cf5f2011-05-07 01:06:41 +0000364 if (log)
Sean Callanan82b74c82010-08-12 01:56:52 +0000365 {
Sean Callanan696cf5f2011-05-07 01:06:41 +0000366 std::string decl_desc_str;
367 raw_string_ostream decl_desc_stream(decl_desc_str);
368 result_decl->print(decl_desc_stream);
369 decl_desc_stream.flush();
Sean Callanan82b74c82010-08-12 01:56:52 +0000370
Sean Callanan696cf5f2011-05-07 01:06:41 +0000371 log->Printf("Found result decl: \"%s\"", decl_desc_str.c_str());
Sean Callanan82b74c82010-08-12 01:56:52 +0000372 }
373
Sean Callanan696cf5f2011-05-07 01:06:41 +0000374 clang::VarDecl *result_var = dyn_cast<clang::VarDecl>(result_decl);
375 if (!result_var)
Sean Callanan82b74c82010-08-12 01:56:52 +0000376 {
377 if (log)
Sean Callanan696cf5f2011-05-07 01:06:41 +0000378 log->PutCString("Result variable Decl isn't a VarDecl");
Sean Callanan97c924e2011-01-27 01:07:04 +0000379
380 if (m_error_stream)
Sean Callanan696cf5f2011-05-07 01:06:41 +0000381 m_error_stream->Printf("Internal error [IRForTarget]: Result variable (%s)'s corresponding Clang entity isn't a variable\n", result_name);
Sean Callanan97c924e2011-01-27 01:07:04 +0000382
Sean Callanan82b74c82010-08-12 01:56:52 +0000383 return false;
384 }
Sean Callanan82b74c82010-08-12 01:56:52 +0000385
Sean Callanan82b74c82010-08-12 01:56:52 +0000386 // Get the next available result name from m_decl_map and create the persistent
387 // variable for it
388
Sean Callanan6a925532011-01-13 08:53:35 +0000389 lldb_private::TypeFromParser result_decl_type;
390
Sean Callanan696cf5f2011-05-07 01:06:41 +0000391 // If the result is an Lvalue, it is emitted as a pointer; see
392 // ASTResultSynthesizer::SynthesizeBodyResult.
Sean Callanan6a925532011-01-13 08:53:35 +0000393 if (m_result_is_pointer)
394 {
Sean Callanan696cf5f2011-05-07 01:06:41 +0000395 clang::QualType pointer_qual_type = result_var->getType();
Sean Callanand5b3c352011-01-27 04:42:51 +0000396 const clang::Type *pointer_type = pointer_qual_type.getTypePtr();
397 const clang::PointerType *pointer_pointertype = dyn_cast<clang::PointerType>(pointer_type);
Sean Callanan6a925532011-01-13 08:53:35 +0000398
399 if (!pointer_pointertype)
400 {
401 if (log)
402 log->PutCString("Expected result to have pointer type, but it did not");
Sean Callanan97c924e2011-01-27 01:07:04 +0000403
404 if (m_error_stream)
405 m_error_stream->Printf("Internal error [IRForTarget]: Lvalue result (%s) is not a pointer variable\n", result_name);
406
Sean Callanan6a925532011-01-13 08:53:35 +0000407 return false;
408 }
409
410 clang::QualType element_qual_type = pointer_pointertype->getPointeeType();
411
412 result_decl_type = lldb_private::TypeFromParser(element_qual_type.getAsOpaquePtr(),
413 &result_decl->getASTContext());
414 }
415 else
416 {
Sean Callanan696cf5f2011-05-07 01:06:41 +0000417 result_decl_type = lldb_private::TypeFromParser(result_var->getType().getAsOpaquePtr(),
Sean Callanan6a925532011-01-13 08:53:35 +0000418 &result_decl->getASTContext());
419 }
420
Sean Callanan696cf5f2011-05-07 01:06:41 +0000421 if (log)
422 {
423 lldb_private::StreamString type_desc_stream;
424 result_decl_type.DumpTypeDescription(&type_desc_stream);
425
426 log->Printf("Result decl type: \"%s\"", type_desc_stream.GetString().c_str());
427 }
428
Sean Callanan6a925532011-01-13 08:53:35 +0000429 m_result_name = m_decl_map->GetPersistentResultName();
Sean Callanan82b74c82010-08-12 01:56:52 +0000430
431 if (log)
Sean Callanan6a925532011-01-13 08:53:35 +0000432 log->Printf("Creating a new result global: \"%s\"", m_result_name.GetCString());
Sean Callanan82b74c82010-08-12 01:56:52 +0000433
434 // Construct a new result global and set up its metadata
435
Greg Clayton8de27c72010-10-15 22:48:33 +0000436 GlobalVariable *new_result_global = new GlobalVariable(llvm_module,
Sean Callanan82b74c82010-08-12 01:56:52 +0000437 result_global->getType()->getElementType(),
438 false, /* not constant */
439 GlobalValue::ExternalLinkage,
440 NULL, /* no initializer */
Sean Callanan6a925532011-01-13 08:53:35 +0000441 m_result_name.GetCString ());
Sean Callanan82b74c82010-08-12 01:56:52 +0000442
443 // It's too late in compilation to create a new VarDecl for this, but we don't
444 // need to. We point the metadata at the old VarDecl. This creates an odd
445 // anomaly: a variable with a Value whose name is something like $0 and a
Greg Clayton8de27c72010-10-15 22:48:33 +0000446 // Decl whose name is $__lldb_expr_result. This condition is handled in
Sean Callanan82b74c82010-08-12 01:56:52 +0000447 // ClangExpressionDeclMap::DoMaterialize, and the name of the variable is
448 // fixed up.
449
Sean Callanan696cf5f2011-05-07 01:06:41 +0000450 ConstantInt *new_constant_int = ConstantInt::get(llvm::Type::getInt64Ty(llvm_module.getContext()),
451 reinterpret_cast<uint64_t>(result_decl),
Sean Callanan82b74c82010-08-12 01:56:52 +0000452 false);
453
454 llvm::Value* values[2];
455 values[0] = new_result_global;
456 values[1] = new_constant_int;
457
Sean Callanan0de254a2011-05-15 22:34:38 +0000458 ArrayRef<Value*> value_ref(values, 2);
459
460 MDNode *persistent_global_md = MDNode::get(llvm_module.getContext(), value_ref);
Sean Callanan696cf5f2011-05-07 01:06:41 +0000461 NamedMDNode *named_metadata = llvm_module.getNamedMetadata("clang.global.decl.ptrs");
Sean Callanan82b74c82010-08-12 01:56:52 +0000462 named_metadata->addOperand(persistent_global_md);
463
464 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +0000465 log->Printf("Replacing \"%s\" with \"%s\"",
Sean Callanan2e2db532010-09-07 22:43:19 +0000466 PrintValue(result_global).c_str(),
Sean Callanan82b74c82010-08-12 01:56:52 +0000467 PrintValue(new_result_global).c_str());
Sean Callanan2e2db532010-09-07 22:43:19 +0000468
469 if (result_global->hasNUses(0))
470 {
471 // We need to synthesize a store for this variable, because otherwise
472 // there's nothing to put into its equivalent persistent variable.
Sean Callanan82b74c82010-08-12 01:56:52 +0000473
Greg Clayton8de27c72010-10-15 22:48:33 +0000474 BasicBlock &entry_block(llvm_function.getEntryBlock());
Sean Callanan2e2db532010-09-07 22:43:19 +0000475 Instruction *first_entry_instruction(entry_block.getFirstNonPHIOrDbg());
476
477 if (!first_entry_instruction)
478 return false;
479
480 if (!result_global->hasInitializer())
481 {
482 if (log)
483 log->Printf("Couldn't find initializer for unused variable");
Sean Callanan97c924e2011-01-27 01:07:04 +0000484
485 if (m_error_stream)
486 m_error_stream->Printf("Internal error [IRForTarget]: Result variable (%s) has no writes and no initializer\n", result_name);
487
Sean Callanan2e2db532010-09-07 22:43:19 +0000488 return false;
489 }
490
491 Constant *initializer = result_global->getInitializer();
Sean Callanan05a5a1b2010-12-16 03:17:46 +0000492
493 // Here we write the initializer into a result variable assuming it
494 // can be computed statically.
495
496 if (!m_has_side_effects)
497 {
498 MaybeSetConstantResult (initializer,
Sean Callanan6a925532011-01-13 08:53:35 +0000499 m_result_name,
Sean Callanan05a5a1b2010-12-16 03:17:46 +0000500 result_decl_type);
501 }
Sean Callanan2e2db532010-09-07 22:43:19 +0000502
Greg Clayton2c344ca2011-02-05 02:28:58 +0000503 StoreInst *synthesized_store = new StoreInst(initializer,
504 new_result_global,
505 first_entry_instruction);
Sean Callanan2e2db532010-09-07 22:43:19 +0000506
507 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +0000508 log->Printf("Synthesized result store \"%s\"\n", PrintValue(synthesized_store).c_str());
Sean Callanan2e2db532010-09-07 22:43:19 +0000509 }
510 else
511 {
Sean Callanan696cf5f2011-05-07 01:06:41 +0000512 if (!m_has_side_effects && lldb_private::ClangASTContext::IsPointerType (result_decl_type.GetOpaqueQualType()))
513 {
514 MaybeSetCastResult (llvm_module, result_decl_type);
515 }
516
Sean Callanan2e2db532010-09-07 22:43:19 +0000517 result_global->replaceAllUsesWith(new_result_global);
518 }
Sean Callanan696cf5f2011-05-07 01:06:41 +0000519
520 if (!m_const_result)
521 m_decl_map->AddPersistentVariable(result_decl,
522 m_result_name,
523 result_decl_type,
524 true,
525 m_result_is_pointer);
Sean Callanan2e2db532010-09-07 22:43:19 +0000526
Sean Callanan82b74c82010-08-12 01:56:52 +0000527 result_global->eraseFromParent();
528
529 return true;
530}
531
Greg Clayton3c7feb42010-11-19 01:05:25 +0000532static void DebugUsers(lldb::LogSP &log, Value *value, uint8_t depth)
Sean Callanan6ba533e2010-11-17 23:00:36 +0000533{
534 if (!depth)
535 return;
536
537 depth--;
538
Greg Clayton3c7feb42010-11-19 01:05:25 +0000539 log->Printf(" <Begin %d users>", value->getNumUses());
Sean Callanan6ba533e2010-11-17 23:00:36 +0000540
Greg Clayton3c7feb42010-11-19 01:05:25 +0000541 for (Value::use_iterator ui = value->use_begin(), ue = value->use_end();
Sean Callanan6ba533e2010-11-17 23:00:36 +0000542 ui != ue;
543 ++ui)
544 {
545 log->Printf(" <Use %p> %s", *ui, PrintValue(*ui).c_str());
546 DebugUsers(log, *ui, depth);
547 }
548
549 log->Printf(" <End uses>");
550}
551
552bool
Greg Clayton3c7feb42010-11-19 01:05:25 +0000553IRForTarget::RewriteObjCConstString (llvm::Module &llvm_module,
554 llvm::GlobalVariable *ns_str,
555 llvm::GlobalVariable *cstr,
556 Instruction *FirstEntryInstruction)
Sean Callanan6ba533e2010-11-17 23:00:36 +0000557{
558 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
559
Greg Clayton3c7feb42010-11-19 01:05:25 +0000560 const Type *i8_ptr_ty = Type::getInt8PtrTy(llvm_module.getContext());
561 const IntegerType *intptr_ty = Type::getIntNTy(llvm_module.getContext(),
562 (llvm_module.getPointerSize() == Module::Pointer64) ? 64 : 32);
563 const Type *i32_ty = Type::getInt32Ty(llvm_module.getContext());
564 const Type *i8_ty = Type::getInt8Ty(llvm_module.getContext());
Sean Callanan6ba533e2010-11-17 23:00:36 +0000565
566 if (!m_CFStringCreateWithBytes)
567 {
568 lldb::addr_t CFStringCreateWithBytes_addr;
569
570 static lldb_private::ConstString g_CFStringCreateWithBytes_str ("CFStringCreateWithBytes");
571
572 if (!m_decl_map->GetFunctionAddress (g_CFStringCreateWithBytes_str, CFStringCreateWithBytes_addr))
573 {
574 if (log)
575 log->PutCString("Couldn't find CFStringCreateWithBytes in the target");
576
Sean Callanan97c924e2011-01-27 01:07:04 +0000577 if (m_error_stream)
578 m_error_stream->Printf("Error [IRForTarget]: Rewriting an Objective-C constant string requires CFStringCreateWithBytes\n");
579
Sean Callanan6ba533e2010-11-17 23:00:36 +0000580 return false;
581 }
582
583 if (log)
584 log->Printf("Found CFStringCreateWithBytes at 0x%llx", CFStringCreateWithBytes_addr);
585
586 // Build the function type:
587 //
588 // CFStringRef CFStringCreateWithBytes (
589 // CFAllocatorRef alloc,
590 // const UInt8 *bytes,
591 // CFIndex numBytes,
592 // CFStringEncoding encoding,
593 // Boolean isExternalRepresentation
594 // );
595 //
596 // We make the following substitutions:
597 //
598 // CFStringRef -> i8*
599 // CFAllocatorRef -> i8*
600 // UInt8 * -> i8*
601 // CFIndex -> long (i32 or i64, as appropriate; we ask the module for its pointer size for now)
602 // CFStringEncoding -> i32
603 // Boolean -> i8
604
605 std::vector <const Type *> CFSCWB_arg_types;
606 CFSCWB_arg_types.push_back(i8_ptr_ty);
607 CFSCWB_arg_types.push_back(i8_ptr_ty);
608 CFSCWB_arg_types.push_back(intptr_ty);
609 CFSCWB_arg_types.push_back(i32_ty);
610 CFSCWB_arg_types.push_back(i8_ty);
611 llvm::Type *CFSCWB_ty = FunctionType::get(i8_ptr_ty, CFSCWB_arg_types, false);
612
613 // Build the constant containing the pointer to the function
614 PointerType *CFSCWB_ptr_ty = PointerType::getUnqual(CFSCWB_ty);
615 Constant *CFSCWB_addr_int = ConstantInt::get(intptr_ty, CFStringCreateWithBytes_addr, false);
616 m_CFStringCreateWithBytes = ConstantExpr::getIntToPtr(CFSCWB_addr_int, CFSCWB_ptr_ty);
617 }
618
Sean Callanan0ece5262011-02-10 22:17:53 +0000619 ConstantArray *string_array;
620
621 if (cstr)
622 string_array = dyn_cast<ConstantArray>(cstr->getInitializer());
623 else
624 string_array = NULL;
Sean Callanan6ba533e2010-11-17 23:00:36 +0000625
626 SmallVector <Value*, 5> CFSCWB_arguments;
627
628 Constant *alloc_arg = Constant::getNullValue(i8_ptr_ty);
Sean Callanan0ece5262011-02-10 22:17:53 +0000629 Constant *bytes_arg = cstr ? ConstantExpr::getBitCast(cstr, i8_ptr_ty) : Constant::getNullValue(i8_ptr_ty);
630 Constant *numBytes_arg = ConstantInt::get(intptr_ty, cstr ? string_array->getType()->getNumElements() - 1 : 0, false);
Sean Callanan6ba533e2010-11-17 23:00:36 +0000631 Constant *encoding_arg = ConstantInt::get(i32_ty, 0x0600, false); /* 0x0600 is kCFStringEncodingASCII */
632 Constant *isExternal_arg = ConstantInt::get(i8_ty, 0x0, false); /* 0x0 is false */
633
634 CFSCWB_arguments.push_back(alloc_arg);
635 CFSCWB_arguments.push_back(bytes_arg);
636 CFSCWB_arguments.push_back(numBytes_arg);
637 CFSCWB_arguments.push_back(encoding_arg);
638 CFSCWB_arguments.push_back(isExternal_arg);
639
640 CallInst *CFSCWB_call = CallInst::Create(m_CFStringCreateWithBytes,
641 CFSCWB_arguments.begin(),
642 CFSCWB_arguments.end(),
643 "CFStringCreateWithBytes",
644 FirstEntryInstruction);
Sean Callananae71e302010-11-18 22:21:58 +0000645
Greg Clayton3c7feb42010-11-19 01:05:25 +0000646 if (!UnfoldConstant(ns_str, CFSCWB_call, FirstEntryInstruction))
Sean Callanan6ba533e2010-11-17 23:00:36 +0000647 {
648 if (log)
649 log->PutCString("Couldn't replace the NSString with the result of the call");
650
Sean Callanan97c924e2011-01-27 01:07:04 +0000651 if (m_error_stream)
652 m_error_stream->Printf("Error [IRForTarget]: Couldn't replace an Objective-C constant string with a dynamic string\n");
653
Sean Callanan6ba533e2010-11-17 23:00:36 +0000654 return false;
655 }
656
Greg Clayton3c7feb42010-11-19 01:05:25 +0000657 ns_str->eraseFromParent();
Sean Callanan6ba533e2010-11-17 23:00:36 +0000658
659 return true;
660}
661
662bool
Greg Clayton3c7feb42010-11-19 01:05:25 +0000663IRForTarget::RewriteObjCConstStrings(Module &llvm_module, Function &llvm_function)
Sean Callanan6ba533e2010-11-17 23:00:36 +0000664{
665 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
666
Greg Clayton3c7feb42010-11-19 01:05:25 +0000667 ValueSymbolTable& value_symbol_table = llvm_module.getValueSymbolTable();
Sean Callanan6ba533e2010-11-17 23:00:36 +0000668
Greg Clayton3c7feb42010-11-19 01:05:25 +0000669 BasicBlock &entry_block(llvm_function.getEntryBlock());
Sean Callanan6ba533e2010-11-17 23:00:36 +0000670 Instruction *FirstEntryInstruction(entry_block.getFirstNonPHIOrDbg());
671
672 if (!FirstEntryInstruction)
673 {
674 if (log)
675 log->PutCString("Couldn't find first instruction for rewritten Objective-C strings");
676
Sean Callanan97c924e2011-01-27 01:07:04 +0000677 if (m_error_stream)
678 m_error_stream->Printf("Internal error [IRForTarget]: Couldn't find the location for calls to CFStringCreateWithBytes\n");
679
Sean Callanan6ba533e2010-11-17 23:00:36 +0000680 return false;
681 }
682
683 for (ValueSymbolTable::iterator vi = value_symbol_table.begin(), ve = value_symbol_table.end();
684 vi != ve;
685 ++vi)
686 {
687 if (strstr(vi->first(), "_unnamed_cfstring_"))
688 {
689 Value *nsstring_value = vi->second;
690
691 GlobalVariable *nsstring_global = dyn_cast<GlobalVariable>(nsstring_value);
692
693 if (!nsstring_global)
694 {
695 if (log)
696 log->PutCString("NSString variable is not a GlobalVariable");
Sean Callanan97c924e2011-01-27 01:07:04 +0000697
698 if (m_error_stream)
699 m_error_stream->Printf("Internal error [IRForTarget]: An Objective-C constant string is not a global variable\n");
700
Sean Callanan6ba533e2010-11-17 23:00:36 +0000701 return false;
702 }
703
704 if (!nsstring_global->hasInitializer())
705 {
706 if (log)
707 log->PutCString("NSString variable does not have an initializer");
Sean Callanan97c924e2011-01-27 01:07:04 +0000708
709 if (m_error_stream)
710 m_error_stream->Printf("Internal error [IRForTarget]: An Objective-C constant string does not have an initializer\n");
711
Sean Callanan6ba533e2010-11-17 23:00:36 +0000712 return false;
713 }
714
715 ConstantStruct *nsstring_struct = dyn_cast<ConstantStruct>(nsstring_global->getInitializer());
716
717 if (!nsstring_struct)
718 {
719 if (log)
720 log->PutCString("NSString variable's initializer is not a ConstantStruct");
Sean Callanan97c924e2011-01-27 01:07:04 +0000721
722 if (m_error_stream)
723 m_error_stream->Printf("Internal error [IRForTarget]: An Objective-C constant string is not a structure constant\n");
724
Sean Callanan6ba533e2010-11-17 23:00:36 +0000725 return false;
726 }
727
728 // We expect the following structure:
729 //
730 // struct {
731 // int *isa;
732 // int flags;
733 // char *str;
734 // long length;
735 // };
736
737 if (nsstring_struct->getNumOperands() != 4)
738 {
739 if (log)
740 log->Printf("NSString variable's initializer structure has an unexpected number of members. Should be 4, is %d", nsstring_struct->getNumOperands());
Sean Callanan97c924e2011-01-27 01:07:04 +0000741
742 if (m_error_stream)
743 m_error_stream->Printf("Internal error [IRForTarget]: The struct for an Objective-C constant string is not as expected\n");
744
Sean Callanan6ba533e2010-11-17 23:00:36 +0000745 return false;
746 }
747
748 Constant *nsstring_member = nsstring_struct->getOperand(2);
749
750 if (!nsstring_member)
751 {
752 if (log)
753 log->PutCString("NSString initializer's str element was empty");
Sean Callanan97c924e2011-01-27 01:07:04 +0000754
755 if (m_error_stream)
756 m_error_stream->Printf("Internal error [IRForTarget]: An Objective-C constant string does not have a string initializer\n");
757
Sean Callanan6ba533e2010-11-17 23:00:36 +0000758 return false;
759 }
760
761 ConstantExpr *nsstring_expr = dyn_cast<ConstantExpr>(nsstring_member);
762
763 if (!nsstring_expr)
764 {
765 if (log)
766 log->PutCString("NSString initializer's str element is not a ConstantExpr");
Sean Callanan97c924e2011-01-27 01:07:04 +0000767
768 if (m_error_stream)
769 m_error_stream->Printf("Internal error [IRForTarget]: An Objective-C constant string's string initializer is not constant\n");
770
Sean Callanan6ba533e2010-11-17 23:00:36 +0000771 return false;
772 }
773
774 if (nsstring_expr->getOpcode() != Instruction::GetElementPtr)
775 {
776 if (log)
777 log->Printf("NSString initializer's str element is not a GetElementPtr expression, it's a %s", nsstring_expr->getOpcodeName());
Sean Callanan97c924e2011-01-27 01:07:04 +0000778
779 if (m_error_stream)
780 m_error_stream->Printf("Internal error [IRForTarget]: An Objective-C constant string's string initializer is not an array\n");
781
Sean Callanan6ba533e2010-11-17 23:00:36 +0000782 return false;
783 }
784
785 Constant *nsstring_cstr = nsstring_expr->getOperand(0);
786
787 GlobalVariable *cstr_global = dyn_cast<GlobalVariable>(nsstring_cstr);
788
789 if (!cstr_global)
790 {
791 if (log)
792 log->PutCString("NSString initializer's str element is not a GlobalVariable");
Sean Callanan97c924e2011-01-27 01:07:04 +0000793
794 if (m_error_stream)
795 m_error_stream->Printf("Internal error [IRForTarget]: An Objective-C constant string's string initializer doesn't point to a global\n");
Sean Callanan65e2aee2010-11-20 02:06:01 +0000796
Sean Callanan6ba533e2010-11-17 23:00:36 +0000797 return false;
798 }
799
800 if (!cstr_global->hasInitializer())
801 {
802 if (log)
803 log->PutCString("NSString initializer's str element does not have an initializer");
Sean Callanan97c924e2011-01-27 01:07:04 +0000804
805 if (m_error_stream)
806 m_error_stream->Printf("Internal error [IRForTarget]: An Objective-C constant string's string initializer doesn't point to initialized data\n");
807
Sean Callanan6ba533e2010-11-17 23:00:36 +0000808 return false;
809 }
Sean Callanan0ece5262011-02-10 22:17:53 +0000810
811 /*
Sean Callanan6ba533e2010-11-17 23:00:36 +0000812 if (!cstr_array)
813 {
814 if (log)
815 log->PutCString("NSString initializer's str element is not a ConstantArray");
Sean Callanan97c924e2011-01-27 01:07:04 +0000816
817 if (m_error_stream)
818 m_error_stream->Printf("Internal error [IRForTarget]: An Objective-C constant string's string initializer doesn't point to an array\n");
819
Sean Callanan6ba533e2010-11-17 23:00:36 +0000820 return false;
821 }
822
823 if (!cstr_array->isCString())
824 {
825 if (log)
826 log->PutCString("NSString initializer's str element is not a C string array");
Sean Callanan97c924e2011-01-27 01:07:04 +0000827
828 if (m_error_stream)
829 m_error_stream->Printf("Internal error [IRForTarget]: An Objective-C constant string's string initializer doesn't point to a C string\n");
830
Sean Callanan6ba533e2010-11-17 23:00:36 +0000831 return false;
832 }
Sean Callanan0ece5262011-02-10 22:17:53 +0000833 */
834
835 ConstantArray *cstr_array = dyn_cast<ConstantArray>(cstr_global->getInitializer());
Sean Callanan6ba533e2010-11-17 23:00:36 +0000836
837 if (log)
Sean Callanan0ece5262011-02-10 22:17:53 +0000838 {
839 if (cstr_array)
840 log->Printf("Found NSString constant %s, which contains \"%s\"", vi->first(), cstr_array->getAsString().c_str());
841 else
842 log->Printf("Found NSString constant %s, which contains \"\"", vi->first());
843 }
844
845 if (!cstr_array)
846 cstr_global = NULL;
Sean Callanan6ba533e2010-11-17 23:00:36 +0000847
Greg Clayton3c7feb42010-11-19 01:05:25 +0000848 if (!RewriteObjCConstString(llvm_module, nsstring_global, cstr_global, FirstEntryInstruction))
Sean Callanan97c924e2011-01-27 01:07:04 +0000849 {
Sean Callanan6ba533e2010-11-17 23:00:36 +0000850 if (log)
851 log->PutCString("Error rewriting the constant string");
Sean Callanan97c924e2011-01-27 01:07:04 +0000852
853 // We don't print an error message here because RewriteObjCConstString has done so for us.
854
Sean Callanan6ba533e2010-11-17 23:00:36 +0000855 return false;
856 }
Sean Callanan6ba533e2010-11-17 23:00:36 +0000857 }
858 }
859
860 for (ValueSymbolTable::iterator vi = value_symbol_table.begin(), ve = value_symbol_table.end();
861 vi != ve;
862 ++vi)
863 {
864 if (!strcmp(vi->first(), "__CFConstantStringClassReference"))
865 {
866 GlobalVariable *gv = dyn_cast<GlobalVariable>(vi->second);
867
868 if (!gv)
869 {
870 if (log)
871 log->PutCString("__CFConstantStringClassReference is not a global variable");
Sean Callanan97c924e2011-01-27 01:07:04 +0000872
873 if (m_error_stream)
874 m_error_stream->Printf("Internal error [IRForTarget]: Found a CFConstantStringClassReference, but it is not a global object\n");
875
Sean Callanan6ba533e2010-11-17 23:00:36 +0000876 return false;
877 }
878
879 gv->eraseFromParent();
880
881 break;
882 }
883 }
884
885 return true;
886}
887
Greg Clayton3c7feb42010-11-19 01:05:25 +0000888static bool IsObjCSelectorRef (Value *value)
Sean Callananf5857a02010-07-31 01:32:05 +0000889{
Greg Clayton3c7feb42010-11-19 01:05:25 +0000890 GlobalVariable *global_variable = dyn_cast<GlobalVariable>(value);
Sean Callananf5857a02010-07-31 01:32:05 +0000891
Greg Clayton3c7feb42010-11-19 01:05:25 +0000892 if (!global_variable || !global_variable->hasName() || !global_variable->getName().startswith("\01L_OBJC_SELECTOR_REFERENCES_"))
Sean Callananf5857a02010-07-31 01:32:05 +0000893 return false;
894
895 return true;
896}
897
Sean Callanan97c924e2011-01-27 01:07:04 +0000898// This function does not report errors; its callers are responsible.
Sean Callananf5857a02010-07-31 01:32:05 +0000899bool
Greg Clayton3c7feb42010-11-19 01:05:25 +0000900IRForTarget::RewriteObjCSelector (Instruction* selector_load, Module &llvm_module)
Sean Callananf5857a02010-07-31 01:32:05 +0000901{
Greg Claytone005f2c2010-11-06 01:53:30 +0000902 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
Sean Callananf5857a02010-07-31 01:32:05 +0000903
904 LoadInst *load = dyn_cast<LoadInst>(selector_load);
905
906 if (!load)
907 return false;
908
909 // Unpack the message name from the selector. In LLVM IR, an objc_msgSend gets represented as
910 //
911 // %tmp = load i8** @"\01L_OBJC_SELECTOR_REFERENCES_" ; <i8*>
912 // %call = call i8* (i8*, i8*, ...)* @objc_msgSend(i8* %obj, i8* %tmp, ...) ; <i8*>
913 //
914 // where %obj is the object pointer and %tmp is the selector.
915 //
Greg Clayton3c7feb42010-11-19 01:05:25 +0000916 // @"\01L_OBJC_SELECTOR_REFERENCES_" is a pointer to a character array called @"\01L_OBJC_llvm_moduleETH_VAR_NAllvm_moduleE_".
917 // @"\01L_OBJC_llvm_moduleETH_VAR_NAllvm_moduleE_" contains the string.
Sean Callananf5857a02010-07-31 01:32:05 +0000918
919 // Find the pointer's initializer (a ConstantExpr with opcode GetElementPtr) and get the string from its target
920
921 GlobalVariable *_objc_selector_references_ = dyn_cast<GlobalVariable>(load->getPointerOperand());
922
923 if (!_objc_selector_references_ || !_objc_selector_references_->hasInitializer())
924 return false;
925
926 Constant *osr_initializer = _objc_selector_references_->getInitializer();
927
928 ConstantExpr *osr_initializer_expr = dyn_cast<ConstantExpr>(osr_initializer);
929
930 if (!osr_initializer_expr || osr_initializer_expr->getOpcode() != Instruction::GetElementPtr)
931 return false;
932
933 Value *osr_initializer_base = osr_initializer_expr->getOperand(0);
934
935 if (!osr_initializer_base)
936 return false;
937
938 // Find the string's initializer (a ConstantArray) and get the string from it
939
940 GlobalVariable *_objc_meth_var_name_ = dyn_cast<GlobalVariable>(osr_initializer_base);
941
942 if (!_objc_meth_var_name_ || !_objc_meth_var_name_->hasInitializer())
943 return false;
944
945 Constant *omvn_initializer = _objc_meth_var_name_->getInitializer();
946
947 ConstantArray *omvn_initializer_array = dyn_cast<ConstantArray>(omvn_initializer);
948
949 if (!omvn_initializer_array->isString())
950 return false;
951
952 std::string omvn_initializer_string = omvn_initializer_array->getAsString();
953
954 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +0000955 log->Printf("Found Objective-C selector reference \"%s\"", omvn_initializer_string.c_str());
Sean Callananf5857a02010-07-31 01:32:05 +0000956
957 // Construct a call to sel_registerName
958
959 if (!m_sel_registerName)
960 {
Greg Claytonb5037af2010-11-15 01:47:11 +0000961 lldb::addr_t sel_registerName_addr;
Sean Callananf5857a02010-07-31 01:32:05 +0000962
Greg Clayton8de27c72010-10-15 22:48:33 +0000963 static lldb_private::ConstString g_sel_registerName_str ("sel_registerName");
Greg Claytonb5037af2010-11-15 01:47:11 +0000964 if (!m_decl_map->GetFunctionAddress (g_sel_registerName_str, sel_registerName_addr))
Sean Callananf5857a02010-07-31 01:32:05 +0000965 return false;
966
Sean Callananc2c6f772010-10-26 00:31:56 +0000967 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +0000968 log->Printf("Found sel_registerName at 0x%llx", sel_registerName_addr);
Sean Callananc2c6f772010-10-26 00:31:56 +0000969
Sean Callananf5857a02010-07-31 01:32:05 +0000970 // Build the function type: struct objc_selector *sel_registerName(uint8_t*)
971
972 // The below code would be "more correct," but in actuality what's required is uint8_t*
Greg Clayton3c7feb42010-11-19 01:05:25 +0000973 //Type *sel_type = StructType::get(llvm_module.getContext());
Sean Callananf5857a02010-07-31 01:32:05 +0000974 //Type *sel_ptr_type = PointerType::getUnqual(sel_type);
Greg Clayton3c7feb42010-11-19 01:05:25 +0000975 const Type *sel_ptr_type = Type::getInt8PtrTy(llvm_module.getContext());
Sean Callananf5857a02010-07-31 01:32:05 +0000976
977 std::vector <const Type *> srN_arg_types;
Greg Clayton3c7feb42010-11-19 01:05:25 +0000978 srN_arg_types.push_back(Type::getInt8PtrTy(llvm_module.getContext()));
Sean Callananf5857a02010-07-31 01:32:05 +0000979 llvm::Type *srN_type = FunctionType::get(sel_ptr_type, srN_arg_types, false);
980
981 // Build the constant containing the pointer to the function
Greg Clayton3c7feb42010-11-19 01:05:25 +0000982 const IntegerType *intptr_ty = Type::getIntNTy(llvm_module.getContext(),
983 (llvm_module.getPointerSize() == Module::Pointer64) ? 64 : 32);
Sean Callananf5857a02010-07-31 01:32:05 +0000984 PointerType *srN_ptr_ty = PointerType::getUnqual(srN_type);
Greg Claytonb5037af2010-11-15 01:47:11 +0000985 Constant *srN_addr_int = ConstantInt::get(intptr_ty, sel_registerName_addr, false);
Sean Callananf5857a02010-07-31 01:32:05 +0000986 m_sel_registerName = ConstantExpr::getIntToPtr(srN_addr_int, srN_ptr_ty);
987 }
988
989 SmallVector <Value*, 1> srN_arguments;
990
Greg Clayton3c7feb42010-11-19 01:05:25 +0000991 Constant *omvn_pointer = ConstantExpr::getBitCast(_objc_meth_var_name_, Type::getInt8PtrTy(llvm_module.getContext()));
Sean Callananf5857a02010-07-31 01:32:05 +0000992
993 srN_arguments.push_back(omvn_pointer);
994
995 CallInst *srN_call = CallInst::Create(m_sel_registerName,
996 srN_arguments.begin(),
997 srN_arguments.end(),
Sean Callanan6ba533e2010-11-17 23:00:36 +0000998 "sel_registerName",
Sean Callananf5857a02010-07-31 01:32:05 +0000999 selector_load);
1000
1001 // Replace the load with the call in all users
1002
1003 selector_load->replaceAllUsesWith(srN_call);
1004
1005 selector_load->eraseFromParent();
1006
1007 return true;
1008}
1009
1010bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001011IRForTarget::RewriteObjCSelectors (Module &llvm_module, BasicBlock &basic_block)
Sean Callananf5857a02010-07-31 01:32:05 +00001012{
Greg Claytone005f2c2010-11-06 01:53:30 +00001013 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
Sean Callananf5857a02010-07-31 01:32:05 +00001014
1015 BasicBlock::iterator ii;
1016
1017 typedef SmallVector <Instruction*, 2> InstrList;
1018 typedef InstrList::iterator InstrIterator;
1019
1020 InstrList selector_loads;
1021
Greg Clayton3c7feb42010-11-19 01:05:25 +00001022 for (ii = basic_block.begin();
1023 ii != basic_block.end();
Sean Callananf5857a02010-07-31 01:32:05 +00001024 ++ii)
1025 {
1026 Instruction &inst = *ii;
1027
1028 if (LoadInst *load = dyn_cast<LoadInst>(&inst))
Greg Clayton3c7feb42010-11-19 01:05:25 +00001029 if (IsObjCSelectorRef(load->getPointerOperand()))
Sean Callananf5857a02010-07-31 01:32:05 +00001030 selector_loads.push_back(&inst);
1031 }
1032
1033 InstrIterator iter;
1034
1035 for (iter = selector_loads.begin();
1036 iter != selector_loads.end();
1037 ++iter)
1038 {
Greg Clayton3c7feb42010-11-19 01:05:25 +00001039 if (!RewriteObjCSelector(*iter, llvm_module))
Sean Callananf5857a02010-07-31 01:32:05 +00001040 {
Sean Callanan97c924e2011-01-27 01:07:04 +00001041 if (m_error_stream)
1042 m_error_stream->Printf("Internal error [IRForTarget]: Couldn't change a static reference to an Objective-C selector to a dynamic reference\n");
1043
Sean Callananf5857a02010-07-31 01:32:05 +00001044 if(log)
1045 log->PutCString("Couldn't rewrite a reference to an Objective-C selector");
Sean Callanan97c924e2011-01-27 01:07:04 +00001046
Sean Callananf5857a02010-07-31 01:32:05 +00001047 return false;
1048 }
1049 }
1050
1051 return true;
1052}
1053
Sean Callanan97c924e2011-01-27 01:07:04 +00001054// This function does not report errors; its callers are responsible.
Sean Callanana48fe162010-08-11 03:57:18 +00001055bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001056IRForTarget::RewritePersistentAlloc (llvm::Instruction *persistent_alloc,
1057 llvm::Module &llvm_module)
Sean Callanana48fe162010-08-11 03:57:18 +00001058{
Sean Callanan97678d12011-01-13 21:23:32 +00001059 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
1060
Sean Callanana48fe162010-08-11 03:57:18 +00001061 AllocaInst *alloc = dyn_cast<AllocaInst>(persistent_alloc);
1062
1063 MDNode *alloc_md = alloc->getMetadata("clang.decl.ptr");
1064
1065 if (!alloc_md || !alloc_md->getNumOperands())
1066 return false;
1067
1068 ConstantInt *constant_int = dyn_cast<ConstantInt>(alloc_md->getOperand(0));
1069
1070 if (!constant_int)
1071 return false;
1072
1073 // We attempt to register this as a new persistent variable with the DeclMap.
1074
1075 uintptr_t ptr = constant_int->getZExtValue();
1076
Sean Callanan82b74c82010-08-12 01:56:52 +00001077 clang::VarDecl *decl = reinterpret_cast<clang::VarDecl *>(ptr);
Sean Callanana48fe162010-08-11 03:57:18 +00001078
Sean Callanan82b74c82010-08-12 01:56:52 +00001079 lldb_private::TypeFromParser result_decl_type (decl->getType().getAsOpaquePtr(),
1080 &decl->getASTContext());
1081
Greg Clayton8de27c72010-10-15 22:48:33 +00001082 StringRef decl_name (decl->getName());
1083 lldb_private::ConstString persistent_variable_name (decl_name.data(), decl_name.size());
Sean Callanan6a925532011-01-13 08:53:35 +00001084 if (!m_decl_map->AddPersistentVariable(decl, persistent_variable_name, result_decl_type, false, false))
Sean Callanana48fe162010-08-11 03:57:18 +00001085 return false;
1086
Sean Callanan97678d12011-01-13 21:23:32 +00001087 GlobalVariable *persistent_global = new GlobalVariable(llvm_module,
1088 alloc->getType(),
Sean Callanana48fe162010-08-11 03:57:18 +00001089 false, /* not constant */
1090 GlobalValue::ExternalLinkage,
1091 NULL, /* no initializer */
1092 alloc->getName().str().c_str());
1093
1094 // What we're going to do here is make believe this was a regular old external
1095 // variable. That means we need to make the metadata valid.
1096
Greg Clayton8de27c72010-10-15 22:48:33 +00001097 NamedMDNode *named_metadata = llvm_module.getNamedMetadata("clang.global.decl.ptrs");
Sean Callanana48fe162010-08-11 03:57:18 +00001098
1099 llvm::Value* values[2];
1100 values[0] = persistent_global;
1101 values[1] = constant_int;
Sean Callanan0de254a2011-05-15 22:34:38 +00001102
1103 ArrayRef<llvm::Value*> value_ref(values, 2);
Sean Callanana48fe162010-08-11 03:57:18 +00001104
Sean Callanan0de254a2011-05-15 22:34:38 +00001105 MDNode *persistent_global_md = MDNode::get(llvm_module.getContext(), value_ref);
Sean Callanana48fe162010-08-11 03:57:18 +00001106 named_metadata->addOperand(persistent_global_md);
1107
Sean Callanan97678d12011-01-13 21:23:32 +00001108 // Now, since the variable is a pointer variable, we will drop in a load of that
1109 // pointer variable.
1110
1111 LoadInst *persistent_load = new LoadInst (persistent_global, "", alloc);
1112
1113 if (log)
1114 log->Printf("Replacing \"%s\" with \"%s\"",
1115 PrintValue(alloc).c_str(),
1116 PrintValue(persistent_load).c_str());
1117
1118 alloc->replaceAllUsesWith(persistent_load);
Sean Callanana48fe162010-08-11 03:57:18 +00001119 alloc->eraseFromParent();
1120
1121 return true;
1122}
1123
1124bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001125IRForTarget::RewritePersistentAllocs(llvm::Module &llvm_module, llvm::BasicBlock &basic_block)
Sean Callanana48fe162010-08-11 03:57:18 +00001126{
Sean Callanane8a59a82010-09-13 21:34:21 +00001127 if (!m_resolve_vars)
1128 return true;
1129
Greg Claytone005f2c2010-11-06 01:53:30 +00001130 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
Sean Callanana48fe162010-08-11 03:57:18 +00001131
1132 BasicBlock::iterator ii;
1133
1134 typedef SmallVector <Instruction*, 2> InstrList;
1135 typedef InstrList::iterator InstrIterator;
1136
1137 InstrList pvar_allocs;
1138
Greg Clayton3c7feb42010-11-19 01:05:25 +00001139 for (ii = basic_block.begin();
1140 ii != basic_block.end();
Sean Callanana48fe162010-08-11 03:57:18 +00001141 ++ii)
1142 {
1143 Instruction &inst = *ii;
1144
1145 if (AllocaInst *alloc = dyn_cast<AllocaInst>(&inst))
Sean Callanan237e4742011-01-21 22:30:25 +00001146 {
1147 llvm::StringRef alloc_name = alloc->getName();
1148
1149 if (alloc_name.startswith("$") &&
1150 !alloc_name.startswith("$__lldb"))
1151 {
1152 if (alloc_name.find_first_of("0123456789") == 1)
1153 {
1154 if (log)
1155 log->Printf("Rejecting a numeric persistent variable.");
1156
Sean Callanan97c924e2011-01-27 01:07:04 +00001157 if (m_error_stream)
1158 m_error_stream->Printf("Error [IRForTarget]: Names starting with $0, $1, ... are reserved for use as result names\n");
1159
Sean Callanan237e4742011-01-21 22:30:25 +00001160 return false;
1161 }
1162
Sean Callanana48fe162010-08-11 03:57:18 +00001163 pvar_allocs.push_back(alloc);
Sean Callanan237e4742011-01-21 22:30:25 +00001164 }
1165 }
Sean Callanana48fe162010-08-11 03:57:18 +00001166 }
1167
1168 InstrIterator iter;
1169
1170 for (iter = pvar_allocs.begin();
1171 iter != pvar_allocs.end();
1172 ++iter)
1173 {
Greg Clayton3c7feb42010-11-19 01:05:25 +00001174 if (!RewritePersistentAlloc(*iter, llvm_module))
Sean Callanana48fe162010-08-11 03:57:18 +00001175 {
Sean Callanan97c924e2011-01-27 01:07:04 +00001176 if (m_error_stream)
1177 m_error_stream->Printf("Internal error [IRForTarget]: Couldn't rewrite the creation of a persistent variable\n");
1178
Sean Callanana48fe162010-08-11 03:57:18 +00001179 if(log)
1180 log->PutCString("Couldn't rewrite the creation of a persistent variable");
Sean Callanan97c924e2011-01-27 01:07:04 +00001181
Sean Callanana48fe162010-08-11 03:57:18 +00001182 return false;
1183 }
1184 }
1185
1186 return true;
1187}
1188
Sean Callanan97c924e2011-01-27 01:07:04 +00001189// This function does not report errors; its callers are responsible.
Sean Callanan8bce6652010-07-13 21:41:46 +00001190bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001191IRForTarget::MaybeHandleVariable (Module &llvm_module, Value *llvm_value_ptr)
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001192{
Greg Claytone005f2c2010-11-06 01:53:30 +00001193 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
Sean Callanan48443652010-12-02 19:47:57 +00001194
1195 if (log)
Sean Callananb9f09a62010-12-06 22:16:55 +00001196 log->Printf("MaybeHandleVariable (%s)", PrintValue(llvm_value_ptr).c_str());
Sean Callananf5857a02010-07-31 01:32:05 +00001197
Greg Clayton8de27c72010-10-15 22:48:33 +00001198 if (ConstantExpr *constant_expr = dyn_cast<ConstantExpr>(llvm_value_ptr))
Sean Callananbc2928a2010-08-03 00:23:29 +00001199 {
Sean Callanan93a4b1a2010-08-04 01:02:13 +00001200 switch (constant_expr->getOpcode())
Sean Callananbc2928a2010-08-03 00:23:29 +00001201 {
Sean Callanan93a4b1a2010-08-04 01:02:13 +00001202 default:
1203 break;
1204 case Instruction::GetElementPtr:
1205 case Instruction::BitCast:
Sean Callananbc2928a2010-08-03 00:23:29 +00001206 Value *s = constant_expr->getOperand(0);
Sean Callanan48443652010-12-02 19:47:57 +00001207 if (!MaybeHandleVariable(llvm_module, s))
1208 return false;
Sean Callananbc2928a2010-08-03 00:23:29 +00001209 }
1210 }
Sean Callananf921cf52010-12-03 19:51:05 +00001211 else if (GlobalVariable *global_variable = dyn_cast<GlobalVariable>(llvm_value_ptr))
Sean Callananf5857a02010-07-31 01:32:05 +00001212 {
Sean Callanan696cf5f2011-05-07 01:06:41 +00001213 clang::NamedDecl *named_decl = DeclForGlobal(llvm_module, global_variable);
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001214
Sean Callananf5857a02010-07-31 01:32:05 +00001215 if (!named_decl)
1216 {
Greg Clayton3c7feb42010-11-19 01:05:25 +00001217 if (IsObjCSelectorRef(llvm_value_ptr))
Sean Callananf5857a02010-07-31 01:32:05 +00001218 return true;
1219
Sean Callanan7cd46742010-12-06 00:56:39 +00001220 if (!global_variable->hasExternalLinkage())
1221 return true;
1222
Sean Callananf5857a02010-07-31 01:32:05 +00001223 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +00001224 log->Printf("Found global variable \"%s\" without metadata", global_variable->getName().str().c_str());
Sean Callanan97c924e2011-01-27 01:07:04 +00001225
Sean Callananf5857a02010-07-31 01:32:05 +00001226 return false;
1227 }
1228
Greg Clayton8de27c72010-10-15 22:48:33 +00001229 std::string name (named_decl->getName().str());
Sean Callanan810f22d2010-07-16 00:09:46 +00001230
Sean Callanan771131d2010-09-30 21:18:25 +00001231 void *opaque_type = NULL;
Sean Callananf328c9f2010-07-20 23:31:16 +00001232 clang::ASTContext *ast_context = NULL;
Sean Callanan810f22d2010-07-16 00:09:46 +00001233
1234 if (clang::ValueDecl *value_decl = dyn_cast<clang::ValueDecl>(named_decl))
Sean Callananf328c9f2010-07-20 23:31:16 +00001235 {
Sean Callanan771131d2010-09-30 21:18:25 +00001236 opaque_type = value_decl->getType().getAsOpaquePtr();
Sean Callananf328c9f2010-07-20 23:31:16 +00001237 ast_context = &value_decl->getASTContext();
1238 }
Sean Callanan810f22d2010-07-16 00:09:46 +00001239 else
Sean Callananf328c9f2010-07-20 23:31:16 +00001240 {
Sean Callanan810f22d2010-07-16 00:09:46 +00001241 return false;
Sean Callananf328c9f2010-07-20 23:31:16 +00001242 }
Sean Callanan771131d2010-09-30 21:18:25 +00001243
Sean Callanan6a925532011-01-13 08:53:35 +00001244 clang::QualType qual_type;
1245 const Type *value_type;
1246
Sean Callanan97678d12011-01-13 21:23:32 +00001247 if (name[0] == '$')
Sean Callanan6a925532011-01-13 08:53:35 +00001248 {
1249 // The $__lldb_expr_result name indicates the the return value has allocated as
1250 // a static variable. Per the comment at ASTResultSynthesizer::SynthesizeBodyResult,
1251 // accesses to this static variable need to be redirected to the result of dereferencing
1252 // a pointer that is passed in as one of the arguments.
1253 //
1254 // Consequently, when reporting the size of the type, we report a pointer type pointing
1255 // to the type of $__lldb_expr_result, not the type itself.
Sean Callanan97678d12011-01-13 21:23:32 +00001256 //
1257 // We also do this for any user-declared persistent variables.
Sean Callananf328c9f2010-07-20 23:31:16 +00001258
Sean Callanan6a925532011-01-13 08:53:35 +00001259 qual_type = ast_context->getPointerType(clang::QualType::getFromOpaquePtr(opaque_type));
1260 value_type = PointerType::get(global_variable->getType(), 0);
1261 }
1262 else
1263 {
1264 qual_type = clang::QualType::getFromOpaquePtr(opaque_type);
1265 value_type = global_variable->getType();
1266 }
Sean Callanan771131d2010-09-30 21:18:25 +00001267
1268 size_t value_size = (ast_context->getTypeSize(qual_type) + 7) / 8;
1269 off_t value_alignment = (ast_context->getTypeAlign(qual_type) + 7) / 8;
Sean Callanan3aa7da52010-12-13 22:46:15 +00001270
Sean Callanan771131d2010-09-30 21:18:25 +00001271 if (log)
Sean Callanan97678d12011-01-13 21:23:32 +00001272 log->Printf("Type of \"%s\" is [clang \"%s\", llvm \"%s\"] [size %d, align %d]",
Sean Callanan771131d2010-09-30 21:18:25 +00001273 name.c_str(),
1274 qual_type.getAsString().c_str(),
1275 PrintType(value_type).c_str(),
1276 value_size,
1277 value_alignment);
Sean Callanan3aa7da52010-12-13 22:46:15 +00001278
Sean Callanan8bce6652010-07-13 21:41:46 +00001279
Sean Callanan8c127202010-08-23 23:09:38 +00001280 if (named_decl && !m_decl_map->AddValueToStruct(named_decl,
Greg Clayton8de27c72010-10-15 22:48:33 +00001281 lldb_private::ConstString (name.c_str()),
1282 llvm_value_ptr,
Sean Callananba992c52010-07-27 02:07:53 +00001283 value_size,
1284 value_alignment))
Sean Callanan8bce6652010-07-13 21:41:46 +00001285 return false;
1286 }
Sean Callananf3143b72010-12-03 03:02:31 +00001287 else if (dyn_cast<llvm::Function>(llvm_value_ptr))
Sean Callanan48443652010-12-02 19:47:57 +00001288 {
1289 if (log)
1290 log->Printf("Function pointers aren't handled right now");
1291
1292 return false;
1293 }
Sean Callanan8bce6652010-07-13 21:41:46 +00001294
1295 return true;
1296}
1297
Sean Callanan97c924e2011-01-27 01:07:04 +00001298// This function does not report errors; its callers are responsible.
Sean Callanan8bce6652010-07-13 21:41:46 +00001299bool
Sean Callananc7674af2011-01-17 23:42:46 +00001300IRForTarget::HandleSymbol (Module &llvm_module,
1301 Value *symbol)
Sean Callanan81974962011-05-08 02:21:26 +00001302{
Sean Callananc7674af2011-01-17 23:42:46 +00001303 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
1304
1305 lldb_private::ConstString name(symbol->getName().str().c_str());
1306
1307 uint64_t symbol_addr;
1308
1309 if (!m_decl_map->GetSymbolAddress (name, symbol_addr))
1310 {
1311 if (log)
1312 log->Printf ("Symbol \"%s\" had no address", name.GetCString());
1313
1314 return false;
1315 }
1316
1317 if (log)
1318 log->Printf("Found \"%s\" at 0x%llx", name.GetCString(), symbol_addr);
1319
1320 const Type *symbol_type = symbol->getType();
1321
1322 const IntegerType *intptr_ty = Type::getIntNTy(llvm_module.getContext(),
1323 (llvm_module.getPointerSize() == Module::Pointer64) ? 64 : 32);
1324
1325 Constant *symbol_addr_int = ConstantInt::get(intptr_ty, symbol_addr, false);
1326
1327 Value *symbol_addr_ptr = ConstantExpr::getIntToPtr(symbol_addr_int, symbol_type);
1328
1329 if (log)
1330 log->Printf("Replacing %s with %s", PrintValue(symbol).c_str(), PrintValue(symbol_addr_ptr).c_str());
1331
1332 symbol->replaceAllUsesWith(symbol_addr_ptr);
1333
1334 return true;
1335}
1336
1337bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001338IRForTarget::MaybeHandleCallArguments (Module &llvm_module, CallInst *Old)
Sean Callanandc27aba2010-10-05 22:26:43 +00001339{
Sean Callanan48443652010-12-02 19:47:57 +00001340 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
1341
1342 if (log)
1343 log->Printf("MaybeHandleCallArguments(%s)", PrintValue(Old).c_str());
Sean Callanandc27aba2010-10-05 22:26:43 +00001344
Sean Callanan6ba533e2010-11-17 23:00:36 +00001345 for (unsigned op_index = 0, num_ops = Old->getNumArgOperands();
Sean Callanandc27aba2010-10-05 22:26:43 +00001346 op_index < num_ops;
1347 ++op_index)
Greg Clayton3c7feb42010-11-19 01:05:25 +00001348 if (!MaybeHandleVariable(llvm_module, Old->getArgOperand(op_index))) // conservatively believe that this is a store
Sean Callanan97c924e2011-01-27 01:07:04 +00001349 {
1350 if (m_error_stream)
1351 m_error_stream->Printf("Internal error [IRForTarget]: Couldn't rewrite one of the arguments of a function call.\n");
1352
Sean Callanandc27aba2010-10-05 22:26:43 +00001353 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001354 }
1355
Sean Callanandc27aba2010-10-05 22:26:43 +00001356 return true;
1357}
1358
1359bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001360IRForTarget::MaybeHandleCall (Module &llvm_module, CallInst *llvm_call_inst)
Sean Callananba992c52010-07-27 02:07:53 +00001361{
Greg Claytone005f2c2010-11-06 01:53:30 +00001362 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
Sean Callananba992c52010-07-27 02:07:53 +00001363
Greg Clayton8de27c72010-10-15 22:48:33 +00001364 Function *fun = llvm_call_inst->getCalledFunction();
Sean Callananba992c52010-07-27 02:07:53 +00001365
1366 if (fun == NULL)
Sean Callanan65af7342010-09-08 20:04:08 +00001367 {
Greg Clayton8de27c72010-10-15 22:48:33 +00001368 Value *val = llvm_call_inst->getCalledValue();
Sean Callanan65af7342010-09-08 20:04:08 +00001369
1370 ConstantExpr *const_expr = dyn_cast<ConstantExpr>(val);
Sean Callanand5b3c352011-01-27 04:42:51 +00001371 LoadInst *load_inst = dyn_cast<LoadInst>(val);
Sean Callanan65af7342010-09-08 20:04:08 +00001372
1373 if (const_expr && const_expr->getOpcode() == Instruction::BitCast)
1374 {
1375 fun = dyn_cast<Function>(const_expr->getOperand(0));
1376
1377 if (!fun)
Sean Callanan97c924e2011-01-27 01:07:04 +00001378 {
1379 if (m_error_stream)
1380 m_error_stream->Printf("Internal error [IRForTarget]: Called entity is a cast of something not a function\n");
1381
Sean Callanan48443652010-12-02 19:47:57 +00001382 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001383 }
Sean Callanan65af7342010-09-08 20:04:08 +00001384 }
Sean Callananc4217a62010-12-06 23:53:20 +00001385 else if (const_expr && const_expr->getOpcode() == Instruction::IntToPtr)
1386 {
1387 return true; // already resolved
1388 }
Sean Callanand5b3c352011-01-27 04:42:51 +00001389 else if (load_inst)
1390 {
1391 return true; // virtual method call
1392 }
Sean Callanan65af7342010-09-08 20:04:08 +00001393 else
1394 {
Sean Callanan97c924e2011-01-27 01:07:04 +00001395 if (m_error_stream)
1396 m_error_stream->Printf("Internal error [IRForTarget]: Called entity is not a function\n");
1397
Sean Callanan48443652010-12-02 19:47:57 +00001398 return false;
Sean Callanan65af7342010-09-08 20:04:08 +00001399 }
1400 }
Sean Callananba992c52010-07-27 02:07:53 +00001401
Greg Clayton8de27c72010-10-15 22:48:33 +00001402 lldb_private::ConstString str;
Sean Callananc04743d2010-09-28 21:13:03 +00001403
Sean Callanan3cb1fd82010-09-28 23:55:00 +00001404 if (fun->isIntrinsic())
Sean Callananc04743d2010-09-28 21:13:03 +00001405 {
Sean Callanan3cb1fd82010-09-28 23:55:00 +00001406 Intrinsic::ID intrinsic_id = (Intrinsic::ID)fun->getIntrinsicID();
Sean Callananc04743d2010-09-28 21:13:03 +00001407
Sean Callanan3cb1fd82010-09-28 23:55:00 +00001408 switch (intrinsic_id)
1409 {
1410 default:
1411 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +00001412 log->Printf("Unresolved intrinsic \"%s\"", Intrinsic::getName(intrinsic_id).c_str());
Sean Callanan97c924e2011-01-27 01:07:04 +00001413
1414 if (m_error_stream)
1415 m_error_stream->Printf("Internal error [IRForTarget]: Call to unhandled compiler intrinsic '%s'\n", Intrinsic::getName(intrinsic_id).c_str());
1416
Sean Callanan3cb1fd82010-09-28 23:55:00 +00001417 return false;
1418 case Intrinsic::memcpy:
Greg Clayton8de27c72010-10-15 22:48:33 +00001419 {
1420 static lldb_private::ConstString g_memcpy_str ("memcpy");
1421 str = g_memcpy_str;
1422 }
Sean Callanan3cb1fd82010-09-28 23:55:00 +00001423 break;
1424 }
Sean Callananc04743d2010-09-28 21:13:03 +00001425
Greg Clayton8de27c72010-10-15 22:48:33 +00001426 if (log && str)
Greg Claytonb5037af2010-11-15 01:47:11 +00001427 log->Printf("Resolved intrinsic name \"%s\"", str.GetCString());
Sean Callanan3cb1fd82010-09-28 23:55:00 +00001428 }
1429 else
1430 {
Greg Clayton8de27c72010-10-15 22:48:33 +00001431 str.SetCStringWithLength (fun->getName().data(), fun->getName().size());
Sean Callananc04743d2010-09-28 21:13:03 +00001432 }
1433
Sean Callanan696cf5f2011-05-07 01:06:41 +00001434 clang::NamedDecl *fun_decl = DeclForGlobal (llvm_module, fun);
Greg Claytonb5037af2010-11-15 01:47:11 +00001435 lldb::addr_t fun_addr = LLDB_INVALID_ADDRESS;
Sean Callananf5857a02010-07-31 01:32:05 +00001436 Value **fun_value_ptr = NULL;
Sean Callananba992c52010-07-27 02:07:53 +00001437
Sean Callananf5857a02010-07-31 01:32:05 +00001438 if (fun_decl)
Sean Callananba992c52010-07-27 02:07:53 +00001439 {
Greg Clayton8de27c72010-10-15 22:48:33 +00001440 if (!m_decl_map->GetFunctionInfo (fun_decl, fun_value_ptr, fun_addr))
Sean Callananf5857a02010-07-31 01:32:05 +00001441 {
Sean Callanan92aa6662010-09-07 21:49:41 +00001442 fun_value_ptr = NULL;
1443
Greg Clayton8de27c72010-10-15 22:48:33 +00001444 if (!m_decl_map->GetFunctionAddress (str, fun_addr))
Sean Callanan92aa6662010-09-07 21:49:41 +00001445 {
1446 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +00001447 log->Printf("Function \"%s\" had no address", str.GetCString());
Sean Callanan92aa6662010-09-07 21:49:41 +00001448
Sean Callanan97c924e2011-01-27 01:07:04 +00001449 if (m_error_stream)
1450 m_error_stream->Printf("Error [IRForTarget]: Call to a function '%s' that is not present in the target\n", str.GetCString());
1451
Sean Callanan92aa6662010-09-07 21:49:41 +00001452 return false;
1453 }
Sean Callananf5857a02010-07-31 01:32:05 +00001454 }
1455 }
1456 else
1457 {
Greg Clayton8de27c72010-10-15 22:48:33 +00001458 if (!m_decl_map->GetFunctionAddress (str, fun_addr))
Sean Callananf5857a02010-07-31 01:32:05 +00001459 {
1460 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +00001461 log->Printf ("Metadataless function \"%s\" had no address", str.GetCString());
Sean Callanan97c924e2011-01-27 01:07:04 +00001462
1463 if (m_error_stream)
1464 m_error_stream->Printf("Error [IRForTarget]: Call to a symbol-only function '%s' that is not present in the target\n", str.GetCString());
1465
1466 return false;
Sean Callananf5857a02010-07-31 01:32:05 +00001467 }
Sean Callananba992c52010-07-27 02:07:53 +00001468 }
1469
1470 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +00001471 log->Printf("Found \"%s\" at 0x%llx", str.GetCString(), fun_addr);
Sean Callananba992c52010-07-27 02:07:53 +00001472
Sean Callananf5857a02010-07-31 01:32:05 +00001473 Value *fun_addr_ptr;
1474
1475 if (!fun_value_ptr || !*fun_value_ptr)
Sean Callanan02fbafa2010-07-27 21:39:39 +00001476 {
Greg Clayton8de27c72010-10-15 22:48:33 +00001477 const IntegerType *intptr_ty = Type::getIntNTy(llvm_module.getContext(),
1478 (llvm_module.getPointerSize() == Module::Pointer64) ? 64 : 32);
Sean Callanan92aa6662010-09-07 21:49:41 +00001479 const FunctionType *fun_ty = fun->getFunctionType();
Sean Callanan02fbafa2010-07-27 21:39:39 +00001480 PointerType *fun_ptr_ty = PointerType::getUnqual(fun_ty);
1481 Constant *fun_addr_int = ConstantInt::get(intptr_ty, fun_addr, false);
Sean Callananf5857a02010-07-31 01:32:05 +00001482 fun_addr_ptr = ConstantExpr::getIntToPtr(fun_addr_int, fun_ptr_ty);
1483
1484 if (fun_value_ptr)
1485 *fun_value_ptr = fun_addr_ptr;
Sean Callanan02fbafa2010-07-27 21:39:39 +00001486 }
Sean Callananf5857a02010-07-31 01:32:05 +00001487
1488 if (fun_value_ptr)
1489 fun_addr_ptr = *fun_value_ptr;
Sean Callanan02fbafa2010-07-27 21:39:39 +00001490
Greg Clayton8de27c72010-10-15 22:48:33 +00001491 llvm_call_inst->setCalledFunction(fun_addr_ptr);
Sean Callanan02fbafa2010-07-27 21:39:39 +00001492
Greg Clayton8de27c72010-10-15 22:48:33 +00001493 ConstantArray *func_name = (ConstantArray*)ConstantArray::get(llvm_module.getContext(), str.GetCString());
Sean Callanane8a59a82010-09-13 21:34:21 +00001494
1495 Value *values[1];
1496 values[0] = func_name;
Sean Callanan0de254a2011-05-15 22:34:38 +00001497 ArrayRef<Value*> value_ref(values, 1);
1498
1499 MDNode *func_metadata = MDNode::get(llvm_module.getContext(), value_ref);
Sean Callanane8a59a82010-09-13 21:34:21 +00001500
Greg Clayton8de27c72010-10-15 22:48:33 +00001501 llvm_call_inst->setMetadata("lldb.call.realName", func_metadata);
Sean Callanane8a59a82010-09-13 21:34:21 +00001502
1503 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +00001504 log->Printf("Set metadata for %p [%d, \"%s\"]", llvm_call_inst, func_name->isString(), func_name->getAsString().c_str());
Sean Callanane8a59a82010-09-13 21:34:21 +00001505
Sean Callananba992c52010-07-27 02:07:53 +00001506 return true;
1507}
1508
1509bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001510IRForTarget::ResolveCalls(Module &llvm_module, BasicBlock &basic_block)
Sean Callanan8bce6652010-07-13 21:41:46 +00001511{
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001512 /////////////////////////////////////////////////////////////////////////
1513 // Prepare the current basic block for execution in the remote process
1514 //
1515
Sean Callanan02fbafa2010-07-27 21:39:39 +00001516 BasicBlock::iterator ii;
Sean Callanan8bce6652010-07-13 21:41:46 +00001517
Greg Clayton3c7feb42010-11-19 01:05:25 +00001518 for (ii = basic_block.begin();
1519 ii != basic_block.end();
Sean Callanan8bce6652010-07-13 21:41:46 +00001520 ++ii)
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001521 {
Sean Callanan8bce6652010-07-13 21:41:46 +00001522 Instruction &inst = *ii;
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001523
Sean Callanan1d1b39c2010-11-08 00:31:32 +00001524 CallInst *call = dyn_cast<CallInst>(&inst);
Sean Callananba992c52010-07-27 02:07:53 +00001525
Sean Callanan97c924e2011-01-27 01:07:04 +00001526 // MaybeHandleCall handles error reporting; we are silent here
Greg Clayton3c7feb42010-11-19 01:05:25 +00001527 if (call && !MaybeHandleCall(llvm_module, call))
Sean Callanan1d1b39c2010-11-08 00:31:32 +00001528 return false;
Sean Callanan48443652010-12-02 19:47:57 +00001529
Sean Callanan97c924e2011-01-27 01:07:04 +00001530 // MaybeHandleCallArguments handles error reporting; we are silent here
Sean Callanan48443652010-12-02 19:47:57 +00001531 if (call && !MaybeHandleCallArguments(llvm_module, call))
1532 return false;
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001533 }
1534
1535 return true;
1536}
1537
Sean Callanan1d1b39c2010-11-08 00:31:32 +00001538bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001539IRForTarget::ResolveExternals (Module &llvm_module, Function &llvm_function)
Sean Callanan1d1b39c2010-11-08 00:31:32 +00001540{
Sean Callananae71e302010-11-18 22:21:58 +00001541 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
1542
Greg Clayton3c7feb42010-11-19 01:05:25 +00001543 for (Module::global_iterator global = llvm_module.global_begin(), end = llvm_module.global_end();
Sean Callanan1d1b39c2010-11-08 00:31:32 +00001544 global != end;
1545 ++global)
1546 {
Greg Clayton3c7feb42010-11-19 01:05:25 +00001547 if (log)
1548 log->Printf("Examining %s, DeclForGlobalValue returns %p",
1549 (*global).getName().str().c_str(),
Sean Callanan696cf5f2011-05-07 01:06:41 +00001550 DeclForGlobal(llvm_module, global));
Greg Clayton3c7feb42010-11-19 01:05:25 +00001551
Sean Callananc7674af2011-01-17 23:42:46 +00001552 if ((*global).getName().str().find("OBJC_IVAR") == 0)
1553 {
1554 if (!HandleSymbol(llvm_module, global))
Sean Callanan97c924e2011-01-27 01:07:04 +00001555 {
1556 if (m_error_stream)
1557 m_error_stream->Printf("Error [IRForTarget]: Couldn't find Objective-C indirect ivar symbol %s\n", (*global).getName().str().c_str());
1558
Sean Callananc7674af2011-01-17 23:42:46 +00001559 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001560 }
Sean Callananc7674af2011-01-17 23:42:46 +00001561 }
Sean Callanan696cf5f2011-05-07 01:06:41 +00001562 else if (DeclForGlobal(llvm_module, global))
Sean Callananc7674af2011-01-17 23:42:46 +00001563 {
1564 if (!MaybeHandleVariable (llvm_module, global))
Sean Callanan97c924e2011-01-27 01:07:04 +00001565 {
1566 if (m_error_stream)
1567 m_error_stream->Printf("Internal error [IRForTarget]: Couldn't rewrite external variable %s\n", (*global).getName().str().c_str());
1568
Sean Callananc7674af2011-01-17 23:42:46 +00001569 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001570 }
Sean Callananc7674af2011-01-17 23:42:46 +00001571 }
Sean Callanan1d1b39c2010-11-08 00:31:32 +00001572 }
1573
1574 return true;
1575}
1576
Sean Callanan02fbafa2010-07-27 21:39:39 +00001577static bool isGuardVariableRef(Value *V)
Sean Callanan45839272010-07-24 01:37:44 +00001578{
Sean Callanan6ba533e2010-11-17 23:00:36 +00001579 Constant *Old;
Sean Callanan45839272010-07-24 01:37:44 +00001580
Sean Callanan6ba533e2010-11-17 23:00:36 +00001581 if (!(Old = dyn_cast<Constant>(V)))
Sean Callanan45839272010-07-24 01:37:44 +00001582 return false;
1583
Sean Callanan47a5c4c2010-09-23 03:01:22 +00001584 ConstantExpr *CE;
1585
1586 if ((CE = dyn_cast<ConstantExpr>(V)))
1587 {
1588 if (CE->getOpcode() != Instruction::BitCast)
1589 return false;
1590
Sean Callanan6ba533e2010-11-17 23:00:36 +00001591 Old = CE->getOperand(0);
Sean Callanan47a5c4c2010-09-23 03:01:22 +00001592 }
1593
Sean Callanan6ba533e2010-11-17 23:00:36 +00001594 GlobalVariable *GV = dyn_cast<GlobalVariable>(Old);
Sean Callanan45839272010-07-24 01:37:44 +00001595
1596 if (!GV || !GV->hasName() || !GV->getName().startswith("_ZGV"))
1597 return false;
1598
1599 return true;
1600}
1601
Greg Clayton3c7feb42010-11-19 01:05:25 +00001602static void TurnGuardLoadIntoZero(Instruction* guard_load, Module &llvm_module)
Sean Callanan45839272010-07-24 01:37:44 +00001603{
Greg Clayton3c7feb42010-11-19 01:05:25 +00001604 Constant* zero(ConstantInt::get(Type::getInt8Ty(llvm_module.getContext()), 0, true));
Sean Callanan45839272010-07-24 01:37:44 +00001605
1606 Value::use_iterator ui;
1607
1608 for (ui = guard_load->use_begin();
1609 ui != guard_load->use_end();
1610 ++ui)
Sean Callananb5b749c2010-07-27 01:17:28 +00001611 {
Greg Clayton6e713402010-07-30 20:30:44 +00001612 if (isa<Constant>(*ui))
Sean Callananb5b749c2010-07-27 01:17:28 +00001613 {
1614 // do nothing for the moment
1615 }
1616 else
1617 {
1618 ui->replaceUsesOfWith(guard_load, zero);
1619 }
1620 }
Sean Callanan45839272010-07-24 01:37:44 +00001621
1622 guard_load->eraseFromParent();
1623}
1624
1625static void ExciseGuardStore(Instruction* guard_store)
1626{
1627 guard_store->eraseFromParent();
1628}
1629
1630bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001631IRForTarget::RemoveGuards(Module &llvm_module, BasicBlock &basic_block)
Sean Callanan45839272010-07-24 01:37:44 +00001632{
1633 ///////////////////////////////////////////////////////
1634 // Eliminate any reference to guard variables found.
1635 //
1636
Sean Callanan02fbafa2010-07-27 21:39:39 +00001637 BasicBlock::iterator ii;
Sean Callanan45839272010-07-24 01:37:44 +00001638
Sean Callanan02fbafa2010-07-27 21:39:39 +00001639 typedef SmallVector <Instruction*, 2> InstrList;
Sean Callanan45839272010-07-24 01:37:44 +00001640 typedef InstrList::iterator InstrIterator;
1641
1642 InstrList guard_loads;
1643 InstrList guard_stores;
1644
Greg Clayton3c7feb42010-11-19 01:05:25 +00001645 for (ii = basic_block.begin();
1646 ii != basic_block.end();
Sean Callanan45839272010-07-24 01:37:44 +00001647 ++ii)
1648 {
1649 Instruction &inst = *ii;
1650
1651 if (LoadInst *load = dyn_cast<LoadInst>(&inst))
1652 if (isGuardVariableRef(load->getPointerOperand()))
1653 guard_loads.push_back(&inst);
1654
1655 if (StoreInst *store = dyn_cast<StoreInst>(&inst))
1656 if (isGuardVariableRef(store->getPointerOperand()))
1657 guard_stores.push_back(&inst);
1658 }
1659
1660 InstrIterator iter;
1661
1662 for (iter = guard_loads.begin();
1663 iter != guard_loads.end();
1664 ++iter)
Greg Clayton3c7feb42010-11-19 01:05:25 +00001665 TurnGuardLoadIntoZero(*iter, llvm_module);
Sean Callanan45839272010-07-24 01:37:44 +00001666
1667 for (iter = guard_stores.begin();
1668 iter != guard_stores.end();
1669 ++iter)
1670 ExciseGuardStore(*iter);
1671
1672 return true;
1673}
1674
Sean Callanan97c924e2011-01-27 01:07:04 +00001675// This function does not report errors; its callers are responsible.
Sean Callanan6ba533e2010-11-17 23:00:36 +00001676bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001677IRForTarget::UnfoldConstant(Constant *old_constant, Value *new_constant, Instruction *first_entry_inst)
Sean Callananbafd6852010-07-14 23:40:29 +00001678{
Greg Claytone005f2c2010-11-06 01:53:30 +00001679 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
Sean Callananbafd6852010-07-14 23:40:29 +00001680
1681 Value::use_iterator ui;
1682
Sean Callanana48fe162010-08-11 03:57:18 +00001683 SmallVector<User*, 16> users;
1684
1685 // We do this because the use list might change, invalidating our iterator.
1686 // Much better to keep a work list ourselves.
Greg Clayton3c7feb42010-11-19 01:05:25 +00001687 for (ui = old_constant->use_begin();
1688 ui != old_constant->use_end();
Sean Callananbafd6852010-07-14 23:40:29 +00001689 ++ui)
Sean Callanana48fe162010-08-11 03:57:18 +00001690 users.push_back(*ui);
Sean Callananbafd6852010-07-14 23:40:29 +00001691
Sean Callanana48fe162010-08-11 03:57:18 +00001692 for (int i = 0;
1693 i < users.size();
1694 ++i)
1695 {
1696 User *user = users[i];
1697
Sean Callananbafd6852010-07-14 23:40:29 +00001698 if (Constant *constant = dyn_cast<Constant>(user))
1699 {
1700 // synthesize a new non-constant equivalent of the constant
1701
1702 if (ConstantExpr *constant_expr = dyn_cast<ConstantExpr>(constant))
1703 {
1704 switch (constant_expr->getOpcode())
1705 {
1706 default:
1707 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +00001708 log->Printf("Unhandled constant expression type: \"%s\"", PrintValue(constant_expr).c_str());
Sean Callananbafd6852010-07-14 23:40:29 +00001709 return false;
1710 case Instruction::BitCast:
1711 {
1712 // UnaryExpr
1713 // OperandList[0] is value
1714
1715 Value *s = constant_expr->getOperand(0);
1716
Greg Clayton3c7feb42010-11-19 01:05:25 +00001717 if (s == old_constant)
1718 s = new_constant;
Sean Callananbafd6852010-07-14 23:40:29 +00001719
Greg Clayton3c7feb42010-11-19 01:05:25 +00001720 BitCastInst *bit_cast(new BitCastInst(s, old_constant->getType(), "", first_entry_inst));
Sean Callananbafd6852010-07-14 23:40:29 +00001721
Greg Clayton3c7feb42010-11-19 01:05:25 +00001722 UnfoldConstant(constant_expr, bit_cast, first_entry_inst);
Sean Callananbafd6852010-07-14 23:40:29 +00001723 }
1724 break;
1725 case Instruction::GetElementPtr:
1726 {
1727 // GetElementPtrConstantExpr
1728 // OperandList[0] is base
1729 // OperandList[1]... are indices
1730
1731 Value *ptr = constant_expr->getOperand(0);
1732
Greg Clayton3c7feb42010-11-19 01:05:25 +00001733 if (ptr == old_constant)
1734 ptr = new_constant;
Sean Callananbafd6852010-07-14 23:40:29 +00001735
1736 SmallVector<Value*, 16> indices;
1737
1738 unsigned operand_index;
1739 unsigned num_operands = constant_expr->getNumOperands();
1740
1741 for (operand_index = 1;
1742 operand_index < num_operands;
1743 ++operand_index)
1744 {
1745 Value *operand = constant_expr->getOperand(operand_index);
1746
Greg Clayton3c7feb42010-11-19 01:05:25 +00001747 if (operand == old_constant)
1748 operand = new_constant;
Sean Callananbafd6852010-07-14 23:40:29 +00001749
1750 indices.push_back(operand);
1751 }
1752
Greg Clayton3c7feb42010-11-19 01:05:25 +00001753 GetElementPtrInst *get_element_ptr(GetElementPtrInst::Create(ptr, indices.begin(), indices.end(), "", first_entry_inst));
Sean Callananbafd6852010-07-14 23:40:29 +00001754
Greg Clayton3c7feb42010-11-19 01:05:25 +00001755 UnfoldConstant(constant_expr, get_element_ptr, first_entry_inst);
Sean Callananbafd6852010-07-14 23:40:29 +00001756 }
1757 break;
1758 }
1759 }
1760 else
1761 {
1762 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +00001763 log->Printf("Unhandled constant type: \"%s\"", PrintValue(constant).c_str());
Sean Callananbafd6852010-07-14 23:40:29 +00001764 return false;
1765 }
1766 }
1767 else
1768 {
1769 // simple fall-through case for non-constants
Greg Clayton3c7feb42010-11-19 01:05:25 +00001770 user->replaceUsesOfWith(old_constant, new_constant);
Sean Callananbafd6852010-07-14 23:40:29 +00001771 }
1772 }
1773
1774 return true;
1775}
1776
Sean Callanan8bce6652010-07-13 21:41:46 +00001777bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001778IRForTarget::ReplaceVariables (Module &llvm_module, Function &llvm_function)
Sean Callanan8bce6652010-07-13 21:41:46 +00001779{
Sean Callanane8a59a82010-09-13 21:34:21 +00001780 if (!m_resolve_vars)
1781 return true;
1782
Greg Claytone005f2c2010-11-06 01:53:30 +00001783 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
Sean Callanan8bce6652010-07-13 21:41:46 +00001784
1785 m_decl_map->DoStructLayout();
1786
1787 if (log)
1788 log->Printf("Element arrangement:");
1789
1790 uint32_t num_elements;
1791 uint32_t element_index;
1792
1793 size_t size;
1794 off_t alignment;
1795
1796 if (!m_decl_map->GetStructInfo (num_elements, size, alignment))
1797 return false;
1798
Greg Clayton3c7feb42010-11-19 01:05:25 +00001799 Function::arg_iterator iter(llvm_function.getArgumentList().begin());
Sean Callanan8bce6652010-07-13 21:41:46 +00001800
Greg Clayton3c7feb42010-11-19 01:05:25 +00001801 if (iter == llvm_function.getArgumentList().end())
Sean Callanan97c924e2011-01-27 01:07:04 +00001802 {
1803 if (m_error_stream)
1804 m_error_stream->Printf("Internal error [IRForTarget]: Wrapper takes no arguments (should take at least a struct pointer)");
1805
Sean Callanan8bce6652010-07-13 21:41:46 +00001806 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001807 }
1808
Sean Callanan02fbafa2010-07-27 21:39:39 +00001809 Argument *argument = iter;
Sean Callanan8bce6652010-07-13 21:41:46 +00001810
Sean Callanan3c9c5eb2010-09-21 00:44:12 +00001811 if (argument->getName().equals("this"))
1812 {
1813 ++iter;
1814
Greg Clayton3c7feb42010-11-19 01:05:25 +00001815 if (iter == llvm_function.getArgumentList().end())
Sean Callanan97c924e2011-01-27 01:07:04 +00001816 {
1817 if (m_error_stream)
1818 m_error_stream->Printf("Internal error [IRForTarget]: Wrapper takes only 'this' argument (should take a struct pointer too)");
1819
Sean Callanan3c9c5eb2010-09-21 00:44:12 +00001820 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001821 }
Sean Callanan3c9c5eb2010-09-21 00:44:12 +00001822
1823 argument = iter;
1824 }
Sean Callanan3aa7da52010-12-13 22:46:15 +00001825 else if (argument->getName().equals("self"))
1826 {
1827 ++iter;
1828
1829 if (iter == llvm_function.getArgumentList().end())
Sean Callanan97c924e2011-01-27 01:07:04 +00001830 {
1831 if (m_error_stream)
1832 m_error_stream->Printf("Internal error [IRForTarget]: Wrapper takes only 'self' argument (should take '_cmd' and a struct pointer too)");
1833
Sean Callanan3aa7da52010-12-13 22:46:15 +00001834 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001835 }
Sean Callanan3aa7da52010-12-13 22:46:15 +00001836
1837 if (!iter->getName().equals("_cmd"))
Sean Callanan97c924e2011-01-27 01:07:04 +00001838 {
1839 if (m_error_stream)
1840 m_error_stream->Printf("Internal error [IRForTarget]: Wrapper takes '%s' after 'self' argument (should take '_cmd')", iter->getName().str().c_str());
1841
Sean Callanan3aa7da52010-12-13 22:46:15 +00001842 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001843 }
Sean Callanan3aa7da52010-12-13 22:46:15 +00001844
1845 ++iter;
1846
1847 if (iter == llvm_function.getArgumentList().end())
Sean Callanan97c924e2011-01-27 01:07:04 +00001848 {
1849 if (m_error_stream)
1850 m_error_stream->Printf("Internal error [IRForTarget]: Wrapper takes only 'self' and '_cmd' arguments (should take a struct pointer too)");
1851
Sean Callanan3aa7da52010-12-13 22:46:15 +00001852 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001853 }
Sean Callanan3aa7da52010-12-13 22:46:15 +00001854
1855 argument = iter;
1856 }
Sean Callanan3c9c5eb2010-09-21 00:44:12 +00001857
Greg Clayton8de27c72010-10-15 22:48:33 +00001858 if (!argument->getName().equals("$__lldb_arg"))
Sean Callanan97c924e2011-01-27 01:07:04 +00001859 {
1860 if (m_error_stream)
1861 m_error_stream->Printf("Internal error [IRForTarget]: Wrapper takes an argument named '%s' instead of the struct pointer", argument->getName().str().c_str());
1862
Sean Callanan8bce6652010-07-13 21:41:46 +00001863 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001864 }
1865
Sean Callanan8bce6652010-07-13 21:41:46 +00001866 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +00001867 log->Printf("Arg: \"%s\"", PrintValue(argument).c_str());
Sean Callanan8bce6652010-07-13 21:41:46 +00001868
Greg Clayton3c7feb42010-11-19 01:05:25 +00001869 BasicBlock &entry_block(llvm_function.getEntryBlock());
Sean Callanan6ba533e2010-11-17 23:00:36 +00001870 Instruction *FirstEntryInstruction(entry_block.getFirstNonPHIOrDbg());
Sean Callanan8bce6652010-07-13 21:41:46 +00001871
Sean Callanan6ba533e2010-11-17 23:00:36 +00001872 if (!FirstEntryInstruction)
Sean Callanan97c924e2011-01-27 01:07:04 +00001873 {
1874 if (m_error_stream)
1875 m_error_stream->Printf("Internal error [IRForTarget]: Couldn't find the first instruction in the wrapper for use in rewriting");
1876
Sean Callanan8bce6652010-07-13 21:41:46 +00001877 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001878 }
Sean Callanan8bce6652010-07-13 21:41:46 +00001879
Greg Clayton3c7feb42010-11-19 01:05:25 +00001880 LLVMContext &context(llvm_module.getContext());
Sean Callanan8bce6652010-07-13 21:41:46 +00001881 const IntegerType *offset_type(Type::getInt32Ty(context));
1882
1883 if (!offset_type)
Sean Callanan97c924e2011-01-27 01:07:04 +00001884 {
1885 if (m_error_stream)
1886 m_error_stream->Printf("Internal error [IRForTarget]: Couldn't produce an offset type");
1887
Sean Callanan8bce6652010-07-13 21:41:46 +00001888 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001889 }
Sean Callanan8bce6652010-07-13 21:41:46 +00001890
1891 for (element_index = 0; element_index < num_elements; ++element_index)
1892 {
1893 const clang::NamedDecl *decl;
Sean Callanan02fbafa2010-07-27 21:39:39 +00001894 Value *value;
Sean Callanan8bce6652010-07-13 21:41:46 +00001895 off_t offset;
Greg Clayton8de27c72010-10-15 22:48:33 +00001896 lldb_private::ConstString name;
Sean Callanan8bce6652010-07-13 21:41:46 +00001897
Sean Callanan45690fe2010-08-30 22:17:16 +00001898 if (!m_decl_map->GetStructElement (decl, value, offset, name, element_index))
Sean Callanan97c924e2011-01-27 01:07:04 +00001899 {
1900 if (m_error_stream)
1901 m_error_stream->Printf("Internal error [IRForTarget]: Structure information is incomplete");
1902
Sean Callanan8bce6652010-07-13 21:41:46 +00001903 return false;
Sean Callanan97c924e2011-01-27 01:07:04 +00001904 }
1905
Sean Callanan8bce6652010-07-13 21:41:46 +00001906 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +00001907 log->Printf(" \"%s\" [\"%s\"] (\"%s\") placed at %d",
Sean Callanan82b74c82010-08-12 01:56:52 +00001908 value->getName().str().c_str(),
Greg Clayton8de27c72010-10-15 22:48:33 +00001909 name.GetCString(),
Sean Callanan8bce6652010-07-13 21:41:46 +00001910 PrintValue(value, true).c_str(),
1911 offset);
1912
1913 ConstantInt *offset_int(ConstantInt::getSigned(offset_type, offset));
Sean Callanan6ba533e2010-11-17 23:00:36 +00001914 GetElementPtrInst *get_element_ptr = GetElementPtrInst::Create(argument, offset_int, "", FirstEntryInstruction);
Sean Callanan6a925532011-01-13 08:53:35 +00001915
1916 Value *replacement;
Sean Callanan8bce6652010-07-13 21:41:46 +00001917
Sean Callanan6a925532011-01-13 08:53:35 +00001918 // Per the comment at ASTResultSynthesizer::SynthesizeBodyResult, in cases where the result
1919 // variable is an rvalue, we have to synthesize a dereference of the appropriate structure
1920 // entry in order to produce the static variable that the AST thinks it is accessing.
1921 if (name == m_result_name && !m_result_is_pointer)
1922 {
1923 BitCastInst *bit_cast = new BitCastInst(get_element_ptr, value->getType()->getPointerTo(), "", FirstEntryInstruction);
1924
1925 LoadInst *load = new LoadInst(bit_cast, "", FirstEntryInstruction);
1926
1927 replacement = load;
1928 }
Sean Callananbafd6852010-07-14 23:40:29 +00001929 else
Sean Callanan6a925532011-01-13 08:53:35 +00001930 {
1931 BitCastInst *bit_cast = new BitCastInst(get_element_ptr, value->getType(), "", FirstEntryInstruction);
1932
1933 replacement = bit_cast;
1934 }
1935
1936 if (Constant *constant = dyn_cast<Constant>(value))
1937 UnfoldConstant(constant, replacement, FirstEntryInstruction);
1938 else
1939 value->replaceAllUsesWith(replacement);
Sean Callananb51ee982010-11-02 23:51:17 +00001940
1941 if (GlobalVariable *var = dyn_cast<GlobalVariable>(value))
1942 var->eraseFromParent();
Sean Callanan8bce6652010-07-13 21:41:46 +00001943 }
1944
1945 if (log)
1946 log->Printf("Total structure [align %d, size %d]", alignment, size);
1947
1948 return true;
1949}
1950
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001951bool
Greg Clayton3c7feb42010-11-19 01:05:25 +00001952IRForTarget::runOnModule (Module &llvm_module)
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001953{
Greg Claytone005f2c2010-11-06 01:53:30 +00001954 lldb::LogSP log(lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_EXPRESSIONS));
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001955
Greg Clayton3c7feb42010-11-19 01:05:25 +00001956 Function* function = llvm_module.getFunction(StringRef(m_func_name.c_str()));
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001957
1958 if (!function)
1959 {
1960 if (log)
Greg Claytonb5037af2010-11-15 01:47:11 +00001961 log->Printf("Couldn't find \"%s()\" in the module", m_func_name.c_str());
Sean Callanan97c924e2011-01-27 01:07:04 +00001962
1963 if (m_error_stream)
1964 m_error_stream->Printf("Internal error [IRForTarget]: Couldn't find wrapper '%s' in the mdoule", m_func_name.c_str());
Sean Callanan6a925532011-01-13 08:53:35 +00001965
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001966 return false;
1967 }
1968
Sean Callanan02fbafa2010-07-27 21:39:39 +00001969 Function::iterator bbi;
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00001970
Sean Callanan05a5a1b2010-12-16 03:17:46 +00001971 m_has_side_effects = HasSideEffects(llvm_module, *function);
1972
Sean Callanan82b74c82010-08-12 01:56:52 +00001973 ////////////////////////////////////////////////////////////
Greg Clayton8de27c72010-10-15 22:48:33 +00001974 // Replace $__lldb_expr_result with a persistent variable
Sean Callanan82b74c82010-08-12 01:56:52 +00001975 //
1976
Greg Clayton3c7feb42010-11-19 01:05:25 +00001977 if (!CreateResultVariable(llvm_module, *function))
Sean Callanan3aa7da52010-12-13 22:46:15 +00001978 {
1979 if (log)
1980 log->Printf("CreateResultVariable() failed");
Sean Callanan97c924e2011-01-27 01:07:04 +00001981
1982 // CreateResultVariable() reports its own errors, so we don't do so here
1983
Sean Callanan82b74c82010-08-12 01:56:52 +00001984 return false;
Sean Callanan3aa7da52010-12-13 22:46:15 +00001985 }
Sean Callanan82b74c82010-08-12 01:56:52 +00001986
Sean Callanan696cf5f2011-05-07 01:06:41 +00001987 if (m_const_result)
1988 return true;
1989
Sean Callanan6ba533e2010-11-17 23:00:36 +00001990 ///////////////////////////////////////////////////////////////////////////////
1991 // Fix all Objective-C constant strings to use NSStringWithCString:encoding:
1992 //
Sean Callanan6ba533e2010-11-17 23:00:36 +00001993
Greg Clayton3c7feb42010-11-19 01:05:25 +00001994 if (!RewriteObjCConstStrings(llvm_module, *function))
Sean Callanan3aa7da52010-12-13 22:46:15 +00001995 {
1996 if (log)
1997 log->Printf("RewriteObjCConstStrings() failed");
Sean Callanan97c924e2011-01-27 01:07:04 +00001998
1999 // RewriteObjCConstStrings() reports its own errors, so we don't do so here
2000
Sean Callanan6ba533e2010-11-17 23:00:36 +00002001 return false;
Sean Callanan3aa7da52010-12-13 22:46:15 +00002002 }
Sean Callanan6ba533e2010-11-17 23:00:36 +00002003
Sean Callananf5857a02010-07-31 01:32:05 +00002004 //////////////////////////////////
2005 // Run basic-block level passes
2006 //
2007
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00002008 for (bbi = function->begin();
2009 bbi != function->end();
2010 ++bbi)
2011 {
Greg Clayton3c7feb42010-11-19 01:05:25 +00002012 if (!RemoveGuards(llvm_module, *bbi))
Sean Callanan3aa7da52010-12-13 22:46:15 +00002013 {
2014 if (log)
2015 log->Printf("RemoveGuards() failed");
Sean Callanan97c924e2011-01-27 01:07:04 +00002016
2017 // RemoveGuards() reports its own errors, so we don't do so here
2018
Sean Callanan8c127202010-08-23 23:09:38 +00002019 return false;
Sean Callanan3aa7da52010-12-13 22:46:15 +00002020 }
Sean Callanan8c127202010-08-23 23:09:38 +00002021
Greg Clayton3c7feb42010-11-19 01:05:25 +00002022 if (!RewritePersistentAllocs(llvm_module, *bbi))
Sean Callanan3aa7da52010-12-13 22:46:15 +00002023 {
2024 if (log)
2025 log->Printf("RewritePersistentAllocs() failed");
Sean Callanan97c924e2011-01-27 01:07:04 +00002026
2027 // RewritePersistentAllocs() reports its own errors, so we don't do so here
2028
Sean Callananf5857a02010-07-31 01:32:05 +00002029 return false;
Sean Callanan3aa7da52010-12-13 22:46:15 +00002030 }
Sean Callananf5857a02010-07-31 01:32:05 +00002031
Greg Clayton3c7feb42010-11-19 01:05:25 +00002032 if (!RewriteObjCSelectors(llvm_module, *bbi))
Sean Callanan3aa7da52010-12-13 22:46:15 +00002033 {
2034 if (log)
2035 log->Printf("RewriteObjCSelectors() failed");
Sean Callanan97c924e2011-01-27 01:07:04 +00002036
2037 // RewriteObjCSelectors() reports its own errors, so we don't do so here
2038
Sean Callanana48fe162010-08-11 03:57:18 +00002039 return false;
Sean Callanan3aa7da52010-12-13 22:46:15 +00002040 }
Sean Callanana48fe162010-08-11 03:57:18 +00002041
Greg Clayton3c7feb42010-11-19 01:05:25 +00002042 if (!ResolveCalls(llvm_module, *bbi))
Sean Callanan3aa7da52010-12-13 22:46:15 +00002043 {
2044 if (log)
2045 log->Printf("ResolveCalls() failed");
Sean Callanan97c924e2011-01-27 01:07:04 +00002046
2047 // ResolveCalls() reports its own errors, so we don't do so here
2048
Sean Callanan8bce6652010-07-13 21:41:46 +00002049 return false;
Sean Callanan3aa7da52010-12-13 22:46:15 +00002050 }
Sean Callanan8bce6652010-07-13 21:41:46 +00002051 }
2052
Sean Callanan771131d2010-09-30 21:18:25 +00002053 ///////////////////////////////
2054 // Run function-level passes
2055 //
2056
Greg Clayton3c7feb42010-11-19 01:05:25 +00002057 if (!ResolveExternals(llvm_module, *function))
Sean Callanan3aa7da52010-12-13 22:46:15 +00002058 {
2059 if (log)
2060 log->Printf("ResolveExternals() failed");
Sean Callanan97c924e2011-01-27 01:07:04 +00002061
2062 // ResolveExternals() reports its own errors, so we don't do so here
2063
Sean Callanan1d1b39c2010-11-08 00:31:32 +00002064 return false;
Sean Callanan3aa7da52010-12-13 22:46:15 +00002065 }
Sean Callanan1d1b39c2010-11-08 00:31:32 +00002066
Greg Clayton3c7feb42010-11-19 01:05:25 +00002067 if (!ReplaceVariables(llvm_module, *function))
Sean Callanan3aa7da52010-12-13 22:46:15 +00002068 {
2069 if (log)
2070 log->Printf("ReplaceVariables() failed");
Sean Callanan97c924e2011-01-27 01:07:04 +00002071
2072 // ReplaceVariables() reports its own errors, so we don't do so here
2073
Sean Callanan771131d2010-09-30 21:18:25 +00002074 return false;
Sean Callanan3aa7da52010-12-13 22:46:15 +00002075 }
Sean Callanan771131d2010-09-30 21:18:25 +00002076
Sean Callanan8bce6652010-07-13 21:41:46 +00002077 if (log)
2078 {
Sean Callanan321fe9e2010-07-28 01:00:59 +00002079 std::string s;
2080 raw_string_ostream oss(s);
Sean Callanan8bce6652010-07-13 21:41:46 +00002081
Greg Clayton3c7feb42010-11-19 01:05:25 +00002082 llvm_module.print(oss, NULL);
Sean Callanan321fe9e2010-07-28 01:00:59 +00002083
2084 oss.flush();
2085
Greg Claytonb5037af2010-11-15 01:47:11 +00002086 log->Printf("Module after preparing for execution: \n\"%s\"", s.c_str());
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00002087 }
2088
2089 return true;
2090}
2091
2092void
Greg Clayton3c7feb42010-11-19 01:05:25 +00002093IRForTarget::assignPassManager (PMStack &pass_mgr_stack, PassManagerType pass_mgr_type)
Sean Callanan5cf4a1c2010-07-03 01:35:46 +00002094{
2095}
2096
2097PassManagerType
2098IRForTarget::getPotentialPassManagerType() const
2099{
2100 return PMT_ModulePassManager;
2101}