From: Volker Lendecke <vl@samba.org>
Date: Fri, 22 Jan 2021 18:40:37 +0000 (+0100)
Subject: librpc: Add a NULL check to dcerpc_binding_build_tower()
X-Git-Tag: tevent-0.11.0~1887
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=2e3e297b259ff32094db6fb4114726201f08ca2a;p=thirdparty%2Fsamba.git

librpc: Add a NULL check to dcerpc_binding_build_tower()

Signed-off-by: Volker Lendecke <vl@samba.org>
Reviewed-by: Samuel Cabrero <scabrero@samba.org>
---

diff --git a/librpc/rpc/binding.c b/librpc/rpc/binding.c
index 80ab1a45134..05daccca7df 100644
--- a/librpc/rpc/binding.c
+++ b/librpc/rpc/binding.c
@@ -1480,6 +1480,9 @@ _PUBLIC_ NTSTATUS dcerpc_binding_build_tower(TALLOC_CTX *mem_ctx,
 
 	tower->num_floors = 2 + num_protocols;
 	tower->floors = talloc_array(mem_ctx, struct epm_floor, tower->num_floors);
+	if (tower->floors == NULL) {
+		return NT_STATUS_NO_MEMORY;
+	}
 
 	/* Floor 0 */
 	tower->floors[0].lhs.protocol = EPM_PROTOCOL_UUID;