| #!/usr/bin/env python3 |
| # -*- coding: utf-8 -*- |
| #*************************************************************************** |
| # _ _ ____ _ |
| # Project ___| | | | _ \| | |
| # / __| | | | |_) | | |
| # | (__| |_| | _ <| |___ |
| # \___|\___/|_| \_\_____| |
| # |
| # Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al. |
| # |
| # This software is licensed as described in the file COPYING, which |
| # you should have received as part of this distribution. The terms |
| # are also available at https://curl.se/docs/copyright.html. |
| # |
| # You may opt to use, copy, modify, merge, publish, distribute and/or sell |
| # copies of the Software, and permit persons to whom the Software is |
| # furnished to do so, under the terms of the COPYING file. |
| # |
| # This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY |
| # KIND, either express or implied. |
| # |
| # SPDX-License-Identifier: curl |
| # |
| ########################################################################### |
| # |
| import difflib |
| import filecmp |
| import logging |
| import os |
| import shutil |
| import pytest |
| |
| from testenv import Env, CurlClient, VsFTPD |
| |
| |
| log = logging.getLogger(__name__) |
| |
| |
| @pytest.mark.skipif(condition=not Env.has_vsftpd(), reason="missing vsftpd") |
| class TestFtpsVsFTPD: |
| |
| SUPPORTS_SSL = True |
| |
| @pytest.fixture(autouse=True, scope='class') |
| def vsftpds(self, env): |
| if not TestFtpsVsFTPD.SUPPORTS_SSL: |
| pytest.skip('vsftpd does not seem to support SSL') |
| vsftpds = VsFTPD(env=env, with_ssl=True, ssl_implicit=True) |
| if not vsftpds.initial_start(): |
| vsftpds.stop() |
| TestFtpsVsFTPD.SUPPORTS_SSL = False |
| pytest.skip('vsftpd does not seem to support SSL') |
| yield vsftpds |
| vsftpds.stop() |
| |
| def _make_docs_file(self, docs_dir: str, fname: str, fsize: int): |
| fpath = os.path.join(docs_dir, fname) |
| data1k = 1024*'x' |
| flen = 0 |
| with open(fpath, 'w') as fd: |
| while flen < fsize: |
| fd.write(data1k) |
| flen += len(data1k) |
| return flen |
| |
| @pytest.fixture(autouse=True, scope='class') |
| def _class_scope(self, env, vsftpds): |
| if os.path.exists(vsftpds.docs_dir): |
| shutil.rmtree(vsftpds.docs_dir) |
| if not os.path.exists(vsftpds.docs_dir): |
| os.makedirs(vsftpds.docs_dir) |
| self._make_docs_file(docs_dir=vsftpds.docs_dir, fname='data-1k', fsize=1024) |
| self._make_docs_file(docs_dir=vsftpds.docs_dir, fname='data-10k', fsize=10*1024) |
| self._make_docs_file(docs_dir=vsftpds.docs_dir, fname='data-1m', fsize=1024*1024) |
| self._make_docs_file(docs_dir=vsftpds.docs_dir, fname='data-10m', fsize=10*1024*1024) |
| env.make_data_file(indir=env.gen_dir, fname="upload-1k", fsize=1024) |
| env.make_data_file(indir=env.gen_dir, fname="upload-100k", fsize=100*1024) |
| env.make_data_file(indir=env.gen_dir, fname="upload-1m", fsize=1024*1024) |
| |
| def test_32_01_list_dir(self, env: Env, vsftpds: VsFTPD): |
| curl = CurlClient(env=env) |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/' |
| r = curl.ftp_get(urls=[url], with_stats=True) |
| r.check_stats(count=1, http_status=226) |
| lines = open(os.path.join(curl.run_dir, 'download_#1.data')).readlines() |
| assert len(lines) == 4, f'list: {lines}' |
| |
| # download 1 file, no SSL |
| @pytest.mark.parametrize("docname", [ |
| 'data-1k', 'data-1m', 'data-10m' |
| ]) |
| def test_32_02_download_1(self, env: Env, vsftpds: VsFTPD, docname): |
| curl = CurlClient(env=env) |
| srcfile = os.path.join(vsftpds.docs_dir, f'{docname}') |
| count = 1 |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/{docname}?[0-{count-1}]' |
| r = curl.ftp_get(urls=[url], with_stats=True) |
| r.check_stats(count=count, http_status=226) |
| self.check_downloads(curl, srcfile, count) |
| |
| @pytest.mark.parametrize("docname", [ |
| 'data-1k', 'data-1m', 'data-10m' |
| ]) |
| def test_32_03_download_10_serial(self, env: Env, vsftpds: VsFTPD, docname): |
| curl = CurlClient(env=env) |
| srcfile = os.path.join(vsftpds.docs_dir, f'{docname}') |
| count = 10 |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/{docname}?[0-{count-1}]' |
| r = curl.ftp_get(urls=[url], with_stats=True) |
| r.check_stats(count=count, http_status=226) |
| self.check_downloads(curl, srcfile, count) |
| assert r.total_connects == count + 1, 'should reuse the control conn' |
| |
| # 2 serial transfers, first with 'ftps://' and second with 'ftp://' |
| # we want connection reuse in this case |
| def test_32_03b_ftp_compat_ftps(self, env: Env, vsftpds: VsFTPD): |
| curl = CurlClient(env=env) |
| docname = 'data-1k' |
| count = 2 |
| url1= f'ftps://{env.ftp_domain}:{vsftpds.port}/{docname}' |
| url2 = f'ftp://{env.ftp_domain}:{vsftpds.port}/{docname}' |
| r = curl.ftp_get(urls=[url1, url2], with_stats=True) |
| r.check_stats(count=count, http_status=226) |
| assert r.total_connects == count + 1, 'should reuse the control conn' |
| |
| @pytest.mark.parametrize("docname", [ |
| 'data-1k', 'data-1m', 'data-10m' |
| ]) |
| def test_32_04_download_10_parallel(self, env: Env, vsftpds: VsFTPD, docname): |
| curl = CurlClient(env=env) |
| srcfile = os.path.join(vsftpds.docs_dir, f'{docname}') |
| count = 10 |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/{docname}?[0-{count-1}]' |
| r = curl.ftp_get(urls=[url], with_stats=True, extra_args=[ |
| '--parallel' |
| ]) |
| r.check_stats(count=count, http_status=226) |
| self.check_downloads(curl, srcfile, count) |
| assert r.total_connects > count + 1, 'should have used several control conns' |
| |
| @pytest.mark.parametrize("docname", [ |
| 'upload-1k', 'upload-100k', 'upload-1m' |
| ]) |
| def test_32_05_upload_1(self, env: Env, vsftpds: VsFTPD, docname): |
| curl = CurlClient(env=env) |
| srcfile = os.path.join(env.gen_dir, docname) |
| dstfile = os.path.join(vsftpds.docs_dir, docname) |
| self._rmf(dstfile) |
| count = 1 |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/' |
| r = curl.ftp_upload(urls=[url], fupload=f'{srcfile}', with_stats=True) |
| r.check_stats(count=count, http_status=226) |
| self.check_upload(env, vsftpds, docname=docname) |
| |
| def _rmf(self, path): |
| if os.path.exists(path): |
| return os.remove(path) |
| |
| # check with `tcpdump` if curl causes any TCP RST packets |
| @pytest.mark.skipif(condition=not Env.tcpdump(), reason="tcpdump not available") |
| @pytest.mark.skipif(condition=not Env.curl_is_debug(), reason="needs curl debug") |
| def test_32_06_shutdownh_download(self, env: Env, vsftpds: VsFTPD): |
| docname = 'data-1k' |
| curl = CurlClient(env=env) |
| count = 1 |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/{docname}?[0-{count-1}]' |
| r = curl.ftp_get(urls=[url], with_stats=True, with_tcpdump=True) |
| r.check_stats(count=count, http_status=226) |
| # vsftp closes control connection without niceties, |
| # look only at ports from DATA connection. |
| data_ports = vsftpds.get_data_ports(r) |
| assert len(data_ports), f'unable to find FTP data port connected to\n{r.dump_logs()}' |
| assert len(r.tcpdump.get_rsts(ports=data_ports)) == 0, 'Unexpected TCP RST packets' |
| |
| # check with `tcpdump` if curl causes any TCP RST packets |
| @pytest.mark.skipif(condition=not Env.tcpdump(), reason="tcpdump not available") |
| @pytest.mark.skipif(condition=not Env.curl_is_debug(), reason="needs curl debug") |
| def test_32_07_shutdownh_upload(self, env: Env, vsftpds: VsFTPD): |
| docname = 'upload-1k' |
| curl = CurlClient(env=env) |
| srcfile = os.path.join(env.gen_dir, docname) |
| dstfile = os.path.join(vsftpds.docs_dir, docname) |
| self._rmf(dstfile) |
| count = 1 |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/' |
| r = curl.ftp_upload(urls=[url], fupload=f'{srcfile}', with_stats=True, with_tcpdump=True) |
| r.check_stats(count=count, http_status=226) |
| # vsftp closes control connection without niceties, |
| # look only at ports from DATA connection. |
| data_ports = vsftpds.get_data_ports(r) |
| assert len(data_ports), f'unable to find FTP data port connected to\n{r.dump_logs()}' |
| assert len(r.tcpdump.get_rsts(ports=data_ports)) == 0, 'Unexpected TCP RST packets' |
| |
| def test_32_08_upload_ascii(self, env: Env, vsftpds: VsFTPD): |
| docname = 'upload-ascii' |
| line_length = 21 |
| srcfile = os.path.join(env.gen_dir, docname) |
| dstfile = os.path.join(vsftpds.docs_dir, docname) |
| env.make_data_file(indir=env.gen_dir, fname=docname, fsize=100*1024, |
| line_length=line_length) |
| srcsize = os.path.getsize(srcfile) |
| self._rmf(dstfile) |
| count = 1 |
| curl = CurlClient(env=env) |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/' |
| r = curl.ftp_upload(urls=[url], fupload=f'{srcfile}', with_stats=True, |
| extra_args=['--use-ascii']) |
| r.check_stats(count=count, http_status=226) |
| # expect the uploaded file to be number of converted newlines larger |
| dstsize = os.path.getsize(dstfile) |
| newlines = len(open(srcfile).readlines()) |
| assert (srcsize + newlines) == dstsize, \ |
| f'expected source with {newlines} lines to be that much larger,'\ |
| f'instead srcsize={srcsize}, upload size={dstsize}, diff={dstsize-srcsize}' |
| |
| def test_32_08_active_download(self, env: Env, vsftpds: VsFTPD): |
| docname = 'data-10k' |
| curl = CurlClient(env=env) |
| srcfile = os.path.join(vsftpds.docs_dir, f'{docname}') |
| count = 1 |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/{docname}?[0-{count-1}]' |
| r = curl.ftp_get(urls=[url], with_stats=True, extra_args=[ |
| '--ftp-port', '127.0.0.1' |
| ]) |
| r.check_stats(count=count, http_status=226) |
| self.check_downloads(curl, srcfile, count) |
| |
| def test_32_09_active_upload(self, env: Env, vsftpds: VsFTPD): |
| docname = 'upload-1k' |
| curl = CurlClient(env=env) |
| srcfile = os.path.join(env.gen_dir, docname) |
| dstfile = os.path.join(vsftpds.docs_dir, docname) |
| self._rmf(dstfile) |
| count = 1 |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/' |
| r = curl.ftp_upload(urls=[url], fupload=f'{srcfile}', with_stats=True, extra_args=[ |
| '--ftp-port', '127.0.0.1' |
| ]) |
| r.check_stats(count=count, http_status=226) |
| self.check_upload(env, vsftpds, docname=docname) |
| |
| @pytest.mark.parametrize("indata", [ |
| pytest.param('1234567890', id='10-bytes'), |
| pytest.param('', id='0-bytes'), |
| ]) |
| def test_32_10_upload_stdin(self, env: Env, vsftpds: VsFTPD, indata): |
| curl = CurlClient(env=env) |
| docname = "upload_31_10" |
| dstfile = os.path.join(vsftpds.docs_dir, docname) |
| self._rmf(dstfile) |
| count = 1 |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/{docname}' |
| r = curl.ftp_upload(urls=[url], updata=indata, with_stats=True) |
| r.check_stats(count=count, http_status=226) |
| assert os.path.exists(dstfile) |
| destdata = open(dstfile).readlines() |
| expdata = [indata] if len(indata) else [] |
| assert expdata == destdata, f'expected: {expdata}, got: {destdata}' |
| |
| def test_32_11_download_non_existing(self, env: Env, vsftpds: VsFTPD): |
| curl = CurlClient(env=env) |
| url = f'ftps://{env.ftp_domain}:{vsftpds.port}/does-not-exist' |
| r = curl.ftp_get(urls=[url], with_stats=True) |
| r.check_exit_code(78) |
| r.check_stats(count=1, exitcode=78) |
| |
| def check_downloads(self, client, srcfile: str, count: int, |
| complete: bool = True): |
| for i in range(count): |
| dfile = client.download_file(i) |
| assert os.path.exists(dfile) |
| if complete and not filecmp.cmp(srcfile, dfile, shallow=False): |
| diff = "".join(difflib.unified_diff(a=open(srcfile).readlines(), |
| b=open(dfile).readlines(), |
| fromfile=srcfile, |
| tofile=dfile, |
| n=1)) |
| assert False, f'download {dfile} differs:\n{diff}' |
| |
| def check_upload(self, env, vsftpd: VsFTPD, docname): |
| srcfile = os.path.join(env.gen_dir, docname) |
| dstfile = os.path.join(vsftpd.docs_dir, docname) |
| assert os.path.exists(srcfile) |
| assert os.path.exists(dstfile) |
| if not filecmp.cmp(srcfile, dstfile, shallow=False): |
| diff = "".join(difflib.unified_diff(a=open(srcfile).readlines(), |
| b=open(dstfile).readlines(), |
| fromfile=srcfile, |
| tofile=dstfile, |
| n=1)) |
| assert False, f'upload {dstfile} differs:\n{diff}' |
