contrib/llvm/tools/lldb/source/Host/openbsd/Host.cpp

   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
  38 extern "C" {
  39 extern char **environ;
  40 }
  41
  42 using namespace lldb;
  43 using namespace lldb_private;
  44
  45 Environment 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
  58 static bool
  59 GetOpenBSDProcessArgs(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
 107 static 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
 117 static 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
 145 uint32_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
 205 bool 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
 219 Status Host::ShellExpandArguments(ProcessLaunchInfo &launch_info) {
 220   return Status("unimplemented");
 221 }