r6690: added ndr_pull_struct_blob_all(), which is like ndr_pull_struct_blob() but...

that all bytes are consumed

diff --git a/source/librpc/ndr/ndr.c b/source/librpc/ndr/ndr.c
index e21ec72ec80fa700e44585946fecbe8201020c19..79af5967c754e7ac56186b6e04dfbc82e6f18234 100644 (file)
--- a/source/librpc/ndr/ndr.c
+++ b/source/librpc/ndr/ndr.c
@@ -759,6 +759,27 @@ NTSTATUS ndr_pull_struct_blob(const DATA_BLOB *blob, TALLOC_CTX *mem_ctx, void *
        return fn(ndr, NDR_SCALARS|NDR_BUFFERS, p);
 }
 
        return fn(ndr, NDR_SCALARS|NDR_BUFFERS, p);
 }
 

+/*
+  pull a struct from a blob using NDR - failing if all bytes are not consumed
+*/
+NTSTATUS ndr_pull_struct_blob_all(const DATA_BLOB *blob, TALLOC_CTX *mem_ctx, void *p,
+                                 ndr_pull_flags_fn_t fn)
+{
+       struct ndr_pull *ndr;
+       NTSTATUS status;
+
+       ndr = ndr_pull_init_blob(blob, mem_ctx);
+       if (!ndr) {
+               return NT_STATUS_NO_MEMORY;
+       }
+       status = fn(ndr, NDR_SCALARS|NDR_BUFFERS, p);
+       if (!NT_STATUS_IS_OK(status)) return status;
+       if (ndr->offset != ndr->data_size) {
+               return NT_STATUS_BUFFER_TOO_SMALL;
+       }
+       return status;
+}
+

 /*
   push a struct to a blob using NDR
 */
 /*
   push a struct to a blob using NDR
 */