Add beacon network crate (#715)

Author: ogenev

Cargo.lock

@@ -53,6 +53,7 @@ members = [
     "ethportal-peertest",
     "light-client",
     "rpc",
+    "trin-beacon",
     "trin-cli",
     "trin-history",
     "trin-state",

Cargo.toml

@@ -21,6 +21,7 @@ COPY ./portalnet ./portalnet
 COPY ./light-client ./light-client
 COPY ./rpc ./rpc
 COPY ./src ./src 
+COPY ./trin-beacon ./trin-beacon
 COPY ./trin-bridge ./trin-bridge
 COPY ./trin-cli ./trin-cli
 COPY ./trin-history ./trin-history 

docker/Dockerfile

@@ -18,6 +18,7 @@ COPY ./Cargo.toml ./Cargo.toml
 COPY ./portalnet ./portalnet
 COPY ./light-client ./light-client
 COPY ./src ./src 
+COPY ./trin-beacon ./trin-beacon
 COPY ./trin-bridge ./trin-bridge
 COPY ./trin-cli ./trin-cli
 COPY ./trin-history ./trin-history 

docker/Dockerfile.bridge

@@ -0,0 +1 @@
+Add beacon portal network crate

newsfragments/715.added.md

@@ -0,0 +1,28 @@
+[package]
+name = "trin-beacon"
+version = "0.1.0"
+edition = "2021"
+repository = "https://github.com/ethereum/trin/tree/master/trin-beacon"
+license = "GPL-3.0"
+readme = "README.md"
+keywords = ["ethereum", "portal-network"]
+categories = ["cryptography::cryptocurrencies"]
+description = "Beacon network subprotocol for Trin."
+authors = ["https://github.com/ethereum/trin/graphs/contributors"]
+
+
+[dependencies]
+anyhow = "1.0.68"
+async-trait = "0.1.53"
+discv5 = { version = "0.2.1", features = ["serde"]}
+ethportal-api = {path = "../ethportal-api"}
+eth2_ssz = "0.4.0"
+parking_lot = "0.11.2"
+portalnet = { path = "../portalnet" }
+serde_json = "1.0.89"
+tokio = {version = "1.14.0", features = ["full"]}
+tracing = "0.1.36"
+trin-types = { path = "../trin-types" }
+trin-validation = { path = "../trin-validation" }
+trin-utils = { path = "../trin-utils" }
+utp-rs = "0.1.0-alpha.4"

trin-beacon/Cargo.toml

@@ -0,0 +1,51 @@
+use crate::network::BeaconNetwork;
+use discv5::TalkRequest;
+use portalnet::types::messages::Message;
+use std::sync::Arc;
+use tokio::sync::mpsc::UnboundedReceiver;
+use tracing::{error, warn, Instrument};
+
+pub struct BeaconEvents {
+    pub network: Arc<BeaconNetwork>,
+    pub event_rx: UnboundedReceiver<TalkRequest>,
+}
+
+impl BeaconEvents {
+    pub async fn start(mut self) {
+        loop {
+            tokio::select! {
+                Some(talk_request) = self.event_rx.recv() => {
+                    self.handle_beacon_talk_request(talk_request);
+                }
+            }
+        }
+    }
+
+    /// Handle beacon network TalkRequest event
+    fn handle_beacon_talk_request(&self, talk_request: TalkRequest) {
+        let network = Arc::clone(&self.network);
+        let talk_request_id = talk_request.id().clone();
+        tokio::spawn(async move {
+            let reply = match network
+                .overlay
+                .process_one_request(&talk_request)
+                .instrument(tracing::info_span!("beacon_network", req = %talk_request_id))
+                .await
+            {
+                Ok(response) => Message::from(response).into(),
+                Err(error) => {
+                    error!(
+                        error = %error,
+                        request.discv5.id = %talk_request_id,
+                        "Error processing portal beacon request, responding with empty TALKRESP"
+                    );
+                    // Return an empty TALKRESP if there was an error executing the request
+                    "".into()
+                }
+            };
+            if let Err(error) = talk_request.respond(reply) {
+                warn!(error = %error, request.discv5.id = %talk_request_id, "Error responding to TALKREQ");
+            }
+        });
+    }
+}