ClickHouse/dbms/programs/server/MySQLHandler.h

60 lines
1.4 KiB
C++
Raw Normal View History

2019-03-16 02:08:21 +00:00
#pragma once
#include <Poco/Net/TCPServerConnection.h>
2019-04-29 06:05:30 +00:00
#include <Poco/Net/SecureStreamSocket.h>
2019-03-16 02:08:21 +00:00
#include <Common/getFQDNOrHostName.h>
#include <Core/MySQLProtocol.h>
2019-05-26 06:52:29 +00:00
#include <openssl/rsa.h>
2019-03-16 02:08:21 +00:00
#include "IServer.h"
namespace DB
{
/// Handler for MySQL wire protocol connections. Allows to connect to ClickHouse using MySQL client.
2019-04-01 09:59:49 +00:00
class MySQLHandler : public Poco::Net::TCPServerConnection
{
2019-03-16 02:08:21 +00:00
public:
2019-05-26 06:52:29 +00:00
MySQLHandler(IServer & server_, const Poco::Net::StreamSocket & socket_, RSA & public_key, RSA & private_key, bool ssl_enabled, size_t connection_id);
2019-03-16 02:08:21 +00:00
void run() final;
private:
2019-04-29 06:05:30 +00:00
/// Enables SSL, if client requested.
MySQLProtocol::HandshakeResponse finishHandshake();
void comQuery(const String & payload);
2019-03-16 02:08:21 +00:00
void comFieldList(const String & payload);
2019-03-16 02:08:21 +00:00
void comPing();
void comInitDB(const String & payload);
2019-04-01 09:59:49 +00:00
static String generateScramble();
void authenticate(const MySQLProtocol::HandshakeResponse &, const String & scramble);
2019-03-16 02:08:21 +00:00
IServer & server;
Poco::Logger * log;
Context connection_context;
std::shared_ptr<MySQLProtocol::PacketSender> packet_sender;
2019-03-16 02:08:21 +00:00
2019-05-26 06:52:29 +00:00
size_t connection_id = 0;
2019-03-16 02:08:21 +00:00
2019-06-16 18:12:14 +00:00
size_t server_capability_flags = 0;
size_t client_capability_flags = 0;
2019-03-16 02:08:21 +00:00
2019-05-26 06:52:29 +00:00
RSA & public_key;
RSA & private_key;
2019-04-29 06:05:30 +00:00
std::shared_ptr<ReadBuffer> in;
std::shared_ptr<WriteBuffer> out;
bool secure_connection = false;
std::shared_ptr<Poco::Net::SecureStreamSocket> ss;
2019-03-16 02:08:21 +00:00
};
}