From 36ce07093b68b07513149577c209ae7a57ab356b Mon Sep 17 00:00:00 2001
From: Gabriele Contini <contini.mailing@gmail.com>
Date: 周日, 15 3月 2020 16:26:21 +0800
Subject: [PATCH] Merge branch 'feature/pc_identifiers' into develop issues #2 #3 #14 #49

---
 src/library/hw_identifier/hw_identifier.cpp |   99 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 99 insertions(+), 0 deletions(-)

diff --git a/src/library/hw_identifier/hw_identifier.cpp b/src/library/hw_identifier/hw_identifier.cpp
new file mode 100644
index 0000000..5c163e7
--- /dev/null
+++ b/src/library/hw_identifier/hw_identifier.cpp
@@ -0,0 +1,99 @@
+/*
+ * hw_identifier.cpp
+ *
+ *  Created on: Dec 22, 2019
+ *      Author: GC
+ */
+
+#include <algorithm>
+#include "hw_identifier.hpp"
+#include "../base/base64.h"
+
+namespace license {
+namespace hw_identifier {
+
+using namespace std;
+
+HwIdentifier::HwIdentifier() {}
+
+HwIdentifier::HwIdentifier(const std::string& param) {
+	string tmp_str(param);  // throw away const
+	std::replace(tmp_str.begin(), tmp_str.end(), '-', '\n');
+	vector<uint8_t> decoded = unbase64(tmp_str);
+	if (decoded.size() != HW_IDENTIFIER_PROPRIETARY_DATA + 1) {
+		cerr << decoded.size();
+		throw logic_error("wrong identifier size " + param);
+	}
+	std::copy_n(decoded.begin(), HW_IDENTIFIER_PROPRIETARY_DATA + 1, m_data.begin());
+}
+
+HwIdentifier::~HwIdentifier() {}
+
+HwIdentifier::HwIdentifier(const HwIdentifier& other) : m_data(other.m_data) {}
+
+void HwIdentifier::set_identification_strategy(LCC_API_IDENTIFICATION_STRATEGY strategy) {
+	if (strategy == STRATEGY_NONE || strategy == STRATEGY_DEFAULT) {
+		throw logic_error("Only known strategies are permitted");
+	}
+	uint8_t stratMov = (strategy << 5);
+	m_data[1] = (m_data[1] & 0x1F) | stratMov;
+}
+
+void HwIdentifier::set_use_environment_var(bool use_env_var) {
+	if (use_env_var) {
+		m_data[0] = m_data[0] | 0x40;
+	} else {
+		m_data[0] = m_data[0] & ~0x40;
+	}
+}
+
+void HwIdentifier::set_virtual_environment(os::VIRTUALIZATION virt) {
+	// 110000 0x30
+	m_data[0] = (m_data[0] & ~0x30) | virt << 4;
+}
+
+void HwIdentifier::set_virtualization(os::VIRTUALIZATION_DETAIL virtualization_detail) {
+	m_data[0] = (m_data[0] & ~0x0F) | virtualization_detail;
+}
+
+void HwIdentifier::set_cloud_provider(os::CLOUD_PROVIDER cloud_provider) {
+	m_data[0] = (m_data[0] & ~0x0F) | cloud_provider | 0x08;
+}
+
+void HwIdentifier::set_data(const std::array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA>& data) {
+	m_data[1] = (m_data[1] & (~0x1f)) | (data[0] & 0x1f);
+	for (int i = 1; i < HW_IDENTIFIER_PROPRIETARY_DATA; i++) {
+		m_data[i + 1] = data[i];
+	}
+}
+
+std::string HwIdentifier::print() const {
+	string result = base64(m_data.data(), m_data.size(), 5);
+	std::replace(result.begin(), result.end(), '\n', '-');
+	return result.substr(0, result.size() - 1);
+}
+
+LCC_API_IDENTIFICATION_STRATEGY HwIdentifier::get_identification_strategy() const {
+	uint8_t stratMov = m_data[1] >> 5;
+	return static_cast<LCC_API_IDENTIFICATION_STRATEGY>(stratMov);
+}
+
+bool HwIdentifier::data_match(const std::array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA>& data) const {
+	bool equals = true;
+	for (int i = 0; i < HW_IDENTIFIER_PROPRIETARY_DATA && equals; i++) {
+		equals = (i == 0) ? ((data[i] & 0x1f) == (m_data[i + 1] & 0x1f)) : (data[i] == m_data[i + 1]);
+	}
+	return equals;
+}
+
+bool operator==(const HwIdentifier& lhs, const HwIdentifier& rhs) {
+	bool equals = lhs.get_identification_strategy() == rhs.get_identification_strategy();
+	for (int i = 0; i < HW_IDENTIFIER_PROPRIETARY_DATA && equals; i++) {
+		equals = (i == 0) ? ((rhs.m_data[i + 1] & 0x1f) == (lhs.m_data[i + 1] & 0x1f))
+						  : (lhs.m_data[i + 1] == rhs.m_data[i + 1]);
+	}
+	return equals;
+}
+
+}  // namespace hw_identifier
+} /* namespace license */

--
Gitblit v1.9.1