1  //===-- GDBRemoteCommunicationServerLLGS.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 <cerrno>
10  
11  #include "lldb/Host/Config.h"
12  
13  #include <chrono>
14  #include <cstring>
15  #include <limits>
16  #include <optional>
17  #include <thread>
18  
19  #include "GDBRemoteCommunicationServerLLGS.h"
20  #include "lldb/Host/ConnectionFileDescriptor.h"
21  #include "lldb/Host/Debug.h"
22  #include "lldb/Host/File.h"
23  #include "lldb/Host/FileAction.h"
24  #include "lldb/Host/FileSystem.h"
25  #include "lldb/Host/Host.h"
26  #include "lldb/Host/HostInfo.h"
27  #include "lldb/Host/PosixApi.h"
28  #include "lldb/Host/Socket.h"
29  #include "lldb/Host/common/NativeProcessProtocol.h"
30  #include "lldb/Host/common/NativeRegisterContext.h"
31  #include "lldb/Host/common/NativeThreadProtocol.h"
32  #include "lldb/Target/MemoryRegionInfo.h"
33  #include "lldb/Utility/Args.h"
34  #include "lldb/Utility/DataBuffer.h"
35  #include "lldb/Utility/Endian.h"
36  #include "lldb/Utility/GDBRemote.h"
37  #include "lldb/Utility/LLDBAssert.h"
38  #include "lldb/Utility/LLDBLog.h"
39  #include "lldb/Utility/Log.h"
40  #include "lldb/Utility/State.h"
41  #include "lldb/Utility/StreamString.h"
42  #include "lldb/Utility/UnimplementedError.h"
43  #include "lldb/Utility/UriParser.h"
44  #include "llvm/Support/JSON.h"
45  #include "llvm/Support/ScopedPrinter.h"
46  #include "llvm/TargetParser/Triple.h"
47  
48  #include "ProcessGDBRemote.h"
49  #include "ProcessGDBRemoteLog.h"
50  #include "lldb/Utility/StringExtractorGDBRemote.h"
51  
52  using namespace lldb;
53  using namespace lldb_private;
54  using namespace lldb_private::process_gdb_remote;
55  using namespace llvm;
56  
57  // GDBRemote Errors
58  
59  namespace {
60  enum GDBRemoteServerError {
61    // Set to the first unused error number in literal form below
62    eErrorFirst = 29,
63    eErrorNoProcess = eErrorFirst,
64    eErrorResume,
65    eErrorExitStatus
66  };
67  }
68  
69  // GDBRemoteCommunicationServerLLGS constructor
GDBRemoteCommunicationServerLLGS(MainLoop & mainloop,NativeProcessProtocol::Manager & process_manager)70  GDBRemoteCommunicationServerLLGS::GDBRemoteCommunicationServerLLGS(
71      MainLoop &mainloop, NativeProcessProtocol::Manager &process_manager)
72      : GDBRemoteCommunicationServerCommon(), m_mainloop(mainloop),
73        m_process_manager(process_manager), m_current_process(nullptr),
74        m_continue_process(nullptr), m_stdio_communication() {
75    RegisterPacketHandlers();
76  }
77  
RegisterPacketHandlers()78  void GDBRemoteCommunicationServerLLGS::RegisterPacketHandlers() {
79    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_C,
80                                  &GDBRemoteCommunicationServerLLGS::Handle_C);
81    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_c,
82                                  &GDBRemoteCommunicationServerLLGS::Handle_c);
83    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_D,
84                                  &GDBRemoteCommunicationServerLLGS::Handle_D);
85    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_H,
86                                  &GDBRemoteCommunicationServerLLGS::Handle_H);
87    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_I,
88                                  &GDBRemoteCommunicationServerLLGS::Handle_I);
89    RegisterMemberFunctionHandler(
90        StringExtractorGDBRemote::eServerPacketType_interrupt,
91        &GDBRemoteCommunicationServerLLGS::Handle_interrupt);
92    RegisterMemberFunctionHandler(
93        StringExtractorGDBRemote::eServerPacketType_m,
94        &GDBRemoteCommunicationServerLLGS::Handle_memory_read);
95    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_M,
96                                  &GDBRemoteCommunicationServerLLGS::Handle_M);
97    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType__M,
98                                  &GDBRemoteCommunicationServerLLGS::Handle__M);
99    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType__m,
100                                  &GDBRemoteCommunicationServerLLGS::Handle__m);
101    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_p,
102                                  &GDBRemoteCommunicationServerLLGS::Handle_p);
103    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_P,
104                                  &GDBRemoteCommunicationServerLLGS::Handle_P);
105    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_qC,
106                                  &GDBRemoteCommunicationServerLLGS::Handle_qC);
107    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_T,
108                                  &GDBRemoteCommunicationServerLLGS::Handle_T);
109    RegisterMemberFunctionHandler(
110        StringExtractorGDBRemote::eServerPacketType_qfThreadInfo,
111        &GDBRemoteCommunicationServerLLGS::Handle_qfThreadInfo);
112    RegisterMemberFunctionHandler(
113        StringExtractorGDBRemote::eServerPacketType_qFileLoadAddress,
114        &GDBRemoteCommunicationServerLLGS::Handle_qFileLoadAddress);
115    RegisterMemberFunctionHandler(
116        StringExtractorGDBRemote::eServerPacketType_qGetWorkingDir,
117        &GDBRemoteCommunicationServerLLGS::Handle_qGetWorkingDir);
118    RegisterMemberFunctionHandler(
119        StringExtractorGDBRemote::eServerPacketType_QThreadSuffixSupported,
120        &GDBRemoteCommunicationServerLLGS::Handle_QThreadSuffixSupported);
121    RegisterMemberFunctionHandler(
122        StringExtractorGDBRemote::eServerPacketType_QListThreadsInStopReply,
123        &GDBRemoteCommunicationServerLLGS::Handle_QListThreadsInStopReply);
124    RegisterMemberFunctionHandler(
125        StringExtractorGDBRemote::eServerPacketType_qMemoryRegionInfo,
126        &GDBRemoteCommunicationServerLLGS::Handle_qMemoryRegionInfo);
127    RegisterMemberFunctionHandler(
128        StringExtractorGDBRemote::eServerPacketType_qMemoryRegionInfoSupported,
129        &GDBRemoteCommunicationServerLLGS::Handle_qMemoryRegionInfoSupported);
130    RegisterMemberFunctionHandler(
131        StringExtractorGDBRemote::eServerPacketType_qProcessInfo,
132        &GDBRemoteCommunicationServerLLGS::Handle_qProcessInfo);
133    RegisterMemberFunctionHandler(
134        StringExtractorGDBRemote::eServerPacketType_qRegisterInfo,
135        &GDBRemoteCommunicationServerLLGS::Handle_qRegisterInfo);
136    RegisterMemberFunctionHandler(
137        StringExtractorGDBRemote::eServerPacketType_QRestoreRegisterState,
138        &GDBRemoteCommunicationServerLLGS::Handle_QRestoreRegisterState);
139    RegisterMemberFunctionHandler(
140        StringExtractorGDBRemote::eServerPacketType_QSaveRegisterState,
141        &GDBRemoteCommunicationServerLLGS::Handle_QSaveRegisterState);
142    RegisterMemberFunctionHandler(
143        StringExtractorGDBRemote::eServerPacketType_QSetDisableASLR,
144        &GDBRemoteCommunicationServerLLGS::Handle_QSetDisableASLR);
145    RegisterMemberFunctionHandler(
146        StringExtractorGDBRemote::eServerPacketType_QSetWorkingDir,
147        &GDBRemoteCommunicationServerLLGS::Handle_QSetWorkingDir);
148    RegisterMemberFunctionHandler(
149        StringExtractorGDBRemote::eServerPacketType_qsThreadInfo,
150        &GDBRemoteCommunicationServerLLGS::Handle_qsThreadInfo);
151    RegisterMemberFunctionHandler(
152        StringExtractorGDBRemote::eServerPacketType_qThreadStopInfo,
153        &GDBRemoteCommunicationServerLLGS::Handle_qThreadStopInfo);
154    RegisterMemberFunctionHandler(
155        StringExtractorGDBRemote::eServerPacketType_jThreadsInfo,
156        &GDBRemoteCommunicationServerLLGS::Handle_jThreadsInfo);
157    RegisterMemberFunctionHandler(
158        StringExtractorGDBRemote::eServerPacketType_qWatchpointSupportInfo,
159        &GDBRemoteCommunicationServerLLGS::Handle_qWatchpointSupportInfo);
160    RegisterMemberFunctionHandler(
161        StringExtractorGDBRemote::eServerPacketType_qXfer,
162        &GDBRemoteCommunicationServerLLGS::Handle_qXfer);
163    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_s,
164                                  &GDBRemoteCommunicationServerLLGS::Handle_s);
165    RegisterMemberFunctionHandler(
166        StringExtractorGDBRemote::eServerPacketType_stop_reason,
167        &GDBRemoteCommunicationServerLLGS::Handle_stop_reason); // ?
168    RegisterMemberFunctionHandler(
169        StringExtractorGDBRemote::eServerPacketType_vAttach,
170        &GDBRemoteCommunicationServerLLGS::Handle_vAttach);
171    RegisterMemberFunctionHandler(
172        StringExtractorGDBRemote::eServerPacketType_vAttachWait,
173        &GDBRemoteCommunicationServerLLGS::Handle_vAttachWait);
174    RegisterMemberFunctionHandler(
175        StringExtractorGDBRemote::eServerPacketType_qVAttachOrWaitSupported,
176        &GDBRemoteCommunicationServerLLGS::Handle_qVAttachOrWaitSupported);
177    RegisterMemberFunctionHandler(
178        StringExtractorGDBRemote::eServerPacketType_vAttachOrWait,
179        &GDBRemoteCommunicationServerLLGS::Handle_vAttachOrWait);
180    RegisterMemberFunctionHandler(
181        StringExtractorGDBRemote::eServerPacketType_vCont,
182        &GDBRemoteCommunicationServerLLGS::Handle_vCont);
183    RegisterMemberFunctionHandler(
184        StringExtractorGDBRemote::eServerPacketType_vCont_actions,
185        &GDBRemoteCommunicationServerLLGS::Handle_vCont_actions);
186    RegisterMemberFunctionHandler(
187        StringExtractorGDBRemote::eServerPacketType_vRun,
188        &GDBRemoteCommunicationServerLLGS::Handle_vRun);
189    RegisterMemberFunctionHandler(
190        StringExtractorGDBRemote::eServerPacketType_x,
191        &GDBRemoteCommunicationServerLLGS::Handle_memory_read);
192    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_Z,
193                                  &GDBRemoteCommunicationServerLLGS::Handle_Z);
194    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_z,
195                                  &GDBRemoteCommunicationServerLLGS::Handle_z);
196    RegisterMemberFunctionHandler(
197        StringExtractorGDBRemote::eServerPacketType_QPassSignals,
198        &GDBRemoteCommunicationServerLLGS::Handle_QPassSignals);
199  
200    RegisterMemberFunctionHandler(
201        StringExtractorGDBRemote::eServerPacketType_jLLDBTraceSupported,
202        &GDBRemoteCommunicationServerLLGS::Handle_jLLDBTraceSupported);
203    RegisterMemberFunctionHandler(
204        StringExtractorGDBRemote::eServerPacketType_jLLDBTraceStart,
205        &GDBRemoteCommunicationServerLLGS::Handle_jLLDBTraceStart);
206    RegisterMemberFunctionHandler(
207        StringExtractorGDBRemote::eServerPacketType_jLLDBTraceStop,
208        &GDBRemoteCommunicationServerLLGS::Handle_jLLDBTraceStop);
209    RegisterMemberFunctionHandler(
210        StringExtractorGDBRemote::eServerPacketType_jLLDBTraceGetState,
211        &GDBRemoteCommunicationServerLLGS::Handle_jLLDBTraceGetState);
212    RegisterMemberFunctionHandler(
213        StringExtractorGDBRemote::eServerPacketType_jLLDBTraceGetBinaryData,
214        &GDBRemoteCommunicationServerLLGS::Handle_jLLDBTraceGetBinaryData);
215  
216    RegisterMemberFunctionHandler(StringExtractorGDBRemote::eServerPacketType_g,
217                                  &GDBRemoteCommunicationServerLLGS::Handle_g);
218  
219    RegisterMemberFunctionHandler(
220        StringExtractorGDBRemote::eServerPacketType_qMemTags,
221        &GDBRemoteCommunicationServerLLGS::Handle_qMemTags);
222  
223    RegisterMemberFunctionHandler(
224        StringExtractorGDBRemote::eServerPacketType_QMemTags,
225        &GDBRemoteCommunicationServerLLGS::Handle_QMemTags);
226  
227    RegisterPacketHandler(StringExtractorGDBRemote::eServerPacketType_k,
228                          [this](StringExtractorGDBRemote packet, Status &error,
229                                 bool &interrupt, bool &quit) {
230                            quit = true;
231                            return this->Handle_k(packet);
232                          });
233  
234    RegisterMemberFunctionHandler(
235        StringExtractorGDBRemote::eServerPacketType_vKill,
236        &GDBRemoteCommunicationServerLLGS::Handle_vKill);
237  
238    RegisterMemberFunctionHandler(
239        StringExtractorGDBRemote::eServerPacketType_qLLDBSaveCore,
240        &GDBRemoteCommunicationServerLLGS::Handle_qSaveCore);
241  
242    RegisterMemberFunctionHandler(
243        StringExtractorGDBRemote::eServerPacketType_QNonStop,
244        &GDBRemoteCommunicationServerLLGS::Handle_QNonStop);
245    RegisterMemberFunctionHandler(
246        StringExtractorGDBRemote::eServerPacketType_vStdio,
247        &GDBRemoteCommunicationServerLLGS::Handle_vStdio);
248    RegisterMemberFunctionHandler(
249        StringExtractorGDBRemote::eServerPacketType_vStopped,
250        &GDBRemoteCommunicationServerLLGS::Handle_vStopped);
251    RegisterMemberFunctionHandler(
252        StringExtractorGDBRemote::eServerPacketType_vCtrlC,
253        &GDBRemoteCommunicationServerLLGS::Handle_vCtrlC);
254  }
255  
SetLaunchInfo(const ProcessLaunchInfo & info)256  void GDBRemoteCommunicationServerLLGS::SetLaunchInfo(const ProcessLaunchInfo &info) {
257    m_process_launch_info = info;
258  }
259  
LaunchProcess()260  Status GDBRemoteCommunicationServerLLGS::LaunchProcess() {
261    Log *log = GetLog(LLDBLog::Process);
262  
263    if (!m_process_launch_info.GetArguments().GetArgumentCount())
264      return Status("%s: no process command line specified to launch",
265                    __FUNCTION__);
266  
267    const bool should_forward_stdio =
268        m_process_launch_info.GetFileActionForFD(STDIN_FILENO) == nullptr ||
269        m_process_launch_info.GetFileActionForFD(STDOUT_FILENO) == nullptr ||
270        m_process_launch_info.GetFileActionForFD(STDERR_FILENO) == nullptr;
271    m_process_launch_info.SetLaunchInSeparateProcessGroup(true);
272    m_process_launch_info.GetFlags().Set(eLaunchFlagDebug);
273  
274    if (should_forward_stdio) {
275      // Temporarily relax the following for Windows until we can take advantage
276      // of the recently added pty support. This doesn't really affect the use of
277      // lldb-server on Windows.
278  #if !defined(_WIN32)
279      if (llvm::Error Err = m_process_launch_info.SetUpPtyRedirection())
280        return Status(std::move(Err));
281  #endif
282    }
283  
284    {
285      std::lock_guard<std::recursive_mutex> guard(m_debugged_process_mutex);
286      assert(m_debugged_processes.empty() && "lldb-server creating debugged "
287                                             "process but one already exists");
288      auto process_or = m_process_manager.Launch(m_process_launch_info, *this);
289      if (!process_or)
290        return Status(process_or.takeError());
291      m_continue_process = m_current_process = process_or->get();
292      m_debugged_processes.emplace(
293          m_current_process->GetID(),
294          DebuggedProcess{std::move(*process_or), DebuggedProcess::Flag{}});
295    }
296  
297    SetEnabledExtensions(*m_current_process);
298  
299    // Handle mirroring of inferior stdout/stderr over the gdb-remote protocol as
300    // needed. llgs local-process debugging may specify PTY paths, which will
301    // make these file actions non-null process launch -i/e/o will also make
302    // these file actions non-null nullptr means that the traffic is expected to
303    // flow over gdb-remote protocol
304    if (should_forward_stdio) {
305      // nullptr means it's not redirected to file or pty (in case of LLGS local)
306      // at least one of stdio will be transferred pty<->gdb-remote we need to
307      // give the pty primary handle to this object to read and/or write
308      LLDB_LOG(log,
309               "pid = {0}: setting up stdout/stderr redirection via $O "
310               "gdb-remote commands",
311               m_current_process->GetID());
312  
313      // Setup stdout/stderr mapping from inferior to $O
314      auto terminal_fd = m_current_process->GetTerminalFileDescriptor();
315      if (terminal_fd >= 0) {
316        LLDB_LOGF(log,
317                  "ProcessGDBRemoteCommunicationServerLLGS::%s setting "
318                  "inferior STDIO fd to %d",
319                  __FUNCTION__, terminal_fd);
320        Status status = SetSTDIOFileDescriptor(terminal_fd);
321        if (status.Fail())
322          return status;
323      } else {
324        LLDB_LOGF(log,
325                  "ProcessGDBRemoteCommunicationServerLLGS::%s ignoring "
326                  "inferior STDIO since terminal fd reported as %d",
327                  __FUNCTION__, terminal_fd);
328      }
329    } else {
330      LLDB_LOG(log,
331               "pid = {0} skipping stdout/stderr redirection via $O: inferior "
332               "will communicate over client-provided file descriptors",
333               m_current_process->GetID());
334    }
335  
336    printf("Launched '%s' as process %" PRIu64 "...\n",
337           m_process_launch_info.GetArguments().GetArgumentAtIndex(0),
338           m_current_process->GetID());
339  
340    return Status();
341  }
342  
AttachToProcess(lldb::pid_t pid)343  Status GDBRemoteCommunicationServerLLGS::AttachToProcess(lldb::pid_t pid) {
344    Log *log = GetLog(LLDBLog::Process);
345    LLDB_LOGF(log, "GDBRemoteCommunicationServerLLGS::%s pid %" PRIu64,
346              __FUNCTION__, pid);
347  
348    // Before we try to attach, make sure we aren't already monitoring something
349    // else.
350    if (!m_debugged_processes.empty())
351      return Status("cannot attach to process %" PRIu64
352                    " when another process with pid %" PRIu64
353                    " is being debugged.",
354                    pid, m_current_process->GetID());
355  
356    // Try to attach.
357    auto process_or = m_process_manager.Attach(pid, *this);
358    if (!process_or) {
359      Status status(process_or.takeError());
360      llvm::errs() << llvm::formatv("failed to attach to process {0}: {1}\n", pid,
361                                    status);
362      return status;
363    }
364    m_continue_process = m_current_process = process_or->get();
365    m_debugged_processes.emplace(
366        m_current_process->GetID(),
367        DebuggedProcess{std::move(*process_or), DebuggedProcess::Flag{}});
368    SetEnabledExtensions(*m_current_process);
369  
370    // Setup stdout/stderr mapping from inferior.
371    auto terminal_fd = m_current_process->GetTerminalFileDescriptor();
372    if (terminal_fd >= 0) {
373      LLDB_LOGF(log,
374                "ProcessGDBRemoteCommunicationServerLLGS::%s setting "
375                "inferior STDIO fd to %d",
376                __FUNCTION__, terminal_fd);
377      Status status = SetSTDIOFileDescriptor(terminal_fd);
378      if (status.Fail())
379        return status;
380    } else {
381      LLDB_LOGF(log,
382                "ProcessGDBRemoteCommunicationServerLLGS::%s ignoring "
383                "inferior STDIO since terminal fd reported as %d",
384                __FUNCTION__, terminal_fd);
385    }
386  
387    printf("Attached to process %" PRIu64 "...\n", pid);
388    return Status();
389  }
390  
AttachWaitProcess(llvm::StringRef process_name,bool include_existing)391  Status GDBRemoteCommunicationServerLLGS::AttachWaitProcess(
392      llvm::StringRef process_name, bool include_existing) {
393    Log *log = GetLog(LLDBLog::Process);
394  
395    std::chrono::milliseconds polling_interval = std::chrono::milliseconds(1);
396  
397    // Create the matcher used to search the process list.
398    ProcessInstanceInfoList exclusion_list;
399    ProcessInstanceInfoMatch match_info;
400    match_info.GetProcessInfo().GetExecutableFile().SetFile(
401        process_name, llvm::sys::path::Style::native);
402    match_info.SetNameMatchType(NameMatch::Equals);
403  
404    if (include_existing) {
405      LLDB_LOG(log, "including existing processes in search");
406    } else {
407      // Create the excluded process list before polling begins.
408      Host::FindProcesses(match_info, exclusion_list);
409      LLDB_LOG(log, "placed '{0}' processes in the exclusion list.",
410               exclusion_list.size());
411    }
412  
413    LLDB_LOG(log, "waiting for '{0}' to appear", process_name);
414  
415    auto is_in_exclusion_list =
416        [&exclusion_list](const ProcessInstanceInfo &info) {
417          for (auto &excluded : exclusion_list) {
418            if (excluded.GetProcessID() == info.GetProcessID())
419              return true;
420          }
421          return false;
422        };
423  
424    ProcessInstanceInfoList loop_process_list;
425    while (true) {
426      loop_process_list.clear();
427      if (Host::FindProcesses(match_info, loop_process_list)) {
428        // Remove all the elements that are in the exclusion list.
429        llvm::erase_if(loop_process_list, is_in_exclusion_list);
430  
431        // One match! We found the desired process.
432        if (loop_process_list.size() == 1) {
433          auto matching_process_pid = loop_process_list[0].GetProcessID();
434          LLDB_LOG(log, "found pid {0}", matching_process_pid);
435          return AttachToProcess(matching_process_pid);
436        }
437  
438        // Multiple matches! Return an error reporting the PIDs we found.
439        if (loop_process_list.size() > 1) {
440          StreamString error_stream;
441          error_stream.Format(
442              "Multiple executables with name: '{0}' found. Pids: ",
443              process_name);
444          for (size_t i = 0; i < loop_process_list.size() - 1; ++i) {
445            error_stream.Format("{0}, ", loop_process_list[i].GetProcessID());
446          }
447          error_stream.Format("{0}.", loop_process_list.back().GetProcessID());
448  
449          Status error;
450          error.SetErrorString(error_stream.GetString());
451          return error;
452        }
453      }
454      // No matches, we have not found the process. Sleep until next poll.
455      LLDB_LOG(log, "sleep {0} seconds", polling_interval);
456      std::this_thread::sleep_for(polling_interval);
457    }
458  }
459  
InitializeDelegate(NativeProcessProtocol * process)460  void GDBRemoteCommunicationServerLLGS::InitializeDelegate(
461      NativeProcessProtocol *process) {
462    assert(process && "process cannot be NULL");
463    Log *log = GetLog(LLDBLog::Process);
464    if (log) {
465      LLDB_LOGF(log,
466                "GDBRemoteCommunicationServerLLGS::%s called with "
467                "NativeProcessProtocol pid %" PRIu64 ", current state: %s",
468                __FUNCTION__, process->GetID(),
469                StateAsCString(process->GetState()));
470    }
471  }
472  
473  GDBRemoteCommunication::PacketResult
SendWResponse(NativeProcessProtocol * process)474  GDBRemoteCommunicationServerLLGS::SendWResponse(
475      NativeProcessProtocol *process) {
476    assert(process && "process cannot be NULL");
477    Log *log = GetLog(LLDBLog::Process);
478  
479    // send W notification
480    auto wait_status = process->GetExitStatus();
481    if (!wait_status) {
482      LLDB_LOG(log, "pid = {0}, failed to retrieve process exit status",
483               process->GetID());
484  
485      StreamGDBRemote response;
486      response.PutChar('E');
487      response.PutHex8(GDBRemoteServerError::eErrorExitStatus);
488      return SendPacketNoLock(response.GetString());
489    }
490  
491    LLDB_LOG(log, "pid = {0}, returning exit type {1}", process->GetID(),
492             *wait_status);
493  
494    // If the process was killed through vKill, return "OK".
495    if (bool(m_debugged_processes.at(process->GetID()).flags &
496             DebuggedProcess::Flag::vkilled))
497      return SendOKResponse();
498  
499    StreamGDBRemote response;
500    response.Format("{0:g}", *wait_status);
501    if (bool(m_extensions_supported &
502             NativeProcessProtocol::Extension::multiprocess))
503      response.Format(";process:{0:x-}", process->GetID());
504    if (m_non_stop)
505      return SendNotificationPacketNoLock("Stop", m_stop_notification_queue,
506                                          response.GetString());
507    return SendPacketNoLock(response.GetString());
508  }
509  
AppendHexValue(StreamString & response,const uint8_t * buf,uint32_t buf_size,bool swap)510  static void AppendHexValue(StreamString &response, const uint8_t *buf,
511                             uint32_t buf_size, bool swap) {
512    int64_t i;
513    if (swap) {
514      for (i = buf_size - 1; i >= 0; i--)
515        response.PutHex8(buf[i]);
516    } else {
517      for (i = 0; i < buf_size; i++)
518        response.PutHex8(buf[i]);
519    }
520  }
521  
GetEncodingNameOrEmpty(const RegisterInfo & reg_info)522  static llvm::StringRef GetEncodingNameOrEmpty(const RegisterInfo &reg_info) {
523    switch (reg_info.encoding) {
524    case eEncodingUint:
525      return "uint";
526    case eEncodingSint:
527      return "sint";
528    case eEncodingIEEE754:
529      return "ieee754";
530    case eEncodingVector:
531      return "vector";
532    default:
533      return "";
534    }
535  }
536  
GetFormatNameOrEmpty(const RegisterInfo & reg_info)537  static llvm::StringRef GetFormatNameOrEmpty(const RegisterInfo &reg_info) {
538    switch (reg_info.format) {
539    case eFormatBinary:
540      return "binary";
541    case eFormatDecimal:
542      return "decimal";
543    case eFormatHex:
544      return "hex";
545    case eFormatFloat:
546      return "float";
547    case eFormatVectorOfSInt8:
548      return "vector-sint8";
549    case eFormatVectorOfUInt8:
550      return "vector-uint8";
551    case eFormatVectorOfSInt16:
552      return "vector-sint16";
553    case eFormatVectorOfUInt16:
554      return "vector-uint16";
555    case eFormatVectorOfSInt32:
556      return "vector-sint32";
557    case eFormatVectorOfUInt32:
558      return "vector-uint32";
559    case eFormatVectorOfFloat32:
560      return "vector-float32";
561    case eFormatVectorOfUInt64:
562      return "vector-uint64";
563    case eFormatVectorOfUInt128:
564      return "vector-uint128";
565    default:
566      return "";
567    };
568  }
569  
GetKindGenericOrEmpty(const RegisterInfo & reg_info)570  static llvm::StringRef GetKindGenericOrEmpty(const RegisterInfo &reg_info) {
571    switch (reg_info.kinds[RegisterKind::eRegisterKindGeneric]) {
572    case LLDB_REGNUM_GENERIC_PC:
573      return "pc";
574    case LLDB_REGNUM_GENERIC_SP:
575      return "sp";
576    case LLDB_REGNUM_GENERIC_FP:
577      return "fp";
578    case LLDB_REGNUM_GENERIC_RA:
579      return "ra";
580    case LLDB_REGNUM_GENERIC_FLAGS:
581      return "flags";
582    case LLDB_REGNUM_GENERIC_ARG1:
583      return "arg1";
584    case LLDB_REGNUM_GENERIC_ARG2:
585      return "arg2";
586    case LLDB_REGNUM_GENERIC_ARG3:
587      return "arg3";
588    case LLDB_REGNUM_GENERIC_ARG4:
589      return "arg4";
590    case LLDB_REGNUM_GENERIC_ARG5:
591      return "arg5";
592    case LLDB_REGNUM_GENERIC_ARG6:
593      return "arg6";
594    case LLDB_REGNUM_GENERIC_ARG7:
595      return "arg7";
596    case LLDB_REGNUM_GENERIC_ARG8:
597      return "arg8";
598    case LLDB_REGNUM_GENERIC_TP:
599      return "tp";
600    default:
601      return "";
602    }
603  }
604  
CollectRegNums(const uint32_t * reg_num,StreamString & response,bool usehex)605  static void CollectRegNums(const uint32_t *reg_num, StreamString &response,
606                             bool usehex) {
607    for (int i = 0; *reg_num != LLDB_INVALID_REGNUM; ++reg_num, ++i) {
608      if (i > 0)
609        response.PutChar(',');
610      if (usehex)
611        response.Printf("%" PRIx32, *reg_num);
612      else
613        response.Printf("%" PRIu32, *reg_num);
614    }
615  }
616  
WriteRegisterValueInHexFixedWidth(StreamString & response,NativeRegisterContext & reg_ctx,const RegisterInfo & reg_info,const RegisterValue * reg_value_p,lldb::ByteOrder byte_order)617  static void WriteRegisterValueInHexFixedWidth(
618      StreamString &response, NativeRegisterContext &reg_ctx,
619      const RegisterInfo &reg_info, const RegisterValue *reg_value_p,
620      lldb::ByteOrder byte_order) {
621    RegisterValue reg_value;
622    if (!reg_value_p) {
623      Status error = reg_ctx.ReadRegister(&reg_info, reg_value);
624      if (error.Success())
625        reg_value_p = &reg_value;
626      // else log.
627    }
628  
629    if (reg_value_p) {
630      AppendHexValue(response, (const uint8_t *)reg_value_p->GetBytes(),
631                     reg_value_p->GetByteSize(),
632                     byte_order == lldb::eByteOrderLittle);
633    } else {
634      // Zero-out any unreadable values.
635      if (reg_info.byte_size > 0) {
636        std::vector<uint8_t> zeros(reg_info.byte_size, '\0');
637        AppendHexValue(response, zeros.data(), zeros.size(), false);
638      }
639    }
640  }
641  
642  static std::optional<json::Object>
GetRegistersAsJSON(NativeThreadProtocol & thread)643  GetRegistersAsJSON(NativeThreadProtocol &thread) {
644    Log *log = GetLog(LLDBLog::Thread);
645  
646    NativeRegisterContext& reg_ctx = thread.GetRegisterContext();
647  
648    json::Object register_object;
649  
650  #ifdef LLDB_JTHREADSINFO_FULL_REGISTER_SET
651    const auto expedited_regs =
652        reg_ctx.GetExpeditedRegisters(ExpeditedRegs::Full);
653  #else
654    const auto expedited_regs =
655        reg_ctx.GetExpeditedRegisters(ExpeditedRegs::Minimal);
656  #endif
657    if (expedited_regs.empty())
658      return std::nullopt;
659  
660    for (auto &reg_num : expedited_regs) {
661      const RegisterInfo *const reg_info_p =
662          reg_ctx.GetRegisterInfoAtIndex(reg_num);
663      if (reg_info_p == nullptr) {
664        LLDB_LOGF(log,
665                  "%s failed to get register info for register index %" PRIu32,
666                  __FUNCTION__, reg_num);
667        continue;
668      }
669  
670      if (reg_info_p->value_regs != nullptr)
671        continue; // Only expedite registers that are not contained in other
672                  // registers.
673  
674      RegisterValue reg_value;
675      Status error = reg_ctx.ReadRegister(reg_info_p, reg_value);
676      if (error.Fail()) {
677        LLDB_LOGF(log, "%s failed to read register '%s' index %" PRIu32 ": %s",
678                  __FUNCTION__,
679                  reg_info_p->name ? reg_info_p->name : "<unnamed-register>",
680                  reg_num, error.AsCString());
681        continue;
682      }
683  
684      StreamString stream;
685      WriteRegisterValueInHexFixedWidth(stream, reg_ctx, *reg_info_p,
686                                        &reg_value, lldb::eByteOrderBig);
687  
688      register_object.try_emplace(llvm::to_string(reg_num),
689                                  stream.GetString().str());
690    }
691  
692    return register_object;
693  }
694  
GetStopReasonString(StopReason stop_reason)695  static const char *GetStopReasonString(StopReason stop_reason) {
696    switch (stop_reason) {
697    case eStopReasonTrace:
698      return "trace";
699    case eStopReasonBreakpoint:
700      return "breakpoint";
701    case eStopReasonWatchpoint:
702      return "watchpoint";
703    case eStopReasonSignal:
704      return "signal";
705    case eStopReasonException:
706      return "exception";
707    case eStopReasonExec:
708      return "exec";
709    case eStopReasonProcessorTrace:
710      return "processor trace";
711    case eStopReasonFork:
712      return "fork";
713    case eStopReasonVFork:
714      return "vfork";
715    case eStopReasonVForkDone:
716      return "vforkdone";
717    case eStopReasonInstrumentation:
718    case eStopReasonInvalid:
719    case eStopReasonPlanComplete:
720    case eStopReasonThreadExiting:
721    case eStopReasonNone:
722      break; // ignored
723    }
724    return nullptr;
725  }
726  
727  static llvm::Expected<json::Array>
GetJSONThreadsInfo(NativeProcessProtocol & process,bool abridged)728  GetJSONThreadsInfo(NativeProcessProtocol &process, bool abridged) {
729    Log *log = GetLog(LLDBLog::Process | LLDBLog::Thread);
730  
731    json::Array threads_array;
732  
733    // Ensure we can get info on the given thread.
734    for (NativeThreadProtocol &thread : process.Threads()) {
735      lldb::tid_t tid = thread.GetID();
736      // Grab the reason this thread stopped.
737      struct ThreadStopInfo tid_stop_info;
738      std::string description;
739      if (!thread.GetStopReason(tid_stop_info, description))
740        return llvm::make_error<llvm::StringError>(
741            "failed to get stop reason", llvm::inconvertibleErrorCode());
742  
743      const int signum = tid_stop_info.signo;
744      if (log) {
745        LLDB_LOGF(log,
746                  "GDBRemoteCommunicationServerLLGS::%s pid %" PRIu64
747                  " tid %" PRIu64
748                  " got signal signo = %d, reason = %d, exc_type = %" PRIu64,
749                  __FUNCTION__, process.GetID(), tid, signum,
750                  tid_stop_info.reason, tid_stop_info.details.exception.type);
751      }
752  
753      json::Object thread_obj;
754  
755      if (!abridged) {
756        if (std::optional<json::Object> registers = GetRegistersAsJSON(thread))
757          thread_obj.try_emplace("registers", std::move(*registers));
758      }
759  
760      thread_obj.try_emplace("tid", static_cast<int64_t>(tid));
761  
762      if (signum != 0)
763        thread_obj.try_emplace("signal", signum);
764  
765      const std::string thread_name = thread.GetName();
766      if (!thread_name.empty())
767        thread_obj.try_emplace("name", thread_name);
768  
769      const char *stop_reason = GetStopReasonString(tid_stop_info.reason);
770      if (stop_reason)
771        thread_obj.try_emplace("reason", stop_reason);
772  
773      if (!description.empty())
774        thread_obj.try_emplace("description", description);
775  
776      if ((tid_stop_info.reason == eStopReasonException) &&
777          tid_stop_info.details.exception.type) {
778        thread_obj.try_emplace(
779            "metype", static_cast<int64_t>(tid_stop_info.details.exception.type));
780  
781        json::Array medata_array;
782        for (uint32_t i = 0; i < tid_stop_info.details.exception.data_count;
783             ++i) {
784          medata_array.push_back(
785              static_cast<int64_t>(tid_stop_info.details.exception.data[i]));
786        }
787        thread_obj.try_emplace("medata", std::move(medata_array));
788      }
789      threads_array.push_back(std::move(thread_obj));
790    }
791    return threads_array;
792  }
793  
794  StreamString
PrepareStopReplyPacketForThread(NativeThreadProtocol & thread)795  GDBRemoteCommunicationServerLLGS::PrepareStopReplyPacketForThread(
796      NativeThreadProtocol &thread) {
797    Log *log = GetLog(LLDBLog::Process | LLDBLog::Thread);
798  
799    NativeProcessProtocol &process = thread.GetProcess();
800  
801    LLDB_LOG(log, "preparing packet for pid {0} tid {1}", process.GetID(),
802             thread.GetID());
803  
804    // Grab the reason this thread stopped.
805    StreamString response;
806    struct ThreadStopInfo tid_stop_info;
807    std::string description;
808    if (!thread.GetStopReason(tid_stop_info, description))
809      return response;
810  
811    // FIXME implement register handling for exec'd inferiors.
812    // if (tid_stop_info.reason == eStopReasonExec) {
813    //     const bool force = true;
814    //     InitializeRegisters(force);
815    // }
816  
817    // Output the T packet with the thread
818    response.PutChar('T');
819    int signum = tid_stop_info.signo;
820    LLDB_LOG(
821        log,
822        "pid {0}, tid {1}, got signal signo = {2}, reason = {3}, exc_type = {4}",
823        process.GetID(), thread.GetID(), signum, int(tid_stop_info.reason),
824        tid_stop_info.details.exception.type);
825  
826    // Print the signal number.
827    response.PutHex8(signum & 0xff);
828  
829    // Include the (pid and) tid.
830    response.PutCString("thread:");
831    AppendThreadIDToResponse(response, process.GetID(), thread.GetID());
832    response.PutChar(';');
833  
834    // Include the thread name if there is one.
835    const std::string thread_name = thread.GetName();
836    if (!thread_name.empty()) {
837      size_t thread_name_len = thread_name.length();
838  
839      if (::strcspn(thread_name.c_str(), "$#+-;:") == thread_name_len) {
840        response.PutCString("name:");
841        response.PutCString(thread_name);
842      } else {
843        // The thread name contains special chars, send as hex bytes.
844        response.PutCString("hexname:");
845        response.PutStringAsRawHex8(thread_name);
846      }
847      response.PutChar(';');
848    }
849  
850    // If a 'QListThreadsInStopReply' was sent to enable this feature, we will
851    // send all thread IDs back in the "threads" key whose value is a list of hex
852    // thread IDs separated by commas:
853    //  "threads:10a,10b,10c;"
854    // This will save the debugger from having to send a pair of qfThreadInfo and
855    // qsThreadInfo packets, but it also might take a lot of room in the stop
856    // reply packet, so it must be enabled only on systems where there are no
857    // limits on packet lengths.
858    if (m_list_threads_in_stop_reply) {
859      response.PutCString("threads:");
860  
861      uint32_t thread_num = 0;
862      for (NativeThreadProtocol &listed_thread : process.Threads()) {
863        if (thread_num > 0)
864          response.PutChar(',');
865        response.Printf("%" PRIx64, listed_thread.GetID());
866        ++thread_num;
867      }
868      response.PutChar(';');
869  
870      // Include JSON info that describes the stop reason for any threads that
871      // actually have stop reasons. We use the new "jstopinfo" key whose values
872      // is hex ascii JSON that contains the thread IDs thread stop info only for
873      // threads that have stop reasons. Only send this if we have more than one
874      // thread otherwise this packet has all the info it needs.
875      if (thread_num > 1) {
876        const bool threads_with_valid_stop_info_only = true;
877        llvm::Expected<json::Array> threads_info = GetJSONThreadsInfo(
878            *m_current_process, threads_with_valid_stop_info_only);
879        if (threads_info) {
880          response.PutCString("jstopinfo:");
881          StreamString unescaped_response;
882          unescaped_response.AsRawOstream() << std::move(*threads_info);
883          response.PutStringAsRawHex8(unescaped_response.GetData());
884          response.PutChar(';');
885        } else {
886          LLDB_LOG_ERROR(log, threads_info.takeError(),
887                         "failed to prepare a jstopinfo field for pid {1}: {0}",
888                         process.GetID());
889        }
890      }
891  
892      response.PutCString("thread-pcs");
893      char delimiter = ':';
894      for (NativeThreadProtocol &thread : process.Threads()) {
895        NativeRegisterContext &reg_ctx = thread.GetRegisterContext();
896  
897        uint32_t reg_to_read = reg_ctx.ConvertRegisterKindToRegisterNumber(
898            eRegisterKindGeneric, LLDB_REGNUM_GENERIC_PC);
899        const RegisterInfo *const reg_info_p =
900            reg_ctx.GetRegisterInfoAtIndex(reg_to_read);
901  
902        RegisterValue reg_value;
903        Status error = reg_ctx.ReadRegister(reg_info_p, reg_value);
904        if (error.Fail()) {
905          LLDB_LOGF(log, "%s failed to read register '%s' index %" PRIu32 ": %s",
906                    __FUNCTION__,
907                    reg_info_p->name ? reg_info_p->name : "<unnamed-register>",
908                    reg_to_read, error.AsCString());
909          continue;
910        }
911  
912        response.PutChar(delimiter);
913        delimiter = ',';
914        WriteRegisterValueInHexFixedWidth(response, reg_ctx, *reg_info_p,
915                                          &reg_value, endian::InlHostByteOrder());
916      }
917  
918      response.PutChar(';');
919    }
920  
921    //
922    // Expedite registers.
923    //
924  
925    // Grab the register context.
926    NativeRegisterContext &reg_ctx = thread.GetRegisterContext();
927    const auto expedited_regs =
928        reg_ctx.GetExpeditedRegisters(ExpeditedRegs::Full);
929  
930    for (auto &reg_num : expedited_regs) {
931      const RegisterInfo *const reg_info_p =
932          reg_ctx.GetRegisterInfoAtIndex(reg_num);
933      // Only expediate registers that are not contained in other registers.
934      if (reg_info_p != nullptr && reg_info_p->value_regs == nullptr) {
935        RegisterValue reg_value;
936        Status error = reg_ctx.ReadRegister(reg_info_p, reg_value);
937        if (error.Success()) {
938          response.Printf("%.02x:", reg_num);
939          WriteRegisterValueInHexFixedWidth(response, reg_ctx, *reg_info_p,
940                                            &reg_value, lldb::eByteOrderBig);
941          response.PutChar(';');
942        } else {
943          LLDB_LOGF(log,
944                    "GDBRemoteCommunicationServerLLGS::%s failed to read "
945                    "register '%s' index %" PRIu32 ": %s",
946                    __FUNCTION__,
947                    reg_info_p->name ? reg_info_p->name : "<unnamed-register>",
948                    reg_num, error.AsCString());
949        }
950      }
951    }
952  
953    const char *reason_str = GetStopReasonString(tid_stop_info.reason);
954    if (reason_str != nullptr) {
955      response.Printf("reason:%s;", reason_str);
956    }
957  
958    if (!description.empty()) {
959      // Description may contains special chars, send as hex bytes.
960      response.PutCString("description:");
961      response.PutStringAsRawHex8(description);
962      response.PutChar(';');
963    } else if ((tid_stop_info.reason == eStopReasonException) &&
964               tid_stop_info.details.exception.type) {
965      response.PutCString("metype:");
966      response.PutHex64(tid_stop_info.details.exception.type);
967      response.PutCString(";mecount:");
968      response.PutHex32(tid_stop_info.details.exception.data_count);
969      response.PutChar(';');
970  
971      for (uint32_t i = 0; i < tid_stop_info.details.exception.data_count; ++i) {
972        response.PutCString("medata:");
973        response.PutHex64(tid_stop_info.details.exception.data[i]);
974        response.PutChar(';');
975      }
976    }
977  
978    // Include child process PID/TID for forks.
979    if (tid_stop_info.reason == eStopReasonFork ||
980        tid_stop_info.reason == eStopReasonVFork) {
981      assert(bool(m_extensions_supported &
982                  NativeProcessProtocol::Extension::multiprocess));
983      if (tid_stop_info.reason == eStopReasonFork)
984        assert(bool(m_extensions_supported &
985                    NativeProcessProtocol::Extension::fork));
986      if (tid_stop_info.reason == eStopReasonVFork)
987        assert(bool(m_extensions_supported &
988                    NativeProcessProtocol::Extension::vfork));
989      response.Printf("%s:p%" PRIx64 ".%" PRIx64 ";", reason_str,
990                      tid_stop_info.details.fork.child_pid,
991                      tid_stop_info.details.fork.child_tid);
992    }
993  
994    return response;
995  }
996  
997  GDBRemoteCommunication::PacketResult
SendStopReplyPacketForThread(NativeProcessProtocol & process,lldb::tid_t tid,bool force_synchronous)998  GDBRemoteCommunicationServerLLGS::SendStopReplyPacketForThread(
999      NativeProcessProtocol &process, lldb::tid_t tid, bool force_synchronous) {
1000    // Ensure we can get info on the given thread.
1001    NativeThreadProtocol *thread = process.GetThreadByID(tid);
1002    if (!thread)
1003      return SendErrorResponse(51);
1004  
1005    StreamString response = PrepareStopReplyPacketForThread(*thread);
1006    if (response.Empty())
1007      return SendErrorResponse(42);
1008  
1009    if (m_non_stop && !force_synchronous) {
1010      PacketResult ret = SendNotificationPacketNoLock(
1011          "Stop", m_stop_notification_queue, response.GetString());
1012      // Queue notification events for the remaining threads.
1013      EnqueueStopReplyPackets(tid);
1014      return ret;
1015    }
1016  
1017    return SendPacketNoLock(response.GetString());
1018  }
1019  
EnqueueStopReplyPackets(lldb::tid_t thread_to_skip)1020  void GDBRemoteCommunicationServerLLGS::EnqueueStopReplyPackets(
1021      lldb::tid_t thread_to_skip) {
1022    if (!m_non_stop)
1023      return;
1024  
1025    for (NativeThreadProtocol &listed_thread : m_current_process->Threads()) {
1026      if (listed_thread.GetID() != thread_to_skip) {
1027        StreamString stop_reply = PrepareStopReplyPacketForThread(listed_thread);
1028        if (!stop_reply.Empty())
1029          m_stop_notification_queue.push_back(stop_reply.GetString().str());
1030      }
1031    }
1032  }
1033  
HandleInferiorState_Exited(NativeProcessProtocol * process)1034  void GDBRemoteCommunicationServerLLGS::HandleInferiorState_Exited(
1035      NativeProcessProtocol *process) {
1036    assert(process && "process cannot be NULL");
1037  
1038    Log *log = GetLog(LLDBLog::Process);
1039    LLDB_LOGF(log, "GDBRemoteCommunicationServerLLGS::%s called", __FUNCTION__);
1040  
1041    PacketResult result = SendStopReasonForState(
1042        *process, StateType::eStateExited, /*force_synchronous=*/false);
1043    if (result != PacketResult::Success) {
1044      LLDB_LOGF(log,
1045                "GDBRemoteCommunicationServerLLGS::%s failed to send stop "
1046                "notification for PID %" PRIu64 ", state: eStateExited",
1047                __FUNCTION__, process->GetID());
1048    }
1049  
1050    if (m_current_process == process)
1051      m_current_process = nullptr;
1052    if (m_continue_process == process)
1053      m_continue_process = nullptr;
1054  
1055    lldb::pid_t pid = process->GetID();
1056    m_mainloop.AddPendingCallback([this, pid](MainLoopBase &loop) {
1057      auto find_it = m_debugged_processes.find(pid);
1058      assert(find_it != m_debugged_processes.end());
1059      bool vkilled = bool(find_it->second.flags & DebuggedProcess::Flag::vkilled);
1060      m_debugged_processes.erase(find_it);
1061      // Terminate the main loop only if vKill has not been used.
1062      // When running in non-stop mode, wait for the vStopped to clear
1063      // the notification queue.
1064      if (m_debugged_processes.empty() && !m_non_stop && !vkilled) {
1065        // Close the pipe to the inferior terminal i/o if we launched it and set
1066        // one up.
1067        MaybeCloseInferiorTerminalConnection();
1068  
1069        // We are ready to exit the debug monitor.
1070        m_exit_now = true;
1071        loop.RequestTermination();
1072      }
1073    });
1074  }
1075  
HandleInferiorState_Stopped(NativeProcessProtocol * process)1076  void GDBRemoteCommunicationServerLLGS::HandleInferiorState_Stopped(
1077      NativeProcessProtocol *process) {
1078    assert(process && "process cannot be NULL");
1079  
1080    Log *log = GetLog(LLDBLog::Process);
1081    LLDB_LOGF(log, "GDBRemoteCommunicationServerLLGS::%s called", __FUNCTION__);
1082  
1083    PacketResult result = SendStopReasonForState(
1084        *process, StateType::eStateStopped, /*force_synchronous=*/false);
1085    if (result != PacketResult::Success) {
1086      LLDB_LOGF(log,
1087                "GDBRemoteCommunicationServerLLGS::%s failed to send stop "
1088                "notification for PID %" PRIu64 ", state: eStateExited",
1089                __FUNCTION__, process->GetID());
1090    }
1091  }
1092  
ProcessStateChanged(NativeProcessProtocol * process,lldb::StateType state)1093  void GDBRemoteCommunicationServerLLGS::ProcessStateChanged(
1094      NativeProcessProtocol *process, lldb::StateType state) {
1095    assert(process && "process cannot be NULL");
1096    Log *log = GetLog(LLDBLog::Process);
1097    if (log) {
1098      LLDB_LOGF(log,
1099                "GDBRemoteCommunicationServerLLGS::%s called with "
1100                "NativeProcessProtocol pid %" PRIu64 ", state: %s",
1101                __FUNCTION__, process->GetID(), StateAsCString(state));
1102    }
1103  
1104    switch (state) {
1105    case StateType::eStateRunning:
1106      break;
1107  
1108    case StateType::eStateStopped:
1109      // Make sure we get all of the pending stdout/stderr from the inferior and
1110      // send it to the lldb host before we send the state change notification
1111      SendProcessOutput();
1112      // Then stop the forwarding, so that any late output (see llvm.org/pr25652)
1113      // does not interfere with our protocol.
1114      if (!m_non_stop)
1115        StopSTDIOForwarding();
1116      HandleInferiorState_Stopped(process);
1117      break;
1118  
1119    case StateType::eStateExited:
1120      // Same as above
1121      SendProcessOutput();
1122      if (!m_non_stop)
1123        StopSTDIOForwarding();
1124      HandleInferiorState_Exited(process);
1125      break;
1126  
1127    default:
1128      if (log) {
1129        LLDB_LOGF(log,
1130                  "GDBRemoteCommunicationServerLLGS::%s didn't handle state "
1131                  "change for pid %" PRIu64 ", new state: %s",
1132                  __FUNCTION__, process->GetID(), StateAsCString(state));
1133      }
1134      break;
1135    }
1136  }
1137  
DidExec(NativeProcessProtocol * process)1138  void GDBRemoteCommunicationServerLLGS::DidExec(NativeProcessProtocol *process) {
1139    ClearProcessSpecificData();
1140  }
1141  
NewSubprocess(NativeProcessProtocol * parent_process,std::unique_ptr<NativeProcessProtocol> child_process)1142  void GDBRemoteCommunicationServerLLGS::NewSubprocess(
1143      NativeProcessProtocol *parent_process,
1144      std::unique_ptr<NativeProcessProtocol> child_process) {
1145    lldb::pid_t child_pid = child_process->GetID();
1146    assert(child_pid != LLDB_INVALID_PROCESS_ID);
1147    assert(m_debugged_processes.find(child_pid) == m_debugged_processes.end());
1148    m_debugged_processes.emplace(
1149        child_pid,
1150        DebuggedProcess{std::move(child_process), DebuggedProcess::Flag{}});
1151  }
1152  
DataAvailableCallback()1153  void GDBRemoteCommunicationServerLLGS::DataAvailableCallback() {
1154    Log *log = GetLog(GDBRLog::Comm);
1155  
1156    bool interrupt = false;
1157    bool done = false;
1158    Status error;
1159    while (true) {
1160      const PacketResult result = GetPacketAndSendResponse(
1161          std::chrono::microseconds(0), error, interrupt, done);
1162      if (result == PacketResult::ErrorReplyTimeout)
1163        break; // No more packets in the queue
1164  
1165      if ((result != PacketResult::Success)) {
1166        LLDB_LOGF(log,
1167                  "GDBRemoteCommunicationServerLLGS::%s processing a packet "
1168                  "failed: %s",
1169                  __FUNCTION__, error.AsCString());
1170        m_mainloop.RequestTermination();
1171        break;
1172      }
1173    }
1174  }
1175  
InitializeConnection(std::unique_ptr<Connection> connection)1176  Status GDBRemoteCommunicationServerLLGS::InitializeConnection(
1177      std::unique_ptr<Connection> connection) {
1178    IOObjectSP read_object_sp = connection->GetReadObject();
1179    GDBRemoteCommunicationServer::SetConnection(std::move(connection));
1180  
1181    Status error;
1182    m_network_handle_up = m_mainloop.RegisterReadObject(
1183        read_object_sp, [this](MainLoopBase &) { DataAvailableCallback(); },
1184        error);
1185    return error;
1186  }
1187  
1188  GDBRemoteCommunication::PacketResult
SendONotification(const char * buffer,uint32_t len)1189  GDBRemoteCommunicationServerLLGS::SendONotification(const char *buffer,
1190                                                      uint32_t len) {
1191    if ((buffer == nullptr) || (len == 0)) {
1192      // Nothing to send.
1193      return PacketResult::Success;
1194    }
1195  
1196    StreamString response;
1197    response.PutChar('O');
1198    response.PutBytesAsRawHex8(buffer, len);
1199  
1200    if (m_non_stop)
1201      return SendNotificationPacketNoLock("Stdio", m_stdio_notification_queue,
1202                                          response.GetString());
1203    return SendPacketNoLock(response.GetString());
1204  }
1205  
SetSTDIOFileDescriptor(int fd)1206  Status GDBRemoteCommunicationServerLLGS::SetSTDIOFileDescriptor(int fd) {
1207    Status error;
1208  
1209    // Set up the reading/handling of process I/O
1210    std::unique_ptr<ConnectionFileDescriptor> conn_up(
1211        new ConnectionFileDescriptor(fd, true));
1212    if (!conn_up) {
1213      error.SetErrorString("failed to create ConnectionFileDescriptor");
1214      return error;
1215    }
1216  
1217    m_stdio_communication.SetCloseOnEOF(false);
1218    m_stdio_communication.SetConnection(std::move(conn_up));
1219    if (!m_stdio_communication.IsConnected()) {
1220      error.SetErrorString(
1221          "failed to set connection for inferior I/O communication");
1222      return error;
1223    }
1224  
1225    return Status();
1226  }
1227  
StartSTDIOForwarding()1228  void GDBRemoteCommunicationServerLLGS::StartSTDIOForwarding() {
1229    // Don't forward if not connected (e.g. when attaching).
1230    if (!m_stdio_communication.IsConnected())
1231      return;
1232  
1233    Status error;
1234    assert(!m_stdio_handle_up);
1235    m_stdio_handle_up = m_mainloop.RegisterReadObject(
1236        m_stdio_communication.GetConnection()->GetReadObject(),
1237        [this](MainLoopBase &) { SendProcessOutput(); }, error);
1238  
1239    if (!m_stdio_handle_up) {
1240      // Not much we can do about the failure. Log it and continue without
1241      // forwarding.
1242      if (Log *log = GetLog(LLDBLog::Process))
1243        LLDB_LOG(log, "Failed to set up stdio forwarding: {0}", error);
1244    }
1245  }
1246  
StopSTDIOForwarding()1247  void GDBRemoteCommunicationServerLLGS::StopSTDIOForwarding() {
1248    m_stdio_handle_up.reset();
1249  }
1250  
SendProcessOutput()1251  void GDBRemoteCommunicationServerLLGS::SendProcessOutput() {
1252    char buffer[1024];
1253    ConnectionStatus status;
1254    Status error;
1255    while (true) {
1256      size_t bytes_read = m_stdio_communication.Read(
1257          buffer, sizeof buffer, std::chrono::microseconds(0), status, &error);
1258      switch (status) {
1259      case eConnectionStatusSuccess:
1260        SendONotification(buffer, bytes_read);
1261        break;
1262      case eConnectionStatusLostConnection:
1263      case eConnectionStatusEndOfFile:
1264      case eConnectionStatusError:
1265      case eConnectionStatusNoConnection:
1266        if (Log *log = GetLog(LLDBLog::Process))
1267          LLDB_LOGF(log,
1268                    "GDBRemoteCommunicationServerLLGS::%s Stopping stdio "
1269                    "forwarding as communication returned status %d (error: "
1270                    "%s)",
1271                    __FUNCTION__, status, error.AsCString());
1272        m_stdio_handle_up.reset();
1273        return;
1274  
1275      case eConnectionStatusInterrupted:
1276      case eConnectionStatusTimedOut:
1277        return;
1278      }
1279    }
1280  }
1281  
1282  GDBRemoteCommunication::PacketResult
Handle_jLLDBTraceSupported(StringExtractorGDBRemote & packet)1283  GDBRemoteCommunicationServerLLGS::Handle_jLLDBTraceSupported(
1284      StringExtractorGDBRemote &packet) {
1285  
1286    // Fail if we don't have a current process.
1287    if (!m_current_process ||
1288        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID))
1289      return SendErrorResponse(Status("Process not running."));
1290  
1291    return SendJSONResponse(m_current_process->TraceSupported());
1292  }
1293  
1294  GDBRemoteCommunication::PacketResult
Handle_jLLDBTraceStop(StringExtractorGDBRemote & packet)1295  GDBRemoteCommunicationServerLLGS::Handle_jLLDBTraceStop(
1296      StringExtractorGDBRemote &packet) {
1297    // Fail if we don't have a current process.
1298    if (!m_current_process ||
1299        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID))
1300      return SendErrorResponse(Status("Process not running."));
1301  
1302    packet.ConsumeFront("jLLDBTraceStop:");
1303    Expected<TraceStopRequest> stop_request =
1304        json::parse<TraceStopRequest>(packet.Peek(), "TraceStopRequest");
1305    if (!stop_request)
1306      return SendErrorResponse(stop_request.takeError());
1307  
1308    if (Error err = m_current_process->TraceStop(*stop_request))
1309      return SendErrorResponse(std::move(err));
1310  
1311    return SendOKResponse();
1312  }
1313  
1314  GDBRemoteCommunication::PacketResult
Handle_jLLDBTraceStart(StringExtractorGDBRemote & packet)1315  GDBRemoteCommunicationServerLLGS::Handle_jLLDBTraceStart(
1316      StringExtractorGDBRemote &packet) {
1317  
1318    // Fail if we don't have a current process.
1319    if (!m_current_process ||
1320        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID))
1321      return SendErrorResponse(Status("Process not running."));
1322  
1323    packet.ConsumeFront("jLLDBTraceStart:");
1324    Expected<TraceStartRequest> request =
1325        json::parse<TraceStartRequest>(packet.Peek(), "TraceStartRequest");
1326    if (!request)
1327      return SendErrorResponse(request.takeError());
1328  
1329    if (Error err = m_current_process->TraceStart(packet.Peek(), request->type))
1330      return SendErrorResponse(std::move(err));
1331  
1332    return SendOKResponse();
1333  }
1334  
1335  GDBRemoteCommunication::PacketResult
Handle_jLLDBTraceGetState(StringExtractorGDBRemote & packet)1336  GDBRemoteCommunicationServerLLGS::Handle_jLLDBTraceGetState(
1337      StringExtractorGDBRemote &packet) {
1338  
1339    // Fail if we don't have a current process.
1340    if (!m_current_process ||
1341        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID))
1342      return SendErrorResponse(Status("Process not running."));
1343  
1344    packet.ConsumeFront("jLLDBTraceGetState:");
1345    Expected<TraceGetStateRequest> request =
1346        json::parse<TraceGetStateRequest>(packet.Peek(), "TraceGetStateRequest");
1347    if (!request)
1348      return SendErrorResponse(request.takeError());
1349  
1350    return SendJSONResponse(m_current_process->TraceGetState(request->type));
1351  }
1352  
1353  GDBRemoteCommunication::PacketResult
Handle_jLLDBTraceGetBinaryData(StringExtractorGDBRemote & packet)1354  GDBRemoteCommunicationServerLLGS::Handle_jLLDBTraceGetBinaryData(
1355      StringExtractorGDBRemote &packet) {
1356  
1357    // Fail if we don't have a current process.
1358    if (!m_current_process ||
1359        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID))
1360      return SendErrorResponse(Status("Process not running."));
1361  
1362    packet.ConsumeFront("jLLDBTraceGetBinaryData:");
1363    llvm::Expected<TraceGetBinaryDataRequest> request =
1364        llvm::json::parse<TraceGetBinaryDataRequest>(packet.Peek(),
1365                                                     "TraceGetBinaryDataRequest");
1366    if (!request)
1367      return SendErrorResponse(Status(request.takeError()));
1368  
1369    if (Expected<std::vector<uint8_t>> bytes =
1370            m_current_process->TraceGetBinaryData(*request)) {
1371      StreamGDBRemote response;
1372      response.PutEscapedBytes(bytes->data(), bytes->size());
1373      return SendPacketNoLock(response.GetString());
1374    } else
1375      return SendErrorResponse(bytes.takeError());
1376  }
1377  
1378  GDBRemoteCommunication::PacketResult
Handle_qProcessInfo(StringExtractorGDBRemote & packet)1379  GDBRemoteCommunicationServerLLGS::Handle_qProcessInfo(
1380      StringExtractorGDBRemote &packet) {
1381    // Fail if we don't have a current process.
1382    if (!m_current_process ||
1383        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID))
1384      return SendErrorResponse(68);
1385  
1386    lldb::pid_t pid = m_current_process->GetID();
1387  
1388    if (pid == LLDB_INVALID_PROCESS_ID)
1389      return SendErrorResponse(1);
1390  
1391    ProcessInstanceInfo proc_info;
1392    if (!Host::GetProcessInfo(pid, proc_info))
1393      return SendErrorResponse(1);
1394  
1395    StreamString response;
1396    CreateProcessInfoResponse_DebugServerStyle(proc_info, response);
1397    return SendPacketNoLock(response.GetString());
1398  }
1399  
1400  GDBRemoteCommunication::PacketResult
Handle_qC(StringExtractorGDBRemote & packet)1401  GDBRemoteCommunicationServerLLGS::Handle_qC(StringExtractorGDBRemote &packet) {
1402    // Fail if we don't have a current process.
1403    if (!m_current_process ||
1404        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID))
1405      return SendErrorResponse(68);
1406  
1407    // Make sure we set the current thread so g and p packets return the data the
1408    // gdb will expect.
1409    lldb::tid_t tid = m_current_process->GetCurrentThreadID();
1410    SetCurrentThreadID(tid);
1411  
1412    NativeThreadProtocol *thread = m_current_process->GetCurrentThread();
1413    if (!thread)
1414      return SendErrorResponse(69);
1415  
1416    StreamString response;
1417    response.PutCString("QC");
1418    AppendThreadIDToResponse(response, m_current_process->GetID(),
1419                             thread->GetID());
1420  
1421    return SendPacketNoLock(response.GetString());
1422  }
1423  
1424  GDBRemoteCommunication::PacketResult
Handle_k(StringExtractorGDBRemote & packet)1425  GDBRemoteCommunicationServerLLGS::Handle_k(StringExtractorGDBRemote &packet) {
1426    Log *log = GetLog(LLDBLog::Process);
1427  
1428    if (!m_non_stop)
1429      StopSTDIOForwarding();
1430  
1431    if (m_debugged_processes.empty()) {
1432      LLDB_LOG(log, "No debugged process found.");
1433      return PacketResult::Success;
1434    }
1435  
1436    for (auto it = m_debugged_processes.begin(); it != m_debugged_processes.end();
1437         ++it) {
1438      LLDB_LOG(log, "Killing process {0}", it->first);
1439      Status error = it->second.process_up->Kill();
1440      if (error.Fail())
1441        LLDB_LOG(log, "Failed to kill debugged process {0}: {1}", it->first,
1442                 error);
1443    }
1444  
1445    // The response to kill packet is undefined per the spec.  LLDB
1446    // follows the same rules as for continue packets, i.e. no response
1447    // in all-stop mode, and "OK" in non-stop mode; in both cases this
1448    // is followed by the actual stop reason.
1449    return SendContinueSuccessResponse();
1450  }
1451  
1452  GDBRemoteCommunication::PacketResult
Handle_vKill(StringExtractorGDBRemote & packet)1453  GDBRemoteCommunicationServerLLGS::Handle_vKill(
1454      StringExtractorGDBRemote &packet) {
1455    if (!m_non_stop)
1456      StopSTDIOForwarding();
1457  
1458    packet.SetFilePos(6); // vKill;
1459    uint32_t pid = packet.GetU32(LLDB_INVALID_PROCESS_ID, 16);
1460    if (pid == LLDB_INVALID_PROCESS_ID)
1461      return SendIllFormedResponse(packet,
1462                                   "vKill failed to parse the process id");
1463  
1464    auto it = m_debugged_processes.find(pid);
1465    if (it == m_debugged_processes.end())
1466      return SendErrorResponse(42);
1467  
1468    Status error = it->second.process_up->Kill();
1469    if (error.Fail())
1470      return SendErrorResponse(error.ToError());
1471  
1472    // OK response is sent when the process dies.
1473    it->second.flags |= DebuggedProcess::Flag::vkilled;
1474    return PacketResult::Success;
1475  }
1476  
1477  GDBRemoteCommunication::PacketResult
Handle_QSetDisableASLR(StringExtractorGDBRemote & packet)1478  GDBRemoteCommunicationServerLLGS::Handle_QSetDisableASLR(
1479      StringExtractorGDBRemote &packet) {
1480    packet.SetFilePos(::strlen("QSetDisableASLR:"));
1481    if (packet.GetU32(0))
1482      m_process_launch_info.GetFlags().Set(eLaunchFlagDisableASLR);
1483    else
1484      m_process_launch_info.GetFlags().Clear(eLaunchFlagDisableASLR);
1485    return SendOKResponse();
1486  }
1487  
1488  GDBRemoteCommunication::PacketResult
Handle_QSetWorkingDir(StringExtractorGDBRemote & packet)1489  GDBRemoteCommunicationServerLLGS::Handle_QSetWorkingDir(
1490      StringExtractorGDBRemote &packet) {
1491    packet.SetFilePos(::strlen("QSetWorkingDir:"));
1492    std::string path;
1493    packet.GetHexByteString(path);
1494    m_process_launch_info.SetWorkingDirectory(FileSpec(path));
1495    return SendOKResponse();
1496  }
1497  
1498  GDBRemoteCommunication::PacketResult
Handle_qGetWorkingDir(StringExtractorGDBRemote & packet)1499  GDBRemoteCommunicationServerLLGS::Handle_qGetWorkingDir(
1500      StringExtractorGDBRemote &packet) {
1501    FileSpec working_dir{m_process_launch_info.GetWorkingDirectory()};
1502    if (working_dir) {
1503      StreamString response;
1504      response.PutStringAsRawHex8(working_dir.GetPath().c_str());
1505      return SendPacketNoLock(response.GetString());
1506    }
1507  
1508    return SendErrorResponse(14);
1509  }
1510  
1511  GDBRemoteCommunication::PacketResult
Handle_QThreadSuffixSupported(StringExtractorGDBRemote & packet)1512  GDBRemoteCommunicationServerLLGS::Handle_QThreadSuffixSupported(
1513      StringExtractorGDBRemote &packet) {
1514    m_thread_suffix_supported = true;
1515    return SendOKResponse();
1516  }
1517  
1518  GDBRemoteCommunication::PacketResult
Handle_QListThreadsInStopReply(StringExtractorGDBRemote & packet)1519  GDBRemoteCommunicationServerLLGS::Handle_QListThreadsInStopReply(
1520      StringExtractorGDBRemote &packet) {
1521    m_list_threads_in_stop_reply = true;
1522    return SendOKResponse();
1523  }
1524  
1525  GDBRemoteCommunication::PacketResult
ResumeProcess(NativeProcessProtocol & process,const ResumeActionList & actions)1526  GDBRemoteCommunicationServerLLGS::ResumeProcess(
1527      NativeProcessProtocol &process, const ResumeActionList &actions) {
1528    Log *log = GetLog(LLDBLog::Process | LLDBLog::Thread);
1529  
1530    // In non-stop protocol mode, the process could be running already.
1531    // We do not support resuming threads independently, so just error out.
1532    if (!process.CanResume()) {
1533      LLDB_LOG(log, "process {0} cannot be resumed (state={1})", process.GetID(),
1534               process.GetState());
1535      return SendErrorResponse(0x37);
1536    }
1537  
1538    Status error = process.Resume(actions);
1539    if (error.Fail()) {
1540      LLDB_LOG(log, "process {0} failed to resume: {1}", process.GetID(), error);
1541      return SendErrorResponse(GDBRemoteServerError::eErrorResume);
1542    }
1543  
1544    LLDB_LOG(log, "process {0} resumed", process.GetID());
1545  
1546    return PacketResult::Success;
1547  }
1548  
1549  GDBRemoteCommunication::PacketResult
Handle_C(StringExtractorGDBRemote & packet)1550  GDBRemoteCommunicationServerLLGS::Handle_C(StringExtractorGDBRemote &packet) {
1551    Log *log = GetLog(LLDBLog::Process | LLDBLog::Thread);
1552    LLDB_LOGF(log, "GDBRemoteCommunicationServerLLGS::%s called", __FUNCTION__);
1553  
1554    // Ensure we have a native process.
1555    if (!m_continue_process) {
1556      LLDB_LOGF(log,
1557                "GDBRemoteCommunicationServerLLGS::%s no debugged process "
1558                "shared pointer",
1559                __FUNCTION__);
1560      return SendErrorResponse(0x36);
1561    }
1562  
1563    // Pull out the signal number.
1564    packet.SetFilePos(::strlen("C"));
1565    if (packet.GetBytesLeft() < 1) {
1566      // Shouldn't be using a C without a signal.
1567      return SendIllFormedResponse(packet, "C packet specified without signal.");
1568    }
1569    const uint32_t signo =
1570        packet.GetHexMaxU32(false, std::numeric_limits<uint32_t>::max());
1571    if (signo == std::numeric_limits<uint32_t>::max())
1572      return SendIllFormedResponse(packet, "failed to parse signal number");
1573  
1574    // Handle optional continue address.
1575    if (packet.GetBytesLeft() > 0) {
1576      // FIXME add continue at address support for $C{signo}[;{continue-address}].
1577      if (*packet.Peek() == ';')
1578        return SendUnimplementedResponse(packet.GetStringRef().data());
1579      else
1580        return SendIllFormedResponse(
1581            packet, "unexpected content after $C{signal-number}");
1582    }
1583  
1584    // In non-stop protocol mode, the process could be running already.
1585    // We do not support resuming threads independently, so just error out.
1586    if (!m_continue_process->CanResume()) {
1587      LLDB_LOG(log, "process cannot be resumed (state={0})",
1588               m_continue_process->GetState());
1589      return SendErrorResponse(0x37);
1590    }
1591  
1592    ResumeActionList resume_actions(StateType::eStateRunning,
1593                                    LLDB_INVALID_SIGNAL_NUMBER);
1594    Status error;
1595  
1596    // We have two branches: what to do if a continue thread is specified (in
1597    // which case we target sending the signal to that thread), or when we don't
1598    // have a continue thread set (in which case we send a signal to the
1599    // process).
1600  
1601    // TODO discuss with Greg Clayton, make sure this makes sense.
1602  
1603    lldb::tid_t signal_tid = GetContinueThreadID();
1604    if (signal_tid != LLDB_INVALID_THREAD_ID) {
1605      // The resume action for the continue thread (or all threads if a continue
1606      // thread is not set).
1607      ResumeAction action = {GetContinueThreadID(), StateType::eStateRunning,
1608                             static_cast<int>(signo)};
1609  
1610      // Add the action for the continue thread (or all threads when the continue
1611      // thread isn't present).
1612      resume_actions.Append(action);
1613    } else {
1614      // Send the signal to the process since we weren't targeting a specific
1615      // continue thread with the signal.
1616      error = m_continue_process->Signal(signo);
1617      if (error.Fail()) {
1618        LLDB_LOG(log, "failed to send signal for process {0}: {1}",
1619                 m_continue_process->GetID(), error);
1620  
1621        return SendErrorResponse(0x52);
1622      }
1623    }
1624  
1625    // NB: this checks CanResume() twice but using a single code path for
1626    // resuming still seems worth it.
1627    PacketResult resume_res = ResumeProcess(*m_continue_process, resume_actions);
1628    if (resume_res != PacketResult::Success)
1629      return resume_res;
1630  
1631    // Don't send an "OK" packet, except in non-stop mode;
1632    // otherwise, the response is the stopped/exited message.
1633    return SendContinueSuccessResponse();
1634  }
1635  
1636  GDBRemoteCommunication::PacketResult
Handle_c(StringExtractorGDBRemote & packet)1637  GDBRemoteCommunicationServerLLGS::Handle_c(StringExtractorGDBRemote &packet) {
1638    Log *log = GetLog(LLDBLog::Process | LLDBLog::Thread);
1639    LLDB_LOGF(log, "GDBRemoteCommunicationServerLLGS::%s called", __FUNCTION__);
1640  
1641    packet.SetFilePos(packet.GetFilePos() + ::strlen("c"));
1642  
1643    // For now just support all continue.
1644    const bool has_continue_address = (packet.GetBytesLeft() > 0);
1645    if (has_continue_address) {
1646      LLDB_LOG(log, "not implemented for c[address] variant [{0} remains]",
1647               packet.Peek());
1648      return SendUnimplementedResponse(packet.GetStringRef().data());
1649    }
1650  
1651    // Ensure we have a native process.
1652    if (!m_continue_process) {
1653      LLDB_LOGF(log,
1654                "GDBRemoteCommunicationServerLLGS::%s no debugged process "
1655                "shared pointer",
1656                __FUNCTION__);
1657      return SendErrorResponse(0x36);
1658    }
1659  
1660    // Build the ResumeActionList
1661    ResumeActionList actions(StateType::eStateRunning,
1662                             LLDB_INVALID_SIGNAL_NUMBER);
1663  
1664    PacketResult resume_res = ResumeProcess(*m_continue_process, actions);
1665    if (resume_res != PacketResult::Success)
1666      return resume_res;
1667  
1668    return SendContinueSuccessResponse();
1669  }
1670  
1671  GDBRemoteCommunication::PacketResult
Handle_vCont_actions(StringExtractorGDBRemote & packet)1672  GDBRemoteCommunicationServerLLGS::Handle_vCont_actions(
1673      StringExtractorGDBRemote &packet) {
1674    StreamString response;
1675    response.Printf("vCont;c;C;s;S;t");
1676  
1677    return SendPacketNoLock(response.GetString());
1678  }
1679  
ResumeActionListStopsAllThreads(ResumeActionList & actions)1680  static bool ResumeActionListStopsAllThreads(ResumeActionList &actions) {
1681    // We're doing a stop-all if and only if our only action is a "t" for all
1682    // threads.
1683    if (const ResumeAction *default_action =
1684            actions.GetActionForThread(LLDB_INVALID_THREAD_ID, false)) {
1685      if (default_action->state == eStateSuspended && actions.GetSize() == 1)
1686        return true;
1687    }
1688  
1689    return false;
1690  }
1691  
1692  GDBRemoteCommunication::PacketResult
Handle_vCont(StringExtractorGDBRemote & packet)1693  GDBRemoteCommunicationServerLLGS::Handle_vCont(
1694      StringExtractorGDBRemote &packet) {
1695    Log *log = GetLog(LLDBLog::Process);
1696    LLDB_LOGF(log, "GDBRemoteCommunicationServerLLGS::%s handling vCont packet",
1697              __FUNCTION__);
1698  
1699    packet.SetFilePos(::strlen("vCont"));
1700  
1701    if (packet.GetBytesLeft() == 0) {
1702      LLDB_LOGF(log,
1703                "GDBRemoteCommunicationServerLLGS::%s missing action from "
1704                "vCont package",
1705                __FUNCTION__);
1706      return SendIllFormedResponse(packet, "Missing action from vCont package");
1707    }
1708  
1709    if (::strcmp(packet.Peek(), ";s") == 0) {
1710      // Move past the ';', then do a simple 's'.
1711      packet.SetFilePos(packet.GetFilePos() + 1);
1712      return Handle_s(packet);
1713    }
1714  
1715    std::unordered_map<lldb::pid_t, ResumeActionList> thread_actions;
1716  
1717    while (packet.GetBytesLeft() && *packet.Peek() == ';') {
1718      // Skip the semi-colon.
1719      packet.GetChar();
1720  
1721      // Build up the thread action.
1722      ResumeAction thread_action;
1723      thread_action.tid = LLDB_INVALID_THREAD_ID;
1724      thread_action.state = eStateInvalid;
1725      thread_action.signal = LLDB_INVALID_SIGNAL_NUMBER;
1726  
1727      const char action = packet.GetChar();
1728      switch (action) {
1729      case 'C':
1730        thread_action.signal = packet.GetHexMaxU32(false, 0);
1731        if (thread_action.signal == 0)
1732          return SendIllFormedResponse(
1733              packet, "Could not parse signal in vCont packet C action");
1734        [[fallthrough]];
1735  
1736      case 'c':
1737        // Continue
1738        thread_action.state = eStateRunning;
1739        break;
1740  
1741      case 'S':
1742        thread_action.signal = packet.GetHexMaxU32(false, 0);
1743        if (thread_action.signal == 0)
1744          return SendIllFormedResponse(
1745              packet, "Could not parse signal in vCont packet S action");
1746        [[fallthrough]];
1747  
1748      case 's':
1749        // Step
1750        thread_action.state = eStateStepping;
1751        break;
1752  
1753      case 't':
1754        // Stop
1755        thread_action.state = eStateSuspended;
1756        break;
1757  
1758      default:
1759        return SendIllFormedResponse(packet, "Unsupported vCont action");
1760        break;
1761      }
1762  
1763      // If there's no thread-id (e.g. "vCont;c"), it's "p-1.-1".
1764      lldb::pid_t pid = StringExtractorGDBRemote::AllProcesses;
1765      lldb::tid_t tid = StringExtractorGDBRemote::AllThreads;
1766  
1767      // Parse out optional :{thread-id} value.
1768      if (packet.GetBytesLeft() && (*packet.Peek() == ':')) {
1769        // Consume the separator.
1770        packet.GetChar();
1771  
1772        auto pid_tid = packet.GetPidTid(LLDB_INVALID_PROCESS_ID);
1773        if (!pid_tid)
1774          return SendIllFormedResponse(packet, "Malformed thread-id");
1775  
1776        pid = pid_tid->first;
1777        tid = pid_tid->second;
1778      }
1779  
1780      if (thread_action.state == eStateSuspended &&
1781          tid != StringExtractorGDBRemote::AllThreads) {
1782        return SendIllFormedResponse(
1783            packet, "'t' action not supported for individual threads");
1784      }
1785  
1786      // If we get TID without PID, it's the current process.
1787      if (pid == LLDB_INVALID_PROCESS_ID) {
1788        if (!m_continue_process) {
1789          LLDB_LOG(log, "no process selected via Hc");
1790          return SendErrorResponse(0x36);
1791        }
1792        pid = m_continue_process->GetID();
1793      }
1794  
1795      assert(pid != LLDB_INVALID_PROCESS_ID);
1796      if (tid == StringExtractorGDBRemote::AllThreads)
1797        tid = LLDB_INVALID_THREAD_ID;
1798      thread_action.tid = tid;
1799  
1800      if (pid == StringExtractorGDBRemote::AllProcesses) {
1801        if (tid != LLDB_INVALID_THREAD_ID)
1802          return SendIllFormedResponse(
1803              packet, "vCont: p-1 is not valid with a specific tid");
1804        for (auto &process_it : m_debugged_processes)
1805          thread_actions[process_it.first].Append(thread_action);
1806      } else
1807        thread_actions[pid].Append(thread_action);
1808    }
1809  
1810    assert(thread_actions.size() >= 1);
1811    if (thread_actions.size() > 1 && !m_non_stop)
1812      return SendIllFormedResponse(
1813          packet,
1814          "Resuming multiple processes is supported in non-stop mode only");
1815  
1816    for (std::pair<lldb::pid_t, ResumeActionList> x : thread_actions) {
1817      auto process_it = m_debugged_processes.find(x.first);
1818      if (process_it == m_debugged_processes.end()) {
1819        LLDB_LOG(log, "vCont failed for process {0}: process not debugged",
1820                 x.first);
1821        return SendErrorResponse(GDBRemoteServerError::eErrorResume);
1822      }
1823  
1824      // There are four possible scenarios here.  These are:
1825      // 1. vCont on a stopped process that resumes at least one thread.
1826      //    In this case, we call Resume().
1827      // 2. vCont on a stopped process that leaves all threads suspended.
1828      //    A no-op.
1829      // 3. vCont on a running process that requests suspending all
1830      //    running threads.  In this case, we call Interrupt().
1831      // 4. vCont on a running process that requests suspending a subset
1832      //    of running threads or resuming a subset of suspended threads.
1833      //    Since we do not support full nonstop mode, this is unsupported
1834      //    and we return an error.
1835  
1836      assert(process_it->second.process_up);
1837      if (ResumeActionListStopsAllThreads(x.second)) {
1838        if (process_it->second.process_up->IsRunning()) {
1839          assert(m_non_stop);
1840  
1841          Status error = process_it->second.process_up->Interrupt();
1842          if (error.Fail()) {
1843            LLDB_LOG(log, "vCont failed to halt process {0}: {1}", x.first,
1844                     error);
1845            return SendErrorResponse(GDBRemoteServerError::eErrorResume);
1846          }
1847  
1848          LLDB_LOG(log, "halted process {0}", x.first);
1849  
1850          // hack to avoid enabling stdio forwarding after stop
1851          // TODO: remove this when we improve stdio forwarding for nonstop
1852          assert(thread_actions.size() == 1);
1853          return SendOKResponse();
1854        }
1855      } else {
1856        PacketResult resume_res =
1857            ResumeProcess(*process_it->second.process_up, x.second);
1858        if (resume_res != PacketResult::Success)
1859          return resume_res;
1860      }
1861    }
1862  
1863    return SendContinueSuccessResponse();
1864  }
1865  
SetCurrentThreadID(lldb::tid_t tid)1866  void GDBRemoteCommunicationServerLLGS::SetCurrentThreadID(lldb::tid_t tid) {
1867    Log *log = GetLog(LLDBLog::Thread);
1868    LLDB_LOG(log, "setting current thread id to {0}", tid);
1869  
1870    m_current_tid = tid;
1871    if (m_current_process)
1872      m_current_process->SetCurrentThreadID(m_current_tid);
1873  }
1874  
SetContinueThreadID(lldb::tid_t tid)1875  void GDBRemoteCommunicationServerLLGS::SetContinueThreadID(lldb::tid_t tid) {
1876    Log *log = GetLog(LLDBLog::Thread);
1877    LLDB_LOG(log, "setting continue thread id to {0}", tid);
1878  
1879    m_continue_tid = tid;
1880  }
1881  
1882  GDBRemoteCommunication::PacketResult
Handle_stop_reason(StringExtractorGDBRemote & packet)1883  GDBRemoteCommunicationServerLLGS::Handle_stop_reason(
1884      StringExtractorGDBRemote &packet) {
1885    // Handle the $? gdbremote command.
1886  
1887    if (m_non_stop) {
1888      // Clear the notification queue first, except for pending exit
1889      // notifications.
1890      llvm::erase_if(m_stop_notification_queue, [](const std::string &x) {
1891        return x.front() != 'W' && x.front() != 'X';
1892      });
1893  
1894      if (m_current_process) {
1895        // Queue stop reply packets for all active threads.  Start with
1896        // the current thread (for clients that don't actually support multiple
1897        // stop reasons).
1898        NativeThreadProtocol *thread = m_current_process->GetCurrentThread();
1899        if (thread) {
1900          StreamString stop_reply = PrepareStopReplyPacketForThread(*thread);
1901          if (!stop_reply.Empty())
1902            m_stop_notification_queue.push_back(stop_reply.GetString().str());
1903        }
1904        EnqueueStopReplyPackets(thread ? thread->GetID()
1905                                       : LLDB_INVALID_THREAD_ID);
1906      }
1907  
1908      // If the notification queue is empty (i.e. everything is running), send OK.
1909      if (m_stop_notification_queue.empty())
1910        return SendOKResponse();
1911  
1912      // Send the first item from the new notification queue synchronously.
1913      return SendPacketNoLock(m_stop_notification_queue.front());
1914    }
1915  
1916    // If no process, indicate error
1917    if (!m_current_process)
1918      return SendErrorResponse(02);
1919  
1920    return SendStopReasonForState(*m_current_process,
1921                                  m_current_process->GetState(),
1922                                  /*force_synchronous=*/true);
1923  }
1924  
1925  GDBRemoteCommunication::PacketResult
SendStopReasonForState(NativeProcessProtocol & process,lldb::StateType process_state,bool force_synchronous)1926  GDBRemoteCommunicationServerLLGS::SendStopReasonForState(
1927      NativeProcessProtocol &process, lldb::StateType process_state,
1928      bool force_synchronous) {
1929    Log *log = GetLog(LLDBLog::Process);
1930  
1931    if (m_disabling_non_stop) {
1932      // Check if we are waiting for any more processes to stop.  If we are,
1933      // do not send the OK response yet.
1934      for (const auto &it : m_debugged_processes) {
1935        if (it.second.process_up->IsRunning())
1936          return PacketResult::Success;
1937      }
1938  
1939      // If all expected processes were stopped after a QNonStop:0 request,
1940      // send the OK response.
1941      m_disabling_non_stop = false;
1942      return SendOKResponse();
1943    }
1944  
1945    switch (process_state) {
1946    case eStateAttaching:
1947    case eStateLaunching:
1948    case eStateRunning:
1949    case eStateStepping:
1950    case eStateDetached:
1951      // NOTE: gdb protocol doc looks like it should return $OK
1952      // when everything is running (i.e. no stopped result).
1953      return PacketResult::Success; // Ignore
1954  
1955    case eStateSuspended:
1956    case eStateStopped:
1957    case eStateCrashed: {
1958      lldb::tid_t tid = process.GetCurrentThreadID();
1959      // Make sure we set the current thread so g and p packets return the data
1960      // the gdb will expect.
1961      SetCurrentThreadID(tid);
1962      return SendStopReplyPacketForThread(process, tid, force_synchronous);
1963    }
1964  
1965    case eStateInvalid:
1966    case eStateUnloaded:
1967    case eStateExited:
1968      return SendWResponse(&process);
1969  
1970    default:
1971      LLDB_LOG(log, "pid {0}, current state reporting not handled: {1}",
1972               process.GetID(), process_state);
1973      break;
1974    }
1975  
1976    return SendErrorResponse(0);
1977  }
1978  
1979  GDBRemoteCommunication::PacketResult
Handle_qRegisterInfo(StringExtractorGDBRemote & packet)1980  GDBRemoteCommunicationServerLLGS::Handle_qRegisterInfo(
1981      StringExtractorGDBRemote &packet) {
1982    // Fail if we don't have a current process.
1983    if (!m_current_process ||
1984        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID))
1985      return SendErrorResponse(68);
1986  
1987    // Ensure we have a thread.
1988    NativeThreadProtocol *thread = m_current_process->GetThreadAtIndex(0);
1989    if (!thread)
1990      return SendErrorResponse(69);
1991  
1992    // Get the register context for the first thread.
1993    NativeRegisterContext &reg_context = thread->GetRegisterContext();
1994  
1995    // Parse out the register number from the request.
1996    packet.SetFilePos(strlen("qRegisterInfo"));
1997    const uint32_t reg_index =
1998        packet.GetHexMaxU32(false, std::numeric_limits<uint32_t>::max());
1999    if (reg_index == std::numeric_limits<uint32_t>::max())
2000      return SendErrorResponse(69);
2001  
2002    // Return the end of registers response if we've iterated one past the end of
2003    // the register set.
2004    if (reg_index >= reg_context.GetUserRegisterCount())
2005      return SendErrorResponse(69);
2006  
2007    const RegisterInfo *reg_info = reg_context.GetRegisterInfoAtIndex(reg_index);
2008    if (!reg_info)
2009      return SendErrorResponse(69);
2010  
2011    // Build the reginfos response.
2012    StreamGDBRemote response;
2013  
2014    response.PutCString("name:");
2015    response.PutCString(reg_info->name);
2016    response.PutChar(';');
2017  
2018    if (reg_info->alt_name && reg_info->alt_name[0]) {
2019      response.PutCString("alt-name:");
2020      response.PutCString(reg_info->alt_name);
2021      response.PutChar(';');
2022    }
2023  
2024    response.Printf("bitsize:%" PRIu32 ";", reg_info->byte_size * 8);
2025  
2026    if (!reg_context.RegisterOffsetIsDynamic())
2027      response.Printf("offset:%" PRIu32 ";", reg_info->byte_offset);
2028  
2029    llvm::StringRef encoding = GetEncodingNameOrEmpty(*reg_info);
2030    if (!encoding.empty())
2031      response << "encoding:" << encoding << ';';
2032  
2033    llvm::StringRef format = GetFormatNameOrEmpty(*reg_info);
2034    if (!format.empty())
2035      response << "format:" << format << ';';
2036  
2037    const char *const register_set_name =
2038        reg_context.GetRegisterSetNameForRegisterAtIndex(reg_index);
2039    if (register_set_name)
2040      response << "set:" << register_set_name << ';';
2041  
2042    if (reg_info->kinds[RegisterKind::eRegisterKindEHFrame] !=
2043        LLDB_INVALID_REGNUM)
2044      response.Printf("ehframe:%" PRIu32 ";",
2045                      reg_info->kinds[RegisterKind::eRegisterKindEHFrame]);
2046  
2047    if (reg_info->kinds[RegisterKind::eRegisterKindDWARF] != LLDB_INVALID_REGNUM)
2048      response.Printf("dwarf:%" PRIu32 ";",
2049                      reg_info->kinds[RegisterKind::eRegisterKindDWARF]);
2050  
2051    llvm::StringRef kind_generic = GetKindGenericOrEmpty(*reg_info);
2052    if (!kind_generic.empty())
2053      response << "generic:" << kind_generic << ';';
2054  
2055    if (reg_info->value_regs && reg_info->value_regs[0] != LLDB_INVALID_REGNUM) {
2056      response.PutCString("container-regs:");
2057      CollectRegNums(reg_info->value_regs, response, true);
2058      response.PutChar(';');
2059    }
2060  
2061    if (reg_info->invalidate_regs && reg_info->invalidate_regs[0]) {
2062      response.PutCString("invalidate-regs:");
2063      CollectRegNums(reg_info->invalidate_regs, response, true);
2064      response.PutChar(';');
2065    }
2066  
2067    return SendPacketNoLock(response.GetString());
2068  }
2069  
AddProcessThreads(StreamGDBRemote & response,NativeProcessProtocol & process,bool & had_any)2070  void GDBRemoteCommunicationServerLLGS::AddProcessThreads(
2071      StreamGDBRemote &response, NativeProcessProtocol &process, bool &had_any) {
2072    Log *log = GetLog(LLDBLog::Thread);
2073  
2074    lldb::pid_t pid = process.GetID();
2075    if (pid == LLDB_INVALID_PROCESS_ID)
2076      return;
2077  
2078    LLDB_LOG(log, "iterating over threads of process {0}", process.GetID());
2079    for (NativeThreadProtocol &thread : process.Threads()) {
2080      LLDB_LOG(log, "iterated thread tid={0}", thread.GetID());
2081      response.PutChar(had_any ? ',' : 'm');
2082      AppendThreadIDToResponse(response, pid, thread.GetID());
2083      had_any = true;
2084    }
2085  }
2086  
2087  GDBRemoteCommunication::PacketResult
Handle_qfThreadInfo(StringExtractorGDBRemote & packet)2088  GDBRemoteCommunicationServerLLGS::Handle_qfThreadInfo(
2089      StringExtractorGDBRemote &packet) {
2090    assert(m_debugged_processes.size() <= 1 ||
2091           bool(m_extensions_supported &
2092                NativeProcessProtocol::Extension::multiprocess));
2093  
2094    bool had_any = false;
2095    StreamGDBRemote response;
2096  
2097    for (auto &pid_ptr : m_debugged_processes)
2098      AddProcessThreads(response, *pid_ptr.second.process_up, had_any);
2099  
2100    if (!had_any)
2101      return SendOKResponse();
2102    return SendPacketNoLock(response.GetString());
2103  }
2104  
2105  GDBRemoteCommunication::PacketResult
Handle_qsThreadInfo(StringExtractorGDBRemote & packet)2106  GDBRemoteCommunicationServerLLGS::Handle_qsThreadInfo(
2107      StringExtractorGDBRemote &packet) {
2108    // FIXME for now we return the full thread list in the initial packet and
2109    // always do nothing here.
2110    return SendPacketNoLock("l");
2111  }
2112  
2113  GDBRemoteCommunication::PacketResult
Handle_g(StringExtractorGDBRemote & packet)2114  GDBRemoteCommunicationServerLLGS::Handle_g(StringExtractorGDBRemote &packet) {
2115    Log *log = GetLog(LLDBLog::Thread);
2116  
2117    // Move past packet name.
2118    packet.SetFilePos(strlen("g"));
2119  
2120    // Get the thread to use.
2121    NativeThreadProtocol *thread = GetThreadFromSuffix(packet);
2122    if (!thread) {
2123      LLDB_LOG(log, "failed, no thread available");
2124      return SendErrorResponse(0x15);
2125    }
2126  
2127    // Get the thread's register context.
2128    NativeRegisterContext &reg_ctx = thread->GetRegisterContext();
2129  
2130    std::vector<uint8_t> regs_buffer;
2131    for (uint32_t reg_num = 0; reg_num < reg_ctx.GetUserRegisterCount();
2132         ++reg_num) {
2133      const RegisterInfo *reg_info = reg_ctx.GetRegisterInfoAtIndex(reg_num);
2134  
2135      if (reg_info == nullptr) {
2136        LLDB_LOG(log, "failed to get register info for register index {0}",
2137                 reg_num);
2138        return SendErrorResponse(0x15);
2139      }
2140  
2141      if (reg_info->value_regs != nullptr)
2142        continue; // skip registers that are contained in other registers
2143  
2144      RegisterValue reg_value;
2145      Status error = reg_ctx.ReadRegister(reg_info, reg_value);
2146      if (error.Fail()) {
2147        LLDB_LOG(log, "failed to read register at index {0}", reg_num);
2148        return SendErrorResponse(0x15);
2149      }
2150  
2151      if (reg_info->byte_offset + reg_info->byte_size >= regs_buffer.size())
2152        // Resize the buffer to guarantee it can store the register offsetted
2153        // data.
2154        regs_buffer.resize(reg_info->byte_offset + reg_info->byte_size);
2155  
2156      // Copy the register offsetted data to the buffer.
2157      memcpy(regs_buffer.data() + reg_info->byte_offset, reg_value.GetBytes(),
2158             reg_info->byte_size);
2159    }
2160  
2161    // Write the response.
2162    StreamGDBRemote response;
2163    response.PutBytesAsRawHex8(regs_buffer.data(), regs_buffer.size());
2164  
2165    return SendPacketNoLock(response.GetString());
2166  }
2167  
2168  GDBRemoteCommunication::PacketResult
Handle_p(StringExtractorGDBRemote & packet)2169  GDBRemoteCommunicationServerLLGS::Handle_p(StringExtractorGDBRemote &packet) {
2170    Log *log = GetLog(LLDBLog::Thread);
2171  
2172    // Parse out the register number from the request.
2173    packet.SetFilePos(strlen("p"));
2174    const uint32_t reg_index =
2175        packet.GetHexMaxU32(false, std::numeric_limits<uint32_t>::max());
2176    if (reg_index == std::numeric_limits<uint32_t>::max()) {
2177      LLDB_LOGF(log,
2178                "GDBRemoteCommunicationServerLLGS::%s failed, could not "
2179                "parse register number from request \"%s\"",
2180                __FUNCTION__, packet.GetStringRef().data());
2181      return SendErrorResponse(0x15);
2182    }
2183  
2184    // Get the thread to use.
2185    NativeThreadProtocol *thread = GetThreadFromSuffix(packet);
2186    if (!thread) {
2187      LLDB_LOG(log, "failed, no thread available");
2188      return SendErrorResponse(0x15);
2189    }
2190  
2191    // Get the thread's register context.
2192    NativeRegisterContext &reg_context = thread->GetRegisterContext();
2193  
2194    // Return the end of registers response if we've iterated one past the end of
2195    // the register set.
2196    if (reg_index >= reg_context.GetUserRegisterCount()) {
2197      LLDB_LOGF(log,
2198                "GDBRemoteCommunicationServerLLGS::%s failed, requested "
2199                "register %" PRIu32 " beyond register count %" PRIu32,
2200                __FUNCTION__, reg_index, reg_context.GetUserRegisterCount());
2201      return SendErrorResponse(0x15);
2202    }
2203  
2204    const RegisterInfo *reg_info = reg_context.GetRegisterInfoAtIndex(reg_index);
2205    if (!reg_info) {
2206      LLDB_LOGF(log,
2207                "GDBRemoteCommunicationServerLLGS::%s failed, requested "
2208                "register %" PRIu32 " returned NULL",
2209                __FUNCTION__, reg_index);
2210      return SendErrorResponse(0x15);
2211    }
2212  
2213    // Build the reginfos response.
2214    StreamGDBRemote response;
2215  
2216    // Retrieve the value
2217    RegisterValue reg_value;
2218    Status error = reg_context.ReadRegister(reg_info, reg_value);
2219    if (error.Fail()) {
2220      LLDB_LOGF(log,
2221                "GDBRemoteCommunicationServerLLGS::%s failed, read of "
2222                "requested register %" PRIu32 " (%s) failed: %s",
2223                __FUNCTION__, reg_index, reg_info->name, error.AsCString());
2224      return SendErrorResponse(0x15);
2225    }
2226  
2227    const uint8_t *const data =
2228        static_cast<const uint8_t *>(reg_value.GetBytes());
2229    if (!data) {
2230      LLDB_LOGF(log,
2231                "GDBRemoteCommunicationServerLLGS::%s failed to get data "
2232                "bytes from requested register %" PRIu32,
2233                __FUNCTION__, reg_index);
2234      return SendErrorResponse(0x15);
2235    }
2236  
2237    // FIXME flip as needed to get data in big/little endian format for this host.
2238    for (uint32_t i = 0; i < reg_value.GetByteSize(); ++i)
2239      response.PutHex8(data[i]);
2240  
2241    return SendPacketNoLock(response.GetString());
2242  }
2243  
2244  GDBRemoteCommunication::PacketResult
Handle_P(StringExtractorGDBRemote & packet)2245  GDBRemoteCommunicationServerLLGS::Handle_P(StringExtractorGDBRemote &packet) {
2246    Log *log = GetLog(LLDBLog::Thread);
2247  
2248    // Ensure there is more content.
2249    if (packet.GetBytesLeft() < 1)
2250      return SendIllFormedResponse(packet, "Empty P packet");
2251  
2252    // Parse out the register number from the request.
2253    packet.SetFilePos(strlen("P"));
2254    const uint32_t reg_index =
2255        packet.GetHexMaxU32(false, std::numeric_limits<uint32_t>::max());
2256    if (reg_index == std::numeric_limits<uint32_t>::max()) {
2257      LLDB_LOGF(log,
2258                "GDBRemoteCommunicationServerLLGS::%s failed, could not "
2259                "parse register number from request \"%s\"",
2260                __FUNCTION__, packet.GetStringRef().data());
2261      return SendErrorResponse(0x29);
2262    }
2263  
2264    // Note debugserver would send an E30 here.
2265    if ((packet.GetBytesLeft() < 1) || (packet.GetChar() != '='))
2266      return SendIllFormedResponse(
2267          packet, "P packet missing '=' char after register number");
2268  
2269    // Parse out the value.
2270    size_t reg_size = packet.GetHexBytesAvail(m_reg_bytes);
2271  
2272    // Get the thread to use.
2273    NativeThreadProtocol *thread = GetThreadFromSuffix(packet);
2274    if (!thread) {
2275      LLDB_LOGF(log,
2276                "GDBRemoteCommunicationServerLLGS::%s failed, no thread "
2277                "available (thread index 0)",
2278                __FUNCTION__);
2279      return SendErrorResponse(0x28);
2280    }
2281  
2282    // Get the thread's register context.
2283    NativeRegisterContext &reg_context = thread->GetRegisterContext();
2284    const RegisterInfo *reg_info = reg_context.GetRegisterInfoAtIndex(reg_index);
2285    if (!reg_info) {
2286      LLDB_LOGF(log,
2287                "GDBRemoteCommunicationServerLLGS::%s failed, requested "
2288                "register %" PRIu32 " returned NULL",
2289                __FUNCTION__, reg_index);
2290      return SendErrorResponse(0x48);
2291    }
2292  
2293    // Return the end of registers response if we've iterated one past the end of
2294    // the register set.
2295    if (reg_index >= reg_context.GetUserRegisterCount()) {
2296      LLDB_LOGF(log,
2297                "GDBRemoteCommunicationServerLLGS::%s failed, requested "
2298                "register %" PRIu32 " beyond register count %" PRIu32,
2299                __FUNCTION__, reg_index, reg_context.GetUserRegisterCount());
2300      return SendErrorResponse(0x47);
2301    }
2302  
2303    if (reg_size != reg_info->byte_size)
2304      return SendIllFormedResponse(packet, "P packet register size is incorrect");
2305  
2306    // Build the reginfos response.
2307    StreamGDBRemote response;
2308  
2309    RegisterValue reg_value(ArrayRef<uint8_t>(m_reg_bytes, reg_size),
2310                            m_current_process->GetArchitecture().GetByteOrder());
2311    Status error = reg_context.WriteRegister(reg_info, reg_value);
2312    if (error.Fail()) {
2313      LLDB_LOGF(log,
2314                "GDBRemoteCommunicationServerLLGS::%s failed, write of "
2315                "requested register %" PRIu32 " (%s) failed: %s",
2316                __FUNCTION__, reg_index, reg_info->name, error.AsCString());
2317      return SendErrorResponse(0x32);
2318    }
2319  
2320    return SendOKResponse();
2321  }
2322  
2323  GDBRemoteCommunication::PacketResult
Handle_H(StringExtractorGDBRemote & packet)2324  GDBRemoteCommunicationServerLLGS::Handle_H(StringExtractorGDBRemote &packet) {
2325    Log *log = GetLog(LLDBLog::Thread);
2326  
2327    // Parse out which variant of $H is requested.
2328    packet.SetFilePos(strlen("H"));
2329    if (packet.GetBytesLeft() < 1) {
2330      LLDB_LOGF(log,
2331                "GDBRemoteCommunicationServerLLGS::%s failed, H command "
2332                "missing {g,c} variant",
2333                __FUNCTION__);
2334      return SendIllFormedResponse(packet, "H command missing {g,c} variant");
2335    }
2336  
2337    const char h_variant = packet.GetChar();
2338    NativeProcessProtocol *default_process;
2339    switch (h_variant) {
2340    case 'g':
2341      default_process = m_current_process;
2342      break;
2343  
2344    case 'c':
2345      default_process = m_continue_process;
2346      break;
2347  
2348    default:
2349      LLDB_LOGF(
2350          log,
2351          "GDBRemoteCommunicationServerLLGS::%s failed, invalid $H variant %c",
2352          __FUNCTION__, h_variant);
2353      return SendIllFormedResponse(packet,
2354                                   "H variant unsupported, should be c or g");
2355    }
2356  
2357    // Parse out the thread number.
2358    auto pid_tid = packet.GetPidTid(default_process ? default_process->GetID()
2359                                                    : LLDB_INVALID_PROCESS_ID);
2360    if (!pid_tid)
2361      return SendErrorResponse(llvm::make_error<StringError>(
2362          inconvertibleErrorCode(), "Malformed thread-id"));
2363  
2364    lldb::pid_t pid = pid_tid->first;
2365    lldb::tid_t tid = pid_tid->second;
2366  
2367    if (pid == StringExtractorGDBRemote::AllProcesses)
2368      return SendUnimplementedResponse("Selecting all processes not supported");
2369    if (pid == LLDB_INVALID_PROCESS_ID)
2370      return SendErrorResponse(llvm::make_error<StringError>(
2371          inconvertibleErrorCode(), "No current process and no PID provided"));
2372  
2373    // Check the process ID and find respective process instance.
2374    auto new_process_it = m_debugged_processes.find(pid);
2375    if (new_process_it == m_debugged_processes.end())
2376      return SendErrorResponse(llvm::make_error<StringError>(
2377          inconvertibleErrorCode(),
2378          llvm::formatv("No process with PID {0} debugged", pid)));
2379  
2380    // Ensure we have the given thread when not specifying -1 (all threads) or 0
2381    // (any thread).
2382    if (tid != LLDB_INVALID_THREAD_ID && tid != 0) {
2383      NativeThreadProtocol *thread =
2384          new_process_it->second.process_up->GetThreadByID(tid);
2385      if (!thread) {
2386        LLDB_LOGF(log,
2387                  "GDBRemoteCommunicationServerLLGS::%s failed, tid %" PRIu64
2388                  " not found",
2389                  __FUNCTION__, tid);
2390        return SendErrorResponse(0x15);
2391      }
2392    }
2393  
2394    // Now switch the given process and thread type.
2395    switch (h_variant) {
2396    case 'g':
2397      m_current_process = new_process_it->second.process_up.get();
2398      SetCurrentThreadID(tid);
2399      break;
2400  
2401    case 'c':
2402      m_continue_process = new_process_it->second.process_up.get();
2403      SetContinueThreadID(tid);
2404      break;
2405  
2406    default:
2407      assert(false && "unsupported $H variant - shouldn't get here");
2408      return SendIllFormedResponse(packet,
2409                                   "H variant unsupported, should be c or g");
2410    }
2411  
2412    return SendOKResponse();
2413  }
2414  
2415  GDBRemoteCommunication::PacketResult
Handle_I(StringExtractorGDBRemote & packet)2416  GDBRemoteCommunicationServerLLGS::Handle_I(StringExtractorGDBRemote &packet) {
2417    Log *log = GetLog(LLDBLog::Thread);
2418  
2419    // Fail if we don't have a current process.
2420    if (!m_current_process ||
2421        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
2422      LLDB_LOGF(
2423          log,
2424          "GDBRemoteCommunicationServerLLGS::%s failed, no process available",
2425          __FUNCTION__);
2426      return SendErrorResponse(0x15);
2427    }
2428  
2429    packet.SetFilePos(::strlen("I"));
2430    uint8_t tmp[4096];
2431    for (;;) {
2432      size_t read = packet.GetHexBytesAvail(tmp);
2433      if (read == 0) {
2434        break;
2435      }
2436      // write directly to stdin *this might block if stdin buffer is full*
2437      // TODO: enqueue this block in circular buffer and send window size to
2438      // remote host
2439      ConnectionStatus status;
2440      Status error;
2441      m_stdio_communication.WriteAll(tmp, read, status, &error);
2442      if (error.Fail()) {
2443        return SendErrorResponse(0x15);
2444      }
2445    }
2446  
2447    return SendOKResponse();
2448  }
2449  
2450  GDBRemoteCommunication::PacketResult
Handle_interrupt(StringExtractorGDBRemote & packet)2451  GDBRemoteCommunicationServerLLGS::Handle_interrupt(
2452      StringExtractorGDBRemote &packet) {
2453    Log *log = GetLog(LLDBLog::Process | LLDBLog::Thread);
2454  
2455    // Fail if we don't have a current process.
2456    if (!m_current_process ||
2457        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
2458      LLDB_LOG(log, "failed, no process available");
2459      return SendErrorResponse(0x15);
2460    }
2461  
2462    // Interrupt the process.
2463    Status error = m_current_process->Interrupt();
2464    if (error.Fail()) {
2465      LLDB_LOG(log, "failed for process {0}: {1}", m_current_process->GetID(),
2466               error);
2467      return SendErrorResponse(GDBRemoteServerError::eErrorResume);
2468    }
2469  
2470    LLDB_LOG(log, "stopped process {0}", m_current_process->GetID());
2471  
2472    // No response required from stop all.
2473    return PacketResult::Success;
2474  }
2475  
2476  GDBRemoteCommunication::PacketResult
Handle_memory_read(StringExtractorGDBRemote & packet)2477  GDBRemoteCommunicationServerLLGS::Handle_memory_read(
2478      StringExtractorGDBRemote &packet) {
2479    Log *log = GetLog(LLDBLog::Process);
2480  
2481    if (!m_current_process ||
2482        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
2483      LLDB_LOGF(
2484          log,
2485          "GDBRemoteCommunicationServerLLGS::%s failed, no process available",
2486          __FUNCTION__);
2487      return SendErrorResponse(0x15);
2488    }
2489  
2490    // Parse out the memory address.
2491    packet.SetFilePos(strlen("m"));
2492    if (packet.GetBytesLeft() < 1)
2493      return SendIllFormedResponse(packet, "Too short m packet");
2494  
2495    // Read the address.  Punting on validation.
2496    // FIXME replace with Hex U64 read with no default value that fails on failed
2497    // read.
2498    const lldb::addr_t read_addr = packet.GetHexMaxU64(false, 0);
2499  
2500    // Validate comma.
2501    if ((packet.GetBytesLeft() < 1) || (packet.GetChar() != ','))
2502      return SendIllFormedResponse(packet, "Comma sep missing in m packet");
2503  
2504    // Get # bytes to read.
2505    if (packet.GetBytesLeft() < 1)
2506      return SendIllFormedResponse(packet, "Length missing in m packet");
2507  
2508    const uint64_t byte_count = packet.GetHexMaxU64(false, 0);
2509    if (byte_count == 0) {
2510      LLDB_LOGF(log,
2511                "GDBRemoteCommunicationServerLLGS::%s nothing to read: "
2512                "zero-length packet",
2513                __FUNCTION__);
2514      return SendOKResponse();
2515    }
2516  
2517    // Allocate the response buffer.
2518    std::string buf(byte_count, '\0');
2519    if (buf.empty())
2520      return SendErrorResponse(0x78);
2521  
2522    // Retrieve the process memory.
2523    size_t bytes_read = 0;
2524    Status error = m_current_process->ReadMemoryWithoutTrap(
2525        read_addr, &buf[0], byte_count, bytes_read);
2526    if (error.Fail()) {
2527      LLDB_LOGF(log,
2528                "GDBRemoteCommunicationServerLLGS::%s pid %" PRIu64
2529                " mem 0x%" PRIx64 ": failed to read. Error: %s",
2530                __FUNCTION__, m_current_process->GetID(), read_addr,
2531                error.AsCString());
2532      return SendErrorResponse(0x08);
2533    }
2534  
2535    if (bytes_read == 0) {
2536      LLDB_LOGF(log,
2537                "GDBRemoteCommunicationServerLLGS::%s pid %" PRIu64
2538                " mem 0x%" PRIx64 ": read 0 of %" PRIu64 " requested bytes",
2539                __FUNCTION__, m_current_process->GetID(), read_addr, byte_count);
2540      return SendErrorResponse(0x08);
2541    }
2542  
2543    StreamGDBRemote response;
2544    packet.SetFilePos(0);
2545    char kind = packet.GetChar('?');
2546    if (kind == 'x')
2547      response.PutEscapedBytes(buf.data(), byte_count);
2548    else {
2549      assert(kind == 'm');
2550      for (size_t i = 0; i < bytes_read; ++i)
2551        response.PutHex8(buf[i]);
2552    }
2553  
2554    return SendPacketNoLock(response.GetString());
2555  }
2556  
2557  GDBRemoteCommunication::PacketResult
Handle__M(StringExtractorGDBRemote & packet)2558  GDBRemoteCommunicationServerLLGS::Handle__M(StringExtractorGDBRemote &packet) {
2559    Log *log = GetLog(LLDBLog::Process);
2560  
2561    if (!m_current_process ||
2562        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
2563      LLDB_LOGF(
2564          log,
2565          "GDBRemoteCommunicationServerLLGS::%s failed, no process available",
2566          __FUNCTION__);
2567      return SendErrorResponse(0x15);
2568    }
2569  
2570    // Parse out the memory address.
2571    packet.SetFilePos(strlen("_M"));
2572    if (packet.GetBytesLeft() < 1)
2573      return SendIllFormedResponse(packet, "Too short _M packet");
2574  
2575    const lldb::addr_t size = packet.GetHexMaxU64(false, LLDB_INVALID_ADDRESS);
2576    if (size == LLDB_INVALID_ADDRESS)
2577      return SendIllFormedResponse(packet, "Address not valid");
2578    if (packet.GetChar() != ',')
2579      return SendIllFormedResponse(packet, "Bad packet");
2580    Permissions perms = {};
2581    while (packet.GetBytesLeft() > 0) {
2582      switch (packet.GetChar()) {
2583      case 'r':
2584        perms |= ePermissionsReadable;
2585        break;
2586      case 'w':
2587        perms |= ePermissionsWritable;
2588        break;
2589      case 'x':
2590        perms |= ePermissionsExecutable;
2591        break;
2592      default:
2593        return SendIllFormedResponse(packet, "Bad permissions");
2594      }
2595    }
2596  
2597    llvm::Expected<addr_t> addr = m_current_process->AllocateMemory(size, perms);
2598    if (!addr)
2599      return SendErrorResponse(addr.takeError());
2600  
2601    StreamGDBRemote response;
2602    response.PutHex64(*addr);
2603    return SendPacketNoLock(response.GetString());
2604  }
2605  
2606  GDBRemoteCommunication::PacketResult
Handle__m(StringExtractorGDBRemote & packet)2607  GDBRemoteCommunicationServerLLGS::Handle__m(StringExtractorGDBRemote &packet) {
2608    Log *log = GetLog(LLDBLog::Process);
2609  
2610    if (!m_current_process ||
2611        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
2612      LLDB_LOGF(
2613          log,
2614          "GDBRemoteCommunicationServerLLGS::%s failed, no process available",
2615          __FUNCTION__);
2616      return SendErrorResponse(0x15);
2617    }
2618  
2619    // Parse out the memory address.
2620    packet.SetFilePos(strlen("_m"));
2621    if (packet.GetBytesLeft() < 1)
2622      return SendIllFormedResponse(packet, "Too short m packet");
2623  
2624    const lldb::addr_t addr = packet.GetHexMaxU64(false, LLDB_INVALID_ADDRESS);
2625    if (addr == LLDB_INVALID_ADDRESS)
2626      return SendIllFormedResponse(packet, "Address not valid");
2627  
2628    if (llvm::Error Err = m_current_process->DeallocateMemory(addr))
2629      return SendErrorResponse(std::move(Err));
2630  
2631    return SendOKResponse();
2632  }
2633  
2634  GDBRemoteCommunication::PacketResult
Handle_M(StringExtractorGDBRemote & packet)2635  GDBRemoteCommunicationServerLLGS::Handle_M(StringExtractorGDBRemote &packet) {
2636    Log *log = GetLog(LLDBLog::Process);
2637  
2638    if (!m_current_process ||
2639        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
2640      LLDB_LOGF(
2641          log,
2642          "GDBRemoteCommunicationServerLLGS::%s failed, no process available",
2643          __FUNCTION__);
2644      return SendErrorResponse(0x15);
2645    }
2646  
2647    // Parse out the memory address.
2648    packet.SetFilePos(strlen("M"));
2649    if (packet.GetBytesLeft() < 1)
2650      return SendIllFormedResponse(packet, "Too short M packet");
2651  
2652    // Read the address.  Punting on validation.
2653    // FIXME replace with Hex U64 read with no default value that fails on failed
2654    // read.
2655    const lldb::addr_t write_addr = packet.GetHexMaxU64(false, 0);
2656  
2657    // Validate comma.
2658    if ((packet.GetBytesLeft() < 1) || (packet.GetChar() != ','))
2659      return SendIllFormedResponse(packet, "Comma sep missing in M packet");
2660  
2661    // Get # bytes to read.
2662    if (packet.GetBytesLeft() < 1)
2663      return SendIllFormedResponse(packet, "Length missing in M packet");
2664  
2665    const uint64_t byte_count = packet.GetHexMaxU64(false, 0);
2666    if (byte_count == 0) {
2667      LLDB_LOG(log, "nothing to write: zero-length packet");
2668      return PacketResult::Success;
2669    }
2670  
2671    // Validate colon.
2672    if ((packet.GetBytesLeft() < 1) || (packet.GetChar() != ':'))
2673      return SendIllFormedResponse(
2674          packet, "Comma sep missing in M packet after byte length");
2675  
2676    // Allocate the conversion buffer.
2677    std::vector<uint8_t> buf(byte_count, 0);
2678    if (buf.empty())
2679      return SendErrorResponse(0x78);
2680  
2681    // Convert the hex memory write contents to bytes.
2682    StreamGDBRemote response;
2683    const uint64_t convert_count = packet.GetHexBytes(buf, 0);
2684    if (convert_count != byte_count) {
2685      LLDB_LOG(log,
2686               "pid {0} mem {1:x}: asked to write {2} bytes, but only found {3} "
2687               "to convert.",
2688               m_current_process->GetID(), write_addr, byte_count, convert_count);
2689      return SendIllFormedResponse(packet, "M content byte length specified did "
2690                                           "not match hex-encoded content "
2691                                           "length");
2692    }
2693  
2694    // Write the process memory.
2695    size_t bytes_written = 0;
2696    Status error = m_current_process->WriteMemory(write_addr, &buf[0], byte_count,
2697                                                  bytes_written);
2698    if (error.Fail()) {
2699      LLDB_LOG(log, "pid {0} mem {1:x}: failed to write. Error: {2}",
2700               m_current_process->GetID(), write_addr, error);
2701      return SendErrorResponse(0x09);
2702    }
2703  
2704    if (bytes_written == 0) {
2705      LLDB_LOG(log, "pid {0} mem {1:x}: wrote 0 of {2} requested bytes",
2706               m_current_process->GetID(), write_addr, byte_count);
2707      return SendErrorResponse(0x09);
2708    }
2709  
2710    return SendOKResponse();
2711  }
2712  
2713  GDBRemoteCommunication::PacketResult
Handle_qMemoryRegionInfoSupported(StringExtractorGDBRemote & packet)2714  GDBRemoteCommunicationServerLLGS::Handle_qMemoryRegionInfoSupported(
2715      StringExtractorGDBRemote &packet) {
2716    Log *log = GetLog(LLDBLog::Process);
2717  
2718    // Currently only the NativeProcessProtocol knows if it can handle a
2719    // qMemoryRegionInfoSupported request, but we're not guaranteed to be
2720    // attached to a process.  For now we'll assume the client only asks this
2721    // when a process is being debugged.
2722  
2723    // Ensure we have a process running; otherwise, we can't figure this out
2724    // since we won't have a NativeProcessProtocol.
2725    if (!m_current_process ||
2726        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
2727      LLDB_LOGF(
2728          log,
2729          "GDBRemoteCommunicationServerLLGS::%s failed, no process available",
2730          __FUNCTION__);
2731      return SendErrorResponse(0x15);
2732    }
2733  
2734    // Test if we can get any region back when asking for the region around NULL.
2735    MemoryRegionInfo region_info;
2736    const Status error = m_current_process->GetMemoryRegionInfo(0, region_info);
2737    if (error.Fail()) {
2738      // We don't support memory region info collection for this
2739      // NativeProcessProtocol.
2740      return SendUnimplementedResponse("");
2741    }
2742  
2743    return SendOKResponse();
2744  }
2745  
2746  GDBRemoteCommunication::PacketResult
Handle_qMemoryRegionInfo(StringExtractorGDBRemote & packet)2747  GDBRemoteCommunicationServerLLGS::Handle_qMemoryRegionInfo(
2748      StringExtractorGDBRemote &packet) {
2749    Log *log = GetLog(LLDBLog::Process);
2750  
2751    // Ensure we have a process.
2752    if (!m_current_process ||
2753        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
2754      LLDB_LOGF(
2755          log,
2756          "GDBRemoteCommunicationServerLLGS::%s failed, no process available",
2757          __FUNCTION__);
2758      return SendErrorResponse(0x15);
2759    }
2760  
2761    // Parse out the memory address.
2762    packet.SetFilePos(strlen("qMemoryRegionInfo:"));
2763    if (packet.GetBytesLeft() < 1)
2764      return SendIllFormedResponse(packet, "Too short qMemoryRegionInfo: packet");
2765  
2766    // Read the address.  Punting on validation.
2767    const lldb::addr_t read_addr = packet.GetHexMaxU64(false, 0);
2768  
2769    StreamGDBRemote response;
2770  
2771    // Get the memory region info for the target address.
2772    MemoryRegionInfo region_info;
2773    const Status error =
2774        m_current_process->GetMemoryRegionInfo(read_addr, region_info);
2775    if (error.Fail()) {
2776      // Return the error message.
2777  
2778      response.PutCString("error:");
2779      response.PutStringAsRawHex8(error.AsCString());
2780      response.PutChar(';');
2781    } else {
2782      // Range start and size.
2783      response.Printf("start:%" PRIx64 ";size:%" PRIx64 ";",
2784                      region_info.GetRange().GetRangeBase(),
2785                      region_info.GetRange().GetByteSize());
2786  
2787      // Permissions.
2788      if (region_info.GetReadable() || region_info.GetWritable() ||
2789          region_info.GetExecutable()) {
2790        // Write permissions info.
2791        response.PutCString("permissions:");
2792  
2793        if (region_info.GetReadable())
2794          response.PutChar('r');
2795        if (region_info.GetWritable())
2796          response.PutChar('w');
2797        if (region_info.GetExecutable())
2798          response.PutChar('x');
2799  
2800        response.PutChar(';');
2801      }
2802  
2803      // Flags
2804      MemoryRegionInfo::OptionalBool memory_tagged =
2805          region_info.GetMemoryTagged();
2806      if (memory_tagged != MemoryRegionInfo::eDontKnow) {
2807        response.PutCString("flags:");
2808        if (memory_tagged == MemoryRegionInfo::eYes) {
2809          response.PutCString("mt");
2810        }
2811        response.PutChar(';');
2812      }
2813  
2814      // Name
2815      ConstString name = region_info.GetName();
2816      if (name) {
2817        response.PutCString("name:");
2818        response.PutStringAsRawHex8(name.GetStringRef());
2819        response.PutChar(';');
2820      }
2821    }
2822  
2823    return SendPacketNoLock(response.GetString());
2824  }
2825  
2826  GDBRemoteCommunication::PacketResult
Handle_Z(StringExtractorGDBRemote & packet)2827  GDBRemoteCommunicationServerLLGS::Handle_Z(StringExtractorGDBRemote &packet) {
2828    // Ensure we have a process.
2829    if (!m_current_process ||
2830        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
2831      Log *log = GetLog(LLDBLog::Process);
2832      LLDB_LOG(log, "failed, no process available");
2833      return SendErrorResponse(0x15);
2834    }
2835  
2836    // Parse out software or hardware breakpoint or watchpoint requested.
2837    packet.SetFilePos(strlen("Z"));
2838    if (packet.GetBytesLeft() < 1)
2839      return SendIllFormedResponse(
2840          packet, "Too short Z packet, missing software/hardware specifier");
2841  
2842    bool want_breakpoint = true;
2843    bool want_hardware = false;
2844    uint32_t watch_flags = 0;
2845  
2846    const GDBStoppointType stoppoint_type =
2847        GDBStoppointType(packet.GetS32(eStoppointInvalid));
2848    switch (stoppoint_type) {
2849    case eBreakpointSoftware:
2850      want_hardware = false;
2851      want_breakpoint = true;
2852      break;
2853    case eBreakpointHardware:
2854      want_hardware = true;
2855      want_breakpoint = true;
2856      break;
2857    case eWatchpointWrite:
2858      watch_flags = 1;
2859      want_hardware = true;
2860      want_breakpoint = false;
2861      break;
2862    case eWatchpointRead:
2863      watch_flags = 2;
2864      want_hardware = true;
2865      want_breakpoint = false;
2866      break;
2867    case eWatchpointReadWrite:
2868      watch_flags = 3;
2869      want_hardware = true;
2870      want_breakpoint = false;
2871      break;
2872    case eStoppointInvalid:
2873      return SendIllFormedResponse(
2874          packet, "Z packet had invalid software/hardware specifier");
2875    }
2876  
2877    if ((packet.GetBytesLeft() < 1) || packet.GetChar() != ',')
2878      return SendIllFormedResponse(
2879          packet, "Malformed Z packet, expecting comma after stoppoint type");
2880  
2881    // Parse out the stoppoint address.
2882    if (packet.GetBytesLeft() < 1)
2883      return SendIllFormedResponse(packet, "Too short Z packet, missing address");
2884    const lldb::addr_t addr = packet.GetHexMaxU64(false, 0);
2885  
2886    if ((packet.GetBytesLeft() < 1) || packet.GetChar() != ',')
2887      return SendIllFormedResponse(
2888          packet, "Malformed Z packet, expecting comma after address");
2889  
2890    // Parse out the stoppoint size (i.e. size hint for opcode size).
2891    const uint32_t size =
2892        packet.GetHexMaxU32(false, std::numeric_limits<uint32_t>::max());
2893    if (size == std::numeric_limits<uint32_t>::max())
2894      return SendIllFormedResponse(
2895          packet, "Malformed Z packet, failed to parse size argument");
2896  
2897    if (want_breakpoint) {
2898      // Try to set the breakpoint.
2899      const Status error =
2900          m_current_process->SetBreakpoint(addr, size, want_hardware);
2901      if (error.Success())
2902        return SendOKResponse();
2903      Log *log = GetLog(LLDBLog::Breakpoints);
2904      LLDB_LOG(log, "pid {0} failed to set breakpoint: {1}",
2905               m_current_process->GetID(), error);
2906      return SendErrorResponse(0x09);
2907    } else {
2908      // Try to set the watchpoint.
2909      const Status error = m_current_process->SetWatchpoint(
2910          addr, size, watch_flags, want_hardware);
2911      if (error.Success())
2912        return SendOKResponse();
2913      Log *log = GetLog(LLDBLog::Watchpoints);
2914      LLDB_LOG(log, "pid {0} failed to set watchpoint: {1}",
2915               m_current_process->GetID(), error);
2916      return SendErrorResponse(0x09);
2917    }
2918  }
2919  
2920  GDBRemoteCommunication::PacketResult
Handle_z(StringExtractorGDBRemote & packet)2921  GDBRemoteCommunicationServerLLGS::Handle_z(StringExtractorGDBRemote &packet) {
2922    // Ensure we have a process.
2923    if (!m_current_process ||
2924        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
2925      Log *log = GetLog(LLDBLog::Process);
2926      LLDB_LOG(log, "failed, no process available");
2927      return SendErrorResponse(0x15);
2928    }
2929  
2930    // Parse out software or hardware breakpoint or watchpoint requested.
2931    packet.SetFilePos(strlen("z"));
2932    if (packet.GetBytesLeft() < 1)
2933      return SendIllFormedResponse(
2934          packet, "Too short z packet, missing software/hardware specifier");
2935  
2936    bool want_breakpoint = true;
2937    bool want_hardware = false;
2938  
2939    const GDBStoppointType stoppoint_type =
2940        GDBStoppointType(packet.GetS32(eStoppointInvalid));
2941    switch (stoppoint_type) {
2942    case eBreakpointHardware:
2943      want_breakpoint = true;
2944      want_hardware = true;
2945      break;
2946    case eBreakpointSoftware:
2947      want_breakpoint = true;
2948      break;
2949    case eWatchpointWrite:
2950      want_breakpoint = false;
2951      break;
2952    case eWatchpointRead:
2953      want_breakpoint = false;
2954      break;
2955    case eWatchpointReadWrite:
2956      want_breakpoint = false;
2957      break;
2958    default:
2959      return SendIllFormedResponse(
2960          packet, "z packet had invalid software/hardware specifier");
2961    }
2962  
2963    if ((packet.GetBytesLeft() < 1) || packet.GetChar() != ',')
2964      return SendIllFormedResponse(
2965          packet, "Malformed z packet, expecting comma after stoppoint type");
2966  
2967    // Parse out the stoppoint address.
2968    if (packet.GetBytesLeft() < 1)
2969      return SendIllFormedResponse(packet, "Too short z packet, missing address");
2970    const lldb::addr_t addr = packet.GetHexMaxU64(false, 0);
2971  
2972    if ((packet.GetBytesLeft() < 1) || packet.GetChar() != ',')
2973      return SendIllFormedResponse(
2974          packet, "Malformed z packet, expecting comma after address");
2975  
2976    /*
2977    // Parse out the stoppoint size (i.e. size hint for opcode size).
2978    const uint32_t size = packet.GetHexMaxU32 (false,
2979    std::numeric_limits<uint32_t>::max ());
2980    if (size == std::numeric_limits<uint32_t>::max ())
2981        return SendIllFormedResponse(packet, "Malformed z packet, failed to parse
2982    size argument");
2983    */
2984  
2985    if (want_breakpoint) {
2986      // Try to clear the breakpoint.
2987      const Status error =
2988          m_current_process->RemoveBreakpoint(addr, want_hardware);
2989      if (error.Success())
2990        return SendOKResponse();
2991      Log *log = GetLog(LLDBLog::Breakpoints);
2992      LLDB_LOG(log, "pid {0} failed to remove breakpoint: {1}",
2993               m_current_process->GetID(), error);
2994      return SendErrorResponse(0x09);
2995    } else {
2996      // Try to clear the watchpoint.
2997      const Status error = m_current_process->RemoveWatchpoint(addr);
2998      if (error.Success())
2999        return SendOKResponse();
3000      Log *log = GetLog(LLDBLog::Watchpoints);
3001      LLDB_LOG(log, "pid {0} failed to remove watchpoint: {1}",
3002               m_current_process->GetID(), error);
3003      return SendErrorResponse(0x09);
3004    }
3005  }
3006  
3007  GDBRemoteCommunication::PacketResult
Handle_s(StringExtractorGDBRemote & packet)3008  GDBRemoteCommunicationServerLLGS::Handle_s(StringExtractorGDBRemote &packet) {
3009    Log *log = GetLog(LLDBLog::Process | LLDBLog::Thread);
3010  
3011    // Ensure we have a process.
3012    if (!m_continue_process ||
3013        (m_continue_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
3014      LLDB_LOGF(
3015          log,
3016          "GDBRemoteCommunicationServerLLGS::%s failed, no process available",
3017          __FUNCTION__);
3018      return SendErrorResponse(0x32);
3019    }
3020  
3021    // We first try to use a continue thread id.  If any one or any all set, use
3022    // the current thread. Bail out if we don't have a thread id.
3023    lldb::tid_t tid = GetContinueThreadID();
3024    if (tid == 0 || tid == LLDB_INVALID_THREAD_ID)
3025      tid = GetCurrentThreadID();
3026    if (tid == LLDB_INVALID_THREAD_ID)
3027      return SendErrorResponse(0x33);
3028  
3029    // Double check that we have such a thread.
3030    // TODO investigate: on MacOSX we might need to do an UpdateThreads () here.
3031    NativeThreadProtocol *thread = m_continue_process->GetThreadByID(tid);
3032    if (!thread)
3033      return SendErrorResponse(0x33);
3034  
3035    // Create the step action for the given thread.
3036    ResumeAction action = {tid, eStateStepping, LLDB_INVALID_SIGNAL_NUMBER};
3037  
3038    // Setup the actions list.
3039    ResumeActionList actions;
3040    actions.Append(action);
3041  
3042    // All other threads stop while we're single stepping a thread.
3043    actions.SetDefaultThreadActionIfNeeded(eStateStopped, 0);
3044  
3045    PacketResult resume_res = ResumeProcess(*m_continue_process, actions);
3046    if (resume_res != PacketResult::Success)
3047      return resume_res;
3048  
3049    // No response here, unless in non-stop mode.
3050    // Otherwise, the stop or exit will come from the resulting action.
3051    return SendContinueSuccessResponse();
3052  }
3053  
3054  llvm::Expected<std::unique_ptr<llvm::MemoryBuffer>>
BuildTargetXml()3055  GDBRemoteCommunicationServerLLGS::BuildTargetXml() {
3056    // Ensure we have a thread.
3057    NativeThreadProtocol *thread = m_current_process->GetThreadAtIndex(0);
3058    if (!thread)
3059      return llvm::createStringError(llvm::inconvertibleErrorCode(),
3060                                     "No thread available");
3061  
3062    Log *log = GetLog(LLDBLog::Process | LLDBLog::Thread);
3063    // Get the register context for the first thread.
3064    NativeRegisterContext &reg_context = thread->GetRegisterContext();
3065  
3066    StreamString response;
3067  
3068    response.Printf("<?xml version=\"1.0\"?>\n");
3069    response.Printf("<target version=\"1.0\">\n");
3070    response.IndentMore();
3071  
3072    response.Indent();
3073    response.Printf("<architecture>%s</architecture>\n",
3074                    m_current_process->GetArchitecture()
3075                        .GetTriple()
3076                        .getArchName()
3077                        .str()
3078                        .c_str());
3079  
3080    response.Indent("<feature>\n");
3081  
3082    const int registers_count = reg_context.GetUserRegisterCount();
3083    if (registers_count)
3084      response.IndentMore();
3085  
3086    llvm::StringSet<> field_enums_seen;
3087    for (int reg_index = 0; reg_index < registers_count; reg_index++) {
3088      const RegisterInfo *reg_info =
3089          reg_context.GetRegisterInfoAtIndex(reg_index);
3090  
3091      if (!reg_info) {
3092        LLDB_LOGF(log,
3093                  "%s failed to get register info for register index %" PRIu32,
3094                  "target.xml", reg_index);
3095        continue;
3096      }
3097  
3098      if (reg_info->flags_type) {
3099        response.IndentMore();
3100        reg_info->flags_type->EnumsToXML(response, field_enums_seen);
3101        reg_info->flags_type->ToXML(response);
3102        response.IndentLess();
3103      }
3104  
3105      response.Indent();
3106      response.Printf("<reg name=\"%s\" bitsize=\"%" PRIu32
3107                      "\" regnum=\"%d\" ",
3108                      reg_info->name, reg_info->byte_size * 8, reg_index);
3109  
3110      if (!reg_context.RegisterOffsetIsDynamic())
3111        response.Printf("offset=\"%" PRIu32 "\" ", reg_info->byte_offset);
3112  
3113      if (reg_info->alt_name && reg_info->alt_name[0])
3114        response.Printf("altname=\"%s\" ", reg_info->alt_name);
3115  
3116      llvm::StringRef encoding = GetEncodingNameOrEmpty(*reg_info);
3117      if (!encoding.empty())
3118        response << "encoding=\"" << encoding << "\" ";
3119  
3120      llvm::StringRef format = GetFormatNameOrEmpty(*reg_info);
3121      if (!format.empty())
3122        response << "format=\"" << format << "\" ";
3123  
3124      if (reg_info->flags_type)
3125        response << "type=\"" << reg_info->flags_type->GetID() << "\" ";
3126  
3127      const char *const register_set_name =
3128          reg_context.GetRegisterSetNameForRegisterAtIndex(reg_index);
3129      if (register_set_name)
3130        response << "group=\"" << register_set_name << "\" ";
3131  
3132      if (reg_info->kinds[RegisterKind::eRegisterKindEHFrame] !=
3133          LLDB_INVALID_REGNUM)
3134        response.Printf("ehframe_regnum=\"%" PRIu32 "\" ",
3135                        reg_info->kinds[RegisterKind::eRegisterKindEHFrame]);
3136  
3137      if (reg_info->kinds[RegisterKind::eRegisterKindDWARF] !=
3138          LLDB_INVALID_REGNUM)
3139        response.Printf("dwarf_regnum=\"%" PRIu32 "\" ",
3140                        reg_info->kinds[RegisterKind::eRegisterKindDWARF]);
3141  
3142      llvm::StringRef kind_generic = GetKindGenericOrEmpty(*reg_info);
3143      if (!kind_generic.empty())
3144        response << "generic=\"" << kind_generic << "\" ";
3145  
3146      if (reg_info->value_regs &&
3147          reg_info->value_regs[0] != LLDB_INVALID_REGNUM) {
3148        response.PutCString("value_regnums=\"");
3149        CollectRegNums(reg_info->value_regs, response, false);
3150        response.Printf("\" ");
3151      }
3152  
3153      if (reg_info->invalidate_regs && reg_info->invalidate_regs[0]) {
3154        response.PutCString("invalidate_regnums=\"");
3155        CollectRegNums(reg_info->invalidate_regs, response, false);
3156        response.Printf("\" ");
3157      }
3158  
3159      response.Printf("/>\n");
3160    }
3161  
3162    if (registers_count)
3163      response.IndentLess();
3164  
3165    response.Indent("</feature>\n");
3166    response.IndentLess();
3167    response.Indent("</target>\n");
3168    return MemoryBuffer::getMemBufferCopy(response.GetString(), "target.xml");
3169  }
3170  
3171  llvm::Expected<std::unique_ptr<llvm::MemoryBuffer>>
ReadXferObject(llvm::StringRef object,llvm::StringRef annex)3172  GDBRemoteCommunicationServerLLGS::ReadXferObject(llvm::StringRef object,
3173                                                   llvm::StringRef annex) {
3174    // Make sure we have a valid process.
3175    if (!m_current_process ||
3176        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
3177      return llvm::createStringError(llvm::inconvertibleErrorCode(),
3178                                     "No process available");
3179    }
3180  
3181    if (object == "auxv") {
3182      // Grab the auxv data.
3183      auto buffer_or_error = m_current_process->GetAuxvData();
3184      if (!buffer_or_error)
3185        return llvm::errorCodeToError(buffer_or_error.getError());
3186      return std::move(*buffer_or_error);
3187    }
3188  
3189    if (object == "siginfo") {
3190      NativeThreadProtocol *thread = m_current_process->GetCurrentThread();
3191      if (!thread)
3192        return llvm::createStringError(llvm::inconvertibleErrorCode(),
3193                                       "no current thread");
3194  
3195      auto buffer_or_error = thread->GetSiginfo();
3196      if (!buffer_or_error)
3197        return buffer_or_error.takeError();
3198      return std::move(*buffer_or_error);
3199    }
3200  
3201    if (object == "libraries-svr4") {
3202      auto library_list = m_current_process->GetLoadedSVR4Libraries();
3203      if (!library_list)
3204        return library_list.takeError();
3205  
3206      StreamString response;
3207      response.Printf("<library-list-svr4 version=\"1.0\">");
3208      for (auto const &library : *library_list) {
3209        response.Printf("<library name=\"%s\" ",
3210                        XMLEncodeAttributeValue(library.name.c_str()).c_str());
3211        response.Printf("lm=\"0x%" PRIx64 "\" ", library.link_map);
3212        response.Printf("l_addr=\"0x%" PRIx64 "\" ", library.base_addr);
3213        response.Printf("l_ld=\"0x%" PRIx64 "\" />", library.ld_addr);
3214      }
3215      response.Printf("</library-list-svr4>");
3216      return MemoryBuffer::getMemBufferCopy(response.GetString(), __FUNCTION__);
3217    }
3218  
3219    if (object == "features" && annex == "target.xml")
3220      return BuildTargetXml();
3221  
3222    return llvm::make_error<UnimplementedError>();
3223  }
3224  
3225  GDBRemoteCommunication::PacketResult
Handle_qXfer(StringExtractorGDBRemote & packet)3226  GDBRemoteCommunicationServerLLGS::Handle_qXfer(
3227      StringExtractorGDBRemote &packet) {
3228    SmallVector<StringRef, 5> fields;
3229    // The packet format is "qXfer:<object>:<action>:<annex>:offset,length"
3230    StringRef(packet.GetStringRef()).split(fields, ':', 4);
3231    if (fields.size() != 5)
3232      return SendIllFormedResponse(packet, "malformed qXfer packet");
3233    StringRef &xfer_object = fields[1];
3234    StringRef &xfer_action = fields[2];
3235    StringRef &xfer_annex = fields[3];
3236    StringExtractor offset_data(fields[4]);
3237    if (xfer_action != "read")
3238      return SendUnimplementedResponse("qXfer action not supported");
3239    // Parse offset.
3240    const uint64_t xfer_offset =
3241        offset_data.GetHexMaxU64(false, std::numeric_limits<uint64_t>::max());
3242    if (xfer_offset == std::numeric_limits<uint64_t>::max())
3243      return SendIllFormedResponse(packet, "qXfer packet missing offset");
3244    // Parse out comma.
3245    if (offset_data.GetChar() != ',')
3246      return SendIllFormedResponse(packet,
3247                                   "qXfer packet missing comma after offset");
3248    // Parse out the length.
3249    const uint64_t xfer_length =
3250        offset_data.GetHexMaxU64(false, std::numeric_limits<uint64_t>::max());
3251    if (xfer_length == std::numeric_limits<uint64_t>::max())
3252      return SendIllFormedResponse(packet, "qXfer packet missing length");
3253  
3254    // Get a previously constructed buffer if it exists or create it now.
3255    std::string buffer_key = (xfer_object + xfer_action + xfer_annex).str();
3256    auto buffer_it = m_xfer_buffer_map.find(buffer_key);
3257    if (buffer_it == m_xfer_buffer_map.end()) {
3258      auto buffer_up = ReadXferObject(xfer_object, xfer_annex);
3259      if (!buffer_up)
3260        return SendErrorResponse(buffer_up.takeError());
3261      buffer_it = m_xfer_buffer_map
3262                      .insert(std::make_pair(buffer_key, std::move(*buffer_up)))
3263                      .first;
3264    }
3265  
3266    // Send back the response
3267    StreamGDBRemote response;
3268    bool done_with_buffer = false;
3269    llvm::StringRef buffer = buffer_it->second->getBuffer();
3270    if (xfer_offset >= buffer.size()) {
3271      // We have nothing left to send.  Mark the buffer as complete.
3272      response.PutChar('l');
3273      done_with_buffer = true;
3274    } else {
3275      // Figure out how many bytes are available starting at the given offset.
3276      buffer = buffer.drop_front(xfer_offset);
3277      // Mark the response type according to whether we're reading the remainder
3278      // of the data.
3279      if (xfer_length >= buffer.size()) {
3280        // There will be nothing left to read after this
3281        response.PutChar('l');
3282        done_with_buffer = true;
3283      } else {
3284        // There will still be bytes to read after this request.
3285        response.PutChar('m');
3286        buffer = buffer.take_front(xfer_length);
3287      }
3288      // Now write the data in encoded binary form.
3289      response.PutEscapedBytes(buffer.data(), buffer.size());
3290    }
3291  
3292    if (done_with_buffer)
3293      m_xfer_buffer_map.erase(buffer_it);
3294  
3295    return SendPacketNoLock(response.GetString());
3296  }
3297  
3298  GDBRemoteCommunication::PacketResult
Handle_QSaveRegisterState(StringExtractorGDBRemote & packet)3299  GDBRemoteCommunicationServerLLGS::Handle_QSaveRegisterState(
3300      StringExtractorGDBRemote &packet) {
3301    Log *log = GetLog(LLDBLog::Thread);
3302  
3303    // Move past packet name.
3304    packet.SetFilePos(strlen("QSaveRegisterState"));
3305  
3306    // Get the thread to use.
3307    NativeThreadProtocol *thread = GetThreadFromSuffix(packet);
3308    if (!thread) {
3309      if (m_thread_suffix_supported)
3310        return SendIllFormedResponse(
3311            packet, "No thread specified in QSaveRegisterState packet");
3312      else
3313        return SendIllFormedResponse(packet,
3314                                     "No thread was is set with the Hg packet");
3315    }
3316  
3317    // Grab the register context for the thread.
3318    NativeRegisterContext& reg_context = thread->GetRegisterContext();
3319  
3320    // Save registers to a buffer.
3321    WritableDataBufferSP register_data_sp;
3322    Status error = reg_context.ReadAllRegisterValues(register_data_sp);
3323    if (error.Fail()) {
3324      LLDB_LOG(log, "pid {0} failed to save all register values: {1}",
3325               m_current_process->GetID(), error);
3326      return SendErrorResponse(0x75);
3327    }
3328  
3329    // Allocate a new save id.
3330    const uint32_t save_id = GetNextSavedRegistersID();
3331    assert((m_saved_registers_map.find(save_id) == m_saved_registers_map.end()) &&
3332           "GetNextRegisterSaveID() returned an existing register save id");
3333  
3334    // Save the register data buffer under the save id.
3335    {
3336      std::lock_guard<std::mutex> guard(m_saved_registers_mutex);
3337      m_saved_registers_map[save_id] = register_data_sp;
3338    }
3339  
3340    // Write the response.
3341    StreamGDBRemote response;
3342    response.Printf("%" PRIu32, save_id);
3343    return SendPacketNoLock(response.GetString());
3344  }
3345  
3346  GDBRemoteCommunication::PacketResult
Handle_QRestoreRegisterState(StringExtractorGDBRemote & packet)3347  GDBRemoteCommunicationServerLLGS::Handle_QRestoreRegisterState(
3348      StringExtractorGDBRemote &packet) {
3349    Log *log = GetLog(LLDBLog::Thread);
3350  
3351    // Parse out save id.
3352    packet.SetFilePos(strlen("QRestoreRegisterState:"));
3353    if (packet.GetBytesLeft() < 1)
3354      return SendIllFormedResponse(
3355          packet, "QRestoreRegisterState packet missing register save id");
3356  
3357    const uint32_t save_id = packet.GetU32(0);
3358    if (save_id == 0) {
3359      LLDB_LOG(log, "QRestoreRegisterState packet has malformed save id, "
3360                    "expecting decimal uint32_t");
3361      return SendErrorResponse(0x76);
3362    }
3363  
3364    // Get the thread to use.
3365    NativeThreadProtocol *thread = GetThreadFromSuffix(packet);
3366    if (!thread) {
3367      if (m_thread_suffix_supported)
3368        return SendIllFormedResponse(
3369            packet, "No thread specified in QRestoreRegisterState packet");
3370      else
3371        return SendIllFormedResponse(packet,
3372                                     "No thread was is set with the Hg packet");
3373    }
3374  
3375    // Grab the register context for the thread.
3376    NativeRegisterContext &reg_context = thread->GetRegisterContext();
3377  
3378    // Retrieve register state buffer, then remove from the list.
3379    DataBufferSP register_data_sp;
3380    {
3381      std::lock_guard<std::mutex> guard(m_saved_registers_mutex);
3382  
3383      // Find the register set buffer for the given save id.
3384      auto it = m_saved_registers_map.find(save_id);
3385      if (it == m_saved_registers_map.end()) {
3386        LLDB_LOG(log,
3387                 "pid {0} does not have a register set save buffer for id {1}",
3388                 m_current_process->GetID(), save_id);
3389        return SendErrorResponse(0x77);
3390      }
3391      register_data_sp = it->second;
3392  
3393      // Remove it from the map.
3394      m_saved_registers_map.erase(it);
3395    }
3396  
3397    Status error = reg_context.WriteAllRegisterValues(register_data_sp);
3398    if (error.Fail()) {
3399      LLDB_LOG(log, "pid {0} failed to restore all register values: {1}",
3400               m_current_process->GetID(), error);
3401      return SendErrorResponse(0x77);
3402    }
3403  
3404    return SendOKResponse();
3405  }
3406  
3407  GDBRemoteCommunication::PacketResult
Handle_vAttach(StringExtractorGDBRemote & packet)3408  GDBRemoteCommunicationServerLLGS::Handle_vAttach(
3409      StringExtractorGDBRemote &packet) {
3410    Log *log = GetLog(LLDBLog::Process);
3411  
3412    // Consume the ';' after vAttach.
3413    packet.SetFilePos(strlen("vAttach"));
3414    if (!packet.GetBytesLeft() || packet.GetChar() != ';')
3415      return SendIllFormedResponse(packet, "vAttach missing expected ';'");
3416  
3417    // Grab the PID to which we will attach (assume hex encoding).
3418    lldb::pid_t pid = packet.GetU32(LLDB_INVALID_PROCESS_ID, 16);
3419    if (pid == LLDB_INVALID_PROCESS_ID)
3420      return SendIllFormedResponse(packet,
3421                                   "vAttach failed to parse the process id");
3422  
3423    // Attempt to attach.
3424    LLDB_LOGF(log,
3425              "GDBRemoteCommunicationServerLLGS::%s attempting to attach to "
3426              "pid %" PRIu64,
3427              __FUNCTION__, pid);
3428  
3429    Status error = AttachToProcess(pid);
3430  
3431    if (error.Fail()) {
3432      LLDB_LOGF(log,
3433                "GDBRemoteCommunicationServerLLGS::%s failed to attach to "
3434                "pid %" PRIu64 ": %s\n",
3435                __FUNCTION__, pid, error.AsCString());
3436      return SendErrorResponse(error);
3437    }
3438  
3439    // Notify we attached by sending a stop packet.
3440    assert(m_current_process);
3441    return SendStopReasonForState(*m_current_process,
3442                                  m_current_process->GetState(),
3443                                  /*force_synchronous=*/false);
3444  }
3445  
3446  GDBRemoteCommunication::PacketResult
Handle_vAttachWait(StringExtractorGDBRemote & packet)3447  GDBRemoteCommunicationServerLLGS::Handle_vAttachWait(
3448      StringExtractorGDBRemote &packet) {
3449    Log *log = GetLog(LLDBLog::Process);
3450  
3451    // Consume the ';' after the identifier.
3452    packet.SetFilePos(strlen("vAttachWait"));
3453  
3454    if (!packet.GetBytesLeft() || packet.GetChar() != ';')
3455      return SendIllFormedResponse(packet, "vAttachWait missing expected ';'");
3456  
3457    // Allocate the buffer for the process name from vAttachWait.
3458    std::string process_name;
3459    if (!packet.GetHexByteString(process_name))
3460      return SendIllFormedResponse(packet,
3461                                   "vAttachWait failed to parse process name");
3462  
3463    LLDB_LOG(log, "attempting to attach to process named '{0}'", process_name);
3464  
3465    Status error = AttachWaitProcess(process_name, false);
3466    if (error.Fail()) {
3467      LLDB_LOG(log, "failed to attach to process named '{0}': {1}", process_name,
3468               error);
3469      return SendErrorResponse(error);
3470    }
3471  
3472    // Notify we attached by sending a stop packet.
3473    assert(m_current_process);
3474    return SendStopReasonForState(*m_current_process,
3475                                  m_current_process->GetState(),
3476                                  /*force_synchronous=*/false);
3477  }
3478  
3479  GDBRemoteCommunication::PacketResult
Handle_qVAttachOrWaitSupported(StringExtractorGDBRemote & packet)3480  GDBRemoteCommunicationServerLLGS::Handle_qVAttachOrWaitSupported(
3481      StringExtractorGDBRemote &packet) {
3482    return SendOKResponse();
3483  }
3484  
3485  GDBRemoteCommunication::PacketResult
Handle_vAttachOrWait(StringExtractorGDBRemote & packet)3486  GDBRemoteCommunicationServerLLGS::Handle_vAttachOrWait(
3487      StringExtractorGDBRemote &packet) {
3488    Log *log = GetLog(LLDBLog::Process);
3489  
3490    // Consume the ';' after the identifier.
3491    packet.SetFilePos(strlen("vAttachOrWait"));
3492  
3493    if (!packet.GetBytesLeft() || packet.GetChar() != ';')
3494      return SendIllFormedResponse(packet, "vAttachOrWait missing expected ';'");
3495  
3496    // Allocate the buffer for the process name from vAttachWait.
3497    std::string process_name;
3498    if (!packet.GetHexByteString(process_name))
3499      return SendIllFormedResponse(packet,
3500                                   "vAttachOrWait failed to parse process name");
3501  
3502    LLDB_LOG(log, "attempting to attach to process named '{0}'", process_name);
3503  
3504    Status error = AttachWaitProcess(process_name, true);
3505    if (error.Fail()) {
3506      LLDB_LOG(log, "failed to attach to process named '{0}': {1}", process_name,
3507               error);
3508      return SendErrorResponse(error);
3509    }
3510  
3511    // Notify we attached by sending a stop packet.
3512    assert(m_current_process);
3513    return SendStopReasonForState(*m_current_process,
3514                                  m_current_process->GetState(),
3515                                  /*force_synchronous=*/false);
3516  }
3517  
3518  GDBRemoteCommunication::PacketResult
Handle_vRun(StringExtractorGDBRemote & packet)3519  GDBRemoteCommunicationServerLLGS::Handle_vRun(
3520      StringExtractorGDBRemote &packet) {
3521    Log *log = GetLog(LLDBLog::Process);
3522  
3523    llvm::StringRef s = packet.GetStringRef();
3524    if (!s.consume_front("vRun;"))
3525      return SendErrorResponse(8);
3526  
3527    llvm::SmallVector<llvm::StringRef, 16> argv;
3528    s.split(argv, ';');
3529  
3530    for (llvm::StringRef hex_arg : argv) {
3531      StringExtractor arg_ext{hex_arg};
3532      std::string arg;
3533      arg_ext.GetHexByteString(arg);
3534      m_process_launch_info.GetArguments().AppendArgument(arg);
3535      LLDB_LOGF(log, "LLGSPacketHandler::%s added arg: \"%s\"", __FUNCTION__,
3536                arg.c_str());
3537    }
3538  
3539    if (argv.empty())
3540      return SendErrorResponse(Status("No arguments"));
3541    m_process_launch_info.GetExecutableFile().SetFile(
3542        m_process_launch_info.GetArguments()[0].ref(), FileSpec::Style::native);
3543    m_process_launch_error = LaunchProcess();
3544    if (m_process_launch_error.Fail())
3545      return SendErrorResponse(m_process_launch_error);
3546    assert(m_current_process);
3547    return SendStopReasonForState(*m_current_process,
3548                                  m_current_process->GetState(),
3549                                  /*force_synchronous=*/true);
3550  }
3551  
3552  GDBRemoteCommunication::PacketResult
Handle_D(StringExtractorGDBRemote & packet)3553  GDBRemoteCommunicationServerLLGS::Handle_D(StringExtractorGDBRemote &packet) {
3554    Log *log = GetLog(LLDBLog::Process);
3555    if (!m_non_stop)
3556      StopSTDIOForwarding();
3557  
3558    lldb::pid_t pid = LLDB_INVALID_PROCESS_ID;
3559  
3560    // Consume the ';' after D.
3561    packet.SetFilePos(1);
3562    if (packet.GetBytesLeft()) {
3563      if (packet.GetChar() != ';')
3564        return SendIllFormedResponse(packet, "D missing expected ';'");
3565  
3566      // Grab the PID from which we will detach (assume hex encoding).
3567      pid = packet.GetU32(LLDB_INVALID_PROCESS_ID, 16);
3568      if (pid == LLDB_INVALID_PROCESS_ID)
3569        return SendIllFormedResponse(packet, "D failed to parse the process id");
3570    }
3571  
3572    // Detach forked children if their PID was specified *or* no PID was requested
3573    // (i.e. detach-all packet).
3574    llvm::Error detach_error = llvm::Error::success();
3575    bool detached = false;
3576    for (auto it = m_debugged_processes.begin();
3577         it != m_debugged_processes.end();) {
3578      if (pid == LLDB_INVALID_PROCESS_ID || pid == it->first) {
3579        LLDB_LOGF(log,
3580                  "GDBRemoteCommunicationServerLLGS::%s detaching %" PRId64,
3581                  __FUNCTION__, it->first);
3582        if (llvm::Error e = it->second.process_up->Detach().ToError())
3583          detach_error = llvm::joinErrors(std::move(detach_error), std::move(e));
3584        else {
3585          if (it->second.process_up.get() == m_current_process)
3586            m_current_process = nullptr;
3587          if (it->second.process_up.get() == m_continue_process)
3588            m_continue_process = nullptr;
3589          it = m_debugged_processes.erase(it);
3590          detached = true;
3591          continue;
3592        }
3593      }
3594      ++it;
3595    }
3596  
3597    if (detach_error)
3598      return SendErrorResponse(std::move(detach_error));
3599    if (!detached)
3600      return SendErrorResponse(Status("PID %" PRIu64 " not traced", pid));
3601    return SendOKResponse();
3602  }
3603  
3604  GDBRemoteCommunication::PacketResult
Handle_qThreadStopInfo(StringExtractorGDBRemote & packet)3605  GDBRemoteCommunicationServerLLGS::Handle_qThreadStopInfo(
3606      StringExtractorGDBRemote &packet) {
3607    Log *log = GetLog(LLDBLog::Thread);
3608  
3609    if (!m_current_process ||
3610        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID))
3611      return SendErrorResponse(50);
3612  
3613    packet.SetFilePos(strlen("qThreadStopInfo"));
3614    const lldb::tid_t tid = packet.GetHexMaxU64(false, LLDB_INVALID_THREAD_ID);
3615    if (tid == LLDB_INVALID_THREAD_ID) {
3616      LLDB_LOGF(log,
3617                "GDBRemoteCommunicationServerLLGS::%s failed, could not "
3618                "parse thread id from request \"%s\"",
3619                __FUNCTION__, packet.GetStringRef().data());
3620      return SendErrorResponse(0x15);
3621    }
3622    return SendStopReplyPacketForThread(*m_current_process, tid,
3623                                        /*force_synchronous=*/true);
3624  }
3625  
3626  GDBRemoteCommunication::PacketResult
Handle_jThreadsInfo(StringExtractorGDBRemote &)3627  GDBRemoteCommunicationServerLLGS::Handle_jThreadsInfo(
3628      StringExtractorGDBRemote &) {
3629    Log *log = GetLog(LLDBLog::Process | LLDBLog::Thread);
3630  
3631    // Ensure we have a debugged process.
3632    if (!m_current_process ||
3633        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID))
3634      return SendErrorResponse(50);
3635    LLDB_LOG(log, "preparing packet for pid {0}", m_current_process->GetID());
3636  
3637    StreamString response;
3638    const bool threads_with_valid_stop_info_only = false;
3639    llvm::Expected<json::Value> threads_info =
3640        GetJSONThreadsInfo(*m_current_process, threads_with_valid_stop_info_only);
3641    if (!threads_info) {
3642      LLDB_LOG_ERROR(log, threads_info.takeError(),
3643                     "failed to prepare a packet for pid {1}: {0}",
3644                     m_current_process->GetID());
3645      return SendErrorResponse(52);
3646    }
3647  
3648    response.AsRawOstream() << *threads_info;
3649    StreamGDBRemote escaped_response;
3650    escaped_response.PutEscapedBytes(response.GetData(), response.GetSize());
3651    return SendPacketNoLock(escaped_response.GetString());
3652  }
3653  
3654  GDBRemoteCommunication::PacketResult
Handle_qWatchpointSupportInfo(StringExtractorGDBRemote & packet)3655  GDBRemoteCommunicationServerLLGS::Handle_qWatchpointSupportInfo(
3656      StringExtractorGDBRemote &packet) {
3657    // Fail if we don't have a current process.
3658    if (!m_current_process ||
3659        m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)
3660      return SendErrorResponse(68);
3661  
3662    packet.SetFilePos(strlen("qWatchpointSupportInfo"));
3663    if (packet.GetBytesLeft() == 0)
3664      return SendOKResponse();
3665    if (packet.GetChar() != ':')
3666      return SendErrorResponse(67);
3667  
3668    auto hw_debug_cap = m_current_process->GetHardwareDebugSupportInfo();
3669  
3670    StreamGDBRemote response;
3671    if (hw_debug_cap == std::nullopt)
3672      response.Printf("num:0;");
3673    else
3674      response.Printf("num:%d;", hw_debug_cap->second);
3675  
3676    return SendPacketNoLock(response.GetString());
3677  }
3678  
3679  GDBRemoteCommunication::PacketResult
Handle_qFileLoadAddress(StringExtractorGDBRemote & packet)3680  GDBRemoteCommunicationServerLLGS::Handle_qFileLoadAddress(
3681      StringExtractorGDBRemote &packet) {
3682    // Fail if we don't have a current process.
3683    if (!m_current_process ||
3684        m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)
3685      return SendErrorResponse(67);
3686  
3687    packet.SetFilePos(strlen("qFileLoadAddress:"));
3688    if (packet.GetBytesLeft() == 0)
3689      return SendErrorResponse(68);
3690  
3691    std::string file_name;
3692    packet.GetHexByteString(file_name);
3693  
3694    lldb::addr_t file_load_address = LLDB_INVALID_ADDRESS;
3695    Status error =
3696        m_current_process->GetFileLoadAddress(file_name, file_load_address);
3697    if (error.Fail())
3698      return SendErrorResponse(69);
3699  
3700    if (file_load_address == LLDB_INVALID_ADDRESS)
3701      return SendErrorResponse(1); // File not loaded
3702  
3703    StreamGDBRemote response;
3704    response.PutHex64(file_load_address);
3705    return SendPacketNoLock(response.GetString());
3706  }
3707  
3708  GDBRemoteCommunication::PacketResult
Handle_QPassSignals(StringExtractorGDBRemote & packet)3709  GDBRemoteCommunicationServerLLGS::Handle_QPassSignals(
3710      StringExtractorGDBRemote &packet) {
3711    std::vector<int> signals;
3712    packet.SetFilePos(strlen("QPassSignals:"));
3713  
3714    // Read sequence of hex signal numbers divided by a semicolon and optionally
3715    // spaces.
3716    while (packet.GetBytesLeft() > 0) {
3717      int signal = packet.GetS32(-1, 16);
3718      if (signal < 0)
3719        return SendIllFormedResponse(packet, "Failed to parse signal number.");
3720      signals.push_back(signal);
3721  
3722      packet.SkipSpaces();
3723      char separator = packet.GetChar();
3724      if (separator == '\0')
3725        break; // End of string
3726      if (separator != ';')
3727        return SendIllFormedResponse(packet, "Invalid separator,"
3728                                              " expected semicolon.");
3729    }
3730  
3731    // Fail if we don't have a current process.
3732    if (!m_current_process)
3733      return SendErrorResponse(68);
3734  
3735    Status error = m_current_process->IgnoreSignals(signals);
3736    if (error.Fail())
3737      return SendErrorResponse(69);
3738  
3739    return SendOKResponse();
3740  }
3741  
3742  GDBRemoteCommunication::PacketResult
Handle_qMemTags(StringExtractorGDBRemote & packet)3743  GDBRemoteCommunicationServerLLGS::Handle_qMemTags(
3744      StringExtractorGDBRemote &packet) {
3745    Log *log = GetLog(LLDBLog::Process);
3746  
3747    // Ensure we have a process.
3748    if (!m_current_process ||
3749        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
3750      LLDB_LOGF(
3751          log,
3752          "GDBRemoteCommunicationServerLLGS::%s failed, no process available",
3753          __FUNCTION__);
3754      return SendErrorResponse(1);
3755    }
3756  
3757    // We are expecting
3758    // qMemTags:<hex address>,<hex length>:<hex type>
3759  
3760    // Address
3761    packet.SetFilePos(strlen("qMemTags:"));
3762    const char *current_char = packet.Peek();
3763    if (!current_char || *current_char == ',')
3764      return SendIllFormedResponse(packet, "Missing address in qMemTags packet");
3765    const lldb::addr_t addr = packet.GetHexMaxU64(/*little_endian=*/false, 0);
3766  
3767    // Length
3768    char previous_char = packet.GetChar();
3769    current_char = packet.Peek();
3770    // If we don't have a separator or the length field is empty
3771    if (previous_char != ',' || (current_char && *current_char == ':'))
3772      return SendIllFormedResponse(packet,
3773                                   "Invalid addr,length pair in qMemTags packet");
3774  
3775    if (packet.GetBytesLeft() < 1)
3776      return SendIllFormedResponse(
3777          packet, "Too short qMemtags: packet (looking for length)");
3778    const size_t length = packet.GetHexMaxU64(/*little_endian=*/false, 0);
3779  
3780    // Type
3781    const char *invalid_type_err = "Invalid type field in qMemTags: packet";
3782    if (packet.GetBytesLeft() < 1 || packet.GetChar() != ':')
3783      return SendIllFormedResponse(packet, invalid_type_err);
3784  
3785    // Type is a signed integer but packed into the packet as its raw bytes.
3786    // However, our GetU64 uses strtoull which allows +/-. We do not want this.
3787    const char *first_type_char = packet.Peek();
3788    if (first_type_char && (*first_type_char == '+' || *first_type_char == '-'))
3789      return SendIllFormedResponse(packet, invalid_type_err);
3790  
3791    // Extract type as unsigned then cast to signed.
3792    // Using a uint64_t here so that we have some value outside of the 32 bit
3793    // range to use as the invalid return value.
3794    uint64_t raw_type =
3795        packet.GetU64(std::numeric_limits<uint64_t>::max(), /*base=*/16);
3796  
3797    if ( // Make sure the cast below would be valid
3798        raw_type > std::numeric_limits<uint32_t>::max() ||
3799        // To catch inputs like "123aardvark" that will parse but clearly aren't
3800        // valid in this case.
3801        packet.GetBytesLeft()) {
3802      return SendIllFormedResponse(packet, invalid_type_err);
3803    }
3804  
3805    // First narrow to 32 bits otherwise the copy into type would take
3806    // the wrong 4 bytes on big endian.
3807    uint32_t raw_type_32 = raw_type;
3808    int32_t type = reinterpret_cast<int32_t &>(raw_type_32);
3809  
3810    StreamGDBRemote response;
3811    std::vector<uint8_t> tags;
3812    Status error = m_current_process->ReadMemoryTags(type, addr, length, tags);
3813    if (error.Fail())
3814      return SendErrorResponse(1);
3815  
3816    // This m is here in case we want to support multi part replies in the future.
3817    // In the same manner as qfThreadInfo/qsThreadInfo.
3818    response.PutChar('m');
3819    response.PutBytesAsRawHex8(tags.data(), tags.size());
3820    return SendPacketNoLock(response.GetString());
3821  }
3822  
3823  GDBRemoteCommunication::PacketResult
Handle_QMemTags(StringExtractorGDBRemote & packet)3824  GDBRemoteCommunicationServerLLGS::Handle_QMemTags(
3825      StringExtractorGDBRemote &packet) {
3826    Log *log = GetLog(LLDBLog::Process);
3827  
3828    // Ensure we have a process.
3829    if (!m_current_process ||
3830        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)) {
3831      LLDB_LOGF(
3832          log,
3833          "GDBRemoteCommunicationServerLLGS::%s failed, no process available",
3834          __FUNCTION__);
3835      return SendErrorResponse(1);
3836    }
3837  
3838    // We are expecting
3839    // QMemTags:<hex address>,<hex length>:<hex type>:<tags as hex bytes>
3840  
3841    // Address
3842    packet.SetFilePos(strlen("QMemTags:"));
3843    const char *current_char = packet.Peek();
3844    if (!current_char || *current_char == ',')
3845      return SendIllFormedResponse(packet, "Missing address in QMemTags packet");
3846    const lldb::addr_t addr = packet.GetHexMaxU64(/*little_endian=*/false, 0);
3847  
3848    // Length
3849    char previous_char = packet.GetChar();
3850    current_char = packet.Peek();
3851    // If we don't have a separator or the length field is empty
3852    if (previous_char != ',' || (current_char && *current_char == ':'))
3853      return SendIllFormedResponse(packet,
3854                                   "Invalid addr,length pair in QMemTags packet");
3855  
3856    if (packet.GetBytesLeft() < 1)
3857      return SendIllFormedResponse(
3858          packet, "Too short QMemtags: packet (looking for length)");
3859    const size_t length = packet.GetHexMaxU64(/*little_endian=*/false, 0);
3860  
3861    // Type
3862    const char *invalid_type_err = "Invalid type field in QMemTags: packet";
3863    if (packet.GetBytesLeft() < 1 || packet.GetChar() != ':')
3864      return SendIllFormedResponse(packet, invalid_type_err);
3865  
3866    // Our GetU64 uses strtoull which allows leading +/-, we don't want that.
3867    const char *first_type_char = packet.Peek();
3868    if (first_type_char && (*first_type_char == '+' || *first_type_char == '-'))
3869      return SendIllFormedResponse(packet, invalid_type_err);
3870  
3871    // The type is a signed integer but is in the packet as its raw bytes.
3872    // So parse first as unsigned then cast to signed later.
3873    // We extract to 64 bit, even though we only expect 32, so that we've
3874    // got some invalid value we can check for.
3875    uint64_t raw_type =
3876        packet.GetU64(std::numeric_limits<uint64_t>::max(), /*base=*/16);
3877    if (raw_type > std::numeric_limits<uint32_t>::max())
3878      return SendIllFormedResponse(packet, invalid_type_err);
3879  
3880    // First narrow to 32 bits. Otherwise the copy below would get the wrong
3881    // 4 bytes on big endian.
3882    uint32_t raw_type_32 = raw_type;
3883    int32_t type = reinterpret_cast<int32_t &>(raw_type_32);
3884  
3885    // Tag data
3886    if (packet.GetBytesLeft() < 1 || packet.GetChar() != ':')
3887      return SendIllFormedResponse(packet,
3888                                   "Missing tag data in QMemTags: packet");
3889  
3890    // Must be 2 chars per byte
3891    const char *invalid_data_err = "Invalid tag data in QMemTags: packet";
3892    if (packet.GetBytesLeft() % 2)
3893      return SendIllFormedResponse(packet, invalid_data_err);
3894  
3895    // This is bytes here and is unpacked into target specific tags later
3896    // We cannot assume that number of bytes == length here because the server
3897    // can repeat tags to fill a given range.
3898    std::vector<uint8_t> tag_data;
3899    // Zero length writes will not have any tag data
3900    // (but we pass them on because it will still check that tagging is enabled)
3901    if (packet.GetBytesLeft()) {
3902      size_t byte_count = packet.GetBytesLeft() / 2;
3903      tag_data.resize(byte_count);
3904      size_t converted_bytes = packet.GetHexBytes(tag_data, 0);
3905      if (converted_bytes != byte_count) {
3906        return SendIllFormedResponse(packet, invalid_data_err);
3907      }
3908    }
3909  
3910    Status status =
3911        m_current_process->WriteMemoryTags(type, addr, length, tag_data);
3912    return status.Success() ? SendOKResponse() : SendErrorResponse(1);
3913  }
3914  
3915  GDBRemoteCommunication::PacketResult
Handle_qSaveCore(StringExtractorGDBRemote & packet)3916  GDBRemoteCommunicationServerLLGS::Handle_qSaveCore(
3917      StringExtractorGDBRemote &packet) {
3918    // Fail if we don't have a current process.
3919    if (!m_current_process ||
3920        (m_current_process->GetID() == LLDB_INVALID_PROCESS_ID))
3921      return SendErrorResponse(Status("Process not running."));
3922  
3923    std::string path_hint;
3924  
3925    StringRef packet_str{packet.GetStringRef()};
3926    assert(packet_str.starts_with("qSaveCore"));
3927    if (packet_str.consume_front("qSaveCore;")) {
3928      for (auto x : llvm::split(packet_str, ';')) {
3929        if (x.consume_front("path-hint:"))
3930          StringExtractor(x).GetHexByteString(path_hint);
3931        else
3932          return SendErrorResponse(Status("Unsupported qSaveCore option"));
3933      }
3934    }
3935  
3936    llvm::Expected<std::string> ret = m_current_process->SaveCore(path_hint);
3937    if (!ret)
3938      return SendErrorResponse(ret.takeError());
3939  
3940    StreamString response;
3941    response.PutCString("core-path:");
3942    response.PutStringAsRawHex8(ret.get());
3943    return SendPacketNoLock(response.GetString());
3944  }
3945  
3946  GDBRemoteCommunication::PacketResult
Handle_QNonStop(StringExtractorGDBRemote & packet)3947  GDBRemoteCommunicationServerLLGS::Handle_QNonStop(
3948      StringExtractorGDBRemote &packet) {
3949    Log *log = GetLog(LLDBLog::Process);
3950  
3951    StringRef packet_str{packet.GetStringRef()};
3952    assert(packet_str.starts_with("QNonStop:"));
3953    packet_str.consume_front("QNonStop:");
3954    if (packet_str == "0") {
3955      if (m_non_stop)
3956        StopSTDIOForwarding();
3957      for (auto &process_it : m_debugged_processes) {
3958        if (process_it.second.process_up->IsRunning()) {
3959          assert(m_non_stop);
3960          Status error = process_it.second.process_up->Interrupt();
3961          if (error.Fail()) {
3962            LLDB_LOG(log,
3963                     "while disabling nonstop, failed to halt process {0}: {1}",
3964                     process_it.first, error);
3965            return SendErrorResponse(0x41);
3966          }
3967          // we must not send stop reasons after QNonStop
3968          m_disabling_non_stop = true;
3969        }
3970      }
3971      m_stdio_notification_queue.clear();
3972      m_stop_notification_queue.clear();
3973      m_non_stop = false;
3974      // If we are stopping anything, defer sending the OK response until we're
3975      // done.
3976      if (m_disabling_non_stop)
3977        return PacketResult::Success;
3978    } else if (packet_str == "1") {
3979      if (!m_non_stop)
3980        StartSTDIOForwarding();
3981      m_non_stop = true;
3982    } else
3983      return SendErrorResponse(Status("Invalid QNonStop packet"));
3984    return SendOKResponse();
3985  }
3986  
3987  GDBRemoteCommunication::PacketResult
HandleNotificationAck(std::deque<std::string> & queue)3988  GDBRemoteCommunicationServerLLGS::HandleNotificationAck(
3989      std::deque<std::string> &queue) {
3990    // Per the protocol, the first message put into the queue is sent
3991    // immediately.  However, it remains the queue until the client ACKs it --
3992    // then we pop it and send the next message.  The process repeats until
3993    // the last message in the queue is ACK-ed, in which case the packet sends
3994    // an OK response.
3995    if (queue.empty())
3996      return SendErrorResponse(Status("No pending notification to ack"));
3997    queue.pop_front();
3998    if (!queue.empty())
3999      return SendPacketNoLock(queue.front());
4000    return SendOKResponse();
4001  }
4002  
4003  GDBRemoteCommunication::PacketResult
Handle_vStdio(StringExtractorGDBRemote & packet)4004  GDBRemoteCommunicationServerLLGS::Handle_vStdio(
4005      StringExtractorGDBRemote &packet) {
4006    return HandleNotificationAck(m_stdio_notification_queue);
4007  }
4008  
4009  GDBRemoteCommunication::PacketResult
Handle_vStopped(StringExtractorGDBRemote & packet)4010  GDBRemoteCommunicationServerLLGS::Handle_vStopped(
4011      StringExtractorGDBRemote &packet) {
4012    PacketResult ret = HandleNotificationAck(m_stop_notification_queue);
4013    // If this was the last notification and all the processes exited,
4014    // terminate the server.
4015    if (m_stop_notification_queue.empty() && m_debugged_processes.empty()) {
4016      m_exit_now = true;
4017      m_mainloop.RequestTermination();
4018    }
4019    return ret;
4020  }
4021  
4022  GDBRemoteCommunication::PacketResult
Handle_vCtrlC(StringExtractorGDBRemote & packet)4023  GDBRemoteCommunicationServerLLGS::Handle_vCtrlC(
4024      StringExtractorGDBRemote &packet) {
4025    if (!m_non_stop)
4026      return SendErrorResponse(Status("vCtrl is only valid in non-stop mode"));
4027  
4028    PacketResult interrupt_res = Handle_interrupt(packet);
4029    // If interrupting the process failed, pass the result through.
4030    if (interrupt_res != PacketResult::Success)
4031      return interrupt_res;
4032    // Otherwise, vCtrlC should issue an OK response (normal interrupts do not).
4033    return SendOKResponse();
4034  }
4035  
4036  GDBRemoteCommunication::PacketResult
Handle_T(StringExtractorGDBRemote & packet)4037  GDBRemoteCommunicationServerLLGS::Handle_T(StringExtractorGDBRemote &packet) {
4038    packet.SetFilePos(strlen("T"));
4039    auto pid_tid = packet.GetPidTid(m_current_process ? m_current_process->GetID()
4040                                                      : LLDB_INVALID_PROCESS_ID);
4041    if (!pid_tid)
4042      return SendErrorResponse(llvm::make_error<StringError>(
4043          inconvertibleErrorCode(), "Malformed thread-id"));
4044  
4045    lldb::pid_t pid = pid_tid->first;
4046    lldb::tid_t tid = pid_tid->second;
4047  
4048    // Technically, this would also be caught by the PID check but let's be more
4049    // explicit about the error.
4050    if (pid == LLDB_INVALID_PROCESS_ID)
4051      return SendErrorResponse(llvm::make_error<StringError>(
4052          inconvertibleErrorCode(), "No current process and no PID provided"));
4053  
4054    // Check the process ID and find respective process instance.
4055    auto new_process_it = m_debugged_processes.find(pid);
4056    if (new_process_it == m_debugged_processes.end())
4057      return SendErrorResponse(1);
4058  
4059    // Check the thread ID
4060    if (!new_process_it->second.process_up->GetThreadByID(tid))
4061      return SendErrorResponse(2);
4062  
4063    return SendOKResponse();
4064  }
4065  
MaybeCloseInferiorTerminalConnection()4066  void GDBRemoteCommunicationServerLLGS::MaybeCloseInferiorTerminalConnection() {
4067    Log *log = GetLog(LLDBLog::Process);
4068  
4069    // Tell the stdio connection to shut down.
4070    if (m_stdio_communication.IsConnected()) {
4071      auto connection = m_stdio_communication.GetConnection();
4072      if (connection) {
4073        Status error;
4074        connection->Disconnect(&error);
4075  
4076        if (error.Success()) {
4077          LLDB_LOGF(log,
4078                    "GDBRemoteCommunicationServerLLGS::%s disconnect process "
4079                    "terminal stdio - SUCCESS",
4080                    __FUNCTION__);
4081        } else {
4082          LLDB_LOGF(log,
4083                    "GDBRemoteCommunicationServerLLGS::%s disconnect process "
4084                    "terminal stdio - FAIL: %s",
4085                    __FUNCTION__, error.AsCString());
4086        }
4087      }
4088    }
4089  }
4090  
GetThreadFromSuffix(StringExtractorGDBRemote & packet)4091  NativeThreadProtocol *GDBRemoteCommunicationServerLLGS::GetThreadFromSuffix(
4092      StringExtractorGDBRemote &packet) {
4093    // We have no thread if we don't have a process.
4094    if (!m_current_process ||
4095        m_current_process->GetID() == LLDB_INVALID_PROCESS_ID)
4096      return nullptr;
4097  
4098    // If the client hasn't asked for thread suffix support, there will not be a
4099    // thread suffix. Use the current thread in that case.
4100    if (!m_thread_suffix_supported) {
4101      const lldb::tid_t current_tid = GetCurrentThreadID();
4102      if (current_tid == LLDB_INVALID_THREAD_ID)
4103        return nullptr;
4104      else if (current_tid == 0) {
4105        // Pick a thread.
4106        return m_current_process->GetThreadAtIndex(0);
4107      } else
4108        return m_current_process->GetThreadByID(current_tid);
4109    }
4110  
4111    Log *log = GetLog(LLDBLog::Thread);
4112  
4113    // Parse out the ';'.
4114    if (packet.GetBytesLeft() < 1 || packet.GetChar() != ';') {
4115      LLDB_LOGF(log,
4116                "GDBRemoteCommunicationServerLLGS::%s gdb-remote parse "
4117                "error: expected ';' prior to start of thread suffix: packet "
4118                "contents = '%s'",
4119                __FUNCTION__, packet.GetStringRef().data());
4120      return nullptr;
4121    }
4122  
4123    if (!packet.GetBytesLeft())
4124      return nullptr;
4125  
4126    // Parse out thread: portion.
4127    if (strncmp(packet.Peek(), "thread:", strlen("thread:")) != 0) {
4128      LLDB_LOGF(log,
4129                "GDBRemoteCommunicationServerLLGS::%s gdb-remote parse "
4130                "error: expected 'thread:' but not found, packet contents = "
4131                "'%s'",
4132                __FUNCTION__, packet.GetStringRef().data());
4133      return nullptr;
4134    }
4135    packet.SetFilePos(packet.GetFilePos() + strlen("thread:"));
4136    const lldb::tid_t tid = packet.GetHexMaxU64(false, 0);
4137    if (tid != 0)
4138      return m_current_process->GetThreadByID(tid);
4139  
4140    return nullptr;
4141  }
4142  
GetCurrentThreadID() const4143  lldb::tid_t GDBRemoteCommunicationServerLLGS::GetCurrentThreadID() const {
4144    if (m_current_tid == 0 || m_current_tid == LLDB_INVALID_THREAD_ID) {
4145      // Use whatever the debug process says is the current thread id since the
4146      // protocol either didn't specify or specified we want any/all threads
4147      // marked as the current thread.
4148      if (!m_current_process)
4149        return LLDB_INVALID_THREAD_ID;
4150      return m_current_process->GetCurrentThreadID();
4151    }
4152    // Use the specific current thread id set by the gdb remote protocol.
4153    return m_current_tid;
4154  }
4155  
GetNextSavedRegistersID()4156  uint32_t GDBRemoteCommunicationServerLLGS::GetNextSavedRegistersID() {
4157    std::lock_guard<std::mutex> guard(m_saved_registers_mutex);
4158    return m_next_saved_registers_id++;
4159  }
4160  
ClearProcessSpecificData()4161  void GDBRemoteCommunicationServerLLGS::ClearProcessSpecificData() {
4162    Log *log = GetLog(LLDBLog::Process);
4163  
4164    LLDB_LOG(log, "clearing {0} xfer buffers", m_xfer_buffer_map.size());
4165    m_xfer_buffer_map.clear();
4166  }
4167  
4168  FileSpec
FindModuleFile(const std::string & module_path,const ArchSpec & arch)4169  GDBRemoteCommunicationServerLLGS::FindModuleFile(const std::string &module_path,
4170                                                   const ArchSpec &arch) {
4171    if (m_current_process) {
4172      FileSpec file_spec;
4173      if (m_current_process
4174              ->GetLoadedModuleFileSpec(module_path.c_str(), file_spec)
4175              .Success()) {
4176        if (FileSystem::Instance().Exists(file_spec))
4177          return file_spec;
4178      }
4179    }
4180  
4181    return GDBRemoteCommunicationServerCommon::FindModuleFile(module_path, arch);
4182  }
4183  
XMLEncodeAttributeValue(llvm::StringRef value)4184  std::string GDBRemoteCommunicationServerLLGS::XMLEncodeAttributeValue(
4185      llvm::StringRef value) {
4186    std::string result;
4187    for (const char &c : value) {
4188      switch (c) {
4189      case '\'':
4190        result += "&apos;";
4191        break;
4192      case '"':
4193        result += "&quot;";
4194        break;
4195      case '<':
4196        result += "&lt;";
4197        break;
4198      case '>':
4199        result += "&gt;";
4200        break;
4201      default:
4202        result += c;
4203        break;
4204      }
4205    }
4206    return result;
4207  }
4208  
HandleFeatures(const llvm::ArrayRef<llvm::StringRef> client_features)4209  std::vector<std::string> GDBRemoteCommunicationServerLLGS::HandleFeatures(
4210      const llvm::ArrayRef<llvm::StringRef> client_features) {
4211    std::vector<std::string> ret =
4212        GDBRemoteCommunicationServerCommon::HandleFeatures(client_features);
4213    ret.insert(ret.end(), {
4214                              "QThreadSuffixSupported+",
4215                              "QListThreadsInStopReply+",
4216                              "qXfer:features:read+",
4217                              "QNonStop+",
4218                          });
4219  
4220    // report server-only features
4221    using Extension = NativeProcessProtocol::Extension;
4222    Extension plugin_features = m_process_manager.GetSupportedExtensions();
4223    if (bool(plugin_features & Extension::pass_signals))
4224      ret.push_back("QPassSignals+");
4225    if (bool(plugin_features & Extension::auxv))
4226      ret.push_back("qXfer:auxv:read+");
4227    if (bool(plugin_features & Extension::libraries_svr4))
4228      ret.push_back("qXfer:libraries-svr4:read+");
4229    if (bool(plugin_features & Extension::siginfo_read))
4230      ret.push_back("qXfer:siginfo:read+");
4231    if (bool(plugin_features & Extension::memory_tagging))
4232      ret.push_back("memory-tagging+");
4233    if (bool(plugin_features & Extension::savecore))
4234      ret.push_back("qSaveCore+");
4235  
4236    // check for client features
4237    m_extensions_supported = {};
4238    for (llvm::StringRef x : client_features)
4239      m_extensions_supported |=
4240          llvm::StringSwitch<Extension>(x)
4241              .Case("multiprocess+", Extension::multiprocess)
4242              .Case("fork-events+", Extension::fork)
4243              .Case("vfork-events+", Extension::vfork)
4244              .Default({});
4245  
4246    m_extensions_supported &= plugin_features;
4247  
4248    // fork & vfork require multiprocess
4249    if (!bool(m_extensions_supported & Extension::multiprocess))
4250      m_extensions_supported &= ~(Extension::fork | Extension::vfork);
4251  
4252    // report only if actually supported
4253    if (bool(m_extensions_supported & Extension::multiprocess))
4254      ret.push_back("multiprocess+");
4255    if (bool(m_extensions_supported & Extension::fork))
4256      ret.push_back("fork-events+");
4257    if (bool(m_extensions_supported & Extension::vfork))
4258      ret.push_back("vfork-events+");
4259  
4260    for (auto &x : m_debugged_processes)
4261      SetEnabledExtensions(*x.second.process_up);
4262    return ret;
4263  }
4264  
SetEnabledExtensions(NativeProcessProtocol & process)4265  void GDBRemoteCommunicationServerLLGS::SetEnabledExtensions(
4266      NativeProcessProtocol &process) {
4267    NativeProcessProtocol::Extension flags = m_extensions_supported;
4268    assert(!bool(flags & ~m_process_manager.GetSupportedExtensions()));
4269    process.SetEnabledExtensions(flags);
4270  }
4271  
4272  GDBRemoteCommunication::PacketResult
SendContinueSuccessResponse()4273  GDBRemoteCommunicationServerLLGS::SendContinueSuccessResponse() {
4274    if (m_non_stop)
4275      return SendOKResponse();
4276    StartSTDIOForwarding();
4277    return PacketResult::Success;
4278  }
4279  
AppendThreadIDToResponse(Stream & response,lldb::pid_t pid,lldb::tid_t tid)4280  void GDBRemoteCommunicationServerLLGS::AppendThreadIDToResponse(
4281      Stream &response, lldb::pid_t pid, lldb::tid_t tid) {
4282    if (bool(m_extensions_supported &
4283             NativeProcessProtocol::Extension::multiprocess))
4284      response.Format("p{0:x-}.", pid);
4285    response.Format("{0:x-}", tid);
4286  }
4287  
4288  std::string
LLGSArgToURL(llvm::StringRef url_arg,bool reverse_connect)4289  lldb_private::process_gdb_remote::LLGSArgToURL(llvm::StringRef url_arg,
4290                                                 bool reverse_connect) {
4291    // Try parsing the argument as URL.
4292    if (std::optional<URI> url = URI::Parse(url_arg)) {
4293      if (reverse_connect)
4294        return url_arg.str();
4295  
4296      // Translate the scheme from LLGS notation to ConnectionFileDescriptor.
4297      // If the scheme doesn't match any, pass it through to support using CFD
4298      // schemes directly.
4299      std::string new_url = llvm::StringSwitch<std::string>(url->scheme)
4300                                .Case("tcp", "listen")
4301                                .Case("unix", "unix-accept")
4302                                .Case("unix-abstract", "unix-abstract-accept")
4303                                .Default(url->scheme.str());
4304      llvm::append_range(new_url, url_arg.substr(url->scheme.size()));
4305      return new_url;
4306    }
4307  
4308    std::string host_port = url_arg.str();
4309    // If host_and_port starts with ':', default the host to be "localhost" and
4310    // expect the remainder to be the port.
4311    if (url_arg.starts_with(":"))
4312      host_port.insert(0, "localhost");
4313  
4314    // Try parsing the (preprocessed) argument as host:port pair.
4315    if (!llvm::errorToBool(Socket::DecodeHostAndPort(host_port).takeError()))
4316      return (reverse_connect ? "connect://" : "listen://") + host_port;
4317  
4318    // If none of the above applied, interpret the argument as UNIX socket path.
4319    return (reverse_connect ? "unix-connect://" : "unix-accept://") +
4320           url_arg.str();
4321  }
4322