handle return value of krb5_storage_{read,write}

git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@24012 ec53bebd-3082-4978-b11e-865c3cabbd6b

1 files changed, 7 insertions, 5 deletions

diff --git a/lib/krb5/kcm.c b/lib/krb5/kcm.c
index 3fb9b1ec6..67720e60b 100644
--- a/lib/krb5/kcm.c
+++ b/lib/krb5/kcm.c
@@ -703,6 +703,7 @@ kcm_get_next (krb5_context context,
     krb5_kcm_cursor c = KCMCURSOR(*cursor);
     krb5_storage *request, *response;
     krb5_data response_data;
+    ssize_t sret;
 
  again:
 
@@ -719,13 +720,14 @@ kcm_get_next (krb5_context context,
 	return ret;
     }
 
-    ret = krb5_storage_write(request, 
-			     &c->uuids[c->offset],
-			     sizeof(c->uuids[c->offset]));
+    sret = krb5_storage_write(request, 
+			      &c->uuids[c->offset],
+			      sizeof(c->uuids[c->offset]));
     c->offset++;
-    if (ret) {
+    if (sret != sizeof(c->uuids[c->offset])) {
 	krb5_storage_free(request);
-	return ret;
+	krb5_clear_error_message(context);
+	return ENOMEM;
     }
 
     ret = kcm_call(context, k, request, &response, &response_data);