mirror of
https://github.com/freebsd/freebsd-src
synced 2024-10-15 21:05:08 +00:00
b3dc6f137b
Fix two copy and waste errors (referencing the wrong variable). MFC after: 3 days
382 lines
12 KiB
C
382 lines
12 KiB
C
/*
|
|
* 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, pf3);
|
|
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, pf4);
|
|
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();
|
|
}
|