astreum 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

astreum/node/crypto/ed25519.py

@@ -0,0 +1,50 @@
+# EdDSA signature algorithm over Curve25519
+
+# ed25519.py
+from cryptography.hazmat.primitives.asymmetric import ed25519
+from cryptography.hazmat.primitives.asymmetric.ed25519 import Ed25519PrivateKey, Ed25519PublicKey
+from cryptography.exceptions import InvalidSignature
+from typing import Tuple
+
+def generate_key_pair() -> Tuple[Ed25519PrivateKey, Ed25519PublicKey]:
+    """
+    Generate an Ed25519 private and public key pair.
+    
+    Returns:
+        Tuple[Ed25519PrivateKey, Ed25519PublicKey]: The generated key pair.
+    """
+    private_key: Ed25519PrivateKey = ed25519.Ed25519PrivateKey.generate()
+    public_key: Ed25519PublicKey = private_key.public_key()
+    return private_key, public_key
+
+def sign_message(private_key: Ed25519PrivateKey, message: bytes) -> bytes:
+    """
+    Sign a message using the provided Ed25519 private key.
+    
+    Args:
+        private_key (Ed25519PrivateKey): The private key used for signing.
+        message (bytes): The message to sign.
+    
+    Returns:
+        bytes: The signature.
+    """
+    signature: bytes = private_key.sign(message)
+    return signature
+
+def verify_signature(public_key: Ed25519PublicKey, message: bytes, signature: bytes) -> bool:
+    """
+    Verify a message signature using the provided Ed25519 public key.
+    
+    Args:
+        public_key (Ed25519PublicKey): The public key corresponding to the private key that signed.
+        message (bytes): The original message.
+        signature (bytes): The signature to verify.
+        
+    Returns:
+        bool: True if the signature is valid, False otherwise.
+    """
+    try:
+        public_key.verify(signature, message)
+        return True
+    except InvalidSignature:
+        return False

astreum/node/crypto/x25519.py

@@ -0,0 +1,31 @@
+# Diffie-Hellman key exchange over Curve25519
+
+# x25519.py
+from cryptography.hazmat.primitives.asymmetric import x25519
+from cryptography.hazmat.primitives.asymmetric.x25519 import X25519PrivateKey, X25519PublicKey
+from typing import Tuple
+
+def generate_key_pair() -> Tuple[X25519PrivateKey, X25519PublicKey]:
+    """
+    Generate an X25519 private and public key pair.
+    
+    Returns:
+        Tuple[X25519PrivateKey, X25519PublicKey]: The generated key pair.
+    """
+    private_key: X25519PrivateKey = x25519.X25519PrivateKey.generate()
+    public_key: X25519PublicKey = private_key.public_key()
+    return private_key, public_key
+
+def generate_shared_key(private_key: X25519PrivateKey, peer_public_key: X25519PublicKey) -> bytes:
+    """
+    Generate a shared key using the provided private key and peer's public key.
+    
+    Args:
+        private_key (X25519PrivateKey): Our private key.
+        peer_public_key (X25519PublicKey): The peer's public key.
+        
+    Returns:
+        bytes: The shared key.
+    """
+    shared_key: bytes = private_key.exchange(peer_public_key)
+    return shared_key

astreum/node/relay/__init__.py

@@ -96,6 +96,9 @@ class Relay:
         # Initialize route table with our node ID
         self.route_table = RouteTable(self)
 
+        # Initialize storage index
+        self.storage_index: Dict[bytes, bytes] = {}
+
         # Start worker threads
         self._start_workers()