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