diff options
author | Matt Caswell <matt@openssl.org> | 2016-09-08 10:01:24 +0100 |
---|---|---|
committer | Matt Caswell <matt@openssl.org> | 2016-09-13 09:41:21 +0100 |
commit | d6c4cc293974e622b387458d2293e29f8f14fbbb (patch) | |
tree | 30b685a2149c549d9f64835e948500bdd36f573a /test/wpackettest.c | |
parent | 796a627e0a816ffbd79f53fa7d349e4edb624573 (diff) |
Add tests for the WPACKET implementation
The tests will only work in no-shared builds because WPACKET is an
internal only API that does not get exported by the shared library.
Reviewed-by: Rich Salz <rsalz@openssl.org>
Diffstat (limited to 'test/wpackettest.c')
-rw-r--r-- | test/wpackettest.c | 446 |
1 files changed, 446 insertions, 0 deletions
diff --git a/test/wpackettest.c b/test/wpackettest.c new file mode 100644 index 0000000000..79248e3ba8 --- /dev/null +++ b/test/wpackettest.c @@ -0,0 +1,446 @@ +/* + * Copyright 2016 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/buffer.h> +#include "../ssl/packet_locl.h" +#include "testutil.h" + +const static unsigned char simple1 = 0xff; +const static unsigned char simple2[] = { 0x01, 0xff }; +const static unsigned char simple3[] = { 0x00, 0x00, 0x00, 0x01, 0xff }; +const static unsigned char nestedsub[] = { 0x03, 0xff, 0x01, 0xff }; +const static unsigned char seqsub[] = { 0x01, 0xff, 0x01, 0xff }; +const static unsigned char empty = 0x00; +const static unsigned char alloc[] = { 0x02, 0xfe, 0xff }; +const static unsigned char submem[] = { 0x03, 0x02, 0xfe, 0xff }; + +static BUF_MEM *buf; + +static void testfail(const char *msg, WPACKET *pkt) +{ + fprintf(stderr, "%s", msg); + WPACKET_cleanup(pkt); +} + +static int test_WPACKET_init(void) +{ + WPACKET pkt; + int i; + size_t written; + + if ( !WPACKET_init(&pkt, buf) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + /* Closing a top level WPACKET should fail */ + || WPACKET_close(&pkt) + /* Finishing a top level WPACKET should succeed */ + || !WPACKET_finish(&pkt) + /* + * Can't call close or finish on a WPACKET that's already + * finished. + */ + || WPACKET_close(&pkt) + || WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(simple1) + || memcmp(buf->data, &simple1, written) != 0) { + testfail("test_WPACKET_init():1 failed\n", &pkt); + return 0; + } + + /* Now try with a one byte length prefix */ + if ( !WPACKET_init_len(&pkt, buf, 1) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(simple2) + || memcmp(buf->data, &simple2, written) != 0) { + testfail("test_WPACKET_init():2 failed\n", &pkt); + return 0; + } + + /* And a longer length prefix */ + if ( !WPACKET_init_len(&pkt, buf, 4) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(simple3) + || memcmp(buf->data, &simple3, written) != 0) { + testfail("test_WPACKET_init():3 failed\n", &pkt); + return 0; + } + + if (!WPACKET_init_len(&pkt, buf, 1)) { + testfail("test_WPACKET_init():4 failed\n", &pkt); + return 0; + } + for (i = 1; i < 257; i++) { + /* + * Putting more bytes in than fit for the size of the length prefix + * should fail + */ + if ((!WPACKET_put_bytes(&pkt, 0xff, 1)) == (i != 256)) { + testfail("test_WPACKET_init():4 failed\n", &pkt); + return 0; + } + } + if (!WPACKET_finish(&pkt)) { + testfail("test_WPACKET_init():4 failed\n", &pkt); + return 0; + } + + return 1; +} + +static int test_WPACKET_set_packet_len(void) +{ + WPACKET pkt; + size_t written; + unsigned char len; + + /* + * Calling set_packet_len when the packet len is already set + * should fail + */ + if ( !WPACKET_init_len(&pkt, buf, 1) + || WPACKET_set_packet_len(&pkt, &len, sizeof(len)) + || !WPACKET_finish(&pkt)) { + testfail("test_WPACKET_set_packet_len():1 failed\n", &pkt); + return 0; + } + + if ( !WPACKET_init(&pkt, buf) + || !WPACKET_set_packet_len(&pkt, &len, sizeof(len)) + /* Can't set it again */ + || WPACKET_set_packet_len(&pkt, &len, sizeof(len)) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(simple1) + || memcmp(buf->data, &simple1, written) != 0 + || len != 1) { + testfail("test_WPACKET_set_packet_len():2 failed\n", &pkt); + return 0; + } + + return 1; +} + +static int test_WPACKET_set_max_size(void) +{ + WPACKET pkt; + size_t written; + unsigned char len; + + if ( !WPACKET_init(&pkt, buf) + /* + * No previous lenbytes set so we should be ok to set the max + * possible max size + */ + || !WPACKET_set_max_size(&pkt, SIZE_MAX) + /* We should be able to set it smaller too */ + || !WPACKET_set_max_size(&pkt, SIZE_MAX -1) + /* And setting it bigger again should be ok */ + || !WPACKET_set_max_size(&pkt, SIZE_MAX) + || !WPACKET_set_packet_len(&pkt, &len, 1) + /* + * Max size can't be bigger than biggest that will fit in + * lenbytes + */ + || WPACKET_set_max_size(&pkt, 0x0101) + /* It can be the same as the maximum possible size */ + || !WPACKET_set_max_size(&pkt, 0xff) + /* Or it can be less */ + || !WPACKET_set_max_size(&pkt, 0x00) + /* + * Should fail because packet is already filled + */ + || WPACKET_put_bytes(&pkt, 0xff, 1) + /* + * You can't put in more bytes than max size + */ + || !WPACKET_set_max_size(&pkt, 0x01) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(simple1) + || memcmp(buf->data, &simple1, written) != 0 + || len != 1) { + testfail("test_WPACKET_set_max_size():1 failed\n", &pkt); + return 0; + } + + if ( !WPACKET_init_len(&pkt, buf, 1) + /* + * Should fail because we already consumed 1 byte with the + * length + */ + || WPACKET_set_max_size(&pkt, 0) + || !WPACKET_set_max_size(&pkt, 1) + || WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_set_max_size(&pkt, 2) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(simple2) + || memcmp(buf->data, &simple2, written) != 0 + || len != 1) { + testfail("test_WPACKET_set_max_size():2 failed\n", &pkt); + return 0; + } + + return 1; +} + +static int test_WPACKET_start_sub_packet(void) +{ + WPACKET pkt; + size_t written; + size_t len; + + if ( !WPACKET_init(&pkt, buf) + || !WPACKET_start_sub_packet(&pkt) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + /* Can't finish because we have a sub packet */ + || WPACKET_finish(&pkt) + || !WPACKET_close(&pkt) + /* Sub packet is closed so can't close again */ + || WPACKET_close(&pkt) + /* Now a top level so finish should succeed */ + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(simple1) + || memcmp(buf->data, &simple1, written) != 0) { + testfail("test_WPACKET_start_sub_packet():1 failed\n", &pkt); + return 0; + } + + /* Single sub-packet with length prefix */ + if ( !WPACKET_init(&pkt, buf) + || !WPACKET_start_sub_packet_len(&pkt, 1) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_close(&pkt) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(simple2) + || memcmp(buf->data, &simple2, written) != 0) { + testfail("test_WPACKET_start_sub_packet():2 failed\n", &pkt); + return 0; + } + + /* Nested sub-packets with length prefixes */ + if ( !WPACKET_init(&pkt, buf) + || !WPACKET_start_sub_packet_len(&pkt, 1) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_start_sub_packet_len(&pkt, 1) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_get_length(&pkt, &len) + || len != 1 + || !WPACKET_close(&pkt) + || !WPACKET_get_length(&pkt, &len) + || len != 3 + || !WPACKET_close(&pkt) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(nestedsub) + || memcmp(buf->data, &nestedsub, written) != 0) { + testfail("test_WPACKET_start_sub_packet():3 failed\n", &pkt); + return 0; + } + + /* Sequential sub-packets with length prefixes */ + if ( !WPACKET_init(&pkt, buf) + || !WPACKET_start_sub_packet_len(&pkt, 1) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_close(&pkt) + || !WPACKET_start_sub_packet_len(&pkt, 1) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_close(&pkt) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(seqsub) + || memcmp(buf->data, &seqsub, written) != 0) { + testfail("test_WPACKET_start_sub_packet():4 failed\n", &pkt); + return 0; + } + + return 1; +} + + +static int test_WPACKET_set_flags(void) +{ + WPACKET pkt; + size_t written; + + /* Set packet to be non-zero length */ + if ( !WPACKET_init(&pkt, buf) + || !WPACKET_set_flags(&pkt, OPENSSL_WPACKET_FLAGS_NON_ZERO_LENGTH) + /* Should fail because of zero length */ + || WPACKET_finish(&pkt) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(simple1) + || memcmp(buf->data, &simple1, written) != 0) { + testfail("test_WPACKET_set_flags():1 failed\n", &pkt); + return 0; + } + + /* Repeat above test in a sub-packet */ + if ( !WPACKET_init(&pkt, buf) + || !WPACKET_start_sub_packet(&pkt) + || !WPACKET_set_flags(&pkt, OPENSSL_WPACKET_FLAGS_NON_ZERO_LENGTH) + /* Should fail because of zero length */ + || WPACKET_close(&pkt) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_close(&pkt) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(simple1) + || memcmp(buf->data, &simple1, written) != 0) { + testfail("test_WPACKET_set_flags():2 failed\n", &pkt); + return 0; + } + + /* Set packet to abandon non-zero length */ + if ( !WPACKET_init_len(&pkt, buf, 1) + || !WPACKET_set_flags(&pkt, + OPENSSL_WPACKET_FLAGS_ABANDON_ON_ZERO_LENGTH) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != 0) { + testfail("test_WPACKET_set_flags():3 failed\n", &pkt); + return 0; + } + + /* Repeat above test but only abandon a sub-packet */ + if ( !WPACKET_init_len(&pkt, buf, 1) + || !WPACKET_start_sub_packet_len(&pkt, 1) + || !WPACKET_set_flags(&pkt, + OPENSSL_WPACKET_FLAGS_ABANDON_ON_ZERO_LENGTH) + || !WPACKET_close(&pkt) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(empty) + || memcmp(buf->data, &empty, written) != 0) { + testfail("test_WPACKET_set_flags():4 failed\n", &pkt); + return 0; + } + + /* And repeat with a non empty sub-packet */ + if ( !WPACKET_init(&pkt, buf) + || !WPACKET_start_sub_packet_len(&pkt, 1) + || !WPACKET_set_flags(&pkt, + OPENSSL_WPACKET_FLAGS_ABANDON_ON_ZERO_LENGTH) + || !WPACKET_put_bytes(&pkt, 0xff, 1) + || !WPACKET_close(&pkt) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(simple2) + || memcmp(buf->data, &simple2, written) != 0) { + testfail("test_WPACKET_set_flags():5 failed\n", &pkt); + return 0; + } + return 1; +} + +static int test_WPACKET_allocate_bytes(void) +{ + WPACKET pkt; + size_t written; + unsigned char *bytes; + + if ( !WPACKET_init_len(&pkt, buf, 1) + || !WPACKET_allocate_bytes(&pkt, 2, &bytes)) { + testfail("test_WPACKET_allocate_bytes():1 failed\n", &pkt); + return 0; + } + bytes[0] = 0xfe; + bytes[1] = 0xff; + if ( !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(alloc) + || memcmp(buf->data, &alloc, written) != 0) { + testfail("test_WPACKET_allocate_bytes():2 failed\n", &pkt); + return 0; + } + + return 1; +} + +static int test_WPACKET_memcpy(void) +{ + WPACKET pkt; + size_t written; + const unsigned char bytes[] = { 0xfe, 0xff }; + + if ( !WPACKET_init_len(&pkt, buf, 1) + || !WPACKET_memcpy(&pkt, bytes, sizeof(bytes)) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(alloc) + || memcmp(buf->data, &alloc, written) != 0) { + testfail("test_WPACKET_memcpy():1 failed\n", &pkt); + return 0; + } + + /* Repeat with WPACKET_sub_memcpy() */ + if ( !WPACKET_init_len(&pkt, buf, 1) + || !WPACKET_sub_memcpy(&pkt, bytes, sizeof(bytes), 1) + || !WPACKET_finish(&pkt) + || !WPACKET_get_total_written(&pkt, &written) + || written != sizeof(submem) + || memcmp(buf->data, &submem, written) != 0) { + testfail("test_WPACKET_memcpy():2 failed\n", &pkt); + return 0; + } + + return 1; +} + +int main(int argc, char *argv[]) +{ + BIO *err = NULL; + int testresult = 0; + + err = BIO_new_fp(stderr, BIO_NOCLOSE | BIO_FP_TEXT); + + CRYPTO_set_mem_debug(1); + CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON); + + buf = BUF_MEM_new(); + if (buf != NULL) { + ADD_TEST(test_WPACKET_init); + ADD_TEST(test_WPACKET_set_packet_len); + ADD_TEST(test_WPACKET_set_max_size); + ADD_TEST(test_WPACKET_start_sub_packet); + ADD_TEST(test_WPACKET_set_flags); + ADD_TEST(test_WPACKET_allocate_bytes); + ADD_TEST(test_WPACKET_memcpy); + + testresult = run_tests(argv[0]); + + BUF_MEM_free(buf); + } + +#ifndef OPENSSL_NO_CRYPTO_MDEBUG + if (CRYPTO_mem_leaks(err) <= 0) + testresult = 1; +#endif + BIO_free(err); + + if (!testresult) + printf("PASS\n"); + + return testresult; +} + |