2014-05-07 22:27:00 +00:00
|
|
|
/*
|
2015-04-19 21:10:59 +00:00
|
|
|
Copyright (c) 2009-2015 Roger Light <roger@atchoo.org>
|
2014-05-07 22:27:00 +00:00
|
|
|
|
|
|
|
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:
|
|
|
|
Roger Light - initial implementation and documentation.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <assert.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <string.h>
|
|
|
|
|
2015-04-29 20:37:47 +00:00
|
|
|
#include "mosquitto.h"
|
|
|
|
#include "logging_mosq.h"
|
|
|
|
#include "memory_mosq.h"
|
|
|
|
#include "messages_mosq.h"
|
|
|
|
#include "mqtt3_protocol.h"
|
|
|
|
#include "net_mosq.h"
|
|
|
|
#include "packet_mosq.h"
|
|
|
|
#include "read_handle.h"
|
|
|
|
#include "send_mosq.h"
|
|
|
|
#include "util_mosq.h"
|
2014-05-07 22:27:00 +00:00
|
|
|
#ifdef WITH_BROKER
|
2015-04-29 20:37:47 +00:00
|
|
|
#include "mosquitto_broker.h"
|
2014-05-07 22:27:00 +00:00
|
|
|
#endif
|
|
|
|
|
2015-05-16 17:43:06 +00:00
|
|
|
int handle__pingreq(struct mosquitto *mosq)
|
2014-05-07 22:27:00 +00:00
|
|
|
{
|
|
|
|
assert(mosq);
|
|
|
|
#ifdef WITH_BROKER
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(NULL, MOSQ_LOG_DEBUG, "Received PINGREQ from %s", mosq->id);
|
2014-05-07 22:27:00 +00:00
|
|
|
#else
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(mosq, MOSQ_LOG_DEBUG, "Client %s received PINGREQ", mosq->id);
|
2014-05-07 22:27:00 +00:00
|
|
|
#endif
|
2015-05-16 14:24:24 +00:00
|
|
|
return send__pingresp(mosq);
|
2014-05-07 22:27:00 +00:00
|
|
|
}
|
|
|
|
|
2015-05-16 17:43:06 +00:00
|
|
|
int handle__pingresp(struct mosquitto *mosq)
|
2014-05-07 22:27:00 +00:00
|
|
|
{
|
|
|
|
assert(mosq);
|
|
|
|
mosq->ping_t = 0; /* No longer waiting for a PINGRESP. */
|
|
|
|
#ifdef WITH_BROKER
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(NULL, MOSQ_LOG_DEBUG, "Received PINGRESP from %s", mosq->id);
|
2014-05-07 22:27:00 +00:00
|
|
|
#else
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(mosq, MOSQ_LOG_DEBUG, "Client %s received PINGRESP", mosq->id);
|
2014-05-07 22:27:00 +00:00
|
|
|
#endif
|
|
|
|
return MOSQ_ERR_SUCCESS;
|
|
|
|
}
|
|
|
|
|
2014-11-17 23:46:02 +00:00
|
|
|
#ifdef WITH_BROKER
|
2015-05-16 17:43:06 +00:00
|
|
|
int handle__pubackcomp(struct mosquitto_db *db, struct mosquitto *mosq, const char *type)
|
2014-11-17 23:46:02 +00:00
|
|
|
#else
|
2015-05-16 17:43:06 +00:00
|
|
|
int handle__pubackcomp(struct mosquitto *mosq, const char *type)
|
2014-11-17 23:46:02 +00:00
|
|
|
#endif
|
2014-05-07 22:27:00 +00:00
|
|
|
{
|
|
|
|
uint16_t mid;
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
assert(mosq);
|
2015-05-16 13:16:40 +00:00
|
|
|
rc = packet__read_uint16(&mosq->in_packet, &mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
if(rc) return rc;
|
|
|
|
#ifdef WITH_BROKER
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(NULL, MOSQ_LOG_DEBUG, "Received %s from %s (Mid: %d)", type, mosq->id, mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
|
|
|
|
if(mid){
|
2015-05-16 14:24:24 +00:00
|
|
|
rc = db__message_delete(db, mosq, mid, mosq_md_out);
|
2014-05-07 22:27:00 +00:00
|
|
|
if(rc) return rc;
|
|
|
|
}
|
|
|
|
#else
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(mosq, MOSQ_LOG_DEBUG, "Client %s received %s (Mid: %d)", mosq->id, type, mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
|
2015-05-16 13:29:54 +00:00
|
|
|
if(!message__delete(mosq, mid, mosq_md_out)){
|
2014-05-07 22:27:00 +00:00
|
|
|
/* Only inform the client the message has been sent once. */
|
|
|
|
pthread_mutex_lock(&mosq->callback_mutex);
|
|
|
|
if(mosq->on_publish){
|
|
|
|
mosq->in_callback = true;
|
|
|
|
mosq->on_publish(mosq, mosq->userdata, mid);
|
|
|
|
mosq->in_callback = false;
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&mosq->callback_mutex);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return MOSQ_ERR_SUCCESS;
|
|
|
|
}
|
|
|
|
|
2015-05-16 17:43:06 +00:00
|
|
|
int handle__pubrec(struct mosquitto *mosq)
|
2014-05-07 22:27:00 +00:00
|
|
|
{
|
|
|
|
uint16_t mid;
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
assert(mosq);
|
2015-05-16 13:16:40 +00:00
|
|
|
rc = packet__read_uint16(&mosq->in_packet, &mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
if(rc) return rc;
|
|
|
|
#ifdef WITH_BROKER
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(NULL, MOSQ_LOG_DEBUG, "Received PUBREC from %s (Mid: %d)", mosq->id, mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
|
2015-05-16 14:24:24 +00:00
|
|
|
rc = db__message_update(mosq, mid, mosq_md_out, mosq_ms_wait_for_pubcomp);
|
2014-05-07 22:27:00 +00:00
|
|
|
#else
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(mosq, MOSQ_LOG_DEBUG, "Client %s received PUBREC (Mid: %d)", mosq->id, mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
|
2015-05-16 13:29:54 +00:00
|
|
|
rc = message__out_update(mosq, mid, mosq_ms_wait_for_pubcomp);
|
2014-05-07 22:27:00 +00:00
|
|
|
#endif
|
|
|
|
if(rc) return rc;
|
2015-05-16 14:24:24 +00:00
|
|
|
rc = send__pubrel(mosq, mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
if(rc) return rc;
|
|
|
|
|
|
|
|
return MOSQ_ERR_SUCCESS;
|
|
|
|
}
|
|
|
|
|
2015-05-16 17:43:06 +00:00
|
|
|
int handle__pubrel(struct mosquitto_db *db, struct mosquitto *mosq)
|
2014-05-07 22:27:00 +00:00
|
|
|
{
|
|
|
|
uint16_t mid;
|
|
|
|
#ifndef WITH_BROKER
|
|
|
|
struct mosquitto_message_all *message = NULL;
|
|
|
|
#endif
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
assert(mosq);
|
|
|
|
if(mosq->protocol == mosq_p_mqtt311){
|
|
|
|
if((mosq->in_packet.command&0x0F) != 0x02){
|
|
|
|
return MOSQ_ERR_PROTOCOL;
|
|
|
|
}
|
|
|
|
}
|
2015-05-16 13:16:40 +00:00
|
|
|
rc = packet__read_uint16(&mosq->in_packet, &mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
if(rc) return rc;
|
|
|
|
#ifdef WITH_BROKER
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(NULL, MOSQ_LOG_DEBUG, "Received PUBREL from %s (Mid: %d)", mosq->id, mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
|
2015-05-16 14:24:24 +00:00
|
|
|
if(db__message_release(db, mosq, mid, mosq_md_in)){
|
2014-05-07 22:27:00 +00:00
|
|
|
/* Message not found. Still send a PUBCOMP anyway because this could be
|
|
|
|
* due to a repeated PUBREL after a client has reconnected. */
|
|
|
|
}
|
|
|
|
#else
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(mosq, MOSQ_LOG_DEBUG, "Client %s received PUBREL (Mid: %d)", mosq->id, mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
|
2015-05-16 13:29:54 +00:00
|
|
|
if(!message__remove(mosq, mid, mosq_md_in, &message)){
|
2014-05-07 22:27:00 +00:00
|
|
|
/* Only pass the message on if we have removed it from the queue - this
|
|
|
|
* prevents multiple callbacks for the same message. */
|
|
|
|
pthread_mutex_lock(&mosq->callback_mutex);
|
|
|
|
if(mosq->on_message){
|
|
|
|
mosq->in_callback = true;
|
|
|
|
mosq->on_message(mosq, mosq->userdata, &message->msg);
|
|
|
|
mosq->in_callback = false;
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&mosq->callback_mutex);
|
2015-05-16 13:29:54 +00:00
|
|
|
message__cleanup(&message);
|
2014-05-07 22:27:00 +00:00
|
|
|
}
|
|
|
|
#endif
|
2015-05-16 14:24:24 +00:00
|
|
|
rc = send__pubcomp(mosq, mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
if(rc) return rc;
|
|
|
|
|
|
|
|
return MOSQ_ERR_SUCCESS;
|
|
|
|
}
|
|
|
|
|
2015-05-16 17:43:06 +00:00
|
|
|
int handle__suback(struct mosquitto *mosq)
|
2014-05-07 22:27:00 +00:00
|
|
|
{
|
|
|
|
uint16_t mid;
|
|
|
|
uint8_t qos;
|
|
|
|
int *granted_qos;
|
|
|
|
int qos_count;
|
|
|
|
int i = 0;
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
assert(mosq);
|
|
|
|
#ifdef WITH_BROKER
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(NULL, MOSQ_LOG_DEBUG, "Received SUBACK from %s", mosq->id);
|
2014-05-07 22:27:00 +00:00
|
|
|
#else
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(mosq, MOSQ_LOG_DEBUG, "Client %s received SUBACK", mosq->id);
|
2014-05-07 22:27:00 +00:00
|
|
|
#endif
|
2015-05-16 13:16:40 +00:00
|
|
|
rc = packet__read_uint16(&mosq->in_packet, &mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
if(rc) return rc;
|
|
|
|
|
|
|
|
qos_count = mosq->in_packet.remaining_length - mosq->in_packet.pos;
|
2015-04-19 21:10:59 +00:00
|
|
|
granted_qos = mosquitto__malloc(qos_count*sizeof(int));
|
2014-05-07 22:27:00 +00:00
|
|
|
if(!granted_qos) return MOSQ_ERR_NOMEM;
|
|
|
|
while(mosq->in_packet.pos < mosq->in_packet.remaining_length){
|
2015-05-16 13:16:40 +00:00
|
|
|
rc = packet__read_byte(&mosq->in_packet, &qos);
|
2014-05-07 22:27:00 +00:00
|
|
|
if(rc){
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__free(granted_qos);
|
2014-05-07 22:27:00 +00:00
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
granted_qos[i] = (int)qos;
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
#ifndef WITH_BROKER
|
|
|
|
pthread_mutex_lock(&mosq->callback_mutex);
|
|
|
|
if(mosq->on_subscribe){
|
|
|
|
mosq->in_callback = true;
|
|
|
|
mosq->on_subscribe(mosq, mosq->userdata, mid, qos_count, granted_qos);
|
|
|
|
mosq->in_callback = false;
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&mosq->callback_mutex);
|
|
|
|
#endif
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__free(granted_qos);
|
2014-05-07 22:27:00 +00:00
|
|
|
|
|
|
|
return MOSQ_ERR_SUCCESS;
|
|
|
|
}
|
|
|
|
|
2015-05-16 17:43:06 +00:00
|
|
|
int handle__unsuback(struct mosquitto *mosq)
|
2014-05-07 22:27:00 +00:00
|
|
|
{
|
|
|
|
uint16_t mid;
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
assert(mosq);
|
|
|
|
#ifdef WITH_BROKER
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(NULL, MOSQ_LOG_DEBUG, "Received UNSUBACK from %s", mosq->id);
|
2014-05-07 22:27:00 +00:00
|
|
|
#else
|
2015-04-19 21:10:59 +00:00
|
|
|
mosquitto__log_printf(mosq, MOSQ_LOG_DEBUG, "Client %s received UNSUBACK", mosq->id);
|
2014-05-07 22:27:00 +00:00
|
|
|
#endif
|
2015-05-16 13:16:40 +00:00
|
|
|
rc = packet__read_uint16(&mosq->in_packet, &mid);
|
2014-05-07 22:27:00 +00:00
|
|
|
if(rc) return rc;
|
|
|
|
#ifndef WITH_BROKER
|
|
|
|
pthread_mutex_lock(&mosq->callback_mutex);
|
|
|
|
if(mosq->on_unsubscribe){
|
|
|
|
mosq->in_callback = true;
|
|
|
|
mosq->on_unsubscribe(mosq, mosq->userdata, mid);
|
|
|
|
mosq->in_callback = false;
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&mosq->callback_mutex);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return MOSQ_ERR_SUCCESS;
|
|
|
|
}
|
|
|
|
|