| /* |
| * Copyright (c) 2022 Nordic Semiconductor |
| * |
| * SPDX-License-Identifier: Apache-2.0 |
| */ |
| |
| #include <zephyr/kernel.h> |
| #include <zephyr/bluetooth/hci.h> |
| #include "mesh_test.h" |
| #include "mesh/adv.h" |
| |
| #define LOG_MODULE_NAME test_adv |
| |
| #include <zephyr/logging/log.h> |
| LOG_MODULE_REGISTER(LOG_MODULE_NAME, LOG_LEVEL_INF); |
| |
| #define WAIT_TIME 60 /*seconds*/ |
| |
| struct bt_mesh_test_adv { |
| uint8_t retr; |
| int64_t interval; |
| }; |
| |
| static struct bt_mesh_send_cb send_cb; |
| static struct bt_mesh_test_adv *xmit_param; |
| static struct k_sem observer_sem; |
| static const char txt_msg[] = "adv test"; |
| static const char cb_msg[] = "cb test"; |
| static int64_t tx_timestamp; |
| static int seq_checker; |
| |
| static void test_tx_init(void) |
| { |
| bt_mesh_test_cfg_set(NULL, WAIT_TIME); |
| } |
| |
| static void test_rx_init(void) |
| { |
| bt_mesh_test_cfg_set(NULL, WAIT_TIME); |
| } |
| |
| static void bt_init(void) |
| { |
| ASSERT_OK(bt_enable(NULL), "Bluetooth init failed"); |
| LOG_INF("Bluetooth initialized"); |
| } |
| |
| static void adv_init(void) |
| { |
| bt_mesh_adv_init(); |
| ASSERT_OK(bt_mesh_adv_enable(), "Mesh adv init failed"); |
| } |
| |
| static void single_start_cb(uint16_t duration, int err, void *cb_data) |
| { |
| int64_t delta; |
| |
| delta = k_uptime_delta(&tx_timestamp); |
| LOG_INF("tx start: +%d ms", delta); |
| ASSERT_TRUE(duration >= 90 && duration <= 200); |
| ASSERT_EQUAL(0, err); |
| ASSERT_EQUAL(cb_msg, cb_data); |
| ASSERT_EQUAL(0, seq_checker & 1); |
| seq_checker++; |
| } |
| |
| static void single_end_cb(int err, void *cb_data) |
| { |
| int64_t delta; |
| |
| delta = k_uptime_delta(&tx_timestamp); |
| LOG_INF("tx end: +%d ms", delta); |
| ASSERT_EQUAL(0, err); |
| ASSERT_EQUAL(cb_msg, cb_data); |
| ASSERT_EQUAL(1, seq_checker & 1); |
| seq_checker++; |
| k_sem_give(&observer_sem); |
| } |
| |
| static void realloc_end_cb(int err, void *cb_data) |
| { |
| struct net_buf *buf = (struct net_buf *)cb_data; |
| |
| ASSERT_EQUAL(0, err); |
| buf = bt_mesh_adv_create(BT_MESH_ADV_DATA, BT_MESH_LOCAL_ADV, |
| BT_MESH_TRANSMIT(2, 20), K_NO_WAIT); |
| ASSERT_FALSE(!buf, "Out of buffers"); |
| |
| k_sem_give(&observer_sem); |
| } |
| |
| static void seq_start_cb(uint16_t duration, int err, void *cb_data) |
| { |
| ASSERT_EQUAL(0, err); |
| ASSERT_EQUAL(seq_checker, (intptr_t)cb_data); |
| } |
| |
| static void seq_end_cb(int err, void *cb_data) |
| { |
| ASSERT_EQUAL(0, err); |
| ASSERT_EQUAL(seq_checker, (intptr_t)cb_data); |
| seq_checker++; |
| |
| if (seq_checker == CONFIG_BT_MESH_ADV_BUF_COUNT) { |
| k_sem_give(&observer_sem); |
| } |
| } |
| |
| static void xmit_scan_cb(const bt_addr_le_t *addr, int8_t rssi, uint8_t adv_type, |
| struct net_buf_simple *buf) |
| { |
| uint8_t length; |
| static int cnt; |
| static int64_t buff_timestamp; |
| int64_t timestamp = k_uptime_get(); |
| static int64_t delta; |
| |
| ASSERT_TRUE(xmit_param, "Xmit checker didn't get parameters"); |
| ASSERT_EQUAL(BT_GAP_ADV_TYPE_ADV_NONCONN_IND, adv_type); |
| |
| length = net_buf_simple_pull_u8(buf); |
| ASSERT_EQUAL(buf->len, length); |
| ASSERT_EQUAL(length, sizeof(uint8_t) + sizeof(txt_msg)); |
| ASSERT_EQUAL(BT_DATA_MESH_MESSAGE, net_buf_simple_pull_u8(buf)); |
| |
| char *data = net_buf_simple_pull_mem(buf, sizeof(txt_msg)); |
| |
| ASSERT_EQUAL(0, memcmp(txt_msg, data, sizeof(txt_msg))); |
| |
| cnt++; |
| |
| if (cnt > 1) { |
| delta = k_uptime_delta(&buff_timestamp); |
| |
| ASSERT_TRUE(delta >= xmit_param->interval && |
| delta < (xmit_param->interval + xmit_param->interval / 2)); |
| } |
| |
| LOG_INF("rx: %s +%d ms", txt_msg, delta); |
| |
| buff_timestamp = timestamp; |
| |
| if (cnt > xmit_param->retr) { |
| cnt = 0; |
| delta = 0; |
| buff_timestamp = 0; |
| k_sem_give(&observer_sem); |
| } |
| } |
| |
| static void rx_xmit_adv(void) |
| { |
| struct bt_le_scan_param scan_param = { |
| .type = BT_HCI_LE_SCAN_PASSIVE, |
| .options = BT_LE_SCAN_OPT_NONE, |
| .interval = BT_MESH_ADV_SCAN_UNIT(1000), |
| .window = BT_MESH_ADV_SCAN_UNIT(1000) |
| }; |
| int err; |
| |
| k_sem_init(&observer_sem, 0, 1); |
| err = bt_le_scan_start(&scan_param, xmit_scan_cb); |
| ASSERT_FALSE(err && err != -EALREADY, "starting scan failed (err %d)", err); |
| |
| err = k_sem_take(&observer_sem, |
| K_MSEC(((xmit_param->retr + 1) * xmit_param->interval) + 10)); |
| ASSERT_OK(err, "Didn't receive adv in time"); |
| |
| err = bt_le_scan_stop(); |
| ASSERT_FALSE(err && err != -EALREADY, "stopping scan failed (err %d)", err); |
| } |
| |
| static void test_tx_cb_single(void) |
| { |
| struct net_buf *buf; |
| int err; |
| |
| bt_init(); |
| adv_init(); |
| |
| buf = bt_mesh_adv_create(BT_MESH_ADV_DATA, BT_MESH_LOCAL_ADV, |
| BT_MESH_TRANSMIT(2, 20), K_NO_WAIT); |
| ASSERT_FALSE(!buf, "Out of buffers"); |
| |
| send_cb.start = single_start_cb; |
| send_cb.end = single_end_cb; |
| |
| k_sem_init(&observer_sem, 0, 1); |
| net_buf_add_mem(buf, txt_msg, sizeof(txt_msg)); |
| seq_checker = 0; |
| tx_timestamp = k_uptime_get(); |
| bt_mesh_adv_send(buf, &send_cb, (void *)cb_msg); |
| net_buf_unref(buf); |
| |
| err = k_sem_take(&observer_sem, K_SECONDS(1)); |
| ASSERT_OK(err, "Didn't call end tx cb."); |
| |
| PASS(); |
| } |
| |
| static void test_rx_xmit(void) |
| { |
| struct bt_mesh_test_adv param = { |
| .retr = 2, |
| .interval = 20 |
| }; |
| |
| bt_init(); |
| xmit_param = ¶m; |
| rx_xmit_adv(); |
| PASS(); |
| } |
| |
| static void test_tx_cb_multi(void) |
| { |
| struct net_buf *buf[CONFIG_BT_MESH_ADV_BUF_COUNT]; |
| struct net_buf *dummy_buf; |
| int err; |
| |
| bt_init(); |
| adv_init(); |
| k_sem_init(&observer_sem, 0, 1); |
| |
| /* Allocate all network buffers. */ |
| for (int i = 0; i < CONFIG_BT_MESH_ADV_BUF_COUNT; i++) { |
| buf[i] = bt_mesh_adv_create(BT_MESH_ADV_DATA, BT_MESH_LOCAL_ADV, |
| BT_MESH_TRANSMIT(2, 20), K_NO_WAIT); |
| ASSERT_FALSE(!buf[i], "Out of buffers"); |
| } |
| |
| dummy_buf = bt_mesh_adv_create(BT_MESH_ADV_DATA, BT_MESH_LOCAL_ADV, |
| BT_MESH_TRANSMIT(2, 20), K_NO_WAIT); |
| ASSERT_TRUE(!dummy_buf, "Unexpected extra buffer"); |
| |
| /* Start single adv to reallocate one network buffer in callback. |
| * Check that the buffer is freed before cb is triggered. |
| */ |
| send_cb.start = NULL; |
| send_cb.end = realloc_end_cb; |
| net_buf_add_mem(buf[0], txt_msg, sizeof(txt_msg)); |
| |
| bt_mesh_adv_send(buf[0], &send_cb, buf[0]); |
| net_buf_unref(buf[0]); |
| |
| err = k_sem_take(&observer_sem, K_SECONDS(1)); |
| ASSERT_OK(err, "Didn't call the end tx cb that reallocates buffer one more time."); |
| |
| /* Start multi advs to check that all buffers are sent and cbs are triggered. */ |
| send_cb.start = seq_start_cb; |
| send_cb.end = seq_end_cb; |
| seq_checker = 0; |
| |
| for (int i = 0; i < CONFIG_BT_MESH_ADV_BUF_COUNT; i++) { |
| net_buf_add_le32(buf[i], i); |
| bt_mesh_adv_send(buf[i], &send_cb, (void *)(intptr_t)i); |
| net_buf_unref(buf[i]); |
| } |
| |
| err = k_sem_take(&observer_sem, K_SECONDS(10)); |
| ASSERT_OK(err, "Didn't call the last end tx cb."); |
| |
| PASS(); |
| } |
| |
| #define TEST_CASE(role, name, description) \ |
| { \ |
| .test_id = "adv_" #role "_" #name, \ |
| .test_descr = description, \ |
| .test_pre_init_f = test_##role##_init, \ |
| .test_tick_f = bt_mesh_test_timeout, \ |
| .test_main_f = test_##role##_##name, \ |
| } |
| |
| static const struct bst_test_instance test_adv[] = { |
| TEST_CASE(tx, cb_single, "ADV: tx cb parameter checker"), |
| TEST_CASE(tx, cb_multi, "ADV: tx cb sequence checker"), |
| |
| TEST_CASE(rx, xmit, "ADV: xmit checker"), |
| BSTEST_END_MARKER |
| }; |
| |
| struct bst_test_list *test_adv_install(struct bst_test_list *tests) |
| { |
| tests = bst_add_tests(tests, test_adv); |
| return tests; |
| } |