Rik Van Dyck / Mbed 2 deprecated Project_Embedded_C

Dependencies:   mbed DS1307 Servo TextLCD

Committer:
rikvandyck
Date:
Thu Dec 04 10:44:07 2014 +0000
Revision:
0:e1edd52b1ee2
test

Who changed what in which revision?

UserRevisionLine numberNew contents of line
rikvandyck 0:e1edd52b1ee2 1 #pragma diag_remark 1293
rikvandyck 0:e1edd52b1ee2 2 /*
rikvandyck 0:e1edd52b1ee2 3 Copyright (c) 2010 Donatien Garnier (donatiengar [at] gmail [dot] com)
rikvandyck 0:e1edd52b1ee2 4
rikvandyck 0:e1edd52b1ee2 5 Permission is hereby granted, free of charge, to any person obtaining a copy
rikvandyck 0:e1edd52b1ee2 6 of this software and associated documentation files (the "Software"), to deal
rikvandyck 0:e1edd52b1ee2 7 in the Software without restriction, including without limitation the rights
rikvandyck 0:e1edd52b1ee2 8 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
rikvandyck 0:e1edd52b1ee2 9 copies of the Software, and to permit persons to whom the Software is
rikvandyck 0:e1edd52b1ee2 10 furnished to do so, subject to the following conditions:
rikvandyck 0:e1edd52b1ee2 11
rikvandyck 0:e1edd52b1ee2 12 The above copyright notice and this permission notice shall be included in
rikvandyck 0:e1edd52b1ee2 13 all copies or substantial portions of the Software.
rikvandyck 0:e1edd52b1ee2 14
rikvandyck 0:e1edd52b1ee2 15 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
rikvandyck 0:e1edd52b1ee2 16 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
rikvandyck 0:e1edd52b1ee2 17 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
rikvandyck 0:e1edd52b1ee2 18 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
rikvandyck 0:e1edd52b1ee2 19 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
rikvandyck 0:e1edd52b1ee2 20 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
rikvandyck 0:e1edd52b1ee2 21 THE SOFTWARE.
rikvandyck 0:e1edd52b1ee2 22 */
rikvandyck 0:e1edd52b1ee2 23
rikvandyck 0:e1edd52b1ee2 24 #include "NTPClient.h"
rikvandyck 0:e1edd52b1ee2 25
rikvandyck 0:e1edd52b1ee2 26 #include <stdio.h>
rikvandyck 0:e1edd52b1ee2 27
rikvandyck 0:e1edd52b1ee2 28 //#define __DEBUG
rikvandyck 0:e1edd52b1ee2 29 #include "dbg/dbg.h"
rikvandyck 0:e1edd52b1ee2 30
rikvandyck 0:e1edd52b1ee2 31 #define NTP_PORT 123
rikvandyck 0:e1edd52b1ee2 32 #define NTP_CLIENT_PORT 0//50420 //Random port
rikvandyck 0:e1edd52b1ee2 33 #define NTP_REQUEST_TIMEOUT 15000
rikvandyck 0:e1edd52b1ee2 34 #define NTP_TIMESTAMP_DELTA 2208988800ull //Diff btw a UNIX timestamp (Starting Jan, 1st 1970) and a NTP timestamp (Starting Jan, 1st 1900)
rikvandyck 0:e1edd52b1ee2 35
rikvandyck 0:e1edd52b1ee2 36 #define htons( x ) ( (( x << 8 ) & 0xFF00) | (( x >> 8 ) & 0x00FF) )
rikvandyck 0:e1edd52b1ee2 37 #define ntohs( x ) (htons(x))
rikvandyck 0:e1edd52b1ee2 38
rikvandyck 0:e1edd52b1ee2 39 #define htonl( x ) ( (( x << 24 ) & 0xFF000000) \
rikvandyck 0:e1edd52b1ee2 40 | (( x << 8 ) & 0x00FF0000) \
rikvandyck 0:e1edd52b1ee2 41 | (( x >> 8 ) & 0x0000FF00) \
rikvandyck 0:e1edd52b1ee2 42 | (( x >> 24 ) & 0x000000FF) )
rikvandyck 0:e1edd52b1ee2 43 #define ntohl( x ) (htonl(x))
rikvandyck 0:e1edd52b1ee2 44
rikvandyck 0:e1edd52b1ee2 45 NTPClient::NTPClient() : NetService(false), m_state(NTP_PING), m_pCbItem(NULL), m_pCbMeth(NULL), m_pCb(NULL),
rikvandyck 0:e1edd52b1ee2 46 m_watchdog(), m_timeout(0), m_closed(true), m_host(), m_pDnsReq(NULL), m_blockingResult(NTP_PROCESSING)
rikvandyck 0:e1edd52b1ee2 47 {
rikvandyck 0:e1edd52b1ee2 48 setTimeout(NTP_REQUEST_TIMEOUT);
rikvandyck 0:e1edd52b1ee2 49 DBG("\r\nNew NTPClient %p\r\n",this);
rikvandyck 0:e1edd52b1ee2 50 }
rikvandyck 0:e1edd52b1ee2 51
rikvandyck 0:e1edd52b1ee2 52 NTPClient::~NTPClient()
rikvandyck 0:e1edd52b1ee2 53 {
rikvandyck 0:e1edd52b1ee2 54 close();
rikvandyck 0:e1edd52b1ee2 55 }
rikvandyck 0:e1edd52b1ee2 56
rikvandyck 0:e1edd52b1ee2 57 //High level setup functions
rikvandyck 0:e1edd52b1ee2 58 NTPResult NTPClient::setTime(const Host& host) //Blocking
rikvandyck 0:e1edd52b1ee2 59 {
rikvandyck 0:e1edd52b1ee2 60 doSetTime(host);
rikvandyck 0:e1edd52b1ee2 61 return blockingProcess();
rikvandyck 0:e1edd52b1ee2 62 }
rikvandyck 0:e1edd52b1ee2 63
rikvandyck 0:e1edd52b1ee2 64 NTPResult NTPClient::setTime(const Host& host, void (*pMethod)(NTPResult)) //Non blocking
rikvandyck 0:e1edd52b1ee2 65 {
rikvandyck 0:e1edd52b1ee2 66 setOnResult(pMethod);
rikvandyck 0:e1edd52b1ee2 67 doSetTime(host);
rikvandyck 0:e1edd52b1ee2 68 return NTP_PROCESSING;
rikvandyck 0:e1edd52b1ee2 69 }
rikvandyck 0:e1edd52b1ee2 70
rikvandyck 0:e1edd52b1ee2 71 #if 0 //For doc only
rikvandyck 0:e1edd52b1ee2 72 template<class T>
rikvandyck 0:e1edd52b1ee2 73 NTPResult NTPClient::setTime(const Host& host, T* pItem, void (T::*pMethod)(NTPResult)) //Non blocking
rikvandyck 0:e1edd52b1ee2 74 {
rikvandyck 0:e1edd52b1ee2 75 setOnResult(pItem, pMethod);
rikvandyck 0:e1edd52b1ee2 76 doSetTime(host);
rikvandyck 0:e1edd52b1ee2 77 return NTP_PROCESSING;
rikvandyck 0:e1edd52b1ee2 78 }
rikvandyck 0:e1edd52b1ee2 79 #endif
rikvandyck 0:e1edd52b1ee2 80
rikvandyck 0:e1edd52b1ee2 81 void NTPClient::doSetTime(const Host& host)
rikvandyck 0:e1edd52b1ee2 82 {
rikvandyck 0:e1edd52b1ee2 83 init();
rikvandyck 0:e1edd52b1ee2 84 resetTimeout();
rikvandyck 0:e1edd52b1ee2 85 m_host = host;
rikvandyck 0:e1edd52b1ee2 86 if(!m_host.getPort())
rikvandyck 0:e1edd52b1ee2 87 {
rikvandyck 0:e1edd52b1ee2 88 m_host.setPort(NTP_PORT);
rikvandyck 0:e1edd52b1ee2 89 }
rikvandyck 0:e1edd52b1ee2 90 if(m_host.getIp().isNull())
rikvandyck 0:e1edd52b1ee2 91 {
rikvandyck 0:e1edd52b1ee2 92 //DNS query required
rikvandyck 0:e1edd52b1ee2 93 m_pDnsReq = new DNSRequest();
rikvandyck 0:e1edd52b1ee2 94 DBG("\r\nNTPClient : DNSRequest %p\r\n", m_pDnsReq);
rikvandyck 0:e1edd52b1ee2 95 m_pDnsReq->setOnReply(this, &NTPClient::onDNSReply);
rikvandyck 0:e1edd52b1ee2 96 m_pDnsReq->resolve(&m_host);
rikvandyck 0:e1edd52b1ee2 97 return;
rikvandyck 0:e1edd52b1ee2 98 }
rikvandyck 0:e1edd52b1ee2 99 open();
rikvandyck 0:e1edd52b1ee2 100 }
rikvandyck 0:e1edd52b1ee2 101
rikvandyck 0:e1edd52b1ee2 102 void NTPClient::setOnResult( void (*pMethod)(NTPResult) )
rikvandyck 0:e1edd52b1ee2 103 {
rikvandyck 0:e1edd52b1ee2 104 m_pCb = pMethod;
rikvandyck 0:e1edd52b1ee2 105 }
rikvandyck 0:e1edd52b1ee2 106
rikvandyck 0:e1edd52b1ee2 107 void NTPClient::close()
rikvandyck 0:e1edd52b1ee2 108 {
rikvandyck 0:e1edd52b1ee2 109 if(m_closed)
rikvandyck 0:e1edd52b1ee2 110 return;
rikvandyck 0:e1edd52b1ee2 111 m_closed = true; //Prevent recursive calling or calling on an object being destructed by someone else
rikvandyck 0:e1edd52b1ee2 112 m_watchdog.stop();
rikvandyck 0:e1edd52b1ee2 113 m_pUDPSocket->resetOnEvent();
rikvandyck 0:e1edd52b1ee2 114 m_pUDPSocket->close();
rikvandyck 0:e1edd52b1ee2 115 delete m_pUDPSocket;
rikvandyck 0:e1edd52b1ee2 116 if( m_pDnsReq )
rikvandyck 0:e1edd52b1ee2 117 {
rikvandyck 0:e1edd52b1ee2 118 m_pDnsReq->close();
rikvandyck 0:e1edd52b1ee2 119 delete m_pDnsReq;
rikvandyck 0:e1edd52b1ee2 120 m_pDnsReq = NULL;
rikvandyck 0:e1edd52b1ee2 121 }
rikvandyck 0:e1edd52b1ee2 122 }
rikvandyck 0:e1edd52b1ee2 123
rikvandyck 0:e1edd52b1ee2 124 void NTPClient::poll() //Called by NetServices
rikvandyck 0:e1edd52b1ee2 125 {
rikvandyck 0:e1edd52b1ee2 126 if( (!m_closed) && (m_watchdog.read_ms() >= m_timeout) )
rikvandyck 0:e1edd52b1ee2 127 {
rikvandyck 0:e1edd52b1ee2 128 onTimeout();
rikvandyck 0:e1edd52b1ee2 129 }
rikvandyck 0:e1edd52b1ee2 130 }
rikvandyck 0:e1edd52b1ee2 131
rikvandyck 0:e1edd52b1ee2 132 void NTPClient::init() //Create and setup socket if needed
rikvandyck 0:e1edd52b1ee2 133 {
rikvandyck 0:e1edd52b1ee2 134 if(!m_closed) //Already opened
rikvandyck 0:e1edd52b1ee2 135 return;
rikvandyck 0:e1edd52b1ee2 136 m_state = NTP_PING;
rikvandyck 0:e1edd52b1ee2 137 m_pUDPSocket = new UDPSocket;
rikvandyck 0:e1edd52b1ee2 138 m_pUDPSocket->setOnEvent(this, &NTPClient::onUDPSocketEvent);
rikvandyck 0:e1edd52b1ee2 139 m_closed = false;
rikvandyck 0:e1edd52b1ee2 140 DBG("NTPClient: Init OK\n");
rikvandyck 0:e1edd52b1ee2 141 }
rikvandyck 0:e1edd52b1ee2 142
rikvandyck 0:e1edd52b1ee2 143 void NTPClient::open()
rikvandyck 0:e1edd52b1ee2 144 {
rikvandyck 0:e1edd52b1ee2 145 resetTimeout();
rikvandyck 0:e1edd52b1ee2 146 DBG("Opening connection\n");
rikvandyck 0:e1edd52b1ee2 147 m_state = NTP_PING;
rikvandyck 0:e1edd52b1ee2 148 Host localhost(IpAddr(), NTP_CLIENT_PORT, "localhost"); //Any local address
rikvandyck 0:e1edd52b1ee2 149 m_pUDPSocket->bind(localhost);
rikvandyck 0:e1edd52b1ee2 150 if ((int)time(NULL) < 1280000000) set_time( 1280000000 ); //End of July 2010... just there to limit offset range
rikvandyck 0:e1edd52b1ee2 151 process();
rikvandyck 0:e1edd52b1ee2 152 }
rikvandyck 0:e1edd52b1ee2 153
rikvandyck 0:e1edd52b1ee2 154 #define MIN(a,b) ((a)<(b))?(a):(b)
rikvandyck 0:e1edd52b1ee2 155 void NTPClient::process() //Main state-machine
rikvandyck 0:e1edd52b1ee2 156 {
rikvandyck 0:e1edd52b1ee2 157 int len;
rikvandyck 0:e1edd52b1ee2 158 Host host;
rikvandyck 0:e1edd52b1ee2 159
rikvandyck 0:e1edd52b1ee2 160 switch(m_state)
rikvandyck 0:e1edd52b1ee2 161 {
rikvandyck 0:e1edd52b1ee2 162 case NTP_PING:
rikvandyck 0:e1edd52b1ee2 163 DBG("\r\nPing\r\n");
rikvandyck 0:e1edd52b1ee2 164 //Prepare NTP Packet:
rikvandyck 0:e1edd52b1ee2 165 m_pkt.li = 0; //Leap Indicator : No warning
rikvandyck 0:e1edd52b1ee2 166 m_pkt.vn = 4; //Version Number : 4
rikvandyck 0:e1edd52b1ee2 167 m_pkt.mode = 3; //Client mode
rikvandyck 0:e1edd52b1ee2 168 m_pkt.stratum = 0; //Not relevant here
rikvandyck 0:e1edd52b1ee2 169 m_pkt.poll = 0; //Not significant as well
rikvandyck 0:e1edd52b1ee2 170 m_pkt.precision = 0; //Neither this one is
rikvandyck 0:e1edd52b1ee2 171
rikvandyck 0:e1edd52b1ee2 172 m_pkt.rootDelay = 0; //Or this one
rikvandyck 0:e1edd52b1ee2 173 m_pkt.rootDispersion = 0; //Or that one
rikvandyck 0:e1edd52b1ee2 174 m_pkt.refId = 0; //...
rikvandyck 0:e1edd52b1ee2 175
rikvandyck 0:e1edd52b1ee2 176 m_pkt.refTm_s = 0;
rikvandyck 0:e1edd52b1ee2 177 m_pkt.origTm_s = 0;
rikvandyck 0:e1edd52b1ee2 178 m_pkt.rxTm_s = 0;
rikvandyck 0:e1edd52b1ee2 179 m_pkt.txTm_s = htonl( NTP_TIMESTAMP_DELTA + time(NULL) ); //WARN: We are in LE format, network byte order is BE
rikvandyck 0:e1edd52b1ee2 180
rikvandyck 0:e1edd52b1ee2 181 m_pkt.refTm_f = m_pkt.origTm_f = m_pkt.rxTm_f = m_pkt.txTm_f = 0;
rikvandyck 0:e1edd52b1ee2 182
rikvandyck 0:e1edd52b1ee2 183 #ifdef __DEBUG
rikvandyck 0:e1edd52b1ee2 184 //Hex Dump:
rikvandyck 0:e1edd52b1ee2 185 DBG("\r\nDump Tx:\r\n");
rikvandyck 0:e1edd52b1ee2 186 for(int i = 0; i< sizeof(NTPPacket); i++)
rikvandyck 0:e1edd52b1ee2 187 {
rikvandyck 0:e1edd52b1ee2 188 DBG("%02x ", *((char*)&m_pkt + i));
rikvandyck 0:e1edd52b1ee2 189 }
rikvandyck 0:e1edd52b1ee2 190 DBG("\r\n\r\n");
rikvandyck 0:e1edd52b1ee2 191 #endif
rikvandyck 0:e1edd52b1ee2 192
rikvandyck 0:e1edd52b1ee2 193 len = m_pUDPSocket->sendto( (char*)&m_pkt, sizeof(NTPPacket), &m_host );
rikvandyck 0:e1edd52b1ee2 194 if(len < sizeof(NTPPacket))
rikvandyck 0:e1edd52b1ee2 195 { onResult(NTP_PRTCL); close(); return; }
rikvandyck 0:e1edd52b1ee2 196
rikvandyck 0:e1edd52b1ee2 197 m_state = NTP_PONG;
rikvandyck 0:e1edd52b1ee2 198
rikvandyck 0:e1edd52b1ee2 199 break;
rikvandyck 0:e1edd52b1ee2 200
rikvandyck 0:e1edd52b1ee2 201 case NTP_PONG:
rikvandyck 0:e1edd52b1ee2 202 DBG("\r\nPong\r\n");
rikvandyck 0:e1edd52b1ee2 203 while( len = m_pUDPSocket->recvfrom( (char*)&m_pkt, sizeof(NTPPacket), &host ) )
rikvandyck 0:e1edd52b1ee2 204 {
rikvandyck 0:e1edd52b1ee2 205 if( len <= 0 )
rikvandyck 0:e1edd52b1ee2 206 break;
rikvandyck 0:e1edd52b1ee2 207 if( !host.getIp().isEq(m_host.getIp()) )
rikvandyck 0:e1edd52b1ee2 208 continue; //Not our packet
rikvandyck 0:e1edd52b1ee2 209 if( len > 0 )
rikvandyck 0:e1edd52b1ee2 210 break;
rikvandyck 0:e1edd52b1ee2 211 }
rikvandyck 0:e1edd52b1ee2 212
rikvandyck 0:e1edd52b1ee2 213 if(len == 0)
rikvandyck 0:e1edd52b1ee2 214 return; //Wait for the next packet
rikvandyck 0:e1edd52b1ee2 215
rikvandyck 0:e1edd52b1ee2 216 if(len < 0)
rikvandyck 0:e1edd52b1ee2 217 { onResult(NTP_PRTCL); close(); return; }
rikvandyck 0:e1edd52b1ee2 218
rikvandyck 0:e1edd52b1ee2 219 if(len < sizeof(NTPPacket)) //TODO: Accept chunks
rikvandyck 0:e1edd52b1ee2 220 { onResult(NTP_PRTCL); close(); return; }
rikvandyck 0:e1edd52b1ee2 221
rikvandyck 0:e1edd52b1ee2 222 #ifdef __DEBUG
rikvandyck 0:e1edd52b1ee2 223 //Hex Dump:
rikvandyck 0:e1edd52b1ee2 224 DBG("\r\nDump Rx:\r\n");
rikvandyck 0:e1edd52b1ee2 225 for(int i = 0; i< sizeof(NTPPacket); i++)
rikvandyck 0:e1edd52b1ee2 226 {
rikvandyck 0:e1edd52b1ee2 227 DBG("%02x ", *((char*)&m_pkt + i));
rikvandyck 0:e1edd52b1ee2 228 }
rikvandyck 0:e1edd52b1ee2 229 DBG("\r\n\r\n");
rikvandyck 0:e1edd52b1ee2 230 #endif
rikvandyck 0:e1edd52b1ee2 231
rikvandyck 0:e1edd52b1ee2 232 if( m_pkt.stratum == 0) //Kiss of death message : Not good !
rikvandyck 0:e1edd52b1ee2 233 {
rikvandyck 0:e1edd52b1ee2 234 onResult(NTP_PRTCL); close(); return;
rikvandyck 0:e1edd52b1ee2 235 }
rikvandyck 0:e1edd52b1ee2 236
rikvandyck 0:e1edd52b1ee2 237 //Correct Endianness
rikvandyck 0:e1edd52b1ee2 238 m_pkt.refTm_s = ntohl( m_pkt.refTm_s );
rikvandyck 0:e1edd52b1ee2 239 m_pkt.refTm_f = ntohl( m_pkt.refTm_f );
rikvandyck 0:e1edd52b1ee2 240 m_pkt.origTm_s = ntohl( m_pkt.origTm_s );
rikvandyck 0:e1edd52b1ee2 241 m_pkt.origTm_f = ntohl( m_pkt.origTm_f );
rikvandyck 0:e1edd52b1ee2 242 m_pkt.rxTm_s = ntohl( m_pkt.rxTm_s );
rikvandyck 0:e1edd52b1ee2 243 m_pkt.rxTm_f = ntohl( m_pkt.rxTm_f );
rikvandyck 0:e1edd52b1ee2 244 m_pkt.txTm_s = ntohl( m_pkt.txTm_s );
rikvandyck 0:e1edd52b1ee2 245 m_pkt.txTm_f = ntohl( m_pkt.txTm_f );
rikvandyck 0:e1edd52b1ee2 246
rikvandyck 0:e1edd52b1ee2 247 //Compute offset, see RFC 4330 p.13
rikvandyck 0:e1edd52b1ee2 248 uint32_t destTm_s = (NTP_TIMESTAMP_DELTA + time(NULL));
rikvandyck 0:e1edd52b1ee2 249 //int32_t origTm = (int32_t) ((uint64_t) m_pkt.origTm - NTP_TIMESTAMP_DELTA); //Convert in local 32 bits timestamps
rikvandyck 0:e1edd52b1ee2 250 //int32_t rxTm = (int32_t) ((uint64_t) m_pkt.rxTm - NTP_TIMESTAMP_DELTA); //Convert in local 32 bits timestamps
rikvandyck 0:e1edd52b1ee2 251 //int32_t txTm = (int32_t) ((uint64_t) m_pkt.txTm - NTP_TIMESTAMP_DELTA); //Convert in local 32 bits timestamps
rikvandyck 0:e1edd52b1ee2 252 // int64_t offset = ( ( ( m_pkt.rxTm_s - m_pkt.origTm_s ) + ( m_pkt.txTm_s - destTm_s ) ) << 32 + ( ( m_pkt.rxTm_f - m_pkt.origTm_f ) + ( m_pkt.txTm_f - 0 ) ) ) / 2;
rikvandyck 0:e1edd52b1ee2 253 int64_t offset = ( (int64_t)( m_pkt.rxTm_s - m_pkt.origTm_s ) + (int64_t) ( m_pkt.txTm_s - destTm_s ) ) / 2; //Avoid overflow
rikvandyck 0:e1edd52b1ee2 254 DBG("\r\nSent @%d\r\n", m_pkt.txTm_s);
rikvandyck 0:e1edd52b1ee2 255 DBG("\r\nOffset: %d\r\n", offset);
rikvandyck 0:e1edd52b1ee2 256 //Set time accordingly
rikvandyck 0:e1edd52b1ee2 257 set_time( time(NULL) + (offset /*>> 32*/) );
rikvandyck 0:e1edd52b1ee2 258
rikvandyck 0:e1edd52b1ee2 259 onResult(NTP_OK);
rikvandyck 0:e1edd52b1ee2 260 close();
rikvandyck 0:e1edd52b1ee2 261
rikvandyck 0:e1edd52b1ee2 262 break;
rikvandyck 0:e1edd52b1ee2 263 }
rikvandyck 0:e1edd52b1ee2 264 }
rikvandyck 0:e1edd52b1ee2 265
rikvandyck 0:e1edd52b1ee2 266 void NTPClient::setTimeout(int ms)
rikvandyck 0:e1edd52b1ee2 267 {
rikvandyck 0:e1edd52b1ee2 268 m_timeout = ms;
rikvandyck 0:e1edd52b1ee2 269 }
rikvandyck 0:e1edd52b1ee2 270
rikvandyck 0:e1edd52b1ee2 271 void NTPClient::resetTimeout()
rikvandyck 0:e1edd52b1ee2 272 {
rikvandyck 0:e1edd52b1ee2 273 m_watchdog.reset();
rikvandyck 0:e1edd52b1ee2 274 m_watchdog.start();
rikvandyck 0:e1edd52b1ee2 275 }
rikvandyck 0:e1edd52b1ee2 276
rikvandyck 0:e1edd52b1ee2 277 void NTPClient::onTimeout() //Connection has timed out
rikvandyck 0:e1edd52b1ee2 278 {
rikvandyck 0:e1edd52b1ee2 279 close();
rikvandyck 0:e1edd52b1ee2 280 onResult(NTP_TIMEOUT);
rikvandyck 0:e1edd52b1ee2 281 }
rikvandyck 0:e1edd52b1ee2 282
rikvandyck 0:e1edd52b1ee2 283 void NTPClient::onDNSReply(DNSReply r)
rikvandyck 0:e1edd52b1ee2 284 {
rikvandyck 0:e1edd52b1ee2 285 if(m_closed)
rikvandyck 0:e1edd52b1ee2 286 {
rikvandyck 0:e1edd52b1ee2 287 DBG("\r\nWARN: Discarded\r\n");
rikvandyck 0:e1edd52b1ee2 288 return;
rikvandyck 0:e1edd52b1ee2 289 }
rikvandyck 0:e1edd52b1ee2 290
rikvandyck 0:e1edd52b1ee2 291 if( r != DNS_FOUND )
rikvandyck 0:e1edd52b1ee2 292 {
rikvandyck 0:e1edd52b1ee2 293 DBG("\r\nCould not resolve hostname.\r\n");
rikvandyck 0:e1edd52b1ee2 294 onResult(NTP_DNS);
rikvandyck 0:e1edd52b1ee2 295 close();
rikvandyck 0:e1edd52b1ee2 296 return;
rikvandyck 0:e1edd52b1ee2 297 }
rikvandyck 0:e1edd52b1ee2 298 DBG("\r\nDNS resolved.\r\n");
rikvandyck 0:e1edd52b1ee2 299 m_pDnsReq->close();
rikvandyck 0:e1edd52b1ee2 300 delete m_pDnsReq;
rikvandyck 0:e1edd52b1ee2 301 m_pDnsReq=NULL;
rikvandyck 0:e1edd52b1ee2 302 open();
rikvandyck 0:e1edd52b1ee2 303 }
rikvandyck 0:e1edd52b1ee2 304
rikvandyck 0:e1edd52b1ee2 305 void NTPClient::onUDPSocketEvent(UDPSocketEvent e)
rikvandyck 0:e1edd52b1ee2 306 {
rikvandyck 0:e1edd52b1ee2 307 resetTimeout();
rikvandyck 0:e1edd52b1ee2 308 switch(e)
rikvandyck 0:e1edd52b1ee2 309 {
rikvandyck 0:e1edd52b1ee2 310 case UDPSOCKET_READABLE: //The only event for now
rikvandyck 0:e1edd52b1ee2 311 resetTimeout();
rikvandyck 0:e1edd52b1ee2 312 process();
rikvandyck 0:e1edd52b1ee2 313 break;
rikvandyck 0:e1edd52b1ee2 314 }
rikvandyck 0:e1edd52b1ee2 315 }
rikvandyck 0:e1edd52b1ee2 316
rikvandyck 0:e1edd52b1ee2 317 void NTPClient::onResult(NTPResult r) //Must be called by impl when the request completes
rikvandyck 0:e1edd52b1ee2 318 {
rikvandyck 0:e1edd52b1ee2 319 if(m_pCbItem && m_pCbMeth)
rikvandyck 0:e1edd52b1ee2 320 (m_pCbItem->*m_pCbMeth)(r);
rikvandyck 0:e1edd52b1ee2 321 else if(m_pCb)
rikvandyck 0:e1edd52b1ee2 322 m_pCb(r);
rikvandyck 0:e1edd52b1ee2 323 m_blockingResult = r; //Blocking mode
rikvandyck 0:e1edd52b1ee2 324 }
rikvandyck 0:e1edd52b1ee2 325
rikvandyck 0:e1edd52b1ee2 326 NTPResult NTPClient::blockingProcess() //Called in blocking mode, calls Net::poll() until return code is available
rikvandyck 0:e1edd52b1ee2 327 {
rikvandyck 0:e1edd52b1ee2 328 m_blockingResult = NTP_PROCESSING;
rikvandyck 0:e1edd52b1ee2 329 do
rikvandyck 0:e1edd52b1ee2 330 {
rikvandyck 0:e1edd52b1ee2 331 Net::poll();
rikvandyck 0:e1edd52b1ee2 332 } while(m_blockingResult == NTP_PROCESSING);
rikvandyck 0:e1edd52b1ee2 333 Net::poll(); //Necessary for cleanup
rikvandyck 0:e1edd52b1ee2 334 return m_blockingResult;
rikvandyck 0:e1edd52b1ee2 335 }