o gQ@sdZddlZddlZddlZddlZddlmZddlm Z ddl Z ddl m Z dZdZd Zd d Zd d ZeddkrBejZnejZGdddejZddZGdddZdS)zP Functions for communicating with Pageant, the basic windows ssh agent program. N) zero_byte)b)_winapilPi JcCstjjddS)NsPageant)ctypeswindlluser32 FindWindowAr r ?/usr/local/lib/python3.10/dist-packages/paramiko/win_pageant.py_get_pageant_window_object+sr cCs ttS)z Check to see if there is a "Pageant" agent we can talk to. This checks both if we have the required libraries (win32all or ctypes) and if there is a Pageant currently running. )boolr r r r r can_talk_to_agent/s r64bitc@s,eZdZdZdefdejjfdejfgZ dS)COPYDATASTRUCTzy ctypes implementation of http://msdn.microsoft.com/en-us/library/windows/desktop/ms649010%28v=vs.85%29.aspx num_data data_sizedata_locN) __name__ __module__ __qualname____doc__ ULONG_PTRrwintypesDWORDc_void_p_fields_r r r r r?s  rc Cst}|sdSdtd}t|tt}|Y||tdt |t }| \}}t t ||}tjj|tt|t|}|dkrj|d|d} td| d} | || WdS WddS1svwYdS)za Communication with the Pageant process is done through a shared memory-mapped file. NPageantRequest08xrrz>I)r thread get_identr MemoryMap_AGENT_MAX_MSGLEN get_security_attributes_for_userwritearrayrr buffer_infor_AGENT_COPYDATA_IDrrr SendMessageAwin32con_WM_COPYDATAsizeofbyrefseekreadstructunpack) msghwndmap_namepymap char_bufferchar_buffer_addresschar_buffer_sizecdsresponsedatalenretlenr r r _query_pageantLs2      $r=c@s0eZdZdZddZddZddZdd Zd S) PageantConnectiona Mock "connection" to an agent which roughly approximates the behavior of a unix local-domain socket (as used by Agent). Requests are sent to the pageant daemon via special Windows magick, and responses are buffered back for subsequent reads. cCs d|_dSN _responseselfr r r __init__zs zPageantConnection.__init__cCst||_dSr?)r=rA)rCdatar r r send}szPageantConnection.sendcCs@|jdurdS|jd|}|j|d|_|jdkrd|_|S)Nr@)rCnretr r r recvs  zPageantConnection.recvcCsdSr?r rBr r r closeszPageantConnection.closeN)rrrrrDrFrJrKr r r r r>rs  r>)rr'ctypes.wintypesrplatformr0paramiko.commonr paramiko.utilr_threadr!rGrr)r$r+r r architecturec_uint64rc_uint32 Structurerr=r>r r r r s(     &