- simcalls_BODY = None
- simcalls_PRE = None
- def __init__(self, name, res, args, has_answer=True):
- self.name = name
- self.res = res
- self.args = args
- self.has_answer = has_answer
-
- def check(self):
- # smx_user.c simcall_BODY_
- # smx_*.c void SIMIX_pre_host_on(smx_simcall_t simcall, smx_host_t h)
- self.check_body()
- self.check_pre()
-
- def check_body(self):
- if self.simcalls_BODY is None:
- f = open('smx_user.c')
- self.simcalls_BODY = set(re.findall('simcall_BODY_(.*?)\(', f.read()))
- f.close()
- if self.name not in self.simcalls_BODY:
- print '# ERROR: No function calling simcall_BODY_%s'%self.name
- print '# Add something like this to smx_user.c:'
- print '''%s simcall_%s(%s)
-{
- return simcall_BODY_%s(%s);
-}\n'''%(self.res.ret()
- ,self.name
- ,', '.join('%s %s'%(arg.ret(), arg.name)
- for arg in self.args)
- ,self.name
- ,', '.join(arg.name for arg in self.args))
- return False
- return True
-
- def check_pre(self):
- if self.simcalls_PRE is None:
- self.simcalls_PRE = set()
- for fn in glob.glob('smx_*') + glob.glob('../mc/*'):
- f = open(fn)
- self.simcalls_PRE |= set(re.findall('SIMIX_pre_(.*?)\(', f.read()))
- f.close()
- if self.name not in self.simcalls_PRE:
- print '# ERROR: No function called SIMIX_pre_%s'%self.name
- print '# Add something like this to smx_.*.c:'
- print '''%s SIMIX_pre_%s(smx_simcall_t simcall%s)
-{
- // Your code handling the simcall
-}\n'''%(self.res.ret()
- ,self.name
- ,''.join(', %s %s'%(arg.ret(), arg.name)
- for arg in self.args))
- return False
- return True
-
- def enum(self):
- return ' SIMCALL_%s,'%(self.name.upper())
-
- def string(self):
- return '[SIMCALL_%s] = "SIMCALL_%s",'%(self.name.upper(), self.name.upper())
-
- def accessors(self):
- res = []
- for i in range(len(self.args)):
- res.append(self.arg_getter(i))
- res.append(self.arg_setter(i))
- if self.res.type != 'void':
- res.append('static inline %s simcall_%s__get__result(smx_simcall_t simcall){'%(self.res.ret(), self.name))
- res.append(' return %s simcall->result.%s;'%(self.res.cast(), self.res.field()))
- res.append('}')
- res.append('static inline void simcall_%s__set__result(smx_simcall_t simcall, %s result){'%(self.name, self.res.type,))
- res.append(' simcall->result.%s = result;'%(self.res.field()))
+ simcalls_body = None
+ simcalls_pre = None
+
+ def __init__(self, name, handler, res, args, call_kind):
+ self.name = name
+ self.res = res
+ self.args = args
+ self.need_handler = handler
+ self.call_kind = call_kind
+
+ def check(self):
+ # libsmx.c simcall_BODY_
+ if self.simcalls_body is None:
+ f = open('libsmx.cpp')
+ self.simcalls_body = set(re.findall(r'simcall_BODY_(.*?)\(', f.read()))
+ f.close()
+ if self.name not in self.simcalls_body:
+ print ('# ERROR: No function calling simcall_BODY_%s' % self.name)
+ print ('# Add something like this to libsmx.c:')
+ print ('%s simcall_%s(%s)' % (self.res.rettype(), self.name, ', '.
+ join('%s %s' % (arg.rettype(), arg.name) for arg in self.args)))
+ print ('{')
+ print (' return simcall_BODY_%s(%s);' % (self.name, "..."))
+ print ('}')
+ return False
+
+ # smx_*.c void simcall_HANDLER_host_on(smx_simcall_t simcall,
+ # smx_host_t h)
+ if self.simcalls_pre is None:
+ self.simcalls_pre = set()
+ for fn in glob.glob('smx_*') + glob.glob('ActorImpl*') + glob.glob('../mc/*cpp'):
+ f = open(fn)
+ self.simcalls_pre |= set(re.findall(r'simcall_HANDLER_(.*?)\(', f.read()))
+ f.close()
+ if self.need_handler:
+ if self.name not in self.simcalls_pre:
+ print ('# ERROR: No function called simcall_HANDLER_%s' % self.name)
+ print ('# Add something like this to the relevant C file (like smx_io.c if it\'s an IO call):')
+ print ('%s simcall_HANDLER_%s(smx_simcall_t simcall%s)' % (self.res.rettype(), self.name, ''.
+ join(', %s %s' % (arg.rettype(), arg.name)for arg in self.args)))
+ print ('{')
+ print (' // Your code handling the simcall')
+ print ('}')
+ return False
+ else:
+ if self.name in self.simcalls_pre:
+ print ('# ERROR: You have a function called simcall_HANDLER_%s, but that simcall is not using any handler' % self.name)
+ print ('# Either change your simcall definition, or kill that function')
+ return False
+ return True
+
+ def enum(self):
+ return ' SIMCALL_%s,' % (self.name.upper())
+
+ def string(self):
+ return ' "SIMCALL_%s",' % self.name.upper()
+
+ def accessors(self):
+ res = []
+ res.append('')
+ regex = re.compile(r"^boost::intrusive_ptr<(.*?)>(.*)$") # to compute the raw type
+ # Arguments getter/setters
+ for i in range(len(self.args)):
+ arg = self.args[i]
+ rawtype = regex.sub(r'\1*\2', arg.rettype())
+ res.append('static inline %s simcall_%s__get__%s(smx_simcall_t simcall)' % (
+ arg.rettype(), self.name, arg.name))
+ res.append('{')
+ res.append(' return simgrid::simix::unmarshal<%s>(simcall->args[%i]);' % (arg.rettype(), i))
+ res.append('}')
+ res.append('static inline %s simcall_%s__getraw__%s(smx_simcall_t simcall)' % (
+ rawtype, self.name, arg.name))
+ res.append('{')
+ res.append(' return simgrid::simix::unmarshal_raw<%s>(simcall->args[%i]);' % (rawtype, i))
+ res.append('}')
+ res.append('static inline void simcall_%s__set__%s(smx_simcall_t simcall, %s arg)' % (
+ self.name, arg.name, arg.rettype()))
+ res.append('{')
+ res.append(' simgrid::simix::marshal<%s>(simcall->args[%i], arg);' % (arg.rettype(), i))
+ res.append('}')
+
+ # Return value getter/setters
+ if self.res.type != 'void':
+ rawtype = regex.sub(r'\1*\2', self.res.rettype())
+ res.append(
+ 'static inline %s simcall_%s__get__result(smx_simcall_t simcall)' % (self.res.rettype(), self.name))
+ res.append('{')
+ res.append(' return simgrid::simix::unmarshal<%s>(simcall->result);' % self.res.rettype())
+ res.append('}')
+ res.append('static inline %s simcall_%s__getraw__result(smx_simcall_t simcall)' % (rawtype, self.name))
+ res.append('{')
+ res.append(' return simgrid::simix::unmarshal_raw<%s>(simcall->result);' % rawtype)
+ res.append('}')
+ res.append(
+ 'static inline void simcall_%s__set__result(smx_simcall_t simcall, %s result)' % (self.name, self.res.rettype()))
+ res.append('{')
+ res.append(' simgrid::simix::marshal<%s>(simcall->result, result);' % (self.res.rettype()))
+ res.append('}')
+ return '\n'.join(res)
+
+ def case(self):
+ res = []
+ args = [ "simgrid::simix::unmarshal<%s>(simcall->args[%d])" % (arg.rettype(), i)
+ for i, arg in enumerate(self.args) ]
+ res.append('case SIMCALL_%s:' % (self.name.upper()))
+ if self.need_handler:
+ call = "simcall_HANDLER_%s(simcall%s%s)" % (self.name,
+ ", " if len(args) > 0 else "",
+ ', '.join(args))
+ else:
+ call = "SIMIX_%s(%s)" % (self.name, ', '.join(args))
+ if self.call_kind == 'Func':
+ res.append(" simgrid::simix::marshal<%s>(simcall->result, %s);" % (self.res.rettype(), call))
+ else:
+ res.append(" " + call + ";");
+ if self.call_kind != 'Blck':
+ res.append(' SIMIX_simcall_answer(simcall);')
+ res.append(' break;')
+ res.append('')
+ return '\n'.join(res)
+
+ def body(self):
+ res = ['']
+ res.append(
+ 'inline static %s simcall_BODY_%s(%s)' % (self.res.rettype(),
+ self.name,
+ ', '.join('%s %s' % (arg.rettype(), arg.name) for arg in self.args)))
+ res.append('{')
+ res.append(' if (0) /* Go to that function to follow the code flow through the simcall barrier */')
+ if self.need_handler:
+ res.append(' simcall_HANDLER_%s(%s);' % (self.name,
+ ', '.join(["&SIMIX_process_self()->simcall"] + [arg.name for arg in self.args])))
+ else:
+ res.append(' SIMIX_%s(%s);' % (self.name,
+ ', '.join(arg.name for arg in self.args)))
+ res.append(' return simcall<%s%s>(SIMCALL_%s%s);' % (
+ self.res.rettype(),
+ "".join([ ", " + arg.rettype() for i, arg in enumerate(self.args) ]),
+ self.name.upper(),
+ "".join([ ", " + arg.name for i, arg in enumerate(self.args) ])
+ ));