diff --git a/lib/kadm5/modify_c.c b/lib/kadm5/modify_c.c index 13ca731dc..c488bd4ac 100644 --- a/lib/kadm5/modify_c.c +++ b/lib/kadm5/modify_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. * @@ -50,20 +50,29 @@ kadm5_c_modify_principal(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_modify); kadm5_store_principal_ent(sp, princ); krb5_store_int32(sp, mask); ret = _kadm5_client_send(context, sp); - sp->seek(sp, SEEK_SET, 0); - ret = _kadm5_client_recv(context, sp); - if(ret) - goto out; - krb5_ret_int32(sp, &tmp); - ret = tmp; -out: krb5_storage_free(sp); - return ret; + if(ret) + return ret; + ret = _kadm5_client_recv(context, &reply); + if(ret) + return ret; + sp = krb5_storage_from_data (&reply); + if (sp == NULL) { + krb5_data_free (&reply); + return ENOMEM; + } + krb5_ret_int32(sp, &tmp); + krb5_storage_free(sp); + krb5_data_free (&reply); + return tmp; }