1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
//! Benchmark utilities for the `cell` module.
pub use super::ClientLayer;
pub use super::CryptInit;
pub use super::InboundClientCrypt;
pub use super::InboundClientLayer;
pub use super::InboundRelayLayer;
pub use super::OutboundClientCrypt;
pub use super::OutboundClientLayer;
pub use super::OutboundRelayLayer;
pub use super::RelayCellBody;
pub use super::RelayLayer;
#[cfg(feature = "counter-galois-onion")]
pub use super::cgo::bench_utils as cgo;
pub use super::tor1::bench_utils as tor1;
use super::*;
/// The channel command used as additional data for the cryptographic operations benchmarks.
pub const BENCH_CHAN_CMD: ChanCmd = ChanCmd::RELAY;
impl InboundClientCrypt {
/// Helper method to add an inbound layer from a client layer pair.
pub fn add_layer_from_pair<F, B>(&mut self, pair: impl ClientLayer<F, B>)
where
F: OutboundClientLayer,
B: InboundClientLayer + Send + 'static,
{
let (_, inbound, _) = pair.split_client_layer();
self.add_layer(Box::new(inbound));
}
}
impl OutboundClientCrypt {
/// Helper method to add an outbound layer from a client layer pair.
pub fn add_layer_from_pair<F, B>(&mut self, pair: impl ClientLayer<F, B>)
where
F: OutboundClientLayer + Send + 'static,
B: InboundClientLayer,
{
let (outbound, _, _) = pair.split_client_layer();
self.add_layer(Box::new(outbound));
}
}
/// Encrypts the given `RelayCellBody` in the inbound direction by all the relays in a circuit.
pub fn circuit_encrypt_inbound<F, B>(
cmd: ChanCmd,
cell: &mut RelayCellBody,
relay_states: Vec<impl RelayLayer<F, B>>,
) where
F: OutboundRelayLayer,
B: InboundRelayLayer,
{
for (i, state) in relay_states.into_iter().rev().enumerate() {
let (_, mut inbound, _) = state.split_relay_layer();
if i == 0 {
inbound.originate(cmd, cell);
} else {
inbound.encrypt_inbound(cmd, cell);
}
}
}
|
