[thirdparty/openssl.git] / test / cmp_client_test.c

/*
 * Copyright 2007-2019 The OpenSSL Project Authors. All Rights Reserved.
 * Copyright Nokia 2007-2019
 * Copyright Siemens AG 2015-2019
 *
 * Licensed under the Apache License 2.0 (the "License").  You may not use
 * this file except in compliance with the License.  You can obtain a copy
 * in the file LICENSE in the source distribution or at
 * https://www.openssl.org/source/license.html
 */

#include "cmp_testlib.h"

#include "apps/cmp_mock_srv.h"

#ifndef NDEBUG /* tests need mock server, which is available only if !NDEBUG */

static const char *server_key_f;
static const char *server_cert_f;
static const char *client_key_f;
static const char *client_cert_f;
static const char *pkcs10_f;

typedef struct test_fixture {
    const char *test_case_name;
    OSSL_CMP_CTX *cmp_ctx;
    OSSL_CMP_SRV_CTX *srv_ctx;
    int expected;
    X509 *(*exec_cert_ses_cb) (OSSL_CMP_CTX *);
    STACK_OF(X509) *caPubs;
} CMP_SES_TEST_FIXTURE;

static EVP_PKEY *server_key = NULL;
static X509 *server_cert = NULL;
static EVP_PKEY *client_key = NULL;
static X509 *client_cert = NULL;
static unsigned char ref[CMP_TEST_REFVALUE_LENGTH];

/*
 * For these unit tests, the client abandons message protection, and for
 * error messages the mock server does so as well.
 * Message protection and verification is tested in cmp_lib_test.c
 */

static void tear_down(CMP_SES_TEST_FIXTURE *fixture)
{
    OSSL_CMP_CTX_free(fixture->cmp_ctx);
    ossl_cmp_mock_srv_free(fixture->srv_ctx);
    sk_X509_free(fixture->caPubs);
    OPENSSL_free(fixture);
}

static CMP_SES_TEST_FIXTURE *set_up(const char *const test_case_name)
{
    CMP_SES_TEST_FIXTURE *fixture;
    OSSL_CMP_CTX *srv_cmp_ctx = NULL;
    OSSL_CMP_CTX *ctx = NULL; /* for client */

    if (!TEST_ptr(fixture = OPENSSL_zalloc(sizeof(*fixture))))
        return NULL;
    fixture->test_case_name = test_case_name;
    if (!TEST_ptr(fixture->srv_ctx = ossl_cmp_mock_srv_new())
            || !OSSL_CMP_SRV_CTX_set_accept_unprotected(fixture->srv_ctx, 1)
            || !ossl_cmp_mock_srv_set1_certOut(fixture->srv_ctx, client_cert)
            || (srv_cmp_ctx =
                OSSL_CMP_SRV_CTX_get0_cmp_ctx(fixture->srv_ctx)) == NULL
            || !OSSL_CMP_CTX_set1_clCert(srv_cmp_ctx, server_cert)
            || !OSSL_CMP_CTX_set1_pkey(srv_cmp_ctx, server_key))
        goto err;
    if (!TEST_ptr(fixture->cmp_ctx = ctx = OSSL_CMP_CTX_new())
            || !OSSL_CMP_CTX_set_log_cb(fixture->cmp_ctx, print_to_bio_out)
            || !OSSL_CMP_CTX_set_transfer_cb(ctx, OSSL_CMP_CTX_server_perform)
            || !OSSL_CMP_CTX_set_transfer_cb_arg(ctx, fixture->srv_ctx)
            || !OSSL_CMP_CTX_set_option(ctx, OSSL_CMP_OPT_UNPROTECTED_SEND, 1)
            || !OSSL_CMP_CTX_set_option(ctx, OSSL_CMP_OPT_UNPROTECTED_ERRORS, 1)
            || !OSSL_CMP_CTX_set1_oldCert(ctx, client_cert)
            || !OSSL_CMP_CTX_set1_pkey(ctx, client_key)
            || !OSSL_CMP_CTX_set1_srvCert(ctx, server_cert)
            || !OSSL_CMP_CTX_set1_referenceValue(ctx, ref, sizeof(ref)))
        goto err;
    fixture->exec_cert_ses_cb = NULL;
    return fixture;

 err:
    tear_down(fixture);
    return NULL;
}

static int execute_exec_RR_ses_test(CMP_SES_TEST_FIXTURE *fixture)
{
    return TEST_int_eq(fixture->expected,
                       OSSL_CMP_exec_RR_ses(fixture->cmp_ctx) == client_cert);
}

static int execute_exec_GENM_ses_test(CMP_SES_TEST_FIXTURE *fixture)
{
    STACK_OF(OSSL_CMP_ITAV) *itavs = NULL;
    if (!TEST_ptr(itavs = OSSL_CMP_exec_GENM_ses(fixture->cmp_ctx)))
        return 0;
    sk_OSSL_CMP_ITAV_pop_free(itavs, OSSL_CMP_ITAV_free);
    /* TODO: check if the returned value is the expected one (same as sent) */
    return 1;
}

static int execute_exec_certrequest_ses_test(CMP_SES_TEST_FIXTURE *fixture)
{
    X509 *res;

    if (fixture->expected == 0)
        return TEST_ptr_null(fixture->exec_cert_ses_cb(fixture->cmp_ctx));

    if (!TEST_ptr(res = fixture->exec_cert_ses_cb(fixture->cmp_ctx))
            || !TEST_int_eq(X509_cmp(res, client_cert), 0))
        return 0;
    /* TODO: check that cerfConf has been exchanged unless implicitConfirm */
    if (fixture->caPubs != NULL) {
        STACK_OF(X509) *caPubs = OSSL_CMP_CTX_get1_caPubs(fixture->cmp_ctx);
        int ret = TEST_int_eq(STACK_OF_X509_cmp(fixture->caPubs, caPubs), 0);

        sk_X509_pop_free(caPubs, X509_free);
        return ret;
    }
    return 1;
}

static int test_exec_RR_ses(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->expected = 1;
    EXECUTE_TEST(execute_exec_RR_ses_test, tear_down);
    return result;
}

static int test_exec_RR_ses_receive_error(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    ossl_cmp_mock_srv_set_statusInfo(fixture->srv_ctx,
                                     OSSL_CMP_PKISTATUS_rejection,
                                     OSSL_CMP_CTX_FAILINFO_signerNotTrusted,
                                     "test string");
    ossl_cmp_mock_srv_set_send_error(fixture->srv_ctx, 1);
    fixture->expected = 0;
    EXECUTE_TEST(execute_exec_RR_ses_test, tear_down);
    return result;
}

static int test_exec_IR_ses(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->exec_cert_ses_cb = OSSL_CMP_exec_IR_ses;
    fixture->expected = 1;
    fixture->caPubs = sk_X509_new_null();
    sk_X509_push(fixture->caPubs, server_cert);
    sk_X509_push(fixture->caPubs, server_cert);
    ossl_cmp_mock_srv_set1_caPubsOut(fixture->srv_ctx, fixture->caPubs);
    EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
    return result;
}

static const int checkAfter = 1;
static int test_exec_IR_ses_poll(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->exec_cert_ses_cb = OSSL_CMP_exec_IR_ses;
    fixture->expected = 1;
    ossl_cmp_mock_srv_set_pollCount(fixture->srv_ctx, 2);
    ossl_cmp_mock_srv_set_checkAfterTime(fixture->srv_ctx, checkAfter);
    EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
    /* TODO: check that 2 rounds are done or session takes 2..3 seconds */
    return result;
}

static int test_exec_IR_ses_poll_timeout(void)
{
    const int pollCount = 3;
    const int tout = pollCount * checkAfter;

    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->exec_cert_ses_cb = OSSL_CMP_exec_IR_ses;
    fixture->expected = 0;
    ossl_cmp_mock_srv_set_pollCount(fixture->srv_ctx, pollCount + 1);
    ossl_cmp_mock_srv_set_checkAfterTime(fixture->srv_ctx, checkAfter);
    OSSL_CMP_CTX_set_option(fixture->cmp_ctx, OSSL_CMP_OPT_TOTAL_TIMEOUT, tout);
    EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
    return result;
}


static int test_exec_CR_ses(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->exec_cert_ses_cb = OSSL_CMP_exec_CR_ses;
    fixture->expected = 1;
    EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
    return result;
}

static int test_exec_CR_ses_implicit_confirm(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->exec_cert_ses_cb = OSSL_CMP_exec_CR_ses;
    fixture->expected = 1;
    OSSL_CMP_CTX_set_option(fixture->cmp_ctx,
                            OSSL_CMP_OPT_IMPLICIT_CONFIRM, 1);
    OSSL_CMP_SRV_CTX_set_grant_implicit_confirm(fixture->srv_ctx, 1);
    EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
    return result;
}

static int test_exec_KUR_ses(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->exec_cert_ses_cb = OSSL_CMP_exec_KUR_ses;
    fixture->expected = 1;
    EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
    return result;
}

static int test_exec_P10CR_ses(void)
{
    X509_REQ *req = NULL;

    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->exec_cert_ses_cb = OSSL_CMP_exec_P10CR_ses;
    fixture->expected = 1;
    if (!TEST_ptr(req = load_csr(pkcs10_f))
            || !TEST_true(OSSL_CMP_CTX_set1_p10CSR(fixture->cmp_ctx, req))) {
        tear_down(fixture);
        fixture = NULL;
    }
    X509_REQ_free(req);
    EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
    return result;
}

static int execute_try_certreq_poll_test(CMP_SES_TEST_FIXTURE *fixture)
{
    OSSL_CMP_CTX *ctx = fixture->cmp_ctx;
    int check_after;
    const int CHECK_AFTER = 5;
    const int TYPE = OSSL_CMP_KUR;

    ossl_cmp_mock_srv_set_pollCount(fixture->srv_ctx, 3);
    ossl_cmp_mock_srv_set_checkAfterTime(fixture->srv_ctx, CHECK_AFTER);
    return TEST_int_eq(-1, OSSL_CMP_try_certreq(ctx, TYPE, &check_after))
        && check_after == CHECK_AFTER
        && TEST_ptr_eq(OSSL_CMP_CTX_get0_newCert(ctx), NULL)
        && TEST_int_eq(-1, OSSL_CMP_try_certreq(ctx, TYPE, &check_after))
        && check_after == CHECK_AFTER
        && TEST_ptr_eq(OSSL_CMP_CTX_get0_newCert(ctx), NULL)
        && TEST_int_eq(fixture->expected, OSSL_CMP_try_certreq(ctx, TYPE, NULL))
        && TEST_int_eq(0, X509_cmp(OSSL_CMP_CTX_get0_newCert(ctx), client_cert));
}

static int test_try_certreq_poll(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->expected = 1;
    EXECUTE_TEST(execute_try_certreq_poll_test, tear_down);
    return result;
}

static int execute_try_certreq_poll_abort_test(CMP_SES_TEST_FIXTURE *fixture)
{
    OSSL_CMP_CTX *ctx = fixture->cmp_ctx;
    int check_after;
    const int CHECK_AFTER = INT_MAX;
    const int TYPE = OSSL_CMP_CR;

    ossl_cmp_mock_srv_set_pollCount(fixture->srv_ctx, 3);
    ossl_cmp_mock_srv_set_checkAfterTime(fixture->srv_ctx, CHECK_AFTER);
    return TEST_int_eq(-1, OSSL_CMP_try_certreq(ctx, TYPE, &check_after))
        && check_after == CHECK_AFTER
        && TEST_ptr_eq(OSSL_CMP_CTX_get0_newCert(ctx), NULL)
        && TEST_int_eq(fixture->expected, OSSL_CMP_try_certreq(ctx, -1, NULL))
        && TEST_ptr_eq(OSSL_CMP_CTX_get0_newCert(fixture->cmp_ctx), NULL);
}

static int test_try_certreq_poll_abort(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->expected = 1;
    EXECUTE_TEST(execute_try_certreq_poll_abort_test, tear_down);
    return result;
}

static int test_exec_GENM_ses(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    EXECUTE_TEST(execute_exec_GENM_ses_test, tear_down);
    return result;
}

static int execute_exchange_certConf_test(CMP_SES_TEST_FIXTURE *fixture)
{
    int res =
        ossl_cmp_exchange_certConf(fixture->cmp_ctx,
                                   OSSL_CMP_PKIFAILUREINFO_addInfoNotAvailable,
                                   "abcdefg");
    return TEST_int_eq(fixture->expected, res);
}

static int execute_exchange_error_test(CMP_SES_TEST_FIXTURE *fixture)
{
    int res =
        ossl_cmp_exchange_error(fixture->cmp_ctx,
                                OSSL_CMP_PKISTATUS_rejection,
                                1 << OSSL_CMP_PKIFAILUREINFO_unsupportedVersion,
                                "foo_status", 999, "foo_details");

    return TEST_int_eq(fixture->expected, res);
}

static int test_exchange_certConf(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->expected = 0; /* client should not send certConf immediately */
    if (!ossl_cmp_ctx_set0_newCert(fixture->cmp_ctx, X509_dup(client_cert))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_exchange_certConf_test, tear_down);
    return result;
}

static int test_exchange_error(void)
{
    SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
    fixture->expected = 1; /* client may send error any time */
    EXECUTE_TEST(execute_exchange_error_test, tear_down);
    return result;
}

void cleanup_tests(void)
{
    X509_free(server_cert);
    EVP_PKEY_free(server_key);
    X509_free(client_cert);
    EVP_PKEY_free(client_key);
    return;
}

int setup_tests(void)
{
    if (!test_skip_common_options()) {
        TEST_error("Error parsing test options\n");
        return 0;
    }

    if (!TEST_ptr(server_key_f = test_get_argument(0))
            || !TEST_ptr(server_cert_f = test_get_argument(1))
            || !TEST_ptr(client_key_f = test_get_argument(2))
            || !TEST_ptr(client_cert_f = test_get_argument(3))
            || !TEST_ptr(pkcs10_f = test_get_argument(4))) {
        TEST_error("usage: cmp_client_test server.key server.crt client.key client.crt client.csr\n");
        return 0;
    }

    if (!TEST_ptr(server_key = load_pem_key(server_key_f))
            || !TEST_ptr(server_cert = load_pem_cert(server_cert_f))
            || !TEST_ptr(client_key = load_pem_key(client_key_f))
            || !TEST_ptr(client_cert = load_pem_cert(client_cert_f))
            || !TEST_int_eq(1, RAND_bytes(ref, sizeof(ref)))) {
        cleanup_tests();
        return 0;
    }

    ADD_TEST(test_exec_RR_ses);
    ADD_TEST(test_exec_RR_ses_receive_error);
    ADD_TEST(test_exec_CR_ses);
    ADD_TEST(test_exec_CR_ses_implicit_confirm);
    ADD_TEST(test_exec_IR_ses);
    ADD_TEST(test_exec_IR_ses_poll);
    ADD_TEST(test_exec_IR_ses_poll_timeout);
    ADD_TEST(test_exec_KUR_ses);
    ADD_TEST(test_exec_P10CR_ses);
    ADD_TEST(test_try_certreq_poll);
    ADD_TEST(test_try_certreq_poll_abort);
    ADD_TEST(test_exec_GENM_ses);
    ADD_TEST(test_exchange_certConf);
    ADD_TEST(test_exchange_error);
    return 1;
}

#else /* !defined (NDEBUG) */

int setup_tests(void)
{
    TEST_note("CMP session tests are disabled in this build (NDEBUG).");
    return 1;
}

#endif
Commit	Line	Data
7e765f46 DDO	1	/*
	2	* Copyright 2007-2019 The OpenSSL Project Authors. All Rights Reserved.
	3	* Copyright Nokia 2007-2019
	4	* Copyright Siemens AG 2015-2019
	5	*
	6	* Licensed under the Apache License 2.0 (the "License"). You may not use
	7	* this file except in compliance with the License. You can obtain a copy
	8	* in the file LICENSE in the source distribution or at
	9	* https://www.openssl.org/source/license.html
	10	*/
	11
	12	#include "cmp_testlib.h"
	13
	14	#include "apps/cmp_mock_srv.h"
	15
	16	#ifndef NDEBUG /* tests need mock server, which is available only if !NDEBUG */
	17
	18	static const char *server_key_f;
	19	static const char *server_cert_f;
	20	static const char *client_key_f;
	21	static const char *client_cert_f;
	22	static const char *pkcs10_f;
	23
	24	typedef struct test_fixture {
	25	const char *test_case_name;
	26	OSSL_CMP_CTX *cmp_ctx;
	27	OSSL_CMP_SRV_CTX *srv_ctx;
	28	int expected;
	29	X509 (exec_cert_ses_cb) (OSSL_CMP_CTX *);
	30	STACK_OF(X509) *caPubs;
	31	} CMP_SES_TEST_FIXTURE;
	32
	33	static EVP_PKEY *server_key = NULL;
	34	static X509 *server_cert = NULL;
	35	static EVP_PKEY *client_key = NULL;
	36	static X509 *client_cert = NULL;
	37	static unsigned char ref[CMP_TEST_REFVALUE_LENGTH];
	38
	39	/*
	40	* For these unit tests, the client abandons message protection, and for
	41	* error messages the mock server does so as well.
	42	* Message protection and verification is tested in cmp_lib_test.c
	43	*/
	44
	45	static void tear_down(CMP_SES_TEST_FIXTURE *fixture)
	46	{
	47	OSSL_CMP_CTX_free(fixture->cmp_ctx);
	48	ossl_cmp_mock_srv_free(fixture->srv_ctx);
	49	sk_X509_free(fixture->caPubs);
	50	OPENSSL_free(fixture);
	51	}
	52
	53	static CMP_SES_TEST_FIXTURE set_up(const char const test_case_name)
	54	{
	55	CMP_SES_TEST_FIXTURE *fixture;
	56	OSSL_CMP_CTX *srv_cmp_ctx = NULL;
	57	OSSL_CMP_CTX ctx = NULL; / for client */
	58
	59	if (!TEST_ptr(fixture = OPENSSL_zalloc(sizeof(*fixture))))
	60	return NULL;
	61	fixture->test_case_name = test_case_name;
	62	if (!TEST_ptr(fixture->srv_ctx = ossl_cmp_mock_srv_new())
	63	\|\| !OSSL_CMP_SRV_CTX_set_accept_unprotected(fixture->srv_ctx, 1)
	64	\|\| !ossl_cmp_mock_srv_set1_certOut(fixture->srv_ctx, client_cert)
65	\|\| (srv_cmp_ctx =
66	OSSL_CMP_SRV_CTX_get0_cmp_ctx(fixture->srv_ctx)) == NULL
67	\|\| !OSSL_CMP_CTX_set1_clCert(srv_cmp_ctx, server_cert)
68	\|\| !OSSL_CMP_CTX_set1_pkey(srv_cmp_ctx, server_key))
69	goto err;
70	if (!TEST_ptr(fixture->cmp_ctx = ctx = OSSL_CMP_CTX_new())
71	\|\| !OSSL_CMP_CTX_set_log_cb(fixture->cmp_ctx, print_to_bio_out)
72	\|\| !OSSL_CMP_CTX_set_transfer_cb(ctx, OSSL_CMP_CTX_server_perform)
73	\|\| !OSSL_CMP_CTX_set_transfer_cb_arg(ctx, fixture->srv_ctx)
74	\|\| !OSSL_CMP_CTX_set_option(ctx, OSSL_CMP_OPT_UNPROTECTED_SEND, 1)
75	\|\| !OSSL_CMP_CTX_set_option(ctx, OSSL_CMP_OPT_UNPROTECTED_ERRORS, 1)
76	\|\| !OSSL_CMP_CTX_set1_oldCert(ctx, client_cert)
77	\|\| !OSSL_CMP_CTX_set1_pkey(ctx, client_key)
78	\|\| !OSSL_CMP_CTX_set1_srvCert(ctx, server_cert)
79	\|\| !OSSL_CMP_CTX_set1_referenceValue(ctx, ref, sizeof(ref)))
80	goto err;
81	fixture->exec_cert_ses_cb = NULL;
82	return fixture;
83
84	err:
85	tear_down(fixture);
86	return NULL;
87	}
88
89	static int execute_exec_RR_ses_test(CMP_SES_TEST_FIXTURE *fixture)
90	{
91	return TEST_int_eq(fixture->expected,
92	OSSL_CMP_exec_RR_ses(fixture->cmp_ctx) == client_cert);
93	}
94
95	static int execute_exec_GENM_ses_test(CMP_SES_TEST_FIXTURE *fixture)
96	{
97	STACK_OF(OSSL_CMP_ITAV) *itavs = NULL;
98	if (!TEST_ptr(itavs = OSSL_CMP_exec_GENM_ses(fixture->cmp_ctx)))
99	return 0;
100	sk_OSSL_CMP_ITAV_pop_free(itavs, OSSL_CMP_ITAV_free);
101	/* TODO: check if the returned value is the expected one (same as sent) */
102	return 1;
103	}
104
105	static int execute_exec_certrequest_ses_test(CMP_SES_TEST_FIXTURE *fixture)
106	{
107	X509 *res;
108
109	if (fixture->expected == 0)
110	return TEST_ptr_null(fixture->exec_cert_ses_cb(fixture->cmp_ctx));
111
112	if (!TEST_ptr(res = fixture->exec_cert_ses_cb(fixture->cmp_ctx))
113	\|\| !TEST_int_eq(X509_cmp(res, client_cert), 0))
114	return 0;
115	/* TODO: check that cerfConf has been exchanged unless implicitConfirm */
116	if (fixture->caPubs != NULL) {
117	STACK_OF(X509) *caPubs = OSSL_CMP_CTX_get1_caPubs(fixture->cmp_ctx);
118	int ret = TEST_int_eq(STACK_OF_X509_cmp(fixture->caPubs, caPubs), 0);
119
120	sk_X509_pop_free(caPubs, X509_free);
121	return ret;
122	}
123	return 1;
124	}
125
126	static int test_exec_RR_ses(void)
127	{
128	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
129	fixture->expected = 1;
130	EXECUTE_TEST(execute_exec_RR_ses_test, tear_down);
131	return result;
132	}
133
134	static int test_exec_RR_ses_receive_error(void)
135	{
136	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
137	ossl_cmp_mock_srv_set_statusInfo(fixture->srv_ctx,
138	OSSL_CMP_PKISTATUS_rejection,
139	OSSL_CMP_CTX_FAILINFO_signerNotTrusted,
140	"test string");
141	ossl_cmp_mock_srv_set_send_error(fixture->srv_ctx, 1);
142	fixture->expected = 0;
143	EXECUTE_TEST(execute_exec_RR_ses_test, tear_down);
144	return result;
145	}
146
147	static int test_exec_IR_ses(void)
148	{
149	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
150	fixture->exec_cert_ses_cb = OSSL_CMP_exec_IR_ses;
151	fixture->expected = 1;
152	fixture->caPubs = sk_X509_new_null();
153	sk_X509_push(fixture->caPubs, server_cert);
154	sk_X509_push(fixture->caPubs, server_cert);
155	ossl_cmp_mock_srv_set1_caPubsOut(fixture->srv_ctx, fixture->caPubs);
156	EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
157	return result;
158	}
159
160	static const int checkAfter = 1;
161	static int test_exec_IR_ses_poll(void)
162	{
163	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
164	fixture->exec_cert_ses_cb = OSSL_CMP_exec_IR_ses;
165	fixture->expected = 1;
166	ossl_cmp_mock_srv_set_pollCount(fixture->srv_ctx, 2);
167	ossl_cmp_mock_srv_set_checkAfterTime(fixture->srv_ctx, checkAfter);
168	EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
169	/* TODO: check that 2 rounds are done or session takes 2..3 seconds */
170	return result;
171	}
172
173	static int test_exec_IR_ses_poll_timeout(void)
174	{
175	const int pollCount = 3;
176	const int tout = pollCount * checkAfter;
177
178	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
179	fixture->exec_cert_ses_cb = OSSL_CMP_exec_IR_ses;
180	fixture->expected = 0;
181	ossl_cmp_mock_srv_set_pollCount(fixture->srv_ctx, pollCount + 1);
182	ossl_cmp_mock_srv_set_checkAfterTime(fixture->srv_ctx, checkAfter);
183	OSSL_CMP_CTX_set_option(fixture->cmp_ctx, OSSL_CMP_OPT_TOTAL_TIMEOUT, tout);
184	EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
185	return result;
186	}
187
188
189	static int test_exec_CR_ses(void)
190	{
191	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
192	fixture->exec_cert_ses_cb = OSSL_CMP_exec_CR_ses;
193	fixture->expected = 1;
194	EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
195	return result;
196	}
197
198	static int test_exec_CR_ses_implicit_confirm(void)
199	{
200	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
201	fixture->exec_cert_ses_cb = OSSL_CMP_exec_CR_ses;
202	fixture->expected = 1;
203	OSSL_CMP_CTX_set_option(fixture->cmp_ctx,
204	OSSL_CMP_OPT_IMPLICIT_CONFIRM, 1);
205	OSSL_CMP_SRV_CTX_set_grant_implicit_confirm(fixture->srv_ctx, 1);
206	EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
207	return result;
208	}
209
210	static int test_exec_KUR_ses(void)
211	{
212	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
213	fixture->exec_cert_ses_cb = OSSL_CMP_exec_KUR_ses;
214	fixture->expected = 1;
215	EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
216	return result;
217	}
218
219	static int test_exec_P10CR_ses(void)
220	{
221	X509_REQ *req = NULL;
222
223	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
224	fixture->exec_cert_ses_cb = OSSL_CMP_exec_P10CR_ses;
225	fixture->expected = 1;
226	if (!TEST_ptr(req = load_csr(pkcs10_f))
227	\|\| !TEST_true(OSSL_CMP_CTX_set1_p10CSR(fixture->cmp_ctx, req))) {
228	tear_down(fixture);
229	fixture = NULL;
230	}
231	X509_REQ_free(req);
232	EXECUTE_TEST(execute_exec_certrequest_ses_test, tear_down);
233	return result;
234	}
235
236	static int execute_try_certreq_poll_test(CMP_SES_TEST_FIXTURE *fixture)
237	{
238	OSSL_CMP_CTX *ctx = fixture->cmp_ctx;
239	int check_after;
240	const int CHECK_AFTER = 5;
241	const int TYPE = OSSL_CMP_KUR;
242
243	ossl_cmp_mock_srv_set_pollCount(fixture->srv_ctx, 3);
244	ossl_cmp_mock_srv_set_checkAfterTime(fixture->srv_ctx, CHECK_AFTER);
245	return TEST_int_eq(-1, OSSL_CMP_try_certreq(ctx, TYPE, &check_after))
246	&& check_after == CHECK_AFTER
247	&& TEST_ptr_eq(OSSL_CMP_CTX_get0_newCert(ctx), NULL)
248	&& TEST_int_eq(-1, OSSL_CMP_try_certreq(ctx, TYPE, &check_after))
249	&& check_after == CHECK_AFTER
250	&& TEST_ptr_eq(OSSL_CMP_CTX_get0_newCert(ctx), NULL)
251	&& TEST_int_eq(fixture->expected, OSSL_CMP_try_certreq(ctx, TYPE, NULL))
252	&& TEST_int_eq(0, X509_cmp(OSSL_CMP_CTX_get0_newCert(ctx), client_cert));
253	}
254
255	static int test_try_certreq_poll(void)
256	{
257	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
258	fixture->expected = 1;
259	EXECUTE_TEST(execute_try_certreq_poll_test, tear_down);
260	return result;
261	}
262
263	static int execute_try_certreq_poll_abort_test(CMP_SES_TEST_FIXTURE *fixture)
264	{
265	OSSL_CMP_CTX *ctx = fixture->cmp_ctx;
266	int check_after;
267	const int CHECK_AFTER = INT_MAX;
268	const int TYPE = OSSL_CMP_CR;
269
270	ossl_cmp_mock_srv_set_pollCount(fixture->srv_ctx, 3);
271	ossl_cmp_mock_srv_set_checkAfterTime(fixture->srv_ctx, CHECK_AFTER);
272	return TEST_int_eq(-1, OSSL_CMP_try_certreq(ctx, TYPE, &check_after))
273	&& check_after == CHECK_AFTER
274	&& TEST_ptr_eq(OSSL_CMP_CTX_get0_newCert(ctx), NULL)
275	&& TEST_int_eq(fixture->expected, OSSL_CMP_try_certreq(ctx, -1, NULL))
276	&& TEST_ptr_eq(OSSL_CMP_CTX_get0_newCert(fixture->cmp_ctx), NULL);
277	}
278
279	static int test_try_certreq_poll_abort(void)
280	{
281	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
282	fixture->expected = 1;
283	EXECUTE_TEST(execute_try_certreq_poll_abort_test, tear_down);
284	return result;
285	}
286
287	static int test_exec_GENM_ses(void)
288	{
289	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
290	EXECUTE_TEST(execute_exec_GENM_ses_test, tear_down);
291	return result;
292	}
293
294	static int execute_exchange_certConf_test(CMP_SES_TEST_FIXTURE *fixture)
295	{
296	int res =
297	ossl_cmp_exchange_certConf(fixture->cmp_ctx,
298	OSSL_CMP_PKIFAILUREINFO_addInfoNotAvailable,
299	"abcdefg");
300	return TEST_int_eq(fixture->expected, res);
301	}
302
303	static int execute_exchange_error_test(CMP_SES_TEST_FIXTURE *fixture)
304	{
305	int res =
306	ossl_cmp_exchange_error(fixture->cmp_ctx,
307	OSSL_CMP_PKISTATUS_rejection,
308	1 << OSSL_CMP_PKIFAILUREINFO_unsupportedVersion,
309	"foo_status", 999, "foo_details");
310
311	return TEST_int_eq(fixture->expected, res);
312	}
313
314	static int test_exchange_certConf(void)
315	{
316	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
317	fixture->expected = 0; /* client should not send certConf immediately */
318	if (!ossl_cmp_ctx_set0_newCert(fixture->cmp_ctx, X509_dup(client_cert))) {
319	tear_down(fixture);
320	fixture = NULL;
321	}
322	EXECUTE_TEST(execute_exchange_certConf_test, tear_down);
323	return result;
324	}
325
326	static int test_exchange_error(void)
327	{
328	SETUP_TEST_FIXTURE(CMP_SES_TEST_FIXTURE, set_up);
329	fixture->expected = 1; /* client may send error any time */
330	EXECUTE_TEST(execute_exchange_error_test, tear_down);
331	return result;
332	}
333
334	void cleanup_tests(void)
335	{
336	X509_free(server_cert);
337	EVP_PKEY_free(server_key);
338	X509_free(client_cert);
339	EVP_PKEY_free(client_key);
340	return;
341	}
342
343	int setup_tests(void)
344	{
345	if (!test_skip_common_options()) {
346	TEST_error("Error parsing test options\n");
347	return 0;
348	}
349
350	if (!TEST_ptr(server_key_f = test_get_argument(0))
351	\|\| !TEST_ptr(server_cert_f = test_get_argument(1))
352	\|\| !TEST_ptr(client_key_f = test_get_argument(2))
353	\|\| !TEST_ptr(client_cert_f = test_get_argument(3))
354	\|\| !TEST_ptr(pkcs10_f = test_get_argument(4))) {
355	TEST_error("usage: cmp_client_test server.key server.crt client.key client.crt client.csr\n");
356	return 0;
357	}
358
359	if (!TEST_ptr(server_key = load_pem_key(server_key_f))
360	\|\| !TEST_ptr(server_cert = load_pem_cert(server_cert_f))
361	\|\| !TEST_ptr(client_key = load_pem_key(client_key_f))
362	\|\| !TEST_ptr(client_cert = load_pem_cert(client_cert_f))
363	\|\| !TEST_int_eq(1, RAND_bytes(ref, sizeof(ref)))) {
364	cleanup_tests();
365	return 0;
366	}
367
368	ADD_TEST(test_exec_RR_ses);
369	ADD_TEST(test_exec_RR_ses_receive_error);
370	ADD_TEST(test_exec_CR_ses);
371	ADD_TEST(test_exec_CR_ses_implicit_confirm);
372	ADD_TEST(test_exec_IR_ses);
373	ADD_TEST(test_exec_IR_ses_poll);
374	ADD_TEST(test_exec_IR_ses_poll_timeout);
375	ADD_TEST(test_exec_KUR_ses);
376	ADD_TEST(test_exec_P10CR_ses);
377	ADD_TEST(test_try_certreq_poll);
378	ADD_TEST(test_try_certreq_poll_abort);
379	ADD_TEST(test_exec_GENM_ses);
380	ADD_TEST(test_exchange_certConf);
381	ADD_TEST(test_exchange_error);
382	return 1;
383	}
384
385	#else /* !defined (NDEBUG) */
386
387	int setup_tests(void)
388	{
389	TEST_note("CMP session tests are disabled in this build (NDEBUG).");
390	return 1;
391	}
392
393	#endif