[SCM] Samba Shared Repository - branch master updated - a6bd295229cf1a245e2ced83ac4e10941207c526

Jelmer Vernooij jelmer at samba.org
Mon Dec 22 02:23:12 GMT 2008


The branch, master has been updated
       via  a6bd295229cf1a245e2ced83ac4e10941207c526 (commit)
       via  7ebc7e677d3d2a70c832a672e9f2e6b12896354f (commit)
       via  2c58c9497cb6a15d03626864585f0437949ee6f6 (commit)
      from  9a7c64f1dce7560c94df99c45eb5c75903cf0e6b (commit)

http://gitweb.samba.org/?p=samba.git;a=shortlog;h=master


- Log -----------------------------------------------------------------
commit a6bd295229cf1a245e2ced83ac4e10941207c526
Author: Jelmer Vernooij <jelmer at samba.org>
Date:   Mon Dec 22 03:22:28 2008 +0100

    pidl/python: Skip unknown types, rather than dieing.

commit 7ebc7e677d3d2a70c832a672e9f2e6b12896354f
Author: Jelmer Vernooij <jelmer at samba.org>
Date:   Mon Dec 22 03:21:10 2008 +0100

    pidl: Allow location argument to warning() and error() to be undef, in case it is
    not known.

commit 2c58c9497cb6a15d03626864585f0437949ee6f6
Author: Jelmer Vernooij <jelmer at samba.org>
Date:   Mon Dec 22 01:51:38 2008 +0100

    Import types from other Python mechanisms using the Python import mechanism, to ensure they are initialized.

-----------------------------------------------------------------------

Summary of changes:
 pidl/lib/Parse/Pidl.pm               |   10 ++-
 pidl/lib/Parse/Pidl/ODL.pm           |    4 +-
 pidl/lib/Parse/Pidl/Samba4/Python.pm |  128 +++++++++++++++++++++++----------
 pidl/lib/Parse/Pidl/Typelist.pm      |   23 ++++---
 pidl/pidl                            |    5 +-
 source4/librpc/rpc/pyrpc.h           |    4 +-
 6 files changed, 118 insertions(+), 56 deletions(-)


Changeset truncated at 500 lines:

diff --git a/pidl/lib/Parse/Pidl.pm b/pidl/lib/Parse/Pidl.pm
index c2c9463..40e3673 100644
--- a/pidl/lib/Parse/Pidl.pm
+++ b/pidl/lib/Parse/Pidl.pm
@@ -20,13 +20,19 @@ $VERSION = '0.02';
 sub warning
 {
 	my ($l,$m) = @_;
-	print STDERR "$l->{FILE}:$l->{LINE}: warning: $m\n";
+	if ($l) {
+		print STDERR "$l->{FILE}:$l->{LINE}: ";
+	}
+	print STDERR "warning: $m\n";
 }
 
 sub error
 {
 	my ($l,$m) = @_;
-	print STDERR "$l->{FILE}:$l->{LINE}: error: $m\n";
+	if ($l) {
+		print STDERR "$l->{FILE}:$l->{LINE}: ";
+	}
+	print STDERR "error: $m\n";
 }
 
 sub fatal($$) 
diff --git a/pidl/lib/Parse/Pidl/ODL.pm b/pidl/lib/Parse/Pidl/ODL.pm
index b323a6a..74d9ac7 100644
--- a/pidl/lib/Parse/Pidl/ODL.pm
+++ b/pidl/lib/Parse/Pidl/ODL.pm
@@ -8,6 +8,7 @@ use Parse::Pidl qw(error);
 use Parse::Pidl::IDL;
 use Parse::Pidl::Util qw(has_property unmake_str);
 use Parse::Pidl::Typelist qw(hasType getType);
+use File::Basename;
 use strict;
 
 use vars qw($VERSION);
@@ -71,8 +72,9 @@ sub ODL2IDL
 				my $podl = Parse::Pidl::IDL::parse_file($idl_path, $opt_incdirs);
 				if (defined(@$podl)) {
 					require Parse::Pidl::Typelist;
+					my $basename = basename($idl_path, ".idl");
 
-					Parse::Pidl::Typelist::LoadIdl($podl);
+					Parse::Pidl::Typelist::LoadIdl($podl, $basename);
 					my $pidl = ODL2IDL($podl, $basedir, $opt_incdirs);
 
 					foreach my $y (@$pidl) {
diff --git a/pidl/lib/Parse/Pidl/Samba4/Python.pm b/pidl/lib/Parse/Pidl/Samba4/Python.pm
index 73ae835..378c40c 100644
--- a/pidl/lib/Parse/Pidl/Samba4/Python.pm
+++ b/pidl/lib/Parse/Pidl/Samba4/Python.pm
@@ -9,7 +9,7 @@ use Exporter;
 @ISA = qw(Exporter);
 
 use strict;
-use Parse::Pidl qw(warning fatal);
+use Parse::Pidl qw(warning fatal error);
 use Parse::Pidl::Typelist qw(hasType resolveType getType mapTypeName expandAlias);
 use Parse::Pidl::Util qw(has_property ParseExpr unmake_str);
 use Parse::Pidl::NDR qw(GetPrevLevel GetNextLevel ContainsDeferred is_charset_array);
@@ -24,6 +24,7 @@ sub new($) {
 	my ($class) = @_;
 	my $self = { res => "", res_hdr => "", tabs => "", constants => {},
 	             module_methods => [], module_objects => [], ready_types => [],
+				 module_imports => [], type_imports => {},
 				 patch_type_calls => [], readycode => [] };
 	bless($self, $class);
 }
@@ -58,6 +59,15 @@ sub deindent($)
 	$self->{tabs} = substr($self->{tabs}, 0, -1);
 }
 
+sub PrettifyTypeName($$)
+{
+	my ($name, $basename) = @_;
+
+	$name =~ s/^$basename\_//;
+
+	return $name;
+}
+
 sub Import
 {
 	my $self = shift;
@@ -65,7 +75,8 @@ sub Import
 	foreach (@imports) {
 		$_ = unmake_str($_);
 		s/\.idl$//;
-		$self->pidl_hdr("#include \"librpc/gen_ndr/py_$_\.h\"\n");
+		$self->pidl_hdr("#include \"librpc/gen_ndr/$_\.h\"\n");
+		$self->register_module_import($_);
 	}
 }
 
@@ -287,14 +298,11 @@ sub PythonStruct($$$$$$)
 		$self->pidl("");
 	}
 
-	$self->pidl_hdr("PyAPI_DATA(PyTypeObject) $name\_Type;\n");
-	$self->pidl_hdr("#define $name\_Check(op) PyObject_TypeCheck(op, &$name\_Type)\n");
-	$self->pidl_hdr("#define $name\_CheckExact(op) ((op)->ob_type == &$name\_Type)\n");
-	$self->pidl_hdr("\n");
+	$self->pidl_hdr("staticforward PyTypeObject $name\_Type;\n");
 	$self->pidl("");
 	my $docstring = $self->DocString($d, $name);
 	my $typeobject = "$name\_Type";
-	$self->pidl("PyTypeObject $typeobject = {");
+	$self->pidl("static PyTypeObject $typeobject = {");
 	$self->indent;
 	$self->pidl("PyObject_HEAD_INIT(NULL) 0,");
 	$self->pidl(".tp_name = \"$modulename.$prettyname\",");
@@ -479,7 +487,7 @@ sub PythonFunctionPackIn($$$)
 		next unless (grep(/in/,@{$e->{DIRECTION}}));
 		if ($metadata_args->{in}->{$e->{NAME}}) {
 			my $py_var = "py_".$metadata_args->{in}->{$e->{NAME}};
-			$self->pidl("PY_CHECK_TYPE(PyList, $py_var, $fail);");
+			$self->pidl("PY_CHECK_TYPE(&PyList_Type, $py_var, $fail);");
 			my $val = "PyList_Size($py_var)";
 			if ($e->{LEVELS}[0]->{TYPE} eq "POINTER") {
 				$self->pidl("r->in.$e->{NAME} = talloc_ptrtype(r, r->in.$e->{NAME});");
@@ -565,11 +573,7 @@ sub PythonType($$$$)
 
 	if ($actual_ctype->{TYPE} eq "STRUCT") {
 		my $typeobject;
-		my $fn_name = $d->{NAME};
-
-		$fn_name =~ s/^$interface->{NAME}_//;
-		$fn_name =~ s/^$basename\_//;
-
+		my $fn_name = PrettifyTypeName($d->{NAME}, $basename);
 
 		if ($d->{TYPE} eq "STRUCT") {
 			$typeobject = $self->PythonStruct($modulename, $fn_name, $d->{NAME}, mapTypeName($d), $d);
@@ -623,11 +627,6 @@ sub Interface($$$)
 {
 	my($self,$interface,$basename) = @_;
 
-	$self->pidl_hdr("#ifndef _HEADER_PYTHON_$interface->{NAME}\n");
-	$self->pidl_hdr("#define _HEADER_PYTHON_$interface->{NAME}\n\n");
-
-	$self->pidl_hdr("\n");
-
 	if (has_property($interface, "pyhelper")) {
 		$self->pidl("#include \"".unmake_str($interface->{PROPERTIES}->{pyhelper})."\"\n");
 	}
@@ -641,7 +640,7 @@ sub Interface($$$)
 	}
 
 	if (defined $interface->{PROPERTIES}->{uuid}) {
-		$self->pidl_hdr("PyAPI_DATA(PyTypeObject) $interface->{NAME}_InterfaceType;\n");
+		$self->pidl_hdr("staticforward PyTypeObject $interface->{NAME}_InterfaceType;\n");
 		$self->pidl("");
 
 		my @fns = ();
@@ -661,7 +660,6 @@ sub Interface($$$)
 		}
 
 		$self->pidl("const struct PyNdrRpcMethodDef py_ndr_$interface->{NAME}\_methods[] = {");
-		$self->pidl_hdr("extern const struct PyNdrRpcMethodDef py_ndr_$interface->{NAME}\_methods[];");
 		$self->indent;
 		foreach my $d (@fns) {
 			my ($infn, $outfn, $callfn, $prettyname, $docstring, $opnum) = @$d;
@@ -775,7 +773,7 @@ sub Interface($$$)
 			$docstring = $signature;
 		}
 
-		$self->pidl("PyTypeObject $interface->{NAME}_InterfaceType = {");
+		$self->pidl("static PyTypeObject $interface->{NAME}_InterfaceType = {");
 		$self->indent;
 		$self->pidl("PyObject_HEAD_INIT(NULL) 0,");
 		$self->pidl(".tp_name = \"$basename.$interface->{NAME}\",");
@@ -794,7 +792,6 @@ sub Interface($$$)
 	}
 
 	$self->pidl_hdr("\n");
-	$self->pidl_hdr("#endif /* _HEADER_NDR_$interface->{NAME} */\n");
 }
 
 sub register_module_method($$$$$)
@@ -821,6 +818,31 @@ sub check_ready_type($$)
 	push (@{$self->{ready_types}}, $py_name) unless (grep(/^$py_name$/,@{$self->{ready_types}}));
 }
 
+sub register_module_import($$)
+{
+	my ($self, $basename) = @_;
+
+	push (@{$self->{module_imports}}, $basename) unless (grep(/^$basename$/,@{$self->{module_imports}}));
+}
+
+sub use_type_variable($$)
+{
+	my ($self, $orig_ctype) = @_;
+	my $ctype = resolveType($orig_ctype);
+	unless (defined($ctype->{BASEFILE})) {
+		return undef;
+	}
+	# If this is an external type, make sure we do the right imports.
+	if (($ctype->{BASEFILE} ne $self->{BASENAME})) {
+		$self->register_module_import($ctype->{BASEFILE});
+		unless (defined($self->{type_imports}->{$ctype->{NAME}})) {
+			$self->{type_imports}->{$ctype->{NAME}} = $ctype->{BASEFILE};
+		}
+		return "$ctype->{NAME}_Type";
+	}
+	return "&$ctype->{NAME}_Type";
+}
+
 sub register_patch_type_call($$$)
 {
 	my ($self, $typename, $cvar) = @_;
@@ -862,20 +884,27 @@ sub ConvertObjectFromPythonData($$$$$$)
 	$ctype = resolveType($ctype);
 
 	my $actual_ctype = $ctype;
-	if ($ctype->{TYPE} eq "TYPEDEF") {
-		$actual_ctype = $ctype->{DATA};
+	if ($actual_ctype->{TYPE} eq "TYPEDEF") {
+		$actual_ctype = $actual_ctype->{DATA};
 	}
 
 	if ($actual_ctype->{TYPE} eq "ENUM" or $actual_ctype->{TYPE} eq "BITMAP" or 
 		$actual_ctype->{TYPE} eq "SCALAR" and (
 		expandAlias($actual_ctype->{NAME}) =~ /^(u?int[0-9]*|hyper|NTTIME|time_t|NTTIME_hyper|NTTIME_1sec|dlong|udlong|udlongr)$/)) {
-		$self->pidl("PY_CHECK_TYPE(PyInt, $cvar, $fail);");
+		$self->pidl("PY_CHECK_TYPE(&PyInt_Type, $cvar, $fail);");
 		$self->pidl("$target = PyInt_AsLong($cvar);");
 		return;
 	}
 
 	if ($actual_ctype->{TYPE} eq "STRUCT" or $actual_ctype->{TYPE} eq "INTERFACE") {
-		$self->pidl("PY_CHECK_TYPE($ctype->{NAME}, $cvar, $fail);");
+		my $ctype_name = $self->use_type_variable($ctype);
+		unless (defined ($ctype_name)) {
+			error(undef, "Unable to determine origin of type " . mapTypeName($ctype));
+			$self->assign($target, "NULL");
+			# FIXME:
+			return;
+		}
+		$self->pidl("PY_CHECK_TYPE($ctype_name, $cvar, $fail);");
 		$self->assign($target, "py_talloc_get_ptr($cvar)");
 		return;
 	}
@@ -952,12 +981,12 @@ sub ConvertObjectFromPythonLevel($$$$$$$$)
 		}
 
 		if (is_charset_array($e, $l)) {
-			$self->pidl("PY_CHECK_TYPE(PyUnicode, $py_var, $fail);");
+			$self->pidl("PY_CHECK_TYPE(&PyUnicode_Type, $py_var, $fail);");
 			# FIXME: Use Unix charset setting rather than utf-8
 			$self->pidl($var_name . " = PyString_AsString(PyUnicode_AsEncodedString($py_var, \"utf-8\", \"ignore\"));");
 		} else {
 			my $counter = "$e->{NAME}_cntr_$l->{LEVEL_INDEX}";
-			$self->pidl("PY_CHECK_TYPE(PyList, $py_var, $fail);");
+			$self->pidl("PY_CHECK_TYPE(&PyList_Type, $py_var, $fail);");
 			$self->pidl("{");
 			$self->indent;
 			$self->pidl("int $counter;");
@@ -1043,8 +1072,8 @@ sub ConvertObjectToPythonData($$$$$)
 	$ctype = resolveType($ctype);
 
 	my $actual_ctype = $ctype;
-	if ($ctype->{TYPE} eq "TYPEDEF") {
-		$actual_ctype = $ctype->{DATA};
+	if ($actual_ctype->{TYPE} eq "TYPEDEF") {
+		$actual_ctype = $actual_ctype->{DATA};
 	} 
 	
 	if ($actual_ctype->{TYPE} eq "ENUM") {
@@ -1056,7 +1085,12 @@ sub ConvertObjectToPythonData($$$$$)
 	} elsif ($actual_ctype->{TYPE} eq "UNION") {
 		fatal($ctype, "union without discriminant: " . mapTypeName($ctype) . ": $cvar");
 	} elsif ($actual_ctype->{TYPE} eq "STRUCT" or $actual_ctype->{TYPE} eq "INTERFACE") {
-		return "py_talloc_import_ex(&$ctype->{NAME}_Type, $mem_ctx, $cvar)";
+		my $ctype_name = $self->use_type_variable($ctype);
+		unless (defined($ctype_name)) {
+			error(undef, "Unable to determine origin of type " . mapTypeName($ctype));
+			return "NULL"; # FIXME!
+		}
+		return "py_talloc_import_ex($ctype_name, $mem_ctx, $cvar)";
 	}
 
 	fatal($ctype, "unknown type $actual_ctype->{TYPE} for ".mapTypeName($ctype) . ": $cvar");
@@ -1158,13 +1192,10 @@ sub ConvertObjectToPython($$$$$$)
 sub Parse($$$$$)
 {
     my($self,$basename,$ndr,$ndr_hdr,$hdr) = @_;
-    
-    my $py_hdr = $hdr;
-    $py_hdr =~ s/ndr_([^\/]+)$/py_$1/g;
 
-    $self->pidl_hdr("/* header auto-generated by pidl */\n\n");
-	
-    $self->pidl("
+	$self->{BASENAME} = $basename;
+
+    $self->pidl_hdr("
 /* Python wrapper functions auto-generated by pidl */
 #include \"includes.h\"
 #include <Python.h>
@@ -1174,7 +1205,6 @@ sub Parse($$$$$)
 #include \"lib/events/events.h\"
 #include \"$hdr\"
 #include \"$ndr_hdr\"
-#include \"$py_hdr\"
 
 ");
 
@@ -1200,8 +1230,28 @@ sub Parse($$$$$)
 	$self->pidl("{");
 	$self->indent;
 	$self->pidl("PyObject *m;");
+	foreach (@{$self->{module_imports}}) {
+		$self->pidl("PyObject *dep_$_;");
+	}
 	$self->pidl("");
 
+	foreach (@{$self->{module_imports}}) {
+		$self->pidl("dep_$_ = PyImport_ImportModule(\"samba.dcerpc.$_\");");
+		$self->pidl("if (dep_$_ == NULL)");
+		$self->pidl("\treturn;");
+		$self->pidl("");
+	}
+
+	foreach (keys %{$self->{type_imports}}) {
+		my $basefile = $self->{type_imports}->{$_};
+		$self->pidl_hdr("static PyTypeObject *$_\_Type;\n");
+		my $pretty_name = PrettifyTypeName($_, $basefile);
+		$self->pidl("$_\_Type = (PyTypeObject *)PyObject_GetAttrString(dep_$basefile, \"$pretty_name\");");
+		$self->pidl("if ($_\_Type == NULL)");
+		$self->pidl("\treturn;");
+		$self->pidl("");
+	}
+
 	foreach (@{$self->{ready_types}}) {
 		$self->pidl("if (PyType_Ready($_) < 0)");
 		$self->pidl("\treturn;");
@@ -1249,7 +1299,7 @@ sub Parse($$$$$)
 	$self->pidl("");
 	$self->deindent;
 	$self->pidl("}");
-    return ($self->{res_hdr}, $self->{res});
+    return ($self->{res_hdr} . $self->{res});
 }
 
 1;
diff --git a/pidl/lib/Parse/Pidl/Typelist.pm b/pidl/lib/Parse/Pidl/Typelist.pm
index c5c458a..0e3fd8d 100644
--- a/pidl/lib/Parse/Pidl/Typelist.pm
+++ b/pidl/lib/Parse/Pidl/Typelist.pm
@@ -177,6 +177,7 @@ sub RegisterScalars()
 		addType({
 			NAME => $_,
 			TYPE => "TYPEDEF",
+			BASEFILE => "<builtin>",
 			DATA => {
 				TYPE => "SCALAR",
 				NAME => $_
@@ -266,9 +267,10 @@ sub mapTypeName($)
 	return mapType($dt, $dt->{NAME});
 }
 
-sub LoadIdl($)
+sub LoadIdl($;$)
 {
-	my ($idl) = @_;
+	my $idl = shift;
+	my $basename = shift;
 
 	foreach my $x (@{$idl}) {
 		next if $x->{TYPE} ne "INTERFACE";
@@ -277,16 +279,19 @@ sub LoadIdl($)
 		addType({
 			NAME => $x->{NAME},
 			TYPE => "TYPEDEF",
-			DATA => $x
+			DATA => $x,
+			BASEFILE => $basename,
 			}) if (has_property($x, "object"));
 
 		foreach my $y (@{$x->{DATA}}) {
-			addType($y) if (
-				$y->{TYPE} eq "TYPEDEF" 
-		 		 or $y->{TYPE} eq "UNION"
-		 		 or $y->{TYPE} eq "STRUCT"
-		         or $y->{TYPE} eq "ENUM"
-		         or $y->{TYPE} eq "BITMAP");
+			if ($y->{TYPE} eq "TYPEDEF" 
+		 		or $y->{TYPE} eq "UNION"
+		 		or $y->{TYPE} eq "STRUCT"
+		        or $y->{TYPE} eq "ENUM"
+		        or $y->{TYPE} eq "BITMAP") {
+				$y->{BASEFILE} = $basename;
+				addType($y);
+			}
 		}
 	}
 }
diff --git a/pidl/pidl b/pidl/pidl
index 02817c0..bc0bb35 100755
--- a/pidl/pidl
+++ b/pidl/pidl
@@ -609,7 +609,7 @@ sub process_file($)
 	}
 
 	require Parse::Pidl::Typelist;
-	Parse::Pidl::Typelist::LoadIdl($pidl);
+	Parse::Pidl::Typelist::LoadIdl($pidl, $basename);
 
 	if (defined($opt_dump_idl_tree)) {
 		my($pidl_file) = ($opt_dump_idl_tree or "$outputdir/$basename.pidl");
@@ -697,10 +697,9 @@ sub process_file($)
 	if (defined($opt_python)) {
 		require Parse::Pidl::Samba4::Python;
 		my $generator = new Parse::Pidl::Samba4::Python(); 
-		my ($hdr,$prsr) = $generator->Parse($basename, $ndr, 
+		my ($prsr) = $generator->Parse($basename, $ndr, 
 					"$outputdir/ndr_$basename\_c.h", $h_filename);
 		FileSave("$outputdir/py_$basename.c", $prsr);
-		FileSave("$outputdir/py_$basename.h", $hdr);
 	}
 
 	if (defined($opt_server)) {
diff --git a/source4/librpc/rpc/pyrpc.h b/source4/librpc/rpc/pyrpc.h
index af9ca72..3b2162c 100644
--- a/source4/librpc/rpc/pyrpc.h
+++ b/source4/librpc/rpc/pyrpc.h
@@ -24,8 +24,8 @@
 #include "librpc/rpc/dcerpc.h"
 
 #define PY_CHECK_TYPE(type, var, fail) \
-	if (!type ## _Check(var)) {\
-		PyErr_Format(PyExc_TypeError, "Expected type %s", type ## _Type.tp_name); \
+	if (!PyObject_TypeCheck(var, type)) {\
+		PyErr_Format(PyExc_TypeError, "Expected type %s", (type)->tp_name); \
 		fail; \
 	}
 


-- 
Samba Shared Repository


More information about the samba-cvs mailing list