Just one more thing...

[apps/pfixtools.git] / postlicyd / filter.c

diff --git a/postlicyd/filter.c b/postlicyd/filter.c
index a150d6d..6d98d26 100644 (file)
--- a/postlicyd/filter.c
+++ b/postlicyd/filter.c
@@ -45,6 +45,7 @@ static bool                 params[FTK_count][ATK_count];
 
 static filter_context_constructor_t ctx_constructors[FTK_count];
 static filter_context_destructor_t  ctx_destructors[FTK_count];
+static filter_async_handler_t       async_handler = NULL;
 
 static const filter_hook_t default_hook = {
     .type      = 0,
@@ -62,6 +63,8 @@ static const filter_hook_t async_hook = {
     .filter_id = 0
 };
 
+uint32_t filter_running = 0;
+
 filter_type_t filter_register(const char *type, filter_constructor_t constructor,
                               filter_destructor_t destructor, filter_runner_t runner,
                               filter_context_constructor_t context_constructor,
@@ -101,6 +104,11 @@ filter_param_id_t filter_param_register(filter_type_t filter,
     return tok;
 }
 
+void filter_async_handler_register(filter_async_handler_t handler)
+{
+    async_handler = handler;
+}
+
 bool filter_build(filter_t *filter)
 {
     bool ret = true;
@@ -180,22 +188,16 @@ void filter_wipe(filter_t *filter)
     p_delete(&filter->name);
 }
 
-const filter_hook_t *filter_run(const filter_t *filter, const query_t *query,
-                                filter_context_t *context)
+static inline const filter_hook_t *filter_hook_for_result(const filter_t *filter,
+                                                          filter_result_t res)
 {
     int start = 0;
     int end   = filter->hooks.len;
-    debug("running filter %s (%s)", filter->name, ftokens[filter->type]);
-    filter_result_t res = runners[filter->type](filter, query, context);
 
-    context->current_filter = NULL;
-
-    debug("filter run, result is %s", htokens[res]);
     if (res == HTK_ABORT) {
         return NULL;
     }
     if (res == HTK_ASYNC) {
-        context->current_filter = filter;
         return &async_hook;
     }
 
@@ -216,6 +218,24 @@ const filter_hook_t *filter_run(const filter_t *filter, const query_t *query,
     return &default_hook;
 }
 
+const filter_hook_t *filter_run(const filter_t *filter, const query_t *query,
+                                filter_context_t *context)
+{
+    debug("running filter %s (%s)", filter->name, ftokens[filter->type]);
+    ++filter_running;
+    filter_result_t res = runners[filter->type](filter, query, context);
+
+    if (res == HTK_ASYNC) {
+        context->current_filter = filter;
+    } else {
+        --filter_running;
+        context->current_filter = NULL;
+    }
+
+    debug("filter run, result is %s", htokens[res]);
+    return filter_hook_for_result(filter, res);
+}
+
 bool filter_test(const filter_t *filter, const query_t *query,
                  filter_context_t *context, filter_result_t result)
 {
@@ -267,10 +287,15 @@ bool filter_add_hook(filter_t *filter, const char *name, int name_len,
             htokens[hook.type], ftokens[filter->type]);
         return false;
     }
-    hook.async   = false;
+    hook.async     = false;
+    hook.filter_id = -1;
+    hook.value      = NULL;
     hook.postfix = (strncmp(value, "postfix:", 8) == 0);
+    if (hook.postfix && query_format(NULL, 0, value + 8, NULL) == -1) {
+        err("invalid formatted text \"%s\"", value + 8);
+        return false;
+    }
     hook.value = m_strdup(hook.postfix ? value + 8 : value);
-    hook.filter_id = -1;
     array_add(filter->hooks, hook);
     return true;
 }
@@ -294,3 +319,16 @@ void filter_context_wipe(filter_context_t *context)
         }
     }
 }
+
+void filter_post_async_result(filter_context_t *context, filter_result_t result)
+{
+    const filter_t *filter = context->current_filter;
+    const filter_hook_t *hook = NULL;
+
+    if (result == HTK_ASYNC) {
+        return;
+    }
+    --filter_running;
+    hook = filter_hook_for_result(filter, result);
+    async_handler(context, hook);
+}