From 1d1082fe695366da3498f528b953203a90fe7385 Mon Sep 17 00:00:00 2001 From: Gabriele Contini <contini.mailing@gmail.com> Date: 周六, 01 2月 2020 02:48:51 +0800 Subject: [PATCH] rename enums and further developments --- src/library/pc_identifier/pc_identifier_facade.cpp | 79 +++++++++++++++++++++++++++++++++++---- 1 files changed, 70 insertions(+), 9 deletions(-) diff --git a/src/library/pc_identifier/pc_identifier_facade.cpp b/src/library/pc_identifier/pc_identifier_facade.cpp index 46263f6..1e0e93c 100644 --- a/src/library/pc_identifier/pc_identifier_facade.cpp +++ b/src/library/pc_identifier/pc_identifier_facade.cpp @@ -4,27 +4,88 @@ * Created on: Dec 26, 2019 * Author: devel */ -#include <unordered_map> -#include <bits/unique_ptr.h> #include "pc_identifier_facade.hpp" -#include "identification_strategy.hpp" -#include "pc_identifier.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))); -std::unordered_map<IDENTIFICATION_STRATEGY, std::unique_ptr<IdentificationStrategy>> STRATEGY_MAP; + return strategy_map; +} -EVENT_TYPE PcIdentifierFacade::validate_pc_signature(const std::string& str_code) { +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); - IDENTIFICATION_STRATEGY id_strategy = pc_id.get_identification_strategy(); + LCC_API_IDENTIFICATION_STRATEGY id_strategy = pc_id.get_identification_strategy(); auto it = STRATEGY_MAP.find(id_strategy); - EVENT_TYPE result = IDENTIFIERS_MISMATCH; + 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(IDENTIFICATION_STRATEGY strategy) { PcIdentifier pc_id; } +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 */ -- Gitblit v1.9.1