blob: 2ff2fc28cdc198259ad27c121f263f6deef2a1b6 [file] [log] [blame]
Jacob Erlbecke04e0b02015-05-06 18:30:48 +02001/*
2 * MsTest.cpp
3 *
4 * Copyright (C) 2015 by Sysmocom s.f.m.c. GmbH
5 *
6 * All Rights Reserved
7 *
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU Affero General Public License as published by
10 * the Free Software Foundation; either version 3 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU Affero General Public License
19 * along with this program. If not, see <http://www.gnu.org/licenses/>.
20 *
21 */
22
23#include "tbf.h"
Pau Espin Pedrol9d1cdb12019-09-25 17:47:02 +020024#include "tbf_ul.h"
Jacob Erlbecke04e0b02015-05-06 18:30:48 +020025#include "gprs_debug.h"
26#include "gprs_ms.h"
Jacob Erlbeck53670862015-05-12 17:54:33 +020027#include "gprs_ms_storage.h"
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +020028#include "bts.h"
Jacob Erlbecke04e0b02015-05-06 18:30:48 +020029
30extern "C" {
31#include "pcu_vty.h"
32
33#include <osmocom/core/application.h>
34#include <osmocom/core/msgb.h>
35#include <osmocom/core/talloc.h>
36#include <osmocom/core/utils.h>
37#include <osmocom/vty/vty.h>
38}
39
40#include <errno.h>
Jacob Erlbeckd9e10242015-05-28 15:43:53 +020041#include <unistd.h>
Jacob Erlbecke04e0b02015-05-06 18:30:48 +020042
Pau Espin Pedrola0cbde72020-03-26 15:56:15 +010043#include <new>
44
Jacob Erlbecke04e0b02015-05-06 18:30:48 +020045void *tall_pcu_ctx;
46int16_t spoof_mnc = 0, spoof_mcc = 0;
Neels Hofmeyrbdc55fa2018-02-21 00:39:07 +010047bool spoof_mnc_3_digits = false;
Jacob Erlbecke04e0b02015-05-06 18:30:48 +020048
Pau Espin Pedrol38f80be2021-05-14 14:33:28 +020049static int ul_tbf_dtor(struct gprs_rlcmac_ul_tbf *tbf)
50{
51 tbf->~gprs_rlcmac_ul_tbf();
52 return 0;
53}
54
55static int dl_tbf_dtor(struct gprs_rlcmac_dl_tbf *tbf)
56{
57 tbf->~gprs_rlcmac_dl_tbf();
58 return 0;
59}
60
61static gprs_rlcmac_ul_tbf *alloc_ul_tbf(struct gprs_rlcmac_bts *bts, GprsMs *ms)
62{
63 gprs_rlcmac_ul_tbf *ul_tbf;
64 ul_tbf = talloc_zero(tall_pcu_ctx, struct gprs_rlcmac_ul_tbf);
65 talloc_set_destructor(ul_tbf, ul_tbf_dtor);
66 new (ul_tbf) gprs_rlcmac_ul_tbf(bts, ms);
67 return ul_tbf;
68}
69
70static gprs_rlcmac_dl_tbf *alloc_dl_tbf(struct gprs_rlcmac_bts *bts, GprsMs *ms)
71{
72 gprs_rlcmac_dl_tbf *dl_tbf;
73 dl_tbf = talloc_zero(tall_pcu_ctx, struct gprs_rlcmac_dl_tbf);
74 talloc_set_destructor(dl_tbf, dl_tbf_dtor);
75 new (dl_tbf) gprs_rlcmac_dl_tbf(bts, ms);
76 return dl_tbf;
77}
78
Jacob Erlbecke04e0b02015-05-06 18:30:48 +020079static void test_ms_state()
80{
81 uint32_t tlli = 0xffeeddbb;
82 gprs_rlcmac_dl_tbf *dl_tbf;
83 gprs_rlcmac_ul_tbf *ul_tbf;
Pau Espin Pedrold1049dc2021-01-18 17:14:14 +010084 struct gprs_rlcmac_bts *bts = bts_alloc(the_pcu, 0);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +020085 GprsMs *ms;
86
87 printf("=== start %s ===\n", __func__);
88
Pau Espin Pedrol2182e622021-01-14 16:48:38 +010089 ms = ms_alloc(bts, tlli);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +010090 OSMO_ASSERT(ms_is_idle(ms));
Jacob Erlbecke04e0b02015-05-06 18:30:48 +020091
Pau Espin Pedrol38f80be2021-05-14 14:33:28 +020092 dl_tbf = alloc_dl_tbf(bts, ms);
93 ul_tbf = alloc_ul_tbf(bts, ms);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +020094
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +010095 ms_attach_tbf(ms, ul_tbf);
96 OSMO_ASSERT(!ms_is_idle(ms));
97 OSMO_ASSERT(ms_ul_tbf(ms) == ul_tbf);
98 OSMO_ASSERT(ms_dl_tbf(ms) == NULL);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +020099
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100100 ms_attach_tbf(ms, dl_tbf);
101 OSMO_ASSERT(!ms_is_idle(ms));
102 OSMO_ASSERT(ms_ul_tbf(ms) == ul_tbf);
103 OSMO_ASSERT(ms_dl_tbf(ms) == dl_tbf);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200104
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100105 OSMO_ASSERT(ms_tbf(ms, GPRS_RLCMAC_UL_TBF) == ul_tbf);
106 OSMO_ASSERT(ms_tbf(ms, GPRS_RLCMAC_DL_TBF) == dl_tbf);
Jacob Erlbeckac89a552015-06-29 14:18:46 +0200107
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100108 ms_detach_tbf(ms, ul_tbf);
109 OSMO_ASSERT(!ms_is_idle(ms));
110 OSMO_ASSERT(ms_ul_tbf(ms) == NULL);
111 OSMO_ASSERT(ms_dl_tbf(ms) == dl_tbf);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200112
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100113 ms_detach_tbf(ms, dl_tbf);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200114 /* The ms object is freed now */
115 ms = NULL;
116
117 talloc_free(dl_tbf);
118 talloc_free(ul_tbf);
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100119 talloc_free(bts);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200120 printf("=== end %s ===\n", __func__);
121}
122
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100123static enum {CB_UNKNOWN, CB_IS_IDLE, CB_IS_ACTIVE} last_cb = CB_UNKNOWN;
124static void ms_idle_cb(struct GprsMs *ms)
125{
126 OSMO_ASSERT(ms_is_idle(ms));
127 printf(" ms_idle() was called\n");
128 last_cb = CB_IS_IDLE;
129}
130static void ms_active_cb(struct GprsMs *ms)
131{
132 OSMO_ASSERT(!ms_is_idle(ms));
133 printf(" ms_active() was called\n");
134 last_cb = CB_IS_ACTIVE;
135}
136static struct gpr_ms_callback ms_cb = {
137 .ms_idle = ms_idle_cb,
138 .ms_active = ms_active_cb
139};
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200140static void test_ms_callback()
141{
142 uint32_t tlli = 0xffeeddbb;
143 gprs_rlcmac_dl_tbf *dl_tbf;
144 gprs_rlcmac_ul_tbf *ul_tbf;
Pau Espin Pedrold1049dc2021-01-18 17:14:14 +0100145 struct gprs_rlcmac_bts *bts = bts_alloc(the_pcu, 0);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200146 GprsMs *ms;
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100147 last_cb = CB_UNKNOWN;
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200148
149 printf("=== start %s ===\n", __func__);
150
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100151 ms = ms_alloc(bts, tlli);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100152 ms_set_callback(ms, &ms_cb);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200153
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100154 OSMO_ASSERT(ms_is_idle(ms));
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200155
Pau Espin Pedrol38f80be2021-05-14 14:33:28 +0200156 dl_tbf = alloc_dl_tbf(bts, ms);
157 ul_tbf = alloc_ul_tbf(bts, ms);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200158
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100159 OSMO_ASSERT(last_cb == CB_UNKNOWN);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200160
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100161 ms_attach_tbf(ms, ul_tbf);
162 OSMO_ASSERT(!ms_is_idle(ms));
163 OSMO_ASSERT(ms_ul_tbf(ms) == ul_tbf);
164 OSMO_ASSERT(ms_dl_tbf(ms) == NULL);
165 OSMO_ASSERT(last_cb == CB_IS_ACTIVE);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200166
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100167 last_cb = CB_UNKNOWN;
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200168
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100169 ms_attach_tbf(ms, dl_tbf);
170 OSMO_ASSERT(!ms_is_idle(ms));
171 OSMO_ASSERT(ms_ul_tbf(ms) == ul_tbf);
172 OSMO_ASSERT(ms_dl_tbf(ms) == dl_tbf);
173 OSMO_ASSERT(last_cb == CB_UNKNOWN);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200174
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100175 ms_detach_tbf(ms, ul_tbf);
176 OSMO_ASSERT(!ms_is_idle(ms));
177 OSMO_ASSERT(ms_ul_tbf(ms) == NULL);
178 OSMO_ASSERT(ms_dl_tbf(ms) == dl_tbf);
179 OSMO_ASSERT(last_cb == CB_UNKNOWN);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200180
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100181 ms_detach_tbf(ms, dl_tbf);
182 OSMO_ASSERT(ms_is_idle(ms));
183 OSMO_ASSERT(ms_ul_tbf(ms) == NULL);
184 OSMO_ASSERT(ms_dl_tbf(ms) == NULL);
185 OSMO_ASSERT(last_cb == CB_IS_IDLE);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200186
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100187 last_cb = CB_UNKNOWN;
188 talloc_free(ms);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200189
190 talloc_free(dl_tbf);
191 talloc_free(ul_tbf);
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100192 talloc_free(bts);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200193 printf("=== end %s ===\n", __func__);
194}
195
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100196static bool was_idle;
197static void ms_replace_tbf_idle_cb(struct GprsMs *ms)
198{
199 OSMO_ASSERT(ms_is_idle(ms));
200 printf(" ms_idle() was called\n");
201 was_idle = true;
202}
203static void ms_replace_tbf_active_cb(struct GprsMs *ms)
204{
205 OSMO_ASSERT(!ms_is_idle(ms));
206 printf(" ms_active() was called\n");
207}
208static struct gpr_ms_callback ms_replace_tbf_cb = {
209 .ms_idle = ms_replace_tbf_idle_cb,
210 .ms_active = ms_replace_tbf_active_cb
211};
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200212static void test_ms_replace_tbf()
213{
214 uint32_t tlli = 0xffeeddbb;
215 gprs_rlcmac_dl_tbf *dl_tbf[2];
216 gprs_rlcmac_ul_tbf *ul_tbf;
Pau Espin Pedrold1049dc2021-01-18 17:14:14 +0100217 struct gprs_rlcmac_bts *bts = bts_alloc(the_pcu, 0);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200218 GprsMs *ms;
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200219
220 printf("=== start %s ===\n", __func__);
221
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100222 ms = ms_alloc(bts, tlli);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100223 ms_set_callback(ms, &ms_replace_tbf_cb);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200224
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100225 OSMO_ASSERT(ms_is_idle(ms));
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200226 was_idle = false;
227
Pau Espin Pedrol38f80be2021-05-14 14:33:28 +0200228 dl_tbf[0] = alloc_dl_tbf(bts, ms);
229 dl_tbf[1] = alloc_dl_tbf(bts, ms);
230 ul_tbf = alloc_ul_tbf(bts, ms);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200231
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100232 ms_attach_tbf(ms, dl_tbf[0]);
233 OSMO_ASSERT(!ms_is_idle(ms));
234 OSMO_ASSERT(ms_ul_tbf(ms) == NULL);
235 OSMO_ASSERT(ms_dl_tbf(ms) == dl_tbf[0]);
236 OSMO_ASSERT(llist_empty(&ms->old_tbfs));
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200237 OSMO_ASSERT(!was_idle);
238
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100239 ms_attach_tbf(ms, dl_tbf[1]);
240 OSMO_ASSERT(!ms_is_idle(ms));
241 OSMO_ASSERT(ms_ul_tbf(ms) == NULL);
242 OSMO_ASSERT(ms_dl_tbf(ms) == dl_tbf[1]);
243 OSMO_ASSERT(!llist_empty(&ms->old_tbfs));
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200244 OSMO_ASSERT(!was_idle);
245
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100246 ms_attach_tbf(ms, ul_tbf);
247 OSMO_ASSERT(!ms_is_idle(ms));
248 OSMO_ASSERT(ms_ul_tbf(ms) == ul_tbf);
249 OSMO_ASSERT(ms_dl_tbf(ms) == dl_tbf[1]);
250 OSMO_ASSERT(!llist_empty(&ms->old_tbfs));
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200251 OSMO_ASSERT(!was_idle);
252
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100253 ms_detach_tbf(ms, ul_tbf);
254 OSMO_ASSERT(!ms_is_idle(ms));
255 OSMO_ASSERT(ms_ul_tbf(ms) == NULL);
256 OSMO_ASSERT(ms_dl_tbf(ms) == dl_tbf[1]);
257 OSMO_ASSERT(!llist_empty(&ms->old_tbfs));
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200258 OSMO_ASSERT(!was_idle);
259
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100260 ms_detach_tbf(ms, dl_tbf[0]);
261 OSMO_ASSERT(!ms_is_idle(ms));
262 OSMO_ASSERT(ms_ul_tbf(ms) == NULL);
263 OSMO_ASSERT(ms_dl_tbf(ms) == dl_tbf[1]);
264 OSMO_ASSERT(llist_empty(&ms->old_tbfs));
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200265 OSMO_ASSERT(!was_idle);
266
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100267 ms_detach_tbf(ms, dl_tbf[1]);
268 OSMO_ASSERT(ms_is_idle(ms));
269 OSMO_ASSERT(ms_ul_tbf(ms) == NULL);
270 OSMO_ASSERT(ms_dl_tbf(ms) == NULL);
271 OSMO_ASSERT(llist_empty(&ms->old_tbfs));
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200272 OSMO_ASSERT(was_idle);
273
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100274 talloc_free(ms);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200275
276 talloc_free(dl_tbf[0]);
277 talloc_free(dl_tbf[1]);
278 talloc_free(ul_tbf);
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100279 talloc_free(bts);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200280 printf("=== end %s ===\n", __func__);
281}
282
Jacob Erlbeck93990462015-05-15 15:50:43 +0200283static void test_ms_change_tlli()
284{
285 uint32_t start_tlli = 0xaa000000;
286 uint32_t new_ms_tlli = 0xff001111;
287 uint32_t other_sgsn_tlli = 0xff00eeee;
Pau Espin Pedrold1049dc2021-01-18 17:14:14 +0100288 struct gprs_rlcmac_bts *bts = bts_alloc(the_pcu, 0);
Jacob Erlbeck93990462015-05-15 15:50:43 +0200289 GprsMs *ms;
290
291 printf("=== start %s ===\n", __func__);
292
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100293 ms = ms_alloc(bts, start_tlli);
Jacob Erlbeck93990462015-05-15 15:50:43 +0200294
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100295 OSMO_ASSERT(ms_is_idle(ms));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200296
297 /* MS announces TLLI, SGSN uses it immediately */
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100298 ms_set_tlli(ms, new_ms_tlli);
299 OSMO_ASSERT(ms_tlli(ms) == new_ms_tlli);
300 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
301 OSMO_ASSERT(ms_check_tlli(ms, start_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200302
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100303 ms_confirm_tlli(ms, new_ms_tlli);
304 OSMO_ASSERT(ms_tlli(ms) == new_ms_tlli);
305 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
306 OSMO_ASSERT(!ms_check_tlli(ms, start_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200307
308 /* MS announces TLLI, SGSN uses it later */
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100309 ms_set_tlli(ms, start_tlli);
310 ms_confirm_tlli(ms, start_tlli);
Jacob Erlbeck93990462015-05-15 15:50:43 +0200311
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100312 ms_set_tlli(ms, new_ms_tlli);
313 OSMO_ASSERT(ms_tlli(ms) == new_ms_tlli);
314 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
315 OSMO_ASSERT(ms_check_tlli(ms, start_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200316
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100317 ms_confirm_tlli(ms, start_tlli);
318 OSMO_ASSERT(ms_tlli(ms) == new_ms_tlli);
319 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
320 OSMO_ASSERT(ms_check_tlli(ms, start_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200321
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100322 ms_set_tlli(ms, new_ms_tlli);
323 OSMO_ASSERT(ms_tlli(ms) == new_ms_tlli);
324 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
325 OSMO_ASSERT(ms_check_tlli(ms, start_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200326
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100327 ms_confirm_tlli(ms, new_ms_tlli);
328 OSMO_ASSERT(ms_tlli(ms) == new_ms_tlli);
329 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
330 OSMO_ASSERT(!ms_check_tlli(ms, start_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200331
332 /* MS announces TLLI, SGSN uses it later after another new TLLI */
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100333 ms_set_tlli(ms, start_tlli);
334 ms_confirm_tlli(ms, start_tlli);
Jacob Erlbeck93990462015-05-15 15:50:43 +0200335
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100336 ms_set_tlli(ms, new_ms_tlli);
337 OSMO_ASSERT(ms_tlli(ms) == new_ms_tlli);
338 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
339 OSMO_ASSERT(ms_check_tlli(ms, start_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200340
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100341 ms_confirm_tlli(ms, other_sgsn_tlli);
342 OSMO_ASSERT(ms_tlli(ms) == new_ms_tlli);
343 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
344 OSMO_ASSERT(ms_check_tlli(ms, other_sgsn_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200345
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100346 ms_set_tlli(ms, new_ms_tlli);
347 OSMO_ASSERT(ms_tlli(ms) == new_ms_tlli);
348 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
349 OSMO_ASSERT(ms_check_tlli(ms, other_sgsn_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200350
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100351 ms_confirm_tlli(ms, new_ms_tlli);
352 OSMO_ASSERT(ms_tlli(ms) == new_ms_tlli);
353 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
354 OSMO_ASSERT(!ms_check_tlli(ms, start_tlli));
355 OSMO_ASSERT(!ms_check_tlli(ms, other_sgsn_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200356
357 /* SGSN uses the new TLLI before it is announced by the MS (shouldn't
358 * happen in normal use) */
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100359 ms_set_tlli(ms, start_tlli);
360 ms_confirm_tlli(ms, start_tlli);
Jacob Erlbeck93990462015-05-15 15:50:43 +0200361
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100362 ms_confirm_tlli(ms, new_ms_tlli);
363 OSMO_ASSERT(ms_tlli(ms) == start_tlli);
364 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
365 OSMO_ASSERT(ms_check_tlli(ms, start_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200366
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100367 ms_set_tlli(ms, new_ms_tlli);
368 OSMO_ASSERT(ms_tlli(ms) == new_ms_tlli);
369 OSMO_ASSERT(ms_check_tlli(ms, new_ms_tlli));
370 OSMO_ASSERT(!ms_check_tlli(ms, start_tlli));
Jacob Erlbeck93990462015-05-15 15:50:43 +0200371
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100372 talloc_free(ms);
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100373 talloc_free(bts);
Jacob Erlbeck93990462015-05-15 15:50:43 +0200374 printf("=== end %s ===\n", __func__);
375}
376
Maxf4d39732019-03-12 12:53:27 +0100377static GprsMs *prepare_ms(GprsMsStorage *st, uint32_t tlli, enum gprs_rlcmac_tbf_direction dir)
378{
379 GprsMs *ms = st->get_ms(tlli);
380 if (ms)
381 return ms;
382
383 ms = st->create_ms();
384
385 if (dir == GPRS_RLCMAC_UL_TBF)
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100386 ms_set_tlli(ms, tlli);
Maxf4d39732019-03-12 12:53:27 +0100387 else
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100388 ms_confirm_tlli(ms, tlli);
Maxf4d39732019-03-12 12:53:27 +0100389
390 return ms;
391}
392
Jacob Erlbeck53670862015-05-12 17:54:33 +0200393static void test_ms_storage()
394{
395 uint32_t tlli = 0xffeeddbb;
Jacob Erlbeck7b9f8252015-05-21 11:07:53 +0200396 const char *imsi1 = "001001987654321";
397 const char *imsi2 = "001001987654322";
398
Jacob Erlbeck53670862015-05-12 17:54:33 +0200399 gprs_rlcmac_ul_tbf *ul_tbf;
Pau Espin Pedrold1049dc2021-01-18 17:14:14 +0100400 struct gprs_rlcmac_bts *bts = bts_alloc(the_pcu, 0);
Jacob Erlbeck0e50ce62015-05-21 16:58:22 +0200401 GprsMs *ms, *ms_tmp;
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100402 GprsMsStorage store(bts);
Jacob Erlbeck53670862015-05-12 17:54:33 +0200403
404 printf("=== start %s ===\n", __func__);
405
Jacob Erlbeck53670862015-05-12 17:54:33 +0200406 ms = store.get_ms(tlli + 0);
407 OSMO_ASSERT(ms == NULL);
408
Maxf4d39732019-03-12 12:53:27 +0100409 ms = prepare_ms(&store, tlli + 0, GPRS_RLCMAC_UL_TBF);
Jacob Erlbeck53670862015-05-12 17:54:33 +0200410 OSMO_ASSERT(ms != NULL);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100411 OSMO_ASSERT(ms_tlli(ms) == tlli + 0);
412 ms_set_imsi(ms, imsi1);
413 OSMO_ASSERT(strcmp(ms_imsi(ms), imsi1) == 0);
Jacob Erlbeck53670862015-05-12 17:54:33 +0200414
Jacob Erlbeck0e50ce62015-05-21 16:58:22 +0200415 ms_tmp = store.get_ms(tlli + 0);
416 OSMO_ASSERT(ms == ms_tmp);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100417 OSMO_ASSERT(ms_tlli(ms) == tlli + 0);
Jacob Erlbeck0e50ce62015-05-21 16:58:22 +0200418
Jacob Erlbeck7b9f8252015-05-21 11:07:53 +0200419 ms_tmp = store.get_ms(0, 0, imsi1);
420 OSMO_ASSERT(ms == ms_tmp);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100421 OSMO_ASSERT(strcmp(ms_imsi(ms), imsi1) == 0);
Jacob Erlbeck7b9f8252015-05-21 11:07:53 +0200422 ms_tmp = store.get_ms(0, 0, imsi2);
423 OSMO_ASSERT(ms_tmp == NULL);
424
Maxf4d39732019-03-12 12:53:27 +0100425 ms = prepare_ms(&store, tlli + 1, GPRS_RLCMAC_UL_TBF);
Jacob Erlbeck0e50ce62015-05-21 16:58:22 +0200426 OSMO_ASSERT(ms != NULL);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100427 OSMO_ASSERT(ms_tlli(ms) == tlli + 1);
428 ms_set_imsi(ms, imsi2);
429 OSMO_ASSERT(strcmp(ms_imsi(ms), imsi2) == 0);
Jacob Erlbeck53670862015-05-12 17:54:33 +0200430
Jacob Erlbeck0e50ce62015-05-21 16:58:22 +0200431 ms_tmp = store.get_ms(tlli + 1);
432 OSMO_ASSERT(ms == ms_tmp);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100433 OSMO_ASSERT(ms_tlli(ms) == tlli + 1);
Jacob Erlbeck53670862015-05-12 17:54:33 +0200434
Jacob Erlbeck7b9f8252015-05-21 11:07:53 +0200435 ms_tmp = store.get_ms(0, 0, imsi1);
436 OSMO_ASSERT(ms_tmp != NULL);
437 OSMO_ASSERT(ms_tmp != ms);
438 ms_tmp = store.get_ms(0, 0, imsi2);
439 OSMO_ASSERT(ms == ms_tmp);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100440 OSMO_ASSERT(strcmp(ms_imsi(ms), imsi2) == 0);
Jacob Erlbeck7b9f8252015-05-21 11:07:53 +0200441
Jacob Erlbeck53670862015-05-12 17:54:33 +0200442 /* delete ms */
443 ms = store.get_ms(tlli + 0);
444 OSMO_ASSERT(ms != NULL);
Pau Espin Pedrol38f80be2021-05-14 14:33:28 +0200445 ul_tbf = alloc_ul_tbf(bts, ms);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100446 ms_attach_tbf(ms, ul_tbf);
447 ms_detach_tbf(ms, ul_tbf);
Jacob Erlbeck53670862015-05-12 17:54:33 +0200448 ms = store.get_ms(tlli + 0);
449 OSMO_ASSERT(ms == NULL);
450 ms = store.get_ms(tlli + 1);
451 OSMO_ASSERT(ms != NULL);
452
453 /* delete ms */
454 ms = store.get_ms(tlli + 1);
455 OSMO_ASSERT(ms != NULL);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100456 ms_attach_tbf(ms, ul_tbf);
457 ms_detach_tbf(ms, ul_tbf);
Jacob Erlbeck53670862015-05-12 17:54:33 +0200458 ms = store.get_ms(tlli + 1);
459 OSMO_ASSERT(ms == NULL);
460
461 talloc_free(ul_tbf);
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100462 talloc_free(bts);
Jacob Erlbeck53670862015-05-12 17:54:33 +0200463 printf("=== end %s ===\n", __func__);
464}
465
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200466static void test_ms_timeout()
467{
468 uint32_t tlli = 0xffeeddbb;
469 gprs_rlcmac_dl_tbf *dl_tbf;
470 gprs_rlcmac_ul_tbf *ul_tbf;
Pau Espin Pedrold1049dc2021-01-18 17:14:14 +0100471 struct gprs_rlcmac_bts *bts = bts_alloc(the_pcu, 0);
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200472 GprsMs *ms;
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100473 last_cb = CB_UNKNOWN;
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200474
475 printf("=== start %s ===\n", __func__);
476
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100477 ms = ms_alloc(bts, tlli);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100478 ms_set_callback(ms, &ms_cb);
479 ms_set_timeout(ms, 1);
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200480
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100481 OSMO_ASSERT(ms_is_idle(ms));
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200482
Pau Espin Pedrol38f80be2021-05-14 14:33:28 +0200483 dl_tbf = alloc_dl_tbf(bts, ms);
484 ul_tbf = alloc_ul_tbf(bts, ms);
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200485
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100486 OSMO_ASSERT(last_cb == CB_UNKNOWN);
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200487
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100488 ms_attach_tbf(ms, ul_tbf);
489 OSMO_ASSERT(!ms_is_idle(ms));
490 OSMO_ASSERT(last_cb == CB_IS_ACTIVE);
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200491
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100492 last_cb = CB_UNKNOWN;
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200493
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100494 ms_attach_tbf(ms, dl_tbf);
495 OSMO_ASSERT(!ms_is_idle(ms));
496 OSMO_ASSERT(last_cb == CB_UNKNOWN);
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200497
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100498 ms_detach_tbf(ms, ul_tbf);
499 OSMO_ASSERT(!ms_is_idle(ms));
500 OSMO_ASSERT(last_cb == CB_UNKNOWN);
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200501
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100502 ms_detach_tbf(ms, dl_tbf);
503 OSMO_ASSERT(!ms_is_idle(ms));
504 OSMO_ASSERT(last_cb == CB_UNKNOWN);
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200505
506 usleep(1100000);
507 osmo_timers_update();
508
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100509 OSMO_ASSERT(ms_is_idle(ms));
510 OSMO_ASSERT(last_cb == CB_IS_IDLE);
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200511
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100512 last_cb = CB_UNKNOWN;
513 talloc_free(ms);
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200514 talloc_free(dl_tbf);
515 talloc_free(ul_tbf);
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100516 talloc_free(bts);
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200517 printf("=== end %s ===\n", __func__);
518}
519
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200520static void test_ms_cs_selection()
521{
Pau Espin Pedrold1049dc2021-01-18 17:14:14 +0100522 struct gprs_rlcmac_bts *bts = bts_alloc(the_pcu, 0);
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200523 uint32_t tlli = 0xffeeddbb;
524
525 gprs_rlcmac_dl_tbf *dl_tbf;
526 GprsMs *ms;
527
528 printf("=== start %s ===\n", __func__);
529
530 bts->initial_cs_dl = 4;
531 bts->initial_cs_ul = 1;
Pau Espin Pedrolad79b852021-01-14 13:20:55 +0100532 the_pcu->vty.cs_downgrade_threshold = 0;
Pau Espin Pedrole8dcf642021-01-14 13:17:01 +0100533 the_pcu->vty.cs_adj_lower_limit = 0;
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200534
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100535 ms = ms_alloc(bts, tlli);
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200536
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100537 OSMO_ASSERT(ms_is_idle(ms));
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200538
Pau Espin Pedrol38f80be2021-05-14 14:33:28 +0200539 dl_tbf = alloc_dl_tbf(bts, ms);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100540 ms_attach_tbf(ms, dl_tbf);
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200541
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100542 OSMO_ASSERT(!ms_is_idle(ms));
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200543
Pau Espin Pedrolfc464932021-01-25 12:05:32 +0100544 OSMO_ASSERT(mcs_chan_code(ms_current_cs_dl(ms, ms_mode(ms))) == 3);
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200545
Pau Espin Pedrolad79b852021-01-14 13:20:55 +0100546 the_pcu->vty.cs_downgrade_threshold = 200;
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200547
Pau Espin Pedrolfc464932021-01-25 12:05:32 +0100548 OSMO_ASSERT(mcs_chan_code(ms_current_cs_dl(ms, ms_mode(ms))) == 2);
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200549
550 talloc_free(dl_tbf);
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100551 talloc_free(bts);
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200552 printf("=== end %s ===\n", __func__);
553}
554
Max41d6b352019-03-06 15:59:09 +0100555static void dump_ms(const GprsMs *ms, const char *pref)
556{
557 printf("%s MS DL %s/%s, UL %s/%s, mode %s, <%s>\n", pref,
Pau Espin Pedrolfc464932021-01-25 12:05:32 +0100558 mcs_name(ms_current_cs_dl(ms, ms_mode(ms))), mcs_name(ms_max_cs_dl(ms)),
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100559 mcs_name(ms_current_cs_ul(ms)), mcs_name(ms_max_cs_ul(ms)),
560 mode_name(ms_mode(ms)),
561 ms_is_idle(ms) ? "IDLE" : "ACTIVE");
Max41d6b352019-03-06 15:59:09 +0100562}
563
564static void test_ms_mcs_mode()
565{
Pau Espin Pedrold1049dc2021-01-18 17:14:14 +0100566 struct gprs_rlcmac_bts *bts = bts_alloc(the_pcu, 0);
Max41d6b352019-03-06 15:59:09 +0100567 uint32_t tlli = 0xdeadbeef;
568
569 gprs_rlcmac_dl_tbf *dl_tbf;
570 GprsMs *ms1, *ms2;
571
572 printf("=== start %s ===\n", __func__);
573
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100574 ms1 = ms_alloc(bts, tlli);
Max41d6b352019-03-06 15:59:09 +0100575 dump_ms(ms1, "1: no BTS defaults ");
576
577 bts->initial_cs_dl = 4;
578 bts->initial_cs_ul = 1;
Pau Espin Pedrolad79b852021-01-14 13:20:55 +0100579 the_pcu->vty.cs_downgrade_threshold = 0;
Max41d6b352019-03-06 15:59:09 +0100580
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100581 ms2 = ms_alloc(bts, tlli + 1);
Max41d6b352019-03-06 15:59:09 +0100582 dump_ms(ms2, "2: with BTS defaults");
583
Pau Espin Pedrol38f80be2021-05-14 14:33:28 +0200584 dl_tbf = alloc_dl_tbf(bts, ms2);
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100585 ms_attach_tbf(ms2, dl_tbf);
Max41d6b352019-03-06 15:59:09 +0100586
Max41d6b352019-03-06 15:59:09 +0100587 dump_ms(ms2, "2: after TBF attach ");
588
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100589 ms_set_mode(ms1, EGPRS);
Max41d6b352019-03-06 15:59:09 +0100590 dump_ms(ms1, "1: after mode set ");
591
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100592 ms_set_mode(ms2, EGPRS);
Max41d6b352019-03-06 15:59:09 +0100593 dump_ms(ms2, "2: after mode set ");
594
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100595 ms_set_current_cs_dl(ms1, MCS7);
Max41d6b352019-03-06 15:59:09 +0100596 dump_ms(ms1, "1: after MCS set ");
597
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100598 ms_set_current_cs_dl(ms2, MCS8);
Max41d6b352019-03-06 15:59:09 +0100599 dump_ms(ms2, "2: after MCS set ");
600
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100601 ms_set_mode(ms1, EGPRS_GMSK);
Max41d6b352019-03-06 15:59:09 +0100602 dump_ms(ms1, "1: after mode set ");
603
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100604 ms_set_mode(ms2, EGPRS_GMSK);
Max41d6b352019-03-06 15:59:09 +0100605 dump_ms(ms2, "2: after mode set ");
606
607 // FIXME: following code triggers ASAN failure:
608 // ms2->detach_tbf(dl_tbf);
609 // dump_ms(ms2, "2: after TBF detach ");
610
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100611 ms_set_mode(ms1, GPRS);
Max41d6b352019-03-06 15:59:09 +0100612 dump_ms(ms1, "1: after mode set ");
613
Pau Espin Pedrolda971ee2020-12-16 15:59:45 +0100614 ms_set_mode(ms2, GPRS);
Max41d6b352019-03-06 15:59:09 +0100615 dump_ms(ms2, "2: after mode set ");
616
617 talloc_free(dl_tbf);
Pau Espin Pedrol2182e622021-01-14 16:48:38 +0100618 talloc_free(bts);
Max41d6b352019-03-06 15:59:09 +0100619 printf("=== end %s ===\n", __func__);
620}
621
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200622int main(int argc, char **argv)
623{
624 struct vty_app_info pcu_vty_info = {0};
625
626 tall_pcu_ctx = talloc_named_const(NULL, 1, "MsTest context");
627 if (!tall_pcu_ctx)
628 abort();
629
Neels Hofmeyr78ce5912017-02-08 17:07:31 +0100630 msgb_talloc_ctx_init(tall_pcu_ctx, 0);
Neels Hofmeyr42f2d612018-04-01 16:54:40 +0200631 osmo_init_logging2(tall_pcu_ctx, &gprs_log_info);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200632 log_set_use_color(osmo_stderr_target, 0);
Pau Espin Pedrol00f52cc2021-02-19 14:01:52 +0100633 log_set_print_filename2(osmo_stderr_target, LOG_FILENAME_NONE);
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200634 log_set_log_level(osmo_stderr_target, LOGL_INFO);
Pau Espin Pedrolb18d2a52021-02-19 14:00:48 +0100635 log_set_print_category(osmo_stderr_target, 0);
636 log_set_print_category_hex(osmo_stderr_target, 0);
Harald Welteac0490a2017-10-29 10:39:32 +0100637 log_parse_category_mask(osmo_stderr_target, "DPCU,3:DRLCMAC,3");
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200638
Pau Espin Pedrolac3fd122021-01-13 18:54:38 +0100639 the_pcu = gprs_pcu_alloc(tall_pcu_ctx);
640
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200641 vty_init(&pcu_vty_info);
Pau Espin Pedrolcd2ac562019-08-05 14:30:44 +0200642 pcu_vty_init();
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200643
644 test_ms_state();
645 test_ms_callback();
646 test_ms_replace_tbf();
Jacob Erlbeck93990462015-05-15 15:50:43 +0200647 test_ms_change_tlli();
Jacob Erlbeck53670862015-05-12 17:54:33 +0200648 test_ms_storage();
Jacob Erlbeckd9e10242015-05-28 15:43:53 +0200649 test_ms_timeout();
Jacob Erlbeck70b96aa2015-06-12 10:52:34 +0200650 test_ms_cs_selection();
Max41d6b352019-03-06 15:59:09 +0100651 test_ms_mcs_mode();
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200652
Pau Espin Pedrolac3fd122021-01-13 18:54:38 +0100653 talloc_free(the_pcu);
654
Jacob Erlbecke04e0b02015-05-06 18:30:48 +0200655 if (getenv("TALLOC_REPORT_FULL"))
656 talloc_report_full(tall_pcu_ctx, stderr);
657
658 return EXIT_SUCCESS;
659}
660
661extern "C" {
662void l1if_pdch_req() { abort(); }
663void l1if_connect_pdch() { abort(); }
664void l1if_close_pdch() { abort(); }
665void l1if_open_pdch() { abort(); }
666}