From c914d298604ef7ad3934c3c213d3f8822d6646d9 Mon Sep 17 00:00:00 2001
From: Guillaume Buisson <contact@guillaume.dev>
Date: 周四, 20 5月 2021 13:50:43 +0800
Subject: [PATCH] Several fixes (#120)

---
 src/library/os/linux/execution_environment.cpp |   80 ++++++++++++++--------------------------
 1 files changed, 28 insertions(+), 52 deletions(-)

diff --git a/src/library/os/linux/execution_environment.cpp b/src/library/os/linux/execution_environment.cpp
index 1f14d62..216df0c 100644
--- a/src/library/os/linux/execution_environment.cpp
+++ b/src/library/os/linux/execution_environment.cpp
@@ -1,16 +1,17 @@
 /*
- * virtualization.cpp
+ * execution_environment.cpp
  *
  *  Created on: Dec 15, 2019
  *      Author: GC
  */
+
 #include <paths.h>
 #include <sys/stat.h>
 #include <unistd.h>
 #include <fstream>
 #include <iostream>
 #include <stdio.h>
-#include <string.h>
+#include <cstring>
 #include <dirent.h>
 #include <sys/utsname.h>
 
@@ -18,64 +19,57 @@
 #include "../cpu_info.hpp"
 #include "../execution_environment.hpp"
 #include "../../base/file_utils.hpp"
-#include "../../base/StringUtils.h"
+#include "../../base/string_utils.h"
 
 namespace license {
 namespace os {
 using namespace std;
 
 // 0=NO 1=Docker/2=Lxc
-static int checkContainerProc() {
+static CONTAINER_TYPE checkContainerProc() {
 	// in docer /proc/self/cgroups contains the "docker" or "lxc" string
 	// https://stackoverflow.com/questions/23513045/how-to-check-if-a-process-is-running-inside-docker-container
-	char path[MAX_PATH] = {0};
-	char proc_path[MAX_PATH], pidStr[64];
-	pid_t pid = getpid();
-	sprintf(pidStr, "%d", pid);
-	strcpy(proc_path, "/proc/");
-	strcat(proc_path, pidStr);
-	strcat(proc_path, "/cgroup");
 
 	FILE *fp;
-	char *line = NULL;
+	char *line = nullptr;
 	size_t len = 0;
 	ssize_t read;
-	int result = 0;
+	CONTAINER_TYPE result = CONTAINER_TYPE::NONE;
 
-	fp = fopen(proc_path, "r");
-	if (fp == NULL) {
-		return 0;
+	fp = fopen("/proc/self/cgroup", "r");
+	if (fp == nullptr) {
+		return CONTAINER_TYPE::NONE;
 	}
 
-	while ((read = getline(&line, &len, fp)) != -1 && result == 0) {
-		// line[len]=0;
-		// printf("Retrieved line of length %zu:\n", read);
-		// printf("%s", line);
+	while ((read = getline(&line, &len, fp)) != -1
+			&& result == CONTAINER_TYPE::NONE) {
 		if (strstr(line, "docker") != NULL) {
-			result = 1;
+			result = CONTAINER_TYPE::DOCKER;
 		}
 		if (strstr(line, "lxc") != NULL) {
-			result = 2;
+			result = CONTAINER_TYPE::LXC;
 		}
 	}
 
+	if (line) {
+		free(line);
+	}
 	fclose(fp);
-	if (line) free(line);
 	return result;
 }
 
 // 0=NO 1=Docker/2=Lxc
-static int checkSystemdContainer() {
+static CONTAINER_TYPE checkSystemdContainer() {
 	ifstream systemd_container("/var/run/systemd/container");
-	int result = 0;
+	CONTAINER_TYPE result = CONTAINER_TYPE::NONE;
 	if (systemd_container.good()) {
-		result = 1;
+		result = CONTAINER_TYPE::DOCKER;
 		for (string line; getline(systemd_container, line);) {
 			if (line.find("docker") != string::npos) {
-				result = 1;
+				result = CONTAINER_TYPE::DOCKER;
 				break;
 			} else if (line.find("lxc") != string::npos) {
-				result = 2;
+				result = CONTAINER_TYPE::LXC;
 				break;
 			}
 		}
@@ -83,33 +77,15 @@
 	return result;
 }
 
-ExecutionEnvironment::ExecutionEnvironment() {
-	try {
-		m_bios_vendor = toupper_copy(trim_copy(get_file_contents("/sys/class/dmi/id/sys_vendor", 256)));
-
-	} catch (...) {
+static CONTAINER_TYPE get_container_type() {
+	CONTAINER_TYPE result = checkContainerProc();
+	if (result == CONTAINER_TYPE::NONE) {
+		result = checkSystemdContainer();
 	}
-	try {
-		m_bios_description = toupper_copy(trim_copy(get_file_contents("/sys/class/dmi/id/modalias", 256)));
-		char last_char = m_bios_description[m_bios_description.length() - 1];
-		if (last_char == '\r' || last_char == '\n') {
-			m_bios_description = m_bios_description.erase(m_bios_description.length() - 1);
-		}
-	} catch (...) {
-	}
-	try {
-		m_sys_vendor = get_file_contents("/sys/class/dmi/id/sys_vendor", 256);
-		char last_char = m_sys_vendor[m_sys_vendor.length() - 2];
-		if (last_char == '\r' || last_char == '\n') {
-			m_sys_vendor = m_sys_vendor.erase(m_sys_vendor.length() - 1);
-		}
-	} catch (...) {
-	}
+	return result;
 }
 
-bool ExecutionEnvironment::is_container() const { return (checkContainerProc() != 0 || checkSystemdContainer() != 0); }
-
-bool ExecutionEnvironment::is_docker() const { return (checkContainerProc() == 1 || checkSystemdContainer() == 1); }
+ExecutionEnvironment::ExecutionEnvironment() : m_container_type(get_container_type()) {}
 
 }  // namespace os
 }  // namespace license

--
Gitblit v1.9.1