1 //===-- ScriptInterpreter.cpp ---------------------------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "lldb/Interpreter/ScriptInterpreter.h" 10 #include "lldb/Core/Debugger.h" 11 #include "lldb/Host/ConnectionFileDescriptor.h" 12 #include "lldb/Host/Pipe.h" 13 #include "lldb/Host/PseudoTerminal.h" 14 #include "lldb/Interpreter/CommandReturnObject.h" 15 #include "lldb/Utility/Status.h" 16 #include "lldb/Utility/Stream.h" 17 #include "lldb/Utility/StringList.h" 18 #if defined(_WIN32) 19 #include "lldb/Host/windows/ConnectionGenericFileWindows.h" 20 #endif 21 #include <cstdio> 22 #include <cstdlib> 23 #include <memory> 24 #include <optional> 25 #include <string> 26 27 using namespace lldb; 28 using namespace lldb_private; 29 30 ScriptInterpreter::ScriptInterpreter( 31 Debugger &debugger, lldb::ScriptLanguage script_lang, 32 lldb::ScriptedProcessInterfaceUP scripted_process_interface_up) 33 : m_debugger(debugger), m_script_lang(script_lang), 34 m_scripted_process_interface_up( 35 std::move(scripted_process_interface_up)) {} 36 37 void ScriptInterpreter::CollectDataForBreakpointCommandCallback( 38 std::vector<std::reference_wrapper<BreakpointOptions>> &bp_options_vec, 39 CommandReturnObject &result) { 40 result.AppendError( 41 "This script interpreter does not support breakpoint callbacks."); 42 } 43 44 void ScriptInterpreter::CollectDataForWatchpointCommandCallback( 45 WatchpointOptions *bp_options, CommandReturnObject &result) { 46 result.AppendError( 47 "This script interpreter does not support watchpoint callbacks."); 48 } 49 50 StructuredData::DictionarySP ScriptInterpreter::GetInterpreterInfo() { 51 return nullptr; 52 } 53 54 bool ScriptInterpreter::LoadScriptingModule(const char *filename, 55 const LoadScriptOptions &options, 56 lldb_private::Status &error, 57 StructuredData::ObjectSP *module_sp, 58 FileSpec extra_search_dir) { 59 error.SetErrorString( 60 "This script interpreter does not support importing modules."); 61 return false; 62 } 63 64 std::string ScriptInterpreter::LanguageToString(lldb::ScriptLanguage language) { 65 switch (language) { 66 case eScriptLanguageNone: 67 return "None"; 68 case eScriptLanguagePython: 69 return "Python"; 70 case eScriptLanguageLua: 71 return "Lua"; 72 case eScriptLanguageUnknown: 73 return "Unknown"; 74 } 75 llvm_unreachable("Unhandled ScriptInterpreter!"); 76 } 77 78 lldb::DataExtractorSP 79 ScriptInterpreter::GetDataExtractorFromSBData(const lldb::SBData &data) const { 80 return data.m_opaque_sp; 81 } 82 83 Status 84 ScriptInterpreter::GetStatusFromSBError(const lldb::SBError &error) const { 85 if (error.m_opaque_up) 86 return *error.m_opaque_up; 87 88 return Status(); 89 } 90 91 std::optional<MemoryRegionInfo> 92 ScriptInterpreter::GetOpaqueTypeFromSBMemoryRegionInfo( 93 const lldb::SBMemoryRegionInfo &mem_region) const { 94 if (!mem_region.m_opaque_up) 95 return std::nullopt; 96 return *mem_region.m_opaque_up.get(); 97 } 98 99 lldb::ScriptLanguage 100 ScriptInterpreter::StringToLanguage(const llvm::StringRef &language) { 101 if (language.equals_insensitive(LanguageToString(eScriptLanguageNone))) 102 return eScriptLanguageNone; 103 if (language.equals_insensitive(LanguageToString(eScriptLanguagePython))) 104 return eScriptLanguagePython; 105 if (language.equals_insensitive(LanguageToString(eScriptLanguageLua))) 106 return eScriptLanguageLua; 107 return eScriptLanguageUnknown; 108 } 109 110 Status ScriptInterpreter::SetBreakpointCommandCallback( 111 std::vector<std::reference_wrapper<BreakpointOptions>> &bp_options_vec, 112 const char *callback_text) { 113 Status error; 114 for (BreakpointOptions &bp_options : bp_options_vec) { 115 error = SetBreakpointCommandCallback(bp_options, callback_text); 116 if (!error.Success()) 117 break; 118 } 119 return error; 120 } 121 122 Status ScriptInterpreter::SetBreakpointCommandCallbackFunction( 123 std::vector<std::reference_wrapper<BreakpointOptions>> &bp_options_vec, 124 const char *function_name, StructuredData::ObjectSP extra_args_sp) { 125 Status error; 126 for (BreakpointOptions &bp_options : bp_options_vec) { 127 error = SetBreakpointCommandCallbackFunction(bp_options, function_name, 128 extra_args_sp); 129 if (!error.Success()) 130 return error; 131 } 132 return error; 133 } 134 135 std::unique_ptr<ScriptInterpreterLocker> 136 ScriptInterpreter::AcquireInterpreterLock() { 137 return std::make_unique<ScriptInterpreterLocker>(); 138 } 139 140 static void ReadThreadBytesReceived(void *baton, const void *src, 141 size_t src_len) { 142 if (src && src_len) { 143 Stream *strm = (Stream *)baton; 144 strm->Write(src, src_len); 145 strm->Flush(); 146 } 147 } 148 149 llvm::Expected<std::unique_ptr<ScriptInterpreterIORedirect>> 150 ScriptInterpreterIORedirect::Create(bool enable_io, Debugger &debugger, 151 CommandReturnObject *result) { 152 if (enable_io) 153 return std::unique_ptr<ScriptInterpreterIORedirect>( 154 new ScriptInterpreterIORedirect(debugger, result)); 155 156 auto nullin = FileSystem::Instance().Open(FileSpec(FileSystem::DEV_NULL), 157 File::eOpenOptionReadOnly); 158 if (!nullin) 159 return nullin.takeError(); 160 161 auto nullout = FileSystem::Instance().Open(FileSpec(FileSystem::DEV_NULL), 162 File::eOpenOptionWriteOnly); 163 if (!nullout) 164 return nullin.takeError(); 165 166 return std::unique_ptr<ScriptInterpreterIORedirect>( 167 new ScriptInterpreterIORedirect(std::move(*nullin), std::move(*nullout))); 168 } 169 170 ScriptInterpreterIORedirect::ScriptInterpreterIORedirect( 171 std::unique_ptr<File> input, std::unique_ptr<File> output) 172 : m_input_file_sp(std::move(input)), 173 m_output_file_sp(std::make_shared<StreamFile>(std::move(output))), 174 m_error_file_sp(m_output_file_sp), 175 m_communication("lldb.ScriptInterpreterIORedirect.comm"), 176 m_disconnect(false) {} 177 178 ScriptInterpreterIORedirect::ScriptInterpreterIORedirect( 179 Debugger &debugger, CommandReturnObject *result) 180 : m_communication("lldb.ScriptInterpreterIORedirect.comm"), 181 m_disconnect(false) { 182 183 if (result) { 184 m_input_file_sp = debugger.GetInputFileSP(); 185 186 Pipe pipe; 187 Status pipe_result = pipe.CreateNew(false); 188 #if defined(_WIN32) 189 lldb::file_t read_file = pipe.GetReadNativeHandle(); 190 pipe.ReleaseReadFileDescriptor(); 191 std::unique_ptr<ConnectionGenericFile> conn_up = 192 std::make_unique<ConnectionGenericFile>(read_file, true); 193 #else 194 std::unique_ptr<ConnectionFileDescriptor> conn_up = 195 std::make_unique<ConnectionFileDescriptor>( 196 pipe.ReleaseReadFileDescriptor(), true); 197 #endif 198 199 if (conn_up->IsConnected()) { 200 m_communication.SetConnection(std::move(conn_up)); 201 m_communication.SetReadThreadBytesReceivedCallback( 202 ReadThreadBytesReceived, &result->GetOutputStream()); 203 m_communication.StartReadThread(); 204 m_disconnect = true; 205 206 FILE *outfile_handle = fdopen(pipe.ReleaseWriteFileDescriptor(), "w"); 207 m_output_file_sp = std::make_shared<StreamFile>(outfile_handle, true); 208 m_error_file_sp = m_output_file_sp; 209 if (outfile_handle) 210 ::setbuf(outfile_handle, nullptr); 211 212 result->SetImmediateOutputFile(debugger.GetOutputStream().GetFileSP()); 213 result->SetImmediateErrorFile(debugger.GetErrorStream().GetFileSP()); 214 } 215 } 216 217 if (!m_input_file_sp || !m_output_file_sp || !m_error_file_sp) 218 debugger.AdoptTopIOHandlerFilesIfInvalid(m_input_file_sp, m_output_file_sp, 219 m_error_file_sp); 220 } 221 222 void ScriptInterpreterIORedirect::Flush() { 223 if (m_output_file_sp) 224 m_output_file_sp->Flush(); 225 if (m_error_file_sp) 226 m_error_file_sp->Flush(); 227 } 228 229 ScriptInterpreterIORedirect::~ScriptInterpreterIORedirect() { 230 if (!m_disconnect) 231 return; 232 233 assert(m_output_file_sp); 234 assert(m_error_file_sp); 235 assert(m_output_file_sp == m_error_file_sp); 236 237 // Close the write end of the pipe since we are done with our one line 238 // script. This should cause the read thread that output_comm is using to 239 // exit. 240 m_output_file_sp->GetFile().Close(); 241 // The close above should cause this thread to exit when it gets to the end 242 // of file, so let it get all its data. 243 m_communication.JoinReadThread(); 244 // Now we can close the read end of the pipe. 245 m_communication.Disconnect(); 246 } 247