Rik Van Dyck / Mbed 2 deprecated Project_Embedded_C

Project_Embedded_C

Dependencies:   mbed DS1307 Servo TextLCD

NetServices/services/mysql/MySQLClient.cpp@2:55b6fd49b738, 2014-12-18 (annotated)

Committer:
rikvandyck
Date:
Thu Dec 18 10:43:07 2014 +0000
Revision:
2:55b6fd49b738
Parent:
0:e1edd52b1ee2 
Project_Embedded_C

Who changed what in which revision?

UserRevisionLine numberNew contents of line
rikvandyck 0:e1edd52b1ee2 1 #pragma diag_remark 177
rikvandyck 0:e1edd52b1ee2 2 /*
rikvandyck 0:e1edd52b1ee2 3 Copyright (c) 2010 Donatien Garnier (donatiengar [at] gmail [dot] com)
rikvandyck 0:e1edd52b1ee2 4
rikvandyck 0:e1edd52b1ee2 5 Permission is hereby granted, free of charge, to any person obtaining a copy
rikvandyck 0:e1edd52b1ee2 6 of this software and associated documentation files (the "Software"), to deal
rikvandyck 0:e1edd52b1ee2 7 in the Software without restriction, including without limitation the rights
rikvandyck 0:e1edd52b1ee2 8 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
rikvandyck 0:e1edd52b1ee2 9 copies of the Software, and to permit persons to whom the Software is
rikvandyck 0:e1edd52b1ee2 10 furnished to do so, subject to the following conditions:
rikvandyck 0:e1edd52b1ee2 11
rikvandyck 0:e1edd52b1ee2 12 The above copyright notice and this permission notice shall be included in
rikvandyck 0:e1edd52b1ee2 13 all copies or substantial portions of the Software.
rikvandyck 0:e1edd52b1ee2 14
rikvandyck 0:e1edd52b1ee2 15 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
rikvandyck 0:e1edd52b1ee2 16 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
rikvandyck 0:e1edd52b1ee2 17 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
rikvandyck 0:e1edd52b1ee2 18 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
rikvandyck 0:e1edd52b1ee2 19 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
rikvandyck 0:e1edd52b1ee2 20 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
rikvandyck 0:e1edd52b1ee2 21 THE SOFTWARE.
rikvandyck 0:e1edd52b1ee2 22 */
rikvandyck 0:e1edd52b1ee2 23
rikvandyck 0:e1edd52b1ee2 24 #include "MySQLClient.h"
rikvandyck 0:e1edd52b1ee2 25 #include "sha1.h" //For 4.1+ passwords
rikvandyck 0:e1edd52b1ee2 26 #include "mycrypt.h" //For 4.0- passwords
rikvandyck 0:e1edd52b1ee2 27
rikvandyck 0:e1edd52b1ee2 28 //#define __DEBUG
rikvandyck 0:e1edd52b1ee2 29 #include "dbg/dbg.h"
rikvandyck 0:e1edd52b1ee2 30
rikvandyck 0:e1edd52b1ee2 31 #define MYSQL_TIMEOUT_MS 45000
rikvandyck 0:e1edd52b1ee2 32 #define MYSQL_PORT 3306
rikvandyck 0:e1edd52b1ee2 33
rikvandyck 0:e1edd52b1ee2 34 #define BUF_SIZE 256
rikvandyck 0:e1edd52b1ee2 35
rikvandyck 0:e1edd52b1ee2 36 #define CLIENT_LONG_PASSWORD 1
rikvandyck 0:e1edd52b1ee2 37 #define CLIENT_CONNECT_WITH_DB 8
rikvandyck 0:e1edd52b1ee2 38 #define CLIENT_PROTOCOL_41 512
rikvandyck 0:e1edd52b1ee2 39 #define CLIENT_INTERACTIVE 1024
rikvandyck 0:e1edd52b1ee2 40 #define CLIENT_SECURE_CONNECTION 32768
rikvandyck 0:e1edd52b1ee2 41
rikvandyck 0:e1edd52b1ee2 42 #define MIN(a,b) ((a)<(b)?(a):(b))
rikvandyck 0:e1edd52b1ee2 43 #define ABS(a) (((a)>0)?(a):0)
rikvandyck 0:e1edd52b1ee2 44
rikvandyck 0:e1edd52b1ee2 45 //MySQL commands
rikvandyck 0:e1edd52b1ee2 46 #define COM_QUIT 0x01 //Exit
rikvandyck 0:e1edd52b1ee2 47 #define COM_QUERY 0x03 //Execute an SQL query
rikvandyck 0:e1edd52b1ee2 48
rikvandyck 0:e1edd52b1ee2 49 //#define htons( x ) ( (( x << 8 ) & 0xFF00) | (( x >> 8 ) & 0x00FF) )
rikvandyck 0:e1edd52b1ee2 50 #define ntohs( x ) (htons(x))
rikvandyck 0:e1edd52b1ee2 51
rikvandyck 0:e1edd52b1ee2 52 /*#define htonl( x ) ( (( x << 24 ) & 0xFF000000) \
rikvandyck 0:e1edd52b1ee2 53 | (( x << 8 ) & 0x00FF0000) \
rikvandyck 0:e1edd52b1ee2 54 | (( x >> 8 ) & 0x0000FF00) \
rikvandyck 0:e1edd52b1ee2 55 | (( x >> 24 ) & 0x000000FF) )*/
rikvandyck 0:e1edd52b1ee2 56 #define htonl( x ) (x)
rikvandyck 0:e1edd52b1ee2 57 #define ntohl( x ) (htonl(x))
rikvandyck 0:e1edd52b1ee2 58
rikvandyck 0:e1edd52b1ee2 59 MySQLClient::MySQLClient() : NetService(false) /*Not owned by the pool*/, m_pCbItem(NULL), m_pCbMeth(NULL), m_pCb(NULL),
rikvandyck 0:e1edd52b1ee2 60 m_pTCPSocket(NULL), m_watchdog(), m_timeout(MYSQL_TIMEOUT_MS*1000), m_pDnsReq(NULL), m_closed(true),
rikvandyck 0:e1edd52b1ee2 61 m_host(), m_user(), m_password(), m_db(), m_state(MYSQL_CLOSED)
rikvandyck 0:e1edd52b1ee2 62 {
rikvandyck 0:e1edd52b1ee2 63 m_buf = new byte[BUF_SIZE];
rikvandyck 0:e1edd52b1ee2 64 m_pPos = m_buf;
rikvandyck 0:e1edd52b1ee2 65 m_len = 0;
rikvandyck 0:e1edd52b1ee2 66 m_size = BUF_SIZE;
rikvandyck 0:e1edd52b1ee2 67 }
rikvandyck 0:e1edd52b1ee2 68
rikvandyck 0:e1edd52b1ee2 69 MySQLClient::~MySQLClient()
rikvandyck 0:e1edd52b1ee2 70 {
rikvandyck 0:e1edd52b1ee2 71 close();
rikvandyck 0:e1edd52b1ee2 72 delete[] m_buf;
rikvandyck 0:e1edd52b1ee2 73 }
rikvandyck 0:e1edd52b1ee2 74
rikvandyck 0:e1edd52b1ee2 75 //High Level setup functions
rikvandyck 0:e1edd52b1ee2 76 MySQLResult MySQLClient::open(Host& host, const string& user, const string& password, const string& db, void (*pMethod)(MySQLResult)) //Non blocking
rikvandyck 0:e1edd52b1ee2 77 {
rikvandyck 0:e1edd52b1ee2 78 setOnResult(pMethod);
rikvandyck 0:e1edd52b1ee2 79 setup(host, user, password, db);
rikvandyck 0:e1edd52b1ee2 80 return MYSQL_PROCESSING;
rikvandyck 0:e1edd52b1ee2 81 }
rikvandyck 0:e1edd52b1ee2 82
rikvandyck 0:e1edd52b1ee2 83 #if 0 //Ref only
rikvandyck 0:e1edd52b1ee2 84 template<class T>
rikvandyck 0:e1edd52b1ee2 85 MySQLResult MySQLClient::open(Host& host, const string& user, const string& password, const string& db, T* pItem, void (T::*pMethod)(MySQLResult)) //Non blocking
rikvandyck 0:e1edd52b1ee2 86 {
rikvandyck 0:e1edd52b1ee2 87 setOnResult(pItem, pMethod);
rikvandyck 0:e1edd52b1ee2 88 setup(host, user, password, db);
rikvandyck 0:e1edd52b1ee2 89 return MYSQL_PROCESSING;
rikvandyck 0:e1edd52b1ee2 90 }
rikvandyck 0:e1edd52b1ee2 91 #endif
rikvandyck 0:e1edd52b1ee2 92
rikvandyck 0:e1edd52b1ee2 93 MySQLResult MySQLClient::sql(string& sqlCommand)
rikvandyck 0:e1edd52b1ee2 94 {
rikvandyck 0:e1edd52b1ee2 95 if(m_state!=MYSQL_COMMANDS)
rikvandyck 0:e1edd52b1ee2 96 return MYSQL_SETUP;
rikvandyck 0:e1edd52b1ee2 97 sendCommand(COM_QUERY, (byte*)sqlCommand.data(), sqlCommand.length());
rikvandyck 0:e1edd52b1ee2 98 return MYSQL_PROCESSING;
rikvandyck 0:e1edd52b1ee2 99 }
rikvandyck 0:e1edd52b1ee2 100
rikvandyck 0:e1edd52b1ee2 101 MySQLResult MySQLClient::exit()
rikvandyck 0:e1edd52b1ee2 102 {
rikvandyck 0:e1edd52b1ee2 103 sendCommand(COM_QUIT, NULL, 0);
rikvandyck 0:e1edd52b1ee2 104 close();
rikvandyck 0:e1edd52b1ee2 105 return MYSQL_OK;
rikvandyck 0:e1edd52b1ee2 106 }
rikvandyck 0:e1edd52b1ee2 107
rikvandyck 0:e1edd52b1ee2 108 void MySQLClient::setOnResult( void (*pMethod)(MySQLResult) )
rikvandyck 0:e1edd52b1ee2 109 {
rikvandyck 0:e1edd52b1ee2 110 m_pCb = pMethod;
rikvandyck 0:e1edd52b1ee2 111 m_pCbItem = NULL;
rikvandyck 0:e1edd52b1ee2 112 m_pCbMeth = NULL;
rikvandyck 0:e1edd52b1ee2 113 }
rikvandyck 0:e1edd52b1ee2 114
rikvandyck 0:e1edd52b1ee2 115 #if 0 //Ref only
rikvandyck 0:e1edd52b1ee2 116 template<class T>
rikvandyck 0:e1edd52b1ee2 117 void MySQLClient::setOnResult( T* pItem, void (T::*pMethod)(MySQLResult) )
rikvandyck 0:e1edd52b1ee2 118 {
rikvandyck 0:e1edd52b1ee2 119 m_pCb = NULL;
rikvandyck 0:e1edd52b1ee2 120 m_pCbItem = (CDummy*) pItem;
rikvandyck 0:e1edd52b1ee2 121 m_pCbMeth = (void (CDummy::*)(MySQLResult)) pMethod;
rikvandyck 0:e1edd52b1ee2 122 }
rikvandyck 0:e1edd52b1ee2 123 #endif
rikvandyck 0:e1edd52b1ee2 124
rikvandyck 0:e1edd52b1ee2 125 void MySQLClient::setTimeout(int ms)
rikvandyck 0:e1edd52b1ee2 126 {
rikvandyck 0:e1edd52b1ee2 127 m_timeout = 1000*ms;
rikvandyck 0:e1edd52b1ee2 128 }
rikvandyck 0:e1edd52b1ee2 129
rikvandyck 0:e1edd52b1ee2 130 void MySQLClient::poll() //Called by NetServices
rikvandyck 0:e1edd52b1ee2 131 {
rikvandyck 0:e1edd52b1ee2 132 if(m_closed)
rikvandyck 0:e1edd52b1ee2 133 {
rikvandyck 0:e1edd52b1ee2 134 return;
rikvandyck 0:e1edd52b1ee2 135 }
rikvandyck 0:e1edd52b1ee2 136 if(m_watchdog.read_us()>m_timeout)
rikvandyck 0:e1edd52b1ee2 137 {
rikvandyck 0:e1edd52b1ee2 138 onTimeout();
rikvandyck 0:e1edd52b1ee2 139 }
rikvandyck 0:e1edd52b1ee2 140 }
rikvandyck 0:e1edd52b1ee2 141
rikvandyck 0:e1edd52b1ee2 142 void MySQLClient::resetTimeout()
rikvandyck 0:e1edd52b1ee2 143 {
rikvandyck 0:e1edd52b1ee2 144 m_watchdog.reset();
rikvandyck 0:e1edd52b1ee2 145 m_watchdog.start();
rikvandyck 0:e1edd52b1ee2 146 }
rikvandyck 0:e1edd52b1ee2 147
rikvandyck 0:e1edd52b1ee2 148 void MySQLClient::init()
rikvandyck 0:e1edd52b1ee2 149 {
rikvandyck 0:e1edd52b1ee2 150 close(); //Remove previous elements
rikvandyck 0:e1edd52b1ee2 151 if(!m_closed) //Already opened
rikvandyck 0:e1edd52b1ee2 152 return;
rikvandyck 0:e1edd52b1ee2 153 m_state = MYSQL_HANDSHAKE;
rikvandyck 0:e1edd52b1ee2 154 m_pTCPSocket = new TCPSocket;
rikvandyck 0:e1edd52b1ee2 155 m_pTCPSocket->setOnEvent(this, &MySQLClient::onTCPSocketEvent);
rikvandyck 0:e1edd52b1ee2 156 m_closed = false;
rikvandyck 0:e1edd52b1ee2 157 }
rikvandyck 0:e1edd52b1ee2 158
rikvandyck 0:e1edd52b1ee2 159 void MySQLClient::close()
rikvandyck 0:e1edd52b1ee2 160 {
rikvandyck 0:e1edd52b1ee2 161 if(m_closed)
rikvandyck 0:e1edd52b1ee2 162 return;
rikvandyck 0:e1edd52b1ee2 163 m_state = MYSQL_CLOSED;
rikvandyck 0:e1edd52b1ee2 164 m_closed = true; //Prevent recursive calling or calling on an object being destructed by someone else
rikvandyck 0:e1edd52b1ee2 165 m_watchdog.stop(); //Stop timeout
rikvandyck 0:e1edd52b1ee2 166 m_watchdog.reset();
rikvandyck 0:e1edd52b1ee2 167 m_pTCPSocket->resetOnEvent();
rikvandyck 0:e1edd52b1ee2 168 m_pTCPSocket->close();
rikvandyck 0:e1edd52b1ee2 169 delete m_pTCPSocket;
rikvandyck 0:e1edd52b1ee2 170 m_pTCPSocket = NULL;
rikvandyck 0:e1edd52b1ee2 171 if( m_pDnsReq )
rikvandyck 0:e1edd52b1ee2 172 {
rikvandyck 0:e1edd52b1ee2 173 m_pDnsReq->close();
rikvandyck 0:e1edd52b1ee2 174 delete m_pDnsReq;
rikvandyck 0:e1edd52b1ee2 175 m_pDnsReq = NULL;
rikvandyck 0:e1edd52b1ee2 176 }
rikvandyck 0:e1edd52b1ee2 177 }
rikvandyck 0:e1edd52b1ee2 178
rikvandyck 0:e1edd52b1ee2 179 void MySQLClient::setup(Host& host, const string& user, const string& password, const string& db) //Setup connection, make DNS Req if necessary
rikvandyck 0:e1edd52b1ee2 180 {
rikvandyck 0:e1edd52b1ee2 181 init(); //Initialize client in known state, create socket
rikvandyck 0:e1edd52b1ee2 182 resetTimeout();
rikvandyck 0:e1edd52b1ee2 183 m_host = host;
rikvandyck 0:e1edd52b1ee2 184 if(!host.getPort())
rikvandyck 0:e1edd52b1ee2 185 host.setPort( MYSQL_PORT ); //Default port
rikvandyck 0:e1edd52b1ee2 186
rikvandyck 0:e1edd52b1ee2 187 m_user = user;
rikvandyck 0:e1edd52b1ee2 188 m_password = password;
rikvandyck 0:e1edd52b1ee2 189
rikvandyck 0:e1edd52b1ee2 190 m_db = db;
rikvandyck 0:e1edd52b1ee2 191
rikvandyck 0:e1edd52b1ee2 192 if( !host.getIp().isNull() )
rikvandyck 0:e1edd52b1ee2 193 {
rikvandyck 0:e1edd52b1ee2 194 connect();
rikvandyck 0:e1edd52b1ee2 195 }
rikvandyck 0:e1edd52b1ee2 196 else //Need to do a DNS Query...
rikvandyck 0:e1edd52b1ee2 197 {
rikvandyck 0:e1edd52b1ee2 198 DBG("DNS Query...\n");
rikvandyck 0:e1edd52b1ee2 199 m_pDnsReq = new DNSRequest();
rikvandyck 0:e1edd52b1ee2 200 m_pDnsReq->setOnReply(this, &MySQLClient::onDNSReply);
rikvandyck 0:e1edd52b1ee2 201 m_pDnsReq->resolve(&m_host);
rikvandyck 0:e1edd52b1ee2 202 DBG("MySQLClient : DNSRequest %p\n", m_pDnsReq);
rikvandyck 0:e1edd52b1ee2 203 }
rikvandyck 0:e1edd52b1ee2 204 }
rikvandyck 0:e1edd52b1ee2 205
rikvandyck 0:e1edd52b1ee2 206 void MySQLClient::connect() //Start Connection
rikvandyck 0:e1edd52b1ee2 207 {
rikvandyck 0:e1edd52b1ee2 208 resetTimeout();
rikvandyck 0:e1edd52b1ee2 209 DBG("Connecting...\n");
rikvandyck 0:e1edd52b1ee2 210 m_pTCPSocket->connect(m_host);
rikvandyck 0:e1edd52b1ee2 211 m_packetId = 0;
rikvandyck 0:e1edd52b1ee2 212 }
rikvandyck 0:e1edd52b1ee2 213
rikvandyck 0:e1edd52b1ee2 214 void MySQLClient::handleHandshake()
rikvandyck 0:e1edd52b1ee2 215 {
rikvandyck 0:e1edd52b1ee2 216 readData();
rikvandyck 0:e1edd52b1ee2 217 if( ! (( m_len > 1 ) && ( memchr( m_buf + 1, 0, m_len ) != NULL )) )
rikvandyck 0:e1edd52b1ee2 218 {
rikvandyck 0:e1edd52b1ee2 219 DBG("Connected but could not find pcsz...\n");
rikvandyck 0:e1edd52b1ee2 220 onResult(MYSQL_PRTCL);
rikvandyck 0:e1edd52b1ee2 221 return;
rikvandyck 0:e1edd52b1ee2 222 }
rikvandyck 0:e1edd52b1ee2 223
rikvandyck 0:e1edd52b1ee2 224 DBG("Connected to server: %d bytes read ; Protocol version %d, mysql-%s.\n", m_len, m_buf[0], &m_buf[1]);
rikvandyck 0:e1edd52b1ee2 225
rikvandyck 0:e1edd52b1ee2 226 m_pPos = (byte*) memchr( (char*)(m_buf + 1), 0, m_len ) + 1;
rikvandyck 0:e1edd52b1ee2 227
rikvandyck 0:e1edd52b1ee2 228 sendAuth();
rikvandyck 0:e1edd52b1ee2 229 }
rikvandyck 0:e1edd52b1ee2 230
rikvandyck 0:e1edd52b1ee2 231 void MySQLClient::sendAuth()
rikvandyck 0:e1edd52b1ee2 232 {
rikvandyck 0:e1edd52b1ee2 233 if( m_len - (m_pPos - m_buf) != 44)
rikvandyck 0:e1edd52b1ee2 234 {
rikvandyck 0:e1edd52b1ee2 235 //We only support protocol >= mysql-4.1
rikvandyck 0:e1edd52b1ee2 236 DBG("Message after pcsz has wrong len (%d != 44)...\n", m_len - (m_pPos - m_buf));
rikvandyck 0:e1edd52b1ee2 237 onResult(MYSQL_PRTCL);
rikvandyck 0:e1edd52b1ee2 238 return;
rikvandyck 0:e1edd52b1ee2 239 }
rikvandyck 0:e1edd52b1ee2 240
rikvandyck 0:e1edd52b1ee2 241 uint16_t serverFlags = *((uint16_t*)&m_pPos[13]);
rikvandyck 0:e1edd52b1ee2 242 DBG("Server capabilities are %04X.\n", serverFlags);
rikvandyck 0:e1edd52b1ee2 243
rikvandyck 0:e1edd52b1ee2 244 uint32_t clientFlags = CLIENT_CONNECT_WITH_DB | CLIENT_PROTOCOL_41 | CLIENT_SECURE_CONNECTION | CLIENT_INTERACTIVE;;
rikvandyck 0:e1edd52b1ee2 245
rikvandyck 0:e1edd52b1ee2 246 //if(serverFlags & CLIENT_LONG_PASSWORD)
rikvandyck 0:e1edd52b1ee2 247
rikvandyck 0:e1edd52b1ee2 248 DBG("Using auth 4.1+\n");
rikvandyck 0:e1edd52b1ee2 249 //Encrypt pw using scramble
rikvandyck 0:e1edd52b1ee2 250 byte scramble[20+20]={0};
rikvandyck 0:e1edd52b1ee2 251 memcpy(scramble, m_pPos+4, 8);
rikvandyck 0:e1edd52b1ee2 252 memcpy(scramble+8, m_pPos+31, 12); // *(m_pPos+43) == 0 (zero-terminated char*)
rikvandyck 0:e1edd52b1ee2 253
rikvandyck 0:e1edd52b1ee2 254 byte stage1_hash[20] = {0};
rikvandyck 0:e1edd52b1ee2 255 sha1( (byte*)m_password.data(), m_password.length(), stage1_hash );
rikvandyck 0:e1edd52b1ee2 256
rikvandyck 0:e1edd52b1ee2 257 sha1( stage1_hash, 20, ((byte*)scramble + 20) );
rikvandyck 0:e1edd52b1ee2 258
rikvandyck 0:e1edd52b1ee2 259 byte token[20] = {0};
rikvandyck 0:e1edd52b1ee2 260 sha1( scramble, 40, token );
rikvandyck 0:e1edd52b1ee2 261
rikvandyck 0:e1edd52b1ee2 262 for(int i=0;i<20;i++)
rikvandyck 0:e1edd52b1ee2 263 token[i] = token[i] ^ stage1_hash[i];
rikvandyck 0:e1edd52b1ee2 264
rikvandyck 0:e1edd52b1ee2 265 clientFlags |= CLIENT_LONG_PASSWORD;
rikvandyck 0:e1edd52b1ee2 266
rikvandyck 0:e1edd52b1ee2 267 DBG("Building response\n");
rikvandyck 0:e1edd52b1ee2 268 //Build response
rikvandyck 0:e1edd52b1ee2 269
rikvandyck 0:e1edd52b1ee2 270 //BE
rikvandyck 0:e1edd52b1ee2 271 *((uint32_t*)&m_buf[0]) = htonl(clientFlags);
rikvandyck 0:e1edd52b1ee2 272 *((uint32_t*)&m_buf[4]) = BUF_SIZE; //Max packets size
rikvandyck 0:e1edd52b1ee2 273 m_buf[8] = 8; //latin1 charset
rikvandyck 0:e1edd52b1ee2 274 memset((char*)(m_buf+9),0,23);
rikvandyck 0:e1edd52b1ee2 275 strcpy((char*)(m_buf+32),m_user.c_str());
rikvandyck 0:e1edd52b1ee2 276 m_pPos = m_buf + 32 + m_user.length() + 1;
rikvandyck 0:e1edd52b1ee2 277 m_pPos[0] = 20;
rikvandyck 0:e1edd52b1ee2 278 memcpy((char*)&m_pPos[1],token,20);
rikvandyck 0:e1edd52b1ee2 279 strcpy((char*)(m_pPos+21),m_db.c_str());
rikvandyck 0:e1edd52b1ee2 280 m_len = 32 + m_user.length() + 1 + 21 + m_db.length() + 1;
rikvandyck 0:e1edd52b1ee2 281
rikvandyck 0:e1edd52b1ee2 282 //Save first part of scramble in case we need it again
rikvandyck 0:e1edd52b1ee2 283 memcpy(&m_buf[BUF_SIZE-8], scramble, 8);
rikvandyck 0:e1edd52b1ee2 284
rikvandyck 0:e1edd52b1ee2 285 m_state = MYSQL_AUTH;
rikvandyck 0:e1edd52b1ee2 286
rikvandyck 0:e1edd52b1ee2 287 DBG("Writing data\n");
rikvandyck 0:e1edd52b1ee2 288 writeData();
rikvandyck 0:e1edd52b1ee2 289 }
rikvandyck 0:e1edd52b1ee2 290
rikvandyck 0:e1edd52b1ee2 291 void MySQLClient::handleAuthResult()
rikvandyck 0:e1edd52b1ee2 292 {
rikvandyck 0:e1edd52b1ee2 293 readData();
rikvandyck 0:e1edd52b1ee2 294 if(m_len==1 && *m_buf==0xfe)
rikvandyck 0:e1edd52b1ee2 295 {
rikvandyck 0:e1edd52b1ee2 296 //Re-send auth using 4.0- auth
rikvandyck 0:e1edd52b1ee2 297 sendAuth323();
rikvandyck 0:e1edd52b1ee2 298 return;
rikvandyck 0:e1edd52b1ee2 299 }
rikvandyck 0:e1edd52b1ee2 300 m_watchdog.stop(); //Stop timeout
rikvandyck 0:e1edd52b1ee2 301 m_watchdog.reset();
rikvandyck 0:e1edd52b1ee2 302 if(m_len<2)
rikvandyck 0:e1edd52b1ee2 303 {
rikvandyck 0:e1edd52b1ee2 304 DBG("Response too short..\n");
rikvandyck 0:e1edd52b1ee2 305 onResult(MYSQL_PRTCL);
rikvandyck 0:e1edd52b1ee2 306 return;
rikvandyck 0:e1edd52b1ee2 307 }
rikvandyck 0:e1edd52b1ee2 308 DBG("RC=%d ",m_buf[0]);
rikvandyck 0:e1edd52b1ee2 309 if(m_buf[0]==0)
rikvandyck 0:e1edd52b1ee2 310 {
rikvandyck 0:e1edd52b1ee2 311 m_buf[m_len] = 0;
rikvandyck 0:e1edd52b1ee2 312 m_pPos = m_buf + 1;
rikvandyck 0:e1edd52b1ee2 313 m_pPos += m_buf[1] +1;
rikvandyck 0:e1edd52b1ee2 314 m_pPos += m_pPos[0];
rikvandyck 0:e1edd52b1ee2 315 m_pPos += 1;
rikvandyck 0:e1edd52b1ee2 316 DBG("(OK) : Server status %d, Message : %s\n", *((uint16_t*)&m_pPos[0]), m_pPos+4);
rikvandyck 0:e1edd52b1ee2 317 onResult(MYSQL_OK);
rikvandyck 0:e1edd52b1ee2 318 }
rikvandyck 0:e1edd52b1ee2 319 else
rikvandyck 0:e1edd52b1ee2 320 {
rikvandyck 0:e1edd52b1ee2 321 m_buf[m_len] = 0;
rikvandyck 0:e1edd52b1ee2 322 DBG("(Error %d) : %s\n", *((uint16_t*)&m_buf[1]), &m_buf[9]); //LE
rikvandyck 0:e1edd52b1ee2 323 onResult(MYSQL_AUTHFAILED);
rikvandyck 0:e1edd52b1ee2 324 return;
rikvandyck 0:e1edd52b1ee2 325 }
rikvandyck 0:e1edd52b1ee2 326 m_state = MYSQL_COMMANDS;
rikvandyck 0:e1edd52b1ee2 327 }
rikvandyck 0:e1edd52b1ee2 328
rikvandyck 0:e1edd52b1ee2 329 void MySQLClient::sendAuth323()
rikvandyck 0:e1edd52b1ee2 330 {
rikvandyck 0:e1edd52b1ee2 331 DBG("Using auth 4.0-\n");
rikvandyck 0:e1edd52b1ee2 332 byte scramble[8]={0};
rikvandyck 0:e1edd52b1ee2 333
rikvandyck 0:e1edd52b1ee2 334 memcpy(scramble, &m_buf[BUF_SIZE-8], 8); //Recover scramble
rikvandyck 0:e1edd52b1ee2 335
rikvandyck 0:e1edd52b1ee2 336 //memcpy(scramble+8, m_pPos+31, 12); // *(m_pPos+43) == 0 (zero-terminated char*)
rikvandyck 0:e1edd52b1ee2 337
rikvandyck 0:e1edd52b1ee2 338 byte token[9]={0};
rikvandyck 0:e1edd52b1ee2 339
rikvandyck 0:e1edd52b1ee2 340 scramble_323((char*)token, (const char*)scramble, m_password.c_str());
rikvandyck 0:e1edd52b1ee2 341
rikvandyck 0:e1edd52b1ee2 342 DBG("Building response\n");
rikvandyck 0:e1edd52b1ee2 343 //Build response
rikvandyck 0:e1edd52b1ee2 344
rikvandyck 0:e1edd52b1ee2 345 memcpy((char*)m_buf,token,9);
rikvandyck 0:e1edd52b1ee2 346 m_len = 9;
rikvandyck 0:e1edd52b1ee2 347
rikvandyck 0:e1edd52b1ee2 348 #if 0
rikvandyck 0:e1edd52b1ee2 349 *((uint32_t*)&m_buf[0]) = htonl(clientFlags);
rikvandyck 0:e1edd52b1ee2 350 *((uint32_t*)&m_buf[4]) = BUF_SIZE; //Max packets size
rikvandyck 0:e1edd52b1ee2 351 m_buf[8] = 8; //latin1 charset
rikvandyck 0:e1edd52b1ee2 352 memset((char*)(m_buf+9),0,23);
rikvandyck 0:e1edd52b1ee2 353 strcpy((char*)(m_buf+32),m_user.c_str());
rikvandyck 0:e1edd52b1ee2 354 m_pPos = m_buf + 32 + m_user.length() + 1;
rikvandyck 0:e1edd52b1ee2 355 m_pPos[0] = 8;
rikvandyck 0:e1edd52b1ee2 356 memcpy((char*)&m_pPos[1],token+1,8);
rikvandyck 0:e1edd52b1ee2 357 strcpy((char*)(m_pPos+9),m_db.c_str());
rikvandyck 0:e1edd52b1ee2 358 m_len = 32 + m_user.length() + 1 + 9 + m_db.length() + 1;
rikvandyck 0:e1edd52b1ee2 359 #endif
rikvandyck 0:e1edd52b1ee2 360
rikvandyck 0:e1edd52b1ee2 361 DBG("Writing data\n");
rikvandyck 0:e1edd52b1ee2 362 writeData();
rikvandyck 0:e1edd52b1ee2 363 }
rikvandyck 0:e1edd52b1ee2 364
rikvandyck 0:e1edd52b1ee2 365 void MySQLClient::sendCommand(byte command, byte* arg, int len)
rikvandyck 0:e1edd52b1ee2 366 {
rikvandyck 0:e1edd52b1ee2 367 DBG("Sending command %d, payload of len %d\n", command, len);
rikvandyck 0:e1edd52b1ee2 368 m_packetId=0;//Reset packet ID (New sequence)
rikvandyck 0:e1edd52b1ee2 369 m_buf[0] = command;
rikvandyck 0:e1edd52b1ee2 370 memcpy(&m_buf[1], arg, len);
rikvandyck 0:e1edd52b1ee2 371 m_len = 1 + len;
rikvandyck 0:e1edd52b1ee2 372 writeData();
rikvandyck 0:e1edd52b1ee2 373 m_watchdog.start();
rikvandyck 0:e1edd52b1ee2 374 }
rikvandyck 0:e1edd52b1ee2 375
rikvandyck 0:e1edd52b1ee2 376 void MySQLClient::handleCommandResult()
rikvandyck 0:e1edd52b1ee2 377 {
rikvandyck 0:e1edd52b1ee2 378 readData();
rikvandyck 0:e1edd52b1ee2 379 m_watchdog.stop(); //Stop timeout
rikvandyck 0:e1edd52b1ee2 380 m_watchdog.reset();
rikvandyck 0:e1edd52b1ee2 381 if(m_len<2)
rikvandyck 0:e1edd52b1ee2 382 {
rikvandyck 0:e1edd52b1ee2 383 DBG("Response too short..\n");
rikvandyck 0:e1edd52b1ee2 384 onResult(MYSQL_PRTCL);
rikvandyck 0:e1edd52b1ee2 385 return;
rikvandyck 0:e1edd52b1ee2 386 }
rikvandyck 0:e1edd52b1ee2 387 DBG("RC=%d ",m_buf[0]);
rikvandyck 0:e1edd52b1ee2 388 if(m_buf[0]==0)
rikvandyck 0:e1edd52b1ee2 389 {
rikvandyck 0:e1edd52b1ee2 390 DBG("(OK)\n");
rikvandyck 0:e1edd52b1ee2 391 onResult(MYSQL_OK);
rikvandyck 0:e1edd52b1ee2 392 }
rikvandyck 0:e1edd52b1ee2 393 else
rikvandyck 0:e1edd52b1ee2 394 {
rikvandyck 0:e1edd52b1ee2 395 m_buf[m_len] = 0;
rikvandyck 0:e1edd52b1ee2 396 DBG("(SQL Error %d) : %s\n", *((uint16_t*)&m_buf[1]), &m_buf[9]); //LE
rikvandyck 0:e1edd52b1ee2 397 onResult(MYSQL_SQL);
rikvandyck 0:e1edd52b1ee2 398 return;
rikvandyck 0:e1edd52b1ee2 399 }
rikvandyck 0:e1edd52b1ee2 400 }
rikvandyck 0:e1edd52b1ee2 401
rikvandyck 0:e1edd52b1ee2 402 void MySQLClient::readData() //Copy to buf
rikvandyck 0:e1edd52b1ee2 403 {
rikvandyck 0:e1edd52b1ee2 404 byte head[4];
rikvandyck 0:e1edd52b1ee2 405 int ret = m_pTCPSocket->recv((char*)head, 4); //Packet header
rikvandyck 0:e1edd52b1ee2 406 m_len = *((uint16_t*)&head[0]);
rikvandyck 0:e1edd52b1ee2 407 m_packetId = head[3];
rikvandyck 0:e1edd52b1ee2 408 DBG("Packet Id %d of length %d\n", head[3], m_len);
rikvandyck 0:e1edd52b1ee2 409 m_packetId++;
rikvandyck 0:e1edd52b1ee2 410 if(ret>0)
rikvandyck 0:e1edd52b1ee2 411 ret = m_pTCPSocket->recv((char*)m_buf, m_len);
rikvandyck 0:e1edd52b1ee2 412 if(ret < 0)//Error
rikvandyck 0:e1edd52b1ee2 413 {
rikvandyck 0:e1edd52b1ee2 414 onResult(MYSQL_CONN);
rikvandyck 0:e1edd52b1ee2 415 return;
rikvandyck 0:e1edd52b1ee2 416 }
rikvandyck 0:e1edd52b1ee2 417 if(ret < m_len)
rikvandyck 0:e1edd52b1ee2 418 {
rikvandyck 0:e1edd52b1ee2 419 DBG("WARN: Incomplete packet\n");
rikvandyck 0:e1edd52b1ee2 420 }
rikvandyck 0:e1edd52b1ee2 421 m_len = ret;
rikvandyck 0:e1edd52b1ee2 422 }
rikvandyck 0:e1edd52b1ee2 423
rikvandyck 0:e1edd52b1ee2 424 void MySQLClient::writeData() //Copy from buf
rikvandyck 0:e1edd52b1ee2 425 {
rikvandyck 0:e1edd52b1ee2 426 byte head[4] = { 0 };
rikvandyck 0:e1edd52b1ee2 427 *((uint16_t*)&head[0]) = m_len;
rikvandyck 0:e1edd52b1ee2 428 head[3] = m_packetId;
rikvandyck 0:e1edd52b1ee2 429 DBG("Packet Id %d\n", head[3]);
rikvandyck 0:e1edd52b1ee2 430 m_packetId++;
rikvandyck 0:e1edd52b1ee2 431 int ret = m_pTCPSocket->send((char*)head, 4); //Packet header
rikvandyck 0:e1edd52b1ee2 432 if(ret>0)
rikvandyck 0:e1edd52b1ee2 433 ret = m_pTCPSocket->send((char*)m_buf, m_len);
rikvandyck 0:e1edd52b1ee2 434 if(ret < 0)//Error
rikvandyck 0:e1edd52b1ee2 435 {
rikvandyck 0:e1edd52b1ee2 436 onResult(MYSQL_CONN);
rikvandyck 0:e1edd52b1ee2 437 return;
rikvandyck 0:e1edd52b1ee2 438 }
rikvandyck 0:e1edd52b1ee2 439 m_len = 0;//FIXME... incomplete packets handling
rikvandyck 0:e1edd52b1ee2 440 }
rikvandyck 0:e1edd52b1ee2 441
rikvandyck 0:e1edd52b1ee2 442 void MySQLClient::onTCPSocketEvent(TCPSocketEvent e)
rikvandyck 0:e1edd52b1ee2 443 {
rikvandyck 0:e1edd52b1ee2 444 DBG("Event %d in MySQLClient::onTCPSocketEvent()\n", e);
rikvandyck 0:e1edd52b1ee2 445
rikvandyck 0:e1edd52b1ee2 446 if(m_closed)
rikvandyck 0:e1edd52b1ee2 447 {
rikvandyck 0:e1edd52b1ee2 448 DBG("WARN: Discarded\n");
rikvandyck 0:e1edd52b1ee2 449 return;
rikvandyck 0:e1edd52b1ee2 450 }
rikvandyck 0:e1edd52b1ee2 451
rikvandyck 0:e1edd52b1ee2 452 switch(e)
rikvandyck 0:e1edd52b1ee2 453 {
rikvandyck 0:e1edd52b1ee2 454 case TCPSOCKET_READABLE: //Incoming data
rikvandyck 0:e1edd52b1ee2 455 resetTimeout();
rikvandyck 0:e1edd52b1ee2 456 if(m_state == MYSQL_HANDSHAKE)
rikvandyck 0:e1edd52b1ee2 457 handleHandshake();
rikvandyck 0:e1edd52b1ee2 458 else if(m_state == MYSQL_AUTH)
rikvandyck 0:e1edd52b1ee2 459 handleAuthResult();
rikvandyck 0:e1edd52b1ee2 460 else if(m_state == MYSQL_COMMANDS)
rikvandyck 0:e1edd52b1ee2 461 handleCommandResult();
rikvandyck 0:e1edd52b1ee2 462 break;
rikvandyck 0:e1edd52b1ee2 463 case TCPSOCKET_WRITEABLE: //We can send data
rikvandyck 0:e1edd52b1ee2 464 resetTimeout();
rikvandyck 0:e1edd52b1ee2 465 break;
rikvandyck 0:e1edd52b1ee2 466 case TCPSOCKET_CONNECTED: //Connected, wait for handshake packet
rikvandyck 0:e1edd52b1ee2 467 resetTimeout();
rikvandyck 0:e1edd52b1ee2 468 break;
rikvandyck 0:e1edd52b1ee2 469 case TCPSOCKET_CONTIMEOUT:
rikvandyck 0:e1edd52b1ee2 470 case TCPSOCKET_CONRST:
rikvandyck 0:e1edd52b1ee2 471 case TCPSOCKET_CONABRT:
rikvandyck 0:e1edd52b1ee2 472 case TCPSOCKET_ERROR:
rikvandyck 0:e1edd52b1ee2 473 DBG("Connection error.\n");
rikvandyck 0:e1edd52b1ee2 474 onResult(MYSQL_CONN);
rikvandyck 0:e1edd52b1ee2 475 case TCPSOCKET_DISCONNECTED:
rikvandyck 0:e1edd52b1ee2 476 //There might still be some data available for reading
rikvandyck 0:e1edd52b1ee2 477 //So if we are in a reading state, do not close the socket yet
rikvandyck 0:e1edd52b1ee2 478 if(m_state != MYSQL_CLOSED)
rikvandyck 0:e1edd52b1ee2 479 {
rikvandyck 0:e1edd52b1ee2 480 onResult(MYSQL_CONN);
rikvandyck 0:e1edd52b1ee2 481 }
rikvandyck 0:e1edd52b1ee2 482 DBG("Connection closed by remote host.\n");
rikvandyck 0:e1edd52b1ee2 483 break;
rikvandyck 0:e1edd52b1ee2 484 }
rikvandyck 0:e1edd52b1ee2 485 }
rikvandyck 0:e1edd52b1ee2 486
rikvandyck 0:e1edd52b1ee2 487 void MySQLClient::onDNSReply(DNSReply r)
rikvandyck 0:e1edd52b1ee2 488 {
rikvandyck 0:e1edd52b1ee2 489 if(m_closed)
rikvandyck 0:e1edd52b1ee2 490 {
rikvandyck 0:e1edd52b1ee2 491 DBG("WARN: Discarded\n");
rikvandyck 0:e1edd52b1ee2 492 return;
rikvandyck 0:e1edd52b1ee2 493 }
rikvandyck 0:e1edd52b1ee2 494
rikvandyck 0:e1edd52b1ee2 495 if( r != DNS_FOUND )
rikvandyck 0:e1edd52b1ee2 496 {
rikvandyck 0:e1edd52b1ee2 497 DBG("Could not resolve hostname.\n");
rikvandyck 0:e1edd52b1ee2 498 onResult(MYSQL_DNS);
rikvandyck 0:e1edd52b1ee2 499 return;
rikvandyck 0:e1edd52b1ee2 500 }
rikvandyck 0:e1edd52b1ee2 501
rikvandyck 0:e1edd52b1ee2 502 DBG("DNS Resolved to %d.%d.%d.%d.\n",m_host.getIp()[0],m_host.getIp()[1],m_host.getIp()[2],m_host.getIp()[3]);
rikvandyck 0:e1edd52b1ee2 503 //If no error, m_host has been updated by m_pDnsReq so we're set to go !
rikvandyck 0:e1edd52b1ee2 504 m_pDnsReq->close();
rikvandyck 0:e1edd52b1ee2 505 delete m_pDnsReq;
rikvandyck 0:e1edd52b1ee2 506 m_pDnsReq = NULL;
rikvandyck 0:e1edd52b1ee2 507 connect();
rikvandyck 0:e1edd52b1ee2 508 }
rikvandyck 0:e1edd52b1ee2 509
rikvandyck 0:e1edd52b1ee2 510 void MySQLClient::onResult(MySQLResult r) //Called when exchange completed or on failure
rikvandyck 0:e1edd52b1ee2 511 {
rikvandyck 0:e1edd52b1ee2 512 if(m_pCbItem && m_pCbMeth)
rikvandyck 0:e1edd52b1ee2 513 (m_pCbItem->*m_pCbMeth)(r);
rikvandyck 0:e1edd52b1ee2 514 else if(m_pCb)
rikvandyck 0:e1edd52b1ee2 515 m_pCb(r);
rikvandyck 0:e1edd52b1ee2 516
rikvandyck 0:e1edd52b1ee2 517 if( (r==MYSQL_DNS) || (r==MYSQL_PRTCL) || (r==MYSQL_AUTHFAILED) || (r==MYSQL_TIMEOUT) || (r==MYSQL_CONN) ) //Fatal error, close connection
rikvandyck 0:e1edd52b1ee2 518 close();
rikvandyck 0:e1edd52b1ee2 519 }
rikvandyck 0:e1edd52b1ee2 520
rikvandyck 0:e1edd52b1ee2 521 void MySQLClient::onTimeout() //Connection has timed out
rikvandyck 0:e1edd52b1ee2 522 {
rikvandyck 0:e1edd52b1ee2 523 DBG("Timed out.\n");
rikvandyck 0:e1edd52b1ee2 524 onResult(MYSQL_TIMEOUT);
rikvandyck 0:e1edd52b1ee2 525 close();
rikvandyck 0:e1edd52b1ee2 526 }