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@113:7238f9b8db17, 2013-12-31 (annotated)

Committer:
jengbrecht
Date:
Tue Dec 31 19:57:21 2013 +0000
Revision:
113:7238f9b8db17
Parent:
108:554585370b4a
Child:
114:cd34b1d64360
Made minor robustness improvements to Wifi close method and added NONE default to the transport class

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 79:f356009dbc12 159 if(socketOpened) {
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 95:4fdf968b5b37 292 return socketOpened;
jengbrecht 69:f3e696bbb0d5 293 }
jengbrecht 69:f3e696bbb0d5 294
jengbrecht 69:f3e696bbb0d5 295 bool Wifi::close()
jengbrecht 69:f3e696bbb0d5 296 {
jengbrecht 95:4fdf968b5b37 297 if(io == NULL) {
jengbrecht 95:4fdf968b5b37 298 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 95:4fdf968b5b37 299 return false;
jengbrecht 95:4fdf968b5b37 300 }
jengbrecht 95:4fdf968b5b37 301
jengbrecht 95:4fdf968b5b37 302 if(!socketOpened) {
jengbrecht 95:4fdf968b5b37 303 printf("[WARNING] Socket close() called, but socket was not open\r\n");
jengbrecht 95:4fdf968b5b37 304 return true;
jengbrecht 95:4fdf968b5b37 305 }
jengbrecht 95:4fdf968b5b37 306
jengbrecht 98:dbeac735109d 307 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 308 printf("[ERROR] Failed to close socket\r\n");
jengbrecht 98:dbeac735109d 309 return false;
jengbrecht 95:4fdf968b5b37 310 }
jengbrecht 95:4fdf968b5b37 311
jengbrecht 95:4fdf968b5b37 312 std::string response = sendCommand("close", 10000, "CLOS");
jengbrecht 113:7238f9b8db17 313 //printf("Close Response: %s\n\r", response.c_str());
sgodinez 97:5e65d8f334d5 314 if(response.find("CLOS") == string::npos) {
jengbrecht 98:dbeac735109d 315 printf("[ERROR] Failed to successfully close socket\r\n");
jengbrecht 95:4fdf968b5b37 316 return false;
jengbrecht 95:4fdf968b5b37 317 }
jengbrecht 95:4fdf968b5b37 318
jengbrecht 95:4fdf968b5b37 319 socketOpened = false;
jengbrecht 113:7238f9b8db17 320
jengbrecht 113:7238f9b8db17 321 io->rxClear();
jengbrecht 113:7238f9b8db17 322 io->txClear();
jengbrecht 113:7238f9b8db17 323
jengbrecht 69:f3e696bbb0d5 324 return true;
jengbrecht 69:f3e696bbb0d5 325 }
jengbrecht 69:f3e696bbb0d5 326
jengbrecht 69:f3e696bbb0d5 327 int Wifi::read(char* data, int max, int timeout)
jengbrecht 69:f3e696bbb0d5 328 {
jengbrecht 98:dbeac735109d 329 if(io == NULL) {
jengbrecht 98:dbeac735109d 330 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 98:dbeac735109d 331 return -1;
jengbrecht 98:dbeac735109d 332 }
jengbrecht 98:dbeac735109d 333
jengbrecht 98:dbeac735109d 334 //Check that nothing is in the rx buffer
jengbrecht 98:dbeac735109d 335 if(!socketOpened && !io->readable()) {
jengbrecht 98:dbeac735109d 336 printf("[ERROR] Socket is not open\r\n");
jengbrecht 98:dbeac735109d 337 return -1;
jengbrecht 98:dbeac735109d 338 }
jengbrecht 98:dbeac735109d 339
jengbrecht 98:dbeac735109d 340 //Check for data mode
jengbrecht 98:dbeac735109d 341 if(!setCmdMode(false)) {
jengbrecht 98:dbeac735109d 342 printf("[ERROR] Failed to read data due to mode\r\n");
jengbrecht 98:dbeac735109d 343 return -1;
jengbrecht 98:dbeac735109d 344 }
jengbrecht 98:dbeac735109d 345
jengbrecht 98:dbeac735109d 346 int bytesRead = 0;
jengbrecht 98:dbeac735109d 347
jengbrecht 98:dbeac735109d 348 if(timeout >= 0) {
jengbrecht 98:dbeac735109d 349 bytesRead = io->read(data, max, static_cast<unsigned int>(timeout));
jengbrecht 98:dbeac735109d 350 } else {
jengbrecht 98:dbeac735109d 351 bytesRead = io->read(data, max);
jengbrecht 98:dbeac735109d 352 }
jengbrecht 98:dbeac735109d 353
jengbrecht 98:dbeac735109d 354 return bytesRead;
jengbrecht 69:f3e696bbb0d5 355 }
jengbrecht 69:f3e696bbb0d5 356
jengbrecht 69:f3e696bbb0d5 357 int Wifi::write(const char* data, int length, int timeout)
jengbrecht 69:f3e696bbb0d5 358 {
jengbrecht 98:dbeac735109d 359 if(io == NULL) {
jengbrecht 98:dbeac735109d 360 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 98:dbeac735109d 361 return -1;
jengbrecht 98:dbeac735109d 362 }
jengbrecht 98:dbeac735109d 363
jengbrecht 98:dbeac735109d 364 if(!socketOpened) {
jengbrecht 98:dbeac735109d 365 printf("[ERROR] Socket is not open\r\n");
jengbrecht 98:dbeac735109d 366 return -1;
jengbrecht 98:dbeac735109d 367 }
jengbrecht 98:dbeac735109d 368
jengbrecht 98:dbeac735109d 369 //Check for data mode
jengbrecht 98:dbeac735109d 370 if(!setCmdMode(false)) {
jengbrecht 98:dbeac735109d 371 printf("[ERROR] Failed to write data due to mode\r\n");
jengbrecht 98:dbeac735109d 372 return -1;
jengbrecht 98:dbeac735109d 373 }
jengbrecht 98:dbeac735109d 374
jengbrecht 98:dbeac735109d 375 int bytesWritten = 0;
jengbrecht 103:da58d27c15d7 376
jengbrecht 98:dbeac735109d 377 if(timeout >= 0) {
jengbrecht 100:9d96b4391151 378 bytesWritten = io->write(data, length, static_cast<unsigned int>(timeout));
jengbrecht 98:dbeac735109d 379 } else {
jengbrecht 100:9d96b4391151 380 bytesWritten = io->write(data, length);
jengbrecht 98:dbeac735109d 381 }
jengbrecht 103:da58d27c15d7 382
jengbrecht 98:dbeac735109d 383 return bytesWritten;
jengbrecht 69:f3e696bbb0d5 384 }
jengbrecht 69:f3e696bbb0d5 385
jengbrecht 69:f3e696bbb0d5 386 unsigned int Wifi::readable()
jengbrecht 69:f3e696bbb0d5 387 {
jengbrecht 69:f3e696bbb0d5 388 if(io == NULL) {
jengbrecht 69:f3e696bbb0d5 389 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 69:f3e696bbb0d5 390 return 0;
jengbrecht 69:f3e696bbb0d5 391 }
jengbrecht 69:f3e696bbb0d5 392 if(!socketOpened) {
jengbrecht 69:f3e696bbb0d5 393 printf("[ERROR] Socket is not open\r\n");
jengbrecht 69:f3e696bbb0d5 394 return 0;
jengbrecht 69:f3e696bbb0d5 395 }
jengbrecht 69:f3e696bbb0d5 396 return io->readable();
jengbrecht 69:f3e696bbb0d5 397 }
jengbrecht 69:f3e696bbb0d5 398
jengbrecht 69:f3e696bbb0d5 399 unsigned int Wifi::writeable()
jengbrecht 69:f3e696bbb0d5 400 {
jengbrecht 69:f3e696bbb0d5 401 if(io == NULL) {
jengbrecht 69:f3e696bbb0d5 402 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 69:f3e696bbb0d5 403 return 0;
jengbrecht 69:f3e696bbb0d5 404 }
jengbrecht 69:f3e696bbb0d5 405 if(!socketOpened) {
jengbrecht 69:f3e696bbb0d5 406 printf("[ERROR] Socket is not open\r\n");
jengbrecht 69:f3e696bbb0d5 407 return 0;
jengbrecht 69:f3e696bbb0d5 408 }
jengbrecht 69:f3e696bbb0d5 409
jengbrecht 69:f3e696bbb0d5 410 return io->writeable();
jengbrecht 69:f3e696bbb0d5 411 }
jengbrecht 69:f3e696bbb0d5 412
jengbrecht 69:f3e696bbb0d5 413 void Wifi::reset()
jengbrecht 69:f3e696bbb0d5 414 {
jengbrecht 69:f3e696bbb0d5 415 }
jengbrecht 69:f3e696bbb0d5 416
jengbrecht 106:358972176b89 417 Code Wifi::setDeviceIP(std::string address)
jengbrecht 106:358972176b89 418 {
jengbrecht 108:554585370b4a 419 //Check for command mode
jengbrecht 108:554585370b4a 420 if(!setCmdMode(true)) {
jengbrecht 108:554585370b4a 421 printf("[ERROR] Failed to set IP due to mode issue\r\n");
jengbrecht 108:554585370b4a 422 return FAILURE;
jengbrecht 108:554585370b4a 423 }
jengbrecht 113:7238f9b8db17 424
jengbrecht 106:358972176b89 425 //Set to DHCP mode
jengbrecht 106:358972176b89 426 if(address.compare("DHCP") == 0) {
jengbrecht 106:358972176b89 427 return sendBasicCommand("set ip dhcp 1", 1000);
jengbrecht 106:358972176b89 428 }
jengbrecht 106:358972176b89 429
jengbrecht 106:358972176b89 430 //Set to static mode and set address
jengbrecht 106:358972176b89 431 Code code = sendBasicCommand("set ip address " + address, 1000);
jengbrecht 106:358972176b89 432 if(code != SUCCESS) {
jengbrecht 106:358972176b89 433 return code;
jengbrecht 106:358972176b89 434 }
jengbrecht 106:358972176b89 435 code = sendBasicCommand("set ip dhcp 0", 1000);
jengbrecht 106:358972176b89 436 if(code != SUCCESS) {
jengbrecht 106:358972176b89 437 return code;
jengbrecht 106:358972176b89 438 }
jengbrecht 106:358972176b89 439 local_address = address;
jengbrecht 106:358972176b89 440 return SUCCESS;
jengbrecht 106:358972176b89 441 }
jengbrecht 106:358972176b89 442
jengbrecht 106:358972176b89 443 std::string Wifi::getDeviceIP()
jengbrecht 106:358972176b89 444 {
jengbrecht 106:358972176b89 445 return local_address;
jengbrecht 106:358972176b89 446 }
jengbrecht 106:358972176b89 447
jengbrecht 103:da58d27c15d7 448 Code Wifi::setNetwork(const std::string& ssid, SecurityType type, const std::string& key)
jengbrecht 69:f3e696bbb0d5 449 {
jengbrecht 98:dbeac735109d 450 //Check the command mode
jengbrecht 98:dbeac735109d 451 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 452 return FAILURE;
jengbrecht 98:dbeac735109d 453 }
jengbrecht 98:dbeac735109d 454
sgodinez 73:bb5bbca971ae 455 Code code;
jengbrecht 69:f3e696bbb0d5 456
jengbrecht 69:f3e696bbb0d5 457 //Set the appropraite SSID
jengbrecht 69:f3e696bbb0d5 458 code = sendBasicCommand("set wlan ssid " + ssid, 1000);
sgodinez 73:bb5bbca971ae 459 if (code != SUCCESS) {
jengbrecht 69:f3e696bbb0d5 460 return code;
jengbrecht 69:f3e696bbb0d5 461 }
jengbrecht 69:f3e696bbb0d5 462
jengbrecht 69:f3e696bbb0d5 463 //Set the security key
jengbrecht 69:f3e696bbb0d5 464 if (type == WEP64 || type == WEP128) {
jengbrecht 69:f3e696bbb0d5 465 //Set the WEP key if using WEP encryption
jengbrecht 69:f3e696bbb0d5 466 code = sendBasicCommand("set wlan key " + key, 1000);
sgodinez 73:bb5bbca971ae 467 if (code != SUCCESS) {
jengbrecht 69:f3e696bbb0d5 468 return code;
jengbrecht 69:f3e696bbb0d5 469 }
jengbrecht 69:f3e696bbb0d5 470 } else if (type == WPA || type == WPA2) {
jengbrecht 69:f3e696bbb0d5 471 //Set the WPA key if using WPA encryption
jengbrecht 69:f3e696bbb0d5 472 code = sendBasicCommand("set wlan phrase " + key, 1000);
sgodinez 73:bb5bbca971ae 473 if (code != SUCCESS) {
jengbrecht 69:f3e696bbb0d5 474 return code;
jengbrecht 69:f3e696bbb0d5 475 }
jengbrecht 69:f3e696bbb0d5 476 }
jengbrecht 69:f3e696bbb0d5 477
jengbrecht 69:f3e696bbb0d5 478 _ssid = ssid;
sgodinez 73:bb5bbca971ae 479 return SUCCESS;
jengbrecht 69:f3e696bbb0d5 480 }
jengbrecht 69:f3e696bbb0d5 481
jengbrecht 94:1baa587e89ae 482 Code Wifi::setDNS(const std::string& dnsName)
jengbrecht 94:1baa587e89ae 483 {
jengbrecht 98:dbeac735109d 484 //Check the command mode
jengbrecht 98:dbeac735109d 485 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 486 return FAILURE;
jengbrecht 98:dbeac735109d 487 }
jengbrecht 98:dbeac735109d 488
jengbrecht 94:1baa587e89ae 489 return sendBasicCommand("set dns name " + dnsName, 1000);
jengbrecht 94:1baa587e89ae 490 }
jengbrecht 94:1baa587e89ae 491
jengbrecht 69:f3e696bbb0d5 492 int Wifi::getSignalStrength()
jengbrecht 69:f3e696bbb0d5 493 {
jengbrecht 103:da58d27c15d7 494 //Signal strength does not report correctly if not connected
jengbrecht 103:da58d27c15d7 495 if(!wifiConnected) {
jengbrecht 103:da58d27c15d7 496 printf("[ERROR] Could not get RSSI, Wifi network not connected.\n\r");
jengbrecht 103:da58d27c15d7 497 return 99;
jengbrecht 103:da58d27c15d7 498 }
jengbrecht 103:da58d27c15d7 499
jengbrecht 98:dbeac735109d 500 //Check the command mode
jengbrecht 98:dbeac735109d 501 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 502 printf("[ERROR] Could not get RSSI\n\r");
jengbrecht 103:da58d27c15d7 503 return 99;
jengbrecht 98:dbeac735109d 504 }
jengbrecht 98:dbeac735109d 505
jengbrecht 93:aa7a48e65974 506 string response = sendCommand("show rssi", 2000, "dBm");
jengbrecht 74:9f87bd22c222 507 if (response.find("RSSI") == string::npos) {
jengbrecht 98:dbeac735109d 508 printf("[ERROR] Could not get RSSI\n\r");
jengbrecht 103:da58d27c15d7 509 return 99;
jengbrecht 74:9f87bd22c222 510 }
jengbrecht 74:9f87bd22c222 511 int start = response.find('(');
jengbrecht 74:9f87bd22c222 512 int stop = response.find(')', start);
jengbrecht 74:9f87bd22c222 513 string signal = response.substr(start + 1, stop - start - 1);
jengbrecht 74:9f87bd22c222 514 int value;
jengbrecht 74:9f87bd22c222 515 sscanf(signal.c_str(), "%d", &value);
jengbrecht 74:9f87bd22c222 516 return value;
jengbrecht 69:f3e696bbb0d5 517 }
jengbrecht 69:f3e696bbb0d5 518
jengbrecht 95:4fdf968b5b37 519 bool Wifi::ping(const std::string& address)
jengbrecht 95:4fdf968b5b37 520 {
jengbrecht 98:dbeac735109d 521 //Check the command mode
jengbrecht 98:dbeac735109d 522 if(!setCmdMode(true)) {
jengbrecht 98:dbeac735109d 523 printf("[ERROR] Could not send ping command\n\r");
jengbrecht 98:dbeac735109d 524 return false;
jengbrecht 98:dbeac735109d 525 }
jengbrecht 98:dbeac735109d 526
jengbrecht 95:4fdf968b5b37 527 std::string response;
jengbrecht 95:4fdf968b5b37 528 for (int i = 0; i < PINGNUM; i++) {
jengbrecht 95:4fdf968b5b37 529 response = sendCommand("ping " + address, PINGDELAY * 1000, "reply");
jengbrecht 95:4fdf968b5b37 530 if (response.find("reply") != std::string::npos) {
jengbrecht 95:4fdf968b5b37 531 return true;
jengbrecht 95:4fdf968b5b37 532 }
jengbrecht 95:4fdf968b5b37 533 }
jengbrecht 95:4fdf968b5b37 534 return false;
jengbrecht 95:4fdf968b5b37 535 }
jengbrecht 95:4fdf968b5b37 536
jengbrecht 79:f356009dbc12 537 bool Wifi::setCmdMode(bool on)
jengbrecht 74:9f87bd22c222 538 {
jengbrecht 79:f356009dbc12 539 if (on) {
jengbrecht 79:f356009dbc12 540 if (cmdOn) {
jengbrecht 79:f356009dbc12 541 return true;
jengbrecht 79:f356009dbc12 542 }
jengbrecht 79:f356009dbc12 543 wait(.5);
jengbrecht 93:aa7a48e65974 544 std::string response = sendCommand("$$", 2000, "CMD", '$');
jengbrecht 79:f356009dbc12 545 if (response.find("CMD") != string::npos) {
jengbrecht 79:f356009dbc12 546 cmdOn = true;
jengbrecht 113:7238f9b8db17 547 wait(.5);
jengbrecht 79:f356009dbc12 548 return true;
jengbrecht 79:f356009dbc12 549 }
jengbrecht 98:dbeac735109d 550 printf("[ERROR] Failed to enter command mode\n\r");
jengbrecht 79:f356009dbc12 551 return false;
jengbrecht 79:f356009dbc12 552 } else {
jengbrecht 79:f356009dbc12 553 if (!cmdOn) {
jengbrecht 79:f356009dbc12 554 return true;
jengbrecht 79:f356009dbc12 555 }
jengbrecht 93:aa7a48e65974 556 std::string response = sendCommand("exit", 2000, "EXIT");
jengbrecht 79:f356009dbc12 557 if (response.find("EXIT") != string::npos) {
jengbrecht 79:f356009dbc12 558 cmdOn = false;
jengbrecht 79:f356009dbc12 559 return true;
jengbrecht 79:f356009dbc12 560 }
jengbrecht 98:dbeac735109d 561 printf("[ERROR] Failed to exit command mode\n\r");
jengbrecht 79:f356009dbc12 562 return false;
jengbrecht 74:9f87bd22c222 563 }
jengbrecht 74:9f87bd22c222 564 }
jengbrecht 69:f3e696bbb0d5 565
sgodinez 73:bb5bbca971ae 566 Code Wifi::sendBasicCommand(string command, int timeoutMillis, char esc)
jengbrecht 69:f3e696bbb0d5 567 {
jengbrecht 69:f3e696bbb0d5 568 if(socketOpened) {
jengbrecht 69:f3e696bbb0d5 569 printf("[ERROR] socket is open. Can not send AT commands\r\n");
sgodinez 73:bb5bbca971ae 570 return ERROR;
jengbrecht 69:f3e696bbb0d5 571 }
jengbrecht 69:f3e696bbb0d5 572
jengbrecht 93:aa7a48e65974 573 string response = sendCommand(command, timeoutMillis, "AOK", esc);
jengbrecht 69:f3e696bbb0d5 574 //printf("Response: %s\n\r", response.c_str());
jengbrecht 69:f3e696bbb0d5 575 if (response.size() == 0) {
sgodinez 73:bb5bbca971ae 576 return NO_RESPONSE;
jengbrecht 69:f3e696bbb0d5 577 } else if (response.find("AOK") != string::npos) {
sgodinez 73:bb5bbca971ae 578 return SUCCESS;
jengbrecht 69:f3e696bbb0d5 579 } else if (response.find("ERR") != string::npos) {
sgodinez 73:bb5bbca971ae 580 return ERROR;
jengbrecht 69:f3e696bbb0d5 581 } else {
sgodinez 73:bb5bbca971ae 582 return FAILURE;
jengbrecht 69:f3e696bbb0d5 583 }
jengbrecht 69:f3e696bbb0d5 584 }
jengbrecht 69:f3e696bbb0d5 585
jengbrecht 93:aa7a48e65974 586 string Wifi::sendCommand(string command, int timeoutMillis, std::string response, char esc)
jengbrecht 69:f3e696bbb0d5 587 {
jengbrecht 69:f3e696bbb0d5 588 if(io == NULL) {
jengbrecht 69:f3e696bbb0d5 589 printf("[ERROR] MTSBufferedIO not set\r\n");
jengbrecht 69:f3e696bbb0d5 590 return "";
jengbrecht 69:f3e696bbb0d5 591 }
jengbrecht 106:358972176b89 592 //if(socketOpened && command.compare("$$") != 0 && command.compare("exit") != 0 && command.compare("close") != 0) {
jengbrecht 106:358972176b89 593 // printf("[ERROR] socket is open. Can not send AT commands\r\n");
jengbrecht 106:358972176b89 594 // return "";
jengbrecht 106:358972176b89 595 //}
jengbrecht 69:f3e696bbb0d5 596
jengbrecht 69:f3e696bbb0d5 597 io->rxClear();
jengbrecht 69:f3e696bbb0d5 598 io->txClear();
jengbrecht 69:f3e696bbb0d5 599 std::string result;
jengbrecht 79:f356009dbc12 600
sgodinez 73:bb5bbca971ae 601 //Attempt to write command
sgodinez 73:bb5bbca971ae 602 if(io->write(command.data(), command.size(), timeoutMillis) != command.size()) {
sgodinez 73:bb5bbca971ae 603 //Failed to write command
sgodinez 73:bb5bbca971ae 604 printf("[ERROR] failed to send command to radio within %d milliseconds\r\n", timeoutMillis);
sgodinez 73:bb5bbca971ae 605 return "";
sgodinez 73:bb5bbca971ae 606 }
jengbrecht 79:f356009dbc12 607
sgodinez 73:bb5bbca971ae 608 //Send Escape Character
sgodinez 73:bb5bbca971ae 609 if (esc != 0x00) {
sgodinez 73:bb5bbca971ae 610 if(io->write(esc, timeoutMillis) != 1) {
sgodinez 73:bb5bbca971ae 611 printf("[ERROR] failed to send '%c' to radio within %d milliseconds\r\n", esc, timeoutMillis);
sgodinez 73:bb5bbca971ae 612 return "";
sgodinez 73:bb5bbca971ae 613 }
sgodinez 73:bb5bbca971ae 614 }
sgodinez 73:bb5bbca971ae 615
jengbrecht 69:f3e696bbb0d5 616 int timer = 0;
sgodinez 73:bb5bbca971ae 617 size_t previous = 0;
jengbrecht 69:f3e696bbb0d5 618 char tmp[256];
jengbrecht 69:f3e696bbb0d5 619 tmp[255] = 0;
jengbrecht 69:f3e696bbb0d5 620 bool done = false;
jengbrecht 69:f3e696bbb0d5 621 do {
jengbrecht 93:aa7a48e65974 622 wait(.2);
jengbrecht 93:aa7a48e65974 623 timer = timer + 200;
sgodinez 73:bb5bbca971ae 624 previous = result.size();
jengbrecht 93:aa7a48e65974 625 int size = io->read(tmp, 255, 0); //1 less than allocated
jengbrecht 69:f3e696bbb0d5 626 if(size > 0) {
jengbrecht 69:f3e696bbb0d5 627 result.append(tmp, size);
jengbrecht 93:aa7a48e65974 628 if (response.size() != 0) {
jengbrecht 93:aa7a48e65974 629 if (result.find(response) != string::npos) {
jengbrecht 93:aa7a48e65974 630 return result;
jengbrecht 93:aa7a48e65974 631 }
jengbrecht 93:aa7a48e65974 632 } else {
jengbrecht 93:aa7a48e65974 633 done = (result.size() == previous);
jengbrecht 69:f3e696bbb0d5 634 }
jengbrecht 69:f3e696bbb0d5 635 }
jengbrecht 69:f3e696bbb0d5 636 if(timer >= timeoutMillis) {
jengbrecht 93:aa7a48e65974 637 printf("[WARNING] sendCommand [%s] timed out after %d milliseconds\r\n", command.c_str(), timeoutMillis);
jengbrecht 69:f3e696bbb0d5 638 done = true;
jengbrecht 69:f3e696bbb0d5 639 }
jengbrecht 69:f3e696bbb0d5 640 } while (!done);
jengbrecht 79:f356009dbc12 641
jengbrecht 93:aa7a48e65974 642 //printf("Result: %s\n\r", result.c_str());
jengbrecht 69:f3e696bbb0d5 643 return result;
jengbrecht 69:f3e696bbb0d5 644 }