File: //usr/lib/python3.9/site-packages/ipaclient/plugins/__pycache__/otptoken_yubikey.cpython-39.pyc
a
}�f� � @ s d dl Z d dlZd dlmZmZmZ d dlmZmZ d dl m
Z
m Z m Z d dl
mZ d dlmZ zd dlZd dlZW n ey� eed�d��Y n0 ejr�eZed �ed
� ed � ed � ed
� Ze� ZdZedd�G dd� de ��Zedd�G dd� de ��Z e� G dd� de
��Z dS )� N)�_�api�IntEnum)�NotFound�SkipPluginModule)�Command�Method�Object)�Registry)�
classpropertyz python-yubico is not installed.��reasonz
YubiKey Tokens
z
Manage YubiKey tokens.
zt
This code is an extension to the otptoken plugin and provides support for
reading/writing YubiKey tokens directly.
z
EXAMPLES:
zP
Add a new token:
ipa otptoken-add-yubikey --owner=jdoe --desc="My YubiKey"
ZotpT)Zno_failc @ s e Zd ZdZdS )�_fake_otptokenZotptokenN)�__name__�
__module__� __qualname__�name� r r �F/usr/lib/python3.9/site-packages/ipaclient/plugins/otptoken_yubikey.pyr ; s r c @ s e Zd ZdZdZdS )�_fake_otptoken_add� otptoken_addTN)r r r r �NO_CLIr r r r r @ s r c s� e Zd Zed�Zedded�dd�fZeZedd� �Z e
e �Z e d d
� �Z
� fd d �Z� fd
d�Z� fdd�Zdd� Zdd� Z� ZS )�otptoken_add_yubikeyz Add a new YubiKey OTP token.zslot?�slotz YubiKey slot)� � )Zcli_nameZlabel�valuesc C s t j�d�tu S )Nr )r r Z
get_pluginr )�clsr r r Z__NO_CLI_getterS s z$otptoken_add_yubikey.__NO_CLI_getterc C s | j jjjS �N)r r r � api_version��selfr r r r Y s z otptoken_add_yubikey.api_versionc # s8 | j jj�� D ]
}|V qtt| ��� D ]
}|V q(d S r )r r r �args�superr �get_args)r! �arg�� __class__r r r$ ] s z otptoken_add_yubikey.get_argsc # sB | j jj�� D ]}|jdvr|V qtt| ��� D ]
}|V q2d S )N) �type�ipatokenvendor�
ipatokenmodel�ipatokenserial�ipatokenotpalgorithm�ipatokenhotpcounter�ipatokenotpkeyZipatokentotpclockoffsetZipatokentotptimestep� no_qrcodeZqrcode�version)r r r �optionsr r# r � get_options)r! �optionr& r r r2 c s
z otptoken_add_yubikey.get_optionsc # s8 | j jj�� D ]
}|V qtt| ��� D ]
}|V q(d S r )r r r Z
output_paramsr# r �get_output_params)r! Zparamr& r r r4 u s z&otptoken_add_yubikey.get_output_paramsc C s | j jj�� S r )r r r �outputr r r r � _iter_output{ s z!otptoken_add_yubikey._iter_outputc O s� z t �� }W n� tjjyD } z td|j d��W Y d }~npd }~0 t jjyv } zt|j d��W Y d }~n>d }~0 t
y� } z tt |�d d��W Y d }~n
d }~0 0 |� � dks�J �|�
dd �d u �r z*|�� �� }tddh�|��d |d< W n" t�y ttd �d��Y n0 t�d
�}|�� }|�||�
d | �d ��� |�d d
� |j||d d� dd� |�� D �}| j j j dg|�R ddt |j �t |�!� �dd|d
d�|��} dD ]"}
|
| �
di �v �r�| d |
= �q�|d | �
di �d< | S )NzNo YubiKey found: %sr z:. Please install 'libyubikey' and 'libusb' packages first.)r r r r r r zNo free YubiKey slot!� �ipatokenotpdigitsZSERIAL_API_VISIBLET)r c S s i | ]\}}|d v r||�qS ))r0 � descriptionZ
ipatokenownerZipatokendisabledZipatokennotbeforeZipatokennotafterr8 r )�.0�k�vr r r �
<dictcomp>� � z0otptoken_add_yubikey.forward.<locals>.<dictcomp>r ZhotpZYubiCoZsha1)r( r) r* r+ r, r- r. r/ )Zurir. �result)"�yubicoZ find_yubikey�usbZcoreZUSBErrorr �strerrorZyubikeyZ YubiKeyErrorr
�
ValueError�strZ version_num�get�statusZ
valid_configs�sorted�
difference�
IndexErrorr �os�urandomZ init_configZmode_oath_hotpZget_default_ofZ
extended_flagZ write_config�itemsZBackendZ rpcclient�forward�unicodeZmodel�serial) r! r" �kwargsZyk�e�used�keyZcfgr1 Zanswerr; r r r rM ~ sV $ (
� �
�
�
z otptoken_add_yubikey.forward)r r r r �__doc__r Z
takes_optionsZhas_output_params� classmethodZ$_otptoken_add_yubikey__NO_CLI_getterr r �propertyr r$ r2 r4 r6 rM �
__classcell__r r r& r r F s$ ��
r ) rJ ZsixZipalibr r r Z
ipalib.errorsr r Zipalib.frontendr r r Zipalib.plugabler
Z ipalib.utilr Zusb.corerA r@ � ImportErrorZPY3rD rN rT �registerZtopicr r r r r r r �<module> s< ��� �