Merge branch 'client_init_refactor' into 'master'

Clean client init

See merge request espressif/esp-mqtt!87
This commit is contained in:
David Čermák
2021-01-21 15:57:52 +08:00

View File

@ -36,8 +36,7 @@ static const char *TAG = "MQTT_CLIENT";
ESP_EVENT_DEFINE_BASE(MQTT_EVENTS); ESP_EVENT_DEFINE_BASE(MQTT_EVENTS);
#endif #endif
typedef struct mqtt_state typedef struct mqtt_state {
{
mqtt_connect_info_t *connect_info; mqtt_connect_info_t *connect_info;
uint8_t *in_buffer; uint8_t *in_buffer;
uint8_t *out_buffer; uint8_t *out_buffer;
@ -168,27 +167,27 @@ static esp_err_t esp_mqtt_set_cert_key_data(esp_transport_handle_t ssl, enum esp
switch (ssl_transport_api_id) { switch (ssl_transport_api_id) {
#ifdef MQTT_SUPPORTED_FEATURE_DER_CERTIFICATES #ifdef MQTT_SUPPORTED_FEATURE_DER_CERTIFICATES
case MQTT_SSL_DATA_API_CA_CERT: case MQTT_SSL_DATA_API_CA_CERT:
esp_transport_ssl_set_cert_data_der(ssl, data, len); esp_transport_ssl_set_cert_data_der(ssl, data, len);
break; break;
case MQTT_SSL_DATA_API_CLIENT_CERT: case MQTT_SSL_DATA_API_CLIENT_CERT:
esp_transport_ssl_set_client_cert_data_der(ssl, data, len); esp_transport_ssl_set_client_cert_data_der(ssl, data, len);
break; break;
case MQTT_SSL_DATA_API_CLIENT_KEY: case MQTT_SSL_DATA_API_CLIENT_KEY:
esp_transport_ssl_set_client_key_data_der(ssl, data, len); esp_transport_ssl_set_client_key_data_der(ssl, data, len);
break; break;
#endif #endif
case MQTT_SSL_DATA_API_CA_CERT + MQTT_SSL_DATA_API_MAX: case MQTT_SSL_DATA_API_CA_CERT + MQTT_SSL_DATA_API_MAX:
esp_transport_ssl_set_cert_data(ssl, data, len); esp_transport_ssl_set_cert_data(ssl, data, len);
break; break;
case MQTT_SSL_DATA_API_CLIENT_CERT + MQTT_SSL_DATA_API_MAX: case MQTT_SSL_DATA_API_CLIENT_CERT + MQTT_SSL_DATA_API_MAX:
esp_transport_ssl_set_client_cert_data(ssl, data, len); esp_transport_ssl_set_client_cert_data(ssl, data, len);
break; break;
case MQTT_SSL_DATA_API_CLIENT_KEY + MQTT_SSL_DATA_API_MAX: case MQTT_SSL_DATA_API_CLIENT_KEY + MQTT_SSL_DATA_API_MAX:
esp_transport_ssl_set_client_key_data(ssl, data, len); esp_transport_ssl_set_client_key_data(ssl, data, len);
break; break;
default: default:
return ESP_ERR_INVALID_ARG; return ESP_ERR_INVALID_ARG;
} }
return ESP_OK; return ESP_OK;
} }
@ -218,7 +217,7 @@ static esp_err_t esp_mqtt_set_ssl_transport_properties(esp_transport_list_handle
#endif /* MQTT_SUPPORTED_FEATURE_SECURE_ELEMENT */ #endif /* MQTT_SUPPORTED_FEATURE_SECURE_ELEMENT */
} }
if(cfg->ds_data != NULL) { if (cfg->ds_data != NULL) {
#ifdef MQTT_SUPPORTED_FEATURE_DIGITAL_SIGNATURE #ifdef MQTT_SUPPORTED_FEATURE_DIGITAL_SIGNATURE
#ifdef CONFIG_ESP_TLS_USE_DS_PERIPHERAL #ifdef CONFIG_ESP_TLS_USE_DS_PERIPHERAL
esp_transport_ssl_set_ds_data(ssl, cfg->ds_data); esp_transport_ssl_set_ds_data(ssl, cfg->ds_data);
@ -239,8 +238,8 @@ static esp_err_t esp_mqtt_set_ssl_transport_properties(esp_transport_list_handle
if (cfg->clientkey_password && cfg->clientkey_password_len) { if (cfg->clientkey_password && cfg->clientkey_password_len) {
#if defined(MQTT_SUPPORTED_FEATURE_CLIENT_KEY_PASSWORD) && MQTT_ENABLE_SSL #if defined(MQTT_SUPPORTED_FEATURE_CLIENT_KEY_PASSWORD) && MQTT_ENABLE_SSL
esp_transport_ssl_set_client_key_password(ssl, esp_transport_ssl_set_client_key_password(ssl,
cfg->clientkey_password, cfg->clientkey_password,
cfg->clientkey_password_len); cfg->clientkey_password_len);
#else #else
ESP_LOGE(TAG, "Password protected keys are not available in IDF version %s", IDF_VER); ESP_LOGE(TAG, "Password protected keys are not available in IDF version %s", IDF_VER);
goto esp_mqtt_set_transport_failed; goto esp_mqtt_set_transport_failed;
@ -277,7 +276,7 @@ static esp_err_t esp_mqtt_set_ssl_transport_properties(esp_transport_list_handle
return ESP_OK; return ESP_OK;
esp_mqtt_set_transport_failed: esp_mqtt_set_transport_failed:
return ESP_FAIL; return ESP_FAIL;
} }
#endif // MQTT_ENABLE_SSL #endif // MQTT_ENABLE_SSL
@ -310,80 +309,60 @@ static esp_err_t esp_mqtt_check_cfg_conflict(const mqtt_config_storage_t *cfg, c
return ret; return ret;
} }
static bool set_if_config(char const *const new_config, char **old_config)
{
if (new_config) {
free(*old_config);
*old_config = strdup(new_config);
if(*old_config == NULL) {
return false;
}
}
return true;
}
esp_err_t esp_mqtt_set_config(esp_mqtt_client_handle_t client, const esp_mqtt_client_config_t *config) esp_err_t esp_mqtt_set_config(esp_mqtt_client_handle_t client, const esp_mqtt_client_config_t *config)
{ {
MQTT_API_LOCK(client); MQTT_API_LOCK(client);
//Copy user configurations to client context //Copy user configurations to client context
esp_err_t err = ESP_OK; esp_err_t err = ESP_OK;
mqtt_config_storage_t *cfg; if (!client->config) {
if (client->config) { client->config = calloc(1, sizeof(mqtt_config_storage_t));
cfg = client->config; ESP_MEM_CHECK(TAG, client->config, {
} else {
cfg = calloc(1, sizeof(mqtt_config_storage_t));
ESP_MEM_CHECK(TAG, cfg, {
MQTT_API_UNLOCK(client); MQTT_API_UNLOCK(client);
return ESP_ERR_NO_MEM; return ESP_ERR_NO_MEM;
}); });
client->config = cfg;
}
if (config->task_prio) {
cfg->task_prio = config->task_prio;
} }
if (cfg->task_prio <= 0) { client->config->task_prio = config->task_prio;
cfg->task_prio = MQTT_TASK_PRIORITY; if (client->config->task_prio <= 0) {
client->config->task_prio = MQTT_TASK_PRIORITY;
} }
if (config->task_stack) {
cfg->task_stack = config->task_stack; client->config->task_stack = config->task_stack;
} if (client->config->task_stack <= 0) {
if (cfg->task_stack == 0) { client->config->task_stack = MQTT_TASK_STACK;
cfg->task_stack = MQTT_TASK_STACK;
} }
if (config->port) { if (config->port) {
cfg->port = config->port; client->config->port = config->port;
} }
err = ESP_ERR_NO_MEM; err = ESP_ERR_NO_MEM;
if (config->host) { ESP_MEM_CHECK(TAG, set_if_config(config->host, &client->config->host), goto _mqtt_set_config_failed);
free(cfg->host); ESP_MEM_CHECK(TAG, set_if_config(config->username, &client->connect_info.username), goto _mqtt_set_config_failed);
cfg->host = strdup(config->host); ESP_MEM_CHECK(TAG, set_if_config(config->password, &client->connect_info.password), goto _mqtt_set_config_failed);
ESP_MEM_CHECK(TAG, cfg->host, goto _mqtt_set_config_failed);
}
if (config->username) {
free(client->connect_info.username);
client->connect_info.username = strdup(config->username);
ESP_MEM_CHECK(TAG, client->connect_info.username, goto _mqtt_set_config_failed);
}
if (config->password) {
free(client->connect_info.password);
client->connect_info.password = strdup(config->password);
ESP_MEM_CHECK(TAG, client->connect_info.password, goto _mqtt_set_config_failed);
}
if (config->client_id) { if (config->client_id) {
free(client->connect_info.client_id); ESP_MEM_CHECK(TAG, set_if_config(config->client_id, &client->connect_info.client_id), goto _mqtt_set_config_failed);
client->connect_info.client_id = strdup(config->client_id);
ESP_MEM_CHECK(TAG, client->connect_info.client_id, goto _mqtt_set_config_failed);
} else if (client->connect_info.client_id == NULL) { } else if (client->connect_info.client_id == NULL) {
client->connect_info.client_id = platform_create_id_string(); client->connect_info.client_id = platform_create_id_string();
} }
ESP_MEM_CHECK(TAG, client->connect_info.client_id, goto _mqtt_set_config_failed); ESP_MEM_CHECK(TAG, client->connect_info.client_id, goto _mqtt_set_config_failed);
ESP_LOGD(TAG, "MQTT client_id=%s", client->connect_info.client_id); ESP_LOGD(TAG, "MQTT client_id=%s", client->connect_info.client_id);
if (config->uri) { ESP_MEM_CHECK(TAG, set_if_config(config->uri, &client->config->uri), goto _mqtt_set_config_failed);
free(cfg->uri); ESP_MEM_CHECK(TAG, set_if_config(config->lwt_topic, &client->connect_info.will_topic), goto _mqtt_set_config_failed);
cfg->uri = strdup(config->uri);
ESP_MEM_CHECK(TAG, cfg->uri, goto _mqtt_set_config_failed);
}
if (config->lwt_topic) {
free(client->connect_info.will_topic);
client->connect_info.will_topic = strdup(config->lwt_topic);
ESP_MEM_CHECK(TAG, client->connect_info.will_topic, goto _mqtt_set_config_failed);
}
if (config->lwt_msg_len && config->lwt_msg) { if (config->lwt_msg_len && config->lwt_msg) {
free(client->connect_info.will_message); free(client->connect_info.will_message);
@ -423,7 +402,7 @@ esp_err_t esp_mqtt_set_config(esp_mqtt_client_handle_t client, const esp_mqtt_cl
if (config->protocol_ver) { if (config->protocol_ver) {
client->connect_info.protocol_ver = config->protocol_ver; client->connect_info.protocol_ver = config->protocol_ver;
} }
if (client->connect_info.protocol_ver== MQTT_PROTOCOL_UNDEFINED) { if (client->connect_info.protocol_ver == MQTT_PROTOCOL_UNDEFINED) {
#ifdef MQTT_PROTOCOL_311 #ifdef MQTT_PROTOCOL_311
client->connect_info.protocol_ver = MQTT_PROTOCOL_V_3_1_1; client->connect_info.protocol_ver = MQTT_PROTOCOL_V_3_1_1;
#else #else
@ -431,100 +410,100 @@ esp_err_t esp_mqtt_set_config(esp_mqtt_client_handle_t client, const esp_mqtt_cl
#endif #endif
} }
cfg->network_timeout_ms = config->network_timeout_ms; client->config->network_timeout_ms = config->network_timeout_ms;
if (cfg->network_timeout_ms <= 0) { if (client->config->network_timeout_ms <= 0) {
cfg->network_timeout_ms = MQTT_NETWORK_TIMEOUT_MS; client->config->network_timeout_ms = MQTT_NETWORK_TIMEOUT_MS;
} }
if (config->user_context) { if (config->user_context) {
cfg->user_context = config->user_context; client->config->user_context = config->user_context;
} }
if (config->event_handle) { if (config->event_handle) {
cfg->event_handle = config->event_handle; client->config->event_handle = config->event_handle;
} }
if (config->refresh_connection_after_ms) { if (config->refresh_connection_after_ms) {
cfg->refresh_connection_after_ms = config->refresh_connection_after_ms; client->config->refresh_connection_after_ms = config->refresh_connection_after_ms;
} }
cfg->auto_reconnect = true; client->config->auto_reconnect = true;
if (config->disable_auto_reconnect == cfg->auto_reconnect) { if (config->disable_auto_reconnect == client->config->auto_reconnect) {
cfg->auto_reconnect = !config->disable_auto_reconnect; client->config->auto_reconnect = !config->disable_auto_reconnect;
} }
if (config->reconnect_timeout_ms) { if (config->reconnect_timeout_ms) {
cfg->reconnect_timeout_ms = config->reconnect_timeout_ms; client->config->reconnect_timeout_ms = config->reconnect_timeout_ms;
} else { } else {
cfg->reconnect_timeout_ms = MQTT_RECON_DEFAULT_MS; client->config->reconnect_timeout_ms = MQTT_RECON_DEFAULT_MS;
} }
if (config->alpn_protos) { if (config->alpn_protos) {
for (int i = 0; i < cfg->num_alpn_protos; i++) { for (int i = 0; i < client->config->num_alpn_protos; i++) {
free(cfg->alpn_protos[i]); free(client->config->alpn_protos[i]);
} }
free(cfg->alpn_protos); free(client->config->alpn_protos);
cfg->num_alpn_protos = 0; client->config->num_alpn_protos = 0;
const char **p; const char **p;
for (p = config->alpn_protos; *p != NULL; p++ ) { for (p = config->alpn_protos; *p != NULL; p++ ) {
cfg->num_alpn_protos++; client->config->num_alpn_protos++;
} }
// mbedTLS expects the list to be null-terminated // mbedTLS expects the list to be null-terminated
cfg->alpn_protos = calloc(cfg->num_alpn_protos + 1, sizeof(config->alpn_protos)); client->config->alpn_protos = calloc(client->config->num_alpn_protos + 1, sizeof(config->alpn_protos));
ESP_MEM_CHECK(TAG, cfg->alpn_protos, goto _mqtt_set_config_failed); ESP_MEM_CHECK(TAG, client->config->alpn_protos, goto _mqtt_set_config_failed);
for (int i = 0; i < cfg->num_alpn_protos; i++) { for (int i = 0; i < client->config->num_alpn_protos; i++) {
cfg->alpn_protos[i] = strdup(config->alpn_protos[i]); client->config->alpn_protos[i] = strdup(config->alpn_protos[i]);
ESP_MEM_CHECK(TAG, cfg->alpn_protos[i], goto _mqtt_set_config_failed); ESP_MEM_CHECK(TAG, client->config->alpn_protos[i], goto _mqtt_set_config_failed);
} }
} }
// configure ssl related parameters // configure ssl related parameters
cfg->use_global_ca_store = config->use_global_ca_store; client->config->use_global_ca_store = config->use_global_ca_store;
cfg->cacert_buf = config->cert_pem; client->config->cacert_buf = config->cert_pem;
cfg->cacert_bytes = config->cert_len; client->config->cacert_bytes = config->cert_len;
cfg->clientcert_buf = config->client_cert_pem; client->config->clientcert_buf = config->client_cert_pem;
cfg->clientcert_bytes = config->client_cert_len; client->config->clientcert_bytes = config->client_cert_len;
cfg->clientkey_buf = config->client_key_pem; client->config->clientkey_buf = config->client_key_pem;
cfg->clientkey_bytes = config->client_key_len; client->config->clientkey_bytes = config->client_key_len;
cfg->psk_hint_key = config->psk_hint_key; client->config->psk_hint_key = config->psk_hint_key;
cfg->skip_cert_common_name_check = config->skip_cert_common_name_check; client->config->skip_cert_common_name_check = config->skip_cert_common_name_check;
cfg->use_secure_element = config->use_secure_element; client->config->use_secure_element = config->use_secure_element;
cfg->ds_data = config->ds_data; client->config->ds_data = config->ds_data;
if (config->clientkey_password && config->clientkey_password_len) { if (config->clientkey_password && config->clientkey_password_len) {
cfg->clientkey_password_len = config->clientkey_password_len; client->config->clientkey_password_len = config->clientkey_password_len;
cfg->clientkey_password = malloc(cfg->clientkey_password_len); client->config->clientkey_password = malloc(client->config->clientkey_password_len);
memcpy(cfg->clientkey_password, config->clientkey_password, cfg->clientkey_password_len); memcpy(client->config->clientkey_password, config->clientkey_password, client->config->clientkey_password_len);
} }
if (config->transport) { if (config->transport) {
free(client->config->scheme); free(client->config->scheme);
if (config->transport == MQTT_TRANSPORT_OVER_WS) { if (config->transport == MQTT_TRANSPORT_OVER_WS) {
cfg->scheme = create_string("ws", 2); client->config->scheme = create_string("ws", 2);
ESP_MEM_CHECK(TAG, cfg->scheme, goto _mqtt_set_config_failed); ESP_MEM_CHECK(TAG, client->config->scheme, goto _mqtt_set_config_failed);
} else if (config->transport == MQTT_TRANSPORT_OVER_TCP) { } else if (config->transport == MQTT_TRANSPORT_OVER_TCP) {
cfg->scheme = create_string("mqtt", 4); client->config->scheme = create_string("mqtt", 4);
ESP_MEM_CHECK(TAG, cfg->scheme, goto _mqtt_set_config_failed); ESP_MEM_CHECK(TAG, client->config->scheme, goto _mqtt_set_config_failed);
} else if (config->transport == MQTT_TRANSPORT_OVER_SSL) { } else if (config->transport == MQTT_TRANSPORT_OVER_SSL) {
cfg->scheme = create_string("mqtts", 5); client->config->scheme = create_string("mqtts", 5);
ESP_MEM_CHECK(TAG, cfg->scheme, goto _mqtt_set_config_failed); ESP_MEM_CHECK(TAG, client->config->scheme, goto _mqtt_set_config_failed);
} else if (config->transport == MQTT_TRANSPORT_OVER_WSS) { } else if (config->transport == MQTT_TRANSPORT_OVER_WSS) {
cfg->scheme = create_string("wss", 3); client->config->scheme = create_string("wss", 3);
ESP_MEM_CHECK(TAG, cfg->scheme, goto _mqtt_set_config_failed); ESP_MEM_CHECK(TAG, client->config->scheme, goto _mqtt_set_config_failed);
} }
} }
// Set uri at the end of config to override separately configured uri elements // Set uri at the end of config to override separately configured uri elements
if (config->uri) { if (config->uri) {
if (esp_mqtt_client_set_uri(client, cfg->uri) != ESP_OK) { if (esp_mqtt_client_set_uri(client, client->config->uri) != ESP_OK) {
err = ESP_FAIL; err = ESP_FAIL;
goto _mqtt_set_config_failed; goto _mqtt_set_config_failed;
} }
} }
esp_mqtt_check_cfg_conflict(cfg, config); esp_mqtt_check_cfg_conflict(client->config, config);
MQTT_API_UNLOCK(client); MQTT_API_UNLOCK(client);
@ -660,33 +639,25 @@ static esp_err_t esp_mqtt_abort_connection(esp_mqtt_client_handle_t client)
return ESP_OK; return ESP_OK;
} }
esp_mqtt_client_handle_t esp_mqtt_client_init(const esp_mqtt_client_config_t *config) static bool create_client_data(esp_mqtt_client_handle_t client)
{ {
esp_mqtt_client_handle_t client = calloc(1, sizeof(struct esp_mqtt_client));
ESP_MEM_CHECK(TAG, client, return NULL);
client->event.error_handle = calloc(1, sizeof(esp_mqtt_error_codes_t)); client->event.error_handle = calloc(1, sizeof(esp_mqtt_error_codes_t));
if (!client->event.error_handle) { ESP_MEM_CHECK(TAG, client->event.error_handle, return false)
free(client);
return NULL;
}
client->api_lock = xSemaphoreCreateRecursiveMutex(); client->api_lock = xSemaphoreCreateRecursiveMutex();
if (!client->api_lock) { ESP_MEM_CHECK(TAG, client->api_lock, return false);
free(client->event.error_handle);
free(client);
return NULL;
}
client->transport_list = esp_transport_list_init(); client->transport_list = esp_transport_list_init();
ESP_MEM_CHECK(TAG, client->transport_list, goto _mqtt_init_failed); ESP_MEM_CHECK(TAG, client->transport_list, return false);
esp_transport_handle_t tcp = esp_transport_tcp_init(); esp_transport_handle_t tcp = esp_transport_tcp_init();
ESP_MEM_CHECK(TAG, tcp, goto _mqtt_init_failed); ESP_MEM_CHECK(TAG, tcp, return false);
esp_transport_set_default_port(tcp, MQTT_TCP_DEFAULT_PORT); esp_transport_set_default_port(tcp, MQTT_TCP_DEFAULT_PORT);
esp_transport_list_add(client->transport_list, tcp, "mqtt"); esp_transport_list_add(client->transport_list, tcp, "mqtt");
#if MQTT_ENABLE_WS #if MQTT_ENABLE_WS
esp_transport_handle_t ws = esp_transport_ws_init(tcp); esp_transport_handle_t ws = esp_transport_ws_init(tcp);
ESP_MEM_CHECK(TAG, ws, goto _mqtt_init_failed); ESP_MEM_CHECK(TAG, ws, return false);
esp_transport_set_default_port(ws, MQTT_WS_DEFAULT_PORT); esp_transport_set_default_port(ws, MQTT_WS_DEFAULT_PORT);
#ifdef MQTT_SUPPORTED_FEATURE_WS_SUBPROTOCOL #ifdef MQTT_SUPPORTED_FEATURE_WS_SUBPROTOCOL
esp_transport_ws_set_subprotocol(ws, "mqtt"); esp_transport_ws_set_subprotocol(ws, "mqtt");
@ -696,7 +667,7 @@ esp_mqtt_client_handle_t esp_mqtt_client_init(const esp_mqtt_client_config_t *co
#if MQTT_ENABLE_SSL #if MQTT_ENABLE_SSL
esp_transport_handle_t ssl = esp_transport_ssl_init(); esp_transport_handle_t ssl = esp_transport_ssl_init();
ESP_MEM_CHECK(TAG, ssl, goto _mqtt_init_failed); ESP_MEM_CHECK(TAG, ssl, return false);
esp_transport_set_default_port(ssl, MQTT_SSL_DEFAULT_PORT); esp_transport_set_default_port(ssl, MQTT_SSL_DEFAULT_PORT);
esp_transport_list_add(client->transport_list, ssl, "mqtts"); esp_transport_list_add(client->transport_list, ssl, "mqtts");
@ -704,22 +675,32 @@ esp_mqtt_client_handle_t esp_mqtt_client_init(const esp_mqtt_client_config_t *co
#if MQTT_ENABLE_WSS #if MQTT_ENABLE_WSS
esp_transport_handle_t wss = esp_transport_ws_init(ssl); esp_transport_handle_t wss = esp_transport_ws_init(ssl);
ESP_MEM_CHECK(TAG, wss, goto _mqtt_init_failed); ESP_MEM_CHECK(TAG, wss, return false);
#ifdef MQTT_SUPPORTED_FEATURE_WS_SUBPROTOCOL #ifdef MQTT_SUPPORTED_FEATURE_WS_SUBPROTOCOL
esp_transport_ws_set_subprotocol(wss, "mqtt"); esp_transport_ws_set_subprotocol(wss, "mqtt");
#endif #endif
esp_transport_set_default_port(wss, MQTT_WSS_DEFAULT_PORT); esp_transport_set_default_port(wss, MQTT_WSS_DEFAULT_PORT);
esp_transport_list_add(client->transport_list, wss, "wss"); esp_transport_list_add(client->transport_list, wss, "wss");
#endif #endif
ESP_MEM_CHECK(TAG, client->transport_list, goto _mqtt_init_failed); ESP_MEM_CHECK(TAG, client->transport_list, return false);
return true;
}
esp_mqtt_client_handle_t esp_mqtt_client_init(const esp_mqtt_client_config_t *config)
{
esp_mqtt_client_handle_t client = calloc(1, sizeof(struct esp_mqtt_client));
ESP_MEM_CHECK(TAG, client, return NULL);
if (!create_client_data(client)) {
goto _mqtt_init_failed;
}
if (esp_mqtt_set_config(client, config) != ESP_OK) { if (esp_mqtt_set_config(client, config) != ESP_OK) {
goto _mqtt_init_failed; goto _mqtt_init_failed;
} }
#ifdef MQTT_SUPPORTED_FEATURE_EVENT_LOOP #ifdef MQTT_SUPPORTED_FEATURE_EVENT_LOOP
esp_event_loop_args_t no_task_loop = { esp_event_loop_args_t no_task_loop = {
.queue_size = 1, .queue_size = 1,
.task_name = NULL, .task_name = NULL,
}; };
esp_event_loop_create(&no_task_loop, &client->config->event_loop_handle); esp_event_loop_create(&no_task_loop, &client->config->event_loop_handle);
#endif #endif
@ -749,7 +730,7 @@ esp_mqtt_client_handle_t esp_mqtt_client_init(const esp_mqtt_client_config_t *co
ESP_MEM_CHECK(TAG, client->status_bits, goto _mqtt_init_failed); ESP_MEM_CHECK(TAG, client->status_bits, goto _mqtt_init_failed);
mqtt_msg_init(&client->mqtt_state.mqtt_connection, client->mqtt_state.out_buffer, mqtt_msg_init(&client->mqtt_state.mqtt_connection, client->mqtt_state.out_buffer,
client->mqtt_state.out_buffer_length); client->mqtt_state.out_buffer_length);
return client; return client;
_mqtt_init_failed: _mqtt_init_failed:
@ -762,7 +743,9 @@ esp_err_t esp_mqtt_client_destroy(esp_mqtt_client_handle_t client)
if (client == NULL) { if (client == NULL) {
return ESP_ERR_INVALID_ARG; return ESP_ERR_INVALID_ARG;
} }
esp_mqtt_client_stop(client); if (client->api_lock) {
esp_mqtt_client_stop(client);
}
esp_mqtt_destroy_config(client); esp_mqtt_destroy_config(client);
if (client->transport_list) { if (client->transport_list) {
esp_transport_list_destroy(client->transport_list); esp_transport_list_destroy(client->transport_list);
@ -775,7 +758,9 @@ esp_err_t esp_mqtt_client_destroy(esp_mqtt_client_handle_t client)
} }
free(client->mqtt_state.in_buffer); free(client->mqtt_state.in_buffer);
free(client->mqtt_state.out_buffer); free(client->mqtt_state.out_buffer);
vSemaphoreDelete(client->api_lock); if (client->api_lock) {
vSemaphoreDelete(client->api_lock);
}
free(client->event.error_handle); free(client->event.error_handle);
free(client); free(client);
return ESP_OK; return ESP_OK;
@ -821,7 +806,7 @@ esp_err_t esp_mqtt_client_set_uri(esp_mqtt_client_handle_t client, const char *u
asprintf(&client->config->path, "/?%.*s", puri.field_data[UF_QUERY].len, uri + puri.field_data[UF_QUERY].off); asprintf(&client->config->path, "/?%.*s", puri.field_data[UF_QUERY].len, uri + puri.field_data[UF_QUERY].off);
} else { } else {
asprintf(&client->config->path, "%.*s?%.*s", puri.field_data[UF_PATH].len, uri + puri.field_data[UF_PATH].off, asprintf(&client->config->path, "%.*s?%.*s", puri.field_data[UF_PATH].len, uri + puri.field_data[UF_PATH].off,
puri.field_data[UF_QUERY].len, uri + puri.field_data[UF_QUERY].off); puri.field_data[UF_QUERY].len, uri + puri.field_data[UF_QUERY].off);
} }
ESP_MEM_CHECK(TAG, client->config->path, { ESP_MEM_CHECK(TAG, client->config->path, {
MQTT_API_UNLOCK(client); MQTT_API_UNLOCK(client);
@ -1403,7 +1388,7 @@ static void esp_mqtt_task(void *pv)
} }
if (client->connect_info.keepalive && // connect_info.keepalive=0 means that the keepslive is disabled if (client->connect_info.keepalive && // connect_info.keepalive=0 means that the keepslive is disabled
platform_tick_get_ms() - client->keepalive_tick > client->connect_info.keepalive * 1000 / 2) { platform_tick_get_ms() - client->keepalive_tick > client->connect_info.keepalive * 1000 / 2) {
//No ping resp from last ping => Disconnected //No ping resp from last ping => Disconnected
if (client->wait_for_ping_resp) { if (client->wait_for_ping_resp) {
ESP_LOGE(TAG, "No PING_RESP, disconnected"); ESP_LOGE(TAG, "No PING_RESP, disconnected");
@ -1527,7 +1512,7 @@ esp_err_t esp_mqtt_client_stop(esp_mqtt_client_handle_t client)
} }
// Only send the disconnect message if the client is connected // Only send the disconnect message if the client is connected
if(client->state == MQTT_STATE_CONNECTED) { if (client->state == MQTT_STATE_CONNECTED) {
// Notify the broker we are disconnecting // Notify the broker we are disconnecting
client->mqtt_state.outbound_message = mqtt_msg_disconnect(&client->mqtt_state.mqtt_connection); client->mqtt_state.outbound_message = mqtt_msg_disconnect(&client->mqtt_state.mqtt_connection);
if (client->mqtt_state.outbound_message->length == 0) { if (client->mqtt_state.outbound_message->length == 0) {
@ -1778,7 +1763,7 @@ int esp_mqtt_client_enqueue(esp_mqtt_client_handle_t client, const char *topic,
return ret; return ret;
} }
esp_err_t esp_mqtt_client_register_event(esp_mqtt_client_handle_t client, esp_mqtt_event_id_t event, esp_event_handler_t event_handler, void* event_handler_arg) esp_err_t esp_mqtt_client_register_event(esp_mqtt_client_handle_t client, esp_mqtt_event_id_t event, esp_event_handler_t event_handler, void *event_handler_arg)
{ {
if (client == NULL) { if (client == NULL) {
return ESP_ERR_INVALID_ARG; return ESP_ERR_INVALID_ARG;
@ -1799,18 +1784,18 @@ esp_err_t esp_mqtt_client_register_event(esp_mqtt_client_handle_t client, esp_mq
static void esp_mqtt_client_dispatch_transport_error(esp_mqtt_client_handle_t client) static void esp_mqtt_client_dispatch_transport_error(esp_mqtt_client_handle_t client)
{ {
client->event.event_id = MQTT_EVENT_ERROR; client->event.event_id = MQTT_EVENT_ERROR;
client->event.error_handle->error_type = MQTT_ERROR_TYPE_TCP_TRANSPORT; client->event.error_handle->error_type = MQTT_ERROR_TYPE_TCP_TRANSPORT;
client->event.error_handle->connect_return_code = 0; client->event.error_handle->connect_return_code = 0;
#ifdef MQTT_SUPPORTED_FEATURE_TRANSPORT_ERR_REPORTING #ifdef MQTT_SUPPORTED_FEATURE_TRANSPORT_ERR_REPORTING
client->event.error_handle->esp_tls_last_esp_err = esp_tls_get_and_clear_last_error(esp_transport_get_error_handle(client->transport), client->event.error_handle->esp_tls_last_esp_err = esp_tls_get_and_clear_last_error(esp_transport_get_error_handle(client->transport),
&client->event.error_handle->esp_tls_stack_err, &client->event.error_handle->esp_tls_stack_err,
&client->event.error_handle->esp_tls_cert_verify_flags); &client->event.error_handle->esp_tls_cert_verify_flags);
#ifdef MQTT_SUPPORTED_FEATURE_TRANSPORT_SOCK_ERRNO_REPORTING #ifdef MQTT_SUPPORTED_FEATURE_TRANSPORT_SOCK_ERRNO_REPORTING
client->event.error_handle->esp_transport_sock_errno = esp_transport_get_errno(client->transport); client->event.error_handle->esp_transport_sock_errno = esp_transport_get_errno(client->transport);
#endif #endif
#endif #endif
esp_mqtt_dispatch_event_with_msgid(client); esp_mqtt_dispatch_event_with_msgid(client);
} }
int esp_mqtt_client_get_outbox_size(esp_mqtt_client_handle_t client) int esp_mqtt_client_get_outbox_size(esp_mqtt_client_handle_t client)