,fdZddlZddlZddlZddlZddlmZddlmZdZefdZ Gdde Z y) zHID Transport for U2F. This module imports the U2F HID Transport protocol as well as methods for discovering devices implementing this protocol. N)errors)hidc&|ddk(r |ddk(ryy)N usage_pageiusageTF)devices %lib/third_party/pyu2f/hidtransport.pyHidUsageSelectorr s  LV#w4(?  c#KtjD]1}||s tj|d}t|3y#t$rY@wxYww)Npath)r EnumerateOpenUsbHidTransportOSError)selectorddevs r DiscoverLocalHIDU2FDevicesr#sR ==?a{ hhqy!c""     s( A%A  A AAAAceZdZdZdZdZdZdZdZdZ dZ d Z e gd Z e d gZGd d eZGddeZddZdZdZdZdZdZdZdZdZdZdZy)ramImplements the U2FHID transport protocol. This class implements the U2FHID transport protocol from the FIDO U2F specs. This protocol manages fragmenting longer messages over a short hid frame (usually 64 bytes). It exposes an APDU channel through the MSG command as well as a series of other commands for configuring and interacting with the device. )r!r!r!c,eZdZdZdZdZedZy)UsbHidTransport.InitPacketaRepresent an initial U2FHID packet. Represent an initial U2FHID packet. This packet contains metadata necessary to interpret the entire packet stream associated with a particular exchange (read or write). Attributes: packet_size: The size of the hid report (packet) used. Usually 64. cid: The channel id for the connection to the device. size: The size of the entire message to be sent (including all continuation packets) payload: The portion of the message to put into the init packet. This must be smaller than packet_size - 7 (the overhead for an init packet). c||_t|dk7s |dkDs|dk\rtjt||jdz kDrtj||_||_||_||_y)Nr!i) packet_sizelenrInvalidPacketErrorcidcmdsizepayload)selfr(r+r,r-r.s r __init__z#UsbHidTransport.InitPacket.__init__Vsr$d SQ#)tu}'')) W((1, ,''))dhdhdidlr ctd}|j|dd|j|d<tjd|d|j |j |ddt|j zttt|S)Serializes the packet.@rr&>Hr') bytearrayr+r,struct pack_intor-r.r)listmapintr/rets r ToWireFormatz'UsbHidTransport.InitPacket.ToWireFormatbsn bMcc!Ahxxc!f tS!TYY/%)\\c!ADLL! !" #c3-  r ct|}t||k7rtj|dd}|d}t j dt |ddd}|dd|z}tj|||||S)amDerializes the packet. Deserializes the packet from wire format. Args: packet_size: The size of all packets (usually 64) data: List of ints or bytearray containing the data from the wire. Returns: InitPacket object for specified data Raises: InvalidPacketError: if the data isn't a valid InitPacket rr&r4r5r') r6r)rr*r7unpackbytesr InitPacket)r(databar+r,r-r.s r FromWireFormatz)UsbHidTransport.InitPacket.FromWireFormatks T?b RK '')) qGc qEc ]]4r!Aw 0 3d1QXg  ' ' S#tW MMr N__name__ __module__ __qualname____doc__r0r> staticmethodrEr r r rBr$Es' !NNr rBc,eZdZdZdZdZedZy)UsbHidTransport.ContPacketaRepresents a continutation U2FHID packet. Represents a continutation U2FHID packet. These packets follow the intial packet and contains the remaining data in a particular message. Attributes: packet_size: The size of the hid report (packet) used. Usually 64. cid: The channel id for the connection to the device. seq: The sequence number for this continuation packet. The first continuation packet is 0 and it increases from there. payload: The payload to put into this continuation packet. This must be less than packet_size - 5 (the overhead of the continuation packet is 5). c||_||_||_||_t ||jdz kDrt j |dkDrt j y)Nr5)r(r+seqr.r)rr*)r/r(r+rPr.s r r0z#UsbHidTransport.ContPacket.__init__s`$ddhdhdl W((1, ,'')) s'')) r ct|j}|j|dd|j|d<|j|ddt |jzt tt|S)r2rr&r5) r6r(r+rPr.r)r9r:r;r<s r r>z'UsbHidTransport.ContPacket.ToWireFormats] d&& 'cc!Ahxxc!f%)\\c!ADLL! !" #c3-  r ct|}t||k7rtj|dd}|d}|dd}tj ||||S)amDerializes the packet. Deserializes the packet from wire format. Args: packet_size: The size of all packets (usually 64) data: List of ints or bytearray containing the data from the wire. Returns: InitPacket object for specified data Raises: InvalidPacketError: if the data isn't a valid ContPacket rr&r5N)r6r)rr*r ContPacket)r(rCrDr+rPr.s r rEz)UsbHidTransport.ContPacket.FromWireFormats` T?b RK '')) qGc qEc 12g  ' ' S#w GGr NrFr r r rSrMs' *!HHr rSc*||_|j}|j}||k7rtjd|dk(rtjd||_||_tjd|_ |jy)Nz6unsupported device with different in/out packet sizes.rzunable to determine packet sizezpyu2f.hidtransport) hid_deviceGetInReportDataLengthGetOutReportDataLengthr HardwareErrorr(read_timeout_secslogging getLoggerlogger InternalInit)r/rUrYin_sizeout_sizes r r0zUsbHidTransport.__init__s DO..0G002H(  B DD!|  !B CCD.D##$89DKr cF|jtj|}|SN)InternalExchanger U2FHID_MSG)r/msgrs r SendMsgByteszUsbHidTransport.SendMsgBytess o88#>A Hr cV|jtjt|gSra)rbr U2FHID_PROMPTr6)r/lengths r SendBlinkzUsbHidTransport.SendBlinks'  !>!>!*F8!4 66r cT|jtjtgSra)rbr U2FHID_WINKr6)r/s r SendWinkzUsbHidTransport.SendWinks  !oo122  //6 77r ct|}|jdz }|d|}tj|j|j|t||}|d|=|t|z}|j |d}|dkDrp|jdz }|d|}|d|=|t|z}tj |j|j||} |j | |dz }|dkDroyy)z8Sends a message to the device, including fragmenting it.r'rr5rN)r)r(rrBr+InternalSendPacketrS) r/r,r.length_to_send max_payload first_frame first_packetrP next_frame next_packets r rzUsbHidTransport.InternalSends\N""Q&K!K(K"--d.>.>#.1'lKIL+ c+&&NL) C 1 $$q(k1[)j !K- J'n#..t/?/?3/9;k k* Qhc 1 r c|j}|jjdt|z|jj |y)Nzsending packet: )r>r\r~rrUWrite)r/packetwires r rz"UsbHidTransport.InternalSendPackets>    DKK(3t945OO$r c|jj}|jjdt |z|S)Nzrecv: )rUReadr\r~r)r/frames r InternalReadFramez!UsbHidTransport.InternalReadFrame s6 OO "EKKhU+, Lr c|j}tjj|j|}|j }|j t|j z }d}|dkDr|j}tjj|j|}|j|jk7rY||jk7rtjd|t|j z}|j|j |dz }|dkDr|d|j }|j|fS)z?Receives a message from the device, including defragmenting it.rzPackets received out of orderr)rrrBrEr(r.r-r)rSr+rPrrXextendr,)r/ first_readrrCto_readrP next_readrs r rzUsbHidTransport.InternalRecv)s)'')J"--<N>N>GIk [__ $   ""#BCC[(())g kk+%%& Qhc) A+. ,## $D   d ##r N)g@)rGrHrIrJrorcrlrhrx U2FHID_LOCKr U2FHID_SYNCr6rurobjectrBrSr0rfrjrmrpr]rbrrrrr r r rr.s+*+-++,+"#;<v&=N6=N~=H6=H~" 6MD  8*. "$r r) rJrZrvr7rpyu2frrr rrrr r r rs;   )9 ]$f]$r