1 files changed, 10 insertions, 17 deletions
diff --git a/src/lib/crypto/krb/default_state.c b/src/lib/crypto/krb/default_state.c
index 121244871..c7bfe323f 100644
--- a/src/lib/crypto/krb/default_state.c
+++ b/src/lib/crypto/krb/default_state.c
@@ -1,6 +1,6 @@
 /* -*- mode: c; c-basic-offset: 4; indent-tabs-mode: nil -*- */
 /*
- * Copyright (C) 2001 by the Massachusetts Institute of Technology.
+ * Copyright (C) 2001, 2014 by the Massachusetts Institute of Technology.
  * All rights reserved.
  *
  * Export of this software from the United States of America may
@@ -34,28 +34,21 @@
 
 krb5_error_code
 krb5int_des_init_state(const krb5_keyblock *key, krb5_keyusage usage,
-                       krb5_data *new_state)
+                       krb5_data *state_out)
 {
-    new_state->length = 8;
-    new_state->data = (void *) malloc(8);
-    if (new_state->data) {
-        memset (new_state->data, 0, new_state->length);
-        /* We need to copy in the key for des-cbc-cr--ick, but that's how it works*/
-        if (key->enctype == ENCTYPE_DES_CBC_CRC) {
-            memcpy (new_state->data, key->contents, new_state->length);
-        }
-    } else {
+    if (alloc_data(state_out, 8))
         return ENOMEM;
-    }
+
+    /* des-cbc-crc uses the key as the initial ivec. */
+    if (key->enctype == ENCTYPE_DES_CBC_CRC)
+        memcpy(state_out->data, key->contents, state_out->length);
+
     return 0;
 }
 
 void
 krb5int_default_free_state(krb5_data *state)
 {
-    if (state->data) {
-        free (state->data);
-        state-> data = NULL;
-        state->length = 0;
-    }
+    free(state->data);
+    *state = empty_data();
 }