250 lines
7.7 KiB
C
250 lines
7.7 KiB
C
|
/*
|
||
|
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 <errno.h>
|
||
|
#include <stdio.h>
|
||
|
#include <string.h>
|
||
|
|
||
|
#ifndef WIN32
|
||
|
#include <netdb.h>
|
||
|
#include <sys/socket.h>
|
||
|
#else
|
||
|
#include <winsock2.h>
|
||
|
#include <ws2tcpip.h>
|
||
|
#endif
|
||
|
|
||
|
#include <config.h>
|
||
|
|
||
|
#include <mosquitto.h>
|
||
|
#include <mosquitto_broker.h>
|
||
|
#include <mosquitto_internal.h>
|
||
|
#include <net_mosq.h>
|
||
|
#include <memory_mosq.h>
|
||
|
#include <send_mosq.h>
|
||
|
#include <time_mosq.h>
|
||
|
#include <tls_mosq.h>
|
||
|
#include <util_mosq.h>
|
||
|
#include <will_mosq.h>
|
||
|
|
||
|
#ifdef WITH_BRIDGE
|
||
|
|
||
|
int mqtt3_bridge_new(struct mosquitto_db *db, struct _mqtt3_bridge *bridge)
|
||
|
{
|
||
|
int i;
|
||
|
struct mosquitto *new_context = NULL;
|
||
|
int null_index = -1;
|
||
|
struct mosquitto **tmp_contexts;
|
||
|
char hostname[256];
|
||
|
int len;
|
||
|
char *id;
|
||
|
|
||
|
assert(db);
|
||
|
assert(bridge);
|
||
|
|
||
|
if(bridge->clientid){
|
||
|
id = _mosquitto_strdup(bridge->clientid);
|
||
|
}else{
|
||
|
if(!gethostname(hostname, 256)){
|
||
|
len = strlen(hostname) + strlen(bridge->name) + 2;
|
||
|
id = _mosquitto_malloc(len);
|
||
|
if(!id){
|
||
|
return MOSQ_ERR_NOMEM;
|
||
|
}
|
||
|
snprintf(id, len, "%s.%s", hostname, bridge->name);
|
||
|
}else{
|
||
|
return 1;
|
||
|
}
|
||
|
}
|
||
|
if(!id){
|
||
|
return MOSQ_ERR_NOMEM;
|
||
|
}
|
||
|
|
||
|
/* Search for existing id (possible from persistent db) and also look for a
|
||
|
* gap in the db->contexts[] array in case the id isn't found. */
|
||
|
for(i=0; i<db->context_count; i++){
|
||
|
if(db->contexts[i]){
|
||
|
if(!strcmp(db->contexts[i]->id, id)){
|
||
|
new_context = db->contexts[i];
|
||
|
break;
|
||
|
}
|
||
|
}else if(db->contexts[i] == NULL && null_index == -1){
|
||
|
null_index = i;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
if(!new_context){
|
||
|
/* id wasn't found, so generate a new context */
|
||
|
new_context = mqtt3_context_init(-1);
|
||
|
if(!new_context){
|
||
|
return MOSQ_ERR_NOMEM;
|
||
|
}
|
||
|
if(null_index == -1){
|
||
|
/* There were no gaps in the db->contexts[] array, so need to append. */
|
||
|
db->context_count++;
|
||
|
tmp_contexts = _mosquitto_realloc(db->contexts, sizeof(struct mosquitto*)*db->context_count);
|
||
|
if(tmp_contexts){
|
||
|
db->contexts = tmp_contexts;
|
||
|
db->contexts[db->context_count-1] = new_context;
|
||
|
}else{
|
||
|
_mosquitto_free(new_context);
|
||
|
return MOSQ_ERR_NOMEM;
|
||
|
}
|
||
|
}else{
|
||
|
db->contexts[null_index] = new_context;
|
||
|
}
|
||
|
new_context->id = id;
|
||
|
}else{
|
||
|
/* id was found, so context->id already in memory. */
|
||
|
_mosquitto_free(id);
|
||
|
}
|
||
|
new_context->bridge = bridge;
|
||
|
new_context->is_bridge = true;
|
||
|
|
||
|
new_context->username = new_context->bridge->username;
|
||
|
new_context->password = new_context->bridge->password;
|
||
|
|
||
|
#ifdef WITH_TLS
|
||
|
new_context->tls_cafile = new_context->bridge->tls_cafile;
|
||
|
new_context->tls_capath = new_context->bridge->tls_capath;
|
||
|
new_context->tls_certfile = new_context->bridge->tls_certfile;
|
||
|
new_context->tls_keyfile = new_context->bridge->tls_keyfile;
|
||
|
new_context->tls_version = new_context->bridge->tls_version;
|
||
|
new_context->tls_insecure = new_context->bridge->tls_insecure;
|
||
|
#ifdef REAL_WITH_TLS_PSK
|
||
|
new_context->tls_psk_identity = new_context->bridge->tls_psk_identity;
|
||
|
new_context->tls_psk = new_context->bridge->tls_psk;
|
||
|
#endif
|
||
|
#endif
|
||
|
|
||
|
bridge->try_private_accepted = true;
|
||
|
|
||
|
return mqtt3_bridge_connect(db, new_context);
|
||
|
}
|
||
|
|
||
|
int mqtt3_bridge_connect(struct mosquitto_db *db, struct mosquitto *context)
|
||
|
{
|
||
|
int rc;
|
||
|
int i;
|
||
|
char *notification_topic;
|
||
|
int notification_topic_len;
|
||
|
uint8_t notification_payload;
|
||
|
|
||
|
if(!context || !context->bridge) return MOSQ_ERR_INVAL;
|
||
|
|
||
|
context->state = mosq_cs_new;
|
||
|
context->sock = -1;
|
||
|
context->last_msg_in = mosquitto_time();
|
||
|
context->last_msg_out = mosquitto_time();
|
||
|
context->keepalive = context->bridge->keepalive;
|
||
|
context->clean_session = context->bridge->clean_session;
|
||
|
context->in_packet.payload = NULL;
|
||
|
context->ping_t = 0;
|
||
|
context->bridge->lazy_reconnect = false;
|
||
|
mqtt3_bridge_packet_cleanup(context);
|
||
|
mqtt3_db_message_reconnect_reset(context);
|
||
|
|
||
|
if(context->clean_session){
|
||
|
mqtt3_db_messages_delete(context);
|
||
|
}
|
||
|
|
||
|
/* Delete all local subscriptions even for clean_session==false. We don't
|
||
|
* remove any messages and the next loop carries out the resubscription
|
||
|
* anyway. This means any unwanted subs will be removed.
|
||
|
*/
|
||
|
mqtt3_subs_clean_session(db, context, &db->subs);
|
||
|
|
||
|
for(i=0; i<context->bridge->topic_count; i++){
|
||
|
if(context->bridge->topics[i].direction == bd_out || context->bridge->topics[i].direction == bd_both){
|
||
|
_mosquitto_log_printf(NULL, MOSQ_LOG_DEBUG, "Bridge %s doing local SUBSCRIBE on topic %s", context->id, context->bridge->topics[i].local_topic);
|
||
|
if(mqtt3_sub_add(db, context, context->bridge->topics[i].local_topic, context->bridge->topics[i].qos, &db->subs)) return 1;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if(context->bridge->notifications){
|
||
|
notification_payload = '0';
|
||
|
if(context->bridge->notification_topic){
|
||
|
mqtt3_db_messages_easy_queue(db, context, context->bridge->notification_topic, 1, 1, ¬ification_payload, 1);
|
||
|
rc = _mosquitto_will_set(context, context->bridge->notification_topic, 1, ¬ification_payload, 1, true);
|
||
|
if(rc != MOSQ_ERR_SUCCESS){
|
||
|
return rc;
|
||
|
}
|
||
|
}else{
|
||
|
notification_topic_len = strlen(context->id)+strlen("$SYS/broker/connection//state");
|
||
|
notification_topic = _mosquitto_malloc(sizeof(char)*(notification_topic_len+1));
|
||
|
if(!notification_topic) return MOSQ_ERR_NOMEM;
|
||
|
|
||
|
snprintf(notification_topic, notification_topic_len+1, "$SYS/broker/connection/%s/state", context->id);
|
||
|
mqtt3_db_messages_easy_queue(db, context, notification_topic, 1, 1, ¬ification_payload, 1);
|
||
|
rc = _mosquitto_will_set(context, notification_topic, 1, ¬ification_payload, 1, true);
|
||
|
if(rc != MOSQ_ERR_SUCCESS){
|
||
|
_mosquitto_free(notification_topic);
|
||
|
return rc;
|
||
|
}
|
||
|
_mosquitto_free(notification_topic);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
_mosquitto_log_printf(NULL, MOSQ_LOG_NOTICE, "Connecting bridge %s (%s:%d)", context->bridge->name, context->bridge->addresses[context->bridge->cur_address].address, context->bridge->addresses[context->bridge->cur_address].port);
|
||
|
rc = _mosquitto_socket_connect(context, context->bridge->addresses[context->bridge->cur_address].address, context->bridge->addresses[context->bridge->cur_address].port, NULL, true);
|
||
|
if(rc != MOSQ_ERR_SUCCESS){
|
||
|
if(rc == MOSQ_ERR_TLS){
|
||
|
return rc; /* Error already printed */
|
||
|
}else if(rc == MOSQ_ERR_ERRNO){
|
||
|
_mosquitto_log_printf(NULL, MOSQ_LOG_ERR, "Error creating bridge: %s.", strerror(errno));
|
||
|
}else if(rc == MOSQ_ERR_EAI){
|
||
|
_mosquitto_log_printf(NULL, MOSQ_LOG_ERR, "Error creating bridge: %s.", gai_strerror(errno));
|
||
|
}
|
||
|
|
||
|
return rc;
|
||
|
}
|
||
|
|
||
|
rc = _mosquitto_send_connect(context, context->keepalive, context->clean_session);
|
||
|
if(rc == MOSQ_ERR_SUCCESS){
|
||
|
return MOSQ_ERR_SUCCESS;
|
||
|
}else if(rc == MOSQ_ERR_ERRNO && errno == ENOTCONN){
|
||
|
return MOSQ_ERR_SUCCESS;
|
||
|
}else{
|
||
|
if(rc == MOSQ_ERR_TLS){
|
||
|
return rc; /* Error already printed */
|
||
|
}else if(rc == MOSQ_ERR_ERRNO){
|
||
|
_mosquitto_log_printf(NULL, MOSQ_LOG_ERR, "Error creating bridge: %s.", strerror(errno));
|
||
|
}else if(rc == MOSQ_ERR_EAI){
|
||
|
_mosquitto_log_printf(NULL, MOSQ_LOG_ERR, "Error creating bridge: %s.", gai_strerror(errno));
|
||
|
}
|
||
|
_mosquitto_socket_close(context);
|
||
|
return rc;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void mqtt3_bridge_packet_cleanup(struct mosquitto *context)
|
||
|
{
|
||
|
struct _mosquitto_packet *packet;
|
||
|
if(!context) return;
|
||
|
|
||
|
_mosquitto_packet_cleanup(context->current_out_packet);
|
||
|
while(context->out_packet){
|
||
|
_mosquitto_packet_cleanup(context->out_packet);
|
||
|
packet = context->out_packet;
|
||
|
context->out_packet = context->out_packet->next;
|
||
|
_mosquitto_free(packet);
|
||
|
}
|
||
|
|
||
|
_mosquitto_packet_cleanup(&(context->in_packet));
|
||
|
}
|
||
|
|
||
|
#endif
|