Michael Spencer / Mbed 2 deprecated LaOS

Important changes to repositories hosted on mbed.com

Mbed hosted mercurial repositories are deprecated and are due to be permanently deleted in July 2026.

To keep a copy of this software download the repository Zip archive or clone locally using Mercurial.

It is also possible to export all your personal repositories from the account settings page.

Dependencies:   mbed

NetServices/services/mysql/MySQLClient.cpp@1:f5ac63519541, 2014-03-05 (annotated)

Committer:
Michael J. Spencer
Date:
Wed Mar 05 06:14:02 2014 -0800
Revision:
1:f5ac63519541
Initial commit.

Who changed what in which revision?

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