Fix usage of OPENSSL_NO_*_METHOD
[openssl.git] / test / clienthellotest.c
1 /* Written by Matt Caswell for the OpenSSL Project */
2 /* ====================================================================
3  * Copyright (c) 1998-2015 The OpenSSL Project.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  *
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in
14  *    the documentation and/or other materials provided with the
15  *    distribution.
16  *
17  * 3. All advertising materials mentioning features or use of this
18  *    software must display the following acknowledgment:
19  *    "This product includes software developed by the OpenSSL Project
20  *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
21  *
22  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
23  *    endorse or promote products derived from this software without
24  *    prior written permission. For written permission, please contact
25  *    openssl-core@openssl.org.
26  *
27  * 5. Products derived from this software may not be called "OpenSSL"
28  *    nor may "OpenSSL" appear in their names without prior written
29  *    permission of the OpenSSL Project.
30  *
31  * 6. Redistributions of any form whatsoever must retain the following
32  *    acknowledgment:
33  *    "This product includes software developed by the OpenSSL Project
34  *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
35  *
36  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
37  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
38  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
39  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
40  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
41  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
42  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
43  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
44  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
45  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
46  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
47  * OF THE POSSIBILITY OF SUCH DAMAGE.
48  * ====================================================================
49  *
50  * This product includes cryptographic software written by Eric Young
51  * (eay@cryptsoft.com).  This product includes software written by Tim
52  * Hudson (tjh@cryptsoft.com).
53  *
54  */
55
56 #include <string.h>
57
58 #include <openssl/opensslconf.h>
59 #include <openssl/bio.h>
60 #include <openssl/crypto.h>
61 #include <openssl/evp.h>
62 #include <openssl/ssl.h>
63 #include <openssl/err.h>
64
65
66 #define CLIENT_VERSION_LEN      2
67 #define SESSION_ID_LEN_LEN      1
68 #define CIPHERS_LEN_LEN         2
69 #define COMPRESSION_LEN_LEN     1
70 #define EXTENSIONS_LEN_LEN      2
71 #define EXTENSION_TYPE_LEN      2
72 #define EXTENSION_SIZE_LEN      2
73
74
75 #define TOTAL_NUM_TESTS                         2
76
77 /*
78  * Test that explicitly setting ticket data results in it appearing in the
79  * ClientHello for TLS1.2
80  */
81 #define TEST_SET_SESSION_TICK_DATA_TLS_1_2      0
82
83 /*
84  * Test that explicitly setting ticket data results in it appearing in the
85  * ClientHello for a negotiated SSL/TLS version
86  */
87 #define TEST_SET_SESSION_TICK_DATA_VER_NEG      1
88
89 int main(int argc, char *argv[])
90 {
91     SSL_CTX *ctx;
92     SSL *con;
93     BIO *rbio;
94     BIO *wbio;
95     BIO *err;
96     long len;
97     unsigned char *data;
98     unsigned char *dataend;
99     char *dummytick = "Hello World!";
100     unsigned int tmplen;
101     unsigned int type;
102     unsigned int size;
103     int testresult = 0;
104     int currtest = 0;
105
106     err = BIO_new_fp(stderr, BIO_NOCLOSE | BIO_FP_TEXT);
107
108     CRYPTO_set_mem_debug(1);
109     CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON);
110
111     /*
112      * For each test set up an SSL_CTX and SSL and see what ClientHello gets
113      * produced when we try to connect
114      */
115     for (; currtest < TOTAL_NUM_TESTS; currtest++) {
116         testresult = 0;
117         if (currtest == TEST_SET_SESSION_TICK_DATA_TLS_1_2) {
118 #ifndef OPENSSL_NO_TLS1_2_METHOD
119             ctx = SSL_CTX_new(TLSv1_2_method());
120 #else
121             testresult = 1;
122             continue;
123 #endif
124         } else {
125             ctx = SSL_CTX_new(TLS_method());
126         }
127         con = SSL_new(ctx);
128
129         rbio = BIO_new(BIO_s_mem());
130         wbio = BIO_new(BIO_s_mem());
131         SSL_set_bio(con, rbio, wbio);
132         SSL_set_connect_state(con);
133
134         if (currtest == TEST_SET_SESSION_TICK_DATA_TLS_1_2
135                 || currtest == TEST_SET_SESSION_TICK_DATA_VER_NEG) {
136             if (!SSL_set_session_ticket_ext(con, dummytick, strlen(dummytick)))
137                 goto end;
138         }
139
140         if (SSL_connect(con) > 0) {
141             /* This shouldn't succeed because we don't have a server! */
142             goto end;
143         }
144
145         len = BIO_get_mem_data(wbio, (char **)&data);
146         dataend = data + len;
147
148         /* Skip the record header */
149         data += SSL3_RT_HEADER_LENGTH;
150         /* Skip the handshake message header */
151         data += SSL3_HM_HEADER_LENGTH;
152         /* Skip client version and random */
153         data += CLIENT_VERSION_LEN + SSL3_RANDOM_SIZE;
154         if (data + SESSION_ID_LEN_LEN > dataend)
155             goto end;
156         /* Skip session id */
157         tmplen = *data;
158         data += SESSION_ID_LEN_LEN + tmplen;
159         if (data + CIPHERS_LEN_LEN > dataend)
160             goto end;
161         /* Skip ciphers */
162         tmplen = ((*data) << 8) | *(data + 1);
163         data += CIPHERS_LEN_LEN + tmplen;
164         if (data + COMPRESSION_LEN_LEN > dataend)
165             goto end;
166         /* Skip compression */
167         tmplen = *data;
168         data += COMPRESSION_LEN_LEN + tmplen;
169         if (data + EXTENSIONS_LEN_LEN > dataend)
170             goto end;
171         /* Extensions len */
172         tmplen = ((*data) << 8) | *(data + 1);
173         data += EXTENSIONS_LEN_LEN;
174         if (data + tmplen > dataend)
175             goto end;
176
177         /* Loop through all extensions */
178         while (tmplen > EXTENSION_TYPE_LEN + EXTENSION_SIZE_LEN) {
179             type = ((*data) << 8) | *(data + 1);
180             data += EXTENSION_TYPE_LEN;
181             size = ((*data) << 8) | *(data + 1);
182             data += EXTENSION_SIZE_LEN;
183             if (data + size > dataend)
184                 goto end;
185
186             if (type == TLSEXT_TYPE_session_ticket) {
187                 if (currtest == TEST_SET_SESSION_TICK_DATA_TLS_1_2
188                         || currtest == TEST_SET_SESSION_TICK_DATA_VER_NEG) {
189                     if (size == strlen(dummytick)
190                             && memcmp(data, dummytick, size) == 0) {
191                         /* Ticket data is as we expected */
192                         testresult = 1;
193                     } else {
194                         printf("Received session ticket is not as expected\n");
195                     }
196                     break;
197                 }
198             }
199
200             tmplen -= EXTENSION_TYPE_LEN + EXTENSION_SIZE_LEN + size;
201             data += size;
202         }
203
204  end:
205         SSL_free(con);
206         SSL_CTX_free(ctx);
207         if (!testresult) {
208             printf("ClientHello test: FAILED (Test %d)\n", currtest);
209             break;
210         }
211     }
212
213 #ifndef OPENSSL_NO_CRYPTO_MDEBUG
214     if (CRYPTO_mem_leaks(err) <= 0)
215         testresult = 0;
216 #endif
217     BIO_free(err);
218
219     return testresult?0:1;
220 }