o d/@sddlZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZddlmZmZmZmZddlmZmZddlmZmZmZmZed Zeeeefd fZ ee d fZ!Gd d d eZ"ed dGdddeZ#ed dGdddee#Z$dS)N) dataclass)wraps) AnyCallableDictListMappingOptionalTupleTypeVarUnion)BrokenResourceError EndOfStreamaclose_forcefullyget_cancelled_exc_class)TypedAttributeSettyped_attribute) AnyByteStream ByteStreamListener TaskGroupT_Retval.c@seZdZUdZeZeeed<eZ e ed<eZ e eee fed<eZeeeeeeeffed<eZee ed<eZeed<eZee eee fed<eZejed <eZeed <eZeed <d S) TLSAttributez5Contains Transport Layer Security related attributes. alpn_protocolchannel_binding_tls_uniquecipherpeer_certificatepeer_certificate_binary server_sideshared_ciphers ssl_objectstandard_compatible tls_versionN)__name__ __module__ __qualname____doc__rrr str__annotations__rbytesrr intrrr _PCTRTTT_PCTRTTrrboolr rr!ssl SSLObjectr"r#r1r1s z,TLSStream.extra_attributes..cru)NTrvr1rdr1r2rxscsjjSre)r6rr1rdr1r2rxscs jSre)r6r r1rdr1r2rxs cjSrer"r1rdr1r2rxcryre)r6r1rdr1r2rxr{)r5extra_attributesrrr6selected_alpn_protocolrget_channel_bindingrrrrr r"r!r#versionrdr1rdr2r|s    zTLSStream.extra_attributesr;N)rh)r$r%r&r'rr)r.r/r0rD classmethodr r( SSLContextrMrrobjectrFr r*rcrgr+rUrSrtpropertyrrr|r1r1r1r2r43sN      7  /   $r4c@seZdZUdZeeed<ejed<dZ e ed<dZ e ed<e ded ed d fd d Z ddeegefdeed d fddZdddZed eeegefffddZd S) TLSListenera A convenience listener that wraps another listener and auto-negotiates a TLS session on every accepted connection. If the TLS handshake times out or raises an exception, :meth:`handle_handshake_error` is called to do whatever post-mortem processing is deemed necessary. Supports only the :attr:`~TLSAttribute.standard_compatible` extra attribute. :param Listener listener: the listener to wrap :param ssl_context: the SSL context object :param standard_compatible: a flag passed through to :meth:`TLSStream.wrap` :param handshake_timeout: time limit for the TLS handshake (passed to :func:`~anyio.fail_after`) listenerr:Tr"handshake_timeoutrbstreamr;NcsTdtdt|IdHt|tsttdt|tr't|tr(dS)Nz Handle an exception raised during the TLS handshake. This method does 3 things: #. Forcefully closes the original stream #. Logs the exception (unless it was a cancellation exception) using the ``z`` logger #. Reraises the exception if it was a base exception or a cancellation exception :param exc: the exception :param stream: the original stream zError during TLS handshake)r$rr\rlogging getLogger exception Exception)rbrr1r1r2handle_handshake_errors z"TLSListener.handle_handshake_errorhandler task_groupcs:tdtddffdd }j||IdHdS)Nrr;c sddlm}z$|jtj|jjdIdH}Wdn1s%wYWntyG}z||IdHWYd}~dSd}~ww|IdHdS)Nr ) fail_after)r:r") rrr4rMr:r"rfr)rrwrapped_streamrbrr_r1r2handler_wrapper$s    z*TLSListener.serve..handler_wrapper)rrrserve)r_rrrr1rr2rszTLSListener.servecs|jIdHdSre)rrgrdr1r1r2rg6szTLSListener.aclosecstjfddiS)Ncryrerzr1rdr1r2rx<r{z.TLSListener.extra_attributes..)rr"rdr1rdr2r|9szTLSListener.extra_attributesrer)r$r%r&r'rrr)r/rr"r.rfloat staticmethodrfrrrr4r rrrgrrr|r1r1r1r2rs&       $r)%rrnr/ dataclassesr functoolsrtypingrrrrrr r r r rrrrr_core._typedattrrrabcrrrrrr(r-r,rr4rr1r1r1r2s$  , ;