| /** |
| * Copyright (c) 2024 Croxel, Inc. |
| * |
| * SPDX-License-Identifier: Apache-2.0 |
| */ |
| #include <zephyr/kernel.h> |
| |
| #include "common.h" |
| |
| #include "bs_types.h" |
| #include "bs_tracing.h" |
| #include "time_machine.h" |
| #include "bstests.h" |
| |
| #include <zephyr/types.h> |
| #include <zephyr/sys/printk.h> |
| |
| #include <zephyr/bluetooth/bluetooth.h> |
| #include <zephyr/bluetooth/conn.h> |
| |
| extern enum bst_result_t bst_result; |
| |
| static struct bt_conn *g_conn; |
| |
| CREATE_FLAG(flag_ext_adv_seen); |
| CREATE_FLAG(flag_connected); |
| CREATE_FLAG(flag_conn_recycled); |
| |
| static void connected(struct bt_conn *conn, uint8_t err) |
| { |
| char addr[BT_ADDR_LE_STR_LEN]; |
| |
| bt_addr_le_to_str(bt_conn_get_dst(conn), addr, sizeof(addr)); |
| |
| if (err != BT_HCI_ERR_SUCCESS) { |
| FAIL("Failed to connect to %s: %u\n", addr, err); |
| bt_conn_unref(g_conn); |
| g_conn = NULL; |
| return; |
| } |
| |
| printk("Connected to %s\n", addr); |
| SET_FLAG(flag_connected); |
| } |
| |
| static void free_conn_object_work_fn(struct k_work *work) |
| { |
| ARG_UNUSED(work); |
| |
| bt_conn_unref(g_conn); |
| g_conn = NULL; |
| } |
| |
| static K_WORK_DELAYABLE_DEFINE(free_conn_object_work, free_conn_object_work_fn); |
| |
| static void disconnected(struct bt_conn *conn, uint8_t reason) |
| { |
| char addr[BT_ADDR_LE_STR_LEN]; |
| |
| bt_addr_le_to_str(bt_conn_get_dst(conn), addr, sizeof(addr)); |
| |
| printk("Disconnected: %s (reason %u)\n", addr, reason); |
| |
| /* Schedule to cause de-sync between disconnected and recycled events, |
| * in order to prove the test is relying properly on it. |
| */ |
| k_work_schedule(&free_conn_object_work, K_MSEC(500)); |
| |
| UNSET_FLAG(flag_connected); |
| } |
| |
| static void recycled(void) |
| { |
| SET_FLAG(flag_conn_recycled); |
| } |
| |
| static struct bt_conn_cb conn_cbs = { |
| .connected = connected, |
| .disconnected = disconnected, |
| .recycled = recycled, |
| }; |
| |
| |
| static void scan_recv(const struct bt_le_scan_recv_info *info, |
| struct net_buf_simple *buf) |
| { |
| printk("Found advertisement. Adv-type: 0x%02x, Adv-prop: 0x%02x\n", |
| info->adv_type, info->adv_props); |
| |
| if (info->adv_type & BT_GAP_ADV_TYPE_EXT_ADV && |
| info->adv_props & BT_GAP_ADV_PROP_EXT_ADV) { |
| printk("Found extended advertisement!\n"); |
| SET_FLAG(flag_ext_adv_seen); |
| } |
| |
| if (!TEST_FLAG(flag_connected) && |
| info->adv_props & BT_GAP_ADV_PROP_CONNECTABLE) { |
| int err; |
| |
| printk("Stopping scan\n"); |
| err = bt_le_scan_stop(); |
| if (err) { |
| FAIL("Failed to stop scan: %d", err); |
| return; |
| } |
| |
| err = bt_conn_le_create(info->addr, BT_CONN_LE_CREATE_CONN, |
| BT_LE_CONN_PARAM_DEFAULT, &g_conn); |
| if (err) { |
| FAIL("Could not connect to peer: %d", err); |
| return; |
| } |
| } |
| } |
| |
| static struct bt_le_scan_cb scan_callbacks = { |
| .recv = scan_recv, |
| }; |
| |
| static void common_init(void) |
| { |
| int err = 0; |
| |
| err = bt_enable(NULL); |
| |
| if (err) { |
| FAIL("Bluetooth init failed: %d\n", err); |
| return; |
| } |
| |
| bt_conn_cb_register(&conn_cbs); |
| bt_le_scan_cb_register(&scan_callbacks); |
| |
| printk("Bluetooth initialized\n"); |
| } |
| |
| static void start_scan(void) |
| { |
| int err; |
| |
| printk("Start scanning..."); |
| err = bt_le_scan_start(BT_LE_SCAN_PASSIVE, NULL); |
| if (err) { |
| FAIL("Failed to start scan: %d\n", err); |
| return; |
| } |
| printk("done.\n"); |
| } |
| |
| static void main_ext_adv_scanner(void) |
| { |
| common_init(); |
| start_scan(); |
| |
| printk("Waiting for extended advertisements...\n"); |
| |
| WAIT_FOR_FLAG(flag_ext_adv_seen); |
| |
| PASS("Extended adv scanner passed\n"); |
| } |
| |
| static void scan_connect_and_disconnect_cycle(void) |
| { |
| start_scan(); |
| |
| printk("Waiting for extended advertisements...\n"); |
| WAIT_FOR_FLAG(flag_ext_adv_seen); |
| |
| printk("Waiting for connection with device...\n"); |
| WAIT_FOR_FLAG(flag_connected); |
| |
| printk("Waiting for device disconnection...\n"); |
| WAIT_FOR_FLAG_UNSET(flag_connected); |
| |
| printk("Waiting for Connection object to be recycled...\n"); |
| WAIT_FOR_FLAG(flag_conn_recycled); |
| |
| /* Iteration cleanup */ |
| printk("Clearing flag for seen extended advertisements...\n"); |
| UNSET_FLAG(flag_ext_adv_seen); |
| UNSET_FLAG(flag_conn_recycled); |
| } |
| |
| static void main_ext_adv_conn_scanner(void) |
| { |
| common_init(); |
| |
| scan_connect_and_disconnect_cycle(); |
| |
| start_scan(); |
| printk("Waiting to extended advertisements (again)...\n"); |
| WAIT_FOR_FLAG(flag_ext_adv_seen); |
| |
| PASS("Extended adv scanner passed\n"); |
| } |
| |
| static void main_ext_adv_conn_scanner_x5(void) |
| { |
| common_init(); |
| |
| for (size_t i = 0 ; i < 5 ; i++) { |
| printk("Iteration %d...\n", i); |
| scan_connect_and_disconnect_cycle(); |
| } |
| |
| start_scan(); |
| printk("Waiting to extended advertisements (again)...\n"); |
| WAIT_FOR_FLAG(flag_ext_adv_seen); |
| |
| PASS("Extended adv scanner x5 passed\n"); |
| } |
| |
| static const struct bst_test_instance ext_adv_scanner[] = { |
| { |
| .test_id = "ext_adv_scanner", |
| .test_descr = "Basic extended advertising scanning test. " |
| "Will just scan an extended advertiser.", |
| .test_post_init_f = test_init, |
| .test_tick_f = test_tick, |
| .test_main_f = main_ext_adv_scanner |
| }, |
| { |
| .test_id = "ext_adv_conn_scanner", |
| .test_descr = "Basic extended advertising scanning test. " |
| "Will scan an extended advertiser, connect " |
| "and verify it's detected after disconnection", |
| .test_post_init_f = test_init, |
| .test_tick_f = test_tick, |
| .test_main_f = main_ext_adv_conn_scanner |
| }, |
| { |
| .test_id = "ext_adv_conn_scanner_x5", |
| .test_descr = "Basic extended advertising scanning test. " |
| "Will scan an extended advertiser, connect " |
| "and verify it's detected after disconnection," |
| "repeated over 5 times", |
| .test_post_init_f = test_init, |
| .test_tick_f = test_tick, |
| .test_main_f = main_ext_adv_conn_scanner_x5 |
| }, |
| BSTEST_END_MARKER |
| }; |
| |
| struct bst_test_list *test_ext_adv_scanner(struct bst_test_list *tests) |
| { |
| return bst_add_tests(tests, ext_adv_scanner); |
| } |
| |
| bst_test_install_t test_installers[] = { |
| test_ext_adv_scanner, |
| NULL |
| }; |
| |
| int main(void) |
| { |
| bst_main(); |
| return 0; |
| } |