From 95d1452eefadffaf1ec75dd0a8336bc2c387eb17 Mon Sep 17 00:00:00 2001
From: gcontini <1121667+gcontini@users.noreply.github.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