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