From 28f73a394389b14f6e19a33991514b877889e195 Mon Sep 17 00:00:00 2001
From: Assar Westerlund <assar@sics.se>
Date: Mon, 13 Sep 1999 02:56:42 +0000
Subject: [PATCH] (kadm5_c_modify_principal): new _kadm5_client_recv

git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@6956 ec53bebd-3082-4978-b11e-865c3cabbd6b
---
 lib/kadm5/modify_c.c | 27 ++++++++++++++++++---------
 1 file changed, 18 insertions(+), 9 deletions(-)

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;
 }