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