2011-07-28 04:04:37 +00:00
|
|
|
/*
|
2018-02-23 09:53:12 +01:00
|
|
|
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
2011-07-28 04:04:37 +00:00
|
|
|
*
|
2016-06-27 14:56:38 +10:00
|
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
|
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
2020-09-14 16:20:40 -07:00
|
|
|
* file, you can obtain one at https://mozilla.org/MPL/2.0/.
|
2018-02-23 09:53:12 +01:00
|
|
|
*
|
|
|
|
|
* See the COPYRIGHT file distributed with this work for additional
|
|
|
|
|
* information regarding copyright ownership.
|
2011-07-28 04:04:37 +00:00
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
/*! \file */
|
|
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
#if HAVE_CMOCKA
|
Include <sched.h> where necessary for musl libc
All unit tests define the UNIT_TESTING macro, which causes <cmocka.h> to
replace malloc(), calloc(), realloc(), and free() with its own functions
tracking memory allocations. In order for this not to break
compilation, the system header declaring the prototypes for these
standard functions must be included before <cmocka.h>.
Normally, these prototypes are only present in <stdlib.h>, so we make
sure it is included before <cmocka.h>. However, musl libc also defines
the prototypes for calloc() and free() in <sched.h>, which is included
by <pthread.h>, which is included e.g. by <isc/mutex.h>. Thus, unit
tests including "dnstest.h" (which includes <isc/mem.h>, which includes
<isc/mutex.h>) after <cmocka.h> will not compile with musl libc as for
these programs, <sched.h> will be included after <cmocka.h>.
Always including <cmocka.h> after all other header files is not a
feasible solution as that causes the mock assertion macros defined in
<isc/util.h> to mangle the contents of <cmocka.h>, thus breaking
compilation. We cannot really use the __noreturn__ or analyzer_noreturn
attributes with cmocka assertion functions because they do return if the
tested condition is true. The problem is that what BIND unit tests do
is incompatible with Clang Static Analyzer's assumptions: since we use
cmocka, our custom assertion handlers are present in a shared library
(i.e. it is the cmocka library that checks the assertion condition, not
a macro in unit test code). Redefining cmocka's assertion macros in
<isc/util.h> is an ugly hack to overcome that problem - unfortunately,
this is the only way we can think of to make Clang Static Analyzer
properly process unit test code. Giving up on Clang Static Analyzer
being able to properly process unit test code is not a satisfactory
solution.
Undefining _GNU_SOURCE for unit test code could work around the problem
(musl libc's <sched.h> only defines the prototypes for calloc() and
free() when _GNU_SOURCE is defined), but doing that could introduce
discrepancies for unit tests including entire *.c files, so it is also
not a good solution.
All in all, including <sched.h> before <cmocka.h> for all affected unit
tests seems to be the most benign way of working around this musl libc
quirk. While quite an ugly solution, it achieves our goals here, which
are to keep the benefit of proper static analysis of unit test code and
to fix compilation against musl libc.
2019-07-30 21:08:40 +02:00
|
|
|
#include <sched.h> /* IWYU pragma: keep */
|
2020-02-12 13:59:18 +01:00
|
|
|
#include <setjmp.h>
|
|
|
|
|
#include <stdarg.h>
|
2018-04-17 08:29:14 -07:00
|
|
|
#include <stdbool.h>
|
2020-02-12 13:59:18 +01:00
|
|
|
#include <stddef.h>
|
2018-10-26 17:07:50 +08:00
|
|
|
#include <stdlib.h>
|
2011-07-28 04:04:37 +00:00
|
|
|
#include <time.h>
|
Include <sched.h> where necessary for musl libc
All unit tests define the UNIT_TESTING macro, which causes <cmocka.h> to
replace malloc(), calloc(), realloc(), and free() with its own functions
tracking memory allocations. In order for this not to break
compilation, the system header declaring the prototypes for these
standard functions must be included before <cmocka.h>.
Normally, these prototypes are only present in <stdlib.h>, so we make
sure it is included before <cmocka.h>. However, musl libc also defines
the prototypes for calloc() and free() in <sched.h>, which is included
by <pthread.h>, which is included e.g. by <isc/mutex.h>. Thus, unit
tests including "dnstest.h" (which includes <isc/mem.h>, which includes
<isc/mutex.h>) after <cmocka.h> will not compile with musl libc as for
these programs, <sched.h> will be included after <cmocka.h>.
Always including <cmocka.h> after all other header files is not a
feasible solution as that causes the mock assertion macros defined in
<isc/util.h> to mangle the contents of <cmocka.h>, thus breaking
compilation. We cannot really use the __noreturn__ or analyzer_noreturn
attributes with cmocka assertion functions because they do return if the
tested condition is true. The problem is that what BIND unit tests do
is incompatible with Clang Static Analyzer's assumptions: since we use
cmocka, our custom assertion handlers are present in a shared library
(i.e. it is the cmocka library that checks the assertion condition, not
a macro in unit test code). Redefining cmocka's assertion macros in
<isc/util.h> is an ugly hack to overcome that problem - unfortunately,
this is the only way we can think of to make Clang Static Analyzer
properly process unit test code. Giving up on Clang Static Analyzer
being able to properly process unit test code is not a satisfactory
solution.
Undefining _GNU_SOURCE for unit test code could work around the problem
(musl libc's <sched.h> only defines the prototypes for calloc() and
free() when _GNU_SOURCE is defined), but doing that could introduce
discrepancies for unit tests including entire *.c files, so it is also
not a good solution.
All in all, including <sched.h> before <cmocka.h> for all affected unit
tests seems to be the most benign way of working around this musl libc
quirk. While quite an ugly solution, it achieves our goals here, which
are to keep the benefit of proper static analysis of unit test code and
to fix compilation against musl libc.
2019-07-30 21:08:40 +02:00
|
|
|
#include <unistd.h>
|
2011-07-28 04:04:37 +00:00
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
#define UNIT_TESTING
|
|
|
|
|
#include <cmocka.h>
|
|
|
|
|
|
2019-07-04 13:51:47 +02:00
|
|
|
#include <isc/atomic.h>
|
2015-12-15 11:58:28 -08:00
|
|
|
#include <isc/platform.h>
|
2020-02-12 13:59:18 +01:00
|
|
|
#include <isc/print.h>
|
2011-07-28 04:04:37 +00:00
|
|
|
#include <isc/socket.h>
|
2017-09-08 13:39:09 -07:00
|
|
|
#include <isc/task.h>
|
2011-07-28 04:04:37 +00:00
|
|
|
|
2021-04-27 00:07:43 +02:00
|
|
|
#include "../socket_p.h"
|
2011-07-28 04:04:37 +00:00
|
|
|
#include "isctest.h"
|
|
|
|
|
|
2020-02-13 14:44:37 -08:00
|
|
|
static bool recv_dscp;
|
2013-03-22 12:27:54 -07:00
|
|
|
static unsigned int recv_dscp_value;
|
2020-02-13 14:44:37 -08:00
|
|
|
static bool recv_trunc;
|
2020-05-14 09:39:57 +02:00
|
|
|
isc_socket_t *s1 = NULL, *s2 = NULL, *s3 = NULL;
|
|
|
|
|
isc_task_t *test_task = NULL;
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2011-07-28 04:04:37 +00:00
|
|
|
/*
|
|
|
|
|
* Helper functions
|
|
|
|
|
*/
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
static int
|
2020-02-13 14:44:37 -08:00
|
|
|
_setup(void **state) {
|
2018-10-26 17:07:50 +08:00
|
|
|
isc_result_t result;
|
|
|
|
|
|
|
|
|
|
UNUSED(state);
|
|
|
|
|
|
|
|
|
|
result = isc_test_begin(NULL, true, 0);
|
|
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
|
|
|
|
|
return (0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static int
|
2020-02-13 14:44:37 -08:00
|
|
|
_teardown(void **state) {
|
2018-10-26 17:07:50 +08:00
|
|
|
UNUSED(state);
|
|
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
if (s1 != NULL) {
|
|
|
|
|
isc_socket_detach(&s1);
|
|
|
|
|
}
|
|
|
|
|
if (s2 != NULL) {
|
|
|
|
|
isc_socket_detach(&s2);
|
|
|
|
|
}
|
|
|
|
|
if (s3 != NULL) {
|
|
|
|
|
isc_socket_detach(&s3);
|
|
|
|
|
}
|
|
|
|
|
if (test_task != NULL) {
|
|
|
|
|
isc_task_detach(&test_task);
|
|
|
|
|
}
|
|
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
isc_test_end();
|
|
|
|
|
|
|
|
|
|
return (0);
|
|
|
|
|
}
|
|
|
|
|
|
2011-07-28 04:04:37 +00:00
|
|
|
typedef struct {
|
2020-02-13 14:44:37 -08:00
|
|
|
atomic_bool done;
|
|
|
|
|
isc_result_t result;
|
2013-03-22 12:27:54 -07:00
|
|
|
isc_socket_t *socket;
|
2011-07-28 04:04:37 +00:00
|
|
|
} completion_t;
|
|
|
|
|
|
|
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
completion_init(completion_t *completion) {
|
2020-02-20 11:39:15 +01:00
|
|
|
atomic_init(&completion->done, false);
|
2013-03-22 12:27:54 -07:00
|
|
|
completion->socket = NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
accept_done(isc_task_t *task, isc_event_t *event) {
|
2013-03-22 12:27:54 -07:00
|
|
|
isc_socket_newconnev_t *nevent = (isc_socket_newconnev_t *)event;
|
2020-02-13 14:44:37 -08:00
|
|
|
completion_t *completion = event->ev_arg;
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
UNUSED(task);
|
|
|
|
|
|
|
|
|
|
completion->result = nevent->result;
|
2019-07-04 13:51:47 +02:00
|
|
|
atomic_store(&completion->done, true);
|
2018-10-26 17:07:50 +08:00
|
|
|
if (completion->result == ISC_R_SUCCESS) {
|
2013-03-22 12:27:54 -07:00
|
|
|
completion->socket = nevent->newsocket;
|
2018-10-26 17:07:50 +08:00
|
|
|
}
|
2013-03-23 23:46:06 +00:00
|
|
|
|
2013-03-22 12:27:54 -07:00
|
|
|
isc_event_free(&event);
|
2011-07-28 04:04:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
event_done(isc_task_t *task, isc_event_t *event) {
|
|
|
|
|
isc_socketevent_t *sev = NULL;
|
2018-11-19 09:43:23 +00:00
|
|
|
isc_socket_connev_t *connev = NULL;
|
2020-02-13 14:44:37 -08:00
|
|
|
completion_t *completion = event->ev_arg;
|
2011-07-28 04:04:37 +00:00
|
|
|
UNUSED(task);
|
|
|
|
|
|
2018-11-19 09:43:23 +00:00
|
|
|
switch (event->ev_type) {
|
|
|
|
|
case ISC_SOCKEVENT_RECVDONE:
|
|
|
|
|
case ISC_SOCKEVENT_SENDDONE:
|
2020-02-12 13:59:18 +01:00
|
|
|
sev = (isc_socketevent_t *)event;
|
2018-11-19 09:43:23 +00:00
|
|
|
completion->result = sev->result;
|
|
|
|
|
if ((sev->attributes & ISC_SOCKEVENTATTR_DSCP) != 0) {
|
|
|
|
|
recv_dscp = true;
|
2020-02-12 13:59:18 +01:00
|
|
|
recv_dscp_value = sev->dscp;
|
2018-11-19 09:43:23 +00:00
|
|
|
} else {
|
|
|
|
|
recv_dscp = false;
|
|
|
|
|
}
|
|
|
|
|
recv_trunc = ((sev->attributes & ISC_SOCKEVENTATTR_TRUNC) != 0);
|
|
|
|
|
break;
|
|
|
|
|
case ISC_SOCKEVENT_CONNECT:
|
2020-02-12 13:59:18 +01:00
|
|
|
connev = (isc_socket_connev_t *)event;
|
2018-11-19 09:43:23 +00:00
|
|
|
completion->result = connev->result;
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
assert_false(true);
|
2018-05-01 16:52:01 +10:00
|
|
|
}
|
2019-07-04 13:51:47 +02:00
|
|
|
atomic_store(&completion->done, true);
|
2011-07-28 04:04:37 +00:00
|
|
|
isc_event_free(&event);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static isc_result_t
|
2020-02-13 14:44:37 -08:00
|
|
|
waitfor(completion_t *completion) {
|
2011-07-28 04:04:37 +00:00
|
|
|
int i = 0;
|
2019-07-04 13:51:47 +02:00
|
|
|
while (!atomic_load(&completion->done) && i++ < 5000) {
|
2011-09-02 21:15:39 +00:00
|
|
|
isc_test_nap(1000);
|
2011-07-28 04:04:37 +00:00
|
|
|
}
|
2019-07-04 13:51:47 +02:00
|
|
|
if (atomic_load(&completion->done)) {
|
2011-07-28 04:04:37 +00:00
|
|
|
return (ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
}
|
|
|
|
|
return (ISC_R_FAILURE);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
waitbody(void) {
|
2020-02-12 13:59:18 +01:00
|
|
|
isc_test_nap(1000);
|
2013-03-22 12:27:54 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static isc_result_t
|
2020-02-13 14:44:37 -08:00
|
|
|
waitfor2(completion_t *c1, completion_t *c2) {
|
2013-03-22 12:27:54 -07:00
|
|
|
int i = 0;
|
|
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
while (!(atomic_load(&c1->done) && atomic_load(&c2->done)) &&
|
|
|
|
|
i++ < 5000) {
|
2013-03-22 12:27:54 -07:00
|
|
|
waitbody();
|
|
|
|
|
}
|
2019-07-04 13:51:47 +02:00
|
|
|
if (atomic_load(&c1->done) && atomic_load(&c2->done)) {
|
2013-03-22 12:27:54 -07:00
|
|
|
return (ISC_R_SUCCESS);
|
2018-10-26 17:07:50 +08:00
|
|
|
}
|
2013-03-22 12:27:54 -07:00
|
|
|
return (ISC_R_FAILURE);
|
|
|
|
|
}
|
|
|
|
|
|
2011-07-28 04:04:37 +00:00
|
|
|
/*
|
|
|
|
|
* Individual unit tests
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
/* Test UDP sendto/recv (IPv4) */
|
2018-10-26 17:07:50 +08:00
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
udp_sendto_test(void **state) {
|
|
|
|
|
isc_result_t result;
|
2011-07-28 04:04:37 +00:00
|
|
|
isc_sockaddr_t addr1, addr2;
|
|
|
|
|
struct in_addr in;
|
2020-02-13 14:44:37 -08:00
|
|
|
char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
|
|
|
|
|
completion_t completion;
|
|
|
|
|
isc_region_t r;
|
2011-07-28 04:04:37 +00:00
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
UNUSED(state);
|
2011-07-28 04:04:37 +00:00
|
|
|
|
|
|
|
|
in.s_addr = inet_addr("127.0.0.1");
|
2015-12-15 12:50:32 +11:00
|
|
|
isc_sockaddr_fromin(&addr1, &in, 0);
|
|
|
|
|
isc_sockaddr_fromin(&addr2, &in, 0);
|
2011-07-28 04:04:37 +00:00
|
|
|
|
|
|
|
|
result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_bind(s1, &addr1, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_getsockname(s1, &addr1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr1) != 0);
|
2011-07-28 04:04:37 +00:00
|
|
|
|
|
|
|
|
result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_bind(s2, &addr2, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_getsockname(s2, &addr2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr2) != 0);
|
2011-07-28 23:47:59 +00:00
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_task_create(taskmgr, 0, &test_task);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2011-07-28 04:04:37 +00:00
|
|
|
|
2017-09-13 00:14:37 -07:00
|
|
|
snprintf(sendbuf, sizeof(sendbuf), "Hello");
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)sendbuf;
|
2011-07-28 04:04:37 +00:00
|
|
|
r.length = strlen(sendbuf) + 1;
|
|
|
|
|
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_sendto(s1, &r, test_task, event_done, &completion,
|
2011-07-28 04:04:37 +00:00
|
|
|
&addr2, NULL);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2011-07-28 04:04:37 +00:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
2011-07-28 04:04:37 +00:00
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)recvbuf;
|
2011-07-28 04:04:37 +00:00
|
|
|
r.length = BUFSIZ;
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_recv(s2, &r, 1, test_task, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2011-07-28 04:04:37 +00:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
|
|
|
|
assert_string_equal(recvbuf, "Hello");
|
2011-07-28 04:04:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Test UDP sendto/recv with duplicated socket */
|
2018-10-26 17:07:50 +08:00
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
udp_dup_test(void **state) {
|
|
|
|
|
isc_result_t result;
|
2011-07-28 04:04:37 +00:00
|
|
|
isc_sockaddr_t addr1, addr2;
|
|
|
|
|
struct in_addr in;
|
2020-02-13 14:44:37 -08:00
|
|
|
char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
|
|
|
|
|
completion_t completion;
|
|
|
|
|
isc_region_t r;
|
2011-07-28 04:04:37 +00:00
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
UNUSED(state);
|
2011-07-28 04:04:37 +00:00
|
|
|
|
|
|
|
|
in.s_addr = inet_addr("127.0.0.1");
|
2015-12-15 12:50:32 +11:00
|
|
|
isc_sockaddr_fromin(&addr1, &in, 0);
|
|
|
|
|
isc_sockaddr_fromin(&addr2, &in, 0);
|
2011-07-28 04:04:37 +00:00
|
|
|
|
|
|
|
|
result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_bind(s1, &addr1, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_getsockname(s1, &addr1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr1) != 0);
|
2011-07-28 04:04:37 +00:00
|
|
|
|
|
|
|
|
result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_bind(s2, &addr2, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_getsockname(s2, &addr2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr2) != 0);
|
2011-11-29 01:03:47 +00:00
|
|
|
|
|
|
|
|
result = isc_socket_dup(s2, &s3);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2011-07-28 23:47:59 +00:00
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_task_create(taskmgr, 0, &test_task);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2011-07-28 04:04:37 +00:00
|
|
|
|
2017-09-13 00:14:37 -07:00
|
|
|
snprintf(sendbuf, sizeof(sendbuf), "Hello");
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)sendbuf;
|
2011-07-28 04:04:37 +00:00
|
|
|
r.length = strlen(sendbuf) + 1;
|
|
|
|
|
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_sendto(s1, &r, test_task, event_done, &completion,
|
2011-07-28 04:04:37 +00:00
|
|
|
&addr2, NULL);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2011-07-28 04:04:37 +00:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
2011-07-28 04:04:37 +00:00
|
|
|
|
2017-09-13 00:14:37 -07:00
|
|
|
snprintf(sendbuf, sizeof(sendbuf), "World");
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)sendbuf;
|
2011-07-28 04:04:37 +00:00
|
|
|
r.length = strlen(sendbuf) + 1;
|
|
|
|
|
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_sendto(s1, &r, test_task, event_done, &completion,
|
2011-07-28 04:04:37 +00:00
|
|
|
&addr2, NULL);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2011-07-28 04:04:37 +00:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
2011-07-28 04:04:37 +00:00
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)recvbuf;
|
2011-07-28 04:04:37 +00:00
|
|
|
r.length = BUFSIZ;
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_recv(s2, &r, 1, test_task, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2011-07-28 04:04:37 +00:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
|
|
|
|
assert_string_equal(recvbuf, "Hello");
|
2011-07-28 04:04:37 +00:00
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)recvbuf;
|
2011-07-28 04:04:37 +00:00
|
|
|
r.length = BUFSIZ;
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_recv(s3, &r, 1, test_task, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2011-07-28 04:04:37 +00:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
|
|
|
|
assert_string_equal(recvbuf, "World");
|
2011-07-28 04:04:37 +00:00
|
|
|
}
|
|
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
/* Test UDP sendto/recv (IPv4) */
|
|
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
udp_dscp_v4_test(void **state) {
|
|
|
|
|
isc_result_t result;
|
|
|
|
|
isc_sockaddr_t addr1, addr2;
|
|
|
|
|
struct in_addr in;
|
|
|
|
|
char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
|
|
|
|
|
completion_t completion;
|
|
|
|
|
isc_region_t r;
|
2013-03-22 12:27:54 -07:00
|
|
|
isc_socketevent_t *socketevent;
|
|
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
UNUSED(state);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
in.s_addr = inet_addr("127.0.0.1");
|
2015-12-15 12:50:32 +11:00
|
|
|
isc_sockaddr_fromin(&addr1, &in, 0);
|
|
|
|
|
isc_sockaddr_fromin(&addr2, &in, 0);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
result = isc_socket_bind(s1, &addr1, ISC_SOCKET_REUSEADDRESS);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_getsockname(s1, &addr1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr1) != 0);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
result = isc_socket_bind(s2, &addr2, ISC_SOCKET_REUSEADDRESS);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_getsockname(s2, &addr2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr2) != 0);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_task_create(taskmgr, 0, &test_task);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2017-09-13 00:14:37 -07:00
|
|
|
snprintf(sendbuf, sizeof(sendbuf), "Hello");
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)sendbuf;
|
2013-03-22 12:27:54 -07:00
|
|
|
r.length = strlen(sendbuf) + 1;
|
|
|
|
|
|
|
|
|
|
completion_init(&completion);
|
2013-03-23 23:46:06 +00:00
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
socketevent = isc_socket_socketevent(
|
|
|
|
|
test_mctx, s1, ISC_SOCKEVENT_SENDDONE, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_non_null(socketevent);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
if ((isc_net_probedscp() & ISC_NET_DSCPPKTV4) != 0) {
|
|
|
|
|
socketevent->dscp = 056; /* EF */
|
|
|
|
|
socketevent->attributes |= ISC_SOCKEVENTATTR_DSCP;
|
|
|
|
|
} else if ((isc_net_probedscp() & ISC_NET_DSCPSETV4) != 0) {
|
2020-02-12 13:59:18 +01:00
|
|
|
isc_socket_dscp(s1, 056); /* EF */
|
2013-03-22 12:27:54 -07:00
|
|
|
socketevent->dscp = 0;
|
|
|
|
|
socketevent->attributes &= ~ISC_SOCKEVENTATTR_DSCP;
|
|
|
|
|
}
|
|
|
|
|
|
2018-04-17 08:29:14 -07:00
|
|
|
recv_dscp = false;
|
2013-03-22 12:27:54 -07:00
|
|
|
recv_dscp_value = 0;
|
|
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_sendto2(s1, &r, test_task, &addr2, NULL,
|
|
|
|
|
socketevent, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)recvbuf;
|
2013-03-22 12:27:54 -07:00
|
|
|
r.length = BUFSIZ;
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_recv(s2, &r, 1, test_task, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
|
|
|
|
assert_string_equal(recvbuf, "Hello");
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
if ((isc_net_probedscp() & ISC_NET_DSCPRECVV4) != 0) {
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_true(recv_dscp);
|
|
|
|
|
assert_int_equal(recv_dscp_value, 056);
|
|
|
|
|
} else {
|
|
|
|
|
assert_false(recv_dscp);
|
|
|
|
|
}
|
2013-03-22 12:27:54 -07:00
|
|
|
}
|
|
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
/* Test UDP sendto/recv (IPv6) */
|
|
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
udp_dscp_v6_test(void **state) {
|
|
|
|
|
isc_result_t result;
|
|
|
|
|
isc_sockaddr_t addr1, addr2;
|
|
|
|
|
struct in6_addr in6;
|
|
|
|
|
char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
|
|
|
|
|
completion_t completion;
|
|
|
|
|
isc_region_t r;
|
2013-03-22 12:27:54 -07:00
|
|
|
isc_socketevent_t *socketevent;
|
2020-02-13 14:44:37 -08:00
|
|
|
int n;
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
UNUSED(state);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2013-04-11 17:07:50 +10:00
|
|
|
n = inet_pton(AF_INET6, "::1", &in6.s6_addr);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_true(n == 1);
|
2015-12-15 12:50:32 +11:00
|
|
|
isc_sockaddr_fromin6(&addr1, &in6, 0);
|
|
|
|
|
isc_sockaddr_fromin6(&addr2, &in6, 0);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-02-13 14:44:37 -08:00
|
|
|
result = isc_socket_create(socketmgr, PF_INET6, isc_sockettype_udp,
|
|
|
|
|
&s1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_bind(s1, &addr1, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_getsockname(s1, &addr1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr1) != 0);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-02-13 14:44:37 -08:00
|
|
|
result = isc_socket_create(socketmgr, PF_INET6, isc_sockettype_udp,
|
|
|
|
|
&s2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_bind(s2, &addr2, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_getsockname(s2, &addr2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr2) != 0);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_task_create(taskmgr, 0, &test_task);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2017-09-13 00:14:37 -07:00
|
|
|
snprintf(sendbuf, sizeof(sendbuf), "Hello");
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)sendbuf;
|
2013-03-22 12:27:54 -07:00
|
|
|
r.length = strlen(sendbuf) + 1;
|
|
|
|
|
|
|
|
|
|
completion_init(&completion);
|
2013-03-23 23:46:06 +00:00
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
socketevent = isc_socket_socketevent(
|
|
|
|
|
test_mctx, s1, ISC_SOCKEVENT_SENDDONE, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_non_null(socketevent);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
if ((isc_net_probedscp() & ISC_NET_DSCPPKTV6) != 0) {
|
|
|
|
|
socketevent->dscp = 056; /* EF */
|
|
|
|
|
socketevent->attributes = ISC_SOCKEVENTATTR_DSCP;
|
2018-10-26 17:07:50 +08:00
|
|
|
} else if ((isc_net_probedscp() & ISC_NET_DSCPSETV6) != 0) {
|
2020-02-12 13:59:18 +01:00
|
|
|
isc_socket_dscp(s1, 056); /* EF */
|
2018-10-26 17:07:50 +08:00
|
|
|
}
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2018-04-17 08:29:14 -07:00
|
|
|
recv_dscp = false;
|
2013-03-22 12:27:54 -07:00
|
|
|
recv_dscp_value = 0;
|
|
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_sendto2(s1, &r, test_task, &addr2, NULL,
|
|
|
|
|
socketevent, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)recvbuf;
|
2013-03-22 12:27:54 -07:00
|
|
|
r.length = BUFSIZ;
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_recv(s2, &r, 1, test_task, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
|
|
|
|
assert_string_equal(recvbuf, "Hello");
|
2013-03-22 12:27:54 -07:00
|
|
|
if ((isc_net_probedscp() & ISC_NET_DSCPRECVV6) != 0) {
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_true(recv_dscp);
|
|
|
|
|
assert_int_equal(recv_dscp_value, 056);
|
|
|
|
|
} else {
|
|
|
|
|
assert_false(recv_dscp);
|
|
|
|
|
}
|
2013-03-22 12:27:54 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Test TCP sendto/recv (IPv4) */
|
2018-10-26 17:07:50 +08:00
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
tcp_dscp_v4_test(void **state) {
|
|
|
|
|
isc_result_t result;
|
2013-03-22 12:27:54 -07:00
|
|
|
isc_sockaddr_t addr1;
|
|
|
|
|
struct in_addr in;
|
2020-02-13 14:44:37 -08:00
|
|
|
char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
|
|
|
|
|
completion_t completion, completion2;
|
|
|
|
|
isc_region_t r;
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
UNUSED(state);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
in.s_addr = inet_addr("127.0.0.1");
|
2015-12-15 12:50:32 +11:00
|
|
|
isc_sockaddr_fromin(&addr1, &in, 0);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_tcp, &s1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_bind(s1, &addr1, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_getsockname(s1, &addr1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr1) != 0);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
result = isc_socket_listen(s1, 3);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_tcp, &s2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_task_create(taskmgr, 0, &test_task);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
completion_init(&completion2);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_accept(s1, test_task, accept_done, &completion2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_connect(s2, &addr1, test_task, event_done,
|
|
|
|
|
&completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
waitfor2(&completion, &completion2);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion2.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion2.result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
s3 = completion2.socket;
|
|
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
isc_socket_dscp(s2, 056); /* EF */
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2017-09-13 00:14:37 -07:00
|
|
|
snprintf(sendbuf, sizeof(sendbuf), "Hello");
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)sendbuf;
|
2013-03-22 12:27:54 -07:00
|
|
|
r.length = strlen(sendbuf) + 1;
|
|
|
|
|
|
2018-04-17 08:29:14 -07:00
|
|
|
recv_dscp = false;
|
2013-03-22 12:27:54 -07:00
|
|
|
recv_dscp_value = 0;
|
|
|
|
|
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_sendto(s2, &r, test_task, event_done, &completion,
|
|
|
|
|
NULL, NULL);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)recvbuf;
|
2013-03-22 12:27:54 -07:00
|
|
|
r.length = BUFSIZ;
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_recv(s3, &r, 1, test_task, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
|
|
|
|
assert_string_equal(recvbuf, "Hello");
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
if ((isc_net_probedscp() & ISC_NET_DSCPRECVV4) != 0) {
|
2018-10-26 17:07:50 +08:00
|
|
|
if (recv_dscp) {
|
|
|
|
|
assert_int_equal(recv_dscp_value, 056);
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
assert_false(recv_dscp);
|
|
|
|
|
}
|
2013-03-22 12:27:54 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Test TCP sendto/recv (IPv6) */
|
2018-10-26 17:07:50 +08:00
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
tcp_dscp_v6_test(void **state) {
|
|
|
|
|
isc_result_t result;
|
|
|
|
|
isc_sockaddr_t addr1;
|
2013-03-22 12:27:54 -07:00
|
|
|
struct in6_addr in6;
|
2020-02-13 14:44:37 -08:00
|
|
|
char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
|
|
|
|
|
completion_t completion, completion2;
|
|
|
|
|
isc_region_t r;
|
|
|
|
|
int n;
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
UNUSED(state);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2013-04-11 17:07:50 +10:00
|
|
|
n = inet_pton(AF_INET6, "::1", &in6.s6_addr);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_true(n == 1);
|
2015-12-15 12:50:32 +11:00
|
|
|
isc_sockaddr_fromin6(&addr1, &in6, 0);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-02-13 14:44:37 -08:00
|
|
|
result = isc_socket_create(socketmgr, PF_INET6, isc_sockettype_tcp,
|
|
|
|
|
&s1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_bind(s1, &addr1, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2015-12-15 12:50:32 +11:00
|
|
|
result = isc_socket_getsockname(s1, &addr1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr1) != 0);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
result = isc_socket_listen(s1, 3);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-02-13 14:44:37 -08:00
|
|
|
result = isc_socket_create(socketmgr, PF_INET6, isc_sockettype_tcp,
|
|
|
|
|
&s2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_task_create(taskmgr, 0, &test_task);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
completion_init(&completion2);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_accept(s1, test_task, accept_done, &completion2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_connect(s2, &addr1, test_task, event_done,
|
|
|
|
|
&completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
waitfor2(&completion, &completion2);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion2.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion2.result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
s3 = completion2.socket;
|
|
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
isc_socket_dscp(s2, 056); /* EF */
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2017-09-13 00:14:37 -07:00
|
|
|
snprintf(sendbuf, sizeof(sendbuf), "Hello");
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)sendbuf;
|
2013-03-22 12:27:54 -07:00
|
|
|
r.length = strlen(sendbuf) + 1;
|
|
|
|
|
|
2018-04-17 08:29:14 -07:00
|
|
|
recv_dscp = false;
|
2013-03-22 12:27:54 -07:00
|
|
|
recv_dscp_value = 0;
|
|
|
|
|
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_sendto(s2, &r, test_task, event_done, &completion,
|
|
|
|
|
NULL, NULL);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)recvbuf;
|
2013-03-22 12:27:54 -07:00
|
|
|
r.length = BUFSIZ;
|
|
|
|
|
completion_init(&completion);
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_recv(s3, &r, 1, test_task, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2013-03-22 12:27:54 -07:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
|
|
|
|
assert_string_equal(recvbuf, "Hello");
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
if ((isc_net_probedscp() & ISC_NET_DSCPRECVV6) != 0) {
|
|
|
|
|
/*
|
|
|
|
|
* IPV6_RECVTCLASS is undefined for TCP however
|
2015-08-27 14:11:27 -07:00
|
|
|
* if we do get it it should be the value we set.
|
2013-03-22 12:27:54 -07:00
|
|
|
*/
|
2018-10-26 17:07:50 +08:00
|
|
|
if (recv_dscp) {
|
|
|
|
|
assert_int_equal(recv_dscp_value, 056);
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
assert_false(recv_dscp);
|
|
|
|
|
}
|
2013-03-22 12:27:54 -07:00
|
|
|
}
|
|
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
/* probe dscp capabilities */
|
|
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
net_probedscp_test(void **state) {
|
2013-03-22 12:27:54 -07:00
|
|
|
unsigned int n;
|
|
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
UNUSED(state);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
n = isc_net_probedscp();
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_true((n & ~ISC_NET_DSCPALL) == 0);
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
/* ISC_NET_DSCPSETV4 MUST be set if any is set. */
|
2020-02-12 13:59:18 +01:00
|
|
|
if (n & (ISC_NET_DSCPPKTV4 | ISC_NET_DSCPRECVV4)) {
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_true((n & ISC_NET_DSCPSETV4) != 0);
|
|
|
|
|
}
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
/* ISC_NET_DSCPSETV6 MUST be set if any is set. */
|
2020-02-12 13:59:18 +01:00
|
|
|
if (n & (ISC_NET_DSCPPKTV6 | ISC_NET_DSCPRECVV6)) {
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_true((n & ISC_NET_DSCPSETV6) != 0);
|
|
|
|
|
}
|
2013-03-22 12:27:54 -07:00
|
|
|
|
|
|
|
|
#if 0
|
2020-02-13 21:48:23 +01:00
|
|
|
fprintf(stdout,"IPv4:%s%s%s\n",
|
2013-03-22 12:27:54 -07:00
|
|
|
(n & ISC_NET_DSCPSETV4) ? " set" : "none",
|
|
|
|
|
(n & ISC_NET_DSCPPKTV4) ? " packet" : "",
|
|
|
|
|
(n & ISC_NET_DSCPRECVV4) ? " receive" : "");
|
|
|
|
|
|
2020-02-13 21:48:23 +01:00
|
|
|
fprintf(stdout,"IPv6:%s%s%s\n",
|
2013-03-22 12:27:54 -07:00
|
|
|
(n & ISC_NET_DSCPSETV6) ? " set" : "none",
|
|
|
|
|
(n & ISC_NET_DSCPPKTV6) ? " packet" : "",
|
|
|
|
|
(n & ISC_NET_DSCPRECVV6) ? " receive" : "");
|
2020-02-13 21:48:23 +01:00
|
|
|
#endif /* if 0 */
|
2013-03-22 12:27:54 -07:00
|
|
|
}
|
|
|
|
|
|
2018-05-01 16:52:01 +10:00
|
|
|
/* Test UDP truncation detection */
|
2018-10-26 17:07:50 +08:00
|
|
|
static void
|
2020-02-13 14:44:37 -08:00
|
|
|
udp_trunc_test(void **state) {
|
|
|
|
|
isc_result_t result;
|
|
|
|
|
isc_sockaddr_t addr1, addr2;
|
|
|
|
|
struct in_addr in;
|
|
|
|
|
char sendbuf[BUFSIZ * 2], recvbuf[BUFSIZ];
|
|
|
|
|
completion_t completion;
|
|
|
|
|
isc_region_t r;
|
2018-05-01 16:52:01 +10:00
|
|
|
isc_socketevent_t *socketevent;
|
|
|
|
|
|
2018-10-26 17:07:50 +08:00
|
|
|
UNUSED(state);
|
2018-05-01 16:52:01 +10:00
|
|
|
|
|
|
|
|
in.s_addr = inet_addr("127.0.0.1");
|
|
|
|
|
isc_sockaddr_fromin(&addr1, &in, 0);
|
|
|
|
|
isc_sockaddr_fromin(&addr2, &in, 0);
|
|
|
|
|
|
|
|
|
|
result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2018-05-01 16:52:01 +10:00
|
|
|
result = isc_socket_bind(s1, &addr1, ISC_SOCKET_REUSEADDRESS);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2018-05-01 16:52:01 +10:00
|
|
|
result = isc_socket_getsockname(s1, &addr1);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr1) != 0);
|
2018-05-01 16:52:01 +10:00
|
|
|
result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2018-05-01 16:52:01 +10:00
|
|
|
result = isc_socket_bind(s2, &addr2, ISC_SOCKET_REUSEADDRESS);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2018-05-01 16:52:01 +10:00
|
|
|
result = isc_socket_getsockname(s2, &addr2);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
|
assert_true(isc_sockaddr_getport(&addr2) != 0);
|
2018-05-01 16:52:01 +10:00
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_task_create(taskmgr, 0, &test_task);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2018-05-01 16:52:01 +10:00
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
* Send a message that will not be truncated.
|
|
|
|
|
*/
|
|
|
|
|
memset(sendbuf, 0xff, sizeof(sendbuf));
|
|
|
|
|
snprintf(sendbuf, sizeof(sendbuf), "Hello");
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)sendbuf;
|
2018-05-01 16:52:01 +10:00
|
|
|
r.length = strlen(sendbuf) + 1;
|
|
|
|
|
|
|
|
|
|
completion_init(&completion);
|
|
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
socketevent = isc_socket_socketevent(
|
|
|
|
|
test_mctx, s1, ISC_SOCKEVENT_SENDDONE, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_non_null(socketevent);
|
2018-05-01 16:52:01 +10:00
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_sendto2(s1, &r, test_task, &addr2, NULL,
|
|
|
|
|
socketevent, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2018-05-01 16:52:01 +10:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
2018-05-01 16:52:01 +10:00
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)recvbuf;
|
2018-05-01 16:52:01 +10:00
|
|
|
r.length = BUFSIZ;
|
|
|
|
|
completion_init(&completion);
|
2018-04-17 08:29:14 -07:00
|
|
|
recv_trunc = false;
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_recv(s2, &r, 1, test_task, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2018-05-01 16:52:01 +10:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
|
|
|
|
assert_string_equal(recvbuf, "Hello");
|
|
|
|
|
assert_false(recv_trunc);
|
2018-05-01 16:52:01 +10:00
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
* Send a message that will be truncated.
|
|
|
|
|
*/
|
|
|
|
|
memset(sendbuf, 0xff, sizeof(sendbuf));
|
|
|
|
|
snprintf(sendbuf, sizeof(sendbuf), "Hello");
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)sendbuf;
|
2018-05-01 16:52:01 +10:00
|
|
|
r.length = sizeof(sendbuf);
|
|
|
|
|
|
|
|
|
|
completion_init(&completion);
|
|
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
socketevent = isc_socket_socketevent(
|
|
|
|
|
test_mctx, s1, ISC_SOCKEVENT_SENDDONE, event_done, &completion);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_non_null(socketevent);
|
2018-05-01 16:52:01 +10:00
|
|
|
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_sendto2(s1, &r, test_task, &addr2, NULL,
|
|
|
|
|
socketevent, 0);
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2018-05-01 16:52:01 +10:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
2018-05-01 16:52:01 +10:00
|
|
|
|
2020-02-12 13:59:18 +01:00
|
|
|
r.base = (void *)recvbuf;
|
2018-05-01 16:52:01 +10:00
|
|
|
r.length = BUFSIZ;
|
|
|
|
|
completion_init(&completion);
|
2018-04-17 08:29:14 -07:00
|
|
|
recv_trunc = false;
|
2020-05-14 09:39:57 +02:00
|
|
|
result = isc_socket_recv(s2, &r, 1, test_task, event_done, &completion);
|
2019-02-13 15:53:41 +11:00
|
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
2018-05-01 16:52:01 +10:00
|
|
|
waitfor(&completion);
|
2019-07-04 13:51:47 +02:00
|
|
|
assert_true(atomic_load(&completion.done));
|
2018-10-26 17:07:50 +08:00
|
|
|
assert_int_equal(completion.result, ISC_R_SUCCESS);
|
|
|
|
|
assert_string_equal(recvbuf, "Hello");
|
|
|
|
|
assert_true(recv_trunc);
|
2018-05-01 16:52:01 +10:00
|
|
|
}
|
|
|
|
|
|
2011-07-28 04:04:37 +00:00
|
|
|
/*
|
|
|
|
|
* Main
|
|
|
|
|
*/
|
2018-10-26 17:07:50 +08:00
|
|
|
int
|
2020-02-13 14:44:37 -08:00
|
|
|
main(void) {
|
2018-10-26 17:07:50 +08:00
|
|
|
const struct CMUnitTest tests[] = {
|
2020-02-12 13:59:18 +01:00
|
|
|
cmocka_unit_test_setup_teardown(udp_sendto_test, _setup,
|
|
|
|
|
_teardown),
|
|
|
|
|
cmocka_unit_test_setup_teardown(udp_dup_test, _setup,
|
|
|
|
|
_teardown),
|
|
|
|
|
cmocka_unit_test_setup_teardown(tcp_dscp_v4_test, _setup,
|
|
|
|
|
_teardown),
|
|
|
|
|
cmocka_unit_test_setup_teardown(tcp_dscp_v6_test, _setup,
|
|
|
|
|
_teardown),
|
|
|
|
|
cmocka_unit_test_setup_teardown(udp_dscp_v4_test, _setup,
|
|
|
|
|
_teardown),
|
|
|
|
|
cmocka_unit_test_setup_teardown(udp_dscp_v6_test, _setup,
|
|
|
|
|
_teardown),
|
|
|
|
|
cmocka_unit_test_setup_teardown(net_probedscp_test, _setup,
|
|
|
|
|
_teardown),
|
|
|
|
|
cmocka_unit_test_setup_teardown(udp_trunc_test, _setup,
|
|
|
|
|
_teardown),
|
2018-10-26 17:07:50 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
return (cmocka_run_group_tests(tests, NULL, NULL));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#else /* HAVE_CMOCKA */
|
|
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
|
|
|
|
|
|
int
|
2020-02-13 14:44:37 -08:00
|
|
|
main(void) {
|
2018-10-26 17:07:50 +08:00
|
|
|
printf("1..0 # Skipped: cmocka not available\n");
|
2021-01-18 19:15:44 +01:00
|
|
|
return (SKIPPED_TEST_EXIT_CODE);
|
2011-07-28 04:04:37 +00:00
|
|
|
}
|
2018-10-26 17:07:50 +08:00
|
|
|
|
2020-02-13 21:48:23 +01:00
|
|
|
#endif /* if HAVE_CMOCKA */
|
