U j,%@sdZddlZddlmZmZddlmZddlmZm Z m Z ddl m Z m Z Gddde jZGd d d e jZd Zd Zd ZdZdZdZdZdZdZdZdZdZdZiZe !D](\Z"Z#e"dddkre"ddee#<qdS)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 deferredsselfr;/tmp/pip-unpacked-wheel-j5h_c8_q/twisted/conch/ssh/agent.py__init__szSSHAgentClient.__init__cCs|j|7_t|jdkr dStd|jddd}t|jd|krPdS|jdd||jd|d}|_t|dd}|jd}|tkr|t dq|t kr| dq| |qdS)N!Lrz agent failurer ) r lenstructunpackordrpop AGENT_FAILUREZerrbackr AGENT_SUCCESScallback)rdatapackLenpacketreqTypedrrr dataReceived s(  zSSHAgentClient.dataReceivedcCs>tdt|d||}|j|t}|j||SNz!LBr) rpackr transportwriterZDeferredrappend)rr"rr&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|ddtkr,tdt|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))rrnumKeysresultiblobcommentrrrr-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)rr5r6reqrrr 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)rr5rr8rrrsignData[s  zSSHAgentClient.signDatacCsDt|ddtkr,tdt|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)rr5r8rrrremoveIdentityqszSSHAgentClient.removeIdentitycCs |tdS)z9 Remove all keys from the running agent. r )r*AGENTC_REMOVE_ALL_IDENTITIESrrrrremoveAllIdentitiesysz"SSHAgentClient.removeAllIdentitiesN)r )__name__ __module__ __qualname____doc__rr$r*r.r-r9r=r<rArCrrrrr 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 rrrrrszSSHAgentServer.__init__cCs|j|7_t|jdkr dStd|jddd}t|jd|krPdS|jdd||jd|d}|_t|dd}t|d}|s|tdqt |d|}t |j dddkr|tdt ||ddqdS)Nrrrrr z agentc_%sr) r rrrrmessagesget sendResponsergetattrfactoryr)rrr r!r"ZreqNamefrrrr$s (  zSSHAgentServer.dataReceivedcCs*tdt|d||}|j|dSr%)rr&rr'r()rr"rr&rrrrKszSSHAgentServer.sendResponsecCsz|dks tt|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) AssertionErrorrrMrr)rr&valuesrr5rKr0join)rrr2respkeyr6rrragentc_REQUEST_IDENTITIESs  z(SSHAgentServer.agentc_REQUEST_IDENTITIEScCsbt|\}}||jjkr$|tdSt|\}}|dksrsign)rrr5r=rrragentc_SIGN_REQUESTs     z"SSHAgentServer.agentc_SIGN_REQUESTcCst|\}}|dkrd}n|dkr(d}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-dssr/zunknown blob type: %sZ private_blobtyper N) rrZ BadKeyErrorrKey fromStringrMr5rKr)rrZkeyTyperestZnmpr6krrragentc_ADD_IDENTITYs  z"SSHAgentServer.agentc_ADD_IDENTITYcCs:t|\}}tjj|dd}|jj|=|tddS)zR Remove a specific key from the agent's collection of identities. r5rYr N)rrr[r\rMr5rKr)rrr5_r^rrragentc_REMOVE_IDENTITYs z%SSHAgentServer.agentc_REMOVE_IDENTITYcCs$|dks ti|j_|tddS)zL Remove all keys from the agent's collection of identities. r N)rOrMrrKrrrrrragentc_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)rKAGENT_RSA_IDENTITIES_ANSWERrr&rbrrragentc_REQUEST_RSA_IDENTITIESsz,SSHAgentServer.agentc_REQUEST_RSA_IDENTITIEScCs|tddS)z v1 message for removing RSA1 keys; superseded by agentc_REMOVE_IDENTITY, which handles different key types. r NrKrrbrrragentc_REMOVE_RSA_IDENTITYsz)SSHAgentServer.agentc_REMOVE_RSA_IDENTITYcCs|tddS)z v1 message for removing all RSA1 keys; superseded by agentc_REMOVE_ALL_IDENTITIES, which handles different key types. r Nrfrbrrr agentc_REMOVE_ALL_RSA_IDENTITIESsz/SSHAgentServer.agentc_REMOVE_ALL_RSA_IDENTITIESN)rDrErFrGrr$rKrTrVr_rarcrergrhrrrrrHs  rHrr/rW ZAGENTC_)$rGrZtwisted.conch.errorrrZtwisted.conch.sshrZtwisted.conch.ssh.commonrrrZtwisted.internetrr Protocolr rHZAGENTC_REQUEST_RSA_IDENTITIESrdrrZAGENTC_REMOVE_RSA_IDENTITYZ AGENTC_REMOVE_ALL_RSA_IDENTITIESr+r0r;r>r7r@rBrIlocalscopyitemsnamevaluerrrrs2  k