From 605560149e544aaa889e352232c26450d7a0c84f Mon Sep 17 00:00:00 2001 From: Tim Potter Date: Sun, 17 Oct 2004 13:24:56 +0000 Subject: r3022: Work on unmarshalling arrays of structs in a buffer (Python string) for spoolss. Doesn't quite work yet. (This used to be commit 9045645ddf4eb8b5596d220ed936c8731641ef0a) --- source4/build/pidl/swig.pm | 38 ++++++++++++++++++++++++++++++++------ 1 file changed, 32 insertions(+), 6 deletions(-) diff --git a/source4/build/pidl/swig.pm b/source4/build/pidl/swig.pm index df63548d41..31a726e32d 100644 --- a/source4/build/pidl/swig.pm +++ b/source4/build/pidl/swig.pm @@ -460,11 +460,13 @@ sub ParseStruct($) $result .= "\n"; $result .= "\treturn obj;\n"; - $result .= "}\n"; + $result .= "}\n\n"; - # Generate function to convert DATA_BLOB to Python dict + # 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"; @@ -473,22 +475,46 @@ sub ParseStruct($) $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\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 .= "%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} * {\n"; - $result .= "\tTALLOC_CTX *mem_ctx = talloc_init(\"typemap(argout) $s->{NAME}\");\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; -- cgit