+/*
+ * Copyright (C) 2010 Timo Kokkonen <timo.t.kokkonen@iki.fi>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
}
}
+static int add_pid_to_pidlist(int pidi, struct list_head *pidlist)
+{
+ struct pidlist *pid = alloc_pidlist();
+
+ if (pid == NULL) {
+ perror("malloc");
+ return -1;
+ }
+ pid->pid = pidi;
+
+ list_add_tail(&pid->list, pidlist);
+
+ return 0;
+}
+
void read_args(int argc, char *argv[], struct parse_opts *opts)
{
- int optind = 0, c;
+ int option_index = 0, c;
static struct option long_options[] = {
{ .val = 'p', .name = "pid", .has_arg = 1, },
{ .val = 'P', .name = "process", .has_arg = 1, },
while (1) {
c = getopt_long(argc, argv, short_options, long_options,
- &optind);
+ &option_index);
if (c == -1)
break;
switch (c) {
case 'p':
{
- struct pidlist *pid = alloc_pidlist();
-
- if (pid == NULL) {
- perror("malloc");
- return;
+ int pid = pidstr_is_ok(optarg);
+ if (!pid) {
+ fprintf(stderr, "Invalid pid number %s\n",
+ optarg);
+ break;
}
- pid->pid = atoi(optarg);
+
opts->parse_mask |= PARSE_PID;
- list_add_tail(&pid->list, &opts->pidlist);
+ add_pid_to_pidlist(pid, &opts->pidlist);
break;
}
case 'P':
break;
}
}
+
+ while (optind < argc) {
+ int pid = pidstr_is_ok(argv[optind]);
+
+ if (pid) {
+ opts->parse_mask |= PARSE_PID;
+ add_pid_to_pidlist(pid, &opts->pidlist);
+ } else {
+ get_all_pids_by_name(opts, argv[optind]);
+ opts->parse_mask |= PARSE_PID;
+ }
+ optind++;
+ }
}
int main(int argc, char *argv[])
{
- struct pageframe pf;
+ struct rb_root root;
struct process process_list;
struct parse_opts opts;
read_args(argc, argv, &opts);
- if (argc < 3) {
- printf("A pid needs to be given as an argument\n");
- print_help_and_die(argv[0]);
+ if (argc < 2) {
+ opts.parse_mask = PARSE_MAP_NAME;
+ opts.name = "";
}
- clear_pageframe(&pf);
-
+ memset(&root, 0, sizeof(root));
memset(&process_list, 0, sizeof(process_list));
INIT_LIST_HEAD(&process_list.list);
- if (scan_all_pids(&pf, &process_list, &opts))
+ printf("Scanning all process IDs\n");
+
+ if (scan_all_pids(&root, &process_list, &opts))
return 1;
- update_kpageflags(&pf);
+ printf("Updating kpageflags\n");
+ update_kpageflags(&root);
+ printf("Preparing to print out results\n");
if (opts.parse_mask & PARSE_DUMP)
- dump_process_maps(&pf, &process_list, &opts);
+ dump_process_maps(&root, &process_list, &opts);
else
- print_pid_stats(&pf, &process_list, &opts);
+ print_pid_stats(&root, &process_list, &opts);
- print_page_stats(&pf);
+ print_page_stats(&root);
return 0;
}