\d,%tdZddlZddlmZmZddlmZddlmZm Z m Z ddl m Z m Z Gdde jZGd d e jZd Zd Zd ZdZdZdZdZdZdZdZdZdZdZiZe !D] \Z"Z#e"dddkr e"ddee#<!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)deferprotocolcPeZdZdZdZdZdZdZdZddZ d Z d Z d Z d Z d S)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. c"d|_g|_dSN)buf deferredsselfs 7lib/python3.11/site-packages/twisted/conch/ssh/agent.py__init__zSSHAgentClient.__init__srcd|xj|z c_ t|jdkrdStjd|jddd}t|jd|zkrdS|jdd|z|jd|zdc}|_t |dd}|jd}|tkr#|tdn6|tkr| dn| | )N!Lrz agent failurer) rlenstructunpackordrpop AGENT_FAILUREerrbackr AGENT_SUCCESScallback)rdatapackLenpacketreqTypeds r dataReceivedzSSHAgentClient.dataReceived s D #48}}!!mD$(2A2,77:G48}}q7{**#xAK8$(1w;==:Q FDH&1+&&G""1%%A-'' *_556666M)) 3 6""" #rctjdt|dz||z}|j|t j}|j||SNz!LBr) rpackr transportwriter Deferredrappend)rr&r#r+r's r sendRequestzSSHAgentClient.sendRequest2sb{5#d))a-99D@ T""" N   a   rcp|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)r0AGENTC_REQUEST_IDENTITIES addCallback_cbRequestIdentities)rr's rrequestIdentitiesz SSHAgentClient.requestIdentities9s4   6 < < d/000rct|ddtkr'tdt|ddztjd|ddd}g}|dd}t |D]=}t |\}}t |\}}|||f>|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#numKeysresultiblobcomments rr4z#SSHAgentClient._cbRequestIdentitiesCs tAaCy>>4 4 46T!A#YGHH H-d1Q3i003ABBxw + +AtJD$!$KKMGT MM4/ * * * * rrc`|}|t|z }|t|S)zK Add a private key blob to the agent's collection of keys. )rr0AGENTC_ADD_IDENTITY)rr=r>reqs r addIdentityzSSHAgentClient.addIdentitySs0 r'{{ 3S999rct|}|t|z }|dz }|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. )rr0AGENTC_SIGN_REQUESTr3 _cbSignData)rr=r#rAs rsignDatazSSHAgentClient.signData[sQhh r$xx "" 3S99EEdFVWWWrct|ddtkr'tdt|ddzt|ddd}|S)Nrrzunexpected data: %i)rAGENT_SIGN_RESPONSErr)rr# signatures rrFzSSHAgentClient._cbSignDataks\ tAaCy>>0 0 02Sac^^CDD D$qrr(OOA& rcVt|}|t|S)zp Remove the private key corresponding to the public key in blob from the running agent. )rr0AGENTC_REMOVE_IDENTITY)rr=rAs rremoveIdentityzSSHAgentClient.removeIdentityqs& hh 6<<%%mS999.000&* ! rctjdt|dz||z}|j|dSr*)rr+rr,r-)rr&r#r+s rr\zSSHAgentServer.sendResponses@{5#d))a-99D@ T"""""rc|dksJt|jj}g}|t jd||jjD][\}}|t||t|\| td |dS)zQ Return all of the identities that have been added to the server rrN) rr^rr/rr+valuesrr=r\r8join)rr#r:respkeyr>s ragentc_REQUEST_IDENTITIESz(SSHAgentServer.agentc_REQUEST_IDENTITIESss{{{{dl'(( FKg../// L-4466 % %LC KK388:: ' ' ' KK7 $ $ $ $ 1388D>>BBBBBrcRt|\}}||jjvr|tdSt|\}}|dksJ|t t |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. rrDrN)rr^rr\rrIrsign)rr#r=rGs ragentc_SIGN_REQUESTz"SSHAgentServer.agentc_SIGN_REQUESTs 4[[ d t|( ( ($$]C88 8t$*****  DL$5d$;A$>$C$CH$M$M!N!N     rct|\}}|dkrd}n |dkrd}ntjd|zt||d}t|\}}tj|d}||f|jj|<|td 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-dssr7zunknown blob type: %s private_blobtyperN) rr BadKeyErrorrKey fromStringr^r=r\r!)rr#keyTyperestnmpr>ks ragentc_ADD_IDENTITYz"SSHAgentServer.agentc_ADD_IDENTITYsd  j CC  " "CC"#:W#DEE ET3  d  H  >  : :'('l !&&((# ------rct|\}}tj|d}|jj|=|tddS)zR Remove a specific key from the agent's collection of identities. r=rorN)rrrrrsr^r=r\r!)rr#r=_rws ragentc_REMOVE_IDENTITYz%SSHAgentServer.agentc_REMOVE_IDENTITYs^++a H  6  2 2 L affhh ' ------rcd|dksJi|j_|tddS)zL Remove all keys from the agent's collection of identities. rN)r^rr\r!rr#s ragentc_REMOVE_ALL_IDENTITIESz+SSHAgentServer.agentc_REMOVE_ALL_IDENTITIESs7s{{{{  ------rcb|ttjdddS)z v1 message for listing RSA1 keys; superseded by agentc_REQUEST_IDENTITIES, which handles different key types. rrN)r\AGENT_RSA_IDENTITIES_ANSWERrr+r}s ragentc_REQUEST_RSA_IDENTITIESz,SSHAgentServer.agentc_REQUEST_RSA_IDENTITIESs- 5v{47K7KLLLLLrc<|tddS)z v1 message for removing RSA1 keys; superseded by agentc_REMOVE_IDENTITY, which handles different key types. rNr\r!r}s ragentc_REMOVE_RSA_IDENTITYz)SSHAgentServer.agentc_REMOVE_RSA_IDENTITY ------rc<|tddS)z v1 message for removing all RSA1 keys; superseded by agentc_REMOVE_ALL_IDENTITIES, which handles different key types. rNrr}s r agentc_REMOVE_ALL_RSA_IDENTITIESz/SSHAgentServer.agentc_REMOVE_ALL_RSA_IDENTITIESrrN)rQrRrSrTrr(r\rgrjrxr{r~rrrrUrrrWrWs (### C C C    ...2......MMM........rrWrr7rl AGENTC_)$rTrtwisted.conch.errorrrtwisted.conch.sshrtwisted.conch.ssh.commonrrrtwisted.internetr r Protocolr rWAGENTC_REQUEST_RSA_IDENTITIESrrr!AGENTC_REMOVE_RSA_IDENTITY AGENTC_REMOVE_ALL_RSA_IDENTITIESr2r8rErIr@rLrOrZlocalscopyitemsnamevaluerUrrrs @@@@@@@@""""""5555555555,,,,,,,,hChChChChCX&hChChCV@.@.@.@.@.X&@.@.@.F!"  #$ ! 688==??((**##KD% BQBx9qrr(##r