Unbreak lua_worker with ucl.

1 files changed, 23 insertions, 0 deletions

diff --git a/src/cfg_rcl.c b/src/cfg_rcl.c
index 41a1cd8c8..3f419ceb8 100644
--- a/src/cfg_rcl.c
+++ b/src/cfg_rcl.c
@@ -507,6 +507,11 @@ rspamd_rcl_worker_handler (struct config_file *cfg, ucl_object_t *obj,
 				}
 			}
 		}
+		if (wparser->def_obj_parser != NULL) {
+			if (! wparser->def_obj_parser (obj, wparser->def_ud)) {
+				return FALSE;
+			}
+		}
 	}
 
 	cfg->workers = g_list_prepend (cfg->workers, wrk);
@@ -1393,3 +1398,21 @@ rspamd_rcl_register_worker_option (struct config_file *cfg, gint type, const gch
 	nhandler->handler = handler;
 	HASH_ADD_STR (nparser->parsers, name, nhandler);
 }
+
+
+void
+rspamd_rcl_register_worker_parser (struct config_file *cfg, gint type,
+		gboolean (*func)(ucl_object_t *, gpointer), gpointer ud)
+{
+	struct rspamd_worker_cfg_parser *nparser;
+	HASH_FIND_INT (cfg->wrk_parsers, &type, nparser);
+	if (nparser == NULL) {
+		/* Allocate new parser for this worker */
+		nparser = memory_pool_alloc0 (cfg->cfg_pool, sizeof (struct rspamd_worker_cfg_parser));
+		nparser->type = type;
+		HASH_ADD_INT (cfg->wrk_parsers, type, nparser);
+	}
+
+	nparser->def_obj_parser = func;
+	nparser->def_ud = ud;
+}