39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
from binascii import unhexlify
|
|
from librespot.common.Utils import Utils
|
|
import os
|
|
import struct
|
|
|
|
|
|
class DiffieHellman:
|
|
prime_bytes: bytearray = bytes([
|
|
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xc9, 0x0f, 0xda, 0xa2,
|
|
0x21, 0x68, 0xc2, 0x34, 0xc4, 0xc6, 0x62, 0x8b, 0x80, 0xdc, 0x1c, 0xd1,
|
|
0x29, 0x02, 0x4e, 0x08, 0x8a, 0x67, 0xcc, 0x74, 0x02, 0x0b, 0xbe, 0xa6,
|
|
0x3b, 0x13, 0x9b, 0x22, 0x51, 0x4a, 0x08, 0x79, 0x8e, 0x34, 0x04, 0xdd,
|
|
0xef, 0x95, 0x19, 0xb3, 0xcd, 0x3a, 0x43, 0x1b, 0x30, 0x2b, 0x0a, 0x6d,
|
|
0xf2, 0x5f, 0x14, 0x37, 0x4f, 0xe1, 0x35, 0x6d, 0x6d, 0x51, 0xc2, 0x45,
|
|
0xe4, 0x85, 0xb5, 0x76, 0x62, 0x5e, 0x7e, 0xc6, 0xf4, 0x4c, 0x42, 0xe9,
|
|
0xa6, 0x3a, 0x36, 0x20, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff
|
|
])
|
|
prime: int = int.from_bytes(prime_bytes, "big")
|
|
private_key: int
|
|
public_key: int
|
|
|
|
def __init__(self):
|
|
key_data = os.urandom(95)
|
|
self.private_key = int.from_bytes(key_data, "big")
|
|
self.public_key = pow(2, self.private_key, self.prime)
|
|
|
|
def compute_shared_key(self, remote_key_bytes: bytes):
|
|
remote_key = int.from_bytes(remote_key_bytes, "big")
|
|
return pow(remote_key, self.private_key, self.prime)
|
|
|
|
def private_key(self):
|
|
return self.private_key
|
|
|
|
def public_key(self):
|
|
return self.public_key
|
|
|
|
def public_key_array(self):
|
|
return Utils.to_byte_array(self.public_key)
|