Home | History | Annotate | Line # | Download | only in ftp 
      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 
     31 atf_test_case custom_headers cleanup
     32 custom_headers_head()
     33 {
     34 	atf_require_prog ftp
     35 	atf_set "descr" "Check for custom HTTP headers"
     36 }
     37 
     38 HTTPD_PID=./.__httpd.pid
     39 custom_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 
     54 custom_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 
     64 atf_init_test_cases()
     65 {
     66 	atf_add_test_case custom_headers
     67 }
     68