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.
 
 

135 lines
3.4 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 <backend.conf> <id>\n");
exit(EXIT_FAILURE);
}
/* returns -1 on error, 0 for 'ban', 1 for 'check' and 3 for 'unban'
* also fills buf with second arg after command name
*/
int
parse_input(char *line, char *buf, size_t bufsize) {
const char *ops[] = { "ban", "check", "unban", NULL };
int code = -1;
assert(line != NULL);
assert(buf != NULL);
while (isspace(*line))
line++;
if (*line == '\0')
return -1; /* empty line */
for (size_t i = 0; ops[i] != NULL; i++) {
size_t len = strlen(ops[i]);
if (strncmp(line, ops[i], len) != 0)
continue;
code = i;
line += len;
break;
}
while (isspace(*line))
line++;
if (*line == '\0')
return -1; /* missing second arg */
for (char *p = line; *p != '\0'; p++) {
if ((*p >= 'a' && *p <= 'f') || *p == '.' || *p == ':' ||
(*p >= 'A' && *p <= 'F') || isdigit(*p))
continue; /* valid ipaddr char */
/* everything else */
*p = '\0';
break;
}
if (*line == '\0')
return -1; /* garbled second arg */
strlcpy(buf, line, bufsize);
return code;
}
int main(int argc, char *argv[]) {
f2b_config_t config;
f2b_config_section_t *section = NULL;
f2b_backend_t *backend = NULL;
bool (*handler)(f2b_backend_t *, const char *) = NULL;
char line[256];
char addr[64];
if (argc < 3)
usage();
memset(&config, 0x0, sizeof(config));
if (f2b_config_load(&config, argv[1], false) != true) {
f2b_log_msg(log_fatal, "can't load config");
return EXIT_FAILURE;
}
if (config.backends == NULL) {
f2b_log_msg(log_fatal, "no backends found in config");
return EXIT_FAILURE;
} else {
section = config.backends;
}
if ((backend = f2b_backend_create(section, argv[2])) == NULL) {
f2b_log_msg(log_fatal, "can't create backend '%s' with id '%s'", section->name, argv[2]);
return EXIT_FAILURE;
}
if (!f2b_backend_start(backend)) {
f2b_log_msg(log_error, "action 'ban' failed: %s", f2b_backend_error(backend));
goto cleanup;
}
fputs("usage: <cmd> <ipaddr>\n", stdout);
fputs("press Ctrl-D for exit\n", stdout);
while (1) {
fputs("test >> ", stdout);
if (!fgets(line, sizeof(line) - 1, stdin)) {
if (feof(stdin)) {
fputc('\n', stdout);
} else {
fputs("read error\n", stdout);
}
break;
}
if (line[0] == '\n')
continue;
switch (parse_input(line, addr, sizeof(addr))) {
case 0 : handler = f2b_backend_ban; break;
case 1 : handler = f2b_backend_check; break;
case 2 : handler = f2b_backend_unban; break;
default :
f2b_log_msg(log_error, "can't parse input");
continue;
break;
} /* switch */
if (handler(backend, addr)) {
fputs("ok\n", stdout);
} else {
fputs("failure: ", stdout);
fputs(f2b_backend_error(backend), stdout);
fputc('\n', stdout);
}
} /* while */
cleanup:
f2b_backend_stop(backend);
f2b_backend_destroy(backend);
f2b_config_free(&config);
return EXIT_SUCCESS;
}