a ^g1@sLddlmZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZmZddlmZmZmZmZdd lmZmZdd lmZmZmZmZejd krdd l mZmZndd l mZmZed Z!edZ"e#e#e$e$fdfZ%e#e%dfZ&GdddeZ'e ddGdddeZ(e ddGdddee(Z)dS)) annotationsN)CallableMapping) dataclass)wraps)AnyTypeVar)BrokenResourceError EndOfStreamaclose_forcefullyget_cancelled_exc_class)TypedAttributeSettyped_attribute) AnyByteStream ByteStreamListener TaskGroup) ) TypeVarTupleUnpackT_RetvalPosArgsT.c@seZdZUdZeZded<eZded<eZded<eZ ded <eZ d ed <eZ d ed <eZ ded<eZ ded<eZd ed<eZded<dS) TLSAttributez5Contains Transport Layer Security related attributes. str | None alpn_protocolbyteschannel_binding_tls_uniqueztuple[str, str, int]cipherz*None | dict[str, str | _PCTRTTT | _PCTRTT]peer_certificatez bytes | Nonepeer_certificate_binarybool server_sidez!list[tuple[str, str, int]] | Noneshared_ciphers ssl.SSLObject ssl_objectstandard_compatiblestr tls_versionN)__name__ __module__ __qualname____doc__rr__annotations__rrr r!r#r$r&r'r)r/r/_/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/anyio/streams/tls.pyr s rF)eqc @seZdZUdZded<ded<ded<ded <ded <ed d d d d dddddddddZddddddZddddZdddd Z d/d"d#d$d%d&Z d#dd'd(d)Z ddd*d+Z e d,dd-d.Zd S)0 TLSStreama A stream wrapper that encrypts all sent data and decrypts received data. This class has no public initializer; use :meth:`wrap` instead. All extra attributes from :class:`~TLSAttribute` are supported. :var AnyByteStream transport_stream: the wrapped stream rtransport_streamr"r'r% _ssl_objectz ssl.MemoryBIO _read_bio _write_bioNT)r#hostname ssl_contextr'z bool | Nonerzssl.SSLContext | None)r3r#r7r8r'returnc s|dur| }|sL|rtjjntjj}t|}ttdrL|jtjM_t}t}|j ||||d} |||| ||d} | | j IdH| S)a Wrap an existing stream with Transport Layer Security. This performs a TLS handshake with the peer. :param transport_stream: a bytes-transporting stream to wrap :param server_side: ``True`` if this is the server side of the connection, ``False`` if this is the client side (if omitted, will be set to ``False`` if ``hostname`` has been provided, ``False`` otherwise). Used only to create a default context when an explicit context has not been provided. :param hostname: host name of the peer (if host name checking is desired) :param ssl_context: the SSLContext object to use (if not provided, a secure default will be created) :param standard_compatible: if ``False``, skip the closing handshake when closing the connection, and don't raise an exception if the peer does the same :raises ~ssl.SSLError: if the TLS handshake fails NOP_IGNORE_UNEXPECTED_EOF)r#server_hostname)r3r'r4r5r6) sslPurpose CLIENT_AUTH SERVER_AUTHcreate_default_contexthasattroptionsr: MemoryBIOwrap_bio_call_sslobject_method do_handshake) clsr3r#r7r8r'purposeZbio_inZbio_outr&wrapperr/r/r0wrapNs,  zTLSStream.wrapz&Callable[[Unpack[PosArgsT]], T_Retval]zUnpack[PosArgsT]r)funcargsr9c sz||}Wntjyz4|jjr@|j|jIdH|jIdH}WnXtyn|j YnJt y}z&|j |j t |WYd}~nd}~00|j |Yqtjy|j|jIdHYqtjy&}z&|j |j t |WYd}~qd}~0tjy}zZ|j |j t|tjsn|jrd|jvr|jr~t |ntdWYd}~qd}~00|jjr|j|jIdH|SqdS)NZUNEXPECTED_EOF_WHILE_READING)r<SSLWantReadErrorr6pendingr3sendreadreceiver r5 write_eofOSErrorr writeSSLWantWriteErrorSSLSyscallErrorSSLError isinstance SSLEOFErrorstrerrorr')selfrKrLresultdataexcr/r/r0rEsF        z TLSStream._call_sslobject_methodztuple[AnyByteStream, bytes]r9cs8||jjIdH|j|j|j|jfS)z Does the TLS closing handshake. :return: a tuple of (wrapped byte stream, bytes left in the read buffer) N)rEr4unwrapr5rRr6r3rPr[r/r/r0r`s  zTLSStream.unwrapNonecsR|jr>z|IdHWn$ty<t|jIdHYn0|jIdHdSN)r'r` BaseExceptionr r3acloserar/r/r0res zTLSStream.acloseintr) max_bytesr9cs"||jj|IdH}|st|Src)rEr4rPr )r[rhr]r/r/r0rQszTLSStream.receive)itemr9cs||jj|IdHdSrc)rEr4rT)r[rir/r/r0rOszTLSStream.sendcsd|tj}td|}|rXt|dt|dp6d}}||fdkrXtd|tddS)NzTLSv(\d+)(?:\.(\d+))?r r)rjrz;send_eof() requires at least TLSv1.3; current session uses z7send_eof() has not yet been implemented for TLS streams)extrarr)rematchrggroupNotImplementedError)r[r)rmmajorminorr/r/r0send_eofs  " zTLSStream.send_eofMapping[Any, Callable[[], Any]]csijjtjjjtjjjtjjjtj fddtj fddtj fddtj fddtj fddtjfddtjjji S)Ncs jdS)NFr4 getpeercertr/rar/r0z,TLSStream.extra_attributes..cs jdS)NTrtr/rar/r0rvscsjjSrc)r4r#r/rar/r0rvrwcsjjrjSdSrc)r4r#r$r/rar/r0rvs csjSrcr'r/rar/r0rvrwcsjSrc)r4r/rar/r0rvrw)r3extra_attributesrrr4selected_alpn_protocolrget_channel_bindingrr r!r#r$r'r&r)versionrar/rar0rys   zTLSStream.extra_attributes)rf)r*r+r,r-r. classmethodrJrEr`rerQrOrrpropertyryr/r/r/r0r2<s(  8.  r2c@seZdZUdZded<ded<dZded<d Zd ed <ed d ddddZdddddddZ ddddZ e ddddZ dS) 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`) z Listener[Any]listenerzssl.SSLContextr8Tr"r'floathandshake_timeoutrdrrb)r^streamr9csJt|IdHt|ts.ttjd|dt|trDt|trFdS)a 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 ``anyio.streams.tls`` logger #. Reraises the exception if it was a base exception or a cancellation exception :param exc: the exception :param stream: the original stream NzError during TLS handshake)exc_info)r rXr logging getLoggerr* exception Exception)r^rr/r/r0handle_handshake_errors  z"TLSListener.handle_handshake_errorNzCallable[[TLSStream], Any]zTaskGroup | None)handler task_groupr9cs6tdddfdd }j||IdHdS)Nrrb)rr9c sddlm}zH|j*tj|jjdIdH}Wdn1sH0YWn6ty}z||IdHWYd}~nd}~00|IdHdS)Nr ) fail_after)r8r') rrr2rJr8r'rdr)rrZwrapped_streamr^rr[r/r0handler_wrapper8s  .(z*TLSListener.serve..handler_wrapper)rrserve)r[rrrr/rr0r3szTLSListener.server_cs|jIdHdSrc)rrerar/r/r0reJszTLSListener.acloserscstjfddiS)NcsjSrcrxr/rar/r0rvPrwz.TLSListener.extra_attributes..)rr'rar/rar0ryMszTLSListener.extra_attributes)N) r*r+r,r-r.r'r staticmethodrrrer~ryr/r/r/r0rs   "r)* __future__rrrlr<syscollections.abcrr dataclassesr functoolsrtypingrrrr r r r Z_core._typedattrrrabcrrrr version_inforrZtyping_extensionsrrtupler(Z_PCTRTTZ_PCTRTTTrr2rr/r/r/r0s0     ?