• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /* //device/libs/android_runtime/android_util_Process.cpp
2 **
3 ** Copyright 2006, The Android Open Source Project
4 **
5 ** Licensed under the Apache License, Version 2.0 (the "License");
6 ** you may not use this file except in compliance with the License.
7 ** You may obtain a copy of the License at
8 **
9 **     http://www.apache.org/licenses/LICENSE-2.0
10 **
11 ** Unless required by applicable law or agreed to in writing, software
12 ** distributed under the License is distributed on an "AS IS" BASIS,
13 ** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 ** See the License for the specific language governing permissions and
15 ** limitations under the License.
16 */
17 
18 #define LOG_TAG "Process"
19 
20 #include <utils/Log.h>
21 #include <binder/IPCThreadState.h>
22 #include <binder/ProcessState.h>
23 #include <binder/IServiceManager.h>
24 #include <cutils/sched_policy.h>
25 #include <utils/String8.h>
26 #include <utils/Vector.h>
27 
28 #include <android_runtime/AndroidRuntime.h>
29 
30 #include "android_util_Binder.h"
31 #include "JNIHelp.h"
32 
33 #include <sys/errno.h>
34 #include <sys/resource.h>
35 #include <sys/types.h>
36 #include <dirent.h>
37 #include <fcntl.h>
38 #include <grp.h>
39 #include <pwd.h>
40 #include <signal.h>
41 #include <unistd.h>
42 
43 #define POLICY_DEBUG 0
44 #define GUARD_THREAD_PRIORITY 0
45 
46 using namespace android;
47 
48 #if GUARD_THREAD_PRIORITY
49 Mutex gKeyCreateMutex;
50 static pthread_key_t gBgKey = -1;
51 #endif
52 
53 // For both of these, err should be in the errno range (positive), not a status_t (negative)
54 
signalExceptionForPriorityError(JNIEnv * env,int err)55 static void signalExceptionForPriorityError(JNIEnv* env, int err)
56 {
57     switch (err) {
58         case EINVAL:
59             jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
60             break;
61         case ESRCH:
62             jniThrowException(env, "java/lang/IllegalArgumentException", "Given thread does not exist");
63             break;
64         case EPERM:
65             jniThrowException(env, "java/lang/SecurityException", "No permission to modify given thread");
66             break;
67         case EACCES:
68             jniThrowException(env, "java/lang/SecurityException", "No permission to set to given priority");
69             break;
70         default:
71             jniThrowException(env, "java/lang/RuntimeException", "Unknown error");
72             break;
73     }
74 }
75 
signalExceptionForGroupError(JNIEnv * env,int err)76 static void signalExceptionForGroupError(JNIEnv* env, int err)
77 {
78     switch (err) {
79         case EINVAL:
80             jniThrowException(env, "java/lang/IllegalArgumentException", NULL);
81             break;
82         case ESRCH:
83             jniThrowException(env, "java/lang/IllegalArgumentException", "Given thread does not exist");
84             break;
85         case EPERM:
86             jniThrowException(env, "java/lang/SecurityException", "No permission to modify given thread");
87             break;
88         case EACCES:
89             jniThrowException(env, "java/lang/SecurityException", "No permission to set to given group");
90             break;
91         default:
92             jniThrowException(env, "java/lang/RuntimeException", "Unknown error");
93             break;
94     }
95 }
96 
android_os_Process_myPid(JNIEnv * env,jobject clazz)97 jint android_os_Process_myPid(JNIEnv* env, jobject clazz)
98 {
99     return getpid();
100 }
101 
android_os_Process_myUid(JNIEnv * env,jobject clazz)102 jint android_os_Process_myUid(JNIEnv* env, jobject clazz)
103 {
104     return getuid();
105 }
106 
android_os_Process_myTid(JNIEnv * env,jobject clazz)107 jint android_os_Process_myTid(JNIEnv* env, jobject clazz)
108 {
109     return androidGetTid();
110 }
111 
android_os_Process_getUidForName(JNIEnv * env,jobject clazz,jstring name)112 jint android_os_Process_getUidForName(JNIEnv* env, jobject clazz, jstring name)
113 {
114     if (name == NULL) {
115         jniThrowNullPointerException(env, NULL);
116         return -1;
117     }
118 
119     const jchar* str16 = env->GetStringCritical(name, 0);
120     String8 name8;
121     if (str16) {
122         name8 = String8(str16, env->GetStringLength(name));
123         env->ReleaseStringCritical(name, str16);
124     }
125 
126     const size_t N = name8.size();
127     if (N > 0) {
128         const char* str = name8.string();
129         for (size_t i=0; i<N; i++) {
130             if (str[i] < '0' || str[i] > '9') {
131                 struct passwd* pwd = getpwnam(str);
132                 if (pwd == NULL) {
133                     return -1;
134                 }
135                 return pwd->pw_uid;
136             }
137         }
138         return atoi(str);
139     }
140     return -1;
141 }
142 
android_os_Process_getGidForName(JNIEnv * env,jobject clazz,jstring name)143 jint android_os_Process_getGidForName(JNIEnv* env, jobject clazz, jstring name)
144 {
145     if (name == NULL) {
146         jniThrowNullPointerException(env, NULL);
147         return -1;
148     }
149 
150     const jchar* str16 = env->GetStringCritical(name, 0);
151     String8 name8;
152     if (str16) {
153         name8 = String8(str16, env->GetStringLength(name));
154         env->ReleaseStringCritical(name, str16);
155     }
156 
157     const size_t N = name8.size();
158     if (N > 0) {
159         const char* str = name8.string();
160         for (size_t i=0; i<N; i++) {
161             if (str[i] < '0' || str[i] > '9') {
162                 struct group* grp = getgrnam(str);
163                 if (grp == NULL) {
164                     return -1;
165                 }
166                 return grp->gr_gid;
167             }
168         }
169         return atoi(str);
170     }
171     return -1;
172 }
173 
android_os_Process_setThreadGroup(JNIEnv * env,jobject clazz,int tid,jint grp)174 void android_os_Process_setThreadGroup(JNIEnv* env, jobject clazz, int tid, jint grp)
175 {
176     ALOGV("%s tid=%d grp=%d", __func__, tid, grp);
177     SchedPolicy sp = (SchedPolicy) grp;
178     int res = set_sched_policy(tid, sp);
179     if (res != NO_ERROR) {
180         signalExceptionForGroupError(env, -res);
181     }
182 }
183 
android_os_Process_setProcessGroup(JNIEnv * env,jobject clazz,int pid,jint grp)184 void android_os_Process_setProcessGroup(JNIEnv* env, jobject clazz, int pid, jint grp)
185 {
186     ALOGV("%s pid=%d grp=%d", __func__, pid, grp);
187     DIR *d;
188     FILE *fp;
189     char proc_path[255];
190     struct dirent *de;
191 
192     if ((grp == SP_FOREGROUND) || (grp > SP_MAX)) {
193         signalExceptionForGroupError(env, EINVAL);
194         return;
195     }
196 
197     bool isDefault = false;
198     if (grp < 0) {
199         grp = SP_FOREGROUND;
200         isDefault = true;
201     }
202     SchedPolicy sp = (SchedPolicy) grp;
203 
204 #if POLICY_DEBUG
205     char cmdline[32];
206     int fd;
207 
208     strcpy(cmdline, "unknown");
209 
210     sprintf(proc_path, "/proc/%d/cmdline", pid);
211     fd = open(proc_path, O_RDONLY);
212     if (fd >= 0) {
213         int rc = read(fd, cmdline, sizeof(cmdline)-1);
214         cmdline[rc] = 0;
215         close(fd);
216     }
217 
218     if (sp == SP_BACKGROUND) {
219         ALOGD("setProcessGroup: vvv pid %d (%s)", pid, cmdline);
220     } else {
221         ALOGD("setProcessGroup: ^^^ pid %d (%s)", pid, cmdline);
222     }
223 #endif
224     sprintf(proc_path, "/proc/%d/task", pid);
225     if (!(d = opendir(proc_path))) {
226         // If the process exited on us, don't generate an exception
227         if (errno != ENOENT)
228             signalExceptionForGroupError(env, errno);
229         return;
230     }
231 
232     while ((de = readdir(d))) {
233         int t_pid;
234         int t_pri;
235 
236         if (de->d_name[0] == '.')
237             continue;
238         t_pid = atoi(de->d_name);
239 
240         if (!t_pid) {
241             ALOGE("Error getting pid for '%s'\n", de->d_name);
242             continue;
243         }
244 
245         t_pri = getpriority(PRIO_PROCESS, t_pid);
246 
247         if (t_pri <= ANDROID_PRIORITY_AUDIO) {
248             int scheduler = sched_getscheduler(t_pid);
249             if ((scheduler == SCHED_FIFO) || (scheduler == SCHED_RR)) {
250                 // This task wants to stay in it's current audio group so it can keep it's budget
251                 continue;
252             }
253         }
254 
255         if (isDefault) {
256             if (t_pri >= ANDROID_PRIORITY_BACKGROUND) {
257                 // This task wants to stay at background
258                 continue;
259             }
260         }
261 
262         int err = set_sched_policy(t_pid, sp);
263         if (err != NO_ERROR) {
264             signalExceptionForGroupError(env, -err);
265             break;
266         }
267     }
268     closedir(d);
269 }
270 
android_os_Process_setCanSelfBackground(JNIEnv * env,jobject clazz,jboolean bgOk)271 static void android_os_Process_setCanSelfBackground(JNIEnv* env, jobject clazz, jboolean bgOk) {
272     // Establishes the calling thread as illegal to put into the background.
273     // Typically used only for the system process's main looper.
274 #if GUARD_THREAD_PRIORITY
275     ALOGV("Process.setCanSelfBackground(%d) : tid=%d", bgOk, androidGetTid());
276     {
277         Mutex::Autolock _l(gKeyCreateMutex);
278         if (gBgKey == -1) {
279             pthread_key_create(&gBgKey, NULL);
280         }
281     }
282 
283     // inverted:  not-okay, we set a sentinel value
284     pthread_setspecific(gBgKey, (void*)(bgOk ? 0 : 0xbaad));
285 #endif
286 }
287 
android_os_Process_setThreadScheduler(JNIEnv * env,jclass clazz,jint tid,jint policy,jint pri)288 void android_os_Process_setThreadScheduler(JNIEnv* env, jclass clazz,
289                                               jint tid, jint policy, jint pri)
290 {
291 #ifdef HAVE_SCHED_SETSCHEDULER
292     struct sched_param param;
293     param.sched_priority = pri;
294     int rc = sched_setscheduler(tid, policy, &param);
295     if (rc) {
296         signalExceptionForPriorityError(env, errno);
297     }
298 #else
299     signalExceptionForPriorityError(env, ENOSYS);
300 #endif
301 }
302 
android_os_Process_setThreadPriority(JNIEnv * env,jobject clazz,jint pid,jint pri)303 void android_os_Process_setThreadPriority(JNIEnv* env, jobject clazz,
304                                               jint pid, jint pri)
305 {
306 #if GUARD_THREAD_PRIORITY
307     // if we're putting the current thread into the background, check the TLS
308     // to make sure this thread isn't guarded.  If it is, raise an exception.
309     if (pri >= ANDROID_PRIORITY_BACKGROUND) {
310         if (pid == androidGetTid()) {
311             void* bgOk = pthread_getspecific(gBgKey);
312             if (bgOk == ((void*)0xbaad)) {
313                 ALOGE("Thread marked fg-only put self in background!");
314                 jniThrowException(env, "java/lang/SecurityException", "May not put this thread into background");
315                 return;
316             }
317         }
318     }
319 #endif
320 
321     int rc = androidSetThreadPriority(pid, pri);
322     if (rc != 0) {
323         if (rc == INVALID_OPERATION) {
324             signalExceptionForPriorityError(env, errno);
325         } else {
326             signalExceptionForGroupError(env, errno);
327         }
328     }
329 
330     //ALOGI("Setting priority of %d: %d, getpriority returns %d\n",
331     //     pid, pri, getpriority(PRIO_PROCESS, pid));
332 }
333 
android_os_Process_setCallingThreadPriority(JNIEnv * env,jobject clazz,jint pri)334 void android_os_Process_setCallingThreadPriority(JNIEnv* env, jobject clazz,
335                                                         jint pri)
336 {
337     jint tid = android_os_Process_myTid(env, clazz);
338     android_os_Process_setThreadPriority(env, clazz, tid, pri);
339 }
340 
android_os_Process_getThreadPriority(JNIEnv * env,jobject clazz,jint pid)341 jint android_os_Process_getThreadPriority(JNIEnv* env, jobject clazz,
342                                               jint pid)
343 {
344     errno = 0;
345     jint pri = getpriority(PRIO_PROCESS, pid);
346     if (errno != 0) {
347         signalExceptionForPriorityError(env, errno);
348     }
349     //ALOGI("Returning priority of %d: %d\n", pid, pri);
350     return pri;
351 }
352 
android_os_Process_setOomAdj(JNIEnv * env,jobject clazz,jint pid,jint adj)353 jboolean android_os_Process_setOomAdj(JNIEnv* env, jobject clazz,
354                                       jint pid, jint adj)
355 {
356 #ifdef HAVE_OOM_ADJ
357     char text[64];
358     sprintf(text, "/proc/%d/oom_adj", pid);
359     int fd = open(text, O_WRONLY);
360     if (fd >= 0) {
361         sprintf(text, "%d", adj);
362         write(fd, text, strlen(text));
363         close(fd);
364     }
365     return true;
366 #endif
367     return false;
368 }
369 
android_os_Process_setArgV0(JNIEnv * env,jobject clazz,jstring name)370 void android_os_Process_setArgV0(JNIEnv* env, jobject clazz, jstring name)
371 {
372     if (name == NULL) {
373         jniThrowNullPointerException(env, NULL);
374         return;
375     }
376 
377     const jchar* str = env->GetStringCritical(name, 0);
378     String8 name8;
379     if (str) {
380         name8 = String8(str, env->GetStringLength(name));
381         env->ReleaseStringCritical(name, str);
382     }
383 
384     if (name8.size() > 0) {
385         ProcessState::self()->setArgV0(name8.string());
386     }
387 }
388 
android_os_Process_setUid(JNIEnv * env,jobject clazz,jint uid)389 jint android_os_Process_setUid(JNIEnv* env, jobject clazz, jint uid)
390 {
391     return setuid(uid) == 0 ? 0 : errno;
392 }
393 
android_os_Process_setGid(JNIEnv * env,jobject clazz,jint uid)394 jint android_os_Process_setGid(JNIEnv* env, jobject clazz, jint uid)
395 {
396     return setgid(uid) == 0 ? 0 : errno;
397 }
398 
pid_compare(const void * v1,const void * v2)399 static int pid_compare(const void* v1, const void* v2)
400 {
401     //ALOGI("Compare %d vs %d\n", *((const jint*)v1), *((const jint*)v2));
402     return *((const jint*)v1) - *((const jint*)v2);
403 }
404 
getFreeMemoryImpl(const char * const sums[],const int sumsLen[],int num)405 static jlong getFreeMemoryImpl(const char* const sums[], const int sumsLen[], int num)
406 {
407     int fd = open("/proc/meminfo", O_RDONLY);
408 
409     if (fd < 0) {
410         ALOGW("Unable to open /proc/meminfo");
411         return -1;
412     }
413 
414     char buffer[256];
415     const int len = read(fd, buffer, sizeof(buffer)-1);
416     close(fd);
417 
418     if (len < 0) {
419         ALOGW("Unable to read /proc/meminfo");
420         return -1;
421     }
422     buffer[len] = 0;
423 
424     int numFound = 0;
425     jlong mem = 0;
426 
427     char* p = buffer;
428     while (*p && numFound < num) {
429         int i = 0;
430         while (sums[i]) {
431             if (strncmp(p, sums[i], sumsLen[i]) == 0) {
432                 p += sumsLen[i];
433                 while (*p == ' ') p++;
434                 char* num = p;
435                 while (*p >= '0' && *p <= '9') p++;
436                 if (*p != 0) {
437                     *p = 0;
438                     p++;
439                     if (*p == 0) p--;
440                 }
441                 mem += atoll(num) * 1024;
442                 numFound++;
443                 break;
444             }
445             i++;
446         }
447         p++;
448     }
449 
450     return numFound > 0 ? mem : -1;
451 }
452 
android_os_Process_getFreeMemory(JNIEnv * env,jobject clazz)453 static jlong android_os_Process_getFreeMemory(JNIEnv* env, jobject clazz)
454 {
455     static const char* const sums[] = { "MemFree:", "Cached:", NULL };
456     static const int sumsLen[] = { strlen("MemFree:"), strlen("Cached:"), 0 };
457     return getFreeMemoryImpl(sums, sumsLen, 2);
458 }
459 
android_os_Process_getTotalMemory(JNIEnv * env,jobject clazz)460 static jlong android_os_Process_getTotalMemory(JNIEnv* env, jobject clazz)
461 {
462     static const char* const sums[] = { "MemTotal:", NULL };
463     static const int sumsLen[] = { strlen("MemTotal:"), 0 };
464     return getFreeMemoryImpl(sums, sumsLen, 1);
465 }
466 
android_os_Process_readProcLines(JNIEnv * env,jobject clazz,jstring fileStr,jobjectArray reqFields,jlongArray outFields)467 void android_os_Process_readProcLines(JNIEnv* env, jobject clazz, jstring fileStr,
468                                       jobjectArray reqFields, jlongArray outFields)
469 {
470     //ALOGI("getMemInfo: %p %p", reqFields, outFields);
471 
472     if (fileStr == NULL || reqFields == NULL || outFields == NULL) {
473         jniThrowNullPointerException(env, NULL);
474         return;
475     }
476 
477     const char* file8 = env->GetStringUTFChars(fileStr, NULL);
478     if (file8 == NULL) {
479         return;
480     }
481     String8 file(file8);
482     env->ReleaseStringUTFChars(fileStr, file8);
483 
484     jsize count = env->GetArrayLength(reqFields);
485     if (count > env->GetArrayLength(outFields)) {
486         jniThrowException(env, "java/lang/IllegalArgumentException", "Array lengths differ");
487         return;
488     }
489 
490     Vector<String8> fields;
491     int i;
492 
493     for (i=0; i<count; i++) {
494         jobject obj = env->GetObjectArrayElement(reqFields, i);
495         if (obj != NULL) {
496             const char* str8 = env->GetStringUTFChars((jstring)obj, NULL);
497             //ALOGI("String at %d: %p = %s", i, obj, str8);
498             if (str8 == NULL) {
499                 jniThrowNullPointerException(env, "Element in reqFields");
500                 return;
501             }
502             fields.add(String8(str8));
503             env->ReleaseStringUTFChars((jstring)obj, str8);
504         } else {
505             jniThrowNullPointerException(env, "Element in reqFields");
506             return;
507         }
508     }
509 
510     jlong* sizesArray = env->GetLongArrayElements(outFields, 0);
511     if (sizesArray == NULL) {
512         return;
513     }
514 
515     //ALOGI("Clearing %d sizes", count);
516     for (i=0; i<count; i++) {
517         sizesArray[i] = 0;
518     }
519 
520     int fd = open(file.string(), O_RDONLY);
521 
522     if (fd >= 0) {
523         const size_t BUFFER_SIZE = 2048;
524         char* buffer = (char*)malloc(BUFFER_SIZE);
525         int len = read(fd, buffer, BUFFER_SIZE-1);
526         close(fd);
527 
528         if (len < 0) {
529             ALOGW("Unable to read %s", file.string());
530             len = 0;
531         }
532         buffer[len] = 0;
533 
534         int foundCount = 0;
535 
536         char* p = buffer;
537         while (*p && foundCount < count) {
538             bool skipToEol = true;
539             //ALOGI("Parsing at: %s", p);
540             for (i=0; i<count; i++) {
541                 const String8& field = fields[i];
542                 if (strncmp(p, field.string(), field.length()) == 0) {
543                     p += field.length();
544                     while (*p == ' ' || *p == '\t') p++;
545                     char* num = p;
546                     while (*p >= '0' && *p <= '9') p++;
547                     skipToEol = *p != '\n';
548                     if (*p != 0) {
549                         *p = 0;
550                         p++;
551                     }
552                     char* end;
553                     sizesArray[i] = strtoll(num, &end, 10);
554                     //ALOGI("Field %s = %d", field.string(), sizesArray[i]);
555                     foundCount++;
556                     break;
557                 }
558             }
559             if (skipToEol) {
560                 while (*p && *p != '\n') {
561                     p++;
562                 }
563                 if (*p == '\n') {
564                     p++;
565                 }
566             }
567         }
568 
569         free(buffer);
570     } else {
571         ALOGW("Unable to open %s", file.string());
572     }
573 
574     //ALOGI("Done!");
575     env->ReleaseLongArrayElements(outFields, sizesArray, 0);
576 }
577 
android_os_Process_getPids(JNIEnv * env,jobject clazz,jstring file,jintArray lastArray)578 jintArray android_os_Process_getPids(JNIEnv* env, jobject clazz,
579                                      jstring file, jintArray lastArray)
580 {
581     if (file == NULL) {
582         jniThrowNullPointerException(env, NULL);
583         return NULL;
584     }
585 
586     const char* file8 = env->GetStringUTFChars(file, NULL);
587     if (file8 == NULL) {
588         jniThrowException(env, "java/lang/OutOfMemoryError", NULL);
589         return NULL;
590     }
591 
592     DIR* dirp = opendir(file8);
593 
594     env->ReleaseStringUTFChars(file, file8);
595 
596     if(dirp == NULL) {
597         return NULL;
598     }
599 
600     jsize curCount = 0;
601     jint* curData = NULL;
602     if (lastArray != NULL) {
603         curCount = env->GetArrayLength(lastArray);
604         curData = env->GetIntArrayElements(lastArray, 0);
605     }
606 
607     jint curPos = 0;
608 
609     struct dirent* entry;
610     while ((entry=readdir(dirp)) != NULL) {
611         const char* p = entry->d_name;
612         while (*p) {
613             if (*p < '0' || *p > '9') break;
614             p++;
615         }
616         if (*p != 0) continue;
617 
618         char* end;
619         int pid = strtol(entry->d_name, &end, 10);
620         //ALOGI("File %s pid=%d\n", entry->d_name, pid);
621         if (curPos >= curCount) {
622             jsize newCount = (curCount == 0) ? 10 : (curCount*2);
623             jintArray newArray = env->NewIntArray(newCount);
624             if (newArray == NULL) {
625                 closedir(dirp);
626                 jniThrowException(env, "java/lang/OutOfMemoryError", NULL);
627                 return NULL;
628             }
629             jint* newData = env->GetIntArrayElements(newArray, 0);
630             if (curData != NULL) {
631                 memcpy(newData, curData, sizeof(jint)*curCount);
632                 env->ReleaseIntArrayElements(lastArray, curData, 0);
633             }
634             lastArray = newArray;
635             curCount = newCount;
636             curData = newData;
637         }
638 
639         curData[curPos] = pid;
640         curPos++;
641     }
642 
643     closedir(dirp);
644 
645     if (curData != NULL && curPos > 0) {
646         qsort(curData, curPos, sizeof(jint), pid_compare);
647     }
648 
649     while (curPos < curCount) {
650         curData[curPos] = -1;
651         curPos++;
652     }
653 
654     if (curData != NULL) {
655         env->ReleaseIntArrayElements(lastArray, curData, 0);
656     }
657 
658     return lastArray;
659 }
660 
661 enum {
662     PROC_TERM_MASK = 0xff,
663     PROC_ZERO_TERM = 0,
664     PROC_SPACE_TERM = ' ',
665     PROC_COMBINE = 0x100,
666     PROC_PARENS = 0x200,
667     PROC_OUT_STRING = 0x1000,
668     PROC_OUT_LONG = 0x2000,
669     PROC_OUT_FLOAT = 0x4000,
670 };
671 
android_os_Process_parseProcLineArray(JNIEnv * env,jobject clazz,char * buffer,jint startIndex,jint endIndex,jintArray format,jobjectArray outStrings,jlongArray outLongs,jfloatArray outFloats)672 jboolean android_os_Process_parseProcLineArray(JNIEnv* env, jobject clazz,
673         char* buffer, jint startIndex, jint endIndex, jintArray format,
674         jobjectArray outStrings, jlongArray outLongs, jfloatArray outFloats)
675 {
676 
677     const jsize NF = env->GetArrayLength(format);
678     const jsize NS = outStrings ? env->GetArrayLength(outStrings) : 0;
679     const jsize NL = outLongs ? env->GetArrayLength(outLongs) : 0;
680     const jsize NR = outFloats ? env->GetArrayLength(outFloats) : 0;
681 
682     jint* formatData = env->GetIntArrayElements(format, 0);
683     jlong* longsData = outLongs ?
684         env->GetLongArrayElements(outLongs, 0) : NULL;
685     jfloat* floatsData = outFloats ?
686         env->GetFloatArrayElements(outFloats, 0) : NULL;
687     if (formatData == NULL || (NL > 0 && longsData == NULL)
688             || (NR > 0 && floatsData == NULL)) {
689         if (formatData != NULL) {
690             env->ReleaseIntArrayElements(format, formatData, 0);
691         }
692         if (longsData != NULL) {
693             env->ReleaseLongArrayElements(outLongs, longsData, 0);
694         }
695         if (floatsData != NULL) {
696             env->ReleaseFloatArrayElements(outFloats, floatsData, 0);
697         }
698         jniThrowException(env, "java/lang/OutOfMemoryError", NULL);
699         return JNI_FALSE;
700     }
701 
702     jsize i = startIndex;
703     jsize di = 0;
704 
705     jboolean res = JNI_TRUE;
706 
707     for (jsize fi=0; fi<NF; fi++) {
708         const jint mode = formatData[fi];
709         if ((mode&PROC_PARENS) != 0) {
710             i++;
711         }
712         const char term = (char)(mode&PROC_TERM_MASK);
713         const jsize start = i;
714         if (i >= endIndex) {
715             res = JNI_FALSE;
716             break;
717         }
718 
719         jsize end = -1;
720         if ((mode&PROC_PARENS) != 0) {
721             while (buffer[i] != ')' && i < endIndex) {
722                 i++;
723             }
724             end = i;
725             i++;
726         }
727         while (buffer[i] != term && i < endIndex) {
728             i++;
729         }
730         if (end < 0) {
731             end = i;
732         }
733 
734         if (i < endIndex) {
735             i++;
736             if ((mode&PROC_COMBINE) != 0) {
737                 while (buffer[i] == term && i < endIndex) {
738                     i++;
739                 }
740             }
741         }
742 
743         //ALOGI("Field %d: %d-%d dest=%d mode=0x%x\n", i, start, end, di, mode);
744 
745         if ((mode&(PROC_OUT_FLOAT|PROC_OUT_LONG|PROC_OUT_STRING)) != 0) {
746             char c = buffer[end];
747             buffer[end] = 0;
748             if ((mode&PROC_OUT_FLOAT) != 0 && di < NR) {
749                 char* end;
750                 floatsData[di] = strtof(buffer+start, &end);
751             }
752             if ((mode&PROC_OUT_LONG) != 0 && di < NL) {
753                 char* end;
754                 longsData[di] = strtoll(buffer+start, &end, 10);
755             }
756             if ((mode&PROC_OUT_STRING) != 0 && di < NS) {
757                 jstring str = env->NewStringUTF(buffer+start);
758                 env->SetObjectArrayElement(outStrings, di, str);
759             }
760             buffer[end] = c;
761             di++;
762         }
763     }
764 
765     env->ReleaseIntArrayElements(format, formatData, 0);
766     if (longsData != NULL) {
767         env->ReleaseLongArrayElements(outLongs, longsData, 0);
768     }
769     if (floatsData != NULL) {
770         env->ReleaseFloatArrayElements(outFloats, floatsData, 0);
771     }
772 
773     return res;
774 }
775 
android_os_Process_parseProcLine(JNIEnv * env,jobject clazz,jbyteArray buffer,jint startIndex,jint endIndex,jintArray format,jobjectArray outStrings,jlongArray outLongs,jfloatArray outFloats)776 jboolean android_os_Process_parseProcLine(JNIEnv* env, jobject clazz,
777         jbyteArray buffer, jint startIndex, jint endIndex, jintArray format,
778         jobjectArray outStrings, jlongArray outLongs, jfloatArray outFloats)
779 {
780         jbyte* bufferArray = env->GetByteArrayElements(buffer, NULL);
781 
782         jboolean result = android_os_Process_parseProcLineArray(env, clazz,
783                 (char*) bufferArray, startIndex, endIndex, format, outStrings,
784                 outLongs, outFloats);
785 
786         env->ReleaseByteArrayElements(buffer, bufferArray, 0);
787 
788         return result;
789 }
790 
android_os_Process_readProcFile(JNIEnv * env,jobject clazz,jstring file,jintArray format,jobjectArray outStrings,jlongArray outLongs,jfloatArray outFloats)791 jboolean android_os_Process_readProcFile(JNIEnv* env, jobject clazz,
792         jstring file, jintArray format, jobjectArray outStrings,
793         jlongArray outLongs, jfloatArray outFloats)
794 {
795     if (file == NULL || format == NULL) {
796         jniThrowNullPointerException(env, NULL);
797         return JNI_FALSE;
798     }
799 
800     const char* file8 = env->GetStringUTFChars(file, NULL);
801     if (file8 == NULL) {
802         jniThrowException(env, "java/lang/OutOfMemoryError", NULL);
803         return JNI_FALSE;
804     }
805     int fd = open(file8, O_RDONLY);
806     env->ReleaseStringUTFChars(file, file8);
807 
808     if (fd < 0) {
809         //ALOGW("Unable to open process file: %s\n", file8);
810         return JNI_FALSE;
811     }
812 
813     char buffer[256];
814     const int len = read(fd, buffer, sizeof(buffer)-1);
815     close(fd);
816 
817     if (len < 0) {
818         //ALOGW("Unable to open process file: %s fd=%d\n", file8, fd);
819         return JNI_FALSE;
820     }
821     buffer[len] = 0;
822 
823     return android_os_Process_parseProcLineArray(env, clazz, buffer, 0, len,
824             format, outStrings, outLongs, outFloats);
825 
826 }
827 
android_os_Process_setApplicationObject(JNIEnv * env,jobject clazz,jobject binderObject)828 void android_os_Process_setApplicationObject(JNIEnv* env, jobject clazz,
829                                              jobject binderObject)
830 {
831     if (binderObject == NULL) {
832         jniThrowNullPointerException(env, NULL);
833         return;
834     }
835 
836     sp<IBinder> binder = ibinderForJavaObject(env, binderObject);
837 }
838 
android_os_Process_sendSignal(JNIEnv * env,jobject clazz,jint pid,jint sig)839 void android_os_Process_sendSignal(JNIEnv* env, jobject clazz, jint pid, jint sig)
840 {
841     if (pid > 0) {
842         ALOGI("Sending signal. PID: %d SIG: %d", pid, sig);
843         kill(pid, sig);
844     }
845 }
846 
android_os_Process_sendSignalQuiet(JNIEnv * env,jobject clazz,jint pid,jint sig)847 void android_os_Process_sendSignalQuiet(JNIEnv* env, jobject clazz, jint pid, jint sig)
848 {
849     if (pid > 0) {
850         kill(pid, sig);
851     }
852 }
853 
android_os_Process_getElapsedCpuTime(JNIEnv * env,jobject clazz)854 static jlong android_os_Process_getElapsedCpuTime(JNIEnv* env, jobject clazz)
855 {
856     struct timespec ts;
857 
858     int res = clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &ts);
859 
860     if (res != 0) {
861         return (jlong) 0;
862     }
863 
864     nsecs_t when = seconds_to_nanoseconds(ts.tv_sec) + ts.tv_nsec;
865     return (jlong) nanoseconds_to_milliseconds(when);
866 }
867 
android_os_Process_getPss(JNIEnv * env,jobject clazz,jint pid)868 static jlong android_os_Process_getPss(JNIEnv* env, jobject clazz, jint pid)
869 {
870     char filename[64];
871 
872     snprintf(filename, sizeof(filename), "/proc/%d/smaps", pid);
873 
874     FILE * file = fopen(filename, "r");
875     if (!file) {
876         return (jlong) -1;
877     }
878 
879     // Tally up all of the Pss from the various maps
880     char line[256];
881     jlong pss = 0;
882     while (fgets(line, sizeof(line), file)) {
883         jlong v;
884         if (sscanf(line, "Pss: %lld kB", &v) == 1) {
885             pss += v;
886         }
887     }
888 
889     fclose(file);
890 
891     // Return the Pss value in bytes, not kilobytes
892     return pss * 1024;
893 }
894 
android_os_Process_getPidsForCommands(JNIEnv * env,jobject clazz,jobjectArray commandNames)895 jintArray android_os_Process_getPidsForCommands(JNIEnv* env, jobject clazz,
896         jobjectArray commandNames)
897 {
898     if (commandNames == NULL) {
899         jniThrowNullPointerException(env, NULL);
900         return NULL;
901     }
902 
903     Vector<String8> commands;
904 
905     jsize count = env->GetArrayLength(commandNames);
906 
907     for (int i=0; i<count; i++) {
908         jobject obj = env->GetObjectArrayElement(commandNames, i);
909         if (obj != NULL) {
910             const char* str8 = env->GetStringUTFChars((jstring)obj, NULL);
911             if (str8 == NULL) {
912                 jniThrowNullPointerException(env, "Element in commandNames");
913                 return NULL;
914             }
915             commands.add(String8(str8));
916             env->ReleaseStringUTFChars((jstring)obj, str8);
917         } else {
918             jniThrowNullPointerException(env, "Element in commandNames");
919             return NULL;
920         }
921     }
922 
923     Vector<jint> pids;
924 
925     DIR *proc = opendir("/proc");
926     if (proc == NULL) {
927         fprintf(stderr, "/proc: %s\n", strerror(errno));
928         return NULL;
929     }
930 
931     struct dirent *d;
932     while ((d = readdir(proc))) {
933         int pid = atoi(d->d_name);
934         if (pid <= 0) continue;
935 
936         char path[PATH_MAX];
937         char data[PATH_MAX];
938         snprintf(path, sizeof(path), "/proc/%d/cmdline", pid);
939 
940         int fd = open(path, O_RDONLY);
941         if (fd < 0) {
942             continue;
943         }
944         const int len = read(fd, data, sizeof(data)-1);
945         close(fd);
946 
947         if (len < 0) {
948             continue;
949         }
950         data[len] = 0;
951 
952         for (int i=0; i<len; i++) {
953             if (data[i] == ' ') {
954                 data[i] = 0;
955                 break;
956             }
957         }
958 
959         for (size_t i=0; i<commands.size(); i++) {
960             if (commands[i] == data) {
961                 pids.add(pid);
962                 break;
963             }
964         }
965     }
966 
967     closedir(proc);
968 
969     jintArray pidArray = env->NewIntArray(pids.size());
970     if (pidArray == NULL) {
971         jniThrowException(env, "java/lang/OutOfMemoryError", NULL);
972         return NULL;
973     }
974 
975     if (pids.size() > 0) {
976         env->SetIntArrayRegion(pidArray, 0, pids.size(), pids.array());
977     }
978 
979     return pidArray;
980 }
981 
982 static const JNINativeMethod methods[] = {
983     {"myPid",       "()I", (void*)android_os_Process_myPid},
984     {"myTid",       "()I", (void*)android_os_Process_myTid},
985     {"myUid",       "()I", (void*)android_os_Process_myUid},
986     {"getUidForName",       "(Ljava/lang/String;)I", (void*)android_os_Process_getUidForName},
987     {"getGidForName",       "(Ljava/lang/String;)I", (void*)android_os_Process_getGidForName},
988     {"setThreadPriority",   "(II)V", (void*)android_os_Process_setThreadPriority},
989     {"setThreadScheduler",  "(III)V", (void*)android_os_Process_setThreadScheduler},
990     {"setCanSelfBackground", "(Z)V", (void*)android_os_Process_setCanSelfBackground},
991     {"setThreadPriority",   "(I)V", (void*)android_os_Process_setCallingThreadPriority},
992     {"getThreadPriority",   "(I)I", (void*)android_os_Process_getThreadPriority},
993     {"setThreadGroup",      "(II)V", (void*)android_os_Process_setThreadGroup},
994     {"setProcessGroup",      "(II)V", (void*)android_os_Process_setProcessGroup},
995     {"setOomAdj",   "(II)Z", (void*)android_os_Process_setOomAdj},
996     {"setArgV0",    "(Ljava/lang/String;)V", (void*)android_os_Process_setArgV0},
997     {"setUid", "(I)I", (void*)android_os_Process_setUid},
998     {"setGid", "(I)I", (void*)android_os_Process_setGid},
999     {"sendSignal", "(II)V", (void*)android_os_Process_sendSignal},
1000     {"sendSignalQuiet", "(II)V", (void*)android_os_Process_sendSignalQuiet},
1001     {"getFreeMemory", "()J", (void*)android_os_Process_getFreeMemory},
1002     {"getTotalMemory", "()J", (void*)android_os_Process_getTotalMemory},
1003     {"readProcLines", "(Ljava/lang/String;[Ljava/lang/String;[J)V", (void*)android_os_Process_readProcLines},
1004     {"getPids", "(Ljava/lang/String;[I)[I", (void*)android_os_Process_getPids},
1005     {"readProcFile", "(Ljava/lang/String;[I[Ljava/lang/String;[J[F)Z", (void*)android_os_Process_readProcFile},
1006     {"parseProcLine", "([BII[I[Ljava/lang/String;[J[F)Z", (void*)android_os_Process_parseProcLine},
1007     {"getElapsedCpuTime", "()J", (void*)android_os_Process_getElapsedCpuTime},
1008     {"getPss", "(I)J", (void*)android_os_Process_getPss},
1009     {"getPidsForCommands", "([Ljava/lang/String;)[I", (void*)android_os_Process_getPidsForCommands},
1010     //{"setApplicationObject", "(Landroid/os/IBinder;)V", (void*)android_os_Process_setApplicationObject},
1011 };
1012 
1013 const char* const kProcessPathName = "android/os/Process";
1014 
register_android_os_Process(JNIEnv * env)1015 int register_android_os_Process(JNIEnv* env)
1016 {
1017     return AndroidRuntime::registerNativeMethods(
1018         env, kProcessPathName,
1019         methods, NELEM(methods));
1020 }
1021