74aa3638f9
dcb_final_free: Router session is not closed anymore when dcb is closed. Router session is shared among all dcbs and is closed and freed with session. dcb_connect: dcb's state must be switched fro DCB_STATE_ALLOC to DCB_STATE_DISCONNECTED before dcb_final_free can be called for it. dcb_close: poll_remove_dcb encapsulates dcb's state transition in the same way as poll_add_dcb. Removed state modification from dcb_close. Read return value of poll_remove_dcb and log accordingly. dcb_set state: remove dassert if dcb_set_state_nomutex returned false. False can be returned indicating that intented change didn't occur but the end state, for example, may be acceptable. Failures in state transitions are asserted in dcb_Set_state_nomutex. poll.c poll_add_dcb: dcb state is now set here to either DCB_STATE_LISTENING or to DCB_STATE_POLLING according to dcb's role. Failures in state setting and in epoll_ctl are detected, logged and handled. poll_remove_dcb: Failures in state setting and epoll_ctl are detected, logged, and handled. mysql_client_server_protocol.h Removed macros MYSQL_FAILED_AUTHENTICATION & MYSQL_SUCCESFUL_AUTHENTICATION as they were not necessary and used with constant values 0 and 1 depending on the case. Renamed variable 'conn' to 'protocol' in cases where it meant protocol. mysql_backend.c gw_read_backend_event: In case when there was nothing to read or read failed, backend dcb is closed because situation is assumed to be such that backend server closed its side of the socket. Removed macros MYSQL_FAILED/SUCCESFUL_AUTHENTICATION gw_create_backend_connection: Assigned protocol with fd which is connected to backend. backend_write_delayqueue: In case where dcb_write fails to write anything, close backend dcb to avoid it getting hanging. mysql_client.c gw_read_client_event: Read return value of routeQuery and if it isn't == 1, call mysql_send_custom_error with client dcb and set client's protocol state to MYSQL_IDLE instead of MYSQL_ROUTING. gw_MySQLAccept: Static reply counter was erroneously used as a criteria for jumping to return point where return value was constantly 'success' (=0). Replaced static reply counter with private. Fixed return value in different cases. mysql_common.c gw_receive_backend_auth: Changed to return boolean values indicating of success or failue. Used integers which were assigned to macroed values on the caller side. Added length check before accessing buffer. readconnroute.c Cut too long lines and removed statements with side effects. skygw_debug.h Added macro STRPROTOCOLSTATE(s) to produce string representation of a given protocol state.
299 lines
11 KiB
C
299 lines
11 KiB
C
#ifndef _MYSQL_PROTOCOL_H
|
|
#define _MYSQL_PROTOCOL_H
|
|
/*
|
|
* This file is distributed as part of the SkySQL Gateway. It 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.
|
|
*
|
|
* 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 Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*
|
|
* Copyright SkySQL Ab 2013
|
|
*/
|
|
|
|
/*
|
|
* Revision History
|
|
*
|
|
* Date Who Description
|
|
* 01-06-2013 Mark Riddoch Initial implementation
|
|
* 14-06-2013 Massimiliano Pinto Added specific data
|
|
* for MySQL session
|
|
* 04-07-2013 Massimiliano Pinto Added new MySQL protocol status for asynchronous connection
|
|
* Added authentication reply status
|
|
* 12-07-2013 Massimiliano Pinto Added routines for change_user
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <openssl/sha.h>
|
|
#include <sys/ioctl.h>
|
|
#include <errno.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
#include <stdbool.h>
|
|
#include <fcntl.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include <service.h>
|
|
#include <router.h>
|
|
#include <poll.h>
|
|
#include <users.h>
|
|
|
|
#define GW_VERSION "0.1.0"
|
|
#define GW_MYSQL_VERSION "5.5.22-SKYSQL-" GW_VERSION
|
|
#define GW_MYSQL_LOOP_TIMEOUT 300000000
|
|
#define GW_MYSQL_READ 0
|
|
#define GW_MYSQL_WRITE 1
|
|
|
|
#define GW_MYSQL_PROTOCOL_VERSION 10 // version is 10
|
|
#define GW_MYSQL_HANDSHAKE_FILLER 0x00
|
|
#define GW_MYSQL_SERVER_CAPABILITIES_BYTE1 0xff
|
|
#define GW_MYSQL_SERVER_CAPABILITIES_BYTE2 0xf7
|
|
#define GW_MYSQL_SERVER_LANGUAGE 0x08
|
|
#define GW_MYSQL_MAX_PACKET_LEN 0xffffffL;
|
|
#define GW_MYSQL_SCRAMBLE_SIZE 20
|
|
#define GW_SCRAMBLE_LENGTH_323 8
|
|
|
|
#ifndef MYSQL_SCRAMBLE_LEN
|
|
#define MYSQL_SCRAMBLE_LEN GW_MYSQL_SCRAMBLE_SIZE
|
|
#endif
|
|
|
|
#define MYSQL_USER_MAXLEN 128
|
|
#define MYSQL_DATABASE_MAXLEN 128
|
|
|
|
#define GW_NOINTR_CALL(A) do { errno = 0; A; } while (errno == EINTR)
|
|
// network buffer is 32K
|
|
#define MAX_BUFFER_SIZE 32768
|
|
// socket send buffer for backend
|
|
#define GW_BACKEND_SO_SNDBUF 1024
|
|
#define SMALL_CHUNK 1024
|
|
#define MAX_CHUNK SMALL_CHUNK * 8 * 4
|
|
#define ToHex(Y) (Y>='0'&&Y<='9'?Y-'0':Y-'A'+10)
|
|
|
|
struct dcb;
|
|
|
|
typedef enum {
|
|
MYSQL_ALLOC,
|
|
MYSQL_PENDING_CONNECT,
|
|
MYSQL_CONNECTED,
|
|
MYSQL_AUTH_SENT,
|
|
MYSQL_AUTH_RECV,
|
|
MYSQL_AUTH_FAILED,
|
|
MYSQL_IDLE,
|
|
MYSQL_ROUTING,
|
|
MYSQL_WAITING_RESULT,
|
|
MYSQL_SESSION_CHANGE
|
|
} mysql_pstate_t;
|
|
|
|
|
|
/*
|
|
* MySQL Protocol specific state data
|
|
*/
|
|
typedef struct {
|
|
#if defined(SS_DEBUG)
|
|
skygw_chk_t protocol_chk_top;
|
|
#endif
|
|
int fd; /* The socket descriptor */
|
|
struct dcb *owner_dcb; /** The DCB of the socket
|
|
* we are running on */
|
|
mysql_pstate_t state; /** Current protocol state */
|
|
uint8_t scramble[MYSQL_SCRAMBLE_LEN]; /** server scramble,
|
|
* created or received */
|
|
uint32_t server_capabilities; /** server capabilities,
|
|
* created or received */
|
|
uint32_t client_capabilities; /** client capabilities,
|
|
* created or received */
|
|
unsigned long tid; /** MySQL Thread ID, in
|
|
* handshake */
|
|
#if defined(SS_DEBUG)
|
|
skygw_chk_t protocol_chk_tail;
|
|
#endif
|
|
} MySQLProtocol;
|
|
|
|
/*
|
|
* MySQL session specific data
|
|
*
|
|
*/
|
|
typedef struct mysql_session {
|
|
uint8_t client_sha1[MYSQL_SCRAMBLE_LEN]; /* SHA1(passowrd) */
|
|
char user[MYSQL_USER_MAXLEN]; /* username */
|
|
char db[MYSQL_DATABASE_MAXLEN]; /* database */
|
|
} MYSQL_session;
|
|
|
|
|
|
|
|
/* Protocol packing macros. */
|
|
#define gw_mysql_set_byte2(__buffer, __int) do { \
|
|
(__buffer)[0]= (uint8_t)((__int) & 0xFF); \
|
|
(__buffer)[1]= (uint8_t)(((__int) >> 8) & 0xFF); } while (0)
|
|
#define gw_mysql_set_byte3(__buffer, __int) do { \
|
|
(__buffer)[0]= (uint8_t)((__int) & 0xFF); \
|
|
(__buffer)[1]= (uint8_t)(((__int) >> 8) & 0xFF); \
|
|
(__buffer)[2]= (uint8_t)(((__int) >> 16) & 0xFF); } while (0)
|
|
#define gw_mysql_set_byte4(__buffer, __int) do { \
|
|
(__buffer)[0]= (uint8_t)((__int) & 0xFF); \
|
|
(__buffer)[1]= (uint8_t)(((__int) >> 8) & 0xFF); \
|
|
(__buffer)[2]= (uint8_t)(((__int) >> 16) & 0xFF); \
|
|
(__buffer)[3]= (uint8_t)(((__int) >> 24) & 0xFF); } while (0)
|
|
|
|
/* Protocol unpacking macros. */
|
|
#define gw_mysql_get_byte2(__buffer) \
|
|
(uint16_t)((__buffer)[0] | \
|
|
((__buffer)[1] << 8))
|
|
#define gw_mysql_get_byte3(__buffer) \
|
|
(uint32_t)((__buffer)[0] | \
|
|
((__buffer)[1] << 8) | \
|
|
((__buffer)[2] << 16))
|
|
#define gw_mysql_get_byte4(__buffer) \
|
|
(uint32_t)((__buffer)[0] | \
|
|
((__buffer)[1] << 8) | \
|
|
((__buffer)[2] << 16) | \
|
|
((__buffer)[3] << 24))
|
|
#define gw_mysql_get_byte8(__buffer) \
|
|
((uint64_t)(__buffer)[0] | \
|
|
((uint64_t)(__buffer)[1] << 8) | \
|
|
((uint64_t)(__buffer)[2] << 16) | \
|
|
((uint64_t)(__buffer)[3] << 24) | \
|
|
((uint64_t)(__buffer)[4] << 32) | \
|
|
((uint64_t)(__buffer)[5] << 40) | \
|
|
((uint64_t)(__buffer)[6] << 48) | \
|
|
((uint64_t)(__buffer)[7] << 56))
|
|
|
|
/* MySQL protocol constants */
|
|
typedef enum
|
|
{
|
|
GW_MYSQL_CAPABILITIES_NONE= 0,
|
|
GW_MYSQL_CAPABILITIES_LONG_PASSWORD= (1 << 0),
|
|
GW_MYSQL_CAPABILITIES_FOUND_ROWS= (1 << 1),
|
|
GW_MYSQL_CAPABILITIES_LONG_FLAG= (1 << 2),
|
|
GW_MYSQL_CAPABILITIES_CONNECT_WITH_DB= (1 << 3),
|
|
GW_MYSQL_CAPABILITIES_NO_SCHEMA= (1 << 4),
|
|
GW_MYSQL_CAPABILITIES_COMPRESS= (1 << 5),
|
|
GW_MYSQL_CAPABILITIES_ODBC= (1 << 6),
|
|
GW_MYSQL_CAPABILITIES_LOCAL_FILES= (1 << 7),
|
|
GW_MYSQL_CAPABILITIES_IGNORE_SPACE= (1 << 8),
|
|
GW_MYSQL_CAPABILITIES_PROTOCOL_41= (1 << 9),
|
|
GW_MYSQL_CAPABILITIES_INTERACTIVE= (1 << 10),
|
|
GW_MYSQL_CAPABILITIES_SSL= (1 << 11),
|
|
GW_MYSQL_CAPABILITIES_IGNORE_SIGPIPE= (1 << 12),
|
|
GW_MYSQL_CAPABILITIES_TRANSACTIONS= (1 << 13),
|
|
GW_MYSQL_CAPABILITIES_RESERVED= (1 << 14),
|
|
GW_MYSQL_CAPABILITIES_SECURE_CONNECTION= (1 << 15),
|
|
GW_MYSQL_CAPABILITIES_MULTI_STATEMENTS= (1 << 16),
|
|
GW_MYSQL_CAPABILITIES_MULTI_RESULTS= (1 << 17),
|
|
GW_MYSQL_CAPABILITIES_PS_MULTI_RESULTS= (1 << 18),
|
|
GW_MYSQL_CAPABILITIES_PLUGIN_AUTH= (1 << 19),
|
|
GW_MYSQL_CAPABILITIES_SSL_VERIFY_SERVER_CERT= (1 << 30),
|
|
GW_MYSQL_CAPABILITIES_REMEMBER_OPTIONS= (1 << 31),
|
|
GW_MYSQL_CAPABILITIES_CLIENT= (GW_MYSQL_CAPABILITIES_LONG_PASSWORD |
|
|
GW_MYSQL_CAPABILITIES_FOUND_ROWS |
|
|
GW_MYSQL_CAPABILITIES_LONG_FLAG |
|
|
GW_MYSQL_CAPABILITIES_CONNECT_WITH_DB |
|
|
GW_MYSQL_CAPABILITIES_LOCAL_FILES |
|
|
GW_MYSQL_CAPABILITIES_PLUGIN_AUTH |
|
|
GW_MYSQL_CAPABILITIES_TRANSACTIONS |
|
|
GW_MYSQL_CAPABILITIES_PROTOCOL_41 |
|
|
GW_MYSQL_CAPABILITIES_MULTI_STATEMENTS |
|
|
GW_MYSQL_CAPABILITIES_MULTI_RESULTS |
|
|
GW_MYSQL_CAPABILITIES_PS_MULTI_RESULTS |
|
|
GW_MYSQL_CAPABILITIES_SECURE_CONNECTION),
|
|
GW_MYSQL_CAPABILITIES_CLIENT_COMPRESS= (GW_MYSQL_CAPABILITIES_LONG_PASSWORD |
|
|
GW_MYSQL_CAPABILITIES_FOUND_ROWS |
|
|
GW_MYSQL_CAPABILITIES_LONG_FLAG |
|
|
GW_MYSQL_CAPABILITIES_CONNECT_WITH_DB |
|
|
GW_MYSQL_CAPABILITIES_LOCAL_FILES |
|
|
GW_MYSQL_CAPABILITIES_PLUGIN_AUTH |
|
|
GW_MYSQL_CAPABILITIES_TRANSACTIONS |
|
|
GW_MYSQL_CAPABILITIES_PROTOCOL_41 |
|
|
GW_MYSQL_CAPABILITIES_MULTI_STATEMENTS |
|
|
GW_MYSQL_CAPABILITIES_MULTI_RESULTS |
|
|
GW_MYSQL_CAPABILITIES_PS_MULTI_RESULTS |
|
|
GW_MYSQL_CAPABILITIES_COMPRESS
|
|
),
|
|
} gw_mysql_capabilities_t;
|
|
|
|
/* Basic mysql commands */
|
|
#define MYSQL_COM_CHANGE_USER 0x11
|
|
#define MYSQL_COM_QUIT 0x1
|
|
#define MYSQL_COM_INIT_DB 0x2
|
|
#define MYSQL_COM_QUERY 0x3
|
|
|
|
#define MYSQL_GET_COMMAND(payload) (payload[4])
|
|
#define MYSQL_GET_PACKET_NO(payload) (payload[3])
|
|
#define MYSQL_GET_PACKET_LEN(payload) (gw_mysql_get_byte3(payload))
|
|
|
|
#endif
|
|
|
|
void gw_mysql_close(MySQLProtocol **ptr);
|
|
MySQLProtocol* mysql_protocol_init(DCB* dcb, int fd);
|
|
MySQLProtocol *gw_mysql_init(MySQLProtocol *data);
|
|
void gw_mysql_close(MySQLProtocol **ptr);
|
|
bool gw_receive_backend_auth(MySQLProtocol *protocol);
|
|
int gw_decode_mysql_server_handshake(MySQLProtocol *protocol, uint8_t *payload);
|
|
int gw_read_backend_handshake(MySQLProtocol *protocol);
|
|
int gw_send_authentication_to_backend(
|
|
char *dbname,
|
|
char *user,
|
|
uint8_t *passwd,
|
|
MySQLProtocol *protocol);
|
|
const char *gw_mysql_protocol_state2string(int state);
|
|
int gw_do_connect_to_backend(char *host, int port, int* fd);
|
|
int mysql_send_custom_error (
|
|
DCB *dcb,
|
|
int packet_number,
|
|
int in_affected_rows,
|
|
const char* mysql_message);
|
|
int gw_send_change_user_to_backend(
|
|
char *dbname,
|
|
char *user,
|
|
uint8_t *passwd,
|
|
MySQLProtocol *protocol);
|
|
int gw_find_mysql_user_password_sha1(
|
|
char *username,
|
|
uint8_t *gateway_password,
|
|
void *repository);
|
|
int gw_check_mysql_scramble_data(
|
|
DCB *dcb,
|
|
uint8_t *token,
|
|
unsigned int token_len,
|
|
uint8_t *scramble,
|
|
unsigned int scramble_len,
|
|
char *username,
|
|
uint8_t *stage1_hash);
|
|
int mysql_send_auth_error (
|
|
DCB *dcb,
|
|
int packet_number,
|
|
int in_affected_rows,
|
|
const char* mysql_message);
|
|
|
|
void gw_sha1_str(const uint8_t *in, int in_len, uint8_t *out);
|
|
void gw_sha1_2_str(
|
|
const uint8_t *in,
|
|
int in_len,
|
|
const uint8_t *in2,
|
|
int in2_len,
|
|
uint8_t *out);
|
|
void gw_str_xor(
|
|
uint8_t *output,
|
|
const uint8_t *input1,
|
|
const uint8_t *input2,
|
|
unsigned int len);
|
|
char *gw_bin2hex(char *out, const uint8_t *in, unsigned int len);
|
|
int gw_hex2bin(uint8_t *out, const char *in, unsigned int len);
|
|
int gw_generate_random_str(char *output, int len);
|
|
char *gw_strend(register const char *s);
|
|
int setnonblocking(int fd);
|
|
void setipaddress(struct in_addr *a, char *p);
|
|
int gw_read_gwbuff(DCB *dcb, GWBUF **head, int b);
|