123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488 |
- #include "sdk_common.h"
- #if NRF_MODULE_ENABLED(NRF_BLE_QWR)
- #include <stdlib.h>
- #include "nrf_ble_qwr.h"
- #include "ble.h"
- #include "ble_srv_common.h"
- #define NRF_BLE_QWR_INITIALIZED 0xDE
- #define MODULE_INITIALIZED (p_qwr->initialized == NRF_BLE_QWR_INITIALIZED)
- #include "sdk_macros.h"
- ret_code_t nrf_ble_qwr_init(nrf_ble_qwr_t * p_qwr,
- nrf_ble_qwr_init_t const * p_qwr_init)
- {
- VERIFY_PARAM_NOT_NULL(p_qwr);
- VERIFY_PARAM_NOT_NULL(p_qwr_init);
- if (MODULE_INITIALIZED)
- {
- return NRF_ERROR_INVALID_STATE;
- }
- p_qwr->error_handler = p_qwr_init->error_handler;
- p_qwr->initialized = NRF_BLE_QWR_INITIALIZED;
- p_qwr->conn_handle = BLE_CONN_HANDLE_INVALID;
- #if (NRF_BLE_QWR_MAX_ATTR > 0)
- memset(p_qwr->attr_handles, 0, sizeof(p_qwr->attr_handles));
- p_qwr->nb_registered_attr = 0;
- p_qwr->is_user_mem_reply_pending = false;
- p_qwr->mem_buffer = p_qwr_init->mem_buffer;
- p_qwr->callback = p_qwr_init->callback;
- p_qwr->nb_written_handles = 0;
- #endif
- return NRF_SUCCESS;
- }
- #if (NRF_BLE_QWR_MAX_ATTR > 0)
- ret_code_t nrf_ble_qwr_attr_register(nrf_ble_qwr_t * p_qwr, uint16_t attr_handle)
- {
- VERIFY_PARAM_NOT_NULL(p_qwr);
- VERIFY_MODULE_INITIALIZED();
- if ((p_qwr->nb_registered_attr == NRF_BLE_QWR_MAX_ATTR)
- || (p_qwr->mem_buffer.p_mem == NULL)
- || (p_qwr->mem_buffer.len == 0))
- {
- return (NRF_ERROR_NO_MEM);
- }
- if (attr_handle == BLE_GATT_HANDLE_INVALID)
- {
- return NRF_ERROR_INVALID_PARAM;
- }
- p_qwr->attr_handles[p_qwr->nb_registered_attr] = attr_handle;
- p_qwr->nb_registered_attr++;
- return NRF_SUCCESS;
- }
- ret_code_t nrf_ble_qwr_value_get(nrf_ble_qwr_t * p_qwr,
- uint16_t attr_handle,
- uint8_t * p_mem,
- uint16_t * p_len)
- {
- VERIFY_PARAM_NOT_NULL(p_qwr);
- VERIFY_PARAM_NOT_NULL(p_mem);
- VERIFY_PARAM_NOT_NULL(p_len);
- VERIFY_MODULE_INITIALIZED();
- uint16_t i = 0;
- uint16_t handle = BLE_GATT_HANDLE_INVALID;
- uint16_t val_len = 0;
- uint16_t val_offset = 0;
- uint32_t cur_len = 0;
- do
- {
- handle = uint16_decode(&(p_qwr->mem_buffer.p_mem[i]));
- if (handle == BLE_GATT_HANDLE_INVALID)
- {
- break;
- }
- i += sizeof(uint16_t);
- val_offset = uint16_decode(&(p_qwr->mem_buffer.p_mem[i]));
- i += sizeof(uint16_t);
- val_len = uint16_decode(&(p_qwr->mem_buffer.p_mem[i]));
- i += sizeof(uint16_t);
- if (handle == attr_handle)
- {
- cur_len = val_offset + val_len;
- if (cur_len <= *p_len)
- {
- memcpy((p_mem + val_offset), &(p_qwr->mem_buffer.p_mem[i]), val_len);
- }
- else
- {
- return NRF_ERROR_NO_MEM;
- }
- }
- i += val_len;
- }
- while (i < p_qwr->mem_buffer.len);
- *p_len = cur_len;
- return NRF_SUCCESS;
- }
- #endif
- ret_code_t nrf_ble_qwr_conn_handle_assign(nrf_ble_qwr_t * p_qwr,
- uint16_t conn_handle)
- {
- VERIFY_PARAM_NOT_NULL(p_qwr);
- VERIFY_MODULE_INITIALIZED();
- p_qwr->conn_handle = conn_handle;
- return NRF_SUCCESS;
- }
- static void user_mem_reply(nrf_ble_qwr_t * p_qwr)
- {
- if (p_qwr->is_user_mem_reply_pending)
- {
- ret_code_t err_code;
- #if (NRF_BLE_QWR_MAX_ATTR == 0)
- err_code = sd_ble_user_mem_reply(p_qwr->conn_handle, NULL);
- #else
- err_code = sd_ble_user_mem_reply(p_qwr->conn_handle, &p_qwr->mem_buffer);
- #endif
- if (err_code == NRF_SUCCESS)
- {
- p_qwr->is_user_mem_reply_pending = false;
- }
- else if (err_code == NRF_ERROR_BUSY)
- {
- p_qwr->is_user_mem_reply_pending = true;
- }
- else
- {
- p_qwr->error_handler(err_code);
- }
- }
- }
- static void on_user_mem_request(nrf_ble_qwr_t * p_qwr,
- ble_common_evt_t const * p_common_evt)
- {
- if ((p_common_evt->params.user_mem_request.type == BLE_USER_MEM_TYPE_GATTS_QUEUED_WRITES) &&
- (p_common_evt->conn_handle == p_qwr->conn_handle))
- {
- p_qwr->is_user_mem_reply_pending = true;
- user_mem_reply(p_qwr);
- }
- }
- static void on_user_mem_release(nrf_ble_qwr_t * p_qwr,
- ble_common_evt_t const * p_common_evt)
- {
- #if (NRF_BLE_QWR_MAX_ATTR > 0)
- if ((p_common_evt->params.user_mem_release.type == BLE_USER_MEM_TYPE_GATTS_QUEUED_WRITES) &&
- (p_common_evt->conn_handle == p_qwr->conn_handle))
- {
-
- p_qwr->nb_written_handles = 0;
- }
- #endif
- }
- #if (NRF_BLE_QWR_MAX_ATTR > 0)
- static void on_prepare_write(nrf_ble_qwr_t * p_qwr,
- ble_gatts_evt_write_t const * p_evt_write)
- {
- uint32_t err_code;
- ble_gatts_rw_authorize_reply_params_t auth_reply;
- memset(&auth_reply, 0, sizeof(auth_reply));
- auth_reply.params.write.gatt_status = NRF_BLE_QWR_REJ_REQUEST_ERR_CODE;
- auth_reply.type = BLE_GATTS_AUTHORIZE_TYPE_WRITE;
- uint32_t i;
- for (i = 0; i < p_qwr->nb_written_handles; i++)
- {
- if (p_qwr->written_attr_handles[i] == p_evt_write->handle)
- {
- auth_reply.params.write.gatt_status = BLE_GATT_STATUS_SUCCESS;
- break;
- }
- }
- if (auth_reply.params.write.gatt_status != BLE_GATT_STATUS_SUCCESS)
- {
- for (i = 0; i < p_qwr->nb_registered_attr; i++)
- {
- if (p_qwr->attr_handles[i] == p_evt_write->handle)
- {
- auth_reply.params.write.gatt_status = BLE_GATT_STATUS_SUCCESS;
- p_qwr->written_attr_handles[p_qwr->nb_written_handles++] = p_evt_write->handle;
- break;
- }
- }
- }
- err_code = sd_ble_gatts_rw_authorize_reply(p_qwr->conn_handle, &auth_reply);
- if (err_code != NRF_SUCCESS)
- {
-
- p_qwr->nb_written_handles = 0;
-
- p_qwr->error_handler(err_code);
- }
- }
- static void on_execute_write(nrf_ble_qwr_t * p_qwr,
- ble_gatts_evt_write_t const * p_evt_write)
- {
- uint32_t err_code;
- ble_gatts_rw_authorize_reply_params_t auth_reply;
- memset(&auth_reply, 0, sizeof(auth_reply));
- auth_reply.params.write.gatt_status = BLE_GATT_STATUS_SUCCESS;
- auth_reply.type = BLE_GATTS_AUTHORIZE_TYPE_WRITE;
- if (p_qwr->nb_written_handles == 0)
- {
- auth_reply.params.write.gatt_status = NRF_BLE_QWR_REJ_REQUEST_ERR_CODE;
- err_code = sd_ble_gatts_rw_authorize_reply(p_qwr->conn_handle, &auth_reply);
- if (err_code != NRF_SUCCESS)
- {
-
- p_qwr->error_handler(err_code);
- }
- return;
- }
- for (uint16_t i = 0; i < p_qwr->nb_written_handles; i++)
- {
- nrf_ble_qwr_evt_t evt;
- uint16_t ret_val;
- evt.evt_type = NRF_BLE_QWR_EVT_AUTH_REQUEST;
- evt.attr_handle = p_qwr->written_attr_handles[i];
- ret_val = p_qwr->callback(p_qwr, &evt);
- if (ret_val != BLE_GATT_STATUS_SUCCESS)
- {
- auth_reply.params.write.gatt_status = ret_val;
- }
- }
- err_code = sd_ble_gatts_rw_authorize_reply(p_qwr->conn_handle, &auth_reply);
- if (err_code != NRF_SUCCESS)
- {
-
- p_qwr->error_handler(err_code);
- }
-
- if (auth_reply.params.write.gatt_status == BLE_GATT_STATUS_SUCCESS)
- {
- for (uint16_t i = 0; i < p_qwr->nb_written_handles; i++)
- {
- nrf_ble_qwr_evt_t evt;
- evt.evt_type = NRF_BLE_QWR_EVT_EXECUTE_WRITE;
- evt.attr_handle = p_qwr->written_attr_handles[i];
-
- p_qwr->callback(p_qwr, &evt);
-
- auth_reply.params.write.gatt_status = BLE_GATT_STATUS_SUCCESS;
- }
- }
- p_qwr->nb_written_handles = 0;
- }
- static void on_cancel_write(nrf_ble_qwr_t * p_qwr,
- ble_gatts_evt_write_t const * p_evt_write)
- {
- uint32_t err_code;
- ble_gatts_rw_authorize_reply_params_t auth_reply;
- memset(&auth_reply, 0, sizeof(auth_reply));
- auth_reply.type = BLE_GATTS_AUTHORIZE_TYPE_WRITE;
- auth_reply.params.write.gatt_status = BLE_GATT_STATUS_SUCCESS;
- err_code = sd_ble_gatts_rw_authorize_reply(p_qwr->conn_handle, &auth_reply);
- if (err_code != NRF_SUCCESS)
- {
-
- p_qwr->error_handler(err_code);
- }
- p_qwr->nb_written_handles = 0;
- }
- #endif
- static void on_rw_authorize_request(nrf_ble_qwr_t * p_qwr,
- ble_gatts_evt_t const * p_gatts_evt)
- {
- if (p_gatts_evt->conn_handle != p_qwr->conn_handle)
- {
- return;
- }
- ble_gatts_evt_rw_authorize_request_t const * p_auth_req = &p_gatts_evt->params.authorize_request;
- if (p_auth_req->type != BLE_GATTS_AUTHORIZE_TYPE_WRITE)
- {
- return;
- }
- #if (NRF_BLE_QWR_MAX_ATTR == 0)
-
- if ((p_auth_req->request.write.op != BLE_GATTS_OP_PREP_WRITE_REQ) &&
- (p_auth_req->request.write.op != BLE_GATTS_OP_EXEC_WRITE_REQ_NOW) &&
- (p_auth_req->request.write.op != BLE_GATTS_OP_EXEC_WRITE_REQ_CANCEL))
- {
- return;
- }
-
- ble_gatts_rw_authorize_reply_params_t auth_reply = {0};
- auth_reply.type = BLE_GATTS_AUTHORIZE_TYPE_WRITE;
- auth_reply.params.write.gatt_status = NRF_BLE_QWR_REJ_REQUEST_ERR_CODE;
- if (p_auth_req->request.write.op == BLE_GATTS_OP_EXEC_WRITE_REQ_CANCEL)
- {
- auth_reply.params.write.gatt_status = BLE_GATT_STATUS_SUCCESS;
- }
- ret_code_t err_code = sd_ble_gatts_rw_authorize_reply(p_gatts_evt->conn_handle, &auth_reply);
- if (err_code != NRF_SUCCESS)
- {
-
- p_qwr->error_handler(err_code);
- }
- #else
- switch (p_auth_req->request.write.op)
- {
- case BLE_GATTS_OP_PREP_WRITE_REQ:
- on_prepare_write(p_qwr, &p_auth_req->request.write);
- break;
- case BLE_GATTS_OP_EXEC_WRITE_REQ_NOW:
- on_execute_write(p_qwr, &p_auth_req->request.write);
- break;
- case BLE_GATTS_OP_EXEC_WRITE_REQ_CANCEL:
- on_cancel_write(p_qwr, &p_auth_req->request.write);
- break;
- default:
-
- break;
- }
- #endif
- }
- void nrf_ble_qwr_on_ble_evt(ble_evt_t const * p_ble_evt, void * p_context)
- {
- VERIFY_PARAM_NOT_NULL_VOID(p_context);
- VERIFY_PARAM_NOT_NULL_VOID(p_ble_evt);
- nrf_ble_qwr_t * p_qwr = (nrf_ble_qwr_t *)p_context;
- VERIFY_MODULE_INITIALIZED_VOID();
- if (p_ble_evt->evt.common_evt.conn_handle == p_qwr->conn_handle)
- {
- user_mem_reply(p_qwr);
- }
- switch (p_ble_evt->header.evt_id)
- {
- case BLE_EVT_USER_MEM_REQUEST:
- on_user_mem_request(p_qwr, &p_ble_evt->evt.common_evt);
- break;
- case BLE_EVT_USER_MEM_RELEASE:
- on_user_mem_release(p_qwr, &p_ble_evt->evt.common_evt);
- break;
- case BLE_GATTS_EVT_RW_AUTHORIZE_REQUEST:
- on_rw_authorize_request(p_qwr, &p_ble_evt->evt.gatts_evt);
- break;
- case BLE_GAP_EVT_DISCONNECTED:
- if (p_ble_evt->evt.gap_evt.conn_handle == p_qwr->conn_handle)
- {
- p_qwr->conn_handle = BLE_CONN_HANDLE_INVALID;
- #if (NRF_BLE_QWR_MAX_ATTR > 0)
- p_qwr->nb_written_handles = 0;
- #endif
- }
- break;
- default:
- break;
- }
- }
- #endif
|