From: Vsevolod Stakhov <vsevolod@rspamd.com>
Date: Tue, 16 May 2023 10:41:53 +0000 (+0100)
Subject: [Minor] Fix newindex
X-Git-Tag: 3.6~115
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=68a07a2c1d66746a9265ddcf539c494b96a5f63f;p=thirdparty%2Frspamd.git

[Minor] Fix newindex
---

diff --git a/src/lua/lua_config.c b/src/lua/lua_config.c
index b0e2a475c0..680203c1e6 100644
--- a/src/lua/lua_config.c
+++ b/src/lua/lua_config.c
@@ -2729,7 +2729,7 @@ lua_config_newindex (lua_State *L)
 
 			/* Deal with flags and ids */
 			lua_pushstring (L, "flags");
-			lua_gettable (L, 2);
+			lua_gettable (L, -2);
 			if (lua_type(L, -1) == LUA_TSTRING) {
 				flags = lua_parse_symbol_flags (lua_tostring (L, -1));
 			}
@@ -2741,7 +2741,7 @@ lua_config_newindex (lua_State *L)
 			lua_pop (L, 1); /* Clean flags */
 
 			lua_pushstring(L, "allowed_ids");
-			lua_gettable (L, 2);
+			lua_gettable (L, -2);
 			if (lua_type(L, -1) == LUA_TSTRING) {
 				allowed_ids = rspamd_process_id_list(lua_tostring (L, -1));
 			}
@@ -2755,6 +2755,21 @@ lua_config_newindex (lua_State *L)
 			}
 			lua_pop (L, 1);
 
+			lua_pushstring(L, "forbidden_ids");
+			lua_gettable (L, -2);
+			if (lua_type(L, -1) == LUA_TSTRING) {
+				forbidden_ids = rspamd_process_id_list(lua_tostring (L, -1));
+			}
+			else if (lua_type(L, -1) == LUA_TTABLE) {
+				forbidden_ids = g_array_sized_new(FALSE, FALSE, sizeof (guint32),
+					rspamd_lua_table_size(L, -1));
+				for (lua_pushnil (L); lua_next (L, -2); lua_pop (L, 1)) {
+					guint32 v = lua_tointeger(L, -1);
+					g_array_append_val(forbidden_ids, v);
+				}
+			}
+			lua_pop (L, 1);
+
 			id = rspamd_register_symbol_fromlua (L,
 					cfg,
 					name,