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

/*
 * Copyright 2007-2023 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 "helpers/cmp_testlib.h"

static const char *newkey_f;
static const char *server_cert_f;
static const char *pkcs10_f;

typedef struct test_fixture {
    const char *test_case_name;
    OSSL_CMP_CTX *cmp_ctx;
    /* for msg create tests */
    int bodytype;
    int err_code;
    /* for certConf */
    int fail_info;
    /* for protection tests */
    OSSL_CMP_MSG *msg;
    int expected;
    /* for error and response messages */
    OSSL_CMP_PKISI *si;
} CMP_MSG_TEST_FIXTURE;

static OSSL_LIB_CTX *libctx = NULL;
static OSSL_PROVIDER *default_null_provider = NULL, *provider = NULL;

static unsigned char ref[CMP_TEST_REFVALUE_LENGTH];

static void tear_down(CMP_MSG_TEST_FIXTURE *fixture)
{
    OSSL_CMP_CTX_free(fixture->cmp_ctx);
    OSSL_CMP_MSG_free(fixture->msg);
    OSSL_CMP_PKISI_free(fixture->si);
    OPENSSL_free(fixture);
}

#define SET_OPT_UNPROTECTED_SEND(ctx, val) \
    OSSL_CMP_CTX_set_option((ctx), OSSL_CMP_OPT_UNPROTECTED_SEND, (val))
static CMP_MSG_TEST_FIXTURE *set_up(const char *const test_case_name)
{
    CMP_MSG_TEST_FIXTURE *fixture;

    if (!TEST_ptr(fixture = OPENSSL_zalloc(sizeof(*fixture))))
        return NULL;
    fixture->test_case_name = test_case_name;

    if (!TEST_ptr(fixture->cmp_ctx = OSSL_CMP_CTX_new(libctx, NULL))
            || !TEST_true(SET_OPT_UNPROTECTED_SEND(fixture->cmp_ctx, 1))
            || !TEST_true(OSSL_CMP_CTX_set1_referenceValue(fixture->cmp_ctx,
                                                           ref, sizeof(ref)))) {
        tear_down(fixture);
        return NULL;
    }
    return fixture;
}

static EVP_PKEY *newkey = NULL;
static X509 *cert = NULL;

#define EXECUTE_MSG_CREATION_TEST(expr) \
    do { \
        OSSL_CMP_MSG *msg = NULL; \
        int good = fixture->expected != 0 ? \
            TEST_ptr(msg = (expr)) && TEST_true(valid_asn1_encoding(msg)) : \
            TEST_ptr_null(msg = (expr)); \
 \
        OSSL_CMP_MSG_free(msg); \
        ERR_print_errors_fp(stderr); \
        return good; \
    } while (0)

/*-
 * The following tests call a cmp message creation function.
 * if fixture->expected != 0:
 *         returns 1 if the message is created and syntactically correct.
 * if fixture->expected == 0
 *         returns 1 if message creation returns NULL
 */
static int execute_certreq_create_test(CMP_MSG_TEST_FIXTURE *fixture)
{
    EXECUTE_MSG_CREATION_TEST(ossl_cmp_certreq_new(fixture->cmp_ctx,
                                                   fixture->bodytype,
                                                   NULL));
}

static int execute_errormsg_create_test(CMP_MSG_TEST_FIXTURE *fixture)
{
    EXECUTE_MSG_CREATION_TEST(ossl_cmp_error_new(fixture->cmp_ctx, fixture->si,
                                                 fixture->err_code,
                                                 "details", 0));
}

static int execute_rr_create_test(CMP_MSG_TEST_FIXTURE *fixture)
{
    EXECUTE_MSG_CREATION_TEST(ossl_cmp_rr_new(fixture->cmp_ctx));
}

static int execute_certconf_create_test(CMP_MSG_TEST_FIXTURE *fixture)
{
    EXECUTE_MSG_CREATION_TEST(ossl_cmp_certConf_new
                              (fixture->cmp_ctx, OSSL_CMP_CERTREQID,
                               fixture->fail_info, NULL));
}

static int execute_genm_create_test(CMP_MSG_TEST_FIXTURE *fixture)
{
    EXECUTE_MSG_CREATION_TEST(ossl_cmp_genm_new(fixture->cmp_ctx));
}

static int execute_pollreq_create_test(CMP_MSG_TEST_FIXTURE *fixture)
{
    EXECUTE_MSG_CREATION_TEST(ossl_cmp_pollReq_new(fixture->cmp_ctx, 4711));
}

static int execute_pkimessage_create_test(CMP_MSG_TEST_FIXTURE *fixture)
{
    EXECUTE_MSG_CREATION_TEST(ossl_cmp_msg_create
                              (fixture->cmp_ctx, fixture->bodytype));
}

static int set1_newPkey(OSSL_CMP_CTX *ctx, EVP_PKEY *pkey)
{
    if (!EVP_PKEY_up_ref(pkey))
        return 0;

    if (!OSSL_CMP_CTX_set0_newPkey(ctx, 1, pkey)) {
        EVP_PKEY_free(pkey);
        return 0;
    }
    return 1;
}

static int test_cmp_create_ir_protection_set(void)
{
    OSSL_CMP_CTX *ctx;
    unsigned char secret[16];

    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);

    ctx = fixture->cmp_ctx;
    fixture->bodytype = OSSL_CMP_PKIBODY_IR;
    fixture->err_code = -1;
    fixture->expected = 1;
    if (!TEST_int_eq(1, RAND_bytes_ex(libctx, secret, sizeof(secret), 0))
            || !TEST_true(SET_OPT_UNPROTECTED_SEND(ctx, 0))
            || !TEST_true(set1_newPkey(ctx, newkey))
            || !TEST_true(OSSL_CMP_CTX_set1_secretValue(ctx, secret,
                                                        sizeof(secret)))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_certreq_create_test, tear_down);
    return result;
}

static int test_cmp_create_ir_protection_fails(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->bodytype = OSSL_CMP_PKIBODY_IR;
    fixture->err_code = -1;
    fixture->expected = 0;
    if (!TEST_true(OSSL_CMP_CTX_set1_pkey(fixture->cmp_ctx, newkey))
            || !TEST_true(SET_OPT_UNPROTECTED_SEND(fixture->cmp_ctx, 0))
            /* newkey used by default for signing does not match cert: */
            || !TEST_true(OSSL_CMP_CTX_set1_cert(fixture->cmp_ctx, cert))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_certreq_create_test, tear_down);
    return result;
}

static int test_cmp_create_cr_without_key(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->bodytype = OSSL_CMP_PKIBODY_CR;
    fixture->err_code = -1;
    fixture->expected = 0;
    EXECUTE_TEST(execute_certreq_create_test, tear_down);
    return result;
}

static int test_cmp_create_cr(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->bodytype = OSSL_CMP_PKIBODY_CR;
    fixture->err_code = -1;
    fixture->expected = 1;
    if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_certreq_create_test, tear_down);
    return result;
}

static int test_cmp_create_certreq_with_invalid_bodytype(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->bodytype = OSSL_CMP_PKIBODY_RR;
    fixture->err_code = -1;
    fixture->expected = 0;
    if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_certreq_create_test, tear_down);
    return result;
}

static int test_cmp_create_p10cr(void)
{
    OSSL_CMP_CTX *ctx;
    X509_REQ *p10cr = NULL;

    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    ctx = fixture->cmp_ctx;
    fixture->bodytype = OSSL_CMP_PKIBODY_P10CR;
    fixture->err_code = CMP_R_ERROR_CREATING_CERTREQ;
    fixture->expected = 1;
    if (!TEST_ptr(p10cr = load_csr_der(pkcs10_f, libctx))
            || !TEST_true(set1_newPkey(ctx, newkey))
            || !TEST_true(OSSL_CMP_CTX_set1_p10CSR(ctx, p10cr))) {
        tear_down(fixture);
        fixture = NULL;
    }
    X509_REQ_free(p10cr);
    EXECUTE_TEST(execute_certreq_create_test, tear_down);
    return result;
}

static int test_cmp_create_p10cr_null(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->bodytype = OSSL_CMP_PKIBODY_P10CR;
    fixture->err_code = CMP_R_ERROR_CREATING_CERTREQ;
    fixture->expected = 0;
    if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_certreq_create_test, tear_down);
    return result;
}

static int test_cmp_create_kur(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->bodytype = OSSL_CMP_PKIBODY_KUR;
    fixture->err_code = -1;
    fixture->expected = 1;
    if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))
            || !TEST_true(OSSL_CMP_CTX_set1_oldCert(fixture->cmp_ctx, cert))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_certreq_create_test, tear_down);
    return result;
}

static int test_cmp_create_kur_without_oldcert(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->bodytype = OSSL_CMP_PKIBODY_KUR;
    fixture->err_code = -1;
    fixture->expected = 0;
    if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_certreq_create_test, tear_down);
    return result;
}

static int test_cmp_create_certconf(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->fail_info = 0;
    fixture->expected = 1;
    if (!TEST_true(ossl_cmp_ctx_set0_newCert(fixture->cmp_ctx,
                                             X509_dup(cert)))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_certconf_create_test, tear_down);
    return result;
}

static int test_cmp_create_certconf_badAlg(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->fail_info = 1 << OSSL_CMP_PKIFAILUREINFO_badAlg;
    fixture->expected = 1;
    if (!TEST_true(ossl_cmp_ctx_set0_newCert(fixture->cmp_ctx,
                                             X509_dup(cert)))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_certconf_create_test, tear_down);
    return result;
}

static int test_cmp_create_certconf_fail_info_max(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->fail_info = 1 << OSSL_CMP_PKIFAILUREINFO_MAX;
    fixture->expected = 1;
    if (!TEST_true(ossl_cmp_ctx_set0_newCert(fixture->cmp_ctx,
                                             X509_dup(cert)))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_certconf_create_test, tear_down);
    return result;
}

static int test_cmp_create_error_msg(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->si = OSSL_CMP_STATUSINFO_new(OSSL_CMP_PKISTATUS_rejection,
                                          OSSL_CMP_PKIFAILUREINFO_systemFailure,
                                          NULL);
    fixture->err_code = -1;
    fixture->expected = 1; /* expected: message creation is successful */
    if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_errormsg_create_test, tear_down);
    return result;
}

static int test_cmp_create_pollreq(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->expected = 1;
    EXECUTE_TEST(execute_pollreq_create_test, tear_down);
    return result;
}

static int test_cmp_create_rr(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->expected = 1;
    if (!TEST_true(OSSL_CMP_CTX_set1_oldCert(fixture->cmp_ctx, cert))) {
        tear_down(fixture);
        fixture = NULL;
    }
    EXECUTE_TEST(execute_rr_create_test, tear_down);
    return result;
}

static int test_cmp_create_genm(void)
{
    OSSL_CMP_ITAV *iv = NULL;

    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    fixture->expected = 1;
    iv = OSSL_CMP_ITAV_create(OBJ_nid2obj(NID_id_it_implicitConfirm), NULL);
    if (!TEST_ptr(iv)
            || !TEST_true(OSSL_CMP_CTX_push0_genm_ITAV(fixture->cmp_ctx, iv))) {
        OSSL_CMP_ITAV_free(iv);
        tear_down(fixture);
        fixture = NULL;
    }

    EXECUTE_TEST(execute_genm_create_test, tear_down);
    return result;
}

static int execute_certrep_create(CMP_MSG_TEST_FIXTURE *fixture)
{
    OSSL_CMP_CTX *ctx = fixture->cmp_ctx;
    OSSL_CMP_CERTREPMESSAGE *crepmsg = OSSL_CMP_CERTREPMESSAGE_new();
    OSSL_CMP_CERTRESPONSE *read_cresp, *cresp = OSSL_CMP_CERTRESPONSE_new();
    X509 *certfromresp = NULL;
    int res = 0;

    if (crepmsg == NULL || cresp == NULL)
        goto err;
    if (!ASN1_INTEGER_set(cresp->certReqId, 99))
        goto err;
    if ((cresp->certifiedKeyPair = OSSL_CMP_CERTIFIEDKEYPAIR_new()) == NULL)
        goto err;
    cresp->certifiedKeyPair->certOrEncCert->type =
        OSSL_CMP_CERTORENCCERT_CERTIFICATE;
    if ((cresp->certifiedKeyPair->certOrEncCert->value.certificate =
         X509_dup(cert)) == NULL
            || !sk_OSSL_CMP_CERTRESPONSE_push(crepmsg->response, cresp))
        goto err;
    cresp = NULL;
    read_cresp = ossl_cmp_certrepmessage_get0_certresponse(crepmsg, 99);
    if (!TEST_ptr(read_cresp))
        goto err;
    if (!TEST_ptr_null(ossl_cmp_certrepmessage_get0_certresponse(crepmsg, 88)))
        goto err;
    certfromresp = ossl_cmp_certresponse_get1_cert(ctx, read_cresp);
    if (certfromresp == NULL || !TEST_int_eq(X509_cmp(cert, certfromresp), 0))
        goto err;

    res = 1;
 err:
    X509_free(certfromresp);
    OSSL_CMP_CERTRESPONSE_free(cresp);
    OSSL_CMP_CERTREPMESSAGE_free(crepmsg);
    return res;
}

static int test_cmp_create_certrep(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    EXECUTE_TEST(execute_certrep_create, tear_down);
    return result;
}

static int execute_rp_create(CMP_MSG_TEST_FIXTURE *fixture)
{
    OSSL_CMP_PKISI *si = OSSL_CMP_STATUSINFO_new(33, 44, "a text");
    X509_NAME *issuer = X509_NAME_new();
    ASN1_INTEGER *serial = ASN1_INTEGER_new();
    OSSL_CRMF_CERTID *cid = NULL;
    OSSL_CMP_MSG *rpmsg = NULL;
    int res = 0;

    if (si == NULL || issuer == NULL || serial == NULL)
        goto err;

    if (!X509_NAME_add_entry_by_txt(issuer, "CN", MBSTRING_ASC,
                                    (unsigned char *)"The Issuer", -1, -1, 0)
            || !ASN1_INTEGER_set(serial, 99)
            || (cid = OSSL_CRMF_CERTID_gen(issuer, serial)) == NULL
            || (rpmsg = ossl_cmp_rp_new(fixture->cmp_ctx, si, cid, 1)) == NULL)
        goto err;

    if (!TEST_ptr(ossl_cmp_revrepcontent_get_CertId(rpmsg->body->value.rp, 0)))
        goto err;

    if (!TEST_ptr(ossl_cmp_revrepcontent_get_pkisi(rpmsg->body->value.rp, 0)))
        goto err;

    res = 1;
 err:
    ASN1_INTEGER_free(serial);
    X509_NAME_free(issuer);
    OSSL_CRMF_CERTID_free(cid);
    OSSL_CMP_PKISI_free(si);
    OSSL_CMP_MSG_free(rpmsg);
    return res;
}

static int test_cmp_create_rp(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    EXECUTE_TEST(execute_rp_create, tear_down);
    return result;
}

static int execute_pollrep_create(CMP_MSG_TEST_FIXTURE *fixture)
{
    OSSL_CMP_MSG *pollrep;
    int res = 0;

    pollrep = ossl_cmp_pollRep_new(fixture->cmp_ctx, 77, 2000);
    if (!TEST_ptr(pollrep))
        return 0;
    if (!TEST_ptr(ossl_cmp_pollrepcontent_get0_pollrep(pollrep->body->
                                                       value.pollRep, 77)))
        goto err;
    if (!TEST_ptr_null(ossl_cmp_pollrepcontent_get0_pollrep(pollrep->body->
                                                            value.pollRep, 88)))
        goto err;

    res = 1;
 err:
    OSSL_CMP_MSG_free(pollrep);
    return res;
}

static int test_cmp_create_pollrep(void)
{
    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
    EXECUTE_TEST(execute_pollrep_create, tear_down);
    return result;
}

static int test_cmp_pkimessage_create(int bodytype)
{
    X509_REQ *p10cr = NULL;

    SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);

    switch (fixture->bodytype = bodytype) {
    case OSSL_CMP_PKIBODY_P10CR:
        fixture->expected = 1;
        p10cr = load_csr_der(pkcs10_f, libctx);
        if (!TEST_true(OSSL_CMP_CTX_set1_p10CSR(fixture->cmp_ctx, p10cr))) {
            tear_down(fixture);
            fixture = NULL;
        }
        X509_REQ_free(p10cr);
        break;
    case OSSL_CMP_PKIBODY_IR:
    case OSSL_CMP_PKIBODY_IP:
    case OSSL_CMP_PKIBODY_CR:
    case OSSL_CMP_PKIBODY_CP:
    case OSSL_CMP_PKIBODY_KUR:
    case OSSL_CMP_PKIBODY_KUP:
    case OSSL_CMP_PKIBODY_RR:
    case OSSL_CMP_PKIBODY_RP:
    case OSSL_CMP_PKIBODY_PKICONF:
    case OSSL_CMP_PKIBODY_GENM:
    case OSSL_CMP_PKIBODY_GENP:
    case OSSL_CMP_PKIBODY_ERROR:
    case OSSL_CMP_PKIBODY_CERTCONF:
    case OSSL_CMP_PKIBODY_POLLREQ:
    case OSSL_CMP_PKIBODY_POLLREP:
        fixture->expected = 1;
        break;
    default:
        fixture->expected = 0;
        break;
    }

    EXECUTE_TEST(execute_pkimessage_create_test, tear_down);
    return result;
}

void cleanup_tests(void)
{
    EVP_PKEY_free(newkey);
    X509_free(cert);
    OSSL_PROVIDER_unload(default_null_provider);
    OSSL_PROVIDER_unload(provider);
    OSSL_LIB_CTX_free(libctx);
}

#define USAGE "new.key server.crt pkcs10.der module_name [module_conf_file]\n"
OPT_TEST_DECLARE_USAGE(USAGE)

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

    if (!TEST_ptr(newkey_f = test_get_argument(0))
            || !TEST_ptr(server_cert_f = test_get_argument(1))
            || !TEST_ptr(pkcs10_f = test_get_argument(2))) {
        TEST_error("usage: cmp_msg_test %s", USAGE);
        return 0;
    }

    if (!test_arg_libctx(&libctx, &default_null_provider, &provider, 3, USAGE))
        return 0;

    if (!TEST_ptr(newkey = load_pkey_pem(newkey_f, libctx))
            || !TEST_ptr(cert = load_cert_pem(server_cert_f, libctx))
            || !TEST_int_eq(1, RAND_bytes_ex(libctx, ref, sizeof(ref), 0))) {
        cleanup_tests();
        return 0;
    }

    /* Message creation tests */
    ADD_TEST(test_cmp_create_certreq_with_invalid_bodytype);
    ADD_TEST(test_cmp_create_ir_protection_fails);
    ADD_TEST(test_cmp_create_ir_protection_set);
    ADD_TEST(test_cmp_create_error_msg);
    ADD_TEST(test_cmp_create_certconf);
    ADD_TEST(test_cmp_create_certconf_badAlg);
    ADD_TEST(test_cmp_create_certconf_fail_info_max);
    ADD_TEST(test_cmp_create_kur);
    ADD_TEST(test_cmp_create_kur_without_oldcert);
    ADD_TEST(test_cmp_create_cr);
    ADD_TEST(test_cmp_create_cr_without_key);
    ADD_TEST(test_cmp_create_p10cr);
    ADD_TEST(test_cmp_create_p10cr_null);
    ADD_TEST(test_cmp_create_pollreq);
    ADD_TEST(test_cmp_create_rr);
    ADD_TEST(test_cmp_create_rp);
    ADD_TEST(test_cmp_create_genm);
    ADD_TEST(test_cmp_create_certrep);
    ADD_TEST(test_cmp_create_pollrep);
    ADD_ALL_TESTS_NOSUBTEST(test_cmp_pkimessage_create,
                            OSSL_CMP_PKIBODY_POLLREP + 1);
    return 1;
}
Commit	Line	Data
3dbc5156	1	/*
da1c088f	2	* Copyright 2007-2023 The OpenSSL Project Authors. All Rights Reserved.
3dbc5156 DDO	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
20f8bc72	12	#include "helpers/cmp_testlib.h"
3dbc5156	13
5a7734cd	14	static const char *newkey_f;
3dbc5156 DDO	15	static const char *server_cert_f;
	16	static const char *pkcs10_f;
	17
	18	typedef struct test_fixture {
	19	const char *test_case_name;
	20	OSSL_CMP_CTX *cmp_ctx;
	21	/* for msg create tests */
	22	int bodytype;
	23	int err_code;
	24	/* for certConf */
	25	int fail_info;
	26	/* for protection tests */
	27	OSSL_CMP_MSG *msg;
	28	int expected;
	29	/* for error and response messages */
	30	OSSL_CMP_PKISI *si;
	31	} CMP_MSG_TEST_FIXTURE;
	32
b4250010	33	static OSSL_LIB_CTX *libctx = NULL;
5a7734cd DDO	34	static OSSL_PROVIDER default_null_provider = NULL, provider = NULL;
5a7734cd DDO	35
3dbc5156 DDO	36	static unsigned char ref[CMP_TEST_REFVALUE_LENGTH];
	37
	38	static void tear_down(CMP_MSG_TEST_FIXTURE *fixture)
	39	{
	40	OSSL_CMP_CTX_free(fixture->cmp_ctx);
	41	OSSL_CMP_MSG_free(fixture->msg);
	42	OSSL_CMP_PKISI_free(fixture->si);
	43	OPENSSL_free(fixture);
	44	}
	45
	46	#define SET_OPT_UNPROTECTED_SEND(ctx, val) \
	47	OSSL_CMP_CTX_set_option((ctx), OSSL_CMP_OPT_UNPROTECTED_SEND, (val))
	48	static CMP_MSG_TEST_FIXTURE set_up(const char const test_case_name)
	49	{
	50	CMP_MSG_TEST_FIXTURE *fixture;
	51
	52	if (!TEST_ptr(fixture = OPENSSL_zalloc(sizeof(*fixture))))
	53	return NULL;
	54	fixture->test_case_name = test_case_name;
	55
5a7734cd	56	if (!TEST_ptr(fixture->cmp_ctx = OSSL_CMP_CTX_new(libctx, NULL))
3dbc5156 DDO	57	\|\| !TEST_true(SET_OPT_UNPROTECTED_SEND(fixture->cmp_ctx, 1))
	58	\|\| !TEST_true(OSSL_CMP_CTX_set1_referenceValue(fixture->cmp_ctx,
	59	ref, sizeof(ref)))) {
	60	tear_down(fixture);
	61	return NULL;
	62	}
	63	return fixture;
	64	}
	65
	66	static EVP_PKEY *newkey = NULL;
	67	static X509 *cert = NULL;
	68
	69	#define EXECUTE_MSG_CREATION_TEST(expr) \
	70	do { \
	71	OSSL_CMP_MSG *msg = NULL; \
	72	int good = fixture->expected != 0 ? \
	73	TEST_ptr(msg = (expr)) && TEST_true(valid_asn1_encoding(msg)) : \
	74	TEST_ptr_null(msg = (expr)); \
	75	\
	76	OSSL_CMP_MSG_free(msg); \
7eb48cfc	77	ERR_print_errors_fp(stderr); \
3dbc5156 DDO	78	return good; \
	79	} while (0)
	80
	81	/*-
	82	* The following tests call a cmp message creation function.
	83	* if fixture->expected != 0:
	84	* returns 1 if the message is created and syntactically correct.
	85	* if fixture->expected == 0
	86	* returns 1 if message creation returns NULL
	87	*/
	88	static int execute_certreq_create_test(CMP_MSG_TEST_FIXTURE *fixture)
	89	{
299e0f1e	90	EXECUTE_MSG_CREATION_TEST(ossl_cmp_certreq_new(fixture->cmp_ctx,
3dbc5156	91	fixture->bodytype,
299e0f1e	92	NULL));
3dbc5156 DDO	93	}
	94
	95	static int execute_errormsg_create_test(CMP_MSG_TEST_FIXTURE *fixture)
	96	{
	97	EXECUTE_MSG_CREATION_TEST(ossl_cmp_error_new(fixture->cmp_ctx, fixture->si,
	98	fixture->err_code,
62dcd2aa	99	"details", 0));
3dbc5156 DDO	100	}
	101
	102	static int execute_rr_create_test(CMP_MSG_TEST_FIXTURE *fixture)
	103	{
	104	EXECUTE_MSG_CREATION_TEST(ossl_cmp_rr_new(fixture->cmp_ctx));
	105	}
	106
	107	static int execute_certconf_create_test(CMP_MSG_TEST_FIXTURE *fixture)
	108	{
	109	EXECUTE_MSG_CREATION_TEST(ossl_cmp_certConf_new
25b18e62 DDO	110	(fixture->cmp_ctx, OSSL_CMP_CERTREQID,
25b18e62 DDO	111	fixture->fail_info, NULL));
3dbc5156 DDO	112	}
	113
	114	static int execute_genm_create_test(CMP_MSG_TEST_FIXTURE *fixture)
	115	{
	116	EXECUTE_MSG_CREATION_TEST(ossl_cmp_genm_new(fixture->cmp_ctx));
	117	}
	118
	119	static int execute_pollreq_create_test(CMP_MSG_TEST_FIXTURE *fixture)
	120	{
	121	EXECUTE_MSG_CREATION_TEST(ossl_cmp_pollReq_new(fixture->cmp_ctx, 4711));
	122	}
	123
	124	static int execute_pkimessage_create_test(CMP_MSG_TEST_FIXTURE *fixture)
	125	{
	126	EXECUTE_MSG_CREATION_TEST(ossl_cmp_msg_create
	127	(fixture->cmp_ctx, fixture->bodytype));
	128	}
	129
235595c4	130	static int set1_newPkey(OSSL_CMP_CTX ctx, EVP_PKEY pkey)
3dbc5156 DDO	131	{
	132	if (!EVP_PKEY_up_ref(pkey))
	133	return 0;
	134
	135	if (!OSSL_CMP_CTX_set0_newPkey(ctx, 1, pkey)) {
	136	EVP_PKEY_free(pkey);
	137	return 0;
	138	}
	139	return 1;
	140	}
	141
	142	static int test_cmp_create_ir_protection_set(void)
	143	{
06cee80a	144	OSSL_CMP_CTX *ctx;
3dbc5156 DDO	145	unsigned char secret[16];
3dbc5156 DDO	146
06cee80a DDO	147	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	148
	149	ctx = fixture->cmp_ctx;
3dbc5156	150	fixture->bodytype = OSSL_CMP_PKIBODY_IR;
bce58e7f	151	fixture->err_code = -1;
3dbc5156	152	fixture->expected = 1;
23e97567	153	if (!TEST_int_eq(1, RAND_bytes_ex(libctx, secret, sizeof(secret), 0))
3dbc5156 DDO	154	\|\| !TEST_true(SET_OPT_UNPROTECTED_SEND(ctx, 0))
	155	\|\| !TEST_true(set1_newPkey(ctx, newkey))
	156	\|\| !TEST_true(OSSL_CMP_CTX_set1_secretValue(ctx, secret,
	157	sizeof(secret)))) {
	158	tear_down(fixture);
	159	fixture = NULL;
	160	}
	161	EXECUTE_TEST(execute_certreq_create_test, tear_down);
	162	return result;
	163	}
	164
	165	static int test_cmp_create_ir_protection_fails(void)
	166	{
	167	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	168	fixture->bodytype = OSSL_CMP_PKIBODY_IR;
bce58e7f	169	fixture->err_code = -1;
3dbc5156 DDO	170	fixture->expected = 0;
	171	if (!TEST_true(OSSL_CMP_CTX_set1_pkey(fixture->cmp_ctx, newkey))
	172	\|\| !TEST_true(SET_OPT_UNPROTECTED_SEND(fixture->cmp_ctx, 0))
7e765f46	173	/* newkey used by default for signing does not match cert: */
63f1883d	174	\|\| !TEST_true(OSSL_CMP_CTX_set1_cert(fixture->cmp_ctx, cert))) {
3dbc5156 DDO	175	tear_down(fixture);
	176	fixture = NULL;
	177	}
	178	EXECUTE_TEST(execute_certreq_create_test, tear_down);
	179	return result;
	180	}
	181
	182	static int test_cmp_create_cr_without_key(void)
	183	{
	184	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	185	fixture->bodytype = OSSL_CMP_PKIBODY_CR;
bce58e7f	186	fixture->err_code = -1;
3dbc5156 DDO	187	fixture->expected = 0;
	188	EXECUTE_TEST(execute_certreq_create_test, tear_down);
	189	return result;
	190	}
	191
	192	static int test_cmp_create_cr(void)
	193	{
	194	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	195	fixture->bodytype = OSSL_CMP_PKIBODY_CR;
bce58e7f	196	fixture->err_code = -1;
3dbc5156 DDO	197	fixture->expected = 1;
	198	if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))) {
	199	tear_down(fixture);
	200	fixture = NULL;
	201	}
	202	EXECUTE_TEST(execute_certreq_create_test, tear_down);
	203	return result;
	204	}
	205
	206	static int test_cmp_create_certreq_with_invalid_bodytype(void)
	207	{
	208	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	209	fixture->bodytype = OSSL_CMP_PKIBODY_RR;
bce58e7f	210	fixture->err_code = -1;
3dbc5156 DDO	211	fixture->expected = 0;
	212	if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))) {
	213	tear_down(fixture);
	214	fixture = NULL;
	215	}
	216	EXECUTE_TEST(execute_certreq_create_test, tear_down);
	217	return result;
	218	}
	219
	220	static int test_cmp_create_p10cr(void)
	221	{
06cee80a	222	OSSL_CMP_CTX *ctx;
3dbc5156 DDO	223	X509_REQ *p10cr = NULL;
3dbc5156 DDO	224
06cee80a DDO	225	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
06cee80a DDO	226	ctx = fixture->cmp_ctx;
3dbc5156	227	fixture->bodytype = OSSL_CMP_PKIBODY_P10CR;
299e0f1e	228	fixture->err_code = CMP_R_ERROR_CREATING_CERTREQ;
3dbc5156	229	fixture->expected = 1;
d2b6c062	230	if (!TEST_ptr(p10cr = load_csr_der(pkcs10_f, libctx))
3dbc5156 DDO	231	\|\| !TEST_true(set1_newPkey(ctx, newkey))
	232	\|\| !TEST_true(OSSL_CMP_CTX_set1_p10CSR(ctx, p10cr))) {
	233	tear_down(fixture);
	234	fixture = NULL;
	235	}
	236	X509_REQ_free(p10cr);
	237	EXECUTE_TEST(execute_certreq_create_test, tear_down);
	238	return result;
	239	}
	240
	241	static int test_cmp_create_p10cr_null(void)
	242	{
	243	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	244	fixture->bodytype = OSSL_CMP_PKIBODY_P10CR;
299e0f1e	245	fixture->err_code = CMP_R_ERROR_CREATING_CERTREQ;
3dbc5156 DDO	246	fixture->expected = 0;
	247	if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))) {
	248	tear_down(fixture);
	249	fixture = NULL;
	250	}
	251	EXECUTE_TEST(execute_certreq_create_test, tear_down);
	252	return result;
	253	}
	254
	255	static int test_cmp_create_kur(void)
	256	{
	257	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	258	fixture->bodytype = OSSL_CMP_PKIBODY_KUR;
bce58e7f	259	fixture->err_code = -1;
3dbc5156 DDO	260	fixture->expected = 1;
	261	if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))
	262	\|\| !TEST_true(OSSL_CMP_CTX_set1_oldCert(fixture->cmp_ctx, cert))) {
	263	tear_down(fixture);
	264	fixture = NULL;
	265	}
	266	EXECUTE_TEST(execute_certreq_create_test, tear_down);
	267	return result;
	268	}
	269
	270	static int test_cmp_create_kur_without_oldcert(void)
	271	{
	272	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	273	fixture->bodytype = OSSL_CMP_PKIBODY_KUR;
bce58e7f	274	fixture->err_code = -1;
3dbc5156 DDO	275	fixture->expected = 0;
	276	if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))) {
	277	tear_down(fixture);
	278	fixture = NULL;
	279	}
	280	EXECUTE_TEST(execute_certreq_create_test, tear_down);
	281	return result;
	282	}
	283
	284	static int test_cmp_create_certconf(void)
	285	{
	286	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	287	fixture->fail_info = 0;
	288	fixture->expected = 1;
	289	if (!TEST_true(ossl_cmp_ctx_set0_newCert(fixture->cmp_ctx,
ee46dfbf	290	X509_dup(cert)))) {
3dbc5156 DDO	291	tear_down(fixture);
	292	fixture = NULL;
	293	}
	294	EXECUTE_TEST(execute_certconf_create_test, tear_down);
	295	return result;
	296	}
	297
	298	static int test_cmp_create_certconf_badAlg(void)
	299	{
	300	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	301	fixture->fail_info = 1 << OSSL_CMP_PKIFAILUREINFO_badAlg;
	302	fixture->expected = 1;
	303	if (!TEST_true(ossl_cmp_ctx_set0_newCert(fixture->cmp_ctx,
ee46dfbf	304	X509_dup(cert)))) {
3dbc5156 DDO	305	tear_down(fixture);
	306	fixture = NULL;
	307	}
	308	EXECUTE_TEST(execute_certconf_create_test, tear_down);
	309	return result;
	310	}
	311
	312	static int test_cmp_create_certconf_fail_info_max(void)
	313	{
	314	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	315	fixture->fail_info = 1 << OSSL_CMP_PKIFAILUREINFO_MAX;
	316	fixture->expected = 1;
	317	if (!TEST_true(ossl_cmp_ctx_set0_newCert(fixture->cmp_ctx,
ee46dfbf	318	X509_dup(cert)))) {
3dbc5156 DDO	319	tear_down(fixture);
	320	fixture = NULL;
	321	}
	322	EXECUTE_TEST(execute_certconf_create_test, tear_down);
	323	return result;
	324	}
	325
	326	static int test_cmp_create_error_msg(void)
	327	{
	328	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
62dcd2aa	329	fixture->si = OSSL_CMP_STATUSINFO_new(OSSL_CMP_PKISTATUS_rejection,
3dbc5156 DDO	330	OSSL_CMP_PKIFAILUREINFO_systemFailure,
	331	NULL);
	332	fixture->err_code = -1;
235595c4	333	fixture->expected = 1; /* expected: message creation is successful */
3dbc5156 DDO	334	if (!TEST_true(set1_newPkey(fixture->cmp_ctx, newkey))) {
	335	tear_down(fixture);
	336	fixture = NULL;
	337	}
	338	EXECUTE_TEST(execute_errormsg_create_test, tear_down);
	339	return result;
	340	}
	341
3dbc5156 DDO	342	static int test_cmp_create_pollreq(void)
	343	{
	344	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	345	fixture->expected = 1;
	346	EXECUTE_TEST(execute_pollreq_create_test, tear_down);
	347	return result;
	348	}
	349
	350	static int test_cmp_create_rr(void)
	351	{
	352	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	353	fixture->expected = 1;
	354	if (!TEST_true(OSSL_CMP_CTX_set1_oldCert(fixture->cmp_ctx, cert))) {
	355	tear_down(fixture);
	356	fixture = NULL;
	357	}
	358	EXECUTE_TEST(execute_rr_create_test, tear_down);
	359	return result;
	360	}
	361
	362	static int test_cmp_create_genm(void)
	363	{
	364	OSSL_CMP_ITAV *iv = NULL;
	365
	366	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	367	fixture->expected = 1;
	368	iv = OSSL_CMP_ITAV_create(OBJ_nid2obj(NID_id_it_implicitConfirm), NULL);
7e765f46	369	if (!TEST_ptr(iv)
3dbc5156 DDO	370	\|\| !TEST_true(OSSL_CMP_CTX_push0_genm_ITAV(fixture->cmp_ctx, iv))) {
	371	OSSL_CMP_ITAV_free(iv);
	372	tear_down(fixture);
	373	fixture = NULL;
	374	}
	375
	376	EXECUTE_TEST(execute_genm_create_test, tear_down);
	377	return result;
	378	}
	379
	380	static int execute_certrep_create(CMP_MSG_TEST_FIXTURE *fixture)
	381	{
6d1f50b5	382	OSSL_CMP_CTX *ctx = fixture->cmp_ctx;
3dbc5156 DDO	383	OSSL_CMP_CERTREPMESSAGE *crepmsg = OSSL_CMP_CERTREPMESSAGE_new();
3dbc5156 DDO	384	OSSL_CMP_CERTRESPONSE read_cresp, cresp = OSSL_CMP_CERTRESPONSE_new();
3dbc5156 DDO	385	X509 *certfromresp = NULL;
	386	int res = 0;
	387
	388	if (crepmsg == NULL \|\| cresp == NULL)
	389	goto err;
	390	if (!ASN1_INTEGER_set(cresp->certReqId, 99))
	391	goto err;
	392	if ((cresp->certifiedKeyPair = OSSL_CMP_CERTIFIEDKEYPAIR_new()) == NULL)
	393	goto err;
	394	cresp->certifiedKeyPair->certOrEncCert->type =
	395	OSSL_CMP_CERTORENCCERT_CERTIFICATE;
	396	if ((cresp->certifiedKeyPair->certOrEncCert->value.certificate =
ee46dfbf	397	X509_dup(cert)) == NULL
3dbc5156 DDO	398	\|\| !sk_OSSL_CMP_CERTRESPONSE_push(crepmsg->response, cresp))
	399	goto err;
	400	cresp = NULL;
	401	read_cresp = ossl_cmp_certrepmessage_get0_certresponse(crepmsg, 99);
	402	if (!TEST_ptr(read_cresp))
	403	goto err;
	404	if (!TEST_ptr_null(ossl_cmp_certrepmessage_get0_certresponse(crepmsg, 88)))
	405	goto err;
e0f1ec3b	406	certfromresp = ossl_cmp_certresponse_get1_cert(ctx, read_cresp);
3dbc5156 DDO	407	if (certfromresp == NULL \|\| !TEST_int_eq(X509_cmp(cert, certfromresp), 0))
	408	goto err;
	409
	410	res = 1;
	411	err:
	412	X509_free(certfromresp);
	413	OSSL_CMP_CERTRESPONSE_free(cresp);
	414	OSSL_CMP_CERTREPMESSAGE_free(crepmsg);
	415	return res;
	416	}
	417
	418	static int test_cmp_create_certrep(void)
	419	{
	420	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	421	EXECUTE_TEST(execute_certrep_create, tear_down);
	422	return result;
	423	}
	424
3dbc5156 DDO	425	static int execute_rp_create(CMP_MSG_TEST_FIXTURE *fixture)
3dbc5156 DDO	426	{
62dcd2aa	427	OSSL_CMP_PKISI *si = OSSL_CMP_STATUSINFO_new(33, 44, "a text");
3dbc5156 DDO	428	X509_NAME *issuer = X509_NAME_new();
	429	ASN1_INTEGER *serial = ASN1_INTEGER_new();
	430	OSSL_CRMF_CERTID *cid = NULL;
	431	OSSL_CMP_MSG *rpmsg = NULL;
	432	int res = 0;
	433
	434	if (si == NULL \|\| issuer == NULL \|\| serial == NULL)
	435	goto err;
	436
	437	if (!X509_NAME_add_entry_by_txt(issuer, "CN", MBSTRING_ASC,
235595c4	438	(unsigned char *)"The Issuer", -1, -1, 0)
3dbc5156 DDO	439	\|\| !ASN1_INTEGER_set(serial, 99)
	440	\|\| (cid = OSSL_CRMF_CERTID_gen(issuer, serial)) == NULL
	441	\|\| (rpmsg = ossl_cmp_rp_new(fixture->cmp_ctx, si, cid, 1)) == NULL)
	442	goto err;
	443
	444	if (!TEST_ptr(ossl_cmp_revrepcontent_get_CertId(rpmsg->body->value.rp, 0)))
	445	goto err;
	446
62dcd2aa	447	if (!TEST_ptr(ossl_cmp_revrepcontent_get_pkisi(rpmsg->body->value.rp, 0)))
3dbc5156 DDO	448	goto err;
	449
	450	res = 1;
	451	err:
	452	ASN1_INTEGER_free(serial);
	453	X509_NAME_free(issuer);
	454	OSSL_CRMF_CERTID_free(cid);
	455	OSSL_CMP_PKISI_free(si);
	456	OSSL_CMP_MSG_free(rpmsg);
	457	return res;
	458	}
	459
	460	static int test_cmp_create_rp(void)
	461	{
	462	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	463	EXECUTE_TEST(execute_rp_create, tear_down);
	464	return result;
	465	}
	466
	467	static int execute_pollrep_create(CMP_MSG_TEST_FIXTURE *fixture)
	468	{
	469	OSSL_CMP_MSG *pollrep;
	470	int res = 0;
	471
	472	pollrep = ossl_cmp_pollRep_new(fixture->cmp_ctx, 77, 2000);
	473	if (!TEST_ptr(pollrep))
	474	return 0;
235595c4 DDO	475	if (!TEST_ptr(ossl_cmp_pollrepcontent_get0_pollrep(pollrep->body->
235595c4 DDO	476	value.pollRep, 77)))
3dbc5156	477	goto err;
235595c4 DDO	478	if (!TEST_ptr_null(ossl_cmp_pollrepcontent_get0_pollrep(pollrep->body->
235595c4 DDO	479	value.pollRep, 88)))
3dbc5156 DDO	480	goto err;
	481
	482	res = 1;
	483	err:
	484	OSSL_CMP_MSG_free(pollrep);
	485	return res;
	486	}
	487
	488	static int test_cmp_create_pollrep(void)
	489	{
	490	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	491	EXECUTE_TEST(execute_pollrep_create, tear_down);
	492	return result;
	493	}
	494
	495	static int test_cmp_pkimessage_create(int bodytype)
	496	{
	497	X509_REQ *p10cr = NULL;
	498
	499	SETUP_TEST_FIXTURE(CMP_MSG_TEST_FIXTURE, set_up);
	500
	501	switch (fixture->bodytype = bodytype) {
	502	case OSSL_CMP_PKIBODY_P10CR:
	503	fixture->expected = 1;
d2b6c062	504	p10cr = load_csr_der(pkcs10_f, libctx);
0b7368dd	505	if (!TEST_true(OSSL_CMP_CTX_set1_p10CSR(fixture->cmp_ctx, p10cr))) {
3dbc5156 DDO	506	tear_down(fixture);
	507	fixture = NULL;
	508	}
	509	X509_REQ_free(p10cr);
	510	break;
	511	case OSSL_CMP_PKIBODY_IR:
	512	case OSSL_CMP_PKIBODY_IP:
	513	case OSSL_CMP_PKIBODY_CR:
	514	case OSSL_CMP_PKIBODY_CP:
	515	case OSSL_CMP_PKIBODY_KUR:
	516	case OSSL_CMP_PKIBODY_KUP:
	517	case OSSL_CMP_PKIBODY_RR:
	518	case OSSL_CMP_PKIBODY_RP:
	519	case OSSL_CMP_PKIBODY_PKICONF:
	520	case OSSL_CMP_PKIBODY_GENM:
	521	case OSSL_CMP_PKIBODY_GENP:
	522	case OSSL_CMP_PKIBODY_ERROR:
	523	case OSSL_CMP_PKIBODY_CERTCONF:
	524	case OSSL_CMP_PKIBODY_POLLREQ:
	525	case OSSL_CMP_PKIBODY_POLLREP:
	526	fixture->expected = 1;
	527	break;
	528	default:
	529	fixture->expected = 0;
	530	break;
	531	}
	532
	533	EXECUTE_TEST(execute_pkimessage_create_test, tear_down);
	534	return result;
	535	}
	536
	537	void cleanup_tests(void)
	538	{
	539	EVP_PKEY_free(newkey);
	540	X509_free(cert);
8835940d DC	541	OSSL_PROVIDER_unload(default_null_provider);
8835940d DC	542	OSSL_PROVIDER_unload(provider);
b4250010	543	OSSL_LIB_CTX_free(libctx);
3dbc5156 DDO	544	}
3dbc5156 DDO	545
5a7734cd DDO	546	#define USAGE "new.key server.crt pkcs10.der module_name [module_conf_file]\n"
	547	OPT_TEST_DECLARE_USAGE(USAGE)
	548
3dbc5156 DDO	549	int setup_tests(void)
3dbc5156 DDO	550	{
8d242823 MC	551	if (!test_skip_common_options()) {
	552	TEST_error("Error parsing test options\n");
	553	return 0;
	554	}
	555
5a7734cd DDO	556	if (!TEST_ptr(newkey_f = test_get_argument(0))
	557	\|\| !TEST_ptr(server_cert_f = test_get_argument(1))
	558	\|\| !TEST_ptr(pkcs10_f = test_get_argument(2))) {
	559	TEST_error("usage: cmp_msg_test %s", USAGE);
3dbc5156 DDO	560	return 0;
	561	}
	562
bca7ad6e	563	if (!test_arg_libctx(&libctx, &default_null_provider, &provider, 3, USAGE))
5a7734cd DDO	564	return 0;
5a7734cd DDO	565
0b7368dd DDO	566	if (!TEST_ptr(newkey = load_pkey_pem(newkey_f, libctx))
0b7368dd DDO	567	\|\| !TEST_ptr(cert = load_cert_pem(server_cert_f, libctx))
23e97567	568	\|\| !TEST_int_eq(1, RAND_bytes_ex(libctx, ref, sizeof(ref), 0))) {
3dbc5156 DDO	569	cleanup_tests();
	570	return 0;
	571	}
	572
	573	/* Message creation tests */
	574	ADD_TEST(test_cmp_create_certreq_with_invalid_bodytype);
	575	ADD_TEST(test_cmp_create_ir_protection_fails);
	576	ADD_TEST(test_cmp_create_ir_protection_set);
	577	ADD_TEST(test_cmp_create_error_msg);
	578	ADD_TEST(test_cmp_create_certconf);
	579	ADD_TEST(test_cmp_create_certconf_badAlg);
	580	ADD_TEST(test_cmp_create_certconf_fail_info_max);
	581	ADD_TEST(test_cmp_create_kur);
	582	ADD_TEST(test_cmp_create_kur_without_oldcert);
	583	ADD_TEST(test_cmp_create_cr);
	584	ADD_TEST(test_cmp_create_cr_without_key);
	585	ADD_TEST(test_cmp_create_p10cr);
	586	ADD_TEST(test_cmp_create_p10cr_null);
	587	ADD_TEST(test_cmp_create_pollreq);
	588	ADD_TEST(test_cmp_create_rr);
	589	ADD_TEST(test_cmp_create_rp);
	590	ADD_TEST(test_cmp_create_genm);
	591	ADD_TEST(test_cmp_create_certrep);
	592	ADD_TEST(test_cmp_create_pollrep);
	593	ADD_ALL_TESTS_NOSUBTEST(test_cmp_pkimessage_create,
	594	OSSL_CMP_PKIBODY_POLLREP + 1);
	595	return 1;
	596	}