From: Volker Lendecke <vl@samba.org>
Date: Tue, 9 Jan 2018 09:23:35 +0000 (+0100)
Subject: libnet: Add NULL checks to py_net_finddc
X-Git-Tag: talloc-2.1.11~30
X-Git-Url: http://git.samba.org/?a=commitdiff_plain;h=3022da1a7267f9038f1f0de98d8d54baabd9c567;p=samba.git

libnet: Add NULL checks to py_net_finddc

Signed-off-by: Volker Lendecke <vl@samba.org>
Reviewed-by: Stefan Metzmacher <metze@samba.org>
---

diff --git a/source4/libnet/py_net.c b/source4/libnet/py_net.c
index 7ddee2df92c..0567dbd6353 100644
--- a/source4/libnet/py_net.c
+++ b/source4/libnet/py_net.c
@@ -697,8 +697,18 @@ static PyObject *py_net_finddc(py_net_Object *self, PyObject *args, PyObject *kw
 	}
 
 	mem_ctx = talloc_new(self->mem_ctx);
+	if (mem_ctx == NULL) {
+		PyErr_NoMemory();
+		return NULL;
+	}
 
 	io = talloc_zero(mem_ctx, struct finddcs);
+	if (io == NULL) {
+		TALLOC_FREE(mem_ctx);
+		PyErr_NoMemory();
+		return NULL;
+	}
+
 	if (domain != NULL) {
 		io->in.domain_name = domain;
 	}