a
    ?ig)                     @   s,  d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddl
mZ ddlmZmZmZ G dd deZG dd	 d	eZG d
d deZdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Ze d kr(d!d"  d#d"  e
!  e D ]Z"e#e" qdS )$z4Handle GnuPG keys used to trust signed repositories.    )print_functionN)gettext)ListOptionalTuplec                   @   s   e Zd ZdS )AptKeyErrorN)__name__
__module____qualname__ r   r   */usr/lib/python3/dist-packages/apt/auth.pyr   +   s   r   c                   @   s   e Zd ZdZdS )AptKeyIDTooShortErrorz!Internal class do not rely on it.N)r   r	   r
   __doc__r   r   r   r   r   /   s   r   c                   @   s    e Zd ZdZdd Zdd ZdS )
TrustedKeyzRepresents a trusted key.c                 C   s    || _ t|| _|| _|| _d S )N)Zraw_name_namekeyiddate)selfr   r   r   r   r   r   __init__7   s    
zTrustedKey.__init__c                 C   s   d| j | j| jf S )Nz%s
%s %s)r   r   r   )r   r   r   r   __str__?   s    zTrustedKey.__str__N)r   r	   r
   r   r   r   r   r   r   r   r   3   s   r   c            	      O   s   d}t jddg}||  tj }d|d< d|d< zt jdd	krtj	d
dd}|
t j d |  |j|d< tj||dtjtjtjd}|dd}||\}}|jrtd|jd|||f n|rtj
| | W |dur|  S n|dur|  0 dS )z0Run the apt-key script with the given arguments.NzDir::Bin::Apt-Keyz/usr/bin/apt-keyCZLANG1Z$APT_KEY_DONT_WARN_ON_DANGEROUS_USAGEZDir/zapt-keyz.conf)prefixsuffixzUTF-8Z
APT_CONFIGT)envuniversal_newlinesstdinstdoutstderrr   zGThe apt-key script failed with return code %s:
%s
stdout: %s
stderr: %s )apt_pkgZconfigZ	find_fileextendosenvironcopyZfind_dirtempfileZNamedTemporaryFilewritedumpencodeflushr   
subprocessPopenPIPEgetcommunicate
returncoder   joinsysr    stripclose)	argskwargsZconfcmdr   procr   outputr    r   r   r   _call_apt_key_scriptD   sL    




 
r;   c                 C   s@   t j| std|  t | t js2td|  td|  dS )zImport a GnuPG key file to trust repositores signed by it.

    Keyword arguments:
    filename -- the absolute path to the public GnuPG key file
    z An absolute path is required: %szKey file cannot be accessed: %saddN)r$   pathabspathr   accessR_OKr;   )filenamer   r   r   add_key_from_fileq   s
    rB   c              
   C   sf   t  }z@zt| || W n ty.    Y n0 W dd }tj||d ndd }tj||d 0 dS )zImport a GnuPG key file to trust repositores signed by it.

    Keyword arguments:
    keyid -- the long keyid (fingerprint) of the key, e.g.
             A1BD8E9D78F7FE5C3E65D8AF8B48AD6246925553
    keyserver -- the URL or hostname of the key server
    c                 S   s(   t |d tr"|d jtjkr"d S  d S )N   )
isinstanceOSErrorerrnoZENOENT)funcr=   exc_infor   r   r   onerror   s
    z'add_key_from_keyserver.<locals>.onerror)rI   N)r'   Zmkdtemp_add_key_from_keyserver	ExceptionshutilZrmtree)r   	keyservertmp_keyring_dirrI   r   r   r   add_key_from_keyserver   s    	
rO   c                 C   sN  t | dddddk r$tdtj|d}tj|d}dd	d
d|g}t|d|d|d|d| g }|dkrtd|| f tj|d}t|d|d|d| g }|dkrtd| tj	|d|ddddg tj
dd d }d }	| D ]"}
|
dr|
dd }	 qq| dd }|	|krBtd||f t| d S )Nr!    Z0xg      D@z,Only fingerprints (v4, 160bit) are supportedzsecring.gpgzpubring.gpgZgpgz--no-default-keyringz--no-optionsz	--homedirz--secret-keyringz	--keyringz--keyserverz--recvr   zrecv from '%s' failed for '%s'zexport-keyring.gpgz--outputz--exportzexport of '%s' failedz--fingerprint--batch--fixed-list-mode--with-colonsT)r   r   zfpr::	   )lenreplacer   r$   r=   r2   r,   Zcallr   r-   r.   r0   
splitlines
startswithsplitupperrB   )r   rM   rN   Ztmp_secret_keyringZtmp_keyringZgpg_default_optionsresZtmp_export_keyringr:   Zgot_fingerprintlineZsigning_key_fingerprintr   r   r   rJ      sl    

	


rJ   c                 C   s   t ddddd| d dS )zImport a GnuPG key to trust repositores signed by it.

    Keyword arguments:
    content -- the content of the GnuPG public key
    advz--quietrQ   z--import-)r   Nr;   )Zcontentr   r   r   add_key   s    ra   c                 C   s   t d|  dS )zRemove a GnuPG key to no longer trust repositores signed by it.

    Keyword arguments:
    fingerprint -- the fingerprint identifying the key
    ZrmNr`   Zfingerprintr   r   r   
remove_key   s    rc   c                 C   s
   t d| S )zxReturn the GnuPG key in text format.

    Keyword arguments:
    fingerprint -- the fingerprint identifying the key
    Zexportr`   rb   r   r   r   
export_key   s    rd   c                   C   s   t dS )a  Update the local keyring with the archive keyring and remove from
    the local keyring the archive keys which are no longer valid. The
    archive keyring is shipped in the archive-keyring package of your
    distribution, e.g. the debian-archive-keyring package in Debian.
    updater`   r   r   r   r   re     s    re   c                   C   s   t dS )ay  Work similar to the update command above, but get the archive
    keyring from an URI instead and validate it against a master key.
    This requires an installed wget(1) and an APT build configured to
    have a server to fetch from and a master keyring to validate. APT
    in Debian does not support this command and relies on update
    instead, but Ubuntu's APT does.
    z
net-updater`   r   r   r   r   
net_update  s    	rf   c                  C   sx   t ddddd} g }| dD ]T}|d}|d d	kr@|d
 }|d dkr|d }|d }t|||}|| q|S )zaReturns a list of TrustedKey instances for each key which is
    used to trust repositories.
    r^   rS   rQ   rR   z--list-keys
rT   r   Zpub   uidrU      )r;   rZ   r   append)r:   r\   r]   fieldsr   ri   Zcreation_datekeyr   r   r   	list_keys  s    
rn   __main__c                   C   s   t dS )Nz;Ubuntu Archive Automatic Signing Key <ftpmaster@ubuntu.com>r   r   r   r   r   <lambda>0      rq   c                   C   s   t dS )Nz:Ubuntu CD Image Automatic Signing Key <cdimage@ubuntu.com>rp   r   r   r   r   rq   1  rr   )$r   Z
__future__r   rF   r$   os.pathrL   r,   r3   r'   r"   r   r   typingr   r   r   rK   r   r   objectr   r;   rB   rO   rJ   ra   rc   rd   re   rf   rn   r   ZinitZtrusted_keyprintr   r   r   r   <module>   s<   -H




