/*

 * Copyright 2016-2020 The OpenSSL Project Authors. All Rights Reserved.

 *

 * Licensed under the OpenSSL license (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 <string.h>

#include <openssl/bio.h>

#include <openssl/crypto.h>

#include <openssl/ssl.h>

#include <openssl/err.h>

#include "ssltestlib.h"

#include "testutil.h"

static char *cert = NULL;

static char *privkey = NULL;

static unsigned int timer_cb_count;

#define NUM_TESTS   2

#define DUMMY_CERT_STATUS_LEN  12

static unsigned char certstatus[] = {

    SSL3_RT_HANDSHAKE, /* Content type */

    0xfe, 0xfd, /* Record version */

    0, 1, /* Epoch */

    0, 0, 0, 0, 0, 0x0f, /* Record sequence number */

    0, DTLS1_HM_HEADER_LENGTH + DUMMY_CERT_STATUS_LEN - 2,

    SSL3_MT_CERTIFICATE_STATUS, /* Cert Status handshake message type */

    0, 0, DUMMY_CERT_STATUS_LEN, /* Message len */

    0, 5, /* Message sequence */

    0, 0, 0, /* Fragment offset */

    0, 0, DUMMY_CERT_STATUS_LEN - 2, /* Fragment len */

    0x80, 0x80, 0x80, 0x80, 0x80,

    0x80, 0x80, 0x80, 0x80, 0x80 /* Dummy data */

};

#define RECORD_SEQUENCE 10

static unsigned int timer_cb(SSL *s, unsigned int timer_us)

{

    ++timer_cb_count;

    if (timer_us == 0)

        return 50000;

    else

        return 2 * timer_us;

}

static int test_dtls_unprocessed(int testidx)

{

    SSL_CTX *sctx = NULL, *cctx = NULL;

    SSL *serverssl1 = NULL, *clientssl1 = NULL;

    BIO *c_to_s_fbio, *c_to_s_mempacket;

    int testresult = 0;

    timer_cb_count = 0;

    if (!TEST_true(create_ssl_ctx_pair(DTLS_server_method(),

                                       DTLS_client_method(),

                                       DTLS1_VERSION, DTLS_MAX_VERSION,

                                       &sctx, &cctx, cert, privkey)))

        return 0;

    if (!TEST_true(SSL_CTX_set_cipher_list(cctx, "AES128-SHA")))

        goto end;

    c_to_s_fbio = BIO_new(bio_f_tls_dump_filter());

    if (!TEST_ptr(c_to_s_fbio))

        goto end;

    /* BIO is freed by create_ssl_connection on error */

    if (!TEST_true(create_ssl_objects(sctx, cctx, &serverssl1, &clientssl1,

                                      NULL, c_to_s_fbio)))

        goto end;

    DTLS_set_timer_cb(clientssl1, timer_cb);

    if (testidx == 1)

        certstatus[RECORD_SEQUENCE] = 0xff;

    /*

     * Inject a dummy record from the next epoch. In test 0, this should never

     * get used because the message sequence number is too big. In test 1 we set

     * the record sequence number to be way off in the future.

     */

    c_to_s_mempacket = SSL_get_wbio(clientssl1);

    c_to_s_mempacket = BIO_next(c_to_s_mempacket);

    mempacket_test_inject(c_to_s_mempacket, (char *)certstatus,

                          sizeof(certstatus), 1, INJECT_PACKET_IGNORE_REC_SEQ);

    /*

     * Create the connection. We use "create_bare_ssl_connection" here so that

     * we can force the connection to not do "SSL_read" once partly connected.

     * We don't want to accidentally read the dummy records we injected because

     * they will fail to decrypt.

     */

    if (!TEST_true(create_bare_ssl_connection(serverssl1, clientssl1,

                                              SSL_ERROR_NONE, 0)))

        goto end;

    if (timer_cb_count == 0) {

        printf("timer_callback was not called.\n");

        goto end;

    }

    testresult = 1;

 end:

    SSL_free(serverssl1);

    SSL_free(clientssl1);

    SSL_CTX_free(sctx);

    SSL_CTX_free(cctx);

    return testresult;

}

#define CLI_TO_SRV_EPOCH_0_RECS 3

#define CLI_TO_SRV_EPOCH_1_RECS 1

#if !defined(OPENSSL_NO_EC) || !defined(OPENSSL_NO_DH)

# define SRV_TO_CLI_EPOCH_0_RECS 10

#else

/*

 * In this case we have no ServerKeyExchange message, because we don't have

 * ECDHE or DHE. When it is present it gets fragmented into 3 records in this

 * test.

 */

# define SRV_TO_CLI_EPOCH_0_RECS 9

#endif

#define SRV_TO_CLI_EPOCH_1_RECS 1

#define TOTAL_FULL_HAND_RECORDS \

            (CLI_TO_SRV_EPOCH_0_RECS + CLI_TO_SRV_EPOCH_1_RECS + \

             SRV_TO_CLI_EPOCH_0_RECS + SRV_TO_CLI_EPOCH_1_RECS)

#define CLI_TO_SRV_RESUME_EPOCH_0_RECS 3

#define CLI_TO_SRV_RESUME_EPOCH_1_RECS 1

#define SRV_TO_CLI_RESUME_EPOCH_0_RECS 2

#define SRV_TO_CLI_RESUME_EPOCH_1_RECS 1

#define TOTAL_RESUME_HAND_RECORDS \

            (CLI_TO_SRV_RESUME_EPOCH_0_RECS + CLI_TO_SRV_RESUME_EPOCH_1_RECS + \

             SRV_TO_CLI_RESUME_EPOCH_0_RECS + SRV_TO_CLI_RESUME_EPOCH_1_RECS)

#define TOTAL_RECORDS (TOTAL_FULL_HAND_RECORDS + TOTAL_RESUME_HAND_RECORDS)

static int test_dtls_drop_records(int idx)

{

    SSL_CTX *sctx = NULL, *cctx = NULL;

    SSL *serverssl = NULL, *clientssl = NULL;

    BIO *c_to_s_fbio, *mempackbio;

    int testresult = 0;

    int epoch = 0;

    SSL_SESSION *sess = NULL;

    int cli_to_srv_epoch0, cli_to_srv_epoch1, srv_to_cli_epoch0;

    if (!TEST_true(create_ssl_ctx_pair(DTLS_server_method(),

                                       DTLS_client_method(),

                                       DTLS1_VERSION, DTLS_MAX_VERSION,

                                       &sctx, &cctx, cert, privkey)))

        return 0;

    if (idx >= TOTAL_FULL_HAND_RECORDS) {

        /* We're going to do a resumption handshake. Get a session first. */

        if (!TEST_true(create_ssl_objects(sctx, cctx, &serverssl, &clientssl,

                                          NULL, NULL))

                || !TEST_true(create_ssl_connection(serverssl, clientssl,

                              SSL_ERROR_NONE))

                || !TEST_ptr(sess = SSL_get1_session(clientssl)))

            goto end;

        SSL_shutdown(clientssl);

        SSL_shutdown(serverssl);

        SSL_free(serverssl);

        SSL_free(clientssl);

        serverssl = clientssl = NULL;

        cli_to_srv_epoch0 = CLI_TO_SRV_RESUME_EPOCH_0_RECS;

        cli_to_srv_epoch1 = CLI_TO_SRV_RESUME_EPOCH_1_RECS;

        srv_to_cli_epoch0 = SRV_TO_CLI_RESUME_EPOCH_0_RECS;

        idx -= TOTAL_FULL_HAND_RECORDS;

    } else {

        cli_to_srv_epoch0 = CLI_TO_SRV_EPOCH_0_RECS;

        cli_to_srv_epoch1 = CLI_TO_SRV_EPOCH_1_RECS;

        srv_to_cli_epoch0 = SRV_TO_CLI_EPOCH_0_RECS;

    }

    c_to_s_fbio = BIO_new(bio_f_tls_dump_filter());

    if (!TEST_ptr(c_to_s_fbio))

        goto end;

    /* BIO is freed by create_ssl_connection on error */

    if (!TEST_true(create_ssl_objects(sctx, cctx, &serverssl, &clientssl,

                                      NULL, c_to_s_fbio)))

        goto end;

    if (sess != NULL) {

        if (!TEST_true(SSL_set_session(clientssl, sess)))

            goto end;

    }

    DTLS_set_timer_cb(clientssl, timer_cb);

    DTLS_set_timer_cb(serverssl, timer_cb);

    /* Work out which record to drop based on the test number */

    if (idx >= cli_to_srv_epoch0 + cli_to_srv_epoch1) {

        mempackbio = SSL_get_wbio(serverssl);

        idx -= cli_to_srv_epoch0 + cli_to_srv_epoch1;

        if (idx >= srv_to_cli_epoch0) {

            epoch = 1;

            idx -= srv_to_cli_epoch0;

        }

    } else {

        mempackbio = SSL_get_wbio(clientssl);

        if (idx >= cli_to_srv_epoch0) {

            epoch = 1;

            idx -= cli_to_srv_epoch0;

        }

         mempackbio = BIO_next(mempackbio);

    }

    BIO_ctrl(mempackbio, MEMPACKET_CTRL_SET_DROP_EPOCH, epoch, NULL);

    BIO_ctrl(mempackbio, MEMPACKET_CTRL_SET_DROP_REC, idx, NULL);

    if (!TEST_true(create_ssl_connection(serverssl, clientssl, SSL_ERROR_NONE)))

        goto end;

    if (sess != NULL && !TEST_true(SSL_session_reused(clientssl)))

        goto end;

    /* If the test did what we planned then it should have dropped a record */

    if (!TEST_int_eq((int)BIO_ctrl(mempackbio, MEMPACKET_CTRL_GET_DROP_REC, 0,

                                   NULL), -1))

        goto end;

    testresult = 1;

 end:

    SSL_SESSION_free(sess);

    SSL_free(serverssl);

    SSL_free(clientssl);

    SSL_CTX_free(sctx);

    SSL_CTX_free(cctx);

    return testresult;

}

static const char dummy_cookie[] = "0123456";

static int generate_cookie_cb(SSL *ssl, unsigned char *cookie,

                              unsigned int *cookie_len)

{

    memcpy(cookie, dummy_cookie, sizeof(dummy_cookie));

    *cookie_len = sizeof(dummy_cookie);

    return 1;

}

static int verify_cookie_cb(SSL *ssl, const unsigned char *cookie,

                            unsigned int cookie_len)

{

    return TEST_mem_eq(cookie, cookie_len, dummy_cookie, sizeof(dummy_cookie));

}

static int test_cookie(void)

{

    SSL_CTX *sctx = NULL, *cctx = NULL;

    SSL *serverssl = NULL, *clientssl = NULL;

    int testresult = 0;

    if (!TEST_true(create_ssl_ctx_pair(DTLS_server_method(),

                                       DTLS_client_method(),

                                       DTLS1_VERSION, DTLS_MAX_VERSION,

                                       &sctx, &cctx, cert, privkey)))

        return 0;

    SSL_CTX_set_options(sctx, SSL_OP_COOKIE_EXCHANGE);

    SSL_CTX_set_cookie_generate_cb(sctx, generate_cookie_cb);

    SSL_CTX_set_cookie_verify_cb(sctx, verify_cookie_cb);

    if (!TEST_true(create_ssl_objects(sctx, cctx, &serverssl, &clientssl,

                                      NULL, NULL))

            || !TEST_true(create_ssl_connection(serverssl, clientssl,

                                                SSL_ERROR_NONE)))

        goto end;

    testresult = 1;

 end:

    SSL_free(serverssl);

    SSL_free(clientssl);

    SSL_CTX_free(sctx);

    SSL_CTX_free(cctx);

    return testresult;

}

static int test_dtls_duplicate_records(void)

{

    SSL_CTX *sctx = NULL, *cctx = NULL;

    SSL *serverssl = NULL, *clientssl = NULL;

    int testresult = 0;

    if (!TEST_true(create_ssl_ctx_pair(DTLS_server_method(),

                                       DTLS_client_method(),

                                       DTLS1_VERSION, DTLS_MAX_VERSION,

                                       &sctx, &cctx, cert, privkey)))

        return 0;

    if (!TEST_true(create_ssl_objects(sctx, cctx, &serverssl, &clientssl,

                                      NULL, NULL)))

        goto end;

    DTLS_set_timer_cb(clientssl, timer_cb);

    DTLS_set_timer_cb(serverssl, timer_cb);

    BIO_ctrl(SSL_get_wbio(clientssl), MEMPACKET_CTRL_SET_DUPLICATE_REC, 1, NULL);

    BIO_ctrl(SSL_get_wbio(serverssl), MEMPACKET_CTRL_SET_DUPLICATE_REC, 1, NULL);

    if (!TEST_true(create_ssl_connection(serverssl, clientssl, SSL_ERROR_NONE)))

        goto end;

    testresult = 1;

 end:

    SSL_free(serverssl);

    SSL_free(clientssl);

    SSL_CTX_free(sctx);

    SSL_CTX_free(cctx);

    return testresult;

}

int setup_tests(void)

{

    if (!TEST_ptr(cert = test_get_argument(0))

            || !TEST_ptr(privkey = test_get_argument(1)))

        return 0;

    ADD_ALL_TESTS(test_dtls_unprocessed, NUM_TESTS);

    ADD_ALL_TESTS(test_dtls_drop_records, TOTAL_RECORDS);

    ADD_TEST(test_cookie);

    ADD_TEST(test_dtls_duplicate_records);

    return 1;

}

void cleanup_tests(void)

{

    bio_f_tls_dump_filter_free();

    bio_s_mempacket_test_free();

}