atomics: Add a ref_get() variant returning non-zero on overflows

This is useful for users using ref_get() for unique identifier allocation,
but the zero value has special meaning.
This commit is contained in:
Martin Willi
2024-02-16 10:11:11 +01:00
parent 31f55ba6e9
commit 0cd46df377
2 changed files with 51 additions and 0 deletions
@@ -256,6 +256,26 @@ START_TEST(test_round)
}
END_TEST
/*******************************************************************************
* ref_get/put
*/
START_TEST(test_refs)
{
refcount_t r = 0xfffffffe;
ck_assert_int_eq(ref_cur(&r), 0xfffffffe);
ck_assert_int_eq(ref_get(&r), 0xffffffff);
ck_assert_int_eq(ref_get_nonzero(&r), 1);
ck_assert_int_eq(ref_get_nonzero(&r), 2);
ck_assert_int_eq(ref_cur(&r), 2);
ck_assert(!ref_put(&r));
ck_assert_int_eq(ref_cur(&r), 1);
ck_assert(ref_put(&r));
ck_assert_int_eq(ref_cur(&r), 0);
}
END_TEST
/*******************************************************************************
* streq
*/
@@ -1272,6 +1292,10 @@ Suite *utils_suite_create()
tcase_add_test(tc, test_round);
suite_add_tcase(s, tc);
tc = tcase_create("refcount");
tcase_add_test(tc, test_refs);
suite_add_tcase(s, tc);
tc = tcase_create("string helper");
tcase_add_loop_test(tc, test_streq, 0, countof(streq_data));
tcase_add_loop_test(tc, test_strneq, 0, countof(strneq_data));