fume-manage-python.git

U  
 ý°d\?ã@sdZddlmZddlZddlZddlZddlZddlZddlZddl    Z    ddl
Z
ddlZddlm Z mZmZmZmZmZmZmZmZmZe dej¡Zddd    d
dZdd dddZdddddZdddddZdddddZdGddd d!d"d#Zdd$d%d&d'Z d(d)d*d+d,Z!d(d)d*d-d.Z"d/d0d1d2Z#d3d4d5d6d7d8Z$d3dd$d9d:d;Z%d<d=d>d?d@dAZ&d4ddBdCdDZ'GdEdFdFZ(dS)Ha¦
Low-level helpers for the SecureTransport bindings.
 
These are Python functions that are not directly related to the high-level APIs
but are necessary to get them to work. They include a whole bunch of low-level
CoreFoundation messing about and memory management. The concerns in this module
are almost entirely about trying to avoid memory leaks and providing
appropriate and useful assistance to the higher-level code.
é)ÚannotationsNé)
ÚCFArrayÚCFConstÚCFDataÚCFDictionaryÚCFMutableArrayÚCFStringÚ    CFTypeRefÚCoreFoundationÚSecKeychainRefÚSecuritys;-----BEGIN CERTIFICATE-----
(.*?)
-----END CERTIFICATE-----Úbytesr)Ú
bytestringÚreturncCst tj|t|¡S)zv
    Given a bytestring, create a CFData object from it. This CFData object must
    be CFReleased by the caller.
    )rÚCFDataCreateÚkCFAllocatorDefaultÚlen)r©rúad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\urllib3/contrib/_securetransport/low_level.pyÚ_cf_data_from_bytes)s
ÿrz#list[tuple[typing.Any, typing.Any]]r)ÚtuplesrcCsZt|}dd|D}dd|D}tj||}tj||}t tj|||tjtj¡S)zK
    Given a list of Python tuples, create an associated CFDictionary.
    css|]}|dVqdS)rNr©Ú.0ÚtrrrÚ    <genexpr><sz-_cf_dictionary_from_tuples.<locals>.<genexpr>css|]}|dVqdS)rNrrrrrr=s)rrr
ZCFDictionaryCreaterZkCFTypeDictionaryKeyCallBacksZkCFTypeDictionaryValueCallBacks)rZdictionary_sizeÚkeysÚvaluesZcf_keysZ    cf_valuesrrrÚ_cf_dictionary_from_tuples3súrr    )Úpy_bstrrcCs t |¡}t tj|tj¡}|S)zi
    Given a Python binary data, create a CFString.
    The string must be CFReleased by the caller.
    )ÚctypesÚc_char_prZCFStringCreateWithCStringrrÚkCFStringEncodingUTF8)rZc_strÚcf_strrrrÚ_cfstrKs
ýr$zlist[bytes]r)Úlstrc
Cs°d}zdt tjdt tj¡¡}|s*td|D]6}t|}|sFtdzt     ||¡W5t |¡Xq.WnBt
k
rª}z$|rt |¡t d|¡dW5d}~XYnX|S)zª
    Given a list of Python binary data, create an associated CFMutableArray.
    The array must be CFReleased by the caller.
 
    Raises an ssl.SSLError on failure.
    NrúUnable to allocate memory!zUnable to allocate array: ) rÚCFArrayCreateMutablerr ÚbyrefÚkCFTypeArrayCallBacksÚMemoryErrorr$Ú    CFReleaseÚCFArrayAppendValueÚ BaseExceptionÚsslÚSSLError)r%Zcf_arrÚitemr#ÚerrrÚ_create_cfstring_arrayYs*
ý
$r2z
str | None)ÚvaluercCsnt |t tj¡¡}t |tj¡}|dkrXt d¡}t     ||dtj¡}|sRt
d|j}|dk    rj| d¡}|S)z¨
    Creates a Unicode string from a CFString object. Used entirely for error
    reporting.
 
    Yes, it annoys me quite a lot that this function is this complex.
    Niz'Error copying C string from CFStringRefúutf-8) r ÚcastÚPOINTERÚc_void_prZCFStringGetCStringPtrrr"Úcreate_string_bufferZCFStringGetCStringÚOSErrorr3Údecode)r3Zvalue_as_void_pÚstringÚbufferÚresultrrrÚ_cf_string_to_unicodexs&ÿ
ÿ
r>Úintztype[BaseException] | NoneÚNone)ÚerrorÚexception_classrcCs^|dkrdSt |d¡}t|}t |¡|dks:|dkrDd|}|dkrRtj}||dS)z[
    Checks the return code and throws an exception if there is an error to
    report
    rNÚz    OSStatus )r ZSecCopyErrorMessageStringr>rr+r.r/)rArBZcf_error_stringÚoutputrrrÚ_assert_no_errors
 
rEr)Ú
pem_bundlercCsÚ| dd¡}ddt |¡D}|s.t d¡t tjdt     tj
¡¡}|sTt d¡z`|D]V}t|}|stt d¡t  tj|¡}t |¡|st d¡t ||¡t |¡qZWn tk
rÔt |¡YnX|S)    z
    Given a bundle of certs in PEM format, turns them into a CFArray of certs
    that can be used to validate a cert chain.
    s 
ó
cSsg|]}t | d¡¡qS)r)Úbase64Ú    b64decodeÚgroup)rÚmatchrrrÚ
<listcomp>°sz(_cert_array_from_pem.<locals>.<listcomp>zNo root certificates specifiedrr&zUnable to build cert object!)ÚreplaceÚ _PEM_CERTS_REÚfinditerr.r/rr'rr r(r)rr ZSecCertificateCreateWithDatar+r,Ú    Exception)rFZ    der_certsZ
cert_arrayZ    der_bytesZcertdataÚcertrrrÚ_cert_array_from_pem¨s>ÿ
 
ý
 
ÿ
 
 
rRr
Úbool)r0rcCst ¡}t |¡|kS)z=
    Returns True if a given CFTypeRef is a certificate.
    )r ZSecCertificateGetTypeIDrÚCFGetTypeID©r0ÚexpectedrrrÚ_is_certÖsrWcCst ¡}t |¡|kS)z;
    Returns True if a given CFTypeRef is an identity.
    )r ZSecIdentityGetTypeIDrrTrUrrrÚ_is_identityÞsrXztuple[SecKeychainRef, str])rc
Cst d¡}t |dd¡ d¡}t |dd¡}t ¡}tj ||¡     d¡}t
 ¡}t
 |t ||ddt |¡¡}t|||fS)a³
    This function creates a temporary Mac keychain that we can use to work with
    credentials. This keychain uses a one-time password and a temporary file to
    store the data. We expect to have one keychain per socket. The returned
    SecKeychainRef must be freed by the caller, including calling
    SecKeychainDelete.
 
    Returns a tuple of the SecKeychainRef and the path to the temporary
    directory that contains it.
    é(Nér4F)ÚosÚurandomrHÚ    b16encoder:ÚtempfileÚmkdtempÚpathÚjoinÚencoder rZSecKeychainCreaterr r(rE)Zrandom_bytesÚfilenameÚpasswordZ tempdirectoryZ keychain_pathÚkeychainÚstatusrrrÚ_temporary_keychainæs 
ÿrgrÚstrz'tuple[list[CFTypeRef], list[CFTypeRef]])rer`rcCsg}g}d}t|d}| ¡}W5QRXz²t tj|t|¡}t ¡}t     |ddddd|t
 |¡¡}t|t  |¡}    t|    D]X}
t ||
¡}t
 |tj¡}t|r¼t |¡| |¡qt|rt |¡| |¡qW5|rìt |¡t |¡X||fS)zÊ
    Given a single file, loads all the trust objects from it into arrays and
    the keychain.
    Returns a tuple of lists: the first list is a list of identities, the
    second a list of certs.
    NÚrbr)ÚopenÚreadrr+rrrZ
CFArrayRefr Z SecItemImportr r(rEZCFArrayGetCountÚrangeZCFArrayGetValueAtIndexr5r
rWZCFRetainÚappendrX)rer`ÚcertificatesÚ
identitiesZresult_arrayÚfZraw_filedataZfiledatar=Zresult_countÚindexr0rrrÚ_load_items_from_file    sJ    ÿø
 
 
 
 
rr)reÚpathsrc 
Gsêg}g}dd|D}z°|D]&}t||\}}| |¡| |¡q|st ¡}    t ||dt     
|    ¡¡}
t|
| |    ¡t |  d¡¡t tjdt     
tj¡¡}t ||¡D]}t ||¡q®|W¢St ||¡D]}t |¡qÔXdS)zü
    Load certificates and maybe keys from a number of files. Has the end goal
    of returning a CFArray containing one SecIdentityRef, and then zero or more
    SecCertificateRef objects, suitable for use as a client certificate trust
    chain.
    css|]}|r|VqdS)Nr)rr`rrrrfsz*_load_client_cert_chain.<locals>.<genexpr>rN)Ú    itertoolsÚchainrr+rrÚextendr ZSecIdentityRefZ SecIdentityCreateWithCertificater r(rErmÚpopr'rr)r,) rersrnroZfiltered_pathsÚobjÚ    file_pathZnew_identitiesZ    new_certsZnew_identityrfZtrust_chainr0rrrÚ_load_client_cert_chainBs8 
ÿ
 
ýrz)ré)ér)r|r)r|r{)r|r|)ZSSLv2ÚSSLv3ÚTLSv1zTLSv1.1zTLSv1.2)Úversionrc    CsHt|\}}d}d}t d||¡}t|}d}t d||||¡|}|S)z6
    Builds a TLS alert record for an unknown CA.
    r{é0z>BBéz>BBBH)ÚTLS_PROTOCOL_VERSIONSÚstructÚpackr)    rZver_majZver_minZseverity_fatalZdescription_unknown_caÚmsgZmsg_lenZrecord_type_alertÚrecordrrrÚ_build_tls_unknown_ca_alertsrc@s°eZdZdZdZdZdZdZdZdZ    dZ
d    ZdZdZ dZdZdZd
ZdZdZdZdZd ZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(d Z)d!Z*d"Z+d#S)$Ú SecurityConstzU
    A class object that acts as essentially a namespace for Security constants.
    rrr{éérZé
içr|ééi¸ÙÿÿiµÙÿÿi³Ùÿÿi¨Ùÿÿi²Ùÿÿi±Ùÿÿi¯Ùÿÿi®ÙÿÿiªÙÿÿi©Ùÿÿi¬Ùÿÿi«ÙÿÿiÙÿÿi ÙÿÿiÙÿÿiÙÿÿiÙÿÿiÙÿÿi ÷þÿiQÿÿi,ÿÿiRÿÿN),Ú__name__Ú
__module__Ú__qualname__Ú__doc__Z"kSSLSessionOptionBreakOnServerAuthZ kSSLProtocol2Z kSSLProtocol3Z kTLSProtocol1ZkTLSProtocol11ZkTLSProtocol12ZkTLSProtocol13ZkTLSProtocolMaxSupportedZkSSLClientSideZkSSLStreamTypeZkSecFormatPEMSequenceZkSecTrustResultInvalidZkSecTrustResultProceedZkSecTrustResultDenyZkSecTrustResultUnspecifiedZ&kSecTrustResultRecoverableTrustFailureZ kSecTrustResultFatalTrustFailureZkSecTrustResultOtherErrorZerrSSLProtocolZerrSSLWouldBlockZerrSSLClosedGracefulZerrSSLClosedNoNotifyZerrSSLClosedAbortZerrSSLXCertChainInvalidZerrSSLCryptoZerrSSLInternalZerrSSLCertExpiredZerrSSLCertNotYetValidZerrSSLUnknownRootCertZerrSSLNoRootCertZerrSSLHostNameMismatchZerrSSLPeerHandshakeFailZerrSSLPeerUserCancelledZerrSSLWeakPeerEphemeralDHKeyZerrSSLServerAuthCompletedZerrSSLRecordOverflowZerrSecVerifyFailedZerrSecNoTrustSettingsZerrSecItemNotFoundZerrSecInvalidTrustSettingsrrrrr¤sRr)N))rÚ
__future__rrHr rtr[Úrer.rr^ÚtypingZbindingsrrrrrr    r
rrr ÚcompileÚDOTALLrNrrr$r2r>rErRrWrXrgrrrzrrrrrrrÚ<module>sH    0ÿ
ÿ.#9Lû