The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 1 | #include <stdio.h> |
| 2 | #include <stdlib.h> |
| 3 | #include <ctype.h> |
| 4 | #include <fcntl.h> |
| 5 | |
| 6 | #include <string.h> |
| 7 | |
| 8 | #include <sys/stat.h> |
| 9 | #include <sys/types.h> |
| 10 | #include <dirent.h> |
| 11 | |
| 12 | #include <pwd.h> |
| 13 | |
San Mehat | 3927441 | 2009-10-27 11:53:22 -0700 | [diff] [blame] | 14 | #include <cutils/sched_policy.h> |
| 15 | |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 16 | static char *nexttoksep(char **strp, char *sep) |
| 17 | { |
| 18 | char *p = strsep(strp,sep); |
| 19 | return (p == 0) ? "" : p; |
| 20 | } |
| 21 | static char *nexttok(char **strp) |
| 22 | { |
| 23 | return nexttoksep(strp, " "); |
| 24 | } |
| 25 | |
| 26 | #define SHOW_PRIO 1 |
| 27 | #define SHOW_TIME 2 |
San Mehat | 3927441 | 2009-10-27 11:53:22 -0700 | [diff] [blame] | 28 | #define SHOW_POLICY 4 |
Dmitry Shmidt | 8b37c91 | 2010-08-18 17:26:26 -0700 | [diff] [blame] | 29 | #define SHOW_CPU 8 |
Stephen Smalley | 8290d10 | 2012-01-13 08:53:56 -0500 | [diff] [blame] | 30 | #define SHOW_MACLABEL 16 |
Kenny Root | 8f197e6 | 2014-05-14 15:07:08 -0700 | [diff] [blame] | 31 | #define SHOW_ABI 32 |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 32 | |
| 33 | static int display_flags = 0; |
| 34 | |
Kenny Root | 8f197e6 | 2014-05-14 15:07:08 -0700 | [diff] [blame] | 35 | static void print_exe_abi(int pid); |
| 36 | |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 37 | static int ps_line(int pid, int tid, char *namefilter) |
| 38 | { |
| 39 | char statline[1024]; |
| 40 | char cmdline[1024]; |
Stephen Smalley | 8290d10 | 2012-01-13 08:53:56 -0500 | [diff] [blame] | 41 | char macline[1024]; |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 42 | char user[32]; |
| 43 | struct stat stats; |
| 44 | int fd, r; |
| 45 | char *ptr, *name, *state; |
Mark Salyzyn | aa90776 | 2014-05-08 09:31:43 -0700 | [diff] [blame] | 46 | int ppid; |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 47 | unsigned wchan, rss, vss, eip; |
| 48 | unsigned utime, stime; |
Dmitry Shmidt | 8b37c91 | 2010-08-18 17:26:26 -0700 | [diff] [blame] | 49 | int prio, nice, rtprio, sched, psr; |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 50 | struct passwd *pw; |
| 51 | |
| 52 | sprintf(statline, "/proc/%d", pid); |
| 53 | stat(statline, &stats); |
| 54 | |
| 55 | if(tid) { |
| 56 | sprintf(statline, "/proc/%d/task/%d/stat", pid, tid); |
| 57 | cmdline[0] = 0; |
Stephen Smalley | 8290d10 | 2012-01-13 08:53:56 -0500 | [diff] [blame] | 58 | snprintf(macline, sizeof(macline), "/proc/%d/task/%d/attr/current", pid, tid); |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 59 | } else { |
| 60 | sprintf(statline, "/proc/%d/stat", pid); |
Stephen Smalley | 8290d10 | 2012-01-13 08:53:56 -0500 | [diff] [blame] | 61 | sprintf(cmdline, "/proc/%d/cmdline", pid); |
| 62 | snprintf(macline, sizeof(macline), "/proc/%d/attr/current", pid); |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 63 | fd = open(cmdline, O_RDONLY); |
| 64 | if(fd == 0) { |
| 65 | r = 0; |
| 66 | } else { |
| 67 | r = read(fd, cmdline, 1023); |
| 68 | close(fd); |
| 69 | if(r < 0) r = 0; |
| 70 | } |
| 71 | cmdline[r] = 0; |
| 72 | } |
| 73 | |
| 74 | fd = open(statline, O_RDONLY); |
| 75 | if(fd == 0) return -1; |
| 76 | r = read(fd, statline, 1023); |
| 77 | close(fd); |
| 78 | if(r < 0) return -1; |
| 79 | statline[r] = 0; |
| 80 | |
| 81 | ptr = statline; |
| 82 | nexttok(&ptr); // skip pid |
| 83 | ptr++; // skip "(" |
| 84 | |
| 85 | name = ptr; |
| 86 | ptr = strrchr(ptr, ')'); // Skip to *last* occurence of ')', |
| 87 | *ptr++ = '\0'; // and null-terminate name. |
| 88 | |
| 89 | ptr++; // skip " " |
| 90 | state = nexttok(&ptr); |
| 91 | ppid = atoi(nexttok(&ptr)); |
| 92 | nexttok(&ptr); // pgrp |
| 93 | nexttok(&ptr); // sid |
Mark Salyzyn | aa90776 | 2014-05-08 09:31:43 -0700 | [diff] [blame] | 94 | nexttok(&ptr); // tty |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 95 | |
| 96 | nexttok(&ptr); // tpgid |
| 97 | nexttok(&ptr); // flags |
| 98 | nexttok(&ptr); // minflt |
| 99 | nexttok(&ptr); // cminflt |
| 100 | nexttok(&ptr); // majflt |
| 101 | nexttok(&ptr); // cmajflt |
| 102 | #if 1 |
| 103 | utime = atoi(nexttok(&ptr)); |
| 104 | stime = atoi(nexttok(&ptr)); |
| 105 | #else |
| 106 | nexttok(&ptr); // utime |
| 107 | nexttok(&ptr); // stime |
| 108 | #endif |
| 109 | nexttok(&ptr); // cutime |
| 110 | nexttok(&ptr); // cstime |
| 111 | prio = atoi(nexttok(&ptr)); |
| 112 | nice = atoi(nexttok(&ptr)); |
| 113 | nexttok(&ptr); // threads |
| 114 | nexttok(&ptr); // itrealvalue |
| 115 | nexttok(&ptr); // starttime |
| 116 | vss = strtoul(nexttok(&ptr), 0, 10); // vsize |
| 117 | rss = strtoul(nexttok(&ptr), 0, 10); // rss |
| 118 | nexttok(&ptr); // rlim |
| 119 | nexttok(&ptr); // startcode |
| 120 | nexttok(&ptr); // endcode |
| 121 | nexttok(&ptr); // startstack |
| 122 | nexttok(&ptr); // kstkesp |
| 123 | eip = strtoul(nexttok(&ptr), 0, 10); // kstkeip |
| 124 | nexttok(&ptr); // signal |
| 125 | nexttok(&ptr); // blocked |
| 126 | nexttok(&ptr); // sigignore |
| 127 | nexttok(&ptr); // sigcatch |
| 128 | wchan = strtoul(nexttok(&ptr), 0, 10); // wchan |
| 129 | nexttok(&ptr); // nswap |
| 130 | nexttok(&ptr); // cnswap |
| 131 | nexttok(&ptr); // exit signal |
Dmitry Shmidt | 8b37c91 | 2010-08-18 17:26:26 -0700 | [diff] [blame] | 132 | psr = atoi(nexttok(&ptr)); // processor |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 133 | rtprio = atoi(nexttok(&ptr)); // rt_priority |
| 134 | sched = atoi(nexttok(&ptr)); // scheduling policy |
| 135 | |
Mark Salyzyn | aa90776 | 2014-05-08 09:31:43 -0700 | [diff] [blame] | 136 | nexttok(&ptr); // tty |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 137 | |
| 138 | if(tid != 0) { |
| 139 | ppid = pid; |
| 140 | pid = tid; |
| 141 | } |
| 142 | |
| 143 | pw = getpwuid(stats.st_uid); |
| 144 | if(pw == 0) { |
| 145 | sprintf(user,"%d",(int)stats.st_uid); |
| 146 | } else { |
| 147 | strcpy(user,pw->pw_name); |
| 148 | } |
| 149 | |
| 150 | if(!namefilter || !strncmp(name, namefilter, strlen(namefilter))) { |
Stephen Smalley | 8290d10 | 2012-01-13 08:53:56 -0500 | [diff] [blame] | 151 | if (display_flags & SHOW_MACLABEL) { |
| 152 | fd = open(macline, O_RDONLY); |
| 153 | strcpy(macline, "-"); |
| 154 | if (fd >= 0) { |
| 155 | r = read(fd, macline, sizeof(macline)-1); |
| 156 | close(fd); |
| 157 | if (r > 0) |
| 158 | macline[r] = 0; |
| 159 | } |
| 160 | printf("%-30s %-9s %-5d %-5d %s\n", macline, user, pid, ppid, cmdline[0] ? cmdline : name); |
| 161 | return 0; |
| 162 | } |
| 163 | |
San Mehat | 3927441 | 2009-10-27 11:53:22 -0700 | [diff] [blame] | 164 | printf("%-9s %-5d %-5d %-6d %-5d", user, pid, ppid, vss / 1024, rss * 4); |
Dmitry Shmidt | 8b37c91 | 2010-08-18 17:26:26 -0700 | [diff] [blame] | 165 | if (display_flags & SHOW_CPU) |
| 166 | printf(" %-2d", psr); |
| 167 | if (display_flags & SHOW_PRIO) |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 168 | printf(" %-5d %-5d %-5d %-5d", prio, nice, rtprio, sched); |
San Mehat | 3927441 | 2009-10-27 11:53:22 -0700 | [diff] [blame] | 169 | if (display_flags & SHOW_POLICY) { |
| 170 | SchedPolicy p; |
| 171 | if (get_sched_policy(pid, &p) < 0) |
| 172 | printf(" un "); |
Glenn Kasten | 86c7cc8 | 2012-03-05 16:14:39 -0800 | [diff] [blame] | 173 | else |
| 174 | printf(" %.2s ", get_sched_policy_name(p)); |
San Mehat | 3927441 | 2009-10-27 11:53:22 -0700 | [diff] [blame] | 175 | } |
Kenny Root | 8f197e6 | 2014-05-14 15:07:08 -0700 | [diff] [blame] | 176 | printf(" %08x %08x %s ", wchan, eip, state); |
| 177 | if (display_flags & SHOW_ABI) { |
| 178 | print_exe_abi(pid); |
| 179 | } |
| 180 | printf("%s", cmdline[0] ? cmdline : name); |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 181 | if(display_flags&SHOW_TIME) |
| 182 | printf(" (u:%d, s:%d)", utime, stime); |
San Mehat | 3927441 | 2009-10-27 11:53:22 -0700 | [diff] [blame] | 183 | |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 184 | printf("\n"); |
| 185 | } |
| 186 | return 0; |
| 187 | } |
| 188 | |
Kenny Root | 8f197e6 | 2014-05-14 15:07:08 -0700 | [diff] [blame] | 189 | static void print_exe_abi(int pid) |
| 190 | { |
| 191 | int fd, r; |
| 192 | char exeline[1024]; |
| 193 | |
| 194 | sprintf(exeline, "/proc/%d/exe", pid); |
| 195 | fd = open(exeline, O_RDONLY); |
| 196 | if(fd == 0) { |
| 197 | printf(" "); |
| 198 | return; |
| 199 | } |
| 200 | r = read(fd, exeline, 5 /* 4 byte ELFMAG + 1 byte EI_CLASS */); |
| 201 | close(fd); |
| 202 | if(r < 0) { |
| 203 | printf(" "); |
| 204 | return; |
| 205 | } |
| 206 | if (memcmp("\177ELF", exeline, 4) != 0) { |
| 207 | printf("?? "); |
| 208 | return; |
| 209 | } |
| 210 | switch (exeline[4]) { |
| 211 | case 1: |
| 212 | printf("32 "); |
| 213 | return; |
| 214 | case 2: |
| 215 | printf("64 "); |
| 216 | return; |
| 217 | default: |
| 218 | printf("?? "); |
| 219 | return; |
| 220 | } |
| 221 | } |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 222 | |
| 223 | void ps_threads(int pid, char *namefilter) |
| 224 | { |
| 225 | char tmp[128]; |
| 226 | DIR *d; |
| 227 | struct dirent *de; |
| 228 | |
| 229 | sprintf(tmp,"/proc/%d/task",pid); |
| 230 | d = opendir(tmp); |
| 231 | if(d == 0) return; |
| 232 | |
| 233 | while((de = readdir(d)) != 0){ |
| 234 | if(isdigit(de->d_name[0])){ |
| 235 | int tid = atoi(de->d_name); |
| 236 | if(tid == pid) continue; |
| 237 | ps_line(pid, tid, namefilter); |
| 238 | } |
| 239 | } |
| 240 | closedir(d); |
| 241 | } |
| 242 | |
| 243 | int ps_main(int argc, char **argv) |
| 244 | { |
| 245 | DIR *d; |
| 246 | struct dirent *de; |
| 247 | char *namefilter = 0; |
| 248 | int pidfilter = 0; |
| 249 | int threads = 0; |
| 250 | |
| 251 | d = opendir("/proc"); |
| 252 | if(d == 0) return -1; |
| 253 | |
| 254 | while(argc > 1){ |
| 255 | if(!strcmp(argv[1],"-t")) { |
| 256 | threads = 1; |
| 257 | } else if(!strcmp(argv[1],"-x")) { |
| 258 | display_flags |= SHOW_TIME; |
Stephen Smalley | 8290d10 | 2012-01-13 08:53:56 -0500 | [diff] [blame] | 259 | } else if(!strcmp(argv[1], "-Z")) { |
| 260 | display_flags |= SHOW_MACLABEL; |
San Mehat | 3927441 | 2009-10-27 11:53:22 -0700 | [diff] [blame] | 261 | } else if(!strcmp(argv[1],"-P")) { |
| 262 | display_flags |= SHOW_POLICY; |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 263 | } else if(!strcmp(argv[1],"-p")) { |
| 264 | display_flags |= SHOW_PRIO; |
Dmitry Shmidt | 8b37c91 | 2010-08-18 17:26:26 -0700 | [diff] [blame] | 265 | } else if(!strcmp(argv[1],"-c")) { |
| 266 | display_flags |= SHOW_CPU; |
Kenny Root | 8f197e6 | 2014-05-14 15:07:08 -0700 | [diff] [blame] | 267 | } else if(!strcmp(argv[1],"--abi")) { |
| 268 | display_flags |= SHOW_ABI; |
| 269 | } else if(isdigit(argv[1][0])){ |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 270 | pidfilter = atoi(argv[1]); |
| 271 | } else { |
| 272 | namefilter = argv[1]; |
| 273 | } |
| 274 | argc--; |
| 275 | argv++; |
| 276 | } |
| 277 | |
Stephen Smalley | 8290d10 | 2012-01-13 08:53:56 -0500 | [diff] [blame] | 278 | if (display_flags & SHOW_MACLABEL) { |
| 279 | printf("LABEL USER PID PPID NAME\n"); |
| 280 | } else { |
Kenny Root | 8f197e6 | 2014-05-14 15:07:08 -0700 | [diff] [blame] | 281 | printf("USER PID PPID VSIZE RSS %s%s %s WCHAN PC %sNAME\n", |
Stephen Smalley | 8290d10 | 2012-01-13 08:53:56 -0500 | [diff] [blame] | 282 | (display_flags&SHOW_CPU)?"CPU ":"", |
| 283 | (display_flags&SHOW_PRIO)?"PRIO NICE RTPRI SCHED ":"", |
Kenny Root | 8f197e6 | 2014-05-14 15:07:08 -0700 | [diff] [blame] | 284 | (display_flags&SHOW_POLICY)?"PCY " : "", |
| 285 | (display_flags&SHOW_ABI)?"ABI " : ""); |
Stephen Smalley | 8290d10 | 2012-01-13 08:53:56 -0500 | [diff] [blame] | 286 | } |
The Android Open Source Project | dd7bc33 | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 287 | while((de = readdir(d)) != 0){ |
| 288 | if(isdigit(de->d_name[0])){ |
| 289 | int pid = atoi(de->d_name); |
| 290 | if(!pidfilter || (pidfilter == pid)) { |
| 291 | ps_line(pid, 0, namefilter); |
| 292 | if(threads) ps_threads(pid, namefilter); |
| 293 | } |
| 294 | } |
| 295 | } |
| 296 | closedir(d); |
| 297 | return 0; |
| 298 | } |
| 299 | |