diff options
| -rw-r--r-- | source4/build/pidl/parser.pm | 22 | ||||
| -rw-r--r-- | source4/build/pidl/util.pm | 24 |
2 files changed, 45 insertions, 1 deletions
diff --git a/source4/build/pidl/parser.pm b/source4/build/pidl/parser.pm index 14314886dc..90a1998a50 100644 --- a/source4/build/pidl/parser.pm +++ b/source4/build/pidl/parser.pm @@ -56,6 +56,20 @@ sub find_size_var($$) ##################################################################### +# work out the correct alignment for a structure +sub struct_alignment($) +{ + my $s = shift; + my $align = 1; + for my $e (@{$s->{ELEMENTS}}) { + if ($align < util::type_align($e)) { + $align = util::type_align($e); + } + } + return $align; +} + +##################################################################### # parse an array - push side sub ParseArrayPush($$) { @@ -312,11 +326,14 @@ sub ParseStructPush($) $res .= "\tndr_push_save(ndr, &_save1);\n"; } + my $align = struct_alignment($struct); + $res .= "\tNDR_CHECK(ndr_push_align(ndr, $align));\n"; + $res .= "\tif (!(ndr_flags & NDR_SCALARS)) goto buffers;\n"; foreach my $e (@{$struct->{ELEMENTS}}) { if (defined($struct_len) && $e == $struct_len) { - $res .= "\tNDR_CHECK(ndr_push_align_$e->{TYPE}(ndr));\n"; + $res .= "\tNDR_CHECK(ndr_push_align(ndr, sizeof($e->{TYPE})));\n"; $res .= "\tndr_push_save(ndr, &_save2);\n"; } ParseElementPushScalar($e, "r->", "NDR_SCALARS"); @@ -394,6 +411,9 @@ sub ParseStructPull($) $res .= "\tndr_pull_save(ndr, &_save);\n"; } + my $align = struct_alignment($struct); + $res .= "\tNDR_CHECK(ndr_pull_align(ndr, $align));\n"; + $res .= "\tif (!(ndr_flags & NDR_SCALARS)) goto buffers;\n"; foreach my $e (@{$struct->{ELEMENTS}}) { ParseElementPullScalar($e, "r->", "NDR_SCALARS"); diff --git a/source4/build/pidl/util.pm b/source4/build/pidl/util.pm index 72176abab3..d6c493398c 100644 --- a/source4/build/pidl/util.pm +++ b/source4/build/pidl/util.pm @@ -187,6 +187,30 @@ sub is_scalar_type($) return 0; } +# return the NDR alignment for a type +sub type_align($) +{ + my($e) = shift; + my $type = $e->{TYPE}; + + if ($e->{POINTERS} || array_size($e)) { + # FIXME: we really should recurse here + return 4; + } + + return 4, if ($type eq "uint32"); + return 4, if ($type eq "long"); + return 2, if ($type eq "short"); + return 1, if ($type eq "char"); + return 1, if ($type eq "uint8"); + return 2, if ($type eq "uint16"); + return 4, if ($type eq "NTTIME"); + return 8, if ($type eq "hyper"); + return 2, if ($type eq "wchar_t"); + + return 0; +} + # this is used to determine if the ndr push/pull functions will need # a ndr_flags field to split by buffers/scalars sub is_builtin_type($) |