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

   1 ###################################################
   2 # client calls generator
   3 # Copyright tridge@samba.org 2003
   4 # Copyright jelmer@samba.org 2005-2006
   5 # released under the GNU GPL
   6
   7 package Parse::Pidl::Samba4::NDR::Client;
   8
   9 use Exporter;
  10 @ISA = qw(Exporter);
  11 @EXPORT_OK = qw(Parse);
  12
  13 use Parse::Pidl qw(fatal warning error);
  14 use Parse::Pidl::Util qw(has_property ParseExpr);
  15 use Parse::Pidl::Typelist qw(mapTypeName);
  16 use Parse::Pidl::Samba4 qw(choose_header is_intree DeclLong);
  17 use Parse::Pidl::Samba4::Header qw(GenerateFunctionInEnv GenerateFunctionOutEnv);
  18
  19 use vars qw($VERSION);
  20 $VERSION = '0.01';
  21
  22 use strict;
  23
  24 sub indent($) { my ($self) = @_; $self->{tabs}.="\t"; }
  25 sub deindent($) { my ($self) = @_; $self->{tabs} = substr($self->{tabs}, 1); }
  26 sub pidl($$) { my ($self,$txt) = @_; $self->{res} .= $txt ? "$self->{tabs}$txt\n" : "\n"; }
  27 sub pidl_hdr($$) { my ($self, $txt) = @_; $self->{res_hdr} .= "$txt\n"; }
  28 sub pidl_both($$) { my ($self, $txt) = @_; $self->{hdr} .= "$txt\n"; $self->{res_hdr} .= "$txt\n"; }
  29 sub fn_declare($$) { my ($self,$n) = @_; $self->pidl($n); $self->pidl_hdr("$n;"); }
  30
  31 sub genpad($)
  32 {
  33         my ($s) = @_;
  34         my $nt = int((length($s)+1)/8);
  35         my $lt = ($nt*8)-1;
  36         my $ns = (length($s)-$lt);
  37         return "\t"x($nt)." "x($ns);
  38 }
  39
  40 sub new($)
  41 {
  42         my ($class) = shift;
  43         my $self = { res => "", res_hdr => "", tabs => "" };
  44         bless($self, $class);
  45 }
  46
  47 sub ParseFunction_r_State($$$$)
  48 {
  49         my ($self, $if, $fn, $name) = @_;
  50         my $uname = uc $name;
  51
  52         $self->pidl("struct dcerpc_$name\_r_state {");
  53         $self->indent;
  54         $self->pidl("TALLOC_CTX *out_mem_ctx;");
  55         $self->deindent;
  56         $self->pidl("};");
  57         $self->pidl("");
  58         $self->pidl("static void dcerpc_$name\_r_done(struct tevent_req *subreq);");
  59         $self->pidl("");
  60 }
  61
  62 sub ParseFunction_r_Send($$$$)
  63 {
  64         my ($self, $if, $fn, $name) = @_;
  65         my $uname = uc $name;
  66
  67         my $proto = "struct tevent_req *dcerpc_$name\_r_send(TALLOC_CTX *mem_ctx,\n";
  68         $proto   .= "\tstruct tevent_context *ev,\n",
  69         $proto   .= "\tstruct dcerpc_binding_handle *h,\n",
  70         $proto   .= "\tstruct $name *r)";
  71
  72         $self->fn_declare($proto);
  73
  74         $self->pidl("{");
  75         $self->indent;
  76
  77         $self->pidl("struct tevent_req *req;");
  78         $self->pidl("struct dcerpc_$name\_r_state *state;");
  79         $self->pidl("struct tevent_req *subreq;");
  80         $self->pidl("");
  81
  82         $self->pidl("req = tevent_req_create(mem_ctx, &state,");
  83         $self->pidl("\t\t\tstruct dcerpc_$name\_r_state);");
  84         $self->pidl("if (req == NULL) {");
  85         $self->indent;
  86         $self->pidl("return NULL;");
  87         $self->deindent;
  88         $self->pidl("}");
  89         $self->pidl("");
  90
  91         my $out_params = 0;
  92         foreach (@{$fn->{ELEMENTS}}) {
  93                 if (grep(/out/, @{$_->{DIRECTION}})) {
  94                         $out_params++;
  95                 }
  96         }
  97
  98         my $submem;
  99         if ($out_params > 0) {
 100                 $self->pidl("state->out_mem_ctx = talloc_new(state);");
 101                 $self->pidl("if (tevent_req_nomem(state->out_mem_ctx, req)) {");
 102                 $self->indent;
 103                 $self->pidl("return tevent_req_post(req, ev);");
 104                 $self->deindent;
 105                 $self->pidl("}");
 106                 $self->pidl("");
 107                 $submem = "state->out_mem_ctx";
 108         } else {
 109                 $self->pidl("state->out_mem_ctx = NULL;");
 110                 $submem = "state";
 111         }
 112
 113         $self->pidl("subreq = dcerpc_binding_handle_call_send(state, ev, h,");
 114         $self->pidl("\t\tNULL, &ndr_table_$if->{NAME},");
 115         $self->pidl("\t\tNDR_$uname, $submem, r);");
 116         $self->pidl("if (tevent_req_nomem(subreq, req)) {");
 117         $self->indent;
 118         $self->pidl("return tevent_req_post(req, ev);");
 119         $self->deindent;
 120         $self->pidl("}");
 121         $self->pidl("tevent_req_set_callback(subreq, dcerpc_$name\_r_done, req);");
 122         $self->pidl("");
 123
 124         $self->pidl("return req;");
 125         $self->deindent;
 126         $self->pidl("}");
 127         $self->pidl("");
 128 }
 129
 130 sub ParseFunction_r_Done($$$$)
 131 {
 132         my ($self, $if, $fn, $name) = @_;
 133         my $uname = uc $name;
 134
 135         my $proto = "static void dcerpc_$name\_r_done(struct tevent_req *subreq)";
 136
 137         $self->pidl("$proto");
 138         $self->pidl("{");
 139         $self->indent;
 140
 141         $self->pidl("struct tevent_req *req =");
 142         $self->pidl("\ttevent_req_callback_data(subreq,");
 143         $self->pidl("\tstruct tevent_req);");
 144         $self->pidl("NTSTATUS status;");
 145         $self->pidl("");
 146
 147         $self->pidl("status = dcerpc_binding_handle_call_recv(subreq);");
 148         $self->pidl("if (!NT_STATUS_IS_OK(status)) {");
 149         $self->indent;
 150         $self->pidl("tevent_req_nterror(req, status);");
 151         $self->pidl("return;");
 152         $self->deindent;
 153         $self->pidl("}");
 154         $self->pidl("");
 155
 156         $self->pidl("tevent_req_done(req);");
 157         $self->deindent;
 158         $self->pidl("}");
 159         $self->pidl("");
 160 }
 161
 162 sub ParseFunction_r_Recv($$$$)
 163 {
 164         my ($if, $fn, $name) = @_;
 165         my ($self, $if, $fn, $name) = @_;
 166         my $uname = uc $name;
 167
 168         my $proto = "NTSTATUS dcerpc_$name\_r_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx)";
 169
 170         $self->fn_declare($proto);
 171
 172         $self->pidl("{");
 173         $self->indent;
 174
 175         $self->pidl("struct dcerpc_$name\_r_state *state =");
 176         $self->pidl("\ttevent_req_data(req,");
 177         $self->pidl("\tstruct dcerpc_$name\_r_state);");
 178         $self->pidl("NTSTATUS status;");
 179         $self->pidl("");
 180
 181         $self->pidl("if (tevent_req_is_nterror(req, &status)) {");
 182         $self->indent;
 183         $self->pidl("tevent_req_received(req);");
 184         $self->pidl("return status;");
 185         $self->deindent;
 186         $self->pidl("}");
 187         $self->pidl("");
 188
 189         $self->pidl("talloc_steal(mem_ctx, state->out_mem_ctx);");
 190         $self->pidl("");
 191
 192         $self->pidl("tevent_req_received(req);");
 193         $self->pidl("return NT_STATUS_OK;");
 194         $self->deindent;
 195         $self->pidl("}");
 196         $self->pidl("");
 197 }
 198
 199 sub ParseFunction_r_Sync($$$$)
 200 {
 201         my ($if, $fn, $name) = @_;
 202         my ($self, $if, $fn, $name) = @_;
 203         my $uname = uc $name;
 204
 205         my $proto = "NTSTATUS dcerpc_$name\_r(struct dcerpc_binding_handle *h, TALLOC_CTX *mem_ctx, struct $name *r)";
 206
 207         $self->fn_declare($proto);
 208
 209         $self->pidl("{");
 210         $self->indent;
 211         $self->pidl("NTSTATUS status;");
 212         $self->pidl("");
 213
 214         $self->pidl("status = dcerpc_binding_handle_call(h,");
 215         $self->pidl("\t\tNULL, &ndr_table_$if->{NAME},");
 216         $self->pidl("\t\tNDR_$uname, mem_ctx, r);");
 217         $self->pidl("");
 218         $self->pidl("return status;");
 219
 220         $self->deindent;
 221         $self->pidl("}");
 222         $self->pidl("");
 223 }
 224
 225 sub ParseFunction_Compat_Sync($$$$)
 226 {
 227         my ($if, $fn, $name) = @_;
 228         my ($self, $if, $fn, $name) = @_;
 229         my $uname = uc $name;
 230
 231         my $proto = "NTSTATUS dcerpc_$name\_compat(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct $name *r)";
 232
 233         $self->pidl_hdr("#ifdef DCERPC_CALL_$uname\_COMPAT");
 234         $self->pidl_hdr("#define dcerpc_$name(p, m, r) dcerpc_$name\_compat(p, m, r)");
 235         $self->pidl_hdr("#endif /* DCERPC_CALL_$uname\_COMPAT */");
 236
 237         $self->fn_declare($proto);
 238         $self->pidl("{");
 239         $self->indent;
 240         $self->pidl("NTSTATUS status;");
 241         $self->pidl("");
 242
 243         $self->pidl("status = dcerpc_$name\_r(p->binding_handle, mem_ctx, r);");
 244         $self->pidl("");
 245
 246         $self->pidl("if (NT_STATUS_IS_RPC(status)) {");
 247         $self->indent;
 248         $self->pidl("status = NT_STATUS_NET_WRITE_FAULT;");
 249         $self->deindent;
 250         $self->pidl("}");
 251         $self->pidl("");
 252
 253         if (defined($fn->{RETURN_TYPE}) and $fn->{RETURN_TYPE} eq "NTSTATUS") {
 254                 $self->pidl("if (NT_STATUS_IS_OK(status)) {");
 255                 $self->indent;
 256                 $self->pidl("status = r->out.result;");
 257                 $self->deindent;
 258                 $self->pidl("}");
 259                 $self->pidl("");
 260         }
 261
 262         $self->pidl("return status;");
 263
 264         $self->deindent;
 265         $self->pidl("}");
 266         $self->pidl("");
 267 }
 268
 269 sub ElementDirection($)
 270 {
 271         my ($e) = @_;
 272
 273         return "[in,out]" if (has_property($e, "in") and has_property($e, "out"));
 274         return "[in]" if (has_property($e, "in"));
 275         return "[out]" if (has_property($e, "out"));
 276         return "[in,out]";
 277 }
 278
 279 sub HeaderProperties($$)
 280 {
 281         my($props,$ignores) = @_;
 282         my $ret = "";
 283
 284         foreach my $d (keys %{$props}) {
 285                 next if (grep(/^$d$/, @$ignores));
 286                 if($props->{$d} ne "1") {
 287                         $ret.= "$d($props->{$d}),";
 288                 } else {
 289                         $ret.="$d,";
 290                 }
 291         }
 292
 293         if ($ret) {
 294                 return "[" . substr($ret, 0, -1) . "]";
 295         }
 296 }
 297
 298 sub ParseCopyArgument($$$$$)
 299 {
 300         my ($self, $fn, $e, $r, $i) = @_;
 301         my $l = $e->{LEVELS}[0];
 302
 303         if ($l->{TYPE} eq "ARRAY" and $l->{IS_FIXED} == 1) {
 304                 $self->pidl("memcpy(${r}$e->{NAME}, ${i}$e->{NAME}, sizeof(${r}$e->{NAME}));");
 305         } else {
 306                 $self->pidl("${r}$e->{NAME} = ${i}$e->{NAME};");
 307         }
 308 }
 309
 310 sub ParseInvalidResponse($$)
 311 {
 312         my ($self, $type) = @_;
 313
 314         if ($type eq "sync") {
 315                 $self->pidl("return NT_STATUS_INVALID_NETWORK_RESPONSE;");
 316         } elsif ($type eq "async") {
 317                 $self->pidl("tevent_req_nterror(req, NT_STATUS_INVALID_NETWORK_RESPONSE);");
 318                 $self->pidl("return;");
 319         } else {
 320                 die("ParseInvalidResponse($type)");
 321         }
 322 }
 323
 324 sub ParseOutputArgument($$$$$$)
 325 {
 326         my ($self, $fn, $e, $r, $o, $invalid_response_type) = @_;
 327         my $level = 0;
 328
 329         if ($e->{LEVELS}[0]->{TYPE} ne "POINTER" and $e->{LEVELS}[0]->{TYPE} ne "ARRAY") {
 330                 fatal($e->{ORIGINAL}, "[out] argument is not a pointer or array");
 331                 return;
 332         }
 333
 334         if ($e->{LEVELS}[0]->{TYPE} eq "POINTER") {
 335                 $level = 1;
 336                 if ($e->{LEVELS}[0]->{POINTER_TYPE} ne "ref") {
 337                         $self->pidl("if ($o$e->{NAME} && ${r}out.$e->{NAME}) {");
 338                         $self->indent;
 339                 }
 340         }
 341
 342         if ($e->{LEVELS}[$level]->{TYPE} eq "ARRAY") {
 343                 # This is a call to GenerateFunctionInEnv intentionally.
 344                 # Since the data is being copied into a user-provided data
 345                 # structure, the user should be able to know the size beforehand
 346                 # to allocate a structure of the right size.
 347                 my $in_env = GenerateFunctionInEnv($fn, $r);
 348                 my $out_env = GenerateFunctionOutEnv($fn, $r);
 349                 my $l = $e->{LEVELS}[$level];
 350
 351                 my $in_var = undef;
 352                 if (grep(/in/, @{$e->{DIRECTION}})) {
 353                         $in_var = ParseExpr($e->{NAME}, $in_env, $e->{ORIGINAL});
 354                 }
 355                 my $out_var = ParseExpr($e->{NAME}, $out_env, $e->{ORIGINAL});
 356
 357                 my $in_size_is = undef;
 358                 my $out_size_is = undef;
 359                 my $out_length_is = undef;
 360
 361                 my $avail_len = undef;
 362                 my $needed_len = undef;
 363
 364                 $self->pidl("{");
 365                 $self->indent;
 366                 my $copy_len_var = "_copy_len_$e->{NAME}";
 367                 $self->pidl("size_t $copy_len_var;");
 368
 369                 unless (defined($l->{SIZE_IS})) {
 370                         fatal($e->{ORIGINAL}, "no size known for [out] array `$e->{NAME}'");
 371                 } else {
 372                         $in_size_is = ParseExpr($l->{SIZE_IS}, $in_env, $e->{ORIGINAL});
 373                         $out_size_is = ParseExpr($l->{SIZE_IS}, $out_env, $e->{ORIGINAL});
 374                         $out_length_is = $out_size_is;
 375                         if (defined($l->{LENGTH_IS})) {
 376                                 $out_length_is = ParseExpr($l->{LENGTH_IS}, $out_env, $e->{ORIGINAL});
 377                         }
 378                         if (has_property($e, "charset")) {
 379                                 if (defined($in_var)) {
 380                                         $avail_len = "ndr_charset_length($in_var, CH_UNIX)";
 381                                 } else {
 382                                         $avail_len = $out_length_is;
 383                                 }
 384                                 $needed_len = "ndr_charset_length($out_var, CH_UNIX)";
 385                         }
 386                 }
 387
 388                 if ($out_size_is ne $in_size_is) {
 389                         $self->pidl("if (($out_size_is) > ($in_size_is)) {");
 390                         $self->indent;
 391                         $self->ParseInvalidResponse($invalid_response_type);
 392                         $self->deindent;
 393                         $self->pidl("}");
 394                 }
 395                 if ($out_length_is ne $out_size_is) {
 396                         $self->pidl("if (($out_length_is) > ($out_size_is)) {");
 397                         $self->indent;
 398                         $self->ParseInvalidResponse($invalid_response_type);
 399                         $self->deindent;
 400                         $self->pidl("}");
 401                 }
 402                 if (defined($needed_len)) {
 403                         $self->pidl("$copy_len_var = $needed_len;");
 404                         $self->pidl("if ($copy_len_var > $avail_len) {");
 405                         $self->indent;
 406                         $self->ParseInvalidResponse($invalid_response_type);
 407                         $self->deindent;
 408                         $self->pidl("}");
 409                 } else {
 410                         $self->pidl("$copy_len_var = $out_length_is;");
 411                 }
 412
 413                 if (has_property($e, "charset")) {
 414                         $self->pidl("memcpy(discard_const_p(uint8_t *, $o$e->{NAME}), $out_var, $copy_len_var * sizeof(*$o$e->{NAME}));");
 415                 } else {
 416                         $self->pidl("memcpy($o$e->{NAME}, $out_var, $copy_len_var * sizeof(*$o$e->{NAME}));");
 417                 }
 418
 419                 $self->deindent;
 420                 $self->pidl("}");
 421         } else {
 422                 $self->pidl("*$o$e->{NAME} = *${r}out.$e->{NAME};");
 423         }
 424
 425         if ($e->{LEVELS}[0]->{TYPE} eq "POINTER") {
 426                 if ($e->{LEVELS}[0]->{POINTER_TYPE} ne "ref") {
 427                         $self->deindent;
 428                         $self->pidl("}");
 429                 }
 430         }
 431 }
 432
 433 sub ParseFunction_State($$$$)
 434 {
 435         my ($self, $if, $fn, $name) = @_;
 436
 437         my $state_str = "struct dcerpc_$name\_state";
 438         my $done_fn = "dcerpc_$name\_done";
 439
 440         $self->pidl("$state_str {");
 441         $self->indent;
 442         $self->pidl("struct $name orig;");
 443         $self->pidl("struct $name tmp;");
 444         $self->pidl("TALLOC_CTX *out_mem_ctx;");
 445         $self->deindent;
 446         $self->pidl("};");
 447         $self->pidl("");
 448         $self->pidl("static void $done_fn(struct tevent_req *subreq);");
 449         $self->pidl("");
 450 }
 451
 452 sub ParseFunction_Send($$$$)
 453 {
 454         my ($self, $if, $fn, $name) = @_;
 455
 456         my $fn_args = "";
 457         my $state_str = "struct dcerpc_$name\_state";
 458         my $done_fn = "dcerpc_$name\_done";
 459         my $out_mem_ctx = "dcerpc_$name\_out_memory";
 460         my $fn_str = "struct tevent_req *dcerpc_$name\_send";
 461         my $pad = genpad($fn_str);
 462
 463         $fn_args .= "TALLOC_CTX *mem_ctx";
 464         $fn_args .= ",\n" . $pad . "struct tevent_context *ev";
 465         $fn_args .= ",\n" . $pad . "struct dcerpc_binding_handle *h";
 466
 467         foreach (@{$fn->{ELEMENTS}}) {
 468                 my $dir = ElementDirection($_);
 469                 my $prop = HeaderProperties($_->{PROPERTIES}, ["in", "out"]);
 470                 $fn_args .= ",\n" . $pad . DeclLong($_, "_") . " /* $dir $prop */";
 471         }
 472
 473         $self->fn_declare("$fn_str($fn_args)");
 474         $self->pidl("{");
 475         $self->indent;
 476         $self->pidl("struct tevent_req *req;");
 477         $self->pidl("$state_str *state;");
 478         $self->pidl("struct tevent_req *subreq;");
 479         $self->pidl("");
 480         $self->pidl("req = tevent_req_create(mem_ctx, &state,");
 481         $self->pidl("\t\t\t$state_str);");
 482         $self->pidl("if (req == NULL) {");
 483         $self->indent;
 484         $self->pidl("return NULL;");
 485         $self->deindent;
 486         $self->pidl("}");
 487         $self->pidl("state->out_mem_ctx = NULL;");
 488         $self->pidl("");
 489
 490         $self->pidl("/* In parameters */");
 491         foreach my $e (@{$fn->{ELEMENTS}}) {
 492                 next unless (grep(/in/, @{$e->{DIRECTION}}));
 493
 494                 $self->ParseCopyArgument($fn, $e, "state->orig.in.", "_");
 495         }
 496         $self->pidl("");
 497
 498         my $out_params = 0;
 499         $self->pidl("/* Out parameters */");
 500         foreach my $e (@{$fn->{ELEMENTS}}) {
 501                 next unless grep(/out/, @{$e->{DIRECTION}});
 502
 503                 $self->ParseCopyArgument($fn, $e, "state->orig.out.", "_");
 504                 $out_params++;
 505         }
 506         $self->pidl("");
 507
 508         if (defined($fn->{RETURN_TYPE})) {
 509                 $self->pidl("/* Result */");
 510                 $self->pidl("ZERO_STRUCT(state->orig.out.result);");
 511                 $self->pidl("");
 512         }
 513
 514         if ($out_params > 0) {
 515                 $self->pidl("state->out_mem_ctx = talloc_named_const(state, 0,");
 516                 $self->pidl("\t\t     \"$out_mem_ctx\");");
 517                 $self->pidl("if (tevent_req_nomem(state->out_mem_ctx, req)) {");
 518                 $self->indent;
 519                 $self->pidl("return tevent_req_post(req, ev);");
 520                 $self->deindent;
 521                 $self->pidl("}");
 522                 $self->pidl("");
 523         }
 524
 525         $self->pidl("/* make a temporary copy, that we pass to the dispatch function */");
 526         $self->pidl("state->tmp = state->orig;");
 527         $self->pidl("");
 528
 529         $self->pidl("subreq = dcerpc_$name\_r_send(state, ev, h, &state->tmp);");
 530         $self->pidl("if (tevent_req_nomem(subreq, req)) {");
 531         $self->indent;
 532         $self->pidl("return tevent_req_post(req, ev);");
 533         $self->deindent;
 534         $self->pidl("}");
 535         $self->pidl("tevent_req_set_callback(subreq, $done_fn, req);");
 536         $self->pidl("return req;");
 537         $self->deindent;
 538         $self->pidl("}");
 539         $self->pidl("");
 540 }
 541
 542 sub ParseFunction_Done($$$$)
 543 {
 544         my ($self, $if, $fn, $name) = @_;
 545
 546         my $state_str = "struct dcerpc_$name\_state";
 547         my $done_fn = "dcerpc_$name\_done";
 548
 549         $self->pidl("static void $done_fn(struct tevent_req *subreq)");
 550         $self->pidl("{");
 551         $self->indent;
 552         $self->pidl("struct tevent_req *req = tevent_req_callback_data(");
 553         $self->pidl("\tsubreq, struct tevent_req);");
 554         $self->pidl("$state_str *state = tevent_req_data(");
 555         $self->pidl("\treq, $state_str);");
 556         $self->pidl("NTSTATUS status;");
 557         $self->pidl("TALLOC_CTX *mem_ctx;");
 558         $self->pidl("");
 559
 560         $self->pidl("if (state->out_mem_ctx) {");
 561         $self->indent;
 562         $self->pidl("mem_ctx = state->out_mem_ctx;");
 563         $self->deindent;
 564         $self->pidl("} else {");
 565         $self->indent;
 566         $self->pidl("mem_ctx = state;");
 567         $self->deindent;
 568         $self->pidl("}");
 569         $self->pidl("");
 570
 571         $self->pidl("status = dcerpc_$name\_r_recv(subreq, mem_ctx);");
 572         $self->pidl("TALLOC_FREE(subreq);");
 573         $self->pidl("if (!NT_STATUS_IS_OK(status)) {");
 574         $self->indent;
 575         $self->pidl("tevent_req_nterror(req, status);");
 576         $self->pidl("return;");
 577         $self->deindent;
 578         $self->pidl("}");
 579         $self->pidl("");
 580
 581         $self->pidl("/* Copy out parameters */");
 582         foreach my $e (@{$fn->{ELEMENTS}}) {
 583                 next unless (grep(/out/, @{$e->{DIRECTION}}));
 584
 585                 $self->ParseOutputArgument($fn, $e,
 586                                            "state->tmp.",
 587                                            "state->orig.out.",
 588                                            "async");
 589         }
 590         $self->pidl("");
 591
 592         if (defined($fn->{RETURN_TYPE})) {
 593                 $self->pidl("/* Copy result */");
 594                 $self->pidl("state->orig.out.result = state->tmp.out.result;");
 595                 $self->pidl("");
 596         }
 597
 598         $self->pidl("/* Reset temporary structure */");
 599         $self->pidl("ZERO_STRUCT(state->tmp);");
 600         $self->pidl("");
 601
 602         $self->pidl("tevent_req_done(req);");
 603         $self->deindent;
 604         $self->pidl("}");
 605         $self->pidl("");
 606 }
 607
 608 sub ParseFunction_Recv($$$$)
 609 {
 610         my ($self, $if, $fn, $name) = @_;
 611
 612         my $fn_args = "";
 613         my $state_str = "struct dcerpc_$name\_state";
 614         my $fn_str = "NTSTATUS dcerpc_$name\_recv";
 615         my $pad = genpad($fn_str);
 616
 617         $fn_args .= "struct tevent_req *req,\n" . $pad . "TALLOC_CTX *mem_ctx";
 618
 619         if (defined($fn->{RETURN_TYPE})) {
 620                 $fn_args .= ",\n" . $pad . mapTypeName($fn->{RETURN_TYPE}). " *result";
 621         }
 622
 623         $self->fn_declare("$fn_str($fn_args)");
 624         $self->pidl("{");
 625         $self->indent;
 626         $self->pidl("$state_str *state = tevent_req_data(");
 627         $self->pidl("\treq, $state_str);");
 628         $self->pidl("NTSTATUS status;");
 629         $self->pidl("");
 630         $self->pidl("if (tevent_req_is_nterror(req, &status)) {");
 631         $self->indent;
 632         $self->pidl("tevent_req_received(req);");
 633         $self->pidl("return status;");
 634         $self->deindent;
 635         $self->pidl("}");
 636         $self->pidl("");
 637
 638         $self->pidl("/* Steal possible out parameters to the callers context */");
 639         $self->pidl("talloc_steal(mem_ctx, state->out_mem_ctx);");
 640         $self->pidl("");
 641
 642         if (defined($fn->{RETURN_TYPE})) {
 643                 $self->pidl("/* Return result */");
 644                 $self->pidl("*result = state->orig.out.result;");
 645                 $self->pidl("");
 646         }
 647
 648         $self->pidl("tevent_req_received(req);");
 649         $self->pidl("return NT_STATUS_OK;");
 650         $self->deindent;
 651         $self->pidl("}");
 652         $self->pidl("");
 653 }
 654
 655 sub ParseFunction_Sync($$$$)
 656 {
 657         my ($self, $if, $fn, $name) = @_;
 658
 659         my $uname = uc $name;
 660         my $fn_args = "";
 661         my $fn_str = "NTSTATUS dcerpc_$name";
 662         my $pad = genpad($fn_str);
 663
 664         $fn_args .= "struct dcerpc_binding_handle *h,\n" . $pad . "TALLOC_CTX *mem_ctx";
 665
 666         foreach (@{$fn->{ELEMENTS}}) {
 667                 my $dir = ElementDirection($_);
 668                 my $prop = HeaderProperties($_->{PROPERTIES}, ["in", "out"]);
 669                 $fn_args .= ",\n" . $pad . DeclLong($_, "_") . " /* $dir $prop */";
 670         }
 671
 672         if (defined($fn->{RETURN_TYPE})) {
 673                 $fn_args .= ",\n" . $pad . mapTypeName($fn->{RETURN_TYPE}). " *result";
 674         }
 675
 676         $self->pidl_hdr("#ifndef DCERPC_CALL_$uname\_COMPAT");
 677         $self->fn_declare("$fn_str($fn_args)");
 678         $self->pidl_hdr("#endif /* DCERPC_CALL_$uname\_COMPAT */");
 679         $self->pidl("{");
 680         $self->indent;
 681         $self->pidl("struct $name r;");
 682         $self->pidl("NTSTATUS status;");
 683         $self->pidl("");
 684
 685         $self->pidl("/* In parameters */");
 686         foreach my $e (@{$fn->{ELEMENTS}}) {
 687                 next unless (grep(/in/, @{$e->{DIRECTION}}));
 688
 689                 $self->ParseCopyArgument($fn, $e, "r.in.", "_");
 690         }
 691         $self->pidl("");
 692
 693         $self->pidl("status = dcerpc_$name\_r(h, mem_ctx, &r);");
 694         $self->pidl("if (!NT_STATUS_IS_OK(status)) {");
 695         $self->indent;
 696         $self->pidl("return status;");
 697         $self->deindent;
 698         $self->pidl("}");
 699         $self->pidl("");
 700
 701         $self->pidl("/* Return variables */");
 702         foreach my $e (@{$fn->{ELEMENTS}}) {
 703                 next unless (grep(/out/, @{$e->{DIRECTION}}));
 704
 705                 $self->ParseOutputArgument($fn, $e, "r.", "_", "sync");
 706         }
 707         $self->pidl("");
 708
 709         $self->pidl("/* Return result */");
 710         if ($fn->{RETURN_TYPE}) {
 711                 $self->pidl("*result = r.out.result;");
 712         }
 713         $self->pidl("");
 714
 715         $self->pidl("return NT_STATUS_OK;");
 716
 717         $self->deindent;
 718         $self->pidl("}");
 719         $self->pidl("");
 720 }
 721
 722 #####################################################################
 723 # parse a function
 724 sub ParseFunction($$$)
 725 {
 726         my ($self, $if, $fn) = @_;
 727
 728         $self->ParseFunction_r_State($if, $fn, $fn->{NAME});
 729         $self->ParseFunction_r_Send($if, $fn, $fn->{NAME});
 730         $self->ParseFunction_r_Done($if, $fn, $fn->{NAME});
 731         $self->ParseFunction_r_Recv($if, $fn, $fn->{NAME});
 732         $self->ParseFunction_r_Sync($if, $fn, $fn->{NAME});
 733
 734         $self->ParseFunction_Compat_Sync($if, $fn, $fn->{NAME});
 735
 736         foreach my $e (@{$fn->{ELEMENTS}}) {
 737                 next unless (grep(/out/, @{$e->{DIRECTION}}));
 738
 739                 my $reason = "is not a pointer or array";
 740
 741                 # TODO: make this fatal at NDR level
 742                 if ($e->{LEVELS}[0]->{TYPE} eq "POINTER") {
 743                         if ($e->{LEVELS}[1]->{TYPE} eq "DATA" and
 744                             $e->{LEVELS}[1]->{DATA_TYPE} eq "string") {
 745                                 $reason = "is a pointer to type 'string'";
 746                         } elsif ($e->{LEVELS}[1]->{TYPE} eq "ARRAY" and
 747                                  not defined($e->{LEVELS}[1]->{SIZE_IS})) {
 748                                 $reason = "is a pointer to an unsized array";
 749                         } else {
 750                                 next;
 751                         }
 752                 }
 753                 if ($e->{LEVELS}[0]->{TYPE} eq "ARRAY") {
 754                         if (not defined($e->{LEVELS}[0]->{SIZE_IS})) {
 755                                 $reason = "is an unsized array";
 756                         } else {
 757                                 next;
 758                         }
 759                 }
 760
 761                 $self->pidl_both("/*");
 762                 $self->pidl_both(" * The following functions are skipped because");
 763                 $self->pidl_both(" * an [out] argument $e->{NAME} $reason:");
 764                 $self->pidl_both(" *");
 765                 $self->pidl_both(" * dcerpc_$fn->{NAME}_send()");
 766                 $self->pidl_both(" * dcerpc_$fn->{NAME}_recv()");
 767                 $self->pidl_both(" * dcerpc_$fn->{NAME}()");
 768                 $self->pidl_both(" */");
 769                 $self->pidl_both("");
 770
 771                 error($e->{ORIGINAL}, "$fn->{NAME}: [out] argument '$e->{NAME}' $reason, skip client functions");
 772                 return;
 773         }
 774
 775         $self->ParseFunction_State($if, $fn, $fn->{NAME});
 776         $self->ParseFunction_Send($if, $fn, $fn->{NAME});
 777         $self->ParseFunction_Done($if, $fn, $fn->{NAME});
 778         $self->ParseFunction_Recv($if, $fn, $fn->{NAME});
 779         $self->ParseFunction_Sync($if, $fn, $fn->{NAME});
 780
 781         $self->pidl_hdr("");
 782 }
 783
 784 my %done;
 785
 786 #####################################################################
 787 # parse the interface definitions
 788 sub ParseInterface($$)
 789 {
 790         my ($self, $if) = @_;
 791         my $ifu = uc($if->{NAME});
 792
 793         $self->pidl_hdr("#ifndef _HEADER_RPC_$if->{NAME}");
 794         $self->pidl_hdr("#define _HEADER_RPC_$if->{NAME}");
 795         $self->pidl_hdr("");
 796
 797         if (defined $if->{PROPERTIES}->{uuid}) {
 798                 $self->pidl_hdr("extern const struct ndr_interface_table ndr_table_$if->{NAME};");
 799                 $self->pidl_hdr("");
 800         }
 801
 802         $self->pidl_hdr("#ifdef DCERPC_IFACE_$ifu\_COMPAT");
 803         foreach my $fn (@{$if->{FUNCTIONS}}) {
 804                 next if has_property($fn, "noopnum");
 805                 next if has_property($fn, "todo");
 806                 my $fnu = uc($fn->{NAME});
 807                 $self->pidl_hdr("#define DCERPC_CALL_$fnu\_COMPAT 1");
 808         }
 809         $self->pidl_hdr("#endif /* DCERPC_IFACE_$ifu\_COMPAT */");
 810         $self->pidl_hdr("");
 811
 812         $self->pidl("/* $if->{NAME} - client functions generated by pidl */");
 813         $self->pidl("");
 814
 815         foreach my $fn (@{$if->{FUNCTIONS}}) {
 816                 next if defined($done{$fn->{NAME}});
 817                 next if has_property($fn, "noopnum");
 818                 next if has_property($fn, "todo");
 819                 $self->ParseFunction($if, $fn);
 820                 $done{$fn->{NAME}} = 1;
 821         }
 822
 823         $self->pidl_hdr("#endif /* _HEADER_RPC_$if->{NAME} */");
 824 }
 825
 826 sub Parse($$$$$$)
 827 {
 828         my($self,$ndr,$header,$ndr_header,$client_header) = @_;
 829
 830         $self->pidl("/* client functions auto-generated by pidl */");
 831         $self->pidl("");
 832         if (is_intree()) {
 833                 $self->pidl("#include \"includes.h\"");
 834         } else {
 835                 $self->pidl("#ifndef _GNU_SOURCE");
 836                 $self->pidl("#define _GNU_SOURCE");
 837                 $self->pidl("#endif");
 838                 $self->pidl("#include <stdio.h>");
 839                 $self->pidl("#include <stdbool.h>");
 840                 $self->pidl("#include <stdlib.h>");
 841                 $self->pidl("#include <stdint.h>");
 842                 $self->pidl("#include <stdarg.h>");
 843                 $self->pidl("#include <string.h>");
 844                 $self->pidl("#include <core/ntstatus.h>");
 845         }
 846         $self->pidl("#include <tevent.h>");
 847         $self->pidl(choose_header("lib/util/tevent_ntstatus.h", "util/tevent_ntstatus.h")."");
 848         $self->pidl("#include \"$ndr_header\"");
 849         $self->pidl("#include \"$client_header\"");
 850         $self->pidl("");
 851
 852         $self->pidl_hdr(choose_header("librpc/rpc/dcerpc.h", "dcerpc.h")."");
 853         $self->pidl_hdr("#include \"$header\"");
 854
 855         foreach my $x (@{$ndr}) {
 856                 ($x->{TYPE} eq "INTERFACE") && $self->ParseInterface($x);
 857         }
 858
 859         return ($self->{res},$self->{res_hdr});
 860 }
 861
 862 1;