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@106:358972176b89, 2013-12-31 (annotated)

Committer:
jengbrecht
Date:
Tue Dec 31 16:34:22 2013 +0000
Revision:
106:358972176b89
Parent:
103:da58d27c15d7
Child:
108:554585370b4a
In Wifi added the ability to get and set the Device's IP as static or DHCP.  Also moved the DHCP setting to the init function for default setting. Added some more documentation.

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