Fixed an issue where post_config could not modify the core data structure (#99)

* Add the validation of the storage keyword in the configuration file

* Fixed an issue where post_config could not modify the core data structure

Author: wa5i

src/cli/config.rs

@@ -101,6 +101,8 @@ pub struct Storage {
     pub config: HashMap<String, Value>,
 }
 
+static STORAGE_TYPE_KEYWORDS: &[&str] = &["file", "mysql"];
+
 fn default_bool_true() -> bool {
     true
 }
@@ -182,11 +184,11 @@ where
 {
     let storage: HashMap<String, Storage> = Deserialize::deserialize(deserializer)?;
 
-    // for key in storage.keys() {
-    //     if key != "file" {
-    //         return Err(serde::de::Error::custom("Invalid storage key"));
-    //     }
-    // }
+    for key in storage.keys() {
+        if !STORAGE_TYPE_KEYWORDS.contains(&key.as_str()) {
+            return Err(serde::de::Error::custom("Invalid storage key"));
+        }
+    }
 
     Ok(storage)
 }

src/core.rs

@@ -135,9 +135,11 @@ impl Core {
         let cert_module = CertModule::new(self);
         self.module_manager.add_module(Arc::new(RwLock::new(Box::new(cert_module))))?;
 
-        let handlers = self.handlers.read()?;
+        let handlers = {
+            self.handlers.read()?.clone()
+        };
         for handler in handlers.iter() {
-            match handler.post_config(Arc::clone(&core), config) {
+            match handler.post_config(self, config) {
                 Ok(_) => {
                     continue;
                 }

src/handler.rs

@@ -6,8 +6,6 @@
 //! The `Handler` trait should be implemented in other module, such as the `rusty_vault::router`
 //! for instance.
 
-use std::sync::{Arc, RwLock};
-
 use derive_more::Display;
 use async_trait::async_trait;
 
@@ -22,7 +20,7 @@ use crate::{
 pub trait Handler: Send + Sync {
     fn name(&self) -> String;
 
-    fn post_config(&self, _core: Arc<RwLock<Core>>, _config: Option<&Config>) -> Result<(), RvError> {
+    fn post_config(&self, _core: &mut Core, _config: Option<&Config>) -> Result<(), RvError> {
         Err(RvError::ErrHandlerDefault)
     }