1 ###################################################
2 # Samba4 parser generator for swig wrappers
3 # Copyright tpot@samba.org 2004
4 # released under the GNU GPL
10 # Some build farm machines don't have the Data::Dumper module installed
12 eval("use Data::Dumper");
14 my(%interfaces, %functions, %structs, %unions);
20 return $unions{$name};
23 # Generate code to convert a Python object to an array
25 sub ArrayFromPython($$)
33 if (util::has_property($e, "size_is")) {
34 $size = util::has_property($e, "size_is");
36 $size = $e->{ARRAY_LEN};
39 if (util::has_property($e, "length_is")) {
40 $size = util::has_property($e, "length_is");
43 if (!util::is_constant($size)) {
44 $size = "s->$prefix$size";
47 my($type) = $e->{TYPE};
49 if (!util::is_scalar_type($type)) {
50 $type = "struct $type";
53 if (!util::is_constant($e->{ARRAY_LEN})) {
54 $result .= "\ts->$prefix$e->{NAME} = talloc(mem_ctx, $size * sizeof($type));\n";
57 $result .= "\tif (!PyDict_GetItemString(obj, \"$e->{NAME}\")) {\n";
58 $result .= "\t\tPyErr_Format(PyExc_ValueError, \"Expecting key '$e->{NAME}'\");\n";
59 $result .= "\t\treturn NULL;\n";
62 $result .= "\tif (!PyList_Check(PyDict_GetItemString(obj, \"$e->{NAME}\"))) {\n";
63 $result .= "\t\tPyErr_Format(PyExc_TypeError, \"Expecting list value for key '$e->{NAME}'\");\n";
64 $result .= "\t\treturn NULL;\n";
69 $result .= "\t\tint i;\n\n";
70 $result .= "\t\tfor (i = 0; i < $size; i++) {\n";
71 if (util::is_scalar_type($e->{TYPE})) {
72 $result .= "\t\t\ts->$prefix$e->{NAME}\[i\] = $e->{TYPE}_from_python(PyList_GetItem(PyDict_GetItemString(obj, \"$e->{NAME}\"), i), \"$e->{NAME}\");\n";
74 $result .= "\t\t\t$e->{TYPE}_from_python(mem_ctx, &s->$prefix$e->{NAME}\[i\], PyList_GetItem(PyDict_GetItemString(obj, \"$e->{NAME}\"), i), \"$e->{NAME}\");\n";
83 # Generate code to convert a Python object to a structure field
85 sub FieldFromPython($$)
90 my($obj) = "PyDict_GetItemString(obj, \"$e->{NAME}\")";
94 if ($e->{TYPE} eq "string") {
95 $result .= "\ts->$prefix$e->{NAME} = string_ptr_from_python(mem_ctx, $obj, \"$e->{NAME}\");\n";
99 if ($e->{TYPE} eq "DATA_BLOB") {
100 $result .= "\tDATA_BLOB_ptr_from_python(mem_ctx, &s->$prefix$e->{NAME}, $obj, \"$e->{NAME}\");\n";
104 # Generate conversion for element
106 if (util::is_scalar_type($e->{TYPE})) {
108 if ($e->{POINTERS} == 0) {
109 if ($e->{ARRAY_LEN}) {
110 $result .= ArrayFromPython($e, $prefix);
112 if (util::has_property($e, "value")) {
113 $result .= "\ts->$prefix$e->{NAME} = 0;\n";
115 $result .= "\ts->$prefix$e->{NAME} = $e->{TYPE}_from_python($obj, \"$e->{NAME}\");\n";
119 $result .= "\ts->$prefix$e->{NAME} = talloc(mem_ctx, sizeof($e->{TYPE}));\n";
120 $result .= "\t*s->$prefix$e->{NAME} = $e->{TYPE}_from_python($obj, \"$e->{NAME}\");\n";
123 if ($e->{POINTERS} == 0) {
124 if ($e->{ARRAY_LEN}) {
125 $result .= ArrayFromPython($e, $prefix);
127 $result .= "\t$e->{TYPE}_from_python(mem_ctx, &s->$prefix$e->{NAME}, $obj, \"$e->{NAME}\");\n";
130 if ($e->{ARRAY_LEN} or util::has_property($e, "size_is")) {
131 $result .= ArrayFromPython($e, $prefix);
133 $result .= "\ts->$prefix$e->{NAME} = $e->{TYPE}_ptr_from_python(mem_ctx, $obj, \"$e->{NAME}\");\n";
141 # Generate code to convert an array to a Python object
143 sub ArrayToPython($$)
149 my($array_len) = util::array_size($e);
151 if ($array_len eq "*" or util::has_property($e, "size_is")) {
152 $array_len = util::has_property($e, "size_is");
155 if (!util::is_constant($array_len)) {
156 $array_len = "s->$prefix$array_len";
159 my($type) = $e->{TYPE};
161 if (!util::is_scalar_type($type)) {
162 $type = "struct $type";
165 $result .= "\n\t{\n";
166 $result .= "\t\tPyObject *temp;\n";
167 $result .= "\t\tint i;\n\n";
169 $result .= "\t\ttemp = PyList_New($array_len);\n\n";
170 $result .= "\t\tfor (i = 0; i < $array_len; i++) {\n";
171 if (util::is_scalar_type($e->{TYPE})) {
172 $result .= "\t\t\tPyList_SetItem(temp, i, $e->{TYPE}_to_python(s->$prefix$e->{NAME}\[i\]));\n";
174 $result .= "\t\t\tPyList_SetItem(temp, i, $e->{TYPE}_ptr_to_python(mem_ctx, &s->$prefix$e->{NAME}\[i\]));\n";
176 $result .= "\t\t}\n";
178 $result .= "\t\tPyDict_SetItemString(obj, \"$e->{NAME}\", temp);\n";
185 # Generate code to convert a structure field to a Python object
187 sub FieldToPython($$)
195 if ($e->{TYPE} eq "string") {
196 $result .= "\tPyDict_SetItemString(obj, \"$e->{NAME}\", string_ptr_to_python(mem_ctx, s->$prefix$e->{NAME}));\n";
200 # Generate conversion for scalars and structures
202 if (util::is_scalar_type($e->{TYPE})) {
203 if ($e->{POINTERS} == 0) {
204 if ($e->{ARRAY_LEN}) {
205 $result .= ArrayToPython($e, $prefix);
207 $result .= "\tPyDict_SetItemString(obj, \"$e->{NAME}\", $e->{TYPE}_to_python(s->$prefix$e->{NAME}));\n";
210 if ($e->{ARRAY_LEN} or util::has_property($e, "size_is")) {
211 $result .= ArrayToPython($e, $prefix);
213 $result .= "\tif (s->$prefix$e->{NAME})\n";
214 $result .= "\t\tPyDict_SetItemString(obj, \"$e->{NAME}\", $e->{TYPE}_to_python(*s->$prefix$e->{NAME}));\n";
215 $result .= "\telse\n";
216 $result .= "\t\tPyDict_SetItemString(obj, \"$e->{NAME}\", Py_None);\n";
221 my($extra_args) = "";
223 if (isunion($e->{TYPE})) {
224 $extra_args = ", $e->{NAME}_switch_is";
227 if ($e->{POINTERS} == 0) {
228 if ($e->{ARRAY_LEN}) {
229 $result .= ArrayToPython($e, $prefix);
231 $result .= "\tPyDict_SetItemString(obj, \"$e->{NAME}\", $e->{TYPE}_ptr_to_python(mem_ctx, &s->$prefix$e->{NAME}$extra_args));\n";
234 if ($e->{ARRAY_LEN} or util::has_property($e, "size_is")) {
235 $result .= ArrayToPython($e, $prefix);
237 $result .= "\tPyDict_SetItemString(obj, \"$e->{NAME}\", $e->{TYPE}_ptr_to_python(mem_ctx, s->$prefix$e->{NAME}$extra_args));\n";
253 $result .= IdlDump::DumpFunction($fn);
256 # Generate function to convert Python dict to structure pointer
258 $result .= "/* Convert Python dict to struct $fn->{NAME}.in */\n\n";
260 $result .= "struct $fn->{NAME} *$fn->{NAME}_ptr_from_python(TALLOC_CTX *mem_ctx, PyObject *obj, char *name)\n";
263 $result .= "\tstruct $fn->{NAME} *s;\n\n";
265 $result .= "\tif (!PyDict_Check(obj)) {\n";
266 $result .= "\t\tPyErr_Format(PyExc_TypeError, \"Expecting dict value for key '%s'\", name);\n";
267 $result .= "\t\t\treturn NULL;\n";
268 $result .= "\t}\n\n";
270 $result .= "\ts = talloc(mem_ctx, sizeof(struct $fn->{NAME}));\n\n";
272 # Remove this when all elements are initialised
273 $result .= "\tmemset(s, 0, sizeof(struct $fn->{NAME}));\n\n";
275 foreach my $e (@{$fn->{DATA}}) {
276 if (util::has_property($e, "in")) {
277 if (util::has_property($e, "ref")) {
278 $result .= "\tif (PyDict_GetItemString(obj, \"$e->{NAME}\") == Py_None) {\n";
279 $result .= "\t\tPyErr_Format(PyExc_ValueError, \"Key '$e->{NAME}' cannot be None\");\n";
280 $result .= "\t\treturn NULL;\n";
283 $result .= FieldFromPython($e, "in.") ;
288 $result .= "\treturn s;\n";
291 # Generate function to convert structure pointer to Python dict
293 $result .= "/* Convert struct $fn->{NAME}.out to Python dict */\n\n";
295 $result .= "PyObject *$fn->{NAME}_ptr_to_python(TALLOC_CTX *mem_ctx, struct $fn->{NAME} *s";
297 foreach my $e (@{$fn->{DATA}}) {
298 if (isunion($e->{TYPE})) {
299 $result .= ", int $e->{NAME}_switch_is";
306 $result .= "\tPyObject *obj = PyDict_New();\n\n";
308 foreach my $e (@{$fn->{DATA}}) {
309 $result .= FieldToPython($e, "out.") if util::has_property($e, "out")
313 $result .= "\treturn obj;\n";
320 $result .= "%typemap(in) struct $fn->{NAME} * {\n";
321 $result .= "\tTALLOC_CTX *mem_ctx = talloc_init(\"typemap(int) $fn->{NAME}\");\n\n";
323 $result .= "\t\$1 = $fn->{NAME}_ptr_from_python(mem_ctx, \$input, \"<function params>\");\n\n";
325 $result .= "\tif (PyErr_Occurred()) return NULL;\n\n";
331 $result .= "%typemap(argout) struct $fn->{NAME} * (PyObject *temp) {\n";
332 $result .= "\tTALLOC_CTX *mem_ctx = talloc_init(\"typemap(argout) $fn->{NAME}\");\n\n";
334 $result .= "\ttemp = $fn->{NAME}_ptr_to_python(mem_ctx, \$1";
336 foreach my $e (@{$fn->{DATA}}) {
337 if ((my $switch_is = util::has_property($e, "switch_is"))) {
338 $result .= ", \$1->in.$switch_is";
344 if ($fn->{RETURN_TYPE} eq "NTSTATUS") {
345 $result .= "\tPyDict_SetItemString(temp, \"result\", resultobj);\n";
347 $result .= "\tPyDict_SetItemString(temp, \"result\", PyLong_FromLong(W_ERROR_V(arg3->out.result)));\n";
352 $result .= "\tresultobj = temp;\n\n";
356 $result .= "\tif (NT_STATUS_IS_ERR(result)) {\n";
357 $result .= "\t\tset_ntstatus_exception(NT_STATUS_V(result));\n";
358 $result .= "\t\tgoto fail;\n";
361 if (!($fn->{RETURN_TYPE} eq "NTSTATUS")) {
362 $result .= "\tif (!W_ERROR_IS_OK(arg3->out.result) && \n";
363 $result .= "\t\t!(W_ERROR_EQUAL(arg3->out.result, WERR_INSUFFICIENT_BUFFER))) {\n";
364 $result .= "\t\tset_werror_exception(W_ERROR_V(arg3->out.result));\n";
365 $result .= "\t\tgoto fail;\n";
371 # Function definitions
373 $result .= "%rename($fn->{NAME}) dcerpc_$fn->{NAME};\n";
374 $result .= "NTSTATUS dcerpc_$fn->{NAME}(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct $fn->{NAME} *r);\n\n";
387 $result .= IdlDump::DumpTypedef($s);
390 # Generate function to convert Python dict to structure pointer
392 $result .= "/* Convert Python dict to struct $s->{NAME} pointer */\n\n";
394 $result .= "struct $s->{NAME} *$s->{NAME}_ptr_from_python(TALLOC_CTX *mem_ctx, PyObject *obj, char *name)\n";
396 $result .= "\tstruct $s->{NAME} *s;\n\n";
398 $result .= "\tif (obj == NULL) {\n";
399 $result .= "\t\tPyErr_Format(PyExc_ValueError, \"Expecting key '%s'\", name);\n";
400 $result .= "\t\treturn NULL;\n";
401 $result .= "\t}\n\n";
403 $result .= "\tif (obj == Py_None) return NULL;\n\n";
405 $result .= "\tif (!PyDict_Check(obj)) {\n";
406 $result .= "\t\tPyErr_Format(PyExc_TypeError, \"Expecting dict value for key '%s'\", name);\n";
407 $result .= "\t\treturn NULL;\n";
408 $result .= "\t}\n\n";
410 $result .= "\ts = talloc(mem_ctx, sizeof(struct $s->{NAME}));\n\n";
412 foreach my $e (@{$s->{DATA}{ELEMENTS}}) {
413 $result .= FieldFromPython($e, "");
417 $result .= "\treturn s;\n";
420 # Generate function to convert Python dict to structure
422 $result .= "/* Convert Python dict to struct $s->{NAME} */\n\n";
424 $result .= "void $s->{NAME}_from_python(TALLOC_CTX *mem_ctx, struct $s->{NAME} *s, PyObject *obj, char *name)\n";
427 $result .= "\tif (obj == NULL) {\n";
428 $result .= "\t\tPyErr_Format(PyExc_ValueError, \"Expecting key '%s'\", name);\n";
429 $result .= "\t\treturn;\n";
430 $result .= "\t}\n\n";
432 $result .= "\tif (!PyDict_Check(obj)) {\n";
433 $result .= "\t\tPyErr_Format(PyExc_TypeError, \"Expecting dict value for key '%s'\", name);\n";
434 $result .= "\t\treturn;\n";
435 $result .= "\t}\n\n";
437 foreach my $e (@{$s->{DATA}{ELEMENTS}}) {
438 $result .= FieldFromPython($e, "");
443 # Generate function to convert structure pointer to Python dict
445 $result .= "/* Convert struct $s->{NAME} pointer to Python dict */\n\n";
447 $result .= "PyObject *$s->{NAME}_ptr_to_python(TALLOC_CTX *mem_ctx, struct $s->{NAME} *s)\n";
450 $result .= "\tPyObject *obj;\n\n";
452 $result .= "\tif (s == NULL) {\n";
453 $result .= "\t\tPy_INCREF(Py_None);\n";
454 $result .= "\t\treturn Py_None;\n";
455 $result .= "\t}\n\n";
457 $result .= "\tobj = PyDict_New();\n\n";
459 foreach my $e (@{$s->{DATA}{ELEMENTS}}) {
460 $result .= FieldToPython($e, "");
464 $result .= "\treturn obj;\n";
467 # Generate function to convert DATA_BLOB to Python dict
469 if (util::has_property($s, "public")) {
470 $result .= "/* Convert DATA_BLOB to python dict of struct $s->{NAME} */\n\n";
472 $result .= "NTSTATUS DATA_BLOB_to_python_$s->{NAME}(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, PyObject **obj)\n";
474 $result .= "\tstruct $s->{NAME} s;\n";
475 $result .= "\tNTSTATUS result = ndr_pull_struct_blob(blob, mem_ctx, &s, ndr_pull_$s->{NAME});\n\n";
476 $result .= "\treturn NT_STATUS_OK;\n";
480 $result .= "\n%}\n\n";
491 $result .= IdlDump::DumpTypedef($u);
494 # Generate function to convert Python dict to union pointer
497 $result .= "/* Convert Python dict to union $u->{NAME} pointer */\n\n";
499 $result .= "union $u->{NAME} *$u->{NAME}_ptr_from_python(TALLOC_CTX *mem_ctx, PyObject *obj, char *name)\n";
502 $result .= "\tunion $u->{NAME} *u;\n";
503 $result .= "\tPyObject *dict;\n\n";
505 $result .= "\tif (obj == NULL) {\n";
506 $result .= "\t\tPyErr_Format(PyExc_ValueError, \"Expecting key '%s'\", name);\n";
507 $result .= "\t\treturn NULL;\n";
508 $result .= "\t}\n\n";
510 $result .= "\tif (!PyDict_Check(obj)) {\n";
511 $result .= "\t\tPyErr_Format(PyExc_TypeError, \"Expecting dict value for key '%s'\", name);\n";
512 $result .= "\t\treturn NULL;\n";
513 $result .= "\t}\n\n";
515 $result .= "\tu = talloc(mem_ctx, sizeof(union $u->{NAME}));\n\n";
517 for my $e (@{$u->{DATA}{DATA}}) {
518 if (defined $e->{DATA}{NAME}) {
519 $result .= "\tif ((dict = PyDict_GetItemString(obj, \"$e->{DATA}{NAME}\"))) {\n";
520 if ($e->{DATA}{POINTERS} == 0) {
521 $result .= "\t\t$e->{DATA}{TYPE}_from_python(mem_ctx, &u->$e->{DATA}{NAME}, dict, \"$e->{DATA}{NAME}\");\n";
522 } elsif ($e->{DATA}{POINTERS} == 1) {
523 $result .= "\t\tu->$e->{DATA}{NAME} = $e->{DATA}{TYPE}_ptr_from_python(mem_ctx, dict, \"$e->{DATA}{NAME}\");\n";
525 $result .= "\t\t// $e->{DATA}{TYPE} pointers=$e->{DATA}{POINTERS}\n";
529 $result .= "\t\treturn u;\n";
530 $result .= "\t}\n\n";
533 $result .= "\treturn NULL;\n";
536 # Generate function to convert Python dict to union
538 $result .= "/* Convert Python dict to union $u->{NAME} */\n\n";
540 $result .= "void $u->{NAME}_from_python(TALLOC_CTX *mem_ctx, union $u->{NAME} *u, PyObject *obj, char *name)\n";
542 $result .= "\tPyObject *dict;\n\n";
544 $result .= "\tif (obj == NULL) {\n";
545 $result .= "\t\tPyErr_Format(PyExc_ValueError, \"Expecting key '%s'\", name);\n";
546 $result .= "\t\treturn;\n";
547 $result .= "\t}\n\n";
549 $result .= "\tif (!PyDict_Check(obj)) {\n";
550 $result .= "\t\tPyErr_Format(PyExc_TypeError, \"Expecting dict value for key '%s'\", name);\n";
551 $result .= "\t\treturn;\n";
552 $result .= "\t}\n\n";
554 for my $e (@{$u->{DATA}{DATA}}) {
555 if (defined $e->{DATA}{NAME}) {
556 $result .= "\tif ((dict = PyDict_GetItemString(obj, \"$e->{DATA}{NAME}\"))) {\n";
557 if ($e->{DATA}{POINTERS} == 0) {
558 $result .= "\t\t$e->{DATA}{TYPE}_from_python(mem_ctx, &u->$e->{DATA}{NAME}, dict, \"$e->{DATA}{NAME}\");\n";
559 } elsif ($e->{DATA}{POINTERS} == 1) {
560 $result .= "\t\tu->$e->{DATA}{NAME} = $e->{DATA}{TYPE}_ptr_from_python(mem_ctx, dict, \"$e->{DATA}{NAME}\");\n";
562 $result .= "\t\t// $e->{DATA}{TYPE} pointers=$e->{DATA}{POINTERS}\n";
565 $result .= "\t\treturn;\n";
566 $result .= "\t}\n\n";
570 # Generate function to convert union pointer to Python dict
572 $result .= "/* Convert union $u->{NAME} pointer to Python dict */\n\n";
574 $result .= "PyObject *$u->{NAME}_ptr_to_python(TALLOC_CTX *mem_ctx, union $u->{NAME} *u, int switch_is)\n";
576 $result .= "\tPyObject *obj;\n\n";
578 $result .= "\tif (u == NULL) {\n";
579 $result .= "\t\tPy_INCREF(Py_None);\n";
580 $result .= "\t\treturn Py_None;\n";
581 $result .= "\t}\n\n";
583 $result .= "\tobj = PyDict_New();\n\n";
585 for my $e (@{$u->{DATA}{DATA}}) {
586 $result .= "\tif (switch_is == $e->{CASE}) {\n";
587 my $prefix = util::c_pull_prefix($e);
588 if (defined $e->{DATA}{NAME}) {
589 $result .= "\t\tPyDict_SetItemString(obj, \"$e->{DATA}{NAME}\", $e->{DATA}{TYPE}_ptr_to_python(mem_ctx, ${prefix}u->$e->{DATA}{NAME}));\n";
594 $result .= "\treturn obj;\n";
598 $result .= "\n%}\n\n";
609 $result .= ParseStruct($e) if $e->{DATA}{TYPE} eq "STRUCT";
610 $result .= ParseUnion($e) if $e->{DATA}{TYPE} eq "UNION";
616 sub ParseInheritedData($)
621 foreach my $e (@{$data}) {
622 $result .= ParseFunction($e) if $e->{TYPE} eq "FUNCTION";
623 $result .= ParseTypedef($e) if $e->{TYPE} eq "TYPEDEF";
634 if ($hdr->{PROPERTIES}{uuid}) {
635 my($name) = $hdr->{NAME};
636 $result .= "#define DCERPC_" . uc($name) . "_UUID \"$hdr->{PROPERTIES}->{uuid}\"\n";
637 $result .= "const int DCERPC_" . uc($name) . "_VERSION = " . $hdr->{PROPERTIES}->{version} . ";\n";
638 $result .= "#define DCERPC_" . uc($name) . "_NAME \"" . $name . "\"\n";
642 $result .= ParseInheritedData($hdr->{INHERITED_DATA});
652 # Make index of functions, structs and unions. Currently unused.
654 foreach my $x (@{$idl}) {
655 my($iname) = $x->{NAME};
656 $interfaces{$iname} = $x->{PROPERTIES};
657 foreach my $i (@{$x->{INHERITED_DATA}}) {
658 $functions{$i->{NAME}} = $i if $i->{TYPE} eq "FUNCTION";
659 if ($i->{TYPE} eq "TYPEDEF") {
660 $structs{$i->{NAME}} = $i->{DATA} if $i->{DATA}{TYPE} eq "STRUCT";
661 $unions{$i->{NAME}} = $i->{DATA} if $i->{DATA}{TYPE} eq "UNION";
668 $result .= "/* Auto-generated by pidl. Tastes like -*- C -*-. */\n\n";
670 foreach my $x (@{$idl}) {
671 $result .= ParseHeader($x) if ($x->{TYPE} eq "INTERFACE");