15ffd83dbSDimitry Andric //===-- ScriptInterpreterPython.cpp ---------------------------------------===//
2*0b57cec5SDimitry Andric //
3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*0b57cec5SDimitry Andric //
7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
8*0b57cec5SDimitry Andric 
9480093f4SDimitry Andric #include "lldb/Host/Config.h"
10e8d8bef9SDimitry Andric #include "lldb/lldb-enumerations.h"
11*0b57cec5SDimitry Andric 
12480093f4SDimitry Andric #if LLDB_ENABLE_PYTHON
13*0b57cec5SDimitry Andric 
14*0b57cec5SDimitry Andric // LLDB Python header must be included first
15*0b57cec5SDimitry Andric #include "lldb-python.h"
16*0b57cec5SDimitry Andric 
17*0b57cec5SDimitry Andric #include "PythonDataObjects.h"
18c14a5a88SDimitry Andric #include "PythonReadline.h"
19fe6060f1SDimitry Andric #include "SWIGPythonBridge.h"
20*0b57cec5SDimitry Andric #include "ScriptInterpreterPythonImpl.h"
21fe6060f1SDimitry Andric #include "ScriptedProcessPythonInterface.h"
22fe6060f1SDimitry Andric 
23fe6060f1SDimitry Andric #include "lldb/API/SBError.h"
24*0b57cec5SDimitry Andric #include "lldb/API/SBFrame.h"
25*0b57cec5SDimitry Andric #include "lldb/API/SBValue.h"
26*0b57cec5SDimitry Andric #include "lldb/Breakpoint/StoppointCallbackContext.h"
27*0b57cec5SDimitry Andric #include "lldb/Breakpoint/WatchpointOptions.h"
28*0b57cec5SDimitry Andric #include "lldb/Core/Communication.h"
29*0b57cec5SDimitry Andric #include "lldb/Core/Debugger.h"
30*0b57cec5SDimitry Andric #include "lldb/Core/PluginManager.h"
31*0b57cec5SDimitry Andric #include "lldb/Core/ValueObject.h"
32*0b57cec5SDimitry Andric #include "lldb/DataFormatters/TypeSummary.h"
33*0b57cec5SDimitry Andric #include "lldb/Host/FileSystem.h"
34*0b57cec5SDimitry Andric #include "lldb/Host/HostInfo.h"
35*0b57cec5SDimitry Andric #include "lldb/Host/Pipe.h"
36*0b57cec5SDimitry Andric #include "lldb/Interpreter/CommandInterpreter.h"
37*0b57cec5SDimitry Andric #include "lldb/Interpreter/CommandReturnObject.h"
38*0b57cec5SDimitry Andric #include "lldb/Target/Thread.h"
39*0b57cec5SDimitry Andric #include "lldb/Target/ThreadPlan.h"
40e8d8bef9SDimitry Andric #include "lldb/Utility/ReproducerInstrumentation.h"
41*0b57cec5SDimitry Andric #include "lldb/Utility/Timer.h"
42*0b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h"
43*0b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h"
445ffd83dbSDimitry Andric #include "llvm/Support/Error.h"
45*0b57cec5SDimitry Andric #include "llvm/Support/FileSystem.h"
469dba64beSDimitry Andric #include "llvm/Support/FormatAdapters.h"
47*0b57cec5SDimitry Andric 
48fe6060f1SDimitry Andric #include <cstdio>
49fe6060f1SDimitry Andric #include <cstdlib>
50*0b57cec5SDimitry Andric #include <memory>
51*0b57cec5SDimitry Andric #include <mutex>
52*0b57cec5SDimitry Andric #include <string>
53*0b57cec5SDimitry Andric 
54*0b57cec5SDimitry Andric using namespace lldb;
55*0b57cec5SDimitry Andric using namespace lldb_private;
569dba64beSDimitry Andric using namespace lldb_private::python;
579dba64beSDimitry Andric using llvm::Expected;
58*0b57cec5SDimitry Andric 
595ffd83dbSDimitry Andric LLDB_PLUGIN_DEFINE(ScriptInterpreterPython)
605ffd83dbSDimitry Andric 
61*0b57cec5SDimitry Andric // Defined in the SWIG source file
62*0b57cec5SDimitry Andric #if PY_MAJOR_VERSION >= 3
63*0b57cec5SDimitry Andric extern "C" PyObject *PyInit__lldb(void);
64*0b57cec5SDimitry Andric 
65*0b57cec5SDimitry Andric #define LLDBSwigPyInit PyInit__lldb
66*0b57cec5SDimitry Andric 
67*0b57cec5SDimitry Andric #else
68*0b57cec5SDimitry Andric extern "C" void init_lldb(void);
69*0b57cec5SDimitry Andric 
70*0b57cec5SDimitry Andric #define LLDBSwigPyInit init_lldb
71*0b57cec5SDimitry Andric #endif
72*0b57cec5SDimitry Andric 
73*0b57cec5SDimitry Andric // These prototypes are the Pythonic implementations of the required callbacks.
74*0b57cec5SDimitry Andric // Although these are scripting-language specific, their definition depends on
75*0b57cec5SDimitry Andric // the public API.
76480093f4SDimitry Andric 
77480093f4SDimitry Andric #pragma clang diagnostic push
78480093f4SDimitry Andric #pragma clang diagnostic ignored "-Wreturn-type-c-linkage"
79480093f4SDimitry Andric 
80480093f4SDimitry Andric // Disable warning C4190: 'LLDBSwigPythonBreakpointCallbackFunction' has
81480093f4SDimitry Andric // C-linkage specified, but returns UDT 'llvm::Expected<bool>' which is
82480093f4SDimitry Andric // incompatible with C
83480093f4SDimitry Andric #if _MSC_VER
84480093f4SDimitry Andric #pragma warning (push)
85480093f4SDimitry Andric #pragma warning (disable : 4190)
86480093f4SDimitry Andric #endif
87480093f4SDimitry Andric 
88480093f4SDimitry Andric extern "C" llvm::Expected<bool> LLDBSwigPythonBreakpointCallbackFunction(
89*0b57cec5SDimitry Andric     const char *python_function_name, const char *session_dictionary_name,
90*0b57cec5SDimitry Andric     const lldb::StackFrameSP &sb_frame,
91480093f4SDimitry Andric     const lldb::BreakpointLocationSP &sb_bp_loc, StructuredDataImpl *args_impl);
92480093f4SDimitry Andric 
93480093f4SDimitry Andric #if _MSC_VER
94480093f4SDimitry Andric #pragma warning (pop)
95480093f4SDimitry Andric #endif
96480093f4SDimitry Andric 
97480093f4SDimitry Andric #pragma clang diagnostic pop
98*0b57cec5SDimitry Andric 
99*0b57cec5SDimitry Andric extern "C" bool LLDBSwigPythonWatchpointCallbackFunction(
100*0b57cec5SDimitry Andric     const char *python_function_name, const char *session_dictionary_name,
101*0b57cec5SDimitry Andric     const lldb::StackFrameSP &sb_frame, const lldb::WatchpointSP &sb_wp);
102*0b57cec5SDimitry Andric 
103*0b57cec5SDimitry Andric extern "C" bool LLDBSwigPythonCallTypeScript(
104*0b57cec5SDimitry Andric     const char *python_function_name, void *session_dictionary,
105*0b57cec5SDimitry Andric     const lldb::ValueObjectSP &valobj_sp, void **pyfunct_wrapper,
106*0b57cec5SDimitry Andric     const lldb::TypeSummaryOptionsSP &options_sp, std::string &retval);
107*0b57cec5SDimitry Andric 
108*0b57cec5SDimitry Andric extern "C" void *
109*0b57cec5SDimitry Andric LLDBSwigPythonCreateSyntheticProvider(const char *python_class_name,
110*0b57cec5SDimitry Andric                                       const char *session_dictionary_name,
111*0b57cec5SDimitry Andric                                       const lldb::ValueObjectSP &valobj_sp);
112*0b57cec5SDimitry Andric 
113*0b57cec5SDimitry Andric extern "C" void *
114*0b57cec5SDimitry Andric LLDBSwigPythonCreateCommandObject(const char *python_class_name,
115*0b57cec5SDimitry Andric                                   const char *session_dictionary_name,
116*0b57cec5SDimitry Andric                                   const lldb::DebuggerSP debugger_sp);
117*0b57cec5SDimitry Andric 
118*0b57cec5SDimitry Andric extern "C" void *LLDBSwigPythonCreateScriptedThreadPlan(
119*0b57cec5SDimitry Andric     const char *python_class_name, const char *session_dictionary_name,
1209dba64beSDimitry Andric     StructuredDataImpl *args_data,
1219dba64beSDimitry Andric     std::string &error_string,
122*0b57cec5SDimitry Andric     const lldb::ThreadPlanSP &thread_plan_sp);
123*0b57cec5SDimitry Andric 
124*0b57cec5SDimitry Andric extern "C" bool LLDBSWIGPythonCallThreadPlan(void *implementor,
125*0b57cec5SDimitry Andric                                              const char *method_name,
126*0b57cec5SDimitry Andric                                              Event *event_sp, bool &got_error);
127*0b57cec5SDimitry Andric 
128*0b57cec5SDimitry Andric extern "C" void *LLDBSwigPythonCreateScriptedBreakpointResolver(
129*0b57cec5SDimitry Andric     const char *python_class_name, const char *session_dictionary_name,
130*0b57cec5SDimitry Andric     lldb_private::StructuredDataImpl *args, lldb::BreakpointSP &bkpt_sp);
131*0b57cec5SDimitry Andric 
132*0b57cec5SDimitry Andric extern "C" unsigned int
133*0b57cec5SDimitry Andric LLDBSwigPythonCallBreakpointResolver(void *implementor, const char *method_name,
134*0b57cec5SDimitry Andric                                      lldb_private::SymbolContext *sym_ctx);
135*0b57cec5SDimitry Andric 
136e8d8bef9SDimitry Andric extern "C" void *LLDBSwigPythonCreateScriptedStopHook(
137e8d8bef9SDimitry Andric     TargetSP target_sp, const char *python_class_name,
138e8d8bef9SDimitry Andric     const char *session_dictionary_name, lldb_private::StructuredDataImpl *args,
139e8d8bef9SDimitry Andric     lldb_private::Status &error);
140e8d8bef9SDimitry Andric 
141e8d8bef9SDimitry Andric extern "C" bool
142e8d8bef9SDimitry Andric LLDBSwigPythonStopHookCallHandleStop(void *implementor,
143e8d8bef9SDimitry Andric                                      lldb::ExecutionContextRefSP exc_ctx,
144e8d8bef9SDimitry Andric                                      lldb::StreamSP stream);
145e8d8bef9SDimitry Andric 
146*0b57cec5SDimitry Andric extern "C" size_t LLDBSwigPython_CalculateNumChildren(void *implementor,
147*0b57cec5SDimitry Andric                                                       uint32_t max);
148*0b57cec5SDimitry Andric 
149*0b57cec5SDimitry Andric extern "C" void *LLDBSwigPython_GetChildAtIndex(void *implementor,
150*0b57cec5SDimitry Andric                                                 uint32_t idx);
151*0b57cec5SDimitry Andric 
152*0b57cec5SDimitry Andric extern "C" int LLDBSwigPython_GetIndexOfChildWithName(void *implementor,
153*0b57cec5SDimitry Andric                                                       const char *child_name);
154*0b57cec5SDimitry Andric 
155*0b57cec5SDimitry Andric extern lldb::ValueObjectSP
156*0b57cec5SDimitry Andric LLDBSWIGPython_GetValueObjectSPFromSBValue(void *data);
157*0b57cec5SDimitry Andric 
158*0b57cec5SDimitry Andric extern "C" bool LLDBSwigPython_UpdateSynthProviderInstance(void *implementor);
159*0b57cec5SDimitry Andric 
160*0b57cec5SDimitry Andric extern "C" bool
161*0b57cec5SDimitry Andric LLDBSwigPython_MightHaveChildrenSynthProviderInstance(void *implementor);
162*0b57cec5SDimitry Andric 
163*0b57cec5SDimitry Andric extern "C" void *
164*0b57cec5SDimitry Andric LLDBSwigPython_GetValueSynthProviderInstance(void *implementor);
165*0b57cec5SDimitry Andric 
166*0b57cec5SDimitry Andric extern "C" bool
167*0b57cec5SDimitry Andric LLDBSwigPythonCallCommand(const char *python_function_name,
168*0b57cec5SDimitry Andric                           const char *session_dictionary_name,
169*0b57cec5SDimitry Andric                           lldb::DebuggerSP &debugger, const char *args,
170*0b57cec5SDimitry Andric                           lldb_private::CommandReturnObject &cmd_retobj,
171*0b57cec5SDimitry Andric                           lldb::ExecutionContextRefSP exe_ctx_ref_sp);
172*0b57cec5SDimitry Andric 
173*0b57cec5SDimitry Andric extern "C" bool
174*0b57cec5SDimitry Andric LLDBSwigPythonCallCommandObject(void *implementor, lldb::DebuggerSP &debugger,
175*0b57cec5SDimitry Andric                                 const char *args,
176*0b57cec5SDimitry Andric                                 lldb_private::CommandReturnObject &cmd_retobj,
177*0b57cec5SDimitry Andric                                 lldb::ExecutionContextRefSP exe_ctx_ref_sp);
178*0b57cec5SDimitry Andric 
179*0b57cec5SDimitry Andric extern "C" bool
180*0b57cec5SDimitry Andric LLDBSwigPythonCallModuleInit(const char *python_module_name,
181*0b57cec5SDimitry Andric                              const char *session_dictionary_name,
182*0b57cec5SDimitry Andric                              lldb::DebuggerSP &debugger);
183*0b57cec5SDimitry Andric 
184*0b57cec5SDimitry Andric extern "C" void *
185*0b57cec5SDimitry Andric LLDBSWIGPythonCreateOSPlugin(const char *python_class_name,
186*0b57cec5SDimitry Andric                              const char *session_dictionary_name,
187*0b57cec5SDimitry Andric                              const lldb::ProcessSP &process_sp);
188*0b57cec5SDimitry Andric 
189*0b57cec5SDimitry Andric extern "C" void *
190*0b57cec5SDimitry Andric LLDBSWIGPython_CreateFrameRecognizer(const char *python_class_name,
191*0b57cec5SDimitry Andric                                      const char *session_dictionary_name);
192*0b57cec5SDimitry Andric 
193*0b57cec5SDimitry Andric extern "C" void *
194*0b57cec5SDimitry Andric LLDBSwigPython_GetRecognizedArguments(void *implementor,
195*0b57cec5SDimitry Andric                                       const lldb::StackFrameSP &frame_sp);
196*0b57cec5SDimitry Andric 
197*0b57cec5SDimitry Andric extern "C" bool LLDBSWIGPythonRunScriptKeywordProcess(
198*0b57cec5SDimitry Andric     const char *python_function_name, const char *session_dictionary_name,
199*0b57cec5SDimitry Andric     lldb::ProcessSP &process, std::string &output);
200*0b57cec5SDimitry Andric 
201*0b57cec5SDimitry Andric extern "C" bool LLDBSWIGPythonRunScriptKeywordThread(
202*0b57cec5SDimitry Andric     const char *python_function_name, const char *session_dictionary_name,
203*0b57cec5SDimitry Andric     lldb::ThreadSP &thread, std::string &output);
204*0b57cec5SDimitry Andric 
205*0b57cec5SDimitry Andric extern "C" bool LLDBSWIGPythonRunScriptKeywordTarget(
206*0b57cec5SDimitry Andric     const char *python_function_name, const char *session_dictionary_name,
207*0b57cec5SDimitry Andric     lldb::TargetSP &target, std::string &output);
208*0b57cec5SDimitry Andric 
209*0b57cec5SDimitry Andric extern "C" bool LLDBSWIGPythonRunScriptKeywordFrame(
210*0b57cec5SDimitry Andric     const char *python_function_name, const char *session_dictionary_name,
211*0b57cec5SDimitry Andric     lldb::StackFrameSP &frame, std::string &output);
212*0b57cec5SDimitry Andric 
213*0b57cec5SDimitry Andric extern "C" bool LLDBSWIGPythonRunScriptKeywordValue(
214*0b57cec5SDimitry Andric     const char *python_function_name, const char *session_dictionary_name,
215*0b57cec5SDimitry Andric     lldb::ValueObjectSP &value, std::string &output);
216*0b57cec5SDimitry Andric 
217*0b57cec5SDimitry Andric extern "C" void *
218*0b57cec5SDimitry Andric LLDBSWIGPython_GetDynamicSetting(void *module, const char *setting,
219*0b57cec5SDimitry Andric                                  const lldb::TargetSP &target_sp);
220*0b57cec5SDimitry Andric 
221e8d8bef9SDimitry Andric static ScriptInterpreterPythonImpl *GetPythonInterpreter(Debugger &debugger) {
222e8d8bef9SDimitry Andric   ScriptInterpreter *script_interpreter =
223e8d8bef9SDimitry Andric       debugger.GetScriptInterpreter(true, lldb::eScriptLanguagePython);
224e8d8bef9SDimitry Andric   return static_cast<ScriptInterpreterPythonImpl *>(script_interpreter);
225e8d8bef9SDimitry Andric }
226e8d8bef9SDimitry Andric 
227*0b57cec5SDimitry Andric static bool g_initialized = false;
228*0b57cec5SDimitry Andric 
229*0b57cec5SDimitry Andric namespace {
230*0b57cec5SDimitry Andric 
231*0b57cec5SDimitry Andric // Initializing Python is not a straightforward process.  We cannot control
232*0b57cec5SDimitry Andric // what external code may have done before getting to this point in LLDB,
233*0b57cec5SDimitry Andric // including potentially having already initialized Python, so we need to do a
234*0b57cec5SDimitry Andric // lot of work to ensure that the existing state of the system is maintained
235*0b57cec5SDimitry Andric // across our initialization.  We do this by using an RAII pattern where we
236*0b57cec5SDimitry Andric // save off initial state at the beginning, and restore it at the end
237*0b57cec5SDimitry Andric struct InitializePythonRAII {
238*0b57cec5SDimitry Andric public:
239fe6060f1SDimitry Andric   InitializePythonRAII() {
240*0b57cec5SDimitry Andric     InitializePythonHome();
241*0b57cec5SDimitry Andric 
242c14a5a88SDimitry Andric #ifdef LLDB_USE_LIBEDIT_READLINE_COMPAT_MODULE
243c14a5a88SDimitry Andric     // Python's readline is incompatible with libedit being linked into lldb.
244c14a5a88SDimitry Andric     // Provide a patched version local to the embedded interpreter.
245c14a5a88SDimitry Andric     bool ReadlinePatched = false;
246c14a5a88SDimitry Andric     for (auto *p = PyImport_Inittab; p->name != NULL; p++) {
247c14a5a88SDimitry Andric       if (strcmp(p->name, "readline") == 0) {
248c14a5a88SDimitry Andric         p->initfunc = initlldb_readline;
249c14a5a88SDimitry Andric         break;
250c14a5a88SDimitry Andric       }
251c14a5a88SDimitry Andric     }
252c14a5a88SDimitry Andric     if (!ReadlinePatched) {
253c14a5a88SDimitry Andric       PyImport_AppendInittab("readline", initlldb_readline);
254c14a5a88SDimitry Andric       ReadlinePatched = true;
255c14a5a88SDimitry Andric     }
256c14a5a88SDimitry Andric #endif
257c14a5a88SDimitry Andric 
258*0b57cec5SDimitry Andric     // Register _lldb as a built-in module.
259*0b57cec5SDimitry Andric     PyImport_AppendInittab("_lldb", LLDBSwigPyInit);
260*0b57cec5SDimitry Andric 
261*0b57cec5SDimitry Andric // Python < 3.2 and Python >= 3.2 reversed the ordering requirements for
262*0b57cec5SDimitry Andric // calling `Py_Initialize` and `PyEval_InitThreads`.  < 3.2 requires that you
263*0b57cec5SDimitry Andric // call `PyEval_InitThreads` first, and >= 3.2 requires that you call it last.
264*0b57cec5SDimitry Andric #if (PY_MAJOR_VERSION == 3 && PY_MINOR_VERSION >= 2) || (PY_MAJOR_VERSION > 3)
265*0b57cec5SDimitry Andric     Py_InitializeEx(0);
266*0b57cec5SDimitry Andric     InitializeThreadsPrivate();
267*0b57cec5SDimitry Andric #else
268*0b57cec5SDimitry Andric     InitializeThreadsPrivate();
269*0b57cec5SDimitry Andric     Py_InitializeEx(0);
270*0b57cec5SDimitry Andric #endif
271*0b57cec5SDimitry Andric   }
272*0b57cec5SDimitry Andric 
273*0b57cec5SDimitry Andric   ~InitializePythonRAII() {
274*0b57cec5SDimitry Andric     if (m_was_already_initialized) {
275*0b57cec5SDimitry Andric       Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT));
276*0b57cec5SDimitry Andric       LLDB_LOGV(log, "Releasing PyGILState. Returning to state = {0}locked",
277*0b57cec5SDimitry Andric                 m_gil_state == PyGILState_UNLOCKED ? "un" : "");
278*0b57cec5SDimitry Andric       PyGILState_Release(m_gil_state);
279*0b57cec5SDimitry Andric     } else {
280*0b57cec5SDimitry Andric       // We initialized the threads in this function, just unlock the GIL.
281*0b57cec5SDimitry Andric       PyEval_SaveThread();
282*0b57cec5SDimitry Andric     }
283*0b57cec5SDimitry Andric   }
284*0b57cec5SDimitry Andric 
285*0b57cec5SDimitry Andric private:
286*0b57cec5SDimitry Andric   void InitializePythonHome() {
2875ffd83dbSDimitry Andric #if LLDB_EMBED_PYTHON_HOME
2885ffd83dbSDimitry Andric #if PY_MAJOR_VERSION >= 3
2895ffd83dbSDimitry Andric     typedef wchar_t* str_type;
2905ffd83dbSDimitry Andric #else
2915ffd83dbSDimitry Andric     typedef char* str_type;
2925ffd83dbSDimitry Andric #endif
2935ffd83dbSDimitry Andric     static str_type g_python_home = []() -> str_type {
2945ffd83dbSDimitry Andric       const char *lldb_python_home = LLDB_PYTHON_HOME;
2955ffd83dbSDimitry Andric       const char *absolute_python_home = nullptr;
2965ffd83dbSDimitry Andric       llvm::SmallString<64> path;
2975ffd83dbSDimitry Andric       if (llvm::sys::path::is_absolute(lldb_python_home)) {
2985ffd83dbSDimitry Andric         absolute_python_home = lldb_python_home;
2995ffd83dbSDimitry Andric       } else {
3005ffd83dbSDimitry Andric         FileSpec spec = HostInfo::GetShlibDir();
3015ffd83dbSDimitry Andric         if (!spec)
3025ffd83dbSDimitry Andric           return nullptr;
3035ffd83dbSDimitry Andric         spec.GetPath(path);
3045ffd83dbSDimitry Andric         llvm::sys::path::append(path, lldb_python_home);
3055ffd83dbSDimitry Andric         absolute_python_home = path.c_str();
3065ffd83dbSDimitry Andric       }
307*0b57cec5SDimitry Andric #if PY_MAJOR_VERSION >= 3
308*0b57cec5SDimitry Andric       size_t size = 0;
3095ffd83dbSDimitry Andric       return Py_DecodeLocale(absolute_python_home, &size);
310*0b57cec5SDimitry Andric #else
3115ffd83dbSDimitry Andric       return strdup(absolute_python_home);
312*0b57cec5SDimitry Andric #endif
3135ffd83dbSDimitry Andric     }();
3145ffd83dbSDimitry Andric     if (g_python_home != nullptr) {
315*0b57cec5SDimitry Andric       Py_SetPythonHome(g_python_home);
3165ffd83dbSDimitry Andric     }
317*0b57cec5SDimitry Andric #else
318*0b57cec5SDimitry Andric #if defined(__APPLE__) && PY_MAJOR_VERSION == 2 && PY_MINOR_VERSION == 7
319*0b57cec5SDimitry Andric     // For Darwin, the only Python version supported is the one shipped in the
320*0b57cec5SDimitry Andric     // OS OS and linked with lldb. Other installation of Python may have higher
321*0b57cec5SDimitry Andric     // priorities in the path, overriding PYTHONHOME and causing
322*0b57cec5SDimitry Andric     // problems/incompatibilities. In order to avoid confusion, always hardcode
323*0b57cec5SDimitry Andric     // the PythonHome to be right, as it's not going to change.
324*0b57cec5SDimitry Andric     static char path[] =
325*0b57cec5SDimitry Andric         "/System/Library/Frameworks/Python.framework/Versions/2.7";
326*0b57cec5SDimitry Andric     Py_SetPythonHome(path);
327*0b57cec5SDimitry Andric #endif
328*0b57cec5SDimitry Andric #endif
329*0b57cec5SDimitry Andric   }
330*0b57cec5SDimitry Andric 
331*0b57cec5SDimitry Andric   void InitializeThreadsPrivate() {
332*0b57cec5SDimitry Andric // Since Python 3.7 `Py_Initialize` calls `PyEval_InitThreads` inside itself,
333*0b57cec5SDimitry Andric // so there is no way to determine whether the embedded interpreter
334*0b57cec5SDimitry Andric // was already initialized by some external code. `PyEval_ThreadsInitialized`
335*0b57cec5SDimitry Andric // would always return `true` and `PyGILState_Ensure/Release` flow would be
336*0b57cec5SDimitry Andric // executed instead of unlocking GIL with `PyEval_SaveThread`. When
337*0b57cec5SDimitry Andric // an another thread calls `PyGILState_Ensure` it would get stuck in deadlock.
338*0b57cec5SDimitry Andric #if (PY_MAJOR_VERSION == 3 && PY_MINOR_VERSION >= 7) || (PY_MAJOR_VERSION > 3)
339*0b57cec5SDimitry Andric     // The only case we should go further and acquire the GIL: it is unlocked.
340*0b57cec5SDimitry Andric     if (PyGILState_Check())
341*0b57cec5SDimitry Andric       return;
342*0b57cec5SDimitry Andric #endif
343*0b57cec5SDimitry Andric 
344*0b57cec5SDimitry Andric     if (PyEval_ThreadsInitialized()) {
345*0b57cec5SDimitry Andric       Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT));
346*0b57cec5SDimitry Andric 
347*0b57cec5SDimitry Andric       m_was_already_initialized = true;
348*0b57cec5SDimitry Andric       m_gil_state = PyGILState_Ensure();
349*0b57cec5SDimitry Andric       LLDB_LOGV(log, "Ensured PyGILState. Previous state = {0}locked\n",
350*0b57cec5SDimitry Andric                 m_gil_state == PyGILState_UNLOCKED ? "un" : "");
351*0b57cec5SDimitry Andric       return;
352*0b57cec5SDimitry Andric     }
353*0b57cec5SDimitry Andric 
354*0b57cec5SDimitry Andric     // InitThreads acquires the GIL if it hasn't been called before.
355*0b57cec5SDimitry Andric     PyEval_InitThreads();
356*0b57cec5SDimitry Andric   }
357*0b57cec5SDimitry Andric 
358fe6060f1SDimitry Andric   PyGILState_STATE m_gil_state = PyGILState_UNLOCKED;
359fe6060f1SDimitry Andric   bool m_was_already_initialized = false;
360*0b57cec5SDimitry Andric };
361*0b57cec5SDimitry Andric } // namespace
362*0b57cec5SDimitry Andric 
363*0b57cec5SDimitry Andric void ScriptInterpreterPython::ComputePythonDirForApple(
364*0b57cec5SDimitry Andric     llvm::SmallVectorImpl<char> &path) {
365*0b57cec5SDimitry Andric   auto style = llvm::sys::path::Style::posix;
366*0b57cec5SDimitry Andric 
367*0b57cec5SDimitry Andric   llvm::StringRef path_ref(path.begin(), path.size());
368*0b57cec5SDimitry Andric   auto rbegin = llvm::sys::path::rbegin(path_ref, style);
369*0b57cec5SDimitry Andric   auto rend = llvm::sys::path::rend(path_ref);
370*0b57cec5SDimitry Andric   auto framework = std::find(rbegin, rend, "LLDB.framework");
371*0b57cec5SDimitry Andric   if (framework == rend) {
3729dba64beSDimitry Andric     ComputePythonDir(path);
373*0b57cec5SDimitry Andric     return;
374*0b57cec5SDimitry Andric   }
375*0b57cec5SDimitry Andric   path.resize(framework - rend);
376*0b57cec5SDimitry Andric   llvm::sys::path::append(path, style, "LLDB.framework", "Resources", "Python");
377*0b57cec5SDimitry Andric }
378*0b57cec5SDimitry Andric 
3799dba64beSDimitry Andric void ScriptInterpreterPython::ComputePythonDir(
380*0b57cec5SDimitry Andric     llvm::SmallVectorImpl<char> &path) {
381*0b57cec5SDimitry Andric   // Build the path by backing out of the lib dir, then building with whatever
382*0b57cec5SDimitry Andric   // the real python interpreter uses.  (e.g. lib for most, lib64 on RHEL
3839dba64beSDimitry Andric   // x86_64, or bin on Windows).
3849dba64beSDimitry Andric   llvm::sys::path::remove_filename(path);
3859dba64beSDimitry Andric   llvm::sys::path::append(path, LLDB_PYTHON_RELATIVE_LIBDIR);
386*0b57cec5SDimitry Andric 
3879dba64beSDimitry Andric #if defined(_WIN32)
388*0b57cec5SDimitry Andric   // This will be injected directly through FileSpec.GetDirectory().SetString(),
389*0b57cec5SDimitry Andric   // so we need to normalize manually.
390*0b57cec5SDimitry Andric   std::replace(path.begin(), path.end(), '\\', '/');
3919dba64beSDimitry Andric #endif
392*0b57cec5SDimitry Andric }
393*0b57cec5SDimitry Andric 
394*0b57cec5SDimitry Andric FileSpec ScriptInterpreterPython::GetPythonDir() {
395*0b57cec5SDimitry Andric   static FileSpec g_spec = []() {
396*0b57cec5SDimitry Andric     FileSpec spec = HostInfo::GetShlibDir();
397*0b57cec5SDimitry Andric     if (!spec)
398*0b57cec5SDimitry Andric       return FileSpec();
399*0b57cec5SDimitry Andric     llvm::SmallString<64> path;
400*0b57cec5SDimitry Andric     spec.GetPath(path);
401*0b57cec5SDimitry Andric 
402*0b57cec5SDimitry Andric #if defined(__APPLE__)
403*0b57cec5SDimitry Andric     ComputePythonDirForApple(path);
404*0b57cec5SDimitry Andric #else
4059dba64beSDimitry Andric     ComputePythonDir(path);
406*0b57cec5SDimitry Andric #endif
407*0b57cec5SDimitry Andric     spec.GetDirectory().SetString(path);
408*0b57cec5SDimitry Andric     return spec;
409*0b57cec5SDimitry Andric   }();
410*0b57cec5SDimitry Andric   return g_spec;
411*0b57cec5SDimitry Andric }
412*0b57cec5SDimitry Andric 
413349cc55cSDimitry Andric static const char GetInterpreterInfoScript[] = R"(
414349cc55cSDimitry Andric import os
415349cc55cSDimitry Andric import sys
416349cc55cSDimitry Andric 
417349cc55cSDimitry Andric def main(lldb_python_dir, python_exe_relative_path):
418349cc55cSDimitry Andric   info = {
419349cc55cSDimitry Andric     "lldb-pythonpath": lldb_python_dir,
420349cc55cSDimitry Andric     "language": "python",
421349cc55cSDimitry Andric     "prefix": sys.prefix,
422349cc55cSDimitry Andric     "executable": os.path.join(sys.prefix, python_exe_relative_path)
423349cc55cSDimitry Andric   }
424349cc55cSDimitry Andric   return info
425349cc55cSDimitry Andric )";
426349cc55cSDimitry Andric 
427349cc55cSDimitry Andric static const char python_exe_relative_path[] = LLDB_PYTHON_EXE_RELATIVE_PATH;
428349cc55cSDimitry Andric 
429349cc55cSDimitry Andric StructuredData::DictionarySP ScriptInterpreterPython::GetInterpreterInfo() {
430349cc55cSDimitry Andric   GIL gil;
431349cc55cSDimitry Andric   FileSpec python_dir_spec = GetPythonDir();
432349cc55cSDimitry Andric   if (!python_dir_spec)
433349cc55cSDimitry Andric     return nullptr;
434349cc55cSDimitry Andric   PythonScript get_info(GetInterpreterInfoScript);
435349cc55cSDimitry Andric   auto info_json = unwrapIgnoringErrors(
436349cc55cSDimitry Andric       As<PythonDictionary>(get_info(PythonString(python_dir_spec.GetPath()),
437349cc55cSDimitry Andric                                     PythonString(python_exe_relative_path))));
438349cc55cSDimitry Andric   if (!info_json)
439349cc55cSDimitry Andric     return nullptr;
440349cc55cSDimitry Andric   return info_json.CreateStructuredDictionary();
441349cc55cSDimitry Andric }
442349cc55cSDimitry Andric 
443fe6060f1SDimitry Andric void ScriptInterpreterPython::SharedLibraryDirectoryHelper(
444fe6060f1SDimitry Andric     FileSpec &this_file) {
445fe6060f1SDimitry Andric   // When we're loaded from python, this_file will point to the file inside the
446fe6060f1SDimitry Andric   // python package directory. Replace it with the one in the lib directory.
447fe6060f1SDimitry Andric #ifdef _WIN32
448fe6060f1SDimitry Andric   // On windows, we need to manually back out of the python tree, and go into
449fe6060f1SDimitry Andric   // the bin directory. This is pretty much the inverse of what ComputePythonDir
450fe6060f1SDimitry Andric   // does.
451fe6060f1SDimitry Andric   if (this_file.GetFileNameExtension() == ConstString(".pyd")) {
452fe6060f1SDimitry Andric     this_file.RemoveLastPathComponent(); // _lldb.pyd or _lldb_d.pyd
453fe6060f1SDimitry Andric     this_file.RemoveLastPathComponent(); // lldb
454fe6060f1SDimitry Andric     llvm::StringRef libdir = LLDB_PYTHON_RELATIVE_LIBDIR;
455fe6060f1SDimitry Andric     for (auto it = llvm::sys::path::begin(libdir),
456fe6060f1SDimitry Andric               end = llvm::sys::path::end(libdir);
457fe6060f1SDimitry Andric          it != end; ++it)
458fe6060f1SDimitry Andric       this_file.RemoveLastPathComponent();
459fe6060f1SDimitry Andric     this_file.AppendPathComponent("bin");
460fe6060f1SDimitry Andric     this_file.AppendPathComponent("liblldb.dll");
461fe6060f1SDimitry Andric   }
462fe6060f1SDimitry Andric #else
463fe6060f1SDimitry Andric   // The python file is a symlink, so we can find the real library by resolving
464fe6060f1SDimitry Andric   // it. We can do this unconditionally.
465fe6060f1SDimitry Andric   FileSystem::Instance().ResolveSymbolicLink(this_file, this_file);
466fe6060f1SDimitry Andric #endif
467fe6060f1SDimitry Andric }
468fe6060f1SDimitry Andric 
469349cc55cSDimitry Andric llvm::StringRef ScriptInterpreterPython::GetPluginDescriptionStatic() {
470*0b57cec5SDimitry Andric   return "Embedded Python interpreter";
471*0b57cec5SDimitry Andric }
472*0b57cec5SDimitry Andric 
473*0b57cec5SDimitry Andric void ScriptInterpreterPython::Initialize() {
474*0b57cec5SDimitry Andric   static llvm::once_flag g_once_flag;
475*0b57cec5SDimitry Andric 
476*0b57cec5SDimitry Andric   llvm::call_once(g_once_flag, []() {
477*0b57cec5SDimitry Andric     PluginManager::RegisterPlugin(GetPluginNameStatic(),
478*0b57cec5SDimitry Andric                                   GetPluginDescriptionStatic(),
479*0b57cec5SDimitry Andric                                   lldb::eScriptLanguagePython,
480*0b57cec5SDimitry Andric                                   ScriptInterpreterPythonImpl::CreateInstance);
481*0b57cec5SDimitry Andric   });
482*0b57cec5SDimitry Andric }
483*0b57cec5SDimitry Andric 
484*0b57cec5SDimitry Andric void ScriptInterpreterPython::Terminate() {}
485*0b57cec5SDimitry Andric 
486*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::Locker::Locker(
487*0b57cec5SDimitry Andric     ScriptInterpreterPythonImpl *py_interpreter, uint16_t on_entry,
4889dba64beSDimitry Andric     uint16_t on_leave, FileSP in, FileSP out, FileSP err)
489*0b57cec5SDimitry Andric     : ScriptInterpreterLocker(),
490*0b57cec5SDimitry Andric       m_teardown_session((on_leave & TearDownSession) == TearDownSession),
491*0b57cec5SDimitry Andric       m_python_interpreter(py_interpreter) {
492e8d8bef9SDimitry Andric   repro::Recorder::PrivateThread();
493*0b57cec5SDimitry Andric   DoAcquireLock();
494*0b57cec5SDimitry Andric   if ((on_entry & InitSession) == InitSession) {
495*0b57cec5SDimitry Andric     if (!DoInitSession(on_entry, in, out, err)) {
496*0b57cec5SDimitry Andric       // Don't teardown the session if we didn't init it.
497*0b57cec5SDimitry Andric       m_teardown_session = false;
498*0b57cec5SDimitry Andric     }
499*0b57cec5SDimitry Andric   }
500*0b57cec5SDimitry Andric }
501*0b57cec5SDimitry Andric 
502*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::Locker::DoAcquireLock() {
503*0b57cec5SDimitry Andric   Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT));
504*0b57cec5SDimitry Andric   m_GILState = PyGILState_Ensure();
505*0b57cec5SDimitry Andric   LLDB_LOGV(log, "Ensured PyGILState. Previous state = {0}locked",
506*0b57cec5SDimitry Andric             m_GILState == PyGILState_UNLOCKED ? "un" : "");
507*0b57cec5SDimitry Andric 
508*0b57cec5SDimitry Andric   // we need to save the thread state when we first start the command because
509*0b57cec5SDimitry Andric   // we might decide to interrupt it while some action is taking place outside
510*0b57cec5SDimitry Andric   // of Python (e.g. printing to screen, waiting for the network, ...) in that
511*0b57cec5SDimitry Andric   // case, _PyThreadState_Current will be NULL - and we would be unable to set
512*0b57cec5SDimitry Andric   // the asynchronous exception - not a desirable situation
513*0b57cec5SDimitry Andric   m_python_interpreter->SetThreadState(PyThreadState_Get());
514*0b57cec5SDimitry Andric   m_python_interpreter->IncrementLockCount();
515*0b57cec5SDimitry Andric   return true;
516*0b57cec5SDimitry Andric }
517*0b57cec5SDimitry Andric 
518*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::Locker::DoInitSession(uint16_t on_entry_flags,
5199dba64beSDimitry Andric                                                         FileSP in, FileSP out,
5209dba64beSDimitry Andric                                                         FileSP err) {
521*0b57cec5SDimitry Andric   if (!m_python_interpreter)
522*0b57cec5SDimitry Andric     return false;
523*0b57cec5SDimitry Andric   return m_python_interpreter->EnterSession(on_entry_flags, in, out, err);
524*0b57cec5SDimitry Andric }
525*0b57cec5SDimitry Andric 
526*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::Locker::DoFreeLock() {
527*0b57cec5SDimitry Andric   Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT));
528*0b57cec5SDimitry Andric   LLDB_LOGV(log, "Releasing PyGILState. Returning to state = {0}locked",
529*0b57cec5SDimitry Andric             m_GILState == PyGILState_UNLOCKED ? "un" : "");
530*0b57cec5SDimitry Andric   PyGILState_Release(m_GILState);
531*0b57cec5SDimitry Andric   m_python_interpreter->DecrementLockCount();
532*0b57cec5SDimitry Andric   return true;
533*0b57cec5SDimitry Andric }
534*0b57cec5SDimitry Andric 
535*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::Locker::DoTearDownSession() {
536*0b57cec5SDimitry Andric   if (!m_python_interpreter)
537*0b57cec5SDimitry Andric     return false;
538*0b57cec5SDimitry Andric   m_python_interpreter->LeaveSession();
539*0b57cec5SDimitry Andric   return true;
540*0b57cec5SDimitry Andric }
541*0b57cec5SDimitry Andric 
542*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::Locker::~Locker() {
543*0b57cec5SDimitry Andric   if (m_teardown_session)
544*0b57cec5SDimitry Andric     DoTearDownSession();
545*0b57cec5SDimitry Andric   DoFreeLock();
546*0b57cec5SDimitry Andric }
547*0b57cec5SDimitry Andric 
548*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::ScriptInterpreterPythonImpl(Debugger &debugger)
549*0b57cec5SDimitry Andric     : ScriptInterpreterPython(debugger), m_saved_stdin(), m_saved_stdout(),
550*0b57cec5SDimitry Andric       m_saved_stderr(), m_main_module(),
551*0b57cec5SDimitry Andric       m_session_dict(PyInitialValue::Invalid),
552*0b57cec5SDimitry Andric       m_sys_module_dict(PyInitialValue::Invalid), m_run_one_line_function(),
553*0b57cec5SDimitry Andric       m_run_one_line_str_global(),
554*0b57cec5SDimitry Andric       m_dictionary_name(m_debugger.GetInstanceName().AsCString()),
5559dba64beSDimitry Andric       m_active_io_handler(eIOHandlerNone), m_session_is_active(false),
5565ffd83dbSDimitry Andric       m_pty_secondary_is_open(false), m_valid_session(true), m_lock_count(0),
5579dba64beSDimitry Andric       m_command_thread_state(nullptr) {
558*0b57cec5SDimitry Andric   InitializePrivate();
559*0b57cec5SDimitry Andric 
560fe6060f1SDimitry Andric   m_scripted_process_interface_up =
561fe6060f1SDimitry Andric       std::make_unique<ScriptedProcessPythonInterface>(*this);
562fe6060f1SDimitry Andric 
563*0b57cec5SDimitry Andric   m_dictionary_name.append("_dict");
564*0b57cec5SDimitry Andric   StreamString run_string;
565*0b57cec5SDimitry Andric   run_string.Printf("%s = dict()", m_dictionary_name.c_str());
566*0b57cec5SDimitry Andric 
567*0b57cec5SDimitry Andric   Locker locker(this, Locker::AcquireLock, Locker::FreeAcquiredLock);
568*0b57cec5SDimitry Andric   PyRun_SimpleString(run_string.GetData());
569*0b57cec5SDimitry Andric 
570*0b57cec5SDimitry Andric   run_string.Clear();
571*0b57cec5SDimitry Andric   run_string.Printf(
572*0b57cec5SDimitry Andric       "run_one_line (%s, 'import copy, keyword, os, re, sys, uuid, lldb')",
573*0b57cec5SDimitry Andric       m_dictionary_name.c_str());
574*0b57cec5SDimitry Andric   PyRun_SimpleString(run_string.GetData());
575*0b57cec5SDimitry Andric 
576*0b57cec5SDimitry Andric   // Reloading modules requires a different syntax in Python 2 and Python 3.
577*0b57cec5SDimitry Andric   // This provides a consistent syntax no matter what version of Python.
578*0b57cec5SDimitry Andric   run_string.Clear();
579*0b57cec5SDimitry Andric   run_string.Printf("run_one_line (%s, 'from six.moves import reload_module')",
580*0b57cec5SDimitry Andric                     m_dictionary_name.c_str());
581*0b57cec5SDimitry Andric   PyRun_SimpleString(run_string.GetData());
582*0b57cec5SDimitry Andric 
583*0b57cec5SDimitry Andric   // WARNING: temporary code that loads Cocoa formatters - this should be done
584*0b57cec5SDimitry Andric   // on a per-platform basis rather than loading the whole set and letting the
585*0b57cec5SDimitry Andric   // individual formatter classes exploit APIs to check whether they can/cannot
586*0b57cec5SDimitry Andric   // do their task
587*0b57cec5SDimitry Andric   run_string.Clear();
588*0b57cec5SDimitry Andric   run_string.Printf(
589*0b57cec5SDimitry Andric       "run_one_line (%s, 'import lldb.formatters, lldb.formatters.cpp, pydoc')",
590*0b57cec5SDimitry Andric       m_dictionary_name.c_str());
591*0b57cec5SDimitry Andric   PyRun_SimpleString(run_string.GetData());
592*0b57cec5SDimitry Andric   run_string.Clear();
593*0b57cec5SDimitry Andric 
594*0b57cec5SDimitry Andric   run_string.Printf("run_one_line (%s, 'import lldb.embedded_interpreter; from "
595*0b57cec5SDimitry Andric                     "lldb.embedded_interpreter import run_python_interpreter; "
596*0b57cec5SDimitry Andric                     "from lldb.embedded_interpreter import run_one_line')",
597*0b57cec5SDimitry Andric                     m_dictionary_name.c_str());
598*0b57cec5SDimitry Andric   PyRun_SimpleString(run_string.GetData());
599*0b57cec5SDimitry Andric   run_string.Clear();
600*0b57cec5SDimitry Andric 
601*0b57cec5SDimitry Andric   run_string.Printf("run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64
602*0b57cec5SDimitry Andric                     "; pydoc.pager = pydoc.plainpager')",
603*0b57cec5SDimitry Andric                     m_dictionary_name.c_str(), m_debugger.GetID());
604*0b57cec5SDimitry Andric   PyRun_SimpleString(run_string.GetData());
605*0b57cec5SDimitry Andric }
606*0b57cec5SDimitry Andric 
607*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::~ScriptInterpreterPythonImpl() {
608*0b57cec5SDimitry Andric   // the session dictionary may hold objects with complex state which means
609*0b57cec5SDimitry Andric   // that they may need to be torn down with some level of smarts and that, in
610*0b57cec5SDimitry Andric   // turn, requires a valid thread state force Python to procure itself such a
611*0b57cec5SDimitry Andric   // thread state, nuke the session dictionary and then release it for others
612*0b57cec5SDimitry Andric   // to use and proceed with the rest of the shutdown
613*0b57cec5SDimitry Andric   auto gil_state = PyGILState_Ensure();
614*0b57cec5SDimitry Andric   m_session_dict.Reset();
615*0b57cec5SDimitry Andric   PyGILState_Release(gil_state);
616*0b57cec5SDimitry Andric }
617*0b57cec5SDimitry Andric 
618*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::IOHandlerActivated(IOHandler &io_handler,
619*0b57cec5SDimitry Andric                                                      bool interactive) {
620*0b57cec5SDimitry Andric   const char *instructions = nullptr;
621*0b57cec5SDimitry Andric 
622*0b57cec5SDimitry Andric   switch (m_active_io_handler) {
623*0b57cec5SDimitry Andric   case eIOHandlerNone:
624*0b57cec5SDimitry Andric     break;
625*0b57cec5SDimitry Andric   case eIOHandlerBreakpoint:
626*0b57cec5SDimitry Andric     instructions = R"(Enter your Python command(s). Type 'DONE' to end.
627*0b57cec5SDimitry Andric def function (frame, bp_loc, internal_dict):
628*0b57cec5SDimitry Andric     """frame: the lldb.SBFrame for the location at which you stopped
629*0b57cec5SDimitry Andric        bp_loc: an lldb.SBBreakpointLocation for the breakpoint location information
630*0b57cec5SDimitry Andric        internal_dict: an LLDB support object not to be used"""
631*0b57cec5SDimitry Andric )";
632*0b57cec5SDimitry Andric     break;
633*0b57cec5SDimitry Andric   case eIOHandlerWatchpoint:
634*0b57cec5SDimitry Andric     instructions = "Enter your Python command(s). Type 'DONE' to end.\n";
635*0b57cec5SDimitry Andric     break;
636*0b57cec5SDimitry Andric   }
637*0b57cec5SDimitry Andric 
638*0b57cec5SDimitry Andric   if (instructions) {
6399dba64beSDimitry Andric     StreamFileSP output_sp(io_handler.GetOutputStreamFileSP());
640*0b57cec5SDimitry Andric     if (output_sp && interactive) {
641*0b57cec5SDimitry Andric       output_sp->PutCString(instructions);
642*0b57cec5SDimitry Andric       output_sp->Flush();
643*0b57cec5SDimitry Andric     }
644*0b57cec5SDimitry Andric   }
645*0b57cec5SDimitry Andric }
646*0b57cec5SDimitry Andric 
647*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::IOHandlerInputComplete(IOHandler &io_handler,
648*0b57cec5SDimitry Andric                                                          std::string &data) {
649*0b57cec5SDimitry Andric   io_handler.SetIsDone(true);
650*0b57cec5SDimitry Andric   bool batch_mode = m_debugger.GetCommandInterpreter().GetBatchCommandMode();
651*0b57cec5SDimitry Andric 
652*0b57cec5SDimitry Andric   switch (m_active_io_handler) {
653*0b57cec5SDimitry Andric   case eIOHandlerNone:
654*0b57cec5SDimitry Andric     break;
655*0b57cec5SDimitry Andric   case eIOHandlerBreakpoint: {
656fe6060f1SDimitry Andric     std::vector<std::reference_wrapper<BreakpointOptions>> *bp_options_vec =
657fe6060f1SDimitry Andric         (std::vector<std::reference_wrapper<BreakpointOptions>> *)
658fe6060f1SDimitry Andric             io_handler.GetUserData();
659fe6060f1SDimitry Andric     for (BreakpointOptions &bp_options : *bp_options_vec) {
660*0b57cec5SDimitry Andric 
6619dba64beSDimitry Andric       auto data_up = std::make_unique<CommandDataPython>();
662*0b57cec5SDimitry Andric       if (!data_up)
663*0b57cec5SDimitry Andric         break;
664*0b57cec5SDimitry Andric       data_up->user_source.SplitIntoLines(data);
665*0b57cec5SDimitry Andric 
666480093f4SDimitry Andric       StructuredData::ObjectSP empty_args_sp;
667*0b57cec5SDimitry Andric       if (GenerateBreakpointCommandCallbackData(data_up->user_source,
668480093f4SDimitry Andric                                                 data_up->script_source,
669480093f4SDimitry Andric                                                 false)
670*0b57cec5SDimitry Andric               .Success()) {
671*0b57cec5SDimitry Andric         auto baton_sp = std::make_shared<BreakpointOptions::CommandBaton>(
672*0b57cec5SDimitry Andric             std::move(data_up));
673fe6060f1SDimitry Andric         bp_options.SetCallback(
674*0b57cec5SDimitry Andric             ScriptInterpreterPythonImpl::BreakpointCallbackFunction, baton_sp);
675*0b57cec5SDimitry Andric       } else if (!batch_mode) {
6769dba64beSDimitry Andric         StreamFileSP error_sp = io_handler.GetErrorStreamFileSP();
677*0b57cec5SDimitry Andric         if (error_sp) {
678*0b57cec5SDimitry Andric           error_sp->Printf("Warning: No command attached to breakpoint.\n");
679*0b57cec5SDimitry Andric           error_sp->Flush();
680*0b57cec5SDimitry Andric         }
681*0b57cec5SDimitry Andric       }
682*0b57cec5SDimitry Andric     }
683*0b57cec5SDimitry Andric     m_active_io_handler = eIOHandlerNone;
684*0b57cec5SDimitry Andric   } break;
685*0b57cec5SDimitry Andric   case eIOHandlerWatchpoint: {
686*0b57cec5SDimitry Andric     WatchpointOptions *wp_options =
687*0b57cec5SDimitry Andric         (WatchpointOptions *)io_handler.GetUserData();
6889dba64beSDimitry Andric     auto data_up = std::make_unique<WatchpointOptions::CommandData>();
689*0b57cec5SDimitry Andric     data_up->user_source.SplitIntoLines(data);
690*0b57cec5SDimitry Andric 
691*0b57cec5SDimitry Andric     if (GenerateWatchpointCommandCallbackData(data_up->user_source,
692*0b57cec5SDimitry Andric                                               data_up->script_source)) {
693*0b57cec5SDimitry Andric       auto baton_sp =
694*0b57cec5SDimitry Andric           std::make_shared<WatchpointOptions::CommandBaton>(std::move(data_up));
695*0b57cec5SDimitry Andric       wp_options->SetCallback(
696*0b57cec5SDimitry Andric           ScriptInterpreterPythonImpl::WatchpointCallbackFunction, baton_sp);
697*0b57cec5SDimitry Andric     } else if (!batch_mode) {
6989dba64beSDimitry Andric       StreamFileSP error_sp = io_handler.GetErrorStreamFileSP();
699*0b57cec5SDimitry Andric       if (error_sp) {
700*0b57cec5SDimitry Andric         error_sp->Printf("Warning: No command attached to breakpoint.\n");
701*0b57cec5SDimitry Andric         error_sp->Flush();
702*0b57cec5SDimitry Andric       }
703*0b57cec5SDimitry Andric     }
704*0b57cec5SDimitry Andric     m_active_io_handler = eIOHandlerNone;
705*0b57cec5SDimitry Andric   } break;
706*0b57cec5SDimitry Andric   }
707*0b57cec5SDimitry Andric }
708*0b57cec5SDimitry Andric 
709*0b57cec5SDimitry Andric lldb::ScriptInterpreterSP
710*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::CreateInstance(Debugger &debugger) {
711*0b57cec5SDimitry Andric   return std::make_shared<ScriptInterpreterPythonImpl>(debugger);
712*0b57cec5SDimitry Andric }
713*0b57cec5SDimitry Andric 
714*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::LeaveSession() {
715*0b57cec5SDimitry Andric   Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT));
716*0b57cec5SDimitry Andric   if (log)
717*0b57cec5SDimitry Andric     log->PutCString("ScriptInterpreterPythonImpl::LeaveSession()");
718*0b57cec5SDimitry Andric 
7199dba64beSDimitry Andric   // Unset the LLDB global variables.
7209dba64beSDimitry Andric   PyRun_SimpleString("lldb.debugger = None; lldb.target = None; lldb.process "
7219dba64beSDimitry Andric                      "= None; lldb.thread = None; lldb.frame = None");
7229dba64beSDimitry Andric 
723*0b57cec5SDimitry Andric   // checking that we have a valid thread state - since we use our own
724*0b57cec5SDimitry Andric   // threading and locking in some (rare) cases during cleanup Python may end
725*0b57cec5SDimitry Andric   // up believing we have no thread state and PyImport_AddModule will crash if
726*0b57cec5SDimitry Andric   // that is the case - since that seems to only happen when destroying the
727*0b57cec5SDimitry Andric   // SBDebugger, we can make do without clearing up stdout and stderr
728*0b57cec5SDimitry Andric 
729*0b57cec5SDimitry Andric   // rdar://problem/11292882
730*0b57cec5SDimitry Andric   // When the current thread state is NULL, PyThreadState_Get() issues a fatal
731*0b57cec5SDimitry Andric   // error.
732*0b57cec5SDimitry Andric   if (PyThreadState_GetDict()) {
733*0b57cec5SDimitry Andric     PythonDictionary &sys_module_dict = GetSysModuleDictionary();
734*0b57cec5SDimitry Andric     if (sys_module_dict.IsValid()) {
735*0b57cec5SDimitry Andric       if (m_saved_stdin.IsValid()) {
736*0b57cec5SDimitry Andric         sys_module_dict.SetItemForKey(PythonString("stdin"), m_saved_stdin);
737*0b57cec5SDimitry Andric         m_saved_stdin.Reset();
738*0b57cec5SDimitry Andric       }
739*0b57cec5SDimitry Andric       if (m_saved_stdout.IsValid()) {
740*0b57cec5SDimitry Andric         sys_module_dict.SetItemForKey(PythonString("stdout"), m_saved_stdout);
741*0b57cec5SDimitry Andric         m_saved_stdout.Reset();
742*0b57cec5SDimitry Andric       }
743*0b57cec5SDimitry Andric       if (m_saved_stderr.IsValid()) {
744*0b57cec5SDimitry Andric         sys_module_dict.SetItemForKey(PythonString("stderr"), m_saved_stderr);
745*0b57cec5SDimitry Andric         m_saved_stderr.Reset();
746*0b57cec5SDimitry Andric       }
747*0b57cec5SDimitry Andric     }
748*0b57cec5SDimitry Andric   }
749*0b57cec5SDimitry Andric 
750*0b57cec5SDimitry Andric   m_session_is_active = false;
751*0b57cec5SDimitry Andric }
752*0b57cec5SDimitry Andric 
7539dba64beSDimitry Andric bool ScriptInterpreterPythonImpl::SetStdHandle(FileSP file_sp,
7549dba64beSDimitry Andric                                                const char *py_name,
7559dba64beSDimitry Andric                                                PythonObject &save_file,
756*0b57cec5SDimitry Andric                                                const char *mode) {
7579dba64beSDimitry Andric   if (!file_sp || !*file_sp) {
7589dba64beSDimitry Andric     save_file.Reset();
7599dba64beSDimitry Andric     return false;
7609dba64beSDimitry Andric   }
7619dba64beSDimitry Andric   File &file = *file_sp;
7629dba64beSDimitry Andric 
763*0b57cec5SDimitry Andric   // Flush the file before giving it to python to avoid interleaved output.
764*0b57cec5SDimitry Andric   file.Flush();
765*0b57cec5SDimitry Andric 
766*0b57cec5SDimitry Andric   PythonDictionary &sys_module_dict = GetSysModuleDictionary();
767*0b57cec5SDimitry Andric 
7689dba64beSDimitry Andric   auto new_file = PythonFile::FromFile(file, mode);
7699dba64beSDimitry Andric   if (!new_file) {
7709dba64beSDimitry Andric     llvm::consumeError(new_file.takeError());
771*0b57cec5SDimitry Andric     return false;
772*0b57cec5SDimitry Andric   }
773*0b57cec5SDimitry Andric 
7749dba64beSDimitry Andric   save_file = sys_module_dict.GetItemForKey(PythonString(py_name));
7759dba64beSDimitry Andric 
7769dba64beSDimitry Andric   sys_module_dict.SetItemForKey(PythonString(py_name), new_file.get());
7779dba64beSDimitry Andric   return true;
7789dba64beSDimitry Andric }
7799dba64beSDimitry Andric 
780*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::EnterSession(uint16_t on_entry_flags,
7819dba64beSDimitry Andric                                                FileSP in_sp, FileSP out_sp,
7829dba64beSDimitry Andric                                                FileSP err_sp) {
783*0b57cec5SDimitry Andric   // If we have already entered the session, without having officially 'left'
784*0b57cec5SDimitry Andric   // it, then there is no need to 'enter' it again.
785*0b57cec5SDimitry Andric   Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT));
786*0b57cec5SDimitry Andric   if (m_session_is_active) {
7879dba64beSDimitry Andric     LLDB_LOGF(
7889dba64beSDimitry Andric         log,
789*0b57cec5SDimitry Andric         "ScriptInterpreterPythonImpl::EnterSession(on_entry_flags=0x%" PRIx16
790*0b57cec5SDimitry Andric         ") session is already active, returning without doing anything",
791*0b57cec5SDimitry Andric         on_entry_flags);
792*0b57cec5SDimitry Andric     return false;
793*0b57cec5SDimitry Andric   }
794*0b57cec5SDimitry Andric 
7959dba64beSDimitry Andric   LLDB_LOGF(
7969dba64beSDimitry Andric       log,
7979dba64beSDimitry Andric       "ScriptInterpreterPythonImpl::EnterSession(on_entry_flags=0x%" PRIx16 ")",
798*0b57cec5SDimitry Andric       on_entry_flags);
799*0b57cec5SDimitry Andric 
800*0b57cec5SDimitry Andric   m_session_is_active = true;
801*0b57cec5SDimitry Andric 
802*0b57cec5SDimitry Andric   StreamString run_string;
803*0b57cec5SDimitry Andric 
804*0b57cec5SDimitry Andric   if (on_entry_flags & Locker::InitGlobals) {
805*0b57cec5SDimitry Andric     run_string.Printf("run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64,
806*0b57cec5SDimitry Andric                       m_dictionary_name.c_str(), m_debugger.GetID());
807*0b57cec5SDimitry Andric     run_string.Printf(
808*0b57cec5SDimitry Andric         "; lldb.debugger = lldb.SBDebugger.FindDebuggerWithID (%" PRIu64 ")",
809*0b57cec5SDimitry Andric         m_debugger.GetID());
810*0b57cec5SDimitry Andric     run_string.PutCString("; lldb.target = lldb.debugger.GetSelectedTarget()");
811*0b57cec5SDimitry Andric     run_string.PutCString("; lldb.process = lldb.target.GetProcess()");
812*0b57cec5SDimitry Andric     run_string.PutCString("; lldb.thread = lldb.process.GetSelectedThread ()");
813*0b57cec5SDimitry Andric     run_string.PutCString("; lldb.frame = lldb.thread.GetSelectedFrame ()");
814*0b57cec5SDimitry Andric     run_string.PutCString("')");
815*0b57cec5SDimitry Andric   } else {
816*0b57cec5SDimitry Andric     // If we aren't initing the globals, we should still always set the
817*0b57cec5SDimitry Andric     // debugger (since that is always unique.)
818*0b57cec5SDimitry Andric     run_string.Printf("run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64,
819*0b57cec5SDimitry Andric                       m_dictionary_name.c_str(), m_debugger.GetID());
820*0b57cec5SDimitry Andric     run_string.Printf(
821*0b57cec5SDimitry Andric         "; lldb.debugger = lldb.SBDebugger.FindDebuggerWithID (%" PRIu64 ")",
822*0b57cec5SDimitry Andric         m_debugger.GetID());
823*0b57cec5SDimitry Andric     run_string.PutCString("')");
824*0b57cec5SDimitry Andric   }
825*0b57cec5SDimitry Andric 
826*0b57cec5SDimitry Andric   PyRun_SimpleString(run_string.GetData());
827*0b57cec5SDimitry Andric   run_string.Clear();
828*0b57cec5SDimitry Andric 
829*0b57cec5SDimitry Andric   PythonDictionary &sys_module_dict = GetSysModuleDictionary();
830*0b57cec5SDimitry Andric   if (sys_module_dict.IsValid()) {
8319dba64beSDimitry Andric     lldb::FileSP top_in_sp;
8329dba64beSDimitry Andric     lldb::StreamFileSP top_out_sp, top_err_sp;
8339dba64beSDimitry Andric     if (!in_sp || !out_sp || !err_sp || !*in_sp || !*out_sp || !*err_sp)
8349dba64beSDimitry Andric       m_debugger.AdoptTopIOHandlerFilesIfInvalid(top_in_sp, top_out_sp,
8359dba64beSDimitry Andric                                                  top_err_sp);
836*0b57cec5SDimitry Andric 
837*0b57cec5SDimitry Andric     if (on_entry_flags & Locker::NoSTDIN) {
838*0b57cec5SDimitry Andric       m_saved_stdin.Reset();
839*0b57cec5SDimitry Andric     } else {
8409dba64beSDimitry Andric       if (!SetStdHandle(in_sp, "stdin", m_saved_stdin, "r")) {
8419dba64beSDimitry Andric         if (top_in_sp)
8429dba64beSDimitry Andric           SetStdHandle(top_in_sp, "stdin", m_saved_stdin, "r");
843*0b57cec5SDimitry Andric       }
844*0b57cec5SDimitry Andric     }
845*0b57cec5SDimitry Andric 
8469dba64beSDimitry Andric     if (!SetStdHandle(out_sp, "stdout", m_saved_stdout, "w")) {
8479dba64beSDimitry Andric       if (top_out_sp)
8489dba64beSDimitry Andric         SetStdHandle(top_out_sp->GetFileSP(), "stdout", m_saved_stdout, "w");
849*0b57cec5SDimitry Andric     }
850*0b57cec5SDimitry Andric 
8519dba64beSDimitry Andric     if (!SetStdHandle(err_sp, "stderr", m_saved_stderr, "w")) {
8529dba64beSDimitry Andric       if (top_err_sp)
8539dba64beSDimitry Andric         SetStdHandle(top_err_sp->GetFileSP(), "stderr", m_saved_stderr, "w");
854*0b57cec5SDimitry Andric     }
855*0b57cec5SDimitry Andric   }
856*0b57cec5SDimitry Andric 
857*0b57cec5SDimitry Andric   if (PyErr_Occurred())
858*0b57cec5SDimitry Andric     PyErr_Clear();
859*0b57cec5SDimitry Andric 
860*0b57cec5SDimitry Andric   return true;
861*0b57cec5SDimitry Andric }
862*0b57cec5SDimitry Andric 
8639dba64beSDimitry Andric PythonModule &ScriptInterpreterPythonImpl::GetMainModule() {
864*0b57cec5SDimitry Andric   if (!m_main_module.IsValid())
8659dba64beSDimitry Andric     m_main_module = unwrapIgnoringErrors(PythonModule::Import("__main__"));
866*0b57cec5SDimitry Andric   return m_main_module;
867*0b57cec5SDimitry Andric }
868*0b57cec5SDimitry Andric 
869*0b57cec5SDimitry Andric PythonDictionary &ScriptInterpreterPythonImpl::GetSessionDictionary() {
870*0b57cec5SDimitry Andric   if (m_session_dict.IsValid())
871*0b57cec5SDimitry Andric     return m_session_dict;
872*0b57cec5SDimitry Andric 
873*0b57cec5SDimitry Andric   PythonObject &main_module = GetMainModule();
874*0b57cec5SDimitry Andric   if (!main_module.IsValid())
875*0b57cec5SDimitry Andric     return m_session_dict;
876*0b57cec5SDimitry Andric 
877*0b57cec5SDimitry Andric   PythonDictionary main_dict(PyRefType::Borrowed,
878*0b57cec5SDimitry Andric                              PyModule_GetDict(main_module.get()));
879*0b57cec5SDimitry Andric   if (!main_dict.IsValid())
880*0b57cec5SDimitry Andric     return m_session_dict;
881*0b57cec5SDimitry Andric 
8829dba64beSDimitry Andric   m_session_dict = unwrapIgnoringErrors(
8839dba64beSDimitry Andric       As<PythonDictionary>(main_dict.GetItem(m_dictionary_name)));
884*0b57cec5SDimitry Andric   return m_session_dict;
885*0b57cec5SDimitry Andric }
886*0b57cec5SDimitry Andric 
887*0b57cec5SDimitry Andric PythonDictionary &ScriptInterpreterPythonImpl::GetSysModuleDictionary() {
888*0b57cec5SDimitry Andric   if (m_sys_module_dict.IsValid())
889*0b57cec5SDimitry Andric     return m_sys_module_dict;
8909dba64beSDimitry Andric   PythonModule sys_module = unwrapIgnoringErrors(PythonModule::Import("sys"));
8919dba64beSDimitry Andric   m_sys_module_dict = sys_module.GetDictionary();
892*0b57cec5SDimitry Andric   return m_sys_module_dict;
893*0b57cec5SDimitry Andric }
894*0b57cec5SDimitry Andric 
895480093f4SDimitry Andric llvm::Expected<unsigned>
896480093f4SDimitry Andric ScriptInterpreterPythonImpl::GetMaxPositionalArgumentsForCallable(
897480093f4SDimitry Andric     const llvm::StringRef &callable_name) {
898480093f4SDimitry Andric   if (callable_name.empty()) {
899480093f4SDimitry Andric     return llvm::createStringError(
900480093f4SDimitry Andric         llvm::inconvertibleErrorCode(),
901480093f4SDimitry Andric         "called with empty callable name.");
902480093f4SDimitry Andric   }
903480093f4SDimitry Andric   Locker py_lock(this, Locker::AcquireLock |
904480093f4SDimitry Andric                  Locker::InitSession |
905480093f4SDimitry Andric                  Locker::NoSTDIN);
906480093f4SDimitry Andric   auto dict = PythonModule::MainModule()
907480093f4SDimitry Andric       .ResolveName<PythonDictionary>(m_dictionary_name);
908480093f4SDimitry Andric   auto pfunc = PythonObject::ResolveNameWithDictionary<PythonCallable>(
909480093f4SDimitry Andric       callable_name, dict);
910480093f4SDimitry Andric   if (!pfunc.IsAllocated()) {
911480093f4SDimitry Andric     return llvm::createStringError(
912480093f4SDimitry Andric         llvm::inconvertibleErrorCode(),
913480093f4SDimitry Andric         "can't find callable: %s", callable_name.str().c_str());
914480093f4SDimitry Andric   }
915480093f4SDimitry Andric   llvm::Expected<PythonCallable::ArgInfo> arg_info = pfunc.GetArgInfo();
916480093f4SDimitry Andric   if (!arg_info)
917480093f4SDimitry Andric     return arg_info.takeError();
918480093f4SDimitry Andric   return arg_info.get().max_positional_args;
919480093f4SDimitry Andric }
920480093f4SDimitry Andric 
921*0b57cec5SDimitry Andric static std::string GenerateUniqueName(const char *base_name_wanted,
922*0b57cec5SDimitry Andric                                       uint32_t &functions_counter,
923*0b57cec5SDimitry Andric                                       const void *name_token = nullptr) {
924*0b57cec5SDimitry Andric   StreamString sstr;
925*0b57cec5SDimitry Andric 
926*0b57cec5SDimitry Andric   if (!base_name_wanted)
927*0b57cec5SDimitry Andric     return std::string();
928*0b57cec5SDimitry Andric 
929*0b57cec5SDimitry Andric   if (!name_token)
930*0b57cec5SDimitry Andric     sstr.Printf("%s_%d", base_name_wanted, functions_counter++);
931*0b57cec5SDimitry Andric   else
932*0b57cec5SDimitry Andric     sstr.Printf("%s_%p", base_name_wanted, name_token);
933*0b57cec5SDimitry Andric 
9345ffd83dbSDimitry Andric   return std::string(sstr.GetString());
935*0b57cec5SDimitry Andric }
936*0b57cec5SDimitry Andric 
937*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GetEmbeddedInterpreterModuleObjects() {
938*0b57cec5SDimitry Andric   if (m_run_one_line_function.IsValid())
939*0b57cec5SDimitry Andric     return true;
940*0b57cec5SDimitry Andric 
941*0b57cec5SDimitry Andric   PythonObject module(PyRefType::Borrowed,
942*0b57cec5SDimitry Andric                       PyImport_AddModule("lldb.embedded_interpreter"));
943*0b57cec5SDimitry Andric   if (!module.IsValid())
944*0b57cec5SDimitry Andric     return false;
945*0b57cec5SDimitry Andric 
946*0b57cec5SDimitry Andric   PythonDictionary module_dict(PyRefType::Borrowed,
947*0b57cec5SDimitry Andric                                PyModule_GetDict(module.get()));
948*0b57cec5SDimitry Andric   if (!module_dict.IsValid())
949*0b57cec5SDimitry Andric     return false;
950*0b57cec5SDimitry Andric 
951*0b57cec5SDimitry Andric   m_run_one_line_function =
952*0b57cec5SDimitry Andric       module_dict.GetItemForKey(PythonString("run_one_line"));
953*0b57cec5SDimitry Andric   m_run_one_line_str_global =
954*0b57cec5SDimitry Andric       module_dict.GetItemForKey(PythonString("g_run_one_line_str"));
955*0b57cec5SDimitry Andric   return m_run_one_line_function.IsValid();
956*0b57cec5SDimitry Andric }
957*0b57cec5SDimitry Andric 
958*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ExecuteOneLine(
959*0b57cec5SDimitry Andric     llvm::StringRef command, CommandReturnObject *result,
960*0b57cec5SDimitry Andric     const ExecuteScriptOptions &options) {
961*0b57cec5SDimitry Andric   std::string command_str = command.str();
962*0b57cec5SDimitry Andric 
963*0b57cec5SDimitry Andric   if (!m_valid_session)
964*0b57cec5SDimitry Andric     return false;
965*0b57cec5SDimitry Andric 
966*0b57cec5SDimitry Andric   if (!command.empty()) {
967*0b57cec5SDimitry Andric     // We want to call run_one_line, passing in the dictionary and the command
968*0b57cec5SDimitry Andric     // string.  We cannot do this through PyRun_SimpleString here because the
969*0b57cec5SDimitry Andric     // command string may contain escaped characters, and putting it inside
970*0b57cec5SDimitry Andric     // another string to pass to PyRun_SimpleString messes up the escaping.  So
971*0b57cec5SDimitry Andric     // we use the following more complicated method to pass the command string
972*0b57cec5SDimitry Andric     // directly down to Python.
9735ffd83dbSDimitry Andric     llvm::Expected<std::unique_ptr<ScriptInterpreterIORedirect>>
9745ffd83dbSDimitry Andric         io_redirect_or_error = ScriptInterpreterIORedirect::Create(
9755ffd83dbSDimitry Andric             options.GetEnableIO(), m_debugger, result);
9765ffd83dbSDimitry Andric     if (!io_redirect_or_error) {
9775ffd83dbSDimitry Andric       if (result)
9785ffd83dbSDimitry Andric         result->AppendErrorWithFormatv(
9795ffd83dbSDimitry Andric             "failed to redirect I/O: {0}\n",
9805ffd83dbSDimitry Andric             llvm::fmt_consume(io_redirect_or_error.takeError()));
9815ffd83dbSDimitry Andric       else
9825ffd83dbSDimitry Andric         llvm::consumeError(io_redirect_or_error.takeError());
9839dba64beSDimitry Andric       return false;
9849dba64beSDimitry Andric     }
9855ffd83dbSDimitry Andric 
9865ffd83dbSDimitry Andric     ScriptInterpreterIORedirect &io_redirect = **io_redirect_or_error;
987*0b57cec5SDimitry Andric 
988*0b57cec5SDimitry Andric     bool success = false;
989*0b57cec5SDimitry Andric     {
990*0b57cec5SDimitry Andric       // WARNING!  It's imperative that this RAII scope be as tight as
991*0b57cec5SDimitry Andric       // possible. In particular, the scope must end *before* we try to join
992*0b57cec5SDimitry Andric       // the read thread.  The reason for this is that a pre-requisite for
993*0b57cec5SDimitry Andric       // joining the read thread is that we close the write handle (to break
994*0b57cec5SDimitry Andric       // the pipe and cause it to wake up and exit).  But acquiring the GIL as
995*0b57cec5SDimitry Andric       // below will redirect Python's stdio to use this same handle.  If we
996*0b57cec5SDimitry Andric       // close the handle while Python is still using it, bad things will
997*0b57cec5SDimitry Andric       // happen.
998*0b57cec5SDimitry Andric       Locker locker(
999*0b57cec5SDimitry Andric           this,
1000*0b57cec5SDimitry Andric           Locker::AcquireLock | Locker::InitSession |
1001*0b57cec5SDimitry Andric               (options.GetSetLLDBGlobals() ? Locker::InitGlobals : 0) |
1002*0b57cec5SDimitry Andric               ((result && result->GetInteractive()) ? 0 : Locker::NoSTDIN),
10035ffd83dbSDimitry Andric           Locker::FreeAcquiredLock | Locker::TearDownSession,
10045ffd83dbSDimitry Andric           io_redirect.GetInputFile(), io_redirect.GetOutputFile(),
10055ffd83dbSDimitry Andric           io_redirect.GetErrorFile());
1006*0b57cec5SDimitry Andric 
1007*0b57cec5SDimitry Andric       // Find the correct script interpreter dictionary in the main module.
1008*0b57cec5SDimitry Andric       PythonDictionary &session_dict = GetSessionDictionary();
1009*0b57cec5SDimitry Andric       if (session_dict.IsValid()) {
1010*0b57cec5SDimitry Andric         if (GetEmbeddedInterpreterModuleObjects()) {
1011*0b57cec5SDimitry Andric           if (PyCallable_Check(m_run_one_line_function.get())) {
1012*0b57cec5SDimitry Andric             PythonObject pargs(
1013*0b57cec5SDimitry Andric                 PyRefType::Owned,
1014*0b57cec5SDimitry Andric                 Py_BuildValue("(Os)", session_dict.get(), command_str.c_str()));
1015*0b57cec5SDimitry Andric             if (pargs.IsValid()) {
1016*0b57cec5SDimitry Andric               PythonObject return_value(
1017*0b57cec5SDimitry Andric                   PyRefType::Owned,
1018*0b57cec5SDimitry Andric                   PyObject_CallObject(m_run_one_line_function.get(),
1019*0b57cec5SDimitry Andric                                       pargs.get()));
1020*0b57cec5SDimitry Andric               if (return_value.IsValid())
1021*0b57cec5SDimitry Andric                 success = true;
1022*0b57cec5SDimitry Andric               else if (options.GetMaskoutErrors() && PyErr_Occurred()) {
1023*0b57cec5SDimitry Andric                 PyErr_Print();
1024*0b57cec5SDimitry Andric                 PyErr_Clear();
1025*0b57cec5SDimitry Andric               }
1026*0b57cec5SDimitry Andric             }
1027*0b57cec5SDimitry Andric           }
1028*0b57cec5SDimitry Andric         }
1029*0b57cec5SDimitry Andric       }
1030*0b57cec5SDimitry Andric 
10315ffd83dbSDimitry Andric       io_redirect.Flush();
1032*0b57cec5SDimitry Andric     }
1033*0b57cec5SDimitry Andric 
1034*0b57cec5SDimitry Andric     if (success)
1035*0b57cec5SDimitry Andric       return true;
1036*0b57cec5SDimitry Andric 
1037*0b57cec5SDimitry Andric     // The one-liner failed.  Append the error message.
1038*0b57cec5SDimitry Andric     if (result) {
1039*0b57cec5SDimitry Andric       result->AppendErrorWithFormat(
1040*0b57cec5SDimitry Andric           "python failed attempting to evaluate '%s'\n", command_str.c_str());
1041*0b57cec5SDimitry Andric     }
1042*0b57cec5SDimitry Andric     return false;
1043*0b57cec5SDimitry Andric   }
1044*0b57cec5SDimitry Andric 
1045*0b57cec5SDimitry Andric   if (result)
1046*0b57cec5SDimitry Andric     result->AppendError("empty command passed to python\n");
1047*0b57cec5SDimitry Andric   return false;
1048*0b57cec5SDimitry Andric }
1049*0b57cec5SDimitry Andric 
1050*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::ExecuteInterpreterLoop() {
1051e8d8bef9SDimitry Andric   LLDB_SCOPED_TIMER();
1052*0b57cec5SDimitry Andric 
1053*0b57cec5SDimitry Andric   Debugger &debugger = m_debugger;
1054*0b57cec5SDimitry Andric 
1055*0b57cec5SDimitry Andric   // At the moment, the only time the debugger does not have an input file
1056*0b57cec5SDimitry Andric   // handle is when this is called directly from Python, in which case it is
1057*0b57cec5SDimitry Andric   // both dangerous and unnecessary (not to mention confusing) to try to embed
1058*0b57cec5SDimitry Andric   // a running interpreter loop inside the already running Python interpreter
1059*0b57cec5SDimitry Andric   // loop, so we won't do it.
1060*0b57cec5SDimitry Andric 
10619dba64beSDimitry Andric   if (!debugger.GetInputFile().IsValid())
1062*0b57cec5SDimitry Andric     return;
1063*0b57cec5SDimitry Andric 
1064*0b57cec5SDimitry Andric   IOHandlerSP io_handler_sp(new IOHandlerPythonInterpreter(debugger, this));
1065*0b57cec5SDimitry Andric   if (io_handler_sp) {
10665ffd83dbSDimitry Andric     debugger.RunIOHandlerAsync(io_handler_sp);
1067*0b57cec5SDimitry Andric   }
1068*0b57cec5SDimitry Andric }
1069*0b57cec5SDimitry Andric 
1070*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::Interrupt() {
1071*0b57cec5SDimitry Andric   Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT));
1072*0b57cec5SDimitry Andric 
1073*0b57cec5SDimitry Andric   if (IsExecutingPython()) {
1074*0b57cec5SDimitry Andric     PyThreadState *state = PyThreadState_GET();
1075*0b57cec5SDimitry Andric     if (!state)
1076*0b57cec5SDimitry Andric       state = GetThreadState();
1077*0b57cec5SDimitry Andric     if (state) {
1078*0b57cec5SDimitry Andric       long tid = state->thread_id;
1079*0b57cec5SDimitry Andric       PyThreadState_Swap(state);
1080*0b57cec5SDimitry Andric       int num_threads = PyThreadState_SetAsyncExc(tid, PyExc_KeyboardInterrupt);
10819dba64beSDimitry Andric       LLDB_LOGF(log,
10829dba64beSDimitry Andric                 "ScriptInterpreterPythonImpl::Interrupt() sending "
1083*0b57cec5SDimitry Andric                 "PyExc_KeyboardInterrupt (tid = %li, num_threads = %i)...",
1084*0b57cec5SDimitry Andric                 tid, num_threads);
1085*0b57cec5SDimitry Andric       return true;
1086*0b57cec5SDimitry Andric     }
1087*0b57cec5SDimitry Andric   }
10889dba64beSDimitry Andric   LLDB_LOGF(log,
1089*0b57cec5SDimitry Andric             "ScriptInterpreterPythonImpl::Interrupt() python code not running, "
1090*0b57cec5SDimitry Andric             "can't interrupt");
1091*0b57cec5SDimitry Andric   return false;
1092*0b57cec5SDimitry Andric }
10939dba64beSDimitry Andric 
1094*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ExecuteOneLineWithReturn(
1095*0b57cec5SDimitry Andric     llvm::StringRef in_string, ScriptInterpreter::ScriptReturnType return_type,
1096*0b57cec5SDimitry Andric     void *ret_value, const ExecuteScriptOptions &options) {
1097*0b57cec5SDimitry Andric 
1098fe6060f1SDimitry Andric   llvm::Expected<std::unique_ptr<ScriptInterpreterIORedirect>>
1099fe6060f1SDimitry Andric       io_redirect_or_error = ScriptInterpreterIORedirect::Create(
1100fe6060f1SDimitry Andric           options.GetEnableIO(), m_debugger, /*result=*/nullptr);
1101fe6060f1SDimitry Andric 
1102fe6060f1SDimitry Andric   if (!io_redirect_or_error) {
1103fe6060f1SDimitry Andric     llvm::consumeError(io_redirect_or_error.takeError());
1104fe6060f1SDimitry Andric     return false;
1105fe6060f1SDimitry Andric   }
1106fe6060f1SDimitry Andric 
1107fe6060f1SDimitry Andric   ScriptInterpreterIORedirect &io_redirect = **io_redirect_or_error;
1108fe6060f1SDimitry Andric 
1109*0b57cec5SDimitry Andric   Locker locker(this,
1110*0b57cec5SDimitry Andric                 Locker::AcquireLock | Locker::InitSession |
1111*0b57cec5SDimitry Andric                     (options.GetSetLLDBGlobals() ? Locker::InitGlobals : 0) |
1112*0b57cec5SDimitry Andric                     Locker::NoSTDIN,
1113fe6060f1SDimitry Andric                 Locker::FreeAcquiredLock | Locker::TearDownSession,
1114fe6060f1SDimitry Andric                 io_redirect.GetInputFile(), io_redirect.GetOutputFile(),
1115fe6060f1SDimitry Andric                 io_redirect.GetErrorFile());
1116*0b57cec5SDimitry Andric 
11179dba64beSDimitry Andric   PythonModule &main_module = GetMainModule();
11189dba64beSDimitry Andric   PythonDictionary globals = main_module.GetDictionary();
1119*0b57cec5SDimitry Andric 
1120*0b57cec5SDimitry Andric   PythonDictionary locals = GetSessionDictionary();
11219dba64beSDimitry Andric   if (!locals.IsValid())
11229dba64beSDimitry Andric     locals = unwrapIgnoringErrors(
11239dba64beSDimitry Andric         As<PythonDictionary>(globals.GetAttribute(m_dictionary_name)));
1124*0b57cec5SDimitry Andric   if (!locals.IsValid())
1125*0b57cec5SDimitry Andric     locals = globals;
1126*0b57cec5SDimitry Andric 
11279dba64beSDimitry Andric   Expected<PythonObject> maybe_py_return =
11289dba64beSDimitry Andric       runStringOneLine(in_string, globals, locals);
1129*0b57cec5SDimitry Andric 
11309dba64beSDimitry Andric   if (!maybe_py_return) {
11319dba64beSDimitry Andric     llvm::handleAllErrors(
11329dba64beSDimitry Andric         maybe_py_return.takeError(),
11339dba64beSDimitry Andric         [&](PythonException &E) {
11349dba64beSDimitry Andric           E.Restore();
11359dba64beSDimitry Andric           if (options.GetMaskoutErrors()) {
11369dba64beSDimitry Andric             if (E.Matches(PyExc_SyntaxError)) {
11379dba64beSDimitry Andric               PyErr_Print();
1138*0b57cec5SDimitry Andric             }
11399dba64beSDimitry Andric             PyErr_Clear();
11409dba64beSDimitry Andric           }
11419dba64beSDimitry Andric         },
11429dba64beSDimitry Andric         [](const llvm::ErrorInfoBase &E) {});
11439dba64beSDimitry Andric     return false;
1144*0b57cec5SDimitry Andric   }
1145*0b57cec5SDimitry Andric 
11469dba64beSDimitry Andric   PythonObject py_return = std::move(maybe_py_return.get());
11479dba64beSDimitry Andric   assert(py_return.IsValid());
11489dba64beSDimitry Andric 
1149*0b57cec5SDimitry Andric   switch (return_type) {
1150*0b57cec5SDimitry Andric   case eScriptReturnTypeCharPtr: // "char *"
1151*0b57cec5SDimitry Andric   {
1152*0b57cec5SDimitry Andric     const char format[3] = "s#";
11539dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (char **)ret_value);
1154*0b57cec5SDimitry Andric   }
1155*0b57cec5SDimitry Andric   case eScriptReturnTypeCharStrOrNone: // char* or NULL if py_return ==
1156*0b57cec5SDimitry Andric                                        // Py_None
1157*0b57cec5SDimitry Andric   {
1158*0b57cec5SDimitry Andric     const char format[3] = "z";
11599dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (char **)ret_value);
1160*0b57cec5SDimitry Andric   }
1161*0b57cec5SDimitry Andric   case eScriptReturnTypeBool: {
1162*0b57cec5SDimitry Andric     const char format[2] = "b";
11639dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (bool *)ret_value);
1164*0b57cec5SDimitry Andric   }
1165*0b57cec5SDimitry Andric   case eScriptReturnTypeShortInt: {
1166*0b57cec5SDimitry Andric     const char format[2] = "h";
11679dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (short *)ret_value);
1168*0b57cec5SDimitry Andric   }
1169*0b57cec5SDimitry Andric   case eScriptReturnTypeShortIntUnsigned: {
1170*0b57cec5SDimitry Andric     const char format[2] = "H";
11719dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (unsigned short *)ret_value);
1172*0b57cec5SDimitry Andric   }
1173*0b57cec5SDimitry Andric   case eScriptReturnTypeInt: {
1174*0b57cec5SDimitry Andric     const char format[2] = "i";
11759dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (int *)ret_value);
1176*0b57cec5SDimitry Andric   }
1177*0b57cec5SDimitry Andric   case eScriptReturnTypeIntUnsigned: {
1178*0b57cec5SDimitry Andric     const char format[2] = "I";
11799dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (unsigned int *)ret_value);
1180*0b57cec5SDimitry Andric   }
1181*0b57cec5SDimitry Andric   case eScriptReturnTypeLongInt: {
1182*0b57cec5SDimitry Andric     const char format[2] = "l";
11839dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (long *)ret_value);
1184*0b57cec5SDimitry Andric   }
1185*0b57cec5SDimitry Andric   case eScriptReturnTypeLongIntUnsigned: {
1186*0b57cec5SDimitry Andric     const char format[2] = "k";
11879dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (unsigned long *)ret_value);
1188*0b57cec5SDimitry Andric   }
1189*0b57cec5SDimitry Andric   case eScriptReturnTypeLongLong: {
1190*0b57cec5SDimitry Andric     const char format[2] = "L";
11919dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (long long *)ret_value);
1192*0b57cec5SDimitry Andric   }
1193*0b57cec5SDimitry Andric   case eScriptReturnTypeLongLongUnsigned: {
1194*0b57cec5SDimitry Andric     const char format[2] = "K";
11959dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format,
11969dba64beSDimitry Andric                        (unsigned long long *)ret_value);
1197*0b57cec5SDimitry Andric   }
1198*0b57cec5SDimitry Andric   case eScriptReturnTypeFloat: {
1199*0b57cec5SDimitry Andric     const char format[2] = "f";
12009dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (float *)ret_value);
1201*0b57cec5SDimitry Andric   }
1202*0b57cec5SDimitry Andric   case eScriptReturnTypeDouble: {
1203*0b57cec5SDimitry Andric     const char format[2] = "d";
12049dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (double *)ret_value);
1205*0b57cec5SDimitry Andric   }
1206*0b57cec5SDimitry Andric   case eScriptReturnTypeChar: {
1207*0b57cec5SDimitry Andric     const char format[2] = "c";
12089dba64beSDimitry Andric     return PyArg_Parse(py_return.get(), format, (char *)ret_value);
1209*0b57cec5SDimitry Andric   }
1210*0b57cec5SDimitry Andric   case eScriptReturnTypeOpaqueObject: {
12119dba64beSDimitry Andric     *((PyObject **)ret_value) = py_return.release();
12129dba64beSDimitry Andric     return true;
1213*0b57cec5SDimitry Andric   }
1214*0b57cec5SDimitry Andric   }
1215480093f4SDimitry Andric   llvm_unreachable("Fully covered switch!");
1216*0b57cec5SDimitry Andric }
1217*0b57cec5SDimitry Andric 
1218*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::ExecuteMultipleLines(
1219*0b57cec5SDimitry Andric     const char *in_string, const ExecuteScriptOptions &options) {
12209dba64beSDimitry Andric 
12219dba64beSDimitry Andric   if (in_string == nullptr)
12229dba64beSDimitry Andric     return Status();
1223*0b57cec5SDimitry Andric 
1224fe6060f1SDimitry Andric   llvm::Expected<std::unique_ptr<ScriptInterpreterIORedirect>>
1225fe6060f1SDimitry Andric       io_redirect_or_error = ScriptInterpreterIORedirect::Create(
1226fe6060f1SDimitry Andric           options.GetEnableIO(), m_debugger, /*result=*/nullptr);
1227fe6060f1SDimitry Andric 
1228fe6060f1SDimitry Andric   if (!io_redirect_or_error)
1229fe6060f1SDimitry Andric     return Status(io_redirect_or_error.takeError());
1230fe6060f1SDimitry Andric 
1231fe6060f1SDimitry Andric   ScriptInterpreterIORedirect &io_redirect = **io_redirect_or_error;
1232fe6060f1SDimitry Andric 
1233*0b57cec5SDimitry Andric   Locker locker(this,
1234*0b57cec5SDimitry Andric                 Locker::AcquireLock | Locker::InitSession |
1235*0b57cec5SDimitry Andric                     (options.GetSetLLDBGlobals() ? Locker::InitGlobals : 0) |
1236*0b57cec5SDimitry Andric                     Locker::NoSTDIN,
1237fe6060f1SDimitry Andric                 Locker::FreeAcquiredLock | Locker::TearDownSession,
1238fe6060f1SDimitry Andric                 io_redirect.GetInputFile(), io_redirect.GetOutputFile(),
1239fe6060f1SDimitry Andric                 io_redirect.GetErrorFile());
1240*0b57cec5SDimitry Andric 
12419dba64beSDimitry Andric   PythonModule &main_module = GetMainModule();
12429dba64beSDimitry Andric   PythonDictionary globals = main_module.GetDictionary();
1243*0b57cec5SDimitry Andric 
1244*0b57cec5SDimitry Andric   PythonDictionary locals = GetSessionDictionary();
1245*0b57cec5SDimitry Andric   if (!locals.IsValid())
12469dba64beSDimitry Andric     locals = unwrapIgnoringErrors(
12479dba64beSDimitry Andric         As<PythonDictionary>(globals.GetAttribute(m_dictionary_name)));
1248*0b57cec5SDimitry Andric   if (!locals.IsValid())
1249*0b57cec5SDimitry Andric     locals = globals;
1250*0b57cec5SDimitry Andric 
12519dba64beSDimitry Andric   Expected<PythonObject> return_value =
12529dba64beSDimitry Andric       runStringMultiLine(in_string, globals, locals);
1253*0b57cec5SDimitry Andric 
12549dba64beSDimitry Andric   if (!return_value) {
12559dba64beSDimitry Andric     llvm::Error error =
12569dba64beSDimitry Andric         llvm::handleErrors(return_value.takeError(), [&](PythonException &E) {
12579dba64beSDimitry Andric           llvm::Error error = llvm::createStringError(
12589dba64beSDimitry Andric               llvm::inconvertibleErrorCode(), E.ReadBacktrace());
12599dba64beSDimitry Andric           if (!options.GetMaskoutErrors())
12609dba64beSDimitry Andric             E.Restore();
1261*0b57cec5SDimitry Andric           return error;
12629dba64beSDimitry Andric         });
12639dba64beSDimitry Andric     return Status(std::move(error));
12649dba64beSDimitry Andric   }
12659dba64beSDimitry Andric 
12669dba64beSDimitry Andric   return Status();
1267*0b57cec5SDimitry Andric }
1268*0b57cec5SDimitry Andric 
1269*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::CollectDataForBreakpointCommandCallback(
1270fe6060f1SDimitry Andric     std::vector<std::reference_wrapper<BreakpointOptions>> &bp_options_vec,
1271*0b57cec5SDimitry Andric     CommandReturnObject &result) {
1272*0b57cec5SDimitry Andric   m_active_io_handler = eIOHandlerBreakpoint;
1273*0b57cec5SDimitry Andric   m_debugger.GetCommandInterpreter().GetPythonCommandsFromIOHandler(
1274480093f4SDimitry Andric       "    ", *this, &bp_options_vec);
1275*0b57cec5SDimitry Andric }
1276*0b57cec5SDimitry Andric 
1277*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::CollectDataForWatchpointCommandCallback(
1278*0b57cec5SDimitry Andric     WatchpointOptions *wp_options, CommandReturnObject &result) {
1279*0b57cec5SDimitry Andric   m_active_io_handler = eIOHandlerWatchpoint;
1280*0b57cec5SDimitry Andric   m_debugger.GetCommandInterpreter().GetPythonCommandsFromIOHandler(
1281480093f4SDimitry Andric       "    ", *this, wp_options);
1282*0b57cec5SDimitry Andric }
1283*0b57cec5SDimitry Andric 
1284480093f4SDimitry Andric Status ScriptInterpreterPythonImpl::SetBreakpointCommandCallbackFunction(
1285fe6060f1SDimitry Andric     BreakpointOptions &bp_options, const char *function_name,
1286480093f4SDimitry Andric     StructuredData::ObjectSP extra_args_sp) {
1287480093f4SDimitry Andric   Status error;
1288*0b57cec5SDimitry Andric   // For now just cons up a oneliner that calls the provided function.
1289*0b57cec5SDimitry Andric   std::string oneliner("return ");
1290*0b57cec5SDimitry Andric   oneliner += function_name;
1291480093f4SDimitry Andric 
1292480093f4SDimitry Andric   llvm::Expected<unsigned> maybe_args =
1293480093f4SDimitry Andric       GetMaxPositionalArgumentsForCallable(function_name);
1294480093f4SDimitry Andric   if (!maybe_args) {
1295480093f4SDimitry Andric     error.SetErrorStringWithFormat(
1296480093f4SDimitry Andric         "could not get num args: %s",
1297480093f4SDimitry Andric         llvm::toString(maybe_args.takeError()).c_str());
1298480093f4SDimitry Andric     return error;
1299480093f4SDimitry Andric   }
1300480093f4SDimitry Andric   size_t max_args = *maybe_args;
1301480093f4SDimitry Andric 
1302480093f4SDimitry Andric   bool uses_extra_args = false;
1303480093f4SDimitry Andric   if (max_args >= 4) {
1304480093f4SDimitry Andric     uses_extra_args = true;
1305480093f4SDimitry Andric     oneliner += "(frame, bp_loc, extra_args, internal_dict)";
1306480093f4SDimitry Andric   } else if (max_args >= 3) {
1307480093f4SDimitry Andric     if (extra_args_sp) {
1308480093f4SDimitry Andric       error.SetErrorString("cannot pass extra_args to a three argument callback"
1309480093f4SDimitry Andric                           );
1310480093f4SDimitry Andric       return error;
1311480093f4SDimitry Andric     }
1312480093f4SDimitry Andric     uses_extra_args = false;
1313*0b57cec5SDimitry Andric     oneliner += "(frame, bp_loc, internal_dict)";
1314480093f4SDimitry Andric   } else {
1315480093f4SDimitry Andric     error.SetErrorStringWithFormat("expected 3 or 4 argument "
1316480093f4SDimitry Andric                                    "function, %s can only take %zu",
1317480093f4SDimitry Andric                                    function_name, max_args);
1318480093f4SDimitry Andric     return error;
1319480093f4SDimitry Andric   }
1320480093f4SDimitry Andric 
1321480093f4SDimitry Andric   SetBreakpointCommandCallback(bp_options, oneliner.c_str(), extra_args_sp,
1322480093f4SDimitry Andric                                uses_extra_args);
1323480093f4SDimitry Andric   return error;
1324*0b57cec5SDimitry Andric }
1325*0b57cec5SDimitry Andric 
1326*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::SetBreakpointCommandCallback(
1327fe6060f1SDimitry Andric     BreakpointOptions &bp_options,
1328*0b57cec5SDimitry Andric     std::unique_ptr<BreakpointOptions::CommandData> &cmd_data_up) {
1329*0b57cec5SDimitry Andric   Status error;
1330*0b57cec5SDimitry Andric   error = GenerateBreakpointCommandCallbackData(cmd_data_up->user_source,
1331480093f4SDimitry Andric                                                 cmd_data_up->script_source,
1332480093f4SDimitry Andric                                                 false);
1333*0b57cec5SDimitry Andric   if (error.Fail()) {
1334*0b57cec5SDimitry Andric     return error;
1335*0b57cec5SDimitry Andric   }
1336*0b57cec5SDimitry Andric   auto baton_sp =
1337*0b57cec5SDimitry Andric       std::make_shared<BreakpointOptions::CommandBaton>(std::move(cmd_data_up));
1338fe6060f1SDimitry Andric   bp_options.SetCallback(
1339*0b57cec5SDimitry Andric       ScriptInterpreterPythonImpl::BreakpointCallbackFunction, baton_sp);
1340*0b57cec5SDimitry Andric   return error;
1341*0b57cec5SDimitry Andric }
1342*0b57cec5SDimitry Andric 
1343*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::SetBreakpointCommandCallback(
1344fe6060f1SDimitry Andric     BreakpointOptions &bp_options, const char *command_body_text) {
1345480093f4SDimitry Andric   return SetBreakpointCommandCallback(bp_options, command_body_text, {},false);
1346480093f4SDimitry Andric }
1347*0b57cec5SDimitry Andric 
1348480093f4SDimitry Andric // Set a Python one-liner as the callback for the breakpoint.
1349480093f4SDimitry Andric Status ScriptInterpreterPythonImpl::SetBreakpointCommandCallback(
1350fe6060f1SDimitry Andric     BreakpointOptions &bp_options, const char *command_body_text,
1351fe6060f1SDimitry Andric     StructuredData::ObjectSP extra_args_sp, bool uses_extra_args) {
1352480093f4SDimitry Andric   auto data_up = std::make_unique<CommandDataPython>(extra_args_sp);
1353*0b57cec5SDimitry Andric   // Split the command_body_text into lines, and pass that to
1354*0b57cec5SDimitry Andric   // GenerateBreakpointCommandCallbackData.  That will wrap the body in an
1355*0b57cec5SDimitry Andric   // auto-generated function, and return the function name in script_source.
1356*0b57cec5SDimitry Andric   // That is what the callback will actually invoke.
1357*0b57cec5SDimitry Andric 
1358*0b57cec5SDimitry Andric   data_up->user_source.SplitIntoLines(command_body_text);
1359*0b57cec5SDimitry Andric   Status error = GenerateBreakpointCommandCallbackData(data_up->user_source,
1360480093f4SDimitry Andric                                                        data_up->script_source,
1361480093f4SDimitry Andric                                                        uses_extra_args);
1362*0b57cec5SDimitry Andric   if (error.Success()) {
1363*0b57cec5SDimitry Andric     auto baton_sp =
1364*0b57cec5SDimitry Andric         std::make_shared<BreakpointOptions::CommandBaton>(std::move(data_up));
1365fe6060f1SDimitry Andric     bp_options.SetCallback(
1366*0b57cec5SDimitry Andric         ScriptInterpreterPythonImpl::BreakpointCallbackFunction, baton_sp);
1367*0b57cec5SDimitry Andric     return error;
13685ffd83dbSDimitry Andric   }
1369*0b57cec5SDimitry Andric   return error;
1370*0b57cec5SDimitry Andric }
1371*0b57cec5SDimitry Andric 
1372*0b57cec5SDimitry Andric // Set a Python one-liner as the callback for the watchpoint.
1373*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::SetWatchpointCommandCallback(
1374*0b57cec5SDimitry Andric     WatchpointOptions *wp_options, const char *oneliner) {
13759dba64beSDimitry Andric   auto data_up = std::make_unique<WatchpointOptions::CommandData>();
1376*0b57cec5SDimitry Andric 
1377*0b57cec5SDimitry Andric   // It's necessary to set both user_source and script_source to the oneliner.
1378*0b57cec5SDimitry Andric   // The former is used to generate callback description (as in watchpoint
1379*0b57cec5SDimitry Andric   // command list) while the latter is used for Python to interpret during the
1380*0b57cec5SDimitry Andric   // actual callback.
1381*0b57cec5SDimitry Andric 
1382*0b57cec5SDimitry Andric   data_up->user_source.AppendString(oneliner);
1383*0b57cec5SDimitry Andric   data_up->script_source.assign(oneliner);
1384*0b57cec5SDimitry Andric 
1385*0b57cec5SDimitry Andric   if (GenerateWatchpointCommandCallbackData(data_up->user_source,
1386*0b57cec5SDimitry Andric                                             data_up->script_source)) {
1387*0b57cec5SDimitry Andric     auto baton_sp =
1388*0b57cec5SDimitry Andric         std::make_shared<WatchpointOptions::CommandBaton>(std::move(data_up));
1389*0b57cec5SDimitry Andric     wp_options->SetCallback(
1390*0b57cec5SDimitry Andric         ScriptInterpreterPythonImpl::WatchpointCallbackFunction, baton_sp);
1391*0b57cec5SDimitry Andric   }
1392*0b57cec5SDimitry Andric 
1393*0b57cec5SDimitry Andric   return;
1394*0b57cec5SDimitry Andric }
1395*0b57cec5SDimitry Andric 
1396*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::ExportFunctionDefinitionToInterpreter(
1397*0b57cec5SDimitry Andric     StringList &function_def) {
1398*0b57cec5SDimitry Andric   // Convert StringList to one long, newline delimited, const char *.
1399*0b57cec5SDimitry Andric   std::string function_def_string(function_def.CopyList());
1400*0b57cec5SDimitry Andric 
1401*0b57cec5SDimitry Andric   Status error = ExecuteMultipleLines(
1402*0b57cec5SDimitry Andric       function_def_string.c_str(),
1403fe6060f1SDimitry Andric       ExecuteScriptOptions().SetEnableIO(false));
1404*0b57cec5SDimitry Andric   return error;
1405*0b57cec5SDimitry Andric }
1406*0b57cec5SDimitry Andric 
1407*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::GenerateFunction(const char *signature,
1408*0b57cec5SDimitry Andric                                                      const StringList &input) {
1409*0b57cec5SDimitry Andric   Status error;
1410*0b57cec5SDimitry Andric   int num_lines = input.GetSize();
1411*0b57cec5SDimitry Andric   if (num_lines == 0) {
1412*0b57cec5SDimitry Andric     error.SetErrorString("No input data.");
1413*0b57cec5SDimitry Andric     return error;
1414*0b57cec5SDimitry Andric   }
1415*0b57cec5SDimitry Andric 
1416*0b57cec5SDimitry Andric   if (!signature || *signature == 0) {
1417*0b57cec5SDimitry Andric     error.SetErrorString("No output function name.");
1418*0b57cec5SDimitry Andric     return error;
1419*0b57cec5SDimitry Andric   }
1420*0b57cec5SDimitry Andric 
1421*0b57cec5SDimitry Andric   StreamString sstr;
1422*0b57cec5SDimitry Andric   StringList auto_generated_function;
1423*0b57cec5SDimitry Andric   auto_generated_function.AppendString(signature);
1424*0b57cec5SDimitry Andric   auto_generated_function.AppendString(
1425*0b57cec5SDimitry Andric       "     global_dict = globals()"); // Grab the global dictionary
1426*0b57cec5SDimitry Andric   auto_generated_function.AppendString(
1427*0b57cec5SDimitry Andric       "     new_keys = internal_dict.keys()"); // Make a list of keys in the
1428*0b57cec5SDimitry Andric                                                // session dict
1429*0b57cec5SDimitry Andric   auto_generated_function.AppendString(
1430*0b57cec5SDimitry Andric       "     old_keys = global_dict.keys()"); // Save list of keys in global dict
1431*0b57cec5SDimitry Andric   auto_generated_function.AppendString(
1432*0b57cec5SDimitry Andric       "     global_dict.update (internal_dict)"); // Add the session dictionary
1433*0b57cec5SDimitry Andric                                                   // to the
1434*0b57cec5SDimitry Andric   // global dictionary.
1435*0b57cec5SDimitry Andric 
1436*0b57cec5SDimitry Andric   // Wrap everything up inside the function, increasing the indentation.
1437*0b57cec5SDimitry Andric 
1438*0b57cec5SDimitry Andric   auto_generated_function.AppendString("     if True:");
1439*0b57cec5SDimitry Andric   for (int i = 0; i < num_lines; ++i) {
1440*0b57cec5SDimitry Andric     sstr.Clear();
1441*0b57cec5SDimitry Andric     sstr.Printf("       %s", input.GetStringAtIndex(i));
1442*0b57cec5SDimitry Andric     auto_generated_function.AppendString(sstr.GetData());
1443*0b57cec5SDimitry Andric   }
1444*0b57cec5SDimitry Andric   auto_generated_function.AppendString(
1445*0b57cec5SDimitry Andric       "     for key in new_keys:"); // Iterate over all the keys from session
1446*0b57cec5SDimitry Andric                                     // dict
1447*0b57cec5SDimitry Andric   auto_generated_function.AppendString(
1448*0b57cec5SDimitry Andric       "         internal_dict[key] = global_dict[key]"); // Update session dict
1449*0b57cec5SDimitry Andric                                                          // values
1450*0b57cec5SDimitry Andric   auto_generated_function.AppendString(
1451*0b57cec5SDimitry Andric       "         if key not in old_keys:"); // If key was not originally in
1452*0b57cec5SDimitry Andric                                            // global dict
1453*0b57cec5SDimitry Andric   auto_generated_function.AppendString(
1454*0b57cec5SDimitry Andric       "             del global_dict[key]"); //  ...then remove key/value from
1455*0b57cec5SDimitry Andric                                             //  global dict
1456*0b57cec5SDimitry Andric 
1457*0b57cec5SDimitry Andric   // Verify that the results are valid Python.
1458*0b57cec5SDimitry Andric 
1459*0b57cec5SDimitry Andric   error = ExportFunctionDefinitionToInterpreter(auto_generated_function);
1460*0b57cec5SDimitry Andric 
1461*0b57cec5SDimitry Andric   return error;
1462*0b57cec5SDimitry Andric }
1463*0b57cec5SDimitry Andric 
1464*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateTypeScriptFunction(
1465*0b57cec5SDimitry Andric     StringList &user_input, std::string &output, const void *name_token) {
1466*0b57cec5SDimitry Andric   static uint32_t num_created_functions = 0;
1467*0b57cec5SDimitry Andric   user_input.RemoveBlankLines();
1468*0b57cec5SDimitry Andric   StreamString sstr;
1469*0b57cec5SDimitry Andric 
1470*0b57cec5SDimitry Andric   // Check to see if we have any data; if not, just return.
1471*0b57cec5SDimitry Andric   if (user_input.GetSize() == 0)
1472*0b57cec5SDimitry Andric     return false;
1473*0b57cec5SDimitry Andric 
1474*0b57cec5SDimitry Andric   // Take what the user wrote, wrap it all up inside one big auto-generated
1475*0b57cec5SDimitry Andric   // Python function, passing in the ValueObject as parameter to the function.
1476*0b57cec5SDimitry Andric 
1477*0b57cec5SDimitry Andric   std::string auto_generated_function_name(
1478*0b57cec5SDimitry Andric       GenerateUniqueName("lldb_autogen_python_type_print_func",
1479*0b57cec5SDimitry Andric                          num_created_functions, name_token));
1480*0b57cec5SDimitry Andric   sstr.Printf("def %s (valobj, internal_dict):",
1481*0b57cec5SDimitry Andric               auto_generated_function_name.c_str());
1482*0b57cec5SDimitry Andric 
1483*0b57cec5SDimitry Andric   if (!GenerateFunction(sstr.GetData(), user_input).Success())
1484*0b57cec5SDimitry Andric     return false;
1485*0b57cec5SDimitry Andric 
1486*0b57cec5SDimitry Andric   // Store the name of the auto-generated function to be called.
1487*0b57cec5SDimitry Andric   output.assign(auto_generated_function_name);
1488*0b57cec5SDimitry Andric   return true;
1489*0b57cec5SDimitry Andric }
1490*0b57cec5SDimitry Andric 
1491*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateScriptAliasFunction(
1492*0b57cec5SDimitry Andric     StringList &user_input, std::string &output) {
1493*0b57cec5SDimitry Andric   static uint32_t num_created_functions = 0;
1494*0b57cec5SDimitry Andric   user_input.RemoveBlankLines();
1495*0b57cec5SDimitry Andric   StreamString sstr;
1496*0b57cec5SDimitry Andric 
1497*0b57cec5SDimitry Andric   // Check to see if we have any data; if not, just return.
1498*0b57cec5SDimitry Andric   if (user_input.GetSize() == 0)
1499*0b57cec5SDimitry Andric     return false;
1500*0b57cec5SDimitry Andric 
1501*0b57cec5SDimitry Andric   std::string auto_generated_function_name(GenerateUniqueName(
1502*0b57cec5SDimitry Andric       "lldb_autogen_python_cmd_alias_func", num_created_functions));
1503*0b57cec5SDimitry Andric 
1504*0b57cec5SDimitry Andric   sstr.Printf("def %s (debugger, args, result, internal_dict):",
1505*0b57cec5SDimitry Andric               auto_generated_function_name.c_str());
1506*0b57cec5SDimitry Andric 
1507*0b57cec5SDimitry Andric   if (!GenerateFunction(sstr.GetData(), user_input).Success())
1508*0b57cec5SDimitry Andric     return false;
1509*0b57cec5SDimitry Andric 
1510*0b57cec5SDimitry Andric   // Store the name of the auto-generated function to be called.
1511*0b57cec5SDimitry Andric   output.assign(auto_generated_function_name);
1512*0b57cec5SDimitry Andric   return true;
1513*0b57cec5SDimitry Andric }
1514*0b57cec5SDimitry Andric 
1515*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateTypeSynthClass(
1516*0b57cec5SDimitry Andric     StringList &user_input, std::string &output, const void *name_token) {
1517*0b57cec5SDimitry Andric   static uint32_t num_created_classes = 0;
1518*0b57cec5SDimitry Andric   user_input.RemoveBlankLines();
1519*0b57cec5SDimitry Andric   int num_lines = user_input.GetSize();
1520*0b57cec5SDimitry Andric   StreamString sstr;
1521*0b57cec5SDimitry Andric 
1522*0b57cec5SDimitry Andric   // Check to see if we have any data; if not, just return.
1523*0b57cec5SDimitry Andric   if (user_input.GetSize() == 0)
1524*0b57cec5SDimitry Andric     return false;
1525*0b57cec5SDimitry Andric 
1526*0b57cec5SDimitry Andric   // Wrap all user input into a Python class
1527*0b57cec5SDimitry Andric 
1528*0b57cec5SDimitry Andric   std::string auto_generated_class_name(GenerateUniqueName(
1529*0b57cec5SDimitry Andric       "lldb_autogen_python_type_synth_class", num_created_classes, name_token));
1530*0b57cec5SDimitry Andric 
1531*0b57cec5SDimitry Andric   StringList auto_generated_class;
1532*0b57cec5SDimitry Andric 
1533*0b57cec5SDimitry Andric   // Create the function name & definition string.
1534*0b57cec5SDimitry Andric 
1535*0b57cec5SDimitry Andric   sstr.Printf("class %s:", auto_generated_class_name.c_str());
1536*0b57cec5SDimitry Andric   auto_generated_class.AppendString(sstr.GetString());
1537*0b57cec5SDimitry Andric 
1538*0b57cec5SDimitry Andric   // Wrap everything up inside the class, increasing the indentation. we don't
1539*0b57cec5SDimitry Andric   // need to play any fancy indentation tricks here because there is no
1540*0b57cec5SDimitry Andric   // surrounding code whose indentation we need to honor
1541*0b57cec5SDimitry Andric   for (int i = 0; i < num_lines; ++i) {
1542*0b57cec5SDimitry Andric     sstr.Clear();
1543*0b57cec5SDimitry Andric     sstr.Printf("     %s", user_input.GetStringAtIndex(i));
1544*0b57cec5SDimitry Andric     auto_generated_class.AppendString(sstr.GetString());
1545*0b57cec5SDimitry Andric   }
1546*0b57cec5SDimitry Andric 
1547*0b57cec5SDimitry Andric   // Verify that the results are valid Python. (even though the method is
1548*0b57cec5SDimitry Andric   // ExportFunctionDefinitionToInterpreter, a class will actually be exported)
1549*0b57cec5SDimitry Andric   // (TODO: rename that method to ExportDefinitionToInterpreter)
1550*0b57cec5SDimitry Andric   if (!ExportFunctionDefinitionToInterpreter(auto_generated_class).Success())
1551*0b57cec5SDimitry Andric     return false;
1552*0b57cec5SDimitry Andric 
1553*0b57cec5SDimitry Andric   // Store the name of the auto-generated class
1554*0b57cec5SDimitry Andric 
1555*0b57cec5SDimitry Andric   output.assign(auto_generated_class_name);
1556*0b57cec5SDimitry Andric   return true;
1557*0b57cec5SDimitry Andric }
1558*0b57cec5SDimitry Andric 
1559*0b57cec5SDimitry Andric StructuredData::GenericSP
1560*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::CreateFrameRecognizer(const char *class_name) {
1561*0b57cec5SDimitry Andric   if (class_name == nullptr || class_name[0] == '\0')
1562*0b57cec5SDimitry Andric     return StructuredData::GenericSP();
1563*0b57cec5SDimitry Andric 
1564*0b57cec5SDimitry Andric   void *ret_val;
1565*0b57cec5SDimitry Andric 
1566*0b57cec5SDimitry Andric   {
1567*0b57cec5SDimitry Andric     Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN,
1568*0b57cec5SDimitry Andric                    Locker::FreeLock);
1569*0b57cec5SDimitry Andric     ret_val = LLDBSWIGPython_CreateFrameRecognizer(class_name,
1570*0b57cec5SDimitry Andric                                                    m_dictionary_name.c_str());
1571*0b57cec5SDimitry Andric   }
1572*0b57cec5SDimitry Andric 
1573*0b57cec5SDimitry Andric   return StructuredData::GenericSP(new StructuredPythonObject(ret_val));
1574*0b57cec5SDimitry Andric }
1575*0b57cec5SDimitry Andric 
1576*0b57cec5SDimitry Andric lldb::ValueObjectListSP ScriptInterpreterPythonImpl::GetRecognizedArguments(
1577*0b57cec5SDimitry Andric     const StructuredData::ObjectSP &os_plugin_object_sp,
1578*0b57cec5SDimitry Andric     lldb::StackFrameSP frame_sp) {
1579*0b57cec5SDimitry Andric   Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock);
1580*0b57cec5SDimitry Andric 
1581*0b57cec5SDimitry Andric   if (!os_plugin_object_sp)
1582*0b57cec5SDimitry Andric     return ValueObjectListSP();
1583*0b57cec5SDimitry Andric 
1584*0b57cec5SDimitry Andric   StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric();
1585*0b57cec5SDimitry Andric   if (!generic)
1586*0b57cec5SDimitry Andric     return nullptr;
1587*0b57cec5SDimitry Andric 
1588*0b57cec5SDimitry Andric   PythonObject implementor(PyRefType::Borrowed,
1589*0b57cec5SDimitry Andric                            (PyObject *)generic->GetValue());
1590*0b57cec5SDimitry Andric 
1591*0b57cec5SDimitry Andric   if (!implementor.IsAllocated())
1592*0b57cec5SDimitry Andric     return ValueObjectListSP();
1593*0b57cec5SDimitry Andric 
1594*0b57cec5SDimitry Andric   PythonObject py_return(PyRefType::Owned,
1595*0b57cec5SDimitry Andric                          (PyObject *)LLDBSwigPython_GetRecognizedArguments(
1596*0b57cec5SDimitry Andric                              implementor.get(), frame_sp));
1597*0b57cec5SDimitry Andric 
1598*0b57cec5SDimitry Andric   // if it fails, print the error but otherwise go on
1599*0b57cec5SDimitry Andric   if (PyErr_Occurred()) {
1600*0b57cec5SDimitry Andric     PyErr_Print();
1601*0b57cec5SDimitry Andric     PyErr_Clear();
1602*0b57cec5SDimitry Andric   }
1603*0b57cec5SDimitry Andric   if (py_return.get()) {
1604*0b57cec5SDimitry Andric     PythonList result_list(PyRefType::Borrowed, py_return.get());
1605*0b57cec5SDimitry Andric     ValueObjectListSP result = ValueObjectListSP(new ValueObjectList());
1606*0b57cec5SDimitry Andric     for (size_t i = 0; i < result_list.GetSize(); i++) {
1607*0b57cec5SDimitry Andric       PyObject *item = result_list.GetItemAtIndex(i).get();
1608*0b57cec5SDimitry Andric       lldb::SBValue *sb_value_ptr =
1609*0b57cec5SDimitry Andric           (lldb::SBValue *)LLDBSWIGPython_CastPyObjectToSBValue(item);
1610*0b57cec5SDimitry Andric       auto valobj_sp = LLDBSWIGPython_GetValueObjectSPFromSBValue(sb_value_ptr);
1611*0b57cec5SDimitry Andric       if (valobj_sp)
1612*0b57cec5SDimitry Andric         result->Append(valobj_sp);
1613*0b57cec5SDimitry Andric     }
1614*0b57cec5SDimitry Andric     return result;
1615*0b57cec5SDimitry Andric   }
1616*0b57cec5SDimitry Andric   return ValueObjectListSP();
1617*0b57cec5SDimitry Andric }
1618*0b57cec5SDimitry Andric 
1619*0b57cec5SDimitry Andric StructuredData::GenericSP
1620*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::OSPlugin_CreatePluginObject(
1621*0b57cec5SDimitry Andric     const char *class_name, lldb::ProcessSP process_sp) {
1622*0b57cec5SDimitry Andric   if (class_name == nullptr || class_name[0] == '\0')
1623*0b57cec5SDimitry Andric     return StructuredData::GenericSP();
1624*0b57cec5SDimitry Andric 
1625*0b57cec5SDimitry Andric   if (!process_sp)
1626*0b57cec5SDimitry Andric     return StructuredData::GenericSP();
1627*0b57cec5SDimitry Andric 
1628*0b57cec5SDimitry Andric   void *ret_val;
1629*0b57cec5SDimitry Andric 
1630*0b57cec5SDimitry Andric   {
1631*0b57cec5SDimitry Andric     Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN,
1632*0b57cec5SDimitry Andric                    Locker::FreeLock);
1633*0b57cec5SDimitry Andric     ret_val = LLDBSWIGPythonCreateOSPlugin(
1634*0b57cec5SDimitry Andric         class_name, m_dictionary_name.c_str(), process_sp);
1635*0b57cec5SDimitry Andric   }
1636*0b57cec5SDimitry Andric 
1637*0b57cec5SDimitry Andric   return StructuredData::GenericSP(new StructuredPythonObject(ret_val));
1638*0b57cec5SDimitry Andric }
1639*0b57cec5SDimitry Andric 
1640*0b57cec5SDimitry Andric StructuredData::DictionarySP ScriptInterpreterPythonImpl::OSPlugin_RegisterInfo(
1641*0b57cec5SDimitry Andric     StructuredData::ObjectSP os_plugin_object_sp) {
1642*0b57cec5SDimitry Andric   Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock);
1643*0b57cec5SDimitry Andric 
1644*0b57cec5SDimitry Andric   static char callee_name[] = "get_register_info";
1645*0b57cec5SDimitry Andric 
1646*0b57cec5SDimitry Andric   if (!os_plugin_object_sp)
1647*0b57cec5SDimitry Andric     return StructuredData::DictionarySP();
1648*0b57cec5SDimitry Andric 
1649*0b57cec5SDimitry Andric   StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric();
1650*0b57cec5SDimitry Andric   if (!generic)
1651*0b57cec5SDimitry Andric     return nullptr;
1652*0b57cec5SDimitry Andric 
1653*0b57cec5SDimitry Andric   PythonObject implementor(PyRefType::Borrowed,
1654*0b57cec5SDimitry Andric                            (PyObject *)generic->GetValue());
1655*0b57cec5SDimitry Andric 
1656*0b57cec5SDimitry Andric   if (!implementor.IsAllocated())
1657*0b57cec5SDimitry Andric     return StructuredData::DictionarySP();
1658*0b57cec5SDimitry Andric 
1659*0b57cec5SDimitry Andric   PythonObject pmeth(PyRefType::Owned,
1660*0b57cec5SDimitry Andric                      PyObject_GetAttrString(implementor.get(), callee_name));
1661*0b57cec5SDimitry Andric 
1662*0b57cec5SDimitry Andric   if (PyErr_Occurred())
1663*0b57cec5SDimitry Andric     PyErr_Clear();
1664*0b57cec5SDimitry Andric 
1665*0b57cec5SDimitry Andric   if (!pmeth.IsAllocated())
1666*0b57cec5SDimitry Andric     return StructuredData::DictionarySP();
1667*0b57cec5SDimitry Andric 
1668*0b57cec5SDimitry Andric   if (PyCallable_Check(pmeth.get()) == 0) {
1669*0b57cec5SDimitry Andric     if (PyErr_Occurred())
1670*0b57cec5SDimitry Andric       PyErr_Clear();
1671*0b57cec5SDimitry Andric 
1672*0b57cec5SDimitry Andric     return StructuredData::DictionarySP();
1673*0b57cec5SDimitry Andric   }
1674*0b57cec5SDimitry Andric 
1675*0b57cec5SDimitry Andric   if (PyErr_Occurred())
1676*0b57cec5SDimitry Andric     PyErr_Clear();
1677*0b57cec5SDimitry Andric 
1678*0b57cec5SDimitry Andric   // right now we know this function exists and is callable..
1679*0b57cec5SDimitry Andric   PythonObject py_return(
1680*0b57cec5SDimitry Andric       PyRefType::Owned,
1681*0b57cec5SDimitry Andric       PyObject_CallMethod(implementor.get(), callee_name, nullptr));
1682*0b57cec5SDimitry Andric 
1683*0b57cec5SDimitry Andric   // if it fails, print the error but otherwise go on
1684*0b57cec5SDimitry Andric   if (PyErr_Occurred()) {
1685*0b57cec5SDimitry Andric     PyErr_Print();
1686*0b57cec5SDimitry Andric     PyErr_Clear();
1687*0b57cec5SDimitry Andric   }
1688*0b57cec5SDimitry Andric   if (py_return.get()) {
1689*0b57cec5SDimitry Andric     PythonDictionary result_dict(PyRefType::Borrowed, py_return.get());
1690*0b57cec5SDimitry Andric     return result_dict.CreateStructuredDictionary();
1691*0b57cec5SDimitry Andric   }
1692*0b57cec5SDimitry Andric   return StructuredData::DictionarySP();
1693*0b57cec5SDimitry Andric }
1694*0b57cec5SDimitry Andric 
1695*0b57cec5SDimitry Andric StructuredData::ArraySP ScriptInterpreterPythonImpl::OSPlugin_ThreadsInfo(
1696*0b57cec5SDimitry Andric     StructuredData::ObjectSP os_plugin_object_sp) {
1697*0b57cec5SDimitry Andric   Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock);
1698*0b57cec5SDimitry Andric 
1699*0b57cec5SDimitry Andric   static char callee_name[] = "get_thread_info";
1700*0b57cec5SDimitry Andric 
1701*0b57cec5SDimitry Andric   if (!os_plugin_object_sp)
1702*0b57cec5SDimitry Andric     return StructuredData::ArraySP();
1703*0b57cec5SDimitry Andric 
1704*0b57cec5SDimitry Andric   StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric();
1705*0b57cec5SDimitry Andric   if (!generic)
1706*0b57cec5SDimitry Andric     return nullptr;
1707*0b57cec5SDimitry Andric 
1708*0b57cec5SDimitry Andric   PythonObject implementor(PyRefType::Borrowed,
1709*0b57cec5SDimitry Andric                            (PyObject *)generic->GetValue());
1710*0b57cec5SDimitry Andric 
1711*0b57cec5SDimitry Andric   if (!implementor.IsAllocated())
1712*0b57cec5SDimitry Andric     return StructuredData::ArraySP();
1713*0b57cec5SDimitry Andric 
1714*0b57cec5SDimitry Andric   PythonObject pmeth(PyRefType::Owned,
1715*0b57cec5SDimitry Andric                      PyObject_GetAttrString(implementor.get(), callee_name));
1716*0b57cec5SDimitry Andric 
1717*0b57cec5SDimitry Andric   if (PyErr_Occurred())
1718*0b57cec5SDimitry Andric     PyErr_Clear();
1719*0b57cec5SDimitry Andric 
1720*0b57cec5SDimitry Andric   if (!pmeth.IsAllocated())
1721*0b57cec5SDimitry Andric     return StructuredData::ArraySP();
1722*0b57cec5SDimitry Andric 
1723*0b57cec5SDimitry Andric   if (PyCallable_Check(pmeth.get()) == 0) {
1724*0b57cec5SDimitry Andric     if (PyErr_Occurred())
1725*0b57cec5SDimitry Andric       PyErr_Clear();
1726*0b57cec5SDimitry Andric 
1727*0b57cec5SDimitry Andric     return StructuredData::ArraySP();
1728*0b57cec5SDimitry Andric   }
1729*0b57cec5SDimitry Andric 
1730*0b57cec5SDimitry Andric   if (PyErr_Occurred())
1731*0b57cec5SDimitry Andric     PyErr_Clear();
1732*0b57cec5SDimitry Andric 
1733*0b57cec5SDimitry Andric   // right now we know this function exists and is callable..
1734*0b57cec5SDimitry Andric   PythonObject py_return(
1735*0b57cec5SDimitry Andric       PyRefType::Owned,
1736*0b57cec5SDimitry Andric       PyObject_CallMethod(implementor.get(), callee_name, nullptr));
1737*0b57cec5SDimitry Andric 
1738*0b57cec5SDimitry Andric   // if it fails, print the error but otherwise go on
1739*0b57cec5SDimitry Andric   if (PyErr_Occurred()) {
1740*0b57cec5SDimitry Andric     PyErr_Print();
1741*0b57cec5SDimitry Andric     PyErr_Clear();
1742*0b57cec5SDimitry Andric   }
1743*0b57cec5SDimitry Andric 
1744*0b57cec5SDimitry Andric   if (py_return.get()) {
1745*0b57cec5SDimitry Andric     PythonList result_list(PyRefType::Borrowed, py_return.get());
1746*0b57cec5SDimitry Andric     return result_list.CreateStructuredArray();
1747*0b57cec5SDimitry Andric   }
1748*0b57cec5SDimitry Andric   return StructuredData::ArraySP();
1749*0b57cec5SDimitry Andric }
1750*0b57cec5SDimitry Andric 
1751*0b57cec5SDimitry Andric StructuredData::StringSP
1752*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::OSPlugin_RegisterContextData(
1753*0b57cec5SDimitry Andric     StructuredData::ObjectSP os_plugin_object_sp, lldb::tid_t tid) {
1754*0b57cec5SDimitry Andric   Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock);
1755*0b57cec5SDimitry Andric 
1756*0b57cec5SDimitry Andric   static char callee_name[] = "get_register_data";
1757*0b57cec5SDimitry Andric   static char *param_format =
1758*0b57cec5SDimitry Andric       const_cast<char *>(GetPythonValueFormatString(tid));
1759*0b57cec5SDimitry Andric 
1760*0b57cec5SDimitry Andric   if (!os_plugin_object_sp)
1761*0b57cec5SDimitry Andric     return StructuredData::StringSP();
1762*0b57cec5SDimitry Andric 
1763*0b57cec5SDimitry Andric   StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric();
1764*0b57cec5SDimitry Andric   if (!generic)
1765*0b57cec5SDimitry Andric     return nullptr;
1766*0b57cec5SDimitry Andric   PythonObject implementor(PyRefType::Borrowed,
1767*0b57cec5SDimitry Andric                            (PyObject *)generic->GetValue());
1768*0b57cec5SDimitry Andric 
1769*0b57cec5SDimitry Andric   if (!implementor.IsAllocated())
1770*0b57cec5SDimitry Andric     return StructuredData::StringSP();
1771*0b57cec5SDimitry Andric 
1772*0b57cec5SDimitry Andric   PythonObject pmeth(PyRefType::Owned,
1773*0b57cec5SDimitry Andric                      PyObject_GetAttrString(implementor.get(), callee_name));
1774*0b57cec5SDimitry Andric 
1775*0b57cec5SDimitry Andric   if (PyErr_Occurred())
1776*0b57cec5SDimitry Andric     PyErr_Clear();
1777*0b57cec5SDimitry Andric 
1778*0b57cec5SDimitry Andric   if (!pmeth.IsAllocated())
1779*0b57cec5SDimitry Andric     return StructuredData::StringSP();
1780*0b57cec5SDimitry Andric 
1781*0b57cec5SDimitry Andric   if (PyCallable_Check(pmeth.get()) == 0) {
1782*0b57cec5SDimitry Andric     if (PyErr_Occurred())
1783*0b57cec5SDimitry Andric       PyErr_Clear();
1784*0b57cec5SDimitry Andric     return StructuredData::StringSP();
1785*0b57cec5SDimitry Andric   }
1786*0b57cec5SDimitry Andric 
1787*0b57cec5SDimitry Andric   if (PyErr_Occurred())
1788*0b57cec5SDimitry Andric     PyErr_Clear();
1789*0b57cec5SDimitry Andric 
1790*0b57cec5SDimitry Andric   // right now we know this function exists and is callable..
1791*0b57cec5SDimitry Andric   PythonObject py_return(
1792*0b57cec5SDimitry Andric       PyRefType::Owned,
1793*0b57cec5SDimitry Andric       PyObject_CallMethod(implementor.get(), callee_name, param_format, tid));
1794*0b57cec5SDimitry Andric 
1795*0b57cec5SDimitry Andric   // if it fails, print the error but otherwise go on
1796*0b57cec5SDimitry Andric   if (PyErr_Occurred()) {
1797*0b57cec5SDimitry Andric     PyErr_Print();
1798*0b57cec5SDimitry Andric     PyErr_Clear();
1799*0b57cec5SDimitry Andric   }
1800*0b57cec5SDimitry Andric 
1801*0b57cec5SDimitry Andric   if (py_return.get()) {
1802*0b57cec5SDimitry Andric     PythonBytes result(PyRefType::Borrowed, py_return.get());
1803*0b57cec5SDimitry Andric     return result.CreateStructuredString();
1804*0b57cec5SDimitry Andric   }
1805*0b57cec5SDimitry Andric   return StructuredData::StringSP();
1806*0b57cec5SDimitry Andric }
1807*0b57cec5SDimitry Andric 
1808*0b57cec5SDimitry Andric StructuredData::DictionarySP ScriptInterpreterPythonImpl::OSPlugin_CreateThread(
1809*0b57cec5SDimitry Andric     StructuredData::ObjectSP os_plugin_object_sp, lldb::tid_t tid,
1810*0b57cec5SDimitry Andric     lldb::addr_t context) {
1811*0b57cec5SDimitry Andric   Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock);
1812*0b57cec5SDimitry Andric 
1813*0b57cec5SDimitry Andric   static char callee_name[] = "create_thread";
1814*0b57cec5SDimitry Andric   std::string param_format;
1815*0b57cec5SDimitry Andric   param_format += GetPythonValueFormatString(tid);
1816*0b57cec5SDimitry Andric   param_format += GetPythonValueFormatString(context);
1817*0b57cec5SDimitry Andric 
1818*0b57cec5SDimitry Andric   if (!os_plugin_object_sp)
1819*0b57cec5SDimitry Andric     return StructuredData::DictionarySP();
1820*0b57cec5SDimitry Andric 
1821*0b57cec5SDimitry Andric   StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric();
1822*0b57cec5SDimitry Andric   if (!generic)
1823*0b57cec5SDimitry Andric     return nullptr;
1824*0b57cec5SDimitry Andric 
1825*0b57cec5SDimitry Andric   PythonObject implementor(PyRefType::Borrowed,
1826*0b57cec5SDimitry Andric                            (PyObject *)generic->GetValue());
1827*0b57cec5SDimitry Andric 
1828*0b57cec5SDimitry Andric   if (!implementor.IsAllocated())
1829*0b57cec5SDimitry Andric     return StructuredData::DictionarySP();
1830*0b57cec5SDimitry Andric 
1831*0b57cec5SDimitry Andric   PythonObject pmeth(PyRefType::Owned,
1832*0b57cec5SDimitry Andric                      PyObject_GetAttrString(implementor.get(), callee_name));
1833*0b57cec5SDimitry Andric 
1834*0b57cec5SDimitry Andric   if (PyErr_Occurred())
1835*0b57cec5SDimitry Andric     PyErr_Clear();
1836*0b57cec5SDimitry Andric 
1837*0b57cec5SDimitry Andric   if (!pmeth.IsAllocated())
1838*0b57cec5SDimitry Andric     return StructuredData::DictionarySP();
1839*0b57cec5SDimitry Andric 
1840*0b57cec5SDimitry Andric   if (PyCallable_Check(pmeth.get()) == 0) {
1841*0b57cec5SDimitry Andric     if (PyErr_Occurred())
1842*0b57cec5SDimitry Andric       PyErr_Clear();
1843*0b57cec5SDimitry Andric     return StructuredData::DictionarySP();
1844*0b57cec5SDimitry Andric   }
1845*0b57cec5SDimitry Andric 
1846*0b57cec5SDimitry Andric   if (PyErr_Occurred())
1847*0b57cec5SDimitry Andric     PyErr_Clear();
1848*0b57cec5SDimitry Andric 
1849*0b57cec5SDimitry Andric   // right now we know this function exists and is callable..
1850*0b57cec5SDimitry Andric   PythonObject py_return(PyRefType::Owned,
1851*0b57cec5SDimitry Andric                          PyObject_CallMethod(implementor.get(), callee_name,
1852*0b57cec5SDimitry Andric                                              &param_format[0], tid, context));
1853*0b57cec5SDimitry Andric 
1854*0b57cec5SDimitry Andric   // if it fails, print the error but otherwise go on
1855*0b57cec5SDimitry Andric   if (PyErr_Occurred()) {
1856*0b57cec5SDimitry Andric     PyErr_Print();
1857*0b57cec5SDimitry Andric     PyErr_Clear();
1858*0b57cec5SDimitry Andric   }
1859*0b57cec5SDimitry Andric 
1860*0b57cec5SDimitry Andric   if (py_return.get()) {
1861*0b57cec5SDimitry Andric     PythonDictionary result_dict(PyRefType::Borrowed, py_return.get());
1862*0b57cec5SDimitry Andric     return result_dict.CreateStructuredDictionary();
1863*0b57cec5SDimitry Andric   }
1864*0b57cec5SDimitry Andric   return StructuredData::DictionarySP();
1865*0b57cec5SDimitry Andric }
1866*0b57cec5SDimitry Andric 
1867*0b57cec5SDimitry Andric StructuredData::ObjectSP ScriptInterpreterPythonImpl::CreateScriptedThreadPlan(
18689dba64beSDimitry Andric     const char *class_name, StructuredDataImpl *args_data,
1869480093f4SDimitry Andric     std::string &error_str, lldb::ThreadPlanSP thread_plan_sp) {
1870*0b57cec5SDimitry Andric   if (class_name == nullptr || class_name[0] == '\0')
1871*0b57cec5SDimitry Andric     return StructuredData::ObjectSP();
1872*0b57cec5SDimitry Andric 
1873*0b57cec5SDimitry Andric   if (!thread_plan_sp.get())
18749dba64beSDimitry Andric     return {};
1875*0b57cec5SDimitry Andric 
1876*0b57cec5SDimitry Andric   Debugger &debugger = thread_plan_sp->GetTarget().GetDebugger();
1877*0b57cec5SDimitry Andric   ScriptInterpreterPythonImpl *python_interpreter =
1878e8d8bef9SDimitry Andric       GetPythonInterpreter(debugger);
1879*0b57cec5SDimitry Andric 
1880e8d8bef9SDimitry Andric   if (!python_interpreter)
18819dba64beSDimitry Andric     return {};
1882*0b57cec5SDimitry Andric 
1883*0b57cec5SDimitry Andric   void *ret_val;
1884*0b57cec5SDimitry Andric 
1885*0b57cec5SDimitry Andric   {
1886*0b57cec5SDimitry Andric     Locker py_lock(this,
1887*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
1888*0b57cec5SDimitry Andric     ret_val = LLDBSwigPythonCreateScriptedThreadPlan(
1889*0b57cec5SDimitry Andric         class_name, python_interpreter->m_dictionary_name.c_str(),
18909dba64beSDimitry Andric         args_data, error_str, thread_plan_sp);
18919dba64beSDimitry Andric     if (!ret_val)
18929dba64beSDimitry Andric       return {};
1893*0b57cec5SDimitry Andric   }
1894*0b57cec5SDimitry Andric 
1895*0b57cec5SDimitry Andric   return StructuredData::ObjectSP(new StructuredPythonObject(ret_val));
1896*0b57cec5SDimitry Andric }
1897*0b57cec5SDimitry Andric 
1898*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ScriptedThreadPlanExplainsStop(
1899*0b57cec5SDimitry Andric     StructuredData::ObjectSP implementor_sp, Event *event, bool &script_error) {
1900*0b57cec5SDimitry Andric   bool explains_stop = true;
1901*0b57cec5SDimitry Andric   StructuredData::Generic *generic = nullptr;
1902*0b57cec5SDimitry Andric   if (implementor_sp)
1903*0b57cec5SDimitry Andric     generic = implementor_sp->GetAsGeneric();
1904*0b57cec5SDimitry Andric   if (generic) {
1905*0b57cec5SDimitry Andric     Locker py_lock(this,
1906*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
1907*0b57cec5SDimitry Andric     explains_stop = LLDBSWIGPythonCallThreadPlan(
1908*0b57cec5SDimitry Andric         generic->GetValue(), "explains_stop", event, script_error);
1909*0b57cec5SDimitry Andric     if (script_error)
1910*0b57cec5SDimitry Andric       return true;
1911*0b57cec5SDimitry Andric   }
1912*0b57cec5SDimitry Andric   return explains_stop;
1913*0b57cec5SDimitry Andric }
1914*0b57cec5SDimitry Andric 
1915*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ScriptedThreadPlanShouldStop(
1916*0b57cec5SDimitry Andric     StructuredData::ObjectSP implementor_sp, Event *event, bool &script_error) {
1917*0b57cec5SDimitry Andric   bool should_stop = true;
1918*0b57cec5SDimitry Andric   StructuredData::Generic *generic = nullptr;
1919*0b57cec5SDimitry Andric   if (implementor_sp)
1920*0b57cec5SDimitry Andric     generic = implementor_sp->GetAsGeneric();
1921*0b57cec5SDimitry Andric   if (generic) {
1922*0b57cec5SDimitry Andric     Locker py_lock(this,
1923*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
1924*0b57cec5SDimitry Andric     should_stop = LLDBSWIGPythonCallThreadPlan(
1925*0b57cec5SDimitry Andric         generic->GetValue(), "should_stop", event, script_error);
1926*0b57cec5SDimitry Andric     if (script_error)
1927*0b57cec5SDimitry Andric       return true;
1928*0b57cec5SDimitry Andric   }
1929*0b57cec5SDimitry Andric   return should_stop;
1930*0b57cec5SDimitry Andric }
1931*0b57cec5SDimitry Andric 
1932*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ScriptedThreadPlanIsStale(
1933*0b57cec5SDimitry Andric     StructuredData::ObjectSP implementor_sp, bool &script_error) {
1934*0b57cec5SDimitry Andric   bool is_stale = true;
1935*0b57cec5SDimitry Andric   StructuredData::Generic *generic = nullptr;
1936*0b57cec5SDimitry Andric   if (implementor_sp)
1937*0b57cec5SDimitry Andric     generic = implementor_sp->GetAsGeneric();
1938*0b57cec5SDimitry Andric   if (generic) {
1939*0b57cec5SDimitry Andric     Locker py_lock(this,
1940*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
1941*0b57cec5SDimitry Andric     is_stale = LLDBSWIGPythonCallThreadPlan(generic->GetValue(), "is_stale",
1942*0b57cec5SDimitry Andric                                             nullptr, script_error);
1943*0b57cec5SDimitry Andric     if (script_error)
1944*0b57cec5SDimitry Andric       return true;
1945*0b57cec5SDimitry Andric   }
1946*0b57cec5SDimitry Andric   return is_stale;
1947*0b57cec5SDimitry Andric }
1948*0b57cec5SDimitry Andric 
1949*0b57cec5SDimitry Andric lldb::StateType ScriptInterpreterPythonImpl::ScriptedThreadPlanGetRunState(
1950*0b57cec5SDimitry Andric     StructuredData::ObjectSP implementor_sp, bool &script_error) {
1951*0b57cec5SDimitry Andric   bool should_step = false;
1952*0b57cec5SDimitry Andric   StructuredData::Generic *generic = nullptr;
1953*0b57cec5SDimitry Andric   if (implementor_sp)
1954*0b57cec5SDimitry Andric     generic = implementor_sp->GetAsGeneric();
1955*0b57cec5SDimitry Andric   if (generic) {
1956*0b57cec5SDimitry Andric     Locker py_lock(this,
1957*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
1958*0b57cec5SDimitry Andric     should_step = LLDBSWIGPythonCallThreadPlan(
1959*0b57cec5SDimitry Andric         generic->GetValue(), "should_step", nullptr, script_error);
1960*0b57cec5SDimitry Andric     if (script_error)
1961*0b57cec5SDimitry Andric       should_step = true;
1962*0b57cec5SDimitry Andric   }
1963*0b57cec5SDimitry Andric   if (should_step)
1964*0b57cec5SDimitry Andric     return lldb::eStateStepping;
1965*0b57cec5SDimitry Andric   return lldb::eStateRunning;
1966*0b57cec5SDimitry Andric }
1967*0b57cec5SDimitry Andric 
1968*0b57cec5SDimitry Andric StructuredData::GenericSP
1969*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::CreateScriptedBreakpointResolver(
1970*0b57cec5SDimitry Andric     const char *class_name, StructuredDataImpl *args_data,
1971*0b57cec5SDimitry Andric     lldb::BreakpointSP &bkpt_sp) {
1972*0b57cec5SDimitry Andric 
1973*0b57cec5SDimitry Andric   if (class_name == nullptr || class_name[0] == '\0')
1974*0b57cec5SDimitry Andric     return StructuredData::GenericSP();
1975*0b57cec5SDimitry Andric 
1976*0b57cec5SDimitry Andric   if (!bkpt_sp.get())
1977*0b57cec5SDimitry Andric     return StructuredData::GenericSP();
1978*0b57cec5SDimitry Andric 
1979*0b57cec5SDimitry Andric   Debugger &debugger = bkpt_sp->GetTarget().GetDebugger();
1980*0b57cec5SDimitry Andric   ScriptInterpreterPythonImpl *python_interpreter =
1981e8d8bef9SDimitry Andric       GetPythonInterpreter(debugger);
1982*0b57cec5SDimitry Andric 
1983e8d8bef9SDimitry Andric   if (!python_interpreter)
1984*0b57cec5SDimitry Andric     return StructuredData::GenericSP();
1985*0b57cec5SDimitry Andric 
1986*0b57cec5SDimitry Andric   void *ret_val;
1987*0b57cec5SDimitry Andric 
1988*0b57cec5SDimitry Andric   {
1989*0b57cec5SDimitry Andric     Locker py_lock(this,
1990*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
1991*0b57cec5SDimitry Andric 
1992*0b57cec5SDimitry Andric     ret_val = LLDBSwigPythonCreateScriptedBreakpointResolver(
1993*0b57cec5SDimitry Andric         class_name, python_interpreter->m_dictionary_name.c_str(), args_data,
1994*0b57cec5SDimitry Andric         bkpt_sp);
1995*0b57cec5SDimitry Andric   }
1996*0b57cec5SDimitry Andric 
1997*0b57cec5SDimitry Andric   return StructuredData::GenericSP(new StructuredPythonObject(ret_val));
1998*0b57cec5SDimitry Andric }
1999*0b57cec5SDimitry Andric 
2000*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ScriptedBreakpointResolverSearchCallback(
2001*0b57cec5SDimitry Andric     StructuredData::GenericSP implementor_sp, SymbolContext *sym_ctx) {
2002*0b57cec5SDimitry Andric   bool should_continue = false;
2003*0b57cec5SDimitry Andric 
2004*0b57cec5SDimitry Andric   if (implementor_sp) {
2005*0b57cec5SDimitry Andric     Locker py_lock(this,
2006*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2007*0b57cec5SDimitry Andric     should_continue = LLDBSwigPythonCallBreakpointResolver(
2008*0b57cec5SDimitry Andric         implementor_sp->GetValue(), "__callback__", sym_ctx);
2009*0b57cec5SDimitry Andric     if (PyErr_Occurred()) {
2010*0b57cec5SDimitry Andric       PyErr_Print();
2011*0b57cec5SDimitry Andric       PyErr_Clear();
2012*0b57cec5SDimitry Andric     }
2013*0b57cec5SDimitry Andric   }
2014*0b57cec5SDimitry Andric   return should_continue;
2015*0b57cec5SDimitry Andric }
2016*0b57cec5SDimitry Andric 
2017*0b57cec5SDimitry Andric lldb::SearchDepth
2018*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::ScriptedBreakpointResolverSearchDepth(
2019*0b57cec5SDimitry Andric     StructuredData::GenericSP implementor_sp) {
2020*0b57cec5SDimitry Andric   int depth_as_int = lldb::eSearchDepthModule;
2021*0b57cec5SDimitry Andric   if (implementor_sp) {
2022*0b57cec5SDimitry Andric     Locker py_lock(this,
2023*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2024*0b57cec5SDimitry Andric     depth_as_int = LLDBSwigPythonCallBreakpointResolver(
2025*0b57cec5SDimitry Andric         implementor_sp->GetValue(), "__get_depth__", nullptr);
2026*0b57cec5SDimitry Andric     if (PyErr_Occurred()) {
2027*0b57cec5SDimitry Andric       PyErr_Print();
2028*0b57cec5SDimitry Andric       PyErr_Clear();
2029*0b57cec5SDimitry Andric     }
2030*0b57cec5SDimitry Andric   }
2031*0b57cec5SDimitry Andric   if (depth_as_int == lldb::eSearchDepthInvalid)
2032*0b57cec5SDimitry Andric     return lldb::eSearchDepthModule;
2033*0b57cec5SDimitry Andric 
2034*0b57cec5SDimitry Andric   if (depth_as_int <= lldb::kLastSearchDepthKind)
2035*0b57cec5SDimitry Andric     return (lldb::SearchDepth)depth_as_int;
2036*0b57cec5SDimitry Andric   return lldb::eSearchDepthModule;
2037*0b57cec5SDimitry Andric }
2038*0b57cec5SDimitry Andric 
2039e8d8bef9SDimitry Andric StructuredData::GenericSP ScriptInterpreterPythonImpl::CreateScriptedStopHook(
2040e8d8bef9SDimitry Andric     TargetSP target_sp, const char *class_name, StructuredDataImpl *args_data,
2041e8d8bef9SDimitry Andric     Status &error) {
2042e8d8bef9SDimitry Andric 
2043e8d8bef9SDimitry Andric   if (!target_sp) {
2044e8d8bef9SDimitry Andric     error.SetErrorString("No target for scripted stop-hook.");
2045e8d8bef9SDimitry Andric     return StructuredData::GenericSP();
2046e8d8bef9SDimitry Andric   }
2047e8d8bef9SDimitry Andric 
2048e8d8bef9SDimitry Andric   if (class_name == nullptr || class_name[0] == '\0') {
2049e8d8bef9SDimitry Andric     error.SetErrorString("No class name for scripted stop-hook.");
2050e8d8bef9SDimitry Andric     return StructuredData::GenericSP();
2051e8d8bef9SDimitry Andric   }
2052e8d8bef9SDimitry Andric 
2053e8d8bef9SDimitry Andric   ScriptInterpreterPythonImpl *python_interpreter =
2054e8d8bef9SDimitry Andric       GetPythonInterpreter(m_debugger);
2055e8d8bef9SDimitry Andric 
2056e8d8bef9SDimitry Andric   if (!python_interpreter) {
2057e8d8bef9SDimitry Andric     error.SetErrorString("No script interpreter for scripted stop-hook.");
2058e8d8bef9SDimitry Andric     return StructuredData::GenericSP();
2059e8d8bef9SDimitry Andric   }
2060e8d8bef9SDimitry Andric 
2061e8d8bef9SDimitry Andric   void *ret_val;
2062e8d8bef9SDimitry Andric 
2063e8d8bef9SDimitry Andric   {
2064e8d8bef9SDimitry Andric     Locker py_lock(this,
2065e8d8bef9SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2066e8d8bef9SDimitry Andric 
2067e8d8bef9SDimitry Andric     ret_val = LLDBSwigPythonCreateScriptedStopHook(
2068e8d8bef9SDimitry Andric         target_sp, class_name, python_interpreter->m_dictionary_name.c_str(),
2069e8d8bef9SDimitry Andric         args_data, error);
2070e8d8bef9SDimitry Andric   }
2071e8d8bef9SDimitry Andric 
2072e8d8bef9SDimitry Andric   return StructuredData::GenericSP(new StructuredPythonObject(ret_val));
2073e8d8bef9SDimitry Andric }
2074e8d8bef9SDimitry Andric 
2075e8d8bef9SDimitry Andric bool ScriptInterpreterPythonImpl::ScriptedStopHookHandleStop(
2076e8d8bef9SDimitry Andric     StructuredData::GenericSP implementor_sp, ExecutionContext &exc_ctx,
2077e8d8bef9SDimitry Andric     lldb::StreamSP stream_sp) {
2078e8d8bef9SDimitry Andric   assert(implementor_sp &&
2079e8d8bef9SDimitry Andric          "can't call a stop hook with an invalid implementor");
2080e8d8bef9SDimitry Andric   assert(stream_sp && "can't call a stop hook with an invalid stream");
2081e8d8bef9SDimitry Andric 
2082e8d8bef9SDimitry Andric   Locker py_lock(this,
2083e8d8bef9SDimitry Andric                  Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2084e8d8bef9SDimitry Andric 
2085e8d8bef9SDimitry Andric   lldb::ExecutionContextRefSP exc_ctx_ref_sp(new ExecutionContextRef(exc_ctx));
2086e8d8bef9SDimitry Andric 
2087e8d8bef9SDimitry Andric   bool ret_val = LLDBSwigPythonStopHookCallHandleStop(
2088e8d8bef9SDimitry Andric       implementor_sp->GetValue(), exc_ctx_ref_sp, stream_sp);
2089e8d8bef9SDimitry Andric   return ret_val;
2090e8d8bef9SDimitry Andric }
2091e8d8bef9SDimitry Andric 
2092*0b57cec5SDimitry Andric StructuredData::ObjectSP
2093*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::LoadPluginModule(const FileSpec &file_spec,
2094*0b57cec5SDimitry Andric                                               lldb_private::Status &error) {
2095*0b57cec5SDimitry Andric   if (!FileSystem::Instance().Exists(file_spec)) {
2096*0b57cec5SDimitry Andric     error.SetErrorString("no such file");
2097*0b57cec5SDimitry Andric     return StructuredData::ObjectSP();
2098*0b57cec5SDimitry Andric   }
2099*0b57cec5SDimitry Andric 
2100*0b57cec5SDimitry Andric   StructuredData::ObjectSP module_sp;
2101*0b57cec5SDimitry Andric 
2102fe6060f1SDimitry Andric   LoadScriptOptions load_script_options =
2103fe6060f1SDimitry Andric       LoadScriptOptions().SetInitSession(true).SetSilent(false);
2104fe6060f1SDimitry Andric   if (LoadScriptingModule(file_spec.GetPath().c_str(), load_script_options,
2105fe6060f1SDimitry Andric                           error, &module_sp))
2106*0b57cec5SDimitry Andric     return module_sp;
2107*0b57cec5SDimitry Andric 
2108*0b57cec5SDimitry Andric   return StructuredData::ObjectSP();
2109*0b57cec5SDimitry Andric }
2110*0b57cec5SDimitry Andric 
2111*0b57cec5SDimitry Andric StructuredData::DictionarySP ScriptInterpreterPythonImpl::GetDynamicSettings(
2112*0b57cec5SDimitry Andric     StructuredData::ObjectSP plugin_module_sp, Target *target,
2113*0b57cec5SDimitry Andric     const char *setting_name, lldb_private::Status &error) {
2114*0b57cec5SDimitry Andric   if (!plugin_module_sp || !target || !setting_name || !setting_name[0])
2115*0b57cec5SDimitry Andric     return StructuredData::DictionarySP();
2116*0b57cec5SDimitry Andric   StructuredData::Generic *generic = plugin_module_sp->GetAsGeneric();
2117*0b57cec5SDimitry Andric   if (!generic)
2118*0b57cec5SDimitry Andric     return StructuredData::DictionarySP();
2119*0b57cec5SDimitry Andric 
2120*0b57cec5SDimitry Andric   Locker py_lock(this,
2121*0b57cec5SDimitry Andric                  Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2122*0b57cec5SDimitry Andric   TargetSP target_sp(target->shared_from_this());
2123*0b57cec5SDimitry Andric 
21249dba64beSDimitry Andric   auto setting = (PyObject *)LLDBSWIGPython_GetDynamicSetting(
21259dba64beSDimitry Andric       generic->GetValue(), setting_name, target_sp);
21269dba64beSDimitry Andric 
21279dba64beSDimitry Andric   if (!setting)
21289dba64beSDimitry Andric     return StructuredData::DictionarySP();
21299dba64beSDimitry Andric 
21309dba64beSDimitry Andric   PythonDictionary py_dict =
21319dba64beSDimitry Andric       unwrapIgnoringErrors(As<PythonDictionary>(Take<PythonObject>(setting)));
21329dba64beSDimitry Andric 
21339dba64beSDimitry Andric   if (!py_dict)
21349dba64beSDimitry Andric     return StructuredData::DictionarySP();
21359dba64beSDimitry Andric 
2136*0b57cec5SDimitry Andric   return py_dict.CreateStructuredDictionary();
2137*0b57cec5SDimitry Andric }
2138*0b57cec5SDimitry Andric 
2139*0b57cec5SDimitry Andric StructuredData::ObjectSP
2140*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::CreateSyntheticScriptedProvider(
2141*0b57cec5SDimitry Andric     const char *class_name, lldb::ValueObjectSP valobj) {
2142*0b57cec5SDimitry Andric   if (class_name == nullptr || class_name[0] == '\0')
2143*0b57cec5SDimitry Andric     return StructuredData::ObjectSP();
2144*0b57cec5SDimitry Andric 
2145*0b57cec5SDimitry Andric   if (!valobj.get())
2146*0b57cec5SDimitry Andric     return StructuredData::ObjectSP();
2147*0b57cec5SDimitry Andric 
2148*0b57cec5SDimitry Andric   ExecutionContext exe_ctx(valobj->GetExecutionContextRef());
2149*0b57cec5SDimitry Andric   Target *target = exe_ctx.GetTargetPtr();
2150*0b57cec5SDimitry Andric 
2151*0b57cec5SDimitry Andric   if (!target)
2152*0b57cec5SDimitry Andric     return StructuredData::ObjectSP();
2153*0b57cec5SDimitry Andric 
2154*0b57cec5SDimitry Andric   Debugger &debugger = target->GetDebugger();
2155*0b57cec5SDimitry Andric   ScriptInterpreterPythonImpl *python_interpreter =
2156e8d8bef9SDimitry Andric       GetPythonInterpreter(debugger);
2157*0b57cec5SDimitry Andric 
2158e8d8bef9SDimitry Andric   if (!python_interpreter)
2159*0b57cec5SDimitry Andric     return StructuredData::ObjectSP();
2160*0b57cec5SDimitry Andric 
2161*0b57cec5SDimitry Andric   void *ret_val = nullptr;
2162*0b57cec5SDimitry Andric 
2163*0b57cec5SDimitry Andric   {
2164*0b57cec5SDimitry Andric     Locker py_lock(this,
2165*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2166*0b57cec5SDimitry Andric     ret_val = LLDBSwigPythonCreateSyntheticProvider(
2167*0b57cec5SDimitry Andric         class_name, python_interpreter->m_dictionary_name.c_str(), valobj);
2168*0b57cec5SDimitry Andric   }
2169*0b57cec5SDimitry Andric 
2170*0b57cec5SDimitry Andric   return StructuredData::ObjectSP(new StructuredPythonObject(ret_val));
2171*0b57cec5SDimitry Andric }
2172*0b57cec5SDimitry Andric 
2173*0b57cec5SDimitry Andric StructuredData::GenericSP
2174*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::CreateScriptCommandObject(const char *class_name) {
2175*0b57cec5SDimitry Andric   DebuggerSP debugger_sp(m_debugger.shared_from_this());
2176*0b57cec5SDimitry Andric 
2177*0b57cec5SDimitry Andric   if (class_name == nullptr || class_name[0] == '\0')
2178*0b57cec5SDimitry Andric     return StructuredData::GenericSP();
2179*0b57cec5SDimitry Andric 
2180*0b57cec5SDimitry Andric   if (!debugger_sp.get())
2181*0b57cec5SDimitry Andric     return StructuredData::GenericSP();
2182*0b57cec5SDimitry Andric 
2183*0b57cec5SDimitry Andric   void *ret_val;
2184*0b57cec5SDimitry Andric 
2185*0b57cec5SDimitry Andric   {
2186*0b57cec5SDimitry Andric     Locker py_lock(this,
2187*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2188*0b57cec5SDimitry Andric     ret_val = LLDBSwigPythonCreateCommandObject(
2189*0b57cec5SDimitry Andric         class_name, m_dictionary_name.c_str(), debugger_sp);
2190*0b57cec5SDimitry Andric   }
2191*0b57cec5SDimitry Andric 
2192*0b57cec5SDimitry Andric   return StructuredData::GenericSP(new StructuredPythonObject(ret_val));
2193*0b57cec5SDimitry Andric }
2194*0b57cec5SDimitry Andric 
2195*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateTypeScriptFunction(
2196*0b57cec5SDimitry Andric     const char *oneliner, std::string &output, const void *name_token) {
2197*0b57cec5SDimitry Andric   StringList input;
2198*0b57cec5SDimitry Andric   input.SplitIntoLines(oneliner, strlen(oneliner));
2199*0b57cec5SDimitry Andric   return GenerateTypeScriptFunction(input, output, name_token);
2200*0b57cec5SDimitry Andric }
2201*0b57cec5SDimitry Andric 
2202*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateTypeSynthClass(
2203*0b57cec5SDimitry Andric     const char *oneliner, std::string &output, const void *name_token) {
2204*0b57cec5SDimitry Andric   StringList input;
2205*0b57cec5SDimitry Andric   input.SplitIntoLines(oneliner, strlen(oneliner));
2206*0b57cec5SDimitry Andric   return GenerateTypeSynthClass(input, output, name_token);
2207*0b57cec5SDimitry Andric }
2208*0b57cec5SDimitry Andric 
2209*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::GenerateBreakpointCommandCallbackData(
2210480093f4SDimitry Andric     StringList &user_input, std::string &output,
2211480093f4SDimitry Andric     bool has_extra_args) {
2212*0b57cec5SDimitry Andric   static uint32_t num_created_functions = 0;
2213*0b57cec5SDimitry Andric   user_input.RemoveBlankLines();
2214*0b57cec5SDimitry Andric   StreamString sstr;
2215*0b57cec5SDimitry Andric   Status error;
2216*0b57cec5SDimitry Andric   if (user_input.GetSize() == 0) {
2217*0b57cec5SDimitry Andric     error.SetErrorString("No input data.");
2218*0b57cec5SDimitry Andric     return error;
2219*0b57cec5SDimitry Andric   }
2220*0b57cec5SDimitry Andric 
2221*0b57cec5SDimitry Andric   std::string auto_generated_function_name(GenerateUniqueName(
2222*0b57cec5SDimitry Andric       "lldb_autogen_python_bp_callback_func_", num_created_functions));
2223480093f4SDimitry Andric   if (has_extra_args)
2224480093f4SDimitry Andric     sstr.Printf("def %s (frame, bp_loc, extra_args, internal_dict):",
2225480093f4SDimitry Andric                 auto_generated_function_name.c_str());
2226480093f4SDimitry Andric   else
2227*0b57cec5SDimitry Andric     sstr.Printf("def %s (frame, bp_loc, internal_dict):",
2228*0b57cec5SDimitry Andric                 auto_generated_function_name.c_str());
2229*0b57cec5SDimitry Andric 
2230*0b57cec5SDimitry Andric   error = GenerateFunction(sstr.GetData(), user_input);
2231*0b57cec5SDimitry Andric   if (!error.Success())
2232*0b57cec5SDimitry Andric     return error;
2233*0b57cec5SDimitry Andric 
2234*0b57cec5SDimitry Andric   // Store the name of the auto-generated function to be called.
2235*0b57cec5SDimitry Andric   output.assign(auto_generated_function_name);
2236*0b57cec5SDimitry Andric   return error;
2237*0b57cec5SDimitry Andric }
2238*0b57cec5SDimitry Andric 
2239*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateWatchpointCommandCallbackData(
2240*0b57cec5SDimitry Andric     StringList &user_input, std::string &output) {
2241*0b57cec5SDimitry Andric   static uint32_t num_created_functions = 0;
2242*0b57cec5SDimitry Andric   user_input.RemoveBlankLines();
2243*0b57cec5SDimitry Andric   StreamString sstr;
2244*0b57cec5SDimitry Andric 
2245*0b57cec5SDimitry Andric   if (user_input.GetSize() == 0)
2246*0b57cec5SDimitry Andric     return false;
2247*0b57cec5SDimitry Andric 
2248*0b57cec5SDimitry Andric   std::string auto_generated_function_name(GenerateUniqueName(
2249*0b57cec5SDimitry Andric       "lldb_autogen_python_wp_callback_func_", num_created_functions));
2250*0b57cec5SDimitry Andric   sstr.Printf("def %s (frame, wp, internal_dict):",
2251*0b57cec5SDimitry Andric               auto_generated_function_name.c_str());
2252*0b57cec5SDimitry Andric 
2253*0b57cec5SDimitry Andric   if (!GenerateFunction(sstr.GetData(), user_input).Success())
2254*0b57cec5SDimitry Andric     return false;
2255*0b57cec5SDimitry Andric 
2256*0b57cec5SDimitry Andric   // Store the name of the auto-generated function to be called.
2257*0b57cec5SDimitry Andric   output.assign(auto_generated_function_name);
2258*0b57cec5SDimitry Andric   return true;
2259*0b57cec5SDimitry Andric }
2260*0b57cec5SDimitry Andric 
2261*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GetScriptedSummary(
2262*0b57cec5SDimitry Andric     const char *python_function_name, lldb::ValueObjectSP valobj,
2263*0b57cec5SDimitry Andric     StructuredData::ObjectSP &callee_wrapper_sp,
2264*0b57cec5SDimitry Andric     const TypeSummaryOptions &options, std::string &retval) {
2265*0b57cec5SDimitry Andric 
2266e8d8bef9SDimitry Andric   LLDB_SCOPED_TIMER();
2267*0b57cec5SDimitry Andric 
2268*0b57cec5SDimitry Andric   if (!valobj.get()) {
2269*0b57cec5SDimitry Andric     retval.assign("<no object>");
2270*0b57cec5SDimitry Andric     return false;
2271*0b57cec5SDimitry Andric   }
2272*0b57cec5SDimitry Andric 
2273*0b57cec5SDimitry Andric   void *old_callee = nullptr;
2274*0b57cec5SDimitry Andric   StructuredData::Generic *generic = nullptr;
2275*0b57cec5SDimitry Andric   if (callee_wrapper_sp) {
2276*0b57cec5SDimitry Andric     generic = callee_wrapper_sp->GetAsGeneric();
2277*0b57cec5SDimitry Andric     if (generic)
2278*0b57cec5SDimitry Andric       old_callee = generic->GetValue();
2279*0b57cec5SDimitry Andric   }
2280*0b57cec5SDimitry Andric   void *new_callee = old_callee;
2281*0b57cec5SDimitry Andric 
2282*0b57cec5SDimitry Andric   bool ret_val;
2283*0b57cec5SDimitry Andric   if (python_function_name && *python_function_name) {
2284*0b57cec5SDimitry Andric     {
2285*0b57cec5SDimitry Andric       Locker py_lock(this, Locker::AcquireLock | Locker::InitSession |
2286*0b57cec5SDimitry Andric                                Locker::NoSTDIN);
2287*0b57cec5SDimitry Andric       {
2288*0b57cec5SDimitry Andric         TypeSummaryOptionsSP options_sp(new TypeSummaryOptions(options));
2289*0b57cec5SDimitry Andric 
2290*0b57cec5SDimitry Andric         static Timer::Category func_cat("LLDBSwigPythonCallTypeScript");
2291*0b57cec5SDimitry Andric         Timer scoped_timer(func_cat, "LLDBSwigPythonCallTypeScript");
2292*0b57cec5SDimitry Andric         ret_val = LLDBSwigPythonCallTypeScript(
2293*0b57cec5SDimitry Andric             python_function_name, GetSessionDictionary().get(), valobj,
2294*0b57cec5SDimitry Andric             &new_callee, options_sp, retval);
2295*0b57cec5SDimitry Andric       }
2296*0b57cec5SDimitry Andric     }
2297*0b57cec5SDimitry Andric   } else {
2298*0b57cec5SDimitry Andric     retval.assign("<no function name>");
2299*0b57cec5SDimitry Andric     return false;
2300*0b57cec5SDimitry Andric   }
2301*0b57cec5SDimitry Andric 
2302*0b57cec5SDimitry Andric   if (new_callee && old_callee != new_callee)
2303*0b57cec5SDimitry Andric     callee_wrapper_sp = std::make_shared<StructuredPythonObject>(new_callee);
2304*0b57cec5SDimitry Andric 
2305*0b57cec5SDimitry Andric   return ret_val;
2306*0b57cec5SDimitry Andric }
2307*0b57cec5SDimitry Andric 
2308*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::BreakpointCallbackFunction(
2309*0b57cec5SDimitry Andric     void *baton, StoppointCallbackContext *context, user_id_t break_id,
2310*0b57cec5SDimitry Andric     user_id_t break_loc_id) {
2311*0b57cec5SDimitry Andric   CommandDataPython *bp_option_data = (CommandDataPython *)baton;
2312*0b57cec5SDimitry Andric   const char *python_function_name = bp_option_data->script_source.c_str();
2313*0b57cec5SDimitry Andric 
2314*0b57cec5SDimitry Andric   if (!context)
2315*0b57cec5SDimitry Andric     return true;
2316*0b57cec5SDimitry Andric 
2317*0b57cec5SDimitry Andric   ExecutionContext exe_ctx(context->exe_ctx_ref);
2318*0b57cec5SDimitry Andric   Target *target = exe_ctx.GetTargetPtr();
2319*0b57cec5SDimitry Andric 
2320*0b57cec5SDimitry Andric   if (!target)
2321*0b57cec5SDimitry Andric     return true;
2322*0b57cec5SDimitry Andric 
2323*0b57cec5SDimitry Andric   Debugger &debugger = target->GetDebugger();
2324*0b57cec5SDimitry Andric   ScriptInterpreterPythonImpl *python_interpreter =
2325e8d8bef9SDimitry Andric       GetPythonInterpreter(debugger);
2326*0b57cec5SDimitry Andric 
2327e8d8bef9SDimitry Andric   if (!python_interpreter)
2328*0b57cec5SDimitry Andric     return true;
2329*0b57cec5SDimitry Andric 
2330*0b57cec5SDimitry Andric   if (python_function_name && python_function_name[0]) {
2331*0b57cec5SDimitry Andric     const StackFrameSP stop_frame_sp(exe_ctx.GetFrameSP());
2332*0b57cec5SDimitry Andric     BreakpointSP breakpoint_sp = target->GetBreakpointByID(break_id);
2333*0b57cec5SDimitry Andric     if (breakpoint_sp) {
2334*0b57cec5SDimitry Andric       const BreakpointLocationSP bp_loc_sp(
2335*0b57cec5SDimitry Andric           breakpoint_sp->FindLocationByID(break_loc_id));
2336*0b57cec5SDimitry Andric 
2337*0b57cec5SDimitry Andric       if (stop_frame_sp && bp_loc_sp) {
2338*0b57cec5SDimitry Andric         bool ret_val = true;
2339*0b57cec5SDimitry Andric         {
2340*0b57cec5SDimitry Andric           Locker py_lock(python_interpreter, Locker::AcquireLock |
2341*0b57cec5SDimitry Andric                                                  Locker::InitSession |
2342*0b57cec5SDimitry Andric                                                  Locker::NoSTDIN);
2343480093f4SDimitry Andric           Expected<bool> maybe_ret_val =
2344480093f4SDimitry Andric               LLDBSwigPythonBreakpointCallbackFunction(
2345*0b57cec5SDimitry Andric                   python_function_name,
2346*0b57cec5SDimitry Andric                   python_interpreter->m_dictionary_name.c_str(), stop_frame_sp,
2347480093f4SDimitry Andric                   bp_loc_sp, bp_option_data->m_extra_args_up.get());
2348480093f4SDimitry Andric 
2349480093f4SDimitry Andric           if (!maybe_ret_val) {
2350480093f4SDimitry Andric 
2351480093f4SDimitry Andric             llvm::handleAllErrors(
2352480093f4SDimitry Andric                 maybe_ret_val.takeError(),
2353480093f4SDimitry Andric                 [&](PythonException &E) {
2354480093f4SDimitry Andric                   debugger.GetErrorStream() << E.ReadBacktrace();
2355480093f4SDimitry Andric                 },
2356480093f4SDimitry Andric                 [&](const llvm::ErrorInfoBase &E) {
2357480093f4SDimitry Andric                   debugger.GetErrorStream() << E.message();
2358480093f4SDimitry Andric                 });
2359480093f4SDimitry Andric 
2360480093f4SDimitry Andric           } else {
2361480093f4SDimitry Andric             ret_val = maybe_ret_val.get();
2362480093f4SDimitry Andric           }
2363*0b57cec5SDimitry Andric         }
2364*0b57cec5SDimitry Andric         return ret_val;
2365*0b57cec5SDimitry Andric       }
2366*0b57cec5SDimitry Andric     }
2367*0b57cec5SDimitry Andric   }
2368*0b57cec5SDimitry Andric   // We currently always true so we stop in case anything goes wrong when
2369*0b57cec5SDimitry Andric   // trying to call the script function
2370*0b57cec5SDimitry Andric   return true;
2371*0b57cec5SDimitry Andric }
2372*0b57cec5SDimitry Andric 
2373*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::WatchpointCallbackFunction(
2374*0b57cec5SDimitry Andric     void *baton, StoppointCallbackContext *context, user_id_t watch_id) {
2375*0b57cec5SDimitry Andric   WatchpointOptions::CommandData *wp_option_data =
2376*0b57cec5SDimitry Andric       (WatchpointOptions::CommandData *)baton;
2377*0b57cec5SDimitry Andric   const char *python_function_name = wp_option_data->script_source.c_str();
2378*0b57cec5SDimitry Andric 
2379*0b57cec5SDimitry Andric   if (!context)
2380*0b57cec5SDimitry Andric     return true;
2381*0b57cec5SDimitry Andric 
2382*0b57cec5SDimitry Andric   ExecutionContext exe_ctx(context->exe_ctx_ref);
2383*0b57cec5SDimitry Andric   Target *target = exe_ctx.GetTargetPtr();
2384*0b57cec5SDimitry Andric 
2385*0b57cec5SDimitry Andric   if (!target)
2386*0b57cec5SDimitry Andric     return true;
2387*0b57cec5SDimitry Andric 
2388*0b57cec5SDimitry Andric   Debugger &debugger = target->GetDebugger();
2389*0b57cec5SDimitry Andric   ScriptInterpreterPythonImpl *python_interpreter =
2390e8d8bef9SDimitry Andric       GetPythonInterpreter(debugger);
2391*0b57cec5SDimitry Andric 
2392e8d8bef9SDimitry Andric   if (!python_interpreter)
2393*0b57cec5SDimitry Andric     return true;
2394*0b57cec5SDimitry Andric 
2395*0b57cec5SDimitry Andric   if (python_function_name && python_function_name[0]) {
2396*0b57cec5SDimitry Andric     const StackFrameSP stop_frame_sp(exe_ctx.GetFrameSP());
2397*0b57cec5SDimitry Andric     WatchpointSP wp_sp = target->GetWatchpointList().FindByID(watch_id);
2398*0b57cec5SDimitry Andric     if (wp_sp) {
2399*0b57cec5SDimitry Andric       if (stop_frame_sp && wp_sp) {
2400*0b57cec5SDimitry Andric         bool ret_val = true;
2401*0b57cec5SDimitry Andric         {
2402*0b57cec5SDimitry Andric           Locker py_lock(python_interpreter, Locker::AcquireLock |
2403*0b57cec5SDimitry Andric                                                  Locker::InitSession |
2404*0b57cec5SDimitry Andric                                                  Locker::NoSTDIN);
2405*0b57cec5SDimitry Andric           ret_val = LLDBSwigPythonWatchpointCallbackFunction(
2406*0b57cec5SDimitry Andric               python_function_name,
2407*0b57cec5SDimitry Andric               python_interpreter->m_dictionary_name.c_str(), stop_frame_sp,
2408*0b57cec5SDimitry Andric               wp_sp);
2409*0b57cec5SDimitry Andric         }
2410*0b57cec5SDimitry Andric         return ret_val;
2411*0b57cec5SDimitry Andric       }
2412*0b57cec5SDimitry Andric     }
2413*0b57cec5SDimitry Andric   }
2414*0b57cec5SDimitry Andric   // We currently always true so we stop in case anything goes wrong when
2415*0b57cec5SDimitry Andric   // trying to call the script function
2416*0b57cec5SDimitry Andric   return true;
2417*0b57cec5SDimitry Andric }
2418*0b57cec5SDimitry Andric 
2419*0b57cec5SDimitry Andric size_t ScriptInterpreterPythonImpl::CalculateNumChildren(
2420*0b57cec5SDimitry Andric     const StructuredData::ObjectSP &implementor_sp, uint32_t max) {
2421*0b57cec5SDimitry Andric   if (!implementor_sp)
2422*0b57cec5SDimitry Andric     return 0;
2423*0b57cec5SDimitry Andric   StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
2424*0b57cec5SDimitry Andric   if (!generic)
2425*0b57cec5SDimitry Andric     return 0;
2426*0b57cec5SDimitry Andric   void *implementor = generic->GetValue();
2427*0b57cec5SDimitry Andric   if (!implementor)
2428*0b57cec5SDimitry Andric     return 0;
2429*0b57cec5SDimitry Andric 
2430*0b57cec5SDimitry Andric   size_t ret_val = 0;
2431*0b57cec5SDimitry Andric 
2432*0b57cec5SDimitry Andric   {
2433*0b57cec5SDimitry Andric     Locker py_lock(this,
2434*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2435*0b57cec5SDimitry Andric     ret_val = LLDBSwigPython_CalculateNumChildren(implementor, max);
2436*0b57cec5SDimitry Andric   }
2437*0b57cec5SDimitry Andric 
2438*0b57cec5SDimitry Andric   return ret_val;
2439*0b57cec5SDimitry Andric }
2440*0b57cec5SDimitry Andric 
2441*0b57cec5SDimitry Andric lldb::ValueObjectSP ScriptInterpreterPythonImpl::GetChildAtIndex(
2442*0b57cec5SDimitry Andric     const StructuredData::ObjectSP &implementor_sp, uint32_t idx) {
2443*0b57cec5SDimitry Andric   if (!implementor_sp)
2444*0b57cec5SDimitry Andric     return lldb::ValueObjectSP();
2445*0b57cec5SDimitry Andric 
2446*0b57cec5SDimitry Andric   StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
2447*0b57cec5SDimitry Andric   if (!generic)
2448*0b57cec5SDimitry Andric     return lldb::ValueObjectSP();
2449*0b57cec5SDimitry Andric   void *implementor = generic->GetValue();
2450*0b57cec5SDimitry Andric   if (!implementor)
2451*0b57cec5SDimitry Andric     return lldb::ValueObjectSP();
2452*0b57cec5SDimitry Andric 
2453*0b57cec5SDimitry Andric   lldb::ValueObjectSP ret_val;
2454*0b57cec5SDimitry Andric   {
2455*0b57cec5SDimitry Andric     Locker py_lock(this,
2456*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2457*0b57cec5SDimitry Andric     void *child_ptr = LLDBSwigPython_GetChildAtIndex(implementor, idx);
2458*0b57cec5SDimitry Andric     if (child_ptr != nullptr && child_ptr != Py_None) {
2459*0b57cec5SDimitry Andric       lldb::SBValue *sb_value_ptr =
2460*0b57cec5SDimitry Andric           (lldb::SBValue *)LLDBSWIGPython_CastPyObjectToSBValue(child_ptr);
2461*0b57cec5SDimitry Andric       if (sb_value_ptr == nullptr)
2462*0b57cec5SDimitry Andric         Py_XDECREF(child_ptr);
2463*0b57cec5SDimitry Andric       else
2464*0b57cec5SDimitry Andric         ret_val = LLDBSWIGPython_GetValueObjectSPFromSBValue(sb_value_ptr);
2465*0b57cec5SDimitry Andric     } else {
2466*0b57cec5SDimitry Andric       Py_XDECREF(child_ptr);
2467*0b57cec5SDimitry Andric     }
2468*0b57cec5SDimitry Andric   }
2469*0b57cec5SDimitry Andric 
2470*0b57cec5SDimitry Andric   return ret_val;
2471*0b57cec5SDimitry Andric }
2472*0b57cec5SDimitry Andric 
2473*0b57cec5SDimitry Andric int ScriptInterpreterPythonImpl::GetIndexOfChildWithName(
2474*0b57cec5SDimitry Andric     const StructuredData::ObjectSP &implementor_sp, const char *child_name) {
2475*0b57cec5SDimitry Andric   if (!implementor_sp)
2476*0b57cec5SDimitry Andric     return UINT32_MAX;
2477*0b57cec5SDimitry Andric 
2478*0b57cec5SDimitry Andric   StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
2479*0b57cec5SDimitry Andric   if (!generic)
2480*0b57cec5SDimitry Andric     return UINT32_MAX;
2481*0b57cec5SDimitry Andric   void *implementor = generic->GetValue();
2482*0b57cec5SDimitry Andric   if (!implementor)
2483*0b57cec5SDimitry Andric     return UINT32_MAX;
2484*0b57cec5SDimitry Andric 
2485*0b57cec5SDimitry Andric   int ret_val = UINT32_MAX;
2486*0b57cec5SDimitry Andric 
2487*0b57cec5SDimitry Andric   {
2488*0b57cec5SDimitry Andric     Locker py_lock(this,
2489*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2490*0b57cec5SDimitry Andric     ret_val = LLDBSwigPython_GetIndexOfChildWithName(implementor, child_name);
2491*0b57cec5SDimitry Andric   }
2492*0b57cec5SDimitry Andric 
2493*0b57cec5SDimitry Andric   return ret_val;
2494*0b57cec5SDimitry Andric }
2495*0b57cec5SDimitry Andric 
2496*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::UpdateSynthProviderInstance(
2497*0b57cec5SDimitry Andric     const StructuredData::ObjectSP &implementor_sp) {
2498*0b57cec5SDimitry Andric   bool ret_val = false;
2499*0b57cec5SDimitry Andric 
2500*0b57cec5SDimitry Andric   if (!implementor_sp)
2501*0b57cec5SDimitry Andric     return ret_val;
2502*0b57cec5SDimitry Andric 
2503*0b57cec5SDimitry Andric   StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
2504*0b57cec5SDimitry Andric   if (!generic)
2505*0b57cec5SDimitry Andric     return ret_val;
2506*0b57cec5SDimitry Andric   void *implementor = generic->GetValue();
2507*0b57cec5SDimitry Andric   if (!implementor)
2508*0b57cec5SDimitry Andric     return ret_val;
2509*0b57cec5SDimitry Andric 
2510*0b57cec5SDimitry Andric   {
2511*0b57cec5SDimitry Andric     Locker py_lock(this,
2512*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2513*0b57cec5SDimitry Andric     ret_val = LLDBSwigPython_UpdateSynthProviderInstance(implementor);
2514*0b57cec5SDimitry Andric   }
2515*0b57cec5SDimitry Andric 
2516*0b57cec5SDimitry Andric   return ret_val;
2517*0b57cec5SDimitry Andric }
2518*0b57cec5SDimitry Andric 
2519*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::MightHaveChildrenSynthProviderInstance(
2520*0b57cec5SDimitry Andric     const StructuredData::ObjectSP &implementor_sp) {
2521*0b57cec5SDimitry Andric   bool ret_val = false;
2522*0b57cec5SDimitry Andric 
2523*0b57cec5SDimitry Andric   if (!implementor_sp)
2524*0b57cec5SDimitry Andric     return ret_val;
2525*0b57cec5SDimitry Andric 
2526*0b57cec5SDimitry Andric   StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
2527*0b57cec5SDimitry Andric   if (!generic)
2528*0b57cec5SDimitry Andric     return ret_val;
2529*0b57cec5SDimitry Andric   void *implementor = generic->GetValue();
2530*0b57cec5SDimitry Andric   if (!implementor)
2531*0b57cec5SDimitry Andric     return ret_val;
2532*0b57cec5SDimitry Andric 
2533*0b57cec5SDimitry Andric   {
2534*0b57cec5SDimitry Andric     Locker py_lock(this,
2535*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2536*0b57cec5SDimitry Andric     ret_val =
2537*0b57cec5SDimitry Andric         LLDBSwigPython_MightHaveChildrenSynthProviderInstance(implementor);
2538*0b57cec5SDimitry Andric   }
2539*0b57cec5SDimitry Andric 
2540*0b57cec5SDimitry Andric   return ret_val;
2541*0b57cec5SDimitry Andric }
2542*0b57cec5SDimitry Andric 
2543*0b57cec5SDimitry Andric lldb::ValueObjectSP ScriptInterpreterPythonImpl::GetSyntheticValue(
2544*0b57cec5SDimitry Andric     const StructuredData::ObjectSP &implementor_sp) {
2545*0b57cec5SDimitry Andric   lldb::ValueObjectSP ret_val(nullptr);
2546*0b57cec5SDimitry Andric 
2547*0b57cec5SDimitry Andric   if (!implementor_sp)
2548*0b57cec5SDimitry Andric     return ret_val;
2549*0b57cec5SDimitry Andric 
2550*0b57cec5SDimitry Andric   StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
2551*0b57cec5SDimitry Andric   if (!generic)
2552*0b57cec5SDimitry Andric     return ret_val;
2553*0b57cec5SDimitry Andric   void *implementor = generic->GetValue();
2554*0b57cec5SDimitry Andric   if (!implementor)
2555*0b57cec5SDimitry Andric     return ret_val;
2556*0b57cec5SDimitry Andric 
2557*0b57cec5SDimitry Andric   {
2558*0b57cec5SDimitry Andric     Locker py_lock(this,
2559*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2560*0b57cec5SDimitry Andric     void *child_ptr = LLDBSwigPython_GetValueSynthProviderInstance(implementor);
2561*0b57cec5SDimitry Andric     if (child_ptr != nullptr && child_ptr != Py_None) {
2562*0b57cec5SDimitry Andric       lldb::SBValue *sb_value_ptr =
2563*0b57cec5SDimitry Andric           (lldb::SBValue *)LLDBSWIGPython_CastPyObjectToSBValue(child_ptr);
2564*0b57cec5SDimitry Andric       if (sb_value_ptr == nullptr)
2565*0b57cec5SDimitry Andric         Py_XDECREF(child_ptr);
2566*0b57cec5SDimitry Andric       else
2567*0b57cec5SDimitry Andric         ret_val = LLDBSWIGPython_GetValueObjectSPFromSBValue(sb_value_ptr);
2568*0b57cec5SDimitry Andric     } else {
2569*0b57cec5SDimitry Andric       Py_XDECREF(child_ptr);
2570*0b57cec5SDimitry Andric     }
2571*0b57cec5SDimitry Andric   }
2572*0b57cec5SDimitry Andric 
2573*0b57cec5SDimitry Andric   return ret_val;
2574*0b57cec5SDimitry Andric }
2575*0b57cec5SDimitry Andric 
2576*0b57cec5SDimitry Andric ConstString ScriptInterpreterPythonImpl::GetSyntheticTypeName(
2577*0b57cec5SDimitry Andric     const StructuredData::ObjectSP &implementor_sp) {
2578*0b57cec5SDimitry Andric   Locker py_lock(this,
2579*0b57cec5SDimitry Andric                  Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2580*0b57cec5SDimitry Andric 
2581*0b57cec5SDimitry Andric   static char callee_name[] = "get_type_name";
2582*0b57cec5SDimitry Andric 
2583*0b57cec5SDimitry Andric   ConstString ret_val;
2584*0b57cec5SDimitry Andric   bool got_string = false;
2585*0b57cec5SDimitry Andric   std::string buffer;
2586*0b57cec5SDimitry Andric 
2587*0b57cec5SDimitry Andric   if (!implementor_sp)
2588*0b57cec5SDimitry Andric     return ret_val;
2589*0b57cec5SDimitry Andric 
2590*0b57cec5SDimitry Andric   StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
2591*0b57cec5SDimitry Andric   if (!generic)
2592*0b57cec5SDimitry Andric     return ret_val;
2593*0b57cec5SDimitry Andric   PythonObject implementor(PyRefType::Borrowed,
2594*0b57cec5SDimitry Andric                            (PyObject *)generic->GetValue());
2595*0b57cec5SDimitry Andric   if (!implementor.IsAllocated())
2596*0b57cec5SDimitry Andric     return ret_val;
2597*0b57cec5SDimitry Andric 
2598*0b57cec5SDimitry Andric   PythonObject pmeth(PyRefType::Owned,
2599*0b57cec5SDimitry Andric                      PyObject_GetAttrString(implementor.get(), callee_name));
2600*0b57cec5SDimitry Andric 
2601*0b57cec5SDimitry Andric   if (PyErr_Occurred())
2602*0b57cec5SDimitry Andric     PyErr_Clear();
2603*0b57cec5SDimitry Andric 
2604*0b57cec5SDimitry Andric   if (!pmeth.IsAllocated())
2605*0b57cec5SDimitry Andric     return ret_val;
2606*0b57cec5SDimitry Andric 
2607*0b57cec5SDimitry Andric   if (PyCallable_Check(pmeth.get()) == 0) {
2608*0b57cec5SDimitry Andric     if (PyErr_Occurred())
2609*0b57cec5SDimitry Andric       PyErr_Clear();
2610*0b57cec5SDimitry Andric     return ret_val;
2611*0b57cec5SDimitry Andric   }
2612*0b57cec5SDimitry Andric 
2613*0b57cec5SDimitry Andric   if (PyErr_Occurred())
2614*0b57cec5SDimitry Andric     PyErr_Clear();
2615*0b57cec5SDimitry Andric 
2616*0b57cec5SDimitry Andric   // right now we know this function exists and is callable..
2617*0b57cec5SDimitry Andric   PythonObject py_return(
2618*0b57cec5SDimitry Andric       PyRefType::Owned,
2619*0b57cec5SDimitry Andric       PyObject_CallMethod(implementor.get(), callee_name, nullptr));
2620*0b57cec5SDimitry Andric 
2621*0b57cec5SDimitry Andric   // if it fails, print the error but otherwise go on
2622*0b57cec5SDimitry Andric   if (PyErr_Occurred()) {
2623*0b57cec5SDimitry Andric     PyErr_Print();
2624*0b57cec5SDimitry Andric     PyErr_Clear();
2625*0b57cec5SDimitry Andric   }
2626*0b57cec5SDimitry Andric 
2627*0b57cec5SDimitry Andric   if (py_return.IsAllocated() && PythonString::Check(py_return.get())) {
2628*0b57cec5SDimitry Andric     PythonString py_string(PyRefType::Borrowed, py_return.get());
2629*0b57cec5SDimitry Andric     llvm::StringRef return_data(py_string.GetString());
2630*0b57cec5SDimitry Andric     if (!return_data.empty()) {
2631*0b57cec5SDimitry Andric       buffer.assign(return_data.data(), return_data.size());
2632*0b57cec5SDimitry Andric       got_string = true;
2633*0b57cec5SDimitry Andric     }
2634*0b57cec5SDimitry Andric   }
2635*0b57cec5SDimitry Andric 
2636*0b57cec5SDimitry Andric   if (got_string)
2637*0b57cec5SDimitry Andric     ret_val.SetCStringWithLength(buffer.c_str(), buffer.size());
2638*0b57cec5SDimitry Andric 
2639*0b57cec5SDimitry Andric   return ret_val;
2640*0b57cec5SDimitry Andric }
2641*0b57cec5SDimitry Andric 
2642*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptFormatKeyword(
2643*0b57cec5SDimitry Andric     const char *impl_function, Process *process, std::string &output,
2644*0b57cec5SDimitry Andric     Status &error) {
2645*0b57cec5SDimitry Andric   bool ret_val;
2646*0b57cec5SDimitry Andric   if (!process) {
2647*0b57cec5SDimitry Andric     error.SetErrorString("no process");
2648*0b57cec5SDimitry Andric     return false;
2649*0b57cec5SDimitry Andric   }
2650*0b57cec5SDimitry Andric   if (!impl_function || !impl_function[0]) {
2651*0b57cec5SDimitry Andric     error.SetErrorString("no function to execute");
2652*0b57cec5SDimitry Andric     return false;
2653*0b57cec5SDimitry Andric   }
2654*0b57cec5SDimitry Andric 
2655*0b57cec5SDimitry Andric   {
2656*0b57cec5SDimitry Andric     ProcessSP process_sp(process->shared_from_this());
2657*0b57cec5SDimitry Andric     Locker py_lock(this,
2658*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2659*0b57cec5SDimitry Andric     ret_val = LLDBSWIGPythonRunScriptKeywordProcess(
2660*0b57cec5SDimitry Andric         impl_function, m_dictionary_name.c_str(), process_sp, output);
2661*0b57cec5SDimitry Andric     if (!ret_val)
2662*0b57cec5SDimitry Andric       error.SetErrorString("python script evaluation failed");
2663*0b57cec5SDimitry Andric   }
2664*0b57cec5SDimitry Andric   return ret_val;
2665*0b57cec5SDimitry Andric }
2666*0b57cec5SDimitry Andric 
2667*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptFormatKeyword(
2668*0b57cec5SDimitry Andric     const char *impl_function, Thread *thread, std::string &output,
2669*0b57cec5SDimitry Andric     Status &error) {
2670*0b57cec5SDimitry Andric   bool ret_val;
2671*0b57cec5SDimitry Andric   if (!thread) {
2672*0b57cec5SDimitry Andric     error.SetErrorString("no thread");
2673*0b57cec5SDimitry Andric     return false;
2674*0b57cec5SDimitry Andric   }
2675*0b57cec5SDimitry Andric   if (!impl_function || !impl_function[0]) {
2676*0b57cec5SDimitry Andric     error.SetErrorString("no function to execute");
2677*0b57cec5SDimitry Andric     return false;
2678*0b57cec5SDimitry Andric   }
2679*0b57cec5SDimitry Andric 
2680*0b57cec5SDimitry Andric   {
2681*0b57cec5SDimitry Andric     ThreadSP thread_sp(thread->shared_from_this());
2682*0b57cec5SDimitry Andric     Locker py_lock(this,
2683*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2684*0b57cec5SDimitry Andric     ret_val = LLDBSWIGPythonRunScriptKeywordThread(
2685*0b57cec5SDimitry Andric         impl_function, m_dictionary_name.c_str(), thread_sp, output);
2686*0b57cec5SDimitry Andric     if (!ret_val)
2687*0b57cec5SDimitry Andric       error.SetErrorString("python script evaluation failed");
2688*0b57cec5SDimitry Andric   }
2689*0b57cec5SDimitry Andric   return ret_val;
2690*0b57cec5SDimitry Andric }
2691*0b57cec5SDimitry Andric 
2692*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptFormatKeyword(
2693*0b57cec5SDimitry Andric     const char *impl_function, Target *target, std::string &output,
2694*0b57cec5SDimitry Andric     Status &error) {
2695*0b57cec5SDimitry Andric   bool ret_val;
2696*0b57cec5SDimitry Andric   if (!target) {
2697*0b57cec5SDimitry Andric     error.SetErrorString("no thread");
2698*0b57cec5SDimitry Andric     return false;
2699*0b57cec5SDimitry Andric   }
2700*0b57cec5SDimitry Andric   if (!impl_function || !impl_function[0]) {
2701*0b57cec5SDimitry Andric     error.SetErrorString("no function to execute");
2702*0b57cec5SDimitry Andric     return false;
2703*0b57cec5SDimitry Andric   }
2704*0b57cec5SDimitry Andric 
2705*0b57cec5SDimitry Andric   {
2706*0b57cec5SDimitry Andric     TargetSP target_sp(target->shared_from_this());
2707*0b57cec5SDimitry Andric     Locker py_lock(this,
2708*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2709*0b57cec5SDimitry Andric     ret_val = LLDBSWIGPythonRunScriptKeywordTarget(
2710*0b57cec5SDimitry Andric         impl_function, m_dictionary_name.c_str(), target_sp, output);
2711*0b57cec5SDimitry Andric     if (!ret_val)
2712*0b57cec5SDimitry Andric       error.SetErrorString("python script evaluation failed");
2713*0b57cec5SDimitry Andric   }
2714*0b57cec5SDimitry Andric   return ret_val;
2715*0b57cec5SDimitry Andric }
2716*0b57cec5SDimitry Andric 
2717*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptFormatKeyword(
2718*0b57cec5SDimitry Andric     const char *impl_function, StackFrame *frame, std::string &output,
2719*0b57cec5SDimitry Andric     Status &error) {
2720*0b57cec5SDimitry Andric   bool ret_val;
2721*0b57cec5SDimitry Andric   if (!frame) {
2722*0b57cec5SDimitry Andric     error.SetErrorString("no frame");
2723*0b57cec5SDimitry Andric     return false;
2724*0b57cec5SDimitry Andric   }
2725*0b57cec5SDimitry Andric   if (!impl_function || !impl_function[0]) {
2726*0b57cec5SDimitry Andric     error.SetErrorString("no function to execute");
2727*0b57cec5SDimitry Andric     return false;
2728*0b57cec5SDimitry Andric   }
2729*0b57cec5SDimitry Andric 
2730*0b57cec5SDimitry Andric   {
2731*0b57cec5SDimitry Andric     StackFrameSP frame_sp(frame->shared_from_this());
2732*0b57cec5SDimitry Andric     Locker py_lock(this,
2733*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2734*0b57cec5SDimitry Andric     ret_val = LLDBSWIGPythonRunScriptKeywordFrame(
2735*0b57cec5SDimitry Andric         impl_function, m_dictionary_name.c_str(), frame_sp, output);
2736*0b57cec5SDimitry Andric     if (!ret_val)
2737*0b57cec5SDimitry Andric       error.SetErrorString("python script evaluation failed");
2738*0b57cec5SDimitry Andric   }
2739*0b57cec5SDimitry Andric   return ret_val;
2740*0b57cec5SDimitry Andric }
2741*0b57cec5SDimitry Andric 
2742*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptFormatKeyword(
2743*0b57cec5SDimitry Andric     const char *impl_function, ValueObject *value, std::string &output,
2744*0b57cec5SDimitry Andric     Status &error) {
2745*0b57cec5SDimitry Andric   bool ret_val;
2746*0b57cec5SDimitry Andric   if (!value) {
2747*0b57cec5SDimitry Andric     error.SetErrorString("no value");
2748*0b57cec5SDimitry Andric     return false;
2749*0b57cec5SDimitry Andric   }
2750*0b57cec5SDimitry Andric   if (!impl_function || !impl_function[0]) {
2751*0b57cec5SDimitry Andric     error.SetErrorString("no function to execute");
2752*0b57cec5SDimitry Andric     return false;
2753*0b57cec5SDimitry Andric   }
2754*0b57cec5SDimitry Andric 
2755*0b57cec5SDimitry Andric   {
2756*0b57cec5SDimitry Andric     ValueObjectSP value_sp(value->GetSP());
2757*0b57cec5SDimitry Andric     Locker py_lock(this,
2758*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
2759*0b57cec5SDimitry Andric     ret_val = LLDBSWIGPythonRunScriptKeywordValue(
2760*0b57cec5SDimitry Andric         impl_function, m_dictionary_name.c_str(), value_sp, output);
2761*0b57cec5SDimitry Andric     if (!ret_val)
2762*0b57cec5SDimitry Andric       error.SetErrorString("python script evaluation failed");
2763*0b57cec5SDimitry Andric   }
2764*0b57cec5SDimitry Andric   return ret_val;
2765*0b57cec5SDimitry Andric }
2766*0b57cec5SDimitry Andric 
2767*0b57cec5SDimitry Andric uint64_t replace_all(std::string &str, const std::string &oldStr,
2768*0b57cec5SDimitry Andric                      const std::string &newStr) {
2769*0b57cec5SDimitry Andric   size_t pos = 0;
2770*0b57cec5SDimitry Andric   uint64_t matches = 0;
2771*0b57cec5SDimitry Andric   while ((pos = str.find(oldStr, pos)) != std::string::npos) {
2772*0b57cec5SDimitry Andric     matches++;
2773*0b57cec5SDimitry Andric     str.replace(pos, oldStr.length(), newStr);
2774*0b57cec5SDimitry Andric     pos += newStr.length();
2775*0b57cec5SDimitry Andric   }
2776*0b57cec5SDimitry Andric   return matches;
2777*0b57cec5SDimitry Andric }
2778*0b57cec5SDimitry Andric 
2779*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::LoadScriptingModule(
2780fe6060f1SDimitry Andric     const char *pathname, const LoadScriptOptions &options,
2781fe6060f1SDimitry Andric     lldb_private::Status &error, StructuredData::ObjectSP *module_sp,
2782fe6060f1SDimitry Andric     FileSpec extra_search_dir) {
2783e8d8bef9SDimitry Andric   namespace fs = llvm::sys::fs;
2784e8d8bef9SDimitry Andric   namespace path = llvm::sys::path;
2785e8d8bef9SDimitry Andric 
2786fe6060f1SDimitry Andric   ExecuteScriptOptions exc_options = ExecuteScriptOptions()
2787fe6060f1SDimitry Andric                                          .SetEnableIO(!options.GetSilent())
2788fe6060f1SDimitry Andric                                          .SetSetLLDBGlobals(false);
2789fe6060f1SDimitry Andric 
2790*0b57cec5SDimitry Andric   if (!pathname || !pathname[0]) {
2791*0b57cec5SDimitry Andric     error.SetErrorString("invalid pathname");
2792*0b57cec5SDimitry Andric     return false;
2793*0b57cec5SDimitry Andric   }
2794*0b57cec5SDimitry Andric 
2795fe6060f1SDimitry Andric   llvm::Expected<std::unique_ptr<ScriptInterpreterIORedirect>>
2796fe6060f1SDimitry Andric       io_redirect_or_error = ScriptInterpreterIORedirect::Create(
2797fe6060f1SDimitry Andric           exc_options.GetEnableIO(), m_debugger, /*result=*/nullptr);
2798fe6060f1SDimitry Andric 
2799fe6060f1SDimitry Andric   if (!io_redirect_or_error) {
2800fe6060f1SDimitry Andric     error = io_redirect_or_error.takeError();
2801fe6060f1SDimitry Andric     return false;
2802fe6060f1SDimitry Andric   }
2803fe6060f1SDimitry Andric 
2804fe6060f1SDimitry Andric   ScriptInterpreterIORedirect &io_redirect = **io_redirect_or_error;
2805*0b57cec5SDimitry Andric   lldb::DebuggerSP debugger_sp = m_debugger.shared_from_this();
2806*0b57cec5SDimitry Andric 
2807*0b57cec5SDimitry Andric   // Before executing Python code, lock the GIL.
2808*0b57cec5SDimitry Andric   Locker py_lock(this,
2809*0b57cec5SDimitry Andric                  Locker::AcquireLock |
2810fe6060f1SDimitry Andric                      (options.GetInitSession() ? Locker::InitSession : 0) |
2811fe6060f1SDimitry Andric                      Locker::NoSTDIN,
2812*0b57cec5SDimitry Andric                  Locker::FreeAcquiredLock |
2813fe6060f1SDimitry Andric                      (options.GetInitSession() ? Locker::TearDownSession : 0),
2814fe6060f1SDimitry Andric                  io_redirect.GetInputFile(), io_redirect.GetOutputFile(),
2815fe6060f1SDimitry Andric                  io_redirect.GetErrorFile());
2816*0b57cec5SDimitry Andric 
2817fe6060f1SDimitry Andric   auto ExtendSysPath = [&](std::string directory) -> llvm::Error {
2818e8d8bef9SDimitry Andric     if (directory.empty()) {
2819e8d8bef9SDimitry Andric       return llvm::make_error<llvm::StringError>(
2820e8d8bef9SDimitry Andric           "invalid directory name", llvm::inconvertibleErrorCode());
2821*0b57cec5SDimitry Andric     }
2822*0b57cec5SDimitry Andric 
2823*0b57cec5SDimitry Andric     replace_all(directory, "\\", "\\\\");
2824*0b57cec5SDimitry Andric     replace_all(directory, "'", "\\'");
2825*0b57cec5SDimitry Andric 
2826e8d8bef9SDimitry Andric     // Make sure that Python has "directory" in the search path.
2827*0b57cec5SDimitry Andric     StreamString command_stream;
2828*0b57cec5SDimitry Andric     command_stream.Printf("if not (sys.path.__contains__('%s')):\n    "
2829*0b57cec5SDimitry Andric                           "sys.path.insert(1,'%s');\n\n",
2830*0b57cec5SDimitry Andric                           directory.c_str(), directory.c_str());
2831*0b57cec5SDimitry Andric     bool syspath_retval =
2832fe6060f1SDimitry Andric         ExecuteMultipleLines(command_stream.GetData(), exc_options).Success();
2833*0b57cec5SDimitry Andric     if (!syspath_retval) {
2834e8d8bef9SDimitry Andric       return llvm::make_error<llvm::StringError>(
2835e8d8bef9SDimitry Andric           "Python sys.path handling failed", llvm::inconvertibleErrorCode());
2836*0b57cec5SDimitry Andric     }
2837*0b57cec5SDimitry Andric 
2838e8d8bef9SDimitry Andric     return llvm::Error::success();
2839e8d8bef9SDimitry Andric   };
2840e8d8bef9SDimitry Andric 
2841e8d8bef9SDimitry Andric   std::string module_name(pathname);
2842fe6060f1SDimitry Andric   bool possible_package = false;
2843e8d8bef9SDimitry Andric 
2844e8d8bef9SDimitry Andric   if (extra_search_dir) {
2845e8d8bef9SDimitry Andric     if (llvm::Error e = ExtendSysPath(extra_search_dir.GetPath())) {
2846e8d8bef9SDimitry Andric       error = std::move(e);
2847e8d8bef9SDimitry Andric       return false;
2848*0b57cec5SDimitry Andric     }
2849*0b57cec5SDimitry Andric   } else {
2850e8d8bef9SDimitry Andric     FileSpec module_file(pathname);
2851e8d8bef9SDimitry Andric     FileSystem::Instance().Resolve(module_file);
2852e8d8bef9SDimitry Andric     FileSystem::Instance().Collect(module_file);
2853e8d8bef9SDimitry Andric 
2854e8d8bef9SDimitry Andric     fs::file_status st;
2855e8d8bef9SDimitry Andric     std::error_code ec = status(module_file.GetPath(), st);
2856e8d8bef9SDimitry Andric 
2857e8d8bef9SDimitry Andric     if (ec || st.type() == fs::file_type::status_error ||
2858e8d8bef9SDimitry Andric         st.type() == fs::file_type::type_unknown ||
2859e8d8bef9SDimitry Andric         st.type() == fs::file_type::file_not_found) {
2860e8d8bef9SDimitry Andric       // if not a valid file of any sort, check if it might be a filename still
2861e8d8bef9SDimitry Andric       // dot can't be used but / and \ can, and if either is found, reject
2862e8d8bef9SDimitry Andric       if (strchr(pathname, '\\') || strchr(pathname, '/')) {
2863e8d8bef9SDimitry Andric         error.SetErrorString("invalid pathname");
2864e8d8bef9SDimitry Andric         return false;
2865e8d8bef9SDimitry Andric       }
2866e8d8bef9SDimitry Andric       // Not a filename, probably a package of some sort, let it go through.
2867fe6060f1SDimitry Andric       possible_package = true;
2868e8d8bef9SDimitry Andric     } else if (is_directory(st) || is_regular_file(st)) {
2869e8d8bef9SDimitry Andric       if (module_file.GetDirectory().IsEmpty()) {
2870e8d8bef9SDimitry Andric         error.SetErrorString("invalid directory name");
2871e8d8bef9SDimitry Andric         return false;
2872e8d8bef9SDimitry Andric       }
2873e8d8bef9SDimitry Andric       if (llvm::Error e =
2874e8d8bef9SDimitry Andric               ExtendSysPath(module_file.GetDirectory().GetCString())) {
2875e8d8bef9SDimitry Andric         error = std::move(e);
2876e8d8bef9SDimitry Andric         return false;
2877e8d8bef9SDimitry Andric       }
2878e8d8bef9SDimitry Andric       module_name = module_file.GetFilename().GetCString();
2879e8d8bef9SDimitry Andric     } else {
2880*0b57cec5SDimitry Andric       error.SetErrorString("no known way to import this module specification");
2881*0b57cec5SDimitry Andric       return false;
2882*0b57cec5SDimitry Andric     }
2883e8d8bef9SDimitry Andric   }
2884e8d8bef9SDimitry Andric 
2885e8d8bef9SDimitry Andric   // Strip .py or .pyc extension
2886e8d8bef9SDimitry Andric   llvm::StringRef extension = llvm::sys::path::extension(module_name);
2887e8d8bef9SDimitry Andric   if (!extension.empty()) {
2888e8d8bef9SDimitry Andric     if (extension == ".py")
2889e8d8bef9SDimitry Andric       module_name.resize(module_name.length() - 3);
2890e8d8bef9SDimitry Andric     else if (extension == ".pyc")
2891e8d8bef9SDimitry Andric       module_name.resize(module_name.length() - 4);
2892e8d8bef9SDimitry Andric   }
2893*0b57cec5SDimitry Andric 
2894fe6060f1SDimitry Andric   if (!possible_package && module_name.find('.') != llvm::StringRef::npos) {
2895fe6060f1SDimitry Andric     error.SetErrorStringWithFormat(
2896fe6060f1SDimitry Andric         "Python does not allow dots in module names: %s", module_name.c_str());
2897fe6060f1SDimitry Andric     return false;
2898fe6060f1SDimitry Andric   }
2899fe6060f1SDimitry Andric 
2900fe6060f1SDimitry Andric   if (module_name.find('-') != llvm::StringRef::npos) {
2901fe6060f1SDimitry Andric     error.SetErrorStringWithFormat(
2902fe6060f1SDimitry Andric         "Python discourages dashes in module names: %s", module_name.c_str());
2903fe6060f1SDimitry Andric     return false;
2904fe6060f1SDimitry Andric   }
2905fe6060f1SDimitry Andric 
2906fe6060f1SDimitry Andric   // Check if the module is already imported.
2907e8d8bef9SDimitry Andric   StreamString command_stream;
2908*0b57cec5SDimitry Andric   command_stream.Clear();
2909e8d8bef9SDimitry Andric   command_stream.Printf("sys.modules.__contains__('%s')", module_name.c_str());
2910*0b57cec5SDimitry Andric   bool does_contain = false;
2911fe6060f1SDimitry Andric   // This call will succeed if the module was ever imported in any Debugger in
2912fe6060f1SDimitry Andric   // the lifetime of the process in which this LLDB framework is living.
2913fe6060f1SDimitry Andric   const bool does_contain_executed = ExecuteOneLineWithReturn(
2914*0b57cec5SDimitry Andric       command_stream.GetData(),
2915fe6060f1SDimitry Andric       ScriptInterpreterPythonImpl::eScriptReturnTypeBool, &does_contain, exc_options);
2916fe6060f1SDimitry Andric 
2917fe6060f1SDimitry Andric   const bool was_imported_globally = does_contain_executed && does_contain;
2918fe6060f1SDimitry Andric   const bool was_imported_locally =
2919fe6060f1SDimitry Andric       GetSessionDictionary()
2920e8d8bef9SDimitry Andric           .GetItemForKey(PythonString(module_name))
2921*0b57cec5SDimitry Andric           .IsAllocated();
2922*0b57cec5SDimitry Andric 
2923*0b57cec5SDimitry Andric   // now actually do the import
2924*0b57cec5SDimitry Andric   command_stream.Clear();
2925*0b57cec5SDimitry Andric 
2926fe6060f1SDimitry Andric   if (was_imported_globally || was_imported_locally) {
2927*0b57cec5SDimitry Andric     if (!was_imported_locally)
2928e8d8bef9SDimitry Andric       command_stream.Printf("import %s ; reload_module(%s)",
2929e8d8bef9SDimitry Andric                             module_name.c_str(), module_name.c_str());
2930*0b57cec5SDimitry Andric     else
2931e8d8bef9SDimitry Andric       command_stream.Printf("reload_module(%s)", module_name.c_str());
2932*0b57cec5SDimitry Andric   } else
2933e8d8bef9SDimitry Andric     command_stream.Printf("import %s", module_name.c_str());
2934*0b57cec5SDimitry Andric 
2935fe6060f1SDimitry Andric   error = ExecuteMultipleLines(command_stream.GetData(), exc_options);
2936*0b57cec5SDimitry Andric   if (error.Fail())
2937*0b57cec5SDimitry Andric     return false;
2938*0b57cec5SDimitry Andric 
2939*0b57cec5SDimitry Andric   // if we are here, everything worked
2940*0b57cec5SDimitry Andric   // call __lldb_init_module(debugger,dict)
2941e8d8bef9SDimitry Andric   if (!LLDBSwigPythonCallModuleInit(module_name.c_str(),
2942*0b57cec5SDimitry Andric                                     m_dictionary_name.c_str(), debugger_sp)) {
2943*0b57cec5SDimitry Andric     error.SetErrorString("calling __lldb_init_module failed");
2944*0b57cec5SDimitry Andric     return false;
2945*0b57cec5SDimitry Andric   }
2946*0b57cec5SDimitry Andric 
2947*0b57cec5SDimitry Andric   if (module_sp) {
2948*0b57cec5SDimitry Andric     // everything went just great, now set the module object
2949*0b57cec5SDimitry Andric     command_stream.Clear();
2950e8d8bef9SDimitry Andric     command_stream.Printf("%s", module_name.c_str());
2951*0b57cec5SDimitry Andric     void *module_pyobj = nullptr;
2952*0b57cec5SDimitry Andric     if (ExecuteOneLineWithReturn(
2953*0b57cec5SDimitry Andric             command_stream.GetData(),
2954fe6060f1SDimitry Andric             ScriptInterpreter::eScriptReturnTypeOpaqueObject, &module_pyobj,
2955fe6060f1SDimitry Andric             exc_options) &&
2956*0b57cec5SDimitry Andric         module_pyobj)
2957*0b57cec5SDimitry Andric       *module_sp = std::make_shared<StructuredPythonObject>(module_pyobj);
2958*0b57cec5SDimitry Andric   }
2959*0b57cec5SDimitry Andric 
2960*0b57cec5SDimitry Andric   return true;
2961*0b57cec5SDimitry Andric }
2962*0b57cec5SDimitry Andric 
2963*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::IsReservedWord(const char *word) {
2964*0b57cec5SDimitry Andric   if (!word || !word[0])
2965*0b57cec5SDimitry Andric     return false;
2966*0b57cec5SDimitry Andric 
2967*0b57cec5SDimitry Andric   llvm::StringRef word_sr(word);
2968*0b57cec5SDimitry Andric 
2969*0b57cec5SDimitry Andric   // filter out a few characters that would just confuse us and that are
2970*0b57cec5SDimitry Andric   // clearly not keyword material anyway
2971*0b57cec5SDimitry Andric   if (word_sr.find('"') != llvm::StringRef::npos ||
2972*0b57cec5SDimitry Andric       word_sr.find('\'') != llvm::StringRef::npos)
2973*0b57cec5SDimitry Andric     return false;
2974*0b57cec5SDimitry Andric 
2975*0b57cec5SDimitry Andric   StreamString command_stream;
2976*0b57cec5SDimitry Andric   command_stream.Printf("keyword.iskeyword('%s')", word);
2977*0b57cec5SDimitry Andric   bool result;
2978*0b57cec5SDimitry Andric   ExecuteScriptOptions options;
2979*0b57cec5SDimitry Andric   options.SetEnableIO(false);
2980*0b57cec5SDimitry Andric   options.SetMaskoutErrors(true);
2981*0b57cec5SDimitry Andric   options.SetSetLLDBGlobals(false);
2982*0b57cec5SDimitry Andric   if (ExecuteOneLineWithReturn(command_stream.GetData(),
2983*0b57cec5SDimitry Andric                                ScriptInterpreter::eScriptReturnTypeBool,
2984*0b57cec5SDimitry Andric                                &result, options))
2985*0b57cec5SDimitry Andric     return result;
2986*0b57cec5SDimitry Andric   return false;
2987*0b57cec5SDimitry Andric }
2988*0b57cec5SDimitry Andric 
2989*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::SynchronicityHandler::SynchronicityHandler(
2990*0b57cec5SDimitry Andric     lldb::DebuggerSP debugger_sp, ScriptedCommandSynchronicity synchro)
2991*0b57cec5SDimitry Andric     : m_debugger_sp(debugger_sp), m_synch_wanted(synchro),
2992*0b57cec5SDimitry Andric       m_old_asynch(debugger_sp->GetAsyncExecution()) {
2993*0b57cec5SDimitry Andric   if (m_synch_wanted == eScriptedCommandSynchronicitySynchronous)
2994*0b57cec5SDimitry Andric     m_debugger_sp->SetAsyncExecution(false);
2995*0b57cec5SDimitry Andric   else if (m_synch_wanted == eScriptedCommandSynchronicityAsynchronous)
2996*0b57cec5SDimitry Andric     m_debugger_sp->SetAsyncExecution(true);
2997*0b57cec5SDimitry Andric }
2998*0b57cec5SDimitry Andric 
2999*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::SynchronicityHandler::~SynchronicityHandler() {
3000*0b57cec5SDimitry Andric   if (m_synch_wanted != eScriptedCommandSynchronicityCurrentValue)
3001*0b57cec5SDimitry Andric     m_debugger_sp->SetAsyncExecution(m_old_asynch);
3002*0b57cec5SDimitry Andric }
3003*0b57cec5SDimitry Andric 
3004*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptBasedCommand(
3005*0b57cec5SDimitry Andric     const char *impl_function, llvm::StringRef args,
3006*0b57cec5SDimitry Andric     ScriptedCommandSynchronicity synchronicity,
3007*0b57cec5SDimitry Andric     lldb_private::CommandReturnObject &cmd_retobj, Status &error,
3008*0b57cec5SDimitry Andric     const lldb_private::ExecutionContext &exe_ctx) {
3009*0b57cec5SDimitry Andric   if (!impl_function) {
3010*0b57cec5SDimitry Andric     error.SetErrorString("no function to execute");
3011*0b57cec5SDimitry Andric     return false;
3012*0b57cec5SDimitry Andric   }
3013*0b57cec5SDimitry Andric 
3014*0b57cec5SDimitry Andric   lldb::DebuggerSP debugger_sp = m_debugger.shared_from_this();
3015*0b57cec5SDimitry Andric   lldb::ExecutionContextRefSP exe_ctx_ref_sp(new ExecutionContextRef(exe_ctx));
3016*0b57cec5SDimitry Andric 
3017*0b57cec5SDimitry Andric   if (!debugger_sp.get()) {
3018*0b57cec5SDimitry Andric     error.SetErrorString("invalid Debugger pointer");
3019*0b57cec5SDimitry Andric     return false;
3020*0b57cec5SDimitry Andric   }
3021*0b57cec5SDimitry Andric 
3022*0b57cec5SDimitry Andric   bool ret_val = false;
3023*0b57cec5SDimitry Andric 
3024*0b57cec5SDimitry Andric   std::string err_msg;
3025*0b57cec5SDimitry Andric 
3026*0b57cec5SDimitry Andric   {
3027*0b57cec5SDimitry Andric     Locker py_lock(this,
3028*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession |
3029*0b57cec5SDimitry Andric                        (cmd_retobj.GetInteractive() ? 0 : Locker::NoSTDIN),
3030*0b57cec5SDimitry Andric                    Locker::FreeLock | Locker::TearDownSession);
3031*0b57cec5SDimitry Andric 
3032*0b57cec5SDimitry Andric     SynchronicityHandler synch_handler(debugger_sp, synchronicity);
3033*0b57cec5SDimitry Andric 
3034*0b57cec5SDimitry Andric     std::string args_str = args.str();
3035*0b57cec5SDimitry Andric     ret_val = LLDBSwigPythonCallCommand(
3036*0b57cec5SDimitry Andric         impl_function, m_dictionary_name.c_str(), debugger_sp, args_str.c_str(),
3037*0b57cec5SDimitry Andric         cmd_retobj, exe_ctx_ref_sp);
3038*0b57cec5SDimitry Andric   }
3039*0b57cec5SDimitry Andric 
3040*0b57cec5SDimitry Andric   if (!ret_val)
3041*0b57cec5SDimitry Andric     error.SetErrorString("unable to execute script function");
3042*0b57cec5SDimitry Andric   else
3043*0b57cec5SDimitry Andric     error.Clear();
3044*0b57cec5SDimitry Andric 
3045*0b57cec5SDimitry Andric   return ret_val;
3046*0b57cec5SDimitry Andric }
3047*0b57cec5SDimitry Andric 
3048*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptBasedCommand(
3049*0b57cec5SDimitry Andric     StructuredData::GenericSP impl_obj_sp, llvm::StringRef args,
3050*0b57cec5SDimitry Andric     ScriptedCommandSynchronicity synchronicity,
3051*0b57cec5SDimitry Andric     lldb_private::CommandReturnObject &cmd_retobj, Status &error,
3052*0b57cec5SDimitry Andric     const lldb_private::ExecutionContext &exe_ctx) {
3053*0b57cec5SDimitry Andric   if (!impl_obj_sp || !impl_obj_sp->IsValid()) {
3054*0b57cec5SDimitry Andric     error.SetErrorString("no function to execute");
3055*0b57cec5SDimitry Andric     return false;
3056*0b57cec5SDimitry Andric   }
3057*0b57cec5SDimitry Andric 
3058*0b57cec5SDimitry Andric   lldb::DebuggerSP debugger_sp = m_debugger.shared_from_this();
3059*0b57cec5SDimitry Andric   lldb::ExecutionContextRefSP exe_ctx_ref_sp(new ExecutionContextRef(exe_ctx));
3060*0b57cec5SDimitry Andric 
3061*0b57cec5SDimitry Andric   if (!debugger_sp.get()) {
3062*0b57cec5SDimitry Andric     error.SetErrorString("invalid Debugger pointer");
3063*0b57cec5SDimitry Andric     return false;
3064*0b57cec5SDimitry Andric   }
3065*0b57cec5SDimitry Andric 
3066*0b57cec5SDimitry Andric   bool ret_val = false;
3067*0b57cec5SDimitry Andric 
3068*0b57cec5SDimitry Andric   std::string err_msg;
3069*0b57cec5SDimitry Andric 
3070*0b57cec5SDimitry Andric   {
3071*0b57cec5SDimitry Andric     Locker py_lock(this,
3072*0b57cec5SDimitry Andric                    Locker::AcquireLock | Locker::InitSession |
3073*0b57cec5SDimitry Andric                        (cmd_retobj.GetInteractive() ? 0 : Locker::NoSTDIN),
3074*0b57cec5SDimitry Andric                    Locker::FreeLock | Locker::TearDownSession);
3075*0b57cec5SDimitry Andric 
3076*0b57cec5SDimitry Andric     SynchronicityHandler synch_handler(debugger_sp, synchronicity);
3077*0b57cec5SDimitry Andric 
3078*0b57cec5SDimitry Andric     std::string args_str = args.str();
3079*0b57cec5SDimitry Andric     ret_val = LLDBSwigPythonCallCommandObject(impl_obj_sp->GetValue(),
3080*0b57cec5SDimitry Andric                                               debugger_sp, args_str.c_str(),
3081*0b57cec5SDimitry Andric                                               cmd_retobj, exe_ctx_ref_sp);
3082*0b57cec5SDimitry Andric   }
3083*0b57cec5SDimitry Andric 
3084*0b57cec5SDimitry Andric   if (!ret_val)
3085*0b57cec5SDimitry Andric     error.SetErrorString("unable to execute script function");
3086*0b57cec5SDimitry Andric   else
3087*0b57cec5SDimitry Andric     error.Clear();
3088*0b57cec5SDimitry Andric 
3089*0b57cec5SDimitry Andric   return ret_val;
3090*0b57cec5SDimitry Andric }
3091*0b57cec5SDimitry Andric 
30925ffd83dbSDimitry Andric /// In Python, a special attribute __doc__ contains the docstring for an object
30935ffd83dbSDimitry Andric /// (function, method, class, ...) if any is defined Otherwise, the attribute's
30945ffd83dbSDimitry Andric /// value is None.
3095*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GetDocumentationForItem(const char *item,
3096*0b57cec5SDimitry Andric                                                           std::string &dest) {
3097*0b57cec5SDimitry Andric   dest.clear();
30985ffd83dbSDimitry Andric 
3099*0b57cec5SDimitry Andric   if (!item || !*item)
3100*0b57cec5SDimitry Andric     return false;
31015ffd83dbSDimitry Andric 
3102*0b57cec5SDimitry Andric   std::string command(item);
3103*0b57cec5SDimitry Andric   command += ".__doc__";
3104*0b57cec5SDimitry Andric 
31055ffd83dbSDimitry Andric   // Python is going to point this to valid data if ExecuteOneLineWithReturn
31065ffd83dbSDimitry Andric   // returns successfully.
31075ffd83dbSDimitry Andric   char *result_ptr = nullptr;
3108*0b57cec5SDimitry Andric 
3109*0b57cec5SDimitry Andric   if (ExecuteOneLineWithReturn(
31105ffd83dbSDimitry Andric           command, ScriptInterpreter::eScriptReturnTypeCharStrOrNone,
3111*0b57cec5SDimitry Andric           &result_ptr,
3112fe6060f1SDimitry Andric           ExecuteScriptOptions().SetEnableIO(false))) {
3113*0b57cec5SDimitry Andric     if (result_ptr)
3114*0b57cec5SDimitry Andric       dest.assign(result_ptr);
3115*0b57cec5SDimitry Andric     return true;
3116*0b57cec5SDimitry Andric   }
31175ffd83dbSDimitry Andric 
31185ffd83dbSDimitry Andric   StreamString str_stream;
31195ffd83dbSDimitry Andric   str_stream << "Function " << item
31205ffd83dbSDimitry Andric              << " was not found. Containing module might be missing.";
31215ffd83dbSDimitry Andric   dest = std::string(str_stream.GetString());
31225ffd83dbSDimitry Andric 
31235ffd83dbSDimitry Andric   return false;
3124*0b57cec5SDimitry Andric }
3125*0b57cec5SDimitry Andric 
3126*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GetShortHelpForCommandObject(
3127*0b57cec5SDimitry Andric     StructuredData::GenericSP cmd_obj_sp, std::string &dest) {
3128*0b57cec5SDimitry Andric   dest.clear();
3129*0b57cec5SDimitry Andric 
3130*0b57cec5SDimitry Andric   Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock);
3131*0b57cec5SDimitry Andric 
3132*0b57cec5SDimitry Andric   static char callee_name[] = "get_short_help";
3133*0b57cec5SDimitry Andric 
3134*0b57cec5SDimitry Andric   if (!cmd_obj_sp)
3135*0b57cec5SDimitry Andric     return false;
3136*0b57cec5SDimitry Andric 
3137*0b57cec5SDimitry Andric   PythonObject implementor(PyRefType::Borrowed,
3138*0b57cec5SDimitry Andric                            (PyObject *)cmd_obj_sp->GetValue());
3139*0b57cec5SDimitry Andric 
3140*0b57cec5SDimitry Andric   if (!implementor.IsAllocated())
3141*0b57cec5SDimitry Andric     return false;
3142*0b57cec5SDimitry Andric 
3143*0b57cec5SDimitry Andric   PythonObject pmeth(PyRefType::Owned,
3144*0b57cec5SDimitry Andric                      PyObject_GetAttrString(implementor.get(), callee_name));
3145*0b57cec5SDimitry Andric 
3146*0b57cec5SDimitry Andric   if (PyErr_Occurred())
3147*0b57cec5SDimitry Andric     PyErr_Clear();
3148*0b57cec5SDimitry Andric 
3149*0b57cec5SDimitry Andric   if (!pmeth.IsAllocated())
3150*0b57cec5SDimitry Andric     return false;
3151*0b57cec5SDimitry Andric 
3152*0b57cec5SDimitry Andric   if (PyCallable_Check(pmeth.get()) == 0) {
3153*0b57cec5SDimitry Andric     if (PyErr_Occurred())
3154*0b57cec5SDimitry Andric       PyErr_Clear();
3155*0b57cec5SDimitry Andric     return false;
3156*0b57cec5SDimitry Andric   }
3157*0b57cec5SDimitry Andric 
3158*0b57cec5SDimitry Andric   if (PyErr_Occurred())
3159*0b57cec5SDimitry Andric     PyErr_Clear();
3160*0b57cec5SDimitry Andric 
31615ffd83dbSDimitry Andric   // Right now we know this function exists and is callable.
3162*0b57cec5SDimitry Andric   PythonObject py_return(
3163*0b57cec5SDimitry Andric       PyRefType::Owned,
3164*0b57cec5SDimitry Andric       PyObject_CallMethod(implementor.get(), callee_name, nullptr));
3165*0b57cec5SDimitry Andric 
31665ffd83dbSDimitry Andric   // If it fails, print the error but otherwise go on.
3167*0b57cec5SDimitry Andric   if (PyErr_Occurred()) {
3168*0b57cec5SDimitry Andric     PyErr_Print();
3169*0b57cec5SDimitry Andric     PyErr_Clear();
3170*0b57cec5SDimitry Andric   }
3171*0b57cec5SDimitry Andric 
3172*0b57cec5SDimitry Andric   if (py_return.IsAllocated() && PythonString::Check(py_return.get())) {
3173*0b57cec5SDimitry Andric     PythonString py_string(PyRefType::Borrowed, py_return.get());
3174*0b57cec5SDimitry Andric     llvm::StringRef return_data(py_string.GetString());
3175*0b57cec5SDimitry Andric     dest.assign(return_data.data(), return_data.size());
31765ffd83dbSDimitry Andric     return true;
3177*0b57cec5SDimitry Andric   }
31785ffd83dbSDimitry Andric 
31795ffd83dbSDimitry Andric   return false;
3180*0b57cec5SDimitry Andric }
3181*0b57cec5SDimitry Andric 
3182*0b57cec5SDimitry Andric uint32_t ScriptInterpreterPythonImpl::GetFlagsForCommandObject(
3183*0b57cec5SDimitry Andric     StructuredData::GenericSP cmd_obj_sp) {
3184*0b57cec5SDimitry Andric   uint32_t result = 0;
3185*0b57cec5SDimitry Andric 
3186*0b57cec5SDimitry Andric   Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock);
3187*0b57cec5SDimitry Andric 
3188*0b57cec5SDimitry Andric   static char callee_name[] = "get_flags";
3189*0b57cec5SDimitry Andric 
3190*0b57cec5SDimitry Andric   if (!cmd_obj_sp)
3191*0b57cec5SDimitry Andric     return result;
3192*0b57cec5SDimitry Andric 
3193*0b57cec5SDimitry Andric   PythonObject implementor(PyRefType::Borrowed,
3194*0b57cec5SDimitry Andric                            (PyObject *)cmd_obj_sp->GetValue());
3195*0b57cec5SDimitry Andric 
3196*0b57cec5SDimitry Andric   if (!implementor.IsAllocated())
3197*0b57cec5SDimitry Andric     return result;
3198*0b57cec5SDimitry Andric 
3199*0b57cec5SDimitry Andric   PythonObject pmeth(PyRefType::Owned,
3200*0b57cec5SDimitry Andric                      PyObject_GetAttrString(implementor.get(), callee_name));
3201*0b57cec5SDimitry Andric 
3202*0b57cec5SDimitry Andric   if (PyErr_Occurred())
3203*0b57cec5SDimitry Andric     PyErr_Clear();
3204*0b57cec5SDimitry Andric 
3205*0b57cec5SDimitry Andric   if (!pmeth.IsAllocated())
3206*0b57cec5SDimitry Andric     return result;
3207*0b57cec5SDimitry Andric 
3208*0b57cec5SDimitry Andric   if (PyCallable_Check(pmeth.get()) == 0) {
3209*0b57cec5SDimitry Andric     if (PyErr_Occurred())
3210*0b57cec5SDimitry Andric       PyErr_Clear();
3211*0b57cec5SDimitry Andric     return result;
3212*0b57cec5SDimitry Andric   }
3213*0b57cec5SDimitry Andric 
3214*0b57cec5SDimitry Andric   if (PyErr_Occurred())
3215*0b57cec5SDimitry Andric     PyErr_Clear();
3216*0b57cec5SDimitry Andric 
32175ffd83dbSDimitry Andric   long long py_return = unwrapOrSetPythonException(
32185ffd83dbSDimitry Andric       As<long long>(implementor.CallMethod(callee_name)));
3219*0b57cec5SDimitry Andric 
3220*0b57cec5SDimitry Andric   // if it fails, print the error but otherwise go on
3221*0b57cec5SDimitry Andric   if (PyErr_Occurred()) {
3222*0b57cec5SDimitry Andric     PyErr_Print();
3223*0b57cec5SDimitry Andric     PyErr_Clear();
32245ffd83dbSDimitry Andric   } else {
32255ffd83dbSDimitry Andric     result = py_return;
3226*0b57cec5SDimitry Andric   }
3227*0b57cec5SDimitry Andric 
3228*0b57cec5SDimitry Andric   return result;
3229*0b57cec5SDimitry Andric }
3230*0b57cec5SDimitry Andric 
3231*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GetLongHelpForCommandObject(
3232*0b57cec5SDimitry Andric     StructuredData::GenericSP cmd_obj_sp, std::string &dest) {
3233*0b57cec5SDimitry Andric   bool got_string = false;
3234*0b57cec5SDimitry Andric   dest.clear();
3235*0b57cec5SDimitry Andric 
3236*0b57cec5SDimitry Andric   Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock);
3237*0b57cec5SDimitry Andric 
3238*0b57cec5SDimitry Andric   static char callee_name[] = "get_long_help";
3239*0b57cec5SDimitry Andric 
3240*0b57cec5SDimitry Andric   if (!cmd_obj_sp)
3241*0b57cec5SDimitry Andric     return false;
3242*0b57cec5SDimitry Andric 
3243*0b57cec5SDimitry Andric   PythonObject implementor(PyRefType::Borrowed,
3244*0b57cec5SDimitry Andric                            (PyObject *)cmd_obj_sp->GetValue());
3245*0b57cec5SDimitry Andric 
3246*0b57cec5SDimitry Andric   if (!implementor.IsAllocated())
3247*0b57cec5SDimitry Andric     return false;
3248*0b57cec5SDimitry Andric 
3249*0b57cec5SDimitry Andric   PythonObject pmeth(PyRefType::Owned,
3250*0b57cec5SDimitry Andric                      PyObject_GetAttrString(implementor.get(), callee_name));
3251*0b57cec5SDimitry Andric 
3252*0b57cec5SDimitry Andric   if (PyErr_Occurred())
3253*0b57cec5SDimitry Andric     PyErr_Clear();
3254*0b57cec5SDimitry Andric 
3255*0b57cec5SDimitry Andric   if (!pmeth.IsAllocated())
3256*0b57cec5SDimitry Andric     return false;
3257*0b57cec5SDimitry Andric 
3258*0b57cec5SDimitry Andric   if (PyCallable_Check(pmeth.get()) == 0) {
3259*0b57cec5SDimitry Andric     if (PyErr_Occurred())
3260*0b57cec5SDimitry Andric       PyErr_Clear();
3261*0b57cec5SDimitry Andric 
3262*0b57cec5SDimitry Andric     return false;
3263*0b57cec5SDimitry Andric   }
3264*0b57cec5SDimitry Andric 
3265*0b57cec5SDimitry Andric   if (PyErr_Occurred())
3266*0b57cec5SDimitry Andric     PyErr_Clear();
3267*0b57cec5SDimitry Andric 
3268*0b57cec5SDimitry Andric   // right now we know this function exists and is callable..
3269*0b57cec5SDimitry Andric   PythonObject py_return(
3270*0b57cec5SDimitry Andric       PyRefType::Owned,
3271*0b57cec5SDimitry Andric       PyObject_CallMethod(implementor.get(), callee_name, nullptr));
3272*0b57cec5SDimitry Andric 
3273*0b57cec5SDimitry Andric   // if it fails, print the error but otherwise go on
3274*0b57cec5SDimitry Andric   if (PyErr_Occurred()) {
3275*0b57cec5SDimitry Andric     PyErr_Print();
3276*0b57cec5SDimitry Andric     PyErr_Clear();
3277*0b57cec5SDimitry Andric   }
3278*0b57cec5SDimitry Andric 
3279*0b57cec5SDimitry Andric   if (py_return.IsAllocated() && PythonString::Check(py_return.get())) {
3280*0b57cec5SDimitry Andric     PythonString str(PyRefType::Borrowed, py_return.get());
3281*0b57cec5SDimitry Andric     llvm::StringRef str_data(str.GetString());
3282*0b57cec5SDimitry Andric     dest.assign(str_data.data(), str_data.size());
3283*0b57cec5SDimitry Andric     got_string = true;
3284*0b57cec5SDimitry Andric   }
3285*0b57cec5SDimitry Andric 
3286*0b57cec5SDimitry Andric   return got_string;
3287*0b57cec5SDimitry Andric }
3288*0b57cec5SDimitry Andric 
3289*0b57cec5SDimitry Andric std::unique_ptr<ScriptInterpreterLocker>
3290*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::AcquireInterpreterLock() {
3291*0b57cec5SDimitry Andric   std::unique_ptr<ScriptInterpreterLocker> py_lock(new Locker(
3292*0b57cec5SDimitry Andric       this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN,
3293*0b57cec5SDimitry Andric       Locker::FreeLock | Locker::TearDownSession));
3294*0b57cec5SDimitry Andric   return py_lock;
3295*0b57cec5SDimitry Andric }
3296*0b57cec5SDimitry Andric 
3297*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::InitializePrivate() {
3298*0b57cec5SDimitry Andric   if (g_initialized)
3299*0b57cec5SDimitry Andric     return;
3300*0b57cec5SDimitry Andric 
3301*0b57cec5SDimitry Andric   g_initialized = true;
3302*0b57cec5SDimitry Andric 
3303e8d8bef9SDimitry Andric   LLDB_SCOPED_TIMER();
3304*0b57cec5SDimitry Andric 
3305*0b57cec5SDimitry Andric   // RAII-based initialization which correctly handles multiple-initialization,
3306*0b57cec5SDimitry Andric   // version- specific differences among Python 2 and Python 3, and saving and
3307*0b57cec5SDimitry Andric   // restoring various other pieces of state that can get mucked with during
3308*0b57cec5SDimitry Andric   // initialization.
3309*0b57cec5SDimitry Andric   InitializePythonRAII initialize_guard;
3310*0b57cec5SDimitry Andric 
3311*0b57cec5SDimitry Andric   LLDBSwigPyInit();
3312*0b57cec5SDimitry Andric 
3313*0b57cec5SDimitry Andric   // Update the path python uses to search for modules to include the current
3314*0b57cec5SDimitry Andric   // directory.
3315*0b57cec5SDimitry Andric 
3316*0b57cec5SDimitry Andric   PyRun_SimpleString("import sys");
3317*0b57cec5SDimitry Andric   AddToSysPath(AddLocation::End, ".");
3318*0b57cec5SDimitry Andric 
3319*0b57cec5SDimitry Andric   // Don't denormalize paths when calling file_spec.GetPath().  On platforms
3320*0b57cec5SDimitry Andric   // that use a backslash as the path separator, this will result in executing
3321*0b57cec5SDimitry Andric   // python code containing paths with unescaped backslashes.  But Python also
3322*0b57cec5SDimitry Andric   // accepts forward slashes, so to make life easier we just use that.
3323*0b57cec5SDimitry Andric   if (FileSpec file_spec = GetPythonDir())
3324*0b57cec5SDimitry Andric     AddToSysPath(AddLocation::Beginning, file_spec.GetPath(false));
3325*0b57cec5SDimitry Andric   if (FileSpec file_spec = HostInfo::GetShlibDir())
3326*0b57cec5SDimitry Andric     AddToSysPath(AddLocation::Beginning, file_spec.GetPath(false));
3327*0b57cec5SDimitry Andric 
3328*0b57cec5SDimitry Andric   PyRun_SimpleString("sys.dont_write_bytecode = 1; import "
3329*0b57cec5SDimitry Andric                      "lldb.embedded_interpreter; from "
3330*0b57cec5SDimitry Andric                      "lldb.embedded_interpreter import run_python_interpreter; "
3331*0b57cec5SDimitry Andric                      "from lldb.embedded_interpreter import run_one_line");
3332*0b57cec5SDimitry Andric }
3333*0b57cec5SDimitry Andric 
3334*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::AddToSysPath(AddLocation location,
3335*0b57cec5SDimitry Andric                                                std::string path) {
3336*0b57cec5SDimitry Andric   std::string path_copy;
3337*0b57cec5SDimitry Andric 
3338*0b57cec5SDimitry Andric   std::string statement;
3339*0b57cec5SDimitry Andric   if (location == AddLocation::Beginning) {
3340*0b57cec5SDimitry Andric     statement.assign("sys.path.insert(0,\"");
3341*0b57cec5SDimitry Andric     statement.append(path);
3342*0b57cec5SDimitry Andric     statement.append("\")");
3343*0b57cec5SDimitry Andric   } else {
3344*0b57cec5SDimitry Andric     statement.assign("sys.path.append(\"");
3345*0b57cec5SDimitry Andric     statement.append(path);
3346*0b57cec5SDimitry Andric     statement.append("\")");
3347*0b57cec5SDimitry Andric   }
3348*0b57cec5SDimitry Andric   PyRun_SimpleString(statement.c_str());
3349*0b57cec5SDimitry Andric }
3350*0b57cec5SDimitry Andric 
3351*0b57cec5SDimitry Andric // We are intentionally NOT calling Py_Finalize here (this would be the logical
3352*0b57cec5SDimitry Andric // place to call it).  Calling Py_Finalize here causes test suite runs to seg
3353*0b57cec5SDimitry Andric // fault:  The test suite runs in Python.  It registers SBDebugger::Terminate to
3354*0b57cec5SDimitry Andric // be called 'at_exit'.  When the test suite Python harness finishes up, it
3355*0b57cec5SDimitry Andric // calls Py_Finalize, which calls all the 'at_exit' registered functions.
3356*0b57cec5SDimitry Andric // SBDebugger::Terminate calls Debugger::Terminate, which calls lldb::Terminate,
3357*0b57cec5SDimitry Andric // which calls ScriptInterpreter::Terminate, which calls
3358*0b57cec5SDimitry Andric // ScriptInterpreterPythonImpl::Terminate.  So if we call Py_Finalize here, we
3359*0b57cec5SDimitry Andric // end up with Py_Finalize being called from within Py_Finalize, which results
3360*0b57cec5SDimitry Andric // in a seg fault. Since this function only gets called when lldb is shutting
3361*0b57cec5SDimitry Andric // down and going away anyway, the fact that we don't actually call Py_Finalize
3362*0b57cec5SDimitry Andric // should not cause any problems (everything should shut down/go away anyway
3363*0b57cec5SDimitry Andric // when the process exits).
3364*0b57cec5SDimitry Andric //
3365*0b57cec5SDimitry Andric // void ScriptInterpreterPythonImpl::Terminate() { Py_Finalize (); }
3366*0b57cec5SDimitry Andric 
3367480093f4SDimitry Andric #endif
3368