Page MenuHomeFreeBSD
Differential D30408 Diff 89905 tests/sys/netinet/libalias/3_natin.c

Changeset View

Standalone View

View Options

tests/sys/netinet/libalias/3_natin.c

  • This file was added.
/*
* SPDX-License-Identifier: BSD-3-Clause
*
* Copyright 2021 Lutz Donnerhacke
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following
* disclaimer in the documentation and/or other materials provided
* with the distribution.
* 3. Neither the name of the copyright holder nor the names of its
* contributors may be used to endorse or promote products derived
* from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
* CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
* TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
* ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
* TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
* THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include <atf-c.h>
#include <alias.h>
#include <stdio.h>
#include <stdlib.h>
#include "util.h"
ATF_TC_WITHOUT_HEAD(1_portforward);
ATF_TC_BODY(1_portforward, dummy)
{
struct libalias *la = LibAliasInit(NULL);
struct alias_link *pf1, *pf2, *pf3, *pf4;
struct ip *p;
struct udphdr *u;
ATF_REQUIRE(la != NULL);
LibAliasSetAddress(la, masq);
LibAliasSetMode(la, PKT_ALIAS_RESET_ON_ADDR_CHANGE, ~0);
LibAliasSetMode(la, PKT_ALIAS_DENY_INCOMING, PKT_ALIAS_DENY_INCOMING);
/*
* Fully specified
*/
pf1 = LibAliasRedirectPort(la, prv1, ntohs(0x1234), ext, ntohs(0x5678), masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf1 != NULL);
p = ip_packet(0, 64);
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv1, 0x1234);
/* try again */
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv1, 0x1234);
/* different source */
UDP_UNNAT_FAIL(p, u, pub, 0x5678, masq, 0xabcd);
UDP_UNNAT_FAIL(p, u, ext, 0xdead, masq, 0xabcd);
/* clear table by keeping the address */
LibAliasSetAddress(la, ext);
LibAliasSetAddress(la, masq);
/* delete and try again */
LibAliasRedirectDelete(la, pf1);
UDP_UNNAT_FAIL(p, u, ext, 0x5678, masq, 0xabcd);
/*
* Any external port
*/
pf2 = LibAliasRedirectPort(la, prv2, ntohs(0x1234), ext, ntohs(0), masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf2 != NULL);
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv2, 0x1234);
/* try again */
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv2, 0x1234);
/* different source */
UDP_UNNAT_FAIL(p, u, pub, 0x5678, masq, 0xabcd);
UDP_UNNAT_CHECK(p, u, ext, 0xdead, masq, 0xabcd, prv2, 0x1234);
/* clear table by keeping the address */
LibAliasSetAddress(la, ext);
LibAliasSetAddress(la, masq);
/* delete and try again */
LibAliasRedirectDelete(la, pf2);
UDP_UNNAT_FAIL(p, u, ext, 0x5678, masq, 0xabcd);
/*
* Any external host
*/
pf3 = LibAliasRedirectPort(la, prv3, ntohs(0x1234), ANY_ADDR, ntohs(0x5678), masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf3 != NULL);
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv3, 0x1234);
/* try again */
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv3, 0x1234);
/* different source */
UDP_UNNAT_CHECK(p, u, pub, 0x5678, masq, 0xabcd, prv3, 0x1234);
UDP_UNNAT_FAIL(p, u, ext, 0xdead, masq, 0xabcd);
/* clear table by keeping the address */
LibAliasSetAddress(la, ext);
LibAliasSetAddress(la, masq);
/* delete and try again */
LibAliasRedirectDelete(la, pf2);
UDP_UNNAT_FAIL(p, u, ext, 0x5678, masq, 0xabcd);
/*
* Any external host, any port
*/
pf4 = LibAliasRedirectPort(la, cgn, ntohs(0x1234), ANY_ADDR, ntohs(0), masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf4 != NULL);
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, cgn, 0x1234);
/* try again */
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, cgn, 0x1234);
/* different source */
UDP_UNNAT_CHECK(p, u, pub, 0x5678, masq, 0xabcd, cgn, 0x1234);
UDP_UNNAT_CHECK(p, u, ext, 0xdead, masq, 0xabcd, cgn, 0x1234);
/* clear table by keeping the address */
LibAliasSetAddress(la, ext);
LibAliasSetAddress(la, masq);
/* delete and try again */
LibAliasRedirectDelete(la, pf2);
UDP_UNNAT_FAIL(p, u, ext, 0x5678, masq, 0xabcd);
free(p);
LibAliasUninit(la);
}
ATF_TC_WITHOUT_HEAD(2_portoverlap);
ATF_TC_BODY(2_portoverlap, dummy)
{
struct libalias *la = LibAliasInit(NULL);
struct alias_link *pf1, *pf2, *pf3, *pf4;
struct ip *p;
struct udphdr *u;
ATF_REQUIRE(la != NULL);
LibAliasSetAddress(la, masq);
LibAliasSetMode(la, PKT_ALIAS_RESET_ON_ADDR_CHANGE, ~0);
LibAliasSetMode(la, PKT_ALIAS_DENY_INCOMING, PKT_ALIAS_DENY_INCOMING);
/*
* Fully specified
*/
pf1 = LibAliasRedirectPort(la, prv2, ntohs(0x1234), ext, ntohs(0x5678), masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf1 != NULL);
p = ip_packet(0, 64);
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv2, 0x1234);
/* clear table by keeping the address */
LibAliasSetAddress(la, ext);
LibAliasSetAddress(la, masq);
/*
* Fully specified (override)
*/
pf1 = LibAliasRedirectPort(la, prv1, ntohs(0x1234), ext, ntohs(0x5678), masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf1 != NULL);
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv1, 0x1234);
/* clear table by keeping the address */
LibAliasSetAddress(la, ext);
LibAliasSetAddress(la, masq);
/*
* Any external port
*/
pf2 = LibAliasRedirectPort(la, prv2, ntohs(0x1234), ext, ntohs(0), masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf2 != NULL);
UDP_UNNAT_CHECK(p, u, ext, 0x5679, masq, 0xabcd, prv2, 0x1234);
/* more specific rule wins */
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv1, 0x1234);
/* clear table by keeping the address */
LibAliasSetAddress(la, ext);
LibAliasSetAddress(la, masq);
/*
* Any external host
*/
pf3 = LibAliasRedirectPort(la, prv3, ntohs(0x1234), ANY_ADDR, ntohs(0x5678), masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf3 != NULL);
UDP_UNNAT_CHECK(p, u, pub, 0x5678, masq, 0xabcd, prv3, 0x1234);
/* more specific rule wins */
UDP_UNNAT_CHECK(p, u, ext, 0x5679, masq, 0xabcd, prv2, 0x1234);
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv1, 0x1234);
/* clear table by keeping the address */
LibAliasSetAddress(la, ext);
LibAliasSetAddress(la, masq);
/*
* Any external host, any port
*/
pf4 = LibAliasRedirectPort(la, cgn, ntohs(0x1234), ANY_ADDR, ntohs(0), masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf4 != NULL);
UDP_UNNAT_CHECK(p, u, prv1, 0x5679, masq, 0xabcd, cgn, 0x1234);
/* more specific rule wins */
UDP_UNNAT_CHECK(p, u, pub, 0x5678, masq, 0xabcd, prv3, 0x1234);
UDP_UNNAT_CHECK(p, u, ext, 0x5679, masq, 0xabcd, prv2, 0x1234);
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv1, 0x1234);
free(p);
LibAliasUninit(la);
}
ATF_TC_WITHOUT_HEAD(3_redirectany);
ATF_TC_BODY(3_redirectany, dummy)
{
struct libalias *la = LibAliasInit(NULL);
struct alias_link *pf;
struct ip *p;
struct udphdr *u;
ATF_REQUIRE(la != NULL);
LibAliasSetMode(la, PKT_ALIAS_DENY_INCOMING, ~0);
p = ip_packet(0, 64);
pf = LibAliasRedirectPort(la, prv1, ntohs(0x1234), ANY_ADDR, 0, ANY_ADDR, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf != NULL);
LibAliasSetAddress(la, masq);
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv1, 0x1234);
UDP_UNNAT_FAIL(p, u, pub, 0x5678, pub, 0xabcd);
LibAliasSetAddress(la, pub);
UDP_UNNAT_CHECK(p, u, pub, 0x5679, pub, 0xabcd, prv1, 0x1234);
UDP_UNNAT_FAIL(p, u, ext, 0x5679, masq, 0xabcd);
free(p);
LibAliasUninit(la);
}
ATF_TC_WITHOUT_HEAD(4_redirectaddr);
ATF_TC_BODY(4_redirectaddr, dummy)
{
struct libalias *la = LibAliasInit(NULL);
struct alias_link *pf1, *pf2;
struct ip *p;
ATF_REQUIRE(la != NULL);
LibAliasSetAddress(la, masq);
pf1 = LibAliasRedirectAddr(la, prv1, pub);
ATF_REQUIRE(pf1 != NULL);
p = ip_packet(254, 64);
UNNAT_CHECK(p, ext, pub, prv1);
UNNAT_CHECK(p, ext, masq, masq);
pf2 = LibAliasRedirectAddr(la, prv2, pub);
ATF_REQUIRE(pf2 != NULL);
UNNAT_CHECK(p, ext, pub, prv1);
p->ip_p = 253; /* new flows */
UNNAT_CHECK(p, ext, pub, prv2);
UNNAT_CHECK(p, ext, masq, masq);
p->ip_p = 252; /* new flows */
NAT_CHECK(p, prv1, ext, pub);
NAT_CHECK(p, prv2, ext, pub);
NAT_CHECK(p, prv3, ext, masq);
LibAliasSetMode(la, PKT_ALIAS_DENY_INCOMING, ~0);
p->ip_p = 251; /* new flows */
UNNAT_FAIL(p, ext, pub);
UNNAT_FAIL(p, ext, masq);
/* unhide older version */
LibAliasRedirectDelete(la, pf2);
LibAliasSetMode(la, 0, ~0);
p->ip_p = 250; /* new flows */
UNNAT_CHECK(p, ext, pub, prv1);
p->ip_p = 249; /* new flows */
NAT_CHECK(p, prv1, ext, pub);
NAT_CHECK(p, prv2, ext, masq);
NAT_CHECK(p, prv3, ext, masq);
free(p);
LibAliasUninit(la);
}
ATF_TC_WITHOUT_HEAD(5_lsnat);
ATF_TC_BODY(5_lsnat, dummy)
{
struct libalias *la = LibAliasInit(NULL);
struct alias_link *pf;
struct ip *p;
struct udphdr *u;
ATF_REQUIRE(la != NULL);
LibAliasSetMode(la, 0, ~0);
p = ip_packet(0, 64);
pf = LibAliasRedirectPort(la, cgn, ntohs(0xdead), ANY_ADDR, 0, masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf != NULL);
ATF_REQUIRE(0 == LibAliasAddServer(la, pf, prv1, ntohs(0x1234)));
ATF_REQUIRE(0 == LibAliasAddServer(la, pf, prv2, ntohs(0x2345)));
ATF_REQUIRE(0 == LibAliasAddServer(la, pf, prv3, ntohs(0x3456)));
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv3, 0x3456);
UDP_UNNAT_CHECK(p, u, ext, 0x5679, masq, 0xabcd, prv2, 0x2345);
UDP_UNNAT_CHECK(p, u, ext, 0x567a, masq, 0xabcd, prv1, 0x1234);
UDP_UNNAT_CHECK(p, u, ext, 0x567b, masq, 0xabcd, prv3, 0x3456);
UDP_UNNAT_CHECK(p, u, ext, 0x567c, masq, 0xabcd, prv2, 0x2345);
UDP_UNNAT_CHECK(p, u, ext, 0x567d, masq, 0xabcd, prv1, 0x1234);
free(p);
LibAliasUninit(la);
}
ATF_TC_WITHOUT_HEAD(6_oneshot);
ATF_TC_BODY(6_oneshot, dummy)
{
struct libalias *la = LibAliasInit(NULL);
struct alias_link *pf;
struct ip *p;
struct udphdr *u;
ATF_REQUIRE(la != NULL);
LibAliasSetMode(la, 0, ~0);
LibAliasSetMode(la, PKT_ALIAS_RESET_ON_ADDR_CHANGE, ~0);
LibAliasSetMode(la, PKT_ALIAS_DENY_INCOMING, PKT_ALIAS_DENY_INCOMING);
pf = LibAliasRedirectPort(la, prv1, ntohs(0x1234), ANY_ADDR, 0, masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf != NULL);
/* only for fully specified links */
ATF_CHECK(-1 == LibAliasRedirectDynamic(la, pf));
LibAliasRedirectDelete(la, pf);
pf = LibAliasRedirectPort(la, prv1, ntohs(0x1234), ext, ntohs(0x5678), masq, ntohs(0xabcd), IPPROTO_UDP);
ATF_REQUIRE(pf != NULL);
ATF_CHECK(0 == LibAliasRedirectDynamic(la, pf));
p = ip_packet(0, 64);
UDP_UNNAT_CHECK(p, u, ext, 0x5678, masq, 0xabcd, prv1, 0x1234);
/* clear table by keeping the address */
LibAliasSetAddress(la, ext);
LibAliasSetAddress(la, masq);
/* does not work anymore */
UDP_UNNAT_FAIL(p, u, ext, 0x5678, masq, 0xabcd);
free(p);
LibAliasUninit(la);
}
ATF_TP_ADD_TCS(natin)
{
/* Use "dd if=/dev/random bs=2 count=1 | od -x" to reproduce */
srand(0xe859);
ATF_TP_ADD_TC(natin, 1_portforward);
ATF_TP_ADD_TC(natin, 2_portoverlap);
ATF_TP_ADD_TC(natin, 3_redirectany);
ATF_TP_ADD_TC(natin, 4_redirectaddr);
ATF_TP_ADD_TC(natin, 5_lsnat);
ATF_TP_ADD_TC(natin, 6_oneshot);
return atf_no_error();
}