X-Git-Url: https://git.lttng.org/?p=lttngtop.git;a=blobdiff_plain;f=src%2Fcommon.c;h=a6c2e4c3cfa17a24d833dce601947e14761969a1;hp=df2ef7a4d376a395587815939dc224fa1926cab9;hb=54645d5cab414bb3391837ef7acb919cf106c71f;hpb=3ba84bed69e8eb8e6794a7250da97406dbbe02c2

diff --git a/src/common.c b/src/common.c
index df2ef7a..a6c2e4c 100644
--- a/src/common.c
+++ b/src/common.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2011 Julien Desfossez
+ * Copyright (C) 2011-2012 Julien Desfossez
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License Version 2 as
@@ -17,12 +17,13 @@
 
 #include <babeltrace/ctf/events.h>
 #include <stdlib.h>
+#include <linux/unistd.h>
 #include <string.h>
 #include "common.h"
 
 uint64_t get_cpu_id(const struct bt_ctf_event *event)
 {
-	const struct definition *scope;
+	const struct bt_definition *scope;
 	uint64_t cpu_id;
 
 	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_PACKET_CONTEXT);
@@ -37,15 +38,19 @@ uint64_t get_cpu_id(const struct bt_ctf_event *event)
 
 uint64_t get_context_tid(const struct bt_ctf_event *event)
 {
-	const struct definition *scope;
+	const struct bt_definition *scope;
 	uint64_t tid;
 
 	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
 	tid = bt_ctf_get_int64(bt_ctf_get_field(event,
 				scope, "_tid"));
 	if (bt_ctf_field_get_error()) {
-		fprintf(stderr, "Missing tid context info\n");
-		return -1ULL;
+		tid = bt_ctf_get_int64(bt_ctf_get_field(event,
+					scope, "_vtid"));
+		if (bt_ctf_field_get_error()) {
+			fprintf(stderr, "Missing tid context info\n");
+			return -1ULL;
+		}
 	}
 
 	return tid;
@@ -53,15 +58,20 @@ uint64_t get_context_tid(const struct bt_ctf_event *event)
 
 uint64_t get_context_pid(const struct bt_ctf_event *event)
 {
-	const struct definition *scope;
+	const struct bt_definition *scope;
 	uint64_t pid;
 
 	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
 	pid = bt_ctf_get_int64(bt_ctf_get_field(event,
 				scope, "_pid"));
 	if (bt_ctf_field_get_error()) {
-		fprintf(stderr, "Missing pid context info\n");
-		return -1ULL;
+		/* Try UST pid */
+		pid = bt_ctf_get_int64(bt_ctf_get_field(event,
+					scope, "_vpid"));
+		if (bt_ctf_field_get_error()) {
+			fprintf(stderr, "Missing pid context info\n");
+			return -1ULL;
+		}
 	}
 
 	return pid;
@@ -69,23 +79,67 @@ uint64_t get_context_pid(const struct bt_ctf_event *event)
 
 uint64_t get_context_ppid(const struct bt_ctf_event *event)
 {
-	const struct definition *scope;
+	const struct bt_definition *scope;
 	uint64_t ppid;
 
 	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
 	ppid = bt_ctf_get_int64(bt_ctf_get_field(event,
 				scope, "_ppid"));
 	if (bt_ctf_field_get_error()) {
-		fprintf(stderr, "Missing ppid context info\n");
 		return -1ULL;
 	}
 
 	return ppid;
 }
 
+uint64_t get_context_vtid(const struct bt_ctf_event *event)
+{
+	const struct bt_definition *scope;
+	uint64_t vtid;
+
+	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
+	vtid = bt_ctf_get_int64(bt_ctf_get_field(event,
+				scope, "_vtid"));
+	if (bt_ctf_field_get_error()) {
+		return -1ULL;
+	}
+
+	return vtid;
+}
+
+uint64_t get_context_vpid(const struct bt_ctf_event *event)
+{
+	const struct bt_definition *scope;
+	uint64_t vpid;
+
+	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
+	vpid = bt_ctf_get_int64(bt_ctf_get_field(event,
+				scope, "_vpid"));
+	if (bt_ctf_field_get_error()) {
+		return -1ULL;
+	}
+
+	return vpid;
+}
+
+uint64_t get_context_vppid(const struct bt_ctf_event *event)
+{
+	const struct bt_definition *scope;
+	uint64_t vppid;
+
+	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
+	vppid = bt_ctf_get_int64(bt_ctf_get_field(event,
+				scope, "_vppid"));
+	if (bt_ctf_field_get_error()) {
+		return -1ULL;
+	}
+
+	return vppid;
+}
+
 char *get_context_comm(const struct bt_ctf_event *event)
 {
-	const struct definition *scope;
+	const struct bt_definition *scope;
 	char *comm;
 
 	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
@@ -99,59 +153,109 @@ char *get_context_comm(const struct bt_ctf_event *event)
 	return comm;
 }
 
-struct processtop *find_process_tid(struct lttngtop *ctx, int tid, char *comm)
+char *get_context_hostname(const struct bt_ctf_event *event)
 {
-	gint i;
-	struct processtop *tmp;
+	const struct bt_definition *scope;
+	char *hostname;
 
-	for (i = 0; i < ctx->process_table->len; i++) {
-		tmp = g_ptr_array_index(ctx->process_table, i);
-		if (tmp && tmp->tid == tid)
-			return tmp;
+	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
+	hostname = bt_ctf_get_char_array(bt_ctf_get_field(event,
+				scope, "_hostname"));
+	if (bt_ctf_field_get_error()) {
+		return NULL;
 	}
-	return NULL;
+
+	return hostname;
+}
+
+/*
+ * To get the parent process, put the pid in the tid field
+ * because the parent process gets pid = tid
+ */
+struct processtop *find_process_tid(struct lttngtop *ctx, int tid, const char *comm)
+{
+	struct processtop *tmp;
+
+	tmp = g_hash_table_lookup(ctx->process_hash_table,
+			(gconstpointer) (unsigned long) tid);
+
+	return tmp;
 }
 
 struct processtop* add_proc(struct lttngtop *ctx, int tid, char *comm,
-		unsigned long timestamp)
+		unsigned long timestamp, char *hostname)
 {
 	struct processtop *newproc;
+	struct host *host;
 
 	/* if the PID already exists, we just rename the process */
 	/* FIXME : need to integrate with clone/fork/exit to be accurate */
 	newproc = find_process_tid(ctx, tid, comm);
+
 	if (!newproc) {
 		newproc = g_new0(struct processtop, 1);
 		newproc->tid = tid;
 		newproc->birth = timestamp;
 		newproc->process_files_table = g_ptr_array_new();
-		newproc->files_history = g_new0(struct file_history, 1);
-		newproc->files_history->next = NULL;
+		newproc->files_history = NULL;
 		newproc->totalfileread = 0;
 		newproc->totalfilewrite = 0;
 		newproc->fileread = 0;
 		newproc->filewrite = 0;
 		newproc->syscall_info = NULL;
+		newproc->threadparent = NULL;
 		newproc->threads = g_ptr_array_new();
 		newproc->perf = g_hash_table_new(g_str_hash, g_str_equal);
 		g_ptr_array_add(ctx->process_table, newproc);
+		g_hash_table_insert(ctx->process_hash_table,
+				(gpointer) (unsigned long) tid, newproc);
+		if (lookup_tid_list(tid)) {
+			add_filter_tid_list(newproc);
+		}
+		ctx->nbnewthreads++;
+		ctx->nbthreads++;
 	}
 	newproc->comm = strdup(comm);
+	if (hostname) {
+		host = lookup_hostname_list(hostname);
+		if (!host)
+			host = add_hostname_list(hostname, 0);
+		if (!newproc->host || (newproc->host != host))
+			newproc->host = host;
+		if (is_hostname_filtered(hostname)) {
+			add_filter_tid_list(newproc);
+		}
+	}
 
 	return newproc;
 }
 
 struct processtop* update_proc(struct processtop* proc, int pid, int tid,
-		int ppid, char *comm)
+		int ppid, int vpid, int vtid, int vppid, char *comm, char *hostname)
 {
+	struct host *host;
+
 	if (proc) {
 		proc->pid = pid;
 		proc->tid = tid;
 		proc->ppid = ppid;
+		proc->vpid = vpid;
+		proc->vtid = vtid;
+		proc->vppid = vppid;
 		if (strcmp(proc->comm, comm) != 0) {
 			free(proc->comm);
 			proc->comm = strdup(comm);
 		}
+		if (hostname && !proc->host) {
+			host = lookup_hostname_list(hostname);
+			if (!host)
+				host = add_hostname_list(hostname, 0);
+			if (!proc->host || (proc->host != host))
+				proc->host = host;
+			if (is_hostname_filtered(hostname)) {
+				add_filter_tid_list(proc);
+			}
+		}
 	}
 	return proc;
 }
@@ -165,18 +269,36 @@ void death_proc(struct lttngtop *ctx, int tid, char *comm,
 {
 	struct processtop *tmp;
 	tmp = find_process_tid(ctx, tid, comm);
-	if (tmp && strcmp(tmp->comm, comm) == 0)
+
+	g_hash_table_remove(ctx->process_hash_table,
+			(gpointer) (unsigned long) tid);
+	if (tmp && strcmp(tmp->comm, comm) == 0) {
 		tmp->death = timestamp;
+		ctx->nbdeadthreads++;
+		ctx->nbthreads--;
+	}
 }
 
 struct processtop* get_proc(struct lttngtop *ctx, int tid, char *comm,
-		unsigned long timestamp)
+		unsigned long timestamp, char *hostname)
 {
 	struct processtop *tmp;
+
 	tmp = find_process_tid(ctx, tid, comm);
-	if (tmp && strcmp(tmp->comm, comm) == 0)
+	if (tmp && strcmp(tmp->comm, comm) == 0) {
+		return tmp;
+	}
+	return add_proc(ctx, tid, comm, timestamp, hostname);
+}
+
+struct processtop *get_proc_pid(struct lttngtop *ctx, int tid, int pid,
+		unsigned long timestamp, char *hostname)
+{
+	struct processtop *tmp;
+	tmp = find_process_tid(ctx, tid, NULL);
+	if (tmp && tmp->pid == pid)
 		return tmp;
-	return add_proc(ctx, tid, comm, timestamp);
+	return add_proc(ctx, tid, NULL, timestamp, hostname);
 }
 
 void add_thread(struct processtop *parent, struct processtop *thread)
@@ -184,6 +306,9 @@ void add_thread(struct processtop *parent, struct processtop *thread)
 	gint i;
 	struct processtop *tmp;
 
+	if (!parent)
+		return;
+
 	for (i = 0; i < parent->threads->len; i++) {
 		tmp = g_ptr_array_index(parent->threads, i);
 		if (tmp == thread)
@@ -252,7 +377,7 @@ void reset_perf_counter(gpointer key, gpointer value, gpointer user_data)
 void copy_perf_counter(gpointer key, gpointer value, gpointer new_table)
 {
 	struct perfcounter *newperf;
-	
+
 	newperf = g_new0(struct perfcounter, 1);
 	newperf->count = ((struct perfcounter *) value)->count;
 	newperf->visible = ((struct perfcounter *) value)->visible;
@@ -260,9 +385,15 @@ void copy_perf_counter(gpointer key, gpointer value, gpointer new_table)
 	g_hash_table_insert((GHashTable *) new_table, strdup(key), newperf);
 }
 
+void copy_process_table(gpointer key, gpointer value, gpointer new_table)
+{
+	g_hash_table_insert((GHashTable *) new_table, key, value);
+}
+
 void rotate_perfcounter() {
 	int i;
 	struct processtop *tmp;
+
 	for (i = 0; i < lttngtop.process_table->len; i++) {
 		tmp = g_ptr_array_index(lttngtop.process_table, i);
 		g_hash_table_foreach(tmp->perf, reset_perf_counter, NULL);
@@ -287,11 +418,40 @@ void cleanup_processtop()
 			if (tmpf != NULL) {
 				tmpf->read = 0;
 				tmpf->write = 0;
+
+				if (tmpf->flag == __NR_close)
+					g_ptr_array_index(
+						tmp->process_files_table, j
+					) = NULL;
 			}
 		}
 	}
 }
 
+void reset_global_counters()
+{
+	lttngtop.nbnewproc = 0;
+	lttngtop.nbdeadproc = 0;
+	lttngtop.nbnewthreads = 0;
+	lttngtop.nbdeadthreads = 0;
+	lttngtop.nbnewfiles = 0;
+	lttngtop.nbclosedfiles = 0;
+}
+
+void copy_global_counters(struct lttngtop *dst)
+{
+	dst->nbproc = lttngtop.nbproc;
+	dst->nbnewproc = lttngtop.nbnewproc;
+	dst->nbdeadproc = lttngtop.nbdeadproc;
+	dst->nbthreads = lttngtop.nbthreads;
+	dst->nbnewthreads = lttngtop.nbnewthreads;
+	dst->nbdeadthreads = lttngtop.nbdeadthreads;
+	dst->nbfiles = lttngtop.nbfiles;
+	dst->nbnewfiles = lttngtop.nbnewfiles;
+	dst->nbclosedfiles = lttngtop.nbclosedfiles;
+	reset_global_counters();
+}
+
 struct lttngtop* get_copy_lttngtop(unsigned long start, unsigned long end)
 {
 	gint i, j;
@@ -300,18 +460,22 @@ struct lttngtop* get_copy_lttngtop(unsigned long start, unsigned long end)
 	struct processtop *tmp, *tmp2, *new;
 	struct cputime *tmpcpu, *newcpu;
 	struct files *tmpfile, *newfile;
+	struct kprobes *tmpprobe, *newprobe;
 
 	dst = g_new0(struct lttngtop, 1);
 	dst->start = start;
 	dst->end = end;
+	copy_global_counters(dst);
 	dst->process_table = g_ptr_array_new();
 	dst->files_table = g_ptr_array_new();
 	dst->cpu_table = g_ptr_array_new();
-	dst->perf_list = g_hash_table_new(g_str_hash, g_str_equal);
+	dst->kprobes_table = g_ptr_array_new();
+	dst->process_hash_table = g_hash_table_new(g_direct_hash, g_direct_equal);
+	g_hash_table_foreach(lttngtop.process_hash_table, copy_process_table,
+			dst->process_hash_table);
 
 	rotate_cputime(end);
 
-	g_hash_table_foreach(lttngtop.perf_list, copy_perf_counter, dst->perf_list);
 	for (i = 0; i < lttngtop.process_table->len; i++) {
 		tmp = g_ptr_array_index(lttngtop.process_table, i);
 		new = g_new0(struct processtop, 1);
@@ -320,6 +484,7 @@ struct lttngtop* get_copy_lttngtop(unsigned long start, unsigned long end)
 		new->threads = g_ptr_array_new();
 		new->comm = strdup(tmp->comm);
 		new->process_files_table = g_ptr_array_new();
+		new->files_history = tmp->files_history;
 		new->perf = g_hash_table_new(g_str_hash, g_str_equal);
 		g_hash_table_foreach(tmp->perf, copy_perf_counter, new->perf);
 
@@ -337,7 +502,10 @@ struct lttngtop* get_copy_lttngtop(unsigned long start, unsigned long end)
 
 			if (tmpfile != NULL) {
 				memcpy(newfile, tmpfile, sizeof(struct files));
-				newfile->name = strdup(tmpfile->name);
+				if (tmpfile->name)
+					newfile->name = strdup(tmpfile->name);
+				else
+					newfile->name = NULL;
 				newfile->ref = new;
 				g_ptr_array_add(new->process_files_table,
 						newfile);
@@ -351,6 +519,7 @@ struct lttngtop* get_copy_lttngtop(unsigned long start, unsigned long end)
 			 * files associated with if after the copy
 			 */
 			if (tmp->death > 0 && tmp->death < end) {
+				/* FIXME : close the files before */
 				g_ptr_array_remove(tmp->process_files_table, tmpfile);
 				g_free(tmpfile);
 			}
@@ -386,6 +555,15 @@ struct lttngtop* get_copy_lttngtop(unsigned long start, unsigned long end)
 		 */
 		g_ptr_array_add(dst->cpu_table, newcpu);
 	}
+	if (lttngtop.kprobes_table) {
+		for (i = 0; i < lttngtop.kprobes_table->len; i++) {
+			tmpprobe = g_ptr_array_index(lttngtop.kprobes_table, i);
+			newprobe = g_new0(struct kprobes, 1);
+			memcpy(newprobe, tmpprobe, sizeof(struct kprobes));
+			tmpprobe->count = 0;
+			g_ptr_array_add(dst->kprobes_table, newprobe);
+		}
+	}
 	/* FIXME : better algo */
 	/* create the threads index if required */
 	for (i = 0; i < dst->process_table->len; i++) {
@@ -407,3 +585,202 @@ struct lttngtop* get_copy_lttngtop(unsigned long start, unsigned long end)
 	return dst;
 }
 
+
+enum bt_cb_ret handle_statedump_process_state(struct bt_ctf_event *call_data,
+		void *private_data)
+{
+	const struct bt_definition *scope;
+	struct processtop *proc;
+	unsigned long timestamp;
+	int64_t pid, tid, ppid, vtid, vpid, vppid;
+	char *procname = NULL, *hostname = NULL;
+
+	timestamp = bt_ctf_get_timestamp(call_data);
+	if (timestamp == -1ULL)
+		goto error;
+
+	scope = bt_ctf_get_top_level_scope(call_data,
+			 BT_EVENT_FIELDS);
+	pid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
+			     scope, "_pid"));
+	if (bt_ctf_field_get_error()) {
+		fprintf(stderr, "Missing pid context info\n");
+		goto error;
+	}
+	ppid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
+				scope, "_ppid"));
+	if (bt_ctf_field_get_error()) {
+		goto end;
+	}
+	tid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
+				scope, "_tid"));
+	if (bt_ctf_field_get_error()) {
+		fprintf(stderr, "Missing tid context info\n");
+		goto error;
+	}
+	vtid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
+				scope, "_vtid"));
+	if (bt_ctf_field_get_error()) {
+		fprintf(stderr, "Missing vtid context info\n");
+		goto error;
+	}
+	vpid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
+				scope, "_vpid"));
+	if (bt_ctf_field_get_error()) {
+		fprintf(stderr, "Missing vpid context info\n");
+		goto error;
+	}
+	vppid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
+				scope, "_vppid"));
+	if (bt_ctf_field_get_error()) {
+		fprintf(stderr, "Missing vppid context info\n");
+		goto error;
+	}
+
+	scope = bt_ctf_get_top_level_scope(call_data,
+			BT_EVENT_FIELDS);
+	procname = bt_ctf_get_char_array(bt_ctf_get_field(call_data,
+				scope, "_name"));
+	if (bt_ctf_field_get_error()) {
+		fprintf(stderr, "Missing process name context info\n");
+		goto error;
+	}
+
+	proc = find_process_tid(&lttngtop, tid, procname);
+	if (proc == NULL)
+		proc = add_proc(&lttngtop, tid, procname, timestamp, hostname);
+	update_proc(proc, pid, tid, ppid, vpid, vtid, vppid, procname, hostname);
+
+	if (proc) {
+		free(proc->comm);
+		proc->comm = strdup(procname);
+		proc->pid = pid;
+	}
+
+end:
+	return BT_CB_OK;
+
+error:
+	return BT_CB_ERROR_STOP;
+}
+
+struct tm format_timestamp(uint64_t timestamp)
+{
+	struct tm tm;
+	uint64_t ts_sec = 0, ts_nsec;
+	time_t time_s;
+
+	ts_nsec = timestamp;
+	ts_sec += ts_nsec / NSEC_PER_SEC;
+	ts_nsec = ts_nsec % NSEC_PER_SEC;
+
+	time_s = (time_t) ts_sec;
+
+	localtime_r(&time_s, &tm);
+
+	return tm;
+}
+
+int *lookup_tid_list(int tid)
+{
+	if (!tid_filter_list)
+		return NULL;
+
+	return g_hash_table_lookup(tid_filter_list, (gpointer) &tid);
+}
+
+struct host *lookup_hostname_list(const char *hostname)
+{
+	if (!hostname || !global_host_list)
+		return NULL;
+
+	return g_hash_table_lookup(global_host_list, (gpointer) hostname);
+}
+
+int is_hostname_filtered(const char *hostname)
+{
+	struct host *host;
+
+	host = lookup_hostname_list(hostname);
+	if (host)
+		return host->filter;
+	return 0;
+}
+
+int *lookup_filter_tid_list(int tid)
+{
+	return g_hash_table_lookup(global_filter_list, (gpointer) &tid);
+}
+
+void add_filter_tid_list(struct processtop *proc)
+{
+	unsigned long *hash_tid;
+
+	hash_tid = malloc(sizeof(unsigned long));
+	*hash_tid = proc->tid;
+	g_hash_table_insert(global_filter_list,
+			(gpointer) (unsigned long) hash_tid, proc);
+}
+
+void remove_filter_tid_list(int tid)
+{
+	g_hash_table_remove(global_filter_list,
+			(gpointer) (unsigned long) &tid);
+}
+
+struct host *add_hostname_list(char *hostname, int filter)
+{
+	struct host *host;
+
+	host = lookup_hostname_list(hostname);
+	if (host)
+		return host;
+
+	host = g_new0(struct host, 1);
+	host->hostname = strdup(hostname);
+	host->filter = filter;
+	g_hash_table_insert(global_host_list,
+			(gpointer) host->hostname,
+			(gpointer) host);
+
+	return host;
+}
+
+void update_hostname_filter(struct host *host)
+{
+	struct processtop *tmp;
+	int i;
+
+	for (i = 0; i < lttngtop.process_table->len; i++) {
+		tmp = g_ptr_array_index(lttngtop.process_table, i);
+		if (tmp->host == host) {
+			if (host->filter)
+				add_filter_tid_list(tmp);
+			else
+				remove_filter_tid_list(tmp->tid);
+		}
+	}
+}
+
+char *lookup_procname(const char *procname)
+{
+	if (!procname || !global_procname_list)
+		return NULL;
+
+	return g_hash_table_lookup(global_procname_list, (gpointer) procname);
+}
+
+char *add_procname_list(char *procname, int filter)
+{
+	char *proc;
+
+	proc = lookup_procname(procname);
+	if (proc)
+		return proc;
+
+	proc = strdup(procname);
+	g_hash_table_insert(global_procname_list,
+			(gpointer) procname, (gpointer) procname);
+
+	return proc;
+}