source: branches/samba-3.5.x/pidl/tests/samba3-cli.pl@ 770

Last change on this file since 770 was 414, checked in by Herwig Bauernfeind, 16 years ago

Samba 3.5.0: Initial import
File size: 7.4 KB
Line 
1#!/usr/bin/perl
2# (C) 2007 Jelmer Vernooij <jelmer@samba.org>
3# Published under the GNU General Public License
4use strict;
5use warnings;
6
7use Test::More tests => 9;
8use FindBin qw($RealBin);
9use lib "$RealBin";
10use Util;
11use Parse::Pidl::Util qw(MyDumper);
12use Parse::Pidl::Samba3::ClientNDR qw(ParseFunction ParseOutputArgument);
13use Parse::Pidl::Samba4::Header qw(GenerateFunctionInEnv GenerateFunctionOutEnv);
14
15# Make sure GenerateFunctionInEnv and GenerateFunctionOutEnv work
16my $fn = { ELEMENTS => [ { DIRECTION => ["in"], NAME => "foo" } ] };
17is_deeply({ "foo" => "r.in.foo" }, GenerateFunctionInEnv($fn, "r."));
18is_deeply({ "foo" => "r.in.foo" }, GenerateFunctionOutEnv($fn, "r."));
19
20$fn = { ELEMENTS => [ { DIRECTION => ["out", "in"], NAME => "foo" } ] };
21is_deeply({ "foo" => "r.in.foo" }, GenerateFunctionInEnv($fn, "r."));
22is_deeply({ "foo" => "r.out.foo" }, GenerateFunctionOutEnv($fn, "r."));
23
24$fn = { ELEMENTS => [ { DIRECTION => ["out"], NAME => "foo" } ] };
25is_deeply({ }, GenerateFunctionInEnv($fn, "r."));
26is_deeply({ "foo" => "r.out.foo" }, GenerateFunctionOutEnv($fn, "r."));
27
28my $x = new Parse::Pidl::Samba3::ClientNDR();
29
30$fn = { NAME => "bar", ELEMENTS => [ ] };
31$x->ParseFunction("foo", $fn);
32is($x->{res},
33"struct rpccli_bar_state {
34 struct bar orig;
35 struct bar tmp;
36 TALLOC_CTX *out_mem_ctx;
37 NTSTATUS (*dispatch_recv)(struct tevent_req *req, TALLOC_CTX *mem_ctx);
38};
39
40static void rpccli_bar_done(struct tevent_req *subreq);
41
42struct tevent_req *rpccli_bar_send(TALLOC_CTX *mem_ctx,
43 struct tevent_context *ev,
44 struct rpc_pipe_client *cli)
45{
46 struct tevent_req *req;
47 struct rpccli_bar_state *state;
48 struct tevent_req *subreq;
49
50 req = tevent_req_create(mem_ctx, &state,
51 struct rpccli_bar_state);
52 if (req == NULL) {
53 return NULL;
54 }
55 state->out_mem_ctx = NULL;
56 state->dispatch_recv = cli->dispatch_recv;
57
58 /* In parameters */
59
60 /* Out parameters */
61
62 if (DEBUGLEVEL >= 10) {
63 NDR_PRINT_IN_DEBUG(bar, &state->orig);
64 }
65
66 /* make a temporary copy, that we pass to the dispatch function */
67 state->tmp = state->orig;
68
69 subreq = cli->dispatch_send(state, ev, cli,
70 &ndr_table_foo,
71 NDR_BAR,
72 &state->tmp);
73 if (tevent_req_nomem(subreq, req)) {
74 return tevent_req_post(req, ev);
75 }
76 tevent_req_set_callback(subreq, rpccli_bar_done, req);
77 return req;
78}
79
80static void rpccli_bar_done(struct tevent_req *subreq)
81{
82 struct tevent_req *req = tevent_req_callback_data(
83 subreq, struct tevent_req);
84 struct rpccli_bar_state *state = tevent_req_data(
85 req, struct rpccli_bar_state);
86 NTSTATUS status;
87 TALLOC_CTX *mem_ctx;
88
89 if (state->out_mem_ctx) {
90 mem_ctx = state->out_mem_ctx;
91 } else {
92 mem_ctx = state;
93 }
94
95 status = state->dispatch_recv(subreq, mem_ctx);
96 TALLOC_FREE(subreq);
97 if (!NT_STATUS_IS_OK(status)) {
98 tevent_req_nterror(req, status);
99 return;
100 }
101
102 /* Copy out parameters */
103
104 /* Reset temporary structure */
105 ZERO_STRUCT(state->tmp);
106
107 if (DEBUGLEVEL >= 10) {
108 NDR_PRINT_OUT_DEBUG(bar, &state->orig);
109 }
110
111 tevent_req_done(req);
112}
113
114NTSTATUS rpccli_bar_recv(struct tevent_req *req,
115 TALLOC_CTX *mem_ctx)
116{
117 struct rpccli_bar_state *state = tevent_req_data(
118 req, struct rpccli_bar_state);
119 NTSTATUS status;
120
121 if (tevent_req_is_nterror(req, &status)) {
122 tevent_req_received(req);
123 return status;
124 }
125
126 /* Steal possbile out parameters to the callers context */
127 talloc_steal(mem_ctx, state->out_mem_ctx);
128
129 tevent_req_received(req);
130 return NT_STATUS_OK;
131}
132
133NTSTATUS rpccli_bar(struct rpc_pipe_client *cli,
134 TALLOC_CTX *mem_ctx)
135{
136\tstruct bar r;
137\tNTSTATUS status;
138
139\t/* In parameters */
140
141\tif (DEBUGLEVEL >= 10) {
142\t\tNDR_PRINT_IN_DEBUG(bar, &r);
143\t}
144
145 status = cli->dispatch(cli,
146 mem_ctx,
147 &ndr_table_foo,
148 NDR_BAR,
149 &r);
150
151\tif (!NT_STATUS_IS_OK(status)) {
152\t\treturn status;
153\t}
154
155\tif (DEBUGLEVEL >= 10) {
156\t\tNDR_PRINT_OUT_DEBUG(bar, &r);
157\t}
158
159\tif (NT_STATUS_IS_ERR(status)) {
160\t\treturn status;
161\t}
162
163\t/* Return variables */
164
165\t/* Return result */
166\treturn NT_STATUS_OK;
167}
168
169");
170
171$x = new Parse::Pidl::Samba3::ClientNDR();
172
173$fn = { NAME => "bar", ELEMENTS => [ ], RETURN_TYPE => "WERROR" };
174$x->ParseFunction("foo", $fn);
175is($x->{res},
176"struct rpccli_bar_state {
177 struct bar orig;
178 struct bar tmp;
179 TALLOC_CTX *out_mem_ctx;
180 NTSTATUS (*dispatch_recv)(struct tevent_req *req, TALLOC_CTX *mem_ctx);
181};
182
183static void rpccli_bar_done(struct tevent_req *subreq);
184
185struct tevent_req *rpccli_bar_send(TALLOC_CTX *mem_ctx,
186 struct tevent_context *ev,
187 struct rpc_pipe_client *cli)
188{
189 struct tevent_req *req;
190 struct rpccli_bar_state *state;
191 struct tevent_req *subreq;
192
193 req = tevent_req_create(mem_ctx, &state,
194 struct rpccli_bar_state);
195 if (req == NULL) {
196 return NULL;
197 }
198 state->out_mem_ctx = NULL;
199 state->dispatch_recv = cli->dispatch_recv;
200
201 /* In parameters */
202
203 /* Out parameters */
204
205 /* Result */
206 ZERO_STRUCT(state->orig.out.result);
207
208 if (DEBUGLEVEL >= 10) {
209 NDR_PRINT_IN_DEBUG(bar, &state->orig);
210 }
211
212 /* make a temporary copy, that we pass to the dispatch function */
213 state->tmp = state->orig;
214
215 subreq = cli->dispatch_send(state, ev, cli,
216 &ndr_table_foo,
217 NDR_BAR,
218 &state->tmp);
219 if (tevent_req_nomem(subreq, req)) {
220 return tevent_req_post(req, ev);
221 }
222 tevent_req_set_callback(subreq, rpccli_bar_done, req);
223 return req;
224}
225
226static void rpccli_bar_done(struct tevent_req *subreq)
227{
228 struct tevent_req *req = tevent_req_callback_data(
229 subreq, struct tevent_req);
230 struct rpccli_bar_state *state = tevent_req_data(
231 req, struct rpccli_bar_state);
232 NTSTATUS status;
233 TALLOC_CTX *mem_ctx;
234
235 if (state->out_mem_ctx) {
236 mem_ctx = state->out_mem_ctx;
237 } else {
238 mem_ctx = state;
239 }
240
241 status = state->dispatch_recv(subreq, mem_ctx);
242 TALLOC_FREE(subreq);
243 if (!NT_STATUS_IS_OK(status)) {
244 tevent_req_nterror(req, status);
245 return;
246 }
247
248 /* Copy out parameters */
249
250 /* Copy result */
251 state->orig.out.result = state->tmp.out.result;
252
253 /* Reset temporary structure */
254 ZERO_STRUCT(state->tmp);
255
256 if (DEBUGLEVEL >= 10) {
257 NDR_PRINT_OUT_DEBUG(bar, &state->orig);
258 }
259
260 tevent_req_done(req);
261}
262
263NTSTATUS rpccli_bar_recv(struct tevent_req *req,
264 TALLOC_CTX *mem_ctx,
265 WERROR *result)
266{
267 struct rpccli_bar_state *state = tevent_req_data(
268 req, struct rpccli_bar_state);
269 NTSTATUS status;
270
271 if (tevent_req_is_nterror(req, &status)) {
272 tevent_req_received(req);
273 return status;
274 }
275
276 /* Steal possbile out parameters to the callers context */
277 talloc_steal(mem_ctx, state->out_mem_ctx);
278
279 /* Return result */
280 *result = state->orig.out.result;
281
282 tevent_req_received(req);
283 return NT_STATUS_OK;
284}
285
286NTSTATUS rpccli_bar(struct rpc_pipe_client *cli,
287 TALLOC_CTX *mem_ctx,
288 WERROR *werror)
289{
290\tstruct bar r;
291\tNTSTATUS status;
292
293\t/* In parameters */
294
295\tif (DEBUGLEVEL >= 10) {
296\t\tNDR_PRINT_IN_DEBUG(bar, &r);
297\t}
298
299 status = cli->dispatch(cli,
300 mem_ctx,
301 &ndr_table_foo,
302 NDR_BAR,
303 &r);
304
305\tif (!NT_STATUS_IS_OK(status)) {
306\t\treturn status;
307\t}
308
309\tif (DEBUGLEVEL >= 10) {
310\t\tNDR_PRINT_OUT_DEBUG(bar, &r);
311\t}
312
313\tif (NT_STATUS_IS_ERR(status)) {
314\t\treturn status;
315\t}
316
317\t/* Return variables */
318
319\t/* Return result */
320\tif (werror) {
321\t\t*werror = r.out.result;
322\t}
323
324\treturn werror_to_ntstatus(r.out.result);
325}
326
327");
328
329$x = new Parse::Pidl::Samba3::ClientNDR();
330
331$fn = { NAME => "bar", ELEMENTS => [ ], RETURN_TYPE => "WERROR" };
332my $e = { NAME => "foo", ORIGINAL => { FILE => "f", LINE => -1 },
333 LEVELS => [ { TYPE => "ARRAY", SIZE_IS => "mysize" }, { TYPE => "DATA", DATA_TYPE => "int" } ]};
334
335$x->ParseOutputArgument($fn, $e);
336is($x->{res}, "memcpy(foo, r.out.foo, (mysize) * sizeof(*foo));\n");
Note: See TracBrowser for help on using the repository browser.