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 |   55 +++++++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 41 insertions(+), 14 deletions(-)

diff --git a/src/library/os/windows/execution_environment.cpp b/src/library/os/windows/execution_environment.cpp
index 86513d5..50a255e 100644
--- a/src/library/os/windows/execution_environment.cpp
+++ b/src/library/os/windows/execution_environment.cpp
@@ -4,16 +4,15 @@
  *  Created on: Dec 15, 2019
  *      Author: GC
  */
+#include <windows.h>
 #include <sys/stat.h>
 #include <fstream>
 #include <iostream>
 #include <stdio.h>
 #include <string>
 
-#include "isvm/BIOSReader.h"
-#include "isvm/Native.h"
 #include "../../base/base.h"
-#include "../../base/StringUtils.h"
+
 #include "../cpu_info.hpp"
 #include "../execution_environment.hpp"
 
@@ -21,19 +20,47 @@
 namespace os {
 using namespace std;
 
-ExecutionEnvironment::ExecutionEnvironment() {
-	if (InitEntryPoints()) {
-		BIOSReader reader;
-		SystemInformation info = reader.readSystemInfo();
-		sys_vendor = toupper_copy(info.Manufacturer);
-		bios_vendor = toupper_copy(info.ProductName);
-		bios_description = toupper_copy(info.SysVersion) + toupper_copy(info.family);
+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};
+
+	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;
 }
 
-//TODO
-bool ExecutionEnvironment::is_docker() const { return false; }
-//TODO
-bool ExecutionEnvironment::is_container() const { return is_docker(); }
 }  // namespace os
 }  // namespace license

--
Gitblit v1.9.1