mirror of
https://github.com/logos-co/nomos-specs.git
synced 2025-02-02 18:44:11 +00:00
74 lines
1.6 KiB
Python
74 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
|
|
import dacite
|
|
import yaml
|
|
from pysphinx.sphinx import X25519PrivateKey
|
|
|
|
from mixnet.config import NodeConfig
|
|
|
|
|
|
@dataclass
|
|
class Config:
|
|
simulation: SimulationConfig
|
|
logic: LogicConfig
|
|
mixnet: MixnetConfig
|
|
|
|
@classmethod
|
|
def load(cls, yaml_path: str) -> Config:
|
|
with open(yaml_path, "r") as f:
|
|
data = yaml.safe_load(f)
|
|
config = dacite.from_dict(data_class=Config, data=data)
|
|
|
|
# Validations
|
|
config.simulation.validate()
|
|
config.logic.validate()
|
|
config.mixnet.validate()
|
|
|
|
return config
|
|
|
|
|
|
@dataclass
|
|
class SimulationConfig:
|
|
duration_sec: int
|
|
net_latency_sec: float
|
|
|
|
def validate(self):
|
|
assert self.duration_sec > 0
|
|
assert self.net_latency_sec > 0
|
|
|
|
|
|
@dataclass
|
|
class LogicConfig:
|
|
lottery_interval_sec: float
|
|
sender_prob: float
|
|
|
|
def validate(self):
|
|
assert self.lottery_interval_sec > 0
|
|
assert self.sender_prob > 0
|
|
|
|
|
|
@dataclass
|
|
class MixnetConfig:
|
|
num_nodes: int
|
|
transmission_rate_per_sec: int
|
|
peering_degree: int
|
|
max_mix_path_length: int
|
|
|
|
def validate(self):
|
|
assert self.num_nodes > 0
|
|
assert self.transmission_rate_per_sec > 0
|
|
assert self.peering_degree > 0
|
|
assert self.max_mix_path_length > 0
|
|
|
|
def node_configs(self) -> list[NodeConfig]:
|
|
return [
|
|
NodeConfig(
|
|
X25519PrivateKey.generate(),
|
|
self.peering_degree,
|
|
self.transmission_rate_per_sec,
|
|
)
|
|
for _ in range(self.num_nodes)
|
|
]
|