logos-co/nomos-simulations
1
0
Fork 0
mirror of https://github.com/logos-co/nomos-simulations.git synced 2025-02-03 14:25:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Measure both data and cover message counts

Browse Source
This commit is contained in:
Youngjoon Lee 2024-11-08 15:56:26 +07:00
parent c220497ffb
commit 1562d5e922
No known key found for this signature in database
GPG Key ID: 25CA11F37F095E5D
2 changed files with 21 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
simlib/mixnet-sims/src/node/mix/mod.rs
View File

@ -190,8 +190,8 @@ impl MixNode {
state: MixnodeState {
node_id: id,
step_id: 0,
data_messages_generated: HashMap::new(),
data_messages_fully_unwrapped: HashMap::new(),
messages_generated: HashMap::new(),
messages_fully_unwrapped: HashMap::new(),
},
data_msg_lottery_update_time_sender,
data_msg_lottery_interval,
@ -268,11 +268,12 @@ impl Node for MixNode {
let waker = futures::task::noop_waker();
let mut cx = futures::task::Context::from_waker(&waker);
// Generate a data message probabilistically
if let Poll::Ready(Some(_)) = pin!(&mut self.data_msg_lottery_interval).poll_next(&mut cx) {
if self.data_msg_lottery.run() {
let payload = Payload::new();
self.state
.data_messages_generated
.messages_generated
.insert(payload.id(), self.state.step_id);
let message = self
.crypto_processor
@ -281,9 +282,9 @@ impl Node for MixNode {
self.persistent_sender.send(message).unwrap();
}
}
// TODO: Generate cover message with probability
// Handle incoming messages
for network_message in self.receive() {
// println!(">>>>> Node {}, message: {message:?}", self.id);
self.forward(
network_message.payload().clone(),
Some(network_message.from),
@ -302,14 +303,23 @@ impl Node for MixNode {
MixOutgoingMessage::FullyUnwrapped(payload) => {
let payload = Payload::load(payload);
self.state
.data_messages_fully_unwrapped
.messages_fully_unwrapped
.insert(payload.id(), self.state.step_id);
//TODO: create a tracing event
}
}
}
if let Poll::Ready(Some(msg)) = pin!(&mut self.cover_traffic).poll_next(&mut cx) {
let message = self.crypto_processor.wrap_message(&msg).unwrap();
// Generate a cover message probabilistically
if let Poll::Ready(Some(_)) = pin!(&mut self.cover_traffic).poll_next(&mut cx) {
let payload = Payload::new();
self.state
.messages_generated
.insert(payload.id(), self.state.step_id);
let message = self
.crypto_processor
.wrap_message(payload.as_bytes())
.unwrap();
self.persistent_sender.send(message).unwrap();
}
@ -327,8 +337,7 @@ impl Node for MixNode {
match ward {
WardCondition::Max(_) => false,
WardCondition::Sum(condition) => {
*condition.step_result.borrow_mut() +=
self.state.data_messages_fully_unwrapped.len();
*condition.step_result.borrow_mut() += self.state.messages_fully_unwrapped.len();
false
}
}

4
simlib/mixnet-sims/src/node/mix/state.rs
View File

@ -16,8 +16,8 @@ pub struct MixnodeState {
#[serde(serialize_with = "serialize_node_id_as_index")]
pub node_id: NodeId,
pub step_id: usize,
pub data_messages_generated: HashMap<PayloadId, usize>,
pub data_messages_fully_unwrapped: HashMap<PayloadId, usize>,
pub messages_generated: HashMap<PayloadId, usize>,
pub messages_fully_unwrapped: HashMap<PayloadId, usize>,
}
#[derive(Serialize)]