isc/tests/socket_test.c

f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt/*
bd911976d51f102751848568ccf56592fd5f6d77Tinderbox User * Copyright (C) 2011-2017  Internet Systems Consortium, Inc. ("ISC")
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt *
0c27b3fe77ac1d5094ba3521e8142d9e7973133fMark Andrews * This Source Code Form is subject to the terms of the Mozilla Public
0c27b3fe77ac1d5094ba3521e8142d9e7973133fMark Andrews * License, v. 2.0. If a copy of the MPL was not distributed with this
0c27b3fe77ac1d5094ba3521e8142d9e7973133fMark Andrews * file, You can obtain one at http://mozilla.org/MPL/2.0/.
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt */
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt/*! \file */
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt#include <config.h>
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt#include <atf-c.h>
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt#include <unistd.h>
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt#include <time.h>
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
9effea437d7973c9ed981ab2e14b219182d86ee7Curtis Blackburn#include <isc/platform.h>
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt#include <isc/socket.h>
3759f10fc543747668b1ca4b4671f35b0dea8445Francis Dupont#include <isc/print.h>
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt#include "../task_p.h"
cd594861c26f7392d05c777a1c62c43af84173bdEvan Hunt#include "../unix/socket_p.h"
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt#include "isctest.h"
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Huntstatic isc_boolean_t recv_dscp;
67adc03ef81fb610f8df093b17f55275ee816754Evan Huntstatic unsigned int recv_dscp_value;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt/*
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt * Helper functions
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt */
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunttypedef struct {
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_boolean_t done;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_result_t result;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_t *socket;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt} completion_t;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Huntstatic void
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Huntcompletion_init(completion_t *completion) {
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion->done = ISC_FALSE;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion->socket = NULL;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Huntstatic void
67adc03ef81fb610f8df093b17f55275ee816754Evan Huntaccept_done(isc_task_t *task, isc_event_t *event) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_newconnev_t *nevent = (isc_socket_newconnev_t *)event;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_t *completion = event->ev_arg;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    UNUSED(task);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion->result = nevent->result;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion->done = ISC_TRUE;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if (completion->result == ISC_R_SUCCESS)
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        completion->socket = nevent->newsocket;
313b0ea9f258edd8530f4454c69e6ba194280162Tinderbox User
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_event_free(&event);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt}
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Huntstatic void
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Huntevent_done(isc_task_t *task, isc_event_t *event) {
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_socketevent_t *dev;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion_t *completion = event->ev_arg;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    UNUSED(task);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    dev = (isc_socketevent_t *) event;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion->result = dev->result;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion->done = ISC_TRUE;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if ((dev->attributes & ISC_SOCKEVENTATTR_DSCP) != 0) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        recv_dscp = ISC_TRUE;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        recv_dscp_value = dev->dscp;;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    } else
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        recv_dscp = ISC_FALSE;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_event_free(&event);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt}
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Huntstatic isc_result_t
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Huntwaitfor(completion_t *completion) {
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    int i = 0;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    while (!completion->done && i++ < 5000) {
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt#ifndef ISC_PLATFORM_USETHREADS
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt        while (isc__taskmgr_ready(taskmgr))
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt            isc__taskmgr_dispatch(taskmgr);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt#endif
8a2ab2b9203120c3e2a883a5ee8c0b5d60c1808cEvan Hunt        isc_test_nap(1000);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    }
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    if (completion->done)
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt        return (ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    return (ISC_R_FAILURE);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt}
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt#if 0
67adc03ef81fb610f8df093b17f55275ee816754Evan Huntstatic isc_result_t
67adc03ef81fb610f8df093b17f55275ee816754Evan Huntwaitfor(completion_t *completion) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    int i = 0;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    while (!completion->done && i++ < 5000) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        waitbody();
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    }
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if (completion->done)
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        return (ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    return (ISC_R_FAILURE);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt#endif
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Huntstatic void
b454c0319685041db3f3e8fd7671e1b364fd20c5Evan Huntwaitbody(void) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt#ifndef ISC_PLATFORM_USETHREADS
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    struct timeval tv;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socketwait_t *swait = NULL;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    while (isc__taskmgr_ready(taskmgr))
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        isc__taskmgr_dispatch(taskmgr);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if (socketmgr != NULL) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        tv.tv_sec = 0;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        tv.tv_usec = 1000 ;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        if (isc__socketmgr_waitevents(socketmgr, &tv, &swait) > 0)
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt            isc__socketmgr_dispatch(socketmgr, swait);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    } else
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt#endif
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        isc_test_nap(1000);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Huntstatic isc_result_t
67adc03ef81fb610f8df093b17f55275ee816754Evan Huntwaitfor2(completion_t *c1, completion_t *c2) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    int i = 0;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    while (!(c1->done && c2->done) && i++ < 5000) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        waitbody();
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    }
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if (c1->done && c2->done)
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        return (ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    return (ISC_R_FAILURE);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt/*
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt * Individual unit tests
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt */
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt/* Test UDP sendto/recv (IPv4) */
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan HuntATF_TC(udp_sendto);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan HuntATF_TC_HEAD(udp_sendto, tc) {
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    atf_tc_set_md_var(tc, "descr", "UDP sendto/recv");
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt}
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan HuntATF_TC_BODY(udp_sendto, tc) {
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_result_t result;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_sockaddr_t addr1, addr2;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    struct in_addr in;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_socket_t *s1 = NULL, *s2 = NULL;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_task_t *task = NULL;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion_t completion;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_region_t r;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    UNUSED(tc);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_test_begin(NULL, ISC_TRUE);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    in.s_addr = inet_addr("127.0.0.1");
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    isc_sockaddr_fromin(&addr1, &in, 0);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    isc_sockaddr_fromin(&addr2, &in, 0);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s1);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_bind(s1, &addr1, 0);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_getsockname(s1, &addr1);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_REQUIRE(isc_sockaddr_getport(&addr1) != 0);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s2);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_bind(s2, &addr2, 0);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_getsockname(s2, &addr2);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_REQUIRE(isc_sockaddr_getport(&addr2) != 0);
2f17ad4545ca552c92c88f7cb1e2525050c10c67Automatic Updater
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_task_create(taskmgr, 0, &task);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
7cd594b8427fe742d44295790ba367e1de22a47dEvan Hunt    snprintf(sendbuf, sizeof(sendbuf), "Hello");
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.base = (void *) sendbuf;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.length = strlen(sendbuf) + 1;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion_init(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_socket_sendto(s1, &r, task, event_done, &completion,
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt                   &addr2, NULL);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    waitfor(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK(completion.done);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.base = (void *) recvbuf;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.length = BUFSIZ;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion_init(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_socket_recv(s2, &r, 1, task, event_done, &completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    waitfor(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK(completion.done);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_STREQ(recvbuf, "Hello");
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_task_detach(&task);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_socket_detach(&s1);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_socket_detach(&s2);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_test_end();
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt}
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt/* Test UDP sendto/recv with duplicated socket */
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan HuntATF_TC(udp_dup);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan HuntATF_TC_HEAD(udp_dup, tc) {
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    atf_tc_set_md_var(tc, "descr", "duplicated socket sendto/recv");
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt}
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan HuntATF_TC_BODY(udp_dup, tc) {
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_result_t result;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_sockaddr_t addr1, addr2;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    struct in_addr in;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_socket_t *s1 = NULL, *s2 = NULL, *s3 = NULL;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_task_t *task = NULL;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion_t completion;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_region_t r;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    UNUSED(tc);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_test_begin(NULL, ISC_TRUE);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    in.s_addr = inet_addr("127.0.0.1");
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    isc_sockaddr_fromin(&addr1, &in, 0);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    isc_sockaddr_fromin(&addr2, &in, 0);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s1);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_bind(s1, &addr1, 0);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_getsockname(s1, &addr1);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_REQUIRE(isc_sockaddr_getport(&addr1) != 0);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s2);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_bind(s2, &addr2, 0);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_getsockname(s2, &addr2);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_REQUIRE(isc_sockaddr_getport(&addr2) != 0);
23daf8948b2e5462ccea082a441e9759dffa50cdMark Andrews
23daf8948b2e5462ccea082a441e9759dffa50cdMark Andrews    result = isc_socket_dup(s2, &s3);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
2f17ad4545ca552c92c88f7cb1e2525050c10c67Automatic Updater
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_task_create(taskmgr, 0, &task);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
7cd594b8427fe742d44295790ba367e1de22a47dEvan Hunt    snprintf(sendbuf, sizeof(sendbuf), "Hello");
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.base = (void *) sendbuf;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.length = strlen(sendbuf) + 1;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion_init(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_socket_sendto(s1, &r, task, event_done, &completion,
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt                   &addr2, NULL);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    waitfor(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK(completion.done);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
7cd594b8427fe742d44295790ba367e1de22a47dEvan Hunt    snprintf(sendbuf, sizeof(sendbuf), "World");
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.base = (void *) sendbuf;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.length = strlen(sendbuf) + 1;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion_init(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_socket_sendto(s1, &r, task, event_done, &completion,
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt                   &addr2, NULL);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    waitfor(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK(completion.done);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.base = (void *) recvbuf;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.length = BUFSIZ;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion_init(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_socket_recv(s2, &r, 1, task, event_done, &completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    waitfor(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK(completion.done);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_STREQ(recvbuf, "Hello");
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.base = (void *) recvbuf;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    r.length = BUFSIZ;
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    completion_init(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    result = isc_socket_recv(s3, &r, 1, task, event_done, &completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    waitfor(&completion);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK(completion.done);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_CHECK_STREQ(recvbuf, "World");
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_task_detach(&task);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_socket_detach(&s1);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_socket_detach(&s2);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_socket_detach(&s3);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    isc_test_end();
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt}
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt/* Test TCP sendto/recv (IPv4) */
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC(udp_dscp_v4);
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC_HEAD(udp_dscp_v4, tc) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    atf_tc_set_md_var(tc, "descr", "UDP DSCP IPV4");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC_BODY(udp_dscp_v4, tc) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_result_t result;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_sockaddr_t addr1, addr2;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    struct in_addr in;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_t *s1 = NULL, *s2 = NULL;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_task_t *task = NULL;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_t completion;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_region_t r;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socketevent_t *socketevent;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    UNUSED(tc);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_test_begin(NULL, ISC_TRUE);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    in.s_addr = inet_addr("127.0.0.1");
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    isc_sockaddr_fromin(&addr1, &in, 0);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    isc_sockaddr_fromin(&addr2, &in, 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s1);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt               isc_result_totext(result));
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_bind(s1, &addr1, ISC_SOCKET_REUSEADDRESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt               isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_getsockname(s1, &addr1);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_REQUIRE(isc_sockaddr_getport(&addr1) != 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp, &s2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt               isc_result_totext(result));
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_bind(s2, &addr2, ISC_SOCKET_REUSEADDRESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt               isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_getsockname(s2, &addr2);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_REQUIRE(isc_sockaddr_getport(&addr2) != 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_task_create(taskmgr, 0, &task);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt               isc_result_totext(result));
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
7cd594b8427fe742d44295790ba367e1de22a47dEvan Hunt    snprintf(sendbuf, sizeof(sendbuf), "Hello");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.base = (void *) sendbuf;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.length = strlen(sendbuf) + 1;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion);
313b0ea9f258edd8530f4454c69e6ba194280162Tinderbox User
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    socketevent = isc_socket_socketevent(mctx, s1, ISC_SOCKEVENT_SENDDONE,
313b0ea9f258edd8530f4454c69e6ba194280162Tinderbox User                         event_done, &completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE(socketevent != NULL);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if ((isc_net_probedscp() & ISC_NET_DSCPPKTV4) != 0) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        socketevent->dscp = 056; /* EF */
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        socketevent->attributes |= ISC_SOCKEVENTATTR_DSCP;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    } else if ((isc_net_probedscp() & ISC_NET_DSCPSETV4) != 0) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        isc_socket_dscp(s1, 056);  /* EF */
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        socketevent->dscp = 0;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        socketevent->attributes &= ~ISC_SOCKEVENTATTR_DSCP;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    }
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    recv_dscp = ISC_FALSE;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    recv_dscp_value = 0;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_sendto2(s1, &r, task, &addr2, NULL, socketevent, 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt               isc_result_totext(result));
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    waitfor(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.base = (void *) recvbuf;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.length = BUFSIZ;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_recv(s2, &r, 1, task, event_done, &completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    waitfor(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_STREQ(recvbuf, "Hello");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if ((isc_net_probedscp() & ISC_NET_DSCPRECVV4) != 0) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        ATF_CHECK(recv_dscp);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        ATF_CHECK_EQ(recv_dscp_value, 056);
313b0ea9f258edd8530f4454c69e6ba194280162Tinderbox User    } else
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        ATF_CHECK(!recv_dscp);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_task_detach(&task);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_detach(&s1);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_detach(&s2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_test_end();
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt/* Test TCP sendto/recv (IPv4) */
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC(udp_dscp_v6);
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC_HEAD(udp_dscp_v6, tc) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    atf_tc_set_md_var(tc, "descr", "udp dscp ipv6");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC_BODY(udp_dscp_v6, tc) {
9effea437d7973c9ed981ab2e14b219182d86ee7Curtis Blackburn#if defined(ISC_PLATFORM_HAVEIPV6) && defined(WANT_IPV6)
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_result_t result;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_sockaddr_t addr1, addr2;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    struct in6_addr in6;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_t *s1 = NULL, *s2 = NULL;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_task_t *task = NULL;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_t completion;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_region_t r;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socketevent_t *socketevent;
45b727f651aba2cbd2f9db51ccfb4b541520a5deMark Andrews    int n;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    UNUSED(tc);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_test_begin(NULL, ISC_TRUE);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
45b727f651aba2cbd2f9db51ccfb4b541520a5deMark Andrews    n = inet_pton(AF_INET6, "::1", &in6.s6_addr);
45b727f651aba2cbd2f9db51ccfb4b541520a5deMark Andrews    ATF_REQUIRE(n == 1);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    isc_sockaddr_fromin6(&addr1, &in6, 0);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    isc_sockaddr_fromin6(&addr2, &in6, 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_create(socketmgr, PF_INET6, isc_sockettype_udp,
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                   &s1);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_bind(s1, &addr1, 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_getsockname(s1, &addr1);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_REQUIRE(isc_sockaddr_getport(&addr1) != 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_create(socketmgr, PF_INET6, isc_sockettype_udp,
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                   &s2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_bind(s2, &addr2, 0);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_getsockname(s2, &addr2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_REQUIRE(isc_sockaddr_getport(&addr2) != 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_task_create(taskmgr, 0, &task);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt             isc_result_totext(result));
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
7cd594b8427fe742d44295790ba367e1de22a47dEvan Hunt    snprintf(sendbuf, sizeof(sendbuf), "Hello");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.base = (void *) sendbuf;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.length = strlen(sendbuf) + 1;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion);
313b0ea9f258edd8530f4454c69e6ba194280162Tinderbox User
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    socketevent = isc_socket_socketevent(mctx, s1, ISC_SOCKEVENT_SENDDONE,
313b0ea9f258edd8530f4454c69e6ba194280162Tinderbox User                         event_done, &completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE(socketevent != NULL);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if ((isc_net_probedscp() & ISC_NET_DSCPPKTV6) != 0) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        socketevent->dscp = 056; /* EF */
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        socketevent->attributes = ISC_SOCKEVENTATTR_DSCP;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    } else if ((isc_net_probedscp() & ISC_NET_DSCPSETV6) != 0)
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        isc_socket_dscp(s1, 056);  /* EF */
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    recv_dscp = ISC_FALSE;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    recv_dscp_value = 0;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_sendto2(s1, &r, task, &addr2, NULL, socketevent, 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    waitfor(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.base = (void *) recvbuf;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.length = BUFSIZ;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_recv(s2, &r, 1, task, event_done, &completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    waitfor(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_STREQ(recvbuf, "Hello");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if ((isc_net_probedscp() & ISC_NET_DSCPRECVV6) != 0) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        ATF_CHECK(recv_dscp);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        ATF_CHECK_EQ(recv_dscp_value, 056);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    } else
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        ATF_CHECK(!recv_dscp);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_task_detach(&task);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_detach(&s1);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_detach(&s2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_test_end();
9effea437d7973c9ed981ab2e14b219182d86ee7Curtis Blackburn#else
9effea437d7973c9ed981ab2e14b219182d86ee7Curtis Blackburn    UNUSED(tc);
9effea437d7973c9ed981ab2e14b219182d86ee7Curtis Blackburn    atf_tc_skip("IPv6 not available");
9effea437d7973c9ed981ab2e14b219182d86ee7Curtis Blackburn#endif
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt/* Test TCP sendto/recv (IPv4) */
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC(tcp_dscp_v4);
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC_HEAD(tcp_dscp_v4, tc) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    atf_tc_set_md_var(tc, "descr", "tcp dscp ipv4");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC_BODY(tcp_dscp_v4, tc) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_result_t result;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_sockaddr_t addr1;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    struct in_addr in;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_t *s1 = NULL, *s2 = NULL, *s3 = NULL;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_task_t *task = NULL;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_t completion, completion2;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_region_t r;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    UNUSED(tc);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_test_begin(NULL, ISC_TRUE);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    in.s_addr = inet_addr("127.0.0.1");
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    isc_sockaddr_fromin(&addr1, &in, 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_tcp, &s1);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_bind(s1, &addr1, 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_getsockname(s1, &addr1);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_REQUIRE(isc_sockaddr_getport(&addr1) != 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_listen(s1, 3);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_create(socketmgr, PF_INET, isc_sockettype_tcp, &s2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_task_create(taskmgr, 0, &task);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_accept(s1, task, accept_done, &completion2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_connect(s2, &addr1, task, event_done, &completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    waitfor2(&completion, &completion2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion2.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion2.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    s3 = completion2.socket;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_dscp(s2, 056);  /* EF */
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
7cd594b8427fe742d44295790ba367e1de22a47dEvan Hunt    snprintf(sendbuf, sizeof(sendbuf), "Hello");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.base = (void *) sendbuf;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.length = strlen(sendbuf) + 1;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    recv_dscp = ISC_FALSE;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    recv_dscp_value = 0;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_sendto(s2, &r, task, event_done, &completion,
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                   NULL, NULL);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    waitfor(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.base = (void *) recvbuf;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.length = BUFSIZ;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_recv(s3, &r, 1, task, event_done, &completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    waitfor(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_STREQ(recvbuf, "Hello");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if ((isc_net_probedscp() & ISC_NET_DSCPRECVV4) != 0) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        if (recv_dscp)
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt            ATF_CHECK_EQ(recv_dscp_value, 056);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    } else
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        ATF_CHECK(!recv_dscp);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_task_detach(&task);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_detach(&s1);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_detach(&s2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_detach(&s3);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_test_end();
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt/* Test TCP sendto/recv (IPv6) */
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC(tcp_dscp_v6);
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC_HEAD(tcp_dscp_v6, tc) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    atf_tc_set_md_var(tc, "descr", "tcp dscp ipv6");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC_BODY(tcp_dscp_v6, tc) {
9effea437d7973c9ed981ab2e14b219182d86ee7Curtis Blackburn#ifdef ISC_PLATFORM_HAVEIPV6
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_result_t result;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_sockaddr_t addr1;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    struct in6_addr in6;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_t *s1 = NULL, *s2 = NULL, *s3 = NULL;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_task_t *task = NULL;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    char sendbuf[BUFSIZ], recvbuf[BUFSIZ];
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_t completion, completion2;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_region_t r;
45b727f651aba2cbd2f9db51ccfb4b541520a5deMark Andrews    int n;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    UNUSED(tc);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_test_begin(NULL, ISC_TRUE);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
45b727f651aba2cbd2f9db51ccfb4b541520a5deMark Andrews    n = inet_pton(AF_INET6, "::1", &in6.s6_addr);
45b727f651aba2cbd2f9db51ccfb4b541520a5deMark Andrews    ATF_REQUIRE(n == 1);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    isc_sockaddr_fromin6(&addr1, &in6, 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_create(socketmgr, PF_INET6, isc_sockettype_tcp,
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                   &s1);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_bind(s1, &addr1, 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    result = isc_socket_getsockname(s1, &addr1);
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_CHECK_EQ_MSG(result, ISC_R_SUCCESS, "%s",
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews             isc_result_totext(result));
94c7301f6f0ada7aa84ede5808ab8fdec5bae6b5Mark Andrews    ATF_REQUIRE(isc_sockaddr_getport(&addr1) != 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_listen(s1, 3);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_create(socketmgr, PF_INET6, isc_sockettype_tcp,
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                   &s2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_task_create(taskmgr, 0, &task);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_accept(s1, task, accept_done, &completion2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_connect(s2, &addr1, task, event_done, &completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    waitfor2(&completion, &completion2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion2.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion2.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    s3 = completion2.socket;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_dscp(s2, 056);  /* EF */
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
7cd594b8427fe742d44295790ba367e1de22a47dEvan Hunt    snprintf(sendbuf, sizeof(sendbuf), "Hello");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.base = (void *) sendbuf;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.length = strlen(sendbuf) + 1;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    recv_dscp = ISC_FALSE;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    recv_dscp_value = 0;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_sendto(s2, &r, task, event_done, &completion,
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                   NULL, NULL);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    waitfor(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.base = (void *) recvbuf;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    r.length = BUFSIZ;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    completion_init(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    result = isc_socket_recv(s3, &r, 1, task, event_done, &completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    waitfor(&completion);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK(completion.done);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_EQ(completion.result, ISC_R_SUCCESS);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK_STREQ(recvbuf, "Hello");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if ((isc_net_probedscp() & ISC_NET_DSCPRECVV6) != 0) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        /*
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt         * IPV6_RECVTCLASS is undefined for TCP however
bcae9a15c1e9c50a6e6433168d5225b1de89d6b9Evan Hunt         * if we do get it it should be the value we set.
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt         */
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        if (recv_dscp)
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt            ATF_CHECK_EQ(recv_dscp_value, 056);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    } else
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        ATF_CHECK(!recv_dscp);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_task_detach(&task);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_detach(&s1);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_detach(&s2);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_socket_detach(&s3);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    isc_test_end();
9effea437d7973c9ed981ab2e14b219182d86ee7Curtis Blackburn#else
9effea437d7973c9ed981ab2e14b219182d86ee7Curtis Blackburn    UNUSED(tc);
9effea437d7973c9ed981ab2e14b219182d86ee7Curtis Blackburn    atf_tc_skip("IPv6 not available");
9effea437d7973c9ed981ab2e14b219182d86ee7Curtis Blackburn#endif
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC(net_probedscp);
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC_HEAD(net_probedscp, tc) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    atf_tc_set_md_var(tc, "descr", "probe dscp capabilities");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan HuntATF_TC_BODY(net_probedscp, tc) {
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    unsigned int n;
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    UNUSED(tc);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    n = isc_net_probedscp();
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_CHECK((n & ~ISC_NET_DSCPALL) == 0);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    /* ISC_NET_DSCPSETV4 MUST be set if any is set. */
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    if (n & (ISC_NET_DSCPSETV4|ISC_NET_DSCPPKTV4|ISC_NET_DSCPRECVV4))
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        ATF_CHECK_MSG((n & ISC_NET_DSCPSETV4) != 0,
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                  "IPv4:%s%s%s\n",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                  (n & ISC_NET_DSCPSETV4) ? " set" : " none",
313b0ea9f258edd8530f4454c69e6ba194280162Tinderbox User                  (n & ISC_NET_DSCPPKTV4) ? " packet" : "",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                  (n & ISC_NET_DSCPRECVV4) ? " receive" : "");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    /* ISC_NET_DSCPSETV6 MUST be set if any is set. */
27deca2bf082e1c4ecbd14d6619cf040b07d2412Mark Andrews    if (n & (ISC_NET_DSCPSETV6|ISC_NET_DSCPPKTV6|ISC_NET_DSCPRECVV6))
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        ATF_CHECK_MSG((n & ISC_NET_DSCPSETV6) != 0,
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                  "IPv6:%s%s%s\n",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                  (n & ISC_NET_DSCPSETV6) ? " set" : " none",
313b0ea9f258edd8530f4454c69e6ba194280162Tinderbox User                  (n & ISC_NET_DSCPPKTV6) ? " packet" : "",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt                  (n & ISC_NET_DSCPRECVV6) ? " receive" : "");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt#if 0
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    fprintf(stdout, "IPv4:%s%s%s\n",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        (n & ISC_NET_DSCPSETV4) ? " set" : "none",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        (n & ISC_NET_DSCPPKTV4) ? " packet" : "",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        (n & ISC_NET_DSCPRECVV4) ? " receive" : "");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
22b5363195e50abd70fd9f06173c0bf84a914667Mark Andrews    fprintf(stdout, "IPv6:%s%s%s\n",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        (n & ISC_NET_DSCPSETV6) ? " set" : "none",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        (n & ISC_NET_DSCPPKTV6) ? " packet" : "",
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt        (n & ISC_NET_DSCPRECVV6) ? " receive" : "");
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt#endif
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt}
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt/*
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt * Main
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt */
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan HuntATF_TP_ADD_TCS(tp) {
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_TP_ADD_TC(tp, udp_sendto);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    ATF_TP_ADD_TC(tp, udp_dup);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_TP_ADD_TC(tp, tcp_dscp_v4);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_TP_ADD_TC(tp, tcp_dscp_v6);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_TP_ADD_TC(tp, udp_dscp_v4);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_TP_ADD_TC(tp, udp_dscp_v6);
67adc03ef81fb610f8df093b17f55275ee816754Evan Hunt    ATF_TP_ADD_TC(tp, net_probedscp);
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt    return (atf_no_error());
f07b2fccaf6be13d9440d323e9e79ee84fe345e2Evan Hunt}