U dRf'3@sddlmZmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z mZmZe jdkp|e jdZGdddeZGd d d eZGd d d eZdS) )absolute_importdivisionN)TTransportException)create_thriftpy_contextRESTRICTED_SERVER_CIPHERSDEFAULT_CIPHERSdarwinZfreebsdc@seZdZdZddddejddddddddefddZddZd d Z d d Z d dZ ddZ ddZ ddZddZddZdS) TAsyncSocketz&Socket implementation for client side.N TcCs|r ||_nF|r2||_d|_d|_d|_tj|_n d|_||_||_d|_tj|_||_|rd|dnd|_ |rv|dn|j |_ |r||_ ||_ nl| s| r||_ t d|d|_ | s| r|j j| | d| r|j j| | d| sd|j _tj|j _n d|_ d|_ dS)uInitialize a TSocket TSocket can be initialized in 3 ways: * host + port. can configure to use AF_INET/AF_INET6 * unix_socket * socket. should pass already opened socket here. @param host(str) The host to connect to. @param port(int) The (TCP) port to connect to. @param unix_socket(str) The filename of a unix socket to connect to. @param sock(socket) Initialize with opened socket directly. If this param used, the host, port and unix_socket params will be ignored. @param socket_family(str) socket.AF_INET or socket.AF_INET6. only take effect when using host/port @param socket_timeout socket timeout in ms @param connect_timeout connect timeout in ms, only used in connection, will be set to socket_timeout if not set. @param validate(bool) Set to False to disable SSL certificate validation and hostname validation. Default enabled. @param cafile(str) Path to a file of concatenated CA certificates in PEM format. @param capath(str) path to a directory containing several CA certificates in PEM format, following an OpenSSL specific layout. @param certfile(str) The certfile string must be the path to a single file in PEM format containing the certificate as well as any number of CA certificates needed to establish the certificate’s authenticity. @param keyfile(str) The keyfile string, if not present, the private key will be taken from certfile as well. @param ciphers(list) The cipher suites to allow @param ssl_context(SSLContext) Customize the SSLContext, can be used to persist SSLContext object. Caution it's easy to get wrong, only use if you know what you're doing. NF server_sideciphers)cafilecapathkeyfile)raw_sock unix_sockethostportasyncioZopen_unix_connection sock_factoryZopen_connection socket_familysocket_timeoutconnect_timeout ssl_contextserver_hostnamerload_verify_locationsload_cert_chaincheck_hostnamessl CERT_NONE verify_mode)selfrrrsockrrrrvalidaterrcertfilerrr(@/tmp/pip-unpacked-wheel-jqs7l_7o/thriftpy2/contrib/aio/socket.py__init__sJ)   zTAsyncSocket.__init__cCsv|jrttjtj}n"t|jtj}|tjtjdt ddd}|tj tj ||tj tj d||_ dS)Niir)rsocketAF_UNIX SOCK_STREAMr setsockopt IPPROTO_TCP TCP_NODELAYstructpack SOL_SOCKET SO_LINGER SO_KEEPALIVEr)r$_sockZlingerr(r(r) _init_sockmszTAsyncSocket._init_sockcCs ||_dSN)r)r$r%r(r(r) set_handle{szTAsyncSocket.set_handlecCs>|r|dkr|dnd|_|j|_|jdk r:|j|jdS)zgBackward compat api, will bind the timeout to both connect_timeout and socket_timeout. rr N)rrr settimeout)r$msr(r(r) set_timeout~s zTAsyncSocket.set_timeoutcCs t|jSr:)boolrr$r(r(r)is_openszTAsyncSocket.is_openc s||jp|j|jf}zz|jr0|j|j|j||jrP|j|j|j|j d}|j rn|j |d<t |j f||jIdH\|_|_Wn0tjtfk rttjdt|dYnXdS)Nr%r!rzCould not connect to %stypemessage)r9rrrrrr<connectrrrrwait_forrreaderwriterr-errorOSErrorrZNOT_OPENstr)r$addrkwargsr(r(r)opens(    zTAsyncSocket.openc sz t|j||jIdH}WnDtjk rd}z$|jtjkrRt rR| d}nW5d}~XYnXt |dkrt t j dd|SNrzTSocket read 0 bytesrC)rrGrHreadrr-rJerrno ECONNRESET MAC_OR_BSDcloselenr END_OF_FILEr$szbuffer(r(r)rRs  zTAsyncSocket.readcCs|j|dSr:rIwriter$r[r(r(r)r^szTAsyncSocket.writecst|j|jIdHdSr:)rrGrIdrainrr@r(r(r)flushszTAsyncSocket.flushc CsH|js dSz|j|jd|_Wntjtfk rBYnXdSr:)rrIrVr-rJrKr@r(r(r)rVs   zTAsyncSocket.close)__name__ __module__ __qualname____doc__r-AF_INETrr*r9r;r>rArOrRr^rarVr(r(r(r)r s. S r c @sXeZdZdZdddejdddddef ddZddZd d Z d d Z d dZ ddZ dS)TAsyncServerSocketz&Socket implementation for server side.Nr c Cs|r ||_d|_d|_tj|_nd|_||_||_tj|_||_|rL|dnd|_||_ |rd||_ nD|rt |t j std|td| d|_ |j j|| dnd|_ dS)aInitialize a TServerSocket TSocket can be initialized in 2 ways: * host + port. can configure to use AF_INET/AF_INET6 * unix_socket @param host(str) The host to connect to @param port(int) The (TCP) port to connect to @param unix_socket(str) The filename of a unix socket to connect to @param socket_family(str) socket.AF_INET or socket.AF_INET6. only take effect when using host/port @param client_timeout client socket timeout @param backlog backlog for server socket @param certfile(str) The server cert pem filename @param keyfile(str) The server cert key filename @param ciphers(list) The cipher suites to allow @param ssl_context(SSLContext) Customize the SSLContext, can be used to persist SSLContext object. Caution it's easy to get wrong, only use if you know what you're doing. Nr zNo such certfile found: %sTr r)rrrrZstart_unix_serverrZ start_serverrclient_timeoutbacklogrosaccessR_OKIOErrorrr) r$rrrrrirjrr'rrr(r(r)r*s,  zTAsyncServerSocket.__init__c Cs|jrjttjtj}z||jWqztjtfk rf}z|jtjkrVt |jW5d}~XYqzXnt|j tj}| tj tjdttdrz| tj tjdWn<tjk r}z|dtjtjfkrnW5d}~XYnX|d||_dS)Nr+ SO_REUSEPORTr)rr-r.r/rFrJrKrSZ ECONNREFUSEDrkunlinkrr0r5 SO_REUSEADDRhasattrroZ ENOPROTOOPTEINVALr<r)r$r8errr(r(r)r9s$    zTAsyncServerSocket._init_sockcCs8||jp|j|jf}|j||j|jdSr:)r9rrrrbindlistenrj)r$rMr(r(r)rvs zTAsyncServerSocket.listencs$|j|||j|jdIdH}|S)NrB)r_create_client_connected_cbrr)r$callbackserverr(r(r)accept%s  zTAsyncServerSocket.acceptcsfdd}|S)NcsHz"tt||jIdHWn tjjk rB|YnXdSr:)rrG StreamHandlerri exceptions TimeoutErrorrVrHrIrxr$r(r)client_connected_cb/s zKTAsyncServerSocket._create_client_connected_cb..client_connected_cbr()r$rxrr(rr)rw-s z.TAsyncServerSocket._create_client_connected_cbc CsF|js dSz|jtj|jWntjtfk r@YnXdSr:)rshutdownr- SHUT_RDWRrVrJrKr@r(r(r)rV:szTAsyncServerSocket.close) rbrcrdrer-rfrr*r9rvrzrwrVr(r(r(r)rgs 3 rgc@s<eZdZddZddZddZddZd d Zd d Zd S)r{cCs|||_|_dSr:r~)r$rHrIr(r(r)r*FszStreamHandler.__init__c szz|j|IdH}WnDtjk rZ}z$|jtjkrHtrH|d}nW5d}~XYnXt|dkrvt t j dd|SrP) rHrRr-rJrSrTrUrVrWrrXrYr(r(r)rRIs zStreamHandler.readcCs|j|dSr:r]r_r(r(r)r^]szStreamHandler.writecs|jIdHdSr:)rIr`r@r(r(r)ra`szStreamHandler.flushc Cs.z|jWntjtfk r(YnXdSr:)rIrVr-rJrKr@r(r(r)rVcszStreamHandler.closecsdSr:r(r@r(r(r)rOiszStreamHandler.openN) rbrcrdr*rRr^rarVrOr(r(r(r)r{Es r{) __future__rrr!rrSrkr-r3sysZthriftpy2.transportrZthriftpy2.transport._sslrrrplatform startswithrUobjectr rgr{r(r(r(r)s :u