1*00c7fec1SAndroid Build Coastguard Worker /*
2*00c7fec1SAndroid Build Coastguard Worker * Copyright 2016, The Android Open Source Project
3*00c7fec1SAndroid Build Coastguard Worker *
4*00c7fec1SAndroid Build Coastguard Worker * Licensed under the Apache License, Version 2.0 (the "License");
5*00c7fec1SAndroid Build Coastguard Worker * you may not use this file except in compliance with the License.
6*00c7fec1SAndroid Build Coastguard Worker * You may obtain a copy of the License at
7*00c7fec1SAndroid Build Coastguard Worker *
8*00c7fec1SAndroid Build Coastguard Worker * http://www.apache.org/licenses/LICENSE-2.0
9*00c7fec1SAndroid Build Coastguard Worker *
10*00c7fec1SAndroid Build Coastguard Worker * Unless required by applicable law or agreed to in writing, software
11*00c7fec1SAndroid Build Coastguard Worker * distributed under the License is distributed on an "AS IS" BASIS,
12*00c7fec1SAndroid Build Coastguard Worker * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*00c7fec1SAndroid Build Coastguard Worker * See the License for the specific language governing permissions and
14*00c7fec1SAndroid Build Coastguard Worker * limitations under the License.
15*00c7fec1SAndroid Build Coastguard Worker */
16*00c7fec1SAndroid Build Coastguard Worker
17*00c7fec1SAndroid Build Coastguard Worker #include "util.h"
18*00c7fec1SAndroid Build Coastguard Worker
19*00c7fec1SAndroid Build Coastguard Worker #include <time.h>
20*00c7fec1SAndroid Build Coastguard Worker
21*00c7fec1SAndroid Build Coastguard Worker #include <functional>
22*00c7fec1SAndroid Build Coastguard Worker #include <string>
23*00c7fec1SAndroid Build Coastguard Worker #include <utility>
24*00c7fec1SAndroid Build Coastguard Worker
25*00c7fec1SAndroid Build Coastguard Worker #include <android-base/file.h>
26*00c7fec1SAndroid Build Coastguard Worker #include <android-base/stringprintf.h>
27*00c7fec1SAndroid Build Coastguard Worker #include <android-base/strings.h>
28*00c7fec1SAndroid Build Coastguard Worker #include "protocol.h"
29*00c7fec1SAndroid Build Coastguard Worker
get_command_line(pid_t pid)30*00c7fec1SAndroid Build Coastguard Worker std::vector<std::string> get_command_line(pid_t pid) {
31*00c7fec1SAndroid Build Coastguard Worker std::vector<std::string> result;
32*00c7fec1SAndroid Build Coastguard Worker
33*00c7fec1SAndroid Build Coastguard Worker std::string cmdline;
34*00c7fec1SAndroid Build Coastguard Worker android::base::ReadFileToString(android::base::StringPrintf("/proc/%d/cmdline", pid), &cmdline);
35*00c7fec1SAndroid Build Coastguard Worker
36*00c7fec1SAndroid Build Coastguard Worker auto it = cmdline.cbegin();
37*00c7fec1SAndroid Build Coastguard Worker while (it != cmdline.cend()) {
38*00c7fec1SAndroid Build Coastguard Worker // string::iterator is a wrapped type, not a raw char*.
39*00c7fec1SAndroid Build Coastguard Worker auto terminator = std::find(it, cmdline.cend(), '\0');
40*00c7fec1SAndroid Build Coastguard Worker result.emplace_back(it, terminator);
41*00c7fec1SAndroid Build Coastguard Worker it = std::find_if(terminator, cmdline.cend(), [](char c) { return c != '\0'; });
42*00c7fec1SAndroid Build Coastguard Worker }
43*00c7fec1SAndroid Build Coastguard Worker if (result.empty()) {
44*00c7fec1SAndroid Build Coastguard Worker result.emplace_back("<unknown>");
45*00c7fec1SAndroid Build Coastguard Worker }
46*00c7fec1SAndroid Build Coastguard Worker
47*00c7fec1SAndroid Build Coastguard Worker return result;
48*00c7fec1SAndroid Build Coastguard Worker }
49*00c7fec1SAndroid Build Coastguard Worker
get_process_name(pid_t pid)50*00c7fec1SAndroid Build Coastguard Worker std::string get_process_name(pid_t pid) {
51*00c7fec1SAndroid Build Coastguard Worker std::string result = "<unknown>";
52*00c7fec1SAndroid Build Coastguard Worker android::base::ReadFileToString(android::base::StringPrintf("/proc/%d/cmdline", pid), &result);
53*00c7fec1SAndroid Build Coastguard Worker // We only want the name, not the whole command line, so truncate at the first NUL.
54*00c7fec1SAndroid Build Coastguard Worker return result.c_str();
55*00c7fec1SAndroid Build Coastguard Worker }
56*00c7fec1SAndroid Build Coastguard Worker
get_thread_name(pid_t tid)57*00c7fec1SAndroid Build Coastguard Worker std::string get_thread_name(pid_t tid) {
58*00c7fec1SAndroid Build Coastguard Worker std::string result = "<unknown>";
59*00c7fec1SAndroid Build Coastguard Worker android::base::ReadFileToString(android::base::StringPrintf("/proc/%d/comm", tid), &result);
60*00c7fec1SAndroid Build Coastguard Worker return android::base::Trim(result);
61*00c7fec1SAndroid Build Coastguard Worker }
62*00c7fec1SAndroid Build Coastguard Worker
get_timestamp()63*00c7fec1SAndroid Build Coastguard Worker std::string get_timestamp() {
64*00c7fec1SAndroid Build Coastguard Worker timespec ts;
65*00c7fec1SAndroid Build Coastguard Worker clock_gettime(CLOCK_REALTIME, &ts);
66*00c7fec1SAndroid Build Coastguard Worker
67*00c7fec1SAndroid Build Coastguard Worker tm tm;
68*00c7fec1SAndroid Build Coastguard Worker localtime_r(&ts.tv_sec, &tm);
69*00c7fec1SAndroid Build Coastguard Worker
70*00c7fec1SAndroid Build Coastguard Worker char buf[strlen("1970-01-01 00:00:00.123456789+0830") + 1];
71*00c7fec1SAndroid Build Coastguard Worker char* s = buf;
72*00c7fec1SAndroid Build Coastguard Worker size_t sz = sizeof(buf), n;
73*00c7fec1SAndroid Build Coastguard Worker n = strftime(s, sz, "%F %H:%M", &tm), s += n, sz -= n;
74*00c7fec1SAndroid Build Coastguard Worker n = snprintf(s, sz, ":%02d.%09ld", tm.tm_sec, ts.tv_nsec), s += n, sz -= n;
75*00c7fec1SAndroid Build Coastguard Worker n = strftime(s, sz, "%z", &tm), s += n, sz -= n;
76*00c7fec1SAndroid Build Coastguard Worker return buf;
77*00c7fec1SAndroid Build Coastguard Worker }
78*00c7fec1SAndroid Build Coastguard Worker
iterate_tids(pid_t pid,std::function<void (pid_t)> callback)79*00c7fec1SAndroid Build Coastguard Worker bool iterate_tids(pid_t pid, std::function<void(pid_t)> callback) {
80*00c7fec1SAndroid Build Coastguard Worker char buf[BUFSIZ];
81*00c7fec1SAndroid Build Coastguard Worker snprintf(buf, sizeof(buf), "/proc/%d/task", pid);
82*00c7fec1SAndroid Build Coastguard Worker std::unique_ptr<DIR, int (*)(DIR*)> dir(opendir(buf), closedir);
83*00c7fec1SAndroid Build Coastguard Worker if (dir == nullptr) {
84*00c7fec1SAndroid Build Coastguard Worker return false;
85*00c7fec1SAndroid Build Coastguard Worker }
86*00c7fec1SAndroid Build Coastguard Worker
87*00c7fec1SAndroid Build Coastguard Worker struct dirent* entry;
88*00c7fec1SAndroid Build Coastguard Worker while ((entry = readdir(dir.get())) != nullptr) {
89*00c7fec1SAndroid Build Coastguard Worker pid_t tid = atoi(entry->d_name);
90*00c7fec1SAndroid Build Coastguard Worker if (tid == 0) {
91*00c7fec1SAndroid Build Coastguard Worker continue;
92*00c7fec1SAndroid Build Coastguard Worker }
93*00c7fec1SAndroid Build Coastguard Worker callback(tid);
94*00c7fec1SAndroid Build Coastguard Worker }
95*00c7fec1SAndroid Build Coastguard Worker return true;
96*00c7fec1SAndroid Build Coastguard Worker }
97