Andrew Reed
/
CITY1082-i2c_master_wifi_mqtt
CITY3032-wifi-mqtt
Diff: src/MQTTNetwork.h
- Revision:
- 3:62825c5f3cd7
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/MQTTNetwork.h Sat Nov 13 09:06:17 2021 +0000 @@ -0,0 +1,81 @@ +/* + * Copyright (c) 2019, ARM Limited, All Rights Reserved + * SPDX-License-Identifier: Apache-2.0 + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may + * not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT + * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#ifndef _MQTTNETWORK_H_ +#define _MQTTNETWORK_H_ + +#include "NetworkInterface.h" +#include "TCPSocket.h" + +class MQTTNetwork { +public: + MQTTNetwork(NetworkInterface *aNetwork) : network(aNetwork) + { + socket = new TCPSocket(); + } + + ~MQTTNetwork() + { + delete socket; + } + + int read(unsigned char *buffer, int len, int timeout) + { + int ret = socket->recv(buffer, len); + if (ret == 0) { + // A receive size of 0 indicates that the socket + // was successfully closed so indicate this to MQTTClient + ret = -1; + } + return ret; + } + + int write(unsigned char *buffer, int len, int timeout) + { + int ret = socket->send(buffer, len); + if (ret == 0) { + // The socket is closed so indicate this to MQTTClient + return -1; + } + return ret; + } + + int connect(const char *hostname, int port) + { + int ret = NSAPI_ERROR_OK; + if ((ret = socket->open(network)) != NSAPI_ERROR_OK) { + return ret; + } + SocketAddress addr; + if (network->gethostbyname(hostname, &addr) != NSAPI_ERROR_OK) { + return NSAPI_ERROR_DNS_FAILURE; + } + addr.set_port(port); + return socket->connect(addr); + } + + int disconnect() + { + return socket->close(); + } + +private: + NetworkInterface *network; + TCPSocket *socket; +}; + +#endif // _MQTTNETWORK_H_