1use std::{env, fmt::Write as _, io::Write as _, process, sync::Arc};
7
8use abscissa_core::{
9 application::{self, AppCell},
10 config::CfgCell,
11 status_err,
12 terminal::{component::Terminal, stderr, stdout, ColorChoice},
13 Application, Component, Configurable, FrameworkError, Shutdown, StandardPaths,
14};
15use semver::{BuildMetadata, Version};
16
17use tokio::sync::watch;
18use zebra_network::constants::PORT_IN_USE_ERROR;
19use zebra_state::{
20 constants::LOCK_FILE_ERROR, state_database_format_version_in_code,
21 state_database_format_version_on_disk,
22};
23
24use crate::{
25 commands::EntryPoint,
26 components::{sync::end_of_support::EOS_PANIC_MESSAGE_HEADER, tracing::Tracing},
27 config::ZebradConfig,
28};
29
30fn fatal_error(app_name: String, err: &dyn std::error::Error) -> ! {
33 status_err!("{} fatal error: {}", app_name, err);
34 process::exit(1)
35}
36
37pub static APPLICATION: AppCell<ZebradApp> = AppCell::new();
39
40lazy_static::lazy_static! {
41 pub static ref LAST_WARN_ERROR_LOG_SENDER: watch::Sender<Option<(String, tracing::Level, chrono::DateTime<chrono::Utc>)>> = watch::Sender::new(None);
43}
44
45pub fn build_version() -> Version {
53 const CARGO_PKG_VERSION: &str = env!("CARGO_PKG_VERSION");
55
56 let fallback_version = CARGO_PKG_VERSION.parse().unwrap_or_else(|error| {
58 panic!(
59 "unexpected invalid CARGO_PKG_VERSION: {error:?} in {CARGO_PKG_VERSION:?}, \
60 should have been checked by cargo"
61 )
62 });
63
64 vergen_build_version().unwrap_or(fallback_version)
65}
66
67fn vergen_build_version() -> Option<Version> {
69 const VERGEN_GIT_DESCRIBE: Option<&str> = option_env!("VERGEN_GIT_DESCRIBE");
77
78 let vergen_git_describe = VERGEN_GIT_DESCRIBE?;
88
89 let vergen_git_describe = vergen_git_describe.replace("dirty", "modified");
92
93 let mut vergen_git_describe = vergen_git_describe.split('-').peekable();
95
96 let mut version = vergen_git_describe.next()?;
98
99 version = version.strip_prefix('v').unwrap_or(version);
101
102 if Version::parse(version).is_err() {
104 return None;
105 }
106
107 let mut semver = version.to_string();
108
109 let Some(part) = vergen_git_describe.peek() else {
112 return semver.parse().ok();
114 };
115
116 if part.starts_with(char::is_alphabetic) {
117 semver.push('-');
119 semver.push_str(part);
120
121 let _ = vergen_git_describe.next();
123 }
124
125 let Some(build) = vergen_git_describe.peek() else {
127 return semver.parse().ok();
129 };
130
131 if !build.starts_with(char::is_numeric) {
132 return None;
134 }
135
136 let build_parts: Vec<_> = vergen_git_describe.collect();
138 let build_parts = build_parts.join(".");
139
140 semver.push('+');
141 semver.push_str(&build_parts);
142
143 semver.parse().ok()
144}
145
146pub fn release_version() -> Version {
148 let mut release_version = build_version();
149
150 release_version.build = BuildMetadata::EMPTY;
151
152 release_version
153}
154
155pub fn user_agent() -> String {
161 let release_version = release_version();
162 format!("/Zebra:{release_version}/")
163}
164
165#[derive(Debug, Default)]
167pub struct ZebradApp {
168 config: CfgCell<ZebradConfig>,
170
171 state: application::State<Self>,
173}
174
175impl ZebradApp {
176 pub fn git_commit() -> Option<&'static str> {
184 const GIT_COMMIT_GCLOUD: Option<&str> = option_env!("SHORT_SHA");
185 const GIT_COMMIT_VERGEN: Option<&str> = option_env!("VERGEN_GIT_SHA");
186
187 GIT_COMMIT_GCLOUD.or(GIT_COMMIT_VERGEN)
188 }
189}
190
191impl Application for ZebradApp {
192 type Cmd = EntryPoint;
194
195 type Cfg = ZebradConfig;
197
198 type Paths = StandardPaths;
200
201 fn config(&self) -> Arc<ZebradConfig> {
203 self.config.read()
204 }
205
206 fn state(&self) -> &application::State<Self> {
208 &self.state
209 }
210
211 fn framework_components(
213 &mut self,
214 _command: &Self::Cmd,
215 ) -> Result<Vec<Box<dyn Component<Self>>>, FrameworkError> {
216 let terminal = Terminal::new(ColorChoice::Never);
226
227 Ok(vec![Box::new(terminal)])
228 }
229
230 #[allow(clippy::print_stderr)]
236 #[allow(clippy::unwrap_in_result)]
237 fn register_components(&mut self, command: &Self::Cmd) -> Result<(), FrameworkError> {
238 use crate::components::{
239 metrics::MetricsEndpoint, tokio::TokioComponent, tracing::TracingEndpoint,
240 };
241
242 let mut components = self.framework_components(command)?;
243
244 let config = match command.config_path() {
247 Some(path) => match self.load_config(&path) {
248 Ok(config) => config,
249 Err(_e) if command.cmd().should_ignore_load_config_error() => Default::default(),
251 Err(e) => {
252 status_err!("Zebra could not parse the provided config file. This might mean you are using a deprecated format of the file. You can generate a valid config by running \"zebrad generate\", and diff it against yours to examine any format inconsistencies.");
253 return Err(e);
254 }
255 },
256 None => ZebradConfig::default(),
257 };
258
259 let config = command.process_config(config)?;
260
261 let theme = if config.tracing.use_color_stdout_and_stderr() {
262 color_eyre::config::Theme::dark()
263 } else {
264 color_eyre::config::Theme::new()
265 };
266
267 let disk_db_version =
272 match state_database_format_version_on_disk(&config.state, &config.network.network) {
273 Ok(Some(version)) => version.to_string(),
274 Ok(None) => "creating.new.database".to_string(),
276 Err(error) => {
277 let mut error = format!("error: {error:?}");
278 error.truncate(100);
279 error
280 }
281 };
282
283 let app_metadata = [
284 ("version", build_version().to_string()),
286 ("Zcash network", config.network.network.to_string()),
288 (
290 "running state version",
291 state_database_format_version_in_code().to_string(),
292 ),
293 ("initial disk state version", disk_db_version),
295 ("features", env!("VERGEN_CARGO_FEATURES").to_string()),
297 ];
298
299 let git_metadata: &[(_, Option<_>)] = &[
302 ("branch", option_env!("VERGEN_GIT_BRANCH")),
303 ("git commit", Self::git_commit()),
304 (
305 "commit timestamp",
306 option_env!("VERGEN_GIT_COMMIT_TIMESTAMP"),
307 ),
308 ];
309 let git_metadata: Vec<(_, String)> = git_metadata
311 .iter()
312 .filter_map(|(k, v)| Some((k, (*v)?)))
313 .map(|(k, v)| (*k, v.to_string()))
314 .collect();
315
316 let build_metadata: Vec<_> = [
317 ("target triple", env!("VERGEN_CARGO_TARGET_TRIPLE")),
318 ("rust compiler", env!("VERGEN_RUSTC_SEMVER")),
319 ("rust release date", env!("VERGEN_RUSTC_COMMIT_DATE")),
320 ("optimization level", env!("VERGEN_CARGO_OPT_LEVEL")),
321 ("debug checks", env!("VERGEN_CARGO_DEBUG")),
322 ]
323 .iter()
324 .map(|(k, v)| (*k, v.to_string()))
325 .collect();
326
327 let panic_metadata: Vec<_> = app_metadata
328 .iter()
329 .chain(git_metadata.iter())
330 .chain(build_metadata.iter())
331 .collect();
332
333 let mut builder = color_eyre::config::HookBuilder::default();
334 let mut metadata_section = "Metadata:".to_string();
335 for (k, v) in panic_metadata {
336 builder = builder.add_issue_metadata(k, v.clone());
337 write!(&mut metadata_section, "\n{k}: {}", &v)
338 .expect("unexpected failure writing to string");
339 }
340
341 builder = builder
342 .theme(theme)
343 .panic_section(metadata_section.clone())
344 .issue_url(concat!(env!("CARGO_PKG_REPOSITORY"), "/issues/new"))
345 .issue_filter(|kind| match kind {
346 color_eyre::ErrorKind::NonRecoverable(error) => {
347 let error_str = match error.downcast_ref::<String>() {
348 Some(as_string) => as_string,
349 None => return true,
350 };
351 if PORT_IN_USE_ERROR.is_match(error_str) {
353 return false;
354 }
355 if LOCK_FILE_ERROR.is_match(error_str) {
357 return false;
358 }
359 if error_str.to_string().contains(EOS_PANIC_MESSAGE_HEADER) {
361 return false;
362 }
363 true
364 }
365 color_eyre::ErrorKind::Recoverable(error) => {
366 if error.is::<tower::timeout::error::Elapsed>()
370 || error.is::<tokio::time::error::Elapsed>()
371 || error.is::<zebra_network::PeerError>()
372 || error.is::<zebra_network::SharedPeerError>()
373 || error.is::<zebra_network::HandshakeError>()
374 {
375 return false;
376 }
377
378 let error_str = error.to_string();
381 !error_str.contains("timed out")
382 && !error_str.contains("duplicate hash")
383 && !error_str.contains("No space left on device")
384 && !error_str.contains("error canonicalizing application path")
386 }
387 });
388
389 let (panic_hook, eyre_hook) = builder.into_hooks();
392 eyre_hook.install().expect("eyre_hook.install() error");
393
394 #[cfg(feature = "sentry")]
397 let guard = sentry::init(sentry::ClientOptions {
398 debug: true,
399 release: Some(build_version().to_string().into()),
400 ..Default::default()
401 });
402
403 std::panic::set_hook(Box::new(move |panic_info| {
404 let panic_report = panic_hook.panic_report(panic_info);
405 eprintln!("{panic_report}");
406
407 #[cfg(feature = "sentry")]
408 {
409 let event = crate::sentry::panic_event_from(panic_report);
410 sentry::capture_event(event);
411
412 if !guard.close(None) {
413 warn!("unable to flush sentry events during panic");
414 }
415 }
416 }));
417
418 rayon::ThreadPoolBuilder::new()
426 .num_threads(config.sync.parallel_cpu_threads)
427 .thread_name(|thread_index| format!("rayon {thread_index}"))
428 .build_global()
429 .expect("unable to initialize rayon thread pool");
430
431 let cfg_ref = &config;
432 let default_filter = command.cmd().default_tracing_filter(command.verbose);
433 let is_server = command.cmd().is_server();
434
435 let mut tracing_config = cfg_ref.tracing.clone();
437 let metrics_config = cfg_ref.metrics.clone();
438 if is_server {
439 tracing_config.filter = tracing_config
441 .filter
442 .clone()
443 .or_else(|| Some(default_filter.to_owned()));
444 } else {
445 tracing_config.filter = Some(default_filter.to_owned());
447 tracing_config.flamegraph = None;
448 }
449 components.push(Box::new(Tracing::new(
450 &config.network.network,
451 tracing_config,
452 command.cmd().uses_intro(),
453 )?));
454
455 if is_server {
457 tracing::info!("Diagnostic {}", metadata_section);
458 info!(config_path = ?command.config_path(), config = ?cfg_ref, "loaded zebrad config");
459 }
460
461 let net = config.network.network.to_string();
465 let net = match net.as_str() {
466 default_net_name @ ("Testnet" | "Mainnet") => &default_net_name[..4],
467 other_net_name => other_net_name,
468 };
469 let global_span = if let Some(git_commit) = ZebradApp::git_commit() {
470 error_span!("", zebrad = git_commit, net)
471 } else {
472 error_span!("", net)
473 };
474
475 let global_guard = global_span.enter();
476 std::mem::forget(global_guard);
478
479 tracing::info!(
480 num_threads = rayon::current_num_threads(),
481 "initialized rayon thread pool for CPU-bound tasks",
482 );
483
484 if is_server {
486 components.push(Box::new(TokioComponent::new()?));
487 components.push(Box::new(TracingEndpoint::new(cfg_ref)?));
488 components.push(Box::new(MetricsEndpoint::new(&metrics_config)?));
489 }
490
491 self.state.components_mut().register(components)?;
492
493 self.after_config(config)
495 }
496
497 #[allow(clippy::unwrap_in_result)]
499 fn init(&mut self, command: &Self::Cmd) -> Result<(), FrameworkError> {
500 self.register_components(command)
504 }
505
506 fn after_config(&mut self, config: Self::Cfg) -> Result<(), FrameworkError> {
512 self.state.components_mut().after_config(&config)?;
514 self.config.set_once(config);
515
516 Ok(())
517 }
518
519 fn shutdown(&self, shutdown: Shutdown) -> ! {
520 let _ = stdout().lock().flush();
527 let _ = stderr().lock().flush();
528
529 let shutdown_result = self.state().components().shutdown(self, shutdown);
530
531 self.state()
532 .components_mut()
533 .get_downcast_mut::<Tracing>()
534 .map(Tracing::shutdown);
535
536 if let Err(e) = shutdown_result {
537 let app_name = self.name().to_string();
538 fatal_error(app_name, &e);
539 }
540
541 match shutdown {
542 Shutdown::Graceful => process::exit(0),
543 Shutdown::Forced => process::exit(1),
544 Shutdown::Crash => process::exit(2),
545 }
546 }
547}
548
549pub fn boot(app_cell: &'static AppCell<ZebradApp>) -> ! {
553 let args =
554 EntryPoint::process_cli_args(env::args_os().collect()).unwrap_or_else(|err| err.exit());
555
556 ZebradApp::run(app_cell, args);
557 process::exit(0);
558}