pidl/lib/Parse/Pidl/Samba4/NDR/Parser.pm

   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-2006
   6 # released under the GNU GPL
   7
   8 package Parse::Pidl::Samba4::NDR::Parser;
   9
  10 require Exporter;
  11 @ISA = qw(Exporter);
  12 @EXPORT_OK = qw(check_null_pointer NeededFunction NeededElement NeededType $res NeededInterface TypeFunctionName ParseElementPrint);
  13
  14 use strict;
  15 use Parse::Pidl::Typelist qw(hasType getType mapTypeName typeHasBody);
  16 use Parse::Pidl::Util qw(has_property ParseExpr ParseExprExt print_uuid unmake_str);
  17 use Parse::Pidl::CUtil qw(get_pointer_to get_value_of get_array_element);
  18 use Parse::Pidl::NDR qw(GetPrevLevel GetNextLevel ContainsDeferred ContainsPipe is_charset_array);
  19 use Parse::Pidl::Samba4 qw(is_intree choose_header ArrayDynamicallyAllocated);
  20 use Parse::Pidl::Samba4::Header qw(GenerateFunctionInEnv GenerateFunctionOutEnv EnvSubstituteValue GenerateStructEnv);
  21 use Parse::Pidl qw(warning);
  22
  23 use vars qw($VERSION);
  24 $VERSION = '0.01';
  25
  26 # list of known types
  27 my %typefamily;
  28
  29 sub new($$) {
  30         my ($class) = @_;
  31         my $self = { res => "", res_hdr => "", deferred => [], tabs => "", defer_tabs => "" };
  32         bless($self, $class);
  33 }
  34
  35 sub get_typefamily($)
  36 {
  37         my $n = shift;
  38         return $typefamily{$n};
  39 }
  40
  41 sub append_prefix($$)
  42 {
  43         my ($e, $var_name) = @_;
  44         my $pointers = 0;
  45         my $arrays = 0;
  46
  47         foreach my $l (@{$e->{LEVELS}}) {
  48                 if ($l->{TYPE} eq "POINTER") {
  49                         $pointers++;
  50                 } elsif ($l->{TYPE} eq "ARRAY") {
  51                         $arrays++;
  52                         if (($pointers == 0) and
  53                             (not $l->{IS_FIXED}) and
  54                             (not $l->{IS_INLINE})) {
  55                                 return get_value_of($var_name);
  56                         }
  57                 } elsif ($l->{TYPE} eq "DATA") {
  58                         if (Parse::Pidl::Typelist::scalar_is_reference($l->{DATA_TYPE})) {
  59                                 return get_value_of($var_name) unless ($pointers or $arrays);
  60                         }
  61                 }
  62         }
  63
  64         return $var_name;
  65 }
  66
  67 sub has_fast_array($$)
  68 {
  69         my ($e,$l) = @_;
  70
  71         return 0 if ($l->{TYPE} ne "ARRAY");
  72
  73         my $nl = GetNextLevel($e,$l);
  74         return 0 unless ($nl->{TYPE} eq "DATA");
  75         return 0 unless (hasType($nl->{DATA_TYPE}));
  76
  77         my $t = getType($nl->{DATA_TYPE});
  78
  79         # Only uint8 and string have fast array functions at the moment
  80         return ($t->{NAME} eq "uint8") or ($t->{NAME} eq "string");
  81 }
  82
  83
  84 ####################################
  85 # pidl() is our basic output routine
  86 sub pidl($$)
  87 {
  88         my ($self, $d) = @_;
  89         if ($d) {
  90                 $self->{res} .= $self->{tabs};
  91                 $self->{res} .= $d;
  92         }
  93         $self->{res} .="\n";
  94 }
  95
  96 sub pidl_hdr($$) { my ($self, $d) = @_; $self->{res_hdr} .= "$d\n"; }
  97
  98 ####################################
  99 # defer() is like pidl(), but adds to
 100 # a deferred buffer which is then added to the
 101 # output buffer at the end of the structure/union/function
 102 # This is needed to cope with code that must be pushed back
 103 # to the end of a block of elements
 104 sub defer_indent($) { my ($self) = @_; $self->{defer_tabs}.="\t"; }
 105 sub defer_deindent($) { my ($self) = @_; $self->{defer_tabs}=substr($self->{defer_tabs}, 0, -1); }
 106
 107 sub defer($$)
 108 {
 109         my ($self, $d) = @_;
 110         if ($d) {
 111                 push(@{$self->{deferred}}, $self->{defer_tabs}.$d);
 112         }
 113 }
 114
 115 ########################################
 116 # add the deferred content to the current
 117 # output
 118 sub add_deferred($)
 119 {
 120         my ($self) = @_;
 121         $self->pidl($_) foreach (@{$self->{deferred}});
 122         $self->{deferred} = [];
 123         $self->{defer_tabs} = "";
 124 }
 125
 126 sub indent($)
 127 {
 128         my ($self) = @_;
 129         $self->{tabs} .= "\t";
 130 }
 131
 132 sub deindent($)
 133 {
 134         my ($self) = @_;
 135         $self->{tabs} = substr($self->{tabs}, 0, -1);
 136 }
 137
 138 #####################################################################
 139 # declare a function public or static, depending on its attributes
 140 sub fn_declare($$$$)
 141 {
 142         my ($self,$type,$fn,$decl) = @_;
 143
 144         if (has_property($fn, "no$type")) {
 145                 $self->pidl_hdr("$decl;");
 146                 return 0;
 147         }
 148
 149         if (has_property($fn, "public")) {
 150                 $self->pidl_hdr("$decl;");
 151                 $self->pidl("_PUBLIC_ $decl");
 152         } else {
 153                 $self->pidl("static $decl");
 154         }
 155
 156         return 1;
 157 }
 158
 159 ###################################################################
 160 # setup any special flags for an element or structure
 161 sub start_flags($$$)
 162 {
 163         my ($self, $e, $ndr) = @_;
 164         my $flags = has_property($e, "flag");
 165         if (defined $flags) {
 166                 $self->pidl("{");
 167                 $self->indent;
 168                 $self->pidl("uint32_t _flags_save_$e->{TYPE} = $ndr->flags;");
 169                 $self->pidl("ndr_set_flags(&$ndr->flags, $flags);");
 170         }
 171 }
 172
 173 ###################################################################
 174 # end any special flags for an element or structure
 175 sub end_flags($$$)
 176 {
 177         my ($self, $e, $ndr) = @_;
 178         my $flags = has_property($e, "flag");
 179         if (defined $flags) {
 180                 $self->pidl("$ndr->flags = _flags_save_$e->{TYPE};");
 181                 $self->deindent;
 182                 $self->pidl("}");
 183         }
 184 }
 185
 186 #####################################################################
 187 # parse the data of an array - push side
 188 sub ParseArrayPushHeader($$$$$$)
 189 {
 190         my ($self,$e,$l,$ndr,$var_name,$env) = @_;
 191
 192         my $size;
 193         my $length;
 194
 195         if ($l->{IS_ZERO_TERMINATED}) {
 196                 if (has_property($e, "charset")) {
 197                         $size = $length = "ndr_charset_length($var_name, CH_$e->{PROPERTIES}->{charset})";
 198                 } else {
 199                         $size = $length = "ndr_string_length($var_name, sizeof(*$var_name))";
 200                 }
 201                 if (defined($l->{SIZE_IS})) {
 202                         $size = ParseExpr($l->{SIZE_IS}, $env, $e);
 203                 }
 204                 if (defined($l->{LENGTH_IS})) {
 205                         $length = ParseExpr($l->{LENGTH_IS}, $env, $e);
 206                 }
 207         } else {
 208                 $size = ParseExpr($l->{SIZE_IS}, $env, $e);
 209                 $length = ParseExpr($l->{LENGTH_IS}, $env, $e);
 210         }
 211
 212         if ((!$l->{IS_SURROUNDING}) and $l->{IS_CONFORMANT}) {
 213                 $self->pidl("NDR_CHECK(ndr_push_uint3264($ndr, NDR_SCALARS, $size));");
 214         }
 215
 216         if ($l->{IS_VARYING}) {
 217                 $self->pidl("NDR_CHECK(ndr_push_uint3264($ndr, NDR_SCALARS, 0));");  # array offset
 218                 $self->pidl("NDR_CHECK(ndr_push_uint3264($ndr, NDR_SCALARS, $length));");
 219         }
 220
 221         return $length;
 222 }
 223
 224 sub check_fully_dereferenced($$)
 225 {
 226         my ($element, $env) = @_;
 227
 228         return sub ($) {
 229                 my $origvar = shift;
 230                 my $check = 0;
 231
 232                 # Figure out the number of pointers in $ptr
 233                 my $expandedvar = $origvar;
 234                 $expandedvar =~ s/^(\**)//;
 235                 my $ptr = $1;
 236
 237                 my $var = undef;
 238                 foreach (keys %$env) {
 239                         if ($env->{$_} eq $expandedvar) {
 240                                 $var = $_;
 241                                 last;
 242                         }
 243                 }
 244
 245                 return($origvar) unless (defined($var));
 246                 my $e;
 247                 foreach (@{$element->{PARENT}->{ELEMENTS}}) {
 248                         if ($_->{NAME} eq $var) {
 249                                 $e = $_;
 250                                 last;
 251                         }
 252                 }
 253
 254                 $e or die("Environment doesn't match siblings");
 255
 256                 # See if pointer at pointer level $level
 257                 # needs to be checked.
 258                 my $nump = 0;
 259                 foreach (@{$e->{LEVELS}}) {
 260                         if ($_->{TYPE} eq "POINTER") {
 261                                 $nump = $_->{POINTER_INDEX}+1;
 262                         }
 263                 }
 264                 warning($element->{ORIGINAL}, "Got pointer for `$e->{NAME}', expected fully derefenced variable") if ($nump > length($ptr));
 265                 return ($origvar);
 266         }
 267 }
 268
 269 sub check_null_pointer($$$$)
 270 {
 271         my ($element, $env, $print_fn, $return) = @_;
 272
 273         return sub ($) {
 274                 my $expandedvar = shift;
 275                 my $check = 0;
 276
 277                 # Figure out the number of pointers in $ptr
 278                 $expandedvar =~ s/^(\**)//;
 279                 my $ptr = $1;
 280
 281                 my $var = undef;
 282                 foreach (keys %$env) {
 283                         if ($env->{$_} eq $expandedvar) {
 284                                 $var = $_;
 285                                 last;
 286                         }
 287                 }
 288
 289                 if (defined($var)) {
 290                         my $e;
 291                         # lookup ptr in $e
 292                         foreach (@{$element->{PARENT}->{ELEMENTS}}) {
 293                                 if ($_->{NAME} eq $var) {
 294                                         $e = $_;
 295                                         last;
 296                                 }
 297                         }
 298
 299                         $e or die("Environment doesn't match siblings");
 300
 301                         # See if pointer at pointer level $level
 302                         # needs to be checked.
 303                         foreach my $l (@{$e->{LEVELS}}) {
 304                                 if ($l->{TYPE} eq "POINTER" and
 305                                         $l->{POINTER_INDEX} == length($ptr)) {
 306                                         # No need to check ref pointers
 307                                         $check = ($l->{POINTER_TYPE} ne "ref");
 308                                         last;
 309                                 }
 310
 311                                 if ($l->{TYPE} eq "DATA") {
 312                                         warning($element, "too much dereferences for `$var'");
 313                                 }
 314                         }
 315                 } else {
 316                         warning($element, "unknown dereferenced expression `$expandedvar'");
 317                         $check = 1;
 318                 }
 319
 320                 $print_fn->("if ($ptr$expandedvar == NULL) $return") if $check;
 321         }
 322 }
 323
 324 sub ParseArrayPullGetSize($$$$$$)
 325 {
 326         my ($self,$e,$l,$ndr,$var_name,$env) = @_;
 327
 328         my $size;
 329
 330         if ($l->{IS_CONFORMANT}) {
 331                 $size = "ndr_get_array_size($ndr, " . get_pointer_to($var_name) . ")";
 332         } elsif ($l->{IS_ZERO_TERMINATED} and $l->{SIZE_IS} == 0 and $l->{LENGTH_IS} == 0) { # Noheader arrays
 333                 $size = "ndr_get_string_size($ndr, sizeof(*$var_name))";
 334         } else {
 335                 $size = ParseExprExt($l->{SIZE_IS}, $env, $e->{ORIGINAL},
 336                         check_null_pointer($e, $env, sub { $self->pidl(shift); },
 337                                            "return ndr_pull_error($ndr, NDR_ERR_INVALID_POINTER, \"NULL Pointer for size_is()\");"),
 338                         check_fully_dereferenced($e, $env));
 339         }
 340
 341         my $array_size = $size;
 342
 343         return $array_size;
 344 }
 345
 346 #####################################################################
 347 # parse an array - pull side
 348 sub ParseArrayPullGetLength($$$$$$;$)
 349 {
 350         my ($self,$e,$l,$ndr,$var_name,$env,$array_size) = @_;
 351
 352         if (not defined($array_size)) {
 353                 $array_size = $self->ParseArrayPullGetSize($e, $l, $ndr, $var_name, $env);
 354         }
 355
 356         my $array_length = $array_size;
 357         if ($l->{IS_VARYING}) {
 358                 my $length = "ndr_get_array_length($ndr, " . get_pointer_to($var_name) .")";
 359                 $array_length = $length;
 360         }
 361
 362         return $array_length;
 363 }
 364
 365 #####################################################################
 366 # parse an array - pull side
 367 sub ParseArrayPullHeader($$$$$$)
 368 {
 369         my ($self,$e,$l,$ndr,$var_name,$env) = @_;
 370
 371         if ((!$l->{IS_SURROUNDING}) and $l->{IS_CONFORMANT}) {
 372                 $self->pidl("NDR_CHECK(ndr_pull_array_size($ndr, " . get_pointer_to($var_name) . "));");
 373         }
 374
 375         if ($l->{IS_VARYING}) {
 376                 $self->pidl("NDR_CHECK(ndr_pull_array_length($ndr, " . get_pointer_to($var_name) . "));");
 377         }
 378
 379         my $array_size = $self->ParseArrayPullGetSize($e, $l, $ndr, $var_name, $env);
 380         my $array_length = $self->ParseArrayPullGetLength($e, $l, $ndr, $var_name, $env, $array_size);
 381
 382         if ($array_length ne $array_size) {
 383                 $self->pidl("if ($array_length > $array_size) {");
 384                 $self->indent;
 385                 $self->pidl("return ndr_pull_error($ndr, NDR_ERR_ARRAY_SIZE, \"Bad array size %u should exceed array length %u\", $array_size, $array_length);");
 386                 $self->deindent;
 387                 $self->pidl("}");
 388         }
 389
 390         if ($l->{IS_CONFORMANT} and (defined($l->{SIZE_IS}) or not $l->{IS_ZERO_TERMINATED})) {
 391                 $self->defer("if ($var_name) {");
 392                 $self->defer_indent;
 393                 my $size = ParseExprExt($l->{SIZE_IS}, $env, $e->{ORIGINAL},
 394                         check_null_pointer($e, $env, sub { $self->defer(shift); },
 395                                            "return ndr_pull_error($ndr, NDR_ERR_INVALID_POINTER, \"NULL Pointer for size_is()\");"),
 396                         check_fully_dereferenced($e, $env));
 397                 $self->defer("NDR_CHECK(ndr_check_array_size($ndr, (void*)" . get_pointer_to($var_name) . ", $size));");
 398                 $self->defer_deindent;
 399                 $self->defer("}");
 400         }
 401
 402         if ($l->{IS_VARYING} and (defined($l->{LENGTH_IS}) or not $l->{IS_ZERO_TERMINATED})) {
 403                 $self->defer("if ($var_name) {");
 404                 $self->defer_indent;
 405                 my $length = ParseExprExt($l->{LENGTH_IS}, $env, $e->{ORIGINAL},
 406                         check_null_pointer($e, $env, sub { $self->defer(shift); },
 407                                            "return ndr_pull_error($ndr, NDR_ERR_INVALID_POINTER, \"NULL Pointer for length_is()\");"),
 408                         check_fully_dereferenced($e, $env));
 409                 $self->defer("NDR_CHECK(ndr_check_array_length($ndr, (void*)" . get_pointer_to($var_name) . ", $length));");
 410                 $self->defer_deindent;
 411                 $self->defer("}");
 412         }
 413
 414         if (ArrayDynamicallyAllocated($e,$l) and not is_charset_array($e,$l)) {
 415                 $self->AllocateArrayLevel($e,$l,$ndr,$var_name,$array_size);
 416         }
 417
 418         return $array_length;
 419 }
 420
 421 sub compression_alg($$)
 422 {
 423         my ($e, $l) = @_;
 424         my ($alg, $clen, $dlen) = split(/,/, $l->{COMPRESSION});
 425
 426         return $alg;
 427 }
 428
 429 sub compression_clen($$$)
 430 {
 431         my ($e, $l, $env) = @_;
 432         my ($alg, $clen, $dlen) = split(/,/, $l->{COMPRESSION});
 433
 434         return ParseExpr($clen, $env, $e->{ORIGINAL});
 435 }
 436
 437 sub compression_dlen($$$)
 438 {
 439         my ($e,$l,$env) = @_;
 440         my ($alg, $clen, $dlen) = split(/,/, $l->{COMPRESSION});
 441
 442         return ParseExpr($dlen, $env, $e->{ORIGINAL});
 443 }
 444
 445 sub ParseCompressionPushStart($$$$$)
 446 {
 447         my ($self,$e,$l,$ndr,$env) = @_;
 448         my $comndr = "$ndr\_compressed";
 449         my $alg = compression_alg($e, $l);
 450         my $dlen = compression_dlen($e, $l, $env);
 451
 452         $self->pidl("{");
 453         $self->indent;
 454         $self->pidl("struct ndr_push *$comndr;");
 455         $self->pidl("NDR_CHECK(ndr_push_compression_start($ndr, &$comndr, $alg, $dlen));");
 456
 457         return $comndr;
 458 }
 459
 460 sub ParseCompressionPushEnd($$$$$)
 461 {
 462         my ($self,$e,$l,$ndr,$env) = @_;
 463         my $comndr = "$ndr\_compressed";
 464         my $alg = compression_alg($e, $l);
 465         my $dlen = compression_dlen($e, $l, $env);
 466
 467         $self->pidl("NDR_CHECK(ndr_push_compression_end($ndr, $comndr, $alg, $dlen));");
 468         $self->deindent;
 469         $self->pidl("}");
 470 }
 471
 472 sub ParseCompressionPullStart($$$$$)
 473 {
 474         my ($self,$e,$l,$ndr,$env) = @_;
 475         my $comndr = "$ndr\_compressed";
 476         my $alg = compression_alg($e, $l);
 477         my $dlen = compression_dlen($e, $l, $env);
 478
 479         $self->pidl("{");
 480         $self->indent;
 481         $self->pidl("struct ndr_pull *$comndr;");
 482         $self->pidl("NDR_CHECK(ndr_pull_compression_start($ndr, &$comndr, $alg, $dlen));");
 483
 484         return $comndr;
 485 }
 486
 487 sub ParseCompressionPullEnd($$$$$)
 488 {
 489         my ($self,$e,$l,$ndr,$env) = @_;
 490         my $comndr = "$ndr\_compressed";
 491         my $alg = compression_alg($e, $l);
 492         my $dlen = compression_dlen($e, $l, $env);
 493
 494         $self->pidl("NDR_CHECK(ndr_pull_compression_end($ndr, $comndr, $alg, $dlen));");
 495         $self->deindent;
 496         $self->pidl("}");
 497 }
 498
 499 sub ParseSubcontextPushStart($$$$$)
 500 {
 501         my ($self,$e,$l,$ndr,$env) = @_;
 502         my $subndr = "_ndr_$e->{NAME}";
 503         my $subcontext_size = ParseExpr($l->{SUBCONTEXT_SIZE}, $env, $e->{ORIGINAL});
 504
 505         $self->pidl("{");
 506         $self->indent;
 507         $self->pidl("struct ndr_push *$subndr;");
 508         $self->pidl("NDR_CHECK(ndr_push_subcontext_start($ndr, &$subndr, $l->{HEADER_SIZE}, $subcontext_size));");
 509
 510         if (defined $l->{COMPRESSION}) {
 511                 $subndr = $self->ParseCompressionPushStart($e, $l, $subndr, $env);
 512         }
 513
 514         return $subndr;
 515 }
 516
 517 sub ParseSubcontextPushEnd($$$$$)
 518 {
 519         my ($self,$e,$l,$ndr,$env) = @_;
 520         my $subndr = "_ndr_$e->{NAME}";
 521         my $subcontext_size = ParseExpr($l->{SUBCONTEXT_SIZE}, $env, $e->{ORIGINAL});
 522
 523         if (defined $l->{COMPRESSION}) {
 524                 $self->ParseCompressionPushEnd($e, $l, $subndr, $env);
 525         }
 526
 527         $self->pidl("NDR_CHECK(ndr_push_subcontext_end($ndr, $subndr, $l->{HEADER_SIZE}, $subcontext_size));");
 528         $self->deindent;
 529         $self->pidl("}");
 530 }
 531
 532 sub ParseSubcontextPullStart($$$$$)
 533 {
 534         my ($self,$e,$l,$ndr,$env) = @_;
 535         my $subndr = "_ndr_$e->{NAME}";
 536         my $subcontext_size = ParseExpr($l->{SUBCONTEXT_SIZE}, $env, $e->{ORIGINAL});
 537
 538         $self->pidl("{");
 539         $self->indent;
 540         $self->pidl("struct ndr_pull *$subndr;");
 541         $self->pidl("NDR_CHECK(ndr_pull_subcontext_start($ndr, &$subndr, $l->{HEADER_SIZE}, $subcontext_size));");
 542
 543         if (defined $l->{COMPRESSION}) {
 544                 $subndr = $self->ParseCompressionPullStart($e, $l, $subndr, $env);
 545         }
 546
 547         return $subndr;
 548 }
 549
 550 sub ParseSubcontextPullEnd($$$$$)
 551 {
 552         my ($self,$e,$l,$ndr,$env) = @_;
 553         my $subndr = "_ndr_$e->{NAME}";
 554         my $subcontext_size = ParseExpr($l->{SUBCONTEXT_SIZE}, $env, $e->{ORIGINAL});
 555
 556         if (defined $l->{COMPRESSION}) {
 557                 $self->ParseCompressionPullEnd($e, $l, $subndr, $env);
 558         }
 559
 560         $self->pidl("NDR_CHECK(ndr_pull_subcontext_end($ndr, $subndr, $l->{HEADER_SIZE}, $subcontext_size));");
 561         $self->deindent;
 562         $self->pidl("}");
 563 }
 564
 565 sub ParseElementPushLevel
 566 {
 567         my ($self,$e,$l,$ndr,$var_name,$env,$primitives,$deferred) = @_;
 568
 569         my $ndr_flags = CalcNdrFlags($l, $primitives, $deferred);
 570
 571         if ($l->{TYPE} eq "ARRAY" and ($l->{IS_CONFORMANT} or $l->{IS_VARYING})) {
 572                 $var_name = get_pointer_to($var_name);
 573         }
 574
 575         if (defined($ndr_flags)) {
 576                 if ($l->{TYPE} eq "SUBCONTEXT") {
 577                         my $subndr = $self->ParseSubcontextPushStart($e, $l, $ndr, $env);
 578                         $self->ParseElementPushLevel($e, GetNextLevel($e, $l), $subndr, $var_name, $env, 1, 1);
 579                         $self->ParseSubcontextPushEnd($e, $l, $ndr, $env);
 580                 } elsif ($l->{TYPE} eq "POINTER") {
 581                         $self->ParsePtrPush($e, $l, $ndr, $var_name);
 582                 } elsif ($l->{TYPE} eq "ARRAY") {
 583                         my $length = $self->ParseArrayPushHeader($e, $l, $ndr, $var_name, $env);
 584
 585                         my $nl = GetNextLevel($e, $l);
 586
 587                         # Allow speedups for arrays of scalar types
 588                         if (is_charset_array($e,$l)) {
 589                                 $self->pidl("NDR_CHECK(ndr_push_charset($ndr, $ndr_flags, $var_name, $length, sizeof(" . mapTypeName($nl->{DATA_TYPE}) . "), CH_$e->{PROPERTIES}->{charset}));");
 590                                 return;
 591                         } elsif (has_fast_array($e,$l)) {
 592                                 $self->pidl("NDR_CHECK(ndr_push_array_$nl->{DATA_TYPE}($ndr, $ndr_flags, $var_name, $length));");
 593                                 return;
 594                         }
 595                 } elsif ($l->{TYPE} eq "SWITCH") {
 596                         $self->ParseSwitchPush($e, $l, $ndr, $var_name, $env);
 597                 } elsif ($l->{TYPE} eq "DATA") {
 598                         $self->ParseDataPush($e, $l, $ndr, $var_name, $primitives, $deferred);
 599                 } elsif ($l->{TYPE} eq "TYPEDEF") {
 600                         $typefamily{$e->{DATA}->{TYPE}}->{PUSH_FN_BODY}->($self, $e->{DATA}, $ndr, $var_name);
 601                 }
 602         }
 603
 604         if ($l->{TYPE} eq "POINTER" and $l->{POINTER_TYPE} eq "ignore") {
 605                 $self->pidl("/* [ignore] '$e->{NAME}' */");
 606         } elsif ($l->{TYPE} eq "POINTER" and $deferred) {
 607                 my $rel_var_name = $var_name;
 608                 if ($l->{POINTER_TYPE} ne "ref") {
 609                         $self->pidl("if ($var_name) {");
 610                         $self->indent;
 611                         if ($l->{POINTER_TYPE} eq "relative") {
 612                                 $self->pidl("NDR_CHECK(ndr_push_relative_ptr2_start($ndr, $rel_var_name));");
 613                         }
 614                         if ($l->{POINTER_TYPE} eq "relative_short") {
 615                                 $self->pidl("NDR_CHECK(ndr_push_short_relative_ptr2($ndr, $var_name));");
 616                         }
 617                 }
 618                 $var_name = get_value_of($var_name);
 619                 $self->ParseElementPushLevel($e, GetNextLevel($e, $l), $ndr, $var_name, $env, 1, 1);
 620
 621                 if ($l->{POINTER_TYPE} ne "ref") {
 622                         if ($l->{POINTER_TYPE} eq "relative") {
 623                                 $self->pidl("NDR_CHECK(ndr_push_relative_ptr2_end($ndr, $rel_var_name));");
 624                         }
 625                         $self->deindent;
 626                         $self->pidl("}");
 627                 }
 628         } elsif ($l->{TYPE} eq "ARRAY" and not has_fast_array($e,$l) and
 629                 not is_charset_array($e, $l)) {
 630                 my $length = ParseExpr($l->{LENGTH_IS}, $env, $e->{ORIGINAL});
 631                 my $counter = "cntr_$e->{NAME}_$l->{LEVEL_INDEX}";
 632
 633                 my $array_pointless = ($length eq "0");
 634
 635                 if ($array_pointless) {
 636                         warning($e->{ORIGINAL}, "pointless array `$e->{NAME}' will always have size 0");
 637                 }
 638
 639                 $var_name = get_array_element($var_name, $counter);
 640
 641                 if ((($primitives and not $l->{IS_DEFERRED}) or ($deferred and $l->{IS_DEFERRED})) and not $array_pointless) {
 642                         $self->pidl("for ($counter = 0; $counter < $length; $counter++) {");
 643                         $self->indent;
 644                         $self->ParseElementPushLevel($e, GetNextLevel($e, $l), $ndr, $var_name, $env, 1, 0);
 645                         $self->deindent;
 646                         $self->pidl("}");
 647                 }
 648
 649                 if ($deferred and ContainsDeferred($e, $l) and not $array_pointless) {
 650                         $self->pidl("for ($counter = 0; $counter < $length; $counter++) {");
 651                         $self->indent;
 652                         $self->ParseElementPushLevel($e, GetNextLevel($e, $l), $ndr, $var_name, $env, 0, 1);
 653                         $self->deindent;
 654                         $self->pidl("}");
 655                 }
 656         } elsif ($l->{TYPE} eq "SWITCH") {
 657                 $self->ParseElementPushLevel($e, GetNextLevel($e, $l), $ndr, $var_name, $env, $primitives, $deferred);
 658         }
 659 }
 660
 661 #####################################################################
 662 # parse scalars in a structure element
 663 sub ParseElementPush($$$$$$)
 664 {
 665         my ($self,$e,$ndr,$env,$primitives,$deferred) = @_;
 666         my $subndr = undef;
 667
 668         my $var_name = $env->{$e->{NAME}};
 669
 670         return if ContainsPipe($e, $e->{LEVELS}[0]);
 671
 672         return unless $primitives or ($deferred and ContainsDeferred($e, $e->{LEVELS}[0]));
 673
 674         # Representation type is different from transmit_as
 675         if ($e->{REPRESENTATION_TYPE} ne $e->{TYPE}) {
 676                 $self->pidl("{");
 677                 $self->indent;
 678                 my $transmit_name = "_transmit_$e->{NAME}";
 679                 $self->pidl(mapTypeName($e->{TYPE}) ." $transmit_name;");
 680                 $self->pidl("NDR_CHECK(ndr_$e->{REPRESENTATION_TYPE}_to_$e->{TYPE}($var_name, " . get_pointer_to($transmit_name) . "));");
 681                 $var_name = $transmit_name;
 682         }
 683
 684         $var_name = append_prefix($e, $var_name);
 685
 686         $self->start_flags($e, $ndr);
 687
 688         if (defined(my $value = has_property($e, "value"))) {
 689                 $var_name = ParseExpr($value, $env, $e->{ORIGINAL});
 690         }
 691
 692         $self->ParseElementPushLevel($e, $e->{LEVELS}[0], $ndr, $var_name, $env, $primitives, $deferred);
 693
 694         $self->end_flags($e, $ndr);
 695
 696         if ($e->{REPRESENTATION_TYPE} ne $e->{TYPE}) {
 697                 $self->deindent;
 698                 $self->pidl("}");
 699         }
 700 }
 701
 702 #####################################################################
 703 # parse a pointer in a struct element or function
 704 sub ParsePtrPush($$$$$)
 705 {
 706         my ($self,$e,$l,$ndr,$var_name) = @_;
 707
 708         if ($l->{POINTER_TYPE} eq "ref") {
 709                 $self->pidl("if ($var_name == NULL) {");
 710                 $self->indent;
 711                 $self->pidl("return ndr_push_error($ndr, NDR_ERR_INVALID_POINTER, \"NULL [ref] pointer\");");
 712                 $self->deindent;
 713                 $self->pidl("}");
 714                 if ($l->{LEVEL} eq "EMBEDDED") {
 715                         $self->pidl("NDR_CHECK(ndr_push_ref_ptr(ndr));");
 716                 }
 717         } elsif ($l->{POINTER_TYPE} eq "relative") {
 718                 $self->pidl("NDR_CHECK(ndr_push_relative_ptr1($ndr, $var_name));");
 719         } elsif ($l->{POINTER_TYPE} eq "relative_short") {
 720                 $self->pidl("NDR_CHECK(ndr_push_short_relative_ptr1($ndr, $var_name));");
 721         } elsif ($l->{POINTER_TYPE} eq "unique") {
 722                 $self->pidl("NDR_CHECK(ndr_push_unique_ptr($ndr, $var_name));");
 723         } elsif ($l->{POINTER_TYPE} eq "full") {
 724                 $self->pidl("NDR_CHECK(ndr_push_full_ptr($ndr, $var_name));");
 725         } elsif ($l->{POINTER_TYPE} eq "ignore") {
 726                 # We don't want this pointer to appear on the wire at all
 727                 $self->pidl("NDR_CHECK(ndr_push_uint3264(ndr, NDR_SCALARS, 0));");
 728         } else {
 729                 die("Unhandled pointer type $l->{POINTER_TYPE}");
 730         }
 731 }
 732
 733 sub need_pointer_to($$$)
 734 {
 735         my ($e, $l, $scalar_only) = @_;
 736
 737         my $t;
 738         if (ref($l->{DATA_TYPE})) {
 739                 $t = "$l->{DATA_TYPE}->{TYPE}_$l->{DATA_TYPE}->{NAME}";
 740         } else {
 741                 $t = $l->{DATA_TYPE};
 742         }
 743
 744         if (not Parse::Pidl::Typelist::is_scalar($t)) {
 745                 return 1 if $scalar_only;
 746         }
 747
 748         my $arrays = 0;
 749
 750         foreach my $tl (@{$e->{LEVELS}}) {
 751                 last if $l == $tl;
 752                 if ($tl->{TYPE} eq "ARRAY") {
 753                         $arrays++;
 754                 }
 755         }
 756
 757         if (Parse::Pidl::Typelist::scalar_is_reference($t)) {
 758                 return 1 unless $arrays;
 759         }
 760
 761         return 0;
 762 }
 763
 764 sub ParseDataPrint($$$$$)
 765 {
 766         my ($self, $e, $l, $ndr, $var_name) = @_;
 767
 768         if (not ref($l->{DATA_TYPE}) or defined($l->{DATA_TYPE}->{NAME})) {
 769
 770                 if (need_pointer_to($e, $l, 1)) {
 771                         $var_name = get_pointer_to($var_name);
 772                 }
 773
 774                 $self->pidl(TypeFunctionName("ndr_print", $l->{DATA_TYPE})."($ndr, \"$e->{NAME}\", $var_name);");
 775         } else {
 776                 $self->ParseTypePrint($l->{DATA_TYPE}, $ndr, $var_name);
 777         }
 778 }
 779
 780 #####################################################################
 781 # print scalars in a structure element
 782 sub ParseElementPrint($$$$$)
 783 {
 784         my($self, $e, $ndr, $var_name, $env) = @_;
 785
 786         return if (has_property($e, "noprint"));
 787         my $cur_depth = 0;
 788         my $ignore_depth = 0xFFFF;
 789
 790         if ($e->{REPRESENTATION_TYPE} ne $e->{TYPE}) {
 791                 $self->pidl("ndr_print_$e->{REPRESENTATION_TYPE}($ndr, \"$e->{NAME}\", $var_name);");
 792                 return;
 793         }
 794
 795         $var_name = append_prefix($e, $var_name);
 796
 797         if (defined(my $value = has_property($e, "value"))) {
 798                 $var_name = "($ndr->flags & LIBNDR_PRINT_SET_VALUES)?" . ParseExpr($value,$env, $e->{ORIGINAL}) . ":$var_name";
 799         }
 800
 801         foreach my $l (@{$e->{LEVELS}}) {
 802                 $cur_depth += 1;
 803
 804                 if ($cur_depth > $ignore_depth) {
 805                         next;
 806                 }
 807
 808                 if ($l->{TYPE} eq "POINTER") {
 809                         $self->pidl("ndr_print_ptr($ndr, \"$e->{NAME}\", $var_name);");
 810                         if ($l->{POINTER_TYPE} eq "ignore") {
 811                                 $self->pidl("/* [ignore] '$e->{NAME}' */");
 812                                 $ignore_depth = $cur_depth;
 813                                 last;
 814                         }
 815                         $self->pidl("$ndr->depth++;");
 816                         if ($l->{POINTER_TYPE} ne "ref") {
 817                                 $self->pidl("if ($var_name) {");
 818                                 $self->indent;
 819                         }
 820                         $var_name = get_value_of($var_name);
 821                 } elsif ($l->{TYPE} eq "ARRAY") {
 822                         my $length;
 823
 824                         if ($l->{IS_CONFORMANT} or $l->{IS_VARYING}) {
 825                                 $var_name = get_pointer_to($var_name);
 826                         }
 827
 828                         if ($l->{IS_ZERO_TERMINATED} and not defined($l->{LENGTH_IS})) {
 829                                 $length = "ndr_string_length($var_name, sizeof(*$var_name))";
 830                         } else {
 831                                 $length = ParseExprExt($l->{LENGTH_IS}, $env, $e->{ORIGINAL},
 832                                                         check_null_pointer($e, $env, sub { $self->pidl(shift); }, "return;"), check_fully_dereferenced($e, $env));
 833                         }
 834
 835                         if (is_charset_array($e,$l)) {
 836                                 $self->pidl("ndr_print_string($ndr, \"$e->{NAME}\", $var_name);");
 837                                 last;
 838                         } elsif (has_fast_array($e, $l)) {
 839                                 my $nl = GetNextLevel($e, $l);
 840                                 $self->pidl("ndr_print_array_$nl->{DATA_TYPE}($ndr, \"$e->{NAME}\", $var_name, $length);");
 841                                 last;
 842                         } else {
 843                                 my $counter = "cntr_$e->{NAME}_$l->{LEVEL_INDEX}";
 844
 845                                 $self->pidl("$ndr->print($ndr, \"\%s: ARRAY(\%d)\", \"$e->{NAME}\", (int)$length);");
 846                                 $self->pidl("$ndr->depth++;");
 847                                 $self->pidl("for ($counter=0;$counter<$length;$counter++) {");
 848                                 $self->indent;
 849
 850                                 $var_name = get_array_element($var_name, $counter);
 851                         }
 852                 } elsif ($l->{TYPE} eq "DATA") {
 853                         $self->ParseDataPrint($e, $l, $ndr, $var_name);
 854                 } elsif ($l->{TYPE} eq "SWITCH") {
 855                         my $switch_var = ParseExprExt($l->{SWITCH_IS}, $env, $e->{ORIGINAL},
 856                                                 check_null_pointer($e, $env, sub { $self->pidl(shift); }, "return;"), check_fully_dereferenced($e, $env));
 857                         $self->pidl("ndr_print_set_switch_value($ndr, " . get_pointer_to($var_name) . ", $switch_var);");
 858                 }
 859         }
 860
 861         foreach my $l (reverse @{$e->{LEVELS}}) {
 862                 $cur_depth -= 1;
 863
 864                 if ($cur_depth > $ignore_depth) {
 865                         next;
 866                 }
 867
 868                 if ($l->{TYPE} eq "POINTER") {
 869                         if ($l->{POINTER_TYPE} eq "ignore") {
 870                                 next;
 871                         }
 872
 873                         if ($l->{POINTER_TYPE} ne "ref") {
 874                                 $self->deindent;
 875                                 $self->pidl("}");
 876                         }
 877                         $self->pidl("$ndr->depth--;");
 878                 } elsif (($l->{TYPE} eq "ARRAY")
 879                         and not is_charset_array($e,$l)
 880                         and not has_fast_array($e,$l)) {
 881                         $self->deindent;
 882                         $self->pidl("}");
 883                         $self->pidl("$ndr->depth--;");
 884                 }
 885         }
 886 }
 887
 888 #####################################################################
 889 # parse scalars in a structure element - pull size
 890 sub ParseSwitchPull($$$$$$)
 891 {
 892         my($self,$e,$l,$ndr,$var_name,$env) = @_;
 893         my $switch_var = ParseExprExt($l->{SWITCH_IS}, $env, $e->{ORIGINAL},
 894                 check_null_pointer($e, $env, sub { $self->pidl(shift); },
 895                                    "return ndr_pull_error($ndr, NDR_ERR_INVALID_POINTER, \"NULL Pointer for switch_is()\");"),
 896                 check_fully_dereferenced($e, $env));
 897
 898         $var_name = get_pointer_to($var_name);
 899         $self->pidl("NDR_CHECK(ndr_pull_set_switch_value($ndr, $var_name, $switch_var));");
 900 }
 901
 902 #####################################################################
 903 # push switch element
 904 sub ParseSwitchPush($$$$$$)
 905 {
 906         my($self,$e,$l,$ndr,$var_name,$env) = @_;
 907         my $switch_var = ParseExprExt($l->{SWITCH_IS}, $env, $e->{ORIGINAL},
 908                 check_null_pointer($e, $env, sub { $self->pidl(shift); },
 909                                    "return ndr_push_error($ndr, NDR_ERR_INVALID_POINTER, \"NULL Pointer for switch_is()\");"),
 910                 check_fully_dereferenced($e, $env));
 911
 912         $var_name = get_pointer_to($var_name);
 913         $self->pidl("NDR_CHECK(ndr_push_set_switch_value($ndr, $var_name, $switch_var));");
 914 }
 915
 916 sub ParseDataPull($$$$$$$)
 917 {
 918         my ($self,$e,$l,$ndr,$var_name,$primitives,$deferred) = @_;
 919
 920         if (not ref($l->{DATA_TYPE}) or defined($l->{DATA_TYPE}->{NAME})) {
 921
 922                 my $ndr_flags = CalcNdrFlags($l, $primitives, $deferred);
 923
 924                 if (need_pointer_to($e, $l, 0)) {
 925                         $var_name = get_pointer_to($var_name);
 926                 }
 927
 928                 $var_name = get_pointer_to($var_name);
 929
 930                 $self->pidl("NDR_CHECK(".TypeFunctionName("ndr_pull", $l->{DATA_TYPE})."($ndr, $ndr_flags, $var_name));");
 931
 932                 my $pl = GetPrevLevel($e, $l);
 933
 934                 my $range = has_property($e, "range");
 935                 if ($range and $pl->{TYPE} ne "ARRAY") {
 936                         $var_name = get_value_of($var_name);
 937                         my $signed = Parse::Pidl::Typelist::is_signed($l->{DATA_TYPE});
 938                         my ($low, $high) = split(/,/, $range, 2);
 939                         if ($low < 0 and not $signed) {
 940                                 warning(0, "$low is invalid for the range of an unsigned type");
 941                         }
 942                         if ($low == 0 and not $signed) {
 943                                 $self->pidl("if ($var_name > $high) {");
 944                         } else {
 945                                 $self->pidl("if ($var_name < $low || $var_name > $high) {");
 946                         }
 947                         $self->pidl("\treturn ndr_pull_error($ndr, NDR_ERR_RANGE, \"value out of range\");");
 948                         $self->pidl("}");
 949                 }
 950         } else {
 951                 $self->ParseTypePull($l->{DATA_TYPE}, $ndr, $var_name, $primitives, $deferred);
 952         }
 953 }
 954
 955 sub ParseDataPush($$$$$$$)
 956 {
 957         my ($self,$e,$l,$ndr,$var_name,$primitives,$deferred) = @_;
 958
 959         if (not ref($l->{DATA_TYPE}) or defined($l->{DATA_TYPE}->{NAME})) {
 960
 961                 my $ndr_flags = CalcNdrFlags($l, $primitives, $deferred);
 962
 963                 # strings are passed by value rather than reference
 964                 if (need_pointer_to($e, $l, 1)) {
 965                         $var_name = get_pointer_to($var_name);
 966                 }
 967
 968                 $self->pidl("NDR_CHECK(".TypeFunctionName("ndr_push", $l->{DATA_TYPE})."($ndr, $ndr_flags, $var_name));");
 969         } else {
 970                 $self->ParseTypePush($l->{DATA_TYPE}, $ndr, $var_name, $primitives, $deferred);
 971         }
 972 }
 973
 974 sub CalcNdrFlags($$$)
 975 {
 976         my ($l,$primitives,$deferred) = @_;
 977
 978         my $scalars = 0;
 979         my $buffers = 0;
 980
 981         # Add NDR_SCALARS if this one is deferred
 982         # and deferreds may be pushed
 983         $scalars = 1 if ($l->{IS_DEFERRED} and $deferred);
 984
 985         # Add NDR_SCALARS if this one is not deferred and
 986         # primitives may be pushed
 987         $scalars = 1 if (!$l->{IS_DEFERRED} and $primitives);
 988
 989         # Add NDR_BUFFERS if this one contains deferred stuff
 990         # and deferreds may be pushed
 991         $buffers = 1 if ($l->{CONTAINS_DEFERRED} and $deferred);
 992
 993         return "NDR_SCALARS|NDR_BUFFERS" if ($scalars and $buffers);
 994         return "NDR_SCALARS" if ($scalars);
 995         return "NDR_BUFFERS" if ($buffers);
 996         return undef;
 997 }
 998
 999 sub ParseMemCtxPullFlags($$$$)
1000 {
1001         my ($self, $e, $l) = @_;
1002
1003         return undef unless ($l->{TYPE} eq "POINTER" or $l->{TYPE} eq "ARRAY");
1004         return undef if (($l->{TYPE} eq "POINTER") and ($l->{POINTER_TYPE} eq "ignore"));
1005
1006         return undef unless ($l->{TYPE} ne "ARRAY" or ArrayDynamicallyAllocated($e,$l));
1007         return undef if has_fast_array($e, $l);
1008         return undef if is_charset_array($e, $l);
1009
1010         my $mem_flags = "0";
1011
1012         if (($l->{TYPE} eq "POINTER") and ($l->{POINTER_TYPE} eq "ref")) {
1013                 my $nl = GetNextLevel($e, $l);
1014                 return undef if ($nl->{TYPE} eq "PIPE");
1015                 return undef if ($nl->{TYPE} eq "ARRAY");
1016                 return undef if (($nl->{TYPE} eq "DATA") and ($nl->{DATA_TYPE} eq "string"));
1017
1018                 if ($l->{LEVEL} eq "TOP") {
1019                         $mem_flags = "LIBNDR_FLAG_REF_ALLOC";
1020                 }
1021         }
1022
1023         return $mem_flags;
1024 }
1025
1026 sub ParseMemCtxPullStart($$$$$)
1027 {
1028         my ($self, $e, $l, $ndr, $ptr_name) = @_;
1029
1030         my $mem_r_ctx = "_mem_save_$e->{NAME}_$l->{LEVEL_INDEX}";
1031         my $mem_c_ctx = $ptr_name;
1032         my $mem_c_flags = $self->ParseMemCtxPullFlags($e, $l);
1033
1034         return unless defined($mem_c_flags);
1035
1036         $self->pidl("$mem_r_ctx = NDR_PULL_GET_MEM_CTX($ndr);");
1037         $self->pidl("NDR_PULL_SET_MEM_CTX($ndr, $mem_c_ctx, $mem_c_flags);");
1038 }
1039
1040 sub ParseMemCtxPullEnd($$$$)
1041 {
1042         my ($self, $e, $l, $ndr) = @_;
1043
1044         my $mem_r_ctx = "_mem_save_$e->{NAME}_$l->{LEVEL_INDEX}";
1045         my $mem_r_flags = $self->ParseMemCtxPullFlags($e, $l);
1046
1047         return unless defined($mem_r_flags);
1048
1049         $self->pidl("NDR_PULL_SET_MEM_CTX($ndr, $mem_r_ctx, $mem_r_flags);");
1050 }
1051
1052 sub CheckStringTerminator($$$$$)
1053 {
1054         my ($self,$ndr,$e,$l,$length) = @_;
1055         my $nl = GetNextLevel($e, $l);
1056
1057         # Make sure last element is zero!
1058         $self->pidl("NDR_CHECK(ndr_check_string_terminator($ndr, $length, sizeof($nl->{DATA_TYPE}_t)));");
1059 }
1060
1061 sub ParseElementPullLevel
1062 {
1063         my($self,$e,$l,$ndr,$var_name,$env,$primitives,$deferred) = @_;
1064
1065         my $ndr_flags = CalcNdrFlags($l, $primitives, $deferred);
1066         my $array_length = undef;
1067
1068         if ($l->{TYPE} eq "ARRAY" and ($l->{IS_VARYING} or $l->{IS_CONFORMANT})) {
1069                 $var_name = get_pointer_to($var_name);
1070         }
1071
1072         # Only pull something if there's actually something to be pulled
1073         if (defined($ndr_flags)) {
1074                 if ($l->{TYPE} eq "SUBCONTEXT") {
1075                         my $subndr = $self->ParseSubcontextPullStart($e, $l, $ndr, $env);
1076                         $self->ParseElementPullLevel($e, GetNextLevel($e,$l), $subndr, $var_name, $env, 1, 1);
1077                         $self->ParseSubcontextPullEnd($e, $l, $ndr, $env);
1078                 } elsif ($l->{TYPE} eq "ARRAY") {
1079                         my $length = $self->ParseArrayPullHeader($e, $l, $ndr, $var_name, $env);
1080                         $array_length = $length;
1081
1082                         if (my $range = has_property($e, "range")) {
1083                                 my ($low, $high) = split(/,/, $range, 2);
1084                                 if ($low < 0) {
1085                                         warning(0, "$low is invalid for the range of an array size");
1086                                 }
1087                                 if ($low == 0) {
1088                                         $self->pidl("if ($length > $high) {");
1089                                 } else {
1090                                         $self->pidl("if ($length < $low || $length > $high) {");
1091                                 }
1092                                 $self->pidl("\treturn ndr_pull_error($ndr, NDR_ERR_RANGE, \"value out of range\");");
1093                                 $self->pidl("}");
1094                         }
1095
1096                         my $nl = GetNextLevel($e, $l);
1097
1098                         if (is_charset_array($e,$l)) {
1099                                 if ($l->{IS_ZERO_TERMINATED}) {
1100                                         $self->CheckStringTerminator($ndr, $e, $l, $length);
1101                                 }
1102                                 if ($l->{IS_TO_NULL}) {
1103                                         $self->pidl("NDR_CHECK(ndr_pull_charset_to_null($ndr, $ndr_flags, ".get_pointer_to($var_name).", $length, sizeof(" . mapTypeName($nl->{DATA_TYPE}) . "), CH_$e->{PROPERTIES}->{charset}));");
1104                                 } else {
1105                                         $self->pidl("NDR_CHECK(ndr_pull_charset($ndr, $ndr_flags, ".get_pointer_to($var_name).", $length, sizeof(" . mapTypeName($nl->{DATA_TYPE}) . "), CH_$e->{PROPERTIES}->{charset}));");
1106                                 }
1107                                 return;
1108                         } elsif (has_fast_array($e, $l)) {
1109                                 if ($l->{IS_ZERO_TERMINATED}) {
1110                                         $self->CheckStringTerminator($ndr,$e,$l,$length);
1111                                 }
1112                                 $self->pidl("NDR_CHECK(ndr_pull_array_$nl->{DATA_TYPE}($ndr, $ndr_flags, $var_name, $length));");
1113                                 return;
1114                         }
1115                 } elsif ($l->{TYPE} eq "POINTER") {
1116                         $self->ParsePtrPull($e, $l, $ndr, $var_name);
1117                 } elsif ($l->{TYPE} eq "SWITCH") {
1118                         $self->ParseSwitchPull($e, $l, $ndr, $var_name, $env);
1119                 } elsif ($l->{TYPE} eq "DATA") {
1120                         $self->ParseDataPull($e, $l, $ndr, $var_name, $primitives, $deferred);
1121                 } elsif ($l->{TYPE} eq "TYPEDEF") {
1122                         $typefamily{$e->{DATA}->{TYPE}}->{PULL_FN_BODY}->($self, $e->{DATA}, $ndr, $var_name);
1123                 }
1124         }
1125
1126         # add additional constructions
1127         if ($l->{TYPE} eq "POINTER" and $l->{POINTER_TYPE} eq "ignore") {
1128                 $self->pidl("/* [ignore] '$e->{NAME}' */");
1129         } elsif ($l->{TYPE} eq "POINTER" and $deferred) {
1130                 if ($l->{POINTER_TYPE} ne "ref") {
1131                         $self->pidl("if ($var_name) {");
1132                         $self->indent;
1133
1134                         if ($l->{POINTER_TYPE} eq "relative" or $l->{POINTER_TYPE} eq "relative_short") {
1135                                 $self->pidl("uint32_t _relative_save_offset;");
1136                                 $self->pidl("_relative_save_offset = $ndr->offset;");
1137                                 $self->pidl("NDR_CHECK(ndr_pull_relative_ptr2($ndr, $var_name));");
1138                         }
1139                 }
1140
1141                 $self->ParseMemCtxPullStart($e, $l, $ndr, $var_name);
1142
1143                 $var_name = get_value_of($var_name);
1144                 $self->ParseElementPullLevel($e, GetNextLevel($e,$l), $ndr, $var_name, $env, 1, 1);
1145
1146                 $self->ParseMemCtxPullEnd($e, $l, $ndr);
1147
1148                 if ($l->{POINTER_TYPE} ne "ref") {
1149                         if ($l->{POINTER_TYPE} eq "relative") {
1150                                 $self->pidl("if ($ndr->offset > $ndr->relative_highest_offset) {");
1151                                 $self->indent;
1152                                 $self->pidl("$ndr->relative_highest_offset = $ndr->offset;");
1153                                 $self->deindent;
1154                                 $self->pidl("}");
1155                                 $self->pidl("$ndr->offset = _relative_save_offset;");
1156                         }
1157                         $self->deindent;
1158                         $self->pidl("}");
1159                 }
1160         } elsif ($l->{TYPE} eq "ARRAY" and
1161                         not has_fast_array($e,$l) and not is_charset_array($e, $l)) {
1162                 my $length = $array_length;
1163                 my $counter = "cntr_$e->{NAME}_$l->{LEVEL_INDEX}";
1164                 my $array_name = $var_name;
1165
1166                 if (not defined($length)) {
1167                         $length = $self->ParseArrayPullGetLength($e, $l, $ndr, $var_name, $env);
1168                 }
1169
1170                 if (my $range = has_property($e, "range")) {
1171                         my ($low, $high) = split(/,/, $range, 2);
1172                         if ($low < 0) {
1173                                 warning(0, "$low is invalid for the range of an array size");
1174                         }
1175                         if ($low == 0) {
1176                                 $self->pidl("if ($length > $high) {");
1177                         } else {
1178                                 $self->pidl("if ($length < $low || $length > $high) {");
1179                         }
1180                         $self->pidl("\treturn ndr_pull_error($ndr, NDR_ERR_RANGE, \"value out of range\");");
1181                         $self->pidl("}");
1182                 }
1183
1184                 $var_name = get_array_element($var_name, $counter);
1185
1186                 $self->ParseMemCtxPullStart($e, $l, $ndr, $array_name);
1187
1188                 if (($primitives and not $l->{IS_DEFERRED}) or ($deferred and $l->{IS_DEFERRED})) {
1189                         my $nl = GetNextLevel($e,$l);
1190
1191                         if ($l->{IS_ZERO_TERMINATED}) {
1192                                 $self->CheckStringTerminator($ndr,$e,$l,$length);
1193                         }
1194
1195                         $self->pidl("for ($counter = 0; $counter < $length; $counter++) {");
1196                         $self->indent;
1197                         $self->ParseElementPullLevel($e, $nl, $ndr, $var_name, $env, 1, 0);
1198                         $self->deindent;
1199                         $self->pidl("}");
1200                 }
1201
1202                 if ($deferred and ContainsDeferred($e, $l)) {
1203                         $self->pidl("for ($counter = 0; $counter < $length; $counter++) {");
1204                         $self->indent;
1205                         $self->ParseElementPullLevel($e,GetNextLevel($e,$l), $ndr, $var_name, $env, 0, 1);
1206                         $self->deindent;
1207                         $self->pidl("}");
1208                 }
1209
1210                 $self->ParseMemCtxPullEnd($e, $l, $ndr);
1211
1212         } elsif ($l->{TYPE} eq "SWITCH") {
1213                 $self->ParseElementPullLevel($e, GetNextLevel($e,$l), $ndr, $var_name, $env, $primitives, $deferred);
1214         }
1215 }
1216
1217 #####################################################################
1218 # parse scalars in a structure element - pull size
1219 sub ParseElementPull($$$$$$)
1220 {
1221         my($self,$e,$ndr,$env,$primitives,$deferred) = @_;
1222
1223         my $var_name = $env->{$e->{NAME}};
1224         my $represent_name;
1225         my $transmit_name;
1226
1227         return if ContainsPipe($e, $e->{LEVELS}[0]);
1228
1229         return unless $primitives or ($deferred and ContainsDeferred($e, $e->{LEVELS}[0]));
1230
1231         if ($e->{REPRESENTATION_TYPE} ne $e->{TYPE}) {
1232                 $self->pidl("{");
1233                 $self->indent;
1234                 $represent_name = $var_name;
1235                 $transmit_name = "_transmit_$e->{NAME}";
1236                 $var_name = $transmit_name;
1237                 $self->pidl(mapTypeName($e->{TYPE})." $var_name;");
1238         }
1239
1240         $var_name = append_prefix($e, $var_name);
1241
1242         $self->start_flags($e, $ndr);
1243
1244         $self->ParseElementPullLevel($e,$e->{LEVELS}[0],$ndr,$var_name,$env,$primitives,$deferred);
1245
1246         $self->end_flags($e, $ndr);
1247
1248         # Representation type is different from transmit_as
1249         if ($e->{REPRESENTATION_TYPE} ne $e->{TYPE}) {
1250                 $self->pidl("NDR_CHECK(ndr_$e->{TYPE}_to_$e->{REPRESENTATION_TYPE}($transmit_name, ".get_pointer_to($represent_name)."));");
1251                 $self->deindent;
1252                 $self->pidl("}");
1253         }
1254 }
1255
1256 #####################################################################
1257 # parse a pointer in a struct element or function
1258 sub ParsePtrPull($$$$$)
1259 {
1260         my($self, $e,$l,$ndr,$var_name) = @_;
1261
1262         my $nl = GetNextLevel($e, $l);
1263         my $next_is_array = ($nl->{TYPE} eq "ARRAY");
1264         my $next_is_string = (($nl->{TYPE} eq "DATA") and
1265                                                  ($nl->{DATA_TYPE} eq "string"));
1266
1267         if ($l->{POINTER_TYPE} eq "ref" and $l->{LEVEL} eq "TOP") {
1268
1269                 if (!$next_is_array and !$next_is_string) {
1270                         $self->pidl("if ($ndr->flags & LIBNDR_FLAG_REF_ALLOC) {");
1271                         $self->pidl("\tNDR_PULL_ALLOC($ndr, $var_name);");
1272                         $self->pidl("}");
1273                 }
1274
1275                 return;
1276         } elsif ($l->{POINTER_TYPE} eq "ref" and $l->{LEVEL} eq "EMBEDDED") {
1277                 $self->pidl("NDR_CHECK(ndr_pull_ref_ptr($ndr, &_ptr_$e->{NAME}));");
1278         } elsif (($l->{POINTER_TYPE} eq "unique") or
1279                  ($l->{POINTER_TYPE} eq "relative") or
1280                  ($l->{POINTER_TYPE} eq "full")) {
1281                 $self->pidl("NDR_CHECK(ndr_pull_generic_ptr($ndr, &_ptr_$e->{NAME}));");
1282         } elsif ($l->{POINTER_TYPE} eq "relative_short") {
1283                 $self->pidl("NDR_CHECK(ndr_pull_relative_ptr_short($ndr, &_ptr_$e->{NAME}));");
1284         } elsif ($l->{POINTER_TYPE} eq "ignore") {
1285                 #We want to consume the pointer bytes, but ignore the pointer value
1286                 $self->pidl("NDR_CHECK(ndr_pull_uint3264(ndr, NDR_SCALARS, &_ptr_$e->{NAME}));");
1287                 $self->pidl("_ptr_$e->{NAME} = 0;");
1288         } else {
1289                 die("Unhandled pointer type $l->{POINTER_TYPE}");
1290         }
1291
1292         $self->pidl("if (_ptr_$e->{NAME}) {");
1293         $self->indent;
1294
1295         if ($l->{POINTER_TYPE} eq "ignore") {
1296                 # Don't do anything, we don't want to do the
1297                 # allocation, as we forced it to NULL just above, and
1298                 # we may not know the declared type anyway.
1299         } else {
1300                 # Don't do this for arrays, they're allocated at the actual level
1301                 # of the array
1302                 unless ($next_is_array or $next_is_string) {
1303                        $self->pidl("NDR_PULL_ALLOC($ndr, $var_name);");
1304                 } else {
1305                        # FIXME: Yes, this is nasty.
1306                        # We allocate an array twice
1307                        # - once just to indicate that it's there,
1308                        # - then the real allocation...
1309                        $self->pidl("NDR_PULL_ALLOC($ndr, $var_name);");
1310                 }
1311         }
1312
1313         #$self->pidl("memset($var_name, 0, sizeof($var_name));");
1314         if ($l->{POINTER_TYPE} eq "relative" or $l->{POINTER_TYPE} eq "relative_short") {
1315                 $self->pidl("NDR_CHECK(ndr_pull_relative_ptr1($ndr, $var_name, _ptr_$e->{NAME}));");
1316         }
1317         $self->deindent;
1318         $self->pidl("} else {");
1319         $self->pidl("\t$var_name = NULL;");
1320         $self->pidl("}");
1321 }
1322
1323 sub ParseStructPushPrimitives($$$$$)
1324 {
1325         my ($self, $struct, $ndr, $varname, $env) = @_;
1326
1327         # see if the structure contains a conformant array. If it
1328         # does, then it must be the last element of the structure, and
1329         # we need to push the conformant length early, as it fits on
1330         # the wire before the structure (and even before the structure
1331         # alignment)
1332         if (defined($struct->{SURROUNDING_ELEMENT})) {
1333                 my $e = $struct->{SURROUNDING_ELEMENT};
1334
1335                 if (defined($e->{LEVELS}[0]) and
1336                         $e->{LEVELS}[0]->{TYPE} eq "ARRAY") {
1337                         my $size;
1338
1339                         if ($e->{LEVELS}[0]->{IS_ZERO_TERMINATED}) {
1340                                 if (has_property($e, "charset")) {
1341                                         $size = "ndr_charset_length($varname->$e->{NAME}, CH_$e->{PROPERTIES}->{charset})";
1342                                 } else {
1343                                         $size = "ndr_string_length($varname->$e->{NAME}, sizeof(*$varname->$e->{NAME}))";
1344                                 }
1345                                 if (defined($e->{LEVELS}[0]->{SIZE_IS})) {
1346                                         $size = ParseExpr($e->{LEVELS}[0]->{SIZE_IS}, $env, $e->{ORIGINAL});
1347                                 }
1348                         } else {
1349                                 $size = ParseExpr($e->{LEVELS}[0]->{SIZE_IS}, $env, $e->{ORIGINAL});
1350                         }
1351
1352                         $self->pidl("NDR_CHECK(ndr_push_uint3264($ndr, NDR_SCALARS, $size));");
1353                 } else {
1354                         $self->pidl("NDR_CHECK(ndr_push_uint3264($ndr, NDR_SCALARS, ndr_string_array_size($ndr, $varname->$e->{NAME})));");
1355                 }
1356         }
1357
1358         $self->pidl("NDR_CHECK(ndr_push_align($ndr, $struct->{ALIGN}));");
1359
1360         if (defined($struct->{PROPERTIES}{relative_base})) {
1361                 # set the current offset as base for relative pointers
1362                 # and store it based on the toplevel struct/union
1363                 $self->pidl("NDR_CHECK(ndr_push_setup_relative_base_offset1($ndr, $varname, $ndr->offset));");
1364         }
1365
1366         $self->ParseElementPush($_, $ndr, $env, 1, 0) foreach (@{$struct->{ELEMENTS}});
1367
1368         $self->pidl("NDR_CHECK(ndr_push_trailer_align($ndr, $struct->{ALIGN}));");
1369 }
1370
1371 sub ParseStructPushDeferred($$$$)
1372 {
1373         my ($self, $struct, $ndr, $varname, $env) = @_;
1374         if (defined($struct->{PROPERTIES}{relative_base})) {
1375                 # retrieve the current offset as base for relative pointers
1376                 # based on the toplevel struct/union
1377                 $self->pidl("NDR_CHECK(ndr_push_setup_relative_base_offset2($ndr, $varname));");
1378         }
1379         $self->ParseElementPush($_, $ndr, $env, 0, 1) foreach (@{$struct->{ELEMENTS}});
1380 }
1381
1382 #####################################################################
1383 # parse a struct
1384 sub ParseStructPush($$$$)
1385 {
1386         my ($self, $struct, $ndr, $varname) = @_;
1387
1388         return unless defined($struct->{ELEMENTS});
1389
1390         my $env = GenerateStructEnv($struct, $varname);
1391
1392         EnvSubstituteValue($env, $struct);
1393
1394         $self->DeclareArrayVariablesNoZero($_, $env) foreach (@{$struct->{ELEMENTS}});
1395
1396         $self->start_flags($struct, $ndr);
1397
1398         $self->pidl("NDR_PUSH_CHECK_FLAGS(ndr, ndr_flags);");
1399         $self->pidl("if (ndr_flags & NDR_SCALARS) {");
1400         $self->indent;
1401         $self->ParseStructPushPrimitives($struct, $ndr, $varname, $env);
1402         $self->deindent;
1403         $self->pidl("}");
1404
1405         $self->pidl("if (ndr_flags & NDR_BUFFERS) {");
1406         $self->indent;
1407         $self->ParseStructPushDeferred($struct, $ndr, $varname, $env);
1408         $self->deindent;
1409         $self->pidl("}");
1410
1411         $self->end_flags($struct, $ndr);
1412 }
1413
1414 #####################################################################
1415 # generate a push function for an enum
1416 sub ParseEnumPush($$$$)
1417 {
1418         my($self,$enum,$ndr,$varname) = @_;
1419         my($type_fn) = $enum->{BASE_TYPE};
1420
1421         $self->start_flags($enum, $ndr);
1422         $self->pidl("NDR_CHECK(ndr_push_enum_$type_fn($ndr, NDR_SCALARS, $varname));");
1423         $self->end_flags($enum, $ndr);
1424 }
1425
1426 #####################################################################
1427 # generate a pull function for an enum
1428 sub ParseEnumPull($$$$)
1429 {
1430         my($self,$enum,$ndr,$varname) = @_;
1431         my($type_fn) = $enum->{BASE_TYPE};
1432         my($type_v_decl) = mapTypeName($type_fn);
1433
1434         $self->pidl("$type_v_decl v;");
1435         $self->start_flags($enum, $ndr);
1436         $self->pidl("NDR_CHECK(ndr_pull_enum_$type_fn($ndr, NDR_SCALARS, &v));");
1437         $self->pidl("*$varname = v;");
1438
1439         $self->end_flags($enum, $ndr);
1440 }
1441
1442 #####################################################################
1443 # generate a print function for an enum
1444 sub ParseEnumPrint($$$$$)
1445 {
1446         my($self,$enum,$ndr,$name,$varname) = @_;
1447
1448         $self->pidl("const char *val = NULL;");
1449         $self->pidl("");
1450
1451         $self->start_flags($enum, $ndr);
1452
1453         $self->pidl("switch ($varname) {");
1454         $self->indent;
1455         my $els = \@{$enum->{ELEMENTS}};
1456         foreach my $i (0 .. $#{$els}) {
1457                 my $e = ${$els}[$i];
1458                 chomp $e;
1459                 if ($e =~ /^(.*)=/) {
1460                         $e = $1;
1461                 }
1462                 $self->pidl("case $e: val = \"$e\"; break;");
1463         }
1464
1465         $self->deindent;
1466         $self->pidl("}");
1467
1468         $self->pidl("ndr_print_enum($ndr, name, \"$enum->{TYPE}\", val, $varname);");
1469
1470         $self->end_flags($enum, $ndr);
1471 }
1472
1473 sub DeclEnum($$$$)
1474 {
1475         my ($e,$t,$name,$varname) = @_;
1476         return "enum $name " .
1477                 ($t eq "pull"?"*":"") . $varname;
1478 }
1479
1480 $typefamily{ENUM} = {
1481         DECL => \&DeclEnum,
1482         PUSH_FN_BODY => \&ParseEnumPush,
1483         PULL_FN_BODY => \&ParseEnumPull,
1484         PRINT_FN_BODY => \&ParseEnumPrint,
1485 };
1486
1487 #####################################################################
1488 # generate a push function for a bitmap
1489 sub ParseBitmapPush($$$$)
1490 {
1491         my($self,$bitmap,$ndr,$varname) = @_;
1492         my($type_fn) = $bitmap->{BASE_TYPE};
1493
1494         $self->start_flags($bitmap, $ndr);
1495
1496         $self->pidl("NDR_CHECK(ndr_push_$type_fn($ndr, NDR_SCALARS, $varname));");
1497
1498         $self->end_flags($bitmap, $ndr);
1499 }
1500
1501 #####################################################################
1502 # generate a pull function for an bitmap
1503 sub ParseBitmapPull($$$$)
1504 {
1505         my($self,$bitmap,$ndr,$varname) = @_;
1506         my $type_fn = $bitmap->{BASE_TYPE};
1507         my($type_decl) = mapTypeName($bitmap->{BASE_TYPE});
1508
1509         $self->pidl("$type_decl v;");
1510         $self->start_flags($bitmap, $ndr);
1511         $self->pidl("NDR_CHECK(ndr_pull_$type_fn($ndr, NDR_SCALARS, &v));");
1512         $self->pidl("*$varname = v;");
1513
1514         $self->end_flags($bitmap, $ndr);
1515 }
1516
1517 #####################################################################
1518 # generate a print function for an bitmap
1519 sub ParseBitmapPrintElement($$$$$$)
1520 {
1521         my($self,$e,$bitmap,$ndr,$name,$varname) = @_;
1522         my($type_decl) = mapTypeName($bitmap->{BASE_TYPE});
1523         my($type_fn) = $bitmap->{BASE_TYPE};
1524         my($flag);
1525
1526         if ($e =~ /^(\w+) .*$/) {
1527                 $flag = "$1";
1528         } else {
1529                 die "Bitmap: \"$name\" invalid Flag: \"$e\"";
1530         }
1531
1532         $self->pidl("ndr_print_bitmap_flag($ndr, sizeof($type_decl), \"$flag\", $flag, $varname);");
1533 }
1534
1535 #####################################################################
1536 # generate a print function for an bitmap
1537 sub ParseBitmapPrint($$$$$)
1538 {
1539         my($self,$bitmap,$ndr,$name,$varname) = @_;
1540         my($type_decl) = mapTypeName($bitmap->{TYPE});
1541         my($type_fn) = $bitmap->{BASE_TYPE};
1542
1543         $self->start_flags($bitmap, $ndr);
1544
1545         $self->pidl("ndr_print_$type_fn($ndr, name, $varname);");
1546
1547         $self->pidl("$ndr->depth++;");
1548         foreach my $e (@{$bitmap->{ELEMENTS}}) {
1549                 $self->ParseBitmapPrintElement($e, $bitmap, $ndr, $name, $varname);
1550         }
1551         $self->pidl("$ndr->depth--;");
1552
1553         $self->end_flags($bitmap, $ndr);
1554 }
1555
1556 sub DeclBitmap($$$$)
1557 {
1558         my ($e,$t,$name,$varname) = @_;
1559         return mapTypeName(Parse::Pidl::Typelist::bitmap_type_fn($e)) .
1560                 ($t eq "pull"?" *":" ") . $varname;
1561 }
1562
1563 $typefamily{BITMAP} = {
1564         DECL => \&DeclBitmap,
1565         PUSH_FN_BODY => \&ParseBitmapPush,
1566         PULL_FN_BODY => \&ParseBitmapPull,
1567         PRINT_FN_BODY => \&ParseBitmapPrint,
1568 };
1569
1570 #####################################################################
1571 # generate a struct print function
1572 sub ParseStructPrint($$$$$)
1573 {
1574         my($self,$struct,$ndr,$name,$varname) = @_;
1575
1576         return unless defined $struct->{ELEMENTS};
1577
1578         my $env = GenerateStructEnv($struct, $varname);
1579
1580         $self->DeclareArrayVariables($_) foreach (@{$struct->{ELEMENTS}});
1581
1582         $self->pidl("ndr_print_struct($ndr, name, \"$name\");");
1583         $self->pidl("if (r == NULL) { ndr_print_null($ndr); return; }");
1584
1585         $self->start_flags($struct, $ndr);
1586
1587         $self->pidl("$ndr->depth++;");
1588
1589         $self->ParseElementPrint($_, $ndr, $env->{$_->{NAME}}, $env)
1590                 foreach (@{$struct->{ELEMENTS}});
1591         $self->pidl("$ndr->depth--;");
1592
1593         $self->end_flags($struct, $ndr);
1594 }
1595
1596 sub DeclarePtrVariables($$)
1597 {
1598         my ($self,$e) = @_;
1599         foreach my $l (@{$e->{LEVELS}}) {
1600                 my $size = 32;
1601                 if ($l->{TYPE} eq "POINTER" and
1602                         not ($l->{POINTER_TYPE} eq "ref" and $l->{LEVEL} eq "TOP")) {
1603                         if ($l->{POINTER_TYPE} eq "relative_short") {
1604                                 $size = 16;
1605                         }
1606                         $self->pidl("uint${size}_t _ptr_$e->{NAME};");
1607                         last;
1608                 }
1609         }
1610 }
1611
1612 sub DeclareArrayVariables($$)
1613 {
1614         my ($self,$e) = @_;
1615
1616         foreach my $l (@{$e->{LEVELS}}) {
1617                 next if ($l->{TYPE} ne "ARRAY");
1618                 next if has_fast_array($e,$l);
1619                 next if is_charset_array($e,$l);
1620                 $self->pidl("uint32_t cntr_$e->{NAME}_$l->{LEVEL_INDEX};");
1621         }
1622 }
1623
1624 sub DeclareArrayVariablesNoZero($$$)
1625 {
1626         my ($self,$e,$env) = @_;
1627
1628         foreach my $l (@{$e->{LEVELS}}) {
1629                 next if ($l->{TYPE} ne "ARRAY");
1630                 next if has_fast_array($e,$l);
1631                 next if is_charset_array($e,$l);
1632                 my $length = ParseExpr($l->{LENGTH_IS}, $env, $e->{ORIGINAL});
1633                 if ($length eq "0") {
1634                         warning($e->{ORIGINAL}, "pointless array cntr: 'cntr_$e->{NAME}_$l->{LEVEL_INDEX}': length=$length");
1635                 } else {
1636                         $self->pidl("uint32_t cntr_$e->{NAME}_$l->{LEVEL_INDEX};");
1637                 }
1638         }
1639 }
1640
1641 sub DeclareMemCtxVariables($$)
1642 {
1643         my ($self,$e) = @_;
1644         foreach my $l (@{$e->{LEVELS}}) {
1645                 my $mem_flags = $self->ParseMemCtxPullFlags($e, $l);
1646
1647                 if (($l->{TYPE} eq "POINTER") and ($l->{POINTER_TYPE} eq "ignore")) {
1648                         last;
1649                 }
1650
1651                 if (defined($mem_flags)) {
1652                         $self->pidl("TALLOC_CTX *_mem_save_$e->{NAME}_$l->{LEVEL_INDEX};");
1653                 }
1654         }
1655 }
1656
1657 sub ParseStructPullPrimitives($$$$$)
1658 {
1659         my($self,$struct,$ndr,$varname,$env) = @_;
1660
1661         if (defined $struct->{SURROUNDING_ELEMENT}) {
1662                 $self->pidl("NDR_CHECK(ndr_pull_array_size($ndr, &$varname->$struct->{SURROUNDING_ELEMENT}->{NAME}));");
1663         }
1664
1665         $self->pidl("NDR_CHECK(ndr_pull_align($ndr, $struct->{ALIGN}));");
1666
1667         if (defined($struct->{PROPERTIES}{relative_base})) {
1668                 # set the current offset as base for relative pointers
1669                 # and store it based on the toplevel struct/union
1670                 $self->pidl("NDR_CHECK(ndr_pull_setup_relative_base_offset1($ndr, $varname, $ndr->offset));");
1671         }
1672
1673         $self->ParseElementPull($_, $ndr, $env, 1, 0) foreach (@{$struct->{ELEMENTS}});
1674
1675         $self->add_deferred();
1676
1677         $self->pidl("NDR_CHECK(ndr_pull_trailer_align($ndr, $struct->{ALIGN}));");
1678 }
1679
1680 sub ParseStructPullDeferred($$$$$)
1681 {
1682         my ($self,$struct,$ndr,$varname,$env) = @_;
1683
1684         if (defined($struct->{PROPERTIES}{relative_base})) {
1685                 # retrieve the current offset as base for relative pointers
1686                 # based on the toplevel struct/union
1687                 $self->pidl("NDR_CHECK(ndr_pull_setup_relative_base_offset2($ndr, $varname));");
1688         }
1689         foreach my $e (@{$struct->{ELEMENTS}}) {
1690                 $self->ParseElementPull($e, $ndr, $env, 0, 1);
1691         }
1692
1693         $self->add_deferred();
1694 }
1695
1696 #####################################################################
1697 # parse a struct - pull side
1698 sub ParseStructPull($$$$)
1699 {
1700         my($self,$struct,$ndr,$varname) = @_;
1701
1702         return unless defined $struct->{ELEMENTS};
1703
1704         # declare any internal pointers we need
1705         foreach my $e (@{$struct->{ELEMENTS}}) {
1706                 $self->DeclarePtrVariables($e);
1707                 $self->DeclareArrayVariables($e);
1708                 $self->DeclareMemCtxVariables($e);
1709         }
1710
1711         $self->start_flags($struct, $ndr);
1712
1713         my $env = GenerateStructEnv($struct, $varname);
1714
1715         $self->pidl("NDR_PULL_CHECK_FLAGS(ndr, ndr_flags);");
1716         $self->pidl("if (ndr_flags & NDR_SCALARS) {");
1717         $self->indent;
1718         $self->ParseStructPullPrimitives($struct,$ndr,$varname,$env);
1719         $self->deindent;
1720         $self->pidl("}");
1721         $self->pidl("if (ndr_flags & NDR_BUFFERS) {");
1722         $self->indent;
1723         $self->ParseStructPullDeferred($struct,$ndr,$varname,$env);
1724         $self->deindent;
1725         $self->pidl("}");
1726
1727         $self->end_flags($struct, $ndr);
1728 }
1729
1730 #####################################################################
1731 # calculate size of ndr struct
1732 sub ParseStructNdrSize($$$$)
1733 {
1734         my ($self,$t, $name, $varname) = @_;
1735         my $sizevar;
1736
1737         if (my $flags = has_property($t, "flag")) {
1738                 $self->pidl("flags |= $flags;");
1739         }
1740         $self->pidl("return ndr_size_struct($varname, flags, (ndr_push_flags_fn_t)ndr_push_$name);");
1741 }
1742
1743 sub DeclStruct($$$$)
1744 {
1745         my ($e,$t,$name,$varname) = @_;
1746         return ($t ne "pull"?"const ":"") . "struct $name *$varname";
1747 }
1748
1749 sub ArgsStructNdrSize($$$)
1750 {
1751         my ($d, $name, $varname) = @_;
1752         return "const struct $name *$varname, int flags";
1753 }
1754
1755 $typefamily{STRUCT} = {
1756         PUSH_FN_BODY => \&ParseStructPush,
1757         DECL => \&DeclStruct,
1758         PULL_FN_BODY => \&ParseStructPull,
1759         PRINT_FN_BODY => \&ParseStructPrint,
1760         SIZE_FN_BODY => \&ParseStructNdrSize,
1761         SIZE_FN_ARGS => \&ArgsStructNdrSize,
1762 };
1763
1764 #####################################################################
1765 # calculate size of ndr struct
1766 sub ParseUnionNdrSize($$$)
1767 {
1768         my ($self, $t, $name, $varname) = @_;
1769         my $sizevar;
1770
1771         if (my $flags = has_property($t, "flag")) {
1772                 $self->pidl("flags |= $flags;");
1773         }
1774
1775         $self->pidl("return ndr_size_union($varname, flags, level, (ndr_push_flags_fn_t)ndr_push_$name);");
1776 }
1777
1778 sub ParseUnionPushPrimitives($$$$)
1779 {
1780         my ($self, $e, $ndr ,$varname) = @_;
1781
1782         my $have_default = 0;
1783
1784         $self->pidl("uint32_t level = ndr_push_get_switch_value($ndr, $varname);");
1785
1786         if (defined($e->{SWITCH_TYPE})) {
1787                 if (defined($e->{ALIGN})) {
1788                         $self->pidl("NDR_CHECK(ndr_push_union_align($ndr, $e->{ALIGN}));");
1789                 }
1790
1791                 $self->pidl("NDR_CHECK(ndr_push_$e->{SWITCH_TYPE}($ndr, NDR_SCALARS, level));");
1792         }
1793
1794         if (defined($e->{ALIGN})) {
1795                 if ($e->{IS_MS_UNION}) {
1796                         $self->pidl("/* ms_union is always aligned to the largest union arm*/");
1797                         $self->pidl("NDR_CHECK(ndr_push_align($ndr, $e->{ALIGN}));");
1798                 } else {
1799                         $self->pidl("NDR_CHECK(ndr_push_union_align($ndr, $e->{ALIGN}));");
1800                 }
1801         }
1802
1803         $self->pidl("switch (level) {");
1804         $self->indent;
1805         foreach my $el (@{$e->{ELEMENTS}}) {
1806                 if ($el->{CASE} eq "default") {
1807                         $have_default = 1;
1808                 }
1809                 $self->pidl("$el->{CASE}: {");
1810
1811                 if ($el->{TYPE} ne "EMPTY") {
1812                         $self->indent;
1813                         if (defined($e->{PROPERTIES}{relative_base})) {
1814                                 $self->pidl("NDR_CHECK(ndr_push_align($ndr, $el->{ALIGN}));");
1815                                 # set the current offset as base for relative pointers
1816                                 # and store it based on the toplevel struct/union
1817                                 $self->pidl("NDR_CHECK(ndr_push_setup_relative_base_offset1($ndr, $varname, $ndr->offset));");
1818                         }
1819                         $self->DeclareArrayVariables($el);
1820                         $self->ParseElementPush($el, $ndr, {$el->{NAME} => "$varname->$el->{NAME}"}, 1, 0);
1821                         $self->deindent;
1822                 }
1823                 $self->pidl("break; }");
1824                 $self->pidl("");
1825         }
1826         if (! $have_default) {
1827                 $self->pidl("default:");
1828                 $self->pidl("\treturn ndr_push_error($ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u at \%s\", level, __location__);");
1829         }
1830         $self->deindent;
1831         $self->pidl("}");
1832 }
1833
1834 sub ParseUnionPushDeferred($$$$)
1835 {
1836         my ($self,$e,$ndr,$varname) = @_;
1837
1838         my $have_default = 0;
1839
1840         $self->pidl("uint32_t level = ndr_push_get_switch_value($ndr, $varname);");
1841         if (defined($e->{PROPERTIES}{relative_base})) {
1842                 # retrieve the current offset as base for relative pointers
1843                 # based on the toplevel struct/union
1844                 $self->pidl("NDR_CHECK(ndr_push_setup_relative_base_offset2($ndr, $varname));");
1845         }
1846         $self->pidl("switch (level) {");
1847         $self->indent;
1848         foreach my $el (@{$e->{ELEMENTS}}) {
1849                 if ($el->{CASE} eq "default") {
1850                         $have_default = 1;
1851                 }
1852
1853                 $self->pidl("$el->{CASE}:");
1854                 if ($el->{TYPE} ne "EMPTY") {
1855                         $self->indent;
1856                         $self->ParseElementPush($el, $ndr, {$el->{NAME} => "$varname->$el->{NAME}"}, 0, 1);
1857                         $self->deindent;
1858                 }
1859                 $self->pidl("break;");
1860                 $self->pidl("");
1861         }
1862         if (! $have_default) {
1863                 $self->pidl("default:");
1864                 $self->pidl("\treturn ndr_push_error($ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u at \%s\", level, __location__);");
1865         }
1866         $self->deindent;
1867         $self->pidl("}");
1868 }
1869
1870 #####################################################################
1871 # parse a union - push side
1872 sub ParseUnionPush($$$$)
1873 {
1874         my ($self,$e,$ndr,$varname) = @_;
1875         my $have_default = 0;
1876
1877         $self->start_flags($e, $ndr);
1878
1879         $self->pidl("NDR_PUSH_CHECK_FLAGS(ndr, ndr_flags);");
1880         $self->pidl("if (ndr_flags & NDR_SCALARS) {");
1881         $self->indent;
1882         $self->ParseUnionPushPrimitives($e, $ndr, $varname);
1883         $self->deindent;
1884         $self->pidl("}");
1885         $self->pidl("if (ndr_flags & NDR_BUFFERS) {");
1886         $self->indent;
1887         $self->ParseUnionPushDeferred($e, $ndr, $varname);
1888         $self->deindent;
1889         $self->pidl("}");
1890         $self->end_flags($e, $ndr);
1891 }
1892
1893 #####################################################################
1894 # print a union
1895 sub ParseUnionPrint($$$$$)
1896 {
1897         my ($self,$e,$ndr,$name,$varname) = @_;
1898         my $have_default = 0;
1899
1900         $self->pidl("uint32_t level;");
1901         foreach my $el (@{$e->{ELEMENTS}}) {
1902                 $self->DeclareArrayVariables($el);
1903         }
1904
1905         $self->start_flags($e, $ndr);
1906
1907         $self->pidl("level = ndr_print_get_switch_value($ndr, $varname);");
1908
1909         $self->pidl("ndr_print_union($ndr, name, level, \"$name\");");
1910
1911         $self->pidl("switch (level) {");
1912         $self->indent;
1913         foreach my $el (@{$e->{ELEMENTS}}) {
1914                 if ($el->{CASE} eq "default") {
1915                         $have_default = 1;
1916                 }
1917                 $self->pidl("$el->{CASE}:");
1918                 if ($el->{TYPE} ne "EMPTY") {
1919                         $self->indent;
1920                         $self->ParseElementPrint($el, $ndr, "$varname->$el->{NAME}", {});
1921                         $self->deindent;
1922                 }
1923                 $self->pidl("break;");
1924                 $self->pidl("");
1925         }
1926         if (! $have_default) {
1927                 $self->pidl("default:");
1928                 $self->pidl("\tndr_print_bad_level($ndr, name, level);");
1929         }
1930         $self->deindent;
1931         $self->pidl("}");
1932
1933         $self->end_flags($e, $ndr);
1934 }
1935
1936 sub ParseUnionPullPrimitives($$$$$)
1937 {
1938         my ($self,$e,$ndr,$varname,$switch_type) = @_;
1939         my $have_default = 0;
1940
1941
1942         if (defined($switch_type)) {
1943                 if (defined($e->{ALIGN})) {
1944                         $self->pidl("NDR_CHECK(ndr_pull_union_align($ndr, $e->{ALIGN}));");
1945                 }
1946
1947                 $self->pidl("NDR_CHECK(ndr_pull_$switch_type($ndr, NDR_SCALARS, &_level));");
1948                 $self->pidl("if (_level != level) {");
1949                 $self->pidl("\treturn ndr_pull_error($ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value %u for $varname at \%s\", _level, __location__);");
1950                 $self->pidl("}");
1951         }
1952
1953         if (defined($e->{ALIGN})) {
1954                 if ($e->{IS_MS_UNION}) {
1955                         $self->pidl("/* ms_union is always aligned to the largest union arm*/");
1956                         $self->pidl("NDR_CHECK(ndr_pull_align($ndr, $e->{ALIGN}));");
1957                 } else {
1958                         $self->pidl("NDR_CHECK(ndr_pull_union_align($ndr, $e->{ALIGN}));");
1959                 }
1960         }
1961
1962         $self->pidl("switch (level) {");
1963         $self->indent;
1964         foreach my $el (@{$e->{ELEMENTS}}) {
1965                 if ($el->{CASE} eq "default") {
1966                         $have_default = 1;
1967                 }
1968                 $self->pidl("$el->{CASE}: {");
1969
1970                 if ($el->{TYPE} ne "EMPTY") {
1971                         $self->indent;
1972                         if (defined($e->{PROPERTIES}{relative_base})) {
1973                                 $self->pidl("NDR_CHECK(ndr_pull_align($ndr, $el->{ALIGN}));");
1974                                 # set the current offset as base for relative pointers
1975                                 # and store it based on the toplevel struct/union
1976                                 $self->pidl("NDR_CHECK(ndr_pull_setup_relative_base_offset1($ndr, $varname, $ndr->offset));");
1977                         }
1978                         $self->ParseElementPull($el, $ndr, {$el->{NAME} => "$varname->$el->{NAME}"}, 1, 0);
1979                         $self->deindent;
1980                 }
1981                 $self->pidl("break; }");
1982                 $self->pidl("");
1983         }
1984         if (! $have_default) {
1985                 $self->pidl("default:");
1986                 $self->pidl("\treturn ndr_pull_error($ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u at \%s\", level, __location__);");
1987         }
1988         $self->deindent;
1989         $self->pidl("}");
1990 }
1991
1992 sub ParseUnionPullDeferred($$$$)
1993 {
1994         my ($self,$e,$ndr,$varname) = @_;
1995         my $have_default = 0;
1996
1997         if (defined($e->{PROPERTIES}{relative_base})) {
1998                 # retrieve the current offset as base for relative pointers
1999                 # based on the toplevel struct/union
2000                 $self->pidl("NDR_CHECK(ndr_pull_setup_relative_base_offset2($ndr, $varname));");
2001         }
2002         $self->pidl("switch (level) {");
2003         $self->indent;
2004         foreach my $el (@{$e->{ELEMENTS}}) {
2005                 if ($el->{CASE} eq "default") {
2006                         $have_default = 1;
2007                 }
2008
2009                 $self->pidl("$el->{CASE}:");
2010                 if ($el->{TYPE} ne "EMPTY") {
2011                         $self->indent;
2012                         $self->ParseElementPull($el, $ndr, {$el->{NAME} => "$varname->$el->{NAME}"}, 0, 1);
2013                         $self->deindent;
2014                 }
2015                 $self->pidl("break;");
2016                 $self->pidl("");
2017         }
2018         if (! $have_default) {
2019                 $self->pidl("default:");
2020                 $self->pidl("\treturn ndr_pull_error($ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u at \%s\", level, __location__);");
2021         }
2022         $self->deindent;
2023         $self->pidl("}");
2024
2025
2026 }
2027
2028 #####################################################################
2029 # parse a union - pull side
2030 sub ParseUnionPull($$$$)
2031 {
2032         my ($self,$e,$ndr,$varname) = @_;
2033         my $switch_type = $e->{SWITCH_TYPE};
2034
2035         $self->pidl("uint32_t level;");
2036         if (defined($switch_type)) {
2037                 if (Parse::Pidl::Typelist::typeIs($switch_type, "ENUM")) {
2038                         $switch_type = Parse::Pidl::Typelist::enum_type_fn(getType($switch_type)->{DATA});
2039                 }
2040                 $self->pidl(mapTypeName($switch_type) . " _level;");
2041         }
2042
2043         my %double_cases = ();
2044         foreach my $el (@{$e->{ELEMENTS}}) {
2045                 next if ($el->{TYPE} eq "EMPTY");
2046                 next if ($double_cases{"$el->{NAME}"});
2047                 $self->DeclareMemCtxVariables($el);
2048                 $self->DeclarePtrVariables($el);
2049                 $self->DeclareArrayVariables($el);
2050                 $double_cases{"$el->{NAME}"} = 1;
2051         }
2052
2053         $self->start_flags($e, $ndr);
2054
2055         $self->pidl("level = ndr_pull_get_switch_value($ndr, $varname);");
2056
2057         $self->pidl("NDR_PULL_CHECK_FLAGS(ndr, ndr_flags);");
2058         $self->pidl("if (ndr_flags & NDR_SCALARS) {");
2059         $self->indent;
2060         $self->ParseUnionPullPrimitives($e,$ndr,$varname,$switch_type);
2061         $self->deindent;
2062         $self->pidl("}");
2063
2064         $self->pidl("if (ndr_flags & NDR_BUFFERS) {");
2065         $self->indent;
2066         $self->ParseUnionPullDeferred($e,$ndr,$varname);
2067         $self->deindent;
2068         $self->pidl("}");
2069
2070         $self->add_deferred();
2071
2072         $self->end_flags($e, $ndr);
2073 }
2074
2075 sub DeclUnion($$$$)
2076 {
2077         my ($e,$t,$name,$varname) = @_;
2078         return ($t ne "pull"?"const ":"") . "union $name *$varname";
2079 }
2080
2081 sub ArgsUnionNdrSize($$)
2082 {
2083         my ($d,$name) = @_;
2084         return "const union $name *r, uint32_t level, int flags";
2085 }
2086
2087 $typefamily{UNION} = {
2088         PUSH_FN_BODY => \&ParseUnionPush,
2089         DECL => \&DeclUnion,
2090         PULL_FN_BODY => \&ParseUnionPull,
2091         PRINT_FN_BODY => \&ParseUnionPrint,
2092         SIZE_FN_ARGS => \&ArgsUnionNdrSize,
2093         SIZE_FN_BODY => \&ParseUnionNdrSize,
2094 };
2095
2096 #####################################################################
2097 # parse a typedef - push side
2098 sub ParseTypedefPush($$$$)
2099 {
2100         my($self,$e,$ndr,$varname) = @_;
2101
2102         my $env;
2103
2104         $env->{$e->{NAME}} = $varname;
2105
2106         $self->ParseElementPushLevel($e, $e->{LEVELS}[0], $ndr, $varname, $env, 1, 1);
2107 }
2108
2109 #####################################################################
2110 # parse a typedef - pull side
2111 sub ParseTypedefPull($$$$)
2112 {
2113         my($self,$e,$ndr,$varname) = @_;
2114
2115         my $env;
2116
2117         $env->{$e->{NAME}} = $varname;
2118
2119         $self->ParseElementPullLevel($e, $e->{LEVELS}[0], $ndr, $varname, $env, 1, 1);
2120 }
2121
2122 #####################################################################
2123 # parse a typedef - print side
2124 sub ParseTypedefPrint($$$$$)
2125 {
2126         my($self,$e,$ndr,$name,$varname) = @_;
2127
2128         $typefamily{$e->{DATA}->{TYPE}}->{PRINT_FN_BODY}->($self, $e->{DATA}, $ndr, $name, $varname);
2129 }
2130
2131 #####################################################################
2132 ## calculate the size of a structure
2133 sub ParseTypedefNdrSize($$$$)
2134 {
2135         my($self,$t,$name,$varname) = @_;
2136
2137         $typefamily{$t->{DATA}->{TYPE}}->{SIZE_FN_BODY}->($self, $t->{DATA}, $name, $varname);
2138 }
2139
2140 sub DeclTypedef($$$$)
2141 {
2142         my ($e, $t, $name, $varname) = @_;
2143
2144         return $typefamily{$e->{DATA}->{TYPE}}->{DECL}->($e->{DATA}, $t, $name, $varname);
2145 }
2146
2147 sub ArgsTypedefNdrSize($$$)
2148 {
2149         my ($d, $name, $varname) = @_;
2150         return $typefamily{$d->{DATA}->{TYPE}}->{SIZE_FN_ARGS}->($d->{DATA}, $name, $varname);
2151 }
2152
2153 $typefamily{TYPEDEF} = {
2154         PUSH_FN_BODY => \&ParseTypedefPush,
2155         DECL => \&DeclTypedef,
2156         PULL_FN_BODY => \&ParseTypedefPull,
2157         PRINT_FN_BODY => \&ParseTypedefPrint,
2158         SIZE_FN_ARGS => \&ArgsTypedefNdrSize,
2159         SIZE_FN_BODY => \&ParseTypedefNdrSize,
2160 };
2161
2162 sub ParsePipePushChunk($$)
2163 {
2164         my ($self, $t) = @_;
2165
2166         my $pipe = $t;
2167         $pipe = $t->{DATA} if ($t->{TYPE} eq "TYPEDEF");
2168         my $struct = $pipe->{DATA};
2169
2170         my $name = "$struct->{NAME}";
2171         my $ndr = "ndr";
2172         my $varname = "r";
2173
2174         my $args = $typefamily{$struct->{TYPE}}->{DECL}->($struct, "push", $name, $varname);
2175
2176         $self->fn_declare("push", $struct, "enum ndr_err_code ndr_push_$name(struct ndr_push *$ndr, int ndr_flags, $args)") or return;
2177
2178         return if has_property($t, "nopush");
2179
2180         $self->pidl("{");
2181         $self->indent;
2182
2183         $self->ParseStructPush($struct, $ndr, $varname);
2184         $self->pidl("");
2185
2186         $self->pidl("NDR_CHECK(ndr_push_pipe_chunk_trailer(ndr, ndr_flags, $varname->count));");
2187         $self->pidl("");
2188
2189         $self->pidl("return NDR_ERR_SUCCESS;");
2190         $self->deindent;
2191         $self->pidl("}");
2192         $self->pidl("");
2193 }
2194
2195 sub ParsePipePullChunk($$)
2196 {
2197         my ($self, $t) = @_;
2198
2199         my $pipe = $t;
2200         $pipe = $t->{DATA} if ($t->{TYPE} eq "TYPEDEF");
2201         my $struct = $pipe->{DATA};
2202
2203         my $name = "$struct->{NAME}";
2204         my $ndr = "ndr";
2205         my $varname = "r";
2206
2207         my $args = $typefamily{$struct->{TYPE}}->{DECL}->($struct, "pull", $name, $varname);
2208
2209         $self->fn_declare("pull", $struct, "enum ndr_err_code ndr_pull_$name(struct ndr_pull *$ndr, int ndr_flags, $args)") or return;
2210
2211         return if has_property($struct, "nopull");
2212
2213         $self->pidl("{");
2214         $self->indent;
2215
2216         $self->ParseStructPull($struct, $ndr, $varname);
2217         $self->pidl("");
2218
2219         $self->pidl("NDR_CHECK(ndr_check_pipe_chunk_trailer($ndr, ndr_flags, $varname->count));");
2220         $self->pidl("");
2221
2222         $self->pidl("return NDR_ERR_SUCCESS;");
2223         $self->deindent;
2224         $self->pidl("}");
2225         $self->pidl("");
2226 }
2227
2228 sub ParsePipePrintChunk($$)
2229 {
2230         my ($self, $t) = @_;
2231
2232         my $pipe = $t;
2233         $pipe = $t->{DATA} if ($t->{TYPE} eq "TYPEDEF");
2234         my $struct = $pipe->{DATA};
2235
2236         my $name = "$struct->{NAME}";
2237         my $ndr = "ndr";
2238         my $varname = "r";
2239
2240         my $args = $typefamily{$struct->{TYPE}}->{DECL}->($struct, "print", $name, $varname);
2241
2242         $self->pidl_hdr("void ndr_print_$name(struct ndr_print *ndr, const char *name, $args);");
2243
2244         return if (has_property($t, "noprint"));
2245
2246         $self->pidl("_PUBLIC_ void ndr_print_$name(struct ndr_print *$ndr, const char *name, $args)");
2247         $self->pidl("{");
2248         $self->indent;
2249         $self->ParseTypePrint($struct, $ndr, $varname);
2250         $self->deindent;
2251         $self->pidl("}");
2252         $self->pidl("");
2253 }
2254
2255 #####################################################################
2256 # parse a function - print side
2257 sub ParseFunctionPrint($$)
2258 {
2259         my($self, $fn) = @_;
2260         my $ndr = "ndr";
2261
2262         $self->pidl_hdr("void ndr_print_$fn->{NAME}(struct ndr_print *$ndr, const char *name, int flags, const struct $fn->{NAME} *r);");
2263
2264         return if has_property($fn, "noprint");
2265
2266         $self->pidl("_PUBLIC_ void ndr_print_$fn->{NAME}(struct ndr_print *$ndr, const char *name, int flags, const struct $fn->{NAME} *r)");
2267         $self->pidl("{");
2268         $self->indent;
2269
2270         foreach my $e (@{$fn->{ELEMENTS}}) {
2271                 $self->DeclareArrayVariables($e);
2272         }
2273
2274         $self->pidl("ndr_print_struct($ndr, name, \"$fn->{NAME}\");");
2275         $self->pidl("if (r == NULL) { ndr_print_null($ndr); return; }");
2276         $self->pidl("$ndr->depth++;");
2277
2278         $self->pidl("if (flags & NDR_SET_VALUES) {");
2279         $self->pidl("\t$ndr->flags |= LIBNDR_PRINT_SET_VALUES;");
2280         $self->pidl("}");
2281
2282         $self->pidl("if (flags & NDR_IN) {");
2283         $self->indent;
2284         $self->pidl("ndr_print_struct($ndr, \"in\", \"$fn->{NAME}\");");
2285         $self->pidl("$ndr->depth++;");
2286
2287         my $env = GenerateFunctionInEnv($fn);
2288
2289         foreach my $e (@{$fn->{ELEMENTS}}) {
2290                 if (grep(/in/,@{$e->{DIRECTION}})) {
2291                         $self->ParseElementPrint($e, $ndr, $env->{$e->{NAME}}, $env);
2292                 }
2293         }
2294         $self->pidl("$ndr->depth--;");
2295         $self->deindent;
2296         $self->pidl("}");
2297
2298         $self->pidl("if (flags & NDR_OUT) {");
2299         $self->indent;
2300         $self->pidl("ndr_print_struct($ndr, \"out\", \"$fn->{NAME}\");");
2301         $self->pidl("$ndr->depth++;");
2302
2303         $env = GenerateFunctionOutEnv($fn);
2304         foreach my $e (@{$fn->{ELEMENTS}}) {
2305                 if (grep(/out/,@{$e->{DIRECTION}})) {
2306                         $self->ParseElementPrint($e, $ndr, $env->{$e->{NAME}}, $env);
2307                 }
2308         }
2309         if ($fn->{RETURN_TYPE}) {
2310                 $self->pidl("ndr_print_$fn->{RETURN_TYPE}($ndr, \"result\", r->out.result);");
2311         }
2312         $self->pidl("$ndr->depth--;");
2313         $self->deindent;
2314         $self->pidl("}");
2315
2316         $self->pidl("$ndr->depth--;");
2317         $self->deindent;
2318         $self->pidl("}");
2319         $self->pidl("");
2320 }
2321
2322 #####################################################################
2323 # parse a function
2324 sub ParseFunctionPush($$)
2325 {
2326         my($self, $fn) = @_;
2327         my $ndr = "ndr";
2328
2329         $self->fn_declare("push", $fn, "enum ndr_err_code ndr_push_$fn->{NAME}(struct ndr_push *$ndr, int flags, const struct $fn->{NAME} *r)") or return;
2330
2331         return if has_property($fn, "nopush");
2332
2333         $self->pidl("{");
2334         $self->indent;
2335
2336         foreach my $e (@{$fn->{ELEMENTS}}) {
2337                 $self->DeclareArrayVariables($e);
2338         }
2339
2340         $self->pidl("NDR_PUSH_CHECK_FN_FLAGS(ndr, flags);");
2341
2342         $self->pidl("if (flags & NDR_IN) {");
2343         $self->indent;
2344
2345         my $env = GenerateFunctionInEnv($fn);
2346
2347         EnvSubstituteValue($env, $fn);
2348
2349         foreach my $e (@{$fn->{ELEMENTS}}) {
2350                 if (grep(/in/,@{$e->{DIRECTION}})) {
2351                         $self->ParseElementPush($e, $ndr, $env, 1, 1);
2352                 }
2353         }
2354
2355         $self->deindent;
2356         $self->pidl("}");
2357
2358         $self->pidl("if (flags & NDR_OUT) {");
2359         $self->indent;
2360
2361         $env = GenerateFunctionOutEnv($fn);
2362         foreach my $e (@{$fn->{ELEMENTS}}) {
2363                 if (grep(/out/,@{$e->{DIRECTION}})) {
2364                         $self->ParseElementPush($e, $ndr, $env, 1, 1);
2365                 }
2366         }
2367
2368         if ($fn->{RETURN_TYPE}) {
2369                 $self->pidl("NDR_CHECK(ndr_push_$fn->{RETURN_TYPE}($ndr, NDR_SCALARS, r->out.result));");
2370         }
2371
2372         $self->deindent;
2373         $self->pidl("}");
2374         $self->pidl("return NDR_ERR_SUCCESS;");
2375         $self->deindent;
2376         $self->pidl("}");
2377         $self->pidl("");
2378 }
2379
2380 sub AllocateArrayLevel($$$$$$)
2381 {
2382         my ($self,$e,$l,$ndr,$var,$size) = @_;
2383
2384         my $pl = GetPrevLevel($e, $l);
2385         if (defined($pl) and
2386             $pl->{TYPE} eq "POINTER" and
2387             $pl->{POINTER_TYPE} eq "ref"
2388             and not $l->{IS_ZERO_TERMINATED}) {
2389                 $self->pidl("if ($ndr->flags & LIBNDR_FLAG_REF_ALLOC) {");
2390                 $self->pidl("\tNDR_PULL_ALLOC_N($ndr, $var, $size);");
2391                 $self->pidl("}");
2392                 if (grep(/in/,@{$e->{DIRECTION}}) and
2393                     grep(/out/,@{$e->{DIRECTION}})) {
2394                         $self->pidl("memcpy(r->out.$e->{NAME}, r->in.$e->{NAME}, ($size) * sizeof(*r->in.$e->{NAME}));");
2395                 }
2396                 return;
2397         }
2398
2399         $self->pidl("NDR_PULL_ALLOC_N($ndr, $var, $size);");
2400 }
2401
2402 #####################################################################
2403 # parse a function
2404 sub ParseFunctionPull($$)
2405 {
2406         my($self,$fn) = @_;
2407         my $ndr = "ndr";
2408
2409         # pull function args
2410         $self->fn_declare("pull", $fn, "enum ndr_err_code ndr_pull_$fn->{NAME}(struct ndr_pull *$ndr, int flags, struct $fn->{NAME} *r)") or return;
2411
2412         $self->pidl("{");
2413         $self->indent;
2414
2415         # declare any internal pointers we need
2416         foreach my $e (@{$fn->{ELEMENTS}}) {
2417                 $self->DeclarePtrVariables($e);
2418                 $self->DeclareArrayVariables($e);
2419         }
2420
2421         my %double_cases = ();
2422         foreach my $e (@{$fn->{ELEMENTS}}) {
2423                 next if ($e->{TYPE} eq "EMPTY");
2424                 next if ($double_cases{"$e->{NAME}"});
2425                 $self->DeclareMemCtxVariables($e);
2426                 $double_cases{"$e->{NAME}"} = 1;
2427         }
2428
2429         $self->pidl("NDR_PULL_CHECK_FN_FLAGS(ndr, flags);");
2430
2431         $self->pidl("if (flags & NDR_IN) {");
2432         $self->indent;
2433
2434         # auto-init the out section of a structure. I originally argued that
2435         # this was a bad idea as it hides bugs, but coping correctly
2436         # with initialisation and not wiping ref vars is turning
2437         # out to be too tricky (tridge)
2438         foreach my $e (@{$fn->{ELEMENTS}}) {
2439                 next unless grep(/out/, @{$e->{DIRECTION}});
2440                 $self->pidl("ZERO_STRUCT(r->out);");
2441                 $self->pidl("");
2442                 last;
2443         }
2444
2445         my $env = GenerateFunctionInEnv($fn);
2446
2447         foreach my $e (@{$fn->{ELEMENTS}}) {
2448                 next unless (grep(/in/, @{$e->{DIRECTION}}));
2449                 $self->ParseElementPull($e, $ndr, $env, 1, 1);
2450         }
2451
2452         # allocate the "simple" out ref variables. FIXME: Shouldn't this have it's
2453         # own flag rather than be in NDR_IN ?
2454
2455         foreach my $e (@{$fn->{ELEMENTS}}) {
2456                 next unless (grep(/out/, @{$e->{DIRECTION}}));
2457                 next unless ($e->{LEVELS}[0]->{TYPE} eq "POINTER" and
2458                              $e->{LEVELS}[0]->{POINTER_TYPE} eq "ref");
2459                 next if (($e->{LEVELS}[1]->{TYPE} eq "DATA") and
2460                                  ($e->{LEVELS}[1]->{DATA_TYPE} eq "string"));
2461                 next if ($e->{LEVELS}[1]->{TYPE} eq "PIPE");
2462                 next if (($e->{LEVELS}[1]->{TYPE} eq "ARRAY")
2463                         and   $e->{LEVELS}[1]->{IS_ZERO_TERMINATED});
2464
2465                 if ($e->{LEVELS}[1]->{TYPE} eq "ARRAY") {
2466                         my $size = ParseExprExt($e->{LEVELS}[1]->{SIZE_IS}, $env, $e->{ORIGINAL},
2467                                 check_null_pointer($e, $env, sub { $self->pidl(shift); },
2468                                                    "return ndr_pull_error($ndr, NDR_ERR_INVALID_POINTER, \"NULL Pointer for size_is()\");"),
2469                                 check_fully_dereferenced($e, $env));
2470                         $self->pidl("NDR_PULL_ALLOC_N($ndr, r->out.$e->{NAME}, $size);");
2471
2472                         if (grep(/in/, @{$e->{DIRECTION}})) {
2473                                 $self->pidl("memcpy(r->out.$e->{NAME}, r->in.$e->{NAME}, ($size) * sizeof(*r->in.$e->{NAME}));");
2474                         } else {
2475                                 $self->pidl("memset(r->out.$e->{NAME}, 0, ($size) * sizeof(*r->out.$e->{NAME}));");
2476                         }
2477                 } elsif ($e->{LEVELS}[1]->{TYPE} eq "ARRAY") {
2478                         if (grep(/in/, @{$e->{DIRECTION}})) {
2479                                 $self->pidl("r->out.$e->{NAME} = r->in.$e->{NAME};");
2480                         } else {
2481                                 $self->pidl("r->out.$e->{NAME} = NULL;");
2482                         }
2483                 } else {
2484                         $self->pidl("NDR_PULL_ALLOC($ndr, r->out.$e->{NAME});");
2485
2486                         if (grep(/in/, @{$e->{DIRECTION}})) {
2487                                 $self->pidl("*r->out.$e->{NAME} = *r->in.$e->{NAME};");
2488                         } else {
2489                                 $self->pidl("ZERO_STRUCTP(r->out.$e->{NAME});");
2490                         }
2491                 }
2492         }
2493
2494         $self->add_deferred();
2495         $self->deindent;
2496         $self->pidl("}");
2497
2498         $self->pidl("if (flags & NDR_OUT) {");
2499         $self->indent;
2500
2501         $env = GenerateFunctionOutEnv($fn);
2502         foreach my $e (@{$fn->{ELEMENTS}}) {
2503                 next unless grep(/out/, @{$e->{DIRECTION}});
2504                 $self->ParseElementPull($e, $ndr, $env, 1, 1);
2505         }
2506
2507         if ($fn->{RETURN_TYPE}) {
2508                 $self->pidl("NDR_CHECK(ndr_pull_$fn->{RETURN_TYPE}($ndr, NDR_SCALARS, &r->out.result));");
2509         }
2510
2511         $self->add_deferred();
2512         $self->deindent;
2513         $self->pidl("}");
2514
2515         $self->pidl("return NDR_ERR_SUCCESS;");
2516         $self->deindent;
2517         $self->pidl("}");
2518         $self->pidl("");
2519 }
2520
2521 sub AuthServiceStruct($$$)
2522 {
2523         my ($self, $ifacename, $authservice) = @_;
2524         my @a = split /,/, $authservice;
2525         my $authservice_count = $#a + 1;
2526
2527         $self->pidl("static const char * const $ifacename\_authservice_strings[] = {");
2528         foreach my $ap (@a) {
2529                 $self->pidl("\t$ap, ");
2530         }
2531         $self->pidl("};");
2532         $self->pidl("");
2533
2534         $self->pidl("static const struct ndr_interface_string_array $ifacename\_authservices = {");
2535         $self->pidl("\t.count\t= $authservice_count,");
2536         $self->pidl("\t.names\t= $ifacename\_authservice_strings");
2537         $self->pidl("};");
2538         $self->pidl("");
2539 }
2540
2541 sub ParseGeneratePipeArray($$$)
2542 {
2543         my ($self, $fn, $direction) = @_;
2544
2545         $self->pidl("static const struct ndr_interface_call_pipe $fn->{NAME}\_$direction\_pipes[] = {");
2546         $self->indent;
2547
2548         foreach my $e (@{$fn->{ELEMENTS}}) {
2549                 next unless ContainsPipe($e, $e->{LEVELS}[0]);
2550                 next unless (grep(/$direction/, @{$e->{DIRECTION}}));
2551
2552                 my $cname = "$e->{TYPE}_chunk";
2553
2554                 $self->pidl("{");
2555                 $self->indent;
2556                 $self->pidl("\"$direction.$e->{NAME}\",");
2557                 $self->pidl("\"$cname\",");
2558                 $self->pidl("sizeof(struct $cname),");
2559                 $self->pidl("(ndr_push_flags_fn_t) ndr_push_$cname,");
2560                 $self->pidl("(ndr_pull_flags_fn_t) ndr_pull_$cname,");
2561                 $self->pidl("(ndr_print_fn_t) ndr_print_$cname,");
2562                 $self->deindent;
2563                 $self->pidl("},");
2564         }
2565         $self->pidl("{ NULL, NULL, 0, NULL, NULL, NULL }");
2566         $self->deindent;
2567         $self->pidl("};");
2568         $self->pidl("");
2569 }
2570
2571 sub FunctionCallPipes($$)
2572 {
2573         my ($self, $d) = @_;
2574         return if not defined($d->{OPNUM});
2575
2576         my $in_pipes = 0;
2577         my $out_pipes = 0;
2578
2579         foreach my $e (@{$d->{ELEMENTS}}) {
2580                 next unless ContainsPipe($e, $e->{LEVELS}[0]);
2581
2582                 if (grep(/in/, @{$e->{DIRECTION}})) {
2583                         $in_pipes++;
2584                 }
2585                 if (grep(/out/, @{$e->{DIRECTION}})) {
2586                         $out_pipes++;
2587                 }
2588         }
2589
2590         if ($in_pipes) {
2591                 $self->ParseGeneratePipeArray($d, "in");
2592         }
2593
2594         if ($out_pipes) {
2595                 $self->ParseGeneratePipeArray($d, "out");
2596         }
2597 }
2598
2599 sub FunctionCallEntry($$)
2600 {
2601         my ($self, $d) = @_;
2602         return 0 if not defined($d->{OPNUM});
2603
2604         my $in_pipes = 0;
2605         my $out_pipes = 0;
2606
2607         foreach my $e (@{$d->{ELEMENTS}}) {
2608                 next unless ContainsPipe($e, $e->{LEVELS}[0]);
2609
2610                 if (grep(/in/, @{$e->{DIRECTION}})) {
2611                         $in_pipes++;
2612                 }
2613                 if (grep(/out/, @{$e->{DIRECTION}})) {
2614                         $out_pipes++;
2615                 }
2616         }
2617
2618         my $in_pipes_ptr = "NULL";
2619         my $out_pipes_ptr = "NULL";
2620
2621         if ($in_pipes) {
2622                 $in_pipes_ptr = "$d->{NAME}_in_pipes";
2623         }
2624
2625         if ($out_pipes) {
2626                 $out_pipes_ptr = "$d->{NAME}_out_pipes";
2627         }
2628
2629         $self->pidl("\t{");
2630         $self->pidl("\t\t\"$d->{NAME}\",");
2631         $self->pidl("\t\tsizeof(struct $d->{NAME}),");
2632         $self->pidl("\t\t(ndr_push_flags_fn_t) ndr_push_$d->{NAME},");
2633         $self->pidl("\t\t(ndr_pull_flags_fn_t) ndr_pull_$d->{NAME},");
2634         $self->pidl("\t\t(ndr_print_function_t) ndr_print_$d->{NAME},");
2635         $self->pidl("\t\t{ $in_pipes, $in_pipes_ptr },");
2636         $self->pidl("\t\t{ $out_pipes, $out_pipes_ptr },");
2637         $self->pidl("\t},");
2638         return 1;
2639 }
2640
2641 #####################################################################
2642 # produce a function call table
2643 sub FunctionTable($$)
2644 {
2645         my($self,$interface) = @_;
2646         my $count = 0;
2647         my $uname = uc $interface->{NAME};
2648
2649         return if ($#{$interface->{FUNCTIONS}}+1 == 0);
2650         return unless defined ($interface->{PROPERTIES}->{uuid});
2651
2652         foreach my $d (@{$interface->{INHERITED_FUNCTIONS}},@{$interface->{FUNCTIONS}}) {
2653                 $self->FunctionCallPipes($d);
2654         }
2655
2656         $self->pidl("static const struct ndr_interface_call $interface->{NAME}\_calls[] = {");
2657
2658         foreach my $d (@{$interface->{INHERITED_FUNCTIONS}},@{$interface->{FUNCTIONS}}) {
2659                 $count += $self->FunctionCallEntry($d);
2660         }
2661         $self->pidl("\t{ NULL, 0, NULL, NULL, NULL }");
2662         $self->pidl("};");
2663         $self->pidl("");
2664
2665         $self->pidl("static const char * const $interface->{NAME}\_endpoint_strings[] = {");
2666         foreach my $ep (@{$interface->{ENDPOINTS}}) {
2667                 $self->pidl("\t$ep, ");
2668         }
2669         my $endpoint_count = $#{$interface->{ENDPOINTS}}+1;
2670
2671         $self->pidl("};");
2672         $self->pidl("");
2673
2674         $self->pidl("static const struct ndr_interface_string_array $interface->{NAME}\_endpoints = {");
2675         $self->pidl("\t.count\t= $endpoint_count,");
2676         $self->pidl("\t.names\t= $interface->{NAME}\_endpoint_strings");
2677         $self->pidl("};");
2678         $self->pidl("");
2679
2680         if (! defined $interface->{PROPERTIES}->{authservice}) {
2681                 $interface->{PROPERTIES}->{authservice} = "\"host\"";
2682         }
2683
2684         $self->AuthServiceStruct($interface->{NAME},
2685                                      $interface->{PROPERTIES}->{authservice});
2686
2687         $self->pidl("\nconst struct ndr_interface_table ndr_table_$interface->{NAME} = {");
2688         $self->pidl("\t.name\t\t= \"$interface->{NAME}\",");
2689         $self->pidl("\t.syntax_id\t= {");
2690         $self->pidl("\t\t" . print_uuid($interface->{UUID}) .",");
2691         $self->pidl("\t\tNDR_$uname\_VERSION");
2692         $self->pidl("\t},");
2693         $self->pidl("\t.helpstring\t= NDR_$uname\_HELPSTRING,");
2694         $self->pidl("\t.num_calls\t= $count,");
2695         $self->pidl("\t.calls\t\t= $interface->{NAME}\_calls,");
2696         $self->pidl("\t.endpoints\t= &$interface->{NAME}\_endpoints,");
2697         $self->pidl("\t.authservices\t= &$interface->{NAME}\_authservices");
2698         $self->pidl("};");
2699         $self->pidl("");
2700
2701 }
2702
2703 #####################################################################
2704 # generate include statements for imported idl files
2705 sub HeaderImport
2706 {
2707         my $self = shift;
2708         my @imports = @_;
2709         foreach (@imports) {
2710                 $_ = unmake_str($_);
2711                 s/\.idl$//;
2712                 $self->pidl(choose_header("librpc/gen_ndr/ndr_$_\.h", "gen_ndr/ndr_$_.h"));
2713         }
2714 }
2715
2716 #####################################################################
2717 # generate include statements for included header files
2718 sub HeaderInclude
2719 {
2720         my $self = shift;
2721         my @includes = @_;
2722         foreach (@includes) {
2723                 $self->pidl_hdr("#include $_");
2724         }
2725 }
2726
2727 #####################################################################
2728 # generate prototypes and defines for the interface definitions
2729 # FIXME: these prototypes are for the DCE/RPC client functions, not the
2730 # NDR parser and so do not belong here, technically speaking
2731 sub HeaderInterface($$$)
2732 {
2733         my($self,$interface,$needed) = @_;
2734
2735         my $count = 0;
2736
2737         if ($needed->{"compression"}) {
2738                 $self->pidl(choose_header("librpc/ndr/ndr_compression.h", "ndr/compression.h"));
2739         }
2740
2741         if (has_property($interface, "object")) {
2742                 $self->pidl(choose_header("librpc/gen_ndr/ndr_orpc.h", "ndr/orpc.h"));
2743         }
2744
2745         if (defined $interface->{PROPERTIES}->{helper}) {
2746                 $self->HeaderInclude(split /,/, $interface->{PROPERTIES}->{helper});
2747         }
2748
2749         if (defined $interface->{PROPERTIES}->{uuid}) {
2750                 my $name = uc $interface->{NAME};
2751                 $self->pidl_hdr("#define NDR_$name\_UUID " .
2752                 Parse::Pidl::Util::make_str(lc($interface->{UUID})));
2753
2754                 $self->pidl_hdr("#define NDR_$name\_VERSION $interface->{VERSION}");
2755
2756                 $self->pidl_hdr("#define NDR_$name\_NAME \"$interface->{NAME}\"");
2757
2758                 if(!defined $interface->{PROPERTIES}->{helpstring}) { $interface->{PROPERTIES}->{helpstring} = "NULL"; }
2759                 $self->pidl_hdr("#define NDR_$name\_HELPSTRING $interface->{PROPERTIES}->{helpstring}");
2760
2761                 $self->pidl_hdr("extern const struct ndr_interface_table ndr_table_$interface->{NAME};");
2762         }
2763
2764         foreach (@{$interface->{FUNCTIONS}}) {
2765                 next if has_property($_, "noopnum");
2766                 next if grep(/^$_->{NAME}$/,@{$interface->{INHERITED_FUNCTIONS}});
2767                 my $u_name = uc $_->{NAME};
2768
2769                 my $val = sprintf("0x%02x", $count);
2770                 if (defined($interface->{BASE})) {
2771                         $val .= " + NDR_" . uc $interface->{BASE} . "_CALL_COUNT";
2772                 }
2773
2774                 $self->pidl_hdr("#define NDR_$u_name ($val)");
2775
2776                 $self->pidl_hdr("");
2777                 $count++;
2778         }
2779
2780         my $val = $count;
2781
2782         if (defined($interface->{BASE})) {
2783                 $val .= " + NDR_" . uc $interface->{BASE} . "_CALL_COUNT";
2784         }
2785
2786         $self->pidl_hdr("#define NDR_" . uc $interface->{NAME} . "_CALL_COUNT ($val)");
2787
2788 }
2789
2790 sub ParseTypePush($$$$$$)
2791 {
2792         my ($self,$e, $ndr, $varname, $primitives, $deferred) = @_;
2793
2794         # save the old relative_base_offset
2795         $self->pidl("uint32_t _save_relative_base_offset = ndr_push_get_relative_base_offset($ndr);") if defined(has_property($e, "relative_base"));
2796         $typefamily{$e->{TYPE}}->{PUSH_FN_BODY}->($self, $e, $ndr, $varname);
2797         # restore the old relative_base_offset
2798         $self->pidl("ndr_push_restore_relative_base_offset($ndr, _save_relative_base_offset);") if defined(has_property($e, "relative_base"));
2799 }
2800
2801 sub ParseTypePushFunction($$$)
2802 {
2803         my ($self, $e, $varname) = @_;
2804         my $ndr = "ndr";
2805
2806         my $args = $typefamily{$e->{TYPE}}->{DECL}->($e, "push", $e->{NAME}, $varname);
2807
2808         $self->fn_declare("push", $e, "enum ndr_err_code ".TypeFunctionName("ndr_push", $e)."(struct ndr_push *$ndr, int ndr_flags, $args)") or return;
2809
2810         $self->pidl("{");
2811         $self->indent;
2812         $self->ParseTypePush($e, $ndr, $varname, 1, 1);
2813         $self->pidl("return NDR_ERR_SUCCESS;");
2814         $self->deindent;
2815         $self->pidl("}");
2816         $self->pidl("");;
2817 }
2818
2819 sub ParseTypePull($$$$$$)
2820 {
2821         my ($self, $e, $ndr, $varname, $primitives, $deferred) = @_;
2822
2823         # save the old relative_base_offset
2824         $self->pidl("uint32_t _save_relative_base_offset = ndr_pull_get_relative_base_offset($ndr);") if defined(has_property($e, "relative_base"));
2825         $typefamily{$e->{TYPE}}->{PULL_FN_BODY}->($self, $e, $ndr, $varname);
2826         # restore the old relative_base_offset
2827         $self->pidl("ndr_pull_restore_relative_base_offset($ndr, _save_relative_base_offset);") if defined(has_property($e, "relative_base"));
2828 }
2829
2830 sub ParseTypePullFunction($$)
2831 {
2832         my ($self, $e, $varname) = @_;
2833         my $ndr = "ndr";
2834
2835         my $args = $typefamily{$e->{TYPE}}->{DECL}->($e, "pull", $e->{NAME}, $varname);
2836
2837         $self->fn_declare("pull", $e, "enum ndr_err_code ".TypeFunctionName("ndr_pull", $e)."(struct ndr_pull *$ndr, int ndr_flags, $args)") or return;
2838
2839         $self->pidl("{");
2840         $self->indent;
2841         $self->ParseTypePull($e, $ndr, $varname, 1, 1);
2842         $self->pidl("return NDR_ERR_SUCCESS;");
2843         $self->deindent;
2844         $self->pidl("}");
2845         $self->pidl("");
2846 }
2847
2848 sub ParseTypePrint($$$$)
2849 {
2850         my ($self, $e, $ndr, $varname) = @_;
2851
2852         $typefamily{$e->{TYPE}}->{PRINT_FN_BODY}->($self, $e, $ndr, $e->{NAME}, $varname);
2853 }
2854
2855 sub ParseTypePrintFunction($$$)
2856 {
2857         my ($self, $e, $varname) = @_;
2858         my $ndr = "ndr";
2859
2860         my $args = $typefamily{$e->{TYPE}}->{DECL}->($e, "print", $e->{NAME}, $varname);
2861
2862         $self->pidl_hdr("void ".TypeFunctionName("ndr_print", $e)."(struct ndr_print *ndr, const char *name, $args);");
2863
2864         return if (has_property($e, "noprint"));
2865
2866         $self->pidl("_PUBLIC_ void ".TypeFunctionName("ndr_print", $e)."(struct ndr_print *$ndr, const char *name, $args)");
2867         $self->pidl("{");
2868         $self->indent;
2869         $self->ParseTypePrint($e, $ndr, $varname);
2870         $self->deindent;
2871         $self->pidl("}");
2872         $self->pidl("");
2873 }
2874
2875 sub ParseTypeNdrSize($$)
2876 {
2877         my ($self,$t) = @_;
2878
2879         my $varname = "r";
2880         my $tf = $typefamily{$t->{TYPE}};
2881         my $args = $tf->{SIZE_FN_ARGS}->($t, $t->{NAME}, $varname);
2882
2883         $self->fn_declare("size", $t, "size_t ndr_size_$t->{NAME}($args)") or return;
2884
2885         $self->pidl("{");
2886         $self->indent;
2887         $typefamily{$t->{TYPE}}->{SIZE_FN_BODY}->($self,$t, $t->{NAME}, $varname);
2888         $self->deindent;
2889         $self->pidl("}");
2890         $self->pidl("");
2891 }
2892
2893 #####################################################################
2894 # parse the interface definitions
2895 sub ParseInterface($$$)
2896 {
2897         my($self,$interface,$needed) = @_;
2898
2899         $self->pidl_hdr("#ifndef _HEADER_NDR_$interface->{NAME}");
2900         $self->pidl_hdr("#define _HEADER_NDR_$interface->{NAME}");
2901
2902         $self->pidl_hdr("");
2903
2904         $self->HeaderInterface($interface, $needed);
2905
2906         # Typedefs
2907         foreach my $d (@{$interface->{TYPES}}) {
2908                 if (Parse::Pidl::Typelist::typeIs($d, "PIPE")) {
2909                         ($needed->{TypeFunctionName("ndr_push", $d)}) &&
2910                                 $self->ParsePipePushChunk($d);
2911                         ($needed->{TypeFunctionName("ndr_pull", $d)}) &&
2912                                 $self->ParsePipePullChunk($d);
2913                         ($needed->{TypeFunctionName("ndr_print", $d)}) &&
2914                                 $self->ParsePipePrintChunk($d);
2915
2916                         $needed->{TypeFunctionName("ndr_pull", $d)} = 0;
2917                         $needed->{TypeFunctionName("ndr_push", $d)} = 0;
2918                         $needed->{TypeFunctionName("ndr_print", $d)} = 0;
2919                         next;
2920                 }
2921
2922                 next unless(typeHasBody($d));
2923
2924                 ($needed->{TypeFunctionName("ndr_push", $d)}) && $self->ParseTypePushFunction($d, "r");
2925                 ($needed->{TypeFunctionName("ndr_pull", $d)}) && $self->ParseTypePullFunction($d, "r");
2926                 ($needed->{TypeFunctionName("ndr_print", $d)}) && $self->ParseTypePrintFunction($d, "r");
2927
2928                 # Make sure we don't generate a function twice...
2929                 $needed->{TypeFunctionName("ndr_push", $d)} =
2930                     $needed->{TypeFunctionName("ndr_pull", $d)} =
2931                         $needed->{TypeFunctionName("ndr_print", $d)} = 0;
2932
2933                 ($needed->{"ndr_size_$d->{NAME}"}) && $self->ParseTypeNdrSize($d);
2934         }
2935
2936         # Functions
2937         foreach my $d (@{$interface->{FUNCTIONS}}) {
2938                 ($needed->{"ndr_push_$d->{NAME}"}) && $self->ParseFunctionPush($d);
2939                 ($needed->{"ndr_pull_$d->{NAME}"}) && $self->ParseFunctionPull($d);
2940                 ($needed->{"ndr_print_$d->{NAME}"}) && $self->ParseFunctionPrint($d);
2941         }
2942
2943         $self->FunctionTable($interface);
2944
2945         $self->pidl_hdr("#endif /* _HEADER_NDR_$interface->{NAME} */");
2946 }
2947
2948 sub GenerateIncludes($)
2949 {
2950         my ($self) = @_;
2951         if (is_intree()) {
2952                 $self->pidl("#include \"includes.h\"");
2953         } else {
2954                 $self->pidl("#ifndef _GNU_SOURCE");
2955                 $self->pidl("#define _GNU_SOURCE");
2956                 $self->pidl("#endif");
2957                 $self->pidl("#include <stdint.h>");
2958                 $self->pidl("#include <stdlib.h>");
2959                 $self->pidl("#include <stdio.h>");
2960                 $self->pidl("#include <stdbool.h>");
2961                 $self->pidl("#include <stdarg.h>");
2962                 $self->pidl("#include <string.h>");
2963         }
2964 }
2965
2966 #####################################################################
2967 # parse a parsed IDL structure back into an IDL file
2968 sub Parse($$$$)
2969 {
2970         my($self, $ndr,$gen_header,$ndr_header) = @_;
2971
2972         $self->pidl_hdr("/* header auto-generated by pidl */");
2973         $self->pidl_hdr("");
2974         $self->pidl_hdr(choose_header("librpc/ndr/libndr.h", "ndr.h"));
2975         $self->pidl_hdr("#include \"$gen_header\"") if ($gen_header);
2976         $self->pidl_hdr("");
2977
2978         $self->pidl("/* parser auto-generated by pidl */");
2979         $self->pidl("");
2980         $self->GenerateIncludes();
2981         $self->pidl("#include \"$ndr_header\"") if ($ndr_header);
2982         $self->pidl("");
2983
2984         my %needed = ();
2985
2986         foreach (@{$ndr}) {
2987                 ($_->{TYPE} eq "INTERFACE") && NeededInterface($_, \%needed);
2988         }
2989
2990         foreach (@{$ndr}) {
2991                 ($_->{TYPE} eq "INTERFACE") && $self->ParseInterface($_, \%needed);
2992                 ($_->{TYPE} eq "IMPORT") && $self->HeaderImport(@{$_->{PATHS}});
2993                 ($_->{TYPE} eq "INCLUDE") && $self->HeaderInclude(@{$_->{PATHS}});
2994         }
2995
2996         return ($self->{res_hdr}, $self->{res});
2997 }
2998
2999 sub NeededElement($$$)
3000 {
3001         my ($e, $dir, $needed) = @_;
3002
3003         return if ($e->{TYPE} eq "EMPTY");
3004
3005         return if (ref($e->{TYPE}) eq "HASH" and
3006                        not defined($e->{TYPE}->{NAME}));
3007
3008         my ($t, $rt);
3009         if (ref($e->{TYPE}) eq "HASH") {
3010                 $t = $e->{TYPE}->{TYPE}."_".$e->{TYPE}->{NAME};
3011         } else {
3012                 $t = $e->{TYPE};
3013         }
3014
3015         if (ref($e->{REPRESENTATION_TYPE}) eq "HASH") {
3016                 $rt = $e->{REPRESENTATION_TYPE}->{TYPE}."_".$e->{REPRESENTATION_TYPE}->{NAME};
3017         } else {
3018                 $rt = $e->{REPRESENTATION_TYPE};
3019         }
3020
3021         die ("$e->{NAME} $t, $rt FOO") unless ($rt ne "");
3022
3023         my @fn = ();
3024         if ($dir eq "print") {
3025                 push(@fn, TypeFunctionName("ndr_print", $e->{REPRESENTATION_TYPE}));
3026         } elsif ($dir eq "pull") {
3027                 push (@fn, TypeFunctionName("ndr_pull", $e->{TYPE}));
3028                 push (@fn, "ndr_$t\_to_$rt")
3029                         if ($rt ne $t);
3030         } elsif ($dir eq "push") {
3031                 push (@fn, TypeFunctionName("ndr_push", $e->{TYPE}));
3032                 push (@fn, "ndr_$rt\_to_$t")
3033                         if ($rt ne $t);
3034         } else {
3035                 die("invalid direction `$dir'");
3036         }
3037
3038         foreach (@fn) {
3039                 unless (defined($needed->{$_})) {
3040                         $needed->{$_} = 1;
3041                 }
3042         }
3043 }
3044
3045 sub NeededFunction($$)
3046 {
3047         my ($fn,$needed) = @_;
3048         $needed->{"ndr_pull_$fn->{NAME}"} = 1;
3049         $needed->{"ndr_push_$fn->{NAME}"} = 1;
3050         $needed->{"ndr_print_$fn->{NAME}"} = 1;
3051         foreach my $e (@{$fn->{ELEMENTS}}) {
3052                 $e->{PARENT} = $fn;
3053                 NeededElement($e, $_, $needed) foreach ("pull", "push", "print");
3054         }
3055 }
3056
3057 sub NeededType($$$)
3058 {
3059         sub NeededType($$$);
3060         my ($t,$needed,$req) = @_;
3061
3062         NeededType($t->{DATA}, $needed, $req) if ($t->{TYPE} eq "TYPEDEF");
3063         NeededType($t->{DATA}, $needed, $req) if ($t->{TYPE} eq "PIPE");
3064
3065         if ($t->{TYPE} eq "STRUCT" or $t->{TYPE} eq "UNION") {
3066                 return unless defined($t->{ELEMENTS});
3067                 for my $e (@{$t->{ELEMENTS}}) {
3068                         $e->{PARENT} = $t;
3069                         if (has_property($e, "compression")) {
3070                                 $needed->{"compression"} = 1;
3071                         }
3072                         NeededElement($e, $req, $needed);
3073                         NeededType($e->{TYPE}, $needed, $req) if (ref($e->{TYPE}) eq "HASH");
3074                 }
3075         }
3076 }
3077
3078 #####################################################################
3079 # work out what parse functions are needed
3080 sub NeededInterface($$)
3081 {
3082         my ($interface,$needed) = @_;
3083         NeededFunction($_, $needed) foreach (@{$interface->{FUNCTIONS}});
3084         foreach (reverse @{$interface->{TYPES}}) {
3085
3086                 if (has_property($_, "public")) {
3087                         $needed->{TypeFunctionName("ndr_pull", $_)} = $needed->{TypeFunctionName("ndr_push", $_)} =
3088                                 $needed->{TypeFunctionName("ndr_print", $_)} = 1;
3089                 }
3090
3091                 NeededType($_, $needed, "pull") if ($needed->{TypeFunctionName("ndr_pull", $_)});
3092                 NeededType($_, $needed, "push") if ($needed->{TypeFunctionName("ndr_push", $_)});
3093                 NeededType($_, $needed, "print") if ($needed->{TypeFunctionName("ndr_print", $_)});
3094                 if (has_property($_, "gensize")) {
3095                         $needed->{"ndr_size_$_->{NAME}"} = 1;
3096                 }
3097         }
3098 }
3099
3100 sub TypeFunctionName($$)
3101 {
3102         my ($prefix, $t) = @_;
3103
3104         return "$prefix\_$t->{NAME}" if (ref($t) eq "HASH" and
3105                         $t->{TYPE} eq "TYPEDEF");
3106         return "$prefix\_$t->{TYPE}_$t->{NAME}" if (ref($t) eq "HASH");
3107         return "$prefix\_$t";
3108 }
3109
3110 1;