U Acj @sdZddlZddlmZddlmZddlmZmZdZ dZ Gdd d e Z Gd d d e Z Gd d d e ZGddde ZGdddejZGdddZGdddejZd d d ddddgZdS)z Ident protocol implementation. N)defer)basic)failurelogic@s"eZdZdZdZedddZdS) IdentErrorz; Can't determine connection owner; reason unknown. z UNKNOWN-ERROR)returncCs|jSN)identDescriptionselfr ;/tmp/pip-unpacked-wheel-nrvc_22y/twisted/protocols/ident.py__str__szIdentError.__str__N)__name__ __module__ __qualname____doc__r strrr r r rrsrc@seZdZdZdZdS)NoUserz The connection specified by the port pair is not currently in use or currently not owned by an identifiable entity. zNO-USERNrrrrr r r r rrsrc@seZdZdZdZdS) InvalidPorta Either the local or foreign port was improperly specified. This should be returned if either or both of the port ids were out of range (TCP port numbers are from 1-65535), negative integers, reals or in any fashion not recognized as a non-negative integer. z INVALID-PORTNrr r r rr'src@seZdZdZdZdS) HiddenUserz The server was able to identify the user of this port, but the information was not returned at the request of the user. z HIDDEN-USERNrr r r rr2src@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) IdentServera The Identification Protocol (a.k.a., "ident", a.k.a., "the Ident Protocol") provides a means to determine the identity of a user of a particular TCP connection. Given a TCP port number pair, it returns a character string which identifies the owner of that connection on the server's system. Server authors should subclass this class and override the lookup method. The default implementation returns an UNKNOWN-ERROR response for every query. cCs|d}t|dkr |nztt|\}}Wntk rN|YnVXt|krdtkrnn&t|kr|tkrnn|||n| t t ||dS)N,) splitlen invalidQuerymapint ValueError _MIN_PORT _MAX_PORT validQuery _ebLookuprZFailurer)r lineparts portOnServer portOnClientr r r lineReceivedHs.      zIdentServer.lineReceivedcCs|jdSr ) transportZloseConnectionr r r rr\szIdentServer.invalidQuerycCsL|jj|f}|jj|f}t|j|||j|| |j ||dS)z Called when a valid query is received to look up and deliver the response. @param portOnServer: The server port from the query. @param portOnClient: The client port from the query. N) r+ZgetHosthostZgetPeerrZ maybeDeferredlookupZ addCallback _cbLookupZ addErrbackr%)r r(r)Z serverAddrZ clientAddrr r rr$_szIdentServer.validQuerycCs"|\}}|d||||fdS)Nz%d, %d : USERID : %s : %s)sendLine)r resultsportcportZsysNameZuserIdr r rr.mszIdentServer._cbLookupcCsJ|tr"|d|||jfn$t||d||t|jfdS)Nz%d, %d : ERROR : %s)checkrr/valuererr)r rr1r2r r rr%qs   zIdentServer._ebLookupcCs tdS)aq Lookup user information about the specified address pair. Return value should be a two-tuple of system name and username. Acceptable values for the system name may be found online at:: U{http://www.iana.org/assignments/operating-system-names} This method may also raise any IdentError subclass (or IdentError itself) to indicate user information will not be provided for the given query. A Deferred may also be returned. @param serverAddress: A two-tuple representing the server endpoint of the address being queried. The first element is a string holding a dotted-quad IP address. The second element is an integer representing the port. @param clientAddress: Like I{serverAddress}, but represents the client endpoint of the address being queried. Nr)r serverAddress clientAddressr r rr-zszIdentServer.lookupN) rrrrr*rr$r.r%r-r r r rr;s  rc@szeZdZdZdZzddlmZefddZ[Wnek rLdddZYnXd d Z d d Z d dZ ddZ ddZ dS)ProcServerMixinzDImplements lookup() to grab entries for responses from /proc/net/tcpZLINUXr)getpwuidcCs ||dSNrr r uidr:r r r getUsernameszProcServerMixin.getUsernameNcCs tdSr r6r<r r rr>sc cs4td"}||D]}|VqW5QRXdS)Nz /proc/net/tcp)openreadlinestrip)r fLr r rentriess zProcServerMixin.entriesc Cs&dtttdtdt|dS)N.Z4Bz=L)joinrrstructunpackpackr )r Zhexstrr r rdottedQuadFromHexStringsz'ProcServerMixin.dottedQuadFromHexStringcCs*|d\}}||}t|d}||fS)N:rF)rrKr )r packedaddrportr r r unpackAddresss  zProcServerMixin.unpackAddresscCsN|}||d\}}||d\}}t|d}||f||f|fS)Nrr)rArrPr )r r&r' localAddrZ localPort remoteAddrZ remotePortr=r r r parseLines   zProcServerMixin.parseLinecCsT|D]@}||\}}}||kr|d|dkr|j||fSqtdS)Nr)rDrT SYSTEM_NAMEr>r)r r7r8entrRrSr=r r rr-s  zProcServerMixin.lookup)N)rrrrrUpwdr:r> ImportErrorrDrKrPrTr-r r r rr9s  r9c@s@eZdZeeeefZddZddZ ddZ ddZ d d Z d S) IdentClientcCs g|_dSr )queriesr r r r__init__szIdentClient.__init__cCsR|jt||ft|jdkr2|jddS|d||f|jddS)zK Lookup user information about the specified address pair. rr%d, %d)rZappendrZDeferredrr/)r r(r)r r rr-s zIdentClient.lookupcCsf|jstd|nJ|jd\}}}||||jrb|d|jdd|jddfdS)NzUnexpected server response: rr]rr)rZrmsgpop parseResponser/)r r&d_r r rr*s  zIdentClient.lineReceivedcCs(|jD]}|dt|qg|_dSr;)rZerrbackr)r reasonqr r rconnectionLosts zIdentClient.connectionLostcCs|dd}t|dkr(|t|nbttj|\}}}|dkr||jD]"}|j|krH|||dSqH|t|n| ||fdS)NrLrERROR) rrrdrrrrA errorTypesr callback)r deferredr&r'ZportstypeZaddInfoetr r rras    zIdentClient.parseResponseN) rrrrrrrrjr[r-r*rgrar r r rrYs    rY)rrHZtwisted.internetrZtwisted.protocolsrZtwisted.pythonrrr"r# ExceptionrrrrZLineOnlyReceiverrr9rY__all__r r r rs*      Y20