From 4d446829c490d84081f7b5fd92d034a8c0f948dc Mon Sep 17 00:00:00 2001
From: Revertron <r@revertron.com>
Date: Sun, 9 May 2021 22:08:29 +0200
Subject: [PATCH] Added error check for domain name decryption.

---
 src/keystore.rs | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/keystore.rs b/src/keystore.rs
index d97005d..f89ba54 100644
--- a/src/keystore.rs
+++ b/src/keystore.rs
@@ -28,6 +28,8 @@ use std::cell::RefCell;
 use self::ed25519_dalek::{Signer, PublicKey, Verifier, SecretKey};
 use self::ed25519_dalek::ed25519::signature::Signature;
 use rand_old::{CryptoRng, RngCore};
+use ecies_ed25519::Error;
+use std::alloc::Global;
 
 #[derive(Debug)]
 pub struct Keystore {
@@ -187,8 +189,13 @@ impl Keystore {
     }
 
     pub fn decrypt(&self, message: &[u8]) -> Bytes {
-        let decrypted = self.crypto_box.reveal(message).unwrap();
-        Bytes::from_bytes(&decrypted)
+        match self.crypto_box.reveal(message) {
+            Ok(decrypted) => { Bytes::from_bytes(&decrypted) }
+            Err(_) => {
+                warn!("Decryption failed");
+                Bytes::default()
+            }
+        }
     }
 }