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.
 
 

75 lines
2.1 KiB

/* 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 "config.h"
#include "log.h"
#include "backend.h"
void usage() {
fprintf(stderr, "Usage: backend-test <config-file.conf> <id>\n");
exit(EXIT_FAILURE);
}
int main(int argc, char *argv[]) {
const char *ip = "127.0.0.17";
f2b_config_t config;
f2b_config_param_t *param = NULL;
f2b_config_section_t *section = NULL;
f2b_backend_t *backend = NULL;
if (argc < 3)
usage();
memset(&config, 0x0, sizeof(config));
if (f2b_config_load(&config, argv[1], false) != true) {
f2b_log_msg(log_error, "can't load config");
return EXIT_FAILURE;
}
if ((section = f2b_config_section_find(config.jails, "test")) == NULL) {
f2b_log_msg(log_error, "can't find config section for jail 'test'");
return EXIT_FAILURE;
}
if ((param = f2b_config_param_find(section->param, "backend")) == NULL) {
f2b_log_msg(log_error, "jail 'test' has not param named 'backend'");
return EXIT_FAILURE;
}
if ((section = f2b_config_section_find(config.backends, param->value)) == NULL) {
f2b_log_msg(log_error, "can't find config section for backend '%s'", param->value);
return EXIT_FAILURE;
}
if ((backend = f2b_backend_create(section, argv[2])) == NULL) {
f2b_log_msg(log_error, "can't create backend '%s' with id '%s'", param->value, argv[2]);
return EXIT_FAILURE;
}
if (!backend->ban(backend->cfg, ip)) {
f2b_log_msg(log_error, "action 'ban' failed");
goto cleanup;
}
if (!backend->check(backend->cfg, ip)) {
f2b_log_msg(log_error, "action 'check' failed");
goto cleanup;
}
if (!backend->unban(backend->cfg, ip)) {
f2b_log_msg(log_error, "action 'unban' failed");
goto cleanup;
}
f2b_log_msg(log_info, "all tests passed");
cleanup:
f2b_backend_destroy(backend);
f2b_config_free(&config);
return EXIT_SUCCESS;
}