mirror of
https://github.com/openssl/openssl.git
synced 2024-11-27 05:21:51 +08:00
ad887416f1
that needed test_main now works using the same infrastructure as tests that used
register_tests.
This meant:
* renaming register_tests to setup_tests and giving it a success/failure return.
* renaming the init_test function to setup_test_framework.
* renaming the finish_test function to pulldown_test_framework.
* adding a user provided global_init function that runs before the test frame
work is initialised. It returns a failure indication that stops the stest.
* adding helper functions that permit tests to access their command line args.
* spliting the BIO initialisation and finalisation out from the test setup and
teardown.
* hiding some of the now test internal functions.
* fix the comments in testutil.h
Reviewed-by: Richard Levitte <levitte@openssl.org>
(Merged from https://github.com/openssl/openssl/pull/3953)
358 lines
12 KiB
C
358 lines
12 KiB
C
/*
|
|
* Copyright 2016-2017 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/ssl.h>
|
|
#include <openssl/bio.h>
|
|
#include <openssl/err.h>
|
|
#include <openssl/conf.h>
|
|
#include "e_os.h"
|
|
#include "testutil.h"
|
|
|
|
#ifndef OPENSSL_NO_SOCK
|
|
|
|
/* Just a ClientHello without a cookie */
|
|
static const unsigned char clienthello_nocookie[] = {
|
|
0x16, /* Handshake */
|
|
0xFE, 0xFF, /* DTLSv1.0 */
|
|
0x00, 0x00, /* Epoch */
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* Record sequence number */
|
|
0x00, 0x3A, /* Record Length */
|
|
0x01, /* ClientHello */
|
|
0x00, 0x00, 0x2E, /* Message length */
|
|
0x00, 0x00, /* Message sequence */
|
|
0x00, 0x00, 0x00, /* Fragment offset */
|
|
0x00, 0x00, 0x2E, /* Fragment length */
|
|
0xFE, 0xFD, /* DTLSv1.2 */
|
|
0xCA, 0x18, 0x9F, 0x76, 0xEC, 0x57, 0xCE, 0xE5, 0xB3, 0xAB, 0x79, 0x90,
|
|
0xAD, 0xAC, 0x6E, 0xD1, 0x58, 0x35, 0x03, 0x97, 0x16, 0x10, 0x82, 0x56,
|
|
0xD8, 0x55, 0xFF, 0xE1, 0x8A, 0xA3, 0x2E, 0xF6, /* Random */
|
|
0x00, /* Session id len */
|
|
0x00, /* Cookie len */
|
|
0x00, 0x04, /* Ciphersuites len */
|
|
0x00, 0x2f, /* AES128-SHA */
|
|
0x00, 0xff, /* Empty reneg info SCSV */
|
|
0x01, /* Compression methods len */
|
|
0x00, /* Null compression */
|
|
0x00, 0x00 /* Extensions len */
|
|
};
|
|
|
|
/* First fragment of a ClientHello without a cookie */
|
|
static const unsigned char clienthello_nocookie_frag[] = {
|
|
0x16, /* Handshake */
|
|
0xFE, 0xFF, /* DTLSv1.0 */
|
|
0x00, 0x00, /* Epoch */
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* Record sequence number */
|
|
0x00, 0x30, /* Record Length */
|
|
0x01, /* ClientHello */
|
|
0x00, 0x00, 0x2E, /* Message length */
|
|
0x00, 0x00, /* Message sequence */
|
|
0x00, 0x00, 0x00, /* Fragment offset */
|
|
0x00, 0x00, 0x24, /* Fragment length */
|
|
0xFE, 0xFD, /* DTLSv1.2 */
|
|
0xCA, 0x18, 0x9F, 0x76, 0xEC, 0x57, 0xCE, 0xE5, 0xB3, 0xAB, 0x79, 0x90,
|
|
0xAD, 0xAC, 0x6E, 0xD1, 0x58, 0x35, 0x03, 0x97, 0x16, 0x10, 0x82, 0x56,
|
|
0xD8, 0x55, 0xFF, 0xE1, 0x8A, 0xA3, 0x2E, 0xF6, /* Random */
|
|
0x00, /* Session id len */
|
|
0x00 /* Cookie len */
|
|
};
|
|
|
|
/* First fragment of a ClientHello which is too short */
|
|
static const unsigned char clienthello_nocookie_short[] = {
|
|
0x16, /* Handshake */
|
|
0xFE, 0xFF, /* DTLSv1.0 */
|
|
0x00, 0x00, /* Epoch */
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* Record sequence number */
|
|
0x00, 0x2F, /* Record Length */
|
|
0x01, /* ClientHello */
|
|
0x00, 0x00, 0x2E, /* Message length */
|
|
0x00, 0x00, /* Message sequence */
|
|
0x00, 0x00, 0x00, /* Fragment offset */
|
|
0x00, 0x00, 0x23, /* Fragment length */
|
|
0xFE, 0xFD, /* DTLSv1.2 */
|
|
0xCA, 0x18, 0x9F, 0x76, 0xEC, 0x57, 0xCE, 0xE5, 0xB3, 0xAB, 0x79, 0x90,
|
|
0xAD, 0xAC, 0x6E, 0xD1, 0x58, 0x35, 0x03, 0x97, 0x16, 0x10, 0x82, 0x56,
|
|
0xD8, 0x55, 0xFF, 0xE1, 0x8A, 0xA3, 0x2E, 0xF6, /* Random */
|
|
0x00 /* Session id len */
|
|
};
|
|
|
|
/* Second fragment of a ClientHello */
|
|
static const unsigned char clienthello_2ndfrag[] = {
|
|
0x16, /* Handshake */
|
|
0xFE, 0xFF, /* DTLSv1.0 */
|
|
0x00, 0x00, /* Epoch */
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* Record sequence number */
|
|
0x00, 0x38, /* Record Length */
|
|
0x01, /* ClientHello */
|
|
0x00, 0x00, 0x2E, /* Message length */
|
|
0x00, 0x00, /* Message sequence */
|
|
0x00, 0x00, 0x02, /* Fragment offset */
|
|
0x00, 0x00, 0x2C, /* Fragment length */
|
|
/* Version skipped - sent in first fragment */
|
|
0xCA, 0x18, 0x9F, 0x76, 0xEC, 0x57, 0xCE, 0xE5, 0xB3, 0xAB, 0x79, 0x90,
|
|
0xAD, 0xAC, 0x6E, 0xD1, 0x58, 0x35, 0x03, 0x97, 0x16, 0x10, 0x82, 0x56,
|
|
0xD8, 0x55, 0xFF, 0xE1, 0x8A, 0xA3, 0x2E, 0xF6, /* Random */
|
|
0x00, /* Session id len */
|
|
0x00, /* Cookie len */
|
|
0x00, 0x04, /* Ciphersuites len */
|
|
0x00, 0x2f, /* AES128-SHA */
|
|
0x00, 0xff, /* Empty reneg info SCSV */
|
|
0x01, /* Compression methods len */
|
|
0x00, /* Null compression */
|
|
0x00, 0x00 /* Extensions len */
|
|
};
|
|
|
|
/* A ClientHello with a good cookie */
|
|
static const unsigned char clienthello_cookie[] = {
|
|
0x16, /* Handshake */
|
|
0xFE, 0xFF, /* DTLSv1.0 */
|
|
0x00, 0x00, /* Epoch */
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* Record sequence number */
|
|
0x00, 0x4E, /* Record Length */
|
|
0x01, /* ClientHello */
|
|
0x00, 0x00, 0x42, /* Message length */
|
|
0x00, 0x00, /* Message sequence */
|
|
0x00, 0x00, 0x00, /* Fragment offset */
|
|
0x00, 0x00, 0x42, /* Fragment length */
|
|
0xFE, 0xFD, /* DTLSv1.2 */
|
|
0xCA, 0x18, 0x9F, 0x76, 0xEC, 0x57, 0xCE, 0xE5, 0xB3, 0xAB, 0x79, 0x90,
|
|
0xAD, 0xAC, 0x6E, 0xD1, 0x58, 0x35, 0x03, 0x97, 0x16, 0x10, 0x82, 0x56,
|
|
0xD8, 0x55, 0xFF, 0xE1, 0x8A, 0xA3, 0x2E, 0xF6, /* Random */
|
|
0x00, /* Session id len */
|
|
0x14, /* Cookie len */
|
|
0x00, 0x01, 0x02, 0x03, 0x04, 005, 0x06, 007, 0x08, 0x09, 0x0A, 0x0B, 0x0C,
|
|
0x0D, 0x0E, 0x0F, 0x10, 0x11, 0x12, 0x13, /* Cookie */
|
|
0x00, 0x04, /* Ciphersuites len */
|
|
0x00, 0x2f, /* AES128-SHA */
|
|
0x00, 0xff, /* Empty reneg info SCSV */
|
|
0x01, /* Compression methods len */
|
|
0x00, /* Null compression */
|
|
0x00, 0x00 /* Extensions len */
|
|
};
|
|
|
|
/* A fragmented ClientHello with a good cookie */
|
|
static const unsigned char clienthello_cookie_frag[] = {
|
|
0x16, /* Handshake */
|
|
0xFE, 0xFF, /* DTLSv1.0 */
|
|
0x00, 0x00, /* Epoch */
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* Record sequence number */
|
|
0x00, 0x44, /* Record Length */
|
|
0x01, /* ClientHello */
|
|
0x00, 0x00, 0x42, /* Message length */
|
|
0x00, 0x00, /* Message sequence */
|
|
0x00, 0x00, 0x00, /* Fragment offset */
|
|
0x00, 0x00, 0x38, /* Fragment length */
|
|
0xFE, 0xFD, /* DTLSv1.2 */
|
|
0xCA, 0x18, 0x9F, 0x76, 0xEC, 0x57, 0xCE, 0xE5, 0xB3, 0xAB, 0x79, 0x90,
|
|
0xAD, 0xAC, 0x6E, 0xD1, 0x58, 0x35, 0x03, 0x97, 0x16, 0x10, 0x82, 0x56,
|
|
0xD8, 0x55, 0xFF, 0xE1, 0x8A, 0xA3, 0x2E, 0xF6, /* Random */
|
|
0x00, /* Session id len */
|
|
0x14, /* Cookie len */
|
|
0x00, 0x01, 0x02, 0x03, 0x04, 005, 0x06, 007, 0x08, 0x09, 0x0A, 0x0B, 0x0C,
|
|
0x0D, 0x0E, 0x0F, 0x10, 0x11, 0x12, 0x13 /* Cookie */
|
|
};
|
|
|
|
|
|
/* A ClientHello with a bad cookie */
|
|
static const unsigned char clienthello_badcookie[] = {
|
|
0x16, /* Handshake */
|
|
0xFE, 0xFF, /* DTLSv1.0 */
|
|
0x00, 0x00, /* Epoch */
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* Record sequence number */
|
|
0x00, 0x4E, /* Record Length */
|
|
0x01, /* ClientHello */
|
|
0x00, 0x00, 0x42, /* Message length */
|
|
0x00, 0x00, /* Message sequence */
|
|
0x00, 0x00, 0x00, /* Fragment offset */
|
|
0x00, 0x00, 0x42, /* Fragment length */
|
|
0xFE, 0xFD, /* DTLSv1.2 */
|
|
0xCA, 0x18, 0x9F, 0x76, 0xEC, 0x57, 0xCE, 0xE5, 0xB3, 0xAB, 0x79, 0x90,
|
|
0xAD, 0xAC, 0x6E, 0xD1, 0x58, 0x35, 0x03, 0x97, 0x16, 0x10, 0x82, 0x56,
|
|
0xD8, 0x55, 0xFF, 0xE1, 0x8A, 0xA3, 0x2E, 0xF6, /* Random */
|
|
0x00, /* Session id len */
|
|
0x14, /* Cookie len */
|
|
0x01, 0x01, 0x02, 0x03, 0x04, 005, 0x06, 007, 0x08, 0x09, 0x0A, 0x0B, 0x0C,
|
|
0x0D, 0x0E, 0x0F, 0x10, 0x11, 0x12, 0x13, /* Cookie */
|
|
0x00, 0x04, /* Ciphersuites len */
|
|
0x00, 0x2f, /* AES128-SHA */
|
|
0x00, 0xff, /* Empty reneg info SCSV */
|
|
0x01, /* Compression methods len */
|
|
0x00, /* Null compression */
|
|
0x00, 0x00 /* Extensions len */
|
|
};
|
|
|
|
/* A fragmented ClientHello with the fragment boundary mid cookie */
|
|
static const unsigned char clienthello_cookie_short[] = {
|
|
0x16, /* Handshake */
|
|
0xFE, 0xFF, /* DTLSv1.0 */
|
|
0x00, 0x00, /* Epoch */
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* Record sequence number */
|
|
0x00, 0x43, /* Record Length */
|
|
0x01, /* ClientHello */
|
|
0x00, 0x00, 0x42, /* Message length */
|
|
0x00, 0x00, /* Message sequence */
|
|
0x00, 0x00, 0x00, /* Fragment offset */
|
|
0x00, 0x00, 0x37, /* Fragment length */
|
|
0xFE, 0xFD, /* DTLSv1.2 */
|
|
0xCA, 0x18, 0x9F, 0x76, 0xEC, 0x57, 0xCE, 0xE5, 0xB3, 0xAB, 0x79, 0x90,
|
|
0xAD, 0xAC, 0x6E, 0xD1, 0x58, 0x35, 0x03, 0x97, 0x16, 0x10, 0x82, 0x56,
|
|
0xD8, 0x55, 0xFF, 0xE1, 0x8A, 0xA3, 0x2E, 0xF6, /* Random */
|
|
0x00, /* Session id len */
|
|
0x14, /* Cookie len */
|
|
0x00, 0x01, 0x02, 0x03, 0x04, 005, 0x06, 007, 0x08, 0x09, 0x0A, 0x0B, 0x0C,
|
|
0x0D, 0x0E, 0x0F, 0x10, 0x11, 0x12 /* Cookie */
|
|
};
|
|
|
|
/* Bad record - too short */
|
|
static const unsigned char record_short[] = {
|
|
0x16, /* Handshake */
|
|
0xFE, 0xFF, /* DTLSv1.0 */
|
|
0x00, 0x00, /* Epoch */
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00 /* Record sequence number */
|
|
};
|
|
|
|
static const unsigned char verify[] = {
|
|
0x16, /* Handshake */
|
|
0xFE, 0xFF, /* DTLSv1.0 */
|
|
0x00, 0x00, /* Epoch */
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* Record sequence number */
|
|
0x00, 0x23, /* Record Length */
|
|
0x03, /* HelloVerifyRequest */
|
|
0x00, 0x00, 0x17, /* Message length */
|
|
0x00, 0x00, /* Message sequence */
|
|
0x00, 0x00, 0x00, /* Fragment offset */
|
|
0x00, 0x00, 0x17, /* Fragment length */
|
|
0xFE, 0xFF, /* DTLSv1.0 */
|
|
0x14, /* Cookie len */
|
|
0x00, 0x01, 0x02, 0x03, 0x04, 005, 0x06, 007, 0x08, 0x09, 0x0A, 0x0B, 0x0C,
|
|
0x0D, 0x0E, 0x0F, 0x10, 0x11, 0x12, 0x13 /* Cookie */
|
|
};
|
|
|
|
typedef struct {
|
|
const unsigned char *in;
|
|
unsigned int inlen;
|
|
/*
|
|
* GOOD == positive return value from DTLSv1_listen, no output yet
|
|
* VERIFY == 0 return value, HelloVerifyRequest sent
|
|
* DROP == 0 return value, no output
|
|
*/
|
|
enum {GOOD, VERIFY, DROP} outtype;
|
|
} tests;
|
|
|
|
static tests testpackets[9] = {
|
|
{ clienthello_nocookie, sizeof(clienthello_nocookie), VERIFY },
|
|
{ clienthello_nocookie_frag, sizeof(clienthello_nocookie_frag), VERIFY },
|
|
{ clienthello_nocookie_short, sizeof(clienthello_nocookie_short), DROP },
|
|
{ clienthello_2ndfrag, sizeof(clienthello_2ndfrag), DROP },
|
|
{ clienthello_cookie, sizeof(clienthello_cookie), GOOD },
|
|
{ clienthello_cookie_frag, sizeof(clienthello_cookie_frag), GOOD },
|
|
{ clienthello_badcookie, sizeof(clienthello_badcookie), VERIFY },
|
|
{ clienthello_cookie_short, sizeof(clienthello_cookie_short), DROP },
|
|
{ record_short, sizeof(record_short), DROP }
|
|
};
|
|
|
|
# define COOKIE_LEN 20
|
|
|
|
static int cookie_gen(SSL *ssl, unsigned char *cookie, unsigned int *cookie_len)
|
|
{
|
|
unsigned int i;
|
|
|
|
for (i = 0; i < COOKIE_LEN; i++, cookie++)
|
|
*cookie = i;
|
|
*cookie_len = COOKIE_LEN;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static int cookie_verify(SSL *ssl, const unsigned char *cookie,
|
|
unsigned int cookie_len)
|
|
{
|
|
unsigned int i;
|
|
|
|
if (cookie_len != COOKIE_LEN)
|
|
return 0;
|
|
|
|
for (i = 0; i < COOKIE_LEN; i++, cookie++) {
|
|
if (*cookie != i)
|
|
return 0;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
static int dtls_listen_test(int i)
|
|
{
|
|
SSL_CTX *ctx = NULL;
|
|
SSL *ssl = NULL;
|
|
BIO *outbio = NULL;
|
|
BIO *inbio = NULL;
|
|
BIO_ADDR *peer = NULL;
|
|
tests *tp = &testpackets[i];
|
|
char *data;
|
|
long datalen;
|
|
int ret, success = 0;
|
|
|
|
if (!TEST_ptr(ctx = SSL_CTX_new(DTLS_server_method()))
|
|
|| !TEST_ptr(peer = BIO_ADDR_new()))
|
|
goto err;
|
|
SSL_CTX_set_cookie_generate_cb(ctx, cookie_gen);
|
|
SSL_CTX_set_cookie_verify_cb(ctx, cookie_verify);
|
|
|
|
/* Create an SSL object and set the BIO */
|
|
if (!TEST_ptr(ssl = SSL_new(ctx))
|
|
|| !TEST_ptr(outbio = BIO_new(BIO_s_mem())))
|
|
goto err;
|
|
SSL_set0_wbio(ssl, outbio);
|
|
|
|
/* Set Non-blocking IO behaviour */
|
|
if (!TEST_ptr(inbio = BIO_new_mem_buf((char *)tp->in, tp->inlen)))
|
|
goto err;
|
|
BIO_set_mem_eof_return(inbio, -1);
|
|
SSL_set0_rbio(ssl, inbio);
|
|
|
|
/* Process the incoming packet */
|
|
if (!TEST_int_ge(ret = DTLSv1_listen(ssl, peer), 0))
|
|
goto err;
|
|
datalen = BIO_get_mem_data(outbio, &data);
|
|
|
|
if (tp->outtype == VERIFY) {
|
|
if (!TEST_int_eq(ret, 0)
|
|
|| !TEST_mem_eq(data, datalen, verify, sizeof(verify)))
|
|
goto err;
|
|
} else if (datalen == 0) {
|
|
if (!TEST_true((ret == 0 && tp->outtype == DROP)
|
|
|| (ret == 1 && tp->outtype == GOOD)))
|
|
goto err;
|
|
} else {
|
|
TEST_info("Test %d: unexpected data output", i);
|
|
goto err;
|
|
}
|
|
(void)BIO_reset(outbio);
|
|
inbio = NULL;
|
|
SSL_set0_rbio(ssl, NULL);
|
|
success = 1;
|
|
|
|
err:
|
|
/* Also frees up outbio */
|
|
SSL_free(ssl);
|
|
SSL_CTX_free(ctx);
|
|
BIO_free(inbio);
|
|
OPENSSL_free(peer);
|
|
return success;
|
|
}
|
|
#endif
|
|
|
|
int setup_tests()
|
|
{
|
|
#ifndef OPENSSL_NO_SOCK
|
|
ADD_ALL_TESTS(dtls_listen_test, (int)OSSL_NELEM(testpackets));
|
|
#endif
|
|
return 1;
|
|
}
|