james/yubikey.rs
Public Access
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Tests for ykpiv_list_readers and ykpiv_set_mgmkey

Browse Source
This commit is contained in:
Trevor Bentley
2017-07-24 14:10:13 +02:00
parent 2ea0e4cbdd
commit 89e8e7864e
1 changed files with 48 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/tests/util.c
+48 -2
View File
@@ -81,7 +81,7 @@ void teardown(void) {
START_TEST(test_devicemodel) { START_TEST(test_devicemodel) {
ykpiv_devmodel model; ykpiv_devmodel model;
model = ykpiv_util_devicemodel(g_state); model = ykpiv_util_devicemodel(g_state);
fprintf(stderr, "Model: %u\n", model); fprintf(stdout, "Model: %u\n", model);
ck_assert(model == DEVTYPE_YK4); ck_assert(model == DEVTYPE_YK4);
} }
END_TEST END_TEST
@@ -103,6 +103,20 @@ START_TEST(test_get_set_cardid) {
} }
END_TEST END_TEST
START_TEST(test_list_readers) {
ykpiv_rc res;
char reader_buf[2048];
size_t num_readers = sizeof(reader_buf);
char *reader_ptr;
res = ykpiv_list_readers(g_state, reader_buf, &num_readers);
ck_assert_int_eq(res, YKPIV_OK);
ck_assert_int_gt(num_readers, 0);
for(reader_ptr = reader_buf; *reader_ptr != '\0'; reader_ptr += strlen(reader_ptr) + 1) {
fprintf(stdout, "Found device: %s\n", reader_ptr);
}
}
END_TEST
START_TEST(test_read_write_list_delete_cert) { START_TEST(test_read_write_list_delete_cert) {
ykpiv_rc res; ykpiv_rc res;
uint8_t *read_cert = NULL; uint8_t *read_cert = NULL;
@@ -184,13 +198,44 @@ END_TEST
START_TEST(test_authenticate) { START_TEST(test_authenticate) {
ykpiv_rc res; ykpiv_rc res;
const char *mgm_key = "010203040506070801020304050607080102030405060708"; const char *default_mgm_key = "010203040506070801020304050607080102030405060708";
const char *mgm_key = "112233445566778811223344556677881122334455667788";
unsigned char key[24]; unsigned char key[24];
size_t key_len = sizeof(key); size_t key_len = sizeof(key);
// Try new key, fail.
res = ykpiv_hex_decode(mgm_key, strlen(mgm_key), key, &key_len); res = ykpiv_hex_decode(mgm_key, strlen(mgm_key), key, &key_len);
ck_assert_int_eq(res, YKPIV_OK); ck_assert_int_eq(res, YKPIV_OK);
res = ykpiv_authenticate(g_state, key);
ck_assert_int_eq(res, YKPIV_AUTHENTICATION_ERROR);
// Try default key, succeed
res = ykpiv_hex_decode(default_mgm_key, strlen(default_mgm_key), key, &key_len);
ck_assert_int_eq(res, YKPIV_OK);
res = ykpiv_authenticate(g_state, key);
ck_assert_int_eq(res, YKPIV_OK);
// Change to new key
res = ykpiv_hex_decode(mgm_key, strlen(mgm_key), key, &key_len);
ck_assert_int_eq(res, YKPIV_OK);
res = ykpiv_set_mgmkey(g_state, key);
ck_assert_int_eq(res, YKPIV_OK);
// Try new key, succeed.
res = ykpiv_hex_decode(mgm_key, strlen(mgm_key), key, &key_len);
ck_assert_int_eq(res, YKPIV_OK);
res = ykpiv_authenticate(g_state, key);
ck_assert_int_eq(res, YKPIV_OK);
// Change back to default key
res = ykpiv_hex_decode(default_mgm_key, strlen(default_mgm_key), key, &key_len);
ck_assert_int_eq(res, YKPIV_OK);
res = ykpiv_set_mgmkey(g_state, key);
ck_assert_int_eq(res, YKPIV_OK);
// Try default key, succeed
res = ykpiv_hex_decode(default_mgm_key, strlen(default_mgm_key), key, &key_len);
ck_assert_int_eq(res, YKPIV_OK);
res = ykpiv_authenticate(g_state, key); res = ykpiv_authenticate(g_state, key);
ck_assert_int_eq(res, YKPIV_OK); ck_assert_int_eq(res, YKPIV_OK);
} }
@@ -347,6 +392,7 @@ Suite *test_suite(void) {
// Test util functionality // Test util functionality
tcase_add_test(tc, test_devicemodel); tcase_add_test(tc, test_devicemodel);
tcase_add_test(tc, test_get_set_cardid); tcase_add_test(tc, test_get_set_cardid);
tcase_add_test(tc, test_list_readers);
tcase_add_test(tc, test_read_write_list_delete_cert); tcase_add_test(tc, test_read_write_list_delete_cert);
tcase_add_test(tc, test_generate_key); tcase_add_test(tc, test_generate_key);