• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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