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