This short program illustrates how to use the DS130x_I2C library. My objective is to share the same RTC with Microchip 18F MCU.

Dependencies:   mbed DebugLibrary

Committer:
Yann
Date:
Fri Feb 11 10:17:20 2011 +0000
Revision:
1:995212d326ca
Parent:
0:f30e2135b0db
V0.0.0.2

Who changed what in which revision?

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