centos fix

gcontini

2020-11-08 ee082d3120f29af98bd8f1e933a9a3706e1f6386

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
/*
 * 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 <cstring>
#include <dirent.h>
#include <sys/utsname.h>
 
#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 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
 
    FILE *fp;
    char *line = nullptr;
    size_t len = 0;
    ssize_t read;
    CONTAINER_TYPE result = CONTAINER_TYPE::NONE;
 
    fp = fopen("/proc/self/cgroup", "r");
    if (fp == nullptr) {
        return CONTAINER_TYPE::NONE;
    }
 
    while ((read = getline(&line, &len, fp)) != -1
            && result == CONTAINER_TYPE::NONE) {
        if (strstr(line, "docker") != NULL) {
            result = CONTAINER_TYPE::DOCKER;
        }
        if (strstr(line, "lxc") != NULL) {
            result = CONTAINER_TYPE::LXC;
        }
    }
 
    if (line) {
        free(line);
    }
    fclose(fp);
    return result;
}
 
// 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