From 36093433d67c2eb2c2a7b86ff4ffb9138e1f1869 Mon Sep 17 00:00:00 2001
From: gcontini <1121667+gcontini@users.noreply.github.com>
Date: 周六, 25 4月 2020 22:22:00 +0800
Subject: [PATCH] improve tests

---
 src/inspector/inspector.cpp |   82 +++++++++++++++++++++++++++++++++--------
 1 files changed, 66 insertions(+), 16 deletions(-)

diff --git a/src/inspector/inspector.cpp b/src/inspector/inspector.cpp
index 7cb15f3..0c0819e 100644
--- a/src/inspector/inspector.cpp
+++ b/src/inspector/inspector.cpp
@@ -4,13 +4,53 @@
 #include <licensecc/licensecc.h>
 #include <fstream>
 #include "../library/os/cpu_info.hpp"
+#include "../library/os/dmi_info.hpp"
 
 using namespace std;
-const map<IDENTIFICATION_STRATEGY, string> stringByStrategyId = {
-	{STRATEGY_DEFAULT, "DEFAULT"}, {STRATEGY_ETHERNET, "MAC"},	 {STRATEGY_IP_ADDRESS, "IP"},
-	{STRATEGY_DISK_NUM, "Disk1"},  {STRATEGY_DISK_LABEL, "Disk2"}, {STRATEGY_PLATFORM_SPECIFIC, "Custom"}};
+using namespace license::os;
 
-const unordered_map<EVENT_TYPE, string> stringByEventType = {
+const map<int, string> stringByStrategyId = {{STRATEGY_DEFAULT, "DEFAULT"},
+											 {STRATEGY_ETHERNET, "MAC"},
+											 {STRATEGY_IP_ADDRESS, "IP"},
+											 {STRATEGY_DISK_NUM, "Disk1"},
+											 {STRATEGY_DISK_LABEL, "Disk2"}};
+
+const unordered_map<int, string> descByVirtDetail = {{BARE_TO_METAL, "No virtualization"},
+													 {VMWARE, "Vmware"},
+													 {VIRTUALBOX, "Virtualbox"},
+													 {V_XEN, "XEN"},
+													 {KVM, "KVM"},
+													 {HV, "Microsoft Hypervisor"},
+													 {V_OTHER, "Other type of vm"}};
+
+const unordered_map<int, string> descByVirt = {
+	{LCC_API_VIRTUALIZATION_SUMMARY::NONE, "No virtualization"},
+	{LCC_API_VIRTUALIZATION_SUMMARY::VM, "Virtual machine"},
+	{LCC_API_VIRTUALIZATION_SUMMARY::CONTAINER, "Container"}};
+
+typedef enum {
+	PROV_UNKNOWN = 0,
+	ON_PREMISE = 1,
+	GOOGLE_CLOUD = 2,
+	AZURE_CLOUD = 3,
+	AWS = 4,
+	/**
+	 * "/sys/class/dmi/id/bios_vendor" SeaBIOS
+	 * "/sys/class/dmi/id/sys_vendor" Alibaba Cloud
+	 * modalias
+	 * "dmi:bvnSeaBIOS:bvrrel-1.7.5-0-ge51488c-20140602_164612-nilsson.home.kraxel.org:bd04/01/2014:svnAlibabaCloud:pnAlibabaCloudECS:pvrpc-i440fx-2.1:cvnAlibabaCloud:ct1:cvrpc-i440fx-2.1:"
+	 */
+	ALI_CLOUD = 5
+} LCC_API_CLOUD_PROVIDER;
+
+const unordered_map<int, string> descByCloudProvider = {{PROV_UNKNOWN, "Provider unknown"},
+														{ON_PREMISE, "On premise hardware (no cloud)"},
+														{GOOGLE_CLOUD, "Google Cloud"},
+														{AZURE_CLOUD, "Microsoft Azure"},
+														{AWS, "Amazon AWS"},
+														{ALI_CLOUD, "Alibaba Cloud (Chinese cloud provider)"}};
+
+const unordered_map<int, string> stringByEventType = {
 	{LICENSE_OK, "OK "},
 	{LICENSE_FILE_NOT_FOUND, "license file not found "},
 	{LICENSE_SERVER_NOT_FOUND, "license server can't be contacted "},
@@ -22,11 +62,11 @@
 	{LICENSE_CORRUPTED, "license signature didn't match with current license "},
 	{IDENTIFIERS_MISMATCH, "Calculated identifier and the one provided in license didn't match"}};
 
-static EVENT_TYPE verifyLicense(const string& fname) {
+static LCC_EVENT_TYPE verifyLicense(const string& fname) {
 	LicenseInfo licenseInfo;
 	LicenseLocation licLocation = {LICENSE_PATH};
 	std::copy(fname.begin(), fname.end(), licLocation.licenseData);
-	EVENT_TYPE result = acquire_license(nullptr, &licLocation, &licenseInfo);
+	LCC_EVENT_TYPE result = acquire_license(nullptr, &licLocation, &licenseInfo);
 	if (result == LICENSE_OK) {
 		cout << "license OK" << endl;
 	} else {
@@ -37,21 +77,31 @@
 }
 
 int main(int argc, char* argv[]) {
-	license::CpuInfo cpu;
-	cout << "CpuVendor      :" << cpu.vendor() << endl;
-	cout << "Virtual machine:" << cpu.cpu_virtual() << endl;
-	cout << "Cpu model      : 0x" << std::hex << ((long)cpu.model()) << std::dec << endl;
-
-	char pc_identifier[PC_IDENTIFIER_SIZE + 1];
-	size_t bufSize = PC_IDENTIFIER_SIZE + 1;
+	char hw_identifier[LCC_API_PC_IDENTIFIER_SIZE + 1];
+	size_t bufSize = LCC_API_PC_IDENTIFIER_SIZE + 1;
+	ExecutionEnvironmentInfo exec_env_info;
 	for (const auto& x : stringByStrategyId) {
-		if (identify_pc(x.first, pc_identifier, &bufSize)) {
-			std::cout << x.second << ':' << pc_identifier << std::endl;
-
+		if (identify_pc(static_cast<LCC_API_HW_IDENTIFICATION_STRATEGY>(x.first), hw_identifier, &bufSize,
+						&exec_env_info)) {
+			std::cout << x.second << ':' << hw_identifier << std::endl;
 		} else {
 			std::cout << x.second << ": NA" << endl;
 		}
 	}
+	cout << "Virtualiz. class :" << descByVirt.find(exec_env_info.virtualization)->second << endl;
+	cout << "Virtualiz. detail:" << descByVirtDetail.find(exec_env_info.virtualization_detail)->second << endl;
+	cout << "Cloud provider   :" << descByCloudProvider.find(exec_env_info.cloud_provider) << endl
+		 << "=============" << endl;
+	license::os::CpuInfo cpu;
+	cout << "Cpu Vendor       :" << cpu.vendor() << endl;
+	cout << "Cpu Brand        :" << cpu.brand() << endl;
+	cout << "Cpu hypervisor   :" << cpu.is_hypervisor_set() << endl;
+	cout << "Cpu model        :0x" << std::hex << ((long)cpu.model()) << std::dec << endl;
+	license::os::DmiInfo dmi_info;
+	cout << "Bios vendor     :" << dmi_info.bios_vendor() << endl;
+	cout << "Bios description:" << dmi_info.bios_description() << endl;
+	cout << "System vendor   :" << dmi_info.sys_vendor() << endl << endl;
+
 
 	if (argc == 2) {
 		const string fname(argv[1]);

--
Gitblit v1.9.1