15ffd83dbSDimitry Andric //===-- ScriptInterpreterPython.cpp ---------------------------------------===// 2*0b57cec5SDimitry Andric // 3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6*0b57cec5SDimitry Andric // 7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 8*0b57cec5SDimitry Andric 9480093f4SDimitry Andric #include "lldb/Host/Config.h" 10e8d8bef9SDimitry Andric #include "lldb/lldb-enumerations.h" 11*0b57cec5SDimitry Andric 12480093f4SDimitry Andric #if LLDB_ENABLE_PYTHON 13*0b57cec5SDimitry Andric 14*0b57cec5SDimitry Andric // LLDB Python header must be included first 15*0b57cec5SDimitry Andric #include "lldb-python.h" 16*0b57cec5SDimitry Andric 17*0b57cec5SDimitry Andric #include "PythonDataObjects.h" 18c14a5a88SDimitry Andric #include "PythonReadline.h" 19fe6060f1SDimitry Andric #include "SWIGPythonBridge.h" 20*0b57cec5SDimitry Andric #include "ScriptInterpreterPythonImpl.h" 21fe6060f1SDimitry Andric #include "ScriptedProcessPythonInterface.h" 22fe6060f1SDimitry Andric 23fe6060f1SDimitry Andric #include "lldb/API/SBError.h" 24*0b57cec5SDimitry Andric #include "lldb/API/SBFrame.h" 25*0b57cec5SDimitry Andric #include "lldb/API/SBValue.h" 26*0b57cec5SDimitry Andric #include "lldb/Breakpoint/StoppointCallbackContext.h" 27*0b57cec5SDimitry Andric #include "lldb/Breakpoint/WatchpointOptions.h" 28*0b57cec5SDimitry Andric #include "lldb/Core/Communication.h" 29*0b57cec5SDimitry Andric #include "lldb/Core/Debugger.h" 30*0b57cec5SDimitry Andric #include "lldb/Core/PluginManager.h" 31*0b57cec5SDimitry Andric #include "lldb/Core/ValueObject.h" 32*0b57cec5SDimitry Andric #include "lldb/DataFormatters/TypeSummary.h" 33*0b57cec5SDimitry Andric #include "lldb/Host/FileSystem.h" 34*0b57cec5SDimitry Andric #include "lldb/Host/HostInfo.h" 35*0b57cec5SDimitry Andric #include "lldb/Host/Pipe.h" 36*0b57cec5SDimitry Andric #include "lldb/Interpreter/CommandInterpreter.h" 37*0b57cec5SDimitry Andric #include "lldb/Interpreter/CommandReturnObject.h" 38*0b57cec5SDimitry Andric #include "lldb/Target/Thread.h" 39*0b57cec5SDimitry Andric #include "lldb/Target/ThreadPlan.h" 4004eeddc0SDimitry Andric #include "lldb/Utility/Instrumentation.h" 4181ad6265SDimitry Andric #include "lldb/Utility/LLDBLog.h" 42*0b57cec5SDimitry Andric #include "lldb/Utility/Timer.h" 43*0b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h" 44*0b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h" 455ffd83dbSDimitry Andric #include "llvm/Support/Error.h" 46*0b57cec5SDimitry Andric #include "llvm/Support/FileSystem.h" 479dba64beSDimitry Andric #include "llvm/Support/FormatAdapters.h" 48*0b57cec5SDimitry Andric 49fe6060f1SDimitry Andric #include <cstdio> 50fe6060f1SDimitry Andric #include <cstdlib> 51*0b57cec5SDimitry Andric #include <memory> 52*0b57cec5SDimitry Andric #include <mutex> 53*0b57cec5SDimitry Andric #include <string> 54*0b57cec5SDimitry Andric 55*0b57cec5SDimitry Andric using namespace lldb; 56*0b57cec5SDimitry Andric using namespace lldb_private; 579dba64beSDimitry Andric using namespace lldb_private::python; 589dba64beSDimitry Andric using llvm::Expected; 59*0b57cec5SDimitry Andric 605ffd83dbSDimitry Andric LLDB_PLUGIN_DEFINE(ScriptInterpreterPython) 615ffd83dbSDimitry Andric 62*0b57cec5SDimitry Andric // Defined in the SWIG source file 63*0b57cec5SDimitry Andric extern "C" PyObject *PyInit__lldb(void); 64*0b57cec5SDimitry Andric 65*0b57cec5SDimitry Andric #define LLDBSwigPyInit PyInit__lldb 66*0b57cec5SDimitry Andric 6704eeddc0SDimitry Andric #if defined(_WIN32) 6804eeddc0SDimitry Andric // Don't mess with the signal handlers on Windows. 6904eeddc0SDimitry Andric #define LLDB_USE_PYTHON_SET_INTERRUPT 0 7004eeddc0SDimitry Andric #else 7104eeddc0SDimitry Andric // PyErr_SetInterrupt was introduced in 3.2. 7204eeddc0SDimitry Andric #define LLDB_USE_PYTHON_SET_INTERRUPT \ 7304eeddc0SDimitry Andric (PY_MAJOR_VERSION == 3 && PY_MINOR_VERSION >= 2) || (PY_MAJOR_VERSION > 3) 7404eeddc0SDimitry Andric #endif 75*0b57cec5SDimitry Andric 76e8d8bef9SDimitry Andric static ScriptInterpreterPythonImpl *GetPythonInterpreter(Debugger &debugger) { 77e8d8bef9SDimitry Andric ScriptInterpreter *script_interpreter = 78e8d8bef9SDimitry Andric debugger.GetScriptInterpreter(true, lldb::eScriptLanguagePython); 79e8d8bef9SDimitry Andric return static_cast<ScriptInterpreterPythonImpl *>(script_interpreter); 80e8d8bef9SDimitry Andric } 81e8d8bef9SDimitry Andric 82*0b57cec5SDimitry Andric namespace { 83*0b57cec5SDimitry Andric 84*0b57cec5SDimitry Andric // Initializing Python is not a straightforward process. We cannot control 85*0b57cec5SDimitry Andric // what external code may have done before getting to this point in LLDB, 86*0b57cec5SDimitry Andric // including potentially having already initialized Python, so we need to do a 87*0b57cec5SDimitry Andric // lot of work to ensure that the existing state of the system is maintained 88*0b57cec5SDimitry Andric // across our initialization. We do this by using an RAII pattern where we 89*0b57cec5SDimitry Andric // save off initial state at the beginning, and restore it at the end 90*0b57cec5SDimitry Andric struct InitializePythonRAII { 91*0b57cec5SDimitry Andric public: 92fe6060f1SDimitry Andric InitializePythonRAII() { 93*0b57cec5SDimitry Andric InitializePythonHome(); 94*0b57cec5SDimitry Andric 95c14a5a88SDimitry Andric #ifdef LLDB_USE_LIBEDIT_READLINE_COMPAT_MODULE 96c14a5a88SDimitry Andric // Python's readline is incompatible with libedit being linked into lldb. 97c14a5a88SDimitry Andric // Provide a patched version local to the embedded interpreter. 98c14a5a88SDimitry Andric bool ReadlinePatched = false; 99c14a5a88SDimitry Andric for (auto *p = PyImport_Inittab; p->name != NULL; p++) { 100c14a5a88SDimitry Andric if (strcmp(p->name, "readline") == 0) { 101c14a5a88SDimitry Andric p->initfunc = initlldb_readline; 102c14a5a88SDimitry Andric break; 103c14a5a88SDimitry Andric } 104c14a5a88SDimitry Andric } 105c14a5a88SDimitry Andric if (!ReadlinePatched) { 106c14a5a88SDimitry Andric PyImport_AppendInittab("readline", initlldb_readline); 107c14a5a88SDimitry Andric ReadlinePatched = true; 108c14a5a88SDimitry Andric } 109c14a5a88SDimitry Andric #endif 110c14a5a88SDimitry Andric 111*0b57cec5SDimitry Andric // Register _lldb as a built-in module. 112*0b57cec5SDimitry Andric PyImport_AppendInittab("_lldb", LLDBSwigPyInit); 113*0b57cec5SDimitry Andric 114*0b57cec5SDimitry Andric // Python < 3.2 and Python >= 3.2 reversed the ordering requirements for 115*0b57cec5SDimitry Andric // calling `Py_Initialize` and `PyEval_InitThreads`. < 3.2 requires that you 116*0b57cec5SDimitry Andric // call `PyEval_InitThreads` first, and >= 3.2 requires that you call it last. 117*0b57cec5SDimitry Andric #if (PY_MAJOR_VERSION == 3 && PY_MINOR_VERSION >= 2) || (PY_MAJOR_VERSION > 3) 118*0b57cec5SDimitry Andric Py_InitializeEx(0); 119*0b57cec5SDimitry Andric InitializeThreadsPrivate(); 120*0b57cec5SDimitry Andric #else 121*0b57cec5SDimitry Andric InitializeThreadsPrivate(); 122*0b57cec5SDimitry Andric Py_InitializeEx(0); 123*0b57cec5SDimitry Andric #endif 124*0b57cec5SDimitry Andric } 125*0b57cec5SDimitry Andric 126*0b57cec5SDimitry Andric ~InitializePythonRAII() { 127*0b57cec5SDimitry Andric if (m_was_already_initialized) { 12881ad6265SDimitry Andric Log *log = GetLog(LLDBLog::Script); 129*0b57cec5SDimitry Andric LLDB_LOGV(log, "Releasing PyGILState. Returning to state = {0}locked", 130*0b57cec5SDimitry Andric m_gil_state == PyGILState_UNLOCKED ? "un" : ""); 131*0b57cec5SDimitry Andric PyGILState_Release(m_gil_state); 132*0b57cec5SDimitry Andric } else { 133*0b57cec5SDimitry Andric // We initialized the threads in this function, just unlock the GIL. 134*0b57cec5SDimitry Andric PyEval_SaveThread(); 135*0b57cec5SDimitry Andric } 136*0b57cec5SDimitry Andric } 137*0b57cec5SDimitry Andric 138*0b57cec5SDimitry Andric private: 139*0b57cec5SDimitry Andric void InitializePythonHome() { 1405ffd83dbSDimitry Andric #if LLDB_EMBED_PYTHON_HOME 1415ffd83dbSDimitry Andric typedef wchar_t *str_type; 1425ffd83dbSDimitry Andric static str_type g_python_home = []() -> str_type { 1435ffd83dbSDimitry Andric const char *lldb_python_home = LLDB_PYTHON_HOME; 1445ffd83dbSDimitry Andric const char *absolute_python_home = nullptr; 1455ffd83dbSDimitry Andric llvm::SmallString<64> path; 1465ffd83dbSDimitry Andric if (llvm::sys::path::is_absolute(lldb_python_home)) { 1475ffd83dbSDimitry Andric absolute_python_home = lldb_python_home; 1485ffd83dbSDimitry Andric } else { 1495ffd83dbSDimitry Andric FileSpec spec = HostInfo::GetShlibDir(); 1505ffd83dbSDimitry Andric if (!spec) 1515ffd83dbSDimitry Andric return nullptr; 1525ffd83dbSDimitry Andric spec.GetPath(path); 1535ffd83dbSDimitry Andric llvm::sys::path::append(path, lldb_python_home); 1545ffd83dbSDimitry Andric absolute_python_home = path.c_str(); 1555ffd83dbSDimitry Andric } 156*0b57cec5SDimitry Andric size_t size = 0; 1575ffd83dbSDimitry Andric return Py_DecodeLocale(absolute_python_home, &size); 1585ffd83dbSDimitry Andric }(); 1595ffd83dbSDimitry Andric if (g_python_home != nullptr) { 160*0b57cec5SDimitry Andric Py_SetPythonHome(g_python_home); 1615ffd83dbSDimitry Andric } 162*0b57cec5SDimitry Andric #endif 163*0b57cec5SDimitry Andric } 164*0b57cec5SDimitry Andric 165*0b57cec5SDimitry Andric void InitializeThreadsPrivate() { 166*0b57cec5SDimitry Andric // Since Python 3.7 `Py_Initialize` calls `PyEval_InitThreads` inside itself, 167*0b57cec5SDimitry Andric // so there is no way to determine whether the embedded interpreter 168*0b57cec5SDimitry Andric // was already initialized by some external code. `PyEval_ThreadsInitialized` 169*0b57cec5SDimitry Andric // would always return `true` and `PyGILState_Ensure/Release` flow would be 170*0b57cec5SDimitry Andric // executed instead of unlocking GIL with `PyEval_SaveThread`. When 171*0b57cec5SDimitry Andric // an another thread calls `PyGILState_Ensure` it would get stuck in deadlock. 172*0b57cec5SDimitry Andric #if (PY_MAJOR_VERSION == 3 && PY_MINOR_VERSION >= 7) || (PY_MAJOR_VERSION > 3) 173*0b57cec5SDimitry Andric // The only case we should go further and acquire the GIL: it is unlocked. 174*0b57cec5SDimitry Andric if (PyGILState_Check()) 175*0b57cec5SDimitry Andric return; 176*0b57cec5SDimitry Andric #endif 177*0b57cec5SDimitry Andric 178*0b57cec5SDimitry Andric if (PyEval_ThreadsInitialized()) { 17981ad6265SDimitry Andric Log *log = GetLog(LLDBLog::Script); 180*0b57cec5SDimitry Andric 181*0b57cec5SDimitry Andric m_was_already_initialized = true; 182*0b57cec5SDimitry Andric m_gil_state = PyGILState_Ensure(); 183*0b57cec5SDimitry Andric LLDB_LOGV(log, "Ensured PyGILState. Previous state = {0}locked\n", 184*0b57cec5SDimitry Andric m_gil_state == PyGILState_UNLOCKED ? "un" : ""); 185*0b57cec5SDimitry Andric return; 186*0b57cec5SDimitry Andric } 187*0b57cec5SDimitry Andric 188*0b57cec5SDimitry Andric // InitThreads acquires the GIL if it hasn't been called before. 189*0b57cec5SDimitry Andric PyEval_InitThreads(); 190*0b57cec5SDimitry Andric } 191*0b57cec5SDimitry Andric 192fe6060f1SDimitry Andric PyGILState_STATE m_gil_state = PyGILState_UNLOCKED; 193fe6060f1SDimitry Andric bool m_was_already_initialized = false; 194*0b57cec5SDimitry Andric }; 19504eeddc0SDimitry Andric 19604eeddc0SDimitry Andric #if LLDB_USE_PYTHON_SET_INTERRUPT 19704eeddc0SDimitry Andric /// Saves the current signal handler for the specified signal and restores 19804eeddc0SDimitry Andric /// it at the end of the current scope. 19904eeddc0SDimitry Andric struct RestoreSignalHandlerScope { 20004eeddc0SDimitry Andric /// The signal handler. 20104eeddc0SDimitry Andric struct sigaction m_prev_handler; 20204eeddc0SDimitry Andric int m_signal_code; 20304eeddc0SDimitry Andric RestoreSignalHandlerScope(int signal_code) : m_signal_code(signal_code) { 20404eeddc0SDimitry Andric // Initialize sigaction to their default state. 20504eeddc0SDimitry Andric std::memset(&m_prev_handler, 0, sizeof(m_prev_handler)); 20604eeddc0SDimitry Andric // Don't install a new handler, just read back the old one. 20704eeddc0SDimitry Andric struct sigaction *new_handler = nullptr; 20804eeddc0SDimitry Andric int signal_err = ::sigaction(m_signal_code, new_handler, &m_prev_handler); 20904eeddc0SDimitry Andric lldbassert(signal_err == 0 && "sigaction failed to read handler"); 21004eeddc0SDimitry Andric } 21104eeddc0SDimitry Andric ~RestoreSignalHandlerScope() { 21204eeddc0SDimitry Andric int signal_err = ::sigaction(m_signal_code, &m_prev_handler, nullptr); 21304eeddc0SDimitry Andric lldbassert(signal_err == 0 && "sigaction failed to restore old handler"); 21404eeddc0SDimitry Andric } 21504eeddc0SDimitry Andric }; 21604eeddc0SDimitry Andric #endif 217*0b57cec5SDimitry Andric } // namespace 218*0b57cec5SDimitry Andric 219*0b57cec5SDimitry Andric void ScriptInterpreterPython::ComputePythonDirForApple( 220*0b57cec5SDimitry Andric llvm::SmallVectorImpl<char> &path) { 221*0b57cec5SDimitry Andric auto style = llvm::sys::path::Style::posix; 222*0b57cec5SDimitry Andric 223*0b57cec5SDimitry Andric llvm::StringRef path_ref(path.begin(), path.size()); 224*0b57cec5SDimitry Andric auto rbegin = llvm::sys::path::rbegin(path_ref, style); 225*0b57cec5SDimitry Andric auto rend = llvm::sys::path::rend(path_ref); 226*0b57cec5SDimitry Andric auto framework = std::find(rbegin, rend, "LLDB.framework"); 227*0b57cec5SDimitry Andric if (framework == rend) { 2289dba64beSDimitry Andric ComputePythonDir(path); 229*0b57cec5SDimitry Andric return; 230*0b57cec5SDimitry Andric } 231*0b57cec5SDimitry Andric path.resize(framework - rend); 232*0b57cec5SDimitry Andric llvm::sys::path::append(path, style, "LLDB.framework", "Resources", "Python"); 233*0b57cec5SDimitry Andric } 234*0b57cec5SDimitry Andric 2359dba64beSDimitry Andric void ScriptInterpreterPython::ComputePythonDir( 236*0b57cec5SDimitry Andric llvm::SmallVectorImpl<char> &path) { 237*0b57cec5SDimitry Andric // Build the path by backing out of the lib dir, then building with whatever 238*0b57cec5SDimitry Andric // the real python interpreter uses. (e.g. lib for most, lib64 on RHEL 2399dba64beSDimitry Andric // x86_64, or bin on Windows). 2409dba64beSDimitry Andric llvm::sys::path::remove_filename(path); 2419dba64beSDimitry Andric llvm::sys::path::append(path, LLDB_PYTHON_RELATIVE_LIBDIR); 242*0b57cec5SDimitry Andric 2439dba64beSDimitry Andric #if defined(_WIN32) 244*0b57cec5SDimitry Andric // This will be injected directly through FileSpec.GetDirectory().SetString(), 245*0b57cec5SDimitry Andric // so we need to normalize manually. 246*0b57cec5SDimitry Andric std::replace(path.begin(), path.end(), '\\', '/'); 2479dba64beSDimitry Andric #endif 248*0b57cec5SDimitry Andric } 249*0b57cec5SDimitry Andric 250*0b57cec5SDimitry Andric FileSpec ScriptInterpreterPython::GetPythonDir() { 251*0b57cec5SDimitry Andric static FileSpec g_spec = []() { 252*0b57cec5SDimitry Andric FileSpec spec = HostInfo::GetShlibDir(); 253*0b57cec5SDimitry Andric if (!spec) 254*0b57cec5SDimitry Andric return FileSpec(); 255*0b57cec5SDimitry Andric llvm::SmallString<64> path; 256*0b57cec5SDimitry Andric spec.GetPath(path); 257*0b57cec5SDimitry Andric 258*0b57cec5SDimitry Andric #if defined(__APPLE__) 259*0b57cec5SDimitry Andric ComputePythonDirForApple(path); 260*0b57cec5SDimitry Andric #else 2619dba64beSDimitry Andric ComputePythonDir(path); 262*0b57cec5SDimitry Andric #endif 263*0b57cec5SDimitry Andric spec.GetDirectory().SetString(path); 264*0b57cec5SDimitry Andric return spec; 265*0b57cec5SDimitry Andric }(); 266*0b57cec5SDimitry Andric return g_spec; 267*0b57cec5SDimitry Andric } 268*0b57cec5SDimitry Andric 269349cc55cSDimitry Andric static const char GetInterpreterInfoScript[] = R"( 270349cc55cSDimitry Andric import os 271349cc55cSDimitry Andric import sys 272349cc55cSDimitry Andric 273349cc55cSDimitry Andric def main(lldb_python_dir, python_exe_relative_path): 274349cc55cSDimitry Andric info = { 275349cc55cSDimitry Andric "lldb-pythonpath": lldb_python_dir, 276349cc55cSDimitry Andric "language": "python", 277349cc55cSDimitry Andric "prefix": sys.prefix, 278349cc55cSDimitry Andric "executable": os.path.join(sys.prefix, python_exe_relative_path) 279349cc55cSDimitry Andric } 280349cc55cSDimitry Andric return info 281349cc55cSDimitry Andric )"; 282349cc55cSDimitry Andric 283349cc55cSDimitry Andric static const char python_exe_relative_path[] = LLDB_PYTHON_EXE_RELATIVE_PATH; 284349cc55cSDimitry Andric 285349cc55cSDimitry Andric StructuredData::DictionarySP ScriptInterpreterPython::GetInterpreterInfo() { 286349cc55cSDimitry Andric GIL gil; 287349cc55cSDimitry Andric FileSpec python_dir_spec = GetPythonDir(); 288349cc55cSDimitry Andric if (!python_dir_spec) 289349cc55cSDimitry Andric return nullptr; 290349cc55cSDimitry Andric PythonScript get_info(GetInterpreterInfoScript); 291349cc55cSDimitry Andric auto info_json = unwrapIgnoringErrors( 292349cc55cSDimitry Andric As<PythonDictionary>(get_info(PythonString(python_dir_spec.GetPath()), 293349cc55cSDimitry Andric PythonString(python_exe_relative_path)))); 294349cc55cSDimitry Andric if (!info_json) 295349cc55cSDimitry Andric return nullptr; 296349cc55cSDimitry Andric return info_json.CreateStructuredDictionary(); 297349cc55cSDimitry Andric } 298349cc55cSDimitry Andric 299fe6060f1SDimitry Andric void ScriptInterpreterPython::SharedLibraryDirectoryHelper( 300fe6060f1SDimitry Andric FileSpec &this_file) { 301fe6060f1SDimitry Andric // When we're loaded from python, this_file will point to the file inside the 302fe6060f1SDimitry Andric // python package directory. Replace it with the one in the lib directory. 303fe6060f1SDimitry Andric #ifdef _WIN32 304fe6060f1SDimitry Andric // On windows, we need to manually back out of the python tree, and go into 305fe6060f1SDimitry Andric // the bin directory. This is pretty much the inverse of what ComputePythonDir 306fe6060f1SDimitry Andric // does. 307fe6060f1SDimitry Andric if (this_file.GetFileNameExtension() == ConstString(".pyd")) { 308fe6060f1SDimitry Andric this_file.RemoveLastPathComponent(); // _lldb.pyd or _lldb_d.pyd 309fe6060f1SDimitry Andric this_file.RemoveLastPathComponent(); // lldb 310fe6060f1SDimitry Andric llvm::StringRef libdir = LLDB_PYTHON_RELATIVE_LIBDIR; 311fe6060f1SDimitry Andric for (auto it = llvm::sys::path::begin(libdir), 312fe6060f1SDimitry Andric end = llvm::sys::path::end(libdir); 313fe6060f1SDimitry Andric it != end; ++it) 314fe6060f1SDimitry Andric this_file.RemoveLastPathComponent(); 315fe6060f1SDimitry Andric this_file.AppendPathComponent("bin"); 316fe6060f1SDimitry Andric this_file.AppendPathComponent("liblldb.dll"); 317fe6060f1SDimitry Andric } 318fe6060f1SDimitry Andric #else 319fe6060f1SDimitry Andric // The python file is a symlink, so we can find the real library by resolving 320fe6060f1SDimitry Andric // it. We can do this unconditionally. 321fe6060f1SDimitry Andric FileSystem::Instance().ResolveSymbolicLink(this_file, this_file); 322fe6060f1SDimitry Andric #endif 323fe6060f1SDimitry Andric } 324fe6060f1SDimitry Andric 325349cc55cSDimitry Andric llvm::StringRef ScriptInterpreterPython::GetPluginDescriptionStatic() { 326*0b57cec5SDimitry Andric return "Embedded Python interpreter"; 327*0b57cec5SDimitry Andric } 328*0b57cec5SDimitry Andric 329*0b57cec5SDimitry Andric void ScriptInterpreterPython::Initialize() { 330*0b57cec5SDimitry Andric static llvm::once_flag g_once_flag; 331*0b57cec5SDimitry Andric llvm::call_once(g_once_flag, []() { 332*0b57cec5SDimitry Andric PluginManager::RegisterPlugin(GetPluginNameStatic(), 333*0b57cec5SDimitry Andric GetPluginDescriptionStatic(), 334*0b57cec5SDimitry Andric lldb::eScriptLanguagePython, 335*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::CreateInstance); 33604eeddc0SDimitry Andric ScriptInterpreterPythonImpl::Initialize(); 337*0b57cec5SDimitry Andric }); 338*0b57cec5SDimitry Andric } 339*0b57cec5SDimitry Andric 340*0b57cec5SDimitry Andric void ScriptInterpreterPython::Terminate() {} 341*0b57cec5SDimitry Andric 342*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::Locker::Locker( 343*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl *py_interpreter, uint16_t on_entry, 3449dba64beSDimitry Andric uint16_t on_leave, FileSP in, FileSP out, FileSP err) 345*0b57cec5SDimitry Andric : ScriptInterpreterLocker(), 346*0b57cec5SDimitry Andric m_teardown_session((on_leave & TearDownSession) == TearDownSession), 347*0b57cec5SDimitry Andric m_python_interpreter(py_interpreter) { 348*0b57cec5SDimitry Andric DoAcquireLock(); 349*0b57cec5SDimitry Andric if ((on_entry & InitSession) == InitSession) { 350*0b57cec5SDimitry Andric if (!DoInitSession(on_entry, in, out, err)) { 351*0b57cec5SDimitry Andric // Don't teardown the session if we didn't init it. 352*0b57cec5SDimitry Andric m_teardown_session = false; 353*0b57cec5SDimitry Andric } 354*0b57cec5SDimitry Andric } 355*0b57cec5SDimitry Andric } 356*0b57cec5SDimitry Andric 357*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::Locker::DoAcquireLock() { 35881ad6265SDimitry Andric Log *log = GetLog(LLDBLog::Script); 359*0b57cec5SDimitry Andric m_GILState = PyGILState_Ensure(); 360*0b57cec5SDimitry Andric LLDB_LOGV(log, "Ensured PyGILState. Previous state = {0}locked", 361*0b57cec5SDimitry Andric m_GILState == PyGILState_UNLOCKED ? "un" : ""); 362*0b57cec5SDimitry Andric 363*0b57cec5SDimitry Andric // we need to save the thread state when we first start the command because 364*0b57cec5SDimitry Andric // we might decide to interrupt it while some action is taking place outside 365*0b57cec5SDimitry Andric // of Python (e.g. printing to screen, waiting for the network, ...) in that 366*0b57cec5SDimitry Andric // case, _PyThreadState_Current will be NULL - and we would be unable to set 367*0b57cec5SDimitry Andric // the asynchronous exception - not a desirable situation 368*0b57cec5SDimitry Andric m_python_interpreter->SetThreadState(PyThreadState_Get()); 369*0b57cec5SDimitry Andric m_python_interpreter->IncrementLockCount(); 370*0b57cec5SDimitry Andric return true; 371*0b57cec5SDimitry Andric } 372*0b57cec5SDimitry Andric 373*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::Locker::DoInitSession(uint16_t on_entry_flags, 3749dba64beSDimitry Andric FileSP in, FileSP out, 3759dba64beSDimitry Andric FileSP err) { 376*0b57cec5SDimitry Andric if (!m_python_interpreter) 377*0b57cec5SDimitry Andric return false; 378*0b57cec5SDimitry Andric return m_python_interpreter->EnterSession(on_entry_flags, in, out, err); 379*0b57cec5SDimitry Andric } 380*0b57cec5SDimitry Andric 381*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::Locker::DoFreeLock() { 38281ad6265SDimitry Andric Log *log = GetLog(LLDBLog::Script); 383*0b57cec5SDimitry Andric LLDB_LOGV(log, "Releasing PyGILState. Returning to state = {0}locked", 384*0b57cec5SDimitry Andric m_GILState == PyGILState_UNLOCKED ? "un" : ""); 385*0b57cec5SDimitry Andric PyGILState_Release(m_GILState); 386*0b57cec5SDimitry Andric m_python_interpreter->DecrementLockCount(); 387*0b57cec5SDimitry Andric return true; 388*0b57cec5SDimitry Andric } 389*0b57cec5SDimitry Andric 390*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::Locker::DoTearDownSession() { 391*0b57cec5SDimitry Andric if (!m_python_interpreter) 392*0b57cec5SDimitry Andric return false; 393*0b57cec5SDimitry Andric m_python_interpreter->LeaveSession(); 394*0b57cec5SDimitry Andric return true; 395*0b57cec5SDimitry Andric } 396*0b57cec5SDimitry Andric 397*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::Locker::~Locker() { 398*0b57cec5SDimitry Andric if (m_teardown_session) 399*0b57cec5SDimitry Andric DoTearDownSession(); 400*0b57cec5SDimitry Andric DoFreeLock(); 401*0b57cec5SDimitry Andric } 402*0b57cec5SDimitry Andric 403*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::ScriptInterpreterPythonImpl(Debugger &debugger) 404*0b57cec5SDimitry Andric : ScriptInterpreterPython(debugger), m_saved_stdin(), m_saved_stdout(), 405*0b57cec5SDimitry Andric m_saved_stderr(), m_main_module(), 406*0b57cec5SDimitry Andric m_session_dict(PyInitialValue::Invalid), 407*0b57cec5SDimitry Andric m_sys_module_dict(PyInitialValue::Invalid), m_run_one_line_function(), 408*0b57cec5SDimitry Andric m_run_one_line_str_global(), 409*0b57cec5SDimitry Andric m_dictionary_name(m_debugger.GetInstanceName().AsCString()), 4109dba64beSDimitry Andric m_active_io_handler(eIOHandlerNone), m_session_is_active(false), 4115ffd83dbSDimitry Andric m_pty_secondary_is_open(false), m_valid_session(true), m_lock_count(0), 4129dba64beSDimitry Andric m_command_thread_state(nullptr) { 413fe6060f1SDimitry Andric m_scripted_process_interface_up = 414fe6060f1SDimitry Andric std::make_unique<ScriptedProcessPythonInterface>(*this); 415fe6060f1SDimitry Andric 416*0b57cec5SDimitry Andric m_dictionary_name.append("_dict"); 417*0b57cec5SDimitry Andric StreamString run_string; 418*0b57cec5SDimitry Andric run_string.Printf("%s = dict()", m_dictionary_name.c_str()); 419*0b57cec5SDimitry Andric 420*0b57cec5SDimitry Andric Locker locker(this, Locker::AcquireLock, Locker::FreeAcquiredLock); 421*0b57cec5SDimitry Andric PyRun_SimpleString(run_string.GetData()); 422*0b57cec5SDimitry Andric 423*0b57cec5SDimitry Andric run_string.Clear(); 424*0b57cec5SDimitry Andric run_string.Printf( 425*0b57cec5SDimitry Andric "run_one_line (%s, 'import copy, keyword, os, re, sys, uuid, lldb')", 426*0b57cec5SDimitry Andric m_dictionary_name.c_str()); 427*0b57cec5SDimitry Andric PyRun_SimpleString(run_string.GetData()); 428*0b57cec5SDimitry Andric 429*0b57cec5SDimitry Andric // Reloading modules requires a different syntax in Python 2 and Python 3. 430*0b57cec5SDimitry Andric // This provides a consistent syntax no matter what version of Python. 431*0b57cec5SDimitry Andric run_string.Clear(); 432*0b57cec5SDimitry Andric run_string.Printf("run_one_line (%s, 'from six.moves import reload_module')", 433*0b57cec5SDimitry Andric m_dictionary_name.c_str()); 434*0b57cec5SDimitry Andric PyRun_SimpleString(run_string.GetData()); 435*0b57cec5SDimitry Andric 436*0b57cec5SDimitry Andric // WARNING: temporary code that loads Cocoa formatters - this should be done 437*0b57cec5SDimitry Andric // on a per-platform basis rather than loading the whole set and letting the 438*0b57cec5SDimitry Andric // individual formatter classes exploit APIs to check whether they can/cannot 439*0b57cec5SDimitry Andric // do their task 440*0b57cec5SDimitry Andric run_string.Clear(); 441*0b57cec5SDimitry Andric run_string.Printf( 442*0b57cec5SDimitry Andric "run_one_line (%s, 'import lldb.formatters, lldb.formatters.cpp, pydoc')", 443*0b57cec5SDimitry Andric m_dictionary_name.c_str()); 444*0b57cec5SDimitry Andric PyRun_SimpleString(run_string.GetData()); 445*0b57cec5SDimitry Andric run_string.Clear(); 446*0b57cec5SDimitry Andric 447*0b57cec5SDimitry Andric run_string.Printf("run_one_line (%s, 'import lldb.embedded_interpreter; from " 448*0b57cec5SDimitry Andric "lldb.embedded_interpreter import run_python_interpreter; " 449*0b57cec5SDimitry Andric "from lldb.embedded_interpreter import run_one_line')", 450*0b57cec5SDimitry Andric m_dictionary_name.c_str()); 451*0b57cec5SDimitry Andric PyRun_SimpleString(run_string.GetData()); 452*0b57cec5SDimitry Andric run_string.Clear(); 453*0b57cec5SDimitry Andric 454*0b57cec5SDimitry Andric run_string.Printf("run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64 455*0b57cec5SDimitry Andric "; pydoc.pager = pydoc.plainpager')", 456*0b57cec5SDimitry Andric m_dictionary_name.c_str(), m_debugger.GetID()); 457*0b57cec5SDimitry Andric PyRun_SimpleString(run_string.GetData()); 458*0b57cec5SDimitry Andric } 459*0b57cec5SDimitry Andric 460*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::~ScriptInterpreterPythonImpl() { 461*0b57cec5SDimitry Andric // the session dictionary may hold objects with complex state which means 462*0b57cec5SDimitry Andric // that they may need to be torn down with some level of smarts and that, in 463*0b57cec5SDimitry Andric // turn, requires a valid thread state force Python to procure itself such a 464*0b57cec5SDimitry Andric // thread state, nuke the session dictionary and then release it for others 465*0b57cec5SDimitry Andric // to use and proceed with the rest of the shutdown 466*0b57cec5SDimitry Andric auto gil_state = PyGILState_Ensure(); 467*0b57cec5SDimitry Andric m_session_dict.Reset(); 468*0b57cec5SDimitry Andric PyGILState_Release(gil_state); 469*0b57cec5SDimitry Andric } 470*0b57cec5SDimitry Andric 471*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::IOHandlerActivated(IOHandler &io_handler, 472*0b57cec5SDimitry Andric bool interactive) { 473*0b57cec5SDimitry Andric const char *instructions = nullptr; 474*0b57cec5SDimitry Andric 475*0b57cec5SDimitry Andric switch (m_active_io_handler) { 476*0b57cec5SDimitry Andric case eIOHandlerNone: 477*0b57cec5SDimitry Andric break; 478*0b57cec5SDimitry Andric case eIOHandlerBreakpoint: 479*0b57cec5SDimitry Andric instructions = R"(Enter your Python command(s). Type 'DONE' to end. 480*0b57cec5SDimitry Andric def function (frame, bp_loc, internal_dict): 481*0b57cec5SDimitry Andric """frame: the lldb.SBFrame for the location at which you stopped 482*0b57cec5SDimitry Andric bp_loc: an lldb.SBBreakpointLocation for the breakpoint location information 483*0b57cec5SDimitry Andric internal_dict: an LLDB support object not to be used""" 484*0b57cec5SDimitry Andric )"; 485*0b57cec5SDimitry Andric break; 486*0b57cec5SDimitry Andric case eIOHandlerWatchpoint: 487*0b57cec5SDimitry Andric instructions = "Enter your Python command(s). Type 'DONE' to end.\n"; 488*0b57cec5SDimitry Andric break; 489*0b57cec5SDimitry Andric } 490*0b57cec5SDimitry Andric 491*0b57cec5SDimitry Andric if (instructions) { 4929dba64beSDimitry Andric StreamFileSP output_sp(io_handler.GetOutputStreamFileSP()); 493*0b57cec5SDimitry Andric if (output_sp && interactive) { 494*0b57cec5SDimitry Andric output_sp->PutCString(instructions); 495*0b57cec5SDimitry Andric output_sp->Flush(); 496*0b57cec5SDimitry Andric } 497*0b57cec5SDimitry Andric } 498*0b57cec5SDimitry Andric } 499*0b57cec5SDimitry Andric 500*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::IOHandlerInputComplete(IOHandler &io_handler, 501*0b57cec5SDimitry Andric std::string &data) { 502*0b57cec5SDimitry Andric io_handler.SetIsDone(true); 503*0b57cec5SDimitry Andric bool batch_mode = m_debugger.GetCommandInterpreter().GetBatchCommandMode(); 504*0b57cec5SDimitry Andric 505*0b57cec5SDimitry Andric switch (m_active_io_handler) { 506*0b57cec5SDimitry Andric case eIOHandlerNone: 507*0b57cec5SDimitry Andric break; 508*0b57cec5SDimitry Andric case eIOHandlerBreakpoint: { 509fe6060f1SDimitry Andric std::vector<std::reference_wrapper<BreakpointOptions>> *bp_options_vec = 510fe6060f1SDimitry Andric (std::vector<std::reference_wrapper<BreakpointOptions>> *) 511fe6060f1SDimitry Andric io_handler.GetUserData(); 512fe6060f1SDimitry Andric for (BreakpointOptions &bp_options : *bp_options_vec) { 513*0b57cec5SDimitry Andric 5149dba64beSDimitry Andric auto data_up = std::make_unique<CommandDataPython>(); 515*0b57cec5SDimitry Andric if (!data_up) 516*0b57cec5SDimitry Andric break; 517*0b57cec5SDimitry Andric data_up->user_source.SplitIntoLines(data); 518*0b57cec5SDimitry Andric 519480093f4SDimitry Andric StructuredData::ObjectSP empty_args_sp; 520*0b57cec5SDimitry Andric if (GenerateBreakpointCommandCallbackData(data_up->user_source, 521480093f4SDimitry Andric data_up->script_source, 522480093f4SDimitry Andric false) 523*0b57cec5SDimitry Andric .Success()) { 524*0b57cec5SDimitry Andric auto baton_sp = std::make_shared<BreakpointOptions::CommandBaton>( 525*0b57cec5SDimitry Andric std::move(data_up)); 526fe6060f1SDimitry Andric bp_options.SetCallback( 527*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::BreakpointCallbackFunction, baton_sp); 528*0b57cec5SDimitry Andric } else if (!batch_mode) { 5299dba64beSDimitry Andric StreamFileSP error_sp = io_handler.GetErrorStreamFileSP(); 530*0b57cec5SDimitry Andric if (error_sp) { 531*0b57cec5SDimitry Andric error_sp->Printf("Warning: No command attached to breakpoint.\n"); 532*0b57cec5SDimitry Andric error_sp->Flush(); 533*0b57cec5SDimitry Andric } 534*0b57cec5SDimitry Andric } 535*0b57cec5SDimitry Andric } 536*0b57cec5SDimitry Andric m_active_io_handler = eIOHandlerNone; 537*0b57cec5SDimitry Andric } break; 538*0b57cec5SDimitry Andric case eIOHandlerWatchpoint: { 539*0b57cec5SDimitry Andric WatchpointOptions *wp_options = 540*0b57cec5SDimitry Andric (WatchpointOptions *)io_handler.GetUserData(); 5419dba64beSDimitry Andric auto data_up = std::make_unique<WatchpointOptions::CommandData>(); 542*0b57cec5SDimitry Andric data_up->user_source.SplitIntoLines(data); 543*0b57cec5SDimitry Andric 544*0b57cec5SDimitry Andric if (GenerateWatchpointCommandCallbackData(data_up->user_source, 545*0b57cec5SDimitry Andric data_up->script_source)) { 546*0b57cec5SDimitry Andric auto baton_sp = 547*0b57cec5SDimitry Andric std::make_shared<WatchpointOptions::CommandBaton>(std::move(data_up)); 548*0b57cec5SDimitry Andric wp_options->SetCallback( 549*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::WatchpointCallbackFunction, baton_sp); 550*0b57cec5SDimitry Andric } else if (!batch_mode) { 5519dba64beSDimitry Andric StreamFileSP error_sp = io_handler.GetErrorStreamFileSP(); 552*0b57cec5SDimitry Andric if (error_sp) { 553*0b57cec5SDimitry Andric error_sp->Printf("Warning: No command attached to breakpoint.\n"); 554*0b57cec5SDimitry Andric error_sp->Flush(); 555*0b57cec5SDimitry Andric } 556*0b57cec5SDimitry Andric } 557*0b57cec5SDimitry Andric m_active_io_handler = eIOHandlerNone; 558*0b57cec5SDimitry Andric } break; 559*0b57cec5SDimitry Andric } 560*0b57cec5SDimitry Andric } 561*0b57cec5SDimitry Andric 562*0b57cec5SDimitry Andric lldb::ScriptInterpreterSP 563*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::CreateInstance(Debugger &debugger) { 564*0b57cec5SDimitry Andric return std::make_shared<ScriptInterpreterPythonImpl>(debugger); 565*0b57cec5SDimitry Andric } 566*0b57cec5SDimitry Andric 567*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::LeaveSession() { 56881ad6265SDimitry Andric Log *log = GetLog(LLDBLog::Script); 569*0b57cec5SDimitry Andric if (log) 570*0b57cec5SDimitry Andric log->PutCString("ScriptInterpreterPythonImpl::LeaveSession()"); 571*0b57cec5SDimitry Andric 5729dba64beSDimitry Andric // Unset the LLDB global variables. 5739dba64beSDimitry Andric PyRun_SimpleString("lldb.debugger = None; lldb.target = None; lldb.process " 5749dba64beSDimitry Andric "= None; lldb.thread = None; lldb.frame = None"); 5759dba64beSDimitry Andric 576*0b57cec5SDimitry Andric // checking that we have a valid thread state - since we use our own 577*0b57cec5SDimitry Andric // threading and locking in some (rare) cases during cleanup Python may end 578*0b57cec5SDimitry Andric // up believing we have no thread state and PyImport_AddModule will crash if 579*0b57cec5SDimitry Andric // that is the case - since that seems to only happen when destroying the 580*0b57cec5SDimitry Andric // SBDebugger, we can make do without clearing up stdout and stderr 581*0b57cec5SDimitry Andric 582*0b57cec5SDimitry Andric // rdar://problem/11292882 583*0b57cec5SDimitry Andric // When the current thread state is NULL, PyThreadState_Get() issues a fatal 584*0b57cec5SDimitry Andric // error. 585*0b57cec5SDimitry Andric if (PyThreadState_GetDict()) { 586*0b57cec5SDimitry Andric PythonDictionary &sys_module_dict = GetSysModuleDictionary(); 587*0b57cec5SDimitry Andric if (sys_module_dict.IsValid()) { 588*0b57cec5SDimitry Andric if (m_saved_stdin.IsValid()) { 589*0b57cec5SDimitry Andric sys_module_dict.SetItemForKey(PythonString("stdin"), m_saved_stdin); 590*0b57cec5SDimitry Andric m_saved_stdin.Reset(); 591*0b57cec5SDimitry Andric } 592*0b57cec5SDimitry Andric if (m_saved_stdout.IsValid()) { 593*0b57cec5SDimitry Andric sys_module_dict.SetItemForKey(PythonString("stdout"), m_saved_stdout); 594*0b57cec5SDimitry Andric m_saved_stdout.Reset(); 595*0b57cec5SDimitry Andric } 596*0b57cec5SDimitry Andric if (m_saved_stderr.IsValid()) { 597*0b57cec5SDimitry Andric sys_module_dict.SetItemForKey(PythonString("stderr"), m_saved_stderr); 598*0b57cec5SDimitry Andric m_saved_stderr.Reset(); 599*0b57cec5SDimitry Andric } 600*0b57cec5SDimitry Andric } 601*0b57cec5SDimitry Andric } 602*0b57cec5SDimitry Andric 603*0b57cec5SDimitry Andric m_session_is_active = false; 604*0b57cec5SDimitry Andric } 605*0b57cec5SDimitry Andric 6069dba64beSDimitry Andric bool ScriptInterpreterPythonImpl::SetStdHandle(FileSP file_sp, 6079dba64beSDimitry Andric const char *py_name, 6089dba64beSDimitry Andric PythonObject &save_file, 609*0b57cec5SDimitry Andric const char *mode) { 6109dba64beSDimitry Andric if (!file_sp || !*file_sp) { 6119dba64beSDimitry Andric save_file.Reset(); 6129dba64beSDimitry Andric return false; 6139dba64beSDimitry Andric } 6149dba64beSDimitry Andric File &file = *file_sp; 6159dba64beSDimitry Andric 616*0b57cec5SDimitry Andric // Flush the file before giving it to python to avoid interleaved output. 617*0b57cec5SDimitry Andric file.Flush(); 618*0b57cec5SDimitry Andric 619*0b57cec5SDimitry Andric PythonDictionary &sys_module_dict = GetSysModuleDictionary(); 620*0b57cec5SDimitry Andric 6219dba64beSDimitry Andric auto new_file = PythonFile::FromFile(file, mode); 6229dba64beSDimitry Andric if (!new_file) { 6239dba64beSDimitry Andric llvm::consumeError(new_file.takeError()); 624*0b57cec5SDimitry Andric return false; 625*0b57cec5SDimitry Andric } 626*0b57cec5SDimitry Andric 6279dba64beSDimitry Andric save_file = sys_module_dict.GetItemForKey(PythonString(py_name)); 6289dba64beSDimitry Andric 6299dba64beSDimitry Andric sys_module_dict.SetItemForKey(PythonString(py_name), new_file.get()); 6309dba64beSDimitry Andric return true; 6319dba64beSDimitry Andric } 6329dba64beSDimitry Andric 633*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::EnterSession(uint16_t on_entry_flags, 6349dba64beSDimitry Andric FileSP in_sp, FileSP out_sp, 6359dba64beSDimitry Andric FileSP err_sp) { 636*0b57cec5SDimitry Andric // If we have already entered the session, without having officially 'left' 637*0b57cec5SDimitry Andric // it, then there is no need to 'enter' it again. 63881ad6265SDimitry Andric Log *log = GetLog(LLDBLog::Script); 639*0b57cec5SDimitry Andric if (m_session_is_active) { 6409dba64beSDimitry Andric LLDB_LOGF( 6419dba64beSDimitry Andric log, 642*0b57cec5SDimitry Andric "ScriptInterpreterPythonImpl::EnterSession(on_entry_flags=0x%" PRIx16 643*0b57cec5SDimitry Andric ") session is already active, returning without doing anything", 644*0b57cec5SDimitry Andric on_entry_flags); 645*0b57cec5SDimitry Andric return false; 646*0b57cec5SDimitry Andric } 647*0b57cec5SDimitry Andric 6489dba64beSDimitry Andric LLDB_LOGF( 6499dba64beSDimitry Andric log, 6509dba64beSDimitry Andric "ScriptInterpreterPythonImpl::EnterSession(on_entry_flags=0x%" PRIx16 ")", 651*0b57cec5SDimitry Andric on_entry_flags); 652*0b57cec5SDimitry Andric 653*0b57cec5SDimitry Andric m_session_is_active = true; 654*0b57cec5SDimitry Andric 655*0b57cec5SDimitry Andric StreamString run_string; 656*0b57cec5SDimitry Andric 657*0b57cec5SDimitry Andric if (on_entry_flags & Locker::InitGlobals) { 658*0b57cec5SDimitry Andric run_string.Printf("run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64, 659*0b57cec5SDimitry Andric m_dictionary_name.c_str(), m_debugger.GetID()); 660*0b57cec5SDimitry Andric run_string.Printf( 661*0b57cec5SDimitry Andric "; lldb.debugger = lldb.SBDebugger.FindDebuggerWithID (%" PRIu64 ")", 662*0b57cec5SDimitry Andric m_debugger.GetID()); 663*0b57cec5SDimitry Andric run_string.PutCString("; lldb.target = lldb.debugger.GetSelectedTarget()"); 664*0b57cec5SDimitry Andric run_string.PutCString("; lldb.process = lldb.target.GetProcess()"); 665*0b57cec5SDimitry Andric run_string.PutCString("; lldb.thread = lldb.process.GetSelectedThread ()"); 666*0b57cec5SDimitry Andric run_string.PutCString("; lldb.frame = lldb.thread.GetSelectedFrame ()"); 667*0b57cec5SDimitry Andric run_string.PutCString("')"); 668*0b57cec5SDimitry Andric } else { 669*0b57cec5SDimitry Andric // If we aren't initing the globals, we should still always set the 670*0b57cec5SDimitry Andric // debugger (since that is always unique.) 671*0b57cec5SDimitry Andric run_string.Printf("run_one_line (%s, 'lldb.debugger_unique_id = %" PRIu64, 672*0b57cec5SDimitry Andric m_dictionary_name.c_str(), m_debugger.GetID()); 673*0b57cec5SDimitry Andric run_string.Printf( 674*0b57cec5SDimitry Andric "; lldb.debugger = lldb.SBDebugger.FindDebuggerWithID (%" PRIu64 ")", 675*0b57cec5SDimitry Andric m_debugger.GetID()); 676*0b57cec5SDimitry Andric run_string.PutCString("')"); 677*0b57cec5SDimitry Andric } 678*0b57cec5SDimitry Andric 679*0b57cec5SDimitry Andric PyRun_SimpleString(run_string.GetData()); 680*0b57cec5SDimitry Andric run_string.Clear(); 681*0b57cec5SDimitry Andric 682*0b57cec5SDimitry Andric PythonDictionary &sys_module_dict = GetSysModuleDictionary(); 683*0b57cec5SDimitry Andric if (sys_module_dict.IsValid()) { 6849dba64beSDimitry Andric lldb::FileSP top_in_sp; 6859dba64beSDimitry Andric lldb::StreamFileSP top_out_sp, top_err_sp; 6869dba64beSDimitry Andric if (!in_sp || !out_sp || !err_sp || !*in_sp || !*out_sp || !*err_sp) 6879dba64beSDimitry Andric m_debugger.AdoptTopIOHandlerFilesIfInvalid(top_in_sp, top_out_sp, 6889dba64beSDimitry Andric top_err_sp); 689*0b57cec5SDimitry Andric 690*0b57cec5SDimitry Andric if (on_entry_flags & Locker::NoSTDIN) { 691*0b57cec5SDimitry Andric m_saved_stdin.Reset(); 692*0b57cec5SDimitry Andric } else { 6939dba64beSDimitry Andric if (!SetStdHandle(in_sp, "stdin", m_saved_stdin, "r")) { 6949dba64beSDimitry Andric if (top_in_sp) 6959dba64beSDimitry Andric SetStdHandle(top_in_sp, "stdin", m_saved_stdin, "r"); 696*0b57cec5SDimitry Andric } 697*0b57cec5SDimitry Andric } 698*0b57cec5SDimitry Andric 6999dba64beSDimitry Andric if (!SetStdHandle(out_sp, "stdout", m_saved_stdout, "w")) { 7009dba64beSDimitry Andric if (top_out_sp) 7019dba64beSDimitry Andric SetStdHandle(top_out_sp->GetFileSP(), "stdout", m_saved_stdout, "w"); 702*0b57cec5SDimitry Andric } 703*0b57cec5SDimitry Andric 7049dba64beSDimitry Andric if (!SetStdHandle(err_sp, "stderr", m_saved_stderr, "w")) { 7059dba64beSDimitry Andric if (top_err_sp) 7069dba64beSDimitry Andric SetStdHandle(top_err_sp->GetFileSP(), "stderr", m_saved_stderr, "w"); 707*0b57cec5SDimitry Andric } 708*0b57cec5SDimitry Andric } 709*0b57cec5SDimitry Andric 710*0b57cec5SDimitry Andric if (PyErr_Occurred()) 711*0b57cec5SDimitry Andric PyErr_Clear(); 712*0b57cec5SDimitry Andric 713*0b57cec5SDimitry Andric return true; 714*0b57cec5SDimitry Andric } 715*0b57cec5SDimitry Andric 7169dba64beSDimitry Andric PythonModule &ScriptInterpreterPythonImpl::GetMainModule() { 717*0b57cec5SDimitry Andric if (!m_main_module.IsValid()) 7189dba64beSDimitry Andric m_main_module = unwrapIgnoringErrors(PythonModule::Import("__main__")); 719*0b57cec5SDimitry Andric return m_main_module; 720*0b57cec5SDimitry Andric } 721*0b57cec5SDimitry Andric 722*0b57cec5SDimitry Andric PythonDictionary &ScriptInterpreterPythonImpl::GetSessionDictionary() { 723*0b57cec5SDimitry Andric if (m_session_dict.IsValid()) 724*0b57cec5SDimitry Andric return m_session_dict; 725*0b57cec5SDimitry Andric 726*0b57cec5SDimitry Andric PythonObject &main_module = GetMainModule(); 727*0b57cec5SDimitry Andric if (!main_module.IsValid()) 728*0b57cec5SDimitry Andric return m_session_dict; 729*0b57cec5SDimitry Andric 730*0b57cec5SDimitry Andric PythonDictionary main_dict(PyRefType::Borrowed, 731*0b57cec5SDimitry Andric PyModule_GetDict(main_module.get())); 732*0b57cec5SDimitry Andric if (!main_dict.IsValid()) 733*0b57cec5SDimitry Andric return m_session_dict; 734*0b57cec5SDimitry Andric 7359dba64beSDimitry Andric m_session_dict = unwrapIgnoringErrors( 7369dba64beSDimitry Andric As<PythonDictionary>(main_dict.GetItem(m_dictionary_name))); 737*0b57cec5SDimitry Andric return m_session_dict; 738*0b57cec5SDimitry Andric } 739*0b57cec5SDimitry Andric 740*0b57cec5SDimitry Andric PythonDictionary &ScriptInterpreterPythonImpl::GetSysModuleDictionary() { 741*0b57cec5SDimitry Andric if (m_sys_module_dict.IsValid()) 742*0b57cec5SDimitry Andric return m_sys_module_dict; 7439dba64beSDimitry Andric PythonModule sys_module = unwrapIgnoringErrors(PythonModule::Import("sys")); 7449dba64beSDimitry Andric m_sys_module_dict = sys_module.GetDictionary(); 745*0b57cec5SDimitry Andric return m_sys_module_dict; 746*0b57cec5SDimitry Andric } 747*0b57cec5SDimitry Andric 748480093f4SDimitry Andric llvm::Expected<unsigned> 749480093f4SDimitry Andric ScriptInterpreterPythonImpl::GetMaxPositionalArgumentsForCallable( 750480093f4SDimitry Andric const llvm::StringRef &callable_name) { 751480093f4SDimitry Andric if (callable_name.empty()) { 752480093f4SDimitry Andric return llvm::createStringError( 753480093f4SDimitry Andric llvm::inconvertibleErrorCode(), 754480093f4SDimitry Andric "called with empty callable name."); 755480093f4SDimitry Andric } 756480093f4SDimitry Andric Locker py_lock(this, Locker::AcquireLock | 757480093f4SDimitry Andric Locker::InitSession | 758480093f4SDimitry Andric Locker::NoSTDIN); 759480093f4SDimitry Andric auto dict = PythonModule::MainModule() 760480093f4SDimitry Andric .ResolveName<PythonDictionary>(m_dictionary_name); 761480093f4SDimitry Andric auto pfunc = PythonObject::ResolveNameWithDictionary<PythonCallable>( 762480093f4SDimitry Andric callable_name, dict); 763480093f4SDimitry Andric if (!pfunc.IsAllocated()) { 764480093f4SDimitry Andric return llvm::createStringError( 765480093f4SDimitry Andric llvm::inconvertibleErrorCode(), 766480093f4SDimitry Andric "can't find callable: %s", callable_name.str().c_str()); 767480093f4SDimitry Andric } 768480093f4SDimitry Andric llvm::Expected<PythonCallable::ArgInfo> arg_info = pfunc.GetArgInfo(); 769480093f4SDimitry Andric if (!arg_info) 770480093f4SDimitry Andric return arg_info.takeError(); 771480093f4SDimitry Andric return arg_info.get().max_positional_args; 772480093f4SDimitry Andric } 773480093f4SDimitry Andric 774*0b57cec5SDimitry Andric static std::string GenerateUniqueName(const char *base_name_wanted, 775*0b57cec5SDimitry Andric uint32_t &functions_counter, 776*0b57cec5SDimitry Andric const void *name_token = nullptr) { 777*0b57cec5SDimitry Andric StreamString sstr; 778*0b57cec5SDimitry Andric 779*0b57cec5SDimitry Andric if (!base_name_wanted) 780*0b57cec5SDimitry Andric return std::string(); 781*0b57cec5SDimitry Andric 782*0b57cec5SDimitry Andric if (!name_token) 783*0b57cec5SDimitry Andric sstr.Printf("%s_%d", base_name_wanted, functions_counter++); 784*0b57cec5SDimitry Andric else 785*0b57cec5SDimitry Andric sstr.Printf("%s_%p", base_name_wanted, name_token); 786*0b57cec5SDimitry Andric 7875ffd83dbSDimitry Andric return std::string(sstr.GetString()); 788*0b57cec5SDimitry Andric } 789*0b57cec5SDimitry Andric 790*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GetEmbeddedInterpreterModuleObjects() { 791*0b57cec5SDimitry Andric if (m_run_one_line_function.IsValid()) 792*0b57cec5SDimitry Andric return true; 793*0b57cec5SDimitry Andric 794*0b57cec5SDimitry Andric PythonObject module(PyRefType::Borrowed, 795*0b57cec5SDimitry Andric PyImport_AddModule("lldb.embedded_interpreter")); 796*0b57cec5SDimitry Andric if (!module.IsValid()) 797*0b57cec5SDimitry Andric return false; 798*0b57cec5SDimitry Andric 799*0b57cec5SDimitry Andric PythonDictionary module_dict(PyRefType::Borrowed, 800*0b57cec5SDimitry Andric PyModule_GetDict(module.get())); 801*0b57cec5SDimitry Andric if (!module_dict.IsValid()) 802*0b57cec5SDimitry Andric return false; 803*0b57cec5SDimitry Andric 804*0b57cec5SDimitry Andric m_run_one_line_function = 805*0b57cec5SDimitry Andric module_dict.GetItemForKey(PythonString("run_one_line")); 806*0b57cec5SDimitry Andric m_run_one_line_str_global = 807*0b57cec5SDimitry Andric module_dict.GetItemForKey(PythonString("g_run_one_line_str")); 808*0b57cec5SDimitry Andric return m_run_one_line_function.IsValid(); 809*0b57cec5SDimitry Andric } 810*0b57cec5SDimitry Andric 811*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ExecuteOneLine( 812*0b57cec5SDimitry Andric llvm::StringRef command, CommandReturnObject *result, 813*0b57cec5SDimitry Andric const ExecuteScriptOptions &options) { 814*0b57cec5SDimitry Andric std::string command_str = command.str(); 815*0b57cec5SDimitry Andric 816*0b57cec5SDimitry Andric if (!m_valid_session) 817*0b57cec5SDimitry Andric return false; 818*0b57cec5SDimitry Andric 819*0b57cec5SDimitry Andric if (!command.empty()) { 820*0b57cec5SDimitry Andric // We want to call run_one_line, passing in the dictionary and the command 821*0b57cec5SDimitry Andric // string. We cannot do this through PyRun_SimpleString here because the 822*0b57cec5SDimitry Andric // command string may contain escaped characters, and putting it inside 823*0b57cec5SDimitry Andric // another string to pass to PyRun_SimpleString messes up the escaping. So 824*0b57cec5SDimitry Andric // we use the following more complicated method to pass the command string 825*0b57cec5SDimitry Andric // directly down to Python. 8265ffd83dbSDimitry Andric llvm::Expected<std::unique_ptr<ScriptInterpreterIORedirect>> 8275ffd83dbSDimitry Andric io_redirect_or_error = ScriptInterpreterIORedirect::Create( 8285ffd83dbSDimitry Andric options.GetEnableIO(), m_debugger, result); 8295ffd83dbSDimitry Andric if (!io_redirect_or_error) { 8305ffd83dbSDimitry Andric if (result) 8315ffd83dbSDimitry Andric result->AppendErrorWithFormatv( 8325ffd83dbSDimitry Andric "failed to redirect I/O: {0}\n", 8335ffd83dbSDimitry Andric llvm::fmt_consume(io_redirect_or_error.takeError())); 8345ffd83dbSDimitry Andric else 8355ffd83dbSDimitry Andric llvm::consumeError(io_redirect_or_error.takeError()); 8369dba64beSDimitry Andric return false; 8379dba64beSDimitry Andric } 8385ffd83dbSDimitry Andric 8395ffd83dbSDimitry Andric ScriptInterpreterIORedirect &io_redirect = **io_redirect_or_error; 840*0b57cec5SDimitry Andric 841*0b57cec5SDimitry Andric bool success = false; 842*0b57cec5SDimitry Andric { 843*0b57cec5SDimitry Andric // WARNING! It's imperative that this RAII scope be as tight as 844*0b57cec5SDimitry Andric // possible. In particular, the scope must end *before* we try to join 845*0b57cec5SDimitry Andric // the read thread. The reason for this is that a pre-requisite for 846*0b57cec5SDimitry Andric // joining the read thread is that we close the write handle (to break 847*0b57cec5SDimitry Andric // the pipe and cause it to wake up and exit). But acquiring the GIL as 848*0b57cec5SDimitry Andric // below will redirect Python's stdio to use this same handle. If we 849*0b57cec5SDimitry Andric // close the handle while Python is still using it, bad things will 850*0b57cec5SDimitry Andric // happen. 851*0b57cec5SDimitry Andric Locker locker( 852*0b57cec5SDimitry Andric this, 853*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | 854*0b57cec5SDimitry Andric (options.GetSetLLDBGlobals() ? Locker::InitGlobals : 0) | 855*0b57cec5SDimitry Andric ((result && result->GetInteractive()) ? 0 : Locker::NoSTDIN), 8565ffd83dbSDimitry Andric Locker::FreeAcquiredLock | Locker::TearDownSession, 8575ffd83dbSDimitry Andric io_redirect.GetInputFile(), io_redirect.GetOutputFile(), 8585ffd83dbSDimitry Andric io_redirect.GetErrorFile()); 859*0b57cec5SDimitry Andric 860*0b57cec5SDimitry Andric // Find the correct script interpreter dictionary in the main module. 861*0b57cec5SDimitry Andric PythonDictionary &session_dict = GetSessionDictionary(); 862*0b57cec5SDimitry Andric if (session_dict.IsValid()) { 863*0b57cec5SDimitry Andric if (GetEmbeddedInterpreterModuleObjects()) { 864*0b57cec5SDimitry Andric if (PyCallable_Check(m_run_one_line_function.get())) { 865*0b57cec5SDimitry Andric PythonObject pargs( 866*0b57cec5SDimitry Andric PyRefType::Owned, 867*0b57cec5SDimitry Andric Py_BuildValue("(Os)", session_dict.get(), command_str.c_str())); 868*0b57cec5SDimitry Andric if (pargs.IsValid()) { 869*0b57cec5SDimitry Andric PythonObject return_value( 870*0b57cec5SDimitry Andric PyRefType::Owned, 871*0b57cec5SDimitry Andric PyObject_CallObject(m_run_one_line_function.get(), 872*0b57cec5SDimitry Andric pargs.get())); 873*0b57cec5SDimitry Andric if (return_value.IsValid()) 874*0b57cec5SDimitry Andric success = true; 875*0b57cec5SDimitry Andric else if (options.GetMaskoutErrors() && PyErr_Occurred()) { 876*0b57cec5SDimitry Andric PyErr_Print(); 877*0b57cec5SDimitry Andric PyErr_Clear(); 878*0b57cec5SDimitry Andric } 879*0b57cec5SDimitry Andric } 880*0b57cec5SDimitry Andric } 881*0b57cec5SDimitry Andric } 882*0b57cec5SDimitry Andric } 883*0b57cec5SDimitry Andric 8845ffd83dbSDimitry Andric io_redirect.Flush(); 885*0b57cec5SDimitry Andric } 886*0b57cec5SDimitry Andric 887*0b57cec5SDimitry Andric if (success) 888*0b57cec5SDimitry Andric return true; 889*0b57cec5SDimitry Andric 890*0b57cec5SDimitry Andric // The one-liner failed. Append the error message. 891*0b57cec5SDimitry Andric if (result) { 892*0b57cec5SDimitry Andric result->AppendErrorWithFormat( 893*0b57cec5SDimitry Andric "python failed attempting to evaluate '%s'\n", command_str.c_str()); 894*0b57cec5SDimitry Andric } 895*0b57cec5SDimitry Andric return false; 896*0b57cec5SDimitry Andric } 897*0b57cec5SDimitry Andric 898*0b57cec5SDimitry Andric if (result) 899*0b57cec5SDimitry Andric result->AppendError("empty command passed to python\n"); 900*0b57cec5SDimitry Andric return false; 901*0b57cec5SDimitry Andric } 902*0b57cec5SDimitry Andric 903*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::ExecuteInterpreterLoop() { 904e8d8bef9SDimitry Andric LLDB_SCOPED_TIMER(); 905*0b57cec5SDimitry Andric 906*0b57cec5SDimitry Andric Debugger &debugger = m_debugger; 907*0b57cec5SDimitry Andric 908*0b57cec5SDimitry Andric // At the moment, the only time the debugger does not have an input file 909*0b57cec5SDimitry Andric // handle is when this is called directly from Python, in which case it is 910*0b57cec5SDimitry Andric // both dangerous and unnecessary (not to mention confusing) to try to embed 911*0b57cec5SDimitry Andric // a running interpreter loop inside the already running Python interpreter 912*0b57cec5SDimitry Andric // loop, so we won't do it. 913*0b57cec5SDimitry Andric 9149dba64beSDimitry Andric if (!debugger.GetInputFile().IsValid()) 915*0b57cec5SDimitry Andric return; 916*0b57cec5SDimitry Andric 917*0b57cec5SDimitry Andric IOHandlerSP io_handler_sp(new IOHandlerPythonInterpreter(debugger, this)); 918*0b57cec5SDimitry Andric if (io_handler_sp) { 9195ffd83dbSDimitry Andric debugger.RunIOHandlerAsync(io_handler_sp); 920*0b57cec5SDimitry Andric } 921*0b57cec5SDimitry Andric } 922*0b57cec5SDimitry Andric 923*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::Interrupt() { 92404eeddc0SDimitry Andric #if LLDB_USE_PYTHON_SET_INTERRUPT 92504eeddc0SDimitry Andric // If the interpreter isn't evaluating any Python at the moment then return 92604eeddc0SDimitry Andric // false to signal that this function didn't handle the interrupt and the 92704eeddc0SDimitry Andric // next component should try handling it. 92804eeddc0SDimitry Andric if (!IsExecutingPython()) 92904eeddc0SDimitry Andric return false; 93004eeddc0SDimitry Andric 93104eeddc0SDimitry Andric // Tell Python that it should pretend to have received a SIGINT. 93204eeddc0SDimitry Andric PyErr_SetInterrupt(); 93304eeddc0SDimitry Andric // PyErr_SetInterrupt has no way to return an error so we can only pretend the 93404eeddc0SDimitry Andric // signal got successfully handled and return true. 93504eeddc0SDimitry Andric // Python 3.10 introduces PyErr_SetInterruptEx that could return an error, but 93604eeddc0SDimitry Andric // the error handling is limited to checking the arguments which would be 93704eeddc0SDimitry Andric // just our (hardcoded) input signal code SIGINT, so that's not useful at all. 93804eeddc0SDimitry Andric return true; 93904eeddc0SDimitry Andric #else 94081ad6265SDimitry Andric Log *log = GetLog(LLDBLog::Script); 941*0b57cec5SDimitry Andric 942*0b57cec5SDimitry Andric if (IsExecutingPython()) { 943*0b57cec5SDimitry Andric PyThreadState *state = PyThreadState_GET(); 944*0b57cec5SDimitry Andric if (!state) 945*0b57cec5SDimitry Andric state = GetThreadState(); 946*0b57cec5SDimitry Andric if (state) { 947*0b57cec5SDimitry Andric long tid = state->thread_id; 948*0b57cec5SDimitry Andric PyThreadState_Swap(state); 949*0b57cec5SDimitry Andric int num_threads = PyThreadState_SetAsyncExc(tid, PyExc_KeyboardInterrupt); 9509dba64beSDimitry Andric LLDB_LOGF(log, 9519dba64beSDimitry Andric "ScriptInterpreterPythonImpl::Interrupt() sending " 952*0b57cec5SDimitry Andric "PyExc_KeyboardInterrupt (tid = %li, num_threads = %i)...", 953*0b57cec5SDimitry Andric tid, num_threads); 954*0b57cec5SDimitry Andric return true; 955*0b57cec5SDimitry Andric } 956*0b57cec5SDimitry Andric } 9579dba64beSDimitry Andric LLDB_LOGF(log, 958*0b57cec5SDimitry Andric "ScriptInterpreterPythonImpl::Interrupt() python code not running, " 959*0b57cec5SDimitry Andric "can't interrupt"); 960*0b57cec5SDimitry Andric return false; 96104eeddc0SDimitry Andric #endif 962*0b57cec5SDimitry Andric } 9639dba64beSDimitry Andric 964*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ExecuteOneLineWithReturn( 965*0b57cec5SDimitry Andric llvm::StringRef in_string, ScriptInterpreter::ScriptReturnType return_type, 966*0b57cec5SDimitry Andric void *ret_value, const ExecuteScriptOptions &options) { 967*0b57cec5SDimitry Andric 968fe6060f1SDimitry Andric llvm::Expected<std::unique_ptr<ScriptInterpreterIORedirect>> 969fe6060f1SDimitry Andric io_redirect_or_error = ScriptInterpreterIORedirect::Create( 970fe6060f1SDimitry Andric options.GetEnableIO(), m_debugger, /*result=*/nullptr); 971fe6060f1SDimitry Andric 972fe6060f1SDimitry Andric if (!io_redirect_or_error) { 973fe6060f1SDimitry Andric llvm::consumeError(io_redirect_or_error.takeError()); 974fe6060f1SDimitry Andric return false; 975fe6060f1SDimitry Andric } 976fe6060f1SDimitry Andric 977fe6060f1SDimitry Andric ScriptInterpreterIORedirect &io_redirect = **io_redirect_or_error; 978fe6060f1SDimitry Andric 979*0b57cec5SDimitry Andric Locker locker(this, 980*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | 981*0b57cec5SDimitry Andric (options.GetSetLLDBGlobals() ? Locker::InitGlobals : 0) | 982*0b57cec5SDimitry Andric Locker::NoSTDIN, 983fe6060f1SDimitry Andric Locker::FreeAcquiredLock | Locker::TearDownSession, 984fe6060f1SDimitry Andric io_redirect.GetInputFile(), io_redirect.GetOutputFile(), 985fe6060f1SDimitry Andric io_redirect.GetErrorFile()); 986*0b57cec5SDimitry Andric 9879dba64beSDimitry Andric PythonModule &main_module = GetMainModule(); 9889dba64beSDimitry Andric PythonDictionary globals = main_module.GetDictionary(); 989*0b57cec5SDimitry Andric 990*0b57cec5SDimitry Andric PythonDictionary locals = GetSessionDictionary(); 9919dba64beSDimitry Andric if (!locals.IsValid()) 9929dba64beSDimitry Andric locals = unwrapIgnoringErrors( 9939dba64beSDimitry Andric As<PythonDictionary>(globals.GetAttribute(m_dictionary_name))); 994*0b57cec5SDimitry Andric if (!locals.IsValid()) 995*0b57cec5SDimitry Andric locals = globals; 996*0b57cec5SDimitry Andric 9979dba64beSDimitry Andric Expected<PythonObject> maybe_py_return = 9989dba64beSDimitry Andric runStringOneLine(in_string, globals, locals); 999*0b57cec5SDimitry Andric 10009dba64beSDimitry Andric if (!maybe_py_return) { 10019dba64beSDimitry Andric llvm::handleAllErrors( 10029dba64beSDimitry Andric maybe_py_return.takeError(), 10039dba64beSDimitry Andric [&](PythonException &E) { 10049dba64beSDimitry Andric E.Restore(); 10059dba64beSDimitry Andric if (options.GetMaskoutErrors()) { 10069dba64beSDimitry Andric if (E.Matches(PyExc_SyntaxError)) { 10079dba64beSDimitry Andric PyErr_Print(); 1008*0b57cec5SDimitry Andric } 10099dba64beSDimitry Andric PyErr_Clear(); 10109dba64beSDimitry Andric } 10119dba64beSDimitry Andric }, 10129dba64beSDimitry Andric [](const llvm::ErrorInfoBase &E) {}); 10139dba64beSDimitry Andric return false; 1014*0b57cec5SDimitry Andric } 1015*0b57cec5SDimitry Andric 10169dba64beSDimitry Andric PythonObject py_return = std::move(maybe_py_return.get()); 10179dba64beSDimitry Andric assert(py_return.IsValid()); 10189dba64beSDimitry Andric 1019*0b57cec5SDimitry Andric switch (return_type) { 1020*0b57cec5SDimitry Andric case eScriptReturnTypeCharPtr: // "char *" 1021*0b57cec5SDimitry Andric { 1022*0b57cec5SDimitry Andric const char format[3] = "s#"; 10239dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (char **)ret_value); 1024*0b57cec5SDimitry Andric } 1025*0b57cec5SDimitry Andric case eScriptReturnTypeCharStrOrNone: // char* or NULL if py_return == 1026*0b57cec5SDimitry Andric // Py_None 1027*0b57cec5SDimitry Andric { 1028*0b57cec5SDimitry Andric const char format[3] = "z"; 10299dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (char **)ret_value); 1030*0b57cec5SDimitry Andric } 1031*0b57cec5SDimitry Andric case eScriptReturnTypeBool: { 1032*0b57cec5SDimitry Andric const char format[2] = "b"; 10339dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (bool *)ret_value); 1034*0b57cec5SDimitry Andric } 1035*0b57cec5SDimitry Andric case eScriptReturnTypeShortInt: { 1036*0b57cec5SDimitry Andric const char format[2] = "h"; 10379dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (short *)ret_value); 1038*0b57cec5SDimitry Andric } 1039*0b57cec5SDimitry Andric case eScriptReturnTypeShortIntUnsigned: { 1040*0b57cec5SDimitry Andric const char format[2] = "H"; 10419dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (unsigned short *)ret_value); 1042*0b57cec5SDimitry Andric } 1043*0b57cec5SDimitry Andric case eScriptReturnTypeInt: { 1044*0b57cec5SDimitry Andric const char format[2] = "i"; 10459dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (int *)ret_value); 1046*0b57cec5SDimitry Andric } 1047*0b57cec5SDimitry Andric case eScriptReturnTypeIntUnsigned: { 1048*0b57cec5SDimitry Andric const char format[2] = "I"; 10499dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (unsigned int *)ret_value); 1050*0b57cec5SDimitry Andric } 1051*0b57cec5SDimitry Andric case eScriptReturnTypeLongInt: { 1052*0b57cec5SDimitry Andric const char format[2] = "l"; 10539dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (long *)ret_value); 1054*0b57cec5SDimitry Andric } 1055*0b57cec5SDimitry Andric case eScriptReturnTypeLongIntUnsigned: { 1056*0b57cec5SDimitry Andric const char format[2] = "k"; 10579dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (unsigned long *)ret_value); 1058*0b57cec5SDimitry Andric } 1059*0b57cec5SDimitry Andric case eScriptReturnTypeLongLong: { 1060*0b57cec5SDimitry Andric const char format[2] = "L"; 10619dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (long long *)ret_value); 1062*0b57cec5SDimitry Andric } 1063*0b57cec5SDimitry Andric case eScriptReturnTypeLongLongUnsigned: { 1064*0b57cec5SDimitry Andric const char format[2] = "K"; 10659dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, 10669dba64beSDimitry Andric (unsigned long long *)ret_value); 1067*0b57cec5SDimitry Andric } 1068*0b57cec5SDimitry Andric case eScriptReturnTypeFloat: { 1069*0b57cec5SDimitry Andric const char format[2] = "f"; 10709dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (float *)ret_value); 1071*0b57cec5SDimitry Andric } 1072*0b57cec5SDimitry Andric case eScriptReturnTypeDouble: { 1073*0b57cec5SDimitry Andric const char format[2] = "d"; 10749dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (double *)ret_value); 1075*0b57cec5SDimitry Andric } 1076*0b57cec5SDimitry Andric case eScriptReturnTypeChar: { 1077*0b57cec5SDimitry Andric const char format[2] = "c"; 10789dba64beSDimitry Andric return PyArg_Parse(py_return.get(), format, (char *)ret_value); 1079*0b57cec5SDimitry Andric } 1080*0b57cec5SDimitry Andric case eScriptReturnTypeOpaqueObject: { 10819dba64beSDimitry Andric *((PyObject **)ret_value) = py_return.release(); 10829dba64beSDimitry Andric return true; 1083*0b57cec5SDimitry Andric } 1084*0b57cec5SDimitry Andric } 1085480093f4SDimitry Andric llvm_unreachable("Fully covered switch!"); 1086*0b57cec5SDimitry Andric } 1087*0b57cec5SDimitry Andric 1088*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::ExecuteMultipleLines( 1089*0b57cec5SDimitry Andric const char *in_string, const ExecuteScriptOptions &options) { 10909dba64beSDimitry Andric 10919dba64beSDimitry Andric if (in_string == nullptr) 10929dba64beSDimitry Andric return Status(); 1093*0b57cec5SDimitry Andric 1094fe6060f1SDimitry Andric llvm::Expected<std::unique_ptr<ScriptInterpreterIORedirect>> 1095fe6060f1SDimitry Andric io_redirect_or_error = ScriptInterpreterIORedirect::Create( 1096fe6060f1SDimitry Andric options.GetEnableIO(), m_debugger, /*result=*/nullptr); 1097fe6060f1SDimitry Andric 1098fe6060f1SDimitry Andric if (!io_redirect_or_error) 1099fe6060f1SDimitry Andric return Status(io_redirect_or_error.takeError()); 1100fe6060f1SDimitry Andric 1101fe6060f1SDimitry Andric ScriptInterpreterIORedirect &io_redirect = **io_redirect_or_error; 1102fe6060f1SDimitry Andric 1103*0b57cec5SDimitry Andric Locker locker(this, 1104*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | 1105*0b57cec5SDimitry Andric (options.GetSetLLDBGlobals() ? Locker::InitGlobals : 0) | 1106*0b57cec5SDimitry Andric Locker::NoSTDIN, 1107fe6060f1SDimitry Andric Locker::FreeAcquiredLock | Locker::TearDownSession, 1108fe6060f1SDimitry Andric io_redirect.GetInputFile(), io_redirect.GetOutputFile(), 1109fe6060f1SDimitry Andric io_redirect.GetErrorFile()); 1110*0b57cec5SDimitry Andric 11119dba64beSDimitry Andric PythonModule &main_module = GetMainModule(); 11129dba64beSDimitry Andric PythonDictionary globals = main_module.GetDictionary(); 1113*0b57cec5SDimitry Andric 1114*0b57cec5SDimitry Andric PythonDictionary locals = GetSessionDictionary(); 1115*0b57cec5SDimitry Andric if (!locals.IsValid()) 11169dba64beSDimitry Andric locals = unwrapIgnoringErrors( 11179dba64beSDimitry Andric As<PythonDictionary>(globals.GetAttribute(m_dictionary_name))); 1118*0b57cec5SDimitry Andric if (!locals.IsValid()) 1119*0b57cec5SDimitry Andric locals = globals; 1120*0b57cec5SDimitry Andric 11219dba64beSDimitry Andric Expected<PythonObject> return_value = 11229dba64beSDimitry Andric runStringMultiLine(in_string, globals, locals); 1123*0b57cec5SDimitry Andric 11249dba64beSDimitry Andric if (!return_value) { 11259dba64beSDimitry Andric llvm::Error error = 11269dba64beSDimitry Andric llvm::handleErrors(return_value.takeError(), [&](PythonException &E) { 11279dba64beSDimitry Andric llvm::Error error = llvm::createStringError( 11289dba64beSDimitry Andric llvm::inconvertibleErrorCode(), E.ReadBacktrace()); 11299dba64beSDimitry Andric if (!options.GetMaskoutErrors()) 11309dba64beSDimitry Andric E.Restore(); 1131*0b57cec5SDimitry Andric return error; 11329dba64beSDimitry Andric }); 11339dba64beSDimitry Andric return Status(std::move(error)); 11349dba64beSDimitry Andric } 11359dba64beSDimitry Andric 11369dba64beSDimitry Andric return Status(); 1137*0b57cec5SDimitry Andric } 1138*0b57cec5SDimitry Andric 1139*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::CollectDataForBreakpointCommandCallback( 1140fe6060f1SDimitry Andric std::vector<std::reference_wrapper<BreakpointOptions>> &bp_options_vec, 1141*0b57cec5SDimitry Andric CommandReturnObject &result) { 1142*0b57cec5SDimitry Andric m_active_io_handler = eIOHandlerBreakpoint; 1143*0b57cec5SDimitry Andric m_debugger.GetCommandInterpreter().GetPythonCommandsFromIOHandler( 1144480093f4SDimitry Andric " ", *this, &bp_options_vec); 1145*0b57cec5SDimitry Andric } 1146*0b57cec5SDimitry Andric 1147*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::CollectDataForWatchpointCommandCallback( 1148*0b57cec5SDimitry Andric WatchpointOptions *wp_options, CommandReturnObject &result) { 1149*0b57cec5SDimitry Andric m_active_io_handler = eIOHandlerWatchpoint; 1150*0b57cec5SDimitry Andric m_debugger.GetCommandInterpreter().GetPythonCommandsFromIOHandler( 1151480093f4SDimitry Andric " ", *this, wp_options); 1152*0b57cec5SDimitry Andric } 1153*0b57cec5SDimitry Andric 1154480093f4SDimitry Andric Status ScriptInterpreterPythonImpl::SetBreakpointCommandCallbackFunction( 1155fe6060f1SDimitry Andric BreakpointOptions &bp_options, const char *function_name, 1156480093f4SDimitry Andric StructuredData::ObjectSP extra_args_sp) { 1157480093f4SDimitry Andric Status error; 1158*0b57cec5SDimitry Andric // For now just cons up a oneliner that calls the provided function. 1159*0b57cec5SDimitry Andric std::string oneliner("return "); 1160*0b57cec5SDimitry Andric oneliner += function_name; 1161480093f4SDimitry Andric 1162480093f4SDimitry Andric llvm::Expected<unsigned> maybe_args = 1163480093f4SDimitry Andric GetMaxPositionalArgumentsForCallable(function_name); 1164480093f4SDimitry Andric if (!maybe_args) { 1165480093f4SDimitry Andric error.SetErrorStringWithFormat( 1166480093f4SDimitry Andric "could not get num args: %s", 1167480093f4SDimitry Andric llvm::toString(maybe_args.takeError()).c_str()); 1168480093f4SDimitry Andric return error; 1169480093f4SDimitry Andric } 1170480093f4SDimitry Andric size_t max_args = *maybe_args; 1171480093f4SDimitry Andric 1172480093f4SDimitry Andric bool uses_extra_args = false; 1173480093f4SDimitry Andric if (max_args >= 4) { 1174480093f4SDimitry Andric uses_extra_args = true; 1175480093f4SDimitry Andric oneliner += "(frame, bp_loc, extra_args, internal_dict)"; 1176480093f4SDimitry Andric } else if (max_args >= 3) { 1177480093f4SDimitry Andric if (extra_args_sp) { 1178480093f4SDimitry Andric error.SetErrorString("cannot pass extra_args to a three argument callback" 1179480093f4SDimitry Andric ); 1180480093f4SDimitry Andric return error; 1181480093f4SDimitry Andric } 1182480093f4SDimitry Andric uses_extra_args = false; 1183*0b57cec5SDimitry Andric oneliner += "(frame, bp_loc, internal_dict)"; 1184480093f4SDimitry Andric } else { 1185480093f4SDimitry Andric error.SetErrorStringWithFormat("expected 3 or 4 argument " 1186480093f4SDimitry Andric "function, %s can only take %zu", 1187480093f4SDimitry Andric function_name, max_args); 1188480093f4SDimitry Andric return error; 1189480093f4SDimitry Andric } 1190480093f4SDimitry Andric 1191480093f4SDimitry Andric SetBreakpointCommandCallback(bp_options, oneliner.c_str(), extra_args_sp, 1192480093f4SDimitry Andric uses_extra_args); 1193480093f4SDimitry Andric return error; 1194*0b57cec5SDimitry Andric } 1195*0b57cec5SDimitry Andric 1196*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::SetBreakpointCommandCallback( 1197fe6060f1SDimitry Andric BreakpointOptions &bp_options, 1198*0b57cec5SDimitry Andric std::unique_ptr<BreakpointOptions::CommandData> &cmd_data_up) { 1199*0b57cec5SDimitry Andric Status error; 1200*0b57cec5SDimitry Andric error = GenerateBreakpointCommandCallbackData(cmd_data_up->user_source, 1201480093f4SDimitry Andric cmd_data_up->script_source, 1202480093f4SDimitry Andric false); 1203*0b57cec5SDimitry Andric if (error.Fail()) { 1204*0b57cec5SDimitry Andric return error; 1205*0b57cec5SDimitry Andric } 1206*0b57cec5SDimitry Andric auto baton_sp = 1207*0b57cec5SDimitry Andric std::make_shared<BreakpointOptions::CommandBaton>(std::move(cmd_data_up)); 1208fe6060f1SDimitry Andric bp_options.SetCallback( 1209*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::BreakpointCallbackFunction, baton_sp); 1210*0b57cec5SDimitry Andric return error; 1211*0b57cec5SDimitry Andric } 1212*0b57cec5SDimitry Andric 1213*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::SetBreakpointCommandCallback( 1214fe6060f1SDimitry Andric BreakpointOptions &bp_options, const char *command_body_text) { 1215480093f4SDimitry Andric return SetBreakpointCommandCallback(bp_options, command_body_text, {},false); 1216480093f4SDimitry Andric } 1217*0b57cec5SDimitry Andric 1218480093f4SDimitry Andric // Set a Python one-liner as the callback for the breakpoint. 1219480093f4SDimitry Andric Status ScriptInterpreterPythonImpl::SetBreakpointCommandCallback( 1220fe6060f1SDimitry Andric BreakpointOptions &bp_options, const char *command_body_text, 1221fe6060f1SDimitry Andric StructuredData::ObjectSP extra_args_sp, bool uses_extra_args) { 1222480093f4SDimitry Andric auto data_up = std::make_unique<CommandDataPython>(extra_args_sp); 1223*0b57cec5SDimitry Andric // Split the command_body_text into lines, and pass that to 1224*0b57cec5SDimitry Andric // GenerateBreakpointCommandCallbackData. That will wrap the body in an 1225*0b57cec5SDimitry Andric // auto-generated function, and return the function name in script_source. 1226*0b57cec5SDimitry Andric // That is what the callback will actually invoke. 1227*0b57cec5SDimitry Andric 1228*0b57cec5SDimitry Andric data_up->user_source.SplitIntoLines(command_body_text); 1229*0b57cec5SDimitry Andric Status error = GenerateBreakpointCommandCallbackData(data_up->user_source, 1230480093f4SDimitry Andric data_up->script_source, 1231480093f4SDimitry Andric uses_extra_args); 1232*0b57cec5SDimitry Andric if (error.Success()) { 1233*0b57cec5SDimitry Andric auto baton_sp = 1234*0b57cec5SDimitry Andric std::make_shared<BreakpointOptions::CommandBaton>(std::move(data_up)); 1235fe6060f1SDimitry Andric bp_options.SetCallback( 1236*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::BreakpointCallbackFunction, baton_sp); 1237*0b57cec5SDimitry Andric return error; 12385ffd83dbSDimitry Andric } 1239*0b57cec5SDimitry Andric return error; 1240*0b57cec5SDimitry Andric } 1241*0b57cec5SDimitry Andric 1242*0b57cec5SDimitry Andric // Set a Python one-liner as the callback for the watchpoint. 1243*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::SetWatchpointCommandCallback( 1244*0b57cec5SDimitry Andric WatchpointOptions *wp_options, const char *oneliner) { 12459dba64beSDimitry Andric auto data_up = std::make_unique<WatchpointOptions::CommandData>(); 1246*0b57cec5SDimitry Andric 1247*0b57cec5SDimitry Andric // It's necessary to set both user_source and script_source to the oneliner. 1248*0b57cec5SDimitry Andric // The former is used to generate callback description (as in watchpoint 1249*0b57cec5SDimitry Andric // command list) while the latter is used for Python to interpret during the 1250*0b57cec5SDimitry Andric // actual callback. 1251*0b57cec5SDimitry Andric 1252*0b57cec5SDimitry Andric data_up->user_source.AppendString(oneliner); 1253*0b57cec5SDimitry Andric data_up->script_source.assign(oneliner); 1254*0b57cec5SDimitry Andric 1255*0b57cec5SDimitry Andric if (GenerateWatchpointCommandCallbackData(data_up->user_source, 1256*0b57cec5SDimitry Andric data_up->script_source)) { 1257*0b57cec5SDimitry Andric auto baton_sp = 1258*0b57cec5SDimitry Andric std::make_shared<WatchpointOptions::CommandBaton>(std::move(data_up)); 1259*0b57cec5SDimitry Andric wp_options->SetCallback( 1260*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::WatchpointCallbackFunction, baton_sp); 1261*0b57cec5SDimitry Andric } 1262*0b57cec5SDimitry Andric 1263*0b57cec5SDimitry Andric return; 1264*0b57cec5SDimitry Andric } 1265*0b57cec5SDimitry Andric 1266*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::ExportFunctionDefinitionToInterpreter( 1267*0b57cec5SDimitry Andric StringList &function_def) { 1268*0b57cec5SDimitry Andric // Convert StringList to one long, newline delimited, const char *. 1269*0b57cec5SDimitry Andric std::string function_def_string(function_def.CopyList()); 1270*0b57cec5SDimitry Andric 1271*0b57cec5SDimitry Andric Status error = ExecuteMultipleLines( 1272*0b57cec5SDimitry Andric function_def_string.c_str(), 1273fe6060f1SDimitry Andric ExecuteScriptOptions().SetEnableIO(false)); 1274*0b57cec5SDimitry Andric return error; 1275*0b57cec5SDimitry Andric } 1276*0b57cec5SDimitry Andric 1277*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::GenerateFunction(const char *signature, 1278*0b57cec5SDimitry Andric const StringList &input) { 1279*0b57cec5SDimitry Andric Status error; 1280*0b57cec5SDimitry Andric int num_lines = input.GetSize(); 1281*0b57cec5SDimitry Andric if (num_lines == 0) { 1282*0b57cec5SDimitry Andric error.SetErrorString("No input data."); 1283*0b57cec5SDimitry Andric return error; 1284*0b57cec5SDimitry Andric } 1285*0b57cec5SDimitry Andric 1286*0b57cec5SDimitry Andric if (!signature || *signature == 0) { 1287*0b57cec5SDimitry Andric error.SetErrorString("No output function name."); 1288*0b57cec5SDimitry Andric return error; 1289*0b57cec5SDimitry Andric } 1290*0b57cec5SDimitry Andric 1291*0b57cec5SDimitry Andric StreamString sstr; 1292*0b57cec5SDimitry Andric StringList auto_generated_function; 1293*0b57cec5SDimitry Andric auto_generated_function.AppendString(signature); 1294*0b57cec5SDimitry Andric auto_generated_function.AppendString( 1295*0b57cec5SDimitry Andric " global_dict = globals()"); // Grab the global dictionary 1296*0b57cec5SDimitry Andric auto_generated_function.AppendString( 1297*0b57cec5SDimitry Andric " new_keys = internal_dict.keys()"); // Make a list of keys in the 1298*0b57cec5SDimitry Andric // session dict 1299*0b57cec5SDimitry Andric auto_generated_function.AppendString( 1300*0b57cec5SDimitry Andric " old_keys = global_dict.keys()"); // Save list of keys in global dict 1301*0b57cec5SDimitry Andric auto_generated_function.AppendString( 1302*0b57cec5SDimitry Andric " global_dict.update (internal_dict)"); // Add the session dictionary 1303*0b57cec5SDimitry Andric // to the 1304*0b57cec5SDimitry Andric // global dictionary. 1305*0b57cec5SDimitry Andric 1306*0b57cec5SDimitry Andric // Wrap everything up inside the function, increasing the indentation. 1307*0b57cec5SDimitry Andric 1308*0b57cec5SDimitry Andric auto_generated_function.AppendString(" if True:"); 1309*0b57cec5SDimitry Andric for (int i = 0; i < num_lines; ++i) { 1310*0b57cec5SDimitry Andric sstr.Clear(); 1311*0b57cec5SDimitry Andric sstr.Printf(" %s", input.GetStringAtIndex(i)); 1312*0b57cec5SDimitry Andric auto_generated_function.AppendString(sstr.GetData()); 1313*0b57cec5SDimitry Andric } 1314*0b57cec5SDimitry Andric auto_generated_function.AppendString( 1315*0b57cec5SDimitry Andric " for key in new_keys:"); // Iterate over all the keys from session 1316*0b57cec5SDimitry Andric // dict 1317*0b57cec5SDimitry Andric auto_generated_function.AppendString( 1318*0b57cec5SDimitry Andric " internal_dict[key] = global_dict[key]"); // Update session dict 1319*0b57cec5SDimitry Andric // values 1320*0b57cec5SDimitry Andric auto_generated_function.AppendString( 1321*0b57cec5SDimitry Andric " if key not in old_keys:"); // If key was not originally in 1322*0b57cec5SDimitry Andric // global dict 1323*0b57cec5SDimitry Andric auto_generated_function.AppendString( 1324*0b57cec5SDimitry Andric " del global_dict[key]"); // ...then remove key/value from 1325*0b57cec5SDimitry Andric // global dict 1326*0b57cec5SDimitry Andric 1327*0b57cec5SDimitry Andric // Verify that the results are valid Python. 1328*0b57cec5SDimitry Andric 1329*0b57cec5SDimitry Andric error = ExportFunctionDefinitionToInterpreter(auto_generated_function); 1330*0b57cec5SDimitry Andric 1331*0b57cec5SDimitry Andric return error; 1332*0b57cec5SDimitry Andric } 1333*0b57cec5SDimitry Andric 1334*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateTypeScriptFunction( 1335*0b57cec5SDimitry Andric StringList &user_input, std::string &output, const void *name_token) { 1336*0b57cec5SDimitry Andric static uint32_t num_created_functions = 0; 1337*0b57cec5SDimitry Andric user_input.RemoveBlankLines(); 1338*0b57cec5SDimitry Andric StreamString sstr; 1339*0b57cec5SDimitry Andric 1340*0b57cec5SDimitry Andric // Check to see if we have any data; if not, just return. 1341*0b57cec5SDimitry Andric if (user_input.GetSize() == 0) 1342*0b57cec5SDimitry Andric return false; 1343*0b57cec5SDimitry Andric 1344*0b57cec5SDimitry Andric // Take what the user wrote, wrap it all up inside one big auto-generated 1345*0b57cec5SDimitry Andric // Python function, passing in the ValueObject as parameter to the function. 1346*0b57cec5SDimitry Andric 1347*0b57cec5SDimitry Andric std::string auto_generated_function_name( 1348*0b57cec5SDimitry Andric GenerateUniqueName("lldb_autogen_python_type_print_func", 1349*0b57cec5SDimitry Andric num_created_functions, name_token)); 1350*0b57cec5SDimitry Andric sstr.Printf("def %s (valobj, internal_dict):", 1351*0b57cec5SDimitry Andric auto_generated_function_name.c_str()); 1352*0b57cec5SDimitry Andric 1353*0b57cec5SDimitry Andric if (!GenerateFunction(sstr.GetData(), user_input).Success()) 1354*0b57cec5SDimitry Andric return false; 1355*0b57cec5SDimitry Andric 1356*0b57cec5SDimitry Andric // Store the name of the auto-generated function to be called. 1357*0b57cec5SDimitry Andric output.assign(auto_generated_function_name); 1358*0b57cec5SDimitry Andric return true; 1359*0b57cec5SDimitry Andric } 1360*0b57cec5SDimitry Andric 1361*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateScriptAliasFunction( 1362*0b57cec5SDimitry Andric StringList &user_input, std::string &output) { 1363*0b57cec5SDimitry Andric static uint32_t num_created_functions = 0; 1364*0b57cec5SDimitry Andric user_input.RemoveBlankLines(); 1365*0b57cec5SDimitry Andric StreamString sstr; 1366*0b57cec5SDimitry Andric 1367*0b57cec5SDimitry Andric // Check to see if we have any data; if not, just return. 1368*0b57cec5SDimitry Andric if (user_input.GetSize() == 0) 1369*0b57cec5SDimitry Andric return false; 1370*0b57cec5SDimitry Andric 1371*0b57cec5SDimitry Andric std::string auto_generated_function_name(GenerateUniqueName( 1372*0b57cec5SDimitry Andric "lldb_autogen_python_cmd_alias_func", num_created_functions)); 1373*0b57cec5SDimitry Andric 1374*0b57cec5SDimitry Andric sstr.Printf("def %s (debugger, args, result, internal_dict):", 1375*0b57cec5SDimitry Andric auto_generated_function_name.c_str()); 1376*0b57cec5SDimitry Andric 1377*0b57cec5SDimitry Andric if (!GenerateFunction(sstr.GetData(), user_input).Success()) 1378*0b57cec5SDimitry Andric return false; 1379*0b57cec5SDimitry Andric 1380*0b57cec5SDimitry Andric // Store the name of the auto-generated function to be called. 1381*0b57cec5SDimitry Andric output.assign(auto_generated_function_name); 1382*0b57cec5SDimitry Andric return true; 1383*0b57cec5SDimitry Andric } 1384*0b57cec5SDimitry Andric 1385*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateTypeSynthClass( 1386*0b57cec5SDimitry Andric StringList &user_input, std::string &output, const void *name_token) { 1387*0b57cec5SDimitry Andric static uint32_t num_created_classes = 0; 1388*0b57cec5SDimitry Andric user_input.RemoveBlankLines(); 1389*0b57cec5SDimitry Andric int num_lines = user_input.GetSize(); 1390*0b57cec5SDimitry Andric StreamString sstr; 1391*0b57cec5SDimitry Andric 1392*0b57cec5SDimitry Andric // Check to see if we have any data; if not, just return. 1393*0b57cec5SDimitry Andric if (user_input.GetSize() == 0) 1394*0b57cec5SDimitry Andric return false; 1395*0b57cec5SDimitry Andric 1396*0b57cec5SDimitry Andric // Wrap all user input into a Python class 1397*0b57cec5SDimitry Andric 1398*0b57cec5SDimitry Andric std::string auto_generated_class_name(GenerateUniqueName( 1399*0b57cec5SDimitry Andric "lldb_autogen_python_type_synth_class", num_created_classes, name_token)); 1400*0b57cec5SDimitry Andric 1401*0b57cec5SDimitry Andric StringList auto_generated_class; 1402*0b57cec5SDimitry Andric 1403*0b57cec5SDimitry Andric // Create the function name & definition string. 1404*0b57cec5SDimitry Andric 1405*0b57cec5SDimitry Andric sstr.Printf("class %s:", auto_generated_class_name.c_str()); 1406*0b57cec5SDimitry Andric auto_generated_class.AppendString(sstr.GetString()); 1407*0b57cec5SDimitry Andric 1408*0b57cec5SDimitry Andric // Wrap everything up inside the class, increasing the indentation. we don't 1409*0b57cec5SDimitry Andric // need to play any fancy indentation tricks here because there is no 1410*0b57cec5SDimitry Andric // surrounding code whose indentation we need to honor 1411*0b57cec5SDimitry Andric for (int i = 0; i < num_lines; ++i) { 1412*0b57cec5SDimitry Andric sstr.Clear(); 1413*0b57cec5SDimitry Andric sstr.Printf(" %s", user_input.GetStringAtIndex(i)); 1414*0b57cec5SDimitry Andric auto_generated_class.AppendString(sstr.GetString()); 1415*0b57cec5SDimitry Andric } 1416*0b57cec5SDimitry Andric 1417*0b57cec5SDimitry Andric // Verify that the results are valid Python. (even though the method is 1418*0b57cec5SDimitry Andric // ExportFunctionDefinitionToInterpreter, a class will actually be exported) 1419*0b57cec5SDimitry Andric // (TODO: rename that method to ExportDefinitionToInterpreter) 1420*0b57cec5SDimitry Andric if (!ExportFunctionDefinitionToInterpreter(auto_generated_class).Success()) 1421*0b57cec5SDimitry Andric return false; 1422*0b57cec5SDimitry Andric 1423*0b57cec5SDimitry Andric // Store the name of the auto-generated class 1424*0b57cec5SDimitry Andric 1425*0b57cec5SDimitry Andric output.assign(auto_generated_class_name); 1426*0b57cec5SDimitry Andric return true; 1427*0b57cec5SDimitry Andric } 1428*0b57cec5SDimitry Andric 1429*0b57cec5SDimitry Andric StructuredData::GenericSP 1430*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::CreateFrameRecognizer(const char *class_name) { 1431*0b57cec5SDimitry Andric if (class_name == nullptr || class_name[0] == '\0') 1432*0b57cec5SDimitry Andric return StructuredData::GenericSP(); 1433*0b57cec5SDimitry Andric 143404eeddc0SDimitry Andric Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 143504eeddc0SDimitry Andric PythonObject ret_val = LLDBSWIGPython_CreateFrameRecognizer( 143604eeddc0SDimitry Andric class_name, m_dictionary_name.c_str()); 1437*0b57cec5SDimitry Andric 143804eeddc0SDimitry Andric return StructuredData::GenericSP( 143904eeddc0SDimitry Andric new StructuredPythonObject(std::move(ret_val))); 1440*0b57cec5SDimitry Andric } 1441*0b57cec5SDimitry Andric 1442*0b57cec5SDimitry Andric lldb::ValueObjectListSP ScriptInterpreterPythonImpl::GetRecognizedArguments( 1443*0b57cec5SDimitry Andric const StructuredData::ObjectSP &os_plugin_object_sp, 1444*0b57cec5SDimitry Andric lldb::StackFrameSP frame_sp) { 1445*0b57cec5SDimitry Andric Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 1446*0b57cec5SDimitry Andric 1447*0b57cec5SDimitry Andric if (!os_plugin_object_sp) 1448*0b57cec5SDimitry Andric return ValueObjectListSP(); 1449*0b57cec5SDimitry Andric 1450*0b57cec5SDimitry Andric StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric(); 1451*0b57cec5SDimitry Andric if (!generic) 1452*0b57cec5SDimitry Andric return nullptr; 1453*0b57cec5SDimitry Andric 1454*0b57cec5SDimitry Andric PythonObject implementor(PyRefType::Borrowed, 1455*0b57cec5SDimitry Andric (PyObject *)generic->GetValue()); 1456*0b57cec5SDimitry Andric 1457*0b57cec5SDimitry Andric if (!implementor.IsAllocated()) 1458*0b57cec5SDimitry Andric return ValueObjectListSP(); 1459*0b57cec5SDimitry Andric 14604824e7fdSDimitry Andric PythonObject py_return( 14614824e7fdSDimitry Andric PyRefType::Owned, 14624824e7fdSDimitry Andric LLDBSwigPython_GetRecognizedArguments(implementor.get(), frame_sp)); 1463*0b57cec5SDimitry Andric 1464*0b57cec5SDimitry Andric // if it fails, print the error but otherwise go on 1465*0b57cec5SDimitry Andric if (PyErr_Occurred()) { 1466*0b57cec5SDimitry Andric PyErr_Print(); 1467*0b57cec5SDimitry Andric PyErr_Clear(); 1468*0b57cec5SDimitry Andric } 1469*0b57cec5SDimitry Andric if (py_return.get()) { 1470*0b57cec5SDimitry Andric PythonList result_list(PyRefType::Borrowed, py_return.get()); 1471*0b57cec5SDimitry Andric ValueObjectListSP result = ValueObjectListSP(new ValueObjectList()); 1472*0b57cec5SDimitry Andric for (size_t i = 0; i < result_list.GetSize(); i++) { 1473*0b57cec5SDimitry Andric PyObject *item = result_list.GetItemAtIndex(i).get(); 1474*0b57cec5SDimitry Andric lldb::SBValue *sb_value_ptr = 1475*0b57cec5SDimitry Andric (lldb::SBValue *)LLDBSWIGPython_CastPyObjectToSBValue(item); 1476*0b57cec5SDimitry Andric auto valobj_sp = LLDBSWIGPython_GetValueObjectSPFromSBValue(sb_value_ptr); 1477*0b57cec5SDimitry Andric if (valobj_sp) 1478*0b57cec5SDimitry Andric result->Append(valobj_sp); 1479*0b57cec5SDimitry Andric } 1480*0b57cec5SDimitry Andric return result; 1481*0b57cec5SDimitry Andric } 1482*0b57cec5SDimitry Andric return ValueObjectListSP(); 1483*0b57cec5SDimitry Andric } 1484*0b57cec5SDimitry Andric 1485*0b57cec5SDimitry Andric StructuredData::GenericSP 1486*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::OSPlugin_CreatePluginObject( 1487*0b57cec5SDimitry Andric const char *class_name, lldb::ProcessSP process_sp) { 1488*0b57cec5SDimitry Andric if (class_name == nullptr || class_name[0] == '\0') 1489*0b57cec5SDimitry Andric return StructuredData::GenericSP(); 1490*0b57cec5SDimitry Andric 1491*0b57cec5SDimitry Andric if (!process_sp) 1492*0b57cec5SDimitry Andric return StructuredData::GenericSP(); 1493*0b57cec5SDimitry Andric 149404eeddc0SDimitry Andric Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 149504eeddc0SDimitry Andric PythonObject ret_val = LLDBSWIGPythonCreateOSPlugin( 1496*0b57cec5SDimitry Andric class_name, m_dictionary_name.c_str(), process_sp); 1497*0b57cec5SDimitry Andric 149804eeddc0SDimitry Andric return StructuredData::GenericSP( 149904eeddc0SDimitry Andric new StructuredPythonObject(std::move(ret_val))); 1500*0b57cec5SDimitry Andric } 1501*0b57cec5SDimitry Andric 1502*0b57cec5SDimitry Andric StructuredData::DictionarySP ScriptInterpreterPythonImpl::OSPlugin_RegisterInfo( 1503*0b57cec5SDimitry Andric StructuredData::ObjectSP os_plugin_object_sp) { 1504*0b57cec5SDimitry Andric Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 1505*0b57cec5SDimitry Andric 1506*0b57cec5SDimitry Andric static char callee_name[] = "get_register_info"; 1507*0b57cec5SDimitry Andric 1508*0b57cec5SDimitry Andric if (!os_plugin_object_sp) 1509*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1510*0b57cec5SDimitry Andric 1511*0b57cec5SDimitry Andric StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric(); 1512*0b57cec5SDimitry Andric if (!generic) 1513*0b57cec5SDimitry Andric return nullptr; 1514*0b57cec5SDimitry Andric 1515*0b57cec5SDimitry Andric PythonObject implementor(PyRefType::Borrowed, 1516*0b57cec5SDimitry Andric (PyObject *)generic->GetValue()); 1517*0b57cec5SDimitry Andric 1518*0b57cec5SDimitry Andric if (!implementor.IsAllocated()) 1519*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1520*0b57cec5SDimitry Andric 1521*0b57cec5SDimitry Andric PythonObject pmeth(PyRefType::Owned, 1522*0b57cec5SDimitry Andric PyObject_GetAttrString(implementor.get(), callee_name)); 1523*0b57cec5SDimitry Andric 1524*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1525*0b57cec5SDimitry Andric PyErr_Clear(); 1526*0b57cec5SDimitry Andric 1527*0b57cec5SDimitry Andric if (!pmeth.IsAllocated()) 1528*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1529*0b57cec5SDimitry Andric 1530*0b57cec5SDimitry Andric if (PyCallable_Check(pmeth.get()) == 0) { 1531*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1532*0b57cec5SDimitry Andric PyErr_Clear(); 1533*0b57cec5SDimitry Andric 1534*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1535*0b57cec5SDimitry Andric } 1536*0b57cec5SDimitry Andric 1537*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1538*0b57cec5SDimitry Andric PyErr_Clear(); 1539*0b57cec5SDimitry Andric 1540*0b57cec5SDimitry Andric // right now we know this function exists and is callable.. 1541*0b57cec5SDimitry Andric PythonObject py_return( 1542*0b57cec5SDimitry Andric PyRefType::Owned, 1543*0b57cec5SDimitry Andric PyObject_CallMethod(implementor.get(), callee_name, nullptr)); 1544*0b57cec5SDimitry Andric 1545*0b57cec5SDimitry Andric // if it fails, print the error but otherwise go on 1546*0b57cec5SDimitry Andric if (PyErr_Occurred()) { 1547*0b57cec5SDimitry Andric PyErr_Print(); 1548*0b57cec5SDimitry Andric PyErr_Clear(); 1549*0b57cec5SDimitry Andric } 1550*0b57cec5SDimitry Andric if (py_return.get()) { 1551*0b57cec5SDimitry Andric PythonDictionary result_dict(PyRefType::Borrowed, py_return.get()); 1552*0b57cec5SDimitry Andric return result_dict.CreateStructuredDictionary(); 1553*0b57cec5SDimitry Andric } 1554*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1555*0b57cec5SDimitry Andric } 1556*0b57cec5SDimitry Andric 1557*0b57cec5SDimitry Andric StructuredData::ArraySP ScriptInterpreterPythonImpl::OSPlugin_ThreadsInfo( 1558*0b57cec5SDimitry Andric StructuredData::ObjectSP os_plugin_object_sp) { 1559*0b57cec5SDimitry Andric Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 1560*0b57cec5SDimitry Andric 1561*0b57cec5SDimitry Andric static char callee_name[] = "get_thread_info"; 1562*0b57cec5SDimitry Andric 1563*0b57cec5SDimitry Andric if (!os_plugin_object_sp) 1564*0b57cec5SDimitry Andric return StructuredData::ArraySP(); 1565*0b57cec5SDimitry Andric 1566*0b57cec5SDimitry Andric StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric(); 1567*0b57cec5SDimitry Andric if (!generic) 1568*0b57cec5SDimitry Andric return nullptr; 1569*0b57cec5SDimitry Andric 1570*0b57cec5SDimitry Andric PythonObject implementor(PyRefType::Borrowed, 1571*0b57cec5SDimitry Andric (PyObject *)generic->GetValue()); 1572*0b57cec5SDimitry Andric 1573*0b57cec5SDimitry Andric if (!implementor.IsAllocated()) 1574*0b57cec5SDimitry Andric return StructuredData::ArraySP(); 1575*0b57cec5SDimitry Andric 1576*0b57cec5SDimitry Andric PythonObject pmeth(PyRefType::Owned, 1577*0b57cec5SDimitry Andric PyObject_GetAttrString(implementor.get(), callee_name)); 1578*0b57cec5SDimitry Andric 1579*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1580*0b57cec5SDimitry Andric PyErr_Clear(); 1581*0b57cec5SDimitry Andric 1582*0b57cec5SDimitry Andric if (!pmeth.IsAllocated()) 1583*0b57cec5SDimitry Andric return StructuredData::ArraySP(); 1584*0b57cec5SDimitry Andric 1585*0b57cec5SDimitry Andric if (PyCallable_Check(pmeth.get()) == 0) { 1586*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1587*0b57cec5SDimitry Andric PyErr_Clear(); 1588*0b57cec5SDimitry Andric 1589*0b57cec5SDimitry Andric return StructuredData::ArraySP(); 1590*0b57cec5SDimitry Andric } 1591*0b57cec5SDimitry Andric 1592*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1593*0b57cec5SDimitry Andric PyErr_Clear(); 1594*0b57cec5SDimitry Andric 1595*0b57cec5SDimitry Andric // right now we know this function exists and is callable.. 1596*0b57cec5SDimitry Andric PythonObject py_return( 1597*0b57cec5SDimitry Andric PyRefType::Owned, 1598*0b57cec5SDimitry Andric PyObject_CallMethod(implementor.get(), callee_name, nullptr)); 1599*0b57cec5SDimitry Andric 1600*0b57cec5SDimitry Andric // if it fails, print the error but otherwise go on 1601*0b57cec5SDimitry Andric if (PyErr_Occurred()) { 1602*0b57cec5SDimitry Andric PyErr_Print(); 1603*0b57cec5SDimitry Andric PyErr_Clear(); 1604*0b57cec5SDimitry Andric } 1605*0b57cec5SDimitry Andric 1606*0b57cec5SDimitry Andric if (py_return.get()) { 1607*0b57cec5SDimitry Andric PythonList result_list(PyRefType::Borrowed, py_return.get()); 1608*0b57cec5SDimitry Andric return result_list.CreateStructuredArray(); 1609*0b57cec5SDimitry Andric } 1610*0b57cec5SDimitry Andric return StructuredData::ArraySP(); 1611*0b57cec5SDimitry Andric } 1612*0b57cec5SDimitry Andric 1613*0b57cec5SDimitry Andric StructuredData::StringSP 1614*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::OSPlugin_RegisterContextData( 1615*0b57cec5SDimitry Andric StructuredData::ObjectSP os_plugin_object_sp, lldb::tid_t tid) { 1616*0b57cec5SDimitry Andric Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 1617*0b57cec5SDimitry Andric 1618*0b57cec5SDimitry Andric static char callee_name[] = "get_register_data"; 1619*0b57cec5SDimitry Andric static char *param_format = 1620*0b57cec5SDimitry Andric const_cast<char *>(GetPythonValueFormatString(tid)); 1621*0b57cec5SDimitry Andric 1622*0b57cec5SDimitry Andric if (!os_plugin_object_sp) 1623*0b57cec5SDimitry Andric return StructuredData::StringSP(); 1624*0b57cec5SDimitry Andric 1625*0b57cec5SDimitry Andric StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric(); 1626*0b57cec5SDimitry Andric if (!generic) 1627*0b57cec5SDimitry Andric return nullptr; 1628*0b57cec5SDimitry Andric PythonObject implementor(PyRefType::Borrowed, 1629*0b57cec5SDimitry Andric (PyObject *)generic->GetValue()); 1630*0b57cec5SDimitry Andric 1631*0b57cec5SDimitry Andric if (!implementor.IsAllocated()) 1632*0b57cec5SDimitry Andric return StructuredData::StringSP(); 1633*0b57cec5SDimitry Andric 1634*0b57cec5SDimitry Andric PythonObject pmeth(PyRefType::Owned, 1635*0b57cec5SDimitry Andric PyObject_GetAttrString(implementor.get(), callee_name)); 1636*0b57cec5SDimitry Andric 1637*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1638*0b57cec5SDimitry Andric PyErr_Clear(); 1639*0b57cec5SDimitry Andric 1640*0b57cec5SDimitry Andric if (!pmeth.IsAllocated()) 1641*0b57cec5SDimitry Andric return StructuredData::StringSP(); 1642*0b57cec5SDimitry Andric 1643*0b57cec5SDimitry Andric if (PyCallable_Check(pmeth.get()) == 0) { 1644*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1645*0b57cec5SDimitry Andric PyErr_Clear(); 1646*0b57cec5SDimitry Andric return StructuredData::StringSP(); 1647*0b57cec5SDimitry Andric } 1648*0b57cec5SDimitry Andric 1649*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1650*0b57cec5SDimitry Andric PyErr_Clear(); 1651*0b57cec5SDimitry Andric 1652*0b57cec5SDimitry Andric // right now we know this function exists and is callable.. 1653*0b57cec5SDimitry Andric PythonObject py_return( 1654*0b57cec5SDimitry Andric PyRefType::Owned, 1655*0b57cec5SDimitry Andric PyObject_CallMethod(implementor.get(), callee_name, param_format, tid)); 1656*0b57cec5SDimitry Andric 1657*0b57cec5SDimitry Andric // if it fails, print the error but otherwise go on 1658*0b57cec5SDimitry Andric if (PyErr_Occurred()) { 1659*0b57cec5SDimitry Andric PyErr_Print(); 1660*0b57cec5SDimitry Andric PyErr_Clear(); 1661*0b57cec5SDimitry Andric } 1662*0b57cec5SDimitry Andric 1663*0b57cec5SDimitry Andric if (py_return.get()) { 1664*0b57cec5SDimitry Andric PythonBytes result(PyRefType::Borrowed, py_return.get()); 1665*0b57cec5SDimitry Andric return result.CreateStructuredString(); 1666*0b57cec5SDimitry Andric } 1667*0b57cec5SDimitry Andric return StructuredData::StringSP(); 1668*0b57cec5SDimitry Andric } 1669*0b57cec5SDimitry Andric 1670*0b57cec5SDimitry Andric StructuredData::DictionarySP ScriptInterpreterPythonImpl::OSPlugin_CreateThread( 1671*0b57cec5SDimitry Andric StructuredData::ObjectSP os_plugin_object_sp, lldb::tid_t tid, 1672*0b57cec5SDimitry Andric lldb::addr_t context) { 1673*0b57cec5SDimitry Andric Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 1674*0b57cec5SDimitry Andric 1675*0b57cec5SDimitry Andric static char callee_name[] = "create_thread"; 1676*0b57cec5SDimitry Andric std::string param_format; 1677*0b57cec5SDimitry Andric param_format += GetPythonValueFormatString(tid); 1678*0b57cec5SDimitry Andric param_format += GetPythonValueFormatString(context); 1679*0b57cec5SDimitry Andric 1680*0b57cec5SDimitry Andric if (!os_plugin_object_sp) 1681*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1682*0b57cec5SDimitry Andric 1683*0b57cec5SDimitry Andric StructuredData::Generic *generic = os_plugin_object_sp->GetAsGeneric(); 1684*0b57cec5SDimitry Andric if (!generic) 1685*0b57cec5SDimitry Andric return nullptr; 1686*0b57cec5SDimitry Andric 1687*0b57cec5SDimitry Andric PythonObject implementor(PyRefType::Borrowed, 1688*0b57cec5SDimitry Andric (PyObject *)generic->GetValue()); 1689*0b57cec5SDimitry Andric 1690*0b57cec5SDimitry Andric if (!implementor.IsAllocated()) 1691*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1692*0b57cec5SDimitry Andric 1693*0b57cec5SDimitry Andric PythonObject pmeth(PyRefType::Owned, 1694*0b57cec5SDimitry Andric PyObject_GetAttrString(implementor.get(), callee_name)); 1695*0b57cec5SDimitry Andric 1696*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1697*0b57cec5SDimitry Andric PyErr_Clear(); 1698*0b57cec5SDimitry Andric 1699*0b57cec5SDimitry Andric if (!pmeth.IsAllocated()) 1700*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1701*0b57cec5SDimitry Andric 1702*0b57cec5SDimitry Andric if (PyCallable_Check(pmeth.get()) == 0) { 1703*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1704*0b57cec5SDimitry Andric PyErr_Clear(); 1705*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1706*0b57cec5SDimitry Andric } 1707*0b57cec5SDimitry Andric 1708*0b57cec5SDimitry Andric if (PyErr_Occurred()) 1709*0b57cec5SDimitry Andric PyErr_Clear(); 1710*0b57cec5SDimitry Andric 1711*0b57cec5SDimitry Andric // right now we know this function exists and is callable.. 1712*0b57cec5SDimitry Andric PythonObject py_return(PyRefType::Owned, 1713*0b57cec5SDimitry Andric PyObject_CallMethod(implementor.get(), callee_name, 1714*0b57cec5SDimitry Andric ¶m_format[0], tid, context)); 1715*0b57cec5SDimitry Andric 1716*0b57cec5SDimitry Andric // if it fails, print the error but otherwise go on 1717*0b57cec5SDimitry Andric if (PyErr_Occurred()) { 1718*0b57cec5SDimitry Andric PyErr_Print(); 1719*0b57cec5SDimitry Andric PyErr_Clear(); 1720*0b57cec5SDimitry Andric } 1721*0b57cec5SDimitry Andric 1722*0b57cec5SDimitry Andric if (py_return.get()) { 1723*0b57cec5SDimitry Andric PythonDictionary result_dict(PyRefType::Borrowed, py_return.get()); 1724*0b57cec5SDimitry Andric return result_dict.CreateStructuredDictionary(); 1725*0b57cec5SDimitry Andric } 1726*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1727*0b57cec5SDimitry Andric } 1728*0b57cec5SDimitry Andric 1729*0b57cec5SDimitry Andric StructuredData::ObjectSP ScriptInterpreterPythonImpl::CreateScriptedThreadPlan( 17300eae32dcSDimitry Andric const char *class_name, const StructuredDataImpl &args_data, 1731480093f4SDimitry Andric std::string &error_str, lldb::ThreadPlanSP thread_plan_sp) { 1732*0b57cec5SDimitry Andric if (class_name == nullptr || class_name[0] == '\0') 1733*0b57cec5SDimitry Andric return StructuredData::ObjectSP(); 1734*0b57cec5SDimitry Andric 1735*0b57cec5SDimitry Andric if (!thread_plan_sp.get()) 17369dba64beSDimitry Andric return {}; 1737*0b57cec5SDimitry Andric 1738*0b57cec5SDimitry Andric Debugger &debugger = thread_plan_sp->GetTarget().GetDebugger(); 1739*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl *python_interpreter = 1740e8d8bef9SDimitry Andric GetPythonInterpreter(debugger); 1741*0b57cec5SDimitry Andric 1742e8d8bef9SDimitry Andric if (!python_interpreter) 17439dba64beSDimitry Andric return {}; 1744*0b57cec5SDimitry Andric 1745*0b57cec5SDimitry Andric Locker py_lock(this, 1746*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 174704eeddc0SDimitry Andric PythonObject ret_val = LLDBSwigPythonCreateScriptedThreadPlan( 174804eeddc0SDimitry Andric class_name, python_interpreter->m_dictionary_name.c_str(), args_data, 174904eeddc0SDimitry Andric error_str, thread_plan_sp); 17509dba64beSDimitry Andric if (!ret_val) 17519dba64beSDimitry Andric return {}; 1752*0b57cec5SDimitry Andric 175304eeddc0SDimitry Andric return StructuredData::ObjectSP( 175404eeddc0SDimitry Andric new StructuredPythonObject(std::move(ret_val))); 1755*0b57cec5SDimitry Andric } 1756*0b57cec5SDimitry Andric 1757*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ScriptedThreadPlanExplainsStop( 1758*0b57cec5SDimitry Andric StructuredData::ObjectSP implementor_sp, Event *event, bool &script_error) { 1759*0b57cec5SDimitry Andric bool explains_stop = true; 1760*0b57cec5SDimitry Andric StructuredData::Generic *generic = nullptr; 1761*0b57cec5SDimitry Andric if (implementor_sp) 1762*0b57cec5SDimitry Andric generic = implementor_sp->GetAsGeneric(); 1763*0b57cec5SDimitry Andric if (generic) { 1764*0b57cec5SDimitry Andric Locker py_lock(this, 1765*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1766*0b57cec5SDimitry Andric explains_stop = LLDBSWIGPythonCallThreadPlan( 1767*0b57cec5SDimitry Andric generic->GetValue(), "explains_stop", event, script_error); 1768*0b57cec5SDimitry Andric if (script_error) 1769*0b57cec5SDimitry Andric return true; 1770*0b57cec5SDimitry Andric } 1771*0b57cec5SDimitry Andric return explains_stop; 1772*0b57cec5SDimitry Andric } 1773*0b57cec5SDimitry Andric 1774*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ScriptedThreadPlanShouldStop( 1775*0b57cec5SDimitry Andric StructuredData::ObjectSP implementor_sp, Event *event, bool &script_error) { 1776*0b57cec5SDimitry Andric bool should_stop = true; 1777*0b57cec5SDimitry Andric StructuredData::Generic *generic = nullptr; 1778*0b57cec5SDimitry Andric if (implementor_sp) 1779*0b57cec5SDimitry Andric generic = implementor_sp->GetAsGeneric(); 1780*0b57cec5SDimitry Andric if (generic) { 1781*0b57cec5SDimitry Andric Locker py_lock(this, 1782*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1783*0b57cec5SDimitry Andric should_stop = LLDBSWIGPythonCallThreadPlan( 1784*0b57cec5SDimitry Andric generic->GetValue(), "should_stop", event, script_error); 1785*0b57cec5SDimitry Andric if (script_error) 1786*0b57cec5SDimitry Andric return true; 1787*0b57cec5SDimitry Andric } 1788*0b57cec5SDimitry Andric return should_stop; 1789*0b57cec5SDimitry Andric } 1790*0b57cec5SDimitry Andric 1791*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ScriptedThreadPlanIsStale( 1792*0b57cec5SDimitry Andric StructuredData::ObjectSP implementor_sp, bool &script_error) { 1793*0b57cec5SDimitry Andric bool is_stale = true; 1794*0b57cec5SDimitry Andric StructuredData::Generic *generic = nullptr; 1795*0b57cec5SDimitry Andric if (implementor_sp) 1796*0b57cec5SDimitry Andric generic = implementor_sp->GetAsGeneric(); 1797*0b57cec5SDimitry Andric if (generic) { 1798*0b57cec5SDimitry Andric Locker py_lock(this, 1799*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1800*0b57cec5SDimitry Andric is_stale = LLDBSWIGPythonCallThreadPlan(generic->GetValue(), "is_stale", 1801*0b57cec5SDimitry Andric nullptr, script_error); 1802*0b57cec5SDimitry Andric if (script_error) 1803*0b57cec5SDimitry Andric return true; 1804*0b57cec5SDimitry Andric } 1805*0b57cec5SDimitry Andric return is_stale; 1806*0b57cec5SDimitry Andric } 1807*0b57cec5SDimitry Andric 1808*0b57cec5SDimitry Andric lldb::StateType ScriptInterpreterPythonImpl::ScriptedThreadPlanGetRunState( 1809*0b57cec5SDimitry Andric StructuredData::ObjectSP implementor_sp, bool &script_error) { 1810*0b57cec5SDimitry Andric bool should_step = false; 1811*0b57cec5SDimitry Andric StructuredData::Generic *generic = nullptr; 1812*0b57cec5SDimitry Andric if (implementor_sp) 1813*0b57cec5SDimitry Andric generic = implementor_sp->GetAsGeneric(); 1814*0b57cec5SDimitry Andric if (generic) { 1815*0b57cec5SDimitry Andric Locker py_lock(this, 1816*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1817*0b57cec5SDimitry Andric should_step = LLDBSWIGPythonCallThreadPlan( 1818*0b57cec5SDimitry Andric generic->GetValue(), "should_step", nullptr, script_error); 1819*0b57cec5SDimitry Andric if (script_error) 1820*0b57cec5SDimitry Andric should_step = true; 1821*0b57cec5SDimitry Andric } 1822*0b57cec5SDimitry Andric if (should_step) 1823*0b57cec5SDimitry Andric return lldb::eStateStepping; 1824*0b57cec5SDimitry Andric return lldb::eStateRunning; 1825*0b57cec5SDimitry Andric } 1826*0b57cec5SDimitry Andric 1827*0b57cec5SDimitry Andric StructuredData::GenericSP 1828*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::CreateScriptedBreakpointResolver( 18290eae32dcSDimitry Andric const char *class_name, const StructuredDataImpl &args_data, 1830*0b57cec5SDimitry Andric lldb::BreakpointSP &bkpt_sp) { 1831*0b57cec5SDimitry Andric 1832*0b57cec5SDimitry Andric if (class_name == nullptr || class_name[0] == '\0') 1833*0b57cec5SDimitry Andric return StructuredData::GenericSP(); 1834*0b57cec5SDimitry Andric 1835*0b57cec5SDimitry Andric if (!bkpt_sp.get()) 1836*0b57cec5SDimitry Andric return StructuredData::GenericSP(); 1837*0b57cec5SDimitry Andric 1838*0b57cec5SDimitry Andric Debugger &debugger = bkpt_sp->GetTarget().GetDebugger(); 1839*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl *python_interpreter = 1840e8d8bef9SDimitry Andric GetPythonInterpreter(debugger); 1841*0b57cec5SDimitry Andric 1842e8d8bef9SDimitry Andric if (!python_interpreter) 1843*0b57cec5SDimitry Andric return StructuredData::GenericSP(); 1844*0b57cec5SDimitry Andric 1845*0b57cec5SDimitry Andric Locker py_lock(this, 1846*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1847*0b57cec5SDimitry Andric 184804eeddc0SDimitry Andric PythonObject ret_val = LLDBSwigPythonCreateScriptedBreakpointResolver( 1849*0b57cec5SDimitry Andric class_name, python_interpreter->m_dictionary_name.c_str(), args_data, 1850*0b57cec5SDimitry Andric bkpt_sp); 1851*0b57cec5SDimitry Andric 185204eeddc0SDimitry Andric return StructuredData::GenericSP( 185304eeddc0SDimitry Andric new StructuredPythonObject(std::move(ret_val))); 1854*0b57cec5SDimitry Andric } 1855*0b57cec5SDimitry Andric 1856*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::ScriptedBreakpointResolverSearchCallback( 1857*0b57cec5SDimitry Andric StructuredData::GenericSP implementor_sp, SymbolContext *sym_ctx) { 1858*0b57cec5SDimitry Andric bool should_continue = false; 1859*0b57cec5SDimitry Andric 1860*0b57cec5SDimitry Andric if (implementor_sp) { 1861*0b57cec5SDimitry Andric Locker py_lock(this, 1862*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1863*0b57cec5SDimitry Andric should_continue = LLDBSwigPythonCallBreakpointResolver( 1864*0b57cec5SDimitry Andric implementor_sp->GetValue(), "__callback__", sym_ctx); 1865*0b57cec5SDimitry Andric if (PyErr_Occurred()) { 1866*0b57cec5SDimitry Andric PyErr_Print(); 1867*0b57cec5SDimitry Andric PyErr_Clear(); 1868*0b57cec5SDimitry Andric } 1869*0b57cec5SDimitry Andric } 1870*0b57cec5SDimitry Andric return should_continue; 1871*0b57cec5SDimitry Andric } 1872*0b57cec5SDimitry Andric 1873*0b57cec5SDimitry Andric lldb::SearchDepth 1874*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::ScriptedBreakpointResolverSearchDepth( 1875*0b57cec5SDimitry Andric StructuredData::GenericSP implementor_sp) { 1876*0b57cec5SDimitry Andric int depth_as_int = lldb::eSearchDepthModule; 1877*0b57cec5SDimitry Andric if (implementor_sp) { 1878*0b57cec5SDimitry Andric Locker py_lock(this, 1879*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1880*0b57cec5SDimitry Andric depth_as_int = LLDBSwigPythonCallBreakpointResolver( 1881*0b57cec5SDimitry Andric implementor_sp->GetValue(), "__get_depth__", nullptr); 1882*0b57cec5SDimitry Andric if (PyErr_Occurred()) { 1883*0b57cec5SDimitry Andric PyErr_Print(); 1884*0b57cec5SDimitry Andric PyErr_Clear(); 1885*0b57cec5SDimitry Andric } 1886*0b57cec5SDimitry Andric } 1887*0b57cec5SDimitry Andric if (depth_as_int == lldb::eSearchDepthInvalid) 1888*0b57cec5SDimitry Andric return lldb::eSearchDepthModule; 1889*0b57cec5SDimitry Andric 1890*0b57cec5SDimitry Andric if (depth_as_int <= lldb::kLastSearchDepthKind) 1891*0b57cec5SDimitry Andric return (lldb::SearchDepth)depth_as_int; 1892*0b57cec5SDimitry Andric return lldb::eSearchDepthModule; 1893*0b57cec5SDimitry Andric } 1894*0b57cec5SDimitry Andric 1895e8d8bef9SDimitry Andric StructuredData::GenericSP ScriptInterpreterPythonImpl::CreateScriptedStopHook( 18960eae32dcSDimitry Andric TargetSP target_sp, const char *class_name, 18970eae32dcSDimitry Andric const StructuredDataImpl &args_data, Status &error) { 1898e8d8bef9SDimitry Andric 1899e8d8bef9SDimitry Andric if (!target_sp) { 1900e8d8bef9SDimitry Andric error.SetErrorString("No target for scripted stop-hook."); 1901e8d8bef9SDimitry Andric return StructuredData::GenericSP(); 1902e8d8bef9SDimitry Andric } 1903e8d8bef9SDimitry Andric 1904e8d8bef9SDimitry Andric if (class_name == nullptr || class_name[0] == '\0') { 1905e8d8bef9SDimitry Andric error.SetErrorString("No class name for scripted stop-hook."); 1906e8d8bef9SDimitry Andric return StructuredData::GenericSP(); 1907e8d8bef9SDimitry Andric } 1908e8d8bef9SDimitry Andric 1909e8d8bef9SDimitry Andric ScriptInterpreterPythonImpl *python_interpreter = 1910e8d8bef9SDimitry Andric GetPythonInterpreter(m_debugger); 1911e8d8bef9SDimitry Andric 1912e8d8bef9SDimitry Andric if (!python_interpreter) { 1913e8d8bef9SDimitry Andric error.SetErrorString("No script interpreter for scripted stop-hook."); 1914e8d8bef9SDimitry Andric return StructuredData::GenericSP(); 1915e8d8bef9SDimitry Andric } 1916e8d8bef9SDimitry Andric 1917e8d8bef9SDimitry Andric Locker py_lock(this, 1918e8d8bef9SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1919e8d8bef9SDimitry Andric 192004eeddc0SDimitry Andric PythonObject ret_val = LLDBSwigPythonCreateScriptedStopHook( 1921e8d8bef9SDimitry Andric target_sp, class_name, python_interpreter->m_dictionary_name.c_str(), 1922e8d8bef9SDimitry Andric args_data, error); 1923e8d8bef9SDimitry Andric 192404eeddc0SDimitry Andric return StructuredData::GenericSP( 192504eeddc0SDimitry Andric new StructuredPythonObject(std::move(ret_val))); 1926e8d8bef9SDimitry Andric } 1927e8d8bef9SDimitry Andric 1928e8d8bef9SDimitry Andric bool ScriptInterpreterPythonImpl::ScriptedStopHookHandleStop( 1929e8d8bef9SDimitry Andric StructuredData::GenericSP implementor_sp, ExecutionContext &exc_ctx, 1930e8d8bef9SDimitry Andric lldb::StreamSP stream_sp) { 1931e8d8bef9SDimitry Andric assert(implementor_sp && 1932e8d8bef9SDimitry Andric "can't call a stop hook with an invalid implementor"); 1933e8d8bef9SDimitry Andric assert(stream_sp && "can't call a stop hook with an invalid stream"); 1934e8d8bef9SDimitry Andric 1935e8d8bef9SDimitry Andric Locker py_lock(this, 1936e8d8bef9SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1937e8d8bef9SDimitry Andric 1938e8d8bef9SDimitry Andric lldb::ExecutionContextRefSP exc_ctx_ref_sp(new ExecutionContextRef(exc_ctx)); 1939e8d8bef9SDimitry Andric 1940e8d8bef9SDimitry Andric bool ret_val = LLDBSwigPythonStopHookCallHandleStop( 1941e8d8bef9SDimitry Andric implementor_sp->GetValue(), exc_ctx_ref_sp, stream_sp); 1942e8d8bef9SDimitry Andric return ret_val; 1943e8d8bef9SDimitry Andric } 1944e8d8bef9SDimitry Andric 1945*0b57cec5SDimitry Andric StructuredData::ObjectSP 1946*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::LoadPluginModule(const FileSpec &file_spec, 1947*0b57cec5SDimitry Andric lldb_private::Status &error) { 1948*0b57cec5SDimitry Andric if (!FileSystem::Instance().Exists(file_spec)) { 1949*0b57cec5SDimitry Andric error.SetErrorString("no such file"); 1950*0b57cec5SDimitry Andric return StructuredData::ObjectSP(); 1951*0b57cec5SDimitry Andric } 1952*0b57cec5SDimitry Andric 1953*0b57cec5SDimitry Andric StructuredData::ObjectSP module_sp; 1954*0b57cec5SDimitry Andric 1955fe6060f1SDimitry Andric LoadScriptOptions load_script_options = 1956fe6060f1SDimitry Andric LoadScriptOptions().SetInitSession(true).SetSilent(false); 1957fe6060f1SDimitry Andric if (LoadScriptingModule(file_spec.GetPath().c_str(), load_script_options, 1958fe6060f1SDimitry Andric error, &module_sp)) 1959*0b57cec5SDimitry Andric return module_sp; 1960*0b57cec5SDimitry Andric 1961*0b57cec5SDimitry Andric return StructuredData::ObjectSP(); 1962*0b57cec5SDimitry Andric } 1963*0b57cec5SDimitry Andric 1964*0b57cec5SDimitry Andric StructuredData::DictionarySP ScriptInterpreterPythonImpl::GetDynamicSettings( 1965*0b57cec5SDimitry Andric StructuredData::ObjectSP plugin_module_sp, Target *target, 1966*0b57cec5SDimitry Andric const char *setting_name, lldb_private::Status &error) { 1967*0b57cec5SDimitry Andric if (!plugin_module_sp || !target || !setting_name || !setting_name[0]) 1968*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1969*0b57cec5SDimitry Andric StructuredData::Generic *generic = plugin_module_sp->GetAsGeneric(); 1970*0b57cec5SDimitry Andric if (!generic) 1971*0b57cec5SDimitry Andric return StructuredData::DictionarySP(); 1972*0b57cec5SDimitry Andric 1973*0b57cec5SDimitry Andric Locker py_lock(this, 1974*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 1975*0b57cec5SDimitry Andric TargetSP target_sp(target->shared_from_this()); 1976*0b57cec5SDimitry Andric 19779dba64beSDimitry Andric auto setting = (PyObject *)LLDBSWIGPython_GetDynamicSetting( 19789dba64beSDimitry Andric generic->GetValue(), setting_name, target_sp); 19799dba64beSDimitry Andric 19809dba64beSDimitry Andric if (!setting) 19819dba64beSDimitry Andric return StructuredData::DictionarySP(); 19829dba64beSDimitry Andric 19839dba64beSDimitry Andric PythonDictionary py_dict = 19849dba64beSDimitry Andric unwrapIgnoringErrors(As<PythonDictionary>(Take<PythonObject>(setting))); 19859dba64beSDimitry Andric 19869dba64beSDimitry Andric if (!py_dict) 19879dba64beSDimitry Andric return StructuredData::DictionarySP(); 19889dba64beSDimitry Andric 1989*0b57cec5SDimitry Andric return py_dict.CreateStructuredDictionary(); 1990*0b57cec5SDimitry Andric } 1991*0b57cec5SDimitry Andric 1992*0b57cec5SDimitry Andric StructuredData::ObjectSP 1993*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::CreateSyntheticScriptedProvider( 1994*0b57cec5SDimitry Andric const char *class_name, lldb::ValueObjectSP valobj) { 1995*0b57cec5SDimitry Andric if (class_name == nullptr || class_name[0] == '\0') 1996*0b57cec5SDimitry Andric return StructuredData::ObjectSP(); 1997*0b57cec5SDimitry Andric 1998*0b57cec5SDimitry Andric if (!valobj.get()) 1999*0b57cec5SDimitry Andric return StructuredData::ObjectSP(); 2000*0b57cec5SDimitry Andric 2001*0b57cec5SDimitry Andric ExecutionContext exe_ctx(valobj->GetExecutionContextRef()); 2002*0b57cec5SDimitry Andric Target *target = exe_ctx.GetTargetPtr(); 2003*0b57cec5SDimitry Andric 2004*0b57cec5SDimitry Andric if (!target) 2005*0b57cec5SDimitry Andric return StructuredData::ObjectSP(); 2006*0b57cec5SDimitry Andric 2007*0b57cec5SDimitry Andric Debugger &debugger = target->GetDebugger(); 2008*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl *python_interpreter = 2009e8d8bef9SDimitry Andric GetPythonInterpreter(debugger); 2010*0b57cec5SDimitry Andric 2011e8d8bef9SDimitry Andric if (!python_interpreter) 2012*0b57cec5SDimitry Andric return StructuredData::ObjectSP(); 2013*0b57cec5SDimitry Andric 2014*0b57cec5SDimitry Andric Locker py_lock(this, 2015*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 201604eeddc0SDimitry Andric PythonObject ret_val = LLDBSwigPythonCreateSyntheticProvider( 2017*0b57cec5SDimitry Andric class_name, python_interpreter->m_dictionary_name.c_str(), valobj); 2018*0b57cec5SDimitry Andric 201904eeddc0SDimitry Andric return StructuredData::ObjectSP( 202004eeddc0SDimitry Andric new StructuredPythonObject(std::move(ret_val))); 2021*0b57cec5SDimitry Andric } 2022*0b57cec5SDimitry Andric 2023*0b57cec5SDimitry Andric StructuredData::GenericSP 2024*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::CreateScriptCommandObject(const char *class_name) { 2025*0b57cec5SDimitry Andric DebuggerSP debugger_sp(m_debugger.shared_from_this()); 2026*0b57cec5SDimitry Andric 2027*0b57cec5SDimitry Andric if (class_name == nullptr || class_name[0] == '\0') 2028*0b57cec5SDimitry Andric return StructuredData::GenericSP(); 2029*0b57cec5SDimitry Andric 2030*0b57cec5SDimitry Andric if (!debugger_sp.get()) 2031*0b57cec5SDimitry Andric return StructuredData::GenericSP(); 2032*0b57cec5SDimitry Andric 2033*0b57cec5SDimitry Andric Locker py_lock(this, 2034*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 203504eeddc0SDimitry Andric PythonObject ret_val = LLDBSwigPythonCreateCommandObject( 2036*0b57cec5SDimitry Andric class_name, m_dictionary_name.c_str(), debugger_sp); 2037*0b57cec5SDimitry Andric 203804eeddc0SDimitry Andric return StructuredData::GenericSP( 203904eeddc0SDimitry Andric new StructuredPythonObject(std::move(ret_val))); 2040*0b57cec5SDimitry Andric } 2041*0b57cec5SDimitry Andric 2042*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateTypeScriptFunction( 2043*0b57cec5SDimitry Andric const char *oneliner, std::string &output, const void *name_token) { 2044*0b57cec5SDimitry Andric StringList input; 2045*0b57cec5SDimitry Andric input.SplitIntoLines(oneliner, strlen(oneliner)); 2046*0b57cec5SDimitry Andric return GenerateTypeScriptFunction(input, output, name_token); 2047*0b57cec5SDimitry Andric } 2048*0b57cec5SDimitry Andric 2049*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateTypeSynthClass( 2050*0b57cec5SDimitry Andric const char *oneliner, std::string &output, const void *name_token) { 2051*0b57cec5SDimitry Andric StringList input; 2052*0b57cec5SDimitry Andric input.SplitIntoLines(oneliner, strlen(oneliner)); 2053*0b57cec5SDimitry Andric return GenerateTypeSynthClass(input, output, name_token); 2054*0b57cec5SDimitry Andric } 2055*0b57cec5SDimitry Andric 2056*0b57cec5SDimitry Andric Status ScriptInterpreterPythonImpl::GenerateBreakpointCommandCallbackData( 2057480093f4SDimitry Andric StringList &user_input, std::string &output, 2058480093f4SDimitry Andric bool has_extra_args) { 2059*0b57cec5SDimitry Andric static uint32_t num_created_functions = 0; 2060*0b57cec5SDimitry Andric user_input.RemoveBlankLines(); 2061*0b57cec5SDimitry Andric StreamString sstr; 2062*0b57cec5SDimitry Andric Status error; 2063*0b57cec5SDimitry Andric if (user_input.GetSize() == 0) { 2064*0b57cec5SDimitry Andric error.SetErrorString("No input data."); 2065*0b57cec5SDimitry Andric return error; 2066*0b57cec5SDimitry Andric } 2067*0b57cec5SDimitry Andric 2068*0b57cec5SDimitry Andric std::string auto_generated_function_name(GenerateUniqueName( 2069*0b57cec5SDimitry Andric "lldb_autogen_python_bp_callback_func_", num_created_functions)); 2070480093f4SDimitry Andric if (has_extra_args) 2071480093f4SDimitry Andric sstr.Printf("def %s (frame, bp_loc, extra_args, internal_dict):", 2072480093f4SDimitry Andric auto_generated_function_name.c_str()); 2073480093f4SDimitry Andric else 2074*0b57cec5SDimitry Andric sstr.Printf("def %s (frame, bp_loc, internal_dict):", 2075*0b57cec5SDimitry Andric auto_generated_function_name.c_str()); 2076*0b57cec5SDimitry Andric 2077*0b57cec5SDimitry Andric error = GenerateFunction(sstr.GetData(), user_input); 2078*0b57cec5SDimitry Andric if (!error.Success()) 2079*0b57cec5SDimitry Andric return error; 2080*0b57cec5SDimitry Andric 2081*0b57cec5SDimitry Andric // Store the name of the auto-generated function to be called. 2082*0b57cec5SDimitry Andric output.assign(auto_generated_function_name); 2083*0b57cec5SDimitry Andric return error; 2084*0b57cec5SDimitry Andric } 2085*0b57cec5SDimitry Andric 2086*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GenerateWatchpointCommandCallbackData( 2087*0b57cec5SDimitry Andric StringList &user_input, std::string &output) { 2088*0b57cec5SDimitry Andric static uint32_t num_created_functions = 0; 2089*0b57cec5SDimitry Andric user_input.RemoveBlankLines(); 2090*0b57cec5SDimitry Andric StreamString sstr; 2091*0b57cec5SDimitry Andric 2092*0b57cec5SDimitry Andric if (user_input.GetSize() == 0) 2093*0b57cec5SDimitry Andric return false; 2094*0b57cec5SDimitry Andric 2095*0b57cec5SDimitry Andric std::string auto_generated_function_name(GenerateUniqueName( 2096*0b57cec5SDimitry Andric "lldb_autogen_python_wp_callback_func_", num_created_functions)); 2097*0b57cec5SDimitry Andric sstr.Printf("def %s (frame, wp, internal_dict):", 2098*0b57cec5SDimitry Andric auto_generated_function_name.c_str()); 2099*0b57cec5SDimitry Andric 2100*0b57cec5SDimitry Andric if (!GenerateFunction(sstr.GetData(), user_input).Success()) 2101*0b57cec5SDimitry Andric return false; 2102*0b57cec5SDimitry Andric 2103*0b57cec5SDimitry Andric // Store the name of the auto-generated function to be called. 2104*0b57cec5SDimitry Andric output.assign(auto_generated_function_name); 2105*0b57cec5SDimitry Andric return true; 2106*0b57cec5SDimitry Andric } 2107*0b57cec5SDimitry Andric 2108*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GetScriptedSummary( 2109*0b57cec5SDimitry Andric const char *python_function_name, lldb::ValueObjectSP valobj, 2110*0b57cec5SDimitry Andric StructuredData::ObjectSP &callee_wrapper_sp, 2111*0b57cec5SDimitry Andric const TypeSummaryOptions &options, std::string &retval) { 2112*0b57cec5SDimitry Andric 2113e8d8bef9SDimitry Andric LLDB_SCOPED_TIMER(); 2114*0b57cec5SDimitry Andric 2115*0b57cec5SDimitry Andric if (!valobj.get()) { 2116*0b57cec5SDimitry Andric retval.assign("<no object>"); 2117*0b57cec5SDimitry Andric return false; 2118*0b57cec5SDimitry Andric } 2119*0b57cec5SDimitry Andric 2120*0b57cec5SDimitry Andric void *old_callee = nullptr; 2121*0b57cec5SDimitry Andric StructuredData::Generic *generic = nullptr; 2122*0b57cec5SDimitry Andric if (callee_wrapper_sp) { 2123*0b57cec5SDimitry Andric generic = callee_wrapper_sp->GetAsGeneric(); 2124*0b57cec5SDimitry Andric if (generic) 2125*0b57cec5SDimitry Andric old_callee = generic->GetValue(); 2126*0b57cec5SDimitry Andric } 2127*0b57cec5SDimitry Andric void *new_callee = old_callee; 2128*0b57cec5SDimitry Andric 2129*0b57cec5SDimitry Andric bool ret_val; 2130*0b57cec5SDimitry Andric if (python_function_name && *python_function_name) { 2131*0b57cec5SDimitry Andric { 2132*0b57cec5SDimitry Andric Locker py_lock(this, Locker::AcquireLock | Locker::InitSession | 2133*0b57cec5SDimitry Andric Locker::NoSTDIN); 2134*0b57cec5SDimitry Andric { 2135*0b57cec5SDimitry Andric TypeSummaryOptionsSP options_sp(new TypeSummaryOptions(options)); 2136*0b57cec5SDimitry Andric 2137*0b57cec5SDimitry Andric static Timer::Category func_cat("LLDBSwigPythonCallTypeScript"); 2138*0b57cec5SDimitry Andric Timer scoped_timer(func_cat, "LLDBSwigPythonCallTypeScript"); 2139*0b57cec5SDimitry Andric ret_val = LLDBSwigPythonCallTypeScript( 2140*0b57cec5SDimitry Andric python_function_name, GetSessionDictionary().get(), valobj, 2141*0b57cec5SDimitry Andric &new_callee, options_sp, retval); 2142*0b57cec5SDimitry Andric } 2143*0b57cec5SDimitry Andric } 2144*0b57cec5SDimitry Andric } else { 2145*0b57cec5SDimitry Andric retval.assign("<no function name>"); 2146*0b57cec5SDimitry Andric return false; 2147*0b57cec5SDimitry Andric } 2148*0b57cec5SDimitry Andric 214904eeddc0SDimitry Andric if (new_callee && old_callee != new_callee) { 215004eeddc0SDimitry Andric Locker py_lock(this, 215104eeddc0SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 215204eeddc0SDimitry Andric callee_wrapper_sp = std::make_shared<StructuredPythonObject>( 215304eeddc0SDimitry Andric PythonObject(PyRefType::Borrowed, static_cast<PyObject *>(new_callee))); 215404eeddc0SDimitry Andric } 2155*0b57cec5SDimitry Andric 2156*0b57cec5SDimitry Andric return ret_val; 2157*0b57cec5SDimitry Andric } 2158*0b57cec5SDimitry Andric 2159*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::BreakpointCallbackFunction( 2160*0b57cec5SDimitry Andric void *baton, StoppointCallbackContext *context, user_id_t break_id, 2161*0b57cec5SDimitry Andric user_id_t break_loc_id) { 2162*0b57cec5SDimitry Andric CommandDataPython *bp_option_data = (CommandDataPython *)baton; 2163*0b57cec5SDimitry Andric const char *python_function_name = bp_option_data->script_source.c_str(); 2164*0b57cec5SDimitry Andric 2165*0b57cec5SDimitry Andric if (!context) 2166*0b57cec5SDimitry Andric return true; 2167*0b57cec5SDimitry Andric 2168*0b57cec5SDimitry Andric ExecutionContext exe_ctx(context->exe_ctx_ref); 2169*0b57cec5SDimitry Andric Target *target = exe_ctx.GetTargetPtr(); 2170*0b57cec5SDimitry Andric 2171*0b57cec5SDimitry Andric if (!target) 2172*0b57cec5SDimitry Andric return true; 2173*0b57cec5SDimitry Andric 2174*0b57cec5SDimitry Andric Debugger &debugger = target->GetDebugger(); 2175*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl *python_interpreter = 2176e8d8bef9SDimitry Andric GetPythonInterpreter(debugger); 2177*0b57cec5SDimitry Andric 2178e8d8bef9SDimitry Andric if (!python_interpreter) 2179*0b57cec5SDimitry Andric return true; 2180*0b57cec5SDimitry Andric 2181*0b57cec5SDimitry Andric if (python_function_name && python_function_name[0]) { 2182*0b57cec5SDimitry Andric const StackFrameSP stop_frame_sp(exe_ctx.GetFrameSP()); 2183*0b57cec5SDimitry Andric BreakpointSP breakpoint_sp = target->GetBreakpointByID(break_id); 2184*0b57cec5SDimitry Andric if (breakpoint_sp) { 2185*0b57cec5SDimitry Andric const BreakpointLocationSP bp_loc_sp( 2186*0b57cec5SDimitry Andric breakpoint_sp->FindLocationByID(break_loc_id)); 2187*0b57cec5SDimitry Andric 2188*0b57cec5SDimitry Andric if (stop_frame_sp && bp_loc_sp) { 2189*0b57cec5SDimitry Andric bool ret_val = true; 2190*0b57cec5SDimitry Andric { 2191*0b57cec5SDimitry Andric Locker py_lock(python_interpreter, Locker::AcquireLock | 2192*0b57cec5SDimitry Andric Locker::InitSession | 2193*0b57cec5SDimitry Andric Locker::NoSTDIN); 2194480093f4SDimitry Andric Expected<bool> maybe_ret_val = 2195480093f4SDimitry Andric LLDBSwigPythonBreakpointCallbackFunction( 2196*0b57cec5SDimitry Andric python_function_name, 2197*0b57cec5SDimitry Andric python_interpreter->m_dictionary_name.c_str(), stop_frame_sp, 21980eae32dcSDimitry Andric bp_loc_sp, bp_option_data->m_extra_args); 2199480093f4SDimitry Andric 2200480093f4SDimitry Andric if (!maybe_ret_val) { 2201480093f4SDimitry Andric 2202480093f4SDimitry Andric llvm::handleAllErrors( 2203480093f4SDimitry Andric maybe_ret_val.takeError(), 2204480093f4SDimitry Andric [&](PythonException &E) { 2205480093f4SDimitry Andric debugger.GetErrorStream() << E.ReadBacktrace(); 2206480093f4SDimitry Andric }, 2207480093f4SDimitry Andric [&](const llvm::ErrorInfoBase &E) { 2208480093f4SDimitry Andric debugger.GetErrorStream() << E.message(); 2209480093f4SDimitry Andric }); 2210480093f4SDimitry Andric 2211480093f4SDimitry Andric } else { 2212480093f4SDimitry Andric ret_val = maybe_ret_val.get(); 2213480093f4SDimitry Andric } 2214*0b57cec5SDimitry Andric } 2215*0b57cec5SDimitry Andric return ret_val; 2216*0b57cec5SDimitry Andric } 2217*0b57cec5SDimitry Andric } 2218*0b57cec5SDimitry Andric } 2219*0b57cec5SDimitry Andric // We currently always true so we stop in case anything goes wrong when 2220*0b57cec5SDimitry Andric // trying to call the script function 2221*0b57cec5SDimitry Andric return true; 2222*0b57cec5SDimitry Andric } 2223*0b57cec5SDimitry Andric 2224*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::WatchpointCallbackFunction( 2225*0b57cec5SDimitry Andric void *baton, StoppointCallbackContext *context, user_id_t watch_id) { 2226*0b57cec5SDimitry Andric WatchpointOptions::CommandData *wp_option_data = 2227*0b57cec5SDimitry Andric (WatchpointOptions::CommandData *)baton; 2228*0b57cec5SDimitry Andric const char *python_function_name = wp_option_data->script_source.c_str(); 2229*0b57cec5SDimitry Andric 2230*0b57cec5SDimitry Andric if (!context) 2231*0b57cec5SDimitry Andric return true; 2232*0b57cec5SDimitry Andric 2233*0b57cec5SDimitry Andric ExecutionContext exe_ctx(context->exe_ctx_ref); 2234*0b57cec5SDimitry Andric Target *target = exe_ctx.GetTargetPtr(); 2235*0b57cec5SDimitry Andric 2236*0b57cec5SDimitry Andric if (!target) 2237*0b57cec5SDimitry Andric return true; 2238*0b57cec5SDimitry Andric 2239*0b57cec5SDimitry Andric Debugger &debugger = target->GetDebugger(); 2240*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl *python_interpreter = 2241e8d8bef9SDimitry Andric GetPythonInterpreter(debugger); 2242*0b57cec5SDimitry Andric 2243e8d8bef9SDimitry Andric if (!python_interpreter) 2244*0b57cec5SDimitry Andric return true; 2245*0b57cec5SDimitry Andric 2246*0b57cec5SDimitry Andric if (python_function_name && python_function_name[0]) { 2247*0b57cec5SDimitry Andric const StackFrameSP stop_frame_sp(exe_ctx.GetFrameSP()); 2248*0b57cec5SDimitry Andric WatchpointSP wp_sp = target->GetWatchpointList().FindByID(watch_id); 2249*0b57cec5SDimitry Andric if (wp_sp) { 2250*0b57cec5SDimitry Andric if (stop_frame_sp && wp_sp) { 2251*0b57cec5SDimitry Andric bool ret_val = true; 2252*0b57cec5SDimitry Andric { 2253*0b57cec5SDimitry Andric Locker py_lock(python_interpreter, Locker::AcquireLock | 2254*0b57cec5SDimitry Andric Locker::InitSession | 2255*0b57cec5SDimitry Andric Locker::NoSTDIN); 2256*0b57cec5SDimitry Andric ret_val = LLDBSwigPythonWatchpointCallbackFunction( 2257*0b57cec5SDimitry Andric python_function_name, 2258*0b57cec5SDimitry Andric python_interpreter->m_dictionary_name.c_str(), stop_frame_sp, 2259*0b57cec5SDimitry Andric wp_sp); 2260*0b57cec5SDimitry Andric } 2261*0b57cec5SDimitry Andric return ret_val; 2262*0b57cec5SDimitry Andric } 2263*0b57cec5SDimitry Andric } 2264*0b57cec5SDimitry Andric } 2265*0b57cec5SDimitry Andric // We currently always true so we stop in case anything goes wrong when 2266*0b57cec5SDimitry Andric // trying to call the script function 2267*0b57cec5SDimitry Andric return true; 2268*0b57cec5SDimitry Andric } 2269*0b57cec5SDimitry Andric 2270*0b57cec5SDimitry Andric size_t ScriptInterpreterPythonImpl::CalculateNumChildren( 2271*0b57cec5SDimitry Andric const StructuredData::ObjectSP &implementor_sp, uint32_t max) { 2272*0b57cec5SDimitry Andric if (!implementor_sp) 2273*0b57cec5SDimitry Andric return 0; 2274*0b57cec5SDimitry Andric StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 2275*0b57cec5SDimitry Andric if (!generic) 2276*0b57cec5SDimitry Andric return 0; 22774824e7fdSDimitry Andric auto *implementor = static_cast<PyObject *>(generic->GetValue()); 2278*0b57cec5SDimitry Andric if (!implementor) 2279*0b57cec5SDimitry Andric return 0; 2280*0b57cec5SDimitry Andric 2281*0b57cec5SDimitry Andric size_t ret_val = 0; 2282*0b57cec5SDimitry Andric 2283*0b57cec5SDimitry Andric { 2284*0b57cec5SDimitry Andric Locker py_lock(this, 2285*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2286*0b57cec5SDimitry Andric ret_val = LLDBSwigPython_CalculateNumChildren(implementor, max); 2287*0b57cec5SDimitry Andric } 2288*0b57cec5SDimitry Andric 2289*0b57cec5SDimitry Andric return ret_val; 2290*0b57cec5SDimitry Andric } 2291*0b57cec5SDimitry Andric 2292*0b57cec5SDimitry Andric lldb::ValueObjectSP ScriptInterpreterPythonImpl::GetChildAtIndex( 2293*0b57cec5SDimitry Andric const StructuredData::ObjectSP &implementor_sp, uint32_t idx) { 2294*0b57cec5SDimitry Andric if (!implementor_sp) 2295*0b57cec5SDimitry Andric return lldb::ValueObjectSP(); 2296*0b57cec5SDimitry Andric 2297*0b57cec5SDimitry Andric StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 2298*0b57cec5SDimitry Andric if (!generic) 2299*0b57cec5SDimitry Andric return lldb::ValueObjectSP(); 23004824e7fdSDimitry Andric auto *implementor = static_cast<PyObject *>(generic->GetValue()); 2301*0b57cec5SDimitry Andric if (!implementor) 2302*0b57cec5SDimitry Andric return lldb::ValueObjectSP(); 2303*0b57cec5SDimitry Andric 2304*0b57cec5SDimitry Andric lldb::ValueObjectSP ret_val; 2305*0b57cec5SDimitry Andric { 2306*0b57cec5SDimitry Andric Locker py_lock(this, 2307*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 23084824e7fdSDimitry Andric PyObject *child_ptr = LLDBSwigPython_GetChildAtIndex(implementor, idx); 2309*0b57cec5SDimitry Andric if (child_ptr != nullptr && child_ptr != Py_None) { 2310*0b57cec5SDimitry Andric lldb::SBValue *sb_value_ptr = 2311*0b57cec5SDimitry Andric (lldb::SBValue *)LLDBSWIGPython_CastPyObjectToSBValue(child_ptr); 2312*0b57cec5SDimitry Andric if (sb_value_ptr == nullptr) 2313*0b57cec5SDimitry Andric Py_XDECREF(child_ptr); 2314*0b57cec5SDimitry Andric else 2315*0b57cec5SDimitry Andric ret_val = LLDBSWIGPython_GetValueObjectSPFromSBValue(sb_value_ptr); 2316*0b57cec5SDimitry Andric } else { 2317*0b57cec5SDimitry Andric Py_XDECREF(child_ptr); 2318*0b57cec5SDimitry Andric } 2319*0b57cec5SDimitry Andric } 2320*0b57cec5SDimitry Andric 2321*0b57cec5SDimitry Andric return ret_val; 2322*0b57cec5SDimitry Andric } 2323*0b57cec5SDimitry Andric 2324*0b57cec5SDimitry Andric int ScriptInterpreterPythonImpl::GetIndexOfChildWithName( 2325*0b57cec5SDimitry Andric const StructuredData::ObjectSP &implementor_sp, const char *child_name) { 2326*0b57cec5SDimitry Andric if (!implementor_sp) 2327*0b57cec5SDimitry Andric return UINT32_MAX; 2328*0b57cec5SDimitry Andric 2329*0b57cec5SDimitry Andric StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 2330*0b57cec5SDimitry Andric if (!generic) 2331*0b57cec5SDimitry Andric return UINT32_MAX; 23324824e7fdSDimitry Andric auto *implementor = static_cast<PyObject *>(generic->GetValue()); 2333*0b57cec5SDimitry Andric if (!implementor) 2334*0b57cec5SDimitry Andric return UINT32_MAX; 2335*0b57cec5SDimitry Andric 2336*0b57cec5SDimitry Andric int ret_val = UINT32_MAX; 2337*0b57cec5SDimitry Andric 2338*0b57cec5SDimitry Andric { 2339*0b57cec5SDimitry Andric Locker py_lock(this, 2340*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2341*0b57cec5SDimitry Andric ret_val = LLDBSwigPython_GetIndexOfChildWithName(implementor, child_name); 2342*0b57cec5SDimitry Andric } 2343*0b57cec5SDimitry Andric 2344*0b57cec5SDimitry Andric return ret_val; 2345*0b57cec5SDimitry Andric } 2346*0b57cec5SDimitry Andric 2347*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::UpdateSynthProviderInstance( 2348*0b57cec5SDimitry Andric const StructuredData::ObjectSP &implementor_sp) { 2349*0b57cec5SDimitry Andric bool ret_val = false; 2350*0b57cec5SDimitry Andric 2351*0b57cec5SDimitry Andric if (!implementor_sp) 2352*0b57cec5SDimitry Andric return ret_val; 2353*0b57cec5SDimitry Andric 2354*0b57cec5SDimitry Andric StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 2355*0b57cec5SDimitry Andric if (!generic) 2356*0b57cec5SDimitry Andric return ret_val; 23574824e7fdSDimitry Andric auto *implementor = static_cast<PyObject *>(generic->GetValue()); 2358*0b57cec5SDimitry Andric if (!implementor) 2359*0b57cec5SDimitry Andric return ret_val; 2360*0b57cec5SDimitry Andric 2361*0b57cec5SDimitry Andric { 2362*0b57cec5SDimitry Andric Locker py_lock(this, 2363*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2364*0b57cec5SDimitry Andric ret_val = LLDBSwigPython_UpdateSynthProviderInstance(implementor); 2365*0b57cec5SDimitry Andric } 2366*0b57cec5SDimitry Andric 2367*0b57cec5SDimitry Andric return ret_val; 2368*0b57cec5SDimitry Andric } 2369*0b57cec5SDimitry Andric 2370*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::MightHaveChildrenSynthProviderInstance( 2371*0b57cec5SDimitry Andric const StructuredData::ObjectSP &implementor_sp) { 2372*0b57cec5SDimitry Andric bool ret_val = false; 2373*0b57cec5SDimitry Andric 2374*0b57cec5SDimitry Andric if (!implementor_sp) 2375*0b57cec5SDimitry Andric return ret_val; 2376*0b57cec5SDimitry Andric 2377*0b57cec5SDimitry Andric StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 2378*0b57cec5SDimitry Andric if (!generic) 2379*0b57cec5SDimitry Andric return ret_val; 23804824e7fdSDimitry Andric auto *implementor = static_cast<PyObject *>(generic->GetValue()); 2381*0b57cec5SDimitry Andric if (!implementor) 2382*0b57cec5SDimitry Andric return ret_val; 2383*0b57cec5SDimitry Andric 2384*0b57cec5SDimitry Andric { 2385*0b57cec5SDimitry Andric Locker py_lock(this, 2386*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2387*0b57cec5SDimitry Andric ret_val = 2388*0b57cec5SDimitry Andric LLDBSwigPython_MightHaveChildrenSynthProviderInstance(implementor); 2389*0b57cec5SDimitry Andric } 2390*0b57cec5SDimitry Andric 2391*0b57cec5SDimitry Andric return ret_val; 2392*0b57cec5SDimitry Andric } 2393*0b57cec5SDimitry Andric 2394*0b57cec5SDimitry Andric lldb::ValueObjectSP ScriptInterpreterPythonImpl::GetSyntheticValue( 2395*0b57cec5SDimitry Andric const StructuredData::ObjectSP &implementor_sp) { 2396*0b57cec5SDimitry Andric lldb::ValueObjectSP ret_val(nullptr); 2397*0b57cec5SDimitry Andric 2398*0b57cec5SDimitry Andric if (!implementor_sp) 2399*0b57cec5SDimitry Andric return ret_val; 2400*0b57cec5SDimitry Andric 2401*0b57cec5SDimitry Andric StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 2402*0b57cec5SDimitry Andric if (!generic) 2403*0b57cec5SDimitry Andric return ret_val; 24044824e7fdSDimitry Andric auto *implementor = static_cast<PyObject *>(generic->GetValue()); 2405*0b57cec5SDimitry Andric if (!implementor) 2406*0b57cec5SDimitry Andric return ret_val; 2407*0b57cec5SDimitry Andric 2408*0b57cec5SDimitry Andric { 2409*0b57cec5SDimitry Andric Locker py_lock(this, 2410*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 24114824e7fdSDimitry Andric PyObject *child_ptr = 24124824e7fdSDimitry Andric LLDBSwigPython_GetValueSynthProviderInstance(implementor); 2413*0b57cec5SDimitry Andric if (child_ptr != nullptr && child_ptr != Py_None) { 2414*0b57cec5SDimitry Andric lldb::SBValue *sb_value_ptr = 2415*0b57cec5SDimitry Andric (lldb::SBValue *)LLDBSWIGPython_CastPyObjectToSBValue(child_ptr); 2416*0b57cec5SDimitry Andric if (sb_value_ptr == nullptr) 2417*0b57cec5SDimitry Andric Py_XDECREF(child_ptr); 2418*0b57cec5SDimitry Andric else 2419*0b57cec5SDimitry Andric ret_val = LLDBSWIGPython_GetValueObjectSPFromSBValue(sb_value_ptr); 2420*0b57cec5SDimitry Andric } else { 2421*0b57cec5SDimitry Andric Py_XDECREF(child_ptr); 2422*0b57cec5SDimitry Andric } 2423*0b57cec5SDimitry Andric } 2424*0b57cec5SDimitry Andric 2425*0b57cec5SDimitry Andric return ret_val; 2426*0b57cec5SDimitry Andric } 2427*0b57cec5SDimitry Andric 2428*0b57cec5SDimitry Andric ConstString ScriptInterpreterPythonImpl::GetSyntheticTypeName( 2429*0b57cec5SDimitry Andric const StructuredData::ObjectSP &implementor_sp) { 2430*0b57cec5SDimitry Andric Locker py_lock(this, 2431*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2432*0b57cec5SDimitry Andric 2433*0b57cec5SDimitry Andric static char callee_name[] = "get_type_name"; 2434*0b57cec5SDimitry Andric 2435*0b57cec5SDimitry Andric ConstString ret_val; 2436*0b57cec5SDimitry Andric bool got_string = false; 2437*0b57cec5SDimitry Andric std::string buffer; 2438*0b57cec5SDimitry Andric 2439*0b57cec5SDimitry Andric if (!implementor_sp) 2440*0b57cec5SDimitry Andric return ret_val; 2441*0b57cec5SDimitry Andric 2442*0b57cec5SDimitry Andric StructuredData::Generic *generic = implementor_sp->GetAsGeneric(); 2443*0b57cec5SDimitry Andric if (!generic) 2444*0b57cec5SDimitry Andric return ret_val; 2445*0b57cec5SDimitry Andric PythonObject implementor(PyRefType::Borrowed, 2446*0b57cec5SDimitry Andric (PyObject *)generic->GetValue()); 2447*0b57cec5SDimitry Andric if (!implementor.IsAllocated()) 2448*0b57cec5SDimitry Andric return ret_val; 2449*0b57cec5SDimitry Andric 2450*0b57cec5SDimitry Andric PythonObject pmeth(PyRefType::Owned, 2451*0b57cec5SDimitry Andric PyObject_GetAttrString(implementor.get(), callee_name)); 2452*0b57cec5SDimitry Andric 2453*0b57cec5SDimitry Andric if (PyErr_Occurred()) 2454*0b57cec5SDimitry Andric PyErr_Clear(); 2455*0b57cec5SDimitry Andric 2456*0b57cec5SDimitry Andric if (!pmeth.IsAllocated()) 2457*0b57cec5SDimitry Andric return ret_val; 2458*0b57cec5SDimitry Andric 2459*0b57cec5SDimitry Andric if (PyCallable_Check(pmeth.get()) == 0) { 2460*0b57cec5SDimitry Andric if (PyErr_Occurred()) 2461*0b57cec5SDimitry Andric PyErr_Clear(); 2462*0b57cec5SDimitry Andric return ret_val; 2463*0b57cec5SDimitry Andric } 2464*0b57cec5SDimitry Andric 2465*0b57cec5SDimitry Andric if (PyErr_Occurred()) 2466*0b57cec5SDimitry Andric PyErr_Clear(); 2467*0b57cec5SDimitry Andric 2468*0b57cec5SDimitry Andric // right now we know this function exists and is callable.. 2469*0b57cec5SDimitry Andric PythonObject py_return( 2470*0b57cec5SDimitry Andric PyRefType::Owned, 2471*0b57cec5SDimitry Andric PyObject_CallMethod(implementor.get(), callee_name, nullptr)); 2472*0b57cec5SDimitry Andric 2473*0b57cec5SDimitry Andric // if it fails, print the error but otherwise go on 2474*0b57cec5SDimitry Andric if (PyErr_Occurred()) { 2475*0b57cec5SDimitry Andric PyErr_Print(); 2476*0b57cec5SDimitry Andric PyErr_Clear(); 2477*0b57cec5SDimitry Andric } 2478*0b57cec5SDimitry Andric 2479*0b57cec5SDimitry Andric if (py_return.IsAllocated() && PythonString::Check(py_return.get())) { 2480*0b57cec5SDimitry Andric PythonString py_string(PyRefType::Borrowed, py_return.get()); 2481*0b57cec5SDimitry Andric llvm::StringRef return_data(py_string.GetString()); 2482*0b57cec5SDimitry Andric if (!return_data.empty()) { 2483*0b57cec5SDimitry Andric buffer.assign(return_data.data(), return_data.size()); 2484*0b57cec5SDimitry Andric got_string = true; 2485*0b57cec5SDimitry Andric } 2486*0b57cec5SDimitry Andric } 2487*0b57cec5SDimitry Andric 2488*0b57cec5SDimitry Andric if (got_string) 2489*0b57cec5SDimitry Andric ret_val.SetCStringWithLength(buffer.c_str(), buffer.size()); 2490*0b57cec5SDimitry Andric 2491*0b57cec5SDimitry Andric return ret_val; 2492*0b57cec5SDimitry Andric } 2493*0b57cec5SDimitry Andric 2494*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptFormatKeyword( 2495*0b57cec5SDimitry Andric const char *impl_function, Process *process, std::string &output, 2496*0b57cec5SDimitry Andric Status &error) { 2497*0b57cec5SDimitry Andric bool ret_val; 2498*0b57cec5SDimitry Andric if (!process) { 2499*0b57cec5SDimitry Andric error.SetErrorString("no process"); 2500*0b57cec5SDimitry Andric return false; 2501*0b57cec5SDimitry Andric } 2502*0b57cec5SDimitry Andric if (!impl_function || !impl_function[0]) { 2503*0b57cec5SDimitry Andric error.SetErrorString("no function to execute"); 2504*0b57cec5SDimitry Andric return false; 2505*0b57cec5SDimitry Andric } 2506*0b57cec5SDimitry Andric 2507*0b57cec5SDimitry Andric { 2508*0b57cec5SDimitry Andric Locker py_lock(this, 2509*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2510*0b57cec5SDimitry Andric ret_val = LLDBSWIGPythonRunScriptKeywordProcess( 25114824e7fdSDimitry Andric impl_function, m_dictionary_name.c_str(), process->shared_from_this(), 25124824e7fdSDimitry Andric output); 2513*0b57cec5SDimitry Andric if (!ret_val) 2514*0b57cec5SDimitry Andric error.SetErrorString("python script evaluation failed"); 2515*0b57cec5SDimitry Andric } 2516*0b57cec5SDimitry Andric return ret_val; 2517*0b57cec5SDimitry Andric } 2518*0b57cec5SDimitry Andric 2519*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptFormatKeyword( 2520*0b57cec5SDimitry Andric const char *impl_function, Thread *thread, std::string &output, 2521*0b57cec5SDimitry Andric Status &error) { 2522*0b57cec5SDimitry Andric if (!thread) { 2523*0b57cec5SDimitry Andric error.SetErrorString("no thread"); 2524*0b57cec5SDimitry Andric return false; 2525*0b57cec5SDimitry Andric } 2526*0b57cec5SDimitry Andric if (!impl_function || !impl_function[0]) { 2527*0b57cec5SDimitry Andric error.SetErrorString("no function to execute"); 2528*0b57cec5SDimitry Andric return false; 2529*0b57cec5SDimitry Andric } 2530*0b57cec5SDimitry Andric 2531*0b57cec5SDimitry Andric Locker py_lock(this, 2532*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 25330eae32dcSDimitry Andric if (llvm::Optional<std::string> result = LLDBSWIGPythonRunScriptKeywordThread( 25340eae32dcSDimitry Andric impl_function, m_dictionary_name.c_str(), 25350eae32dcSDimitry Andric thread->shared_from_this())) { 25360eae32dcSDimitry Andric output = std::move(*result); 25370eae32dcSDimitry Andric return true; 2538*0b57cec5SDimitry Andric } 25390eae32dcSDimitry Andric error.SetErrorString("python script evaluation failed"); 25400eae32dcSDimitry Andric return false; 2541*0b57cec5SDimitry Andric } 2542*0b57cec5SDimitry Andric 2543*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptFormatKeyword( 2544*0b57cec5SDimitry Andric const char *impl_function, Target *target, std::string &output, 2545*0b57cec5SDimitry Andric Status &error) { 2546*0b57cec5SDimitry Andric bool ret_val; 2547*0b57cec5SDimitry Andric if (!target) { 2548*0b57cec5SDimitry Andric error.SetErrorString("no thread"); 2549*0b57cec5SDimitry Andric return false; 2550*0b57cec5SDimitry Andric } 2551*0b57cec5SDimitry Andric if (!impl_function || !impl_function[0]) { 2552*0b57cec5SDimitry Andric error.SetErrorString("no function to execute"); 2553*0b57cec5SDimitry Andric return false; 2554*0b57cec5SDimitry Andric } 2555*0b57cec5SDimitry Andric 2556*0b57cec5SDimitry Andric { 2557*0b57cec5SDimitry Andric TargetSP target_sp(target->shared_from_this()); 2558*0b57cec5SDimitry Andric Locker py_lock(this, 2559*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2560*0b57cec5SDimitry Andric ret_val = LLDBSWIGPythonRunScriptKeywordTarget( 2561*0b57cec5SDimitry Andric impl_function, m_dictionary_name.c_str(), target_sp, output); 2562*0b57cec5SDimitry Andric if (!ret_val) 2563*0b57cec5SDimitry Andric error.SetErrorString("python script evaluation failed"); 2564*0b57cec5SDimitry Andric } 2565*0b57cec5SDimitry Andric return ret_val; 2566*0b57cec5SDimitry Andric } 2567*0b57cec5SDimitry Andric 2568*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptFormatKeyword( 2569*0b57cec5SDimitry Andric const char *impl_function, StackFrame *frame, std::string &output, 2570*0b57cec5SDimitry Andric Status &error) { 2571*0b57cec5SDimitry Andric if (!frame) { 2572*0b57cec5SDimitry Andric error.SetErrorString("no frame"); 2573*0b57cec5SDimitry Andric return false; 2574*0b57cec5SDimitry Andric } 2575*0b57cec5SDimitry Andric if (!impl_function || !impl_function[0]) { 2576*0b57cec5SDimitry Andric error.SetErrorString("no function to execute"); 2577*0b57cec5SDimitry Andric return false; 2578*0b57cec5SDimitry Andric } 2579*0b57cec5SDimitry Andric 2580*0b57cec5SDimitry Andric Locker py_lock(this, 2581*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 25820eae32dcSDimitry Andric if (llvm::Optional<std::string> result = LLDBSWIGPythonRunScriptKeywordFrame( 25830eae32dcSDimitry Andric impl_function, m_dictionary_name.c_str(), 25840eae32dcSDimitry Andric frame->shared_from_this())) { 25850eae32dcSDimitry Andric output = std::move(*result); 25860eae32dcSDimitry Andric return true; 2587*0b57cec5SDimitry Andric } 25880eae32dcSDimitry Andric error.SetErrorString("python script evaluation failed"); 25890eae32dcSDimitry Andric return false; 2590*0b57cec5SDimitry Andric } 2591*0b57cec5SDimitry Andric 2592*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptFormatKeyword( 2593*0b57cec5SDimitry Andric const char *impl_function, ValueObject *value, std::string &output, 2594*0b57cec5SDimitry Andric Status &error) { 2595*0b57cec5SDimitry Andric bool ret_val; 2596*0b57cec5SDimitry Andric if (!value) { 2597*0b57cec5SDimitry Andric error.SetErrorString("no value"); 2598*0b57cec5SDimitry Andric return false; 2599*0b57cec5SDimitry Andric } 2600*0b57cec5SDimitry Andric if (!impl_function || !impl_function[0]) { 2601*0b57cec5SDimitry Andric error.SetErrorString("no function to execute"); 2602*0b57cec5SDimitry Andric return false; 2603*0b57cec5SDimitry Andric } 2604*0b57cec5SDimitry Andric 2605*0b57cec5SDimitry Andric { 2606*0b57cec5SDimitry Andric Locker py_lock(this, 2607*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN); 2608*0b57cec5SDimitry Andric ret_val = LLDBSWIGPythonRunScriptKeywordValue( 26094824e7fdSDimitry Andric impl_function, m_dictionary_name.c_str(), value->GetSP(), output); 2610*0b57cec5SDimitry Andric if (!ret_val) 2611*0b57cec5SDimitry Andric error.SetErrorString("python script evaluation failed"); 2612*0b57cec5SDimitry Andric } 2613*0b57cec5SDimitry Andric return ret_val; 2614*0b57cec5SDimitry Andric } 2615*0b57cec5SDimitry Andric 2616*0b57cec5SDimitry Andric uint64_t replace_all(std::string &str, const std::string &oldStr, 2617*0b57cec5SDimitry Andric const std::string &newStr) { 2618*0b57cec5SDimitry Andric size_t pos = 0; 2619*0b57cec5SDimitry Andric uint64_t matches = 0; 2620*0b57cec5SDimitry Andric while ((pos = str.find(oldStr, pos)) != std::string::npos) { 2621*0b57cec5SDimitry Andric matches++; 2622*0b57cec5SDimitry Andric str.replace(pos, oldStr.length(), newStr); 2623*0b57cec5SDimitry Andric pos += newStr.length(); 2624*0b57cec5SDimitry Andric } 2625*0b57cec5SDimitry Andric return matches; 2626*0b57cec5SDimitry Andric } 2627*0b57cec5SDimitry Andric 2628*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::LoadScriptingModule( 2629fe6060f1SDimitry Andric const char *pathname, const LoadScriptOptions &options, 2630fe6060f1SDimitry Andric lldb_private::Status &error, StructuredData::ObjectSP *module_sp, 2631fe6060f1SDimitry Andric FileSpec extra_search_dir) { 2632e8d8bef9SDimitry Andric namespace fs = llvm::sys::fs; 2633e8d8bef9SDimitry Andric namespace path = llvm::sys::path; 2634e8d8bef9SDimitry Andric 2635fe6060f1SDimitry Andric ExecuteScriptOptions exc_options = ExecuteScriptOptions() 2636fe6060f1SDimitry Andric .SetEnableIO(!options.GetSilent()) 2637fe6060f1SDimitry Andric .SetSetLLDBGlobals(false); 2638fe6060f1SDimitry Andric 2639*0b57cec5SDimitry Andric if (!pathname || !pathname[0]) { 2640fcaf7f86SDimitry Andric error.SetErrorString("empty path"); 2641*0b57cec5SDimitry Andric return false; 2642*0b57cec5SDimitry Andric } 2643*0b57cec5SDimitry Andric 2644fe6060f1SDimitry Andric llvm::Expected<std::unique_ptr<ScriptInterpreterIORedirect>> 2645fe6060f1SDimitry Andric io_redirect_or_error = ScriptInterpreterIORedirect::Create( 2646fe6060f1SDimitry Andric exc_options.GetEnableIO(), m_debugger, /*result=*/nullptr); 2647fe6060f1SDimitry Andric 2648fe6060f1SDimitry Andric if (!io_redirect_or_error) { 2649fe6060f1SDimitry Andric error = io_redirect_or_error.takeError(); 2650fe6060f1SDimitry Andric return false; 2651fe6060f1SDimitry Andric } 2652fe6060f1SDimitry Andric 2653fe6060f1SDimitry Andric ScriptInterpreterIORedirect &io_redirect = **io_redirect_or_error; 2654*0b57cec5SDimitry Andric 2655*0b57cec5SDimitry Andric // Before executing Python code, lock the GIL. 2656*0b57cec5SDimitry Andric Locker py_lock(this, 2657*0b57cec5SDimitry Andric Locker::AcquireLock | 2658fe6060f1SDimitry Andric (options.GetInitSession() ? Locker::InitSession : 0) | 2659fe6060f1SDimitry Andric Locker::NoSTDIN, 2660*0b57cec5SDimitry Andric Locker::FreeAcquiredLock | 2661fe6060f1SDimitry Andric (options.GetInitSession() ? Locker::TearDownSession : 0), 2662fe6060f1SDimitry Andric io_redirect.GetInputFile(), io_redirect.GetOutputFile(), 2663fe6060f1SDimitry Andric io_redirect.GetErrorFile()); 2664*0b57cec5SDimitry Andric 2665fe6060f1SDimitry Andric auto ExtendSysPath = [&](std::string directory) -> llvm::Error { 2666e8d8bef9SDimitry Andric if (directory.empty()) { 2667e8d8bef9SDimitry Andric return llvm::make_error<llvm::StringError>( 2668e8d8bef9SDimitry Andric "invalid directory name", llvm::inconvertibleErrorCode()); 2669*0b57cec5SDimitry Andric } 2670*0b57cec5SDimitry Andric 2671*0b57cec5SDimitry Andric replace_all(directory, "\\", "\\\\"); 2672*0b57cec5SDimitry Andric replace_all(directory, "'", "\\'"); 2673*0b57cec5SDimitry Andric 2674e8d8bef9SDimitry Andric // Make sure that Python has "directory" in the search path. 2675*0b57cec5SDimitry Andric StreamString command_stream; 2676*0b57cec5SDimitry Andric command_stream.Printf("if not (sys.path.__contains__('%s')):\n " 2677*0b57cec5SDimitry Andric "sys.path.insert(1,'%s');\n\n", 2678*0b57cec5SDimitry Andric directory.c_str(), directory.c_str()); 2679*0b57cec5SDimitry Andric bool syspath_retval = 2680fe6060f1SDimitry Andric ExecuteMultipleLines(command_stream.GetData(), exc_options).Success(); 2681*0b57cec5SDimitry Andric if (!syspath_retval) { 2682e8d8bef9SDimitry Andric return llvm::make_error<llvm::StringError>( 2683e8d8bef9SDimitry Andric "Python sys.path handling failed", llvm::inconvertibleErrorCode()); 2684*0b57cec5SDimitry Andric } 2685*0b57cec5SDimitry Andric 2686e8d8bef9SDimitry Andric return llvm::Error::success(); 2687e8d8bef9SDimitry Andric }; 2688e8d8bef9SDimitry Andric 2689e8d8bef9SDimitry Andric std::string module_name(pathname); 2690fe6060f1SDimitry Andric bool possible_package = false; 2691e8d8bef9SDimitry Andric 2692e8d8bef9SDimitry Andric if (extra_search_dir) { 2693e8d8bef9SDimitry Andric if (llvm::Error e = ExtendSysPath(extra_search_dir.GetPath())) { 2694e8d8bef9SDimitry Andric error = std::move(e); 2695e8d8bef9SDimitry Andric return false; 2696*0b57cec5SDimitry Andric } 2697*0b57cec5SDimitry Andric } else { 2698e8d8bef9SDimitry Andric FileSpec module_file(pathname); 2699e8d8bef9SDimitry Andric FileSystem::Instance().Resolve(module_file); 2700e8d8bef9SDimitry Andric 2701e8d8bef9SDimitry Andric fs::file_status st; 2702e8d8bef9SDimitry Andric std::error_code ec = status(module_file.GetPath(), st); 2703e8d8bef9SDimitry Andric 2704e8d8bef9SDimitry Andric if (ec || st.type() == fs::file_type::status_error || 2705e8d8bef9SDimitry Andric st.type() == fs::file_type::type_unknown || 2706e8d8bef9SDimitry Andric st.type() == fs::file_type::file_not_found) { 2707e8d8bef9SDimitry Andric // if not a valid file of any sort, check if it might be a filename still 2708e8d8bef9SDimitry Andric // dot can't be used but / and \ can, and if either is found, reject 2709e8d8bef9SDimitry Andric if (strchr(pathname, '\\') || strchr(pathname, '/')) { 2710fcaf7f86SDimitry Andric error.SetErrorStringWithFormatv("invalid pathname '{0}'", pathname); 2711e8d8bef9SDimitry Andric return false; 2712e8d8bef9SDimitry Andric } 2713e8d8bef9SDimitry Andric // Not a filename, probably a package of some sort, let it go through. 2714fe6060f1SDimitry Andric possible_package = true; 2715e8d8bef9SDimitry Andric } else if (is_directory(st) || is_regular_file(st)) { 2716e8d8bef9SDimitry Andric if (module_file.GetDirectory().IsEmpty()) { 2717fcaf7f86SDimitry Andric error.SetErrorStringWithFormatv("invalid directory name '{0}'", pathname); 2718e8d8bef9SDimitry Andric return false; 2719e8d8bef9SDimitry Andric } 2720e8d8bef9SDimitry Andric if (llvm::Error e = 2721e8d8bef9SDimitry Andric ExtendSysPath(module_file.GetDirectory().GetCString())) { 2722e8d8bef9SDimitry Andric error = std::move(e); 2723e8d8bef9SDimitry Andric return false; 2724e8d8bef9SDimitry Andric } 2725e8d8bef9SDimitry Andric module_name = module_file.GetFilename().GetCString(); 2726e8d8bef9SDimitry Andric } else { 2727*0b57cec5SDimitry Andric error.SetErrorString("no known way to import this module specification"); 2728*0b57cec5SDimitry Andric return false; 2729*0b57cec5SDimitry Andric } 2730e8d8bef9SDimitry Andric } 2731e8d8bef9SDimitry Andric 2732e8d8bef9SDimitry Andric // Strip .py or .pyc extension 2733e8d8bef9SDimitry Andric llvm::StringRef extension = llvm::sys::path::extension(module_name); 2734e8d8bef9SDimitry Andric if (!extension.empty()) { 2735e8d8bef9SDimitry Andric if (extension == ".py") 2736e8d8bef9SDimitry Andric module_name.resize(module_name.length() - 3); 2737e8d8bef9SDimitry Andric else if (extension == ".pyc") 2738e8d8bef9SDimitry Andric module_name.resize(module_name.length() - 4); 2739e8d8bef9SDimitry Andric } 2740*0b57cec5SDimitry Andric 2741fe6060f1SDimitry Andric if (!possible_package && module_name.find('.') != llvm::StringRef::npos) { 2742fe6060f1SDimitry Andric error.SetErrorStringWithFormat( 2743fe6060f1SDimitry Andric "Python does not allow dots in module names: %s", module_name.c_str()); 2744fe6060f1SDimitry Andric return false; 2745fe6060f1SDimitry Andric } 2746fe6060f1SDimitry Andric 2747fe6060f1SDimitry Andric if (module_name.find('-') != llvm::StringRef::npos) { 2748fe6060f1SDimitry Andric error.SetErrorStringWithFormat( 2749fe6060f1SDimitry Andric "Python discourages dashes in module names: %s", module_name.c_str()); 2750fe6060f1SDimitry Andric return false; 2751fe6060f1SDimitry Andric } 2752fe6060f1SDimitry Andric 2753fe6060f1SDimitry Andric // Check if the module is already imported. 2754e8d8bef9SDimitry Andric StreamString command_stream; 2755*0b57cec5SDimitry Andric command_stream.Clear(); 2756e8d8bef9SDimitry Andric command_stream.Printf("sys.modules.__contains__('%s')", module_name.c_str()); 2757*0b57cec5SDimitry Andric bool does_contain = false; 2758fe6060f1SDimitry Andric // This call will succeed if the module was ever imported in any Debugger in 2759fe6060f1SDimitry Andric // the lifetime of the process in which this LLDB framework is living. 2760fe6060f1SDimitry Andric const bool does_contain_executed = ExecuteOneLineWithReturn( 2761*0b57cec5SDimitry Andric command_stream.GetData(), 2762fe6060f1SDimitry Andric ScriptInterpreterPythonImpl::eScriptReturnTypeBool, &does_contain, exc_options); 2763fe6060f1SDimitry Andric 2764fe6060f1SDimitry Andric const bool was_imported_globally = does_contain_executed && does_contain; 2765fe6060f1SDimitry Andric const bool was_imported_locally = 2766fe6060f1SDimitry Andric GetSessionDictionary() 2767e8d8bef9SDimitry Andric .GetItemForKey(PythonString(module_name)) 2768*0b57cec5SDimitry Andric .IsAllocated(); 2769*0b57cec5SDimitry Andric 2770*0b57cec5SDimitry Andric // now actually do the import 2771*0b57cec5SDimitry Andric command_stream.Clear(); 2772*0b57cec5SDimitry Andric 2773fe6060f1SDimitry Andric if (was_imported_globally || was_imported_locally) { 2774*0b57cec5SDimitry Andric if (!was_imported_locally) 2775e8d8bef9SDimitry Andric command_stream.Printf("import %s ; reload_module(%s)", 2776e8d8bef9SDimitry Andric module_name.c_str(), module_name.c_str()); 2777*0b57cec5SDimitry Andric else 2778e8d8bef9SDimitry Andric command_stream.Printf("reload_module(%s)", module_name.c_str()); 2779*0b57cec5SDimitry Andric } else 2780e8d8bef9SDimitry Andric command_stream.Printf("import %s", module_name.c_str()); 2781*0b57cec5SDimitry Andric 2782fe6060f1SDimitry Andric error = ExecuteMultipleLines(command_stream.GetData(), exc_options); 2783*0b57cec5SDimitry Andric if (error.Fail()) 2784*0b57cec5SDimitry Andric return false; 2785*0b57cec5SDimitry Andric 2786*0b57cec5SDimitry Andric // if we are here, everything worked 2787*0b57cec5SDimitry Andric // call __lldb_init_module(debugger,dict) 2788e8d8bef9SDimitry Andric if (!LLDBSwigPythonCallModuleInit(module_name.c_str(), 27890eae32dcSDimitry Andric m_dictionary_name.c_str(), 27900eae32dcSDimitry Andric m_debugger.shared_from_this())) { 2791*0b57cec5SDimitry Andric error.SetErrorString("calling __lldb_init_module failed"); 2792*0b57cec5SDimitry Andric return false; 2793*0b57cec5SDimitry Andric } 2794*0b57cec5SDimitry Andric 2795*0b57cec5SDimitry Andric if (module_sp) { 2796*0b57cec5SDimitry Andric // everything went just great, now set the module object 2797*0b57cec5SDimitry Andric command_stream.Clear(); 2798e8d8bef9SDimitry Andric command_stream.Printf("%s", module_name.c_str()); 2799*0b57cec5SDimitry Andric void *module_pyobj = nullptr; 2800*0b57cec5SDimitry Andric if (ExecuteOneLineWithReturn( 2801*0b57cec5SDimitry Andric command_stream.GetData(), 2802fe6060f1SDimitry Andric ScriptInterpreter::eScriptReturnTypeOpaqueObject, &module_pyobj, 2803fe6060f1SDimitry Andric exc_options) && 2804*0b57cec5SDimitry Andric module_pyobj) 280504eeddc0SDimitry Andric *module_sp = std::make_shared<StructuredPythonObject>(PythonObject( 280604eeddc0SDimitry Andric PyRefType::Owned, static_cast<PyObject *>(module_pyobj))); 2807*0b57cec5SDimitry Andric } 2808*0b57cec5SDimitry Andric 2809*0b57cec5SDimitry Andric return true; 2810*0b57cec5SDimitry Andric } 2811*0b57cec5SDimitry Andric 2812*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::IsReservedWord(const char *word) { 2813*0b57cec5SDimitry Andric if (!word || !word[0]) 2814*0b57cec5SDimitry Andric return false; 2815*0b57cec5SDimitry Andric 2816*0b57cec5SDimitry Andric llvm::StringRef word_sr(word); 2817*0b57cec5SDimitry Andric 2818*0b57cec5SDimitry Andric // filter out a few characters that would just confuse us and that are 2819*0b57cec5SDimitry Andric // clearly not keyword material anyway 2820*0b57cec5SDimitry Andric if (word_sr.find('"') != llvm::StringRef::npos || 2821*0b57cec5SDimitry Andric word_sr.find('\'') != llvm::StringRef::npos) 2822*0b57cec5SDimitry Andric return false; 2823*0b57cec5SDimitry Andric 2824*0b57cec5SDimitry Andric StreamString command_stream; 2825*0b57cec5SDimitry Andric command_stream.Printf("keyword.iskeyword('%s')", word); 2826*0b57cec5SDimitry Andric bool result; 2827*0b57cec5SDimitry Andric ExecuteScriptOptions options; 2828*0b57cec5SDimitry Andric options.SetEnableIO(false); 2829*0b57cec5SDimitry Andric options.SetMaskoutErrors(true); 2830*0b57cec5SDimitry Andric options.SetSetLLDBGlobals(false); 2831*0b57cec5SDimitry Andric if (ExecuteOneLineWithReturn(command_stream.GetData(), 2832*0b57cec5SDimitry Andric ScriptInterpreter::eScriptReturnTypeBool, 2833*0b57cec5SDimitry Andric &result, options)) 2834*0b57cec5SDimitry Andric return result; 2835*0b57cec5SDimitry Andric return false; 2836*0b57cec5SDimitry Andric } 2837*0b57cec5SDimitry Andric 2838*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::SynchronicityHandler::SynchronicityHandler( 2839*0b57cec5SDimitry Andric lldb::DebuggerSP debugger_sp, ScriptedCommandSynchronicity synchro) 2840*0b57cec5SDimitry Andric : m_debugger_sp(debugger_sp), m_synch_wanted(synchro), 2841*0b57cec5SDimitry Andric m_old_asynch(debugger_sp->GetAsyncExecution()) { 2842*0b57cec5SDimitry Andric if (m_synch_wanted == eScriptedCommandSynchronicitySynchronous) 2843*0b57cec5SDimitry Andric m_debugger_sp->SetAsyncExecution(false); 2844*0b57cec5SDimitry Andric else if (m_synch_wanted == eScriptedCommandSynchronicityAsynchronous) 2845*0b57cec5SDimitry Andric m_debugger_sp->SetAsyncExecution(true); 2846*0b57cec5SDimitry Andric } 2847*0b57cec5SDimitry Andric 2848*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::SynchronicityHandler::~SynchronicityHandler() { 2849*0b57cec5SDimitry Andric if (m_synch_wanted != eScriptedCommandSynchronicityCurrentValue) 2850*0b57cec5SDimitry Andric m_debugger_sp->SetAsyncExecution(m_old_asynch); 2851*0b57cec5SDimitry Andric } 2852*0b57cec5SDimitry Andric 2853*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptBasedCommand( 2854*0b57cec5SDimitry Andric const char *impl_function, llvm::StringRef args, 2855*0b57cec5SDimitry Andric ScriptedCommandSynchronicity synchronicity, 2856*0b57cec5SDimitry Andric lldb_private::CommandReturnObject &cmd_retobj, Status &error, 2857*0b57cec5SDimitry Andric const lldb_private::ExecutionContext &exe_ctx) { 2858*0b57cec5SDimitry Andric if (!impl_function) { 2859*0b57cec5SDimitry Andric error.SetErrorString("no function to execute"); 2860*0b57cec5SDimitry Andric return false; 2861*0b57cec5SDimitry Andric } 2862*0b57cec5SDimitry Andric 2863*0b57cec5SDimitry Andric lldb::DebuggerSP debugger_sp = m_debugger.shared_from_this(); 2864*0b57cec5SDimitry Andric lldb::ExecutionContextRefSP exe_ctx_ref_sp(new ExecutionContextRef(exe_ctx)); 2865*0b57cec5SDimitry Andric 2866*0b57cec5SDimitry Andric if (!debugger_sp.get()) { 2867*0b57cec5SDimitry Andric error.SetErrorString("invalid Debugger pointer"); 2868*0b57cec5SDimitry Andric return false; 2869*0b57cec5SDimitry Andric } 2870*0b57cec5SDimitry Andric 2871*0b57cec5SDimitry Andric bool ret_val = false; 2872*0b57cec5SDimitry Andric 2873*0b57cec5SDimitry Andric std::string err_msg; 2874*0b57cec5SDimitry Andric 2875*0b57cec5SDimitry Andric { 2876*0b57cec5SDimitry Andric Locker py_lock(this, 2877*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | 2878*0b57cec5SDimitry Andric (cmd_retobj.GetInteractive() ? 0 : Locker::NoSTDIN), 2879*0b57cec5SDimitry Andric Locker::FreeLock | Locker::TearDownSession); 2880*0b57cec5SDimitry Andric 2881*0b57cec5SDimitry Andric SynchronicityHandler synch_handler(debugger_sp, synchronicity); 2882*0b57cec5SDimitry Andric 2883*0b57cec5SDimitry Andric std::string args_str = args.str(); 2884*0b57cec5SDimitry Andric ret_val = LLDBSwigPythonCallCommand( 2885*0b57cec5SDimitry Andric impl_function, m_dictionary_name.c_str(), debugger_sp, args_str.c_str(), 2886*0b57cec5SDimitry Andric cmd_retobj, exe_ctx_ref_sp); 2887*0b57cec5SDimitry Andric } 2888*0b57cec5SDimitry Andric 2889*0b57cec5SDimitry Andric if (!ret_val) 2890*0b57cec5SDimitry Andric error.SetErrorString("unable to execute script function"); 2891*0b57cec5SDimitry Andric else 2892*0b57cec5SDimitry Andric error.Clear(); 2893*0b57cec5SDimitry Andric 2894*0b57cec5SDimitry Andric return ret_val; 2895*0b57cec5SDimitry Andric } 2896*0b57cec5SDimitry Andric 2897*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::RunScriptBasedCommand( 2898*0b57cec5SDimitry Andric StructuredData::GenericSP impl_obj_sp, llvm::StringRef args, 2899*0b57cec5SDimitry Andric ScriptedCommandSynchronicity synchronicity, 2900*0b57cec5SDimitry Andric lldb_private::CommandReturnObject &cmd_retobj, Status &error, 2901*0b57cec5SDimitry Andric const lldb_private::ExecutionContext &exe_ctx) { 2902*0b57cec5SDimitry Andric if (!impl_obj_sp || !impl_obj_sp->IsValid()) { 2903*0b57cec5SDimitry Andric error.SetErrorString("no function to execute"); 2904*0b57cec5SDimitry Andric return false; 2905*0b57cec5SDimitry Andric } 2906*0b57cec5SDimitry Andric 2907*0b57cec5SDimitry Andric lldb::DebuggerSP debugger_sp = m_debugger.shared_from_this(); 2908*0b57cec5SDimitry Andric lldb::ExecutionContextRefSP exe_ctx_ref_sp(new ExecutionContextRef(exe_ctx)); 2909*0b57cec5SDimitry Andric 2910*0b57cec5SDimitry Andric if (!debugger_sp.get()) { 2911*0b57cec5SDimitry Andric error.SetErrorString("invalid Debugger pointer"); 2912*0b57cec5SDimitry Andric return false; 2913*0b57cec5SDimitry Andric } 2914*0b57cec5SDimitry Andric 2915*0b57cec5SDimitry Andric bool ret_val = false; 2916*0b57cec5SDimitry Andric 2917*0b57cec5SDimitry Andric std::string err_msg; 2918*0b57cec5SDimitry Andric 2919*0b57cec5SDimitry Andric { 2920*0b57cec5SDimitry Andric Locker py_lock(this, 2921*0b57cec5SDimitry Andric Locker::AcquireLock | Locker::InitSession | 2922*0b57cec5SDimitry Andric (cmd_retobj.GetInteractive() ? 0 : Locker::NoSTDIN), 2923*0b57cec5SDimitry Andric Locker::FreeLock | Locker::TearDownSession); 2924*0b57cec5SDimitry Andric 2925*0b57cec5SDimitry Andric SynchronicityHandler synch_handler(debugger_sp, synchronicity); 2926*0b57cec5SDimitry Andric 2927*0b57cec5SDimitry Andric std::string args_str = args.str(); 29284824e7fdSDimitry Andric ret_val = LLDBSwigPythonCallCommandObject( 29294824e7fdSDimitry Andric static_cast<PyObject *>(impl_obj_sp->GetValue()), debugger_sp, 29304824e7fdSDimitry Andric args_str.c_str(), cmd_retobj, exe_ctx_ref_sp); 2931*0b57cec5SDimitry Andric } 2932*0b57cec5SDimitry Andric 2933*0b57cec5SDimitry Andric if (!ret_val) 2934*0b57cec5SDimitry Andric error.SetErrorString("unable to execute script function"); 2935*0b57cec5SDimitry Andric else 2936*0b57cec5SDimitry Andric error.Clear(); 2937*0b57cec5SDimitry Andric 2938*0b57cec5SDimitry Andric return ret_val; 2939*0b57cec5SDimitry Andric } 2940*0b57cec5SDimitry Andric 29415ffd83dbSDimitry Andric /// In Python, a special attribute __doc__ contains the docstring for an object 29425ffd83dbSDimitry Andric /// (function, method, class, ...) if any is defined Otherwise, the attribute's 29435ffd83dbSDimitry Andric /// value is None. 2944*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GetDocumentationForItem(const char *item, 2945*0b57cec5SDimitry Andric std::string &dest) { 2946*0b57cec5SDimitry Andric dest.clear(); 29475ffd83dbSDimitry Andric 2948*0b57cec5SDimitry Andric if (!item || !*item) 2949*0b57cec5SDimitry Andric return false; 29505ffd83dbSDimitry Andric 2951*0b57cec5SDimitry Andric std::string command(item); 2952*0b57cec5SDimitry Andric command += ".__doc__"; 2953*0b57cec5SDimitry Andric 29545ffd83dbSDimitry Andric // Python is going to point this to valid data if ExecuteOneLineWithReturn 29555ffd83dbSDimitry Andric // returns successfully. 29565ffd83dbSDimitry Andric char *result_ptr = nullptr; 2957*0b57cec5SDimitry Andric 2958*0b57cec5SDimitry Andric if (ExecuteOneLineWithReturn( 29595ffd83dbSDimitry Andric command, ScriptInterpreter::eScriptReturnTypeCharStrOrNone, 2960*0b57cec5SDimitry Andric &result_ptr, 2961fe6060f1SDimitry Andric ExecuteScriptOptions().SetEnableIO(false))) { 2962*0b57cec5SDimitry Andric if (result_ptr) 2963*0b57cec5SDimitry Andric dest.assign(result_ptr); 2964*0b57cec5SDimitry Andric return true; 2965*0b57cec5SDimitry Andric } 29665ffd83dbSDimitry Andric 29675ffd83dbSDimitry Andric StreamString str_stream; 29685ffd83dbSDimitry Andric str_stream << "Function " << item 29695ffd83dbSDimitry Andric << " was not found. Containing module might be missing."; 29705ffd83dbSDimitry Andric dest = std::string(str_stream.GetString()); 29715ffd83dbSDimitry Andric 29725ffd83dbSDimitry Andric return false; 2973*0b57cec5SDimitry Andric } 2974*0b57cec5SDimitry Andric 2975*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GetShortHelpForCommandObject( 2976*0b57cec5SDimitry Andric StructuredData::GenericSP cmd_obj_sp, std::string &dest) { 2977*0b57cec5SDimitry Andric dest.clear(); 2978*0b57cec5SDimitry Andric 2979*0b57cec5SDimitry Andric Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 2980*0b57cec5SDimitry Andric 2981*0b57cec5SDimitry Andric static char callee_name[] = "get_short_help"; 2982*0b57cec5SDimitry Andric 2983*0b57cec5SDimitry Andric if (!cmd_obj_sp) 2984*0b57cec5SDimitry Andric return false; 2985*0b57cec5SDimitry Andric 2986*0b57cec5SDimitry Andric PythonObject implementor(PyRefType::Borrowed, 2987*0b57cec5SDimitry Andric (PyObject *)cmd_obj_sp->GetValue()); 2988*0b57cec5SDimitry Andric 2989*0b57cec5SDimitry Andric if (!implementor.IsAllocated()) 2990*0b57cec5SDimitry Andric return false; 2991*0b57cec5SDimitry Andric 2992*0b57cec5SDimitry Andric PythonObject pmeth(PyRefType::Owned, 2993*0b57cec5SDimitry Andric PyObject_GetAttrString(implementor.get(), callee_name)); 2994*0b57cec5SDimitry Andric 2995*0b57cec5SDimitry Andric if (PyErr_Occurred()) 2996*0b57cec5SDimitry Andric PyErr_Clear(); 2997*0b57cec5SDimitry Andric 2998*0b57cec5SDimitry Andric if (!pmeth.IsAllocated()) 2999*0b57cec5SDimitry Andric return false; 3000*0b57cec5SDimitry Andric 3001*0b57cec5SDimitry Andric if (PyCallable_Check(pmeth.get()) == 0) { 3002*0b57cec5SDimitry Andric if (PyErr_Occurred()) 3003*0b57cec5SDimitry Andric PyErr_Clear(); 3004*0b57cec5SDimitry Andric return false; 3005*0b57cec5SDimitry Andric } 3006*0b57cec5SDimitry Andric 3007*0b57cec5SDimitry Andric if (PyErr_Occurred()) 3008*0b57cec5SDimitry Andric PyErr_Clear(); 3009*0b57cec5SDimitry Andric 30105ffd83dbSDimitry Andric // Right now we know this function exists and is callable. 3011*0b57cec5SDimitry Andric PythonObject py_return( 3012*0b57cec5SDimitry Andric PyRefType::Owned, 3013*0b57cec5SDimitry Andric PyObject_CallMethod(implementor.get(), callee_name, nullptr)); 3014*0b57cec5SDimitry Andric 30155ffd83dbSDimitry Andric // If it fails, print the error but otherwise go on. 3016*0b57cec5SDimitry Andric if (PyErr_Occurred()) { 3017*0b57cec5SDimitry Andric PyErr_Print(); 3018*0b57cec5SDimitry Andric PyErr_Clear(); 3019*0b57cec5SDimitry Andric } 3020*0b57cec5SDimitry Andric 3021*0b57cec5SDimitry Andric if (py_return.IsAllocated() && PythonString::Check(py_return.get())) { 3022*0b57cec5SDimitry Andric PythonString py_string(PyRefType::Borrowed, py_return.get()); 3023*0b57cec5SDimitry Andric llvm::StringRef return_data(py_string.GetString()); 3024*0b57cec5SDimitry Andric dest.assign(return_data.data(), return_data.size()); 30255ffd83dbSDimitry Andric return true; 3026*0b57cec5SDimitry Andric } 30275ffd83dbSDimitry Andric 30285ffd83dbSDimitry Andric return false; 3029*0b57cec5SDimitry Andric } 3030*0b57cec5SDimitry Andric 3031*0b57cec5SDimitry Andric uint32_t ScriptInterpreterPythonImpl::GetFlagsForCommandObject( 3032*0b57cec5SDimitry Andric StructuredData::GenericSP cmd_obj_sp) { 3033*0b57cec5SDimitry Andric uint32_t result = 0; 3034*0b57cec5SDimitry Andric 3035*0b57cec5SDimitry Andric Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 3036*0b57cec5SDimitry Andric 3037*0b57cec5SDimitry Andric static char callee_name[] = "get_flags"; 3038*0b57cec5SDimitry Andric 3039*0b57cec5SDimitry Andric if (!cmd_obj_sp) 3040*0b57cec5SDimitry Andric return result; 3041*0b57cec5SDimitry Andric 3042*0b57cec5SDimitry Andric PythonObject implementor(PyRefType::Borrowed, 3043*0b57cec5SDimitry Andric (PyObject *)cmd_obj_sp->GetValue()); 3044*0b57cec5SDimitry Andric 3045*0b57cec5SDimitry Andric if (!implementor.IsAllocated()) 3046*0b57cec5SDimitry Andric return result; 3047*0b57cec5SDimitry Andric 3048*0b57cec5SDimitry Andric PythonObject pmeth(PyRefType::Owned, 3049*0b57cec5SDimitry Andric PyObject_GetAttrString(implementor.get(), callee_name)); 3050*0b57cec5SDimitry Andric 3051*0b57cec5SDimitry Andric if (PyErr_Occurred()) 3052*0b57cec5SDimitry Andric PyErr_Clear(); 3053*0b57cec5SDimitry Andric 3054*0b57cec5SDimitry Andric if (!pmeth.IsAllocated()) 3055*0b57cec5SDimitry Andric return result; 3056*0b57cec5SDimitry Andric 3057*0b57cec5SDimitry Andric if (PyCallable_Check(pmeth.get()) == 0) { 3058*0b57cec5SDimitry Andric if (PyErr_Occurred()) 3059*0b57cec5SDimitry Andric PyErr_Clear(); 3060*0b57cec5SDimitry Andric return result; 3061*0b57cec5SDimitry Andric } 3062*0b57cec5SDimitry Andric 3063*0b57cec5SDimitry Andric if (PyErr_Occurred()) 3064*0b57cec5SDimitry Andric PyErr_Clear(); 3065*0b57cec5SDimitry Andric 30665ffd83dbSDimitry Andric long long py_return = unwrapOrSetPythonException( 30675ffd83dbSDimitry Andric As<long long>(implementor.CallMethod(callee_name))); 3068*0b57cec5SDimitry Andric 3069*0b57cec5SDimitry Andric // if it fails, print the error but otherwise go on 3070*0b57cec5SDimitry Andric if (PyErr_Occurred()) { 3071*0b57cec5SDimitry Andric PyErr_Print(); 3072*0b57cec5SDimitry Andric PyErr_Clear(); 30735ffd83dbSDimitry Andric } else { 30745ffd83dbSDimitry Andric result = py_return; 3075*0b57cec5SDimitry Andric } 3076*0b57cec5SDimitry Andric 3077*0b57cec5SDimitry Andric return result; 3078*0b57cec5SDimitry Andric } 3079*0b57cec5SDimitry Andric 3080*0b57cec5SDimitry Andric bool ScriptInterpreterPythonImpl::GetLongHelpForCommandObject( 3081*0b57cec5SDimitry Andric StructuredData::GenericSP cmd_obj_sp, std::string &dest) { 3082*0b57cec5SDimitry Andric bool got_string = false; 3083*0b57cec5SDimitry Andric dest.clear(); 3084*0b57cec5SDimitry Andric 3085*0b57cec5SDimitry Andric Locker py_lock(this, Locker::AcquireLock | Locker::NoSTDIN, Locker::FreeLock); 3086*0b57cec5SDimitry Andric 3087*0b57cec5SDimitry Andric static char callee_name[] = "get_long_help"; 3088*0b57cec5SDimitry Andric 3089*0b57cec5SDimitry Andric if (!cmd_obj_sp) 3090*0b57cec5SDimitry Andric return false; 3091*0b57cec5SDimitry Andric 3092*0b57cec5SDimitry Andric PythonObject implementor(PyRefType::Borrowed, 3093*0b57cec5SDimitry Andric (PyObject *)cmd_obj_sp->GetValue()); 3094*0b57cec5SDimitry Andric 3095*0b57cec5SDimitry Andric if (!implementor.IsAllocated()) 3096*0b57cec5SDimitry Andric return false; 3097*0b57cec5SDimitry Andric 3098*0b57cec5SDimitry Andric PythonObject pmeth(PyRefType::Owned, 3099*0b57cec5SDimitry Andric PyObject_GetAttrString(implementor.get(), callee_name)); 3100*0b57cec5SDimitry Andric 3101*0b57cec5SDimitry Andric if (PyErr_Occurred()) 3102*0b57cec5SDimitry Andric PyErr_Clear(); 3103*0b57cec5SDimitry Andric 3104*0b57cec5SDimitry Andric if (!pmeth.IsAllocated()) 3105*0b57cec5SDimitry Andric return false; 3106*0b57cec5SDimitry Andric 3107*0b57cec5SDimitry Andric if (PyCallable_Check(pmeth.get()) == 0) { 3108*0b57cec5SDimitry Andric if (PyErr_Occurred()) 3109*0b57cec5SDimitry Andric PyErr_Clear(); 3110*0b57cec5SDimitry Andric 3111*0b57cec5SDimitry Andric return false; 3112*0b57cec5SDimitry Andric } 3113*0b57cec5SDimitry Andric 3114*0b57cec5SDimitry Andric if (PyErr_Occurred()) 3115*0b57cec5SDimitry Andric PyErr_Clear(); 3116*0b57cec5SDimitry Andric 3117*0b57cec5SDimitry Andric // right now we know this function exists and is callable.. 3118*0b57cec5SDimitry Andric PythonObject py_return( 3119*0b57cec5SDimitry Andric PyRefType::Owned, 3120*0b57cec5SDimitry Andric PyObject_CallMethod(implementor.get(), callee_name, nullptr)); 3121*0b57cec5SDimitry Andric 3122*0b57cec5SDimitry Andric // if it fails, print the error but otherwise go on 3123*0b57cec5SDimitry Andric if (PyErr_Occurred()) { 3124*0b57cec5SDimitry Andric PyErr_Print(); 3125*0b57cec5SDimitry Andric PyErr_Clear(); 3126*0b57cec5SDimitry Andric } 3127*0b57cec5SDimitry Andric 3128*0b57cec5SDimitry Andric if (py_return.IsAllocated() && PythonString::Check(py_return.get())) { 3129*0b57cec5SDimitry Andric PythonString str(PyRefType::Borrowed, py_return.get()); 3130*0b57cec5SDimitry Andric llvm::StringRef str_data(str.GetString()); 3131*0b57cec5SDimitry Andric dest.assign(str_data.data(), str_data.size()); 3132*0b57cec5SDimitry Andric got_string = true; 3133*0b57cec5SDimitry Andric } 3134*0b57cec5SDimitry Andric 3135*0b57cec5SDimitry Andric return got_string; 3136*0b57cec5SDimitry Andric } 3137*0b57cec5SDimitry Andric 3138*0b57cec5SDimitry Andric std::unique_ptr<ScriptInterpreterLocker> 3139*0b57cec5SDimitry Andric ScriptInterpreterPythonImpl::AcquireInterpreterLock() { 3140*0b57cec5SDimitry Andric std::unique_ptr<ScriptInterpreterLocker> py_lock(new Locker( 3141*0b57cec5SDimitry Andric this, Locker::AcquireLock | Locker::InitSession | Locker::NoSTDIN, 3142*0b57cec5SDimitry Andric Locker::FreeLock | Locker::TearDownSession)); 3143*0b57cec5SDimitry Andric return py_lock; 3144*0b57cec5SDimitry Andric } 3145*0b57cec5SDimitry Andric 314604eeddc0SDimitry Andric void ScriptInterpreterPythonImpl::Initialize() { 3147e8d8bef9SDimitry Andric LLDB_SCOPED_TIMER(); 3148*0b57cec5SDimitry Andric 3149*0b57cec5SDimitry Andric // RAII-based initialization which correctly handles multiple-initialization, 3150*0b57cec5SDimitry Andric // version- specific differences among Python 2 and Python 3, and saving and 3151*0b57cec5SDimitry Andric // restoring various other pieces of state that can get mucked with during 3152*0b57cec5SDimitry Andric // initialization. 3153*0b57cec5SDimitry Andric InitializePythonRAII initialize_guard; 3154*0b57cec5SDimitry Andric 3155*0b57cec5SDimitry Andric LLDBSwigPyInit(); 3156*0b57cec5SDimitry Andric 3157*0b57cec5SDimitry Andric // Update the path python uses to search for modules to include the current 3158*0b57cec5SDimitry Andric // directory. 3159*0b57cec5SDimitry Andric 3160*0b57cec5SDimitry Andric PyRun_SimpleString("import sys"); 3161*0b57cec5SDimitry Andric AddToSysPath(AddLocation::End, "."); 3162*0b57cec5SDimitry Andric 3163*0b57cec5SDimitry Andric // Don't denormalize paths when calling file_spec.GetPath(). On platforms 3164*0b57cec5SDimitry Andric // that use a backslash as the path separator, this will result in executing 3165*0b57cec5SDimitry Andric // python code containing paths with unescaped backslashes. But Python also 3166*0b57cec5SDimitry Andric // accepts forward slashes, so to make life easier we just use that. 3167*0b57cec5SDimitry Andric if (FileSpec file_spec = GetPythonDir()) 3168*0b57cec5SDimitry Andric AddToSysPath(AddLocation::Beginning, file_spec.GetPath(false)); 3169*0b57cec5SDimitry Andric if (FileSpec file_spec = HostInfo::GetShlibDir()) 3170*0b57cec5SDimitry Andric AddToSysPath(AddLocation::Beginning, file_spec.GetPath(false)); 3171*0b57cec5SDimitry Andric 3172*0b57cec5SDimitry Andric PyRun_SimpleString("sys.dont_write_bytecode = 1; import " 3173*0b57cec5SDimitry Andric "lldb.embedded_interpreter; from " 3174*0b57cec5SDimitry Andric "lldb.embedded_interpreter import run_python_interpreter; " 3175*0b57cec5SDimitry Andric "from lldb.embedded_interpreter import run_one_line"); 317604eeddc0SDimitry Andric 317704eeddc0SDimitry Andric #if LLDB_USE_PYTHON_SET_INTERRUPT 317804eeddc0SDimitry Andric // Python will not just overwrite its internal SIGINT handler but also the 317904eeddc0SDimitry Andric // one from the process. Backup the current SIGINT handler to prevent that 318004eeddc0SDimitry Andric // Python deletes it. 318104eeddc0SDimitry Andric RestoreSignalHandlerScope save_sigint(SIGINT); 318204eeddc0SDimitry Andric 318304eeddc0SDimitry Andric // Setup a default SIGINT signal handler that works the same way as the 318404eeddc0SDimitry Andric // normal Python REPL signal handler which raises a KeyboardInterrupt. 318504eeddc0SDimitry Andric // Also make sure to not pollute the user's REPL with the signal module nor 318604eeddc0SDimitry Andric // our utility function. 318704eeddc0SDimitry Andric PyRun_SimpleString("def lldb_setup_sigint_handler():\n" 318804eeddc0SDimitry Andric " import signal;\n" 318904eeddc0SDimitry Andric " def signal_handler(sig, frame):\n" 319004eeddc0SDimitry Andric " raise KeyboardInterrupt()\n" 319104eeddc0SDimitry Andric " signal.signal(signal.SIGINT, signal_handler);\n" 319204eeddc0SDimitry Andric "lldb_setup_sigint_handler();\n" 319304eeddc0SDimitry Andric "del lldb_setup_sigint_handler\n"); 319404eeddc0SDimitry Andric #endif 3195*0b57cec5SDimitry Andric } 3196*0b57cec5SDimitry Andric 3197*0b57cec5SDimitry Andric void ScriptInterpreterPythonImpl::AddToSysPath(AddLocation location, 3198*0b57cec5SDimitry Andric std::string path) { 3199*0b57cec5SDimitry Andric std::string path_copy; 3200*0b57cec5SDimitry Andric 3201*0b57cec5SDimitry Andric std::string statement; 3202*0b57cec5SDimitry Andric if (location == AddLocation::Beginning) { 3203*0b57cec5SDimitry Andric statement.assign("sys.path.insert(0,\""); 3204*0b57cec5SDimitry Andric statement.append(path); 3205*0b57cec5SDimitry Andric statement.append("\")"); 3206*0b57cec5SDimitry Andric } else { 3207*0b57cec5SDimitry Andric statement.assign("sys.path.append(\""); 3208*0b57cec5SDimitry Andric statement.append(path); 3209*0b57cec5SDimitry Andric statement.append("\")"); 3210*0b57cec5SDimitry Andric } 3211*0b57cec5SDimitry Andric PyRun_SimpleString(statement.c_str()); 3212*0b57cec5SDimitry Andric } 3213*0b57cec5SDimitry Andric 3214*0b57cec5SDimitry Andric // We are intentionally NOT calling Py_Finalize here (this would be the logical 3215*0b57cec5SDimitry Andric // place to call it). Calling Py_Finalize here causes test suite runs to seg 3216*0b57cec5SDimitry Andric // fault: The test suite runs in Python. It registers SBDebugger::Terminate to 3217*0b57cec5SDimitry Andric // be called 'at_exit'. When the test suite Python harness finishes up, it 3218*0b57cec5SDimitry Andric // calls Py_Finalize, which calls all the 'at_exit' registered functions. 3219*0b57cec5SDimitry Andric // SBDebugger::Terminate calls Debugger::Terminate, which calls lldb::Terminate, 3220*0b57cec5SDimitry Andric // which calls ScriptInterpreter::Terminate, which calls 3221*0b57cec5SDimitry Andric // ScriptInterpreterPythonImpl::Terminate. So if we call Py_Finalize here, we 3222*0b57cec5SDimitry Andric // end up with Py_Finalize being called from within Py_Finalize, which results 3223*0b57cec5SDimitry Andric // in a seg fault. Since this function only gets called when lldb is shutting 3224*0b57cec5SDimitry Andric // down and going away anyway, the fact that we don't actually call Py_Finalize 3225*0b57cec5SDimitry Andric // should not cause any problems (everything should shut down/go away anyway 3226*0b57cec5SDimitry Andric // when the process exits). 3227*0b57cec5SDimitry Andric // 3228*0b57cec5SDimitry Andric // void ScriptInterpreterPythonImpl::Terminate() { Py_Finalize (); } 3229*0b57cec5SDimitry Andric 3230480093f4SDimitry Andric #endif 3231