[qemu.git] / tests / test-crypto-tlssession.c

/*
 * Copyright (C) 2015 Red Hat, Inc.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library.  If not, see
 * <http://www.gnu.org/licenses/>.
 *
 * Author: Daniel P. Berrange <[email protected]>
 */

#include <stdlib.h>
#include <fcntl.h>

#include "config-host.h"
#include "crypto-tls-x509-helpers.h"
#include "crypto/tlscredsx509.h"
#include "crypto/tlssession.h"
#include "qom/object_interfaces.h"
#include "qemu/sockets.h"
#include "qemu/acl.h"

#ifdef QCRYPTO_HAVE_TLS_TEST_SUPPORT

#define WORKDIR "tests/test-crypto-tlssession-work/"
#define KEYFILE WORKDIR "key-ctx.pem"

struct QCryptoTLSSessionTestData {
    const char *servercacrt;
    const char *clientcacrt;
    const char *servercrt;
    const char *clientcrt;
    bool expectServerFail;
    bool expectClientFail;
    const char *hostname;
    const char *const *wildcards;
};


static ssize_t testWrite(const char *buf, size_t len, void *opaque)
{
    int *fd = opaque;

    return write(*fd, buf, len);
}

static ssize_t testRead(char *buf, size_t len, void *opaque)
{
    int *fd = opaque;

    return read(*fd, buf, len);
}

static QCryptoTLSCreds *test_tls_creds_create(QCryptoTLSCredsEndpoint endpoint,
                                              const char *certdir,
                                              Error **errp)
{
    Error *err = NULL;
    Object *parent = object_get_objects_root();
    Object *creds = object_new_with_props(
        TYPE_QCRYPTO_TLS_CREDS_X509,
        parent,
        (endpoint == QCRYPTO_TLS_CREDS_ENDPOINT_SERVER ?
         "testtlscredsserver" : "testtlscredsclient"),
        &err,
        "endpoint", (endpoint == QCRYPTO_TLS_CREDS_ENDPOINT_SERVER ?
                     "server" : "client"),
        "dir", certdir,
        "verify-peer", "yes",
        /* We skip initial sanity checks here because we
         * want to make sure that problems are being
         * detected at the TLS session validation stage,
         * and the test-crypto-tlscreds test already
         * validate the sanity check code.
         */
        "sanity-check", "no",
        NULL
        );

    if (err) {
        error_propagate(errp, err);
        return NULL;
    }
    return QCRYPTO_TLS_CREDS(creds);
}


/*
 * This tests validation checking of peer certificates
 *
 * This is replicating the checks that are done for an
 * active TLS session after handshake completes. To
 * simulate that we create our TLS contexts, skipping
 * sanity checks. We then get a socketpair, and
 * initiate a TLS session across them. Finally do
 * do actual cert validation tests
 */
static void test_crypto_tls_session(const void *opaque)
{
    struct QCryptoTLSSessionTestData *data =
        (struct QCryptoTLSSessionTestData *)opaque;
    QCryptoTLSCreds *clientCreds;
    QCryptoTLSCreds *serverCreds;
    QCryptoTLSSession *clientSess = NULL;
    QCryptoTLSSession *serverSess = NULL;
    qemu_acl *acl;
    const char * const *wildcards;
    int channel[2];
    bool clientShake = false;
    bool serverShake = false;
    Error *err = NULL;
    int ret;

    /* We'll use this for our fake client-server connection */
    ret = socketpair(AF_UNIX, SOCK_STREAM, 0, channel);
    g_assert(ret == 0);

    /*
     * We have an evil loop to do the handshake in a single
     * thread, so we need these non-blocking to avoid deadlock
     * of ourselves
     */
    qemu_set_nonblock(channel[0]);
    qemu_set_nonblock(channel[1]);

#define CLIENT_CERT_DIR "tests/test-crypto-tlssession-client/"
#define SERVER_CERT_DIR "tests/test-crypto-tlssession-server/"
    mkdir(CLIENT_CERT_DIR, 0700);
    mkdir(SERVER_CERT_DIR, 0700);

    unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT);
    unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_CERT);
    unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_KEY);

    unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT);
    unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_CERT);
    unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_KEY);

    g_assert(link(data->servercacrt,
                  SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT) == 0);
    g_assert(link(data->servercrt,
                  SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_CERT) == 0);
    g_assert(link(KEYFILE,
                  SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_KEY) == 0);

    g_assert(link(data->clientcacrt,
                  CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT) == 0);
    g_assert(link(data->clientcrt,
                  CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_CERT) == 0);
    g_assert(link(KEYFILE,
                  CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_KEY) == 0);

    clientCreds = test_tls_creds_create(
        QCRYPTO_TLS_CREDS_ENDPOINT_CLIENT,
        CLIENT_CERT_DIR,
        &err);
    g_assert(clientCreds != NULL);

    serverCreds = test_tls_creds_create(
        QCRYPTO_TLS_CREDS_ENDPOINT_SERVER,
        SERVER_CERT_DIR,
        &err);
    g_assert(serverCreds != NULL);

    acl = qemu_acl_init("tlssessionacl");
    qemu_acl_reset(acl);
    wildcards = data->wildcards;
    while (wildcards && *wildcards) {
        qemu_acl_append(acl, 0, *wildcards);
        wildcards++;
    }

    /* Now the real part of the test, setup the sessions */
    clientSess = qcrypto_tls_session_new(
        clientCreds, data->hostname, NULL,
        QCRYPTO_TLS_CREDS_ENDPOINT_CLIENT, &err);
    serverSess = qcrypto_tls_session_new(
        serverCreds, NULL,
        data->wildcards ? "tlssessionacl" : NULL,
        QCRYPTO_TLS_CREDS_ENDPOINT_SERVER, &err);

    g_assert(clientSess != NULL);
    g_assert(serverSess != NULL);

    /* For handshake to work, we need to set the I/O callbacks
     * to read/write over the socketpair
     */
    qcrypto_tls_session_set_callbacks(serverSess,
                                      testWrite, testRead,
                                      &channel[0]);
    qcrypto_tls_session_set_callbacks(clientSess,
                                      testWrite, testRead,
                                      &channel[1]);

    /*
     * Finally we loop around & around doing handshake on each
     * session until we get an error, or the handshake completes.
     * This relies on the socketpair being nonblocking to avoid
     * deadlocking ourselves upon handshake
     */
    do {
        int rv;
        if (!serverShake) {
            rv = qcrypto_tls_session_handshake(serverSess,
                                               &err);
            g_assert(rv >= 0);
            if (qcrypto_tls_session_get_handshake_status(serverSess) ==
                QCRYPTO_TLS_HANDSHAKE_COMPLETE) {
                serverShake = true;
            }
        }
        if (!clientShake) {
            rv = qcrypto_tls_session_handshake(clientSess,
                                               &err);
            g_assert(rv >= 0);
            if (qcrypto_tls_session_get_handshake_status(clientSess) ==
                QCRYPTO_TLS_HANDSHAKE_COMPLETE) {
                clientShake = true;
            }
        }
    } while (!clientShake && !serverShake);


    /* Finally make sure the server validation does what
     * we were expecting
     */
    if (qcrypto_tls_session_check_credentials(serverSess, &err) < 0) {
        g_assert(data->expectServerFail);
        error_free(err);
        err = NULL;
    } else {
        g_assert(!data->expectServerFail);
    }

    /*
     * And the same for the client validation check
     */
    if (qcrypto_tls_session_check_credentials(clientSess, &err) < 0) {
        g_assert(data->expectClientFail);
        error_free(err);
        err = NULL;
    } else {
        g_assert(!data->expectClientFail);
    }

    unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT);
    unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_CERT);
    unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_KEY);

    unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT);
    unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_CERT);
    unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_KEY);

    rmdir(CLIENT_CERT_DIR);
    rmdir(SERVER_CERT_DIR);

    object_unparent(OBJECT(serverCreds));
    object_unparent(OBJECT(clientCreds));

    qcrypto_tls_session_free(serverSess);
    qcrypto_tls_session_free(clientSess);

    close(channel[0]);
    close(channel[1]);
}


int main(int argc, char **argv)
{
    int ret;

    module_call_init(MODULE_INIT_QOM);
    g_test_init(&argc, &argv, NULL);
    setenv("GNUTLS_FORCE_FIPS_MODE", "2", 1);

    mkdir(WORKDIR, 0700);

    test_tls_init(KEYFILE);

# define TEST_SESS_REG(name, caCrt,                                     \
                       serverCrt, clientCrt,                            \
                       expectServerFail, expectClientFail,              \
                       hostname, wildcards)                             \
    struct QCryptoTLSSessionTestData name = {                           \
        caCrt, caCrt, serverCrt, clientCrt,                             \
        expectServerFail, expectClientFail,                             \
        hostname, wildcards                                             \
    };                                                                  \
    g_test_add_data_func("/qcrypto/tlssession/" # name,                 \
                         &name, test_crypto_tls_session);               \


# define TEST_SESS_REG_EXT(name, serverCaCrt, clientCaCrt,              \
                           serverCrt, clientCrt,                        \
                           expectServerFail, expectClientFail,          \
                           hostname, wildcards)                         \
    struct QCryptoTLSSessionTestData name = {                           \
        serverCaCrt, clientCaCrt, serverCrt, clientCrt,                 \
        expectServerFail, expectClientFail,                             \
        hostname, wildcards                                             \
    };                                                                  \
    g_test_add_data_func("/qcrypto/tlssession/" # name,                 \
                         &name, test_crypto_tls_session);               \

    /* A perfect CA, perfect client & perfect server */

    /* Basic:CA:critical */
    TLS_ROOT_REQ(cacertreq,
                 "UK", "qemu CA", NULL, NULL, NULL, NULL,
                 true, true, true,
                 true, true, GNUTLS_KEY_KEY_CERT_SIGN,
                 false, false, NULL, NULL,
                 0, 0);

    TLS_ROOT_REQ(altcacertreq,
                 "UK", "qemu CA 1", NULL, NULL, NULL, NULL,
                 true, true, true,
                 false, false, 0,
                 false, false, NULL, NULL,
                 0, 0);

    TLS_CERT_REQ(servercertreq, cacertreq,
                 "UK", "qemu.org", NULL, NULL, NULL, NULL,
                 true, true, false,
                 true, true,
                 GNUTLS_KEY_DIGITAL_SIGNATURE | GNUTLS_KEY_KEY_ENCIPHERMENT,
                 true, true, GNUTLS_KP_TLS_WWW_SERVER, NULL,
                 0, 0);
    TLS_CERT_REQ(clientcertreq, cacertreq,
                 "UK", "qemu", NULL, NULL, NULL, NULL,
                 true, true, false,
                 true, true,
                 GNUTLS_KEY_DIGITAL_SIGNATURE | GNUTLS_KEY_KEY_ENCIPHERMENT,
                 true, true, GNUTLS_KP_TLS_WWW_CLIENT, NULL,
                 0, 0);

    TLS_CERT_REQ(clientcertaltreq, altcacertreq,
                 "UK", "qemu", NULL, NULL, NULL, NULL,
                 true, true, false,
                 true, true,
                 GNUTLS_KEY_DIGITAL_SIGNATURE | GNUTLS_KEY_KEY_ENCIPHERMENT,
                 true, true, GNUTLS_KP_TLS_WWW_CLIENT, NULL,
                 0, 0);

    TEST_SESS_REG(basicca, cacertreq.filename,
                  servercertreq.filename, clientcertreq.filename,
                  false, false, "qemu.org", NULL);
    TEST_SESS_REG_EXT(differentca, cacertreq.filename,
                      altcacertreq.filename, servercertreq.filename,
                      clientcertaltreq.filename, true, true, "qemu.org", NULL);


    /* When an altname is set, the CN is ignored, so it must be duplicated
     * as an altname for it to match */
    TLS_CERT_REQ(servercertalt1req, cacertreq,
                 "UK", "qemu.org", "www.qemu.org", "qemu.org",
                 "192.168.122.1", "fec0::dead:beaf",
                 true, true, false,
                 true, true,
                 GNUTLS_KEY_DIGITAL_SIGNATURE | GNUTLS_KEY_KEY_ENCIPHERMENT,
                 true, true, GNUTLS_KP_TLS_WWW_SERVER, NULL,
                 0, 0);
    /* This intentionally doesn't replicate */
    TLS_CERT_REQ(servercertalt2req, cacertreq,
                 "UK", "qemu.org", "www.qemu.org", "wiki.qemu.org",
                 "192.168.122.1", "fec0::dead:beaf",
                 true, true, false,
                 true, true,
                 GNUTLS_KEY_DIGITAL_SIGNATURE | GNUTLS_KEY_KEY_ENCIPHERMENT,
                 true, true, GNUTLS_KP_TLS_WWW_SERVER, NULL,
                 0, 0);

    TEST_SESS_REG(altname1, cacertreq.filename,
                  servercertalt1req.filename, clientcertreq.filename,
                  false, false, "qemu.org", NULL);
    TEST_SESS_REG(altname2, cacertreq.filename,
                  servercertalt1req.filename, clientcertreq.filename,
                  false, false, "www.qemu.org", NULL);
    TEST_SESS_REG(altname3, cacertreq.filename,
                  servercertalt1req.filename, clientcertreq.filename,
                  false, true, "wiki.qemu.org", NULL);

    TEST_SESS_REG(altname4, cacertreq.filename,
                  servercertalt2req.filename, clientcertreq.filename,
                  false, true, "qemu.org", NULL);
    TEST_SESS_REG(altname5, cacertreq.filename,
                  servercertalt2req.filename, clientcertreq.filename,
                  false, false, "www.qemu.org", NULL);
    TEST_SESS_REG(altname6, cacertreq.filename,
                  servercertalt2req.filename, clientcertreq.filename,
                  false, false, "wiki.qemu.org", NULL);

    const char *const wildcards1[] = {
        "C=UK,CN=dogfood",
        NULL,
    };
    const char *const wildcards2[] = {
        "C=UK,CN=qemu",
        NULL,
    };
    const char *const wildcards3[] = {
        "C=UK,CN=dogfood",
        "C=UK,CN=qemu",
        NULL,
    };
    const char *const wildcards4[] = {
        "C=UK,CN=qemustuff",
        NULL,
    };
    const char *const wildcards5[] = {
        "C=UK,CN=qemu*",
        NULL,
    };
    const char *const wildcards6[] = {
        "C=UK,CN=*emu*",
        NULL,
    };

    TEST_SESS_REG(wildcard1, cacertreq.filename,
                  servercertreq.filename, clientcertreq.filename,
                  true, false, "qemu.org", wildcards1);
    TEST_SESS_REG(wildcard2, cacertreq.filename,
                  servercertreq.filename, clientcertreq.filename,
                  false, false, "qemu.org", wildcards2);
    TEST_SESS_REG(wildcard3, cacertreq.filename,
                  servercertreq.filename, clientcertreq.filename,
                  false, false, "qemu.org", wildcards3);
    TEST_SESS_REG(wildcard4, cacertreq.filename,
                  servercertreq.filename, clientcertreq.filename,
                  true, false, "qemu.org", wildcards4);
    TEST_SESS_REG(wildcard5, cacertreq.filename,
                  servercertreq.filename, clientcertreq.filename,
                  false, false, "qemu.org", wildcards5);
    TEST_SESS_REG(wildcard6, cacertreq.filename,
                  servercertreq.filename, clientcertreq.filename,
                  false, false, "qemu.org", wildcards6);

    TLS_ROOT_REQ(cacertrootreq,
                 "UK", "qemu root", NULL, NULL, NULL, NULL,
                 true, true, true,
                 true, true, GNUTLS_KEY_KEY_CERT_SIGN,
                 false, false, NULL, NULL,
                 0, 0);
    TLS_CERT_REQ(cacertlevel1areq, cacertrootreq,
                 "UK", "qemu level 1a", NULL, NULL, NULL, NULL,
                 true, true, true,
                 true, true, GNUTLS_KEY_KEY_CERT_SIGN,
                 false, false, NULL, NULL,
                 0, 0);
    TLS_CERT_REQ(cacertlevel1breq, cacertrootreq,
                 "UK", "qemu level 1b", NULL, NULL, NULL, NULL,
                 true, true, true,
                 true, true, GNUTLS_KEY_KEY_CERT_SIGN,
                 false, false, NULL, NULL,
                 0, 0);
    TLS_CERT_REQ(cacertlevel2areq, cacertlevel1areq,
                 "UK", "qemu level 2a", NULL, NULL, NULL, NULL,
                 true, true, true,
                 true, true, GNUTLS_KEY_KEY_CERT_SIGN,
                 false, false, NULL, NULL,
                 0, 0);
    TLS_CERT_REQ(servercertlevel3areq, cacertlevel2areq,
                 "UK", "qemu.org", NULL, NULL, NULL, NULL,
                 true, true, false,
                 true, true,
                 GNUTLS_KEY_DIGITAL_SIGNATURE | GNUTLS_KEY_KEY_ENCIPHERMENT,
                 true, true, GNUTLS_KP_TLS_WWW_SERVER, NULL,
                 0, 0);
    TLS_CERT_REQ(clientcertlevel2breq, cacertlevel1breq,
                 "UK", "qemu client level 2b", NULL, NULL, NULL, NULL,
                 true, true, false,
                 true, true,
                 GNUTLS_KEY_DIGITAL_SIGNATURE | GNUTLS_KEY_KEY_ENCIPHERMENT,
                 true, true, GNUTLS_KP_TLS_WWW_CLIENT, NULL,
                 0, 0);

    gnutls_x509_crt_t certchain[] = {
        cacertrootreq.crt,
        cacertlevel1areq.crt,
        cacertlevel1breq.crt,
        cacertlevel2areq.crt,
    };

    test_tls_write_cert_chain(WORKDIR "cacertchain-sess.pem",
                              certchain,
                              G_N_ELEMENTS(certchain));

    TEST_SESS_REG(cachain, WORKDIR "cacertchain-sess.pem",
                  servercertlevel3areq.filename, clientcertlevel2breq.filename,
                  false, false, "qemu.org", NULL);

    ret = g_test_run();

    test_tls_discard_cert(&clientcertreq);
    test_tls_discard_cert(&clientcertaltreq);

    test_tls_discard_cert(&servercertreq);
    test_tls_discard_cert(&servercertalt1req);
    test_tls_discard_cert(&servercertalt2req);

    test_tls_discard_cert(&cacertreq);
    test_tls_discard_cert(&altcacertreq);

    test_tls_discard_cert(&cacertrootreq);
    test_tls_discard_cert(&cacertlevel1areq);
    test_tls_discard_cert(&cacertlevel1breq);
    test_tls_discard_cert(&cacertlevel2areq);
    test_tls_discard_cert(&servercertlevel3areq);
    test_tls_discard_cert(&clientcertlevel2breq);
    unlink(WORKDIR "cacertchain-sess.pem");

    test_tls_cleanup(KEYFILE);
    rmdir(WORKDIR);

    return ret == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
}

#else /* ! QCRYPTO_HAVE_TLS_TEST_SUPPORT */

int
main(void)
{
    return EXIT_SUCCESS;
}

#endif /* ! QCRYPTO_HAVE_TLS_TEST_SUPPORT */
Commit	Line	Data
d321e1e5 DB	1	/*
	2	* Copyright (C) 2015 Red Hat, Inc.
	3	*
	4	* This library is free software; you can redistribute it and/or
	5	* modify it under the terms of the GNU Lesser General Public
	6	* License as published by the Free Software Foundation; either
	7	* version 2.1 of the License, or (at your option) any later version.
	8	*
	9	* This library is distributed in the hope that it will be useful,
	10	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	11	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	12	* Lesser General Public License for more details.
	13	*
	14	* You should have received a copy of the GNU Lesser General Public
	15	* License along with this library. If not, see
	16	* <http://www.gnu.org/licenses/>.
	17	*
	18	* Author: Daniel P. Berrange <[email protected]>
	19	*/
	20
	21	#include <stdlib.h>
	22	#include <fcntl.h>
	23
	24	#include "config-host.h"
	25	#include "crypto-tls-x509-helpers.h"
	26	#include "crypto/tlscredsx509.h"
	27	#include "crypto/tlssession.h"
	28	#include "qom/object_interfaces.h"
	29	#include "qemu/sockets.h"
	30	#include "qemu/acl.h"
	31
	32	#ifdef QCRYPTO_HAVE_TLS_TEST_SUPPORT
	33
	34	#define WORKDIR "tests/test-crypto-tlssession-work/"
	35	#define KEYFILE WORKDIR "key-ctx.pem"
	36
	37	struct QCryptoTLSSessionTestData {
	38	const char *servercacrt;
	39	const char *clientcacrt;
	40	const char *servercrt;
	41	const char *clientcrt;
	42	bool expectServerFail;
	43	bool expectClientFail;
	44	const char *hostname;
	45	const char const wildcards;
	46	};
	47
	48
	49	static ssize_t testWrite(const char buf, size_t len, void opaque)
	50	{
	51	int *fd = opaque;
	52
	53	return write(*fd, buf, len);
	54	}
	55
	56	static ssize_t testRead(char buf, size_t len, void opaque)
	57	{
	58	int *fd = opaque;
	59
	60	return read(*fd, buf, len);
	61	}
	62
	63	static QCryptoTLSCreds *test_tls_creds_create(QCryptoTLSCredsEndpoint endpoint,
	64	const char *certdir,
65	Error **errp)
66	{
67	Error *err = NULL;
68	Object *parent = object_get_objects_root();
69	Object *creds = object_new_with_props(
70	TYPE_QCRYPTO_TLS_CREDS_X509,
71	parent,
72	(endpoint == QCRYPTO_TLS_CREDS_ENDPOINT_SERVER ?
73	"testtlscredsserver" : "testtlscredsclient"),
74	&err,
75	"endpoint", (endpoint == QCRYPTO_TLS_CREDS_ENDPOINT_SERVER ?
76	"server" : "client"),
77	"dir", certdir,
78	"verify-peer", "yes",
79	/* We skip initial sanity checks here because we
80	* want to make sure that problems are being
81	* detected at the TLS session validation stage,
82	* and the test-crypto-tlscreds test already
83	* validate the sanity check code.
84	*/
85	"sanity-check", "no",
86	NULL
87	);
88
89	if (err) {
90	error_propagate(errp, err);
91	return NULL;
92	}
93	return QCRYPTO_TLS_CREDS(creds);
94	}
95
96
97	/*
98	* This tests validation checking of peer certificates
99	*
100	* This is replicating the checks that are done for an
101	* active TLS session after handshake completes. To
102	* simulate that we create our TLS contexts, skipping
103	* sanity checks. We then get a socketpair, and
104	* initiate a TLS session across them. Finally do
105	* do actual cert validation tests
106	*/
107	static void test_crypto_tls_session(const void *opaque)
108	{
109	struct QCryptoTLSSessionTestData *data =
110	(struct QCryptoTLSSessionTestData *)opaque;
111	QCryptoTLSCreds *clientCreds;
112	QCryptoTLSCreds *serverCreds;
113	QCryptoTLSSession *clientSess = NULL;
114	QCryptoTLSSession *serverSess = NULL;
115	qemu_acl *acl;
116	const char * const *wildcards;
117	int channel[2];
118	bool clientShake = false;
119	bool serverShake = false;
120	Error *err = NULL;
121	int ret;
122
123	/* We'll use this for our fake client-server connection */
124	ret = socketpair(AF_UNIX, SOCK_STREAM, 0, channel);
125	g_assert(ret == 0);
126
127	/*
128	* We have an evil loop to do the handshake in a single
129	* thread, so we need these non-blocking to avoid deadlock
130	* of ourselves
131	*/
132	qemu_set_nonblock(channel[0]);
133	qemu_set_nonblock(channel[1]);
134
135	#define CLIENT_CERT_DIR "tests/test-crypto-tlssession-client/"
136	#define SERVER_CERT_DIR "tests/test-crypto-tlssession-server/"
137	mkdir(CLIENT_CERT_DIR, 0700);
138	mkdir(SERVER_CERT_DIR, 0700);
139
140	unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT);
141	unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_CERT);
142	unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_KEY);
143
144	unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT);
145	unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_CERT);
146	unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_KEY);
147
148	g_assert(link(data->servercacrt,
149	SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT) == 0);
150	g_assert(link(data->servercrt,
151	SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_CERT) == 0);
152	g_assert(link(KEYFILE,
153	SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_KEY) == 0);
154
155	g_assert(link(data->clientcacrt,
156	CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT) == 0);
157	g_assert(link(data->clientcrt,
158	CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_CERT) == 0);
159	g_assert(link(KEYFILE,
160	CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_KEY) == 0);
161
162	clientCreds = test_tls_creds_create(
163	QCRYPTO_TLS_CREDS_ENDPOINT_CLIENT,
164	CLIENT_CERT_DIR,
165	&err);
166	g_assert(clientCreds != NULL);
167
168	serverCreds = test_tls_creds_create(
169	QCRYPTO_TLS_CREDS_ENDPOINT_SERVER,
170	SERVER_CERT_DIR,
171	&err);
172	g_assert(serverCreds != NULL);
173
174	acl = qemu_acl_init("tlssessionacl");
175	qemu_acl_reset(acl);
176	wildcards = data->wildcards;
177	while (wildcards && *wildcards) {
178	qemu_acl_append(acl, 0, *wildcards);
179	wildcards++;
180	}
181
182	/* Now the real part of the test, setup the sessions */
183	clientSess = qcrypto_tls_session_new(
184	clientCreds, data->hostname, NULL,
185	QCRYPTO_TLS_CREDS_ENDPOINT_CLIENT, &err);
186	serverSess = qcrypto_tls_session_new(
187	serverCreds, NULL,
188	data->wildcards ? "tlssessionacl" : NULL,
189	QCRYPTO_TLS_CREDS_ENDPOINT_SERVER, &err);
190
191	g_assert(clientSess != NULL);
192	g_assert(serverSess != NULL);
193
194	/* For handshake to work, we need to set the I/O callbacks
195	* to read/write over the socketpair
196	*/
197	qcrypto_tls_session_set_callbacks(serverSess,
198	testWrite, testRead,
199	&channel[0]);
200	qcrypto_tls_session_set_callbacks(clientSess,
201	testWrite, testRead,
202	&channel[1]);
203
204	/*
205	* Finally we loop around & around doing handshake on each
206	* session until we get an error, or the handshake completes.
207	* This relies on the socketpair being nonblocking to avoid
208	* deadlocking ourselves upon handshake
209	*/
210	do {
211	int rv;
212	if (!serverShake) {
213	rv = qcrypto_tls_session_handshake(serverSess,
214	&err);
215	g_assert(rv >= 0);
216	if (qcrypto_tls_session_get_handshake_status(serverSess) ==
217	QCRYPTO_TLS_HANDSHAKE_COMPLETE) {
218	serverShake = true;
219	}
220	}
221	if (!clientShake) {
222	rv = qcrypto_tls_session_handshake(clientSess,
223	&err);
224	g_assert(rv >= 0);
225	if (qcrypto_tls_session_get_handshake_status(clientSess) ==
226	QCRYPTO_TLS_HANDSHAKE_COMPLETE) {
227	clientShake = true;
228	}
229	}
230	} while (!clientShake && !serverShake);
231
232
233	/* Finally make sure the server validation does what
234	* we were expecting
235	*/
236	if (qcrypto_tls_session_check_credentials(serverSess, &err) < 0) {
237	g_assert(data->expectServerFail);
238	error_free(err);
239	err = NULL;
240	} else {
241	g_assert(!data->expectServerFail);
242	}
243
244	/*
245	* And the same for the client validation check
246	*/
247	if (qcrypto_tls_session_check_credentials(clientSess, &err) < 0) {
248	g_assert(data->expectClientFail);
249	error_free(err);
250	err = NULL;
251	} else {
252	g_assert(!data->expectClientFail);
253	}
254
255	unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT);
256	unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_CERT);
257	unlink(SERVER_CERT_DIR QCRYPTO_TLS_CREDS_X509_SERVER_KEY);
258
259	unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CA_CERT);
260	unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_CERT);
261	unlink(CLIENT_CERT_DIR QCRYPTO_TLS_CREDS_X509_CLIENT_KEY);
262
263	rmdir(CLIENT_CERT_DIR);
264	rmdir(SERVER_CERT_DIR);
265
266	object_unparent(OBJECT(serverCreds));
267	object_unparent(OBJECT(clientCreds));
268
269	qcrypto_tls_session_free(serverSess);
270	qcrypto_tls_session_free(clientSess);
271
272	close(channel[0]);
273	close(channel[1]);
274	}
275
276
277	int main(int argc, char **argv)
278	{
279	int ret;
280
281	module_call_init(MODULE_INIT_QOM);
282	g_test_init(&argc, &argv, NULL);
283	setenv("GNUTLS_FORCE_FIPS_MODE", "2", 1);
284
285	mkdir(WORKDIR, 0700);
286
287	test_tls_init(KEYFILE);
288
289	# define TEST_SESS_REG(name, caCrt, \
290	serverCrt, clientCrt, \
291	expectServerFail, expectClientFail, \
292	hostname, wildcards) \
293	struct QCryptoTLSSessionTestData name = { \
294	caCrt, caCrt, serverCrt, clientCrt, \
295	expectServerFail, expectClientFail, \
296	hostname, wildcards \
297	}; \
298	g_test_add_data_func("/qcrypto/tlssession/" # name, \
299	&name, test_crypto_tls_session); \
300
301
302	# define TEST_SESS_REG_EXT(name, serverCaCrt, clientCaCrt, \
303	serverCrt, clientCrt, \
304	expectServerFail, expectClientFail, \
305	hostname, wildcards) \
306	struct QCryptoTLSSessionTestData name = { \
307	serverCaCrt, clientCaCrt, serverCrt, clientCrt, \
308	expectServerFail, expectClientFail, \
309	hostname, wildcards \
310	}; \
311	g_test_add_data_func("/qcrypto/tlssession/" # name, \
312	&name, test_crypto_tls_session); \
313
314	/* A perfect CA, perfect client & perfect server */
315
316	/* Basic:CA:critical */
317	TLS_ROOT_REQ(cacertreq,
318	"UK", "qemu CA", NULL, NULL, NULL, NULL,
319	true, true, true,
320	true, true, GNUTLS_KEY_KEY_CERT_SIGN,
321	false, false, NULL, NULL,
322	0, 0);
323
324	TLS_ROOT_REQ(altcacertreq,
325	"UK", "qemu CA 1", NULL, NULL, NULL, NULL,
326	true, true, true,
327	false, false, 0,
328	false, false, NULL, NULL,
329	0, 0);
330
331	TLS_CERT_REQ(servercertreq, cacertreq,
332	"UK", "qemu.org", NULL, NULL, NULL, NULL,
333	true, true, false,
334	true, true,
335	GNUTLS_KEY_DIGITAL_SIGNATURE \| GNUTLS_KEY_KEY_ENCIPHERMENT,
336	true, true, GNUTLS_KP_TLS_WWW_SERVER, NULL,
337	0, 0);
338	TLS_CERT_REQ(clientcertreq, cacertreq,
339	"UK", "qemu", NULL, NULL, NULL, NULL,
340	true, true, false,
341	true, true,
342	GNUTLS_KEY_DIGITAL_SIGNATURE \| GNUTLS_KEY_KEY_ENCIPHERMENT,
343	true, true, GNUTLS_KP_TLS_WWW_CLIENT, NULL,
344	0, 0);
345
346	TLS_CERT_REQ(clientcertaltreq, altcacertreq,
347	"UK", "qemu", NULL, NULL, NULL, NULL,
348	true, true, false,
349	true, true,
350	GNUTLS_KEY_DIGITAL_SIGNATURE \| GNUTLS_KEY_KEY_ENCIPHERMENT,
351	true, true, GNUTLS_KP_TLS_WWW_CLIENT, NULL,
352	0, 0);
353
354	TEST_SESS_REG(basicca, cacertreq.filename,
355	servercertreq.filename, clientcertreq.filename,
356	false, false, "qemu.org", NULL);
357	TEST_SESS_REG_EXT(differentca, cacertreq.filename,
358	altcacertreq.filename, servercertreq.filename,
359	clientcertaltreq.filename, true, true, "qemu.org", NULL);
360
361
362	/* When an altname is set, the CN is ignored, so it must be duplicated
363	* as an altname for it to match */
364	TLS_CERT_REQ(servercertalt1req, cacertreq,
365	"UK", "qemu.org", "www.qemu.org", "qemu.org",
366	"192.168.122.1", "fec0::dead:beaf",
367	true, true, false,
368	true, true,
369	GNUTLS_KEY_DIGITAL_SIGNATURE \| GNUTLS_KEY_KEY_ENCIPHERMENT,
370	true, true, GNUTLS_KP_TLS_WWW_SERVER, NULL,
371	0, 0);
372	/* This intentionally doesn't replicate */
373	TLS_CERT_REQ(servercertalt2req, cacertreq,
374	"UK", "qemu.org", "www.qemu.org", "wiki.qemu.org",
375	"192.168.122.1", "fec0::dead:beaf",
376	true, true, false,
377	true, true,
378	GNUTLS_KEY_DIGITAL_SIGNATURE \| GNUTLS_KEY_KEY_ENCIPHERMENT,
379	true, true, GNUTLS_KP_TLS_WWW_SERVER, NULL,
380	0, 0);
381
382	TEST_SESS_REG(altname1, cacertreq.filename,
383	servercertalt1req.filename, clientcertreq.filename,
384	false, false, "qemu.org", NULL);
385	TEST_SESS_REG(altname2, cacertreq.filename,
386	servercertalt1req.filename, clientcertreq.filename,
387	false, false, "www.qemu.org", NULL);
388	TEST_SESS_REG(altname3, cacertreq.filename,
389	servercertalt1req.filename, clientcertreq.filename,
390	false, true, "wiki.qemu.org", NULL);
391
392	TEST_SESS_REG(altname4, cacertreq.filename,
393	servercertalt2req.filename, clientcertreq.filename,
394	false, true, "qemu.org", NULL);
395	TEST_SESS_REG(altname5, cacertreq.filename,
396	servercertalt2req.filename, clientcertreq.filename,
397	false, false, "www.qemu.org", NULL);
398	TEST_SESS_REG(altname6, cacertreq.filename,
399	servercertalt2req.filename, clientcertreq.filename,
400	false, false, "wiki.qemu.org", NULL);
401
402	const char *const wildcards1[] = {
403	"C=UK,CN=dogfood",
404	NULL,
405	};
406	const char *const wildcards2[] = {
407	"C=UK,CN=qemu",
408	NULL,
409	};
410	const char *const wildcards3[] = {
411	"C=UK,CN=dogfood",
412	"C=UK,CN=qemu",
413	NULL,
414	};
415	const char *const wildcards4[] = {
416	"C=UK,CN=qemustuff",
417	NULL,
418	};
419	const char *const wildcards5[] = {
420	"C=UK,CN=qemu*",
421	NULL,
422	};
423	const char *const wildcards6[] = {
424	"C=UK,CN=emu",
425	NULL,
426	};
427
428	TEST_SESS_REG(wildcard1, cacertreq.filename,
429	servercertreq.filename, clientcertreq.filename,
430	true, false, "qemu.org", wildcards1);
431	TEST_SESS_REG(wildcard2, cacertreq.filename,
432	servercertreq.filename, clientcertreq.filename,
433	false, false, "qemu.org", wildcards2);
434	TEST_SESS_REG(wildcard3, cacertreq.filename,
435	servercertreq.filename, clientcertreq.filename,
436	false, false, "qemu.org", wildcards3);
437	TEST_SESS_REG(wildcard4, cacertreq.filename,
438	servercertreq.filename, clientcertreq.filename,
439	true, false, "qemu.org", wildcards4);
440	TEST_SESS_REG(wildcard5, cacertreq.filename,
441	servercertreq.filename, clientcertreq.filename,
442	false, false, "qemu.org", wildcards5);
443	TEST_SESS_REG(wildcard6, cacertreq.filename,
444	servercertreq.filename, clientcertreq.filename,
445	false, false, "qemu.org", wildcards6);
446
447	TLS_ROOT_REQ(cacertrootreq,
448	"UK", "qemu root", NULL, NULL, NULL, NULL,
449	true, true, true,
450	true, true, GNUTLS_KEY_KEY_CERT_SIGN,
451	false, false, NULL, NULL,
452	0, 0);
453	TLS_CERT_REQ(cacertlevel1areq, cacertrootreq,
454	"UK", "qemu level 1a", NULL, NULL, NULL, NULL,
455	true, true, true,
456	true, true, GNUTLS_KEY_KEY_CERT_SIGN,
457	false, false, NULL, NULL,
458	0, 0);
459	TLS_CERT_REQ(cacertlevel1breq, cacertrootreq,
460	"UK", "qemu level 1b", NULL, NULL, NULL, NULL,
461	true, true, true,
462	true, true, GNUTLS_KEY_KEY_CERT_SIGN,
463	false, false, NULL, NULL,
464	0, 0);
465	TLS_CERT_REQ(cacertlevel2areq, cacertlevel1areq,
466	"UK", "qemu level 2a", NULL, NULL, NULL, NULL,
467	true, true, true,
468	true, true, GNUTLS_KEY_KEY_CERT_SIGN,
469	false, false, NULL, NULL,
470	0, 0);
471	TLS_CERT_REQ(servercertlevel3areq, cacertlevel2areq,
472	"UK", "qemu.org", NULL, NULL, NULL, NULL,
473	true, true, false,
474	true, true,
475	GNUTLS_KEY_DIGITAL_SIGNATURE \| GNUTLS_KEY_KEY_ENCIPHERMENT,
476	true, true, GNUTLS_KP_TLS_WWW_SERVER, NULL,
477	0, 0);
478	TLS_CERT_REQ(clientcertlevel2breq, cacertlevel1breq,
479	"UK", "qemu client level 2b", NULL, NULL, NULL, NULL,
480	true, true, false,
481	true, true,
482	GNUTLS_KEY_DIGITAL_SIGNATURE \| GNUTLS_KEY_KEY_ENCIPHERMENT,
483	true, true, GNUTLS_KP_TLS_WWW_CLIENT, NULL,
484	0, 0);
485
486	gnutls_x509_crt_t certchain[] = {
487	cacertrootreq.crt,
488	cacertlevel1areq.crt,
489	cacertlevel1breq.crt,
490	cacertlevel2areq.crt,
491	};
492
493	test_tls_write_cert_chain(WORKDIR "cacertchain-sess.pem",
494	certchain,
495	G_N_ELEMENTS(certchain));
496
497	TEST_SESS_REG(cachain, WORKDIR "cacertchain-sess.pem",
498	servercertlevel3areq.filename, clientcertlevel2breq.filename,
499	false, false, "qemu.org", NULL);
500
501	ret = g_test_run();
502
503	test_tls_discard_cert(&clientcertreq);
504	test_tls_discard_cert(&clientcertaltreq);
505
506	test_tls_discard_cert(&servercertreq);
507	test_tls_discard_cert(&servercertalt1req);
508	test_tls_discard_cert(&servercertalt2req);
509
510	test_tls_discard_cert(&cacertreq);
511	test_tls_discard_cert(&altcacertreq);
512
513	test_tls_discard_cert(&cacertrootreq);
514	test_tls_discard_cert(&cacertlevel1areq);
515	test_tls_discard_cert(&cacertlevel1breq);
516	test_tls_discard_cert(&cacertlevel2areq);
517	test_tls_discard_cert(&servercertlevel3areq);
518	test_tls_discard_cert(&clientcertlevel2breq);
519	unlink(WORKDIR "cacertchain-sess.pem");
520
521	test_tls_cleanup(KEYFILE);
522	rmdir(WORKDIR);
523
524	return ret == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
525	}
526
527	#else /* ! QCRYPTO_HAVE_TLS_TEST_SUPPORT */
528
529	int
530	main(void)
531	{
532	return EXIT_SUCCESS;
533	}
534
535	#endif /* ! QCRYPTO_HAVE_TLS_TEST_SUPPORT */