teshsuite/mc/dwarf/dwarf.cpp

   1 /* Copyright (c) 2014. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #ifdef NDEBUG
   8 #undef NDEBUG
   9 #endif
  10
  11 #include <string.h>
  12 #include <assert.h>
  13
  14 #include <xbt.h>
  15 #include <mc/mc.h>
  16
  17 #include "../../src/include/mc/datatypes.h"
  18 #include "../../src/mc/mc_object_info.h"
  19 #include "../../src/mc/mc_private.h"
  20
  21 int test_some_array[4][5][6];
  22 struct some_struct { int first; int second[4][5]; } test_some_struct;
  23
  24 static mc_type_t find_type_by_name(mc_object_info_t info, const char* name) {
  25   xbt_dict_cursor_t cursor = NULL;
  26   char *key;
  27   mc_type_t type;
  28   xbt_dict_foreach(info->types, cursor, key, type)
  29     if(type->name == name)
  30       return type;
  31   return NULL;
  32 }
  33
  34 static dw_frame_t find_function_by_name(mc_object_info_t info, const char* name) {
  35   xbt_dict_cursor_t cursor = 0;
  36   dw_frame_t subprogram;
  37   char* key;
  38   xbt_dict_foreach(info->subprograms, cursor, key, subprogram){
  39     if(!strcmp(name, subprogram->name))
  40       return subprogram;
  41   }
  42
  43   return NULL;
  44 }
  45
  46 static dw_variable_t find_local_variable(dw_frame_t frame, const char* argument_name) {
  47   unsigned int cursor = 0;
  48   dw_variable_t variable;
  49   xbt_dynar_foreach(frame->variables, cursor, variable){
  50     if(!strcmp(argument_name, variable->name))
  51       return variable;
  52   }
  53
  54   dw_frame_t scope = NULL;
  55   xbt_dynar_foreach(frame->scopes, cursor, scope) {
  56     variable = find_local_variable(scope, argument_name);
  57     if(variable)
  58       return variable;
  59   }
  60
  61   return NULL;
  62 }
  63
  64 static void test_local_variable(mc_object_info_t info, const char* function, const char* variable, void* address, unw_cursor_t* cursor) {
  65   dw_frame_t subprogram = find_function_by_name(info, function);
  66   assert(subprogram);
  67   // TODO, Lookup frame by IP and test against name instead
  68
  69   dw_variable_t var = find_local_variable(subprogram, variable);
  70   assert(var);
  71
  72   void* frame_base = mc_find_frame_base(subprogram, info, cursor);
  73   s_mc_location_t location;
  74
  75   mc_dwarf_resolve_locations(&location,
  76     &var->locations, info, cursor, frame_base, NULL, -1);
  77
  78   xbt_assert(mc_get_location_type(&location)==MC_LOCATION_TYPE_ADDRESS,
  79     "Unexpected location type for variable %s of %s", variable, function);
  80
  81   xbt_assert(location.memory_location == address,
  82     "Bad resolution of local variable %s of %s", variable, function);
  83
  84 }
  85
  86 static dw_variable_t test_global_variable(mc_process_t process, mc_object_info_t info, const char* name, void* address, long byte_size) {
  87
  88   dw_variable_t variable = info->find_variable(name);
  89   xbt_assert(variable, "Global variable %s was not found", name);
  90   xbt_assert(!strcmp(variable->name, name), "Name mismatch for %s", name);
  91   xbt_assert(variable->global, "Variable %s is not global", name);
  92   xbt_assert(variable->address == address,
  93       "Address mismatch for %s : %p expected but %p found", name, address, variable->address);
  94
  95   mc_type_t type = (mc_type_t) xbt_dict_get_or_null(process->binary_info->types, variable->type_origin);
  96   xbt_assert(type!=NULL, "Missing type for %s", name);
  97   xbt_assert(type->byte_size = byte_size, "Byte size mismatch for %s", name);
  98   return variable;
  99 }
 100
 101 static mc_type_t find_member(mc_object_info_t info, const char* name, mc_type_t type) {
 102   unsigned int cursor = 0;
 103   mc_type_t member;
 104   xbt_dynar_foreach(type->members, cursor, member)
 105     if(member->name == name)
 106       return member;
 107   return NULL;
 108 }
 109
 110 int some_local_variable = 0;
 111
 112 typedef struct foo {int i;} s_foo;
 113
 114 static void test_type_by_name(mc_process_t process, s_foo my_foo) {
 115   assert(xbt_dict_get_or_null(process->binary_info->full_types_by_name, "struct foo"));
 116 }
 117
 118 int main(int argc, char** argv)
 119 {
 120   SIMIX_global_init(&argc, argv);
 121
 122   dw_variable_t var;
 123   mc_type_t type;
 124
 125   s_mc_process_t p(getpid(), -1);
 126   mc_process_t process = &p;
 127
 128   test_global_variable(process, process->binary_info.get(),
 129     "some_local_variable", &some_local_variable, sizeof(int));
 130
 131   var = test_global_variable(process, process->binary_info.get(),
 132     "test_some_array", &test_some_array, sizeof(test_some_array));
 133   type = (mc_type_t) xbt_dict_get_or_null(process->binary_info->types, var->type_origin);
 134   xbt_assert(type->element_count == 6*5*4, "element_count mismatch in test_some_array : %i / %i", type->element_count, 6*5*4);
 135
 136   var = test_global_variable(process, process->binary_info.get(),
 137     "test_some_struct", &test_some_struct, sizeof(test_some_struct));
 138   type = (mc_type_t) xbt_dict_get_or_null(process->binary_info->types, var->type_origin);
 139   assert(find_member(process->binary_info.get(), "first", type)->offset == 0);
 140   assert(find_member(process->binary_info.get(), "second", type)->offset
 141       == ((const char*)&test_some_struct.second) - (const char*)&test_some_struct);
 142
 143   unw_context_t context;
 144   unw_cursor_t cursor;
 145   unw_getcontext(&context);
 146   unw_init_local(&cursor, &context);
 147
 148   test_local_variable(process->binary_info.get(), "main", "argc", &argc, &cursor);
 149
 150   {
 151     int lexical_block_variable = 50;
 152     test_local_variable(process->binary_info.get(), "main",
 153       "lexical_block_variable", &lexical_block_variable, &cursor);
 154   }
 155
 156   s_foo my_foo;
 157   test_type_by_name(process, my_foo);
 158
 159   _exit(0);
 160 }