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/windows/execution_environment.cpp | 92 +++++++++++++++++++++++----------------------
1 files changed, 47 insertions(+), 45 deletions(-)
diff --git a/src/library/os/windows/execution_environment.cpp b/src/library/os/windows/execution_environment.cpp
index f5a0fb8..50a255e 100644
--- a/src/library/os/windows/execution_environment.cpp
+++ b/src/library/os/windows/execution_environment.cpp
@@ -1,64 +1,66 @@
-#include <paths.h>
+/*
+ * virtualization.cpp
+ *
+ * Created on: Dec 15, 2019
+ * Author: GC
+ */
+#include <windows.h>
#include <sys/stat.h>
-#include <unistd.h>
#include <fstream>
#include <iostream>
#include <stdio.h>
-#include <string.h>
-#include <dirent.h>
-#include <sys/utsname.h>
+#include <string>
-#include "isvm/BIOSReader.h"
-#include "isvm/Native.h"
#include "../../base/base.h"
+
#include "../cpu_info.hpp"
#include "../execution_environment.hpp"
namespace license {
+namespace os {
using namespace std;
+ExecutionEnvironment::ExecutionEnvironment() : m_container_type(CONTAINER_TYPE::NONE) {}
+#define MAX_UNITS 20
+CONTAINER_TYPE wine_container() {
+ size_t ndrives = 0;
+ const DWORD dwSize = MAX_PATH;
+ char szLogicalDrives[MAX_PATH] = {0};
-VIRTUALIZATION ExecutionEnvironment::getVirtualization() {
- VIRTUALIZATION result;
- CpuInfo cpuInfo;
- bool isContainer = false;
- if (isContainer) {
- result = CONTAINER;
- } else if (cpuInfo.cpu_virtual() || is_cloud()) {
- result = VM;
- } else {
- result = NONE;
+ CONTAINER_TYPE result = CONTAINER_TYPE::NONE;
+ const DWORD dwResult = GetLogicalDriveStrings(dwSize, szLogicalDrives);
+
+ //FIXME! this code missed the increment in the loop...
+ if (dwResult > 0 && dwResult <= MAX_PATH) {
+ char* szSingleDrive = szLogicalDrives;
+ while (*szSingleDrive && ndrives < MAX_UNITS) {
+ // get the next drive
+ UINT driveType = GetDriveType(szSingleDrive);
+ if (driveType == DRIVE_FIXED) {
+ string name = szSingleDrive + string("/var/run/systemd/container");
+ try {
+ ifstream systemd_container(name);
+ if (systemd_container.good()) {
+ result = CONTAINER_TYPE::DOCKER;
+ for (string line; getline(systemd_container, line);) {
+ if (line.find("docker") != string::npos) {
+ result = CONTAINER_TYPE::DOCKER;
+ break;
+ } else if (line.find("lxc") != string::npos) {
+ result = CONTAINER_TYPE::LXC;
+ break;
+ }
+ }
+ }
+ } catch (...) {
+ // no problem,we're just guessing
+ }
+ }
+ }
}
return result;
}
-bool ExecutionEnvironment::is_cloud() { return getCloudProvider() != ON_PREMISE; }
-
-bool ExecutionEnvironment::is_docker() { return false; }
-
-CLOUD_PROVIDER ExecutionEnvironment::getCloudProvider() {
- if (InitEntryPoints()) {
- BIOSReader reader;
- SystemInformation info = reader.readSystemInfo();
-
- const char *vmVendors[] = {
- "VMware", "Microsoft Corporation", "Virtual Machine", "innotek GmbH", "PowerVM", "Bochs", "KVM"};
-
- const int count = _countof(vmVendors);
- for (int i = 0; i != count; ++i) {
- const char *vendor = vmVendors[i];
-
- if (std::string::npos != info.Manufacturer.find(vendor) ||
- std::string::npos != info.ProductName.find(vendor) ||
- std::string::npos != info.SerialNum.find(vendor)) {
- std::cout << "Inside virual machine!";
- return 1;
- }
- }
- } else {
- return -1;
- }
-}
-
+} // namespace os
} // namespace license
--
Gitblit v1.9.1