File: //lib/python3.6/site-packages/awscli/customizations/ec2/__pycache__/decryptpassword.cpython-36.pyc
3
�T_� � @ sn d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z e j
e �Z dZ
dd� ZG dd � d e �ZdS )
� N)�six)�model)�BaseCLIArgumentz�<p>The file that contains the private key used to launch
the instance (e.g. windows-keypair.pem). If this is supplied, the
password data sent from EC2 will be decrypted before display.</p>c K s t ||d�| d< dS )z�
This handler gets called after the argument table for the
operation has been created. It's job is to add the
``priv-launch-key`` parameter.
zpriv-launch-keyN)�LaunchKeyArgument)Zargument_table�operation_model�session�kwargs� r �%/usr/lib/python3.6/decryptpassword.py�ec2_add_priv_launch_key s r c @ s^ e Zd Zdd� Zedd� �Zedd� �Zejdd� �Zedd � �Zd
d � Z d d
� Z
dd� Z dS )r c C s4 || _ tjdddi�| _|| _|| _d | _d| _d S )Nr �type�stringF)�_sessionr ZShapeZargument_model�_operation_model�_name� _key_path� _required)�selfr r �namer r r
�__init__- s zLaunchKeyArgument.__init__c C s dS )Nr
r )r r r r
�
cli_type_name5 s z LaunchKeyArgument.cli_type_namec C s | j S )N)r )r r r r
�required9 s zLaunchKeyArgument.requiredc C s
|| _ d S )N)r )r �valuer r r
r = s c C s t S )N)�HELP)r r r r
�
documentationA s z LaunchKeyArgument.documentationc C s |j | j| jdd� d S )NzSSH Private Key file)�dest�help)� add_argumentZcli_nameZpy_name)r �parserr r r
�
add_to_parserE s z LaunchKeyArgument.add_to_parserc C sj |rft jj|�}t jj|�}t jj|�rZ|| _| jjj}d|| jj f }| j
j || j � n d}t
|��dS )a
This gets called with the value of our ``--priv-launch-key``
if it is specified. It needs to determine if the path
provided is valid and, if it is, it stores it in the instance
variable ``_key_path`` for use by the decrypt routine.
zafter-call.%s.%sz_priv-launch-key should be a path to the local SSH private key file used to launch the instance.N)�os�path�
expandvars�
expanduser�isfiler r Z
service_model�endpoint_prefixr r �register�_decrypt_password_data�
ValueError)r Z
parametersr r! r% Zevent�msgr r r
�
add_to_paramsI s
z LaunchKeyArgument.add_to_paramsc K s� | j dk r�tjd| j � |jd�}|s*dS ylt| j ��X}|j� }tjjt j
|��}t j |�}tj
||�}tj|� |jd�|d< tj|� W dQ R X W n. tk
r� tjddd� d}t|��Y nX dS ) aq
This handler gets called after the GetPasswordData command has been
executed. It is called with the and the ``parsed`` data. It checks to
see if a private launch key was specified on the command. If it was,
it tries to use that private key to decrypt the password data and
replace it in the returned data dictionary.
Nz"Decrypting password data using: %sZ PasswordDatazutf-8z Unable to decrypt PasswordDataT)�exc_infoz@Unable to decrypt password data using provided private key file.)r �logger�debug�get�open�read�rsaZ
PrivateKeyZ
load_pkcs1r �b�base64Z b64decodeZdecrypt�decode� Exceptionr( )r Zparsedr r Zpk_fileZ pk_contentsZ private_keyr) r r r
r' ` s$
z(LaunchKeyArgument._decrypt_password_dataN) �__name__�
__module__� __qualname__r �propertyr r �setterr r r* r' r r r r
r + s r )Zloggingr r3 r1 Z
awscli.compatr Zbotocorer Zawscli.argumentsr Z getLoggerr6 r, r r r r r r r
�<module>
s