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