cronekorkn/bundlewrap
1
0
Fork 0
Code Issues Pull requests 8 Projects Releases Wiki Activity

format and comments

Browse source
This commit is contained in:
mwiegand 2022-02-15 16:45:43 +01:00
parent 1e4713cb3a
commit 1b5b354cc9
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
libs/ssh.py
View file

@ -6,18 +6,17 @@ from cryptography.hazmat.primitives.serialization import Encoding, PrivateFormat
def generate_ed25519_key_pair(secret): def generate_ed25519_key_pair(secret):
privkey_bytes = Ed25519PrivateKey.from_private_bytes(secret)
# PRIVATE KEY # PRIVATE KEY
privkey_bytes = Ed25519PrivateKey.from_private_bytes(secret)
nondeterministic_privatekey = privkey_bytes.private_bytes( nondeterministic_privatekey = privkey_bytes.private_bytes(
encoding=Encoding.PEM, encoding=Encoding.PEM,
format=PrivateFormat.OpenSSH, format=PrivateFormat.OpenSSH,
encryption_algorithm=NoEncryption() encryption_algorithm=NoEncryption()
).decode() ).decode()
# handle random 32bit number, occuring twice in a row # get relevant lines from string
nondeterministic_bytes = b64decode(''.join(nondeterministic_privatekey.split('\n')[1:-2])) nondeterministic_bytes = b64decode(''.join(nondeterministic_privatekey.split('\n')[1:-2]))
# sanity check # sanity check
@ -28,6 +27,7 @@ def generate_ed25519_key_pair(secret):
random_bytes = sha3_224(secret).digest()[0:4] random_bytes = sha3_224(secret).digest()[0:4]
deterministic_bytes = nondeterministic_bytes[:98] + random_bytes + random_bytes + nondeterministic_bytes[106:] deterministic_bytes = nondeterministic_bytes[:98] + random_bytes + random_bytes + nondeterministic_bytes[106:]
# reassemble file
deterministic_privatekey = '\n'.join([ deterministic_privatekey = '\n'.join([
'-----BEGIN OPENSSH PRIVATE KEY-----', '-----BEGIN OPENSSH PRIVATE KEY-----',
b64encode(deterministic_bytes).decode(), b64encode(deterministic_bytes).decode(),
@ -41,4 +41,6 @@ def generate_ed25519_key_pair(secret):
format=PublicFormat.OpenSSH, format=PublicFormat.OpenSSH,
).decode() ).decode()
# RETURN
return (deterministic_privatekey, public_key) return (deterministic_privatekey, public_key)