Steve Kim / ESP8266Interface

Added monitoring feature of ESP8266's UART

Dependents:   ESP8266_W7500_Example DualNetworkInterface-Basic

Fork of ESP8266Interface by ESP8266

Wifly/Wifly.cpp@12:c5f0eac67a8a, 2014-10-08 (annotated)

Committer:
michaeljkoster
Date:
Wed Oct 08 19:57:58 2014 +0000
Revision:
12:c5f0eac67a8a
Parent:
11:fc3d86645d23 
Debug

Who changed what in which revision?

UserRevisionLine numberNew contents of line
samux 1:fb4494783863 1 /* Copyright (C) 2012 mbed.org, MIT License
samux 1:fb4494783863 2 *
samux 1:fb4494783863 3 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software
samux 1:fb4494783863 4 * and associated documentation files (the "Software"), to deal in the Software without restriction,
samux 1:fb4494783863 5 * including without limitation the rights to use, copy, modify, merge, publish, distribute,
samux 1:fb4494783863 6 * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
samux 1:fb4494783863 7 * furnished to do so, subject to the following conditions:
samux 1:fb4494783863 8 *
samux 1:fb4494783863 9 * The above copyright notice and this permission notice shall be included in all copies or
samux 1:fb4494783863 10 * substantial portions of the Software.
samux 1:fb4494783863 11 *
samux 1:fb4494783863 12 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
samux 1:fb4494783863 13 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
samux 1:fb4494783863 14 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
samux 1:fb4494783863 15 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
samux 1:fb4494783863 16 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
samux 1:fb4494783863 17 */
samux 1:fb4494783863 18
samux 1:fb4494783863 19 #include "mbed.h"
samux 1:fb4494783863 20 #include "Wifly.h"
samux 1:fb4494783863 21 #include <string>
samux 1:fb4494783863 22 #include <algorithm>
samux 1:fb4494783863 23
samux 1:fb4494783863 24 //Debug is disabled by default
screamer 10:131675c17372 25 #if (defined(DEBUG))
samux 1:fb4494783863 26 #define DBG(x, ...) std::printf("[Wifly : DBG]"x"\r\n", ##__VA_ARGS__);
samux 1:fb4494783863 27 #define WARN(x, ...) std::printf("[Wifly : WARN]"x"\r\n", ##__VA_ARGS__);
samux 1:fb4494783863 28 #define ERR(x, ...) std::printf("[Wifly : ERR]"x"\r\n", ##__VA_ARGS__);
samux 1:fb4494783863 29 #else
samux 1:fb4494783863 30 #define DBG(x, ...)
samux 1:fb4494783863 31 #define WARN(x, ...)
samux 1:fb4494783863 32 #define ERR(x, ...)
samux 1:fb4494783863 33 #endif
samux 1:fb4494783863 34
screamer 10:131675c17372 35 #if defined(DEBUG)
samux 1:fb4494783863 36 #define INFO(x, ...) printf("[Wifly : INFO]"x"\r\n", ##__VA_ARGS__);
samux 1:fb4494783863 37 #else
samux 1:fb4494783863 38 #define INFO(x, ...)
samux 1:fb4494783863 39 #endif
samux 1:fb4494783863 40
samux 1:fb4494783863 41 #define MAX_TRY_JOIN 3
samux 1:fb4494783863 42
michaeljkoster 12:c5f0eac67a8a 43 extern Serial pc;
michaeljkoster 12:c5f0eac67a8a 44
samux 1:fb4494783863 45 Wifly * Wifly::inst;
samux 1:fb4494783863 46
samux 1:fb4494783863 47 Wifly::Wifly( PinName tx, PinName rx, PinName _reset, PinName tcp_status, const char * ssid, const char * phrase, Security sec):
samux 1:fb4494783863 48 wifi(tx, rx), reset_pin(_reset), tcp_status(tcp_status), buf_wifly(256)
samux 1:fb4494783863 49 {
samux 1:fb4494783863 50 memset(&state, 0, sizeof(state));
samux 1:fb4494783863 51 state.sec = sec;
samux 1:fb4494783863 52
samux 1:fb4494783863 53 // change all ' ' in '$' in the ssid and the passphrase
samux 1:fb4494783863 54 strcpy(this->ssid, ssid);
samux 1:fb4494783863 55 for (int i = 0; i < strlen(ssid); i++) {
samux 1:fb4494783863 56 if (this->ssid[i] == ' ')
samux 1:fb4494783863 57 this->ssid[i] = '$';
samux 1:fb4494783863 58 }
samux 1:fb4494783863 59 strcpy(this->phrase, phrase);
samux 1:fb4494783863 60 for (int i = 0; i < strlen(phrase); i++) {
samux 1:fb4494783863 61 if (this->phrase[i] == ' ')
samux 1:fb4494783863 62 this->phrase[i] = '$';
samux 1:fb4494783863 63 }
samux 1:fb4494783863 64
samux 1:fb4494783863 65 inst = this;
samux 1:fb4494783863 66 attach_rx(false);
samux 1:fb4494783863 67 state.cmd_mode = false;
samux 1:fb4494783863 68 }
samux 1:fb4494783863 69
samux 1:fb4494783863 70 bool Wifly::join()
samux 1:fb4494783863 71 {
samux 1:fb4494783863 72 char cmd[20];
samux 1:fb4494783863 73
samux 1:fb4494783863 74 for (int i= 0; i < MAX_TRY_JOIN; i++) {
samux 2:8e54830d0df7 75
samux 2:8e54830d0df7 76 // no auto join
michaeljkoster 12:c5f0eac67a8a 77 if (!sendCommand("set w j 0\r\n", "AOK"))
samux 2:8e54830d0df7 78 continue;
samux 2:8e54830d0df7 79
samux 2:8e54830d0df7 80 //no echo
michaeljkoster 12:c5f0eac67a8a 81 if (!sendCommand("set u m 1\r\n", "AOK"))
samux 2:8e54830d0df7 82 continue;
samux 2:8e54830d0df7 83
samux 1:fb4494783863 84 // set time
michaeljkoster 12:c5f0eac67a8a 85 if (!sendCommand("set c t 30\r\n", "AOK"))
samux 1:fb4494783863 86 continue;
samux 1:fb4494783863 87
samux 1:fb4494783863 88 // set size
michaeljkoster 12:c5f0eac67a8a 89 if (!sendCommand("set c s 1024\r\n", "AOK"))
samux 1:fb4494783863 90 continue;
samux 1:fb4494783863 91
samux 1:fb4494783863 92 // red led on when tcp connection active
michaeljkoster 12:c5f0eac67a8a 93 if (!sendCommand("set s i 0x40\r\n", "AOK"))
samux 1:fb4494783863 94 continue;
samux 1:fb4494783863 95
samux 1:fb4494783863 96 // no string sent to the tcp client
michaeljkoster 12:c5f0eac67a8a 97 if (!sendCommand("set c r 0\r\n", "AOK"))
samux 1:fb4494783863 98 continue;
samux 1:fb4494783863 99
samux 1:fb4494783863 100 // tcp protocol
michaeljkoster 12:c5f0eac67a8a 101 if (!sendCommand("set i p 2\r\n", "AOK"))
samux 1:fb4494783863 102 continue;
samux 1:fb4494783863 103
samux 1:fb4494783863 104 // tcp retry
michaeljkoster 12:c5f0eac67a8a 105 if (!sendCommand("set i f 0x7\r\n", "AOK"))
samux 1:fb4494783863 106 continue;
samux 2:8e54830d0df7 107
samux 2:8e54830d0df7 108 // set dns server
michaeljkoster 12:c5f0eac67a8a 109 if (!sendCommand("set d n rn.microchip.com\r\n", "AOK"))
samux 1:fb4494783863 110 continue;
samux 1:fb4494783863 111
samux 1:fb4494783863 112 //dhcp
michaeljkoster 12:c5f0eac67a8a 113 sprintf(cmd, "set i d %d\r\n", (state.dhcp) ? 1 : 0);
samux 1:fb4494783863 114 if (!sendCommand(cmd, "AOK"))
samux 1:fb4494783863 115 continue;
samux 1:fb4494783863 116
samux 1:fb4494783863 117 // ssid
michaeljkoster 12:c5f0eac67a8a 118 sprintf(cmd, "set w s %s\r\n", ssid);
lz307 7:3152fcc74390 119 if (!sendCommand(cmd, "AOK", NULL, 1000))
samux 1:fb4494783863 120 continue;
samux 1:fb4494783863 121
samux 1:fb4494783863 122 //auth
michaeljkoster 12:c5f0eac67a8a 123 sprintf(cmd, "set w a %d\r\n", state.sec);
samux 1:fb4494783863 124 if (!sendCommand(cmd, "AOK"))
samux 1:fb4494783863 125 continue;
samux 1:fb4494783863 126
samux 1:fb4494783863 127 // if no dhcp, set ip, netmask and gateway
samux 1:fb4494783863 128 if (!state.dhcp) {
michaeljkoster 12:c5f0eac67a8a 129 DBG("not dhcp\r\n");
samux 1:fb4494783863 130
samux 1:fb4494783863 131 sprintf(cmd, "set i a %s\r\n", ip);
samux 1:fb4494783863 132 if (!sendCommand(cmd, "AOK"))
samux 1:fb4494783863 133 continue;
samux 1:fb4494783863 134
michaeljkoster 12:c5f0eac67a8a 135 sprintf(cmd, "set i n %s\r\n", netmask);
samux 1:fb4494783863 136 if (!sendCommand(cmd, "AOK"))
samux 1:fb4494783863 137 continue;
samux 1:fb4494783863 138
samux 1:fb4494783863 139 sprintf(cmd, "set i g %s\r", gateway);
samux 1:fb4494783863 140 if (!sendCommand(cmd, "AOK"))
samux 1:fb4494783863 141 continue;
samux 1:fb4494783863 142 }
samux 1:fb4494783863 143
samux 1:fb4494783863 144 //key step
samux 1:fb4494783863 145 if (state.sec != NONE) {
samux 1:fb4494783863 146 if (state.sec == WPA)
michaeljkoster 12:c5f0eac67a8a 147 sprintf(cmd, "set w p %s\r\n", phrase);
samux 1:fb4494783863 148 else if (state.sec == WEP_128)
michaeljkoster 12:c5f0eac67a8a 149 sprintf(cmd, "set w k %s\r\n", phrase);
samux 1:fb4494783863 150
lz307 7:3152fcc74390 151 if (!sendCommand(cmd, "AOK", NULL, 1000))
samux 1:fb4494783863 152 continue;
samux 1:fb4494783863 153 }
samux 1:fb4494783863 154
samux 1:fb4494783863 155 //join the network
michaeljkoster 12:c5f0eac67a8a 156 sprintf(cmd, "join\r\n");
samux 1:fb4494783863 157 if (!sendCommand(cmd, "Associated", NULL, 3000))
samux 1:fb4494783863 158 continue;
lz307 7:3152fcc74390 159
lz307 7:3152fcc74390 160 if (!sendCommand("", "IP=", NULL, 10000))
lz307 7:3152fcc74390 161 continue;
samux 1:fb4494783863 162
samux 1:fb4494783863 163 if (state.dhcp) {
michaeljkoster 12:c5f0eac67a8a 164 if (!sendCommand("get i\r\n", "DHCP=ON", NULL, 3000))
samux 1:fb4494783863 165 continue;
samux 1:fb4494783863 166 }
samux 1:fb4494783863 167
michaeljkoster 12:c5f0eac67a8a 168 if (!sendCommand("save\r\n", "Stor"))
samux 2:8e54830d0df7 169 continue;
samux 2:8e54830d0df7 170
samux 1:fb4494783863 171 exit();
samux 1:fb4494783863 172
samux 1:fb4494783863 173 state.associated = true;
samux 1:fb4494783863 174 INFO("\r\nssid: %s\r\nphrase: %s\r\nsecurity: %s\r\n\r\n", this->ssid, this->phrase, getStringSecurity());
samux 1:fb4494783863 175 return true;
samux 1:fb4494783863 176 }
samux 1:fb4494783863 177 return false;
samux 1:fb4494783863 178 }
samux 1:fb4494783863 179
samux 1:fb4494783863 180
samux 1:fb4494783863 181 bool Wifly::setProtocol(Protocol p)
samux 1:fb4494783863 182 {
samux 1:fb4494783863 183 // use udp auto pairing
samux 1:fb4494783863 184 char cmd[20];
michaeljkoster 12:c5f0eac67a8a 185 sprintf(cmd, "set i p %d\r\n", p);
samux 1:fb4494783863 186 if (!sendCommand(cmd, "AOK"))
samux 1:fb4494783863 187 return false;
samux 1:fb4494783863 188
samux 1:fb4494783863 189 switch(p) {
samux 1:fb4494783863 190 case TCP:
samux 1:fb4494783863 191 // set ip flags: tcp retry enabled
michaeljkoster 12:c5f0eac67a8a 192 if (!sendCommand("set i f 0x07\r\n", "AOK"))
samux 1:fb4494783863 193 return false;
samux 1:fb4494783863 194 break;
samux 1:fb4494783863 195 case UDP:
samux 1:fb4494783863 196 // set ip flags: udp auto pairing enabled
michaeljkoster 12:c5f0eac67a8a 197 if (!sendCommand("set i h 0.0.0.0\r\n", "AOK"))
samux 1:fb4494783863 198 return false;
michaeljkoster 12:c5f0eac67a8a 199 if (!sendCommand("set i f 0x40\r\n", "AOK"))
michaeljkoster 12:c5f0eac67a8a 200 // if (!sendCommand("set i f 0x00\r\n", "AOK")) // TEST turn off autopairing
samux 1:fb4494783863 201 return false;
samux 1:fb4494783863 202 break;
samux 1:fb4494783863 203 }
samux 1:fb4494783863 204 state.proto = p;
samux 1:fb4494783863 205 return true;
samux 1:fb4494783863 206 }
samux 1:fb4494783863 207
samux 1:fb4494783863 208 char * Wifly::getStringSecurity()
samux 1:fb4494783863 209 {
samux 1:fb4494783863 210 switch(state.sec) {
samux 1:fb4494783863 211 case NONE:
samux 1:fb4494783863 212 return "NONE";
samux 1:fb4494783863 213 case WEP_128:
samux 1:fb4494783863 214 return "WEP_128";
samux 1:fb4494783863 215 case WPA:
samux 1:fb4494783863 216 return "WPA";
samux 1:fb4494783863 217 }
samux 1:fb4494783863 218 return "UNKNOWN";
samux 1:fb4494783863 219 }
samux 1:fb4494783863 220
samux 1:fb4494783863 221 bool Wifly::connect(const char * host, int port)
samux 1:fb4494783863 222 {
samux 1:fb4494783863 223 char rcv[20];
samux 1:fb4494783863 224 char cmd[20];
samux 1:fb4494783863 225
samux 2:8e54830d0df7 226 // try to open
michaeljkoster 12:c5f0eac67a8a 227 sprintf(cmd, "open %s %d\r\n", host, port);
samux 2:8e54830d0df7 228 if (sendCommand(cmd, "OPEN", NULL, 10000)) {
samux 2:8e54830d0df7 229 state.tcp = true;
samux 2:8e54830d0df7 230 state.cmd_mode = false;
samux 2:8e54830d0df7 231 return true;
samux 1:fb4494783863 232 }
samux 1:fb4494783863 233
samux 2:8e54830d0df7 234 // if failed, retry and parse the response
samux 2:8e54830d0df7 235 if (sendCommand(cmd, NULL, rcv, 5000)) {
samux 1:fb4494783863 236 if (strstr(rcv, "OPEN") == NULL) {
samux 1:fb4494783863 237 if (strstr(rcv, "Connected") != NULL) {
samux 2:8e54830d0df7 238 wait(0.25);
michaeljkoster 12:c5f0eac67a8a 239 if (!sendCommand("close\r\n", "CLOS"))
samux 1:fb4494783863 240 return false;
samux 2:8e54830d0df7 241 wait(0.25);
samux 2:8e54830d0df7 242 if (!sendCommand(cmd, "OPEN", NULL, 10000))
samux 1:fb4494783863 243 return false;
samux 1:fb4494783863 244 } else {
samux 1:fb4494783863 245 return false;
samux 1:fb4494783863 246 }
samux 1:fb4494783863 247 }
samux 1:fb4494783863 248 } else {
samux 1:fb4494783863 249 return false;
samux 1:fb4494783863 250 }
samux 2:8e54830d0df7 251
samux 1:fb4494783863 252 state.tcp = true;
samux 1:fb4494783863 253 state.cmd_mode = false;
samux 1:fb4494783863 254
samux 1:fb4494783863 255 return true;
samux 1:fb4494783863 256 }
samux 1:fb4494783863 257
samux 1:fb4494783863 258
samux 1:fb4494783863 259 bool Wifly::gethostbyname(const char * host, char * ip)
samux 1:fb4494783863 260 {
samux 1:fb4494783863 261 string h = host;
samux 1:fb4494783863 262 char cmd[30], rcv[100];
samux 1:fb4494783863 263 int l = 0;
samux 1:fb4494783863 264 char * point;
samux 1:fb4494783863 265 int nb_digits = 0;
samux 1:fb4494783863 266
samux 1:fb4494783863 267 // no dns needed
samux 1:fb4494783863 268 int pos = h.find(".");
samux 1:fb4494783863 269 if (pos != string::npos) {
samux 1:fb4494783863 270 string sub = h.substr(0, h.find("."));
samux 1:fb4494783863 271 nb_digits = atoi(sub.c_str());
samux 1:fb4494783863 272 }
samux 1:fb4494783863 273 //printf("substrL %s\r\n", sub.c_str());
samux 1:fb4494783863 274 if (count(h.begin(), h.end(), '.') == 3 && nb_digits > 0) {
samux 1:fb4494783863 275 strcpy(ip, host);
samux 1:fb4494783863 276 }
samux 1:fb4494783863 277 // dns needed
samux 1:fb4494783863 278 else {
samux 1:fb4494783863 279 nb_digits = 0;
michaeljkoster 12:c5f0eac67a8a 280 sprintf(cmd, "lookup %s\r\n", host);
samux 1:fb4494783863 281 if (!sendCommand(cmd, NULL, rcv))
samux 1:fb4494783863 282 return false;
samux 1:fb4494783863 283
samux 1:fb4494783863 284 // look for the ip address
samux 1:fb4494783863 285 char * begin = strstr(rcv, "=") + 1;
samux 1:fb4494783863 286 for (int i = 0; i < 3; i++) {
samux 1:fb4494783863 287 point = strstr(begin + l, ".");
samux 1:fb4494783863 288 DBG("str: %s", begin + l);
samux 1:fb4494783863 289 l += point - (begin + l) + 1;
samux 1:fb4494783863 290 }
samux 1:fb4494783863 291 DBG("str: %s", begin + l);
samux 1:fb4494783863 292 while(*(begin + l + nb_digits) >= '0' && *(begin + l + nb_digits) <= '9') {
samux 1:fb4494783863 293 DBG("digit: %c", *(begin + l + nb_digits));
samux 1:fb4494783863 294 nb_digits++;
samux 1:fb4494783863 295 }
samux 1:fb4494783863 296 memcpy(ip, begin, l + nb_digits);
samux 1:fb4494783863 297 ip[l+nb_digits] = 0;
samux 1:fb4494783863 298 DBG("ip from dns: %s", ip);
samux 1:fb4494783863 299 }
samux 1:fb4494783863 300 return true;
samux 1:fb4494783863 301 }
samux 1:fb4494783863 302
samux 1:fb4494783863 303
samux 1:fb4494783863 304 void Wifly::flush()
samux 1:fb4494783863 305 {
samux 1:fb4494783863 306 buf_wifly.flush();
samux 1:fb4494783863 307 }
samux 1:fb4494783863 308
samux 1:fb4494783863 309 bool Wifly::sendCommand(const char * cmd, const char * ack, char * res, int timeout)
samux 1:fb4494783863 310 {
samux 1:fb4494783863 311 if (!state.cmd_mode) {
samux 1:fb4494783863 312 cmdMode();
samux 1:fb4494783863 313 }
samux 1:fb4494783863 314 if (send(cmd, strlen(cmd), ack, res, timeout) == -1) {
samux 1:fb4494783863 315 ERR("sendCommand: cannot %s\r\n", cmd);
samux 1:fb4494783863 316 exit();
samux 1:fb4494783863 317 return false;
samux 1:fb4494783863 318 }
samux 1:fb4494783863 319 return true;
samux 1:fb4494783863 320 }
samux 1:fb4494783863 321
samux 1:fb4494783863 322 bool Wifly::cmdMode()
samux 1:fb4494783863 323 {
samux 1:fb4494783863 324 // if already in cmd mode, return
samux 1:fb4494783863 325 if (state.cmd_mode)
samux 1:fb4494783863 326 return true;
samux 1:fb4494783863 327
lz307 7:3152fcc74390 328 if (send("$$$", 3, "CMD") == -1 && send("\r",1,">") != true) {
samux 1:fb4494783863 329 ERR("cannot enter in cmd mode\r\n");
samux 2:8e54830d0df7 330 exit();
samux 1:fb4494783863 331 return false;
samux 1:fb4494783863 332 }
samux 1:fb4494783863 333 state.cmd_mode = true;
samux 1:fb4494783863 334 return true;
samux 1:fb4494783863 335 }
samux 1:fb4494783863 336
samux 1:fb4494783863 337 bool Wifly::disconnect()
samux 1:fb4494783863 338 {
samux 1:fb4494783863 339 // if already disconnected, return
samux 1:fb4494783863 340 if (!state.associated)
samux 1:fb4494783863 341 return true;
samux 2:8e54830d0df7 342
samux 1:fb4494783863 343 if (!sendCommand("leave\r", "DeAuth"))
samux 1:fb4494783863 344 return false;
samux 1:fb4494783863 345 exit();
samux 2:8e54830d0df7 346
samux 1:fb4494783863 347 state.associated = false;
samux 1:fb4494783863 348 return true;
samux 1:fb4494783863 349
samux 1:fb4494783863 350 }
samux 1:fb4494783863 351
samux 1:fb4494783863 352 bool Wifly::is_connected()
samux 1:fb4494783863 353 {
samux 1:fb4494783863 354 return (tcp_status.read() == 1) ? true : false;
samux 1:fb4494783863 355 }
samux 1:fb4494783863 356
samux 1:fb4494783863 357
samux 1:fb4494783863 358 void Wifly::reset()
samux 1:fb4494783863 359 {
samux 1:fb4494783863 360 reset_pin = 0;
samux 1:fb4494783863 361 wait(0.2);
samux 1:fb4494783863 362 reset_pin = 1;
samux 1:fb4494783863 363 wait(0.2);
samux 1:fb4494783863 364 }
samux 1:fb4494783863 365
samux 3:9aa05e19c62e 366 bool Wifly::reboot()
samux 3:9aa05e19c62e 367 {
samux 3:9aa05e19c62e 368 // if already in cmd mode, return
michaeljkoster 12:c5f0eac67a8a 369 if (!sendCommand("reboot\r\n"))
samux 3:9aa05e19c62e 370 return false;
samux 3:9aa05e19c62e 371
samux 3:9aa05e19c62e 372 wait(0.3);
samux 3:9aa05e19c62e 373
samux 3:9aa05e19c62e 374 state.cmd_mode = false;
samux 3:9aa05e19c62e 375 return true;
samux 3:9aa05e19c62e 376 }
samux 3:9aa05e19c62e 377
samux 1:fb4494783863 378 bool Wifly::close()
samux 1:fb4494783863 379 {
samux 1:fb4494783863 380 // if not connected, return
samux 1:fb4494783863 381 if (!state.tcp)
samux 1:fb4494783863 382 return true;
samux 2:8e54830d0df7 383
samux 1:fb4494783863 384 wait(0.25);
michaeljkoster 12:c5f0eac67a8a 385 if (!sendCommand("close\r\n", "CLOS"))
samux 1:fb4494783863 386 return false;
samux 1:fb4494783863 387 exit();
samux 2:8e54830d0df7 388
samux 1:fb4494783863 389 state.tcp = false;
samux 1:fb4494783863 390 return true;
samux 1:fb4494783863 391 }
samux 1:fb4494783863 392
samux 1:fb4494783863 393
samux 1:fb4494783863 394 int Wifly::putc(char c)
samux 1:fb4494783863 395 {
samux 1:fb4494783863 396 while (!wifi.writeable());
samux 1:fb4494783863 397 return wifi.putc(c);
samux 1:fb4494783863 398 }
samux 1:fb4494783863 399
samux 1:fb4494783863 400
samux 1:fb4494783863 401 bool Wifly::exit()
samux 1:fb4494783863 402 {
samux 1:fb4494783863 403 flush();
samux 1:fb4494783863 404 if (!state.cmd_mode)
samux 1:fb4494783863 405 return true;
michaeljkoster 12:c5f0eac67a8a 406 if (!sendCommand("exit\r\n", "EXIT"))
samux 1:fb4494783863 407 return false;
samux 1:fb4494783863 408 state.cmd_mode = false;
samux 1:fb4494783863 409 flush();
samux 1:fb4494783863 410 return true;
samux 1:fb4494783863 411 }
samux 1:fb4494783863 412
samux 1:fb4494783863 413
samux 1:fb4494783863 414 int Wifly::readable()
samux 1:fb4494783863 415 {
samux 1:fb4494783863 416 return buf_wifly.available();
samux 1:fb4494783863 417 }
samux 1:fb4494783863 418
samux 1:fb4494783863 419 int Wifly::writeable()
samux 1:fb4494783863 420 {
samux 1:fb4494783863 421 return wifi.writeable();
samux 1:fb4494783863 422 }
samux 1:fb4494783863 423
samux 1:fb4494783863 424 char Wifly::getc()
samux 1:fb4494783863 425 {
samux 1:fb4494783863 426 char c;
samux 1:fb4494783863 427 while (!buf_wifly.available());
samux 1:fb4494783863 428 buf_wifly.dequeue(&c);
samux 1:fb4494783863 429 return c;
samux 1:fb4494783863 430 }
samux 1:fb4494783863 431
samux 1:fb4494783863 432 void Wifly::handler_rx(void)
samux 1:fb4494783863 433 {
samux 1:fb4494783863 434 //read characters
samux 1:fb4494783863 435 while (wifi.readable())
samux 1:fb4494783863 436 buf_wifly.queue(wifi.getc());
samux 1:fb4494783863 437 }
samux 1:fb4494783863 438
samux 1:fb4494783863 439 void Wifly::attach_rx(bool callback)
samux 1:fb4494783863 440 {
samux 1:fb4494783863 441 if (!callback)
samux 1:fb4494783863 442 wifi.attach(NULL);
samux 1:fb4494783863 443 else
samux 1:fb4494783863 444 wifi.attach(this, &Wifly::handler_rx);
samux 1:fb4494783863 445 }
samux 1:fb4494783863 446
samux 1:fb4494783863 447
samux 1:fb4494783863 448 int Wifly::send(const char * str, int len, const char * ACK, char * res, int timeout)
samux 1:fb4494783863 449 {
samux 1:fb4494783863 450 char read;
samux 1:fb4494783863 451 size_t found = string::npos;
samux 1:fb4494783863 452 string checking;
samux 1:fb4494783863 453 Timer tmr;
samux 1:fb4494783863 454 int result = 0;
samux 1:fb4494783863 455
samux 1:fb4494783863 456 DBG("will send: %s\r\n",str);
samux 1:fb4494783863 457
samux 1:fb4494783863 458 attach_rx(false);
samux 1:fb4494783863 459
samux 1:fb4494783863 460 //We flush the buffer
samux 1:fb4494783863 461 while (wifi.readable())
samux 1:fb4494783863 462 wifi.getc();
samux 1:fb4494783863 463
samux 1:fb4494783863 464 if (!ACK || !strcmp(ACK, "NO")) {
samux 1:fb4494783863 465 for (int i = 0; i < len; i++)
samux 1:fb4494783863 466 result = (putc(str[i]) == str[i]) ? result + 1 : result;
samux 1:fb4494783863 467 } else {
samux 1:fb4494783863 468 //We flush the buffer
samux 1:fb4494783863 469 while (wifi.readable())
samux 1:fb4494783863 470 wifi.getc();
samux 1:fb4494783863 471
samux 1:fb4494783863 472 tmr.start();
samux 1:fb4494783863 473 for (int i = 0; i < len; i++)
samux 1:fb4494783863 474 result = (putc(str[i]) == str[i]) ? result + 1 : result;
samux 1:fb4494783863 475
samux 1:fb4494783863 476 while (1) {
samux 1:fb4494783863 477 if (tmr.read_ms() > timeout) {
samux 1:fb4494783863 478 //We flush the buffer
samux 1:fb4494783863 479 while (wifi.readable())
samux 1:fb4494783863 480 wifi.getc();
samux 1:fb4494783863 481
samux 1:fb4494783863 482 DBG("check: %s\r\n", checking.c_str());
samux 1:fb4494783863 483
samux 1:fb4494783863 484 attach_rx(true);
samux 1:fb4494783863 485 return -1;
samux 1:fb4494783863 486 } else if (wifi.readable()) {
samux 1:fb4494783863 487 read = wifi.getc();
samux 1:fb4494783863 488 if ( read != '\r' && read != '\n') {
samux 1:fb4494783863 489 checking += read;
samux 1:fb4494783863 490 found = checking.find(ACK);
samux 1:fb4494783863 491 if (found != string::npos) {
samux 1:fb4494783863 492 wait(0.01);
samux 1:fb4494783863 493
samux 1:fb4494783863 494 //We flush the buffer
samux 1:fb4494783863 495 while (wifi.readable())
samux 1:fb4494783863 496 wifi.getc();
samux 1:fb4494783863 497
samux 1:fb4494783863 498 break;
samux 1:fb4494783863 499 }
samux 1:fb4494783863 500 }
samux 1:fb4494783863 501 }
samux 1:fb4494783863 502 }
samux 1:fb4494783863 503 DBG("check: %s\r\n", checking.c_str());
samux 1:fb4494783863 504
samux 1:fb4494783863 505 attach_rx(true);
samux 1:fb4494783863 506 return result;
samux 1:fb4494783863 507 }
samux 1:fb4494783863 508
samux 1:fb4494783863 509 //the user wants the result from the command (ACK == NULL, res != NULL)
samux 1:fb4494783863 510 if ( res != NULL) {
samux 1:fb4494783863 511 int i = 0;
samux 1:fb4494783863 512 Timer timeout;
samux 1:fb4494783863 513 timeout.start();
samux 1:fb4494783863 514 tmr.reset();
samux 1:fb4494783863 515 while (1) {
samux 1:fb4494783863 516 if (timeout.read() > 2) {
samux 1:fb4494783863 517 if (i == 0) {
samux 1:fb4494783863 518 res = NULL;
samux 1:fb4494783863 519 break;
samux 1:fb4494783863 520 }
samux 1:fb4494783863 521 res[i] = '\0';
samux 1:fb4494783863 522 DBG("user str 1: %s\r\n", res);
samux 1:fb4494783863 523
samux 1:fb4494783863 524 break;
samux 1:fb4494783863 525 } else {
samux 1:fb4494783863 526 if (tmr.read_ms() > 300) {
samux 1:fb4494783863 527 res[i] = '\0';
samux 1:fb4494783863 528 DBG("user str: %s\r\n", res);
samux 1:fb4494783863 529
samux 1:fb4494783863 530 break;
samux 1:fb4494783863 531 }
samux 1:fb4494783863 532 if (wifi.readable()) {
samux 1:fb4494783863 533 tmr.start();
samux 1:fb4494783863 534 read = wifi.getc();
samux 1:fb4494783863 535
samux 1:fb4494783863 536 // we drop \r and \n
samux 1:fb4494783863 537 if ( read != '\r' && read != '\n') {
samux 1:fb4494783863 538 res[i++] = read;
samux 1:fb4494783863 539 }
samux 1:fb4494783863 540 }
samux 1:fb4494783863 541 }
samux 1:fb4494783863 542 }
samux 1:fb4494783863 543 DBG("user str: %s\r\n", res);
samux 1:fb4494783863 544 }
samux 1:fb4494783863 545
samux 1:fb4494783863 546 //We flush the buffer
samux 1:fb4494783863 547 while (wifi.readable())
samux 1:fb4494783863 548 wifi.getc();
samux 1:fb4494783863 549
samux 1:fb4494783863 550 attach_rx(true);
samux 1:fb4494783863 551 DBG("result: %d\r\n", result)
samux 1:fb4494783863 552 return result;
samux 1:fb4494783863 553 }