From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Tue, 18 Aug 2020 15:06:21 +0000 (+0100)
Subject: [Minor] Add some fail-safety
X-Git-Tag: 2.6~147
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=3c461d3ef5bd61514a3d20d0927f5791c2a8be2f;p=thirdparty%2Frspamd.git

[Minor] Add some fail-safety
---

diff --git a/src/lua/lua_task.c b/src/lua/lua_task.c
index 27fee395d8..09479ea215 100644
--- a/src/lua/lua_task.c
+++ b/src/lua/lua_task.c
@@ -4732,6 +4732,12 @@ lua_task_get_groups (lua_State *L)
 			mres = rspamd_find_metric_result (task, lua_tostring (L, 3));
 		}
 
+		if (mres == NULL) {
+			lua_pushnil (L);
+
+			return 1;
+		}
+
 		lua_createtable (L, 0, kh_size (mres->sym_groups));
 
 		kh_foreach (mres->sym_groups, gr, gr_score, {
@@ -5943,6 +5949,12 @@ lua_task_get_metric_result (lua_State *L)
 
 		if (lua_isstring (L, 2)) {
 			metric_res = rspamd_find_metric_result (task, lua_tostring (L, 2));
+
+			if (metric_res == NULL) {
+				lua_pushnil (L);
+
+				return 1;
+			}
 		}
 
 		/* Fields added:
@@ -6043,6 +6055,12 @@ lua_task_get_metric_action (lua_State *L)
 			mres = rspamd_find_metric_result (task, lua_tostring (L, 2));
 		}
 
+		if (mres == NULL) {
+			lua_pushnil (L);
+
+			return 1;
+		}
+
 		action = rspamd_check_action_metric (task, NULL, mres);
 		lua_pushstring (L, action->name);
 	}