summaryrefslogtreecommitdiff
path: root/procmem/procmem.c
diff options
context:
space:
mode:
Diffstat (limited to 'procmem/procmem.c')
-rw-r--r--procmem/procmem.c322
1 files changed, 322 insertions, 0 deletions
diff --git a/procmem/procmem.c b/procmem/procmem.c
new file mode 100644
index 00000000..03c9f51f
--- /dev/null
+++ b/procmem/procmem.c
@@ -0,0 +1,322 @@
+/*
+ * 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 <errno.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+#include <pagemap/pagemap.h>
+
+/* Information about a single mapping */
+struct map_info {
+ pm_map_t *map;
+ pm_memusage_t usage;
+ /* page counts */
+ unsigned long shared_clean;
+ unsigned long shared_dirty;
+ unsigned long private_clean;
+ unsigned long private_dirty;
+};
+
+/* display the help screen */
+static void usage(const char *cmd);
+
+/* qsort compare function to compare maps by PSS */
+int comp_pss(const void *a, const void *b);
+
+int main(int argc, char *argv[]) {
+ pid_t pid;
+
+ /* libpagemap context */
+ pm_kernel_t *ker;
+ int pagesize; /* cached for speed */
+ pm_process_t *proc;
+
+ /* maps and such */
+ pm_map_t **maps; int num_maps;
+
+ struct map_info **mis;
+ struct map_info *mi;
+
+ /* pagemap information */
+ uint64_t *pagemap; int num_pages;
+ unsigned long address; uint64_t mapentry;
+ uint64_t count, flags;
+
+ /* totals */
+ unsigned long total_shared_clean, total_shared_dirty, total_private_clean, total_private_dirty;
+ pm_memusage_t total_usage;
+
+ /* command-line options */
+ int ws;
+#define WS_OFF (0)
+#define WS_ONLY (1)
+#define WS_RESET (2)
+ int (*compfn)(const void *a, const void *b);
+ int hide_zeros;
+
+ /* temporary variables */
+ int i, j;
+ char *endptr;
+ int error;
+
+ if (argc < 2) {
+ usage(argv[0]);
+ exit(EXIT_FAILURE);
+ }
+
+ ws = WS_OFF;
+ compfn = NULL;
+ hide_zeros = 0;
+ for (i = 1; i < argc - 1; i++) {
+ if (!strcmp(argv[i], "-w")) { ws = WS_ONLY; continue; }
+ if (!strcmp(argv[i], "-W")) { ws = WS_RESET; continue; }
+ if (!strcmp(argv[i], "-m")) { compfn = NULL; continue; }
+ if (!strcmp(argv[i], "-p")) { compfn = &comp_pss; continue; }
+ if (!strcmp(argv[i], "-h")) { hide_zeros = 1; continue; }
+ fprintf(stderr, "Invalid argument \"%s\".\n", argv[i]);
+ usage(argv[0]);
+ exit(EXIT_FAILURE);
+ }
+
+ pid = (pid_t)strtol(argv[argc - 1], &endptr, 10);
+ if (*endptr != '\0') {
+ fprintf(stderr, "Invalid PID \"%s\".\n", argv[argc - 1]);
+ exit(EXIT_FAILURE);
+ }
+
+ error = pm_kernel_create(&ker);
+ if (error) {
+ fprintf(stderr, "error creating kernel interface -- "
+ "does this kernel have pagemap?\n");
+ exit(EXIT_FAILURE);
+ }
+
+ pagesize = pm_kernel_pagesize(ker);
+
+ error = pm_process_create(ker, pid, &proc);
+ if (error) {
+ fprintf(stderr, "error creating process interface -- "
+ "does process %d really exist?\n", pid);
+ exit(EXIT_FAILURE);
+ }
+
+ if (ws == WS_RESET) {
+ error = pm_process_workingset(proc, NULL, 1);
+ if (error) {
+ fprintf(stderr, "error resetting working set for process.\n");
+ exit(EXIT_FAILURE);
+ }
+ exit(EXIT_SUCCESS);
+ }
+
+ /* get maps, and allocate our map_info array */
+ error = pm_process_maps(proc, &maps, &num_maps);
+ if (error) {
+ fprintf(stderr, "error listing maps.\n");
+ exit(EXIT_FAILURE);
+ }
+
+ mis = (struct map_info **)calloc(num_maps, sizeof(struct map_info *));
+ if (!mis) {
+ fprintf(stderr, "error allocating map_info array: %s\n", strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+ /* print header */
+ if (ws == WS_ONLY) {
+ printf("%7s %7s %7s %7s %7s %7s %7s %s\n",
+ "WRss", "WPss", "WUss", "WShCl", "WShDi", "WPrCl", "WPrDi", "Name");
+ printf("%7s %7s %7s %7s %7s %7s %7s %s\n",
+ "-------", "-------", "-------", "-------", "-------", "-------", "-------", "");
+ } else {
+ printf("%7s %7s %7s %7s %7s %7s %7s %7s %s\n",
+ "Vss", "Rss", "Pss", "Uss", "ShCl", "ShDi", "PrCl", "PrDi", "Name");
+ printf("%7s %7s %7s %7s %7s %7s %7s %7s %s\n",
+ "-------", "-------", "-------", "-------", "-------", "-------", "-------", "-------", "");
+ }
+
+ /* zero things */
+ pm_memusage_zero(&total_usage);
+ total_shared_clean = total_shared_dirty = total_private_clean = total_private_dirty = 0;
+
+ for (i = 0; i < num_maps; i++) {
+ mi = (struct map_info *)calloc(1, sizeof(struct map_info));
+ if (!mi) {
+ fprintf(stderr, "error allocating map_info: %s\n", strerror(errno));
+ exit(EXIT_FAILURE);
+ }
+
+ mi->map = maps[i];
+
+ /* get, and sum, memory usage */
+
+ if (ws == WS_ONLY)
+ error = pm_map_workingset(mi->map, &mi->usage);
+ else
+ error = pm_map_usage(mi->map, &mi->usage);
+ if (error) {
+ fflush(stdout);
+ fprintf(stderr, "error getting usage for map.\n");
+ continue;
+ }
+
+ pm_memusage_add(&total_usage, &mi->usage);
+
+ /* get, and sum, individual page counts */
+
+ error = pm_map_pagemap(mi->map, &pagemap, &num_pages);
+ if (error) {
+ fflush(stdout);
+ fprintf(stderr, "error getting pagemap for map.\n");
+ continue;
+ }
+
+ mi->shared_clean = mi->shared_dirty = mi->private_clean = mi->private_dirty = 0;
+
+ for (j = 0; j < num_pages; j++) {
+ address = pm_map_start(mi->map) + j * ker->pagesize;
+ mapentry = pagemap[j];
+
+ if (PM_PAGEMAP_PRESENT(mapentry) && !PM_PAGEMAP_SWAPPED(mapentry)) {
+
+ error = pm_kernel_count(ker, PM_PAGEMAP_PFN(mapentry), &count);
+ if (error) {
+ fflush(stdout);
+ fprintf(stderr, "error getting count for frame.\n");
+ }
+
+ error = pm_kernel_flags(ker, PM_PAGEMAP_PFN(mapentry), &flags);
+ if (error) {
+ fflush(stdout);
+ fprintf(stderr, "error getting flags for frame.\n");
+ }
+
+ if ((ws != WS_ONLY) || (flags & PM_PAGE_REFERENCED)) {
+ if (count > 1) {
+ if (flags & PM_PAGE_DIRTY)
+ mi->shared_dirty++;
+ else
+ mi->shared_clean++;
+ } else {
+ if (flags & PM_PAGE_DIRTY)
+ mi->private_dirty++;
+ else
+ mi->private_clean++;
+ }
+ }
+ }
+ }
+
+ total_shared_clean += mi->shared_clean;
+ total_shared_dirty += mi->shared_dirty;
+ total_private_clean += mi->private_clean;
+ total_private_dirty += mi->private_dirty;
+
+ /* add to array */
+ mis[i] = mi;
+ }
+
+ /* sort the array, if requested (compfn == NULL for original order) */
+ if (compfn)
+ qsort(mis, num_maps, sizeof(mis[0]), compfn);
+
+ for (i = 0; i < num_maps; i++) {
+ mi = mis[i];
+
+ if (hide_zeros && !mi->usage.rss)
+ continue;
+
+ if (ws == WS_ONLY) {
+ printf("%6ldK %6ldK %6ldK %6ldK %6ldK %6ldK %6ldK %s\n",
+ (long)mi->usage.rss / 1024,
+ (long)mi->usage.pss / 1024,
+ (long)mi->usage.uss / 1024,
+ mi->shared_clean * pagesize / 1024,
+ mi->shared_dirty * pagesize / 1024,
+ mi->private_clean * pagesize / 1024,
+ mi->private_dirty * pagesize / 1024,
+ pm_map_name(mi->map)
+ );
+ } else {
+ printf("%6ldK %6ldK %6ldK %6ldK %6ldK %6ldK %6ldK %6ldK %s\n",
+ (long)mi->usage.vss / 1024,
+ (long)mi->usage.rss / 1024,
+ (long)mi->usage.pss / 1024,
+ (long)mi->usage.uss / 1024,
+ mi->shared_clean * pagesize / 1024,
+ mi->shared_dirty * pagesize / 1024,
+ mi->private_clean * pagesize / 1024,
+ mi->private_dirty * pagesize / 1024,
+ pm_map_name(mi->map)
+ );
+ }
+ }
+
+ /* print totals */
+ if (ws == WS_ONLY) {
+ printf("%7s %7s %7s %7s %7s %7s %7s %s\n",
+ "-------", "-------", "-------", "-------", "-------", "-------", "-------", "");
+ printf("%6ldK %6ldK %6ldK %6ldK %6ldK %6ldK %6ldK %s\n",
+ (long)total_usage.rss / 1024,
+ (long)total_usage.pss / 1024,
+ (long)total_usage.uss / 1024,
+ total_shared_clean * pagesize / 1024,
+ total_shared_dirty * pagesize / 1024,
+ total_private_clean * pagesize / 1024,
+ total_private_dirty * pagesize / 1024,
+ "TOTAL"
+ );
+ } else {
+ printf("%7s %7s %7s %7s %7s %7s %7s %7s %s\n",
+ "-------", "-------", "-------", "-------", "-------", "-------", "-------", "-------", "");
+ printf("%6ldK %6ldK %6ldK %6ldK %6ldK %6ldK %6ldK %6ldK %s\n",
+ (long)total_usage.vss / 1024,
+ (long)total_usage.rss / 1024,
+ (long)total_usage.pss / 1024,
+ (long)total_usage.uss / 1024,
+ total_shared_clean * pagesize / 1024,
+ total_shared_dirty * pagesize / 1024,
+ total_private_clean * pagesize / 1024,
+ total_private_dirty * pagesize / 1024,
+ "TOTAL"
+ );
+ }
+
+ return 0;
+}
+
+static void usage(const char *cmd) {
+ fprintf(stderr, "Usage: %s [ -w | -W ] [ -p | -m ] [ -h ] pid\n"
+ " -w Displays statistics for the working set only.\n"
+ " -W Resets the working set of the process.\n"
+ " -p Sort by PSS.\n"
+ " -m Sort by mapping order (as read from /proc).\n"
+ " -h Hide maps with no RSS.\n",
+ cmd);
+}
+
+int comp_pss(const void *a, const void *b) {
+ struct map_info *ma, *mb;
+
+ ma = *((struct map_info **)a);
+ mb = *((struct map_info **)b);
+
+ if (mb->usage.pss < ma->usage.pss) return -1;
+ if (mb->usage.pss > ma->usage.pss) return 1;
+ return 0;
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-05-23 21:29:17 +0000