1 /*
2 * Copyright (c) 2022 Nordic Semiconductor ASA
3 *
4 * SPDX-License-Identifier: Apache-2.0
5 */
6
7 #include "mocks/crypto.h"
8 #include "mocks/crypto_expects.h"
9 #include "mocks/hci_core.h"
10 #include "mocks/rpa.h"
11 #include "mocks/rpa_expects.h"
12 #include "testing_common_defs.h"
13
14 #include <zephyr/bluetooth/hci.h>
15 #include <zephyr/fff.h>
16 #include <zephyr/kernel.h>
17
18 #include <host/hci_core.h>
19 #include <host/id.h>
20
21 DEFINE_FFF_GLOBALS;
22
fff_reset_rule_before(const struct ztest_unit_test * test,void * fixture)23 static void fff_reset_rule_before(const struct ztest_unit_test *test, void *fixture)
24 {
25 memset(&bt_dev, 0x00, sizeof(struct bt_dev));
26 bt_addr_le_copy(&bt_dev.random_addr, &bt_addr_le_none);
27 #if defined(CONFIG_BT_RPA_SHARING)
28 bt_addr_copy(&bt_dev.rpa[BT_ID_DEFAULT], BT_ADDR_NONE);
29 #endif
30
31 RPA_FFF_FAKES_LIST(RESET_FAKE);
32 CRYPTO_FFF_FAKES_LIST(RESET_FAKE);
33 HCI_CORE_FFF_FAKES_LIST(RESET_FAKE);
34 }
35
36 ZTEST_RULE(fff_reset_rule, fff_reset_rule_before, NULL);
37
38 ZTEST_SUITE(bt_id_set_adv_private_addr, NULL, NULL, NULL, NULL, NULL);
39
bt_rand_custom_fake(void * buf,size_t len)40 static int bt_rand_custom_fake(void *buf, size_t len)
41 {
42 __ASSERT_NO_MSG(buf != NULL);
43 __ASSERT_NO_MSG(len == sizeof(BT_ADDR->val));
44
45 /* This will make set_random_address() succeeds and returns 0 */
46 memcpy(buf, &BT_ADDR->val, len);
47 bt_addr_copy(&bt_dev.random_addr.a, BT_ADDR);
48
49 return 0;
50 }
51
bt_rpa_create_custom_fake(const uint8_t irk[16],bt_addr_t * rpa)52 static int bt_rpa_create_custom_fake(const uint8_t irk[16], bt_addr_t *rpa)
53 {
54 __ASSERT_NO_MSG(irk != NULL);
55 __ASSERT_NO_MSG(rpa != NULL);
56
57 /* This will make set_random_address() succeeds and returns 0 */
58 bt_addr_copy(rpa, &BT_RPA_LE_ADDR->a);
59 bt_addr_copy(&bt_dev.random_addr.a, &BT_RPA_LE_ADDR->a);
60
61 return 0;
62 }
63
64 /*
65 * Test setting advertising private address with a valid advertise parameters reference while
66 * 'CONFIG_BT_PRIVACY' is enabled and 'CONFIG_BT_EXT_ADV' isn't enabled
67 *
68 * Constraints:
69 * - A valid advertise parameters ID is used (<= CONFIG_BT_ID_MAX)
70 * - 'CONFIG_BT_PRIVACY' is enabled
71 * - 'CONFIG_BT_EXT_ADV' isn't enabled
72 *
73 * Expected behaviour:
74 * - bt_id_set_adv_private_addr() returns 0 (success)
75 */
ZTEST(bt_id_set_adv_private_addr,test_set_adv_private_address_with_valid_ref_privacy_succeeds)76 ZTEST(bt_id_set_adv_private_addr, test_set_adv_private_address_with_valid_ref_privacy_succeeds)
77 {
78 int err;
79 struct bt_le_ext_adv adv_param = {0};
80
81 Z_TEST_SKIP_IFNDEF(CONFIG_BT_PRIVACY);
82 Z_TEST_SKIP_IFDEF(CONFIG_BT_EXT_ADV);
83
84 bt_rpa_create_fake.custom_fake = bt_rpa_create_custom_fake;
85
86 err = bt_id_set_adv_private_addr(&adv_param);
87
88 #if defined(CONFIG_BT_PRIVACY)
89 expect_single_call_bt_rpa_create(bt_dev.irk[adv_param.id]);
90 #endif
91
92 zassert_true(atomic_test_bit(bt_dev.flags, BT_DEV_RPA_VALID),
93 "Flags were not correctly set");
94
95 zassert_ok(err, "Unexpected error code '%d' was returned", err);
96 }
97
98 /*
99 * Test setting advertising private address with a valid advertise parameters reference while
100 * 'CONFIG_BT_PRIVACY' and 'CONFIG_BT_EXT_ADV' are enabled
101 *
102 * Constraints:
103 * - A valid advertise parameters ID is used (<= CONFIG_BT_ID_MAX)
104 * - 'CONFIG_BT_PRIVACY' is enabled
105 * - 'CONFIG_BT_EXT_ADV' is enabled
106 *
107 * Expected behaviour:
108 * - bt_id_set_adv_private_addr() returns 0 (success)
109 */
ZTEST(bt_id_set_adv_private_addr,test_set_adv_private_address_with_valid_ref_priv_ext_adv_succeeds)110 ZTEST(bt_id_set_adv_private_addr, test_set_adv_private_address_with_valid_ref_priv_ext_adv_succeeds)
111 {
112 int err;
113 struct bt_le_ext_adv adv_param = {0};
114
115 Z_TEST_SKIP_IFNDEF(CONFIG_BT_PRIVACY);
116 Z_TEST_SKIP_IFNDEF(CONFIG_BT_EXT_ADV);
117
118 bt_rpa_create_fake.custom_fake = bt_rpa_create_custom_fake;
119
120 err = bt_id_set_adv_private_addr(&adv_param);
121
122 #if defined(CONFIG_BT_PRIVACY)
123 expect_single_call_bt_rpa_create(bt_dev.irk[adv_param.id]);
124 #endif
125
126 #if defined(CONFIG_BT_EXT_ADV)
127 zassert_true(atomic_test_bit(adv_param.flags, BT_ADV_RANDOM_ADDR_PENDING),
128 "Flags were not correctly set");
129
130 zassert_mem_equal(&adv_param.random_addr, BT_RPA_LE_ADDR, sizeof(bt_addr_le_t),
131 "Incorrect address was set");
132 #endif
133
134 zassert_true(atomic_test_bit(adv_param.flags, BT_ADV_RPA_VALID),
135 "Flags were not correctly set");
136
137 zassert_ok(err, "Unexpected error code '%d' was returned", err);
138 }
139
140 /*
141 * Test setting advertising private address with a valid advertise parameters reference after it
142 * was set previously while 'CONFIG_BT_PRIVACY' and 'CONFIG_BT_EXT_ADV' are enabled
143 *
144 * Constraints:
145 * - A valid advertise parameters ID is used (<= CONFIG_BT_ID_MAX)
146 * - 'CONFIG_BT_PRIVACY' is enabled
147 * - 'CONFIG_BT_EXT_ADV' is enabled
148 * - 'BT_ADV_RPA_VALID' flag in advertise parameters reference is set
149 *
150 * Expected behaviour:
151 * - bt_id_set_adv_private_addr() returns 0 (success)
152 */
ZTEST(bt_id_set_adv_private_addr,test_set_adv_private_address_with_valid_ref_priv_ext_adv_again)153 ZTEST(bt_id_set_adv_private_addr, test_set_adv_private_address_with_valid_ref_priv_ext_adv_again)
154 {
155 int err;
156 struct bt_le_ext_adv adv_param = {0};
157
158 Z_TEST_SKIP_IFNDEF(CONFIG_BT_PRIVACY);
159 Z_TEST_SKIP_IFNDEF(CONFIG_BT_EXT_ADV);
160
161 atomic_set_bit(adv_param.flags, BT_ADV_RPA_VALID);
162
163 err = bt_id_set_adv_private_addr(&adv_param);
164
165 expect_not_called_bt_rpa_create();
166
167 zassert_ok(err, "Unexpected error code '%d' was returned", err);
168 }
169
170 /*
171 * Test setting advertising private address with a valid advertise parameters reference while
172 * 'CONFIG_BT_PRIVACY' isn't enabled
173 *
174 * Constraints:
175 * - 'CONFIG_BT_PRIVACY' isn't enabled
176 *
177 * Expected behaviour:
178 * - bt_id_set_adv_private_addr() returns 0 (success)
179 */
ZTEST(bt_id_set_adv_private_addr,test_set_adv_private_address_with_valid_ref_succeeds)180 ZTEST(bt_id_set_adv_private_addr, test_set_adv_private_address_with_valid_ref_succeeds)
181 {
182 int err;
183 struct bt_le_ext_adv adv_param;
184
185 Z_TEST_SKIP_IFDEF(CONFIG_BT_PRIVACY);
186
187 bt_rand_fake.custom_fake = bt_rand_custom_fake;
188
189 err = bt_id_set_adv_private_addr(&adv_param);
190
191 expect_not_called_bt_rpa_create();
192
193 zassert_ok(err, "Unexpected error code '%d' was returned", err);
194 }
195