pidl-python: cope with NULL pointers in more places

NULL is converted to Py_None

diff --git a/lib/talloc/pytalloc.c b/lib/talloc/pytalloc.c
index ccf27f5fa0fe3b9d8ec1f252ad94c31925242965..c9c6027fe08621615b5b0a1d551bb9bb6849e649 100644 (file)
--- a/lib/talloc/pytalloc.c
+++ b/lib/talloc/pytalloc.c
@@ -106,5 +106,25 @@ static void py_cobject_talloc_free(void *ptr)
 
 PyObject *PyCObject_FromTallocPtr(void *ptr)
 {
 
 PyObject *PyCObject_FromTallocPtr(void *ptr)
 {

+       if (ptr == NULL) {
+               Py_RETURN_NONE;
+       }

        return PyCObject_FromVoidPtr(ptr, py_cobject_talloc_free);
 }
        return PyCObject_FromVoidPtr(ptr, py_cobject_talloc_free);
 }

+
+PyObject *PyString_FromString_check_null(const char *ptr)
+{
+       if (ptr == NULL) {
+               Py_RETURN_NONE;
+       }
+       return PyString_FromString(ptr);
+}
+
+PyObject *PyUnicode_Decode_check_null(const void *ptr, size_t len,
+                                     const char *charset, const char *options)
+{
+       if (ptr == NULL) {
+               Py_RETURN_NONE;
+       }
+       return PyUnicode_Decode(ptr, len, charset, options);
+}

diff --git a/lib/talloc/pytalloc.h b/lib/talloc/pytalloc.h
index cdcc57bf85de1fae541c222fe7feaf98cb292c83..caa388ac5063194bcbfaf82303a3ebf407496d7a 100644 (file)
--- a/lib/talloc/pytalloc.h
+++ b/lib/talloc/pytalloc.h
@@ -54,4 +54,8 @@ PyObject *py_talloc_default_repr(PyObject *py_obj);
 
 PyObject *PyCObject_FromTallocPtr(void *);
 
 
 PyObject *PyCObject_FromTallocPtr(void *);
 

+PyObject *PyString_FromString_check_null(const char *ptr);
+PyObject *PyUnicode_Decode_check_null(const void *ptr, size_t len,
+                                     const char *charset, const char *options);
+

 #endif /* _PY_TALLOC_H_ */
 #endif /* _PY_TALLOC_H_ */

diff --git a/pidl/lib/Parse/Pidl/Samba4/Python.pm b/pidl/lib/Parse/Pidl/Samba4/Python.pm
index 05c11be07f47fc004b31c6de0ce9f58b96fa71c4..85ce2d64bae266f5698324e417d340e4501c1310 100644 (file)
--- a/pidl/lib/Parse/Pidl/Samba4/Python.pm
+++ b/pidl/lib/Parse/Pidl/Samba4/Python.pm
@@ -873,7 +873,7 @@ sub ConvertObjectFromPythonData($$$$$$;$)
                my $ctype_name = $self->use_type_variable($ctype);
                unless (defined ($ctype_name)) {
                        error($location, "Unable to determine origin of type `" . mapTypeName($ctype) . "'");
                my $ctype_name = $self->use_type_variable($ctype);
                unless (defined ($ctype_name)) {
                        error($location, "Unable to determine origin of type `" . mapTypeName($ctype) . "'");

-                       $self->pidl("PyErr_SetString(PyExc_TypeError, \"Can not convert C Type " . mapTypeName($ctype) . " to Python\");");
+                       $self->pidl("PyErr_SetString(PyExc_TypeError, \"Can not convert C Type " . mapTypeName($ctype) . " from Python\");");

                        return;
                }
                $self->pidl("PY_CHECK_TYPE($ctype_name, $cvar, $fail);");
                        return;
                }
                $self->pidl("PY_CHECK_TYPE($ctype_name, $cvar, $fail);");


@@ -895,12 +895,12 @@ sub ConvertObjectFromPythonData($$$$$$;$)
        if ($actual_ctype->{TYPE} eq "SCALAR" and $actual_ctype->{NAME} eq "ipv4address") {
                $self->pidl("$target = PyString_AsString($cvar);");
                return;
        if ($actual_ctype->{TYPE} eq "SCALAR" and $actual_ctype->{NAME} eq "ipv4address") {
                $self->pidl("$target = PyString_AsString($cvar);");
                return;

-               }
+       }

 
        if ($actual_ctype->{TYPE} eq "SCALAR" and $actual_ctype->{NAME} eq "dnsp_name") {
                $self->pidl("$target = PyString_AsString($cvar);");
                return;
 
        if ($actual_ctype->{TYPE} eq "SCALAR" and $actual_ctype->{NAME} eq "dnsp_name") {
                $self->pidl("$target = PyString_AsString($cvar);");
                return;

-               }
+       }

 
 
        if ($actual_ctype->{TYPE} eq "SCALAR" and $actual_ctype->{NAME} eq "NTSTATUS") {
 
 
        if ($actual_ctype->{TYPE} eq "SCALAR" and $actual_ctype->{NAME} eq "NTSTATUS") {


@@ -1050,13 +1050,13 @@ sub ConvertScalarToPython($$$)
        }
 
        if (($ctypename eq "string" or $ctypename eq "nbt_string" or $ctypename eq "nbt_name" or $ctypename eq "wrepl_nbt_name")) {
        }
 
        if (($ctypename eq "string" or $ctypename eq "nbt_string" or $ctypename eq "nbt_name" or $ctypename eq "wrepl_nbt_name")) {

-               return "PyString_FromString($cvar)";
+               return "PyString_FromString_check_null($cvar)";

        }
 
        # Not yet supported
        if ($ctypename eq "string_array") { return "PyCObject_FromTallocPtr($cvar)"; }
        }
 
        # Not yet supported
        if ($ctypename eq "string_array") { return "PyCObject_FromTallocPtr($cvar)"; }

-       if ($ctypename eq "ipv4address") { return "PyString_FromString($cvar)"; }
-       if ($ctypename eq "dnsp_name") { return "PyString_FromString($cvar)"; }
+       if ($ctypename eq "ipv4address") { return "PyString_FromString_check_null($cvar)"; }
+       if ($ctypename eq "dnsp_name") { return "PyString_FromString_check_null($cvar)"; }

        if ($ctypename eq "pointer") {
                return "PyCObject_FromTallocPtr($cvar)";
        }
        if ($ctypename eq "pointer") {
                return "PyCObject_FromTallocPtr($cvar)";
        }


@@ -1139,7 +1139,7 @@ sub ConvertObjectToPythonLevel($$$$$$)
 
                if (is_charset_array($e, $l)) {
                        # FIXME: Use Unix charset setting rather than utf-8
 
                if (is_charset_array($e, $l)) {
                        # FIXME: Use Unix charset setting rather than utf-8

-                       $self->pidl("$py_var = PyUnicode_Decode($var_name, strlen($var_name), \"utf-8\", \"ignore\");");
+                       $self->pidl("$py_var = PyUnicode_Decode_check_null($var_name, strlen($var_name), \"utf-8\", \"ignore\");");

                } else {
                        die("No SIZE_IS for array $var_name") unless (defined($l->{SIZE_IS}));
                        my $length = $l->{SIZE_IS};
                } else {
                        die("No SIZE_IS for array $var_name") unless (defined($l->{SIZE_IS}));
                        my $length = $l->{SIZE_IS};