You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

645 lines
23 KiB

1 year ago
1 year ago
1 year ago
  1. /**
  2. * Copyright (c) 2016 - 2021, Nordic Semiconductor ASA
  3. *
  4. * All rights reserved.
  5. *
  6. * Redistribution and use in source and binary forms, with or without modification,
  7. * are permitted provided that the following conditions are met:
  8. *
  9. * 1. Redistributions of source code must retain the above copyright notice, this
  10. * list of conditions and the following disclaimer.
  11. *
  12. * 2. Redistributions in binary form, except as embedded into a Nordic
  13. * Semiconductor ASA integrated circuit in a product or a software update for
  14. * such product, must reproduce the above copyright notice, this list of
  15. * conditions and the following disclaimer in the documentation and/or other
  16. * materials provided with the distribution.
  17. *
  18. * 3. Neither the name of Nordic Semiconductor ASA nor the names of its
  19. * contributors may be used to endorse or promote products derived from this
  20. * software without specific prior written permission.
  21. *
  22. * 4. This software, with or without modification, must only be used with a
  23. * Nordic Semiconductor ASA integrated circuit.
  24. *
  25. * 5. Any software provided in binary form under this license must not be reverse
  26. * engineered, decompiled, modified and/or disassembled.
  27. *
  28. * THIS SOFTWARE IS PROVIDED BY NORDIC SEMICONDUCTOR ASA "AS IS" AND ANY EXPRESS
  29. * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  30. * OF MERCHANTABILITY, NONINFRINGEMENT, AND FITNESS FOR A PARTICULAR PURPOSE ARE
  31. * DISCLAIMED. IN NO EVENT SHALL NORDIC SEMICONDUCTOR ASA OR CONTRIBUTORS BE
  32. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  33. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
  34. * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  35. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  36. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  37. * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  38. *
  39. */
  40. #include <stdbool.h>
  41. #include <stdint.h>
  42. #include <stdio.h>
  43. #include "app_error.h"
  44. #include "app_timer.h"
  45. #include "app_uart.h"
  46. #include "app_util.h"
  47. #include "ble.h"
  48. #include "ble_db_discovery.h"
  49. #include "ble_gap.h"
  50. #include "ble_hci.h"
  51. #include "bsp_btn_ble.h"
  52. #include "nordic_common.h"
  53. #include "nrf_ble_gatt.h"
  54. #include "nrf_ble_scan.h"
  55. #include "nrf_log.h"
  56. #include "nrf_log_ctrl.h"
  57. #include "nrf_log_default_backends.h"
  58. #include "nrf_pwr_mgmt.h"
  59. #include "nrf_sdh.h"
  60. #include "nrf_sdh_ble.h"
  61. #include "nrf_sdh_soc.h"
  62. #include "zble_nus_c.h"
  63. #define APP_BLE_CONN_CFG_TAG 1 /**< Tag that refers to the BLE stack configuration set with @ref sd_ble_cfg_set. The default tag is @ref BLE_CONN_CFG_TAG_DEFAULT. */
  64. #define APP_BLE_OBSERVER_PRIO 3 /**< BLE observer priority of the application. There is no need to modify this value. */
  65. #define UART_TX_BUF_SIZE 256 /**< UART TX buffer size. */
  66. #define UART_RX_BUF_SIZE 256 /**< UART RX buffer size. */
  67. #define NUS_SERVICE_UUID_TYPE BLE_UUID_TYPE_VENDOR_BEGIN /**< UUID type for the Nordic UART Service (vendor specific). */
  68. #define ECHOBACK_BLE_UART_DATA 0 /**< Echo the UART data that is received over the Nordic UART Service (NUS) back to the sender. */
  69. BLE_NUS_C_DEF(m_ble_nus_c); /**< BLE Nordic UART Service (NUS) client instance. */
  70. NRF_BLE_GATT_DEF(m_gatt); /**< GATT module instance. */
  71. BLE_DB_DISCOVERY_DEF(m_db_disc); /**< Database discovery module instance. */
  72. NRF_BLE_SCAN_DEF(m_scan); /**< Scanning Module instance. */
  73. NRF_BLE_GQ_DEF(m_ble_gatt_queue, /**< BLE GATT Queue instance. */
  74. NRF_SDH_BLE_CENTRAL_LINK_COUNT, NRF_BLE_GQ_QUEUE_SIZE);
  75. static uint16_t m_ble_nus_max_data_len = BLE_GATT_ATT_MTU_DEFAULT - OPCODE_LENGTH - HANDLE_LENGTH; /**< Maximum length of data (in bytes) that can be transmitted to the peer by the Nordic UART service module. */
  76. /**@brief NUS UUID. */
  77. static ble_uuid_t const m_nus_uuid = {.uuid = BLE_UUID_NUS_SERVICE, .type = NUS_SERVICE_UUID_TYPE};
  78. /**@brief Function for handling asserts in the SoftDevice.
  79. *
  80. * @details This function is called in case of an assert in the SoftDevice.
  81. *
  82. * @warning This handler is only an example and is not meant for the final product. You need to analyze
  83. * how your product is supposed to react in case of assert.
  84. * @warning On assert from the SoftDevice, the system can only recover on reset.
  85. *
  86. * @param[in] line_num Line number of the failing assert call.
  87. * @param[in] p_file_name File name of the failing assert call.
  88. */
  89. void assert_nrf_callback(uint16_t line_num, const uint8_t* p_file_name) { app_error_handler(0xDEADBEEF, line_num, p_file_name); }
  90. /**@brief Function for handling the Nordic UART Service Client errors.
  91. *
  92. * @param[in] nrf_error Error code containing information about what went wrong.
  93. */
  94. static void nus_error_handler(uint32_t nrf_error) { APP_ERROR_HANDLER(nrf_error); }
  95. /**@brief Function to start scanning. */
  96. static void scan_start(void) {
  97. ret_code_t ret;
  98. ret = nrf_ble_scan_start(&m_scan);
  99. APP_ERROR_CHECK(ret);
  100. ret = bsp_indication_set(BSP_INDICATE_SCANNING);
  101. APP_ERROR_CHECK(ret);
  102. }
  103. /**@brief Function for handling Scanning Module events.
  104. */
  105. static void scan_evt_handler(scan_evt_t const* p_scan_evt) {
  106. ret_code_t err_code;
  107. switch (p_scan_evt->scan_evt_id) {
  108. case NRF_BLE_SCAN_EVT_CONNECTING_ERROR: {
  109. err_code = p_scan_evt->params.connecting_err.err_code;
  110. APP_ERROR_CHECK(err_code);
  111. } break;
  112. case NRF_BLE_SCAN_EVT_CONNECTED: {
  113. ble_gap_evt_connected_t const* p_connected = p_scan_evt->params.connected.p_connected;
  114. // Scan is automatically stopped by the connection.
  115. NRF_LOG_INFO("Connecting to target %02x%02x%02x%02x%02x%02x", p_connected->peer_addr.addr[0], p_connected->peer_addr.addr[1], p_connected->peer_addr.addr[2], p_connected->peer_addr.addr[3], p_connected->peer_addr.addr[4], p_connected->peer_addr.addr[5]);
  116. } break;
  117. case NRF_BLE_SCAN_EVT_SCAN_TIMEOUT: {
  118. NRF_LOG_INFO("Scan timed out.");
  119. scan_start();
  120. } break;
  121. default:
  122. break;
  123. }
  124. }
  125. /**@brief Function for initializing the scanning and setting the filters.
  126. */
  127. static char m_target_periph_name0[20] = "M1002"; /**< Name of the device we try to connect to. This name is searched in the scan report data*/
  128. static char m_target_periph_name1[20] = "M1001"; /**< Name of the device we try to connect to. This name is searched in the scan report data*/
  129. static char m_target_periph_name2[20] = "iflytop"; /**< Name of the device we try to connect to. This name is searched in the scan report data*/
  130. static char m_target_periph_name3[20] = "ADS1293"; /**< Name of the device we try to connect to. This name is searched in the scan report data*/
  131. static char m_target_periph_name4[20] = "IFLYTOP"; /**< Name of the device we try to connect to. This name is searched in the scan report data*/
  132. static char m_target_periph_name5[20] = "M1003"; /**< Name of the device we try to connect to. This name is searched in the scan report data*/
  133. static void scan_init(void) {
  134. ret_code_t err_code;
  135. nrf_ble_scan_init_t init_scan;
  136. memset(&init_scan, 0, sizeof(init_scan));
  137. init_scan.connect_if_match = true;
  138. init_scan.conn_cfg_tag = APP_BLE_CONN_CFG_TAG;
  139. err_code = nrf_ble_scan_init(&m_scan, &init_scan, scan_evt_handler);
  140. APP_ERROR_CHECK(err_code);
  141. // err_code = nrf_ble_scan_filter_set(&m_scan, SCAN_UUID_FILTER, &m_nus_uuid);
  142. // APP_ERROR_CHECK(err_code);
  143. APP_ERROR_CHECK(nrf_ble_scan_filter_set(&m_scan, SCAN_NAME_FILTER, m_target_periph_name0));
  144. APP_ERROR_CHECK(nrf_ble_scan_filter_set(&m_scan, SCAN_NAME_FILTER, m_target_periph_name1));
  145. APP_ERROR_CHECK(nrf_ble_scan_filter_set(&m_scan, SCAN_NAME_FILTER, m_target_periph_name2));
  146. APP_ERROR_CHECK(nrf_ble_scan_filter_set(&m_scan, SCAN_NAME_FILTER, m_target_periph_name3));
  147. APP_ERROR_CHECK(nrf_ble_scan_filter_set(&m_scan, SCAN_NAME_FILTER, m_target_periph_name4));
  148. APP_ERROR_CHECK(nrf_ble_scan_filter_set(&m_scan, SCAN_NAME_FILTER, m_target_periph_name5));
  149. // err_code = nrf_ble_scan_filters_enable(&m_scan, NRF_BLE_SCAN_NAME_FILTER, false);
  150. // APP_ERROR_CHECK(err_code);
  151. err_code = nrf_ble_scan_filters_enable(&m_scan, NRF_BLE_SCAN_NAME_FILTER, false);
  152. APP_ERROR_CHECK(err_code);
  153. }
  154. /**@brief Function for handling database discovery events.
  155. *
  156. * @details This function is a callback function to handle events from the database discovery module.
  157. * Depending on the UUIDs that are discovered, this function forwards the events
  158. * to their respective services.
  159. *
  160. * @param[in] p_event Pointer to the database discovery event.
  161. */
  162. static void db_disc_handler(ble_db_discovery_evt_t* p_evt) { ble_nus_c_on_db_disc_evt(&m_ble_nus_c, p_evt); }
  163. /**@brief Function for handling characters received by the Nordic UART Service (NUS).
  164. *
  165. * @details This function takes a list of characters of length data_len and prints the characters out on UART.
  166. * If @ref ECHOBACK_BLE_UART_DATA is set, the data is sent back to sender.
  167. */
  168. static void app_uart_force_put(uint8_t data) {
  169. ret_code_t ret_val;
  170. do {
  171. ret_val = app_uart_put(data);
  172. if ((ret_val != NRF_SUCCESS) && (ret_val != NRF_ERROR_BUSY)) {
  173. NRF_LOG_ERROR("app_uart_put failed for index 0x%04x.", 0);
  174. APP_ERROR_CHECK(ret_val);
  175. }
  176. } while (ret_val == NRF_ERROR_BUSY);
  177. }
  178. static void ble_nus_chars_received_uart_print(uint8_t* p_data, uint16_t data_len) {
  179. ret_code_t ret_val;
  180. NRF_LOG_DEBUG("Receiving data.");
  181. NRF_LOG_HEXDUMP_DEBUG(p_data, data_len);
  182. app_uart_force_put(0x5A);
  183. app_uart_force_put(0xA5);
  184. uint8_t sumcheck = 0;
  185. for (uint32_t i = 0; i < data_len; i++) {
  186. do {
  187. ret_val = app_uart_put(p_data[i]);
  188. sumcheck += p_data[i];
  189. if ((ret_val != NRF_SUCCESS) && (ret_val != NRF_ERROR_BUSY)) {
  190. NRF_LOG_ERROR("app_uart_put failed for index 0x%04x.", i);
  191. APP_ERROR_CHECK(ret_val);
  192. }
  193. } while (ret_val == NRF_ERROR_BUSY);
  194. }
  195. app_uart_force_put(sumcheck);
  196. app_uart_force_put(0x5B);
  197. app_uart_force_put(0xB5);
  198. }
  199. static void ble_nus_chars_received_uart_print_ch4(uint8_t* p_data, uint16_t data_len) {
  200. ret_code_t ret_val;
  201. NRF_LOG_DEBUG("Receiving data.");
  202. NRF_LOG_HEXDUMP_DEBUG(p_data, data_len);
  203. app_uart_force_put(0x4A);
  204. app_uart_force_put(0xA4);
  205. for (uint32_t i = 0; i < data_len; i++) {
  206. do {
  207. ret_val = app_uart_put(p_data[i]);
  208. if ((ret_val != NRF_SUCCESS) && (ret_val != NRF_ERROR_BUSY)) {
  209. NRF_LOG_ERROR("app_uart_put failed for index 0x%04x.", i);
  210. APP_ERROR_CHECK(ret_val);
  211. }
  212. } while (ret_val == NRF_ERROR_BUSY);
  213. }
  214. app_uart_force_put(0x4B);
  215. app_uart_force_put(0xB4);
  216. }
  217. /**@brief Function for handling app_uart events.
  218. *
  219. * @details This function receives a single character from the app_uart module and appends it to
  220. * a string. The string is sent over BLE when the last character received is a
  221. * 'new line' '\n' (hex 0x0A) or if the string reaches the maximum data length.
  222. */
  223. static uint8_t data_array[BLE_NUS_MAX_DATA_LEN];
  224. static uint16_t data_array_index = 0;
  225. // static uint32_t
  226. void uart_event_handle(app_uart_evt_t* p_event) {
  227. uint32_t ret_val;
  228. switch (p_event->evt_type) {
  229. /**@snippet [Handling data from UART] */
  230. case APP_UART_DATA_READY:
  231. UNUSED_VARIABLE(app_uart_get(&data_array[data_array_index]));
  232. data_array_index++;
  233. if (data_array_index >= (m_ble_nus_max_data_len)) {
  234. NRF_LOG_DEBUG("Ready to send data over BLE NUS");
  235. NRF_LOG_HEXDUMP_DEBUG(data_array, data_array_index);
  236. do {
  237. ret_val = ble_nus_c_string_send(&m_ble_nus_c, data_array, data_array_index);
  238. if ((ret_val != NRF_ERROR_INVALID_STATE) && (ret_val != NRF_ERROR_RESOURCES)) {
  239. APP_ERROR_CHECK(ret_val);
  240. }
  241. } while (ret_val == NRF_ERROR_RESOURCES);
  242. data_array_index = 0;
  243. }
  244. break;
  245. /**@snippet [Handling data from UART] */
  246. case APP_UART_COMMUNICATION_ERROR:
  247. NRF_LOG_ERROR("Communication error occurred while handling UART.");
  248. APP_ERROR_HANDLER(p_event->data.error_communication);
  249. break;
  250. case APP_UART_FIFO_ERROR:
  251. NRF_LOG_ERROR("Error occurred in FIFO module used by UART.");
  252. APP_ERROR_HANDLER(p_event->data.error_code);
  253. break;
  254. default:
  255. break;
  256. }
  257. }
  258. /**@brief Callback handling Nordic UART Service (NUS) client events.
  259. *
  260. * @details This function is called to notify the application of NUS client events.
  261. *
  262. * @param[in] p_ble_nus_c NUS client handle. This identifies the NUS client.
  263. * @param[in] p_ble_nus_evt Pointer to the NUS client event.
  264. */
  265. /**@snippet [Handling events from the ble_nus_c module] */
  266. typedef struct {
  267. uint32_t rxcnt;
  268. uint32_t m_rx_sum_cnt;
  269. } block_data_rx_state_t;
  270. block_data_rx_state_t block_data_rx_state;
  271. static void ble_nus_c_evt_handler(ble_nus_c_t* p_ble_nus_c, ble_nus_c_evt_t const* p_ble_nus_evt) {
  272. ret_code_t err_code;
  273. switch (p_ble_nus_evt->evt_type) {
  274. case BLE_NUS_C_EVT_DISCOVERY_COMPLETE:
  275. NRF_LOG_INFO("Discovery complete.");
  276. err_code = ble_nus_c_handles_assign(p_ble_nus_c, p_ble_nus_evt->conn_handle, &p_ble_nus_evt->handles);
  277. APP_ERROR_CHECK(err_code);
  278. err_code = ble_nus_c_tx_notif_enable(p_ble_nus_c);
  279. APP_ERROR_CHECK(err_code);
  280. NRF_LOG_INFO("Connected to device with Nordic UART Service.");
  281. break;
  282. case BLE_NUS_C_EVT_NUS_TX_EVT:
  283. ble_nus_chars_received_uart_print(p_ble_nus_evt->p_data, p_ble_nus_evt->data_len);
  284. break;
  285. case BLE_NUS_C_EVT_NUS_TX_EVT_2:
  286. #if 1
  287. block_data_rx_state.rxcnt += p_ble_nus_evt->data_len;
  288. for (size_t i = 0; i < p_ble_nus_evt->data_len; i++) {
  289. block_data_rx_state.m_rx_sum_cnt += p_ble_nus_evt->p_data[i];
  290. }
  291. ble_nus_chars_received_uart_print_ch4((uint8_t*)&block_data_rx_state, sizeof(block_data_rx_state));
  292. #else
  293. ble_nus_chars_received_uart_print_ch4((uint8_t*)p_ble_nus_evt->p_data, p_ble_nus_evt->data_len);
  294. #endif
  295. break;
  296. case BLE_NUS_C_EVT_DISCONNECTED:
  297. NRF_LOG_INFO("Disconnected.");
  298. scan_start();
  299. break;
  300. }
  301. }
  302. /**@snippet [Handling events from the ble_nus_c module] */
  303. /**
  304. * @brief Function for handling shutdown events.
  305. *
  306. * @param[in] event Shutdown type.
  307. */
  308. static bool shutdown_handler(nrf_pwr_mgmt_evt_t event) {
  309. ret_code_t err_code;
  310. err_code = bsp_indication_set(BSP_INDICATE_IDLE);
  311. APP_ERROR_CHECK(err_code);
  312. switch (event) {
  313. case NRF_PWR_MGMT_EVT_PREPARE_WAKEUP:
  314. // Prepare wakeup buttons.
  315. err_code = bsp_btn_ble_sleep_mode_prepare();
  316. APP_ERROR_CHECK(err_code);
  317. break;
  318. default:
  319. break;
  320. }
  321. return true;
  322. }
  323. NRF_PWR_MGMT_HANDLER_REGISTER(shutdown_handler, APP_SHUTDOWN_HANDLER_PRIORITY);
  324. /**@brief Function for handling BLE events.
  325. *
  326. * @param[in] p_ble_evt Bluetooth stack event.
  327. * @param[in] p_context Unused.
  328. */
  329. static void ble_evt_handler(ble_evt_t const* p_ble_evt, void* p_context) {
  330. ret_code_t err_code;
  331. ble_gap_evt_t const* p_gap_evt = &p_ble_evt->evt.gap_evt;
  332. switch (p_ble_evt->header.evt_id) {
  333. case BLE_GAP_EVT_CONNECTED:
  334. err_code = ble_nus_c_handles_assign(&m_ble_nus_c, p_ble_evt->evt.gap_evt.conn_handle, NULL);
  335. APP_ERROR_CHECK(err_code);
  336. err_code = bsp_indication_set(BSP_INDICATE_CONNECTED);
  337. APP_ERROR_CHECK(err_code);
  338. // start discovery of services. The NUS Client waits for a discovery result
  339. err_code = ble_db_discovery_start(&m_db_disc, p_ble_evt->evt.gap_evt.conn_handle);
  340. APP_ERROR_CHECK(err_code);
  341. break;
  342. case BLE_GAP_EVT_DISCONNECTED:
  343. NRF_LOG_INFO("Disconnected. conn_handle: 0x%x, reason: 0x%x", p_gap_evt->conn_handle, p_gap_evt->params.disconnected.reason);
  344. break;
  345. case BLE_GAP_EVT_TIMEOUT:
  346. if (p_gap_evt->params.timeout.src == BLE_GAP_TIMEOUT_SRC_CONN) {
  347. NRF_LOG_INFO("Connection Request timed out.");
  348. }
  349. break;
  350. case BLE_GAP_EVT_SEC_PARAMS_REQUEST:
  351. // Pairing not supported.
  352. err_code = sd_ble_gap_sec_params_reply(p_ble_evt->evt.gap_evt.conn_handle, BLE_GAP_SEC_STATUS_PAIRING_NOT_SUPP, NULL, NULL);
  353. APP_ERROR_CHECK(err_code);
  354. break;
  355. case BLE_GAP_EVT_CONN_PARAM_UPDATE_REQUEST:
  356. // Accepting parameters requested by peer.
  357. err_code = sd_ble_gap_conn_param_update(p_gap_evt->conn_handle, &p_gap_evt->params.conn_param_update_request.conn_params);
  358. APP_ERROR_CHECK(err_code);
  359. break;
  360. case BLE_GAP_EVT_PHY_UPDATE_REQUEST: {
  361. NRF_LOG_DEBUG("PHY update request.");
  362. ble_gap_phys_t const phys = {
  363. .rx_phys = BLE_GAP_PHY_AUTO,
  364. .tx_phys = BLE_GAP_PHY_AUTO,
  365. };
  366. err_code = sd_ble_gap_phy_update(p_ble_evt->evt.gap_evt.conn_handle, &phys);
  367. APP_ERROR_CHECK(err_code);
  368. } break;
  369. case BLE_GATTC_EVT_TIMEOUT:
  370. // Disconnect on GATT Client timeout event.
  371. NRF_LOG_DEBUG("GATT Client Timeout.");
  372. err_code = sd_ble_gap_disconnect(p_ble_evt->evt.gattc_evt.conn_handle, BLE_HCI_REMOTE_USER_TERMINATED_CONNECTION);
  373. APP_ERROR_CHECK(err_code);
  374. break;
  375. case BLE_GATTS_EVT_TIMEOUT:
  376. // Disconnect on GATT Server timeout event.
  377. NRF_LOG_DEBUG("GATT Server Timeout.");
  378. err_code = sd_ble_gap_disconnect(p_ble_evt->evt.gatts_evt.conn_handle, BLE_HCI_REMOTE_USER_TERMINATED_CONNECTION);
  379. APP_ERROR_CHECK(err_code);
  380. break;
  381. default:
  382. break;
  383. }
  384. }
  385. /**@brief Function for initializing the BLE stack.
  386. *
  387. * @details Initializes the SoftDevice and the BLE event interrupt.
  388. */
  389. static void ble_stack_init(void) {
  390. ret_code_t err_code;
  391. err_code = nrf_sdh_enable_request();
  392. APP_ERROR_CHECK(err_code);
  393. // Configure the BLE stack using the default settings.
  394. // Fetch the start address of the application RAM.
  395. uint32_t ram_start = 0;
  396. err_code = nrf_sdh_ble_default_cfg_set(APP_BLE_CONN_CFG_TAG, &ram_start);
  397. APP_ERROR_CHECK(err_code);
  398. // Enable BLE stack.
  399. err_code = nrf_sdh_ble_enable(&ram_start);
  400. APP_ERROR_CHECK(err_code);
  401. // Register a handler for BLE events.
  402. NRF_SDH_BLE_OBSERVER(m_ble_observer, APP_BLE_OBSERVER_PRIO, ble_evt_handler, NULL);
  403. }
  404. /**@brief Function for handling events from the GATT library. */
  405. void gatt_evt_handler(nrf_ble_gatt_t* p_gatt, nrf_ble_gatt_evt_t const* p_evt) {
  406. if (p_evt->evt_id == NRF_BLE_GATT_EVT_ATT_MTU_UPDATED) {
  407. NRF_LOG_INFO("ATT MTU exchange completed.");
  408. m_ble_nus_max_data_len = p_evt->params.att_mtu_effective - OPCODE_LENGTH - HANDLE_LENGTH;
  409. NRF_LOG_INFO("Ble NUS max data length set to 0x%X(%d)", m_ble_nus_max_data_len, m_ble_nus_max_data_len);
  410. }
  411. }
  412. /**@brief Function for initializing the GATT library. */
  413. void gatt_init(void) {
  414. ret_code_t err_code;
  415. err_code = nrf_ble_gatt_init(&m_gatt, gatt_evt_handler);
  416. APP_ERROR_CHECK(err_code);
  417. err_code = nrf_ble_gatt_att_mtu_central_set(&m_gatt, NRF_SDH_BLE_GATT_MAX_MTU_SIZE);
  418. APP_ERROR_CHECK(err_code);
  419. }
  420. /**@brief Function for handling events from the BSP module.
  421. *
  422. * @param[in] event Event generated by button press.
  423. */
  424. void bsp_event_handler(bsp_event_t event) {
  425. ret_code_t err_code;
  426. switch (event) {
  427. case BSP_EVENT_SLEEP:
  428. nrf_pwr_mgmt_shutdown(NRF_PWR_MGMT_SHUTDOWN_GOTO_SYSOFF);
  429. break;
  430. case BSP_EVENT_DISCONNECT:
  431. err_code = sd_ble_gap_disconnect(m_ble_nus_c.conn_handle, BLE_HCI_REMOTE_USER_TERMINATED_CONNECTION);
  432. if (err_code != NRF_ERROR_INVALID_STATE) {
  433. APP_ERROR_CHECK(err_code);
  434. }
  435. break;
  436. default:
  437. break;
  438. }
  439. }
  440. /**@brief Function for initializing the UART. */
  441. static void uart_init(void) {
  442. ret_code_t err_code;
  443. app_uart_comm_params_t const comm_params = {.rx_pin_no = RX_PIN_NUMBER, .tx_pin_no = TX_PIN_NUMBER, .rts_pin_no = RTS_PIN_NUMBER, .cts_pin_no = CTS_PIN_NUMBER, .flow_control = APP_UART_FLOW_CONTROL_DISABLED, .use_parity = false, .baud_rate = UART_BAUDRATE_BAUDRATE_Baud460800};
  444. APP_UART_FIFO_INIT(&comm_params, UART_RX_BUF_SIZE, UART_TX_BUF_SIZE, uart_event_handle, APP_IRQ_PRIORITY_LOWEST, err_code);
  445. APP_ERROR_CHECK(err_code);
  446. }
  447. /**@brief Function for initializing the Nordic UART Service (NUS) client. */
  448. static void nus_c_init(void) {
  449. ret_code_t err_code;
  450. ble_nus_c_init_t init;
  451. init.evt_handler = ble_nus_c_evt_handler;
  452. init.error_handler = nus_error_handler;
  453. init.p_gatt_queue = &m_ble_gatt_queue;
  454. err_code = ble_nus_c_init(&m_ble_nus_c, &init);
  455. APP_ERROR_CHECK(err_code);
  456. }
  457. /**@brief Function for initializing buttons and leds. */
  458. static void buttons_leds_init(void) {
  459. ret_code_t err_code;
  460. bsp_event_t startup_event;
  461. err_code = bsp_init(BSP_INIT_LEDS, bsp_event_handler);
  462. APP_ERROR_CHECK(err_code);
  463. err_code = bsp_btn_ble_init(NULL, &startup_event);
  464. APP_ERROR_CHECK(err_code);
  465. }
  466. /**@brief Function for initializing the timer. */
  467. static void timer_init(void) {
  468. ret_code_t err_code = app_timer_init();
  469. APP_ERROR_CHECK(err_code);
  470. }
  471. /**@brief Function for initializing the nrf log module. */
  472. static void log_init(void) {
  473. ret_code_t err_code = NRF_LOG_INIT(NULL);
  474. APP_ERROR_CHECK(err_code);
  475. NRF_LOG_DEFAULT_BACKENDS_INIT();
  476. }
  477. /**@brief Function for initializing power management.
  478. */
  479. static void power_management_init(void) {
  480. ret_code_t err_code;
  481. err_code = nrf_pwr_mgmt_init();
  482. APP_ERROR_CHECK(err_code);
  483. }
  484. /** @brief Function for initializing the database discovery module. */
  485. static void db_discovery_init(void) {
  486. ble_db_discovery_init_t db_init;
  487. memset(&db_init, 0, sizeof(ble_db_discovery_init_t));
  488. db_init.evt_handler = db_disc_handler;
  489. db_init.p_gatt_queue = &m_ble_gatt_queue;
  490. ret_code_t err_code = ble_db_discovery_init(&db_init);
  491. APP_ERROR_CHECK(err_code);
  492. }
  493. /**@brief Function for handling the idle state (main loop).
  494. *
  495. * @details Handles any pending log operations, then sleeps until the next event occurs.
  496. */
  497. static void idle_state_handle(void) {
  498. if (NRF_LOG_PROCESS() == false) {
  499. nrf_pwr_mgmt_run();
  500. }
  501. }
  502. APP_TIMER_DEF(m_uart_send_tmr); // ״̬��������ʱ��
  503. static void uart_send_tmr_cb(void* p_context) { //
  504. static uint32_t last_data_array_index;
  505. ret_code_t ret_val;
  506. if (data_array_index != 0 && data_array_index == last_data_array_index) {
  507. do {
  508. ret_val = ble_nus_c_string_send(&m_ble_nus_c, data_array, data_array_index);
  509. if ((ret_val != NRF_ERROR_INVALID_STATE) && (ret_val != NRF_ERROR_RESOURCES)) {
  510. APP_ERROR_CHECK(ret_val);
  511. }
  512. } while (ret_val == NRF_ERROR_RESOURCES);
  513. data_array_index = 0;
  514. last_data_array_index = 0;
  515. }
  516. last_data_array_index = data_array_index;
  517. }
  518. int main(void) {
  519. // Initialize.
  520. log_init();
  521. timer_init();
  522. uart_init();
  523. buttons_leds_init();
  524. db_discovery_init();
  525. power_management_init();
  526. ble_stack_init();
  527. gatt_init();
  528. nus_c_init();
  529. scan_init();
  530. app_timer_create(&m_uart_send_tmr, APP_TIMER_MODE_REPEATED, uart_send_tmr_cb);
  531. app_timer_start(m_uart_send_tmr, APP_TIMER_TICKS(2), NULL);
  532. // Start execution.
  533. // printf("BLE UART central example started.\r\n");
  534. NRF_LOG_INFO("BLE UART central example started.");
  535. scan_start();
  536. // Enter main loop.
  537. for (;;) {
  538. idle_state_handle();
  539. }
  540. }