svn commit: samba r11642 - in branches/SAMBA_4_0/source/lib/stream: .

[metze at samba.org]

Author: metze
Date: 2005-11-10 13:52:07 +0000 (Thu, 10 Nov 2005)
New Revision: 11642

WebSVN: http://websvn.samba.org/cgi-bin/viewcvs.cgi?view=rev&root=samba&rev=11642

Log:
add some error checks

metze
Modified:
   branches/SAMBA_4_0/source/lib/stream/packet.c


Changeset:
Modified: branches/SAMBA_4_0/source/lib/stream/packet.c
===================================================================
--- branches/SAMBA_4_0/source/lib/stream/packet.c	2005-11-10 13:02:56 UTC (rev 11641)
+++ branches/SAMBA_4_0/source/lib/stream/packet.c	2005-11-10 13:52:07 UTC (rev 11642)
@@ -317,7 +317,11 @@
 			packet_error(pc, NT_STATUS_NO_MEMORY);
 			return;
 		}
-		data_blob_realloc(pc, &blob, pc->packet_size);
+		status = data_blob_realloc(pc, &blob, pc->packet_size);
+		if (!NT_STATUS_IS_OK(status)) {
+			packet_error(pc, status);
+			return;
+		}
 	} else {
 		pc->partial = data_blob(NULL, 0);
 	}
@@ -440,7 +444,9 @@
 	/* if we aren't going to free the packet then we must reference it
 	   to ensure it doesn't disappear before going out */
 	if (pc->nofree) {
-		talloc_reference(el, blob.data);
+		if (!talloc_reference(el, blob.data)) {
+			return NT_STATUS_NO_MEMORY;
+		}
 	} else {
 		talloc_steal(el, blob.data);
 	}