aboutsummaryrefslogtreecommitdiffstats
path: root/src/lua
diff options
context:
space:
mode:
Diffstat (limited to 'src/lua')
-rw-r--r--src/lua/lua_config.c77
-rw-r--r--src/lua/lua_task.c90
2 files changed, 128 insertions, 39 deletions
diff --git a/src/lua/lua_config.c b/src/lua/lua_config.c
index 266dbd111..24487f1b5 100644
--- a/src/lua/lua_config.c
+++ b/src/lua/lua_config.c
@@ -1267,24 +1267,45 @@ lua_metric_symbol_callback (struct rspamd_task *task,
for (i = level + first_opt; i <= last_pos; i++) {
if (lua_type (L, i) == LUA_TSTRING) {
- const char *opt = lua_tostring (L, i);
+ gsize optlen;
+ const char *opt = lua_tolstring (L, i, &optlen);
- rspamd_task_add_result_option (task, s, opt);
+ rspamd_task_add_result_option (task, s, opt, optlen);
+ }
+ else if (lua_type (L, i) == LUA_TUSERDATA) {
+ struct rspamd_lua_text *t = lua_check_text (L, i);
+
+ if (t) {
+ rspamd_task_add_result_option (task, s, t->start,
+ t->len);
+ }
}
else if (lua_type (L, i) == LUA_TTABLE) {
- lua_pushvalue (L, i);
+ gsize objlen = rspamd_lua_table_size (L, i);
- for (lua_pushnil (L); lua_next (L, -2); lua_pop (L, 1)) {
- const char *opt = lua_tostring (L, -1);
+ for (guint j = 1; j <= objlen; j ++) {
+ lua_rawgeti (L, i, j);
- rspamd_task_add_result_option (task, s, opt);
- }
+ if (lua_type (L, -1) == LUA_TSTRING) {
+ gsize optlen;
+ const char *opt = lua_tolstring (L, -1, &optlen);
- lua_pop (L, 1);
+ rspamd_task_add_result_option (task, s, opt, optlen);
+ }
+ else if (lua_type (L, -1) == LUA_TUSERDATA) {
+ struct rspamd_lua_text *t = lua_check_text (L, -1);
+
+ if (t) {
+ rspamd_task_add_result_option (task, s, t->start,
+ t->len);
+ }
+ }
+
+ lua_pop (L, 1);
+ }
}
}
}
-
}
lua_pop (L, nresults);
@@ -1403,20 +1424,42 @@ lua_metric_symbol_callback_return (struct thread_entry *thread_entry, int ret)
for (i = cd->stack_level + first_opt; i <= last_pos; i++) {
if (lua_type (L, i) == LUA_TSTRING) {
- const char *opt = lua_tostring (L, i);
+ gsize optlen;
+ const char *opt = lua_tolstring (L, i, &optlen);
+
+ rspamd_task_add_result_option (task, s, opt, optlen);
+ }
+ else if (lua_type (L, i) == LUA_TUSERDATA) {
+ struct rspamd_lua_text *t = lua_check_text (L, i);
- rspamd_task_add_result_option (task, s, opt);
+ if (t) {
+ rspamd_task_add_result_option (task, s, t->start,
+ t->len);
+ }
}
else if (lua_type (L, i) == LUA_TTABLE) {
- lua_pushvalue (L, i);
+ gsize objlen = rspamd_lua_table_size (L, i);
- for (lua_pushnil (L); lua_next (L, -2); lua_pop (L, 1)) {
- const char *opt = lua_tostring (L, -1);
+ for (guint j = 1; j <= objlen; j ++) {
+ lua_rawgeti (L, i, j);
- rspamd_task_add_result_option (task, s, opt);
- }
+ if (lua_type (L, -1) == LUA_TSTRING) {
+ gsize optlen;
+ const char *opt = lua_tolstring (L, -1, &optlen);
+
+ rspamd_task_add_result_option (task, s, opt, optlen);
+ }
+ else if (lua_type (L, -1) == LUA_TUSERDATA) {
+ struct rspamd_lua_text *t = lua_check_text (L, -1);
+
+ if (t) {
+ rspamd_task_add_result_option (task, s, t->start,
+ t->len);
+ }
+ }
- lua_pop (L, 1);
+ lua_pop (L, 1);
+ }
}
}
}
diff --git a/src/lua/lua_task.c b/src/lua/lua_task.c
index 80a00d97b..0b5fab09e 100644
--- a/src/lua/lua_task.c
+++ b/src/lua/lua_task.c
@@ -1837,7 +1837,7 @@ lua_task_insert_result (lua_State * L)
{
LUA_TRACE_POINT;
struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *symbol_name, *param;
+ const gchar *symbol_name;
double weight;
struct rspamd_symbol_result *s;
enum rspamd_symbol_insert_flags flags = RSPAMD_SYMBOL_INSERT_DEFAULT;
@@ -1868,17 +1868,43 @@ lua_task_insert_result (lua_State * L)
gint ltype = lua_type (L, i);
if (ltype == LUA_TSTRING) {
- param = luaL_checkstring (L, i);
- rspamd_task_add_result_option (task, s, param);
+ gsize optlen;
+ const char *opt = lua_tolstring (L, i, &optlen);
+
+ rspamd_task_add_result_option (task, s, opt, optlen);
+ }
+ else if (ltype == LUA_TUSERDATA) {
+ struct rspamd_lua_text *t = lua_check_text (L, i);
+
+ if (t) {
+ rspamd_task_add_result_option (task, s, t->start,
+ t->len);
+ }
}
else if (ltype == LUA_TTABLE) {
- lua_pushvalue (L, i);
- lua_pushnil (L);
+ gsize objlen = rspamd_lua_table_size (L, i);
+
+ for (guint j = 1; j <= objlen; j ++) {
+ lua_rawgeti (L, i, j);
+
+ if (lua_type (L, -1) == LUA_TSTRING) {
+ gsize optlen;
+ const char *opt = lua_tolstring (L, -1, &optlen);
- while (lua_next (L, -2)) {
- if (lua_isstring (L, -1)) {
- param = lua_tostring (L, -1);
- rspamd_task_add_result_option (task, s, param);
+ rspamd_task_add_result_option (task, s, opt, optlen);
+ }
+ else if (lua_type (L, -1) == LUA_TUSERDATA) {
+ struct rspamd_lua_text *t = lua_check_text (L, -1);
+
+ if (t) {
+ rspamd_task_add_result_option (task, s, t->start,
+ t->len);
+ }
+ else {
+ return luaL_error (L, "not rspamd_text option in a table "
+ "when adding symbol %s: %s type",
+ s->name);
+ }
}
else {
const gchar *tname = lua_typename (L, lua_type (L, -1));
@@ -1891,8 +1917,6 @@ lua_task_insert_result (lua_State * L)
lua_pop (L, 1);
}
-
- lua_pop (L, 1);
}
else if (ltype == LUA_TNIL) {
/* We have received a NULL option, it is not good but not a fatal error */
@@ -1923,7 +1947,7 @@ lua_task_adjust_result (lua_State * L)
{
LUA_TRACE_POINT;
struct rspamd_task *task = lua_check_task (L, 1);
- const gchar *symbol_name, *param;
+ const gchar *symbol_name;
struct rspamd_scan_result *metric_res;
struct rspamd_symbol_result *s = NULL;
double weight;
@@ -1956,20 +1980,42 @@ lua_task_adjust_result (lua_State * L)
if (s) {
for (i = 4; i <= top; i++) {
if (lua_type (L, i) == LUA_TSTRING) {
- param = luaL_checkstring (L, i);
- rspamd_task_add_result_option (task, s, param);
+ gsize optlen;
+ const char *opt = lua_tolstring (L, i, &optlen);
+
+ rspamd_task_add_result_option (task, s, opt, optlen);
+ }
+ else if (lua_type (L, i) == LUA_TUSERDATA) {
+ struct rspamd_lua_text *t = lua_check_text (L, i);
+
+ if (t) {
+ rspamd_task_add_result_option (task, s, t->start,
+ t->len);
+ }
}
else if (lua_type (L, i) == LUA_TTABLE) {
- lua_pushvalue (L, i);
- lua_pushnil (L);
+ gsize objlen = rspamd_lua_table_size (L, i);
+
+ for (guint j = 1; j <= objlen; j ++) {
+ lua_rawgeti (L, i, j);
+
+ if (lua_type (L, -1) == LUA_TSTRING) {
+ gsize optlen;
+ const char *opt = lua_tolstring (L, -1, &optlen);
+
+ rspamd_task_add_result_option (task, s, opt, optlen);
+ }
+ else if (lua_type (L, -1) == LUA_TUSERDATA) {
+ struct rspamd_lua_text *t = lua_check_text (L, -1);
+
+ if (t) {
+ rspamd_task_add_result_option (task, s, t->start,
+ t->len);
+ }
+ }
- while (lua_next (L, -2)) {
- param = lua_tostring (L, -1);
- rspamd_task_add_result_option (task, s, param);
lua_pop (L, 1);
}
-
- lua_pop (L, 1);
}
}
}
@@ -4374,7 +4420,7 @@ lua_push_symbol_result (lua_State *L,
lua_createtable (L, kh_size (s->options), 0);
DL_FOREACH (s->opts_head, opt) {
- lua_pushstring (L, (const char*)opt->option);
+ lua_pushlstring (L, opt->option, opt->optlen);
lua_rawseti (L, -2, j++);
}
generated by cgit v1.2.3 (git 2.39.1) at 2024-12-23 16:17:38 +0000