12c1f46dcSZachary Turner //===-- ScriptInterpreterPython.cpp -----------------------------*- C++ -*-===//
22c1f46dcSZachary Turner //
32c1f46dcSZachary Turner //                     The LLVM Compiler Infrastructure
42c1f46dcSZachary Turner //
52c1f46dcSZachary Turner // This file is distributed under the University of Illinois Open Source
62c1f46dcSZachary Turner // License. See LICENSE.TXT for details.
72c1f46dcSZachary Turner //
82c1f46dcSZachary Turner //===----------------------------------------------------------------------===//
92c1f46dcSZachary Turner 
102c1f46dcSZachary Turner #ifdef LLDB_DISABLE_PYTHON
112c1f46dcSZachary Turner 
122c1f46dcSZachary Turner // Python is disabled in this build
132c1f46dcSZachary Turner 
142c1f46dcSZachary Turner #else
152c1f46dcSZachary Turner 
162c1f46dcSZachary Turner #include "lldb-python.h"
172c1f46dcSZachary Turner #include "ScriptInterpreterPython.h"
182c1f46dcSZachary Turner #include "PythonDataObjects.h"
192c1f46dcSZachary Turner 
202c1f46dcSZachary Turner #include <stdlib.h>
212c1f46dcSZachary Turner #include <stdio.h>
222c1f46dcSZachary Turner 
232c1f46dcSZachary Turner #include <mutex>
242c1f46dcSZachary Turner #include <string>
252c1f46dcSZachary Turner 
262c1f46dcSZachary Turner #include "lldb/API/SBValue.h"
272c1f46dcSZachary Turner #include "lldb/Breakpoint/BreakpointLocation.h"
282c1f46dcSZachary Turner #include "lldb/Breakpoint/StoppointCallbackContext.h"
292c1f46dcSZachary Turner #include "lldb/Breakpoint/WatchpointOptions.h"
302c1f46dcSZachary Turner #include "lldb/Core/Communication.h"
312c1f46dcSZachary Turner #include "lldb/Core/Debugger.h"
322c1f46dcSZachary Turner #include "lldb/Core/PluginManager.h"
332c1f46dcSZachary Turner #include "lldb/Core/Timer.h"
342c1f46dcSZachary Turner #include "lldb/Core/ValueObject.h"
352c1f46dcSZachary Turner #include "lldb/DataFormatters/TypeSummary.h"
362c1f46dcSZachary Turner #include "lldb/Host/ConnectionFileDescriptor.h"
374eff2d31SZachary Turner #include "lldb/Host/FileSystem.h"
382c1f46dcSZachary Turner #include "lldb/Host/HostInfo.h"
392c1f46dcSZachary Turner #include "lldb/Host/Pipe.h"
402c1f46dcSZachary Turner #include "lldb/Interpreter/CommandInterpreter.h"
412c1f46dcSZachary Turner #include "lldb/Interpreter/CommandReturnObject.h"
422c1f46dcSZachary Turner #include "lldb/Target/Thread.h"
432c1f46dcSZachary Turner #include "lldb/Target/ThreadPlan.h"
442c1f46dcSZachary Turner 
452c1f46dcSZachary Turner #if defined(_WIN32)
462c1f46dcSZachary Turner #include "lldb/Host/windows/ConnectionGenericFileWindows.h"
472c1f46dcSZachary Turner #endif
482c1f46dcSZachary Turner 
492c1f46dcSZachary Turner #include "llvm/ADT/StringRef.h"
5022c8efcdSZachary Turner #include "llvm/ADT/STLExtras.h"
512c1f46dcSZachary Turner 
522c1f46dcSZachary Turner using namespace lldb;
532c1f46dcSZachary Turner using namespace lldb_private;
542c1f46dcSZachary Turner 
552c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGInitCallback g_swig_init_callback = nullptr;
562c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGBreakpointCallbackFunction g_swig_breakpoint_callback = nullptr;
572c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGWatchpointCallbackFunction g_swig_watchpoint_callback = nullptr;
582c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonTypeScriptCallbackFunction g_swig_typescript_callback = nullptr;
592c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonCreateSyntheticProvider g_swig_synthetic_script = nullptr;
602c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonCreateCommandObject g_swig_create_cmd = nullptr;
612c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonCalculateNumChildren g_swig_calc_children = nullptr;
622c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonGetChildAtIndex g_swig_get_child_index = nullptr;
632c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonGetIndexOfChildWithName g_swig_get_index_child = nullptr;
642c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonCastPyObjectToSBValue g_swig_cast_to_sbvalue  = nullptr;
652c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonGetValueObjectSPFromSBValue g_swig_get_valobj_sp_from_sbvalue = nullptr;
662c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonUpdateSynthProviderInstance g_swig_update_provider = nullptr;
672c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonMightHaveChildrenSynthProviderInstance g_swig_mighthavechildren_provider = nullptr;
682c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonGetValueSynthProviderInstance g_swig_getvalue_provider = nullptr;
692c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonCallCommand g_swig_call_command = nullptr;
702c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonCallCommandObject g_swig_call_command_object = nullptr;
712c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonCallModuleInit g_swig_call_module_init = nullptr;
722c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonCreateOSPlugin g_swig_create_os_plugin = nullptr;
732c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonScriptKeyword_Process g_swig_run_script_keyword_process = nullptr;
742c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonScriptKeyword_Thread g_swig_run_script_keyword_thread = nullptr;
752c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonScriptKeyword_Target g_swig_run_script_keyword_target = nullptr;
762c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonScriptKeyword_Frame g_swig_run_script_keyword_frame = nullptr;
772c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonScriptKeyword_Value g_swig_run_script_keyword_value = nullptr;
782c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPython_GetDynamicSetting g_swig_plugin_get = nullptr;
792c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonCreateScriptedThreadPlan g_swig_thread_plan_script = nullptr;
802c1f46dcSZachary Turner static ScriptInterpreterPython::SWIGPythonCallThreadPlan g_swig_call_thread_plan = nullptr;
812c1f46dcSZachary Turner 
822c1f46dcSZachary Turner static bool g_initialized = false;
832c1f46dcSZachary Turner 
84079fe48aSZachary Turner namespace
85079fe48aSZachary Turner {
8622c8efcdSZachary Turner 
87079fe48aSZachary Turner // Initializing Python is not a straightforward process.  We cannot control what
88079fe48aSZachary Turner // external code may have done before getting to this point in LLDB, including
89079fe48aSZachary Turner // potentially having already initialized Python, so we need to do a lot of work
90079fe48aSZachary Turner // to ensure that the existing state of the system is maintained across our
91079fe48aSZachary Turner // initialization.  We do this by using an RAII pattern where we save off initial
92079fe48aSZachary Turner // state at the beginning, and restore it at the end
93079fe48aSZachary Turner struct InitializePythonRAII
94079fe48aSZachary Turner {
95079fe48aSZachary Turner public:
96079fe48aSZachary Turner     InitializePythonRAII() :
9776866446SStephane Sezer         m_gil_state(PyGILState_UNLOCKED),
9876866446SStephane Sezer         m_was_already_initialized(false)
99079fe48aSZachary Turner     {
100079fe48aSZachary Turner         // Python will muck with STDIN terminal state, so save off any current TTY
101079fe48aSZachary Turner         // settings so we can restore them.
102079fe48aSZachary Turner         m_stdin_tty_state.Save(STDIN_FILENO, false);
103079fe48aSZachary Turner 
104079fe48aSZachary Turner         InitializePythonHome();
105079fe48aSZachary Turner 
106079fe48aSZachary Turner         // Python < 3.2 and Python >= 3.2 reversed the ordering requirements for
107079fe48aSZachary Turner         // calling `Py_Initialize` and `PyEval_InitThreads`.  < 3.2 requires that you
108079fe48aSZachary Turner         // call `PyEval_InitThreads` first, and >= 3.2 requires that you call it last.
109079fe48aSZachary Turner #if (PY_MAJOR_VERSION == 3 && PY_MINOR_VERSION >= 2) || (PY_MAJOR_VERSION > 3)
110079fe48aSZachary Turner         Py_InitializeEx(0);
111079fe48aSZachary Turner         InitializeThreadsPrivate();
112079fe48aSZachary Turner #else
113079fe48aSZachary Turner         InitializeThreadsPrivate();
114079fe48aSZachary Turner         Py_InitializeEx(0);
115079fe48aSZachary Turner #endif
116079fe48aSZachary Turner     }
117079fe48aSZachary Turner 
118079fe48aSZachary Turner     ~InitializePythonRAII()
119079fe48aSZachary Turner     {
120079fe48aSZachary Turner         if (m_was_already_initialized)
121079fe48aSZachary Turner         {
122079fe48aSZachary Turner             Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT | LIBLLDB_LOG_VERBOSE));
123079fe48aSZachary Turner 
124079fe48aSZachary Turner             if (log)
125079fe48aSZachary Turner             {
126079fe48aSZachary Turner                 log->Printf("Releasing PyGILState. Returning to state = %slocked\n",
127079fe48aSZachary Turner                     m_was_already_initialized == PyGILState_UNLOCKED ? "un" : "");
128079fe48aSZachary Turner             }
129079fe48aSZachary Turner             PyGILState_Release(m_gil_state);
130079fe48aSZachary Turner         }
131079fe48aSZachary Turner         else
132079fe48aSZachary Turner         {
133079fe48aSZachary Turner             // We initialized the threads in this function, just unlock the GIL.
134079fe48aSZachary Turner             PyEval_SaveThread();
135079fe48aSZachary Turner         }
136079fe48aSZachary Turner 
137079fe48aSZachary Turner         m_stdin_tty_state.Restore();
138079fe48aSZachary Turner     }
139079fe48aSZachary Turner 
140079fe48aSZachary Turner private:
141079fe48aSZachary Turner     void InitializePythonHome()
14222c8efcdSZachary Turner     {
14322c8efcdSZachary Turner #if defined(LLDB_PYTHON_HOME)
14422c8efcdSZachary Turner #if PY_MAJOR_VERSION >= 3
14522c8efcdSZachary Turner         size_t size = 0;
146079fe48aSZachary Turner         static wchar_t *g_python_home = Py_DecodeLocale(LLDB_PYTHON_HOME, &size);
14722c8efcdSZachary Turner #else
148079fe48aSZachary Turner         static char *g_python_home = LLDB_PYTHON_HOME;
14922c8efcdSZachary Turner #endif
150079fe48aSZachary Turner         Py_SetPythonHome(g_python_home);
15122c8efcdSZachary Turner #endif
15222c8efcdSZachary Turner     }
15322c8efcdSZachary Turner 
154079fe48aSZachary Turner     void InitializeThreadsPrivate()
155079fe48aSZachary Turner     {
156079fe48aSZachary Turner         if (PyEval_ThreadsInitialized())
157079fe48aSZachary Turner         {
158079fe48aSZachary Turner             Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT | LIBLLDB_LOG_VERBOSE));
159079fe48aSZachary Turner 
160079fe48aSZachary Turner             m_was_already_initialized = true;
161079fe48aSZachary Turner             m_gil_state = PyGILState_Ensure();
162079fe48aSZachary Turner             if (log)
163079fe48aSZachary Turner             {
164079fe48aSZachary Turner                 log->Printf("Ensured PyGILState. Previous state = %slocked\n",
165079fe48aSZachary Turner                     m_gil_state == PyGILState_UNLOCKED ? "un" : "");
166079fe48aSZachary Turner             }
167079fe48aSZachary Turner             return;
168079fe48aSZachary Turner         }
169079fe48aSZachary Turner 
170079fe48aSZachary Turner         // InitThreads acquires the GIL if it hasn't been called before.
171079fe48aSZachary Turner         PyEval_InitThreads();
172079fe48aSZachary Turner     }
173079fe48aSZachary Turner 
174079fe48aSZachary Turner     TerminalState m_stdin_tty_state;
175079fe48aSZachary Turner     PyGILState_STATE m_gil_state;
176079fe48aSZachary Turner     bool m_was_already_initialized;
177079fe48aSZachary Turner };
178079fe48aSZachary Turner 
179079fe48aSZachary Turner }
180079fe48aSZachary Turner 
181f8b22f8fSZachary Turner static std::string ReadPythonBacktrace(PythonObject py_backtrace);
1822c1f46dcSZachary Turner 
1832c1f46dcSZachary Turner ScriptInterpreterPython::Locker::Locker (ScriptInterpreterPython *py_interpreter,
1842c1f46dcSZachary Turner                                          uint16_t on_entry,
1852c1f46dcSZachary Turner                                          uint16_t on_leave,
1862c1f46dcSZachary Turner                                          FILE *in,
1872c1f46dcSZachary Turner                                          FILE *out,
1882c1f46dcSZachary Turner                                          FILE *err) :
1892c1f46dcSZachary Turner     ScriptInterpreterLocker (),
1902c1f46dcSZachary Turner     m_teardown_session( (on_leave & TearDownSession) == TearDownSession ),
1912c1f46dcSZachary Turner     m_python_interpreter(py_interpreter)
1922c1f46dcSZachary Turner {
1932c1f46dcSZachary Turner     DoAcquireLock();
1942c1f46dcSZachary Turner     if ((on_entry & InitSession) == InitSession)
1952c1f46dcSZachary Turner     {
1962c1f46dcSZachary Turner         if (DoInitSession(on_entry, in, out, err) == false)
1972c1f46dcSZachary Turner         {
1982c1f46dcSZachary Turner             // Don't teardown the session if we didn't init it.
1992c1f46dcSZachary Turner             m_teardown_session = false;
2002c1f46dcSZachary Turner         }
2012c1f46dcSZachary Turner     }
2022c1f46dcSZachary Turner }
2032c1f46dcSZachary Turner 
2042c1f46dcSZachary Turner bool
2052c1f46dcSZachary Turner ScriptInterpreterPython::Locker::DoAcquireLock()
2062c1f46dcSZachary Turner {
2072c1f46dcSZachary Turner     Log *log (lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_SCRIPT | LIBLLDB_LOG_VERBOSE));
2082c1f46dcSZachary Turner     m_GILState = PyGILState_Ensure();
2092c1f46dcSZachary Turner     if (log)
2102c1f46dcSZachary Turner         log->Printf("Ensured PyGILState. Previous state = %slocked\n", m_GILState == PyGILState_UNLOCKED ? "un" : "");
2112c1f46dcSZachary Turner 
2122c1f46dcSZachary Turner     // we need to save the thread state when we first start the command
2132c1f46dcSZachary Turner     // because we might decide to interrupt it while some action is taking
2142c1f46dcSZachary Turner     // place outside of Python (e.g. printing to screen, waiting for the network, ...)
2152c1f46dcSZachary Turner     // in that case, _PyThreadState_Current will be NULL - and we would be unable
2162c1f46dcSZachary Turner     // to set the asynchronous exception - not a desirable situation
21722c8efcdSZachary Turner     m_python_interpreter->SetThreadState(PyThreadState_Get());
2182c1f46dcSZachary Turner     m_python_interpreter->IncrementLockCount();
2192c1f46dcSZachary Turner     return true;
2202c1f46dcSZachary Turner }
2212c1f46dcSZachary Turner 
2222c1f46dcSZachary Turner bool
2232c1f46dcSZachary Turner ScriptInterpreterPython::Locker::DoInitSession(uint16_t on_entry_flags, FILE *in, FILE *out, FILE *err)
2242c1f46dcSZachary Turner {
2252c1f46dcSZachary Turner     if (!m_python_interpreter)
2262c1f46dcSZachary Turner         return false;
2272c1f46dcSZachary Turner     return m_python_interpreter->EnterSession (on_entry_flags, in, out, err);
2282c1f46dcSZachary Turner }
2292c1f46dcSZachary Turner 
2302c1f46dcSZachary Turner bool
2312c1f46dcSZachary Turner ScriptInterpreterPython::Locker::DoFreeLock()
2322c1f46dcSZachary Turner {
2332c1f46dcSZachary Turner     Log *log (lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_SCRIPT | LIBLLDB_LOG_VERBOSE));
2342c1f46dcSZachary Turner     if (log)
2352c1f46dcSZachary Turner         log->Printf("Releasing PyGILState. Returning to state = %slocked\n", m_GILState == PyGILState_UNLOCKED ? "un" : "");
2362c1f46dcSZachary Turner     PyGILState_Release(m_GILState);
2372c1f46dcSZachary Turner     m_python_interpreter->DecrementLockCount();
2382c1f46dcSZachary Turner     return true;
2392c1f46dcSZachary Turner }
2402c1f46dcSZachary Turner 
2412c1f46dcSZachary Turner bool
2422c1f46dcSZachary Turner ScriptInterpreterPython::Locker::DoTearDownSession()
2432c1f46dcSZachary Turner {
2442c1f46dcSZachary Turner     if (!m_python_interpreter)
2452c1f46dcSZachary Turner         return false;
2462c1f46dcSZachary Turner     m_python_interpreter->LeaveSession ();
2472c1f46dcSZachary Turner     return true;
2482c1f46dcSZachary Turner }
2492c1f46dcSZachary Turner 
2502c1f46dcSZachary Turner ScriptInterpreterPython::Locker::~Locker()
2512c1f46dcSZachary Turner {
2522c1f46dcSZachary Turner     if (m_teardown_session)
2532c1f46dcSZachary Turner         DoTearDownSession();
2542c1f46dcSZachary Turner     DoFreeLock();
2552c1f46dcSZachary Turner }
2562c1f46dcSZachary Turner 
2572c1f46dcSZachary Turner ScriptInterpreterPython::ScriptInterpreterPython(CommandInterpreter &interpreter) :
2582c1f46dcSZachary Turner     ScriptInterpreter(interpreter, eScriptLanguagePython),
2592c1f46dcSZachary Turner     IOHandlerDelegateMultiline("DONE"),
2602c1f46dcSZachary Turner     m_saved_stdin(),
2612c1f46dcSZachary Turner     m_saved_stdout(),
2622c1f46dcSZachary Turner     m_saved_stderr(),
2632c1f46dcSZachary Turner     m_main_module(),
2642c1f46dcSZachary Turner     m_lldb_module(),
265f8b22f8fSZachary Turner     m_session_dict(PyInitialValue::Invalid),
266f8b22f8fSZachary Turner     m_sys_module_dict(PyInitialValue::Invalid),
2672c1f46dcSZachary Turner     m_run_one_line_function(),
2682c1f46dcSZachary Turner     m_run_one_line_str_global(),
2692c1f46dcSZachary Turner     m_dictionary_name(interpreter.GetDebugger().GetInstanceName().AsCString()),
2702c1f46dcSZachary Turner     m_terminal_state(),
2712c1f46dcSZachary Turner     m_active_io_handler(eIOHandlerNone),
2722c1f46dcSZachary Turner     m_session_is_active(false),
2732c1f46dcSZachary Turner     m_pty_slave_is_open(false),
2742c1f46dcSZachary Turner     m_valid_session(true),
2752c1f46dcSZachary Turner     m_lock_count(0),
2762c1f46dcSZachary Turner     m_command_thread_state(nullptr)
2772c1f46dcSZachary Turner {
2782c1f46dcSZachary Turner     assert(g_initialized && "ScriptInterpreterPython created but InitializePrivate has not been called!");
2792c1f46dcSZachary Turner 
2802c1f46dcSZachary Turner     m_dictionary_name.append("_dict");
2812c1f46dcSZachary Turner     StreamString run_string;
2822c1f46dcSZachary Turner     run_string.Printf ("%s = dict()", m_dictionary_name.c_str());
2832c1f46dcSZachary Turner 
2842c1f46dcSZachary Turner     Locker locker(this,
2852c1f46dcSZachary Turner                   ScriptInterpreterPython::Locker::AcquireLock,
2862c1f46dcSZachary Turner                   ScriptInterpreterPython::Locker::FreeAcquiredLock);
2872c1f46dcSZachary Turner     PyRun_SimpleString (run_string.GetData());
2882c1f46dcSZachary Turner 
2892c1f46dcSZachary Turner     run_string.Clear();
2902c1f46dcSZachary Turner 
2912c1f46dcSZachary Turner     run_string.Printf ("run_one_line (%s, 'import copy, keyword, os, re, sys, uuid, lldb')", m_dictionary_name.c_str());
2922c1f46dcSZachary Turner     PyRun_SimpleString (run_string.GetData());
2932c1f46dcSZachary Turner 
2942c1f46dcSZachary Turner     // WARNING: temporary code that loads Cocoa formatters - this should be done on a per-platform basis rather than loading the whole set
2952c1f46dcSZachary Turner     // and letting the individual formatter classes exploit APIs to check whether they can/cannot do their task
2962c1f46dcSZachary Turner     run_string.Clear();
2972c1f46dcSZachary Turner     run_string.Printf ("run_one_line (%s, 'import lldb.formatters, lldb.formatters.cpp, pydoc')", m_dictionary_name.c_str());
2982c1f46dcSZachary Turner     PyRun_SimpleString (run_string.GetData());
2992c1f46dcSZachary Turner     run_string.Clear();
3002c1f46dcSZachary Turner 
3012c1f46dcSZachary Turner     run_string.Printf ("run_one_line (%s, 'import lldb.embedded_interpreter; from lldb.embedded_interpreter import run_python_interpreter; from lldb.embedded_interpreter import run_one_line')", m_dictionary_name.c_str());
3022c1f46dcSZachary Turner     PyRun_SimpleString (run_string.GetData());
3032c1f46dcSZachary Turner     run_string.Clear();
3042c1f46dcSZachary Turner 
3052c1f46dcSZachary Turner     run_string.Printf ("run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64 "; pydoc.pager = pydoc.plainpager')", m_dictionary_name.c_str(),
3062c1f46dcSZachary Turner                        interpreter.GetDebugger().GetID());
3072c1f46dcSZachary Turner     PyRun_SimpleString (run_string.GetData());
3082c1f46dcSZachary Turner }
3092c1f46dcSZachary Turner 
3102c1f46dcSZachary Turner ScriptInterpreterPython::~ScriptInterpreterPython ()
3112c1f46dcSZachary Turner {
3122c1f46dcSZachary Turner     // the session dictionary may hold objects with complex state
3132c1f46dcSZachary Turner     // which means that they may need to be torn down with some level of smarts
3142c1f46dcSZachary Turner     // and that, in turn, requires a valid thread state
3152c1f46dcSZachary Turner     // force Python to procure itself such a thread state, nuke the session dictionary
3162c1f46dcSZachary Turner     // and then release it for others to use and proceed with the rest of the shutdown
3172c1f46dcSZachary Turner     auto gil_state = PyGILState_Ensure();
3182c1f46dcSZachary Turner     m_session_dict.Reset();
3192c1f46dcSZachary Turner     PyGILState_Release(gil_state);
3202c1f46dcSZachary Turner }
3212c1f46dcSZachary Turner 
3222c1f46dcSZachary Turner void
3232c1f46dcSZachary Turner ScriptInterpreterPython::Initialize()
3242c1f46dcSZachary Turner {
3252c1f46dcSZachary Turner     static std::once_flag g_once_flag;
3262c1f46dcSZachary Turner 
3272c1f46dcSZachary Turner     std::call_once(g_once_flag, []()
3282c1f46dcSZachary Turner     {
3292c1f46dcSZachary Turner         InitializePrivate();
3302c1f46dcSZachary Turner 
3312c1f46dcSZachary Turner         PluginManager::RegisterPlugin(GetPluginNameStatic(),
3322c1f46dcSZachary Turner                                       GetPluginDescriptionStatic(),
3332c1f46dcSZachary Turner                                       lldb::eScriptLanguagePython,
3342c1f46dcSZachary Turner                                       CreateInstance);
3352c1f46dcSZachary Turner     });
3362c1f46dcSZachary Turner }
3372c1f46dcSZachary Turner 
3382c1f46dcSZachary Turner void
3392c1f46dcSZachary Turner ScriptInterpreterPython::Terminate()
3402c1f46dcSZachary Turner {
3412c1f46dcSZachary Turner 
3422c1f46dcSZachary Turner }
3432c1f46dcSZachary Turner 
3442c1f46dcSZachary Turner lldb::ScriptInterpreterSP
3452c1f46dcSZachary Turner ScriptInterpreterPython::CreateInstance(CommandInterpreter &interpreter)
3462c1f46dcSZachary Turner {
3472c1f46dcSZachary Turner     return std::make_shared<ScriptInterpreterPython>(interpreter);
3482c1f46dcSZachary Turner }
3492c1f46dcSZachary Turner 
3502c1f46dcSZachary Turner lldb_private::ConstString
3512c1f46dcSZachary Turner ScriptInterpreterPython::GetPluginNameStatic()
3522c1f46dcSZachary Turner {
3532c1f46dcSZachary Turner     static ConstString g_name("script-python");
3542c1f46dcSZachary Turner     return g_name;
3552c1f46dcSZachary Turner }
3562c1f46dcSZachary Turner 
3572c1f46dcSZachary Turner const char *
3582c1f46dcSZachary Turner ScriptInterpreterPython::GetPluginDescriptionStatic()
3592c1f46dcSZachary Turner {
3602c1f46dcSZachary Turner     return "Embedded Python interpreter";
3612c1f46dcSZachary Turner }
3622c1f46dcSZachary Turner 
3632c1f46dcSZachary Turner lldb_private::ConstString
3642c1f46dcSZachary Turner ScriptInterpreterPython::GetPluginName()
3652c1f46dcSZachary Turner {
3662c1f46dcSZachary Turner     return GetPluginNameStatic();
3672c1f46dcSZachary Turner }
3682c1f46dcSZachary Turner 
3692c1f46dcSZachary Turner uint32_t
3702c1f46dcSZachary Turner ScriptInterpreterPython::GetPluginVersion()
3712c1f46dcSZachary Turner {
3722c1f46dcSZachary Turner     return 1;
3732c1f46dcSZachary Turner }
3742c1f46dcSZachary Turner 
3752c1f46dcSZachary Turner void
3762c1f46dcSZachary Turner ScriptInterpreterPython::IOHandlerActivated (IOHandler &io_handler)
3772c1f46dcSZachary Turner {
3782c1f46dcSZachary Turner     const char *instructions = nullptr;
3792c1f46dcSZachary Turner 
3802c1f46dcSZachary Turner     switch (m_active_io_handler)
3812c1f46dcSZachary Turner     {
3822c1f46dcSZachary Turner     case eIOHandlerNone:
3832c1f46dcSZachary Turner             break;
3842c1f46dcSZachary Turner     case eIOHandlerBreakpoint:
3852c1f46dcSZachary Turner             instructions = R"(Enter your Python command(s). Type 'DONE' to end.
3862c1f46dcSZachary Turner def function (frame, bp_loc, internal_dict):
3872c1f46dcSZachary Turner     """frame: the lldb.SBFrame for the location at which you stopped
3882c1f46dcSZachary Turner        bp_loc: an lldb.SBBreakpointLocation for the breakpoint location information
3892c1f46dcSZachary Turner        internal_dict: an LLDB support object not to be used"""
3902c1f46dcSZachary Turner )";
3912c1f46dcSZachary Turner             break;
3922c1f46dcSZachary Turner     case eIOHandlerWatchpoint:
3932c1f46dcSZachary Turner             instructions = "Enter your Python command(s). Type 'DONE' to end.\n";
3942c1f46dcSZachary Turner             break;
3952c1f46dcSZachary Turner     }
3962c1f46dcSZachary Turner 
3972c1f46dcSZachary Turner     if (instructions)
3982c1f46dcSZachary Turner     {
3992c1f46dcSZachary Turner         StreamFileSP output_sp(io_handler.GetOutputStreamFile());
4002c1f46dcSZachary Turner         if (output_sp)
4012c1f46dcSZachary Turner         {
4022c1f46dcSZachary Turner             output_sp->PutCString(instructions);
4032c1f46dcSZachary Turner             output_sp->Flush();
4042c1f46dcSZachary Turner         }
4052c1f46dcSZachary Turner     }
4062c1f46dcSZachary Turner }
4072c1f46dcSZachary Turner 
4082c1f46dcSZachary Turner void
4092c1f46dcSZachary Turner ScriptInterpreterPython::IOHandlerInputComplete (IOHandler &io_handler, std::string &data)
4102c1f46dcSZachary Turner {
4112c1f46dcSZachary Turner     io_handler.SetIsDone(true);
4122c1f46dcSZachary Turner     bool batch_mode = m_interpreter.GetBatchCommandMode();
4132c1f46dcSZachary Turner 
4142c1f46dcSZachary Turner     switch (m_active_io_handler)
4152c1f46dcSZachary Turner     {
4162c1f46dcSZachary Turner     case eIOHandlerNone:
4172c1f46dcSZachary Turner         break;
4182c1f46dcSZachary Turner     case eIOHandlerBreakpoint:
4192c1f46dcSZachary Turner         {
4202c1f46dcSZachary Turner             std::vector<BreakpointOptions *> *bp_options_vec = (std::vector<BreakpointOptions *> *)io_handler.GetUserData();
4212c1f46dcSZachary Turner             for (auto bp_options : *bp_options_vec)
4222c1f46dcSZachary Turner             {
4232c1f46dcSZachary Turner                 if (!bp_options)
4242c1f46dcSZachary Turner                     continue;
4252c1f46dcSZachary Turner 
4262c1f46dcSZachary Turner                 std::unique_ptr<BreakpointOptions::CommandData> data_ap(new BreakpointOptions::CommandData());
4272c1f46dcSZachary Turner                 if (data_ap.get())
4282c1f46dcSZachary Turner                 {
4292c1f46dcSZachary Turner                     data_ap->user_source.SplitIntoLines(data);
4302c1f46dcSZachary Turner 
4312c1f46dcSZachary Turner                     if (GenerateBreakpointCommandCallbackData (data_ap->user_source, data_ap->script_source).Success())
4322c1f46dcSZachary Turner                     {
4332c1f46dcSZachary Turner                         BatonSP baton_sp (new BreakpointOptions::CommandBaton (data_ap.release()));
4342c1f46dcSZachary Turner                         bp_options->SetCallback (ScriptInterpreterPython::BreakpointCallbackFunction, baton_sp);
4352c1f46dcSZachary Turner                     }
4362c1f46dcSZachary Turner                     else if (!batch_mode)
4372c1f46dcSZachary Turner                     {
4382c1f46dcSZachary Turner                         StreamFileSP error_sp = io_handler.GetErrorStreamFile();
4392c1f46dcSZachary Turner                         if (error_sp)
4402c1f46dcSZachary Turner                         {
4412c1f46dcSZachary Turner                             error_sp->Printf ("Warning: No command attached to breakpoint.\n");
4422c1f46dcSZachary Turner                             error_sp->Flush();
4432c1f46dcSZachary Turner                         }
4442c1f46dcSZachary Turner                     }
4452c1f46dcSZachary Turner                 }
4462c1f46dcSZachary Turner             }
4472c1f46dcSZachary Turner             m_active_io_handler = eIOHandlerNone;
4482c1f46dcSZachary Turner         }
4492c1f46dcSZachary Turner         break;
4502c1f46dcSZachary Turner     case eIOHandlerWatchpoint:
4512c1f46dcSZachary Turner         {
4522c1f46dcSZachary Turner             WatchpointOptions *wp_options = (WatchpointOptions *)io_handler.GetUserData();
4532c1f46dcSZachary Turner             std::unique_ptr<WatchpointOptions::CommandData> data_ap(new WatchpointOptions::CommandData());
4542c1f46dcSZachary Turner             if (data_ap.get())
4552c1f46dcSZachary Turner             {
4562c1f46dcSZachary Turner                 data_ap->user_source.SplitIntoLines(data);
4572c1f46dcSZachary Turner 
4582c1f46dcSZachary Turner                 if (GenerateWatchpointCommandCallbackData (data_ap->user_source, data_ap->script_source))
4592c1f46dcSZachary Turner                 {
4602c1f46dcSZachary Turner                     BatonSP baton_sp (new WatchpointOptions::CommandBaton (data_ap.release()));
4612c1f46dcSZachary Turner                     wp_options->SetCallback (ScriptInterpreterPython::WatchpointCallbackFunction, baton_sp);
4622c1f46dcSZachary Turner                 }
4632c1f46dcSZachary Turner                 else if (!batch_mode)
4642c1f46dcSZachary Turner                 {
4652c1f46dcSZachary Turner                     StreamFileSP error_sp = io_handler.GetErrorStreamFile();
4662c1f46dcSZachary Turner                     if (error_sp)
4672c1f46dcSZachary Turner                     {
4682c1f46dcSZachary Turner                         error_sp->Printf ("Warning: No command attached to breakpoint.\n");
4692c1f46dcSZachary Turner                         error_sp->Flush();
4702c1f46dcSZachary Turner                     }
4712c1f46dcSZachary Turner                 }
4722c1f46dcSZachary Turner             }
4732c1f46dcSZachary Turner             m_active_io_handler = eIOHandlerNone;
4742c1f46dcSZachary Turner         }
4752c1f46dcSZachary Turner         break;
4762c1f46dcSZachary Turner     }
4772c1f46dcSZachary Turner }
4782c1f46dcSZachary Turner 
4792c1f46dcSZachary Turner 
4802c1f46dcSZachary Turner void
4812c1f46dcSZachary Turner ScriptInterpreterPython::ResetOutputFileHandle (FILE *fh)
4822c1f46dcSZachary Turner {
4832c1f46dcSZachary Turner }
4842c1f46dcSZachary Turner 
4852c1f46dcSZachary Turner void
4862c1f46dcSZachary Turner ScriptInterpreterPython::SaveTerminalState (int fd)
4872c1f46dcSZachary Turner {
4882c1f46dcSZachary Turner     // Python mucks with the terminal state of STDIN. If we can possibly avoid
4892c1f46dcSZachary Turner     // this by setting the file handles up correctly prior to entering the
4902c1f46dcSZachary Turner     // interpreter we should. For now we save and restore the terminal state
4912c1f46dcSZachary Turner     // on the input file handle.
4922c1f46dcSZachary Turner     m_terminal_state.Save (fd, false);
4932c1f46dcSZachary Turner }
4942c1f46dcSZachary Turner 
4952c1f46dcSZachary Turner void
4962c1f46dcSZachary Turner ScriptInterpreterPython::RestoreTerminalState ()
4972c1f46dcSZachary Turner {
4982c1f46dcSZachary Turner     // Python mucks with the terminal state of STDIN. If we can possibly avoid
4992c1f46dcSZachary Turner     // this by setting the file handles up correctly prior to entering the
5002c1f46dcSZachary Turner     // interpreter we should. For now we save and restore the terminal state
5012c1f46dcSZachary Turner     // on the input file handle.
5022c1f46dcSZachary Turner     m_terminal_state.Restore();
5032c1f46dcSZachary Turner }
5042c1f46dcSZachary Turner 
5052c1f46dcSZachary Turner void
5062c1f46dcSZachary Turner ScriptInterpreterPython::LeaveSession ()
5072c1f46dcSZachary Turner {
5082c1f46dcSZachary Turner     Log *log (lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_SCRIPT));
5092c1f46dcSZachary Turner     if (log)
5102c1f46dcSZachary Turner         log->PutCString("ScriptInterpreterPython::LeaveSession()");
5112c1f46dcSZachary Turner 
5122c1f46dcSZachary Turner     // checking that we have a valid thread state - since we use our own threading and locking
5132c1f46dcSZachary Turner     // in some (rare) cases during cleanup Python may end up believing we have no thread state
5142c1f46dcSZachary Turner     // and PyImport_AddModule will crash if that is the case - since that seems to only happen
5152c1f46dcSZachary Turner     // when destroying the SBDebugger, we can make do without clearing up stdout and stderr
5162c1f46dcSZachary Turner 
5172c1f46dcSZachary Turner     // rdar://problem/11292882
5182c1f46dcSZachary Turner     // When the current thread state is NULL, PyThreadState_Get() issues a fatal error.
5192c1f46dcSZachary Turner     if (PyThreadState_GetDict())
5202c1f46dcSZachary Turner     {
5212c1f46dcSZachary Turner         PythonDictionary &sys_module_dict = GetSysModuleDictionary ();
522f8b22f8fSZachary Turner         if (sys_module_dict.IsValid())
5232c1f46dcSZachary Turner         {
524f8b22f8fSZachary Turner             if (m_saved_stdin.IsValid())
5252c1f46dcSZachary Turner             {
526f8b22f8fSZachary Turner                 sys_module_dict.SetItemForKey(PythonString("stdin"), m_saved_stdin);
5272c1f46dcSZachary Turner                 m_saved_stdin.Reset ();
5282c1f46dcSZachary Turner             }
529f8b22f8fSZachary Turner             if (m_saved_stdout.IsValid())
5302c1f46dcSZachary Turner             {
531f8b22f8fSZachary Turner                 sys_module_dict.SetItemForKey(PythonString("stdout"), m_saved_stdout);
5322c1f46dcSZachary Turner                 m_saved_stdout.Reset ();
5332c1f46dcSZachary Turner             }
534f8b22f8fSZachary Turner             if (m_saved_stderr.IsValid())
5352c1f46dcSZachary Turner             {
536f8b22f8fSZachary Turner                 sys_module_dict.SetItemForKey(PythonString("stderr"), m_saved_stderr);
5372c1f46dcSZachary Turner                 m_saved_stderr.Reset ();
5382c1f46dcSZachary Turner             }
5392c1f46dcSZachary Turner         }
5402c1f46dcSZachary Turner     }
5412c1f46dcSZachary Turner 
5422c1f46dcSZachary Turner     m_session_is_active = false;
5432c1f46dcSZachary Turner }
5442c1f46dcSZachary Turner 
5452c1f46dcSZachary Turner bool
5462c1f46dcSZachary Turner ScriptInterpreterPython::EnterSession (uint16_t on_entry_flags,
5472c1f46dcSZachary Turner                                        FILE *in,
5482c1f46dcSZachary Turner                                        FILE *out,
5492c1f46dcSZachary Turner                                        FILE *err)
5502c1f46dcSZachary Turner {
5512c1f46dcSZachary Turner     // If we have already entered the session, without having officially 'left' it, then there is no need to
5522c1f46dcSZachary Turner     // 'enter' it again.
5532c1f46dcSZachary Turner     Log *log (lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_SCRIPT));
5542c1f46dcSZachary Turner     if (m_session_is_active)
5552c1f46dcSZachary Turner     {
5562c1f46dcSZachary Turner         if (log)
5572c1f46dcSZachary Turner             log->Printf("ScriptInterpreterPython::EnterSession(on_entry_flags=0x%" PRIx16 ") session is already active, returning without doing anything", on_entry_flags);
5582c1f46dcSZachary Turner         return false;
5592c1f46dcSZachary Turner     }
5602c1f46dcSZachary Turner 
5612c1f46dcSZachary Turner     if (log)
5622c1f46dcSZachary Turner         log->Printf("ScriptInterpreterPython::EnterSession(on_entry_flags=0x%" PRIx16 ")", on_entry_flags);
5632c1f46dcSZachary Turner 
5642c1f46dcSZachary Turner 
5652c1f46dcSZachary Turner     m_session_is_active = true;
5662c1f46dcSZachary Turner 
5672c1f46dcSZachary Turner     StreamString run_string;
5682c1f46dcSZachary Turner 
5692c1f46dcSZachary Turner     if (on_entry_flags & Locker::InitGlobals)
5702c1f46dcSZachary Turner     {
5712c1f46dcSZachary Turner         run_string.Printf (    "run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64, m_dictionary_name.c_str(), GetCommandInterpreter().GetDebugger().GetID());
5722c1f46dcSZachary Turner         run_string.Printf (    "; lldb.debugger = lldb.SBDebugger.FindDebuggerWithID (%" PRIu64 ")", GetCommandInterpreter().GetDebugger().GetID());
5732c1f46dcSZachary Turner         run_string.PutCString ("; lldb.target = lldb.debugger.GetSelectedTarget()");
5742c1f46dcSZachary Turner         run_string.PutCString ("; lldb.process = lldb.target.GetProcess()");
5752c1f46dcSZachary Turner         run_string.PutCString ("; lldb.thread = lldb.process.GetSelectedThread ()");
5762c1f46dcSZachary Turner         run_string.PutCString ("; lldb.frame = lldb.thread.GetSelectedFrame ()");
5772c1f46dcSZachary Turner         run_string.PutCString ("')");
5782c1f46dcSZachary Turner     }
5792c1f46dcSZachary Turner     else
5802c1f46dcSZachary Turner     {
5812c1f46dcSZachary Turner         // If we aren't initing the globals, we should still always set the debugger (since that is always unique.)
5822c1f46dcSZachary Turner         run_string.Printf (    "run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64, m_dictionary_name.c_str(), GetCommandInterpreter().GetDebugger().GetID());
5832c1f46dcSZachary Turner         run_string.Printf (    "; lldb.debugger = lldb.SBDebugger.FindDebuggerWithID (%" PRIu64 ")", GetCommandInterpreter().GetDebugger().GetID());
5842c1f46dcSZachary Turner         run_string.PutCString ("')");
5852c1f46dcSZachary Turner     }
5862c1f46dcSZachary Turner 
5872c1f46dcSZachary Turner     PyRun_SimpleString (run_string.GetData());
5882c1f46dcSZachary Turner     run_string.Clear();
5892c1f46dcSZachary Turner 
5902c1f46dcSZachary Turner     PythonDictionary &sys_module_dict = GetSysModuleDictionary ();
591f8b22f8fSZachary Turner     if (sys_module_dict.IsValid())
5922c1f46dcSZachary Turner     {
5939c40264fSZachary Turner         File in_file(in, false);
5949c40264fSZachary Turner         File out_file(out, false);
5959c40264fSZachary Turner         File err_file(err, false);
5969c40264fSZachary Turner 
5972c1f46dcSZachary Turner         lldb::StreamFileSP in_sp;
5982c1f46dcSZachary Turner         lldb::StreamFileSP out_sp;
5992c1f46dcSZachary Turner         lldb::StreamFileSP err_sp;
6009c40264fSZachary Turner         if (!in_file.IsValid() || !out_file.IsValid() || !err_file.IsValid())
6012c1f46dcSZachary Turner             m_interpreter.GetDebugger().AdoptTopIOHandlerFilesIfInvalid (in_sp, out_sp, err_sp);
6022c1f46dcSZachary Turner 
6032c1f46dcSZachary Turner         m_saved_stdin.Reset();
6042c1f46dcSZachary Turner 
6052c1f46dcSZachary Turner         if ((on_entry_flags & Locker::NoSTDIN) == 0)
6062c1f46dcSZachary Turner         {
6072c1f46dcSZachary Turner             // STDIN is enabled
6089c40264fSZachary Turner             if (!in_file.IsValid() && in_sp)
6099c40264fSZachary Turner                 in_file = in_sp->GetFile();
6109c40264fSZachary Turner             if (in_file.IsValid())
6112c1f46dcSZachary Turner             {
6129c40264fSZachary Turner                 // Flush the file before giving it to python to avoid interleaved output.
6139c40264fSZachary Turner                 in_file.Flush();
6149c40264fSZachary Turner 
615*32064024SZachary Turner                 m_saved_stdin = sys_module_dict.GetItemForKey(PythonString("stdin")).AsType<PythonFile>();
6162c1f46dcSZachary Turner                 // This call can deadlock your process if the file is locked
6179c40264fSZachary Turner                 PythonFile new_file(in_file, "r");
618f8b22f8fSZachary Turner                 sys_module_dict.SetItemForKey (PythonString("stdin"), new_file);
6192c1f46dcSZachary Turner             }
6202c1f46dcSZachary Turner         }
6212c1f46dcSZachary Turner 
6229c40264fSZachary Turner         if (!out_file.IsValid() && out_sp)
6239c40264fSZachary Turner             out_file = out_sp->GetFile();
6249c40264fSZachary Turner         if (out_file.IsValid())
6252c1f46dcSZachary Turner         {
6269c40264fSZachary Turner             // Flush the file before giving it to python to avoid interleaved output.
6279c40264fSZachary Turner             out_file.Flush();
6289c40264fSZachary Turner 
629*32064024SZachary Turner             m_saved_stdout = sys_module_dict.GetItemForKey(PythonString("stdout")).AsType<PythonFile>();
6302c1f46dcSZachary Turner 
6319c40264fSZachary Turner             PythonFile new_file(out_file, "w");
632f8b22f8fSZachary Turner             sys_module_dict.SetItemForKey (PythonString("stdout"), new_file);
6332c1f46dcSZachary Turner         }
6342c1f46dcSZachary Turner         else
6352c1f46dcSZachary Turner             m_saved_stdout.Reset();
6362c1f46dcSZachary Turner 
6379c40264fSZachary Turner         if (!err_file.IsValid() && err_sp)
6389c40264fSZachary Turner             err_file = err_sp->GetFile();
6399c40264fSZachary Turner         if (err_file.IsValid())
6402c1f46dcSZachary Turner         {
6419c40264fSZachary Turner             // Flush the file before giving it to python to avoid interleaved output.
6429c40264fSZachary Turner             err_file.Flush();
6439c40264fSZachary Turner 
644*32064024SZachary Turner             m_saved_stderr = sys_module_dict.GetItemForKey(PythonString("stderr")).AsType<PythonFile>();
6452c1f46dcSZachary Turner 
6469c40264fSZachary Turner             PythonFile new_file(err_file, "w");
647f8b22f8fSZachary Turner             sys_module_dict.SetItemForKey (PythonString("stderr"), new_file);
6482c1f46dcSZachary Turner         }
6492c1f46dcSZachary Turner         else
6502c1f46dcSZachary Turner             m_saved_stderr.Reset();
6512c1f46dcSZachary Turner     }
6522c1f46dcSZachary Turner 
6532c1f46dcSZachary Turner     if (PyErr_Occurred())
6542c1f46dcSZachary Turner         PyErr_Clear ();
6552c1f46dcSZachary Turner 
6562c1f46dcSZachary Turner     return true;
6572c1f46dcSZachary Turner }
6582c1f46dcSZachary Turner 
6592c1f46dcSZachary Turner PythonObject &
6602c1f46dcSZachary Turner ScriptInterpreterPython::GetMainModule()
6612c1f46dcSZachary Turner {
662f8b22f8fSZachary Turner     if (!m_main_module.IsValid())
663f8b22f8fSZachary Turner         m_main_module.Reset(PyRefType::Borrowed, PyImport_AddModule("__main__"));
6642c1f46dcSZachary Turner     return m_main_module;
6652c1f46dcSZachary Turner }
6662c1f46dcSZachary Turner 
6672c1f46dcSZachary Turner PythonDictionary &
6682c1f46dcSZachary Turner ScriptInterpreterPython::GetSessionDictionary ()
6692c1f46dcSZachary Turner {
670f8b22f8fSZachary Turner     if (m_session_dict.IsValid())
671f8b22f8fSZachary Turner         return m_session_dict;
672f8b22f8fSZachary Turner 
6732c1f46dcSZachary Turner     PythonObject &main_module = GetMainModule();
674f8b22f8fSZachary Turner     if (!main_module.IsValid())
675f8b22f8fSZachary Turner         return m_session_dict;
676f8b22f8fSZachary Turner 
677f8b22f8fSZachary Turner     PythonDictionary main_dict(PyRefType::Borrowed, PyModule_GetDict(main_module.get()));
678f8b22f8fSZachary Turner     if (!main_dict.IsValid())
679f8b22f8fSZachary Turner         return m_session_dict;
680f8b22f8fSZachary Turner 
681f8b22f8fSZachary Turner     PythonObject item = main_dict.GetItemForKey(PythonString(m_dictionary_name));
682f8b22f8fSZachary Turner     m_session_dict.Reset(PyRefType::Borrowed, item.get());
6832c1f46dcSZachary Turner     return m_session_dict;
6842c1f46dcSZachary Turner }
6852c1f46dcSZachary Turner 
6862c1f46dcSZachary Turner PythonDictionary &
6872c1f46dcSZachary Turner ScriptInterpreterPython::GetSysModuleDictionary ()
6882c1f46dcSZachary Turner {
689f8b22f8fSZachary Turner     if (m_sys_module_dict.IsValid())
690f8b22f8fSZachary Turner         return m_sys_module_dict;
691f8b22f8fSZachary Turner 
692f8b22f8fSZachary Turner     PythonObject sys_module(PyRefType::Borrowed, PyImport_AddModule("sys"));
693f8b22f8fSZachary Turner     if (sys_module.IsValid())
694f8b22f8fSZachary Turner         m_sys_module_dict.Reset(PyRefType::Borrowed, PyModule_GetDict(sys_module.get()));
6952c1f46dcSZachary Turner     return m_sys_module_dict;
6962c1f46dcSZachary Turner }
6972c1f46dcSZachary Turner 
6982c1f46dcSZachary Turner static std::string
6992c1f46dcSZachary Turner GenerateUniqueName (const char* base_name_wanted,
7002c1f46dcSZachary Turner                     uint32_t& functions_counter,
7012c1f46dcSZachary Turner                     const void* name_token = nullptr)
7022c1f46dcSZachary Turner {
7032c1f46dcSZachary Turner     StreamString sstr;
7042c1f46dcSZachary Turner 
7052c1f46dcSZachary Turner     if (!base_name_wanted)
7062c1f46dcSZachary Turner         return std::string();
7072c1f46dcSZachary Turner 
7082c1f46dcSZachary Turner     if (!name_token)
7092c1f46dcSZachary Turner         sstr.Printf ("%s_%d", base_name_wanted, functions_counter++);
7102c1f46dcSZachary Turner     else
7112c1f46dcSZachary Turner         sstr.Printf ("%s_%p", base_name_wanted, name_token);
7122c1f46dcSZachary Turner 
7132c1f46dcSZachary Turner     return sstr.GetString();
7142c1f46dcSZachary Turner }
7152c1f46dcSZachary Turner 
7162c1f46dcSZachary Turner bool
7172c1f46dcSZachary Turner ScriptInterpreterPython::GetEmbeddedInterpreterModuleObjects ()
7182c1f46dcSZachary Turner {
719f8b22f8fSZachary Turner     if (m_run_one_line_function.IsValid())
720f8b22f8fSZachary Turner         return true;
721f8b22f8fSZachary Turner 
722f8b22f8fSZachary Turner     PythonObject module(PyRefType::Borrowed, PyImport_AddModule ("lldb.embedded_interpreter"));
723f8b22f8fSZachary Turner     if (!module.IsValid())
724f8b22f8fSZachary Turner         return false;
725f8b22f8fSZachary Turner 
726f8b22f8fSZachary Turner     PythonDictionary module_dict(PyRefType::Borrowed, PyModule_GetDict(module.get()));
727f8b22f8fSZachary Turner     if (!module_dict.IsValid())
728f8b22f8fSZachary Turner         return false;
729f8b22f8fSZachary Turner 
730f8b22f8fSZachary Turner     m_run_one_line_function = module_dict.GetItemForKey(PythonString("run_one_line"));
731f8b22f8fSZachary Turner     m_run_one_line_str_global = module_dict.GetItemForKey(PythonString("g_run_one_line_str"));
732f8b22f8fSZachary Turner     return m_run_one_line_function.IsValid();
7332c1f46dcSZachary Turner }
7342c1f46dcSZachary Turner 
7352c1f46dcSZachary Turner static void
7362c1f46dcSZachary Turner ReadThreadBytesReceived(void *baton, const void *src, size_t src_len)
7372c1f46dcSZachary Turner {
7382c1f46dcSZachary Turner     if (src && src_len)
7392c1f46dcSZachary Turner     {
7402c1f46dcSZachary Turner         Stream *strm = (Stream *)baton;
7412c1f46dcSZachary Turner         strm->Write(src, src_len);
7422c1f46dcSZachary Turner         strm->Flush();
7432c1f46dcSZachary Turner     }
7442c1f46dcSZachary Turner }
7452c1f46dcSZachary Turner 
7462c1f46dcSZachary Turner bool
7472c1f46dcSZachary Turner ScriptInterpreterPython::ExecuteOneLine (const char *command, CommandReturnObject *result, const ExecuteScriptOptions &options)
7482c1f46dcSZachary Turner {
7492c1f46dcSZachary Turner     if (!m_valid_session)
7502c1f46dcSZachary Turner         return false;
7512c1f46dcSZachary Turner 
7522c1f46dcSZachary Turner     if (command && command[0])
7532c1f46dcSZachary Turner     {
7542c1f46dcSZachary Turner         // We want to call run_one_line, passing in the dictionary and the command string.  We cannot do this through
7552c1f46dcSZachary Turner         // PyRun_SimpleString here because the command string may contain escaped characters, and putting it inside
7562c1f46dcSZachary Turner         // another string to pass to PyRun_SimpleString messes up the escaping.  So we use the following more complicated
7572c1f46dcSZachary Turner         // method to pass the command string directly down to Python.
7582c1f46dcSZachary Turner         Debugger &debugger = m_interpreter.GetDebugger();
7592c1f46dcSZachary Turner 
7602c1f46dcSZachary Turner         StreamFileSP input_file_sp;
7612c1f46dcSZachary Turner         StreamFileSP output_file_sp;
7622c1f46dcSZachary Turner         StreamFileSP error_file_sp;
7632c1f46dcSZachary Turner         Communication output_comm ("lldb.ScriptInterpreterPython.ExecuteOneLine.comm");
7642c1f46dcSZachary Turner         bool join_read_thread = false;
7652c1f46dcSZachary Turner         if (options.GetEnableIO())
7662c1f46dcSZachary Turner         {
7672c1f46dcSZachary Turner             if (result)
7682c1f46dcSZachary Turner             {
7692c1f46dcSZachary Turner                 input_file_sp = debugger.GetInputFile();
7702c1f46dcSZachary Turner                 // Set output to a temporary file so we can forward the results on to the result object
7712c1f46dcSZachary Turner 
7722c1f46dcSZachary Turner                 Pipe pipe;
7732c1f46dcSZachary Turner                 Error pipe_result = pipe.CreateNew(false);
7742c1f46dcSZachary Turner                 if (pipe_result.Success())
7752c1f46dcSZachary Turner                 {
7762c1f46dcSZachary Turner #if defined(_WIN32)
7772c1f46dcSZachary Turner                     lldb::file_t read_file = pipe.GetReadNativeHandle();
7782c1f46dcSZachary Turner                     pipe.ReleaseReadFileDescriptor();
7792c1f46dcSZachary Turner                     std::unique_ptr<ConnectionGenericFile> conn_ap(new ConnectionGenericFile(read_file, true));
7802c1f46dcSZachary Turner #else
7812c1f46dcSZachary Turner                     std::unique_ptr<ConnectionFileDescriptor> conn_ap(new ConnectionFileDescriptor(pipe.ReleaseReadFileDescriptor(), true));
7822c1f46dcSZachary Turner #endif
7832c1f46dcSZachary Turner                     if (conn_ap->IsConnected())
7842c1f46dcSZachary Turner                     {
7852c1f46dcSZachary Turner                         output_comm.SetConnection(conn_ap.release());
7862c1f46dcSZachary Turner                         output_comm.SetReadThreadBytesReceivedCallback(ReadThreadBytesReceived, &result->GetOutputStream());
7872c1f46dcSZachary Turner                         output_comm.StartReadThread();
7882c1f46dcSZachary Turner                         join_read_thread = true;
7892c1f46dcSZachary Turner                         FILE *outfile_handle = fdopen (pipe.ReleaseWriteFileDescriptor(), "w");
7902c1f46dcSZachary Turner                         output_file_sp.reset(new StreamFile(outfile_handle, true));
7912c1f46dcSZachary Turner                         error_file_sp = output_file_sp;
7922c1f46dcSZachary Turner                         if (outfile_handle)
7932c1f46dcSZachary Turner                             ::setbuf (outfile_handle, nullptr);
7942c1f46dcSZachary Turner 
7952c1f46dcSZachary Turner                         result->SetImmediateOutputFile(debugger.GetOutputFile()->GetFile().GetStream());
7962c1f46dcSZachary Turner                         result->SetImmediateErrorFile(debugger.GetErrorFile()->GetFile().GetStream());
7972c1f46dcSZachary Turner                     }
7982c1f46dcSZachary Turner                 }
7992c1f46dcSZachary Turner             }
8002c1f46dcSZachary Turner             if (!input_file_sp || !output_file_sp || !error_file_sp)
8012c1f46dcSZachary Turner                 debugger.AdoptTopIOHandlerFilesIfInvalid(input_file_sp, output_file_sp, error_file_sp);
8022c1f46dcSZachary Turner         }
8032c1f46dcSZachary Turner         else
8042c1f46dcSZachary Turner         {
8052c1f46dcSZachary Turner             input_file_sp.reset (new StreamFile ());
8064eff2d31SZachary Turner             input_file_sp->GetFile().Open(FileSystem::DEV_NULL, File::eOpenOptionRead);
8072c1f46dcSZachary Turner             output_file_sp.reset (new StreamFile ());
8084eff2d31SZachary Turner             output_file_sp->GetFile().Open(FileSystem::DEV_NULL, File::eOpenOptionWrite);
8092c1f46dcSZachary Turner             error_file_sp = output_file_sp;
8102c1f46dcSZachary Turner         }
8112c1f46dcSZachary Turner 
8122c1f46dcSZachary Turner         FILE *in_file = input_file_sp->GetFile().GetStream();
8132c1f46dcSZachary Turner         FILE *out_file = output_file_sp->GetFile().GetStream();
8142c1f46dcSZachary Turner         FILE *err_file = error_file_sp->GetFile().GetStream();
815*32064024SZachary Turner         bool success = false;
816*32064024SZachary Turner         {
817*32064024SZachary Turner             // WARNING!  It's imperative that this RAII scope be as tight as possible.  In particular, the
818*32064024SZachary Turner             // scope must end *before* we try to join the read thread.  The reason for this is that a
819*32064024SZachary Turner             // pre-requisite for joining the read thread is that we close the write handle (to break the
820*32064024SZachary Turner             // pipe and cause it to wake up and exit).  But acquiring the GIL as below will redirect Python's
821*32064024SZachary Turner             // stdio to use this same handle.  If we close the handle while Python is still using it, bad
822*32064024SZachary Turner             // things will happen.
8232c1f46dcSZachary Turner             Locker locker(this,
8242c1f46dcSZachary Turner                           ScriptInterpreterPython::Locker::AcquireLock |
8252c1f46dcSZachary Turner                           ScriptInterpreterPython::Locker::InitSession |
8262c1f46dcSZachary Turner                           (options.GetSetLLDBGlobals() ? ScriptInterpreterPython::Locker::InitGlobals : 0) |
8272c1f46dcSZachary Turner                           ((result && result->GetInteractive()) ? 0: Locker::NoSTDIN),
8282c1f46dcSZachary Turner                           ScriptInterpreterPython::Locker::FreeAcquiredLock |
8292c1f46dcSZachary Turner                           ScriptInterpreterPython::Locker::TearDownSession,
8302c1f46dcSZachary Turner                           in_file,
8312c1f46dcSZachary Turner                           out_file,
8322c1f46dcSZachary Turner                           err_file);
8332c1f46dcSZachary Turner 
8342c1f46dcSZachary Turner             // Find the correct script interpreter dictionary in the main module.
8352c1f46dcSZachary Turner             PythonDictionary &session_dict = GetSessionDictionary ();
836f8b22f8fSZachary Turner             if (session_dict.IsValid())
8372c1f46dcSZachary Turner             {
8382c1f46dcSZachary Turner                 if (GetEmbeddedInterpreterModuleObjects ())
8392c1f46dcSZachary Turner                 {
840f8b22f8fSZachary Turner                     if (PyCallable_Check(m_run_one_line_function.get()))
8412c1f46dcSZachary Turner                     {
842f8b22f8fSZachary Turner                         PythonObject pargs(PyRefType::Owned, Py_BuildValue("(Os)", session_dict.get(), command));
843f8b22f8fSZachary Turner                         if (pargs.IsValid())
8442c1f46dcSZachary Turner                         {
845f8b22f8fSZachary Turner                             PythonObject return_value(PyRefType::Owned,
846f8b22f8fSZachary Turner                                 PyObject_CallObject(m_run_one_line_function.get(), pargs.get()));
847f8b22f8fSZachary Turner                             if (return_value.IsValid())
8482c1f46dcSZachary Turner                                 success = true;
8492c1f46dcSZachary Turner                             else if (options.GetMaskoutErrors() && PyErr_Occurred ())
8502c1f46dcSZachary Turner                             {
8512c1f46dcSZachary Turner                                 PyErr_Print();
8522c1f46dcSZachary Turner                                 PyErr_Clear();
8532c1f46dcSZachary Turner                             }
8542c1f46dcSZachary Turner                         }
8552c1f46dcSZachary Turner                     }
8562c1f46dcSZachary Turner                 }
8572c1f46dcSZachary Turner             }
8582c1f46dcSZachary Turner 
8592c1f46dcSZachary Turner             // Flush our output and error file handles
8602c1f46dcSZachary Turner             ::fflush (out_file);
8612c1f46dcSZachary Turner             if (out_file != err_file)
8622c1f46dcSZachary Turner                 ::fflush (err_file);
863*32064024SZachary Turner         }
8642c1f46dcSZachary Turner 
8652c1f46dcSZachary Turner         if (join_read_thread)
8662c1f46dcSZachary Turner         {
8672c1f46dcSZachary Turner             // Close the write end of the pipe since we are done with our
8682c1f46dcSZachary Turner             // one line script. This should cause the read thread that
8692c1f46dcSZachary Turner             // output_comm is using to exit
8702c1f46dcSZachary Turner             output_file_sp->GetFile().Close();
8712c1f46dcSZachary Turner             // The close above should cause this thread to exit when it gets
8722c1f46dcSZachary Turner             // to the end of file, so let it get all its data
8732c1f46dcSZachary Turner             output_comm.JoinReadThread();
8742c1f46dcSZachary Turner             // Now we can close the read end of the pipe
8752c1f46dcSZachary Turner             output_comm.Disconnect();
8762c1f46dcSZachary Turner         }
8772c1f46dcSZachary Turner 
8782c1f46dcSZachary Turner 
8792c1f46dcSZachary Turner         if (success)
8802c1f46dcSZachary Turner             return true;
8812c1f46dcSZachary Turner 
8822c1f46dcSZachary Turner         // The one-liner failed.  Append the error message.
8832c1f46dcSZachary Turner         if (result)
8842c1f46dcSZachary Turner             result->AppendErrorWithFormat ("python failed attempting to evaluate '%s'\n", command);
8852c1f46dcSZachary Turner         return false;
8862c1f46dcSZachary Turner     }
8872c1f46dcSZachary Turner 
8882c1f46dcSZachary Turner     if (result)
8892c1f46dcSZachary Turner         result->AppendError ("empty command passed to python\n");
8902c1f46dcSZachary Turner     return false;
8912c1f46dcSZachary Turner }
8922c1f46dcSZachary Turner 
8932c1f46dcSZachary Turner 
8942c1f46dcSZachary Turner class IOHandlerPythonInterpreter :
8952c1f46dcSZachary Turner     public IOHandler
8962c1f46dcSZachary Turner {
8972c1f46dcSZachary Turner public:
8982c1f46dcSZachary Turner 
8992c1f46dcSZachary Turner     IOHandlerPythonInterpreter (Debugger &debugger,
9002c1f46dcSZachary Turner                                 ScriptInterpreterPython *python) :
9012c1f46dcSZachary Turner         IOHandler (debugger, IOHandler::Type::PythonInterpreter),
9022c1f46dcSZachary Turner         m_python(python)
9032c1f46dcSZachary Turner     {
9042c1f46dcSZachary Turner 
9052c1f46dcSZachary Turner     }
9062c1f46dcSZachary Turner 
9072c1f46dcSZachary Turner     ~IOHandlerPythonInterpreter() override
9082c1f46dcSZachary Turner     {
9092c1f46dcSZachary Turner 
9102c1f46dcSZachary Turner     }
9112c1f46dcSZachary Turner 
9122c1f46dcSZachary Turner     ConstString
9132c1f46dcSZachary Turner     GetControlSequence (char ch) override
9142c1f46dcSZachary Turner     {
9152c1f46dcSZachary Turner         if (ch == 'd')
9162c1f46dcSZachary Turner             return ConstString("quit()\n");
9172c1f46dcSZachary Turner         return ConstString();
9182c1f46dcSZachary Turner     }
9192c1f46dcSZachary Turner 
9202c1f46dcSZachary Turner     void
9212c1f46dcSZachary Turner     Run () override
9222c1f46dcSZachary Turner     {
9232c1f46dcSZachary Turner         if (m_python)
9242c1f46dcSZachary Turner         {
9252c1f46dcSZachary Turner             int stdin_fd = GetInputFD();
9262c1f46dcSZachary Turner             if (stdin_fd >= 0)
9272c1f46dcSZachary Turner             {
9282c1f46dcSZachary Turner                 Terminal terminal(stdin_fd);
9292c1f46dcSZachary Turner                 TerminalState terminal_state;
9302c1f46dcSZachary Turner                 const bool is_a_tty = terminal.IsATerminal();
9312c1f46dcSZachary Turner 
9322c1f46dcSZachary Turner                 if (is_a_tty)
9332c1f46dcSZachary Turner                 {
9342c1f46dcSZachary Turner                     terminal_state.Save (stdin_fd, false);
9352c1f46dcSZachary Turner                     terminal.SetCanonical(false);
9362c1f46dcSZachary Turner                     terminal.SetEcho(true);
9372c1f46dcSZachary Turner                 }
9382c1f46dcSZachary Turner 
9392c1f46dcSZachary Turner                 ScriptInterpreterPython::Locker locker (m_python,
9402c1f46dcSZachary Turner                                                         ScriptInterpreterPython::Locker::AcquireLock |
9412c1f46dcSZachary Turner                                                         ScriptInterpreterPython::Locker::InitSession |
9422c1f46dcSZachary Turner                                                         ScriptInterpreterPython::Locker::InitGlobals,
9432c1f46dcSZachary Turner                                                         ScriptInterpreterPython::Locker::FreeAcquiredLock |
9442c1f46dcSZachary Turner                                                         ScriptInterpreterPython::Locker::TearDownSession);
9452c1f46dcSZachary Turner 
9462c1f46dcSZachary Turner                 // The following call drops into the embedded interpreter loop and stays there until the
9472c1f46dcSZachary Turner                 // user chooses to exit from the Python interpreter.
9482c1f46dcSZachary Turner                 // This embedded interpreter will, as any Python code that performs I/O, unlock the GIL before
9492c1f46dcSZachary Turner                 // a system call that can hang, and lock it when the syscall has returned.
9502c1f46dcSZachary Turner 
9512c1f46dcSZachary Turner                 // We need to surround the call to the embedded interpreter with calls to PyGILState_Ensure and
9522c1f46dcSZachary Turner                 // PyGILState_Release (using the Locker above). This is because Python has a global lock which must be held whenever we want
9532c1f46dcSZachary Turner                 // to touch any Python objects. Otherwise, if the user calls Python code, the interpreter state will be off,
9542c1f46dcSZachary Turner                 // and things could hang (it's happened before).
9552c1f46dcSZachary Turner 
9562c1f46dcSZachary Turner                 StreamString run_string;
9572c1f46dcSZachary Turner                 run_string.Printf ("run_python_interpreter (%s)", m_python->GetDictionaryName ());
9582c1f46dcSZachary Turner                 PyRun_SimpleString (run_string.GetData());
9592c1f46dcSZachary Turner 
9602c1f46dcSZachary Turner                 if (is_a_tty)
9612c1f46dcSZachary Turner                     terminal_state.Restore();
9622c1f46dcSZachary Turner             }
9632c1f46dcSZachary Turner         }
9642c1f46dcSZachary Turner         SetIsDone(true);
9652c1f46dcSZachary Turner     }
9662c1f46dcSZachary Turner 
9672c1f46dcSZachary Turner     void
9682c1f46dcSZachary Turner     Cancel () override
9692c1f46dcSZachary Turner     {
9702c1f46dcSZachary Turner 
9712c1f46dcSZachary Turner     }
9722c1f46dcSZachary Turner 
9732c1f46dcSZachary Turner     bool
9742c1f46dcSZachary Turner     Interrupt () override
9752c1f46dcSZachary Turner     {
9762c1f46dcSZachary Turner         return m_python->Interrupt();
9772c1f46dcSZachary Turner     }
9782c1f46dcSZachary Turner 
9792c1f46dcSZachary Turner     void
9802c1f46dcSZachary Turner     GotEOF() override
9812c1f46dcSZachary Turner     {
9822c1f46dcSZachary Turner 
9832c1f46dcSZachary Turner     }
9842c1f46dcSZachary Turner protected:
9852c1f46dcSZachary Turner     ScriptInterpreterPython *m_python;
9862c1f46dcSZachary Turner };
9872c1f46dcSZachary Turner 
9882c1f46dcSZachary Turner 
9892c1f46dcSZachary Turner void
9902c1f46dcSZachary Turner ScriptInterpreterPython::ExecuteInterpreterLoop ()
9912c1f46dcSZachary Turner {
9922c1f46dcSZachary Turner     Timer scoped_timer (__PRETTY_FUNCTION__, __PRETTY_FUNCTION__);
9932c1f46dcSZachary Turner 
9942c1f46dcSZachary Turner     Debugger &debugger = GetCommandInterpreter().GetDebugger();
9952c1f46dcSZachary Turner 
9962c1f46dcSZachary Turner     // At the moment, the only time the debugger does not have an input file handle is when this is called
9972c1f46dcSZachary Turner     // directly from Python, in which case it is both dangerous and unnecessary (not to mention confusing) to
9982c1f46dcSZachary Turner     // try to embed a running interpreter loop inside the already running Python interpreter loop, so we won't
9992c1f46dcSZachary Turner     // do it.
10002c1f46dcSZachary Turner 
10012c1f46dcSZachary Turner     if (!debugger.GetInputFile()->GetFile().IsValid())
10022c1f46dcSZachary Turner         return;
10032c1f46dcSZachary Turner 
10042c1f46dcSZachary Turner     IOHandlerSP io_handler_sp (new IOHandlerPythonInterpreter (debugger, this));
10052c1f46dcSZachary Turner     if (io_handler_sp)
10062c1f46dcSZachary Turner     {
10072c1f46dcSZachary Turner         debugger.PushIOHandler(io_handler_sp);
10082c1f46dcSZachary Turner     }
10092c1f46dcSZachary Turner }
10102c1f46dcSZachary Turner 
10112c1f46dcSZachary Turner bool
10122c1f46dcSZachary Turner ScriptInterpreterPython::Interrupt()
10132c1f46dcSZachary Turner {
10142c1f46dcSZachary Turner     Log *log (lldb_private::GetLogIfAllCategoriesSet (LIBLLDB_LOG_SCRIPT));
10152c1f46dcSZachary Turner 
10162c1f46dcSZachary Turner     if (IsExecutingPython())
10172c1f46dcSZachary Turner     {
101822c8efcdSZachary Turner         PyThreadState *state = PyThreadState_Get();
10192c1f46dcSZachary Turner         if (!state)
10202c1f46dcSZachary Turner             state = GetThreadState();
10212c1f46dcSZachary Turner         if (state)
10222c1f46dcSZachary Turner         {
10232c1f46dcSZachary Turner             long tid = state->thread_id;
102422c8efcdSZachary Turner             PyThreadState_Swap(state);
10252c1f46dcSZachary Turner             int num_threads = PyThreadState_SetAsyncExc(tid, PyExc_KeyboardInterrupt);
10262c1f46dcSZachary Turner             if (log)
10272c1f46dcSZachary Turner                 log->Printf("ScriptInterpreterPython::Interrupt() sending PyExc_KeyboardInterrupt (tid = %li, num_threads = %i)...", tid, num_threads);
10282c1f46dcSZachary Turner             return true;
10292c1f46dcSZachary Turner         }
10302c1f46dcSZachary Turner     }
10312c1f46dcSZachary Turner     if (log)
10322c1f46dcSZachary Turner         log->Printf("ScriptInterpreterPython::Interrupt() python code not running, can't interrupt");
10332c1f46dcSZachary Turner     return false;
10342c1f46dcSZachary Turner 
10352c1f46dcSZachary Turner }
10362c1f46dcSZachary Turner bool
10372c1f46dcSZachary Turner ScriptInterpreterPython::ExecuteOneLineWithReturn (const char *in_string,
10382c1f46dcSZachary Turner                                                    ScriptInterpreter::ScriptReturnType return_type,
10392c1f46dcSZachary Turner                                                    void *ret_value,
10402c1f46dcSZachary Turner                                                    const ExecuteScriptOptions &options)
10412c1f46dcSZachary Turner {
10422c1f46dcSZachary Turner 
10432c1f46dcSZachary Turner     Locker locker(this,
10442c1f46dcSZachary Turner                   ScriptInterpreterPython::Locker::AcquireLock | ScriptInterpreterPython::Locker::InitSession | (options.GetSetLLDBGlobals() ? ScriptInterpreterPython::Locker::InitGlobals : 0) | Locker::NoSTDIN,
10452c1f46dcSZachary Turner                   ScriptInterpreterPython::Locker::FreeAcquiredLock | ScriptInterpreterPython::Locker::TearDownSession);
10462c1f46dcSZachary Turner 
1047f8b22f8fSZachary Turner     PythonObject py_return;
10482c1f46dcSZachary Turner     PythonObject &main_module = GetMainModule();
1049f8b22f8fSZachary Turner     PythonDictionary globals(PyRefType::Borrowed, PyModule_GetDict(main_module.get()));
1050f8b22f8fSZachary Turner     PythonObject py_error;
10512c1f46dcSZachary Turner     bool ret_success = false;
10522c1f46dcSZachary Turner     int success;
10532c1f46dcSZachary Turner 
10542c1f46dcSZachary Turner     PythonDictionary locals = GetSessionDictionary ();
10552c1f46dcSZachary Turner 
1056f8b22f8fSZachary Turner     if (!locals.IsValid())
10572c1f46dcSZachary Turner     {
1058f8b22f8fSZachary Turner         locals.Reset(PyRefType::Owned, PyObject_GetAttrString(globals.get(), m_dictionary_name.c_str()));
10592c1f46dcSZachary Turner     }
10602c1f46dcSZachary Turner 
1061f8b22f8fSZachary Turner     if (!locals.IsValid())
10622c1f46dcSZachary Turner         locals = globals;
10632c1f46dcSZachary Turner 
1064f8b22f8fSZachary Turner     py_error.Reset(PyRefType::Borrowed, PyErr_Occurred());
1065f8b22f8fSZachary Turner     if (py_error.IsValid())
10662c1f46dcSZachary Turner         PyErr_Clear();
10672c1f46dcSZachary Turner 
10682c1f46dcSZachary Turner     if (in_string != nullptr)
10692c1f46dcSZachary Turner     {
10702c1f46dcSZachary Turner         { // scope for PythonInputReaderManager
10712c1f46dcSZachary Turner             //PythonInputReaderManager py_input(options.GetEnableIO() ? this : NULL);
1072f8b22f8fSZachary Turner             py_return.Reset(PyRefType::Owned, PyRun_String(in_string, Py_eval_input, globals.get(), locals.get()));
1073f8b22f8fSZachary Turner             if (!py_return.IsValid())
10742c1f46dcSZachary Turner             {
1075f8b22f8fSZachary Turner                 py_error.Reset(PyRefType::Borrowed, PyErr_Occurred());
1076f8b22f8fSZachary Turner                 if (py_error.IsValid())
10772c1f46dcSZachary Turner                     PyErr_Clear();
10782c1f46dcSZachary Turner 
1079f8b22f8fSZachary Turner                 py_return.Reset(PyRefType::Owned, PyRun_String(in_string, Py_single_input, globals.get(), locals.get()));
10802c1f46dcSZachary Turner             }
10812c1f46dcSZachary Turner         }
10822c1f46dcSZachary Turner 
1083f8b22f8fSZachary Turner         if (py_return.IsValid())
10842c1f46dcSZachary Turner         {
10852c1f46dcSZachary Turner             switch (return_type)
10862c1f46dcSZachary Turner             {
10872c1f46dcSZachary Turner                 case eScriptReturnTypeCharPtr: // "char *"
10882c1f46dcSZachary Turner                 {
10892c1f46dcSZachary Turner                     const char format[3] = "s#";
1090f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (char **) ret_value);
10912c1f46dcSZachary Turner                     break;
10922c1f46dcSZachary Turner                 }
10932c1f46dcSZachary Turner                 case eScriptReturnTypeCharStrOrNone: // char* or NULL if py_return == Py_None
10942c1f46dcSZachary Turner                 {
10952c1f46dcSZachary Turner                     const char format[3] = "z";
1096f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (char **) ret_value);
10972c1f46dcSZachary Turner                     break;
10982c1f46dcSZachary Turner                 }
10992c1f46dcSZachary Turner                 case eScriptReturnTypeBool:
11002c1f46dcSZachary Turner                 {
11012c1f46dcSZachary Turner                     const char format[2] = "b";
1102f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (bool *) ret_value);
11032c1f46dcSZachary Turner                     break;
11042c1f46dcSZachary Turner                 }
11052c1f46dcSZachary Turner                 case eScriptReturnTypeShortInt:
11062c1f46dcSZachary Turner                 {
11072c1f46dcSZachary Turner                     const char format[2] = "h";
1108f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (short *) ret_value);
11092c1f46dcSZachary Turner                     break;
11102c1f46dcSZachary Turner                 }
11112c1f46dcSZachary Turner                 case eScriptReturnTypeShortIntUnsigned:
11122c1f46dcSZachary Turner                 {
11132c1f46dcSZachary Turner                     const char format[2] = "H";
1114f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (unsigned short *) ret_value);
11152c1f46dcSZachary Turner                     break;
11162c1f46dcSZachary Turner                 }
11172c1f46dcSZachary Turner                 case eScriptReturnTypeInt:
11182c1f46dcSZachary Turner                 {
11192c1f46dcSZachary Turner                     const char format[2] = "i";
1120f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (int *) ret_value);
11212c1f46dcSZachary Turner                     break;
11222c1f46dcSZachary Turner                 }
11232c1f46dcSZachary Turner                 case eScriptReturnTypeIntUnsigned:
11242c1f46dcSZachary Turner                 {
11252c1f46dcSZachary Turner                     const char format[2] = "I";
1126f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (unsigned int *) ret_value);
11272c1f46dcSZachary Turner                     break;
11282c1f46dcSZachary Turner                 }
11292c1f46dcSZachary Turner                 case eScriptReturnTypeLongInt:
11302c1f46dcSZachary Turner                 {
11312c1f46dcSZachary Turner                     const char format[2] = "l";
1132f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (long *) ret_value);
11332c1f46dcSZachary Turner                     break;
11342c1f46dcSZachary Turner                 }
11352c1f46dcSZachary Turner                 case eScriptReturnTypeLongIntUnsigned:
11362c1f46dcSZachary Turner                 {
11372c1f46dcSZachary Turner                     const char format[2] = "k";
1138f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (unsigned long *) ret_value);
11392c1f46dcSZachary Turner                     break;
11402c1f46dcSZachary Turner                 }
11412c1f46dcSZachary Turner                 case eScriptReturnTypeLongLong:
11422c1f46dcSZachary Turner                 {
11432c1f46dcSZachary Turner                     const char format[2] = "L";
1144f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (long long *) ret_value);
11452c1f46dcSZachary Turner                     break;
11462c1f46dcSZachary Turner                 }
11472c1f46dcSZachary Turner                 case eScriptReturnTypeLongLongUnsigned:
11482c1f46dcSZachary Turner                 {
11492c1f46dcSZachary Turner                     const char format[2] = "K";
1150f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (unsigned long long *) ret_value);
11512c1f46dcSZachary Turner                     break;
11522c1f46dcSZachary Turner                 }
11532c1f46dcSZachary Turner                 case eScriptReturnTypeFloat:
11542c1f46dcSZachary Turner                 {
11552c1f46dcSZachary Turner                     const char format[2] = "f";
1156f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (float *) ret_value);
11572c1f46dcSZachary Turner                     break;
11582c1f46dcSZachary Turner                 }
11592c1f46dcSZachary Turner                 case eScriptReturnTypeDouble:
11602c1f46dcSZachary Turner                 {
11612c1f46dcSZachary Turner                     const char format[2] = "d";
1162f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (double *) ret_value);
11632c1f46dcSZachary Turner                     break;
11642c1f46dcSZachary Turner                 }
11652c1f46dcSZachary Turner                 case eScriptReturnTypeChar:
11662c1f46dcSZachary Turner                 {
11672c1f46dcSZachary Turner                     const char format[2] = "c";
1168f8b22f8fSZachary Turner                     success = PyArg_Parse (py_return.get(), format, (char *) ret_value);
11692c1f46dcSZachary Turner                     break;
11702c1f46dcSZachary Turner                 }
11712c1f46dcSZachary Turner                 case eScriptReturnTypeOpaqueObject:
11722c1f46dcSZachary Turner                 {
11732c1f46dcSZachary Turner                     success = true;
1174f8b22f8fSZachary Turner                     PyObject *saved_value = py_return.get();
1175f8b22f8fSZachary Turner                     Py_XINCREF(saved_value);
1176f8b22f8fSZachary Turner                     *((PyObject **)ret_value) = saved_value;
11772c1f46dcSZachary Turner                     break;
11782c1f46dcSZachary Turner                 }
11792c1f46dcSZachary Turner             }
1180f8b22f8fSZachary Turner 
11812c1f46dcSZachary Turner             if (success)
11822c1f46dcSZachary Turner                 ret_success = true;
11832c1f46dcSZachary Turner             else
11842c1f46dcSZachary Turner                 ret_success = false;
11852c1f46dcSZachary Turner         }
11862c1f46dcSZachary Turner     }
11872c1f46dcSZachary Turner 
1188f8b22f8fSZachary Turner     py_error.Reset(PyRefType::Borrowed, PyErr_Occurred());
1189f8b22f8fSZachary Turner     if (py_error.IsValid())
11902c1f46dcSZachary Turner     {
11912c1f46dcSZachary Turner         ret_success = false;
11922c1f46dcSZachary Turner         if (options.GetMaskoutErrors())
11932c1f46dcSZachary Turner         {
1194f8b22f8fSZachary Turner             if (PyErr_GivenExceptionMatches (py_error.get(), PyExc_SyntaxError))
11952c1f46dcSZachary Turner                 PyErr_Print ();
11962c1f46dcSZachary Turner             PyErr_Clear();
11972c1f46dcSZachary Turner         }
11982c1f46dcSZachary Turner     }
11992c1f46dcSZachary Turner 
12002c1f46dcSZachary Turner     return ret_success;
12012c1f46dcSZachary Turner }
12022c1f46dcSZachary Turner 
12032c1f46dcSZachary Turner Error
12042c1f46dcSZachary Turner ScriptInterpreterPython::ExecuteMultipleLines (const char *in_string, const ExecuteScriptOptions &options)
12052c1f46dcSZachary Turner {
12062c1f46dcSZachary Turner     Error error;
12072c1f46dcSZachary Turner 
12082c1f46dcSZachary Turner     Locker locker(this,
12092c1f46dcSZachary Turner                   ScriptInterpreterPython::Locker::AcquireLock      | ScriptInterpreterPython::Locker::InitSession | (options.GetSetLLDBGlobals() ? ScriptInterpreterPython::Locker::InitGlobals : 0) | Locker::NoSTDIN,
12102c1f46dcSZachary Turner                   ScriptInterpreterPython::Locker::FreeAcquiredLock | ScriptInterpreterPython::Locker::TearDownSession);
12112c1f46dcSZachary Turner 
12122c1f46dcSZachary Turner     PythonObject return_value;
12132c1f46dcSZachary Turner     PythonObject &main_module = GetMainModule();
1214f8b22f8fSZachary Turner     PythonDictionary globals(PyRefType::Borrowed, PyModule_GetDict(main_module.get()));
1215f8b22f8fSZachary Turner     PythonObject py_error;
12162c1f46dcSZachary Turner 
12172c1f46dcSZachary Turner     PythonDictionary locals = GetSessionDictionary();
12182c1f46dcSZachary Turner 
1219f8b22f8fSZachary Turner     if (!locals.IsValid())
1220f8b22f8fSZachary Turner         locals.Reset(PyRefType::Owned, PyObject_GetAttrString(globals.get(), m_dictionary_name.c_str()));
12212c1f46dcSZachary Turner 
1222f8b22f8fSZachary Turner     if (!locals.IsValid())
12232c1f46dcSZachary Turner         locals = globals;
12242c1f46dcSZachary Turner 
1225f8b22f8fSZachary Turner     py_error.Reset(PyRefType::Borrowed, PyErr_Occurred());
1226f8b22f8fSZachary Turner     if (py_error.IsValid())
12272c1f46dcSZachary Turner         PyErr_Clear();
12282c1f46dcSZachary Turner 
12292c1f46dcSZachary Turner     if (in_string != nullptr)
12302c1f46dcSZachary Turner     {
1231f8b22f8fSZachary Turner         PythonObject code_object;
1232f8b22f8fSZachary Turner         code_object.Reset(PyRefType::Owned, Py_CompileString(in_string, "temp.py", Py_file_input));
1233f8b22f8fSZachary Turner 
1234f8b22f8fSZachary Turner         if (code_object.IsValid())
1235f8b22f8fSZachary Turner         {
1236f8b22f8fSZachary Turner             // In Python 2.x, PyEval_EvalCode takes a PyCodeObject, but in Python 3.x, it takes
1237f8b22f8fSZachary Turner             // a PyObject.  They are convertible (hence the function PyCode_Check(PyObject*), so
1238f8b22f8fSZachary Turner             // we have to do the cast for Python 2.x
123922c8efcdSZachary Turner #if PY_MAJOR_VERSION >= 3
1240f8b22f8fSZachary Turner             PyObject *py_code_obj = code_object.get();
124122c8efcdSZachary Turner #else
1242f8b22f8fSZachary Turner             PyCodeObject *py_code_obj = reinterpret_cast<PyCodeObject *>(code_object.get());
124322c8efcdSZachary Turner #endif
1244f8b22f8fSZachary Turner             return_value.Reset(PyRefType::Owned, PyEval_EvalCode(py_code_obj, globals.get(), locals.get()));
1245f8b22f8fSZachary Turner         }
12462c1f46dcSZachary Turner     }
12472c1f46dcSZachary Turner 
1248f8b22f8fSZachary Turner     py_error.Reset(PyRefType::Borrowed, PyErr_Occurred());
1249f8b22f8fSZachary Turner     if (py_error.IsValid())
12502c1f46dcSZachary Turner     {
12512c1f46dcSZachary Turner         // puts(in_string);
12522c1f46dcSZachary Turner         // _PyObject_Dump (py_error);
12532c1f46dcSZachary Turner         // PyErr_Print();
12542c1f46dcSZachary Turner         // success = false;
12552c1f46dcSZachary Turner 
1256f8b22f8fSZachary Turner         PyObject *py_type = nullptr;
1257f8b22f8fSZachary Turner         PyObject *py_value = nullptr;
1258f8b22f8fSZachary Turner         PyObject *py_traceback = nullptr;
1259f8b22f8fSZachary Turner         PyErr_Fetch(&py_type, &py_value, &py_traceback);
1260f8b22f8fSZachary Turner 
1261f8b22f8fSZachary Turner         PythonObject type(PyRefType::Owned, py_type);
1262f8b22f8fSZachary Turner         PythonObject value(PyRefType::Owned, py_value);
1263f8b22f8fSZachary Turner         PythonObject traceback(PyRefType::Owned, py_traceback);
12642c1f46dcSZachary Turner 
12652c1f46dcSZachary Turner         // get the backtrace
12662c1f46dcSZachary Turner         std::string bt = ReadPythonBacktrace(traceback);
12672c1f46dcSZachary Turner 
1268f8b22f8fSZachary Turner         if (value.IsAllocated())
126922c8efcdSZachary Turner         {
1270f8b22f8fSZachary Turner             PythonString str = value.Str();
127122c8efcdSZachary Turner             llvm::StringRef value_str(str.GetString());
127222c8efcdSZachary Turner             error.SetErrorStringWithFormat("%s\n%s", value_str.str().c_str(), bt.c_str());
127322c8efcdSZachary Turner         }
12742c1f46dcSZachary Turner         else
12752c1f46dcSZachary Turner             error.SetErrorStringWithFormat("%s",bt.c_str());
12762c1f46dcSZachary Turner         if (options.GetMaskoutErrors())
12772c1f46dcSZachary Turner             PyErr_Clear();
12782c1f46dcSZachary Turner     }
12792c1f46dcSZachary Turner 
12802c1f46dcSZachary Turner     return error;
12812c1f46dcSZachary Turner }
12822c1f46dcSZachary Turner 
12832c1f46dcSZachary Turner 
12842c1f46dcSZachary Turner void
12852c1f46dcSZachary Turner ScriptInterpreterPython::CollectDataForBreakpointCommandCallback (std::vector<BreakpointOptions *> &bp_options_vec,
12862c1f46dcSZachary Turner                                                                   CommandReturnObject &result)
12872c1f46dcSZachary Turner {
12882c1f46dcSZachary Turner     m_active_io_handler = eIOHandlerBreakpoint;
12892c1f46dcSZachary Turner     m_interpreter.GetPythonCommandsFromIOHandler ("    ", *this, true, &bp_options_vec);
12902c1f46dcSZachary Turner }
12912c1f46dcSZachary Turner 
12922c1f46dcSZachary Turner void
12932c1f46dcSZachary Turner ScriptInterpreterPython::CollectDataForWatchpointCommandCallback (WatchpointOptions *wp_options,
12942c1f46dcSZachary Turner                                                                   CommandReturnObject &result)
12952c1f46dcSZachary Turner {
12962c1f46dcSZachary Turner     m_active_io_handler = eIOHandlerWatchpoint;
12972c1f46dcSZachary Turner     m_interpreter.GetPythonCommandsFromIOHandler ("    ", *this, true, wp_options);
12982c1f46dcSZachary Turner }
12992c1f46dcSZachary Turner 
13002c1f46dcSZachary Turner void
13012c1f46dcSZachary Turner ScriptInterpreterPython::SetBreakpointCommandCallbackFunction (BreakpointOptions *bp_options,
13022c1f46dcSZachary Turner                                                                const char *function_name)
13032c1f46dcSZachary Turner {
13042c1f46dcSZachary Turner     // For now just cons up a oneliner that calls the provided function.
13052c1f46dcSZachary Turner     std::string oneliner("return ");
13062c1f46dcSZachary Turner     oneliner += function_name;
13072c1f46dcSZachary Turner     oneliner += "(frame, bp_loc, internal_dict)";
13082c1f46dcSZachary Turner     m_interpreter.GetScriptInterpreter()->SetBreakpointCommandCallback (bp_options,
13092c1f46dcSZachary Turner                                                                         oneliner.c_str());
13102c1f46dcSZachary Turner }
13112c1f46dcSZachary Turner 
13122c1f46dcSZachary Turner // Set a Python one-liner as the callback for the breakpoint.
13132c1f46dcSZachary Turner Error
13142c1f46dcSZachary Turner ScriptInterpreterPython::SetBreakpointCommandCallback (BreakpointOptions *bp_options,
13152c1f46dcSZachary Turner                                                        const char *command_body_text)
13162c1f46dcSZachary Turner {
13172c1f46dcSZachary Turner     std::unique_ptr<BreakpointOptions::CommandData> data_ap(new BreakpointOptions::CommandData());
13182c1f46dcSZachary Turner 
13192c1f46dcSZachary Turner     // Split the command_body_text into lines, and pass that to GenerateBreakpointCommandCallbackData.  That will
13202c1f46dcSZachary Turner     // wrap the body in an auto-generated function, and return the function name in script_source.  That is what
13212c1f46dcSZachary Turner     // the callback will actually invoke.
13222c1f46dcSZachary Turner 
13232c1f46dcSZachary Turner     data_ap->user_source.SplitIntoLines(command_body_text);
13242c1f46dcSZachary Turner     Error error = GenerateBreakpointCommandCallbackData (data_ap->user_source, data_ap->script_source);
13252c1f46dcSZachary Turner     if (error.Success())
13262c1f46dcSZachary Turner     {
13272c1f46dcSZachary Turner         BatonSP baton_sp (new BreakpointOptions::CommandBaton (data_ap.release()));
13282c1f46dcSZachary Turner         bp_options->SetCallback (ScriptInterpreterPython::BreakpointCallbackFunction, baton_sp);
13292c1f46dcSZachary Turner         return error;
13302c1f46dcSZachary Turner     }
13312c1f46dcSZachary Turner     else
13322c1f46dcSZachary Turner         return error;
13332c1f46dcSZachary Turner }
13342c1f46dcSZachary Turner 
13352c1f46dcSZachary Turner // Set a Python one-liner as the callback for the watchpoint.
13362c1f46dcSZachary Turner void
13372c1f46dcSZachary Turner ScriptInterpreterPython::SetWatchpointCommandCallback (WatchpointOptions *wp_options,
13382c1f46dcSZachary Turner                                                        const char *oneliner)
13392c1f46dcSZachary Turner {
13402c1f46dcSZachary Turner     std::unique_ptr<WatchpointOptions::CommandData> data_ap(new WatchpointOptions::CommandData());
13412c1f46dcSZachary Turner 
13422c1f46dcSZachary Turner     // It's necessary to set both user_source and script_source to the oneliner.
13432c1f46dcSZachary Turner     // The former is used to generate callback description (as in watchpoint command list)
13442c1f46dcSZachary Turner     // while the latter is used for Python to interpret during the actual callback.
13452c1f46dcSZachary Turner 
13462c1f46dcSZachary Turner     data_ap->user_source.AppendString (oneliner);
13472c1f46dcSZachary Turner     data_ap->script_source.assign (oneliner);
13482c1f46dcSZachary Turner 
13492c1f46dcSZachary Turner     if (GenerateWatchpointCommandCallbackData (data_ap->user_source, data_ap->script_source))
13502c1f46dcSZachary Turner     {
13512c1f46dcSZachary Turner         BatonSP baton_sp (new WatchpointOptions::CommandBaton (data_ap.release()));
13522c1f46dcSZachary Turner         wp_options->SetCallback (ScriptInterpreterPython::WatchpointCallbackFunction, baton_sp);
13532c1f46dcSZachary Turner     }
13542c1f46dcSZachary Turner 
13552c1f46dcSZachary Turner     return;
13562c1f46dcSZachary Turner }
13572c1f46dcSZachary Turner 
13582c1f46dcSZachary Turner Error
13592c1f46dcSZachary Turner ScriptInterpreterPython::ExportFunctionDefinitionToInterpreter (StringList &function_def)
13602c1f46dcSZachary Turner {
13612c1f46dcSZachary Turner     // Convert StringList to one long, newline delimited, const char *.
13622c1f46dcSZachary Turner     std::string function_def_string(function_def.CopyList());
13632c1f46dcSZachary Turner 
13642c1f46dcSZachary Turner     Error error = ExecuteMultipleLines (function_def_string.c_str(), ScriptInterpreter::ExecuteScriptOptions().SetEnableIO(false));
13652c1f46dcSZachary Turner     return error;
13662c1f46dcSZachary Turner }
13672c1f46dcSZachary Turner 
13682c1f46dcSZachary Turner Error
13692c1f46dcSZachary Turner ScriptInterpreterPython::GenerateFunction(const char *signature, const StringList &input)
13702c1f46dcSZachary Turner {
13712c1f46dcSZachary Turner     Error error;
13722c1f46dcSZachary Turner     int num_lines = input.GetSize ();
13732c1f46dcSZachary Turner     if (num_lines == 0)
13742c1f46dcSZachary Turner     {
13752c1f46dcSZachary Turner         error.SetErrorString ("No input data.");
13762c1f46dcSZachary Turner         return error;
13772c1f46dcSZachary Turner     }
13782c1f46dcSZachary Turner 
13792c1f46dcSZachary Turner     if (!signature || *signature == 0)
13802c1f46dcSZachary Turner     {
13812c1f46dcSZachary Turner         error.SetErrorString("No output function name.");
13822c1f46dcSZachary Turner         return error;
13832c1f46dcSZachary Turner     }
13842c1f46dcSZachary Turner 
13852c1f46dcSZachary Turner     StreamString sstr;
13862c1f46dcSZachary Turner     StringList auto_generated_function;
13872c1f46dcSZachary Turner     auto_generated_function.AppendString (signature);
13882c1f46dcSZachary Turner     auto_generated_function.AppendString ("     global_dict = globals()");   // Grab the global dictionary
13892c1f46dcSZachary Turner     auto_generated_function.AppendString ("     new_keys = internal_dict.keys()");    // Make a list of keys in the session dict
13902c1f46dcSZachary Turner     auto_generated_function.AppendString ("     old_keys = global_dict.keys()"); // Save list of keys in global dict
13912c1f46dcSZachary Turner     auto_generated_function.AppendString ("     global_dict.update (internal_dict)"); // Add the session dictionary to the
13922c1f46dcSZachary Turner     // global dictionary.
13932c1f46dcSZachary Turner 
13942c1f46dcSZachary Turner     // Wrap everything up inside the function, increasing the indentation.
13952c1f46dcSZachary Turner 
13962c1f46dcSZachary Turner     auto_generated_function.AppendString("     if True:");
13972c1f46dcSZachary Turner     for (int i = 0; i < num_lines; ++i)
13982c1f46dcSZachary Turner     {
13992c1f46dcSZachary Turner         sstr.Clear ();
14002c1f46dcSZachary Turner         sstr.Printf ("       %s", input.GetStringAtIndex (i));
14012c1f46dcSZachary Turner         auto_generated_function.AppendString (sstr.GetData());
14022c1f46dcSZachary Turner     }
14032c1f46dcSZachary Turner     auto_generated_function.AppendString ("     for key in new_keys:");  // Iterate over all the keys from session dict
14042c1f46dcSZachary Turner     auto_generated_function.AppendString ("         internal_dict[key] = global_dict[key]");  // Update session dict values
14052c1f46dcSZachary Turner     auto_generated_function.AppendString ("         if key not in old_keys:");       // If key was not originally in global dict
14062c1f46dcSZachary Turner     auto_generated_function.AppendString ("             del global_dict[key]");      //  ...then remove key/value from global dict
14072c1f46dcSZachary Turner 
14082c1f46dcSZachary Turner     // Verify that the results are valid Python.
14092c1f46dcSZachary Turner 
14102c1f46dcSZachary Turner     error = ExportFunctionDefinitionToInterpreter (auto_generated_function);
14112c1f46dcSZachary Turner 
14122c1f46dcSZachary Turner     return error;
14132c1f46dcSZachary Turner }
14142c1f46dcSZachary Turner 
14152c1f46dcSZachary Turner bool
14162c1f46dcSZachary Turner ScriptInterpreterPython::GenerateTypeScriptFunction (StringList &user_input, std::string& output, const void* name_token)
14172c1f46dcSZachary Turner {
14182c1f46dcSZachary Turner     static uint32_t num_created_functions = 0;
14192c1f46dcSZachary Turner     user_input.RemoveBlankLines ();
14202c1f46dcSZachary Turner     StreamString sstr;
14212c1f46dcSZachary Turner 
14222c1f46dcSZachary Turner     // Check to see if we have any data; if not, just return.
14232c1f46dcSZachary Turner     if (user_input.GetSize() == 0)
14242c1f46dcSZachary Turner         return false;
14252c1f46dcSZachary Turner 
14262c1f46dcSZachary Turner     // Take what the user wrote, wrap it all up inside one big auto-generated Python function, passing in the
14272c1f46dcSZachary Turner     // ValueObject as parameter to the function.
14282c1f46dcSZachary Turner 
14292c1f46dcSZachary Turner     std::string auto_generated_function_name(GenerateUniqueName("lldb_autogen_python_type_print_func", num_created_functions, name_token));
14302c1f46dcSZachary Turner     sstr.Printf ("def %s (valobj, internal_dict):", auto_generated_function_name.c_str());
14312c1f46dcSZachary Turner 
14322c1f46dcSZachary Turner     if (!GenerateFunction(sstr.GetData(), user_input).Success())
14332c1f46dcSZachary Turner         return false;
14342c1f46dcSZachary Turner 
14352c1f46dcSZachary Turner     // Store the name of the auto-generated function to be called.
14362c1f46dcSZachary Turner     output.assign(auto_generated_function_name);
14372c1f46dcSZachary Turner     return true;
14382c1f46dcSZachary Turner }
14392c1f46dcSZachary Turner 
14402c1f46dcSZachary Turner bool
14412c1f46dcSZachary Turner ScriptInterpreterPython::GenerateScriptAliasFunction (StringList &user_input, std::string &output)
14422c1f46dcSZachary Turner {
14432c1f46dcSZachary Turner     static uint32_t num_created_functions = 0;
14442c1f46dcSZachary Turner     user_input.RemoveBlankLines ();
14452c1f46dcSZachary Turner     StreamString sstr;
14462c1f46dcSZachary Turner 
14472c1f46dcSZachary Turner     // Check to see if we have any data; if not, just return.
14482c1f46dcSZachary Turner     if (user_input.GetSize() == 0)
14492c1f46dcSZachary Turner         return false;
14502c1f46dcSZachary Turner 
14512c1f46dcSZachary Turner     std::string auto_generated_function_name(GenerateUniqueName("lldb_autogen_python_cmd_alias_func", num_created_functions));
14522c1f46dcSZachary Turner 
14532c1f46dcSZachary Turner     sstr.Printf ("def %s (debugger, args, result, internal_dict):", auto_generated_function_name.c_str());
14542c1f46dcSZachary Turner 
14552c1f46dcSZachary Turner     if (!GenerateFunction(sstr.GetData(),user_input).Success())
14562c1f46dcSZachary Turner         return false;
14572c1f46dcSZachary Turner 
14582c1f46dcSZachary Turner     // Store the name of the auto-generated function to be called.
14592c1f46dcSZachary Turner     output.assign(auto_generated_function_name);
14602c1f46dcSZachary Turner     return true;
14612c1f46dcSZachary Turner }
14622c1f46dcSZachary Turner 
14632c1f46dcSZachary Turner 
14642c1f46dcSZachary Turner bool
14652c1f46dcSZachary Turner ScriptInterpreterPython::GenerateTypeSynthClass (StringList &user_input, std::string &output, const void* name_token)
14662c1f46dcSZachary Turner {
14672c1f46dcSZachary Turner     static uint32_t num_created_classes = 0;
14682c1f46dcSZachary Turner     user_input.RemoveBlankLines ();
14692c1f46dcSZachary Turner     int num_lines = user_input.GetSize ();
14702c1f46dcSZachary Turner     StreamString sstr;
14712c1f46dcSZachary Turner 
14722c1f46dcSZachary Turner     // Check to see if we have any data; if not, just return.
14732c1f46dcSZachary Turner     if (user_input.GetSize() == 0)
14742c1f46dcSZachary Turner         return false;
14752c1f46dcSZachary Turner 
14762c1f46dcSZachary Turner     // Wrap all user input into a Python class
14772c1f46dcSZachary Turner 
14782c1f46dcSZachary Turner     std::string auto_generated_class_name(GenerateUniqueName("lldb_autogen_python_type_synth_class",num_created_classes,name_token));
14792c1f46dcSZachary Turner 
14802c1f46dcSZachary Turner     StringList auto_generated_class;
14812c1f46dcSZachary Turner 
14822c1f46dcSZachary Turner     // Create the function name & definition string.
14832c1f46dcSZachary Turner 
14842c1f46dcSZachary Turner     sstr.Printf ("class %s:", auto_generated_class_name.c_str());
14852c1f46dcSZachary Turner     auto_generated_class.AppendString (sstr.GetData());
14862c1f46dcSZachary Turner 
14872c1f46dcSZachary Turner     // Wrap everything up inside the class, increasing the indentation.
14882c1f46dcSZachary Turner     // we don't need to play any fancy indentation tricks here because there is no
14892c1f46dcSZachary Turner     // surrounding code whose indentation we need to honor
14902c1f46dcSZachary Turner     for (int i = 0; i < num_lines; ++i)
14912c1f46dcSZachary Turner     {
14922c1f46dcSZachary Turner         sstr.Clear ();
14932c1f46dcSZachary Turner         sstr.Printf ("     %s", user_input.GetStringAtIndex (i));
14942c1f46dcSZachary Turner         auto_generated_class.AppendString (sstr.GetData());
14952c1f46dcSZachary Turner     }
14962c1f46dcSZachary Turner 
14972c1f46dcSZachary Turner 
14982c1f46dcSZachary Turner     // Verify that the results are valid Python.
14992c1f46dcSZachary Turner     // (even though the method is ExportFunctionDefinitionToInterpreter, a class will actually be exported)
15002c1f46dcSZachary Turner     // (TODO: rename that method to ExportDefinitionToInterpreter)
15012c1f46dcSZachary Turner     if (!ExportFunctionDefinitionToInterpreter (auto_generated_class).Success())
15022c1f46dcSZachary Turner         return false;
15032c1f46dcSZachary Turner 
15042c1f46dcSZachary Turner     // Store the name of the auto-generated class
15052c1f46dcSZachary Turner 
15062c1f46dcSZachary Turner     output.assign(auto_generated_class_name);
15072c1f46dcSZachary Turner     return true;
15082c1f46dcSZachary Turner }
15092c1f46dcSZachary Turner 
15102c1f46dcSZachary Turner StructuredData::GenericSP
15112c1f46dcSZachary Turner ScriptInterpreterPython::OSPlugin_CreatePluginObject(const char *class_name, lldb::ProcessSP process_sp)
15122c1f46dcSZachary Turner {
15132c1f46dcSZachary Turner     if (class_name == nullptr || class_name[0] == '\0')
15142c1f46dcSZachary Turner         return StructuredData::GenericSP();
15152c1f46dcSZachary Turner 
15162c1f46dcSZachary Turner     if (!process_sp)
15172c1f46dcSZachary Turner         return StructuredData::GenericSP();
15182c1f46dcSZachary Turner 
15192c1f46dcSZachary Turner     void* ret_val;
15202c1f46dcSZachary Turner 
15212c1f46dcSZachary Turner     {
15222c1f46dcSZachary Turner         Locker py_lock  (this,
15232c1f46dcSZachary Turner                          Locker::AcquireLock | Locker::NoSTDIN,
15242c1f46dcSZachary Turner                          Locker::FreeLock);
15252c1f46dcSZachary Turner         ret_val = g_swig_create_os_plugin    (class_name,
15262c1f46dcSZachary Turner                                               m_dictionary_name.c_str(),
15272c1f46dcSZachary Turner                                               process_sp);
15282c1f46dcSZachary Turner     }
15292c1f46dcSZachary Turner 
15302c1f46dcSZachary Turner     return StructuredData::GenericSP(new StructuredPythonObject(ret_val));
15312c1f46dcSZachary Turner }
15322c1f46dcSZachary Turner 
15332c1f46dcSZachary Turner StructuredData::DictionarySP
15342c1f46dcSZachary Turner ScriptInterpreterPython::OSPlugin_RegisterInfo(StructuredData::ObjectSP os_plugin_object_sp)
15352c1f46dcSZachary Turner {
15362c1f46dcSZachary Turner     Locker py_lock(this,
15372c1f46dcSZachary Turner                    Locker::AcquireLock | Locker::NoSTDIN,
15382c1f46dcSZachary Turner                    Locker::FreeLock);
15392c1f46dcSZachary Turner 
15402c1f46dcSZachary Turner     static char callee_name[] = "get_register_info";
15412c1f46dcSZachary Turner 
15422c1f46dcSZachary Turner     if (!os_plugin_object_sp)
15432c1f46dcSZachary Turner         return StructuredData::DictionarySP();
15442c1f46dcSZachary Turner 
15452c1f46dcSZachary Turner     StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric();
15462c1f46dcSZachary Turner     if (!generic)
15472c1f46dcSZachary Turner         return nullptr;
15482c1f46dcSZachary Turner 
1549f8b22f8fSZachary Turner     PythonObject implementor(PyRefType::Borrowed, (PyObject *)generic->GetValue());
15502c1f46dcSZachary Turner 
1551f8b22f8fSZachary Turner     if (!implementor.IsAllocated())
15522c1f46dcSZachary Turner         return StructuredData::DictionarySP();
15532c1f46dcSZachary Turner 
1554f8b22f8fSZachary Turner     PythonObject pmeth(PyRefType::Owned, PyObject_GetAttrString(implementor.get(), callee_name));
15552c1f46dcSZachary Turner 
15562c1f46dcSZachary Turner     if (PyErr_Occurred())
15572c1f46dcSZachary Turner         PyErr_Clear();
15582c1f46dcSZachary Turner 
1559f8b22f8fSZachary Turner     if (!pmeth.IsAllocated())
15602c1f46dcSZachary Turner         return StructuredData::DictionarySP();
15612c1f46dcSZachary Turner 
1562f8b22f8fSZachary Turner     if (PyCallable_Check(pmeth.get()) == 0)
15632c1f46dcSZachary Turner     {
15642c1f46dcSZachary Turner         if (PyErr_Occurred())
15652c1f46dcSZachary Turner             PyErr_Clear();
15662c1f46dcSZachary Turner 
15672c1f46dcSZachary Turner         return StructuredData::DictionarySP();
15682c1f46dcSZachary Turner     }
15692c1f46dcSZachary Turner 
15702c1f46dcSZachary Turner     if (PyErr_Occurred())
15712c1f46dcSZachary Turner         PyErr_Clear();
15722c1f46dcSZachary Turner 
15732c1f46dcSZachary Turner     // right now we know this function exists and is callable..
1574f8b22f8fSZachary Turner     PythonObject py_return(PyRefType::Owned, PyObject_CallMethod(implementor.get(), callee_name, nullptr));
15752c1f46dcSZachary Turner 
15762c1f46dcSZachary Turner     // if it fails, print the error but otherwise go on
15772c1f46dcSZachary Turner     if (PyErr_Occurred())
15782c1f46dcSZachary Turner     {
15792c1f46dcSZachary Turner         PyErr_Print();
15802c1f46dcSZachary Turner         PyErr_Clear();
15812c1f46dcSZachary Turner     }
1582f8b22f8fSZachary Turner     assert(PythonDictionary::Check(py_return.get()) && "get_register_info returned unknown object type!");
15832c1f46dcSZachary Turner 
1584f8b22f8fSZachary Turner     PythonDictionary result_dict(PyRefType::Borrowed, py_return.get());
15852c1f46dcSZachary Turner     return result_dict.CreateStructuredDictionary();
15862c1f46dcSZachary Turner }
15872c1f46dcSZachary Turner 
15882c1f46dcSZachary Turner StructuredData::ArraySP
15892c1f46dcSZachary Turner ScriptInterpreterPython::OSPlugin_ThreadsInfo(StructuredData::ObjectSP os_plugin_object_sp)
15902c1f46dcSZachary Turner {
15912c1f46dcSZachary Turner     Locker py_lock (this,
15922c1f46dcSZachary Turner                     Locker::AcquireLock | Locker::NoSTDIN,
15932c1f46dcSZachary Turner                     Locker::FreeLock);
15942c1f46dcSZachary Turner 
15952c1f46dcSZachary Turner     static char callee_name[] = "get_thread_info";
15962c1f46dcSZachary Turner 
15972c1f46dcSZachary Turner     if (!os_plugin_object_sp)
15982c1f46dcSZachary Turner         return StructuredData::ArraySP();
15992c1f46dcSZachary Turner 
16002c1f46dcSZachary Turner     StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric();
16012c1f46dcSZachary Turner     if (!generic)
16022c1f46dcSZachary Turner         return nullptr;
16032c1f46dcSZachary Turner 
1604f8b22f8fSZachary Turner     PythonObject implementor(PyRefType::Borrowed, (PyObject *)generic->GetValue());
1605f8b22f8fSZachary Turner 
1606f8b22f8fSZachary Turner     if (!implementor.IsAllocated())
16072c1f46dcSZachary Turner         return StructuredData::ArraySP();
16082c1f46dcSZachary Turner 
1609f8b22f8fSZachary Turner     PythonObject pmeth(PyRefType::Owned, PyObject_GetAttrString(implementor.get(), callee_name));
16102c1f46dcSZachary Turner 
16112c1f46dcSZachary Turner     if (PyErr_Occurred())
16122c1f46dcSZachary Turner         PyErr_Clear();
16132c1f46dcSZachary Turner 
1614f8b22f8fSZachary Turner     if (!pmeth.IsAllocated())
16152c1f46dcSZachary Turner         return StructuredData::ArraySP();
16162c1f46dcSZachary Turner 
1617f8b22f8fSZachary Turner     if (PyCallable_Check(pmeth.get()) == 0)
16182c1f46dcSZachary Turner     {
16192c1f46dcSZachary Turner         if (PyErr_Occurred())
16202c1f46dcSZachary Turner             PyErr_Clear();
16212c1f46dcSZachary Turner 
16222c1f46dcSZachary Turner         return StructuredData::ArraySP();
16232c1f46dcSZachary Turner     }
16242c1f46dcSZachary Turner 
16252c1f46dcSZachary Turner     if (PyErr_Occurred())
16262c1f46dcSZachary Turner         PyErr_Clear();
16272c1f46dcSZachary Turner 
16282c1f46dcSZachary Turner     // right now we know this function exists and is callable..
1629f8b22f8fSZachary Turner     PythonObject py_return(PyRefType::Owned, PyObject_CallMethod(implementor.get(), callee_name, nullptr));
16302c1f46dcSZachary Turner 
16312c1f46dcSZachary Turner     // if it fails, print the error but otherwise go on
16322c1f46dcSZachary Turner     if (PyErr_Occurred())
16332c1f46dcSZachary Turner     {
16342c1f46dcSZachary Turner         PyErr_Print();
16352c1f46dcSZachary Turner         PyErr_Clear();
16362c1f46dcSZachary Turner     }
16372c1f46dcSZachary Turner 
1638f8b22f8fSZachary Turner     assert(PythonList::Check(py_return.get()) && "get_thread_info returned unknown object type!");
1639f8b22f8fSZachary Turner 
1640f8b22f8fSZachary Turner     PythonList result_list(PyRefType::Borrowed, py_return.get());
1641f8b22f8fSZachary Turner     return result_list.CreateStructuredArray();
16422c1f46dcSZachary Turner }
16432c1f46dcSZachary Turner 
16442c1f46dcSZachary Turner // GetPythonValueFormatString provides a system independent type safe way to
16452c1f46dcSZachary Turner // convert a variable's type into a python value format. Python value formats
16462c1f46dcSZachary Turner // are defined in terms of builtin C types and could change from system to
16472c1f46dcSZachary Turner // as the underlying typedef for uint* types, size_t, off_t and other values
16482c1f46dcSZachary Turner // change.
16492c1f46dcSZachary Turner 
16502c1f46dcSZachary Turner template <typename T>
16512c1f46dcSZachary Turner const char *GetPythonValueFormatString(T t)
16522c1f46dcSZachary Turner {
16532c1f46dcSZachary Turner     assert(!"Unhandled type passed to GetPythonValueFormatString(T), make a specialization of GetPythonValueFormatString() to support this type.");
16542c1f46dcSZachary Turner     return nullptr;
16552c1f46dcSZachary Turner }
16562c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (char *)             { return "s"; }
16572c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (char)               { return "b"; }
16582c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (unsigned char)      { return "B"; }
16592c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (short)              { return "h"; }
16602c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (unsigned short)     { return "H"; }
16612c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (int)                { return "i"; }
16622c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (unsigned int)       { return "I"; }
16632c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (long)               { return "l"; }
16642c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (unsigned long)      { return "k"; }
16652c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (long long)          { return "L"; }
16662c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (unsigned long long) { return "K"; }
16672c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (float t)            { return "f"; }
16682c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString (double t)           { return "d"; }
16692c1f46dcSZachary Turner 
16702c1f46dcSZachary Turner StructuredData::StringSP
16712c1f46dcSZachary Turner ScriptInterpreterPython::OSPlugin_RegisterContextData(StructuredData::ObjectSP os_plugin_object_sp, lldb::tid_t tid)
16722c1f46dcSZachary Turner {
16732c1f46dcSZachary Turner     Locker py_lock (this,
16742c1f46dcSZachary Turner                     Locker::AcquireLock | Locker::NoSTDIN,
16752c1f46dcSZachary Turner                     Locker::FreeLock);
16762c1f46dcSZachary Turner 
16772c1f46dcSZachary Turner     static char callee_name[] = "get_register_data";
16782c1f46dcSZachary Turner     static char *param_format = const_cast<char *>(GetPythonValueFormatString(tid));
16792c1f46dcSZachary Turner 
16802c1f46dcSZachary Turner     if (!os_plugin_object_sp)
16812c1f46dcSZachary Turner         return StructuredData::StringSP();
16822c1f46dcSZachary Turner 
16832c1f46dcSZachary Turner     StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric();
16842c1f46dcSZachary Turner     if (!generic)
16852c1f46dcSZachary Turner         return nullptr;
1686f8b22f8fSZachary Turner     PythonObject implementor(PyRefType::Borrowed, (PyObject *)generic->GetValue());
16872c1f46dcSZachary Turner 
1688f8b22f8fSZachary Turner     if (!implementor.IsAllocated())
16892c1f46dcSZachary Turner         return StructuredData::StringSP();
16902c1f46dcSZachary Turner 
1691f8b22f8fSZachary Turner     PythonObject pmeth(PyRefType::Owned, PyObject_GetAttrString(implementor.get(), callee_name));
16922c1f46dcSZachary Turner 
16932c1f46dcSZachary Turner     if (PyErr_Occurred())
16942c1f46dcSZachary Turner         PyErr_Clear();
16952c1f46dcSZachary Turner 
1696f8b22f8fSZachary Turner     if (!pmeth.IsAllocated())
16972c1f46dcSZachary Turner         return StructuredData::StringSP();
16982c1f46dcSZachary Turner 
1699f8b22f8fSZachary Turner     if (PyCallable_Check(pmeth.get()) == 0)
17002c1f46dcSZachary Turner     {
17012c1f46dcSZachary Turner         if (PyErr_Occurred())
17022c1f46dcSZachary Turner             PyErr_Clear();
17032c1f46dcSZachary Turner         return StructuredData::StringSP();
17042c1f46dcSZachary Turner     }
17052c1f46dcSZachary Turner 
17062c1f46dcSZachary Turner     if (PyErr_Occurred())
17072c1f46dcSZachary Turner         PyErr_Clear();
17082c1f46dcSZachary Turner 
17092c1f46dcSZachary Turner     // right now we know this function exists and is callable..
1710f8b22f8fSZachary Turner     PythonObject py_return(PyRefType::Owned, PyObject_CallMethod(implementor.get(), callee_name, param_format, tid));
17112c1f46dcSZachary Turner 
17122c1f46dcSZachary Turner     // if it fails, print the error but otherwise go on
17132c1f46dcSZachary Turner     if (PyErr_Occurred())
17142c1f46dcSZachary Turner     {
17152c1f46dcSZachary Turner         PyErr_Print();
17162c1f46dcSZachary Turner         PyErr_Clear();
17172c1f46dcSZachary Turner     }
1718f8b22f8fSZachary Turner 
1719f8b22f8fSZachary Turner     assert(PythonString::Check(py_return.get()) && "get_register_data returned unknown object type!");
1720f8b22f8fSZachary Turner 
1721f8b22f8fSZachary Turner     PythonString result_string(PyRefType::Borrowed, py_return.get());
17222c1f46dcSZachary Turner     return result_string.CreateStructuredString();
17232c1f46dcSZachary Turner }
17242c1f46dcSZachary Turner 
17252c1f46dcSZachary Turner StructuredData::DictionarySP
17262c1f46dcSZachary Turner ScriptInterpreterPython::OSPlugin_CreateThread(StructuredData::ObjectSP os_plugin_object_sp, lldb::tid_t tid, lldb::addr_t context)
17272c1f46dcSZachary Turner {
17282c1f46dcSZachary Turner     Locker py_lock(this,
17292c1f46dcSZachary Turner                    Locker::AcquireLock | Locker::NoSTDIN,
17302c1f46dcSZachary Turner                    Locker::FreeLock);
17312c1f46dcSZachary Turner 
17322c1f46dcSZachary Turner     static char callee_name[] = "create_thread";
17332c1f46dcSZachary Turner     std::string param_format;
17342c1f46dcSZachary Turner     param_format += GetPythonValueFormatString(tid);
17352c1f46dcSZachary Turner     param_format += GetPythonValueFormatString(context);
17362c1f46dcSZachary Turner 
17372c1f46dcSZachary Turner     if (!os_plugin_object_sp)
17382c1f46dcSZachary Turner         return StructuredData::DictionarySP();
17392c1f46dcSZachary Turner 
17402c1f46dcSZachary Turner     StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric();
17412c1f46dcSZachary Turner     if (!generic)
17422c1f46dcSZachary Turner         return nullptr;
17432c1f46dcSZachary Turner 
1744f8b22f8fSZachary Turner     PythonObject implementor(PyRefType::Borrowed, (PyObject *)generic->GetValue());
1745f8b22f8fSZachary Turner 
1746f8b22f8fSZachary Turner     if (!implementor.IsAllocated())
17472c1f46dcSZachary Turner         return StructuredData::DictionarySP();
17482c1f46dcSZachary Turner 
1749f8b22f8fSZachary Turner     PythonObject pmeth(PyRefType::Owned, PyObject_GetAttrString(implementor.get(), callee_name));
17502c1f46dcSZachary Turner 
17512c1f46dcSZachary Turner     if (PyErr_Occurred())
17522c1f46dcSZachary Turner         PyErr_Clear();
17532c1f46dcSZachary Turner 
1754f8b22f8fSZachary Turner     if (!pmeth.IsAllocated())
17552c1f46dcSZachary Turner         return StructuredData::DictionarySP();
17562c1f46dcSZachary Turner 
1757f8b22f8fSZachary Turner     if (PyCallable_Check(pmeth.get()) == 0)
17582c1f46dcSZachary Turner     {
17592c1f46dcSZachary Turner         if (PyErr_Occurred())
17602c1f46dcSZachary Turner             PyErr_Clear();
17612c1f46dcSZachary Turner         return StructuredData::DictionarySP();
17622c1f46dcSZachary Turner     }
17632c1f46dcSZachary Turner 
17642c1f46dcSZachary Turner     if (PyErr_Occurred())
17652c1f46dcSZachary Turner         PyErr_Clear();
17662c1f46dcSZachary Turner 
17672c1f46dcSZachary Turner     // right now we know this function exists and is callable..
1768f8b22f8fSZachary Turner     PythonObject py_return(PyRefType::Owned, PyObject_CallMethod(implementor.get(), callee_name, &param_format[0], tid, context));
17692c1f46dcSZachary Turner 
17702c1f46dcSZachary Turner     // if it fails, print the error but otherwise go on
17712c1f46dcSZachary Turner     if (PyErr_Occurred())
17722c1f46dcSZachary Turner     {
17732c1f46dcSZachary Turner         PyErr_Print();
17742c1f46dcSZachary Turner         PyErr_Clear();
17752c1f46dcSZachary Turner     }
17762c1f46dcSZachary Turner 
1777f8b22f8fSZachary Turner     assert(PythonDictionary::Check(py_return.get()) && "create_thread returned unknown object type!");
1778f8b22f8fSZachary Turner 
1779f8b22f8fSZachary Turner     PythonDictionary result_dict(PyRefType::Borrowed, py_return.get());
17802c1f46dcSZachary Turner     return result_dict.CreateStructuredDictionary();
17812c1f46dcSZachary Turner }
17822c1f46dcSZachary Turner 
17832c1f46dcSZachary Turner StructuredData::ObjectSP
17842c1f46dcSZachary Turner ScriptInterpreterPython::CreateScriptedThreadPlan(const char *class_name, lldb::ThreadPlanSP thread_plan_sp)
17852c1f46dcSZachary Turner {
17862c1f46dcSZachary Turner     if (class_name == nullptr || class_name[0] == '\0')
17872c1f46dcSZachary Turner         return StructuredData::ObjectSP();
17882c1f46dcSZachary Turner 
17892c1f46dcSZachary Turner     if (!thread_plan_sp.get())
17902c1f46dcSZachary Turner         return StructuredData::ObjectSP();
17912c1f46dcSZachary Turner 
17922c1f46dcSZachary Turner     Debugger &debugger = thread_plan_sp->GetTarget().GetDebugger();
17932c1f46dcSZachary Turner     ScriptInterpreter *script_interpreter = debugger.GetCommandInterpreter().GetScriptInterpreter();
17942c1f46dcSZachary Turner     ScriptInterpreterPython *python_interpreter = static_cast<ScriptInterpreterPython *>(script_interpreter);
17952c1f46dcSZachary Turner 
17962c1f46dcSZachary Turner     if (!script_interpreter)
17972c1f46dcSZachary Turner         return StructuredData::ObjectSP();
17982c1f46dcSZachary Turner 
17992c1f46dcSZachary Turner     void* ret_val;
18002c1f46dcSZachary Turner 
18012c1f46dcSZachary Turner     {
18022c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
18032c1f46dcSZachary Turner 
18042c1f46dcSZachary Turner         ret_val = g_swig_thread_plan_script (class_name,
18052c1f46dcSZachary Turner                                              python_interpreter->m_dictionary_name.c_str(),
18062c1f46dcSZachary Turner                                              thread_plan_sp);
18072c1f46dcSZachary Turner     }
18082c1f46dcSZachary Turner 
18092c1f46dcSZachary Turner     return StructuredData::ObjectSP(new StructuredPythonObject(ret_val));
18102c1f46dcSZachary Turner }
18112c1f46dcSZachary Turner 
18122c1f46dcSZachary Turner bool
18132c1f46dcSZachary Turner ScriptInterpreterPython::ScriptedThreadPlanExplainsStop(StructuredData::ObjectSP implementor_sp, Event *event, bool &script_error)
18142c1f46dcSZachary Turner {
18152c1f46dcSZachary Turner     bool explains_stop = true;
18162c1f46dcSZachary Turner     StructuredData::Generic *generic = nullptr;
18172c1f46dcSZachary Turner     if (implementor_sp)
18182c1f46dcSZachary Turner         generic = implementor_sp->GetAsGeneric();
18192c1f46dcSZachary Turner     if (generic)
18202c1f46dcSZachary Turner     {
18212c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
18222c1f46dcSZachary Turner         explains_stop = g_swig_call_thread_plan(generic->GetValue(), "explains_stop", event, script_error);
18232c1f46dcSZachary Turner         if (script_error)
18242c1f46dcSZachary Turner             return true;
18252c1f46dcSZachary Turner     }
18262c1f46dcSZachary Turner     return explains_stop;
18272c1f46dcSZachary Turner }
18282c1f46dcSZachary Turner 
18292c1f46dcSZachary Turner bool
18302c1f46dcSZachary Turner ScriptInterpreterPython::ScriptedThreadPlanShouldStop(StructuredData::ObjectSP implementor_sp, Event *event, bool &script_error)
18312c1f46dcSZachary Turner {
18322c1f46dcSZachary Turner     bool should_stop = true;
18332c1f46dcSZachary Turner     StructuredData::Generic *generic = nullptr;
18342c1f46dcSZachary Turner     if (implementor_sp)
18352c1f46dcSZachary Turner         generic = implementor_sp->GetAsGeneric();
18362c1f46dcSZachary Turner     if (generic)
18372c1f46dcSZachary Turner     {
18382c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
18392c1f46dcSZachary Turner         should_stop = g_swig_call_thread_plan(generic->GetValue(), "should_stop", event, script_error);
18402c1f46dcSZachary Turner         if (script_error)
18412c1f46dcSZachary Turner             return true;
18422c1f46dcSZachary Turner     }
18432c1f46dcSZachary Turner     return should_stop;
18442c1f46dcSZachary Turner }
18452c1f46dcSZachary Turner 
18462c1f46dcSZachary Turner lldb::StateType
18472c1f46dcSZachary Turner ScriptInterpreterPython::ScriptedThreadPlanGetRunState(StructuredData::ObjectSP implementor_sp, bool &script_error)
18482c1f46dcSZachary Turner {
18492c1f46dcSZachary Turner     bool should_step = false;
18502c1f46dcSZachary Turner     StructuredData::Generic *generic = nullptr;
18512c1f46dcSZachary Turner     if (implementor_sp)
18522c1f46dcSZachary Turner         generic = implementor_sp->GetAsGeneric();
18532c1f46dcSZachary Turner     if (generic)
18542c1f46dcSZachary Turner     {
18552c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
18562c1f46dcSZachary Turner         should_step = g_swig_call_thread_plan(generic->GetValue(), "should_step", NULL, script_error);
18572c1f46dcSZachary Turner         if (script_error)
18582c1f46dcSZachary Turner             should_step = true;
18592c1f46dcSZachary Turner     }
18602c1f46dcSZachary Turner     if (should_step)
18612c1f46dcSZachary Turner         return lldb::eStateStepping;
18622c1f46dcSZachary Turner     else
18632c1f46dcSZachary Turner         return lldb::eStateRunning;
18642c1f46dcSZachary Turner }
18652c1f46dcSZachary Turner 
18662c1f46dcSZachary Turner StructuredData::ObjectSP
18672c1f46dcSZachary Turner ScriptInterpreterPython::LoadPluginModule(const FileSpec &file_spec, lldb_private::Error &error)
18682c1f46dcSZachary Turner {
18692c1f46dcSZachary Turner     if (!file_spec.Exists())
18702c1f46dcSZachary Turner     {
18712c1f46dcSZachary Turner         error.SetErrorString("no such file");
18722c1f46dcSZachary Turner         return StructuredData::ObjectSP();
18732c1f46dcSZachary Turner     }
18742c1f46dcSZachary Turner 
18752c1f46dcSZachary Turner     StructuredData::ObjectSP module_sp;
18762c1f46dcSZachary Turner 
18772c1f46dcSZachary Turner     if (LoadScriptingModule(file_spec.GetPath().c_str(),true,true,error,&module_sp))
18782c1f46dcSZachary Turner         return module_sp;
18792c1f46dcSZachary Turner 
18802c1f46dcSZachary Turner     return StructuredData::ObjectSP();
18812c1f46dcSZachary Turner }
18822c1f46dcSZachary Turner 
18832c1f46dcSZachary Turner StructuredData::DictionarySP
18842c1f46dcSZachary Turner ScriptInterpreterPython::GetDynamicSettings(StructuredData::ObjectSP plugin_module_sp, Target *target, const char *setting_name,
18852c1f46dcSZachary Turner                                             lldb_private::Error &error)
18862c1f46dcSZachary Turner {
18872c1f46dcSZachary Turner     if (!plugin_module_sp || !target || !setting_name || !setting_name[0] || !g_swig_plugin_get)
18882c1f46dcSZachary Turner         return StructuredData::DictionarySP();
18892c1f46dcSZachary Turner     StructuredData::Generic *generic = plugin_module_sp->GetAsGeneric();
18902c1f46dcSZachary Turner     if (!generic)
18912c1f46dcSZachary Turner         return StructuredData::DictionarySP();
18922c1f46dcSZachary Turner 
1893f8b22f8fSZachary Turner     PythonObject reply_pyobj;
18942c1f46dcSZachary Turner     {
18952c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
18962c1f46dcSZachary Turner         TargetSP target_sp(target->shared_from_this());
1897f8b22f8fSZachary Turner         reply_pyobj.Reset(PyRefType::Owned,
1898f8b22f8fSZachary Turner                           (PyObject *)g_swig_plugin_get(generic->GetValue(), setting_name, target_sp));
18992c1f46dcSZachary Turner     }
19002c1f46dcSZachary Turner 
1901f8b22f8fSZachary Turner     PythonDictionary py_dict(PyRefType::Borrowed, reply_pyobj.get());
19022c1f46dcSZachary Turner     return py_dict.CreateStructuredDictionary();
19032c1f46dcSZachary Turner }
19042c1f46dcSZachary Turner 
19052c1f46dcSZachary Turner StructuredData::ObjectSP
19062c1f46dcSZachary Turner ScriptInterpreterPython::CreateSyntheticScriptedProvider(const char *class_name, lldb::ValueObjectSP valobj)
19072c1f46dcSZachary Turner {
19082c1f46dcSZachary Turner     if (class_name == nullptr || class_name[0] == '\0')
19092c1f46dcSZachary Turner         return StructuredData::ObjectSP();
19102c1f46dcSZachary Turner 
19112c1f46dcSZachary Turner     if (!valobj.get())
19122c1f46dcSZachary Turner         return StructuredData::ObjectSP();
19132c1f46dcSZachary Turner 
19142c1f46dcSZachary Turner     ExecutionContext exe_ctx (valobj->GetExecutionContextRef());
19152c1f46dcSZachary Turner     Target *target = exe_ctx.GetTargetPtr();
19162c1f46dcSZachary Turner 
19172c1f46dcSZachary Turner     if (!target)
19182c1f46dcSZachary Turner         return StructuredData::ObjectSP();
19192c1f46dcSZachary Turner 
19202c1f46dcSZachary Turner     Debugger &debugger = target->GetDebugger();
19212c1f46dcSZachary Turner     ScriptInterpreter *script_interpreter = debugger.GetCommandInterpreter().GetScriptInterpreter();
19222c1f46dcSZachary Turner     ScriptInterpreterPython *python_interpreter = (ScriptInterpreterPython *) script_interpreter;
19232c1f46dcSZachary Turner 
19242c1f46dcSZachary Turner     if (!script_interpreter)
19252c1f46dcSZachary Turner         return StructuredData::ObjectSP();
19262c1f46dcSZachary Turner 
19272c1f46dcSZachary Turner     void *ret_val = nullptr;
19282c1f46dcSZachary Turner 
19292c1f46dcSZachary Turner     {
19302c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
19312c1f46dcSZachary Turner         ret_val = g_swig_synthetic_script (class_name,
19322c1f46dcSZachary Turner                                            python_interpreter->m_dictionary_name.c_str(),
19332c1f46dcSZachary Turner                                            valobj);
19342c1f46dcSZachary Turner     }
19352c1f46dcSZachary Turner 
19362c1f46dcSZachary Turner     return StructuredData::ObjectSP(new StructuredPythonObject(ret_val));
19372c1f46dcSZachary Turner }
19382c1f46dcSZachary Turner 
19392c1f46dcSZachary Turner StructuredData::GenericSP
19402c1f46dcSZachary Turner ScriptInterpreterPython::CreateScriptCommandObject (const char *class_name)
19412c1f46dcSZachary Turner {
19422c1f46dcSZachary Turner     DebuggerSP debugger_sp(GetCommandInterpreter().GetDebugger().shared_from_this());
19432c1f46dcSZachary Turner 
19442c1f46dcSZachary Turner     if (class_name == nullptr || class_name[0] == '\0')
19452c1f46dcSZachary Turner         return StructuredData::GenericSP();
19462c1f46dcSZachary Turner 
19472c1f46dcSZachary Turner     if (!debugger_sp.get())
19482c1f46dcSZachary Turner         return StructuredData::GenericSP();
19492c1f46dcSZachary Turner 
19502c1f46dcSZachary Turner     void* ret_val;
19512c1f46dcSZachary Turner 
19522c1f46dcSZachary Turner     {
19532c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
19542c1f46dcSZachary Turner         ret_val = g_swig_create_cmd (class_name,
19552c1f46dcSZachary Turner                                      m_dictionary_name.c_str(),
19562c1f46dcSZachary Turner                                      debugger_sp);
19572c1f46dcSZachary Turner     }
19582c1f46dcSZachary Turner 
19592c1f46dcSZachary Turner     return StructuredData::GenericSP(new StructuredPythonObject(ret_val));
19602c1f46dcSZachary Turner }
19612c1f46dcSZachary Turner 
19622c1f46dcSZachary Turner bool
19632c1f46dcSZachary Turner ScriptInterpreterPython::GenerateTypeScriptFunction (const char* oneliner, std::string& output, const void* name_token)
19642c1f46dcSZachary Turner {
19652c1f46dcSZachary Turner     StringList input;
19662c1f46dcSZachary Turner     input.SplitIntoLines(oneliner, strlen(oneliner));
19672c1f46dcSZachary Turner     return GenerateTypeScriptFunction(input, output, name_token);
19682c1f46dcSZachary Turner }
19692c1f46dcSZachary Turner 
19702c1f46dcSZachary Turner bool
19712c1f46dcSZachary Turner ScriptInterpreterPython::GenerateTypeSynthClass (const char* oneliner, std::string& output, const void* name_token)
19722c1f46dcSZachary Turner {
19732c1f46dcSZachary Turner     StringList input;
19742c1f46dcSZachary Turner     input.SplitIntoLines(oneliner, strlen(oneliner));
19752c1f46dcSZachary Turner     return GenerateTypeSynthClass(input, output, name_token);
19762c1f46dcSZachary Turner }
19772c1f46dcSZachary Turner 
19782c1f46dcSZachary Turner 
19792c1f46dcSZachary Turner Error
19802c1f46dcSZachary Turner ScriptInterpreterPython::GenerateBreakpointCommandCallbackData (StringList &user_input, std::string& output)
19812c1f46dcSZachary Turner {
19822c1f46dcSZachary Turner     static uint32_t num_created_functions = 0;
19832c1f46dcSZachary Turner     user_input.RemoveBlankLines ();
19842c1f46dcSZachary Turner     StreamString sstr;
19852c1f46dcSZachary Turner     Error error;
19862c1f46dcSZachary Turner     if (user_input.GetSize() == 0)
19872c1f46dcSZachary Turner     {
19882c1f46dcSZachary Turner         error.SetErrorString("No input data.");
19892c1f46dcSZachary Turner         return error;
19902c1f46dcSZachary Turner     }
19912c1f46dcSZachary Turner 
19922c1f46dcSZachary Turner     std::string auto_generated_function_name(GenerateUniqueName("lldb_autogen_python_bp_callback_func_",num_created_functions));
19932c1f46dcSZachary Turner     sstr.Printf ("def %s (frame, bp_loc, internal_dict):", auto_generated_function_name.c_str());
19942c1f46dcSZachary Turner 
19952c1f46dcSZachary Turner     error = GenerateFunction(sstr.GetData(), user_input);
19962c1f46dcSZachary Turner     if (!error.Success())
19972c1f46dcSZachary Turner         return error;
19982c1f46dcSZachary Turner 
19992c1f46dcSZachary Turner     // Store the name of the auto-generated function to be called.
20002c1f46dcSZachary Turner     output.assign(auto_generated_function_name);
20012c1f46dcSZachary Turner     return error;
20022c1f46dcSZachary Turner }
20032c1f46dcSZachary Turner 
20042c1f46dcSZachary Turner bool
20052c1f46dcSZachary Turner ScriptInterpreterPython::GenerateWatchpointCommandCallbackData (StringList &user_input, std::string& output)
20062c1f46dcSZachary Turner {
20072c1f46dcSZachary Turner     static uint32_t num_created_functions = 0;
20082c1f46dcSZachary Turner     user_input.RemoveBlankLines ();
20092c1f46dcSZachary Turner     StreamString sstr;
20102c1f46dcSZachary Turner 
20112c1f46dcSZachary Turner     if (user_input.GetSize() == 0)
20122c1f46dcSZachary Turner         return false;
20132c1f46dcSZachary Turner 
20142c1f46dcSZachary Turner     std::string auto_generated_function_name(GenerateUniqueName("lldb_autogen_python_wp_callback_func_",num_created_functions));
20152c1f46dcSZachary Turner     sstr.Printf ("def %s (frame, wp, internal_dict):", auto_generated_function_name.c_str());
20162c1f46dcSZachary Turner 
20172c1f46dcSZachary Turner     if (!GenerateFunction(sstr.GetData(), user_input).Success())
20182c1f46dcSZachary Turner         return false;
20192c1f46dcSZachary Turner 
20202c1f46dcSZachary Turner     // Store the name of the auto-generated function to be called.
20212c1f46dcSZachary Turner     output.assign(auto_generated_function_name);
20222c1f46dcSZachary Turner     return true;
20232c1f46dcSZachary Turner }
20242c1f46dcSZachary Turner 
20252c1f46dcSZachary Turner bool
20262c1f46dcSZachary Turner ScriptInterpreterPython::GetScriptedSummary(const char *python_function_name, lldb::ValueObjectSP valobj,
20272c1f46dcSZachary Turner                                             StructuredData::ObjectSP &callee_wrapper_sp, const TypeSummaryOptions &options,
20282c1f46dcSZachary Turner                                             std::string &retval)
20292c1f46dcSZachary Turner {
20302c1f46dcSZachary Turner 
20312c1f46dcSZachary Turner     Timer scoped_timer (__PRETTY_FUNCTION__, __PRETTY_FUNCTION__);
20322c1f46dcSZachary Turner 
20332c1f46dcSZachary Turner     if (!valobj.get())
20342c1f46dcSZachary Turner     {
20352c1f46dcSZachary Turner         retval.assign("<no object>");
20362c1f46dcSZachary Turner         return false;
20372c1f46dcSZachary Turner     }
20382c1f46dcSZachary Turner 
20392c1f46dcSZachary Turner     void *old_callee = nullptr;
20402c1f46dcSZachary Turner     StructuredData::Generic *generic = nullptr;
20412c1f46dcSZachary Turner     if (callee_wrapper_sp)
20422c1f46dcSZachary Turner     {
20432c1f46dcSZachary Turner         generic = callee_wrapper_sp->GetAsGeneric();
20442c1f46dcSZachary Turner         if (generic)
20452c1f46dcSZachary Turner             old_callee = generic->GetValue();
20462c1f46dcSZachary Turner     }
20472c1f46dcSZachary Turner     void* new_callee = old_callee;
20482c1f46dcSZachary Turner 
20492c1f46dcSZachary Turner     bool ret_val;
20502c1f46dcSZachary Turner     if (python_function_name && *python_function_name)
20512c1f46dcSZachary Turner     {
20522c1f46dcSZachary Turner         {
20532c1f46dcSZachary Turner             Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
20542c1f46dcSZachary Turner             {
20552c1f46dcSZachary Turner                 TypeSummaryOptionsSP options_sp(new TypeSummaryOptions(options));
20562c1f46dcSZachary Turner 
20572c1f46dcSZachary Turner                 Timer scoped_timer ("g_swig_typescript_callback","g_swig_typescript_callback");
20582c1f46dcSZachary Turner                 ret_val = g_swig_typescript_callback (python_function_name,
20592c1f46dcSZachary Turner                                                       GetSessionDictionary().get(),
20602c1f46dcSZachary Turner                                                       valobj,
20612c1f46dcSZachary Turner                                                       &new_callee,
20622c1f46dcSZachary Turner                                                       options_sp,
20632c1f46dcSZachary Turner                                                       retval);
20642c1f46dcSZachary Turner             }
20652c1f46dcSZachary Turner         }
20662c1f46dcSZachary Turner     }
20672c1f46dcSZachary Turner     else
20682c1f46dcSZachary Turner     {
20692c1f46dcSZachary Turner         retval.assign("<no function name>");
20702c1f46dcSZachary Turner         return false;
20712c1f46dcSZachary Turner     }
20722c1f46dcSZachary Turner 
20732c1f46dcSZachary Turner     if (new_callee && old_callee != new_callee)
20742c1f46dcSZachary Turner         callee_wrapper_sp.reset(new StructuredPythonObject(new_callee));
20752c1f46dcSZachary Turner 
20762c1f46dcSZachary Turner     return ret_val;
20772c1f46dcSZachary Turner }
20782c1f46dcSZachary Turner 
20792c1f46dcSZachary Turner void
20802c1f46dcSZachary Turner ScriptInterpreterPython::Clear ()
20812c1f46dcSZachary Turner {
20822c1f46dcSZachary Turner     // Release any global variables that might have strong references to
20832c1f46dcSZachary Turner     // LLDB objects when clearing the python script interpreter.
20842c1f46dcSZachary Turner     Locker locker(this,
20852c1f46dcSZachary Turner                   ScriptInterpreterPython::Locker::AcquireLock,
20862c1f46dcSZachary Turner                   ScriptInterpreterPython::Locker::FreeAcquiredLock);
20872c1f46dcSZachary Turner 
20882c1f46dcSZachary Turner     // This may be called as part of Py_Finalize.  In that case the modules are destroyed in random
20892c1f46dcSZachary Turner     // order and we can't guarantee that we can access these.
20902c1f46dcSZachary Turner     if (Py_IsInitialized())
20912c1f46dcSZachary Turner         PyRun_SimpleString("lldb.debugger = None; lldb.target = None; lldb.process = None; lldb.thread = None; lldb.frame = None");
20922c1f46dcSZachary Turner }
20932c1f46dcSZachary Turner 
20942c1f46dcSZachary Turner bool
20952c1f46dcSZachary Turner ScriptInterpreterPython::BreakpointCallbackFunction
20962c1f46dcSZachary Turner (
20972c1f46dcSZachary Turner     void *baton,
20982c1f46dcSZachary Turner     StoppointCallbackContext *context,
20992c1f46dcSZachary Turner     user_id_t break_id,
21002c1f46dcSZachary Turner     user_id_t break_loc_id
21012c1f46dcSZachary Turner )
21022c1f46dcSZachary Turner {
21032c1f46dcSZachary Turner     BreakpointOptions::CommandData *bp_option_data = (BreakpointOptions::CommandData *) baton;
21042c1f46dcSZachary Turner     const char *python_function_name = bp_option_data->script_source.c_str();
21052c1f46dcSZachary Turner 
21062c1f46dcSZachary Turner     if (!context)
21072c1f46dcSZachary Turner         return true;
21082c1f46dcSZachary Turner 
21092c1f46dcSZachary Turner     ExecutionContext exe_ctx (context->exe_ctx_ref);
21102c1f46dcSZachary Turner     Target *target = exe_ctx.GetTargetPtr();
21112c1f46dcSZachary Turner 
21122c1f46dcSZachary Turner     if (!target)
21132c1f46dcSZachary Turner         return true;
21142c1f46dcSZachary Turner 
21152c1f46dcSZachary Turner     Debugger &debugger = target->GetDebugger();
21162c1f46dcSZachary Turner     ScriptInterpreter *script_interpreter = debugger.GetCommandInterpreter().GetScriptInterpreter();
21172c1f46dcSZachary Turner     ScriptInterpreterPython *python_interpreter = (ScriptInterpreterPython *) script_interpreter;
21182c1f46dcSZachary Turner 
21192c1f46dcSZachary Turner     if (!script_interpreter)
21202c1f46dcSZachary Turner         return true;
21212c1f46dcSZachary Turner 
21222c1f46dcSZachary Turner     if (python_function_name && python_function_name[0])
21232c1f46dcSZachary Turner     {
21242c1f46dcSZachary Turner         const StackFrameSP stop_frame_sp (exe_ctx.GetFrameSP());
21252c1f46dcSZachary Turner         BreakpointSP breakpoint_sp = target->GetBreakpointByID (break_id);
21262c1f46dcSZachary Turner         if (breakpoint_sp)
21272c1f46dcSZachary Turner         {
21282c1f46dcSZachary Turner             const BreakpointLocationSP bp_loc_sp (breakpoint_sp->FindLocationByID (break_loc_id));
21292c1f46dcSZachary Turner 
21302c1f46dcSZachary Turner             if (stop_frame_sp && bp_loc_sp)
21312c1f46dcSZachary Turner             {
21322c1f46dcSZachary Turner                 bool ret_val = true;
21332c1f46dcSZachary Turner                 {
21342c1f46dcSZachary Turner                     Locker py_lock(python_interpreter, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
21352c1f46dcSZachary Turner                     ret_val = g_swig_breakpoint_callback (python_function_name,
21362c1f46dcSZachary Turner                                                           python_interpreter->m_dictionary_name.c_str(),
21372c1f46dcSZachary Turner                                                           stop_frame_sp,
21382c1f46dcSZachary Turner                                                           bp_loc_sp);
21392c1f46dcSZachary Turner                 }
21402c1f46dcSZachary Turner                 return ret_val;
21412c1f46dcSZachary Turner             }
21422c1f46dcSZachary Turner         }
21432c1f46dcSZachary Turner     }
21442c1f46dcSZachary Turner     // We currently always true so we stop in case anything goes wrong when
21452c1f46dcSZachary Turner     // trying to call the script function
21462c1f46dcSZachary Turner     return true;
21472c1f46dcSZachary Turner }
21482c1f46dcSZachary Turner 
21492c1f46dcSZachary Turner bool
21502c1f46dcSZachary Turner ScriptInterpreterPython::WatchpointCallbackFunction
21512c1f46dcSZachary Turner (
21522c1f46dcSZachary Turner     void *baton,
21532c1f46dcSZachary Turner     StoppointCallbackContext *context,
21542c1f46dcSZachary Turner     user_id_t watch_id
21552c1f46dcSZachary Turner )
21562c1f46dcSZachary Turner {
21572c1f46dcSZachary Turner     WatchpointOptions::CommandData *wp_option_data = (WatchpointOptions::CommandData *) baton;
21582c1f46dcSZachary Turner     const char *python_function_name = wp_option_data->script_source.c_str();
21592c1f46dcSZachary Turner 
21602c1f46dcSZachary Turner     if (!context)
21612c1f46dcSZachary Turner         return true;
21622c1f46dcSZachary Turner 
21632c1f46dcSZachary Turner     ExecutionContext exe_ctx (context->exe_ctx_ref);
21642c1f46dcSZachary Turner     Target *target = exe_ctx.GetTargetPtr();
21652c1f46dcSZachary Turner 
21662c1f46dcSZachary Turner     if (!target)
21672c1f46dcSZachary Turner         return true;
21682c1f46dcSZachary Turner 
21692c1f46dcSZachary Turner     Debugger &debugger = target->GetDebugger();
21702c1f46dcSZachary Turner     ScriptInterpreter *script_interpreter = debugger.GetCommandInterpreter().GetScriptInterpreter();
21712c1f46dcSZachary Turner     ScriptInterpreterPython *python_interpreter = (ScriptInterpreterPython *) script_interpreter;
21722c1f46dcSZachary Turner 
21732c1f46dcSZachary Turner     if (!script_interpreter)
21742c1f46dcSZachary Turner         return true;
21752c1f46dcSZachary Turner 
21762c1f46dcSZachary Turner     if (python_function_name && python_function_name[0])
21772c1f46dcSZachary Turner     {
21782c1f46dcSZachary Turner         const StackFrameSP stop_frame_sp (exe_ctx.GetFrameSP());
21792c1f46dcSZachary Turner         WatchpointSP wp_sp = target->GetWatchpointList().FindByID (watch_id);
21802c1f46dcSZachary Turner         if (wp_sp)
21812c1f46dcSZachary Turner         {
21822c1f46dcSZachary Turner             if (stop_frame_sp && wp_sp)
21832c1f46dcSZachary Turner             {
21842c1f46dcSZachary Turner                 bool ret_val = true;
21852c1f46dcSZachary Turner                 {
21862c1f46dcSZachary Turner                     Locker py_lock(python_interpreter, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
21872c1f46dcSZachary Turner                     ret_val = g_swig_watchpoint_callback (python_function_name,
21882c1f46dcSZachary Turner                                                           python_interpreter->m_dictionary_name.c_str(),
21892c1f46dcSZachary Turner                                                           stop_frame_sp,
21902c1f46dcSZachary Turner                                                           wp_sp);
21912c1f46dcSZachary Turner                 }
21922c1f46dcSZachary Turner                 return ret_val;
21932c1f46dcSZachary Turner             }
21942c1f46dcSZachary Turner         }
21952c1f46dcSZachary Turner     }
21962c1f46dcSZachary Turner     // We currently always true so we stop in case anything goes wrong when
21972c1f46dcSZachary Turner     // trying to call the script function
21982c1f46dcSZachary Turner     return true;
21992c1f46dcSZachary Turner }
22002c1f46dcSZachary Turner 
22012c1f46dcSZachary Turner size_t
22022c1f46dcSZachary Turner ScriptInterpreterPython::CalculateNumChildren(const StructuredData::ObjectSP &implementor_sp)
22032c1f46dcSZachary Turner {
22042c1f46dcSZachary Turner     if (!implementor_sp)
22052c1f46dcSZachary Turner         return 0;
22062c1f46dcSZachary Turner     StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
22072c1f46dcSZachary Turner     if (!generic)
22082c1f46dcSZachary Turner         return 0;
22092c1f46dcSZachary Turner     void *implementor = generic->GetValue();
22102c1f46dcSZachary Turner     if (!implementor)
22112c1f46dcSZachary Turner         return 0;
22122c1f46dcSZachary Turner 
22132c1f46dcSZachary Turner     if (!g_swig_calc_children)
22142c1f46dcSZachary Turner         return 0;
22152c1f46dcSZachary Turner 
22162c1f46dcSZachary Turner     size_t ret_val = 0;
22172c1f46dcSZachary Turner 
22182c1f46dcSZachary Turner     {
22192c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
22202c1f46dcSZachary Turner         ret_val = g_swig_calc_children (implementor);
22212c1f46dcSZachary Turner     }
22222c1f46dcSZachary Turner 
22232c1f46dcSZachary Turner     return ret_val;
22242c1f46dcSZachary Turner }
22252c1f46dcSZachary Turner 
22262c1f46dcSZachary Turner lldb::ValueObjectSP
22272c1f46dcSZachary Turner ScriptInterpreterPython::GetChildAtIndex(const StructuredData::ObjectSP &implementor_sp, uint32_t idx)
22282c1f46dcSZachary Turner {
22292c1f46dcSZachary Turner     if (!implementor_sp)
22302c1f46dcSZachary Turner         return lldb::ValueObjectSP();
22312c1f46dcSZachary Turner 
22322c1f46dcSZachary Turner     StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
22332c1f46dcSZachary Turner     if (!generic)
22342c1f46dcSZachary Turner         return lldb::ValueObjectSP();
22352c1f46dcSZachary Turner     void *implementor = generic->GetValue();
22362c1f46dcSZachary Turner     if (!implementor)
22372c1f46dcSZachary Turner         return lldb::ValueObjectSP();
22382c1f46dcSZachary Turner 
22392c1f46dcSZachary Turner     if (!g_swig_get_child_index || !g_swig_cast_to_sbvalue)
22402c1f46dcSZachary Turner         return lldb::ValueObjectSP();
22412c1f46dcSZachary Turner 
22422c1f46dcSZachary Turner     lldb::ValueObjectSP ret_val;
22432c1f46dcSZachary Turner 
22442c1f46dcSZachary Turner     {
22452c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
22462c1f46dcSZachary Turner         void* child_ptr = g_swig_get_child_index (implementor,idx);
22472c1f46dcSZachary Turner         if (child_ptr != nullptr && child_ptr != Py_None)
22482c1f46dcSZachary Turner         {
22492c1f46dcSZachary Turner             lldb::SBValue* sb_value_ptr = (lldb::SBValue*)g_swig_cast_to_sbvalue(child_ptr);
22502c1f46dcSZachary Turner             if (sb_value_ptr == nullptr)
22512c1f46dcSZachary Turner                 Py_XDECREF(child_ptr);
22522c1f46dcSZachary Turner             else
22532c1f46dcSZachary Turner                 ret_val = g_swig_get_valobj_sp_from_sbvalue (sb_value_ptr);
22542c1f46dcSZachary Turner         }
22552c1f46dcSZachary Turner         else
22562c1f46dcSZachary Turner         {
22572c1f46dcSZachary Turner             Py_XDECREF(child_ptr);
22582c1f46dcSZachary Turner         }
22592c1f46dcSZachary Turner     }
22602c1f46dcSZachary Turner 
22612c1f46dcSZachary Turner     return ret_val;
22622c1f46dcSZachary Turner }
22632c1f46dcSZachary Turner 
22642c1f46dcSZachary Turner int
22652c1f46dcSZachary Turner ScriptInterpreterPython::GetIndexOfChildWithName(const StructuredData::ObjectSP &implementor_sp, const char *child_name)
22662c1f46dcSZachary Turner {
22672c1f46dcSZachary Turner     if (!implementor_sp)
22682c1f46dcSZachary Turner         return UINT32_MAX;
22692c1f46dcSZachary Turner 
22702c1f46dcSZachary Turner     StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
22712c1f46dcSZachary Turner     if (!generic)
22722c1f46dcSZachary Turner         return UINT32_MAX;
22732c1f46dcSZachary Turner     void *implementor = generic->GetValue();
22742c1f46dcSZachary Turner     if (!implementor)
22752c1f46dcSZachary Turner         return UINT32_MAX;
22762c1f46dcSZachary Turner 
22772c1f46dcSZachary Turner     if (!g_swig_get_index_child)
22782c1f46dcSZachary Turner         return UINT32_MAX;
22792c1f46dcSZachary Turner 
22802c1f46dcSZachary Turner     int ret_val = UINT32_MAX;
22812c1f46dcSZachary Turner 
22822c1f46dcSZachary Turner     {
22832c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
22842c1f46dcSZachary Turner         ret_val = g_swig_get_index_child (implementor, child_name);
22852c1f46dcSZachary Turner     }
22862c1f46dcSZachary Turner 
22872c1f46dcSZachary Turner     return ret_val;
22882c1f46dcSZachary Turner }
22892c1f46dcSZachary Turner 
22902c1f46dcSZachary Turner bool
22912c1f46dcSZachary Turner ScriptInterpreterPython::UpdateSynthProviderInstance(const StructuredData::ObjectSP &implementor_sp)
22922c1f46dcSZachary Turner {
22932c1f46dcSZachary Turner     bool ret_val = false;
22942c1f46dcSZachary Turner 
22952c1f46dcSZachary Turner     if (!implementor_sp)
22962c1f46dcSZachary Turner         return ret_val;
22972c1f46dcSZachary Turner 
22982c1f46dcSZachary Turner     StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
22992c1f46dcSZachary Turner     if (!generic)
23002c1f46dcSZachary Turner         return ret_val;
23012c1f46dcSZachary Turner     void *implementor = generic->GetValue();
23022c1f46dcSZachary Turner     if (!implementor)
23032c1f46dcSZachary Turner         return ret_val;
23042c1f46dcSZachary Turner 
23052c1f46dcSZachary Turner     if (!g_swig_update_provider)
23062c1f46dcSZachary Turner         return ret_val;
23072c1f46dcSZachary Turner 
23082c1f46dcSZachary Turner     {
23092c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
23102c1f46dcSZachary Turner         ret_val = g_swig_update_provider (implementor);
23112c1f46dcSZachary Turner     }
23122c1f46dcSZachary Turner 
23132c1f46dcSZachary Turner     return ret_val;
23142c1f46dcSZachary Turner }
23152c1f46dcSZachary Turner 
23162c1f46dcSZachary Turner bool
23172c1f46dcSZachary Turner ScriptInterpreterPython::MightHaveChildrenSynthProviderInstance(const StructuredData::ObjectSP &implementor_sp)
23182c1f46dcSZachary Turner {
23192c1f46dcSZachary Turner     bool ret_val = false;
23202c1f46dcSZachary Turner 
23212c1f46dcSZachary Turner     if (!implementor_sp)
23222c1f46dcSZachary Turner         return ret_val;
23232c1f46dcSZachary Turner 
23242c1f46dcSZachary Turner     StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
23252c1f46dcSZachary Turner     if (!generic)
23262c1f46dcSZachary Turner         return ret_val;
23272c1f46dcSZachary Turner     void *implementor = generic->GetValue();
23282c1f46dcSZachary Turner     if (!implementor)
23292c1f46dcSZachary Turner         return ret_val;
23302c1f46dcSZachary Turner 
23312c1f46dcSZachary Turner     if (!g_swig_mighthavechildren_provider)
23322c1f46dcSZachary Turner         return ret_val;
23332c1f46dcSZachary Turner 
23342c1f46dcSZachary Turner     {
23352c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
23362c1f46dcSZachary Turner         ret_val = g_swig_mighthavechildren_provider (implementor);
23372c1f46dcSZachary Turner     }
23382c1f46dcSZachary Turner 
23392c1f46dcSZachary Turner     return ret_val;
23402c1f46dcSZachary Turner }
23412c1f46dcSZachary Turner 
23422c1f46dcSZachary Turner lldb::ValueObjectSP
23432c1f46dcSZachary Turner ScriptInterpreterPython::GetSyntheticValue(const StructuredData::ObjectSP &implementor_sp)
23442c1f46dcSZachary Turner {
23452c1f46dcSZachary Turner     lldb::ValueObjectSP ret_val(nullptr);
23462c1f46dcSZachary Turner 
23472c1f46dcSZachary Turner     if (!implementor_sp)
23482c1f46dcSZachary Turner         return ret_val;
23492c1f46dcSZachary Turner 
23502c1f46dcSZachary Turner     StructuredData::Generic *generic = implementor_sp->GetAsGeneric();
23512c1f46dcSZachary Turner     if (!generic)
23522c1f46dcSZachary Turner         return ret_val;
23532c1f46dcSZachary Turner     void *implementor = generic->GetValue();
23542c1f46dcSZachary Turner     if (!implementor)
23552c1f46dcSZachary Turner         return ret_val;
23562c1f46dcSZachary Turner 
23572c1f46dcSZachary Turner     if (!g_swig_getvalue_provider || !g_swig_cast_to_sbvalue || !g_swig_get_valobj_sp_from_sbvalue)
23582c1f46dcSZachary Turner         return ret_val;
23592c1f46dcSZachary Turner 
23602c1f46dcSZachary Turner     {
23612c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
23622c1f46dcSZachary Turner         void* child_ptr = g_swig_getvalue_provider (implementor);
23632c1f46dcSZachary Turner         if (child_ptr != nullptr && child_ptr != Py_None)
23642c1f46dcSZachary Turner         {
23652c1f46dcSZachary Turner             lldb::SBValue* sb_value_ptr = (lldb::SBValue*)g_swig_cast_to_sbvalue(child_ptr);
23662c1f46dcSZachary Turner             if (sb_value_ptr == nullptr)
23672c1f46dcSZachary Turner                 Py_XDECREF(child_ptr);
23682c1f46dcSZachary Turner             else
23692c1f46dcSZachary Turner                 ret_val = g_swig_get_valobj_sp_from_sbvalue (sb_value_ptr);
23702c1f46dcSZachary Turner         }
23712c1f46dcSZachary Turner         else
23722c1f46dcSZachary Turner         {
23732c1f46dcSZachary Turner             Py_XDECREF(child_ptr);
23742c1f46dcSZachary Turner         }
23752c1f46dcSZachary Turner     }
23762c1f46dcSZachary Turner 
23772c1f46dcSZachary Turner     return ret_val;
23782c1f46dcSZachary Turner }
23792c1f46dcSZachary Turner 
23802c1f46dcSZachary Turner static std::string
2381f8b22f8fSZachary Turner ReadPythonBacktrace(PythonObject py_backtrace)
23822c1f46dcSZachary Turner {
2383f8b22f8fSZachary Turner     PythonObject traceback_module;
2384f8b22f8fSZachary Turner     PythonObject stringIO_module;
2385f8b22f8fSZachary Turner     PythonObject stringIO_builder;
2386f8b22f8fSZachary Turner     PythonObject stringIO_buffer;
2387f8b22f8fSZachary Turner     PythonObject printTB;
2388f8b22f8fSZachary Turner     PythonObject printTB_args;
2389f8b22f8fSZachary Turner     PythonObject printTB_result;
2390f8b22f8fSZachary Turner     PythonObject stringIO_getvalue;
2391f8b22f8fSZachary Turner     PythonObject printTB_string;
23922c1f46dcSZachary Turner 
23932c1f46dcSZachary Turner     std::string retval("backtrace unavailable");
23942c1f46dcSZachary Turner 
2395f8b22f8fSZachary Turner     if (!py_backtrace.IsAllocated())
2396f8b22f8fSZachary Turner         return retval;
23972c1f46dcSZachary Turner 
2398f8b22f8fSZachary Turner     traceback_module.Reset(PyRefType::Owned, PyImport_ImportModule("traceback"));
2399f8b22f8fSZachary Turner     stringIO_module.Reset(PyRefType::Owned, PyImport_ImportModule("StringIO"));
2400f8b22f8fSZachary Turner     if (!traceback_module.IsAllocated() || !stringIO_module.IsAllocated())
2401f8b22f8fSZachary Turner         return retval;
2402f8b22f8fSZachary Turner 
2403f8b22f8fSZachary Turner     stringIO_builder.Reset(PyRefType::Owned, PyObject_GetAttrString(stringIO_module.get(), "StringIO"));
2404f8b22f8fSZachary Turner     if (!stringIO_builder.IsAllocated())
2405f8b22f8fSZachary Turner         return retval;
2406f8b22f8fSZachary Turner 
2407f8b22f8fSZachary Turner     stringIO_buffer.Reset(PyRefType::Owned, PyObject_CallObject(stringIO_builder.get(), nullptr));
2408f8b22f8fSZachary Turner     if (!stringIO_buffer.IsAllocated())
2409f8b22f8fSZachary Turner         return retval;
2410f8b22f8fSZachary Turner 
2411f8b22f8fSZachary Turner     printTB.Reset(PyRefType::Owned, PyObject_GetAttrString(traceback_module.get(), "print_tb"));
2412f8b22f8fSZachary Turner     if (!printTB.IsAllocated())
2413f8b22f8fSZachary Turner         return retval;
2414f8b22f8fSZachary Turner 
2415f8b22f8fSZachary Turner     printTB_args.Reset(PyRefType::Owned, Py_BuildValue("OOO", py_backtrace.get(), Py_None, stringIO_buffer.get()));
2416f8b22f8fSZachary Turner     printTB_result.Reset(PyRefType::Owned, PyObject_CallObject(printTB.get(), printTB_args.get()));
2417f8b22f8fSZachary Turner     stringIO_getvalue.Reset(PyRefType::Owned, PyObject_GetAttrString(stringIO_buffer.get(), "getvalue"));
2418f8b22f8fSZachary Turner     if (!stringIO_getvalue.IsAllocated())
2419f8b22f8fSZachary Turner         return retval;
2420f8b22f8fSZachary Turner 
2421f8b22f8fSZachary Turner     printTB_string.Reset(PyRefType::Owned, PyObject_CallObject(stringIO_getvalue.get(), nullptr));
2422f8b22f8fSZachary Turner     if (!printTB_string.IsAllocated())
2423f8b22f8fSZachary Turner         return retval;
2424f8b22f8fSZachary Turner 
2425f8b22f8fSZachary Turner     PythonString str(PyRefType::Borrowed, printTB_string.get());
242622c8efcdSZachary Turner     llvm::StringRef string_data(str.GetString());
242722c8efcdSZachary Turner     retval.assign(string_data.data(), string_data.size());
2428f8b22f8fSZachary Turner 
24292c1f46dcSZachary Turner     return retval;
24302c1f46dcSZachary Turner }
24312c1f46dcSZachary Turner 
24322c1f46dcSZachary Turner bool
24332c1f46dcSZachary Turner ScriptInterpreterPython::RunScriptFormatKeyword (const char* impl_function,
24342c1f46dcSZachary Turner                                                  Process* process,
24352c1f46dcSZachary Turner                                                  std::string& output,
24362c1f46dcSZachary Turner                                                  Error& error)
24372c1f46dcSZachary Turner {
24382c1f46dcSZachary Turner     bool ret_val;
24392c1f46dcSZachary Turner     if (!process)
24402c1f46dcSZachary Turner     {
24412c1f46dcSZachary Turner         error.SetErrorString("no process");
24422c1f46dcSZachary Turner         return false;
24432c1f46dcSZachary Turner     }
24442c1f46dcSZachary Turner     if (!impl_function || !impl_function[0])
24452c1f46dcSZachary Turner     {
24462c1f46dcSZachary Turner         error.SetErrorString("no function to execute");
24472c1f46dcSZachary Turner         return false;
24482c1f46dcSZachary Turner     }
24492c1f46dcSZachary Turner     if (!g_swig_run_script_keyword_process)
24502c1f46dcSZachary Turner     {
24512c1f46dcSZachary Turner         error.SetErrorString("internal helper function missing");
24522c1f46dcSZachary Turner         return false;
24532c1f46dcSZachary Turner     }
24542c1f46dcSZachary Turner     {
24552c1f46dcSZachary Turner         ProcessSP process_sp(process->shared_from_this());
24562c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
24572c1f46dcSZachary Turner         ret_val = g_swig_run_script_keyword_process (impl_function, m_dictionary_name.c_str(), process_sp, output);
24582c1f46dcSZachary Turner         if (!ret_val)
24592c1f46dcSZachary Turner             error.SetErrorString("python script evaluation failed");
24602c1f46dcSZachary Turner     }
24612c1f46dcSZachary Turner     return ret_val;
24622c1f46dcSZachary Turner }
24632c1f46dcSZachary Turner 
24642c1f46dcSZachary Turner bool
24652c1f46dcSZachary Turner ScriptInterpreterPython::RunScriptFormatKeyword (const char* impl_function,
24662c1f46dcSZachary Turner                                                  Thread* thread,
24672c1f46dcSZachary Turner                                                  std::string& output,
24682c1f46dcSZachary Turner                                                  Error& error)
24692c1f46dcSZachary Turner {
24702c1f46dcSZachary Turner     bool ret_val;
24712c1f46dcSZachary Turner     if (!thread)
24722c1f46dcSZachary Turner     {
24732c1f46dcSZachary Turner         error.SetErrorString("no thread");
24742c1f46dcSZachary Turner         return false;
24752c1f46dcSZachary Turner     }
24762c1f46dcSZachary Turner     if (!impl_function || !impl_function[0])
24772c1f46dcSZachary Turner     {
24782c1f46dcSZachary Turner         error.SetErrorString("no function to execute");
24792c1f46dcSZachary Turner         return false;
24802c1f46dcSZachary Turner     }
24812c1f46dcSZachary Turner     if (!g_swig_run_script_keyword_thread)
24822c1f46dcSZachary Turner     {
24832c1f46dcSZachary Turner         error.SetErrorString("internal helper function missing");
24842c1f46dcSZachary Turner         return false;
24852c1f46dcSZachary Turner     }
24862c1f46dcSZachary Turner     {
24872c1f46dcSZachary Turner         ThreadSP thread_sp(thread->shared_from_this());
24882c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
24892c1f46dcSZachary Turner         ret_val = g_swig_run_script_keyword_thread (impl_function, m_dictionary_name.c_str(), thread_sp, output);
24902c1f46dcSZachary Turner         if (!ret_val)
24912c1f46dcSZachary Turner             error.SetErrorString("python script evaluation failed");
24922c1f46dcSZachary Turner     }
24932c1f46dcSZachary Turner     return ret_val;
24942c1f46dcSZachary Turner }
24952c1f46dcSZachary Turner 
24962c1f46dcSZachary Turner bool
24972c1f46dcSZachary Turner ScriptInterpreterPython::RunScriptFormatKeyword (const char* impl_function,
24982c1f46dcSZachary Turner                                                  Target* target,
24992c1f46dcSZachary Turner                                                  std::string& output,
25002c1f46dcSZachary Turner                                                  Error& error)
25012c1f46dcSZachary Turner {
25022c1f46dcSZachary Turner     bool ret_val;
25032c1f46dcSZachary Turner     if (!target)
25042c1f46dcSZachary Turner     {
25052c1f46dcSZachary Turner         error.SetErrorString("no thread");
25062c1f46dcSZachary Turner         return false;
25072c1f46dcSZachary Turner     }
25082c1f46dcSZachary Turner     if (!impl_function || !impl_function[0])
25092c1f46dcSZachary Turner     {
25102c1f46dcSZachary Turner         error.SetErrorString("no function to execute");
25112c1f46dcSZachary Turner         return false;
25122c1f46dcSZachary Turner     }
25132c1f46dcSZachary Turner     if (!g_swig_run_script_keyword_target)
25142c1f46dcSZachary Turner     {
25152c1f46dcSZachary Turner         error.SetErrorString("internal helper function missing");
25162c1f46dcSZachary Turner         return false;
25172c1f46dcSZachary Turner     }
25182c1f46dcSZachary Turner     {
25192c1f46dcSZachary Turner         TargetSP target_sp(target->shared_from_this());
25202c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
25212c1f46dcSZachary Turner         ret_val = g_swig_run_script_keyword_target (impl_function, m_dictionary_name.c_str(), target_sp, output);
25222c1f46dcSZachary Turner         if (!ret_val)
25232c1f46dcSZachary Turner             error.SetErrorString("python script evaluation failed");
25242c1f46dcSZachary Turner     }
25252c1f46dcSZachary Turner     return ret_val;
25262c1f46dcSZachary Turner }
25272c1f46dcSZachary Turner 
25282c1f46dcSZachary Turner bool
25292c1f46dcSZachary Turner ScriptInterpreterPython::RunScriptFormatKeyword (const char* impl_function,
25302c1f46dcSZachary Turner                                                  StackFrame* frame,
25312c1f46dcSZachary Turner                                                  std::string& output,
25322c1f46dcSZachary Turner                                                  Error& error)
25332c1f46dcSZachary Turner {
25342c1f46dcSZachary Turner     bool ret_val;
25352c1f46dcSZachary Turner     if (!frame)
25362c1f46dcSZachary Turner     {
25372c1f46dcSZachary Turner         error.SetErrorString("no frame");
25382c1f46dcSZachary Turner         return false;
25392c1f46dcSZachary Turner     }
25402c1f46dcSZachary Turner     if (!impl_function || !impl_function[0])
25412c1f46dcSZachary Turner     {
25422c1f46dcSZachary Turner         error.SetErrorString("no function to execute");
25432c1f46dcSZachary Turner         return false;
25442c1f46dcSZachary Turner     }
25452c1f46dcSZachary Turner     if (!g_swig_run_script_keyword_frame)
25462c1f46dcSZachary Turner     {
25472c1f46dcSZachary Turner         error.SetErrorString("internal helper function missing");
25482c1f46dcSZachary Turner         return false;
25492c1f46dcSZachary Turner     }
25502c1f46dcSZachary Turner     {
25512c1f46dcSZachary Turner         StackFrameSP frame_sp(frame->shared_from_this());
25522c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
25532c1f46dcSZachary Turner         ret_val = g_swig_run_script_keyword_frame (impl_function, m_dictionary_name.c_str(), frame_sp, output);
25542c1f46dcSZachary Turner         if (!ret_val)
25552c1f46dcSZachary Turner             error.SetErrorString("python script evaluation failed");
25562c1f46dcSZachary Turner     }
25572c1f46dcSZachary Turner     return ret_val;
25582c1f46dcSZachary Turner }
25592c1f46dcSZachary Turner 
25602c1f46dcSZachary Turner bool
25612c1f46dcSZachary Turner ScriptInterpreterPython::RunScriptFormatKeyword (const char* impl_function,
25622c1f46dcSZachary Turner                                                  ValueObject *value,
25632c1f46dcSZachary Turner                                                  std::string& output,
25642c1f46dcSZachary Turner                                                  Error& error)
25652c1f46dcSZachary Turner {
25662c1f46dcSZachary Turner     bool ret_val;
25672c1f46dcSZachary Turner     if (!value)
25682c1f46dcSZachary Turner     {
25692c1f46dcSZachary Turner         error.SetErrorString("no value");
25702c1f46dcSZachary Turner         return false;
25712c1f46dcSZachary Turner     }
25722c1f46dcSZachary Turner     if (!impl_function || !impl_function[0])
25732c1f46dcSZachary Turner     {
25742c1f46dcSZachary Turner         error.SetErrorString("no function to execute");
25752c1f46dcSZachary Turner         return false;
25762c1f46dcSZachary Turner     }
25772c1f46dcSZachary Turner     if (!g_swig_run_script_keyword_value)
25782c1f46dcSZachary Turner     {
25792c1f46dcSZachary Turner         error.SetErrorString("internal helper function missing");
25802c1f46dcSZachary Turner         return false;
25812c1f46dcSZachary Turner     }
25822c1f46dcSZachary Turner     {
25832c1f46dcSZachary Turner         ValueObjectSP value_sp(value->GetSP());
25842c1f46dcSZachary Turner         Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN);
25852c1f46dcSZachary Turner         ret_val = g_swig_run_script_keyword_value (impl_function, m_dictionary_name.c_str(), value_sp, output);
25862c1f46dcSZachary Turner         if (!ret_val)
25872c1f46dcSZachary Turner             error.SetErrorString("python script evaluation failed");
25882c1f46dcSZachary Turner     }
25892c1f46dcSZachary Turner     return ret_val;
25902c1f46dcSZachary Turner }
25912c1f46dcSZachary Turner 
25922c1f46dcSZachary Turner uint64_t replace_all(std::string& str, const std::string& oldStr, const std::string& newStr)
25932c1f46dcSZachary Turner {
25942c1f46dcSZachary Turner     size_t pos = 0;
25952c1f46dcSZachary Turner     uint64_t matches = 0;
25962c1f46dcSZachary Turner     while((pos = str.find(oldStr, pos)) != std::string::npos)
25972c1f46dcSZachary Turner     {
25982c1f46dcSZachary Turner         matches++;
25992c1f46dcSZachary Turner         str.replace(pos, oldStr.length(), newStr);
26002c1f46dcSZachary Turner         pos += newStr.length();
26012c1f46dcSZachary Turner     }
26022c1f46dcSZachary Turner     return matches;
26032c1f46dcSZachary Turner }
26042c1f46dcSZachary Turner 
26052c1f46dcSZachary Turner bool
26062c1f46dcSZachary Turner ScriptInterpreterPython::LoadScriptingModule(const char *pathname, bool can_reload, bool init_session, lldb_private::Error &error,
26072c1f46dcSZachary Turner                                              StructuredData::ObjectSP *module_sp)
26082c1f46dcSZachary Turner {
26092c1f46dcSZachary Turner     if (!pathname || !pathname[0])
26102c1f46dcSZachary Turner     {
26112c1f46dcSZachary Turner         error.SetErrorString("invalid pathname");
26122c1f46dcSZachary Turner         return false;
26132c1f46dcSZachary Turner     }
26142c1f46dcSZachary Turner 
26152c1f46dcSZachary Turner     if (!g_swig_call_module_init)
26162c1f46dcSZachary Turner     {
26172c1f46dcSZachary Turner         error.SetErrorString("internal helper function missing");
26182c1f46dcSZachary Turner         return false;
26192c1f46dcSZachary Turner     }
26202c1f46dcSZachary Turner 
26212c1f46dcSZachary Turner     lldb::DebuggerSP debugger_sp = m_interpreter.GetDebugger().shared_from_this();
26222c1f46dcSZachary Turner 
26232c1f46dcSZachary Turner     {
26242c1f46dcSZachary Turner         FileSpec target_file(pathname, true);
26252c1f46dcSZachary Turner         std::string basename(target_file.GetFilename().GetCString());
26262c1f46dcSZachary Turner 
26272c1f46dcSZachary Turner         StreamString command_stream;
26282c1f46dcSZachary Turner 
26292c1f46dcSZachary Turner         // Before executing Pyton code, lock the GIL.
26302c1f46dcSZachary Turner         Locker py_lock (this,
26312c1f46dcSZachary Turner                         Locker::AcquireLock      | (init_session ? Locker::InitSession     : 0) | Locker::NoSTDIN,
26322c1f46dcSZachary Turner                         Locker::FreeAcquiredLock | (init_session ? Locker::TearDownSession : 0));
26332c1f46dcSZachary Turner 
26342c1f46dcSZachary Turner         if (target_file.GetFileType() == FileSpec::eFileTypeInvalid ||
26352c1f46dcSZachary Turner             target_file.GetFileType() == FileSpec::eFileTypeUnknown)
26362c1f46dcSZachary Turner         {
26372c1f46dcSZachary Turner             // if not a valid file of any sort, check if it might be a filename still
26382c1f46dcSZachary Turner             // dot can't be used but / and \ can, and if either is found, reject
26392c1f46dcSZachary Turner             if (strchr(pathname,'\\') || strchr(pathname,'/'))
26402c1f46dcSZachary Turner             {
26412c1f46dcSZachary Turner                 error.SetErrorString("invalid pathname");
26422c1f46dcSZachary Turner                 return false;
26432c1f46dcSZachary Turner             }
26442c1f46dcSZachary Turner             basename = pathname; // not a filename, probably a package of some sort, let it go through
26452c1f46dcSZachary Turner         }
26462c1f46dcSZachary Turner         else if (target_file.GetFileType() == FileSpec::eFileTypeDirectory ||
26472c1f46dcSZachary Turner                  target_file.GetFileType() == FileSpec::eFileTypeRegular ||
26482c1f46dcSZachary Turner                  target_file.GetFileType() == FileSpec::eFileTypeSymbolicLink)
26492c1f46dcSZachary Turner         {
26502c1f46dcSZachary Turner             std::string directory(target_file.GetDirectory().GetCString());
26512c1f46dcSZachary Turner             replace_all(directory,"'","\\'");
26522c1f46dcSZachary Turner 
26532c1f46dcSZachary Turner             // now make sure that Python has "directory" in the search path
26542c1f46dcSZachary Turner             StreamString command_stream;
26552c1f46dcSZachary Turner             command_stream.Printf("if not (sys.path.__contains__('%s')):\n    sys.path.insert(1,'%s');\n\n",
26562c1f46dcSZachary Turner                                   directory.c_str(),
26572c1f46dcSZachary Turner                                   directory.c_str());
26582c1f46dcSZachary Turner             bool syspath_retval = ExecuteMultipleLines(command_stream.GetData(), ScriptInterpreter::ExecuteScriptOptions().SetEnableIO(false).SetSetLLDBGlobals(false)).Success();
26592c1f46dcSZachary Turner             if (!syspath_retval)
26602c1f46dcSZachary Turner             {
26612c1f46dcSZachary Turner                 error.SetErrorString("Python sys.path handling failed");
26622c1f46dcSZachary Turner                 return false;
26632c1f46dcSZachary Turner             }
26642c1f46dcSZachary Turner 
26652c1f46dcSZachary Turner             // strip .py or .pyc extension
26662c1f46dcSZachary Turner             ConstString extension = target_file.GetFileNameExtension();
26672c1f46dcSZachary Turner             if (extension)
26682c1f46dcSZachary Turner             {
26692c1f46dcSZachary Turner                 if (::strcmp(extension.GetCString(), "py") == 0)
26702c1f46dcSZachary Turner                     basename.resize(basename.length()-3);
26712c1f46dcSZachary Turner                 else if(::strcmp(extension.GetCString(), "pyc") == 0)
26722c1f46dcSZachary Turner                     basename.resize(basename.length()-4);
26732c1f46dcSZachary Turner             }
26742c1f46dcSZachary Turner         }
26752c1f46dcSZachary Turner         else
26762c1f46dcSZachary Turner         {
26772c1f46dcSZachary Turner             error.SetErrorString("no known way to import this module specification");
26782c1f46dcSZachary Turner             return false;
26792c1f46dcSZachary Turner         }
26802c1f46dcSZachary Turner 
26812c1f46dcSZachary Turner         // check if the module is already import-ed
26822c1f46dcSZachary Turner         command_stream.Clear();
26832c1f46dcSZachary Turner         command_stream.Printf("sys.modules.__contains__('%s')",basename.c_str());
26842c1f46dcSZachary Turner         bool does_contain = false;
26852c1f46dcSZachary Turner         // this call will succeed if the module was ever imported in any Debugger in the lifetime of the process
26862c1f46dcSZachary Turner         // in which this LLDB framework is living
26872c1f46dcSZachary Turner         bool was_imported_globally = (ExecuteOneLineWithReturn(command_stream.GetData(),
26882c1f46dcSZachary Turner                                                                ScriptInterpreterPython::eScriptReturnTypeBool,
26892c1f46dcSZachary Turner                                                                &does_contain,
26902c1f46dcSZachary Turner                                                                ScriptInterpreter::ExecuteScriptOptions().SetEnableIO(false).SetSetLLDBGlobals(false)) && does_contain);
26912c1f46dcSZachary Turner         // this call will fail if the module was not imported in this Debugger before
26922c1f46dcSZachary Turner         command_stream.Clear();
26932c1f46dcSZachary Turner         command_stream.Printf("sys.getrefcount(%s)",basename.c_str());
2694f8b22f8fSZachary Turner         bool was_imported_locally = GetSessionDictionary().GetItemForKey(PythonString(basename)).IsAllocated();
26952c1f46dcSZachary Turner 
26962c1f46dcSZachary Turner         bool was_imported = (was_imported_globally || was_imported_locally);
26972c1f46dcSZachary Turner 
26982c1f46dcSZachary Turner         if (was_imported == true && can_reload == false)
26992c1f46dcSZachary Turner         {
27002c1f46dcSZachary Turner             error.SetErrorString("module already imported");
27012c1f46dcSZachary Turner             return false;
27022c1f46dcSZachary Turner         }
27032c1f46dcSZachary Turner 
27042c1f46dcSZachary Turner         // now actually do the import
27052c1f46dcSZachary Turner         command_stream.Clear();
27062c1f46dcSZachary Turner 
27072c1f46dcSZachary Turner         if (was_imported)
27082c1f46dcSZachary Turner         {
27092c1f46dcSZachary Turner             if (!was_imported_locally)
27102c1f46dcSZachary Turner                 command_stream.Printf("import %s ; reload(%s)",basename.c_str(),basename.c_str());
27112c1f46dcSZachary Turner             else
27122c1f46dcSZachary Turner                 command_stream.Printf("reload(%s)",basename.c_str());
27132c1f46dcSZachary Turner         }
27142c1f46dcSZachary Turner         else
27152c1f46dcSZachary Turner             command_stream.Printf("import %s",basename.c_str());
27162c1f46dcSZachary Turner 
27172c1f46dcSZachary Turner         error = ExecuteMultipleLines(command_stream.GetData(), ScriptInterpreter::ExecuteScriptOptions().SetEnableIO(false).SetSetLLDBGlobals(false));
27182c1f46dcSZachary Turner         if (error.Fail())
27192c1f46dcSZachary Turner             return false;
27202c1f46dcSZachary Turner 
27212c1f46dcSZachary Turner         // if we are here, everything worked
27222c1f46dcSZachary Turner         // call __lldb_init_module(debugger,dict)
27232c1f46dcSZachary Turner         if (!g_swig_call_module_init (basename.c_str(),
27242c1f46dcSZachary Turner                                       m_dictionary_name.c_str(),
27252c1f46dcSZachary Turner                                       debugger_sp))
27262c1f46dcSZachary Turner         {
27272c1f46dcSZachary Turner             error.SetErrorString("calling __lldb_init_module failed");
27282c1f46dcSZachary Turner             return false;
27292c1f46dcSZachary Turner         }
27302c1f46dcSZachary Turner 
27312c1f46dcSZachary Turner         if (module_sp)
27322c1f46dcSZachary Turner         {
27332c1f46dcSZachary Turner             // everything went just great, now set the module object
27342c1f46dcSZachary Turner             command_stream.Clear();
27352c1f46dcSZachary Turner             command_stream.Printf("%s",basename.c_str());
27362c1f46dcSZachary Turner             void* module_pyobj = nullptr;
27372c1f46dcSZachary Turner             if (ExecuteOneLineWithReturn(command_stream.GetData(),ScriptInterpreter::eScriptReturnTypeOpaqueObject,&module_pyobj) && module_pyobj)
27382c1f46dcSZachary Turner                 module_sp->reset(new StructuredPythonObject(module_pyobj));
27392c1f46dcSZachary Turner         }
27402c1f46dcSZachary Turner 
27412c1f46dcSZachary Turner         return true;
27422c1f46dcSZachary Turner     }
27432c1f46dcSZachary Turner }
27442c1f46dcSZachary Turner 
27452c1f46dcSZachary Turner bool
27462c1f46dcSZachary Turner ScriptInterpreterPython::IsReservedWord (const char* word)
27472c1f46dcSZachary Turner {
27482c1f46dcSZachary Turner     if (!word || !word[0])
27492c1f46dcSZachary Turner         return false;
27502c1f46dcSZachary Turner 
27512c1f46dcSZachary Turner     llvm::StringRef word_sr(word);
27522c1f46dcSZachary Turner 
27532c1f46dcSZachary Turner     // filter out a few characters that would just confuse us
27542c1f46dcSZachary Turner     // and that are clearly not keyword material anyway
27552c1f46dcSZachary Turner     if (word_sr.find_first_of("'\"") != llvm::StringRef::npos)
27562c1f46dcSZachary Turner         return false;
27572c1f46dcSZachary Turner 
27582c1f46dcSZachary Turner     StreamString command_stream;
27592c1f46dcSZachary Turner     command_stream.Printf("keyword.iskeyword('%s')", word);
27602c1f46dcSZachary Turner     bool result;
27612c1f46dcSZachary Turner     ExecuteScriptOptions options;
27622c1f46dcSZachary Turner     options.SetEnableIO(false);
27632c1f46dcSZachary Turner     options.SetMaskoutErrors(true);
27642c1f46dcSZachary Turner     options.SetSetLLDBGlobals(false);
27652c1f46dcSZachary Turner     if (ExecuteOneLineWithReturn(command_stream.GetData(), ScriptInterpreter::eScriptReturnTypeBool, &result, options))
27662c1f46dcSZachary Turner         return result;
27672c1f46dcSZachary Turner     return false;
27682c1f46dcSZachary Turner }
27692c1f46dcSZachary Turner 
27702c1f46dcSZachary Turner ScriptInterpreterPython::SynchronicityHandler::SynchronicityHandler (lldb::DebuggerSP debugger_sp,
27712c1f46dcSZachary Turner                                                                      ScriptedCommandSynchronicity synchro) :
27722c1f46dcSZachary Turner     m_debugger_sp(debugger_sp),
27732c1f46dcSZachary Turner     m_synch_wanted(synchro),
27742c1f46dcSZachary Turner     m_old_asynch(debugger_sp->GetAsyncExecution())
27752c1f46dcSZachary Turner {
27762c1f46dcSZachary Turner     if (m_synch_wanted == eScriptedCommandSynchronicitySynchronous)
27772c1f46dcSZachary Turner         m_debugger_sp->SetAsyncExecution(false);
27782c1f46dcSZachary Turner     else if (m_synch_wanted == eScriptedCommandSynchronicityAsynchronous)
27792c1f46dcSZachary Turner         m_debugger_sp->SetAsyncExecution(true);
27802c1f46dcSZachary Turner }
27812c1f46dcSZachary Turner 
27822c1f46dcSZachary Turner ScriptInterpreterPython::SynchronicityHandler::~SynchronicityHandler()
27832c1f46dcSZachary Turner {
27842c1f46dcSZachary Turner     if (m_synch_wanted != eScriptedCommandSynchronicityCurrentValue)
27852c1f46dcSZachary Turner         m_debugger_sp->SetAsyncExecution(m_old_asynch);
27862c1f46dcSZachary Turner }
27872c1f46dcSZachary Turner 
27882c1f46dcSZachary Turner bool
27892c1f46dcSZachary Turner ScriptInterpreterPython::RunScriptBasedCommand(const char* impl_function,
27902c1f46dcSZachary Turner                                                const char* args,
27912c1f46dcSZachary Turner                                                ScriptedCommandSynchronicity synchronicity,
27922c1f46dcSZachary Turner                                                lldb_private::CommandReturnObject& cmd_retobj,
27932c1f46dcSZachary Turner                                                Error& error,
27942c1f46dcSZachary Turner                                                const lldb_private::ExecutionContext& exe_ctx)
27952c1f46dcSZachary Turner {
27962c1f46dcSZachary Turner     if (!impl_function)
27972c1f46dcSZachary Turner     {
27982c1f46dcSZachary Turner         error.SetErrorString("no function to execute");
27992c1f46dcSZachary Turner         return false;
28002c1f46dcSZachary Turner     }
28012c1f46dcSZachary Turner 
28022c1f46dcSZachary Turner     if (!g_swig_call_command)
28032c1f46dcSZachary Turner     {
28042c1f46dcSZachary Turner         error.SetErrorString("no helper function to run scripted commands");
28052c1f46dcSZachary Turner         return false;
28062c1f46dcSZachary Turner     }
28072c1f46dcSZachary Turner 
28082c1f46dcSZachary Turner     lldb::DebuggerSP debugger_sp = m_interpreter.GetDebugger().shared_from_this();
28092c1f46dcSZachary Turner     lldb::ExecutionContextRefSP exe_ctx_ref_sp(new ExecutionContextRef(exe_ctx));
28102c1f46dcSZachary Turner 
28112c1f46dcSZachary Turner     if (!debugger_sp.get())
28122c1f46dcSZachary Turner     {
28132c1f46dcSZachary Turner         error.SetErrorString("invalid Debugger pointer");
28142c1f46dcSZachary Turner         return false;
28152c1f46dcSZachary Turner    }
28162c1f46dcSZachary Turner 
28172c1f46dcSZachary Turner     bool ret_val = false;
28182c1f46dcSZachary Turner 
28192c1f46dcSZachary Turner     std::string err_msg;
28202c1f46dcSZachary Turner 
28212c1f46dcSZachary Turner     {
28222c1f46dcSZachary Turner         Locker py_lock(this,
28232c1f46dcSZachary Turner                        Locker::AcquireLock | Locker::InitSession | (cmd_retobj.GetInteractive() ? 0 : Locker::NoSTDIN),
28242c1f46dcSZachary Turner                        Locker::FreeLock    | Locker::TearDownSession);
28252c1f46dcSZachary Turner 
28262c1f46dcSZachary Turner         SynchronicityHandler synch_handler(debugger_sp,
28272c1f46dcSZachary Turner                                            synchronicity);
28282c1f46dcSZachary Turner 
28292c1f46dcSZachary Turner         ret_val = g_swig_call_command       (impl_function,
28302c1f46dcSZachary Turner                                              m_dictionary_name.c_str(),
28312c1f46dcSZachary Turner                                              debugger_sp,
28322c1f46dcSZachary Turner                                              args,
28332c1f46dcSZachary Turner                                              cmd_retobj,
28342c1f46dcSZachary Turner                                              exe_ctx_ref_sp);
28352c1f46dcSZachary Turner     }
28362c1f46dcSZachary Turner 
28372c1f46dcSZachary Turner     if (!ret_val)
28382c1f46dcSZachary Turner         error.SetErrorString("unable to execute script function");
28392c1f46dcSZachary Turner     else
28402c1f46dcSZachary Turner         error.Clear();
28412c1f46dcSZachary Turner 
28422c1f46dcSZachary Turner     return ret_val;
28432c1f46dcSZachary Turner }
28442c1f46dcSZachary Turner 
28452c1f46dcSZachary Turner bool
28462c1f46dcSZachary Turner ScriptInterpreterPython::RunScriptBasedCommand (StructuredData::GenericSP impl_obj_sp,
28472c1f46dcSZachary Turner                                                 const char* args,
28482c1f46dcSZachary Turner                                                 ScriptedCommandSynchronicity synchronicity,
28492c1f46dcSZachary Turner                                                 lldb_private::CommandReturnObject& cmd_retobj,
28502c1f46dcSZachary Turner                                                 Error& error,
28512c1f46dcSZachary Turner                                                 const lldb_private::ExecutionContext& exe_ctx)
28522c1f46dcSZachary Turner {
28532c1f46dcSZachary Turner     if (!impl_obj_sp || !impl_obj_sp->IsValid())
28542c1f46dcSZachary Turner     {
28552c1f46dcSZachary Turner         error.SetErrorString("no function to execute");
28562c1f46dcSZachary Turner         return false;
28572c1f46dcSZachary Turner     }
28582c1f46dcSZachary Turner 
28592c1f46dcSZachary Turner     if (!g_swig_call_command_object)
28602c1f46dcSZachary Turner     {
28612c1f46dcSZachary Turner         error.SetErrorString("no helper function to run scripted commands");
28622c1f46dcSZachary Turner         return false;
28632c1f46dcSZachary Turner     }
28642c1f46dcSZachary Turner 
28652c1f46dcSZachary Turner     lldb::DebuggerSP debugger_sp = m_interpreter.GetDebugger().shared_from_this();
28662c1f46dcSZachary Turner     lldb::ExecutionContextRefSP exe_ctx_ref_sp(new ExecutionContextRef(exe_ctx));
28672c1f46dcSZachary Turner 
28682c1f46dcSZachary Turner     if (!debugger_sp.get())
28692c1f46dcSZachary Turner     {
28702c1f46dcSZachary Turner         error.SetErrorString("invalid Debugger pointer");
28712c1f46dcSZachary Turner         return false;
28722c1f46dcSZachary Turner     }
28732c1f46dcSZachary Turner 
28742c1f46dcSZachary Turner     bool ret_val = false;
28752c1f46dcSZachary Turner 
28762c1f46dcSZachary Turner     std::string err_msg;
28772c1f46dcSZachary Turner 
28782c1f46dcSZachary Turner     {
28792c1f46dcSZachary Turner         Locker py_lock(this,
28802c1f46dcSZachary Turner                        Locker::AcquireLock | Locker::InitSession | (cmd_retobj.GetInteractive() ? 0 : Locker::NoSTDIN),
28812c1f46dcSZachary Turner                        Locker::FreeLock    | Locker::TearDownSession);
28822c1f46dcSZachary Turner 
28832c1f46dcSZachary Turner         SynchronicityHandler synch_handler(debugger_sp,
28842c1f46dcSZachary Turner                                            synchronicity);
28852c1f46dcSZachary Turner 
28862c1f46dcSZachary Turner         ret_val = g_swig_call_command_object      (impl_obj_sp->GetValue(),
28872c1f46dcSZachary Turner                                                    debugger_sp,
28882c1f46dcSZachary Turner                                                    args,
28892c1f46dcSZachary Turner                                                    cmd_retobj,
28902c1f46dcSZachary Turner                                                    exe_ctx_ref_sp);
28912c1f46dcSZachary Turner     }
28922c1f46dcSZachary Turner 
28932c1f46dcSZachary Turner     if (!ret_val)
28942c1f46dcSZachary Turner         error.SetErrorString("unable to execute script function");
28952c1f46dcSZachary Turner     else
28962c1f46dcSZachary Turner         error.Clear();
28972c1f46dcSZachary Turner 
28982c1f46dcSZachary Turner     return ret_val;
28992c1f46dcSZachary Turner }
29002c1f46dcSZachary Turner 
29012c1f46dcSZachary Turner // in Python, a special attribute __doc__ contains the docstring
29022c1f46dcSZachary Turner // for an object (function, method, class, ...) if any is defined
29032c1f46dcSZachary Turner // Otherwise, the attribute's value is None
29042c1f46dcSZachary Turner bool
29052c1f46dcSZachary Turner ScriptInterpreterPython::GetDocumentationForItem(const char* item, std::string& dest)
29062c1f46dcSZachary Turner {
29072c1f46dcSZachary Turner 	dest.clear();
29082c1f46dcSZachary Turner 	if (!item || !*item)
29092c1f46dcSZachary Turner 		return false;
29102c1f46dcSZachary Turner     std::string command(item);
29112c1f46dcSZachary Turner     command += ".__doc__";
29122c1f46dcSZachary Turner 
29132c1f46dcSZachary Turner     char* result_ptr = nullptr; // Python is going to point this to valid data if ExecuteOneLineWithReturn returns successfully
29142c1f46dcSZachary Turner 
29152c1f46dcSZachary Turner     if (ExecuteOneLineWithReturn (command.c_str(),
29162c1f46dcSZachary Turner                                   ScriptInterpreter::eScriptReturnTypeCharStrOrNone,
29172c1f46dcSZachary Turner                                   &result_ptr,
29182c1f46dcSZachary Turner                                   ScriptInterpreter::ExecuteScriptOptions().SetEnableIO(false)))
29192c1f46dcSZachary Turner     {
29202c1f46dcSZachary Turner         if (result_ptr)
29212c1f46dcSZachary Turner             dest.assign(result_ptr);
29222c1f46dcSZachary Turner         return true;
29232c1f46dcSZachary Turner     }
29242c1f46dcSZachary Turner     else
29252c1f46dcSZachary Turner     {
29262c1f46dcSZachary Turner         StreamString str_stream;
29272c1f46dcSZachary Turner         str_stream.Printf("Function %s was not found. Containing module might be missing.",item);
29282c1f46dcSZachary Turner         dest.assign(str_stream.GetData());
29292c1f46dcSZachary Turner         return false;
29302c1f46dcSZachary Turner     }
29312c1f46dcSZachary Turner }
29322c1f46dcSZachary Turner 
29332c1f46dcSZachary Turner bool
29342c1f46dcSZachary Turner ScriptInterpreterPython::GetShortHelpForCommandObject (StructuredData::GenericSP cmd_obj_sp,
29352c1f46dcSZachary Turner                                                        std::string& dest)
29362c1f46dcSZachary Turner {
29372c1f46dcSZachary Turner     bool got_string = false;
29382c1f46dcSZachary Turner     dest.clear();
29392c1f46dcSZachary Turner 
29402c1f46dcSZachary Turner     Locker py_lock (this,
29412c1f46dcSZachary Turner                     Locker::AcquireLock | Locker::NoSTDIN,
29422c1f46dcSZachary Turner                     Locker::FreeLock);
29432c1f46dcSZachary Turner 
29442c1f46dcSZachary Turner     static char callee_name[] = "get_short_help";
29452c1f46dcSZachary Turner 
29462c1f46dcSZachary Turner     if (!cmd_obj_sp)
29472c1f46dcSZachary Turner         return false;
29482c1f46dcSZachary Turner 
2949f8b22f8fSZachary Turner     PythonObject implementor(PyRefType::Borrowed, (PyObject *)cmd_obj_sp->GetValue());
29502c1f46dcSZachary Turner 
2951f8b22f8fSZachary Turner     if (!implementor.IsAllocated())
29522c1f46dcSZachary Turner         return false;
29532c1f46dcSZachary Turner 
2954f8b22f8fSZachary Turner     PythonObject pmeth(PyRefType::Owned, PyObject_GetAttrString(implementor.get(), callee_name));
29552c1f46dcSZachary Turner 
29562c1f46dcSZachary Turner     if (PyErr_Occurred())
29572c1f46dcSZachary Turner         PyErr_Clear();
29582c1f46dcSZachary Turner 
2959f8b22f8fSZachary Turner     if (!pmeth.IsAllocated())
29602c1f46dcSZachary Turner         return false;
29612c1f46dcSZachary Turner 
2962f8b22f8fSZachary Turner     if (PyCallable_Check(pmeth.get()) == 0)
29632c1f46dcSZachary Turner     {
29642c1f46dcSZachary Turner         if (PyErr_Occurred())
29652c1f46dcSZachary Turner             PyErr_Clear();
29662c1f46dcSZachary Turner         return false;
29672c1f46dcSZachary Turner     }
29682c1f46dcSZachary Turner 
29692c1f46dcSZachary Turner     if (PyErr_Occurred())
29702c1f46dcSZachary Turner         PyErr_Clear();
29712c1f46dcSZachary Turner 
29722c1f46dcSZachary Turner     // right now we know this function exists and is callable..
2973f8b22f8fSZachary Turner     PythonObject py_return(PyRefType::Owned, PyObject_CallMethod(implementor.get(), callee_name, nullptr));
29742c1f46dcSZachary Turner 
29752c1f46dcSZachary Turner     // if it fails, print the error but otherwise go on
29762c1f46dcSZachary Turner     if (PyErr_Occurred())
29772c1f46dcSZachary Turner     {
29782c1f46dcSZachary Turner         PyErr_Print();
29792c1f46dcSZachary Turner         PyErr_Clear();
29802c1f46dcSZachary Turner     }
29812c1f46dcSZachary Turner 
2982f8b22f8fSZachary Turner     if (py_return.IsAllocated() && PythonString::Check(py_return.get()))
29832c1f46dcSZachary Turner     {
2984f8b22f8fSZachary Turner         PythonString py_string(PyRefType::Borrowed, py_return.get());
298522c8efcdSZachary Turner         llvm::StringRef return_data(py_string.GetString());
298622c8efcdSZachary Turner         dest.assign(return_data.data(), return_data.size());
29872c1f46dcSZachary Turner         got_string = true;
29882c1f46dcSZachary Turner     }
29892c1f46dcSZachary Turner     return got_string;
29902c1f46dcSZachary Turner }
29912c1f46dcSZachary Turner 
29922c1f46dcSZachary Turner uint32_t
29932c1f46dcSZachary Turner ScriptInterpreterPython::GetFlagsForCommandObject (StructuredData::GenericSP cmd_obj_sp)
29942c1f46dcSZachary Turner {
29952c1f46dcSZachary Turner     uint32_t result = 0;
29962c1f46dcSZachary Turner 
29972c1f46dcSZachary Turner     Locker py_lock (this,
29982c1f46dcSZachary Turner                     Locker::AcquireLock | Locker::NoSTDIN,
29992c1f46dcSZachary Turner                     Locker::FreeLock);
30002c1f46dcSZachary Turner 
30012c1f46dcSZachary Turner     static char callee_name[] = "get_flags";
30022c1f46dcSZachary Turner 
30032c1f46dcSZachary Turner     if (!cmd_obj_sp)
30042c1f46dcSZachary Turner         return result;
30052c1f46dcSZachary Turner 
3006f8b22f8fSZachary Turner     PythonObject implementor(PyRefType::Borrowed, (PyObject *)cmd_obj_sp->GetValue());
30072c1f46dcSZachary Turner 
3008f8b22f8fSZachary Turner     if (!implementor.IsAllocated())
30092c1f46dcSZachary Turner         return result;
30102c1f46dcSZachary Turner 
3011f8b22f8fSZachary Turner     PythonObject pmeth(PyRefType::Owned, PyObject_GetAttrString(implementor.get(), callee_name));
30122c1f46dcSZachary Turner 
30132c1f46dcSZachary Turner     if (PyErr_Occurred())
30142c1f46dcSZachary Turner         PyErr_Clear();
30152c1f46dcSZachary Turner 
3016f8b22f8fSZachary Turner     if (!pmeth.IsAllocated())
30172c1f46dcSZachary Turner         return result;
30182c1f46dcSZachary Turner 
3019f8b22f8fSZachary Turner     if (PyCallable_Check(pmeth.get()) == 0)
30202c1f46dcSZachary Turner     {
30212c1f46dcSZachary Turner         if (PyErr_Occurred())
30222c1f46dcSZachary Turner             PyErr_Clear();
30232c1f46dcSZachary Turner         return result;
30242c1f46dcSZachary Turner     }
30252c1f46dcSZachary Turner 
30262c1f46dcSZachary Turner     if (PyErr_Occurred())
30272c1f46dcSZachary Turner         PyErr_Clear();
30282c1f46dcSZachary Turner 
30292c1f46dcSZachary Turner     // right now we know this function exists and is callable..
3030f8b22f8fSZachary Turner     PythonObject py_return(PyRefType::Owned, PyObject_CallMethod(implementor.get(), callee_name, nullptr));
30312c1f46dcSZachary Turner 
30322c1f46dcSZachary Turner     // if it fails, print the error but otherwise go on
30332c1f46dcSZachary Turner     if (PyErr_Occurred())
30342c1f46dcSZachary Turner     {
30352c1f46dcSZachary Turner         PyErr_Print();
30362c1f46dcSZachary Turner         PyErr_Clear();
30372c1f46dcSZachary Turner     }
30382c1f46dcSZachary Turner 
3039f8b22f8fSZachary Turner     if (py_return.IsAllocated() && PythonInteger::Check(py_return.get()))
30402c1f46dcSZachary Turner     {
3041f8b22f8fSZachary Turner         PythonInteger int_value(PyRefType::Borrowed, py_return.get());
304222c8efcdSZachary Turner         result = int_value.GetInteger();
30432c1f46dcSZachary Turner     }
30442c1f46dcSZachary Turner 
30452c1f46dcSZachary Turner     return result;
30462c1f46dcSZachary Turner }
30472c1f46dcSZachary Turner 
30482c1f46dcSZachary Turner bool
30492c1f46dcSZachary Turner ScriptInterpreterPython::GetLongHelpForCommandObject (StructuredData::GenericSP cmd_obj_sp,
30502c1f46dcSZachary Turner                                                       std::string& dest)
30512c1f46dcSZachary Turner {
30522c1f46dcSZachary Turner     bool got_string = false;
30532c1f46dcSZachary Turner     dest.clear();
30542c1f46dcSZachary Turner 
30552c1f46dcSZachary Turner     Locker py_lock (this,
30562c1f46dcSZachary Turner                     Locker::AcquireLock | Locker::NoSTDIN,
30572c1f46dcSZachary Turner                     Locker::FreeLock);
30582c1f46dcSZachary Turner 
30592c1f46dcSZachary Turner     static char callee_name[] = "get_long_help";
30602c1f46dcSZachary Turner 
30612c1f46dcSZachary Turner     if (!cmd_obj_sp)
30622c1f46dcSZachary Turner         return false;
30632c1f46dcSZachary Turner 
3064f8b22f8fSZachary Turner     PythonObject implementor(PyRefType::Borrowed, (PyObject *)cmd_obj_sp->GetValue());
30652c1f46dcSZachary Turner 
3066f8b22f8fSZachary Turner     if (!implementor.IsAllocated())
30672c1f46dcSZachary Turner         return false;
30682c1f46dcSZachary Turner 
3069f8b22f8fSZachary Turner     PythonObject pmeth(PyRefType::Owned, PyObject_GetAttrString(implementor.get(), callee_name));
30702c1f46dcSZachary Turner 
30712c1f46dcSZachary Turner     if (PyErr_Occurred())
30722c1f46dcSZachary Turner         PyErr_Clear();
30732c1f46dcSZachary Turner 
3074f8b22f8fSZachary Turner     if (!pmeth.IsAllocated())
30752c1f46dcSZachary Turner         return false;
30762c1f46dcSZachary Turner 
3077f8b22f8fSZachary Turner     if (PyCallable_Check(pmeth.get()) == 0)
30782c1f46dcSZachary Turner     {
30792c1f46dcSZachary Turner         if (PyErr_Occurred())
30802c1f46dcSZachary Turner             PyErr_Clear();
30812c1f46dcSZachary Turner 
30822c1f46dcSZachary Turner         return false;
30832c1f46dcSZachary Turner     }
30842c1f46dcSZachary Turner 
30852c1f46dcSZachary Turner     if (PyErr_Occurred())
30862c1f46dcSZachary Turner         PyErr_Clear();
30872c1f46dcSZachary Turner 
30882c1f46dcSZachary Turner     // right now we know this function exists and is callable..
3089f8b22f8fSZachary Turner     PythonObject py_return(PyRefType::Owned, PyObject_CallMethod(implementor.get(), callee_name, nullptr));
30902c1f46dcSZachary Turner 
30912c1f46dcSZachary Turner     // if it fails, print the error but otherwise go on
30922c1f46dcSZachary Turner     if (PyErr_Occurred())
30932c1f46dcSZachary Turner     {
30942c1f46dcSZachary Turner         PyErr_Print();
30952c1f46dcSZachary Turner         PyErr_Clear();
30962c1f46dcSZachary Turner     }
30972c1f46dcSZachary Turner 
3098f8b22f8fSZachary Turner     if (py_return.IsAllocated() && PythonString::Check(py_return.get()))
30992c1f46dcSZachary Turner     {
3100f8b22f8fSZachary Turner         PythonString str(PyRefType::Borrowed, py_return.get());
310122c8efcdSZachary Turner         llvm::StringRef str_data(str.GetString());
310222c8efcdSZachary Turner         dest.assign(str_data.data(), str_data.size());
31032c1f46dcSZachary Turner         got_string = true;
31042c1f46dcSZachary Turner     }
31052c1f46dcSZachary Turner 
31062c1f46dcSZachary Turner     return got_string;
31072c1f46dcSZachary Turner }
31082c1f46dcSZachary Turner 
31092c1f46dcSZachary Turner std::unique_ptr<ScriptInterpreterLocker>
31102c1f46dcSZachary Turner ScriptInterpreterPython::AcquireInterpreterLock ()
31112c1f46dcSZachary Turner {
31122c1f46dcSZachary Turner     std::unique_ptr<ScriptInterpreterLocker> py_lock(new Locker(this,
31132c1f46dcSZachary Turner                                                                 Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN,
31142c1f46dcSZachary Turner                                                                 Locker::FreeLock | Locker::TearDownSession));
31152c1f46dcSZachary Turner     return py_lock;
31162c1f46dcSZachary Turner }
31172c1f46dcSZachary Turner 
31182c1f46dcSZachary Turner void
31192c1f46dcSZachary Turner ScriptInterpreterPython::InitializeInterpreter (SWIGInitCallback swig_init_callback,
31202c1f46dcSZachary Turner                                                 SWIGBreakpointCallbackFunction swig_breakpoint_callback,
31212c1f46dcSZachary Turner                                                 SWIGWatchpointCallbackFunction swig_watchpoint_callback,
31222c1f46dcSZachary Turner                                                 SWIGPythonTypeScriptCallbackFunction swig_typescript_callback,
31232c1f46dcSZachary Turner                                                 SWIGPythonCreateSyntheticProvider swig_synthetic_script,
31242c1f46dcSZachary Turner                                                 SWIGPythonCreateCommandObject swig_create_cmd,
31252c1f46dcSZachary Turner                                                 SWIGPythonCalculateNumChildren swig_calc_children,
31262c1f46dcSZachary Turner                                                 SWIGPythonGetChildAtIndex swig_get_child_index,
31272c1f46dcSZachary Turner                                                 SWIGPythonGetIndexOfChildWithName swig_get_index_child,
31282c1f46dcSZachary Turner                                                 SWIGPythonCastPyObjectToSBValue swig_cast_to_sbvalue ,
31292c1f46dcSZachary Turner                                                 SWIGPythonGetValueObjectSPFromSBValue swig_get_valobj_sp_from_sbvalue,
31302c1f46dcSZachary Turner                                                 SWIGPythonUpdateSynthProviderInstance swig_update_provider,
31312c1f46dcSZachary Turner                                                 SWIGPythonMightHaveChildrenSynthProviderInstance swig_mighthavechildren_provider,
31322c1f46dcSZachary Turner                                                 SWIGPythonGetValueSynthProviderInstance swig_getvalue_provider,
31332c1f46dcSZachary Turner                                                 SWIGPythonCallCommand swig_call_command,
31342c1f46dcSZachary Turner                                                 SWIGPythonCallCommandObject swig_call_command_object,
31352c1f46dcSZachary Turner                                                 SWIGPythonCallModuleInit swig_call_module_init,
31362c1f46dcSZachary Turner                                                 SWIGPythonCreateOSPlugin swig_create_os_plugin,
31372c1f46dcSZachary Turner                                                 SWIGPythonScriptKeyword_Process swig_run_script_keyword_process,
31382c1f46dcSZachary Turner                                                 SWIGPythonScriptKeyword_Thread swig_run_script_keyword_thread,
31392c1f46dcSZachary Turner                                                 SWIGPythonScriptKeyword_Target swig_run_script_keyword_target,
31402c1f46dcSZachary Turner                                                 SWIGPythonScriptKeyword_Frame swig_run_script_keyword_frame,
31412c1f46dcSZachary Turner                                                 SWIGPythonScriptKeyword_Value swig_run_script_keyword_value,
31422c1f46dcSZachary Turner                                                 SWIGPython_GetDynamicSetting swig_plugin_get,
31432c1f46dcSZachary Turner                                                 SWIGPythonCreateScriptedThreadPlan swig_thread_plan_script,
31442c1f46dcSZachary Turner                                                 SWIGPythonCallThreadPlan swig_call_thread_plan)
31452c1f46dcSZachary Turner {
31462c1f46dcSZachary Turner     g_swig_init_callback = swig_init_callback;
31472c1f46dcSZachary Turner     g_swig_breakpoint_callback = swig_breakpoint_callback;
31482c1f46dcSZachary Turner     g_swig_watchpoint_callback = swig_watchpoint_callback;
31492c1f46dcSZachary Turner     g_swig_typescript_callback = swig_typescript_callback;
31502c1f46dcSZachary Turner     g_swig_synthetic_script = swig_synthetic_script;
31512c1f46dcSZachary Turner     g_swig_create_cmd = swig_create_cmd;
31522c1f46dcSZachary Turner     g_swig_calc_children = swig_calc_children;
31532c1f46dcSZachary Turner     g_swig_get_child_index = swig_get_child_index;
31542c1f46dcSZachary Turner     g_swig_get_index_child = swig_get_index_child;
31552c1f46dcSZachary Turner     g_swig_cast_to_sbvalue = swig_cast_to_sbvalue;
31562c1f46dcSZachary Turner     g_swig_get_valobj_sp_from_sbvalue = swig_get_valobj_sp_from_sbvalue;
31572c1f46dcSZachary Turner     g_swig_update_provider = swig_update_provider;
31582c1f46dcSZachary Turner     g_swig_mighthavechildren_provider = swig_mighthavechildren_provider;
31592c1f46dcSZachary Turner     g_swig_getvalue_provider = swig_getvalue_provider;
31602c1f46dcSZachary Turner     g_swig_call_command = swig_call_command;
31612c1f46dcSZachary Turner     g_swig_call_command_object = swig_call_command_object;
31622c1f46dcSZachary Turner     g_swig_call_module_init = swig_call_module_init;
31632c1f46dcSZachary Turner     g_swig_create_os_plugin = swig_create_os_plugin;
31642c1f46dcSZachary Turner     g_swig_run_script_keyword_process = swig_run_script_keyword_process;
31652c1f46dcSZachary Turner     g_swig_run_script_keyword_thread = swig_run_script_keyword_thread;
31662c1f46dcSZachary Turner     g_swig_run_script_keyword_target = swig_run_script_keyword_target;
31672c1f46dcSZachary Turner     g_swig_run_script_keyword_frame = swig_run_script_keyword_frame;
31682c1f46dcSZachary Turner     g_swig_run_script_keyword_value = swig_run_script_keyword_value;
31692c1f46dcSZachary Turner     g_swig_plugin_get = swig_plugin_get;
31702c1f46dcSZachary Turner     g_swig_thread_plan_script = swig_thread_plan_script;
31712c1f46dcSZachary Turner     g_swig_call_thread_plan = swig_call_thread_plan;
31722c1f46dcSZachary Turner }
31732c1f46dcSZachary Turner 
31742c1f46dcSZachary Turner void
31752c1f46dcSZachary Turner ScriptInterpreterPython::InitializePrivate ()
31762c1f46dcSZachary Turner {
31772c1f46dcSZachary Turner     assert(!g_initialized && "ScriptInterpreterPython::InitializePrivate() called more than once!");
31782c1f46dcSZachary Turner     g_initialized = true;
31792c1f46dcSZachary Turner 
31802c1f46dcSZachary Turner     Timer scoped_timer (__PRETTY_FUNCTION__, __PRETTY_FUNCTION__);
31812c1f46dcSZachary Turner 
3182079fe48aSZachary Turner     // RAII-based initialization which correctly handles multiple-initialization, version-
3183079fe48aSZachary Turner     // specific differences among Python 2 and Python 3, and saving and restoring various
3184079fe48aSZachary Turner     // other pieces of state that can get mucked with during initialization.
3185079fe48aSZachary Turner     InitializePythonRAII initialize_guard;
31862c1f46dcSZachary Turner 
31872c1f46dcSZachary Turner     if (g_swig_init_callback)
31882c1f46dcSZachary Turner         g_swig_init_callback ();
31892c1f46dcSZachary Turner 
31902c1f46dcSZachary Turner     // Update the path python uses to search for modules to include the current directory.
31912c1f46dcSZachary Turner 
31922c1f46dcSZachary Turner     PyRun_SimpleString ("import sys");
31932c1f46dcSZachary Turner     AddToSysPath(AddLocation::End, ".");
31942c1f46dcSZachary Turner 
31952c1f46dcSZachary Turner     FileSpec file_spec;
31962c1f46dcSZachary Turner     // Don't denormalize paths when calling file_spec.GetPath().  On platforms that use
31972c1f46dcSZachary Turner     // a backslash as the path separator, this will result in executing python code containing
31982c1f46dcSZachary Turner     // paths with unescaped backslashes.  But Python also accepts forward slashes, so to make
31992c1f46dcSZachary Turner     // life easier we just use that.
32002c1f46dcSZachary Turner     if (HostInfo::GetLLDBPath(ePathTypePythonDir, file_spec))
32012c1f46dcSZachary Turner         AddToSysPath(AddLocation::Beginning, file_spec.GetPath(false));
32022c1f46dcSZachary Turner     if (HostInfo::GetLLDBPath(ePathTypeLLDBShlibDir, file_spec))
32032c1f46dcSZachary Turner         AddToSysPath(AddLocation::Beginning, file_spec.GetPath(false));
32042c1f46dcSZachary Turner 
32052c1f46dcSZachary Turner     PyRun_SimpleString ("sys.dont_write_bytecode = 1; import lldb.embedded_interpreter; from lldb.embedded_interpreter import run_python_interpreter; from lldb.embedded_interpreter import run_one_line");
32062c1f46dcSZachary Turner }
32072c1f46dcSZachary Turner 
32082c1f46dcSZachary Turner void
32092c1f46dcSZachary Turner ScriptInterpreterPython::AddToSysPath(AddLocation location, std::string path)
32102c1f46dcSZachary Turner {
32112c1f46dcSZachary Turner     std::string path_copy;
32122c1f46dcSZachary Turner 
32132c1f46dcSZachary Turner     std::string statement;
32142c1f46dcSZachary Turner     if (location == AddLocation::Beginning)
32152c1f46dcSZachary Turner     {
32162c1f46dcSZachary Turner         statement.assign("sys.path.insert(0,\"");
32172c1f46dcSZachary Turner         statement.append (path);
32182c1f46dcSZachary Turner         statement.append ("\")");
32192c1f46dcSZachary Turner     }
32202c1f46dcSZachary Turner     else
32212c1f46dcSZachary Turner     {
32222c1f46dcSZachary Turner         statement.assign("sys.path.append(\"");
32232c1f46dcSZachary Turner         statement.append(path);
32242c1f46dcSZachary Turner         statement.append("\")");
32252c1f46dcSZachary Turner     }
32262c1f46dcSZachary Turner     PyRun_SimpleString (statement.c_str());
32272c1f46dcSZachary Turner }
32282c1f46dcSZachary Turner 
32292c1f46dcSZachary Turner 
32302c1f46dcSZachary Turner //void
32312c1f46dcSZachary Turner //ScriptInterpreterPython::Terminate ()
32322c1f46dcSZachary Turner //{
32332c1f46dcSZachary Turner //    // We are intentionally NOT calling Py_Finalize here (this would be the logical place to call it).  Calling
32342c1f46dcSZachary Turner //    // Py_Finalize here causes test suite runs to seg fault:  The test suite runs in Python.  It registers
32352c1f46dcSZachary Turner //    // SBDebugger::Terminate to be called 'at_exit'.  When the test suite Python harness finishes up, it calls
32362c1f46dcSZachary Turner //    // Py_Finalize, which calls all the 'at_exit' registered functions.  SBDebugger::Terminate calls Debugger::Terminate,
32372c1f46dcSZachary Turner //    // which calls lldb::Terminate, which calls ScriptInterpreter::Terminate, which calls
32382c1f46dcSZachary Turner //    // ScriptInterpreterPython::Terminate.  So if we call Py_Finalize here, we end up with Py_Finalize being called from
32392c1f46dcSZachary Turner //    // within Py_Finalize, which results in a seg fault.
32402c1f46dcSZachary Turner //    //
32412c1f46dcSZachary Turner //    // Since this function only gets called when lldb is shutting down and going away anyway, the fact that we don't
32422c1f46dcSZachary Turner //    // actually call Py_Finalize should not cause any problems (everything should shut down/go away anyway when the
32432c1f46dcSZachary Turner //    // process exits).
32442c1f46dcSZachary Turner //    //
32452c1f46dcSZachary Turner ////    Py_Finalize ();
32462c1f46dcSZachary Turner //}
32472c1f46dcSZachary Turner 
32482c1f46dcSZachary Turner #endif // #ifdef LLDB_DISABLE_PYTHON
3249