Add main() test methods to reduce test boilerplate.
[openssl.git] / test / dtlstest.c
1 /*
2  * Copyright 2016 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the OpenSSL license (the "License").  You may not use
5  * this file except in compliance with the License.  You can obtain a copy
6  * in the file LICENSE in the source distribution or at
7  * https://www.openssl.org/source/license.html
8  */
9
10 #include <openssl/bio.h>
11 #include <openssl/crypto.h>
12 #include <openssl/ssl.h>
13 #include <openssl/err.h>
14
15 #include "ssltestlib.h"
16 #include "testutil.h"
17 #include "test_main_custom.h"
18
19 static char *cert = NULL;
20 static char *privkey = NULL;
21
22 #define NUM_TESTS   2
23
24
25 #define DUMMY_CERT_STATUS_LEN  12
26
27 static unsigned char certstatus[] = {
28     SSL3_RT_HANDSHAKE, /* Content type */
29     0xfe, 0xfd, /* Record version */
30     0, 1, /* Epoch */
31     0, 0, 0, 0, 0, 0x0f, /* Record sequence number */
32     0, DTLS1_HM_HEADER_LENGTH + DUMMY_CERT_STATUS_LEN - 2,
33     SSL3_MT_CERTIFICATE_STATUS, /* Cert Status handshake message type */
34     0, 0, DUMMY_CERT_STATUS_LEN, /* Message len */
35     0, 5, /* Message sequence */
36     0, 0, 0, /* Fragment offset */
37     0, 0, DUMMY_CERT_STATUS_LEN - 2, /* Fragment len */
38     0x80, 0x80, 0x80, 0x80, 0x80,
39     0x80, 0x80, 0x80, 0x80, 0x80 /* Dummy data */
40 };
41
42 #define RECORD_SEQUENCE 10
43
44 static int test_dtls_unprocessed(int testidx)
45 {
46     SSL_CTX *sctx = NULL, *cctx = NULL;
47     SSL *serverssl1 = NULL, *clientssl1 = NULL;
48     BIO *c_to_s_fbio, *c_to_s_mempacket;
49     int testresult = 0;
50
51     printf("Starting Test %d\n", testidx);
52
53     if (!create_ssl_ctx_pair(DTLS_server_method(), DTLS_client_method(), &sctx,
54                              &cctx, cert, privkey)) {
55         printf("Unable to create SSL_CTX pair\n");
56         return 0;
57     }
58
59     if (!SSL_CTX_set_cipher_list(cctx, "AES128-SHA")) {
60         printf("Failed setting cipher list\n");
61     }
62
63     c_to_s_fbio = BIO_new(bio_f_tls_dump_filter());
64     if (c_to_s_fbio == NULL) {
65         printf("Failed to create filter BIO\n");
66         goto end;
67     }
68
69     /* BIO is freed by create_ssl_connection on error */
70     if (!create_ssl_objects(sctx, cctx, &serverssl1, &clientssl1, NULL,
71                                c_to_s_fbio)) {
72         printf("Unable to create SSL objects\n");
73         ERR_print_errors_fp(stdout);
74         goto end;
75     }
76
77     if (testidx == 1)
78         certstatus[RECORD_SEQUENCE] = 0xff;
79
80     /*
81      * Inject a dummy record from the next epoch. In test 0, this should never
82      * get used because the message sequence number is too big. In test 1 we set
83      * the record sequence number to be way off in the future. This should not
84      * have an impact on the record replay protection because the record should
85      * be dropped before it is marked as arrived
86      */
87     c_to_s_mempacket = SSL_get_wbio(clientssl1);
88     c_to_s_mempacket = BIO_next(c_to_s_mempacket);
89     mempacket_test_inject(c_to_s_mempacket, (char *)certstatus,
90                           sizeof(certstatus), 1, INJECT_PACKET_IGNORE_REC_SEQ);
91
92     if (!create_ssl_connection(serverssl1, clientssl1)) {
93         printf("Unable to create SSL connection\n");
94         ERR_print_errors_fp(stdout);
95         goto end;
96     }
97
98     testresult = 1;
99  end:
100     SSL_free(serverssl1);
101     SSL_free(clientssl1);
102     SSL_CTX_free(sctx);
103     SSL_CTX_free(cctx);
104
105     return testresult;
106 }
107
108 int test_main(int argc, char *argv[])
109 {
110     int testresult = 1;
111
112     if (argc != 3) {
113         printf("Invalid argument count\n");
114         return 1;
115     }
116
117     cert = argv[1];
118     privkey = argv[2];
119
120     ADD_ALL_TESTS(test_dtls_unprocessed, NUM_TESTS);
121
122     testresult = run_tests(argv[0]);
123
124     bio_f_tls_dump_filter_free();
125     bio_s_mempacket_test_free();
126
127     return testresult;
128 }