123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750 |
- #!/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 typing import List
- from testenv import Env, CurlClient, LocalClient
- log = logging.getLogger(__name__)
- class TestUpload:
- @pytest.fixture(autouse=True, scope='class')
- def _class_scope(self, env, httpd, nghttpx):
- if env.have_h3():
- nghttpx.start_if_needed()
- env.make_data_file(indir=env.gen_dir, fname="data-10k", fsize=10*1024)
- env.make_data_file(indir=env.gen_dir, fname="data-63k", fsize=63*1024)
- env.make_data_file(indir=env.gen_dir, fname="data-64k", fsize=64*1024)
- env.make_data_file(indir=env.gen_dir, fname="data-100k", fsize=100*1024)
- env.make_data_file(indir=env.gen_dir, fname="data-1m+", fsize=(1024*1024)+1)
- env.make_data_file(indir=env.gen_dir, fname="data-10m", fsize=10*1024*1024)
- httpd.clear_extra_configs()
- httpd.reload()
- # upload small data, check that this is what was echoed
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_01_upload_1_small(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- data = '0123456789'
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]'
- r = curl.http_upload(urls=[url], data=data, alpn_proto=proto)
- r.check_stats(count=1, http_status=200, exitcode=0)
- respdata = open(curl.response_file(0)).readlines()
- assert respdata == [data]
- # upload large data, check that this is what was echoed
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_02_upload_1_large(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- fdata = os.path.join(env.gen_dir, 'data-100k')
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]'
- r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto)
- r.check_stats(count=1, http_status=200, exitcode=0)
- indata = open(fdata).readlines()
- respdata = open(curl.response_file(0)).readlines()
- assert respdata == indata
- # upload data sequentially, check that they were echoed
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_10_upload_sequential(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 stalls here")
- count = 20
- data = '0123456789'
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-{count-1}]'
- r = curl.http_upload(urls=[url], data=data, alpn_proto=proto)
- 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]
- # upload data parallel, check that they were echoed
- @pytest.mark.parametrize("proto", ['h2', 'h3'])
- def test_07_11_upload_parallel(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 stalls here")
- # limit since we use a separate connection in h1
- count = 20
- data = '0123456789'
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/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]
- # upload large data sequentially, check that this is what was echoed
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_12_upload_seq_large(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 stalls here")
- fdata = os.path.join(env.gen_dir, 'data-100k')
- count = 10
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-{count-1}]'
- r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto)
- r.check_response(count=count, http_status=200)
- indata = open(fdata).readlines()
- r.check_stats(count=count, http_status=200, exitcode=0)
- for i in range(count):
- respdata = open(curl.response_file(i)).readlines()
- assert respdata == indata
- # upload very large data sequentially, check that this is what was echoed
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_13_upload_seq_large(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 stalls here")
- fdata = os.path.join(env.gen_dir, 'data-10m')
- count = 2
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-{count-1}]'
- r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto)
- r.check_stats(count=count, http_status=200, exitcode=0)
- indata = open(fdata).readlines()
- for i in range(count):
- respdata = open(curl.response_file(i)).readlines()
- assert respdata == indata
- # upload from stdin, issue #14870
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- @pytest.mark.parametrize("indata", [
- '', '1', '123\n456andsomething\n\n'
- ])
- def test_07_14_upload_stdin(self, env: Env, httpd, nghttpx, proto, indata):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 stalls here")
- count = 1
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?id=[0-{count-1}]'
- r = curl.http_put(urls=[url], data=indata, alpn_proto=proto)
- r.check_stats(count=count, http_status=200, exitcode=0)
- for i in range(count):
- respdata = open(curl.response_file(i)).readlines()
- assert respdata == [f'{len(indata)}']
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_15_hx_put(self, env: Env, httpd, nghttpx, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- count = 2
- upload_size = 128*1024
- url = f'https://localhost:{env.https_port}/curltest/put?id=[0-{count-1}]'
- client = LocalClient(name='hx-upload', env=env)
- if not client.exists():
- pytest.skip(f'example client not built: {client.name}')
- r = client.run(args=[
- '-n', f'{count}', '-S', f'{upload_size}', '-V', proto, url
- ])
- r.check_exit_code(0)
- self.check_downloads(client, [f"{upload_size}"], count)
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_16_hx_put_reuse(self, env: Env, httpd, nghttpx, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- count = 2
- upload_size = 128*1024
- url = f'https://localhost:{env.https_port}/curltest/put?id=[0-{count-1}]'
- client = LocalClient(name='hx-upload', env=env)
- if not client.exists():
- pytest.skip(f'example client not built: {client.name}')
- r = client.run(args=[
- '-n', f'{count}', '-S', f'{upload_size}', '-R', '-V', proto, url
- ])
- r.check_exit_code(0)
- self.check_downloads(client, [f"{upload_size}"], count)
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_17_hx_post_reuse(self, env: Env, httpd, nghttpx, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- count = 2
- upload_size = 128*1024
- url = f'https://localhost:{env.https_port}/curltest/echo?id=[0-{count-1}]'
- client = LocalClient(name='hx-upload', env=env)
- if not client.exists():
- pytest.skip(f'example client not built: {client.name}')
- r = client.run(args=[
- '-n', f'{count}', '-M', 'POST', '-S', f'{upload_size}', '-R', '-V', proto, url
- ])
- r.check_exit_code(0)
- self.check_downloads(client, ["x" * upload_size], count)
- # upload data parallel, check that they were echoed
- @pytest.mark.parametrize("proto", ['h2', 'h3'])
- def test_07_20_upload_parallel(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 stalls here")
- # limit since we use a separate connection in h1
- count = 10
- data = '0123456789'
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/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]
- # upload large data parallel, check that this is what was echoed
- @pytest.mark.parametrize("proto", ['h2', 'h3'])
- def test_07_21_upload_parallel_large(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 stalls here")
- fdata = os.path.join(env.gen_dir, 'data-100k')
- # limit since we use a separate connection in h1
- count = 10
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-{count-1}]'
- r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto,
- extra_args=['--parallel'])
- r.check_response(count=count, http_status=200)
- self.check_download(count, fdata, curl)
- # upload large data parallel to a URL that denies uploads
- @pytest.mark.parametrize("proto", ['h2', 'h3'])
- def test_07_22_upload_parallel_fail(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 stalls here")
- fdata = os.path.join(env.gen_dir, 'data-10m')
- count = 20
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}'\
- f'/curltest/tweak?status=400&delay=5ms&chunks=1&body_error=reset&id=[0-{count-1}]'
- r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto,
- extra_args=['--parallel'])
- exp_exit = 92 if proto == 'h2' else 95
- r.check_stats(count=count, exitcode=exp_exit)
- # PUT 100k
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_30_put_100k(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- fdata = os.path.join(env.gen_dir, 'data-100k')
- count = 1
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?id=[0-{count-1}]'
- r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto,
- extra_args=['--parallel'])
- r.check_stats(count=count, http_status=200, exitcode=0)
- 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
- # PUT 10m
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_31_put_10m(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- fdata = os.path.join(env.gen_dir, 'data-10m')
- count = 1
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?id=[0-{count-1}]&chunk_delay=2ms'
- r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto,
- extra_args=['--parallel'])
- r.check_stats(count=count, http_status=200, exitcode=0)
- 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
- # issue #10591
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_32_issue_10591(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- fdata = os.path.join(env.gen_dir, 'data-10m')
- count = 1
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?id=[0-{count-1}]'
- r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto)
- r.check_stats(count=count, http_status=200, exitcode=0)
- # issue #11157, upload that is 404'ed by server, needs to terminate
- # correctly and not time out on sending
- def test_07_33_issue_11157a(self, env: Env, httpd, nghttpx, repeat):
- proto = 'h2'
- fdata = os.path.join(env.gen_dir, 'data-10m')
- # send a POST to our PUT handler which will send immediately a 404 back
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put'
- curl = CurlClient(env=env)
- r = curl.run_direct(with_stats=True, args=[
- '--resolve', f'{env.authority_for(env.domain1, proto)}:127.0.0.1',
- '--cacert', env.ca.cert_file,
- '--request', 'POST',
- '--max-time', '5', '-v',
- '--url', url,
- '--form', 'idList=12345678',
- '--form', 'pos=top',
- '--form', 'name=mr_test',
- '--form', f'fileSource=@{fdata};type=application/pdf',
- ])
- assert r.exit_code == 0, f'{r}'
- r.check_stats(1, 404)
- # issue #11157, send upload that is slowly read in
- def test_07_33_issue_11157b(self, env: Env, httpd, nghttpx, repeat):
- proto = 'h2'
- fdata = os.path.join(env.gen_dir, 'data-10m')
- # tell our test PUT handler to read the upload more slowly, so
- # that the send buffering and transfer loop needs to wait
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?chunk_delay=2ms'
- curl = CurlClient(env=env)
- r = curl.run_direct(with_stats=True, args=[
- '--verbose', '--trace-config', 'ids,time',
- '--resolve', f'{env.authority_for(env.domain1, proto)}:127.0.0.1',
- '--cacert', env.ca.cert_file,
- '--request', 'PUT',
- '--max-time', '10', '-v',
- '--url', url,
- '--form', 'idList=12345678',
- '--form', 'pos=top',
- '--form', 'name=mr_test',
- '--form', f'fileSource=@{fdata};type=application/pdf',
- ])
- assert r.exit_code == 0, r.dump_logs()
- r.check_stats(1, 200)
- def test_07_34_issue_11194(self, env: Env, httpd, nghttpx, repeat):
- proto = 'h2'
- # tell our test PUT handler to read the upload more slowly, so
- # that the send buffering and transfer loop needs to wait
- fdata = os.path.join(env.gen_dir, 'data-100k')
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put'
- curl = CurlClient(env=env)
- r = curl.run_direct(with_stats=True, args=[
- '--verbose', '--trace-config', 'ids,time',
- '--resolve', f'{env.authority_for(env.domain1, proto)}:127.0.0.1',
- '--cacert', env.ca.cert_file,
- '--request', 'PUT',
- '--digest', '--user', 'test:test',
- '--data-binary', f'@{fdata}',
- '--url', url,
- ])
- assert r.exit_code == 0, r.dump_logs()
- r.check_stats(1, 200)
- # upload large data on a h1 to h2 upgrade
- def test_07_35_h1_h2_upgrade_upload(self, env: Env, httpd, nghttpx, repeat):
- fdata = os.path.join(env.gen_dir, 'data-100k')
- curl = CurlClient(env=env)
- url = f'http://{env.domain1}:{env.http_port}/curltest/echo?id=[0-0]'
- r = curl.http_upload(urls=[url], data=f'@{fdata}', extra_args=[
- '--http2'
- ])
- r.check_response(count=1, http_status=200)
- # apache does not Upgrade on request with a body
- assert r.stats[0]['http_version'] == '1.1', f'{r}'
- indata = open(fdata).readlines()
- respdata = open(curl.response_file(0)).readlines()
- assert respdata == indata
- # upload to a 301,302,303 response
- @pytest.mark.parametrize("redir", ['301', '302', '303'])
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_36_upload_30x(self, env: Env, httpd, nghttpx, repeat, redir, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- data = '0123456789' * 10
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo{redir}?id=[0-0]'
- r = curl.http_upload(urls=[url], data=data, alpn_proto=proto, extra_args=[
- '-L', '--trace-config', 'http/2,http/3'
- ])
- r.check_response(count=1, http_status=200)
- respdata = open(curl.response_file(0)).readlines()
- assert respdata == [] # was transformed to a GET
- # upload to a 307 response
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_37_upload_307(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- data = '0123456789' * 10
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo307?id=[0-0]'
- r = curl.http_upload(urls=[url], data=data, alpn_proto=proto, extra_args=[
- '-L', '--trace-config', 'http/2,http/3'
- ])
- r.check_response(count=1, http_status=200)
- respdata = open(curl.response_file(0)).readlines()
- assert respdata == [data] # was POST again
- # POST form data, yet another code path in transfer
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_38_form_small(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]'
- r = curl.http_form(urls=[url], alpn_proto=proto, form={
- 'name1': 'value1',
- })
- r.check_stats(count=1, http_status=200, exitcode=0)
- # POST data urlencoded, small enough to be sent with request headers
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_39_post_urlenc_small(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- fdata = os.path.join(env.gen_dir, 'data-63k')
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]'
- r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto, extra_args=[
- '--trace-config', 'http/2,http/3'
- ])
- r.check_stats(count=1, http_status=200, exitcode=0)
- indata = open(fdata).readlines()
- respdata = open(curl.response_file(0)).readlines()
- assert respdata == indata
- # POST data urlencoded, large enough to be sent separate from request headers
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_40_post_urlenc_large(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- fdata = os.path.join(env.gen_dir, 'data-64k')
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]'
- r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto, extra_args=[
- '--trace-config', 'http/2,http/3'
- ])
- r.check_stats(count=1, http_status=200, exitcode=0)
- indata = open(fdata).readlines()
- respdata = open(curl.response_file(0)).readlines()
- assert respdata == indata
- # POST data urlencoded, small enough to be sent with request headers
- # and request headers are so large that the first send is larger
- # than our default upload buffer length (64KB).
- # Unfixed, this will fail when run with CURL_DBG_SOCK_WBLOCK=80 most
- # of the time
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_41_post_urlenc_small(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- if proto == 'h3' and env.curl_uses_lib('quiche'):
- pytest.skip("quiche has CWND issues with large requests")
- fdata = os.path.join(env.gen_dir, 'data-63k')
- curl = CurlClient(env=env)
- extra_args = ['--trace-config', 'http/2,http/3']
- # add enough headers so that the first send chunk is > 64KB
- for i in range(63):
- extra_args.extend(['-H', f'x{i:02d}: {"y"*1019}'])
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]'
- r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto, extra_args=extra_args)
- r.check_stats(count=1, http_status=200, exitcode=0)
- indata = open(fdata).readlines()
- respdata = open(curl.response_file(0)).readlines()
- assert respdata == indata
- def check_download(self, count, srcfile, curl):
- for i in range(count):
- dfile = curl.download_file(i)
- assert os.path.exists(dfile)
- if 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}'
- # upload data, pause, let connection die with an incomplete response
- # issues #11769 #13260
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_42a_upload_disconnect(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- client = LocalClient(name='upload-pausing', env=env, timeout=60)
- if not client.exists():
- pytest.skip(f'example client not built: {client.name}')
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]&die_after=0'
- r = client.run(['-V', proto, url])
- if r.exit_code == 18: # PARTIAL_FILE is always ok
- pass
- elif proto == 'h2':
- r.check_exit_code(92) # CURLE_HTTP2_STREAM also ok
- elif proto == 'h3':
- r.check_exit_code(95) # CURLE_HTTP3 also ok
- else:
- r.check_exit_code(18) # will fail as it should
- # upload data, pause, let connection die without any response at all
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_42b_upload_disconnect(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- client = LocalClient(name='upload-pausing', env=env, timeout=60)
- if not client.exists():
- pytest.skip(f'example client not built: {client.name}')
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]&just_die=1'
- r = client.run(['-V', proto, url])
- exp_code = 52 # GOT_NOTHING
- if proto == 'h2' or proto == 'h3':
- exp_code = 0 # we get a 500 from the server
- r.check_exit_code(exp_code) # GOT_NOTHING
- # upload data, pause, let connection die after 100 continue
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_42c_upload_disconnect(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- client = LocalClient(name='upload-pausing', env=env, timeout=60)
- if not client.exists():
- pytest.skip(f'example client not built: {client.name}')
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]&die_after_100=1'
- r = client.run(['-V', proto, url])
- exp_code = 52 # GOT_NOTHING
- if proto == 'h2' or proto == 'h3':
- exp_code = 0 # we get a 500 from the server
- r.check_exit_code(exp_code) # GOT_NOTHING
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_43_upload_denied(self, env: Env, httpd, nghttpx, repeat, proto):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- if proto == 'h3' and env.curl_uses_lib('msh3'):
- pytest.skip("msh3 fails here")
- fdata = os.path.join(env.gen_dir, 'data-10m')
- count = 1
- max_upload = 128 * 1024
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?'\
- f'id=[0-{count-1}]&max_upload={max_upload}'
- r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto,
- extra_args=['--trace-config', 'all'])
- r.check_stats(count=count, http_status=413, exitcode=0)
- # speed limited on put handler
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_50_put_speed_limit(self, env: Env, httpd, nghttpx, proto, repeat):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- count = 1
- fdata = os.path.join(env.gen_dir, 'data-100k')
- up_len = 100 * 1024
- speed_limit = 50 * 1024
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?id=[0-0]'
- r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto,
- with_headers=True, extra_args=[
- '--limit-rate', f'{speed_limit}'
- ])
- r.check_response(count=count, http_status=200)
- assert r.responses[0]['header']['received-length'] == f'{up_len}', f'{r.responses[0]}'
- up_speed = r.stats[0]['speed_upload']
- assert (speed_limit * 0.5) <= up_speed <= (speed_limit * 1.5), f'{r.stats[0]}'
- # speed limited on echo handler
- @pytest.mark.parametrize("proto", ['http/1.1', 'h2', 'h3'])
- def test_07_51_echo_speed_limit(self, env: Env, httpd, nghttpx, proto, repeat):
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- count = 1
- fdata = os.path.join(env.gen_dir, 'data-100k')
- speed_limit = 50 * 1024
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]'
- r = curl.http_upload(urls=[url], data=f'@{fdata}', alpn_proto=proto,
- with_headers=True, extra_args=[
- '--limit-rate', f'{speed_limit}'
- ])
- r.check_response(count=count, http_status=200)
- up_speed = r.stats[0]['speed_upload']
- assert (speed_limit * 0.5) <= up_speed <= (speed_limit * 1.5), f'{r.stats[0]}'
- # upload larger data, triggering "Expect: 100-continue" code paths
- @pytest.mark.parametrize("proto", ['http/1.1'])
- def test_07_60_upload_exp100(self, env: Env, httpd, nghttpx, repeat, proto):
- fdata = os.path.join(env.gen_dir, 'data-1m+')
- read_delay = 1
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?id=[0-0]'\
- f'&read_delay={read_delay}s'
- r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto, extra_args=[
- '--expect100-timeout', f'{read_delay+1}'
- ])
- r.check_stats(count=1, http_status=200, exitcode=0)
- # upload larger data, triggering "Expect: 100-continue" code paths
- @pytest.mark.parametrize("proto", ['http/1.1'])
- def test_07_61_upload_exp100_timeout(self, env: Env, httpd, nghttpx, repeat, proto):
- fdata = os.path.join(env.gen_dir, 'data-1m+')
- read_delay = 2
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/put?id=[0-0]'\
- f'&read_delay={read_delay}s'
- r = curl.http_put(urls=[url], fdata=fdata, alpn_proto=proto, extra_args=[
- '--expect100-timeout', f'{read_delay-1}'
- ])
- r.check_stats(count=1, http_status=200, exitcode=0)
- # issue #15688 when posting a form and cr_mime_read() is called with
- # length < 4, we did not progress
- @pytest.mark.parametrize("proto", ['http/1.1'])
- def test_07_62_upload_issue_15688(self, env: Env, httpd, proto):
- # this length leads to (including multipart formatting) to a
- # client reader invocation with length 1.
- upload_len = 196169
- fname = f'data-{upload_len}'
- env.make_data_file(indir=env.gen_dir, fname=fname, fsize=upload_len)
- fdata = os.path.join(env.gen_dir, fname)
- curl = CurlClient(env=env)
- url = f'https://{env.authority_for(env.domain1, proto)}/curltest/echo?id=[0-0]'
- r = curl.http_form(urls=[url], form={
- 'file': f'@{fdata}',
- }, alpn_proto=proto, extra_args=[
- '--max-time', '10'
- ])
- r.check_stats(count=1, http_status=200, exitcode=0)
- # nghttpx is the only server we have that supports TLS early data and
- # has a limit of 16k it announces
- @pytest.mark.skipif(condition=not Env.have_nghttpx(), reason="no nghttpx")
- @pytest.mark.parametrize("proto,upload_size,exp_early", [
- ['http/1.1', 100, 203], # headers+body
- ['http/1.1', 10*1024, 10345], # headers+body
- ['http/1.1', 32*1024, 16384], # headers+body, limited by server max
- ['h2', 10*1024, 10378], # headers+body
- ['h2', 32*1024, 16384], # headers+body, limited by server max
- ['h3', 1024, 1126], # headers+body (app data)
- ['h3', 1024 * 1024, 131177], # headers+body (long app data). The 0RTT
- # size is limited by our sendbuf size
- # of 128K.
- ])
- def test_07_70_put_earlydata(self, env: Env, httpd, nghttpx, proto, upload_size, exp_early):
- if not env.curl_uses_lib('gnutls'):
- pytest.skip('TLS earlydata only implemented in GnuTLS')
- if proto == 'h3' and not env.have_h3():
- pytest.skip("h3 not supported")
- count = 2
- # we want this test to always connect to nghttpx, since it is
- # the only server we have that supports TLS earlydata
- port = env.port_for(proto)
- if proto != 'h3':
- port = env.nghttpx_https_port
- url = f'https://{env.domain1}:{port}/curltest/put?id=[0-{count-1}]'
- client = LocalClient(name='hx-upload', 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
- '-l', # announce upload length, no 'Expect: 100'
- '-S', f'{upload_size}',
- '-r', f'{env.domain1}:{port}:127.0.0.1',
- '-V', proto, url
- ])
- r.check_exit_code(0)
- self.check_downloads(client, [f"{upload_size}"], 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}'
- assert earlydata[1] == exp_early, f'{earlydata}'
- def check_downloads(self, client, source: List[str], count: int,
- complete: bool = True):
- for i in range(count):
- dfile = client.download_file(i)
- assert os.path.exists(dfile)
- if complete:
- diff = "".join(difflib.unified_diff(a=source,
- b=open(dfile).readlines(),
- fromfile='-',
- tofile=dfile,
- n=1))
- assert not diff, f'download {dfile} differs:\n{diff}'
|