r5671: Bunch of fixes related to arrays and pointers to arrays.
[samba.git] / source4 / build / pidl / ndr.pm
index 1a11c66563da8c0e063880b30b165ff1ea550782..333f20e78f19423153d308fe7d799cae56e9262a 100644 (file)
@@ -117,8 +117,7 @@ sub is_inline_array($)
 sub is_varying_array($)
 {
        my $e = shift;
-       return 1 if (util::has_property($e, "length_is"));
-       return 0;
+       return util::has_property($e, "length_is");
 }
 
 # return 1 if this is a surrounding array (sometimes 
@@ -168,17 +167,16 @@ sub need_wire_pointer($)
        return $n;
 }
 
-# determine if an element is a pure scalar. pure scalars do not
-# have a "buffers" section in NDR
-sub is_pure_scalar($)
+# determine if an element needs a "buffers" section in NDR
+sub need_buffers_section($)
 {
        my $e = shift;
-       if (is_scalar_type($e->{TYPE}) && 
-           need_wire_pointer($e) == 0 && 
+       if ((is_scalar_type($e->{TYPE}) || util::has_property($e, "subcontext")) &&
+           $e->{POINTERS} == 0 && 
            !util::array_size($e)) {
-               return 1;
+               return 0;
        }
-       return 0;
+       return 1;
 }
 
 # see if a variable needs to be allocated by the NDR subsystem on pull
@@ -211,9 +209,9 @@ sub c_push_prefix($)
 
        if ($e->{TYPE} =~ "string") {
                $ret = "";
-       } elsif (is_scalar_type($e->{TYPE}) &&
-           $e->{POINTERS}) {
-               $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)) {
@@ -292,6 +290,8 @@ sub ParseExpr($$$)
 
        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";
        }
@@ -425,11 +425,13 @@ sub ParseArrayPushPreceding($$$)
 
 #####################################################################
 # parse the data of an array - push side
-sub ParseArrayPush($$$)
+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);
 
@@ -441,15 +443,15 @@ sub ParseArrayPush($$$)
        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));";
+               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, $var_prefix$e->{NAME}, $size));";
+               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, $var_prefix$e->{NAME}, sizeof($var_prefix$e->{NAME}\[0]), $size, (ndr_push_flags_fn_t)ndr_push_$e->{TYPE}));";
+               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}));";
        }
 }
 
@@ -460,15 +462,16 @@ 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}\", $var_prefix$e->{NAME}, $size);";
+               pidl "ndr_print_array_$e->{TYPE}(ndr, \"$e->{NAME}\", $cprefix$var_prefix$e->{NAME}, $size);";
        } else {
-               pidl "ndr_print_array(ndr, \"$e->{NAME}\", $var_prefix$e->{NAME}, sizeof($var_prefix$e->{NAME}\[0]), $size, (ndr_print_fn_t)ndr_print_$e->{TYPE});";
+               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});";
        }
 }
 
@@ -479,7 +482,7 @@ sub CheckArraySizes($$)
        my $e = shift;
        my $var_prefix = shift;
 
-       if (!is_surrounding_array($e) && is_conformant_array($e)) {
+       if (is_conformant_array($e)) {
                my $size = ParseExpr($e, util::array_size($e), $var_prefix);
                pidl "if ($var_prefix$e->{NAME}) {";
                indent;
@@ -515,65 +518,124 @@ sub ParseArrayPullPreceding($$$)
 
 #####################################################################
 # parse an array - pull side
-sub ParseArrayPull($$$)
+sub ParseArrayPull($$$$)
 {
        my $e = shift;
+       my $ndr = shift;
        my $var_prefix = shift;
        my $ndr_flags = shift;
 
-       my $size = ParseExpr($e, util::array_size($e), $var_prefix);
-       my $alloc_size = $size;
+       my $cprefix = c_pull_prefix($e);
+       my $length = ParseExpr($e, util::array_size($e), $var_prefix);
+       my $size = $length;
+
+       if (is_conformant_array($e)) {
+               $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_conformant_array($e) and is_surrounding_array($e)) {
-               $alloc_size = "ndr_get_array_size(ndr, &$var_prefix$e->{NAME})";
-               check_null_pointer($size);
-               pidl "if ($size > $alloc_size) {";
-               indent;
-               pidl "return ndr_pull_error(ndr, NDR_ERR_CONFORMANT_SIZE, \"Bad conformant size %u should be %u\", $alloc_size, $size);";
-               deindent;
-               pidl "}";
-       } elsif (!is_inline_array($e)) {
-               if ($var_prefix =~ /^r->out/ && $size =~ /^\*r->in/) {
-                       my $size2 = substr($size, 1);
-                       pidl "if (ndr->flags & LIBNDR_FLAG_REF_ALLOC) { NDR_ALLOC(ndr, $size2); }";
+       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);
+       }
 
-               $alloc_size = "ndr_get_array_size(ndr, &$var_prefix$e->{NAME})";
+       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}, $alloc_size);";
+                       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) {";
-                       indent;
-                       pidl "NDR_ALLOC_N(ndr, $var_prefix$e->{NAME}, $alloc_size);";
-                       deindent;
+                       pidl "if ($ndr->flags & LIBNDR_FLAG_REF_ALLOC) {";
+                       pidl "\tNDR_ALLOC_N($ndr, $var_prefix$e->{NAME}, $size);";
                        pidl "}";
                }
        }
 
-       if (is_varying_array($e)) {
-               pidl "NDR_CHECK(ndr_pull_array_length(ndr, &$var_prefix$e->{NAME}));";
-               $size = "ndr_get_array_length(ndr, &$var_prefix$e->{NAME})";
+       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}));";
        }
+}
 
-       check_null_pointer($size);
+sub ParseSubcontextPushStart($)
+{
+       my $e = shift;
+       my $sub_size = util::has_property($e, "subcontext");
 
-       if (is_scalar_type($e->{TYPE})) {
-               pidl "NDR_CHECK(ndr_pull_array_$e->{TYPE}(ndr, $ndr_flags, $var_prefix$e->{NAME}, $size));";
+       pidl "{";
+       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 $sub_size = util::has_property($e, "subcontext");
+       pidl "NDR_CHECK(ndr_push_subcontext_header(ndr, $sub_size, _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 $sub_size = util::has_property($e, "subcontext");
+
+       pidl "{";
+       indent;
+       pidl "struct ndr_pull *_ndr_$e->{NAME};";
+       pidl "NDR_ALLOC(ndr, _ndr_$e->{NAME});";
+       pidl "NDR_CHECK(ndr_pull_subcontext_header(ndr, $sub_size, _ndr_$e->{NAME}));"; 
+
+       return "_ndr_$e->{NAME}";
+}
+
+sub ParseSubcontextPullEnd($)
+{
+       my $e = shift;
+       my $sub_size = util::has_property($e, "subcontext");
+
+       my $advance;
+       if ($sub_size) {
+               $advance = "_ndr_$e->{NAME}->data_size";
        } else {
-               pidl "NDR_CHECK(ndr_pull_array(ndr, $ndr_flags, (void **)$var_prefix$e->{NAME}, sizeof($var_prefix$e->{NAME}\[0]), $size, (ndr_pull_flags_fn_t)ndr_pull_$e->{TYPE}));";
+               $advance = "_ndr_$e->{NAME}->offset";
        }
+       pidl "NDR_CHECK(ndr_pull_advance(ndr, $advance));";
+       deindent;
+       pidl "}";
 }
 
 #####################################################################
@@ -586,6 +648,7 @@ sub ParseElementPushScalar($$$)
        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);
 
@@ -593,18 +656,24 @@ sub ParseElementPushScalar($$$)
                pidl "$cprefix$var_prefix$e->{NAME} = $value;";
        }
 
+       if (defined $sub_size and $e->{POINTERS} == 0) {
+               $ndr = ParseSubcontextPushStart($e);
+       }
+
        if (need_wire_pointer($e)) {
-               ParseElementPushPtr($e, $ptr_prefix.$var_prefix);
+               ParsePtrPush($e, $ptr_prefix.$var_prefix);
        } elsif (is_inline_array($e)) {
-               ParseArrayPush($e, "r->", "NDR_SCALARS");
+               ParseArrayPush($e, $ndr, "r->", "NDR_SCALARS");
        } elsif (need_alloc($e)) {
                # no scalar component
        } elsif (my $switch = util::has_property($e, "switch_is")) {
-               ParseElementPushSwitch($e, $var_prefix, $ndr_flags, $switch);
-       } elsif (defined $sub_size) {
-               pidl "NDR_CHECK(ndr_push_subcontext_flags_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_push_flags_fn_t) ndr_push_$e->{TYPE}));";
+               ParseSwitchPush($e, $ndr, $var_prefix, $ndr_flags, $switch);
        } else {
-               pidl "NDR_CHECK(ndr_push_$e->{TYPE}(ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}));";
+               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);
@@ -612,7 +681,7 @@ sub ParseElementPushScalar($$$)
 
 #####################################################################
 # parse a pointer in a struct element or function
-sub ParseElementPushPtr($$)
+sub ParsePtrPush($$)
 {
        my $e = shift;
        my $var_prefix = shift;
@@ -675,9 +744,10 @@ sub ParseElementPrint($$)
 
 #####################################################################
 # parse scalars in a structure element - pull size
-sub ParseElementPullSwitch($$$$)
+sub ParseSwitchPull($$$$$)
 {
        my($e) = shift;
+       my $ndr = shift;
        my($var_prefix) = shift;
        my($ndr_flags) = shift;
        my $switch = shift;
@@ -696,15 +766,15 @@ sub ParseElementPullSwitch($$$$)
                pidl "if (($ndr_flags) & NDR_SCALARS) {";
                indent;
                pidl "$type_decl _level;";
-               pidl "NDR_CHECK(ndr_pull_$e2->{TYPE}(ndr, NDR_SCALARS, &_level));";
+               pidl "NDR_CHECK(ndr_pull_$e2->{TYPE}($ndr, NDR_SCALARS, &_level));";
                if ($switch_var =~ /r->in/) {
-                       pidl "if (!(ndr->flags & LIBNDR_FLAG_REF_ALLOC) && _level != $switch_var) {";
+                       pidl "if (!($ndr->flags & LIBNDR_FLAG_REF_ALLOC) && _level != $switch_var) {";
                        indent;
                } else {
                        pidl "if (_level != $switch_var) {"; 
                        indent;
                }
-               pidl "return ndr_pull_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value %u in $e->{NAME}\", _level);";
+               pidl "return ndr_pull_error($ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value %u in $e->{NAME}\", _level);";
                deindent;
                if ($switch_var =~ /r->/) {
                        pidl "} else { $switch_var = _level; }";
@@ -715,21 +785,15 @@ sub ParseElementPullSwitch($$$$)
                pidl "}";
        }
 
-       my $sub_size = util::has_property($e, "subcontext");
-       if (defined $sub_size) {
-               pidl "if (($ndr_flags) & NDR_SCALARS) {";
-               pidl "\tNDR_CHECK(ndr_pull_subcontext_union_fn(ndr, $sub_size, $switch_var, $cprefix$var_prefix$e->{NAME}, (ndr_pull_union_fn_t) ndr_pull_$e->{TYPE}));";
-               pidl "}";
-       } else {
-               pidl "NDR_CHECK(ndr_pull_$e->{TYPE}(ndr, $ndr_flags, $switch_var, $cprefix$var_prefix$e->{NAME}));";
-       }
+       pidl "NDR_CHECK(ndr_pull_$e->{TYPE}($ndr, $ndr_flags, $switch_var, $cprefix$var_prefix$e->{NAME}));";
 }
 
 #####################################################################
 # push switch element
-sub ParseElementPushSwitch($$$$)
+sub ParseSwitchPush($$$$$)
 {
        my($e) = shift;
+       my $ndr = shift;
        my($var_prefix) = shift;
        my($ndr_flags) = shift;
        my $switch = shift;
@@ -743,18 +807,11 @@ sub ParseElementPushSwitch($$$$)
            !util::has_property($utype, "nodiscriminant")) {
                my $e2 = util::find_sibling($e, $switch);
                pidl "if (($ndr_flags) & NDR_SCALARS) {";
-               pidl "\tNDR_CHECK(ndr_push_$e2->{TYPE}(ndr, NDR_SCALARS, $switch_var));";
+               pidl "\tNDR_CHECK(ndr_push_$e2->{TYPE}($ndr, NDR_SCALARS, $switch_var));";
                pidl "}";
        }
 
-       my $sub_size = util::has_property($e, "subcontext");
-       if (defined $sub_size) {
-               pidl "if(($ndr_flags) & NDR_SCALARS) {";
-               pidl "\tNDR_CHECK(ndr_push_subcontext_union_fn(ndr, $sub_size, $switch_var, $cprefix$var_prefix$e->{NAME}, (ndr_push_union_fn_t) ndr_push_$e->{TYPE}));";
-               pidl "}";
-       } else {
-               pidl "NDR_CHECK(ndr_push_$e->{TYPE}(ndr, $ndr_flags, $switch_var, $cprefix$var_prefix$e->{NAME}));";
-       }
+       pidl "NDR_CHECK(ndr_push_$e->{TYPE}($ndr, $ndr_flags, $switch_var, $cprefix$var_prefix$e->{NAME}));";
 }
 
 #####################################################################
@@ -767,35 +824,43 @@ sub ParseElementPullScalar($$$)
        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_SCALARS|NDR_BUFFERS";
+       }
+
        if (is_inline_array($e)) {
-               ParseArrayPull($e, "r->", "NDR_SCALARS");
+               ParseArrayPull($e, $ndr, "r->", "NDR_SCALARS");
        } elsif (need_wire_pointer($e)) {
-               ParseElementPullPtr($e, $ptr_prefix.$var_prefix);
+               ParsePtrPull($e, $ptr_prefix.$var_prefix);
        } elsif (is_surrounding_array($e)) {
        } elsif (my $switch = util::has_property($e, "switch_is")) {
-               ParseElementPullSwitch($e, $var_prefix, $ndr_flags, $switch);
-       } elsif (defined $sub_size) {
-               pidl "NDR_CHECK(ndr_pull_subcontext_flags_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_pull_flags_fn_t) ndr_pull_$e->{TYPE}));";
+               ParseSwitchPull($e, $ndr, $var_prefix, $ndr_flags, $switch);
        } else {
-               pidl "NDR_CHECK(ndr_pull_$e->{TYPE}(ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}));";
+               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 "\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 ParseElementPullPtr($$)
+sub ParsePtrPull($$)
 {
        my($e) = shift;
        my($var_prefix) = shift;
@@ -821,15 +886,16 @@ sub ParseElementPushBuffer($$)
        my($var_prefix) = shift;
        my $cprefix = c_push_prefix($e);
        my $sub_size = util::has_property($e, "subcontext");
+       my $ndr = "ndr";
 
-       return if (is_pure_scalar($e));
+       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) {
-                       ParseElementPushPtr($e,$pointers.$var_prefix);
+                       ParsePtrPush($e,$pointers.$var_prefix);
                }
                pidl "if ($pointers$var_prefix$e->{NAME}) {";
                indent;
@@ -845,19 +911,24 @@ sub ParseElementPushBuffer($$)
        {
                $ndr_flags="NDR_SCALARS|$ndr_flags" 
        }
+
+       if (defined $sub_size) {
+               $ndr = ParseSubcontextPushStart($e);
+               $ndr_flags = "NDR_SCALARS|NDR_BUFFERS";
+       }
            
        if (util::array_size($e)) {
-               ParseArrayPush($e, "r->", $ndr_flags);
+               ParseArrayPush($e, $ndr, "r->", $ndr_flags);
        } elsif (my $switch = util::has_property($e, "switch_is")) {
-               ParseElementPushSwitch($e, $var_prefix, $ndr_flags, $switch);
-       } elsif (defined $sub_size) {
-               if ($e->{POINTERS}) {
-                       pidl "NDR_CHECK(ndr_push_subcontext_flags_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_push_flags_fn_t) ndr_push_$e->{TYPE}));";
-               }
+               ParseSwitchPush($e, $ndr, $var_prefix, $ndr_flags, $switch);
        } else {
                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 "}";
@@ -874,15 +945,16 @@ sub ParseElementPullBuffer($$)
        my($var_prefix) = shift;
        my $cprefix = c_pull_prefix($e);
        my $sub_size = util::has_property($e, "subcontext");
+       my $ndr = "ndr";
 
-       return if (is_pure_scalar($e));
+       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) {
-                       ParseElementPullPtr($e,$pointers.$var_prefix);
+                       ParsePtrPull($e,$pointers.$var_prefix);
                }
                pidl "if ($pointers$var_prefix$e->{NAME}) {";
                indent;
@@ -900,17 +972,22 @@ sub ParseElementPullBuffer($$)
        {
                $ndr_flags="NDR_SCALARS|$ndr_flags" 
        }
-           
+
+       if (defined $sub_size) {
+               $ndr = ParseSubcontextPullStart($e);
+               $ndr_flags = "NDR_SCALARS|NDR_BUFFERS";
+       }
+
        if (util::array_size($e)) {
-               ParseArrayPull($e, "r->", $ndr_flags);
+               ParseArrayPull($e, $ndr, "r->", $ndr_flags);
        } elsif (my $switch = util::has_property($e, "switch_is")) {
-               ParseElementPullSwitch($e, $var_prefix, $ndr_flags, $switch);
-       } elsif (defined $sub_size) {
-               if ($e->{POINTERS}) {
-                       pidl "NDR_CHECK(ndr_pull_subcontext_flags_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_pull_flags_fn_t) ndr_pull_$e->{TYPE}));";
-               }
+               ParseSwitchPull($e, $ndr, $var_prefix, $ndr_flags, $switch);
        } else {
-               pidl "NDR_CHECK(ndr_pull_$e->{TYPE}(ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}));";
+               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")) {
@@ -1680,11 +1757,11 @@ sub ParseFunctionElementPush($$)
                        pidl "NDR_CHECK(ndr_push_unique_ptr(ndr, r->$inout.$e->{NAME}));";
                        pidl "if (r->$inout.$e->{NAME}) {";
                        indent;
-                       ParseArrayPush($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
+                       ParseArrayPush($e, "ndr", "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
                        deindent;
                        pidl "}";
                } else {
-                       ParseArrayPush($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
+                       ParseArrayPush($e, "ndr", "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
                }
        } else {
                ParseElementPushScalar($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
@@ -1749,15 +1826,18 @@ sub ParseFunctionElementPull($$)
                        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}) {";
-               } else {
-                       pidl "{";
+                       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 "}";
                }
-               indent;
-               ParseArrayPull($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
-               deindent;
-               pidl "}";
        } else {
                if ($inout eq "out" && util::has_property($e, "ref")) {
                        pidl "if (ndr->flags & LIBNDR_FLAG_REF_ALLOC) {";
@@ -1937,18 +2017,40 @@ sub FunctionTable($)
        pidl "";
 
        pidl "static const struct dcerpc_endpoint_list $interface->{NAME}\_endpoints = {";
-       pidl "\t$endpoint_count, $interface->{NAME}\_endpoint_strings";
+       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\"$interface->{NAME}\",";
-       pidl "\tDCERPC_$uname\_UUID,";
-       pidl "\tDCERPC_$uname\_VERSION,";
-       pidl "\tDCERPC_$uname\_HELPSTRING,";
-       pidl "\t$count,";
-       pidl "\t$interface->{NAME}\_calls,";
-       pidl "\t&$interface->{NAME}\_endpoints";
+       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 "";