From e047dbe884f5288943d5ba2f8843a078d647d7ef Mon Sep 17 00:00:00 2001
From: gcontini <1121667+gcontini@users.noreply.github.com>
Date: 周六, 25 4月 2020 21:01:33 +0800
Subject: [PATCH] first docker & vm detection

---
 src/inspector/inspector.cpp |   43 +++++++++++++++++++++++--------------------
 1 files changed, 23 insertions(+), 20 deletions(-)

diff --git a/src/inspector/inspector.cpp b/src/inspector/inspector.cpp
index af9e61c..dd44f76 100644
--- a/src/inspector/inspector.cpp
+++ b/src/inspector/inspector.cpp
@@ -4,7 +4,7 @@
 #include <licensecc/licensecc.h>
 #include <fstream>
 #include "../library/os/cpu_info.hpp"
-#include "../library/os/execution_environment.hpp"
+#include "../library/os/dmi_info.hpp"
 
 using namespace std;
 using namespace license::os;
@@ -23,8 +23,10 @@
 													 {HV, "Microsoft Hypervisor"},
 													 {V_OTHER, "Other type of vm"}};
 
-const unordered_map<int, string> descByVirt = {
-	{VIRTUALIZATION::NONE, "No virtualization"}, {VIRTUALIZATION::VM, "VM"}, {VIRTUALIZATION::CONTAINER, "Container"}};
+const unordered_map<LCC_API_VIRTUALIZATION_SUMMARY, string> descByVirt = {
+	{LCC_API_VIRTUALIZATION_SUMMARY::NONE, "No virtualization"},
+	{LCC_API_VIRTUALIZATION_SUMMARY::VM, "VM"},
+	{LCC_API_VIRTUALIZATION_SUMMARY::CONTAINER, "Container"}};
 
 const unordered_map<int, string> stringByEventType = {
 	{LICENSE_OK, "OK "},
@@ -53,31 +55,32 @@
 }
 
 int main(int argc, char* argv[]) {
-	license::os::CpuInfo cpu;
-	cout << "Cpu Vendor      :" << cpu.vendor() << endl;
-	cout << "Cpu Brand       :" << cpu.brand() << endl;
-	cout << "Cpu is hypervis.:" << cpu.cpu_virtual() << endl;
-	cout << "Cpu model       :0x" << std::hex << ((long)cpu.model()) << std::dec << endl;
-	cout << "Virt. detail cpu:" << descByVirtDetail.find(cpu.virtualization_details())->second << endl <<endl;
-	ExecutionEnvironment execEnv;
-	cout << "Running in cloud:" << execEnv.is_cloud() << endl;
-	cout << "Docker          :" << execEnv.is_docker() << endl;
-	cout << "other container :" << execEnv.is_container() << endl;
-	cout << "Virtualiz. class:" << descByVirt.find(execEnv.getVirtualization())->second << endl;
-
-	cout << "Bios vendor     :" << execEnv.bios_vendor() << endl;
-	cout << "Bios description:" << execEnv.bios_description() << endl;
-	cout << "System vendor   :" << execEnv.sys_vendor() << endl <<endl;
-
 	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(static_cast<LCC_API_HW_IDENTIFICATION_STRATEGY>(x.first), hw_identifier, &bufSize)) {
+		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   :" << 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