12c1f46dcSZachary Turner //===-- ScriptInterpreterPython.cpp -----------------------------*- C++ -*-===// 22c1f46dcSZachary Turner // 32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information. 52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 62c1f46dcSZachary Turner // 72c1f46dcSZachary Turner //===----------------------------------------------------------------------===// 82c1f46dcSZachary Turner 9d68983e3SPavel Labath #ifdef LLDB_DISABLE_PYTHON 10d68983e3SPavel Labath 11d68983e3SPavel Labath // Python is disabled in this build 12d68983e3SPavel Labath 13d68983e3SPavel Labath #else 14d68983e3SPavel Labath 1541de9a97SKate Stone // LLDB Python header must be included first 162c1f46dcSZachary Turner #include "lldb-python.h" 1741de9a97SKate Stone 182c1f46dcSZachary Turner #include "PythonDataObjects.h" 193946247cSZachary Turner #include "PythonExceptionState.h" 20b9c1b51eSKate Stone #include "ScriptInterpreterPython.h" 212c1f46dcSZachary Turner 222c1f46dcSZachary Turner #include <stdio.h> 23b9c1b51eSKate Stone #include <stdlib.h> 242c1f46dcSZachary Turner 25796ac80bSJonas Devlieghere #include <memory> 262c1f46dcSZachary Turner #include <mutex> 272c1f46dcSZachary Turner #include <string> 282c1f46dcSZachary Turner 292c1f46dcSZachary Turner #include "lldb/API/SBValue.h" 3041ae8e74SKuba Mracek #include "lldb/API/SBFrame.h" 312c1f46dcSZachary Turner #include "lldb/Breakpoint/BreakpointLocation.h" 322c1f46dcSZachary Turner #include "lldb/Breakpoint/StoppointCallbackContext.h" 332c1f46dcSZachary Turner #include "lldb/Breakpoint/WatchpointOptions.h" 342c1f46dcSZachary Turner #include "lldb/Core/Communication.h" 352c1f46dcSZachary Turner #include "lldb/Core/Debugger.h" 362c1f46dcSZachary Turner #include "lldb/Core/PluginManager.h" 372c1f46dcSZachary Turner #include "lldb/Core/ValueObject.h" 382c1f46dcSZachary Turner #include "lldb/DataFormatters/TypeSummary.h" 392c1f46dcSZachary Turner #include "lldb/Host/ConnectionFileDescriptor.h" 404eff2d31SZachary Turner #include "lldb/Host/FileSystem.h" 412c1f46dcSZachary Turner #include "lldb/Host/HostInfo.h" 422c1f46dcSZachary Turner #include "lldb/Host/Pipe.h" 432c1f46dcSZachary Turner #include "lldb/Interpreter/CommandInterpreter.h" 442c1f46dcSZachary Turner #include "lldb/Interpreter/CommandReturnObject.h" 452c1f46dcSZachary Turner #include "lldb/Target/Thread.h" 462c1f46dcSZachary Turner #include "lldb/Target/ThreadPlan.h" 4738d0632eSPavel Labath #include "lldb/Utility/Timer.h" 482c1f46dcSZachary Turner 492c1f46dcSZachary Turner #if defined(_WIN32) 502c1f46dcSZachary Turner #include "lldb/Host/windows/ConnectionGenericFileWindows.h" 512c1f46dcSZachary Turner #endif 522c1f46dcSZachary Turner 5322c8efcdSZachary Turner #include "llvm/ADT/STLExtras.h" 54b9c1b51eSKate Stone #include "llvm/ADT/StringRef.h" 557d86ee5aSZachary Turner #include "llvm/Support/FileSystem.h" 562c1f46dcSZachary Turner 572c1f46dcSZachary Turner using namespace lldb; 582c1f46dcSZachary Turner using namespace lldb_private; 592c1f46dcSZachary Turner 60b01b1087SJonas Devlieghere #ifndef LLDB_DISABLE_PYTHON 61b01b1087SJonas Devlieghere 62b01b1087SJonas Devlieghere // Defined in the SWIG source file 63b01b1087SJonas Devlieghere #if PY_MAJOR_VERSION >= 3 64b01b1087SJonas Devlieghere extern "C" PyObject *PyInit__lldb(void); 65b01b1087SJonas Devlieghere 66b01b1087SJonas Devlieghere #define LLDBSwigPyInit PyInit__lldb 67b01b1087SJonas Devlieghere 68b01b1087SJonas Devlieghere #else 69b01b1087SJonas Devlieghere extern "C" void init_lldb(void); 70b01b1087SJonas Devlieghere 71b01b1087SJonas Devlieghere #define LLDBSwigPyInit init_lldb 72b01b1087SJonas Devlieghere #endif 73b01b1087SJonas Devlieghere 74*05495c5dSJonas Devlieghere // These prototypes are the Pythonic implementations of the required callbacks. 75*05495c5dSJonas Devlieghere // Although these are scripting-language specific, their definition depends on 76*05495c5dSJonas Devlieghere // the public API. 77b01b1087SJonas Devlieghere extern "C" bool LLDBSwigPythonBreakpointCallbackFunction( 78b01b1087SJonas Devlieghere const char *python_function_name, const char *session_dictionary_name, 79b01b1087SJonas Devlieghere const lldb::StackFrameSP &sb_frame, 80b01b1087SJonas Devlieghere const lldb::BreakpointLocationSP &sb_bp_loc); 81b01b1087SJonas Devlieghere 82b01b1087SJonas Devlieghere extern "C" bool LLDBSwigPythonWatchpointCallbackFunction( 83b01b1087SJonas Devlieghere const char *python_function_name, const char *session_dictionary_name, 84b01b1087SJonas Devlieghere const lldb::StackFrameSP &sb_frame, const lldb::WatchpointSP &sb_wp); 85b01b1087SJonas Devlieghere 86b01b1087SJonas Devlieghere extern "C" bool LLDBSwigPythonCallTypeScript( 87b01b1087SJonas Devlieghere const char *python_function_name, void *session_dictionary, 88b01b1087SJonas Devlieghere const lldb::ValueObjectSP &valobj_sp, void **pyfunct_wrapper, 89b01b1087SJonas Devlieghere const lldb::TypeSummaryOptionsSP &options_sp, std::string &retval); 90b01b1087SJonas Devlieghere 91b01b1087SJonas Devlieghere extern "C" void * 92b01b1087SJonas Devlieghere LLDBSwigPythonCreateSyntheticProvider(const char *python_class_name, 93b01b1087SJonas Devlieghere const char *session_dictionary_name, 94b01b1087SJonas Devlieghere const lldb::ValueObjectSP &valobj_sp); 95b01b1087SJonas Devlieghere 96b01b1087SJonas Devlieghere extern "C" void * 97b01b1087SJonas Devlieghere LLDBSwigPythonCreateCommandObject(const char *python_class_name, 98b01b1087SJonas Devlieghere const char *session_dictionary_name, 99b01b1087SJonas Devlieghere const lldb::DebuggerSP debugger_sp); 100b01b1087SJonas Devlieghere 101b01b1087SJonas Devlieghere extern "C" void *LLDBSwigPythonCreateScriptedThreadPlan( 102b01b1087SJonas Devlieghere const char *python_class_name, const char *session_dictionary_name, 103b01b1087SJonas Devlieghere const lldb::ThreadPlanSP &thread_plan_sp); 104b01b1087SJonas Devlieghere 105b01b1087SJonas Devlieghere extern "C" bool LLDBSWIGPythonCallThreadPlan(void *implementor, 106b01b1087SJonas Devlieghere const char *method_name, 107b01b1087SJonas Devlieghere Event *event_sp, bool &got_error); 108b01b1087SJonas Devlieghere 109b01b1087SJonas Devlieghere extern "C" void *LLDBSwigPythonCreateScriptedBreakpointResolver( 110b01b1087SJonas Devlieghere const char *python_class_name, const char *session_dictionary_name, 111b01b1087SJonas Devlieghere lldb_private::StructuredDataImpl *args, lldb::BreakpointSP &bkpt_sp); 112b01b1087SJonas Devlieghere 113b01b1087SJonas Devlieghere extern "C" unsigned int 114b01b1087SJonas Devlieghere LLDBSwigPythonCallBreakpointResolver(void *implementor, const char *method_name, 115b01b1087SJonas Devlieghere lldb_private::SymbolContext *sym_ctx); 116b01b1087SJonas Devlieghere 117b01b1087SJonas Devlieghere extern "C" size_t LLDBSwigPython_CalculateNumChildren(void *implementor, 118b01b1087SJonas Devlieghere uint32_t max); 119b01b1087SJonas Devlieghere 120b01b1087SJonas Devlieghere extern "C" void *LLDBSwigPython_GetChildAtIndex(void *implementor, 121b01b1087SJonas Devlieghere uint32_t idx); 122b01b1087SJonas Devlieghere 123b01b1087SJonas Devlieghere extern "C" int LLDBSwigPython_GetIndexOfChildWithName(void *implementor, 124b01b1087SJonas Devlieghere const char *child_name); 125b01b1087SJonas Devlieghere 126b01b1087SJonas Devlieghere extern "C" void *LLDBSWIGPython_CastPyObjectToSBValue(void *data); 127b01b1087SJonas Devlieghere 128b01b1087SJonas Devlieghere extern lldb::ValueObjectSP 129b01b1087SJonas Devlieghere LLDBSWIGPython_GetValueObjectSPFromSBValue(void *data); 130b01b1087SJonas Devlieghere 131b01b1087SJonas Devlieghere extern "C" bool LLDBSwigPython_UpdateSynthProviderInstance(void *implementor); 132b01b1087SJonas Devlieghere 133b01b1087SJonas Devlieghere extern "C" bool 134b01b1087SJonas Devlieghere LLDBSwigPython_MightHaveChildrenSynthProviderInstance(void *implementor); 135b01b1087SJonas Devlieghere 136b01b1087SJonas Devlieghere extern "C" void * 137b01b1087SJonas Devlieghere LLDBSwigPython_GetValueSynthProviderInstance(void *implementor); 138b01b1087SJonas Devlieghere 139b01b1087SJonas Devlieghere extern "C" bool 140b01b1087SJonas Devlieghere LLDBSwigPythonCallCommand(const char *python_function_name, 141b01b1087SJonas Devlieghere const char *session_dictionary_name, 142b01b1087SJonas Devlieghere lldb::DebuggerSP &debugger, const char *args, 143b01b1087SJonas Devlieghere lldb_private::CommandReturnObject &cmd_retobj, 144b01b1087SJonas Devlieghere lldb::ExecutionContextRefSP exe_ctx_ref_sp); 145b01b1087SJonas Devlieghere 146b01b1087SJonas Devlieghere extern "C" bool 147b01b1087SJonas Devlieghere LLDBSwigPythonCallCommandObject(void *implementor, lldb::DebuggerSP &debugger, 148b01b1087SJonas Devlieghere const char *args, 149b01b1087SJonas Devlieghere lldb_private::CommandReturnObject &cmd_retobj, 150b01b1087SJonas Devlieghere lldb::ExecutionContextRefSP exe_ctx_ref_sp); 151b01b1087SJonas Devlieghere 152b01b1087SJonas Devlieghere extern "C" bool 153b01b1087SJonas Devlieghere LLDBSwigPythonCallModuleInit(const char *python_module_name, 154b01b1087SJonas Devlieghere const char *session_dictionary_name, 155b01b1087SJonas Devlieghere lldb::DebuggerSP &debugger); 156b01b1087SJonas Devlieghere 157b01b1087SJonas Devlieghere extern "C" void * 158b01b1087SJonas Devlieghere LLDBSWIGPythonCreateOSPlugin(const char *python_class_name, 159b01b1087SJonas Devlieghere const char *session_dictionary_name, 160b01b1087SJonas Devlieghere const lldb::ProcessSP &process_sp); 161b01b1087SJonas Devlieghere 162b01b1087SJonas Devlieghere extern "C" void * 163b01b1087SJonas Devlieghere LLDBSWIGPython_CreateFrameRecognizer(const char *python_class_name, 164b01b1087SJonas Devlieghere const char *session_dictionary_name); 165b01b1087SJonas Devlieghere 166b01b1087SJonas Devlieghere extern "C" void * 167b01b1087SJonas Devlieghere LLDBSwigPython_GetRecognizedArguments(void *implementor, 168b01b1087SJonas Devlieghere const lldb::StackFrameSP &frame_sp); 169b01b1087SJonas Devlieghere 170b01b1087SJonas Devlieghere extern "C" bool LLDBSWIGPythonRunScriptKeywordProcess( 171b01b1087SJonas Devlieghere const char *python_function_name, const char *session_dictionary_name, 172b01b1087SJonas Devlieghere lldb::ProcessSP &process, std::string &output); 173b01b1087SJonas Devlieghere 174b01b1087SJonas Devlieghere extern "C" bool LLDBSWIGPythonRunScriptKeywordThread( 175b01b1087SJonas Devlieghere const char *python_function_name, const char *session_dictionary_name, 176b01b1087SJonas Devlieghere lldb::ThreadSP &thread, std::string &output); 177b01b1087SJonas Devlieghere 178b01b1087SJonas Devlieghere extern "C" bool LLDBSWIGPythonRunScriptKeywordTarget( 179b01b1087SJonas Devlieghere const char *python_function_name, const char *session_dictionary_name, 180b01b1087SJonas Devlieghere lldb::TargetSP &target, std::string &output); 181b01b1087SJonas Devlieghere 182b01b1087SJonas Devlieghere extern "C" bool LLDBSWIGPythonRunScriptKeywordFrame( 183b01b1087SJonas Devlieghere const char *python_function_name, const char *session_dictionary_name, 184b01b1087SJonas Devlieghere lldb::StackFrameSP &frame, std::string &output); 185b01b1087SJonas Devlieghere 186b01b1087SJonas Devlieghere extern "C" bool LLDBSWIGPythonRunScriptKeywordValue( 187b01b1087SJonas Devlieghere const char *python_function_name, const char *session_dictionary_name, 188b01b1087SJonas Devlieghere lldb::ValueObjectSP &value, std::string &output); 189b01b1087SJonas Devlieghere 190b01b1087SJonas Devlieghere extern "C" void * 191b01b1087SJonas Devlieghere LLDBSWIGPython_GetDynamicSetting(void *module, const char *setting, 192b01b1087SJonas Devlieghere const lldb::TargetSP &target_sp); 193b01b1087SJonas Devlieghere 194b01b1087SJonas Devlieghere #endif 195b01b1087SJonas Devlieghere 1962c1f46dcSZachary Turner static bool g_initialized = false; 1972c1f46dcSZachary Turner 198b9c1b51eSKate Stone namespace { 19922c8efcdSZachary Turner 20005097246SAdrian Prantl // Initializing Python is not a straightforward process. We cannot control 20105097246SAdrian Prantl // what external code may have done before getting to this point in LLDB, 20205097246SAdrian Prantl // including potentially having already initialized Python, so we need to do a 20305097246SAdrian Prantl // lot of work to ensure that the existing state of the system is maintained 20405097246SAdrian Prantl // across our initialization. We do this by using an RAII pattern where we 20505097246SAdrian Prantl // save off initial state at the beginning, and restore it at the end 206b9c1b51eSKate Stone struct InitializePythonRAII { 207079fe48aSZachary Turner public: 208b9c1b51eSKate Stone InitializePythonRAII() 209b9c1b51eSKate Stone : m_gil_state(PyGILState_UNLOCKED), m_was_already_initialized(false) { 210079fe48aSZachary Turner // Python will muck with STDIN terminal state, so save off any current TTY 211079fe48aSZachary Turner // settings so we can restore them. 212079fe48aSZachary Turner m_stdin_tty_state.Save(STDIN_FILENO, false); 213079fe48aSZachary Turner 214079fe48aSZachary Turner InitializePythonHome(); 215079fe48aSZachary Turner 21674587a0eSVadim Chugunov // Register _lldb as a built-in module. 217*05495c5dSJonas Devlieghere PyImport_AppendInittab("_lldb", LLDBSwigPyInit); 21874587a0eSVadim Chugunov 219079fe48aSZachary Turner // Python < 3.2 and Python >= 3.2 reversed the ordering requirements for 220079fe48aSZachary Turner // calling `Py_Initialize` and `PyEval_InitThreads`. < 3.2 requires that you 221079fe48aSZachary Turner // call `PyEval_InitThreads` first, and >= 3.2 requires that you call it last. 222079fe48aSZachary Turner #if (PY_MAJOR_VERSION == 3 && PY_MINOR_VERSION >= 2) || (PY_MAJOR_VERSION > 3) 223079fe48aSZachary Turner Py_InitializeEx(0); 224079fe48aSZachary Turner InitializeThreadsPrivate(); 225079fe48aSZachary Turner #else 226079fe48aSZachary Turner InitializeThreadsPrivate(); 227079fe48aSZachary Turner Py_InitializeEx(0); 228079fe48aSZachary Turner #endif 229079fe48aSZachary Turner } 230079fe48aSZachary Turner 231b9c1b51eSKate Stone ~InitializePythonRAII() { 232b9c1b51eSKate Stone if (m_was_already_initialized) { 2333b7e1981SPavel Labath Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT)); 2343b7e1981SPavel Labath LLDB_LOGV(log, "Releasing PyGILState. Returning to state = {0}locked", 2351b6700efSTatyana Krasnukha m_gil_state == PyGILState_UNLOCKED ? "un" : ""); 236079fe48aSZachary Turner PyGILState_Release(m_gil_state); 237b9c1b51eSKate Stone } else { 238079fe48aSZachary Turner // We initialized the threads in this function, just unlock the GIL. 239079fe48aSZachary Turner PyEval_SaveThread(); 240079fe48aSZachary Turner } 241079fe48aSZachary Turner 242079fe48aSZachary Turner m_stdin_tty_state.Restore(); 243079fe48aSZachary Turner } 244079fe48aSZachary Turner 245079fe48aSZachary Turner private: 246b9c1b51eSKate Stone void InitializePythonHome() { 24722c8efcdSZachary Turner #if defined(LLDB_PYTHON_HOME) 24822c8efcdSZachary Turner #if PY_MAJOR_VERSION >= 3 24922c8efcdSZachary Turner size_t size = 0; 250079fe48aSZachary Turner static wchar_t *g_python_home = Py_DecodeLocale(LLDB_PYTHON_HOME, &size); 25122c8efcdSZachary Turner #else 252bdc52ef3SAidan Dodds static char g_python_home[] = LLDB_PYTHON_HOME; 25322c8efcdSZachary Turner #endif 254079fe48aSZachary Turner Py_SetPythonHome(g_python_home); 255386f00dbSDavide Italiano #else 256386f00dbSDavide Italiano #if defined(__APPLE__) && PY_MAJOR_VERSION == 2 && PY_MINOR_VERSION == 7 257386f00dbSDavide Italiano // For Darwin, the only Python version supported is the one shipped in the OS 2584f9cb260SDavide Italiano // OS and linked with lldb. Other installation of Python may have higher 2594f9cb260SDavide Italiano // priorities in the path, overriding PYTHONHOME and causing 2604f9cb260SDavide Italiano // problems/incompatibilities. In order to avoid confusion, always hardcode 2614f9cb260SDavide Italiano // the PythonHome to be right, as it's not going to change. 2625c4fad0cSDavide Italiano static char path[] = "/System/Library/Frameworks/Python.framework/Versions/2.7"; 2634f9cb260SDavide Italiano Py_SetPythonHome(path); 264386f00dbSDavide Italiano #endif 265386f00dbSDavide Italiano #endif 26622c8efcdSZachary Turner } 26722c8efcdSZachary Turner 268b9c1b51eSKate Stone void InitializeThreadsPrivate() { 2691b6700efSTatyana Krasnukha // Since Python 3.7 `Py_Initialize` calls `PyEval_InitThreads` inside itself, 2701b6700efSTatyana Krasnukha // so there is no way to determine whether the embedded interpreter 2711b6700efSTatyana Krasnukha // was already initialized by some external code. `PyEval_ThreadsInitialized` 2721b6700efSTatyana Krasnukha // would always return `true` and `PyGILState_Ensure/Release` flow would be 2731b6700efSTatyana Krasnukha // executed instead of unlocking GIL with `PyEval_SaveThread`. When 2741b6700efSTatyana Krasnukha // an another thread calls `PyGILState_Ensure` it would get stuck in deadlock. 2751b6700efSTatyana Krasnukha #if (PY_MAJOR_VERSION == 3 && PY_MINOR_VERSION >= 7) || (PY_MAJOR_VERSION > 3) 2761b6700efSTatyana Krasnukha // The only case we should go further and acquire the GIL: it is unlocked. 2771b6700efSTatyana Krasnukha if (PyGILState_Check()) 2781b6700efSTatyana Krasnukha return; 2791b6700efSTatyana Krasnukha #endif 2801b6700efSTatyana Krasnukha 281b9c1b51eSKate Stone if (PyEval_ThreadsInitialized()) { 2823b7e1981SPavel Labath Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT)); 283079fe48aSZachary Turner 284079fe48aSZachary Turner m_was_already_initialized = true; 285079fe48aSZachary Turner m_gil_state = PyGILState_Ensure(); 2863b7e1981SPavel Labath LLDB_LOGV(log, "Ensured PyGILState. Previous state = {0}locked\n", 287079fe48aSZachary Turner m_gil_state == PyGILState_UNLOCKED ? "un" : ""); 288079fe48aSZachary Turner return; 289079fe48aSZachary Turner } 290079fe48aSZachary Turner 291079fe48aSZachary Turner // InitThreads acquires the GIL if it hasn't been called before. 292079fe48aSZachary Turner PyEval_InitThreads(); 293079fe48aSZachary Turner } 294079fe48aSZachary Turner 295079fe48aSZachary Turner TerminalState m_stdin_tty_state; 296079fe48aSZachary Turner PyGILState_STATE m_gil_state; 297079fe48aSZachary Turner bool m_was_already_initialized; 298079fe48aSZachary Turner }; 299079fe48aSZachary Turner } 300079fe48aSZachary Turner 3012c1f46dcSZachary Turner ScriptInterpreterPython::Locker::Locker(ScriptInterpreterPython *py_interpreter, 302b9c1b51eSKate Stone uint16_t on_entry, uint16_t on_leave, 303b9c1b51eSKate Stone FILE *in, FILE *out, FILE *err) 304b9c1b51eSKate Stone : ScriptInterpreterLocker(), 3052c1f46dcSZachary Turner m_teardown_session((on_leave & TearDownSession) == TearDownSession), 306b9c1b51eSKate Stone m_python_interpreter(py_interpreter) { 3072c1f46dcSZachary Turner DoAcquireLock(); 308b9c1b51eSKate Stone if ((on_entry & InitSession) == InitSession) { 309a6682a41SJonas Devlieghere if (!DoInitSession(on_entry, in, out, err)) { 3102c1f46dcSZachary Turner // Don't teardown the session if we didn't init it. 3112c1f46dcSZachary Turner m_teardown_session = false; 3122c1f46dcSZachary Turner } 3132c1f46dcSZachary Turner } 3142c1f46dcSZachary Turner } 3152c1f46dcSZachary Turner 316b9c1b51eSKate Stone bool ScriptInterpreterPython::Locker::DoAcquireLock() { 3173b7e1981SPavel Labath Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT)); 3182c1f46dcSZachary Turner m_GILState = PyGILState_Ensure(); 3193b7e1981SPavel Labath LLDB_LOGV(log, "Ensured PyGILState. Previous state = {0}locked", 320b9c1b51eSKate Stone m_GILState == PyGILState_UNLOCKED ? "un" : ""); 3212c1f46dcSZachary Turner 32205097246SAdrian Prantl // we need to save the thread state when we first start the command because 32305097246SAdrian Prantl // we might decide to interrupt it while some action is taking place outside 32405097246SAdrian Prantl // of Python (e.g. printing to screen, waiting for the network, ...) in that 32505097246SAdrian Prantl // case, _PyThreadState_Current will be NULL - and we would be unable to set 32605097246SAdrian Prantl // the asynchronous exception - not a desirable situation 32722c8efcdSZachary Turner m_python_interpreter->SetThreadState(PyThreadState_Get()); 3282c1f46dcSZachary Turner m_python_interpreter->IncrementLockCount(); 3292c1f46dcSZachary Turner return true; 3302c1f46dcSZachary Turner } 3312c1f46dcSZachary Turner 332b9c1b51eSKate Stone bool ScriptInterpreterPython::Locker::DoInitSession(uint16_t on_entry_flags, 333b9c1b51eSKate Stone FILE *in, FILE *out, 334b9c1b51eSKate Stone FILE *err) { 3352c1f46dcSZachary Turner if (!m_python_interpreter) 3362c1f46dcSZachary Turner return false; 3372c1f46dcSZachary Turner return m_python_interpreter->EnterSession(on_entry_flags, in, out, err); 3382c1f46dcSZachary Turner } 3392c1f46dcSZachary Turner 340b9c1b51eSKate Stone bool ScriptInterpreterPython::Locker::DoFreeLock() { 3413b7e1981SPavel Labath Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT)); 3423b7e1981SPavel Labath LLDB_LOGV(log, "Releasing PyGILState. Returning to state = {0}locked", 343b9c1b51eSKate Stone m_GILState == PyGILState_UNLOCKED ? "un" : ""); 3442c1f46dcSZachary Turner PyGILState_Release(m_GILState); 3452c1f46dcSZachary Turner m_python_interpreter->DecrementLockCount(); 3462c1f46dcSZachary Turner return true; 3472c1f46dcSZachary Turner } 3482c1f46dcSZachary Turner 349b9c1b51eSKate Stone bool ScriptInterpreterPython::Locker::DoTearDownSession() { 3502c1f46dcSZachary Turner if (!m_python_interpreter) 3512c1f46dcSZachary Turner return false; 3522c1f46dcSZachary Turner m_python_interpreter->LeaveSession(); 3532c1f46dcSZachary Turner return true; 3542c1f46dcSZachary Turner } 3552c1f46dcSZachary Turner 356b9c1b51eSKate Stone ScriptInterpreterPython::Locker::~Locker() { 3572c1f46dcSZachary Turner if (m_teardown_session) 3582c1f46dcSZachary Turner DoTearDownSession(); 3592c1f46dcSZachary Turner DoFreeLock(); 3602c1f46dcSZachary Turner } 3612c1f46dcSZachary Turner 362b9c1b51eSKate Stone ScriptInterpreterPython::ScriptInterpreterPython( 363b9c1b51eSKate Stone CommandInterpreter &interpreter) 364b9c1b51eSKate Stone : ScriptInterpreter(interpreter, eScriptLanguagePython), 365b9c1b51eSKate Stone IOHandlerDelegateMultiline("DONE"), m_saved_stdin(), m_saved_stdout(), 366b9c1b51eSKate Stone m_saved_stderr(), m_main_module(), m_lldb_module(), 367f8b22f8fSZachary Turner m_session_dict(PyInitialValue::Invalid), 368b9c1b51eSKate Stone m_sys_module_dict(PyInitialValue::Invalid), m_run_one_line_function(), 3692c1f46dcSZachary Turner m_run_one_line_str_global(), 370b9c1b51eSKate Stone m_dictionary_name( 371b9c1b51eSKate Stone interpreter.GetDebugger().GetInstanceName().AsCString()), 372b9c1b51eSKate Stone m_terminal_state(), m_active_io_handler(eIOHandlerNone), 373b9c1b51eSKate Stone m_session_is_active(false), m_pty_slave_is_open(false), 374b9c1b51eSKate Stone m_valid_session(true), m_lock_count(0), m_command_thread_state(nullptr) { 37515d1b4e2SEnrico Granata InitializePrivate(); 3762c1f46dcSZachary Turner 3772c1f46dcSZachary Turner m_dictionary_name.append("_dict"); 3782c1f46dcSZachary Turner StreamString run_string; 3792c1f46dcSZachary Turner run_string.Printf("%s = dict()", m_dictionary_name.c_str()); 3802c1f46dcSZachary Turner 381b9c1b51eSKate Stone Locker locker(this, ScriptInterpreterPython::Locker::AcquireLock, 3822c1f46dcSZachary Turner ScriptInterpreterPython::Locker::FreeAcquiredLock); 3832c1f46dcSZachary Turner PyRun_SimpleString(run_string.GetData()); 3842c1f46dcSZachary Turner 3852c1f46dcSZachary Turner run_string.Clear(); 386b9c1b51eSKate Stone run_string.Printf( 387b9c1b51eSKate Stone "run_one_line (%s, 'import copy, keyword, os, re, sys, uuid, lldb')", 388b9c1b51eSKate Stone m_dictionary_name.c_str()); 3892c1f46dcSZachary Turner PyRun_SimpleString(run_string.GetData()); 3902c1f46dcSZachary Turner 391b9c1b51eSKate Stone // Reloading modules requires a different syntax in Python 2 and Python 3. 39205097246SAdrian Prantl // This provides a consistent syntax no matter what version of Python. 3937d2d0984SZachary Turner run_string.Clear(); 394b9c1b51eSKate Stone run_string.Printf("run_one_line (%s, 'from six.moves import reload_module')", 395b9c1b51eSKate Stone m_dictionary_name.c_str()); 3967d2d0984SZachary Turner PyRun_SimpleString(run_string.GetData()); 3977d2d0984SZachary Turner 398b9c1b51eSKate Stone // WARNING: temporary code that loads Cocoa formatters - this should be done 39905097246SAdrian Prantl // on a per-platform basis rather than loading the whole set and letting the 40005097246SAdrian Prantl // individual formatter classes exploit APIs to check whether they can/cannot 40105097246SAdrian Prantl // do their task 4022c1f46dcSZachary Turner run_string.Clear(); 403b9c1b51eSKate Stone run_string.Printf( 404b9c1b51eSKate Stone "run_one_line (%s, 'import lldb.formatters, lldb.formatters.cpp, pydoc')", 405b9c1b51eSKate Stone m_dictionary_name.c_str()); 4062c1f46dcSZachary Turner PyRun_SimpleString(run_string.GetData()); 4072c1f46dcSZachary Turner run_string.Clear(); 4082c1f46dcSZachary Turner 409b9c1b51eSKate Stone run_string.Printf("run_one_line (%s, 'import lldb.embedded_interpreter; from " 410b9c1b51eSKate Stone "lldb.embedded_interpreter import run_python_interpreter; " 411b9c1b51eSKate Stone "from lldb.embedded_interpreter import run_one_line')", 412b9c1b51eSKate Stone m_dictionary_name.c_str()); 4132c1f46dcSZachary Turner PyRun_SimpleString(run_string.GetData()); 4142c1f46dcSZachary Turner run_string.Clear(); 4152c1f46dcSZachary Turner 416b9c1b51eSKate Stone run_string.Printf("run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64 417b9c1b51eSKate Stone "; pydoc.pager = pydoc.plainpager')", 418b9c1b51eSKate Stone m_dictionary_name.c_str(), 4192c1f46dcSZachary Turner interpreter.GetDebugger().GetID()); 4202c1f46dcSZachary Turner PyRun_SimpleString(run_string.GetData()); 4212c1f46dcSZachary Turner } 4222c1f46dcSZachary Turner 423b9c1b51eSKate Stone ScriptInterpreterPython::~ScriptInterpreterPython() { 42405097246SAdrian Prantl // the session dictionary may hold objects with complex state which means 42505097246SAdrian Prantl // that they may need to be torn down with some level of smarts and that, in 42605097246SAdrian Prantl // turn, requires a valid thread state force Python to procure itself such a 42705097246SAdrian Prantl // thread state, nuke the session dictionary and then release it for others 42805097246SAdrian Prantl // to use and proceed with the rest of the shutdown 4292c1f46dcSZachary Turner auto gil_state = PyGILState_Ensure(); 4302c1f46dcSZachary Turner m_session_dict.Reset(); 4312c1f46dcSZachary Turner PyGILState_Release(gil_state); 4322c1f46dcSZachary Turner } 4332c1f46dcSZachary Turner 434b9c1b51eSKate Stone void ScriptInterpreterPython::Initialize() { 435c5f28e2aSKamil Rytarowski static llvm::once_flag g_once_flag; 4362c1f46dcSZachary Turner 437c5f28e2aSKamil Rytarowski llvm::call_once(g_once_flag, []() { 4382c1f46dcSZachary Turner PluginManager::RegisterPlugin(GetPluginNameStatic(), 4392c1f46dcSZachary Turner GetPluginDescriptionStatic(), 440b9c1b51eSKate Stone lldb::eScriptLanguagePython, CreateInstance); 4412c1f46dcSZachary Turner }); 4422c1f46dcSZachary Turner } 4432c1f46dcSZachary Turner 444b9c1b51eSKate Stone void ScriptInterpreterPython::Terminate() {} 4452c1f46dcSZachary Turner 4462c1f46dcSZachary Turner lldb::ScriptInterpreterSP 447b9c1b51eSKate Stone ScriptInterpreterPython::CreateInstance(CommandInterpreter &interpreter) { 4482c1f46dcSZachary Turner return std::make_shared<ScriptInterpreterPython>(interpreter); 4492c1f46dcSZachary Turner } 4502c1f46dcSZachary Turner 451b9c1b51eSKate Stone lldb_private::ConstString ScriptInterpreterPython::GetPluginNameStatic() { 4522c1f46dcSZachary Turner static ConstString g_name("script-python"); 4532c1f46dcSZachary Turner return g_name; 4542c1f46dcSZachary Turner } 4552c1f46dcSZachary Turner 456b9c1b51eSKate Stone const char *ScriptInterpreterPython::GetPluginDescriptionStatic() { 4572c1f46dcSZachary Turner return "Embedded Python interpreter"; 4582c1f46dcSZachary Turner } 4592c1f46dcSZachary Turner 4602df331b0SPavel Labath void ScriptInterpreterPython::ComputePythonDirForApple( 4612df331b0SPavel Labath llvm::SmallVectorImpl<char> &path) { 4622df331b0SPavel Labath auto style = llvm::sys::path::Style::posix; 4632df331b0SPavel Labath 4642df331b0SPavel Labath llvm::StringRef path_ref(path.begin(), path.size()); 4652df331b0SPavel Labath auto rbegin = llvm::sys::path::rbegin(path_ref, style); 4662df331b0SPavel Labath auto rend = llvm::sys::path::rend(path_ref); 4672df331b0SPavel Labath auto framework = std::find(rbegin, rend, "LLDB.framework"); 4682df331b0SPavel Labath if (framework == rend) { 4692df331b0SPavel Labath ComputePythonDirForPosix(path); 4702df331b0SPavel Labath return; 4712df331b0SPavel Labath } 4722df331b0SPavel Labath path.resize(framework - rend); 4732df331b0SPavel Labath llvm::sys::path::append(path, style, "LLDB.framework", "Resources", "Python"); 4742df331b0SPavel Labath } 4752df331b0SPavel Labath 4762df331b0SPavel Labath void ScriptInterpreterPython::ComputePythonDirForPosix( 4772df331b0SPavel Labath llvm::SmallVectorImpl<char> &path) { 4782df331b0SPavel Labath auto style = llvm::sys::path::Style::posix; 4792df331b0SPavel Labath #if defined(LLDB_PYTHON_RELATIVE_LIBDIR) 4802df331b0SPavel Labath // Build the path by backing out of the lib dir, then building with whatever 4812df331b0SPavel Labath // the real python interpreter uses. (e.g. lib for most, lib64 on RHEL 4822df331b0SPavel Labath // x86_64). 4832df331b0SPavel Labath llvm::sys::path::remove_filename(path, style); 4842df331b0SPavel Labath llvm::sys::path::append(path, style, LLDB_PYTHON_RELATIVE_LIBDIR); 4852df331b0SPavel Labath #else 4862df331b0SPavel Labath llvm::sys::path::append(path, style, 4872df331b0SPavel Labath "python" + llvm::Twine(PY_MAJOR_VERSION) + "." + 4882df331b0SPavel Labath llvm::Twine(PY_MINOR_VERSION), 4892df331b0SPavel Labath "site-packages"); 4902df331b0SPavel Labath #endif 4912df331b0SPavel Labath } 4922df331b0SPavel Labath 4932df331b0SPavel Labath void ScriptInterpreterPython::ComputePythonDirForWindows( 4942df331b0SPavel Labath llvm::SmallVectorImpl<char> &path) { 4952df331b0SPavel Labath auto style = llvm::sys::path::Style::windows; 4962df331b0SPavel Labath llvm::sys::path::remove_filename(path, style); 4972df331b0SPavel Labath llvm::sys::path::append(path, style, "lib", "site-packages"); 4982df331b0SPavel Labath 4992df331b0SPavel Labath // This will be injected directly through FileSpec.GetDirectory().SetString(), 5002df331b0SPavel Labath // so we need to normalize manually. 5012df331b0SPavel Labath std::replace(path.begin(), path.end(), '\\', '/'); 5022df331b0SPavel Labath } 5032df331b0SPavel Labath 5042df331b0SPavel Labath FileSpec ScriptInterpreterPython::GetPythonDir() { 5052df331b0SPavel Labath static FileSpec g_spec = []() { 5062df331b0SPavel Labath FileSpec spec = HostInfo::GetShlibDir(); 5072df331b0SPavel Labath if (!spec) 5082df331b0SPavel Labath return FileSpec(); 5092df331b0SPavel Labath llvm::SmallString<64> path; 5102df331b0SPavel Labath spec.GetPath(path); 5112df331b0SPavel Labath 5122df331b0SPavel Labath #if defined(__APPLE__) 5132df331b0SPavel Labath ComputePythonDirForApple(path); 5142df331b0SPavel Labath #elif defined(_WIN32) 5152df331b0SPavel Labath ComputePythonDirForWindows(path); 5162df331b0SPavel Labath #else 5172df331b0SPavel Labath ComputePythonDirForPosix(path); 5182df331b0SPavel Labath #endif 5192df331b0SPavel Labath spec.GetDirectory().SetString(path); 5202df331b0SPavel Labath return spec; 5212df331b0SPavel Labath }(); 5222df331b0SPavel Labath return g_spec; 5232df331b0SPavel Labath } 5242df331b0SPavel Labath 525b9c1b51eSKate Stone lldb_private::ConstString ScriptInterpreterPython::GetPluginName() { 5262c1f46dcSZachary Turner return GetPluginNameStatic(); 5272c1f46dcSZachary Turner } 5282c1f46dcSZachary Turner 529b9c1b51eSKate Stone uint32_t ScriptInterpreterPython::GetPluginVersion() { return 1; } 5302c1f46dcSZachary Turner 5310affb582SDave Lee void ScriptInterpreterPython::IOHandlerActivated(IOHandler &io_handler, bool interactive) { 5322c1f46dcSZachary Turner const char *instructions = nullptr; 5332c1f46dcSZachary Turner 534b9c1b51eSKate Stone switch (m_active_io_handler) { 5352c1f46dcSZachary Turner case eIOHandlerNone: 5362c1f46dcSZachary Turner break; 5372c1f46dcSZachary Turner case eIOHandlerBreakpoint: 5382c1f46dcSZachary Turner instructions = R"(Enter your Python command(s). Type 'DONE' to end. 5392c1f46dcSZachary Turner def function (frame, bp_loc, internal_dict): 5402c1f46dcSZachary Turner """frame: the lldb.SBFrame for the location at which you stopped 5412c1f46dcSZachary Turner bp_loc: an lldb.SBBreakpointLocation for the breakpoint location information 5422c1f46dcSZachary Turner internal_dict: an LLDB support object not to be used""" 5432c1f46dcSZachary Turner )"; 5442c1f46dcSZachary Turner break; 5452c1f46dcSZachary Turner case eIOHandlerWatchpoint: 5462c1f46dcSZachary Turner instructions = "Enter your Python command(s). Type 'DONE' to end.\n"; 5472c1f46dcSZachary Turner break; 5482c1f46dcSZachary Turner } 5492c1f46dcSZachary Turner 550b9c1b51eSKate Stone if (instructions) { 5512c1f46dcSZachary Turner StreamFileSP output_sp(io_handler.GetOutputStreamFile()); 5520affb582SDave Lee if (output_sp && interactive) { 5532c1f46dcSZachary Turner output_sp->PutCString(instructions); 5542c1f46dcSZachary Turner output_sp->Flush(); 5552c1f46dcSZachary Turner } 5562c1f46dcSZachary Turner } 5572c1f46dcSZachary Turner } 5582c1f46dcSZachary Turner 559b9c1b51eSKate Stone void ScriptInterpreterPython::IOHandlerInputComplete(IOHandler &io_handler, 560b9c1b51eSKate Stone std::string &data) { 5612c1f46dcSZachary Turner io_handler.SetIsDone(true); 5622c1f46dcSZachary Turner bool batch_mode = m_interpreter.GetBatchCommandMode(); 5632c1f46dcSZachary Turner 564b9c1b51eSKate Stone switch (m_active_io_handler) { 5652c1f46dcSZachary Turner case eIOHandlerNone: 5662c1f46dcSZachary Turner break; 567b9c1b51eSKate Stone case eIOHandlerBreakpoint: { 568b9c1b51eSKate Stone std::vector<BreakpointOptions *> *bp_options_vec = 569b9c1b51eSKate Stone (std::vector<BreakpointOptions *> *)io_handler.GetUserData(); 570b9c1b51eSKate Stone for (auto bp_options : *bp_options_vec) { 5712c1f46dcSZachary Turner if (!bp_options) 5722c1f46dcSZachary Turner continue; 5732c1f46dcSZachary Turner 574d5b44036SJonas Devlieghere auto data_up = llvm::make_unique<CommandDataPython>(); 575d5b44036SJonas Devlieghere if (!data_up) 5764e4fbe82SZachary Turner break; 577d5b44036SJonas Devlieghere data_up->user_source.SplitIntoLines(data); 5782c1f46dcSZachary Turner 579d5b44036SJonas Devlieghere if (GenerateBreakpointCommandCallbackData(data_up->user_source, 580d5b44036SJonas Devlieghere data_up->script_source) 581b9c1b51eSKate Stone .Success()) { 5824e4fbe82SZachary Turner auto baton_sp = std::make_shared<BreakpointOptions::CommandBaton>( 583d5b44036SJonas Devlieghere std::move(data_up)); 584b9c1b51eSKate Stone bp_options->SetCallback( 585b9c1b51eSKate Stone ScriptInterpreterPython::BreakpointCallbackFunction, baton_sp); 586b9c1b51eSKate Stone } else if (!batch_mode) { 5872c1f46dcSZachary Turner StreamFileSP error_sp = io_handler.GetErrorStreamFile(); 588b9c1b51eSKate Stone if (error_sp) { 5892c1f46dcSZachary Turner error_sp->Printf("Warning: No command attached to breakpoint.\n"); 5902c1f46dcSZachary Turner error_sp->Flush(); 5912c1f46dcSZachary Turner } 5922c1f46dcSZachary Turner } 5932c1f46dcSZachary Turner } 5942c1f46dcSZachary Turner m_active_io_handler = eIOHandlerNone; 595b9c1b51eSKate Stone } break; 596b9c1b51eSKate Stone case eIOHandlerWatchpoint: { 597b9c1b51eSKate Stone WatchpointOptions *wp_options = 598b9c1b51eSKate Stone (WatchpointOptions *)io_handler.GetUserData(); 599d5b44036SJonas Devlieghere auto data_up = llvm::make_unique<WatchpointOptions::CommandData>(); 600d5b44036SJonas Devlieghere data_up->user_source.SplitIntoLines(data); 6012c1f46dcSZachary Turner 602d5b44036SJonas Devlieghere if (GenerateWatchpointCommandCallbackData(data_up->user_source, 603d5b44036SJonas Devlieghere data_up->script_source)) { 6044e4fbe82SZachary Turner auto baton_sp = 605d5b44036SJonas Devlieghere std::make_shared<WatchpointOptions::CommandBaton>(std::move(data_up)); 606b9c1b51eSKate Stone wp_options->SetCallback( 607b9c1b51eSKate Stone ScriptInterpreterPython::WatchpointCallbackFunction, baton_sp); 608b9c1b51eSKate Stone } else if (!batch_mode) { 6092c1f46dcSZachary Turner StreamFileSP error_sp = io_handler.GetErrorStreamFile(); 610b9c1b51eSKate Stone if (error_sp) { 6112c1f46dcSZachary Turner error_sp->Printf("Warning: No command attached to breakpoint.\n"); 6122c1f46dcSZachary Turner error_sp->Flush(); 6132c1f46dcSZachary Turner } 6142c1f46dcSZachary Turner } 6152c1f46dcSZachary Turner m_active_io_handler = eIOHandlerNone; 616b9c1b51eSKate Stone } break; 6172c1f46dcSZachary Turner } 6182c1f46dcSZachary Turner } 6192c1f46dcSZachary Turner 620b9c1b51eSKate Stone void ScriptInterpreterPython::ResetOutputFileHandle(FILE *fh) {} 6212c1f46dcSZachary Turner 622b9c1b51eSKate Stone void ScriptInterpreterPython::SaveTerminalState(int fd) { 6232c1f46dcSZachary Turner // Python mucks with the terminal state of STDIN. If we can possibly avoid 6242c1f46dcSZachary Turner // this by setting the file handles up correctly prior to entering the 62505097246SAdrian Prantl // interpreter we should. For now we save and restore the terminal state on 62605097246SAdrian Prantl // the input file handle. 6272c1f46dcSZachary Turner m_terminal_state.Save(fd, false); 6282c1f46dcSZachary Turner } 6292c1f46dcSZachary Turner 630b9c1b51eSKate Stone void ScriptInterpreterPython::RestoreTerminalState() { 6312c1f46dcSZachary Turner // Python mucks with the terminal state of STDIN. If we can possibly avoid 6322c1f46dcSZachary Turner // this by setting the file handles up correctly prior to entering the 63305097246SAdrian Prantl // interpreter we should. For now we save and restore the terminal state on 63405097246SAdrian Prantl // the input file handle. 6352c1f46dcSZachary Turner m_terminal_state.Restore(); 6362c1f46dcSZachary Turner } 6372c1f46dcSZachary Turner 638b9c1b51eSKate Stone void ScriptInterpreterPython::LeaveSession() { 6392c1f46dcSZachary Turner Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT)); 6402c1f46dcSZachary Turner if (log) 6412c1f46dcSZachary Turner log->PutCString("ScriptInterpreterPython::LeaveSession()"); 6422c1f46dcSZachary Turner 64305097246SAdrian Prantl // checking that we have a valid thread state - since we use our own 64405097246SAdrian Prantl // threading and locking in some (rare) cases during cleanup Python may end 64505097246SAdrian Prantl // up believing we have no thread state and PyImport_AddModule will crash if 64605097246SAdrian Prantl // that is the case - since that seems to only happen when destroying the 64705097246SAdrian Prantl // SBDebugger, we can make do without clearing up stdout and stderr 6482c1f46dcSZachary Turner 6492c1f46dcSZachary Turner // rdar://problem/11292882 650b9c1b51eSKate Stone // When the current thread state is NULL, PyThreadState_Get() issues a fatal 651b9c1b51eSKate Stone // error. 652b9c1b51eSKate Stone if (PyThreadState_GetDict()) { 6532c1f46dcSZachary Turner PythonDictionary &sys_module_dict = GetSysModuleDictionary(); 654b9c1b51eSKate Stone if (sys_module_dict.IsValid()) { 655b9c1b51eSKate Stone if (m_saved_stdin.IsValid()) { 656f8b22f8fSZachary Turner sys_module_dict.SetItemForKey(PythonString("stdin"), m_saved_stdin); 6572c1f46dcSZachary Turner m_saved_stdin.Reset(); 6582c1f46dcSZachary Turner } 659b9c1b51eSKate Stone if (m_saved_stdout.IsValid()) { 660f8b22f8fSZachary Turner sys_module_dict.SetItemForKey(PythonString("stdout"), m_saved_stdout); 6612c1f46dcSZachary Turner m_saved_stdout.Reset(); 6622c1f46dcSZachary Turner } 663b9c1b51eSKate Stone if (m_saved_stderr.IsValid()) { 664f8b22f8fSZachary Turner sys_module_dict.SetItemForKey(PythonString("stderr"), m_saved_stderr); 6652c1f46dcSZachary Turner m_saved_stderr.Reset(); 6662c1f46dcSZachary Turner } 6672c1f46dcSZachary Turner } 6682c1f46dcSZachary Turner } 6692c1f46dcSZachary Turner 6702c1f46dcSZachary Turner m_session_is_active = false; 6712c1f46dcSZachary Turner } 6722c1f46dcSZachary Turner 673b9c1b51eSKate Stone bool ScriptInterpreterPython::SetStdHandle(File &file, const char *py_name, 67423502721SJason Molenda PythonFile &save_file, 675b9c1b51eSKate Stone const char *mode) { 676b9c1b51eSKate Stone if (file.IsValid()) { 677a31baf08SGreg Clayton // Flush the file before giving it to python to avoid interleaved output. 678a31baf08SGreg Clayton file.Flush(); 679a31baf08SGreg Clayton 680a31baf08SGreg Clayton PythonDictionary &sys_module_dict = GetSysModuleDictionary(); 681a31baf08SGreg Clayton 68223502721SJason Molenda save_file = sys_module_dict.GetItemForKey(PythonString(py_name)) 68323502721SJason Molenda .AsType<PythonFile>(); 684a31baf08SGreg Clayton 685a31baf08SGreg Clayton PythonFile new_file(file, mode); 686a31baf08SGreg Clayton sys_module_dict.SetItemForKey(PythonString(py_name), new_file); 687a31baf08SGreg Clayton return true; 688b9c1b51eSKate Stone } else 689a31baf08SGreg Clayton save_file.Reset(); 690a31baf08SGreg Clayton return false; 691a31baf08SGreg Clayton } 692a31baf08SGreg Clayton 693b9c1b51eSKate Stone bool ScriptInterpreterPython::EnterSession(uint16_t on_entry_flags, FILE *in, 694b9c1b51eSKate Stone FILE *out, FILE *err) { 695b9c1b51eSKate Stone // If we have already entered the session, without having officially 'left' 69605097246SAdrian Prantl // it, then there is no need to 'enter' it again. 6972c1f46dcSZachary Turner Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT)); 698b9c1b51eSKate Stone if (m_session_is_active) { 6992c1f46dcSZachary Turner if (log) 700b9c1b51eSKate Stone log->Printf( 701b9c1b51eSKate Stone "ScriptInterpreterPython::EnterSession(on_entry_flags=0x%" PRIx16 702b9c1b51eSKate Stone ") session is already active, returning without doing anything", 703b9c1b51eSKate Stone on_entry_flags); 7042c1f46dcSZachary Turner return false; 7052c1f46dcSZachary Turner } 7062c1f46dcSZachary Turner 7072c1f46dcSZachary Turner if (log) 708b9c1b51eSKate Stone log->Printf( 709b9c1b51eSKate Stone "ScriptInterpreterPython::EnterSession(on_entry_flags=0x%" PRIx16 ")", 710b9c1b51eSKate Stone on_entry_flags); 7112c1f46dcSZachary Turner 7122c1f46dcSZachary Turner m_session_is_active = true; 7132c1f46dcSZachary Turner 7142c1f46dcSZachary Turner StreamString run_string; 7152c1f46dcSZachary Turner 716b9c1b51eSKate Stone if (on_entry_flags & Locker::InitGlobals) { 717b9c1b51eSKate Stone run_string.Printf("run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64, 718b9c1b51eSKate Stone m_dictionary_name.c_str(), 719b9c1b51eSKate Stone GetCommandInterpreter().GetDebugger().GetID()); 720b9c1b51eSKate Stone run_string.Printf( 721b9c1b51eSKate Stone "; lldb.debugger = lldb.SBDebugger.FindDebuggerWithID (%" PRIu64 ")", 722b9c1b51eSKate Stone GetCommandInterpreter().GetDebugger().GetID()); 7232c1f46dcSZachary Turner run_string.PutCString("; lldb.target = lldb.debugger.GetSelectedTarget()"); 7242c1f46dcSZachary Turner run_string.PutCString("; lldb.process = lldb.target.GetProcess()"); 7252c1f46dcSZachary Turner run_string.PutCString("; lldb.thread = lldb.process.GetSelectedThread ()"); 7262c1f46dcSZachary Turner run_string.PutCString("; lldb.frame = lldb.thread.GetSelectedFrame ()"); 7272c1f46dcSZachary Turner run_string.PutCString("')"); 728b9c1b51eSKate Stone } else { 72905097246SAdrian Prantl // If we aren't initing the globals, we should still always set the 73005097246SAdrian Prantl // debugger (since that is always unique.) 731b9c1b51eSKate Stone run_string.Printf("run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64, 732b9c1b51eSKate Stone m_dictionary_name.c_str(), 733b9c1b51eSKate Stone GetCommandInterpreter().GetDebugger().GetID()); 734b9c1b51eSKate Stone run_string.Printf( 735b9c1b51eSKate Stone "; lldb.debugger = lldb.SBDebugger.FindDebuggerWithID (%" PRIu64 ")", 736b9c1b51eSKate Stone GetCommandInterpreter().GetDebugger().GetID()); 7372c1f46dcSZachary Turner run_string.PutCString("')"); 7382c1f46dcSZachary Turner } 7392c1f46dcSZachary Turner 7402c1f46dcSZachary Turner PyRun_SimpleString(run_string.GetData()); 7412c1f46dcSZachary Turner run_string.Clear(); 7422c1f46dcSZachary Turner 7432c1f46dcSZachary Turner PythonDictionary &sys_module_dict = GetSysModuleDictionary(); 744b9c1b51eSKate Stone if (sys_module_dict.IsValid()) { 7459c40264fSZachary Turner File in_file(in, false); 7469c40264fSZachary Turner File out_file(out, false); 7479c40264fSZachary Turner File err_file(err, false); 7489c40264fSZachary Turner 7492c1f46dcSZachary Turner lldb::StreamFileSP in_sp; 7502c1f46dcSZachary Turner lldb::StreamFileSP out_sp; 7512c1f46dcSZachary Turner lldb::StreamFileSP err_sp; 7529c40264fSZachary Turner if (!in_file.IsValid() || !out_file.IsValid() || !err_file.IsValid()) 753b9c1b51eSKate Stone m_interpreter.GetDebugger().AdoptTopIOHandlerFilesIfInvalid(in_sp, out_sp, 754b9c1b51eSKate Stone err_sp); 7552c1f46dcSZachary Turner 756b9c1b51eSKate Stone if (on_entry_flags & Locker::NoSTDIN) { 7572c1f46dcSZachary Turner m_saved_stdin.Reset(); 758b9c1b51eSKate Stone } else { 759b9c1b51eSKate Stone if (!SetStdHandle(in_file, "stdin", m_saved_stdin, "r")) { 760a31baf08SGreg Clayton if (in_sp) 761a31baf08SGreg Clayton SetStdHandle(in_sp->GetFile(), "stdin", m_saved_stdin, "r"); 7622c1f46dcSZachary Turner } 763a31baf08SGreg Clayton } 764a31baf08SGreg Clayton 765b9c1b51eSKate Stone if (!SetStdHandle(out_file, "stdout", m_saved_stdout, "w")) { 766a31baf08SGreg Clayton if (out_sp) 767a31baf08SGreg Clayton SetStdHandle(out_sp->GetFile(), "stdout", m_saved_stdout, "w"); 768a31baf08SGreg Clayton } 769a31baf08SGreg Clayton 770b9c1b51eSKate Stone if (!SetStdHandle(err_file, "stderr", m_saved_stderr, "w")) { 771a31baf08SGreg Clayton if (err_sp) 772a31baf08SGreg Clayton SetStdHandle(err_sp->GetFile(), "stderr", m_saved_stderr, "w"); 773a31baf08SGreg Clayton } 7742c1f46dcSZachary Turner } 7752c1f46dcSZachary Turner 7762c1f46dcSZachary Turner if (PyErr_Occurred()) 7772c1f46dcSZachary Turner PyErr_Clear(); 7782c1f46dcSZachary Turner 7792c1f46dcSZachary Turner return true; 7802c1f46dcSZachary Turner } 7812c1f46dcSZachary Turner 782b9c1b51eSKate Stone PythonObject &ScriptInterpreterPython::GetMainModule() { 783f8b22f8fSZachary Turner if (!m_main_module.IsValid()) 784f8b22f8fSZachary Turner m_main_module.Reset(PyRefType::Borrowed, PyImport_AddModule("__main__")); 7852c1f46dcSZachary Turner return m_main_module; 7862c1f46dcSZachary Turner } 7872c1f46dcSZachary Turner 788b9c1b51eSKate Stone PythonDictionary &ScriptInterpreterPython::GetSessionDictionary() { 789f8b22f8fSZachary Turner if (m_session_dict.IsValid()) 790f8b22f8fSZachary Turner return m_session_dict; 791f8b22f8fSZachary Turner 7922c1f46dcSZachary Turner PythonObject &main_module = GetMainModule(); 793f8b22f8fSZachary Turner if (!main_module.IsValid()) 794f8b22f8fSZachary Turner return m_session_dict; 795f8b22f8fSZachary Turner 796b9c1b51eSKate Stone PythonDictionary main_dict(PyRefType::Borrowed, 797b9c1b51eSKate Stone PyModule_GetDict(main_module.get())); 798f8b22f8fSZachary Turner if (!main_dict.IsValid()) 799f8b22f8fSZachary Turner return m_session_dict; 800f8b22f8fSZachary Turner 801f8b22f8fSZachary Turner PythonObject item = main_dict.GetItemForKey(PythonString(m_dictionary_name)); 802f8b22f8fSZachary Turner m_session_dict.Reset(PyRefType::Borrowed, item.get()); 8032c1f46dcSZachary Turner return m_session_dict; 8042c1f46dcSZachary Turner } 8052c1f46dcSZachary Turner 806b9c1b51eSKate Stone PythonDictionary &ScriptInterpreterPython::GetSysModuleDictionary() { 807f8b22f8fSZachary Turner if (m_sys_module_dict.IsValid()) 808f8b22f8fSZachary Turner return m_sys_module_dict; 809f8b22f8fSZachary Turner 810f8b22f8fSZachary Turner PythonObject sys_module(PyRefType::Borrowed, PyImport_AddModule("sys")); 811f8b22f8fSZachary Turner if (sys_module.IsValid()) 812b9c1b51eSKate Stone m_sys_module_dict.Reset(PyRefType::Borrowed, 813b9c1b51eSKate Stone PyModule_GetDict(sys_module.get())); 8142c1f46dcSZachary Turner return m_sys_module_dict; 8152c1f46dcSZachary Turner } 8162c1f46dcSZachary Turner 817b9c1b51eSKate Stone static std::string GenerateUniqueName(const char *base_name_wanted, 8182c1f46dcSZachary Turner uint32_t &functions_counter, 819b9c1b51eSKate Stone const void *name_token = nullptr) { 8202c1f46dcSZachary Turner StreamString sstr; 8212c1f46dcSZachary Turner 8222c1f46dcSZachary Turner if (!base_name_wanted) 8232c1f46dcSZachary Turner return std::string(); 8242c1f46dcSZachary Turner 8252c1f46dcSZachary Turner if (!name_token) 8262c1f46dcSZachary Turner sstr.Printf("%s_%d", base_name_wanted, functions_counter++); 8272c1f46dcSZachary Turner else 8282c1f46dcSZachary Turner sstr.Printf("%s_%p", base_name_wanted, name_token); 8292c1f46dcSZachary Turner 8302c1f46dcSZachary Turner return sstr.GetString(); 8312c1f46dcSZachary Turner } 8322c1f46dcSZachary Turner 833b9c1b51eSKate Stone bool ScriptInterpreterPython::GetEmbeddedInterpreterModuleObjects() { 834f8b22f8fSZachary Turner if (m_run_one_line_function.IsValid()) 835f8b22f8fSZachary Turner return true; 836f8b22f8fSZachary Turner 837b9c1b51eSKate Stone PythonObject module(PyRefType::Borrowed, 838b9c1b51eSKate Stone PyImport_AddModule("lldb.embedded_interpreter")); 839f8b22f8fSZachary Turner if (!module.IsValid()) 840f8b22f8fSZachary Turner return false; 841f8b22f8fSZachary Turner 842b9c1b51eSKate Stone PythonDictionary module_dict(PyRefType::Borrowed, 843b9c1b51eSKate Stone PyModule_GetDict(module.get())); 844f8b22f8fSZachary Turner if (!module_dict.IsValid()) 845f8b22f8fSZachary Turner return false; 846f8b22f8fSZachary Turner 847b9c1b51eSKate Stone m_run_one_line_function = 848b9c1b51eSKate Stone module_dict.GetItemForKey(PythonString("run_one_line")); 849b9c1b51eSKate Stone m_run_one_line_str_global = 850b9c1b51eSKate Stone module_dict.GetItemForKey(PythonString("g_run_one_line_str")); 851f8b22f8fSZachary Turner return m_run_one_line_function.IsValid(); 8522c1f46dcSZachary Turner } 8532c1f46dcSZachary Turner 854b9c1b51eSKate Stone static void ReadThreadBytesReceived(void *baton, const void *src, 855b9c1b51eSKate Stone size_t src_len) { 856b9c1b51eSKate Stone if (src && src_len) { 8572c1f46dcSZachary Turner Stream *strm = (Stream *)baton; 8582c1f46dcSZachary Turner strm->Write(src, src_len); 8592c1f46dcSZachary Turner strm->Flush(); 8602c1f46dcSZachary Turner } 8612c1f46dcSZachary Turner } 8622c1f46dcSZachary Turner 863b9c1b51eSKate Stone bool ScriptInterpreterPython::ExecuteOneLine( 8644d51a902SRaphael Isemann llvm::StringRef command, CommandReturnObject *result, 865b9c1b51eSKate Stone const ExecuteScriptOptions &options) { 866d6c062bcSRaphael Isemann std::string command_str = command.str(); 867d6c062bcSRaphael Isemann 8682c1f46dcSZachary Turner if (!m_valid_session) 8692c1f46dcSZachary Turner return false; 8702c1f46dcSZachary Turner 8714d51a902SRaphael Isemann if (!command.empty()) { 872b9c1b51eSKate Stone // We want to call run_one_line, passing in the dictionary and the command 87305097246SAdrian Prantl // string. We cannot do this through PyRun_SimpleString here because the 87405097246SAdrian Prantl // command string may contain escaped characters, and putting it inside 875b9c1b51eSKate Stone // another string to pass to PyRun_SimpleString messes up the escaping. So 87605097246SAdrian Prantl // we use the following more complicated method to pass the command string 87705097246SAdrian Prantl // directly down to Python. 8782c1f46dcSZachary Turner Debugger &debugger = m_interpreter.GetDebugger(); 8792c1f46dcSZachary Turner 8802c1f46dcSZachary Turner StreamFileSP input_file_sp; 8812c1f46dcSZachary Turner StreamFileSP output_file_sp; 8822c1f46dcSZachary Turner StreamFileSP error_file_sp; 883b9c1b51eSKate Stone Communication output_comm( 884b9c1b51eSKate Stone "lldb.ScriptInterpreterPython.ExecuteOneLine.comm"); 8852c1f46dcSZachary Turner bool join_read_thread = false; 886b9c1b51eSKate Stone if (options.GetEnableIO()) { 887b9c1b51eSKate Stone if (result) { 8882c1f46dcSZachary Turner input_file_sp = debugger.GetInputFile(); 889b9c1b51eSKate Stone // Set output to a temporary file so we can forward the results on to 890b9c1b51eSKate Stone // the result object 8912c1f46dcSZachary Turner 8922c1f46dcSZachary Turner Pipe pipe; 89397206d57SZachary Turner Status pipe_result = pipe.CreateNew(false); 894b9c1b51eSKate Stone if (pipe_result.Success()) { 8952c1f46dcSZachary Turner #if defined(_WIN32) 8962c1f46dcSZachary Turner lldb::file_t read_file = pipe.GetReadNativeHandle(); 8972c1f46dcSZachary Turner pipe.ReleaseReadFileDescriptor(); 898d5b44036SJonas Devlieghere std::unique_ptr<ConnectionGenericFile> conn_up( 899b9c1b51eSKate Stone new ConnectionGenericFile(read_file, true)); 9002c1f46dcSZachary Turner #else 901d5b44036SJonas Devlieghere std::unique_ptr<ConnectionFileDescriptor> conn_up( 902b9c1b51eSKate Stone new ConnectionFileDescriptor(pipe.ReleaseReadFileDescriptor(), 903b9c1b51eSKate Stone true)); 9042c1f46dcSZachary Turner #endif 905d5b44036SJonas Devlieghere if (conn_up->IsConnected()) { 906d5b44036SJonas Devlieghere output_comm.SetConnection(conn_up.release()); 907b9c1b51eSKate Stone output_comm.SetReadThreadBytesReceivedCallback( 908b9c1b51eSKate Stone ReadThreadBytesReceived, &result->GetOutputStream()); 9092c1f46dcSZachary Turner output_comm.StartReadThread(); 9102c1f46dcSZachary Turner join_read_thread = true; 911b9c1b51eSKate Stone FILE *outfile_handle = 912b9c1b51eSKate Stone fdopen(pipe.ReleaseWriteFileDescriptor(), "w"); 913796ac80bSJonas Devlieghere output_file_sp = std::make_shared<StreamFile>(outfile_handle, true); 9142c1f46dcSZachary Turner error_file_sp = output_file_sp; 9152c1f46dcSZachary Turner if (outfile_handle) 9162c1f46dcSZachary Turner ::setbuf(outfile_handle, nullptr); 9172c1f46dcSZachary Turner 918b9c1b51eSKate Stone result->SetImmediateOutputFile( 919b9c1b51eSKate Stone debugger.GetOutputFile()->GetFile().GetStream()); 920b9c1b51eSKate Stone result->SetImmediateErrorFile( 921b9c1b51eSKate Stone debugger.GetErrorFile()->GetFile().GetStream()); 9222c1f46dcSZachary Turner } 9232c1f46dcSZachary Turner } 9242c1f46dcSZachary Turner } 9252c1f46dcSZachary Turner if (!input_file_sp || !output_file_sp || !error_file_sp) 926b9c1b51eSKate Stone debugger.AdoptTopIOHandlerFilesIfInvalid(input_file_sp, output_file_sp, 927b9c1b51eSKate Stone error_file_sp); 928b9c1b51eSKate Stone } else { 929796ac80bSJonas Devlieghere input_file_sp = std::make_shared<StreamFile>(); 93050bc1ed2SJonas Devlieghere FileSystem::Instance().Open(input_file_sp->GetFile(), 93150bc1ed2SJonas Devlieghere FileSpec(FileSystem::DEV_NULL), 932b9c1b51eSKate Stone File::eOpenOptionRead); 93350bc1ed2SJonas Devlieghere 934796ac80bSJonas Devlieghere output_file_sp = std::make_shared<StreamFile>(); 93550bc1ed2SJonas Devlieghere FileSystem::Instance().Open(output_file_sp->GetFile(), 93650bc1ed2SJonas Devlieghere FileSpec(FileSystem::DEV_NULL), 937b9c1b51eSKate Stone File::eOpenOptionWrite); 93850bc1ed2SJonas Devlieghere 9392c1f46dcSZachary Turner error_file_sp = output_file_sp; 9402c1f46dcSZachary Turner } 9412c1f46dcSZachary Turner 9422c1f46dcSZachary Turner FILE *in_file = input_file_sp->GetFile().GetStream(); 9432c1f46dcSZachary Turner FILE *out_file = output_file_sp->GetFile().GetStream(); 9442c1f46dcSZachary Turner FILE *err_file = error_file_sp->GetFile().GetStream(); 94532064024SZachary Turner bool success = false; 94632064024SZachary Turner { 94705097246SAdrian Prantl // WARNING! It's imperative that this RAII scope be as tight as 94805097246SAdrian Prantl // possible. In particular, the scope must end *before* we try to join 94905097246SAdrian Prantl // the read thread. The reason for this is that a pre-requisite for 95005097246SAdrian Prantl // joining the read thread is that we close the write handle (to break 95105097246SAdrian Prantl // the pipe and cause it to wake up and exit). But acquiring the GIL as 95205097246SAdrian Prantl // below will redirect Python's stdio to use this same handle. If we 95305097246SAdrian Prantl // close the handle while Python is still using it, bad things will 95405097246SAdrian Prantl // happen. 955b9c1b51eSKate Stone Locker locker( 956b9c1b51eSKate Stone this, 9572c1f46dcSZachary Turner ScriptInterpreterPython::Locker::AcquireLock | 9582c1f46dcSZachary Turner ScriptInterpreterPython::Locker::InitSession | 959b9c1b51eSKate Stone (options.GetSetLLDBGlobals() 960b9c1b51eSKate Stone ? ScriptInterpreterPython::Locker::InitGlobals 961b9c1b51eSKate Stone : 0) | 9622c1f46dcSZachary Turner ((result && result->GetInteractive()) ? 0 : Locker::NoSTDIN), 9632c1f46dcSZachary Turner ScriptInterpreterPython::Locker::FreeAcquiredLock | 9642c1f46dcSZachary Turner ScriptInterpreterPython::Locker::TearDownSession, 965b9c1b51eSKate Stone in_file, out_file, err_file); 9662c1f46dcSZachary Turner 9672c1f46dcSZachary Turner // Find the correct script interpreter dictionary in the main module. 9682c1f46dcSZachary Turner PythonDictionary &session_dict = GetSessionDictionary(); 969b9c1b51eSKate Stone if (session_dict.IsValid()) { 970b9c1b51eSKate Stone if (GetEmbeddedInterpreterModuleObjects()) { 971b9c1b51eSKate Stone if (PyCallable_Check(m_run_one_line_function.get())) { 972b9c1b51eSKate Stone PythonObject pargs( 973b9c1b51eSKate Stone PyRefType::Owned, 974d6c062bcSRaphael Isemann Py_BuildValue("(Os)", session_dict.get(), command_str.c_str())); 975b9c1b51eSKate Stone if (pargs.IsValid()) { 976b9c1b51eSKate Stone PythonObject return_value( 977b9c1b51eSKate Stone PyRefType::Owned, 978b9c1b51eSKate Stone PyObject_CallObject(m_run_one_line_function.get(), 979b9c1b51eSKate Stone pargs.get())); 980f8b22f8fSZachary Turner if (return_value.IsValid()) 9812c1f46dcSZachary Turner success = true; 982b9c1b51eSKate Stone else if (options.GetMaskoutErrors() && PyErr_Occurred()) { 9832c1f46dcSZachary Turner PyErr_Print(); 9842c1f46dcSZachary Turner PyErr_Clear(); 9852c1f46dcSZachary Turner } 9862c1f46dcSZachary Turner } 9872c1f46dcSZachary Turner } 9882c1f46dcSZachary Turner } 9892c1f46dcSZachary Turner } 9902c1f46dcSZachary Turner 9912c1f46dcSZachary Turner // Flush our output and error file handles 9922c1f46dcSZachary Turner ::fflush(out_file); 9932c1f46dcSZachary Turner if (out_file != err_file) 9942c1f46dcSZachary Turner ::fflush(err_file); 99532064024SZachary Turner } 9962c1f46dcSZachary Turner 997b9c1b51eSKate Stone if (join_read_thread) { 99805097246SAdrian Prantl // Close the write end of the pipe since we are done with our one line 99905097246SAdrian Prantl // script. This should cause the read thread that output_comm is using to 100005097246SAdrian Prantl // exit 10012c1f46dcSZachary Turner output_file_sp->GetFile().Close(); 100205097246SAdrian Prantl // The close above should cause this thread to exit when it gets to the 100305097246SAdrian Prantl // end of file, so let it get all its data 10042c1f46dcSZachary Turner output_comm.JoinReadThread(); 10052c1f46dcSZachary Turner // Now we can close the read end of the pipe 10062c1f46dcSZachary Turner output_comm.Disconnect(); 10072c1f46dcSZachary Turner } 10082c1f46dcSZachary Turner 10092c1f46dcSZachary Turner if (success) 10102c1f46dcSZachary Turner return true; 10112c1f46dcSZachary Turner 10122c1f46dcSZachary Turner // The one-liner failed. Append the error message. 10134d51a902SRaphael Isemann if (result) { 1014b9c1b51eSKate Stone result->AppendErrorWithFormat( 10154d51a902SRaphael Isemann "python failed attempting to evaluate '%s'\n", command_str.c_str()); 10164d51a902SRaphael Isemann } 10172c1f46dcSZachary Turner return false; 10182c1f46dcSZachary Turner } 10192c1f46dcSZachary Turner 10202c1f46dcSZachary Turner if (result) 10212c1f46dcSZachary Turner result->AppendError("empty command passed to python\n"); 10222c1f46dcSZachary Turner return false; 10232c1f46dcSZachary Turner } 10242c1f46dcSZachary Turner 1025b9c1b51eSKate Stone class IOHandlerPythonInterpreter : public IOHandler { 10262c1f46dcSZachary Turner public: 10272c1f46dcSZachary Turner IOHandlerPythonInterpreter(Debugger &debugger, 1028b9c1b51eSKate Stone ScriptInterpreterPython *python) 1029b9c1b51eSKate Stone : IOHandler(debugger, IOHandler::Type::PythonInterpreter), 1030b9c1b51eSKate Stone m_python(python) {} 10312c1f46dcSZachary Turner 1032b9c1b51eSKate Stone ~IOHandlerPythonInterpreter() override {} 10332c1f46dcSZachary Turner 1034b9c1b51eSKate Stone ConstString GetControlSequence(char ch) override { 10352c1f46dcSZachary Turner if (ch == 'd') 10362c1f46dcSZachary Turner return ConstString("quit()\n"); 10372c1f46dcSZachary Turner return ConstString(); 10382c1f46dcSZachary Turner } 10392c1f46dcSZachary Turner 1040b9c1b51eSKate Stone void Run() override { 1041b9c1b51eSKate Stone if (m_python) { 10422c1f46dcSZachary Turner int stdin_fd = GetInputFD(); 1043b9c1b51eSKate Stone if (stdin_fd >= 0) { 10442c1f46dcSZachary Turner Terminal terminal(stdin_fd); 10452c1f46dcSZachary Turner TerminalState terminal_state; 10462c1f46dcSZachary Turner const bool is_a_tty = terminal.IsATerminal(); 10472c1f46dcSZachary Turner 1048b9c1b51eSKate Stone if (is_a_tty) { 10492c1f46dcSZachary Turner terminal_state.Save(stdin_fd, false); 10502c1f46dcSZachary Turner terminal.SetCanonical(false); 10512c1f46dcSZachary Turner terminal.SetEcho(true); 10522c1f46dcSZachary Turner } 10532c1f46dcSZachary Turner 1054b9c1b51eSKate Stone ScriptInterpreterPython::Locker locker( 1055b9c1b51eSKate Stone m_python, ScriptInterpreterPython::Locker::AcquireLock | 10562c1f46dcSZachary Turner ScriptInterpreterPython::Locker::InitSession | 10572c1f46dcSZachary Turner ScriptInterpreterPython::Locker::InitGlobals, 10582c1f46dcSZachary Turner ScriptInterpreterPython::Locker::FreeAcquiredLock | 10592c1f46dcSZachary Turner ScriptInterpreterPython::Locker::TearDownSession); 10602c1f46dcSZachary Turner 106105097246SAdrian Prantl // The following call drops into the embedded interpreter loop and 106205097246SAdrian Prantl // stays there until the user chooses to exit from the Python 106305097246SAdrian Prantl // interpreter. This embedded interpreter will, as any Python code that 106405097246SAdrian Prantl // performs I/O, unlock the GIL before a system call that can hang, and 106505097246SAdrian Prantl // lock it when the syscall has returned. 10662c1f46dcSZachary Turner 1067b9c1b51eSKate Stone // We need to surround the call to the embedded interpreter with calls 106805097246SAdrian Prantl // to PyGILState_Ensure and PyGILState_Release (using the Locker 106905097246SAdrian Prantl // above). This is because Python has a global lock which must be held 107005097246SAdrian Prantl // whenever we want to touch any Python objects. Otherwise, if the user 107105097246SAdrian Prantl // calls Python code, the interpreter state will be off, and things 107205097246SAdrian Prantl // could hang (it's happened before). 10732c1f46dcSZachary Turner 10742c1f46dcSZachary Turner StreamString run_string; 1075b9c1b51eSKate Stone run_string.Printf("run_python_interpreter (%s)", 1076b9c1b51eSKate Stone m_python->GetDictionaryName()); 10772c1f46dcSZachary Turner PyRun_SimpleString(run_string.GetData()); 10782c1f46dcSZachary Turner 10792c1f46dcSZachary Turner if (is_a_tty) 10802c1f46dcSZachary Turner terminal_state.Restore(); 10812c1f46dcSZachary Turner } 10822c1f46dcSZachary Turner } 10832c1f46dcSZachary Turner SetIsDone(true); 10842c1f46dcSZachary Turner } 10852c1f46dcSZachary Turner 1086b9c1b51eSKate Stone void Cancel() override {} 10872c1f46dcSZachary Turner 1088b9c1b51eSKate Stone bool Interrupt() override { return m_python->Interrupt(); } 10892c1f46dcSZachary Turner 1090b9c1b51eSKate Stone void GotEOF() override {} 10912c1f46dcSZachary Turner 10922c1f46dcSZachary Turner protected: 10932c1f46dcSZachary Turner ScriptInterpreterPython *m_python; 10942c1f46dcSZachary Turner }; 10952c1f46dcSZachary Turner 1096b9c1b51eSKate Stone void ScriptInterpreterPython::ExecuteInterpreterLoop() { 1097f9d16476SPavel Labath static Timer::Category func_cat(LLVM_PRETTY_FUNCTION); 1098f9d16476SPavel Labath Timer scoped_timer(func_cat, LLVM_PRETTY_FUNCTION); 10992c1f46dcSZachary Turner 11002c1f46dcSZachary Turner Debugger &debugger = GetCommandInterpreter().GetDebugger(); 11012c1f46dcSZachary Turner 1102b9c1b51eSKate Stone // At the moment, the only time the debugger does not have an input file 110305097246SAdrian Prantl // handle is when this is called directly from Python, in which case it is 110405097246SAdrian Prantl // both dangerous and unnecessary (not to mention confusing) to try to embed 110505097246SAdrian Prantl // a running interpreter loop inside the already running Python interpreter 110605097246SAdrian Prantl // loop, so we won't do it. 11072c1f46dcSZachary Turner 11082c1f46dcSZachary Turner if (!debugger.GetInputFile()->GetFile().IsValid()) 11092c1f46dcSZachary Turner return; 11102c1f46dcSZachary Turner 11112c1f46dcSZachary Turner IOHandlerSP io_handler_sp(new IOHandlerPythonInterpreter(debugger, this)); 1112b9c1b51eSKate Stone if (io_handler_sp) { 11132c1f46dcSZachary Turner debugger.PushIOHandler(io_handler_sp); 11142c1f46dcSZachary Turner } 11152c1f46dcSZachary Turner } 11162c1f46dcSZachary Turner 1117b9c1b51eSKate Stone bool ScriptInterpreterPython::Interrupt() { 11182c1f46dcSZachary Turner Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_SCRIPT)); 11192c1f46dcSZachary Turner 1120b9c1b51eSKate Stone if (IsExecutingPython()) { 1121b1cf558dSEnrico Granata PyThreadState *state = PyThreadState_GET(); 11222c1f46dcSZachary Turner if (!state) 11232c1f46dcSZachary Turner state = GetThreadState(); 1124b9c1b51eSKate Stone if (state) { 11252c1f46dcSZachary Turner long tid = state->thread_id; 112622c8efcdSZachary Turner PyThreadState_Swap(state); 11272c1f46dcSZachary Turner int num_threads = PyThreadState_SetAsyncExc(tid, PyExc_KeyboardInterrupt); 11282c1f46dcSZachary Turner if (log) 1129b9c1b51eSKate Stone log->Printf("ScriptInterpreterPython::Interrupt() sending " 1130b9c1b51eSKate Stone "PyExc_KeyboardInterrupt (tid = %li, num_threads = %i)...", 1131b9c1b51eSKate Stone tid, num_threads); 11322c1f46dcSZachary Turner return true; 11332c1f46dcSZachary Turner } 11342c1f46dcSZachary Turner } 11352c1f46dcSZachary Turner if (log) 1136b9c1b51eSKate Stone log->Printf("ScriptInterpreterPython::Interrupt() python code not running, " 1137b9c1b51eSKate Stone "can't interrupt"); 11382c1f46dcSZachary Turner return false; 11392c1f46dcSZachary Turner } 1140b9c1b51eSKate Stone bool ScriptInterpreterPython::ExecuteOneLineWithReturn( 11414d51a902SRaphael Isemann llvm::StringRef in_string, ScriptInterpreter::ScriptReturnType return_type, 1142b9c1b51eSKate Stone void *ret_value, const ExecuteScriptOptions &options) { 11432c1f46dcSZachary Turner 1144b9c1b51eSKate Stone Locker locker(this, ScriptInterpreterPython::Locker::AcquireLock | 1145b9c1b51eSKate Stone ScriptInterpreterPython::Locker::InitSession | 1146b9c1b51eSKate Stone (options.GetSetLLDBGlobals() 1147b9c1b51eSKate Stone ? ScriptInterpreterPython::Locker::InitGlobals 1148b9c1b51eSKate Stone : 0) | 1149b9c1b51eSKate Stone Locker::NoSTDIN, 1150b9c1b51eSKate Stone ScriptInterpreterPython::Locker::FreeAcquiredLock | 1151b9c1b51eSKate Stone ScriptInterpreterPython::Locker::TearDownSession); 11522c1f46dcSZachary Turner 1153f8b22f8fSZachary Turner PythonObject py_return; 11542c1f46dcSZachary Turner PythonObject &main_module = GetMainModule(); 1155b9c1b51eSKate Stone PythonDictionary globals(PyRefType::Borrowed, 1156b9c1b51eSKate Stone PyModule_GetDict(main_module.get())); 1157f8b22f8fSZachary Turner PythonObject py_error; 11582c1f46dcSZachary Turner bool ret_success = false; 11592c1f46dcSZachary Turner int success; 11602c1f46dcSZachary Turner 11612c1f46dcSZachary Turner PythonDictionary locals = GetSessionDictionary(); 11622c1f46dcSZachary Turner 1163b9c1b51eSKate Stone if (!locals.IsValid()) { 1164b9c1b51eSKate Stone locals.Reset( 1165b9c1b51eSKate Stone PyRefType::Owned, 1166b9c1b51eSKate Stone PyObject_GetAttrString(globals.get(), m_dictionary_name.c_str())); 11672c1f46dcSZachary Turner } 11682c1f46dcSZachary Turner 1169f8b22f8fSZachary Turner if (!locals.IsValid()) 11702c1f46dcSZachary Turner locals = globals; 11712c1f46dcSZachary Turner 1172f8b22f8fSZachary Turner py_error.Reset(PyRefType::Borrowed, PyErr_Occurred()); 1173f8b22f8fSZachary Turner if (py_error.IsValid()) 11742c1f46dcSZachary Turner PyErr_Clear(); 11752c1f46dcSZachary Turner 11764d51a902SRaphael Isemann std::string as_string = in_string.str(); 11772c1f46dcSZachary Turner { // scope for PythonInputReaderManager 11782c1f46dcSZachary Turner // PythonInputReaderManager py_input(options.GetEnableIO() ? this : NULL); 11794d51a902SRaphael Isemann py_return.Reset(PyRefType::Owned, 11804d51a902SRaphael Isemann PyRun_String(as_string.c_str(), Py_eval_input, 11814d51a902SRaphael Isemann globals.get(), locals.get())); 1182b9c1b51eSKate Stone if (!py_return.IsValid()) { 1183f8b22f8fSZachary Turner py_error.Reset(PyRefType::Borrowed, PyErr_Occurred()); 1184f8b22f8fSZachary Turner if (py_error.IsValid()) 11852c1f46dcSZachary Turner PyErr_Clear(); 11862c1f46dcSZachary Turner 1187b9c1b51eSKate Stone py_return.Reset(PyRefType::Owned, 11884d51a902SRaphael Isemann PyRun_String(as_string.c_str(), Py_single_input, 11894d51a902SRaphael Isemann globals.get(), locals.get())); 11902c1f46dcSZachary Turner } 11912c1f46dcSZachary Turner } 11922c1f46dcSZachary Turner 1193b9c1b51eSKate Stone if (py_return.IsValid()) { 1194b9c1b51eSKate Stone switch (return_type) { 11952c1f46dcSZachary Turner case eScriptReturnTypeCharPtr: // "char *" 11962c1f46dcSZachary Turner { 11972c1f46dcSZachary Turner const char format[3] = "s#"; 1198f8b22f8fSZachary Turner success = PyArg_Parse(py_return.get(), format, (char **)ret_value); 11992c1f46dcSZachary Turner break; 12002c1f46dcSZachary Turner } 1201b9c1b51eSKate Stone case eScriptReturnTypeCharStrOrNone: // char* or NULL if py_return == 1202b9c1b51eSKate Stone // Py_None 12032c1f46dcSZachary Turner { 12042c1f46dcSZachary Turner const char format[3] = "z"; 1205f8b22f8fSZachary Turner success = PyArg_Parse(py_return.get(), format, (char **)ret_value); 12062c1f46dcSZachary Turner break; 12072c1f46dcSZachary Turner } 1208b9c1b51eSKate Stone case eScriptReturnTypeBool: { 12092c1f46dcSZachary Turner const char format[2] = "b"; 1210f8b22f8fSZachary Turner success = PyArg_Parse(py_return.get(), format, (bool *)ret_value); 12112c1f46dcSZachary Turner break; 12122c1f46dcSZachary Turner } 1213b9c1b51eSKate Stone case eScriptReturnTypeShortInt: { 12142c1f46dcSZachary Turner const char format[2] = "h"; 1215f8b22f8fSZachary Turner success = PyArg_Parse(py_return.get(), format, (short *)ret_value); 12162c1f46dcSZachary Turner break; 12172c1f46dcSZachary Turner } 1218b9c1b51eSKate Stone case eScriptReturnTypeShortIntUnsigned: { 12192c1f46dcSZachary Turner const char format[2] = "H"; 1220b9c1b51eSKate Stone success = 1221b9c1b51eSKate Stone PyArg_Parse(py_return.get(), format, (unsigned short *)ret_value); 12222c1f46dcSZachary Turner break; 12232c1f46dcSZachary Turner } 1224b9c1b51eSKate Stone case eScriptReturnTypeInt: { 12252c1f46dcSZachary Turner const char format[2] = "i"; 1226f8b22f8fSZachary Turner success = PyArg_Parse(py_return.get(), format, (int *)ret_value); 12272c1f46dcSZachary Turner break; 12282c1f46dcSZachary Turner } 1229b9c1b51eSKate Stone case eScriptReturnTypeIntUnsigned: { 12302c1f46dcSZachary Turner const char format[2] = "I"; 12314d51a902SRaphael Isemann success = PyArg_Parse(py_return.get(), format, (unsigned int *)ret_value); 12322c1f46dcSZachary Turner break; 12332c1f46dcSZachary Turner } 1234b9c1b51eSKate Stone case eScriptReturnTypeLongInt: { 12352c1f46dcSZachary Turner const char format[2] = "l"; 1236f8b22f8fSZachary Turner success = PyArg_Parse(py_return.get(), format, (long *)ret_value); 12372c1f46dcSZachary Turner break; 12382c1f46dcSZachary Turner } 1239b9c1b51eSKate Stone case eScriptReturnTypeLongIntUnsigned: { 12402c1f46dcSZachary Turner const char format[2] = "k"; 1241b9c1b51eSKate Stone success = 1242b9c1b51eSKate Stone PyArg_Parse(py_return.get(), format, (unsigned long *)ret_value); 12432c1f46dcSZachary Turner break; 12442c1f46dcSZachary Turner } 1245b9c1b51eSKate Stone case eScriptReturnTypeLongLong: { 12462c1f46dcSZachary Turner const char format[2] = "L"; 1247f8b22f8fSZachary Turner success = PyArg_Parse(py_return.get(), format, (long long *)ret_value); 12482c1f46dcSZachary Turner break; 12492c1f46dcSZachary Turner } 1250b9c1b51eSKate Stone case eScriptReturnTypeLongLongUnsigned: { 12512c1f46dcSZachary Turner const char format[2] = "K"; 12524d51a902SRaphael Isemann success = 12534d51a902SRaphael Isemann PyArg_Parse(py_return.get(), format, (unsigned long long *)ret_value); 12542c1f46dcSZachary Turner break; 12552c1f46dcSZachary Turner } 1256b9c1b51eSKate Stone case eScriptReturnTypeFloat: { 12572c1f46dcSZachary Turner const char format[2] = "f"; 1258f8b22f8fSZachary Turner success = PyArg_Parse(py_return.get(), format, (float *)ret_value); 12592c1f46dcSZachary Turner break; 12602c1f46dcSZachary Turner } 1261b9c1b51eSKate Stone case eScriptReturnTypeDouble: { 12622c1f46dcSZachary Turner const char format[2] = "d"; 1263f8b22f8fSZachary Turner success = PyArg_Parse(py_return.get(), format, (double *)ret_value); 12642c1f46dcSZachary Turner break; 12652c1f46dcSZachary Turner } 1266b9c1b51eSKate Stone case eScriptReturnTypeChar: { 12672c1f46dcSZachary Turner const char format[2] = "c"; 1268f8b22f8fSZachary Turner success = PyArg_Parse(py_return.get(), format, (char *)ret_value); 12692c1f46dcSZachary Turner break; 12702c1f46dcSZachary Turner } 1271b9c1b51eSKate Stone case eScriptReturnTypeOpaqueObject: { 12722c1f46dcSZachary Turner success = true; 1273f8b22f8fSZachary Turner PyObject *saved_value = py_return.get(); 1274f8b22f8fSZachary Turner Py_XINCREF(saved_value); 1275f8b22f8fSZachary Turner *((PyObject **)ret_value) = saved_value; 12762c1f46dcSZachary Turner break; 12772c1f46dcSZachary Turner } 12782c1f46dcSZachary Turner } 1279f8b22f8fSZachary Turner 128017af5b6eSRaphael Isemann ret_success = success; 12812c1f46dcSZachary Turner } 12822c1f46dcSZachary Turner 1283f8b22f8fSZachary Turner py_error.Reset(PyRefType::Borrowed, PyErr_Occurred()); 1284b9c1b51eSKate Stone if (py_error.IsValid()) { 12852c1f46dcSZachary Turner ret_success = false; 1286b9c1b51eSKate Stone if (options.GetMaskoutErrors()) { 1287f8b22f8fSZachary Turner if (PyErr_GivenExceptionMatches(py_error.get(), PyExc_SyntaxError)) 12882c1f46dcSZachary Turner PyErr_Print(); 12892c1f46dcSZachary Turner PyErr_Clear(); 12902c1f46dcSZachary Turner } 12912c1f46dcSZachary Turner } 12922c1f46dcSZachary Turner 12932c1f46dcSZachary Turner return ret_success; 12942c1f46dcSZachary Turner } 12952c1f46dcSZachary Turner 129697206d57SZachary Turner Status ScriptInterpreterPython::ExecuteMultipleLines( 1297b9c1b51eSKate Stone const char *in_string, const ExecuteScriptOptions &options) { 129897206d57SZachary Turner Status error; 12992c1f46dcSZachary Turner 1300b9c1b51eSKate Stone Locker locker(this, ScriptInterpreterPython::Locker::AcquireLock | 1301b9c1b51eSKate Stone ScriptInterpreterPython::Locker::InitSession | 1302b9c1b51eSKate Stone (options.GetSetLLDBGlobals() 1303b9c1b51eSKate Stone ? ScriptInterpreterPython::Locker::InitGlobals 1304b9c1b51eSKate Stone : 0) | 1305b9c1b51eSKate Stone Locker::NoSTDIN, 1306b9c1b51eSKate Stone ScriptInterpreterPython::Locker::FreeAcquiredLock | 1307b9c1b51eSKate Stone ScriptInterpreterPython::Locker::TearDownSession); 13082c1f46dcSZachary Turner 13092c1f46dcSZachary Turner PythonObject return_value; 13102c1f46dcSZachary Turner PythonObject &main_module = GetMainModule(); 1311b9c1b51eSKate Stone PythonDictionary globals(PyRefType::Borrowed, 1312b9c1b51eSKate Stone PyModule_GetDict(main_module.get())); 1313f8b22f8fSZachary Turner PythonObject py_error; 13142c1f46dcSZachary Turner 13152c1f46dcSZachary Turner PythonDictionary locals = GetSessionDictionary(); 13162c1f46dcSZachary Turner 1317f8b22f8fSZachary Turner if (!locals.IsValid()) 1318b9c1b51eSKate Stone locals.Reset( 1319b9c1b51eSKate Stone PyRefType::Owned, 1320b9c1b51eSKate Stone PyObject_GetAttrString(globals.get(), m_dictionary_name.c_str())); 13212c1f46dcSZachary Turner 1322f8b22f8fSZachary Turner if (!locals.IsValid()) 13232c1f46dcSZachary Turner locals = globals; 13242c1f46dcSZachary Turner 1325f8b22f8fSZachary Turner py_error.Reset(PyRefType::Borrowed, PyErr_Occurred()); 1326f8b22f8fSZachary Turner if (py_error.IsValid()) 13272c1f46dcSZachary Turner PyErr_Clear(); 13282c1f46dcSZachary Turner 1329b9c1b51eSKate Stone if (in_string != nullptr) { 1330f8b22f8fSZachary Turner PythonObject code_object; 1331b9c1b51eSKate Stone code_object.Reset(PyRefType::Owned, 1332b9c1b51eSKate Stone Py_CompileString(in_string, "temp.py", Py_file_input)); 1333f8b22f8fSZachary Turner 1334b9c1b51eSKate Stone if (code_object.IsValid()) { 1335b9c1b51eSKate Stone // In Python 2.x, PyEval_EvalCode takes a PyCodeObject, but in Python 3.x, it 133605097246SAdrian Prantl // takes a PyObject. They are convertible (hence the function 133705097246SAdrian Prantl // PyCode_Check(PyObject*), so we have to do the cast for Python 2.x 133822c8efcdSZachary Turner #if PY_MAJOR_VERSION >= 3 1339f8b22f8fSZachary Turner PyObject *py_code_obj = code_object.get(); 134022c8efcdSZachary Turner #else 1341b9c1b51eSKate Stone PyCodeObject *py_code_obj = 1342b9c1b51eSKate Stone reinterpret_cast<PyCodeObject *>(code_object.get()); 134322c8efcdSZachary Turner #endif 1344b9c1b51eSKate Stone return_value.Reset( 1345b9c1b51eSKate Stone PyRefType::Owned, 1346b9c1b51eSKate Stone PyEval_EvalCode(py_code_obj, globals.get(), locals.get())); 1347f8b22f8fSZachary Turner } 13482c1f46dcSZachary Turner } 13492c1f46dcSZachary Turner 13503946247cSZachary Turner PythonExceptionState exception_state(!options.GetMaskoutErrors()); 13513946247cSZachary Turner if (exception_state.IsError()) 13523946247cSZachary Turner error.SetErrorString(exception_state.Format().c_str()); 13532c1f46dcSZachary Turner 13542c1f46dcSZachary Turner return error; 13552c1f46dcSZachary Turner } 13562c1f46dcSZachary Turner 1357b9c1b51eSKate Stone void ScriptInterpreterPython::CollectDataForBreakpointCommandCallback( 1358b9c1b51eSKate Stone std::vector<BreakpointOptions *> &bp_options_vec, 1359b9c1b51eSKate Stone CommandReturnObject &result) { 13602c1f46dcSZachary Turner m_active_io_handler = eIOHandlerBreakpoint; 1361b9c1b51eSKate Stone m_interpreter.GetPythonCommandsFromIOHandler(" ", *this, true, 1362b9c1b51eSKate Stone &bp_options_vec); 13632c1f46dcSZachary Turner } 13642c1f46dcSZachary Turner 1365b9c1b51eSKate Stone void ScriptInterpreterPython::CollectDataForWatchpointCommandCallback( 1366b9c1b51eSKate Stone WatchpointOptions *wp_options, CommandReturnObject &result) { 13672c1f46dcSZachary Turner m_active_io_handler = eIOHandlerWatchpoint; 13682c1f46dcSZachary Turner m_interpreter.GetPythonCommandsFromIOHandler(" ", *this, true, wp_options); 13692c1f46dcSZachary Turner } 13702c1f46dcSZachary Turner 1371b9c1b51eSKate Stone void ScriptInterpreterPython::SetBreakpointCommandCallbackFunction( 1372b9c1b51eSKate Stone BreakpointOptions *bp_options, const char *function_name) { 13732c1f46dcSZachary Turner // For now just cons up a oneliner that calls the provided function. 13742c1f46dcSZachary Turner std::string oneliner("return "); 13752c1f46dcSZachary Turner oneliner += function_name; 13762c1f46dcSZachary Turner oneliner += "(frame, bp_loc, internal_dict)"; 1377b9c1b51eSKate Stone m_interpreter.GetScriptInterpreter()->SetBreakpointCommandCallback( 1378b9c1b51eSKate Stone bp_options, oneliner.c_str()); 13792c1f46dcSZachary Turner } 13802c1f46dcSZachary Turner 138197206d57SZachary Turner Status ScriptInterpreterPython::SetBreakpointCommandCallback( 1382f7e07256SJim Ingham BreakpointOptions *bp_options, 1383f7e07256SJim Ingham std::unique_ptr<BreakpointOptions::CommandData> &cmd_data_up) { 138497206d57SZachary Turner Status error; 1385f7e07256SJim Ingham error = GenerateBreakpointCommandCallbackData(cmd_data_up->user_source, 1386f7e07256SJim Ingham cmd_data_up->script_source); 1387f7e07256SJim Ingham if (error.Fail()) { 1388f7e07256SJim Ingham return error; 1389f7e07256SJim Ingham } 1390f7e07256SJim Ingham auto baton_sp = 1391f7e07256SJim Ingham std::make_shared<BreakpointOptions::CommandBaton>(std::move(cmd_data_up)); 1392f7e07256SJim Ingham bp_options->SetCallback(ScriptInterpreterPython::BreakpointCallbackFunction, 1393f7e07256SJim Ingham baton_sp); 1394f7e07256SJim Ingham return error; 1395f7e07256SJim Ingham } 1396f7e07256SJim Ingham 13972c1f46dcSZachary Turner // Set a Python one-liner as the callback for the breakpoint. 139897206d57SZachary Turner Status ScriptInterpreterPython::SetBreakpointCommandCallback( 1399b9c1b51eSKate Stone BreakpointOptions *bp_options, const char *command_body_text) { 1400d5b44036SJonas Devlieghere auto data_up = llvm::make_unique<CommandDataPython>(); 14012c1f46dcSZachary Turner 1402b9c1b51eSKate Stone // Split the command_body_text into lines, and pass that to 140305097246SAdrian Prantl // GenerateBreakpointCommandCallbackData. That will wrap the body in an 140405097246SAdrian Prantl // auto-generated function, and return the function name in script_source. 140505097246SAdrian Prantl // That is what the callback will actually invoke. 14062c1f46dcSZachary Turner 1407d5b44036SJonas Devlieghere data_up->user_source.SplitIntoLines(command_body_text); 1408d5b44036SJonas Devlieghere Status error = GenerateBreakpointCommandCallbackData(data_up->user_source, 1409d5b44036SJonas Devlieghere data_up->script_source); 1410b9c1b51eSKate Stone if (error.Success()) { 14114e4fbe82SZachary Turner auto baton_sp = 1412d5b44036SJonas Devlieghere std::make_shared<BreakpointOptions::CommandBaton>(std::move(data_up)); 1413b9c1b51eSKate Stone bp_options->SetCallback(ScriptInterpreterPython::BreakpointCallbackFunction, 1414b9c1b51eSKate Stone baton_sp); 14152c1f46dcSZachary Turner return error; 1416b9c1b51eSKate Stone } else 14172c1f46dcSZachary Turner return error; 14182c1f46dcSZachary Turner } 14192c1f46dcSZachary Turner 14202c1f46dcSZachary Turner // Set a Python one-liner as the callback for the watchpoint. 1421b9c1b51eSKate Stone void ScriptInterpreterPython::SetWatchpointCommandCallback( 1422b9c1b51eSKate Stone WatchpointOptions *wp_options, const char *oneliner) { 1423d5b44036SJonas Devlieghere auto data_up = llvm::make_unique<WatchpointOptions::CommandData>(); 14242c1f46dcSZachary Turner 14252c1f46dcSZachary Turner // It's necessary to set both user_source and script_source to the oneliner. 1426b9c1b51eSKate Stone // The former is used to generate callback description (as in watchpoint 142705097246SAdrian Prantl // command list) while the latter is used for Python to interpret during the 142805097246SAdrian Prantl // actual callback. 14292c1f46dcSZachary Turner 1430d5b44036SJonas Devlieghere data_up->user_source.AppendString(oneliner); 1431d5b44036SJonas Devlieghere data_up->script_source.assign(oneliner); 14322c1f46dcSZachary Turner 1433d5b44036SJonas Devlieghere if (GenerateWatchpointCommandCallbackData(data_up->user_source, 1434d5b44036SJonas Devlieghere data_up->script_source)) { 14354e4fbe82SZachary Turner auto baton_sp = 1436d5b44036SJonas Devlieghere std::make_shared<WatchpointOptions::CommandBaton>(std::move(data_up)); 1437b9c1b51eSKate Stone wp_options->SetCallback(ScriptInterpreterPython::WatchpointCallbackFunction, 1438b9c1b51eSKate Stone baton_sp); 14392c1f46dcSZachary Turner } 14402c1f46dcSZachary Turner 14412c1f46dcSZachary Turner return; 14422c1f46dcSZachary Turner } 14432c1f46dcSZachary Turner 144497206d57SZachary Turner Status ScriptInterpreterPython::ExportFunctionDefinitionToInterpreter( 1445b9c1b51eSKate Stone StringList &function_def) { 14462c1f46dcSZachary Turner // Convert StringList to one long, newline delimited, const char *. 14472c1f46dcSZachary Turner std::string function_def_string(function_def.CopyList()); 14482c1f46dcSZachary Turner 144997206d57SZachary Turner Status error = ExecuteMultipleLines( 1450b9c1b51eSKate Stone function_def_string.c_str(), 1451b9c1b51eSKate Stone ScriptInterpreter::ExecuteScriptOptions().SetEnableIO(false)); 14522c1f46dcSZachary Turner return error; 14532c1f46dcSZachary Turner } 14542c1f46dcSZachary Turner 145597206d57SZachary Turner Status ScriptInterpreterPython::GenerateFunction(const char *signature, 1456b9c1b51eSKate Stone const StringList &input) { 145797206d57SZachary Turner Status error; 14582c1f46dcSZachary Turner int num_lines = input.GetSize(); 1459b9c1b51eSKate Stone if (num_lines == 0) { 14602c1f46dcSZachary Turner error.SetErrorString("No input data."); 14612c1f46dcSZachary Turner return error; 14622c1f46dcSZachary Turner } 14632c1f46dcSZachary Turner 1464b9c1b51eSKate Stone if (!signature || *signature == 0) { 14652c1f46dcSZachary Turner error.SetErrorString("No output function name."); 14662c1f46dcSZachary Turner return error; 14672c1f46dcSZachary Turner } 14682c1f46dcSZachary Turner 14692c1f46dcSZachary Turner StreamString sstr; 14702c1f46dcSZachary Turner StringList auto_generated_function; 14712c1f46dcSZachary Turner auto_generated_function.AppendString(signature); 1472b9c1b51eSKate Stone auto_generated_function.AppendString( 1473b9c1b51eSKate Stone " global_dict = globals()"); // Grab the global dictionary 1474b9c1b51eSKate Stone auto_generated_function.AppendString( 1475b9c1b51eSKate Stone " new_keys = internal_dict.keys()"); // Make a list of keys in the 1476b9c1b51eSKate Stone // session dict 1477b9c1b51eSKate Stone auto_generated_function.AppendString( 1478b9c1b51eSKate Stone " old_keys = global_dict.keys()"); // Save list of keys in global dict 1479b9c1b51eSKate Stone auto_generated_function.AppendString( 1480b9c1b51eSKate Stone " global_dict.update (internal_dict)"); // Add the session dictionary 1481b9c1b51eSKate Stone // to the 14822c1f46dcSZachary Turner // global dictionary. 14832c1f46dcSZachary Turner 14842c1f46dcSZachary Turner // Wrap everything up inside the function, increasing the indentation. 14852c1f46dcSZachary Turner 14862c1f46dcSZachary Turner auto_generated_function.AppendString(" if True:"); 1487b9c1b51eSKate Stone for (int i = 0; i < num_lines; ++i) { 14882c1f46dcSZachary Turner sstr.Clear(); 14892c1f46dcSZachary Turner sstr.Printf(" %s", input.GetStringAtIndex(i)); 14902c1f46dcSZachary Turner auto_generated_function.AppendString(sstr.GetData()); 14912c1f46dcSZachary Turner } 1492b9c1b51eSKate Stone auto_generated_function.AppendString( 1493b9c1b51eSKate Stone " for key in new_keys:"); // Iterate over all the keys from session 1494b9c1b51eSKate Stone // dict 1495b9c1b51eSKate Stone auto_generated_function.AppendString( 1496b9c1b51eSKate Stone " internal_dict[key] = global_dict[key]"); // Update session dict 1497b9c1b51eSKate Stone // values 1498b9c1b51eSKate Stone auto_generated_function.AppendString( 1499b9c1b51eSKate Stone " if key not in old_keys:"); // If key was not originally in 1500b9c1b51eSKate Stone // global dict 1501b9c1b51eSKate Stone auto_generated_function.AppendString( 1502b9c1b51eSKate Stone " del global_dict[key]"); // ...then remove key/value from 1503b9c1b51eSKate Stone // global dict 15042c1f46dcSZachary Turner 15052c1f46dcSZachary Turner // Verify that the results are valid Python. 15062c1f46dcSZachary Turner 15072c1f46dcSZachary Turner error = ExportFunctionDefinitionToInterpreter(auto_generated_function); 15082c1f46dcSZachary Turner 15092c1f46dcSZachary Turner return error; 15102c1f46dcSZachary Turner } 15112c1f46dcSZachary Turner 1512b9c1b51eSKate Stone bool ScriptInterpreterPython::GenerateTypeScriptFunction( 1513b9c1b51eSKate Stone StringList &user_input, std::string &output, const void *name_token) { 15142c1f46dcSZachary Turner static uint32_t num_created_functions = 0; 15152c1f46dcSZachary Turner user_input.RemoveBlankLines(); 15162c1f46dcSZachary Turner StreamString sstr; 15172c1f46dcSZachary Turner 15182c1f46dcSZachary Turner // Check to see if we have any data; if not, just return. 15192c1f46dcSZachary Turner if (user_input.GetSize() == 0) 15202c1f46dcSZachary Turner return false; 15212c1f46dcSZachary Turner 1522b9c1b51eSKate Stone // Take what the user wrote, wrap it all up inside one big auto-generated 152305097246SAdrian Prantl // Python function, passing in the ValueObject as parameter to the function. 15242c1f46dcSZachary Turner 1525b9c1b51eSKate Stone std::string auto_generated_function_name( 1526b9c1b51eSKate Stone GenerateUniqueName("lldb_autogen_python_type_print_func", 1527b9c1b51eSKate Stone num_created_functions, name_token)); 1528b9c1b51eSKate Stone sstr.Printf("def %s (valobj, internal_dict):", 1529b9c1b51eSKate Stone auto_generated_function_name.c_str()); 15302c1f46dcSZachary Turner 15312c1f46dcSZachary Turner if (!GenerateFunction(sstr.GetData(), user_input).Success()) 15322c1f46dcSZachary Turner return false; 15332c1f46dcSZachary Turner 15342c1f46dcSZachary Turner // Store the name of the auto-generated function to be called. 15352c1f46dcSZachary Turner output.assign(auto_generated_function_name); 15362c1f46dcSZachary Turner return true; 15372c1f46dcSZachary Turner } 15382c1f46dcSZachary Turner 1539b9c1b51eSKate Stone bool ScriptInterpreterPython::GenerateScriptAliasFunction( 1540b9c1b51eSKate Stone StringList &user_input, std::string &output) { 15412c1f46dcSZachary Turner static uint32_t num_created_functions = 0; 15422c1f46dcSZachary Turner user_input.RemoveBlankLines(); 15432c1f46dcSZachary Turner StreamString sstr; 15442c1f46dcSZachary Turner 15452c1f46dcSZachary Turner // Check to see if we have any data; if not, just return. 15462c1f46dcSZachary Turner if (user_input.GetSize() == 0) 15472c1f46dcSZachary Turner return false; 15482c1f46dcSZachary Turner 1549b9c1b51eSKate Stone std::string auto_generated_function_name(GenerateUniqueName( 1550b9c1b51eSKate Stone "lldb_autogen_python_cmd_alias_func", num_created_functions)); 15512c1f46dcSZachary Turner 1552b9c1b51eSKate Stone sstr.Printf("def %s (debugger, args, result, internal_dict):", 1553b9c1b51eSKate Stone auto_generated_function_name.c_str()); 15542c1f46dcSZachary Turner 15552c1f46dcSZachary Turner if (!GenerateFunction(sstr.GetData(), user_input).Success()) 15562c1f46dcSZachary Turner return false; 15572c1f46dcSZachary Turner 15582c1f46dcSZachary Turner // Store the name of the auto-generated function to be called. 15592c1f46dcSZachary Turner output.assign(auto_generated_function_name); 15602c1f46dcSZachary Turner return true; 15612c1f46dcSZachary Turner } 15622c1f46dcSZachary Turner 1563b9c1b51eSKate Stone bool ScriptInterpreterPython::GenerateTypeSynthClass(StringList &user_input, 1564b9c1b51eSKate Stone std::string &output, 1565b9c1b51eSKate Stone const void *name_token) { 15662c1f46dcSZachary Turner static uint32_t num_created_classes = 0; 15672c1f46dcSZachary Turner user_input.RemoveBlankLines(); 15682c1f46dcSZachary Turner int num_lines = user_input.GetSize(); 15692c1f46dcSZachary Turner StreamString sstr; 15702c1f46dcSZachary Turner 15712c1f46dcSZachary Turner // Check to see if we have any data; if not, just return. 15722c1f46dcSZachary Turner if (user_input.GetSize() == 0) 15732c1f46dcSZachary Turner return false; 15742c1f46dcSZachary Turner 15752c1f46dcSZachary Turner // Wrap all user input into a Python class 15762c1f46dcSZachary Turner 1577b9c1b51eSKate Stone std::string auto_generated_class_name(GenerateUniqueName( 1578b9c1b51eSKate Stone "lldb_autogen_python_type_synth_class", num_created_classes, name_token)); 15792c1f46dcSZachary Turner 15802c1f46dcSZachary Turner StringList auto_generated_class; 15812c1f46dcSZachary Turner 15822c1f46dcSZachary Turner // Create the function name & definition string. 15832c1f46dcSZachary Turner 15842c1f46dcSZachary Turner sstr.Printf("class %s:", auto_generated_class_name.c_str()); 1585c156427dSZachary Turner auto_generated_class.AppendString(sstr.GetString()); 15862c1f46dcSZachary Turner 158705097246SAdrian Prantl // Wrap everything up inside the class, increasing the indentation. we don't 158805097246SAdrian Prantl // need to play any fancy indentation tricks here because there is no 15892c1f46dcSZachary Turner // surrounding code whose indentation we need to honor 1590b9c1b51eSKate Stone for (int i = 0; i < num_lines; ++i) { 15912c1f46dcSZachary Turner sstr.Clear(); 15922c1f46dcSZachary Turner sstr.Printf(" %s", user_input.GetStringAtIndex(i)); 1593c156427dSZachary Turner auto_generated_class.AppendString(sstr.GetString()); 15942c1f46dcSZachary Turner } 15952c1f46dcSZachary Turner 159605097246SAdrian Prantl // Verify that the results are valid Python. (even though the method is 159705097246SAdrian Prantl // ExportFunctionDefinitionToInterpreter, a class will actually be exported) 15982c1f46dcSZachary Turner // (TODO: rename that method to ExportDefinitionToInterpreter) 15992c1f46dcSZachary Turner if (!ExportFunctionDefinitionToInterpreter(auto_generated_class).Success()) 16002c1f46dcSZachary Turner return false; 16012c1f46dcSZachary Turner 16022c1f46dcSZachary Turner // Store the name of the auto-generated class 16032c1f46dcSZachary Turner 16042c1f46dcSZachary Turner output.assign(auto_generated_class_name); 16052c1f46dcSZachary Turner return true; 16062c1f46dcSZachary Turner } 16072c1f46dcSZachary Turner 160841ae8e74SKuba Mracek StructuredData::GenericSP ScriptInterpreterPython::CreateFrameRecognizer( 160941ae8e74SKuba Mracek const char *class_name) { 161041ae8e74SKuba Mracek if (class_name == nullptr || class_name[0] == '\0') 161141ae8e74SKuba Mracek return StructuredData::GenericSP(); 161241ae8e74SKuba Mracek 161341ae8e74SKuba Mracek void *ret_val; 161441ae8e74SKuba Mracek 161541ae8e74SKuba Mracek { 161641ae8e74SKuba Mracek Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, 161741ae8e74SKuba Mracek Locker::FreeLock); 1618*05495c5dSJonas Devlieghere ret_val = LLDBSWIGPython_CreateFrameRecognizer(class_name, 1619*05495c5dSJonas Devlieghere m_dictionary_name.c_str()); 162041ae8e74SKuba Mracek } 162141ae8e74SKuba Mracek 162241ae8e74SKuba Mracek return StructuredData::GenericSP(new StructuredPythonObject(ret_val)); 162341ae8e74SKuba Mracek } 162441ae8e74SKuba Mracek 162541ae8e74SKuba Mracek lldb::ValueObjectListSP ScriptInterpreterPython::GetRecognizedArguments( 162641ae8e74SKuba Mracek const StructuredData::ObjectSP &os_plugin_object_sp, 162741ae8e74SKuba Mracek lldb::StackFrameSP frame_sp) { 162841ae8e74SKuba Mracek Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 162941ae8e74SKuba Mracek 163041ae8e74SKuba Mracek if (!os_plugin_object_sp) return ValueObjectListSP(); 163141ae8e74SKuba Mracek 163241ae8e74SKuba Mracek StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric(); 163341ae8e74SKuba Mracek if (!generic) return nullptr; 163441ae8e74SKuba Mracek 163541ae8e74SKuba Mracek PythonObject implementor(PyRefType::Borrowed, 163641ae8e74SKuba Mracek (PyObject *)generic->GetValue()); 163741ae8e74SKuba Mracek 163841ae8e74SKuba Mracek if (!implementor.IsAllocated()) return ValueObjectListSP(); 163941ae8e74SKuba Mracek 1640*05495c5dSJonas Devlieghere PythonObject py_return(PyRefType::Owned, 1641*05495c5dSJonas Devlieghere (PyObject *)LLDBSwigPython_GetRecognizedArguments( 1642*05495c5dSJonas Devlieghere implementor.get(), frame_sp)); 164341ae8e74SKuba Mracek 164441ae8e74SKuba Mracek // if it fails, print the error but otherwise go on 164541ae8e74SKuba Mracek if (PyErr_Occurred()) { 164641ae8e74SKuba Mracek PyErr_Print(); 164741ae8e74SKuba Mracek PyErr_Clear(); 164841ae8e74SKuba Mracek } 164941ae8e74SKuba Mracek if (py_return.get()) { 165041ae8e74SKuba Mracek PythonList result_list(PyRefType::Borrowed, py_return.get()); 165141ae8e74SKuba Mracek ValueObjectListSP result = ValueObjectListSP(new ValueObjectList()); 16528f81aed1SDavid Bolvansky for (size_t i = 0; i < result_list.GetSize(); i++) { 165341ae8e74SKuba Mracek PyObject *item = result_list.GetItemAtIndex(i).get(); 165441ae8e74SKuba Mracek lldb::SBValue *sb_value_ptr = 1655*05495c5dSJonas Devlieghere (lldb::SBValue *)LLDBSWIGPython_CastPyObjectToSBValue(item); 1656*05495c5dSJonas Devlieghere auto valobj_sp = LLDBSWIGPython_GetValueObjectSPFromSBValue(sb_value_ptr); 165741ae8e74SKuba Mracek if (valobj_sp) result->Append(valobj_sp); 165841ae8e74SKuba Mracek } 165941ae8e74SKuba Mracek return result; 166041ae8e74SKuba Mracek } 166141ae8e74SKuba Mracek return ValueObjectListSP(); 166241ae8e74SKuba Mracek } 166341ae8e74SKuba Mracek 1664b9c1b51eSKate Stone StructuredData::GenericSP ScriptInterpreterPython::OSPlugin_CreatePluginObject( 1665b9c1b51eSKate Stone const char *class_name, lldb::ProcessSP process_sp) { 16662c1f46dcSZachary Turner if (class_name == nullptr || class_name[0] == '\0') 16672c1f46dcSZachary Turner return StructuredData::GenericSP(); 16682c1f46dcSZachary Turner 16692c1f46dcSZachary Turner if (!process_sp) 16702c1f46dcSZachary Turner return StructuredData::GenericSP(); 16712c1f46dcSZachary Turner 16722c1f46dcSZachary Turner void *ret_val; 16732c1f46dcSZachary Turner 16742c1f46dcSZachary Turner { 1675b9c1b51eSKate Stone Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, 16762c1f46dcSZachary Turner Locker::FreeLock); 1677*05495c5dSJonas Devlieghere ret_val = LLDBSWIGPythonCreateOSPlugin( 1678*05495c5dSJonas Devlieghere class_name, m_dictionary_name.c_str(), process_sp); 16792c1f46dcSZachary Turner } 16802c1f46dcSZachary Turner 16812c1f46dcSZachary Turner return StructuredData::GenericSP(new StructuredPythonObject(ret_val)); 16822c1f46dcSZachary Turner } 16832c1f46dcSZachary Turner 1684b9c1b51eSKate Stone StructuredData::DictionarySP ScriptInterpreterPython::OSPlugin_RegisterInfo( 1685b9c1b51eSKate Stone StructuredData::ObjectSP os_plugin_object_sp) { 1686b9c1b51eSKate Stone Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 16872c1f46dcSZachary Turner 16882c1f46dcSZachary Turner static char callee_name[] = "get_register_info"; 16892c1f46dcSZachary Turner 16902c1f46dcSZachary Turner if (!os_plugin_object_sp) 16912c1f46dcSZachary Turner return StructuredData::DictionarySP(); 16922c1f46dcSZachary Turner 16932c1f46dcSZachary Turner StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric(); 16942c1f46dcSZachary Turner if (!generic) 16952c1f46dcSZachary Turner return nullptr; 16962c1f46dcSZachary Turner 1697b9c1b51eSKate Stone PythonObject implementor(PyRefType::Borrowed, 1698b9c1b51eSKate Stone (PyObject *)generic->GetValue()); 16992c1f46dcSZachary Turner 1700f8b22f8fSZachary Turner if (!implementor.IsAllocated()) 17012c1f46dcSZachary Turner return StructuredData::DictionarySP(); 17022c1f46dcSZachary Turner 1703b9c1b51eSKate Stone PythonObject pmeth(PyRefType::Owned, 1704b9c1b51eSKate Stone PyObject_GetAttrString(implementor.get(), callee_name)); 17052c1f46dcSZachary Turner 17062c1f46dcSZachary Turner if (PyErr_Occurred()) 17072c1f46dcSZachary Turner PyErr_Clear(); 17082c1f46dcSZachary Turner 1709f8b22f8fSZachary Turner if (!pmeth.IsAllocated()) 17102c1f46dcSZachary Turner return StructuredData::DictionarySP(); 17112c1f46dcSZachary Turner 1712b9c1b51eSKate Stone if (PyCallable_Check(pmeth.get()) == 0) { 17132c1f46dcSZachary Turner if (PyErr_Occurred()) 17142c1f46dcSZachary Turner PyErr_Clear(); 17152c1f46dcSZachary Turner 17162c1f46dcSZachary Turner return StructuredData::DictionarySP(); 17172c1f46dcSZachary Turner } 17182c1f46dcSZachary Turner 17192c1f46dcSZachary Turner if (PyErr_Occurred()) 17202c1f46dcSZachary Turner PyErr_Clear(); 17212c1f46dcSZachary Turner 17222c1f46dcSZachary Turner // right now we know this function exists and is callable.. 1723b9c1b51eSKate Stone PythonObject py_return( 1724b9c1b51eSKate Stone PyRefType::Owned, 1725b9c1b51eSKate Stone PyObject_CallMethod(implementor.get(), callee_name, nullptr)); 17262c1f46dcSZachary Turner 17272c1f46dcSZachary Turner // if it fails, print the error but otherwise go on 1728b9c1b51eSKate Stone if (PyErr_Occurred()) { 17292c1f46dcSZachary Turner PyErr_Print(); 17302c1f46dcSZachary Turner PyErr_Clear(); 17312c1f46dcSZachary Turner } 1732b9c1b51eSKate Stone if (py_return.get()) { 1733f8b22f8fSZachary Turner PythonDictionary result_dict(PyRefType::Borrowed, py_return.get()); 17342c1f46dcSZachary Turner return result_dict.CreateStructuredDictionary(); 17352c1f46dcSZachary Turner } 173658b794aeSGreg Clayton return StructuredData::DictionarySP(); 173758b794aeSGreg Clayton } 17382c1f46dcSZachary Turner 1739b9c1b51eSKate Stone StructuredData::ArraySP ScriptInterpreterPython::OSPlugin_ThreadsInfo( 1740b9c1b51eSKate Stone StructuredData::ObjectSP os_plugin_object_sp) { 1741b9c1b51eSKate Stone Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 17422c1f46dcSZachary Turner 17432c1f46dcSZachary Turner static char callee_name[] = "get_thread_info"; 17442c1f46dcSZachary Turner 17452c1f46dcSZachary Turner if (!os_plugin_object_sp) 17462c1f46dcSZachary Turner return StructuredData::ArraySP(); 17472c1f46dcSZachary Turner 17482c1f46dcSZachary Turner StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric(); 17492c1f46dcSZachary Turner if (!generic) 17502c1f46dcSZachary Turner return nullptr; 17512c1f46dcSZachary Turner 1752b9c1b51eSKate Stone PythonObject implementor(PyRefType::Borrowed, 1753b9c1b51eSKate Stone (PyObject *)generic->GetValue()); 1754f8b22f8fSZachary Turner 1755f8b22f8fSZachary Turner if (!implementor.IsAllocated()) 17562c1f46dcSZachary Turner return StructuredData::ArraySP(); 17572c1f46dcSZachary Turner 1758b9c1b51eSKate Stone PythonObject pmeth(PyRefType::Owned, 1759b9c1b51eSKate Stone PyObject_GetAttrString(implementor.get(), callee_name)); 17602c1f46dcSZachary Turner 17612c1f46dcSZachary Turner if (PyErr_Occurred()) 17622c1f46dcSZachary Turner PyErr_Clear(); 17632c1f46dcSZachary Turner 1764f8b22f8fSZachary Turner if (!pmeth.IsAllocated()) 17652c1f46dcSZachary Turner return StructuredData::ArraySP(); 17662c1f46dcSZachary Turner 1767b9c1b51eSKate Stone if (PyCallable_Check(pmeth.get()) == 0) { 17682c1f46dcSZachary Turner if (PyErr_Occurred()) 17692c1f46dcSZachary Turner PyErr_Clear(); 17702c1f46dcSZachary Turner 17712c1f46dcSZachary Turner return StructuredData::ArraySP(); 17722c1f46dcSZachary Turner } 17732c1f46dcSZachary Turner 17742c1f46dcSZachary Turner if (PyErr_Occurred()) 17752c1f46dcSZachary Turner PyErr_Clear(); 17762c1f46dcSZachary Turner 17772c1f46dcSZachary Turner // right now we know this function exists and is callable.. 1778b9c1b51eSKate Stone PythonObject py_return( 1779b9c1b51eSKate Stone PyRefType::Owned, 1780b9c1b51eSKate Stone PyObject_CallMethod(implementor.get(), callee_name, nullptr)); 17812c1f46dcSZachary Turner 17822c1f46dcSZachary Turner // if it fails, print the error but otherwise go on 1783b9c1b51eSKate Stone if (PyErr_Occurred()) { 17842c1f46dcSZachary Turner PyErr_Print(); 17852c1f46dcSZachary Turner PyErr_Clear(); 17862c1f46dcSZachary Turner } 17872c1f46dcSZachary Turner 1788b9c1b51eSKate Stone if (py_return.get()) { 1789f8b22f8fSZachary Turner PythonList result_list(PyRefType::Borrowed, py_return.get()); 1790f8b22f8fSZachary Turner return result_list.CreateStructuredArray(); 17912c1f46dcSZachary Turner } 179258b794aeSGreg Clayton return StructuredData::ArraySP(); 179358b794aeSGreg Clayton } 17942c1f46dcSZachary Turner 17952c1f46dcSZachary Turner // GetPythonValueFormatString provides a system independent type safe way to 17962c1f46dcSZachary Turner // convert a variable's type into a python value format. Python value formats 179705097246SAdrian Prantl // are defined in terms of builtin C types and could change from system to as 179805097246SAdrian Prantl // the underlying typedef for uint* types, size_t, off_t and other values 17992c1f46dcSZachary Turner // change. 18002c1f46dcSZachary Turner 1801684c2c93SPavel Labath template <typename T> const char *GetPythonValueFormatString(T t); 18022c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString(char *) { return "s"; } 18032c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString(char) { return "b"; } 1804b9c1b51eSKate Stone template <> const char *GetPythonValueFormatString(unsigned char) { 1805b9c1b51eSKate Stone return "B"; 1806b9c1b51eSKate Stone } 18072c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString(short) { return "h"; } 1808b9c1b51eSKate Stone template <> const char *GetPythonValueFormatString(unsigned short) { 1809b9c1b51eSKate Stone return "H"; 1810b9c1b51eSKate Stone } 18112c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString(int) { return "i"; } 18122c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString(unsigned int) { return "I"; } 18132c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString(long) { return "l"; } 1814b9c1b51eSKate Stone template <> const char *GetPythonValueFormatString(unsigned long) { 1815b9c1b51eSKate Stone return "k"; 1816b9c1b51eSKate Stone } 18172c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString(long long) { return "L"; } 1818b9c1b51eSKate Stone template <> const char *GetPythonValueFormatString(unsigned long long) { 1819b9c1b51eSKate Stone return "K"; 1820b9c1b51eSKate Stone } 18212c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString(float t) { return "f"; } 18222c1f46dcSZachary Turner template <> const char *GetPythonValueFormatString(double t) { return "d"; } 18232c1f46dcSZachary Turner 1824b9c1b51eSKate Stone StructuredData::StringSP ScriptInterpreterPython::OSPlugin_RegisterContextData( 1825b9c1b51eSKate Stone StructuredData::ObjectSP os_plugin_object_sp, lldb::tid_t tid) { 1826b9c1b51eSKate Stone Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 18272c1f46dcSZachary Turner 18282c1f46dcSZachary Turner static char callee_name[] = "get_register_data"; 1829b9c1b51eSKate Stone static char *param_format = 1830b9c1b51eSKate Stone const_cast<char *>(GetPythonValueFormatString(tid)); 18312c1f46dcSZachary Turner 18322c1f46dcSZachary Turner if (!os_plugin_object_sp) 18332c1f46dcSZachary Turner return StructuredData::StringSP(); 18342c1f46dcSZachary Turner 18352c1f46dcSZachary Turner StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric(); 18362c1f46dcSZachary Turner if (!generic) 18372c1f46dcSZachary Turner return nullptr; 1838b9c1b51eSKate Stone PythonObject implementor(PyRefType::Borrowed, 1839b9c1b51eSKate Stone (PyObject *)generic->GetValue()); 18402c1f46dcSZachary Turner 1841f8b22f8fSZachary Turner if (!implementor.IsAllocated()) 18422c1f46dcSZachary Turner return StructuredData::StringSP(); 18432c1f46dcSZachary Turner 1844b9c1b51eSKate Stone PythonObject pmeth(PyRefType::Owned, 1845b9c1b51eSKate Stone PyObject_GetAttrString(implementor.get(), callee_name)); 18462c1f46dcSZachary Turner 18472c1f46dcSZachary Turner if (PyErr_Occurred()) 18482c1f46dcSZachary Turner PyErr_Clear(); 18492c1f46dcSZachary Turner 1850f8b22f8fSZachary Turner if (!pmeth.IsAllocated()) 18512c1f46dcSZachary Turner return StructuredData::StringSP(); 18522c1f46dcSZachary Turner 1853b9c1b51eSKate Stone if (PyCallable_Check(pmeth.get()) == 0) { 18542c1f46dcSZachary Turner if (PyErr_Occurred()) 18552c1f46dcSZachary Turner PyErr_Clear(); 18562c1f46dcSZachary Turner return StructuredData::StringSP(); 18572c1f46dcSZachary Turner } 18582c1f46dcSZachary Turner 18592c1f46dcSZachary Turner if (PyErr_Occurred()) 18602c1f46dcSZachary Turner PyErr_Clear(); 18612c1f46dcSZachary Turner 18622c1f46dcSZachary Turner // right now we know this function exists and is callable.. 1863b9c1b51eSKate Stone PythonObject py_return( 1864b9c1b51eSKate Stone PyRefType::Owned, 1865b9c1b51eSKate Stone PyObject_CallMethod(implementor.get(), callee_name, param_format, tid)); 18662c1f46dcSZachary Turner 18672c1f46dcSZachary Turner // if it fails, print the error but otherwise go on 1868b9c1b51eSKate Stone if (PyErr_Occurred()) { 18692c1f46dcSZachary Turner PyErr_Print(); 18702c1f46dcSZachary Turner PyErr_Clear(); 18712c1f46dcSZachary Turner } 1872f8b22f8fSZachary Turner 1873b9c1b51eSKate Stone if (py_return.get()) { 18747a76845cSZachary Turner PythonBytes result(PyRefType::Borrowed, py_return.get()); 18757a76845cSZachary Turner return result.CreateStructuredString(); 18762c1f46dcSZachary Turner } 187758b794aeSGreg Clayton return StructuredData::StringSP(); 187858b794aeSGreg Clayton } 18792c1f46dcSZachary Turner 1880b9c1b51eSKate Stone StructuredData::DictionarySP ScriptInterpreterPython::OSPlugin_CreateThread( 1881b9c1b51eSKate Stone StructuredData::ObjectSP os_plugin_object_sp, lldb::tid_t tid, 1882b9c1b51eSKate Stone lldb::addr_t context) { 1883b9c1b51eSKate Stone Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 18842c1f46dcSZachary Turner 18852c1f46dcSZachary Turner static char callee_name[] = "create_thread"; 18862c1f46dcSZachary Turner std::string param_format; 18872c1f46dcSZachary Turner param_format += GetPythonValueFormatString(tid); 18882c1f46dcSZachary Turner param_format += GetPythonValueFormatString(context); 18892c1f46dcSZachary Turner 18902c1f46dcSZachary Turner if (!os_plugin_object_sp) 18912c1f46dcSZachary Turner return StructuredData::DictionarySP(); 18922c1f46dcSZachary Turner 18932c1f46dcSZachary Turner StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric(); 18942c1f46dcSZachary Turner if (!generic) 18952c1f46dcSZachary Turner return nullptr; 18962c1f46dcSZachary Turner 1897b9c1b51eSKate Stone PythonObject implementor(PyRefType::Borrowed, 1898b9c1b51eSKate Stone (PyObject *)generic->GetValue()); 1899f8b22f8fSZachary Turner 1900f8b22f8fSZachary Turner if (!implementor.IsAllocated()) 19012c1f46dcSZachary Turner return StructuredData::DictionarySP(); 19022c1f46dcSZachary Turner 1903b9c1b51eSKate Stone PythonObject pmeth(PyRefType::Owned, 1904b9c1b51eSKate Stone PyObject_GetAttrString(implementor.get(), callee_name)); 19052c1f46dcSZachary Turner 19062c1f46dcSZachary Turner if (PyErr_Occurred()) 19072c1f46dcSZachary Turner PyErr_Clear(); 19082c1f46dcSZachary Turner 1909f8b22f8fSZachary Turner if (!pmeth.IsAllocated()) 19102c1f46dcSZachary Turner return StructuredData::DictionarySP(); 19112c1f46dcSZachary Turner 1912b9c1b51eSKate Stone if (PyCallable_Check(pmeth.get()) == 0) { 19132c1f46dcSZachary Turner if (PyErr_Occurred()) 19142c1f46dcSZachary Turner PyErr_Clear(); 19152c1f46dcSZachary Turner return StructuredData::DictionarySP(); 19162c1f46dcSZachary Turner } 19172c1f46dcSZachary Turner 19182c1f46dcSZachary Turner if (PyErr_Occurred()) 19192c1f46dcSZachary Turner PyErr_Clear(); 19202c1f46dcSZachary Turner 19212c1f46dcSZachary Turner // right now we know this function exists and is callable.. 1922b9c1b51eSKate Stone PythonObject py_return(PyRefType::Owned, 1923b9c1b51eSKate Stone PyObject_CallMethod(implementor.get(), callee_name, 1924b9c1b51eSKate Stone ¶m_format[0], tid, context)); 19252c1f46dcSZachary Turner 19262c1f46dcSZachary Turner // if it fails, print the error but otherwise go on 1927b9c1b51eSKate Stone if (PyErr_Occurred()) { 19282c1f46dcSZachary Turner PyErr_Print(); 19292c1f46dcSZachary Turner PyErr_Clear(); 19302c1f46dcSZachary Turner } 19312c1f46dcSZachary Turner 1932b9c1b51eSKate Stone if (py_return.get()) { 1933f8b22f8fSZachary Turner PythonDictionary result_dict(PyRefType::Borrowed, py_return.get()); 19342c1f46dcSZachary Turner return result_dict.CreateStructuredDictionary(); 19352c1f46dcSZachary Turner } 193658b794aeSGreg Clayton return StructuredData::DictionarySP(); 193758b794aeSGreg Clayton } 19382c1f46dcSZachary Turner 1939b9c1b51eSKate Stone StructuredData::ObjectSP ScriptInterpreterPython::CreateScriptedThreadPlan( 1940b9c1b51eSKate Stone const char *class_name, lldb::ThreadPlanSP thread_plan_sp) { 19412c1f46dcSZachary Turner if (class_name == nullptr || class_name[0] == '\0') 19422c1f46dcSZachary Turner return StructuredData::ObjectSP(); 19432c1f46dcSZachary Turner 19442c1f46dcSZachary Turner if (!thread_plan_sp.get()) 19452c1f46dcSZachary Turner return StructuredData::ObjectSP(); 19462c1f46dcSZachary Turner 19472c1f46dcSZachary Turner Debugger &debugger = thread_plan_sp->GetTarget().GetDebugger(); 1948b9c1b51eSKate Stone ScriptInterpreter *script_interpreter = 1949b9c1b51eSKate Stone debugger.GetCommandInterpreter().GetScriptInterpreter(); 1950b9c1b51eSKate Stone ScriptInterpreterPython *python_interpreter = 1951b9c1b51eSKate Stone static_cast<ScriptInterpreterPython *>(script_interpreter); 19522c1f46dcSZachary Turner 19532c1f46dcSZachary Turner if (!script_interpreter) 19542c1f46dcSZachary Turner return StructuredData::ObjectSP(); 19552c1f46dcSZachary Turner 19562c1f46dcSZachary Turner void *ret_val; 19572c1f46dcSZachary Turner 19582c1f46dcSZachary Turner { 1959b9c1b51eSKate Stone Locker py_lock(this, 1960b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 19612c1f46dcSZachary Turner 1962*05495c5dSJonas Devlieghere ret_val = LLDBSwigPythonCreateScriptedThreadPlan( 1963b9c1b51eSKate Stone class_name, python_interpreter->m_dictionary_name.c_str(), 19642c1f46dcSZachary Turner thread_plan_sp); 19652c1f46dcSZachary Turner } 19662c1f46dcSZachary Turner 19672c1f46dcSZachary Turner return StructuredData::ObjectSP(new StructuredPythonObject(ret_val)); 19682c1f46dcSZachary Turner } 19692c1f46dcSZachary Turner 1970b9c1b51eSKate Stone bool ScriptInterpreterPython::ScriptedThreadPlanExplainsStop( 1971b9c1b51eSKate Stone StructuredData::ObjectSP implementor_sp, Event *event, bool &script_error) { 19722c1f46dcSZachary Turner bool explains_stop = true; 19732c1f46dcSZachary Turner StructuredData::Generic *generic = nullptr; 19742c1f46dcSZachary Turner if (implementor_sp) 19752c1f46dcSZachary Turner generic = implementor_sp->GetAsGeneric(); 1976b9c1b51eSKate Stone if (generic) { 1977b9c1b51eSKate Stone Locker py_lock(this, 1978b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1979*05495c5dSJonas Devlieghere explains_stop = LLDBSWIGPythonCallThreadPlan( 1980b9c1b51eSKate Stone generic->GetValue(), "explains_stop", event, script_error); 19812c1f46dcSZachary Turner if (script_error) 19822c1f46dcSZachary Turner return true; 19832c1f46dcSZachary Turner } 19842c1f46dcSZachary Turner return explains_stop; 19852c1f46dcSZachary Turner } 19862c1f46dcSZachary Turner 1987b9c1b51eSKate Stone bool ScriptInterpreterPython::ScriptedThreadPlanShouldStop( 1988b9c1b51eSKate Stone StructuredData::ObjectSP implementor_sp, Event *event, bool &script_error) { 19892c1f46dcSZachary Turner bool should_stop = true; 19902c1f46dcSZachary Turner StructuredData::Generic *generic = nullptr; 19912c1f46dcSZachary Turner if (implementor_sp) 19922c1f46dcSZachary Turner generic = implementor_sp->GetAsGeneric(); 1993b9c1b51eSKate Stone if (generic) { 1994b9c1b51eSKate Stone Locker py_lock(this, 1995b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1996*05495c5dSJonas Devlieghere should_stop = LLDBSWIGPythonCallThreadPlan( 1997*05495c5dSJonas Devlieghere generic->GetValue(), "should_stop", event, script_error); 19982c1f46dcSZachary Turner if (script_error) 19992c1f46dcSZachary Turner return true; 20002c1f46dcSZachary Turner } 20012c1f46dcSZachary Turner return should_stop; 20022c1f46dcSZachary Turner } 20032c1f46dcSZachary Turner 2004b9c1b51eSKate Stone bool ScriptInterpreterPython::ScriptedThreadPlanIsStale( 2005b9c1b51eSKate Stone StructuredData::ObjectSP implementor_sp, bool &script_error) { 2006c915a7d2SJim Ingham bool is_stale = true; 2007c915a7d2SJim Ingham StructuredData::Generic *generic = nullptr; 2008c915a7d2SJim Ingham if (implementor_sp) 2009c915a7d2SJim Ingham generic = implementor_sp->GetAsGeneric(); 2010b9c1b51eSKate Stone if (generic) { 2011b9c1b51eSKate Stone Locker py_lock(this, 2012b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2013*05495c5dSJonas Devlieghere is_stale = LLDBSWIGPythonCallThreadPlan(generic->GetValue(), "is_stale", 2014*05495c5dSJonas Devlieghere nullptr, script_error); 2015c915a7d2SJim Ingham if (script_error) 2016c915a7d2SJim Ingham return true; 2017c915a7d2SJim Ingham } 2018c915a7d2SJim Ingham return is_stale; 2019c915a7d2SJim Ingham } 2020c915a7d2SJim Ingham 2021b9c1b51eSKate Stone lldb::StateType ScriptInterpreterPython::ScriptedThreadPlanGetRunState( 2022b9c1b51eSKate Stone StructuredData::ObjectSP implementor_sp, bool &script_error) { 20232c1f46dcSZachary Turner bool should_step = false; 20242c1f46dcSZachary Turner StructuredData::Generic *generic = nullptr; 20252c1f46dcSZachary Turner if (implementor_sp) 20262c1f46dcSZachary Turner generic = implementor_sp->GetAsGeneric(); 2027b9c1b51eSKate Stone if (generic) { 2028b9c1b51eSKate Stone Locker py_lock(this, 2029b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2030*05495c5dSJonas Devlieghere should_step = LLDBSWIGPythonCallThreadPlan( 2031*05495c5dSJonas Devlieghere generic->GetValue(), "should_step", NULL, script_error); 20322c1f46dcSZachary Turner if (script_error) 20332c1f46dcSZachary Turner should_step = true; 20342c1f46dcSZachary Turner } 20352c1f46dcSZachary Turner if (should_step) 20362c1f46dcSZachary Turner return lldb::eStateStepping; 20372c1f46dcSZachary Turner else 20382c1f46dcSZachary Turner return lldb::eStateRunning; 20392c1f46dcSZachary Turner } 20402c1f46dcSZachary Turner 20413815e702SJim Ingham StructuredData::GenericSP 20423815e702SJim Ingham ScriptInterpreterPython::CreateScriptedBreakpointResolver( 20433815e702SJim Ingham const char *class_name, 20443815e702SJim Ingham StructuredDataImpl *args_data, 20453815e702SJim Ingham lldb::BreakpointSP &bkpt_sp) { 20463815e702SJim Ingham 20473815e702SJim Ingham if (class_name == nullptr || class_name[0] == '\0') 20483815e702SJim Ingham return StructuredData::GenericSP(); 20493815e702SJim Ingham 20503815e702SJim Ingham if (!bkpt_sp.get()) 20513815e702SJim Ingham return StructuredData::GenericSP(); 20523815e702SJim Ingham 20533815e702SJim Ingham Debugger &debugger = bkpt_sp->GetTarget().GetDebugger(); 20543815e702SJim Ingham ScriptInterpreter *script_interpreter = 20553815e702SJim Ingham debugger.GetCommandInterpreter().GetScriptInterpreter(); 20563815e702SJim Ingham ScriptInterpreterPython *python_interpreter = 20573815e702SJim Ingham static_cast<ScriptInterpreterPython *>(script_interpreter); 20583815e702SJim Ingham 20593815e702SJim Ingham if (!script_interpreter) 20603815e702SJim Ingham return StructuredData::GenericSP(); 20613815e702SJim Ingham 20623815e702SJim Ingham void *ret_val; 20633815e702SJim Ingham 20643815e702SJim Ingham { 20653815e702SJim Ingham Locker py_lock(this, 20663815e702SJim Ingham Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 20673815e702SJim Ingham 2068*05495c5dSJonas Devlieghere ret_val = LLDBSwigPythonCreateScriptedBreakpointResolver( 2069*05495c5dSJonas Devlieghere class_name, python_interpreter->m_dictionary_name.c_str(), args_data, 2070*05495c5dSJonas Devlieghere bkpt_sp); 20713815e702SJim Ingham } 20723815e702SJim Ingham 20733815e702SJim Ingham return StructuredData::GenericSP(new StructuredPythonObject(ret_val)); 20743815e702SJim Ingham } 20753815e702SJim Ingham 20763815e702SJim Ingham bool 20773815e702SJim Ingham ScriptInterpreterPython::ScriptedBreakpointResolverSearchCallback( 20783815e702SJim Ingham StructuredData::GenericSP implementor_sp, 20793815e702SJim Ingham SymbolContext *sym_ctx) { 20803815e702SJim Ingham bool should_continue = false; 20813815e702SJim Ingham 20823815e702SJim Ingham if (implementor_sp) { 20833815e702SJim Ingham Locker py_lock(this, 20843815e702SJim Ingham Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2085*05495c5dSJonas Devlieghere should_continue = LLDBSwigPythonCallBreakpointResolver( 2086*05495c5dSJonas Devlieghere implementor_sp->GetValue(), "__callback__", sym_ctx); 20873815e702SJim Ingham if (PyErr_Occurred()) { 20883815e702SJim Ingham PyErr_Print(); 20893815e702SJim Ingham PyErr_Clear(); 20903815e702SJim Ingham } 20913815e702SJim Ingham } 20923815e702SJim Ingham return should_continue; 20933815e702SJim Ingham } 20943815e702SJim Ingham 20953815e702SJim Ingham lldb::SearchDepth 20963815e702SJim Ingham ScriptInterpreterPython::ScriptedBreakpointResolverSearchDepth( 20973815e702SJim Ingham StructuredData::GenericSP implementor_sp) { 20983815e702SJim Ingham int depth_as_int = lldb::eSearchDepthModule; 20993815e702SJim Ingham if (implementor_sp) { 21003815e702SJim Ingham Locker py_lock(this, 21013815e702SJim Ingham Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2102*05495c5dSJonas Devlieghere depth_as_int = LLDBSwigPythonCallBreakpointResolver( 2103*05495c5dSJonas Devlieghere implementor_sp->GetValue(), "__get_depth__", nullptr); 21043815e702SJim Ingham if (PyErr_Occurred()) { 21053815e702SJim Ingham PyErr_Print(); 21063815e702SJim Ingham PyErr_Clear(); 21073815e702SJim Ingham } 21083815e702SJim Ingham } 21093815e702SJim Ingham if (depth_as_int == lldb::eSearchDepthInvalid) 21103815e702SJim Ingham return lldb::eSearchDepthModule; 21113815e702SJim Ingham 21123815e702SJim Ingham if (depth_as_int <= lldb::kLastSearchDepthKind) 21133815e702SJim Ingham return (lldb::SearchDepth) depth_as_int; 21143815e702SJim Ingham else 21153815e702SJim Ingham return lldb::eSearchDepthModule; 21163815e702SJim Ingham } 21173815e702SJim Ingham 21182c1f46dcSZachary Turner StructuredData::ObjectSP 2119b9c1b51eSKate Stone ScriptInterpreterPython::LoadPluginModule(const FileSpec &file_spec, 212097206d57SZachary Turner lldb_private::Status &error) { 2121dbd7fabaSJonas Devlieghere if (!FileSystem::Instance().Exists(file_spec)) { 21222c1f46dcSZachary Turner error.SetErrorString("no such file"); 21232c1f46dcSZachary Turner return StructuredData::ObjectSP(); 21242c1f46dcSZachary Turner } 21252c1f46dcSZachary Turner 21262c1f46dcSZachary Turner StructuredData::ObjectSP module_sp; 21272c1f46dcSZachary Turner 2128b9c1b51eSKate Stone if (LoadScriptingModule(file_spec.GetPath().c_str(), true, true, error, 2129b9c1b51eSKate Stone &module_sp)) 21302c1f46dcSZachary Turner return module_sp; 21312c1f46dcSZachary Turner 21322c1f46dcSZachary Turner return StructuredData::ObjectSP(); 21332c1f46dcSZachary Turner } 21342c1f46dcSZachary Turner 2135b9c1b51eSKate Stone StructuredData::DictionarySP ScriptInterpreterPython::GetDynamicSettings( 2136b9c1b51eSKate Stone StructuredData::ObjectSP plugin_module_sp, Target *target, 213797206d57SZachary Turner const char *setting_name, lldb_private::Status &error) { 2138*05495c5dSJonas Devlieghere if (!plugin_module_sp || !target || !setting_name || !setting_name[0]) 21392c1f46dcSZachary Turner return StructuredData::DictionarySP(); 21402c1f46dcSZachary Turner StructuredData::Generic *generic = plugin_module_sp->GetAsGeneric(); 21412c1f46dcSZachary Turner if (!generic) 21422c1f46dcSZachary Turner return StructuredData::DictionarySP(); 21432c1f46dcSZachary Turner 2144f8b22f8fSZachary Turner PythonObject reply_pyobj; 2145b9c1b51eSKate Stone Locker py_lock(this, 2146b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 21472c1f46dcSZachary Turner TargetSP target_sp(target->shared_from_this()); 2148f8b22f8fSZachary Turner reply_pyobj.Reset(PyRefType::Owned, 2149*05495c5dSJonas Devlieghere (PyObject *)LLDBSWIGPython_GetDynamicSetting( 2150*05495c5dSJonas Devlieghere generic->GetValue(), setting_name, target_sp)); 21512c1f46dcSZachary Turner 2152f8b22f8fSZachary Turner PythonDictionary py_dict(PyRefType::Borrowed, reply_pyobj.get()); 21532c1f46dcSZachary Turner return py_dict.CreateStructuredDictionary(); 21542c1f46dcSZachary Turner } 21552c1f46dcSZachary Turner 21562c1f46dcSZachary Turner StructuredData::ObjectSP 2157b9c1b51eSKate Stone ScriptInterpreterPython::CreateSyntheticScriptedProvider( 2158b9c1b51eSKate Stone const char *class_name, lldb::ValueObjectSP valobj) { 21592c1f46dcSZachary Turner if (class_name == nullptr || class_name[0] == '\0') 21602c1f46dcSZachary Turner return StructuredData::ObjectSP(); 21612c1f46dcSZachary Turner 21622c1f46dcSZachary Turner if (!valobj.get()) 21632c1f46dcSZachary Turner return StructuredData::ObjectSP(); 21642c1f46dcSZachary Turner 21652c1f46dcSZachary Turner ExecutionContext exe_ctx(valobj->GetExecutionContextRef()); 21662c1f46dcSZachary Turner Target *target = exe_ctx.GetTargetPtr(); 21672c1f46dcSZachary Turner 21682c1f46dcSZachary Turner if (!target) 21692c1f46dcSZachary Turner return StructuredData::ObjectSP(); 21702c1f46dcSZachary Turner 21712c1f46dcSZachary Turner Debugger &debugger = target->GetDebugger(); 2172b9c1b51eSKate Stone ScriptInterpreter *script_interpreter = 2173b9c1b51eSKate Stone debugger.GetCommandInterpreter().GetScriptInterpreter(); 2174b9c1b51eSKate Stone ScriptInterpreterPython *python_interpreter = 2175b9c1b51eSKate Stone (ScriptInterpreterPython *)script_interpreter; 21762c1f46dcSZachary Turner 21772c1f46dcSZachary Turner if (!script_interpreter) 21782c1f46dcSZachary Turner return StructuredData::ObjectSP(); 21792c1f46dcSZachary Turner 21802c1f46dcSZachary Turner void *ret_val = nullptr; 21812c1f46dcSZachary Turner 21822c1f46dcSZachary Turner { 2183b9c1b51eSKate Stone Locker py_lock(this, 2184b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2185*05495c5dSJonas Devlieghere ret_val = LLDBSwigPythonCreateSyntheticProvider( 2186b9c1b51eSKate Stone class_name, python_interpreter->m_dictionary_name.c_str(), valobj); 21872c1f46dcSZachary Turner } 21882c1f46dcSZachary Turner 21892c1f46dcSZachary Turner return StructuredData::ObjectSP(new StructuredPythonObject(ret_val)); 21902c1f46dcSZachary Turner } 21912c1f46dcSZachary Turner 21922c1f46dcSZachary Turner StructuredData::GenericSP 2193b9c1b51eSKate Stone ScriptInterpreterPython::CreateScriptCommandObject(const char *class_name) { 2194b9c1b51eSKate Stone DebuggerSP debugger_sp( 2195b9c1b51eSKate Stone GetCommandInterpreter().GetDebugger().shared_from_this()); 21962c1f46dcSZachary Turner 21972c1f46dcSZachary Turner if (class_name == nullptr || class_name[0] == '\0') 21982c1f46dcSZachary Turner return StructuredData::GenericSP(); 21992c1f46dcSZachary Turner 22002c1f46dcSZachary Turner if (!debugger_sp.get()) 22012c1f46dcSZachary Turner return StructuredData::GenericSP(); 22022c1f46dcSZachary Turner 22032c1f46dcSZachary Turner void *ret_val; 22042c1f46dcSZachary Turner 22052c1f46dcSZachary Turner { 2206b9c1b51eSKate Stone Locker py_lock(this, 2207b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2208*05495c5dSJonas Devlieghere ret_val = LLDBSwigPythonCreateCommandObject( 2209*05495c5dSJonas Devlieghere class_name, m_dictionary_name.c_str(), debugger_sp); 22102c1f46dcSZachary Turner } 22112c1f46dcSZachary Turner 22122c1f46dcSZachary Turner return StructuredData::GenericSP(new StructuredPythonObject(ret_val)); 22132c1f46dcSZachary Turner } 22142c1f46dcSZachary Turner 2215b9c1b51eSKate Stone bool ScriptInterpreterPython::GenerateTypeScriptFunction( 2216b9c1b51eSKate Stone const char *oneliner, std::string &output, const void *name_token) { 22172c1f46dcSZachary Turner StringList input; 22182c1f46dcSZachary Turner input.SplitIntoLines(oneliner, strlen(oneliner)); 22192c1f46dcSZachary Turner return GenerateTypeScriptFunction(input, output, name_token); 22202c1f46dcSZachary Turner } 22212c1f46dcSZachary Turner 2222b9c1b51eSKate Stone bool ScriptInterpreterPython::GenerateTypeSynthClass(const char *oneliner, 2223b9c1b51eSKate Stone std::string &output, 2224b9c1b51eSKate Stone const void *name_token) { 22252c1f46dcSZachary Turner StringList input; 22262c1f46dcSZachary Turner input.SplitIntoLines(oneliner, strlen(oneliner)); 22272c1f46dcSZachary Turner return GenerateTypeSynthClass(input, output, name_token); 22282c1f46dcSZachary Turner } 22292c1f46dcSZachary Turner 223097206d57SZachary Turner Status ScriptInterpreterPython::GenerateBreakpointCommandCallbackData( 2231b9c1b51eSKate Stone StringList &user_input, std::string &output) { 22322c1f46dcSZachary Turner static uint32_t num_created_functions = 0; 22332c1f46dcSZachary Turner user_input.RemoveBlankLines(); 22342c1f46dcSZachary Turner StreamString sstr; 223597206d57SZachary Turner Status error; 2236b9c1b51eSKate Stone if (user_input.GetSize() == 0) { 22372c1f46dcSZachary Turner error.SetErrorString("No input data."); 22382c1f46dcSZachary Turner return error; 22392c1f46dcSZachary Turner } 22402c1f46dcSZachary Turner 2241b9c1b51eSKate Stone std::string auto_generated_function_name(GenerateUniqueName( 2242b9c1b51eSKate Stone "lldb_autogen_python_bp_callback_func_", num_created_functions)); 2243b9c1b51eSKate Stone sstr.Printf("def %s (frame, bp_loc, internal_dict):", 2244b9c1b51eSKate Stone auto_generated_function_name.c_str()); 22452c1f46dcSZachary Turner 22462c1f46dcSZachary Turner error = GenerateFunction(sstr.GetData(), user_input); 22472c1f46dcSZachary Turner if (!error.Success()) 22482c1f46dcSZachary Turner return error; 22492c1f46dcSZachary Turner 22502c1f46dcSZachary Turner // Store the name of the auto-generated function to be called. 22512c1f46dcSZachary Turner output.assign(auto_generated_function_name); 22522c1f46dcSZachary Turner return error; 22532c1f46dcSZachary Turner } 22542c1f46dcSZachary Turner 2255b9c1b51eSKate Stone bool ScriptInterpreterPython::GenerateWatchpointCommandCallbackData( 2256b9c1b51eSKate Stone StringList &user_input, std::string &output) { 22572c1f46dcSZachary Turner static uint32_t num_created_functions = 0; 22582c1f46dcSZachary Turner user_input.RemoveBlankLines(); 22592c1f46dcSZachary Turner StreamString sstr; 22602c1f46dcSZachary Turner 22612c1f46dcSZachary Turner if (user_input.GetSize() == 0) 22622c1f46dcSZachary Turner return false; 22632c1f46dcSZachary Turner 2264b9c1b51eSKate Stone std::string auto_generated_function_name(GenerateUniqueName( 2265b9c1b51eSKate Stone "lldb_autogen_python_wp_callback_func_", num_created_functions)); 2266b9c1b51eSKate Stone sstr.Printf("def %s (frame, wp, internal_dict):", 2267b9c1b51eSKate Stone auto_generated_function_name.c_str()); 22682c1f46dcSZachary Turner 22692c1f46dcSZachary Turner if (!GenerateFunction(sstr.GetData(), user_input).Success()) 22702c1f46dcSZachary Turner return false; 22712c1f46dcSZachary Turner 22722c1f46dcSZachary Turner // Store the name of the auto-generated function to be called. 22732c1f46dcSZachary Turner output.assign(auto_generated_function_name); 22742c1f46dcSZachary Turner return true; 22752c1f46dcSZachary Turner } 22762c1f46dcSZachary Turner 2277b9c1b51eSKate Stone bool ScriptInterpreterPython::GetScriptedSummary( 2278b9c1b51eSKate Stone const char *python_function_name, lldb::ValueObjectSP valobj, 2279b9c1b51eSKate Stone StructuredData::ObjectSP &callee_wrapper_sp, 2280b9c1b51eSKate Stone const TypeSummaryOptions &options, std::string &retval) { 22812c1f46dcSZachary Turner 2282f9d16476SPavel Labath static Timer::Category func_cat(LLVM_PRETTY_FUNCTION); 2283f9d16476SPavel Labath Timer scoped_timer(func_cat, LLVM_PRETTY_FUNCTION); 22842c1f46dcSZachary Turner 2285b9c1b51eSKate Stone if (!valobj.get()) { 22862c1f46dcSZachary Turner retval.assign("<no object>"); 22872c1f46dcSZachary Turner return false; 22882c1f46dcSZachary Turner } 22892c1f46dcSZachary Turner 22902c1f46dcSZachary Turner void *old_callee = nullptr; 22912c1f46dcSZachary Turner StructuredData::Generic *generic = nullptr; 2292b9c1b51eSKate Stone if (callee_wrapper_sp) { 22932c1f46dcSZachary Turner generic = callee_wrapper_sp->GetAsGeneric(); 22942c1f46dcSZachary Turner if (generic) 22952c1f46dcSZachary Turner old_callee = generic->GetValue(); 22962c1f46dcSZachary Turner } 22972c1f46dcSZachary Turner void *new_callee = old_callee; 22982c1f46dcSZachary Turner 22992c1f46dcSZachary Turner bool ret_val; 2300b9c1b51eSKate Stone if (python_function_name && *python_function_name) { 23012c1f46dcSZachary Turner { 2302b9c1b51eSKate Stone Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | 2303b9c1b51eSKate Stone Locker::NoSTDIN); 23042c1f46dcSZachary Turner { 23052c1f46dcSZachary Turner TypeSummaryOptionsSP options_sp(new TypeSummaryOptions(options)); 23062c1f46dcSZachary Turner 2307*05495c5dSJonas Devlieghere static Timer::Category func_cat("LLDBSwigPythonCallTypeScript"); 2308*05495c5dSJonas Devlieghere Timer scoped_timer(func_cat, "LLDBSwigPythonCallTypeScript"); 2309*05495c5dSJonas Devlieghere ret_val = LLDBSwigPythonCallTypeScript( 2310b9c1b51eSKate Stone python_function_name, GetSessionDictionary().get(), valobj, 2311b9c1b51eSKate Stone &new_callee, options_sp, retval); 23122c1f46dcSZachary Turner } 23132c1f46dcSZachary Turner } 2314b9c1b51eSKate Stone } else { 23152c1f46dcSZachary Turner retval.assign("<no function name>"); 23162c1f46dcSZachary Turner return false; 23172c1f46dcSZachary Turner } 23182c1f46dcSZachary Turner 23192c1f46dcSZachary Turner if (new_callee && old_callee != new_callee) 2320796ac80bSJonas Devlieghere callee_wrapper_sp = std::make_shared<StructuredPythonObject>(new_callee); 23212c1f46dcSZachary Turner 23222c1f46dcSZachary Turner return ret_val; 23232c1f46dcSZachary Turner } 23242c1f46dcSZachary Turner 2325b9c1b51eSKate Stone void ScriptInterpreterPython::Clear() { 23262c1f46dcSZachary Turner // Release any global variables that might have strong references to 23272c1f46dcSZachary Turner // LLDB objects when clearing the python script interpreter. 2328b9c1b51eSKate Stone Locker locker(this, ScriptInterpreterPython::Locker::AcquireLock, 23292c1f46dcSZachary Turner ScriptInterpreterPython::Locker::FreeAcquiredLock); 23302c1f46dcSZachary Turner 2331b9c1b51eSKate Stone // This may be called as part of Py_Finalize. In that case the modules are 233205097246SAdrian Prantl // destroyed in random order and we can't guarantee that we can access these. 23332c1f46dcSZachary Turner if (Py_IsInitialized()) 2334b9c1b51eSKate Stone PyRun_SimpleString("lldb.debugger = None; lldb.target = None; lldb.process " 2335b9c1b51eSKate Stone "= None; lldb.thread = None; lldb.frame = None"); 23362c1f46dcSZachary Turner } 23372c1f46dcSZachary Turner 2338b9c1b51eSKate Stone bool ScriptInterpreterPython::BreakpointCallbackFunction( 2339b9c1b51eSKate Stone void *baton, StoppointCallbackContext *context, user_id_t break_id, 2340b9c1b51eSKate Stone user_id_t break_loc_id) { 2341f7e07256SJim Ingham CommandDataPython *bp_option_data = (CommandDataPython *)baton; 23422c1f46dcSZachary Turner const char *python_function_name = bp_option_data->script_source.c_str(); 23432c1f46dcSZachary Turner 23442c1f46dcSZachary Turner if (!context) 23452c1f46dcSZachary Turner return true; 23462c1f46dcSZachary Turner 23472c1f46dcSZachary Turner ExecutionContext exe_ctx(context->exe_ctx_ref); 23482c1f46dcSZachary Turner Target *target = exe_ctx.GetTargetPtr(); 23492c1f46dcSZachary Turner 23502c1f46dcSZachary Turner if (!target) 23512c1f46dcSZachary Turner return true; 23522c1f46dcSZachary Turner 23532c1f46dcSZachary Turner Debugger &debugger = target->GetDebugger(); 2354b9c1b51eSKate Stone ScriptInterpreter *script_interpreter = 2355b9c1b51eSKate Stone debugger.GetCommandInterpreter().GetScriptInterpreter(); 2356b9c1b51eSKate Stone ScriptInterpreterPython *python_interpreter = 2357b9c1b51eSKate Stone (ScriptInterpreterPython *)script_interpreter; 23582c1f46dcSZachary Turner 23592c1f46dcSZachary Turner if (!script_interpreter) 23602c1f46dcSZachary Turner return true; 23612c1f46dcSZachary Turner 2362b9c1b51eSKate Stone if (python_function_name && python_function_name[0]) { 23632c1f46dcSZachary Turner const StackFrameSP stop_frame_sp(exe_ctx.GetFrameSP()); 23642c1f46dcSZachary Turner BreakpointSP breakpoint_sp = target->GetBreakpointByID(break_id); 2365b9c1b51eSKate Stone if (breakpoint_sp) { 2366b9c1b51eSKate Stone const BreakpointLocationSP bp_loc_sp( 2367b9c1b51eSKate Stone breakpoint_sp->FindLocationByID(break_loc_id)); 23682c1f46dcSZachary Turner 2369b9c1b51eSKate Stone if (stop_frame_sp && bp_loc_sp) { 23702c1f46dcSZachary Turner bool ret_val = true; 23712c1f46dcSZachary Turner { 2372b9c1b51eSKate Stone Locker py_lock(python_interpreter, Locker::AcquireLock | 2373b9c1b51eSKate Stone Locker::InitSession | 2374b9c1b51eSKate Stone Locker::NoSTDIN); 2375*05495c5dSJonas Devlieghere ret_val = LLDBSwigPythonBreakpointCallbackFunction( 2376b9c1b51eSKate Stone python_function_name, 2377b9c1b51eSKate Stone python_interpreter->m_dictionary_name.c_str(), stop_frame_sp, 23782c1f46dcSZachary Turner bp_loc_sp); 23792c1f46dcSZachary Turner } 23802c1f46dcSZachary Turner return ret_val; 23812c1f46dcSZachary Turner } 23822c1f46dcSZachary Turner } 23832c1f46dcSZachary Turner } 23842c1f46dcSZachary Turner // We currently always true so we stop in case anything goes wrong when 23852c1f46dcSZachary Turner // trying to call the script function 23862c1f46dcSZachary Turner return true; 23872c1f46dcSZachary Turner } 23882c1f46dcSZachary Turner 2389b9c1b51eSKate Stone bool ScriptInterpreterPython::WatchpointCallbackFunction( 2390b9c1b51eSKate Stone void *baton, StoppointCallbackContext *context, user_id_t watch_id) { 2391b9c1b51eSKate Stone WatchpointOptions::CommandData *wp_option_data = 2392b9c1b51eSKate Stone (WatchpointOptions::CommandData *)baton; 23932c1f46dcSZachary Turner const char *python_function_name = wp_option_data->script_source.c_str(); 23942c1f46dcSZachary Turner 23952c1f46dcSZachary Turner if (!context) 23962c1f46dcSZachary Turner return true; 23972c1f46dcSZachary Turner 23982c1f46dcSZachary Turner ExecutionContext exe_ctx(context->exe_ctx_ref); 23992c1f46dcSZachary Turner Target *target = exe_ctx.GetTargetPtr(); 24002c1f46dcSZachary Turner 24012c1f46dcSZachary Turner if (!target) 24022c1f46dcSZachary Turner return true; 24032c1f46dcSZachary Turner 24042c1f46dcSZachary Turner Debugger &debugger = target->GetDebugger(); 2405b9c1b51eSKate Stone ScriptInterpreter *script_interpreter = 2406b9c1b51eSKate Stone debugger.GetCommandInterpreter().GetScriptInterpreter(); 2407b9c1b51eSKate Stone ScriptInterpreterPython *python_interpreter = 2408b9c1b51eSKate Stone (ScriptInterpreterPython *)script_interpreter; 24092c1f46dcSZachary Turner 24102c1f46dcSZachary Turner if (!script_interpreter) 24112c1f46dcSZachary Turner return true; 24122c1f46dcSZachary Turner 2413b9c1b51eSKate Stone if (python_function_name && python_function_name[0]) { 24142c1f46dcSZachary Turner const StackFrameSP stop_frame_sp(exe_ctx.GetFrameSP()); 24152c1f46dcSZachary Turner WatchpointSP wp_sp = target->GetWatchpointList().FindByID(watch_id); 2416b9c1b51eSKate Stone if (wp_sp) { 2417b9c1b51eSKate Stone if (stop_frame_sp && wp_sp) { 24182c1f46dcSZachary Turner bool ret_val = true; 24192c1f46dcSZachary Turner { 2420b9c1b51eSKate Stone Locker py_lock(python_interpreter, Locker::AcquireLock | 2421b9c1b51eSKate Stone Locker::InitSession | 2422b9c1b51eSKate Stone Locker::NoSTDIN); 2423*05495c5dSJonas Devlieghere ret_val = LLDBSwigPythonWatchpointCallbackFunction( 2424b9c1b51eSKate Stone python_function_name, 2425b9c1b51eSKate Stone python_interpreter->m_dictionary_name.c_str(), stop_frame_sp, 24262c1f46dcSZachary Turner wp_sp); 24272c1f46dcSZachary Turner } 24282c1f46dcSZachary Turner return ret_val; 24292c1f46dcSZachary Turner } 24302c1f46dcSZachary Turner } 24312c1f46dcSZachary Turner } 24322c1f46dcSZachary Turner // We currently always true so we stop in case anything goes wrong when 24332c1f46dcSZachary Turner // trying to call the script function 24342c1f46dcSZachary Turner return true; 24352c1f46dcSZachary Turner } 24362c1f46dcSZachary Turner 2437b9c1b51eSKate Stone size_t ScriptInterpreterPython::CalculateNumChildren( 2438b9c1b51eSKate Stone const StructuredData::ObjectSP &implementor_sp, uint32_t max) { 24392c1f46dcSZachary Turner if (!implementor_sp) 24402c1f46dcSZachary Turner return 0; 24412c1f46dcSZachary Turner StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 24422c1f46dcSZachary Turner if (!generic) 24432c1f46dcSZachary Turner return 0; 24442c1f46dcSZachary Turner void *implementor = generic->GetValue(); 24452c1f46dcSZachary Turner if (!implementor) 24462c1f46dcSZachary Turner return 0; 24472c1f46dcSZachary Turner 24482c1f46dcSZachary Turner size_t ret_val = 0; 24492c1f46dcSZachary Turner 24502c1f46dcSZachary Turner { 2451b9c1b51eSKate Stone Locker py_lock(this, 2452b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2453*05495c5dSJonas Devlieghere ret_val = LLDBSwigPython_CalculateNumChildren(implementor, max); 24542c1f46dcSZachary Turner } 24552c1f46dcSZachary Turner 24562c1f46dcSZachary Turner return ret_val; 24572c1f46dcSZachary Turner } 24582c1f46dcSZachary Turner 2459b9c1b51eSKate Stone lldb::ValueObjectSP ScriptInterpreterPython::GetChildAtIndex( 2460b9c1b51eSKate Stone const StructuredData::ObjectSP &implementor_sp, uint32_t idx) { 24612c1f46dcSZachary Turner if (!implementor_sp) 24622c1f46dcSZachary Turner return lldb::ValueObjectSP(); 24632c1f46dcSZachary Turner 24642c1f46dcSZachary Turner StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 24652c1f46dcSZachary Turner if (!generic) 24662c1f46dcSZachary Turner return lldb::ValueObjectSP(); 24672c1f46dcSZachary Turner void *implementor = generic->GetValue(); 24682c1f46dcSZachary Turner if (!implementor) 24692c1f46dcSZachary Turner return lldb::ValueObjectSP(); 24702c1f46dcSZachary Turner 24712c1f46dcSZachary Turner 24722c1f46dcSZachary Turner lldb::ValueObjectSP ret_val; 24732c1f46dcSZachary Turner { 2474b9c1b51eSKate Stone Locker py_lock(this, 2475b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2476*05495c5dSJonas Devlieghere void *child_ptr = LLDBSwigPython_GetChildAtIndex(implementor, idx); 2477b9c1b51eSKate Stone if (child_ptr != nullptr && child_ptr != Py_None) { 2478b9c1b51eSKate Stone lldb::SBValue *sb_value_ptr = 2479*05495c5dSJonas Devlieghere (lldb::SBValue *)LLDBSWIGPython_CastPyObjectToSBValue(child_ptr); 24802c1f46dcSZachary Turner if (sb_value_ptr == nullptr) 24812c1f46dcSZachary Turner Py_XDECREF(child_ptr); 24822c1f46dcSZachary Turner else 2483*05495c5dSJonas Devlieghere ret_val = LLDBSWIGPython_GetValueObjectSPFromSBValue(sb_value_ptr); 2484b9c1b51eSKate Stone } else { 24852c1f46dcSZachary Turner Py_XDECREF(child_ptr); 24862c1f46dcSZachary Turner } 24872c1f46dcSZachary Turner } 24882c1f46dcSZachary Turner 24892c1f46dcSZachary Turner return ret_val; 24902c1f46dcSZachary Turner } 24912c1f46dcSZachary Turner 2492b9c1b51eSKate Stone int ScriptInterpreterPython::GetIndexOfChildWithName( 2493b9c1b51eSKate Stone const StructuredData::ObjectSP &implementor_sp, const char *child_name) { 24942c1f46dcSZachary Turner if (!implementor_sp) 24952c1f46dcSZachary Turner return UINT32_MAX; 24962c1f46dcSZachary Turner 24972c1f46dcSZachary Turner StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 24982c1f46dcSZachary Turner if (!generic) 24992c1f46dcSZachary Turner return UINT32_MAX; 25002c1f46dcSZachary Turner void *implementor = generic->GetValue(); 25012c1f46dcSZachary Turner if (!implementor) 25022c1f46dcSZachary Turner return UINT32_MAX; 25032c1f46dcSZachary Turner 25042c1f46dcSZachary Turner int ret_val = UINT32_MAX; 25052c1f46dcSZachary Turner 25062c1f46dcSZachary Turner { 2507b9c1b51eSKate Stone Locker py_lock(this, 2508b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2509*05495c5dSJonas Devlieghere ret_val = LLDBSwigPython_GetIndexOfChildWithName(implementor, child_name); 25102c1f46dcSZachary Turner } 25112c1f46dcSZachary Turner 25122c1f46dcSZachary Turner return ret_val; 25132c1f46dcSZachary Turner } 25142c1f46dcSZachary Turner 2515b9c1b51eSKate Stone bool ScriptInterpreterPython::UpdateSynthProviderInstance( 2516b9c1b51eSKate Stone const StructuredData::ObjectSP &implementor_sp) { 25172c1f46dcSZachary Turner bool ret_val = false; 25182c1f46dcSZachary Turner 25192c1f46dcSZachary Turner if (!implementor_sp) 25202c1f46dcSZachary Turner return ret_val; 25212c1f46dcSZachary Turner 25222c1f46dcSZachary Turner StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 25232c1f46dcSZachary Turner if (!generic) 25242c1f46dcSZachary Turner return ret_val; 25252c1f46dcSZachary Turner void *implementor = generic->GetValue(); 25262c1f46dcSZachary Turner if (!implementor) 25272c1f46dcSZachary Turner return ret_val; 25282c1f46dcSZachary Turner 25292c1f46dcSZachary Turner { 2530b9c1b51eSKate Stone Locker py_lock(this, 2531b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2532*05495c5dSJonas Devlieghere ret_val = LLDBSwigPython_UpdateSynthProviderInstance(implementor); 25332c1f46dcSZachary Turner } 25342c1f46dcSZachary Turner 25352c1f46dcSZachary Turner return ret_val; 25362c1f46dcSZachary Turner } 25372c1f46dcSZachary Turner 2538b9c1b51eSKate Stone bool ScriptInterpreterPython::MightHaveChildrenSynthProviderInstance( 2539b9c1b51eSKate Stone const StructuredData::ObjectSP &implementor_sp) { 25402c1f46dcSZachary Turner bool ret_val = false; 25412c1f46dcSZachary Turner 25422c1f46dcSZachary Turner if (!implementor_sp) 25432c1f46dcSZachary Turner return ret_val; 25442c1f46dcSZachary Turner 25452c1f46dcSZachary Turner StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 25462c1f46dcSZachary Turner if (!generic) 25472c1f46dcSZachary Turner return ret_val; 25482c1f46dcSZachary Turner void *implementor = generic->GetValue(); 25492c1f46dcSZachary Turner if (!implementor) 25502c1f46dcSZachary Turner return ret_val; 25512c1f46dcSZachary Turner 25522c1f46dcSZachary Turner 25532c1f46dcSZachary Turner { 2554b9c1b51eSKate Stone Locker py_lock(this, 2555b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2556*05495c5dSJonas Devlieghere ret_val = 2557*05495c5dSJonas Devlieghere LLDBSwigPython_MightHaveChildrenSynthProviderInstance(implementor); 25582c1f46dcSZachary Turner } 25592c1f46dcSZachary Turner 25602c1f46dcSZachary Turner return ret_val; 25612c1f46dcSZachary Turner } 25622c1f46dcSZachary Turner 2563b9c1b51eSKate Stone lldb::ValueObjectSP ScriptInterpreterPython::GetSyntheticValue( 2564b9c1b51eSKate Stone const StructuredData::ObjectSP &implementor_sp) { 25652c1f46dcSZachary Turner lldb::ValueObjectSP ret_val(nullptr); 25662c1f46dcSZachary Turner 25672c1f46dcSZachary Turner if (!implementor_sp) 25682c1f46dcSZachary Turner return ret_val; 25692c1f46dcSZachary Turner 25702c1f46dcSZachary Turner StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 25712c1f46dcSZachary Turner if (!generic) 25722c1f46dcSZachary Turner return ret_val; 25732c1f46dcSZachary Turner void *implementor = generic->GetValue(); 25742c1f46dcSZachary Turner if (!implementor) 25752c1f46dcSZachary Turner return ret_val; 25762c1f46dcSZachary Turner 25772c1f46dcSZachary Turner { 2578b9c1b51eSKate Stone Locker py_lock(this, 2579b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2580*05495c5dSJonas Devlieghere void *child_ptr = LLDBSwigPython_GetValueSynthProviderInstance(implementor); 2581b9c1b51eSKate Stone if (child_ptr != nullptr && child_ptr != Py_None) { 2582b9c1b51eSKate Stone lldb::SBValue *sb_value_ptr = 2583*05495c5dSJonas Devlieghere (lldb::SBValue *)LLDBSWIGPython_CastPyObjectToSBValue(child_ptr); 25842c1f46dcSZachary Turner if (sb_value_ptr == nullptr) 25852c1f46dcSZachary Turner Py_XDECREF(child_ptr); 25862c1f46dcSZachary Turner else 2587*05495c5dSJonas Devlieghere ret_val = LLDBSWIGPython_GetValueObjectSPFromSBValue(sb_value_ptr); 2588b9c1b51eSKate Stone } else { 25892c1f46dcSZachary Turner Py_XDECREF(child_ptr); 25902c1f46dcSZachary Turner } 25912c1f46dcSZachary Turner } 25922c1f46dcSZachary Turner 25932c1f46dcSZachary Turner return ret_val; 25942c1f46dcSZachary Turner } 25952c1f46dcSZachary Turner 2596b9c1b51eSKate Stone ConstString ScriptInterpreterPython::GetSyntheticTypeName( 2597b9c1b51eSKate Stone const StructuredData::ObjectSP &implementor_sp) { 2598b9c1b51eSKate Stone Locker py_lock(this, 2599b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 26006eec8d6cSEnrico Granata 26016eec8d6cSEnrico Granata static char callee_name[] = "get_type_name"; 26026eec8d6cSEnrico Granata 26036eec8d6cSEnrico Granata ConstString ret_val; 26046eec8d6cSEnrico Granata bool got_string = false; 26056eec8d6cSEnrico Granata std::string buffer; 26066eec8d6cSEnrico Granata 26076eec8d6cSEnrico Granata if (!implementor_sp) 26086eec8d6cSEnrico Granata return ret_val; 26096eec8d6cSEnrico Granata 26106eec8d6cSEnrico Granata StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 26116eec8d6cSEnrico Granata if (!generic) 26126eec8d6cSEnrico Granata return ret_val; 2613b9c1b51eSKate Stone PythonObject implementor(PyRefType::Borrowed, 2614b9c1b51eSKate Stone (PyObject *)generic->GetValue()); 26156eec8d6cSEnrico Granata if (!implementor.IsAllocated()) 26166eec8d6cSEnrico Granata return ret_val; 26176eec8d6cSEnrico Granata 2618b9c1b51eSKate Stone PythonObject pmeth(PyRefType::Owned, 2619b9c1b51eSKate Stone PyObject_GetAttrString(implementor.get(), callee_name)); 26206eec8d6cSEnrico Granata 26216eec8d6cSEnrico Granata if (PyErr_Occurred()) 26226eec8d6cSEnrico Granata PyErr_Clear(); 26236eec8d6cSEnrico Granata 26246eec8d6cSEnrico Granata if (!pmeth.IsAllocated()) 26256eec8d6cSEnrico Granata return ret_val; 26266eec8d6cSEnrico Granata 2627b9c1b51eSKate Stone if (PyCallable_Check(pmeth.get()) == 0) { 26286eec8d6cSEnrico Granata if (PyErr_Occurred()) 26296eec8d6cSEnrico Granata PyErr_Clear(); 26306eec8d6cSEnrico Granata return ret_val; 26316eec8d6cSEnrico Granata } 26326eec8d6cSEnrico Granata 26336eec8d6cSEnrico Granata if (PyErr_Occurred()) 26346eec8d6cSEnrico Granata PyErr_Clear(); 26356eec8d6cSEnrico Granata 26366eec8d6cSEnrico Granata // right now we know this function exists and is callable.. 2637b9c1b51eSKate Stone PythonObject py_return( 2638b9c1b51eSKate Stone PyRefType::Owned, 2639b9c1b51eSKate Stone PyObject_CallMethod(implementor.get(), callee_name, nullptr)); 26406eec8d6cSEnrico Granata 26416eec8d6cSEnrico Granata // if it fails, print the error but otherwise go on 2642b9c1b51eSKate Stone if (PyErr_Occurred()) { 26436eec8d6cSEnrico Granata PyErr_Print(); 26446eec8d6cSEnrico Granata PyErr_Clear(); 26456eec8d6cSEnrico Granata } 26466eec8d6cSEnrico Granata 2647b9c1b51eSKate Stone if (py_return.IsAllocated() && PythonString::Check(py_return.get())) { 26486eec8d6cSEnrico Granata PythonString py_string(PyRefType::Borrowed, py_return.get()); 26496eec8d6cSEnrico Granata llvm::StringRef return_data(py_string.GetString()); 2650b9c1b51eSKate Stone if (!return_data.empty()) { 26516eec8d6cSEnrico Granata buffer.assign(return_data.data(), return_data.size()); 26526eec8d6cSEnrico Granata got_string = true; 26536eec8d6cSEnrico Granata } 26546eec8d6cSEnrico Granata } 26556eec8d6cSEnrico Granata 26566eec8d6cSEnrico Granata if (got_string) 26576eec8d6cSEnrico Granata ret_val.SetCStringWithLength(buffer.c_str(), buffer.size()); 26586eec8d6cSEnrico Granata 26596eec8d6cSEnrico Granata return ret_val; 26606eec8d6cSEnrico Granata } 26616eec8d6cSEnrico Granata 2662b9c1b51eSKate Stone bool ScriptInterpreterPython::RunScriptFormatKeyword(const char *impl_function, 26632c1f46dcSZachary Turner Process *process, 26642c1f46dcSZachary Turner std::string &output, 266597206d57SZachary Turner Status &error) { 26662c1f46dcSZachary Turner bool ret_val; 2667b9c1b51eSKate Stone if (!process) { 26682c1f46dcSZachary Turner error.SetErrorString("no process"); 26692c1f46dcSZachary Turner return false; 26702c1f46dcSZachary Turner } 2671b9c1b51eSKate Stone if (!impl_function || !impl_function[0]) { 26722c1f46dcSZachary Turner error.SetErrorString("no function to execute"); 26732c1f46dcSZachary Turner return false; 26742c1f46dcSZachary Turner } 2675*05495c5dSJonas Devlieghere 26762c1f46dcSZachary Turner { 26772c1f46dcSZachary Turner ProcessSP process_sp(process->shared_from_this()); 2678b9c1b51eSKate Stone Locker py_lock(this, 2679b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2680*05495c5dSJonas Devlieghere ret_val = LLDBSWIGPythonRunScriptKeywordProcess( 2681b9c1b51eSKate Stone impl_function, m_dictionary_name.c_str(), process_sp, output); 26822c1f46dcSZachary Turner if (!ret_val) 26832c1f46dcSZachary Turner error.SetErrorString("python script evaluation failed"); 26842c1f46dcSZachary Turner } 26852c1f46dcSZachary Turner return ret_val; 26862c1f46dcSZachary Turner } 26872c1f46dcSZachary Turner 2688b9c1b51eSKate Stone bool ScriptInterpreterPython::RunScriptFormatKeyword(const char *impl_function, 26892c1f46dcSZachary Turner Thread *thread, 26902c1f46dcSZachary Turner std::string &output, 269197206d57SZachary Turner Status &error) { 26922c1f46dcSZachary Turner bool ret_val; 2693b9c1b51eSKate Stone if (!thread) { 26942c1f46dcSZachary Turner error.SetErrorString("no thread"); 26952c1f46dcSZachary Turner return false; 26962c1f46dcSZachary Turner } 2697b9c1b51eSKate Stone if (!impl_function || !impl_function[0]) { 26982c1f46dcSZachary Turner error.SetErrorString("no function to execute"); 26992c1f46dcSZachary Turner return false; 27002c1f46dcSZachary Turner } 2701*05495c5dSJonas Devlieghere 27022c1f46dcSZachary Turner { 27032c1f46dcSZachary Turner ThreadSP thread_sp(thread->shared_from_this()); 2704b9c1b51eSKate Stone Locker py_lock(this, 2705b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2706*05495c5dSJonas Devlieghere ret_val = LLDBSWIGPythonRunScriptKeywordThread( 2707b9c1b51eSKate Stone impl_function, m_dictionary_name.c_str(), thread_sp, output); 27082c1f46dcSZachary Turner if (!ret_val) 27092c1f46dcSZachary Turner error.SetErrorString("python script evaluation failed"); 27102c1f46dcSZachary Turner } 27112c1f46dcSZachary Turner return ret_val; 27122c1f46dcSZachary Turner } 27132c1f46dcSZachary Turner 2714b9c1b51eSKate Stone bool ScriptInterpreterPython::RunScriptFormatKeyword(const char *impl_function, 27152c1f46dcSZachary Turner Target *target, 27162c1f46dcSZachary Turner std::string &output, 271797206d57SZachary Turner Status &error) { 27182c1f46dcSZachary Turner bool ret_val; 2719b9c1b51eSKate Stone if (!target) { 27202c1f46dcSZachary Turner error.SetErrorString("no thread"); 27212c1f46dcSZachary Turner return false; 27222c1f46dcSZachary Turner } 2723b9c1b51eSKate Stone if (!impl_function || !impl_function[0]) { 27242c1f46dcSZachary Turner error.SetErrorString("no function to execute"); 27252c1f46dcSZachary Turner return false; 27262c1f46dcSZachary Turner } 2727*05495c5dSJonas Devlieghere 27282c1f46dcSZachary Turner { 27292c1f46dcSZachary Turner TargetSP target_sp(target->shared_from_this()); 2730b9c1b51eSKate Stone Locker py_lock(this, 2731b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2732*05495c5dSJonas Devlieghere ret_val = LLDBSWIGPythonRunScriptKeywordTarget( 2733b9c1b51eSKate Stone impl_function, m_dictionary_name.c_str(), target_sp, output); 27342c1f46dcSZachary Turner if (!ret_val) 27352c1f46dcSZachary Turner error.SetErrorString("python script evaluation failed"); 27362c1f46dcSZachary Turner } 27372c1f46dcSZachary Turner return ret_val; 27382c1f46dcSZachary Turner } 27392c1f46dcSZachary Turner 2740b9c1b51eSKate Stone bool ScriptInterpreterPython::RunScriptFormatKeyword(const char *impl_function, 27412c1f46dcSZachary Turner StackFrame *frame, 27422c1f46dcSZachary Turner std::string &output, 274397206d57SZachary Turner Status &error) { 27442c1f46dcSZachary Turner bool ret_val; 2745b9c1b51eSKate Stone if (!frame) { 27462c1f46dcSZachary Turner error.SetErrorString("no frame"); 27472c1f46dcSZachary Turner return false; 27482c1f46dcSZachary Turner } 2749b9c1b51eSKate Stone if (!impl_function || !impl_function[0]) { 27502c1f46dcSZachary Turner error.SetErrorString("no function to execute"); 27512c1f46dcSZachary Turner return false; 27522c1f46dcSZachary Turner } 2753*05495c5dSJonas Devlieghere 27542c1f46dcSZachary Turner { 27552c1f46dcSZachary Turner StackFrameSP frame_sp(frame->shared_from_this()); 2756b9c1b51eSKate Stone Locker py_lock(this, 2757b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2758*05495c5dSJonas Devlieghere ret_val = LLDBSWIGPythonRunScriptKeywordFrame( 2759b9c1b51eSKate Stone impl_function, m_dictionary_name.c_str(), frame_sp, output); 27602c1f46dcSZachary Turner if (!ret_val) 27612c1f46dcSZachary Turner error.SetErrorString("python script evaluation failed"); 27622c1f46dcSZachary Turner } 27632c1f46dcSZachary Turner return ret_val; 27642c1f46dcSZachary Turner } 27652c1f46dcSZachary Turner 2766b9c1b51eSKate Stone bool ScriptInterpreterPython::RunScriptFormatKeyword(const char *impl_function, 27672c1f46dcSZachary Turner ValueObject *value, 27682c1f46dcSZachary Turner std::string &output, 276997206d57SZachary Turner Status &error) { 27702c1f46dcSZachary Turner bool ret_val; 2771b9c1b51eSKate Stone if (!value) { 27722c1f46dcSZachary Turner error.SetErrorString("no value"); 27732c1f46dcSZachary Turner return false; 27742c1f46dcSZachary Turner } 2775b9c1b51eSKate Stone if (!impl_function || !impl_function[0]) { 27762c1f46dcSZachary Turner error.SetErrorString("no function to execute"); 27772c1f46dcSZachary Turner return false; 27782c1f46dcSZachary Turner } 2779*05495c5dSJonas Devlieghere 27802c1f46dcSZachary Turner { 27812c1f46dcSZachary Turner ValueObjectSP value_sp(value->GetSP()); 2782b9c1b51eSKate Stone Locker py_lock(this, 2783b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2784*05495c5dSJonas Devlieghere ret_val = LLDBSWIGPythonRunScriptKeywordValue( 2785b9c1b51eSKate Stone impl_function, m_dictionary_name.c_str(), value_sp, output); 27862c1f46dcSZachary Turner if (!ret_val) 27872c1f46dcSZachary Turner error.SetErrorString("python script evaluation failed"); 27882c1f46dcSZachary Turner } 27892c1f46dcSZachary Turner return ret_val; 27902c1f46dcSZachary Turner } 27912c1f46dcSZachary Turner 2792b9c1b51eSKate Stone uint64_t replace_all(std::string &str, const std::string &oldStr, 2793b9c1b51eSKate Stone const std::string &newStr) { 27942c1f46dcSZachary Turner size_t pos = 0; 27952c1f46dcSZachary Turner uint64_t matches = 0; 2796b9c1b51eSKate Stone while ((pos = str.find(oldStr, pos)) != std::string::npos) { 27972c1f46dcSZachary Turner matches++; 27982c1f46dcSZachary Turner str.replace(pos, oldStr.length(), newStr); 27992c1f46dcSZachary Turner pos += newStr.length(); 28002c1f46dcSZachary Turner } 28012c1f46dcSZachary Turner return matches; 28022c1f46dcSZachary Turner } 28032c1f46dcSZachary Turner 2804b9c1b51eSKate Stone bool ScriptInterpreterPython::LoadScriptingModule( 2805b9c1b51eSKate Stone const char *pathname, bool can_reload, bool init_session, 280697206d57SZachary Turner lldb_private::Status &error, StructuredData::ObjectSP *module_sp) { 2807b9c1b51eSKate Stone if (!pathname || !pathname[0]) { 28082c1f46dcSZachary Turner error.SetErrorString("invalid pathname"); 28092c1f46dcSZachary Turner return false; 28102c1f46dcSZachary Turner } 28112c1f46dcSZachary Turner 28122c1f46dcSZachary Turner lldb::DebuggerSP debugger_sp = m_interpreter.GetDebugger().shared_from_this(); 28132c1f46dcSZachary Turner 28142c1f46dcSZachary Turner { 28158f3be7a3SJonas Devlieghere FileSpec target_file(pathname); 28168f3be7a3SJonas Devlieghere FileSystem::Instance().Resolve(target_file); 28172c1f46dcSZachary Turner std::string basename(target_file.GetFilename().GetCString()); 28182c1f46dcSZachary Turner 28192c1f46dcSZachary Turner StreamString command_stream; 28202c1f46dcSZachary Turner 2821f9f36097SAdrian McCarthy // Before executing Python code, lock the GIL. 2822b9c1b51eSKate Stone Locker py_lock(this, Locker::AcquireLock | 2823b9c1b51eSKate Stone (init_session ? Locker::InitSession : 0) | 2824b9c1b51eSKate Stone Locker::NoSTDIN, 2825b9c1b51eSKate Stone Locker::FreeAcquiredLock | 2826b9c1b51eSKate Stone (init_session ? Locker::TearDownSession : 0)); 28277d86ee5aSZachary Turner namespace fs = llvm::sys::fs; 28287d86ee5aSZachary Turner fs::file_status st; 28297d86ee5aSZachary Turner std::error_code ec = status(target_file.GetPath(), st); 28302c1f46dcSZachary Turner 28317d86ee5aSZachary Turner if (ec || st.type() == fs::file_type::status_error || 28327d86ee5aSZachary Turner st.type() == fs::file_type::type_unknown || 28337d86ee5aSZachary Turner st.type() == fs::file_type::file_not_found) { 28342c1f46dcSZachary Turner // if not a valid file of any sort, check if it might be a filename still 28352c1f46dcSZachary Turner // dot can't be used but / and \ can, and if either is found, reject 2836b9c1b51eSKate Stone if (strchr(pathname, '\\') || strchr(pathname, '/')) { 28372c1f46dcSZachary Turner error.SetErrorString("invalid pathname"); 28382c1f46dcSZachary Turner return false; 28392c1f46dcSZachary Turner } 2840b9c1b51eSKate Stone basename = pathname; // not a filename, probably a package of some sort, 2841b9c1b51eSKate Stone // let it go through 28427d86ee5aSZachary Turner } else if (is_directory(st) || is_regular_file(st)) { 284342a9da7bSStefan Granitz if (target_file.GetDirectory().IsEmpty()) { 284442a9da7bSStefan Granitz error.SetErrorString("invalid directory name"); 284542a9da7bSStefan Granitz return false; 284642a9da7bSStefan Granitz } 284742a9da7bSStefan Granitz 2848f9f36097SAdrian McCarthy std::string directory = target_file.GetDirectory().GetCString(); 2849f9f36097SAdrian McCarthy replace_all(directory, "\\", "\\\\"); 28502c1f46dcSZachary Turner replace_all(directory, "'", "\\'"); 28512c1f46dcSZachary Turner 28522c1f46dcSZachary Turner // now make sure that Python has "directory" in the search path 28532c1f46dcSZachary Turner StreamString command_stream; 2854b9c1b51eSKate Stone command_stream.Printf("if not (sys.path.__contains__('%s')):\n " 2855b9c1b51eSKate Stone "sys.path.insert(1,'%s');\n\n", 2856b9c1b51eSKate Stone directory.c_str(), directory.c_str()); 2857b9c1b51eSKate Stone bool syspath_retval = 2858b9c1b51eSKate Stone ExecuteMultipleLines(command_stream.GetData(), 2859b9c1b51eSKate Stone ScriptInterpreter::ExecuteScriptOptions() 2860b9c1b51eSKate Stone .SetEnableIO(false) 2861b9c1b51eSKate Stone .SetSetLLDBGlobals(false)) 2862b9c1b51eSKate Stone .Success(); 2863b9c1b51eSKate Stone if (!syspath_retval) { 28642c1f46dcSZachary Turner error.SetErrorString("Python sys.path handling failed"); 28652c1f46dcSZachary Turner return false; 28662c1f46dcSZachary Turner } 28672c1f46dcSZachary Turner 28682c1f46dcSZachary Turner // strip .py or .pyc extension 28692c1f46dcSZachary Turner ConstString extension = target_file.GetFileNameExtension(); 2870b9c1b51eSKate Stone if (extension) { 2871ad8d48f9SJonas Devlieghere if (llvm::StringRef(extension.GetCString()) == ".py") 28722c1f46dcSZachary Turner basename.resize(basename.length() - 3); 2873ad8d48f9SJonas Devlieghere else if (llvm::StringRef(extension.GetCString()) == ".pyc") 28742c1f46dcSZachary Turner basename.resize(basename.length() - 4); 28752c1f46dcSZachary Turner } 2876b9c1b51eSKate Stone } else { 28772c1f46dcSZachary Turner error.SetErrorString("no known way to import this module specification"); 28782c1f46dcSZachary Turner return false; 28792c1f46dcSZachary Turner } 28802c1f46dcSZachary Turner 28812c1f46dcSZachary Turner // check if the module is already import-ed 28822c1f46dcSZachary Turner command_stream.Clear(); 28832c1f46dcSZachary Turner command_stream.Printf("sys.modules.__contains__('%s')", basename.c_str()); 28842c1f46dcSZachary Turner bool does_contain = false; 288505097246SAdrian Prantl // this call will succeed if the module was ever imported in any Debugger 288605097246SAdrian Prantl // in the lifetime of the process in which this LLDB framework is living 2887b9c1b51eSKate Stone bool was_imported_globally = 2888b9c1b51eSKate Stone (ExecuteOneLineWithReturn( 2889b9c1b51eSKate Stone command_stream.GetData(), 2890b9c1b51eSKate Stone ScriptInterpreterPython::eScriptReturnTypeBool, &does_contain, 2891b9c1b51eSKate Stone ScriptInterpreter::ExecuteScriptOptions() 2892b9c1b51eSKate Stone .SetEnableIO(false) 2893b9c1b51eSKate Stone .SetSetLLDBGlobals(false)) && 2894b9c1b51eSKate Stone does_contain); 2895b9c1b51eSKate Stone // this call will fail if the module was not imported in this Debugger 2896b9c1b51eSKate Stone // before 28972c1f46dcSZachary Turner command_stream.Clear(); 28982c1f46dcSZachary Turner command_stream.Printf("sys.getrefcount(%s)", basename.c_str()); 2899b9c1b51eSKate Stone bool was_imported_locally = GetSessionDictionary() 2900b9c1b51eSKate Stone .GetItemForKey(PythonString(basename)) 2901b9c1b51eSKate Stone .IsAllocated(); 29022c1f46dcSZachary Turner 29032c1f46dcSZachary Turner bool was_imported = (was_imported_globally || was_imported_locally); 29042c1f46dcSZachary Turner 2905a6682a41SJonas Devlieghere if (was_imported && !can_reload) { 29062c1f46dcSZachary Turner error.SetErrorString("module already imported"); 29072c1f46dcSZachary Turner return false; 29082c1f46dcSZachary Turner } 29092c1f46dcSZachary Turner 29102c1f46dcSZachary Turner // now actually do the import 29112c1f46dcSZachary Turner command_stream.Clear(); 29122c1f46dcSZachary Turner 2913b9c1b51eSKate Stone if (was_imported) { 29142c1f46dcSZachary Turner if (!was_imported_locally) 2915b9c1b51eSKate Stone command_stream.Printf("import %s ; reload_module(%s)", basename.c_str(), 2916b9c1b51eSKate Stone basename.c_str()); 29172c1f46dcSZachary Turner else 29187d2d0984SZachary Turner command_stream.Printf("reload_module(%s)", basename.c_str()); 2919b9c1b51eSKate Stone } else 29202c1f46dcSZachary Turner command_stream.Printf("import %s", basename.c_str()); 29212c1f46dcSZachary Turner 2922b9c1b51eSKate Stone error = ExecuteMultipleLines(command_stream.GetData(), 2923b9c1b51eSKate Stone ScriptInterpreter::ExecuteScriptOptions() 2924b9c1b51eSKate Stone .SetEnableIO(false) 2925b9c1b51eSKate Stone .SetSetLLDBGlobals(false)); 29262c1f46dcSZachary Turner if (error.Fail()) 29272c1f46dcSZachary Turner return false; 29282c1f46dcSZachary Turner 29292c1f46dcSZachary Turner // if we are here, everything worked 29302c1f46dcSZachary Turner // call __lldb_init_module(debugger,dict) 2931*05495c5dSJonas Devlieghere if (!LLDBSwigPythonCallModuleInit(basename.c_str(), 2932*05495c5dSJonas Devlieghere m_dictionary_name.c_str(), debugger_sp)) { 29332c1f46dcSZachary Turner error.SetErrorString("calling __lldb_init_module failed"); 29342c1f46dcSZachary Turner return false; 29352c1f46dcSZachary Turner } 29362c1f46dcSZachary Turner 2937b9c1b51eSKate Stone if (module_sp) { 29382c1f46dcSZachary Turner // everything went just great, now set the module object 29392c1f46dcSZachary Turner command_stream.Clear(); 29402c1f46dcSZachary Turner command_stream.Printf("%s", basename.c_str()); 29412c1f46dcSZachary Turner void *module_pyobj = nullptr; 2942b9c1b51eSKate Stone if (ExecuteOneLineWithReturn( 2943b9c1b51eSKate Stone command_stream.GetData(), 2944b9c1b51eSKate Stone ScriptInterpreter::eScriptReturnTypeOpaqueObject, 2945b9c1b51eSKate Stone &module_pyobj) && 2946b9c1b51eSKate Stone module_pyobj) 2947796ac80bSJonas Devlieghere *module_sp = std::make_shared<StructuredPythonObject>(module_pyobj); 29482c1f46dcSZachary Turner } 29492c1f46dcSZachary Turner 29502c1f46dcSZachary Turner return true; 29512c1f46dcSZachary Turner } 29522c1f46dcSZachary Turner } 29532c1f46dcSZachary Turner 2954b9c1b51eSKate Stone bool ScriptInterpreterPython::IsReservedWord(const char *word) { 29552c1f46dcSZachary Turner if (!word || !word[0]) 29562c1f46dcSZachary Turner return false; 29572c1f46dcSZachary Turner 29582c1f46dcSZachary Turner llvm::StringRef word_sr(word); 29592c1f46dcSZachary Turner 296005097246SAdrian Prantl // filter out a few characters that would just confuse us and that are 296105097246SAdrian Prantl // clearly not keyword material anyway 29622c1f46dcSZachary Turner if (word_sr.find_first_of("'\"") != llvm::StringRef::npos) 29632c1f46dcSZachary Turner return false; 29642c1f46dcSZachary Turner 29652c1f46dcSZachary Turner StreamString command_stream; 29662c1f46dcSZachary Turner command_stream.Printf("keyword.iskeyword('%s')", word); 29672c1f46dcSZachary Turner bool result; 29682c1f46dcSZachary Turner ExecuteScriptOptions options; 29692c1f46dcSZachary Turner options.SetEnableIO(false); 29702c1f46dcSZachary Turner options.SetMaskoutErrors(true); 29712c1f46dcSZachary Turner options.SetSetLLDBGlobals(false); 2972b9c1b51eSKate Stone if (ExecuteOneLineWithReturn(command_stream.GetData(), 2973b9c1b51eSKate Stone ScriptInterpreter::eScriptReturnTypeBool, 2974b9c1b51eSKate Stone &result, options)) 29752c1f46dcSZachary Turner return result; 29762c1f46dcSZachary Turner return false; 29772c1f46dcSZachary Turner } 29782c1f46dcSZachary Turner 2979b9c1b51eSKate Stone ScriptInterpreterPython::SynchronicityHandler::SynchronicityHandler( 2980b9c1b51eSKate Stone lldb::DebuggerSP debugger_sp, ScriptedCommandSynchronicity synchro) 2981b9c1b51eSKate Stone : m_debugger_sp(debugger_sp), m_synch_wanted(synchro), 2982b9c1b51eSKate Stone m_old_asynch(debugger_sp->GetAsyncExecution()) { 29832c1f46dcSZachary Turner if (m_synch_wanted == eScriptedCommandSynchronicitySynchronous) 29842c1f46dcSZachary Turner m_debugger_sp->SetAsyncExecution(false); 29852c1f46dcSZachary Turner else if (m_synch_wanted == eScriptedCommandSynchronicityAsynchronous) 29862c1f46dcSZachary Turner m_debugger_sp->SetAsyncExecution(true); 29872c1f46dcSZachary Turner } 29882c1f46dcSZachary Turner 2989b9c1b51eSKate Stone ScriptInterpreterPython::SynchronicityHandler::~SynchronicityHandler() { 29902c1f46dcSZachary Turner if (m_synch_wanted != eScriptedCommandSynchronicityCurrentValue) 29912c1f46dcSZachary Turner m_debugger_sp->SetAsyncExecution(m_old_asynch); 29922c1f46dcSZachary Turner } 29932c1f46dcSZachary Turner 2994b9c1b51eSKate Stone bool ScriptInterpreterPython::RunScriptBasedCommand( 29954d51a902SRaphael Isemann const char *impl_function, llvm::StringRef args, 29962c1f46dcSZachary Turner ScriptedCommandSynchronicity synchronicity, 299797206d57SZachary Turner lldb_private::CommandReturnObject &cmd_retobj, Status &error, 2998b9c1b51eSKate Stone const lldb_private::ExecutionContext &exe_ctx) { 2999b9c1b51eSKate Stone if (!impl_function) { 30002c1f46dcSZachary Turner error.SetErrorString("no function to execute"); 30012c1f46dcSZachary Turner return false; 30022c1f46dcSZachary Turner } 30032c1f46dcSZachary Turner 30042c1f46dcSZachary Turner lldb::DebuggerSP debugger_sp = m_interpreter.GetDebugger().shared_from_this(); 30052c1f46dcSZachary Turner lldb::ExecutionContextRefSP exe_ctx_ref_sp(new ExecutionContextRef(exe_ctx)); 30062c1f46dcSZachary Turner 3007b9c1b51eSKate Stone if (!debugger_sp.get()) { 30082c1f46dcSZachary Turner error.SetErrorString("invalid Debugger pointer"); 30092c1f46dcSZachary Turner return false; 30102c1f46dcSZachary Turner } 30112c1f46dcSZachary Turner 30122c1f46dcSZachary Turner bool ret_val = false; 30132c1f46dcSZachary Turner 30142c1f46dcSZachary Turner std::string err_msg; 30152c1f46dcSZachary Turner 30162c1f46dcSZachary Turner { 30172c1f46dcSZachary Turner Locker py_lock(this, 3018b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | 3019b9c1b51eSKate Stone (cmd_retobj.GetInteractive() ? 0 : Locker::NoSTDIN), 30202c1f46dcSZachary Turner Locker::FreeLock | Locker::TearDownSession); 30212c1f46dcSZachary Turner 3022b9c1b51eSKate Stone SynchronicityHandler synch_handler(debugger_sp, synchronicity); 30232c1f46dcSZachary Turner 30244d51a902SRaphael Isemann std::string args_str = args.str(); 3025*05495c5dSJonas Devlieghere ret_val = LLDBSwigPythonCallCommand( 3026*05495c5dSJonas Devlieghere impl_function, m_dictionary_name.c_str(), debugger_sp, args_str.c_str(), 3027*05495c5dSJonas Devlieghere cmd_retobj, exe_ctx_ref_sp); 30282c1f46dcSZachary Turner } 30292c1f46dcSZachary Turner 30302c1f46dcSZachary Turner if (!ret_val) 30312c1f46dcSZachary Turner error.SetErrorString("unable to execute script function"); 30322c1f46dcSZachary Turner else 30332c1f46dcSZachary Turner error.Clear(); 30342c1f46dcSZachary Turner 30352c1f46dcSZachary Turner return ret_val; 30362c1f46dcSZachary Turner } 30372c1f46dcSZachary Turner 3038b9c1b51eSKate Stone bool ScriptInterpreterPython::RunScriptBasedCommand( 30394d51a902SRaphael Isemann StructuredData::GenericSP impl_obj_sp, llvm::StringRef args, 30402c1f46dcSZachary Turner ScriptedCommandSynchronicity synchronicity, 304197206d57SZachary Turner lldb_private::CommandReturnObject &cmd_retobj, Status &error, 3042b9c1b51eSKate Stone const lldb_private::ExecutionContext &exe_ctx) { 3043b9c1b51eSKate Stone if (!impl_obj_sp || !impl_obj_sp->IsValid()) { 30442c1f46dcSZachary Turner error.SetErrorString("no function to execute"); 30452c1f46dcSZachary Turner return false; 30462c1f46dcSZachary Turner } 30472c1f46dcSZachary Turner 30482c1f46dcSZachary Turner lldb::DebuggerSP debugger_sp = m_interpreter.GetDebugger().shared_from_this(); 30492c1f46dcSZachary Turner lldb::ExecutionContextRefSP exe_ctx_ref_sp(new ExecutionContextRef(exe_ctx)); 30502c1f46dcSZachary Turner 3051b9c1b51eSKate Stone if (!debugger_sp.get()) { 30522c1f46dcSZachary Turner error.SetErrorString("invalid Debugger pointer"); 30532c1f46dcSZachary Turner return false; 30542c1f46dcSZachary Turner } 30552c1f46dcSZachary Turner 30562c1f46dcSZachary Turner bool ret_val = false; 30572c1f46dcSZachary Turner 30582c1f46dcSZachary Turner std::string err_msg; 30592c1f46dcSZachary Turner 30602c1f46dcSZachary Turner { 30612c1f46dcSZachary Turner Locker py_lock(this, 3062b9c1b51eSKate Stone Locker::AcquireLock | Locker::InitSession | 3063b9c1b51eSKate Stone (cmd_retobj.GetInteractive() ? 0 : Locker::NoSTDIN), 30642c1f46dcSZachary Turner Locker::FreeLock | Locker::TearDownSession); 30652c1f46dcSZachary Turner 3066b9c1b51eSKate Stone SynchronicityHandler synch_handler(debugger_sp, synchronicity); 30672c1f46dcSZachary Turner 30684d51a902SRaphael Isemann std::string args_str = args.str(); 3069*05495c5dSJonas Devlieghere ret_val = LLDBSwigPythonCallCommandObject(impl_obj_sp->GetValue(), 3070*05495c5dSJonas Devlieghere debugger_sp, args_str.c_str(), 3071*05495c5dSJonas Devlieghere cmd_retobj, exe_ctx_ref_sp); 30722c1f46dcSZachary Turner } 30732c1f46dcSZachary Turner 30742c1f46dcSZachary Turner if (!ret_val) 30752c1f46dcSZachary Turner error.SetErrorString("unable to execute script function"); 30762c1f46dcSZachary Turner else 30772c1f46dcSZachary Turner error.Clear(); 30782c1f46dcSZachary Turner 30792c1f46dcSZachary Turner return ret_val; 30802c1f46dcSZachary Turner } 30812c1f46dcSZachary Turner 308205097246SAdrian Prantl // in Python, a special attribute __doc__ contains the docstring for an object 308305097246SAdrian Prantl // (function, method, class, ...) if any is defined Otherwise, the attribute's 308405097246SAdrian Prantl // value is None 3085b9c1b51eSKate Stone bool ScriptInterpreterPython::GetDocumentationForItem(const char *item, 3086b9c1b51eSKate Stone std::string &dest) { 30872c1f46dcSZachary Turner dest.clear(); 30882c1f46dcSZachary Turner if (!item || !*item) 30892c1f46dcSZachary Turner return false; 30902c1f46dcSZachary Turner std::string command(item); 30912c1f46dcSZachary Turner command += ".__doc__"; 30922c1f46dcSZachary Turner 3093b9c1b51eSKate Stone char *result_ptr = nullptr; // Python is going to point this to valid data if 3094b9c1b51eSKate Stone // ExecuteOneLineWithReturn returns successfully 30952c1f46dcSZachary Turner 3096b9c1b51eSKate Stone if (ExecuteOneLineWithReturn( 3097b9c1b51eSKate Stone command.c_str(), ScriptInterpreter::eScriptReturnTypeCharStrOrNone, 30982c1f46dcSZachary Turner &result_ptr, 3099b9c1b51eSKate Stone ScriptInterpreter::ExecuteScriptOptions().SetEnableIO(false))) { 31002c1f46dcSZachary Turner if (result_ptr) 31012c1f46dcSZachary Turner dest.assign(result_ptr); 31022c1f46dcSZachary Turner return true; 3103b9c1b51eSKate Stone } else { 31042c1f46dcSZachary Turner StreamString str_stream; 3105b9c1b51eSKate Stone str_stream.Printf( 3106b9c1b51eSKate Stone "Function %s was not found. Containing module might be missing.", item); 3107c156427dSZachary Turner dest = str_stream.GetString(); 31082c1f46dcSZachary Turner return false; 31092c1f46dcSZachary Turner } 31102c1f46dcSZachary Turner } 31112c1f46dcSZachary Turner 3112b9c1b51eSKate Stone bool ScriptInterpreterPython::GetShortHelpForCommandObject( 3113b9c1b51eSKate Stone StructuredData::GenericSP cmd_obj_sp, std::string &dest) { 31142c1f46dcSZachary Turner bool got_string = false; 31152c1f46dcSZachary Turner dest.clear(); 31162c1f46dcSZachary Turner 3117b9c1b51eSKate Stone Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 31182c1f46dcSZachary Turner 31192c1f46dcSZachary Turner static char callee_name[] = "get_short_help"; 31202c1f46dcSZachary Turner 31212c1f46dcSZachary Turner if (!cmd_obj_sp) 31222c1f46dcSZachary Turner return false; 31232c1f46dcSZachary Turner 3124b9c1b51eSKate Stone PythonObject implementor(PyRefType::Borrowed, 3125b9c1b51eSKate Stone (PyObject *)cmd_obj_sp->GetValue()); 31262c1f46dcSZachary Turner 3127f8b22f8fSZachary Turner if (!implementor.IsAllocated()) 31282c1f46dcSZachary Turner return false; 31292c1f46dcSZachary Turner 3130b9c1b51eSKate Stone PythonObject pmeth(PyRefType::Owned, 3131b9c1b51eSKate Stone PyObject_GetAttrString(implementor.get(), callee_name)); 31322c1f46dcSZachary Turner 31332c1f46dcSZachary Turner if (PyErr_Occurred()) 31342c1f46dcSZachary Turner PyErr_Clear(); 31352c1f46dcSZachary Turner 3136f8b22f8fSZachary Turner if (!pmeth.IsAllocated()) 31372c1f46dcSZachary Turner return false; 31382c1f46dcSZachary Turner 3139b9c1b51eSKate Stone if (PyCallable_Check(pmeth.get()) == 0) { 31402c1f46dcSZachary Turner if (PyErr_Occurred()) 31412c1f46dcSZachary Turner PyErr_Clear(); 31422c1f46dcSZachary Turner return false; 31432c1f46dcSZachary Turner } 31442c1f46dcSZachary Turner 31452c1f46dcSZachary Turner if (PyErr_Occurred()) 31462c1f46dcSZachary Turner PyErr_Clear(); 31472c1f46dcSZachary Turner 31482c1f46dcSZachary Turner // right now we know this function exists and is callable.. 3149b9c1b51eSKate Stone PythonObject py_return( 3150b9c1b51eSKate Stone PyRefType::Owned, 3151b9c1b51eSKate Stone PyObject_CallMethod(implementor.get(), callee_name, nullptr)); 31522c1f46dcSZachary Turner 31532c1f46dcSZachary Turner // if it fails, print the error but otherwise go on 3154b9c1b51eSKate Stone if (PyErr_Occurred()) { 31552c1f46dcSZachary Turner PyErr_Print(); 31562c1f46dcSZachary Turner PyErr_Clear(); 31572c1f46dcSZachary Turner } 31582c1f46dcSZachary Turner 3159b9c1b51eSKate Stone if (py_return.IsAllocated() && PythonString::Check(py_return.get())) { 3160f8b22f8fSZachary Turner PythonString py_string(PyRefType::Borrowed, py_return.get()); 316122c8efcdSZachary Turner llvm::StringRef return_data(py_string.GetString()); 316222c8efcdSZachary Turner dest.assign(return_data.data(), return_data.size()); 31632c1f46dcSZachary Turner got_string = true; 31642c1f46dcSZachary Turner } 31652c1f46dcSZachary Turner return got_string; 31662c1f46dcSZachary Turner } 31672c1f46dcSZachary Turner 3168b9c1b51eSKate Stone uint32_t ScriptInterpreterPython::GetFlagsForCommandObject( 3169b9c1b51eSKate Stone StructuredData::GenericSP cmd_obj_sp) { 31702c1f46dcSZachary Turner uint32_t result = 0; 31712c1f46dcSZachary Turner 3172b9c1b51eSKate Stone Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 31732c1f46dcSZachary Turner 31742c1f46dcSZachary Turner static char callee_name[] = "get_flags"; 31752c1f46dcSZachary Turner 31762c1f46dcSZachary Turner if (!cmd_obj_sp) 31772c1f46dcSZachary Turner return result; 31782c1f46dcSZachary Turner 3179b9c1b51eSKate Stone PythonObject implementor(PyRefType::Borrowed, 3180b9c1b51eSKate Stone (PyObject *)cmd_obj_sp->GetValue()); 31812c1f46dcSZachary Turner 3182f8b22f8fSZachary Turner if (!implementor.IsAllocated()) 31832c1f46dcSZachary Turner return result; 31842c1f46dcSZachary Turner 3185b9c1b51eSKate Stone PythonObject pmeth(PyRefType::Owned, 3186b9c1b51eSKate Stone PyObject_GetAttrString(implementor.get(), callee_name)); 31872c1f46dcSZachary Turner 31882c1f46dcSZachary Turner if (PyErr_Occurred()) 31892c1f46dcSZachary Turner PyErr_Clear(); 31902c1f46dcSZachary Turner 3191f8b22f8fSZachary Turner if (!pmeth.IsAllocated()) 31922c1f46dcSZachary Turner return result; 31932c1f46dcSZachary Turner 3194b9c1b51eSKate Stone if (PyCallable_Check(pmeth.get()) == 0) { 31952c1f46dcSZachary Turner if (PyErr_Occurred()) 31962c1f46dcSZachary Turner PyErr_Clear(); 31972c1f46dcSZachary Turner return result; 31982c1f46dcSZachary Turner } 31992c1f46dcSZachary Turner 32002c1f46dcSZachary Turner if (PyErr_Occurred()) 32012c1f46dcSZachary Turner PyErr_Clear(); 32022c1f46dcSZachary Turner 32032c1f46dcSZachary Turner // right now we know this function exists and is callable.. 3204b9c1b51eSKate Stone PythonObject py_return( 3205b9c1b51eSKate Stone PyRefType::Owned, 3206b9c1b51eSKate Stone PyObject_CallMethod(implementor.get(), callee_name, nullptr)); 32072c1f46dcSZachary Turner 32082c1f46dcSZachary Turner // if it fails, print the error but otherwise go on 3209b9c1b51eSKate Stone if (PyErr_Occurred()) { 32102c1f46dcSZachary Turner PyErr_Print(); 32112c1f46dcSZachary Turner PyErr_Clear(); 32122c1f46dcSZachary Turner } 32132c1f46dcSZachary Turner 3214b9c1b51eSKate Stone if (py_return.IsAllocated() && PythonInteger::Check(py_return.get())) { 3215f8b22f8fSZachary Turner PythonInteger int_value(PyRefType::Borrowed, py_return.get()); 321622c8efcdSZachary Turner result = int_value.GetInteger(); 32172c1f46dcSZachary Turner } 32182c1f46dcSZachary Turner 32192c1f46dcSZachary Turner return result; 32202c1f46dcSZachary Turner } 32212c1f46dcSZachary Turner 3222b9c1b51eSKate Stone bool ScriptInterpreterPython::GetLongHelpForCommandObject( 3223b9c1b51eSKate Stone StructuredData::GenericSP cmd_obj_sp, std::string &dest) { 32242c1f46dcSZachary Turner bool got_string = false; 32252c1f46dcSZachary Turner dest.clear(); 32262c1f46dcSZachary Turner 3227b9c1b51eSKate Stone Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 32282c1f46dcSZachary Turner 32292c1f46dcSZachary Turner static char callee_name[] = "get_long_help"; 32302c1f46dcSZachary Turner 32312c1f46dcSZachary Turner if (!cmd_obj_sp) 32322c1f46dcSZachary Turner return false; 32332c1f46dcSZachary Turner 3234b9c1b51eSKate Stone PythonObject implementor(PyRefType::Borrowed, 3235b9c1b51eSKate Stone (PyObject *)cmd_obj_sp->GetValue()); 32362c1f46dcSZachary Turner 3237f8b22f8fSZachary Turner if (!implementor.IsAllocated()) 32382c1f46dcSZachary Turner return false; 32392c1f46dcSZachary Turner 3240b9c1b51eSKate Stone PythonObject pmeth(PyRefType::Owned, 3241b9c1b51eSKate Stone PyObject_GetAttrString(implementor.get(), callee_name)); 32422c1f46dcSZachary Turner 32432c1f46dcSZachary Turner if (PyErr_Occurred()) 32442c1f46dcSZachary Turner PyErr_Clear(); 32452c1f46dcSZachary Turner 3246f8b22f8fSZachary Turner if (!pmeth.IsAllocated()) 32472c1f46dcSZachary Turner return false; 32482c1f46dcSZachary Turner 3249b9c1b51eSKate Stone if (PyCallable_Check(pmeth.get()) == 0) { 32502c1f46dcSZachary Turner if (PyErr_Occurred()) 32512c1f46dcSZachary Turner PyErr_Clear(); 32522c1f46dcSZachary Turner 32532c1f46dcSZachary Turner return false; 32542c1f46dcSZachary Turner } 32552c1f46dcSZachary Turner 32562c1f46dcSZachary Turner if (PyErr_Occurred()) 32572c1f46dcSZachary Turner PyErr_Clear(); 32582c1f46dcSZachary Turner 32592c1f46dcSZachary Turner // right now we know this function exists and is callable.. 3260b9c1b51eSKate Stone PythonObject py_return( 3261b9c1b51eSKate Stone PyRefType::Owned, 3262b9c1b51eSKate Stone PyObject_CallMethod(implementor.get(), callee_name, nullptr)); 32632c1f46dcSZachary Turner 32642c1f46dcSZachary Turner // if it fails, print the error but otherwise go on 3265b9c1b51eSKate Stone if (PyErr_Occurred()) { 32662c1f46dcSZachary Turner PyErr_Print(); 32672c1f46dcSZachary Turner PyErr_Clear(); 32682c1f46dcSZachary Turner } 32692c1f46dcSZachary Turner 3270b9c1b51eSKate Stone if (py_return.IsAllocated() && PythonString::Check(py_return.get())) { 3271f8b22f8fSZachary Turner PythonString str(PyRefType::Borrowed, py_return.get()); 327222c8efcdSZachary Turner llvm::StringRef str_data(str.GetString()); 327322c8efcdSZachary Turner dest.assign(str_data.data(), str_data.size()); 32742c1f46dcSZachary Turner got_string = true; 32752c1f46dcSZachary Turner } 32762c1f46dcSZachary Turner 32772c1f46dcSZachary Turner return got_string; 32782c1f46dcSZachary Turner } 32792c1f46dcSZachary Turner 32802c1f46dcSZachary Turner std::unique_ptr<ScriptInterpreterLocker> 3281b9c1b51eSKate Stone ScriptInterpreterPython::AcquireInterpreterLock() { 3282b9c1b51eSKate Stone std::unique_ptr<ScriptInterpreterLocker> py_lock(new Locker( 3283b9c1b51eSKate Stone this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN, 32842c1f46dcSZachary Turner Locker::FreeLock | Locker::TearDownSession)); 32852c1f46dcSZachary Turner return py_lock; 32862c1f46dcSZachary Turner } 32872c1f46dcSZachary Turner 3288b9c1b51eSKate Stone void ScriptInterpreterPython::InitializePrivate() { 328915d1b4e2SEnrico Granata if (g_initialized) 329015d1b4e2SEnrico Granata return; 329115d1b4e2SEnrico Granata 32922c1f46dcSZachary Turner g_initialized = true; 32932c1f46dcSZachary Turner 3294f9d16476SPavel Labath static Timer::Category func_cat(LLVM_PRETTY_FUNCTION); 3295f9d16476SPavel Labath Timer scoped_timer(func_cat, LLVM_PRETTY_FUNCTION); 32962c1f46dcSZachary Turner 3297b9c1b51eSKate Stone // RAII-based initialization which correctly handles multiple-initialization, 329805097246SAdrian Prantl // version- specific differences among Python 2 and Python 3, and saving and 329905097246SAdrian Prantl // restoring various other pieces of state that can get mucked with during 330005097246SAdrian Prantl // initialization. 3301079fe48aSZachary Turner InitializePythonRAII initialize_guard; 33022c1f46dcSZachary Turner 3303*05495c5dSJonas Devlieghere LLDBSwigPyInit(); 33042c1f46dcSZachary Turner 3305b9c1b51eSKate Stone // Update the path python uses to search for modules to include the current 3306b9c1b51eSKate Stone // directory. 33072c1f46dcSZachary Turner 33082c1f46dcSZachary Turner PyRun_SimpleString("import sys"); 33092c1f46dcSZachary Turner AddToSysPath(AddLocation::End, "."); 33102c1f46dcSZachary Turner 3311b9c1b51eSKate Stone // Don't denormalize paths when calling file_spec.GetPath(). On platforms 331205097246SAdrian Prantl // that use a backslash as the path separator, this will result in executing 331305097246SAdrian Prantl // python code containing paths with unescaped backslashes. But Python also 331405097246SAdrian Prantl // accepts forward slashes, so to make life easier we just use that. 33152df331b0SPavel Labath if (FileSpec file_spec = GetPythonDir()) 33162c1f46dcSZachary Turner AddToSysPath(AddLocation::Beginning, file_spec.GetPath(false)); 331760f028ffSPavel Labath if (FileSpec file_spec = HostInfo::GetShlibDir()) 33182c1f46dcSZachary Turner AddToSysPath(AddLocation::Beginning, file_spec.GetPath(false)); 33192c1f46dcSZachary Turner 3320b9c1b51eSKate Stone PyRun_SimpleString("sys.dont_write_bytecode = 1; import " 3321b9c1b51eSKate Stone "lldb.embedded_interpreter; from " 3322b9c1b51eSKate Stone "lldb.embedded_interpreter import run_python_interpreter; " 3323b9c1b51eSKate Stone "from lldb.embedded_interpreter import run_one_line"); 33242c1f46dcSZachary Turner } 33252c1f46dcSZachary Turner 3326b9c1b51eSKate Stone void ScriptInterpreterPython::AddToSysPath(AddLocation location, 3327b9c1b51eSKate Stone std::string path) { 33282c1f46dcSZachary Turner std::string path_copy; 33292c1f46dcSZachary Turner 33302c1f46dcSZachary Turner std::string statement; 3331b9c1b51eSKate Stone if (location == AddLocation::Beginning) { 33322c1f46dcSZachary Turner statement.assign("sys.path.insert(0,\""); 33332c1f46dcSZachary Turner statement.append(path); 33342c1f46dcSZachary Turner statement.append("\")"); 3335b9c1b51eSKate Stone } else { 33362c1f46dcSZachary Turner statement.assign("sys.path.append(\""); 33372c1f46dcSZachary Turner statement.append(path); 33382c1f46dcSZachary Turner statement.append("\")"); 33392c1f46dcSZachary Turner } 33402c1f46dcSZachary Turner PyRun_SimpleString(statement.c_str()); 33412c1f46dcSZachary Turner } 33422c1f46dcSZachary Turner 3343bcadb5a3SPavel Labath // We are intentionally NOT calling Py_Finalize here (this would be the logical 3344bcadb5a3SPavel Labath // place to call it). Calling Py_Finalize here causes test suite runs to seg 3345bcadb5a3SPavel Labath // fault: The test suite runs in Python. It registers SBDebugger::Terminate to 3346bcadb5a3SPavel Labath // be called 'at_exit'. When the test suite Python harness finishes up, it 3347bcadb5a3SPavel Labath // calls Py_Finalize, which calls all the 'at_exit' registered functions. 3348bcadb5a3SPavel Labath // SBDebugger::Terminate calls Debugger::Terminate, which calls lldb::Terminate, 3349bcadb5a3SPavel Labath // which calls ScriptInterpreter::Terminate, which calls 3350bcadb5a3SPavel Labath // ScriptInterpreterPython::Terminate. So if we call Py_Finalize here, we end 3351bcadb5a3SPavel Labath // up with Py_Finalize being called from within Py_Finalize, which results in a 3352bcadb5a3SPavel Labath // seg fault. Since this function only gets called when lldb is shutting down 3353bcadb5a3SPavel Labath // and going away anyway, the fact that we don't actually call Py_Finalize 3354bcadb5a3SPavel Labath // should not cause any problems (everything should shut down/go away anyway 3355bcadb5a3SPavel Labath // when the process exits). 3356bcadb5a3SPavel Labath // 3357bcadb5a3SPavel Labath // void ScriptInterpreterPython::Terminate() { Py_Finalize (); } 3358d68983e3SPavel Labath 3359d68983e3SPavel Labath #endif // LLDB_DISABLE_PYTHON 3360