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 
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
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, false,
72                                                  FileSpec::Style::native);
73 
74         if (!(match_info_ptr == NULL ||
75               NameMatches(
76                   process_info.GetExecutableFile().GetFilename().GetCString(),
77                   match_info_ptr->GetNameMatchType(),
78                   match_info_ptr->GetProcessInfo().GetName())))
79           return false;
80 
81         Args &proc_args = process_info.GetArguments();
82         while (1) {
83           const uint8_t *p = data.PeekData(offset, 1);
84           while ((p != NULL) && (*p == '\0') && offset < arg_data_size) {
85             ++offset;
86             p = data.PeekData(offset, 1);
87           }
88           if (p == NULL || offset >= arg_data_size)
89             return true;
90 
91           cstr = data.GetCStr(&offset);
92           if (cstr)
93             proc_args.AppendArgument(llvm::StringRef(cstr));
94           else
95             return true;
96         }
97       }
98     }
99   }
100   return false;
101 }
102 
103 static bool GetOpenBSDProcessCPUType(ProcessInstanceInfo &process_info) {
104   if (process_info.ProcessIDIsValid()) {
105     process_info.GetArchitecture() =
106         HostInfo::GetArchitecture(HostInfo::eArchKindDefault);
107     return true;
108   }
109   process_info.GetArchitecture().Clear();
110   return false;
111 }
112 
113 static bool GetOpenBSDProcessUserAndGroup(ProcessInstanceInfo &process_info) {
114   struct kinfo_proc proc_kinfo;
115   size_t proc_kinfo_size;
116 
117   if (process_info.ProcessIDIsValid()) {
118     int mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_PID,
119                   (int)process_info.GetProcessID()};
120     proc_kinfo_size = sizeof(struct kinfo_proc);
121 
122     if (::sysctl(mib, 4, &proc_kinfo, &proc_kinfo_size, NULL, 0) == 0) {
123       if (proc_kinfo_size > 0) {
124         process_info.SetParentProcessID(proc_kinfo.p_ppid);
125         process_info.SetUserID(proc_kinfo.p_ruid);
126         process_info.SetGroupID(proc_kinfo.p_rgid);
127         process_info.SetEffectiveUserID(proc_kinfo.p_uid);
128 	process_info.SetEffectiveGroupID(proc_kinfo.p_gid);
129         return true;
130       }
131     }
132   }
133   process_info.SetParentProcessID(LLDB_INVALID_PROCESS_ID);
134   process_info.SetUserID(UINT32_MAX);
135   process_info.SetGroupID(UINT32_MAX);
136   process_info.SetEffectiveUserID(UINT32_MAX);
137   process_info.SetEffectiveGroupID(UINT32_MAX);
138   return false;
139 }
140 
141 uint32_t Host::FindProcesses(const ProcessInstanceInfoMatch &match_info,
142                              ProcessInstanceInfoList &process_infos) {
143   std::vector<struct kinfo_proc> kinfos;
144 
145   int mib[3] = {CTL_KERN, KERN_PROC, KERN_PROC_ALL};
146 
147   size_t pid_data_size = 0;
148   if (::sysctl(mib, 3, NULL, &pid_data_size, NULL, 0) != 0)
149     return 0;
150 
151   // Add a few extra in case a few more show up
152   const size_t estimated_pid_count =
153       (pid_data_size / sizeof(struct kinfo_proc)) + 10;
154 
155   kinfos.resize(estimated_pid_count);
156   pid_data_size = kinfos.size() * sizeof(struct kinfo_proc);
157 
158   if (::sysctl(mib, 3, &kinfos[0], &pid_data_size, NULL, 0) != 0)
159     return 0;
160 
161   const size_t actual_pid_count = (pid_data_size / sizeof(struct kinfo_proc));
162 
163   bool all_users = match_info.GetMatchAllUsers();
164   const ::pid_t our_pid = getpid();
165   const uid_t our_uid = getuid();
166   for (size_t i = 0; i < actual_pid_count; i++) {
167     const struct kinfo_proc &kinfo = kinfos[i];
168     const bool kinfo_user_matches = (all_users || (kinfo.p_ruid == our_uid) ||
169                                      // Special case, if lldb is being run as
170                                      // root we can attach to anything.
171                                      (our_uid == 0));
172 
173     if (kinfo_user_matches == false || // Make sure the user is acceptable
174         kinfo.p_pid == our_pid ||     // Skip this process
175         kinfo.p_pid == 0 ||           // Skip kernel (kernel pid is zero)
176         kinfo.p_stat == SZOMB ||      // Zombies are bad, they like brains...
177         kinfo.p_psflags & PS_TRACED || // Being debugged?
178         kinfo.p_flag & P_WEXIT)       // Working on exiting
179       continue;
180 
181     ProcessInstanceInfo process_info;
182     process_info.SetProcessID(kinfo.p_pid);
183     process_info.SetParentProcessID(kinfo.p_ppid);
184     process_info.SetUserID(kinfo.p_ruid);
185     process_info.SetGroupID(kinfo.p_rgid);
186     process_info.SetEffectiveUserID(kinfo.p_svuid);
187     process_info.SetEffectiveGroupID(kinfo.p_svgid);
188 
189     // Make sure our info matches before we go fetch the name and cpu type
190     if (match_info.Matches(process_info) &&
191         GetOpenBSDProcessArgs(&match_info, process_info)) {
192       GetOpenBSDProcessCPUType(process_info);
193       if (match_info.Matches(process_info))
194         process_infos.Append(process_info);
195     }
196   }
197 
198   return process_infos.GetSize();
199 }
200 
201 bool Host::GetProcessInfo(lldb::pid_t pid, ProcessInstanceInfo &process_info) {
202   process_info.SetProcessID(pid);
203 
204   if (GetOpenBSDProcessArgs(NULL, process_info)) {
205     // should use libprocstat instead of going right into sysctl?
206     GetOpenBSDProcessCPUType(process_info);
207     GetOpenBSDProcessUserAndGroup(process_info);
208     return true;
209   }
210 
211   process_info.Clear();
212   return false;
213 }
214 
215 Status Host::ShellExpandArguments(ProcessLaunchInfo &launch_info) {
216   return Status("unimplemented");
217 }
218