o jF/@s*dZddlmZmZmZmZddlZddlZddlm Z ddl m Z ddl m Z mZmZmZgdZe d krdd lmZiZeZ d d Zd dZddZddZdgddZddZddZddZdhddZddZdd Zd!d"Z d#d$Z!d%d&Z"d'd(Z#d)d*Z$dhd+d,Z%dgd-d.Z&d/d0Z'd1d2Z(d3d4Z)d5d6Z*d7d8Z+d9d:Z,d;d<Z-d=d>Z.d Z/nddl0Z0dd?l0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6ee3e07e0j8fZ9d@d@d@d@dAZ:e5e6e3e07e3e2e4e0j;e0jZ>e:?d@d@d@d@dEe=?e0j8e3e6e3e6e>j@e>jAe07e0j8e07e07e0j8dF dGdHZBdId ZdJdZdKdZdgdLdZdMdZdNdZdOdZdhdPdZdQdZdRd ZdSd"Z dTd$Z!dUd&Z"dVd(Z#dWd*Z$dhdXd,Z%dgdYd.Z&dZd0Z'd[d2Z(d\d4Z)d]d6Z*d^d8Z+d_d:Z,d`d<Z-dad>Z.dbZ/dcddZCGdedfdfeDZEdS)izK Exceptions and compatibility shims for consistently using ctypes and cffi )unicode_literalsdivisionabsolute_importprint_functionN) find_library)ffi)str_clsbyte_cls int_types bytes_to_list)array_from_pointer array_setbuffer_from_bytesbuffer_from_unicodebuffer_pointer byte_arraybyte_string_from_bufferbytes_from_buffercallbackcastdereferrnoFFIEngineError get_libraryis_nullnativenewnull pointer_setref register_ffisizeofstruct struct_bytesstruct_from_bufferunwrapwrite_to_buffercffi)FFIcCs |t|<dSN) _ffi_registrylibraryffi_objr//root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/oscrypto/_ffi.pyr!8 r!cCs|tvrt|StSr*)r+r)r-r/r/r0_get_ffi;sr2cCs"tjdkr td|Std|S)Nwin32zunsigned char[]zchar[])sysplatformrr initializerr/r/r0r@s   rcCs td|S)Nz wchar_t []rrr6r/r/r0rEr1rcCs||||t|<dSr*)lenbufferdataoffsetr/r/r0r'Hsr'cCstd|gS)Nzchar *[]r8r;r/r/r0rKrcCt|}|||Sr*)r2r)r-type_valuer.r/r/r0rN rcCst|}||Sr*)r2r")r-rBr.r/r/r0r"Rs r"cCs.|durt||ddSt|ddSr*rr;r;maxlenr/r/r0rVsrcC t|Sr*)rstringr>r/r/r0r[ rcCs|Sr*r/ byte_stringr/r/r0r^rcCs ||d<dSNrr/pointer_rBr/r/r0rar1rcCt|D]\}}|||<qdSr* enumeratearrayrBindexvalr/r/r0rd rcCtjSr*)rNULLr/r/r/r0rhrcCsH|durdS|tjkr dStt|dkrdS|dtjkr"dSdS)NTvoid *Fr)rrYgetctypetypeofpointr/r/r0rks rcCrXr*)rrr/r/r/r0rvrZrcCsLt|}g}|dur|||tddgvr||dS|j|g|RS)NBCRYPT_KEY_HANDLEBCRYPT_ALG_HANDLEr)r2appendsetrr)r-rArBr.paramsr/r/r0rys  rcCs||Sr*r/rBr=r/r/r0r r cCs4|tkr t|S|tkrt|ddS||Sr*)r rrHr r;rArBr/r/r0rs  rcC|dSrMr/r^r/r/r0rrfrcCrhrMr/r^r/r/r0r&rfr&cCst|}|d|S)N%s *)r2r)r-namer.r/r/r0r#sr#cCst|ddSr*rDstruct_r/r/r0r$sr$cCsRt|}|d|}|d}t||}||}|||dd|dd<|S)Nrir)r2rr"r;)r-rjr;r.new_struct_pointer new_struct struct_size struct_bufferr/r/r0r%s  r%c Cst|}|d||f|}||}|dkrgSg}ddddddd}||v} td|D]} || } | r:|| } || q-|S)Nz%s[%s]rT)LPSTRLPCSTRLPWSTRLPCWSTRchar * wchar_t *)r2rr"rangerHrb) r-rjr_sizer.rT total_bytesoutput string_types string_typeirBr/r/r0r s(   r cCr@r*)r2r)r-signature_namefuncr.r/r/r0rrCr)pointerc_intc_char_pc_uintc_void_p c_wchar_pT)r[rvruchar **)r[rvrurintz unsigned intsize_tuint32_tr3)wintypes)rqrsrrrt) BYTErqrsrrrtULONGDWORDrurcCs|dddk}|r|dd}|dddko|tv}|r$|dd}|ddk}|rO||dd|d}|d krBd }nt|}|d |d}|tvrXt|}nt||}|rdt|}|||fS) Nz **z *[r]Tr)_pointer_typesfindr _type_mapgetattrctypesPOINTER)r-rAis_double_pointer is_pointeris_arrayr/r/r0 _type_infos&      rcCdSr*r/r,r/r/r0r!rLcCrGr*)rcreate_string_bufferr6r/r/r0r rIcCrGr*)rcreate_unicode_bufferr6r/r/r0rrIcCsRt|ttjrt||t|dS|dkr||_dS|jd|||_dSrM) isinstancerrc_bytememmover9rBrawr:r/r/r0r's  cCstt|tSr*)rrrrr>r/r/r0rcCs8t||\}}}|rt|}n|r||}t||Sr*)rrrr)r-rArBrrr/r/r0rs   cCs t|Sr*)rr")r-rBr/r/r0r"(rIcCs2t|tr t||S|dur|jd|S|jSrM)r_pointer_int_typesr string_atrrEr/r/r0r+s  cC|jSr*)rBr>r/r/r0r2rZcCstjt|t|Sr*)rrr9r rJr/r/r0r5scCs ||j_dSr*)contentsrBrNr/r/r0r8r1cCrPr*rQrSr/r/r0r;rWcCrr*r/r/r/r/r0r?rLcCs t| Sr*)boolr^r/r/r0rBrIcCstSr*)r get_errnor/r/r/r0rErfcCs`t||\}}}|r|dur||}d}n||}g}|dur$||||}|r.t|}|S)NT)rrbr)r-rArBrrrdrzr/r/r0rHs cCs0|dkr t|Stt||ttjSrM)rbyrefr addressofrrrer/r/r0r [s cCsdt||r|Stjdkr|tkrt|tr|St|tjr-|jtjkr-t t ||j S||j S)N)) rr4 version_inforr rArray_type_rrr_length_rBrgr/r/r0r`s  cCrhrMr/r^r/r/r0rirfcCrr*)rr^r/r/r0r&lrZcCstt||Sr*)rr)r-rjr/r/r0r#orcCst|t|jSr*)rrr"rrkr/r/r0r$rscCs4t||}|}tt||t|t|Sr*)rrrrr"r)r-rAr;class_rBr/r/r0r%us  cCsHt||\}}}t|t|}g}td|D] }|||q|SrM)rrrrrwrb)r-rAr_rx_rTrzr}r/r/r0r {s cCst|||Sr*)r)r-signature_typerr/r/r0rr?rcCsrt|}tjdkr7d|}|dd|}ttttdd}|s-|dkr-|}|S|dkr7||kr7|}|S) a Retrieve the C library path with special handling for Mac :param name: A unicode string of the library to search the system for :param dylib_name: Mac only - a unicode string of the unversioned dylib name :param version: Mac only - a unicode string of the dylib version to use. Used on macOS 10.15+ when the unversioned dylib is found, since unversioned OpenSSL/LibreSSL are just placeholders, and a versioned dylib must be imported. Used on macOS 10.16+ when find_library() doesn't return a result, due to system dylibs not being present on the filesystem any longer. :return: A unicode string of the path to the library darwinz /usr/lib/%sz.dylibz .%s.dylibr.) )r) rr4r5replacetuplemaprmac_versplit)rj dylib_nameversionr- unversioned versionedrr/r/r0rs  rc@seZdZdZdS)rz@ An exception when trying to instantiate ctypes or cffi N)__name__ __module__ __qualname____doc__r/r/r/r0rsr)rr*)Fr __future__rrrrr4r5 ctypes.utilrrr_typesr r r r __all__r(r)r+r!r2rrr'rrr"rrrrrrrrrr rrr&r#r$r%r renginerrrrrrrrrrrc_size_tc_uint32rrupdaterrrr Exceptionrr/r/r/r0s                   &