File: //proc/self/root/usr/lib/python3.6/site-packages/yubico/__pycache__/yubikey_usb_hid.cpython-36.pyc
3
y@�V�X � @ sN d Z dddgZddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lm Z dd lm
Z
dd l m Z m Z m
Z
dd
l
mZ ddlZddlZddlZddlZd!ZdZdZdZdZdZdZdZdZe j e j d�e j e j d�d�Z G dd� dej!�Z"G dd� de
j#�Z$G dd � d e%�Z&G d d� de�Z'G d d� de%�Z(G d d � d ej)�Z*dS )"z(
module for accessing a USB HID YubiKey
�
YubiKeyUSBHID�YubiKeyUSBHIDError�YubiKeyUSBHIDStatus� )� __version__)� yubico_util)�yubico_exception)�
yubikey_frame)�yubikey_config)� yubikey_defs)� yubikey_base)�SLOT�
YUBICO_VID�PID)�YubiKey� N� � � i� � � )r � )�HMAC�OTPc @ s e Zd ZdZdS )r z= Exception raised for errors with the USB HID communication. N)�__name__�
__module__� __qualname__�__doc__� r r �%/usr/lib/python3.6/yubikey_usb_hid.pyr 4 s c sp e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Z� ZS )�YubiKeyUSBHIDCapabilitiesz�
Capture the capabilities of the various versions of YubiKeys.
Overrides just the functions from YubiKeyCapabilities() that are available
in one or more versions, leaving the other ones at False through default_answer.
c s t t| �j|||d� d S )N)�model�version�default_answer)�superr �__init__)�selfr r! r" )� __class__r r r$ ? s
z"YubiKeyUSBHIDCapabilities.__init__c C s dS )zG Yubico OTP support has always been available in the standard YubiKey. Tr )r% r r r �have_yubico_OTPE s z)YubiKeyUSBHIDCapabilities.have_yubico_OTPc C s |dkr dS | j dkS )z* OATH HOTP was introduced in YubiKey 2.2. �HOTPFr r r )r( )r r r )r! )r% �moder r r � have_OATHI s z#YubiKeyUSBHIDCapabilities.have_OATHc C s |dkr dS | j dkS )z3 Challenge-response was introduced in YubiKey 2.2. r r Fr r )r r )r r r )r! )r% r) r r r �have_challenge_responseO s z1YubiKeyUSBHIDCapabilities.have_challenge_responsec C s
| j dkS )zW Reading serial number was introduced in YubiKey 2.2, but depends on extflags set too. r r )r r r )r! )r% r r r �have_serial_numberU s z,YubiKeyUSBHIDCapabilities.have_serial_numberc C s |j | j| jd�S )N)r r! )�
is_compatibler r! )r% �flagr r r �have_ticket_flagY s z*YubiKeyUSBHIDCapabilities.have_ticket_flagc C s |j | j| jd�S )N)r r! )r- r r! )r% r. r r r �have_config_flag\ s z*YubiKeyUSBHIDCapabilities.have_config_flagc C s |j | j| jd�S )N)r r! )r- r r! )r% r. r r r �have_extended_flag_ s z,YubiKeyUSBHIDCapabilities.have_extended_flagc C s
| j dkS )Nr r )r r r )r! )r% r r r � have_extended_scan_code_modeb s z6YubiKeyUSBHIDCapabilities.have_extended_scan_code_modec C s
| j dkS )Nr r )r r r )r! )r% r r r �have_shifted_1_modee s z-YubiKeyUSBHIDCapabilities.have_shifted_1_modec C s |dkS )Nr r )r r r )r% �slotr r r �have_configuration_sloth s z1YubiKeyUSBHIDCapabilities.have_configuration_slot)r r r r r$ r' r* r+ r, r/ r0 r1 r2 r3 r5 �
__classcell__r r )r& r r 8 s r c @ s� e Zd ZdZd'dd�Zdd� Zdd � Zd
d � Zd(d d
�Zdd� Z dd� Z
dd� Z d)dd�Z d*dd�Z
d+dd�Zd,d d �Zd-d d �Zd d!� Zd.d"d#�Zd/d%d&�ZdS )0�YubiKeyHIDDevicezP
High-level wrapper for low-level HID commands for a HID based YubiKey.
Fr c C s* || _ d| _| j|�s td��| j� dS )z�
Find and connect to a YubiKey (USB HID).
Attributes :
skip -- number of YubiKeys to skip
debug -- True or False
Nz%YubiKey USB HID initialization failed)�debug� _usb_handle�_openr �status)r% r8 �skipr r r r$ q s
zYubiKeyHIDDevice.__init__c C s | j � }t|�| _| jS )z*
Poll YubiKey for status.
)�_readr �_status)r% �datar r r r; s
zYubiKeyHIDDevice.statusc C s, y| j r| j� W n tk
r& Y nX d S )N)r9 �_close�IOError)r% r r r �__del__� s
zYubiKeyHIDDevice.__del__c C s� | j j}|j|d�}| jdtj|j�|f � | j|� | jt j
� | j � | jd||| j jf � | j j � }| r�| j jdkr�dS | j j|d kr�dS t
d|| j jf ��dS )z! Write configuration to YubiKey. )r4 zWriting %s frame :
%s
z&Programmed slot %i, sequence %i -> %i
r Nr z6YubiKey programming failed (seq %i not increased (%i)))r> �pgm_seqZto_frame�_debugr Z command2str�command�_write�_waitfor_clearr
�SLOT_WRITE_FLAGr; �
valid_configsr )r% �cfgr4 Z old_pgm_seq�frameZcfgsr r r �
_write_config� s
z YubiKeyHIDDevice._write_configc C sr | j tj|�dd� }xN| j� }tj|d �}|tj@ r`|d@ }|rN|dkrNP ||dd� 7 }qP qW | j� |S )z7 Wait for a response to become available, and read it. N� � r )� _waitfor_setr
ZRESP_PENDING_FLAGr= r �ord_byte� _write_reset)r% � may_block�res�this�flags�seqr r r �_read_response� s
z YubiKeyHIDDevice._read_responsec C s� t tB tB }td> }| jj|tt|td�}t |�tkrP| j
dt|f � t d��dj dd� |D ��}| j
dt
j|d d
� � |S ) z1 Read a USB HID feature report from the YubiKey. r )�value�timeoutz7Failed reading %i bytes (got %i) from USB HID YubiKey.
z#Failed reading from USB HID YubiKey� c s s | ]}t j|�V qd S )N)r Zchr_byte)�.0�cr r r � <genexpr>� s z)YubiKeyHIDDevice._read.<locals>.<genexpr>z
READ : %sT)�colorize)�_USB_TYPE_CLASS�_USB_RECIP_INTERFACE�_USB_ENDPOINT_IN�_REPORT_TYPE_FEATUREr9 �
controlMsg�_HID_GET_REPORT�_FEATURE_RPT_SIZE�_USB_TIMEOUT_MS�lenrD r �joinr �hexdump)r% � request_typerX Zrecvr? r r r r= � s zYubiKeyHIDDevice._readc C sF x@|j | jd�D ].}d}| jr&|\}}| jtj� | j||� qW dS )zy
Write a YubiKeyFrame to the USB HID.
Includes polling for YubiKey readiness before each write.
)r8 NT)Zto_feature_reportsr8 rG r
rH �
_raw_write)r% rK r? � debug_strr r r rF � s zYubiKeyHIDDevice._writec C s d}| j |� | jtj� dS )z;
Reset read mode by issuing a dummy write.
s �T)rk rG r
rH )r% r? r r r rQ � s
z YubiKeyHIDDevice._write_resetNc C s� | j r6|sd}tj|dd�dd � }| jd||f � ttB tB }td> }| jj |t
||t d�}|t kr�| j d t |f � t
d
��|S ) z(
Write data to YubiKey.
� T)r^ Nr zWRITE : %s %s
r )rX rY z7Failed writing %i bytes (wrote %i) to USB HID YubiKey.
z!Failed talking to USB HID YubiKey���)r8 r ri rD r_ r` �_USB_ENDPOINT_OUTrb r9 rc �_HID_SET_REPORTrf re r )r% r? rl ri rj rX Zsentr r r rk � s" zYubiKeyHIDDevice._raw_writec C s | j d||�S )z�
Wait for the YubiKey to turn OFF the bits in 'mask' in status responses.
Returns the 8 bytes last read.
�nand)�_waitfor)r% �maskrR r r r rG � s z YubiKeyHIDDevice._waitfor_clearc C s | j d||�S )z�
Wait for the YubiKey to turn ON the bits in 'mask' in status responses.
Returns the 8 bytes last read.
�and)rr )r% rs rR r r r rO � s z YubiKeyHIDDevice._waitfor_setr c
C sh d}d}|d d d }d}�xD|�sbt j|� | j� } tj| d �}
|
tj@ r�|s�d}|
tj@ } | jd| � |r�t d | �} | d d d }|d
kr�|
|@ |ks�d}n | jd t
|
�|
t
|�|f � nF|d k�r|
|@ |kr�d}n | jd
t
|
�|
t
|�|f � n
f �st �|�s\|d8 }|dk�rL|d
k�r:d| } nd| } t j
| ��t || d�}q | S q W dS )z�
Wait for the YubiKey to either turn ON or OFF certain bits in the status byte.
mode is either 'and' or 'nand'
timeout is a number of seconds (precision about ~0.5 seconds)
Fg{�G�z�?r r � rM Tz0Device indicates RESP_TIMEOUT (%i seconds left)
� rq z0Status %s (0x%x) has not cleared bits %s (0x%x)
rt z,Status %s (0x%x) has not set bits %s (0x%x)
r z2Timed out waiting for YubiKey to clear status 0x%xz0Timed out waiting for YubiKey to set status 0x%xg �?N)�time�sleepr= r rP r
ZRESP_TIMEOUT_WAIT_FLAGZRESP_TIMEOUT_WAIT_MASKrD �min�bin�AssertionErrorr ZYubiKeyTimeout)
r% r) rs rR rY Zfinishedrx Zwait_numZ resp_timeoutrT rU Z seconds_left�reasonr r r rr sJ
zYubiKeyHIDDevice._waitforc C s� | j |�}|r*|jd }|jd d | _ntd��y|j� | _| jjd� W nD tk
r� } z(dt |�krv| j
d� n
| j
d� W Y dd}~X nX y| jj d� W n t j
k
r� | j
d� Y nX | jj| j� d S )
z Perform HID initialization r zNo USB YubiKey foundz-could not detach kernel driver from interfacez3The in-kernel-HID driver has already been detached
z"detachKernelDriver not supported!
Nr z)Unable to set configuration, ignoring...
T)�_get_usb_deviceZconfigurationsZ
interfaces�_usb_intr �openr9 ZdetachKernelDriver� Exception�strrD ZsetConfiguration�usbZUSBErrorZclaimInterface)r% r< Z
usb_deviceZusb_conf�errorr r r r: ; s$
zYubiKeyHIDDevice._openc C s: | j j� y| j jjd� W n Y nX d| _d| _ dS )z2
Release the USB interface again.
r NT)r9 ZreleaseInterfaceZdevZattach_kernel_driverr~ )r% r r r r@ U s
zYubiKeyHIDDevice._closec
s� y2ddl � ddl� � fdd�� jjdtd�D �}W n. tk
r` ddl� dd� � j� D �}Y nX x<|D ]4}|jtkrh|j t
j dd�krh|dkr�|S |d 8 }qhW dS )
z�
Get YubiKey USB device.
Optionally allows you to skip n devices, to support multiple attached YubiKeys.
r Nc s g | ]}� j j|��qS r )�legacyZDevice)r[ �d)r� r r �
<listcomp>n s z4YubiKeyHIDDevice._get_usb_device.<locals>.<listcomp>T)Zfind_all�idVendorc S s g | ]}|j D ]}|�qqS r )�devices)r[ Zbusr� r r r r� s s )Zotpr ) Zusb.coreZ
usb.legacyZcore�findr
� ImportErrorr� Zbussesr� Z idProductr �all)r% r< r� Zdevicer )r� r r} c s
z YubiKeyHIDDevice._get_usb_deviceTc C sF | j rB|r6| jj}t| d�r&t| d�}tjjd| � tjj|� dS )z/ Print out to stderr, if debugging is enabled. Z debug_prefixz%s: N)r8 r& r �hasattr�getattr�sys�stderr�write)r% �outZ print_prefixZprer r r rD | s
zYubiKeyHIDDevice._debug)Fr )F)N)F)F)r )r )r )T)r r r r r$ r; rB rL rW r= rF rQ rk rG rO rr r: r@ r} rD r r r r r7 l s"
7
r7 c s� e Zd ZdZdZdZeZd"� fdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
d#dd�Zd$dd�Zdd� Zd%d d �Zd d � Zd d!� Z� ZS )&r a�
Class for accessing a YubiKey over USB HID.
This class is for communicating specifically with standard YubiKeys
(USB vendor id = 0x1050, product id = 0x10) using USB HID.
There is another class for the YubiKey NEO BETA, even though that
product also goes by product id 0x10 for the BETA versions. The
expectation is that the final YubiKey NEO will have it's own product id.
Tested with YubiKey versions 1.3 and 2.2.
r zYubiKey (or YubiKey NANO)Fr Nc sH t t| �j|� |dkr&t||�| _n|| _| j| j| j� dd�| _dS )z�
Find and connect to a YubiKey (USB HID).
Attributes :
skip -- number of YubiKeys to skip
debug -- True or False
NF)r r! r" ) r# r r$ r7 �_device�_capabilities_clsr � version_num� capabilities)r% r8 r< Z
hid_device)r& r r r$ � s zYubiKeyUSBHID.__init__c C s d| j jtt| ��| j� f S )Nz'<%s instance at %s: YubiKey version %s>)r& r �hex�idr! )r% r r r �__repr__� s
zYubiKeyUSBHID.__repr__c C s d| j | j� f S )Nz%s (%s))r r! )r% r r r �__str__� s zYubiKeyUSBHID.__str__c C s
| j j� S )z*
Poll YubiKey for status.
)r� r; )r% r r r r; � s zYubiKeyUSBHID.statusc C s | j jj� S )z; Get the YubiKey version as a tuple (major, minor, build). )r� r> �ykver)r% r r r r� � s zYubiKeyUSBHID.version_numc C s | j jj� S )z Get the YubiKey version. )r� r> r! )r% r r r r! � s zYubiKeyUSBHID.versionTc C s& | j j� s tjd| j� ��| j|�S )z7 Get the YubiKey serial number (requires YubiKey 2.2). z'Serial number unsupported in YubiKey %s)r� r, r �YubiKeyVersionErrorr! � _read_serial)r% rR r r r �serial� s
zYubiKeyUSBHID.serialr r c C s4 | j j|�s"tjd|| j� f ��| j|||||�S )zR Issue a challenge to the YubiKey and return the response (requires YubiKey 2.2). z/%s challenge-response unsupported in YubiKey %s)r� r+ r r� r! �_challenge_response)r% � challenger) r4 �variablerR r r r �challenge_response� s z YubiKeyUSBHID.challenge_responsec K s t f | j� | jd�|��S )z6 Get a configuration object for this type of YubiKey. )r� r� )�YubiKeyConfigUSBHIDr� r� )r% �kwr r r � init_config� s
zYubiKeyUSBHID.init_configc C sZ |j � }|| j� kr4tjd|d |d | j� f ��| jj|�sLtd| ��| jj ||�S )z' Write a configuration to the YubiKey. z9Configuration requires YubiKey version %i.%i (this is %s)r r z$Can't write configuration to slot %i)
Zversion_requiredr� r r� r! r� r5 r r� rL )r% rJ r4 Z cfg_req_verr r r � write_config� s zYubiKeyUSBHID.write_configc C sV t jtjd�}| jj|� | jj|d�}tj|dd� �sBt d��t
j d|�}|d S )z. Read the serial number from a YubiKey > 2.2. )rE )rR Nru z!Read from device failed CRC checkz>lxxxr ) r � YubiKeyFramer Z
DEVICE_SERIALr� rF rW r �validate_crc16r �struct�unpack)r% rR rK �responser� r r r r� � s zYubiKeyUSBHID._read_serialc C s> |dkrnt |�tjkr.tjdtjt |�f ��t |�tjk rfd}|rX|dd� |krXd}|jtj|�}tj}nP|dkr�t |�tjkr�tjdtjt |�f ��|jtjd�}d }ntjd
| ��yt| | }W n tjd | ��Y nX t j
||d �} | j j | � | j j
|d
�}
tj|
d|d � ��s2td��|
d|� S )z- Do challenge-response with a YubiKey > 2.0. r z#Mode HMAC challenge too big (%i/%i)� r N� �r z,Mode OTP challenge must be %i bytes (got %i)� z9Invalid mode supplied (%s, valid values are HMAC and OTP)zInvalid slot specified (%s))rE Zpayload)rR r z!Read from device failed CRC checkrn )rg r
ZSHA1_MAX_BLOCK_SIZEr Z
InputError�ljustZSHA1_DIGEST_SIZEZUID_SIZE�_CMD_CHALLENGEr r� r� rF rW r r� r ) r% r� r) r4 r� rR Zpad_withZ response_lenrE rK r� r r r r� � s8 z!YubiKeyUSBHID._challenge_response)Fr N)T)r r TT)r )r r r r r � descriptionr r� r$ r� r� r; r� r! r� r� r� r� r� r� r6 r r )r& r r � s
c @ s@ e Zd ZdZdZdZdd� Zdd� Zdd � Zd
d � Z d d
� Z
dS )r zD Class to represent the status information we get from the YubiKey. r r c C s, d}t j||�\| _| _| _| _| _| _d S )Nz <x BBB B H B)r� r� �
version_major�
version_minor�
version_buildrC � touch_levelrU )r% r? Zfmtr r r r$ s
z YubiKeyUSBHIDStatus.__init__c C sZ d}d}| j � dkr d| j� }| jr0d| j }d| jjtt| ��| j� | j| j ||f S ) Nrm r r r z
, valid=%sz
(flags 0x%x)zG<%s instance at %s: YubiKey version %s, pgm_seq=%i, touch_level=%i%s%s>)r r r )
r� rI rU r& r r� r� r! rC r� )r% Z valid_strZ flags_strr r r r� ' s
z YubiKeyUSBHIDStatus.__repr__c C s | j | j| jfS )zQ Returns a tuple with the (major, minor, build) version of the YubiKey firmware. )r� r� r� )r% r r r r� 9 s zYubiKeyUSBHIDStatus.ykverc C s d| j � }|S )z2 Return the YubiKey firmware version as a string. z%d.%d.%d)r� )r% r! r r r r! = s zYubiKeyUSBHIDStatus.versionc C s\ | j � dk r td| j� ��g }| j| j@ | jkr<|jd� | j| j@ | jkrX|jd� |S )zL Return a list of slots having a valid configurtion. Requires firmware 2.1. r r r z(Valid configs unsupported in firmware %s)r r r )r� r r! r� �
CONFIG1_VALID�append�
CONFIG2_VALID)r% rS r r r rI B s
z!YubiKeyUSBHIDStatus.valid_configsN) r r r r r� r� r$ r� r� r! rI r r r r r s c s" e Zd ZdZd� fdd� Z� ZS )r� z3
Configuration class for USB HID YubiKeys.
Nc s t t| �jf ||d�|�� d S )N)r� r� )r# r� r$ )r% r� r� r� )r& r r r$ R s z YubiKeyConfigUSBHID.__init__)N)r r r r r$ r6 r r )r& r r� N s r� � )+r �__all__Zyubico_versionr rm r r r r r
r r r
r r r� rw r� r� r_ r` ra ro rd rp rf re rb Z
CHAL_HMAC1Z
CHAL_HMAC2Z CHAL_OTP1Z CHAL_OTP2r� Z YubicoErrorr ZYubiKeyCapabilitiesr �objectr7 r r Z
YubiKeyConfigr� r r r r �<module> sH 4 ?