o n~b@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)datavaluerH/usr/local/lib/python3.10/dist-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-ctrZbcryptz#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 bytearrayrappendjoinziprZMODE_CTRdecrypt)rpasswordZ ciphernameZkdfnameZ kdfoptionsZnumber_of_keys_Z encryptedZ decryptedsaltZ iterationsZ pwd_sha512ZstripesZconstantcountZ salt_sha512Zout_leoutaccresultcipherZ checkint1Z checkint2Zssh_namerrrimport_openssh_private_genericAsf                  rC)r ZCryptodome.CipherrZCryptodome.HashrZCryptodome.Protocol.KDFrZCryptodome.Util.strxorrZCryptodome.Util.py3compatrrrrrrrrCrrrrs