diff --git a/lib/kadm5/get_princs_c.c b/lib/kadm5/get_princs_c.c index 4357e8353..d8260b80a 100644 --- a/lib/kadm5/get_princs_c.c +++ b/lib/kadm5/get_princs_c.c @@ -1,5 +1,5 @@ /* - * Copyright (c) 1997 Kungliga Tekniska Högskolan + * Copyright (c) 1997 - 1999 Kungliga Tekniska Högskolan * (Royal Institute of Technology, Stockholm, Sweden). * All rights reserved. * @@ -51,28 +51,41 @@ kadm5_c_get_principals(void *server_handle, krb5_storage *sp; unsigned char buf[1024]; int32_t tmp; + krb5_data reply; sp = krb5_storage_from_mem(buf, sizeof(buf)); + if (sp == NULL) + return ENOMEM; krb5_store_int32(sp, kadm_get_princs); krb5_store_int32(sp, exp != NULL); if(exp) krb5_store_string(sp, exp); ret = _kadm5_client_send(context, sp); - sp->seek(sp, SEEK_SET, 0); - ret = _kadm5_client_recv(context, sp); + krb5_storage_free(sp); + ret = _kadm5_client_recv(context, &reply); if(ret) - goto out; + return ret; + sp = krb5_storage_from_data (&reply); + if (sp == NULL) { + krb5_data_free (&reply); + return ENOMEM; + } krb5_ret_int32(sp, &tmp); ret = tmp; - if(ret == 0){ + if(ret == 0) { int i; krb5_ret_int32(sp, &tmp); *princs = calloc(tmp + 1, sizeof(**princs)); + if (*princs == NULL) { + ret = ENOMEM; + goto out; + } for(i = 0; i < tmp; i++) krb5_ret_string(sp, &(*princs)[i]); *count = tmp; } out: krb5_storage_free(sp); + krb5_data_free (&reply); return ret; }