prepare the 0.3 release

[lttngtop.git] / src / cputop.c

diff --git a/src/cputop.c b/src/cputop.c
index a0ff279e3eb95fc81c6bffa909fb14264c9a4f01..4596f28b674ddd3b0b0c4a2e43ccc2c45bdeb11b 100644 (file)
--- a/src/cputop.c
+++ b/src/cputop.c
@@ -20,9 +20,10 @@
 #include "lttngtoptypes.h"
 #include "common.h"
 #include "cputop.h"
+#include "lttngtop.h"
 
 void update_cputop_data(unsigned long timestamp, int64_t cpu, int prev_pid,
-               int next_pid, char *prev_comm, char *next_comm)
+               int next_pid, char *prev_comm, char *next_comm, char *hostname)
 {
        struct cputime *tmpcpu;
        unsigned long elapsed;
@@ -33,12 +34,14 @@ void update_cputop_data(unsigned long timestamp, int64_t cpu, int prev_pid,
                elapsed = timestamp - tmpcpu->task_start;
                tmpcpu->current_task->totalcpunsec += elapsed;
                tmpcpu->current_task->threadstotalcpunsec += elapsed;
-               if (tmpcpu->current_task->pid != tmpcpu->current_task->tid)
+               if (tmpcpu->current_task->threadparent &&
+                               tmpcpu->current_task->pid != tmpcpu->current_task->tid)
                        tmpcpu->current_task->threadparent->threadstotalcpunsec += elapsed;
        }
 
        if (next_pid != 0)
-               tmpcpu->current_task = get_proc(&lttngtop, next_pid, next_comm, timestamp);
+               tmpcpu->current_task = get_proc(&lttngtop, next_pid, next_comm,
+                               timestamp, hostname);
        else
                tmpcpu->current_task = NULL;
 
@@ -53,6 +56,7 @@ enum bt_cb_ret handle_sched_switch(struct bt_ctf_event *call_data,
        uint64_t cpu_id;
        char *prev_comm, *next_comm;
        int prev_tid, next_tid;
+       char *hostname = NULL;
 
        timestamp = bt_ctf_get_timestamp(call_data);
        if (timestamp == -1ULL)
@@ -91,7 +95,7 @@ enum bt_cb_ret handle_sched_switch(struct bt_ctf_event *call_data,
        cpu_id = get_cpu_id(call_data);
 
        update_cputop_data(timestamp, cpu_id, prev_tid, next_tid,
-                       prev_comm, next_comm);
+                       prev_comm, next_comm, hostname);
 
        return BT_CB_OK;
 
@@ -126,6 +130,9 @@ enum bt_cb_ret handle_sched_process_free(struct bt_ctf_event *call_data,
                fprintf(stderr, "Missing tid field\n");
                goto error;
        }
+       if (tid == opt_exec_pid) {
+               quit = 1;
+       }
 
        death_proc(&lttngtop, tid, comm, timestamp);
 
@@ -136,3 +143,55 @@ error:
 
 }
 
+enum bt_cb_ret handle_sched_process_fork(struct bt_ctf_event *call_data,
+               void *private_data)
+{
+       const struct bt_definition *scope;
+       struct processtop *tmp;
+       int tid, *hash_tid, parent_pid;
+       unsigned long timestamp;
+       char *comm;
+
+       timestamp = bt_ctf_get_timestamp(call_data);
+       if (timestamp == -1ULL)
+               goto error;
+
+       scope = bt_ctf_get_top_level_scope(call_data,
+                       BT_EVENT_FIELDS);
+       comm = bt_ctf_get_char_array(bt_ctf_get_field(call_data,
+                               scope, "_child_comm"));
+       if (bt_ctf_field_get_error()) {
+               fprintf(stderr, "Missing procname context info\n");
+               goto error;
+       }
+
+       tid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
+                               scope, "_child_tid"));
+       if (bt_ctf_field_get_error()) {
+               fprintf(stderr, "Missing child_tid field\n");
+               goto error;
+       }
+
+       parent_pid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
+                               scope, "_parent_pid"));
+       if (bt_ctf_field_get_error()) {
+               fprintf(stderr, "Missing parent_pid field\n");
+               goto error;
+       }
+
+       tmp = get_proc(&lttngtop, tid, comm, timestamp, NULL);
+
+       if (opt_child) {
+               hash_tid = lookup_filter_tid_list(parent_pid);
+               if (hash_tid) {
+                       add_filter_tid_list(tmp);
+               }
+       }
+
+       return BT_CB_OK;
+
+error:
+       return BT_CB_ERROR_STOP;
+
+}
+