Browse Source

[Minor] Add some fail-safety

tags/2.6
Vsevolod Stakhov 3 years ago
parent
commit
3c461d3ef5
1 changed files with 18 additions and 0 deletions
  1. 18
    0
      src/lua/lua_task.c

+ 18
- 0
src/lua/lua_task.c View File

@@ -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);
}

Loading…
Cancel
Save