Refine the interactions between the hooks provided by the different modules.

[lttv.git] / ltt / branches / poly / lttv / state.c

#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);
  }
}