
    &g_B                    P   % S r SSKJr  SSKrSSKrSSKrSSKrSSKJrJ	r	  SSK
JrJrJrJrJrJrJr  SSKJr  SSKJr  SSKJrJrJrJr  SS	KJrJr  SS
KJr  SSK J!r!  SSK"J#r#J$r$  SSK%J&r&  \(       a  SSK'J(r(  SSK)J*r*  Sr+Sr, SSK-r.\/" \0" \1\.Rd                  Rg                  S5      SS 5      5      S:  a  Sr,Sr5S/S jr6S0S jr7S1S jr8S2S jr9S3S jr:S3S jr;S3S jr<S3S jr=\:\<\!\#\;\R|                  " \6SS9\R|                  " \6SS9\=S .r?S!\@S"'    " S# S$5      rA " S% S&\A5      rB " S' S(\A5      rC " S) S*\A5      rD\C\R|                  " \BSS9\R|                  " \BSS9\D\R|                  " \BSS9S+.rES,\@S-'    S4       S5S. jjrFg! \4 a     SSK.r. N! \4 a    Sr+  Nf = ff = f)6zAuthentication helpers.    )annotationsN)standard_b64decodestandard_b64encode)TYPE_CHECKINGAnyCallableMappingMutableMappingOptionalcast)quote)Binary)MongoCredential_authenticate_scram_start_parse_scram_response_xor)ConfigurationErrorOperationFailure)saslprep)_authenticate_aws)_authenticate_oidc_get_authenticator)_getaddrinfo)Hello)
ConnectionTF.   )r      c                   U R                   nUS:X  a7  Sn[        R                  n[        U R                  5      R                  S5      nO6Sn[        R                  n[        X0R                  5      R                  S5      nU R                  nU R                  n[        R                  n	UR                  n
U
(       aV  U
R                  5       (       aA  [        U
[        5      (       d   eU
R                   c   eU
R                   u  pU
R"                  nO[%        X5      u  pnUR'                  X~5      nUc   eUS   n[)        U5      n[+        US   5      nUS:  a  [-        S	5      eUS
   nUS   nUR/                  U5      (       d  [-        S5      eSU-   nUR0                  (       a  UR0                  u  nnnnOSu  nnnnU(       a  UU:w  d  UU:w  a\  [        R2                  " XF[5        U5      U5      nU	" USU5      R7                  5       nU	" USU5      R7                  5       nUUUU4Ul        U" U5      R7                  5       nSR9                  XU45      nU	" UUU5      R7                  5       nS[;        [=        UU5      5      -   nSR9                  UU45      n[;        U	" UUU5      R7                  5       5      nSUS   [?        U5      S.nUR'                  X~5      n[)        US   5      n[        R@                  " US   U5      (       d  [-        S5      eUS   (       d9  SUS   [?        S5      S.nUR'                  X~5      nUS   (       d  [-        S5      egg)zAuthenticate using SCRAM.SCRAM-SHA-256sha256utf-8sha1Npayload   ii   z+Server returned an invalid iteration count.   s   rz!Server returned an invalid nonce.s	   c=biws,r=)NNNNs
   Client Keys
   Server Key   ,s   p=   conversationIdsaslContinuer*   r$      vz%Server returned an invalid signature.done    z%SASL conversation failed to complete.)!usernamehashlibr!   r   passwordencoder#   _password_digestsourcecachehmacHMACauth_ctxspeculate_succeeded
isinstance_ScramContext
scram_dataspeculative_authenticater   commandr   intr   
startswithdatapbkdf2_hmacr   digestjoinr   r   r   compare_digest) credentialsconn	mechanismr0   rD   	digestmodrB   r5   r6   _hmacctxnonce
first_barerescmdserver_firstparsed
iterationssaltrnoncewithout_proof
client_key
server_keycsaltciterationssalted_pass
stored_keyauth_msg
client_sigclient_proofclient_final
server_sigs                                    ]C:\Suresh\moveshuttle\MDcreated\moveengine\venv\Lib\site-packages\pymongo/synchronous/auth.py_authenticate_scramrc   G   s4   ##HO#NN	,,-44W=LL	*>*>?FFwOFE IIE
--C
s&&((#}----~~)))NN**!:;!R3ll6'??y>L"<0FVD\"JDLMM$<DD\FU##BCC 6)Mzz5:ZZ2
J{5K2
J{ **C))&8J48PR\];yAHHJ
;yAHHJ
 *dJ?
:&--/Jyy*MBCHz8Y7>>@J-d:z.JKKL99m\:;L#E*h	$J$Q$Q$STJ ./,'C
 ,,v
#C"3y>2Fvd|Z88FGG v;!"23c{

 ll6'6{"#JKK  r/   c                f   [        U[        5      (       d  [        S5      e[        U5      S:X  a  [	        S5      e[        U [        5      (       d  [        S[        U 5       35      e[        R                  " 5       nU  SU 3nUR                  UR                  S5      5        UR                  5       $ )z0Get a password digest to use for authentication.z#password must be an instance of strr   zpassword can't be emptyz)username must be an instance of str, not z:mongo:r"   )r;   str	TypeErrorlen
ValueErrortyper1   md5updater3   	hexdigest)r0   r2   md5hashrB   s       rb   r4   r4      s    h$$=>>
8}233h$$CDNCSTUUkkmGZwxj)DNN4;;w'(r/   c                    [        X5      n[        R                  " 5       nU  U U 3nUR                  UR	                  S5      5        UR                  5       $ )z*Get an auth key to use for authentication.r"   )r4   r1   rj   rk   r3   rl   )rM   r0   r2   rD   rm   rB   s         rb   	_auth_keyro      sM    h1FkkmGWXJvh'DNN4;;w'(r/   c           	     f   US;   a  U $ [        U SSS[        R                  [        R                  S9S   u  p#pEnUS:X  a  UR	                  5       $  [        R
                  " U[        R                  5      nUS   R	                  5       $ ! [        R                   a    UR	                  5       s $ f = f)z2Canonicalize hostname following MIT-krb5 behavior.)FnoneNr   )familyri   protoflagsforward)r   socketIPPROTO_TCPAI_CANONNAMElowergetnameinfoNI_NAMEREQDgaierror)hostnameoptionafsocktypers   	canonnamesockaddrnames           rb   _canonicalize_hostnamer      s       	$$%%	
 	0	,B%H   !!!(F,>,>? 7==? ?? !  !s   %B	 	$B0/B0c                    [         (       d  [        S5      e U R                  nU R                  nU R                  nUR
                  =(       d    UR                  S   n[        XTR                  5      nUR                  S-   U-   nUR                  b  US-   UR                  -   nUb  [        (       aK  SR                  [        U5      [        U5      45      n[        R                  " Xg[        R                   S9u  pOlSU;   a  UR#                  SS5      u  pOUSp[        R                  " U[        R                   U
UUS9u  pO%[        R                  " U[        R                   S9u  pU[        R$                  :w  a  ['        S	5      e [        R(                  " U	S
5      S:w  a  ['        S5      e[        R*                  " U	5      nSSUSS.nUR-                  SU5      n[/        S5       H  n[        R(                  " U	[1        US   5      5      nUS:X  a  ['        S5      e[        R*                  " U	5      =(       d    S
nSUS   US.nUR-                  SU5      nU[        R$                  :X  d  M    O   ['        S5      e[        R2                  " U	[1        US   5      5      S:w  a  ['        S5      e[        R4                  " U	[        R*                  " U	5      U5      S:w  a  ['        S5      e[        R*                  " U	5      nSUS   US.nUR-                  SU5        [        R6                  " U	5        g! [        R6                  " U	5        f = f! [        R8                   a  n['        [1        U5      5      SeSnAff = f)zAuthenticate using GSSAPI.zEThe "kerberos" module must be installed to use GSSAPI authentication.r   @N:)gssflagsr)   )r   userdomainr2   z&Kerberos context failed to initialize. z*Unknown kerberos failure in step function.GSSAPI	saslStartrI   r$   autoAuthorize	$external
   r$   r*   r+   z+Kerberos authentication failed to complete.z0Unknown kerberos failure during GSS_Unwrap step.z.Unknown kerberos failure during GSS_Wrap step.)HAVE_KERBEROSr   r0   r2   mechanism_propertiesservice_hostaddressr   canonicalize_host_nameservice_nameservice_realm_USE_PRINCIPALrE   r   kerberosauthGSSClientInitGSS_C_MUTUAL_FLAGsplitAUTH_GSS_COMPLETEr   authGSSClientStepauthGSSClientResponser?   rangere   authGSSClientUnwrapauthGSSClientWrapauthGSSClientCleanKrbError)rG   rH   r0   r2   propshostservice	principalresultrL   r   r   r$   rP   response_excs                    rb   _authenticate_gssapir      s;   = S
 	
d3''''00 !!4T\\!_%d,H,HI$$s*T1*me&9&99G~  HHeHouX%GH	&881K1K (?#+>>#q#9LD&#+T&&88%77!% #44WxGaGabKFX///"#KLL:	-
 ))#r2a7&'STT 44S9G%"!"	C ||K5H 2Y!33CXi=P9QRR<*+WXX"88=C %&&./?&@&
  <<S9X777 " ''TUU ++CXi5H1IJaO&'YZZ))#x/M/Mc/RT\]abb&'WXX44S9G !"*+;"<"C
 LLc* '',H'', 3s3x(d23s8   E0M 	C#L4 0B-L4 M 4MM M=#M88M=c                    U R                   nU R                  nU R                  nSU SU 3R                  5       nSS[	        U5      SS.nUR                  X&5        g)z(Authenticate using SASL PLAIN (RFC 4616) r)   PLAINr   N)r5   r0   r2   r3   r   r?   )rG   rH   r5   r0   r2   r$   rP   s          rb   _authenticate_plainr   =  sc    F##H##HhZtH:.668G'?	C 	LLr/   c                    UR                   nU(       a  UR                  5       (       a  g[        XR                  5      R	                  5       nUR                  SU5        g)z Authenticate using MONGODB-X509.Nr   )r9   r:   _X509Contextr   speculate_commandr?   )rG   rH   rL   rP   s       rb   _authenticate_x509r   L  sD    
--C
s&&((
{LL
1
C
C
ECLLc"r/   c                T   UR                   S:  a  UR                  (       a  UR                  nOQU R                  nUR                  5       nUS-   U R                  -   US'   UR                  X4SS9R                  S/ 5      nSU;   a  [        XS5      $ [        XS5      $ [        XS5      $ )N   r   saslSupportedMechsF)publish_eventsr    SCRAM-SHA-1)max_wire_versionnegotiated_mechsr5   	hello_cmdr0   r?   getrc   )rG   rH   mechsr5   rP   s        rb   _authenticate_defaultr   W  s    !  ))E ''F.."C(.{7K7K(KC$%\\&e\DIIJ^`bcEe#&{/JJ&{-HH";mDDr/   r   )rI   r    )r   MONGODB-X509zMONGODB-AWSMONGODB-OIDCr   r   r    DEFAULTz!Mapping[str, Callable[..., None]]	_AUTH_MAPc                  \    \ rS rSrS	S jr\      S
S j5       rSS jrSS jrSS jr	Sr
g)_AuthContextit  c                *    Xl         S U l        X l        g N)rG   r>   r   )selfrG   r   s      rb   __init___AuthContext.__init__u  s    &EI%r/   c                |    [         R                  U R                  5      nU(       a  [        [        U" X5      5      $ g r   )_SPECULATIVE_AUTH_MAPr   rI   r   r   )credsr   spec_clss      rb   from_credentials_AuthContext.from_credentialsz  s0     ),,U__=hu&>??r/   c                    [         er   )NotImplementedErrorr   s    rb   r   _AuthContext.speculate_command  s    !!r/   c                &    UR                   U l         g r   )r>   )r   hellos     rb   parse_response_AuthContext.parse_response  s    (-(F(F%r/   c                ,    [        U R                  5      $ r   )boolr>   r   s    rb   r:    _AuthContext.speculate_succeeded  s    D1122r/   )r   rG   r>   N)rG   r   r   tuple[str, int]returnNone)r   r   r   r   r   zOptional[_AuthContext]r   z"Optional[MutableMapping[str, Any]])r   zHello[Mapping[str, Any]]r   r   )r   r   )__name__
__module____qualname____firstlineno__r   staticmethodr   r   r   r:   __static_attributes__ r/   rb   r   r   t  sC    
 )8	 "G3r/   r   c                  F   ^  \ rS rSr        SU 4S jjrSS jrSrU =r$ )r<   i  c                >   > [         TU ]  X5        S U l        X0l        g r   )superr   r=   rI   )r   rG   r   rI   	__class__s       rb   r   _ScramContext.__init__  s     	.9="r/   c                    [        U R                  U R                  5      u  pnU R                  R                  US'   X4U l        U$ Ndb)r   rG   rI   r5   r=   )r   rM   rN   rP   s       rb   r   _ScramContext.speculate_command  sA    !:4;K;KT^^!\3$$++D	 -
r/   )rI   r=   )rG   r   r   r   rI   re   r   r   r   )r   r   r   r   r   r   r   __classcell__)r   s   @rb   r<   r<     s2    #*#5D#QT#	# r/   r<   c                      \ rS rSrSS jrSrg)r   i  c                p    SSS.nU R                   R                  b  U R                   R                  US'   U$ )Nr)   r   )authenticaterI   r   )rG   r0   )r   rP   s     rb   r   _X509Context.speculate_command  s8     ~>$$0**33CK
r/   r   N)r   zMutableMapping[str, Any]r   r   r   r   r   r   r   r/   rb   r   r     s    r/   r   c                      \ rS rSrSS jrSrg)_OIDCContexti  c                    [        U R                  U R                  5      nUR                  5       nUc  g U R                  R                  US'   U$ r   )r   rG   r   get_spec_auth_cmdr5   )r   authenticatorrP   s      rb   r   _OIDCContext.speculate_command  sH    *4+;+;T\\J--/;$$++D	
r/   r   Nr   r   r   r/   rb   r   r     s    r/   r   )r   r   r    r   r   zMapping[str, Any]r   c                d    U R                   n[        U   nUS:X  a  [        XU5        gU" X5        g)zAuthenticate connection.r   N)rI   r   r   )rG   rH   reauthenticaterI   	auth_funcs        rb   r   r     s3     %%I)$IN";n=+$r/   )rG   r   rH   r   rI   re   r   r   )r0   re   r2   re   r   re   )rM   re   r0   re   r2   re   r   re   )r}   re   r~   z
str | boolr   re   )rG   r   rH   r   r   r   )F)rG   r   rH   r   r   r   r   r   )G__doc__
__future__r   	functoolsr1   r7   rv   base64r   r   typingr   r   r   r	   r
   r   r   urllib.parser   bson.binaryr   pymongo.auth_sharedr   r   r   r   pymongo.errorsr   r   pymongo.saslprepr   pymongo.synchronous.auth_awsr   pymongo.synchronous.auth_oidcr   r   pymongo.synchronous.helpersr   pymongo.hellor   pymongo.synchronous.poolr   r   r   winkerberosr   tuplemapr@   __version__r   ImportError_IS_SYNCrc   r4   ro   r   r   r   r   r   partialr   __annotations__r   r<   r   r   r   r   r   r/   rb   <module>r     s    "     9      @ % : 5#3	"Sh**005bq9:;vE PLf:k3\#E$ #&$& $$%8MR&&':oV$	0	, 	3 32L "< <  !$$]mL&&}P   /J, (  LQ	% 	%(2	%DH	%		%y   s*   5F F%FF!F% F!!F%