1use serde::{Deserialize, Serialize};
99use smallvec::SmallVec;
100use std::borrow::Cow;
101use std::collections::HashMap;
102use std::io;
103use std::net::SocketAddr;
104use std::time::Duration;
105use thiserror::Error;
106
107#[derive(Debug, Error)]
113pub enum P2PError {
114 #[error("Network error: {0}")]
116 Network(#[from] NetworkError),
117
118 #[error("DHT error: {0}")]
120 Dht(#[from] DhtError),
121
122 #[error("Identity error: {0}")]
124 Identity(#[from] IdentityError),
125
126 #[error("Cryptography error: {0}")]
128 Crypto(#[from] CryptoError),
129
130 #[error("Storage error: {0}")]
132 Storage(#[from] StorageError),
133
134 #[error("Transport error: {0}")]
136 Transport(#[from] TransportError),
137
138 #[error("Configuration error: {0}")]
140 Config(#[from] ConfigError),
141
142 #[error("Security error: {0}")]
144 Security(#[from] SecurityError),
145
146 #[error("Bootstrap error: {0}")]
148 Bootstrap(#[from] BootstrapError),
149
150 #[error("IO error: {0}")]
152 Io(#[from] io::Error),
153
154 #[error("Serialization error: {0}")]
156 Serialization(Cow<'static, str>),
157
158 #[error("Validation error: {0}")]
160 Validation(Cow<'static, str>),
161
162 #[error("Operation timed out after {0:?}")]
164 Timeout(Duration),
165
166 #[error("Resource exhausted: {0}")]
168 ResourceExhausted(Cow<'static, str>),
169
170 #[error("Internal error: {0}")]
172 Internal(Cow<'static, str>),
173
174 #[error("Encoding error: {0}")]
176 Encoding(Cow<'static, str>),
177
178 #[error("Record too large: {0} bytes (max 512)")]
180 RecordTooLarge(usize),
181
182 #[error("Time error")]
184 TimeError,
185
186 #[error("Invalid input: {0}")]
188 InvalidInput(String),
189
190 #[error("WebRTC error: {0}")]
192 WebRtcError(String),
193}
194
195#[derive(Debug, Error)]
197pub enum NetworkError {
198 #[error("Connection failed to {addr}: {reason}")]
199 ConnectionFailed {
200 addr: SocketAddr,
201 reason: Cow<'static, str>,
202 },
203
204 #[error("Connection closed unexpectedly")]
205 ConnectionClosed,
206
207 #[error("Invalid network address: {0}")]
208 InvalidAddress(Cow<'static, str>),
209
210 #[error("Peer not found: {0}")]
211 PeerNotFound(Cow<'static, str>),
212
213 #[error("Peer disconnected - peer: {peer}, reason: {reason}")]
214 PeerDisconnected { peer: String, reason: String },
215
216 #[error("Network timeout")]
217 Timeout,
218
219 #[error("Too many connections")]
220 TooManyConnections,
221
222 #[error("Protocol error: {0}")]
223 ProtocolError(Cow<'static, str>),
224
225 #[error("Bind error: {0}")]
226 BindError(Cow<'static, str>),
227}
228
229#[derive(Debug, Error)]
231pub enum DhtError {
232 #[error("Key not found: {0}")]
233 KeyNotFound(Cow<'static, str>),
234
235 #[error("Store operation failed: {0}")]
236 StoreFailed(Cow<'static, str>),
237
238 #[error("Invalid key format: {0}")]
239 InvalidKey(Cow<'static, str>),
240
241 #[error("Routing table full")]
242 RoutingTableFull,
243
244 #[error("No suitable peers found")]
245 NoPeersFound,
246
247 #[error("Replication failed: {0}")]
248 ReplicationFailed(Cow<'static, str>),
249
250 #[error("Query timeout")]
251 QueryTimeout,
252
253 #[error("Routing error: {0}")]
254 RoutingError(Cow<'static, str>),
255
256 #[error("Storage failed: {0}")]
257 StorageFailed(Cow<'static, str>),
258
259 #[error("Insufficient replicas: {0}")]
260 InsufficientReplicas(Cow<'static, str>),
261}
262
263#[derive(Debug, Error)]
265pub enum IdentityError {
266 #[error("Invalid three-word address: {0}")]
267 InvalidThreeWordAddress(Cow<'static, str>),
268
269 #[error("Invalid four-word address: {0}")]
270 InvalidFourWordAddress(Cow<'static, str>),
271
272 #[error("Identity not found: {0}")]
273 IdentityNotFound(Cow<'static, str>),
274
275 #[error("Identity already exists: {0}")]
276 IdentityExists(Cow<'static, str>),
277
278 #[error("Invalid signature")]
279 InvalidSignature,
280
281 #[error("Invalid canonical bytes")]
282 InvalidCanonicalBytes,
283
284 #[error("Membership conflict")]
285 MembershipConflict,
286
287 #[error("Missing group key")]
288 MissingGroupKey,
289
290 #[error("Website root update refused")]
291 WebsiteRootUpdateRefused,
292
293 #[error("Key derivation failed: {0}")]
294 KeyDerivationFailed(Cow<'static, str>),
295
296 #[error("Permission denied")]
297 PermissionDenied,
298
299 #[error("Invalid peer ID: {0}")]
300 InvalidPeerId(Cow<'static, str>),
301
302 #[error("Invalid format: {0}")]
303 InvalidFormat(Cow<'static, str>),
304
305 #[error("System time error: {0}")]
306 SystemTime(Cow<'static, str>),
307
308 #[error("Not found: {0}")]
309 NotFound(Cow<'static, str>),
310
311 #[error("Verification failed: {0}")]
312 VerificationFailed(Cow<'static, str>),
313
314 #[error("Insufficient entropy")]
315 InsufficientEntropy,
316
317 #[error("Access denied: {0}")]
318 AccessDenied(Cow<'static, str>),
319}
320
321#[derive(Debug, Error)]
323pub enum CryptoError {
324 #[error("Encryption failed: {0}")]
325 EncryptionFailed(Cow<'static, str>),
326
327 #[error("Decryption failed: {0}")]
328 DecryptionFailed(Cow<'static, str>),
329
330 #[error("Invalid key length: expected {expected}, got {actual}")]
331 InvalidKeyLength { expected: usize, actual: usize },
332
333 #[error("Signature verification failed")]
334 SignatureVerificationFailed,
335
336 #[error("Key generation failed: {0}")]
337 KeyGenerationFailed(Cow<'static, str>),
338
339 #[error("Invalid public key")]
340 InvalidPublicKey,
341
342 #[error("Invalid private key")]
343 InvalidPrivateKey,
344
345 #[error("HKDF expansion failed: {0}")]
346 HkdfError(Cow<'static, str>),
347}
348
349#[derive(Debug, Error)]
351pub enum StorageError {
352 #[error("Database error: {0}")]
353 Database(Cow<'static, str>),
354
355 #[error("Disk full")]
356 DiskFull,
357
358 #[error("Corrupt data: {0}")]
359 CorruptData(Cow<'static, str>),
360
361 #[error("Storage path not found: {0}")]
362 PathNotFound(Cow<'static, str>),
363
364 #[error("Permission denied: {0}")]
365 PermissionDenied(Cow<'static, str>),
366
367 #[error("Lock acquisition failed")]
368 LockFailed,
369
370 #[error("Lock poisoned: {0}")]
371 LockPoisoned(Cow<'static, str>),
372
373 #[error("File not found: {0}")]
374 FileNotFound(Cow<'static, str>),
375
376 #[error("Corruption detected: {0}")]
377 CorruptionDetected(Cow<'static, str>),
378}
379
380#[derive(Debug, Error)]
382pub enum TransportError {
383 #[error("QUIC error: {0}")]
384 Quic(Cow<'static, str>),
385
386 #[error("TCP error: {0}")]
387 Tcp(Cow<'static, str>),
388
389 #[error("Invalid transport configuration: {0}")]
390 InvalidConfig(Cow<'static, str>),
391
392 #[error("Transport not supported: {0}")]
393 NotSupported(Cow<'static, str>),
394
395 #[error("Stream error: {0}")]
396 StreamError(Cow<'static, str>),
397
398 #[error("Certificate error: {0}")]
399 CertificateError(Cow<'static, str>),
400
401 #[error("Setup failed: {0}")]
402 SetupFailed(Cow<'static, str>),
403
404 #[error("Connection failed to {addr}: {reason}")]
405 ConnectionFailed {
406 addr: SocketAddr,
407 reason: Cow<'static, str>,
408 },
409
410 #[error("Bind error: {0}")]
411 BindError(Cow<'static, str>),
412
413 #[error("Accept failed: {0}")]
414 AcceptFailed(Cow<'static, str>),
415
416 #[error("Not listening")]
417 NotListening,
418
419 #[error("Not initialized")]
420 NotInitialized,
421}
422
423#[derive(Debug, Error)]
425pub enum ConfigError {
426 #[error("Missing required field: {0}")]
427 MissingField(Cow<'static, str>),
428
429 #[error("Invalid value for {field}: {reason}")]
430 InvalidValue {
431 field: Cow<'static, str>,
432 reason: Cow<'static, str>,
433 },
434
435 #[error("Configuration file not found: {0}")]
436 FileNotFound(Cow<'static, str>),
437
438 #[error("Parse error: {0}")]
439 ParseError(Cow<'static, str>),
440
441 #[error("Validation failed: {0}")]
442 ValidationFailed(Cow<'static, str>),
443
444 #[error("IO error for {path}: {source}")]
445 IoError {
446 path: Cow<'static, str>,
447 #[source]
448 source: std::io::Error,
449 },
450}
451
452#[derive(Debug, Error)]
454pub enum SecurityError {
455 #[error("Authentication failed")]
456 AuthenticationFailed,
457
458 #[error("Authorization denied")]
459 AuthorizationDenied,
460
461 #[error("Invalid credentials")]
462 InvalidCredentials,
463
464 #[error("Certificate error: {0}")]
465 CertificateError(Cow<'static, str>),
466
467 #[error("Encryption failed: {0}")]
468 EncryptionFailed(Cow<'static, str>),
469
470 #[error("Decryption failed: {0}")]
471 DecryptionFailed(Cow<'static, str>),
472
473 #[error("Invalid key: {0}")]
474 InvalidKey(Cow<'static, str>),
475
476 #[error("Signature verification failed: {0}")]
477 SignatureVerificationFailed(Cow<'static, str>),
478
479 #[error("Key generation failed: {0}")]
480 KeyGenerationFailed(Cow<'static, str>),
481
482 #[error("Authorization failed: {0}")]
483 AuthorizationFailed(Cow<'static, str>),
484}
485
486#[derive(Debug, Error)]
488pub enum BootstrapError {
489 #[error("No bootstrap nodes available")]
490 NoBootstrapNodes,
491
492 #[error("Bootstrap failed: {0}")]
493 BootstrapFailed(Cow<'static, str>),
494
495 #[error("Invalid bootstrap node: {0}")]
496 InvalidBootstrapNode(Cow<'static, str>),
497
498 #[error("Bootstrap timeout")]
499 BootstrapTimeout,
500
501 #[error("Cache error: {0}")]
502 CacheError(Cow<'static, str>),
503
504 #[error("Invalid data: {0}")]
505 InvalidData(Cow<'static, str>),
506}
507
508pub type P2pResult<T> = Result<T, P2PError>;
510
511pub trait Recoverable {
515 fn is_transient(&self) -> bool;
517
518 fn suggested_retry_after(&self) -> Option<Duration>;
520
521 fn max_retries(&self) -> usize;
523}
524
525impl Recoverable for P2PError {
526 fn is_transient(&self) -> bool {
527 match self {
528 P2PError::Network(NetworkError::ConnectionFailed { .. }) => true,
529 P2PError::Network(NetworkError::Timeout) => true,
530 P2PError::Transport(TransportError::ConnectionFailed { .. }) => true,
531 P2PError::Dht(DhtError::QueryTimeout) => true,
532 P2PError::Timeout(_) => true,
533 P2PError::ResourceExhausted(_) => true,
534 P2PError::Io(err) => matches!(
535 err.kind(),
536 io::ErrorKind::WouldBlock | io::ErrorKind::TimedOut | io::ErrorKind::Interrupted
537 ),
538 _ => false,
539 }
540 }
541
542 fn suggested_retry_after(&self) -> Option<Duration> {
543 match self {
544 P2PError::Network(NetworkError::Timeout) => Some(Duration::from_secs(5)),
545 P2PError::Timeout(duration) => Some(*duration * 2),
546 P2PError::ResourceExhausted(_) => Some(Duration::from_secs(30)),
547 P2PError::Transport(TransportError::ConnectionFailed { .. }) => {
548 Some(Duration::from_secs(1))
549 }
550 _ => None,
551 }
552 }
553
554 fn max_retries(&self) -> usize {
555 match self {
556 P2PError::Network(NetworkError::ConnectionFailed { .. }) => 3,
557 P2PError::Transport(TransportError::ConnectionFailed { .. }) => 3,
558 P2PError::Timeout(_) => 2,
559 P2PError::ResourceExhausted(_) => 1,
560 _ => 0,
561 }
562 }
563}
564
565pub trait ErrorContext<T> {
567 fn context(self, msg: &str) -> Result<T, P2PError>;
569
570 fn with_context<F>(self, f: F) -> Result<T, P2PError>
572 where
573 F: FnOnce() -> String;
574}
575
576impl<T, E> ErrorContext<T> for Result<T, E>
577where
578 E: Into<P2PError>,
579{
580 fn context(self, msg: &str) -> Result<T, P2PError> {
581 self.map_err(|e| {
582 let base_error = e.into();
583 P2PError::Internal(format!("{}: {}", msg, base_error).into())
584 })
585 }
586
587 fn with_context<F>(self, f: F) -> Result<T, P2PError>
588 where
589 F: FnOnce() -> String,
590 {
591 self.map_err(|e| {
592 let base_error = e.into();
593 P2PError::Internal(format!("{}: {}", f(), base_error).into())
594 })
595 }
596}
597
598impl P2PError {
600 pub fn connection_failed(addr: SocketAddr, reason: impl Into<String>) -> Self {
602 P2PError::Network(NetworkError::ConnectionFailed {
603 addr,
604 reason: reason.into().into(),
605 })
606 }
607
608 pub fn timeout(duration: Duration) -> Self {
610 P2PError::Timeout(duration)
611 }
612
613 pub fn validation(msg: impl Into<Cow<'static, str>>) -> Self {
615 P2PError::Validation(msg.into())
616 }
617
618 pub fn internal(msg: impl Into<Cow<'static, str>>) -> Self {
620 P2PError::Internal(msg.into())
621 }
622}
623
624impl P2PError {
626 pub fn log(&self) {
628 use tracing::{error, warn};
629
630 match self {
631 P2PError::Network(NetworkError::Timeout) | P2PError::Timeout(_) => warn!("{}", self),
632
633 P2PError::Validation(_) | P2PError::Config(_) => warn!("{}", self),
634
635 _ => error!("{}", self),
636 }
637 }
638
639 pub fn log_with_context(&self, context: &str) {
641 use tracing::error;
642 error!("{}: {}", context, self);
643 }
644}
645
646impl From<serde_json::Error> for P2PError {
649 fn from(err: serde_json::Error) -> Self {
650 P2PError::Serialization(err.to_string().into())
651 }
652}
653
654impl From<bincode::Error> for P2PError {
655 fn from(err: bincode::Error) -> Self {
656 P2PError::Serialization(err.to_string().into())
657 }
658}
659
660impl From<std::net::AddrParseError> for P2PError {
661 fn from(err: std::net::AddrParseError) -> Self {
662 P2PError::Network(NetworkError::InvalidAddress(err.to_string().into()))
663 }
664}
665
666impl From<tokio::time::error::Elapsed> for P2PError {
667 fn from(_: tokio::time::error::Elapsed) -> Self {
668 P2PError::Network(NetworkError::Timeout)
669 }
670}
671
672impl From<crate::adaptive::AdaptiveNetworkError> for P2PError {
673 fn from(err: crate::adaptive::AdaptiveNetworkError) -> Self {
674 use crate::adaptive::AdaptiveNetworkError;
675 match err {
676 AdaptiveNetworkError::Network(io_err) => P2PError::Io(io_err),
677 AdaptiveNetworkError::Io(io_err) => P2PError::Io(io_err),
678 AdaptiveNetworkError::Serialization(ser_err) => {
679 P2PError::Serialization(ser_err.to_string().into())
680 }
681 AdaptiveNetworkError::Routing(msg) => {
682 P2PError::Internal(format!("Routing error: {}", msg).into())
683 }
684 AdaptiveNetworkError::Trust(msg) => {
685 P2PError::Internal(format!("Trust error: {}", msg).into())
686 }
687 AdaptiveNetworkError::Learning(msg) => {
688 P2PError::Internal(format!("Learning error: {}", msg).into())
689 }
690 AdaptiveNetworkError::Gossip(msg) => {
691 P2PError::Internal(format!("Gossip error: {}", msg).into())
692 }
693 AdaptiveNetworkError::Other(msg) => P2PError::Internal(msg.into()),
694 }
695 }
696}
697
698#[derive(Debug, Clone, Serialize, Deserialize)]
702pub enum ErrorValue {
703 String(Cow<'static, str>),
704 Number(i64),
705 Bool(bool),
706 Duration(Duration),
707 Address(SocketAddr),
708}
709
710#[derive(Debug, Serialize, Deserialize)]
712pub struct ErrorLog {
713 pub timestamp: i64, pub error_type: &'static str,
715 pub message: Cow<'static, str>,
716 pub context: SmallVec<[(&'static str, ErrorValue); 4]>, pub stack_trace: Option<Cow<'static, str>>,
718}
719
720impl ErrorLog {
721 pub fn from_error(error: &P2PError) -> Self {
723 let mut context = SmallVec::new();
724
725 match error {
727 P2PError::Network(NetworkError::ConnectionFailed { addr, reason }) => {
728 context.push(("address", ErrorValue::Address(*addr)));
729 context.push(("reason", ErrorValue::String(reason.clone())));
730 }
731 P2PError::Timeout(duration) => {
732 context.push(("timeout", ErrorValue::Duration(*duration)));
733 }
734 P2PError::Crypto(CryptoError::InvalidKeyLength { expected, actual }) => {
735 context.push(("expected_length", ErrorValue::Number(*expected as i64)));
736 context.push(("actual_length", ErrorValue::Number(*actual as i64)));
737 }
738 _ => {}
739 }
740
741 ErrorLog {
742 timestamp: chrono::Utc::now().timestamp(),
743 error_type: error_type_name(error),
744 message: error.to_string().into(),
745 context,
746 stack_trace: None,
747 }
748 }
749
750 pub fn with_context(mut self, key: &'static str, value: ErrorValue) -> Self {
751 self.context.push((key, value));
752 self
753 }
754
755 pub fn log(&self) {
756 use log::{error, warn};
757
758 let json = serde_json::to_string(self).unwrap_or_else(|_| self.message.to_string());
759
760 match self.error_type {
761 "Validation" | "Config" => warn!("{}", json),
762 _ => error!("{}", json),
763 }
764 }
765}
766
767fn error_type_name(error: &P2PError) -> &'static str {
768 match error {
769 P2PError::Network(_) => "Network",
770 P2PError::Dht(_) => "DHT",
771 P2PError::Identity(_) => "Identity",
772 P2PError::Crypto(_) => "Crypto",
773 P2PError::Storage(_) => "Storage",
774 P2PError::Transport(_) => "Transport",
775 P2PError::Config(_) => "Config",
776 P2PError::Io(_) => "IO",
777 P2PError::Serialization(_) => "Serialization",
778 P2PError::Validation(_) => "Validation",
779 P2PError::Timeout(_) => "Timeout",
780 P2PError::ResourceExhausted(_) => "ResourceExhausted",
781 P2PError::Internal(_) => "Internal",
782 P2PError::Security(_) => "Security",
783 P2PError::Bootstrap(_) => "Bootstrap",
784 P2PError::Encoding(_) => "Encoding",
785 P2PError::RecordTooLarge(_) => "RecordTooLarge",
786 P2PError::TimeError => "TimeError",
787 P2PError::InvalidInput(_) => "InvalidInput",
788 P2PError::WebRtcError(_) => "WebRTC",
789 }
790}
791
792pub trait ErrorReporting {
794 fn report(&self) -> ErrorLog;
795 fn report_with_context(&self, context: HashMap<String, serde_json::Value>) -> ErrorLog;
796}
797
798impl ErrorReporting for P2PError {
799 fn report(&self) -> ErrorLog {
800 ErrorLog::from_error(self)
801 }
802
803 fn report_with_context(&self, context: HashMap<String, serde_json::Value>) -> ErrorLog {
804 let log = ErrorLog::from_error(self);
805 for (_key, _value) in context {
807 }
811 log
812 }
813}
814
815pub trait IntoAnyhow<T> {
819 fn into_anyhow(self) -> anyhow::Result<T>;
820}
821
822impl<T> IntoAnyhow<T> for P2pResult<T> {
823 fn into_anyhow(self) -> anyhow::Result<T> {
824 self.map_err(|e| anyhow::anyhow!(e))
825 }
826}
827
828pub trait FromAnyhowExt<T> {
829 fn into_p2p_result(self) -> P2pResult<T>;
830}
831
832impl<T> FromAnyhowExt<T> for anyhow::Result<T> {
833 fn into_p2p_result(self) -> P2pResult<T> {
834 self.map_err(|e| P2PError::Internal(e.to_string().into()))
835 }
836}
837
838pub use anyhow::{Context as AnyhowContext, Result as AnyhowResult};
840
841#[cfg(test)]
842mod tests {
843 use super::*;
844
845 #[test]
846 fn test_error_display() {
847 let err =
848 P2PError::connection_failed("127.0.0.1:8080".parse().unwrap(), "Connection refused");
849 assert_eq!(
850 err.to_string(),
851 "Network error: Connection failed to 127.0.0.1:8080: Connection refused"
852 );
853 }
854
855 #[test]
856 fn test_error_context() {
857 let result: Result<(), io::Error> =
858 Err(io::Error::new(io::ErrorKind::NotFound, "file not found"));
859
860 let with_context = crate::error::ErrorContext::context(result, "Failed to load config");
861 assert!(with_context.is_err());
862 assert!(
863 with_context
864 .unwrap_err()
865 .to_string()
866 .contains("Failed to load config")
867 );
868 }
869
870 #[test]
871 fn test_timeout_error() {
872 let err = P2PError::timeout(Duration::from_secs(30));
873 assert_eq!(err.to_string(), "Operation timed out after 30s");
874 }
875
876 #[test]
877 fn test_crypto_error() {
878 let err = P2PError::Crypto(CryptoError::InvalidKeyLength {
879 expected: 32,
880 actual: 16,
881 });
882 assert_eq!(
883 err.to_string(),
884 "Cryptography error: Invalid key length: expected 32, got 16"
885 );
886 }
887
888 #[test]
889 fn test_error_log_serialization() {
890 let error = P2PError::Network(NetworkError::ConnectionFailed {
891 addr: "127.0.0.1:8080".parse().unwrap(),
892 reason: "Connection refused".into(),
893 });
894
895 let log = error
896 .report()
897 .with_context("peer_id", ErrorValue::String("peer123".into()))
898 .with_context("retry_count", ErrorValue::Number(3));
899
900 let json = serde_json::to_string_pretty(&log).unwrap();
901 assert!(json.contains("Network"));
902 assert!(json.contains("127.0.0.1:8080"));
903 assert!(json.contains("peer123"));
904 }
905
906 #[test]
907 fn test_anyhow_conversion() {
908 let p2p_result: P2pResult<()> = Err(P2PError::validation("Invalid input"));
909 let anyhow_result = p2p_result.into_anyhow();
910 assert!(anyhow_result.is_err());
911
912 let anyhow_err = anyhow::anyhow!("Test error");
913 let anyhow_result: anyhow::Result<()> = Err(anyhow_err);
914 let p2p_result = crate::error::FromAnyhowExt::into_p2p_result(anyhow_result);
915 assert!(p2p_result.is_err());
916 match p2p_result.unwrap_err() {
917 P2PError::Internal(msg) => assert!(msg.contains("Test error")),
918 _ => panic!("Expected Internal error"),
919 }
920 }
921}