From 8c596dc1efa8a1267c222b168a4de9c8ba254760 Mon Sep 17 00:00:00 2001
From: Li Jin <dragon-fly@qq.com>
Date: Sat, 18 Jul 2020 16:45:50 +0800
Subject: fix issue for using return statement with export.

---
 src/lua/liolib.c | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/lua/liolib.c')

diff --git a/src/lua/liolib.c b/src/lua/liolib.c
index 7ac3444..60ab1bf 100644
--- a/src/lua/liolib.c
+++ b/src/lua/liolib.c
@@ -52,6 +52,12 @@ static int l_checkmode (const char *mode) {
 ** =======================================================
 */
 
+#if !defined(l_checkmodep)
+/* By default, Lua accepts only "r" or "w" as mode */
+#define l_checkmodep(m)	((m[0] == 'r' || m[0] == 'w') && m[1] == '\0')
+#endif
+
+
 #if !defined(l_popen)		/* { */
 
 #if defined(LUA_USE_POSIX)	/* { */
@@ -279,6 +285,7 @@ static int io_popen (lua_State *L) {
   const char *filename = luaL_checkstring(L, 1);
   const char *mode = luaL_optstring(L, 2, "r");
   LStream *p = newprefile(L);
+  luaL_argcheck(L, l_checkmodep(mode), 2, "invalid mode");
   p->f = l_popen(L, filename, mode);
   p->closef = &io_pclose;
   return (p->f == NULL) ? luaL_fileresult(L, 0, filename) : 1;
-- 
cgit v1.2.3-55-g6feb