o j@sxddlZddlmZddlmZddlmZddlmZddl m Z m Z m Z ddZ d d Zd d Zd dZddZdS)N)AES)SHA512) _bcrypt_hash)strxor)tostrbchrbordcCs<t|dkr tdtd|ddd}||ddfS)NzInsufficient data>Ir)len ValueErrorstructunpack)datavaluer/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/Cryptodome/PublicKey/_openssh.py read_int4(s rcCs8t|\}}t||krtd|d|||dfS)NzInsufficient data (V))rr r )rsizerrr read_bytes/s  rcCst|\}}t||fS)N)rr)rsdrrr read_string6s  rcCs2t|D]\}}t||dd@krtdqdS)NzIncorrect padding) enumeraterr )padvxrrr check_padding;s rcCsV|ds td|dd}t|\}}t|\}}t|\}}t|\}}|dkr/tdt|\}}t|\}}|rAtdt|ddkrMtd |d krT|}n||fd krbtd ||ft|\} }t|\} }t| d krxtd|r~tdt|} g} d} t ddD]Z}t| t d|}t | d|| d}t j dgt d|R}t|}t d| D]#}t | dt|| d}t j dgt d|R}t|||dq| |ddqdddt| D}tj|ddtjd|ddd}||}t|\}}t|\}}||kr!tdt|\}}||fS) Nsopenssh-key-v1zIncorrect magic valuerzWe only handle 1 key at a timez Too much datarzIncorrect payload lengthnone)z aes256-ctrbcryptz#Unsupported encryption scheme %s/%szIncorrect salt lengthzToo much data in kdfoptionss OxychromaticBlowfishSwatDynamiter Fz IIIIIIII)outputcSs g|] \}}t|t|qSr)r).0abrrr ys z2import_openssh_private_generic.. 0)nonce initial_valuezIncorrect checksum) startswithr rrrr rnewdigestranger packrr bytearrayrappendjoinziprMODE_CTRdecrypt)rpassword ciphernamekdfname kdfoptionsnumber_of_keys_ encrypted decryptedsalt iterations pwd_sha512stripesconstantcount salt_sha512out_leoutaccresultcipher checkint1 checkint2ssh_namerrrimport_openssh_private_genericAsf                  rT)r Cryptodome.CipherrCryptodome.HashrCryptodome.Protocol.KDFrCryptodome.Util.strxorrCryptodome.Util.py3compatrrrrrrrrTrrrrs