1# 2# SPDX-License-Identifier: BSD-2-Clause 3# 4# Copyright (c) 2021 Rubicon Communications, LLC (Netgate) 5# 6# Redistribution and use in source and binary forms, with or without 7# modification, are permitted provided that the following conditions 8# are met: 9# 1. Redistributions of source code must retain the above copyright 10# notice, this list of conditions and the following disclaimer. 11# 2. Redistributions in binary form must reproduce the above copyright 12# notice, this list of conditions and the following disclaimer in the 13# documentation and/or other materials provided with the distribution. 14# 15# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25# SUCH DAMAGE. 26 27. $(atf_get_srcdir)/utils.subr 28 29common_dir=$(atf_get_srcdir)/../common 30 31atf_test_case "ftp" "cleanup" 32ftp_head() 33{ 34 atf_set descr 'Test ftp-proxy' 35 atf_set require.user root 36 atf_set require.progs twistd 37} 38 39ftp_body() 40{ 41 pft_init 42 43 epair_client=$(vnet_mkepair) 44 epair_link=$(vnet_mkepair) 45 46 ifconfig ${epair_client}a 192.0.2.2/24 up 47 route add -net 198.51.100.0/24 192.0.2.1 48 49 vnet_mkjail fwd ${epair_client}b ${epair_link}a 50 jexec fwd ifconfig ${epair_client}b 192.0.2.1/24 up 51 jexec fwd ifconfig ${epair_link}a 198.51.100.1/24 up 52 jexec fwd ifconfig lo0 127.0.0.1/8 up 53 jexec fwd sysctl net.inet.ip.forwarding=1 54 55 vnet_mkjail srv ${epair_link}b 56 jexec srv ifconfig ${epair_link}b 198.51.100.2/24 up 57 jexec srv route add default 198.51.100.1 58 59 # Start FTP server in srv 60 jexec srv twistd --logfile=/dev/null ftp -r `pwd` -p 21 61 62 # Sanity check 63 atf_check -s exit:0 -o ignore ping -c 1 198.51.100.2 64 65 jexec fwd pfctl -e 66 pft_set_rules fwd \ 67 "nat on ${epair_link}a inet from 192.0.2.0/24 to any -> (${epair_link}a)" \ 68 "nat-anchor \"ftp-proxy/*\"" \ 69 "rdr-anchor \"ftp-proxy/*\"" \ 70 "rdr pass on ${epair_client}b proto tcp from 192.0.2.0/24 to any port 21 -> 127.0.0.1 port 8021" \ 71 "anchor \"ftp-proxy/*\"" \ 72 "pass out proto tcp from 127.0.0.1 to any port 21" 73 jexec fwd /usr/sbin/ftp-proxy 74 75 # Create a dummy file to download 76 echo 'foo' > remote.txt 77 echo -e 'epsv\nget remote.txt local.txt' | ftp -a 198.51.100.2 78 79 # Compare the downloaded file to the original 80 if ! diff -q local.txt remote.txt; 81 then 82 atf_fail 'Failed to retrieve file' 83 fi 84} 85 86ftp_cleanup() 87{ 88 pft_cleanup 89} 90 91atf_init_test_cases() 92{ 93 atf_add_test_case "ftp" 94} 95