feat: add secret-key command line option (#1946)

This commit is contained in:
Ryan Orendorff
2023-04-08 03:48:21 +01:00
committed by GitHub
parent 1bf5c7dce3
commit eca6dd01ae
11 changed files with 137 additions and 14 deletions

View File

@ -30,6 +30,11 @@ tracing = "0.1.37"
# crypto
rand = { version = "0.8", optional = true }
secp256k1 = { version = "0.26.0", features = [
"global-context",
"rand-std",
"recovery",
] }
# errors
thiserror = "1"

View File

@ -4,7 +4,8 @@ use reth_downloaders::{
bodies::bodies::BodiesDownloaderBuilder,
headers::reverse_headers::ReverseHeadersDownloaderBuilder,
};
use reth_network::{config::rng_secret_key, NetworkConfigBuilder, PeersConfig};
use reth_network::{NetworkConfigBuilder, PeersConfig};
use secp256k1::SecretKey;
use serde::{Deserialize, Serialize};
use std::path::PathBuf;
@ -25,15 +26,17 @@ impl Config {
&self,
nat_resolution_method: reth_net_nat::NatResolver,
peers_file: Option<PathBuf>,
secret_key: SecretKey,
) -> NetworkConfigBuilder {
let peer_config = self
.peers
.clone()
.with_basic_nodes_from_file(peers_file)
.unwrap_or_else(|_| self.peers.clone());
let discv4 =
Discv4Config::builder().external_ip_resolver(Some(nat_resolution_method)).clone();
NetworkConfigBuilder::new(rng_secret_key()).peer_config(peer_config).discovery(discv4)
NetworkConfigBuilder::new(secret_key).peer_config(peer_config).discovery(discv4)
}
}