2014-05-07 22:27:00 +00:00
|
|
|
/*
|
|
|
|
Copyright (c) 2009-2014 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 <assert.h>
|
|
|
|
|
|
|
|
#include <config.h>
|
|
|
|
|
|
|
|
#include <mosquitto_broker.h>
|
|
|
|
#include <memory_mosq.h>
|
|
|
|
#include <time_mosq.h>
|
|
|
|
|
|
|
|
#include "uthash.h"
|
|
|
|
|
2014-06-23 22:47:48 +00:00
|
|
|
struct mosquitto *mqtt3_context_init(struct mosquitto_db *db, int sock)
|
2014-05-07 22:27:00 +00:00
|
|
|
{
|
|
|
|
struct mosquitto *context;
|
|
|
|
char address[1024];
|
|
|
|
|
|
|
|
context = _mosquitto_calloc(1, sizeof(struct mosquitto));
|
|
|
|
if(!context) return NULL;
|
|
|
|
|
|
|
|
context->state = mosq_cs_new;
|
|
|
|
context->sock = sock;
|
|
|
|
context->last_msg_in = mosquitto_time();
|
|
|
|
context->last_msg_out = mosquitto_time();
|
|
|
|
context->keepalive = 60; /* Default to 60s */
|
|
|
|
context->clean_session = true;
|
|
|
|
context->disconnect_t = 0;
|
|
|
|
context->id = NULL;
|
|
|
|
context->last_mid = 0;
|
|
|
|
context->will = NULL;
|
|
|
|
context->username = NULL;
|
|
|
|
context->password = NULL;
|
|
|
|
context->listener = NULL;
|
|
|
|
context->acl_list = NULL;
|
|
|
|
/* 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;
|
|
|
|
_mosquitto_packet_cleanup(&context->in_packet);
|
|
|
|
context->out_packet = NULL;
|
|
|
|
context->current_out_packet = NULL;
|
|
|
|
|
|
|
|
context->address = NULL;
|
|
|
|
if(sock != -1){
|
|
|
|
if(!_mosquitto_socket_get_address(sock, address, 1024)){
|
|
|
|
context->address = _mosquitto_strdup(address);
|
|
|
|
}
|
|
|
|
if(!context->address){
|
|
|
|
/* getpeername and inet_ntop failed and not a bridge */
|
|
|
|
_mosquitto_free(context);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
context->bridge = NULL;
|
|
|
|
context->msgs = NULL;
|
|
|
|
context->last_msg = NULL;
|
|
|
|
context->msg_count = 0;
|
|
|
|
context->msg_count12 = 0;
|
|
|
|
#ifdef WITH_TLS
|
|
|
|
context->ssl = NULL;
|
|
|
|
#endif
|
|
|
|
|
2014-06-23 22:47:48 +00:00
|
|
|
if(context->sock != INVALID_SOCKET){
|
|
|
|
HASH_ADD(hh_sock, db->contexts_by_sock, sock, sizeof(context->sock), context);
|
|
|
|
}
|
2014-05-07 22:27:00 +00:00
|
|
|
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 mqtt3_context_cleanup(struct mosquitto_db *db, struct mosquitto *context, bool do_free)
|
|
|
|
{
|
|
|
|
struct _mosquitto_packet *packet;
|
|
|
|
struct mosquitto_client_msg *msg, *next;
|
|
|
|
|
|
|
|
if(!context) return;
|
|
|
|
|
|
|
|
if(context->username){
|
|
|
|
_mosquitto_free(context->username);
|
|
|
|
context->username = NULL;
|
|
|
|
}
|
|
|
|
if(context->password){
|
|
|
|
_mosquitto_free(context->password);
|
|
|
|
context->password = NULL;
|
|
|
|
}
|
|
|
|
#ifdef WITH_BRIDGE
|
|
|
|
if(context->bridge){
|
|
|
|
if(context->bridge->username){
|
|
|
|
context->bridge->username = NULL;
|
|
|
|
}
|
|
|
|
if(context->bridge->password){
|
|
|
|
context->bridge->password = NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
2014-06-23 16:57:35 +00:00
|
|
|
_mosquitto_socket_close(db, context);
|
2014-05-07 22:27:00 +00:00
|
|
|
if(context->clean_session && db){
|
|
|
|
mqtt3_subs_clean_session(db, context, &db->subs);
|
|
|
|
mqtt3_db_messages_delete(context);
|
|
|
|
}
|
|
|
|
if(context->address){
|
|
|
|
_mosquitto_free(context->address);
|
|
|
|
context->address = NULL;
|
|
|
|
}
|
|
|
|
if(context->id){
|
|
|
|
assert(db); /* db can only be NULL here if the client hasn't sent a
|
|
|
|
CONNECT and hence wouldn't have an id. */
|
|
|
|
|
2014-06-23 16:57:35 +00:00
|
|
|
HASH_DELETE(hh_id, db->contexts_by_id, context);
|
2014-05-07 22:27:00 +00:00
|
|
|
_mosquitto_free(context->id);
|
|
|
|
context->id = NULL;
|
|
|
|
}
|
|
|
|
_mosquitto_packet_cleanup(&(context->in_packet));
|
|
|
|
_mosquitto_packet_cleanup(context->current_out_packet);
|
|
|
|
context->current_out_packet = NULL;
|
|
|
|
while(context->out_packet){
|
|
|
|
_mosquitto_packet_cleanup(context->out_packet);
|
|
|
|
packet = context->out_packet;
|
|
|
|
context->out_packet = context->out_packet->next;
|
|
|
|
_mosquitto_free(packet);
|
|
|
|
}
|
|
|
|
if(context->will){
|
|
|
|
if(context->will->topic) _mosquitto_free(context->will->topic);
|
|
|
|
if(context->will->payload) _mosquitto_free(context->will->payload);
|
|
|
|
_mosquitto_free(context->will);
|
|
|
|
context->will = NULL;
|
|
|
|
}
|
|
|
|
if(do_free || context->clean_session){
|
|
|
|
msg = context->msgs;
|
|
|
|
while(msg){
|
|
|
|
next = msg->next;
|
|
|
|
msg->store->ref_count--;
|
|
|
|
_mosquitto_free(msg);
|
|
|
|
msg = next;
|
|
|
|
}
|
|
|
|
context->msgs = NULL;
|
|
|
|
context->last_msg = NULL;
|
|
|
|
}
|
|
|
|
if(do_free){
|
|
|
|
_mosquitto_free(context);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void mqtt3_context_disconnect(struct mosquitto_db *db, struct mosquitto *ctxt)
|
|
|
|
{
|
|
|
|
if(ctxt->state != mosq_cs_disconnecting && ctxt->will){
|
|
|
|
/* Unexpected disconnect, queue the client will. */
|
|
|
|
mqtt3_db_messages_easy_queue(db, ctxt, ctxt->will->topic, ctxt->will->qos, ctxt->will->payloadlen, ctxt->will->payload, ctxt->will->retain);
|
|
|
|
}
|
|
|
|
if(ctxt->will){
|
|
|
|
if(ctxt->will->topic) _mosquitto_free(ctxt->will->topic);
|
|
|
|
if(ctxt->will->payload) _mosquitto_free(ctxt->will->payload);
|
|
|
|
_mosquitto_free(ctxt->will);
|
|
|
|
ctxt->will = NULL;
|
|
|
|
}
|
2014-06-04 21:14:16 +00:00
|
|
|
ctxt->disconnect_t = time(NULL);
|
2014-06-02 21:08:40 +00:00
|
|
|
#ifdef WITH_SYS_TREE
|
|
|
|
db->connected_count--;
|
|
|
|
if(!ctxt->clean_session){
|
|
|
|
db->disconnected_count++;
|
|
|
|
}
|
|
|
|
#endif
|
2014-06-23 16:57:35 +00:00
|
|
|
_mosquitto_socket_close(db, ctxt);
|
2014-05-07 22:27:00 +00:00
|
|
|
}
|
|
|
|
|