[CDN] Rate limiting - DDoS protection

Cargo.toml

@@ -2,7 +2,7 @@
 version = "0.5.79"                                        # same as `hotshot`, but workspace subcrate can also release its own version
 authors = ["Espresso Systems <hello@espressosys.com>"]
 edition = "2021"
-rust-version = "1.76.0"
+rust-version = "1.80.0"
 homepage = "https://github.com/EspressoSystems/HotShot"
 documentation = "https://hotshot.docs.espressosys.com"
 repository = "https://github.com/EspressoSystems/HotShot"

crates/examples/combined/all.rs

@@ -14,7 +14,7 @@ use cdn_broker::{reexports::def::hook::NoMessageHook, Broker};
 use cdn_marshal::Marshal;
 use hotshot::{
     helpers::initialize_logging,
-    traits::implementations::{KeyPair, TestingDef, WrappedSignatureKey},
+    traits::implementations::{HotShotMessageHook, KeyPair, TestingDef, WrappedSignatureKey},
     types::SignatureKey,
 };
 use hotshot_example_types::{node_types::TestVersions, state_types::TestTypes};
@@ -82,7 +82,7 @@ async fn main() {
                     private_key: broker_private_key.clone(),
                 },
 
-                user_message_hook: NoMessageHook,
+                user_message_hook: HotShotMessageHook::default(),
                 broker_message_hook: NoMessageHook,
 
                 metrics_bind_endpoint: None,

crates/examples/push-cdn/all.rs

@@ -17,7 +17,7 @@ use cdn_broker::{
 use cdn_marshal::Marshal;
 use hotshot::{
     helpers::initialize_logging,
-    traits::implementations::{TestingDef, WrappedSignatureKey},
+    traits::implementations::{HotShotMessageHook, TestingDef, WrappedSignatureKey},
     types::SignatureKey,
 };
 use hotshot_example_types::{node_types::TestVersions, state_types::TestTypes};
@@ -93,7 +93,7 @@ async fn main() {
                     private_key: broker_private_key.clone(),
                 },
 
-                user_message_hook: NoMessageHook,
+                user_message_hook: HotShotMessageHook::default(),
                 broker_message_hook: NoMessageHook,
 
                 metrics_bind_endpoint: None,

crates/examples/push-cdn/broker.rs

@@ -9,7 +9,9 @@
 use anyhow::Result;
 use cdn_broker::{reexports::def::hook::NoMessageHook, Broker, Config};
 use clap::Parser;
-use hotshot::traits::implementations::{KeyPair, ProductionDef, WrappedSignatureKey};
+use hotshot::traits::implementations::{
+    HotShotMessageHook, KeyPair, ProductionDef, WrappedSignatureKey,
+};
 use hotshot_example_types::node_types::TestTypes;
 use hotshot_types::traits::{node_implementation::NodeType, signature_key::SignatureKey};
 use sha2::Digest;
@@ -103,7 +105,7 @@ async fn main() -> Result<()> {
             private_key,
         },
 
-        user_message_hook: NoMessageHook,
+        user_message_hook: HotShotMessageHook::default(),
         broker_message_hook: NoMessageHook,
 
         public_bind_endpoint: args.public_bind_endpoint,

crates/hotshot/Cargo.toml

@@ -49,6 +49,7 @@ rand = { workspace = true }
 serde = { workspace = true, features = ["rc"] }
 sha2 = { workspace = true }
 time = { workspace = true }
+gcr = "0.1"
 
 tokio = { workspace = true }
 tracing = { workspace = true }

crates/hotshot/src/traits.rs

@@ -24,8 +24,10 @@ pub mod implementations {
         },
         memory_network::{MasterMap, MemoryNetwork},
         push_cdn_network::{
-            CdnMetricsValue, KeyPair, ProductionDef, PushCdnNetwork, TestingDef, Topic as CdnTopic,
-            WrappedSignatureKey,
+            definition::message_hook::HotShotMessageHook,
+            definition::signature_key::WrappedSignatureKey, definition::ProductionDef,
+            definition::TestingDef, definition::Topic as CdnTopic, metrics::CdnMetricsValue,
+            KeyPair, PushCdnNetwork,
         },
     };
 }