s4-smbtorture: add smbcli_rap_netprintqueueresume() and smbcli_rap_netprintqueuepurge().
[Samba/ekacnet.git] / source4 / torture / rap / rap.c
blobd4899b1423729a213e3a11788cecdf06629b753b
1 /*
2 Unix SMB/CIFS implementation.
3 test suite for various RAP operations
4 Copyright (C) Volker Lendecke 2004
5 Copyright (C) Tim Potter 2005
6 Copyright (C) Jelmer Vernooij 2007
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "includes.h"
23 #include "libcli/libcli.h"
24 #include "torture/smbtorture.h"
25 #include "torture/util.h"
26 #include "../librpc/gen_ndr/ndr_rap.h"
27 #include "librpc/ndr/libndr.h"
28 #include "param/param.h"
29 #include "torture/rap/proto.h"
31 #define RAP_GOTO(call) do { \
32 NTSTATUS _status; \
33 _status = call; \
34 if (!NT_STATUS_IS_OK(_status)) { \
35 result = _status; \
36 goto done; \
37 } \
38 } while (0)
40 #define RAP_RETURN(call) do { \
41 NTSTATUS _status; \
42 _status = call; \
43 if (!NT_STATUS_IS_OK(_status)) { \
44 return _status; \
45 } \
46 } while (0)
49 #define NDR_GOTO(call) do { \
50 enum ndr_err_code _ndr_err; \
51 _ndr_err = call; \
52 if (!NDR_ERR_CODE_IS_SUCCESS(_ndr_err)) { \
53 result = ndr_map_error2ntstatus(_ndr_err); \
54 goto done; \
55 } \
56 } while (0)
58 #define NDR_RETURN(call) do { \
59 enum ndr_err_code _ndr_err; \
60 _ndr_err = call; \
61 if (!NDR_ERR_CODE_IS_SUCCESS(_ndr_err)) { \
62 return ndr_map_error2ntstatus(_ndr_err); \
63 } \
64 } while (0)
66 struct rap_call {
67 uint16_t callno;
68 char *paramdesc;
69 const char *datadesc;
71 uint16_t status;
72 uint16_t convert;
74 uint16_t rcv_paramlen, rcv_datalen;
76 struct ndr_push *ndr_push_param;
77 struct ndr_push *ndr_push_data;
78 struct ndr_pull *ndr_pull_param;
79 struct ndr_pull *ndr_pull_data;
82 #define RAPNDR_FLAGS (LIBNDR_FLAG_NOALIGN|LIBNDR_FLAG_STR_ASCII|LIBNDR_FLAG_STR_NULLTERM);
84 static struct rap_call *new_rap_cli_call(TALLOC_CTX *mem_ctx, struct smb_iconv_convenience *iconv_convenience, uint16_t callno)
86 struct rap_call *call;
88 call = talloc(mem_ctx, struct rap_call);
90 if (call == NULL)
91 return NULL;
93 call->callno = callno;
94 call->rcv_paramlen = 4;
96 call->paramdesc = NULL;
97 call->datadesc = NULL;
99 call->ndr_push_param = ndr_push_init_ctx(mem_ctx, iconv_convenience);
100 call->ndr_push_param->flags = RAPNDR_FLAGS;
102 call->ndr_push_data = ndr_push_init_ctx(mem_ctx, iconv_convenience);
103 call->ndr_push_data->flags = RAPNDR_FLAGS;
105 return call;
108 static void rap_cli_push_paramdesc(struct rap_call *call, char desc)
110 int len = 0;
112 if (call->paramdesc != NULL)
113 len = strlen(call->paramdesc);
115 call->paramdesc = talloc_realloc(call,
116 call->paramdesc,
117 char,
118 len+2);
120 call->paramdesc[len] = desc;
121 call->paramdesc[len+1] = '\0';
124 static void rap_cli_push_word(struct rap_call *call, uint16_t val)
126 rap_cli_push_paramdesc(call, 'W');
127 ndr_push_uint16(call->ndr_push_param, NDR_SCALARS, val);
130 static void rap_cli_push_dword(struct rap_call *call, uint32_t val)
132 rap_cli_push_paramdesc(call, 'D');
133 ndr_push_uint32(call->ndr_push_param, NDR_SCALARS, val);
136 static void rap_cli_push_rcvbuf(struct rap_call *call, int len)
138 rap_cli_push_paramdesc(call, 'r');
139 rap_cli_push_paramdesc(call, 'L');
140 ndr_push_uint16(call->ndr_push_param, NDR_SCALARS, len);
141 call->rcv_datalen = len;
144 static void rap_cli_expect_multiple_entries(struct rap_call *call)
146 rap_cli_push_paramdesc(call, 'e');
147 rap_cli_push_paramdesc(call, 'h');
148 call->rcv_paramlen += 4; /* uint16_t entry count, uint16_t total */
151 static void rap_cli_expect_word(struct rap_call *call)
153 rap_cli_push_paramdesc(call, 'h');
154 call->rcv_paramlen += 2;
157 static void rap_cli_push_string(struct rap_call *call, const char *str)
159 if (str == NULL) {
160 rap_cli_push_paramdesc(call, 'O');
161 return;
163 rap_cli_push_paramdesc(call, 'z');
164 ndr_push_string(call->ndr_push_param, NDR_SCALARS, str);
167 static void rap_cli_expect_format(struct rap_call *call, const char *format)
169 call->datadesc = format;
172 static NTSTATUS rap_pull_string(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr,
173 uint16_t convert, const char **dest)
175 uint16_t string_offset;
176 uint16_t ignore;
177 const char *p;
178 size_t len;
180 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &string_offset));
181 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &ignore));
183 string_offset -= convert;
185 if (string_offset+1 > ndr->data_size)
186 return NT_STATUS_INVALID_PARAMETER;
188 p = (const char *)(ndr->data + string_offset);
189 len = strnlen(p, ndr->data_size-string_offset);
191 if ( string_offset + len + 1 > ndr->data_size )
192 return NT_STATUS_INVALID_PARAMETER;
194 *dest = talloc_zero_array(mem_ctx, char, len+1);
195 pull_string((char *)*dest, p, len+1, len, STR_ASCII);
197 return NT_STATUS_OK;
200 static NTSTATUS rap_cli_do_call(struct smbcli_tree *tree,
201 struct smb_iconv_convenience *iconv_convenience,
202 struct rap_call *call)
204 NTSTATUS result;
205 DATA_BLOB param_blob;
206 struct ndr_push *params;
207 struct smb_trans2 trans;
209 params = ndr_push_init_ctx(call, iconv_convenience);
211 if (params == NULL)
212 return NT_STATUS_NO_MEMORY;
214 params->flags = RAPNDR_FLAGS;
216 trans.in.max_param = call->rcv_paramlen;
217 trans.in.max_data = call->rcv_datalen;
218 trans.in.max_setup = 0;
219 trans.in.flags = 0;
220 trans.in.timeout = 0;
221 trans.in.setup_count = 0;
222 trans.in.setup = NULL;
223 trans.in.trans_name = "\\PIPE\\LANMAN";
225 NDR_RETURN(ndr_push_uint16(params, NDR_SCALARS, call->callno));
226 if (call->paramdesc)
227 NDR_RETURN(ndr_push_string(params, NDR_SCALARS, call->paramdesc));
228 if (call->datadesc)
229 NDR_RETURN(ndr_push_string(params, NDR_SCALARS, call->datadesc));
231 param_blob = ndr_push_blob(call->ndr_push_param);
232 NDR_RETURN(ndr_push_bytes(params, param_blob.data,
233 param_blob.length));
235 trans.in.params = ndr_push_blob(params);
236 trans.in.data = data_blob(NULL, 0);
238 result = smb_raw_trans(tree, call, &trans);
240 if (!NT_STATUS_IS_OK(result))
241 return result;
243 call->ndr_pull_param = ndr_pull_init_blob(&trans.out.params, call,
244 iconv_convenience);
245 call->ndr_pull_param->flags = RAPNDR_FLAGS;
247 call->ndr_pull_data = ndr_pull_init_blob(&trans.out.data, call,
248 iconv_convenience);
249 call->ndr_pull_data->flags = RAPNDR_FLAGS;
251 return result;
255 static NTSTATUS smbcli_rap_netshareenum(struct smbcli_tree *tree,
256 struct smb_iconv_convenience *iconv_convenience,
257 TALLOC_CTX *mem_ctx,
258 struct rap_NetShareEnum *r)
260 struct rap_call *call;
261 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
262 int i;
264 call = new_rap_cli_call(tree, iconv_convenience, RAP_WshareEnum);
266 if (call == NULL)
267 return NT_STATUS_NO_MEMORY;
269 rap_cli_push_word(call, r->in.level); /* Level */
270 rap_cli_push_rcvbuf(call, r->in.bufsize);
271 rap_cli_expect_multiple_entries(call);
273 switch(r->in.level) {
274 case 0:
275 rap_cli_expect_format(call, "B13");
276 break;
277 case 1:
278 rap_cli_expect_format(call, "B13BWz");
279 break;
282 if (DEBUGLEVEL >= 10) {
283 NDR_PRINT_IN_DEBUG(rap_NetShareEnum, r);
286 result = rap_cli_do_call(tree, iconv_convenience, call);
288 if (!NT_STATUS_IS_OK(result))
289 goto done;
291 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
292 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
293 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
294 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
296 r->out.info = talloc_array(mem_ctx, union rap_share_info, r->out.count);
298 if (r->out.info == NULL) {
299 result = NT_STATUS_NO_MEMORY;
300 goto done;
303 for (i=0; i<r->out.count; i++) {
304 switch(r->in.level) {
305 case 0:
306 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
307 r->out.info[i].info0.share_name, 13));
308 break;
309 case 1:
310 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
311 r->out.info[i].info1.share_name, 13));
312 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
313 &r->out.info[i].info1.reserved1, 1));
314 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_data,
315 NDR_SCALARS, &r->out.info[i].info1.share_type));
316 RAP_GOTO(rap_pull_string(mem_ctx, call->ndr_pull_data,
317 r->out.convert,
318 &r->out.info[i].info1.comment));
319 break;
323 if (DEBUGLEVEL >= 10) {
324 NDR_PRINT_OUT_DEBUG(rap_NetShareEnum, r);
326 result = NT_STATUS_OK;
328 done:
329 talloc_free(call);
330 return result;
333 static bool test_netshareenum(struct torture_context *tctx,
334 struct smbcli_state *cli)
336 struct rap_NetShareEnum r;
337 int i;
339 r.in.level = 1;
340 r.in.bufsize = 8192;
342 torture_assert_ntstatus_ok(tctx,
343 smbcli_rap_netshareenum(cli->tree, lp_iconv_convenience(tctx->lp_ctx), tctx, &r), "");
345 for (i=0; i<r.out.count; i++) {
346 printf("%s %d %s\n", r.out.info[i].info1.share_name,
347 r.out.info[i].info1.share_type,
348 r.out.info[i].info1.comment);
351 return true;
354 static NTSTATUS smbcli_rap_netserverenum2(struct smbcli_tree *tree,
355 struct smb_iconv_convenience *iconv_convenience,
356 TALLOC_CTX *mem_ctx,
357 struct rap_NetServerEnum2 *r)
359 struct rap_call *call;
360 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
361 int i;
363 call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_NetServerEnum2);
365 if (call == NULL)
366 return NT_STATUS_NO_MEMORY;
368 rap_cli_push_word(call, r->in.level);
369 rap_cli_push_rcvbuf(call, r->in.bufsize);
370 rap_cli_expect_multiple_entries(call);
371 rap_cli_push_dword(call, r->in.servertype);
372 rap_cli_push_string(call, r->in.domain);
374 switch(r->in.level) {
375 case 0:
376 rap_cli_expect_format(call, "B16");
377 break;
378 case 1:
379 rap_cli_expect_format(call, "B16BBDz");
380 break;
383 if (DEBUGLEVEL >= 10) {
384 NDR_PRINT_IN_DEBUG(rap_NetServerEnum2, r);
387 result = rap_cli_do_call(tree, iconv_convenience, call);
389 if (!NT_STATUS_IS_OK(result))
390 goto done;
392 result = NT_STATUS_INVALID_PARAMETER;
394 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
395 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
396 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
397 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
399 r->out.info = talloc_array(mem_ctx, union rap_server_info, r->out.count);
401 if (r->out.info == NULL) {
402 result = NT_STATUS_NO_MEMORY;
403 goto done;
406 for (i=0; i<r->out.count; i++) {
407 switch(r->in.level) {
408 case 0:
409 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
410 r->out.info[i].info0.name, 16));
411 break;
412 case 1:
413 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
414 r->out.info[i].info1.name, 16));
415 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
416 &r->out.info[i].info1.version_major, 1));
417 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
418 &r->out.info[i].info1.version_minor, 1));
419 NDR_GOTO(ndr_pull_uint32(call->ndr_pull_data,
420 NDR_SCALARS, &r->out.info[i].info1.servertype));
421 RAP_GOTO(rap_pull_string(mem_ctx, call->ndr_pull_data,
422 r->out.convert,
423 &r->out.info[i].info1.comment));
427 if (DEBUGLEVEL >= 10) {
428 NDR_PRINT_OUT_DEBUG(rap_NetServerEnum2, r);
431 result = NT_STATUS_OK;
433 done:
434 talloc_free(call);
435 return result;
438 static bool test_netserverenum(struct torture_context *tctx,
439 struct smbcli_state *cli)
441 struct rap_NetServerEnum2 r;
442 int i;
444 r.in.level = 0;
445 r.in.bufsize = 8192;
446 r.in.servertype = 0xffffffff;
447 r.in.servertype = 0x80000000;
448 r.in.domain = NULL;
450 torture_assert_ntstatus_ok(tctx,
451 smbcli_rap_netserverenum2(cli->tree, lp_iconv_convenience(tctx->lp_ctx), tctx, &r), "");
453 for (i=0; i<r.out.count; i++) {
454 switch (r.in.level) {
455 case 0:
456 printf("%s\n", r.out.info[i].info0.name);
457 break;
458 case 1:
459 printf("%s %x %s\n", r.out.info[i].info1.name,
460 r.out.info[i].info1.servertype,
461 r.out.info[i].info1.comment);
462 break;
466 return true;
469 NTSTATUS smbcli_rap_netservergetinfo(struct smbcli_tree *tree,
470 struct smb_iconv_convenience *iconv_convenience,
471 TALLOC_CTX *mem_ctx,
472 struct rap_WserverGetInfo *r)
474 struct rap_call *call;
475 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
477 if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WserverGetInfo))) {
478 return NT_STATUS_NO_MEMORY;
481 rap_cli_push_word(call, r->in.level);
482 rap_cli_push_rcvbuf(call, r->in.bufsize);
483 rap_cli_expect_word(call);
485 switch(r->in.level) {
486 case 0:
487 rap_cli_expect_format(call, "B16");
488 break;
489 case 1:
490 rap_cli_expect_format(call, "B16BBDz");
491 break;
492 default:
493 result = NT_STATUS_INVALID_PARAMETER;
494 goto done;
497 if (DEBUGLEVEL >= 10) {
498 NDR_PRINT_IN_DEBUG(rap_WserverGetInfo, r);
501 result = rap_cli_do_call(tree, iconv_convenience, call);
503 if (!NT_STATUS_IS_OK(result))
504 goto done;
506 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
507 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
508 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
510 switch(r->in.level) {
511 case 0:
512 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
513 r->out.info.info0.name, 16));
514 break;
515 case 1:
516 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
517 r->out.info.info1.name, 16));
518 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
519 &r->out.info.info1.version_major, 1));
520 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
521 &r->out.info.info1.version_minor, 1));
522 NDR_GOTO(ndr_pull_uint32(call->ndr_pull_data,
523 NDR_SCALARS, &r->out.info.info1.servertype));
524 RAP_GOTO(rap_pull_string(mem_ctx, call->ndr_pull_data,
525 r->out.convert,
526 &r->out.info.info1.comment));
529 if (DEBUGLEVEL >= 10) {
530 NDR_PRINT_OUT_DEBUG(rap_WserverGetInfo, r);
532 done:
533 talloc_free(call);
534 return result;
537 static NTSTATUS rap_pull_rap_JobInfo1(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintJobInfo1 *r)
539 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->JobID));
540 NDR_RETURN(ndr_pull_charset(ndr, NDR_SCALARS, &r->UserName, 21, sizeof(uint8_t), CH_DOS));
541 NDR_RETURN(ndr_pull_uint8(ndr, NDR_SCALARS, &r->Pad));
542 NDR_RETURN(ndr_pull_charset(ndr, NDR_SCALARS, &r->NotifyName, 16, sizeof(uint8_t), CH_DOS));
543 NDR_RETURN(ndr_pull_charset(ndr, NDR_SCALARS, &r->DataType, 10, sizeof(uint8_t), CH_DOS));
544 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintParameterString));
545 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->JobPosition));
546 NDR_RETURN(ndr_pull_rap_PrintJStatusCode(ndr, NDR_SCALARS, &r->JobStatus));
547 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->JobStatusString));
548 NDR_RETURN(ndr_pull_time_t(ndr, NDR_SCALARS, &r->TimeSubmitted));
549 NDR_RETURN(ndr_pull_uint32(ndr, NDR_SCALARS, &r->JobSize));
550 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->JobCommentString));
552 return NT_STATUS_OK;
555 static NTSTATUS rap_pull_rap_JobInfo2(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintJobInfo2 *r)
557 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->JobID));
558 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->Priority));
559 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->UserName));
560 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->JobPosition));
561 NDR_RETURN(ndr_pull_rap_PrintJStatusCode(ndr, NDR_SCALARS, &r->JobStatus));
562 NDR_RETURN(ndr_pull_time_t(ndr, NDR_SCALARS, &r->TimeSubmitted));
563 NDR_RETURN(ndr_pull_uint32(ndr, NDR_SCALARS, &r->JobSize));
564 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->JobCommentString));
565 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->DocumentName));
567 return NT_STATUS_OK;
570 static NTSTATUS rap_pull_rap_PrintQueue0(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue0 *r)
572 NDR_RETURN(ndr_pull_charset(ndr, NDR_SCALARS, &r->PrintQName, 13, sizeof(uint8_t), CH_DOS));
574 return NT_STATUS_OK;
577 static NTSTATUS rap_pull_rap_PrintQueue1(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue1 *r)
579 NDR_RETURN(ndr_pull_charset(ndr, NDR_SCALARS, &r->PrintQName, 13, sizeof(uint8_t), CH_DOS));
580 NDR_RETURN(ndr_pull_uint8(ndr, NDR_SCALARS, &r->Pad1));
581 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->Priority));
582 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->StartTime));
583 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->UntilTime));
584 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->SeparatorPageFilename));
585 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintProcessorDllName));
586 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintDestinationsName));
587 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintParameterString));
588 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->CommentString));
589 NDR_RETURN(ndr_pull_rap_PrintQStatusCode(ndr, NDR_SCALARS, &r->PrintQStatus));
590 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->PrintJobCount));
592 return NT_STATUS_OK;
595 static NTSTATUS rap_pull_rap_PrintQueue2(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue2 *r)
597 int i;
598 RAP_RETURN(rap_pull_rap_PrintQueue1(mem_ctx, ndr, convert, &r->queue));
599 r->job = talloc_zero_array(mem_ctx, struct rap_PrintJobInfo1, r->queue.PrintJobCount);
600 if (r->job == NULL) {
601 return NT_STATUS_NO_MEMORY;
603 for (i=0; i < r->queue.PrintJobCount; i++) {
604 RAP_RETURN(rap_pull_rap_JobInfo1(mem_ctx, ndr, convert, &r->job[i]));
607 return NT_STATUS_OK;
610 static NTSTATUS rap_pull_rap_PrintQueue3(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue3 *r)
612 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintQueueName));
613 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->Priority));
614 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->StartTime));
615 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->UntilTime));
616 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->Pad));
617 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->SeparatorPageFilename));
618 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintProcessorDllName));
619 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintParameterString));
620 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->CommentString));
621 NDR_RETURN(ndr_pull_rap_PrintQStatusCode(ndr, NDR_SCALARS, &r->PrintQStatus));
622 NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->PrintJobCount));
623 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->Printers));
624 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->DriverName));
625 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintDriverData));
627 return NT_STATUS_OK;
630 static NTSTATUS rap_pull_rap_PrintQueue4(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue4 *r)
632 int i;
633 RAP_RETURN(rap_pull_rap_PrintQueue3(mem_ctx, ndr, convert, &r->queue));
634 r->job = talloc_zero_array(mem_ctx, struct rap_PrintJobInfo2, r->queue.PrintJobCount);
635 if (r->job == NULL) {
636 return NT_STATUS_NO_MEMORY;
638 for (i=0; i < r->queue.PrintJobCount; i++) {
639 RAP_RETURN(rap_pull_rap_JobInfo2(mem_ctx, ndr, convert, &r->job[i]));
642 return NT_STATUS_OK;
645 static NTSTATUS rap_pull_rap_PrintQueue5(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue5 *r)
647 RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintQueueName));
649 return NT_STATUS_OK;
652 NTSTATUS smbcli_rap_netprintqenum(struct smbcli_tree *tree,
653 struct smb_iconv_convenience *iconv_convenience,
654 TALLOC_CTX *mem_ctx,
655 struct rap_NetPrintQEnum *r)
657 struct rap_call *call;
658 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
659 int i;
661 if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WPrintQEnum))) {
662 return NT_STATUS_NO_MEMORY;
665 rap_cli_push_word(call, r->in.level);
666 rap_cli_push_rcvbuf(call, r->in.bufsize);
667 rap_cli_expect_multiple_entries(call);
669 switch(r->in.level) {
670 case 0:
671 rap_cli_expect_format(call, "B13");
672 break;
673 case 1:
674 rap_cli_expect_format(call, "B13BWWWzzzzzWW");
675 break;
676 case 2:
677 rap_cli_expect_format(call, "B13BWWWzzzzzWN");
678 break;
679 case 3:
680 rap_cli_expect_format(call, "zWWWWzzzzWWzzl");
681 break;
682 case 4:
683 rap_cli_expect_format(call, "zWWWWzzzzWNzzl");
684 break;
685 case 5:
686 rap_cli_expect_format(call, "z");
687 break;
688 default:
689 result = NT_STATUS_INVALID_PARAMETER;
690 goto done;
693 if (DEBUGLEVEL >= 10) {
694 NDR_PRINT_IN_DEBUG(rap_NetPrintQEnum, r);
697 result = rap_cli_do_call(tree, iconv_convenience, call);
699 if (!NT_STATUS_IS_OK(result))
700 goto done;
702 result = NT_STATUS_INVALID_PARAMETER;
704 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
705 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
706 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
707 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
709 r->out.info = talloc_zero_array(mem_ctx, union rap_printq_info, r->out.count);
711 if (r->out.info == NULL) {
712 result = NT_STATUS_NO_MEMORY;
713 goto done;
716 for (i=0; i<r->out.count; i++) {
717 switch(r->in.level) {
718 case 0:
719 result = rap_pull_rap_PrintQueue0(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info0);
720 break;
721 case 1:
722 result = rap_pull_rap_PrintQueue1(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info1);
723 break;
724 case 2:
725 result = rap_pull_rap_PrintQueue2(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info2);
726 break;
727 case 3:
728 result = rap_pull_rap_PrintQueue3(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info3);
729 break;
730 case 4:
731 result = rap_pull_rap_PrintQueue4(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info4);
732 break;
733 case 5:
734 result = rap_pull_rap_PrintQueue5(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info5);
735 break;
739 if (!NT_STATUS_IS_OK(result)) {
740 goto done;
743 if (DEBUGLEVEL >= 10) {
744 NDR_PRINT_OUT_DEBUG(rap_NetPrintQEnum, r);
747 result = NT_STATUS_OK;
749 done:
750 talloc_free(call);
751 return result;
754 NTSTATUS smbcli_rap_netprintqgetinfo(struct smbcli_tree *tree,
755 struct smb_iconv_convenience *iconv_convenience,
756 TALLOC_CTX *mem_ctx,
757 struct rap_NetPrintQGetInfo *r)
759 struct rap_call *call;
760 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
762 if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WPrintQGetInfo))) {
763 return NT_STATUS_NO_MEMORY;
766 rap_cli_push_string(call, r->in.PrintQueueName);
767 rap_cli_push_word(call, r->in.level);
768 rap_cli_push_rcvbuf(call, r->in.bufsize);
769 rap_cli_expect_word(call);
771 switch(r->in.level) {
772 case 0:
773 rap_cli_expect_format(call, "B13");
774 break;
775 case 1:
776 rap_cli_expect_format(call, "B13BWWWzzzzzWW");
777 break;
778 case 2:
779 rap_cli_expect_format(call, "B13BWWWzzzzzWN");
780 break;
781 case 3:
782 rap_cli_expect_format(call, "zWWWWzzzzWWzzl");
783 break;
784 case 4:
785 rap_cli_expect_format(call, "zWWWWzzzzWNzzl");
786 break;
787 case 5:
788 rap_cli_expect_format(call, "z");
789 break;
790 default:
791 result = NT_STATUS_INVALID_PARAMETER;
792 goto done;
795 if (DEBUGLEVEL >= 10) {
796 NDR_PRINT_IN_DEBUG(rap_NetPrintQGetInfo, r);
799 result = rap_cli_do_call(tree, iconv_convenience, call);
801 if (!NT_STATUS_IS_OK(result))
802 goto done;
804 result = NT_STATUS_INVALID_PARAMETER;
806 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
807 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
808 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
810 switch(r->in.level) {
811 case 0:
812 result = rap_pull_rap_PrintQueue0(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info.info0);
813 break;
814 case 1:
815 result = rap_pull_rap_PrintQueue1(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info.info1);
816 break;
817 case 2:
818 result = rap_pull_rap_PrintQueue2(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info.info2);
819 break;
820 case 3:
821 result = rap_pull_rap_PrintQueue3(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info.info3);
822 break;
823 case 4:
824 result = rap_pull_rap_PrintQueue4(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info.info4);
825 break;
826 case 5:
827 result = rap_pull_rap_PrintQueue5(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info.info5);
828 break;
831 if (!NT_STATUS_IS_OK(result)) {
832 goto done;
835 if (DEBUGLEVEL >= 10) {
836 NDR_PRINT_OUT_DEBUG(rap_NetPrintQGetInfo, r);
839 done:
840 talloc_free(call);
841 return result;
844 NTSTATUS smbcli_rap_netprintjobpause(struct smbcli_tree *tree,
845 struct smb_iconv_convenience *iconv_convenience,
846 TALLOC_CTX *mem_ctx,
847 struct rap_NetPrintJobPause *r)
849 struct rap_call *call;
850 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
852 if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WPrintJobPause))) {
853 return NT_STATUS_NO_MEMORY;
856 rap_cli_push_word(call, r->in.JobID);
858 rap_cli_expect_format(call, "W");
860 if (DEBUGLEVEL >= 10) {
861 NDR_PRINT_IN_DEBUG(rap_NetPrintJobPause, r);
864 result = rap_cli_do_call(tree, iconv_convenience, call);
866 if (!NT_STATUS_IS_OK(result))
867 goto done;
869 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
870 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
872 if (DEBUGLEVEL >= 10) {
873 NDR_PRINT_OUT_DEBUG(rap_NetPrintJobPause, r);
876 done:
877 talloc_free(call);
878 return result;
881 NTSTATUS smbcli_rap_netprintjobcontinue(struct smbcli_tree *tree,
882 struct smb_iconv_convenience *iconv_convenience,
883 TALLOC_CTX *mem_ctx,
884 struct rap_NetPrintJobContinue *r)
886 struct rap_call *call;
887 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
889 if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WPrintJobContinue))) {
890 return NT_STATUS_NO_MEMORY;
893 rap_cli_push_word(call, r->in.JobID);
895 rap_cli_expect_format(call, "W");
897 if (DEBUGLEVEL >= 10) {
898 NDR_PRINT_IN_DEBUG(rap_NetPrintJobContinue, r);
901 result = rap_cli_do_call(tree, iconv_convenience, call);
903 if (!NT_STATUS_IS_OK(result))
904 goto done;
906 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
907 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
909 if (DEBUGLEVEL >= 10) {
910 NDR_PRINT_OUT_DEBUG(rap_NetPrintJobContinue, r);
913 done:
914 talloc_free(call);
915 return result;
918 NTSTATUS smbcli_rap_netprintjobdelete(struct smbcli_tree *tree,
919 struct smb_iconv_convenience *iconv_convenience,
920 TALLOC_CTX *mem_ctx,
921 struct rap_NetPrintJobDelete *r)
923 struct rap_call *call;
924 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
926 if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WPrintJobDel))) {
927 return NT_STATUS_NO_MEMORY;
930 rap_cli_push_word(call, r->in.JobID);
932 rap_cli_expect_format(call, "W");
934 if (DEBUGLEVEL >= 10) {
935 NDR_PRINT_IN_DEBUG(rap_NetPrintJobDelete, r);
938 result = rap_cli_do_call(tree, iconv_convenience, call);
940 if (!NT_STATUS_IS_OK(result))
941 goto done;
943 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
944 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
946 if (DEBUGLEVEL >= 10) {
947 NDR_PRINT_OUT_DEBUG(rap_NetPrintJobDelete, r);
950 done:
951 talloc_free(call);
952 return result;
955 NTSTATUS smbcli_rap_netprintqueuepause(struct smbcli_tree *tree,
956 struct smb_iconv_convenience *iconv_convenience,
957 TALLOC_CTX *mem_ctx,
958 struct rap_NetPrintQueuePause *r)
960 struct rap_call *call;
961 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
963 if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WPrintQPause))) {
964 return NT_STATUS_NO_MEMORY;
967 rap_cli_push_string(call, r->in.PrintQueueName);
969 rap_cli_expect_format(call, "");
971 if (DEBUGLEVEL >= 10) {
972 NDR_PRINT_IN_DEBUG(rap_NetPrintQueuePause, r);
975 result = rap_cli_do_call(tree, iconv_convenience, call);
977 if (!NT_STATUS_IS_OK(result))
978 goto done;
980 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
981 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
983 if (DEBUGLEVEL >= 10) {
984 NDR_PRINT_OUT_DEBUG(rap_NetPrintQueuePause, r);
987 done:
988 talloc_free(call);
989 return result;
992 NTSTATUS smbcli_rap_netprintqueueresume(struct smbcli_tree *tree,
993 struct smb_iconv_convenience *iconv_convenience,
994 TALLOC_CTX *mem_ctx,
995 struct rap_NetPrintQueueResume *r)
997 struct rap_call *call;
998 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1000 if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WPrintQContinue))) {
1001 return NT_STATUS_NO_MEMORY;
1004 rap_cli_push_string(call, r->in.PrintQueueName);
1006 rap_cli_expect_format(call, "");
1008 if (DEBUGLEVEL >= 10) {
1009 NDR_PRINT_IN_DEBUG(rap_NetPrintQueueResume, r);
1012 result = rap_cli_do_call(tree, iconv_convenience, call);
1014 if (!NT_STATUS_IS_OK(result))
1015 goto done;
1017 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
1018 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
1020 if (DEBUGLEVEL >= 10) {
1021 NDR_PRINT_OUT_DEBUG(rap_NetPrintQueueResume, r);
1024 done:
1025 talloc_free(call);
1026 return result;
1029 NTSTATUS smbcli_rap_netprintqueuepurge(struct smbcli_tree *tree,
1030 struct smb_iconv_convenience *iconv_convenience,
1031 TALLOC_CTX *mem_ctx,
1032 struct rap_NetPrintQueuePurge *r)
1034 struct rap_call *call;
1035 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1037 if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WPrintQPurge))) {
1038 return NT_STATUS_NO_MEMORY;
1041 rap_cli_push_string(call, r->in.PrintQueueName);
1043 rap_cli_expect_format(call, "");
1045 if (DEBUGLEVEL >= 10) {
1046 NDR_PRINT_IN_DEBUG(rap_NetPrintQueuePurge, r);
1049 result = rap_cli_do_call(tree, iconv_convenience, call);
1051 if (!NT_STATUS_IS_OK(result))
1052 goto done;
1054 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
1055 NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
1057 if (DEBUGLEVEL >= 10) {
1058 NDR_PRINT_OUT_DEBUG(rap_NetPrintQueuePurge, r);
1061 done:
1062 talloc_free(call);
1063 return result;
1066 static bool test_netservergetinfo(struct torture_context *tctx,
1067 struct smbcli_state *cli)
1069 struct rap_WserverGetInfo r;
1070 bool res = true;
1072 r.in.bufsize = 0xffff;
1074 r.in.level = 0;
1075 torture_assert_ntstatus_ok(tctx,
1076 smbcli_rap_netservergetinfo(cli->tree, lp_iconv_convenience(tctx->lp_ctx), tctx, &r),
1077 "rap_netservergetinfo level 0 failed");
1079 if (torture_setting_bool(tctx, "samba3", false)) {
1080 torture_skip(tctx, "skipping netservergetinfo level 1 against samba3");
1083 r.in.level = 1;
1084 torture_assert_ntstatus_ok(tctx,
1085 smbcli_rap_netservergetinfo(cli->tree, lp_iconv_convenience(tctx->lp_ctx), tctx, &r),
1086 "rap_netservergetinfo level 1 failed");
1088 return res;
1091 bool torture_rap_scan(struct torture_context *torture, struct smbcli_state *cli)
1093 int callno;
1095 for (callno = 0; callno < 0xffff; callno++) {
1096 struct rap_call *call = new_rap_cli_call(torture, lp_iconv_convenience(torture->lp_ctx), callno);
1097 NTSTATUS result;
1099 result = rap_cli_do_call(cli->tree, lp_iconv_convenience(torture->lp_ctx), call);
1101 if (!NT_STATUS_EQUAL(result, NT_STATUS_INVALID_PARAMETER))
1102 continue;
1104 printf("callno %d is RAP call\n", callno);
1107 return true;
1110 NTSTATUS torture_rap_init(void)
1112 struct torture_suite *suite = torture_suite_create(talloc_autofree_context(), "RAP");
1113 struct torture_suite *suite_basic = torture_suite_create(suite, "BASIC");
1115 torture_suite_add_suite(suite, suite_basic);
1116 torture_suite_add_suite(suite, torture_rap_rpc(suite));
1117 torture_suite_add_suite(suite, torture_rap_printing(suite));
1119 torture_suite_add_1smb_test(suite_basic, "netserverenum",
1120 test_netserverenum);
1121 torture_suite_add_1smb_test(suite_basic, "netshareenum",
1122 test_netshareenum);
1123 torture_suite_add_1smb_test(suite_basic, "netservergetinfo",
1124 test_netservergetinfo);
1126 torture_suite_add_1smb_test(suite, "SCAN", torture_rap_scan);
1128 suite->description = talloc_strdup(suite,
1129 "Remote Administration Protocol tests");
1131 torture_register_suite(suite);
1133 return NT_STATUS_OK;