Make rate limits configureable.

Add 'rate_limit_count' and 'rate_limit_period' to configuration option.

config_spec.toml

@@ -71,3 +71,15 @@ name = "response_invoice_timeout"
 type = "u32"
 optional = true
 doc = "Amount of time in seconds that server waits for an offer creator to respond with an invoice. Defaults to 15s."
+
+[[param]]
+name = "rate_limit_count"
+type = "u8"
+default = "10"
+doc = "The number of calls each peer allowed to make per rate limit period."
+
+[[param]]
+name = "rate_limit_period"
+type = "u64"
+default = "1"
+doc = "The period in seconds over which peers are rate limited."

src/lib.rs

@@ -129,6 +129,8 @@ pub struct Cfg {
     pub lnd: LndCfg,
     pub signals: LifecycleSignals,
     pub skip_version_check: bool,
+    pub rate_limit_count: u8,
+    pub rate_limit_period: u64
 }
 
 #[derive(Clone)]
@@ -236,6 +238,8 @@ impl LndkOnionMessenger {
             onion_messenger,
             network,
             args.signals,
+            args.rate_limit_count,
+            args.rate_limit_period
         )
         .await
     }

src/main.rs

@@ -54,6 +54,8 @@ async fn main() -> Result<(), ()> {
         lnd: lnd_args,
         signals,
         skip_version_check: config.skip_version_check,
+        rate_limit_count: config.rate_limit_count,
+        rate_limit_period: config.rate_limit_period,
     };
 
     let response_invoice_timeout = config.response_invoice_timeout;

src/onion_messenger.rs

@@ -46,12 +46,6 @@ const ONION_MESSAGE_TYPE: u32 = 513;
 /// MSG_POLL_INTERVAL is the interval at which we poll for outgoing onion messages.
 const MSG_POLL_INTERVAL: Duration = Duration::from_millis(100);
 
-/// DEFAULT_CALL_COUNT is the default number of calls each peer gets per rate limited period.
-const DEFAULT_CALL_COUNT: u8 = 10;
-
-/// DEFAULT_CALL_FREQUENCY is the default period over which peers are rate limited.
-const DEFAULT_CALL_FREQUENCY: Duration = Duration::from_secs(1);
-
 /// Node Id LookUp is a utility struct implementing NodeIdLookUp trait for LDK's OnionMessenger.
 pub struct LndkNodeIdLookUp {
     client: Client,
@@ -167,6 +161,8 @@ impl LndkOnionMessenger {
         onion_messenger: OnionMessenger<ES, NS, L, NL, MR, OMH, CMH>,
         network: Network,
         signals: LifecycleSignals,
+        rate_limit_count: u8,
+        rate_limit_period: u64,
     ) -> Result<(), ()>
     where
         ES::Target: EntropySource,
@@ -273,8 +269,8 @@ impl LndkOnionMessenger {
         // events we need).
         let rate_limiter = &mut TokenLimiter::new(
             current_peers.keys().copied(),
-            DEFAULT_CALL_COUNT,
-            DEFAULT_CALL_FREQUENCY,
+            rate_limit_count,
+            Duration::from_secs(rate_limit_period),
             TokioClock::new(),
         );
         let mut message_sender = CustomMessenger {