source4/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 Parse::Pidl::Samba4 qw(choose_header is_intree);
  10 use Parse::Pidl::Util qw(has_property);
  11
  12 use vars qw($VERSION);
  13 $VERSION = '0.01';
  14
  15 use strict;
  16
  17 my($res,$res_hdr);
  18
  19 sub ParseFunctionSend($$$)
  20 {
  21         my ($interface, $fn, $name) = @_;
  22         my $uname = uc $name;
  23
  24         my $proto = "struct rpc_request *dcerpc_$name\_send(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct $name *r)";
  25
  26         $res_hdr .= "\n$proto;\n";
  27
  28         $res .= "$proto\n{\n";
  29
  30         if (has_property($fn, "todo")) {
  31                 $res .= "\treturn NULL;\n";
  32         } else {
  33                 $res .= "
  34         if (p->conn->flags & DCERPC_DEBUG_PRINT_IN) {
  35                 NDR_PRINT_IN_DEBUG($name, r);
  36         }
  37
  38         return dcerpc_ndr_request_send(p, NULL, &ndr_table_$interface->{NAME}, NDR_$uname, mem_ctx, r);
  39 ";
  40         }
  41
  42         $res .= "}\n\n";
  43 }
  44
  45 sub ParseFunctionSync($$$)
  46 {
  47         my ($interface, $fn, $name) = @_;
  48
  49         my $proto = "NTSTATUS dcerpc_$name(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct $name *r)";
  50
  51         $res_hdr .= "\n$proto;\n";
  52         $res .= "$proto\n{\n";
  53
  54         if (has_property($fn, "todo")) {
  55                 $res .= "\treturn NT_STATUS_NOT_IMPLEMENTED;\n";
  56         } else {
  57                 $res .= "
  58         struct rpc_request *req;
  59         NTSTATUS status;
  60
  61         req = dcerpc_$name\_send(p, mem_ctx, r);
  62         if (req == NULL) return NT_STATUS_NO_MEMORY;
  63
  64         status = dcerpc_ndr_request_recv(req);
  65
  66         if (NT_STATUS_IS_OK(status) && (p->conn->flags & DCERPC_DEBUG_PRINT_OUT)) {
  67                 NDR_PRINT_OUT_DEBUG($name, r);
  68         }
  69 ";
  70
  71         if (defined($fn->{RETURN_TYPE}) and $fn->{RETURN_TYPE} eq "NTSTATUS") {
  72              $res .= "\tif (NT_STATUS_IS_OK(status)) status = r->out.result;\n";
  73         }
  74         $res .=
  75 "
  76         return status;
  77 ";
  78         }
  79
  80         $res .= "}\n\n";
  81 }
  82
  83 #####################################################################
  84 # parse a function
  85 sub ParseFunction($$)
  86 {
  87         my ($interface, $fn) = @_;
  88
  89         ParseFunctionSend($interface, $fn, $fn->{NAME});
  90         ParseFunctionSync($interface, $fn, $fn->{NAME});
  91 }
  92
  93 my %done;
  94
  95 #####################################################################
  96 # parse the interface definitions
  97 sub ParseInterface($)
  98 {
  99         my($interface) = shift;
 100
 101         $res_hdr .= "#ifndef _HEADER_RPC_$interface->{NAME}\n";
 102         $res_hdr .= "#define _HEADER_RPC_$interface->{NAME}\n\n";
 103
 104         if (defined $interface->{PROPERTIES}->{uuid}) {
 105                 $res_hdr .= "extern const struct ndr_interface_table ndr_table_$interface->{NAME};\n";
 106         }
 107
 108         $res .= "/* $interface->{NAME} - client functions generated by pidl */\n\n";
 109
 110         foreach my $fn (@{$interface->{FUNCTIONS}}) {
 111                 next if not defined($fn->{OPNUM});
 112                 next if defined($done{$fn->{NAME}});
 113                 ParseFunction($interface, $fn);
 114                 $done{$fn->{NAME}} = 1;
 115         }
 116
 117         $res_hdr .= "#endif /* _HEADER_RPC_$interface->{NAME} */\n";
 118
 119         return $res;
 120 }
 121
 122 sub Parse($$$$)
 123 {
 124         my($ndr,$header,$ndr_header,$client_header) = @_;
 125
 126         $res = "";
 127         $res_hdr = "";
 128
 129         $res .= "/* client functions auto-generated by pidl */\n";
 130         $res .= "\n";
 131         if (is_intree()) {
 132                 $res .= "#include \"includes.h\"\n";
 133         } else {
 134                 $res .= "#define _GNU_SOURCE\n";
 135                 $res .= "#include <stdio.h>\n";
 136                 $res .= "#include <stdbool.h>\n";
 137                 $res .= "#include <stdlib.h>\n";
 138                 $res .= "#include <stdint.h>\n";
 139                 $res .= "#include <stdarg.h>\n";
 140                 $res .= "#include <core/ntstatus.h>\n";
 141         }
 142         $res .= "#include \"$ndr_header\"\n";
 143         $res .= "#include \"$client_header\"\n";
 144         $res .= "\n";
 145
 146         $res_hdr .= choose_header("librpc/rpc/dcerpc.h", "dcerpc.h")."\n";
 147         $res_hdr .= "#include \"$header\"\n";
 148
 149         foreach my $x (@{$ndr}) {
 150                 ($x->{TYPE} eq "INTERFACE") && ParseInterface($x);
 151         }
 152
 153         return ($res,$res_hdr);
 154 }
 155
 156 1;