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