1 // SPDX-License-Identifier: BSD-3-Clause OR GPL-2.0
60 	if (!world->key_refs[key_id_index(key_id)]) {  in world_obj_get()
61 		world->objagg_objs[key_id_index(key_id)] = objagg_obj;  in world_obj_get()
62 	} else if (world->objagg_objs[key_id_index(key_id)] != objagg_obj) {  in world_obj_get()
65 		err = -EINVAL;  in world_obj_get()
68 	world->key_refs[key_id_index(key_id)]++;  in world_obj_get()
81 	if (!world->key_refs[key_id_index(key_id)])  in world_obj_put()
83 	objagg_obj = world->objagg_objs[key_id_index(key_id)];  in world_obj_put()
85 	world->key_refs[key_id_index(key_id)]--;  in world_obj_put()
88 #define MAX_KEY_ID_DIFF 5
94 	int diff = key->id - parent_key->id;  in delta_check()
104 	int diff = key->id - parent_key->id;  in delta_create()
108 		return ERR_PTR(-EINVAL);  in delta_create()
112 		return ERR_PTR(-ENOMEM);  in delta_create()
113 	delta->key_id_diff = diff;  in delta_create()
114 	world->delta_count++;  in delta_create()
123 	world->delta_count--;  in delta_destroy()
135 		return ERR_PTR(-ENOMEM);  in root_create()
136 	memcpy(&root->key, key, sizeof(root->key));  in root_create()
137 	memcpy(root->buf, world->next_root_buf, sizeof(root->buf));  in root_create()
138 	world->root_count++;  in root_create()
147 	world->root_count--;  in root_destroy()
154 	unsigned int orig_root_count = world->root_count;  in test_nodelta_obj_get()
160 		get_random_bytes(world->next_root_buf,  in test_nodelta_obj_get()
161 			      sizeof(world->next_root_buf));  in test_nodelta_obj_get()
169 		if (world->root_count != orig_root_count + 1) {  in test_nodelta_obj_get()
171 			err = -EINVAL;  in test_nodelta_obj_get()
175 		if (world->root_count != orig_root_count) {  in test_nodelta_obj_get()
178 			err = -EINVAL;  in test_nodelta_obj_get()
183 	if (root->key.id != key_id) {  in test_nodelta_obj_get()
185 		err = -EINVAL;  in test_nodelta_obj_get()
189 	    memcmp(world->next_root_buf, root->buf, sizeof(root->buf))) {  in test_nodelta_obj_get()
192 		err = -EINVAL;  in test_nodelta_obj_get()
207 	unsigned int orig_root_count = world->root_count;  in test_nodelta_obj_put()
212 		if (world->root_count != orig_root_count - 1) {  in test_nodelta_obj_put()
214 			return -EINVAL;  in test_nodelta_obj_put()
217 		if (world->root_count != orig_root_count) {  in test_nodelta_obj_put()
220 			return -EINVAL;  in test_nodelta_obj_put()
235 	if (stats->stats_info_count != 0) {  in check_stats_zero()
237 		err = -EINVAL;  in check_stats_zero()
254 	if (stats->stats_info_count != NUM_KEYS) {  in check_stats_nodelta()
256 		       NUM_KEYS, stats->stats_info_count);  in check_stats_nodelta()
257 		err = -EINVAL;  in check_stats_nodelta()
261 	for (i = 0; i < stats->stats_info_count; i++) {  in check_stats_nodelta()
262 		if (stats->stats_info[i].stats.user_count != 2) {  in check_stats_nodelta()
264 			err = -EINVAL;  in check_stats_nodelta()
267 		if (stats->stats_info[i].stats.delta_user_count != 2) {  in check_stats_nodelta()
269 			err = -EINVAL;  in check_stats_nodelta()
288 	return ERR_PTR(-EOPNOTSUPP);  in delta_create_dummy()
339 	for (i = NUM_KEYS - 1; i >= 0; i--) {  in test_nodelta()
344 	for (i = NUM_KEYS - 1; i >= 0; i--) {  in test_nodelta()
360 	for (i--; i >= 0; i--)  in test_nodelta()
366 	for (i--; i >= 0; i--)  in test_nodelta()
446 		5, ACTION_GET, EXPECT_DELTA_INC, EXPECT_ROOT_SAME,
448 				DELTA(3, 1), DELTA(5, 1)),
449 	},	/* r: 1, 7		d: 3^1, 5^1 */
453 				DELTA(3, 2), DELTA(5, 1)),
454 	},	/* r: 1, 7		d: 3^1, 3^1, 5^1 */
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 */
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 */
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 */
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 */
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 */
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 */
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 */
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 */
496 		5, ACTION_PUT, EXPECT_DELTA_DEC, EXPECT_ROOT_DEC,
498 				DELTA(8, 2)),
499 	},	/* r: 7, 30		d: 8^7, 8^7 */
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 */
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 */
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 */
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 */
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 */
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),
529 	},	/* r: 7, 30, 5		d: 6^5 */
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 */
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 */
542 		EXPECT_STATS(3, ROOT(5, 1, 3),
543 				DELTA(6, 1), DELTA(8, 1)),
544 	},	/* r: 5			d: 6^5, 8^5 */
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 */
552 		EXPECT_STATS(2, ROOT(5, 0, 1),
553 				DELTA(8, 1)),
554 	},	/* r:			d: 6^5 */
556 		8, ACTION_PUT, EXPECT_DELTA_DEC, EXPECT_ROOT_DEC,
566 	unsigned int key_id = action_item->key_id;  in check_expect()
568 	switch (action_item->expect_delta) {  in check_expect()
570 		if (orig_delta_count != world->delta_count) {  in check_expect()
573 			return -EINVAL;  in check_expect()
577 		if (WARN_ON(action_item->action == ACTION_PUT))  in check_expect()
578 			return -EINVAL;  in check_expect()
579 		if (orig_delta_count + 1 != world->delta_count) {  in check_expect()
582 			return -EINVAL;  in check_expect()
586 		if (WARN_ON(action_item->action == ACTION_GET))  in check_expect()
587 			return -EINVAL;  in check_expect()
588 		if (orig_delta_count - 1 != world->delta_count) {  in check_expect()
591 			return -EINVAL;  in check_expect()
596 	switch (action_item->expect_root) {  in check_expect()
598 		if (orig_root_count != world->root_count) {  in check_expect()
601 			return -EINVAL;  in check_expect()
605 		if (WARN_ON(action_item->action == ACTION_PUT))  in check_expect()
606 			return -EINVAL;  in check_expect()
607 		if (orig_root_count + 1 != world->root_count) {  in check_expect()
610 			return -EINVAL;  in check_expect()
614 		if (WARN_ON(action_item->action == ACTION_GET))  in check_expect()
615 			return -EINVAL;  in check_expect()
616 		if (orig_root_count - 1 != world->root_count) {  in check_expect()
619 			return -EINVAL;  in check_expect()
633 	key_id = root_key->id;  in obj_to_key_id()
636 		key_id += delta->key_id_diff;  in obj_to_key_id()
645 	if (stats_info->is_root != expect_stats_info->is_root) {  in check_expect_stats_nums()
648 		return -EINVAL;  in check_expect_stats_nums()
650 	if (stats_info->stats.user_count !=  in check_expect_stats_nums()
651 	    expect_stats_info->stats.user_count) {  in check_expect_stats_nums()
654 		return -EINVAL;  in check_expect_stats_nums()
656 	if (stats_info->stats.delta_user_count !=  in check_expect_stats_nums()
657 	    expect_stats_info->stats.delta_user_count) {  in check_expect_stats_nums()
660 		return -EINVAL;  in check_expect_stats_nums()
670 	if (obj_to_key_id(stats_info->objagg_obj) !=  in check_expect_stats_key_id()
671 	    expect_stats_info->key_id) {  in check_expect_stats_key_id()
674 		return -EINVAL;  in check_expect_stats_key_id()
686 	for (i = pos - 1; i >= 0; i--) {  in check_expect_stats_neigh()
687 		err = check_expect_stats_nums(&stats->stats_info[i],  in check_expect_stats_neigh()
688 					      &expect_stats->info[pos], NULL);  in check_expect_stats_neigh()
691 		err = check_expect_stats_key_id(&stats->stats_info[i],  in check_expect_stats_neigh()
692 						&expect_stats->info[pos], NULL);  in check_expect_stats_neigh()
696 	for (i = pos + 1; i < stats->stats_info_count; i++) {  in check_expect_stats_neigh()
697 		err = check_expect_stats_nums(&stats->stats_info[i],  in check_expect_stats_neigh()
698 					      &expect_stats->info[pos], NULL);  in check_expect_stats_neigh()
701 		err = check_expect_stats_key_id(&stats->stats_info[i],  in check_expect_stats_neigh()
702 						&expect_stats->info[pos], NULL);  in check_expect_stats_neigh()
706 	return -EINVAL;  in check_expect_stats_neigh()
716 	if (stats->stats_info_count != expect_stats->info_count) {  in __check_expect_stats()
718 		return -EINVAL;  in __check_expect_stats()
721 	for (i = 0; i < stats->stats_info_count; i++) {  in __check_expect_stats()
722 		err = check_expect_stats_nums(&stats->stats_info[i],  in __check_expect_stats()
723 					      &expect_stats->info[i], errmsg);  in __check_expect_stats()
726 		err = check_expect_stats_key_id(&stats->stats_info[i],  in __check_expect_stats()
727 						&expect_stats->info[i], errmsg);  in __check_expect_stats()
762 	unsigned int orig_delta_count = world->delta_count;  in test_delta_action_item()
763 	unsigned int orig_root_count = world->root_count;  in test_delta_action_item()
764 	unsigned int key_id = action_item->key_id;  in test_delta_action_item()
765 	enum action action = action_item->action;  in test_delta_action_item()
791 	err = check_expect_stats(objagg, &action_item->expect_stats, &errmsg);  in test_delta_action_item()
793 		pr_err("Key %u: Stats: %s\n", action_item->key_id, errmsg);  in test_delta_action_item()
829 	for (i--; i >= 0; i--)  in test_delta()
844 	1, 7, 3, 5, 3, 1, 30, 8, 8, 5, 6, 8,
852 				DELTA(8, 3), DELTA(3, 2),
853 				DELTA(5, 2), DELTA(6, 1)),
856 				DELTA(8, 3), DELTA(5, 2),
864 	for (i = 0; i < stats->stats_info_count; i++)  in __pr_debug_stats()
866 			 obj_to_key_id(stats->stats_info[i].objagg_obj),  in __pr_debug_stats()
867 			 stats->stats_info[i].stats.user_count,  in __pr_debug_stats()
868 			 stats->stats_info[i].stats.delta_user_count,  in __pr_debug_stats()
869 			 stats->stats_info[i].is_root ? "root" : "noroot");  in __pr_debug_stats()
925 	for (i = 0; i < hints_case->key_ids_count; i++) {  in test_hints_case()
927 					   hints_case->key_ids[i]);  in test_hints_case()
935 	err = check_expect_stats(objagg, &hints_case->expect_stats, &errmsg);  in test_hints_case()
948 	err = check_expect_hints_stats(hints, &hints_case->expect_stats_hints,  in test_hints_case()
959 	for (i = 0; i < hints_case->key_ids_count; i++) {  in test_hints_case()
961 					   hints_case->key_ids[i]);  in test_hints_case()
969 	err = check_expect_stats(objagg2, &hints_case->expect_stats_hints,  in test_hints_case()
980 	for (i--; i >= 0; i--)  in test_hints_case()
981 		world_obj_put(&world2, objagg, hints_case->key_ids[i]);  in test_hints_case()
982 	i = hints_case->key_ids_count;  in test_hints_case()
989 	for (i--; i >= 0; i--)  in test_hints_case()
990 		world_obj_put(&world, objagg, hints_case->key_ids[i]);  in test_hints_case()