src/surf/surfxml_parseplatf.cpp

   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/cpu_interface.hpp"
  14 #include "src/surf/surf_private.h"
  15
  16 #ifdef HAVE_LUA
  17 #include "src/bindings/lua/simgrid_lua.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_one)
  41 {
  42   if (!new_one)
  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_one)
  51 {
  52   if (!new_one)
  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_host_speed = NULL;
  67 xbt_dict_t trace_connect_list_link_avail = NULL;
  68 xbt_dict_t trace_connect_list_link_bw = NULL;
  69 xbt_dict_t trace_connect_list_link_lat = 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     lua_State* L = luaL_newstate();
 103     luaL_openlibs(L);
 104
 105     luaL_loadfile(L, file); // This loads the file without executing it.
 106
 107     /* Run the script */
 108     if (lua_pcall(L, 0, 0, 0)) {
 109         XBT_ERROR("FATAL ERROR:\n  %s: %s\n\n", "Lua call failed. Errormessage:", lua_tostring(L, -1));
 110         xbt_die("Lua call failed. See Log");
 111     }
 112   }
 113   else
 114 #endif
 115   { // Use XML parser
 116
 117     int parse_status;
 118
 119     /* init the flex parser */
 120     surfxml_buffer_stack_stack_ptr = 1;
 121     surfxml_buffer_stack_stack[0] = 0;
 122     after_config_done = 0;
 123     surf_parse_open(file);
 124
 125     traces_set_list = xbt_dict_new_homogeneous(NULL);
 126     trace_connect_list_host_avail = xbt_dict_new_homogeneous(free);
 127     trace_connect_list_host_speed = xbt_dict_new_homogeneous(free);
 128     trace_connect_list_link_avail = xbt_dict_new_homogeneous(free);
 129     trace_connect_list_link_bw = xbt_dict_new_homogeneous(free);
 130     trace_connect_list_link_lat = xbt_dict_new_homogeneous(free);
 131
 132     /* Init my data */
 133     if (!surfxml_bufferstack_stack)
 134       surfxml_bufferstack_stack = xbt_dynar_new(sizeof(char *), NULL);
 135
 136     /* Do the actual parsing */
 137     parse_status = surf_parse();
 138
 139     /* connect all traces relative to hosts */
 140     xbt_dict_cursor_t cursor = NULL;
 141     char *trace_name, *elm;
 142
 143     xbt_dict_foreach(trace_connect_list_host_avail, cursor, trace_name, elm) {
 144       tmgr_trace_t trace = (tmgr_trace_t) xbt_dict_get_or_null(traces_set_list, trace_name);
 145       xbt_assert(trace, "Trace %s undefined", trace_name);
 146
 147       simgrid::s4u::Host *host = sg_host_by_name(elm);
 148       xbt_assert(host, "Host %s undefined", elm);
 149       simgrid::surf::Cpu *cpu = host->pimpl_cpu;
 150
 151       cpu->set_state_trace(trace);
 152     }
 153
 154     /* Free my data */
 155     xbt_dict_free(&trace_connect_list_host_avail);
 156     xbt_dict_free(&trace_connect_list_host_speed);
 157     xbt_dict_free(&trace_connect_list_link_avail);
 158     xbt_dict_free(&trace_connect_list_link_bw);
 159     xbt_dict_free(&trace_connect_list_link_lat);
 160     xbt_dict_free(&traces_set_list);
 161     xbt_dict_free(&random_data_list);
 162     xbt_dynar_free(&surfxml_bufferstack_stack);
 163
 164     surf_parse_close();
 165
 166     if (parse_status)
 167       surf_parse_error("Parse error in %s", file);
 168
 169   }
 170
 171
 172 }