Multi-Hackers / SocketModem

A library for talking to Multi-Tech's Cellular SocketModem Devices.

Dependents:   M2X_dev axeda_wrapper_dev MTS_M2x_Example1 MTS_Cellular_Connect_Example ... more

wifi/Wifi.cpp@117:e59a616ffe16, 2013-12-31 (annotated)

Committer:
jengbrecht
Date:
Tue Dec 31 22:00:53 2013 +0000
Revision:
117:e59a616ffe16
Parent:
114:cd34b1d64360
Child:
119:544578fd290c
In Wifi made a few more robustness improvements related to the isOpen method throughout.

Who changed what in which revision?

UserRevisionLine numberNew contents of line
jengbrecht 69:f3e696bbb0d5 1 #include "Wifi.h"
jengbrecht 93:aa7a48e65974 2 #include <string>
jengbrecht 69:f3e696bbb0d5 3
jengbrecht 69:f3e696bbb0d5 4 Wifi* Wifi::instance = NULL;
jengbrecht 69:f3e696bbb0d5 5
jengbrecht 69:f3e696bbb0d5 6 Wifi* Wifi::getInstance()
jengbrecht 69:f3e696bbb0d5 7 {
jengbrecht 69:f3e696bbb0d5 8 if(instance == NULL) {
jengbrecht 69:f3e696bbb0d5 9 instance = new Wifi(NULL);
jengbrecht 69:f3e696bbb0d5 10 }
jengbrecht 69:f3e696bbb0d5 11 return instance;
jengbrecht 69:f3e696bbb0d5 12 }
jengbrecht 69:f3e696bbb0d5 13
jengbrecht 69:f3e696bbb0d5 14 bool Wifi::init(MTSBufferedIO* io)
jengbrecht 69:f3e696bbb0d5 15 {
jengbrecht 69:f3e696bbb0d5 16 if (io == NULL) {
jengbrecht 69:f3e696bbb0d5 17 return false;
jengbrecht 69:f3e696bbb0d5 18 }
jengbrecht 69:f3e696bbb0d5 19 instance->io = io;
jengbrecht 94:1baa587e89ae 20 //Set device into command mode
jengbrecht 94:1baa587e89ae 21 if (!setCmdMode(true)) {
jengbrecht 94:1baa587e89ae 22 return false;
jengbrecht 94:1baa587e89ae 23 }
jengbrecht 94:1baa587e89ae 24
jengbrecht 94:1baa587e89ae 25 //Set device to non-echo mode
jengbrecht 94:1baa587e89ae 26 if (sendBasicCommand("set uart mode 1", 1000) != SUCCESS) {
jengbrecht 94:1baa587e89ae 27 printf("[ERROR] Failed to set to non-echo mode\n\r");
jengbrecht 94:1baa587e89ae 28 return false;
jengbrecht 74:9f87bd22c222 29 }
jengbrecht 94:1baa587e89ae 30
jengbrecht 94:1baa587e89ae 31 //Set device to manual infrastructure mode
jengbrecht 94:1baa587e89ae 32 if (sendBasicCommand("set wlan join 0", 1000) != SUCCESS) {
jengbrecht 94:1baa587e89ae 33 printf("[ERROR] Failed to set join mode\n\r");
jengbrecht 94:1baa587e89ae 34 return false;
jengbrecht 94:1baa587e89ae 35 }
jengbrecht 94:1baa587e89ae 36
jengbrecht 94:1baa587e89ae 37 //Set device to channel auto-scanning mode
jengbrecht 94:1baa587e89ae 38 if (sendBasicCommand("set wlan channel 0", 1000) != SUCCESS) {
jengbrecht 94:1baa587e89ae 39 printf("[ERROR] Failed to set auto-scanning mode\n\r");
jengbrecht 94:1baa587e89ae 40 return false;
jengbrecht 94:1baa587e89ae 41 }
jengbrecht 94:1baa587e89ae 42
jengbrecht 94:1baa587e89ae 43 //Set device so no data is transmitted immediately following a socket connection
jengbrecht 94:1baa587e89ae 44 if (sendBasicCommand("set comm remote 0", 1000) != SUCCESS) {
jengbrecht 94:1baa587e89ae 45 printf("[ERROR] Failed to set remote transmit mode\n\r");
jengbrecht 94:1baa587e89ae 46 return false;
jengbrecht 94:1baa587e89ae 47 }
jengbrecht 106:358972176b89 48
jengbrecht 106:358972176b89 49 //Set device into DHCP mode by default
jengbrecht 106:358972176b89 50 if (sendBasicCommand("set ip dhcp 1", 1000) != SUCCESS) {
jengbrecht 106:358972176b89 51 printf("[ERROR] Failed to set to default DHCP mode\n\r");
jengbrecht 106:358972176b89 52 return false;
jengbrecht 106:358972176b89 53 }
jengbrecht 106:358972176b89 54
jengbrecht 94:1baa587e89ae 55 return true;
jengbrecht 69:f3e696bbb0d5 56 }
jengbrecht 69:f3e696bbb0d5 57
jengbrecht 69:f3e696bbb0d5 58 Wifi::Wifi(MTSBufferedIO* io)
jengbrecht 69:f3e696bbb0d5 59 : io(io)
jengbrecht 69:f3e696bbb0d5 60 , wifiConnected(false)
jengbrecht 79:f356009dbc12 61 , _ssid("")
jengbrecht 69:f3e696bbb0d5 62 , mode(TCP)
jengbrecht 69:f3e696bbb0d5 63 , socketOpened(false)
jengbrecht 69:f3e696bbb0d5 64 , socketCloseable(true)
jengbrecht 69:f3e696bbb0d5 65 , local_port(0)
jengbrecht 106:358972176b89 66 , local_address("")
jengbrecht 69:f3e696bbb0d5 67 , host_port(0)
jengbrecht 79:f356009dbc12 68 , cmdOn(false)
jengbrecht 69:f3e696bbb0d5 69 {
jengbrecht 69:f3e696bbb0d5 70
jengbrecht 69:f3e696bbb0d5 71 }
jengbrecht 69:f3e696bbb0d5 72
jengbrecht 69:f3e696bbb0d5 73 Wifi::~Wifi()
jengbrecht 69:f3e696bbb0d5 74 {
jengbrecht 69:f3e696bbb0d5 75 }
jengbrecht 69:f3e696bbb0d5 76
jengbrecht 69:f3e696bbb0d5 77 bool Wifi::connect()
jengbrecht 69:f3e696bbb0d5 78 {
jengbrecht 69:f3e696bbb0d5 79 //Check if socket is open
jengbrecht 69:f3e696bbb0d5 80 if(socketOpened) {
jengbrecht 69:f3e696bbb0d5 81 return true;
jengbrecht 69:f3e696bbb0d5 82 }
jengbrecht 69:f3e696bbb0d5 83
jengbrecht 69:f3e696bbb0d5 84 //Run Test first to validate a good state
jengbrecht 69:f3e696bbb0d5 85 if(isConnected()) {
jengbrecht 69:f3e696bbb0d5 86 return true;
jengbrecht 69:f3e696bbb0d5 87 }
jengbrecht 69:f3e696bbb0d5 88
jengbrecht 93:aa7a48e65974 89 if (_ssid.size() == 0) {
jengbrecht 93:aa7a48e65974 90 printf("[ERROR] No SSID has been set\n\r");
jengbrecht 93:aa7a48e65974 91 return false;
jengbrecht 93:aa7a48e65974 92 }
jengbrecht 93:aa7a48e65974 93
jengbrecht 98:dbeac735109d 94 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 95 return false;
jengbrecht 93:aa7a48e65974 96 }
jengbrecht 93:aa7a48e65974 97
jengbrecht 69:f3e696bbb0d5 98 //Possibly add a scan command here and look for the network....
jengbrecht 69:f3e696bbb0d5 99
jengbrecht 69:f3e696bbb0d5 100 //join my_network
jengbrecht 69:f3e696bbb0d5 101 printf("[DEBUG] Making SSID Connection Attempt. SSID[%s]\r\n", _ssid.c_str());
jengbrecht 93:aa7a48e65974 102 std::string result = sendCommand("join " + _ssid, 15000, "Listen");
jengbrecht 93:aa7a48e65974 103 //printf("Connect Status: %s\n\r", result.c_str());
jengbrecht 69:f3e696bbb0d5 104
jengbrecht 93:aa7a48e65974 105 //Check whether connection was successful
jengbrecht 93:aa7a48e65974 106 if(result.find("Associated!") != string::npos) {
jengbrecht 106:358972176b89 107 if(result.find("Static") == string::npos) {
jengbrecht 106:358972176b89 108 int start = result.find("IP=");
jengbrecht 106:358972176b89 109 int stop = result.find(":", start);
jengbrecht 106:358972176b89 110 local_address = result.substr(start + 3, stop - start - 3);
jengbrecht 106:358972176b89 111 }
jengbrecht 93:aa7a48e65974 112 printf("[INFO] WiFi Connection Established: IP[%s]\r\n", local_address.c_str());
jengbrecht 93:aa7a48e65974 113 wifiConnected = true;
jengbrecht 106:358972176b89 114
jengbrecht 103:da58d27c15d7 115 //Report Signal Strength of new connection
jengbrecht 103:da58d27c15d7 116 wait(1); //Needed for signal strength to be available
jengbrecht 103:da58d27c15d7 117 int rssi = getSignalStrength();
jengbrecht 103:da58d27c15d7 118 printf("[DEBUG] Signal strength (dBm): %d\r\n", rssi);
jengbrecht 93:aa7a48e65974 119 } else {
jengbrecht 93:aa7a48e65974 120 wifiConnected = false;
jengbrecht 93:aa7a48e65974 121 }
jengbrecht 69:f3e696bbb0d5 122
jengbrecht 69:f3e696bbb0d5 123 return wifiConnected;
jengbrecht 69:f3e696bbb0d5 124 }
jengbrecht 69:f3e696bbb0d5 125
jengbrecht 69:f3e696bbb0d5 126 void Wifi::disconnect()
jengbrecht 69:f3e696bbb0d5 127 {
jengbrecht 79:f356009dbc12 128 printf("[DEBUG] Disconnecting from network\r\n");
jengbrecht 79:f356009dbc12 129
jengbrecht 79:f356009dbc12 130 if(socketOpened) {
jengbrecht 79:f356009dbc12 131 close();
jengbrecht 79:f356009dbc12 132 }
jengbrecht 79:f356009dbc12 133
jengbrecht 98:dbeac735109d 134 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 135 printf("[ERROR] Failed in disconnecting from network. Continuing ...\r\n");
jengbrecht 79:f356009dbc12 136 }
jengbrecht 79:f356009dbc12 137
jengbrecht 100:9d96b4391151 138 std::string response = sendCommand("leave", 10000, "<4.00>");
jengbrecht 100:9d96b4391151 139 response = sendCommand("show net", 5000, "Links");
jengbrecht 100:9d96b4391151 140 //printf("Response: %s\n\r", response.c_str());
jengbrecht 100:9d96b4391151 141 if (response.find("Assoc=FAIL") != string::npos) {
jengbrecht 79:f356009dbc12 142 printf("[DEBUG] Successfully disconnected from network\r\n");
jengbrecht 79:f356009dbc12 143 } else {
jengbrecht 79:f356009dbc12 144 printf("[ERROR] Failed in disconnecting from network. Continuing ...\r\n");
jengbrecht 79:f356009dbc12 145 }
jengbrecht 79:f356009dbc12 146
jengbrecht 79:f356009dbc12 147 wifiConnected = false;
jengbrecht 69:f3e696bbb0d5 148 }
jengbrecht 69:f3e696bbb0d5 149
jengbrecht 69:f3e696bbb0d5 150 bool Wifi::isConnected()
jengbrecht 69:f3e696bbb0d5 151 {
jengbrecht 79:f356009dbc12 152 //1) Check if SSID was set
jengbrecht 79:f356009dbc12 153 if(_ssid.size() == 0) {
jengbrecht 79:f356009dbc12 154 printf("[DEBUG] SSID is not set\r\n");
jengbrecht 79:f356009dbc12 155 return false;
jengbrecht 79:f356009dbc12 156 }
jengbrecht 79:f356009dbc12 157
jengbrecht 79:f356009dbc12 158 //1) Check that we do not have a live connection up
jengbrecht 117:e59a616ffe16 159 if(isOpen()) {
jengbrecht 79:f356009dbc12 160 printf("[DEBUG] Socket is opened\r\n");
jengbrecht 79:f356009dbc12 161 return true;
jengbrecht 79:f356009dbc12 162 }
jengbrecht 79:f356009dbc12 163
jengbrecht 98:dbeac735109d 164 //Check command mode.
jengbrecht 98:dbeac735109d 165 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 166 return false;
jengbrecht 98:dbeac735109d 167 }
jengbrecht 98:dbeac735109d 168
jengbrecht 79:f356009dbc12 169 //2) Query the wifi module
jengbrecht 79:f356009dbc12 170 wifiConnected = false;
jengbrecht 93:aa7a48e65974 171 std::string result = sendCommand("show net", 5000, "Links");
jengbrecht 93:aa7a48e65974 172 //printf("netResult: %s\n\r", result);
jengbrecht 93:aa7a48e65974 173 if(result.find("Assoc=OK") != std::string::npos) {
jengbrecht 79:f356009dbc12 174 wifiConnected = true;
jengbrecht 79:f356009dbc12 175 }
jengbrecht 79:f356009dbc12 176
jengbrecht 79:f356009dbc12 177 return wifiConnected;
jengbrecht 69:f3e696bbb0d5 178 }
jengbrecht 69:f3e696bbb0d5 179
jengbrecht 69:f3e696bbb0d5 180 bool Wifi::bind(unsigned int port)
jengbrecht 69:f3e696bbb0d5 181 {
jengbrecht 94:1baa587e89ae 182 if(socketOpened) {
jengbrecht 94:1baa587e89ae 183 printf("[ERROR] socket is open. Can not set local port\r\n");
jengbrecht 94:1baa587e89ae 184 return false;
jengbrecht 94:1baa587e89ae 185 }
jengbrecht 94:1baa587e89ae 186 if(port > 65535) {
jengbrecht 94:1baa587e89ae 187 printf("[ERROR] port out of range (0-65535)\r\n");
jengbrecht 94:1baa587e89ae 188 return false;
jengbrecht 94:1baa587e89ae 189 }
jengbrecht 94:1baa587e89ae 190 local_port = port;
jengbrecht 69:f3e696bbb0d5 191 return true;
jengbrecht 69:f3e696bbb0d5 192 }
jengbrecht 69:f3e696bbb0d5 193
jengbrecht 69:f3e696bbb0d5 194 bool Wifi::open(const std::string& address, unsigned int port, Mode mode)
jengbrecht 69:f3e696bbb0d5 195 {
jengbrecht 95:4fdf968b5b37 196 char buffer[256] = {0};
jengbrecht 95:4fdf968b5b37 197 printf("[DEBUG] Attempting to Open Socket\r\n");
jengbrecht 95:4fdf968b5b37 198
jengbrecht 95:4fdf968b5b37 199 //1) Check that we do not have a live connection up
jengbrecht 95:4fdf968b5b37 200 if(socketOpened) {
jengbrecht 95:4fdf968b5b37 201 //Check that the address, port, and mode match
jengbrecht 95:4fdf968b5b37 202 if(host_address != address || host_port != port || this->mode != mode) {
jengbrecht 95:4fdf968b5b37 203 if(this->mode == TCP) {
jengbrecht 95:4fdf968b5b37 204 printf("[ERROR] TCP socket already opened (%s:%d)\r\n", host_address.c_str(), host_port);
jengbrecht 95:4fdf968b5b37 205 } else {
jengbrecht 95:4fdf968b5b37 206 printf("[ERROR] UDP socket already opened (%s:%d)\r\n", host_address.c_str(), host_port);
jengbrecht 95:4fdf968b5b37 207 }
jengbrecht 95:4fdf968b5b37 208 return false;
jengbrecht 95:4fdf968b5b37 209 }
jengbrecht 95:4fdf968b5b37 210
jengbrecht 95:4fdf968b5b37 211 printf("[DEBUG] Socket already opened\r\n");
jengbrecht 95:4fdf968b5b37 212 return true;
jengbrecht 95:4fdf968b5b37 213 }
jengbrecht 95:4fdf968b5b37 214
jengbrecht 95:4fdf968b5b37 215 //2) Check Parameters
jengbrecht 95:4fdf968b5b37 216 if(port > 65535) {
jengbrecht 95:4fdf968b5b37 217 printf("[ERROR] port out of range (0-65535)\r\n");
jengbrecht 95:4fdf968b5b37 218 return false;
jengbrecht 95:4fdf968b5b37 219 }
jengbrecht 95:4fdf968b5b37 220
jengbrecht 95:4fdf968b5b37 221
jengbrecht 95:4fdf968b5b37 222 //3) Check Wifi network connection
jengbrecht 95:4fdf968b5b37 223 if(!isConnected()) {
jengbrecht 95:4fdf968b5b37 224 printf("[ERROR] Wifi network not connected. Attempting to connect\r\n");
jengbrecht 95:4fdf968b5b37 225 if(!connect()) {
jengbrecht 95:4fdf968b5b37 226 printf("[ERROR] Wifi network connection failed\r\n");
jengbrecht 95:4fdf968b5b37 227 return false;
jengbrecht 95:4fdf968b5b37 228 } else {
jengbrecht 95:4fdf968b5b37 229 printf("[DEBUG] Wifi connection established\r\n");
jengbrecht 95:4fdf968b5b37 230 }
jengbrecht 95:4fdf968b5b37 231 }
jengbrecht 95:4fdf968b5b37 232
jengbrecht 95:4fdf968b5b37 233 //Check command mode
jengbrecht 98:dbeac735109d 234 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 235 return false;
jengbrecht 95:4fdf968b5b37 236 }
jengbrecht 95:4fdf968b5b37 237
jengbrecht 95:4fdf968b5b37 238 //Set Local Port
jengbrecht 95:4fdf968b5b37 239 if(local_port != 0) {
jengbrecht 95:4fdf968b5b37 240 //Attempt to set local port
jengbrecht 95:4fdf968b5b37 241 sprintf(buffer, "set ip localport %d", local_port);
jengbrecht 95:4fdf968b5b37 242 Code code = sendBasicCommand(buffer, 1000);
jengbrecht 95:4fdf968b5b37 243 if(code != SUCCESS) {
jengbrecht 95:4fdf968b5b37 244 printf("[WARNING] Unable to set local port (%d) [%d]. Continuing...\r\n", local_port, (int) code);
jengbrecht 95:4fdf968b5b37 245 }
jengbrecht 95:4fdf968b5b37 246 }
jengbrecht 95:4fdf968b5b37 247
jengbrecht 95:4fdf968b5b37 248 //Set TCP/UDP parameters
jengbrecht 95:4fdf968b5b37 249 sprintf(buffer, "set ip remote %d", port);
jengbrecht 95:4fdf968b5b37 250 if(sendBasicCommand(buffer, 1000) == SUCCESS) {
jengbrecht 95:4fdf968b5b37 251 host_port = port;
jengbrecht 95:4fdf968b5b37 252 } else {
jengbrecht 95:4fdf968b5b37 253 printf("[ERROR] Host port could not be set\r\n");
jengbrecht 95:4fdf968b5b37 254 }
jengbrecht 95:4fdf968b5b37 255
jengbrecht 95:4fdf968b5b37 256 if(sendBasicCommand("set ip host " + address, 1000) == SUCCESS) {
jengbrecht 95:4fdf968b5b37 257 host_address = address;
jengbrecht 95:4fdf968b5b37 258 } else {
jengbrecht 95:4fdf968b5b37 259 printf("[ERROR] Host address could not be set\r\n");
jengbrecht 95:4fdf968b5b37 260 }
jengbrecht 95:4fdf968b5b37 261
jengbrecht 100:9d96b4391151 262 if(sendBasicCommand("set ip protocol 8", 1000) != SUCCESS) {
jengbrecht 100:9d96b4391151 263 printf("[ERROR] Failed to set TCP mode\r\n");
jengbrecht 95:4fdf968b5b37 264 }
jengbrecht 95:4fdf968b5b37 265
jengbrecht 95:4fdf968b5b37 266 // Try and Connect
jengbrecht 95:4fdf968b5b37 267 std::string sMode;
jengbrecht 95:4fdf968b5b37 268 std::string sOpenSocketCmd;
jengbrecht 95:4fdf968b5b37 269 if(mode == TCP) {
jengbrecht 95:4fdf968b5b37 270 sOpenSocketCmd = "open";
jengbrecht 95:4fdf968b5b37 271 sMode = "TCP";
jengbrecht 95:4fdf968b5b37 272 } else {
jengbrecht 100:9d96b4391151 273 //TODO
jengbrecht 100:9d96b4391151 274 //sOpenSocketCmd = "AT#OUDP";
jengbrecht 100:9d96b4391151 275 //sMode = "UDP";
jengbrecht 95:4fdf968b5b37 276 }
jengbrecht 100:9d96b4391151 277 string response = sendCommand(sOpenSocketCmd, 10000, "OPEN");
jengbrecht 95:4fdf968b5b37 278 if (response.find("OPEN") != string::npos) {
jengbrecht 95:4fdf968b5b37 279 printf("[INFO] Opened %s Socket [%s:%d]\r\n", sMode.c_str(), address.c_str(), port);
jengbrecht 95:4fdf968b5b37 280 socketOpened = true;
jengbrecht 100:9d96b4391151 281 cmdOn = false;
jengbrecht 95:4fdf968b5b37 282 } else {
jengbrecht 95:4fdf968b5b37 283 printf("[WARNING] Unable to open %s Socket [%s:%d]\r\n", sMode.c_str(), address.c_str(), port);
jengbrecht 95:4fdf968b5b37 284 socketOpened = false;
jengbrecht 95:4fdf968b5b37 285 }
jengbrecht 95:4fdf968b5b37 286
jengbrecht 95:4fdf968b5b37 287 return socketOpened;
jengbrecht 69:f3e696bbb0d5 288 }
jengbrecht 69:f3e696bbb0d5 289
jengbrecht 69:f3e696bbb0d5 290 bool Wifi::isOpen()
jengbrecht 69:f3e696bbb0d5 291 {
jengbrecht 114:cd34b1d64360 292 if(!setCmdMode(true)) {
jengbrecht 114:cd34b1d64360 293 printf("[ERROR] Failed to properly check if TCP connection is open.\r\n");
jengbrecht 114:cd34b1d64360 294 return socketOpened;
jengbrecht 114:cd34b1d64360 295 }
jengbrecht 114:cd34b1d64360 296 std::string response = sendCommand("show connection", 2000, "\n");
jengbrecht 114:cd34b1d64360 297 int start = response.find("f");
jengbrecht 114:cd34b1d64360 298 if(start != string::npos && response.size() >= (start + 3)) {
jengbrecht 114:cd34b1d64360 299 if(response[start + 3] == '1') {
jengbrecht 114:cd34b1d64360 300 socketOpened = true;
jengbrecht 114:cd34b1d64360 301 } else {
jengbrecht 114:cd34b1d64360 302 socketOpened = false;
jengbrecht 114:cd34b1d64360 303 }
jengbrecht 114:cd34b1d64360 304 } else {
jengbrecht 114:cd34b1d64360 305 printf("[WARNING] Trouble checking TCP Connection status.\n\r");
jengbrecht 114:cd34b1d64360 306 }
jengbrecht 95:4fdf968b5b37 307 return socketOpened;
jengbrecht 69:f3e696bbb0d5 308 }
jengbrecht 69:f3e696bbb0d5 309
jengbrecht 69:f3e696bbb0d5 310 bool Wifi::close()
jengbrecht 69:f3e696bbb0d5 311 {
jengbrecht 114:cd34b1d64360 312 wait(1);
jengbrecht 95:4fdf968b5b37 313 if(io == NULL) {
jengbrecht 95:4fdf968b5b37 314 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 95:4fdf968b5b37 315 return false;
jengbrecht 95:4fdf968b5b37 316 }
jengbrecht 95:4fdf968b5b37 317
jengbrecht 95:4fdf968b5b37 318 if(!socketOpened) {
jengbrecht 95:4fdf968b5b37 319 printf("[WARNING] Socket close() called, but socket was not open\r\n");
jengbrecht 95:4fdf968b5b37 320 return true;
jengbrecht 95:4fdf968b5b37 321 }
jengbrecht 95:4fdf968b5b37 322
jengbrecht 98:dbeac735109d 323 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 324 printf("[ERROR] Failed to close socket\r\n");
jengbrecht 98:dbeac735109d 325 return false;
jengbrecht 95:4fdf968b5b37 326 }
jengbrecht 95:4fdf968b5b37 327
jengbrecht 114:cd34b1d64360 328 if(isOpen()) {
jengbrecht 114:cd34b1d64360 329 std::string response = sendCommand("close", 3000, "CLOS");
jengbrecht 114:cd34b1d64360 330 if(response.find("CLOS") == string::npos) {
jengbrecht 114:cd34b1d64360 331 printf("[WARNING] Failed to successfully close socket...\r\n");
jengbrecht 114:cd34b1d64360 332 return false;
jengbrecht 114:cd34b1d64360 333 }
jengbrecht 95:4fdf968b5b37 334 }
jengbrecht 95:4fdf968b5b37 335
jengbrecht 113:7238f9b8db17 336 io->rxClear();
jengbrecht 113:7238f9b8db17 337 io->txClear();
jengbrecht 114:cd34b1d64360 338
jengbrecht 69:f3e696bbb0d5 339 return true;
jengbrecht 69:f3e696bbb0d5 340 }
jengbrecht 69:f3e696bbb0d5 341
jengbrecht 69:f3e696bbb0d5 342 int Wifi::read(char* data, int max, int timeout)
jengbrecht 69:f3e696bbb0d5 343 {
jengbrecht 98:dbeac735109d 344 if(io == NULL) {
jengbrecht 98:dbeac735109d 345 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 98:dbeac735109d 346 return -1;
jengbrecht 98:dbeac735109d 347 }
jengbrecht 98:dbeac735109d 348
jengbrecht 98:dbeac735109d 349 //Check that nothing is in the rx buffer
jengbrecht 98:dbeac735109d 350 if(!socketOpened && !io->readable()) {
jengbrecht 98:dbeac735109d 351 printf("[ERROR] Socket is not open\r\n");
jengbrecht 98:dbeac735109d 352 return -1;
jengbrecht 98:dbeac735109d 353 }
jengbrecht 98:dbeac735109d 354
jengbrecht 98:dbeac735109d 355 //Check for data mode
jengbrecht 98:dbeac735109d 356 if(!setCmdMode(false)) {
jengbrecht 98:dbeac735109d 357 printf("[ERROR] Failed to read data due to mode\r\n");
jengbrecht 98:dbeac735109d 358 return -1;
jengbrecht 98:dbeac735109d 359 }
jengbrecht 98:dbeac735109d 360
jengbrecht 98:dbeac735109d 361 int bytesRead = 0;
jengbrecht 98:dbeac735109d 362
jengbrecht 98:dbeac735109d 363 if(timeout >= 0) {
jengbrecht 98:dbeac735109d 364 bytesRead = io->read(data, max, static_cast<unsigned int>(timeout));
jengbrecht 98:dbeac735109d 365 } else {
jengbrecht 98:dbeac735109d 366 bytesRead = io->read(data, max);
jengbrecht 98:dbeac735109d 367 }
jengbrecht 98:dbeac735109d 368
jengbrecht 98:dbeac735109d 369 return bytesRead;
jengbrecht 69:f3e696bbb0d5 370 }
jengbrecht 69:f3e696bbb0d5 371
jengbrecht 69:f3e696bbb0d5 372 int Wifi::write(const char* data, int length, int timeout)
jengbrecht 69:f3e696bbb0d5 373 {
jengbrecht 98:dbeac735109d 374 if(io == NULL) {
jengbrecht 98:dbeac735109d 375 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 98:dbeac735109d 376 return -1;
jengbrecht 98:dbeac735109d 377 }
jengbrecht 98:dbeac735109d 378
jengbrecht 98:dbeac735109d 379 if(!socketOpened) {
jengbrecht 98:dbeac735109d 380 printf("[ERROR] Socket is not open\r\n");
jengbrecht 98:dbeac735109d 381 return -1;
jengbrecht 98:dbeac735109d 382 }
jengbrecht 98:dbeac735109d 383
jengbrecht 98:dbeac735109d 384 //Check for data mode
jengbrecht 98:dbeac735109d 385 if(!setCmdMode(false)) {
jengbrecht 98:dbeac735109d 386 printf("[ERROR] Failed to write data due to mode\r\n");
jengbrecht 98:dbeac735109d 387 return -1;
jengbrecht 98:dbeac735109d 388 }
jengbrecht 98:dbeac735109d 389
jengbrecht 98:dbeac735109d 390 int bytesWritten = 0;
jengbrecht 103:da58d27c15d7 391
jengbrecht 98:dbeac735109d 392 if(timeout >= 0) {
jengbrecht 100:9d96b4391151 393 bytesWritten = io->write(data, length, static_cast<unsigned int>(timeout));
jengbrecht 98:dbeac735109d 394 } else {
jengbrecht 100:9d96b4391151 395 bytesWritten = io->write(data, length);
jengbrecht 98:dbeac735109d 396 }
jengbrecht 103:da58d27c15d7 397
jengbrecht 98:dbeac735109d 398 return bytesWritten;
jengbrecht 69:f3e696bbb0d5 399 }
jengbrecht 69:f3e696bbb0d5 400
jengbrecht 69:f3e696bbb0d5 401 unsigned int Wifi::readable()
jengbrecht 69:f3e696bbb0d5 402 {
jengbrecht 69:f3e696bbb0d5 403 if(io == NULL) {
jengbrecht 69:f3e696bbb0d5 404 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 69:f3e696bbb0d5 405 return 0;
jengbrecht 69:f3e696bbb0d5 406 }
jengbrecht 69:f3e696bbb0d5 407 if(!socketOpened) {
jengbrecht 69:f3e696bbb0d5 408 printf("[ERROR] Socket is not open\r\n");
jengbrecht 69:f3e696bbb0d5 409 return 0;
jengbrecht 69:f3e696bbb0d5 410 }
jengbrecht 69:f3e696bbb0d5 411 return io->readable();
jengbrecht 69:f3e696bbb0d5 412 }
jengbrecht 69:f3e696bbb0d5 413
jengbrecht 69:f3e696bbb0d5 414 unsigned int Wifi::writeable()
jengbrecht 69:f3e696bbb0d5 415 {
jengbrecht 69:f3e696bbb0d5 416 if(io == NULL) {
jengbrecht 69:f3e696bbb0d5 417 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 69:f3e696bbb0d5 418 return 0;
jengbrecht 69:f3e696bbb0d5 419 }
jengbrecht 69:f3e696bbb0d5 420 if(!socketOpened) {
jengbrecht 69:f3e696bbb0d5 421 printf("[ERROR] Socket is not open\r\n");
jengbrecht 69:f3e696bbb0d5 422 return 0;
jengbrecht 69:f3e696bbb0d5 423 }
jengbrecht 69:f3e696bbb0d5 424
jengbrecht 69:f3e696bbb0d5 425 return io->writeable();
jengbrecht 69:f3e696bbb0d5 426 }
jengbrecht 69:f3e696bbb0d5 427
jengbrecht 69:f3e696bbb0d5 428 void Wifi::reset()
jengbrecht 69:f3e696bbb0d5 429 {
jengbrecht 69:f3e696bbb0d5 430 }
jengbrecht 69:f3e696bbb0d5 431
jengbrecht 106:358972176b89 432 Code Wifi::setDeviceIP(std::string address)
jengbrecht 106:358972176b89 433 {
jengbrecht 108:554585370b4a 434 //Check for command mode
jengbrecht 108:554585370b4a 435 if(!setCmdMode(true)) {
jengbrecht 108:554585370b4a 436 printf("[ERROR] Failed to set IP due to mode issue\r\n");
jengbrecht 108:554585370b4a 437 return FAILURE;
jengbrecht 108:554585370b4a 438 }
jengbrecht 113:7238f9b8db17 439
jengbrecht 106:358972176b89 440 //Set to DHCP mode
jengbrecht 106:358972176b89 441 if(address.compare("DHCP") == 0) {
jengbrecht 106:358972176b89 442 return sendBasicCommand("set ip dhcp 1", 1000);
jengbrecht 106:358972176b89 443 }
jengbrecht 106:358972176b89 444
jengbrecht 106:358972176b89 445 //Set to static mode and set address
jengbrecht 106:358972176b89 446 Code code = sendBasicCommand("set ip address " + address, 1000);
jengbrecht 106:358972176b89 447 if(code != SUCCESS) {
jengbrecht 106:358972176b89 448 return code;
jengbrecht 106:358972176b89 449 }
jengbrecht 106:358972176b89 450 code = sendBasicCommand("set ip dhcp 0", 1000);
jengbrecht 106:358972176b89 451 if(code != SUCCESS) {
jengbrecht 106:358972176b89 452 return code;
jengbrecht 106:358972176b89 453 }
jengbrecht 106:358972176b89 454 local_address = address;
jengbrecht 106:358972176b89 455 return SUCCESS;
jengbrecht 106:358972176b89 456 }
jengbrecht 106:358972176b89 457
jengbrecht 106:358972176b89 458 std::string Wifi::getDeviceIP()
jengbrecht 106:358972176b89 459 {
jengbrecht 106:358972176b89 460 return local_address;
jengbrecht 106:358972176b89 461 }
jengbrecht 106:358972176b89 462
jengbrecht 103:da58d27c15d7 463 Code Wifi::setNetwork(const std::string& ssid, SecurityType type, const std::string& key)
jengbrecht 69:f3e696bbb0d5 464 {
jengbrecht 98:dbeac735109d 465 //Check the command mode
jengbrecht 98:dbeac735109d 466 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 467 return FAILURE;
jengbrecht 98:dbeac735109d 468 }
jengbrecht 98:dbeac735109d 469
sgodinez 73:bb5bbca971ae 470 Code code;
jengbrecht 69:f3e696bbb0d5 471
jengbrecht 69:f3e696bbb0d5 472 //Set the appropraite SSID
jengbrecht 69:f3e696bbb0d5 473 code = sendBasicCommand("set wlan ssid " + ssid, 1000);
sgodinez 73:bb5bbca971ae 474 if (code != SUCCESS) {
jengbrecht 69:f3e696bbb0d5 475 return code;
jengbrecht 69:f3e696bbb0d5 476 }
jengbrecht 69:f3e696bbb0d5 477
jengbrecht 69:f3e696bbb0d5 478 //Set the security key
jengbrecht 69:f3e696bbb0d5 479 if (type == WEP64 || type == WEP128) {
jengbrecht 69:f3e696bbb0d5 480 //Set the WEP key if using WEP encryption
jengbrecht 69:f3e696bbb0d5 481 code = sendBasicCommand("set wlan key " + key, 1000);
sgodinez 73:bb5bbca971ae 482 if (code != SUCCESS) {
jengbrecht 69:f3e696bbb0d5 483 return code;
jengbrecht 69:f3e696bbb0d5 484 }
jengbrecht 69:f3e696bbb0d5 485 } else if (type == WPA || type == WPA2) {
jengbrecht 69:f3e696bbb0d5 486 //Set the WPA key if using WPA encryption
jengbrecht 69:f3e696bbb0d5 487 code = sendBasicCommand("set wlan phrase " + key, 1000);
sgodinez 73:bb5bbca971ae 488 if (code != SUCCESS) {
jengbrecht 69:f3e696bbb0d5 489 return code;
jengbrecht 69:f3e696bbb0d5 490 }
jengbrecht 69:f3e696bbb0d5 491 }
jengbrecht 69:f3e696bbb0d5 492
jengbrecht 69:f3e696bbb0d5 493 _ssid = ssid;
sgodinez 73:bb5bbca971ae 494 return SUCCESS;
jengbrecht 69:f3e696bbb0d5 495 }
jengbrecht 69:f3e696bbb0d5 496
jengbrecht 94:1baa587e89ae 497 Code Wifi::setDNS(const std::string& dnsName)
jengbrecht 94:1baa587e89ae 498 {
jengbrecht 98:dbeac735109d 499 //Check the command mode
jengbrecht 98:dbeac735109d 500 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 501 return FAILURE;
jengbrecht 98:dbeac735109d 502 }
jengbrecht 98:dbeac735109d 503
jengbrecht 94:1baa587e89ae 504 return sendBasicCommand("set dns name " + dnsName, 1000);
jengbrecht 94:1baa587e89ae 505 }
jengbrecht 94:1baa587e89ae 506
jengbrecht 69:f3e696bbb0d5 507 int Wifi::getSignalStrength()
jengbrecht 69:f3e696bbb0d5 508 {
jengbrecht 103:da58d27c15d7 509 //Signal strength does not report correctly if not connected
jengbrecht 103:da58d27c15d7 510 if(!wifiConnected) {
jengbrecht 103:da58d27c15d7 511 printf("[ERROR] Could not get RSSI, Wifi network not connected.\n\r");
jengbrecht 103:da58d27c15d7 512 return 99;
jengbrecht 103:da58d27c15d7 513 }
jengbrecht 103:da58d27c15d7 514
jengbrecht 98:dbeac735109d 515 //Check the command mode
jengbrecht 98:dbeac735109d 516 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 517 printf("[ERROR] Could not get RSSI\n\r");
jengbrecht 103:da58d27c15d7 518 return 99;
jengbrecht 98:dbeac735109d 519 }
jengbrecht 98:dbeac735109d 520
jengbrecht 93:aa7a48e65974 521 string response = sendCommand("show rssi", 2000, "dBm");
jengbrecht 74:9f87bd22c222 522 if (response.find("RSSI") == string::npos) {
jengbrecht 98:dbeac735109d 523 printf("[ERROR] Could not get RSSI\n\r");
jengbrecht 103:da58d27c15d7 524 return 99;
jengbrecht 74:9f87bd22c222 525 }
jengbrecht 74:9f87bd22c222 526 int start = response.find('(');
jengbrecht 74:9f87bd22c222 527 int stop = response.find(')', start);
jengbrecht 74:9f87bd22c222 528 string signal = response.substr(start + 1, stop - start - 1);
jengbrecht 74:9f87bd22c222 529 int value;
jengbrecht 74:9f87bd22c222 530 sscanf(signal.c_str(), "%d", &value);
jengbrecht 74:9f87bd22c222 531 return value;
jengbrecht 69:f3e696bbb0d5 532 }
jengbrecht 69:f3e696bbb0d5 533
jengbrecht 95:4fdf968b5b37 534 bool Wifi::ping(const std::string& address)
jengbrecht 95:4fdf968b5b37 535 {
jengbrecht 98:dbeac735109d 536 //Check the command mode
jengbrecht 98:dbeac735109d 537 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 538 printf("[ERROR] Could not send ping command\n\r");
jengbrecht 98:dbeac735109d 539 return false;
jengbrecht 98:dbeac735109d 540 }
jengbrecht 98:dbeac735109d 541
jengbrecht 95:4fdf968b5b37 542 std::string response;
jengbrecht 95:4fdf968b5b37 543 for (int i = 0; i < PINGNUM; i++) {
jengbrecht 95:4fdf968b5b37 544 response = sendCommand("ping " + address, PINGDELAY * 1000, "reply");
jengbrecht 95:4fdf968b5b37 545 if (response.find("reply") != std::string::npos) {
jengbrecht 95:4fdf968b5b37 546 return true;
jengbrecht 95:4fdf968b5b37 547 }
jengbrecht 95:4fdf968b5b37 548 }
jengbrecht 95:4fdf968b5b37 549 return false;
jengbrecht 95:4fdf968b5b37 550 }
jengbrecht 95:4fdf968b5b37 551
jengbrecht 79:f356009dbc12 552 bool Wifi::setCmdMode(bool on)
jengbrecht 74:9f87bd22c222 553 {
jengbrecht 79:f356009dbc12 554 if (on) {
jengbrecht 79:f356009dbc12 555 if (cmdOn) {
jengbrecht 79:f356009dbc12 556 return true;
jengbrecht 79:f356009dbc12 557 }
jengbrecht 79:f356009dbc12 558 wait(.5);
jengbrecht 93:aa7a48e65974 559 std::string response = sendCommand("$$", 2000, "CMD", '$');
jengbrecht 79:f356009dbc12 560 if (response.find("CMD") != string::npos) {
jengbrecht 79:f356009dbc12 561 cmdOn = true;
jengbrecht 113:7238f9b8db17 562 wait(.5);
jengbrecht 79:f356009dbc12 563 return true;
jengbrecht 79:f356009dbc12 564 }
jengbrecht 98:dbeac735109d 565 printf("[ERROR] Failed to enter command mode\n\r");
jengbrecht 79:f356009dbc12 566 return false;
jengbrecht 79:f356009dbc12 567 } else {
jengbrecht 79:f356009dbc12 568 if (!cmdOn) {
jengbrecht 79:f356009dbc12 569 return true;
jengbrecht 79:f356009dbc12 570 }
jengbrecht 93:aa7a48e65974 571 std::string response = sendCommand("exit", 2000, "EXIT");
jengbrecht 79:f356009dbc12 572 if (response.find("EXIT") != string::npos) {
jengbrecht 79:f356009dbc12 573 cmdOn = false;
jengbrecht 79:f356009dbc12 574 return true;
jengbrecht 79:f356009dbc12 575 }
jengbrecht 98:dbeac735109d 576 printf("[ERROR] Failed to exit command mode\n\r");
jengbrecht 79:f356009dbc12 577 return false;
jengbrecht 74:9f87bd22c222 578 }
jengbrecht 74:9f87bd22c222 579 }
jengbrecht 69:f3e696bbb0d5 580
sgodinez 73:bb5bbca971ae 581 Code Wifi::sendBasicCommand(string command, int timeoutMillis, char esc)
jengbrecht 69:f3e696bbb0d5 582 {
jengbrecht 69:f3e696bbb0d5 583 if(socketOpened) {
jengbrecht 69:f3e696bbb0d5 584 printf("[ERROR] socket is open. Can not send AT commands\r\n");
sgodinez 73:bb5bbca971ae 585 return ERROR;
jengbrecht 69:f3e696bbb0d5 586 }
jengbrecht 69:f3e696bbb0d5 587
jengbrecht 93:aa7a48e65974 588 string response = sendCommand(command, timeoutMillis, "AOK", esc);
jengbrecht 69:f3e696bbb0d5 589 //printf("Response: %s\n\r", response.c_str());
jengbrecht 69:f3e696bbb0d5 590 if (response.size() == 0) {
sgodinez 73:bb5bbca971ae 591 return NO_RESPONSE;
jengbrecht 69:f3e696bbb0d5 592 } else if (response.find("AOK") != string::npos) {
sgodinez 73:bb5bbca971ae 593 return SUCCESS;
jengbrecht 69:f3e696bbb0d5 594 } else if (response.find("ERR") != string::npos) {
sgodinez 73:bb5bbca971ae 595 return ERROR;
jengbrecht 69:f3e696bbb0d5 596 } else {
sgodinez 73:bb5bbca971ae 597 return FAILURE;
jengbrecht 69:f3e696bbb0d5 598 }
jengbrecht 69:f3e696bbb0d5 599 }
jengbrecht 69:f3e696bbb0d5 600
jengbrecht 93:aa7a48e65974 601 string Wifi::sendCommand(string command, int timeoutMillis, std::string response, char esc)
jengbrecht 69:f3e696bbb0d5 602 {
jengbrecht 69:f3e696bbb0d5 603 if(io == NULL) {
jengbrecht 69:f3e696bbb0d5 604 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 69:f3e696bbb0d5 605 return "";
jengbrecht 69:f3e696bbb0d5 606 }
jengbrecht 106:358972176b89 607 //if(socketOpened && command.compare("$$") != 0 && command.compare("exit") != 0 && command.compare("close") != 0) {
jengbrecht 106:358972176b89 608 // printf("[ERROR] socket is open. Can not send AT commands\r\n");
jengbrecht 106:358972176b89 609 // return "";
jengbrecht 106:358972176b89 610 //}
jengbrecht 69:f3e696bbb0d5 611
jengbrecht 69:f3e696bbb0d5 612 io->rxClear();
jengbrecht 69:f3e696bbb0d5 613 io->txClear();
jengbrecht 69:f3e696bbb0d5 614 std::string result;
jengbrecht 79:f356009dbc12 615
sgodinez 73:bb5bbca971ae 616 //Attempt to write command
sgodinez 73:bb5bbca971ae 617 if(io->write(command.data(), command.size(), timeoutMillis) != command.size()) {
sgodinez 73:bb5bbca971ae 618 //Failed to write command
sgodinez 73:bb5bbca971ae 619 printf("[ERROR] failed to send command to radio within %d milliseconds\r\n", timeoutMillis);
sgodinez 73:bb5bbca971ae 620 return "";
sgodinez 73:bb5bbca971ae 621 }
jengbrecht 79:f356009dbc12 622
sgodinez 73:bb5bbca971ae 623 //Send Escape Character
sgodinez 73:bb5bbca971ae 624 if (esc != 0x00) {
sgodinez 73:bb5bbca971ae 625 if(io->write(esc, timeoutMillis) != 1) {
sgodinez 73:bb5bbca971ae 626 printf("[ERROR] failed to send '%c' to radio within %d milliseconds\r\n", esc, timeoutMillis);
sgodinez 73:bb5bbca971ae 627 return "";
sgodinez 73:bb5bbca971ae 628 }
sgodinez 73:bb5bbca971ae 629 }
sgodinez 73:bb5bbca971ae 630
jengbrecht 69:f3e696bbb0d5 631 int timer = 0;
sgodinez 73:bb5bbca971ae 632 size_t previous = 0;
jengbrecht 69:f3e696bbb0d5 633 char tmp[256];
jengbrecht 69:f3e696bbb0d5 634 tmp[255] = 0;
jengbrecht 69:f3e696bbb0d5 635 bool done = false;
jengbrecht 69:f3e696bbb0d5 636 do {
jengbrecht 93:aa7a48e65974 637 wait(.2);
jengbrecht 93:aa7a48e65974 638 timer = timer + 200;
sgodinez 73:bb5bbca971ae 639 previous = result.size();
jengbrecht 93:aa7a48e65974 640 int size = io->read(tmp, 255, 0); //1 less than allocated
jengbrecht 69:f3e696bbb0d5 641 if(size > 0) {
jengbrecht 69:f3e696bbb0d5 642 result.append(tmp, size);
jengbrecht 93:aa7a48e65974 643 if (response.size() != 0) {
jengbrecht 93:aa7a48e65974 644 if (result.find(response) != string::npos) {
jengbrecht 93:aa7a48e65974 645 return result;
jengbrecht 93:aa7a48e65974 646 }
jengbrecht 93:aa7a48e65974 647 } else {
jengbrecht 93:aa7a48e65974 648 done = (result.size() == previous);
jengbrecht 69:f3e696bbb0d5 649 }
jengbrecht 69:f3e696bbb0d5 650 }
jengbrecht 69:f3e696bbb0d5 651 if(timer >= timeoutMillis) {
jengbrecht 93:aa7a48e65974 652 printf("[WARNING] sendCommand [%s] timed out after %d milliseconds\r\n", command.c_str(), timeoutMillis);
jengbrecht 69:f3e696bbb0d5 653 done = true;
jengbrecht 69:f3e696bbb0d5 654 }
jengbrecht 69:f3e696bbb0d5 655 } while (!done);
jengbrecht 79:f356009dbc12 656
jengbrecht 93:aa7a48e65974 657 //printf("Result: %s\n\r", result.c_str());
jengbrecht 69:f3e696bbb0d5 658 return result;
jengbrecht 69:f3e696bbb0d5 659 }