+
+#include <lttv/state.h>
+
+
+static void
+init(LttvTracesetState *self, LttvTraceset *ts)
+{
+ guint i, j, nb_trace, nb_tracefile;
+
+ LttvTraceState *tc;
+
+ LttvTracefileState *tfc;
+
+ LTTV_TRACESET_CONTEXT_CLASS(g_type_class_peek_parent(LTTV_TRACESET_STATE_TYPE))->init(self, ts);
+
+ nb_trace = lttv_traceset_number(ts);
+ for(i = 0 ; i < nb_trace ; i++) {
+ tc = (LttvTraceState *)self->traces[i];
+ tc->processes = g_hash_table_new(g_direct_hash, g_direct_equal);
+
+ nb_tracefile = ltt_trace_control_tracefile_number(tc->t);
+ for(j = 0 ; j < nb_tracefile ; j++) {
+ tfc = tc->control_tracefiles[j];
+ tfc->process = NULL;
+ }
+
+ nb_tracefile = ltt_trace_per_cpu_tracefile_number(tc->t);
+ for(j = 0 ; j < nb_tracefile ; j++) {
+ tfc = tc->per_cpu_tracefiles[j];
+ tfc->process = NULL;
+ }
+ }
+}
+
+
+static void
+fini(LttvTracesetState *self)
+{
+ guint i, j, nb_trace, nb_tracefile;
+
+ LttvTraceState *tc;
+
+ LttvTracefileState *tfc;
+
+ nb_trace = lttv_traceset_number(ts);
+ for(i = 0 ; i < nb_trace ; i++) {
+ tc = (LttvTraceState *)self->traces[i];
+ remove_all_processes(tc->processes);
+ g_hash_table_destroy(tc->processes);
+ }
+ LTTV_TRACESET_CONTEXT_CLASS(g_type_class_peek_parent(LTTV_TRACESET_STATE_TYPE))->fini(self);
+}
+
+
+LttvTracesetContext *
+new_traceset_context(LttvTracesetContext *self)
+{
+ return LTTV_TRACESET_CONTEXT(g_object_new(LTTV_TRACESET_STATE_TYPE));
+}
+
+
+LttvTraceContext *
+new_trace_context(LttvTracesetContext *self)
+{
+ return LTTV_TRACE_CONTEXT(g_object_new(LTTV_TRACE_STATE_TYPE));
+}
+
+
+LttvTracefileContext *
+new_tracefile_context(LttvTracesetContext *self)
+{
+ return LTTV_TRACEFILE_CONTEXT(g_object_new(LTTV_TRACEFILE_STATE_TYPE));
+}
+
+
+GType
+lttv_traceset_state_get_type(void)
+{
+ static GType type = 0;
+ if (type == 0) {
+ static const GTypeInfo info = {
+ sizeof (LttvTracesetStateClass),
+ NULL, /* base_init */
+ NULL, /* base_finalize */
+ traceset_state_class_init, /* class_init */
+ NULL, /* class_finalize */
+ NULL, /* class_data */
+ sizeof (LttvTracesetContext),
+ 0, /* n_preallocs */
+ traceset_state_instance_init /* instance_init */
+ };
+
+ type = g_type_register_static (LTTV_TRACESET_CONTEXT_TYPE, "LttvTracesetStateType",
+ &info, 0);
+ }
+ return type;
+}
+
+
+static void
+traceset_state_instance_init (GTypeInstance *instance, gpointer g_class)
+{
+}
+
+
+static void
+traceset_context_finalize (LttvTracesetContext *self)
+{
+ G_OBJECT_CLASS(g_type_class_peek_parent(LTTV_TRACESET_STATE_TYPE))->finalize(self);
+}
+
+
+static void
+traceset_state_class_init (LttvTracesetContextClass *klass)
+{
+ GObjectClass *gobject_class = G_OBJECT_CLASS(klass);
+
+ gobject_class->finalize = traceset_state_finalize;
+ klass->init = init;
+ klass->fini = fini;
+ klass->new_traceset_context = new_traceset_context;
+ klass->new_trace_context = new_trace_context;
+ klass->new_tracefile_context = new_tracefile_context;
+}
+
+
+GType
+lttv_trace_state_get_type(void)
+{
+ static GType type = 0;
+ if (type == 0) {
+ static const GTypeInfo info = {
+ sizeof (LttvTraceStateClass),
+ NULL, /* base_init */
+ NULL, /* base_finalize */
+ trace_state_class_init, /* class_init */
+ NULL, /* class_finalize */
+ NULL, /* class_data */
+ sizeof (LttvTracesetState),
+ 0, /* n_preallocs */
+ trace_state_instance_init /* instance_init */
+ };
+
+ type = g_type_register_static (LTTV_TRACE_CONTEXT_TYPE,
+ "LttvTraceStateType", &info, 0);
+ }
+ return type;
+}
+
+
+static void
+trace_state_instance_init (GTypeInstance *instance, gpointer g_class)
+{
+}
+
+
+static void
+trace_state_finalize (LttvTraceContext *self)
+{
+ G_OBJECT_CLASS(g_type_class_peek_parent(LTTV_TRACE_STATE_TYPE))->finalize(self);
+}
+
+
+static void
+trace_state_class_init (LttvTraceContextClass *klass)
+{
+ GObjectClass *gobject_class = G_OBJECT_CLASS(klass);
+
+ gobject_class->finalize = trace_state_finalize;
+}
+
+
+GType
+lttv_tracefile_state_get_type(void)
+{
+ static GType type = 0;
+ if (type == 0) {
+ static const GTypeInfo info = {
+ sizeof (LttvTracefileStateClass),
+ NULL, /* base_init */
+ NULL, /* base_finalize */
+ tracefile_state_class_init, /* class_init */
+ NULL, /* class_finalize */
+ NULL, /* class_data */
+ sizeof (LttvTracefileState),
+ 0, /* n_preallocs */
+ tracefile_state_instance_init /* instance_init */
+ };
+
+ type = g_type_register_static (LTTV_TRACEFILE_CONTEXT_TYPE,
+ "LttvTracefileStateType", &info, 0);
+ }
+ return type;
+}
+
+
+static void
+tracefile_state_instance_init (GTypeInstance *instance, gpointer g_class)
+{
+}
+
+
+static void
+tracefile_state_finalize (LttvTracefileState *self)
+{
+ G_OBJECT_CLASS(g_type_class_peek_parent(LTTV_TRACEFILE_STATE_TYPE))->finalize(self);
+}
+
+
+static void
+tracefile_state_class_init (LttvTracefileStateClass *klass)
+{
+ GObjectClass *gobject_class = G_OBJECT_CLASS(klass);
+
+ gobject_class->finalize = tracefile_state_finalize;
+}
+
+
+struct HookData {
+ LttField *f1;
+ LttField *f2;
+ LttField *f3;
+}
+
+
+struct HookId {
+ LttvHook h;
+ guint id;
+ void *hook_data;
+ gboolean free_hook_data;
+}
+
+
+void push_state(LttvTracefileState *tfs, LttvInterruptType t, guint state_id)
+{
+ LttvInterruptState intr;
+
+ LttvProcessState *process = tfs->process;
+
+ guint depth = process->interrupt_stack->len;
+
+ g_array_set_size(process->interrupt_stack, depth + 1);
+ intr = &g_array_index(process->interrupt_stack, LttvInterruptState, depth);
+ intr->t = t;
+ intr->n = state_id;
+ intr->entry = tfs->time;
+ intr->last_change = tfs->time;
+ intr->s = process->state->s;
+}
+
+
+void pop_state(LttvTracefileState *tfs, LttvInterruptType t)
+{
+ LttvProcessState *process = tfs->process;
+
+ guint depth = process->interrupt_stack->len - 1;
+
+ g_assert(process->state->t == t);
+ g_array_remove_index(process->interrupt_stack, depth);
+ depth--;
+ process->state = &g_array_index(process->interrupt_stack, LttvInterruptState,
+ depth);
+}
+
+
+LttvProcessState *create_process(LttvTraceState *tfs, LttvProcessState *parent,
+ guint pid)
+{
+ LttvProcessState *process = g_new(LttvProcessState, 1);
+
+ LttvInterruptState *intr;
+
+ g_hash_table_insert(tfs->ts->processes, pid, process);
+ process->pid = pid;
+ process->birth = tfs->time;
+ process->name = LTTV_STATE_UNNAMED;
+ process->interrupt_stack = g_array_new(FALSE, FALSE,
+ sizeof(LttvInterruptState));
+ g_array_set_size(process->interrupt_stack, 1);
+ intr = process->state = process->interrupt_stack->data;
+ intr->t = LTTV_STATE_USER_MODE
+ intr->n = 0;
+ intr->entry = tfs->time;
+ intr->last_change = tfs->time;
+ intr->s = LTTV_STATE_WAIT_FORK;
+}
+
+
+LttvProcessState *find_process(LttvTraceState *tfs, guint pid)
+{
+ LttvProcessState *process = g_hash_table_lookup(tfs->processes, pid);
+ if(process == NULL) process = create_process(tfs, NULL, pid);
+ return process;
+}
+
+
+void exit_process(LttvTraceState *ts, LttvProcessState *process)
+{
+ g_hash_table_remove(tfs-ts->processes, process->pid);
+ g_array_free(process->interrupt_stack, TRUE);
+ g_free(process);
+}
+
+
+void free_process_state(gpointer key, gpointer value, gpointer user_data)
+{
+ g_array_free(((LttvProcessState *)value)->interrupt_stack, TRUE);
+ g_free(value);
+}
+
+
+void remove_all_processes(GHashTable *processes)
+{
+ g_hash_table_foreach(processes, free_process_state, NULL);
+}
+
+
+gboolean syscall_entry(void *hook_data, void *call_data)
+{
+ LttField *f = (LttField *)hook_data;
+
+ struct LttvTracefileState *s = (LttvTraceFileState *)call_data;
+
+ push_state(s, STATE_SYSCALL, ltt_event_get_unsigned(s->e, f));
+ return FALSE;
+}
+
+
+gboolean syscall_exit(void *hook_data, void *call_data)
+{
+ struct LttvTracefileState *s = (LttvTraceFileState *)call_data;
+
+ pop_state(s, STATE_SYSCALL);
+ return FALSE;
+}
+
+
+gboolean trap_entry(void *hook_data, void *call_data)
+{
+ LttField *f = (LttField *)hook_data;
+
+ struct LttvTracefileState *s = (LttvTraceFileState *)call_data;
+
+ push_state(s, STATE_TRAP, ltt_event_get_unsigned(s->e, f));
+ return FALSE;
+}
+
+
+gboolean trap_exit(void *hook_data, void *call_data)
+{
+ struct LttvTracefileState *s = (LttvTraceFileState *)call_data;
+
+ pop_state(s, STATE_TRAP);
+ return FALSE;
+}
+
+
+gboolean irq_entry(void *hook_data, void *call_data)
+{
+ LttField *f = (LttField *)hook_data;
+
+ struct LttvTracefileState *s = (LttvTraceFileState *)call_data;
+
+ /* Do something with the info about being in user or system mode when int? */
+ push_state(s, STATE_IRQ, ltt_event_get_unsigned(s->e, f));
+ return FALSE;
+}
+
+
+gboolean irq_exit(void *hook_data, void *call_data)
+{
+ struct LttvTracefileState *s = (LttvTraceFileState *)call_data;
+
+ pop_state(s, STATE_IRQ);
+ return FALSE;
+}
+
+
+gboolean schedchange(void *hook_data, void *call_data)
+{
+ struct HookData *h = (struct HookData *)hook_data;
+
+ struct LttvTracefileState *s = (LttvTraceFileState *)call_data;
+
+ guint pid_in, pid_out, state_out;
+
+ pid_in = ltt_event_get_int(s->e, h->f1);
+ pid_out = ltt_event_get_int(s->e, h->f2);
+ state_out = ltt_event_get_int(s->e, h->f3);
+ if(s->process != NULL) {
+ if(state_out == 0) s->process->state->s = STATE_WAIT_CPU;
+ else if(s->process->state->s == STATE_EXIT)
+ exit_process(s->tc, s->process);
+ else s->process->state->s = STATE_WAIT;
+ }
+ s->process = find_process(s->tc, pid_in);
+ s->process->state->s = STATE_RUN;
+ return FALSE;
+}
+
+
+gboolean process_fork(void *hook_data, void *call_data)
+{
+ LttField *f = (LttField *)hook_data;
+
+ struct LttvTracefileState *s = (LttvTraceFileState *)call_data;
+
+ guint child_pid;
+
+ child_pid = ltt_event_get_int(s->e, f);
+ create_process(s->tc, s->process, child_pid);
+ return FALSE;
+}
+
+
+gboolean process_exit(void *hook_data, void *call_data)
+{
+ struct LttvTracefileState *s = (LttvTraceFileState *)call_data;
+
+ if(s->process != NULL) {
+ s->process->state->s = STATE_EXIT;
+ }
+ return FALSE;
+}
+
+
+static LttField *
+find_field(LttEventType *et, const char *field)
+{
+ LttType *t;
+
+ LttField *f;
+
+ guint i, nb;
+
+ char *name;
+
+ if(field == NULL) return NULL;
+
+ f = ltt_eventtype_field(et);
+ t = ltt_eventtype_type(et);
+ g_assert(ltt_type_class(t) == LTT_STRUCT);
+ nb = ltt_type_member_number(t);
+ for(i = 0 ; i < nb ; i++) {
+ ltt_type_member_type(t, i, &name);
+ if(strcmp(name, field) == 0) break;
+ }
+ g_assert(i < nb);
+ return ltt_field_member(f, i);
+}
+
+
+static HookId
+find_hook(LttTrace *t, const char *facility, const char *event,
+ const char *field1, *field2, *field3, LttHook h)
+{
+ LttFacility *f;
+
+ LttEventType *et;
+
+ guint nb, pos, i;
+
+ struct HookId hook_id;
+
+ struct HookData hook_data, *phook_data;
+
+ char *name;
+
+ nb = ltt_trace_facility_find(t, facility, &pos);
+ if(nb < 1) g_error("No %s facility", facility);
+ f = ltt_facility_get(t, pos);
+ et = ltt_facility_get_by_name(f, event);
+ if(et == NULL) g_error("Event %s does not exist", event);
+
+ hook_id.id = ltt_eventtype_id(et);
+ hook_id.h = h;
+ hook_id.free_hook_data = FALSE;
+ hook_data.f1 = find_field(et, field1);
+ hook_data.f2 = find_field(et, field2);
+ hook_data.f3 = find_field(et, field3);
+ if(hook_data->f1 == NULL) hook_id->hook_data = NULL;
+ else if(hook_data->f2 == NULL) hook_id->hook_data = hook_data->f1;
+ else {
+ phook_data = g_new(struct HookData, 1);
+ *phook_data = hook_data;
+ hook_id.hook_data = phook_data;
+ hook_id.free_hook_data = TRUE;
+ }
+ return hook_id;
+}
+
+
+lttv_state_add_event_hooks(LttvTracesetState *self)
+{
+ LttvTraceset *ts = self->ts;
+
+ guint i, j, nb_trace, nb_control, nb_per_cpu, nb_tracefile;
+
+ LttFacility *f;
+
+ LttEventType *et;
+
+ LttvTraceState *tc;
+
+ LttvTracefileState *tfc;
+
+ void *hook_data;
+
+ GArray *hooks;
+
+ struct HookId hook_id;
+
+ LttvAttributeValue val;
+
+ nb_trace = lttv_traceset_number(ts);
+ for(i = 0 ; i < nb_trace ; i++) {
+ tc = (LttvTraceState *)self->traces[i];
+
+ /* Find the eventtype id for the following events and register the
+ associated by id hooks. */
+
+ hooks = g_array_new(FALSE, FALSE, struct HookId);
+ g_array_add(hooks, find_hook(tc->t, "core","syscall_entry", "syscall_id",
+ NULL, NULL, syscall_entry));
+ g_array_add(hooks, find_hook(tc->t, "core", "syscall_exit", NULL, NULL,
+ NULL, syscall_exit));
+ g_array_add(hooks, find_hook(tc->t, "core", "trap_entry", "trap_id",
+ NULL, trap_entry));
+ g_array_add(hooks, find_hook(tc->t, "core", "trap_exit", NULL, NULL,
+ NULL, trap_exit));
+ g_array_add(hooks, find_hook(tc->t, "core", "irq_entry", "irq_id",
+ NULL, irq_entry));
+ g_array_add(hooks, find_hook(tc->t, "core", "irq_exit", NULL, NULL,
+ NULL, irq_exit));
+ g_array_add(hooks, find_hook(tc->t, "core", "schedchange", "in", "out",
+ "out_state", schedchange));
+ g_array_add(hooks, find_hook(tc->t, "core", "process_fork", "child_pid",
+ NULL, NULL, process_fork));
+ g_array_add(hooks, find_hook(tc->t, "core", "process_exit", NULL, NULL,
+ NULL, process_exit));
+
+ /* Add these hooks to each before_event_by_id hooks list */
+
+ nb_control = ltt_trace_control_tracefile_number(tc->t);
+ nb_per_cpu = ltt_trace_control_tracefile_number(tc->t);
+ nb_tracefile = nb_control + nb_per_cpu;
+ for(j = 0 ; j < nb_tracefile ; j++) {
+ if(j < nb_control) {
+ tfc = tc->control_tracefiles[j];
+ }
+ else {
+ tfc = tc->per_cpu_tracefiles[j];
+ }
+
+ for(k = 0 ; k < hooks->len ; k++) {
+ hook_id = g_array_index(hooks, struct HookId, k);
+ lttv_hooks_add(lttv_hooks_by_id_find(tfc->before_event_by_id,
+ hook_id->id), hook_id->h, hook_id->hook_data);
+ }
+ lttv_attribute_find(self->a, STATE_HOOKS, LTTV_POINTER, val);
+ val->v_pointer = hooks;
+ }
+}
+
+
+lttv_state_remove_event_hooks(LttvTracesetState *self)
+{
+ LttvTraceset *ts = self->ts;
+
+ guint i, j, nb_trace, nb_control, nb_per_cpu, nb_tracefile;
+
+ LttvTraceState *tc;
+
+ LttvTracefileState *tfc;
+
+ void *hook_data;
+
+ GArray *hooks;
+
+ struct HookId hook_id;
+
+ LttvAttributeValue val;
+
+ nb_trace = lttv_traceset_number(ts);
+ for(i = 0 ; i < nb_trace ; i++) {
+ tc = (LttvTraceState *)self->traces[i];
+ lttv_attribute_find(self->a, STATE_HOOKS, LTTV_POINTER, val);
+ hooks = val->v_pointer;
+
+ /* Add these hooks to each before_event_by_id hooks list */
+
+ nb_control = ltt_trace_control_tracefile_number(tc->t);
+ nb_per_cpu = ltt_trace_control_tracefile_number(tc->t);
+ nb_tracefile = nb_control + nb_per_cpu;
+ for(j = 0 ; j < nb_tracefile ; j++) {
+ if(j < nb_control) {
+ tfc = tc->control_tracefiles[j];
+ }
+ else {
+ tfc = tc->per_cpu_tracefiles[j];
+ }
+
+ for(k = 0 ; k < hooks->len ; k++) {
+ hook_id = g_array_index(hooks, struct HookId, k);
+ lttv_hooks_remove_data(lttv_hooks_by_id_find(tfc->before_event_by_id,
+ hook_id->id), hook_id->h, hook_id->hook_data);
+ }
+ for(k = 0 ; k < hooks->len ; k++) {
+ hook_id = g_array_index(hooks, struct HookId, k);
+ if(hook_id->free_hook_data) g_free(hook_id->hook_data);
+ }
+ g_array_free(hooks, TRUE);
+ }
+}
+
+
+
+
+
+
+
+
+
+