1 ###################################################
2 # DCOM parser for Samba
3 # Basically the glue between COM and DCE/RPC with NDR
4 # Copyright jelmer@samba.org 2003-2005
5 # released under the GNU GPL
7 package Parse::Pidl::Samba4::COM::Proxy;
9 use Parse::Pidl::Samba4::COM::Header;
10 use Parse::Pidl::Typelist qw(mapTypeName);
11 use Parse::Pidl::Util qw(has_property);
13 use vars qw($VERSION);
22 my ($interface, $name) = @_;
25 $res .="\tstruct $interface->{NAME}_vtable $name = {";
27 if (defined($interface->{BASE})) {
31 my $data = $interface->{DATA};
33 foreach my $d (@{$data}) {
34 if ($d->{TYPE} eq "FUNCTION") {
35 $res .= "\n\t\tdcom_proxy_$interface->{NAME}_$d->{NAME}";
45 my $interface = shift;
47 $res .= "static NTSTATUS dcom_proxy_$interface->{NAME}_init(void)
49 struct $interface->{NAME}_vtable *proxy_vtable = talloc(talloc_autofree_context(), struct $interface->{NAME}_vtable);
52 if (defined($interface->{BASE})) {
55 const void *base_vtable;
57 base_iid = ndr_table_$interface->{BASE}.syntax_id.uuid;
59 base_vtable = dcom_proxy_vtable_by_iid(&base_iid);
60 if (base_vtable == NULL) {
61 DEBUG(0, (\"No proxy registered for base interface '$interface->{BASE}'\\n\"));
62 return NT_STATUS_FOOBAR;
65 memcpy(&proxy_vtable, base_vtable, sizeof(struct $interface->{BASE}_vtable));
69 foreach my $x (@{$interface->{DATA}}) {
70 next unless ($x->{TYPE} eq "FUNCTION");
72 $res .= "\tproxy_vtable->$x->{NAME} = dcom_proxy_$interface->{NAME}_$x->{NAME};\n";
76 proxy_vtable->iid = ndr_table_$interface->{NAME}.syntax_id.uuid;
78 return dcom_register_proxy((struct IUnknown_vtable *)proxy_vtable);
82 #####################################################################
86 my ($interface, $fn) = @_;
87 my $name = $fn->{NAME};
90 my $tn = mapTypeName($fn->{RETURN_TYPE});
93 static $tn dcom_proxy_$interface->{NAME}_$name(struct $interface->{NAME} *d, TALLOC_CTX *mem_ctx" . Parse::Pidl::Samba4::COM::Header::GetArgumentProtoList($fn) . ")
95 struct dcerpc_pipe *p;
96 NTSTATUS status = dcom_get_pipe(d, &p);
98 struct rpc_request *req;
100 if (NT_STATUS_IS_ERR(status)) {
104 ZERO_STRUCT(r.in.ORPCthis);
105 r.in.ORPCthis.version.MajorVersion = COM_MAJOR_VERSION;
106 r.in.ORPCthis.version.MinorVersion = COM_MINOR_VERSION;
109 # Put arguments into r
110 foreach my $a (@{$fn->{ELEMENTS}}) {
111 next unless (has_property($a, "in"));
112 if (Parse::Pidl::Typelist::typeIs($a->{TYPE}, "INTERFACE")) {
113 $res .="\tNDR_CHECK(dcom_OBJREF_from_IUnknown(mem_ctx, &r.in.$a->{NAME}.obj, $a->{NAME}));\n";
115 $res .= "\tr.in.$a->{NAME} = $a->{NAME};\n";
120 if (p->conn->flags & DCERPC_DEBUG_PRINT_IN) {
121 NDR_PRINT_IN_DEBUG($name, &r);
124 status = dcerpc_ndr_request(p, &d->ipid, &ndr_table_$interface->{NAME}, NDR_$uname, mem_ctx, &r);
126 if (NT_STATUS_IS_OK(status) && (p->conn->flags & DCERPC_DEBUG_PRINT_OUT)) {
127 NDR_PRINT_OUT_DEBUG($name, r);
132 # Put r info back into arguments
133 foreach my $a (@{$fn->{ELEMENTS}}) {
134 next unless (has_property($a, "out"));
136 if (Parse::Pidl::Typelist::typeIs($a->{TYPE}, "INTERFACE")) {
137 $res .="\tNDR_CHECK(dcom_IUnknown_from_OBJREF(d->ctx, &$a->{NAME}, r.out.$a->{NAME}.obj));\n";
139 $res .= "\t*$a->{NAME} = r.out.$a->{NAME};\n";
144 if ($fn->{RETURN_TYPE} eq "NTSTATUS") {
145 $res .= "\tif (NT_STATUS_IS_OK(status)) status = r.out.result;\n";
154 #####################################################################
155 # parse the interface definitions
156 sub ParseInterface($)
158 my($interface) = shift;
159 my($data) = $interface->{DATA};
160 $res = "/* DCOM proxy for $interface->{NAME} generated by pidl */\n\n";
161 foreach my $d (@{$data}) {
162 ($d->{TYPE} eq "FUNCTION") &&
163 ParseFunction($interface, $d);
166 ParseRegFunc($interface);
169 sub RegistrationFunction($$)
172 my $basename = shift;
174 my $res = "\n\nNTSTATUS dcom_$basename\_init(void)\n";
176 $res .="\tNTSTATUS status = NT_STATUS_OK;\n";
177 foreach my $interface (@{$idl}) {
178 next if $interface->{TYPE} ne "INTERFACE";
179 next if not has_property($interface, "object");
181 my $data = $interface->{DATA};
183 foreach my $d (@{$data}) {
184 if ($d->{TYPE} eq "FUNCTION") { $count++; }
187 next if ($count == 0);
189 $res .= "\tstatus = dcom_$interface->{NAME}_init();\n";
190 $res .= "\tif (NT_STATUS_IS_ERR(status)) {\n";
191 $res .= "\t\treturn status;\n";
194 $res .= "\treturn status;\n";
202 my ($pidl,$comh_filename) = @_;
205 $res .= "#include \"includes.h\"\n" .
206 "#include \"lib/com/dcom/dcom.h\"\n" .
207 "#include \"$comh_filename\"\n" .
208 "#include \"librpc/rpc/dcerpc.h\"\n";
211 next if ($_->{TYPE} ne "INTERFACE");
212 next if has_property($_, "local");
213 next unless has_property($_, "object");
215 $res .= ParseInterface($_);