blob: 235a24db1ad2d55181b825c1942cda41b2ed5e45 (
plain)
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
static const char *const default_admin_pin = "12345678";
static const char *const default_user_pin = "123456";
const char * temporary_password = "123456789012345678901234";
const char * RFC_SECRET = "12345678901234567890";
#include "catch.hpp"
#include <iostream>
#include <NitrokeyManager.h>
#include <iostream>
#include <stick20_commands.h>
using namespace nitrokey;
TEST_CASE("List devices", "[BASIC]") {
shared_ptr<Stick20> d = make_shared<Stick20>();
auto v = d->enumerate();
REQUIRE(v.size() > 0);
for (auto a : v){
std::cout << a;
d->set_path(a);
d->connect();
auto res = GetStatus::CommandTransaction::run(d);
auto res2 = GetDeviceStatus::CommandTransaction::run(d);
std::cout << " " << res.data().card_serial_u32 << " "
<< res.data().get_card_serial_hex()
<< " " << std::to_string(res2.data().versionInfo.minor)
<< std::endl;
d->disconnect();
}
}
TEST_CASE("Regenerate AES keys", "[BASIC]") {
shared_ptr<Stick20> d = make_shared<Stick20>();
auto v = d->enumerate();
REQUIRE(v.size() > 0);
std::vector<shared_ptr<Stick20>> devices;
for (auto a : v){
std::cout << a << endl;
d = make_shared<Stick20>();
d->set_path(a);
d->connect();
devices.push_back(d);
}
for (auto d : devices){
auto res2 = GetDeviceStatus::CommandTransaction::run(d);
std::cout << std::to_string(res2.data().versionInfo.minor) << std::endl;
// nitrokey::proto::stick20::CreateNewKeys::CommandPayload p;
// p.set_defaults();
// memcpy(p.password, "12345678", 8);
// auto res3 = nitrokey::proto::stick20::CreateNewKeys::CommandTransaction::run(d, p);
}
for (auto d : devices){
//TODO watch out for multiple hid_exit calls
d->disconnect();
}
}
TEST_CASE("Use API", "[BASIC]") {
auto nm = NitrokeyManager::instance();
nm->set_loglevel(2);
auto v = nm->list_devices();
REQUIRE(v.size() > 0);
for (int i=0; i<10; i++){
for (auto a : v) {
std::cout <<"Connect with: " << a <<
" " << std::boolalpha << nm->connect_with_path(a) << " ";
try{
auto status_storage = nm->get_status_storage();
std::cout << status_storage.ActiveSmartCardID_u32
<< " " << status_storage.ActiveSD_CardID_u32
<< std::endl;
// nm->fill_SD_card_with_random_data("12345678");
}
catch (const LongOperationInProgressException &e){
std::cout << "long operation in progress on " << a
<< " " << std::to_string(e.progress_bar_value) << std::endl;
// this_thread::sleep_for(1000ms);
}
}
std::cout <<"Iteration: " << i << std::endl;
}
}
TEST_CASE("Use API ID", "[BASIC]") {
auto nm = NitrokeyManager::instance();
nm->set_loglevel(2);
auto v = nm->list_devices_by_cpuID();
REQUIRE(v.size() > 0);
for(int i=0; i<1000; i++) {
auto v = nm->list_devices_by_cpuID();
REQUIRE(v.size() > 0);
for (auto i : v) {
nm->connect_with_ID(i);
auto retry_count = nm->get_admin_retry_count();
std::cout << i << " " << to_string(retry_count) << std::endl;
}
}
std::cout << "finished" << std::endl;
}
|