g:dZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZ ddlZdZGd d ZGddZ y#e$r dd lmZmZddlZej,j.Zej2j4ej2j6ej2j6ge_ej2j:e_d Zn#e$rd ZYnwxYwYwxYw)z%Utilities to manage open connections.N)suppress)errors) IS_WINDOWS)MakeFilec|j}tj|tj}tj|tj|tjzy)z4Mark the given socket fd as non-inheritable (POSIX).N)filenofcntlF_GETFDF_SETFD FD_CLOEXEC)sockfd old_flagss (/opt/Tautulli/lib/cheroot/connections.pyprevent_socket_inheritancer)s@ [[]KKEMM2  B y53C3C'CD)windllWinErrorcNt|jdds ty)z6Mark the given socket fd as non-inheritable (Windows).rrN)_SetHandleInformationr rrs rrr$s!(1=j >rcy)ztStub inheritance prevention. Dummy function, since neither fcntl nor ctypes are available. Nrs rrrs rcHeZdZdZdZdZedZd dZdZ d dZ d Z y) _ThreadsafeSelectoraThread-safe wrapper around a DefaultSelector. There are 2 thread contexts in which it may be accessed: * the selector thread * one of the worker threads in workers/threadpool.py The expected read/write patterns are: * :py:func:`~iter`: selector thread * :py:meth:`register`: selector thread and threadpool, via :py:meth:`~cheroot.workers.threadpool.ThreadPool.put` * :py:meth:`unregister`: selector thread only Notably, this means :py:class:`_ThreadsafeSelector` never needs to worry that connections will be removed behind its back. The lock is held when iterating or modifying the selector but is not required when :py:meth:`select()ing ` on it. chtj|_tj|_yN) selectorsDefaultSelector _selector threadingLock_lockselfs r__init__z_ThreadsafeSelector.__init__Ds "224^^% rc|j5t|jjxsicdddS#1swYyxYwr)r$lenr!get_mapr%s r__len__z_ThreadsafeSelector.__len__Hs8 ZZ 7t~~--/526 7 7 7s '>Ac#K|j5|jjxsi}|jD]\}\}}}}||f dddy#1swYyxYww)z2Retrieve connections registered with the selector.N)r$r!r*items)r&mapping_sock_fdconns r connectionsz_ThreadsafeSelector.connectionsLsgZZ &nn,,.4"G,3MMO &((Aw4o% & & & &s A'AA A'A$ A'Nc|j5|jj|||cdddS#1swYyxYw)z'Register ``fileobj`` with the selector.N)r$r!register)r&fileobjeventsdatas rr4z_ThreadsafeSelector.registerTs8 ZZ B>>**7FDA B B Bs4=c||j5|jj|cdddS#1swYyxYw)z)Unregister ``fileobj`` from the selector.N)r$r! unregister)r&r5s rr9z_ThreadsafeSelector.unregisterYs0 ZZ 6>>,,W5 6 6 6s2;cHd|jj|DS)zReturn socket fd and data pairs from selectors.select call. Returns entries ready to read in the form: (socket_file_descriptor, connection) c3RK|]\}}|j|jf!ywr)rr7).0keyr/s r z-_ThreadsafeSelector.select..ds) QVVSXX  s%'timeout)r!select)r&r@s rrAz_ThreadsafeSelector.select^s&  ..///@  rcz|j5|jjdddy#1swYyxYw)zClose the selector.N)r$r!closer%s rrCz_ThreadsafeSelector.closeis. ZZ # NN " # # #s1:r) __name__ __module__ __qualname____doc__r'r+propertyr2r4r9rArCrrrrr0s:&&7&&B 6  #rrcfeZdZdZdZdZdZdZdZdZ dZ d Z d Z e d Ze d Zy )ConnectionManagerzClass which manages HTTPConnection objects. This is for connections which are being kept-alive for follow-up requests. cd|_d|_||_t|_|jj |j jtj|y)zInitialize ConnectionManager object. Args: server (cheroot.server.HTTPServer): web server object that uses this ConnectionManager instance. Fr7N) _serving_stop_requestedserverrr!r4socketr r EVENT_READ)r&rOs rr'zConnectionManager.__init__usV $ ,.  MM "  v  rc*tj|_|jjr|jj |y|j j|jjtj|y)zPut idle connection into the ConnectionManager to be managed. :param conn: HTTP connection to be managed :type conn: cheroot.server.HTTPConnection rLN) time last_usedrfilehas_datarO process_connr!r4rPr rrQ)r&r1s rputzConnectionManager.putsf  ::   KK $ $T * NN # # ""$i&:&: $ rc |jjDcgc]'\}}||jk7r|j|kr||f)}}}|D]0\}}|jj ||j 2ycc}}w)aIExpire least recently used connections. :param threshold: Connections that have not been used within this \ duration (in seconds), are considered expired and \ are closed and removed. :type threshold: float This should be called periodically. N)r!r2rOrTr9rC)r& thresholdr0r1timed_out_connectionss r_expirezConnectionManager._expires$(>>#=#=! $t{{"t~~ 'AdO! ! 3 MGT NN % %g . JJL  ! s,A?cpd|_|jr#tjd|jr"yy)z^Stop the selector loop in run() synchronously. May take up to half a second. Tg{Gz?N)rNrMrSsleepr%s rstopzConnectionManager.stops( $mm JJt mmrc\d|_ |j|d|_y#d|_wxYw)aRun the connections selector indefinitely. Args: expiration_interval (float): Interval, in seconds, at which connections will be checked for expiration. Connections that are ready to process are submitted via self.server.process_conn() Connections submitted for processing must be `put()` back if they should be examined again for another request. Can be shut down by calling `stop()`. TFN)rM_run)r&expiration_intervals rrunzConnectionManager.runs,  " II) *!DMEDMs" +ctj}tr t|d}n|}|js |jj |}|D]\}}||jurD|j|jj}|<|jj|X|jj||jj|tj}||z |kDr+|j||jjz |}|jsyy#t $r|jY/wxYw)aRun connection handler loop until stop was requested. :param expiration_interval: Interval, in seconds, at which \ connections will be checked for \ expiration. :type expiration_interval: float Use ``expiration_interval`` as ``select()`` timeout to assure expired connections are closed in time. On Windows cap the timeout to 0.05 seconds as ``select()`` does not return when a socket is ready. g?r?N)rZ)rSrminrNr!rAOSError_remove_invalid_socketsrO_from_server_socketrPrWr9r\r@) r&rblast_expiration_checkselect_timeout active_listr0r1new_connnows rrazConnectionManager._runs/!%  !!4d;N0N&& "nn33N3K $/ 3$4;;&#77 8J8JKH+ 00:NN--g6KK,,T2 3))+C++/BB sT[[-@-@'@ A(+%-&& ,,. sD==EEcg}|jjD]*\}}||jur tj|,|D]H\}}|jj|tt 5|jdddJy#t $r|j ||fYwxYw#1swYwxYw)aClean up the resources of any broken connections. This method attempts to detect any connections in an invalid state, unregisters them from the selector and closes the file descriptors of the corresponding network sockets where possible. N) r!r2rOosfstatrfappendr9rrC)r& invalid_connsr0r1s rrgz)ConnectionManager._remove_invalid_socketss !^^77 6MGTt{{" 6!  6+ MGT NN % %g .'"       6$$gt_5 6  sB8B6B32B36B? cN |j\}}|jjdr!|jjdxxdz cc<t|t |dr%|j |jj t}i}|jjz |jjj|\}}|jjj0}t |dr%|j |jj |jj3|j||} t5|jj6t8t:fs7|!t|j=dk(rd}nd}|d| _|d| _ || _!| S#tj$r+}|jjd|d|Yd}~yd}~wtj$r}|jjd|d|d}d |jjzd t|zd |g} ||d t j"} | j%d j'| j)dn6#t*$r*} | j,dtj.vrYd} ~ nd} ~ wwxYwYd}~yd}~wwxYw#tDj $rYyt*$r} |jjdr!|jjdxxdz cc<| j,dtjFvrYd} ~ y| j,dtjHvrYd} ~ y| j,dtj.vrYd} ~ yd} ~ wwxYw)NEnabledAcceptsr settimeoutzClient uF lost — peer dropped the TLS connection suddenly, during handshake: u attempted to speak plain HTTP into a TCP connection configured for TLS-only traffic — trying to send back a plain HTTP error response: zUThe client sent a plain HTTP request, but this server only speaks HTTPS on this port.z%s 400 Bad Request zContent-Length: %s zContent-Type: text/plain wbz ISO-8859-1r)z0.0.0.0r)z::rz Socket Errors)%acceptrOstatsrhasattrrvr@r ssl_adapterwrapr FatalSSLAlert error_log NoSSLErrorprotocolr)ioDEFAULT_BUFFER_SIZEwritejoinencoderfargssocket_errors_to_ignoremakefileConnectionClass isinstance bind_addrstrbytes getsockname remote_addr remote_portssl_envrPsocket_error_eintrsocket_errors_nonblocking) r& server_socketsaddrmfrtls_connection_drop_errorhttp_over_https_errmsgbufwfileexr1s rrhz%ConnectionManager._from_server_sockets-^ #**,GAt{{  + !!),1, &q )q,' T[[001BG{{&&2!!%!8!8!=!=a!@JAwB[[,,551l+LL!4!45;;..t{{ArBDdkk33c5\B<1==?+q0- )#'7 #'7 "DLKm++KK))!$+B48: ((KK))!$+L/24F 14;;3G3GG03s8;: Cq$(>(>?E" BGGCL$7$7 $EF""771:V-K-KK!L"1b~~   {{  + !!/2a72wwqzV666wwqzV===wwqzV;;; ' sB'K*(F4C!K4K!G-(K-KA$K(/JK K ! KKK  KKKKN$0N$8ANN:NNN$c|jjD]$\}}||jus|j&|jjy)z Close all monitored connections.N)r!r2rOrC)r&r/r1s rrCzConnectionManager.closeysF33 IQ4;;&   rc2t|jdz S)zReturn the current number of connections. Includes all connections registered with the selector, minus one for the server socket, which is always registered with the selector. r)r)r!r%s r_num_connectionsz"ConnectionManager._num_connectionss4>>"Q&&rcX|jj}|duxs|j|kS)z>Flag whether it is allowed to add a new keep-alive connection.N)rOkeep_alive_conn_limitr)r&ka_limits rcan_add_keepalive_connectionz.ConnectionManager.can_add_keepalive_connections.;;444C4#8#88#CCrN)rDrErFrGr'rXr\r_rcrargrhrCrHrrrrrrJrJosa  $ *"*3,j2_B''DDrrJ)!rGrrorPr"rSr contextlibrrxr_compatrrrr r ImportErrorctypesrrctypes.wintypeskernel32SetHandleInformationrwintypesHANDLEDWORDargtypesBOOLrestyperrJrrrrs+  E2E<#<#~_D_D}!!+ & D D OO " " OO ! ! OO ! !* & )/(<(<% !  !s6AC.BCC.C(%C.'C((C.-C.