gcontini
2020-10-31 9c33ad84278353286bbb4118cd11ca3fcfec439a
src/library/os/linux/execution_environment.cpp
@@ -1,12 +1,15 @@
/*
 * virtualization.cpp
 * execution_environment.cpp
 *
 *  Created on: Dec 15, 2019
 *      Author: GC
 */
#define __STDC_WANT_LIB_EXT1__1
#include <paths.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fstream>
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <dirent.h>
@@ -15,41 +18,42 @@
#include "../../base/base.h"
#include "../cpu_info.hpp"
#include "../execution_environment.hpp"
#include "../../base/file_utils.hpp"
#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");
   snprintf(pidStr, sizeof(pidStr), "%d", pid);
   strncpy(proc_path, "/proc/", sizeof(proc_path));
   strncat(proc_path, pidStr, sizeof(proc_path));
   strncpy(proc_path, "/cgroup", sizeof(proc_path));
   FILE *fp;
   char *line = NULL;
   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;
      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);
      if (strstr(line, "docker") != NULL) {
         result = 1;
         result = CONTAINER_TYPE::DOCKER;
      }
      if (strstr(line, "lxc") != NULL) {
         result = 2;
         result = CONTAINER_TYPE::LXC;
      }
   }
@@ -58,21 +62,34 @@
   return result;
}
// 0=NO 1=Docker/Lxc
static int checkLXC() { return (access("/var/run/systemd/container", F_OK) == 0) ? 1 : 0; }
VIRTUALIZATION ExecutionEnvironment::getVirtualization() {
   VIRTUALIZATION result = NONE;
   CpuInfo cpuInfo;
   int isContainer = checkContainerProc();
   if (isContainer == 1) {
      result = CONTAINER;
   } else if (isContainer == 2 || checkLXC()) {
      result = CONTAINER;
   } else if (cpuInfo.cpu_virtual()) {
      result = VM;
   } else {
// 0=NO 1=Docker/2=Lxc
static CONTAINER_TYPE checkSystemdContainer() {
   ifstream systemd_container("/var/run/systemd/container");
   CONTAINER_TYPE result = CONTAINER_TYPE::NONE;
   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;
         }
      }
   }
   return result;
}
static CONTAINER_TYPE get_container_type() {
   CONTAINER_TYPE result = checkContainerProc();
   if (result == CONTAINER_TYPE::NONE) {
      result = checkSystemdContainer();
   }
   return result;
}
ExecutionEnvironment::ExecutionEnvironment() : m_container_type(get_container_type()) {}
}  // namespace os
}  // namespace license