Compare commits

...

3 Commits

Author SHA1 Message Date
Mike Yuan e063993af0
Merge da80af70d2 into 94eacb9329 2024-11-24 19:20:11 +01:00
Mike Yuan da80af70d2
basic/user-util: modernize getgroups_alloc() a bit
- Make sure ret is initialized if we return >= 0
- Reduce variable scope
2024-11-21 17:02:15 +01:00
Mike Yuan 62d6972874
basic/user-util: use FOREACH_ARRAY at one more place 2024-11-21 15:23:26 +01:00
3 changed files with 25 additions and 25 deletions

View File

@ -467,9 +467,12 @@ char* gid_to_name(gid_t gid) {
}
static bool gid_list_has(const gid_t *list, size_t size, gid_t val) {
for (size_t i = 0; i < size; i++)
if (list[i] == val)
assert(list || size == 0);
FOREACH_ARRAY(i, list, size)
if (*i == val)
return true;
return false;
}
@ -526,20 +529,24 @@ int merge_gid_lists(const gid_t *list1, size_t size1, const gid_t *list2, size_t
return (int)nresult;
}
int getgroups_alloc(gid_t** gids) {
gid_t *allocated;
_cleanup_free_ gid_t *p = NULL;
int getgroups_alloc(gid_t **ret) {
int ngroups = 8;
unsigned attempt = 0;
allocated = new(gid_t, ngroups);
if (!allocated)
return -ENOMEM;
p = allocated;
assert(ret);
for (unsigned attempt = 0;;) {
_cleanup_free_ gid_t *p = NULL;
p = new(gid_t, ngroups);
if (!p)
return -ENOMEM;
for (;;) {
ngroups = getgroups(ngroups, p);
if (ngroups >= 0)
if (ngroups > 0) {
*ret = TAKE_PTR(p);
return ngroups;
}
if (ngroups == 0)
break;
if (errno != EINVAL)
return -errno;
@ -554,17 +561,11 @@ int getgroups_alloc(gid_t** gids) {
if (ngroups < 0)
return -errno;
if (ngroups == 0)
return false;
free(allocated);
p = allocated = new(gid_t, ngroups);
if (!allocated)
return -ENOMEM;
break;
}
*gids = TAKE_PTR(p);
return ngroups;
*ret = NULL;
return 0;
}
int get_home_dir(char **ret) {

View File

@ -64,7 +64,7 @@ int in_gid(gid_t gid);
int in_group(const char *name);
int merge_gid_lists(const gid_t *list1, size_t size1, const gid_t *list2, size_t size2, gid_t **result);
int getgroups_alloc(gid_t** gids);
int getgroups_alloc(gid_t **ret);
int get_home_dir(char **ret);
int get_shell(char **ret);

View File

@ -444,9 +444,8 @@ TEST(gid_lists_ops) {
assert_se(nresult >= 0);
assert_se(memcmp_nn(result2, ELEMENTSOF(result2), res4, nresult) == 0);
nresult = getgroups_alloc(&gids);
assert_se(nresult >= 0 || nresult == -EINVAL || nresult == -ENOMEM);
assert_se(gids);
ASSERT_OK(nresult = getgroups_alloc(&gids));
assert_se(gids || nresult == 0);
}
TEST(parse_uid_range) {