File: //lib/python3/dist-packages/awscli/customizations/ec2/__pycache__/decryptpassword.cpython-310.pyc
o
.&�a� � @ sn d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z e �
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)�argument_table�operation_model�session�kwargs� r
�K/usr/lib/python3/dist-packages/awscli/customizations/ec2/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 || _ t�dddi�| _|| _|| _d | _d| _d S )Nr �type�stringF)�_sessionr �Shape�argument_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 � z LaunchKeyArgument.cli_type_namec C s | j S �N�r r r
r
r �required9 s zLaunchKeyArgument.requiredc C s
|| _ d S r r )r �valuer
r
r r = s
c C s t S r )�HELPr r
r
r �
documentationA r z LaunchKeyArgument.documentationc C s |j | j| jdd� d S )NzSSH Private Key file)�dest�help)� add_argument�cli_name�py_name)r �parserr
r
r �
add_to_parserE s
�z LaunchKeyArgument.add_to_parserc C sl |r4t j�|�}t j�|�}t j�|�r.|| _| jjj}d|| jj f }| j
� || j � dS 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 �
service_model�endpoint_prefixr r �register�_decrypt_password_data�
ValueError)r �
parametersr r* r/ �event�msgr
r
r �
add_to_paramsI s ���z LaunchKeyArgument.add_to_paramsc K s� | j durmt�d| j � |�d�}|sdS zCt| j ��3}|�� }tj�t �
|��}t � |�}t�
||�}t�|� |�d�|d< t�|� W d � W dS 1 sQw Y W dS tyl tjddd� d}t|��w 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: %s� PasswordDatazutf-8z Unable to decrypt PasswordDataT)�exc_infoz@Unable to decrypt password data using provided private key file.)r �logger�debug�get�open�read�rsa�
PrivateKey�
load_pkcs1r �b�base64� b64decode�decrypt�decode� Exceptionr2 )r �parsedr r �pk_file� pk_contents� private_keyr5 r
r
r r1 ` s*
&� ��z(LaunchKeyArgument._decrypt_password_dataN) �__name__�
__module__� __qualname__r �propertyr r �setterr! r( r6 r1 r
r
r
r r + s
r )�loggingr) rB r> �
awscli.compatr �botocorer �awscli.argumentsr � getLoggerrK r9 r r r r
r
r
r �<module> s