Skip to content

Commit b7d2046

Browse files
committed
Use network name in Noise prologue.
To avoid accidentally connecting peers of different networks we now include the network name (e.g. "sailfish") in the Noise prologue. The handshake will fail, if differently named network peers try to connect.
1 parent 0ee031b commit b7d2046

File tree

3 files changed

+10
-6
lines changed

3 files changed

+10
-6
lines changed

cliquenet/src/net.rs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -872,6 +872,8 @@ where
872872
let h = Builder::new(NOISE_PARAMS.parse().expect("valid noise params"))
873873
.local_private_key(&self.keypair.secret_key().as_bytes())
874874
.expect("valid private key")
875+
.prologue(self.name.as_bytes())
876+
.expect("1st time we set the prologue")
875877
.build_responder()
876878
.expect("valid noise params yield valid handshake state");
877879
self.handshake_tasks.spawn(async move {
@@ -967,6 +969,8 @@ async fn connect<T: tcp::Stream + Unpin>(
967969
.expect("valid private key")
968970
.remote_public_key(to.1.as_slice())
969971
.expect("valid remote pub key")
972+
.prologue(name.as_bytes())
973+
.expect("1st time we set the prologue")
970974
.build_initiator()
971975
.expect("valid noise params yield valid handshake state")
972976
};

cliquenet/tests/frame-handling.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ async fn multiple_frames() {
3131

3232
let mut net_a = Overlay::new(
3333
Network::create(
34-
"f1",
34+
"frames",
3535
all_parties[0].2,
3636
party_a_sign.public_key(),
3737
party_a_dh,
@@ -43,7 +43,7 @@ async fn multiple_frames() {
4343
);
4444
let mut net_b = Overlay::new(
4545
Network::create(
46-
"f2",
46+
"frames",
4747
all_parties[1].2,
4848
party_b_sign.public_key(),
4949
party_b_dh,

tests/src/tests/rbc.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ fn mk_host<A, const N: usize>(
6060
let p = peers.clone();
6161
async move {
6262
let comm = Network::create_turmoil(
63-
"test",
63+
"rbc",
6464
a,
6565
k.public_key(),
6666
x.clone(),
@@ -114,7 +114,7 @@ fn small_committee() {
114114

115115
sim.client("C", async move {
116116
let addr = (UNSPECIFIED, ports[2]);
117-
let comm = Network::create_turmoil("C", addr, k.public_key(), x, peers, NetworkMetrics::default()).await?;
117+
let comm = Network::create_turmoil("rbc", addr, k.public_key(), x, peers, NetworkMetrics::default()).await?;
118118
let cfg = RbcConfig::new(k.clone(), c.id(), c.clone()).recover(false);
119119
let rbc = Rbc::new(10, Overlay::new(comm), cfg);
120120
let cons = Consensus::new(k, c, EmptyBlocks);
@@ -172,7 +172,7 @@ fn medium_committee() {
172172

173173
sim.client("E", async move {
174174
let addr = (UNSPECIFIED, ports[4]);
175-
let comm = Network::create_turmoil("E", addr, k.public_key(), x, peers, NetworkMetrics::default()).await?;
175+
let comm = Network::create_turmoil("rbc", addr, k.public_key(), x, peers, NetworkMetrics::default()).await?;
176176
let cfg = RbcConfig::new(k.clone(), c.id(), c.clone()).recover(false);
177177
let rbc = Rbc::new(10, Overlay::new(comm), cfg);
178178
let cons = Consensus::new(k, c, EmptyBlocks);
@@ -229,7 +229,7 @@ fn medium_committee_partition_network() {
229229

230230
sim.client("E", async move {
231231
let addr = (UNSPECIFIED, ports[4]);
232-
let comm = Network::create_turmoil("E", addr, k.public_key(), x, peers, NetworkMetrics::default()).await?;
232+
let comm = Network::create_turmoil("rbc", addr, k.public_key(), x, peers, NetworkMetrics::default()).await?;
233233
let cfg = RbcConfig::new(k.clone(), c.id(), c.clone()).recover(false);
234234
let rbc = Rbc::new(10, Overlay::new(comm), cfg);
235235
let cons = Consensus::new(k, c, EmptyBlocks);

0 commit comments

Comments
 (0)