1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
*
* Copyright (C) 2008 Sun Microsystems, Inc.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; version 2 of the License.
*
* 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 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 PLUGIN_MYSQL_PROTOCOL_MYSQL_PROTOCOL_H
#define PLUGIN_MYSQL_PROTOCOL_MYSQL_PROTOCOL_H
#include <drizzled/plugin/listen_tcp.h>
#include <drizzled/plugin/client.h>
#include <drizzled/atomics.h>
#include "drizzled/plugin/table_function.h"
#include "net_serv.h"
namespace drizzle_plugin
{
void compose_ip_addresses(std::vector<std::string> options);
class ProtocolCounters
{
public:
ProtocolCounters():
max_connections(1000)
{ }
drizzled::atomic<uint64_t> connectionCount;
drizzled::atomic<uint64_t> adminConnectionCount;
drizzled::atomic<uint64_t> failedConnections;
drizzled::atomic<uint64_t> connected;
drizzled::atomic<uint64_t> adminConnected;
uint32_t max_connections;
};
typedef drizzled::constrained_check<uint32_t, 300, 1> timeout_constraint;
typedef drizzled::constrained_check<uint32_t, 300, 1> retry_constraint;
typedef drizzled::constrained_check<uint32_t, 1048576, 1024, 1024> buffer_constraint;
class ListenMySQLProtocol: public drizzled::plugin::ListenTcp
{
protected:
const std::string _hostname;
bool _using_mysql41_protocol;
public:
ListenMySQLProtocol(std::string name,
const std::string &hostname,
bool using_mysql41_protocol):
drizzled::plugin::ListenTcp(name),
_hostname(hostname),
_using_mysql41_protocol(using_mysql41_protocol)
{ }
virtual ~ListenMySQLProtocol();
virtual const std::string getHost(void) const;
virtual in_port_t getPort(void) const;
virtual drizzled::plugin::Client *getClient(int fd);
static ProtocolCounters *mysql_counters;
virtual ProtocolCounters *getCounters(void) const { return mysql_counters; }
};
class ClientMySQLProtocol: public drizzled::plugin::Client
{
protected:
NET net;
drizzled::String packet;
uint32_t client_capabilities;
bool is_admin_connection;
bool _using_mysql41_protocol;
bool checkConnection(void);
bool netStoreData(const unsigned char *from, size_t length);
void writeEOFPacket(uint32_t server_status, uint32_t total_warn_count);
unsigned char *storeLength(unsigned char *packet, uint64_t length);
void makeScramble(char *scramble);
public:
ClientMySQLProtocol(int fd, bool _using_mysql41_protocol, ProtocolCounters *set_counters);
virtual ~ClientMySQLProtocol();
ProtocolCounters *counters;
virtual int getFileDescriptor(void);
virtual bool isConnected();
virtual bool isReading(void);
virtual bool isWriting(void);
virtual bool flush(void);
virtual void close(void);
virtual bool authenticate(void);
virtual bool readCommand(char **packet, uint32_t *packet_length);
virtual void sendOK(void);
virtual void sendEOF(void);
virtual void sendError(uint32_t sql_errno, const char *err);
virtual bool sendFields(drizzled::List<drizzled::Item> *list);
using Client::store;
virtual bool store(drizzled::Field *from);
virtual bool store(void);
virtual bool store(int32_t from);
virtual bool store(uint32_t from);
virtual bool store(int64_t from);
virtual bool store(uint64_t from);
virtual bool store(double from, uint32_t decimals, drizzled::String *buffer);
virtual bool store(const char *from, size_t length);
virtual bool haveError(void);
virtual bool haveMoreData(void);
virtual bool wasAborted(void);
virtual bool isAdminAllowed(void);
static std::vector<std::string> mysql_admin_ip_addresses;
static void mysql_compose_ip_addresses(std::vector<std::string> options);
};
} /* namespace drizzle_plugin */
#endif /* PLUGIN_MYSQL_PROTOCOL_MYSQL_PROTOCOL_H */
|