1911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao/* 2911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * Copyright (C) 2016 The Android Open Source Project 3911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * 4911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * Licensed under the Apache License, Version 2.0 (the "License"); 5911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * you may not use this file except in compliance with the License. 6911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * You may obtain a copy of the License at 7911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * 8911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * http://www.apache.org/licenses/LICENSE-2.0 9911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * 10911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * Unless required by applicable law or agreed to in writing, software 11911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * distributed under the License is distributed on an "AS IS" BASIS, 12911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * See the License for the specific language governing permissions and 14911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao * limitations under the License. 15911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao */ 16911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 17911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#pragma once 18911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 19911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#include <dirent.h> 20911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#include <fcntl.h> 21911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#include <stdlib.h> 22911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#include <sys/types.h> 23911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#include <unistd.h> 24911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 25911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#include <memory> 26911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#include <string> 27911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#include <type_traits> 28911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 29911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#include <android-base/logging.h> 30911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#include <android-base/parseint.h> 31911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#include <android-base/unique_fd.h> 32911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 33911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gaonamespace android { 34911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gaonamespace procinfo { 35911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 36911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#if defined(__linux__) 37911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 38911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gaostruct ProcessInfo { 39911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao std::string name; 40911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao pid_t tid; 41911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao pid_t pid; 42911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao pid_t ppid; 43911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao pid_t tracer; 44911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao uid_t uid; 45911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao uid_t gid; 46911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao}; 47911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 48911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao// Parse the contents of /proc/<tid>/status into |process_info|. 49911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gaobool GetProcessInfo(pid_t tid, ProcessInfo* process_info); 50911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 51911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao// Parse the contents of <fd>/status into |process_info|. 52911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao// |fd| should be an fd pointing at a /proc/<pid> directory. 53911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gaobool GetProcessInfoFromProcPidFd(int fd, ProcessInfo* process_info); 54911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 55911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao// Fetch the list of threads from a given process's /proc/<pid> directory. 56911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao// |fd| should be an fd pointing at a /proc/<pid> directory. 57911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gaotemplate <typename Collection> 58911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gaoauto GetProcessTidsFromProcPidFd(int fd, Collection* out) -> 59911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao typename std::enable_if<sizeof(typename Collection::value_type) >= sizeof(pid_t), bool>::type { 60911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao out->clear(); 61911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 62911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao int task_fd = openat(fd, "task", O_DIRECTORY | O_RDONLY | O_CLOEXEC); 63911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao std::unique_ptr<DIR, int (*)(DIR*)> dir(fdopendir(task_fd), closedir); 64911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao if (!dir) { 65911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao PLOG(ERROR) << "failed to open task directory"; 66911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao return false; 67911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao } 68911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 69911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao struct dirent* dent; 70911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao while ((dent = readdir(dir.get()))) { 71911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao if (strcmp(dent->d_name, ".") != 0 && strcmp(dent->d_name, "..") != 0) { 72911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao pid_t tid; 73911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao if (!android::base::ParseInt(dent->d_name, &tid, 1, std::numeric_limits<pid_t>::max())) { 74911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao LOG(ERROR) << "failed to parse task id: " << dent->d_name; 75911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao return false; 76911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao } 77911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 78911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao out->insert(out->end(), tid); 79911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao } 80911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao } 81911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 82911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao return true; 83911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao} 84911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 85911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gaotemplate <typename Collection> 86911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gaoauto GetProcessTids(pid_t pid, Collection* out) -> 87911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao typename std::enable_if<sizeof(typename Collection::value_type) >= sizeof(pid_t), bool>::type { 88911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao char task_path[PATH_MAX]; 89911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao if (snprintf(task_path, PATH_MAX, "/proc/%d", pid) >= PATH_MAX) { 90911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao LOG(ERROR) << "task path overflow (pid = " << pid << ")"; 91911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao return false; 92911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao } 93911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 94911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao android::base::unique_fd fd(open(task_path, O_DIRECTORY | O_RDONLY | O_CLOEXEC)); 95911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao if (fd == -1) { 96911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao PLOG(ERROR) << "failed to open " << task_path; 97911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao return false; 98911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao } 99911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 100911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao return GetProcessTidsFromProcPidFd(fd.get(), out); 101911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao} 102911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 103911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao#endif 104911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao 105911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao} /* namespace procinfo */ 106911d729c8a01fbc052a8e15322bad5413a1138a9Josh Gao} /* namespace android */ 107