From 29d105d5dda7427ce514f0e461300265402878a7 Mon Sep 17 00:00:00 2001
From: Timo Kokkonen <kaapeli@itanic.dy.fi>
Date: Fri, 23 Jul 2010 13:41:22 +0300
Subject: [PATCH] parser: Add support for parsing processes by name

Signed-off-by: Timo Kokkonen <kaapeli@itanic.dy.fi>
---
 parse.c | 59 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)

diff --git a/parse.c b/parse.c
index 117026b..678dda3 100644
--- a/parse.c
+++ b/parse.c
@@ -4,6 +4,7 @@
 #include <stdlib.h>
 #include <string.h>
 #include <errno.h>
+#include <libgen.h>
 
 #include "parse.h"
 #include "pagemap.h"
@@ -144,6 +145,26 @@ static int read_cmdline(int pid, int tid, char *cmdline, size_t len)
 	return ret > 0 ? 0 : -1;
 }
 
+static char *get_name_by_pid(int pid)
+{
+	static int last_pid;
+	static char cmdline[128];
+	static char *bname;
+
+	if (last_pid == pid)
+		return bname;
+
+	if (read_cmdline(pid, pid, cmdline, sizeof(cmdline))) {
+		bname = NULL;
+		return NULL;
+	}
+
+	bname = basename(cmdline);
+
+	last_pid = pid;
+	return bname;
+}
+
 static int check_parse_opts(struct parse_opts *opts, struct pageframe *pf,
 		struct maps *map)
 {
@@ -157,6 +178,11 @@ static int check_parse_opts(struct parse_opts *opts, struct pageframe *pf,
 			return 1;
 	}
 
+	if (opts->parse_mask & PARSE_PROCESS_NAME) {
+		if (!strcmp(opts->name, get_name_by_pid(map->pid)))
+			return 1;
+	}
+
 	return 0;
 }
 
@@ -369,6 +395,31 @@ static int get_next_pid(DIR **dir)
 	return parse_pid(dir);
 }
 
+static int get_next_pid_by_name(DIR **dir, char *name)
+{
+	int pid;
+	char *pname;
+
+	if (opendir_check(dir, "/proc"))
+		return -1;
+
+	while (1) {
+		pid = parse_pid(dir);
+		if (pid <= 0)
+			break;
+
+		pname = get_name_by_pid(pid);
+		if (pname == NULL)
+			continue;
+		if (strcmp(pname, name))
+			continue;
+
+		return pid;
+	}
+
+	return 0;
+}
+
 static void read_pageframe_with_threads(int pid,
 					struct pageframe *pageframe,
 					struct process **process_list,
@@ -399,6 +450,14 @@ void scan_all_pids(struct pageframe *pf, struct process **process_list,
 	DIR *dir = NULL;
 	int pid;
 
+	if (opts->parse_mask & PARSE_PROCESS_NAME) {
+		while ((pid = get_next_pid_by_name(&dir, opts->name))) {
+			read_pageframe_with_threads(pid, pf, process_list,
+						opts);
+		}
+		dir = NULL;
+	}
+
 	if (opts->parse_mask & PARSE_PID)
 		read_pageframe_with_threads(opts->pid, pf, process_list, opts);
 
-- 
2.45.0