基于QT实现的MQTT客户端

/* * qmqtt_client_p.cpp - qmqtt client private * * Copyright (c) 2013 Ery Lee <ery.lee at gmail dot com> * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * * Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * Neither the name of mqttc nor the names of its contributors may be used * to endorse or promote products derived from this software without * specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE * POSSIBILITY OF SUCH DAMAGE. * */ #include "qmqtt_client_p.h" #include "qmqtt_network_p.h" #include "qmqtt_frame.h" #include "qmqtt_message.h" #include <QLoggingCategory> #include <QUuid> #ifndef QT_NO_SSL #include <QFile> #include <QSslConfiguration> #include <QSslKey> #endif // QT_NO_SSL Q_LOGGING_CATEGORY(client, "qmqtt.client") static const quint8 QOS0 = 0x00; static const quint8 QOS1 = 0x01; static const quint8 QOS2 = 0x02; QMQTT::ClientPrivate::ClientPrivate(Client* qq_ptr) : _host(QHostAddress::LocalHost) , _port(1883) , _gmid(1) , _version(MQTTVersion::V3_1_0) , _clientId(QUuid::createUuid().toString()) , _cleanSession(false) , _connectionState(STATE_INIT) , _willQos(0) , _willRetain(false) , q_ptr(qq_ptr) { setKeepAlive(300); } QMQTT::ClientPrivate::~ClientPrivate() { } void QMQTT::ClientPrivate::init(const QHostAddress& host, const quint16 port, NetworkInterface* network) { _host = host; _port = port; if(network == NULL) { init(new Network); } else { init(network); } } #ifndef QT_NO_SSL void QMQTT::ClientPrivate::init(const QString& hostName, const quint16 port, const QSslConfiguration &config, const bool ignoreSelfSigned) { _hostName = hostName; _port = port; init(new Network(config, ignoreSelfSigned)); } #endif // QT_NO_SSL void QMQTT::ClientPrivate::init(const QString& hostName, const quint16 port, const bool ssl, const bool ignoreSelfSigned) { _hostName = hostName; _port = port; if (ssl) { #ifndef QT_NO_SSL QSslConfiguration sslConf = QSslConfiguration::defaultConfiguration(); QList<QSslCertificate> certs = QSslCertificate::fromPath(QStringLiteral("./cert.crt")); if (!certs.isEmpty()) sslConf.setLocalCertificate(certs.first()); QFile file(QStringLiteral("./cert.key")); if (file.open(QIODevice::ReadOnly)) { sslConf.setPrivateKey(QSslKey(file.readAll(), QSsl::Rsa)); } sslConf.setPeerVerifyMode(QSslSocket::VerifyNone); init(hostName, port, sslConf, ignoreSelfSigned); #else Q_UNUSED(ignoreSelfSigned) qCritical() << "SSL not supported in this QT build"; #endif // QT_NO_SSL } else { init(new Network); } } #ifdef QT_WEBSOCKETS_LIB void QMQTT::ClientPrivate::init(const QString& url, const QString& origin, QWebSocketProtocol::Version version, const QSslConfiguration* sslConfig, bool ignoreSelfSigned) { _hostName = url; init(new Network(origin, version, sslConfig, ignoreSelfSigned)); } #endif // QT_WEBSOCKETS_LIB void QMQTT::ClientPrivate::init(NetworkInterface* network) { Q_Q(Client); _network.reset(network); _timer.setSingleShot(true); _pingResponseTimer.setSingleShot(true); QObject::connect(&_timer, &QTimer::timeout, q, &Client::onTimerPingReq); QObject::connect(&_pingResponseTimer, &QTimer::timeout, q, &Client::onPingTimeout); QObject::connect(_network.data(), &Network::connected, q, &Client::onNetworkConnected); QObject::connect(_network.data(), &Network::disconnected, q, &Client::onNetworkDisconnected); QObject::connect(_network.data(), &Network::received, q, &Client::onNetworkReceived); QObject::connect(_network.data(), &Network::error, q, &Client::onNetworkError); } void QMQTT::ClientPrivate::connectToHost() { if (_hostName.isEmpty()) { _network->connectToHost(_host, _port); } else { _network->connectToHost(_hostName, _port); } } void QMQTT::ClientPrivate::onNetworkConnected() { sendConnect(); } void QMQTT::ClientPrivate::sendConnect() { quint8 header = CONNECT; quint8 flags = 0; //header Frame frame(header); //flags flags = FLAG_CLEANSESS(flags, _cleanSession ? 1 : 0 ); flags = FLAG_WILL(flags, willTopic().isEmpty() ? 0 : 1); if (!willTopic().isEmpty()) { flags = FLAG_WILLQOS(flags, willQos()); flags = FLAG_WILLRETAIN(flags, willRetain() ? 1 : 0); } if (!username().isEmpty()) { flags = FLAG_USERNAME(flags, 1); flags = FLAG_PASSWD(flags, !password().isEmpty() ? 1 : 0); } //payload if(_version == V3_1_1) { frame.writeString(QStringLiteral(PROTOCOL_MAGIC_3_1_1)); } else { frame.writeString(QStringLiteral(PROTOCOL_MAGIC_3_1_0)); } frame.writeChar(_version); frame.writeChar(flags); frame.writeInt(keepAlive()); frame.writeString(_clientId); if(!willTopic().isEmpty()) { frame.writeString(willTopic()); // According to the specs (http://docs.oasis-open.org/mqtt/mqtt/v3.1.1/os/mqtt-v3.1.1-os.html#_Toc398718031) // the will message gets always sent together with the topic, also when it is empty which is perfectly valid. frame.writeByteArray(_willMessage); } if (!_username.isEmpty()) { frame.writeString(_username); if (!_password.isEmpty()) { frame.writeByteArray(_password); } } sendFrame(frame); } quint16 QMQTT::ClientPrivate::sendPublish(const Message &message) { quint16 msgid = message.id(); quint8 header = PUBLISH; header = SETRETAIN(header, message.retain() ? 1 : 0); header = SETQOS(header, message.qos()); header = SETDUP(header, message.dup() ? 1 : 0); Frame frame(header); frame.writeString(message.topic()); if(message.qos() > QOS0) { if (msgid == 0) msgid = nextmid(); frame.writeInt(msgid); } if(!message.payload().isEmpty()) { frame.writeRawData(message.payload()); } sendFrame(frame); return msgid; } void QMQTT::ClientPrivate::sendPuback(const quint8 type, const quint16 mid) { Frame frame(type); frame.writeInt(mid); sendFrame(frame); } quint16 QMQTT::ClientPrivate::sendSubscribe(const QString & topic, const quint8 qos) { quint16 mid = nextmid(); Frame frame(SETQOS(SUBSCRIBE, QOS1)); frame.writeInt(mid); frame.writeString(topic); frame.writeChar(qos); sendFrame(frame);