mirror of
https://github.com/curl/curl.git
synced 2026-04-11 12:01:42 +08:00
0a7e1982fe
Test that connecting to a server that immediately closes the connection produces an error instead of hanging/timing out.
215 lines
9.6 KiB
Python
215 lines
9.6 KiB
Python
#!/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 logging
|
|
import os
|
|
import socket
|
|
import threading
|
|
|
|
import pytest
|
|
from testenv import CurlClient, Env
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
@pytest.mark.skipif(condition=not Env.httpd_is_at_least('2.4.55'),
|
|
reason=f"httpd version too old for this: {Env.httpd_version()}")
|
|
class TestErrors:
|
|
|
|
# download 1 file, check that we get CURLE_PARTIAL_FILE
|
|
@pytest.mark.parametrize("proto", Env.http_protos())
|
|
def test_05_01_partial_1(self, env: Env, httpd, nghttpx, proto):
|
|
if proto == 'h3' and env.curl_uses_lib('quiche') and \
|
|
not env.curl_lib_version_at_least('quiche', '0.24.8'):
|
|
pytest.skip("quiche issue #2277 not fixed")
|
|
count = 1
|
|
curl = CurlClient(env=env)
|
|
urln = f'https://{env.authority_for(env.domain1, proto)}' \
|
|
f'/curltest/tweak?id=[0-{count - 1}]'\
|
|
'&chunks=3&chunk_size=16000&body_error=reset'
|
|
r = curl.http_download(urls=[urln], alpn_proto=proto, extra_args=[
|
|
'--retry', '0'
|
|
])
|
|
r.check_exit_code(False)
|
|
invalid_stats = []
|
|
for idx, s in enumerate(r.stats):
|
|
if 'exitcode' not in s or s['exitcode'] not in [18, 56, 92, 95]:
|
|
invalid_stats.append(f'request {idx} exit with {s["exitcode"]}')
|
|
assert len(invalid_stats) == 0, f'failed: {invalid_stats}'
|
|
|
|
# download files, check that we get CURLE_PARTIAL_FILE for all
|
|
@pytest.mark.parametrize("proto", Env.http_mplx_protos())
|
|
def test_05_02_partial_20(self, env: Env, httpd, nghttpx, proto):
|
|
if proto == 'h3' and env.curl_uses_lib('quiche') and \
|
|
not env.curl_lib_version_at_least('quiche', '0.24.8'):
|
|
pytest.skip("quiche issue #2277 not fixed")
|
|
count = 20
|
|
curl = CurlClient(env=env)
|
|
urln = f'https://{env.authority_for(env.domain1, proto)}' \
|
|
f'/curltest/tweak?id=[0-{count - 1}]'\
|
|
'&chunks=5&chunk_size=16000&body_error=reset'
|
|
r = curl.http_download(urls=[urln], alpn_proto=proto, extra_args=[
|
|
'--retry', '0', '--parallel',
|
|
])
|
|
r.check_exit_code(False)
|
|
assert len(r.stats) == count, f'did not get all stats: {r}'
|
|
invalid_stats = []
|
|
for idx, s in enumerate(r.stats):
|
|
if 'exitcode' not in s or s['exitcode'] not in [18, 55, 56, 92, 95]:
|
|
invalid_stats.append(f'request {idx} exit with {s["exitcode"]}\n{r.dump_logs()}')
|
|
assert len(invalid_stats) == 0, f'failed: {invalid_stats}\n{r.dump_logs()}'
|
|
|
|
# access a resource that, on h2, RST the stream with HTTP_1_1_REQUIRED
|
|
@pytest.mark.skipif(condition=not Env.have_h2_curl(), reason="curl without h2")
|
|
def test_05_03_required(self, env: Env, httpd, nghttpx):
|
|
curl = CurlClient(env=env)
|
|
proto = 'http/1.1'
|
|
urln = f'https://{env.authority_for(env.domain1, proto)}/curltest/1_1'
|
|
r = curl.http_download(urls=[urln], alpn_proto=proto)
|
|
r.check_exit_code(0)
|
|
r.check_response(http_status=200, count=1)
|
|
proto = 'h2'
|
|
urln = f'https://{env.authority_for(env.domain1, proto)}/curltest/1_1'
|
|
r = curl.http_download(urls=[urln], alpn_proto=proto)
|
|
r.check_exit_code(0)
|
|
r.check_response(http_status=200, count=1)
|
|
# check that we did a downgrade
|
|
assert r.stats[0]['http_version'] == '1.1', r.dump_logs()
|
|
|
|
# On the URL used here, Apache is doing an "unclean" TLS shutdown,
|
|
# meaning it sends no shutdown notice and just closes TCP.
|
|
# The HTTP response delivers a body without Content-Length. We expect:
|
|
# - http/1.0 to fail since it relies on a clean connection close to
|
|
# detect the end of the body
|
|
# - http/1.1 to work since it will used "chunked" transfer encoding
|
|
# and stop receiving when that signals the end
|
|
# - h2 to work since it will signal the end of the response before
|
|
# and not see the "unclean" close either
|
|
@pytest.mark.parametrize("proto", ['http/1.0', 'http/1.1', 'h2'])
|
|
def test_05_04_unclean_tls_shutdown(self, env: Env, httpd, nghttpx, proto):
|
|
if proto == 'h2' and not env.have_h2_curl():
|
|
pytest.skip("h2 not supported")
|
|
if proto == 'h3' and not env.have_h3():
|
|
pytest.skip("h3 not supported")
|
|
count = 10 if proto == 'h2' else 1
|
|
curl = CurlClient(env=env)
|
|
url = f'https://{env.authority_for(env.domain1, proto)}'\
|
|
f'/curltest/shutdown_unclean?id=[0-{count-1}]&chunks=4'
|
|
r = curl.http_download(urls=[url], alpn_proto=proto, extra_args=[
|
|
'--parallel', '--trace-config', 'ssl'
|
|
])
|
|
if proto == 'http/1.0':
|
|
# we are inconsistent if we fail or not in missing TLS shutdown
|
|
# openssl code ignore such errors intentionally in non-debug builds
|
|
r.check_exit_code(56)
|
|
else:
|
|
r.check_exit_code(0)
|
|
r.check_response(http_status=200, count=count)
|
|
|
|
# Make a connect with a fail for '::1' and wrong certificate for '127.0.0.1'
|
|
# The error message reported needs to be from the certificate.
|
|
# verifies issue #20608
|
|
@pytest.mark.parametrize("proto", Env.http_h1_h2_protos())
|
|
def test_05_05_failed_peer(self, env: Env, proto, httpd, nghttpx):
|
|
domain = f'invalid.{env.tld}'
|
|
url = f'https://{domain}:{env.port_for(proto)}/'
|
|
run_env = os.environ.copy()
|
|
run_env['CURL_DBG_SOCK_FAIL_IPV6'] = '1'
|
|
curl = CurlClient(env=env, run_env=run_env)
|
|
r = curl.http_download(urls=[url], alpn_proto=proto, extra_args=[
|
|
'--resolve', f'{domain}:{env.port_for(proto)}:::1,127.0.0.1',
|
|
])
|
|
assert r.exit_code == 60, f'{r}'
|
|
assert r.stats[0]['errormsg'] != 'CURL_DBG_SOCK_FAIL_IPV6: failed to open socket'
|
|
|
|
# Get, retry on 502
|
|
def test_05_06_retry_502(self, env: Env, httpd, nghttpx):
|
|
proto = 'http/1.1'
|
|
curl = CurlClient(env=env)
|
|
url = f'https://{env.authority_for(env.domain1, proto)}/curltest/tweak?status=502'
|
|
r = curl.http_download(urls=[url], alpn_proto=proto, extra_args=[
|
|
'--retry', '2', '--retry-all-errors', '--retry-delay', '1',
|
|
])
|
|
r.check_response(http_status=502)
|
|
assert r.stats[0]['num_retries'] == 2, f'{r}'
|
|
# curious, since curl does the retries, it finds the previous
|
|
# connection in the cache and reports that no connects were done
|
|
assert r.stats[0]['num_connects'] == 0, f'{r}'
|
|
|
|
# Get, retry on 502 in parallel mode
|
|
def test_05_07_retry_502_parallel(self, env: Env, httpd, nghttpx):
|
|
proto = 'http/1.1'
|
|
curl = CurlClient(env=env)
|
|
url = f'https://{env.authority_for(env.domain1, proto)}/curltest/tweak?status=502'
|
|
r = curl.http_download(urls=[url], alpn_proto=proto, extra_args=[
|
|
'--retry', '2', '--retry-all-errors', '--retry-delay', '1', '--parallel'
|
|
])
|
|
r.check_response(http_status=502)
|
|
assert r.stats[0]['num_retries'] == 2, f'{r}'
|
|
|
|
# Get, retry on 401, not happening
|
|
def test_05_08_retry_401(self, env: Env, httpd, nghttpx):
|
|
proto = 'http/1.1'
|
|
curl = CurlClient(env=env)
|
|
url = f'https://{env.authority_for(env.domain1, proto)}/curltest/tweak?status=401'
|
|
r = curl.http_download(urls=[url], alpn_proto=proto, extra_args=[
|
|
'--retry', '2', '--retry-all-errors', '--retry-delay', '1'
|
|
])
|
|
r.check_response(http_status=401)
|
|
# No retries on a 401
|
|
assert r.stats[0]['num_retries'] == 0, f'{r}'
|
|
|
|
# Server closes the connection immediately after accept,
|
|
def test_05_09_handshake_eof(self, env: Env, httpd, nghttpx):
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server:
|
|
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|
server.bind(('127.0.0.1', 0))
|
|
server.listen(1)
|
|
port = server.getsockname()[1]
|
|
|
|
# accept one connection and immediately close it
|
|
def accept_and_close():
|
|
try:
|
|
conn, _ = server.accept()
|
|
conn.close()
|
|
except Exception:
|
|
pass
|
|
|
|
t = threading.Thread(target=accept_and_close)
|
|
t.start()
|
|
|
|
curl = CurlClient(env=env, timeout=5)
|
|
url = f'https://127.0.0.1:{port}/'
|
|
r = curl.run_direct(args=[url, '--insecure'])
|
|
|
|
t.join(timeout=2)
|
|
|
|
# We expect an error code, not success (0) and not timeout (-1)
|
|
# Expected error code is:
|
|
# - CURLE_SSL_CONNECT_ERROR (35) - common for handshake failures
|
|
assert r.exit_code == 35, \
|
|
f'Expected error 35, got {r.exit_code}\n{r.dump_logs()}'
|