oysteikt/heimdal
0
0
Fork 0
Code Issues 5 Pull Requests Releases Activity

kadm5: Make kadm5_get_principals() not quadratic

Browse Source
This commit is contained in:
Nicolas Williams
2022-03-18 18:33:01 -05:00
parent fa92fe37e7
commit 7827b3a91c
1 changed files with 24 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
lib/kadm5/get_princs_s.c
View File

@@ -39,17 +39,27 @@ struct foreach_data {
const char *exp;
char *exp2;
char **princs;
int count;
size_t nalloced;
size_t count;
};
static krb5_error_code
add_princ(krb5_context context, struct foreach_data *d, char *princ)
{
if (d->count == INT_MAX)
return ERANGE;
if (d->nalloced == d->count) {
size_t n = d->nalloced + (d->nalloced >> 1) + 128; /* No O(N^2) pls */
char **tmp;
tmp = realloc(d->princs, (d->count + 1) * sizeof(*tmp));
if (tmp == NULL)
if (SIZE_MAX / sizeof(*tmp) <= n)
return ERANGE;
if ((tmp = realloc(d->princs, n * sizeof(*tmp))) == NULL)
return krb5_enomem(context);
d->princs = tmp;
d->nalloced = n;
}
d->princs[d->count++] = princ;
return 0;
}
@@ -109,16 +119,20 @@ kadm5_s_get_principals(void *server_handle,
}
}
d.princs = NULL;
d.nalloced = 0;
d.count = 0;
ret = hdb_foreach(context->context, context->db, HDB_F_ADMIN_DATA, foreach, &d);
if (ret == 0)
ret = add_princ(context->context, &d, NULL);
if (ret == 0){
*princs = d.princs;
if (d.count >= INT_MAX)
*count = INT_MAX;
else
*count = d.count - 1;
} else
kadm5_free_name_list(context, d.princs, &d.count);
if (ret == 0)
*princs = d.princs;
else
kadm5_free_name_list(context, d.princs, count);
free(d.exp2);
out:
if (!context->keep_open)