/* Copyright (c) 2019, 2024, Oracle and/or its affiliates. This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License, version 2.0, as published by the Free Software Foundation. This program is designed to work with certain software (including but not limited to OpenSSL) that is licensed under separate terms, as designated in a particular file or component or in included license documentation. The authors of MySQL hereby grant you an additional permission to link the program and your derivative works with the separately licensed software that they have either included with the program or referenced in the documentation. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License, version 2.0, for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */ #ifndef MYSQL_ASYNC_INCLUDED #define MYSQL_ASYNC_INCLUDED #include /** @file mysql_async.h Declarations for asynchronous client communication. @note this file should not be included as part of packaging. */ /** This enum is to represent different asynchronous operations like reading the network, writing to network, idle state, or complete state. */ enum net_async_operation { NET_ASYNC_OP_IDLE = 0, /**< default state */ NET_ASYNC_OP_READING, /**< used by my_net_read calls */ NET_ASYNC_OP_WRITING, /**< used by my_net_write calls */ NET_ASYNC_OP_COMPLETE /**< network read or write is complete */ }; /** Reading a packet is a multi-step process, so we have a state machine. */ enum net_async_read_packet_state { NET_ASYNC_PACKET_READ_IDLE = 0, /**< default packet read state */ NET_ASYNC_PACKET_READ_HEADER, /**< read packet header */ NET_ASYNC_PACKET_READ_BODY, /**< read packet contents */ NET_ASYNC_PACKET_READ_COMPLETE /**< state to define if packet is completely read */ }; /** Different states when reading a query result. */ enum net_read_query_result_status { NET_ASYNC_READ_QUERY_RESULT_IDLE = 0, /**< default state */ NET_ASYNC_READ_QUERY_RESULT_FIELD_COUNT, /**< read Ok or read field count sent as part of COM_QUERY */ NET_ASYNC_READ_QUERY_RESULT_FIELD_INFO /**< read result of above COM_* command */ }; /** Sending a command involves the write as well as reading the status. */ enum net_send_command_status { NET_ASYNC_SEND_COMMAND_IDLE = 0, /**< default send command state */ NET_ASYNC_SEND_COMMAND_WRITE_COMMAND, /**< send COM_* command */ NET_ASYNC_SEND_COMMAND_READ_STATUS /**< read result of above COM_* command */ }; /** Async operations are broadly classified into 3 phases: Connection phase, phase of sending data to server (which is writing phase) and reading data from server (which is reading phase). Below enum describes the same */ enum net_async_block_state { NET_NONBLOCKING_CONNECT = 0, NET_NONBLOCKING_READ, NET_NONBLOCKING_WRITE }; /** Represents the packet to be sent on wire asynchronously. */ struct io_vec { void *iov_base; /**< Starting address */ size_t iov_len; /**< Number of bytes to transfer */ }; typedef struct NET_ASYNC { /** The position in buff we continue reads from when data is next available */ unsigned char *cur_pos; /** Blocking state */ enum net_async_block_state async_blocking_state; /** Our current operation */ enum net_async_operation async_operation; /** How many bytes we want to read */ size_t async_bytes_wanted; /** Simple state to know if we're reading the first row, and command/query statuses. */ bool read_rows_is_first_read; enum net_send_command_status async_send_command_status; enum net_read_query_result_status async_read_query_result_status; /** State when waiting on an async read */ enum net_async_read_packet_state async_packet_read_state; /** Size of the packet we're currently reading */ size_t async_packet_length; /** Headers and vector for our async writes; see net_serv.c for detailed description. */ unsigned char *async_write_headers; struct io_vec *async_write_vector; size_t async_write_vector_size; size_t async_write_vector_current; /** If the packet length is less than MAX_PACKET_LENGTH then use a static array to hold the meta packet header. The array either holds the usual packet header or a compressed meta packet header as following. The compressed meta packet header is followwed by usual compresses packet heder that is 7 bytes in length. Packet Header ~~~~~~~~~~~~~~~~~~~ B1 B2 B3 : Packet length B4 : Packet number ~~~~~~~~~~~~~~~~~~~ Payload ~~~~~~~~~~~~~~~~~~~ B5 : COM_COMMAND ~~~~~~~~~~~~~~~~~~~ Compressed Packet Header ~~~~~~~~~~~~~~~~~~~ B1 B2 B3 : Compress packet length B4 : Compress Packet Nunmber 00 00 00 : Indicates following payload is uncompressed ~~~~~~~~~~~~~~~~~~~ Payload ~~~~~~~~~~~~~~~~~~~ B8 B9 B10 : Packet size B11 : Packet number B12 : COM_COMMAND ~~~~~~~~~~~~~~~~~~~ */ unsigned char inline_async_write_header[NET_HEADER_SIZE + COMP_HEADER_SIZE + NET_HEADER_SIZE + 1]; struct io_vec inline_async_write_vector[3]; /** Keep track of compressed buffers */ unsigned char **compressed_write_buffers; /** Size of the compressed buffer */ size_t compressed_buffers_size; } NET_ASYNC; struct NET_EXTENSION { NET_ASYNC *net_async_context; mysql_compress_context compress_ctx; }; NET_EXTENSION *net_extension_init(); void net_extension_free(NET *); #define NET_EXTENSION_PTR(N) \ ((NET_EXTENSION *)((N)->extension ? (N)->extension : NULL)) #define NET_ASYNC_DATA(M) \ ((NET_EXTENSION_PTR(M)) ? (NET_EXTENSION_PTR(M)->net_async_context) : NULL) /** Asynchronous operations are broadly classified into 2 categories. 1. Connection 2. Query execution This classification is defined in below enum */ enum mysql_async_operation_status { ASYNC_OP_UNSET = 0, ASYNC_OP_CONNECT, ASYNC_OP_QUERY }; /** Query execution in an asynchronous fashion is broadly divided into 3 states which is described in below enum */ enum mysql_async_query_state_enum { QUERY_IDLE = 0, QUERY_SENDING, QUERY_READING_RESULT }; typedef struct MYSQL_ASYNC { /** Buffer storing the rows result for cli_read_rows_nonblocking */ MYSQL_DATA *rows_result_buffer; /** a pointer to keep track of the previous row of the current result row */ MYSQL_ROWS **prev_row_ptr; /** Context needed to track the state of a connection being established */ struct mysql_async_connect *connect_context; /** Status of the current async op */ enum mysql_async_operation_status async_op_status; /** Size of the current running async query */ size_t async_query_length; /** If a query is running, this is its state */ enum mysql_async_query_state_enum async_query_state; /** context needed to support metadata read operation */ unsigned long *async_read_metadata_field_len; MYSQL_FIELD *async_read_metadata_fields; MYSQL_ROWS async_read_metadata_data; unsigned int async_read_metadata_cur_field; /** a pointer to keep track of the result sets */ struct MYSQL_RES *async_store_result_result; /** the query parameters data */ uchar *async_qp_data; /** the query parameters data length */ unsigned long async_qp_data_length; } MYSQL_ASYNC; enum net_async_status my_net_write_nonblocking(NET *net, const unsigned char *packet, size_t len, bool *res); enum net_async_status net_write_command_nonblocking( NET *net, unsigned char command, const unsigned char *prefix, size_t prefix_len, const unsigned char *packet, size_t packet_len, bool *res); enum net_async_status my_net_read_nonblocking(NET *net, unsigned long *len_ptr); int mysql_get_socket_descriptor(MYSQL *mysql); #endif /* MYSQL_ASYNC_INCLUDED */