From e19d04719fa3fcc865c7f540ed3bc8da05897649 Mon Sep 17 00:00:00 2001
From: Liu Jiang <gerry@linux.alibaba.com>
Date: Thu, 30 Dec 2021 16:29:03 +0800
Subject: [PATCH] libs/types: implement KataConfig to wrap TomlConfig

The TomlConfig structure is a parsed form of Kata configuration file,
but it's a little inconveneient to access those configuration
information directly. So introduce a wrapper KataConfig to easily
access those configuration information.

Two singletons of KataConfig is provided:
- KATA_DEFAULT_CONFIG: the original version directly loaded from Kata
configuration file.
- KATA_ACTIVE_CONFIG: the active version is the KATA_DEFAULT_CONFIG
patched by annotations.

So the recommended to way to use these two singletons:
- Load TomlConfig from configuration file and set it as the default one.
- Clone the default one and patch it with values from annotations.
- Use the default one for permission checks, such as to check for
  allowed annotation keys/values.
- The patched version may be set as the active one or passed to clients.
- The clients directly accesses information from the active/passed one,
  and do not need to check annotation for override.

Signed-off-by: Liu Jiang <gerry@linux.alibaba.com>
---
 src/libs/kata-types/src/config/mod.rs | 91 +++++++++++++++++++++++++++
 1 file changed, 91 insertions(+)

diff --git a/src/libs/kata-types/src/config/mod.rs b/src/libs/kata-types/src/config/mod.rs
index 12f42794d8..cfb6316e33 100644
--- a/src/libs/kata-types/src/config/mod.rs
+++ b/src/libs/kata-types/src/config/mod.rs
@@ -8,6 +8,9 @@ use std::collections::HashMap;
 use std::fs;
 use std::io::{self, Result};
 use std::path::{Path, PathBuf};
+use std::sync::{Arc, Mutex};
+
+use lazy_static::lazy_static;
 
 use crate::{eother, sl};
 
@@ -176,6 +179,94 @@ pub fn validate_path_pattern<P: AsRef<Path>>(patterns: &[String], path: P) -> Re
     Err(eother!("Path {} is not permitted", path))
 }
 
+/// Kata configuration information.
+pub struct KataConfig {
+    config: TomlConfig,
+    agent: String,
+    hypervisor: String,
+}
+
+impl KataConfig {
+    /// Set the default Kata configuration object.
+    ///
+    /// The default Kata configuration information is loaded from system configuration file.
+    pub fn set_default_config(config: TomlConfig, hypervisor: &str, agent: &str) {
+        let kata = KataConfig {
+            config,
+            agent: agent.to_string(),
+            hypervisor: hypervisor.to_string(),
+        };
+        *KATA_DEFAULT_CONFIG.lock().unwrap() = Arc::new(kata);
+    }
+
+    /// Get the default Kata configuration object.
+    ///
+    /// The default Kata configuration information is loaded from system configuration file.
+    pub fn get_default_config() -> Arc<KataConfig> {
+        KATA_DEFAULT_CONFIG.lock().unwrap().clone()
+    }
+
+    /// Set the active Kata configuration object.
+    ///
+    /// The active Kata configuration information is default configuration information patched
+    /// with tunable configuration information from annotations.
+    pub fn set_active_config(config: TomlConfig, hypervisor: &str, agent: &str) {
+        let kata = KataConfig {
+            config,
+            agent: agent.to_string(),
+            hypervisor: hypervisor.to_string(),
+        };
+        *KATA_ACTIVE_CONFIG.lock().unwrap() = Arc::new(kata);
+    }
+
+    /// Get the active Kata configuration object.
+    ///
+    /// The active Kata configuration information is default configuration information patched
+    /// with tunable configuration information from annotations.
+    pub fn get_active_config() -> Arc<KataConfig> {
+        KATA_ACTIVE_CONFIG.lock().unwrap().clone()
+    }
+
+    /// Get the agent configuration in use.
+    pub fn get_agent(&self) -> Option<&Agent> {
+        if !self.agent.is_empty() {
+            self.config.agent.get(&self.agent)
+        } else {
+            None
+        }
+    }
+
+    /// Get the hypervisor configuration in use.
+    pub fn get_hypervisor(&self) -> Option<&Hypervisor> {
+        if !self.hypervisor.is_empty() {
+            self.config.hypervisor.get(&self.hypervisor)
+        } else {
+            None
+        }
+    }
+}
+
+lazy_static! {
+    static ref KATA_DEFAULT_CONFIG: Mutex<Arc<KataConfig>> = {
+        let config = TomlConfig::load("").unwrap();
+        let kata = KataConfig {
+            config,
+            agent: String::new(),
+            hypervisor: String::new(),
+        };
+        Mutex::new(Arc::new(kata))
+    };
+    static ref KATA_ACTIVE_CONFIG: Mutex<Arc<KataConfig>> = {
+        let config = TomlConfig::load("").unwrap();
+        let kata = KataConfig {
+            config,
+            agent: String::new(),
+            hypervisor: String::new(),
+        };
+        Mutex::new(Arc::new(kata))
+    };
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;