U  W[%@sdZddlmZmZddlZddlmZmZmZddl m Z m Z ddl m Z ddlmZmZddlmZGd d d ejZGd d d ejZd ZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"iZ#e$%&D](\Z'Z(e'dddkre'dde#e(<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 )absolute_importdivisionN)NSgetNSgetMP) ConchErrorMissingKeyStoreError)keys)deferprotocol) itervaluesc@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 deferredsselfr9/usr/lib/python3/dist-packages/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) rlenstructunpackordrpop AGENT_FAILUREZerrbackr AGENT_SUCCESScallback)rdatapackLenpacketreqTypedrrr dataReceived$s(  zSSHAgentClient.dataReceivedcCs>tdt|d||}|j|t}|j||SNz!LBr) rpackr transportwriter ZDeferredrappend)rr%r"r)r&rrr sendRequest7s   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&rrrrequestIdentities?s  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,)rr"numKeysresultiblobcommentrrrr0Js    z#SSHAgentClient._cbRequestIdentitiesrcCs|}|t|7}|t|S)zK Add a private key blob to the agent's collection of keys. )rr-AGENTC_ADD_IDENTITY)rr8r9reqrrr addIdentity[s 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)rr8r"r;rrrsignDatads  zSSHAgentClient.signDatacCsDt|ddtkr,tdt|ddt|ddd}|S)Nrrzunexpected data: %i)rAGENT_SIGN_RESPONSErr)rr"Z signaturerrrr?uszSSHAgentClient._cbSignDatacCst|}|t|S)zp Remove the private key corresponding to the public key in blob from the running agent. )rr-AGENTC_REMOVE_IDENTITY)rr8r;rrrremoveIdentity|szSSHAgentClient.removeIdentitycCs |tdS)z9 Remove all keys from the running agent. r)r-AGENTC_REMOVE_ALL_IDENTITIESrrrrremoveAllIdentitiessz"SSHAgentClient.removeAllIdentitiesN)r)__name__ __module__ __qualname____doc__rr'r-r1r0r<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)rrrrrrszSSHAgentServer.__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)Nrrrrrz agentc_%sr ) rrrrrmessagesget sendResponsergetattrfactoryr)rr"r#r$r%ZreqNamefrrrr's (  zSSHAgentServer.dataReceivedcCs*tdt|d||}|j|dSr()rr)rr*r+)rr%r"r)rrrrMszSSHAgentServer.sendResponsecCsz|dks tt|jj}g}|td|t|jjD](\}}|t| |t|q:| t d |dS)zQ Return all of the identities that have been added to the server rrN) AssertionErrorrrOr r,rr)r rr8rMr3join)rr"r5Zrespkeyr9rrragentc_REQUEST_IDENTITIESs  z(SSHAgentServer.agentc_REQUEST_IDENTITIEScCsbt|\}}||jjkr$|tdSt|\}}|dksrAr:rBrDrKlocalscopyitemsnamevaluerrrrs6  u