o 9Bi,%@sdZddlZddlmZmZddlmZddlmZm Z m Z ddl m Z m Z Gddde jZGd d d e jZd Zd Zd ZdZdZdZdZdZdZdZdZdZdZiZe !D]\Z"Z#e"dddkroe"ddee#<q[dS)z Implements the SSH v2 key agent protocol. This protocol is documented in the SSH source code, in the file U{PROTOCOL.agent}. Maintainer: Paul Swartz N) ConchErrorMissingKeyStoreError)keys)NSgetMPgetNS)deferprotocolc@sbeZdZdZddZddZddZdd Zd d Zdd dZ ddZ ddZ ddZ ddZ dS)SSHAgentClientz The client side of the SSH agent protocol. This is equivalent to ssh-add(1) and can be used with either ssh-agent(1) or the SSHAgentServer protocol, also in this package. cCsd|_g|_dSN)buf deferredsselfrW/var/www/html/Trade-python/venv/lib/python3.10/site-packages/twisted/conch/ssh/agent.py__init__s zSSHAgentClient.__init__cCs|j|7_ t|jdkrdStd|jddd}t|jd|kr)dS|jdd||jd|d}|_t|dd}|jd}|tkrW|t dn|t kra| dn| |q)N!Lrz agent failurer ) r lenstructunpackordrpop AGENT_FAILUREerrbackr AGENT_SUCCESScallback)rdatapackLenpacketreqTypedrrr dataReceived s (   zSSHAgentClient.dataReceivedcCs>tdt|d||}|j|t}|j||SNz!LBr) rpackr transportwriterDeferredrappend)rr#r r'r$rrr sendRequest2s   zSSHAgentClient.sendRequestcCs|td}||j|S)z @return: A L{Deferred} which will fire with a list of all keys found in the SSH agent. The list of keys is comprised of (public key blob, comment) tuples. r )r,AGENTC_REQUEST_IDENTITIES addCallback_cbRequestIdentities)rr$rrrrequestIdentities9s  z SSHAgentClient.requestIdentitiescCst|ddtkrtdt|ddtd|ddd}g}|dd}t|D]}t|\}}t|\}}|||fq.|S)z} Unpack a collection of identities into a list of tuples comprised of public key blobs and comments. rrzunexpected response: %irN)rAGENT_IDENTITIES_ANSWERrrrrangerr+)rr numKeysresultiblobcommentrrrr/Cs    z#SSHAgentClient._cbRequestIdentitiesr cCs|}|t|7}|t|S)zK Add a private key blob to the agent's collection of keys. )rr,AGENTC_ADD_IDENTITY)rr7r8reqrrr addIdentitySs  zSSHAgentClient.addIdentitycCs0t|}|t|7}|d7}|t||jS)a Request that the agent sign the given C{data} with the private key which corresponds to the public key given by C{blob}. The private key should have been added to the agent already. @type blob: L{bytes} @type data: L{bytes} @return: A L{Deferred} which fires with a signature for given data created with the given key. )rr,AGENTC_SIGN_REQUESTr. _cbSignData)rr7r r:rrrsignData[s zSSHAgentClient.signDatacCsDt|ddtkrtdt|ddt|ddd}|S)Nrrzunexpected data: %i)rAGENT_SIGN_RESPONSErr)rr signaturerrrr>kszSSHAgentClient._cbSignDatacCst|}|t|S)zp Remove the private key corresponding to the public key in blob from the running agent. )rr,AGENTC_REMOVE_IDENTITY)rr7r:rrrremoveIdentityqs zSSHAgentClient.removeIdentitycCs |tdS)z9 Remove all keys from the running agent. r )r,AGENTC_REMOVE_ALL_IDENTITIESrrrrremoveAllIdentitiesys z"SSHAgentClient.removeAllIdentitiesN)r )__name__ __module__ __qualname____doc__rr%r,r0r/r;r?r>rCrErrrrr s  r c@sheZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZdS)SSHAgentServerz The server side of the SSH agent protocol. This is equivalent to ssh-agent(1) and can be used with either ssh-add(1) or the SSHAgentClient protocol, also in this package. cCs d|_dSr )r rrrrrs zSSHAgentServer.__init__cCs|j|7_ t|jdkrdStd|jddd}t|jd|kr)dS|jdd||jd|d}|_t|dd}t|d}|sT|tdn!t |d|}t |j dddurm|tdt ||ddq)Nrrrrr z agentc_%sr) r rrrrmessagesget sendResponsergetattrfactoryr)rr r!r"r#reqNamefrrrr%s$(  zSSHAgentServer.dataReceivedcCs*tdt|d||}|j|dSr&)rr'rr(r))rr#r r'rrrrMszSSHAgentServer.sendResponsecCsz|dksJt|jj}g}|td||jjD]\}}|t||t|q| t d |dS)zQ Return all of the identities that have been added to the server r rN) rrOrr+rr'valuesrr7rMr2join)rr r4respkeyr8rrragentc_REQUEST_IDENTITIESs  z(SSHAgentServer.agentc_REQUEST_IDENTITIEScCsbt|\}}||jjvr|tdSt|\}}|dksJ|tt|jj|d|dS)z Data is a structure with a reference to an already added key object and some data that the clients wants signed with that key. If the key object wasn't loaded, return AGENT_FAILURE, else return the signature. r r<rN)rrOrrMrr@rsign)rr r7r?rrragentc_SIGN_REQUESTs     z"SSHAgentServer.agentc_SIGN_REQUESTcCst|\}}|dkr d}n|dkrd}ntd|t||d}t|\}}tjj|dd}||f|jj|<|t d d S) z Adds a private key to the agent's collection of identities. On subsequent interactions, the private key can be accessed using only the corresponding public key. sssh-rsasssh-dssr1zunknown blob type: %s private_blobtyper N) rr BadKeyErrorrKey fromStringrOr7rMr)rr keyTyperestnmpr8krrragentc_ADD_IDENTITYs  z"SSHAgentServer.agentc_ADD_IDENTITYcCs:t|\}}tjj|dd}|jj|=|tddS)zR Remove a specific key from the agent's collection of identities. r7r\r N)rrr_r`rOr7rMr)rr r7_rdrrragentc_REMOVE_IDENTITYs z%SSHAgentServer.agentc_REMOVE_IDENTITYcCs$|dksJi|j_|tddS)zL Remove all keys from the agent's collection of identities. r N)rOrrMrrr rrragentc_REMOVE_ALL_IDENTITIESs z+SSHAgentServer.agentc_REMOVE_ALL_IDENTITIEScCs|ttdddS)z v1 message for listing RSA1 keys; superseded by agentc_REQUEST_IDENTITIES, which handles different key types. rrN)rMAGENT_RSA_IDENTITIES_ANSWERrr'rhrrragentc_REQUEST_RSA_IDENTITIESsz,SSHAgentServer.agentc_REQUEST_RSA_IDENTITIEScC|tddS)z v1 message for removing RSA1 keys; superseded by agentc_REMOVE_IDENTITY, which handles different key types. r NrMrrhrrragentc_REMOVE_RSA_IDENTITYz)SSHAgentServer.agentc_REMOVE_RSA_IDENTITYcCrl)z v1 message for removing all RSA1 keys; superseded by agentc_REMOVE_ALL_IDENTITIES, which handles different key types. r Nrmrhrrr agentc_REMOVE_ALL_RSA_IDENTITIESroz/SSHAgentServer.agentc_REMOVE_ALL_RSA_IDENTITIESN)rFrGrHrIrr%rMrVrXrergrirkrnrprrrrrJs   rJrr1rY AGENTC_)$rIrtwisted.conch.errorrrtwisted.conch.sshrtwisted.conch.ssh.commonrrrtwisted.internetrr Protocolr rJAGENTC_REQUEST_RSA_IDENTITIESrjrrAGENTC_REMOVE_RSA_IDENTITY AGENTC_REMOVE_ALL_RSA_IDENTITIESr-r2r=r@r9rBrDrKlocalscopyitemsnamevaluerrrrs8  k