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.
135 lines
3.2 KiB
C
135 lines
3.2 KiB
C
/*
|
|
* Nsock regression test suite
|
|
* Same license as nmap -- see http://nmap.org/book/man-legal.html
|
|
*/
|
|
|
|
#include "test-common.h"
|
|
|
|
|
|
struct basic_test_data {
|
|
nsock_pool nsp;
|
|
};
|
|
|
|
|
|
static void cancel_handler(nsock_pool nsp, nsock_event nse, void *udata) {
|
|
int *ev_done = (int *)udata;
|
|
|
|
if (nse_status(nse) == NSE_STATUS_CANCELLED)
|
|
*ev_done = 1;
|
|
}
|
|
|
|
static int cancel_setup(void **tdata) {
|
|
struct basic_test_data *btd;
|
|
|
|
btd = calloc(1, sizeof(struct basic_test_data));
|
|
if (btd == NULL)
|
|
return -ENOMEM;
|
|
|
|
btd->nsp = nsock_pool_new(NULL);
|
|
|
|
*tdata = btd;
|
|
return 0;
|
|
}
|
|
|
|
static int cancel_teardown(void *tdata) {
|
|
struct basic_test_data *btd = (struct basic_test_data *)tdata;
|
|
|
|
if (tdata) {
|
|
nsock_pool_delete(btd->nsp);
|
|
free(tdata);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int cancel_tcp_run(void *tdata) {
|
|
struct basic_test_data *btd = (struct basic_test_data *)tdata;
|
|
struct sockaddr_in peer;
|
|
nsock_iod iod;
|
|
nsock_event_id id;
|
|
int done = 0;
|
|
|
|
iod = nsock_iod_new(btd->nsp, NULL);
|
|
AssertNonNull(iod);
|
|
|
|
memset(&peer, 0, sizeof(peer));
|
|
peer.sin_family = AF_INET;
|
|
inet_aton("127.0.0.1", &peer.sin_addr);
|
|
|
|
id = nsock_connect_tcp(btd->nsp, iod, cancel_handler, 4000, (void *)&done,
|
|
(struct sockaddr *)&peer, sizeof(peer), PORT_TCP);
|
|
nsock_event_cancel(btd->nsp, id, 1);
|
|
|
|
nsock_iod_delete(iod, NSOCK_PENDING_SILENT);
|
|
|
|
return (done == 1) ? 0 : -ENOEXEC;
|
|
}
|
|
|
|
static int cancel_udp_run(void *tdata) {
|
|
struct basic_test_data *btd = (struct basic_test_data *)tdata;
|
|
struct sockaddr_in peer;
|
|
nsock_iod iod;
|
|
nsock_event_id id;
|
|
int done = 0;
|
|
|
|
iod = nsock_iod_new(btd->nsp, NULL);
|
|
AssertNonNull(iod);
|
|
|
|
memset(&peer, 0, sizeof(peer));
|
|
peer.sin_family = AF_INET;
|
|
inet_aton("127.0.0.1", &peer.sin_addr);
|
|
|
|
id = nsock_connect_udp(btd->nsp, iod, cancel_handler, (void *)&done,
|
|
(struct sockaddr *)&peer, sizeof(peer), PORT_UDP);
|
|
nsock_event_cancel(btd->nsp, id, 1);
|
|
|
|
nsock_iod_delete(iod, NSOCK_PENDING_SILENT);
|
|
|
|
return (done == 1) ? 0 : -ENOEXEC;
|
|
}
|
|
|
|
static int cancel_ssl_run(void *tdata) {
|
|
struct basic_test_data *btd = (struct basic_test_data *)tdata;
|
|
struct sockaddr_in peer;
|
|
nsock_iod iod;
|
|
nsock_event_id id;
|
|
int done = 0;
|
|
|
|
iod = nsock_iod_new(btd->nsp, NULL);
|
|
AssertNonNull(iod);
|
|
|
|
memset(&peer, 0, sizeof(peer));
|
|
peer.sin_family = AF_INET;
|
|
inet_aton("127.0.0.1", &peer.sin_addr);
|
|
|
|
id = nsock_connect_ssl(btd->nsp, iod, cancel_handler, 4000, (void *)&done,
|
|
(struct sockaddr *)&peer, sizeof(peer), IPPROTO_TCP,
|
|
PORT_TCPSSL, NULL);
|
|
nsock_event_cancel(btd->nsp, id, 1);
|
|
|
|
nsock_iod_delete(iod, NSOCK_PENDING_SILENT);
|
|
|
|
return (done == 1) ? 0 : -ENOEXEC;
|
|
}
|
|
|
|
|
|
const struct test_case TestCancelTCP = {
|
|
.t_name = "schedule and cancel TCP connect",
|
|
.t_setup = cancel_setup,
|
|
.t_run = cancel_tcp_run,
|
|
.t_teardown = cancel_teardown
|
|
};
|
|
|
|
const struct test_case TestCancelUDP = {
|
|
.t_name = "schedule and cancel UDP pseudo-connect",
|
|
.t_setup = cancel_setup,
|
|
.t_run = cancel_udp_run,
|
|
.t_teardown = cancel_teardown
|
|
};
|
|
|
|
const struct test_case TestCancelSSL = {
|
|
.t_name = "schedule and cancel SSL connect",
|
|
.t_setup = cancel_setup,
|
|
.t_run = cancel_ssl_run,
|
|
.t_teardown = cancel_teardown
|
|
};
|