From ecd2f38f3d9c8b28aaa4c47bc09789f11dcaa7f7 Mon Sep 17 00:00:00 2001
From: gcontini <1121667+gcontini@users.noreply.github.com>
Date: 周日, 29 11月 2020 11:19:33 +0800
Subject: [PATCH] fix documentation warnings and broken links

---
 src/library/os/execution_environment_common.cpp |   80 ++++++++++++++++++++++++++++++++++------
 1 files changed, 68 insertions(+), 12 deletions(-)

diff --git a/src/library/os/execution_environment_common.cpp b/src/library/os/execution_environment_common.cpp
index 9fa2a38..953c0bc 100644
--- a/src/library/os/execution_environment_common.cpp
+++ b/src/library/os/execution_environment_common.cpp
@@ -8,6 +8,7 @@
 #include <string.h>
 #include <unordered_map>
 #include <array>
+#include <licensecc/datatypes.h>
 
 #include "../base/base.h"
 #include "cpu_info.hpp"
@@ -17,36 +18,91 @@
 namespace os {
 using namespace std;
 
-VIRTUALIZATION ExecutionEnvironment::getVirtualization() const {
-	VIRTUALIZATION result;
-	CpuInfo cpuInfo;
+const unordered_map<string, LCC_API_VIRTUALIZATION_DETAIL> virtual_cpu_names{
+	{"bhyve bhyve ", V_OTHER}, {"KVM", KVM},	   {"MICROSOFT", HV},		{" lrpepyh vr", HV},
+	{"prl hyperv  ", PARALLELS}, {"VMWARE", VMWARE}, {"XenVMMXenVMM", V_XEN}, {"ACRNACRNACRN", V_OTHER},
+	{"VBOX", VIRTUALBOX}};
+
+const unordered_map<string, LCC_API_VIRTUALIZATION_DETAIL> vm_vendors{{"VMWARE", VMWARE},
+																	  {"MICROSOFT", HV},
+																	  {"PARALLELS", PARALLELS},
+																	  {"VITRUAL MACHINE", V_OTHER},
+																	  {"INNOTEK GMBH", VIRTUALBOX},
+																	  {"POWERVM", V_OTHER},
+																	  {"BOCHS", V_OTHER},
+																	  {"KVM", KVM}};
+
+static LCC_API_VIRTUALIZATION_DETAIL find_in_map(const unordered_map<string, LCC_API_VIRTUALIZATION_DETAIL>& map,
+												 const string& data) {
+	for (auto it : map) {
+		if (data.find(it.first) != string::npos) {
+			return it.second;
+		}
+	}
+	return BARE_TO_METAL;
+}
+
+LCC_API_VIRTUALIZATION_SUMMARY ExecutionEnvironment::virtualization() const {
+	LCC_API_VIRTUALIZATION_SUMMARY result;
 	bool isContainer = is_container();
 	if (isContainer) {
-		result = CONTAINER;
-	} else if (cpuInfo.cpu_virtual() || is_cloud()) {
-		result = VM;
+		result = LCC_API_VIRTUALIZATION_SUMMARY::CONTAINER;
+	} else if (virtualization_detail() != BARE_TO_METAL || is_cloud()) {
+		result = LCC_API_VIRTUALIZATION_SUMMARY::VM;
 	} else {
-		result = NONE;
+		result = LCC_API_VIRTUALIZATION_SUMMARY::NONE;
 	}
 	return result;
 }
 
-bool ExecutionEnvironment::is_cloud() const { return getCloudProvider() != ON_PREMISE; }
+LCC_API_VIRTUALIZATION_DETAIL ExecutionEnvironment::virtualization_detail() const {
+	LCC_API_VIRTUALIZATION_DETAIL result = BARE_TO_METAL;
+	const string bios_description = m_dmi_info.bios_description();
+	const string bios_vendor = m_dmi_info.bios_vendor();
+	const string sys_vendor = m_dmi_info.sys_vendor();
+	if ((result = find_in_map(vm_vendors, bios_description)) == BARE_TO_METAL) {
+		if ((result = find_in_map(vm_vendors, bios_vendor)) == BARE_TO_METAL) {
+			if ((result = find_in_map(vm_vendors, sys_vendor)) == BARE_TO_METAL) {
+				if ((result = find_in_map(virtual_cpu_names, m_cpu_info.vendor())) == BARE_TO_METAL) {
+					result = find_in_map(virtual_cpu_names, m_cpu_info.brand());
+				}
+			}
+		}
+	}
+	if (result == BARE_TO_METAL) {
+		if (m_cpu_info.is_hypervisor_set() || is_cloud()) {
+			result = V_OTHER;
+		}
+	}
+	return result;
+}
+
+bool ExecutionEnvironment::is_cloud() const {
+	const LCC_API_CLOUD_PROVIDER prov = cloud_provider();
+	return prov != ON_PREMISE && prov != PROV_UNKNOWN;
+}
 
 // TODO test and azure
-CLOUD_PROVIDER ExecutionEnvironment::getCloudProvider() const {
-	CLOUD_PROVIDER result = PROV_UNKNOWN;
+LCC_API_CLOUD_PROVIDER ExecutionEnvironment::cloud_provider() const {
+	LCC_API_CLOUD_PROVIDER result = PROV_UNKNOWN;
+	const string bios_description = m_dmi_info.bios_description();
+	const string bios_vendor = m_dmi_info.bios_vendor();
+	const string sys_vendor = m_dmi_info.sys_vendor();
 	if (bios_description.size() > 0 || bios_vendor.size() > 0 || sys_vendor.size() > 0) {
 		if (bios_vendor.find("SEABIOS") != string::npos || bios_description.find("ALIBABA") != string::npos ||
 			sys_vendor.find("ALIBABA") != string::npos) {
 			result = ALI_CLOUD;
-		} else if (sys_vendor.find("GOOGLE") != string::npos || bios_description.find("GOOGLE") != string::npos) {
+		} else if (sys_vendor.find("GOOGLE") != string::npos ||
+				   bios_description.find("GOOGLECOMPUTEENGINE") != string::npos) {
 			result = GOOGLE_CLOUD;
 		} else if (bios_vendor.find("AWS") != string::npos || bios_description.find("AMAZON") != string::npos ||
 				   sys_vendor.find("AWS") != string::npos) {
 			result = AWS;
+		} else if (bios_description.find("HP-COMPAQ") != string::npos ||
+				   bios_description.find("ASUS") != string::npos || bios_description.find("DELL") != string::npos) {
+			result = ON_PREMISE;
 		}
-	} 
+	}
 	return result;
 }
 }  // namespace os

--
Gitblit v1.9.1