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",
+ .counter = -1,
+ .cost = 0,
.postfix = true,
+ .async = false,
.filter_id = 0
};
+static const filter_hook_t async_hook = {
+ .type = 0,
+ .value = NULL,
+ .counter = -1,
+ .cost = 0,
+ .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_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);
- syslog(LOG_DEBUG, "filter type %s registered", type);
-
runners[tok] = runner;
constructors[tok] = constructor;
destructors[tok] = destructor;
+
+ ctx_constructors[tok] = context_constructor;
+ ctx_destructors[tok] = context_destructor;
return tok;
}
CHECK_FILTER(filter);
CHECK_HOOK(tok);
- syslog(LOG_DEBUG, "hook %s registered for filter type %s", name,
- ftokens[filter]);
-
hooks[filter][tok] = true;
return tok;
}
CHECK_FILTER(filter);
CHECK_PARAM(tok);
- syslog(LOG_INFO, "param %s registered for filter type %s", name,
- ftokens[filter]);
-
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)
{
bool ret = true;
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, htokens[hook->type]);
+ err("invalid filter name %s for hook %s",
+ hook->value, htokens[hook->type]);
return false;
}
p_delete(&hook->value);
return true;
}
-static inline bool filter_check_loop(filter_t *filter, A(filter_t) *array, bool cleanup)
+static inline bool filter_check_loop(filter_t *filter, A(filter_t) *array, int level)
{
- if (cleanup) {
- foreach (filter_t *filter2, *array) {
- filter2->seen = false;
- }}
- } else if (filter->seen) {
- return false;
- }
- if (filter->safe) {
+ if (filter->last_seen == level) {
return true;
}
- filter->seen = true;
+ filter->last_seen = level;
foreach (filter_hook_t *hook, filter->hooks) {
if (hook->postfix) {
continue;
}
- if (!filter_check_loop(array_ptr(*array, hook->filter_id), array, false)) {
+ if (hook->filter_id == level) {
+ return false;
+ }
+ if (!filter_check_loop(array_ptr(*array, hook->filter_id), array, level)) {
return false;
}
}}
- filter->safe = true;
return true;
}
bool filter_check_safety(A(filter_t) *array)
{
foreach (filter_t *filter, *array) {
- if (!filter_check_loop(filter, array, true)) {
- syslog(LOG_ERR, "the filter tree contains a loop");
+ if (!filter_check_loop(filter, array, __Ai)) {
+ err("the filter tree contains a loop");
return false;
}
}}
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;
- //syslog(LOG_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;
}
- //syslog(LOG_DEBUG, "filter run, result is %s", htokens[res]);
+ 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) {
- //syslog(LOG_DEBUG, "return hook of type %s, value %s",
- // htokens[hook->type], hook->value);
+ debug("return hook of type %s, value %s",
+ htokens[hook->type], hook->value);
return hook;
} else if (res < hook->type) {
end = mid;
start = mid + 1;
}
}
- syslog(LOG_WARNING, "missing hook %s for filter %s",
- htokens[res], filter->name);
+ warn("missing hook %s for filter %s", htokens[res], filter->name);
return &default_hook;
}
-void filter_set_name(filter_t *filter, const char *name, ssize_t len)
+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, 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_param_t param;
param.type = param_tokenize(name, name_len);
if (param.type == ATK_UNKNOWN) {
- syslog(LOG_ERR, "unknown parameter %.*s", name_len, name);
+ err("unknown parameter %.*s", name_len, name);
return false;
}
if (!params[filter->type][param.type]) {
- syslog(LOG_ERR, "hook %s is not valid for filter %s",
- atokens[param.type], ftokens[filter->type]);
+ err("hook %s is not valid for filter %s",
+ atokens[param.type], ftokens[filter->type]);
return false;
}
- param.value = m_strdup(value);
+ 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.type = hook_tokenize(name, name_len);
if (hook.type == HTK_UNKNOWN) {
- syslog(LOG_ERR, "unknown hook type %.*s", name_len, name);
+ err("unknown hook type %.*s", name_len, name);
return false;
}
if (!hooks[filter->type][hook.type] || hook.type == HTK_ABORT) {
- syslog(LOG_ERR, "hook %s not is valid for filter %s",
- htokens[hook.type], ftokens[filter->type]);
+ err("hook %s not is valid for filter %s",
+ htokens[hook.type], ftokens[filter->type]);
return false;
}
+ hook.async = false;
+
+ /* Value format is (counter:id:incr)?(postfix:reply|filter_name)
+ */
+ if (strncmp(value, "counter:", 8) == 0) {
+ char *end = NULL;
+ value += 8;
+ hook.counter = strtol(value, &end, 10);
+ if (end == value || *end != ':') {
+ err("hook %s, cannot read counter id", htokens[hook.type]);
+ return false;
+ } else if (hook.counter < 0 || hook.counter >= MAX_COUNTERS) {
+ err("hook %s, invalid counter id %d", htokens[hook.type], hook.counter);
+ return false;
+ }
+ value = end + 1;
+ hook.cost = strtol(value, &end, 10);
+ if (end == value || *end != ':') {
+ err("hook %s, cannot read counter increment", htokens[hook.type]);
+ return false;
+ } else if (hook.cost < 0) {
+ err("hook %s, invalid counter increment value %d", htokens[hook.type],
+ hook.cost);
+ return false;
+ }
+ value = end + 1;
+ } else {
+ hook.counter = -1;
+ hook.cost = 0;
+ }
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_context_clean(filter_context_t *context)
+{
+ p_clear(&context->counters, 1);
+ context->instance[0] = '\0';
+}
+
+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);
+}