From 9c33ad84278353286bbb4118cd11ca3fcfec439a Mon Sep 17 00:00:00 2001 From: gcontini <1121667+gcontini@users.noreply.github.com> Date: 周六, 31 10月 2020 23:36:25 +0800 Subject: [PATCH] codacy cleanup --- 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..f3bc1d4 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") || + bios_description.find("DELL")) { + result = ON_PREMISE; } - } + } return result; } } // namespace os -- Gitblit v1.9.1