Mirror/Momentum-Firmware
1
0
Fork 0
mirror of https://github.com/Next-Flip/Momentum-Firmware.git synced 2026-05-13 07:48:35 -07:00
Code Issues Packages Projects Releases Wiki Activity

Format

Browse Source
This commit is contained in:
Willy-JL
2024-03-09 05:09:48 +00:00
parent fca5da1c42
commit 74c6440a27
1 changed files with 55 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

87
applications/system/findmy/generate_keys.py
View File

@@ -4,6 +4,7 @@ from cryptography.hazmat.primitives.asymmetric import ec
from cryptography.hazmat.backends import default_backend from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives import hashes from cryptography.hazmat.primitives import hashes
def advertisement_template(): def advertisement_template():
adv = "" adv = ""
adv += "1e" # length (30) adv += "1e" # length (30)
@@ -17,13 +18,17 @@ def advertisement_template():
adv += "00" # hint adv += "00" # hint
return bytearray.fromhex(adv) return bytearray.fromhex(adv)
def convert_key_to_hex(private_key, public_key): def convert_key_to_hex(private_key, public_key):
private_key_hex = private_key.private_numbers().private_value.to_bytes(28, byteorder='big').hex() private_key_hex = (
public_key_hex = public_key.public_numbers().x.to_bytes(28, byteorder='big').hex() private_key.private_numbers().private_value.to_bytes(28, byteorder="big").hex()
)
public_key_hex = public_key.public_numbers().x.to_bytes(28, byteorder="big").hex()
return private_key_hex, public_key_hex return private_key_hex, public_key_hex
def generate_mac_and_payload(public_key): def generate_mac_and_payload(public_key):
key = public_key.public_numbers().x.to_bytes(28, byteorder='big') key = public_key.public_numbers().x.to_bytes(28, byteorder="big")
addr = bytearray(key[:6]) addr = bytearray(key[:6])
addr[0] |= 0b11000000 addr[0] |= 0b11000000
@@ -34,56 +39,74 @@ def generate_mac_and_payload(public_key):
return addr.hex(), adv.hex() return addr.hex(), adv.hex()
def main(): def main():
nkeys = int(input('Enter the number of keys to generate: ')) nkeys = int(input("Enter the number of keys to generate: "))
prefix = input('Enter a name for the keyfiles (optional, press enter to skip): ') prefix = input("Enter a name for the keyfiles (optional, press enter to skip): ")
print() print()
if not os.path.exists('keys'): if not os.path.exists("keys"):
os.makedirs('keys') os.makedirs("keys")
for i in range(nkeys): for i in range(nkeys):
while True: while True:
private_key = ec.generate_private_key(ec.SECP224R1(), default_backend()) private_key = ec.generate_private_key(ec.SECP224R1(), default_backend())
public_key = private_key.public_key() public_key = private_key.public_key()
private_key_bytes = private_key.private_numbers().private_value.to_bytes(28, byteorder='big') private_key_bytes = private_key.private_numbers().private_value.to_bytes(
public_key_bytes = public_key.public_numbers().x.to_bytes(28, byteorder='big') 28, byteorder="big"
)
public_key_bytes = public_key.public_numbers().x.to_bytes(
28, byteorder="big"
)
private_key_b64 = base64.b64encode(private_key_bytes).decode("ascii") private_key_b64 = base64.b64encode(private_key_bytes).decode("ascii")
public_key_b64 = base64.b64encode(public_key_bytes).decode("ascii") public_key_b64 = base64.b64encode(public_key_bytes).decode("ascii")
private_key_hex, public_key_hex = convert_key_to_hex(private_key, public_key) private_key_hex, public_key_hex = convert_key_to_hex(
private_key, public_key
)
mac, payload = generate_mac_and_payload(public_key) mac, payload = generate_mac_and_payload(public_key)
public_key_hash = hashes.Hash(hashes.SHA256()) public_key_hash = hashes.Hash(hashes.SHA256())
public_key_hash.update(public_key_bytes) public_key_hash.update(public_key_bytes)
s256_b64 = base64.b64encode(public_key_hash.finalize()).decode("ascii") s256_b64 = base64.b64encode(public_key_hash.finalize()).decode("ascii")
if '/' not in s256_b64[:7]: if "/" not in s256_b64[:7]:
fname = f"{prefix}_{s256_b64[:7]}.keys" if prefix else f"{s256_b64[:7]}.keys" fname = (
f"{prefix}_{s256_b64[:7]}.keys"
if prefix
else f"{s256_b64[:7]}.keys"
)
print(f'{i + 1})') print(f"{i + 1})")
print('Private key (Base64):', private_key_b64) print("Private key (Base64):", private_key_b64)
print('Public key (Base64):', public_key_b64) print("Public key (Base64):", public_key_b64)
print('Hashed adv key (Base64):', s256_b64) print("Hashed adv key (Base64):", s256_b64)
print('---------------------------------------------------------------------------------') print(
print('Private key (Hex):', private_key_hex) "---------------------------------------------------------------------------------"
print('Public key (Hex):', public_key_hex) )
print('---------------------------------------------------------------------------------') print("Private key (Hex):", private_key_hex)
print('MAC:', mac) print("Public key (Hex):", public_key_hex)
print('Payload:', payload) print(
"---------------------------------------------------------------------------------"
)
print("MAC:", mac)
print("Payload:", payload)
print() print()
print('Place the .keys file onto your Flipper or input the MAC and Payload manually.') print(
"Place the .keys file onto your Flipper or input the MAC and Payload manually."
)
with open(f"keys/{fname}", 'w') as f: with open(f"keys/{fname}", "w") as f:
f.write(f'Private key: {private_key_b64}\n') f.write(f"Private key: {private_key_b64}\n")
f.write(f'Public key: {public_key_b64}\n') f.write(f"Public key: {public_key_b64}\n")
f.write(f'Hashed adv key: {s256_b64}\n') f.write(f"Hashed adv key: {s256_b64}\n")
f.write(f'Private key (Hex): {private_key_hex}\n') f.write(f"Private key (Hex): {private_key_hex}\n")
f.write(f'Public key (Hex): {public_key_hex}\n') f.write(f"Public key (Hex): {public_key_hex}\n")
f.write(f'MAC: {mac}\n') f.write(f"MAC: {mac}\n")
f.write(f'Payload: {payload}\n') f.write(f"Payload: {payload}\n")
break break
main()
main()