src/surf/xml/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 "src/instr/instr_private.h" // TRACE_start(). FIXME: remove by subscribing tracing to the surf signals
   8 #include "src/surf/cpu_interface.hpp"
   9 #include "src/surf/network_interface.hpp"
  10 #include "xbt/dict.h"
  11 #include "xbt/log.h"
  12 #include "xbt/misc.h"
  13 #include "xbt/str.h"
  14 #include <vector>
  15
  16 #include "src/surf/xml/platf_private.hpp"
  17
  18 #if SIMGRID_HAVE_LUA
  19 extern "C" {
  20 #include "src/bindings/lua/simgrid_lua.h"
  21
  22 #include <lua.h>                /* Always include this when calling Lua */
  23 #include <lauxlib.h>            /* Always include this when calling Lua */
  24 #include <lualib.h>             /* Prototype for luaL_openlibs(), */
  25 }
  26 #endif
  27
  28 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(surf_parse);
  29
  30 SG_BEGIN_DECL()
  31
  32 /*
  33  *  Allow the cluster tag to mess with the parsing buffer.
  34  * (this will probably become obsolete once the cluster tag do not mess with the parsing callbacks directly)
  35  */
  36
  37 /* This buffer is used to store the original buffer before substituting it by out own buffer. Useful for the cluster tag */
  38 static std::vector<char*> surfxml_bufferstack_stack;
  39 int surfxml_bufferstack_size = 2048;
  40
  41 static char *old_buff = nullptr;
  42
  43 XBT_IMPORT_NO_EXPORT(unsigned int) surfxml_buffer_stack_stack_ptr;
  44 XBT_IMPORT_NO_EXPORT(unsigned int) surfxml_buffer_stack_stack[1024];
  45
  46 void surfxml_bufferstack_push(int new_one)
  47 {
  48   if (!new_one)
  49     old_buff = surfxml_bufferstack;
  50   else {
  51     surfxml_bufferstack_stack.push_back(surfxml_bufferstack);
  52     surfxml_bufferstack = xbt_new0(char, surfxml_bufferstack_size);
  53   }
  54 }
  55
  56 void surfxml_bufferstack_pop(int new_one)
  57 {
  58   if (!new_one)
  59     surfxml_bufferstack = old_buff;
  60   else {
  61     free(surfxml_bufferstack);
  62     surfxml_bufferstack = surfxml_bufferstack_stack.back();
  63     surfxml_bufferstack_stack.pop_back();
  64   }
  65 }
  66
  67 /*
  68  * Trace related stuff
  69  */
  70
  71 xbt_dict_t traces_set_list = nullptr;
  72 XBT_PRIVATE xbt_dict_t trace_connect_list_host_avail = nullptr;
  73 XBT_PRIVATE xbt_dict_t trace_connect_list_host_speed = nullptr;
  74 XBT_PRIVATE xbt_dict_t trace_connect_list_link_avail = nullptr;
  75 XBT_PRIVATE xbt_dict_t trace_connect_list_link_bw = nullptr;
  76 XBT_PRIVATE xbt_dict_t trace_connect_list_link_lat = nullptr;
  77
  78 void sg_platf_trace_connect(sg_platf_trace_connect_cbarg_t trace_connect)
  79 {
  80   xbt_assert(xbt_dict_get_or_null(traces_set_list, trace_connect->trace),
  81               "Cannot connect trace %s to %s: trace unknown",
  82               trace_connect->trace,
  83               trace_connect->element);
  84
  85   switch (trace_connect->kind) {
  86   case SURF_TRACE_CONNECT_KIND_HOST_AVAIL:
  87     xbt_dict_set(trace_connect_list_host_avail,
  88         trace_connect->trace,
  89         xbt_strdup(trace_connect->element), nullptr);
  90     break;
  91   case SURF_TRACE_CONNECT_KIND_SPEED:
  92     xbt_dict_set(trace_connect_list_host_speed, trace_connect->trace,
  93         xbt_strdup(trace_connect->element), nullptr);
  94     break;
  95   case SURF_TRACE_CONNECT_KIND_LINK_AVAIL:
  96     xbt_dict_set(trace_connect_list_link_avail,
  97         trace_connect->trace,
  98         xbt_strdup(trace_connect->element), nullptr);
  99     break;
 100   case SURF_TRACE_CONNECT_KIND_BANDWIDTH:
 101     xbt_dict_set(trace_connect_list_link_bw,
 102         trace_connect->trace,
 103         xbt_strdup(trace_connect->element), nullptr);
 104     break;
 105   case SURF_TRACE_CONNECT_KIND_LATENCY:
 106     xbt_dict_set(trace_connect_list_link_lat, trace_connect->trace,
 107         xbt_strdup(trace_connect->element), nullptr);
 108     break;
 109   default:
 110   surf_parse_error("Cannot connect trace %s to %s: kind of trace unknown",
 111         trace_connect->trace, trace_connect->element);
 112     break;
 113   }
 114 }
 115
 116
 117 /* ***************************************** */
 118
 119 static int after_config_done;
 120 void parse_after_config() {
 121   if (!after_config_done) {
 122     TRACE_start();
 123
 124     /* Register classical callbacks */
 125     storage_register_callbacks();
 126
 127     after_config_done = 1;
 128   }
 129 }
 130
 131 /* This function acts as a main in the parsing area. */
 132 void parse_platform_file(const char *file)
 133 {
 134 #if SIMGRID_HAVE_LUA
 135   int len    = (file == nullptr ? 0 : strlen(file));
 136   int is_lua = (file != nullptr && len > 3 && file[len - 3] == 'l' && file[len - 2] == 'u' && file[len - 1] == 'a');
 137 #endif
 138
 139   sg_platf_init();
 140
 141 #if SIMGRID_HAVE_LUA
 142   /* Check if file extension is "lua". If so, we will use
 143    * the lua bindings to parse the platform file (since it is
 144    * written in lua). If not, we will use the (old?) XML parser
 145    */
 146   if (is_lua) {
 147     lua_State* L = luaL_newstate();
 148     luaL_openlibs(L);
 149
 150     luaL_loadfile(L, file); // This loads the file without executing it.
 151
 152     /* Run the script */
 153     if (lua_pcall(L, 0, 0, 0)) {
 154       XBT_ERROR("FATAL ERROR:\n  %s: %s\n\n", "Lua call failed. Error message:", lua_tostring(L, -1));
 155       xbt_die("Lua call failed. See Log");
 156     }
 157     lua_close(L);
 158   }
 159   else
 160 #endif
 161   { // Use XML parser
 162
 163     int parse_status;
 164
 165     /* init the flex parser */
 166     surfxml_buffer_stack_stack_ptr = 1;
 167     surfxml_buffer_stack_stack[0] = 0;
 168     after_config_done = 0;
 169     surf_parse_open(file);
 170
 171     traces_set_list = xbt_dict_new_homogeneous(nullptr);
 172     trace_connect_list_host_avail = xbt_dict_new_homogeneous(free);
 173     trace_connect_list_host_speed = xbt_dict_new_homogeneous(free);
 174     trace_connect_list_link_avail = xbt_dict_new_homogeneous(free);
 175     trace_connect_list_link_bw = xbt_dict_new_homogeneous(free);
 176     trace_connect_list_link_lat = xbt_dict_new_homogeneous(free);
 177
 178     /* Do the actual parsing */
 179     parse_status = surf_parse();
 180
 181     /* connect all traces relative to hosts */
 182     xbt_dict_cursor_t cursor = nullptr;
 183     char *trace_name, *elm;
 184
 185     xbt_dict_foreach(trace_connect_list_host_avail, cursor, trace_name, elm) {
 186       tmgr_trace_t trace = (tmgr_trace_t) xbt_dict_get_or_null(traces_set_list, trace_name);
 187       xbt_assert(trace, "Trace %s undefined", trace_name);
 188
 189       simgrid::s4u::Host *host = sg_host_by_name(elm);
 190       xbt_assert(host, "Host %s undefined", elm);
 191       simgrid::surf::Cpu *cpu = host->pimpl_cpu;
 192
 193       cpu->setStateTrace(trace);
 194     }
 195     xbt_dict_foreach(trace_connect_list_host_speed, cursor, trace_name, elm) {
 196       tmgr_trace_t trace = (tmgr_trace_t) xbt_dict_get_or_null(traces_set_list, trace_name);
 197       xbt_assert(trace, "Trace %s undefined", trace_name);
 198
 199       simgrid::s4u::Host *host = sg_host_by_name(elm);
 200       xbt_assert(host, "Host %s undefined", elm);
 201       simgrid::surf::Cpu *cpu = host->pimpl_cpu;
 202
 203       cpu->setSpeedTrace(trace);
 204     }
 205     xbt_dict_foreach(trace_connect_list_link_avail, cursor, trace_name, elm) {
 206       tmgr_trace_t trace = (tmgr_trace_t) xbt_dict_get_or_null(traces_set_list, trace_name);
 207       xbt_assert(trace, "Trace %s undefined", trace_name);
 208       sg_link_t link = simgrid::s4u::Link::byName(elm);
 209       xbt_assert(link, "Link %s undefined", elm);
 210       link->setStateTrace(trace);
 211     }
 212
 213     xbt_dict_foreach(trace_connect_list_link_bw, cursor, trace_name, elm) {
 214       tmgr_trace_t trace = (tmgr_trace_t) xbt_dict_get_or_null(traces_set_list, trace_name);
 215       xbt_assert(trace, "Trace %s undefined", trace_name);
 216       sg_link_t link = simgrid::s4u::Link::byName(elm);
 217       xbt_assert(link, "Link %s undefined", elm);
 218       link->setBandwidthTrace(trace);
 219     }
 220
 221     xbt_dict_foreach(trace_connect_list_link_lat, cursor, trace_name, elm) {
 222       tmgr_trace_t trace = (tmgr_trace_t) xbt_dict_get_or_null(traces_set_list, trace_name);
 223       xbt_assert(trace, "Trace %s undefined", trace_name);
 224       sg_link_t link = simgrid::s4u::Link::byName(elm);
 225       xbt_assert(link, "Link %s undefined", elm);
 226       link->setLatencyTrace(trace);
 227     }
 228
 229     /* Free my data */
 230     xbt_dict_free(&trace_connect_list_host_avail);
 231     xbt_dict_free(&trace_connect_list_host_speed);
 232     xbt_dict_free(&trace_connect_list_link_avail);
 233     xbt_dict_free(&trace_connect_list_link_bw);
 234     xbt_dict_free(&trace_connect_list_link_lat);
 235     xbt_dict_free(&traces_set_list);
 236
 237     surf_parse_close();
 238
 239     if (parse_status)
 240       surf_parse_error("Parse error in %s", file);
 241   }
 242 }
 243
 244 SG_END_DECL()