mirror of
https://github.com/logos-co/nomos-specs.git
synced 2025-02-08 05:25:50 +00:00
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import List, Tuple
|
|
|
|
from pysphinx.payload import Payload
|
|
from pysphinx.sphinx import SphinxPacket
|
|
|
|
from mixnet.config import MixMembership, NodeInfo
|
|
|
|
|
|
class SphinxPacketBuilder:
|
|
@staticmethod
|
|
def build(
|
|
message: bytes, membership: MixMembership, path_len: int
|
|
) -> Tuple[SphinxPacket, List[NodeInfo]]:
|
|
if path_len <= 0:
|
|
raise ValueError("path_len must be greater than 0")
|
|
if len(message) > Payload.max_plain_payload_size():
|
|
raise ValueError("message is too long")
|
|
|
|
route = membership.generate_route(path_len)
|
|
# We don't need the destination (defined in the Loopix Sphinx spec)
|
|
# because the last mix will broadcast the fully unwrapped message.
|
|
# Later, we will optimize the Sphinx according to our requirements.
|
|
dummy_destination = route[-1]
|
|
|
|
packet = SphinxPacket.build(
|
|
message,
|
|
route=[mixnode.sphinx_node() for mixnode in route],
|
|
destination=dummy_destination.sphinx_node(),
|
|
)
|
|
return (packet, route)
|