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/disk_strategy.cpp | 97 ++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 97 insertions(+), 0 deletions(-)
diff --git a/src/library/hw_identifier/disk_strategy.cpp b/src/library/hw_identifier/disk_strategy.cpp
new file mode 100644
index 0000000..d9f6831
--- /dev/null
+++ b/src/library/hw_identifier/disk_strategy.cpp
@@ -0,0 +1,97 @@
+/*
+ * disk_strategy.cpp
+ *
+ * Created on: Jan 14, 2020
+ * Author: devel
+ */
+#include <string.h>
+#include "../os/os.h"
+#include "disk_strategy.hpp"
+
+using namespace std;
+namespace license {
+namespace hw_identifier {
+
+static FUNCTION_RETURN generate_disk_pc_id(vector<array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA>> &v_disk_id,
+ bool use_id) {
+ size_t disk_num = 0;
+ size_t available_disk_info = 0;
+ FUNCTION_RETURN result_diskinfos;
+
+ result_diskinfos = getDiskInfos(nullptr, &disk_num);
+ if (result_diskinfos != FUNC_RET_OK && result_diskinfos != FUNC_RET_BUFFER_TOO_SMALL) {
+ return result_diskinfos;
+ }
+ if (disk_num == 0) {
+ return FUNC_RET_NOT_AVAIL;
+ }
+ size_t mem = disk_num * sizeof(DiskInfo);
+ DiskInfo *diskInfos = (DiskInfo *)malloc(mem);
+ if (diskInfos == nullptr) {
+ return FUNC_RET_NOT_AVAIL;
+ }
+ memset(diskInfos, 0, mem);
+ result_diskinfos = getDiskInfos(diskInfos, &disk_num);
+
+ if (result_diskinfos != FUNC_RET_OK) {
+ free(diskInfos);
+ return result_diskinfos;
+ }
+ for (unsigned int i = 0; i < disk_num; i++) {
+ char firstChar = use_id ? diskInfos[i].label[0] : diskInfos[i].disk_sn[0];
+ available_disk_info += firstChar == 0 ? 0 : 1;
+ }
+ if (available_disk_info == 0) {
+ free(diskInfos);
+ return FUNC_RET_NOT_AVAIL;
+ }
+ v_disk_id.reserve(available_disk_info);
+ for (int j = 0; j < 2; j++) {
+ int preferred = (j == 0 ? 1 : 0);
+ for (unsigned int i = 0; i < disk_num; i++) {
+ array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA> a_disk_id;
+ a_disk_id.fill(0);
+ if (use_id) {
+ if (diskInfos[i].disk_sn[0] != 0 && diskInfos[i].preferred == preferred) {
+ size_t size = min(a_disk_id.size(), sizeof(&diskInfos[i].disk_sn));
+ memcpy(&a_disk_id[0], diskInfos[i].disk_sn, size);
+ v_disk_id.push_back(a_disk_id);
+ }
+ } else {
+ if (diskInfos[i].label[0] != 0 && diskInfos[i].preferred == preferred) {
+ strncpy((char *)&a_disk_id[0], diskInfos[i].label, a_disk_id.size() - 1);
+ v_disk_id.push_back(a_disk_id);
+ }
+ }
+ }
+ }
+ free(diskInfos);
+ return FUNC_RET_OK;
+}
+
+DiskStrategy::DiskStrategy(bool use_id) : m_use_id(use_id) {}
+
+DiskStrategy::~DiskStrategy() {}
+
+LCC_API_IDENTIFICATION_STRATEGY DiskStrategy::identification_strategy() const {
+ return m_use_id ? STRATEGY_DISK_NUM : STRATEGY_DISK_LABEL;
+}
+
+std::vector<HwIdentifier> DiskStrategy::alternative_ids() const {
+ vector<array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA>> data;
+ FUNCTION_RETURN result = generate_disk_pc_id(data, m_use_id);
+ vector<HwIdentifier> identifiers;
+ if (result == FUNC_RET_OK) {
+ identifiers.reserve(data.size());
+ for (auto &it : data) {
+ HwIdentifier pc_id;
+ pc_id.set_identification_strategy(identification_strategy());
+ pc_id.set_data(it);
+ identifiers.push_back(pc_id);
+ }
+ }
+ return identifiers;
+}
+
+} // namespace hw_identifier
+} /* namespace license */
--
Gitblit v1.9.1