#include "policy_tokens.h"
#include "server.h"
#include "config.h"
-#include "postlicyd.h"
+#include "query.h"
#define DAEMON_NAME "postlicyd"
-#define DAEMON_VERSION "0.3"
+#define DAEMON_VERSION "0.4"
#define DEFAULT_PORT 10000
#define RUNAS_USER "nobody"
#define RUNAS_GROUP "nogroup"
DECLARE_MAIN
-static config_t *config = NULL;
+typedef struct query_context_t {
+ query_t query;
+ filter_context_t context;
+ client_t *client;
+} query_context_t;
+static config_t *config = NULL;
+static bool refresh = false;
+static PA(client_t) busy = ARRAY_INIT;
-static void *query_starter(server_t* server)
+static void *query_starter(listener_t* server)
{
query_context_t *context = p_new(query_context_t, 1);
filter_context_prepare(&context->context, context);
static bool config_refresh(void *mconfig)
{
+ refresh = true;
if (filter_running > 0) {
- sleep(1);
return true;
}
- return config_reload(mconfig);
+ bool ret = config_reload(mconfig);
+ foreach (client_t **server, busy) {
+ client_io_ro(*server);
+ }}
+ array_len(busy) = 0;
+ refresh = false;
+ return ret;
}
-static void policy_answer(server_t *pcy, const char *message)
+static void policy_answer(client_t *pcy, const char *message)
{
- query_context_t *context = pcy->data;
+ query_context_t *context = client_data(pcy);
const query_t* query = &context->query;
+ buffer_t *buf = client_output_buffer(pcy);
- buffer_addstr(&pcy->obuf, "action=");
- buffer_ensure(&pcy->obuf, m_strlen(message) + 64);
+ /* Write reply "action=ACTION [text]" */
+ buffer_addstr(buf, "action=");
+ buffer_ensure(buf, m_strlen(message) + 64);
- ssize_t size = array_size(pcy->obuf) - array_len(pcy->obuf);
- ssize_t format_size = query_format(array_ptr(pcy->obuf, array_len(pcy->obuf)),
+ ssize_t size = array_size(*buf) - array_len(*buf);
+ ssize_t format_size = query_format(array_ptr(*buf, array_len(*buf)),
size, message, query);
if (format_size == -1) {
- buffer_addstr(&pcy->obuf, message);
+ buffer_addstr(buf, message);
} else if (format_size > size) {
- buffer_ensure(&pcy->obuf, format_size + 1);
- query_format(array_ptr(pcy->obuf, array_len(pcy->obuf)),
- array_size(pcy->obuf) - array_len(pcy->obuf),
+ buffer_ensure(buf, format_size + 1);
+ query_format(array_ptr(*buf, array_len(*buf)),
+ array_size(*buf) - array_len(*buf),
message, query);
- array_len(pcy->obuf) += format_size;
+ array_len(*buf) += format_size;
} else {
- array_len(pcy->obuf) += format_size;
+ array_len(*buf) += format_size;
}
- buffer_addstr(&pcy->obuf, "\n\n");
- buffer_consume(&pcy->ibuf, query->eoq - pcy->ibuf.data);
- server_rw(pcy);
+ buffer_addstr(buf, "\n\n");
+
+ /* Finalize query. */
+ buf = client_input_buffer(pcy);
+ buffer_consume(buf, query->eoq - buf->data);
+ client_io_rw(pcy);
}
-static const filter_t *next_filter(server_t *pcy, const filter_t *filter,
+static const filter_t *next_filter(client_t *pcy, const filter_t *filter,
const query_t *query, const filter_hook_t *hook, bool *ok) {
+#define MESSAGE_FORMAT "request client=%s from=<%s> to=<%s> at %s: "
+#define MESSAGE_PARAMS query->client_name, \
+ query->sender == NULL ? "undefined" : query->sender, \
+ query->recipient == NULL ? "undefined" : query->recipient, \
+ smtp_state_names[query->state]
+
+ if (hook != NULL) {
+ query_context_t *context = client_data(pcy);
+ if (hook->counter >= 0 && hook->counter < MAX_COUNTERS && hook->cost > 0) {
+ context->context.counters[hook->counter] += hook->cost;
+ debug(MESSAGE_FORMAT "added %d to counter %d (now %u)", MESSAGE_PARAMS,
+ hook->cost, hook->counter, context->context.counters[hook->counter]);
+ }
+ }
if (hook == NULL) {
- warn("request client=%s, from=<%s>, to=<%s>: aborted",
- query->client_name,
- query->sender == NULL ? "undefined" : query->sender,
- query->recipient == NULL ? "undefined" : query->recipient);
+ warn(MESSAGE_FORMAT "aborted", MESSAGE_PARAMS);
*ok = false;
return NULL;
} else if (hook->async) {
- debug("request client=%s, from=<%s>, to=<%s>: "
- "asynchronous filter from filter %s",
- query->client_name,
- query->sender == NULL ? "undefined" : query->sender,
- query->recipient == NULL ? "undefined" : query->recipient,
- filter->name);
+ debug(MESSAGE_FORMAT "asynchronous filter from filter %s",
+ MESSAGE_PARAMS, filter->name);
*ok = true;
return NULL;
} else if (hook->postfix) {
- info("request client=%s, from=<%s>, to=<%s>: "
- "awswer %s from filter %s: \"%s\"",
- query->client_name,
- query->sender == NULL ? "undefined" : query->sender,
- query->recipient == NULL ? "undefined" : query->recipient,
+ info(MESSAGE_FORMAT "awswer %s from filter %s: \"%s\"", MESSAGE_PARAMS,
htokens[hook->type], filter->name, hook->value);
policy_answer(pcy, hook->value);
*ok = true;
return NULL;
} else {
- debug("request client=%s, from=<%s>, to=<%s>: "
- "awswer %s from filter %s: next filter %s",
- query->client_name,
- query->sender == NULL ? "undefined" : query->sender,
- query->recipient == NULL ? "undefined" : query->recipient,
- htokens[hook->type], filter->name,
- (array_ptr(config->filters, hook->filter_id))->name);
+ debug(MESSAGE_FORMAT "awswer %s from filter %s: next filter %s",
+ MESSAGE_PARAMS, htokens[hook->type], filter->name,
+ (array_ptr(config->filters, hook->filter_id))->name);
return array_ptr(config->filters, hook->filter_id);
}
+#undef MESSAGE_PARAMS
+#undef MESSAGE_FORMAT
}
-static bool policy_process(server_t *pcy, const config_t *mconfig)
+static bool policy_process(client_t *pcy, const config_t *mconfig)
{
- query_context_t *context = pcy->data;
+ query_context_t *context = client_data(pcy);
const query_t* query = &context->query;
const filter_t *filter;
if (mconfig->entry_points[query->state] == -1) {
- warn("no filter defined for current protocol_state (%d)", query->state);
+ warn("no filter defined for current protocol_state (%s)", smtp_state_names[query->state]);
return false;
}
if (context->context.current_filter != NULL) {
}
}
-static int policy_run(server_t *pcy, void* vconfig)
+static int policy_run(client_t *pcy, void* vconfig)
{
- int search_offs = MAX(0, (int)(pcy->ibuf.len - 1));
- int nb = buffer_read(&pcy->ibuf, pcy->fd, -1);
- const char *eoq;
- query_context_t *context = pcy->data;
- query_t *query = &context->query;
- context->server = pcy;
const config_t *mconfig = vconfig;
+ if (refresh) {
+ array_add(busy, pcy);
+ return 0;
+ }
+
+ query_context_t *context = client_data(pcy);
+ query_t *query = &context->query;
+ context->client = pcy;
+
+ buffer_t *buf = client_input_buffer(pcy);
+ int search_offs = MAX(0, (int)(buf->len - 1));
+ int nb = client_read(pcy);
+ const char *eoq;
if (nb < 0) {
if (errno == EAGAIN || errno == EINTR)
return -1;
}
if (nb == 0) {
- if (pcy->ibuf.len)
+ if (buf->len)
err("unexpected end of data");
return -1;
}
- if (!(eoq = strstr(pcy->ibuf.data + search_offs, "\n\n")))
+ if (!(eoq = strstr(buf->data + search_offs, "\n\n"))) {
return 0;
+ }
- if (!query_parse(pcy->data, pcy->ibuf.data))
+ if (!query_parse(query, buf->data)) {
return -1;
+ }
query->eoq = eoq + strlen("\n\n");
- server_none(pcy);
+
+ /* The instance changed => reset the static context */
+ if (query->instance == NULL || strcmp(context->context.instance, query->instance) != 0) {
+ filter_context_clean(&context->context);
+ m_strcat(context->context.instance, 64, query->instance);
+ }
+ client_io_none(pcy);
return policy_process(pcy, mconfig) ? 0 : -1;
}
const filter_t *filter = context->current_filter;
query_context_t *qctx = context->data;
query_t *query = &qctx->query;
- server_t *server = qctx->server;
+ client_t *server = qctx->client;
context->current_filter = next_filter(server, filter, query, hook, &ok);
if (context->current_filter != NULL) {
ok = policy_process(server, config);
}
if (!ok) {
- server_release(server);
+ client_release(server);
+ }
+ if (refresh && filter_running == 0) {
+ config_refresh(config);
}
}
filter_async_handler_register(policy_async_handler);
return 0;
}
-module_init(postlicyd_init);
-int start_listener(int port)
+static void postlicyd_shutdown(void)
{
- return start_server(port, NULL, NULL);
+ array_deep_wipe(busy, client_delete);
}
+module_init(postlicyd_init);
+module_exit(postlicyd_shutdown);
/* administrivia {{{ */
return EXIT_FAILURE;
}
- info("starting %s v%s...", DAEMON_NAME, DAEMON_VERSION);
+ info("%s v%s...", DAEMON_NAME, DAEMON_VERSION);
if (pidfile_open(pidfile) < 0) {
crit("unable to write pidfile %s", pidfile);
pidfile_refresh();
- if (start_listener(config->port) < 0) {
+ if (start_listener(config->port) == NULL) {
return EXIT_FAILURE;
} else {
return server_loop(query_starter, query_stopper,
projects / apps / pfixtools.git / blobdiff