From Lua 5.2: coroutine.running(). Needs -DLUAJIT_ENABLE_LUA52COMPAT.

1 files changed, 7 insertions, 1 deletions

diff --git a/src/lib_base.c b/src/lib_base.c
index 746a4fd6..bac49a0b 100644
--- a/src/lib_base.c
+++ b/src/lib_base.c
@@ -3,7 +3,7 @@
 ** Copyright (C) 2005-2011 Mike Pall. See Copyright Notice in luajit.h
 **
 ** Major portions taken verbatim or adapted from the Lua interpreter.
-** Copyright (C) 1994-2008 Lua.org, PUC-Rio. See Copyright Notice in lua.h
+** Copyright (C) 1994-2011 Lua.org, PUC-Rio. See Copyright Notice in lua.h
 */
 
 #include <stdio.h>
@@ -522,9 +522,15 @@ LJLIB_CF(coroutine_status)
 
 LJLIB_CF(coroutine_running)
 {
+#ifdef LUAJIT_ENABLE_LUA52COMPAT
+  int ismain = lua_pushthread(L);
+  setboolV(L->top++, ismain);
+  return 2;
+#else
   if (lua_pushthread(L))
     setnilV(L->top++);
   return 1;
+#endif
 }
 
 LJLIB_CF(coroutine_create)