commit 83e04f2023cb80366cf4f06a43f4df6ad2b5f08b
author Damien Miller <djm@mindrot.org> Mon Sep 17 16:11:01 2007 +1000
committer Damien Miller <djm@mindrot.org> Mon Sep 17 16:11:01 2007 +1000
tree 482cac77c11d376b8e81297f9d6f24d619de52c1
parent 1235cd387e0fee0595129f0fa55216ab0d040c94

   - stevesk@cvs.openbsd.org 2007/09/12 19:39:19
     [umac.c]
     use xmalloc() and xfree(); ok markus@ pvalchev@

umac.c

diff --git a/umac.c b/umac.c
index c2fdcf4..ca5e08b 100644
--- a/umac.c
+++ b/umac.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: umac.c,v 1.1 2007/06/07 19:37:34 pvalchev Exp $ */
+/* $OpenBSD: umac.c,v 1.2 2007/09/12 19:39:19 stevesk Exp $ */
 /* -----------------------------------------------------------------------
  * 
  * umac.c -- C Implementation UMAC Message Authentication
@@ -66,6 +66,7 @@
 #include "includes.h"
 #include <sys/types.h>
 
+#include "xmalloc.h"
 #include "umac.h"
 #include <string.h>
 #include <stdlib.h>
@@ -1196,7 +1197,7 @@
     if (ctx) {
         if (ALLOC_BOUNDARY)
             ctx = (struct umac_ctx *)ctx->free_ptr;
-        free(ctx);
+        xfree(ctx);
     }
     return (1);
 }
@@ -1212,7 +1213,7 @@
     size_t bytes_to_add;
     aes_int_key prf_key;
     
-    octx = ctx = malloc(sizeof(*ctx) + ALLOC_BOUNDARY);
+    octx = ctx = xmalloc(sizeof(*ctx) + ALLOC_BOUNDARY);
     if (ctx) {
         if (ALLOC_BOUNDARY) {
             bytes_to_add = ALLOC_BOUNDARY -