1/*
2 * Copyright 2014-2021 The OpenSSL Project Authors. All Rights Reserved.
3 *
4 * Licensed under the Apache License 2.0 (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/* Custom extension utility functions */
11
12#include <openssl/ct.h>
13#include "../ssl_local.h"
14#include "internal/cryptlib.h"
15#include "statem_local.h"
16
17typedef struct {
18    void *add_arg;
19    custom_ext_add_cb add_cb;
20    custom_ext_free_cb free_cb;
21} custom_ext_add_cb_wrap;
22
23typedef struct {
24    void *parse_arg;
25    custom_ext_parse_cb parse_cb;
26} custom_ext_parse_cb_wrap;
27
28/*
29 * Provide thin wrapper callbacks which convert new style arguments to old style
30 */
31static int custom_ext_add_old_cb_wrap(SSL *s, unsigned int ext_type,
32                                      unsigned int context,
33                                      const unsigned char **out,
34                                      size_t *outlen, X509 *x, size_t chainidx,
35                                      int *al, void *add_arg)
36{
37    custom_ext_add_cb_wrap *add_cb_wrap = (custom_ext_add_cb_wrap *)add_arg;
38
39    if (add_cb_wrap->add_cb == NULL)
40        return 1;
41
42    return add_cb_wrap->add_cb(s, ext_type, out, outlen, al,
43                               add_cb_wrap->add_arg);
44}
45
46static void custom_ext_free_old_cb_wrap(SSL *s, unsigned int ext_type,
47                                        unsigned int context,
48                                        const unsigned char *out, void *add_arg)
49{
50    custom_ext_add_cb_wrap *add_cb_wrap = (custom_ext_add_cb_wrap *)add_arg;
51
52    if (add_cb_wrap->free_cb == NULL)
53        return;
54
55    add_cb_wrap->free_cb(s, ext_type, out, add_cb_wrap->add_arg);
56}
57
58static int custom_ext_parse_old_cb_wrap(SSL *s, unsigned int ext_type,
59                                        unsigned int context,
60                                        const unsigned char *in,
61                                        size_t inlen, X509 *x, size_t chainidx,
62                                        int *al, void *parse_arg)
63{
64    custom_ext_parse_cb_wrap *parse_cb_wrap =
65        (custom_ext_parse_cb_wrap *)parse_arg;
66
67    if (parse_cb_wrap->parse_cb == NULL)
68        return 1;
69
70    return parse_cb_wrap->parse_cb(s, ext_type, in, inlen, al,
71                                   parse_cb_wrap->parse_arg);
72}
73
74/*
75 * Find a custom extension from the list. The |role| param is there to
76 * support the legacy API where custom extensions for client and server could
77 * be set independently on the same SSL_CTX. It is set to ENDPOINT_SERVER if we
78 * are trying to find a method relevant to the server, ENDPOINT_CLIENT for the
79 * client, or ENDPOINT_BOTH for either
80 */
81custom_ext_method *custom_ext_find(const custom_ext_methods *exts,
82                                   ENDPOINT role, unsigned int ext_type,
83                                   size_t *idx)
84{
85    size_t i;
86    custom_ext_method *meth = exts->meths;
87
88    for (i = 0; i < exts->meths_count; i++, meth++) {
89        if (ext_type == meth->ext_type
90                && (role == ENDPOINT_BOTH || role == meth->role
91                    || meth->role == ENDPOINT_BOTH)) {
92            if (idx != NULL)
93                *idx = i;
94            return meth;
95        }
96    }
97    return NULL;
98}
99
100/*
101 * Initialise custom extensions flags to indicate neither sent nor received.
102 */
103void custom_ext_init(custom_ext_methods *exts)
104{
105    size_t i;
106    custom_ext_method *meth = exts->meths;
107
108    for (i = 0; i < exts->meths_count; i++, meth++)
109        meth->ext_flags = 0;
110}
111
112/* Pass received custom extension data to the application for parsing. */
113int custom_ext_parse(SSL *s, unsigned int context, unsigned int ext_type,
114                     const unsigned char *ext_data, size_t ext_size, X509 *x,
115                     size_t chainidx)
116{
117    int al;
118    custom_ext_methods *exts = &s->cert->custext;
119    custom_ext_method *meth;
120    ENDPOINT role = ENDPOINT_BOTH;
121
122    if ((context & (SSL_EXT_CLIENT_HELLO | SSL_EXT_TLS1_2_SERVER_HELLO)) != 0)
123        role = s->server ? ENDPOINT_SERVER : ENDPOINT_CLIENT;
124
125    meth = custom_ext_find(exts, role, ext_type, NULL);
126    /* If not found return success */
127    if (!meth)
128        return 1;
129
130    /* Check if extension is defined for our protocol. If not, skip */
131    if (!extension_is_relevant(s, meth->context, context))
132        return 1;
133
134    if ((context & (SSL_EXT_TLS1_2_SERVER_HELLO
135                    | SSL_EXT_TLS1_3_SERVER_HELLO
136                    | SSL_EXT_TLS1_3_ENCRYPTED_EXTENSIONS)) != 0) {
137        /*
138         * If it's ServerHello or EncryptedExtensions we can't have any
139         * extensions not sent in ClientHello.
140         */
141        if ((meth->ext_flags & SSL_EXT_FLAG_SENT) == 0) {
142            SSLfatal(s, TLS1_AD_UNSUPPORTED_EXTENSION, SSL_R_BAD_EXTENSION);
143            return 0;
144        }
145    }
146
147    /*
148     * Extensions received in the ClientHello or CertificateRequest are marked
149     * with the SSL_EXT_FLAG_RECEIVED. This is so we know to add the equivalent
150     * extensions in the response messages
151     */
152    if ((context & (SSL_EXT_CLIENT_HELLO | SSL_EXT_TLS1_3_CERTIFICATE_REQUEST))
153            != 0)
154        meth->ext_flags |= SSL_EXT_FLAG_RECEIVED;
155
156    /* If no parse function set return success */
157    if (!meth->parse_cb)
158        return 1;
159
160    if (meth->parse_cb(s, ext_type, context, ext_data, ext_size, x, chainidx,
161                       &al, meth->parse_arg) <= 0) {
162        SSLfatal(s, al, SSL_R_BAD_EXTENSION);
163        return 0;
164    }
165
166    return 1;
167}
168
169/*
170 * Request custom extension data from the application and add to the return
171 * buffer.
172 */
173int custom_ext_add(SSL *s, int context, WPACKET *pkt, X509 *x, size_t chainidx,
174                   int maxversion)
175{
176    custom_ext_methods *exts = &s->cert->custext;
177    custom_ext_method *meth;
178    size_t i;
179    int al;
180
181    for (i = 0; i < exts->meths_count; i++) {
182        const unsigned char *out = NULL;
183        size_t outlen = 0;
184
185        meth = exts->meths + i;
186
187        if (!should_add_extension(s, meth->context, context, maxversion))
188            continue;
189
190        if ((context & (SSL_EXT_TLS1_2_SERVER_HELLO
191                        | SSL_EXT_TLS1_3_SERVER_HELLO
192                        | SSL_EXT_TLS1_3_ENCRYPTED_EXTENSIONS
193                        | SSL_EXT_TLS1_3_CERTIFICATE
194                        | SSL_EXT_TLS1_3_HELLO_RETRY_REQUEST)) != 0) {
195            /* Only send extensions present in ClientHello/CertificateRequest */
196            if (!(meth->ext_flags & SSL_EXT_FLAG_RECEIVED))
197                continue;
198        }
199        /*
200         * We skip it if the callback is absent - except for a ClientHello where
201         * we add an empty extension.
202         */
203        if ((context & SSL_EXT_CLIENT_HELLO) == 0 && meth->add_cb == NULL)
204            continue;
205
206        if (meth->add_cb != NULL) {
207            int cb_retval = meth->add_cb(s, meth->ext_type, context, &out,
208                                         &outlen, x, chainidx, &al,
209                                         meth->add_arg);
210
211            if (cb_retval < 0) {
212                SSLfatal(s, al, SSL_R_CALLBACK_FAILED);
213                return 0;       /* error */
214            }
215            if (cb_retval == 0)
216                continue;       /* skip this extension */
217        }
218
219        if (!WPACKET_put_bytes_u16(pkt, meth->ext_type)
220                || !WPACKET_start_sub_packet_u16(pkt)
221                || (outlen > 0 && !WPACKET_memcpy(pkt, out, outlen))
222                || !WPACKET_close(pkt)) {
223            if (meth->free_cb != NULL)
224                meth->free_cb(s, meth->ext_type, context, out, meth->add_arg);
225            SSLfatal(s, SSL_AD_INTERNAL_ERROR, ERR_R_INTERNAL_ERROR);
226            return 0;
227        }
228        if ((context & SSL_EXT_CLIENT_HELLO) != 0) {
229            /*
230             * We can't send duplicates: code logic should prevent this.
231             */
232            if (!ossl_assert((meth->ext_flags & SSL_EXT_FLAG_SENT) == 0)) {
233                if (meth->free_cb != NULL)
234                    meth->free_cb(s, meth->ext_type, context, out,
235                                  meth->add_arg);
236                SSLfatal(s, SSL_AD_INTERNAL_ERROR, ERR_R_INTERNAL_ERROR);
237                return 0;
238            }
239            /*
240             * Indicate extension has been sent: this is both a sanity check to
241             * ensure we don't send duplicate extensions and indicates that it
242             * is not an error if the extension is present in ServerHello.
243             */
244            meth->ext_flags |= SSL_EXT_FLAG_SENT;
245        }
246        if (meth->free_cb != NULL)
247            meth->free_cb(s, meth->ext_type, context, out, meth->add_arg);
248    }
249    return 1;
250}
251
252/* Copy the flags from src to dst for any extensions that exist in both */
253int custom_exts_copy_flags(custom_ext_methods *dst,
254                           const custom_ext_methods *src)
255{
256    size_t i;
257    custom_ext_method *methsrc = src->meths;
258
259    for (i = 0; i < src->meths_count; i++, methsrc++) {
260        custom_ext_method *methdst = custom_ext_find(dst, methsrc->role,
261                                                     methsrc->ext_type, NULL);
262
263        if (methdst == NULL)
264            continue;
265
266        methdst->ext_flags = methsrc->ext_flags;
267    }
268
269    return 1;
270}
271
272/* Copy table of custom extensions */
273int custom_exts_copy(custom_ext_methods *dst, const custom_ext_methods *src)
274{
275    size_t i;
276    int err = 0;
277
278    if (src->meths_count > 0) {
279        dst->meths =
280            OPENSSL_memdup(src->meths,
281                           sizeof(*src->meths) * src->meths_count);
282        if (dst->meths == NULL)
283            return 0;
284        dst->meths_count = src->meths_count;
285
286        for (i = 0; i < src->meths_count; i++) {
287            custom_ext_method *methsrc = src->meths + i;
288            custom_ext_method *methdst = dst->meths + i;
289
290            if (methsrc->add_cb != custom_ext_add_old_cb_wrap)
291                continue;
292
293            /*
294             * We have found an old style API wrapper. We need to copy the
295             * arguments too.
296             */
297
298            if (err) {
299                methdst->add_arg = NULL;
300                methdst->parse_arg = NULL;
301                continue;
302            }
303
304            methdst->add_arg = OPENSSL_memdup(methsrc->add_arg,
305                                              sizeof(custom_ext_add_cb_wrap));
306            methdst->parse_arg = OPENSSL_memdup(methsrc->parse_arg,
307                                            sizeof(custom_ext_parse_cb_wrap));
308
309            if (methdst->add_arg == NULL || methdst->parse_arg == NULL)
310                err = 1;
311        }
312    }
313
314    if (err) {
315        custom_exts_free(dst);
316        return 0;
317    }
318
319    return 1;
320}
321
322void custom_exts_free(custom_ext_methods *exts)
323{
324    size_t i;
325    custom_ext_method *meth;
326
327    for (i = 0, meth = exts->meths; i < exts->meths_count; i++, meth++) {
328        if (meth->add_cb != custom_ext_add_old_cb_wrap)
329            continue;
330
331        /* Old style API wrapper. Need to free the arguments too */
332        OPENSSL_free(meth->add_arg);
333        OPENSSL_free(meth->parse_arg);
334    }
335    OPENSSL_free(exts->meths);
336}
337
338/* Return true if a client custom extension exists, false otherwise */
339int SSL_CTX_has_client_custom_ext(const SSL_CTX *ctx, unsigned int ext_type)
340{
341    return custom_ext_find(&ctx->cert->custext, ENDPOINT_CLIENT, ext_type,
342                           NULL) != NULL;
343}
344
345static int add_custom_ext_intern(SSL_CTX *ctx, ENDPOINT role,
346                                 unsigned int ext_type,
347                                 unsigned int context,
348                                 SSL_custom_ext_add_cb_ex add_cb,
349                                 SSL_custom_ext_free_cb_ex free_cb,
350                                 void *add_arg,
351                                 SSL_custom_ext_parse_cb_ex parse_cb,
352                                 void *parse_arg)
353{
354    custom_ext_methods *exts = &ctx->cert->custext;
355    custom_ext_method *meth, *tmp;
356
357    /*
358     * Check application error: if add_cb is not set free_cb will never be
359     * called.
360     */
361    if (add_cb == NULL && free_cb != NULL)
362        return 0;
363
364#ifndef OPENSSL_NO_CT
365    /*
366     * We don't want applications registering callbacks for SCT extensions
367     * whilst simultaneously using the built-in SCT validation features, as
368     * these two things may not play well together.
369     */
370    if (ext_type == TLSEXT_TYPE_signed_certificate_timestamp
371            && (context & SSL_EXT_CLIENT_HELLO) != 0
372            && SSL_CTX_ct_is_enabled(ctx))
373        return 0;
374#endif
375
376    /*
377     * Don't add if extension supported internally, but make exception
378     * for extension types that previously were not supported, but now are.
379     */
380    if (SSL_extension_supported(ext_type)
381            && ext_type != TLSEXT_TYPE_signed_certificate_timestamp)
382        return 0;
383
384    /* Extension type must fit in 16 bits */
385    if (ext_type > 0xffff)
386        return 0;
387    /* Search for duplicate */
388    if (custom_ext_find(exts, role, ext_type, NULL))
389        return 0;
390    tmp = OPENSSL_realloc(exts->meths,
391                          (exts->meths_count + 1) * sizeof(custom_ext_method));
392    if (tmp == NULL)
393        return 0;
394
395    exts->meths = tmp;
396    meth = exts->meths + exts->meths_count;
397    memset(meth, 0, sizeof(*meth));
398    meth->role = role;
399    meth->context = context;
400    meth->parse_cb = parse_cb;
401    meth->add_cb = add_cb;
402    meth->free_cb = free_cb;
403    meth->ext_type = ext_type;
404    meth->add_arg = add_arg;
405    meth->parse_arg = parse_arg;
406    exts->meths_count++;
407    return 1;
408}
409
410static int add_old_custom_ext(SSL_CTX *ctx, ENDPOINT role,
411                              unsigned int ext_type,
412                              unsigned int context,
413                              custom_ext_add_cb add_cb,
414                              custom_ext_free_cb free_cb,
415                              void *add_arg,
416                              custom_ext_parse_cb parse_cb, void *parse_arg)
417{
418    custom_ext_add_cb_wrap *add_cb_wrap
419        = OPENSSL_malloc(sizeof(*add_cb_wrap));
420    custom_ext_parse_cb_wrap *parse_cb_wrap
421        = OPENSSL_malloc(sizeof(*parse_cb_wrap));
422    int ret;
423
424    if (add_cb_wrap == NULL || parse_cb_wrap == NULL) {
425        OPENSSL_free(add_cb_wrap);
426        OPENSSL_free(parse_cb_wrap);
427        return 0;
428    }
429
430    add_cb_wrap->add_arg = add_arg;
431    add_cb_wrap->add_cb = add_cb;
432    add_cb_wrap->free_cb = free_cb;
433    parse_cb_wrap->parse_arg = parse_arg;
434    parse_cb_wrap->parse_cb = parse_cb;
435
436    ret = add_custom_ext_intern(ctx, role, ext_type,
437                                context,
438                                custom_ext_add_old_cb_wrap,
439                                custom_ext_free_old_cb_wrap,
440                                add_cb_wrap,
441                                custom_ext_parse_old_cb_wrap,
442                                parse_cb_wrap);
443
444    if (!ret) {
445        OPENSSL_free(add_cb_wrap);
446        OPENSSL_free(parse_cb_wrap);
447    }
448
449    return ret;
450}
451
452/* Application level functions to add the old custom extension callbacks */
453int SSL_CTX_add_client_custom_ext(SSL_CTX *ctx, unsigned int ext_type,
454                                  custom_ext_add_cb add_cb,
455                                  custom_ext_free_cb free_cb,
456                                  void *add_arg,
457                                  custom_ext_parse_cb parse_cb, void *parse_arg)
458{
459    return add_old_custom_ext(ctx, ENDPOINT_CLIENT, ext_type,
460                              SSL_EXT_TLS1_2_AND_BELOW_ONLY
461                              | SSL_EXT_CLIENT_HELLO
462                              | SSL_EXT_TLS1_2_SERVER_HELLO
463                              | SSL_EXT_IGNORE_ON_RESUMPTION,
464                              add_cb, free_cb, add_arg, parse_cb, parse_arg);
465}
466
467int SSL_CTX_add_server_custom_ext(SSL_CTX *ctx, unsigned int ext_type,
468                                  custom_ext_add_cb add_cb,
469                                  custom_ext_free_cb free_cb,
470                                  void *add_arg,
471                                  custom_ext_parse_cb parse_cb, void *parse_arg)
472{
473    return add_old_custom_ext(ctx, ENDPOINT_SERVER, ext_type,
474                              SSL_EXT_TLS1_2_AND_BELOW_ONLY
475                              | SSL_EXT_CLIENT_HELLO
476                              | SSL_EXT_TLS1_2_SERVER_HELLO
477                              | SSL_EXT_IGNORE_ON_RESUMPTION,
478                              add_cb, free_cb, add_arg, parse_cb, parse_arg);
479}
480
481int SSL_CTX_add_custom_ext(SSL_CTX *ctx, unsigned int ext_type,
482                           unsigned int context,
483                           SSL_custom_ext_add_cb_ex add_cb,
484                           SSL_custom_ext_free_cb_ex free_cb,
485                           void *add_arg,
486                           SSL_custom_ext_parse_cb_ex parse_cb, void *parse_arg)
487{
488    return add_custom_ext_intern(ctx, ENDPOINT_BOTH, ext_type, context, add_cb,
489                                 free_cb, add_arg, parse_cb, parse_arg);
490}
491
492int SSL_extension_supported(unsigned int ext_type)
493{
494    switch (ext_type) {
495        /* Internally supported extensions. */
496    case TLSEXT_TYPE_application_layer_protocol_negotiation:
497    case TLSEXT_TYPE_ec_point_formats:
498    case TLSEXT_TYPE_supported_groups:
499    case TLSEXT_TYPE_key_share:
500#ifndef OPENSSL_NO_NEXTPROTONEG
501    case TLSEXT_TYPE_next_proto_neg:
502#endif
503    case TLSEXT_TYPE_padding:
504    case TLSEXT_TYPE_renegotiate:
505    case TLSEXT_TYPE_max_fragment_length:
506    case TLSEXT_TYPE_server_name:
507    case TLSEXT_TYPE_session_ticket:
508    case TLSEXT_TYPE_signature_algorithms:
509#ifndef OPENSSL_NO_SRP
510    case TLSEXT_TYPE_srp:
511#endif
512#ifndef OPENSSL_NO_OCSP
513    case TLSEXT_TYPE_status_request:
514#endif
515#ifndef OPENSSL_NO_CT
516    case TLSEXT_TYPE_signed_certificate_timestamp:
517#endif
518#ifndef OPENSSL_NO_SRTP
519    case TLSEXT_TYPE_use_srtp:
520#endif
521    case TLSEXT_TYPE_encrypt_then_mac:
522    case TLSEXT_TYPE_supported_versions:
523    case TLSEXT_TYPE_extended_master_secret:
524    case TLSEXT_TYPE_psk_kex_modes:
525    case TLSEXT_TYPE_cookie:
526    case TLSEXT_TYPE_early_data:
527    case TLSEXT_TYPE_certificate_authorities:
528    case TLSEXT_TYPE_psk:
529    case TLSEXT_TYPE_post_handshake_auth:
530        return 1;
531    default:
532        return 0;
533    }
534}
535