ssl/quic/quic_impl.c

   1 /*
   2  * Copyright 2022 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 #include <openssl/macros.h>
  11 #include <openssl/objects.h>
  12 #include <openssl/sslerr.h>
  13 #include <crypto/rand.h>
  14 #include "quic_local.h"
  15 #include "internal/quic_dummy_handshake.h"
  16 #include "internal/quic_rx_depack.h"
  17 #include "internal/quic_error.h"
  18 #include "internal/time.h"
  19
  20 static void aon_write_finish(QUIC_CONNECTION *qc);
  21
  22 /*
  23  * QUIC Front-End I/O API: Common Utilities
  24  * ========================================
  25  */
  26
  27 /*
  28  * Block until a predicate is met.
  29  *
  30  * Precondition: Must have a channel.
  31  */
  32 static int block_until_pred(QUIC_CONNECTION *qc,
  33                             int (*pred)(void *arg), void *pred_arg,
  34                             uint32_t flags)
  35 {
  36     QUIC_REACTOR *rtor;
  37
  38     assert(qc->ch != NULL);
  39
  40     rtor = ossl_quic_channel_get_reactor(qc->ch);
  41     return ossl_quic_reactor_block_until_pred(rtor, pred, pred_arg, flags);
  42 }
  43
  44 /*
  45  * Raise a 'normal' error, meaning one that can be reported via SSL_get_error()
  46  * rather than via ERR.
  47  */
  48 static int quic_raise_normal_error(QUIC_CONNECTION *qc,
  49                                    int err)
  50 {
  51     qc->last_error = err;
  52     return 0;
  53 }
  54
  55 /*
  56  * Raise a 'non-normal' error, meaning any error that is not reported via
  57  * SSL_get_error() and must be reported via ERR.
  58  */
  59 static int quic_raise_non_normal_error(QUIC_CONNECTION *qc,
  60                                        const char *file,
  61                                        int line,
  62                                        const char *func,
  63                                        int reason,
  64                                        const char *fmt,
  65                                        ...)
  66 {
  67     va_list args;
  68
  69     ERR_new();
  70     ERR_set_debug(file, line, func);
  71
  72     va_start(args, fmt);
  73     ERR_vset_error(ERR_LIB_SSL, reason, fmt, args);
  74     va_end(args);
  75
  76     qc->last_error = SSL_ERROR_SSL;
  77     return 0;
  78 }
  79
  80 #define QUIC_RAISE_NORMAL_ERROR(qc, err)                        \
  81     quic_raise_normal_error((qc), (err))
  82
  83 #define QUIC_RAISE_NON_NORMAL_ERROR(qc, reason, msg)            \
  84     quic_raise_non_normal_error((qc),                           \
  85                                 OPENSSL_FILE, OPENSSL_LINE,     \
  86                                 OPENSSL_FUNC,                   \
  87                                 (reason),                       \
  88                                 (msg))
  89
  90 /*
  91  * Should be called at entry of every public function to confirm we have a valid
  92  * QUIC_CONNECTION.
  93  */
  94 static ossl_inline int expect_quic_conn(const QUIC_CONNECTION *qc)
  95 {
  96     if (!ossl_assert(qc != NULL))
  97         return QUIC_RAISE_NON_NORMAL_ERROR(NULL, ERR_R_INTERNAL_ERROR, NULL);
  98
  99     return 1;
 100
 101 }
 102
 103 /*
 104  * QUIC Front-End I/O API: Initialization
 105  * ======================================
 106  *
 107  *         SSL_new                  => ossl_quic_new
 108  *                                     ossl_quic_init
 109  *         SSL_reset                => ossl_quic_reset
 110  *         SSL_clear                => ossl_quic_clear
 111  *                                     ossl_quic_deinit
 112  *         SSL_free                 => ossl_quic_free
 113  *
 114  */
 115
 116 /* SSL_new */
 117 SSL *ossl_quic_new(SSL_CTX *ctx)
 118 {
 119     QUIC_CONNECTION *qc = NULL;
 120     SSL *ssl_base = NULL;
 121
 122     qc = OPENSSL_zalloc(sizeof(*qc));
 123     if (qc == NULL)
 124         goto err;
 125
 126     /* Initialise the QUIC_CONNECTION's stub header. */
 127     ssl_base = &qc->ssl;
 128     if (!ossl_ssl_init(ssl_base, ctx, SSL_TYPE_QUIC_CONNECTION)) {
 129         ssl_base = NULL;
 130         goto err;
 131     }
 132
 133     /* Channel is not created yet. */
 134     qc->ssl_mode   = qc->ssl.ctx->mode;
 135     qc->last_error = SSL_ERROR_NONE;
 136     qc->blocking   = 1;
 137     return ssl_base;
 138
 139 err:
 140     OPENSSL_free(qc);
 141     return NULL;
 142 }
 143
 144 /* SSL_free */
 145 void ossl_quic_free(SSL *s)
 146 {
 147     QUIC_CONNECTION *qc = QUIC_CONNECTION_FROM_SSL(s);
 148
 149     /* We should never be called on anything but a QUIC_CONNECTION. */
 150     if (!expect_quic_conn(qc))
 151         return;
 152
 153     ossl_quic_channel_free(qc->ch);
 154
 155     BIO_free(qc->net_rbio);
 156     BIO_free(qc->net_wbio);
 157
 158     /* Note: SSL_free calls OPENSSL_free(qc) for us */
 159 }
 160
 161 /* SSL method init */
 162 int ossl_quic_init(SSL *s)
 163 {
 164     QUIC_CONNECTION *qc = QUIC_CONNECTION_FROM_SSL(s);
 165
 166     if (!expect_quic_conn(qc))
 167         return 0;
 168
 169     /* Same op as SSL_clear, forward the call. */
 170     return ossl_quic_clear(s);
 171 }
 172
 173 /* SSL method deinit */
 174 void ossl_quic_deinit(SSL *s)
 175 {
 176     /* No-op. */
 177 }
 178
 179 /* SSL_reset */
 180 int ossl_quic_reset(SSL *s)
 181 {
 182     QUIC_CONNECTION *qc = QUIC_CONNECTION_FROM_SSL(s);
 183
 184     if (!expect_quic_conn(qc))
 185         return 0;
 186
 187     /* TODO(QUIC); Currently a no-op. */
 188     return 1;
 189 }
 190
 191 /* SSL_clear */
 192 int ossl_quic_clear(SSL *s)
 193 {
 194     QUIC_CONNECTION *qc = QUIC_CONNECTION_FROM_SSL(s);
 195
 196     if (!expect_quic_conn(qc))
 197         return 0;
 198
 199     /* TODO(QUIC): Currently a no-op. */
 200     return 1;
 201 }
 202
 203 /*
 204  * QUIC Front-End I/O API: Network BIO Configuration
 205  * =================================================
 206  *
 207  * Handling the different BIOs is difficult:
 208  *
 209  *   - It is more or less a requirement that we use non-blocking network I/O;
 210  *     we need to be able to have timeouts on recv() calls, and make best effort
 211  *     (non blocking) send() and recv() calls.
 212  *
 213  *     The only sensible way to do this is to configure the socket into
 214  *     non-blocking mode. We could try to do select() before calling send() or
 215  *     recv() to get a guarantee that the call will not block, but this will
 216  *     probably run into issues with buggy OSes which generate spurious socket
 217  *     readiness events. In any case, relying on this to work reliably does not
 218  *     seem sane.
 219  *
 220  *     Timeouts could be handled via setsockopt() socket timeout options, but
 221  *     this depends on OS support and adds another syscall to every network I/O
 222  *     operation. It also has obvious thread safety concerns if we want to move
 223  *     to concurrent use of a single socket at some later date.
 224  *
 225  *     Some OSes support a MSG_DONTWAIT flag which allows a single I/O option to
 226  *     be made non-blocking. However some OSes (e.g. Windows) do not support
 227  *     this, so we cannot rely on this.
 228  *
 229  *     As such, we need to configure any FD in non-blocking mode. This may
 230  *     confound users who pass a blocking socket to libssl. However, in practice
 231  *     it would be extremely strange for a user of QUIC to pass an FD to us,
 232  *     then also try and send receive traffic on the same socket(!). Thus the
 233  *     impact of this should be limited, and can be documented.
 234  *
 235  *   - We support both blocking and non-blocking operation in terms of the API
 236  *     presented to the user. One prospect is to set the blocking mode based on
 237  *     whether the socket passed to us was already in blocking mode. However,
 238  *     Windows has no API for determining if a socket is in blocking mode (!),
 239  *     therefore this cannot be done portably. Currently therefore we expose an
 240  *     explicit API call to set this, and default to blocking mode.
 241  *
 242  *   - We need to determine our initial destination UDP address. The "natural"
 243  *     way for a user to do this is to set the peer variable on a BIO_dgram.
 244  *     However, this has problems because BIO_dgram's peer variable is used for
 245  *     both transmission and reception. This means it can be constantly being
 246  *     changed to a malicious value (e.g. if some random unrelated entity on the
 247  *     network starts sending traffic to us) on every read call. This is not a
 248  *     direct issue because we use the 'stateless' BIO_sendmmsg and BIO_recvmmsg
 249  *     calls only, which do not use this variable. However, we do need to let
 250  *     the user specify the peer in a 'normal' manner. The compromise here is
 251  *     that we grab the current peer value set at the time the write BIO is set
 252  *     and do not read the value again.
 253  *
 254  *   - We also need to support memory BIOs (e.g. BIO_dgram_pair) or custom BIOs.
 255  *     Currently we do this by only supporting non-blocking mode.
 256  *
 257  */
 258
 259 /*
 260  * Determines what initial destination UDP address we should use, if possible.
 261  * If this fails the client must set the destination address manually, or use a
 262  * BIO which does not need a destination address.
 263  */
 264 static int csm_analyse_init_peer_addr(BIO *net_wbio, BIO_ADDR *peer)
 265 {
 266     if (!BIO_dgram_get_peer(net_wbio, peer))
 267         return 0;
 268
 269     return 1;
 270 }
 271
 272 void ossl_quic_conn_set0_net_rbio(QUIC_CONNECTION *qc, BIO *net_rbio)
 273 {
 274     if (qc->net_rbio == net_rbio)
 275         return;
 276
 277     if (qc->ch != NULL && !ossl_quic_channel_set_net_rbio(qc->ch, net_rbio))
 278         return;
 279
 280     BIO_free(qc->net_rbio);
 281     qc->net_rbio = net_rbio;
 282
 283     /*
 284      * If what we have is not pollable (e.g. a BIO_dgram_pair) disable blocking
 285      * mode as we do not support it for non-pollable BIOs.
 286      */
 287     if (net_rbio != NULL) {
 288         BIO_POLL_DESCRIPTOR d = {0};
 289
 290         if (!BIO_get_rpoll_descriptor(net_rbio, &d)
 291             || d.type != BIO_POLL_DESCRIPTOR_TYPE_SOCK_FD) {
 292             qc->blocking = 0;
 293             qc->can_poll_net_rbio = 0;
 294         } else {
 295             qc->can_poll_net_rbio = 1;
 296         }
 297     }
 298 }
 299
 300 void ossl_quic_conn_set0_net_wbio(QUIC_CONNECTION *qc, BIO *net_wbio)
 301 {
 302     if (qc->net_wbio == net_wbio)
 303         return;
 304
 305     if (qc->ch != NULL && !ossl_quic_channel_set_net_wbio(qc->ch, net_wbio))
 306         return;
 307
 308     BIO_free(qc->net_wbio);
 309     qc->net_wbio = net_wbio;
 310
 311     if (net_wbio != NULL) {
 312         BIO_POLL_DESCRIPTOR d = {0};
 313
 314         if (!BIO_get_wpoll_descriptor(net_wbio, &d)
 315             || d.type != BIO_POLL_DESCRIPTOR_TYPE_SOCK_FD) {
 316             qc->blocking = 0;
 317             qc->can_poll_net_wbio = 0;
 318         } else {
 319             qc->can_poll_net_wbio = 1;
 320         }
 321
 322         /*
 323          * If we do not have a peer address yet, and we have not started trying
 324          * to connect yet, try to autodetect one.
 325          */
 326         if (BIO_ADDR_family(&qc->init_peer_addr) == AF_UNSPEC
 327             && !qc->started) {
 328             if (!csm_analyse_init_peer_addr(net_wbio, &qc->init_peer_addr))
 329                 /* best effort */
 330                 BIO_ADDR_clear(&qc->init_peer_addr);
 331
 332             if (qc->ch != NULL)
 333                 ossl_quic_channel_set_peer_addr(qc->ch, &qc->init_peer_addr);
 334         }
 335     }
 336 }
 337
 338 BIO *ossl_quic_conn_get_net_rbio(const QUIC_CONNECTION *qc)
 339 {
 340     return qc->net_rbio;
 341 }
 342
 343 BIO *ossl_quic_conn_get_net_wbio(const QUIC_CONNECTION *qc)
 344 {
 345     return qc->net_wbio;
 346 }
 347
 348 int ossl_quic_conn_get_blocking_mode(const QUIC_CONNECTION *qc)
 349 {
 350     return qc->blocking;
 351 }
 352
 353 int ossl_quic_conn_set_blocking_mode(QUIC_CONNECTION *qc, int blocking)
 354 {
 355     /* Cannot enable blocking mode if we do not have pollable FDs. */
 356     if (blocking != 0 &&
 357         (!qc->can_poll_net_rbio || !qc->can_poll_net_wbio))
 358         return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_UNSUPPORTED, NULL);
 359
 360     qc->blocking = (blocking != 0);
 361     return 1;
 362 }
 363
 364 int ossl_quic_conn_set_initial_peer_addr(QUIC_CONNECTION *qc,
 365                                          const BIO_ADDR *peer_addr)
 366 {
 367     if (qc->started)
 368         return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_SHOULD_NOT_HAVE_BEEN_CALLED,
 369                                            NULL);
 370
 371     if (peer_addr == NULL) {
 372         BIO_ADDR_clear(&qc->init_peer_addr);
 373         return 1;
 374     }
 375
 376     qc->init_peer_addr = *peer_addr;
 377     return 1;
 378 }
 379
 380 /*
 381  * QUIC Front-End I/O API: Asynchronous I/O Management
 382  * ===================================================
 383  *
 384  *   (BIO/)SSL_tick                 => ossl_quic_tick
 385  *   (BIO/)SSL_get_tick_timeout     => ossl_quic_get_tick_timeout
 386  *   (BIO/)SSL_get_poll_fd          => ossl_quic_get_poll_fd
 387  *
 388  */
 389
 390 /* Returns 1 if the connection is being used in blocking mode. */
 391 static int blocking_mode(const QUIC_CONNECTION *qc)
 392 {
 393     return qc->blocking;
 394 }
 395
 396 /* SSL_tick; ticks the reactor. */
 397 int ossl_quic_tick(QUIC_CONNECTION *qc)
 398 {
 399     if (qc->ch == NULL)
 400         return 1;
 401
 402     ossl_quic_reactor_tick(ossl_quic_channel_get_reactor(qc->ch));
 403     return 1;
 404 }
 405
 406 /*
 407  * SSL_get_tick_timeout. Get the time in milliseconds until the SSL object
 408  * should be ticked by the application by calling SSL_tick(). tv is set to 0 if
 409  * the object should be ticked immediately and tv->tv_sec is set to -1 if no
 410  * timeout is currently active.
 411  */
 412 int ossl_quic_get_tick_timeout(QUIC_CONNECTION *qc, struct timeval *tv)
 413 {
 414     OSSL_TIME deadline = ossl_time_infinite();
 415
 416     if (qc->ch != NULL)
 417         deadline
 418             = ossl_quic_reactor_get_tick_deadline(ossl_quic_channel_get_reactor(qc->ch));
 419
 420     if (ossl_time_is_infinite(deadline)) {
 421         tv->tv_sec  = -1;
 422         tv->tv_usec = 0;
 423         return 1;
 424     }
 425
 426     *tv = ossl_time_to_timeval(ossl_time_subtract(deadline, ossl_time_now()));
 427     return 1;
 428 }
 429
 430 /* SSL_get_rpoll_descriptor */
 431 int ossl_quic_get_rpoll_descriptor(QUIC_CONNECTION *qc, BIO_POLL_DESCRIPTOR *desc)
 432 {
 433     if (desc == NULL || qc->net_rbio == NULL)
 434         return 0;
 435
 436     return BIO_get_rpoll_descriptor(qc->net_rbio, desc);
 437 }
 438
 439 /* SSL_get_wpoll_descriptor */
 440 int ossl_quic_get_wpoll_descriptor(QUIC_CONNECTION *qc, BIO_POLL_DESCRIPTOR *desc)
 441 {
 442     if (desc == NULL || qc->net_wbio == NULL)
 443         return 0;
 444
 445     return BIO_get_wpoll_descriptor(qc->net_wbio, desc);
 446 }
 447
 448 /* SSL_want_net_read */
 449 int ossl_quic_get_want_net_read(QUIC_CONNECTION *qc)
 450 {
 451     if (qc->ch == NULL)
 452         return 0;
 453
 454     return ossl_quic_reactor_want_net_read(ossl_quic_channel_get_reactor(qc->ch));
 455 }
 456
 457 /* SSL_want_net_write */
 458 int ossl_quic_get_want_net_write(QUIC_CONNECTION *qc)
 459 {
 460     if (qc->ch == NULL)
 461         return 0;
 462
 463     return ossl_quic_reactor_want_net_write(ossl_quic_channel_get_reactor(qc->ch));
 464 }
 465
 466 /*
 467  * QUIC Front-End I/O API: Connection Lifecycle Operations
 468  * =======================================================
 469  *
 470  *         SSL_do_handshake         => ossl_quic_do_handshake
 471  *         SSL_set_connect_state    => ossl_quic_set_connect_state
 472  *         SSL_set_accept_state     => ossl_quic_set_accept_state
 473  *         SSL_shutdown             => ossl_quic_shutdown
 474  *         SSL_ctrl                 => ossl_quic_ctrl
 475  *   (BIO/)SSL_connect              => ossl_quic_connect
 476  *   (BIO/)SSL_accept               => ossl_quic_accept
 477  *
 478  */
 479
 480 /* SSL_shutdown */
 481 int ossl_quic_shutdown(SSL *s)
 482 {
 483     QUIC_CONNECTION *qc = QUIC_CONNECTION_FROM_SSL(s);
 484
 485     if (!expect_quic_conn(qc))
 486         return 0;
 487
 488     if (qc->ch != NULL)
 489         ossl_quic_channel_local_close(qc->ch);
 490
 491     return 1;
 492 }
 493
 494 /* SSL_ctrl */
 495 static void fixup_mode_change(QUIC_CONNECTION *qc)
 496 {
 497     /* If enabling EPW mode, cancel any AON write */
 498     if ((qc->ssl_mode & SSL_MODE_ENABLE_PARTIAL_WRITE) != 0)
 499         aon_write_finish(qc);
 500 }
 501
 502 long ossl_quic_ctrl(SSL *s, int cmd, long larg, void *parg)
 503 {
 504     QUIC_CONNECTION *qc = QUIC_CONNECTION_FROM_SSL(s);
 505
 506     if (!expect_quic_conn(qc))
 507         return 0;
 508
 509     switch (cmd) {
 510     case SSL_CTRL_MODE:
 511         qc->ssl_mode |= (uint32_t)larg;
 512         fixup_mode_change(qc);
 513         return qc->ssl_mode;
 514     case SSL_CTRL_CLEAR_MODE:
 515         qc->ssl_mode &= ~(uint32_t)larg;
 516         fixup_mode_change(qc);
 517         return qc->ssl_mode;
 518     default:
 519         return 0;
 520     }
 521 }
 522
 523 /* SSL_set_connect_state */
 524 void ossl_quic_set_connect_state(QUIC_CONNECTION *qc)
 525 {
 526     /* Cannot be changed after handshake started */
 527     if (qc->started)
 528         return;
 529
 530     qc->as_server = 0;
 531 }
 532
 533 /* SSL_set_accept_state */
 534 void ossl_quic_set_accept_state(QUIC_CONNECTION *qc)
 535 {
 536     /* Cannot be changed after handshake started */
 537     if (qc->started)
 538         return;
 539
 540     qc->as_server = 1;
 541 }
 542
 543 /* SSL_do_handshake */
 544 struct quic_handshake_wait_args {
 545     QUIC_CONNECTION     *qc;
 546 };
 547
 548 static int quic_handshake_wait(void *arg)
 549 {
 550     struct quic_handshake_wait_args *args = arg;
 551
 552     if (!ossl_quic_channel_is_active(args->qc->ch))
 553         return -1;
 554
 555     if (ossl_quic_channel_is_handshake_complete(args->qc->ch))
 556         return 1;
 557
 558     return 0;
 559 }
 560
 561 static int configure_channel(QUIC_CONNECTION *qc)
 562 {
 563     assert(qc->ch != NULL);
 564
 565     if (!ossl_quic_channel_set_net_rbio(qc->ch, qc->net_rbio)
 566         || !ossl_quic_channel_set_net_wbio(qc->ch, qc->net_wbio)
 567         || !ossl_quic_channel_set_peer_addr(qc->ch, &qc->init_peer_addr))
 568         return 0;
 569
 570     return 1;
 571 }
 572
 573 /*
 574  * Creates a channel and configures it with the information we have accumulated
 575  * via calls made to us from the application prior to starting a handshake
 576  * attempt.
 577  */
 578 static int ensure_channel_and_start(QUIC_CONNECTION *qc)
 579 {
 580     QUIC_CHANNEL_ARGS args = {0};
 581
 582     if (qc->ch != NULL)
 583         return 1;
 584
 585     args.libctx     = qc->ssl.ctx->libctx;
 586     args.propq      = qc->ssl.ctx->propq;
 587     args.is_server  = 0;
 588
 589     qc->ch = ossl_quic_channel_new(&args);
 590     if (qc->ch == NULL)
 591         return 0;
 592
 593     if (!configure_channel(qc)
 594         || !ossl_quic_channel_start(qc->ch)) {
 595         ossl_quic_channel_free(qc->ch);
 596         qc->ch = NULL;
 597         return 0;
 598     }
 599
 600     qc->stream0 = ossl_quic_channel_get_stream_by_id(qc->ch, 0);
 601     if (qc->stream0 == NULL) {
 602         ossl_quic_channel_free(qc->ch);
 603         qc->ch = NULL;
 604         return 0;
 605     }
 606
 607     qc->started = 1;
 608     return 1;
 609 }
 610
 611 int ossl_quic_do_handshake(QUIC_CONNECTION *qc)
 612 {
 613     int ret;
 614
 615     if (qc->ch != NULL && ossl_quic_channel_is_handshake_complete(qc->ch))
 616         /* Handshake already completed. */
 617         return 1;
 618
 619     if (qc->ch != NULL && ossl_quic_channel_is_term_any(qc->ch))
 620         return QUIC_RAISE_NON_NORMAL_ERROR(qc, SSL_R_PROTOCOL_IS_SHUTDOWN, NULL);
 621
 622     if (BIO_ADDR_family(&qc->init_peer_addr) == AF_UNSPEC) {
 623         /* Peer address must have been set. */
 624         QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_PASSED_INVALID_ARGUMENT, NULL);
 625         return -1; /* Non-protocol error */
 626     }
 627
 628     if (qc->as_server) {
 629         /* TODO(QUIC): Server mode not currently supported */
 630         QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_PASSED_INVALID_ARGUMENT, NULL);
 631         return -1; /* Non-protocol error */
 632     }
 633
 634     if (qc->net_rbio == NULL || qc->net_wbio == NULL) {
 635         /* Need read and write BIOs. */
 636         QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_PASSED_INVALID_ARGUMENT, NULL);
 637         return -1; /* Non-protocol error */
 638     }
 639
 640     /*
 641      * Start connection process. Note we may come here multiple times in
 642      * non-blocking mode, which is fine.
 643      */
 644     if (!ensure_channel_and_start(qc)) {
 645         QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
 646         return -1; /* Non-protocol error */
 647     }
 648
 649     if (ossl_quic_channel_is_handshake_complete(qc->ch))
 650         /* The handshake is now done. */
 651         return 1;
 652
 653     if (blocking_mode(qc)) {
 654         /* In blocking mode, wait for the handshake to complete. */
 655         struct quic_handshake_wait_args args;
 656
 657         args.qc     = qc;
 658
 659         ret = block_until_pred(qc, quic_handshake_wait, &args, 0);
 660         if (!ossl_quic_channel_is_active(qc->ch)) {
 661             QUIC_RAISE_NON_NORMAL_ERROR(qc, SSL_R_PROTOCOL_IS_SHUTDOWN, NULL);
 662             return 0; /* Shutdown before completion */
 663         } else if (ret <= 0) {
 664             QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
 665             return -1; /* Non-protocol error */
 666         }
 667
 668         assert(ossl_quic_channel_is_handshake_complete(qc->ch));
 669         return 1;
 670     } else {
 671         /* Try to advance the reactor. */
 672         ossl_quic_reactor_tick(ossl_quic_channel_get_reactor(qc->ch));
 673
 674         if (ossl_quic_channel_is_handshake_complete(qc->ch))
 675             /* The handshake is now done. */
 676             return 1;
 677
 678         /* Otherwise, indicate that the handshake isn't done yet. */
 679         QUIC_RAISE_NORMAL_ERROR(qc, SSL_ERROR_WANT_READ);
 680         return -1; /* Non-protocol error */
 681     }
 682 }
 683
 684 /* SSL_connect */
 685 int ossl_quic_connect(SSL *s)
 686 {
 687     QUIC_CONNECTION *qc = QUIC_CONNECTION_FROM_SSL(s);
 688
 689     if (!expect_quic_conn(qc))
 690         return 0;
 691
 692     /* Ensure we are in connect state (no-op if non-idle). */
 693     ossl_quic_set_connect_state(qc);
 694
 695     /* Begin or continue the handshake */
 696     return ossl_quic_do_handshake(qc);
 697 }
 698
 699 /* SSL_accept */
 700 int ossl_quic_accept(SSL *s)
 701 {
 702     QUIC_CONNECTION *qc = QUIC_CONNECTION_FROM_SSL(s);
 703
 704     if (!expect_quic_conn(qc))
 705         return 0;
 706
 707     /* Ensure we are in accept state (no-op if non-idle). */
 708     ossl_quic_set_accept_state(qc);
 709
 710     /* Begin or continue the handshake */
 711     return ossl_quic_do_handshake(qc);
 712 }
 713
 714 /*
 715  * QUIC Front-End I/O API: Steady-State Operations
 716  * ===============================================
 717  *
 718  * Here we dispatch calls to the steady-state front-end I/O API functions; that
 719  * is, the functions used during the established phase of a QUIC connection
 720  * (e.g. SSL_read, SSL_write).
 721  *
 722  * Each function must handle both blocking and non-blocking modes. As discussed
 723  * above, all QUIC I/O is implemented using non-blocking mode internally.
 724  *
 725  *         SSL_get_error        => partially implemented by ossl_quic_get_error
 726  *   (BIO/)SSL_read             => ossl_quic_read
 727  *   (BIO/)SSL_write            => ossl_quic_write
 728  *         SSL_pending          => ossl_quic_pending
 729  */
 730
 731 /* SSL_get_error */
 732 int ossl_quic_get_error(const QUIC_CONNECTION *qc, int i)
 733 {
 734     return qc->last_error;
 735 }
 736
 737 /*
 738  * SSL_write
 739  * ---------
 740  *
 741  * The set of functions below provide the implementation of the public SSL_write
 742  * function. We must handle:
 743  *
 744  *   - both blocking and non-blocking operation at the application level,
 745  *     depending on how we are configured;
 746  *
 747  *   - SSL_MODE_ENABLE_PARTIAL_WRITE being on or off;
 748  *
 749  *   - SSL_MODE_ACCEPT_MOVING_WRITE_BUFFER.
 750  *
 751  */
 752 static void quic_post_write(QUIC_CONNECTION *qc, int did_append, int do_tick)
 753 {
 754     /*
 755      * We have appended at least one byte to the stream.
 756      * Potentially mark stream as active, depending on FC.
 757      */
 758     if (did_append)
 759         ossl_quic_stream_map_update_state(ossl_quic_channel_get_qsm(qc->ch),
 760                                           qc->stream0);
 761
 762     /*
 763      * Try and send.
 764      *
 765      * TODO(QUIC): It is probably inefficient to try and do this immediately,
 766      * plus we should eventually consider Nagle's algorithm.
 767      */
 768     if (do_tick)
 769         ossl_quic_reactor_tick(ossl_quic_channel_get_reactor(qc->ch));
 770 }
 771
 772 struct quic_write_again_args {
 773     QUIC_CONNECTION     *qc;
 774     const unsigned char *buf;
 775     size_t              len;
 776     size_t              total_written;
 777 };
 778
 779 static int quic_write_again(void *arg)
 780 {
 781     struct quic_write_again_args *args = arg;
 782     size_t actual_written = 0;
 783
 784     if (!ossl_quic_channel_is_active(args->qc->ch))
 785         /* If connection is torn down due to an error while blocking, stop. */
 786         return -2;
 787
 788     if (!ossl_quic_sstream_append(args->qc->stream0->sstream,
 789                                   args->buf, args->len, &actual_written))
 790         return -2;
 791
 792     quic_post_write(args->qc, actual_written > 0, 0);
 793
 794     args->buf           += actual_written;
 795     args->len           -= actual_written;
 796     args->total_written += actual_written;
 797
 798     if (actual_written == 0)
 799         /* Written everything, done. */
 800         return 1;
 801
 802     /* Not written everything yet, keep trying. */
 803     return 0;
 804 }
 805
 806 static int quic_write_blocking(QUIC_CONNECTION *qc, const void *buf, size_t len,
 807                                size_t *written)
 808 {
 809     int res;
 810     struct quic_write_again_args args;
 811     size_t actual_written = 0;
 812
 813     /* First make a best effort to append as much of the data as possible. */
 814     if (!ossl_quic_sstream_append(qc->stream0->sstream, buf, len,
 815                                   &actual_written)) {
 816         /* Stream already finished or allocation error. */
 817         *written = 0;
 818         return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
 819     }
 820
 821     quic_post_write(qc, actual_written > 0, 1);
 822
 823     if (actual_written == len) {
 824         /* Managed to append everything on the first try. */
 825         *written = actual_written;
 826         return 1;
 827     }
 828
 829     /*
 830      * We did not manage to append all of the data immediately, so the stream
 831      * buffer has probably filled up. This means we need to block until some of
 832      * it is freed up.
 833      */
 834     args.qc             = qc;
 835     args.buf            = (const unsigned char *)buf + actual_written;
 836     args.len            = len - actual_written;
 837     args.total_written  = 0;
 838
 839     res = block_until_pred(qc, quic_write_again, &args, 0);
 840     if (res <= 0) {
 841         if (!ossl_quic_channel_is_active(qc->ch))
 842             return QUIC_RAISE_NON_NORMAL_ERROR(qc, SSL_R_PROTOCOL_IS_SHUTDOWN, NULL);
 843         else
 844             return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
 845     }
 846
 847     *written = args.total_written;
 848     return 1;
 849 }
 850
 851 /*
 852  * Functions to manage All-or-Nothing (AON) (that is, non-ENABLE_PARTIAL_WRITE)
 853  * write semantics.
 854  */
 855 static void aon_write_begin(QUIC_CONNECTION *qc, const unsigned char *buf,
 856                             size_t buf_len, size_t already_sent)
 857 {
 858     assert(!qc->aon_write_in_progress);
 859
 860     qc->aon_write_in_progress = 1;
 861     qc->aon_buf_base          = buf;
 862     qc->aon_buf_pos           = already_sent;
 863     qc->aon_buf_len           = buf_len;
 864 }
 865
 866 static void aon_write_finish(QUIC_CONNECTION *qc)
 867 {
 868     qc->aon_write_in_progress   = 0;
 869     qc->aon_buf_base            = NULL;
 870     qc->aon_buf_pos             = 0;
 871     qc->aon_buf_len             = 0;
 872 }
 873
 874 static int quic_write_nonblocking_aon(QUIC_CONNECTION *qc, const void *buf,
 875                                       size_t len, size_t *written)
 876 {
 877     const void *actual_buf;
 878     size_t actual_len, actual_written = 0;
 879     int accept_moving_buffer
 880         = ((qc->ssl_mode & SSL_MODE_ACCEPT_MOVING_WRITE_BUFFER) != 0);
 881
 882     if (qc->aon_write_in_progress) {
 883         /*
 884          * We are in the middle of an AON write (i.e., a previous write did not
 885          * manage to append all data to the SSTREAM and we have EPW mode
 886          * disabled.)
 887          */
 888         if ((!accept_moving_buffer && qc->aon_buf_base != buf)
 889             || len != qc->aon_buf_len)
 890             /*
 891              * Pointer must not have changed if we are not in accept moving
 892              * buffer mode. Length must never change.
 893              */
 894             return QUIC_RAISE_NON_NORMAL_ERROR(qc, SSL_R_BAD_WRITE_RETRY, NULL);
 895
 896         actual_buf = (unsigned char *)buf + qc->aon_buf_pos;
 897         actual_len = len - qc->aon_buf_pos;
 898         assert(actual_len > 0);
 899     } else {
 900         actual_buf = buf;
 901         actual_len = len;
 902     }
 903
 904     /* First make a best effort to append as much of the data as possible. */
 905     if (!ossl_quic_sstream_append(qc->stream0->sstream, actual_buf, actual_len,
 906                                   &actual_written)) {
 907         /* Stream already finished or allocation error. */
 908         *written = 0;
 909         return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
 910     }
 911
 912     quic_post_write(qc, actual_written > 0, 1);
 913
 914     if (actual_written == actual_len) {
 915         /* We have sent everything. */
 916         if (qc->aon_write_in_progress) {
 917             /*
 918              * We have sent everything, and we were in the middle of an AON
 919              * write. The output write length is the total length of the AON
 920              * buffer, not however many bytes we managed to write to the stream
 921              * in this call.
 922              */
 923             *written = qc->aon_buf_len;
 924             aon_write_finish(qc);
 925         } else {
 926             *written = actual_written;
 927         }
 928
 929         return 1;
 930     }
 931
 932     if (qc->aon_write_in_progress) {
 933         /*
 934          * AON write is in progress but we have not written everything yet. We
 935          * may have managed to send zero bytes, or some number of bytes less
 936          * than the total remaining which need to be appended during this
 937          * AON operation.
 938          */
 939         qc->aon_buf_pos += actual_written;
 940         assert(qc->aon_buf_pos < qc->aon_buf_len);
 941         return QUIC_RAISE_NORMAL_ERROR(qc, SSL_ERROR_WANT_WRITE);
 942     }
 943
 944     /*
 945      * Not in an existing AON operation but partial write is not enabled, so we
 946      * need to begin a new AON operation. However we needn't bother if we didn't
 947      * actually append anything.
 948      */
 949     if (actual_written > 0)
 950         aon_write_begin(qc, buf, len, actual_written);
 951
 952     /*
 953      * AON - We do not publicly admit to having appended anything until AON
 954      * completes.
 955      */
 956     *written = 0;
 957     return QUIC_RAISE_NORMAL_ERROR(qc, SSL_ERROR_WANT_WRITE);
 958 }
 959
 960 static int quic_write_nonblocking_epw(QUIC_CONNECTION *qc, const void *buf, size_t len,
 961                                       size_t *written)
 962 {
 963     /* Simple best effort operation. */
 964     if (!ossl_quic_sstream_append(qc->stream0->sstream, buf, len, written)) {
 965         /* Stream already finished or allocation error. */
 966         *written = 0;
 967         return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
 968     }
 969
 970     quic_post_write(qc, *written > 0, 1);
 971     return 1;
 972 }
 973
 974 int ossl_quic_write(SSL *s, const void *buf, size_t len, size_t *written)
 975 {
 976     QUIC_CONNECTION *qc = QUIC_CONNECTION_FROM_SSL(s);
 977     int partial_write = ((qc->ssl_mode & SSL_MODE_ENABLE_PARTIAL_WRITE) != 0);
 978
 979     *written = 0;
 980
 981     if (!expect_quic_conn(qc))
 982         return 0;
 983
 984     if (qc->ch != NULL && ossl_quic_channel_is_term_any(qc->ch))
 985         return QUIC_RAISE_NON_NORMAL_ERROR(qc, SSL_R_PROTOCOL_IS_SHUTDOWN, NULL);
 986
 987     /*
 988      * If we haven't finished the handshake, try to advance it.
 989      * We don't accept writes until the handshake is completed.
 990      */
 991     if (ossl_quic_do_handshake(qc) < 1)
 992         return 0;
 993
 994     if (qc->stream0 == NULL || qc->stream0->sstream == NULL)
 995         return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
 996
 997     if (blocking_mode(qc))
 998         return quic_write_blocking(qc, buf, len, written);
 999     else if (partial_write)
1000         return quic_write_nonblocking_epw(qc, buf, len, written);
1001     else
1002         return quic_write_nonblocking_aon(qc, buf, len, written);
1003 }
1004
1005 /*
1006  * SSL_read
1007  * --------
1008  */
1009 struct quic_read_again_args {
1010     QUIC_CONNECTION *qc;
1011     QUIC_STREAM     *stream;
1012     void            *buf;
1013     size_t          len;
1014     size_t          *bytes_read;
1015     int             peek;
1016 };
1017
1018 static int quic_read_actual(QUIC_CONNECTION *qc,
1019                             QUIC_STREAM *stream,
1020                             void *buf, size_t buf_len,
1021                             size_t *bytes_read,
1022                             int peek)
1023 {
1024     int is_fin = 0;
1025
1026     if (stream->rstream == NULL)
1027         return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
1028
1029     if (peek) {
1030         if (!ossl_quic_rstream_peek(stream->rstream, buf, buf_len,
1031                                     bytes_read, &is_fin))
1032             return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
1033
1034     } else {
1035         if (!ossl_quic_rstream_read(stream->rstream, buf, buf_len,
1036                                     bytes_read, &is_fin))
1037             return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
1038     }
1039
1040     if (!peek) {
1041         if (*bytes_read > 0) {
1042             /*
1043              * We have read at least one byte from the stream. Inform stream-level
1044              * RXFC of the retirement of controlled bytes. Update the active stream
1045              * status (the RXFC may now want to emit a frame granting more credit to
1046              * the peer).
1047              */
1048             OSSL_RTT_INFO rtt_info;
1049
1050             ossl_statm_get_rtt_info(ossl_quic_channel_get_statm(qc->ch), &rtt_info);
1051
1052             if (!ossl_quic_rxfc_on_retire(&qc->stream0->rxfc, *bytes_read,
1053                                           rtt_info.smoothed_rtt))
1054                 return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
1055         }
1056
1057         if (is_fin)
1058             stream->recv_fin_retired = 1;
1059
1060         if (*bytes_read > 0)
1061             ossl_quic_stream_map_update_state(ossl_quic_channel_get_qsm(qc->ch),
1062                                               qc->stream0);
1063     }
1064
1065     return 1;
1066 }
1067
1068 static int quic_read_again(void *arg)
1069 {
1070     struct quic_read_again_args *args = arg;
1071
1072     if (!ossl_quic_channel_is_active(args->qc->ch))
1073         /* If connection is torn down due to an error while blocking, stop. */
1074         return -1;
1075
1076     if (!quic_read_actual(args->qc, args->stream,
1077                           args->buf, args->len, args->bytes_read,
1078                           args->peek))
1079         return -1;
1080
1081     if (*args->bytes_read > 0)
1082         /* got at least one byte, the SSL_read op can finish now */
1083         return 1;
1084
1085     return 0; /* did not write anything, keep trying */
1086 }
1087
1088 static int quic_read(SSL *s, void *buf, size_t len, size_t *bytes_read, int peek)
1089 {
1090     int res;
1091     QUIC_CONNECTION *qc = QUIC_CONNECTION_FROM_SSL(s);
1092     struct quic_read_again_args args;
1093
1094     *bytes_read = 0;
1095
1096     if (!expect_quic_conn(qc))
1097         return 0;
1098
1099     if (qc->ch != NULL && ossl_quic_channel_is_term_any(qc->ch))
1100         return QUIC_RAISE_NON_NORMAL_ERROR(qc, SSL_R_PROTOCOL_IS_SHUTDOWN, NULL);
1101
1102     /* If we haven't finished the handshake, try to advance it.*/
1103     if (ossl_quic_do_handshake(qc) < 1)
1104         return 0;
1105
1106     if (qc->stream0 == NULL)
1107         return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
1108
1109     if (!quic_read_actual(qc, qc->stream0, buf, len, bytes_read, peek))
1110         return 0;
1111
1112     if (*bytes_read > 0) {
1113         /*
1114          * Even though we succeeded, tick the reactor here to ensure we are
1115          * handling other aspects of the QUIC connection.
1116          */
1117         ossl_quic_reactor_tick(ossl_quic_channel_get_reactor(qc->ch));
1118         return 1;
1119     } else if (blocking_mode(qc)) {
1120         /*
1121          * We were not able to read anything immediately, so our stream
1122          * buffer is empty. This means we need to block until we get
1123          * at least one byte.
1124          */
1125         args.qc         = qc;
1126         args.stream     = qc->stream0;
1127         args.buf        = buf;
1128         args.len        = len;
1129         args.bytes_read = bytes_read;
1130         args.peek       = peek;
1131
1132         res = block_until_pred(qc, quic_read_again, &args, 0);
1133         if (res <= 0) {
1134             if (!ossl_quic_channel_is_active(qc->ch))
1135                 return QUIC_RAISE_NON_NORMAL_ERROR(qc, SSL_R_PROTOCOL_IS_SHUTDOWN, NULL);
1136             else
1137                 return QUIC_RAISE_NON_NORMAL_ERROR(qc, ERR_R_INTERNAL_ERROR, NULL);
1138         }
1139     }
1140
1141     return 1;
1142 }
1143
1144 int ossl_quic_read(SSL *s, void *buf, size_t len, size_t *bytes_read)
1145 {
1146     return quic_read(s, buf, len, bytes_read, 0);
1147 }
1148
1149 int ossl_quic_peek(SSL *s, void *buf, size_t len, size_t *bytes_read)
1150 {
1151     return quic_read(s, buf, len, bytes_read, 1);
1152 }
1153
1154 /*
1155  * SSL_pending
1156  * -----------
1157  */
1158 size_t ossl_quic_pending(const SSL *s)
1159 {
1160     const QUIC_CONNECTION *qc = QUIC_CONNECTION_FROM_CONST_SSL(s);
1161     size_t avail = 0;
1162     int fin = 0;
1163
1164     if (!expect_quic_conn(qc))
1165         return 0;
1166
1167     if (qc->stream0 == NULL || qc->stream0->rstream == NULL)
1168         /* Cannot raise errors here because we are const, just fail. */
1169         return 0;
1170
1171     if (!ossl_quic_rstream_available(qc->stream0->rstream, &avail, &fin))
1172         return 0;
1173
1174     return avail;
1175 }
1176
1177 /*
1178  * QUIC Front-End I/O API: SSL_CTX Management
1179  * ==========================================
1180  */
1181
1182 long ossl_quic_ctx_ctrl(SSL_CTX *ctx, int cmd, long larg, void *parg)
1183 {
1184     switch (cmd) {
1185     default:
1186         return ssl3_ctx_ctrl(ctx, cmd, larg, parg);
1187     }
1188 }
1189
1190 long ossl_quic_callback_ctrl(SSL *s, int cmd, void (*fp) (void))
1191 {
1192     return ssl3_callback_ctrl(s, cmd, fp);
1193 }
1194
1195 long ossl_quic_ctx_callback_ctrl(SSL_CTX *ctx, int cmd, void (*fp) (void))
1196 {
1197     return ssl3_ctx_callback_ctrl(ctx, cmd, fp);
1198 }
1199
1200 int ossl_quic_renegotiate_check(SSL *ssl, int initok)
1201 {
1202     /* We never do renegotiation. */
1203     return 0;
1204 }
1205
1206 /*
1207  * This is the subset of TLS1.3 ciphers which can be used with QUIC and which we
1208  * actually support.
1209  */
1210 static SSL_CIPHER tls13_quic_ciphers[] = {
1211     {
1212         1,
1213         TLS1_3_RFC_AES_128_GCM_SHA256,
1214         TLS1_3_RFC_AES_128_GCM_SHA256,
1215         TLS1_3_CK_AES_128_GCM_SHA256,
1216         SSL_kANY,
1217         SSL_aANY,
1218         SSL_AES128GCM,
1219         SSL_AEAD,
1220         TLS1_3_VERSION, TLS1_3_VERSION,
1221         0, 0,
1222         SSL_HIGH,
1223         SSL_HANDSHAKE_MAC_SHA256,
1224         128,
1225         128,
1226     }, {
1227         1,
1228         TLS1_3_RFC_AES_256_GCM_SHA384,
1229         TLS1_3_RFC_AES_256_GCM_SHA384,
1230         TLS1_3_CK_AES_256_GCM_SHA384,
1231         SSL_kANY,
1232         SSL_aANY,
1233         SSL_AES256GCM,
1234         SSL_AEAD,
1235         TLS1_3_VERSION, TLS1_3_VERSION,
1236         0, 0,
1237         SSL_HIGH,
1238         SSL_HANDSHAKE_MAC_SHA384,
1239         256,
1240         256,
1241     },
1242     {
1243         1,
1244         TLS1_3_RFC_CHACHA20_POLY1305_SHA256,
1245         TLS1_3_RFC_CHACHA20_POLY1305_SHA256,
1246         TLS1_3_CK_CHACHA20_POLY1305_SHA256,
1247         SSL_kANY,
1248         SSL_aANY,
1249         SSL_CHACHA20POLY1305,
1250         SSL_AEAD,
1251         TLS1_3_VERSION, TLS1_3_VERSION,
1252         0, 0,
1253         SSL_HIGH,
1254         SSL_HANDSHAKE_MAC_SHA256,
1255         256,
1256         256,
1257     }
1258 };
1259
1260 int ossl_quic_num_ciphers(void)
1261 {
1262     return OSSL_NELEM(tls13_quic_ciphers);
1263 }
1264
1265 const SSL_CIPHER *ossl_quic_get_cipher(unsigned int u)
1266 {
1267     if (u >= OSSL_NELEM(tls13_quic_ciphers))
1268         return NULL;
1269
1270     return &tls13_quic_ciphers[u];
1271 }