125 lines
3.5 KiB
C++
125 lines
3.5 KiB
C++
/*
|
|
* Copyright (C) 2016 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <dirent.h>
|
|
#include <fcntl.h>
|
|
#include <stdlib.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <type_traits>
|
|
|
|
#include <android-base/logging.h>
|
|
#include <android-base/parseint.h>
|
|
#include <android-base/unique_fd.h>
|
|
|
|
namespace android {
|
|
namespace procinfo {
|
|
|
|
#if defined(__linux__)
|
|
|
|
enum ProcessState {
|
|
kProcessStateUnknown,
|
|
kProcessStateRunning,
|
|
kProcessStateSleeping,
|
|
kProcessStateUninterruptibleWait,
|
|
kProcessStateStopped,
|
|
kProcessStateZombie,
|
|
};
|
|
|
|
struct ProcessInfo {
|
|
std::string name;
|
|
ProcessState state;
|
|
pid_t tid;
|
|
pid_t pid;
|
|
pid_t ppid;
|
|
pid_t tracer;
|
|
uid_t uid;
|
|
uid_t gid;
|
|
};
|
|
|
|
// Parse the contents of /proc/<tid>/status into |process_info|.
|
|
bool GetProcessInfo(pid_t tid, ProcessInfo* process_info, std::string* error = nullptr);
|
|
|
|
// Parse the contents of <fd>/status into |process_info|.
|
|
// |fd| should be an fd pointing at a /proc/<pid> directory.
|
|
bool GetProcessInfoFromProcPidFd(int fd, ProcessInfo* process_info, std::string* error = nullptr);
|
|
|
|
// Fetch the list of threads from a given process's /proc/<pid> directory.
|
|
// |fd| should be an fd pointing at a /proc/<pid> directory.
|
|
template <typename Collection>
|
|
auto GetProcessTidsFromProcPidFd(int fd, Collection* out, std::string* error = nullptr) ->
|
|
typename std::enable_if<sizeof(typename Collection::value_type) >= sizeof(pid_t), bool>::type {
|
|
out->clear();
|
|
|
|
int task_fd = openat(fd, "task", O_DIRECTORY | O_RDONLY | O_CLOEXEC);
|
|
std::unique_ptr<DIR, int (*)(DIR*)> dir(fdopendir(task_fd), closedir);
|
|
if (!dir) {
|
|
if (error != nullptr) {
|
|
*error = "failed to open task directory";
|
|
}
|
|
return false;
|
|
}
|
|
|
|
struct dirent* dent;
|
|
while ((dent = readdir(dir.get()))) {
|
|
if (strcmp(dent->d_name, ".") != 0 && strcmp(dent->d_name, "..") != 0) {
|
|
pid_t tid;
|
|
if (!android::base::ParseInt(dent->d_name, &tid, 1, std::numeric_limits<pid_t>::max())) {
|
|
if (error != nullptr) {
|
|
*error = std::string("failed to parse task id: ") + dent->d_name;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
out->insert(out->end(), tid);
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
template <typename Collection>
|
|
auto GetProcessTids(pid_t pid, Collection* out, std::string* error = nullptr) ->
|
|
typename std::enable_if<sizeof(typename Collection::value_type) >= sizeof(pid_t), bool>::type {
|
|
char task_path[PATH_MAX];
|
|
if (snprintf(task_path, PATH_MAX, "/proc/%d", pid) >= PATH_MAX) {
|
|
if (error != nullptr) {
|
|
*error = "task path overflow (pid = " + std::to_string(pid) + ")";
|
|
}
|
|
return false;
|
|
}
|
|
|
|
android::base::unique_fd fd(open(task_path, O_DIRECTORY | O_RDONLY | O_CLOEXEC));
|
|
if (fd == -1) {
|
|
if (error != nullptr) {
|
|
*error = std::string("failed to open ") + task_path;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
return GetProcessTidsFromProcPidFd(fd.get(), out, error);
|
|
}
|
|
|
|
#endif
|
|
|
|
} /* namespace procinfo */
|
|
} /* namespace android */
|