Let device identify itself by model

Signed-off-by: Szczepan Zalega <szczepan@nitrokey.com>

2 files changed, 10 insertions, 3 deletions

diff --git a/device.cc b/device.cc
index 0427d29..4f11e23 100644
--- a/device.cc
+++ b/device.cc
@@ -45,8 +45,6 @@ int Device::send(const void *packet) {
 }
 
 int Device::recv(void *packet) {
-  // FIXME change CommError return value to int (return value of
-  // hid_get_feature_report)
   int status;
   int retry_count = 0;
 
@@ -86,10 +84,13 @@ int Device::recv(void *packet) {
 Stick10::Stick10() {
   m_vid = 0x20a0;
   m_pid = 0x4108;
+  m_model = DeviceModel::PRO;
 }
 
 Stick20::Stick20() {
   m_vid = 0x20a0;
   m_pid = 0x4109;
   m_retry_timeout = std::chrono::milliseconds(500);
+  m_model = DeviceModel::STORAGE;
+
 }
diff --git a/include/device.h b/include/device.h
index 1c95c56..1f6d486 100644
--- a/include/device.h
+++ b/include/device.h
@@ -19,6 +19,11 @@ enum class CommError {
   ERR_STATUS_NOT_OK = -4
 };
 
+enum class DeviceModel{
+    PRO,
+    STORAGE
+};
+
 class Device {
  public:
   Device();
@@ -45,13 +50,14 @@ class Device {
     int get_last_command_status() const;
     void set_last_command_status(uint8_t _err) { last_command_status = _err;} ;
     bool last_command_sucessfull() const {return last_command_status == 0;};
-
+    DeviceModel get_device_model() const {return m_model;}
 private:
     uint8_t last_command_status;
 
  protected:
   uint16_t m_vid;
   uint16_t m_pid;
+    DeviceModel m_model;
 
   /*
    *	While the project uses Signal11 portable HIDAPI