1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638 |
- /*
- Copyright (c) 2013, 2014 Paolo Patierno
- All rights reserved. This program and the accompanying materials
- are made available under the terms of the Eclipse Public License v1.0
- and Eclipse Distribution License v1.0 which accompany this distribution.
- The Eclipse Public License is available at
- http://www.eclipse.org/legal/epl-v10.html
- and the Eclipse Distribution License is available at
- http://www.eclipse.org/org/documents/edl-v10.php.
- Contributors:
- Paolo Patierno - initial API and implementation and/or initial documentation
- ----------------------------------------------------------------------------
- Giovanni Paolo Vigano' - preprocessor directives for platform dependent compilation in Unity
- */
- using System;
- using System.Net;
- #if !(WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP))
- using System.Net.Sockets;
- using System.Security.Cryptography.X509Certificates;
- #endif
- using System.Threading;
- using uPLibrary.Networking.M2Mqtt.Exceptions;
- using uPLibrary.Networking.M2Mqtt.Messages;
- using uPLibrary.Networking.M2Mqtt.Session;
- using uPLibrary.Networking.M2Mqtt.Utility;
- using uPLibrary.Networking.M2Mqtt.Internal;
- // if .Net Micro Framework
- #if (MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3)
- using Microsoft.SPOT;
- #if SSL
- using Microsoft.SPOT.Net.Security;
- #endif
- // else other frameworks (.Net, .Net Compact, Mono, Windows Phone)
- #else
- using System.Collections.Generic;
- #if (SSL && !(WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP)))
- using System.Security.Authentication;
- using System.Net.Security;
- #endif
- #endif
- #if (WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR && UNITY_WSA_10_0 && !ENABLE_IL2CPP))
- using Windows.Networking.Sockets;
- #endif
- using System.Collections;
- // alias needed due to Microsoft.SPOT.Trace in .Net Micro Framework
- // (it's ambiguos with uPLibrary.Networking.M2Mqtt.Utility.Trace)
- using MqttUtility = uPLibrary.Networking.M2Mqtt.Utility;
- using System.IO;
- using System.Net.Security;
- using UnityEngine;
- namespace uPLibrary.Networking.M2Mqtt
- {
- /// <summary>
- /// MQTT Client
- /// </summary>
- public class MqttClient
- {
- #if BROKER
- #region Constants ...
- // thread names
- private const string RECEIVE_THREAD_NAME = "ReceiveThread";
- private const string RECEIVE_EVENT_THREAD_NAME = "DispatchEventThread";
- private const string PROCESS_INFLIGHT_THREAD_NAME = "ProcessInflightThread";
- private const string KEEP_ALIVE_THREAD = "KeepAliveThread";
- #endregion
- #endif
- /// <summary>
- /// Delagate that defines event handler for PUBLISH message received
- /// </summary>
- public delegate void MqttMsgPublishEventHandler(object sender, MqttMsgPublishEventArgs e);
- /// <summary>
- /// Delegate that defines event handler for published message
- /// </summary>
- public delegate void MqttMsgPublishedEventHandler(object sender, MqttMsgPublishedEventArgs e);
- /// <summary>
- /// Delagate that defines event handler for subscribed topic
- /// </summary>
- public delegate void MqttMsgSubscribedEventHandler(object sender, MqttMsgSubscribedEventArgs e);
- /// <summary>
- /// Delagate that defines event handler for unsubscribed topic
- /// </summary>
- public delegate void MqttMsgUnsubscribedEventHandler(object sender, MqttMsgUnsubscribedEventArgs e);
- #if BROKER
- /// <summary>
- /// Delagate that defines event handler for SUBSCRIBE message received
- /// </summary>
- public delegate void MqttMsgSubscribeEventHandler(object sender, MqttMsgSubscribeEventArgs e);
- /// <summary>
- /// Delagate that defines event handler for UNSUBSCRIBE message received
- /// </summary>
- public delegate void MqttMsgUnsubscribeEventHandler(object sender, MqttMsgUnsubscribeEventArgs e);
- /// <summary>
- /// Delagate that defines event handler for CONNECT message received
- /// </summary>
- public delegate void MqttMsgConnectEventHandler(object sender, MqttMsgConnectEventArgs e);
- /// <summary>
- /// Delegate that defines event handler for client disconnection (DISCONNECT message or not)
- /// </summary>
- public delegate void MqttMsgDisconnectEventHandler(object sender, EventArgs e);
- #endif
- /// <summary>
- /// Delegate that defines event handler for cliet/peer disconnection
- /// </summary>
- public delegate void ConnectionClosedEventHandler(object sender, EventArgs e);
- // broker hostname (or ip address) and port
- private string brokerHostName;
- private int brokerPort;
- // running status of threads
- private bool isRunning;
- // event for raising received message event
- private AutoResetEvent receiveEventWaitHandle;
- // event for starting process inflight queue asynchronously
- private AutoResetEvent inflightWaitHandle;
- // event for signaling synchronous receive
- AutoResetEvent syncEndReceiving;
- // message received
- MqttMsgBase msgReceived;
- // exeption thrown during receiving
- Exception exReceiving;
- // keep alive period (in ms)
- private int keepAlivePeriod;
- // events for signaling on keep alive thread
- private AutoResetEvent keepAliveEvent;
- private AutoResetEvent keepAliveEventEnd;
- // last communication time in ticks
- private int lastCommTime;
- // event for PUBLISH message received
- public event MqttMsgPublishEventHandler MqttMsgPublishReceived;
- // event for published message
- public event MqttMsgPublishedEventHandler MqttMsgPublished;
- // event for subscribed topic
- public event MqttMsgSubscribedEventHandler MqttMsgSubscribed;
- // event for unsubscribed topic
- public event MqttMsgUnsubscribedEventHandler MqttMsgUnsubscribed;
- #if BROKER
- // event for SUBSCRIBE message received
- public event MqttMsgSubscribeEventHandler MqttMsgSubscribeReceived;
- // event for USUBSCRIBE message received
- public event MqttMsgUnsubscribeEventHandler MqttMsgUnsubscribeReceived;
- // event for CONNECT message received
- public event MqttMsgConnectEventHandler MqttMsgConnected;
- // event for DISCONNECT message received
- public event MqttMsgDisconnectEventHandler MqttMsgDisconnected;
- #endif
- // event for peer/client disconnection
- public event ConnectionClosedEventHandler ConnectionClosed;
-
- // channel to communicate over the network
- private IMqttNetworkChannel channel;
- // inflight messages queue
- private Queue inflightQueue;
- // internal queue for received messages about inflight messages
- private Queue internalQueue;
- // internal queue for dispatching events
- private Queue eventQueue;
- // session
- private MqttClientSession session;
- // reference to avoid access to singleton via property
- private MqttSettings settings;
- // current message identifier generated
- private ushort messageIdCounter = 0;
- // connection is closing due to peer
- private bool isConnectionClosing;
- /// <summary>
- /// Connection status between client and broker
- /// </summary>
- public bool IsConnected { get; private set; }
- /// <summary>
- /// Client identifier
- /// </summary>
- public string ClientId { get; private set; }
- /// <summary>
- /// Clean session flag
- /// </summary>
- public bool CleanSession { get; private set; }
- /// <summary>
- /// Will flag
- /// </summary>
- public bool WillFlag { get; private set; }
- /// <summary>
- /// Will QOS level
- /// </summary>
- public byte WillQosLevel { get; private set; }
- /// <summary>
- /// Will topic
- /// </summary>
- public string WillTopic { get; private set; }
- /// <summary>
- /// Will message
- /// </summary>
- public string WillMessage { get; private set; }
- /// <summary>
- /// MQTT protocol version
- /// </summary>
- public MqttProtocolVersion ProtocolVersion { get; set; }
- #if BROKER
- /// <summary>
- /// MQTT Client Session
- /// </summary>
- public MqttClientSession Session
- {
- get { return this.session; }
- set { this.session = value; }
- }
- #endif
- /// <summary>
- /// MQTT client settings
- /// </summary>
- public MqttSettings Settings
- {
- get { return this.settings; }
- }
- #if !(WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP))
- /// <summary>
- /// Constructor
- /// </summary>
- /// <param name="brokerIpAddress">Broker IP address</param>
- [Obsolete("Use this ctor MqttClient(string brokerHostName) insted")]
- public MqttClient(IPAddress brokerIpAddress) :
- this(brokerIpAddress, MqttSettings.MQTT_BROKER_DEFAULT_PORT, false, null, null, MqttSslProtocols.None)
- {
- }
- /// <summary>
- /// Constructor
- /// </summary>
- /// <param name="brokerIpAddress">Broker IP address</param>
- /// <param name="brokerPort">Broker port</param>
- /// <param name="secure">Using secure connection</param>
- /// <param name="caCert">CA certificate for secure connection</param>
- /// <param name="clientCert">Client certificate</param>
- /// <param name="sslProtocol">SSL/TLS protocol version</param>
- [Obsolete("Use this ctor MqttClient(string brokerHostName, int brokerPort, bool secure, X509Certificate caCert) insted")]
- public MqttClient(IPAddress brokerIpAddress, int brokerPort, bool secure, X509Certificate caCert, X509Certificate clientCert, MqttSslProtocols sslProtocol)
- {
- #if !(MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK)
- this.Init(brokerIpAddress.ToString(), brokerPort, secure, caCert, clientCert, sslProtocol, null, null);
- #else
- this.Init(brokerIpAddress.ToString(), brokerPort, secure, caCert, clientCert, sslProtocol);
- #endif
- }
- #endif
- /// <summary>
- /// Constructor
- /// </summary>
- /// <param name="brokerHostName">Broker Host Name or IP Address</param>
- public MqttClient(string brokerHostName) :
- #if !(WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP))
- this(brokerHostName, MqttSettings.MQTT_BROKER_DEFAULT_PORT, false, null, null, MqttSslProtocols.None)
- #else
- this(brokerHostName, MqttSettings.MQTT_BROKER_DEFAULT_PORT, false, MqttSslProtocols.None)
- #endif
- {
- }
- /// <summary>
- /// Constructor
- /// </summary>
- /// <param name="brokerHostName">Broker Host Name or IP Address</param>
- /// <param name="brokerPort">Broker port</param>
- /// <param name="secure">使用安全连接</param>
- /// <param name="sslProtocol">SSL/TLS protocol version</param>
- #if !(WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR && UNITY_WSA_10_0 && !ENABLE_IL2CPP))
- /// <param name="caCert">CA certificate for secure connection</param>
- /// <param name="clientCert">Client certificate</param>
- public MqttClient(string brokerHostName, int brokerPort, bool secure, X509Certificate caCert, X509Certificate clientCert, MqttSslProtocols sslProtocol)
- #else
- public MqttClient(string brokerHostName, int brokerPort, bool secure, MqttSslProtocols sslProtocol)
- #endif
- {
- #if !(MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK || WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP))
- this.Init(brokerHostName, brokerPort, secure, caCert, clientCert, sslProtocol, null, null);
- #elif (WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP))
- this.Init(brokerHostName, brokerPort, secure, sslProtocol);
- #else
- this.Init(brokerHostName, brokerPort, secure, caCert, clientCert, sslProtocol);
- #endif
- }
- #if !(MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK || WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP))
- /// <summary>
- /// Constructor
- /// </summary>
- /// <param name="brokerHostName">Broker Host Name or IP Address</param>
- /// <param name="brokerPort">Broker port</param>
- /// <param name="secure">Using secure connection</param>
- /// <param name="caCert">CA certificate for secure connection</param>
- /// <param name="clientCert">Client certificate</param>
- /// <param name="sslProtocol">SSL/TLS protocol version</param>
- /// <param name="userCertificateValidationCallback">A RemoteCertificateValidationCallback delegate responsible for validating the certificate supplied by the remote party</param>
- public MqttClient(string brokerHostName, int brokerPort, bool secure, X509Certificate caCert, X509Certificate clientCert, MqttSslProtocols sslProtocol,
- RemoteCertificateValidationCallback userCertificateValidationCallback)
- : this(brokerHostName, brokerPort, secure, caCert, clientCert, sslProtocol, userCertificateValidationCallback, null)
- {
- }
- /// <summary>
- /// Constructor
- /// </summary>
- /// <param name="brokerHostName">Broker Host Name or IP Address</param>
- /// <param name="brokerPort">Broker port</param>
- /// <param name="secure">Using secure connection</param>
- /// <param name="sslProtocol">SSL/TLS protocol version</param>
- /// <param name="userCertificateValidationCallback">A RemoteCertificateValidationCallback delegate responsible for validating the certificate supplied by the remote party</param>
- /// <param name="userCertificateSelectionCallback">A LocalCertificateSelectionCallback delegate responsible for selecting the certificate used for authentication</param>
- public MqttClient(string brokerHostName, int brokerPort, bool secure, MqttSslProtocols sslProtocol,
- RemoteCertificateValidationCallback userCertificateValidationCallback,
- LocalCertificateSelectionCallback userCertificateSelectionCallback)
- : this(brokerHostName, brokerPort, secure, null, null, sslProtocol, userCertificateValidationCallback, userCertificateSelectionCallback)
- {
- }
- /// <summary>
- /// Constructor
- /// </summary>
- /// <param name="brokerHostName">Broker Host Name or IP Address</param>
- /// <param name="brokerPort">Broker port</param>
- /// <param name="secure">Using secure connection</param>
- /// <param name="caCert">CA certificate for secure connection</param>
- /// <param name="clientCert">Client certificate</param>
- /// <param name="sslProtocol">SSL/TLS protocol version</param>
- /// <param name="userCertificateValidationCallback">A RemoteCertificateValidationCallback delegate responsible for validating the certificate supplied by the remote party</param>
- /// <param name="userCertificateSelectionCallback">A LocalCertificateSelectionCallback delegate responsible for selecting the certificate used for authentication</param>
- public MqttClient(string brokerHostName, int brokerPort, bool secure, X509Certificate caCert, X509Certificate clientCert, MqttSslProtocols sslProtocol,
- RemoteCertificateValidationCallback userCertificateValidationCallback,
- LocalCertificateSelectionCallback userCertificateSelectionCallback)
- {
- this.Init(brokerHostName, brokerPort, secure, caCert, clientCert, sslProtocol, userCertificateValidationCallback, userCertificateSelectionCallback);
- }
- #endif
- #if BROKER
- /// <summary>
- /// Constructor
- /// </summary>
- /// <param name="channel">Network channel for communication</param>
- public MqttClient(IMqttNetworkChannel channel)
- {
- // set default MQTT protocol version (default is 3.1.1)
- this.ProtocolVersion = MqttProtocolVersion.Version_3_1_1;
- this.channel = channel;
- // reference to MQTT settings
- this.settings = MqttSettings.Instance;
- // client not connected yet (CONNACK not send from client), some default values
- this.IsConnected = false;
- this.ClientId = null;
- this.CleanSession = true;
- this.keepAliveEvent = new AutoResetEvent(false);
- // queue for handling inflight messages (publishing and acknowledge)
- this.inflightWaitHandle = new AutoResetEvent(false);
- this.inflightQueue = new Queue();
- // queue for received message
- this.receiveEventWaitHandle = new AutoResetEvent(false);
- this.eventQueue = new Queue();
- this.internalQueue = new Queue();
- // session
- this.session = null;
- }
- #endif
- /// <summary>
- /// MqttClient initialization
- /// </summary>
- /// <param name="brokerHostName">Broker Host Name or IP Address</param>
- /// <param name="brokerPort">Broker port</param>
- /// <param name="secure">>Using secure connection</param>
- /// <param name="caCert">CA certificate for secure connection</param>
- /// <param name="clientCert">Client certificate</param>
- /// <param name="sslProtocol">SSL/TLS protocol version</param>
- #if !(MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK || WINDOWS_APP || WINDOWS_PHONE_APP || ((!UNITY_EDITOR && UNITY_WSA_10_0 && !ENABLE_IL2CPP)))
- /// <param name="userCertificateSelectionCallback">A RemoteCertificateValidationCallback delegate responsible for validating the certificate supplied by the remote party</param>
- /// <param name="userCertificateValidationCallback">A LocalCertificateSelectionCallback delegate responsible for selecting the certificate used for authentication</param>
- private void Init(string brokerHostName, int brokerPort, bool secure, X509Certificate caCert, X509Certificate clientCert, MqttSslProtocols sslProtocol,
- RemoteCertificateValidationCallback userCertificateValidationCallback,
- LocalCertificateSelectionCallback userCertificateSelectionCallback)
- #elif (WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP))
- private void Init(string brokerHostName, int brokerPort, bool secure, MqttSslProtocols sslProtocol)
- #else
- private void Init(string brokerHostName, int brokerPort, bool secure, X509Certificate caCert, X509Certificate clientCert, MqttSslProtocols sslProtocol)
- #endif
- {
- // set default MQTT protocol version (default is 3.1.1)
- this.ProtocolVersion = MqttProtocolVersion.Version_3_1_1;
- #if !SSL
- // check security parameters
- if (secure)
- throw new ArgumentException("Library compiled without SSL support");
- #endif
- this.brokerHostName = brokerHostName;
- this.brokerPort = brokerPort;
- // reference to MQTT settings
- this.settings = MqttSettings.Instance;
- // set settings port based on secure connection or not
- if (!secure)
- this.settings.Port = this.brokerPort;
- else
- this.settings.SslPort = this.brokerPort;
- this.syncEndReceiving = new AutoResetEvent(false);
- this.keepAliveEvent = new AutoResetEvent(false);
- // queue for handling inflight messages (publishing and acknowledge)
- this.inflightWaitHandle = new AutoResetEvent(false);
- this.inflightQueue = new Queue();
- // queue for received message
- this.receiveEventWaitHandle = new AutoResetEvent(false);
- this.eventQueue = new Queue();
- this.internalQueue = new Queue();
- // session
- this.session = null;
- // create network channel
- #if !(MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK || WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP))
- this.channel = new MqttNetworkChannel(this.brokerHostName, this.brokerPort, secure, caCert, clientCert, sslProtocol, userCertificateValidationCallback, userCertificateSelectionCallback);
- #elif (WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP))
- this.channel = new MqttNetworkChannel(this.brokerHostName, this.brokerPort, secure, sslProtocol);
- #else
- this.channel = new MqttNetworkChannel(this.brokerHostName, this.brokerPort, secure, caCert, clientCert, sslProtocol);
- #endif
- }
- /// <summary>
- /// Connect to broker
- /// </summary>
- /// <param name="clientId">Client identifier</param>
- /// <returns>Return code of CONNACK message from broker</returns>
- public byte Connect(string clientId)
- {
- return this.Connect(clientId, null, null, false, MqttMsgConnect.QOS_LEVEL_AT_MOST_ONCE, false, null, null, true, MqttMsgConnect.KEEP_ALIVE_PERIOD_DEFAULT);
- }
- /// <summary>
- /// Connect to broker
- /// </summary>
- /// <param name="clientId">Client identifier</param>
- /// <param name="username">Username</param>
- /// <param name="password">Password</param>
- /// <returns>Return code of CONNACK message from broker</returns>
- public byte Connect(string clientId,
- string username,
- string password)
- {
- return this.Connect(clientId, username, password, false, MqttMsgConnect.QOS_LEVEL_AT_MOST_ONCE, false, null, null, true, MqttMsgConnect.KEEP_ALIVE_PERIOD_DEFAULT);
- }
- /// <summary>
- /// Connect to broker
- /// </summary>
- /// <param name="clientId">Client identifier</param>
- /// <param name="username">Username</param>
- /// <param name="password">Password</param>
- /// <param name="cleanSession">Clean sessione flag</param>
- /// <param name="keepAlivePeriod">Keep alive period</param>
- /// <returns>Return code of CONNACK message from broker</returns>
- public byte Connect(string clientId,
- string username,
- string password,
- bool cleanSession,
- ushort keepAlivePeriod)
- {
- return this.Connect(clientId, username, password, false, MqttMsgConnect.QOS_LEVEL_AT_MOST_ONCE, false, null, null, cleanSession, keepAlivePeriod);
- }
- /// <summary>
- /// Connect to broker
- /// </summary>
- /// <param name="clientId">Client identifier</param>
- /// <param name="username">Username</param>
- /// <param name="password">Password</param>
- /// <param name="willRetain">Will retain flag</param>
- /// <param name="willQosLevel">Will QOS level</param>
- /// <param name="willFlag">Will flag</param>
- /// <param name="willTopic">Will topic</param>
- /// <param name="willMessage">Will message</param>
- /// <param name="cleanSession">Clean sessione flag</param>
- /// <param name="keepAlivePeriod">Keep alive period</param>
- /// <returns>Return code of CONNACK message from broker</returns>
- public byte Connect(string clientId,
- string username,
- string password,
- bool willRetain,
- byte willQosLevel,
- bool willFlag,
- string willTopic,
- string willMessage,
- bool cleanSession,
- ushort keepAlivePeriod)
- {
- // create CONNECT message
- MqttMsgConnect connect = new MqttMsgConnect(clientId,
- username,
- password,
- willRetain,
- willQosLevel,
- willFlag,
- willTopic,
- willMessage,
- cleanSession,
- keepAlivePeriod,
- (byte)this.ProtocolVersion);
- try
- {
- // connect to the broker
- this.channel.Connect();
- }
- catch (Exception ex)
- {
- throw new MqttConnectionException("Exception connecting to the broker", ex);
- }
- this.lastCommTime = 0;
- this.isRunning = true;
- this.isConnectionClosing = false;
- // start thread for receiving messages from broker
- Fx.StartThread(this.ReceiveThread);
-
- MqttMsgConnack connack = (MqttMsgConnack)this.SendReceive(connect);
- // if connection accepted, start keep alive timer and
- if (connack.ReturnCode == MqttMsgConnack.CONN_ACCEPTED)
- {
- // set all client properties
- this.ClientId = clientId;
- this.CleanSession = cleanSession;
- this.WillFlag = willFlag;
- this.WillTopic = willTopic;
- this.WillMessage = willMessage;
- this.WillQosLevel = willQosLevel;
- this.keepAlivePeriod = keepAlivePeriod * 1000; // convert in ms
- // restore previous session
- this.RestoreSession();
- // keep alive period equals zero means turning off keep alive mechanism
- if (this.keepAlivePeriod != 0)
- {
- // start thread for sending keep alive message to the broker
- Fx.StartThread(this.KeepAliveThread);
- }
- // start thread for raising received message event from broker
- Fx.StartThread(this.DispatchEventThread);
-
- // start thread for handling inflight messages queue to broker asynchronously (publish and acknowledge)
- Fx.StartThread(this.ProcessInflightThread);
- this.IsConnected = true;
- }
- return connack.ReturnCode;
- }
- /// <summary>
- /// Disconnect from broker
- /// </summary>
- public void Disconnect()
- {
- MqttMsgDisconnect disconnect = new MqttMsgDisconnect();
- this.Send(disconnect);
- // close client
- this.OnConnectionClosing();
- }
- #if BROKER
- /// <summary>
- /// Open client communication
- /// </summary>
- public void Open()
- {
- this.isRunning = true;
- // start thread for receiving messages from client
- Fx.StartThread(this.ReceiveThread);
- // start thread for raising received message event from client
- Fx.StartThread(this.DispatchEventThread);
- // start thread for handling inflight messages queue to client asynchronously (publish and acknowledge)
- Fx.StartThread(this.ProcessInflightThread);
- }
- #endif
- /// <summary>
- /// Close client
- /// </summary>
- #if BROKER
- public void Close()
- #else
- private void Close()
- #endif
- {
- // stop receiving thread
- this.isRunning = false;
- // wait end receive event thread
- if (this.receiveEventWaitHandle != null)
- this.receiveEventWaitHandle.Set();
- // wait end process inflight thread
- if (this.inflightWaitHandle != null)
- this.inflightWaitHandle.Set();
- #if BROKER
- // unlock keep alive thread
- this.keepAliveEvent.Set();
- #else
- // unlock keep alive thread and wait
- this.keepAliveEvent.Set();
- if (this.keepAliveEventEnd != null)
- this.keepAliveEventEnd.WaitOne();
- #endif
- // clear all queues
- this.inflightQueue.Clear();
- this.internalQueue.Clear();
- this.eventQueue.Clear();
- // close network channel
- this.channel.Close();
- this.IsConnected = false;
- }
- /// <summary>
- /// Execute ping to broker for keep alive
- /// </summary>
- /// <returns>PINGRESP message from broker</returns>
- private MqttMsgPingResp Ping()
- {
- MqttMsgPingReq pingreq = new MqttMsgPingReq();
- try
- {
- // broker must send PINGRESP within timeout equal to keep alive period
- return (MqttMsgPingResp)this.SendReceive(pingreq, this.keepAlivePeriod);
- }
- catch (Exception e)
- {
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Error, "Exception occurred: {0}", e.ToString());
- #endif
- // client must close connection
- this.OnConnectionClosing();
- return null;
- }
- }
- #if BROKER
- /// <summary>
- /// Send CONNACK message to the client (connection accepted or not)
- /// </summary>
- /// <param name="connect">CONNECT message with all client information</param>
- /// <param name="returnCode">Return code for CONNACK message</param>
- /// <param name="clientId">If not null, client id assigned by broker</param>
- /// <param name="sessionPresent">Session present on the broker</param>
- public void Connack(MqttMsgConnect connect, byte returnCode, string clientId, bool sessionPresent)
- {
- this.lastCommTime = 0;
- // create CONNACK message and ...
- MqttMsgConnack connack = new MqttMsgConnack();
- connack.ReturnCode = returnCode;
- // [v3.1.1] session present flag
- if (this.ProtocolVersion == MqttProtocolVersion.Version_3_1_1)
- connack.SessionPresent = sessionPresent;
- // ... send it to the client
- this.Send(connack);
- // connection accepted, start keep alive thread checking
- if (connack.ReturnCode == MqttMsgConnack.CONN_ACCEPTED)
- {
- // [v3.1.1] if client id isn't null, the CONNECT message has a cliend id with zero bytes length
- // and broker assigned a unique identifier to the client
- this.ClientId = (clientId == null) ? connect.ClientId : clientId;
- this.CleanSession = connect.CleanSession;
- this.WillFlag = connect.WillFlag;
- this.WillTopic = connect.WillTopic;
- this.WillMessage = connect.WillMessage;
- this.WillQosLevel = connect.WillQosLevel;
- this.keepAlivePeriod = connect.KeepAlivePeriod * 1000; // convert in ms
- // broker has a tolerance of 1.5 specified keep alive period
- this.keepAlivePeriod += (this.keepAlivePeriod / 2);
- // start thread for checking keep alive period timeout
- Fx.StartThread(this.KeepAliveThread);
- this.isConnectionClosing = false;
- this.IsConnected = true;
- }
- // connection refused, close TCP/IP channel
- else
- {
- this.Close();
- }
- }
- /// <summary>
- /// Send SUBACK message to the client
- /// </summary>
- /// <param name="messageId">Message Id for the SUBSCRIBE message that is being acknowledged</param>
- /// <param name="grantedQosLevels">Granted QoS Levels</param>
- public void Suback(ushort messageId, byte[] grantedQosLevels)
- {
- MqttMsgSuback suback = new MqttMsgSuback();
- suback.MessageId = messageId;
- suback.GrantedQoSLevels = grantedQosLevels;
- this.Send(suback);
- }
- /// <summary>
- /// Send UNSUBACK message to the client
- /// </summary>
- /// <param name="messageId">Message Id for the UNSUBSCRIBE message that is being acknowledged</param>
- public void Unsuback(ushort messageId)
- {
- MqttMsgUnsuback unsuback = new MqttMsgUnsuback();
- unsuback.MessageId = messageId;
- this.Send(unsuback);
- }
- #endif
- /// <summary>
- /// 订阅消息主题
- /// </summary>
- /// <param name="topics">要监听的主题数组</param>
- /// <param name="qosLevels">主题相关的 QOS levels</param>
- /// <returns>监听消息相关的消息ID</returns>
- public ushort Subscribe(string[] topics, byte[] qosLevels)
- {
- MqttMsgSubscribe subscribe = new MqttMsgSubscribe(topics, qosLevels);
- subscribe.MessageId = this.GetMessageId();
- // 将订阅请求排入机上队列
- this.EnqueueInflight(subscribe, MqttMsgFlow.ToPublish);
- //UnityEngine.Debug.LogError("订阅消息主题"+subscribe);
- return subscribe.MessageId;
- }
- /// <summary>
- /// Unsubscribe for message topics
- /// </summary>
- /// <param name="topics">List of topics to unsubscribe</param>
- /// <returns>Message Id in UNSUBACK message from broker</returns>
- public ushort Unsubscribe(string[] topics)
- {
- MqttMsgUnsubscribe unsubscribe =
- new MqttMsgUnsubscribe(topics);
- unsubscribe.MessageId = this.GetMessageId();
- // enqueue unsubscribe request into the inflight queue
- this.EnqueueInflight(unsubscribe, MqttMsgFlow.ToPublish);
- return unsubscribe.MessageId;
- }
- /// <summary>
- /// 异步发布消息(QoS级别0且未保留)
- /// Publish a message asynchronously (QoS Level 0 and not retained)
- /// </summary>
- /// <param name="topic">Message topic</param>
- /// <param name="message">Message data (payload)</param>
- /// <returns>Message Id related to PUBLISH message</returns>
- public ushort Publish(string topic, byte[] message)
- {
- return this.Publish(topic, message, MqttMsgBase.QOS_LEVEL_AT_MOST_ONCE, false);
- }
- /// <summary>
- /// 异步发布消息
- /// </summary>
- /// <param name="topic">Message topic</param>
- /// <param name="message">Message data (payload)</param>
- /// <param name="qosLevel">QoS Level</param>
- /// <param name="retain">Retain flag</param>
- /// <returns>Message Id related to PUBLISH message</returns>
- public ushort Publish(string topic, byte[] message, byte qosLevel, bool retain)
- {
- MqttMsgPublish publish =
- new MqttMsgPublish(topic, message, false, qosLevel, retain);
- publish.MessageId = this.GetMessageId();
- // enqueue message to publish into the inflight queue
- bool enqueue = this.EnqueueInflight(publish, MqttMsgFlow.ToPublish);
- // message enqueued
- if (enqueue)
- return publish.MessageId;
- // infligh queue full, message not enqueued
- else
- throw new MqttClientException(MqttClientErrorCode.InflightQueueFull);
- }
- /// <summary>
- /// Wrapper method for raising events
- /// </summary>
- /// <param name="internalEvent">Internal event</param>
- private void OnInternalEvent(InternalEvent internalEvent)
- {
- lock (this.eventQueue)
- {
- this.eventQueue.Enqueue(internalEvent);
- }
- this.receiveEventWaitHandle.Set();
- }
- /// <summary>
- /// Wrapper method for raising closing connection event
- /// </summary>
- private void OnConnectionClosing()
- {
- if (!this.isConnectionClosing)
- {
- this.isConnectionClosing = true;
- this.receiveEventWaitHandle.Set();
- }
- }
- /// <summary>
- /// Wrapper method for raising PUBLISH message received event
- /// </summary>
- /// <param name="publish">PUBLISH message received</param>
- private void OnMqttMsgPublishReceived(MqttMsgPublish publish)
- {
- if (this.MqttMsgPublishReceived != null)
- {
- this.MqttMsgPublishReceived(this,
- new MqttMsgPublishEventArgs(publish.Topic, publish.Message, publish.DupFlag, publish.QosLevel, publish.Retain));
- }
- }
- /// <summary>
- /// Wrapper method for raising published message event
- /// </summary>
- /// <param name="messageId">Message identifier for published message</param>
- /// <param name="isPublished">Publish flag</param>
- private void OnMqttMsgPublished(ushort messageId, bool isPublished)
- {
- if (this.MqttMsgPublished != null)
- {
- this.MqttMsgPublished(this,
- new MqttMsgPublishedEventArgs(messageId, isPublished));
- }
- }
- /// <summary>
- /// Wrapper method for raising subscribed topic event
- /// </summary>
- /// <param name="suback">SUBACK message received</param>
- private void OnMqttMsgSubscribed(MqttMsgSuback suback)
- {
- if (this.MqttMsgSubscribed != null)
- {
- this.MqttMsgSubscribed(this,
- new MqttMsgSubscribedEventArgs(suback.MessageId, suback.GrantedQoSLevels));
- }
- }
- /// <summary>
- /// Wrapper method for raising unsubscribed topic event
- /// </summary>
- /// <param name="messageId">Message identifier for unsubscribed topic</param>
- private void OnMqttMsgUnsubscribed(ushort messageId)
- {
- if (this.MqttMsgUnsubscribed != null)
- {
- this.MqttMsgUnsubscribed(this,
- new MqttMsgUnsubscribedEventArgs(messageId));
- }
- }
- #if BROKER
- /// <summary>
- /// Wrapper method for raising SUBSCRIBE message event
- /// </summary>
- /// <param name="messageId">Message identifier for subscribe topics request</param>
- /// <param name="topics">Topics requested to subscribe</param>
- /// <param name="qosLevels">List of QOS Levels requested</param>
- private void OnMqttMsgSubscribeReceived(ushort messageId, string[] topics, byte[] qosLevels)
- {
- if (this.MqttMsgSubscribeReceived != null)
- {
- this.MqttMsgSubscribeReceived(this,
- new MqttMsgSubscribeEventArgs(messageId, topics, qosLevels));
- }
- }
- /// <summary>
- /// Wrapper method for raising UNSUBSCRIBE message event
- /// </summary>
- /// <param name="messageId">Message identifier for unsubscribe topics request</param>
- /// <param name="topics">Topics requested to unsubscribe</param>
- private void OnMqttMsgUnsubscribeReceived(ushort messageId, string[] topics)
- {
- if (this.MqttMsgUnsubscribeReceived != null)
- {
- this.MqttMsgUnsubscribeReceived(this,
- new MqttMsgUnsubscribeEventArgs(messageId, topics));
- }
- }
- /// <summary>
- /// Wrapper method for raising CONNECT message event
- /// </summary>
- private void OnMqttMsgConnected(MqttMsgConnect connect)
- {
- if (this.MqttMsgConnected != null)
- {
- this.ProtocolVersion = (MqttProtocolVersion)connect.ProtocolVersion;
- this.MqttMsgConnected(this, new MqttMsgConnectEventArgs(connect));
- }
- }
- /// <summary>
- /// Wrapper method for raising DISCONNECT message event
- /// </summary>
- private void OnMqttMsgDisconnected()
- {
- if (this.MqttMsgDisconnected != null)
- {
- this.MqttMsgDisconnected(this, EventArgs.Empty);
- }
- }
- #endif
- /// <summary>
- /// Wrapper method for peer/client disconnection
- /// </summary>
- private void OnConnectionClosed()
- {
- if (this.ConnectionClosed != null)
- {
- this.ConnectionClosed(this, EventArgs.Empty);
- }
- }
- /// <summary>
- /// Send a message
- /// </summary>
- /// <param name="msgBytes">Message bytes</param>
- private void Send(byte[] msgBytes)
- {
- try
- {
- // send message
- this.channel.Send(msgBytes);
- #if !BROKER
- // update last message sent ticks
- this.lastCommTime = Environment.TickCount;
- #endif
- }
- catch (Exception e)
- {
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Error, "Exception occurred: {0}", e.ToString());
- #endif
- throw new MqttCommunicationException(e);
- }
- }
- /// <summary>
- /// Send a message
- /// </summary>
- /// <param name="msg">Message</param>
- private void Send(MqttMsgBase msg)
- {
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Frame, "SEND {0}", msg);
- #endif
- this.Send(msg.GetBytes((byte)this.ProtocolVersion));
- }
- /// <summary>
- /// Send a message to the broker and wait answer
- /// </summary>
- /// <param name="msgBytes">Message bytes</param>
- /// <returns>MQTT message response</returns>
- private MqttMsgBase SendReceive(byte[] msgBytes)
- {
- return this.SendReceive(msgBytes, MqttSettings.MQTT_DEFAULT_TIMEOUT);
- }
- /// <summary>
- /// Send a message to the broker and wait answer
- /// </summary>
- /// <param name="msgBytes">Message bytes</param>
- /// <param name="timeout">Timeout for receiving answer</param>
- /// <returns>MQTT message response</returns>
- private MqttMsgBase SendReceive(byte[] msgBytes, int timeout)
- {
- // reset handle before sending
- this.syncEndReceiving.Reset();
- try
- {
- // send message
- this.channel.Send(msgBytes);
- // update last message sent ticks
- this.lastCommTime = Environment.TickCount;
- }
- catch (Exception e)
- {
- #if !(MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK || WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP))
- if (typeof(SocketException) == e.GetType())
- {
- // connection reset by broker
- if (((SocketException)e).SocketErrorCode == SocketError.ConnectionReset)
- this.IsConnected = false;
- }
- #endif
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Error, "Exception occurred: {0}", e.ToString());
- #endif
- throw new MqttCommunicationException(e);
- }
- #if (MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK)
- // wait for answer from broker
- if (this.syncEndReceiving.WaitOne(timeout, false))
- #else
- // wait for answer from broker
- if (this.syncEndReceiving.WaitOne(timeout))
- #endif
- {
- // message received without exception
- if (this.exReceiving == null)
- return this.msgReceived;
- // receiving thread catched exception
- else
- throw this.exReceiving;
- }
- else
- {
- // throw timeout exception
- throw new MqttCommunicationException();
- }
- }
- /// <summary>
- /// Send a message to the broker and wait answer
- /// </summary>
- /// <param name="msg">Message</param>
- /// <returns>MQTT message response</returns>
- private MqttMsgBase SendReceive(MqttMsgBase msg)
- {
- return this.SendReceive(msg, MqttSettings.MQTT_DEFAULT_TIMEOUT);
- }
- /// <summary>
- /// Send a message to the broker and wait answer
- /// </summary>
- /// <param name="msg">Message</param>
- /// <param name="timeout">Timeout for receiving answer</param>
- /// <returns>MQTT message response</returns>
- private MqttMsgBase SendReceive(MqttMsgBase msg, int timeout)
- {
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Frame, "SEND {0}", msg);
- #endif
- return this.SendReceive(msg.GetBytes((byte)this.ProtocolVersion), timeout);
- }
- /// <summary>
- /// Enqueue a message into the inflight queue
- /// </summary>
- /// <param name="msg">Message to enqueue</param>
- /// <param name="flow">Message flow (publish, acknowledge)</param>
- /// <returns>Message enqueued or not</returns>
- private bool EnqueueInflight(MqttMsgBase msg, MqttMsgFlow flow)
- {
- // enqueue is needed (or not)
- bool enqueue = true;
- // if it is a PUBLISH message with QoS Level 2
- if ((msg.Type == MqttMsgBase.MQTT_MSG_PUBLISH_TYPE) &&
- (msg.QosLevel == MqttMsgBase.QOS_LEVEL_EXACTLY_ONCE))
- {
- lock (this.inflightQueue)
- {
- // if it is a PUBLISH message already received (it is in the inflight queue), the publisher
- // re-sent it because it didn't received the PUBREC. In this case, we have to re-send PUBREC
- // NOTE : I need to find on message id and flow because the broker could be publish/received
- // to/from client and message id could be the same (one tracked by broker and the other by client)
- MqttMsgContextFinder msgCtxFinder = new MqttMsgContextFinder(msg.MessageId, MqttMsgFlow.ToAcknowledge);
- MqttMsgContext msgCtx = (MqttMsgContext)this.inflightQueue.Get(msgCtxFinder.Find);
- // the PUBLISH message is alredy in the inflight queue, we don't need to re-enqueue but we need
- // to change state to re-send PUBREC
- if (msgCtx != null)
- {
- msgCtx.State = MqttMsgState.QueuedQos2;
- msgCtx.Flow = MqttMsgFlow.ToAcknowledge;
- enqueue = false;
- }
- }
- }
- if (enqueue)
- {
- // set a default state
- MqttMsgState state = MqttMsgState.QueuedQos0;
- // based on QoS level, the messages flow between broker and client changes
- switch (msg.QosLevel)
- {
- // QoS Level 0
- case MqttMsgBase.QOS_LEVEL_AT_MOST_ONCE:
- state = MqttMsgState.QueuedQos0;
- break;
- // QoS Level 1
- case MqttMsgBase.QOS_LEVEL_AT_LEAST_ONCE:
- state = MqttMsgState.QueuedQos1;
- break;
- // QoS Level 2
- case MqttMsgBase.QOS_LEVEL_EXACTLY_ONCE:
- state = MqttMsgState.QueuedQos2;
- break;
- }
- // [v3.1.1] SUBSCRIBE and UNSUBSCRIBE aren't "officially" QOS = 1
- // so QueuedQos1 state isn't valid for them
- if (msg.Type == MqttMsgBase.MQTT_MSG_SUBSCRIBE_TYPE)
- state = MqttMsgState.SendSubscribe;
- else if (msg.Type == MqttMsgBase.MQTT_MSG_UNSUBSCRIBE_TYPE)
- state = MqttMsgState.SendUnsubscribe;
- // queue message context
- MqttMsgContext msgContext = new MqttMsgContext()
- {
- Message = msg,
- State = state,
- Flow = flow,
- Attempt = 0
- };
- lock (this.inflightQueue)
- {
- // check number of messages inside inflight queue
- enqueue = (this.inflightQueue.Count < this.settings.InflightQueueSize);
- if (enqueue)
- {
- // enqueue message and unlock send thread
- this.inflightQueue.Enqueue(msgContext);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "enqueued {0}", msg);
- #endif
- // PUBLISH message
- if (msg.Type == MqttMsgBase.MQTT_MSG_PUBLISH_TYPE)
- {
- // to publish and QoS level 1 or 2
- if ((msgContext.Flow == MqttMsgFlow.ToPublish) &&
- ((msg.QosLevel == MqttMsgBase.QOS_LEVEL_AT_LEAST_ONCE) ||
- (msg.QosLevel == MqttMsgBase.QOS_LEVEL_EXACTLY_ONCE)))
- {
- if (this.session != null)
- this.session.InflightMessages.Add(msgContext.Key, msgContext);
- }
- // to acknowledge and QoS level 2
- else if ((msgContext.Flow == MqttMsgFlow.ToAcknowledge) &&
- (msg.QosLevel == MqttMsgBase.QOS_LEVEL_EXACTLY_ONCE))
- {
- if (this.session != null)
- this.session.InflightMessages.Add(msgContext.Key, msgContext);
- }
- }
- }
- }
- }
- this.inflightWaitHandle.Set();
- return enqueue;
- }
- /// <summary>
- /// Enqueue a message into the internal queue
- /// </summary>
- /// <param name="msg">Message to enqueue</param>
- private void EnqueueInternal(MqttMsgBase msg)
- {
- // enqueue is needed (or not)
- bool enqueue = true;
- // if it is a PUBREL message (for QoS Level 2)
- if (msg.Type == MqttMsgBase.MQTT_MSG_PUBREL_TYPE)
- {
- lock (this.inflightQueue)
- {
- // if it is a PUBREL but the corresponding PUBLISH isn't in the inflight queue,
- // it means that we processed PUBLISH message and received PUBREL and we sent PUBCOMP
- // but publisher didn't receive PUBCOMP so it re-sent PUBREL. We need only to re-send PUBCOMP.
- // NOTE : I need to find on message id and flow because the broker could be publish/received
- // to/from client and message id could be the same (one tracked by broker and the other by client)
- MqttMsgContextFinder msgCtxFinder = new MqttMsgContextFinder(msg.MessageId, MqttMsgFlow.ToAcknowledge);
- MqttMsgContext msgCtx = (MqttMsgContext)this.inflightQueue.Get(msgCtxFinder.Find);
- // the PUBLISH message isn't in the inflight queue, it was already processed so
- // we need to re-send PUBCOMP only
- if (msgCtx == null)
- {
- MqttMsgPubcomp pubcomp = new MqttMsgPubcomp();
- pubcomp.MessageId = msg.MessageId;
- this.Send(pubcomp);
- enqueue = false;
- }
- }
- }
- // if it is a PUBCOMP message (for QoS Level 2)
- else if (msg.Type == MqttMsgBase.MQTT_MSG_PUBCOMP_TYPE)
- {
- lock (this.inflightQueue)
- {
- // if it is a PUBCOMP but the corresponding PUBLISH isn't in the inflight queue,
- // it means that we sent PUBLISH message, sent PUBREL (after receiving PUBREC) and already received PUBCOMP
- // but publisher didn't receive PUBREL so it re-sent PUBCOMP. We need only to ignore this PUBCOMP.
- // NOTE : I need to find on message id and flow because the broker could be publish/received
- // to/from client and message id could be the same (one tracked by broker and the other by client)
- MqttMsgContextFinder msgCtxFinder = new MqttMsgContextFinder(msg.MessageId, MqttMsgFlow.ToPublish);
- MqttMsgContext msgCtx = (MqttMsgContext)this.inflightQueue.Get(msgCtxFinder.Find);
- // the PUBLISH message isn't in the inflight queue, it was already sent so we need to ignore this PUBCOMP
- if (msgCtx == null)
- {
- enqueue = false;
- }
- }
- }
- // if it is a PUBREC message (for QoS Level 2)
- else if (msg.Type == MqttMsgBase.MQTT_MSG_PUBREC_TYPE)
- {
- lock (this.inflightQueue)
- {
- // if it is a PUBREC but the corresponding PUBLISH isn't in the inflight queue,
- // it means that we sent PUBLISH message more times (retries) but broker didn't send PUBREC in time
- // the publish is failed and we need only to ignore this PUBREC.
- // NOTE : I need to find on message id and flow because the broker could be publish/received
- // to/from client and message id could be the same (one tracked by broker and the other by client)
- MqttMsgContextFinder msgCtxFinder = new MqttMsgContextFinder(msg.MessageId, MqttMsgFlow.ToPublish);
- MqttMsgContext msgCtx = (MqttMsgContext)this.inflightQueue.Get(msgCtxFinder.Find);
- // the PUBLISH message isn't in the inflight queue, it was already sent so we need to ignore this PUBREC
- if (msgCtx == null)
- {
- enqueue = false;
- }
- }
- }
- if (enqueue)
- {
- lock (this.internalQueue)
- {
- this.internalQueue.Enqueue(msg);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "enqueued {0}", msg);
- #endif
- this.inflightWaitHandle.Set();
- }
- }
- }
- /// <summary>
- /// Thread for receiving messages
- /// </summary>
- private void ReceiveThread()
- {
- int readBytes = 0;
- byte[] fixedHeaderFirstByte = new byte[1];
- byte msgType;
- while (this.isRunning)
- {
- try
- {
- // read first byte (fixed header)
- readBytes = this.channel.Receive(fixedHeaderFirstByte);
- if (readBytes > 0)
- {
- #if BROKER
- // update last message received ticks
- this.lastCommTime = Environment.TickCount;
- #endif
- // extract message type from received byte
- msgType = (byte)((fixedHeaderFirstByte[0] & MqttMsgBase.MSG_TYPE_MASK) >> MqttMsgBase.MSG_TYPE_OFFSET);
- switch (msgType)
- {
- // CONNECT message received
- case MqttMsgBase.MQTT_MSG_CONNECT_TYPE:
- #if BROKER
- MqttMsgConnect connect = MqttMsgConnect.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- Trace.WriteLine(TraceLevel.Frame, "RECV {0}", connect);
- #endif
- // raise message received event
- this.OnInternalEvent(new MsgInternalEvent(connect));
- break;
- #else
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #endif
-
- // CONNACK message received
- case MqttMsgBase.MQTT_MSG_CONNACK_TYPE:
- #if BROKER
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #else
- this.msgReceived = MqttMsgConnack.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Frame, "RECV {0}", this.msgReceived);
- #endif
- this.syncEndReceiving.Set();
- break;
- #endif
- // PINGREQ message received
- case MqttMsgBase.MQTT_MSG_PINGREQ_TYPE:
- #if BROKER
- this.msgReceived = MqttMsgPingReq.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- Trace.WriteLine(TraceLevel.Frame, "RECV {0}", this.msgReceived);
- #endif
- MqttMsgPingResp pingresp = new MqttMsgPingResp();
- this.Send(pingresp);
- break;
- #else
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #endif
- // PINGRESP message received
- case MqttMsgBase.MQTT_MSG_PINGRESP_TYPE:
- #if BROKER
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #else
- this.msgReceived = MqttMsgPingResp.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Frame, "RECV {0}", this.msgReceived);
- #endif
- this.syncEndReceiving.Set();
- break;
- #endif
- // SUBSCRIBE message received
- case MqttMsgBase.MQTT_MSG_SUBSCRIBE_TYPE:
- #if BROKER
- MqttMsgSubscribe subscribe = MqttMsgSubscribe.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- Trace.WriteLine(TraceLevel.Frame, "RECV {0}", subscribe);
- #endif
- // raise message received event
- this.OnInternalEvent(new MsgInternalEvent(subscribe));
- break;
- #else
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #endif
- // SUBACK message received
- case MqttMsgBase.MQTT_MSG_SUBACK_TYPE:
- #if BROKER
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #else
- // enqueue SUBACK message received (for QoS Level 1) into the internal queue
- MqttMsgSuback suback = MqttMsgSuback.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Frame, "RECV {0}", suback);
- #endif
- // enqueue SUBACK message into the internal queue
- this.EnqueueInternal(suback);
- break;
- #endif
- // PUBLISH message received
- case MqttMsgBase.MQTT_MSG_PUBLISH_TYPE:
- MqttMsgPublish publish = MqttMsgPublish.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Frame, "RECV {0}", publish);
- #endif
- // enqueue PUBLISH message to acknowledge into the inflight queue
- this.EnqueueInflight(publish, MqttMsgFlow.ToAcknowledge);
- break;
- // PUBACK message received
- case MqttMsgBase.MQTT_MSG_PUBACK_TYPE:
- // enqueue PUBACK message received (for QoS Level 1) into the internal queue
- MqttMsgPuback puback = MqttMsgPuback.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Frame, "RECV {0}", puback);
- #endif
- // enqueue PUBACK message into the internal queue
- this.EnqueueInternal(puback);
- break;
- // PUBREC message received
- case MqttMsgBase.MQTT_MSG_PUBREC_TYPE:
- // enqueue PUBREC message received (for QoS Level 2) into the internal queue
- MqttMsgPubrec pubrec = MqttMsgPubrec.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Frame, "RECV {0}", pubrec);
- #endif
- // enqueue PUBREC message into the internal queue
- this.EnqueueInternal(pubrec);
- break;
- // PUBREL message received
- case MqttMsgBase.MQTT_MSG_PUBREL_TYPE:
- // enqueue PUBREL message received (for QoS Level 2) into the internal queue
- MqttMsgPubrel pubrel = MqttMsgPubrel.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Frame, "RECV {0}", pubrel);
- #endif
- // enqueue PUBREL message into the internal queue
- this.EnqueueInternal(pubrel);
- break;
-
- // PUBCOMP message received
- case MqttMsgBase.MQTT_MSG_PUBCOMP_TYPE:
- // enqueue PUBCOMP message received (for QoS Level 2) into the internal queue
- MqttMsgPubcomp pubcomp = MqttMsgPubcomp.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Frame, "RECV {0}", pubcomp);
- #endif
- // enqueue PUBCOMP message into the internal queue
- this.EnqueueInternal(pubcomp);
- break;
- // UNSUBSCRIBE message received
- case MqttMsgBase.MQTT_MSG_UNSUBSCRIBE_TYPE:
- #if BROKER
- MqttMsgUnsubscribe unsubscribe = MqttMsgUnsubscribe.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- Trace.WriteLine(TraceLevel.Frame, "RECV {0}", unsubscribe);
- #endif
- // raise message received event
- this.OnInternalEvent(new MsgInternalEvent(unsubscribe));
- break;
- #else
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #endif
- // UNSUBACK message received
- case MqttMsgBase.MQTT_MSG_UNSUBACK_TYPE:
- #if BROKER
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #else
- // enqueue UNSUBACK message received (for QoS Level 1) into the internal queue
- MqttMsgUnsuback unsuback = MqttMsgUnsuback.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Frame, "RECV {0}", unsuback);
- #endif
- // enqueue UNSUBACK message into the internal queue
- this.EnqueueInternal(unsuback);
- break;
- #endif
- // DISCONNECT message received
- case MqttMsgDisconnect.MQTT_MSG_DISCONNECT_TYPE:
- #if BROKER
- MqttMsgDisconnect disconnect = MqttMsgDisconnect.Parse(fixedHeaderFirstByte[0], (byte)this.ProtocolVersion, this.channel);
- #if TRACE
- Trace.WriteLine(TraceLevel.Frame, "RECV {0}", disconnect);
- #endif
- // raise message received event
- this.OnInternalEvent(new MsgInternalEvent(disconnect));
- break;
- #else
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #endif
- default:
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- }
- this.exReceiving = null;
- }
- // zero bytes read, peer gracefully closed socket
- else
- {
- // wake up thread that will notify connection is closing
- this.OnConnectionClosing();
- }
- }
- catch (Exception e)
- {
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Error, "Exception occurred: {0}", e.ToString());
- #endif
- this.exReceiving = new MqttCommunicationException(e);
- bool close = false;
- if (e.GetType() == typeof(MqttClientException))
- {
- // [v3.1.1] scenarios the receiver MUST close the network connection
- MqttClientException ex = e as MqttClientException;
- close = ((ex.ErrorCode == MqttClientErrorCode.InvalidFlagBits) ||
- (ex.ErrorCode == MqttClientErrorCode.InvalidProtocolName) ||
- (ex.ErrorCode == MqttClientErrorCode.InvalidConnectFlags));
- }
- #if !(WINDOWS_APP || WINDOWS_PHONE_APP || (!UNITY_EDITOR&&UNITY_WSA_10_0&&!ENABLE_IL2CPP))
- else if ((e.GetType() == typeof(IOException)) || (e.GetType() == typeof(SocketException)) ||
- ((e.InnerException != null) && (e.InnerException.GetType() == typeof(SocketException)))) // added for SSL/TLS incoming connection that use SslStream that wraps SocketException
- {
- close = true;
- }
- #endif
-
- if (close)
- {
- // wake up thread that will notify connection is closing
- this.OnConnectionClosing();
- }
- }
- }
- }
- /// <summary>
- /// Thread for handling keep alive message
- /// </summary>
- private void KeepAliveThread()
- {
- int delta = 0;
- int wait = this.keepAlivePeriod;
-
- // create event to signal that current thread is end
- this.keepAliveEventEnd = new AutoResetEvent(false);
- while (this.isRunning)
- {
- #if (MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK)
- // waiting...
- this.keepAliveEvent.WaitOne(wait, false);
- #else
- // waiting...
- this.keepAliveEvent.WaitOne(wait);
- #endif
- if (this.isRunning)
- {
- delta = Environment.TickCount - this.lastCommTime;
- // if timeout exceeded ...
- if (delta >= this.keepAlivePeriod)
- {
- #if BROKER
- // client must close connection
- this.OnConnectionClosing();
- #else
- // ... send keep alive
- this.Ping();
- wait = this.keepAlivePeriod;
- #endif
- }
- else
- {
- // update waiting time
- wait = this.keepAlivePeriod - delta;
- }
- }
- }
- // signal thread end
- this.keepAliveEventEnd.Set();
- }
- /// <summary>
- /// Thread for raising event
- /// </summary>
- private void DispatchEventThread()
- {
- while (this.isRunning)
- {
- #if BROKER
- if ((this.eventQueue.Count == 0) && !this.isConnectionClosing)
- {
- // broker need to receive the first message (CONNECT)
- // within a reasonable amount of time after TCP/IP connection
- if (!this.IsConnected)
- {
- // wait on receiving message from client with a connection timeout
- if (!this.receiveEventWaitHandle.WaitOne(this.settings.TimeoutOnConnection))
- {
- // client must close connection
- this.Close();
- // client raw disconnection
- this.OnConnectionClosed();
- }
- }
- else
- {
- // wait on receiving message from client
- this.receiveEventWaitHandle.WaitOne();
- }
- }
- #else
- if ((this.eventQueue.Count == 0) && !this.isConnectionClosing)
- // wait on receiving message from client
- this.receiveEventWaitHandle.WaitOne();
- #endif
- // check if it is running or we are closing client
- if (this.isRunning)
- {
- // get event from queue
- InternalEvent internalEvent = null;
- lock (this.eventQueue)
- {
- if (this.eventQueue.Count > 0)
- internalEvent = (InternalEvent)this.eventQueue.Dequeue();
- }
- // it's an event with a message inside
- if (internalEvent != null)
- {
- MqttMsgBase msg = ((MsgInternalEvent)internalEvent).Message;
- if (msg != null)
- {
- switch (msg.Type)
- {
- // CONNECT message received
- case MqttMsgBase.MQTT_MSG_CONNECT_TYPE:
- #if BROKER
- // raise connected client event (CONNECT message received)
- this.OnMqttMsgConnected((MqttMsgConnect)msg);
- break;
- #else
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #endif
- // SUBSCRIBE message received
- case MqttMsgBase.MQTT_MSG_SUBSCRIBE_TYPE:
- #if BROKER
- MqttMsgSubscribe subscribe = (MqttMsgSubscribe)msg;
- // raise subscribe topic event (SUBSCRIBE message received)
- this.OnMqttMsgSubscribeReceived(subscribe.MessageId, subscribe.Topics, subscribe.QoSLevels);
- break;
- #else
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #endif
- // SUBACK message received
- case MqttMsgBase.MQTT_MSG_SUBACK_TYPE:
- // raise subscribed topic event (SUBACK message received)
- this.OnMqttMsgSubscribed((MqttMsgSuback)msg);
- break;
- // PUBLISH message received
- case MqttMsgBase.MQTT_MSG_PUBLISH_TYPE:
- // PUBLISH message received in a published internal event, no publish succeeded
- if (internalEvent.GetType() == typeof(MsgPublishedInternalEvent))
- this.OnMqttMsgPublished(msg.MessageId, false);
- else
- // raise PUBLISH message received event
- this.OnMqttMsgPublishReceived((MqttMsgPublish)msg);
- break;
- // PUBACK message received
- case MqttMsgBase.MQTT_MSG_PUBACK_TYPE:
- // raise published message event
- // (PUBACK received for QoS Level 1)
- this.OnMqttMsgPublished(msg.MessageId, true);
- break;
- // PUBREL message received
- case MqttMsgBase.MQTT_MSG_PUBREL_TYPE:
- // raise message received event
- // (PUBREL received for QoS Level 2)
- this.OnMqttMsgPublishReceived((MqttMsgPublish)msg);
- break;
- // PUBCOMP message received
- case MqttMsgBase.MQTT_MSG_PUBCOMP_TYPE:
- // raise published message event
- // (PUBCOMP received for QoS Level 2)
- this.OnMqttMsgPublished(msg.MessageId, true);
- break;
- // UNSUBSCRIBE message received from client
- case MqttMsgBase.MQTT_MSG_UNSUBSCRIBE_TYPE:
- #if BROKER
- MqttMsgUnsubscribe unsubscribe = (MqttMsgUnsubscribe)msg;
- // raise unsubscribe topic event (UNSUBSCRIBE message received)
- this.OnMqttMsgUnsubscribeReceived(unsubscribe.MessageId, unsubscribe.Topics);
- break;
- #else
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #endif
- // UNSUBACK message received
- case MqttMsgBase.MQTT_MSG_UNSUBACK_TYPE:
- // raise unsubscribed topic event
- this.OnMqttMsgUnsubscribed(msg.MessageId);
- break;
- // DISCONNECT message received from client
- case MqttMsgDisconnect.MQTT_MSG_DISCONNECT_TYPE:
- #if BROKER
- // raise disconnected client event (DISCONNECT message received)
- this.OnMqttMsgDisconnected();
- break;
- #else
- throw new MqttClientException(MqttClientErrorCode.WrongBrokerMessage);
- #endif
- }
- }
- }
-
- // all events for received messages dispatched, check if there is closing connection
- if ((this.eventQueue.Count == 0) && this.isConnectionClosing)
- {
- // client must close connection
- this.Close();
- // client raw disconnection
- this.OnConnectionClosed();
- }
- }
- }
- }
- /// <summary>
- /// Process inflight messages queue
- /// </summary>
- private void ProcessInflightThread()
- {
- MqttMsgContext msgContext = null;
- MqttMsgBase msgInflight = null;
- MqttMsgBase msgReceived = null;
- InternalEvent internalEvent = null;
- bool acknowledge = false;
- int timeout = Timeout.Infinite;
- int delta;
- bool msgReceivedProcessed = false;
- try
- {
- while (this.isRunning)
- {
- #if (MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK)
- // wait on message queueud to inflight
- this.inflightWaitHandle.WaitOne(timeout, false);
- #else
- // wait on message queueud to inflight
- this.inflightWaitHandle.WaitOne(timeout);
- #endif
- // it could be unblocked because Close() method is joining
- if (this.isRunning)
- {
- lock (this.inflightQueue)
- {
- // message received and peeked from internal queue is processed
- // NOTE : it has the corresponding message in inflight queue based on messageId
- // (ex. a PUBREC for a PUBLISH, a SUBACK for a SUBSCRIBE, ...)
- // if it's orphan we need to remove from internal queue
- msgReceivedProcessed = false;
- acknowledge = false;
- msgReceived = null;
- // set timeout tu MaxValue instead of Infinte (-1) to perform
- // compare with calcultad current msgTimeout
- timeout = Int32.MaxValue;
- // a message inflight could be re-enqueued but we have to
- // analyze it only just one time for cycle
- int count = this.inflightQueue.Count;
- // process all inflight queued messages
- while (count > 0)
- {
- count--;
- acknowledge = false;
- msgReceived = null;
- // check to be sure that client isn't closing and all queues are now empty !
- if (!this.isRunning)
- break;
- // dequeue message context from queue
- msgContext = (MqttMsgContext)this.inflightQueue.Dequeue();
- // get inflight message
- msgInflight = (MqttMsgBase)msgContext.Message;
- switch (msgContext.State)
- {
- case MqttMsgState.QueuedQos0:
- // QoS 0, PUBLISH message to send to broker, no state change, no acknowledge
- if (msgContext.Flow == MqttMsgFlow.ToPublish)
- {
- this.Send(msgInflight);
- }
- // QoS 0, no need acknowledge
- else if (msgContext.Flow == MqttMsgFlow.ToAcknowledge)
- {
- internalEvent = new MsgInternalEvent(msgInflight);
- // notify published message from broker (no need acknowledged)
- this.OnInternalEvent(internalEvent);
- }
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "processed {0}", msgInflight);
- #endif
- break;
- case MqttMsgState.QueuedQos1:
- // [v3.1.1] SUBSCRIBE and UNSIBSCRIBE aren't "officially" QOS = 1
- case MqttMsgState.SendSubscribe:
- case MqttMsgState.SendUnsubscribe:
- // QoS 1, PUBLISH or SUBSCRIBE/UNSUBSCRIBE message to send to broker, state change to wait PUBACK or SUBACK/UNSUBACK
- if (msgContext.Flow == MqttMsgFlow.ToPublish)
- {
- msgContext.Timestamp = Environment.TickCount;
- msgContext.Attempt++;
- if (msgInflight.Type == MqttMsgBase.MQTT_MSG_PUBLISH_TYPE)
- {
- // PUBLISH message to send, wait for PUBACK
- msgContext.State = MqttMsgState.WaitForPuback;
- // retry ? set dup flag [v3.1.1] only for PUBLISH message
- if (msgContext.Attempt > 1)
- msgInflight.DupFlag = true;
- }
- else if (msgInflight.Type == MqttMsgBase.MQTT_MSG_SUBSCRIBE_TYPE)
- // SUBSCRIBE message to send, wait for SUBACK
- msgContext.State = MqttMsgState.WaitForSuback;
- else if (msgInflight.Type == MqttMsgBase.MQTT_MSG_UNSUBSCRIBE_TYPE)
- // UNSUBSCRIBE message to send, wait for UNSUBACK
- msgContext.State = MqttMsgState.WaitForUnsuback;
- this.Send(msgInflight);
- // update timeout : minimum between delay (based on current message sent) or current timeout
- timeout = (this.settings.DelayOnRetry < timeout) ? this.settings.DelayOnRetry : timeout;
- // re-enqueue message (I have to re-analyze for receiving PUBACK, SUBACK or UNSUBACK)
- this.inflightQueue.Enqueue(msgContext);
- }
- // QoS 1, PUBLISH message received from broker to acknowledge, send PUBACK
- else if (msgContext.Flow == MqttMsgFlow.ToAcknowledge)
- {
- MqttMsgPuback puback = new MqttMsgPuback();
- puback.MessageId = msgInflight.MessageId;
- this.Send(puback);
- internalEvent = new MsgInternalEvent(msgInflight);
- // notify published message from broker and acknowledged
- this.OnInternalEvent(internalEvent);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "processed {0}", msgInflight);
- #endif
- }
- break;
- case MqttMsgState.QueuedQos2:
- // QoS 2, PUBLISH message to send to broker, state change to wait PUBREC
- if (msgContext.Flow == MqttMsgFlow.ToPublish)
- {
- msgContext.Timestamp = Environment.TickCount;
- msgContext.Attempt++;
- msgContext.State = MqttMsgState.WaitForPubrec;
- // retry ? set dup flag
- if (msgContext.Attempt > 1)
- msgInflight.DupFlag = true;
- this.Send(msgInflight);
- // update timeout : minimum between delay (based on current message sent) or current timeout
- timeout = (this.settings.DelayOnRetry < timeout) ? this.settings.DelayOnRetry : timeout;
- // re-enqueue message (I have to re-analyze for receiving PUBREC)
- this.inflightQueue.Enqueue(msgContext);
- }
- // QoS 2, PUBLISH message received from broker to acknowledge, send PUBREC, state change to wait PUBREL
- else if (msgContext.Flow == MqttMsgFlow.ToAcknowledge)
- {
- MqttMsgPubrec pubrec = new MqttMsgPubrec();
- pubrec.MessageId = msgInflight.MessageId;
- msgContext.State = MqttMsgState.WaitForPubrel;
- this.Send(pubrec);
- // re-enqueue message (I have to re-analyze for receiving PUBREL)
- this.inflightQueue.Enqueue(msgContext);
- }
- break;
- case MqttMsgState.WaitForPuback:
- case MqttMsgState.WaitForSuback:
- case MqttMsgState.WaitForUnsuback:
- // QoS 1, waiting for PUBACK of a PUBLISH message sent or
- // waiting for SUBACK of a SUBSCRIBE message sent or
- // waiting for UNSUBACK of a UNSUBSCRIBE message sent or
- if (msgContext.Flow == MqttMsgFlow.ToPublish)
- {
- acknowledge = false;
- lock (this.internalQueue)
- {
- if (this.internalQueue.Count > 0)
- msgReceived = (MqttMsgBase)this.internalQueue.Peek();
- }
- // it is a PUBACK message or a SUBACK/UNSUBACK message
- if (msgReceived != null)
- {
- // PUBACK message or SUBACK/UNSUBACK message for the current message
- if (((msgReceived.Type == MqttMsgBase.MQTT_MSG_PUBACK_TYPE) && (msgInflight.Type == MqttMsgBase.MQTT_MSG_PUBLISH_TYPE) && (msgReceived.MessageId == msgInflight.MessageId)) ||
- ((msgReceived.Type == MqttMsgBase.MQTT_MSG_SUBACK_TYPE) && (msgInflight.Type == MqttMsgBase.MQTT_MSG_SUBSCRIBE_TYPE) && (msgReceived.MessageId == msgInflight.MessageId)) ||
- ((msgReceived.Type == MqttMsgBase.MQTT_MSG_UNSUBACK_TYPE) && (msgInflight.Type == MqttMsgBase.MQTT_MSG_UNSUBSCRIBE_TYPE) && (msgReceived.MessageId == msgInflight.MessageId)))
- {
- lock (this.internalQueue)
- {
- // received message processed
- this.internalQueue.Dequeue();
- acknowledge = true;
- msgReceivedProcessed = true;
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "dequeued {0}", msgReceived);
- #endif
- }
- // if PUBACK received, confirm published with flag
- if (msgReceived.Type == MqttMsgBase.MQTT_MSG_PUBACK_TYPE)
- internalEvent = new MsgPublishedInternalEvent(msgReceived, true);
- else
- internalEvent = new MsgInternalEvent(msgReceived);
- // notify received acknowledge from broker of a published message or subscribe/unsubscribe message
- this.OnInternalEvent(internalEvent);
- // PUBACK received for PUBLISH message with QoS Level 1, remove from session state
- if ((msgInflight.Type == MqttMsgBase.MQTT_MSG_PUBLISH_TYPE) &&
- (this.session != null) &&
- #if (MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK)
- (this.session.InflightMessages.Contains(msgContext.Key)))
- #else
- (this.session.InflightMessages.ContainsKey(msgContext.Key)))
- #endif
- {
- this.session.InflightMessages.Remove(msgContext.Key);
- }
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "processed {0}", msgInflight);
- #endif
- }
- }
- // current message not acknowledged, no PUBACK or SUBACK/UNSUBACK or not equal messageid
- if (!acknowledge)
- {
- delta = Environment.TickCount - msgContext.Timestamp;
- // check timeout for receiving PUBACK since PUBLISH was sent or
- // for receiving SUBACK since SUBSCRIBE was sent or
- // for receiving UNSUBACK since UNSUBSCRIBE was sent
- if (delta >= this.settings.DelayOnRetry)
- {
- // max retry not reached, resend
- if (msgContext.Attempt < this.settings.AttemptsOnRetry)
- {
- msgContext.State = MqttMsgState.QueuedQos1;
- // re-enqueue message
- this.inflightQueue.Enqueue(msgContext);
- // update timeout (0 -> reanalyze queue immediately)
- timeout = 0;
- }
- else
- {
- // if PUBACK for a PUBLISH message not received after retries, raise event for not published
- if (msgInflight.Type == MqttMsgBase.MQTT_MSG_PUBLISH_TYPE)
- {
- // PUBACK not received in time, PUBLISH retries failed, need to remove from session inflight messages too
- if ((this.session != null) &&
- #if (MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK)
- (this.session.InflightMessages.Contains(msgContext.Key)))
- #else
- (this.session.InflightMessages.ContainsKey(msgContext.Key)))
- #endif
- {
- this.session.InflightMessages.Remove(msgContext.Key);
- }
- internalEvent = new MsgPublishedInternalEvent(msgInflight, false);
- // notify not received acknowledge from broker and message not published
- this.OnInternalEvent(internalEvent);
- }
- // NOTE : not raise events for SUBACK or UNSUBACK not received
- // for the user no event raised means subscribe/unsubscribe failed
- }
- }
- else
- {
- // re-enqueue message (I have to re-analyze for receiving PUBACK, SUBACK or UNSUBACK)
- this.inflightQueue.Enqueue(msgContext);
- // update timeout
- int msgTimeout = (this.settings.DelayOnRetry - delta);
- timeout = (msgTimeout < timeout) ? msgTimeout : timeout;
- }
- }
- }
- break;
- case MqttMsgState.WaitForPubrec:
- // QoS 2, waiting for PUBREC of a PUBLISH message sent
- if (msgContext.Flow == MqttMsgFlow.ToPublish)
- {
- acknowledge = false;
- lock (this.internalQueue)
- {
- if (this.internalQueue.Count > 0)
- msgReceived = (MqttMsgBase)this.internalQueue.Peek();
- }
- // it is a PUBREC message
- if ((msgReceived != null) && (msgReceived.Type == MqttMsgBase.MQTT_MSG_PUBREC_TYPE))
- {
- // PUBREC message for the current PUBLISH message, send PUBREL, wait for PUBCOMP
- if (msgReceived.MessageId == msgInflight.MessageId)
- {
- lock (this.internalQueue)
- {
- // received message processed
- this.internalQueue.Dequeue();
- acknowledge = true;
- msgReceivedProcessed = true;
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "dequeued {0}", msgReceived);
- #endif
- }
- MqttMsgPubrel pubrel = new MqttMsgPubrel();
- pubrel.MessageId = msgInflight.MessageId;
- msgContext.State = MqttMsgState.WaitForPubcomp;
- msgContext.Timestamp = Environment.TickCount;
- msgContext.Attempt = 1;
- this.Send(pubrel);
- // update timeout : minimum between delay (based on current message sent) or current timeout
- timeout = (this.settings.DelayOnRetry < timeout) ? this.settings.DelayOnRetry : timeout;
- // re-enqueue message
- this.inflightQueue.Enqueue(msgContext);
- }
- }
- // current message not acknowledged
- if (!acknowledge)
- {
- delta = Environment.TickCount - msgContext.Timestamp;
- // check timeout for receiving PUBREC since PUBLISH was sent
- if (delta >= this.settings.DelayOnRetry)
- {
- // max retry not reached, resend
- if (msgContext.Attempt < this.settings.AttemptsOnRetry)
- {
- msgContext.State = MqttMsgState.QueuedQos2;
- // re-enqueue message
- this.inflightQueue.Enqueue(msgContext);
- // update timeout (0 -> reanalyze queue immediately)
- timeout = 0;
- }
- else
- {
- // PUBREC not received in time, PUBLISH retries failed, need to remove from session inflight messages too
- if ((this.session != null) &&
- #if (MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK)
- (this.session.InflightMessages.Contains(msgContext.Key)))
- #else
- (this.session.InflightMessages.ContainsKey(msgContext.Key)))
- #endif
- {
- this.session.InflightMessages.Remove(msgContext.Key);
- }
- // if PUBREC for a PUBLISH message not received after retries, raise event for not published
- internalEvent = new MsgPublishedInternalEvent(msgInflight, false);
- // notify not received acknowledge from broker and message not published
- this.OnInternalEvent(internalEvent);
- }
- }
- else
- {
- // re-enqueue message
- this.inflightQueue.Enqueue(msgContext);
- // update timeout
- int msgTimeout = (this.settings.DelayOnRetry - delta);
- timeout = (msgTimeout < timeout) ? msgTimeout : timeout;
- }
- }
- }
- break;
- case MqttMsgState.WaitForPubrel:
- // QoS 2, waiting for PUBREL of a PUBREC message sent
- if (msgContext.Flow == MqttMsgFlow.ToAcknowledge)
- {
- lock (this.internalQueue)
- {
- if (this.internalQueue.Count > 0)
- msgReceived = (MqttMsgBase)this.internalQueue.Peek();
- }
- // it is a PUBREL message
- if ((msgReceived != null) && (msgReceived.Type == MqttMsgBase.MQTT_MSG_PUBREL_TYPE))
- {
- // PUBREL message for the current message, send PUBCOMP
- if (msgReceived.MessageId == msgInflight.MessageId)
- {
- lock (this.internalQueue)
- {
- // received message processed
- this.internalQueue.Dequeue();
- msgReceivedProcessed = true;
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "dequeued {0}", msgReceived);
- #endif
- }
- MqttMsgPubcomp pubcomp = new MqttMsgPubcomp();
- pubcomp.MessageId = msgInflight.MessageId;
- this.Send(pubcomp);
- internalEvent = new MsgInternalEvent(msgInflight);
- // notify published message from broker and acknowledged
- this.OnInternalEvent(internalEvent);
- // PUBREL received (and PUBCOMP sent) for PUBLISH message with QoS Level 2, remove from session state
- if ((msgInflight.Type == MqttMsgBase.MQTT_MSG_PUBLISH_TYPE) &&
- (this.session != null) &&
- #if (MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK)
- (this.session.InflightMessages.Contains(msgContext.Key)))
- #else
- (this.session.InflightMessages.ContainsKey(msgContext.Key)))
- #endif
- {
- this.session.InflightMessages.Remove(msgContext.Key);
- }
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "processed {0}", msgInflight);
- #endif
- }
- else
- {
- // re-enqueue message
- this.inflightQueue.Enqueue(msgContext);
- }
- }
- else
- {
- // re-enqueue message
- this.inflightQueue.Enqueue(msgContext);
- }
- }
- break;
- case MqttMsgState.WaitForPubcomp:
- // QoS 2, waiting for PUBCOMP of a PUBREL message sent
- if (msgContext.Flow == MqttMsgFlow.ToPublish)
- {
- acknowledge = false;
- lock (this.internalQueue)
- {
- if (this.internalQueue.Count > 0)
- msgReceived = (MqttMsgBase)this.internalQueue.Peek();
- }
- // it is a PUBCOMP message
- if ((msgReceived != null) && (msgReceived.Type == MqttMsgBase.MQTT_MSG_PUBCOMP_TYPE))
- {
- // PUBCOMP message for the current message
- if (msgReceived.MessageId == msgInflight.MessageId)
- {
- lock (this.internalQueue)
- {
- // received message processed
- this.internalQueue.Dequeue();
- acknowledge = true;
- msgReceivedProcessed = true;
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "dequeued {0}", msgReceived);
- #endif
- }
- internalEvent = new MsgPublishedInternalEvent(msgReceived, true);
- // notify received acknowledge from broker of a published message
- this.OnInternalEvent(internalEvent);
- // PUBCOMP received for PUBLISH message with QoS Level 2, remove from session state
- if ((msgInflight.Type == MqttMsgBase.MQTT_MSG_PUBLISH_TYPE) &&
- (this.session != null) &&
- #if (MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK)
- (this.session.InflightMessages.Contains(msgContext.Key)))
- #else
- (this.session.InflightMessages.ContainsKey(msgContext.Key)))
- #endif
- {
- this.session.InflightMessages.Remove(msgContext.Key);
- }
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "processed {0}", msgInflight);
- #endif
- }
- }
- // it is a PUBREC message
- else if ((msgReceived != null) && (msgReceived.Type == MqttMsgBase.MQTT_MSG_PUBREC_TYPE))
- {
- // another PUBREC message for the current message due to a retransmitted PUBLISH
- // I'm in waiting for PUBCOMP, so I can discard this PUBREC
- if (msgReceived.MessageId == msgInflight.MessageId)
- {
- lock (this.internalQueue)
- {
- // received message processed
- this.internalQueue.Dequeue();
- acknowledge = true;
- msgReceivedProcessed = true;
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "dequeued {0}", msgReceived);
- #endif
- // re-enqueue message
- this.inflightQueue.Enqueue(msgContext);
- }
- }
- }
- // current message not acknowledged
- if (!acknowledge)
- {
- delta = Environment.TickCount - msgContext.Timestamp;
- // check timeout for receiving PUBCOMP since PUBREL was sent
- if (delta >= this.settings.DelayOnRetry)
- {
- // max retry not reached, resend
- if (msgContext.Attempt < this.settings.AttemptsOnRetry)
- {
- msgContext.State = MqttMsgState.SendPubrel;
- // re-enqueue message
- this.inflightQueue.Enqueue(msgContext);
- // update timeout (0 -> reanalyze queue immediately)
- timeout = 0;
- }
- else
- {
- // PUBCOMP not received, PUBREL retries failed, need to remove from session inflight messages too
- if ((this.session != null) &&
- #if (MF_FRAMEWORK_VERSION_V4_2 || MF_FRAMEWORK_VERSION_V4_3 || COMPACT_FRAMEWORK)
- (this.session.InflightMessages.Contains(msgContext.Key)))
- #else
- (this.session.InflightMessages.ContainsKey(msgContext.Key)))
- #endif
- {
- this.session.InflightMessages.Remove(msgContext.Key);
- }
- // if PUBCOMP for a PUBLISH message not received after retries, raise event for not published
- internalEvent = new MsgPublishedInternalEvent(msgInflight, false);
- // notify not received acknowledge from broker and message not published
- this.OnInternalEvent(internalEvent);
- }
- }
- else
- {
- // re-enqueue message
- this.inflightQueue.Enqueue(msgContext);
- // update timeout
- int msgTimeout = (this.settings.DelayOnRetry - delta);
- timeout = (msgTimeout < timeout) ? msgTimeout : timeout;
- }
- }
- }
- break;
- case MqttMsgState.SendPubrec:
- // TODO : impossible ? --> QueuedQos2 ToAcknowledge
- break;
- case MqttMsgState.SendPubrel:
- // QoS 2, PUBREL message to send to broker, state change to wait PUBCOMP
- if (msgContext.Flow == MqttMsgFlow.ToPublish)
- {
- MqttMsgPubrel pubrel = new MqttMsgPubrel();
- pubrel.MessageId = msgInflight.MessageId;
- msgContext.State = MqttMsgState.WaitForPubcomp;
- msgContext.Timestamp = Environment.TickCount;
- msgContext.Attempt++;
- // retry ? set dup flag [v3.1.1] no needed
- if (this.ProtocolVersion == MqttProtocolVersion.Version_3_1)
- {
- if (msgContext.Attempt > 1)
- pubrel.DupFlag = true;
- }
-
- this.Send(pubrel);
- // update timeout : minimum between delay (based on current message sent) or current timeout
- timeout = (this.settings.DelayOnRetry < timeout) ? this.settings.DelayOnRetry : timeout;
- // re-enqueue message
- this.inflightQueue.Enqueue(msgContext);
- }
- break;
- case MqttMsgState.SendPubcomp:
- // TODO : impossible ?
- break;
- case MqttMsgState.SendPuback:
- // TODO : impossible ? --> QueuedQos1 ToAcknowledge
- break;
- default:
- break;
- }
- }
- // if calculated timeout is MaxValue, it means that must be Infinite (-1)
- if (timeout == Int32.MaxValue)
- timeout = Timeout.Infinite;
- // if message received is orphan, no corresponding message in inflight queue
- // based on messageId, we need to remove from the queue
- if ((msgReceived != null) && !msgReceivedProcessed)
- {
- this.internalQueue.Dequeue();
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Queuing, "dequeued {0} orphan", msgReceived);
- #endif
- }
- }
- }
- }
- }
- catch (MqttCommunicationException e)
- {
- // possible exception on Send, I need to re-enqueue not sent message
- if (msgContext != null)
- // re-enqueue message
- this.inflightQueue.Enqueue(msgContext);
- #if TRACE
- MqttUtility.Trace.WriteLine(TraceLevel.Error, "Exception occurred: {0}", e.ToString());
- #endif
- // raise disconnection client event
- this.OnConnectionClosing();
- }
- }
- /// <summary>
- /// Restore session
- /// </summary>
- private void RestoreSession()
- {
- // if not clean session
- if (!this.CleanSession)
- {
- // there is a previous session
- if (this.session != null)
- {
- lock (this.inflightQueue)
- {
- foreach (MqttMsgContext msgContext in this.session.InflightMessages.Values)
- {
- this.inflightQueue.Enqueue(msgContext);
- // if it is a PUBLISH message to publish
- if ((msgContext.Message.Type == MqttMsgBase.MQTT_MSG_PUBLISH_TYPE) &&
- (msgContext.Flow == MqttMsgFlow.ToPublish))
- {
- // it's QoS 1 and we haven't received PUBACK
- if ((msgContext.Message.QosLevel == MqttMsgBase.QOS_LEVEL_AT_LEAST_ONCE) &&
- (msgContext.State == MqttMsgState.WaitForPuback))
- {
- // we haven't received PUBACK, we need to resend PUBLISH message
- msgContext.State = MqttMsgState.QueuedQos1;
- }
- // it's QoS 2
- else if (msgContext.Message.QosLevel == MqttMsgBase.QOS_LEVEL_EXACTLY_ONCE)
- {
- // we haven't received PUBREC, we need to resend PUBLISH message
- if (msgContext.State == MqttMsgState.WaitForPubrec)
- {
- msgContext.State = MqttMsgState.QueuedQos2;
- }
- // we haven't received PUBCOMP, we need to resend PUBREL for it
- else if (msgContext.State == MqttMsgState.WaitForPubcomp)
- {
- msgContext.State = MqttMsgState.SendPubrel;
- }
- }
- }
- }
- }
- // unlock process inflight queue
- this.inflightWaitHandle.Set();
- }
- else
- {
- // create new session
- this.session = new MqttClientSession(this.ClientId);
- }
- }
- // clean any previous session
- else
- {
- if (this.session != null)
- this.session.Clear();
- }
- }
- #if BROKER
- /// <summary>
- /// Load a given session
- /// </summary>
- /// <param name="session">MQTT Client session to load</param>
- public void LoadSession(MqttClientSession session)
- {
- // if not clean session
- if (!this.CleanSession)
- {
- // set the session ...
- this.session = session;
- // ... and restore it
- this.RestoreSession();
- }
- }
- #endif
- /// <summary>
- /// 生成下一个消息标识符
- /// </summary>
- /// <returns>消息标识符</returns>
- private ushort GetMessageId()
- {
- // 如果为0或最大UInt16,则变为1(第一个有效的messageId)
- this.messageIdCounter = ((this.messageIdCounter % UInt16.MaxValue) != 0) ? (ushort)(this.messageIdCounter + 1) : (ushort)1;
- return this.messageIdCounter;
- }
- /// <summary>
- /// Finder class for PUBLISH message inside a queue
- /// </summary>
- internal class MqttMsgContextFinder
- {
- // PUBLISH message id
- internal ushort MessageId { get; set; }
- // message flow into inflight queue
- internal MqttMsgFlow Flow { get; set; }
- /// <summary>
- /// Constructor
- /// </summary>
- /// <param name="messageId">Message Id</param>
- /// <param name="flow">Message flow inside inflight queue</param>
- internal MqttMsgContextFinder(ushort messageId, MqttMsgFlow flow)
- {
- this.MessageId = messageId;
- this.Flow = flow;
- }
- internal bool Find(object item)
- {
- MqttMsgContext msgCtx = (MqttMsgContext)item;
- return ((msgCtx.Message.Type == MqttMsgBase.MQTT_MSG_PUBLISH_TYPE) &&
- (msgCtx.Message.MessageId == this.MessageId) &&
- msgCtx.Flow == this.Flow);
- }
- }
- }
- /// <summary>
- /// MQTT protocol version
- /// </summary>
- public enum MqttProtocolVersion
- {
- Version_3_1 = MqttMsgConnect.PROTOCOL_VERSION_V3_1,
- Version_3_1_1 = MqttMsgConnect.PROTOCOL_VERSION_V3_1_1
- }
- }
|