disk strategy failing on windows

gcontini

2020-10-25 4b7e3271e3ad8b2fda6fadead733a3fa9bad8dab

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
/*
 * 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 array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA> generate_id_by_sn(const DiskInfo &disk_info) {
    array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA> a_disk_id;
    a_disk_id.fill(0);
    size_t size = min(a_disk_id.size(), sizeof(disk_info.disk_sn));
    memcpy(&a_disk_id[0], disk_info.disk_sn, size);
 
    return a_disk_id;
}
 
static array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA> generate_id_by_label(const DiskInfo &disk_info) {
    array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA> a_disk_id;
    a_disk_id.fill(0);
    strncpy((char *)&a_disk_id[0], disk_info.label, a_disk_id.size() - 1);
    return a_disk_id;
}
 
static FUNCTION_RETURN generate_disk_pc_id(vector<array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA>> &v_disk_id) {
    std::vector<DiskInfo> disk_infos;
    FUNCTION_RETURN result_diskinfos = getDiskInfos(disk_infos);
    if (result_diskinfos != FUNC_RET_OK && result_diskinfos != FUNC_RET_BUFFER_TOO_SMALL) {
        return result_diskinfos;
    }
    if (disk_infos.size() == 0) {
        return FUNC_RET_NOT_AVAIL;
    }
 
    v_disk_id.reserve(disk_infos.size() * 2);
    for (int j = 0; j < 2; j++) {
        bool preferred = (j == 0);
        for (size_t i = 0; i < disk_infos.size(); i++) {
            if (disk_infos[i].preferred == preferred) {
                if (disk_infos[i].sn_initialized) {
                    array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA> a_disk_id = generate_id_by_sn(disk_infos[i]);
                    v_disk_id.push_back(a_disk_id);
                }
                if (disk_infos[i].label_initialized) {
                    array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA> l_disk_id = generate_id_by_label(disk_infos[i]);
                    v_disk_id.push_back(l_disk_id);
                }
            }
        }
    }
    return v_disk_id.size() > 0 ? FUNC_RET_OK : FUNC_RET_NOT_AVAIL;
}
 
DiskStrategy::~DiskStrategy() {}
 
LCC_API_HW_IDENTIFICATION_STRATEGY DiskStrategy::identification_strategy() const {
    return LCC_API_HW_IDENTIFICATION_STRATEGY::STRATEGY_DISK;
}
 
std::vector<HwIdentifier> DiskStrategy::alternative_ids() const {
    vector<array<uint8_t, HW_IDENTIFIER_PROPRIETARY_DATA>> data;
    FUNCTION_RETURN result = generate_disk_pc_id(data);
    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 */