From f4c7dc933d0df00c8ecc9308553a135a998399e2 Mon Sep 17 00:00:00 2001
From: William Wilgus <wilgus.william@gmail.com>
Date: Thu, 21 Oct 2021 22:26:27 -0400
Subject: lua add ability to pass PLUGIN_ ret vals with os.exit() + add defines

Change-Id: I45998245271fbd8fee4534f8497b014025e4dcc7
---
 apps/plugins/lua/rocklua.c | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

(limited to 'apps/plugins/lua/rocklua.c')

diff --git a/apps/plugins/lua/rocklua.c b/apps/plugins/lua/rocklua.c
index 3909f3008f..3cf0fce945 100644
--- a/apps/plugins/lua/rocklua.c
+++ b/apps/plugins/lua/rocklua.c
@@ -175,9 +175,10 @@ static int loadfile_newstate(lua_State **L, const char *filename)
 static void lua_atexit(void)
 {
   char *filename;
-
+  int err_n;
   if(Ls && lua_gettop(Ls) > 1)
   {
+    err_n = lua_tointeger(Ls, -1); /* os.exit? */
     if (Ls == lua_touserdata(Ls, -1)) /* signal from restart_lua */
     {
       filename = (char *) malloc((MAX_PATH * 2) + 1);
@@ -195,7 +196,12 @@ static void lua_atexit(void)
       free(filename);
       plugin_start(NULL);
     }
-    else if (lua_tointeger(Ls, -1) != 0) /* os.exit */
+    else if (err_n >= PLUGIN_USB_CONNECTED) /* INTERNAL PLUGIN RETVAL */
+    {
+      lua_close(Ls);
+      _exit(err_n);  /* don't call exit handler */
+    }
+    else if (err_n != 0)
     {
 ERR_RUN:
       lu_status = LUA_ERRRUN;
@@ -205,7 +211,7 @@ ERR_RUN:
     else
       lua_close(Ls);
   }
-  _exit(0); /* don't call exit handler */
+  _exit(PLUGIN_OK); /* don't call exit handler */
 }
 
 /* split filename at argchar
-- 
cgit v1.2.3