1#!/bin/sh
2
3# Copyright (c) 2015 Oracle and/or its affiliates. All Rights Reserved.
4# Copyright (c) International Business Machines  Corp., 2005
5#
6# This program is free software; you can redistribute it and/or
7# modify it under the terms of the GNU General Public License as
8# published by the Free Software Foundation; either version 2 of
9# the License, or (at your option) any later version.
10#
11# This program is distributed in the hope that it would be useful,
12# but WITHOUT ANY WARRANTY; without even the implied warranty of
13# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14# GNU General Public License for more details.
15#
16# You should have received a copy of the GNU General Public License
17# along with this program; if not, write the Free Software Foundation,
18# Inc.,  51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19#
20# Author: Mitsuru Chinen <mitch@jp.ibm.com>
21
22TCID=ftp-upload-stress
23TST_TOTAL=2
24TST_CLEANUP="cleanup"
25
26. test_net.sh
27
28# Big file size to upload (byte)
29UPLOAD_BIGFILESIZE=${UPLOAD_BIGFILESIZE:-2147483647}  # 2GB - 1
30# Regular file size to upload(byte)
31UPLOAD_REGFILESIZE=${UPLOAD_REGFILESIZE:-1024}     # 1K byte
32
33cleanup()
34{
35	rm -f $FTP_UPLOAD_DIR/ftp_file*
36	rm -f $FTP_UPLOAD_DIR/ftp_reg_file*
37	tst_rmdir
38	pkill vsftpd
39}
40
41setup()
42{
43	tst_require_root
44	tst_check_cmds pkill vsftpd
45	tst_tmpdir
46
47	ipver=${TST_IPV6:-'4'}
48	tst_resm TINFO "run FTP over IPv$ipver"
49
50	trap "tst_brkm TBROK 'test interrupted'" INT
51
52	[ -d "$FTP_UPLOAD_DIR" ] || \
53		tst_brkm TCONF "Start ftp server and set FTP_UPLOAD_DIR var"
54
55	chmod o+w $FTP_UPLOAD_DIR
56
57	getenforce 2> /dev/null | grep -q Enforcing
58	if [ $? -eq 0 ]; then
59		tst_resm TINFO "configuring SELinux FTP parameters"
60		tst_check_cmds chcon setsebool
61		setsebool allow_ftpd_anon_write 1 || \
62			tst_brkm TBROK "Failed to allow ftpd anonymous write"
63		chcon -R -t public_content_rw_t $FTP_UPLOAD_DIR || \
64			tst_brkm TBROK "Failed to apply public_content_rw_t"
65	fi
66
67	tst_resm TINFO "restart vsftpd with custom options"
68	pkill vsftpd
69	touch vsftpd.conf
70
71	local ip_opt="-olisten=YES -olisten_ipv6=NO"
72	[ $TST_IPV6 ] && ip_opt="-olisten=NO -olisten_ipv6=YES"
73
74	upload_opt="-owrite_enable=YES -oanon_upload_enable=YES"
75
76	vsftpd $ip_opt $upload_opt -oanonymous_enable=YES vsftpd.conf
77}
78
79test01()
80{
81	tst_resm TINFO "upload file with size '$UPLOAD_BIGFILESIZE'"
82
83	# Run the script at the remote host
84	tst_rhost_run -s -c "ftp-upload-stress01-rmt $(tst_ipaddr) \
85		$FTP_UPLOAD_URLDIR ftp_file $UPLOAD_BIGFILESIZE"
86
87	tst_resm TPASS "Test is finished successfully"
88}
89
90test02()
91{
92	tst_resm TINFO "Upload data asynchronously in $NS_DURATION sec"
93
94	tst_rhost_run -s -c "ftp-upload-stress02-rmt \
95		$(tst_ipaddr) $FTP_UPLOAD_URLDIR ftp_reg_file \
96		$UPLOAD_REGFILESIZE $NS_DURATION $CONNECTION_TOTAL"
97
98	tst_resm TPASS "Test is finished successfully"
99}
100
101setup
102
103test01
104test02
105
106tst_exit
107