llvm-project

Форк
0
219 строк · 6.9 Кб
1
//===-- source/Host/openbsd/Host.cpp --------------------------------------===//
2
//
3
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4
// See https://llvm.org/LICENSE.txt for license information.
5
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6
//
7
//===----------------------------------------------------------------------===//
8

9
#include <sys/types.h>
10

11
#include <sys/signal.h>
12
#include <sys/exec.h>
13
#include <sys/proc.h>
14
#include <sys/ptrace.h>
15
#include <sys/sysctl.h>
16
#include <sys/user.h>
17

18
#include <cstdio>
19

20
#include "lldb/Host/Host.h"
21
#include "lldb/Host/HostInfo.h"
22
#include "lldb/Utility/DataBufferHeap.h"
23
#include "lldb/Utility/DataExtractor.h"
24
#include "lldb/Utility/Endian.h"
25
#include "lldb/Utility/Log.h"
26
#include "lldb/Utility/NameMatches.h"
27
#include "lldb/Utility/ProcessInfo.h"
28
#include "lldb/Utility/Status.h"
29
#include "lldb/Utility/StreamString.h"
30

31
#include "llvm/TargetParser/Host.h"
32

33
extern "C" {
34
extern char **environ;
35
}
36

37
using namespace lldb;
38
using namespace lldb_private;
39

40
namespace lldb_private {
41
class ProcessLaunchInfo;
42
}
43

44
Environment Host::GetEnvironment() {
45
  Environment env;
46
  char *v;
47
  char **var = environ;
48
  for (; var != NULL && *var != NULL; ++var) {
49
    v = strchr(*var, (int)'-');
50
    if (v == NULL)
51
      continue;
52
    env.insert(v);
53
  }
54
  return env;
55
}
56

57
static bool
58
GetOpenBSDProcessArgs(const ProcessInstanceInfoMatch *match_info_ptr,
59
                      ProcessInstanceInfo &process_info) {
60
  if (process_info.ProcessIDIsValid()) {
61
    int mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_ARGS,
62
                  (int)process_info.GetProcessID()};
63

64
    char arg_data[8192];
65
    size_t arg_data_size = sizeof(arg_data);
66
    if (::sysctl(mib, 4, arg_data, &arg_data_size, NULL, 0) == 0) {
67
      DataExtractor data(arg_data, arg_data_size, endian::InlHostByteOrder(),
68
                         sizeof(void *));
69
      lldb::offset_t offset = 0;
70
      const char *cstr;
71

72
      cstr = data.GetCStr(&offset);
73
      if (cstr) {
74
        process_info.GetExecutableFile().SetFile(cstr, FileSpec::Style::native);
75

76
        if (!(match_info_ptr == NULL ||
77
              NameMatches(
78
                  process_info.GetExecutableFile().GetFilename().GetCString(),
79
                  match_info_ptr->GetNameMatchType(),
80
                  match_info_ptr->GetProcessInfo().GetName())))
81
          return false;
82

83
        Args &proc_args = process_info.GetArguments();
84
        while (1) {
85
          const uint8_t *p = data.PeekData(offset, 1);
86
          while ((p != NULL) && (*p == '\0') && offset < arg_data_size) {
87
            ++offset;
88
            p = data.PeekData(offset, 1);
89
          }
90
          if (p == NULL || offset >= arg_data_size)
91
            return true;
92

93
          cstr = data.GetCStr(&offset);
94
          if (cstr)
95
            proc_args.AppendArgument(llvm::StringRef(cstr));
96
          else
97
            return true;
98
        }
99
      }
100
    }
101
  }
102
  return false;
103
}
104

105
static bool GetOpenBSDProcessCPUType(ProcessInstanceInfo &process_info) {
106
  if (process_info.ProcessIDIsValid()) {
107
    process_info.GetArchitecture() =
108
        HostInfo::GetArchitecture(HostInfo::eArchKindDefault);
109
    return true;
110
  }
111
  process_info.GetArchitecture().Clear();
112
  return false;
113
}
114

115
static bool GetOpenBSDProcessUserAndGroup(ProcessInstanceInfo &process_info) {
116
  struct kinfo_proc proc_kinfo;
117
  size_t proc_kinfo_size;
118

119
  if (process_info.ProcessIDIsValid()) {
120
    int mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_PID,
121
                  (int)process_info.GetProcessID()};
122
    proc_kinfo_size = sizeof(struct kinfo_proc);
123

124
    if (::sysctl(mib, 4, &proc_kinfo, &proc_kinfo_size, NULL, 0) == 0) {
125
      if (proc_kinfo_size > 0) {
126
        process_info.SetParentProcessID(proc_kinfo.p_ppid);
127
        process_info.SetUserID(proc_kinfo.p_ruid);
128
        process_info.SetGroupID(proc_kinfo.p_rgid);
129
        process_info.SetEffectiveUserID(proc_kinfo.p_uid);
130
	process_info.SetEffectiveGroupID(proc_kinfo.p_gid);
131
        return true;
132
      }
133
    }
134
  }
135
  process_info.SetParentProcessID(LLDB_INVALID_PROCESS_ID);
136
  process_info.SetUserID(UINT32_MAX);
137
  process_info.SetGroupID(UINT32_MAX);
138
  process_info.SetEffectiveUserID(UINT32_MAX);
139
  process_info.SetEffectiveGroupID(UINT32_MAX);
140
  return false;
141
}
142

143
uint32_t Host::FindProcessesImpl(const ProcessInstanceInfoMatch &match_info,
144
                                 ProcessInstanceInfoList &process_infos) {
145
  std::vector<struct kinfo_proc> kinfos;
146

147
  int mib[3] = {CTL_KERN, KERN_PROC, KERN_PROC_ALL};
148

149
  size_t pid_data_size = 0;
150
  if (::sysctl(mib, 3, NULL, &pid_data_size, NULL, 0) != 0)
151
    return 0;
152

153
  // Add a few extra in case a few more show up
154
  const size_t estimated_pid_count =
155
      (pid_data_size / sizeof(struct kinfo_proc)) + 10;
156

157
  kinfos.resize(estimated_pid_count);
158
  pid_data_size = kinfos.size() * sizeof(struct kinfo_proc);
159

160
  if (::sysctl(mib, 3, &kinfos[0], &pid_data_size, NULL, 0) != 0)
161
    return 0;
162

163
  const size_t actual_pid_count = (pid_data_size / sizeof(struct kinfo_proc));
164

165
  bool all_users = match_info.GetMatchAllUsers();
166
  const ::pid_t our_pid = getpid();
167
  const uid_t our_uid = getuid();
168
  for (size_t i = 0; i < actual_pid_count; i++) {
169
    const struct kinfo_proc &kinfo = kinfos[i];
170
    const bool kinfo_user_matches = (all_users || (kinfo.p_ruid == our_uid) ||
171
                                     // Special case, if lldb is being run as
172
                                     // root we can attach to anything.
173
                                     (our_uid == 0));
174

175
    if (kinfo_user_matches == false || // Make sure the user is acceptable
176
        kinfo.p_pid == our_pid ||     // Skip this process
177
        kinfo.p_pid == 0 ||           // Skip kernel (kernel pid is zero)
178
        kinfo.p_stat == SZOMB ||      // Zombies are bad, they like brains...
179
        kinfo.p_psflags & PS_TRACED || // Being debugged?
180
        kinfo.p_flag & P_WEXIT)       // Working on exiting
181
      continue;
182

183
    ProcessInstanceInfo process_info;
184
    process_info.SetProcessID(kinfo.p_pid);
185
    process_info.SetParentProcessID(kinfo.p_ppid);
186
    process_info.SetUserID(kinfo.p_ruid);
187
    process_info.SetGroupID(kinfo.p_rgid);
188
    process_info.SetEffectiveUserID(kinfo.p_svuid);
189
    process_info.SetEffectiveGroupID(kinfo.p_svgid);
190

191
    // Make sure our info matches before we go fetch the name and cpu type
192
    if (match_info.Matches(process_info) &&
193
        GetOpenBSDProcessArgs(&match_info, process_info)) {
194
      GetOpenBSDProcessCPUType(process_info);
195
      if (match_info.Matches(process_info))
196
        process_infos.push_back(process_info);
197
    }
198
  }
199

200
  return process_infos.size();
201
}
202

203
bool Host::GetProcessInfo(lldb::pid_t pid, ProcessInstanceInfo &process_info) {
204
  process_info.SetProcessID(pid);
205

206
  if (GetOpenBSDProcessArgs(NULL, process_info)) {
207
    // should use libprocstat instead of going right into sysctl?
208
    GetOpenBSDProcessCPUType(process_info);
209
    GetOpenBSDProcessUserAndGroup(process_info);
210
    return true;
211
  }
212

213
  process_info.Clear();
214
  return false;
215
}
216

217
Status Host::ShellExpandArguments(ProcessLaunchInfo &launch_info) {
218
  return Status("unimplemented");
219
}
220

Использование cookies

Мы используем файлы cookie в соответствии с Политикой конфиденциальности и Политикой использования cookies.

Нажимая кнопку «Принимаю», Вы даете АО «СберТех» согласие на обработку Ваших персональных данных в целях совершенствования нашего веб-сайта и Сервиса GitVerse, а также повышения удобства их использования.

Запретить использование cookies Вы можете самостоятельно в настройках Вашего браузера.