aboutsummaryrefslogtreecommitdiff
path: root/device.cc
blob: 3aca2b25dca3a940867eb6e258a85f5174a93ed2 (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
#include <chrono>
#include <thread>
#include <cstddef>
#include <stdexcept>
#include <hidapi/hidapi.h>
#include "device.h"
#include "log.h"
#include "misc.h"

using namespace nitrokey::device;
using namespace nitrokey::log;

Device::Device()
: m_vid(0), m_pid(0),
  m_retry_count(4),
  m_retry_timeout(50),
  mp_devhandle(NULL) {}

bool Device::disconnect() {
	Log::instance()(__PRETTY_FUNCTION__, Loglevel::DEBUG_L2);

	hid_exit();
	mp_devhandle = NULL;
	return true;
}
bool Device::connect() {
	Log::instance()(__PRETTY_FUNCTION__, Loglevel::DEBUG_L2);

	hid_init();
	mp_devhandle = hid_open(m_vid, m_pid, NULL);
	hid_init();
	return mp_devhandle != NULL;
}

CommError Device::send(const void *packet) {
	Log::instance()(__PRETTY_FUNCTION__, Loglevel::DEBUG_L2);

	if (mp_devhandle == NULL)
		throw std::runtime_error("Attempted HID send on an invalid descriptor.");

	return (CommError)(
		hid_send_feature_report(mp_devhandle, (const unsigned char *)(packet),
			HID_REPORT_SIZE));
}

CommError Device::recv(void *packet) {
	CommError status;
	int retry_count = 0;

	Log::instance()(__PRETTY_FUNCTION__, Loglevel::DEBUG_L2);

	if (mp_devhandle == NULL)
		throw std::runtime_error("Attempted HID receive on an invalid descriptor.");

	for(;;) {
		status = (CommError)(
			hid_get_feature_report(mp_devhandle, (unsigned char *)(packet),
				HID_REPORT_SIZE));
		if ((int)status >0 || retry_count++ >= m_retry_count)
			break;
		std::this_thread::sleep_for(m_retry_timeout);
	}

	return status;
}

Stick10::Stick10() {
	m_vid = 0x20a0;
	m_pid = 0x4108;
}

Stick20::Stick20() {
	m_vid = 0x20a0;
	m_pid = 0x4109;
}