1# $NetBSD: t_custom_headers.sh,v 1.1 2024/10/12 22:19:37 riastradh Exp $ 2# 3# Copyright (c) 2024 The NetBSD Foundation, Inc. 4# All rights reserved. 5# 6# This code is derived from software contributed to The NetBSD Foundation 7# by Sunil Nimmagadda. 8# 9# Redistribution and use in source and binary forms, with or without 10# modification, are permitted provided that the following conditions 11# are met: 12# 1. Redistributions of source code must retain the above copyright 13# notice, this list of conditions and the following disclaimer. 14# 2. Redistributions in binary form must reproduce the above copyright 15# notice, this list of conditions and the following disclaimer in the 16# documentation and/or other materials provided with the distribution. 17# 18# THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 19# ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 20# TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 21# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 22# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 23# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 24# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 25# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 26# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 27# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 28# POSSIBILITY OF SUCH DAMAGE. 29# 30 31atf_test_case custom_headers cleanup 32custom_headers_head() 33{ 34 atf_require_prog ftp 35 atf_set "descr" "Check for custom HTTP headers" 36} 37 38HTTPD_PID=./.__httpd.pid 39custom_headers_body() 40{ 41 # start httpd in daemon mode 42 atf_check -s exit:0 \ 43 /usr/libexec/httpd -P $HTTPD_PID -I 8080 -b -C .sh /bin/sh \ 44 -c "$(atf_get_srcdir)" . 45 46 atf_check \ 47 -o inline:'HTTP_X_ORIGIN=example.com\nHTTP_X_RATE_LIMIT=1000\n' \ 48 ftp -V -o - \ 49 -H 'X-Origin: example.com' \ 50 -H 'X-Rate-Limit: 1000' \ 51 http://127.0.0.1:8080/cgi-bin/custom_headers.sh 52} 53 54custom_headers_cleanup() 55{ 56 if [ -f "$HTTPD_PID" ]; then 57 echo kill -9 "$(cat $HTTPD_PID)" 58 kill -9 "$(cat $HTTPD_PID)" 59 echo '# wait for httpd to exit' 60 sleep 1 61 fi 62} 63 64atf_init_test_cases() 65{ 66 atf_add_test_case custom_headers 67} 68