svn commit: samba r3204 - in branches/SAMBA_4_0/source: build/pidl scripting/swig/torture

tpot at samba.org tpot at samba.org
Mon Oct 25 07:32:00 GMT 2004


Author: tpot
Date: 2004-10-25 07:31:59 +0000 (Mon, 25 Oct 2004)
New Revision: 3204

WebSVN: http://websvn.samba.org/websvn/changeset.php?rep=samba&path=/branches/SAMBA_4_0/source&rev=3204&nolog=1

Log:
Got unpacking of an array of PrinterInfo union elements working.  Phew.

Modified:
   branches/SAMBA_4_0/source/build/pidl/swig.pm
   branches/SAMBA_4_0/source/scripting/swig/torture/spoolss.py


Changeset:
Modified: branches/SAMBA_4_0/source/build/pidl/swig.pm
===================================================================
--- branches/SAMBA_4_0/source/build/pidl/swig.pm	2004-10-25 07:24:46 UTC (rev 3203)
+++ branches/SAMBA_4_0/source/build/pidl/swig.pm	2004-10-25 07:31:59 UTC (rev 3204)
@@ -462,61 +462,8 @@
     $result .= "\treturn obj;\n";
     $result .= "}\n\n";
 
-    # Generate function to convert DATA_BLOB to Python dict and an array
-    # of Python dicts.
-
-    if (util::has_property($s->{DATA}, "public")) {
-
-	$result .= "/* Convert a Python string to a struct $s->{NAME} python dict */\n\n";
-	$result .= "NTSTATUS unmarshall_$s->{NAME}(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, struct $s->{NAME} *s)\n";
-	$result .= "{\n";
-	$result .= "\tstruct ndr_pull *ndr = ndr_pull_init_blob(blob, mem_ctx);\n";
-	$result .= "\tNTSTATUS result;\n\n";
-	$result .= "\tresult = ndr_pull_struct_blob(blob, mem_ctx, s, ndr_pull_$s->{NAME});\n";
-	$result .= "\treturn result;\n";
-	$result .= "}\n\n";
-
-	$result .= "/* Convert a Python string to an array of struct $s->{NAME} python dicts */\n\n";
-	$result .= "NTSTATUS unmarshall_$s->{NAME}_array(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, uint32 count, struct $s->{NAME} **s)\n";
-	$result .= "{\n";
-	$result .= "\tstruct ndr_pull *ndr = ndr_pull_init_blob(blob, mem_ctx);\n";
-	$result .= "\tNTSTATUS result;\n\n";
-	$result .= "\t*s = talloc(mem_ctx, sizeof(struct $s->{NAME}) * count);\n\n";
-	$result .= "\tresult = ndr_pull_array(ndr, NDR_SCALARS|NDR_BUFFERS, *s, sizeof(struct $s->{NAME}), count, ndr_pull_$s->{NAME});\n\n";
-	$result .= "\treturn result;\n";
-	$result .= "}\n\n";
-    }
-
     $result .= "%}\n\n";    
 
-    if (util::has_property($s->{DATA}, "public")) {
-
-	$result .= "%typemap(in, numinputs=0) struct $s->{NAME} *EMPTY (struct $s->{NAME} temp_$s->{NAME}) {\n";
-	$result .= "\t\$1 = &temp_$s->{NAME};\n";
-	$result .= "}\n\n";
-
-	$result .= "%typemap(argout) struct $s->{NAME} *EMPTY {\n";
-	$result .= "\tTALLOC_CTX *mem_ctx = talloc_init(\"typemap(argout) $s->{NAME} *\");\n\n";
-	$result .= "\t\$result = $s->{NAME}_ptr_to_python(mem_ctx, \$1);\n";
-	$result .= "}\n\n";
-
-	$result .= "NTSTATUS unmarshall_$s->{NAME}(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, struct $s->{NAME} *EMPTY);\n\n";
-
-	$result .= "%typemap(in, numinputs=0) struct $s->{NAME} **ARRAY (struct $s->{NAME} *temp_$s->{NAME}) {\n";
-	$result .= "\t\$1 = &temp_$s->{NAME};\n";
-	$result .= "}\n\n";
-
-	$result .= "%typemap(argout) (uint32 count, struct $s->{NAME} **ARRAY) {\n";
-	$result .= "\tTALLOC_CTX *mem_ctx = talloc_init(\"typemap(argout) $s->{NAME} **\");\n";
-	$result .= "\tuint32 i;\n\n";
-	$result .= "\t\$result = PyList_New(\$1);\n\n";
-	$result .= "\tfor (i = 0; i < \$1; i++)\n";
-	$result .= "\t\tPyList_SetItem(\$result, i, spoolss_PrinterInfo1_ptr_to_python(mem_ctx, \$2[i]));\n";
-	$result .= "}\n\n";
-
-	$result .= "NTSTATUS unmarshall_$s->{NAME}_array(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, uint32 count, struct $s->{NAME} **ARRAY);\n\n";
-    }
-
     return $result;
 }
 
@@ -633,8 +580,52 @@
 
     $result .= "}\n\n";
 
+    if (util::has_property($u->{DATA}, "public")) {
+
+	# Generate function to unmarshall an array of structures.
+	# Used exclusively (?) in the spoolss pipe.
+
+	$result .= "/* Unmarshall an array of structures from a Python string */\n\n";
+
+	$result .= "NTSTATUS unmarshall_$u->{NAME}_array(DATA_BLOB *blob, TALLOC_CTX *mem_ctx, uint32 level, uint32 count, union $u->{NAME} **info)\n";
+	$result .= "{\n";
+	$result .= "\tint i;\n";
+	$result .= "\tstruct ndr_pull *ndr;\n";
+	$result .= "\tndr = ndr_pull_init_blob(blob, mem_ctx);\n";
+	$result .= "\tif (!ndr) {\n";
+	$result .= "\t\treturn NT_STATUS_NO_MEMORY;\n";
+	$result .= "\t}\n";
+	$result .= "\tNDR_ALLOC_N(ndr, (*info), count);\n";
+	$result .= "\tfor (i=0;i<count;i++) {\n";
+	$result .= "\t\tndr->data += ndr->offset;\n";
+	$result .= "\t\tndr->offset = 0;\n";
+	$result .= "\t\tNDR_CHECK(ndr_pull_spoolss_PrinterInfo(ndr, NDR_SCALARS|NDR_BUFFERS, level, &(*info)[i]));\n";
+	$result .= "\t}\n\n";
+	$result .= "\treturn NT_STATUS_OK;\n";
+	$result .= "\t}\n";
+    }
+
     $result .= "%}\n\n";    
 
+    if (util::has_property($u->{DATA}, "public")) {
+
+	$result .= "%typemap(in, numinputs=0) union $u->{NAME} **EMPTY (union $u->{NAME} *temp_$u->{NAME}) {\n";
+	$result .= "\t\$1 = &temp_$u->{NAME};\n";
+	$result .= "}\n\n";
+
+	$result .= "%typemap(argout) (uint32 level, uint32 count, union $u->{NAME} **EMPTY) {\n";
+	$result .= "\tTALLOC_CTX *mem_ctx = talloc_init(\"unmarshall_$u->{NAME}_array\");\n";
+	$result .= "\tint i;\n\n";
+	$result .= "\t\$result = PyList_New(\$2);\n\n";
+	$result .= "\tfor (i = 0; i < \$2; i++) {\n";
+	$result .= "\t\tPyList_SetItem(\$result, i, $u->{NAME}_ptr_to_python(mem_ctx, &(*\$3)[i], \$1));\n";
+	$result .= "\t}\n\n";
+	$result .= "\ttalloc_free(mem_ctx);\n";
+	$result .= "}\n\n";
+
+	$result .= "NTSTATUS unmarshall_$u->{NAME}_array(DATA_BLOB *blob, TALLOC_CTX *mem_ctx, uint32 level, uint32 count, union $u->{NAME} **EMPTY);\n\n";
+    }
+
     return $result;
 }
 

Modified: branches/SAMBA_4_0/source/scripting/swig/torture/spoolss.py
===================================================================
--- branches/SAMBA_4_0/source/scripting/swig/torture/spoolss.py	2004-10-25 07:24:46 UTC (rev 3203)
+++ branches/SAMBA_4_0/source/scripting/swig/torture/spoolss.py	2004-10-25 07:31:59 UTC (rev 3204)
@@ -17,7 +17,7 @@
 
         result = dcerpc.spoolss_EnumPrinters(pipe, r)
 
-    print dcerpc.unmarshall_spoolss_PrinterInfo1(result['buffer'])
+    print dcerpc.unmarshall_spoolss_PrinterInfo_array(result['buffer'], r['level'], result['count'])
 
 def runtests(binding, domain, username, password):
     



More information about the samba-cvs mailing list