1 | //===-- RNBContext.h --------------------------------------------*- C++ -*-===// |
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 | // Created by Greg Clayton on 12/12/07. |
10 | // |
11 | //===----------------------------------------------------------------------===// |
12 | |
13 | #ifndef LLDB_TOOLS_DEBUGSERVER_SOURCE_RNBCONTEXT_H |
14 | #define LLDB_TOOLS_DEBUGSERVER_SOURCE_RNBCONTEXT_H |
15 | |
16 | #include "DNBError.h" |
17 | #include "PThreadEvent.h" |
18 | #include "RNBDefs.h" |
19 | #include <string> |
20 | #include <vector> |
21 | |
22 | class RNBContext { |
23 | public: |
24 | using IgnoredExceptions = std::vector<exception_mask_t>; |
25 | enum { |
26 | event_proc_state_changed = 0x001, |
27 | event_proc_thread_running = 0x002, // Sticky |
28 | event_proc_thread_exiting = 0x004, |
29 | event_proc_stdio_available = 0x008, |
30 | event_proc_profile_data = 0x010, |
31 | event_read_packet_available = 0x020, |
32 | event_read_thread_running = 0x040, // Sticky |
33 | event_read_thread_exiting = 0x080, |
34 | |
35 | normal_event_bits = event_proc_state_changed | event_proc_thread_exiting | |
36 | event_proc_stdio_available | event_proc_profile_data | |
37 | event_read_packet_available | |
38 | event_read_thread_exiting , |
39 | |
40 | sticky_event_bits = event_proc_thread_running | event_read_thread_running, |
41 | |
42 | all_event_bits = sticky_event_bits | normal_event_bits |
43 | } event_t; |
44 | // Constructors and Destructors |
45 | RNBContext() = default; |
46 | virtual ~RNBContext(); |
47 | |
48 | nub_process_t ProcessID() const { return m_pid; } |
49 | bool HasValidProcessID() const { return m_pid != INVALID_NUB_PROCESS; } |
50 | void SetProcessID(nub_process_t pid); |
51 | nub_size_t GetProcessStopCount() const { return m_pid_stop_count; } |
52 | bool SetProcessStopCount(nub_size_t count) { |
53 | // Returns true if this class' notion of the PID state changed |
54 | if (m_pid_stop_count == count) |
55 | return false; // Didn't change |
56 | m_pid_stop_count = count; |
57 | return true; // The stop count has changed. |
58 | } |
59 | |
60 | bool ProcessStateRunning() const; |
61 | PThreadEvent &Events() { return m_events; } |
62 | nub_event_t AllEventBits() const { return all_event_bits; } |
63 | nub_event_t NormalEventBits() const { return normal_event_bits; } |
64 | nub_event_t StickyEventBits() const { return sticky_event_bits; } |
65 | const char *EventsAsString(nub_event_t events, std::string &s); |
66 | |
67 | size_t ArgumentCount() const { return m_arg_vec.size(); } |
68 | const char *ArgumentAtIndex(size_t index); |
69 | void PushArgument(const char *arg) { |
70 | if (arg) |
71 | m_arg_vec.push_back(x: arg); |
72 | } |
73 | void ClearArgv() { m_arg_vec.erase(first: m_arg_vec.begin(), last: m_arg_vec.end()); } |
74 | |
75 | size_t EnvironmentCount() const { return m_env_vec.size(); } |
76 | const char *EnvironmentAtIndex(size_t index); |
77 | void PushEnvironment(const char *arg) { |
78 | if (arg) |
79 | m_env_vec.push_back(x: arg); |
80 | } |
81 | void PushEnvironmentIfNeeded(const char *arg); |
82 | void ClearEnvironment() { |
83 | m_env_vec.erase(first: m_env_vec.begin(), last: m_env_vec.end()); |
84 | } |
85 | DNBError &LaunchStatus() { return m_launch_status; } |
86 | const char *LaunchStatusAsString(std::string &s); |
87 | nub_launch_flavor_t LaunchFlavor() const { return m_launch_flavor; } |
88 | void SetLaunchFlavor(nub_launch_flavor_t flavor) { m_launch_flavor = flavor; } |
89 | |
90 | const char *GetWorkingDirectory() const { |
91 | if (!m_working_directory.empty()) |
92 | return m_working_directory.c_str(); |
93 | return NULL; |
94 | } |
95 | |
96 | bool SetWorkingDirectory(const char *path); |
97 | |
98 | std::string &GetSTDIN() { return m_stdin; } |
99 | std::string &GetSTDOUT() { return m_stdout; } |
100 | std::string &GetSTDERR() { return m_stderr; } |
101 | std::string &GetWorkingDir() { return m_working_dir; } |
102 | |
103 | const char *GetSTDINPath() { |
104 | return m_stdin.empty() ? NULL : m_stdin.c_str(); |
105 | } |
106 | const char *GetSTDOUTPath() { |
107 | return m_stdout.empty() ? NULL : m_stdout.c_str(); |
108 | } |
109 | const char *GetSTDERRPath() { |
110 | return m_stderr.empty() ? NULL : m_stderr.c_str(); |
111 | } |
112 | const char *GetWorkingDirPath() { |
113 | return m_working_dir.empty() ? NULL : m_working_dir.c_str(); |
114 | } |
115 | |
116 | void PushProcessEvent(const char *p) { m_process_event.assign(s: p); } |
117 | const char *GetProcessEvent() { return m_process_event.c_str(); } |
118 | |
119 | void SetDetachOnError(bool detach) { m_detach_on_error = detach; } |
120 | bool GetDetachOnError() { return m_detach_on_error; } |
121 | |
122 | bool AddIgnoredException(const char *exception_name); |
123 | |
124 | void AddDefaultIgnoredExceptions(); |
125 | |
126 | const IgnoredExceptions &GetIgnoredExceptions() { |
127 | return m_ignored_exceptions; |
128 | } |
129 | |
130 | protected: |
131 | // Classes that inherit from RNBContext can see and modify these |
132 | nub_process_t m_pid = INVALID_NUB_PROCESS; |
133 | std::string m_stdin; |
134 | std::string m_stdout; |
135 | std::string m_stderr; |
136 | std::string m_working_dir; |
137 | nub_size_t m_pid_stop_count = 0; |
138 | /// Threaded events that we can wait for. |
139 | PThreadEvent m_events{0, all_event_bits}; |
140 | pthread_t m_pid_pthread; |
141 | /// How to launch our inferior process. |
142 | nub_launch_flavor_t m_launch_flavor = eLaunchFlavorDefault; |
143 | /// This holds the status from the last launch attempt. |
144 | DNBError m_launch_status; |
145 | std::vector<std::string> m_arg_vec; |
146 | /// This will be unparsed entries FOO=value |
147 | std::vector<std::string> m_env_vec; |
148 | std::string m_working_directory; |
149 | std::string m_process_event; |
150 | bool m_detach_on_error = false; |
151 | IgnoredExceptions m_ignored_exceptions; |
152 | |
153 | void StartProcessStatusThread(); |
154 | void StopProcessStatusThread(); |
155 | static void *ThreadFunctionProcessStatus(void *arg); |
156 | |
157 | private: |
158 | RNBContext(const RNBContext &rhs) = delete; |
159 | RNBContext &operator=(const RNBContext &rhs) = delete; |
160 | }; |
161 | |
162 | #endif // LLDB_TOOLS_DEBUGSERVER_SOURCE_RNBCONTEXT_H |
163 | |