1use crate::{
22 arg_enums::Database, error::Result, DatabaseParams, ImportParams, KeystoreParams,
23 NetworkParams, NodeKeyParams, OffchainWorkerParams, PruningParams, RpcEndpoint, SharedParams,
24 SubstrateCli,
25};
26use log::warn;
27use names::{Generator, Name};
28use sc_service::{
29 config::{
30 BasePath, Configuration, DatabaseSource, ExecutorConfiguration, IpNetwork, KeystoreConfig,
31 NetworkConfiguration, NodeKeyConfig, OffchainWorkerConfig, PrometheusConfig, PruningMode,
32 Role, RpcBatchRequestConfig, RpcConfiguration, RpcMethods, TelemetryEndpoints,
33 TransactionPoolOptions, WasmExecutionMethod,
34 },
35 BlocksPruning, ChainSpec, TracingReceiver,
36};
37use sc_tracing::logging::LoggerBuilder;
38use std::{num::NonZeroU32, path::PathBuf};
39
40pub(crate) const NODE_NAME_MAX_LENGTH: usize = 64;
42
43pub(crate) const DEFAULT_NETWORK_CONFIG_PATH: &str = "network";
45
46const RECOMMENDED_OPEN_FILE_DESCRIPTOR_LIMIT: u64 = 10_000;
48
49pub const RPC_DEFAULT_PORT: u16 = 9944;
51pub const RPC_DEFAULT_MAX_SUBS_PER_CONN: u32 = 1024;
53pub const RPC_DEFAULT_MAX_REQUEST_SIZE_MB: u32 = 15;
55pub const RPC_DEFAULT_MAX_RESPONSE_SIZE_MB: u32 = 15;
57pub const RPC_DEFAULT_MAX_CONNECTIONS: u32 = 100;
59pub const RPC_DEFAULT_MESSAGE_CAPACITY_PER_CONN: u32 = 64;
62
63pub trait DefaultConfigurationValues {
68 fn p2p_listen_port() -> u16 {
72 30333
73 }
74
75 fn rpc_listen_port() -> u16 {
79 RPC_DEFAULT_PORT
80 }
81
82 fn prometheus_listen_port() -> u16 {
86 9615
87 }
88}
89
90impl DefaultConfigurationValues for () {}
91
92pub trait CliConfiguration<DCV: DefaultConfigurationValues = ()>: Sized {
94 fn shared_params(&self) -> &SharedParams;
96
97 fn import_params(&self) -> Option<&ImportParams> {
99 None
100 }
101
102 fn pruning_params(&self) -> Option<&PruningParams> {
104 self.import_params().map(|x| &x.pruning_params)
105 }
106
107 fn keystore_params(&self) -> Option<&KeystoreParams> {
109 None
110 }
111
112 fn network_params(&self) -> Option<&NetworkParams> {
114 None
115 }
116
117 fn offchain_worker_params(&self) -> Option<&OffchainWorkerParams> {
119 None
120 }
121
122 fn node_key_params(&self) -> Option<&NodeKeyParams> {
124 self.network_params().map(|x| &x.node_key_params)
125 }
126
127 fn database_params(&self) -> Option<&DatabaseParams> {
129 self.import_params().map(|x| &x.database_params)
130 }
131
132 fn base_path(&self) -> Result<Option<BasePath>> {
136 self.shared_params().base_path()
137 }
138
139 fn is_dev(&self) -> Result<bool> {
143 Ok(self.shared_params().is_dev())
144 }
145
146 fn role(&self, _is_dev: bool) -> Result<Role> {
150 Ok(Role::Full)
151 }
152
153 fn transaction_pool(&self, _is_dev: bool) -> Result<TransactionPoolOptions> {
157 Ok(Default::default())
158 }
159
160 fn network_config(
166 &self,
167 chain_spec: &Box<dyn ChainSpec>,
168 is_dev: bool,
169 is_validator: bool,
170 net_config_dir: PathBuf,
171 client_id: &str,
172 node_name: &str,
173 node_key: NodeKeyConfig,
174 default_listen_port: u16,
175 ) -> Result<NetworkConfiguration> {
176 let network_config = if let Some(network_params) = self.network_params() {
177 network_params.network_config(
178 chain_spec,
179 is_dev,
180 is_validator,
181 Some(net_config_dir),
182 client_id,
183 node_name,
184 node_key,
185 default_listen_port,
186 )
187 } else {
188 NetworkConfiguration::new(node_name, client_id, node_key, Some(net_config_dir))
189 };
190
191 Ok(network_config)
196 }
197
198 fn keystore_config(&self, config_dir: &PathBuf) -> Result<KeystoreConfig> {
203 self.keystore_params()
204 .map(|x| x.keystore_config(config_dir))
205 .unwrap_or_else(|| Ok(KeystoreConfig::InMemory))
206 }
207
208 fn database_cache_size(&self) -> Result<Option<usize>> {
212 Ok(self.database_params().map(|x| x.database_cache_size()).unwrap_or_default())
213 }
214
215 fn database(&self) -> Result<Option<Database>> {
219 Ok(self.database_params().and_then(|x| x.database()))
220 }
221
222 fn database_config(
224 &self,
225 base_path: &PathBuf,
226 cache_size: usize,
227 database: Database,
228 ) -> Result<DatabaseSource> {
229 let role_dir = "full";
230 let rocksdb_path = base_path.join("db").join(role_dir);
231 let paritydb_path = base_path.join("paritydb").join(role_dir);
232 Ok(match database {
233 #[cfg(feature = "rocksdb")]
234 Database::RocksDb => DatabaseSource::RocksDb { path: rocksdb_path, cache_size },
235 Database::ParityDb => DatabaseSource::ParityDb { path: paritydb_path },
236 Database::ParityDbDeprecated => {
237 eprintln!(
238 "WARNING: \"paritydb-experimental\" database setting is deprecated and will be removed in future releases. \
239 Please update your setup to use the new value: \"paritydb\"."
240 );
241 DatabaseSource::ParityDb { path: paritydb_path }
242 },
243 Database::Auto => DatabaseSource::Auto { paritydb_path, rocksdb_path, cache_size },
244 })
245 }
246
247 fn trie_cache_maximum_size(&self) -> Result<Option<usize>> {
252 Ok(self.import_params().map(|x| x.trie_cache_maximum_size()).unwrap_or_default())
253 }
254
255 fn state_pruning(&self) -> Result<Option<PruningMode>> {
260 self.pruning_params()
261 .map(|x| x.state_pruning())
262 .unwrap_or_else(|| Ok(Default::default()))
263 }
264
265 fn blocks_pruning(&self) -> Result<BlocksPruning> {
270 self.pruning_params()
271 .map(|x| x.blocks_pruning())
272 .unwrap_or_else(|| Ok(BlocksPruning::KeepFinalized))
273 }
274
275 fn chain_id(&self, is_dev: bool) -> Result<String> {
279 Ok(self.shared_params().chain_id(is_dev))
280 }
281
282 fn node_name(&self) -> Result<String> {
286 Ok(generate_node_name())
287 }
288
289 fn wasm_method(&self) -> Result<WasmExecutionMethod> {
294 Ok(self.import_params().map(|x| x.wasm_method()).unwrap_or_default())
295 }
296
297 fn wasm_runtime_overrides(&self) -> Option<PathBuf> {
301 self.import_params().map(|x| x.wasm_runtime_overrides()).unwrap_or_default()
302 }
303
304 fn rpc_addr(&self, _default_listen_port: u16) -> Result<Option<Vec<RpcEndpoint>>> {
306 Ok(None)
307 }
308
309 fn rpc_methods(&self) -> Result<RpcMethods> {
314 Ok(Default::default())
315 }
316
317 fn rpc_max_connections(&self) -> Result<u32> {
319 Ok(RPC_DEFAULT_MAX_CONNECTIONS)
320 }
321
322 fn rpc_cors(&self, _is_dev: bool) -> Result<Option<Vec<String>>> {
326 Ok(Some(Vec::new()))
327 }
328
329 fn rpc_max_request_size(&self) -> Result<u32> {
331 Ok(RPC_DEFAULT_MAX_REQUEST_SIZE_MB)
332 }
333
334 fn rpc_max_response_size(&self) -> Result<u32> {
336 Ok(RPC_DEFAULT_MAX_RESPONSE_SIZE_MB)
337 }
338
339 fn rpc_max_subscriptions_per_connection(&self) -> Result<u32> {
341 Ok(RPC_DEFAULT_MAX_SUBS_PER_CONN)
342 }
343
344 fn rpc_buffer_capacity_per_connection(&self) -> Result<u32> {
346 Ok(RPC_DEFAULT_MESSAGE_CAPACITY_PER_CONN)
347 }
348
349 fn rpc_batch_config(&self) -> Result<RpcBatchRequestConfig> {
351 Ok(RpcBatchRequestConfig::Unlimited)
352 }
353
354 fn rpc_rate_limit(&self) -> Result<Option<NonZeroU32>> {
356 Ok(None)
357 }
358
359 fn rpc_rate_limit_whitelisted_ips(&self) -> Result<Vec<IpNetwork>> {
361 Ok(vec![])
362 }
363
364 fn rpc_rate_limit_trust_proxy_headers(&self) -> Result<bool> {
366 Ok(false)
367 }
368
369 fn prometheus_config(
373 &self,
374 _default_listen_port: u16,
375 _chain_spec: &Box<dyn ChainSpec>,
376 ) -> Result<Option<PrometheusConfig>> {
377 Ok(None)
378 }
379
380 fn telemetry_endpoints(
384 &self,
385 chain_spec: &Box<dyn ChainSpec>,
386 ) -> Result<Option<TelemetryEndpoints>> {
387 Ok(chain_spec.telemetry_endpoints().clone())
388 }
389
390 fn default_heap_pages(&self) -> Result<Option<u64>> {
394 Ok(None)
395 }
396
397 fn offchain_worker(&self, role: &Role) -> Result<OffchainWorkerConfig> {
401 self.offchain_worker_params()
402 .map(|x| x.offchain_worker(role))
403 .unwrap_or_else(|| Ok(OffchainWorkerConfig::default()))
404 }
405
406 fn force_authoring(&self) -> Result<bool> {
410 Ok(Default::default())
411 }
412
413 fn disable_grandpa(&self) -> Result<bool> {
417 Ok(Default::default())
418 }
419
420 fn dev_key_seed(&self, _is_dev: bool) -> Result<Option<String>> {
424 Ok(Default::default())
425 }
426
427 fn tracing_targets(&self) -> Result<Option<String>> {
432 Ok(self.shared_params().tracing_targets())
433 }
434
435 fn tracing_receiver(&self) -> Result<TracingReceiver> {
440 Ok(self.shared_params().tracing_receiver())
441 }
442
443 fn node_key(&self, net_config_dir: &PathBuf) -> Result<NodeKeyConfig> {
448 let is_dev = self.is_dev()?;
449 let role = self.role(is_dev)?;
450 self.node_key_params()
451 .map(|x| x.node_key(net_config_dir, role, is_dev))
452 .unwrap_or_else(|| Ok(Default::default()))
453 }
454
455 fn max_runtime_instances(&self) -> Result<Option<usize>> {
459 Ok(Default::default())
460 }
461
462 fn runtime_cache_size(&self) -> Result<u8> {
466 Ok(2)
467 }
468
469 fn announce_block(&self) -> Result<bool> {
473 Ok(true)
474 }
475
476 fn create_configuration<C: SubstrateCli>(
478 &self,
479 cli: &C,
480 tokio_handle: tokio::runtime::Handle,
481 ) -> Result<Configuration> {
482 let is_dev = self.is_dev()?;
483 let chain_id = self.chain_id(is_dev)?;
484 let chain_spec = cli.load_spec(&chain_id)?;
485 let base_path = base_path_or_default(self.base_path()?, &C::executable_name());
486 let config_dir = build_config_dir(&base_path, chain_spec.id());
487 let net_config_dir = build_net_config_dir(&config_dir);
488 let client_id = C::client_id();
489 let database_cache_size = self.database_cache_size()?.unwrap_or(1024);
490 let database = self.database()?.unwrap_or(
491 #[cfg(feature = "rocksdb")]
492 {
493 Database::RocksDb
494 },
495 #[cfg(not(feature = "rocksdb"))]
496 {
497 Database::ParityDb
498 },
499 );
500 let node_key = self.node_key(&net_config_dir)?;
501 let role = self.role(is_dev)?;
502 let max_runtime_instances = self.max_runtime_instances()?.unwrap_or(8);
503 let is_validator = role.is_authority();
504 let keystore = self.keystore_config(&config_dir)?;
505 let telemetry_endpoints = self.telemetry_endpoints(&chain_spec)?;
506 let runtime_cache_size = self.runtime_cache_size()?;
507
508 let rpc_addrs: Option<Vec<sc_service::config::RpcEndpoint>> = self
509 .rpc_addr(DCV::rpc_listen_port())?
510 .map(|addrs| addrs.into_iter().map(Into::into).collect());
511
512 Ok(Configuration {
513 impl_name: C::impl_name(),
514 impl_version: C::impl_version(),
515 tokio_handle,
516 transaction_pool: self.transaction_pool(is_dev)?,
517 network: self.network_config(
518 &chain_spec,
519 is_dev,
520 is_validator,
521 net_config_dir,
522 client_id.as_str(),
523 self.node_name()?.as_str(),
524 node_key,
525 DCV::p2p_listen_port(),
526 )?,
527 keystore,
528 database: self.database_config(&config_dir, database_cache_size, database)?,
529 data_path: config_dir,
530 trie_cache_maximum_size: self.trie_cache_maximum_size()?,
531 state_pruning: self.state_pruning()?,
532 blocks_pruning: self.blocks_pruning()?,
533 executor: ExecutorConfiguration {
534 wasm_method: self.wasm_method()?,
535 default_heap_pages: self.default_heap_pages()?,
536 max_runtime_instances,
537 runtime_cache_size,
538 },
539 wasm_runtime_overrides: self.wasm_runtime_overrides(),
540 rpc: RpcConfiguration {
541 addr: rpc_addrs,
542 methods: self.rpc_methods()?,
543 max_connections: self.rpc_max_connections()?,
544 cors: self.rpc_cors(is_dev)?,
545 max_request_size: self.rpc_max_request_size()?,
546 max_response_size: self.rpc_max_response_size()?,
547 id_provider: None,
548 max_subs_per_conn: self.rpc_max_subscriptions_per_connection()?,
549 port: DCV::rpc_listen_port(),
550 message_buffer_capacity: self.rpc_buffer_capacity_per_connection()?,
551 batch_config: self.rpc_batch_config()?,
552 rate_limit: self.rpc_rate_limit()?,
553 rate_limit_whitelisted_ips: self.rpc_rate_limit_whitelisted_ips()?,
554 rate_limit_trust_proxy_headers: self.rpc_rate_limit_trust_proxy_headers()?,
555 },
556 prometheus_config: self
557 .prometheus_config(DCV::prometheus_listen_port(), &chain_spec)?,
558 telemetry_endpoints,
559 offchain_worker: self.offchain_worker(&role)?,
560 force_authoring: self.force_authoring()?,
561 disable_grandpa: self.disable_grandpa()?,
562 dev_key_seed: self.dev_key_seed(is_dev)?,
563 tracing_targets: self.tracing_targets()?,
564 tracing_receiver: self.tracing_receiver()?,
565 chain_spec,
566 announce_block: self.announce_block()?,
567 role,
568 base_path,
569 })
570 }
571
572 fn log_filters(&self) -> Result<String> {
579 Ok(self.shared_params().log_filters().join(","))
580 }
581
582 fn detailed_log_output(&self) -> Result<bool> {
584 Ok(self.shared_params().detailed_log_output())
585 }
586
587 fn enable_log_reloading(&self) -> Result<bool> {
589 Ok(self.shared_params().enable_log_reloading())
590 }
591
592 fn disable_log_color(&self) -> Result<bool> {
594 Ok(self.shared_params().disable_log_color())
595 }
596
597 fn init<F>(&self, support_url: &String, impl_version: &String, logger_hook: F) -> Result<()>
626 where
627 F: FnOnce(&mut LoggerBuilder),
628 {
629 sp_panic_handler::set(support_url, impl_version);
630
631 let mut logger = LoggerBuilder::new(self.log_filters()?);
632 logger
633 .with_log_reloading(self.enable_log_reloading()?)
634 .with_detailed_output(self.detailed_log_output()?);
635
636 if let Some(tracing_targets) = self.tracing_targets()? {
637 let tracing_receiver = self.tracing_receiver()?;
638 logger.with_profiling(tracing_receiver, tracing_targets);
639 }
640
641 if self.disable_log_color()? {
642 logger.with_colors(false);
643 }
644
645 logger_hook(&mut logger);
647
648 logger.init()?;
649
650 match fdlimit::raise_fd_limit() {
651 Ok(fdlimit::Outcome::LimitRaised { to, .. }) =>
652 if to < RECOMMENDED_OPEN_FILE_DESCRIPTOR_LIMIT {
653 warn!(
654 "Low open file descriptor limit configured for the process. \
655 Current value: {:?}, recommended value: {:?}.",
656 to, RECOMMENDED_OPEN_FILE_DESCRIPTOR_LIMIT,
657 );
658 },
659 Ok(fdlimit::Outcome::Unsupported) => {
660 },
662 Err(error) => {
663 warn!(
664 "Failed to configure file descriptor limit for the process: \
665 {}, recommended value: {:?}.",
666 error, RECOMMENDED_OPEN_FILE_DESCRIPTOR_LIMIT,
667 );
668 },
669 }
670
671 Ok(())
672 }
673}
674
675pub fn generate_node_name() -> String {
677 loop {
678 let node_name = Generator::with_naming(Name::Numbered)
679 .next()
680 .expect("RNG is available on all supported platforms; qed");
681 let count = node_name.chars().count();
682
683 if count < NODE_NAME_MAX_LENGTH {
684 return node_name
685 }
686 }
687}
688
689pub(crate) fn base_path_or_default(
691 base_path: Option<BasePath>,
692 executable_name: &String,
693) -> BasePath {
694 base_path.unwrap_or_else(|| BasePath::from_project("", "", executable_name))
695}
696
697pub(crate) fn build_config_dir(base_path: &BasePath, chain_spec_id: &str) -> PathBuf {
699 base_path.config_dir(chain_spec_id)
700}
701
702pub(crate) fn build_net_config_dir(config_dir: &PathBuf) -> PathBuf {
704 config_dir.join(DEFAULT_NETWORK_CONFIG_PATH)
705}
706
707pub(crate) fn build_network_key_dir_or_default(
710 base_path: Option<BasePath>,
711 chain_spec_id: &str,
712 executable_name: &String,
713) -> PathBuf {
714 let config_dir =
715 build_config_dir(&base_path_or_default(base_path, executable_name), chain_spec_id);
716 build_net_config_dir(&config_dir)
717}