X-Git-Url: https://git.lttng.org/?p=lttngtop.git;a=blobdiff_plain;f=src%2Fcommon.c;h=500c80d674f9056ced114e75a54af1e04fc51fc2;hp=f6c20c9c8dee3487dd480bfa37d6bb0cb570d2b4;hb=da4353bbbb28c19bd28447018edb94845ae71673;hpb=c8d75a13bc5c025053b0c2c14fa1a7dfdaa1b448

diff --git a/src/common.c b/src/common.c
index f6c20c9..500c80d 100644
--- a/src/common.c
+++ b/src/common.c
@@ -200,7 +200,9 @@ struct processtop* add_proc(struct lttngtop *ctx, int tid, char *comm,
 		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(tid, newproc);
+		}
 		ctx->nbnewthreads++;
 		ctx->nbthreads++;
 	}
@@ -228,6 +230,9 @@ struct processtop* update_proc(struct processtop* proc, int pid, int tid,
 				free(proc->hostname);
 			}
 			proc->hostname = strdup(hostname);
+			if (lookup_hostname_list(hostname)) {
+				add_filter_tid_list(tid, proc);
+			}
 		}
 	}
 	return proc;
@@ -430,6 +435,7 @@ 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;
@@ -438,6 +444,7 @@ struct lttngtop* get_copy_lttngtop(unsigned long start, unsigned long end)
 	dst->process_table = g_ptr_array_new();
 	dst->files_table = g_ptr_array_new();
 	dst->cpu_table = g_ptr_array_new();
+	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);
@@ -520,6 +527,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++) {
@@ -639,13 +655,37 @@ struct tm format_timestamp(uint64_t timestamp)
 
 int *lookup_tid_list(int tid)
 {
+	if (!tid_list)
+		return NULL;
+
 	return g_hash_table_lookup(tid_list, (gpointer) &tid);
 }
 
 char *lookup_hostname_list(const char *hostname)
 {
-	if (!hostname)
+	if (!hostname || !hostname_list)
 		return NULL;
 
 	return g_hash_table_lookup(hostname_list, (gpointer) hostname);
 }
+
+int *lookup_filter_tid_list(int tid)
+{
+	return g_hash_table_lookup(global_filter_list, (gpointer) &tid);
+}
+
+void add_filter_tid_list(int tid, struct processtop *newproc)
+{
+	unsigned long *hash_tid;
+
+	hash_tid = malloc(sizeof(unsigned long));
+	*hash_tid = tid;
+	g_hash_table_insert(global_filter_list,
+			(gpointer) (unsigned long) hash_tid, newproc);
+}
+
+void remove_filter_tid_list(int tid)
+{
+	g_hash_table_remove(global_filter_list,
+			(gpointer) (unsigned long) &tid);
+}