| #!/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 re |
| import pytest |
| |
| from testenv import Env, CurlClient, Caddy, LocalClient |
| |
| |
| log = logging.getLogger(__name__) |
| |
| |
| @pytest.mark.skipif(condition=not Env.has_caddy(), reason="missing caddy") |
| @pytest.mark.skipif(condition=not Env.have_ssl_curl(), reason="curl without SSL") |
| class TestCaddy: |
| |
| @pytest.fixture(autouse=True, scope='class') |
| def caddy(self, env): |
| caddy = Caddy(env=env) |
| assert caddy.initial_start() |
| yield caddy |
| caddy.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, caddy): |
| self._make_docs_file(docs_dir=caddy.docs_dir, fname='data10k.data', fsize=10*1024) |
| self._make_docs_file(docs_dir=caddy.docs_dir, fname='data1.data', fsize=1024*1024) |
| self._make_docs_file(docs_dir=caddy.docs_dir, fname='data5.data', fsize=5*1024*1024) |
| self._make_docs_file(docs_dir=caddy.docs_dir, fname='data10.data', fsize=10*1024*1024) |
| self._make_docs_file(docs_dir=caddy.docs_dir, fname='data100.data', fsize=100*1024*1024) |
| env.make_data_file(indir=env.gen_dir, fname="data-10m", fsize=10*1024*1024) |
| |
| # download 1 file |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_08_01_download_1(self, env: Env, caddy: Caddy, proto): |
| if proto == 'h3' and not env.have_h3_curl(): |
| pytest.skip("h3 not supported in curl") |
| curl = CurlClient(env=env) |
| url = f'https://{env.domain1}:{caddy.port}/data.json' |
| r = curl.http_download(urls=[url], alpn_proto=proto) |
| r.check_response(count=1, http_status=200) |
| |
| # download 1MB files sequentially |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_08_02_download_1mb_sequential(self, env: Env, caddy: Caddy, proto): |
| if proto == 'h3' and not env.have_h3_curl(): |
| pytest.skip("h3 not supported in curl") |
| count = 50 |
| curl = CurlClient(env=env) |
| urln = f'https://{env.domain1}:{caddy.port}/data1.data?[0-{count-1}]' |
| r = curl.http_download(urls=[urln], alpn_proto=proto) |
| r.check_response(count=count, http_status=200, connect_count=1) |
| |
| # download 1MB files parallel |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_08_03_download_1mb_parallel(self, env: Env, caddy: Caddy, proto): |
| if proto == 'h3' and not env.have_h3_curl(): |
| pytest.skip("h3 not supported in curl") |
| count = 20 |
| curl = CurlClient(env=env) |
| urln = f'https://{env.domain1}:{caddy.port}/data1.data?[0-{count-1}]' |
| r = curl.http_download(urls=[urln], alpn_proto=proto, extra_args=[ |
| '--parallel' |
| ]) |
| r.check_response(count=count, http_status=200) |
| if proto == 'http/1.1': |
| # http/1.1 parallel transfers will open multiple connections |
| assert r.total_connects > 1, r.dump_logs() |
| else: |
| assert r.total_connects == 1, r.dump_logs() |
| |
| # download 5MB files sequentially |
| @pytest.mark.skipif(condition=Env().slow_network, reason="not suitable for slow network tests") |
| @pytest.mark.skipif(condition=Env().ci_run, reason="not suitable for CI runs") |
| @pytest.mark.parametrize("proto", ['h2', 'h3']) |
| def test_08_04a_download_10mb_sequential(self, env: Env, caddy: Caddy, proto): |
| if proto == 'h3' and not env.have_h3_curl(): |
| pytest.skip("h3 not supported in curl") |
| count = 40 |
| curl = CurlClient(env=env) |
| urln = f'https://{env.domain1}:{caddy.port}/data5.data?[0-{count-1}]' |
| r = curl.http_download(urls=[urln], alpn_proto=proto) |
| r.check_response(count=count, http_status=200, connect_count=1) |
| |
| # download 10MB files sequentially |
| @pytest.mark.skipif(condition=Env().slow_network, reason="not suitable for slow network tests") |
| @pytest.mark.skipif(condition=Env().ci_run, reason="not suitable for CI runs") |
| @pytest.mark.parametrize("proto", ['h2', 'h3']) |
| def test_08_04b_download_10mb_sequential(self, env: Env, caddy: Caddy, proto): |
| if proto == 'h3' and not env.have_h3_curl(): |
| pytest.skip("h3 not supported in curl") |
| count = 20 |
| curl = CurlClient(env=env) |
| urln = f'https://{env.domain1}:{caddy.port}/data10.data?[0-{count-1}]' |
| r = curl.http_download(urls=[urln], alpn_proto=proto) |
| r.check_response(count=count, http_status=200, connect_count=1) |
| |
| # download 10MB files parallel |
| @pytest.mark.skipif(condition=Env().slow_network, reason="not suitable for slow network tests") |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| @pytest.mark.skipif(condition=Env().ci_run, reason="not suitable for CI runs") |
| def test_08_05_download_1mb_parallel(self, env: Env, caddy: Caddy, proto): |
| if proto == 'h3' and not env.have_h3_curl(): |
| pytest.skip("h3 not supported in curl") |
| if proto == 'http/1.1' and env.curl_uses_lib('mbedtls'): |
| pytest.skip("mbedtls 3.6.0 fails on 50 connections with: " |
| "ssl_handshake returned: (-0x7F00) SSL - Memory allocation failed") |
| count = 50 |
| curl = CurlClient(env=env) |
| urln = f'https://{env.domain1}:{caddy.port}/data10.data?[0-{count-1}]' |
| r = curl.http_download(urls=[urln], alpn_proto=proto, extra_args=[ |
| '--parallel' |
| ]) |
| r.check_response(count=count, http_status=200) |
| if proto == 'http/1.1': |
| # http/1.1 parallel transfers will open multiple connections |
| assert r.total_connects > 1, r.dump_logs() |
| else: |
| assert r.total_connects == 1, r.dump_logs() |
| |
| # post data parallel, check that they were echoed |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_08_06_post_parallel(self, env: Env, httpd, caddy, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| # limit since we use a separate connection in h1 |
| count = 20 |
| data = '0123456789' |
| curl = CurlClient(env=env) |
| url = f'https://{env.domain2}:{caddy.port}/curltest/echo?id=[0-{count-1}]' |
| r = curl.http_upload(urls=[url], data=data, alpn_proto=proto, |
| extra_args=['--parallel']) |
| r.check_stats(count=count, http_status=200, exitcode=0) |
| for i in range(count): |
| respdata = open(curl.response_file(i)).readlines() |
| assert respdata == [data] |
| |
| # put large file, check that they length were echoed |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_08_07_put_large(self, env: Env, httpd, caddy, proto): |
| if proto == 'h3' and not env.have_h3(): |
| pytest.skip("h3 not supported") |
| # limit since we use a separate connection in h1< |
| count = 1 |
| fdata = os.path.join(env.gen_dir, 'data-10m') |
| curl = CurlClient(env=env) |
| url = f'https://{env.domain2}:{caddy.port}/curltest/put?id=[0-{count-1}]' |
| r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto) |
| exp_data = [f'{os.path.getsize(fdata)}'] |
| r.check_response(count=count, http_status=200) |
| for i in range(count): |
| respdata = open(curl.response_file(i)).readlines() |
| assert respdata == exp_data |
| |
| @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3']) |
| def test_08_08_earlydata(self, env: Env, httpd, caddy, proto): |
| if not env.curl_can_early_data(): |
| pytest.skip('TLS earlydata not implemented') |
| if proto == 'h3' and \ |
| (not env.have_h3() or not env.curl_can_h3_early_data()): |
| pytest.skip("h3 not supported") |
| count = 2 |
| docname = 'data10k.data' |
| url = f'https://{env.domain1}:{caddy.port}/{docname}' |
| client = LocalClient(name='cli_hx_download', env=env) |
| if not client.exists(): |
| pytest.skip(f'example client not built: {client.name}') |
| r = client.run(args=[ |
| '-n', f'{count}', |
| '-e', # use TLS earlydata |
| '-f', # forbid reuse of connections |
| '-r', f'{env.domain1}:{caddy.port}:127.0.0.1', |
| '-V', proto, url |
| ]) |
| r.check_exit_code(0) |
| srcfile = os.path.join(caddy.docs_dir, docname) |
| self.check_downloads(client, srcfile, count) |
| earlydata = {} |
| for line in r.trace_lines: |
| m = re.match(r'^\[t-(\d+)] EarlyData: (-?\d+)', line) |
| if m: |
| earlydata[int(m.group(1))] = int(m.group(2)) |
| assert earlydata[0] == 0, f'{earlydata}' |
| if proto == 'h3': |
| assert earlydata[1] == 113, f'{earlydata}' |
| else: |
| # Caddy does not support early data on TCP |
| assert earlydata[1] == 0, f'{earlydata}' |
| |
| 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}' |
