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
|
#define CATCH_CONFIG_MAIN // This tells Catch to provide a main()
#include "catch.hpp"
#include <iostream>
#include <string.h>
#include <NitrokeyManager.h>
#include "device_proto.h"
#include "log.h"
//#include "stick10_commands.h"
#include "stick20_commands.h"
using namespace std;
using namespace nitrokey::device;
using namespace nitrokey::proto;
using namespace nitrokey::proto::stick20;
using namespace nitrokey::log;
using namespace nitrokey::misc;
template<typename CMDTYPE>
void execute_password_command(Device &stick, const char *password, const char kind = 'P') {
auto p = get_payload<CMDTYPE>();
if (kind == 'P'){
p.set_kind_user();
} else {
p.set_kind_admin();
}
strcpyT(p.password, password);
CMDTYPE::CommandTransaction::run(stick, p);
this_thread::sleep_for(1000ms);
}
TEST_CASE("test", "[test]") {
Stick20 stick;
bool connected = stick.connect();
REQUIRE(connected == true);
Log::instance().set_loglevel(Loglevel::DEBUG_L2);
stick10::LockDevice::CommandTransaction::run(stick);
// execute_password_command<EnableEncryptedPartition>(stick, "123456");
// execute_password_command<DisableEncryptedPartition>(stick, "123456");
this_thread::sleep_for(1000ms);
execute_password_command<EnableEncryptedPartition>(stick, "123456");
this_thread::sleep_for(4000ms);
bool passed = false;
for(int i=0; i<5; i++){
try {
execute_password_command<EnableHiddenEncryptedPartition>(stick, "123123123");
CHECK(true);
passed=true;
break;
}
catch (CommandFailedException &e){
this_thread::sleep_for(2000ms);
}
}
if(!passed){
CHECK(false);
}
execute_password_command<DisableHiddenEncryptedPartition>(stick, "123123123");
execute_password_command<SendSetReadonlyToUncryptedVolume>(stick, "123456");
execute_password_command<SendSetReadwriteToUncryptedVolume>(stick, "123456");
execute_password_command<SendClearNewSdCardFound>(stick, "12345678", 'A');
this_thread::sleep_for(1000ms);
// execute_password_command<LockFirmware>(stick, "123123123"); //CAUTION
// execute_password_command<EnableFirmwareUpdate>(stick, "123123123"); //CAUTION FIRMWARE PIN
execute_password_command<ExportFirmware>(stick, "12345678", 'A');
// execute_password_command<FillSDCardWithRandomChars>(stick, "12345678", 'A');
stick10::LockDevice::CommandTransaction::run(stick);
stick.disconnect();
}
|
