2 files changed, 3 insertions, 5 deletions
diff --git a/source4/libcli/ndr/libndr.h b/source4/libcli/ndr/libndr.h
index 931fc1c341..7bd92b4cd1 100644
--- a/source4/libcli/ndr/libndr.h
+++ b/source4/libcli/ndr/libndr.h
@@ -58,10 +58,6 @@ struct ndr_push_save {
 	uint32 offset;
 };
 
-#define NDR_BASE_MARSHALL_SIZE 1024
-
-
-
 #define LIBNDR_FLAG_BIGENDIAN 1
 
 
diff --git a/source4/libcli/ndr/ndr.c b/source4/libcli/ndr/ndr.c
index eca5799d8e..4e5f199835 100644
--- a/source4/libcli/ndr/ndr.c
+++ b/source4/libcli/ndr/ndr.c
@@ -29,6 +29,8 @@
 
 #include "includes.h"
 
+#define NDR_BASE_MARSHALL_SIZE 1024
+
 /*
   initialise a ndr parse structure from a data blob
 */
@@ -166,7 +168,7 @@ NTSTATUS ndr_push_expand(struct ndr_push *ndr, uint32 size)
 	}
 
 	ndr->alloc_size = size;
-	ndr->data = realloc(ndr->data, ndr->alloc_size);
+	ndr->data = talloc_realloc(ndr->mem_ctx, ndr->data, ndr->alloc_size);
 	if (!ndr->data) {
 		return NT_STATUS_NO_MEMORY;
 	}