mirror of
https://github.com/nmap/nmap.git
synced 2025-12-15 04:09:01 +00:00
b55ff2d68f
Make current loglevel and current log callback global to the library. Attaching them to the nsock pool doesn't bring any benefit and prevents from logging activity in code sections that don't have access to a pool (such as proxy chain specification parsing). Updated external calls and nsock tests accordingly.
176 lines
3.4 KiB
C
176 lines
3.4 KiB
C
/*
|
|
* Nsock regression test suite
|
|
* Same license as nmap -- see http://nmap.org/book/man-legal.html
|
|
*/
|
|
|
|
#include "test-common.h"
|
|
|
|
|
|
struct log_test_data {
|
|
nsock_pool nsp;
|
|
nsock_loglevel_t current_level;
|
|
unsigned int got_dbgfull: 1;
|
|
unsigned int got_dbg: 1;
|
|
unsigned int got_info: 1;
|
|
unsigned int got_error: 1;
|
|
unsigned int total;
|
|
int errcode;
|
|
};
|
|
|
|
static struct log_test_data *GlobalLTD;
|
|
|
|
static void log_handler(const struct nsock_log_rec *rec) {
|
|
GlobalLTD->total++;
|
|
switch(rec->level) {
|
|
case NSOCK_LOG_DBG_ALL:
|
|
GlobalLTD->got_dbgfull = 1;
|
|
break;
|
|
|
|
case NSOCK_LOG_DBG:
|
|
GlobalLTD->got_dbg = 1;
|
|
break;
|
|
|
|
case NSOCK_LOG_INFO:
|
|
GlobalLTD->got_info = 1;
|
|
break;
|
|
|
|
case NSOCK_LOG_ERROR:
|
|
GlobalLTD->got_error = 1;
|
|
break;
|
|
|
|
default:
|
|
fprintf(stderr, "UNEXPECTED LOG LEVEL (%d)!\n", (int)rec->level);
|
|
GlobalLTD->errcode = -EINVAL;
|
|
}
|
|
}
|
|
|
|
static void nop_handler(nsock_pool nsp, nsock_event nse, void *udata) {
|
|
}
|
|
|
|
static int check_loglevel(struct log_test_data *ltd, nsock_loglevel_t level) {
|
|
int rc = 0;
|
|
nsock_event_id id;
|
|
|
|
nsock_set_loglevel(level);
|
|
|
|
ltd->current_level = level;
|
|
|
|
ltd->got_dbgfull = 0;
|
|
ltd->got_dbg = 0;
|
|
ltd->got_info = 0;
|
|
ltd->got_error = 0;
|
|
|
|
ltd->total = 0;
|
|
ltd->errcode = 0;
|
|
|
|
id = nsock_timer_create(ltd->nsp, nop_handler, 200, NULL);
|
|
nsock_event_cancel(ltd->nsp, id, 0);
|
|
|
|
if (ltd->errcode)
|
|
return ltd->errcode;
|
|
|
|
if (ltd->total < 1)
|
|
return -EINVAL;
|
|
|
|
return rc;
|
|
}
|
|
|
|
static int check_errlevel(struct log_test_data *ltd, nsock_loglevel_t level) {
|
|
nsock_event_id id;
|
|
|
|
nsock_set_loglevel(level);
|
|
|
|
ltd->current_level = level;
|
|
|
|
ltd->got_dbgfull = 0;
|
|
ltd->got_dbg = 0;
|
|
ltd->got_info = 0;
|
|
ltd->got_error = 0;
|
|
|
|
ltd->total = 0;
|
|
ltd->errcode = 0;
|
|
|
|
id = nsock_timer_create(ltd->nsp, nop_handler, 200, NULL);
|
|
nsock_event_cancel(ltd->nsp, id, 0);
|
|
|
|
if (ltd->errcode)
|
|
return ltd->errcode;
|
|
|
|
if (ltd->total > 0)
|
|
return -EINVAL;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int log_setup(void **tdata) {
|
|
struct log_test_data *ltd;
|
|
|
|
ltd = calloc(1, sizeof(struct log_test_data));
|
|
if (ltd == NULL)
|
|
return -ENOMEM;
|
|
|
|
ltd->nsp = nsock_pool_new(ltd);
|
|
AssertNonNull(ltd->nsp);
|
|
|
|
*tdata = GlobalLTD = ltd;
|
|
return 0;
|
|
}
|
|
|
|
static int log_teardown(void *tdata) {
|
|
struct log_test_data *ltd = (struct log_test_data *)tdata;
|
|
|
|
if (tdata) {
|
|
nsock_pool_delete(ltd->nsp);
|
|
free(tdata);
|
|
}
|
|
GlobalLTD = NULL;
|
|
return 0;
|
|
}
|
|
|
|
static int log_check_std_levels(void *tdata) {
|
|
struct log_test_data *ltd = (struct log_test_data *)tdata;
|
|
nsock_loglevel_t lvl;
|
|
int rc = 0;
|
|
|
|
nsock_set_log_function(log_handler);
|
|
|
|
for (lvl = NSOCK_LOG_DBG_ALL; lvl < NSOCK_LOG_ERROR; lvl++) {
|
|
rc = check_loglevel(ltd, lvl);
|
|
if (rc)
|
|
return rc;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int log_check_err_levels(void *tdata) {
|
|
struct log_test_data *ltd = (struct log_test_data *)tdata;
|
|
nsock_loglevel_t lvl;
|
|
int rc = 0;
|
|
|
|
nsock_set_log_function(log_handler);
|
|
|
|
for (lvl = NSOCK_LOG_ERROR; lvl <= NSOCK_LOG_NONE; lvl++) {
|
|
rc = check_errlevel(ltd, NSOCK_LOG_ERROR);
|
|
if (rc)
|
|
return rc;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
const struct test_case TestLogLevels = {
|
|
.t_name = "set standard log levels",
|
|
.t_setup = log_setup,
|
|
.t_run = log_check_std_levels,
|
|
.t_teardown = log_teardown
|
|
};
|
|
|
|
const struct test_case TestErrLevels = {
|
|
.t_name = "check error log levels",
|
|
.t_setup = log_setup,
|
|
.t_run = log_check_err_levels,
|
|
.t_teardown = log_teardown
|
|
};
|