@@ -2397,7 +2397,10 @@ def wrap_conn(self):
2397
2397
sys .stdout .write (" client cert is " + pprint .pformat (cert ) + "\n " )
2398
2398
cert_binary = self .sslconn .getpeercert (True )
2399
2399
if support .verbose and self .server .chatty :
2400
- sys .stdout .write (" cert binary is " + str (len (cert_binary )) + " bytes\n " )
2400
+ if cert_binary is None :
2401
+ sys .stdout .write (" client did not provide a cert\n " )
2402
+ else :
2403
+ sys .stdout .write (f" cert binary is { len (cert_binary )} b\n " )
2401
2404
cipher = self .sslconn .cipher ()
2402
2405
if support .verbose and self .server .chatty :
2403
2406
sys .stdout .write (" server: connection cipher is now " + str (cipher ) + "\n " )
@@ -2495,31 +2498,22 @@ def run(self):
2495
2498
sys .stdout .write (" server: read %r (%s), sending back %r (%s)...\n "
2496
2499
% (msg , ctype , msg .lower (), ctype ))
2497
2500
self .write (msg .lower ())
2498
- except (ConnectionResetError , ConnectionAbortedError ):
2499
- # XXX: OpenSSL 1.1.1 sometimes raises ConnectionResetError
2500
- # when connection is not shut down gracefully.
2501
+ except OSError as e :
2502
+ # handles SSLError and socket errors
2501
2503
if self .server .chatty and support .verbose :
2502
- sys .stdout .write (
2503
- " Connection reset by peer: {}\n " .format (
2504
- self .addr )
2505
- )
2506
- self .close ()
2507
- self .running = False
2508
- except ssl .SSLError as err :
2509
- # On Windows sometimes test_pha_required_nocert receives the
2510
- # PEER_DID_NOT_RETURN_A_CERTIFICATE exception
2511
- # before the 'tlsv13 alert certificate required' exception.
2512
- # If the server is stopped when PEER_DID_NOT_RETURN_A_CERTIFICATE
2513
- # is received test_pha_required_nocert fails with ConnectionResetError
2514
- # because the underlying socket is closed
2515
- if 'PEER_DID_NOT_RETURN_A_CERTIFICATE' == err .reason :
2516
- if self .server .chatty and support .verbose :
2517
- sys .stdout .write (err .args [1 ])
2518
- # test_pha_required_nocert is expecting this exception
2519
- raise ssl .SSLError ('tlsv13 alert certificate required' )
2520
- except OSError :
2521
- if self .server .chatty :
2522
- handle_error ("Test server failure:\n " )
2504
+ if isinstance (e , ConnectionError ):
2505
+ # OpenSSL 1.1.1 sometimes raises
2506
+ # ConnectionResetError when connection is not
2507
+ # shut down gracefully.
2508
+ print (
2509
+ f" Connection reset by peer: { self .addr } "
2510
+ )
2511
+ else :
2512
+ handle_error ("Test server failure:\n " )
2513
+ try :
2514
+ self .write (b"ERROR\n " )
2515
+ except OSError :
2516
+ pass
2523
2517
self .close ()
2524
2518
self .running = False
2525
2519
@@ -4502,24 +4496,30 @@ def test_pha_required_nocert(self):
4502
4496
server_context .verify_mode = ssl .CERT_REQUIRED
4503
4497
client_context .post_handshake_auth = True
4504
4498
4505
- # Ignore expected SSLError in ConnectionHandler of ThreadedEchoServer
4506
- # (it is only raised sometimes on Windows)
4507
- with support .catch_threading_exception () as cm :
4508
- server = ThreadedEchoServer (context = server_context , chatty = False )
4509
- with server :
4510
- with client_context .wrap_socket (socket .socket (),
4511
- server_hostname = hostname ) as s :
4512
- s .connect ((HOST , server .port ))
4513
- s .write (b'PHA' )
4499
+ def msg_cb (conn , direction , version , content_type , msg_type , data ):
4500
+ if support .verbose and content_type == _TLSContentType .ALERT :
4501
+ info = (conn , direction , version , content_type , msg_type , data )
4502
+ sys .stdout .write (f"TLS: { info !r} \n " )
4503
+
4504
+ server_context ._msg_callback = msg_cb
4505
+ client_context ._msg_callback = msg_cb
4506
+
4507
+ server = ThreadedEchoServer (context = server_context , chatty = True )
4508
+ with server :
4509
+ with client_context .wrap_socket (socket .socket (),
4510
+ server_hostname = hostname ) as s :
4511
+ s .connect ((HOST , server .port ))
4512
+ s .write (b'PHA' )
4513
+ with self .assertRaisesRegex (
4514
+ ssl .SSLError ,
4515
+ 'tlsv13 alert certificate required'
4516
+ ):
4514
4517
# receive CertificateRequest
4515
4518
self .assertEqual (s .recv (1024 ), b'OK\n ' )
4516
4519
# send empty Certificate + Finish
4517
4520
s .write (b'HASCERT' )
4518
4521
# receive alert
4519
- with self .assertRaisesRegex (
4520
- ssl .SSLError ,
4521
- 'tlsv13 alert certificate required' ):
4522
- s .recv (1024 )
4522
+ s .recv (1024 )
4523
4523
4524
4524
def test_pha_optional (self ):
4525
4525
if support .verbose :
0 commit comments