private static MqttMsgBase WaitForPubrel(MqttClientConnection clientConnection, MqttMsgContext msgContext, MqttMsgBase msgInflight, ref bool msgReceivedProcessed)
{
// QoS 2, waiting for PUBREL of a PUBREC message sent
if (msgContext.Flow != MqttMsgFlow.ToAcknowledge)
{
return(null);
}
MqttMsgBase msgReceived;
if (!clientConnection.InternalQueue.TryPeek(out msgReceived))
{
return(null);
}
InternalEvent internalEvent;
// 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)
{
// received message processed
MqttMsgBase dequeuedMsg;
clientConnection.InternalQueue.TryDequeue(out dequeuedMsg);
msgReceivedProcessed = true;
MqttOutgoingMessageManager.Pubcomp(clientConnection, msgInflight.MessageId);
internalEvent = new MsgInternalEvent(msgInflight);
// notify published message from broker and acknowledged
clientConnection.EnqueueInternalEvent(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) && (clientConnection.Session != null) &&
(clientConnection.Session.InflightMessages.ContainsKey(msgContext.Key)))
{
MqttMsgContext contextToBeRemoved;
clientConnection.Session.InflightMessages.TryRemove(msgContext.Key, out contextToBeRemoved);
}
}
else
{
// re-enqueue message
clientConnection.EnqueueInflight(msgContext);
}
}
else
{
// re-enqueue message
clientConnection.EnqueueInflight(msgContext);
}
return(msgReceived);
}
private static int HandleQueuedQos1SendSubscribeAndSendUnsubscribe(
MqttClientConnection clientConnection,
MqttMsgContext msgContext,
MqttMsgBase msgInflight,
int timeout)
{
InternalEvent internalEvent;
// 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;
}
MqttOutgoingMessageManager.Send(clientConnection, msgInflight);
// update timeout : minimum between delay (based on current message sent) or current timeout
timeout = (clientConnection.Settings.DelayOnRetry < timeout) ? clientConnection.Settings.DelayOnRetry : timeout;
// re-enqueue message (I have to re-analyze for receiving PUBACK, SUBACK or UNSUBACK)
clientConnection.EnqueueInflight(msgContext);
}
// QoS 1, PUBLISH message received from broker to acknowledge, send PUBACK
else if (msgContext.Flow == MqttMsgFlow.ToAcknowledge)
{
MqttOutgoingMessageManager.Puback(clientConnection, msgInflight.MessageId);
internalEvent = new MsgInternalEvent(msgInflight);
// notify published message from broker and acknowledged
clientConnection.EnqueueInternalEvent(internalEvent);
}
return(timeout);
}
private static void HandleQueuedQos0(
MqttClientConnection clientConnection,
MqttMsgContext msgContext,
MqttMsgBase msgInflight)
{
// QoS 0, PUBLISH message to send to broker, no state change, no acknowledge
if (msgContext.Flow == MqttMsgFlow.ToPublish)
{
MqttOutgoingMessageManager.Send(clientConnection, msgInflight);
}
// QoS 0, no need acknowledge
else if (msgContext.Flow == MqttMsgFlow.ToAcknowledge)
{
var internalEvent = new MsgInternalEvent(msgInflight);
clientConnection.EnqueueInternalEvent(internalEvent);
}
}
private static MqttMsgBase HandleWaitForPubackSubackUbsuback(
MqttClientConnection clientConnection,
MqttMsgContext msgContext,
MqttMsgBase msgInflight,
ref bool msgReceivedProcessed,
ref int timeout)
{
// 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)
{
return(null);
}
MqttMsgBase msgReceived;
if (!clientConnection.InternalQueue.TryPeek(out msgReceived))
{
return(null);
}
bool acknowledge = false;
InternalEvent internalEvent;
// 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)))
{
// received message processed
MqttMsgBase dequeuedMsg;
clientConnection.InternalQueue.TryDequeue(out dequeuedMsg);
acknowledge = true;
msgReceivedProcessed = true;
// 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
clientConnection.EnqueueInternalEvent(internalEvent);
// PUBACK received for PUBLISH message with QoS Level 1, remove from session state
if ((msgInflight.Type == MqttMsgBase.MQTT_MSG_PUBLISH_TYPE) && (clientConnection.Session != null) &&
(clientConnection.Session.InflightMessages.ContainsKey(msgContext.Key)))
{
MqttMsgContext contextToBeRemoved;
clientConnection.Session.InflightMessages.TryRemove(msgContext.Key, out contextToBeRemoved);
}
}
// current message not acknowledged, no PUBACK or SUBACK/UNSUBACK or not equal messageid
if (!acknowledge)
{
var 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 >= clientConnection.Settings.DelayOnRetry)
{
// max retry not reached, resend
if (msgContext.Attempt < clientConnection.Settings.AttemptsOnRetry)
{
msgContext.State = MqttMsgState.QueuedQos1;
// re-enqueue message
clientConnection.EnqueueInflight(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 ((clientConnection.Session != null) &&
(clientConnection.Session.InflightMessages.ContainsKey(msgContext.Key)))
{
MqttMsgContext contextToBeRemoved;
clientConnection.Session.InflightMessages.TryRemove(msgContext.Key, out contextToBeRemoved);
}
internalEvent = new MsgPublishedInternalEvent(msgInflight, false);
// notify not received acknowledge from broker and message not published
clientConnection.EnqueueInternalEvent(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)
clientConnection.EnqueueInflight(msgContext);
// update timeout
int msgTimeout = (clientConnection.Settings.DelayOnRetry - delta);
timeout = (msgTimeout < timeout) ? msgTimeout : timeout;
}
}
return(msgReceived);
}
