From f4e683fc6bd6ac76f9979f49d737588b6ddf16b0 Mon Sep 17 00:00:00 2001
From: gcontini <1121667+gcontini@users.noreply.github.com>
Date: 周六, 14 11月 2020 22:13:13 +0800
Subject: [PATCH] windows smbios
---
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