o i@sHdZddlZddlmZdZdZdZGdddeZGd d d eZ dS) aImplement the U2F variant of ISO 7816 extended APDU. This module implements a subset ISO 7816 APDU encoding. In particular, it only supports extended length encoding, it only supports commands that expect a reply, and it does not support explicitly specifying the length of the expected reply. It also implements the U2F variant of ISO 7816 where the Lc field is always specified, even if there is no data. N)errorsc@sFeZdZdZdZdZdZdZdZd ddZ ddZ ddZ d d Z dS) CommandApduzRepresents a Command APDU. Represents a Command APDU sent to the security key. Encoding is specified in FIDO U2F standards. NcCsB||_||_||_||_|rt|dkrt|r||_dSdS)Ni)clainsp1p2lenrInvalidCommandErrordata)selfrrr r r r3/tmp/google-cloud-sdk/lib/third_party/pyu2f/apdu.py__init__.s zCommandApdu.__init__cCsx|}td}|j|d<|j|d<|j|d<|j|d<|jr3||||j|ddg|S|gd|S)zSerialize the command. Encodes the command as per the U2F specs, using the standard ISO 7816-4 extended encoding. All Commands expect data, so Le is always present. Returns: Python bytearray of the encoded command. rrrr)rrr)InternalEncodeLc bytearrayrrr r r extendrlcoutrrr ToByteArray8s       zCommandApdu.ToByteArraycCsf|}td}|j|d<|j|d<|j|d<|j|d<|||jr*||j|ddg|S)zSerialize the command in the legacy format. Encodes the command as per the U2F specs, using the legacy encoding in which LC is always present. Returns: Python bytearray of the encoded command. rrrrr)rrrrr r rr rrrrToLegacyU2FByteArrayQs       z CommandApdu.ToLegacyU2FByteArraycCs0d}|jr t|j}td|}t|ddS)Nrz>Ir)r r structpackr)rdlfourbyterrrris   zCommandApdu.InternalEncodeLc)N) __name__ __module__ __qualname____doc__rrr r r rrrrrrrrr"s   rc@s4eZdZdZdZdZdZddZddZddZ dS) ResponseApduzRepresents a Response APDU. Represents a Response APU sent by the security key. Encoding is specified in FIDO U2F standards. NcCsP||_|r t|dkrtt|dkr|dd|_|d|_|d|_dS)Nr)dbg_full_packetr rInvalidResponseErrorbodysw1sw2)rr rrrr}s  zResponseApdu.__init__cCs|jdko |jdkS)Nr)r)r*rrrr IsSuccessszResponseApdu.IsSuccesscCsp|jdkr|jdkrt|jdkr|jdkrt|jdkr*|jdkr*t|s6t|j|jdS)Nijgr)r)r*rTUPRequiredErrorInvalidKeyHandleErrorr- ApduErrorr,rrrCheckSuccessOrRaisesz ResponseApdu.CheckSuccessOrRaise) rr r!r"r(r)r*rr-r6rrrrr#ss r#) r"rpyu2fr CMD_REGISTERCMD_AUTH CMD_VERSIONobjectrr#rrrrs Q