From 32487505f658b61a768680e9b4aadf616726ef12 Mon Sep 17 00:00:00 2001
From: Gabriele Contini <contini.mailing@gmail.com>
Date: 周日, 09 2月 2020 04:48:55 +0800
Subject: [PATCH] identifiers next work
---
src/library/pc_identifier/pc_identifier_facade.cpp | 69 +++++++++++++++++++++++++++++-----
1 files changed, 58 insertions(+), 11 deletions(-)
diff --git a/src/library/pc_identifier/pc_identifier_facade.cpp b/src/library/pc_identifier/pc_identifier_facade.cpp
index 46263f6..a9aa9b8 100644
--- a/src/library/pc_identifier/pc_identifier_facade.cpp
+++ b/src/library/pc_identifier/pc_identifier_facade.cpp
@@ -4,27 +4,74 @@
* Created on: Dec 26, 2019
* Author: devel
*/
-#include <unordered_map>
-#include <bits/unique_ptr.h>
#include "pc_identifier_facade.hpp"
+
+#include <cstdlib>
+#include <stdexcept>
+
+#include "../base/base.h"
+#include "../base/logger.h"
+#include "../os/cpu_info.hpp"
+#include "../os/execution_environment.hpp"
#include "identification_strategy.hpp"
#include "pc_identifier.hpp"
-#include "../os/execution_environment.hpp"
+
namespace license {
+namespace pc_identifier {
-std::unordered_map<IDENTIFICATION_STRATEGY, std::unique_ptr<IdentificationStrategy>> STRATEGY_MAP;
+using namespace std;
-EVENT_TYPE PcIdentifierFacade::validate_pc_signature(const std::string& str_code) {
+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();
- auto it = STRATEGY_MAP.find(id_strategy);
- EVENT_TYPE result = IDENTIFIERS_MISMATCH;
- if (it != STRATEGY_MAP.end()) {
- result = it->second->validate_identifier(pc_id);
+ LCC_API_IDENTIFICATION_STRATEGY id_strategy = pc_id.get_identification_strategy();
+ LCC_EVENT_TYPE result = IDENTIFIERS_MISMATCH;
+ try {
+ unique_ptr<IdentificationStrategy> strategy = IdentificationStrategy::get_strategy(id_strategy);
+ result = strategy->validate_identifier(pc_id);
+ } catch (logic_error& e) {
+ LOG_ERROR("Error validating identifier %s: %s", str_code.c_str(), e.what());
}
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;
+ vector<LCC_API_IDENTIFICATION_STRATEGY> strategies_to_try;
+ 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;
+ }
+ }
+ }
+
+ unique_ptr<IdentificationStrategy> strategy_ptr = IdentificationStrategy::get_strategy(strategy);
+ PcIdentifier pc_id;
+ FUNCTION_RETURN result = strategy_ptr->identify_pc(pc_id);
+ if (result != FUNC_RET_OK) {
+ /// FIXME
+ }
+ 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 pc_identifier
} /* namespace license */
--
Gitblit v1.9.1