HEX

File: //lib64/python3.6/site-packages/gpg/__pycache__/core.cpython-36.pyc
3

���\j��@sddlmZmZmZddlZddlZddlZddlZddlm	Z	ddl
mZmZddlm
Z
ddlm
Z
ddlmZ[[[Gd	d
�d
e�ZGdd�de�ZGd
d�de�Zdd�Zdd�Zdd�Zdd�Zdd�Zd%dd�Ze�dd�Zdd�Zd&dd �Zd!d"�Zd#d$�ZdS)'�)�absolute_import�print_function�unicode_literalsN�)�gpgme)�
errorcheck�
GPGMEError)�	constants)�errors)�utilcs�eZdZdZdd�Z�fdd�Zdd�Zdd	�Zd
d�Ze	dd
��Z
e	dd��Zdd�Ze
�Zddd�Zejd�Zdd�Z�fdd�Z�ZS)�GpgmeWrapperz>Base wrapper class

    Not to be instantiated directly.

    cCsd|_||_dS)N)�_callback_excinfo�wrapped)�selfr�r�/usr/lib64/python3.6/core.py�__init__3szGpgmeWrapper.__init__csdjtt|�j�|j�S)Nz	<{}/{!r}>)�format�superr�__repr__r)r)�	__class__rrr7szGpgmeWrapper.__repr__csPdjt�jj�g}�fdd��jD�}|r@|jdjdj|���djdj|��S)Nz{}.{}csg|]}t�|�r|�qSr)�getattr)�.0�f)rrr�
<listcomp>=sz(GpgmeWrapper.__str__.<locals>.<listcomp>z({})� z<{}>)r�__name__r�_boolean_properties�append�join)rZacc�flagsr)rr�__str__;s
zGpgmeWrapper.__str__cCstt|j��S)N)�hash�reprr)rrrr�__hash__CszGpgmeWrapper.__hash__cCs$|dkrdSt|j�t|j�kSdS)NF)r#r)r�otherrrr�__eq__FszGpgmeWrapper.__eq__cCs
t��dS)z]The name of the c type wrapped by this class

        Must be set by child classes.

        N)�NotImplementedError)rrrr�_ctypeLszGpgmeWrapper._ctypecCs
t��dS)zgThe common prefix of c functions wrapped by this class

        Must be set by child classes.

        N)r')rrrr�_cprefixUszGpgmeWrapper._cprefixcCs
t��dS)z�Must be implemented by child classes.

        This function must return a trueish value for all c functions
        returning gpgme_error_t.N)r')r�namerrr�_errorcheck^szGpgmeWrapper._errorcheckFNcs�ttdj|j|���ttdj|j|����fdd�}�fdd�}t||dj|�d�}t|j||�|rv||t|��n||�SdS)	Nz{}get_{}z{}set_{}cst�|j��S)N)�boolr)�slf)�get_funcrr�getlsz1GpgmeWrapper.__wrap_boolean_property.<locals>.getcs�|jt|��dS)N)rr,)r-�value)�set_funcrr�set_osz2GpgmeWrapper.__wrap_boolean_property.<locals>.set_z{} flag)�doc)rrrr)�property�setattrrr,)r�keyZdo_setr0r/r2�pr)r.r1rZ__wrap_boolean_propertyhsz$GpgmeWrapper.__wrap_boolean_propertyz$gpgme_([^(]*)\(([^,]*), (.*\) -> .*)cs�|ddks�jdkrdS|�jkr.�j|�S�j|�tt����j��r\��fdd��n�fdd���jjdt�d��}|�_t	�j
|����fd	d
�}||_|S)z7On-the-fly generation of wrapper methods and propertiesr�_Ncs*�|jf|��}|jr tj|�t|��S)N)rr
r�gpg_raise_callback_exceptionr)r-�args�result)�funcr*rr�	_funcwrap�s
z+GpgmeWrapper.__getattr__.<locals>._funcwrapcs$�|jf|��}|jr tj|�|S)N)rr
rr9)r-r:r;)r<rrr=�s
z\2.\1(\3�__doc__cs��f|��S)Nr)r:)r=rrr�wrapper�sz)GpgmeWrapper.__getattr__.<locals>.wrapper)r)r�$_GpgmeWrapper__wrap_boolean_propertyrrr+�_munge_docstring�subr>r5r)rr6r3r?r)r=r<r*rr�__getattr__|s




zGpgmeWrapper.__getattr__cs0||jkr|j|d|�ntt|�j||�dS)z#On-the-fly generation of propertiesTN)rr@rr�__setattr__)rr6r0)rrrrD�s
zGpgmeWrapper.__setattr__)FN)r�
__module__�__qualname__r>rrr!r$r&r4r(r)r+�setrr@�re�compilerArCrD�
__classcell__rr)rrr,s		

(rc	s<eZdZdZdddgejejddf�fdd�	Zdd�Zdd	�Z	gd
ddddddd
f	dd�Z
dgd
d�Zdejfdd�Z
ddgfdd�Zdd�Zdhdd�Zdidd�Zdjdd�Zddejjjdfdd�Zdkdd�Zdld d!�Zd"d#�Zd$d%�Zdmd&d'�Zd(d)�Zdnd*d+�Zdod,d-�Zed.d/��Zej d0d/��Zed1d2��Z!e!j d3d2��Z!ed4d5��Z"e"j d6d5��Z"ed7d8��Z#e#j d9d8��Z#d:Z$d;Z%d<d=�Z&d>d?d@hZ'dAdB�Z(dCdD�Z)dEdF�Z*dGdH�Z+dIdJ�Z,dpdKdL�Z-dMdN�Z.dOdP�Z/dqdQdR�Z0dSdT�Z1drdUdV�Z2dWdX�Z3dsdYdZ�Z4d[d\�Z5ed]d^��Z6d_d`�Z7dtdadb�Z8dcdd�Z9dedf�Z:�Z;S)u�ContextaContext for cryptographic operations

    All cryptographic operations in GPGME are performed within a
    context, which contains the internal state of the operation as
    well as configuration parameters.  By using several contexts you
    can run several cryptographic operations in parallel, with
    different configuration.

    Access to a context must be synchronized.

    FNc	
sz|rd|_n0tj�}	ttj|	��tj|	�}tj|	�d|_tt|�j	|�||_
||_||_||_
||_||_||_dS)aConstruct a context object

        Keyword arguments:
        armor		-- enable ASCII armoring (default False)
        textmode	-- enable canonical text mode (default False)
        offline		-- do not contact external key sources (default False)
        signers		-- list of keys used for signing (default [])
        pinentry_mode	-- pinentry mode (default PINENTRY_MODE_DEFAULT)
        protocol	-- protocol to use (default PROTOCOL_OpenPGP)
        home_dir        -- state directory (default is the engine default)

        FTN)�ownrZnew_gpgme_ctx_t_pr�	gpgme_newZgpgme_ctx_t_p_valueZdelete_gpgme_ctx_t_prrKr�armor�textmode�offline�signers�
pinentry_mode�protocol�home_dir)
rrNrOrPrQrRrSrrT�tmp)rrrr�s

zContext.__init__cCs&|s|dkrdS|jdtj�|j�S)zxRead helper

        Helper function to retrieve the results of an operation, or
        None if SINK is given.
        Nr)�seek�os�SEEK_SET�read)r�sink�datarrr�__read__�szContext.__read__cCs
dj|�S)Nz�Context(armor={0.armor}, textmode={0.textmode}, offline={0.offline}, signers={0.signers}, pinentry_mode={0.pinentry_mode}, protocol={0.protocol}, home_dir={0.home_dir}))r)rrrrr�szContext.__repr__Tcs�|r|nt�}d}||tjO}||tjO}||tjO}||	tjO}||
tjO}�dk	r�|j}
t|dd�}tj	|_d�fdd�	}|j
|�z�y*|r�|j||||�n|j||||�Wn�t
jk
�r|}z�|j�}|r�|j�nd}|j||�||f}|j�t
jk�r.|j�r.t
j|j|j|d��|j�t
jk�rb|j�}|j�rbt
j|j|j|d��||_|�WYdd}~XnXWd�dk	�r�|
|_|�r�|j
|dd��X|j�}|j�s�t�|�r�|j�nd}|�s�|j�s�t�|j||�||fS)	a%Encrypt data

        Encrypt the given plaintext for the given recipients.  If the
        list of recipients is empty, the data is encrypted
        symmetrically with a passphrase.

        The passphrase can be given as parameter, using a callback
        registered at the context, or out-of-band via pinentry.

        Keyword arguments:
        recipients	-- list of keys to encrypt to
        sign		-- sign plaintext (default True)
        sink		-- write result to sink instead of returning it
        passphrase	-- for symmetric encryption
        always_trust	-- always trust the keys (default False)
        add_encrypt_to	-- encrypt to configured additional keys (default False)
        prepare		-- (ui) prepare for encryption (default False)
        expect_sign	-- (ui) prepare for signing (default False)
        compress	-- compress plaintext (default True)

        Returns:
        ciphertext	-- the encrypted data (or None if sink is given)
        result		-- additional information about the encryption
        sign_result	-- additional information about the signature(s)

        Raises:
        InvalidRecipients -- if encryption using a particular key failed
        InvalidSigners	-- if signing using a particular key failed
        GPGMEError	-- as signaled by the underlying library

        rN�_passphrase_cbcs�S)Nr)�hint�desc�prev_bad�hook)�
passphraserr�
passphrase_cb(sz&Context.encrypt.<locals>.passphrase_cb)�error�resultsr)N)�Datar	ZENCRYPT_ALWAYS_TRUSTZENCRYPT_NO_ENCRYPT_TOZENCRYPT_PREPAREZENCRYPT_EXPECT_SIGNZENCRYPT_NO_COMPRESSrRr�PINENTRY_MODE_LOOPBACK�set_passphrase_cbZop_encrypt_signZ
op_encryptr
rZop_encrypt_result�op_sign_resultr\�getcodeZUNUSABLE_PUBKEYZinvalid_recipientsZInvalidRecipientsrd�UNUSABLE_SECKEY�invalid_signers�InvalidSignersre�AssertionError)r�	plaintextZ
recipients�signrZrbZalways_trustZadd_encrypt_toZprepareZexpect_sign�compress�
ciphertextr �old_pinentry_mode�old_passphrase_cbrc�er;Z
sig_resultrer)rbr�encrypt�sZ*

zContext.encryptcs<d}d}|r|nt�}�dk	rP|j}t|dd�}	tj|_d�fdd�	}
|j|
�z�yVt|t�rd|}n$|dkr�tj	dt
d�d}n|}d}|r�|j||�n|j||�WnXt
jk
�r}z8|j�}|r�|j�}
nd}
|j||�||
f|_|�WYdd}~XnXWd�dk	�r.||_|	�r.|j|	d	d��X|j�}|�rH|j�}
nd}
|j||�||
f}|j�rvt
j|j|d
��|�r8ttdd�|
j��|
_|dk	�r8g}x�|D]x}d}x\|jD]R}xB|
jD]8}|jtj@d
k�rܐq�|j�r�|j|jk�r�d}P�q�W|�r�P�q�W|�s�|j|��q�W|�r8t
j|
||d
��|S)a�Decrypt data

        Decrypt the given ciphertext and verify any signatures.  If
        VERIFY is an iterable of keys, the ciphertext must be signed
        by all those keys, otherwise a MissingSignatures error is
        raised.  Note: if VERIFY is an empty iterable, that is treated
        the same as passing verify=True (that is, verify signatures
        and return data about any valid signatures found, but no
        signatures are required and no MissingSignatures error will be
        raised).

        If the ciphertext is symmetrically encrypted using a
        passphrase, that passphrase can be given as parameter, using a
        callback registered at the context, or out-of-band via
        pinentry.

        Keyword arguments:
        sink            -- write result to sink instead of returning it
        passphrase      -- for symmetric decryption
        verify          -- check signatures (boolean or iterable of keys,
                           see above) (default True)

        Returns:
        plaintext       -- the decrypted data (or None if sink is given)
        result          -- additional information about the decryption
        verify_result   -- additional information about the valid
                           signature(s) found

        Raises:
        UnsupportedAlgorithm -- if an unsupported algorithm was used
        MissingSignatures    -- if expected signatures are missing or bad
        GPGMEError           -- as signaled by the underlying library

        FNr]cs�S)Nr)r^r_r`ra)rbrrrc�sz&Context.decrypt.<locals>.passphrase_cbzTctx.decrypt called with verify=None, should be bool or iterable (treating as False).)�categoryTr)recSs|jtjkS)N)�statusr
�NO_ERROR)�srrr�<lambda>�sz!Context.decrypt.<locals>.<lambda>r)N)rfrRrr	rgrh�
isinstancer,�warnings�warn�DeprecationWarningZop_decrypt_verifyZ
op_decryptr
rZop_decrypt_result�op_verify_resultr\reZunsupported_algorithmZUnsupportedAlgorithm�list�filter�
signatures�subkeys�summary�SIGSUM_VALID�can_sign�fprr�MissingSignatures)rrrrZrb�verifyZdo_sig_verificationZ
required_keysrorsrtrcrur;Z
verify_resultre�missingr6�ok�subkey�sigr)rbr�decryptTs|#






zContext.decryptcCs�|r|nt�}y|j|||�Wnrtjk
r�}zT|j||�|j�f}|j�tjkrx|djrxtj	|dj|j
|d��||_|�WYdd}~XnX|j�}|js�t�|j||�|fS)aSign data

        Sign the given data with either the configured default local
        key, or the 'signers' keys of this context.

        Keyword arguments:
        mode		-- signature mode (default: normal, see below)
        sink		-- write result to sink instead of returning it

        Returns:
        either
          signed_data	-- encoded data and signature (normal mode)
          signature	-- only the signature data (detached mode)
          cleartext	-- data and signature as text (cleartext mode)
            (or None if sink is given)
        result		-- additional information about the signature(s)

        Raises:
        InvalidSigners	-- if signing using a particular key failed
        GPGMEError	-- as signaled by the underlying library

        r)rdreN)
rfZop_signr
rr\rirjrkrlrmrdrern)rr[rZ�modeZ
signeddatarurer;rrrrp�s 
zContext.signc
CsT|r
d}n|r|nt�}y&|r.|j||d�n|j|d|�Wn>tjk
r|}z |j||�|j�f|_|�WYdd}~XnX|j||�|j�f}tdd�|djD��r�tj	|d|d��t
�}xr|D]j}	d}
xR|	jD]H}x<|djD].}|jt
j@dkr�q�|jr�|j|jkr�d}
Pq�W|
r�Pq�W|
s�|j|	�q�W|�rPtj|d||d��|S)	a�Verify signatures

        Verify signatures over data.  If VERIFY is an iterable of
        keys, the ciphertext must be signed by all those keys,
        otherwise an error is raised.

        Keyword arguments:
        signature	-- detached signature data
        sink		-- write result to sink instead of returning it

        Returns:
        data		-- the plain data
            (or None if sink is given, or we verified a detached signature)
        result		-- additional information about the signature(s)

        Raises:
        BadSignatures	-- if a bad signature is encountered
        MissingSignatures -- if expected signatures are missing or bad
        GPGMEError	-- as signaled by the underlying library

        Ncss|]}|jtjkVqdS)N)rxr
ry)rrzrrr�	<genexpr>sz!Context.verify.<locals>.<genexpr>r)reFrT)rfZ	op_verifyr
rr\r�re�anyr�Z
BadSignaturesr�r�r�r	r�r�r�rr�)
rZsigned_dataZ	signaturerZr�r[rurer�r6r�r�r�rrrr��s>
zContext.verifycCs�y.|j|�|j�}|jdkr&tj}ntj}Wn�tk
r�}zl|tjkrd|j	dkr\tj
}q�tj}nB|tkr�t
|d�dkr�tj}n$|tkr�t
|d�dkr�tj}ntj}WYdd}~XnX|tjkr�|}n|}|S)a�Import data

        Imports the given data into the Context.

        Returns:
                -- an object describing the results of imported or updated
                   keys

        Raises:
        TypeError      -- Very rarely.
        GPGMEError     -- as signaled by the underlying library:

                          Import status errors, when they occur, will usually
                          be of NODATA.  NO_PUBKEY indicates something
                          managed to run the function without any
                          arguments, while an argument of None triggers
                          the first NODATA of errors.GPGME in the
                          exception.
        rzNo data�decodeT�encodeN)Z	op_importZop_import_resultZ
consideredr	ZSTATUS_IMPORT_PROBLEMZSTATUS_KEY_CONSIDERED�	Exceptionr
rZcode_strZ
STATUS_NODATAZSTATUS_FILE_ERROR�	TypeError�hasattrZSTATUS_NO_PUBKEYZSTATUS_ERROR)rr[r;rxruZ
import_resultrrr�
key_import3s(





zContext.key_importcCstt�}d}y(|j|||�|jdtj�|j�}Wn&tk
rX}z
|}WYdd}~XnXt|�dkrl|}nd}|S)aQExport keys.

        Exports public keys matching the pattern specified.  If no
        pattern is specified then exports all available keys.

        Keyword arguments:
        pattern	-- return keys matching pattern (default: all keys)

        Returns:
                -- A key block containing one or more OpenPGP keys in
                   either ASCII armoured or binary format as determined
                   by the Context().  If there are no matching keys it
                   returns None.

        Raises:
        GPGMEError     -- as signaled by the underlying library.
        rN)rf�	op_exportrVrWrXrYr�len)r�patternr[r��	pk_resultrur;rrr�
key_exportbszContext.key_exportcCsvt�}tj}y(|j|||�|jdtj�|j�}Wn&tk
rZ}z
|}WYdd}~XnXt	|�dkrn|}nd}|S)ayExport keys.

        Exports public keys matching the pattern specified in a
        minimised format.  If no pattern is specified then exports all
        available keys.

        Keyword arguments:
        pattern	-- return keys matching pattern (default: all keys)

        Returns:
                -- A key block containing one or more minimised OpenPGP
                   keys in either ASCII armoured or binary format as
                   determined by the Context().  If there are no matching
                   keys it returns None.

        Raises:
        GPGMEError     -- as signaled by the underlying library.
        rN)
rfrZGPGME_EXPORT_MODE_MINIMALr�rVrWrXrYrr�)rr�r[r�r�rur;rrr�key_export_minimal�szContext.key_export_minimalcCsvt�}tj}y(|j|||�|jdtj�|j�}Wn&tk
rZ}z
|}WYdd}~XnXt	|�dkrn|}nd}|S)a�Export secret keys.

        Exports secret keys matching the pattern specified.  If no
        pattern is specified then exports or attempts to export all
        available secret keys.

        IMPORTANT: Each secret key to be exported will prompt for its
        passphrase via an invocation of pinentry and gpg-agent.  If the
        passphrase is not entered or does not match then no data will be
        exported.  This is the same result as when specifying a pattern
        that is not matched by the available keys.

        Keyword arguments:
        pattern	-- return keys matching pattern (default: all keys)

        Returns:
                -- On success a key block containing one or more OpenPGP
                   secret keys in either ASCII armoured or binary format
                   as determined by the Context().
                -- On failure while not raising an exception, returns None.

        Raises:
        GPGMEError     -- as signaled by the underlying library.
        rN)
rfrZGPGME_EXPORT_MODE_SECRETr�rVrWrXrYrr�)rr�r[r�Z	sk_resultrur;rrr�key_export_secret�szContext.key_export_secretccsh|s|j|�|j||�n t|t�s0t|d�}|j|d�|j�}x|rZ|V|j�}qFW|j�dS)aList keys

        Keyword arguments:
        pattern	-- return keys matching pattern (default: all keys)
        secret	-- return only secret keys (default: False)
        mode    -- keylist mode (default: list local keys)
        source  -- read keys from source instead from the keyring
                       (all other options are ignored in this case)

        Returns:
                -- an iterator returning key objects

        Raises:
        GPGMEError	-- as signaled by the underlying library
        )�filerN)Zset_keylist_mode�op_keylist_startr|rfZop_keylist_from_data_start�op_keylist_next�op_keylist_end)rr��secretr��sourcer6rrr�keylist�s


zContext.keylistrcs�tj��r<|j}t|dd�}tj|_d�fdd�	}
|j|
�z�|j||d|d|rXtjj	nd|rftjj
ndB|rvtjjndB|r�tjjndB�dkr�tjj
ndB|r�dntjjB|
r�tjjndB�Wdtj��r�||_|r�|j|dd��X|j�S)a	Create a primary key

        Create a primary key for the user id USERID.

        ALGORITHM may be used to specify the public key encryption
        algorithm for the new key.  By default, a reasonable default
        is chosen.  You may use "future-default" to select an
        algorithm that will be the default in a future implementation
        of the engine.  ALGORITHM may be a string like "rsa", or
        "rsa2048" to explicitly request an algorithm and a key size.

        EXPIRES_IN specifies the expiration time of the key in number
        of seconds since the keys creation.  By default, a reasonable
        expiration time is chosen.  If you want to create a key that
        does not expire, use the keyword argument EXPIRES.

        SIGN, ENCRYPT, CERTIFY, and AUTHENTICATE can be used to
        request the capabilities of the new key.  If you don't request
        any, a reasonable set of capabilities is selected, and in case
        of OpenPGP, a subkey with a reasonable set of capabilities is
        created.

        If PASSPHRASE is None (the default), then the key will not be
        protected with a passphrase.  If PASSPHRASE is a string, it
        will be used to protect the key.  If PASSPHRASE is True, the
        passphrase must be supplied using a passphrase callback or
        out-of-band with a pinentry.

        Keyword arguments:
        algorithm    -- public key algorithm, see above (default: reasonable)
        expires_in   -- expiration time in seconds (default: reasonable)
        expires      -- whether or not the key should expire (default: True)
        sign         -- request the signing capability (see above)
        encrypt      -- request the encryption capability (see above)
        certify      -- request the certification capability (see above)
        authenticate -- request the authentication capability (see above)
        passphrase   -- protect the key with a passphrase (default: no
                        passphrase)
        force        -- force key creation even if a key with the same userid
                        exists (default: False)

        Returns:
                     -- an object describing the result of the key creation

        Raises:
        GPGMEError   -- as signaled by the underlying library

        r]Ncs�S)Nr)r^r_r`ra)rbrrrc5sz)Context.create_key.<locals>.passphrase_cbrr)N)r�is_a_stringrRrr	rgrhZop_createkey�create�SIGN�ENCRZCERT�AUTH�NOPASSWD�NOEXPIREZFORCE�op_genkey_result)rZuserid�	algorithm�
expires_in�expiresrprvZcertify�authenticaterb�forcersrtrcr)rbr�
create_key�s(;

b
zContext.create_keyc	s�tj��r<|j}	t|dd�}
tj|_d�fdd�	}|j|�zf|j||d||rVtjj	nd|rdtjj
ndB|rttjjndB�dkr�tjjndB|r�dntjj
B�Wdtj��r�|	|_|
r�|j|
dd��X|j�S)a@Create a subkey

        Create a subkey for the given KEY.  As subkeys are a concept
        of OpenPGP, calling this is only valid for the OpenPGP
        protocol.

        ALGORITHM may be used to specify the public key encryption
        algorithm for the new subkey.  By default, a reasonable
        default is chosen.  You may use "future-default" to select an
        algorithm that will be the default in a future implementation
        of the engine.  ALGORITHM may be a string like "rsa", or
        "rsa2048" to explicitly request an algorithm and a key size.

        EXPIRES_IN specifies the expiration time of the subkey in
        number of seconds since the subkeys creation.  By default, a
        reasonable expiration time is chosen.  If you want to create a
        subkey that does not expire, use the keyword argument EXPIRES.

        SIGN, ENCRYPT, and AUTHENTICATE can be used to request the
        capabilities of the new subkey.  If you don't request any, an
        encryption subkey is generated.

        If PASSPHRASE is None (the default), then the subkey will not
        be protected with a passphrase.  If PASSPHRASE is a string, it
        will be used to protect the subkey.  If PASSPHRASE is True,
        the passphrase must be supplied using a passphrase callback or
        out-of-band with a pinentry.

        Keyword arguments:
        algorithm    -- public key algorithm, see above (default: reasonable)
        expires_in   -- expiration time in seconds (default: reasonable)
        expires      -- whether or not the subkey should expire (default: True)
        sign         -- request the signing capability (see above)
        encrypt      -- request the encryption capability (see above)
        authenticate -- request the authentication capability (see above)
        passphrase   -- protect the subkey with a passphrase (default: no
                        passphrase)

        Returns:
                     -- an object describing the result of the subkey creation

        Raises:
        GPGMEError   -- as signaled by the underlying library

        r]Ncs�S)Nr)r^r_r`ra)rbrrrc�sz,Context.create_subkey.<locals>.passphrase_cbrr)N)rr�rRrr	rgrhZop_createsubkeyr�r�r�r�r�r�r�)rr6r�r�r�rprvr�rbrsrtrcr)rbr�
create_subkeyPs&6

B
zContext.create_subkeycCs|j||d�dS)z�Add a UID

        Add the uid UID to the given KEY.  Calling this function is
        only valid for the OpenPGP protocol.

        Raises:
        GPGMEError   -- as signaled by the underlying library

        rN)Z	op_adduid)rr6�uidrrr�key_add_uid�s
zContext.key_add_uidcCs|j||d�dS)z�Revoke a UID

        Revoke the uid UID from the given KEY.  Calling this function
        is only valid for the OpenPGP protocol.

        Raises:
        GPGMEError   -- as signaled by the underlying library

        rN)Z	op_revuid)rr6r�rrr�key_revoke_uid�s
zContext.key_revoke_uidcCsbd}|dks.tj|�rn|tjjO}dj|�}|s>|tjjO}|rN|tjjO}|j||||�dS)a�Sign a key

        Sign a key with the current set of signing keys.  Calling this
        function is only valid for the OpenPGP protocol.

        If UIDS is None (the default), then all UIDs are signed.  If
        it is a string, then only the matching UID is signed.  If it
        is a list of strings, then all matching UIDs are signed.  Note
        that a case-sensitive exact string comparison is done.

        EXPIRES_IN specifies the expiration time of the signature in
        seconds.  If EXPIRES_IN is False, the signature does not
        expire.

        Keyword arguments:
        uids         -- user ids to sign, see above (default: sign all)
        expires_in   -- validity period of the signature in seconds
                                               (default: do not expire)
        local        -- create a local, non-exportable signature
                                               (default: False)

        Raises:
        GPGMEError   -- as signaled by the underlying library

        rN�
)	rr�r	ZkeysignZLFSEPrr��LOCALZ
op_keysign)rr6Zuidsr�Zlocalr rrr�key_sign�s
zContext.key_signcCs|j||�dS)z�Set a keys' TOFU policy

        Set the TOFU policy associated with KEY to POLICY.  Calling
        this function is only valid for the OpenPGP protocol.

        Raises:
        GPGMEError   -- as signaled by the underlying library

        N)Zop_tofu_policy)rr6Zpolicyrrr�key_tofu_policy�s
zContext.key_tofu_policyc	Cs�tj|�st|t�r|}ndjdd�|D��}tj�}tj|j||rRt	j
|�|fnd|rft	j
|�|fnd|rzt	j
|�|fnd|�}|jr�tj|�t
|�tj|�}tj|�|dkr�t|�SdS)a�Issue a raw assuan command

        This function can be used to issue a raw assuan command to the
        engine.

        If command is a string or bytes, it will be used as-is.  If it
        is an iterable of strings, it will be properly escaped and
        joined into an well-formed assuan command.

        Keyword arguments:
        data_cb		-- a callback receiving data lines
        inquire_cb	-- a callback providing more information
        status_cb	-- a callback receiving status lines

        Returns:
        result		-- the result of command as GPGMEError

        Raises:
        GPGMEError	-- as signaled by the underlying library

        rcss|]}tj|�VqdS)N)rZpercent_escape)rrrrrr�sz*Context.assuan_transact.<locals>.<genexpr>Nr)rr�r|�bytesrr�new_gpgme_error_t_pZgpgme_op_assuan_transact_extr�weakref�refr
r9r�gpgme_error_t_p_value�delete_gpgme_error_t_pr)	rZcommandZdata_cbZ
inquire_cbZ	status_cb�cmdZerrptr�errrxrrr�assuan_transact�s


zContext.assuan_transactcCsr|dkrtd��|dkrt�}|r4tj|�||f}ntj|�|f}tj|j||||�}|jrftj|�t	|�dS)a�Interact with the engine

        This method can be used to edit keys and cards interactively.
        KEY is the key to edit, FUNC is called repeatedly with two
        unicode arguments, 'keyword' and 'args'.  See the GPGME manual
        for details.

        Keyword arguments:
        sink		-- if given, additional output is written here
        flags		-- use constants.INTERACT_CARD to edit a card

        Raises:
        GPGMEError	-- as signaled by the underlying library

        NzFirst argument cannot be None)
�
ValueErrorrfr�r�rZgpgme_op_interactrr
r9r)rr6r<rZr �	fnc_valueZ
opaquedatar;rrr�interact"s
zContext.interactcs�fdd�t�j��D�S)zKeys used for signingcsg|]}�j|��qSr)Zsigners_enum)r�i)rrrrFsz#Context.signers.<locals>.<listcomp>)�rangeZ
signers_count)rr)rrrQCszContext.signersc	CsD|j}|j�yx|D]}|j|�qWWn||_�YnXdS)N)rQZ
signers_clearZsigners_add)rrQ�oldr6rrrrQHs
cCs|j�S)z
Pinentry mode)Zget_pinentry_mode)rrrrrRSszContext.pinentry_modecCs|j|�dS)N)Zset_pinentry_mode)rr0rrrrRXscCs|j�S)zProtocol to use)Zget_protocol)rrrrrS\szContext.protocolcCsttj|��|j|�dS)N)rr�gpgme_engine_check_versionZset_protocol)rr0rrrrSascCs|jjS)zEngine's home directory)�engine_inforT)rrrrrTfszContext.home_dircCs|j|j|d�dS)N)rT)�set_engine_inforS)rr0rrrrTksZgpgme_ctx_tZgpgme_cCs|jd�r|jd�p|dkS)z?This function should list all functions returning gpgme_error_tZ	gpgme_op_Z_resultrM�gpgme_set_ctx_flag�gpgme_set_protocol�gpgme_set_sub_protocol�gpgme_set_keylist_mode�gpgme_set_pinentry_mode�gpgme_set_locale�gpgme_ctx_set_engine_info�gpgme_signers_add�gpgme_sig_notation_add�gpgme_set_sender�gpgme_cancel�gpgme_cancel_async�
gpgme_get_key�gpgme_get_sig_key>r�r�r�r�r�r�r�r�r�r�r�r�r�rMr�)�
startswith�endswith)rr*rrrr+rs
zContext._errorcheckrNrOrPcCsHtsdS|j�|j�|j�|jrD|jrDtjrDtj|j�d|_dS)N)r�_free_passcb�_free_progresscb�_free_statuscbrLrZ
gpgme_release)rrrr�__del__�szContext.__del__cCs|S)Nr)rrrr�	__enter__�szContext.__enter__cCs|j�dS)N)r�)r�typer0�tbrrr�__exit__�szContext.__exit__cos8|j||�|j�}x|r*|V|j�}qW|j�dS)N)r�r�r�)rr:�kwargsr6rrr�op_keylist_all�szContext.op_keylist_allcCs�tj�}y ttj|j|��tj|�}Wn:tjk
rb}zd}|j�tj	krR|�WYdd}~XnXtj
|�|r�dd�|_|SdS)z~Returns the next key in the list created
        by a call to op_keylist_start().  The object returned
        is of type Key.NcSs
tj|�S)N)r�gpgme_key_unref)rrrrr{�sz)Context.op_keylist_next.<locals>.<lambda>)r�new_gpgme_key_t_prZgpgme_op_keylist_nextr�gpgme_key_t_p_valuer
rrj�EOF�delete_gpgme_key_t_pr�)r�ptrr6�excprrrr��s

zContext.op_keylist_nextcCs�tj�}yttj|j|||��Wn@tjk
rb}z"|j�tjkrNtj	|��|�WYdd}~XnXtj
|�}tj|�|s�t�dd�|_
|S)a&Get a key given a fingerprint

        Keyword arguments:
        secret		-- to request a secret key

        Returns:
                        -- the matching key

        Raises:
        KeyError	-- if the key was not found
        GPGMEError	-- as signaled by the underlying library

        NcSs
tj|�S)N)rr�)rrrrr{�sz!Context.get_key.<locals>.<lambda>)rr�rr�rr
rrjr�ZKeyNotFoundr�r�rnr�)rr�r�r�rur6rrr�get_key�s



zContext.get_keycos8|j||�|j�}x|r*|V|j�}qW|j�dS)N)Zop_trustlist_start�op_trustlist_nextZop_trustlist_end)rr:r��trustrrr�op_trustlist_all�szContext.op_trustlist_allcCsptj�}y ttj|j|��tj|�}Wn8tjk
r`}zd}|j�tj	krP�WYdd}~XnXtj
|�|S)z�Returns the next trust item in the list created
        by a call to op_trustlist_start().  The object returned
        is of type TrustItem.N)rZnew_gpgme_trust_item_t_prZgpgme_op_trustlist_nextrZgpgme_trust_item_t_p_valuer
rrjr�Zdelete_gpgme_trust_item_t_p)rr�r�r�rrrr��s
zContext.op_trustlist_nextcCsF|dkrd}n(|dkr&tj|�|f}ntj|�||f}tj||�dS)a*Sets the passphrase callback to the function specified by func.

        When the system needs a passphrase, it will call func with three args:
        hint, a string describing the key it needs the passphrase for;
        desc, a string describing the passphrase it needs;
        prev_bad, a boolean equal True if this is a call made after
        unsuccessful previous attempt.

        If hook has a value other than None it will be passed into the func
        as a forth argument.

        Please see the GPGME manual for more information.
        N)r�r�r�gpg_set_passphrase_cb)rr<ra�hookdatarrrrh�szContext.set_passphrase_cbcCstjr|jd�dS)N)rr�rh)rrrrr��szContext._free_passcbcCsF|dkrd}n(|dkr&tj|�|f}ntj|�||f}tj||�dS)a�Sets the progress meter callback to the function specified by FUNC.
        If FUNC is None, the callback will be cleared.

        This function will be called to provide an interactive update
        of the system's progress.  The function will be called with
        three arguments, type, total, and current.  If HOOK is not
        None, it will be supplied as fourth argument.

        Please see the GPGME manual for more information.

        N)r�r�r�gpg_set_progress_cb)rr<rar�rrr�set_progress_cbszContext.set_progress_cbcCstjr|jd�dS)N)rr�r�)rrrrr�szContext._free_progresscbcCsF|dkrd}n(|dkr&tj|�|f}ntj|�||f}tj||�dS)aPSets the status callback to the function specified by FUNC.  If
        FUNC is None, the callback will be cleared.

        The function will be called with two arguments, keyword and
        args.  If HOOK is not None, it will be supplied as third
        argument.

        Please see the GPGME manual for more information.

        N)r�r�r�gpg_set_status_cb)rr<rar�rrr�
set_status_cbszContext.set_status_cbcCstjr|jd�dS)N)rr�r�)rrrrr�-szContext._free_statuscbcs4|j��fdd�|j�D�}t|�dks,t�|dS)z,Configuration of the engine currently in usecsg|]}|j�kr|�qSr)rS)rr�)r7rrr5sz'Context.engine_info.<locals>.<listcomp>rr)rS�get_engine_infor�rn)rZinfosr)r7rr�1szContext.engine_infocCstj|j�S)z�Get engine configuration

        Returns information about all configured and installed
        engines.

        Returns:
        infos		-- a list of engine infos

        )rZgpgme_ctx_get_engine_infor)rrrrr�9s
zContext.get_engine_infocCs|j|||�dS)a6Change engine configuration

        Changes the configuration of the crypto engine implementing
        the protocol 'proto' for the context.

        Keyword arguments:
        file_name	-- engine program file name (unchanged if None)
        home_dir	-- configuration directory (unchanged if None)

        N)Zctx_set_engine_info)r�proto�	file_namerTrrrr�EszContext.set_engine_infocCs8tj�}tj|j||�tj|�}tj|�t|�dS)z�Wait for asynchronous call to finish. Wait forever if hang is True.
        Raises an exception on errors.

        Please read the GPGME manual for more information.

        N)rr��
gpgme_waitrr�r�r)r�hangr�rxrrr�waitRs


zContext.waitcCs tjdtd�|j||||d�S)a�Start key editing using supplied callback function

        Note: This interface is deprecated and will be removed with
        GPGME 1.8.  Please use .interact instead.  Furthermore, we
        implement this using gpgme_op_interact, so callbacks will get
        called with string keywords instead of numeric status
        messages.  Code that is using constants.STATUS_X or
        constants.status.X will continue to work, whereas code using
        magic numbers will break as a result.

        z"Call to deprecated method op_edit.)rw)rZr�)r}r~rr�)rr6r<r��outrrr�op_edit_s
zContext.op_edit)NNT)N)N)N)	NrTFFFFNF)NrTFFFN)NFF)NNN)NrN)F)N)N)N)NN)<rrErFr>r	ZPINENTRY_MODE_DEFAULTZPROTOCOL_OpenPGPrr\rrvr�ZSIG_MODE_NORMALrpr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r4rQ�setterrRrSrTr(r)r+rr�r�r�r�r�r�r�r�rhr�r�r�r�r�r�r�r�r�rrJrr)rrrK�s�	Y
v+>/
"
#
*#
S
K
)
.
!








rKcs�eZdZdZdZdZdd�Zd'�fdd	�	Zd
d�Zdd
�Z	dd�Z
dd�Zdd�Zd(dd�Z
d)dd�Zd*dd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd,d%d&�Z�ZS)-rfaJData buffer

    A lot of data has to be exchanged between the user and the crypto
    engine, like plaintext messages, ciphertext, signatures and
    information about the keys.  The technical details about
    exchanging the data information are completely abstracted by
    GPGME.  The user provides and receives the data via `gpgme_data_t'
    objects, regardless of the communication protocol between GPGME
    and the crypto engine in use.

    This Data class is the implementation of the GpgmeData objects.

    Please see the information about __init__ for instantiation.

    Zgpgme_data_tZgpgme_data_c
Cs|d
kS)z?This function should list all functions returning gpgme_error_t�gpgme_data_read�gpgme_data_write�gpgme_data_seek�gpgme_data_release�gpgme_data_release_and_get_mem�gpgme_data_get_encoding�gpgme_data_get_file_name�gpgme_data_set_flag�gpgme_data_identify>	rrr	r
rrrrrr)rr*rrrr+�szData._errorcheckNTcs�tt|�jd�d|_|dk	r*|j|�nr|dk	r@|j||�n\|dk	rh|dk	rh|dk	rh|j|||�n4|dk	r�tj|�r�|j	||�q�|j
|�n|j�dS)a�Initialize a new gpgme_data_t object.

        If no args are specified, make it an empty object.

        If string alone is specified, initialize it with the data
        contained there.

        If file, offset, and length are all specified, file must
        be either a filename or a file-like object, and the object
        will be initialized by reading the specified chunk from the file.

        If cbs is specified, it MUST be a tuple of the form:

        (read_cb, write_cb, seek_cb, release_cb[, hook])

        where the first four items are functions implementing reading,
        writing, seeking the data, and releasing any resources once
        the data object is deallocated.  The functions must match the
        following prototypes:

            def read(amount, hook=None):
                return <a b"bytes" object>

            def write(data, hook=None):
                return <the number of bytes written>

            def seek(offset, whence, hook=None):
                return <the new file position>

            def release(hook=None):
                <return value and exceptions are ignored>

        The functions may be bound methods.  In that case, you can
        simply use the 'self' reference instead of using a hook.

        If file is specified without any other arguments, then
        it must be a filename, and the object will be initialized from
        that file.

        N)rrfrZdata_cbs�new_from_cbs�new_from_mem�new_from_filepartrr��
new_from_file�new_from_fd�new)r�stringr��offset�lengthZcbs�copy)rrrr�s/
z
Data.__init__cCsFtsdS|jdk	r:tjr:tj|j�|jr4tj|�d|_|j�dS)N)rrrr
r9�
_free_datacbs)rrrrr��s
zData.__del__cCs|S)Nr)rrrrr��szData.__enter__cCs|j�dS)N)r�)rr�r0r�rrrr��sz
Data.__exit__cCs
d|_dS)N)Z	_data_cbs)rrrrr�szData._free_datacbscCs0tj�}ttj|��tj|�|_tj|�dS)N)r�new_gpgme_data_t_prZgpgme_data_new�gpgme_data_t_p_valuer�delete_gpgme_data_t_p)rrUrrrr�szData.newcCs:tj�}ttj||t|�|��tj|�|_tj|�dS)N)rrrZgpgme_data_new_from_memr�rrr)rrrrUrrrr
�s
zData.new_from_memcCs�tj�}yttj|||��WnFtjk
rd}z(|j�tjkrP|rPtd��n|�WYdd}~XnXtj	|�|_
tj|�dS)Nz#delayed reads are not yet supported)rrrZgpgme_data_new_from_filer
rrjZ	INV_VALUEr�rrr)r�filenamerrUrurrrr�s
zData.new_from_filecCsdtj�}|dk	r(tj|�|||||f}ntj|�||||f}tj|||�tj|�|_tj|�dS)N)rrr�r�Zgpg_data_new_from_cbsrrr)rZread_cbZwrite_cbZseek_cbZ
release_cbrarUr�rrrrs
zData.new_from_cbscCs�tj�}d}d}tj|�r |}n6tj|j�|j�}|dkrVtdtt	|��t|�f��t
tj|||||��tj|�|_
tj|�dS)z�This wraps the GPGME gpgme_data_new_from_filepart() function.
        The argument "file" may be:

        * a string specifying a file name, or
        * a file-like object supporting the fileno() and the mode attribute.

        Nz"Failed to open file from %s arg %s)rrrr��fdopen�filenor�r��strr�rZgpgme_data_new_from_filepartrrr)rr�rrrUr�fprrrrs	
zData.new_from_filepartcCs6tj�}ttj||j���tj|�|_tj|�dS)z�This wraps the GPGME gpgme_data_new_from_fd() function.  The
        argument "file" must be a file-like object, supporting the
        fileno() method.

        N)rrrZgpgme_data_new_from_fdrrrr)rr�rUrrrr-szData.new_from_fdcCs|j|�dS)z�This wrap around gpgme_data_new_from_stream is an alias for
        new_from_fd() method since in python there's no difference
        between file stream and file descriptor.N)r)rr�rrr�new_from_stream8szData.new_from_streamcCs|j|�dS)z�This wrap around gpgme_data_new_from_estream is an alias for
        new_from_fd() method since in python there's no difference
        between file stream and file descriptor, but using fd broke.N)r)rr�rrr�new_from_estream>szData.new_from_estreamcCs4tj|j|�}|dkr0|jr(tj|�ntj��|S)zkWrite buffer given as string or bytes.

        If a string is given, it is implicitly encoded using UTF-8.r)rrrr
r9rZfromSyserror)r�bufferZwrittenrrr�writeDsz
Data.writerc	Cs�|dkrdS|dkrLytj|j|�}Wn |jr@tj|�n�YnX|Sg}xPytj|jd�}Wn |jr~tj|�n�YnXt|�dkr�P|j|�qRWdj|�SdS)z�Read at most size bytes, returned as bytes.

        If the size argument is negative or omitted, read until EOF is reached.

        Returns the data read, or the empty string if there was no data
        to read before EOF was reached.r�i�N)rrrr
r9r�rr)r�sizer;ZchunksrrrrYPs,z	Data.read)NNNNNT)T)T)N���)r&)rrErFr>r(r)r+rr�r�r�rrr
rrrrrr r"rYrJrr)rrrfps0:



rfcCs
tj|�S)z�Return short algorithm string

    Return a public key algorithm string (e.g. "rsa2048") for a given
    SUBKEY.

    Returns:
    algo      - a string

    )rZgpgme_pubkey_algo_string)r�rrr�pubkey_algo_stringts
r'cCs
tj|�S)z�Return name of public key algorithm

    Return the name of the public key algorithm for a given numeric
    algorithm id ALGO (cf. RFC4880).

    Returns:
    algo      - a string

    )rZgpgme_pubkey_algo_name)�algorrr�pubkey_algo_name�s
r)cCs
tj|�S)z�Return name of hash algorithm

    Return the name of the hash algorithm for a given numeric
    algorithm id ALGO (cf. RFC4880).

    Returns:
    algo      - a string

    )rZgpgme_hash_algo_name)r(rrr�hash_algo_name�s
r*cCs
tj|�S)ztGet protocol description

    Get the string describing protocol PROTO.

    Returns:
    proto     - a string

    )rZgpgme_get_protocol_name)r�rrr�get_protocol_name�s	r+cCs
tj|�S)z�Return the address spec

    Return the addr-spec (cf. RFC2822 section 4.3) from a user id UID.

    Returns:
    addr_spec - a string

    )rZgpgme_addrspec_from_uid)r�rrr�addrspec_from_uid�s	r,cCs
tj|�S)N)rZgpgme_check_version)�versionrrr�
check_version�sr.cCs.yttj|��dStjk
r(dSXdS)NTF)rrr�r
r)r�rrr�engine_check_version�s
r/cCsNtj�}yttj|��tj|�}Wntjk
r>d}YnXtj|�|S)N)rZnew_gpgme_engine_info_t_prZgpgme_get_engine_infoZgpgme_engine_info_t_p_valuer
rZdelete_gpgme_engine_info_t_p)r��inforrrr��s

r�cCsttj|||��dS)a#Changes the default configuration of the crypto engine implementing
    the protocol 'proto'. 'file_name' is the file name of
    the executable program implementing this protocol. 'home_dir' is the
    directory name of the configuration directory (engine's default is
    used if omitted).N)rrZgpgme_set_engine_info)r�r�rTrrrr��sr�cCsttjd||��dS)z(Sets the default locale used by contextsN)rrr�)rwr0rrr�
set_locale�sr1cCsLtj�}tjd||�}tj|�}tj|�|dkr<t|�nt|�}||fS)aFWait for asynchronous call on any Context  to finish.
    Wait forever if hang is True.

    For finished anynch calls it returns a tuple (status, context):
        status  - status return by asnynchronous call.
        context - context which caused this call to return.

    Please read the GPGME manual of more information.N)rr�r�r�r�rrK)r�r��contextrxrrrr��s	


r�)N)N)Z
__future__rrrrHrWr}r�r#rr
rrr	r�objectrrKrfr'r)r*r+r,r.r/r�r�r1r�rrrr�<module>sJM