<[j5< F ddlmZddlZddlZddlZddlZddlmZmZddl m Z m Z m Z m Z ejeZdZdZdZd Zd Zd Zd Zd ZdZGddejZGddeZGddeZGddeZGddeZGddeZ GddeZ!ej"eZ#ej"eZ$ej"eZ%ej"e Z&ej'Z(ej"e!Z)ej*Z+ej*Z,ej-Z.ej/Z0ej1Z2ej3Z4dZ5ej6Z7ej6Z8ej6Z9ej6Z:ej6Z;e:Zej"eZ?ej@dejAe7ej*e8ej3ej"ejBe0ZCej@dejAe7ej*ZDdZEdZFdZGdZHdZIdZJdZKd ZLejMNejOPd!pd"ZQejMNejOPd#pd$ZRe$SeRd%ZTdeRjU_Ve#ej"ej*geRjU_We$eRjX_Vej*ejYej3geRjX_WejZeRj[_Ve%ejYe0e4geRj[_We.eRj\_Ve%geRj\_We.eRj]_Ve.eRj^_Vej_eRj`_Ve&eRja_VgeRja_We(eRjb_Ve$e2ejZgeRjb_We7eRjc_Ve%geRjc_We:geQjd_We>eQje_Ve)e9geQje_We#eQjf_Ve>geQjf_WdeQjg_Ve>e%geQjg_We,eQjh_VejigeQjh_We7eQjj_Ve=ej"ejigeQjj_We?eQjk_Ve)eQ R R R ) K  . **:24Mrc&eZdZdZdZdZdZdS)MacCtapHidConnectionc ||_t|j|_t|jd}|t krtd|t|_ d|_ tj |j z|_t|j|j|jj t tj|j dS)Nrz)Failed to open device for communication: ) descriptor_handle_from_pathpathr@r>IOHIDDeviceOpenK_IO_RETURN_SUCCESSOSErrorrr+r7r)c_uint8report_size_inin_report_buffer&IOHIDDeviceRegisterInputReportCallbackREGISTERED_READ_CALLBACKrD)selfrRr,s r__init__zMacCtapHidConnection.__init__s$' 88 &&t{A66 ( ( (NfNNOO O"'!"(*2K!K N N 44 K  ! O * $  T_ - -      rc t|j|j|jjt jdtddS)Nr) r>r[r@rZrRrYr)castIO_HID_REPORT_CALLBACK)r]s rclosezMacCtapHidConnection.close8sJ 44 K  ! O * K1 2 2       rc t|jtd|t |}|t krt d|dS)Nrz"Failed to write report to device: )r>IOHIDDeviceSetReportr@K_IO_HID_REPORT_TYPE_OUTPUTlenrVrW)r]datar,s r write_packetz!MacCtapHidConnection.write_packetAs[++ K '  II    ( ( (GvGGHH H ) (rcP |jdS#t$rtjt |f}|| |jdcYS#t$rtdwxYwwxYw)NF)targetargszFailed reading a response) r+getr threadingThreadrNstartjoinrW)r] read_threads r read_packetz MacCtapHidConnection.read_packetNs ;?&&u-- - ; ; ;#*2B$QQQK             ;**511111 ; ; ;9::: ;  ;s"AB%+BB%B!!B%N)rrrr^rbrhrrrrrrPrPsS   2    I I I ; ; ; ; ;rrPcLtd|d}t||}t||st d|dt|tkr*tdt|tj }t |ttj |std|d|jS)z'Reads int property from the HID device.Nrz Property 'z ' not foundzExpected number type, got zFailed to read property '')r5CFStringCreateWithCStringr>IOHIDDeviceGetProperty CFRelease ValueError CFGetTypeIDCFNumberGetTypeIDrWr)c_int32CFNumberGetValueK_CF_NUMBER_SINT32_TYPEbyrefvalue)devkeycf_keytype_refouts rget_int_propertyr[s  ) )$Q 7 7F++C88HLL 86c666777 ~~h2#7#7#9#999M2>>(3K3KMMNNN .  C  x)@&,sBSBS T T:8#888999 9rc^td|d}t||}t||sdSt|tkr*tdt|tj d}t ||tj |t}|sdS |j dpdS#t$rYdSwxYw)z*Reads string property from the HID device.NrzExpected string type, got zutf-8)r5rur>rvrwryCFStringGetTypeIDrWr)create_string_bufferCFStringGetCStringsizeof CF_STRING_BUILTIN_ENCODINGS_UTF8rdecodeUnicodeDecodeError)rrrrrrets rget_string_propertyrms  ) )$Q 7 7F++C88HLL t ~~h2#7#7#9#999M2>>(3K3KMMNNN  %c * *C  #v}S))+K  C ty((0D0 ttsD D,+D,ct|}tj}t|tj|}|t krtd||jS)zObtains the unique IORegistry entry ID for the device. Args: handle: reference to the device Returns: A unique ID for the device, obtained from the IO Registry z&Failed to obtain IORegistry entry ID: ) r>IOHIDDeviceGetServicer)c_uint64!IORegistryEntryGetRegistryEntryIDr~rVrWr)r@io_service_objentry_idr,s r get_device_idrsu0088N  H  4 4 X..F$$$GvGGHHH >rc"tjt|}t|}tt |}|std|dtt|S)Nz Device ID z, does not match any HID device on the system) r)rintr>IORegistryEntryIDMatchingIOServiceGetMatchingServiceK_IO_MASTER_PORT_DEFAULTrWIOHIDDeviceCreateK_CF_ALLOCATOR_DEFAULT)rTr matching_dict device_entrys rrSrSss4yy))H33H==M44 -L WU4UUUVVV  " "#9< H HHrc t|Sr4)rP)rRs ropen_connectionrs  + ++rc t|t}t|t}|tkr|tkrt |}t|t }t|t}t|t}t|t}t|t}t|t} tt||||| ||Std)NzNot a CTAP device)r&HID_DEVICE_PROPERTY_PRIMARY_USAGE_PAGE!HID_DEVICE_PROPERTY_PRIMARY_USAGErrrHID_DEVICE_PROPERTY_VENDOR_IDHID_DEVICE_PROPERTY_PRODUCT_IDrHID_DEVICE_PROPERTY_PRODUCT!HID_DEVICE_PROPERTY_SERIAL_NUMBER)HID_DEVICE_PROPERTY_MAX_INPUT_REPORT_SIZE*HID_DEVICE_PROPERTY_MAX_OUTPUT_REPORT_SIZEr strrx) r@ usage_pageusage device_idvidpidproductserialsize_insize_outs r_get_descriptor_from_handlers!&*PQQJ V%F G GE_$$*)<)<!&)) v'DEEv'EFF%f.IJJ$V-NOO"6+TUU#F,VWW  NNCgx&    ( ) ))rc:tt|Sr4)rrS)rTs rget_descriptorrs &'8'>'> ? ??rctdd}|std t|dt|}|std t|}t |z}t||g}|D]7} t|}| |(#t$rY4wxYw|t |t |S#t |wxYw#t |wxYw)Nrz&Unable to obtain HID manager referencez)Failed to obtain devices from HID manager) r>IOHIDManagerCreaterWIOHIDManagerSetDeviceMatchingIOHIDManagerCopyDevices CFSetGetCountIO_HID_DEVICE_REFCFSetGetValuesrappendrxr5rw)hid_mgrdevice_set_refnumdevices descriptorsr@rRs rlist_descriptorsrsr&&tQ//G @>??? ++GT:::66w?? GEFF F )%%n55C(3.11G   9 9 9K!  !r5in_dllrArrestypeargtypesruc_char_pc_boolrryrzrc_intr|r;rErwIOObjectReleaserrrrrrrIOHIDDeviceCloser?rIrvrdrr2r\r9rCrNrPrrrrSrrrrrrrrs$#""""" OOOOOOOOOOOO  8 $ $!,!-($3!$3!)<&,A)-C* +     f                W        G        7                7   V^F # # y)) fnW%%  &.,,!6>,//O"O ^ =?_#, M ]m m  !#V^M22"FN<00))   O OFN6>""   #&"4)99foVV#$    KW%%:9   [ K-..LK *005LMM!(.&.*I*IJ'4$ O O O)% !'  O  " $&-))$l!0"$ 6"/1A6=!Q  $  "-#5 %5~$F!(2%*<)=&.2#+0BK/P#,*C',2O+<(2;'/FN6?##4'0 #4$4l#C !*#4fo"F/3$,1$- 26&.3&/ (3$):M(J%%."  O '#-9!).9;L-M!*<<<212DEE... #?#899444n:;:;:;:;:;,:;:;:;z$0* I I I,,,***"@@@r