File: //usr/lib/python3.6/site-packages/ipaclient/plugins/__pycache__/otptoken_yubikey.cpython-36.pyc
3
�d[e
�
@ s" d dl Z d dlZd dlmZmZmZ d dlmZmZ d dl m
Z
m Z m Z d dl
mZ d dlmZ yd dlZd dlZW n" ek
r� eed�d��Y nX 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 �&/usr/lib/python3.6/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� fdd�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 jjd�tkS )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 # s@ x | j jjj� D ]
}|V qW x tt| �j� D ]
}|V q.W d S )N)r r r �args�superr �get_args)r �arg)� __class__r r r! ] s
z otptoken_add_yubikey.get_argsc # sJ x&| j jjj� D ]}|jd
kr|V qW x tt| �j� D ]
}|V q8W d S )N�type�ipatokenvendor�
ipatokenmodel�ipatokenserial�ipatokenotpalgorithm�ipatokenhotpcounter�ipatokenotpkey�ipatokentotpclockoffset�ipatokentotptimestep� no_qrcode�qrcode�version) r$ r% r& r' r( r) r* r+ r, r- r. r/ )r r r �optionsr r r � get_options)r Zoption)r# r r r1 c s
z otptoken_add_yubikey.get_optionsc # s@ x | j jjj� D ]
}|V qW x tt| �j� D ]
}|V q.W d S )N)r r r Z
output_paramsr r �get_output_params)r Zparam)r# r r r2 u s
z&otptoken_add_yubikey.get_output_paramsc C s | j jjj� S )N)r r r �output)r r r r � _iter_output{ s z!otptoken_add_yubikey._iter_outputc ! O s� y t j� }W n� tjjk
rB } ztd|j d��W Y d d }~X nd t jjk
rr } zt|j d��W Y d d }~X n4 t
k
r� } ztt |�d d��W Y d d }~X nX |j � dks�t
�|jdd �d k�ry*|j� j� }tddhj|��d |d< W n$ tk
�r ttd�d��Y nX tjd �}|j� }|j||jd
| jd
��� |jd d � |j||d d
� dd� |j � D �}| j j j d|�ddt |j!�t |j"� �dd|d d�|��} x*dD ]"}
|
| jdi �k�r�| d |
= �q�W |d | jdi �d< | S )NzNo YubiKey found: %s)r z:. Please install 'libyubikey' and 'libusb' packages first.r r r r zNo free YubiKey slot!� �ipatokenotpdigitsZSERIAL_API_VISIBLET)r c
S s i | ]\}}|dkr||�qS )r/ � description�
ipatokenowner�ipatokendisabled�ipatokennotbefore�ipatokennotafterr6 )r/ r7 r8 r9 r: r; r6 r )�.0�k�vr r r �
<dictcomp>� s z0otptoken_add_yubikey.forward.<locals>.<dictcomp>r ZhotpZYubiCoZsha1)r$ r% r& r' r( r) r* r- �urir* �result)r r )r )r@ r* )#�yubicoZ find_yubikey�usbZcoreZUSBErrorr �strerrorZyubikeyZ YubiKeyErrorr �
ValueError�strZ version_num�AssertionError�getZstatusZ
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�eZused�keyZcfgr0 Zanswerr= r r r rO ~ sN $
z otptoken_add_yubikey.forward)r r )r r r r �__doc__r Z
takes_optionsZhas_output_params� classmethodZ$_otptoken_add_yubikey__NO_CLI_getterr r �propertyr r! r1 r2 r4 rO �
__classcell__r r )r# r r F s
r ) rL ZsixZipalibr r r Z
ipalib.errorsr r Zipalib.frontendr r r Zipalib.plugabler
Z ipalib.utilr Zusb.corerC rB � ImportErrorZPY3rF rP rU �registerZtopicr
r r r r r r �<module> s.