/*************************************************************************** * * Copyright 2015-2019 BES. * All rights reserved. All unpublished rights reserved. * * No part of this work may be used or reproduced in any form or by any * means, or stored in a database or retrieval system, without prior written * permission of BES. * * Use of this work is governed by a license granted by BES. * This work contains confidential and proprietary information of * BES. which is protected by copyright, trade secret, * trademark and other intellectual property rights. * ****************************************************************************/ /** **************************************************************************************** * @addtogroup OTATASK * @{ **************************************************************************************** */ /* * INCLUDE FILES **************************************************************************************** */ #include "rwip_config.h" #if (BLE_OTA) #include "gap.h" #include "gattc_task.h" #include "gapc_task.h" #include "attm.h" #include "ota.h" #include "ota_task.h" #include "prf_utils.h" #include "ke_mem.h" #include "co_utils.h" /* * LOCAL FUNCTIONS DEFINITIONS **************************************************************************************** */ static int gapc_disconnect_ind_handler(ke_msg_id_t const msgid, struct gapc_disconnect_ind const *param, ke_task_id_t const dest_id, ke_task_id_t const src_id) { uint8_t conidx = KE_IDX_GET(src_id); struct ota_env_tag *ota_env = PRF_ENV_GET(OTA, ota); ota_env->isNotificationEnabled[conidx] = false; return KE_MSG_CONSUMED; } void app_ota_send_rx_cfm(uint8_t conidx) { struct ota_env_tag *ota_env = PRF_ENV_GET(OTA, ota); struct gattc_write_cfm *cfm = KE_MSG_ALLOC(GATTC_WRITE_CFM, KE_BUILD_ID(TASK_GATTC, conidx), prf_src_task_get(&ota_env->prf_env, conidx), gattc_write_cfm); cfm->handle = ota_env->shdl + OTA_IDX_VAL; cfm->status = GAP_ERR_NO_ERROR; ke_msg_send(cfm); } /** **************************************************************************************** * @brief Handles reception of the @ref GL2C_CODE_ATT_WR_CMD_IND message. * The handler compares the new values with current ones and notifies them if they changed. * @param[in] msgid Id of the message received (probably unused). * @param[in] param Pointer to the parameters of the message. * @param[in] dest_id ID of the receiving task instance (probably unused). * @param[in] src_id ID of the sending task instance. * @return If the message was consumed or not. **************************************************************************************** */ __STATIC int gattc_write_req_ind_handler(ke_msg_id_t const msgid, struct gattc_write_req_ind const *param, ke_task_id_t const dest_id, ke_task_id_t const src_id) { // Get the address of the environment struct ota_env_tag *ota_env = PRF_ENV_GET(OTA, ota); uint8_t conidx = KE_IDX_GET(src_id); uint8_t status = GAP_ERR_NO_ERROR; BLE_GATT_DBG("gattc_write_req_ind_handler ota_env 0x%x write handle %d shdl %d", ota_env, param->handle, ota_env->shdl); #if 0 //reply write-response so fast which results in rx queue oveflow //Send write response struct gattc_write_cfm *cfm = KE_MSG_ALLOC( GATTC_WRITE_CFM, src_id, dest_id, gattc_write_cfm); cfm->handle = param->handle; cfm->status = status; ke_msg_send(cfm); #endif if (ota_env != NULL) { if (param->handle == (ota_env->shdl + OTA_IDX_NTF_CFG)) { uint16_t value = 0x0000; struct gattc_write_cfm *cfm = KE_MSG_ALLOC(GATTC_WRITE_CFM, src_id, dest_id, gattc_write_cfm); cfm->handle = param->handle; cfm->status = status; ke_msg_send(cfm); //Extract value before check memcpy(&value, &(param->value), sizeof(uint16_t)); if (value == PRF_CLI_STOP_NTFIND) { ota_env->isNotificationEnabled[conidx] = false; } else if (value == PRF_CLI_START_NTF) { ota_env->isNotificationEnabled[conidx] = true; } else { status = PRF_APP_ERROR; } if (status == GAP_ERR_NO_ERROR) { //Inform APP of TX ccc change struct ble_ota_tx_notif_config_t * ind = KE_MSG_ALLOC(OTA_CCC_CHANGED, prf_dst_task_get(&ota_env->prf_env, conidx), prf_src_task_get(&ota_env->prf_env, conidx), ble_ota_tx_notif_config_t); ind->isNotificationEnabled = ota_env->isNotificationEnabled[conidx]; ke_msg_send(ind); } } else if (param->handle == (ota_env->shdl + OTA_IDX_VAL)) { TRACE(1,"[OTA] ble cmd %x",param->value[0]); //inform APP of data struct ble_ota_rx_data_ind_t * ind = KE_MSG_ALLOC_DYN(OTA_DATA_RECEIVED, prf_dst_task_get(&ota_env->prf_env, conidx), prf_src_task_get(&ota_env->prf_env, conidx), ble_ota_rx_data_ind_t, sizeof(struct ble_ota_rx_data_ind_t) + param->length); ind->length = param->length; memcpy((uint8_t *)(ind->data), &(param->value), param->length); ke_msg_send(ind); return KE_MSG_CONSUMED; } else { status = PRF_APP_ERROR; } } return (KE_MSG_CONSUMED); } /** **************************************************************************************** * @brief Handles @ref GATTC_CMP_EVT for GATTC_NOTIFY message meaning that Measurement * notification has been correctly sent to peer device (but not confirmed by peer device). * * * @param[in] msgid Id of the message received. * @param[in] param Pointer to the parameters of the message. * @param[in] dest_id ID of the receiving task instance * @param[in] src_id ID of the sending task instance. * @return If the message was consumed or not. **************************************************************************************** */ __STATIC int gattc_cmp_evt_handler(ke_msg_id_t const msgid, struct gattc_cmp_evt const *param, ke_task_id_t const dest_id, ke_task_id_t const src_id) { // Get the address of the environment struct ota_env_tag *ota_env = PRF_ENV_GET(OTA, ota); uint8_t conidx = KE_IDX_GET(dest_id); // notification or write request has been sent out if ((GATTC_NOTIFY == param->operation) || (GATTC_INDICATE == param->operation)) { struct ble_ota_tx_sent_ind_t * ind = KE_MSG_ALLOC(OTA_TX_DATA_SENT, prf_dst_task_get(&ota_env->prf_env, conidx), prf_src_task_get(&ota_env->prf_env, conidx), ble_ota_tx_sent_ind_t); ind->status = param->status; ke_msg_send(ind); } ke_state_set(dest_id, OTA_IDLE); return (KE_MSG_CONSUMED); } /** **************************************************************************************** * @brief Handles reception of the read request from peer device * * @param[in] msgid Id of the message received (probably unused). * @param[in] param Pointer to the parameters of the message. * @param[in] dest_id ID of the receiving task instance (probably unused). * @param[in] src_id ID of the sending task instance. * @return If the message was consumed or not. **************************************************************************************** */ __STATIC int gattc_read_req_ind_handler(ke_msg_id_t const msgid, struct gattc_read_req_ind const *param, ke_task_id_t const dest_id, ke_task_id_t const src_id) { // Get the address of the environment struct ota_env_tag *ota_env = PRF_ENV_GET(OTA, ota); struct gattc_read_cfm* cfm; uint8_t status = GAP_ERR_NO_ERROR; uint8_t conidx = KE_IDX_GET(src_id); BLE_GATT_DBG("gattc_read_req_ind_handler read handle %d shdl %d", param->handle, ota_env->shdl); if (param->handle == (ota_env->shdl + OTA_IDX_NTF_CFG)) { uint16_t notify_ccc; cfm = KE_MSG_ALLOC_DYN(GATTC_READ_CFM, src_id, dest_id, gattc_read_cfm, sizeof(notify_ccc)); if (ota_env->isNotificationEnabled[conidx]) { notify_ccc = 1; } else { notify_ccc = 0; } cfm->length = sizeof(notify_ccc); memcpy(cfm->value, (uint8_t *)¬ify_ccc, cfm->length); } else { cfm = KE_MSG_ALLOC(GATTC_READ_CFM, src_id, dest_id, gattc_read_cfm); cfm->length = 0; status = ATT_ERR_REQUEST_NOT_SUPPORTED; } cfm->handle = param->handle; cfm->status = status; ke_msg_send(cfm); ke_state_set(dest_id, OTA_IDLE); return (KE_MSG_CONSUMED); } static void send_notifiction(uint8_t conidx, const uint8_t* ptrData, uint32_t length) { uint16_t handle_offset = 0xFFFF; struct ota_env_tag *ota_env = PRF_ENV_GET(OTA, ota); if (ota_env->isNotificationEnabled[conidx]) { handle_offset = OTA_IDX_VAL; } TRACE(1,"Send ota notificationto handle offset %d:", handle_offset); DUMP8("%02x ", ptrData, length); if (0xFFFF != handle_offset) { // Allocate the GATT notification message struct gattc_send_evt_cmd *report_ntf = KE_MSG_ALLOC_DYN(GATTC_SEND_EVT_CMD, KE_BUILD_ID(TASK_GATTC, conidx), prf_src_task_get(&ota_env->prf_env, conidx), gattc_send_evt_cmd, length); // Fill in the parameter structure report_ntf->operation = GATTC_NOTIFY; report_ntf->handle = ota_env->shdl + handle_offset; // pack measured value in database report_ntf->length = length; memcpy(report_ntf->value, ptrData, length); // send notification to peer device ke_msg_send(report_ntf); } } static void send_indication(uint8_t conidx, const uint8_t* ptrData, uint32_t length) { uint16_t handle_offset = 0xFFFF; struct ota_env_tag *ota_env = PRF_ENV_GET(OTA, ota); if (ota_env->isNotificationEnabled[conidx]) { handle_offset = OTA_IDX_VAL; } if (0xFFFF != handle_offset) { // Allocate the GATT notification message struct gattc_send_evt_cmd *report_ntf = KE_MSG_ALLOC_DYN(GATTC_SEND_EVT_CMD, KE_BUILD_ID(TASK_GATTC, conidx), prf_src_task_get(&ota_env->prf_env, conidx), gattc_send_evt_cmd, length); // Fill in the parameter structure report_ntf->operation = GATTC_INDICATE; report_ntf->handle = ota_env->shdl + handle_offset; // pack measured value in database report_ntf->length = length; memcpy(report_ntf->value, ptrData, length); // send notification to peer device ke_msg_send(report_ntf); } } __STATIC int send_notification_handler(ke_msg_id_t const msgid, struct ble_ota_send_data_req_t const *param, ke_task_id_t const dest_id, ke_task_id_t const src_id) { send_notifiction(param->connecionIndex, param->value, param->length); return (KE_MSG_CONSUMED); } __STATIC int send_indication_handler(ke_msg_id_t const msgid, struct ble_ota_send_data_req_t const *param, ke_task_id_t const dest_id, ke_task_id_t const src_id) { send_indication(param->connecionIndex, param->value, param->length); return (KE_MSG_CONSUMED); } /** **************************************************************************************** * @brief Handles reception of the attribute info request message. * * @param[in] msgid Id of the message received (probably unused). * @param[in] param Pointer to the parameters of the message. * @param[in] dest_id ID of the receiving task instance (probably unused). * @param[in] src_id ID of the sending task instance. * @return If the message was consumed or not. **************************************************************************************** */ static int gattc_att_info_req_ind_handler(ke_msg_id_t const msgid, struct gattc_att_info_req_ind *param, ke_task_id_t const dest_id, ke_task_id_t const src_id) { // Get the address of the environment struct gattc_att_info_cfm * cfm; //Send write response cfm = KE_MSG_ALLOC(GATTC_ATT_INFO_CFM, src_id, dest_id, gattc_att_info_cfm); cfm->handle = param->handle; struct ota_env_tag *ota_env = PRF_ENV_GET(OTA, ota); if (param->handle == (ota_env->shdl + OTA_IDX_NTF_CFG)) { // CCC attribute length = 2 cfm->length = 2; cfm->status = GAP_ERR_NO_ERROR; } else if (param->handle == (ota_env->shdl + OTA_IDX_VAL)) { // force length to zero to reject any write starting from something != 0 cfm->length = 0; cfm->status = GAP_ERR_NO_ERROR; } else { cfm->length = 0; cfm->status = ATT_ERR_WRITE_NOT_PERMITTED; } ke_msg_send(cfm); return (KE_MSG_CONSUMED); } /* * GLOBAL VARIABLE DEFINITIONS **************************************************************************************** */ /* Default State handlers definition. */ KE_MSG_HANDLER_TAB(ota) { {GAPC_DISCONNECT_IND, (ke_msg_func_t) gapc_disconnect_ind_handler}, {GATTC_WRITE_REQ_IND, (ke_msg_func_t) gattc_write_req_ind_handler}, {GATTC_CMP_EVT, (ke_msg_func_t) gattc_cmp_evt_handler}, {GATTC_READ_REQ_IND, (ke_msg_func_t) gattc_read_req_ind_handler}, {OTA_SEND_NOTIFICATION, (ke_msg_func_t) send_notification_handler}, {OTA_SEND_INDICATION, (ke_msg_func_t) send_indication_handler}, {GATTC_ATT_INFO_REQ_IND, (ke_msg_func_t) gattc_att_info_req_ind_handler }, }; void ota_task_init(struct ke_task_desc *task_desc) { // Get the address of the environment struct ota_env_tag *ota_env = PRF_ENV_GET(OTA, ota); task_desc->msg_handler_tab = ota_msg_handler_tab; task_desc->msg_cnt = ARRAY_LEN(ota_msg_handler_tab); task_desc->state = &(ota_env->state); task_desc->idx_max = BLE_CONNECTION_MAX; } #endif /* #if (BLE_OTA) */ /// @} OTATASK