Add luaL_execresult to C API.

2 files changed, 42 insertions, 2 deletions

diff --git a/c-api/compat-5.3.c b/c-api/compat-5.3.c
index 962b951..0e06176 100644
--- a/c-api/compat-5.3.c
+++ b/c-api/compat-5.3.c
@@ -14,7 +14,7 @@
 
 
 /* definitions for Lua 5.1 only */
-#if defined( LUA_VERSION_NUM ) && LUA_VERSION_NUM == 501
+#if defined(LUA_VERSION_NUM) && LUA_VERSION_NUM == 501
 
 
 COMPAT53_API int lua_absindex (lua_State *L, int i) {
@@ -366,6 +366,44 @@ COMPAT53_API int luaL_fileresult (lua_State *L, int stat, const char *fname) {
     return 3;
   }
 }
+
+
+#if !defined(l_inspectstat) && \
+    (defined(unix) || defined(__unix) || defined(__unix__) || \
+     defined(__TOS_AIX__) || defined(_SYSTYPE_BSD))
+/* some form of unix; check feature macros in unistd.h for details */
+#  include <unistd.h>
+/* check posix version; the relevant include files and macros probably
+ * were available before 2001, but I'm not sure */
+#  if defined(_POSIX_VERSION) && _POSIX_VERSION >= 200112L
+#    include <sys/wait.h>
+#    define l_inspectstat(stat,what) \
+  if (WIFEXITED(stat)) { stat = WEXITSTATUS(stat); } \
+  else if (WIFSIGNALED(stat)) { stat = WTERMSIG(stat); what = "signal"; }
+#  endif
+#endif
+
+/* provide default (no-op) version */
+#if !defined(l_inspectstat)
+#  define l_inspectstat(stat,what) ((void)0)
+#endif
+
+
+COMPAT53_API int luaL_execresult (lua_State *L, int stat) {
+  const char *what = "exit";
+  if (stat == -1)
+    return luaL_fileresult(L, 0, NULL);
+  else {
+    l_inspectstat(stat, what);
+    if (*what == 'e' && stat == 0)
+      lua_pushboolean(L, 1);
+    else
+      lua_pushnil(L);
+    lua_pushstring(L, what);
+    lua_pushinteger(L, stat);
+    return 3;
+  }
+}
 #endif /* not COMPAT53_IS_LUAJIT */
 
 
diff --git a/c-api/compat-5.3.h b/c-api/compat-5.3.h
index dc8d3ea..cd6764c 100644
--- a/c-api/compat-5.3.h
+++ b/c-api/compat-5.3.h
@@ -45,7 +45,6 @@
  * lua_upvaluejoin
  * lua_version
  * lua_yieldk
- * luaL_execresult
  * luaL_loadbufferx
  * luaL_loadfilex
  */
@@ -150,6 +149,9 @@ COMPAT53_API void luaL_traceback (lua_State *L, lua_State *L1, const char *msg,
 
 #define luaL_fileresult COMPAT53_CONCAT(COMPAT53_PREFIX, L_fileresult)
 COMPAT53_API int luaL_fileresult (lua_State *L, int stat, const char *fname);
+
+#define luaL_execresult COMPAT53_CONCAT(COMPAT53_PREFIX, L_execresult)
+COMPAT53_API int luaL_execresult (lua_State *L, int stat);
 #endif /* COMPAT53_IS_LUAJIT */
 
 #define lua_callk(L, na, nr, ctx, cont) \