mirror of
https://github.com/logos-co/nomos-specs.git
synced 2025-02-12 15:28:02 +00:00
71 lines
1.6 KiB
Python
71 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
import random
|
|
from dataclasses import dataclass
|
|
from typing import List
|
|
|
|
from cryptography.hazmat.primitives.asymmetric.x25519 import (
|
|
X25519PrivateKey,
|
|
X25519PublicKey,
|
|
)
|
|
from pysphinx.sphinx import Node as SphinxNode
|
|
|
|
|
|
@dataclass
|
|
class GlobalConfig:
|
|
"""
|
|
Global parameters used across all nodes in the network
|
|
"""
|
|
|
|
membership: MixMembership
|
|
transmission_rate_per_sec: int # Global Transmission Rate
|
|
# TODO: use this to make the size of Sphinx packet constant
|
|
max_mix_path_length: int
|
|
|
|
|
|
@dataclass
|
|
class NodeConfig:
|
|
"""
|
|
Node-specific parameters
|
|
"""
|
|
|
|
private_key: X25519PrivateKey
|
|
mix_path_length: int
|
|
nomssip: NomssipConfig
|
|
|
|
|
|
@dataclass
|
|
class NomssipConfig:
|
|
# The target number of peers a node should maintain in its p2p network
|
|
peering_degree: int
|
|
|
|
|
|
@dataclass
|
|
class MixMembership:
|
|
"""
|
|
A list of public information of nodes in the network.
|
|
We assume that this list is known to all nodes in the network.
|
|
"""
|
|
|
|
nodes: List[NodeInfo]
|
|
|
|
def generate_route(self, length: int) -> list[NodeInfo]:
|
|
"""
|
|
Generate a mix route for a Sphinx packet.
|
|
"""
|
|
return [random.choice(self.nodes) for _ in range(length)]
|
|
|
|
|
|
@dataclass
|
|
class NodeInfo:
|
|
"""
|
|
Public information of a node to be shared to all nodes in the network
|
|
"""
|
|
|
|
public_key: X25519PublicKey
|
|
|
|
def sphinx_node(self) -> SphinxNode:
|
|
# TODO: Use a pre-signed incentive tx, instead of NodeAddress
|
|
dummy_node_addr = bytes(32)
|
|
return SphinxNode(self.public_key, dummy_node_addr)
|