/*
 * Copyright 2022 The OpenSSL Project Authors. All Rights Reserved.
 *
 * Licensed under the Apache License 2.0 (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
 */

#ifndef OSSL_QUIC_LOCAL_H
# define OSSL_QUIC_LOCAL_H

# include <openssl/ssl.h>
# include "internal/quic_ssl.h"       /* QUIC_CONNECTION */
# include "internal/quic_fc.h"
# include "internal/quic_stream.h"
# include "../ssl_local.h"

struct quic_stream_st {
    /* type identifier and common data for the public SSL object */
    struct ssl_st ssl;

    /* QUIC_CONNECTION that this stream belongs to */
    QUIC_CONNECTION *conn;
    /* receive flow controller */
    QUIC_RXFC *rxfc;
    /* receive and send stream objects */
    QUIC_RSTREAM *rstream;
    QUIC_SSTREAM *sstream;
};

struct quic_conn_st {
    /* QUIC connection is always a stream (the stream id 0) */
    struct quic_stream_st stream;
    /* the associated tls-1.3 connection data */
    SSL *tls;

    /* QUIC ack manager */
    OSSL_ACKM *ackm;
    /* QUIC receive record layer */
    OSSL_QRX *qrx;
};

# define QUIC_CONNECTION_FROM_SSL_int(ssl, c)   \
    ((ssl) == NULL ? NULL                       \
     : ((ssl)->type == SSL_TYPE_QUIC_CONNECTION \
        ? (c QUIC_CONNECTION *)(ssl)            \
        : NULL))

# define QUIC_STREAM_FROM_SSL_int(ssl, c)       \
    ((ssl) == NULL ? NULL                       \
     : ((ssl)->type == SSL_TYPE_QUIC_CONNECTION \
         || (ssl)->type == SSL_TYPE_QUIC_STREAM \
        ? (c QUIC_STREAM *)(ssl)                \
        : NULL))

# define SSL_CONNECTION_FROM_QUIC_SSL_int(ssl, c)               \
    ((ssl) == NULL ? NULL                                       \
     : ((ssl)->type == SSL_TYPE_QUIC_CONNECTION                 \
        ? (c SSL_CONNECTION *)((c QUIC_CONNECTION *)(ssl))->tls \
        : NULL))

# define QUIC_CONNECTION_FROM_SSL(ssl) \
    QUIC_CONNECTION_FROM_SSL_int(ssl, SSL_CONNECTION_NO_CONST)
# define QUIC_CONNECTION_FROM_CONST_SSL(ssl) \
    QUIC_CONNECTION_FROM_SSL_int(ssl, const)
# define QUIC_STREAM_FROM_SSL(ssl) \
    QUIC_STREAM_FROM_SSL_int(ssl, SSL_CONNECTION_NO_CONST)
# define QUIC_STREAM_FROM_CONST_SSL(ssl) \
    QUIC_STREAM_FROM_SSL_int(ssl, const)
# define SSL_CONNECTION_FROM_QUIC_SSL(ssl) \
    SSL_CONNECTION_FROM_QUIC_SSL_int(ssl, SSL_CONNECTION_NO_CONST)
# define SSL_CONNECTION_FROM_CONST_QUIC_SSL(ssl) \
    SSL_CONNECTION_FROM_CONST_QUIC_SSL_int(ssl, const)

# define OSSL_QUIC_ANY_VERSION 0xFFFFF

# define IMPLEMENT_quic_meth_func(version, func_name, q_accept, \
                                 q_connect, enc_data) \
const SSL_METHOD *func_name(void)  \
        { \
        static const SSL_METHOD func_name##_data= { \
                version, \
                0, \
                0, \
                ossl_quic_new, \
                ossl_quic_free, \
                ossl_quic_reset, \
                ossl_quic_init, \
                ossl_quic_clear, \
                ossl_quic_deinit, \
                q_accept, \
                q_connect, \
                ossl_quic_read, \
                ossl_quic_peek, \
                ossl_quic_write, \
                ossl_quic_shutdown, \
                NULL /* renegotiate */, \
                ossl_quic_renegotiate_check, \
                NULL /* read_bytes */, \
                NULL /* write_bytes */, \
                NULL /* dispatch_alert */, \
                ossl_quic_ctrl, \
                ossl_quic_ctx_ctrl, \
                NULL /* get_cipher_by_char */, \
                NULL /* put_cipher_by_char */, \
                ossl_quic_pending, \
                ossl_quic_num_ciphers, \
                ossl_quic_get_cipher, \
                ossl_quic_default_timeout, \
                &enc_data, \
                ssl_undefined_void_function, \
                ossl_quic_callback_ctrl, \
                ossl_quic_ctx_callback_ctrl, \
        }; \
        return &func_name##_data; \
        }

#endif