U  W[e@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)logfailureic@seZdZdZdZddZdS) IdentErrorz; Can't determine connection owner; reason unknown. z UNKNOWN-ERRORcCs|jSN)identDescriptionselfr 9/usr/lib/python3/dist-packages/twisted/protocols/ident.py__str__szIdentError.__str__N)__name__ __module__ __qualname____doc__r rr r r r rsrc@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 r rsrc@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 r r'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 r r2src@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   0zIdentServer.lineReceivedcCs|jdSr) transportZloseConnectionr r r r rXszIdentServer.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)ZgetHostZhostZgetPeerrZ maybeDeferredlookupZ addCallback _cbLookupZ addErrbackr#)r r&r'Z serverAddrZ clientAddrr r r r"\szIdentServer.validQuerycCs"|\}}|d||||fdS)Nz%d, %d : USERID : %s : %s)sendLine)r resultsportcportZsysNameZuserIdr r r r+lszIdentServer._cbLookupcCsJ|tr"|d|||jfn$t||d||t|jfdS)Nz%d, %d : ERROR : %s)Zcheckrr,valuererr)r rr.r/r r r r#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 r r*yszIdentServer.lookupN) rrrrr(rr"r+r#r*r r r r r;s rc@sxeZdZdZdZzddlmZefddZ[Wnek rJddZYnXdd Z d d Z d d Z ddZ ddZ dS)ProcServerMixinzIImplements lookup() to grab entries for responses from /proc/net/tcp ZLINUXr)getpwuidcCs ||dSNrr )r uidr6r r r getUsernameszProcServerMixin.getUsernamecCs tdSrr2)r r8r r r r9sc cs4td"}||D]}|VqW5QRXdS)Nz /proc/net/tcp)openreadlinestrip)r fLr r r entriess zProcServerMixin.entriesc Cs&dtttdtdt|dS)N.Z4Bz=L)joinrstrstructZunpackZpackr)r Zhexstrr r r dottedQuadFromHexStringsz'ProcServerMixin.dottedQuadFromHexStringcCs*|d\}}||}t|d}||fS)N:rA)rrEr)r ZpackedZaddrZportr r r unpackAddresss  zProcServerMixin.unpackAddresscCsN|}||d\}}||d\}}t|d}||f||f|fS)Nrr)r<rrGr)r r$r% localAddrZ localPort remoteAddrZ remotePortr8r r r parseLines   zProcServerMixin.parseLinecCsT|D]@}||\}}}||kr|d|dkr|j||fSqtdS)Nr)r?rK SYSTEM_NAMEr9r)r r3r4ZentrIrJr8r r r r*s  zProcServerMixin.lookupN)rrrrrLpwdr6r9 ImportErrorr?rErGrKr*r r r r r5s  r5c@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)rPappendrZDeferredrr,)r r&r'r r r r*s zIdentClient.lookupcCsf|jstd|fnJ|jd\}}}||||jrb|d|jdd|jddfdS)NzUnexpected server response: %rrrSrr)rPrmsgpop parseResponser,)r r$d_r r r r(s  zIdentClient.lineReceivedcCs(|jD]}|dt|qg|_dSr7)rPerrbackr)r reasonqr r r connectionLosts zIdentClient.connectionLostcCs|dd}t|dkr(|t|nbttj|\}}}|dkr||jD]"}|j|krH|||dSqH|t|n| ||fdS)NrFrZERROR) rrrZrrrCr< errorTypesr callback)r Zdeferredr$r%ZportstypeZaddInfoZetr r r rWs    zIdentClient.parseResponseN) rrrrrrrr_rQr*r(r]rWr r r r rOs    rO)rrDZtwisted.internetrZtwisted.protocolsrZtwisted.pythonrrr r! ExceptionrrrrZLineOnlyReceiverrr5rO__all__r r r r s$      Y45