Gabriele Contini
2020-01-11 1d1082fe695366da3498f528b953203a90fe7385
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
/*
 * pc_identifier_facade.cpp
 *
 *  Created on: Dec 26, 2019
 *      Author: devel
 */
 
#include "pc_identifier_facade.hpp"
 
#include <cstdlib>
#include <stdexcept>
 
#include "../../../projects/DEFAULT/include/licensecc/DEFAULT/licensecc_properties.h"
#include "../base/base.h"
#include "../base/logger.h"
#include "../os/cpu_info.hpp"
#include "../os/execution_environment.hpp"
#include "default_strategy.hpp"
#include "ethernet.hpp"
#include "pc_identifier.hpp"
 
namespace license {
using namespace std;
static std::unordered_map<LCC_API_IDENTIFICATION_STRATEGY, std::unique_ptr<IdentificationStrategy>> init() {
    unordered_map<LCC_API_IDENTIFICATION_STRATEGY, std::unique_ptr<IdentificationStrategy>> strategy_map;
    strategy_map[STRATEGY_DEFAULT] =
        unique_ptr<IdentificationStrategy>(dynamic_cast<IdentificationStrategy*>(new DefaultStrategy()));
    strategy_map[STRATEGY_ETHERNET] =
        unique_ptr<IdentificationStrategy>(dynamic_cast<IdentificationStrategy*>(new Ethernet(false)));
    strategy_map[STRATEGY_IP_ADDRESS] =
        unique_ptr<IdentificationStrategy>(dynamic_cast<IdentificationStrategy*>(new Ethernet(true)));
 
    return strategy_map;
}
 
std::unordered_map<LCC_API_IDENTIFICATION_STRATEGY, std::unique_ptr<IdentificationStrategy>>
    PcIdentifierFacade::STRATEGY_MAP = init();
 
LCC_EVENT_TYPE PcIdentifierFacade::validate_pc_signature(const std::string& str_code) {
    PcIdentifier pc_id(str_code);
    LCC_API_IDENTIFICATION_STRATEGY id_strategy = pc_id.get_identification_strategy();
    auto it = STRATEGY_MAP.find(id_strategy);
    LCC_EVENT_TYPE result = IDENTIFIERS_MISMATCH;
    if (it != STRATEGY_MAP.end()) {
        result = it->second->validate_identifier(pc_id);
    } else {
        // TODO: log
    }
    return result;
}
 
std::string PcIdentifierFacade::generate_user_pc_signature(LCC_API_IDENTIFICATION_STRATEGY strategy) {
    bool use_env_var = false;
    if (strategy == STRATEGY_DEFAULT) {
        char* env_var_value = getenv(LCC_IDENTIFICATION_STRATEGY_ENV_VAR);
        if (env_var_value != nullptr && env_var_value[0] != '\0') {
            int strategy_int = atoi(env_var_value);
            if (strategy_int < 0 || strategy_int > 3) {
                LOG_WARN("unknown " LCC_IDENTIFICATION_STRATEGY_ENV_VAR " %s", env_var_value);
            } else {
                strategy = (LCC_API_IDENTIFICATION_STRATEGY)strategy_int;
                use_env_var = true;
            }
        }
    }
    auto it = STRATEGY_MAP.find(strategy);
    PcIdentifier pc_id;
    if (it != STRATEGY_MAP.end()) {
        FUNCTION_RETURN result = it->second->identify_pc(pc_id);
        if (result != FUNC_RET_OK) {
            /// FIXME
        }
    } else {
        throw logic_error("Specified identification strategy invalid");
    }
    ExecutionEnvironment exec;
    VIRTUALIZATION virtualization = exec.getVirtualization();
    pc_id.set_virtual_environment(virtualization);
    pc_id.set_use_environment_var(use_env_var);
    if (virtualization != NONE) {
        bool isCloud = exec.is_cloud();
        if (isCloud) {
            pc_id.set_cloud_provider(exec.getCloudProvider());
        } else {
            CpuInfo cpu;
            pc_id.set_virtualization(cpu.getVirtualizationDetail());
        }
    }
    return pc_id.print();
}
} /* namespace license */