X-Git-Url: http://git.madism.org/?a=blobdiff_plain;f=postlicyd%2Ffilter.c;h=7861af4f05c5db6b42ec07657727c80424d67818;hb=6deab7a7086ccb592daa0c41e12759ec0b9c7aa0;hp=d277542dec496de225419693f944b9f94b33715d;hpb=520b2f3bb198bfbb88b90b058ef610f0a9b980c6;p=apps%2Fpfixtools.git diff --git a/postlicyd/filter.c b/postlicyd/filter.c index d277542..7861af4 100644 --- a/postlicyd/filter.c +++ b/postlicyd/filter.c @@ -43,15 +43,30 @@ static filter_destructor_t destructors[FTK_count]; static bool hooks[FTK_count][HTK_count]; 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, .value = (char*)"DUNNO", .postfix = true, + .async = false, + .filter_id = 0 +}; + +static const filter_hook_t async_hook = { + .type = 0, + .value = NULL, + .postfix = false, + .async = true, .filter_id = 0 }; filter_type_t filter_register(const char *type, filter_constructor_t constructor, - filter_destructor_t destructor, filter_runner_t runner) + filter_destructor_t destructor, filter_runner_t runner, + filter_context_constructor_t context_constructor, + filter_context_destructor_t context_destructor) { filter_token tok = filter_tokenize(type, m_strlen(type)); CHECK_FILTER(tok); @@ -59,6 +74,9 @@ filter_type_t filter_register(const char *type, filter_constructor_t constructor runners[tok] = runner; constructors[tok] = constructor; destructors[tok] = destructor; + + ctx_constructors[tok] = context_constructor; + ctx_destructors[tok] = context_destructor; return tok; } @@ -84,6 +102,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; @@ -163,17 +186,18 @@ void filter_wipe(filter_t *filter) p_delete(&filter->name); } -const filter_hook_t *filter_run(const filter_t *filter, const query_t *query) +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); if (res == HTK_ABORT) { return NULL; } - debug("filter run, result is %s", htokens[res]); + if (res == HTK_ASYNC) { + return &async_hook; + } while (start < end) { int mid = (start + end) / 2; @@ -192,24 +216,41 @@ const filter_hook_t *filter_run(const filter_t *filter, const query_t *query) return &default_hook; } -bool filter_test(const filter_t *filter, const query_t *query, filter_result_t result) +const filter_hook_t *filter_run(const filter_t *filter, const query_t *query, + filter_context_t *context) { - return !!(runners[filter->type](filter, query) == result); + debug("running filter %s (%s)", filter->name, ftokens[filter->type]); + filter_result_t res = runners[filter->type](filter, query, context); + + if (res == HTK_ASYNC) { + context->current_filter = filter; + } else { + context->current_filter = NULL; + } + + debug("filter run, result is %s", htokens[res]); + return filter_hook_for_result(filter, res); } -void filter_set_name(filter_t *filter, const char *name, ssize_t len) +bool filter_test(const filter_t *filter, const query_t *query, + filter_context_t *context, filter_result_t result) +{ + return !!(runners[filter->type](filter, query, context) == result); +} + +void filter_set_name(filter_t *filter, const char *name, int len) { filter->name = p_dupstr(name, len); } -bool filter_set_type(filter_t *filter, const char *type, ssize_t len) +bool filter_set_type(filter_t *filter, const char *type, int len) { filter->type = filter_tokenize(type, len); return filter->type != FTK_UNKNOWN; } -bool filter_add_param(filter_t *filter, const char *name, ssize_t name_len, - const char *value, ssize_t value_len) +bool filter_add_param(filter_t *filter, const char *name, int name_len, + const char *value, int value_len) { filter_param_t param; param.type = param_tokenize(name, name_len); @@ -228,8 +269,8 @@ bool filter_add_param(filter_t *filter, const char *name, ssize_t name_len, return true; } -bool filter_add_hook(filter_t *filter, const char *name, ssize_t name_len, - const char *value, ssize_t value_len) +bool filter_add_hook(filter_t *filter, const char *name, int name_len, + const char *value, int value_len) { filter_hook_t hook; hook.type = hook_tokenize(name, name_len); @@ -242,9 +283,42 @@ bool filter_add_hook(filter_t *filter, const char *name, ssize_t name_len, htokens[hook.type], ftokens[filter->type]); return false; } + hook.async = false; hook.postfix = (strncmp(value, "postfix:", 8) == 0); hook.value = m_strdup(hook.postfix ? value + 8 : value); hook.filter_id = -1; array_add(filter->hooks, hook); return true; } + +void filter_context_prepare(filter_context_t *context, void *qctx) +{ + for (int i = 0 ; i < FTK_count ; ++i) { + if (ctx_constructors[i] != NULL) { + context->contexts[i] = ctx_constructors[i](); + } + } + context->current_filter = NULL; + context->data = qctx; +} + +void filter_context_wipe(filter_context_t *context) +{ + for (int i = 0 ; i < FTK_count ; ++i) { + if (ctx_destructors[i] != NULL) { + ctx_destructors[i](context->contexts[i]); + } + } +} + +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; + } + hook = filter_hook_for_result(filter, result); + async_handler(context, hook); +}