Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
model-checker : initialize variables
[simgrid.git] / src / mc / mc_checkpoint.c
index d96dd72..340b1b0 100644 (file)
@@ -1,4 +1,4 @@
-/* Copyright (c) 2008-2012 Da SimGrid Team. All rights reserved.            */
+/* Copyright (c) 2008-2013 Da SimGrid Team. All rights reserved.            */
 
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
@@ -39,10 +39,12 @@ static void *get_stack_pointer(void *stack_context, void *heap);
 static void snapshot_stack_free(mc_snapshot_stack_t s);
 static xbt_dynar_t take_snapshot_ignore(void);
 
+static void get_hash_global(char *snapshot_hash, void *data1, void *data2);
+static void get_hash_local(char *snapshot_hash, xbt_dynar_t stacks);
+
 static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size)
 {
   mc_mem_region_t new_reg = xbt_new0(s_mc_mem_region_t, 1);
-  new_reg->type = type;
   new_reg->start_addr = start_addr;
   new_reg->size = size;
   new_reg->data = xbt_malloc0(size);
@@ -72,19 +74,118 @@ static void MC_region_destroy(mc_mem_region_t reg)
 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size)
 {
   mc_mem_region_t new_reg = MC_region_new(type, start_addr, size);
-  snapshot->regions = xbt_realloc(snapshot->regions, (snapshot->num_reg + 1) * sizeof(mc_mem_region_t));
-  snapshot->regions[snapshot->num_reg] = new_reg;
-  snapshot->region_type[snapshot->num_reg] = type;
-  snapshot->num_reg++;
+  snapshot->regions[type] = new_reg;
   return;
 } 
 
-void MC_init_memory_map_info(){
+static void get_memory_regions(mc_snapshot_t snapshot){
 
-  int raw_mem_set = (mmalloc_get_current_heap() == raw_heap);
+  FILE *fp;
+  char *line = NULL;
+  ssize_t read;
+  size_t n = 0;
+  
+  char *lfields[6] = {0}, *tok;
+  void *start_addr, *start_addr1, *end_addr;
+  size_t size;
+  int i;
 
-  MC_SET_RAW_MEM;
+  fp = fopen("/proc/self/maps", "r");
   
+  xbt_assert(fp, 
+             "Cannot open /proc/self/maps to investigate the memory map of the process. Please report this bug.");
+
+  setbuf(fp, NULL);
+
+  while((read = xbt_getline(&line, &n, fp)) != -1){
+
+    /* Wipeout the new line character */
+    line[read - 1] = '\0';
+
+    /* Tokenize the line using spaces as delimiters and store each token */
+    lfields[0] = strtok(line, " ");
+
+    for (i = 1; i < 6 && lfields[i - 1] != NULL; i++) {
+      lfields[i] = strtok(NULL, " ");
+    }
+
+    /* First get the permissions flags, need write permission */
+    if(lfields[1][1] == 'w'){
+
+      /* Get the start address of the map */
+      tok = strtok(lfields[0], "-");
+      start_addr = (void *)strtoul(tok, NULL, 16);
+    
+      if(start_addr == std_heap){     /* Std_heap ? */
+        tok = strtok(NULL, "-");
+        end_addr = (void *)strtoul(tok, NULL, 16);
+        MC_snapshot_add_region(snapshot, 0, start_addr, (char*)end_addr - (char*)start_addr);
+        snapshot->heap_bytes_used = mmalloc_get_bytes_used(std_heap);
+      }else{ /* map name == libsimgrid || binary_name ? */
+        if(lfields[5] != NULL){
+          if(!memcmp(basename(lfields[5]), "libsimgrid", 10)){
+            tok = strtok(NULL, "-");
+            end_addr = (void *)strtoul(tok, NULL, 16);
+            size = (char*)end_addr - (char*)start_addr;
+            /* BSS and data segments may be separated according to the OS */
+            if((read = xbt_getline(&line, &n, fp)) != -1){
+              line[read - 1] = '\0';
+              lfields[0] = strtok(line, " ");
+              for (i = 1; i < 6 && lfields[i - 1] != NULL; i++) {
+                lfields[i] = strtok(NULL, " ");
+              }
+              if(lfields[1][1] == 'w' && lfields[5] == NULL){
+                tok = strtok(lfields[0], "-");
+                start_addr1 = (void *)strtoul(tok, NULL, 16);
+                tok = strtok(NULL, "-");
+                size += (char *)(void *)strtoul(tok, NULL, 16) - (char*)start_addr1;
+              }
+            }
+            MC_snapshot_add_region(snapshot, 1, start_addr, size);
+          }else if(!memcmp(basename(lfields[5]), basename(xbt_binary_name), strlen(basename(xbt_binary_name)))){
+            tok = strtok(NULL, "-");
+            end_addr = (void *)strtoul(tok, NULL, 16);
+            size = (char*)end_addr - (char*)start_addr;
+             /* BSS and data segments may be separated according to the OS */
+            if((read = xbt_getline(&line, &n, fp)) != -1){
+              line[read - 1] = '\0';
+              lfields[0] = strtok(line, " ");
+              for (i = 1; i < 6 && lfields[i - 1] != NULL; i++) {
+                lfields[i] = strtok(NULL, " ");
+              }
+              tok = strtok(lfields[0], "-");
+              start_addr1 = (void *)strtoul(tok, NULL, 16);
+              if(lfields[1][1] == 'w' && lfields[5] == NULL){
+                if(start_addr1 == std_heap){     /* Std_heap ? */
+                  tok = strtok(NULL, "-");
+                  end_addr = (void *)strtoul(tok, NULL, 16);
+                  MC_snapshot_add_region(snapshot, 0, start_addr1, (char*)end_addr - (char*)start_addr1);
+                  snapshot->heap_bytes_used = mmalloc_get_bytes_used(std_heap);
+                }else if(start_addr1 != raw_heap){
+                  tok = strtok(NULL, "-");
+                  size += (char *)(void *)strtoul(tok, NULL, 16) - (char *)start_addr1;
+                }
+              }
+            }
+            MC_snapshot_add_region(snapshot, 2, start_addr, size);
+          }else if (!memcmp(lfields[5], "[stack]", 7)){
+            maestro_stack_start = start_addr;
+            tok = strtok(NULL, "-");
+            maestro_stack_end = (void *)strtoul(tok, NULL, 16);
+          }
+        }
+      }
+    }
+    
+  }
+
+  free(line);
+  fclose(fp);
+
+}
+
+void MC_init_memory_map_info(){
   unsigned int i = 0;
   s_map_region_t reg;
   memory_map_t maps = get_memory_map();
@@ -125,22 +226,14 @@ void MC_init_memory_map_info(){
     }
     i++;
   }
-  
+   
   free_memory_map(maps);
 
-  MC_UNSET_RAW_MEM;
-
-  if(raw_mem_set)
-    MC_SET_RAW_MEM;
-
-}
-
-mc_snapshot_t SIMIX_pre_mc_snapshot(smx_simcall_t simcall){
-  return MC_take_snapshot();
 }
 
 mc_snapshot_t MC_take_snapshot()
 {
+
   int raw_mem = (mmalloc_get_current_heap() == raw_heap);
   
   MC_SET_RAW_MEM;
@@ -148,66 +241,16 @@ mc_snapshot_t MC_take_snapshot()
   mc_snapshot_t snapshot = xbt_new0(s_mc_snapshot_t, 1);
   snapshot->nb_processes = xbt_swag_size(simix_global->process_list);
 
-  unsigned int i = 0;
-  s_map_region_t reg;
-  memory_map_t maps = get_memory_map();
-  void *heap = NULL;
-  size_t size = 0;
-  void *start = NULL;
-
-  /* Save the std heap and the writable mapped pages of libsimgrid */
-  while (i < maps->mapsize) {
-    reg = maps->regions[i];
-    if ((reg.prot & PROT_WRITE)){
-      if (maps->regions[i].pathname == NULL){
-        if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
-          MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
-          snapshot->heap_bytes_used = mmalloc_get_bytes_used(std_heap);
-          heap = snapshot->regions[snapshot->num_reg - 1]->data;
-        }
-        i++;
-      } else{ 
-        if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
-          size = (char*)reg.end_addr - (char*)reg.start_addr;
-          start = reg.start_addr;
-          i++;
-          reg = maps->regions[i];
-          if(reg.pathname == NULL && (reg.prot & PROT_WRITE) && i < maps->mapsize){
-            size += (char*)reg.end_addr - (char*)reg.start_addr;
-            reg = maps->regions[i];
-            i++;
-          }
-          MC_snapshot_add_region(snapshot, 1, start, size);
-        }else if(!memcmp(maps->regions[i].pathname, "[stack]", 7)){
-          maestro_stack_start = reg.start_addr;
-          maestro_stack_end = reg.end_addr;
-          i++;
-        } else if (!memcmp(basename(maps->regions[i].pathname), basename(xbt_binary_name), strlen(basename(xbt_binary_name)))){
-          size = (char*)reg.end_addr - (char*)reg.start_addr;
-          start = reg.start_addr;
-          i++;
-          reg = maps->regions[i];
-          if(reg.pathname == NULL && (reg.prot & PROT_WRITE) && reg.start_addr != std_heap && reg.start_addr != raw_heap && i < maps->mapsize){
-            size += (char*)reg.end_addr - (char*)reg.start_addr;
-            reg = maps->regions[i];
-            i++;
-          }
-          MC_snapshot_add_region(snapshot, 2, start, size);
-        }else{
-          i++;
-        }
-      }
-    }else{
-      i++;
-    }
-  }
+  /* Save the std heap and the writable mapped pages of libsimgrid and binary */
+  get_memory_regions(snapshot);
 
   snapshot->to_ignore = take_snapshot_ignore();
 
-  if(_sg_mc_visited > 0 || strcmp(_sg_mc_property_file,""))
-    snapshot->stacks = take_snapshot_stacks(&snapshot, heap);
-  
-  free_memory_map(maps);
+  if(_sg_mc_visited > 0 || strcmp(_sg_mc_property_file,"")){
+    snapshot->stacks = take_snapshot_stacks(&snapshot, snapshot->regions[0]->data);
+    get_hash_global(snapshot->hash_global, snapshot->regions[1]->data, snapshot->regions[2]->data);
+    get_hash_local(snapshot->hash_local, snapshot->stacks);
+  }
 
   MC_UNSET_RAW_MEM;
 
@@ -221,7 +264,7 @@ mc_snapshot_t MC_take_snapshot()
 void MC_restore_snapshot(mc_snapshot_t snapshot)
 {
   unsigned int i;
-  for(i=0; i < snapshot->num_reg; i++){
+  for(i=0; i < NB_REGIONS; i++){
     MC_region_restore(snapshot->regions[i]);
   }
 
@@ -230,10 +273,10 @@ void MC_restore_snapshot(mc_snapshot_t snapshot)
 void MC_free_snapshot(mc_snapshot_t snapshot)
 {
   unsigned int i;
-  for(i=0; i < snapshot->num_reg; i++)
+  for(i=0; i < NB_REGIONS; i++)
     MC_region_destroy(snapshot->regions[i]);
 
-  xbt_free(snapshot->regions);
+  xbt_free(snapshot->stack_sizes);
   xbt_dynar_free(&(snapshot->stacks));
   xbt_dynar_free(&(snapshot->to_ignore));
   xbt_free(snapshot);
@@ -463,10 +506,13 @@ static xbt_strbuff_t get_local_variables_values(void *stack_context, void *heap)
       return variables;
     }
 
-    to_append = bprintf("ip=%s\n", frame_name);
+    to_append = bprintf("frame_name=%s\n", frame_name);
     xbt_strbuff_append(variables, to_append);
     xbt_free(to_append);
-
+    to_append = bprintf("ip=%lx\n", (unsigned long)ip);
+    xbt_strbuff_append(variables, to_append);
+    xbt_free(to_append);
+    
     true_ip = (long)frame->low_pc + (long)off;
 
     /* Get frame pointer */
@@ -498,8 +544,9 @@ static xbt_strbuff_t get_local_variables_values(void *stack_context, void *heap)
               cursor2++;
             }
 
-            if(xbt_dynar_length(compose) > 0){
+            if(!xbt_dynar_is_empty(compose)){
               frame_pointer_address = xbt_dynar_get_as(compose, xbt_dynar_length(compose) - 1, variable_value_t)->value.address ; 
+              xbt_dynar_reset(compose);
             }
             break;
           default :
@@ -546,7 +593,7 @@ static xbt_strbuff_t get_local_variables_values(void *stack_context, void *heap)
             cursor++;
           }
           
-          if(xbt_dynar_length(compose) > 0){
+          if(!xbt_dynar_is_empty(compose)){
             if(strcmp(xbt_dynar_get_as(compose, xbt_dynar_length(compose) - 1, variable_value_t)->type, "value") == 0){
               to_append = bprintf("%s=%lx\n", current_variable->name, xbt_dynar_get_as(compose, xbt_dynar_length(compose) - 1, variable_value_t)->value.res);
               xbt_strbuff_append(variables, to_append);
@@ -566,6 +613,7 @@ static xbt_strbuff_t get_local_variables_values(void *stack_context, void *heap)
                 xbt_free(to_append);
               }
             }
+            xbt_dynar_reset(compose);
           }else{
             to_append = bprintf("%s=undefined\n", current_variable->name);
             xbt_strbuff_append(variables, to_append);
@@ -616,6 +664,10 @@ void snapshot_stack_free_voidp(void *s){
   snapshot_stack_free((mc_snapshot_stack_t) * (void **) s);
 }
 
+mc_snapshot_t SIMIX_pre_mc_snapshot(smx_simcall_t simcall){
+  return MC_take_snapshot();
+}
+
 void *MC_snapshot(void){
 
   return simcall_mc_snapshot();
@@ -633,6 +685,101 @@ void variable_value_free_voidp(void* v){
   variable_value_free((variable_value_t) * (void **)v);
 }
 
+static void get_hash_global(char *snapshot_hash, void *data1, void *data2){
+  
+  unsigned int cursor = 0;
+  size_t offset; 
+  global_variable_t current_var; 
+  void *addr_pointed = NULL;
+  void *res = NULL;
+
+  xbt_strbuff_t clear = xbt_strbuff_new();
+  
+  xbt_dynar_foreach(mc_global_variables, cursor, current_var){
+    if(current_var->address < start_data_libsimgrid){ /* binary */
+      offset = (char *)current_var->address - (char *)start_data_binary;
+      addr_pointed = *((void **)((char *)data2 + offset));
+      if(((addr_pointed >= start_plt_binary && addr_pointed <= end_plt_binary)) || ((addr_pointed >= std_heap && (char *)addr_pointed <= (char *)std_heap + STD_HEAP_SIZE )))
+        continue;
+      res = xbt_malloc0(current_var->size + 1);
+      memset(res, 0, current_var->size + 1);
+      memcpy(res, (char*)data2 + offset, current_var->size);
+    }else{ /* libsimgrid */
+      offset = (char *)current_var->address - (char *)start_data_libsimgrid;
+      addr_pointed = *((void **)((char *)data1 + offset));
+      if((addr_pointed >= start_plt_libsimgrid && addr_pointed <= end_plt_libsimgrid) || (addr_pointed >= std_heap && (char *)addr_pointed <= (char *)std_heap + STD_HEAP_SIZE ))
+        continue;
+      res = xbt_malloc0(current_var->size + 1);
+      memset(res, 0, current_var->size + 1);
+      memcpy(res, (char*)data1 + offset, current_var->size);
+    }
+    if(res != NULL){
+      xbt_strbuff_append(clear, (const char*)res);
+      xbt_free(res);
+      res = NULL;
+    }
+  }
+
+  xbt_sha(clear->data, snapshot_hash);
+
+  xbt_strbuff_free(clear);
+
+}
+
+static void get_hash_local(char *snapshot_hash, xbt_dynar_t stacks){
+
+  xbt_dynar_t tokens = NULL, s_tokens = NULL;
+  unsigned int cursor1 = 0, cursor2 = 0;
+  mc_snapshot_stack_t current_stack;
+  char *frame_name = NULL;
+  void *addr;
+
+  xbt_strbuff_t clear = xbt_strbuff_new();
+
+  while(cursor1 < xbt_dynar_length(stacks)){
+    current_stack = xbt_dynar_get_as(stacks, cursor1, mc_snapshot_stack_t);
+    tokens = xbt_str_split(current_stack->local_variables->data, NULL);
+    cursor2 = 0;
+    while(cursor2 < xbt_dynar_length(tokens)){
+      s_tokens = xbt_str_split(xbt_dynar_get_as(tokens, cursor2, char *), "=");
+      if(xbt_dynar_length(s_tokens) > 1){
+        if(strcmp(xbt_dynar_get_as(s_tokens, 0, char *), "frame_name") == 0){
+          xbt_free(frame_name);
+          frame_name = xbt_strdup(xbt_dynar_get_as(s_tokens, 1, char *));
+          xbt_strbuff_append(clear, (const char*)xbt_dynar_get_as(tokens, cursor2, char *));
+          cursor2++;
+          xbt_dynar_free(&s_tokens);
+          continue;
+        }
+        addr = (void *) strtoul(xbt_dynar_get_as(s_tokens, 1, char *), NULL, 16);
+        if(addr > std_heap && (char *)addr <= (char *)std_heap + STD_HEAP_SIZE){
+          cursor2++;
+          xbt_dynar_free(&s_tokens);
+          continue;
+        }
+        if(is_stack_ignore_variable(frame_name, xbt_dynar_get_as(s_tokens, 0, char *))){
+          cursor2++;
+          xbt_dynar_free(&s_tokens);
+          continue;
+        }
+        xbt_strbuff_append(clear, (const char *)xbt_dynar_get_as(tokens, cursor2, char *));
+      }
+      xbt_dynar_free(&s_tokens);
+      cursor2++;
+    }
+    xbt_dynar_free(&tokens);
+    cursor1++;
+  }
+
+  xbt_free(frame_name);
+
+  xbt_sha(clear->data, snapshot_hash);
+
+  xbt_strbuff_free(clear);
+
+}
+
+
 static xbt_dynar_t take_snapshot_ignore(){
   
   if(mc_heap_comparison_ignore == NULL)