diff options
author | The Android Open Source Project <initial-contribution@android.com> | 2009-03-03 18:29:04 -0800 |
---|---|---|
committer | The Android Open Source Project <initial-contribution@android.com> | 2009-03-03 18:29:04 -0800 |
commit | 19ddb4b1680760e2d6863c3003976882ebd9d0fa (patch) | |
tree | 4b825dc642cb6eb9a060e54bf8d69288fbee4904 /librank/librank.c | |
parent | dbdf1561ad9b4c6100fb93028929ece02bf66a74 (diff) | |
download | extras-19ddb4b1680760e2d6863c3003976882ebd9d0fa.tar.gz |
auto import from //depot/cupcake/@135843
Diffstat (limited to 'librank/librank.c')
-rw-r--r-- | librank/librank.c | 349 |
1 files changed, 0 insertions, 349 deletions
diff --git a/librank/librank.c b/librank/librank.c deleted file mode 100644 index 13b20974..00000000 --- a/librank/librank.c +++ /dev/null @@ -1,349 +0,0 @@ -/* - * Copyright (C) 2008 The Android Open Source Project - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -#include <assert.h> -#include <dirent.h> -#include <errno.h> -#include <stdlib.h> -#include <string.h> -#include <sys/types.h> -#include <unistd.h> - -#include <pagemap/pagemap.h> - -#define MAX_CMDLINE 256 - -struct process_info { - pid_t pid; - char cmdline[MAX_CMDLINE]; -}; - -struct mapping_info { - struct process_info *proc; - pm_memusage_t usage; -}; - -struct library_info { - struct library_info *next; - char *name; - struct mapping_info **mappings; - int mappings_count; - int mappings_size; - pm_memusage_t total_usage; -}; - -static void usage(char *myname); -static int getprocname(pid_t pid, char *buf, size_t len); -static int numcmp(long long a, long long b); -static int licmp(const void *a, const void *b); - -char *library_name_blacklist[] = { "[heap]", "[stack]", "", NULL }; - -#define declare_sort(field) \ - static int sort_by_ ## field (const void *a, const void *b) - -declare_sort(vss); -declare_sort(rss); -declare_sort(pss); -declare_sort(uss); - -#define INIT_LIBRARIES 16 -#define INIT_MAPPINGS 4 - -static int order; - -struct library_info **libraries; -int libraries_count; -int libraries_size; - -struct library_info *get_library(char *name) { - int i; - struct library_info *library; - - for (i = 0; library_name_blacklist[i]; i++) - if (!strcmp(name, library_name_blacklist[i])) - return NULL; - - for (i = 0; i < libraries_count; i++) { - if (!strcmp(libraries[i]->name, name)) - return libraries[i]; - } - - if (libraries_count >= libraries_size) { - libraries = realloc(libraries, 2 * libraries_size * sizeof(struct library_info *)); - if (!libraries) { - fprintf(stderr, "Couldn't resize libraries array: %s\n", strerror(errno)); - exit(EXIT_FAILURE); - } - libraries_size = 2 * libraries_size; - } - - library = calloc(1, sizeof(*library)); - if (!library) { - fprintf(stderr, "Couldn't allocate space for library struct: %s\n", strerror(errno)); - exit(EXIT_FAILURE); - } - library->name = malloc(strlen(name) + 1); - if (!library->name) { - fprintf(stderr, "Couldn't allocate space for library name: %s\n", strerror(errno)); - exit(EXIT_FAILURE); - } - strcpy(library->name, name); - library->mappings = malloc(INIT_MAPPINGS * sizeof(struct mapping_info *)); - if (!library->mappings) { - fprintf(stderr, "Couldn't allocate space for library mappings array: %s\n", strerror(errno)); - exit(EXIT_FAILURE); - } - library->mappings_count = 0; library->mappings_size = INIT_MAPPINGS; - pm_memusage_zero(&library->total_usage); - - libraries[libraries_count++] = library; - - return library; -} - -struct mapping_info *get_mapping(struct library_info *library, struct process_info *proc) { - struct mapping_info *mapping; - int i; - - for (i = 0; i < library->mappings_count; i++) { - if (library->mappings[i]->proc == proc) - return library->mappings[i]; - } - - if (library->mappings_count >= library->mappings_size) { - library->mappings = realloc(library->mappings, - 2 * library->mappings_size * sizeof(struct mapping*)); - if (!library->mappings) { - fprintf(stderr, "Couldn't resize mappings array: %s\n", strerror(errno)); - exit(EXIT_FAILURE); - } - library->mappings_size = 2 * library->mappings_size; - } - - mapping = calloc(1, sizeof(*mapping)); - if (!mapping) { - fprintf(stderr, "Couldn't allocate space for mapping struct: %s\n", strerror(errno)); - exit(EXIT_FAILURE); - } - mapping->proc = proc; - pm_memusage_zero(&mapping->usage); - - library->mappings[library->mappings_count++] = mapping; - - return mapping; -} - -struct process_info *get_process(pid_t pid) { - struct process_info *process; - - process = calloc(1, sizeof(*process)); - if (!process) { - fprintf(stderr, "Couldn't allocate space for process struct: %s\n", strerror(errno)); - exit(EXIT_FAILURE); - } - - process->pid = pid; - getprocname(pid, process->cmdline, sizeof(process->cmdline)); - - return process; -} - -int main(int argc, char *argv[]) { - char cmdline[256]; - char *prefix; - size_t prefix_len; - int (*compfn)(const void *a, const void *b); - - pm_kernel_t *ker; - pm_process_t *proc; - - pid_t *pids; - size_t num_procs; - - pm_map_t **maps; - size_t num_maps; - pm_memusage_t map_usage; - - struct library_info *li, **lis; - struct mapping_info *mi, **mis; - struct process_info *pi; - - int i, j, error; - - compfn = &sort_by_pss; - order = -1; - prefix = NULL; - prefix_len = 0; - - for (i = 1; i < argc; i++) { - if (!strcmp(argv[i], "-P")) { - if (i + 1 >= argc) { - fprintf(stderr, "Option -P requires an argument.\n"); - usage(argv[0]); - exit(EXIT_FAILURE); - } - prefix = argv[++i]; - prefix_len = strlen(prefix); - continue; - } - if (!strcmp(argv[i], "-v")) { compfn = &sort_by_vss; continue; } - if (!strcmp(argv[i], "-r")) { compfn = &sort_by_rss; continue; } - if (!strcmp(argv[i], "-p")) { compfn = &sort_by_pss; continue; } - if (!strcmp(argv[i], "-u")) { compfn = &sort_by_uss; continue; } - if (!strcmp(argv[i], "-R")) { order *= -1; continue; } - if (!strcmp(argv[i], "-h")) { usage(argv[0]); exit(0); } - fprintf(stderr, "Invalid argument \"%s\".\n", argv[i]); - usage(argv[0]); - exit(EXIT_FAILURE); - } - - libraries = malloc(INIT_LIBRARIES * sizeof(struct library_info *)); - libraries_count = 0; libraries_size = INIT_LIBRARIES; - - error = pm_kernel_create(&ker); - if (error) { - fprintf(stderr, "Error initializing kernel interface -- " - "does this kernel have pagemap?\n"); - exit(EXIT_FAILURE); - } - - error = pm_kernel_pids(ker, &pids, &num_procs); - if (error) { - fprintf(stderr, "Error listing processes.\n"); - exit(EXIT_FAILURE); - } - - for (i = 0; i < num_procs; i++) { - error = pm_process_create(ker, pids[i], &proc); - if (error) { - fprintf(stderr, "warning: could not create process interface for %d\n", pids[i]); - continue; - } - - pi = get_process(pids[i]); - - error = pm_process_maps(proc, &maps, &num_maps); - if (error) { - fprintf(stderr, "Error listing maps for process %d.\n", proc->pid); - exit(EXIT_FAILURE); - } - - for (j = 0; j < num_maps; j++) { - if (prefix && (strncmp(pm_map_name(maps[j]), prefix, prefix_len))) - continue; - - li = get_library(pm_map_name(maps[j])); - if (!li) - continue; - - mi = get_mapping(li, pi); - - error = pm_map_usage(maps[j], &map_usage); - if (error) { - fprintf(stderr, "Error getting map memory usage of " - "map %s in process %d.\n", - pm_map_name(maps[j]), proc->pid); - exit(EXIT_FAILURE); - } - pm_memusage_add(&mi->usage, &map_usage); - pm_memusage_add(&li->total_usage, &map_usage); - } - } - - printf( " %6s %6s %6s %6s %6s %s\n", "RSStot", "VSS", "RSS", "PSS", "USS", "Name/PID"); - fflush(stdout); - - qsort(libraries, libraries_count, sizeof(libraries[0]), &licmp); - - for (i = 0; i < libraries_count; i++) { - li = libraries[i]; - - printf("%6dK %6s %6s %6s %6s %s\n", li->total_usage.pss / 1024, "", "", "", "", li->name); - fflush(stdout); - - qsort(li->mappings, li->mappings_count, sizeof(li->mappings[0]), compfn); - - for (j = 0; j < li->mappings_count; j++) { - mi = li->mappings[j]; - pi = mi->proc; - printf( " %6s %6dK %6dK %6dK %6dK %s [%d]\n", "", - mi->usage.vss / 1024, - mi->usage.rss / 1024, - mi->usage.pss / 1024, - mi->usage.uss / 1024, - pi->cmdline, - pi->pid); - } - printf("\n"); - fflush(stdout); - } - - return 0; -} - -static void usage(char *myname) { - fprintf(stderr, "Usage: %s [ -P | -L ] [ -v | -r | -p | -u | -h ]\n" - "\n" - "Sort options:\n" - " -v Sort processes by VSS.\n" - " -r Sort processes by RSS.\n" - " -p Sort processes by PSS.\n" - " -u Sort processes by USS.\n" - " (Default sort order is PSS.)\n" - " -P /path Limit libraries displayed to those in path.\n" - " -R Reverse sort order (default is descending).\n" - " -h Display this help screen.\n", - myname); -} - -static int getprocname(pid_t pid, char *buf, size_t len) { - char filename[20]; - FILE *f; - - sprintf(filename, "/proc/%d/cmdline", pid); - f = fopen(filename, "r"); - if (!f) { *buf = '\0'; return 1; } - if (!fgets(buf, len, f)) { *buf = '\0'; return 2; } - fclose(f); - return 0; -} - -static int numcmp(long long a, long long b) { - if (a < b) return -1; - if (a > b) return 1; - return 0; -} - -static int licmp(const void *a, const void *b) { - return order * numcmp( - (*((struct library_info**)a))->total_usage.pss, - (*((struct library_info**)b))->total_usage.pss - ); -} - -#define create_sort(field, compfn) \ - static int sort_by_ ## field (const void *a, const void *b) { \ - return order * compfn( \ - (*((struct mapping_info**)a))->usage.field, \ - (*((struct mapping_info**)b))->usage.field \ - ); \ - } - -create_sort(vss, numcmp) -create_sort(rss, numcmp) -create_sort(pss, numcmp) -create_sort(uss, numcmp) |