| //===-- source/Host/freebsd/Host.cpp ------------------------------*- C++ |
| //-*-===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| // C Includes |
| #include <sys/types.h> |
| |
| #include <sys/exec.h> |
| #include <sys/proc.h> |
| #include <sys/ptrace.h> |
| #include <sys/sysctl.h> |
| #include <sys/user.h> |
| |
| #include <machine/elf.h> |
| |
| #include <dlfcn.h> |
| #include <execinfo.h> |
| #include <stdio.h> |
| |
| // C++ Includes |
| // Other libraries and framework includes |
| // Project includes |
| #include "lldb/Host/Host.h" |
| #include "lldb/Host/HostInfo.h" |
| #include "lldb/Target/Process.h" |
| #include "lldb/Utility/DataBufferHeap.h" |
| #include "lldb/Utility/DataExtractor.h" |
| #include "lldb/Utility/Endian.h" |
| #include "lldb/Utility/Log.h" |
| #include "lldb/Utility/NameMatches.h" |
| #include "lldb/Utility/Status.h" |
| #include "lldb/Utility/StreamString.h" |
| |
| #include "llvm/Support/Host.h" |
| |
| extern "C" { |
| extern char **environ; |
| } |
| |
| using namespace lldb; |
| using namespace lldb_private; |
| |
| static bool |
| GetFreeBSDProcessArgs(const ProcessInstanceInfoMatch *match_info_ptr, |
| ProcessInstanceInfo &process_info) { |
| if (!process_info.ProcessIDIsValid()) |
| return false; |
| |
| int pid = process_info.GetProcessID(); |
| |
| int mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_ARGS, pid}; |
| |
| char arg_data[8192]; |
| size_t arg_data_size = sizeof(arg_data); |
| if (::sysctl(mib, 4, arg_data, &arg_data_size, NULL, 0) != 0) |
| return false; |
| |
| DataExtractor data(arg_data, arg_data_size, endian::InlHostByteOrder(), |
| sizeof(void *)); |
| lldb::offset_t offset = 0; |
| const char *cstr; |
| |
| cstr = data.GetCStr(&offset); |
| if (!cstr) |
| return false; |
| |
| // Get pathname for pid. If that fails fall back to argv[0]. |
| char pathname[MAXPATHLEN]; |
| size_t pathname_len = sizeof(pathname); |
| mib[2] = KERN_PROC_PATHNAME; |
| if (::sysctl(mib, 4, pathname, &pathname_len, NULL, 0) == 0) |
| process_info.GetExecutableFile().SetFile(pathname, false, |
| FileSpec::Style::native); |
| else |
| process_info.GetExecutableFile().SetFile(cstr, false, |
| FileSpec::Style::native); |
| |
| if (!(match_info_ptr == NULL || |
| NameMatches(process_info.GetExecutableFile().GetFilename().GetCString(), |
| match_info_ptr->GetNameMatchType(), |
| match_info_ptr->GetProcessInfo().GetName()))) |
| return false; |
| |
| Args &proc_args = process_info.GetArguments(); |
| while (1) { |
| const uint8_t *p = data.PeekData(offset, 1); |
| while ((p != NULL) && (*p == '\0') && offset < arg_data_size) { |
| ++offset; |
| p = data.PeekData(offset, 1); |
| } |
| if (p == NULL || offset >= arg_data_size) |
| break; |
| |
| cstr = data.GetCStr(&offset); |
| if (!cstr) |
| break; |
| |
| proc_args.AppendArgument(llvm::StringRef(cstr)); |
| } |
| |
| return true; |
| } |
| |
| static bool GetFreeBSDProcessCPUType(ProcessInstanceInfo &process_info) { |
| if (process_info.ProcessIDIsValid()) { |
| process_info.GetArchitecture() = |
| HostInfo::GetArchitecture(HostInfo::eArchKindDefault); |
| return true; |
| } |
| process_info.GetArchitecture().Clear(); |
| return false; |
| } |
| |
| static bool GetFreeBSDProcessUserAndGroup(ProcessInstanceInfo &process_info) { |
| struct kinfo_proc proc_kinfo; |
| size_t proc_kinfo_size; |
| const int pid = process_info.GetProcessID(); |
| int mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_PID, pid}; |
| |
| if (!process_info.ProcessIDIsValid()) |
| goto error; |
| |
| proc_kinfo_size = sizeof(struct kinfo_proc); |
| |
| if (::sysctl(mib, 4, &proc_kinfo, &proc_kinfo_size, NULL, 0) != 0) |
| goto error; |
| |
| if (proc_kinfo_size == 0) |
| goto error; |
| |
| process_info.SetParentProcessID(proc_kinfo.ki_ppid); |
| process_info.SetUserID(proc_kinfo.ki_ruid); |
| process_info.SetGroupID(proc_kinfo.ki_rgid); |
| process_info.SetEffectiveUserID(proc_kinfo.ki_uid); |
| if (proc_kinfo.ki_ngroups > 0) |
| process_info.SetEffectiveGroupID(proc_kinfo.ki_groups[0]); |
| else |
| process_info.SetEffectiveGroupID(UINT32_MAX); |
| return true; |
| |
| error: |
| process_info.SetParentProcessID(LLDB_INVALID_PROCESS_ID); |
| process_info.SetUserID(UINT32_MAX); |
| process_info.SetGroupID(UINT32_MAX); |
| process_info.SetEffectiveUserID(UINT32_MAX); |
| process_info.SetEffectiveGroupID(UINT32_MAX); |
| return false; |
| } |
| |
| uint32_t Host::FindProcesses(const ProcessInstanceInfoMatch &match_info, |
| ProcessInstanceInfoList &process_infos) { |
| const ::pid_t our_pid = ::getpid(); |
| const ::uid_t our_uid = ::getuid(); |
| std::vector<struct kinfo_proc> kinfos; |
| // Special case, if lldb is being run as root we can attach to anything. |
| bool all_users = match_info.GetMatchAllUsers() || (our_uid == 0); |
| |
| int mib[3] = {CTL_KERN, KERN_PROC, KERN_PROC_ALL}; |
| |
| size_t pid_data_size = 0; |
| if (::sysctl(mib, 3, NULL, &pid_data_size, NULL, 0) != 0) |
| return 0; |
| |
| // Add a few extra in case a few more show up |
| const size_t estimated_pid_count = |
| (pid_data_size / sizeof(struct kinfo_proc)) + 10; |
| |
| kinfos.resize(estimated_pid_count); |
| pid_data_size = kinfos.size() * sizeof(struct kinfo_proc); |
| |
| if (::sysctl(mib, 3, &kinfos[0], &pid_data_size, NULL, 0) != 0) |
| return 0; |
| |
| const size_t actual_pid_count = (pid_data_size / sizeof(struct kinfo_proc)); |
| |
| for (size_t i = 0; i < actual_pid_count; i++) { |
| const struct kinfo_proc &kinfo = kinfos[i]; |
| |
| /* Make sure the user is acceptable */ |
| if (!all_users && kinfo.ki_ruid != our_uid) |
| continue; |
| |
| if (kinfo.ki_pid == our_pid || // Skip this process |
| kinfo.ki_pid == 0 || // Skip kernel (kernel pid is 0) |
| kinfo.ki_stat == SZOMB || // Zombies are bad |
| kinfo.ki_flag & P_TRACED || // Being debugged? |
| kinfo.ki_flag & P_WEXIT) // Working on exiting |
| continue; |
| |
| // Every thread is a process in FreeBSD, but all the threads of a single |
| // process have the same pid. Do not store the process info in the result |
| // list if a process with given identifier is already registered there. |
| bool already_registered = false; |
| for (uint32_t pi = 0; |
| !already_registered && (const int)kinfo.ki_numthreads > 1 && |
| pi < (const uint32_t)process_infos.GetSize(); |
| pi++) |
| already_registered = |
| (process_infos.GetProcessIDAtIndex(pi) == (uint32_t)kinfo.ki_pid); |
| |
| if (already_registered) |
| continue; |
| |
| ProcessInstanceInfo process_info; |
| process_info.SetProcessID(kinfo.ki_pid); |
| process_info.SetParentProcessID(kinfo.ki_ppid); |
| process_info.SetUserID(kinfo.ki_ruid); |
| process_info.SetGroupID(kinfo.ki_rgid); |
| process_info.SetEffectiveUserID(kinfo.ki_svuid); |
| process_info.SetEffectiveGroupID(kinfo.ki_svgid); |
| |
| // Make sure our info matches before we go fetch the name and cpu type |
| if (match_info.Matches(process_info) && |
| GetFreeBSDProcessArgs(&match_info, process_info)) { |
| GetFreeBSDProcessCPUType(process_info); |
| if (match_info.Matches(process_info)) |
| process_infos.Append(process_info); |
| } |
| } |
| |
| return process_infos.GetSize(); |
| } |
| |
| bool Host::GetProcessInfo(lldb::pid_t pid, ProcessInstanceInfo &process_info) { |
| process_info.SetProcessID(pid); |
| |
| if (GetFreeBSDProcessArgs(NULL, process_info)) { |
| // should use libprocstat instead of going right into sysctl? |
| GetFreeBSDProcessCPUType(process_info); |
| GetFreeBSDProcessUserAndGroup(process_info); |
| return true; |
| } |
| |
| process_info.Clear(); |
| return false; |
| } |
| |
| Environment Host::GetEnvironment() { return Environment(environ); } |
| |
| Status Host::ShellExpandArguments(ProcessLaunchInfo &launch_info) { |
| return Status("unimplemented"); |
| } |