source4/build/pidl/parser.pm

   1 ###################################################
   2 # Samba4 parser generator for IDL structures
   3 # Copyright tridge@samba.org 2000-2003
   4 # Copyright tpot@samba.org 2001
   5 # Copyright jelmer@samba.org 2004
   6 # released under the GNU GPL
   7
   8 package IdlParser;
   9
  10 use strict;
  11 use client;
  12 use needed;
  13
  14 # the list of needed functions
  15 my %structs;
  16
  17 sub pidl($)
  18 {
  19         print OUT shift;
  20 }
  21
  22 #####################################################################
  23 # parse a properties list
  24 sub ParseProperties($)
  25 {
  26     my($props) = shift;
  27     foreach my $d (@{$props}) {
  28         if (ref($d) ne "HASH") {
  29             pidl "[$d] ";
  30         } else {
  31             foreach my $k (keys %{$d}) {
  32                 pidl "[$k($d->{$k})] ";
  33             }
  34         }
  35     }
  36 }
  37
  38 ###################################
  39 # find a sibling var in a structure
  40 sub find_sibling($$)
  41 {
  42         my($e) = shift;
  43         my($name) = shift;
  44         my($fn) = $e->{PARENT};
  45
  46         if ($name =~ /\*(.*)/) {
  47                 $name = $1;
  48         }
  49
  50         if ($fn->{TYPE} eq "FUNCTION") {
  51                 for my $e2 (@{$fn->{DATA}}) {
  52                         if ($e2->{NAME} eq $name) {
  53                                 return $e2;
  54                         }
  55                 }
  56         }
  57
  58         for my $e2 (@{$fn->{ELEMENTS}}) {
  59                 if ($e2->{NAME} eq $name) {
  60                         return $e2;
  61                 }
  62         }
  63         die "invalid sibling '$name'";
  64 }
  65
  66 ####################################################################
  67 # work out the name of a size_is() variable
  68 sub find_size_var($$$)
  69 {
  70         my($e) = shift;
  71         my($size) = shift;
  72         my($var_prefix) = shift;
  73
  74         my($fn) = $e->{PARENT};
  75
  76         if (util::is_constant($size)) {
  77                 return $size;
  78         }
  79
  80         if ($size =~ /ndr->|\(/) {
  81                 return $size;
  82         }
  83
  84         my $prefix = "";
  85
  86         if ($size =~ /\*(.*)/) {
  87                 $size = $1;
  88                 $prefix = "*";
  89         }
  90
  91         if ($fn->{TYPE} ne "FUNCTION") {
  92                 return $prefix . "r->$size";
  93         }
  94
  95         my $e2 = find_sibling($e, $size);
  96
  97         if (util::has_property($e2, "in") && util::has_property($e2, "out")) {
  98                 return $prefix . "$var_prefix$size";
  99         }
 100         if (util::has_property($e2, "in")) {
 101                 return $prefix . "r->in.$size";
 102         }
 103         if (util::has_property($e2, "out")) {
 104                 return $prefix . "r->out.$size";
 105         }
 106
 107         die "invalid variable in $size for element $e->{NAME} in $fn->{NAME}\n";
 108 }
 109
 110
 111 #####################################################################
 112 # work out is a parse function should be declared static or not
 113 sub fn_prefix($)
 114 {
 115         my $fn = shift;
 116         if ($fn->{TYPE} eq "TYPEDEF") {
 117                 if (util::has_property($fn->{DATA}, "public")) {
 118                         return "";
 119                 }
 120         }
 121
 122         if ($fn->{TYPE} eq "FUNCTION") {
 123                 if (util::has_property($fn, "public")) {
 124                         return "";
 125                 }
 126         }
 127         return "static ";
 128 }
 129
 130
 131 ###################################################################
 132 # setup any special flags for an element or structure
 133 sub start_flags($)
 134 {
 135         my $e = shift;
 136         my $flags = util::has_property($e, "flag");
 137         if (defined $flags) {
 138                 pidl "\t{ uint32_t _flags_save_$e->{TYPE} = ndr->flags;\n";
 139                 pidl "\tndr_set_flags(&ndr->flags, $flags);\n";
 140         }
 141 }
 142
 143 ###################################################################
 144 # end any special flags for an element or structure
 145 sub end_flags($)
 146 {
 147         my $e = shift;
 148         my $flags = util::has_property($e, "flag");
 149         if (defined $flags) {
 150                 pidl "\tndr->flags = _flags_save_$e->{TYPE};\n\t}\n";
 151         }
 152 }
 153
 154
 155 #####################################################################
 156 # work out the correct alignment for a structure
 157 sub struct_alignment
 158 {
 159         my $s = shift;
 160
 161         my $align = 1;
 162         for my $e (@{$s->{ELEMENTS}}) {
 163                 my $a = 1;
 164
 165                 if (!util::need_wire_pointer($e)
 166                     && defined $structs{$e->{TYPE}}) {
 167                         if ($structs{$e->{TYPE}}->{DATA}->{TYPE} eq "STRUCT") {
 168                                 $a = struct_alignment($structs{$e->{TYPE}}->{DATA});
 169                         } elsif ($structs{$e->{TYPE}}->{DATA}->{TYPE} eq "UNION") {
 170                                 if (defined $structs{$e->{TYPE}}->{DATA}) {
 171                                         $a = union_alignment($structs{$e->{TYPE}}->{DATA});
 172                                 }
 173                         }
 174                 } else {
 175                         $a = util::type_align($e);
 176                 }
 177
 178                 if ($align < $a) {
 179                         $align = $a;
 180                 }
 181         }
 182
 183         return $align;
 184 }
 185
 186 #####################################################################
 187 # work out the correct alignment for a union
 188 sub union_alignment
 189 {
 190         my $u = shift;
 191
 192         my $align = 1;
 193
 194         foreach my $e (@{$u->{DATA}}) {
 195                 my $a = 1;
 196
 197                 if ($e->{TYPE} eq "EMPTY") {
 198                         next;
 199                 }
 200
 201                 if (!util::need_wire_pointer($e)
 202                     && defined $structs{$e->{DATA}->{TYPE}}) {
 203                         my $s = $structs{$e->{DATA}->{TYPE}};
 204                         if ($s->{DATA}->{TYPE} eq "STRUCT") {
 205                                 $a = struct_alignment($s->{DATA});
 206                         } elsif ($s->{DATA}->{TYPE} eq "UNION") {
 207                                 $a = union_alignment($s->{DATA});
 208                         }
 209                 } else {
 210                         $a = util::type_align($e->{DATA});
 211                 }
 212
 213                 if ($align < $a) {
 214                         $align = $a;
 215                 }
 216         }
 217
 218         return $align;
 219 }
 220
 221 #####################################################################
 222 # parse an array - push side
 223 sub ParseArrayPush($$$)
 224 {
 225         my $e = shift;
 226         my $var_prefix = shift;
 227         my $ndr_flags = shift;
 228
 229         my $size = find_size_var($e, util::array_size($e), $var_prefix);
 230
 231         if (defined $e->{CONFORMANT_SIZE}) {
 232                 # the conformant size has already been pushed
 233         } elsif (!util::is_inline_array($e)) {
 234                 # we need to emit the array size
 235                 pidl "\t\tNDR_CHECK(ndr_push_uint32(ndr, $size));\n";
 236         }
 237
 238         if (my $length = util::has_property($e, "length_is")) {
 239                 $length = find_size_var($e, $length, $var_prefix);
 240                 pidl "\t\tNDR_CHECK(ndr_push_uint32(ndr, 0));\n";
 241                 pidl "\t\tNDR_CHECK(ndr_push_uint32(ndr, $length));\n";
 242                 $size = $length;
 243         }
 244
 245         if (util::is_scalar_type($e->{TYPE})) {
 246                 pidl "\t\tNDR_CHECK(ndr_push_array_$e->{TYPE}(ndr, $ndr_flags, $var_prefix$e->{NAME}, $size));\n";
 247         } else {
 248                 pidl "\t\tNDR_CHECK(ndr_push_array(ndr, $ndr_flags, $var_prefix$e->{NAME}, sizeof($var_prefix$e->{NAME}\[0]), $size, (ndr_push_flags_fn_t)ndr_push_$e->{TYPE}));\n";
 249         }
 250 }
 251
 252 #####################################################################
 253 # print an array
 254 sub ParseArrayPrint($$)
 255 {
 256         my $e = shift;
 257         my $var_prefix = shift;
 258         my $size = find_size_var($e, util::array_size($e), $var_prefix);
 259         my $length = util::has_property($e, "length_is");
 260
 261         if (defined $length) {
 262                 $size = find_size_var($e, $length, $var_prefix);
 263         }
 264
 265         if (util::is_scalar_type($e->{TYPE})) {
 266                 pidl "\t\tndr_print_array_$e->{TYPE}(ndr, \"$e->{NAME}\", $var_prefix$e->{NAME}, $size);\n";
 267         } else {
 268                 pidl "\t\tndr_print_array(ndr, \"$e->{NAME}\", $var_prefix$e->{NAME}, sizeof($var_prefix$e->{NAME}\[0]), $size, (ndr_print_fn_t)ndr_print_$e->{TYPE});\n";
 269         }
 270 }
 271
 272 #####################################################################
 273 # parse an array - pull side
 274 sub ParseArrayPull($$$)
 275 {
 276         my $e = shift;
 277         my $var_prefix = shift;
 278         my $ndr_flags = shift;
 279
 280         my $size = find_size_var($e, util::array_size($e), $var_prefix);
 281         my $alloc_size = $size;
 282
 283         # if this is a conformant array then we use that size to allocate, and make sure
 284         # we allocate enough to pull the elements
 285         if (defined $e->{CONFORMANT_SIZE}) {
 286                 $alloc_size = $e->{CONFORMANT_SIZE};
 287
 288                 pidl "\tif ($size > $alloc_size) {\n";
 289                 pidl "\t\treturn ndr_pull_error(ndr, NDR_ERR_CONFORMANT_SIZE, \"Bad conformant size %u should be %u\", $alloc_size, $size);\n";
 290                 pidl "\t}\n";
 291         } elsif (!util::is_inline_array($e)) {
 292                 if ($var_prefix =~ /^r->out/ && $size =~ /^\*r->in/) {
 293                         my $size2 = substr($size, 1);
 294                         pidl "if (ndr->flags & LIBNDR_FLAG_REF_ALLOC) { NDR_ALLOC(ndr, $size2); }\n";
 295                 }
 296
 297                 # non fixed arrays encode the size just before the array
 298                 pidl "\t{\n";
 299                 pidl "\t\tuint32_t _array_size;\n";
 300                 pidl "\t\tNDR_CHECK(ndr_pull_uint32(ndr, &_array_size));\n";
 301                 if ($size =~ /r->in/) {
 302                         pidl "\t\tif (!(ndr->flags & LIBNDR_FLAG_REF_ALLOC) && _array_size != $size) {\n";
 303                 } else {
 304                         pidl "\t\tif ($size != _array_size) {\n";
 305                 }
 306                 pidl "\t\t\treturn ndr_pull_error(ndr, NDR_ERR_ARRAY_SIZE, \"Bad array size %u should be %u\", _array_size, $size);\n";
 307                 pidl "\t\t}\n";
 308                 if ($size =~ /r->in/) {
 309                         pidl "else { $size = _array_size; }\n";
 310                 }
 311                 pidl "\t}\n";
 312         }
 313
 314         if ((util::need_alloc($e) && !util::is_fixed_array($e)) ||
 315             ($var_prefix eq "r->in." && util::has_property($e, "ref"))) {
 316                 if (!util::is_inline_array($e) || $ndr_flags eq "NDR_SCALARS") {
 317                         pidl "\t\tNDR_ALLOC_N(ndr, $var_prefix$e->{NAME}, MAX(1, $alloc_size));\n";
 318                 }
 319         }
 320
 321         if (($var_prefix eq "r->out." && util::has_property($e, "ref"))) {
 322                 if (!util::is_inline_array($e) || $ndr_flags eq "NDR_SCALARS") {
 323                         pidl "\tif (ndr->flags & LIBNDR_FLAG_REF_ALLOC) {";
 324                         pidl "\t\tNDR_ALLOC_N(ndr, $var_prefix$e->{NAME}, MAX(1, $alloc_size));\n";
 325                         pidl "\t}\n";
 326                 }
 327         }
 328
 329         pidl "\t{\n";
 330
 331         if (my $length = util::has_property($e, "length_is")) {
 332                 $length = find_size_var($e, $length, $var_prefix);
 333                 pidl "\t\tuint32_t _offset, _length;\n";
 334                 pidl "\t\tNDR_CHECK(ndr_pull_uint32(ndr, &_offset));\n";
 335                 pidl "\t\tNDR_CHECK(ndr_pull_uint32(ndr, &_length));\n";
 336                 pidl "\t\tif (_offset != 0) return ndr_pull_error(ndr, NDR_ERR_OFFSET, \"Bad array offset 0x%08x\", _offset);\n";
 337                 pidl "\t\tif (_length > $size || _length != $length) return ndr_pull_error(ndr, NDR_ERR_LENGTH, \"Bad array length 0x%08x > size 0x%08x\", _offset, $size);\n\n";
 338                 $size = "_length";
 339         }
 340
 341         if (util::is_scalar_type($e->{TYPE})) {
 342                 pidl "\t\tNDR_CHECK(ndr_pull_array_$e->{TYPE}(ndr, $ndr_flags, $var_prefix$e->{NAME}, $size));\n";
 343         } else {
 344                 pidl "\t\tNDR_CHECK(ndr_pull_array(ndr, $ndr_flags, (void **)$var_prefix$e->{NAME}, sizeof($var_prefix$e->{NAME}\[0]), $size, (ndr_pull_flags_fn_t)ndr_pull_$e->{TYPE}));\n";
 345         }
 346
 347         pidl "\t}\n";
 348 }
 349
 350
 351 #####################################################################
 352 # parse scalars in a structure element
 353 sub ParseElementPushScalar($$$)
 354 {
 355         my($e) = shift;
 356         my($var_prefix) = shift;
 357         my($ndr_flags) = shift;
 358         my $cprefix = util::c_push_prefix($e);
 359         my $sub_size = util::has_property($e, "subcontext");
 360
 361         start_flags($e);
 362
 363         if (my $value = util::has_property($e, "value")) {
 364                 pidl "\t$cprefix$var_prefix$e->{NAME} = $value;\n";
 365         }
 366
 367         if (util::has_property($e, "relative")) {
 368                 pidl "\tNDR_CHECK(ndr_push_relative1(ndr, $var_prefix$e->{NAME}));\n";
 369         } elsif (util::is_inline_array($e)) {
 370                 ParseArrayPush($e, "r->", "NDR_SCALARS");
 371         } elsif (util::need_wire_pointer($e)) {
 372                 pidl "\tNDR_CHECK(ndr_push_ptr(ndr, $var_prefix$e->{NAME}));\n";
 373         } elsif (util::need_alloc($e)) {
 374                 # no scalar component
 375         } elsif (my $switch = util::has_property($e, "switch_is")) {
 376                 ParseElementPushSwitch($e, $var_prefix, $ndr_flags, $switch);
 377         } elsif (defined $sub_size) {
 378                 if (util::is_builtin_type($e->{TYPE})) {
 379                         pidl "\tNDR_CHECK(ndr_push_subcontext_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_push_fn_t) ndr_push_$e->{TYPE}));\n";
 380                 } else {
 381                         pidl "\tNDR_CHECK(ndr_push_subcontext_flags_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_push_flags_fn_t) ndr_push_$e->{TYPE}));\n";
 382                 }
 383         } elsif (util::is_builtin_type($e->{TYPE})) {
 384                 pidl "\tNDR_CHECK(ndr_push_$e->{TYPE}(ndr, $cprefix$var_prefix$e->{NAME}));\n";
 385         } else {
 386                 pidl "\tNDR_CHECK(ndr_push_$e->{TYPE}(ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}));\n";
 387         }
 388
 389         end_flags($e);
 390 }
 391
 392 #####################################################################
 393 # print scalars in a structure element
 394 sub ParseElementPrintScalar($$)
 395 {
 396         my($e) = shift;
 397         my($var_prefix) = shift;
 398         my $cprefix = util::c_push_prefix($e);
 399
 400         if (util::has_property($e, "noprint")) {
 401                 return;
 402         }
 403
 404         if (my $value = util::has_property($e, "value")) {
 405                 pidl "\tif (ndr->flags & LIBNDR_PRINT_SET_VALUES) {\n";
 406                 pidl "\t\t$cprefix$var_prefix$e->{NAME} = $value;\n";
 407                 pidl "\t}\n";
 408         }
 409
 410         if (util::is_fixed_array($e)) {
 411                 ParseElementPrintBuffer($e, $var_prefix);
 412         } elsif (util::has_direct_buffers($e)) {
 413                 pidl "\tndr_print_ptr(ndr, \"$e->{NAME}\", $var_prefix$e->{NAME});\n";
 414                 pidl "\tndr->depth++;\n";
 415                 ParseElementPrintBuffer($e, $var_prefix);
 416                 pidl "\tndr->depth--;\n";
 417         } elsif (my $switch = util::has_property($e, "switch_is")) {
 418                 ParseElementPrintSwitch($e, $var_prefix, $switch);
 419         } else {
 420                 pidl "\tndr_print_$e->{TYPE}(ndr, \"$e->{NAME}\", $cprefix$var_prefix$e->{NAME});\n";
 421         }
 422 }
 423
 424 #####################################################################
 425 # parse scalars in a structure element - pull size
 426 sub ParseElementPullSwitch($$$$)
 427 {
 428         my($e) = shift;
 429         my($var_prefix) = shift;
 430         my($ndr_flags) = shift;
 431         my $switch = shift;
 432         my $switch_var = find_size_var($e, $switch, $var_prefix);
 433
 434         my $cprefix = util::c_pull_prefix($e);
 435
 436         my $utype = $structs{$e->{TYPE}};
 437         if (!defined $utype ||
 438             !util::has_property($utype->{DATA}, "nodiscriminant")) {
 439                 my $e2 = find_sibling($e, $switch);
 440                 pidl "\tif (($ndr_flags) & NDR_SCALARS) {\n";
 441                 pidl "\t\t $e2->{TYPE} _level;\n";
 442                 pidl "\t\tNDR_CHECK(ndr_pull_$e2->{TYPE}(ndr, &_level));\n";
 443                 if ($switch_var =~ /r->in/) {
 444                         pidl "\t\tif (!(ndr->flags & LIBNDR_FLAG_REF_ALLOC) && _level != $switch_var) {\n";
 445                 } else {
 446                         pidl "\t\tif (_level != $switch_var) {\n";
 447                 }
 448                 pidl "\t\t\treturn ndr_pull_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value %u in $e->{NAME}\", _level);\n";
 449                 pidl "\t\t}\n";
 450                 if ($switch_var =~ /r->/) {
 451                         pidl "else { $switch_var = _level; }\n";
 452                 }
 453                 pidl "\t}\n";
 454         }
 455
 456         my $sub_size = util::has_property($e, "subcontext");
 457         if (defined $sub_size) {
 458                 pidl "\tif (($ndr_flags) & NDR_SCALARS) {\n";
 459                 pidl "\t\tNDR_CHECK(ndr_pull_subcontext_union_fn(ndr, $sub_size, $switch_var, $cprefix$var_prefix$e->{NAME}, (ndr_pull_union_fn_t) ndr_pull_$e->{TYPE}));\n";
 460                 pidl "\t}\n";
 461         } else {
 462                 pidl "\tNDR_CHECK(ndr_pull_$e->{TYPE}(ndr, $ndr_flags, $switch_var, $cprefix$var_prefix$e->{NAME}));\n";
 463         }
 464
 465
 466 }
 467
 468 #####################################################################
 469 # push switch element
 470 sub ParseElementPushSwitch($$$$)
 471 {
 472         my($e) = shift;
 473         my($var_prefix) = shift;
 474         my($ndr_flags) = shift;
 475         my $switch = shift;
 476         my $switch_var = find_size_var($e, $switch, $var_prefix);
 477         my $cprefix = util::c_push_prefix($e);
 478
 479         my $utype = $structs{$e->{TYPE}};
 480         if (!defined $utype ||
 481             !util::has_property($utype->{DATA}, "nodiscriminant")) {
 482                 my $e2 = find_sibling($e, $switch);
 483                 pidl "\tif (($ndr_flags) & NDR_SCALARS) {\n";
 484                 pidl "\t\tNDR_CHECK(ndr_push_$e2->{TYPE}(ndr, $switch_var));\n";
 485                 pidl "\t}\n";
 486         }
 487
 488         my $sub_size = util::has_property($e, "subcontext");
 489         if (defined $sub_size) {
 490                 pidl "\tif(($ndr_flags) & NDR_SCALARS) {\n";
 491                 pidl "\t\tNDR_CHECK(ndr_push_subcontext_union_fn(ndr, $sub_size, $switch_var, $cprefix$var_prefix$e->{NAME}, (ndr_push_union_fn_t) ndr_push_$e->{TYPE}));\n";
 492                 pidl "\t}\n";
 493         } else {
 494                 pidl "\tNDR_CHECK(ndr_push_$e->{TYPE}(ndr, $ndr_flags, $switch_var, $cprefix$var_prefix$e->{NAME}));\n";
 495         }
 496 }
 497
 498 #####################################################################
 499 # print scalars in a structure element
 500 sub ParseElementPrintSwitch($$$)
 501 {
 502         my($e) = shift;
 503         my($var_prefix) = shift;
 504         my $switch = shift;
 505         my $switch_var = find_size_var($e, $switch, $var_prefix);
 506         my $cprefix = util::c_push_prefix($e);
 507
 508         pidl "\tndr_print_$e->{TYPE}(ndr, \"$e->{NAME}\", $switch_var, $cprefix$var_prefix$e->{NAME});\n";
 509 }
 510
 511
 512 #####################################################################
 513 # parse scalars in a structure element - pull size
 514 sub ParseElementPullScalar($$$)
 515 {
 516         my($e) = shift;
 517         my($var_prefix) = shift;
 518         my($ndr_flags) = shift;
 519         my $cprefix = util::c_pull_prefix($e);
 520         my $sub_size = util::has_property($e, "subcontext");
 521
 522         start_flags($e);
 523
 524         if (util::is_inline_array($e)) {
 525                 ParseArrayPull($e, "r->", "NDR_SCALARS");
 526         } elsif (util::need_wire_pointer($e)) {
 527                 pidl "\tNDR_CHECK(ndr_pull_ptr(ndr, &_ptr_$e->{NAME}));\n";
 528                 pidl "\tif (_ptr_$e->{NAME}) {\n";
 529                 pidl "\t\tNDR_ALLOC(ndr, $var_prefix$e->{NAME});\n";
 530                 if (util::has_property($e, "relative")) {
 531                         pidl "\t\tNDR_CHECK(ndr_pull_relative1(ndr, $var_prefix$e->{NAME}, _ptr_$e->{NAME}));";
 532                 }
 533                 pidl "\t} else {\n";
 534                 pidl "\t\t$var_prefix$e->{NAME} = NULL;\n";
 535                 pidl "\t}\n";
 536         } elsif (util::need_alloc($e)) {
 537                 # no scalar component
 538         } elsif (my $switch = util::has_property($e, "switch_is")) {
 539                 ParseElementPullSwitch($e, $var_prefix, $ndr_flags, $switch);
 540         } elsif (defined $sub_size) {
 541                 if (util::is_builtin_type($e->{TYPE})) {
 542                         pidl "\tNDR_CHECK(ndr_pull_subcontext_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_pull_fn_t) ndr_pull_$e->{TYPE}));\n";
 543                 } else {
 544                         pidl "\tNDR_CHECK(ndr_pull_subcontext_flags_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_pull_flags_fn_t) ndr_pull_$e->{TYPE}));\n";
 545                 }
 546         } elsif (util::is_builtin_type($e->{TYPE})) {
 547                 pidl "\tNDR_CHECK(ndr_pull_$e->{TYPE}(ndr, $cprefix$var_prefix$e->{NAME}));\n";
 548         } else {
 549                 pidl "\tNDR_CHECK(ndr_pull_$e->{TYPE}(ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}));\n";
 550         }
 551         if (my $range = util::has_property($e, "range")) {
 552                 my ($low, $high) = split(/ /, $range, 2);
 553                 pidl "\tif ($var_prefix$e->{NAME} < $low || $var_prefix$e->{NAME} > $high) {\n";
 554                 pidl "\t\treturn ndr_pull_error(ndr, NDR_ERR_RANGE, \"value out of range\");\n\t}\n";
 555         }
 556
 557         end_flags($e);
 558 }
 559
 560 #####################################################################
 561 # parse buffers in a structure element
 562 sub ParseElementPushBuffer($$$)
 563 {
 564         my($e) = shift;
 565         my($var_prefix) = shift;
 566         my($ndr_flags) = shift;
 567         my $cprefix = util::c_push_prefix($e);
 568         my $sub_size = util::has_property($e, "subcontext");
 569
 570         if (util::is_pure_scalar($e)) {
 571                 return;
 572         }
 573
 574         start_flags($e);
 575
 576         if (util::need_wire_pointer($e)) {
 577                 pidl "\tif ($var_prefix$e->{NAME}) {\n";
 578                 if (util::has_property($e, "relative")) {
 579                         pidl "\t\tNDR_CHECK(ndr_push_relative2(ndr, $var_prefix$e->{NAME}));\n";
 580                 }
 581         }
 582
 583         if (util::is_inline_array($e)) {
 584                 ParseArrayPush($e, "r->", "NDR_BUFFERS");
 585         } elsif (util::array_size($e)) {
 586                 ParseArrayPush($e, "r->", "NDR_SCALARS|NDR_BUFFERS");
 587         } elsif (my $switch = util::has_property($e, "switch_is")) {
 588                 if ($e->{POINTERS}) {
 589                         ParseElementPushSwitch($e, $var_prefix, "NDR_BUFFERS|NDR_SCALARS", $switch);
 590                 } else {
 591                         ParseElementPushSwitch($e, $var_prefix, "NDR_BUFFERS", $switch);
 592                 }
 593         } elsif (defined $sub_size) {
 594                 if ($e->{POINTERS}) {
 595                         if (util::is_builtin_type($e->{TYPE})) {
 596                                 pidl "\tNDR_CHECK(ndr_push_subcontext_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_push_fn_t) ndr_push_$e->{TYPE}));\n";
 597                         } else {
 598                                 pidl "\tNDR_CHECK(ndr_push_subcontext_flags_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_push_flags_fn_t) ndr_push_$e->{TYPE}));\n";
 599                         }
 600                 }
 601         } elsif (util::is_builtin_type($e->{TYPE})) {
 602                 pidl "\t\tNDR_CHECK(ndr_push_$e->{TYPE}(ndr, $cprefix$var_prefix$e->{NAME}));\n";
 603         } elsif ($e->{POINTERS}) {
 604                 pidl "\t\tNDR_CHECK(ndr_push_$e->{TYPE}(ndr, NDR_SCALARS|NDR_BUFFERS, $cprefix$var_prefix$e->{NAME}));\n";
 605         } else {
 606                 pidl "\t\tNDR_CHECK(ndr_push_$e->{TYPE}(ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}));\n";
 607         }
 608
 609         if (util::need_wire_pointer($e)) {
 610                 pidl "\t}\n";
 611         }
 612
 613         end_flags($e);
 614 }
 615
 616 #####################################################################
 617 # print buffers in a structure element
 618 sub ParseElementPrintBuffer($$)
 619 {
 620         my($e) = shift;
 621         my($var_prefix) = shift;
 622         my $cprefix = util::c_push_prefix($e);
 623
 624         if (util::need_wire_pointer($e)) {
 625                 pidl "\tif ($var_prefix$e->{NAME}) {\n";
 626         }
 627
 628         if (util::array_size($e)) {
 629                 ParseArrayPrint($e, $var_prefix)
 630         } elsif (my $switch = util::has_property($e, "switch_is")) {
 631                 ParseElementPrintSwitch($e, $var_prefix, $switch);
 632         } else {
 633                 pidl "\t\tndr_print_$e->{TYPE}(ndr, \"$e->{NAME}\", $cprefix$var_prefix$e->{NAME});\n";
 634         }
 635
 636         if (util::need_wire_pointer($e)) {
 637                 pidl "\t}\n";
 638         }
 639 }
 640
 641
 642 #####################################################################
 643 # parse buffers in a structure element - pull side
 644 sub ParseElementPullBuffer($$$)
 645 {
 646         my($e) = shift;
 647         my($var_prefix) = shift;
 648         my($ndr_flags) = shift;
 649         my $cprefix = util::c_pull_prefix($e);
 650         my $sub_size = util::has_property($e, "subcontext");
 651
 652         if (util::is_pure_scalar($e)) {
 653                 return;
 654         }
 655
 656         start_flags($e);
 657
 658         if (util::need_wire_pointer($e)) {
 659                 pidl "\tif ($var_prefix$e->{NAME}) {\n";
 660                 if (util::has_property($e, "relative")) {
 661                         pidl "\t\tstruct ndr_pull_save _relative_save;\n";
 662                         pidl "\t\tndr_pull_save(ndr, &_relative_save);\n";
 663                         pidl "\t\tNDR_CHECK(ndr_pull_relative2(ndr, $var_prefix$e->{NAME}));\n";
 664                 }
 665         }
 666
 667         if (util::is_inline_array($e)) {
 668                 ParseArrayPull($e, "r->", "NDR_BUFFERS");
 669         } elsif (util::array_size($e)) {
 670                 ParseArrayPull($e, "r->", "NDR_SCALARS|NDR_BUFFERS");
 671         } elsif (my $switch = util::has_property($e, "switch_is")) {
 672                 if ($e->{POINTERS}) {
 673                         ParseElementPullSwitch($e, $var_prefix, "NDR_SCALARS|NDR_BUFFERS", $switch);
 674                 } else {
 675                         ParseElementPullSwitch($e, $var_prefix, "NDR_BUFFERS", $switch);
 676                 }
 677         } elsif (defined $sub_size) {
 678                 if ($e->{POINTERS}) {
 679                         if (util::is_builtin_type($e->{TYPE})) {
 680                                 pidl "\tNDR_CHECK(ndr_pull_subcontext_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_pull_fn_t) ndr_pull_$e->{TYPE}));\n";
 681                         } else {
 682                                 pidl "\tNDR_CHECK(ndr_pull_subcontext_flags_fn(ndr, $sub_size, $cprefix$var_prefix$e->{NAME}, (ndr_pull_flags_fn_t) ndr_pull_$e->{TYPE}));\n";
 683                         }
 684                 }
 685         } elsif (util::is_builtin_type($e->{TYPE})) {
 686                 pidl "\t\tNDR_CHECK(ndr_pull_$e->{TYPE}(ndr, $cprefix$var_prefix$e->{NAME}));\n";
 687         } elsif ($e->{POINTERS}) {
 688                 pidl "\t\tNDR_CHECK(ndr_pull_$e->{TYPE}(ndr, NDR_SCALARS|NDR_BUFFERS, $cprefix$var_prefix$e->{NAME}));\n";
 689         } else {
 690                 pidl "\t\tNDR_CHECK(ndr_pull_$e->{TYPE}(ndr, $ndr_flags, $cprefix$var_prefix$e->{NAME}));\n";
 691         }
 692
 693         if (util::need_wire_pointer($e)) {
 694                 if (util::has_property($e, "relative")) {
 695                         pidl "\t\tndr_pull_restore(ndr, &_relative_save);\n";
 696                 }
 697                 pidl "\t}\n";
 698         }
 699
 700         end_flags($e);
 701 }
 702
 703 #####################################################################
 704 # parse a struct
 705 sub ParseStructPush($)
 706 {
 707         my($struct) = shift;
 708         my $conform_e;
 709
 710         if (! defined $struct->{ELEMENTS}) {
 711                 return;
 712         }
 713
 714         start_flags($struct);
 715
 716         # see if the structure contains a conformant array. If it
 717         # does, then it must be the last element of the structure, and
 718         # we need to push the conformant length early, as it fits on
 719         # the wire before the structure (and even before the structure
 720         # alignment)
 721         my $e = $struct->{ELEMENTS}[-1];
 722         if (defined $e->{ARRAY_LEN} && $e->{ARRAY_LEN} eq "*") {
 723                 my $size = find_size_var($e, util::array_size($e), "r->");
 724                 $e->{CONFORMANT_SIZE} = $size;
 725                 $conform_e = $e;
 726                 pidl "\tNDR_CHECK(ndr_push_uint32(ndr, $size));\n";
 727         }
 728
 729         pidl "\tif (!(ndr_flags & NDR_SCALARS)) goto buffers;\n";
 730
 731         pidl "\tNDR_CHECK(ndr_push_struct_start(ndr));\n";
 732
 733         my $align = struct_alignment($struct);
 734         pidl "\tNDR_CHECK(ndr_push_align(ndr, $align));\n";
 735
 736         foreach my $e (@{$struct->{ELEMENTS}}) {
 737                 ParseElementPushScalar($e, "r->", "NDR_SCALARS");
 738         }
 739
 740         pidl "buffers:\n";
 741         pidl "\tif (!(ndr_flags & NDR_BUFFERS)) goto done;\n";
 742         foreach my $e (@{$struct->{ELEMENTS}}) {
 743                 ParseElementPushBuffer($e, "r->", "NDR_BUFFERS");
 744         }
 745
 746         pidl "\tndr_push_struct_end(ndr);\n";
 747
 748         pidl "done:\n";
 749
 750         end_flags($struct);
 751 }
 752
 753 #####################################################################
 754 # generate a struct print function
 755 sub ParseStructPrint($)
 756 {
 757         my($struct) = shift;
 758
 759         if (! defined $struct->{ELEMENTS}) {
 760                 return;
 761         }
 762
 763         start_flags($struct);
 764
 765         pidl "\tndr->depth++;\n";
 766         foreach my $e (@{$struct->{ELEMENTS}}) {
 767                 ParseElementPrintScalar($e, "r->");
 768         }
 769         pidl "\tndr->depth--;\n";
 770
 771         end_flags($struct);
 772 }
 773
 774 #####################################################################
 775 # parse a struct - pull side
 776 sub ParseStructPull($)
 777 {
 778         my($struct) = shift;
 779         my $conform_e;
 780
 781         if (! defined $struct->{ELEMENTS}) {
 782                 return;
 783         }
 784
 785         # see if the structure contains a conformant array. If it
 786         # does, then it must be the last element of the structure, and
 787         # we need to pull the conformant length early, as it fits on
 788         # the wire before the structure (and even before the structure
 789         # alignment)
 790         my $e = $struct->{ELEMENTS}[-1];
 791         if (defined $e->{ARRAY_LEN} && $e->{ARRAY_LEN} eq "*") {
 792                 $conform_e = $e;
 793                 pidl "\tuint32_t _conformant_size;\n";
 794                 $conform_e->{CONFORMANT_SIZE} = "_conformant_size";
 795         }
 796
 797         # declare any internal pointers we need
 798         foreach my $e (@{$struct->{ELEMENTS}}) {
 799                 if (util::need_wire_pointer($e)) {
 800                         pidl "\tuint32_t _ptr_$e->{NAME};\n";
 801                 }
 802         }
 803
 804         start_flags($struct);
 805
 806         pidl "\tif (!(ndr_flags & NDR_SCALARS)) goto buffers;\n";
 807
 808         pidl "\tNDR_CHECK(ndr_pull_struct_start(ndr));\n";
 809
 810         if (defined $conform_e) {
 811                 pidl "\tNDR_CHECK(ndr_pull_uint32(ndr, &$conform_e->{CONFORMANT_SIZE}));\n";
 812         }
 813
 814         my $align = struct_alignment($struct);
 815         pidl "\tNDR_CHECK(ndr_pull_align(ndr, $align));\n";
 816
 817         foreach my $e (@{$struct->{ELEMENTS}}) {
 818                 ParseElementPullScalar($e, "r->", "NDR_SCALARS");
 819         }
 820
 821         pidl "buffers:\n";
 822         pidl "\tif (!(ndr_flags & NDR_BUFFERS)) goto done;\n";
 823         foreach my $e (@{$struct->{ELEMENTS}}) {
 824                 ParseElementPullBuffer($e, "r->", "NDR_BUFFERS");
 825         }
 826
 827         pidl "\tndr_pull_struct_end(ndr);\n";
 828
 829         pidl "done:\n";
 830
 831         end_flags($struct);
 832 }
 833
 834 #####################################################################
 835 # calculate size of ndr struct
 836
 837 sub ParseStructNdrSize($)
 838 {
 839         my $t = shift;
 840         my $static = fn_prefix($t);
 841         my $sizevar;
 842
 843         pidl $static . "size_t ndr_size_$t->{NAME}(int ret, struct $t->{NAME} *r, int flags)\n";
 844         pidl "{\n";
 845
 846         if (util::has_property($t->{DATA}, "flag")) {
 847
 848                 pidl "\tflags = flags | " . $t->{DATA}->{PROPERTIES}->{flag} . ";\n";
 849         }
 850
 851         pidl "\tif(!r) return 0;\n";
 852
 853         pidl "\tret = NDR_SIZE_ALIGN(ret, " . struct_alignment($t->{DATA}) . ", flags);\n";
 854
 855         for my $e (@{$t->{DATA}->{ELEMENTS}}) {
 856                 my $switch = "";
 857
 858                 if (util::has_property($e, "subcontext")) {
 859                         pidl "\tret += $e->{PROPERTIES}->{subcontext}; /* Subcontext length */\n";
 860                 }
 861
 862                 if (util::has_property($e, "switch_is")) {
 863                         $switch = ", r->$e->{PROPERTIES}->{switch_is}";
 864                 }
 865
 866                 if ($e->{POINTERS} > 0) {
 867                         pidl "\tret = ndr_size_ptr(ret, &r->$e->{NAME}, flags); \n";
 868                 } elsif (util::is_inline_array($e)) {
 869                         $sizevar = find_size_var($e, util::array_size($e), "r->");
 870                         pidl "\t{\n";
 871                         pidl "\t\tint i;\n";
 872                         pidl "\t\tfor(i = 0; i < $sizevar; i++) {\n";
 873                         pidl "\t\t\tret = ndr_size_$e->{TYPE}(ret, &r->" . $e->{NAME} . "[i], flags);\n";
 874                         pidl "\t\t}\n";
 875                         pidl "\t}\n";
 876                 } else {
 877                         pidl "\tret = ndr_size_$e->{TYPE}(ret, &r->$e->{NAME}$switch, flags); \n";
 878                 }
 879         }
 880
 881         # Add lengths of relative members
 882         for my $e (@{$t->{DATA}->{ELEMENTS}}) {
 883                 next unless (util::has_property($e, "relative"));
 884
 885                 pidl "\tif (r->$e->{NAME}) {\n";
 886                 pidl "\t\tret = ndr_size_$e->{TYPE}(ret, r->$e->{NAME}, flags); \n";
 887                 pidl "\t}\n";
 888         }
 889
 890         pidl "\treturn ret;\n";
 891         pidl "}\n\n";
 892 }
 893
 894 #####################################################################
 895 # parse a union - push side
 896 sub ParseUnionPush($)
 897 {
 898         my $e = shift;
 899         my $have_default = 0;
 900
 901         start_flags($e);
 902
 903         pidl "\tif (!(ndr_flags & NDR_SCALARS)) goto buffers;\n";
 904
 905         pidl "\tNDR_CHECK(ndr_push_struct_start(ndr));\n";
 906
 907 #       my $align = union_alignment($e);
 908 #       pidl "\tNDR_CHECK(ndr_push_align(ndr, $align));\n";
 909
 910         pidl "\tswitch (level) {\n";
 911         foreach my $el (@{$e->{DATA}}) {
 912                 if ($el->{CASE} eq "default") {
 913                         pidl "\tdefault:\n";
 914                         $have_default = 1;
 915                 } else {
 916                         pidl "\tcase $el->{CASE}:\n";
 917                 }
 918                 if ($el->{TYPE} eq "UNION_ELEMENT") {
 919                         ParseElementPushScalar($el->{DATA}, "r->", "NDR_SCALARS");
 920                 }
 921                 pidl "\tbreak;\n\n";
 922         }
 923         if (! $have_default) {
 924                 pidl "\tdefault:\n";
 925                 pidl "\t\treturn ndr_push_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);\n";
 926         }
 927         pidl "\t}\n";
 928         pidl "buffers:\n";
 929         pidl "\tif (!(ndr_flags & NDR_BUFFERS)) goto done;\n";
 930         pidl "\tswitch (level) {\n";
 931         foreach my $el (@{$e->{DATA}}) {
 932                 if ($el->{CASE} eq "default") {
 933                         pidl "\tdefault:\n";
 934                 } else {
 935                         pidl "\tcase $el->{CASE}:\n";
 936                 }
 937                 if ($el->{TYPE} eq "UNION_ELEMENT") {
 938                         ParseElementPushBuffer($el->{DATA}, "r->", "NDR_BUFFERS");
 939                 }
 940                 pidl "\tbreak;\n\n";
 941         }
 942         if (! $have_default) {
 943                 pidl "\tdefault:\n";
 944                 pidl "\t\treturn ndr_push_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);\n";
 945         }
 946         pidl "\t}\n";
 947         pidl "\tndr_push_struct_end(ndr);\n";
 948         pidl "done:\n";
 949         end_flags($e);
 950 }
 951
 952 #####################################################################
 953 # print a union
 954 sub ParseUnionPrint($)
 955 {
 956         my $e = shift;
 957         my $have_default = 0;
 958
 959         start_flags($e);
 960
 961         pidl "\tswitch (level) {\n";
 962         foreach my $el (@{$e->{DATA}}) {
 963                 if ($el->{CASE} eq "default") {
 964                         $have_default = 1;
 965                         pidl "\tdefault:\n";
 966                 } else {
 967                         pidl "\tcase $el->{CASE}:\n";
 968                 }
 969                 if ($el->{TYPE} eq "UNION_ELEMENT") {
 970                         ParseElementPrintScalar($el->{DATA}, "r->");
 971                 }
 972                 pidl "\tbreak;\n\n";
 973         }
 974         if (! $have_default) {
 975                 pidl "\tdefault:\n\t\tndr_print_bad_level(ndr, name, level);\n";
 976         }
 977         pidl "\t}\n";
 978
 979         end_flags($e);
 980 }
 981
 982 #####################################################################
 983 # parse a union - pull side
 984 sub ParseUnionPull($)
 985 {
 986         my $e = shift;
 987         my $have_default = 0;
 988
 989         start_flags($e);
 990
 991         pidl "\tif (!(ndr_flags & NDR_SCALARS)) goto buffers;\n";
 992
 993         pidl "\tNDR_CHECK(ndr_pull_struct_start(ndr));\n";
 994
 995 #       my $align = union_alignment($e);
 996 #       pidl "\tNDR_CHECK(ndr_pull_align(ndr, $align));\n";
 997
 998         pidl "\tswitch (level) {\n";
 999         foreach my $el (@{$e->{DATA}}) {
1000                 if ($el->{CASE} eq "default") {
1001                         pidl "\tdefault: {\n";
1002                         $have_default = 1;
1003                 } else {
1004                         pidl "\tcase $el->{CASE}: {\n";
1005                 }
1006                 if ($el->{TYPE} eq "UNION_ELEMENT") {
1007                         my $e2 = $el->{DATA};
1008                         if ($e2->{POINTERS}) {
1009                                 pidl "\t\tuint32_t _ptr_$e2->{NAME};\n";
1010                         }
1011                         ParseElementPullScalar($el->{DATA}, "r->", "NDR_SCALARS");
1012                 }
1013                 pidl "\tbreak; }\n\n";
1014         }
1015         if (! $have_default) {
1016                 pidl "\tdefault:\n";
1017                 pidl "\t\treturn ndr_pull_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);\n";
1018         }
1019         pidl "\t}\n";
1020         pidl "buffers:\n";
1021         pidl "\tif (!(ndr_flags & NDR_BUFFERS)) goto done;\n";
1022         pidl "\tswitch (level) {\n";
1023         foreach my $el (@{$e->{DATA}}) {
1024                 if ($el->{CASE} eq "default") {
1025                         pidl "\tdefault:\n";
1026                 } else {
1027                         pidl "\tcase $el->{CASE}:\n";
1028                 }
1029                 if ($el->{TYPE} eq "UNION_ELEMENT") {
1030                         ParseElementPullBuffer($el->{DATA}, "r->", "NDR_BUFFERS");
1031                 }
1032                 pidl "\tbreak;\n\n";
1033         }
1034         if (! $have_default) {
1035                 pidl "\tdefault:\n";
1036                 pidl "\t\treturn ndr_pull_error(ndr, NDR_ERR_BAD_SWITCH, \"Bad switch value \%u\", level);\n";
1037         }
1038         pidl "\t}\n";
1039         pidl "\tndr_pull_struct_end(ndr);\n";
1040         pidl "done:\n";
1041         end_flags($e);
1042 }
1043
1044 #####################################################################
1045 # calculate size of ndr union
1046
1047 sub ParseUnionNdrSize($)
1048 {
1049         my $t = shift;
1050         my $static = fn_prefix($t);
1051
1052         pidl $static . "size_t ndr_size_$t->{NAME}(int ret, union $t->{NAME} *data, uint16 level, int flags)\n";
1053         pidl "{\n";
1054         if (util::has_property($t->{DATA}, "flag")) {
1055                 pidl "\tflags = flags | " . $t->{DATA}->{PROPERTIES}->{flag} . ";\n";
1056         }
1057         pidl "\tif(!data) return 0;\n\n";
1058
1059         pidl "\tret = NDR_SIZE_ALIGN(ret, " . union_alignment($t->{DATA}) . ", flags);\n";
1060
1061         pidl "\tswitch(level) {\n";
1062
1063         for my $e (@{$t->{DATA}->{DATA}}) {
1064                 if ($e->{TYPE} eq "UNION_ELEMENT") {
1065
1066                         if ($e->{CASE} eq "default") {
1067                                 pidl "\t\tdefault:";
1068                         } else {
1069                                 pidl "\t\tcase $e->{CASE}:";
1070                         }
1071
1072                         pidl " return ndr_size_$e->{DATA}->{TYPE}(ret, &data->$e->{DATA}->{NAME}, flags); \n";
1073
1074                 }
1075         }
1076         pidl "\t}\n";
1077         pidl "\treturn ret;\n";
1078         pidl "}\n\n";
1079 }
1080
1081 #####################################################################
1082 # parse a type
1083 sub ParseTypePush($)
1084 {
1085         my($data) = shift;
1086
1087         if (ref($data) eq "HASH") {
1088                 ($data->{TYPE} eq "STRUCT") &&
1089                     ParseStructPush($data);
1090                 ($data->{TYPE} eq "UNION") &&
1091                     ParseUnionPush($data);
1092         }
1093 }
1094
1095 #####################################################################
1096 # generate a print function for a type
1097 sub ParseTypePrint($)
1098 {
1099         my($data) = shift;
1100
1101         if (ref($data) eq "HASH") {
1102                 ($data->{TYPE} eq "STRUCT") &&
1103                     ParseStructPrint($data);
1104                 ($data->{TYPE} eq "UNION") &&
1105                     ParseUnionPrint($data);
1106         }
1107 }
1108
1109 #####################################################################
1110 # parse a type
1111 sub ParseTypePull($)
1112 {
1113         my($data) = shift;
1114
1115         if (ref($data) eq "HASH") {
1116                 ($data->{TYPE} eq "STRUCT") &&
1117                     ParseStructPull($data);
1118                 ($data->{TYPE} eq "UNION") &&
1119                     ParseUnionPull($data);
1120         }
1121 }
1122
1123 #####################################################################
1124 # parse a typedef - push side
1125 sub ParseTypedefPush($)
1126 {
1127         my($e) = shift;
1128         my $static = fn_prefix($e);
1129
1130         if (! needed::is_needed("push_$e->{NAME}")) {
1131 #               print "push_$e->{NAME} not needed\n";
1132                 return;
1133         }
1134
1135         if ($e->{DATA}->{TYPE} eq "STRUCT") {
1136                 pidl $static . "NTSTATUS ndr_push_$e->{NAME}(struct ndr_push *ndr, int ndr_flags, struct $e->{NAME} *r)";
1137                 pidl "\n{\n";
1138                 ParseTypePush($e->{DATA});
1139                 pidl "\treturn NT_STATUS_OK;\n";
1140                 pidl "}\n\n";
1141         }
1142
1143         if ($e->{DATA}->{TYPE} eq "UNION") {
1144                 pidl $static . "NTSTATUS ndr_push_$e->{NAME}(struct ndr_push *ndr, int ndr_flags, int level, union $e->{NAME} *r)";
1145                 pidl "\n{\n";
1146                 ParseTypePush($e->{DATA});
1147                 pidl "\treturn NT_STATUS_OK;\n";
1148                 pidl "}\n\n";
1149         }
1150 }
1151
1152
1153 #####################################################################
1154 # parse a typedef - pull side
1155 sub ParseTypedefPull($)
1156 {
1157         my($e) = shift;
1158         my $static = fn_prefix($e);
1159
1160         if (! needed::is_needed("pull_$e->{NAME}")) {
1161 #               print "pull_$e->{NAME} not needed\n";
1162                 return;
1163         }
1164
1165         if ($e->{DATA}->{TYPE} eq "STRUCT") {
1166                 pidl $static . "NTSTATUS ndr_pull_$e->{NAME}(struct ndr_pull *ndr, int ndr_flags, struct $e->{NAME} *r)";
1167                 pidl "\n{\n";
1168                 ParseTypePull($e->{DATA});
1169                 pidl "\treturn NT_STATUS_OK;\n";
1170                 pidl "}\n\n";
1171         }
1172
1173         if ($e->{DATA}->{TYPE} eq "UNION") {
1174                 pidl $static . "NTSTATUS ndr_pull_$e->{NAME}(struct ndr_pull *ndr, int ndr_flags, int level, union $e->{NAME} *r)";
1175                 pidl "\n{\n";
1176                 ParseTypePull($e->{DATA});
1177                 pidl "\treturn NT_STATUS_OK;\n";
1178                 pidl "}\n\n";
1179         }
1180 }
1181
1182
1183 #####################################################################
1184 # parse a typedef - print side
1185 sub ParseTypedefPrint($)
1186 {
1187         my($e) = shift;
1188
1189         if ($e->{DATA}->{TYPE} eq "STRUCT") {
1190                 pidl "void ndr_print_$e->{NAME}(struct ndr_print *ndr, const char *name, struct $e->{NAME} *r)";
1191                 pidl "\n{\n";
1192                 pidl "\tndr_print_struct(ndr, name, \"$e->{NAME}\");\n";
1193                 ParseTypePrint($e->{DATA});
1194                 pidl "}\n\n";
1195         }
1196
1197         if ($e->{DATA}->{TYPE} eq "UNION") {
1198                 pidl "void ndr_print_$e->{NAME}(struct ndr_print *ndr, const char *name, int level, union $e->{NAME} *r)";
1199                 pidl "\n{\n";
1200                 pidl "\tndr_print_union(ndr, name, level, \"$e->{NAME}\");\n";
1201                 ParseTypePrint($e->{DATA});
1202                 pidl "}\n\n";
1203         }
1204 }
1205
1206 #####################################################################
1207 ## calculate the size of a structure
1208 sub ParseTypedefNdrSize($)
1209 {
1210         my($t) = shift;
1211         if (! needed::is_needed("ndr_size_$t->{NAME}")) {
1212                 return;
1213         }
1214
1215         ($t->{DATA}->{TYPE} eq "STRUCT") &&
1216                 ParseStructNdrSize($t);
1217
1218         ($t->{DATA}->{TYPE} eq "UNION") &&
1219                 ParseUnionNdrSize($t);
1220
1221 }
1222
1223 #####################################################################
1224 # parse a function - print side
1225 sub ParseFunctionPrint($)
1226 {
1227         my($fn) = shift;
1228
1229         pidl "void ndr_print_$fn->{NAME}(struct ndr_print *ndr, const char *name, int flags, struct $fn->{NAME} *r)";
1230         pidl "\n{\n";
1231         pidl "\tndr_print_struct(ndr, name, \"$fn->{NAME}\");\n";
1232         pidl "\tndr->depth++;\n";
1233
1234         pidl "\tif (flags & NDR_SET_VALUES) {\n";
1235         pidl "\t\tndr->flags |= LIBNDR_PRINT_SET_VALUES;\n";
1236         pidl "\t}\n";
1237
1238         pidl "\tif (flags & NDR_IN) {\n";
1239         pidl "\t\tndr_print_struct(ndr, \"in\", \"$fn->{NAME}\");\n";
1240         pidl "\tndr->depth++;\n";
1241         foreach my $e (@{$fn->{DATA}}) {
1242                 if (util::has_property($e, "in")) {
1243                         ParseElementPrintScalar($e, "r->in.");
1244                 }
1245         }
1246         pidl "\tndr->depth--;\n";
1247         pidl "\t}\n";
1248
1249         pidl "\tif (flags & NDR_OUT) {\n";
1250         pidl "\t\tndr_print_struct(ndr, \"out\", \"$fn->{NAME}\");\n";
1251         pidl "\tndr->depth++;\n";
1252         foreach my $e (@{$fn->{DATA}}) {
1253                 if (util::has_property($e, "out")) {
1254                         ParseElementPrintScalar($e, "r->out.");
1255                 }
1256         }
1257         if ($fn->{RETURN_TYPE} && $fn->{RETURN_TYPE} ne "void") {
1258                 if (util::is_scalar_type($fn->{RETURN_TYPE})) {
1259                         pidl "\tndr_print_$fn->{RETURN_TYPE}(ndr, \"result\", r->out.result);\n";
1260                 } else {
1261                         pidl "\tndr_print_$fn->{RETURN_TYPE}(ndr, \"result\", &r->out.result);\n";
1262                 }
1263         }
1264         pidl "\tndr->depth--;\n";
1265         pidl "\t}\n";
1266
1267         pidl "\tndr->depth--;\n";
1268         pidl "}\n\n";
1269 }
1270
1271
1272 #####################################################################
1273 # parse a function element
1274 sub ParseFunctionElementPush($$)
1275 {
1276         my $e = shift;
1277         my $inout = shift;
1278
1279         if (util::array_size($e)) {
1280                 if (util::need_wire_pointer($e)) {
1281                         pidl "\tNDR_CHECK(ndr_push_ptr(ndr, r->$inout.$e->{NAME}));\n";
1282                         pidl "\tif (r->$inout.$e->{NAME}) {\n";
1283                         ParseArrayPush($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
1284                         pidl "\t}\n";
1285                 } else {
1286                         ParseArrayPush($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
1287                 }
1288         } else {
1289                 ParseElementPushScalar($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
1290                 if ($e->{POINTERS}) {
1291                         ParseElementPushBuffer($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
1292                 }
1293         }
1294 }
1295
1296 #####################################################################
1297 # parse a function
1298 sub ParseFunctionPush($)
1299 {
1300         my($fn) = shift;
1301         my $static = fn_prefix($fn);
1302
1303         pidl $static . "NTSTATUS ndr_push_$fn->{NAME}(struct ndr_push *ndr, int flags, struct $fn->{NAME} *r)\n{\n";
1304
1305         pidl "\n\tif (!(flags & NDR_IN)) goto ndr_out;\n\n";
1306         foreach my $e (@{$fn->{DATA}}) {
1307                 if (util::has_property($e, "in")) {
1308                         ParseFunctionElementPush($e, "in");
1309                 }
1310         }
1311
1312         pidl "\nndr_out:\n";
1313         pidl "\tif (!(flags & NDR_OUT)) goto done;\n\n";
1314         foreach my $e (@{$fn->{DATA}}) {
1315                 if (util::has_property($e, "out")) {
1316                         ParseFunctionElementPush($e, "out");
1317                 }
1318         }
1319
1320         if ($fn->{RETURN_TYPE} && $fn->{RETURN_TYPE} ne "void") {
1321                 pidl "\tNDR_CHECK(ndr_push_$fn->{RETURN_TYPE}(ndr, r->out.result));\n";
1322         }
1323
1324         pidl "\ndone:\n";
1325         pidl "\n\treturn NT_STATUS_OK;\n}\n\n";
1326 }
1327
1328 #####################################################################
1329 # parse a function element
1330 sub ParseFunctionElementPull($$)
1331 {
1332         my $e = shift;
1333         my $inout = shift;
1334
1335         if (util::array_size($e)) {
1336                 if (util::need_wire_pointer($e)) {
1337                         pidl "\tNDR_CHECK(ndr_pull_ptr(ndr, &_ptr_$e->{NAME}));\n";
1338                         pidl "\tif (_ptr_$e->{NAME}) {\n";
1339                 } elsif ($inout eq "out" && util::has_property($e, "ref")) {
1340                         pidl "\tif (r->$inout.$e->{NAME}) {\n";
1341                 } else {
1342                         pidl "\t{\n";
1343                 }
1344                 ParseArrayPull($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
1345                 pidl "\t}\n";
1346         } else {
1347                 if ($inout eq "out" && util::has_property($e, "ref")) {
1348                         pidl "\tif (ndr->flags & LIBNDR_FLAG_REF_ALLOC) {\n";
1349                         pidl "\tNDR_ALLOC(ndr, r->out.$e->{NAME});\n";
1350                         pidl "\t}\n";
1351                 }
1352                 if ($inout eq "in" && util::has_property($e, "ref")) {
1353                         pidl "\tNDR_ALLOC(ndr, r->in.$e->{NAME});\n";
1354                 }
1355
1356                 ParseElementPullScalar($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
1357                 if ($e->{POINTERS}) {
1358                         ParseElementPullBuffer($e, "r->$inout.", "NDR_SCALARS|NDR_BUFFERS");
1359                 }
1360         }
1361 }
1362
1363
1364 ############################################################
1365 # allocate ref variables
1366 sub AllocateRefVars($)
1367 {
1368         my $e = shift;
1369         my $asize = util::array_size($e);
1370
1371         # note that if the variable is also an "in"
1372         # variable then we copy the initial value from
1373         # the in side
1374
1375         if (!defined $asize) {
1376                 # its a simple variable
1377                 pidl "\tNDR_ALLOC(ndr, r->out.$e->{NAME});\n";
1378                 if (util::has_property($e, "in")) {
1379                         pidl "\t*r->out.$e->{NAME} = *r->in.$e->{NAME};\n";
1380                 } else {
1381                         pidl "\tZERO_STRUCTP(r->out.$e->{NAME});\n";
1382                 }
1383                 return;
1384         }
1385
1386         # its an array
1387         my $size = find_size_var($e, $asize, "r->out.");
1388         pidl "\tNDR_ALLOC_N(ndr, r->out.$e->{NAME}, MAX(1, $size));\n";
1389         if (util::has_property($e, "in")) {
1390                 pidl "\tmemcpy(r->out.$e->{NAME},r->in.$e->{NAME},$size * sizeof(*r->in.$e->{NAME}));\n";
1391         } else {
1392                 pidl "\tmemset(r->out.$e->{NAME}, 0, $size * sizeof(*r->out.$e->{NAME}));\n";
1393         }
1394 }
1395
1396
1397 #####################################################################
1398 # parse a function
1399 sub ParseFunctionPull($)
1400 {
1401         my($fn) = shift;
1402         my $static = fn_prefix($fn);
1403
1404         # pull function args
1405         pidl $static . "NTSTATUS ndr_pull_$fn->{NAME}(struct ndr_pull *ndr, int flags, struct $fn->{NAME} *r)\n{\n";
1406
1407         # declare any internal pointers we need
1408         foreach my $e (@{$fn->{DATA}}) {
1409                 if (util::need_wire_pointer($e)) {
1410                         pidl "\tuint32_t _ptr_$e->{NAME};\n";
1411                 }
1412         }
1413
1414         pidl "\n\tif (!(flags & NDR_IN)) goto ndr_out;\n\n";
1415
1416         # auto-init the out section of a structure. I originally argued that
1417         # this was a bad idea as it hides bugs, but coping correctly
1418         # with initialisation and not wiping ref vars is turning
1419         # out to be too tricky (tridge)
1420         foreach my $e (@{$fn->{DATA}}) {
1421                 if (util::has_property($e, "out")) {
1422                         pidl "\tZERO_STRUCT(r->out);\n\n";
1423                         last;
1424                 }
1425         }
1426
1427         foreach my $e (@{$fn->{DATA}}) {
1428                 if (util::has_property($e, "in")) {
1429                         ParseFunctionElementPull($e, "in");
1430                 }
1431                 # we need to allocate any reference output variables, so that
1432                 # a dcerpc backend can be sure they are non-null
1433                 if (util::has_property($e, "out") && util::has_property($e, "ref")) {
1434                         AllocateRefVars($e);
1435                 }
1436         }
1437
1438         pidl "\nndr_out:\n";
1439         pidl "\tif (!(flags & NDR_OUT)) goto done;\n\n";
1440         foreach my $e (@{$fn->{DATA}}) {
1441                 if (util::has_property($e, "out")) {
1442                         ParseFunctionElementPull($e, "out");
1443                 }
1444         }
1445
1446         if ($fn->{RETURN_TYPE} && $fn->{RETURN_TYPE} ne "void") {
1447                 pidl "\tNDR_CHECK(ndr_pull_$fn->{RETURN_TYPE}(ndr, &r->out.result));\n";
1448         }
1449
1450         pidl "\ndone:\n";
1451         pidl "\n\treturn NT_STATUS_OK;\n}\n\n";
1452 }
1453
1454 #####################################################################
1455 # produce a function call table
1456 sub FunctionTable($)
1457 {
1458         my($interface) = shift;
1459         my($data) = $interface->{INHERITED_DATA};
1460         my $count = 0;
1461         my $uname = uc $interface->{NAME};
1462
1463         foreach my $d (@{$data}) {
1464                 if ($d->{TYPE} eq "FUNCTION") { $count++; }
1465         }
1466
1467         if ($count == 0) {
1468                 return;
1469         }
1470
1471         pidl "static const struct dcerpc_interface_call $interface->{NAME}\_calls[] = {\n";
1472         foreach my $d (@{$data}) {
1473                 if ($d->{TYPE} eq "FUNCTION") {
1474                         pidl "\t{\n";
1475                         pidl "\t\t\"$d->{NAME}\",\n";
1476                         pidl "\t\tsizeof(struct $d->{NAME}),\n";
1477                         pidl "\t\t(ndr_push_flags_fn_t) ndr_push_$d->{NAME},\n";
1478                         pidl "\t\t(ndr_pull_flags_fn_t) ndr_pull_$d->{NAME},\n";
1479                         pidl "\t\t(ndr_print_function_t) ndr_print_$d->{NAME}\n";
1480                         pidl "\t},\n";
1481                 }
1482         }
1483         pidl "\t{ NULL, 0, NULL, NULL }\n};\n\n";
1484
1485         # If no endpoint is set, default to the interface name as a named pipe
1486         if (! defined $interface->{PROPERTIES}->{endpoint}) {
1487                 $interface->{PROPERTIES}->{endpoint} = "\"ncacn_np:[\\\\pipe\\\\" . $interface->{NAME} . "]\"";
1488         }
1489
1490         my @e = split / /, $interface->{PROPERTIES}->{endpoint};
1491         my $endpoint_count = $#e + 1;
1492
1493         pidl "static const char * const $interface->{NAME}\_endpoint_strings[] = {\n";
1494         foreach my $ep (@e) {
1495                 pidl "\t$ep, \n";
1496         }
1497         pidl "};\n\n";
1498
1499         pidl "static const struct dcerpc_endpoint_list $interface->{NAME}\_endpoints = {\n";
1500         pidl "\t$endpoint_count, $interface->{NAME}\_endpoint_strings\n";
1501         pidl "};\n\n";
1502
1503         pidl "\nconst struct dcerpc_interface_table dcerpc_table_$interface->{NAME} = {\n";
1504         pidl "\t\"$interface->{NAME}\",\n";
1505         pidl "\tDCERPC_$uname\_UUID,\n";
1506         pidl "\tDCERPC_$uname\_VERSION,\n";
1507         pidl "\tDCERPC_$uname\_HELPSTRING,\n";
1508         pidl "\t$count,\n";
1509         pidl "\t$interface->{NAME}\_calls,\n";
1510         pidl "\t&$interface->{NAME}\_endpoints\n";
1511         pidl "};\n\n";
1512 }
1513
1514
1515 #####################################################################
1516 # parse the interface definitions
1517 sub ParseInterface($)
1518 {
1519         my($interface) = shift;
1520         my($data) = $interface->{DATA};
1521
1522         foreach my $d (@{$data}) {
1523                 if ($d->{TYPE} eq "TYPEDEF") {
1524                     $structs{$d->{NAME}} = $d;
1525             }
1526         }
1527
1528         foreach my $d (@{$data}) {
1529                 ($d->{TYPE} eq "TYPEDEF") &&
1530                         ParseTypedefNdrSize($d);
1531         }
1532
1533         foreach my $d (@{$data}) {
1534                 ($d->{TYPE} eq "TYPEDEF") &&
1535                     ParseTypedefPush($d);
1536                 ($d->{TYPE} eq "FUNCTION") &&
1537                     ParseFunctionPush($d);
1538         }
1539         foreach my $d (@{$data}) {
1540                 ($d->{TYPE} eq "TYPEDEF") &&
1541                     ParseTypedefPull($d);
1542                 ($d->{TYPE} eq "FUNCTION") &&
1543                     ParseFunctionPull($d);
1544         }
1545         foreach my $d (@{$data}) {
1546                 if ($d->{TYPE} eq "TYPEDEF" &&
1547                     !util::has_property($d->{DATA}, "noprint")) {
1548                         ParseTypedefPrint($d);
1549                 }
1550                 if ($d->{TYPE} eq "FUNCTION" &&
1551                     !util::has_property($d, "noprint")) {
1552                         ParseFunctionPrint($d);
1553                 }
1554         }
1555
1556         FunctionTable($interface);
1557
1558 }
1559
1560 #####################################################################
1561 # parse a parsed IDL structure back into an IDL file
1562 sub Parse($$)
1563 {
1564         my($idl) = shift;
1565         my($filename) = shift;
1566         my $h_filename = $filename;
1567
1568         if ($h_filename =~ /(.*)\.c/) {
1569                 $h_filename = "$1.h";
1570         }
1571
1572         open(OUT, ">$filename") || die "can't open $filename";
1573
1574         pidl "/* parser auto-generated by pidl */\n\n";
1575         pidl "#include \"includes.h\"\n";
1576         pidl "#include \"$h_filename\"\n\n";
1577
1578         foreach my $x (@{$idl}) {
1579                 if ($x->{TYPE} eq "INTERFACE") {
1580                         needed::BuildNeeded($x);
1581                         ParseInterface($x);
1582                 }
1583         }
1584
1585         pidl IdlClient::Parse($idl);
1586
1587         close(OUT);
1588 }
1589
1590 1;