o ÞÜgP(ã@s¬dZddlZddlmZmZddlmZddlmZm Z m Z m Z ddl m Z ddlmZedd ƒ\ZZZZZd d „edd ƒDƒ\ZZZZZGd d „d ƒZGdd„deƒZdS)zÒ Variant on `KexGroup1 ` where the prime "p" and generator "g" are provided by the server. A bit more work is required on the client side, and a **lot** more on the server side. éN)Úsha1Úsha256)Úutil)ÚDEBUGÚbyte_chrÚbyte_ordÚ byte_mask)ÚMessage)Ú SSHExceptionéé#cCsg|]}t|ƒ‘qS©)r)Ú.0Úcr r úJ/var/www/html/api-tag/env/lib/python3.10/site-packages/paramiko/kex_gex.pyÚ 0src@sjeZdZdZdZdZdZeZdd„Z ddd „Z d d „Z d d „Z dd„Z dd„Zdd„Zdd„Zdd„ZdS)ÚKexGexz"diffie-hellman-group-exchange-sha1éé icCs4||_d|_d|_d|_d|_d|_d|_d|_dS)NF)Ú transportÚpÚqÚgÚxÚeÚfÚ old_style)Úselfrr r rÚ__init__;s zKexGex.__init__FcCsŒ|jjr |j tt¡dStƒ}|r!| t¡| |j ¡d|_ n| t ¡| |j ¡| |j ¡| |j ¡|j |¡|j t¡dS)NT)rÚ server_modeÚ_expect_packetÚ_MSG_KEXDH_GEX_REQUESTÚ_MSG_KEXDH_GEX_REQUEST_OLDr Úadd_byteÚc_MSG_KEXDH_GEX_REQUEST_OLDÚadd_intÚpreferred_bitsrÚc_MSG_KEXDH_GEX_REQUESTÚmin_bitsÚmax_bitsÚ _send_messageÚ_MSG_KEXDH_GEX_GROUP)rÚ_test_old_styleÚmr r rÚ start_kexEs ÿ       zKexGex.start_kexcCsp|tkr | |¡S|tkr| |¡S|tkr| |¡S|tkr$| |¡S|tkr-|  |¡Sd}t |  |j |¡ƒ‚)Nz*KexGex {} asked to handle packet type {:d}) r!Ú_parse_kexdh_gex_requestr+Ú_parse_kexdh_gex_groupÚ_MSG_KEXDH_GEX_INITÚ_parse_kexdh_gex_initÚ_MSG_KEXDH_GEX_REPLYÚ_parse_kexdh_gex_replyr"Ú_parse_kexdh_gex_request_oldr ÚformatÚname)rÚptyper-Úmsgr r rÚ parse_next\s     zKexGex.parse_nextcCs¢|jdd}t |d¡}t|dƒ}t|ƒ}d}|d@s)|dK}|dL}|d@r t |¡}t|d|ƒ|dd…}t |d¡}|dkrK||krKnq*||_ dS)Nééréÿé€) rrÚ deflate_longrÚlenÚosÚurandomrÚ inflate_longr)rrÚqnormÚqhbyteÚ byte_countÚqmaskÚx_bytesrr r rÚ _generate_xls"  þ  û zKexGex._generate_xcCsî| ¡}| ¡}| ¡}||jkr|j}||jkr|j}||kr"|}||kr(|}||_||_||_|j ¡}|dur>tdƒ‚|j td  |||¡¡|  |||¡\|_ |_ t ƒ}| t¡| |j ¡| |j ¡|j |¡|j t¡dS)Nú-Can't do server-side gex with no modulus packzPicking p ({} <= {} <= {} bits))Úget_intr)r(r&rÚ_get_modulus_packr Ú_logrr6Ú get_modulusrrr r#Úc_MSG_KEXDH_GEX_GROUPÚ add_mpintr*r r1)rr-ÚminbitsÚ preferredbitsÚmaxbitsÚpackr r rr/~s<   ÿþ    zKexGex._parse_kexdh_gex_requestcCsÈ| ¡|_|j|jkr|j|_|j|jkr|j|_|j ¡}|dur&tdƒ‚|j td  |j¡¡|  |j|j|j¡\|_ |_ t ƒ}| t¡| |j ¡| |j ¡|j |¡|j t¡d|_dS)NrJzPicking p (~ {} bits)T)rKr&r)r(rrLr rMrr6rNrrr r#rOrPr*r r1r)rr-rTr r rr5¤s*     ÿ  ÿ      z#KexGex._parse_kexdh_gex_request_oldcCs¦| ¡|_| ¡|_t |j¡}|dks|dkrtd |¡ƒ‚|j t d |¡¡|  ¡t |j|j |jƒ|_ tƒ}| t¡| |j ¡|j |¡|j t¡dS)Nrrzs0    ú úl