Only let active Serai validators connect over P2P

coordinator/src/p2p/mod.rs

@@ -1,12 +1,46 @@
-use serai_client::primitives::NetworkId;
+use std::{
+  sync::{Arc, RwLock},
+  collections::{HashSet, HashMap},
+};
 
+use serai_client::primitives::{NetworkId, PublicKey};
+
+use tokio::sync::mpsc;
+
+use futures_util::StreamExt;
+use libp2p::{
+  multihash::Multihash,
+  identity::PeerId,
+  swarm::{dial_opts::DialOpts, NetworkBehaviour, SwarmEvent, Swarm},
+};
+
+/// A struct to sync the validators from the Serai node in order to keep track of them.
+mod validators;
+
+/// The authentication protocol upgrade to limit the P2P network to active validators.
+mod authenticate;
+
+/// The dial task, to find new peers to connect to
+mod dial;
+
+/// The request-response messages and behavior
 mod reqres;
 use reqres::{Request, Response};
 
+/// The gossip messages and behavior
 mod gossip;
 
+/// The heartbeat task, effecting sync of Tributaries
 mod heartbeat;
 
+const PORT: u16 = 30563; // 5132 ^ (('c' << 8) | 'o')
+
+fn peer_id_from_public(public: PublicKey) -> PeerId {
+  // 0 represents the identity Multihash, that no hash was performed
+  // It's an internal constant so we can't refer to the constant inside libp2p
+  PeerId::from_multihash(Multihash::wrap(0, &public.0).unwrap()).unwrap()
+}
+
 struct Peer;
 impl Peer {
   async fn send(&self, request: Request) -> Result<Response, tokio::time::error::Elapsed> {
@@ -14,6 +48,11 @@ impl Peer {
   }
 }
 
+#[derive(Clone)]
+struct Peers {
+  peers: Arc<RwLock<HashMap<NetworkId, HashSet<PeerId>>>>,
+}
+
 #[derive(Clone, Debug)]
 struct P2p;
 impl P2p {
@@ -28,3 +67,9 @@ impl tributary::P2p for P2p {
     todo!("TODO")
   }
 }
+
+#[derive(NetworkBehaviour)]
+struct Behavior {
+  reqres: reqres::Behavior,
+  gossip: gossip::Behavior,
+}