FBI FRATER / Mbed 2 deprecated InetTest

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@0:4edb816d21e1, 2015-10-16 (annotated)

Committer:
ICTFBI
Date:
Fri Oct 16 14:28:26 2015 +0000
Revision:
0:4edb816d21e1
Pre-update 16-10-15

Who changed what in which revision?

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