2 # -*- coding: utf-8 -*-
4 # Copyright (c) 2014-2015. The SimGrid Team. All rights reserved.
6 # This program is free software; you can redistribute it and/or modify it
7 # under the terms of the license (GNU LGPL) which comes with this package.
14 def __init__(self, name, type):
19 return self.simcall_types[self.type]
24 class Simcall(object):
28 def __init__(self, name, handler, res, args, call_kind):
32 self.need_handler = handler
33 self.call_kind = call_kind
36 # libsmx.c simcall_BODY_
37 if self.simcalls_BODY is None:
38 f = open('libsmx.cpp')
39 self.simcalls_BODY = set(
40 re.findall('simcall_BODY_(.*?)\(', f.read()))
42 if self.name not in self.simcalls_BODY:
43 print '# ERROR: No function calling simcall_BODY_%s' % self.name
44 print '# Add something like this to libsmx.c:'
45 print '%s simcall_%s(%s) {' % (self.res.rettype(), self.name, ', '.join('%s %s' % (arg.rettype(), arg.name) for arg in self.args))
46 print ' return simcall_BODY_%s(%s);' % (self.name, "...")
50 # smx_*.c void simcall_HANDLER_host_on(smx_simcall_t simcall,
52 if self.simcalls_PRE is None:
53 self.simcalls_PRE = set()
54 for fn in glob.glob('smx_*') + glob.glob('../mc/*'):
56 self.simcalls_PRE |= set(
57 re.findall('simcall_HANDLER_(.*?)\(', f.read()))
60 if (self.name not in self.simcalls_PRE):
61 print '# ERROR: No function called simcall_HANDLER_%s' % self.name
62 print '# Add something like this to the relevant C file (like smx_io.c if it\'s an IO call):'
63 print '%s simcall_HANDLER_%s(smx_simcall_t simcall%s) {' % (self.res.rettype(), self.name, ''.join(', %s %s' % (arg.rettype(), arg.name)
64 for arg in self.args))
65 print ' // Your code handling the simcall'
69 if (self.name in self.simcalls_PRE):
70 print '# ERROR: You have a function called simcall_HANDLER_%s, but that simcall is not using any handler' % self.name
71 print '# Either change your simcall definition, or kill that function'
76 return ' SIMCALL_%s,' % (self.name.upper())
79 return ' "SIMCALL_%s",' % self.name.upper()
84 # Arguments getter/setters
85 for i in range(len(self.args)):
87 res.append('static inline %s simcall_%s__get__%s(smx_simcall_t simcall) {' % (
88 arg.rettype(), self.name, arg.name))
90 ' return simgrid::simix::unmarshal<%s>(simcall->args[%i]);' % (arg.rettype(), i))
92 res.append('static inline void simcall_%s__set__%s(smx_simcall_t simcall, %s arg) {' % (
93 self.name, arg.name, arg.rettype()))
94 res.append(' simgrid::simix::marshal<%s>(simcall->args[%i], arg);' % (arg.rettype(), i))
97 # Return value getter/setters
98 if self.res.type != 'void':
100 'static inline %s simcall_%s__get__result(smx_simcall_t simcall){' % (self.res.rettype(), self.name))
101 res.append(' return simgrid::simix::unmarshal<%s>(simcall->result);' % self.res.rettype())
104 'static inline void simcall_%s__set__result(smx_simcall_t simcall, %s result){' % (self.name, self.res.rettype()))
105 res.append(' simgrid::simix::marshal<%s>(simcall->result, result);' % (self.res.rettype()))
107 return '\n'.join(res)
111 args = [ ("simgrid::simix::unmarshal<%s>(simcall->args[%d])" % (arg.rettype(), i))
112 for i, arg in enumerate(self.args) ]
113 res.append('case SIMCALL_%s:' % (self.name.upper()))
114 if self.need_handler:
115 call = "simcall_HANDLER_%s(simcall%s%s)" % (self.name,
116 ", " if len(args) > 0 else "",
119 call = "SIMIX_%s(%s)" % (self.name, ', '.join(args))
120 if self.call_kind == 'Func':
121 res.append(" simgrid::simix::marshal<%s>(simcall->result, %s);" % (self.res.rettype(), call))
123 res.append(" " + call + ";");
124 res.append(' %sbreak; \n' %
125 ('SIMIX_simcall_answer(simcall);\n ' if self.call_kind != 'Blck' else ' '))
126 return '\n'.join(res)
131 'inline static %s simcall_BODY_%s(%s) {' % (self.res.rettype(),
133 ', '.join('%s %s' % (arg.rettype(), arg.name) for arg in self.args)))
135 ' /* Go to that function to follow the code flow through the simcall barrier */')
136 if self.need_handler:
137 res.append(' if (0) simcall_HANDLER_%s(%s);' % (self.name,
138 ', '.join(["&SIMIX_process_self()->simcall"] + [arg.name for arg in self.args])))
140 res.append(' if (0) SIMIX_%s(%s);' % (self.name,
141 ', '.join(arg.name for arg in self.args)))
142 res.append(' return simcall<%s%s>(SIMCALL_%s%s);' % (
144 "".join([ ", " + arg.rettype() for i, arg in enumerate(self.args) ]),
146 "".join([ ", " + arg.name for i, arg in enumerate(self.args) ])
149 return '\n'.join(res)
151 def handler_prototype(self):
152 if self.need_handler:
153 return "XBT_PRIVATE %s simcall_HANDLER_%s(smx_simcall_t simcall%s);" % (self.res.rettype() if self.call_kind == 'Func' else 'void',
155 ''.join(', %s %s' % (arg.rettype(), arg.name)
156 for i, arg in enumerate(self.args)))
164 simcalls_guarded = {}
165 for line in open(fn).read().split('\n'):
166 if line.startswith('##'):
168 simcalls_guarded[re.search(r'## *(.*)', line).group(1)] = resdi
169 if line.startswith('#') or not line:
172 r'^(Proc|Func|Blck)\s*([H-])\s*(\S*)\s*(\S*)\s*\(*([^\(\)]*)\)\s*;?$', line)
174 ans, handler, ret, name, args = match.groups()
175 assert (ans == 'Proc' or ans == 'Func' or ans == 'Blck'), "Invalid call type: '%s'. Faulty line:\n%s\n" % (
177 assert (handler == 'H' or handler == '-'), "Invalid need_handler indication: '%s'. Faulty line:\n%s\n" % (
180 if not re.match("^\s*$", args):
181 for arg in re.split(",", args):
183 match = re.match("^(.*?)\s*?(\S+)$", arg)
184 t, n = match.groups()
187 sargs.append(Arg(n, t))
188 sim = Simcall(name, handler == 'H',
189 Arg('result', ret), sargs, ans)
194 return simcalls, simcalls_guarded
200 '/**********************************************************************/\n')
202 '/* File generated by src/simix/simcalls.py from src/simix/simcalls.in */\n')
206 '/* DO NOT EVER CHANGE THIS FILE */\n')
210 '/* change simcalls specification in src/simix/simcalls.in */\n')
212 '/**********************************************************************/\n\n')
215 ' * Note that the name comes from http://en.wikipedia.org/wiki/Popping\n')
217 ' * Indeed, the control flow is doing a strange dance in there.\n')
220 ' * That\'s not about http://en.wikipedia.org/wiki/Poop, despite the odor :)\n')
225 def handle(fd, func, simcalls, guarded_simcalls):
226 def nonempty(e): return e != ''
228 '\n'.join(filter(nonempty, (func(simcall) for simcall in simcalls))))
230 for guard, list in guarded_simcalls.items():
231 fd.write('\n#if %s\n' % (guard))
232 fd.write('\n'.join(func(simcall) for simcall in list))
233 fd.write('\n#endif\n')
235 if __name__ == '__main__':
237 simcalls, simcalls_dict = parse('simcalls.in')
240 ok &= all(map(Simcall.check, simcalls))
241 for k, v in simcalls_dict.items():
242 ok &= all(map(Simcall.check, v))
243 # FIXME: we should not hide it
245 # print ("Some checks fail!")
249 # smx_popping_accessors.c
251 fd = header('popping_accessors.h')
252 fd.write('#include "src/simix/popping_private.h"');
253 handle(fd, Simcall.accessors, simcalls, simcalls_dict)
255 "\n\n/* The prototype of all simcall handlers, automatically generated for you */\n\n")
256 handle(fd, Simcall.handler_prototype, simcalls, simcalls_dict)
262 fd = header("popping_enum.h")
264 fd.write(' * @brief All possible simcalls.\n')
266 fd.write('typedef enum {\n')
267 fd.write(' SIMCALL_NONE,\n')
269 handle(fd, Simcall.enum, simcalls, simcalls_dict)
271 fd.write(' NUM_SIMCALLS\n')
272 fd.write('} e_smx_simcall_t;\n')
276 # smx_popping_generated.cpp
279 fd = header("popping_generated.cpp")
281 fd.write('#include <xbt/base.h>\n')
282 fd.write('#include "smx_private.h"\n')
283 fd.write('#if HAVE_MC\n')
284 fd.write('#include "src/mc/mc_forward.hpp"\n')
287 fd.write('XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simix_popping);\n\n')
290 '/** @brief Simcalls\' names (generated from src/simix/simcalls.in) */\n')
291 fd.write('const char* simcall_names[] = {\n')
293 fd.write(' "SIMCALL_NONE",')
294 handle(fd, Simcall.string, simcalls, simcalls_dict)
298 fd.write('/** @private\n')
300 ' * @brief (in kernel mode) unpack the simcall and activate the handler\n')
302 fd.write(' * This function is generated from src/simix/simcalls.in\n')
305 'void SIMIX_simcall_handle(smx_simcall_t simcall, int value) {\n')
307 ' XBT_DEBUG("Handling simcall %p: %s", simcall, SIMIX_simcall_name(simcall->call));\n')
308 fd.write(' SIMCALL_SET_MC_VALUE(simcall, value);\n')
310 ' if (simcall->issuer->context->iwannadie && simcall->call != SIMCALL_PROCESS_CLEANUP)\n')
311 fd.write(' return;\n')
312 fd.write(' switch (simcall->call) {\n')
314 handle(fd, Simcall.case, simcalls, simcalls_dict)
316 fd.write(' case NUM_SIMCALLS:\n')
317 fd.write(' break;\n')
318 fd.write(' case SIMCALL_NONE:\n')
320 ' THROWF(arg_error,0,"Asked to do the noop syscall on %s@%s",\n')
321 fd.write(' SIMIX_process_get_name(simcall->issuer),\n')
323 ' sg_host_get_name(SIMIX_process_get_host(simcall->issuer))\n')
325 fd.write(' break;\n')
333 # smx_popping_bodies.cpp
335 fd = header('popping_bodies.cpp')
336 fd.write('#include "smx_private.h"\n')
337 fd.write('#include "src/mc/mc_forward.hpp"\n')
338 fd.write('#include "xbt/ex.h"\n')
339 fd.write('#include <simgrid/simix.hpp>\n')
340 fd.write("/** @cond */ // Please Doxygen, don't look at this\n")
342 template<class R, class... T>
343 inline static R simcall(e_smx_simcall_t call, T const&... t)
345 smx_process_t self = SIMIX_process_self();
346 simgrid::simix::marshal(&self->simcall, call, t...);
347 if (self != simix_global->maestro_process) {
348 XBT_DEBUG("Yield process '%s' on simcall %s (%d)", self->name.c_str(),
349 SIMIX_simcall_name(self->simcall.call), (int)self->simcall.call);
350 SIMIX_process_yield(self);
352 SIMIX_simcall_handle(&self->simcall, 0);
354 return simgrid::simix::unmarshal<R>(self->simcall.result);
357 handle(fd, Simcall.body, simcalls, simcalls_dict)
358 fd.write("/** @endcond */\n");