3 \�� @s8dgZddlZddlZddlZddlZddlZddlZyddlZWnek rZdZYnXddl m Z ddl mZddl mZddl m Z ddl mZdd l mZdd l mZddl mZddlmZdd lmZdd�ZGdd�de j�ZGdd�dejej�ZGdd�de�ZGdd�de�ZGdd�de�ZdS)�BaseSelectorEventLoop�N�)�base_events)�compat)� constants)�events)�futures)� selectors)� transports)�sslproto)� coroutine)�loggercCs6y|j|�}Wntk r"dSXt|j|@�SdS)NF)�get_key�KeyError�boolr)�selector�fdZevent�key�r�//usr/lib64/python3.6/asyncio/selector_events.py�_test_selector_event s rcsneZdZdN�fdd� ZdOddd�dd�ZdPddddd�d d �Zddddd�dd�ZdQd d�Z�fdd�Zdd�Z dd�Z dd�Zdd�Zdd�Z dd�ZdRdd�ZdSd d!�ZedTd"d#��Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd0d1�Zd2d3�Zd4d5�Zd6d7�Zd8d9�Zd:d;�Zdd?��Z d@dA�Z!dBdC�Z"dDdE�Z#dFdG�Z$dHdI�Z%dJdK�Z&dLdM�Z'�Z(S)UrNcsFt�j�|dkrtj�}tjd|jj�||_|j �t j�|_dS)NzUsing selector: %s) �super�__init__r ZDefaultSelectorr �debug� __class__�__name__� _selector�_make_self_pipe�weakref�WeakValueDictionary�_transports)�selfr)rrrr1s zBaseSelectorEventLoop.__init__)�extra�servercCst||||||�S)N)�_SelectorSocketTransport)r!�sock�protocol�waiterr"r#rrr�_make_socket_transport;s z,BaseSelectorEventLoop._make_socket_transportF)�server_side�server_hostnamer"r#c CsNtj�s"|j||||||||d�Stj||||||�} t||| ||d�| jS)N)r)r*r"r#)r"r#)rZ_is_sslproto_available�_make_legacy_ssl_transportZSSLProtocolr$Z_app_transport) r!�rawsockr&� sslcontextr'r)r*r"r#Zssl_protocolrrr�_make_ssl_transport@s z)BaseSelectorEventLoop._make_ssl_transportc Cst|||||||||� S)N)�_SelectorSslTransport) r!r,r&r-r'r)r*r"r#rrrr+Os z0BaseSelectorEventLoop._make_legacy_ssl_transportcCst||||||�S)N)�_SelectorDatagramTransport)r!r%r&�addressr'r"rrr�_make_datagram_transportYsz.BaseSelectorEventLoop._make_datagram_transportcsL|j�rtd��|j�rdS|j�t�j�|jdk rH|jj�d|_dS)Nz!Cannot close a running event loop)Z is_running�RuntimeError� is_closed�_close_self_piper�closer)r!)rrrr6^s zBaseSelectorEventLoop.closecCst�dS)N)�NotImplementedError)r!rrr�_socketpairisz!BaseSelectorEventLoop._socketpaircCsB|j|jj��|jj�d|_|jj�d|_|jd8_dS)Nr)�_remove_reader�_ssock�filenor6�_csock� _internal_fds)r!rrrr5ls z&BaseSelectorEventLoop._close_self_pipecCsN|j�\|_|_|jjd�|jjd�|jd7_|j|jj�|j�dS)NFr)r8r:r<�setblockingr=�_add_readerr;�_read_from_self)r!rrrrts z%BaseSelectorEventLoop._make_self_pipecCsdS)Nr)r!�datarrr�_process_self_data|sz(BaseSelectorEventLoop._process_self_datacCsVxPy |jjd�}|sP|j|�Wqtk r8wYqtk rLPYqXqWdS)Ni)r:�recvrB�InterruptedError�BlockingIOError)r!rArrrr@sz%BaseSelectorEventLoop._read_from_selfcCsJ|j}|dk rFy|jd�Wn(tk rD|jr@tjddd�YnXdS)N�z3Fail to write a null byte into the self-pipe socketT)�exc_info)r<�send�OSError�_debugr r)r!Zcsockrrr�_write_to_self�sz$BaseSelectorEventLoop._write_to_self�dcCs |j|j�|j|||||�dS)N)r?r;�_accept_connection)r!�protocol_factoryr%r-r#�backlogrrr�_start_serving�sz$BaseSelectorEventLoop._start_servingcCs�x�t|�D]�}y0|j�\}}|jr2tjd|||�|jd�Wn�tttfk rXdSt k r�} z^| j t jt jt j t jfkr�|jd| |d��|j|j��|jtj|j|||||�n�WYdd} ~ Xq Xd|i} |j||| ||�}|j|�q WdS)Nz#%r got a new connection from %r: %rFz&socket.accept() out of system resource)�message� exception�socket�peername)�range�acceptrJr rr>rErD�ConnectionAbortedErrorrI�errnoZEMFILEZENFILEZENOBUFSZENOMEM�call_exception_handlerr9r;Z call_laterrZACCEPT_RETRY_DELAYrP�_accept_connection2Zcreate_task)r!rNr%r-r#rO�_�conn�addr�excr"rVrrrrM�s4 z(BaseSelectorEventLoop._accept_connectionccs�d}d}yj|�}|j�}|r6|j||||d||d�}n|j|||||d�}y|EdHWn|j��YnXWn\tk r�} z@|jr�d| d�} |dk r�|| d<|dk r�|| d<|j| �WYdd} ~ XnXdS)NT)r'r)r"r#)r'r"r#z3Error on transport creation for incoming connection)rQrRr&� transport)� create_futurer.r(r6� ExceptionrJrY)r!rNr\r"r-r#r&r_r'r^�contextrrrrZ�s4z)BaseSelectorEventLoop._accept_connection2cCs@y|j|}Wntk r"YnX|j�sX|j|j }\}}|jj ||tjB||f�|dk r�|j�dS)N)� _check_closedr�Handlerrr�registerr � EVENT_READrA�modify�cancel) r!r�callback�args�handler�mask�reader�writerrrrr?�s z!BaseSelectorEventLoop._add_readerc Cs�|j�rdSy|jj|�}Wntk r0dSX|j|j}\}}|tjM}|sb|jj|�n|jj ||d|f�|dk r�|j �dSdSdS)NFT)r4rrrrrAr ri� unregisterrjrk)r!rrrorprqrrrr9sz$BaseSelectorEventLoop._remove_readerc Gs�|j�tj|||�}y|jj|�}Wn*tk rP|jj|tjd|f�Yn>X|j|j }\}}|jj ||tjB||f�|dk r�|j�dS)N)rfrrgrrrrhr �EVENT_WRITErArjrk) r!rrlrmrnrrorprqrrr�_add_writers z!BaseSelectorEventLoop._add_writerc Cs�|j�rdSy|jj|�}Wntk r0dSX|j|j}\}}|tjM}|sb|jj|�n|jj |||df�|dk r�|j �dSdSdS)NFT)r4rrrrrAr rsrrrjrk)r!rrrorprqrrr�_remove_writer,sz$BaseSelectorEventLoop._remove_writercGs|j|�|j||f|��S)N)rer?)r!rrlrmrrr� add_readerCs z BaseSelectorEventLoop.add_readercCs|j|�|j|�S)N)rer9)r!rrrr� remove_readerHs z#BaseSelectorEventLoop.remove_readercGs|j|�|j||f|��S)N)rert)r!rrlrmrrr� add_writerMs z BaseSelectorEventLoop.add_writercCs|j|�|j|�S)N)reru)r!rrrr� remove_writerRs z#BaseSelectorEventLoop.remove_writercCs6|jr|j�dkrtd��|j�}|j|d||�|S)Nrzthe socket must be non-blocking)rJ� gettimeout� ValueErrorr`� _sock_recv)r!r%�n�futrrr� sock_recvWs zBaseSelectorEventLoop.sock_recvcCs�|dk r|j|�|j�rdSy|j|�}Wn`ttfk rb|j�}|j||j||||�Yn6tk r�}z|j |�WYdd}~XnX|j |�dS)N)rw� cancelledrCrErDr;rvr|ra� set_exception� set_result)r!r~� registered_fdr%r}rArr^rrrr|fs z BaseSelectorEventLoop._sock_recvcCsF|jr|j�dkrtd��|j�}|r8|j|d||�n |jd�|S)Nrzthe socket must be non-blocking)rJrzr{r`� _sock_sendallr�)r!r%rAr~rrr�sock_sendall{s z"BaseSelectorEventLoop.sock_sendallcCs�|dk r|j|�|j�rdSy|j|�}WnDttfk rHd}Yn*tk rp}z|j|�dSd}~XnX|t|�kr�|jd�n.|r�||d�}|j �}|j ||j||||�dS)Nr)ryr�rHrErDrar��lenr�r;rxr�)r!r~r�r%rAr}r^rrrrr��s" z#BaseSelectorEventLoop._sock_sendallccs�|jr|j�dkrtd��ttd�s2|jtjkrptj||j|j |d�}|j �sZ|EdH|j�d\}}}}}|j�}|j |||�|EdHS)Nrzthe socket must be non-blocking�AF_UNIX)�family�proto�loop)rJrzr{�hasattrrSr�r�rZ_ensure_resolvedr��done�resultr`� _sock_connect)r!r%r1Zresolvedr[r~rrr�sock_connect�s z"BaseSelectorEventLoop.sock_connectcCs�|j�}y|j|�Wnjttfk rV|jtj|j|��|j||j |||�Yn6t k r�}z|j|�WYdd}~XnX|jd�dS)N) r;ZconnectrErDZadd_done_callback� functools�partial�_sock_connect_donerx�_sock_connect_cbrar�r�)r!r~r%r1rr^rrrr��sz#BaseSelectorEventLoop._sock_connectcCs|j|�dS)N)ry)r!rr~rrrr��sz(BaseSelectorEventLoop._sock_connect_donecCs�|j�rdSy,|jtjtj�}|dkr6t|d|f��WnBttfk rPYn6tk rz}z|j |�WYdd}~XnX|j d�dS)NrzConnect call failed %s)r�Z getsockoptrSZ SOL_SOCKETZSO_ERRORrIrErDrar�r�)r!r~r%r1�errr^rrrr��sz&BaseSelectorEventLoop._sock_connect_cbcCs4|jr|j�dkrtd��|j�}|j|d|�|S)Nrzthe socket must be non-blockingF)rJrzr{r`�_sock_accept)r!r%r~rrr�sock_accept�s z!BaseSelectorEventLoop.sock_acceptcCs�|j�}|r|j|�|j�r"dSy|j�\}}|jd�WnVttfk rh|j||j|d|�Yn:t k r�}z|j |�WYdd}~XnX|j||f�dS)NFT)r;rwr�rVr>rErDrvr�rar�r�)r!r~Z registeredr%rr\r1r^rrrr��s z"BaseSelectorEventLoop._sock_acceptcCs�x~|D]v\}}|j|j}\}}|tj@rN|dk rN|jrD|j|�n |j|�|tj@r|dk r|jrr|j|�q|j|�qWdS)N) �fileobjrAr riZ _cancelledr9Z _add_callbackrsru)r!Z event_listrror�rprqrrr�_process_events�s z%BaseSelectorEventLoop._process_eventscCs|j|j��|j�dS)N)r9r;r6)r!r%rrr� _stop_servingsz#BaseSelectorEventLoop._stop_serving)N)N)N)NNN)NNrL)NNrL)NN))r� __module__�__qualname__rr(r.r+r2r6r8r5rrBr@rKrPrMrrZrer?r9rtrurvrwrxryrr|r�r�r�r�r�r�r�r�r�r�� __classcell__rr)rrr+sR (#cs�eZdZdZeZdZd �fdd� Zdd�Zdd �Z d d�Z dd �Zdd�Zdd�Z ejr`dd�Zd!dd�Zdd�Zdd�Zdd�Zdd�Z�ZS)"�_SelectorTransport�iNcs�t�j||�||jd<|j�|jd<d|jkrdy|j�|jd<Wn tjk rbd|jd<YnX||_|j�|_ ||_ d|_||_|j �|_d|_d|_|jdk r�|jj�||j|j <dS)NrSZsocknamerTTrF)rr�_extraZgetsocknameZgetpeernamerS�error�_sockr;�_sock_fd� _protocol�_protocol_connected�_server�_buffer_factory�_buffer� _conn_lost�_closingZ_attachr )r!r�r%r&r"r#)rrrrs& z_SelectorTransport.__init__cCs�|jjg}|jdkr |jd�n|jr0|jd�|jd|j�|jdk r�|jj�r�t|jj |jt j�}|rz|jd�n |jd�t|jj |jt j�}|r�d}nd}|j �}|jd||f�d d j|�S)N�closed�closingzfd=%szread=pollingz read=idle�pollingZidlezwrite=<%s, bufsize=%s>z<%s>� )rrr��appendr�r��_loopr4rrr rirs�get_write_buffer_size�join)r!�infor��state�bufsizerrr�__repr__2s* z_SelectorTransport.__repr__cCs|jd�dS)N)�_force_close)r!rrr�abortNsz_SelectorTransport.abortcCs ||_dS)N)r�)r!r&rrr�set_protocolQsz_SelectorTransport.set_protocolcCs|jS)N)r�)r!rrr�get_protocolTsz_SelectorTransport.get_protocolcCs|jS)N)r�)r!rrrrcWsz_SelectorTransport.is_closingcCsT|jr dSd|_|jj|j�|jsP|jd7_|jj|j�|jj|jd�dS)NTr) r�r�r9r�r�r�ru� call_soon�_call_connection_lost)r!rrrr6Zsz_SelectorTransport.closecCs,|jdk r(tjd|t|d�|jj�dS)Nzunclosed transport %r)�source)r��warnings�warn�ResourceWarningr6)r!rrr�__del__hs z_SelectorTransport.__del__�Fatal error on transportcCsPt|tj�r*|jj�rBtjd||dd�n|jj||||jd��|j |�dS)Nz%r: %sT)rG)rQrRr_r&) � isinstancerZ_FATAL_ERROR_IGNOREr�� get_debugr rrYr�r�)r!r^rQrrr�_fatal_errorns z_SelectorTransport._fatal_errorcCsd|jr dS|jr(|jj�|jj|j�|jsBd|_|jj|j�|jd7_|jj|j |�dS)NTr) r�r��clearr�rur�r�r9r�r�)r!r^rrrr�|s z_SelectorTransport._force_closecCsVz|jr|jj|�Wd|jj�d|_d|_d|_|j}|dk rP|j�d|_XdS)N)r�r�Zconnection_lostr�r6r�r�Z_detach)r!r^r#rrrr��s z(_SelectorTransport._call_connection_lostcCs t|j�S)N)r�r�)r!rrrr��sz(_SelectorTransport.get_write_buffer_sizecGs"|jr dS|jj||f|��dS)N)r�r�r?)r!rrlrmrrrr?�sz_SelectorTransport._add_readeri)NN)r�)rr�r��max_size� bytearrayr�r�rr�r�r�r�rcr6rZPY34r�r�r�r�r�r?r�rr)rrr�s" r�csVeZdZd�fdd� Zdd�Zdd�Zdd �Zd d�Zdd �Zdd�Z dd�Z �ZS)r$Ncsrt�j|||||�d|_d|_tj|j�|jj|j j |�|jj|j|j|j �|dk rn|jjtj|d�dS)NF)rr�_eof�_pausedrZ_set_nodelayr�r�r�r��connection_mader?r��_read_readyr�_set_result_unless_cancelled)r!r�r%r&r'r"r#)rrrr�s z!_SelectorSocketTransport.__init__cCs>|js|jrdSd|_|jj|j�|jj�r:tjd|�dS)NTz%r pauses reading)r�r�r�r9r�r�r r)r!rrr� pause_reading�s z&_SelectorSocketTransport.pause_readingcCsB|js|jrdSd|_|j|j|j�|jj�r>tjd|�dS)NFz%r resumes reading) r�r�r?r�r�r�r�r r)r!rrr�resume_reading�s z'_SelectorSocketTransport.resume_readingcCs�|jr dSy|jj|j�}WnDttfk r4Yn|tk r`}z|j|d�WYdd}~XnPX|rt|jj |�n<|j j�r�tj d|�|jj�}|r�|j j|j�n|j�dS)Nz$Fatal read error on socket transportz%r received EOF)r�r�rCr�rErDrar�r�� data_receivedr�r�r r�eof_receivedr9r�r6)r!rAr^� keep_openrrrr��s z$_SelectorSocketTransport._read_readycCs�t|tttf�s"tdt|�j��|jr0td��|s8dS|j rf|j t jkrTtj d�|j d7_ dS|js�y|jj|�}WnBttfk r�Yn@tk r�}z|j|d�dSd}~XnX||d�}|s�dS|jj|j|j�|jj|�|j�dS)Nz1data argument must be a bytes-like object, not %rz%Cannot call write() after write_eof()zsocket.send() raised exception.rz%Fatal write error on socket transport)r��bytesr�� memoryview� TypeError�typerr�r3r�r�!LOG_THRESHOLD_FOR_CONNLOST_WRITESr �warningr�r�rHrErDrar�r�rtr��_write_ready�extend�_maybe_pause_protocol)r!rAr}r^rrr�write�s4 z_SelectorSocketTransport.writecCs�|jr dSy|jj|j�}Wn\ttfk r4Yn�tk rx}z*|jj|j �|jj �|j|d�WYdd}~XnTX|r�|jd|�=|j�|js�|jj|j �|j r�|jd�n|jr�|jjtj�dS)Nz%Fatal write error on socket transport)r�r�rHr�rErDrar�rur�r�r��_maybe_resume_protocolr�r�r��shutdownrS�SHUT_WR)r!r}r^rrrr�s& z%_SelectorSocketTransport._write_readycCs.|js|jrdSd|_|js*|jjtj�dS)NT)r�r�r�r�r�rSr�)r!rrr� write_eofs z"_SelectorSocketTransport.write_eofcCsdS)NTr)r!rrr� can_write_eof sz&_SelectorSocketTransport.can_write_eof)NNN)rr�r�rr�r�r�r�r�r�r�r�rr)rrr$�s#r$csdeZdZeZd�fdd� Zddd�Zdd�Zd d �Zdd�Z d d�Z dd�Zdd�Zdd�Z �ZS)r/NFc s�tdkrtd��|s tj||�}|dd�} |r<|r<|| d<|j|f| �}t�j||||| �d|_||_||_ ||_ d|_|jj |d�|jj�r�tjd|�|jj�}nd}|j|�dS)Nzstdlib ssl module not availableF)r)Zdo_handshake_on_connectr*)r-z%r starts SSL handshake)�sslr3rZ_create_transport_contextZwrap_socketrrr��_server_hostname�_waiter�_sslcontextr�r��updater�r�r r�time� _on_handshake) r!r�r,r&r-r'r)r*r"r#Zwrap_kwargsZsslsock� start_time)rrrr(s* z_SelectorSslTransport.__init__cCsD|jdkrdS|jj�s:|dk r.|jj|�n|jjd�d|_dS)N)r�r�r�r�)r!r^rrr�_wakeup_waiterLs z$_SelectorSslTransport._wakeup_waiterc"Cs$y|jj�Wn�tjk r8|jj|j|j|�dStjk r`|jj |j|j|�dSt k r�}z`|jj�r�tj d|dd�|jj|j�|jj|j�|jj�|j|�t|t�r�dS�WYdd}~XnX|jj|j�|jj|j�|jj�}t|jd��s�|j�r�|jjtjk�r�ytj||j�WnRtk �r�}z4|jj��rjtj d|dd�|jj�|j|�dSd}~XnX|jj||jj�|jj�|jd�d|_d|_ |jj|j|j!�d|_"|jj#|j$j%|�|jj#|j�|jj��r |jj&�|}tj'd||d �dS) Nz%r: SSL handshake failedT)rGZcheck_hostnamez1%r: SSL handshake failed on matching the hostname)�peercert�cipher�compressionZ ssl_objectFz%r: SSL handshake took %.1f msg@�@)(r�Zdo_handshaker��SSLWantReadErrorr�r?r�r��SSLWantWriteErrorrt� BaseExceptionr�r r�r9rur6r�r�raZgetpeercertr�r�r�Zverify_modeZ CERT_NONEZmatch_hostnamer�r�r�r��_read_wants_write�_write_wants_readr�r�r�r�r�r�r)r!r�r^r�Zdtrrrr�Vsb z#_SelectorSslTransport._on_handshakecCsJ|jrtd��|jrtd��d|_|jj|j�|jj�rFtjd|�dS)Nz#Cannot pause_reading() when closingzAlready pausedTz%r pauses reading) r�r3r�r�r9r�r�r r)r!rrrr��s z#_SelectorSslTransport.pause_readingcCsJ|jstd��d|_|jrdS|jj|j|j�|jj�rFtj d|�dS)Nz Not pausedFz%r resumes reading) r�r3r�r�r?r�r�r�r r)r!rrrr��s z$_SelectorSslTransport.resume_readingcCs"|jr dS|jr6d|_|j�|jr6|jj|j|j�y|jj|j �}Wn�t ttj fk rdYn�tjk r�d|_|jj|j�|jj|j|j�Yn�tk r�}z|j|d�WYdd}~XnTX|r�|jj|�n@z4|jj�r�tjd|�|jj�}|�rtjd�Wd|j�XdS)NFTz!Fatal read error on SSL transportz%r received EOFz?returning true from eof_received() has no effect when using ssl)r�r�r�r�r�rtr�r�rCr�rErDr�r�r�r�r9rar�r�r�r�r rr�r�r6)r!rAr^r�rrrr��s4 z!_SelectorSslTransport._read_readycCs(|jr dS|jrszC_SelectorDatagramTransport.get_write_buffer_size..)�sumr�)r!rrrr�sz0_SelectorDatagramTransport.get_write_buffer_sizecCs�|jr dSy|jj|j�\}}Wnpttfk r8Ynhtk rd}z|jj|�WYdd}~Xn<t k r�}z|j |d�WYdd}~XnX|jj||�dS)Nz&Fatal read error on datagram transport)r�r�Zrecvfromr�rErDrIr��error_receivedrar�Zdatagram_received)r!rAr]r^rrrr� sz&_SelectorDatagramTransport._read_readycCsTt|tttf�s"tdt|�j��|s*dS|jrN|d|jfkrNtd|jf��|j r�|jr�|j t jkrptj d�|j d7_ dS|j�s4y&|jr�|jj|�n|jj||�dSttfk r�|jj|j|j�YnZtk �r}z|jj|�dSd}~Xn.tk �r2}z|j|d�dSd}~XnX|jjt|�|f�|j�dS)Nz1data argument must be a bytes-like object, not %rz#Invalid address: must be None or %szsocket.send() raised exception.rz'Fatal write error on datagram transport)r�r�r�r�r�r�rr�r{r�rr�r r�r�r�rH�sendtorErDr�rtr�� _sendto_readyrIr�r�rar�r�r�)r!rAr]r^rrrr�.s< z!_SelectorDatagramTransport.sendtocCs�x�|jr�|jj�\}}y&|jr,|jj|�n|jj||�Wqttfk rf|jj||f�PYqt k r�}z|j j|�dSd}~Xqtk r�}z|j |d�dSd}~XqXqW|j�|js�|jj|j�|jr�|jd�dS)Nz'Fatal write error on datagram transport)r��popleftr�r�rHr�rErD� appendleftrIr�r�rar�r�r�rur�r�r�)r!rAr]r^rrrr�Us*z(_SelectorDatagramTransport._sendto_ready)NNN)N)rr�r��collections�dequer�rr�r�r�r�r�rr)rrr0s 'r0)�__all__r�rXr�rSr�rr��ImportError�rrrrrr r rZ coroutinesr�logr rZ BaseEventLooprZ_FlowControlMixinZ Transportr�r$r/r0rrrr�sB ii