From bcbb54f49fad3f4bb3e8e6aca081d9d300e74d2b Mon Sep 17 00:00:00 2001 From: Szczepan Zalega Date: Thu, 8 Sep 2016 16:05:11 +0200 Subject: Function for getting device's serial number in hex. Fixes #33 Signed-off-by: Szczepan Zalega --- NK_C_API.cc | 9 +++++++++ NK_C_API.h | 6 ++++++ NitrokeyManager.cc | 7 ++++++- include/NitrokeyManager.h | 1 + include/stick10_commands.h | 9 +++++++-- unittest/test_bindings.py | 7 +++++++ 6 files changed, 36 insertions(+), 3 deletions(-) diff --git a/NK_C_API.cc b/NK_C_API.cc index b755c70..6265215 100644 --- a/NK_C_API.cc +++ b/NK_C_API.cc @@ -175,6 +175,15 @@ extern const char * NK_status() { }); } +extern const char * NK_device_serial_number(){ + auto m = NitrokeyManager::instance(); + return get_with_string_result([&](){ + string && s = m->get_serial_number(); + char * rs = strdup(s.c_str()); + clear_string(s); + return rs; + }); +} extern uint32_t NK_get_hotp_code(uint8_t slot_number) { return NK_get_hotp_code_PIN(slot_number, ""); diff --git a/NK_C_API.h b/NK_C_API.h index 5960885..3c851a5 100644 --- a/NK_C_API.h +++ b/NK_C_API.h @@ -39,6 +39,12 @@ extern int NK_logout(); */ extern const char * NK_status(); +/** + * Return the device's serial number string in hex. + * @return string device's serial number in hex + */ +extern const char * NK_device_serial_number(); + /** * Get last command processing status. Useful for commands which returns the results of their own and could not return * an error code. diff --git a/NitrokeyManager.cc b/NitrokeyManager.cc index 161f7aa..5b648b5 100644 --- a/NitrokeyManager.cc +++ b/NitrokeyManager.cc @@ -88,6 +88,11 @@ namespace nitrokey{ } } + string NitrokeyManager::get_serial_number() { + auto response = GetStatus::CommandTransaction::run(*device); + return response.data().get_card_serial_hex(); + } + string NitrokeyManager::get_status() { auto response = GetStatus::CommandTransaction::run(*device); return response.data().dissect(); @@ -410,4 +415,4 @@ namespace nitrokey{ return true; } -} \ No newline at end of file +} diff --git a/include/NitrokeyManager.h b/include/NitrokeyManager.h index d2ea991..1e518f4 100644 --- a/include/NitrokeyManager.h +++ b/include/NitrokeyManager.h @@ -40,6 +40,7 @@ namespace nitrokey { bool disconnect(); void set_debug(bool state); string get_status(); + string get_serial_number(); const char * get_totp_slot_name(uint8_t slot_number); const char * get_hotp_slot_name(uint8_t slot_number); diff --git a/include/stick10_commands.h b/include/stick10_commands.h index ef83747..a60be59 100644 --- a/include/stick10_commands.h +++ b/include/stick10_commands.h @@ -324,15 +324,20 @@ class GetStatus : Command { }; bool isValid() const { return enable_user_password!=delete_user_password; } + std::string get_card_serial_hex() const { + return ::nitrokey::misc::hexdump((const char *)(card_serial), + sizeof card_serial, false); + } + std::string dissect() const { std::stringstream ss; ss << "firmware_version:\t" << firmware_version << std::endl; ss << "card_serial:\t" << ::nitrokey::misc::hexdump((const char *)(card_serial), - sizeof card_serial); + sizeof card_serial, false); ss << "general_config:\t" << ::nitrokey::misc::hexdump((const char *)(general_config), - sizeof general_config); + sizeof general_config, false); ss << "numlock:\t" << (int)numlock << std::endl; ss << "capslock:\t" << (int)capslock << std::endl; ss << "scrolllock:\t" << (int)scrolllock << std::endl; diff --git a/unittest/test_bindings.py b/unittest/test_bindings.py index 2ffa046..eeda247 100644 --- a/unittest/test_bindings.py +++ b/unittest/test_bindings.py @@ -496,3 +496,10 @@ def test_get_status(C): status = C.NK_status() s = gs(status) assert len(s) > 0 + + +def test_get_serial_number(C): + sn = C.NK_device_serial_number() + sn = gs(sn) + assert len(sn) > 0 + print(('Serial number of the device: ', sn)) -- cgit v1.2.3