s4-pydsdb: Use local memory context in py_dsdb_get_oid_from_attid()

[kamenim/samba.git] / source4 / dsdb / pydsdb.c

diff --git a/source4/dsdb/pydsdb.c b/source4/dsdb/pydsdb.c
index af035042e641ddbbb8e6a5099c60af99d65429e6..1a51b86455483356a18b60396bf01bc575c5b9b4 100644 (file)
--- a/source4/dsdb/pydsdb.c
+++ b/source4/dsdb/pydsdb.c
@@ -249,25 +249,37 @@ static PyObject *py_dsdb_get_oid_from_attid(PyObject *self, PyObject *args)
        const char *oid;
        PyObject *ret;
        WERROR status;
+       TALLOC_CTX *mem_ctx;
 
        if (!PyArg_ParseTuple(args, "Oi", &py_ldb, &attid))
                return NULL;
 
        PyErr_LDB_OR_RAISE(py_ldb, ldb);
 
-       schema = dsdb_get_schema(ldb, NULL);
+       mem_ctx = talloc_new(NULL);
+       if (!mem_ctx) {
+               PyErr_NoMemory();
+               return NULL;
+       }
 
+       schema = dsdb_get_schema(ldb, mem_ctx);
        if (!schema) {
                PyErr_SetString(PyExc_RuntimeError, "Failed to find a schema from ldb \n");
+               talloc_free(mem_ctx);
                return NULL;
        }
        
        status = dsdb_schema_pfm_oid_from_attid(schema->prefixmap, attid,
-                                               NULL, &oid);
-       PyErr_WERROR_IS_ERR_RAISE(status);
+                                               mem_ctx, &oid);
+       if (!W_ERROR_IS_OK(status)) {
+               PyErr_SetWERROR(status);
+               talloc_free(mem_ctx);
+               return NULL;
+       }
 
        ret = PyString_FromString(oid);
-       talloc_free(discard_const_p(char, oid));
+
+       talloc_free(mem_ctx);
 
        return ret;
 }