varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-10 23:20:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add Windows compatibility using coincurve

Browse Source
This commit is contained in:
Paschal
2025-02-23 12:47:00 +01:00
committed by Paul Robinson
parent 479b12f64d
commit 3bdbde57c7
8 changed files with 175 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
libp2p/crypto/key_exchange.py
View File

@ -1,11 +1,26 @@
import sys
from typing import (
Callable,
cast,
)
from fastecdsa.encoding import (
util,
)
if sys.platform != "win32":
from fastecdsa.encoding import (
util,
)
int_bytelen = util.int_bytelen
else:
from math import (
ceil,
log2,
)
def int_bytelen(n: int) -> int:
if n == 0:
return 1
return ceil(log2(abs(n) + 1) / 8)
from libp2p.crypto.ecc import (
ECCPrivateKey,
@ -18,8 +33,6 @@ from libp2p.crypto.keys import (
SharedKeyGenerator = Callable[[bytes], bytes]
int_bytelen = util.int_bytelen
def create_ephemeral_key_pair(curve_type: str) -> tuple[PublicKey, SharedKeyGenerator]:
"""Facilitates ECDH key exchange."""
@ -32,9 +45,15 @@ def create_ephemeral_key_pair(curve_type: str) -> tuple[PublicKey, SharedKeyGene
private_key = cast(ECCPrivateKey, key_pair.private_key)
remote_point = ECCPublicKey.from_bytes(serialized_remote_public_key, curve_type)
secret_point = remote_point.impl * private_key.impl
secret_x_coordinate = secret_point.x
byte_size = int_bytelen(secret_x_coordinate)
return secret_x_coordinate.to_bytes(byte_size, byteorder="big")
if sys.platform != "win32":
secret_point = remote_point.impl * private_key.impl
secret_x_coordinate = secret_point.x
byte_size = int_bytelen(secret_x_coordinate)
return secret_x_coordinate.to_bytes(byte_size, byteorder="big")
else:
# Windows implementation using coincurve
shared_key = private_key.impl.ecdh(remote_point.impl.public_key)
return shared_key
return key_pair.public_key, _key_exchange