py_net: make use of pytalloc_GenericObject_steal()

BUG: https://bugzilla.samba.org/show_bug.cgi?id=12601

Signed-off-by: Stefan Metzmacher <metze@samba.org>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

diff --git a/source4/libnet/py_net.c b/source4/libnet/py_net.c
index 3e70c79ea743fb38276ae489f1bae14f005e5eb1..5180c247a716b44f0fa1dc91f0c2005bc5d28f12 100644 (file)
--- a/source4/libnet/py_net.c
+++ b/source4/libnet/py_net.c
@@ -457,7 +457,7 @@ static PyObject *py_net_replicate_init(py_net_Object *self, PyObject *args, PyOb
        s->chunk.forest = &s->forest;
        s->chunk.dest_dsa = &s->dest_dsa;
 
-       return pytalloc_CObject_FromTallocPtr(s);
+       return pytalloc_GenericObject_steal(s);
 }
 
 
@@ -485,9 +485,8 @@ static PyObject *py_net_replicate_chunk(py_net_Object *self, PyObject *args, PyO
                return NULL;
        }
 
-       s = talloc_get_type(PyCObject_AsVoidPtr(py_state), struct replicate_state);
+       s = pytalloc_get_type(py_state, struct replicate_state);
        if (!s) {
-               PyErr_SetString(PyExc_TypeError, "Expected replication_state");
                return NULL;
        }