X-Git-Url: http://git.madism.org/?a=blobdiff_plain;f=postlicyd%2Ffilter.c;h=6d98d261d11c8912a5b25708a20e0f4f3a2ea500;hb=43cbaab1fc66139d3dfd87a2f8b6be5d906ee317;hp=deb31902a69efb123eab3f20dc1fa36e2d06e619;hpb=0df827a28124af3bb45cea7a3b80e5d1800bb1e2;p=apps%2Fpfixtools.git diff --git a/postlicyd/filter.c b/postlicyd/filter.c index deb3190..6d98d26 100644 --- a/postlicyd/filter.c +++ b/postlicyd/filter.c @@ -40,18 +40,73 @@ static filter_runner_t runners[FTK_count]; static filter_constructor_t constructors[FTK_count]; static filter_destructor_t destructors[FTK_count]; +static bool hooks[FTK_count][HTK_count]; +static bool params[FTK_count][ATK_count]; -void filter_register(const char *type, filter_constructor_t constructor, - filter_destructor_t destructor, filter_runner_t runner) +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 +}; + +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, + filter_context_destructor_t context_destructor) { filter_token tok = filter_tokenize(type, m_strlen(type)); - assert(tok != FTK_UNKNOWN && "Unknown filter type"); - - syslog(LOG_INFO, "filter type %s registered", type); + CHECK_FILTER(tok); runners[tok] = runner; constructors[tok] = constructor; destructors[tok] = destructor; + + ctx_constructors[tok] = context_constructor; + ctx_destructors[tok] = context_destructor; + return tok; +} + +filter_result_t filter_hook_register(filter_type_t filter, + const char *name) +{ + filter_result_t tok = hook_tokenize(name, m_strlen(name)); + CHECK_FILTER(filter); + CHECK_HOOK(tok); + + hooks[filter][tok] = true; + return tok; +} + +filter_param_id_t filter_param_register(filter_type_t filter, + const char *name) +{ + filter_param_id_t tok = param_tokenize(name, m_strlen(name)); + CHECK_FILTER(filter); + CHECK_PARAM(tok); + + params[filter][tok] = true; + return tok; +} + +void filter_async_handler_register(filter_async_handler_t handler) +{ + async_handler = handler; } bool filter_build(filter_t *filter) @@ -64,7 +119,7 @@ bool filter_build(filter_t *filter) # define QSORT_TYPE filter_hook_t # define QSORT_BASE filter->hooks.data # define QSORT_NELT filter->hooks.len -# define QSORT_LT(a,b) strcmp(a->name, b->name) < 0 +# define QSORT_LT(a,b) a->type < b->type # include "qsort.c" } filter_constructor_t constructor = constructors[filter->type]; @@ -81,8 +136,8 @@ bool filter_update_references(filter_t *filter, A(filter_t) *filter_list) if (!hook->postfix) { hook->filter_id = filter_find_with_name(filter_list, hook->value); if (hook->filter_id == -1) { - syslog(LOG_ERR, "invalid filter name %s for hook %s", - hook->value, hook->name); + err("invalid filter name %s for hook %s", + hook->value, htokens[hook->type]); return false; } p_delete(&hook->value); @@ -91,6 +146,37 @@ bool filter_update_references(filter_t *filter, A(filter_t) *filter_list) return true; } +static inline bool filter_check_loop(filter_t *filter, A(filter_t) *array, int level) +{ + if (filter->last_seen == level) { + return true; + } + filter->last_seen = level; + foreach (filter_hook_t *hook, filter->hooks) { + if (hook->postfix) { + continue; + } + if (hook->filter_id == level) { + return false; + } + if (!filter_check_loop(array_ptr(*array, hook->filter_id), array, level)) { + return false; + } + }} + return true; +} + +bool filter_check_safety(A(filter_t) *array) +{ + foreach (filter_t *filter, *array) { + if (!filter_check_loop(filter, array, __Ai)) { + err("the filter tree contains a loop"); + return false; + } + }} + return true; +} + void filter_wipe(filter_t *filter) { filter_destructor_t destructor = destructors[filter->type]; @@ -102,42 +188,147 @@ void filter_wipe(filter_t *filter) p_delete(&filter->name); } -filter_result_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) { - return runners[filter->type](filter, query); + int start = 0; + int end = filter->hooks.len; + + if (res == HTK_ABORT) { + return NULL; + } + if (res == HTK_ASYNC) { + return &async_hook; + } + + while (start < end) { + int mid = (start + end) / 2; + filter_hook_t *hook = array_ptr(filter->hooks, mid); + if (hook->type == res) { + debug("return hook of type %s, value %s", + htokens[hook->type], hook->value); + return hook; + } else if (res < hook->type) { + end = mid; + } else { + start = mid + 1; + } + } + warn("missing hook %s for filter %s", htokens[res], filter->name); + 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) +{ + return !!(runners[filter->type](filter, query, context) == result); } -void filter_set_name(filter_t *filter, const char *name, ssize_t len) +void filter_set_name(filter_t *filter, const char *name, int len) { - filter->name = p_new(char, len + 1); - memcpy(filter->name, name, len); - filter->name[len] = '\0'; + 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_params_t param; - param.name = m_strdup(name); - param.value = m_strdup(value); + filter_param_t param; + param.type = param_tokenize(name, name_len); + if (param.type == ATK_UNKNOWN) { + err("unknown parameter %.*s", name_len, name); + return false; + } + if (!params[filter->type][param.type]) { + err("hook %s is not valid for filter %s", + atokens[param.type], ftokens[filter->type]); + return false; + } + param.value = p_dupstr(value, value_len); + param.value_len = value_len; array_add(filter->params, param); 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.name = m_strdup(name); + hook.type = hook_tokenize(name, name_len); + if (hook.type == HTK_UNKNOWN) { + err("unknown hook type %.*s", name_len, name); + return false; + } + if (!hooks[filter->type][hook.type] || hook.type == HTK_ABORT) { + err("hook %s not is valid for filter %s", + htokens[hook.type], ftokens[filter->type]); + return 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; } + +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; + } + --filter_running; + hook = filter_hook_for_result(filter, result); + async_handler(context, hook); +}