1 ###################################################
2 # Samba4 NDR parser generator for IDL structures
3 # Copyright tridge@samba.org 2000-2003
4 # Copyright tpot@samba.org 2001
5 # Copyright jelmer@samba.org 2004-2005
6 # released under the GNU GPL
8 package Parse::Pidl::Samba4::NDR::Parser;
11 use Parse::Pidl::Typelist qw(hasType getType mapType);
12 use Parse::Pidl::Util qw(has_property ParseExpr print_uuid);
13 use Parse::Pidl::NDR qw(GetPrevLevel GetNextLevel ContainsDeferred);
15 use vars qw($VERSION);
24 return $typefamily{$n};
29 my ($e, $var_name) = @_;
32 foreach my $l (@{$e->{LEVELS}}) {
33 if ($l->{TYPE} eq "POINTER") {
35 } elsif ($l->{TYPE} eq "ARRAY") {
36 if (($pointers == 0) and
37 (not $l->{IS_FIXED}) and
38 (not $l->{IS_INLINE})) {
39 return get_value_of($var_name);
41 } elsif ($l->{TYPE} eq "DATA") {
42 if (Parse::Pidl::Typelist::scalar_is_reference($l->{DATA_TYPE})) {
43 return get_value_of($var_name) unless ($pointers);
51 sub has_fast_array($$)
55 return 0 if ($l->{TYPE} ne "ARRAY");
57 my $nl = GetNextLevel($e,$l);
58 return 0 unless ($nl->{TYPE} eq "DATA");
59 return 0 unless (hasType($nl->{DATA_TYPE}));
61 my $t = getType($nl->{DATA_TYPE});
63 # Only uint8 and string have fast array functions at the moment
64 return ($t->{NAME} eq "uint8") or ($t->{NAME} eq "string");
67 sub is_charset_array($$)
71 return 0 if ($l->{TYPE} ne "ARRAY");
73 my $nl = GetNextLevel($e,$l);
75 return 0 unless ($nl->{TYPE} eq "DATA");
77 return has_property($e, "charset");
84 if ($var_name =~ /^\*(.*)$/) {
86 } elsif ($var_name =~ /^\&(.*)$/) {
87 return "&($var_name)";
97 if ($var_name =~ /^\&(.*)$/) {
108 ####################################
109 # pidl() is our basic output routine
122 sub pidl_hdr ($) { my $d = shift; $res_hdr .= "$d\n"; }
124 ####################################
125 # defer() is like pidl(), but adds to
126 # a deferred buffer which is then added to the
127 # output buffer at the end of the structure/union/function
128 # This is needed to cope with code that must be pushed back
129 # to the end of a block of elements
140 ########################################
141 # add the deferred content to the current
156 $tabs = substr($tabs, 0, -1);
159 #####################################################################
160 # check that a variable we get from ParseExpr isn't a null pointer
161 sub check_null_pointer($)
164 if ($size =~ /^\*/) {
165 my $size2 = substr($size, 1);
166 pidl "if ($size2 == NULL) return NT_STATUS_INVALID_PARAMETER_MIX;";
170 #####################################################################
171 # check that a variable we get from ParseExpr isn't a null pointer,
172 # putting the check at the end of the structure/function
173 sub check_null_pointer_deferred($)
176 if ($size =~ /^\*/) {
177 my $size2 = substr($size, 1);
178 defer "if ($size2 == NULL) return NT_STATUS_INVALID_PARAMETER_MIX;";
182 #####################################################################
183 # check that a variable we get from ParseExpr isn't a null pointer
184 # void return varient
185 sub check_null_pointer_void($)
188 if ($size =~ /^\*/) {
189 my $size2 = substr($size, 1);
190 pidl "if ($size2 == NULL) return;";
194 #####################################################################
195 # declare a function public or static, depending on its attributes
200 if (has_property($fn, "public")) {
202 pidl "_PUBLIC_ $decl";
208 ###################################################################
209 # setup any special flags for an element or structure
213 my $flags = has_property($e, "flag");
214 if (defined $flags) {
217 pidl "uint32_t _flags_save_$e->{TYPE} = ndr->flags;";
218 pidl "ndr_set_flags(&ndr->flags, $flags);";
222 ###################################################################
223 # end any special flags for an element or structure
227 my $flags = has_property($e, "flag");
228 if (defined $flags) {
229 pidl "ndr->flags = _flags_save_$e->{TYPE};";
235 sub GenerateStructEnv($)
240 foreach my $e (@{$x->{ELEMENTS}}) {
241 $env{$e->{NAME}} = "r->$e->{NAME}";
249 sub EnvSubstituteValue($$)
253 # Substitute the value() values in the env
254 foreach my $e (@{$s->{ELEMENTS}}) {
255 next unless (my $v = has_property($e, "value"));
257 $env->{$e->{NAME}} = ParseExpr($v, $env);
263 sub GenerateFunctionInEnv($)
268 foreach my $e (@{$fn->{ELEMENTS}}) {
269 if (grep (/in/, @{$e->{DIRECTION}})) {
270 $env{$e->{NAME}} = "r->in.$e->{NAME}";
277 sub GenerateFunctionOutEnv($)
282 foreach my $e (@{$fn->{ELEMENTS}}) {
283 if (grep (/out/, @{$e->{DIRECTION}})) {
284 $env{$e->{NAME}} = "r->out.$e->{NAME}";
285 } elsif (grep (/in/, @{$e->{DIRECTION}})) {
286 $env{$e->{NAME}} = "r->in.$e->{NAME}";
293 #####################################################################
294 # parse the data of an array - push side
295 sub ParseArrayPushHeader($$$$$)
297 my ($e,$l,$ndr,$var_name,$env) = @_;
302 if ($l->{IS_ZERO_TERMINATED}) {
303 if (has_property($e, "charset")) {
304 $size = $length = "ndr_charset_length($var_name, CH_$e->{PROPERTIES}->{charset})";
306 $size = $length = "ndr_string_length($var_name, sizeof(*$var_name))";
309 $size = ParseExpr($l->{SIZE_IS}, $env);
310 $length = ParseExpr($l->{LENGTH_IS}, $env);
313 if ((!$l->{IS_SURROUNDING}) and $l->{IS_CONFORMANT}) {
314 pidl "NDR_CHECK(ndr_push_uint32($ndr, NDR_SCALARS, $size));";
317 if ($l->{IS_VARYING}) {
318 pidl "NDR_CHECK(ndr_push_uint32($ndr, NDR_SCALARS, 0));"; # array offset
319 pidl "NDR_CHECK(ndr_push_uint32($ndr, NDR_SCALARS, $length));";
325 #####################################################################
326 # parse an array - pull side
327 sub ParseArrayPullHeader($$$$$)
329 my ($e,$l,$ndr,$var_name,$env) = @_;
334 if ($l->{IS_CONFORMANT}) {
335 $length = $size = "ndr_get_array_size($ndr, " . get_pointer_to($var_name) . ")";
336 } elsif ($l->{IS_ZERO_TERMINATED}) { # Noheader arrays
337 $length = $size = "ndr_get_string_size($ndr, sizeof(*$var_name))";
339 $length = $size = ParseExpr($l->{SIZE_IS}, $env);
342 if ((!$l->{IS_SURROUNDING}) and $l->{IS_CONFORMANT}) {
343 pidl "NDR_CHECK(ndr_pull_array_size(ndr, " . get_pointer_to($var_name) . "));";
347 if ($l->{IS_VARYING}) {
348 pidl "NDR_CHECK(ndr_pull_array_length($ndr, " . get_pointer_to($var_name) . "));";
349 $length = "ndr_get_array_length($ndr, " . get_pointer_to($var_name) .")";
352 check_null_pointer($length);
354 if ($length ne $size) {
355 pidl "if ($length > $size) {";
357 pidl "return ndr_pull_error($ndr, NDR_ERR_ARRAY_SIZE, \"Bad array size %u should exceed array length %u\", $size, $length);";
362 if ($l->{IS_CONFORMANT} and not $l->{IS_ZERO_TERMINATED}) {
363 my $size = ParseExpr($l->{SIZE_IS}, $env);
364 defer "if ($var_name) {";
365 check_null_pointer_deferred($size);
366 defer "NDR_CHECK(ndr_check_array_size(ndr, (void*)" . get_pointer_to($var_name) . ", $size));";
370 if ($l->{IS_VARYING} and not $l->{IS_ZERO_TERMINATED}) {
371 my $length = ParseExpr($l->{LENGTH_IS}, $env);
372 defer "if ($var_name) {";
373 check_null_pointer_deferred($length);
374 defer "NDR_CHECK(ndr_check_array_length(ndr, (void*)" . get_pointer_to($var_name) . ", $length));";
378 if (not $l->{IS_FIXED} and not is_charset_array($e, $l)) {
379 AllocateArrayLevel($e,$l,$ndr,$env,$size);
385 sub compression_alg($$)
388 my $compression = $l->{COMPRESSION};
389 my ($alg, $clen, $dlen) = split(/ /, $compression);
394 sub compression_clen($$$)
396 my ($e,$l,$env) = @_;
397 my $compression = $l->{COMPRESSION};
398 my ($alg, $clen, $dlen) = split(/ /, $compression);
400 return ParseExpr($clen, $env);
403 sub compression_dlen($$$)
405 my ($e,$l,$env) = @_;
406 my $compression = $l->{COMPRESSION};
407 my ($alg, $clen, $dlen) = split(/ /, $compression);
409 return ParseExpr($dlen, $env);
412 sub ParseCompressionPushStart($$$$)
414 my ($e,$l,$ndr,$env) = @_;
415 my $comndr = "$ndr\_compressed";
416 my $alg = compression_alg($e, $l);
417 my $dlen = compression_dlen($e, $l, $env);
421 pidl "struct ndr_push *$comndr;";
422 pidl "NDR_CHECK(ndr_push_compression_start($ndr, &$comndr, $alg, $dlen));";
427 sub ParseCompressionPushEnd($$$$)
429 my ($e,$l,$ndr,$env) = @_;
430 my $comndr = "$ndr\_compressed";
431 my $alg = compression_alg($e, $l);
432 my $dlen = compression_dlen($e, $l, $env);
434 pidl "NDR_CHECK(ndr_push_compression_end($ndr, $comndr, $alg, $dlen));";
439 sub ParseCompressionPullStart($$$$)
441 my ($e,$l,$ndr,$env) = @_;
442 my $comndr = "$ndr\_compressed";
443 my $alg = compression_alg($e, $l);
444 my $dlen = compression_dlen($e, $l, $env);
448 pidl "struct ndr_pull *$comndr;";
449 pidl "NDR_CHECK(ndr_pull_compression_start($ndr, &$comndr, $alg, $dlen));";
454 sub ParseCompressionPullEnd($$$$)
456 my ($e,$l,$ndr,$env) = @_;
457 my $comndr = "$ndr\_compressed";
458 my $alg = compression_alg($e, $l);
459 my $dlen = compression_dlen($e, $l, $env);
461 pidl "NDR_CHECK(ndr_pull_compression_end($ndr, $comndr, $alg, $dlen));";
466 sub ParseObfuscationPushStart($$)
469 my $obfuscation = has_property($e, "obfuscation");
471 pidl "NDR_CHECK(ndr_push_obfuscation_start($ndr, $obfuscation));";
476 sub ParseObfuscationPushEnd($$)
479 my $obfuscation = has_property($e, "obfuscation");
481 pidl "NDR_CHECK(ndr_push_obfuscation_end($ndr, $obfuscation));";
484 sub ParseObfuscationPullStart($$)
487 my $obfuscation = has_property($e, "obfuscation");
489 pidl "NDR_CHECK(ndr_pull_obfuscation_start($ndr, $obfuscation));";
494 sub ParseObfuscationPullEnd($$)
497 my $obfuscation = has_property($e, "obfuscation");
499 pidl "NDR_CHECK(ndr_pull_obfuscation_end($ndr, $obfuscation));";
502 sub ParseSubcontextPushStart($$$$)
504 my ($e,$l,$ndr,$env) = @_;
505 my $subndr = "_ndr_$e->{NAME}";
506 my $subcontext_size = ParseExpr($l->{SUBCONTEXT_SIZE},$env);
510 pidl "struct ndr_push *$subndr;";
511 pidl "NDR_CHECK(ndr_push_subcontext_start($ndr, &$subndr, $l->{HEADER_SIZE}, $subcontext_size));";
513 if (defined $l->{COMPRESSION}) {
514 $subndr = ParseCompressionPushStart($e, $l, $subndr, $env);
517 if (defined $l->{OBFUSCATION}) {
518 $subndr = ParseObfuscationPushStart($e, $subndr);
524 sub ParseSubcontextPushEnd($$$$)
526 my ($e,$l,$ndr,$env) = @_;
527 my $subndr = "_ndr_$e->{NAME}";
528 my $subcontext_size = ParseExpr($l->{SUBCONTEXT_SIZE},$env);
530 if (defined $l->{COMPRESSION}) {
531 ParseCompressionPushEnd($e, $l, $subndr, $env);
534 if (defined $l->{OBFUSCATION}) {
535 ParseObfuscationPushEnd($e, $subndr);
538 pidl "NDR_CHECK(ndr_push_subcontext_end($ndr, $subndr, $l->{HEADER_SIZE}, $subcontext_size));";
543 sub ParseSubcontextPullStart($$$$)
545 my ($e,$l,$ndr,$env) = @_;
546 my $subndr = "_ndr_$e->{NAME}";
547 my $subcontext_size = ParseExpr($l->{SUBCONTEXT_SIZE},$env);
551 pidl "struct ndr_pull *$subndr;";
552 pidl "NDR_CHECK(ndr_pull_subcontext_start($ndr, &$subndr, $l->{HEADER_SIZE}, $subcontext_size));";
554 if (defined $l->{COMPRESSION}) {
555 $subndr = ParseCompressionPullStart($e, $l, $subndr, $env);
558 if (defined $l->{OBFUSCATION}) {
559 $subndr = ParseObfuscationPullStart($e, $subndr);
565 sub ParseSubcontextPullEnd($$$$)
567 my ($e,$l,$ndr,$env) = @_;
568 my $subndr = "_ndr_$e->{NAME}";
569 my $subcontext_size = ParseExpr($l->{SUBCONTEXT_SIZE},$env);
571 if (defined $l->{COMPRESSION}) {
572 ParseCompressionPullEnd($e, $l, $subndr, $env);
575 if (defined $l->{OBFUSCATION}) {
576 ParseObfuscationPullEnd($e, $subndr);
579 pidl "NDR_CHECK(ndr_pull_subcontext_end($ndr, $subndr, $l->{HEADER_SIZE}, $subcontext_size));";
584 sub ParseElementPushLevel
586 my ($e,$l,$ndr,$var_name,$env,$primitives,$deferred) = @_;
588 my $ndr_flags = CalcNdrFlags($l, $primitives, $deferred);
590 if ($l->{TYPE} eq "ARRAY" and ($l->{IS_CONFORMANT} or $l->{IS_VARYING})) {
591 $var_name = get_pointer_to($var_name);
594 if (defined($ndr_flags)) {
595 if ($l->{TYPE} eq "SUBCONTEXT") {
596 my $subndr = ParseSubcontextPushStart($e, $l, $ndr, $env);
597 ParseElementPushLevel($e, GetNextLevel($e, $l), $subndr, $var_name, $env, 1, 1);
598 ParseSubcontextPushEnd($e, $l, $ndr, $env);
599 } elsif ($l->{TYPE} eq "POINTER") {
600 ParsePtrPush($e, $l, $var_name);
601 } elsif ($l->{TYPE} eq "ARRAY") {
602 my $length = ParseArrayPushHeader($e, $l, $ndr, $var_name, $env);
604 my $nl = GetNextLevel($e, $l);
606 # Allow speedups for arrays of scalar types
607 if (is_charset_array($e,$l)) {
608 pidl "NDR_CHECK(ndr_push_charset($ndr, $ndr_flags, $var_name, $length, sizeof(" . mapType($nl->{DATA_TYPE}) . "), CH_$e->{PROPERTIES}->{charset}));";
610 } elsif (has_fast_array($e,$l)) {
611 pidl "NDR_CHECK(ndr_push_array_$nl->{DATA_TYPE}($ndr, $ndr_flags, $var_name, $length));";
614 } elsif ($l->{TYPE} eq "SWITCH") {
615 ParseSwitchPush($e, $l, $ndr, $var_name, $ndr_flags, $env);
616 } elsif ($l->{TYPE} eq "DATA") {
617 ParseDataPush($e, $l, $ndr, $var_name, $ndr_flags);
621 if ($l->{TYPE} eq "POINTER" and $deferred) {
622 if ($l->{POINTER_TYPE} ne "ref") {
623 pidl "if ($var_name) {";
625 if ($l->{POINTER_TYPE} eq "relative") {
626 pidl "NDR_CHECK(ndr_push_relative_ptr2(ndr, $var_name));";
629 $var_name = get_value_of($var_name);
630 ParseElementPushLevel($e, GetNextLevel($e, $l), $ndr, $var_name, $env, 1, 1);
632 if ($l->{POINTER_TYPE} ne "ref") {
636 } elsif ($l->{TYPE} eq "ARRAY" and not has_fast_array($e,$l) and
637 not is_charset_array($e, $l)) {
638 my $length = ParseExpr($l->{LENGTH_IS}, $env);
639 my $counter = "cntr_$e->{NAME}_$l->{LEVEL_INDEX}";
641 $var_name = $var_name . "[$counter]";
643 if (($primitives and not $l->{IS_DEFERRED}) or ($deferred and $l->{IS_DEFERRED})) {
644 pidl "for ($counter = 0; $counter < $length; $counter++) {";
646 ParseElementPushLevel($e, GetNextLevel($e, $l), $ndr, $var_name, $env, 1, 0);
651 if ($deferred and ContainsDeferred($e, $l)) {
652 pidl "for ($counter = 0; $counter < $length; $counter++) {";
654 ParseElementPushLevel($e, GetNextLevel($e, $l), $ndr, $var_name, $env, 0, 1);
658 } elsif ($l->{TYPE} eq "SWITCH") {
659 ParseElementPushLevel($e, GetNextLevel($e, $l), $ndr, $var_name, $env, $primitives, $deferred);
663 #####################################################################
664 # parse scalars in a structure element
665 sub ParseElementPush($$$$$$)
667 my ($e,$ndr,$var_prefix,$env,$primitives,$deferred) = @_;
670 my $var_name = $var_prefix.$e->{NAME};
672 $var_name = append_prefix($e, $var_name);
674 return unless $primitives or ($deferred and ContainsDeferred($e, $e->{LEVELS}[0]));
678 if (my $value = has_property($e, "value")) {
679 $var_name = ParseExpr($value, $env);
682 ParseElementPushLevel($e, $e->{LEVELS}[0], $ndr, $var_name, $env, $primitives, $deferred);
687 #####################################################################
688 # parse a pointer in a struct element or function
689 sub ParsePtrPush($$$)
691 my ($e,$l,$var_name) = @_;
693 if ($l->{POINTER_TYPE} eq "ref") {
694 if ($l->{LEVEL} eq "EMBEDDED") {
695 pidl "NDR_CHECK(ndr_push_ref_ptr(ndr, $var_name));";
697 check_null_pointer(get_value_of($var_name));
699 } elsif ($l->{POINTER_TYPE} eq "relative") {
700 pidl "NDR_CHECK(ndr_push_relative_ptr1(ndr, $var_name));";
701 } elsif ($l->{POINTER_TYPE} eq "unique") {
702 pidl "NDR_CHECK(ndr_push_unique_ptr(ndr, $var_name));";
703 } elsif ($l->{POINTER_TYPE} eq "sptr") {
704 pidl "NDR_CHECK(ndr_push_sptr_ptr(ndr, $var_name));";
706 die("Unhandled pointer type $l->{POINTER_TYPE}");
710 #####################################################################
711 # print scalars in a structure element
712 sub ParseElementPrint($$$)
714 my($e,$var_name,$env) = @_;
716 $var_name = append_prefix($e, $var_name);
717 return if (has_property($e, "noprint"));
719 if (my $value = has_property($e, "value")) {
720 $var_name = "(ndr->flags & LIBNDR_PRINT_SET_VALUES)?" . ParseExpr($value,$env) . ":$var_name";
723 foreach my $l (@{$e->{LEVELS}}) {
724 if ($l->{TYPE} eq "POINTER") {
725 pidl "ndr_print_ptr(ndr, \"$e->{NAME}\", $var_name);";
726 pidl "ndr->depth++;";
727 if ($l->{POINTER_TYPE} ne "ref") {
728 pidl "if ($var_name) {";
731 $var_name = get_value_of($var_name);
732 } elsif ($l->{TYPE} eq "ARRAY") {
735 if ($l->{IS_CONFORMANT} or $l->{IS_VARYING}) {
736 $var_name = get_pointer_to($var_name);
739 if ($l->{IS_ZERO_TERMINATED}) {
740 $length = "ndr_string_length($var_name, sizeof(*$var_name))";
742 $length = ParseExpr($l->{LENGTH_IS}, $env);
745 if (is_charset_array($e,$l)) {
746 pidl "ndr_print_string(ndr, \"$e->{NAME}\", $var_name);";
748 } elsif (has_fast_array($e, $l)) {
749 my $nl = GetNextLevel($e, $l);
750 pidl "ndr_print_array_$nl->{DATA_TYPE}(ndr, \"$e->{NAME}\", $var_name, $length);";
753 my $counter = "cntr_$e->{NAME}_$l->{LEVEL_INDEX}";
755 pidl "ndr->print(ndr, \"\%s: ARRAY(\%d)\", \"$e->{NAME}\", $length);";
756 pidl 'ndr->depth++;';
757 pidl "for ($counter=0;$counter<$length;$counter++) {";
759 pidl "char *idx_$l->{LEVEL_INDEX}=NULL;";
760 pidl "asprintf(&idx_$l->{LEVEL_INDEX}, \"[\%d]\", $counter);";
761 pidl "if (idx_$l->{LEVEL_INDEX}) {";
764 $var_name = $var_name . "[$counter]";
766 } elsif ($l->{TYPE} eq "DATA") {
767 if (not Parse::Pidl::Typelist::is_scalar($l->{DATA_TYPE}) or Parse::Pidl::Typelist::scalar_is_reference($l->{DATA_TYPE})) {
768 $var_name = get_pointer_to($var_name);
770 pidl "ndr_print_$l->{DATA_TYPE}(ndr, \"$e->{NAME}\", $var_name);";
771 } elsif ($l->{TYPE} eq "SWITCH") {
772 my $switch_var = ParseExpr($l->{SWITCH_IS}, $env);
773 check_null_pointer_void($switch_var);
774 pidl "ndr_print_set_switch_value(ndr, " . get_pointer_to($var_name) . ", $switch_var);";
778 foreach my $l (reverse @{$e->{LEVELS}}) {
779 if ($l->{TYPE} eq "POINTER") {
780 if ($l->{POINTER_TYPE} ne "ref") {
784 pidl "ndr->depth--;";
785 } elsif (($l->{TYPE} eq "ARRAY")
786 and not is_charset_array($e,$l)
787 and not has_fast_array($e,$l)) {
788 pidl "free(idx_$l->{LEVEL_INDEX});";
793 pidl "ndr->depth--;";
798 #####################################################################
799 # parse scalars in a structure element - pull size
800 sub ParseSwitchPull($$$$$$)
802 my($e,$l,$ndr,$var_name,$ndr_flags,$env) = @_;
803 my $switch_var = ParseExpr($l->{SWITCH_IS}, $env);
805 check_null_pointer($switch_var);
807 $var_name = get_pointer_to($var_name);
808 pidl "NDR_CHECK(ndr_pull_set_switch_value($ndr, $var_name, $switch_var));";
811 #####################################################################
812 # push switch element
813 sub ParseSwitchPush($$$$$$)
815 my($e,$l,$ndr,$var_name,$ndr_flags,$env) = @_;
816 my $switch_var = ParseExpr($l->{SWITCH_IS}, $env);
818 check_null_pointer($switch_var);
819 $var_name = get_pointer_to($var_name);
820 pidl "NDR_CHECK(ndr_push_set_switch_value($ndr, $var_name, $switch_var));";
823 sub ParseDataPull($$$$$)
825 my ($e,$l,$ndr,$var_name,$ndr_flags) = @_;
827 if (Parse::Pidl::Typelist::scalar_is_reference($l->{DATA_TYPE})) {
828 $var_name = get_pointer_to($var_name);
831 $var_name = get_pointer_to($var_name);
833 pidl "NDR_CHECK(ndr_pull_$l->{DATA_TYPE}($ndr, $ndr_flags, $var_name));";
835 if (my $range = has_property($e, "range")) {
836 $var_name = get_value_of($var_name);
837 my ($low, $high) = split(/ /, $range, 2);
838 pidl "if ($var_name < $low || $var_name > $high) {";
839 pidl "\treturn ndr_pull_error($ndr, NDR_ERR_RANGE, \"value out of range\");";
844 sub ParseDataPush($$$$$)
846 my ($e,$l,$ndr,$var_name,$ndr_flags) = @_;
848 # strings are passed by value rather then reference
849 if (not Parse::Pidl::Typelist::is_scalar($l->{DATA_TYPE}) or Parse::Pidl::Typelist::scalar_is_reference($l->{DATA_TYPE})) {
850 $var_name = get_pointer_to($var_name);
853 pidl "NDR_CHECK(ndr_push_$l->{DATA_TYPE}($ndr, $ndr_flags, $var_name));";
856 sub CalcNdrFlags($$$)
858 my ($l,$primitives,$deferred) = @_;
863 # Add NDR_SCALARS if this one is deferred
864 # and deferreds may be pushed
865 $scalars = 1 if ($l->{IS_DEFERRED} and $deferred);
867 # Add NDR_SCALARS if this one is not deferred and
868 # primitives may be pushed
869 $scalars = 1 if (!$l->{IS_DEFERRED} and $primitives);
871 # Add NDR_BUFFERS if this one contains deferred stuff
872 # and deferreds may be pushed
873 $buffers = 1 if ($l->{CONTAINS_DEFERRED} and $deferred);
875 return "NDR_SCALARS|NDR_BUFFERS" if ($scalars and $buffers);
876 return "NDR_SCALARS" if ($scalars);
877 return "NDR_BUFFERS" if ($buffers);
881 sub ParseMemCtxPullStart($$$)
885 my $ptr_name = shift;
887 my $mem_r_ctx = "_mem_save_$e->{NAME}_$l->{LEVEL_INDEX}";
888 my $mem_c_ctx = $ptr_name;
889 my $mem_c_flags = "0";
891 return if ($l->{TYPE} eq "ARRAY" and $l->{IS_FIXED});
893 if (($l->{TYPE} eq "POINTER") and ($l->{POINTER_TYPE} eq "ref")) {
894 my $nl = GetNextLevel($e, $l);
895 my $next_is_array = ($nl->{TYPE} eq "ARRAY");
896 my $next_is_string = (($nl->{TYPE} eq "DATA") and
897 ($nl->{DATA_TYPE} eq "string"));
898 if ($next_is_array or $next_is_string) {
901 $mem_c_flags = "LIBNDR_FLAG_REF_ALLOC";
905 pidl "$mem_r_ctx = NDR_PULL_GET_MEM_CTX(ndr);";
906 pidl "NDR_PULL_SET_MEM_CTX(ndr, $mem_c_ctx, $mem_c_flags);";
909 sub ParseMemCtxPullEnd($$)
914 my $mem_r_ctx = "_mem_save_$e->{NAME}_$l->{LEVEL_INDEX}";
915 my $mem_r_flags = "0";
917 return if ($l->{TYPE} eq "ARRAY" and $l->{IS_FIXED});
919 if (($l->{TYPE} eq "POINTER") and ($l->{POINTER_TYPE} eq "ref")) {
920 my $nl = GetNextLevel($e, $l);
921 my $next_is_array = ($nl->{TYPE} eq "ARRAY");
922 my $next_is_string = (($nl->{TYPE} eq "DATA") and
923 ($nl->{DATA_TYPE} eq "string"));
924 if ($next_is_array or $next_is_string) {
927 $mem_r_flags = "LIBNDR_FLAG_REF_ALLOC";
931 pidl "NDR_PULL_SET_MEM_CTX(ndr, $mem_r_ctx, $mem_r_flags);";
934 sub CheckStringTerminator($$$$)
936 my ($ndr,$e,$l,$length) = @_;
937 my $nl = GetNextLevel($e, $l);
939 # Make sure last element is zero!
940 pidl "NDR_CHECK(ndr_check_string_terminator($ndr, $length, sizeof($nl->{DATA_TYPE}_t)));";
943 sub ParseElementPullLevel
945 my($e,$l,$ndr,$var_name,$env,$primitives,$deferred) = @_;
947 my $ndr_flags = CalcNdrFlags($l, $primitives, $deferred);
949 if ($l->{TYPE} eq "ARRAY" and ($l->{IS_VARYING} or $l->{IS_CONFORMANT})) {
950 $var_name = get_pointer_to($var_name);
953 # Only pull something if there's actually something to be pulled
954 if (defined($ndr_flags)) {
955 if ($l->{TYPE} eq "SUBCONTEXT") {
956 my $subndr = ParseSubcontextPullStart($e, $l, $ndr, $env);
957 ParseElementPullLevel($e, GetNextLevel($e,$l), $subndr, $var_name, $env, 1, 1);
958 ParseSubcontextPullEnd($e, $l, $ndr, $env);
959 } elsif ($l->{TYPE} eq "ARRAY") {
960 my $length = ParseArrayPullHeader($e, $l, $ndr, $var_name, $env);
962 my $nl = GetNextLevel($e, $l);
964 if (is_charset_array($e,$l)) {
965 if ($l->{IS_ZERO_TERMINATED}) {
966 CheckStringTerminator($ndr, $e, $l, $length);
968 pidl "NDR_CHECK(ndr_pull_charset($ndr, $ndr_flags, ".get_pointer_to($var_name).", $length, sizeof(" . mapType($nl->{DATA_TYPE}) . "), CH_$e->{PROPERTIES}->{charset}));";
970 } elsif (has_fast_array($e, $l)) {
971 if ($l->{IS_ZERO_TERMINATED}) {
972 CheckStringTerminator($ndr,$e,$l,$length);
974 pidl "NDR_CHECK(ndr_pull_array_$nl->{DATA_TYPE}($ndr, $ndr_flags, $var_name, $length));";
977 } elsif ($l->{TYPE} eq "POINTER") {
978 ParsePtrPull($e, $l, $ndr, $var_name);
979 } elsif ($l->{TYPE} eq "SWITCH") {
980 ParseSwitchPull($e, $l, $ndr, $var_name, $ndr_flags, $env);
981 } elsif ($l->{TYPE} eq "DATA") {
982 ParseDataPull($e, $l, $ndr, $var_name, $ndr_flags);
986 # add additional constructions
987 if ($l->{TYPE} eq "POINTER" and $deferred) {
988 if ($l->{POINTER_TYPE} ne "ref") {
989 pidl "if ($var_name) {";
992 if ($l->{POINTER_TYPE} eq "relative") {
993 pidl "struct ndr_pull_save _relative_save;";
994 pidl "ndr_pull_save(ndr, &_relative_save);";
995 pidl "NDR_CHECK(ndr_pull_relative_ptr2(ndr, $var_name));";
999 ParseMemCtxPullStart($e,$l, $var_name);
1001 $var_name = get_value_of($var_name);
1002 ParseElementPullLevel($e,GetNextLevel($e,$l), $ndr, $var_name, $env, 1, 1);
1004 ParseMemCtxPullEnd($e,$l);
1006 if ($l->{POINTER_TYPE} ne "ref") {
1007 if ($l->{POINTER_TYPE} eq "relative") {
1008 pidl "ndr_pull_restore(ndr, &_relative_save);";
1013 } elsif ($l->{TYPE} eq "ARRAY" and
1014 not has_fast_array($e,$l) and not is_charset_array($e, $l)) {
1015 my $length = ParseExpr($l->{LENGTH_IS}, $env);
1016 my $counter = "cntr_$e->{NAME}_$l->{LEVEL_INDEX}";
1017 my $array_name = $var_name;
1019 $var_name = $var_name . "[$counter]";
1021 ParseMemCtxPullStart($e,$l, $array_name);
1023 if (($primitives and not $l->{IS_DEFERRED}) or ($deferred and $l->{IS_DEFERRED})) {
1024 my $nl = GetNextLevel($e,$l);
1026 if ($l->{IS_ZERO_TERMINATED}) {
1027 CheckStringTerminator($ndr,$e,$l,$length);
1030 pidl "for ($counter = 0; $counter < $length; $counter++) {";
1032 ParseElementPullLevel($e, $nl, $ndr, $var_name, $env, 1, 0);
1037 if ($deferred and ContainsDeferred($e, $l)) {
1038 pidl "for ($counter = 0; $counter < $length; $counter++) {";
1040 ParseElementPullLevel($e,GetNextLevel($e,$l), $ndr, $var_name, $env, 0, 1);
1045 ParseMemCtxPullEnd($e,$l);
1047 } elsif ($l->{TYPE} eq "SWITCH") {
1048 ParseElementPullLevel($e,GetNextLevel($e,$l), $ndr, $var_name, $env, $primitives, $deferred);
1052 #####################################################################
1053 # parse scalars in a structure element - pull size
1054 sub ParseElementPull($$$$$$)
1056 my($e,$ndr,$var_prefix,$env,$primitives,$deferred) = @_;
1058 my $var_name = $var_prefix.$e->{NAME};
1060 $var_name = append_prefix($e, $var_name);
1062 return unless $primitives or ($deferred and ContainsDeferred($e, $e->{LEVELS}[0]));
1066 ParseElementPullLevel($e,$e->{LEVELS}[0],$ndr,$var_name,$env,$primitives,$deferred);
1071 #####################################################################
1072 # parse a pointer in a struct element or function
1073 sub ParsePtrPull($$$$)
1075 my($e,$l,$ndr,$var_name) = @_;
1077 my $nl = GetNextLevel($e, $l);
1078 my $next_is_array = ($nl->{TYPE} eq "ARRAY");
1079 my $next_is_string = (($nl->{TYPE} eq "DATA") and
1080 ($nl->{DATA_TYPE} eq "string"));
1082 if ($l->{POINTER_TYPE} eq "ref") {
1083 unless ($l->{LEVEL} eq "TOP") {
1084 pidl "NDR_CHECK(ndr_pull_ref_ptr($ndr, &_ptr_$e->{NAME}));";
1087 unless ($next_is_array or $next_is_string) {
1088 pidl "if (ndr->flags & LIBNDR_FLAG_REF_ALLOC) {";
1089 pidl "\tNDR_PULL_ALLOC($ndr, $var_name);";
1094 } elsif (($l->{POINTER_TYPE} eq "unique") or
1095 ($l->{POINTER_TYPE} eq "relative") or
1096 ($l->{POINTER_TYPE} eq "sptr")) {
1097 pidl "NDR_CHECK(ndr_pull_generic_ptr($ndr, &_ptr_$e->{NAME}));";
1098 pidl "if (_ptr_$e->{NAME}) {";
1101 die("Unhandled pointer type $l->{POINTER_TYPE}");
1104 # Don't do this for arrays, they're allocated at the actual level
1106 unless ($next_is_array or $next_is_string) {
1107 pidl "NDR_PULL_ALLOC($ndr, $var_name);";
1109 # FIXME: Yes, this is nasty.
1110 # We allocate an array twice
1111 # - once just to indicate that it's there,
1112 # - then the real allocation...
1113 pidl "NDR_PULL_ALLOC_SIZE($ndr, $var_name, 1);";
1116 #pidl "memset($var_name, 0, sizeof($var_name));";
1117 if ($l->{POINTER_TYPE} eq "relative") {
1118 pidl "NDR_CHECK(ndr_pull_relative_ptr1($ndr, $var_name, _ptr_$e->{NAME}));";
1122 pidl "\t$var_name = NULL;";
1126 #####################################################################
1128 sub ParseStructPush($$)
1130 my($struct,$name) = @_;
1132 return unless defined($struct->{ELEMENTS});
1134 my $env = GenerateStructEnv($struct);
1136 EnvSubstituteValue($env, $struct);
1138 # save the old relative_base_offset
1139 pidl "uint32_t _save_relative_base_offset = ndr_push_get_relative_base_offset(ndr);" if defined($struct->{PROPERTIES}{relative_base});
1141 foreach my $e (@{$struct->{ELEMENTS}}) {
1142 DeclareArrayVariables($e);
1145 start_flags($struct);
1147 # see if the structure contains a conformant array. If it
1148 # does, then it must be the last element of the structure, and
1149 # we need to push the conformant length early, as it fits on
1150 # the wire before the structure (and even before the structure
1152 if (defined($struct->{SURROUNDING_ELEMENT})) {
1153 my $e = $struct->{SURROUNDING_ELEMENT};
1155 if (defined($e->{LEVELS}[0]) and
1156 $e->{LEVELS}[0]->{TYPE} eq "ARRAY") {
1159 if ($e->{LEVELS}[0]->{IS_ZERO_TERMINATED}) {
1160 if (has_property($e, "charset")) {
1161 $size = "ndr_charset_length(r->$e->{NAME}, CH_$e->{PROPERTIES}->{charset})";
1163 $size = "ndr_string_length(r->$e->{NAME}, sizeof(*r->$e->{NAME}))";
1166 $size = ParseExpr($e->{LEVELS}[0]->{SIZE_IS}, $env);
1169 pidl "NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, $size));";
1171 pidl "NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, ndr_string_array_size(ndr, r->$e->{NAME})));";
1175 pidl "if (ndr_flags & NDR_SCALARS) {";
1178 pidl "NDR_CHECK(ndr_push_align(ndr, $struct->{ALIGN}));";
1180 if (defined($struct->{PROPERTIES}{relative_base})) {
1181 # set the current offset as base for relative pointers
1182 # and store it based on the toplevel struct/union
1183 pidl "NDR_CHECK(ndr_push_setup_relative_base_offset1(ndr, r, ndr->offset));";
1186 foreach my $e (@{$struct->{ELEMENTS}}) {
1187 ParseElementPush($e, "ndr", "r->", $env, 1, 0);
1193 pidl "if (ndr_flags & NDR_BUFFERS) {";
1195 if (defined($struct->{PROPERTIES}{relative_base})) {
1196 # retrieve the current offset as base for relative pointers
1197 # based on the toplevel struct/union
1198 pidl "NDR_CHECK(ndr_push_setup_relative_base_offset2(ndr, r));";
1200 foreach my $e (@{$struct->{ELEMENTS}}) {
1201 ParseElementPush($e, "ndr", "r->", $env, 0, 1);
1208 # restore the old relative_base_offset
1209 pidl "ndr_push_restore_relative_base_offset(ndr, _save_relative_base_offset);" if defined($struct->{PROPERTIES}{relative_base});
1212 #####################################################################
1213 # generate a push function for an enum
1214 sub ParseEnumPush($$)
1216 my($enum,$name) = @_;
1217 my($type_fn) = $enum->{BASE_TYPE};
1220 pidl "NDR_CHECK(ndr_push_$type_fn(ndr, NDR_SCALARS, r));";
1224 #####################################################################
1225 # generate a pull function for an enum
1226 sub ParseEnumPull($$)
1228 my($enum,$name) = @_;
1229 my($type_fn) = $enum->{BASE_TYPE};
1230 my($type_v_decl) = mapType($type_fn);
1232 pidl "$type_v_decl v;";
1234 pidl "NDR_CHECK(ndr_pull_$type_fn(ndr, NDR_SCALARS, &v));";
1240 #####################################################################
1241 # generate a print function for an enum
1242 sub ParseEnumPrint($$)
1244 my($enum,$name) = @_;
1246 pidl "const char *val = NULL;";
1251 pidl "switch (r) {";
1253 my $els = \@{$enum->{ELEMENTS}};
1254 foreach my $i (0 .. $#{$els}) {
1255 my $e = ${$els}[$i];
1257 if ($e =~ /^(.*)=/) {
1260 pidl "case $e: val = \"$e\"; break;";
1266 pidl "ndr_print_enum(ndr, name, \"$enum->{TYPE}\", val, r);";
1274 return "enum $e->{NAME} " .
1275 ($t eq "pull"?"*":"") . "r";
1278 $typefamily{ENUM} = {
1280 PUSH_FN_BODY => \&ParseEnumPush,
1281 PULL_FN_BODY => \&ParseEnumPull,
1282 PRINT_FN_BODY => \&ParseEnumPrint,
1285 #####################################################################
1286 # generate a push function for a bitmap
1287 sub ParseBitmapPush($$)
1289 my($bitmap,$name) = @_;
1290 my($type_fn) = $bitmap->{BASE_TYPE};
1292 start_flags($bitmap);
1294 pidl "NDR_CHECK(ndr_push_$type_fn(ndr, NDR_SCALARS, r));";
1299 #####################################################################
1300 # generate a pull function for an bitmap
1301 sub ParseBitmapPull($$)
1303 my($bitmap,$name) = @_;
1304 my $type_fn = $bitmap->{BASE_TYPE};
1305 my($type_decl) = mapType($bitmap->{BASE_TYPE});
1307 pidl "$type_decl v;";
1308 start_flags($bitmap);
1309 pidl "NDR_CHECK(ndr_pull_$type_fn(ndr, NDR_SCALARS, &v));";
1315 #####################################################################
1316 # generate a print function for an bitmap
1317 sub ParseBitmapPrintElement($$$)
1319 my($e,$bitmap,$name) = @_;
1320 my($type_decl) = mapType($bitmap->{BASE_TYPE});
1321 my($type_fn) = $bitmap->{BASE_TYPE};
1324 if ($e =~ /^(\w+) .*$/) {
1327 die "Bitmap: \"$name\" invalid Flag: \"$e\"";
1330 pidl "ndr_print_bitmap_flag(ndr, sizeof($type_decl), \"$flag\", $flag, r);";
1333 #####################################################################
1334 # generate a print function for an bitmap
1335 sub ParseBitmapPrint($$)
1337 my($bitmap,$name) = @_;
1338 my($type_decl) = mapType($bitmap->{TYPE});
1339 my($type_fn) = $bitmap->{BASE_TYPE};
1341 start_flags($bitmap);
1343 pidl "ndr_print_$type_fn(ndr, name, r);";
1345 pidl "ndr->depth++;";
1346 foreach my $e (@{$bitmap->{ELEMENTS}}) {
1347 ParseBitmapPrintElement($e, $bitmap, $name);
1349 pidl "ndr->depth--;";
1357 return mapType(Parse::Pidl::Typelist::bitmap_type_fn($e->{DATA})) .
1358 ($t eq "pull"?" *":" ") . "r";
1361 $typefamily{BITMAP} = {
1362 DECL => \&DeclBitmap,
1363 PUSH_FN_BODY => \&ParseBitmapPush,
1364 PULL_FN_BODY => \&ParseBitmapPull,
1365 PRINT_FN_BODY => \&ParseBitmapPrint,
1368 #####################################################################
1369 # generate a struct print function
1370 sub ParseStructPrint($$)
1372 my($struct,$name) = @_;
1374 return unless defined $struct->{ELEMENTS};
1376 my $env = GenerateStructEnv($struct);
1378 EnvSubstituteValue($env, $struct);
1380 foreach my $e (@{$struct->{ELEMENTS}}) {
1381 DeclareArrayVariables($e);
1384 pidl "ndr_print_struct(ndr, name, \"$name\");";
1386 start_flags($struct);
1388 pidl "ndr->depth++;";
1389 foreach my $e (@{$struct->{ELEMENTS}}) {
1390 ParseElementPrint($e, "r->$e->{NAME}", $env);
1392 pidl "ndr->depth--;";
1397 sub DeclarePtrVariables($)
1400 foreach my $l (@{$e->{LEVELS}}) {
1401 if ($l->{TYPE} eq "POINTER" and
1402 not ($l->{POINTER_TYPE} eq "ref" and $l->{LEVEL} eq "TOP")) {
1403 pidl "uint32_t _ptr_$e->{NAME};";
1409 sub DeclareArrayVariables($)
1413 foreach my $l (@{$e->{LEVELS}}) {
1414 next if has_fast_array($e,$l);
1415 next if is_charset_array($e,$l);
1416 if ($l->{TYPE} eq "ARRAY") {
1417 pidl "uint32_t cntr_$e->{NAME}_$l->{LEVEL_INDEX};";
1422 sub need_decl_mem_ctx($$)
1427 return 0 if has_fast_array($e,$l);
1428 return 0 if is_charset_array($e,$l);
1429 return 1 if (($l->{TYPE} eq "ARRAY") and not $l->{IS_FIXED});
1431 if (($l->{TYPE} eq "POINTER") and ($l->{POINTER_TYPE} eq "ref")) {
1432 my $nl = GetNextLevel($e, $l);
1433 my $next_is_array = ($nl->{TYPE} eq "ARRAY");
1434 my $next_is_string = (($nl->{TYPE} eq "DATA") and
1435 ($nl->{DATA_TYPE} eq "string"));
1436 return 0 if ($next_is_array or $next_is_string);
1438 return 1 if ($l->{TYPE} eq "POINTER");
1443 sub DeclareMemCtxVariables($)
1446 foreach my $l (@{$e->{LEVELS}}) {
1447 if (need_decl_mem_ctx($e, $l)) {
1448 pidl "TALLOC_CTX *_mem_save_$e->{NAME}_$l->{LEVEL_INDEX};";
1453 #####################################################################
1454 # parse a struct - pull side
1455 sub ParseStructPull($$)
1457 my($struct,$name) = @_;
1459 return unless defined $struct->{ELEMENTS};
1461 my $env = GenerateStructEnv($struct);
1463 # declare any internal pointers we need
1464 foreach my $e (@{$struct->{ELEMENTS}}) {
1465 DeclarePtrVariables($e);
1466 DeclareArrayVariables($e);
1467 DeclareMemCtxVariables($e);
1470 # save the old relative_base_offset
1471 pidl "uint32_t _save_relative_base_offset = ndr_pull_get_relative_base_offset(ndr);" if defined($struct->{PROPERTIES}{relative_base});
1473 start_flags($struct);
1475 pidl "if (ndr_flags & NDR_SCALARS) {";
1478 if (defined $struct->{SURROUNDING_ELEMENT}) {
1479 pidl "NDR_CHECK(ndr_pull_array_size(ndr, &r->$struct->{SURROUNDING_ELEMENT}->{NAME}));";
1482 pidl "NDR_CHECK(ndr_pull_align(ndr, $struct->{ALIGN}));";
1484 if (defined($struct->{PROPERTIES}{relative_base})) {
1485 # set the current offset as base for relative pointers
1486 # and store it based on the toplevel struct/union
1487 pidl "NDR_CHECK(ndr_pull_setup_relative_base_offset1(ndr, r, ndr->offset));";
1490 foreach my $e (@{$struct->{ELEMENTS}}) {
1491 ParseElementPull($e, "ndr", "r->", $env, 1, 0);
1498 pidl "if (ndr_flags & NDR_BUFFERS) {";
1500 if (defined($struct->{PROPERTIES}{relative_base})) {
1501 # retrieve the current offset as base for relative pointers
1502 # based on the toplevel struct/union
1503 pidl "NDR_CHECK(ndr_pull_setup_relative_base_offset2(ndr, r));";
1505 foreach my $e (@{$struct->{ELEMENTS}}) {
1506 ParseElementPull($e, "ndr", "r->", $env, 0, 1);
1515 # restore the old relative_base_offset
1516 pidl "ndr_pull_restore_relative_base_offset(ndr, _save_relative_base_offset);" if defined($struct->{PROPERTIES}{relative_base});
1519 #####################################################################
1520 # calculate size of ndr struct
1521 sub ParseStructNdrSize($)
1526 if (my $flags = has_property($t, "flag")) {
1527 pidl "flags |= $flags;";
1529 pidl "return ndr_size_struct(r, flags, (ndr_push_flags_fn_t)ndr_push_$t->{NAME});";
1535 return ($t ne "pull"?"const ":"") . "struct $e->{NAME} *r";
1538 sub ArgsStructNdrSize($)
1541 return "const struct $d->{NAME} *r, int flags";
1544 $typefamily{STRUCT} = {
1545 PUSH_FN_BODY => \&ParseStructPush,
1546 DECL => \&DeclStruct,
1547 PULL_FN_BODY => \&ParseStructPull,
1548 PRINT_FN_BODY => \&ParseStructPrint,
1549 SIZE_FN_BODY => \&ParseStructNdrSize,
1550 SIZE_FN_ARGS => \&ArgsStructNdrSize,
1553 #####################################################################
1554 # calculate size of ndr struct
1555 sub ParseUnionNdrSize($)
1560 if (my $flags = has_property($t, "flag")) {
1561 pidl "flags |= $flags;";
1564 pidl "return ndr_size_union(r, flags, level, (ndr_push_flags_fn_t)ndr_push_$t->{NAME});";
1567 #####################################################################
1568 # parse a union - push side
1569 sub ParseUnionPush($$)
1572 my $have_default = 0;
1574 # save the old relative_base_offset
1575 pidl "uint32_t _save_relative_base_offset = ndr_push_get_relative_base_offset(ndr);" if defined($e->{PROPERTIES}{relative_base});
1580 pidl "level = ndr_push_get_switch_value(ndr, r);";
1582 pidl "if (ndr_flags & NDR_SCALARS) {";
1585 if (defined($e->{SWITCH_TYPE})) {
1586 pidl "NDR_CHECK(ndr_push_$e->{SWITCH_TYPE}(ndr, NDR_SCALARS, level));";
1589 pidl "switch (level) {";
1591 foreach my $el (@{$e->{ELEMENTS}}) {
1592 if ($el->{CASE} eq "default") {
1595 pidl "$el->{CASE}:";
1597 if ($el->{TYPE} ne "EMPTY") {
1599 if (defined($e->{PROPERTIES}{relative_base})) {
1600 pidl "NDR_CHECK(ndr_push_align(ndr, $el->{ALIGN}));";
1601 # set the current offset as base for relative pointers
1602 # and store it based on the toplevel struct/union
1603 pidl "NDR_CHECK(ndr_push_setup_relative_base_offset1(ndr, r, ndr->offset));";
1605 DeclareArrayVariables($el);
1606 ParseElementPush($el, "ndr", "r->", {}, 1, 0);
1612 if (! $have_default) {
1614 pidl "\treturn ndr_push_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);";
1620 pidl "if (ndr_flags & NDR_BUFFERS) {";
1622 if (defined($e->{PROPERTIES}{relative_base})) {
1623 # retrieve the current offset as base for relative pointers
1624 # based on the toplevel struct/union
1625 pidl "NDR_CHECK(ndr_push_setup_relative_base_offset2(ndr, r));";
1627 pidl "switch (level) {";
1629 foreach my $el (@{$e->{ELEMENTS}}) {
1630 pidl "$el->{CASE}:";
1631 if ($el->{TYPE} ne "EMPTY") {
1633 ParseElementPush($el, "ndr", "r->", {}, 0, 1);
1639 if (! $have_default) {
1641 pidl "\treturn ndr_push_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);";
1649 # restore the old relative_base_offset
1650 pidl "ndr_push_restore_relative_base_offset(ndr, _save_relative_base_offset);" if defined($e->{PROPERTIES}{relative_base});
1653 #####################################################################
1655 sub ParseUnionPrint($$)
1658 my $have_default = 0;
1661 foreach my $el (@{$e->{ELEMENTS}}) {
1662 DeclareArrayVariables($el);
1667 pidl "level = ndr_print_get_switch_value(ndr, r);";
1669 pidl "ndr_print_union(ndr, name, level, \"$name\");";
1671 pidl "switch (level) {";
1673 foreach my $el (@{$e->{ELEMENTS}}) {
1674 if ($el->{CASE} eq "default") {
1677 pidl "$el->{CASE}:";
1678 if ($el->{TYPE} ne "EMPTY") {
1680 ParseElementPrint($el, "r->$el->{NAME}", {});
1686 if (! $have_default) {
1688 pidl "\tndr_print_bad_level(ndr, name, level);";
1696 #####################################################################
1697 # parse a union - pull side
1698 sub ParseUnionPull($$)
1701 my $have_default = 0;
1702 my $switch_type = $e->{SWITCH_TYPE};
1704 # save the old relative_base_offset
1705 pidl "uint32_t _save_relative_base_offset = ndr_pull_get_relative_base_offset(ndr);" if defined($e->{PROPERTIES}{relative_base});
1707 if (defined($switch_type)) {
1708 if (Parse::Pidl::Typelist::typeIs($switch_type, "ENUM")) {
1709 $switch_type = Parse::Pidl::Typelist::enum_type_fn(getType($switch_type));
1711 pidl mapType($switch_type) . " _level;";
1714 my %double_cases = ();
1715 foreach my $el (@{$e->{ELEMENTS}}) {
1716 next if ($el->{TYPE} eq "EMPTY");
1717 next if ($double_cases{"$el->{NAME}"});
1718 DeclareMemCtxVariables($el);
1719 $double_cases{"$el->{NAME}"} = 1;
1724 pidl "level = ndr_pull_get_switch_value(ndr, r);";
1726 pidl "if (ndr_flags & NDR_SCALARS) {";
1729 if (defined($switch_type)) {
1730 pidl "NDR_CHECK(ndr_pull_$switch_type(ndr, NDR_SCALARS, &_level));";
1731 pidl "if (_level != level) {";
1732 pidl "\treturn ndr_pull_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value %u for $name\", _level);";
1736 pidl "switch (level) {";
1738 foreach my $el (@{$e->{ELEMENTS}}) {
1739 if ($el->{CASE} eq "default") {
1742 pidl "$el->{CASE}: {";
1744 if ($el->{TYPE} ne "EMPTY") {
1746 DeclarePtrVariables($el);
1747 DeclareArrayVariables($el);
1748 if (defined($e->{PROPERTIES}{relative_base})) {
1749 pidl "NDR_CHECK(ndr_pull_align(ndr, $el->{ALIGN}));";
1750 # set the current offset as base for relative pointers
1751 # and store it based on the toplevel struct/union
1752 pidl "NDR_CHECK(ndr_pull_setup_relative_base_offset1(ndr, r, ndr->offset));";
1754 ParseElementPull($el, "ndr", "r->", {}, 1, 0);
1760 if (! $have_default) {
1762 pidl "\treturn ndr_pull_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);";
1768 pidl "if (ndr_flags & NDR_BUFFERS) {";
1770 if (defined($e->{PROPERTIES}{relative_base})) {
1771 # retrieve the current offset as base for relative pointers
1772 # based on the toplevel struct/union
1773 pidl "NDR_CHECK(ndr_pull_setup_relative_base_offset2(ndr, r));";
1775 pidl "switch (level) {";
1777 foreach my $el (@{$e->{ELEMENTS}}) {
1778 pidl "$el->{CASE}:";
1779 if ($el->{TYPE} ne "EMPTY") {
1781 ParseElementPull($el, "ndr", "r->", {}, 0, 1);
1787 if (! $have_default) {
1789 pidl "\treturn ndr_pull_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);";
1800 # restore the old relative_base_offset
1801 pidl "ndr_pull_restore_relative_base_offset(ndr, _save_relative_base_offset);" if defined($e->{PROPERTIES}{relative_base});
1807 return ($t ne "pull"?"const ":"") . "union $e->{NAME} *r";
1810 sub ArgsUnionNdrSize($)
1813 return "const union $d->{NAME} *r, uint32_t level, int flags";
1816 $typefamily{UNION} = {
1817 PUSH_FN_BODY => \&ParseUnionPush,
1818 DECL => \&DeclUnion,
1819 PULL_FN_BODY => \&ParseUnionPull,
1820 PRINT_FN_BODY => \&ParseUnionPrint,
1821 SIZE_FN_ARGS => \&ArgsUnionNdrSize,
1822 SIZE_FN_BODY => \&ParseUnionNdrSize,
1825 #####################################################################
1826 # parse a typedef - push side
1827 sub ParseTypedefPush($)
1831 my $args = $typefamily{$e->{DATA}->{TYPE}}->{DECL}->($e,"push");
1832 fn_declare($e, "NTSTATUS ndr_push_$e->{NAME}(struct ndr_push *ndr, int ndr_flags, $args)");
1836 $typefamily{$e->{DATA}->{TYPE}}->{PUSH_FN_BODY}->($e->{DATA}, $e->{NAME});
1837 pidl "return NT_STATUS_OK;";
1843 #####################################################################
1844 # parse a typedef - pull side
1845 sub ParseTypedefPull($)
1849 my $args = $typefamily{$e->{DATA}->{TYPE}}->{DECL}->($e,"pull");
1851 fn_declare($e, "NTSTATUS ndr_pull_$e->{NAME}(struct ndr_pull *ndr, int ndr_flags, $args)");
1855 $typefamily{$e->{DATA}->{TYPE}}->{PULL_FN_BODY}->($e->{DATA}, $e->{NAME});
1856 pidl "return NT_STATUS_OK;";
1862 #####################################################################
1863 # parse a typedef - print side
1864 sub ParseTypedefPrint($)
1868 my $args = $typefamily{$e->{DATA}->{TYPE}}->{DECL}->($e,"print");
1870 pidl "_PUBLIC_ void ndr_print_$e->{NAME}(struct ndr_print *ndr, const char *name, $args)";
1871 pidl_hdr "void ndr_print_$e->{NAME}(struct ndr_print *ndr, const char *name, $args);";
1874 $typefamily{$e->{DATA}->{TYPE}}->{PRINT_FN_BODY}->($e->{DATA}, $e->{NAME});
1880 #####################################################################
1881 ## calculate the size of a structure
1882 sub ParseTypedefNdrSize($)
1886 my $tf = $typefamily{$t->{DATA}->{TYPE}};
1887 my $args = $tf->{SIZE_FN_ARGS}->($t);
1889 fn_declare($t, "size_t ndr_size_$t->{NAME}($args)");
1893 $typefamily{$t->{DATA}->{TYPE}}->{SIZE_FN_BODY}->($t);
1899 #####################################################################
1900 # parse a function - print side
1901 sub ParseFunctionPrint($)
1905 return if has_property($fn, "noprint");
1907 pidl "_PUBLIC_ void ndr_print_$fn->{NAME}(struct ndr_print *ndr, const char *name, int flags, const struct $fn->{NAME} *r)";
1908 pidl_hdr "void ndr_print_$fn->{NAME}(struct ndr_print *ndr, const char *name, int flags, const struct $fn->{NAME} *r);";
1912 foreach my $e (@{$fn->{ELEMENTS}}) {
1913 DeclareArrayVariables($e);
1916 pidl "ndr_print_struct(ndr, name, \"$fn->{NAME}\");";
1917 pidl "ndr->depth++;";
1919 pidl "if (flags & NDR_SET_VALUES) {";
1920 pidl "\tndr->flags |= LIBNDR_PRINT_SET_VALUES;";
1923 pidl "if (flags & NDR_IN) {";
1925 pidl "ndr_print_struct(ndr, \"in\", \"$fn->{NAME}\");";
1926 pidl "ndr->depth++;";
1928 my $env = GenerateFunctionInEnv($fn);
1929 EnvSubstituteValue($env, $fn);
1931 foreach my $e (@{$fn->{ELEMENTS}}) {
1932 if (grep(/in/,@{$e->{DIRECTION}})) {
1933 ParseElementPrint($e, "r->in.$e->{NAME}", $env);
1936 pidl "ndr->depth--;";
1940 pidl "if (flags & NDR_OUT) {";
1942 pidl "ndr_print_struct(ndr, \"out\", \"$fn->{NAME}\");";
1943 pidl "ndr->depth++;";
1945 $env = GenerateFunctionOutEnv($fn);
1946 foreach my $e (@{$fn->{ELEMENTS}}) {
1947 if (grep(/out/,@{$e->{DIRECTION}})) {
1948 ParseElementPrint($e, "r->out.$e->{NAME}", $env);
1951 if ($fn->{RETURN_TYPE}) {
1952 pidl "ndr_print_$fn->{RETURN_TYPE}(ndr, \"result\", r->out.result);";
1954 pidl "ndr->depth--;";
1958 pidl "ndr->depth--;";
1964 #####################################################################
1966 sub ParseFunctionPush($)
1970 return if has_property($fn, "nopush");
1972 fn_declare($fn, "NTSTATUS ndr_push_$fn->{NAME}(struct ndr_push *ndr, int flags, const struct $fn->{NAME} *r)");
1977 foreach my $e (@{$fn->{ELEMENTS}}) {
1978 DeclareArrayVariables($e);
1981 pidl "if (flags & NDR_IN) {";
1984 my $env = GenerateFunctionInEnv($fn);
1986 EnvSubstituteValue($env, $fn);
1988 foreach my $e (@{$fn->{ELEMENTS}}) {
1989 if (grep(/in/,@{$e->{DIRECTION}})) {
1990 ParseElementPush($e, "ndr", "r->in.", $env, 1, 1);
1997 pidl "if (flags & NDR_OUT) {";
2000 $env = GenerateFunctionOutEnv($fn);
2001 foreach my $e (@{$fn->{ELEMENTS}}) {
2002 if (grep(/out/,@{$e->{DIRECTION}})) {
2003 ParseElementPush($e, "ndr", "r->out.", $env, 1, 1);
2007 if ($fn->{RETURN_TYPE}) {
2008 pidl "NDR_CHECK(ndr_push_$fn->{RETURN_TYPE}(ndr, NDR_SCALARS, r->out.result));";
2013 pidl "return NT_STATUS_OK;";
2019 sub AllocateArrayLevel($$$$$)
2021 my ($e,$l,$ndr,$env,$size) = @_;
2023 my $var = ParseExpr($e->{NAME}, $env);
2025 check_null_pointer($size);
2026 my $pl = GetPrevLevel($e, $l);
2027 if (defined($pl) and
2028 $pl->{TYPE} eq "POINTER" and
2029 $pl->{POINTER_TYPE} eq "ref"
2030 and not $l->{IS_ZERO_TERMINATED}) {
2031 pidl "if (ndr->flags & LIBNDR_FLAG_REF_ALLOC) {";
2032 pidl "\tNDR_PULL_ALLOC_N($ndr, $var, $size);";
2034 if (grep(/in/,@{$e->{DIRECTION}}) and
2035 grep(/out/,@{$e->{DIRECTION}})) {
2036 pidl "memcpy(r->out.$e->{NAME},r->in.$e->{NAME},$size * sizeof(*r->in.$e->{NAME}));";
2041 pidl "NDR_PULL_ALLOC_N($ndr, $var, $size);";
2044 #####################################################################
2046 sub ParseFunctionPull($)
2050 return if has_property($fn, "nopull");
2052 # pull function args
2053 fn_declare($fn, "NTSTATUS ndr_pull_$fn->{NAME}(struct ndr_pull *ndr, int flags, struct $fn->{NAME} *r)");
2057 # declare any internal pointers we need
2058 foreach my $e (@{$fn->{ELEMENTS}}) {
2059 DeclarePtrVariables($e);
2060 DeclareArrayVariables($e);
2063 my %double_cases = ();
2064 foreach my $e (@{$fn->{ELEMENTS}}) {
2065 next if ($e->{TYPE} eq "EMPTY");
2066 next if ($double_cases{"$e->{NAME}"});
2067 DeclareMemCtxVariables($e);
2068 $double_cases{"$e->{NAME}"} = 1;
2071 pidl "if (flags & NDR_IN) {";
2074 # auto-init the out section of a structure. I originally argued that
2075 # this was a bad idea as it hides bugs, but coping correctly
2076 # with initialisation and not wiping ref vars is turning
2077 # out to be too tricky (tridge)
2078 foreach my $e (@{$fn->{ELEMENTS}}) {
2079 next unless grep(/out/, @{$e->{DIRECTION}});
2080 pidl "ZERO_STRUCT(r->out);";
2085 my $env = GenerateFunctionInEnv($fn);
2087 foreach my $e (@{$fn->{ELEMENTS}}) {
2088 next unless (grep(/in/, @{$e->{DIRECTION}}));
2089 ParseElementPull($e, "ndr", "r->in.", $env, 1, 1);
2092 # allocate the "simple" out ref variables. FIXME: Shouldn't this have it's
2093 # own flag rather then be in NDR_IN ?
2095 foreach my $e (@{$fn->{ELEMENTS}}) {
2096 next unless (grep(/out/, @{$e->{DIRECTION}}));
2097 next unless ($e->{LEVELS}[0]->{TYPE} eq "POINTER" and
2098 $e->{LEVELS}[0]->{POINTER_TYPE} eq "ref");
2099 next if (($e->{LEVELS}[1]->{TYPE} eq "DATA") and
2100 ($e->{LEVELS}[1]->{DATA_TYPE} eq "string"));
2101 next if (($e->{LEVELS}[1]->{TYPE} eq "ARRAY")
2102 and $e->{LEVELS}[1]->{IS_ZERO_TERMINATED});
2104 if ($e->{LEVELS}[1]->{TYPE} eq "ARRAY") {
2105 my $size = ParseExpr($e->{LEVELS}[1]->{SIZE_IS}, $env);
2106 check_null_pointer($size);
2108 pidl "NDR_PULL_ALLOC_N(ndr, r->out.$e->{NAME}, $size);";
2110 if (grep(/in/, @{$e->{DIRECTION}})) {
2111 pidl "memcpy(r->out.$e->{NAME}, r->in.$e->{NAME}, $size * sizeof(*r->in.$e->{NAME}));";
2113 pidl "memset(r->out.$e->{NAME}, 0, $size * sizeof(*r->out.$e->{NAME}));";
2116 pidl "NDR_PULL_ALLOC(ndr, r->out.$e->{NAME});";
2118 if (grep(/in/, @{$e->{DIRECTION}})) {
2119 pidl "*r->out.$e->{NAME} = *r->in.$e->{NAME};";
2121 pidl "ZERO_STRUCTP(r->out.$e->{NAME});";
2130 pidl "if (flags & NDR_OUT) {";
2133 $env = GenerateFunctionOutEnv($fn);
2134 foreach my $e (@{$fn->{ELEMENTS}}) {
2135 next unless grep(/out/, @{$e->{DIRECTION}});
2136 ParseElementPull($e, "ndr", "r->out.", $env, 1, 1);
2139 if ($fn->{RETURN_TYPE}) {
2140 pidl "NDR_CHECK(ndr_pull_$fn->{RETURN_TYPE}(ndr, NDR_SCALARS, &r->out.result));";
2147 pidl "return NT_STATUS_OK;";
2153 #####################################################################
2154 # produce a function call table
2155 sub FunctionTable($)
2157 my($interface) = shift;
2159 my $uname = uc $interface->{NAME};
2161 return if ($#{$interface->{FUNCTIONS}}+1 == 0);
2162 return unless defined ($interface->{PROPERTIES}->{uuid});
2164 pidl "static const struct dcerpc_interface_call $interface->{NAME}\_calls[] = {";
2165 foreach my $d (@{$interface->{FUNCTIONS}}) {
2166 next if not defined($d->{OPNUM});
2168 pidl "\t\t\"$d->{NAME}\",";
2169 pidl "\t\tsizeof(struct $d->{NAME}),";
2170 pidl "\t\t(ndr_push_flags_fn_t) ndr_push_$d->{NAME},";
2171 pidl "\t\t(ndr_pull_flags_fn_t) ndr_pull_$d->{NAME},";
2172 pidl "\t\t(ndr_print_function_t) ndr_print_$d->{NAME},";
2173 pidl "\t\t".($d->{ASYNC}?"True":"False").",";
2177 pidl "\t{ NULL, 0, NULL, NULL, NULL, False }";
2181 pidl "static const char * const $interface->{NAME}\_endpoint_strings[] = {";
2182 foreach my $ep (@{$interface->{ENDPOINTS}}) {
2185 my $endpoint_count = $#{$interface->{ENDPOINTS}}+1;
2190 pidl "static const struct dcerpc_endpoint_list $interface->{NAME}\_endpoints = {";
2191 pidl "\t.count\t= $endpoint_count,";
2192 pidl "\t.names\t= $interface->{NAME}\_endpoint_strings";
2196 if (! defined $interface->{PROPERTIES}->{authservice}) {
2197 $interface->{PROPERTIES}->{authservice} = "\"host\"";
2200 my @a = split / /, $interface->{PROPERTIES}->{authservice};
2201 my $authservice_count = $#a + 1;
2203 pidl "static const char * const $interface->{NAME}\_authservice_strings[] = {";
2204 foreach my $ap (@a) {
2210 pidl "static const struct dcerpc_authservice_list $interface->{NAME}\_authservices = {";
2211 pidl "\t.count\t= $endpoint_count,";
2212 pidl "\t.names\t= $interface->{NAME}\_authservice_strings";
2216 pidl "\nconst struct dcerpc_interface_table dcerpc_table_$interface->{NAME} = {";
2217 pidl "\t.name\t\t= \"$interface->{NAME}\",";
2218 pidl "\t.uuid\t\t= ". print_uuid($interface->{UUID}) .",";
2219 pidl "\t.if_version\t= DCERPC_$uname\_VERSION,";
2220 pidl "\t.helpstring\t= DCERPC_$uname\_HELPSTRING,";
2221 pidl "\t.num_calls\t= $count,";
2222 pidl "\t.calls\t\t= $interface->{NAME}\_calls,";
2223 pidl "\t.endpoints\t= &$interface->{NAME}\_endpoints,";
2224 pidl "\t.authservices\t= &$interface->{NAME}\_authservices";
2230 #####################################################################
2231 # generate prototypes and defines for the interface definitions
2232 # FIXME: these prototypes are for the DCE/RPC client functions, not the
2233 # NDR parser and so do not belong here, technically speaking
2234 sub HeaderInterface($)
2236 my($interface) = shift;
2240 if (defined $interface->{PROPERTIES}->{depends}) {
2241 my @d = split / /, $interface->{PROPERTIES}->{depends};
2242 foreach my $i (@d) {
2243 pidl_hdr "#include \"librpc/gen_ndr/ndr_$i\.h\"";
2247 if (defined $interface->{PROPERTIES}->{helper}) {
2248 foreach (split / /, $interface->{PROPERTIES}->{helper}) {
2249 pidl_hdr "#include $_";
2253 if (defined $interface->{PROPERTIES}->{uuid}) {
2254 my $name = uc $interface->{NAME};
2255 pidl_hdr "#define DCERPC_$name\_UUID " .
2256 Parse::Pidl::Util::make_str(lc($interface->{PROPERTIES}->{uuid}));
2258 if(!defined $interface->{PROPERTIES}->{version}) { $interface->{PROPERTIES}->{version} = "0.0"; }
2259 pidl_hdr "#define DCERPC_$name\_VERSION $interface->{PROPERTIES}->{version}";
2261 pidl_hdr "#define DCERPC_$name\_NAME \"$interface->{NAME}\"";
2263 if(!defined $interface->{PROPERTIES}->{helpstring}) { $interface->{PROPERTIES}->{helpstring} = "NULL"; }
2264 pidl_hdr "#define DCERPC_$name\_HELPSTRING $interface->{PROPERTIES}->{helpstring}";
2266 pidl_hdr "extern const struct dcerpc_interface_table dcerpc_table_$interface->{NAME};";
2267 pidl_hdr "NTSTATUS dcerpc_server_$interface->{NAME}_init(void);";
2270 foreach (@{$interface->{FUNCTIONS}}) {
2271 next if has_property($_, "noopnum");
2272 next if grep(/$_->{NAME}/,@{$interface->{INHERITED_FUNCTIONS}});
2273 my $u_name = uc $_->{NAME};
2275 my $val = sprintf("0x%02x", $count);
2276 if (defined($interface->{BASE})) {
2277 $val .= " + DCERPC_" . uc $interface->{BASE} . "_CALL_COUNT";
2280 pidl_hdr "#define DCERPC_$u_name ($val)";
2282 pidl_hdr "NTSTATUS dcerpc_$_->{NAME}(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct $_->{NAME} *r);";
2283 pidl_hdr "struct rpc_request *dcerpc_$_->{NAME}\_send(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct $_->{NAME} *r);";
2291 if (defined($interface->{BASE})) {
2292 $val .= " + DCERPC_" . uc $interface->{BASE} . "_CALL_COUNT";
2295 pidl_hdr "#define DCERPC_" . uc $interface->{NAME} . "_CALL_COUNT ($val)";
2299 #####################################################################
2300 # parse the interface definitions
2301 sub ParseInterface($$)
2303 my($interface,$needed) = @_;
2305 pidl_hdr "#ifndef _HEADER_RPC_$interface->{NAME}";
2306 pidl_hdr "#define _HEADER_RPC_$interface->{NAME}";
2310 if ($needed->{"compression"}) {
2311 pidl "#include \"librpc/ndr/ndr_compression.h\"";
2314 HeaderInterface($interface);
2317 foreach my $d (@{$interface->{TYPES}}) {
2318 ($needed->{"push_$d->{NAME}"}) && ParseTypedefPush($d);
2319 ($needed->{"pull_$d->{NAME}"}) && ParseTypedefPull($d);
2320 ($needed->{"print_$d->{NAME}"}) && ParseTypedefPrint($d);
2322 # Make sure we don't generate a function twice...
2323 $needed->{"push_$d->{NAME}"} = $needed->{"pull_$d->{NAME}"} =
2324 $needed->{"print_$d->{NAME}"} = 0;
2326 ($needed->{"ndr_size_$d->{NAME}"}) && ParseTypedefNdrSize($d);
2330 foreach my $d (@{$interface->{FUNCTIONS}}) {
2331 ($needed->{"push_$d->{NAME}"}) && ParseFunctionPush($d);
2332 ($needed->{"pull_$d->{NAME}"}) && ParseFunctionPull($d);
2333 ($needed->{"print_$d->{NAME}"}) && ParseFunctionPrint($d);
2335 # Make sure we don't generate a function twice...
2336 $needed->{"push_$d->{NAME}"} = $needed->{"pull_$d->{NAME}"} =
2337 $needed->{"print_$d->{NAME}"} = 0;
2340 FunctionTable($interface);
2342 pidl_hdr "#endif /* _HEADER_RPC_$interface->{NAME} */";
2345 #####################################################################
2346 # parse a parsed IDL structure back into an IDL file
2349 my($ndr,$basename) = @_;
2355 pidl_hdr "/* header auto-generated by pidl */";
2358 pidl "/* parser auto-generated by pidl */";
2364 ($_->{TYPE} eq "INTERFACE") && NeededInterface($_, \%needed);
2368 ($_->{TYPE} eq "INTERFACE") && ParseInterface($_, \%needed);
2371 return ($res_hdr, $res);
2374 sub NeededFunction($$)
2376 my ($fn,$needed) = @_;
2377 $needed->{"pull_$fn->{NAME}"} = 1;
2378 $needed->{"push_$fn->{NAME}"} = 1;
2379 $needed->{"print_$fn->{NAME}"} = 1;
2380 foreach my $e (@{$fn->{ELEMENTS}}) {
2382 unless(defined($needed->{"pull_$e->{TYPE}"})) {
2383 $needed->{"pull_$e->{TYPE}"} = 1;
2385 unless(defined($needed->{"push_$e->{TYPE}"})) {
2386 $needed->{"push_$e->{TYPE}"} = 1;
2388 unless(defined($needed->{"print_$e->{TYPE}"})) {
2389 $needed->{"print_$e->{TYPE}"} = 1;
2394 sub NeededTypedef($$)
2396 my ($t,$needed) = @_;
2397 if (has_property($t, "public")) {
2398 $needed->{"pull_$t->{NAME}"} = not has_property($t, "nopull");
2399 $needed->{"push_$t->{NAME}"} = not has_property($t, "nopush");
2400 $needed->{"print_$t->{NAME}"} = not has_property($t, "noprint");
2403 if ($t->{DATA}->{TYPE} eq "STRUCT" or $t->{DATA}->{TYPE} eq "UNION") {
2404 if (has_property($t, "gensize")) {
2405 $needed->{"ndr_size_$t->{NAME}"} = 1;
2408 for my $e (@{$t->{DATA}->{ELEMENTS}}) {
2409 $e->{PARENT} = $t->{DATA};
2410 if (has_property($e, "compression")) {
2411 $needed->{"compression"} = 1;
2413 if ($needed->{"pull_$t->{NAME}"} and
2414 not defined($needed->{"pull_$e->{TYPE}"})) {
2415 $needed->{"pull_$e->{TYPE}"} = 1;
2417 if ($needed->{"push_$t->{NAME}"} and
2418 not defined($needed->{"push_$e->{TYPE}"})) {
2419 $needed->{"push_$e->{TYPE}"} = 1;
2421 if ($needed->{"print_$t->{NAME}"} and
2422 not defined($needed->{"print_$e->{TYPE}"})) {
2423 $needed->{"print_$e->{TYPE}"} = 1;
2429 #####################################################################
2430 # work out what parse functions are needed
2431 sub NeededInterface($$)
2433 my ($interface,$needed) = @_;
2434 NeededFunction($_, $needed) foreach (@{$interface->{FUNCTIONS}});
2435 NeededTypedef($_, $needed) foreach (reverse @{$interface->{TYPES}});