You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

120 lines
3.6 KiB

9 years ago
/* Copyright 2016 Alex 'AdUser' Z (ad_user@runbox.com)
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2 as
* published by the Free Software Foundation.
*/
#include "common.h"
#include "backend.h"
#define BACKEND_LIBRARY_PARAM "load"
f2b_backend_t *
f2b_backend_create(f2b_config_section_t *config, const char *id) {
f2b_config_param_t *param = NULL;
f2b_backend_t *backend = NULL;
int flags = RTLD_NOW | RTLD_LOCAL;
9 years ago
const char *dlerr = NULL;
assert(config != NULL);
assert(config->type == t_backend);
param = f2b_config_param_find(config->param, BACKEND_LIBRARY_PARAM);
9 years ago
if (!param) {
f2b_log_msg(log_error, "can't find '%s' param in backend config", BACKEND_LIBRARY_PARAM);
return NULL;
9 years ago
}
if ((backend = calloc(1, sizeof(f2b_backend_t))) == NULL)
return NULL;
if ((backend->h = dlopen(param->value, flags)) == NULL)
goto cleanup;
if ((*(void **) (&backend->create) = dlsym(backend->h, "create")) == NULL)
goto cleanup;
if ((*(void **) (&backend->config) = dlsym(backend->h, "config")) == NULL)
goto cleanup;
if ((*(void **) (&backend->ready) = dlsym(backend->h, "ready")) == NULL)
goto cleanup;
if ((*(void **) (&backend->error) = dlsym(backend->h, "error")) == NULL)
goto cleanup;
if ((*(void **) (&backend->start) = dlsym(backend->h, "start")) == NULL)
goto cleanup;
if ((*(void **) (&backend->stop) = dlsym(backend->h, "stop")) == NULL)
goto cleanup;
if ((*(void **) (&backend->ping) = dlsym(backend->h, "ping")) == NULL)
goto cleanup;
if ((*(void **) (&backend->ban) = dlsym(backend->h, "ban")) == NULL)
goto cleanup;
if ((*(void **) (&backend->check) = dlsym(backend->h, "check")) == NULL)
goto cleanup;
if ((*(void **) (&backend->unban) = dlsym(backend->h, "unban")) == NULL)
goto cleanup;
if ((*(void **) (&backend->destroy) = dlsym(backend->h, "destroy")) == NULL)
goto cleanup;
if ((backend->cfg = backend->create(id)) == NULL) {
f2b_log_msg(log_error, "backend create config failed");
goto cleanup;
}
/* try init */
9 years ago
for (param = config->param; param != NULL; param = param->next) {
if (strcmp(param->name, BACKEND_LIBRARY_PARAM) == 0)
continue;
if (backend->config(backend->cfg, param->name, param->value))
continue;
f2b_log_msg(log_warn, "param pair not accepted by backend '%s': %s=%s",
config->name, param->name, param->value);
}
if (backend->ready(backend->cfg))
return backend;
/* still not ready */
f2b_log_msg(log_error, "backend '%s' not fully configured", config->name);
cleanup:
9 years ago
dlerr = dlerror();
if (dlerr)
f2b_log_msg(log_error, "backend load error: %s", dlerr);
if (backend->h) {
if (backend->cfg && backend->destroy)
backend->destroy(backend->cfg);
dlclose(backend->h);
}
free(backend);
return NULL;
}
void
f2b_backend_destroy(f2b_backend_t *backend) {
assert(backend != NULL);
backend->destroy(backend->cfg);
dlclose(backend->h);
free(backend);
}
#define BACKEND_CMD_ARG0(CMD, RETURNS) \
RETURNS \
f2b_backend_ ## CMD(f2b_backend_t *backend) { \
assert(backend != NULL); \
return backend->CMD(backend->cfg); \
}
#define BACKEND_CMD_ARG1(CMD, RETURNS) \
RETURNS \
f2b_backend_ ## CMD(f2b_backend_t *backend, const char *ip) { \
assert(backend != NULL); \
return backend->CMD(backend->cfg, ip); \
}
BACKEND_CMD_ARG0(error, const char *)
BACKEND_CMD_ARG0(start, bool)
BACKEND_CMD_ARG0(stop, bool)
BACKEND_CMD_ARG0(ping, bool)
BACKEND_CMD_ARG1(check, bool)
BACKEND_CMD_ARG1(ban, bool)
BACKEND_CMD_ARG1(unban, bool)