1 //===-- source/Host/netbsd/Host.cpp ------------------------------*- C++
2 //-*-===//
3 //
4 //                     The LLVM Compiler Infrastructure
5 //
6 // This file is distributed under the University of Illinois Open Source
7 // License. See LICENSE.TXT for details.
8 //
9 //===----------------------------------------------------------------------===//
10 
11 // C Includes
12 #include <dlfcn.h>
13 #include <execinfo.h>
14 #include <stdio.h>
15 #include <sys/proc.h>
16 #include <sys/sysctl.h>
17 #include <sys/types.h>
18 #include <sys/user.h>
19 
20 #include <limits.h>
21 
22 #include <elf.h>
23 #include <kvm.h>
24 #include <sys/exec.h>
25 #include <sys/ptrace.h>
26 
27 // C++ Includes
28 // Other libraries and framework includes
29 // Project includes
30 #include "lldb/Core/DataExtractor.h"
31 #include "lldb/Core/Error.h"
32 #include "lldb/Core/Log.h"
33 #include "lldb/Core/Module.h"
34 #include "lldb/Core/StreamFile.h"
35 #include "lldb/Core/StreamString.h"
36 #include "lldb/Host/Endian.h"
37 #include "lldb/Host/Host.h"
38 #include "lldb/Host/HostInfo.h"
39 #include "lldb/Target/Platform.h"
40 #include "lldb/Target/Process.h"
41 
42 #include "lldb/Core/DataBufferHeap.h"
43 #include "lldb/Core/DataExtractor.h"
44 #include "lldb/Utility/CleanUp.h"
45 #include "lldb/Utility/NameMatches.h"
46 
47 #include "llvm/Support/Host.h"
48 
49 extern "C" {
50 extern char **environ;
51 }
52 
53 using namespace lldb;
54 using namespace lldb_private;
55 
56 size_t Host::GetEnvironment(StringList &env) {
57   char *v;
58   char **var = environ;
59   for (; var != NULL && *var != NULL; ++var) {
60     v = ::strchr(*var, (int)'-');
61     if (v == NULL)
62       continue;
63     env.AppendString(v);
64   }
65   return env.GetSize();
66 }
67 
68 static bool GetNetBSDProcessArgs(const ProcessInstanceInfoMatch *match_info_ptr,
69                                  ProcessInstanceInfo &process_info) {
70   if (!process_info.ProcessIDIsValid())
71     return false;
72 
73   int pid = process_info.GetProcessID();
74 
75   int mib[4] = {CTL_KERN, KERN_PROC_ARGS, pid, KERN_PROC_ARGV};
76 
77   char arg_data[8192];
78   size_t arg_data_size = sizeof(arg_data);
79   if (::sysctl(mib, 4, arg_data, &arg_data_size, NULL, 0) != 0)
80     return false;
81 
82   DataExtractor data(arg_data, arg_data_size, endian::InlHostByteOrder(),
83                      sizeof(void *));
84   lldb::offset_t offset = 0;
85   const char *cstr;
86 
87   cstr = data.GetCStr(&offset);
88   if (!cstr)
89     return false;
90 
91   process_info.GetExecutableFile().SetFile(cstr, false);
92 
93   if (!(match_info_ptr == NULL ||
94         NameMatches(process_info.GetExecutableFile().GetFilename().GetCString(),
95                     match_info_ptr->GetNameMatchType(),
96                     match_info_ptr->GetProcessInfo().GetName())))
97     return false;
98 
99   Args &proc_args = process_info.GetArguments();
100   while (1) {
101     const uint8_t *p = data.PeekData(offset, 1);
102     while ((p != NULL) && (*p == '\0') && offset < arg_data_size) {
103       ++offset;
104       p = data.PeekData(offset, 1);
105     }
106     if (p == NULL || offset >= arg_data_size)
107       break;
108 
109     cstr = data.GetCStr(&offset);
110     if (!cstr)
111       break;
112 
113     proc_args.AppendArgument(llvm::StringRef(cstr));
114   }
115 
116   return true;
117 }
118 
119 static bool GetNetBSDProcessCPUType(ProcessInstanceInfo &process_info) {
120   if (process_info.ProcessIDIsValid()) {
121     process_info.GetArchitecture() =
122         HostInfo::GetArchitecture(HostInfo::eArchKindDefault);
123     return true;
124   }
125   process_info.GetArchitecture().Clear();
126   return false;
127 }
128 
129 static bool GetNetBSDProcessUserAndGroup(ProcessInstanceInfo &process_info) {
130   ::kvm_t *kdp;
131   char errbuf[_POSIX2_LINE_MAX]; /* XXX: error string unused */
132 
133   struct ::kinfo_proc2 *proc_kinfo;
134   const int pid = process_info.GetProcessID();
135   int nproc;
136 
137   if (!process_info.ProcessIDIsValid())
138     goto error;
139 
140   if ((kdp = ::kvm_openfiles(NULL, NULL, NULL, KVM_NO_FILES, errbuf)) == NULL)
141     goto error;
142 
143   if ((proc_kinfo = ::kvm_getproc2(kdp, KERN_PROC_PID, pid,
144                                    sizeof(struct ::kinfo_proc2), &nproc)) ==
145       NULL) {
146     ::kvm_close(kdp);
147     goto error;
148   }
149 
150   if (nproc < 1) {
151     ::kvm_close(kdp); /* XXX: we don't check for error here */
152     goto error;
153   }
154 
155   process_info.SetParentProcessID(proc_kinfo->p_ppid);
156   process_info.SetUserID(proc_kinfo->p_ruid);
157   process_info.SetGroupID(proc_kinfo->p_rgid);
158   process_info.SetEffectiveUserID(proc_kinfo->p_uid);
159   process_info.SetEffectiveGroupID(proc_kinfo->p_gid);
160 
161   ::kvm_close(kdp); /* XXX: we don't check for error here */
162 
163   return true;
164 
165 error:
166   process_info.SetParentProcessID(LLDB_INVALID_PROCESS_ID);
167   process_info.SetUserID(UINT32_MAX);
168   process_info.SetGroupID(UINT32_MAX);
169   process_info.SetEffectiveUserID(UINT32_MAX);
170   process_info.SetEffectiveGroupID(UINT32_MAX);
171   return false;
172 }
173 
174 uint32_t Host::FindProcesses(const ProcessInstanceInfoMatch &match_info,
175                              ProcessInstanceInfoList &process_infos) {
176   const ::pid_t our_pid = ::getpid();
177   const ::uid_t our_uid = ::getuid();
178 
179   const bool all_users =
180       match_info.GetMatchAllUsers() ||
181       // Special case, if lldb is being run as root we can attach to anything
182       (our_uid == 0);
183 
184   kvm_t *kdp;
185   char errbuf[_POSIX2_LINE_MAX]; /* XXX: error string unused */
186   if ((kdp = ::kvm_openfiles(NULL, NULL, NULL, KVM_NO_FILES, errbuf)) == NULL)
187     return 0;
188 
189   struct ::kinfo_proc2 *proc_kinfo;
190   int nproc;
191   if ((proc_kinfo = ::kvm_getproc2(kdp, KERN_PROC_ALL, 0,
192                                    sizeof(struct ::kinfo_proc2), &nproc)) ==
193       NULL) {
194     ::kvm_close(kdp);
195     return 0;
196   }
197 
198   for (int i = 0; i < nproc; i++) {
199     if (proc_kinfo[i].p_pid < 1)
200       continue; /* not valid */
201     /* Make sure the user is acceptable */
202     if (!all_users && proc_kinfo[i].p_ruid != our_uid)
203       continue;
204 
205     if (proc_kinfo[i].p_pid == our_pid ||  // Skip this process
206         proc_kinfo[i].p_pid == 0 ||        // Skip kernel (kernel pid is 0)
207         proc_kinfo[i].p_stat == LSZOMB ||  // Zombies are bad
208         proc_kinfo[i].p_flag & P_TRACED || // Being debugged?
209         proc_kinfo[i].p_flag & P_WEXIT)    // Working on exiting
210       continue;
211 
212     // Every thread is a process in NetBSD, but all the threads of a single
213     // process have the same pid. Do not store the process info in the
214     // result list if a process with given identifier is already registered
215     // there.
216     if (proc_kinfo[i].p_nlwps > 1) {
217       bool already_registered = false;
218       for (size_t pi = 0; pi < process_infos.GetSize(); pi++) {
219         if (process_infos.GetProcessIDAtIndex(pi) == proc_kinfo[i].p_pid) {
220           already_registered = true;
221           break;
222         }
223       }
224 
225       if (already_registered)
226         continue;
227     }
228     ProcessInstanceInfo process_info;
229     process_info.SetProcessID(proc_kinfo[i].p_pid);
230     process_info.SetParentProcessID(proc_kinfo[i].p_ppid);
231     process_info.SetUserID(proc_kinfo[i].p_ruid);
232     process_info.SetGroupID(proc_kinfo[i].p_rgid);
233     process_info.SetEffectiveUserID(proc_kinfo[i].p_uid);
234     process_info.SetEffectiveGroupID(proc_kinfo[i].p_gid);
235     // Make sure our info matches before we go fetch the name and cpu type
236     if (match_info.Matches(process_info) &&
237         GetNetBSDProcessArgs(&match_info, process_info)) {
238       GetNetBSDProcessCPUType(process_info);
239       if (match_info.Matches(process_info))
240         process_infos.Append(process_info);
241     }
242   }
243 
244   kvm_close(kdp); /* XXX: we don't check for error here */
245 
246   return process_infos.GetSize();
247 }
248 
249 bool Host::GetProcessInfo(lldb::pid_t pid, ProcessInstanceInfo &process_info) {
250   process_info.SetProcessID(pid);
251 
252   if (GetNetBSDProcessArgs(NULL, process_info)) {
253     GetNetBSDProcessCPUType(process_info);
254     GetNetBSDProcessUserAndGroup(process_info);
255     return true;
256   }
257 
258   process_info.Clear();
259   return false;
260 }
261 
262 lldb::DataBufferSP Host::GetAuxvData(lldb_private::Process *process) {
263   return lldb::DataBufferSP();
264 }
265 
266 Error Host::ShellExpandArguments(ProcessLaunchInfo &launch_info) {
267   return Error("unimplemented");
268 }
269