summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--source4/build/pidl/swig.pm20
1 files changed, 18 insertions, 2 deletions
diff --git a/source4/build/pidl/swig.pm b/source4/build/pidl/swig.pm
index dcd3efa698..e43454a2fc 100644
--- a/source4/build/pidl/swig.pm
+++ b/source4/build/pidl/swig.pm
@@ -231,7 +231,13 @@ sub ParseStruct($)
$res .= "struct $s->{NAME} *$s->{NAME}_ptr_from_python(TALLOC_CTX *mem_ctx, PyObject *obj)\n";
$res .= "{\n";
- $res .= "\tstruct $s->{NAME} *s = talloc(mem_ctx, sizeof(struct $s->{NAME}));\n\n";
+ $res .= "\tstruct $s->{NAME} *s;\n\n";
+
+ $res .= "\tif (obj == Py_None) {\n";
+ $res .= "\t\treturn NULL;\n";
+ $res .= "\t}\n\n";
+
+ $res .= "\ts = talloc(mem_ctx, sizeof(struct $s->{NAME}));\n\n";
foreach my $e (@{$s->{DATA}{ELEMENTS}}) {
$res .= XFromPython($e, "");
@@ -259,6 +265,11 @@ sub ParseStruct($)
$res .= "\tPyObject *obj = PyDict_New();\n\n";
+ $res .= "\tif (s == NULL) {\n";
+ $res .= "\t\tPy_INCREF(Py_None);\n";
+ $res .= "\t\treturn Py_None;\n";
+ $res .= "\t}\n\n";
+
foreach my $e (@{$s->{DATA}{ELEMENTS}}) {
$res .= XToPython($e, "");
}
@@ -282,7 +293,6 @@ sub ParseUnion($)
$res .= "\tunion $u->{NAME} *u = talloc(mem_ctx, sizeof(union $u->{NAME}));\n";
$res .= "\tPyObject *dict;\n\n";
-
for my $e (@{$u->{DATA}{DATA}}) {
$res .= "\tif ((dict = PyDict_GetItem(obj, PyString_FromString(\"$e->{DATA}{NAME}\")))) {\n";
@@ -305,6 +315,12 @@ sub ParseUnion($)
$res .= "PyObject *$u->{NAME}_ptr_to_python(TALLOC_CTX *mem_ctx, union $u->{NAME} *u)\n";
$res .= "{\n";
+
+ $res .= "\tif (u == NULL) {\n";
+ $res .= "\t\tPy_INCREF(Py_None);\n";
+ $res .= "\t\treturn Py_None;\n";
+ $res .= "\t}\n\n";
+
$res .= "\treturn PyDict_New();\n";
$res .= "}\n\n";