1 //===-- source/Host/openbsd/Host.cpp ----------------------------*- C++ -*-===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include <sys/types.h>
11
12 #include <sys/signal.h>
13 #include <sys/exec.h>
14 #include <sys/proc.h>
15 #include <sys/ptrace.h>
16 #include <sys/sysctl.h>
17 #include <sys/user.h>
18
19 #include <stdio.h>
20
21 #include "lldb/Host/Host.h"
22 #include "lldb/Host/HostInfo.h"
23 #include "lldb/Target/Process.h"
24 #include "lldb/Utility/DataBufferHeap.h"
25 #include "lldb/Utility/DataExtractor.h"
26 #include "lldb/Utility/Endian.h"
27 #include "lldb/Utility/Log.h"
28 #include "lldb/Utility/NameMatches.h"
29 #include "lldb/Utility/Status.h"
30 #include "lldb/Utility/StreamString.h"
31
32 #include "llvm/Support/Host.h"
33
34 extern "C" {
35 extern char **environ;
36 }
37
38 using namespace lldb;
39 using namespace lldb_private;
40
GetEnvironment()41 Environment Host::GetEnvironment() {
42 Environment env;
43 char *v;
44 char **var = environ;
45 for (; var != NULL && *var != NULL; ++var) {
46 v = strchr(*var, (int)'-');
47 if (v == NULL)
48 continue;
49 env.insert(v);
50 }
51 return env;
52 }
53
54 static bool
GetOpenBSDProcessArgs(const ProcessInstanceInfoMatch * match_info_ptr,ProcessInstanceInfo & process_info)55 GetOpenBSDProcessArgs(const ProcessInstanceInfoMatch *match_info_ptr,
56 ProcessInstanceInfo &process_info) {
57 if (process_info.ProcessIDIsValid()) {
58 int mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_ARGS,
59 (int)process_info.GetProcessID()};
60
61 char arg_data[8192];
62 size_t arg_data_size = sizeof(arg_data);
63 if (::sysctl(mib, 4, arg_data, &arg_data_size, NULL, 0) == 0) {
64 DataExtractor data(arg_data, arg_data_size, endian::InlHostByteOrder(),
65 sizeof(void *));
66 lldb::offset_t offset = 0;
67 const char *cstr;
68
69 cstr = data.GetCStr(&offset);
70 if (cstr) {
71 process_info.GetExecutableFile().SetFile(cstr, FileSpec::Style::native);
72
73 if (!(match_info_ptr == NULL ||
74 NameMatches(
75 process_info.GetExecutableFile().GetFilename().GetCString(),
76 match_info_ptr->GetNameMatchType(),
77 match_info_ptr->GetProcessInfo().GetName())))
78 return false;
79
80 Args &proc_args = process_info.GetArguments();
81 while (1) {
82 const uint8_t *p = data.PeekData(offset, 1);
83 while ((p != NULL) && (*p == '\0') && offset < arg_data_size) {
84 ++offset;
85 p = data.PeekData(offset, 1);
86 }
87 if (p == NULL || offset >= arg_data_size)
88 return true;
89
90 cstr = data.GetCStr(&offset);
91 if (cstr)
92 proc_args.AppendArgument(llvm::StringRef(cstr));
93 else
94 return true;
95 }
96 }
97 }
98 }
99 return false;
100 }
101
GetOpenBSDProcessCPUType(ProcessInstanceInfo & process_info)102 static bool GetOpenBSDProcessCPUType(ProcessInstanceInfo &process_info) {
103 if (process_info.ProcessIDIsValid()) {
104 process_info.GetArchitecture() =
105 HostInfo::GetArchitecture(HostInfo::eArchKindDefault);
106 return true;
107 }
108 process_info.GetArchitecture().Clear();
109 return false;
110 }
111
GetOpenBSDProcessUserAndGroup(ProcessInstanceInfo & process_info)112 static bool GetOpenBSDProcessUserAndGroup(ProcessInstanceInfo &process_info) {
113 struct kinfo_proc proc_kinfo;
114 size_t proc_kinfo_size;
115
116 if (process_info.ProcessIDIsValid()) {
117 int mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_PID,
118 (int)process_info.GetProcessID()};
119 proc_kinfo_size = sizeof(struct kinfo_proc);
120
121 if (::sysctl(mib, 4, &proc_kinfo, &proc_kinfo_size, NULL, 0) == 0) {
122 if (proc_kinfo_size > 0) {
123 process_info.SetParentProcessID(proc_kinfo.p_ppid);
124 process_info.SetUserID(proc_kinfo.p_ruid);
125 process_info.SetGroupID(proc_kinfo.p_rgid);
126 process_info.SetEffectiveUserID(proc_kinfo.p_uid);
127 process_info.SetEffectiveGroupID(proc_kinfo.p_gid);
128 return true;
129 }
130 }
131 }
132 process_info.SetParentProcessID(LLDB_INVALID_PROCESS_ID);
133 process_info.SetUserID(UINT32_MAX);
134 process_info.SetGroupID(UINT32_MAX);
135 process_info.SetEffectiveUserID(UINT32_MAX);
136 process_info.SetEffectiveGroupID(UINT32_MAX);
137 return false;
138 }
139
FindProcesses(const ProcessInstanceInfoMatch & match_info,ProcessInstanceInfoList & process_infos)140 uint32_t Host::FindProcesses(const ProcessInstanceInfoMatch &match_info,
141 ProcessInstanceInfoList &process_infos) {
142 std::vector<struct kinfo_proc> kinfos;
143
144 int mib[3] = {CTL_KERN, KERN_PROC, KERN_PROC_ALL};
145
146 size_t pid_data_size = 0;
147 if (::sysctl(mib, 3, NULL, &pid_data_size, NULL, 0) != 0)
148 return 0;
149
150 // Add a few extra in case a few more show up
151 const size_t estimated_pid_count =
152 (pid_data_size / sizeof(struct kinfo_proc)) + 10;
153
154 kinfos.resize(estimated_pid_count);
155 pid_data_size = kinfos.size() * sizeof(struct kinfo_proc);
156
157 if (::sysctl(mib, 3, &kinfos[0], &pid_data_size, NULL, 0) != 0)
158 return 0;
159
160 const size_t actual_pid_count = (pid_data_size / sizeof(struct kinfo_proc));
161
162 bool all_users = match_info.GetMatchAllUsers();
163 const ::pid_t our_pid = getpid();
164 const uid_t our_uid = getuid();
165 for (size_t i = 0; i < actual_pid_count; i++) {
166 const struct kinfo_proc &kinfo = kinfos[i];
167 const bool kinfo_user_matches = (all_users || (kinfo.p_ruid == our_uid) ||
168 // Special case, if lldb is being run as
169 // root we can attach to anything.
170 (our_uid == 0));
171
172 if (kinfo_user_matches == false || // Make sure the user is acceptable
173 kinfo.p_pid == our_pid || // Skip this process
174 kinfo.p_pid == 0 || // Skip kernel (kernel pid is zero)
175 kinfo.p_stat == SZOMB || // Zombies are bad, they like brains...
176 kinfo.p_psflags & PS_TRACED || // Being debugged?
177 kinfo.p_flag & P_WEXIT) // Working on exiting
178 continue;
179
180 ProcessInstanceInfo process_info;
181 process_info.SetProcessID(kinfo.p_pid);
182 process_info.SetParentProcessID(kinfo.p_ppid);
183 process_info.SetUserID(kinfo.p_ruid);
184 process_info.SetGroupID(kinfo.p_rgid);
185 process_info.SetEffectiveUserID(kinfo.p_svuid);
186 process_info.SetEffectiveGroupID(kinfo.p_svgid);
187
188 // Make sure our info matches before we go fetch the name and cpu type
189 if (match_info.Matches(process_info) &&
190 GetOpenBSDProcessArgs(&match_info, process_info)) {
191 GetOpenBSDProcessCPUType(process_info);
192 if (match_info.Matches(process_info))
193 process_infos.Append(process_info);
194 }
195 }
196
197 return process_infos.GetSize();
198 }
199
GetProcessInfo(lldb::pid_t pid,ProcessInstanceInfo & process_info)200 bool Host::GetProcessInfo(lldb::pid_t pid, ProcessInstanceInfo &process_info) {
201 process_info.SetProcessID(pid);
202
203 if (GetOpenBSDProcessArgs(NULL, process_info)) {
204 // should use libprocstat instead of going right into sysctl?
205 GetOpenBSDProcessCPUType(process_info);
206 GetOpenBSDProcessUserAndGroup(process_info);
207 return true;
208 }
209
210 process_info.Clear();
211 return false;
212 }
213
ShellExpandArguments(ProcessLaunchInfo & launch_info)214 Status Host::ShellExpandArguments(ProcessLaunchInfo &launch_info) {
215 return Status("unimplemented");
216 }
217