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/library/os/execution_environment.hpp |   45 +++++++++++++++++++++++++++------------------
 1 files changed, 27 insertions(+), 18 deletions(-)

diff --git a/src/library/os/execution_environment.hpp b/src/library/os/execution_environment.hpp
index 6fbd6c6..f52f8db 100644
--- a/src/library/os/execution_environment.hpp
+++ b/src/library/os/execution_environment.hpp
@@ -8,32 +8,41 @@
 #ifndef SRC_LIBRARY_OS_VIRTUALIZATION_HPP_
 #define SRC_LIBRARY_OS_VIRTUALIZATION_HPP_
 
+#include <string>
+#include <licensecc/datatypes.h>
+#include "dmi_info.hpp"
+#include "cpu_info.hpp"
+
 namespace license {
+namespace os {
 
-typedef enum { NONE, CONTAINER, VM } VIRTUALIZATION;
-typedef enum {
-	ON_PREMISE,
-	GOOGLE_CLOUD,
-	AZURE_CLOUD,
-	AWS,
-	/**
-	 * "/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
-} CLOUD_PROVIDER;
+/*
+ * windows bios sometimes reports vm names add execution environment detection from bios
+const char *vmVendors[] = {
+	"VMware", "Microsoft Corporation", "Virtual Machine", "innotek GmbH", "PowerVM", "Bochs", "KVM"};
+*/
 
-typedef enum { BARE_TO_METAL, VMWARE, VIRTUALBOX, XEN, KVM } VIRTUALIZATION_DETAIL;
+enum CONTAINER_TYPE { NONE, DOCKER, LXC };
 
 class ExecutionEnvironment {
+private:
+	const CpuInfo m_cpu_info;
+	const DmiInfo m_dmi_info;
+	const CONTAINER_TYPE m_container_type;
+
 public:
-	ExecutionEnvironment(){};
-	virtual ~ExecutionEnvironment(){};
-	VIRTUALIZATION getVirtualization();
+	ExecutionEnvironment();
+	~ExecutionEnvironment(){};
+	LCC_API_VIRTUALIZATION_SUMMARY virtualization() const;
+	bool is_cloud() const;
+	bool is_docker() const { return m_container_type != DOCKER; }
+	// detect if it's a kind of container technology (docker or lxc)
+	bool is_container() const { return m_container_type != NONE; }
+	LCC_API_CLOUD_PROVIDER cloud_provider() const;
+	LCC_API_VIRTUALIZATION_DETAIL virtualization_detail() const;
 };
 
+}  // namespace os
 }  // namespace license
 
 #endif /* SRC_LIBRARY_OS_VIRTUALIZATION_HPP_ */

--
Gitblit v1.9.1