sq_slave_vm.cpp 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691
  1. /*
  2. ** Rings: Multiple Lua States
  3. ** $Id: rings.c,v 1.24 2008/06/30 17:52:31 carregal Exp $
  4. ** See Copyright Notice in license.html
  5. */
  6. #include <string.h>
  7. #include <stdio.h>
  8. #include "squirrel.h"
  9. #include <sqstdblob.h>
  10. #include <sqstdsystem.h>
  11. #include <sqstdio.h>
  12. #include <sqstdmath.h>
  13. #include <sqstdstring.h>
  14. #ifdef SLAVE_VM_WITH_OS_THREADS
  15. #include "mongoose.h"
  16. #endif // SLAVE_VM_WITH_OS_THREADS
  17. /*
  18. ** Copies values from State src to State dst.
  19. */
  20. static SQRESULT copy_values_between_vms (HSQUIRRELVM dst, HSQUIRRELVM src, int argc, int argIdx)
  21. {
  22. SQRESULT _rc_;
  23. sq_reservestack(dst, argc + 20);
  24. argc += argIdx; //we will work with argc args starting at argIdx
  25. for (; argIdx < argc; argIdx++)
  26. {
  27. switch (sq_gettype(src, argIdx))
  28. {
  29. case OT_INTEGER:
  30. SQ_GET_INTEGER(src, argIdx, vint);
  31. sq_pushinteger(dst, vint);
  32. break;
  33. case OT_FLOAT:
  34. SQ_GET_FLOAT(src, argIdx, vfloat);
  35. sq_pushfloat (dst, vfloat);
  36. break;
  37. case OT_BOOL:
  38. SQ_GET_BOOL(src, argIdx, vbool);
  39. sq_pushbool (dst, vbool);
  40. break;
  41. case OT_STRING:
  42. {
  43. SQ_GET_STRING(src, argIdx, vstr)
  44. sq_pushstring (dst, vstr, vstr_size);
  45. }
  46. break;
  47. case OT_ARRAY:
  48. {
  49. SQInteger size = sq_getsize(src, argIdx);
  50. sq_newarray(dst, size);
  51. for(SQInteger i=0; i<size; ++i)
  52. {
  53. sq_pushinteger(src, i);
  54. sq_get(src, -2);
  55. sq_pushinteger(dst, i);
  56. if(copy_values_between_vms(dst, src, 1, sq_gettop(src)) != SQ_OK) return SQ_ERROR;
  57. sq_poptop(src);
  58. sq_set(dst, -3);
  59. }
  60. }
  61. break;
  62. case OT_TABLE:
  63. {
  64. sq_newtable(dst);
  65. sq_pushnull(src);
  66. while(sq_next(src, -2) == SQ_OK)
  67. {
  68. SQInteger src_top = sq_gettop(src);
  69. if(copy_values_between_vms(dst, src, 1, src_top-1) != SQ_OK
  70. || copy_values_between_vms(dst, src, 1, src_top) != SQ_OK) return SQ_ERROR;
  71. sq_newslot(dst, -3, SQFalse);
  72. sq_pop(src, 2);
  73. }
  74. sq_pop(src,1);
  75. }
  76. break;
  77. case OT_USERPOINTER:
  78. {
  79. SQUserPointer ptr;
  80. sq_getuserpointer(src, argIdx, &ptr);
  81. sq_pushuserpointer(dst, ptr);
  82. }
  83. break;
  84. case OT_NULL:
  85. sq_pushnull(dst);
  86. break;
  87. default:
  88. return SQ_ERROR;
  89. }
  90. }
  91. return SQ_OK;
  92. }
  93. struct SlaveVM_st {
  94. HSQUIRRELVM v;
  95. HSQUIRRELVM master_vm;
  96. #ifdef SLAVE_VM_WITH_OS_THREADS
  97. mg_thread_mutex_t mutex;
  98. mg_thread_cond_t cond_var;
  99. mg_thread_t thread_id;
  100. int slave_state, master_state;
  101. bool isRunningAsThread;
  102. #endif // SLAVE_VM_WITH_OS_THREADS
  103. };
  104. static const SQChar sq_slave_vm_TAG[] = _SC("SlaveVM");
  105. static SQRESULT get_slave_vm_instance(HSQUIRRELVM v, SQInteger idx, SlaveVM_st **svm_st)
  106. {
  107. if(sq_getinstanceup(v, idx, (SQUserPointer*)svm_st, (void*)sq_slave_vm_TAG) != SQ_OK) return SQ_ERROR;
  108. if(!*svm_st) return sq_throwerror(v, _SC("slave vm already closed"));
  109. return SQ_OK;
  110. }
  111. #define GET_sq_slave_vm_INSTANCE(v, idx) \
  112. SlaveVM_st *self=NULL; HSQUIRRELVM svm; \
  113. if(get_slave_vm_instance(v, idx, &self) != SQ_OK) return SQ_ERROR; \
  114. svm = self->v;
  115. static SQRESULT sq_slave_vm__tostring (HSQUIRRELVM v)
  116. {
  117. GET_sq_slave_vm_INSTANCE(v, 1);
  118. sq_pushfstring (v, _SC("Squirrel vm (%p)"), svm);
  119. return 1;
  120. }
  121. #ifdef SLAVE_VM_WITH_OS_THREADS
  122. static int releaseSlaveThread(SlaveVM_st *self)
  123. {
  124. if(self->isRunningAsThread)
  125. {
  126. mg_thread_cond_destroy(&self->cond_var);
  127. mg_thread_mutex_destroy(&self->mutex);
  128. }
  129. return 0;
  130. }
  131. #endif // SLAVE_VM_WITH_OS_THREADS
  132. static SQRESULT sq_slave_vm_release_hook(SQUserPointer p, SQInteger size, void */*ep*/)
  133. {
  134. SlaveVM_st *self = (SlaveVM_st*)p;
  135. if(self && self->v)
  136. {
  137. #ifdef SLAVE_VM_WITH_OS_THREADS
  138. releaseSlaveThread(self);
  139. #endif // SLAVE_VM_WITH_OS_THREADS
  140. sq_close(self->v);
  141. self->v = NULL;
  142. sq_free(self, sizeof(*self));
  143. }
  144. return 0;
  145. }
  146. #ifdef SLAVE_VM_WITH_OS_THREADS
  147. #define SQ_getforeignptr_slave(v) \
  148. SlaveVM_st *self = (SlaveVM_st*)sq_getforeignptr(v);\
  149. if(!self) return sq_throwerror(v, _SC("No slavevm foreignptr found"));
  150. static SQRESULT sq_vm_thread_lock(HSQUIRRELVM v)
  151. {
  152. SQ_getforeignptr_slave(v);
  153. sq_pushinteger(v, mg_thread_mutex_lock(&self->mutex));
  154. return 1;
  155. }
  156. /*
  157. static SQRESULT sq_vm_thread_unlock(HSQUIRRELVM v)
  158. {
  159. SQ_getforeignptr_slave(v);
  160. sq_pushinteger(v, mg_thread_mutex_unlock(&self->mutex));
  161. return 1;
  162. }
  163. */
  164. static SQRESULT sq_vm_thread_cond_wait(HSQUIRRELVM v)
  165. {
  166. SQ_getforeignptr_slave(v);
  167. sq_pushinteger(v, mg_thread_cond_wait(&self->cond_var, &self->mutex));
  168. return 1;
  169. }
  170. static SQRESULT sq_vm_thread_master_state(HSQUIRRELVM v)
  171. {
  172. SQ_getforeignptr_slave(v);
  173. sq_pushinteger(v, self->master_state);
  174. return 1;
  175. }
  176. static SQRESULT sq_vm_thread_slave_state(HSQUIRRELVM v)
  177. {
  178. SQ_FUNC_VARS(v);
  179. SQ_getforeignptr_slave(v);
  180. SQ_OPT_INTEGER(v, 2, state, 0);
  181. sq_pushinteger(v, self->slave_state);
  182. if(sq_gettop(v) > 1) self->slave_state = state;
  183. return 1;
  184. }
  185. #endif // SLAVE_VM_WITH_OS_THREADS
  186. /*
  187. ** Creates a new SQuirrel vm.
  188. */
  189. static SQRESULT sq_slave_vm_constructor (HSQUIRRELVM v)
  190. {
  191. SQ_FUNC_VARS(v);
  192. SQ_OPT_INTEGER(v, 2, stack_size, 1024);
  193. SQ_OPT_BOOL(v, 3, with_libraries, false);
  194. SQ_OPT_BOOL(v, 4, run_as_thread, false);
  195. HSQUIRRELVM self = sq_open(stack_size);
  196. /* Initialize environment */
  197. sq_setprintfunc(self,sq_getprintfunc(v),sq_geterrorfunc(v));
  198. if(with_libraries)
  199. {
  200. /* load base libraries */
  201. sq_pushroottable(self);
  202. sqstd_register_bloblib(self);
  203. sqstd_register_iolib(self);
  204. sqstd_register_systemlib(self);
  205. sqstd_register_mathlib(self);
  206. sqstd_register_stringlib(self);
  207. sq_poptop(self); //remove root table
  208. }
  209. SlaveVM_st *svm_st = (SlaveVM_st *)sq_malloc(sizeof(SlaveVM_st));
  210. memset(svm_st, 0, sizeof(*svm_st));
  211. #ifdef SLAVE_VM_WITH_OS_THREADS
  212. sq_setforeignptr(self, svm_st);
  213. if(run_as_thread)
  214. {
  215. svm_st->isRunningAsThread = true;
  216. svm_st->master_vm = v;
  217. mg_thread_mutex_init(&svm_st->mutex, NULL);
  218. mg_thread_cond_init(&svm_st->cond_var, NULL);
  219. sq_pushroottable(self);
  220. //sq_insertfunc(self, _SC("slavevm_thread_lock"), sq_vm_thread_lock, 1, _SC("."), SQFalse);
  221. //sq_insertfunc(self, _SC("slavevm_unlock"), sq_vm_thread_unlock, 1, _SC("."), SQFalse);
  222. sq_insertfunc(self, _SC("slavevm_thread_cond_wait"), sq_vm_thread_cond_wait, 1, _SC("."), SQFalse);
  223. sq_insertfunc(self, _SC("slavevm_thread_master_state"), sq_vm_thread_master_state, 1, _SC("."), SQFalse);
  224. sq_insertfunc(self, _SC("slavevm_thread_slave_state"), sq_vm_thread_slave_state, -1, _SC(".i"), SQFalse);
  225. sq_poptop(self); //remove root table
  226. }
  227. #endif // SLAVE_VM_WITH_OS_THREADS
  228. svm_st->v = self;
  229. svm_st->master_vm = v;
  230. sq_setinstanceup(v, 1, svm_st);
  231. sq_setreleasehook(v, 1, sq_slave_vm_release_hook);
  232. return 1;
  233. }
  234. #ifdef SLAVE_VM_WITH_OS_THREADS
  235. static SQRESULT checkTryLock(HSQUIRRELVM v, SlaveVM_st *self)
  236. {
  237. if(self->isRunningAsThread && mg_thread_mutex_trylock(&self->mutex))
  238. {
  239. return sq_throwerror(v, _SC("Could not aquire slave lock"));
  240. }
  241. return SQ_OK;
  242. }
  243. #endif // SLAVE_VM_WITH_OS_THREADS
  244. static SQRESULT sq_slave_vm_close(HSQUIRRELVM v)
  245. {
  246. GET_sq_slave_vm_INSTANCE(v, 1);
  247. #ifdef SLAVE_VM_WITH_OS_THREADS
  248. if(checkTryLock(v, self)) return SQ_ERROR;
  249. releaseSlaveThread(self);
  250. #endif // SLAVE_VM_WITH_OS_THREADS
  251. sq_close(svm);
  252. sq_setinstanceup(v, 1, 0); //next calls will fail with "vm is closed"
  253. return 0;
  254. }
  255. static SQRESULT sq_slave_vm_call(HSQUIRRELVM v)
  256. {
  257. SQ_FUNC_VARS_NO_TOP(v);
  258. GET_sq_slave_vm_INSTANCE(v, 1);
  259. SQ_GET_BOOL(v, 2, has_ret_val);
  260. SQ_GET_STRING(v, 3, func_name);
  261. #ifdef SLAVE_VM_WITH_OS_THREADS
  262. if(checkTryLock(v, self)) return SQ_ERROR;
  263. #endif // SLAVE_VM_WITH_OS_THREADS
  264. SQInteger top = sq_gettop(svm);
  265. SQRESULT result = SQ_ERROR;
  266. sq_pushroottable(svm);
  267. sq_pushstring(svm, func_name, func_name_size);
  268. if(sq_get(svm, -2) == SQ_OK)
  269. {
  270. switch(sq_gettype(svm, -1))
  271. {
  272. case OT_CLOSURE:
  273. case OT_NATIVECLOSURE:
  274. case OT_CLASS:
  275. {
  276. sq_pushroottable(svm);
  277. int argc = sq_gettop(v) - 3;
  278. if(argc && copy_values_between_vms(svm, v, argc, 4) != SQ_OK)
  279. {
  280. result = sq_throwerror(v, sq_getlasterror_str(svm));
  281. goto cleanup;
  282. }
  283. if(sq_call(svm, argc+1, has_ret_val, SQFalse) != SQ_OK)
  284. {
  285. result = sq_throwerror(v, sq_getlasterror_str(svm));
  286. goto cleanup;
  287. }
  288. if(has_ret_val)
  289. {
  290. if(copy_values_between_vms(v, svm, 1, sq_gettop(svm)) == SQ_OK) result = 1;
  291. }
  292. else result = SQ_OK;
  293. }
  294. }
  295. }
  296. else
  297. {
  298. result = sq_throwerror(v, sq_getlasterror_str(svm));
  299. }
  300. cleanup:
  301. sq_settop(svm, top);
  302. #ifdef SLAVE_VM_WITH_OS_THREADS
  303. if(self->isRunningAsThread) mg_thread_mutex_unlock(&self->mutex);
  304. #endif // SLAVE_VM_WITH_OS_THREADS
  305. return result;
  306. }
  307. static SQRESULT sq_slave_vm_set(HSQUIRRELVM v)
  308. {
  309. GET_sq_slave_vm_INSTANCE(v, 1);
  310. #ifdef SLAVE_VM_WITH_OS_THREADS
  311. if(checkTryLock(v, self)) return SQ_ERROR;
  312. #endif // SLAVE_VM_WITH_OS_THREADS
  313. SQInteger top = sq_gettop(svm);
  314. SQRESULT result;
  315. sq_pushroottable(svm);
  316. if(copy_values_between_vms(svm, v, 1, 2) == SQ_OK
  317. && copy_values_between_vms(svm, v, 1, 3) == SQ_OK)
  318. {
  319. result = sq_newslot(svm, -3, SQFalse);
  320. if(result == SQ_ERROR) sq_throwerror(v, sq_getlasterror_str(svm));
  321. }
  322. else result = SQ_ERROR;
  323. sq_settop(svm, top);
  324. #ifdef SLAVE_VM_WITH_OS_THREADS
  325. if(self->isRunningAsThread) mg_thread_mutex_unlock(&self->mutex);
  326. #endif // SLAVE_VM_WITH_OS_THREADS
  327. return result;
  328. }
  329. static SQRESULT sq_slave_vm_get(HSQUIRRELVM v)
  330. {
  331. GET_sq_slave_vm_INSTANCE(v, 1);
  332. #ifdef SLAVE_VM_WITH_OS_THREADS
  333. if(checkTryLock(v, self)) return SQ_ERROR;
  334. #endif // SLAVE_VM_WITH_OS_THREADS
  335. SQInteger top = sq_gettop(svm);
  336. SQRESULT result = SQ_ERROR;
  337. sq_pushroottable(svm);
  338. if(copy_values_between_vms(svm, v, 1, 2) == SQ_OK)
  339. {
  340. if(sq_get(svm, -2) == SQ_OK
  341. && copy_values_between_vms(v, svm, 1, sq_gettop(svm)) == SQ_OK)
  342. {
  343. result = 1;
  344. }
  345. else
  346. {
  347. if(sq_gettop(v) == 3) result = 1; //we have a default value
  348. else sq_throwerror(v, sq_getlasterror_str(svm));
  349. }
  350. }
  351. sq_settop(svm, top);
  352. #ifdef SLAVE_VM_WITH_OS_THREADS
  353. if(self->isRunningAsThread) mg_thread_mutex_unlock(&self->mutex);
  354. #endif // SLAVE_VM_WITH_OS_THREADS
  355. return result;
  356. }
  357. static SQRESULT sq_slave_vm_dofile(HSQUIRRELVM v)
  358. {
  359. SQ_FUNC_VARS(v);
  360. GET_sq_slave_vm_INSTANCE(v, 1);
  361. SQ_GET_STRING(v, 2, file_name);
  362. SQ_OPT_BOOL(v, 3, retval, false);
  363. SQ_OPT_BOOL(v, 4, printerror, false);
  364. SQ_OPT_BOOL(v, 5, show_warnings, false);
  365. #ifdef SLAVE_VM_WITH_OS_THREADS
  366. if(checkTryLock(v, self)) return SQ_ERROR;
  367. #endif // SLAVE_VM_WITH_OS_THREADS
  368. SQInteger top = sq_gettop(svm);
  369. SQRESULT result = SQ_ERROR;
  370. sq_pushroottable(svm); //important always push the root table, because sqstd_dofile will try to do a sq_push(v, -2)
  371. if(sqstd_dofile(svm, file_name, retval, printerror, show_warnings) >= 0)
  372. {
  373. if(retval)
  374. {
  375. if(copy_values_between_vms(v, svm, 1, sq_gettop(svm)) == SQ_OK) result = 1;
  376. }
  377. else result = SQ_OK;
  378. }
  379. else sq_throwerror(v, sq_getlasterror_str(svm));
  380. sq_settop(svm, top);
  381. #ifdef SLAVE_VM_WITH_OS_THREADS
  382. if(self->isRunningAsThread) mg_thread_mutex_unlock(&self->mutex);
  383. #endif // SLAVE_VM_WITH_OS_THREADS
  384. return result;
  385. }
  386. static SQRESULT sq_slave_vm_loadfile(HSQUIRRELVM v)
  387. {
  388. SQ_FUNC_VARS(v);
  389. GET_sq_slave_vm_INSTANCE(v, 1);
  390. SQ_GET_STRING(v, 2, func_name);
  391. SQ_GET_STRING(v, 3, file_name);
  392. SQ_OPT_BOOL(v, 4, printerror, false);
  393. SQ_OPT_BOOL(v, 5, show_warnings, false);
  394. #ifdef SLAVE_VM_WITH_OS_THREADS
  395. if(checkTryLock(v, self)) return SQ_ERROR;
  396. #endif // SLAVE_VM_WITH_OS_THREADS
  397. SQInteger top = sq_gettop(svm);
  398. SQRESULT result = SQ_ERROR;
  399. sq_pushroottable(svm);
  400. sq_pushstring(svm, func_name, func_name_size);
  401. if(sqstd_loadfile(svm, file_name, printerror, show_warnings) >= 0)
  402. {
  403. result = sq_newslot(svm, -3, SQFalse);
  404. }
  405. sq_settop(svm, top);
  406. #ifdef SLAVE_VM_WITH_OS_THREADS
  407. if(self->isRunningAsThread) mg_thread_mutex_unlock(&self->mutex);
  408. #endif // SLAVE_VM_WITH_OS_THREADS
  409. return result;
  410. }
  411. static SQRESULT sq_slave_vm_compilestring(HSQUIRRELVM v)
  412. {
  413. SQ_FUNC_VARS(v);
  414. GET_sq_slave_vm_INSTANCE(v, 1);
  415. SQ_GET_STRING(v, 2, func_name);
  416. SQ_GET_STRING(v, 3, str_script);
  417. SQ_OPT_BOOL(v, 4, printerror, false);
  418. SQ_OPT_BOOL(v, 5, show_warnings, false);
  419. SQ_OPT_INTEGER(v, 6, max_nested_includes, 0);
  420. #ifdef SLAVE_VM_WITH_OS_THREADS
  421. if(checkTryLock(v, self)) return SQ_ERROR;
  422. #endif // SLAVE_VM_WITH_OS_THREADS
  423. SQInteger top = sq_gettop(svm);
  424. SQRESULT result = SQ_ERROR;
  425. sq_pushroottable(svm);
  426. sq_pushstring(svm, func_name, func_name_size);
  427. if(sq_compilebuffer(svm, str_script, str_script_size, func_name, printerror, show_warnings, max_nested_includes) >= 0)
  428. {
  429. result = sq_newslot(svm, -3, SQFalse);
  430. }
  431. else
  432. {
  433. const SQChar *last_error = sq_getlasterror_str(svm);
  434. if(last_error) {
  435. SQInteger line, column;
  436. sq_getlasterror_line_col(svm, &line, &column);
  437. result = sq_throwerror(v, _SC("compilestring %s %d:%d: %s"), func_name, line, column, last_error);
  438. }
  439. }
  440. sq_settop(svm, top);
  441. #ifdef SLAVE_VM_WITH_OS_THREADS
  442. if(self->isRunningAsThread) mg_thread_mutex_unlock(&self->mutex);
  443. #endif // SLAVE_VM_WITH_OS_THREADS
  444. return result;
  445. }
  446. extern SQInteger _g_io_dostring(HSQUIRRELVM v);
  447. static SQRESULT sq_slave_vm_dostring(HSQUIRRELVM v)
  448. {
  449. SQ_FUNC_VARS(v);
  450. GET_sq_slave_vm_INSTANCE(v, 1);
  451. SQ_GET_STRING(v, 2, str);
  452. SQ_OPT_BOOL(v, 3, retval, false);
  453. #ifdef SLAVE_VM_WITH_OS_THREADS
  454. if(checkTryLock(v, self)) return SQ_ERROR;
  455. #endif // SLAVE_VM_WITH_OS_THREADS
  456. SQInteger top = sq_gettop(svm);
  457. sq_pushroottable(svm); //important always push the root table, because sqstd_dofile will try to do a sq_push(v, -2)
  458. sq_pushstring(svm, str, str_size);
  459. sq_pushbool(svm, retval);
  460. SQRESULT result = _g_io_dostring(svm);
  461. if(result == SQ_ERROR) sq_settop(svm, top);
  462. #ifdef SLAVE_VM_WITH_OS_THREADS
  463. if(self->isRunningAsThread) mg_thread_mutex_unlock(&self->mutex);
  464. #endif // SLAVE_VM_WITH_OS_THREADS
  465. return result;
  466. }
  467. static SQRESULT sq_slave_vm_is_thread_idle(HSQUIRRELVM v)
  468. {
  469. #ifdef SLAVE_VM_WITH_OS_THREADS
  470. GET_sq_slave_vm_INSTANCE(v, 1);
  471. int rc = mg_thread_mutex_trylock(&self->mutex) == 0;
  472. if(rc) mg_thread_mutex_unlock(&self->mutex);
  473. sq_pushbool(v, rc);
  474. #else
  475. sq_pushbool(v, SQTrue);
  476. #endif
  477. return 1;
  478. }
  479. static SQRESULT sq_slave_vm_is_runing_as_thread(HSQUIRRELVM v)
  480. {
  481. #ifdef SLAVE_VM_WITH_OS_THREADS
  482. GET_sq_slave_vm_INSTANCE(v, 1);
  483. sq_pushbool(v, self->isRunningAsThread);
  484. #else
  485. sq_pushbool(v, SQFalse);
  486. #endif
  487. return 1;
  488. }
  489. #ifdef SLAVE_VM_WITH_OS_THREADS
  490. #define check_run_as_thread(v) \
  491. if(!self->isRunningAsThread) return sq_throwerror(v, _SC("Slave VM not running as thread"))
  492. static void *slaveThreadTask(void *p)
  493. {
  494. SlaveVM_st *self = (SlaveVM_st*)p;
  495. self->thread_id = mg_thread_self();
  496. HSQUIRRELVM svm = self->v;
  497. int rc;
  498. SQInteger top, argc;
  499. rc = sq_getinteger(svm, -1, &top) == SQ_OK;
  500. if(rc)
  501. {
  502. rc = sq_getinteger(svm, -2, &argc) == SQ_OK;
  503. if(rc)
  504. {
  505. sq_pop(svm, 2); //remove top & argc
  506. rc = sq_call(svm, argc+1, SQFalse, SQFalse);
  507. }
  508. sq_settop(svm, top);
  509. }
  510. rc = mg_thread_mutex_unlock(&self->mutex);
  511. self->thread_id = 0;
  512. //printf("%d:%d:%s\n", __LINE__, rc, __FILE__);
  513. return NULL;
  514. }
  515. extern void sq_system_sleep(int n);
  516. static SQRESULT sq_slave_vm_thread_run(HSQUIRRELVM v)
  517. {
  518. GET_sq_slave_vm_INSTANCE(v, 1);
  519. check_run_as_thread(v);
  520. int rc;
  521. const SQChar *func_name;
  522. SQInteger func_name_size;
  523. if(self->thread_id) return sq_throwerror(v, _SC("slavevm already running in a thread"));
  524. SQInteger top = sq_gettop(svm);
  525. rc = mg_thread_mutex_trylock(&self->mutex) == 0;
  526. if(rc)
  527. {
  528. sq_getstr_and_size(v, 2, &func_name, &func_name_size);
  529. sq_pushstring(svm, func_name, func_name_size);
  530. sq_getonroottable(svm);
  531. //printf("%d:%s\n", __LINE__, __FILE__);
  532. rc = (sq_gettype(svm, -1) == OT_CLOSURE);
  533. if( rc )
  534. {
  535. //printf("%d:%s\n", __LINE__, __FILE__);
  536. sq_pushroottable(svm);
  537. int argc = sq_gettop(v) - 2;
  538. rc = ( (argc == 0) || (copy_values_between_vms(svm, v, argc, 3) == SQ_OK));
  539. if(rc)
  540. {
  541. sq_pushinteger(svm, argc);
  542. sq_pushinteger(svm, top);
  543. rc = mg_start_thread(slaveThreadTask, self) == 0;
  544. if(rc) goto done;
  545. }
  546. //printf("%d:%d:%s\n", __LINE__, rc, __FILE__);
  547. }
  548. sq_settop(svm, top); //something went wrong reset slave stack
  549. }
  550. done:
  551. sq_pushbool(v, rc);
  552. return 1;
  553. }
  554. static SQRESULT sq_slave_vm_thread_trylock(HSQUIRRELVM v)
  555. {
  556. GET_sq_slave_vm_INSTANCE(v, 1);
  557. check_run_as_thread(v);
  558. sq_pushinteger(v, mg_thread_mutex_trylock(&self->mutex));
  559. return 1;
  560. }
  561. static SQRESULT sq_slave_vm_thread_unlock(HSQUIRRELVM v)
  562. {
  563. GET_sq_slave_vm_INSTANCE(v, 1);
  564. check_run_as_thread(v);
  565. sq_pushinteger(v, mg_thread_mutex_unlock(&self->mutex));
  566. return 1;
  567. }
  568. static SQRESULT sq_slave_vm_thread_cond_signal(HSQUIRRELVM v)
  569. {
  570. GET_sq_slave_vm_INSTANCE(v, 1);
  571. check_run_as_thread(v);
  572. sq_pushinteger(v, mg_thread_cond_signal(&self->cond_var));
  573. return 1;
  574. }
  575. static SQRESULT sq_slave_vm_thread_master_state(HSQUIRRELVM v)
  576. {
  577. SQ_FUNC_VARS(v);
  578. GET_sq_slave_vm_INSTANCE(v, 1);
  579. check_run_as_thread(v);
  580. SQ_OPT_INTEGER(v, 2, state, 0);
  581. sq_pushinteger(v, self->master_state);
  582. if(sq_gettop(v) > 1) self->master_state = state;
  583. return 1;
  584. }
  585. static SQRESULT sq_slave_vm_thread_slave_state(HSQUIRRELVM v)
  586. {
  587. GET_sq_slave_vm_INSTANCE(v, 1);
  588. check_run_as_thread(v);
  589. sq_pushinteger(v, self->slave_state);
  590. return 1;
  591. }
  592. #endif // SLAVE_VM_WITH_OS_THREADS
  593. #ifdef __cplusplus
  594. extern "C" {
  595. #endif
  596. SQRESULT sqext_register_sq_slave_vm(HSQUIRRELVM v)
  597. {
  598. const SQChar get_set_validation_mask[] = _SC("x s|n|p s|n|b|a|t|p|o");
  599. sq_pushstring(v,sq_slave_vm_TAG, -1);
  600. sq_newclass(v, SQFalse);
  601. sq_settypetag(v,-1,(void*)sq_slave_vm_TAG);
  602. sq_insertfunc(v, _SC("constructor"), sq_slave_vm_constructor, -1, _SC("xibb"), SQFalse);
  603. sq_insertfunc(v, _SC("_tostring"), sq_slave_vm__tostring, 1, _SC("x"), SQFalse);
  604. sq_insertfunc(v, _SC("close"), sq_slave_vm_close, 1, _SC("x"), SQFalse);
  605. sq_insertfunc(v, _SC("set"), sq_slave_vm_set, 3, get_set_validation_mask, SQFalse);
  606. sq_insertfunc(v, _SC("_set"), sq_slave_vm_set, 3, get_set_validation_mask, SQFalse);
  607. sq_insertfunc(v, _SC("get"), sq_slave_vm_get, -2, get_set_validation_mask, SQFalse);
  608. sq_insertfunc(v, _SC("_get"), sq_slave_vm_get, -2, get_set_validation_mask, SQFalse);
  609. sq_insertfunc(v, _SC("dofile"), sq_slave_vm_dofile, -2, _SC("xsbbb"), SQFalse);
  610. sq_insertfunc(v, _SC("loadfile"), sq_slave_vm_loadfile, -3, _SC("xssbb"), SQFalse);
  611. sq_insertfunc(v, _SC("dostring"), sq_slave_vm_dostring, -2, _SC("xsb"), SQFalse);
  612. sq_insertfunc(v, _SC("compilestring"), sq_slave_vm_compilestring, -3, _SC("xssbbi"), SQFalse);
  613. sq_insertfunc(v, _SC("call"), sq_slave_vm_call, -3, _SC("xbs"), SQFalse);
  614. sq_insertfunc(v, _SC("is_thread_idle"), sq_slave_vm_is_thread_idle, 1, _SC("x"), SQFalse);
  615. sq_insertfunc(v, _SC("is_runing_as_thread"), sq_slave_vm_is_runing_as_thread, 1, _SC("x"), SQFalse);
  616. #ifdef SLAVE_VM_WITH_OS_THREADS
  617. sq_insertfunc(v, _SC("thread_run"), sq_slave_vm_thread_run, -2, _SC("xs"), SQFalse);
  618. sq_insertfunc(v, _SC("thread_trylock"), sq_slave_vm_thread_trylock, 1, _SC("x"), SQFalse);
  619. sq_insertfunc(v, _SC("thread_unlock"), sq_slave_vm_thread_unlock, 1, _SC("x"), SQFalse);
  620. sq_insertfunc(v, _SC("thread_cond_signal"), sq_slave_vm_thread_cond_signal, 1, _SC("x"), SQFalse);
  621. sq_insertfunc(v, _SC("thread_master_state"), sq_slave_vm_thread_master_state, -1, _SC("xi"), SQFalse);
  622. sq_insertfunc(v, _SC("thread_slave_state"), sq_slave_vm_thread_slave_state, 1, _SC("x"), SQFalse);
  623. #endif // SLAVE_VM_WITH_OS_THREADS
  624. sq_newslot(v,-3,SQTrue); //push sq_slave_vm class
  625. return 0;
  626. }
  627. #ifdef __cplusplus
  628. }
  629. #endif