[144] | 1 | // The MIT License (MIT)
|
---|
| 2 | //
|
---|
| 3 | // Copyright (c) 2015 Adafruit Industries
|
---|
| 4 | //
|
---|
| 5 | // Permission is hereby granted, free of charge, to any person obtaining a copy
|
---|
| 6 | // of this software and associated documentation files (the "Software"), to deal
|
---|
| 7 | // in the Software without restriction, including without limitation the rights
|
---|
| 8 | // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
---|
| 9 | // copies of the Software, and to permit persons to whom the Software is
|
---|
| 10 | // furnished to do so, subject to the following conditions:
|
---|
| 11 | //
|
---|
| 12 | // The above copyright notice and this permission notice shall be included in all
|
---|
| 13 | // copies or substantial portions of the Software.
|
---|
| 14 | //
|
---|
| 15 | // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
---|
| 16 | // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
---|
| 17 | // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
---|
| 18 | // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
---|
| 19 | // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
---|
| 20 | // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
---|
| 21 | // SOFTWARE.
|
---|
| 22 | #ifndef _ADAFRUIT_MQTT_CC3000_H_
|
---|
| 23 | #define _ADAFRUIT_MQTT_CC3000_H_
|
---|
| 24 |
|
---|
| 25 | #include <Adafruit_SleepyDog.h>
|
---|
| 26 | #include <Adafruit_CC3000.h>
|
---|
| 27 | #include "Adafruit_MQTT.h"
|
---|
| 28 |
|
---|
| 29 |
|
---|
| 30 | // delay in ms between calls of available()
|
---|
| 31 | #define MQTT_CC3000_INTERAVAILDELAY 10
|
---|
| 32 |
|
---|
| 33 |
|
---|
| 34 | // CC3000-specific version of the Adafruit_MQTT class.
|
---|
| 35 | // Note that this is defined as a header-only class to prevent issues with using
|
---|
| 36 | // the library on non-CC3000 platforms (since Arduino will include all .cpp files
|
---|
| 37 | // in the compilation of the library).
|
---|
| 38 | class Adafruit_MQTT_CC3000 : public Adafruit_MQTT {
|
---|
| 39 | public:
|
---|
| 40 | Adafruit_MQTT_CC3000(Adafruit_CC3000 *cc3k, const char *server, uint16_t port,
|
---|
| 41 | const char *cid, const char *user, const char *pass):
|
---|
| 42 | Adafruit_MQTT(server, port, cid, user, pass),
|
---|
| 43 | cc3000(cc3k)
|
---|
| 44 | {}
|
---|
| 45 |
|
---|
| 46 | bool connectServer() {
|
---|
| 47 | uint32_t ip = 0;
|
---|
| 48 |
|
---|
| 49 | Watchdog.reset();
|
---|
| 50 |
|
---|
| 51 | // look up IP address
|
---|
| 52 | if (serverip == 0) {
|
---|
| 53 | // Try looking up the website's IP address using CC3K's built in getHostByName
|
---|
| 54 | strcpy_P((char *)buffer, servername);
|
---|
| 55 | Serial.print((char *)buffer); Serial.print(F(" -> "));
|
---|
| 56 | uint8_t dnsretries = 5;
|
---|
| 57 |
|
---|
| 58 | Watchdog.reset();
|
---|
| 59 | while (ip == 0) {
|
---|
| 60 | if (! cc3000->getHostByName((char *)buffer, &ip)) {
|
---|
| 61 | Serial.println(F("Couldn't resolve!"));
|
---|
| 62 | dnsretries--;
|
---|
| 63 | Watchdog.reset();
|
---|
| 64 | }
|
---|
| 65 | //Serial.println("OK"); Serial.println(ip, HEX);
|
---|
| 66 | if (!dnsretries) return false;
|
---|
| 67 | delay(500);
|
---|
| 68 | }
|
---|
| 69 |
|
---|
| 70 | serverip = ip;
|
---|
| 71 | cc3000->printIPdotsRev(serverip);
|
---|
| 72 | Serial.println();
|
---|
| 73 | }
|
---|
| 74 |
|
---|
| 75 | Watchdog.reset();
|
---|
| 76 |
|
---|
| 77 | // connect to server
|
---|
| 78 | DEBUG_PRINTLN(F("Connecting to TCP"));
|
---|
| 79 | mqttclient = cc3000->connectTCP(serverip, portnum);
|
---|
| 80 |
|
---|
| 81 | return mqttclient.connected();
|
---|
| 82 | }
|
---|
| 83 |
|
---|
| 84 | bool disconnect() {
|
---|
| 85 | if (connected()) {
|
---|
| 86 | return (mqttclient.close() == 0);
|
---|
| 87 | }
|
---|
| 88 | else {
|
---|
| 89 | return true;
|
---|
| 90 | }
|
---|
| 91 | }
|
---|
| 92 |
|
---|
| 93 | bool connected() {
|
---|
| 94 | return mqttclient.connected();
|
---|
| 95 | }
|
---|
| 96 |
|
---|
| 97 | uint16_t readPacket(uint8_t *buffer, uint8_t maxlen, int16_t timeout,
|
---|
| 98 | bool checkForValidPubPacket = false) {
|
---|
| 99 | /* Read data until either the connection is closed, or the idle timeout is reached. */
|
---|
| 100 | uint16_t len = 0;
|
---|
| 101 | int16_t t = timeout;
|
---|
| 102 |
|
---|
| 103 | while (mqttclient.connected() && (timeout >= 0)) {
|
---|
| 104 | //DEBUG_PRINT('.');
|
---|
| 105 | while (mqttclient.available()) {
|
---|
| 106 | //DEBUG_PRINT('!');
|
---|
| 107 | char c = mqttclient.read();
|
---|
| 108 | timeout = t; // reset the timeout
|
---|
| 109 | buffer[len] = c;
|
---|
| 110 | //DEBUG_PRINTLN((uint8_t)c, HEX);
|
---|
| 111 | len++;
|
---|
| 112 | if (len == maxlen) { // we read all we want, bail
|
---|
| 113 | DEBUG_PRINT(F("Read packet:\t"));
|
---|
| 114 | DEBUG_PRINTBUFFER(buffer, len);
|
---|
| 115 | return len;
|
---|
| 116 | }
|
---|
| 117 |
|
---|
| 118 | // special case where we just one one publication packet at a time
|
---|
| 119 | if (checkForValidPubPacket) {
|
---|
| 120 | if ((buffer[0] == (MQTT_CTRL_PUBLISH << 4)) && (buffer[1] == len-2)) {
|
---|
| 121 | // oooh a valid publish packet!
|
---|
| 122 | DEBUG_PRINT(F("Read PUBLISH packet:\t"));
|
---|
| 123 | DEBUG_PRINTBUFFER(buffer, len);
|
---|
| 124 | return len;
|
---|
| 125 | }
|
---|
| 126 | }
|
---|
| 127 | }
|
---|
| 128 | Watchdog.reset();
|
---|
| 129 | timeout -= MQTT_CC3000_INTERAVAILDELAY;
|
---|
| 130 | delay(MQTT_CC3000_INTERAVAILDELAY);
|
---|
| 131 | }
|
---|
| 132 | return len;
|
---|
| 133 | }
|
---|
| 134 |
|
---|
| 135 | bool sendPacket(uint8_t *buffer, uint8_t len) {
|
---|
| 136 | if (mqttclient.connected()) {
|
---|
| 137 | uint16_t ret = mqttclient.write(buffer, len);
|
---|
| 138 | DEBUG_PRINT(F("sendPacket returned: ")); DEBUG_PRINTLN(ret);
|
---|
| 139 | if (ret != len) {
|
---|
| 140 | DEBUG_PRINTLN("Failed to send complete packet.")
|
---|
| 141 | return false;
|
---|
| 142 | }
|
---|
| 143 | } else {
|
---|
| 144 | DEBUG_PRINTLN(F("Connection failed!"));
|
---|
| 145 | return false;
|
---|
| 146 | }
|
---|
| 147 | return true;
|
---|
| 148 | }
|
---|
| 149 |
|
---|
| 150 | private:
|
---|
| 151 | uint32_t serverip;
|
---|
| 152 | Adafruit_CC3000 *cc3000;
|
---|
| 153 | Adafruit_CC3000_Client mqttclient;
|
---|
| 154 | };
|
---|
| 155 |
|
---|
| 156 |
|
---|
| 157 | #endif
|
---|