/** * @file test_ks_ts.c * @author Roman Janota * @brief libnetconf2 Keystore and trustore usage test. * * @copyright * Copyright (c) 2023 - 2024 CESNET, z.s.p.o. * * This source code is licensed under BSD 3-Clause License (the "License"). * You may not use this file except in compliance with the License. * You may obtain a copy of the License at * * https://opensource.org/licenses/BSD-3-Clause */ #define _GNU_SOURCE #include #include #include #include #include #include #include #include #include "ln2_test.h" int TEST_PORT = 10050; const char *TEST_PORT_STR = "10050"; static void * client_thread_ssh(void *arg) { int ret; struct nc_session *session = NULL; struct ln2_test_ctx *test_ctx = arg; /* skip all hostkey and known_hosts checks */ nc_client_ssh_set_knownhosts_mode(NC_SSH_KNOWNHOSTS_SKIP); ret = nc_client_set_schema_searchpath(MODULES_DIR); assert_int_equal(ret, 0); ret = nc_client_ssh_set_username("client"); assert_int_equal(ret, 0); ret = nc_client_ssh_add_keypair(TESTS_DIR "/data/id_ed25519.pub", TESTS_DIR "/data/id_ed25519"); assert_int_equal(ret, 0); pthread_barrier_wait(&test_ctx->barrier); session = nc_connect_ssh("127.0.0.1", TEST_PORT, NULL); assert_non_null(session); nc_session_free(session, NULL); return NULL; } static void test_nc_ks_ts_ssh(void **state) { int ret, i; pthread_t tids[2]; assert_non_null(state); ret = pthread_create(&tids[0], NULL, client_thread_ssh, *state); assert_int_equal(ret, 0); ret = pthread_create(&tids[1], NULL, ln2_glob_test_server_thread, *state); assert_int_equal(ret, 0); for (i = 0; i < 2; i++) { pthread_join(tids[i], NULL); } } static int setup_ssh(void **state) { int ret; struct lyd_node *tree = NULL; struct ln2_test_ctx *test_ctx; ret = ln2_glob_test_setup(&test_ctx); assert_int_equal(ret, 0); *state = test_ctx; ret = nc_server_config_add_address_port(test_ctx->ctx, "endpt", NC_TI_SSH, "127.0.0.1", TEST_PORT, &tree); assert_int_equal(ret, 0); ret = nc_server_config_add_ssh_keystore_ref(test_ctx->ctx, "endpt", "hostkey", "test_keystore", &tree); assert_int_equal(ret, 0); ret = nc_server_config_add_ssh_truststore_ref(test_ctx->ctx, "endpt", "client", "test_truststore", &tree); assert_int_equal(ret, 0); ret = nc_server_config_add_keystore_asym_key(test_ctx->ctx, NC_TI_SSH, "test_keystore", TESTS_DIR "/data/key_rsa", NULL, &tree); assert_int_equal(ret, 0); ret = nc_server_config_add_truststore_pubkey(test_ctx->ctx, "test_truststore", "pubkey", TESTS_DIR "/data/id_ed25519.pub", &tree); assert_int_equal(ret, 0); /* configure the server based on the data */ ret = nc_server_config_setup_data(tree); assert_int_equal(ret, 0); lyd_free_all(tree); return 0; } static void * client_thread_tls(void *arg) { int ret; struct nc_session *session = NULL; struct ln2_test_ctx *test_ctx = arg; ret = nc_client_set_schema_searchpath(MODULES_DIR); assert_int_equal(ret, 0); /* set client cert */ ret = nc_client_tls_set_cert_key_paths(TESTS_DIR "/data/client.crt", TESTS_DIR "/data/client.key"); assert_int_equal(ret, 0); /* set client ca */ ret = nc_client_tls_set_trusted_ca_paths(NULL, TESTS_DIR "/data"); assert_int_equal(ret, 0); pthread_barrier_wait(&test_ctx->barrier); session = nc_connect_tls("127.0.0.1", TEST_PORT, NULL); assert_non_null(session); nc_session_free(session, NULL); return NULL; } static void test_nc_ks_ts_tls(void **state) { int ret, i; pthread_t tids[2]; assert_non_null(state); ret = pthread_create(&tids[0], NULL, client_thread_tls, *state); assert_int_equal(ret, 0); ret = pthread_create(&tids[1], NULL, ln2_glob_test_server_thread, *state); assert_int_equal(ret, 0); for (i = 0; i < 2; i++) { pthread_join(tids[i], NULL); } } static int setup_tls(void **state) { int ret; struct lyd_node *tree = NULL; struct ln2_test_ctx *test_ctx; ret = ln2_glob_test_setup(&test_ctx); assert_int_equal(ret, 0); *state = test_ctx; /* new tls bind */ ret = nc_server_config_add_address_port(test_ctx->ctx, "endpt", NC_TI_TLS, "127.0.0.1", TEST_PORT, &tree); assert_int_equal(ret, 0); /* new keystore asym key pair */ ret = nc_server_config_add_keystore_asym_key(test_ctx->ctx, NC_TI_TLS, "server_key", TESTS_DIR "/data/server.key", NULL, &tree); assert_int_equal(ret, 0); /* new keystore cert belonging to the key pair */ ret = nc_server_config_add_keystore_cert(test_ctx->ctx, "server_key", "server_cert", TESTS_DIR "/data/server.crt", &tree); assert_int_equal(ret, 0); /* new truststore client cert */ ret = nc_server_config_add_truststore_cert(test_ctx->ctx, "ee_cert_bag", "ee_cert", TESTS_DIR "/data/client.crt", &tree); assert_int_equal(ret, 0); /* new truststore client CA cert */ ret = nc_server_config_add_truststore_cert(test_ctx->ctx, "ca_cert_bag", "ca_cert", TESTS_DIR "/data/serverca.pem", &tree); assert_int_equal(ret, 0); /* new keystore ref for the TLS server cert */ ret = nc_server_config_add_tls_keystore_ref(test_ctx->ctx, "endpt", "server_key", "server_cert", &tree); assert_int_equal(ret, 0); /* new truststore ref for the client cert */ ret = nc_server_config_add_tls_client_cert_truststore_ref(test_ctx->ctx, "endpt", "ee_cert_bag", &tree); assert_int_equal(ret, 0); /* new truststore ref for the client CA cert */ ret = nc_server_config_add_tls_ca_cert_truststore_ref(test_ctx->ctx, "endpt", "ca_cert_bag", &tree); assert_int_equal(ret, 0); /* new cert-to-name */ ret = nc_server_config_add_tls_ctn(test_ctx->ctx, "endpt", 1, "04:85:6B:75:D1:1A:86:E0:D8:FE:5B:BD:72:F5:73:1D:07:EA:32:BF:09:11:21:6A:6E:23:78:8E:B6:D5:73:C3:2D", NC_TLS_CTN_SPECIFIED, "client", &tree); assert_int_equal(ret, 0); /* configure the server based on the data */ ret = nc_server_config_setup_data(tree); assert_int_equal(ret, 0); lyd_free_all(tree); return 0; } int main(void) { const struct CMUnitTest tests[] = { cmocka_unit_test_setup_teardown(test_nc_ks_ts_ssh, setup_ssh, ln2_glob_test_teardown), cmocka_unit_test_setup_teardown(test_nc_ks_ts_tls, setup_tls, ln2_glob_test_teardown), }; /* try to get ports from the environment, otherwise use the default */ if (ln2_glob_test_get_ports(1, &TEST_PORT, &TEST_PORT_STR)) { return 1; } setenv("CMOCKA_TEST_ABORT", "1", 1); return cmocka_run_group_tests(tests, NULL, NULL); }