/*
|
Copyright (c) 2009-2019 Roger Light <roger@atchoo.org>
|
|
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 "config.h"
|
|
#include <assert.h>
|
#include <stdio.h>
|
#include <string.h>
|
|
#ifdef WITH_BROKER
|
# include "mosquitto_broker_internal.h"
|
# include "sys_tree.h"
|
#else
|
# define G_PUB_BYTES_SENT_INC(A)
|
#endif
|
|
#include "mosquitto.h"
|
#include "mosquitto_internal.h"
|
#include "logging_mosq.h"
|
#include "mqtt_protocol.h"
|
#include "memory_mosq.h"
|
#include "net_mosq.h"
|
#include "packet_mosq.h"
|
#include "property_mosq.h"
|
#include "send_mosq.h"
|
#include "time_mosq.h"
|
#include "util_mosq.h"
|
|
int send__pingreq(struct mosquitto *mosq)
|
{
|
int rc;
|
assert(mosq);
|
#ifdef WITH_BROKER
|
log__printf(NULL, MOSQ_LOG_DEBUG, "Sending PINGREQ to %s", mosq->id);
|
#else
|
log__printf(mosq, MOSQ_LOG_DEBUG, "Client %s sending PINGREQ", mosq->id);
|
#endif
|
rc = send__simple_command(mosq, CMD_PINGREQ);
|
if(rc == MOSQ_ERR_SUCCESS){
|
mosq->ping_t = mosquitto_time();
|
}
|
return rc;
|
}
|
|
int send__pingresp(struct mosquitto *mosq)
|
{
|
#ifdef WITH_BROKER
|
log__printf(NULL, MOSQ_LOG_DEBUG, "Sending PINGRESP to %s", mosq->id);
|
#else
|
log__printf(mosq, MOSQ_LOG_DEBUG, "Client %s sending PINGRESP", mosq->id);
|
#endif
|
return send__simple_command(mosq, CMD_PINGRESP);
|
}
|
|
int send__puback(struct mosquitto *mosq, uint16_t mid, uint8_t reason_code)
|
{
|
#ifdef WITH_BROKER
|
log__printf(NULL, MOSQ_LOG_DEBUG, "Sending PUBACK to %s (m%d, rc%d)", mosq->id, mid, reason_code);
|
#else
|
log__printf(mosq, MOSQ_LOG_DEBUG, "Client %s sending PUBACK (m%d, rc%d)", mosq->id, mid, reason_code);
|
#endif
|
util__increment_receive_quota(mosq);
|
/* We don't use Reason String or User Property yet. */
|
return send__command_with_mid(mosq, CMD_PUBACK, mid, false, reason_code, NULL);
|
}
|
|
int send__pubcomp(struct mosquitto *mosq, uint16_t mid)
|
{
|
#ifdef WITH_BROKER
|
log__printf(NULL, MOSQ_LOG_DEBUG, "Sending PUBCOMP to %s (m%d)", mosq->id, mid);
|
#else
|
log__printf(mosq, MOSQ_LOG_DEBUG, "Client %s sending PUBCOMP (m%d)", mosq->id, mid);
|
#endif
|
util__increment_receive_quota(mosq);
|
/* We don't use Reason String or User Property yet. */
|
return send__command_with_mid(mosq, CMD_PUBCOMP, mid, false, 0, NULL);
|
}
|
|
|
int send__pubrec(struct mosquitto *mosq, uint16_t mid, uint8_t reason_code)
|
{
|
#ifdef WITH_BROKER
|
log__printf(NULL, MOSQ_LOG_DEBUG, "Sending PUBREC to %s (m%d, rc%d)", mosq->id, mid, reason_code);
|
#else
|
log__printf(mosq, MOSQ_LOG_DEBUG, "Client %s sending PUBREC (m%d, rc%d)", mosq->id, mid, reason_code);
|
#endif
|
if(reason_code >= 0x80 && mosq->protocol == mosq_p_mqtt5){
|
util__increment_receive_quota(mosq);
|
}
|
/* We don't use Reason String or User Property yet. */
|
return send__command_with_mid(mosq, CMD_PUBREC, mid, false, reason_code, NULL);
|
}
|
|
int send__pubrel(struct mosquitto *mosq, uint16_t mid)
|
{
|
#ifdef WITH_BROKER
|
log__printf(NULL, MOSQ_LOG_DEBUG, "Sending PUBREL to %s (m%d)", mosq->id, mid);
|
#else
|
log__printf(mosq, MOSQ_LOG_DEBUG, "Client %s sending PUBREL (m%d)", mosq->id, mid);
|
#endif
|
/* We don't use Reason String or User Property yet. */
|
return send__command_with_mid(mosq, CMD_PUBREL|2, mid, false, 0, NULL);
|
}
|
|
/* For PUBACK, PUBCOMP, PUBREC, and PUBREL */
|
int send__command_with_mid(struct mosquitto *mosq, uint8_t command, uint16_t mid, bool dup, uint8_t reason_code, const mosquitto_property *properties)
|
{
|
struct mosquitto__packet *packet = NULL;
|
int rc;
|
int proplen, varbytes;
|
|
assert(mosq);
|
packet = mosquitto__calloc(1, sizeof(struct mosquitto__packet));
|
if(!packet) return MOSQ_ERR_NOMEM;
|
|
packet->command = command;
|
if(dup){
|
packet->command |= 8;
|
}
|
packet->remaining_length = 2;
|
|
if(mosq->protocol == mosq_p_mqtt5){
|
if(reason_code != 0 || properties){
|
packet->remaining_length += 1;
|
}
|
|
if(properties){
|
proplen = property__get_length_all(properties);
|
varbytes = packet__varint_bytes(proplen);
|
packet->remaining_length += varbytes + proplen;
|
}
|
}
|
|
rc = packet__alloc(packet);
|
if(rc){
|
mosquitto__free(packet);
|
return rc;
|
}
|
|
packet__write_uint16(packet, mid);
|
|
if(mosq->protocol == mosq_p_mqtt5){
|
if(reason_code != 0 || properties){
|
packet__write_byte(packet, reason_code);
|
}
|
if(properties){
|
property__write_all(packet, properties, true);
|
}
|
}
|
|
return packet__queue(mosq, packet);
|
}
|
|
/* For DISCONNECT, PINGREQ and PINGRESP */
|
int send__simple_command(struct mosquitto *mosq, uint8_t command)
|
{
|
struct mosquitto__packet *packet = NULL;
|
int rc;
|
|
assert(mosq);
|
packet = mosquitto__calloc(1, sizeof(struct mosquitto__packet));
|
if(!packet) return MOSQ_ERR_NOMEM;
|
|
packet->command = command;
|
packet->remaining_length = 0;
|
|
rc = packet__alloc(packet);
|
if(rc){
|
mosquitto__free(packet);
|
return rc;
|
}
|
|
return packet__queue(mosq, packet);
|
}
|
