1 //===-- ClangUserExpression.cpp ---------------------------------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include <stdio.h>
10 #if HAVE_SYS_TYPES_H
11 #include <sys/types.h>
12 #endif
13 
14 #include <cstdlib>
15 #include <map>
16 #include <string>
17 
18 #include "ClangUserExpression.h"
19 
20 #include "ASTResultSynthesizer.h"
21 #include "ClangDiagnostic.h"
22 #include "ClangExpressionDeclMap.h"
23 #include "ClangExpressionParser.h"
24 #include "ClangExpressionSourceCode.h"
25 #include "ClangModulesDeclVendor.h"
26 #include "ClangPersistentVariables.h"
27 
28 #include "lldb/Core/Debugger.h"
29 #include "lldb/Core/Module.h"
30 #include "lldb/Core/StreamFile.h"
31 #include "lldb/Core/ValueObjectConstResult.h"
32 #include "lldb/Expression/ExpressionSourceCode.h"
33 #include "lldb/Expression/IRExecutionUnit.h"
34 #include "lldb/Expression/IRInterpreter.h"
35 #include "lldb/Expression/Materializer.h"
36 #include "lldb/Host/HostInfo.h"
37 #include "lldb/Symbol/Block.h"
38 #include "lldb/Symbol/ClangASTContext.h"
39 #include "lldb/Symbol/ClangExternalASTSourceCommon.h"
40 #include "lldb/Symbol/CompileUnit.h"
41 #include "lldb/Symbol/Function.h"
42 #include "lldb/Symbol/ObjectFile.h"
43 #include "lldb/Symbol/SymbolVendor.h"
44 #include "lldb/Symbol/Type.h"
45 #include "lldb/Symbol/VariableList.h"
46 #include "lldb/Target/ExecutionContext.h"
47 #include "lldb/Target/Process.h"
48 #include "lldb/Target/StackFrame.h"
49 #include "lldb/Target/Target.h"
50 #include "lldb/Target/ThreadPlan.h"
51 #include "lldb/Target/ThreadPlanCallUserExpression.h"
52 #include "lldb/Utility/ConstString.h"
53 #include "lldb/Utility/Log.h"
54 #include "lldb/Utility/StreamString.h"
55 
56 #include "clang/AST/DeclCXX.h"
57 #include "clang/AST/DeclObjC.h"
58 
59 using namespace lldb_private;
60 
61 ClangUserExpression::ClangUserExpression(
62     ExecutionContextScope &exe_scope, llvm::StringRef expr,
63     llvm::StringRef prefix, lldb::LanguageType language,
64     ResultType desired_type, const EvaluateExpressionOptions &options,
65     ValueObject *ctx_obj)
66     : LLVMUserExpression(exe_scope, expr, prefix, language, desired_type,
67                          options, eKindClangUserExpression),
68       m_type_system_helper(*m_target_wp.lock(), options.GetExecutionPolicy() ==
69                                                     eExecutionPolicyTopLevel),
70       m_result_delegate(exe_scope.CalculateTarget()), m_ctx_obj(ctx_obj) {
71   switch (m_language) {
72   case lldb::eLanguageTypeC_plus_plus:
73     m_allow_cxx = true;
74     break;
75   case lldb::eLanguageTypeObjC:
76     m_allow_objc = true;
77     break;
78   case lldb::eLanguageTypeObjC_plus_plus:
79   default:
80     m_allow_cxx = true;
81     m_allow_objc = true;
82     break;
83   }
84 }
85 
86 ClangUserExpression::~ClangUserExpression() {}
87 
88 void ClangUserExpression::ScanContext(ExecutionContext &exe_ctx, Status &err) {
89   Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
90 
91   if (log)
92     log->Printf("ClangUserExpression::ScanContext()");
93 
94   m_target = exe_ctx.GetTargetPtr();
95 
96   if (!(m_allow_cxx || m_allow_objc)) {
97     if (log)
98       log->Printf("  [CUE::SC] Settings inhibit C++ and Objective-C");
99     return;
100   }
101 
102   StackFrame *frame = exe_ctx.GetFramePtr();
103   if (frame == NULL) {
104     if (log)
105       log->Printf("  [CUE::SC] Null stack frame");
106     return;
107   }
108 
109   SymbolContext sym_ctx = frame->GetSymbolContext(lldb::eSymbolContextFunction |
110                                                   lldb::eSymbolContextBlock);
111 
112   if (!sym_ctx.function) {
113     if (log)
114       log->Printf("  [CUE::SC] Null function");
115     return;
116   }
117 
118   // Find the block that defines the function represented by "sym_ctx"
119   Block *function_block = sym_ctx.GetFunctionBlock();
120 
121   if (!function_block) {
122     if (log)
123       log->Printf("  [CUE::SC] Null function block");
124     return;
125   }
126 
127   CompilerDeclContext decl_context = function_block->GetDeclContext();
128 
129   if (!decl_context) {
130     if (log)
131       log->Printf("  [CUE::SC] Null decl context");
132     return;
133   }
134 
135   if (m_ctx_obj) {
136     switch (m_ctx_obj->GetObjectRuntimeLanguage()) {
137     case lldb::eLanguageTypeC:
138     case lldb::eLanguageTypeC89:
139     case lldb::eLanguageTypeC99:
140     case lldb::eLanguageTypeC11:
141     case lldb::eLanguageTypeC_plus_plus:
142     case lldb::eLanguageTypeC_plus_plus_03:
143     case lldb::eLanguageTypeC_plus_plus_11:
144     case lldb::eLanguageTypeC_plus_plus_14:
145       m_in_cplusplus_method = true;
146       break;
147     case lldb::eLanguageTypeObjC:
148     case lldb::eLanguageTypeObjC_plus_plus:
149       m_in_objectivec_method = true;
150       break;
151     default:
152       break;
153     }
154     m_needs_object_ptr = true;
155   } else if (clang::CXXMethodDecl *method_decl =
156           ClangASTContext::DeclContextGetAsCXXMethodDecl(decl_context)) {
157     if (m_allow_cxx && method_decl->isInstance()) {
158       if (m_enforce_valid_object) {
159         lldb::VariableListSP variable_list_sp(
160             function_block->GetBlockVariableList(true));
161 
162         const char *thisErrorString = "Stopped in a C++ method, but 'this' "
163                                       "isn't available; pretending we are in a "
164                                       "generic context";
165 
166         if (!variable_list_sp) {
167           err.SetErrorString(thisErrorString);
168           return;
169         }
170 
171         lldb::VariableSP this_var_sp(
172             variable_list_sp->FindVariable(ConstString("this")));
173 
174         if (!this_var_sp || !this_var_sp->IsInScope(frame) ||
175             !this_var_sp->LocationIsValidForFrame(frame)) {
176           err.SetErrorString(thisErrorString);
177           return;
178         }
179       }
180 
181       m_in_cplusplus_method = true;
182       m_needs_object_ptr = true;
183     }
184   } else if (clang::ObjCMethodDecl *method_decl =
185                  ClangASTContext::DeclContextGetAsObjCMethodDecl(
186                      decl_context)) {
187     if (m_allow_objc) {
188       if (m_enforce_valid_object) {
189         lldb::VariableListSP variable_list_sp(
190             function_block->GetBlockVariableList(true));
191 
192         const char *selfErrorString = "Stopped in an Objective-C method, but "
193                                       "'self' isn't available; pretending we "
194                                       "are in a generic context";
195 
196         if (!variable_list_sp) {
197           err.SetErrorString(selfErrorString);
198           return;
199         }
200 
201         lldb::VariableSP self_variable_sp =
202             variable_list_sp->FindVariable(ConstString("self"));
203 
204         if (!self_variable_sp || !self_variable_sp->IsInScope(frame) ||
205             !self_variable_sp->LocationIsValidForFrame(frame)) {
206           err.SetErrorString(selfErrorString);
207           return;
208         }
209       }
210 
211       m_in_objectivec_method = true;
212       m_needs_object_ptr = true;
213 
214       if (!method_decl->isInstanceMethod())
215         m_in_static_method = true;
216     }
217   } else if (clang::FunctionDecl *function_decl =
218                  ClangASTContext::DeclContextGetAsFunctionDecl(decl_context)) {
219     // We might also have a function that said in the debug information that it
220     // captured an object pointer.  The best way to deal with getting to the
221     // ivars at present is by pretending that this is a method of a class in
222     // whatever runtime the debug info says the object pointer belongs to.  Do
223     // that here.
224 
225     ClangASTMetadata *metadata =
226         ClangASTContext::DeclContextGetMetaData(decl_context, function_decl);
227     if (metadata && metadata->HasObjectPtr()) {
228       lldb::LanguageType language = metadata->GetObjectPtrLanguage();
229       if (language == lldb::eLanguageTypeC_plus_plus) {
230         if (m_enforce_valid_object) {
231           lldb::VariableListSP variable_list_sp(
232               function_block->GetBlockVariableList(true));
233 
234           const char *thisErrorString = "Stopped in a context claiming to "
235                                         "capture a C++ object pointer, but "
236                                         "'this' isn't available; pretending we "
237                                         "are in a generic context";
238 
239           if (!variable_list_sp) {
240             err.SetErrorString(thisErrorString);
241             return;
242           }
243 
244           lldb::VariableSP this_var_sp(
245               variable_list_sp->FindVariable(ConstString("this")));
246 
247           if (!this_var_sp || !this_var_sp->IsInScope(frame) ||
248               !this_var_sp->LocationIsValidForFrame(frame)) {
249             err.SetErrorString(thisErrorString);
250             return;
251           }
252         }
253 
254         m_in_cplusplus_method = true;
255         m_needs_object_ptr = true;
256       } else if (language == lldb::eLanguageTypeObjC) {
257         if (m_enforce_valid_object) {
258           lldb::VariableListSP variable_list_sp(
259               function_block->GetBlockVariableList(true));
260 
261           const char *selfErrorString =
262               "Stopped in a context claiming to capture an Objective-C object "
263               "pointer, but 'self' isn't available; pretending we are in a "
264               "generic context";
265 
266           if (!variable_list_sp) {
267             err.SetErrorString(selfErrorString);
268             return;
269           }
270 
271           lldb::VariableSP self_variable_sp =
272               variable_list_sp->FindVariable(ConstString("self"));
273 
274           if (!self_variable_sp || !self_variable_sp->IsInScope(frame) ||
275               !self_variable_sp->LocationIsValidForFrame(frame)) {
276             err.SetErrorString(selfErrorString);
277             return;
278           }
279 
280           Type *self_type = self_variable_sp->GetType();
281 
282           if (!self_type) {
283             err.SetErrorString(selfErrorString);
284             return;
285           }
286 
287           CompilerType self_clang_type = self_type->GetForwardCompilerType();
288 
289           if (!self_clang_type) {
290             err.SetErrorString(selfErrorString);
291             return;
292           }
293 
294           if (ClangASTContext::IsObjCClassType(self_clang_type)) {
295             return;
296           } else if (ClangASTContext::IsObjCObjectPointerType(
297                          self_clang_type)) {
298             m_in_objectivec_method = true;
299             m_needs_object_ptr = true;
300           } else {
301             err.SetErrorString(selfErrorString);
302             return;
303           }
304         } else {
305           m_in_objectivec_method = true;
306           m_needs_object_ptr = true;
307         }
308       }
309     }
310   }
311 }
312 
313 // This is a really nasty hack, meant to fix Objective-C expressions of the
314 // form (int)[myArray count].  Right now, because the type information for
315 // count is not available, [myArray count] returns id, which can't be directly
316 // cast to int without causing a clang error.
317 static void ApplyObjcCastHack(std::string &expr) {
318 #define OBJC_CAST_HACK_FROM "(int)["
319 #define OBJC_CAST_HACK_TO "(int)(long long)["
320 
321   size_t from_offset;
322 
323   while ((from_offset = expr.find(OBJC_CAST_HACK_FROM)) != expr.npos)
324     expr.replace(from_offset, sizeof(OBJC_CAST_HACK_FROM) - 1,
325                  OBJC_CAST_HACK_TO);
326 
327 #undef OBJC_CAST_HACK_TO
328 #undef OBJC_CAST_HACK_FROM
329 }
330 
331 namespace {
332 // Utility guard that calls a callback when going out of scope.
333 class OnExit {
334 public:
335   typedef std::function<void(void)> Callback;
336 
337   OnExit(Callback const &callback) : m_callback(callback) {}
338 
339   ~OnExit() { m_callback(); }
340 
341 private:
342   Callback m_callback;
343 };
344 } // namespace
345 
346 bool ClangUserExpression::SetupPersistentState(DiagnosticManager &diagnostic_manager,
347                                  ExecutionContext &exe_ctx) {
348   if (Target *target = exe_ctx.GetTargetPtr()) {
349     if (PersistentExpressionState *persistent_state =
350             target->GetPersistentExpressionStateForLanguage(
351                 lldb::eLanguageTypeC)) {
352       m_result_delegate.RegisterPersistentState(persistent_state);
353     } else {
354       diagnostic_manager.PutString(
355           eDiagnosticSeverityError,
356           "couldn't start parsing (no persistent data)");
357       return false;
358     }
359   } else {
360     diagnostic_manager.PutString(eDiagnosticSeverityError,
361                                  "error: couldn't start parsing (no target)");
362     return false;
363   }
364   return true;
365 }
366 
367 static void SetupDeclVendor(ExecutionContext &exe_ctx, Target *target) {
368   if (ClangModulesDeclVendor *decl_vendor =
369           target->GetClangModulesDeclVendor()) {
370     const ClangModulesDeclVendor::ModuleVector &hand_imported_modules =
371         llvm::cast<ClangPersistentVariables>(
372             target->GetPersistentExpressionStateForLanguage(
373                 lldb::eLanguageTypeC))
374             ->GetHandLoadedClangModules();
375     ClangModulesDeclVendor::ModuleVector modules_for_macros;
376 
377     for (ClangModulesDeclVendor::ModuleID module : hand_imported_modules) {
378       modules_for_macros.push_back(module);
379     }
380 
381     if (target->GetEnableAutoImportClangModules()) {
382       if (StackFrame *frame = exe_ctx.GetFramePtr()) {
383         if (Block *block = frame->GetFrameBlock()) {
384           SymbolContext sc;
385 
386           block->CalculateSymbolContext(&sc);
387 
388           if (sc.comp_unit) {
389             StreamString error_stream;
390 
391             decl_vendor->AddModulesForCompileUnit(
392                 *sc.comp_unit, modules_for_macros, error_stream);
393           }
394         }
395       }
396     }
397   }
398 }
399 
400 void ClangUserExpression::UpdateLanguageForExpr(
401     DiagnosticManager &diagnostic_manager, ExecutionContext &exe_ctx,
402     std::vector<std::string> modules_to_import) {
403   m_expr_lang = lldb::LanguageType::eLanguageTypeUnknown;
404 
405   std::string prefix = m_expr_prefix;
406 
407   if (m_options.GetExecutionPolicy() == eExecutionPolicyTopLevel) {
408     m_transformed_text = m_expr_text;
409   } else {
410     std::unique_ptr<ClangExpressionSourceCode> source_code(
411         ClangExpressionSourceCode::CreateWrapped(prefix.c_str(),
412                                             m_expr_text.c_str()));
413 
414     if (m_in_cplusplus_method)
415       m_expr_lang = lldb::eLanguageTypeC_plus_plus;
416     else if (m_in_objectivec_method)
417       m_expr_lang = lldb::eLanguageTypeObjC;
418     else
419       m_expr_lang = lldb::eLanguageTypeC;
420 
421     if (!source_code->GetText(m_transformed_text, m_expr_lang,
422                               m_in_static_method, exe_ctx, !m_ctx_obj,
423                               modules_to_import)) {
424       diagnostic_manager.PutString(eDiagnosticSeverityError,
425                                    "couldn't construct expression body");
426       return;
427     }
428 
429     // Find and store the start position of the original code inside the
430     // transformed code. We need this later for the code completion.
431     std::size_t original_start;
432     std::size_t original_end;
433     bool found_bounds = source_code->GetOriginalBodyBounds(
434         m_transformed_text, m_expr_lang, original_start, original_end);
435     if (found_bounds) {
436       m_user_expression_start_pos = original_start;
437     }
438   }
439 }
440 
441 static bool SupportsCxxModuleImport(lldb::LanguageType language) {
442   switch (language) {
443   case lldb::eLanguageTypeC_plus_plus:
444   case lldb::eLanguageTypeC_plus_plus_03:
445   case lldb::eLanguageTypeC_plus_plus_11:
446   case lldb::eLanguageTypeC_plus_plus_14:
447   case lldb::eLanguageTypeObjC_plus_plus:
448     return true;
449   default:
450     return false;
451   }
452 }
453 
454 std::vector<std::string>
455 ClangUserExpression::GetModulesToImport(ExecutionContext &exe_ctx) {
456   Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
457 
458   if (!SupportsCxxModuleImport(Language()))
459     return {};
460 
461   Target *target = exe_ctx.GetTargetPtr();
462   if (!target || !target->GetEnableImportStdModule())
463     return {};
464 
465   StackFrame *frame = exe_ctx.GetFramePtr();
466   if (!frame)
467     return {};
468 
469   Block *block = frame->GetFrameBlock();
470   if (!block)
471     return {};
472 
473   SymbolContext sc;
474   block->CalculateSymbolContext(&sc);
475   if (!sc.comp_unit)
476     return {};
477 
478   if (log) {
479     for (const SourceModule &m : sc.comp_unit->GetImportedModules()) {
480       LLDB_LOG(log, "Found module in compile unit: {0:$[.]} - include dir: {1}",
481                   llvm::make_range(m.path.begin(), m.path.end()), m.search_path);
482     }
483   }
484 
485   for (const SourceModule &m : sc.comp_unit->GetImportedModules())
486     m_include_directories.push_back(m.search_path);
487 
488   // Check if we imported 'std' or any of its submodules.
489   // We currently don't support importing any other modules in the expression
490   // parser.
491   for (const SourceModule &m : sc.comp_unit->GetImportedModules())
492     if (!m.path.empty() && m.path.front() == ConstString("std"))
493       return {"std"};
494 
495   return {};
496 }
497 
498 bool ClangUserExpression::PrepareForParsing(
499     DiagnosticManager &diagnostic_manager, ExecutionContext &exe_ctx) {
500   Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
501 
502   InstallContext(exe_ctx);
503 
504   if (!SetupPersistentState(diagnostic_manager, exe_ctx))
505     return false;
506 
507   Status err;
508   ScanContext(exe_ctx, err);
509 
510   if (!err.Success()) {
511     diagnostic_manager.PutString(eDiagnosticSeverityWarning, err.AsCString());
512   }
513 
514   ////////////////////////////////////
515   // Generate the expression
516   //
517 
518   ApplyObjcCastHack(m_expr_text);
519 
520   SetupDeclVendor(exe_ctx, m_target);
521 
522   std::vector<std::string> used_modules = GetModulesToImport(exe_ctx);
523   m_imported_cpp_modules = !used_modules.empty();
524 
525   LLDB_LOG(log, "List of imported modules in expression: {0}",
526            llvm::make_range(used_modules.begin(), used_modules.end()));
527 
528   UpdateLanguageForExpr(diagnostic_manager, exe_ctx, used_modules);
529   return true;
530 }
531 
532 bool ClangUserExpression::Parse(DiagnosticManager &diagnostic_manager,
533                                 ExecutionContext &exe_ctx,
534                                 lldb_private::ExecutionPolicy execution_policy,
535                                 bool keep_result_in_memory,
536                                 bool generate_debug_info) {
537   Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
538 
539   if (!PrepareForParsing(diagnostic_manager, exe_ctx))
540     return false;
541 
542   if (log)
543     log->Printf("Parsing the following code:\n%s", m_transformed_text.c_str());
544 
545   ////////////////////////////////////
546   // Set up the target and compiler
547   //
548 
549   Target *target = exe_ctx.GetTargetPtr();
550 
551   if (!target) {
552     diagnostic_manager.PutString(eDiagnosticSeverityError, "invalid target");
553     return false;
554   }
555 
556   //////////////////////////
557   // Parse the expression
558   //
559 
560   m_materializer_up.reset(new Materializer());
561 
562   ResetDeclMap(exe_ctx, m_result_delegate, keep_result_in_memory);
563 
564   OnExit on_exit([this]() { ResetDeclMap(); });
565 
566   if (!DeclMap()->WillParse(exe_ctx, m_materializer_up.get())) {
567     diagnostic_manager.PutString(
568         eDiagnosticSeverityError,
569         "current process state is unsuitable for expression parsing");
570     return false;
571   }
572 
573   if (m_options.GetExecutionPolicy() == eExecutionPolicyTopLevel) {
574     DeclMap()->SetLookupsEnabled(true);
575   }
576 
577   Process *process = exe_ctx.GetProcessPtr();
578   ExecutionContextScope *exe_scope = process;
579 
580   if (!exe_scope)
581     exe_scope = exe_ctx.GetTargetPtr();
582 
583   // We use a shared pointer here so we can use the original parser - if it
584   // succeeds or the rewrite parser we might make if it fails.  But the
585   // parser_sp will never be empty.
586 
587   ClangExpressionParser parser(exe_scope, *this, generate_debug_info,
588                                m_include_directories);
589 
590   unsigned num_errors = parser.Parse(diagnostic_manager);
591 
592   // Check here for FixItHints.  If there are any try to apply the fixits and
593   // set the fixed text in m_fixed_text before returning an error.
594   if (num_errors) {
595     if (diagnostic_manager.HasFixIts()) {
596       if (parser.RewriteExpression(diagnostic_manager)) {
597         size_t fixed_start;
598         size_t fixed_end;
599         const std::string &fixed_expression =
600             diagnostic_manager.GetFixedExpression();
601         if (ClangExpressionSourceCode::GetOriginalBodyBounds(
602                 fixed_expression, m_expr_lang, fixed_start, fixed_end))
603           m_fixed_text =
604               fixed_expression.substr(fixed_start, fixed_end - fixed_start);
605       }
606     }
607     return false;
608   }
609 
610   //////////////////////////////////////////////////////////////////////////////////////////
611   // Prepare the output of the parser for execution, evaluating it statically
612   // if possible
613   //
614 
615   {
616     Status jit_error = parser.PrepareForExecution(
617         m_jit_start_addr, m_jit_end_addr, m_execution_unit_sp, exe_ctx,
618         m_can_interpret, execution_policy);
619 
620     if (!jit_error.Success()) {
621       const char *error_cstr = jit_error.AsCString();
622       if (error_cstr && error_cstr[0])
623         diagnostic_manager.PutString(eDiagnosticSeverityError, error_cstr);
624       else
625         diagnostic_manager.PutString(eDiagnosticSeverityError,
626                                      "expression can't be interpreted or run");
627       return false;
628     }
629   }
630 
631   if (exe_ctx.GetProcessPtr() && execution_policy == eExecutionPolicyTopLevel) {
632     Status static_init_error =
633         parser.RunStaticInitializers(m_execution_unit_sp, exe_ctx);
634 
635     if (!static_init_error.Success()) {
636       const char *error_cstr = static_init_error.AsCString();
637       if (error_cstr && error_cstr[0])
638         diagnostic_manager.Printf(eDiagnosticSeverityError,
639                                   "couldn't run static initializers: %s\n",
640                                   error_cstr);
641       else
642         diagnostic_manager.PutString(eDiagnosticSeverityError,
643                                      "couldn't run static initializers\n");
644       return false;
645     }
646   }
647 
648   if (m_execution_unit_sp) {
649     bool register_execution_unit = false;
650 
651     if (m_options.GetExecutionPolicy() == eExecutionPolicyTopLevel) {
652       register_execution_unit = true;
653     }
654 
655     // If there is more than one external function in the execution unit, it
656     // needs to keep living even if it's not top level, because the result
657     // could refer to that function.
658 
659     if (m_execution_unit_sp->GetJittedFunctions().size() > 1) {
660       register_execution_unit = true;
661     }
662 
663     if (register_execution_unit) {
664       llvm::cast<PersistentExpressionState>(
665           exe_ctx.GetTargetPtr()->GetPersistentExpressionStateForLanguage(
666               m_language))
667           ->RegisterExecutionUnit(m_execution_unit_sp);
668     }
669   }
670 
671   if (generate_debug_info) {
672     lldb::ModuleSP jit_module_sp(m_execution_unit_sp->GetJITModule());
673 
674     if (jit_module_sp) {
675       ConstString const_func_name(FunctionName());
676       FileSpec jit_file;
677       jit_file.GetFilename() = const_func_name;
678       jit_module_sp->SetFileSpecAndObjectName(jit_file, ConstString());
679       m_jit_module_wp = jit_module_sp;
680       target->GetImages().Append(jit_module_sp);
681     }
682   }
683 
684   if (process && m_jit_start_addr != LLDB_INVALID_ADDRESS)
685     m_jit_process_wp = lldb::ProcessWP(process->shared_from_this());
686   return true;
687 }
688 
689 //------------------------------------------------------------------
690 /// Converts an absolute position inside a given code string into
691 /// a column/line pair.
692 ///
693 /// \param[in] abs_pos
694 ///     A absolute position in the code string that we want to convert
695 ///     to a column/line pair.
696 ///
697 /// \param[in] code
698 ///     A multi-line string usually representing source code.
699 ///
700 /// \param[out] line
701 ///     The line in the code that contains the given absolute position.
702 ///     The first line in the string is indexed as 1.
703 ///
704 /// \param[out] column
705 ///     The column in the line that contains the absolute position.
706 ///     The first character in a line is indexed as 0.
707 //------------------------------------------------------------------
708 static void AbsPosToLineColumnPos(size_t abs_pos, llvm::StringRef code,
709                                   unsigned &line, unsigned &column) {
710   // Reset to code position to beginning of the file.
711   line = 0;
712   column = 0;
713 
714   assert(abs_pos <= code.size() && "Absolute position outside code string?");
715 
716   // We have to walk up to the position and count lines/columns.
717   for (std::size_t i = 0; i < abs_pos; ++i) {
718     // If we hit a line break, we go back to column 0 and enter a new line.
719     // We only handle \n because that's what we internally use to make new
720     // lines for our temporary code strings.
721     if (code[i] == '\n') {
722       ++line;
723       column = 0;
724       continue;
725     }
726     ++column;
727   }
728 }
729 
730 bool ClangUserExpression::Complete(ExecutionContext &exe_ctx,
731                                    CompletionRequest &request,
732                                    unsigned complete_pos) {
733   Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_EXPRESSIONS));
734 
735   // We don't want any visible feedback when completing an expression. Mostly
736   // because the results we get from an incomplete invocation are probably not
737   // correct.
738   DiagnosticManager diagnostic_manager;
739 
740   if (!PrepareForParsing(diagnostic_manager, exe_ctx))
741     return false;
742 
743   if (log)
744     log->Printf("Parsing the following code:\n%s", m_transformed_text.c_str());
745 
746   //////////////////////////
747   // Parse the expression
748   //
749 
750   m_materializer_up.reset(new Materializer());
751 
752   ResetDeclMap(exe_ctx, m_result_delegate, /*keep result in memory*/ true);
753 
754   OnExit on_exit([this]() { ResetDeclMap(); });
755 
756   if (!DeclMap()->WillParse(exe_ctx, m_materializer_up.get())) {
757     diagnostic_manager.PutString(
758         eDiagnosticSeverityError,
759         "current process state is unsuitable for expression parsing");
760 
761     return false;
762   }
763 
764   if (m_options.GetExecutionPolicy() == eExecutionPolicyTopLevel) {
765     DeclMap()->SetLookupsEnabled(true);
766   }
767 
768   Process *process = exe_ctx.GetProcessPtr();
769   ExecutionContextScope *exe_scope = process;
770 
771   if (!exe_scope)
772     exe_scope = exe_ctx.GetTargetPtr();
773 
774   ClangExpressionParser parser(exe_scope, *this, false);
775 
776   // We have to find the source code location where the user text is inside
777   // the transformed expression code. When creating the transformed text, we
778   // already stored the absolute position in the m_transformed_text string. The
779   // only thing left to do is to transform it into the line:column format that
780   // Clang expects.
781 
782   // The line and column of the user expression inside the transformed source
783   // code.
784   unsigned user_expr_line, user_expr_column;
785   if (m_user_expression_start_pos.hasValue())
786     AbsPosToLineColumnPos(*m_user_expression_start_pos, m_transformed_text,
787                           user_expr_line, user_expr_column);
788   else
789     return false;
790 
791   // The actual column where we have to complete is the start column of the
792   // user expression + the offset inside the user code that we were given.
793   const unsigned completion_column = user_expr_column + complete_pos;
794   parser.Complete(request, user_expr_line, completion_column, complete_pos);
795 
796   return true;
797 }
798 
799 bool ClangUserExpression::AddArguments(ExecutionContext &exe_ctx,
800                                        std::vector<lldb::addr_t> &args,
801                                        lldb::addr_t struct_address,
802                                        DiagnosticManager &diagnostic_manager) {
803   lldb::addr_t object_ptr = LLDB_INVALID_ADDRESS;
804   lldb::addr_t cmd_ptr = LLDB_INVALID_ADDRESS;
805 
806   if (m_needs_object_ptr) {
807     lldb::StackFrameSP frame_sp = exe_ctx.GetFrameSP();
808     if (!frame_sp)
809       return true;
810 
811     ConstString object_name;
812 
813     if (m_in_cplusplus_method) {
814       object_name.SetCString("this");
815     } else if (m_in_objectivec_method) {
816       object_name.SetCString("self");
817     } else {
818       diagnostic_manager.PutString(
819           eDiagnosticSeverityError,
820           "need object pointer but don't know the language");
821       return false;
822     }
823 
824     Status object_ptr_error;
825 
826     if (m_ctx_obj) {
827       AddressType address_type;
828       object_ptr = m_ctx_obj->GetAddressOf(false, &address_type);
829       if (object_ptr == LLDB_INVALID_ADDRESS ||
830           address_type != eAddressTypeLoad)
831         object_ptr_error.SetErrorString("Can't get context object's "
832                                         "debuggee address");
833     } else
834       object_ptr = GetObjectPointer(frame_sp, object_name, object_ptr_error);
835 
836     if (!object_ptr_error.Success()) {
837       exe_ctx.GetTargetRef().GetDebugger().GetAsyncOutputStream()->Printf(
838           "warning: `%s' is not accessible (substituting 0)\n",
839           object_name.AsCString());
840       object_ptr = 0;
841     }
842 
843     if (m_in_objectivec_method) {
844       ConstString cmd_name("_cmd");
845 
846       cmd_ptr = GetObjectPointer(frame_sp, cmd_name, object_ptr_error);
847 
848       if (!object_ptr_error.Success()) {
849         diagnostic_manager.Printf(
850             eDiagnosticSeverityWarning,
851             "couldn't get cmd pointer (substituting NULL): %s",
852             object_ptr_error.AsCString());
853         cmd_ptr = 0;
854       }
855     }
856 
857     args.push_back(object_ptr);
858 
859     if (m_in_objectivec_method)
860       args.push_back(cmd_ptr);
861 
862     args.push_back(struct_address);
863   } else {
864     args.push_back(struct_address);
865   }
866   return true;
867 }
868 
869 lldb::ExpressionVariableSP ClangUserExpression::GetResultAfterDematerialization(
870     ExecutionContextScope *exe_scope) {
871   return m_result_delegate.GetVariable();
872 }
873 
874 void ClangUserExpression::ClangUserExpressionHelper::ResetDeclMap(
875     ExecutionContext &exe_ctx,
876     Materializer::PersistentVariableDelegate &delegate,
877     bool keep_result_in_memory,
878     ValueObject *ctx_obj) {
879   m_expr_decl_map_up.reset(
880       new ClangExpressionDeclMap(keep_result_in_memory, &delegate, exe_ctx,
881                                  ctx_obj));
882 }
883 
884 clang::ASTConsumer *
885 ClangUserExpression::ClangUserExpressionHelper::ASTTransformer(
886     clang::ASTConsumer *passthrough) {
887   m_result_synthesizer_up.reset(
888       new ASTResultSynthesizer(passthrough, m_top_level, m_target));
889 
890   return m_result_synthesizer_up.get();
891 }
892 
893 void ClangUserExpression::ClangUserExpressionHelper::CommitPersistentDecls() {
894   if (m_result_synthesizer_up) {
895     m_result_synthesizer_up->CommitPersistentDecls();
896   }
897 }
898 
899 ConstString ClangUserExpression::ResultDelegate::GetName() {
900   auto prefix = m_persistent_state->GetPersistentVariablePrefix();
901   return m_persistent_state->GetNextPersistentVariableName(*m_target_sp,
902                                                            prefix);
903 }
904 
905 void ClangUserExpression::ResultDelegate::DidDematerialize(
906     lldb::ExpressionVariableSP &variable) {
907   m_variable = variable;
908 }
909 
910 void ClangUserExpression::ResultDelegate::RegisterPersistentState(
911     PersistentExpressionState *persistent_state) {
912   m_persistent_state = persistent_state;
913 }
914 
915 lldb::ExpressionVariableSP &ClangUserExpression::ResultDelegate::GetVariable() {
916   return m_variable;
917 }
918