chore: added is_hardened() mechanism to conclusion()

Author: kernelwernel

src/cli.cpp

@@ -194,6 +194,10 @@ static const char* color(const u8 score) {
         return "";
     }
 
+    if (VM::is_hardened()) {
+        return green.c_str();
+    }
+
     if (arg_bitset.test(DYNAMIC)) {
         if      (score == 0)   { return red.c_str(); }
         else if (score <= 12)  { return red.c_str(); }

src/vmaware.hpp

@@ -12348,6 +12348,7 @@ struct VM {
     static std::string conclusion(const flagset &flags = core::generate_default()) {
         std::string brand_tmp = brand(flags);
         const u8 percent_tmp = percentage(flags);
+        const bool has_hardener = is_hardened();
 
         constexpr const char* very_unlikely = "Very unlikely a";
         constexpr const char* unlikely = "Unlikely a";
@@ -12362,30 +12363,38 @@ struct VM {
                 return memo::conclusion::fetch();
             }
 
+            const char* hardener = "";
+            
+            if (has_hardener) {
+                hardener = "hardened ";
+            }
+            
             const char* addition = " a ";
 
             // this basically just fixes the grammatical syntax
             // by either having "a" or "an" before the VM brand
             // name. It would look weird if the conclusion 
             // message was "an VirtualBox" or "a Anubis", so this
-            // lambda fixes that issue.    
+            // condition fixes that issue.    
             if (
-                (brand_tmp == brands::ACRN) ||
-                (brand_tmp == brands::ANUBIS) ||
-                (brand_tmp == brands::BSD_VMM) ||
-                (brand_tmp == brands::INTEL_HAXM) ||
-                (brand_tmp == brands::APPLE_VZ) ||
-                (brand_tmp == brands::INTEL_KGT) ||
-                (brand_tmp == brands::POWERVM) ||
-                (brand_tmp == brands::OPENSTACK) ||
-                (brand_tmp == brands::AWS_NITRO) ||
-                (brand_tmp == brands::OPENVZ) ||
-                (brand_tmp == brands::INTEL_TDX) ||
-                (brand_tmp == brands::AMD_SEV) ||
-                (brand_tmp == brands::AMD_SEV_ES) ||
-                (brand_tmp == brands::AMD_SEV_SNP) ||
-                (brand_tmp == brands::NSJAIL) ||
-                (brand_tmp == brands::NULL_BRAND)
+                !hardener && (
+                    (brand_tmp == brands::ACRN) ||
+                    (brand_tmp == brands::ANUBIS) ||
+                    (brand_tmp == brands::BSD_VMM) ||
+                    (brand_tmp == brands::INTEL_HAXM) ||
+                    (brand_tmp == brands::APPLE_VZ) ||
+                    (brand_tmp == brands::INTEL_KGT) ||
+                    (brand_tmp == brands::POWERVM) ||
+                    (brand_tmp == brands::OPENSTACK) ||
+                    (brand_tmp == brands::AWS_NITRO) ||
+                    (brand_tmp == brands::OPENVZ) ||
+                    (brand_tmp == brands::INTEL_TDX) ||
+                    (brand_tmp == brands::AMD_SEV) ||
+                    (brand_tmp == brands::AMD_SEV_ES) ||
+                    (brand_tmp == brands::AMD_SEV_SNP) ||
+                    (brand_tmp == brands::NSJAIL) ||
+                    (brand_tmp == brands::NULL_BRAND)
+                )
             ) {
                 addition = " an ";
             }
@@ -12399,17 +12408,20 @@ struct VM {
             // Hyper-V artifacts are an exception due to how unique the circumstance is
             std::string result;
             if (brand_tmp == brands::HYPERV_ARTIFACT) {
-                result = std::string(category) + addition + brand_tmp;
-            }
-            else {
-                result = std::string(category) + addition + brand_tmp + " VM";
+                result = std::string(category) + addition + hardener + brand_tmp;
+            } else {
+                result = std::string(category) + addition + hardener + brand_tmp + " VM";
             }
 
             memo::conclusion::store(result.c_str());
 
             return result;
         };
 
+        if (has_hardener) {
+            return make_conclusion(inside_vm);
+        }
+
         if (core::is_enabled(flags, DYNAMIC)) {
             if (percent_tmp == 0) { return "Running on baremetal"; }
             else if (percent_tmp <= 20) { return make_conclusion(very_unlikely); }
@@ -12418,14 +12430,13 @@ struct VM {
             else if (percent_tmp <= 62) { return make_conclusion(might); }
             else if (percent_tmp <= 75) { return make_conclusion(likely); }
             else if (percent_tmp < 100) { return make_conclusion(very_likely); }
-            else { return make_conclusion(inside_vm); }
         }
 
         if (percent_tmp == 100) {
             return make_conclusion(inside_vm);
-        } else {
-            return "Running on baremetal";
         }
+
+        return "Running on baremetal";
     }