1 // SPDX-License-Identifier: BSD-3-Clause OR GPL-2.0
2 /* Copyright (c) 2018 Mellanox Technologies. All rights reserved */
3
4 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
5
6 #include <linux/kernel.h>
7 #include <linux/module.h>
8 #include <linux/slab.h>
9 #include <linux/random.h>
10 #include <linux/objagg.h>
11
12 struct tokey {
13 unsigned int id;
14 };
15
16 #define NUM_KEYS 32
17
key_id_index(unsigned int key_id)18 static int key_id_index(unsigned int key_id)
19 {
20 if (key_id >= NUM_KEYS) {
21 WARN_ON(1);
22 return 0;
23 }
24 return key_id;
25 }
26
27 #define BUF_LEN 128
28
29 struct world {
30 unsigned int root_count;
31 unsigned int delta_count;
32 char next_root_buf[BUF_LEN];
33 struct objagg_obj *objagg_objs[NUM_KEYS];
34 unsigned int key_refs[NUM_KEYS];
35 };
36
37 struct root {
38 struct tokey key;
39 char buf[BUF_LEN];
40 };
41
42 struct delta {
43 unsigned int key_id_diff;
44 };
45
world_obj_get(struct world * world,struct objagg * objagg,unsigned int key_id)46 static struct objagg_obj *world_obj_get(struct world *world,
47 struct objagg *objagg,
48 unsigned int key_id)
49 {
50 struct objagg_obj *objagg_obj;
51 struct tokey key;
52 int err;
53
54 key.id = key_id;
55 objagg_obj = objagg_obj_get(objagg, &key);
56 if (IS_ERR(objagg_obj)) {
57 pr_err("Key %u: Failed to get object.\n", key_id);
58 return objagg_obj;
59 }
60 if (!world->key_refs[key_id_index(key_id)]) {
61 world->objagg_objs[key_id_index(key_id)] = objagg_obj;
62 } else if (world->objagg_objs[key_id_index(key_id)] != objagg_obj) {
63 pr_err("Key %u: Got another object for the same key.\n",
64 key_id);
65 err = -EINVAL;
66 goto err_key_id_check;
67 }
68 world->key_refs[key_id_index(key_id)]++;
69 return objagg_obj;
70
71 err_key_id_check:
72 objagg_obj_put(objagg, objagg_obj);
73 return ERR_PTR(err);
74 }
75
world_obj_put(struct world * world,struct objagg * objagg,unsigned int key_id)76 static void world_obj_put(struct world *world, struct objagg *objagg,
77 unsigned int key_id)
78 {
79 struct objagg_obj *objagg_obj;
80
81 if (!world->key_refs[key_id_index(key_id)])
82 return;
83 objagg_obj = world->objagg_objs[key_id_index(key_id)];
84 objagg_obj_put(objagg, objagg_obj);
85 world->key_refs[key_id_index(key_id)]--;
86 }
87
88 #define MAX_KEY_ID_DIFF 5
89
delta_check(void * priv,const void * parent_obj,const void * obj)90 static bool delta_check(void *priv, const void *parent_obj, const void *obj)
91 {
92 const struct tokey *parent_key = parent_obj;
93 const struct tokey *key = obj;
94 int diff = key->id - parent_key->id;
95
96 return diff >= 0 && diff <= MAX_KEY_ID_DIFF;
97 }
98
delta_create(void * priv,void * parent_obj,void * obj)99 static void *delta_create(void *priv, void *parent_obj, void *obj)
100 {
101 struct tokey *parent_key = parent_obj;
102 struct world *world = priv;
103 struct tokey *key = obj;
104 int diff = key->id - parent_key->id;
105 struct delta *delta;
106
107 if (!delta_check(priv, parent_obj, obj))
108 return ERR_PTR(-EINVAL);
109
110 delta = kzalloc(sizeof(*delta), GFP_KERNEL);
111 if (!delta)
112 return ERR_PTR(-ENOMEM);
113 delta->key_id_diff = diff;
114 world->delta_count++;
115 return delta;
116 }
117
delta_destroy(void * priv,void * delta_priv)118 static void delta_destroy(void *priv, void *delta_priv)
119 {
120 struct delta *delta = delta_priv;
121 struct world *world = priv;
122
123 world->delta_count--;
124 kfree(delta);
125 }
126
root_create(void * priv,void * obj,unsigned int id)127 static void *root_create(void *priv, void *obj, unsigned int id)
128 {
129 struct world *world = priv;
130 struct tokey *key = obj;
131 struct root *root;
132
133 root = kzalloc(sizeof(*root), GFP_KERNEL);
134 if (!root)
135 return ERR_PTR(-ENOMEM);
136 memcpy(&root->key, key, sizeof(root->key));
137 memcpy(root->buf, world->next_root_buf, sizeof(root->buf));
138 world->root_count++;
139 return root;
140 }
141
root_destroy(void * priv,void * root_priv)142 static void root_destroy(void *priv, void *root_priv)
143 {
144 struct root *root = root_priv;
145 struct world *world = priv;
146
147 world->root_count--;
148 kfree(root);
149 }
150
test_nodelta_obj_get(struct world * world,struct objagg * objagg,unsigned int key_id,bool should_create_root)151 static int test_nodelta_obj_get(struct world *world, struct objagg *objagg,
152 unsigned int key_id, bool should_create_root)
153 {
154 unsigned int orig_root_count = world->root_count;
155 struct objagg_obj *objagg_obj;
156 const struct root *root;
157 int err;
158
159 if (should_create_root)
160 get_random_bytes(world->next_root_buf,
161 sizeof(world->next_root_buf));
162
163 objagg_obj = world_obj_get(world, objagg, key_id);
164 if (IS_ERR(objagg_obj)) {
165 pr_err("Key %u: Failed to get object.\n", key_id);
166 return PTR_ERR(objagg_obj);
167 }
168 if (should_create_root) {
169 if (world->root_count != orig_root_count + 1) {
170 pr_err("Key %u: Root was not created\n", key_id);
171 err = -EINVAL;
172 goto err_check_root_count;
173 }
174 } else {
175 if (world->root_count != orig_root_count) {
176 pr_err("Key %u: Root was incorrectly created\n",
177 key_id);
178 err = -EINVAL;
179 goto err_check_root_count;
180 }
181 }
182 root = objagg_obj_root_priv(objagg_obj);
183 if (root->key.id != key_id) {
184 pr_err("Key %u: Root has unexpected key id\n", key_id);
185 err = -EINVAL;
186 goto err_check_key_id;
187 }
188 if (should_create_root &&
189 memcmp(world->next_root_buf, root->buf, sizeof(root->buf))) {
190 pr_err("Key %u: Buffer does not match the expected content\n",
191 key_id);
192 err = -EINVAL;
193 goto err_check_buf;
194 }
195 return 0;
196
197 err_check_buf:
198 err_check_key_id:
199 err_check_root_count:
200 objagg_obj_put(objagg, objagg_obj);
201 return err;
202 }
203
test_nodelta_obj_put(struct world * world,struct objagg * objagg,unsigned int key_id,bool should_destroy_root)204 static int test_nodelta_obj_put(struct world *world, struct objagg *objagg,
205 unsigned int key_id, bool should_destroy_root)
206 {
207 unsigned int orig_root_count = world->root_count;
208
209 world_obj_put(world, objagg, key_id);
210
211 if (should_destroy_root) {
212 if (world->root_count != orig_root_count - 1) {
213 pr_err("Key %u: Root was not destroyed\n", key_id);
214 return -EINVAL;
215 }
216 } else {
217 if (world->root_count != orig_root_count) {
218 pr_err("Key %u: Root was incorrectly destroyed\n",
219 key_id);
220 return -EINVAL;
221 }
222 }
223 return 0;
224 }
225
check_stats_zero(struct objagg * objagg)226 static int check_stats_zero(struct objagg *objagg)
227 {
228 const struct objagg_stats *stats;
229 int err = 0;
230
231 stats = objagg_stats_get(objagg);
232 if (IS_ERR(stats))
233 return PTR_ERR(stats);
234
235 if (stats->stats_info_count != 0) {
236 pr_err("Stats: Object count is not zero while it should be\n");
237 err = -EINVAL;
238 }
239
240 objagg_stats_put(stats);
241 return err;
242 }
243
check_stats_nodelta(struct objagg * objagg)244 static int check_stats_nodelta(struct objagg *objagg)
245 {
246 const struct objagg_stats *stats;
247 int i;
248 int err;
249
250 stats = objagg_stats_get(objagg);
251 if (IS_ERR(stats))
252 return PTR_ERR(stats);
253
254 if (stats->stats_info_count != NUM_KEYS) {
255 pr_err("Stats: Unexpected object count (%u expected, %u returned)\n",
256 NUM_KEYS, stats->stats_info_count);
257 err = -EINVAL;
258 goto stats_put;
259 }
260
261 for (i = 0; i < stats->stats_info_count; i++) {
262 if (stats->stats_info[i].stats.user_count != 2) {
263 pr_err("Stats: incorrect user count\n");
264 err = -EINVAL;
265 goto stats_put;
266 }
267 if (stats->stats_info[i].stats.delta_user_count != 2) {
268 pr_err("Stats: incorrect delta user count\n");
269 err = -EINVAL;
270 goto stats_put;
271 }
272 }
273 err = 0;
274
275 stats_put:
276 objagg_stats_put(stats);
277 return err;
278 }
279
delta_check_dummy(void * priv,const void * parent_obj,const void * obj)280 static bool delta_check_dummy(void *priv, const void *parent_obj,
281 const void *obj)
282 {
283 return false;
284 }
285
delta_create_dummy(void * priv,void * parent_obj,void * obj)286 static void *delta_create_dummy(void *priv, void *parent_obj, void *obj)
287 {
288 return ERR_PTR(-EOPNOTSUPP);
289 }
290
delta_destroy_dummy(void * priv,void * delta_priv)291 static void delta_destroy_dummy(void *priv, void *delta_priv)
292 {
293 }
294
295 static const struct objagg_ops nodelta_ops = {
296 .obj_size = sizeof(struct tokey),
297 .delta_check = delta_check_dummy,
298 .delta_create = delta_create_dummy,
299 .delta_destroy = delta_destroy_dummy,
300 .root_create = root_create,
301 .root_destroy = root_destroy,
302 };
303
test_nodelta(void)304 static int test_nodelta(void)
305 {
306 struct world world = {};
307 struct objagg *objagg;
308 int i;
309 int err;
310
311 objagg = objagg_create(&nodelta_ops, NULL, &world);
312 if (IS_ERR(objagg))
313 return PTR_ERR(objagg);
314
315 err = check_stats_zero(objagg);
316 if (err)
317 goto err_stats_first_zero;
318
319 /* First round of gets, the root objects should be created */
320 for (i = 0; i < NUM_KEYS; i++) {
321 err = test_nodelta_obj_get(&world, objagg, i, true);
322 if (err)
323 goto err_obj_first_get;
324 }
325
326 /* Do the second round of gets, all roots are already created,
327 * make sure that no new root is created
328 */
329 for (i = 0; i < NUM_KEYS; i++) {
330 err = test_nodelta_obj_get(&world, objagg, i, false);
331 if (err)
332 goto err_obj_second_get;
333 }
334
335 err = check_stats_nodelta(objagg);
336 if (err)
337 goto err_stats_nodelta;
338
339 for (i = NUM_KEYS - 1; i >= 0; i--) {
340 err = test_nodelta_obj_put(&world, objagg, i, false);
341 if (err)
342 goto err_obj_first_put;
343 }
344 for (i = NUM_KEYS - 1; i >= 0; i--) {
345 err = test_nodelta_obj_put(&world, objagg, i, true);
346 if (err)
347 goto err_obj_second_put;
348 }
349
350 err = check_stats_zero(objagg);
351 if (err)
352 goto err_stats_second_zero;
353
354 objagg_destroy(objagg);
355 return 0;
356
357 err_stats_nodelta:
358 err_obj_first_put:
359 err_obj_second_get:
360 for (i--; i >= 0; i--)
361 world_obj_put(&world, objagg, i);
362
363 i = NUM_KEYS;
364 err_obj_first_get:
365 err_obj_second_put:
366 for (i--; i >= 0; i--)
367 world_obj_put(&world, objagg, i);
368 err_stats_first_zero:
369 err_stats_second_zero:
370 objagg_destroy(objagg);
371 return err;
372 }
373
374 static const struct objagg_ops delta_ops = {
375 .obj_size = sizeof(struct tokey),
376 .delta_check = delta_check,
377 .delta_create = delta_create,
378 .delta_destroy = delta_destroy,
379 .root_create = root_create,
380 .root_destroy = root_destroy,
381 };
382
383 enum action {
384 ACTION_GET,
385 ACTION_PUT,
386 };
387
388 enum expect_delta {
389 EXPECT_DELTA_SAME,
390 EXPECT_DELTA_INC,
391 EXPECT_DELTA_DEC,
392 };
393
394 enum expect_root {
395 EXPECT_ROOT_SAME,
396 EXPECT_ROOT_INC,
397 EXPECT_ROOT_DEC,
398 };
399
400 struct expect_stats_info {
401 struct objagg_obj_stats stats;
402 bool is_root;
403 unsigned int key_id;
404 };
405
406 struct expect_stats {
407 unsigned int info_count;
408 struct expect_stats_info info[NUM_KEYS];
409 };
410
411 struct action_item {
412 unsigned int key_id;
413 enum action action;
414 enum expect_delta expect_delta;
415 enum expect_root expect_root;
416 struct expect_stats expect_stats;
417 };
418
419 #define EXPECT_STATS(count, ...) \
420 { \
421 .info_count = count, \
422 .info = { __VA_ARGS__ } \
423 }
424
425 #define ROOT(key_id, user_count, delta_user_count) \
426 {{user_count, delta_user_count}, true, key_id}
427
428 #define DELTA(key_id, user_count) \
429 {{user_count, user_count}, false, key_id}
430
431 static const struct action_item action_items[] = {
432 {
433 1, ACTION_GET, EXPECT_DELTA_SAME, EXPECT_ROOT_INC,
434 EXPECT_STATS(1, ROOT(1, 1, 1)),
435 }, /* r: 1 d: */
436 {
437 7, ACTION_GET, EXPECT_DELTA_SAME, EXPECT_ROOT_INC,
438 EXPECT_STATS(2, ROOT(1, 1, 1), ROOT(7, 1, 1)),
439 }, /* r: 1, 7 d: */
440 {
441 3, ACTION_GET, EXPECT_DELTA_INC, EXPECT_ROOT_SAME,
442 EXPECT_STATS(3, ROOT(1, 1, 2), ROOT(7, 1, 1),
443 DELTA(3, 1)),
444 }, /* r: 1, 7 d: 3^1 */
445 {
446 5, ACTION_GET, EXPECT_DELTA_INC, EXPECT_ROOT_SAME,
447 EXPECT_STATS(4, ROOT(1, 1, 3), ROOT(7, 1, 1),
448 DELTA(3, 1), DELTA(5, 1)),
449 }, /* r: 1, 7 d: 3^1, 5^1 */
450 {
451 3, ACTION_GET, EXPECT_DELTA_SAME, EXPECT_ROOT_SAME,
452 EXPECT_STATS(4, ROOT(1, 1, 4), ROOT(7, 1, 1),
453 DELTA(3, 2), DELTA(5, 1)),
454 }, /* r: 1, 7 d: 3^1, 3^1, 5^1 */
455 {
456 1, ACTION_GET, EXPECT_DELTA_SAME, EXPECT_ROOT_SAME,
457 EXPECT_STATS(4, ROOT(1, 2, 5), ROOT(7, 1, 1),
458 DELTA(3, 2), DELTA(5, 1)),
459 }, /* r: 1, 1, 7 d: 3^1, 3^1, 5^1 */
460 {
461 30, ACTION_GET, EXPECT_DELTA_SAME, EXPECT_ROOT_INC,
462 EXPECT_STATS(5, ROOT(1, 2, 5), ROOT(7, 1, 1), ROOT(30, 1, 1),
463 DELTA(3, 2), DELTA(5, 1)),
464 }, /* r: 1, 1, 7, 30 d: 3^1, 3^1, 5^1 */
465 {
466 8, ACTION_GET, EXPECT_DELTA_INC, EXPECT_ROOT_SAME,
467 EXPECT_STATS(6, ROOT(1, 2, 5), ROOT(7, 1, 2), ROOT(30, 1, 1),
468 DELTA(3, 2), DELTA(5, 1), DELTA(8, 1)),
469 }, /* r: 1, 1, 7, 30 d: 3^1, 3^1, 5^1, 8^7 */
470 {
471 8, ACTION_GET, EXPECT_DELTA_SAME, EXPECT_ROOT_SAME,
472 EXPECT_STATS(6, ROOT(1, 2, 5), ROOT(7, 1, 3), ROOT(30, 1, 1),
473 DELTA(3, 2), DELTA(8, 2), DELTA(5, 1)),
474 }, /* r: 1, 1, 7, 30 d: 3^1, 3^1, 5^1, 8^7, 8^7 */
475 {
476 3, ACTION_PUT, EXPECT_DELTA_SAME, EXPECT_ROOT_SAME,
477 EXPECT_STATS(6, ROOT(1, 2, 4), ROOT(7, 1, 3), ROOT(30, 1, 1),
478 DELTA(8, 2), DELTA(3, 1), DELTA(5, 1)),
479 }, /* r: 1, 1, 7, 30 d: 3^1, 5^1, 8^7, 8^7 */
480 {
481 3, ACTION_PUT, EXPECT_DELTA_DEC, EXPECT_ROOT_SAME,
482 EXPECT_STATS(5, ROOT(1, 2, 3), ROOT(7, 1, 3), ROOT(30, 1, 1),
483 DELTA(8, 2), DELTA(5, 1)),
484 }, /* r: 1, 1, 7, 30 d: 5^1, 8^7, 8^7 */
485 {
486 1, ACTION_PUT, EXPECT_DELTA_SAME, EXPECT_ROOT_SAME,
487 EXPECT_STATS(5, ROOT(7, 1, 3), ROOT(1, 1, 2), ROOT(30, 1, 1),
488 DELTA(8, 2), DELTA(5, 1)),
489 }, /* r: 1, 7, 30 d: 5^1, 8^7, 8^7 */
490 {
491 1, ACTION_PUT, EXPECT_DELTA_SAME, EXPECT_ROOT_SAME,
492 EXPECT_STATS(5, ROOT(7, 1, 3), ROOT(30, 1, 1), ROOT(1, 0, 1),
493 DELTA(8, 2), DELTA(5, 1)),
494 }, /* r: 7, 30 d: 5^1, 8^7, 8^7 */
495 {
496 5, ACTION_PUT, EXPECT_DELTA_DEC, EXPECT_ROOT_DEC,
497 EXPECT_STATS(3, ROOT(7, 1, 3), ROOT(30, 1, 1),
498 DELTA(8, 2)),
499 }, /* r: 7, 30 d: 8^7, 8^7 */
500 {
501 5, ACTION_GET, EXPECT_DELTA_SAME, EXPECT_ROOT_INC,
502 EXPECT_STATS(4, ROOT(7, 1, 3), ROOT(30, 1, 1), ROOT(5, 1, 1),
503 DELTA(8, 2)),
504 }, /* r: 7, 30, 5 d: 8^7, 8^7 */
505 {
506 6, ACTION_GET, EXPECT_DELTA_INC, EXPECT_ROOT_SAME,
507 EXPECT_STATS(5, ROOT(7, 1, 3), ROOT(5, 1, 2), ROOT(30, 1, 1),
508 DELTA(8, 2), DELTA(6, 1)),
509 }, /* r: 7, 30, 5 d: 8^7, 8^7, 6^5 */
510 {
511 8, ACTION_GET, EXPECT_DELTA_SAME, EXPECT_ROOT_SAME,
512 EXPECT_STATS(5, ROOT(7, 1, 4), ROOT(5, 1, 2), ROOT(30, 1, 1),
513 DELTA(8, 3), DELTA(6, 1)),
514 }, /* r: 7, 30, 5 d: 8^7, 8^7, 8^7, 6^5 */
515 {
516 8, ACTION_PUT, EXPECT_DELTA_SAME, EXPECT_ROOT_SAME,
517 EXPECT_STATS(5, ROOT(7, 1, 3), ROOT(5, 1, 2), ROOT(30, 1, 1),
518 DELTA(8, 2), DELTA(6, 1)),
519 }, /* r: 7, 30, 5 d: 8^7, 8^7, 6^5 */
520 {
521 8, ACTION_PUT, EXPECT_DELTA_SAME, EXPECT_ROOT_SAME,
522 EXPECT_STATS(5, ROOT(7, 1, 2), ROOT(5, 1, 2), ROOT(30, 1, 1),
523 DELTA(8, 1), DELTA(6, 1)),
524 }, /* r: 7, 30, 5 d: 8^7, 6^5 */
525 {
526 8, ACTION_PUT, EXPECT_DELTA_DEC, EXPECT_ROOT_SAME,
527 EXPECT_STATS(4, ROOT(5, 1, 2), ROOT(7, 1, 1), ROOT(30, 1, 1),
528 DELTA(6, 1)),
529 }, /* r: 7, 30, 5 d: 6^5 */
530 {
531 8, ACTION_GET, EXPECT_DELTA_INC, EXPECT_ROOT_SAME,
532 EXPECT_STATS(5, ROOT(5, 1, 3), ROOT(7, 1, 1), ROOT(30, 1, 1),
533 DELTA(6, 1), DELTA(8, 1)),
534 }, /* r: 7, 30, 5 d: 6^5, 8^5 */
535 {
536 7, ACTION_PUT, EXPECT_DELTA_SAME, EXPECT_ROOT_DEC,
537 EXPECT_STATS(4, ROOT(5, 1, 3), ROOT(30, 1, 1),
538 DELTA(6, 1), DELTA(8, 1)),
539 }, /* r: 30, 5 d: 6^5, 8^5 */
540 {
541 30, ACTION_PUT, EXPECT_DELTA_SAME, EXPECT_ROOT_DEC,
542 EXPECT_STATS(3, ROOT(5, 1, 3),
543 DELTA(6, 1), DELTA(8, 1)),
544 }, /* r: 5 d: 6^5, 8^5 */
545 {
546 5, ACTION_PUT, EXPECT_DELTA_SAME, EXPECT_ROOT_SAME,
547 EXPECT_STATS(3, ROOT(5, 0, 2),
548 DELTA(6, 1), DELTA(8, 1)),
549 }, /* r: d: 6^5, 8^5 */
550 {
551 6, ACTION_PUT, EXPECT_DELTA_DEC, EXPECT_ROOT_SAME,
552 EXPECT_STATS(2, ROOT(5, 0, 1),
553 DELTA(8, 1)),
554 }, /* r: d: 6^5 */
555 {
556 8, ACTION_PUT, EXPECT_DELTA_DEC, EXPECT_ROOT_DEC,
557 EXPECT_STATS(0, ),
558 }, /* r: d: */
559 };
560
check_expect(struct world * world,const struct action_item * action_item,unsigned int orig_delta_count,unsigned int orig_root_count)561 static int check_expect(struct world *world,
562 const struct action_item *action_item,
563 unsigned int orig_delta_count,
564 unsigned int orig_root_count)
565 {
566 unsigned int key_id = action_item->key_id;
567
568 switch (action_item->expect_delta) {
569 case EXPECT_DELTA_SAME:
570 if (orig_delta_count != world->delta_count) {
571 pr_err("Key %u: Delta count changed while expected to remain the same.\n",
572 key_id);
573 return -EINVAL;
574 }
575 break;
576 case EXPECT_DELTA_INC:
577 if (WARN_ON(action_item->action == ACTION_PUT))
578 return -EINVAL;
579 if (orig_delta_count + 1 != world->delta_count) {
580 pr_err("Key %u: Delta count was not incremented.\n",
581 key_id);
582 return -EINVAL;
583 }
584 break;
585 case EXPECT_DELTA_DEC:
586 if (WARN_ON(action_item->action == ACTION_GET))
587 return -EINVAL;
588 if (orig_delta_count - 1 != world->delta_count) {
589 pr_err("Key %u: Delta count was not decremented.\n",
590 key_id);
591 return -EINVAL;
592 }
593 break;
594 }
595
596 switch (action_item->expect_root) {
597 case EXPECT_ROOT_SAME:
598 if (orig_root_count != world->root_count) {
599 pr_err("Key %u: Root count changed while expected to remain the same.\n",
600 key_id);
601 return -EINVAL;
602 }
603 break;
604 case EXPECT_ROOT_INC:
605 if (WARN_ON(action_item->action == ACTION_PUT))
606 return -EINVAL;
607 if (orig_root_count + 1 != world->root_count) {
608 pr_err("Key %u: Root count was not incremented.\n",
609 key_id);
610 return -EINVAL;
611 }
612 break;
613 case EXPECT_ROOT_DEC:
614 if (WARN_ON(action_item->action == ACTION_GET))
615 return -EINVAL;
616 if (orig_root_count - 1 != world->root_count) {
617 pr_err("Key %u: Root count was not decremented.\n",
618 key_id);
619 return -EINVAL;
620 }
621 }
622
623 return 0;
624 }
625
obj_to_key_id(struct objagg_obj * objagg_obj)626 static unsigned int obj_to_key_id(struct objagg_obj *objagg_obj)
627 {
628 const struct tokey *root_key;
629 const struct delta *delta;
630 unsigned int key_id;
631
632 root_key = objagg_obj_root_priv(objagg_obj);
633 key_id = root_key->id;
634 delta = objagg_obj_delta_priv(objagg_obj);
635 if (delta)
636 key_id += delta->key_id_diff;
637 return key_id;
638 }
639
640 static int
check_expect_stats_nums(const struct objagg_obj_stats_info * stats_info,const struct expect_stats_info * expect_stats_info,const char ** errmsg)641 check_expect_stats_nums(const struct objagg_obj_stats_info *stats_info,
642 const struct expect_stats_info *expect_stats_info,
643 const char **errmsg)
644 {
645 if (stats_info->is_root != expect_stats_info->is_root) {
646 if (errmsg)
647 *errmsg = "Incorrect root/delta indication";
648 return -EINVAL;
649 }
650 if (stats_info->stats.user_count !=
651 expect_stats_info->stats.user_count) {
652 if (errmsg)
653 *errmsg = "Incorrect user count";
654 return -EINVAL;
655 }
656 if (stats_info->stats.delta_user_count !=
657 expect_stats_info->stats.delta_user_count) {
658 if (errmsg)
659 *errmsg = "Incorrect delta user count";
660 return -EINVAL;
661 }
662 return 0;
663 }
664
665 static int
check_expect_stats_key_id(const struct objagg_obj_stats_info * stats_info,const struct expect_stats_info * expect_stats_info,const char ** errmsg)666 check_expect_stats_key_id(const struct objagg_obj_stats_info *stats_info,
667 const struct expect_stats_info *expect_stats_info,
668 const char **errmsg)
669 {
670 if (obj_to_key_id(stats_info->objagg_obj) !=
671 expect_stats_info->key_id) {
672 if (errmsg)
673 *errmsg = "incorrect key id";
674 return -EINVAL;
675 }
676 return 0;
677 }
678
check_expect_stats_neigh(const struct objagg_stats * stats,const struct expect_stats * expect_stats,int pos)679 static int check_expect_stats_neigh(const struct objagg_stats *stats,
680 const struct expect_stats *expect_stats,
681 int pos)
682 {
683 int i;
684 int err;
685
686 for (i = pos - 1; i >= 0; i--) {
687 err = check_expect_stats_nums(&stats->stats_info[i],
688 &expect_stats->info[pos], NULL);
689 if (err)
690 break;
691 err = check_expect_stats_key_id(&stats->stats_info[i],
692 &expect_stats->info[pos], NULL);
693 if (!err)
694 return 0;
695 }
696 for (i = pos + 1; i < stats->stats_info_count; i++) {
697 err = check_expect_stats_nums(&stats->stats_info[i],
698 &expect_stats->info[pos], NULL);
699 if (err)
700 break;
701 err = check_expect_stats_key_id(&stats->stats_info[i],
702 &expect_stats->info[pos], NULL);
703 if (!err)
704 return 0;
705 }
706 return -EINVAL;
707 }
708
__check_expect_stats(const struct objagg_stats * stats,const struct expect_stats * expect_stats,const char ** errmsg)709 static int __check_expect_stats(const struct objagg_stats *stats,
710 const struct expect_stats *expect_stats,
711 const char **errmsg)
712 {
713 int i;
714 int err;
715
716 if (stats->stats_info_count != expect_stats->info_count) {
717 *errmsg = "Unexpected object count";
718 return -EINVAL;
719 }
720
721 for (i = 0; i < stats->stats_info_count; i++) {
722 err = check_expect_stats_nums(&stats->stats_info[i],
723 &expect_stats->info[i], errmsg);
724 if (err)
725 return err;
726 err = check_expect_stats_key_id(&stats->stats_info[i],
727 &expect_stats->info[i], errmsg);
728 if (err) {
729 /* It is possible that one of the neighbor stats with
730 * same numbers have the correct key id, so check it
731 */
732 err = check_expect_stats_neigh(stats, expect_stats, i);
733 if (err)
734 return err;
735 }
736 }
737 return 0;
738 }
739
check_expect_stats(struct objagg * objagg,const struct expect_stats * expect_stats,const char ** errmsg)740 static int check_expect_stats(struct objagg *objagg,
741 const struct expect_stats *expect_stats,
742 const char **errmsg)
743 {
744 const struct objagg_stats *stats;
745 int err;
746
747 stats = objagg_stats_get(objagg);
748 if (IS_ERR(stats)) {
749 *errmsg = "objagg_stats_get() failed.";
750 return PTR_ERR(stats);
751 }
752 err = __check_expect_stats(stats, expect_stats, errmsg);
753 objagg_stats_put(stats);
754 return err;
755 }
756
test_delta_action_item(struct world * world,struct objagg * objagg,const struct action_item * action_item,bool inverse)757 static int test_delta_action_item(struct world *world,
758 struct objagg *objagg,
759 const struct action_item *action_item,
760 bool inverse)
761 {
762 unsigned int orig_delta_count = world->delta_count;
763 unsigned int orig_root_count = world->root_count;
764 unsigned int key_id = action_item->key_id;
765 enum action action = action_item->action;
766 struct objagg_obj *objagg_obj;
767 const char *errmsg;
768 int err;
769
770 if (inverse)
771 action = action == ACTION_GET ? ACTION_PUT : ACTION_GET;
772
773 switch (action) {
774 case ACTION_GET:
775 objagg_obj = world_obj_get(world, objagg, key_id);
776 if (IS_ERR(objagg_obj))
777 return PTR_ERR(objagg_obj);
778 break;
779 case ACTION_PUT:
780 world_obj_put(world, objagg, key_id);
781 break;
782 }
783
784 if (inverse)
785 return 0;
786 err = check_expect(world, action_item,
787 orig_delta_count, orig_root_count);
788 if (err)
789 goto errout;
790
791 err = check_expect_stats(objagg, &action_item->expect_stats, &errmsg);
792 if (err) {
793 pr_err("Key %u: Stats: %s\n", action_item->key_id, errmsg);
794 goto errout;
795 }
796
797 return 0;
798
799 errout:
800 /* This can only happen when action is not inversed.
801 * So in case of an error, cleanup by doing inverse action.
802 */
803 test_delta_action_item(world, objagg, action_item, true);
804 return err;
805 }
806
test_delta(void)807 static int test_delta(void)
808 {
809 struct world world = {};
810 struct objagg *objagg;
811 int i;
812 int err;
813
814 objagg = objagg_create(&delta_ops, NULL, &world);
815 if (IS_ERR(objagg))
816 return PTR_ERR(objagg);
817
818 for (i = 0; i < ARRAY_SIZE(action_items); i++) {
819 err = test_delta_action_item(&world, objagg,
820 &action_items[i], false);
821 if (err)
822 goto err_do_action_item;
823 }
824
825 objagg_destroy(objagg);
826 return 0;
827
828 err_do_action_item:
829 for (i--; i >= 0; i--)
830 test_delta_action_item(&world, objagg, &action_items[i], true);
831
832 objagg_destroy(objagg);
833 return err;
834 }
835
836 struct hints_case {
837 const unsigned int *key_ids;
838 size_t key_ids_count;
839 struct expect_stats expect_stats;
840 struct expect_stats expect_stats_hints;
841 };
842
843 static const unsigned int hints_case_key_ids[] = {
844 1, 7, 3, 5, 3, 1, 30, 8, 8, 5, 6, 8,
845 };
846
847 static const struct hints_case hints_case = {
848 .key_ids = hints_case_key_ids,
849 .key_ids_count = ARRAY_SIZE(hints_case_key_ids),
850 .expect_stats =
851 EXPECT_STATS(7, ROOT(1, 2, 7), ROOT(7, 1, 4), ROOT(30, 1, 1),
852 DELTA(8, 3), DELTA(3, 2),
853 DELTA(5, 2), DELTA(6, 1)),
854 .expect_stats_hints =
855 EXPECT_STATS(7, ROOT(3, 2, 9), ROOT(1, 2, 2), ROOT(30, 1, 1),
856 DELTA(8, 3), DELTA(5, 2),
857 DELTA(6, 1), DELTA(7, 1)),
858 };
859
__pr_debug_stats(const struct objagg_stats * stats)860 static void __pr_debug_stats(const struct objagg_stats *stats)
861 {
862 int i;
863
864 for (i = 0; i < stats->stats_info_count; i++)
865 pr_debug("Stat index %d key %u: u %d, d %d, %s\n", i,
866 obj_to_key_id(stats->stats_info[i].objagg_obj),
867 stats->stats_info[i].stats.user_count,
868 stats->stats_info[i].stats.delta_user_count,
869 stats->stats_info[i].is_root ? "root" : "noroot");
870 }
871
pr_debug_stats(struct objagg * objagg)872 static void pr_debug_stats(struct objagg *objagg)
873 {
874 const struct objagg_stats *stats;
875
876 stats = objagg_stats_get(objagg);
877 if (IS_ERR(stats))
878 return;
879 __pr_debug_stats(stats);
880 objagg_stats_put(stats);
881 }
882
pr_debug_hints_stats(struct objagg_hints * objagg_hints)883 static void pr_debug_hints_stats(struct objagg_hints *objagg_hints)
884 {
885 const struct objagg_stats *stats;
886
887 stats = objagg_hints_stats_get(objagg_hints);
888 if (IS_ERR(stats))
889 return;
890 __pr_debug_stats(stats);
891 objagg_stats_put(stats);
892 }
893
check_expect_hints_stats(struct objagg_hints * objagg_hints,const struct expect_stats * expect_stats,const char ** errmsg)894 static int check_expect_hints_stats(struct objagg_hints *objagg_hints,
895 const struct expect_stats *expect_stats,
896 const char **errmsg)
897 {
898 const struct objagg_stats *stats;
899 int err;
900
901 stats = objagg_hints_stats_get(objagg_hints);
902 if (IS_ERR(stats))
903 return PTR_ERR(stats);
904 err = __check_expect_stats(stats, expect_stats, errmsg);
905 objagg_stats_put(stats);
906 return err;
907 }
908
test_hints_case(const struct hints_case * hints_case)909 static int test_hints_case(const struct hints_case *hints_case)
910 {
911 struct objagg_obj *objagg_obj;
912 struct objagg_hints *hints;
913 struct world world2 = {};
914 struct world world = {};
915 struct objagg *objagg2;
916 struct objagg *objagg;
917 const char *errmsg;
918 int i;
919 int err;
920
921 objagg = objagg_create(&delta_ops, NULL, &world);
922 if (IS_ERR(objagg))
923 return PTR_ERR(objagg);
924
925 for (i = 0; i < hints_case->key_ids_count; i++) {
926 objagg_obj = world_obj_get(&world, objagg,
927 hints_case->key_ids[i]);
928 if (IS_ERR(objagg_obj)) {
929 err = PTR_ERR(objagg_obj);
930 goto err_world_obj_get;
931 }
932 }
933
934 pr_debug_stats(objagg);
935 err = check_expect_stats(objagg, &hints_case->expect_stats, &errmsg);
936 if (err) {
937 pr_err("Stats: %s\n", errmsg);
938 goto err_check_expect_stats;
939 }
940
941 hints = objagg_hints_get(objagg, OBJAGG_OPT_ALGO_SIMPLE_GREEDY);
942 if (IS_ERR(hints)) {
943 err = PTR_ERR(hints);
944 goto err_hints_get;
945 }
946
947 pr_debug_hints_stats(hints);
948 err = check_expect_hints_stats(hints, &hints_case->expect_stats_hints,
949 &errmsg);
950 if (err) {
951 pr_err("Hints stats: %s\n", errmsg);
952 goto err_check_expect_hints_stats;
953 }
954
955 objagg2 = objagg_create(&delta_ops, hints, &world2);
956 if (IS_ERR(objagg2))
957 return PTR_ERR(objagg2);
958
959 for (i = 0; i < hints_case->key_ids_count; i++) {
960 objagg_obj = world_obj_get(&world2, objagg2,
961 hints_case->key_ids[i]);
962 if (IS_ERR(objagg_obj)) {
963 err = PTR_ERR(objagg_obj);
964 goto err_world2_obj_get;
965 }
966 }
967
968 pr_debug_stats(objagg2);
969 err = check_expect_stats(objagg2, &hints_case->expect_stats_hints,
970 &errmsg);
971 if (err) {
972 pr_err("Stats2: %s\n", errmsg);
973 goto err_check_expect_stats2;
974 }
975
976 err = 0;
977
978 err_check_expect_stats2:
979 err_world2_obj_get:
980 for (i--; i >= 0; i--)
981 world_obj_put(&world2, objagg, hints_case->key_ids[i]);
982 i = hints_case->key_ids_count;
983 objagg_destroy(objagg2);
984 err_check_expect_hints_stats:
985 objagg_hints_put(hints);
986 err_hints_get:
987 err_check_expect_stats:
988 err_world_obj_get:
989 for (i--; i >= 0; i--)
990 world_obj_put(&world, objagg, hints_case->key_ids[i]);
991
992 objagg_destroy(objagg);
993 return err;
994 }
test_hints(void)995 static int test_hints(void)
996 {
997 return test_hints_case(&hints_case);
998 }
999
test_objagg_init(void)1000 static int __init test_objagg_init(void)
1001 {
1002 int err;
1003
1004 err = test_nodelta();
1005 if (err)
1006 return err;
1007 err = test_delta();
1008 if (err)
1009 return err;
1010 return test_hints();
1011 }
1012
test_objagg_exit(void)1013 static void __exit test_objagg_exit(void)
1014 {
1015 }
1016
1017 module_init(test_objagg_init);
1018 module_exit(test_objagg_exit);
1019 MODULE_LICENSE("Dual BSD/GPL");
1020 MODULE_AUTHOR("Jiri Pirko <jiri@mellanox.com>");
1021 MODULE_DESCRIPTION("Test module for objagg");
1022