src/surf/surfxml_parseplatf.c

   1 /* Copyright (c) 2006-2015. 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 #include "xbt/misc.h"
   8 #include "xbt/log.h"
   9 #include "xbt/str.h"
  10 #include "xbt/dict.h"
  11 #include "simgrid/platf.h"
  12 #include "surf/surfxml_parse.h"
  13 #include "src/surf/surf_private.h"
  14
  15 #ifdef HAVE_LUA
  16 #include "src/bindings/lua/simgrid_lua.h"
  17 #include "src/bindings/lua/lua_state_cloner.h"
  18
  19 #include <lua.h>                /* Always include this when calling Lua */
  20 #include <lauxlib.h>            /* Always include this when calling Lua */
  21 #include <lualib.h>             /* Prototype for luaL_openlibs(), */
  22 #endif
  23
  24 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(surf_parse);
  25
  26 /*
  27  *  Allow the cluster tag to mess with the parsing buffer.
  28  * (this will probably become obsolete once the cluster tag do not mess with the parsing callbacks directly)
  29  */
  30
  31 /* This buffer is used to store the original buffer before substituting it by out own buffer. Useful for the cluster tag */
  32 static xbt_dynar_t surfxml_bufferstack_stack = NULL;
  33 int surfxml_bufferstack_size = 2048;
  34
  35 static char *old_buff = NULL;
  36
  37 XBT_IMPORT_NO_EXPORT(unsigned int) surfxml_buffer_stack_stack_ptr;
  38 XBT_IMPORT_NO_EXPORT(unsigned int) surfxml_buffer_stack_stack[1024];
  39
  40 void surfxml_bufferstack_push(int new)
  41 {
  42   if (!new)
  43     old_buff = surfxml_bufferstack;
  44   else {
  45     xbt_dynar_push(surfxml_bufferstack_stack, &surfxml_bufferstack);
  46     surfxml_bufferstack = xbt_new0(char, surfxml_bufferstack_size);
  47   }
  48 }
  49
  50 void surfxml_bufferstack_pop(int new)
  51 {
  52   if (!new)
  53     surfxml_bufferstack = old_buff;
  54   else {
  55     free(surfxml_bufferstack);
  56     xbt_dynar_pop(surfxml_bufferstack_stack, &surfxml_bufferstack);
  57   }
  58 }
  59
  60 /*
  61  * Trace related stuff
  62  */
  63
  64 xbt_dict_t traces_set_list = NULL;
  65 xbt_dict_t trace_connect_list_host_avail = NULL;
  66 xbt_dict_t trace_connect_list_power = NULL;
  67 xbt_dict_t trace_connect_list_link_avail = NULL;
  68 xbt_dict_t trace_connect_list_bandwidth = NULL;
  69 xbt_dict_t trace_connect_list_latency = NULL;
  70
  71 /* ***************************************** */
  72
  73 static int after_config_done;
  74 void parse_after_config() {
  75   if (!after_config_done) {
  76           TRACE_start();
  77
  78     /* Register classical callbacks */
  79     storage_register_callbacks();
  80     routing_register_callbacks();
  81
  82     after_config_done = 1;
  83   }
  84 }
  85
  86 /* This function acts as a main in the parsing area. */
  87 void parse_platform_file(const char *file)
  88 {
  89 #ifdef HAVE_LUA
  90   int is_lua = (file != NULL && strlen(file) > 3 && file[strlen(file)-3] == 'l' && file[strlen(file)-2] == 'u'
  91         && file[strlen(file)-1] == 'a');
  92 #endif
  93
  94   surf_parse_init_callbacks();
  95
  96 #ifdef HAVE_LUA
  97   /* Check if file extension is "lua". If so, we will use
  98    * the lua bindings to parse the platform file (since it is
  99    * written in lua). If not, we will use the (old?) XML parser
 100    */
 101   if (is_lua) {
 102     // Get maestro state. In case we're calling Lua from
 103     // C only, this will be NULL -- no Lua code has been
 104     // executed yet and hence, the SimGrid module has not
 105     // yet been loaded.
 106     // NOTE: After executing the lua_pcall() below,
 107     // sglua_get_maestro() will not be NULL, since the
 108     // SimGrid module was loaded!
 109     lua_State* L = sglua_get_maestro();
 110
 111     // We may want to remove the task_copy_callback from
 112     // the SimGrid module if we're using C code only (this
 113     // callback is used for Lua-only code).
 114     int remove_callback = FALSE;
 115     if (L == NULL) {
 116         L = luaL_newstate();
 117         remove_callback = TRUE;
 118     }
 119     luaL_openlibs(L);
 120
 121     luaL_loadfile(L, file); // This loads the file without executing it.
 122
 123     /* Run the script */
 124     if (lua_pcall(L, 0, 0, 0)) {
 125         XBT_ERROR("FATAL ERROR:\n  %s: %s\n\n", "Lua call failed. Errormessage:", lua_tostring(L, -1));
 126         xbt_die("Lua call failed. See Log");
 127     }
 128     // Without this, task_copy_callback() will try to copy
 129     // some tasks -- but these don't exist in case we're using
 130     // C. Hence, we need to remove the callback -- we don't
 131     // want to segfault.
 132     if (remove_callback) {
 133       MSG_task_set_copy_callback(NULL);
 134     }
 135
 136   }
 137   else
 138 #endif
 139   { // Use XML parser
 140
 141     int parse_status;
 142
 143     /* init the flex parser */
 144     surfxml_buffer_stack_stack_ptr = 1;
 145     surfxml_buffer_stack_stack[0] = 0;
 146     after_config_done = 0;
 147     surf_parse_open(file);
 148
 149     traces_set_list = xbt_dict_new_homogeneous(NULL);
 150     trace_connect_list_host_avail = xbt_dict_new_homogeneous(free);
 151     trace_connect_list_power = xbt_dict_new_homogeneous(free);
 152     trace_connect_list_link_avail = xbt_dict_new_homogeneous(free);
 153     trace_connect_list_bandwidth = xbt_dict_new_homogeneous(free);
 154     trace_connect_list_latency = xbt_dict_new_homogeneous(free);
 155
 156     /* Init my data */
 157     if (!surfxml_bufferstack_stack)
 158       surfxml_bufferstack_stack = xbt_dynar_new(sizeof(char *), NULL);
 159
 160     /* Do the actual parsing */
 161     parse_status = surf_parse();
 162
 163     /* Free my data */
 164     xbt_dict_free(&trace_connect_list_host_avail);
 165     xbt_dict_free(&trace_connect_list_power);
 166     xbt_dict_free(&trace_connect_list_link_avail);
 167     xbt_dict_free(&trace_connect_list_bandwidth);
 168     xbt_dict_free(&trace_connect_list_latency);
 169     xbt_dict_free(&traces_set_list);
 170     xbt_dict_free(&random_data_list);
 171     xbt_dynar_free(&surfxml_bufferstack_stack);
 172
 173     surf_parse_close();
 174
 175     if (parse_status)
 176       surf_parse_error("Parse error in %s", file);
 177
 178   }
 179
 180
 181 }