X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;ds=sidebyside;f=src%2Fcommon.c;h=3a2fa0c184cbf8712dd1e00153db5ec770047b93;hb=72201e5a82ab8d4e7c67fe05242388454d3438f6;hp=a09ad2c0c7f4c4e37b155fb2a126c7293d8d2562;hpb=19cc9a75b815d2cce16b2d6c0c7ce5990512b18d;p=lttngtop.git

diff --git a/src/common.c b/src/common.c
index a09ad2c..3a2fa0c 100644
--- a/src/common.c
+++ b/src/common.c
@@ -145,6 +145,21 @@ char *get_context_comm(const struct bt_ctf_event *event)
 	return comm;
 }
 
+char *get_context_hostname(const struct bt_ctf_event *event)
+{
+	const struct definition *scope;
+	char *hostname;
+
+	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 hostname;
+}
+
 /*
  * To get the parent process, put the pid in the tid field
  * because the parent process gets pid = tid
@@ -167,6 +182,7 @@ struct processtop* add_proc(struct lttngtop *ctx, int tid, char *comm,
 	/* 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;
@@ -194,7 +210,7 @@ struct processtop* add_proc(struct lttngtop *ctx, int tid, char *comm,
 }
 
 struct processtop* update_proc(struct processtop* proc, int pid, int tid,
-		int ppid, int vpid, int vtid, int vppid, char *comm)
+		int ppid, int vpid, int vtid, int vppid, char *comm, char *hostname)
 {
 	if (proc) {
 		proc->pid = pid;
@@ -207,6 +223,12 @@ struct processtop* update_proc(struct processtop* proc, int pid, int tid,
 			free(proc->comm);
 			proc->comm = strdup(comm);
 		}
+		if (hostname) {
+			if (proc->hostname && strcmp(proc->hostname, hostname) != 0) {
+				free(proc->hostname);
+			}
+			proc->hostname = strdup(hostname);
+		}
 	}
 	return proc;
 }
@@ -255,6 +277,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)
@@ -581,11 +606,13 @@ enum bt_cb_ret handle_statedump_process_state(struct bt_ctf_event *call_data,
 	proc = find_process_tid(&lttngtop, tid, procname);
 	if (proc == NULL)
 		proc = add_proc(&lttngtop, tid, procname, timestamp);
-	update_proc(proc, pid, tid, ppid, vpid, vtid, vppid, procname);
+	update_proc(proc, pid, tid, ppid, vpid, vtid, vppid, procname, NULL);
 
-	free(proc->comm);
-	proc->comm = strdup(procname);
-	proc->pid = pid;
+	if (proc) {
+		free(proc->comm);
+		proc->comm = strdup(procname);
+		proc->pid = pid;
+	}
 
 	return BT_CB_OK;
 
@@ -609,3 +636,16 @@ struct tm format_timestamp(uint64_t timestamp)
 
 	return tm;
 }
+
+int *lookup_tid_list(int tid)
+{
+	return g_hash_table_lookup(tid_list, (gpointer) &tid);
+}
+
+char *lookup_hostname_list(const char *hostname)
+{
+	if (!hostname)
+		return NULL;
+
+	return g_hash_table_lookup(hostname_list, (gpointer) hostname);
+}