Test cases for ykpiv_util_* functions

2017-07-10 17:16:05 +02:00
parent 6c5d5545bf
commit cffe862bbc
2 changed files with 212 additions and 1 deletions
@@ -31,7 +31,7 @@ AM_CPPFLAGS = -I$(top_srcdir)/lib -I$(top_builddir)/lib
 AM_LDFLAGS = -no-install @CHECK_LIBS@
 LDADD = ../libykpiv.la

-check_PROGRAMS = basic parse_key
+check_PROGRAMS = basic parse_key util
 TESTS = $(check_PROGRAMS)

 LOG_COMPILER = $(VALGRIND)
@@ -0,0 +1,211 @@
+/*
+ * Copyright (c) 2014-2016 Yubico AB
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ *
+ *   * Redistributions of source code must retain the above copyright
+ *     notice, this list of conditions and the following disclaimer.
+ *
+ *   * Redistributions in binary form must reproduce the above
+ *     copyright notice, this list of conditions and the following
+ *     disclaimer in the documentation and/or other materials provided
+ *     with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ */
+
+#include "ykpiv.h"
+#include "internal.h"
+
+#include <stddef.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include <check.h>
+
+ykpiv_state *g_state;
+const uint8_t g_cert[] = {
+  "0123456789ABCDEFGHIK0123456789ABCDEFGHIK0123456789ABCDEFGHIK0123456789ABCDEFGHIK"
+  "0123456789ABCDEFGHIK0123456789ABCDEFGHIK0123456789ABCDEFGHIK0123456789ABCDEFGHIK"
+  "0123456789ABCDEFGHIK0123456789ABCDEFGHIK0123456789ABCDEFGHIK0123456789ABCDEFGHIK"
+  "0123456789ABCDEFGHIK0123456789ABCDEFGHIK0123456789ABCDEFGHIK0123456789ABCDEFGHIK"
+  "0123456789ABCDEFGHIK0123456789ABCDEFGHIK0123456789ABCDEFGHIK0123456789ABCDEFGHIK"
+};
+
+void setup(void) {
+  ykpiv_rc res;
+  const char *mgm_key = "010203040506070801020304050607080102030405060708";
+  unsigned char key[24];
+  size_t key_len = sizeof(key);
+
+  res = ykpiv_init(&g_state, true);
+  ck_assert_int_eq(res, YKPIV_OK);
+
+  res = ykpiv_connect(g_state, NULL);
+  ck_assert_int_eq(res, YKPIV_OK);
+
+  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);
+}
+void teardown(void) {
+  ykpiv_done(g_state);
+}
+
+START_TEST(test_devicemodel) {
+  ykpiv_devmodel model;
+  model = ykpiv_util_devicemodel(g_state);
+  fprintf(stderr, "Model: %u\n", model);
+  ck_assert(model == DEVTYPE_YK4);
+}
+END_TEST
+
+START_TEST(test_get_set_cardid) {
+  ykpiv_rc res;
+  ykpiv_cardid set_id;
+  ykpiv_cardid get_id;
+
+  memset(&set_id.data, 'i', sizeof(set_id.data));
+  memset(&get_id.data, 0, sizeof(get_id.data));
+
+  res = ykpiv_util_set_cardid(g_state, &set_id);
+  ck_assert_int_eq(res, YKPIV_OK);
+
+  res = ykpiv_util_get_cardid(g_state, &get_id);
+  ck_assert_int_eq(res, YKPIV_OK);
+  ck_assert_mem_eq(&set_id.data, &get_id.data, sizeof(set_id.data));
+}
+END_TEST
+
+START_TEST(test_read_write_list_delete_cert) {
+  ykpiv_rc res;
+  uint8_t *read_cert = NULL;
+  size_t read_cert_len = 0;
+
+  {
+    res = ykpiv_util_write_cert(g_state, YKPIV_KEY_AUTHENTICATION, (uint8_t*)g_cert, sizeof(g_cert));
+    ck_assert_int_eq(res, YKPIV_OK);
+
+    res = ykpiv_util_read_cert(g_state, YKPIV_KEY_AUTHENTICATION, &read_cert, &read_cert_len);
+    ck_assert_int_eq(res, YKPIV_OK);
+    ck_assert_ptr_nonnull(read_cert);
+    ck_assert_int_eq(read_cert_len, sizeof(g_cert));
+    ck_assert_mem_eq(g_cert, read_cert, sizeof(g_cert));
+
+    res = ykpiv_util_free(g_state, read_cert);
+    ck_assert_int_eq(res, YKPIV_OK);
+  }
+
+  {
+    ykpiv_key *keys = NULL;
+    size_t data_len;
+    uint8_t key_count;
+    res = ykpiv_util_list_keys(g_state, &key_count, &keys, &data_len);
+    ck_assert_int_eq(res, YKPIV_OK);
+    ck_assert_ptr_nonnull(keys);
+    ck_assert_int_gt(key_count, 0);
+
+    res = ykpiv_util_free(g_state, keys);
+    ck_assert_int_eq(res, YKPIV_OK);
+  }
+
+  {
+    res = ykpiv_util_delete_cert(g_state, YKPIV_KEY_AUTHENTICATION);
+    ck_assert_int_eq(res, YKPIV_OK);
+
+    res = ykpiv_util_read_cert(g_state, YKPIV_KEY_AUTHENTICATION, &read_cert, &read_cert_len);
+    ck_assert_int_eq(res, YKPIV_GENERIC_ERROR);
+
+    res = ykpiv_util_free(g_state, read_cert);
+    ck_assert_int_eq(res, YKPIV_OK);
+  }
+}
+END_TEST
+
+START_TEST(test_generate_key) {
+  ykpiv_rc res;
+  uint8_t *mod, *exp;
+  size_t mod_len, exp_len;
+  res = ykpiv_util_write_cert(g_state, YKPIV_KEY_AUTHENTICATION, (uint8_t*)g_cert, sizeof(g_cert));
+  ck_assert_int_eq(res, YKPIV_OK);
+  res = ykpiv_util_generate_key(g_state,
+                                YKPIV_KEY_AUTHENTICATION,
+                                YKPIV_ALGO_RSA2048,
+                                YKPIV_PINPOLICY_ONCE,
+                                YKPIV_TOUCHPOLICY_DEFAULT,
+                                &mod,
+                                &mod_len,
+                                &exp,
+                                &exp_len,
+                                NULL,
+                                NULL);
+  ck_assert_int_eq(res, YKPIV_OK);
+  res = ykpiv_util_free(g_state, mod);
+  ck_assert_int_eq(res, YKPIV_OK);
+  res = ykpiv_util_free(g_state, exp);
+  ck_assert_int_eq(res, YKPIV_OK);
+  // TODO: and??
+}
+END_TEST
+
+START_TEST(test_read_write_mscmap) {
+}
+END_TEST
+
+START_TEST(test_read_write_msroots) {
+}
+END_TEST
+
+START_TEST(test_reset) {
+}
+END_TEST
+
+Suite *test_suite(void) {
+  Suite *s;
+  TCase *tc;
+
+  s = suite_create("libykpiv util");
+  tc = tcase_create("util");
+#ifdef HW_TESTS
+  tcase_add_unchecked_fixture(tc, setup, teardown);
+  tcase_add_test(tc, test_devicemodel);
+  tcase_add_test(tc, test_get_set_cardid);
+  tcase_add_test(tc, test_read_write_list_delete_cert);
+  tcase_add_test(tc, test_generate_key);
+#endif
+  suite_add_tcase(s, tc);
+
+  return s;
+}
+
+int main(void)
+{
+  int number_failed;
+  Suite *s;
+  SRunner *sr;
+
+  s = test_suite();
+  sr = srunner_create(s);
+  srunner_set_fork_status(sr, CK_NOFORK);
+  srunner_run_all(sr, CK_NORMAL);
+  number_failed = srunner_ntests_failed(sr);
+  srunner_free(sr);
+  return (number_failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
+}