25#ifndef MYSQLD_MOCK_STATEMENT_READER_INCLUDED
26#define MYSQLD_MOCK_STATEMENT_READER_INCLUDED
35#include <openssl/bio.h>
63 std::vector<classic_protocol::message::server::ColumnMeta>
columns;
64 std::vector<RowValueType>
rows;
94 ssl_ = std::move(rhs.ssl_);
106 virtual void encode_ok(
const uint64_t affected_rows = 0,
107 const uint64_t last_insert_id = 0,
108 const uint16_t server_status = 0,
109 const uint16_t warning_count = 0) = 0;
112 encode_ok(resp.affected_rows(), resp.last_insert_id(),
113 resp.status_flags().to_ulong(), resp.warning_count());
127 template <
class CompletionToken>
134 auto write_ec = write_res.error();
140 [
this, compl_handler = std::move(
init.completion_handler)](
141 std::error_code ec)
mutable {
143 compl_handler(ec, {});
151 [compl_handler = std::move(
init.completion_handler),
152 ec = write_res.error()]() { compl_handler(ec, {}); });
158 [compl_handler = std::move(
init.completion_handler),
159 transferred = write_res.value()]() {
160 compl_handler({}, transferred);
164 return init.result.get();
167 template <
class CompletionToken>
170 async_send_tls(std::forward<CompletionToken>(token));
173 std::forward<CompletionToken>(token));
182 template <
class CompletionToken>
189 auto orig_size =
buf.size();
190 auto grow_size = 16 * 1024;
193 size_t transferred{};
194 auto read_res = read_ssl(
buf.data(orig_size, grow_size));
196 transferred = read_res.value();
199 buf.shrink(grow_size - transferred);
202 const auto read_ec = read_res.error();
204 client_socket_.async_wait(
207 [
this, compl_handler = std::move(
init.completion_handler)](
208 std::error_code ec)
mutable {
210 compl_handler(ec, {});
214 async_receive_tls(std::move(compl_handler));
221 client_socket_.get_executor(),
222 [compl_handler = std::move(
init.completion_handler),
223 ec = map_tls_error_code(read_ec)]() { compl_handler(ec, {}); });
228 [compl_handler = std::move(
init.completion_handler),
229 transferred]() { compl_handler({}, transferred); });
232 return init.result.get();
235 template <
class CompletionToken>
238 return async_receive_tls(std::forward<CompletionToken>(token));
241 std::forward<CompletionToken>(token));
245 template <
class CompletionToken>
250 auto res = tls_accept();
252 auto ec = res.error();
257 client_socket_.async_wait(
258 wt, [&, compl_handler = std::move(
init.completion_handler)](
259 std::error_code ec)
mutable {
266 async_tls_accept(std::move(compl_handler));
269 net::defer(client_socket_.get_executor().context(),
270 [ec, compl_handler = std::move(
init.completion_handler)]() {
275 net::defer(client_socket_.get_executor().context(),
276 [compl_handler = std::move(
init.completion_handler)]() {
281 return init.result.get();
284 const std::vector<uint8_t> &
send_buffer()
const {
return send_buffer_; }
293 auth_method_name_ = auth_method_name;
299 auth_method_data_ = auth_method_data;
304 static bool authenticate(
const std::string &auth_method_name,
305 const std::string &auth_method_data,
307 const std::vector<uint8_t> &auth_response);
313 const SSL *
ssl()
const {
return ssl_.get(); }
330 std::string username_{};
331 std::string auth_method_name_{};
332 std::string auth_method_data_{};
341 std::unique_ptr<SSL, SSL_Deleter>
ssl_;
354 bool cert_required{
false};
static mysql_service_status_t init()
Component initialization.
Definition: audit_api_message_emit.cc:570
TLS Context for the server side.
Definition: tls_server_context.h:45
Definition: executor.h:71
Definition: socket.h:1143
io_context & context() const noexcept
Definition: io_context.h:1001
Definition: io_context.h:60
executor_type get_executor() noexcept
Definition: io_context.h:1079
TCP protocol.
Definition: internet.h:1154
basic_endpoint< tcp > endpoint
Definition: internet.h:1156
basic_stream_socket< tcp > socket
Definition: internet.h:1158
static constexpr wait_type wait_read
Definition: socket.h:167
static constexpr wait_type wait_write
Definition: socket.h:168
Definition: statement_reader.h:336
void operator()(SSL *v)
Definition: statement_reader.h:338
Definition: statement_reader.h:79
std::string username() const
Definition: statement_reader.h:290
virtual ~ProtocolBase()=default
net::steady_timer & exec_timer()
Definition: statement_reader.h:317
TlsServerContext & tls_ctx_
Definition: statement_reader.h:334
ProtocolBase(socket_type client_sock, endpoint_type client_ep, TlsServerContext &tls_ctx)
Definition: protocol_base.cc:35
stdx::expected< size_t, std::error_code > avail_ssl()
Definition: protocol_base.cc:73
const std::vector< uint8_t > & send_buffer() const
Definition: statement_reader.h:284
void async_send_tls(CompletionToken &&token)
Definition: statement_reader.h:128
socket_type client_socket_
Definition: statement_reader.h:326
const SSL * ssl() const
Definition: statement_reader.h:313
std::string auth_method_data() const
Definition: statement_reader.h:302
virtual void encode_ok(const uint64_t affected_rows=0, const uint64_t last_insert_id=0, const uint16_t server_status=0, const uint16_t warning_count=0)=0
void encode_ok(const OkResponse &resp)
Definition: statement_reader.h:111
void auth_method_name(const std::string &auth_method_name)
Definition: statement_reader.h:292
ProtocolBase(ProtocolBase &&)=default
ProtocolBase & operator=(ProtocolBase &&rhs)
Definition: statement_reader.h:92
void async_tls_accept(CompletionToken &&token)
Definition: statement_reader.h:246
virtual void encode_resultset(const ResultsetResponse &response)=0
endpoint_type client_ep_
Definition: statement_reader.h:327
std::vector< uint8_t > recv_buffer_
Definition: statement_reader.h:343
const net::ip::tcp::socket & client_socket() const
Definition: statement_reader.h:286
void async_receive_tls(CompletionToken &&token)
Definition: statement_reader.h:183
void auth_method_data(const std::string &auth_method_data)
Definition: statement_reader.h:298
bool is_tls()
Definition: statement_reader.h:311
void async_receive(CompletionToken &&token)
Definition: statement_reader.h:236
typename protocol_type::endpoint endpoint_type
Definition: statement_reader.h:83
void async_send(CompletionToken &&token)
Definition: statement_reader.h:168
std::string auth_method_name() const
Definition: statement_reader.h:296
virtual void encode_error(const ErrorResponse &resp)=0
stdx::expected< size_t, std::error_code > read_ssl(const net::mutable_buffer &buf)
Definition: protocol_base.cc:58
static std::error_code map_tls_error_code(std::error_code ec)
Definition: statement_reader.h:178
void username(const std::string &username)
Definition: statement_reader.h:288
ProtocolBase & operator=(const ProtocolBase &)=delete
typename protocol_type::socket socket_type
Definition: statement_reader.h:82
stdx::expected< size_t, std::error_code > write_ssl(const net::const_buffer &buf)
Definition: protocol_base.cc:47
std::vector< uint8_t > send_buffer_
Definition: statement_reader.h:344
net::io_context & io_context()
Definition: statement_reader.h:321
ProtocolBase(const ProtocolBase &)=delete
std::unique_ptr< SSL, SSL_Deleter > ssl_
Definition: statement_reader.h:341
Definition: statement_reader.h:347
StatementReaderBase & operator=(StatementReaderBase &&)=default
virtual std::vector< AsyncNotice > get_async_notices()=0
virtual stdx::expected< handshake_data, ErrorResponse > handshake()=0
StatementReaderBase(StatementReaderBase &&)=default
virtual ~StatementReaderBase()=default
virtual void set_session_ssl_info(const SSL *ssl)=0
virtual void handle_statement(const std::string &statement, ProtocolBase *protocol)=0
Returns the data about the next statement from the json file.
virtual stdx::expected< classic_protocol::message::server::Greeting, std::error_code > server_greeting(bool with_tls)=0
StatementReaderBase & operator=(const StatementReaderBase &)=default
virtual std::chrono::microseconds server_greeting_exec_time()=0
StatementReaderBase()=default
StatementReaderBase(const StatementReaderBase &)=default
virtual std::chrono::microseconds get_default_exec_time()=0
Returns the default execution time in microseconds.
static char * password
Definition: mysql_secure_installation.cc:55
Definition: buf0block_hint.cc:29
constexpr value_type ssl
Definition: classic_protocol_constants.h:48
borrowable::message::server::Error< false > Error
Definition: classic_protocol_message.h:1357
borrowable::message::server::Greeting< false > Greeting
Definition: classic_protocol_message.h:1359
borrowable::message::server::Ok< false > Ok
Definition: classic_protocol_message.h:1356
dynamic_vector_buffer< T, Allocator > dynamic_buffer(std::vector< T, Allocator > &vec) noexcept
Definition: buffer.h:662
auto defer(CompletionToken &&token)
Definition: executor.h:803
mutable_buffer buffer(void *p, size_t n) noexcept
Definition: buffer.h:419
std::enable_if_t< is_dynamic_buffer< DynamicBuffer >::value, void > async_read(AsyncReadStream &stream, DynamicBuffer &&b, CompletionCondition completion_condition, CompletionToken &&token)
Definition: buffer.h:927
std::enable_if_t< is_dynamic_buffer< DynamicBuffer >::value, void > async_write(AsyncWriteStream &stream, DynamicBuffer &&b, CompletionCondition cond, CompletionToken &&token)
Definition: buffer.h:1073
Definition: mock_server_component.h:35
classic_protocol::message::server::Ok OkResponse
Definition: statement_reader.h:67
std::vector< std::optional< std::string > > RowValueType
Vector for keeping has_value|string representation of the values of the single row (ordered by column...
Definition: statement_reader.h:57
classic_protocol::message::server::Error ErrorResponse
Definition: statement_reader.h:68
Definition: statement_reader.h:70
std::chrono::milliseconds send_offset_ms
Definition: statement_reader.h:73
bool is_local
Definition: statement_reader.h:75
std::string payload
Definition: statement_reader.h:76
unsigned type
Definition: statement_reader.h:74
Keeps result data for single SQL statement that returns resultset.
Definition: statement_reader.h:62
std::vector< RowValueType > rows
Definition: statement_reader.h:64
std::vector< classic_protocol::message::server::ColumnMeta > columns
Definition: statement_reader.h:63
Definition: statement_reader.h:349
std::optional< std::string > username
Definition: statement_reader.h:352
std::optional< ErrorResponse > error
Definition: statement_reader.h:350
std::optional< std::string > cert_issuer
Definition: statement_reader.h:356
std::optional< std::string > password
Definition: statement_reader.h:353
std::optional< std::string > cert_subject
Definition: statement_reader.h:355
Definition: mysqlslap.cc:216