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.
144 lines
3.9 KiB
C
144 lines
3.9 KiB
C
/*
|
|
Copyright (c) 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 <math.h>
|
|
#include <stdio.h>
|
|
#include <utlist.h>
|
|
|
|
#include "mosquitto_broker_internal.h"
|
|
#include "memory_mosq.h"
|
|
#include "sys_tree.h"
|
|
#include "time_mosq.h"
|
|
|
|
static struct session_expiry_list *expiry_list = NULL;
|
|
static time_t last_check = 0;
|
|
|
|
|
|
static int session_expiry__cmp(struct session_expiry_list *i1, struct session_expiry_list *i2)
|
|
{
|
|
if(i1->context->session_expiry_time == i2->context->session_expiry_time){
|
|
return 0;
|
|
}else if(i1->context->session_expiry_time > i2->context->session_expiry_time){
|
|
return 1;
|
|
}else{
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
|
|
int session_expiry__add(struct mosquitto_db *db, struct mosquitto *context)
|
|
{
|
|
struct session_expiry_list *item;
|
|
|
|
if(db->config->persistent_client_expiration == 0){
|
|
if(context->session_expiry_interval == UINT32_MAX){
|
|
/* There isn't a global expiry set, and the client has asked to
|
|
* never expire, so we don't add it to the list. */
|
|
return MOSQ_ERR_SUCCESS;
|
|
}
|
|
}
|
|
|
|
item = mosquitto__calloc(1, sizeof(struct session_expiry_list));
|
|
if(!item) return MOSQ_ERR_NOMEM;
|
|
|
|
item->context = context;
|
|
item->context->session_expiry_time = time(NULL);
|
|
|
|
if(db->config->persistent_client_expiration == 0){
|
|
/* No global expiry, so use the client expiration interval */
|
|
item->context->session_expiry_time += item->context->session_expiry_interval;
|
|
}else{
|
|
/* We have a global expiry interval */
|
|
if(db->config->persistent_client_expiration < item->context->session_expiry_interval){
|
|
/* The client expiry is longer than the global expiry, so use the global */
|
|
item->context->session_expiry_time += db->config->persistent_client_expiration;
|
|
}else{
|
|
/* The global expiry is longer than the client expiry, so use the client */
|
|
item->context->session_expiry_time += item->context->session_expiry_interval;
|
|
}
|
|
}
|
|
context->expiry_list_item = item;
|
|
|
|
DL_INSERT_INORDER(expiry_list, item, session_expiry__cmp);
|
|
|
|
return MOSQ_ERR_SUCCESS;
|
|
}
|
|
|
|
|
|
void session_expiry__remove(struct mosquitto *context)
|
|
{
|
|
if(context->expiry_list_item){
|
|
DL_DELETE(expiry_list, context->expiry_list_item);
|
|
mosquitto__free(context->expiry_list_item);
|
|
context->expiry_list_item = NULL;
|
|
}
|
|
}
|
|
|
|
|
|
/* Call on broker shutdown only */
|
|
void session_expiry__remove_all(struct mosquitto_db *db)
|
|
{
|
|
struct session_expiry_list *item, *tmp;
|
|
struct mosquitto *context;
|
|
|
|
DL_FOREACH_SAFE(expiry_list, item, tmp){
|
|
context = item->context;
|
|
session_expiry__remove(context);
|
|
context->session_expiry_interval = 0;
|
|
context->will_delay_interval = 0;
|
|
will_delay__remove(context);
|
|
context__disconnect(db, context);
|
|
}
|
|
|
|
}
|
|
|
|
void session_expiry__check(struct mosquitto_db *db, time_t now)
|
|
{
|
|
struct session_expiry_list *item, *tmp;
|
|
struct mosquitto *context;
|
|
|
|
if(now <= last_check) return;
|
|
|
|
last_check = now;
|
|
|
|
DL_FOREACH_SAFE(expiry_list, item, tmp){
|
|
if(item->context->session_expiry_time < now){
|
|
|
|
context = item->context;
|
|
session_expiry__remove(context);
|
|
|
|
if(context->id){
|
|
log__printf(NULL, MOSQ_LOG_NOTICE, "Expiring client %s due to timeout.", context->id);
|
|
}
|
|
G_CLIENTS_EXPIRED_INC();
|
|
|
|
/* Session has now expired, so clear interval */
|
|
context->session_expiry_interval = 0;
|
|
/* Session has expired, so will delay should be cleared. */
|
|
context->will_delay_interval = 0;
|
|
will_delay__remove(context);
|
|
context__send_will(db, context);
|
|
context__add_to_disused(db, context);
|
|
}else{
|
|
return;
|
|
}
|
|
}
|
|
|
|
}
|
|
|