U ¹êW[ÍUã@sdZddlmZmZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z mZddlmZmZmZeedƒszedƒ‚ddlmZmZmZmZmZmZdd lmZmZdd lmZm Z m!Z!m"Z"dd l#m$Z$dd l%m&Z&dd l'm(Z(zddlm)Z)Wnek rdZ)YnXdd„Z*eej+ƒGdd„de,ƒƒZ-Gdd„de,ƒZ.e)rPe-Z-ne.Z-Gdd„de-ej/ƒZ/dd„Z0Gdd„de,ƒZ1Gdd„de1ej2ƒZ2Gdd„de-ej3ƒZ4Gdd „d ej5ƒZ6eej7ƒGd!d"„d"e1ej2ƒƒZ8eej9feej:ƒžŽGd#d$„d$e8ƒƒZ;dS)%z” UNIX socket support for Twisted. End users shouldn't use this module directly - use the reactor APIs instead. Maintainer: Itamar Shtull-Trauring é)ÚdivisionÚabsolute_importN)ÚEINTRÚEMSGSIZEÚEAGAINÚ EWOULDBLOCKÚ ECONNREFUSEDÚENOBUFS)Ú implementerÚimplementer_onlyÚ implementedByÚAF_UNIXz+UNIX sockets not supported on this platform)ÚmainÚbaseÚtcpÚudpÚerrorÚ interfaces)ÚprotocolÚaddress)ÚlockfileÚlogÚreflectÚfailure)Ú_coerceToFilesystemEncoding)ÚuntilConcludes)Ú lazyByteSlice)ÚsendmsgcCst d|¡}tjtj|fgS)zh Pack an integer into an ancillary data structure suitable for use with L{sendmsg.sendmsg}. Úi)ÚstructZpackÚsocketÚ SOL_SOCKETrÚ SCM_RIGHTS)ÚfdZpacked©r$ú7/usr/lib/python3/dist-packages/twisted/internet/unix.pyÚ_ancillaryDescriptor)s r&c@sHeZdZdZdZdZdd„Zdd„Zdd „Zd d „Z d d „Z dd„Z dS)Ú _SendmsgMixina  Mixin for stream-oriented UNIX transports which uses sendmsg and recvmsg to offer additional functionality, such as copying file descriptors into other processes. @ivar _writeSomeDataBase: The class which provides the basic implementation of C{writeSomeData}. Ultimately this should be a subclass of L{twisted.internet.abstract.FileDescriptor}. Subclasses which mix in L{_SendmsgMixin} must define this. @ivar _sendmsgQueue: A C{list} of C{int} holding file descriptors which are currently buffered before being sent. @ivar _fileDescriptorBufferSize: An C{int} giving the maximum number of file descriptors to accept and queue for sending before pausing the registered producer, if there is one. Né@cCs g|_dS©N)Ú _sendmsgQueue©Úselfr$r$r%Ú__init__Jsz_SendmsgMixin.__init__cCst|jƒ|jkp|j |¡S)a¦ Determine whether the user-space send buffer for this transport is full or not. This extends the base determination by adding consideration of how many file descriptors need to be sent using L{sendmsg.sendmsg}. When there are more than C{self._fileDescriptorBufferSize}, the buffer is considered full. @return: C{True} if it is full, C{False} otherwise. )Úlenr*Ú_fileDescriptorBufferSizeÚ_writeSomeDataBaseÚ_isSendBufferFullr+r$r$r%r1Ns þz_SendmsgMixin._isSendBufferFullcCs |j |¡| ¡| ¡dS)zY Queue the given file descriptor to be sent and start trying to send it. N)r*ÚappendZ_maybePauseProducerZ startWriting)r,Úfilenor$r$r%ÚsendFileDescriptorfs z _SendmsgMixin.sendFileDescriptorc Cst|jƒt|ƒkrt ¡Sd}z¦|t|jƒkrÂ|j|}z&ttj|j|||d…t|ƒƒWnXtjk r¶}z8|jdt t fkr”|WY¢W¢4St j WY¢ W¢"SW5d}~XYq X|d7}q W5|jd|…=Xt ||ƒ}|j ||¡}z ||WStk r|YSXdS)zg Send as much of C{data} as possible. Also send any pending file descriptors. rNé)r.r*rZFileDescriptorOverrunrrr r&Úargsrr rÚCONNECTION_LOSTrr0Ú writeSomeDataÚ TypeError)r,ÚdataÚindexr#ÚseZ limitedDataÚresultr$r$r%r8os2  þ$  z_SendmsgMixin.writeSomeDatac CsÈzttj|j|jƒ\}}}WnJtjk rf}z*|jdtkrHWY¢dStj WY¢SW5d}~XYnX|D]P\}}}|tj kr–|tj kr–|  |¡qlt jd| ¡| ¡| |j¡||dql| |¡S)aö Calls {IProtocol.dataReceived} with all available data and L{IFileDescriptorReceiver.fileDescriptorReceived} once for each received file descriptor in ancillary data. This reads up to C{self.bufferSize} bytes of data from its socket, then dispatches the data to protocol callbacks to be handled. If the connection is not lost through an error in the underlying recvmsg(), this function will return the result of the dataReceived call. rNzˆ%(protocolName)s (on %(hostAddress)r) received unsupported ancillary data (level=%(cmsgLevel)r, type=%(cmsgType)r) from %(peerAddress)r.)ÚformatÚ hostAddressÚ peerAddressÚ protocolNameÚ cmsgLevelÚcmsgType)rrZrecvmsgr Z bufferSizerr6rrr7r!r"Ú%_ancillaryLevelSOLSOCKETTypeSCMRIGHTSrÚmsgÚgetHostÚgetPeerÚ _getLogPrefixrZ _dataReceived)r,r:Z ancillaryÚflagsr<rBrCÚcmsgDatar$r$r%ÚdoRead›s0 ÿ   ÿ  ø z_SendmsgMixin.doReadcCs|t|ƒd}t d||¡}tj |j¡rB|D]}|j |¡q.n6tj d|  ¡|  ¡|  |j¡d|D]}t  |¡qhdS)a¼ Processes ancillary data with level SOL_SOCKET and type SCM_RIGHTS, indicating that the ancillary data payload holds file descriptors. Calls L{IFileDescriptorReceiver.fileDescriptorReceived} once for each received file descriptor or logs a message if the protocol does not implement L{IFileDescriptorReceiver}. @param cmsgData: Ancillary data payload. @type cmsgData: L{bytes} érzˆ%(protocolName)s (on %(hostAddress)r) does not provide IFileDescriptorReceiver; closing file descriptor received (from %(peerAddress)r).)r>r?r@rAN)r.rZunpackrZIFileDescriptorReceiverZ providedByrZfileDescriptorReceivedrrErFrGrHÚosÚclose)r,rJZfdCountZfdsr#r$r$r%rDÂs  úz3_SendmsgMixin._ancillaryLevelSOLSOCKETTypeSCMRIGHTS) Ú__name__Ú __module__Ú __qualname__Ú__doc__r0r/r-r1r4r8rKrDr$r$r$r%r'3s ,'r'c@seZdZdZdS)Ú_UnsupportedSendmsgMixinz Behaviorless placeholder used when C{twisted.python.sendmsg} is not available, preventing L{IUNIXTransport} from being supported. N)rOrPrQrRr$r$r$r%rSásrSc@s6eZdZejZdd„Zedd„ƒZdd„Z dd„Z d S) ÚServerc Cs*t |¡tj ||||df|||¡dSr))r'r-rrT)r,ZsockrZclientZserverÚ sessionnoÚreactorr$r$r%r-ôs zServer.__init__c Csžt |tjtj¡}t | ¡¡}| |¡}|dkr>| ¡dSd}||||  ¡d||ƒ}d|j j j |j | ¡f|_d|j j j |j | ¡f|_| |¡|S)a Create a new L{Server} based on an existing connected I{SOCK_STREAM} socket. Arguments are the same as to L{Server.__init__}, except where noted. @param fileDescriptor: An integer file descriptor associated with a connected socket. The socket must be in non-blocking mode. Any additional attributes desired, such as I{FD_CLOEXEC}, must also be set already. @return: A new instance of C{cls} wrapping the socket given by C{fileDescriptor}. Nrz<%s #%s on %s>z%s,%s,%s)r Úfromfdr Ú SOCK_STREAMrÚ UNIXAddressÚ getsocknameZ buildProtocolrNZ getpeernamerÚ __class__rOrUZrepstrÚlogstrZmakeConnection) ÚclsZfileDescriptorÚfactoryrVÚsktZ protocolAddrÚprotorUr,r$r$r%Ú_fromConnectedSocketøs( ÿÿ zServer._fromConnectedSocketcCst |j ¡¡Sr)©rrYr rZr+r$r$r%rFszServer.getHostcCst |jp d¡Sr))rrYZhostnamer+r$r$r%rGszServer.getPeerN) rOrPrQrrTr0r-Ú classmethodrarFrGr$r$r$r%rTðs  !rTcCs|dd…dkS)a› Determine whether the given unix socket path is in a filesystem namespace. While most PF_UNIX sockets are entries in the filesystem, Linux 2.2 and above support PF_UNIX sockets in an "abstract namespace" that does not correspond to any path. This function returns C{True} if the given socket path is stored in the filesystem and C{False} if the path is in this abstract namespace. Nr5)óúr$)Úpathr$r$r%Ú_inFilesystemNamespace"s rgc@seZdZdd„ZdS)Ú _UNIXPortcCst |j ¡¡S)zV Returns a UNIXAddress. This indicates the server's address. rbr+r$r$r%rF0sz_UNIXPort.getHostN)rOrPrQrFr$r$r$r%rh/srhc@s^eZdZejZejZeZ dZ ddd„Z e dd„ƒZ d d „Zd d „Zd d„Zdd„Zdd„ZdS)ÚPortNé2é¶rcCs4tjj|| |¡j|||d||_||_d|_dS)N©rV)rrir-Ú _buildAddrÚnameÚmodeÚwantPIDÚ_preexistingSocket)r,ZfileNamer^ÚbacklogrorVrpr$r$r%r-Asÿz Port.__init__cCs.t ||j|j¡}|| ¡||d}||_|S)aé Create a new L{Port} based on an existing listening I{SOCK_STREAM} socket. Arguments are the same as to L{Port.__init__}, except where noted. @param fd: An integer file descriptor associated with a listening socket. The socket must be in non-blocking mode. Any additional attributes desired, such as I{FD_CLOEXEC}, must also be set already. @return: A new instance of C{cls} wrapping the socket given by C{fd}. rl)r rWÚ addressFamilyÚ socketTyperZrq)r]rVr#r^Úportr,r$r$r%Ú_fromListeningDescriptorIszPort._fromListeningDescriptorcCs:t |jj¡}t|dƒr,d|td|jƒfSd|fSdS)Nr ú <%s on %r>Úú<%s (not listening)>)rÚqualr^r[Úhasattrrru)r,Z factoryNamer$r$r%Ú__repr__\s  ÿz Port.__repr__cCs t |¡Sr)©rrY)r,rnr$r$r%rmdszPort._buildAddrc Cs\tj ¡t d| |j¡td|jƒf¡|j ršt   |jd¡|_ |j   ¡s`t d|jd¡‚n:|j jsšz$t t |j¡j¡rŠt |j¡Wn YnX|j ¡z0|jdk r¾|j}d|_n| ¡}| |j¡Wn6tjk r }zt d|j|¡‚W5d}~XYnNXt|jƒr(t |j|j¡| |j¡d|_ ||_|jj!|_!d|_"| #¡dS)zÐ Create and bind my socket, and begin listening on it. This is called on unserialization, and must be called after creating a server to begin listening on the specified port. z%s starting on %rrxó.lockNzCannot acquire lockTéd)$rZ _reservedFDZreserverrErHr^rrurprZFilesystemLockÚlockFileÚlockrÚCannotListenErrorZcleanÚstatÚS_ISSOCKrMÚst_modeÚremoveZdoStartrqÚcreateInternetSocketÚbindr rgÚchmodroZlistenrrÚ connectedr3Z numberAcceptsZ startReading©r,r_Úler$r$r%ÚstartListeninggsD   þ  ÿ  "   zPort.startListeningcCst dtd|jƒ¡dS)z0 Log message for closing socket z(UNIX Port %s Closed)rxN)rrErrur+r$r$r%Ú_logConnectionLostMsg›s ÿzPort._logConnectionLostMsgcCs<t|jƒrt |j¡|jdk r*|j ¡tj ||¡dSr)) rgrurMÚunlinkr€ZunlockrriÚconnectionLost©r,Úreasonr$r$r%r£s     zPort.connectionLost)rjrkNr)rOrPrQr r rsrXrtrTZ transportr€r-rcrvr|rmrrŽrr$r$r$r%ri:sÿ  4ric@s<eZdZdZejZejZe j Z d dd„Z dd„Z dd „ZdS) ÚClientzA client for Unix sockets.NrcCsjt |¡t |¡j}||_||_|_|rPt  |d¡sP|  ddt   |¡|¡|  |j | ¡d|¡dS)Nr~)r'r-rrYrnÚ connectorZ realAddressÚaddrrZisLockedZ _finishInitrZ BadFileErrorZ doConnectr‡)r,Úfilenamer”rVÚcheckPIDr$r$r%r-²s   ÿzClient.__init__cCs t |j¡Sr))rrYr•r+r$r$r%rG½szClient.getPeercCs t d¡Sr)r}r+r$r$r%rFÀszClient.getHost)Nr)rOrPrQrRr r rsrXrtrÚ BaseClientr0r-rGrFr$r$r$r%r“¬s r“c@s$eZdZdd„Zdd„Zdd„ZdS)Ú ConnectorcCs"tj ||||¡||_||_dSr))rÚ BaseConnectorr-rr—)r,rr^ZtimeoutrVr—r$r$r%r-ÅszConnector.__init__cCst|j||j|jƒSr))r“rrVr—r+r$r$r%Ú_makeTransportÊszConnector._makeTransportcCs t |j¡Sr)r}r+r$r$r%ÚgetDestinationÍszConnector.getDestinationN)rOrPrQr-r›rœr$r$r$r%r™Äsr™c@sJeZdZdZejZddd„Zdd„Zd d „Z d d „Z dd d„Z dd„Z dS)Ú DatagramPortz4 Datagram UNIX port, listening for packets. é rkNcCs tjj|||||d||_dS)z.Initialize with address to listen on. )Ú maxPacketSizerVN)rrir-ro)r,r•r`rŸrorVr$r$r%r-ÙszDatagramPort.__init__cCs4t |jj¡}t|dƒr&d||jfSd|fSdS)Nr rwry)rrzrr[r{ru)r,rAr$r$r%r|às zDatagramPort.__repr__c Csªt d|jjt|jƒf¡z| ¡}|jr8| |j¡Wn4tj k rn}zt   d|j|¡‚W5d}~XYnX|jrt |jƒrt   |j|j¡d|_||_|jj|_dS)Nz%s starting on %sr5)rrErr[Úreprrur‡rˆr rr‚rgrMr‰rorŠr3r‹r$r$r%Ú _bindSocketès"zDatagramPort._bindSocketc Cs€z|j ||¡WStjk rz}zJ|jd}|tkrJ| ||¡WY¢&S|tkr^t d¡‚n |tkrhn‚W5d}~XYnXdS)zWrite a datagram.rúmessage too longN) r Zsendtorr6rÚwriterÚMessageLengthErrorr)r,Zdatagramrr<Únor$r$r%r£÷s  zDatagramPort.writecCslt dt|jƒ¡tj ||¡t|dƒr6|j  ¡d|_ |j   ¡|` |` t|dƒrh|j d¡|`dS)zCleans up my socket. z(Port %s Closed)rrÚdN)rrEr rurÚBasePortrr{rZdoStoprŠr rNr3r¦Úcallbackr‘r$r$r%r s     zDatagramPort.connectionLostcCst |jj¡d|_dS)Nz (UDP))rrzrr[r\r+r$r$r%Ú setLogStrszDatagramPort.setLogStr)ržrkN)N) rOrPrQrRr r rsr-r|r¡r£rr©r$r$r$r%rÑs  rc@sBeZdZdZddd„Zdd„Zd d „Zd d „Zd d„Zdd„Z dS)ÚConnectedDatagramPortz+ A connected datagram UNIX socket. ržrkNcCs.t|tjƒst‚t ||||||¡||_dSr))Ú isinstancerZConnectedDatagramProtocolÚAssertionErrorrr-Ú remoteaddr)r,r•r`rŸroZ bindAddressrVr$r$r%r-%s ÿzConnectedDatagramPort.__init__cCsBz"| ¡|j |j¡| ¡Wn| t ¡¡YnXdSr))r¡r Zconnectr­Z_connectToProtocolÚconnectionFailedrZFailurer+r$r$r%r-s  z$ConnectedDatagramPort.startListeningcCs| ¡|j |¡|`dS)z£ Called when a connection fails. Stop listening on the socket. @type reason: L{Failure} @param reason: Why the connection failed. N)Z stopListeningrr®r‘r$r$r%r®6s z&ConnectedDatagramPort.connectionFailedc Cs²d}||jkr®z.|j |j¡\}}|t|ƒ7}|j |¡Wqtjk r˜}z<|jd}|t t t fkrrWY¢dS|t kr†|j  ¡n‚W5d}~XYqt ¡YqXqdS)z= Called when my socket is ready for reading. rN)Z maxThroughputr ZrecvfromrŸr.rZdatagramReceivedrr6rrrrÚconnectionRefusedrZdeferr)r,Úreadr:r•r<r¥r$r$r%rKBs     zConnectedDatagramPort.doReadc Csz|j |¡WStjk rŠ}z\|jd}|tkrF| |¡WY¢:S|tkrZt d¡‚n |tkrn|j   ¡n |t krxn‚W5d}~XYnXdS)z# Write a datagram. rr¢N) r Úsendrr6rr£rr¤rrr¯r)r,r:r<r¥r$r$r%r£Xs   zConnectedDatagramPort.writecCs t |j¡Sr))rrYr­r+r$r$r%rGoszConnectedDatagramPort.getPeer)ržrkNN) rOrPrQrRr-rr®rKr£rGr$r$r$r%rªsÿ   rª)sN        .2  r Mÿ