You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
329 lines
8.4 KiB
C
329 lines
8.4 KiB
C
/*
|
|
Copyright (c) 2009-2021 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 2.0
|
|
and Eclipse Distribution License v1.0 which accompany this distribution.
|
|
|
|
The Eclipse Public License is available at
|
|
https://www.eclipse.org/legal/epl-2.0/
|
|
and the Eclipse Distribution License is available at
|
|
http://www.eclipse.org/org/documents/edl-v10.php.
|
|
|
|
SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause
|
|
|
|
Contributors:
|
|
Roger Light - initial implementation and documentation.
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <assert.h>
|
|
#include <time.h>
|
|
#ifdef __APPLE__
|
|
#include <sys/socket.h>
|
|
#endif
|
|
|
|
#include "mosquitto_broker_internal.h"
|
|
#include "alias_mosq.h"
|
|
#include "memory_mosq.h"
|
|
#include "packet_mosq.h"
|
|
#include "property_mosq.h"
|
|
#include "sys_tree.h"
|
|
#include "time_mosq.h"
|
|
#include "util_mosq.h"
|
|
#include "will_mosq.h"
|
|
|
|
#include "uthash.h"
|
|
|
|
int context__init_sock(struct mosquitto *context, mosq_sock_t sock)
|
|
{
|
|
char address[1024];
|
|
|
|
context->sock = sock;
|
|
|
|
if((int)context->sock >= 0){
|
|
if(!net__socket_get_address(context->sock,
|
|
address, sizeof(address),
|
|
&context->remote_port)){
|
|
|
|
context->address = mosquitto__strdup(address);
|
|
}
|
|
if(!context->address){
|
|
/* getpeername and inet_ntop failed and not a bridge */
|
|
return MOSQ_ERR_NOMEM;
|
|
}
|
|
HASH_ADD(hh_sock, db.contexts_by_sock, sock, sizeof(context->sock), context);
|
|
}
|
|
return MOSQ_ERR_SUCCESS;
|
|
}
|
|
|
|
struct mosquitto *context__init(void)
|
|
{
|
|
struct mosquitto *context;
|
|
|
|
context = mosquitto__calloc(1, sizeof(struct mosquitto));
|
|
if(!context) return NULL;
|
|
|
|
#if defined(WITH_EPOLL) || defined(WITH_KQUEUE)
|
|
context->ident = id_client;
|
|
#else
|
|
context->pollfd_index = -1;
|
|
#endif
|
|
mosquitto__set_state(context, mosq_cs_new);
|
|
context->sock = INVALID_SOCKET;
|
|
context->last_msg_in = db.now_s;
|
|
context->next_msg_out = db.now_s + 20;
|
|
context->keepalive = 20; /* Default to 20s */
|
|
context->clean_start = true;
|
|
context->id = NULL;
|
|
context->last_mid = 0;
|
|
context->will = NULL;
|
|
context->username = NULL;
|
|
context->password = NULL;
|
|
context->listener = NULL;
|
|
context->acl_list = NULL;
|
|
context->retain_available = true;
|
|
#if defined(WITH_WEBSOCKETS) && WITH_WEBSOCKETS == WS_IS_BUILTIN
|
|
memset(&context->wsd, 0, sizeof(context->wsd));
|
|
context->wsd.opcode = UINT8_MAX;
|
|
context->wsd.mask = UINT8_MAX;
|
|
context->wsd.disconnect_reason = 0xE8;
|
|
#endif
|
|
|
|
/* is_bridge records whether this client is a bridge or not. This could be
|
|
* done by looking at context->bridge for bridges that we create ourself,
|
|
* but incoming bridges need some other way of being recorded. */
|
|
context->is_bridge = false;
|
|
|
|
context->in_packet.payload = NULL;
|
|
packet__cleanup(&context->in_packet);
|
|
context->out_packet = NULL;
|
|
context->out_packet_count = 0;
|
|
context->out_packet_bytes = 0;
|
|
|
|
context->address = NULL;
|
|
context->bridge = NULL;
|
|
context->msgs_in.inflight_maximum = db.config->max_inflight_messages;
|
|
context->msgs_out.inflight_maximum = db.config->max_inflight_messages;
|
|
context->msgs_in.inflight_quota = db.config->max_inflight_messages;
|
|
context->msgs_out.inflight_quota = db.config->max_inflight_messages;
|
|
context->max_qos = 2;
|
|
#ifdef WITH_TLS
|
|
context->ssl = NULL;
|
|
#endif
|
|
|
|
return context;
|
|
}
|
|
|
|
/*
|
|
* This will result in any outgoing packets going unsent. If we're disconnected
|
|
* forcefully then it is usually an error condition and shouldn't be a problem,
|
|
* but it will mean that CONNACK messages will never get sent for bad protocol
|
|
* versions for example.
|
|
*/
|
|
void context__cleanup(struct mosquitto *context, bool force_free)
|
|
{
|
|
struct mosquitto__packet *packet;
|
|
|
|
if(!context) return;
|
|
|
|
if(force_free){
|
|
context->clean_start = true;
|
|
}
|
|
|
|
#ifdef WITH_BRIDGE
|
|
if(context->bridge){
|
|
bridge__cleanup(context);
|
|
}
|
|
#endif
|
|
|
|
alias__free_all(context);
|
|
keepalive__remove(context);
|
|
|
|
mosquitto__FREE(context->auth_method);
|
|
mosquitto__FREE(context->username);
|
|
mosquitto__FREE(context->password);
|
|
|
|
net__socket_close(context);
|
|
if(force_free){
|
|
sub__clean_session(context);
|
|
}
|
|
db__messages_delete(context, force_free);
|
|
|
|
mosquitto__FREE(context->address);
|
|
|
|
context__send_will(context);
|
|
|
|
if(context->id){
|
|
context__remove_from_by_id(context);
|
|
mosquitto__FREE(context->id);
|
|
}
|
|
packet__cleanup(&(context->in_packet));
|
|
while(context->out_packet){
|
|
packet = context->out_packet;
|
|
context->out_packet = context->out_packet->next;
|
|
mosquitto__FREE(packet);
|
|
}
|
|
G_OUT_PACKET_COUNT_DEC(context->out_packet_count);
|
|
G_OUT_PACKET_BYTES_DEC(context->out_packet_bytes);
|
|
context->out_packet_count = 0;
|
|
context->out_packet_bytes = 0;
|
|
#if defined(WITH_BROKER) && defined(__GLIBC__) && defined(WITH_ADNS)
|
|
if(context->adns){
|
|
gai_cancel(context->adns);
|
|
struct addrinfo *ar_request = (struct addrinfo *)context->adns->ar_request;
|
|
mosquitto__FREE(ar_request);
|
|
mosquitto__FREE(context->adns);
|
|
}
|
|
#endif
|
|
|
|
#if defined(WITH_WEBSOCKETS) && WITH_WEBSOCKETS == WS_IS_BUILTIN
|
|
mosquitto__FREE(context->http_request);
|
|
#endif
|
|
if(force_free){
|
|
mosquitto__FREE(context);
|
|
}
|
|
}
|
|
|
|
|
|
void context__send_will(struct mosquitto *ctxt)
|
|
{
|
|
if(ctxt->state != mosq_cs_disconnecting && ctxt->will){
|
|
if(ctxt->will_delay_interval > 0){
|
|
will_delay__add(ctxt);
|
|
return;
|
|
}
|
|
|
|
if(mosquitto_acl_check(ctxt,
|
|
ctxt->will->msg.topic,
|
|
(uint32_t)ctxt->will->msg.payloadlen,
|
|
ctxt->will->msg.payload,
|
|
(uint8_t)ctxt->will->msg.qos,
|
|
ctxt->will->msg.retain,
|
|
MOSQ_ACL_WRITE) == MOSQ_ERR_SUCCESS){
|
|
|
|
/* Unexpected disconnect, queue the client will. */
|
|
db__messages_easy_queue(ctxt,
|
|
ctxt->will->msg.topic,
|
|
(uint8_t)ctxt->will->msg.qos,
|
|
(uint32_t)ctxt->will->msg.payloadlen,
|
|
ctxt->will->msg.payload,
|
|
ctxt->will->msg.retain,
|
|
ctxt->will->expiry_interval,
|
|
&ctxt->will->properties);
|
|
}
|
|
}
|
|
will__clear(ctxt);
|
|
}
|
|
|
|
|
|
void context__disconnect(struct mosquitto *context)
|
|
{
|
|
if(mosquitto__get_state(context) == mosq_cs_disconnected){
|
|
return;
|
|
}
|
|
|
|
#if defined(WITH_WEBSOCKETS) && WITH_WEBSOCKETS == WS_IS_BUILTIN
|
|
if(context->transport == mosq_t_ws){
|
|
uint8_t buf[4] = {0x88, 0x02, 0x03, context->wsd.disconnect_reason};
|
|
/* Send the disconnect reason, but don't care if it fails */
|
|
if(send(context->sock, buf, 4, 0)){};
|
|
}
|
|
#endif
|
|
plugin__handle_disconnect(context, -1);
|
|
|
|
context__send_will(context);
|
|
net__socket_close(context);
|
|
if(context->session_expiry_interval == 0){
|
|
plugin_persist__handle_client_delete(context);
|
|
/* Client session is due to be expired now */
|
|
#ifdef WITH_BRIDGE
|
|
if(context->bridge == NULL)
|
|
#endif
|
|
{
|
|
if(context->will_delay_interval == 0){
|
|
/* This will be done later, after the will is published for delay>0. */
|
|
context__add_to_disused(context);
|
|
}
|
|
}
|
|
}else{
|
|
session_expiry__add(context);
|
|
}
|
|
keepalive__remove(context);
|
|
mosquitto__set_state(context, mosq_cs_disconnected);
|
|
alias__free_all(context);
|
|
}
|
|
|
|
void context__add_to_disused(struct mosquitto *context)
|
|
{
|
|
if(context->state == mosq_cs_disused) return;
|
|
|
|
mosquitto__set_state(context, mosq_cs_disused);
|
|
|
|
context__remove_from_by_id(context);
|
|
|
|
context->for_free_next = db.ll_for_free;
|
|
db.ll_for_free = context;
|
|
}
|
|
|
|
void context__free_disused(void)
|
|
{
|
|
struct mosquitto *context, *next;
|
|
#if defined(WITH_WEBSOCKETS) && WITH_WEBSOCKETS == WS_IS_LWS
|
|
struct mosquitto *last = NULL;
|
|
#endif
|
|
|
|
context = db.ll_for_free;
|
|
db.ll_for_free = NULL;
|
|
while(context){
|
|
#if defined(WITH_WEBSOCKETS) && WITH_WEBSOCKETS == WS_IS_LWS
|
|
if(context->wsi){
|
|
/* Don't delete yet, lws hasn't finished with it */
|
|
if(last){
|
|
last->for_free_next = context;
|
|
}else{
|
|
db.ll_for_free = context;
|
|
}
|
|
next = context->for_free_next;
|
|
context->for_free_next = NULL;
|
|
last = context;
|
|
context = next;
|
|
}else
|
|
#endif
|
|
{
|
|
next = context->for_free_next;
|
|
context__cleanup(context, true);
|
|
context = next;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void context__add_to_by_id(struct mosquitto *context)
|
|
{
|
|
if(context->in_by_id == false){
|
|
context->in_by_id = true;
|
|
HASH_ADD_KEYPTR(hh_id, db.contexts_by_id, context->id, strlen(context->id), context);
|
|
}
|
|
}
|
|
|
|
|
|
void context__remove_from_by_id(struct mosquitto *context)
|
|
{
|
|
struct mosquitto *context_found;
|
|
|
|
if(context->in_by_id == true && context->id){
|
|
HASH_FIND(hh_id, db.contexts_by_id_delayed_auth, context->id, strlen(context->id), context_found);
|
|
if(context_found){
|
|
HASH_DELETE(hh_id, db.contexts_by_id_delayed_auth, context_found);
|
|
}
|
|
|
|
HASH_FIND(hh_id, db.contexts_by_id, context->id, strlen(context->id), context_found);
|
|
if(context_found){
|
|
HASH_DELETE(hh_id, db.contexts_by_id, context_found);
|
|
}
|
|
context->in_by_id = false;
|
|
}
|
|
}
|