diff options
author | Jelmer Vernooij <jelmer@samba.org> | 2005-03-17 12:12:57 +0000 |
---|---|---|
committer | Gerald (Jerry) Carter <jerry@samba.org> | 2007-10-10 13:11:05 -0500 |
commit | 795f59375721ac46b3d50297c782b0077b721870 (patch) | |
tree | 9ae361f31a416e683e1dd8dad88bf0cf694effbb /source4/build/pidl/ndr_parser.pm | |
parent | 85e0b90cf3de8f9cff68c0fdadd6c9b315d48d37 (diff) | |
download | samba-795f59375721ac46b3d50297c782b0077b721870.tar.gz samba-795f59375721ac46b3d50297c782b0077b721870.tar.bz2 samba-795f59375721ac46b3d50297c782b0077b721870.zip |
r5852: Rename ndr.pm to ndr_parser.pm
I'm going to add a ndr.pm later on that'll generate a
tree with necessary information for the two NDR backends
(eparser, ndr_parser) containing alignment info, etc.
(This used to be commit 5162daa9464cd64930f5a8fd0d7b381b122c931d)
Diffstat (limited to 'source4/build/pidl/ndr_parser.pm')
-rw-r--r-- | source4/build/pidl/ndr_parser.pm | 2275 |
1 files changed, 2275 insertions, 0 deletions
diff --git a/source4/build/pidl/ndr_parser.pm b/source4/build/pidl/ndr_parser.pm new file mode 100644 index 0000000000..fde9f05d2f --- /dev/null +++ b/source4/build/pidl/ndr_parser.pm @@ -0,0 +1,2275 @@ +################################################### +# Samba4 NDR parser generator for IDL structures +# Copyright tridge@samba.org 2000-2003 +# Copyright tpot@samba.org 2001 +# Copyright jelmer@samba.org 2004-2005 +# released under the GNU GPL + +package NdrParser; + +use strict; +use needed; +use typelist; + +# list of known types +my %typefamily; + +sub get_typefamily($) +{ + my $n = shift; + return $typefamily{$n}; +} + +my %scalar_alignments = +( + "char" => 1, + "int8" => 1, + "uint8" => 1, + "short" => 2, + "wchar_t" => 2, + "int16" => 2, + "uint16" => 2, + "long" => 4, + "int32" => 4, + "uint32" => 4, + "dlong" => 4, + "udlong" => 4, + "udlongr" => 4, + "NTTIME" => 4, + "NTTIME_1sec" => 4, + "time_t" => 4, + "DATA_BLOB" => 4, + "error_status_t" => 4, + "WERROR" => 4, + "NTSTATUS" => 4, + "boolean32" => 4, + "unsigned32" => 4, + "ipv4address" => 4, + "hyper" => 8, + "NTTIME_hyper" => 8 +); + +$typefamily{SCALAR} = { + ALIGN => sub { + my $t = shift; + return $scalar_alignments{$t->{NAME}}; + } +}; + +sub is_scalar_type($) +{ + my $type = shift; + + return 0 unless typelist::hasType($type); + + if (my $dt = typelist::getType($type)->{DATA}->{TYPE}) { + return 1 if ($dt eq "SCALAR" or $dt eq "ENUM" or $dt eq "BITMAP"); + } + + return 0; +} + +sub pointer_type($) +{ + my $e = shift; + + return undef unless $e->{POINTERS}; + + return "ref" if (util::has_property($e, "ref")); + return "ptr" if (util::has_property($e, "ptr")); + return "unique" if (util::has_property($e, "unique")); + return "relative" if (util::has_property($e, "relative")); + return "ignore" if (util::has_property($e, "ignore")); + + return undef; +} + +# return 1 if this is a fixed array +sub is_fixed_array($) +{ + my $e = shift; + my $len = $e->{"ARRAY_LEN"}; + return 1 if (defined $len && util::is_constant($len)); + return 0; +} + +# return 1 if this is a conformant array +sub is_conformant_array($) +{ + my $e = shift; + return 1 if (util::has_property($e, "size_is")); + return 0; +} + +# return 1 if this is a inline array +sub is_inline_array($) +{ + my $e = shift; + my $len = $e->{"ARRAY_LEN"}; + if (is_fixed_array($e) || + defined $len && $len ne "*") { + return 1; + } + return 0; +} + +# return 1 if this is a varying array +sub is_varying_array($) +{ + my $e = shift; + return util::has_property($e, "length_is"); +} + +# return 1 if this is a surrounding array (sometimes +# referred to as an embedded array). Can only occur as +# the last element in a struct and can not contain any pointers. +sub is_surrounding_array($) +{ + my $e = shift; + + return ($e->{POINTERS} == 0 + and defined $e->{ARRAY_LEN} + and $e->{ARRAY_LEN} eq "*" + and $e == $e->{PARENT}->{ELEMENTS}[-1] + and $e->{PARENT}->{TYPE} ne "FUNCTION"); +} + +sub array_type($) +{ + my $e = shift; + + return "conformant-varying" if (is_varying_array($e) and is_conformant_array($e)); + return "conformant" if (is_varying_array($e)); + return "varying" if (is_varying_array($e)); + return "inline" if (is_inline_array($e)); + return "fixed" if (is_fixed_array($e)); + + return undef; +} + +# determine if an element needs a reference pointer on the wire +# in its NDR representation +sub need_wire_pointer($) +{ + my $e = shift; + + my $n = $e->{POINTERS}; + my $pt = pointer_type($e); + + # Top level "ref" pointers do not have a referrent identifier + if ( defined($pt) + and $pt eq "ref" + and $e->{PARENT}->{TYPE} eq "FUNCTION") + { + $n--; + } + + return $n; +} + +# determine if an element needs a "buffers" section in NDR +sub need_buffers_section($) +{ + my $e = shift; + if (!can_contain_deferred($e) && + !util::array_size($e)) { + return 0; + } + return 1; +} + +# see if a variable needs to be allocated by the NDR subsystem on pull +sub need_alloc($) +{ + my $e = shift; + + return 0 if (util::has_property($e, "ref") && $e->{PARENT}->{TYPE} eq "FUNCTION"); + return 1 if ($e->{POINTERS} || util::array_size($e)); + return 0; +} + +# Prefix to get the actual value of a variable +sub c_ptr_prefix($) +{ + my $e = shift; + my $pointers = ""; + foreach my $i (need_wire_pointer($e)..$e->{POINTERS}-1) { $pointers.="*"; } + return $pointers; +} + +# determine the C prefix used to refer to a variable when passing to a push +# function. This will be '*' for pointers to scalar types, '' for scalar +# types and normal pointers and '&' for pass-by-reference structures +sub c_push_prefix($) +{ + my $e = shift; + + my $ret = ""; + + if ($e->{TYPE} =~ "string") { + $ret = ""; + } elsif (is_scalar_type($e->{TYPE}) and $e->{POINTERS} and + !util::array_size($e)) { + $ret .="*"; + } elsif (!is_scalar_type($e->{TYPE}) && + !$e->{POINTERS} && + !util::array_size($e)) { + return "&"; + } + + foreach my $i (2..$e->{POINTERS}) { $ret.="*"; } + + return $ret; +} + +# determine the C prefix used to refer to a variable when passing to a pull +# return '&' or '' +sub c_pull_prefix($) +{ + my $e = shift; + + if (!$e->{POINTERS} && !util::array_size($e)) { + return "&"; + } + + if ($e->{TYPE} =~ "string") { + return "&"; + } + + my $ret = ""; + foreach my $i (2..$e->{POINTERS}) { $ret.="*"; } + return $ret; +} +my $res = ""; +my $tabs = ""; +sub pidl($) +{ + my $d = shift; + if ($d) { + $res .= $tabs; + $res .= $d; + } + $res .="\n"; +} + +sub indent() +{ + $tabs .= "\t"; +} + +sub deindent() +{ + $tabs = substr($tabs, 0, -1); +} + +#################################################################### +# work out the name of a size_is() variable +sub ParseExpr($$$) +{ + my($e) = shift; + my($size) = shift; + my($var_prefix) = shift; + + my($fn) = $e->{PARENT}; + + return $size if (util::is_constant($size)); + + return $size if ($size =~ /ndr->|\(/); + + my $prefix = ""; + + if ($size =~ /\*(.*)/) { + $size = $1; + $prefix = "*"; + } + + if ($fn->{TYPE} ne "FUNCTION") { + return $prefix . "r->$size"; + } + + my $e2 = util::find_sibling($e, $size); + + die("Invalid sibling '$size'") unless defined($e2); + + if (util::has_property($e2, "in") && util::has_property($e2, "out")) { + return $prefix . "$var_prefix$size"; + } + + if (util::has_property($e2, "in")) { + return $prefix . "r->in.$size"; + } + + if (util::has_property($e2, "out")) { + return $prefix . "r->out.$size"; + } + + die "invalid variable in $size for element $e->{NAME} in $fn->{NAME}\n"; +} + +##################################################################### +# check that a variable we get from ParseExpr isn't a null pointer +sub check_null_pointer($) +{ + my $size = shift; + if ($size =~ /^\*/) { + my $size2 = substr($size, 1); + pidl "if ($size2 == NULL) return NT_STATUS_INVALID_PARAMETER_MIX;"; + } +} + +##################################################################### +# check that a variable we get from ParseExpr isn't a null pointer +# void return varient +sub check_null_pointer_void($) +{ + my $size = shift; + if ($size =~ /^\*/) { + my $size2 = substr($size, 1); + pidl "if ($size2 == NULL) return;"; + } +} + +##################################################################### +# work out is a parse function should be declared static or not +sub fn_prefix($) +{ + my $fn = shift; + + if ($fn->{TYPE} eq "TYPEDEF" or + $fn->{TYPE} eq "FUNCTION") { + return "" if (util::has_property($fn, "public")); + } + + return "static "; +} + +################################################################### +# setup any special flags for an element or structure +sub start_flags($) +{ + my $e = shift; + my $flags = util::has_property($e, "flag"); + if (defined $flags) { + pidl "{ uint32_t _flags_save_$e->{TYPE} = ndr->flags;"; + pidl "ndr_set_flags(&ndr->flags, $flags);"; + } +} + +################################################################### +# end any special flags for an element or structure +sub end_flags($) +{ + my $e = shift; + my $flags = util::has_property($e, "flag"); + if (defined $flags) { + pidl "ndr->flags = _flags_save_$e->{TYPE};\n\t}"; + } +} + +##################################################################### +# work out the correct alignment for a structure or union +sub find_largest_alignment($) +{ + my $s = shift; + + my $align = 1; + for my $e (@{$s->{ELEMENTS}}) { + my $a = 1; + + if (need_wire_pointer($e)) { + $a = 4; + } else { + $a = align_type($e->{TYPE}); + } + + $align = $a if ($align < $a); + } + + return $align; +} + +##################################################################### +# align a type +sub align_type +{ + my $e = shift; + + unless (typelist::hasType($e)) { + # it must be an external type - all we can do is guess + # print "Warning: assuming alignment of unknown type '$e' is 4\n"; + return 4; + } + + my $dt = typelist::getType($e)->{DATA}; + + my $tmp = $typefamily{$dt->{TYPE}}->{ALIGN}->($dt); + return $tmp; +} + +##################################################################### +# see if a type contains any deferred data +sub can_contain_deferred +{ + my $e = shift; + + return 1 if ($e->{POINTERS}); + return 0 if (is_scalar_type($e->{TYPE})); + return 0 if (util::has_property($e, "subcontext")); + return 1 unless (typelist::hasType($e->{TYPE})); # assume the worst + + my $type = typelist::getType($e->{TYPE}); + + foreach my $x (@{$type->{DATA}->{ELEMENTS}}) { + return 1 if (can_contain_deferred ($x)); + } + + return 0; +} + +##################################################################### +# parse array preceding data - push side +sub ParseArrayPushPreceding($$$) +{ + my $e = shift; + my $var_prefix = shift; + my $ndr_flags = shift; + + my $size = ParseExpr($e, util::array_size($e), $var_prefix); + + if (!is_inline_array($e)) { + # we need to emit the array size + pidl "NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, $size));"; + } +} + +##################################################################### +# parse the data of an array - push side +sub ParseArrayPush($$$$) +{ + my $e = shift; + my $ndr = shift; + my $var_prefix = shift; + my $ndr_flags = shift; + my $cprefix = c_push_prefix($e); + + my $size = ParseExpr($e, util::array_size($e), $var_prefix); + + # See whether the array size has been pushed yet + if (!is_surrounding_array($e)) { + ParseArrayPushPreceding($e, $var_prefix, $ndr_flags); + } + + if (is_varying_array($e)) { + my $length = util::has_property($e, "length_is"); + $length = ParseExpr($e, $length, $var_prefix); + pidl "NDR_CHECK(ndr_push_uint32($ndr, NDR_SCALARS, 0));"; + pidl "NDR_CHECK(ndr_push_uint32($ndr, NDR_SCALARS, $length));"; + $size = $length; + } + + if (is_scalar_type($e->{TYPE})) { + pidl "NDR_CHECK(ndr_push_array_$e->{TYPE}($ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}, $size));"; + } else { + pidl "NDR_CHECK(ndr_push_array($ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}, sizeof($cprefix$var_prefix$e->{NAME}\[0]), $size, (ndr_push_flags_fn_t)ndr_push_$e->{TYPE}));"; + } +} + +##################################################################### +# print an array +sub ParseArrayPrint($$) +{ + my $e = shift; + my $var_prefix = shift; + my $size = ParseExpr($e, util::array_size($e), $var_prefix); + my $cprefix = c_push_prefix($e); + + if (is_varying_array($e)) { + $size = ParseExpr($e, util::has_property($e, "length_is"), $var_prefix); + } + + if (is_scalar_type($e->{TYPE})) { + pidl "ndr_print_array_$e->{TYPE}(ndr, \"$e->{NAME}\", $cprefix$var_prefix$e->{NAME}, $size);"; + } else { + pidl "ndr_print_array(ndr, \"$e->{NAME}\", $cprefix$var_prefix$e->{NAME}, sizeof($cprefix$var_prefix$e->{NAME}\[0]), $size, (ndr_print_fn_t)ndr_print_$e->{TYPE});"; + } +} + +##################################################################### +# check the size_is and length_is constraints +sub CheckArraySizes($$) +{ + my $e = shift; + my $var_prefix = shift; + + if (is_conformant_array($e)) { + my $size = ParseExpr($e, util::array_size($e), $var_prefix); + pidl "if ($var_prefix$e->{NAME}) {"; + indent; + check_null_pointer($size); + pidl "NDR_CHECK(ndr_check_array_size(ndr, (void*)&$var_prefix$e->{NAME}, $size));"; + deindent; + pidl "}"; + } + + if (is_varying_array($e)) { + my $length = util::has_property($e, "length_is"); + $length = ParseExpr($e, $length, $var_prefix); + pidl "if ($var_prefix$e->{NAME}) {"; + indent; + check_null_pointer($length); + pidl "NDR_CHECK(ndr_check_array_length(ndr, (void*)&$var_prefix$e->{NAME}, $length));"; + deindent; + pidl "}" + } +} + +sub ParseArrayPullPreceding($$$) +{ + my $e = shift; + my $var_prefix = shift; + my $ndr_flags = shift; + + if (!is_inline_array($e)) { + # non fixed arrays encode the size just before the array + pidl "NDR_CHECK(ndr_pull_array_size(ndr, &$var_prefix$e->{NAME}));"; + } +} + +##################################################################### +# parse an array - pull side +sub ParseArrayPull($$$$) +{ + my $e = shift; + my $ndr = shift; + my $var_prefix = shift; + my $ndr_flags = shift; + + my $cprefix = c_pull_prefix($e); + my $length = ParseExpr($e, util::array_size($e), $var_prefix); + my $size = $length; + + if (is_conformant_array($e)) { + $length = $size = "ndr_get_array_size($ndr, &$var_prefix$e->{NAME})"; + } + + # if this is a conformant array then we use that size to allocate, and make sure + # we allocate enough to pull the elements + if (!is_inline_array($e) and not is_surrounding_array($e)) { + if ($var_prefix =~ /^r->out/ && $length =~ /^\*r->in/) { + my $length2 = substr($length, 1); + pidl "if (ndr->flags & LIBNDR_FLAG_REF_ALLOC) { NDR_ALLOC($ndr, $length2); }"; + } + + ParseArrayPullPreceding($e, $var_prefix, $ndr_flags); + } + + if (is_varying_array($e)) { + pidl "NDR_CHECK(ndr_pull_array_length($ndr, &$var_prefix$e->{NAME}));"; + $length = "ndr_get_array_length($ndr, &$var_prefix$e->{NAME})"; + } + + check_null_pointer($length); + + if ($length ne $size) { + pidl "if ($length > $size) {"; + indent; + pidl "return ndr_pull_error($ndr, NDR_ERR_CONFORMANT_SIZE, \"Bad conformant size %u should be %u\", $size, $length);"; + deindent; + pidl "}"; + } + + if ((need_alloc($e) && !is_fixed_array($e)) || + ($var_prefix eq "r->in." && util::has_property($e, "ref"))) { + if (!is_inline_array($e) || $ndr_flags eq "NDR_SCALARS") { + pidl "NDR_ALLOC_N($ndr, $var_prefix$e->{NAME}, $size);"; + } + } + + if (($var_prefix eq "r->out." && util::has_property($e, "ref"))) { + if (!is_inline_array($e) || $ndr_flags eq "NDR_SCALARS") { + pidl "if ($ndr->flags & LIBNDR_FLAG_REF_ALLOC) {"; + pidl "\tNDR_ALLOC_N($ndr, $var_prefix$e->{NAME}, $size);"; + pidl "}"; + } + } + + if (is_scalar_type($e->{TYPE})) { + pidl "NDR_CHECK(ndr_pull_array_$e->{TYPE}($ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}, $length));"; + } else { + pidl "NDR_CHECK(ndr_pull_array($ndr, $ndr_flags, (void **)$cprefix$var_prefix$e->{NAME}, sizeof($cprefix$var_prefix$e->{NAME}\[0]), $length, (ndr_pull_flags_fn_t)ndr_pull_$e->{TYPE}));"; + } +} + +sub ParseSubcontextPushStart($$) +{ + my $e = shift; + my $ndr_flags = shift; + + pidl "if (($ndr_flags) & NDR_SCALARS) {"; + indent; + pidl "struct ndr_push *_ndr_$e->{NAME};"; + pidl ""; + pidl "_ndr_$e->{NAME} = ndr_push_init_ctx(ndr);"; + pidl "if (!_ndr_$e->{NAME}) return NT_STATUS_NO_MEMORY;"; + pidl "_ndr_$e->{NAME}->flags = ndr->flags;"; + pidl ""; + + return "_ndr_$e->{NAME}"; +} + +sub ParseSubcontextPushEnd($) +{ + my $e = shift; + my $header_size = util::has_property($e, "subcontext"); + my $size_is = util::has_property($e, "subcontext_size"); + + if (not defined($size_is)) { + $size_is = "-1"; + } + + pidl "NDR_CHECK(ndr_push_subcontext_header(ndr, $header_size, $size_is, _ndr_$e->{NAME}));"; + pidl "NDR_CHECK(ndr_push_bytes(ndr, _ndr_$e->{NAME}->data, _ndr_$e->{NAME}->offset));"; + deindent; + pidl "}"; +} + +sub ParseSubcontextPullStart($$) +{ + my $e = shift; + my $ndr_flags = shift; + my $header_size = util::has_property($e, "subcontext"); + my $size_is = util::has_property($e, "subcontext_size"); + + if (not defined($size_is)) { + $size_is = "-1"; + } + + pidl "if (($ndr_flags) & NDR_SCALARS) {"; + indent; + pidl "struct ndr_pull *_ndr_$e->{NAME};"; + pidl "NDR_ALLOC(ndr, _ndr_$e->{NAME});"; + pidl "NDR_CHECK(ndr_pull_subcontext_header(ndr, $header_size, $size_is, _ndr_$e->{NAME}));"; + return "_ndr_$e->{NAME}"; +} + +sub ParseSubcontextPullEnd($) +{ + my $e = shift; + my $header_size = util::has_property($e, "subcontext"); + my $size_is = util::has_property($e, "subcontext_size"); + + my $advance; + if (defined ($size_is)) { + $advance = "$size_is"; + } elsif ($header_size) { + $advance = "_ndr_$e->{NAME}->data_size"; + } else { + $advance = "_ndr_$e->{NAME}->offset"; + } + pidl "NDR_CHECK(ndr_pull_advance(ndr, $advance));"; + deindent; + pidl "}"; +} + +##################################################################### +# parse scalars in a structure element +sub ParseElementPushScalar($$$) +{ + my($e) = shift; + my($var_prefix) = shift; + my($ndr_flags) = shift; + my $cprefix = c_push_prefix($e); + my $ptr_prefix = c_ptr_prefix($e); + my $sub_size = util::has_property($e, "subcontext"); + my $ndr = "ndr"; + + start_flags($e); + + if (my $value = util::has_property($e, "value")) { + pidl "$cprefix$var_prefix$e->{NAME} = $value;"; + } + + if (defined $sub_size and $e->{POINTERS} == 0) { + $ndr = ParseSubcontextPushStart($e, "NDR_SCALARS"); + } + + if (need_wire_pointer($e)) { + ParsePtrPush($e, $ptr_prefix.$var_prefix); + } elsif (is_inline_array($e)) { + ParseArrayPush($e, $ndr, "r->", "NDR_SCALARS"); + } elsif (need_alloc($e)) { + # no scalar component + } else { + if (my $switch = util::has_property($e, "switch_is")) { + ParseSwitchPush($e, $ndr, $var_prefix, $ndr_flags, $switch); + } + + pidl "NDR_CHECK(ndr_push_$e->{TYPE}($ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}));"; + } + + if (defined $sub_size and $e->{POINTERS} == 0) { + ParseSubcontextPushEnd($e); + } + + end_flags($e); +} + +##################################################################### +# parse a pointer in a struct element or function +sub ParsePtrPush($$) +{ + my $e = shift; + my $var_prefix = shift; + + if (util::has_property($e, "ref")) { + pidl "NDR_CHECK(ndr_push_ref_ptr(ndr, $var_prefix$e->{NAME}));"; + } elsif (util::has_property($e, "relative")) { + pidl "NDR_CHECK(ndr_push_relative_ptr1(ndr, $var_prefix$e->{NAME}));"; + } else { + pidl "NDR_CHECK(ndr_push_unique_ptr(ndr, $var_prefix$e->{NAME}));"; + } +} + +##################################################################### +# print scalars in a structure element +sub ParseElementPrint($$) +{ + my($e) = shift; + my($var_prefix) = shift; + my $cprefix = c_push_prefix($e); + my $ptr_prefix = c_ptr_prefix($e); + + return if (util::has_property($e, "noprint")); + + if (my $value = util::has_property($e, "value")) { + pidl "if (ndr->flags & LIBNDR_PRINT_SET_VALUES) {"; + pidl "\t$cprefix$var_prefix$e->{NAME} = $value;"; + pidl "}"; + } + + my $l = $e->{POINTERS}; + $l++ if (util::array_size($e) and $l == 0 and !is_fixed_array($e)); + + foreach my $i (1..$l) { + pidl "ndr_print_ptr(ndr, \"$e->{NAME}\", $var_prefix$e->{NAME});"; + pidl "ndr->depth++;"; + if ($i > $l-need_wire_pointer($e)) { + pidl "if ($ptr_prefix$var_prefix$e->{NAME}) {"; + indent; + } + } + + if (util::array_size($e)) { + ParseArrayPrint($e, $var_prefix) + } elsif (my $switch = util::has_property($e, "switch_is")) { + my $switch_var = ParseExpr($e, $switch, $var_prefix); + check_null_pointer_void($switch_var); + + pidl "ndr_print_$e->{TYPE}(ndr, \"$e->{NAME}\", $switch_var, $cprefix$var_prefix$e->{NAME});"; + } else { + pidl "ndr_print_$e->{TYPE}(ndr, \"$e->{NAME}\", $cprefix$var_prefix$e->{NAME});"; + } + + foreach my $i (1..$l) { + if ($i > $l-need_wire_pointer($e)) { + deindent; + pidl "}"; + } + pidl "ndr->depth--;"; + } +} + +##################################################################### +# parse scalars in a structure element - pull size +sub ParseSwitchPull($$$$$) +{ + my($e) = shift; + my $ndr = shift; + my($var_prefix) = shift; + my($ndr_flags) = shift; + my $switch = shift; + my $switch_var = ParseExpr($e, $switch, $var_prefix); + + my $cprefix = c_pull_prefix($e); + + my $utype = typelist::getType($e->{TYPE}); + + check_null_pointer($switch_var); + + pidl "NDR_CHECK(ndr_pull_set_switch_value($ndr, $cprefix$var_prefix$e->{NAME}, $switch_var));"; + +} + +##################################################################### +# push switch element +sub ParseSwitchPush($$$$$) +{ + my($e) = shift; + my $ndr = shift; + my($var_prefix) = shift; + my($ndr_flags) = shift; + my $switch = shift; + my $switch_var = ParseExpr($e, $switch, $var_prefix); + my $cprefix = c_push_prefix($e); + + check_null_pointer($switch_var); + + pidl "NDR_CHECK(ndr_push_set_switch_value($ndr, $cprefix$var_prefix$e->{NAME}, $switch_var));"; + +} + +##################################################################### +# parse scalars in a structure element - pull size +sub ParseElementPullScalar($$$) +{ + my($e) = shift; + my($var_prefix) = shift; + my($ndr_flags) = shift; + my $cprefix = c_pull_prefix($e); + my $ptr_prefix = c_ptr_prefix($e); + my $sub_size = util::has_property($e, "subcontext"); + my $ndr = "ndr"; + + start_flags($e); + + if (defined $sub_size && $e->{POINTERS} == 0) { + $ndr = ParseSubcontextPullStart($e, $ndr_flags); + $ndr_flags = "NDR_SCALARS|NDR_BUFFERS"; + } + + if (is_inline_array($e)) { + ParseArrayPull($e, $ndr, "r->", "NDR_SCALARS"); + } elsif (need_wire_pointer($e)) { + ParsePtrPull($e, $ptr_prefix.$var_prefix); + } elsif (is_surrounding_array($e)) { + } else { + if (my $switch = util::has_property($e, "switch_is")) { + ParseSwitchPull($e, $ndr, $var_prefix, $ndr_flags, $switch); + } + + pidl "NDR_CHECK(ndr_pull_$e->{TYPE}($ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}));"; + } + + if (my $range = util::has_property($e, "range")) { + my ($low, $high) = split(/ /, $range, 2); + pidl "if ($var_prefix$e->{NAME} < $low || $var_prefix$e->{NAME} > $high) {"; + pidl "\treturn ndr_pull_error($ndr, NDR_ERR_RANGE, \"value out of range\");"; + pidl "}"; + } + + if (defined $sub_size && $e->{POINTERS} == 0) { + ParseSubcontextPullEnd($e); + } + + end_flags($e); +} + +##################################################################### +# parse a pointer in a struct element or function +sub ParsePtrPull($$) +{ + my($e) = shift; + my($var_prefix) = shift; + + if (util::has_property($e, "ref")) { + pidl "NDR_CHECK(ndr_pull_ref_ptr(ndr, &_ptr_$e->{NAME}));"; + } else { + pidl "NDR_CHECK(ndr_pull_unique_ptr(ndr, &_ptr_$e->{NAME}));"; + } + pidl "if (_ptr_$e->{NAME}) {"; + indent; + pidl "NDR_ALLOC(ndr, $var_prefix$e->{NAME});"; + if (util::has_property($e, "relative")) { + pidl "NDR_CHECK(ndr_pull_relative_ptr1(ndr, $var_prefix$e->{NAME}, _ptr_$e->{NAME}));"; + } + deindent; + pidl "} else {"; + pidl "\t$var_prefix$e->{NAME} = NULL;"; + pidl "}"; +} + +##################################################################### +# parse buffers in a structure element +sub ParseElementPushBuffer($$) +{ + my($e) = shift; + my($var_prefix) = shift; + my $cprefix = c_push_prefix($e); + my $sub_size = util::has_property($e, "subcontext"); + my $ndr = "ndr"; + + return unless (need_buffers_section($e)); + + start_flags($e); + + my $pointers = c_ptr_prefix($e); + for my $i (1..need_wire_pointer($e)) { + if ($i > 1) { + ParsePtrPush($e,$pointers.$var_prefix); + } + pidl "if ($pointers$var_prefix$e->{NAME}) {"; + indent; + $pointers.="*"; + } + + if (util::has_property($e, "relative")) { + pidl "NDR_CHECK(ndr_push_relative_ptr2(ndr, $var_prefix$e->{NAME}));"; + } + + my $ndr_flags = "NDR_BUFFERS"; + if ($e->{POINTERS} || (util::array_size($e) && !is_inline_array($e))) + { + $ndr_flags="NDR_SCALARS|$ndr_flags" + } + + if (defined $sub_size) { + $ndr = ParseSubcontextPushStart($e, $ndr_flags); + $ndr_flags = "NDR_SCALARS|NDR_BUFFERS"; + } + + if (util::array_size($e)) { + ParseArrayPush($e, $ndr, "r->", $ndr_flags); + } else { + if (my $switch = util::has_property($e, "switch_is")) { + ParseSwitchPush($e, $ndr, $var_prefix, $ndr_flags, $switch); + } + + pidl "NDR_CHECK(ndr_push_$e->{TYPE}($ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}));"; + } + + if (defined $sub_size) { + ParseSubcontextPushEnd($e); + } + + for my $i (1..need_wire_pointer($e)) { + deindent; + pidl "}"; + } + + end_flags($e); +} + +##################################################################### +# parse buffers in a structure element - pull side +sub ParseElementPullBuffer($$) +{ + my($e) = shift; + my($var_prefix) = shift; + my $cprefix = c_pull_prefix($e); + my $sub_size = util::has_property($e, "subcontext"); + my $ndr = "ndr"; + + return unless (need_buffers_section($e)); + + start_flags($e); + + my $pointers = c_ptr_prefix($e); + for my $i (1..need_wire_pointer($e)) { + if ($i > 1) { + ParsePtrPull($e,$pointers.$var_prefix); + } + pidl "if ($pointers$var_prefix$e->{NAME}) {"; + indent; + $pointers.="*"; + } + + if (util::has_property($e, "relative")) { + pidl "struct ndr_pull_save _relative_save;"; + pidl "ndr_pull_save(ndr, &_relative_save);"; + pidl "NDR_CHECK(ndr_pull_relative_ptr2(ndr, $var_prefix$e->{NAME}));"; + } + + my $ndr_flags = "NDR_BUFFERS"; + if ($e->{POINTERS} || (util::array_size($e) && !is_inline_array($e))) + { + $ndr_flags="NDR_SCALARS|$ndr_flags" + } + + if (defined $sub_size) { + $ndr = ParseSubcontextPullStart($e, $ndr_flags); + $ndr_flags = "NDR_SCALARS|NDR_BUFFERS"; + } + + if (util::array_size($e)) { + ParseArrayPull($e, $ndr, "r->", $ndr_flags); + } else { + if (my $switch = util::has_property($e, "switch_is")) { + ParseSwitchPull($e, $ndr, $var_prefix, $ndr_flags, $switch); + } + + pidl "NDR_CHECK(ndr_pull_$e->{TYPE}($ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}));"; + } + + if (defined $sub_size) { + ParseSubcontextPullEnd($e); + } + + if (util::has_property($e, "relative")) { + pidl "ndr_pull_restore(ndr, &_relative_save);"; + } + + for my $i (1..need_wire_pointer($e)) { + deindent; + pidl "}"; + } + + end_flags($e); +} + +##################################################################### +# parse a struct +sub ParseStructPush($) +{ + my($struct) = shift; + + return unless defined($struct->{ELEMENTS}); + + start_flags($struct); + + # see if the structure contains a conformant array. If it + # does, then it must be the last element of the structure, and + # we need to push the conformant length early, as it fits on + # the wire before the structure (and even before the structure + # alignment) + my $e = $struct->{ELEMENTS}[-1]; + if (is_conformant_array($e) and is_surrounding_array($e)) { + ParseArrayPushPreceding($e, "r->", "NDR_SCALARS"); + } + + if (defined $e->{TYPE} && $e->{TYPE} eq "string" + && util::property_matches($e, "flag", ".*LIBNDR_FLAG_STR_CONFORMANT.*")) { + pidl "NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, ndr_string_array_size(ndr, r->$e->{NAME})));"; + } + + pidl "if (!(ndr_flags & NDR_SCALARS)) goto buffers;"; + + pidl "NDR_CHECK(ndr_push_struct_start(ndr));"; + + my $align = find_largest_alignment($struct); + pidl "NDR_CHECK(ndr_push_align(ndr, $align));"; + + foreach my $e (@{$struct->{ELEMENTS}}) { + ParseElementPushScalar($e, "r->", "NDR_SCALARS"); + } + + pidl "buffers:"; + pidl "if (!(ndr_flags & NDR_BUFFERS)) goto done;"; + foreach my $e (@{$struct->{ELEMENTS}}) { + ParseElementPushBuffer($e, "r->"); + } + + pidl "ndr_push_struct_end(ndr);"; + + pidl "done:"; + + end_flags($struct); +} + +##################################################################### +# generate a push function for an enum +sub ParseEnumPush($) +{ + my($enum) = shift; + my($type_fn) = typelist::enum_type_fn($enum); + + start_flags($enum); + + pidl "NDR_CHECK(ndr_push_$type_fn(ndr, NDR_SCALARS, r));"; + + end_flags($enum); +} + +##################################################################### +# generate a pull function for an enum +sub ParseEnumPull($) +{ + my($enum) = shift; + my($type_fn) = typelist::enum_type_fn($enum); + my($type_v_decl) = typelist::mapScalarType(typelist::enum_type_fn($enum)); + + pidl "$type_v_decl v;"; + start_flags($enum); + pidl "NDR_CHECK(ndr_pull_$type_fn(ndr, NDR_SCALARS, &v));"; + pidl "*r = v;"; + + end_flags($enum); +} + +##################################################################### +# generate a print function for an enum +sub ParseEnumPrint($) +{ + my($enum) = shift; + + pidl "const char *val = NULL;"; + pidl ""; + + start_flags($enum); + + pidl "switch (r) {"; + indent; + my $els = \@{$enum->{ELEMENTS}}; + foreach my $i (0 .. $#{$els}) { + my $e = ${$els}[$i]; + chomp $e; + if ($e =~ /^(.*)=/) { + $e = $1; + } + pidl "case $e: val = \"$e\"; break;"; + } + + deindent; + pidl "}"; + + pidl "ndr_print_enum(ndr, name, \"$enum->{TYPE}\", val, r);"; + + end_flags($enum); +} + +sub ArgsEnumPush($) +{ + my $e = shift; + return "struct ndr_push *ndr, int ndr_flags, enum $e->{NAME} r"; +} + +sub ArgsEnumPrint($) +{ + my $e = shift; + return "struct ndr_print *ndr, const char *name, enum $e->{NAME} r"; +} + +sub ArgsEnumPull($) +{ + my $e = shift; + return "struct ndr_pull *ndr, int ndr_flags, enum $e->{NAME} *r"; +} + +$typefamily{ENUM} = { + PUSH_FN_BODY => \&ParseEnumPush, + PUSH_FN_ARGS => \&ArgsEnumPush, + PULL_FN_BODY => \&ParseEnumPull, + PULL_FN_ARGS => \&ArgsEnumPull, + PRINT_FN_BODY => \&ParseEnumPrint, + PRINT_FN_ARGS => \&ArgsEnumPrint, + ALIGN => sub { return align_type(typelist::enum_type_fn(shift)); } +}; + +##################################################################### +# generate a push function for a bitmap +sub ParseBitmapPush($) +{ + my($bitmap) = shift; + my($type_fn) = typelist::bitmap_type_fn($bitmap); + + start_flags($bitmap); + + pidl "NDR_CHECK(ndr_push_$type_fn(ndr, NDR_SCALARS, r));"; + + end_flags($bitmap); +} + +##################################################################### +# generate a pull function for an bitmap +sub ParseBitmapPull($) +{ + my($bitmap) = shift; + my($type_fn) = typelist::bitmap_type_fn($bitmap); + my($type_decl) = typelist::mapType($bitmap); + + pidl "$type_decl v;"; + start_flags($bitmap); + pidl "NDR_CHECK(ndr_pull_$type_fn(ndr, NDR_SCALARS, &v));"; + pidl "*r = v;"; + + end_flags($bitmap); +} + +##################################################################### +# generate a print function for an bitmap +sub ParseBitmapPrintElement($$) +{ + my($e) = shift; + my($bitmap) = shift; + my($type_decl) = typelist::mapType($bitmap); + my($type_fn) = typelist::bitmap_type_fn($bitmap); + my($name) = $bitmap->{PARENT}->{NAME}; + my($flag); + + if ($e =~ /^(\w+) .*$/) { + $flag = "$1"; + } else { + die "Bitmap: \"$name\" invalid Flag: \"$e\""; + } + + pidl "ndr_print_bitmap_flag(ndr, sizeof($type_decl), \"$flag\", $flag, r);"; +} + +##################################################################### +# generate a print function for an bitmap +sub ParseBitmapPrint($) +{ + my($bitmap) = shift; + my($type_decl) = typelist::mapType($bitmap); + my($type_fn) = typelist::bitmap_type_fn($bitmap); + + start_flags($bitmap); + + pidl "ndr_print_$type_fn(ndr, name, r);"; + + pidl "ndr->depth++;"; + foreach my $e (@{$bitmap->{ELEMENTS}}) { + ParseBitmapPrintElement($e, $bitmap); + } + pidl "ndr->depth--;"; + + end_flags($bitmap); +} + +sub ArgsBitmapPush($) +{ + my $e = shift; + my $type_decl = typelist::mapType($e->{DATA}); + return "struct ndr_push *ndr, int ndr_flags, $type_decl r"; +} + +sub ArgsBitmapPrint($) +{ + my $e = shift; + my $type_decl = typelist::mapType($e->{DATA}); + return "struct ndr_print *ndr, const char *name, $type_decl r"; +} + +sub ArgsBitmapPull($) +{ + my $e = shift; + my $type_decl = typelist::mapType($e->{DATA}); + return "struct ndr_pull *ndr, int ndr_flags, $type_decl *r"; +} + +$typefamily{BITMAP} = { + PUSH_FN_BODY => \&ParseBitmapPush, + PUSH_FN_ARGS => \&ArgsBitmapPush, + PULL_FN_BODY => \&ParseBitmapPull, + PULL_FN_ARGS => \&ArgsBitmapPull, + PRINT_FN_BODY => \&ParseBitmapPrint, + PRINT_FN_ARGS => \&ArgsBitmapPrint, + ALIGN => sub { return align_type(typelist::bitmap_type_fn(shift)); } +}; + +##################################################################### +# generate a struct print function +sub ParseStructPrint($) +{ + my($struct) = shift; + my($name) = $struct->{PARENT}->{NAME}; + + return unless defined $struct->{ELEMENTS}; + + pidl "ndr_print_struct(ndr, name, \"$name\");"; + + start_flags($struct); + + pidl "ndr->depth++;"; + foreach my $e (@{$struct->{ELEMENTS}}) { + ParseElementPrint($e, "r->"); + } + pidl "ndr->depth--;"; + + end_flags($struct); +} + +##################################################################### +# parse a struct - pull side +sub ParseStructPull($) +{ + my($struct) = shift; + my $conform_e; + + return unless defined $struct->{ELEMENTS}; + + # see if the structure contains a conformant array. If it + # does, then it must be the last element of the structure, and + # we need to pull the conformant length early, as it fits on + # the wire before the structure (and even before the structure + # alignment) + my $e = $struct->{ELEMENTS}[-1]; + if (is_conformant_array($e) and is_surrounding_array($e)) { + $conform_e = $e; + } + + if (defined $e->{TYPE} && $e->{TYPE} eq "string" + && util::property_matches($e, "flag", ".*LIBNDR_FLAG_STR_CONFORMANT.*")) { + $conform_e = $e; + } + + # declare any internal pointers we need + foreach my $e (@{$struct->{ELEMENTS}}) { + if (need_wire_pointer($e)) { + pidl "uint32_t _ptr_$e->{NAME};"; + } + } + + start_flags($struct); + + pidl "if (!(ndr_flags & NDR_SCALARS)) goto buffers;"; + + pidl "NDR_CHECK(ndr_pull_struct_start(ndr));"; + + if (defined $conform_e) { + ParseArrayPullPreceding($conform_e, "r->", "NDR_SCALARS"); + } + + my $align = find_largest_alignment($struct); + pidl "NDR_CHECK(ndr_pull_align(ndr, $align));"; + + foreach my $e (@{$struct->{ELEMENTS}}) { + ParseElementPullScalar($e, "r->", "NDR_SCALARS"); + } + + pidl "buffers:\n"; + pidl "if (!(ndr_flags & NDR_BUFFERS)) goto done;"; + foreach my $e (@{$struct->{ELEMENTS}}) { + ParseElementPullBuffer($e, "r->"); + } + + foreach my $e (@{$struct->{ELEMENTS}}) { + CheckArraySizes($e, "r->"); + } + + pidl "ndr_pull_struct_end(ndr);"; + + pidl "done:"; + + end_flags($struct); +} + +##################################################################### +# calculate size of ndr struct +sub ParseStructNdrSize($) +{ + my $t = shift; + my $static = fn_prefix($t); + my $sizevar; + + if (my $flags = util::has_property($t, "flag")) { + pidl "flags |= $flags;"; + } + pidl "return ndr_size_struct(r, flags, (ndr_push_flags_fn_t)ndr_push_$t->{NAME});"; +} + +sub ArgsStructPush($) +{ + my $e = shift; + return "struct ndr_push *ndr, int ndr_flags, struct $e->{NAME} *r"; +} + +sub ArgsStructPrint($) +{ + my $e = shift; + return "struct ndr_print *ndr, const char *name, struct $e->{NAME} *r"; +} + +sub ArgsStructPull($) +{ + my $e = shift; + return "struct ndr_pull *ndr, int ndr_flags, struct $e->{NAME} *r"; +} + +sub ArgsStructNdrSize($) +{ + my $d = shift; + return "const struct $d->{NAME} *r, int flags"; +} + +$typefamily{STRUCT} = { + PUSH_FN_BODY => \&ParseStructPush, + PUSH_FN_ARGS => \&ArgsStructPush, + PULL_FN_BODY => \&ParseStructPull, + PULL_FN_ARGS => \&ArgsStructPull, + PRINT_FN_BODY => \&ParseStructPrint, + PRINT_FN_ARGS => \&ArgsStructPrint, + SIZE_FN_BODY => \&ParseStructNdrSize, + SIZE_FN_ARGS => \&ArgsStructNdrSize, + ALIGN => \&find_largest_alignment +}; + +##################################################################### +# calculate size of ndr struct +sub ParseUnionNdrSize($) +{ + my $t = shift; + my $static = fn_prefix($t); + my $sizevar; + + if (my $flags = util::has_property($t, "flag")) { + pidl "flags |= $flags;"; + } + pidl "return ndr_size_union(r, flags, level, (ndr_push_union_fn_t)ndr_push_$t->{NAME});"; +} + +##################################################################### +# parse a union - push side +sub ParseUnionPush($) +{ + my $e = shift; + my $have_default = 0; + + pidl "int level;"; + + start_flags($e); + + pidl "level = ndr_push_get_switch_value(ndr, r);"; + + pidl "if (!(ndr_flags & NDR_SCALARS)) goto buffers;"; + + if (!util::has_property($e, "nodiscriminant")) { + my $switch_type = util::has_property($e, "switch_type"); + $switch_type = "uint32" unless (defined ($switch_type)); + pidl "NDR_CHECK(ndr_push_$switch_type(ndr, NDR_SCALARS, level));"; + } + + pidl "NDR_CHECK(ndr_push_struct_start(ndr));"; + +# my $align = union_alignment($e); +# pidl "NDR_CHECK(ndr_push_align(ndr, $align));"; + + pidl "switch (level) {"; + indent; + foreach my $el (@{$e->{ELEMENTS}}) { + if (util::has_property($el, "default")) { + pidl "default:"; + $have_default = 1; + } else { + pidl "case $el->{PROPERTIES}->{case}:"; + + } + if ($el->{TYPE} ne "EMPTY") { + indent; + ParseElementPushScalar($el, "r->", "NDR_SCALARS"); + deindent; + } + pidl "break;"; + pidl ""; + } + if (! $have_default) { + pidl "default:"; + pidl "\treturn ndr_push_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);"; + } + deindent; + pidl "}"; + pidl "buffers:"; + pidl "if (!(ndr_flags & NDR_BUFFERS)) goto done;"; + pidl "switch (level) {"; + indent; + foreach my $el (@{$e->{ELEMENTS}}) { + if (util::has_property($el, "default")) { + pidl "default:"; + } else { + pidl "case $el->{PROPERTIES}->{case}:"; + } + if ($el->{TYPE} ne "EMPTY") { + indent; + ParseElementPushBuffer($el, "r->"); + deindent; + } + pidl "break;"; + pidl ""; + } + if (! $have_default) { + pidl "default:"; + pidl "\treturn ndr_push_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);"; + } + deindent; + pidl "}"; + pidl "ndr_push_struct_end(ndr);"; + pidl "done:"; + end_flags($e); +} + +##################################################################### +# print a union +sub ParseUnionPrint($) +{ + my $e = shift; + my $have_default = 0; + my($name) = $e->{PARENT}->{NAME}; + + pidl "ndr_print_union(ndr, name, level, \"$name\");"; + start_flags($e); + + pidl "switch (level) {"; + indent; + foreach my $el (@{$e->{ELEMENTS}}) { + if (util::has_property($el, "default")) { + $have_default = 1; + pidl "default:"; + } else { + pidl "case $el->{PROPERTIES}->{case}:"; + } + if ($el->{TYPE} ne "EMPTY") { + indent; + ParseElementPrint($el, "r->"); + deindent; + } + pidl "break;"; + pidl ""; + } + if (! $have_default) { + pidl "default:"; + pidl "\tndr_print_bad_level(ndr, name, level);"; + } + deindent; + pidl "}"; + + end_flags($e); +} + +##################################################################### +# parse a union - pull side +sub ParseUnionPull($) +{ + my $e = shift; + my $have_default = 0; + my $switch_type = util::has_property($e, "switch_type"); + $switch_type = "uint32" unless defined($switch_type); + + pidl "int level;"; + if (!util::has_property($e, "nodiscriminant")) { + if (typelist::typeIs($switch_type, "ENUM")) { + $switch_type = typelist::enum_type_fn(typelist::getType($switch_type)); + } + pidl typelist::mapScalarType($switch_type) . " _level;"; + } + + start_flags($e); + + pidl "level = ndr_pull_get_switch_value(ndr, r);"; + + pidl "if (!(ndr_flags & NDR_SCALARS)) goto buffers;"; + + if (!util::has_property($e, "nodiscriminant")) { + pidl "NDR_CHECK(ndr_pull_$switch_type(ndr, NDR_SCALARS, &_level));"; + pidl "if (_level != level) {"; + pidl "\treturn ndr_pull_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value %u for $e->{PARENT}->{NAME}\", _level);"; + pidl "}"; + } + + pidl "NDR_CHECK(ndr_pull_struct_start(ndr));"; + +# my $align = union_alignment($e); +# pidl "\tNDR_CHECK(ndr_pull_align(ndr, $align));\n"; + + pidl "switch (level) {"; + indent; + foreach my $el (@{$e->{ELEMENTS}}) { + if (util::has_property($el, "default")) { + pidl "default: {"; + $have_default = 1; + } else { + pidl "case $el->{PROPERTIES}->{case}: {"; + } + if ($el->{TYPE} ne "EMPTY") { + indent; + if ($el->{POINTERS}) { + pidl "uint32_t _ptr_$el->{NAME};"; + } + ParseElementPullScalar($el, "r->", "NDR_SCALARS"); + deindent; + } + pidl "break; }"; + pidl ""; + } + if (! $have_default) { + pidl "default:"; + pidl "\treturn ndr_pull_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);"; + } + deindent; + pidl "}"; + pidl "buffers:"; + pidl "if (!(ndr_flags & NDR_BUFFERS)) goto done;"; + pidl "switch (level) {"; + indent; + foreach my $el (@{$e->{ELEMENTS}}) { + if (util::has_property($el, "default")) { + pidl "default:"; + } else { + pidl "case $el->{PROPERTIES}->{case}:"; + } + if ($el->{TYPE} ne "EMPTY") { + indent; + ParseElementPullBuffer($el, "r->"); + deindent; + } + pidl "break;"; + pidl ""; + } + if (! $have_default) { + pidl "default:"; + pidl "\treturn ndr_pull_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);"; + } + deindent; + pidl "}"; + pidl "ndr_pull_struct_end(ndr);"; + pidl "done:"; + end_flags($e); +} + +sub ArgsUnionPush($) +{ + my $e = shift; + return "struct ndr_push *ndr, int ndr_flags, union $e->{NAME} *r"; +} + +sub ArgsUnionPrint($) +{ + my $e = shift; + return "struct ndr_print *ndr, const char *name, int level, union $e->{NAME} *r"; +} + +sub ArgsUnionPull($) +{ + my $e = shift; + return "struct ndr_pull *ndr, int ndr_flags, union $e->{NAME} *r"; +} + +sub ArgsUnionNdrSize($) +{ + my $d = shift; + return "const union $d->{NAME} *r, uint32_t level, int flags"; +} + +$typefamily{UNION} = { + PUSH_FN_BODY => \&ParseUnionPush, + PUSH_FN_ARGS => \&ArgsUnionPush, + PULL_FN_BODY => \&ParseUnionPull, + PULL_FN_ARGS => \&ArgsUnionPull, + PRINT_FN_BODY => \&ParseUnionPrint, + PRINT_FN_ARGS => \&ArgsUnionPrint, + SIZE_FN_ARGS => \&ArgsUnionNdrSize, + SIZE_FN_BODY => \&ParseUnionNdrSize, + ALIGN => \&find_largest_alignment +}; + +##################################################################### +# parse a typedef - push side +sub ParseTypedefPush($) +{ + my($e) = shift; + my $static = fn_prefix($e); + + return unless needed::is_needed("push_$e->{NAME}"); + + my $args = $typefamily{$e->{DATA}->{TYPE}}->{PUSH_FN_ARGS}->($e); + pidl $static . "NTSTATUS ndr_push_$e->{NAME}($args)"; + + pidl "{"; + indent; + $typefamily{$e->{DATA}->{TYPE}}->{PUSH_FN_BODY}->($e->{DATA}); + pidl "return NT_STATUS_OK;"; + deindent; + pidl "}"; + pidl "";; +} + + +##################################################################### +# parse a typedef - pull side +sub ParseTypedefPull($) +{ + my($e) = shift; + my $static = fn_prefix($e); + + return unless needed::is_needed("pull_$e->{NAME}"); + + my $args = $typefamily{$e->{DATA}->{TYPE}}->{PULL_FN_ARGS}->($e); + + pidl $static . "NTSTATUS ndr_pull_$e->{NAME}($args)"; + + pidl "{"; + indent; + $typefamily{$e->{DATA}->{TYPE}}->{PULL_FN_BODY}->($e->{DATA}); + pidl "return NT_STATUS_OK;"; + deindent; + pidl "}"; + pidl ""; +} + +##################################################################### +# parse a typedef - print side +sub ParseTypedefPrint($) +{ + my($e) = shift; + + my $args = $typefamily{$e->{DATA}->{TYPE}}->{PRINT_FN_ARGS}->($e); + + return unless !util::has_property($e, "noprint"); + + pidl "void ndr_print_$e->{NAME}($args)"; + pidl "{"; + indent; + $typefamily{$e->{DATA}->{TYPE}}->{PRINT_FN_BODY}->($e->{DATA}); + deindent; + pidl "}"; +} + +##################################################################### +## calculate the size of a structure +sub ParseTypedefNdrSize($) +{ + my($t) = shift; + + return unless needed::is_needed("ndr_size_$t->{NAME}"); + + my $tf = $typefamily{$t->{DATA}->{TYPE}}; + my $args = $tf->{SIZE_FN_ARGS}->($t); + + pidl "size_t ndr_size_$t->{NAME}($args)"; + pidl "{"; + indent; + $typefamily{$t->{DATA}->{TYPE}}->{SIZE_FN_BODY}->($t); + deindent; + pidl "}"; + pidl ""; +} + +##################################################################### +# parse a function - print side +sub ParseFunctionPrint($) +{ + my($fn) = shift; + + return unless !util::has_property($fn, "noprint"); + + pidl "void ndr_print_$fn->{NAME}(struct ndr_print *ndr, const char *name, int flags, struct $fn->{NAME} *r)"; + pidl "{"; + indent; + pidl "ndr_print_struct(ndr, name, \"$fn->{NAME}\");"; + pidl "ndr->depth++;"; + + pidl "if (flags & NDR_SET_VALUES) {"; + pidl "\tndr->flags |= LIBNDR_PRINT_SET_VALUES;"; + pidl "}"; + + pidl "if (flags & NDR_IN) {"; + indent; + pidl "ndr_print_struct(ndr, \"in\", \"$fn->{NAME}\");"; + pidl "ndr->depth++;"; + + foreach my $e (@{$fn->{ELEMENTS}}) { + if (util::has_property($e, "in")) { + ParseElementPrint($e, "r->in."); + } + } + pidl "ndr->depth--;"; + deindent; + pidl "}"; + + pidl "if (flags & NDR_OUT) {"; + indent; + pidl "ndr_print_struct(ndr, \"out\", \"$fn->{NAME}\");"; + pidl "ndr->depth++;"; + foreach my $e (@{$fn->{ELEMENTS}}) { + if (util::has_property($e, "out")) { + ParseElementPrint($e, "r->out."); + } + } + if ($fn->{RETURN_TYPE} && $fn->{RETURN_TYPE} ne "void") { + my $cprefix = "&"; + $cprefix = "" if (is_scalar_type($fn->{RETURN_TYPE})) ; # FIXME: Should really use util::c_push_prefix here + pidl "ndr_print_$fn->{RETURN_TYPE}(ndr, \"result\", $cprefix"."r->out.result);"; + } + pidl "ndr->depth--;"; + deindent; + pidl "}"; + + pidl "ndr->depth--;"; + deindent; + pidl "}"; + pidl ""; +} + +##################################################################### +# parse a function element +sub ParseFunctionElementPush($$) +{ + my $e = shift; + my $inout = shift; + + if (util::array_size($e)) { + if (need_wire_pointer($e)) { + ParsePtrPush($e, "r->$inout."); + pidl "if (r->$inout.$e->{NAME}) {"; + indent; + ParseArrayPush($e, "ndr", "r->$inout.", "NDR_SCALARS|NDR_BUFFERS"); + deindent; + pidl "}"; + } else { + ParseArrayPush($e, "ndr", "r->$inout.", "NDR_SCALARS|NDR_BUFFERS"); + } + } else { + ParseElementPushScalar($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS"); + if (need_wire_pointer($e)) { + ParseElementPushBuffer($e, "r->$inout."); + } + } +} + +##################################################################### +# parse a function +sub ParseFunctionPush($) +{ + my($fn) = shift; + my $static = fn_prefix($fn); + + return unless !util::has_property($fn, "nopush"); + + pidl $static . "NTSTATUS ndr_push_$fn->{NAME}(struct ndr_push *ndr, int flags, struct $fn->{NAME} *r)"; + pidl "{"; + indent; + + pidl "if (!(flags & NDR_IN)) goto ndr_out;"; + pidl ""; + + foreach my $e (@{$fn->{ELEMENTS}}) { + if (util::has_property($e, "in")) { + if (util::has_property($e, "ref")) { + check_null_pointer("*r->in.$e->{NAME}"); + } + ParseFunctionElementPush($e, "in"); + } + } + + pidl "ndr_out:"; + pidl "if (!(flags & NDR_OUT)) goto done;"; + pidl ""; + + foreach my $e (@{$fn->{ELEMENTS}}) { + if (util::has_property($e, "out")) { + if (util::has_property($e, "ref")) { + check_null_pointer("*r->out.$e->{NAME}"); + } + ParseFunctionElementPush($e, "out"); + } + } + + if ($fn->{RETURN_TYPE} && $fn->{RETURN_TYPE} ne "void") { + pidl "NDR_CHECK(ndr_push_$fn->{RETURN_TYPE}(ndr, NDR_SCALARS, r->out.result));"; + } + + pidl "done:"; + pidl "return NT_STATUS_OK;"; + deindent; + pidl "}"; + pidl ""; +} + +##################################################################### +# parse a function element +sub ParseFunctionElementPull($$) +{ + my $e = shift; + my $inout = shift; + + if (util::array_size($e)) { + if (need_wire_pointer($e)) { + pidl "NDR_CHECK(ndr_pull_unique_ptr(ndr, &_ptr_$e->{NAME}));"; + pidl "r->$inout.$e->{NAME} = NULL;"; + pidl "if (_ptr_$e->{NAME}) {"; + indent; + } elsif ($inout eq "out" && util::has_property($e, "ref")) { + pidl "if (r->$inout.$e->{NAME}) {"; + indent; + } + + ParseArrayPull($e, "ndr", "r->$inout.", "NDR_SCALARS|NDR_BUFFERS"); + + if (need_wire_pointer($e) or ($inout eq "out" and util::has_property($e, "ref"))) { + deindent; + pidl "}"; + } + } else { + if ($inout eq "out" && util::has_property($e, "ref")) { + pidl "if (ndr->flags & LIBNDR_FLAG_REF_ALLOC) {"; + pidl "\tNDR_ALLOC(ndr, r->out.$e->{NAME});"; + pidl "}"; + } + + if ($inout eq "in" && util::has_property($e, "ref")) { + pidl "NDR_ALLOC(ndr, r->in.$e->{NAME});"; + } + + ParseElementPullScalar($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS"); + if (need_wire_pointer($e)) { + ParseElementPullBuffer($e, "r->$inout."); + } + } +} + +############################################################ +# allocate ref variables +sub AllocateRefVars($) +{ + my $e = shift; + my $asize = util::array_size($e); + + # note that if the variable is also an "in" + # variable then we copy the initial value from + # the in side + + if (!defined $asize) { + # its a simple variable + pidl "NDR_ALLOC(ndr, r->out.$e->{NAME});"; + if (util::has_property($e, "in")) { + pidl "*r->out.$e->{NAME} = *r->in.$e->{NAME};"; + } else { + pidl "ZERO_STRUCTP(r->out.$e->{NAME});"; + } + return; + } + + # its an array + my $size = ParseExpr($e, $asize, "r->out."); + check_null_pointer($size); + pidl "NDR_ALLOC_N(ndr, r->out.$e->{NAME}, $size);"; + if (util::has_property($e, "in")) { + pidl "memcpy(r->out.$e->{NAME},r->in.$e->{NAME},$size * sizeof(*r->in.$e->{NAME}));"; + } else { + pidl "memset(r->out.$e->{NAME}, 0, $size * sizeof(*r->out.$e->{NAME}));"; + } +} + +##################################################################### +# parse a function +sub ParseFunctionPull($) +{ + my($fn) = shift; + my $static = fn_prefix($fn); + + return unless !util::has_property($fn, "nopull"); + + # pull function args + pidl $static . "NTSTATUS ndr_pull_$fn->{NAME}(struct ndr_pull *ndr, int flags, struct $fn->{NAME} *r)"; + pidl "{"; + indent; + + # declare any internal pointers we need + foreach my $e (@{$fn->{ELEMENTS}}) { + if (need_wire_pointer($e)) { + pidl "uint32_t _ptr_$e->{NAME};"; + } + } + + pidl "if (!(flags & NDR_IN)) goto ndr_out;"; + pidl ""; + + # auto-init the out section of a structure. I originally argued that + # this was a bad idea as it hides bugs, but coping correctly + # with initialisation and not wiping ref vars is turning + # out to be too tricky (tridge) + foreach my $e (@{$fn->{ELEMENTS}}) { + if (util::has_property($e, "out")) { + pidl "ZERO_STRUCT(r->out);"; + pidl ""; + last; + } + } + + foreach my $e (@{$fn->{ELEMENTS}}) { + if (util::has_property($e, "in")) { + ParseFunctionElementPull($e, "in"); + } + # we need to allocate any reference output variables, so that + # a dcerpc backend can be sure they are non-null + if (util::has_property($e, "out") && util::has_property($e, "ref")) { + AllocateRefVars($e); + } + } + + foreach my $e (@{$fn->{ELEMENTS}}) { + if (util::has_property($e, "in")) { + CheckArraySizes($e, "r->in."); + } + } + + pidl "ndr_out:"; + pidl "if (!(flags & NDR_OUT)) goto done;"; + pidl ""; + + foreach my $e (@{$fn->{ELEMENTS}}) { + if (util::has_property($e, "out")) { + ParseFunctionElementPull($e, "out"); + } + } + + foreach my $e (@{$fn->{ELEMENTS}}) { + if (util::has_property($e, "out")) { + CheckArraySizes($e, "r->out."); + } + } + + if ($fn->{RETURN_TYPE} && $fn->{RETURN_TYPE} ne "void") { + pidl "NDR_CHECK(ndr_pull_$fn->{RETURN_TYPE}(ndr, NDR_SCALARS, &r->out.result));"; + } + + pidl "done:"; + pidl ""; + pidl "return NT_STATUS_OK;"; + deindent; + pidl "}"; + pidl ""; +} + +##################################################################### +# produce a function call table +sub FunctionTable($) +{ + my($interface) = shift; + my($data) = $interface->{INHERITED_DATA}; + my $count = 0; + my $uname = uc $interface->{NAME}; + + foreach my $d (@{$data}) { + if ($d->{TYPE} eq "FUNCTION") { $count++; } + } + + return if ($count == 0); + + pidl "static const struct dcerpc_interface_call $interface->{NAME}\_calls[] = {"; + foreach my $d (@{$data}) { + if ($d->{TYPE} eq "FUNCTION") { + pidl "\t{"; + pidl "\t\t\"$d->{NAME}\","; + pidl "\t\tsizeof(struct $d->{NAME}),"; + pidl "\t\t(ndr_push_flags_fn_t) ndr_push_$d->{NAME},"; + pidl "\t\t(ndr_pull_flags_fn_t) ndr_pull_$d->{NAME},"; + pidl "\t\t(ndr_print_function_t) ndr_print_$d->{NAME}"; + pidl "\t},"; + } + } + pidl "\t{ NULL, 0, NULL, NULL, NULL }"; + pidl "};"; + pidl ""; + + # If no endpoint is set, default to the interface name as a named pipe + if (! defined $interface->{PROPERTIES}->{endpoint}) { + $interface->{PROPERTIES}->{endpoint} = "\"ncacn_np:[\\\\pipe\\\\" . $interface->{NAME} . "]\""; + } + + my @e = split / /, $interface->{PROPERTIES}->{endpoint}; + my $endpoint_count = $#e + 1; + + pidl "static const char * const $interface->{NAME}\_endpoint_strings[] = {"; + foreach my $ep (@e) { + pidl "\t$ep, "; + } + pidl "};"; + pidl ""; + + pidl "static const struct dcerpc_endpoint_list $interface->{NAME}\_endpoints = {"; + pidl "\t.count\t= $endpoint_count,"; + pidl "\t.names\t= $interface->{NAME}\_endpoint_strings"; + pidl "};"; + pidl ""; + + if (! defined $interface->{PROPERTIES}->{authservice}) { + $interface->{PROPERTIES}->{authservice} = "\"host\""; + } + + my @a = split / /, $interface->{PROPERTIES}->{authservice}; + my $authservice_count = $#a + 1; + + pidl "static const char * const $interface->{NAME}\_authservice_strings[] = {"; + foreach my $ap (@a) { + pidl "\t$ap, "; + } + pidl "};"; + pidl ""; + + pidl "static const struct dcerpc_authservice_list $interface->{NAME}\_authservices = {"; + pidl "\t.count\t= $endpoint_count,"; + pidl "\t.names\t= $interface->{NAME}\_authservice_strings"; + pidl "};"; + pidl ""; + + pidl "\nconst struct dcerpc_interface_table dcerpc_table_$interface->{NAME} = {"; + pidl "\t.name\t\t= \"$interface->{NAME}\","; + pidl "\t.uuid\t\t= DCERPC_$uname\_UUID,"; + pidl "\t.if_version\t= DCERPC_$uname\_VERSION,"; + pidl "\t.helpstring\t= DCERPC_$uname\_HELPSTRING,"; + pidl "\t.num_calls\t= $count,"; + pidl "\t.calls\t\t= $interface->{NAME}\_calls,"; + pidl "\t.endpoints\t= &$interface->{NAME}\_endpoints,"; + pidl "\t.authservices\t= &$interface->{NAME}\_authservices"; + pidl "};"; + pidl ""; + + pidl "static NTSTATUS dcerpc_ndr_$interface->{NAME}_init(void)"; + pidl "{"; + pidl "\treturn librpc_register_interface(&dcerpc_table_$interface->{NAME});"; + pidl "}"; + pidl ""; +} + +##################################################################### +# parse the interface definitions +sub ParseInterface($) +{ + my($interface) = shift; + my($data) = $interface->{DATA}; + + # Push functions + foreach my $d (@{$data}) { + ($d->{TYPE} eq "TYPEDEF") && + ParseTypedefPush($d); + ($d->{TYPE} eq "FUNCTION") && + ParseFunctionPush($d); + } + + # Pull functions + foreach my $d (@{$data}) { + ($d->{TYPE} eq "TYPEDEF") && + ParseTypedefPull($d); + ($d->{TYPE} eq "FUNCTION") && + ParseFunctionPull($d); + } + + # Print functions + foreach my $d (@{$data}) { + ($d->{TYPE} eq "TYPEDEF") && + ParseTypedefPrint($d); + ($d->{TYPE} eq "FUNCTION") && + ParseFunctionPrint($d); + } + + # Size functions + foreach my $d (@{$data}) { + ($d->{TYPE} eq "TYPEDEF") && + ParseTypedefNdrSize($d); + } + + FunctionTable($interface); +} + +sub RegistrationFunction($$) +{ + my $idl = shift; + my $filename = shift; + + $filename =~ /.*\/ndr_(.*).c/; + my $basename = $1; + pidl "NTSTATUS dcerpc_$basename\_init(void)"; + pidl "{"; + indent; + pidl "NTSTATUS status = NT_STATUS_OK;"; + foreach my $interface (@{$idl}) { + next if $interface->{TYPE} ne "INTERFACE"; + + my $data = $interface->{INHERITED_DATA}; + my $count = 0; + foreach my $d (@{$data}) { + if ($d->{TYPE} eq "FUNCTION") { $count++; } + } + + next if ($count == 0); + + pidl "status = dcerpc_ndr_$interface->{NAME}_init();"; + pidl "if (NT_STATUS_IS_ERR(status)) {"; + pidl "\treturn status;"; + pidl "}"; + pidl ""; + } + pidl "return status;"; + deindent; + pidl "}"; + pidl ""; +} + +sub CheckPointerTypes($$) +{ + my $s = shift; + my $default = shift; + + foreach my $e (@{$s->{ELEMENTS}}) { + if ($e->{POINTERS}) { + if (not defined(pointer_type($e))) { + $e->{PROPERTIES}->{$default} = 1; + } + + if (pointer_type($e) eq "ptr") { + print "Warning: ptr is not supported by pidl yet\n"; + } + } + } +} + +sub LoadInterface($) +{ + my $x = shift; + + if (not util::has_property($x, "pointer_default")) { + # MIDL defaults to "ptr" in DCE compatible mode (/osf) + # and "unique" in Microsoft Extensions mode (default) + $x->{PROPERTIES}->{pointer_default} = "unique"; + } + + foreach my $d (@{$x->{DATA}}) { + if (($d->{TYPE} eq "DECLARE") or ($d->{TYPE} eq "TYPEDEF")) { + if ($d->{DATA}->{TYPE} eq "STRUCT" or $d->{DATA}->{TYPE} eq "UNION") { + CheckPointerTypes($d->{DATA}, $x->{PROPERTIES}->{pointer_default}); + } + + if (defined($d->{PROPERTIES}) && !defined($d->{DATA}->{PROPERTIES})) { + $d->{DATA}->{PROPERTIES} = $d->{PROPERTIES}; + } + } + if ($d->{TYPE} eq "FUNCTION") { + CheckPointerTypes($d, + $x->{PROPERTIES}->{pointer_default} # MIDL defaults to "ref" + ); + } + } +} + +sub Load($) +{ + my $idl = shift; + + foreach my $x (@{$idl}) { + LoadInterface($x); + } +} + +##################################################################### +# parse a parsed IDL structure back into an IDL file +sub Parse($$) +{ + my($idl) = shift; + my($filename) = shift; + my $h_filename = $filename; + $res = ""; + + Load($idl); + + if ($h_filename =~ /(.*)\.c/) { + $h_filename = "$1.h"; + } + + pidl "/* parser auto-generated by pidl */"; + pidl ""; + pidl "#include \"includes.h\""; + pidl "#include \"$h_filename\""; + pidl ""; + + foreach my $x (@{$idl}) { + if ($x->{TYPE} eq "INTERFACE") { + needed::BuildNeeded($x); + ParseInterface($x); + } + } + + RegistrationFunction($idl, $filename); + + return $res; +} + +1; |