pyglue: Check generate_random_str() return value

Signed-off-by: Joseph Sutton <josephsutton@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

diff --git a/python/pyglue.c b/python/pyglue.c
index cbc1c791ec1735b2369644090708082cb8a9558b..b320a0b348d0b693fb17d52585cc5e670aeeb01d 100644 (file)
--- a/python/pyglue.c
+++ b/python/pyglue.c
@@ -48,6 +48,9 @@ static PyObject *py_generate_random_str(PyObject *self, PyObject *args)
                return NULL;
        }
        retstr = generate_random_str(NULL, len);
+       if (retstr == NULL) {
+               return PyErr_NoMemory();
+       }
        ret = PyUnicode_FromStringAndSize(retstr, len);
        talloc_free(retstr);
        return ret;