2 Unix SMB/CIFS implementation.
3 test suite for spoolss rpc operations as performed by various win versions
5 Copyright (C) Kai Blin 2007
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "torture/torture.h"
23 #include "torture/rpc/rpc.h"
24 #include "librpc/gen_ndr/ndr_spoolss_c.h"
25 #include "rpc_server/dcerpc_server.h"
26 #include "ntvfs/ntvfs.h"
27 #include "param/param.h"
29 struct test_spoolss_win_context {
31 uint32_t printer_count;
32 union spoolss_PrinterInfo *printer_info;
33 union spoolss_PrinterInfo *current_info;
36 const char **printer_keys;
38 bool printer_has_driver;
41 /* This is a convenience function for all OpenPrinterEx calls */
42 static bool test_OpenPrinterEx(struct torture_context *tctx,
43 struct dcerpc_pipe *p,
44 struct policy_handle *handle,
45 const char *printer_name,
49 struct spoolss_OpenPrinterEx op;
50 struct spoolss_UserLevel1 ul_1;
52 torture_comment(tctx, "Opening printer '%s'\n", printer_name);
54 op.in.printername = talloc_strdup(tctx, printer_name);
55 op.in.datatype = NULL;
56 op.in.devmode_ctr.devmode = NULL;
57 op.in.access_mask = access_mask;
59 op.in.userlevel.level1 = &ul_1;
60 op.out.handle = handle;
63 ul_1.client = "\\clientname";
64 ul_1.user = "username";
68 ul_1.processor = 4567;
70 status = dcerpc_spoolss_OpenPrinterEx(p, tctx, &op);
71 torture_assert_ntstatus_ok(tctx, status, "OpenPrinterEx failed");
72 torture_assert_werr_ok(tctx, op.out.result, "OpenPrinterEx failed");
77 static bool test_OpenPrinterAsAdmin(struct torture_context *tctx,
78 struct dcerpc_pipe *p,
79 const char *printername)
82 struct spoolss_OpenPrinterEx op;
83 struct spoolss_ClosePrinter cp;
84 struct spoolss_UserLevel1 ul_1;
85 struct policy_handle handle;
88 ul_1.client = "\\clientname";
89 ul_1.user = "username";
93 ul_1.processor = 4567;
95 op.in.printername = talloc_strdup(tctx, printername);
96 op.in.datatype = NULL;
97 op.in.devmode_ctr.devmode = NULL;
98 op.in.access_mask = SERVER_ALL_ACCESS;
100 op.in.userlevel.level1 = &ul_1;
101 op.out.handle = &handle;
103 cp.in.handle = &handle;
104 cp.out.handle = &handle;
106 torture_comment(tctx, "Testing OpenPrinterEx(%s) with admin rights\n",
109 status = dcerpc_spoolss_OpenPrinterEx(p, tctx, &op);
111 if (NT_STATUS_IS_OK(status) && W_ERROR_IS_OK(op.out.result)) {
112 status = dcerpc_spoolss_ClosePrinter(p, tctx, &cp);
113 torture_assert_ntstatus_ok(tctx, status, "ClosePrinter failed");
120 static bool test_ClosePrinter(struct torture_context *tctx,
121 struct dcerpc_pipe *p,
122 struct policy_handle *handle);
124 /* This replicates the opening sequence of OpenPrinterEx calls XP does */
125 static bool test_OpenPrinterSequence(struct torture_context *tctx,
126 struct dcerpc_pipe *p,
127 struct policy_handle *handle)
130 char *printername = talloc_asprintf(tctx, "\\\\%s",
131 dcerpc_server_name(p));
133 /* First, see if we can open the printer read_only */
134 ret = test_OpenPrinterEx(tctx, p, handle, printername, 0);
135 torture_assert(tctx, ret == true, "OpenPrinterEx failed.");
137 ret = test_ClosePrinter(tctx, p, handle);
138 torture_assert(tctx, ret, "ClosePrinter failed");
140 /* Now let's see if we have admin rights to it. */
141 ret = test_OpenPrinterAsAdmin(tctx, p, printername);
142 torture_assert(tctx, ret == true,
143 "OpenPrinterEx as admin failed unexpectedly.");
145 ret = test_OpenPrinterEx(tctx, p, handle, printername, SERVER_EXECUTE);
146 torture_assert(tctx, ret == true, "OpenPrinterEx failed.");
151 static bool test_GetPrinterData(struct torture_context *tctx,
152 struct dcerpc_pipe *p,
153 struct policy_handle *handle,
154 const char *value_name,
155 WERROR expected_werr,
156 uint32_t expected_value)
159 struct spoolss_GetPrinterData gpd;
161 enum winreg_Type type;
162 union spoolss_PrinterData data;
164 torture_comment(tctx, "Testing GetPrinterData(%s).\n", value_name);
165 gpd.in.handle = handle;
166 gpd.in.value_name = value_name;
168 gpd.out.needed = &needed;
169 gpd.out.type = &type;
170 gpd.out.data = &data;
172 status = dcerpc_spoolss_GetPrinterData(p, tctx, &gpd);
173 torture_assert_ntstatus_ok(tctx, status, "GetPrinterData failed.");
174 torture_assert_werr_equal(tctx, gpd.out.result, expected_werr,
175 "GetPrinterData did not return expected error value.");
177 if (W_ERROR_IS_OK(expected_werr)) {
178 torture_assert_int_equal(tctx, data.value,
180 "GetPrinterData did not return expected value.");
185 static bool test_EnumPrinters(struct torture_context *tctx,
186 struct dcerpc_pipe *p,
187 struct test_spoolss_win_context *ctx,
188 uint32_t initial_blob_size)
191 struct spoolss_EnumPrinters ep;
192 DATA_BLOB blob = data_blob_talloc_zero(ctx, initial_blob_size);
195 union spoolss_PrinterInfo *info;
197 ep.in.flags = PRINTER_ENUM_NAME;
198 ep.in.server = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
200 ep.in.buffer = &blob;
201 ep.in.offered = initial_blob_size;
202 ep.out.needed = &needed;
203 ep.out.count = &count;
206 status = dcerpc_spoolss_EnumPrinters(p, ctx, &ep);
207 torture_assert_ntstatus_ok(tctx, status, "EnumPrinters failed.");
209 if (W_ERROR_EQUAL(ep.out.result, WERR_INSUFFICIENT_BUFFER)) {
210 blob = data_blob_talloc_zero(ctx, needed);
211 ep.in.buffer = &blob;
212 ep.in.offered = needed;
213 status = dcerpc_spoolss_EnumPrinters(p, ctx, &ep);
214 torture_assert_ntstatus_ok(tctx, status,"EnumPrinters failed.");
217 torture_assert_werr_ok(tctx, ep.out.result, "EnumPrinters failed.");
219 ctx->printer_count = count;
220 ctx->printer_info = info;
222 torture_comment(tctx, "Found %d printer(s).\n", ctx->printer_count);
227 static bool test_GetPrinter(struct torture_context *tctx,
228 struct dcerpc_pipe *p,
229 struct policy_handle *handle,
230 struct test_spoolss_win_context *ctx,
232 uint32_t initial_blob_size)
235 struct spoolss_GetPrinter gp;
236 DATA_BLOB blob = data_blob_talloc_zero(ctx, initial_blob_size);
239 torture_comment(tctx, "Test GetPrinter level %d\n", level);
241 gp.in.handle = handle;
243 gp.in.buffer = (initial_blob_size == 0)?NULL:&blob;
244 gp.in.offered = initial_blob_size;
245 gp.out.needed = &needed;
247 status = dcerpc_spoolss_GetPrinter(p, tctx, &gp);
248 torture_assert_ntstatus_ok(tctx, status, "GetPrinter failed");
250 if (W_ERROR_EQUAL(gp.out.result, WERR_INSUFFICIENT_BUFFER)) {
251 blob = data_blob_talloc_zero(ctx, needed);
252 gp.in.buffer = &blob;
253 gp.in.offered = needed;
254 status = dcerpc_spoolss_GetPrinter(p, tctx, &gp);
255 torture_assert_ntstatus_ok(tctx, status, "GetPrinter failed");
258 torture_assert_werr_ok(tctx, gp.out.result, "GetPrinter failed");
260 ctx->current_info = gp.out.info;
262 if (level == 2 && gp.out.info) {
263 ctx->printer_has_driver = gp.out.info->info2.drivername &&
264 strlen(gp.out.info->info2.drivername);
270 static bool test_EnumJobs(struct torture_context *tctx,
271 struct dcerpc_pipe *p,
272 struct policy_handle *handle)
275 struct spoolss_EnumJobs ej;
276 DATA_BLOB blob = data_blob_talloc_zero(tctx, 1024);
279 union spoolss_JobInfo *info;
281 torture_comment(tctx, "Test EnumJobs\n");
283 ej.in.handle = handle;
285 ej.in.buffer = &blob;
286 ej.in.offered = 1024;
287 ej.out.needed = &needed;
288 ej.out.count = &count;
291 status = dcerpc_spoolss_EnumJobs(p, tctx, &ej);
292 torture_assert_ntstatus_ok(tctx, status, "EnumJobs failed");
293 if (W_ERROR_EQUAL(ej.out.result, WERR_INSUFFICIENT_BUFFER)) {
294 blob = data_blob_talloc_zero(tctx, needed);
295 ej.in.offered = needed;
296 ej.in.buffer = &blob;
297 status = dcerpc_spoolss_EnumJobs(p, tctx, &ej);
298 torture_assert_ntstatus_ok(tctx, status, "EnumJobs failed");
300 torture_assert_werr_ok(tctx, ej.out.result, "EnumJobs failed");
305 static bool test_GetPrinterDriver2(struct torture_context *tctx,
306 struct dcerpc_pipe *p,
307 struct test_spoolss_win_context *ctx,
308 struct policy_handle *handle)
311 struct spoolss_GetPrinterDriver2 gpd2;
312 DATA_BLOB blob = data_blob_talloc_zero(tctx, 87424);
314 uint32_t server_major_version;
315 uint32_t server_minor_version;
317 torture_comment(tctx, "Testing GetPrinterDriver2\n");
319 gpd2.in.handle = handle;
320 gpd2.in.architecture = "Windows NT x86";
322 gpd2.in.buffer = &blob;
323 gpd2.in.offered = 87424;
324 gpd2.in.client_major_version = 3;
325 gpd2.in.client_minor_version = 0;
326 gpd2.out.needed = &needed;
327 gpd2.out.server_major_version = &server_major_version;
328 gpd2.out.server_minor_version = &server_minor_version;
330 status = dcerpc_spoolss_GetPrinterDriver2(p, tctx, &gpd2);
331 torture_assert_ntstatus_ok(tctx, status, "GetPrinterDriver2 failed");
333 if (ctx->printer_has_driver) {
334 torture_assert_werr_ok(tctx, gpd2.out.result,
335 "GetPrinterDriver2 failed.");
341 static bool test_EnumForms(struct torture_context *tctx,
342 struct dcerpc_pipe *p,
343 struct policy_handle *handle,
344 uint32_t initial_blob_size)
347 struct spoolss_EnumForms ef;
348 DATA_BLOB blob = data_blob_talloc_zero(tctx, initial_blob_size);
351 union spoolss_FormInfo *info;
353 torture_comment(tctx, "Testing EnumForms\n");
355 ef.in.handle = handle;
357 ef.in.buffer = (initial_blob_size == 0)?NULL:&blob;
358 ef.in.offered = initial_blob_size;
359 ef.out.needed = &needed;
360 ef.out.count = &count;
363 status = dcerpc_spoolss_EnumForms(p, tctx, &ef);
364 torture_assert_ntstatus_ok(tctx, status, "EnumForms failed");
366 if (W_ERROR_EQUAL(ef.out.result, WERR_INSUFFICIENT_BUFFER)) {
367 blob = data_blob_talloc_zero(tctx, needed);
368 ef.in.buffer = &blob;
369 ef.in.offered = needed;
370 status = dcerpc_spoolss_EnumForms(p, tctx, &ef);
371 torture_assert_ntstatus_ok(tctx, status, "EnumForms failed");
374 torture_assert_werr_ok(tctx, ef.out.result, "EnumForms failed");
379 static bool test_EnumPrinterKey(struct torture_context *tctx,
380 struct dcerpc_pipe *p,
381 struct policy_handle *handle,
383 struct test_spoolss_win_context *ctx)
386 struct spoolss_EnumPrinterKey epk;
388 const char **key_buffer = NULL;
390 torture_comment(tctx, "Testing EnumPrinterKey(%s)\n", key);
392 epk.in.handle = handle;
393 epk.in.key_name = talloc_strdup(tctx, key);
395 epk.out.needed = &needed;
396 epk.out.key_buffer = &key_buffer;
398 status = dcerpc_spoolss_EnumPrinterKey(p, tctx, &epk);
399 torture_assert_ntstatus_ok(tctx, status, "EnumPrinterKey failed");
402 if (W_ERROR_EQUAL(epk.out.result, WERR_MORE_DATA)) {
403 epk.in.offered = needed;
404 status = dcerpc_spoolss_EnumPrinterKey(p, tctx, &epk);
405 torture_assert_ntstatus_ok(tctx, status,
406 "EnumPrinterKey failed");
409 torture_assert_werr_ok(tctx, epk.out.result, "EnumPrinterKey failed");
411 ctx->printer_keys = key_buffer;
416 static bool test_EnumPrinterDataEx(struct torture_context *tctx,
417 struct dcerpc_pipe *p,
418 struct policy_handle *handle,
420 uint32_t initial_blob_size,
421 WERROR expected_error)
424 struct spoolss_EnumPrinterDataEx epde;
425 struct spoolss_PrinterEnumValues *info;
429 torture_comment(tctx, "Testing EnumPrinterDataEx(%s)\n", key);
431 epde.in.handle = handle;
432 epde.in.key_name = talloc_strdup(tctx, key);
434 epde.out.needed = &needed;
435 epde.out.count = &count;
436 epde.out.info = &info;
438 status = dcerpc_spoolss_EnumPrinterDataEx(p, tctx, &epde);
439 torture_assert_ntstatus_ok(tctx, status, "EnumPrinterDataEx failed.");
440 if (W_ERROR_EQUAL(epde.out.result, WERR_MORE_DATA)) {
441 epde.in.offered = needed;
442 status = dcerpc_spoolss_EnumPrinterDataEx(p, tctx, &epde);
443 torture_assert_ntstatus_ok(tctx, status,
444 "EnumPrinterDataEx failed.");
447 torture_assert_werr_equal(tctx, epde.out.result, expected_error,
448 "EnumPrinterDataEx failed.");
453 static bool test_ClosePrinter(struct torture_context *tctx,
454 struct dcerpc_pipe *p,
455 struct policy_handle *handle)
458 struct spoolss_ClosePrinter cp;
460 cp.in.handle = handle;
461 cp.out.handle = handle;
463 status = dcerpc_spoolss_ClosePrinter(p, tctx, &cp);
464 torture_assert_ntstatus_ok(tctx, status, "ClosePrinter failed");
469 static bool test_WinXP(struct torture_context *tctx, struct dcerpc_pipe *p)
472 struct test_spoolss_win_context *ctx, *tmp_ctx;
473 struct policy_handle handle01, handle02, handle03, handle04;
474 /* Sometimes a handle stays unused. In order to make this clear in the
475 * code, the unused_handle structures are used for that. */
476 struct policy_handle unused_handle1, unused_handle2;
480 ntvfs_init(tctx->lp_ctx);
482 ctx = talloc_zero(tctx, struct test_spoolss_win_context);
483 tmp_ctx = talloc_zero(tctx, struct test_spoolss_win_context);
485 ret &= test_OpenPrinterSequence(tctx, p, &handle01);
486 ret &= test_GetPrinterData(tctx, p, &handle01,"UISingleJobStatusString",
487 WERR_INVALID_PARAM, 0);
488 torture_comment(tctx, "Skip RemoteFindNextPrinterChangeNotifyEx test\n");
490 server_name = talloc_asprintf(ctx, "\\\\%s", dcerpc_server_name(p));
491 ret &= test_OpenPrinterEx(tctx, p, &unused_handle1, server_name, 0);
493 ret &= test_EnumPrinters(tctx, p, ctx, 1024);
495 ret &= test_OpenPrinterEx(tctx, p, &handle02, server_name, 0);
496 ret &= test_GetPrinterData(tctx, p, &handle02, "MajorVersion", WERR_OK,
498 ret &= test_ClosePrinter(tctx, p, &handle02);
500 /* If no printers were found, skip all tests that need a printer */
501 if (ctx->printer_count == 0) {
505 ret &= test_OpenPrinterEx(tctx, p, &handle02,
506 ctx->printer_info[0].info2.printername,
508 ret &= test_GetPrinter(tctx, p, &handle02, ctx, 2, 0);
510 torture_assert_str_equal(tctx, ctx->current_info->info2.printername,
511 ctx->printer_info[0].info2.printername,
512 "GetPrinter returned unexpected printername");
513 /*FIXME: Test more components of the PrinterInfo2 struct */
515 ret &= test_OpenPrinterEx(tctx, p, &handle03,
516 ctx->printer_info[0].info2.printername, 0);
517 ret &= test_GetPrinter(tctx, p, &handle03, ctx, 0, 1164);
518 ret &= test_GetPrinter(tctx, p, &handle03, ctx, 2, 0);
520 ret &= test_OpenPrinterEx(tctx, p, &handle04,
521 ctx->printer_info[0].info2.printername, 0);
522 ret &= test_GetPrinter(tctx, p, &handle04, ctx, 2, 0);
523 ret &= test_ClosePrinter(tctx, p, &handle04);
525 ret &= test_OpenPrinterEx(tctx, p, &handle04,
526 ctx->printer_info[0].info2.printername, 0);
527 ret &= test_GetPrinter(tctx, p, &handle04, ctx, 2, 4096);
528 ret &= test_ClosePrinter(tctx, p, &handle04);
530 ret &= test_OpenPrinterAsAdmin(tctx, p,
531 ctx->printer_info[0].info2.printername);
533 ret &= test_OpenPrinterEx(tctx, p, &handle04,
534 ctx->printer_info[0].info2.printername, PRINTER_READ);
535 ret &= test_GetPrinterData(tctx, p, &handle04,"UISingleJobStatusString",
537 torture_comment(tctx, "Skip RemoteFindNextPrinterChangeNotifyEx test\n");
539 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
540 ctx->printer_info[0].info2.printername, 0);
542 ret &= test_EnumJobs(tctx, p, &handle04);
543 ret &= test_GetPrinter(tctx, p, &handle04, ctx, 2, 4096);
545 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
546 ret &= test_ClosePrinter(tctx, p, &handle04);
548 ret &= test_EnumPrinters(tctx, p, ctx, 1556);
549 ret &= test_GetPrinterDriver2(tctx, p, ctx, &handle03);
550 ret &= test_EnumForms(tctx, p, &handle03, 0);
552 ret &= test_EnumPrinterKey(tctx, p, &handle03, "", ctx);
554 for (i=0; ctx->printer_keys[i] != NULL; i++) {
556 ret &= test_EnumPrinterKey(tctx, p, &handle03,
557 ctx->printer_keys[i],
559 ret &= test_EnumPrinterDataEx(tctx, p, &handle03,
560 ctx->printer_keys[i], 0,
564 ret &= test_EnumPrinterDataEx(tctx, p, &handle03, "", 0,
567 ret &= test_GetPrinter(tctx, p, &handle03, tmp_ctx, 2, 0);
569 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
570 ctx->printer_info[0].info2.printername, 0);
571 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
573 ret &= test_GetPrinter(tctx, p, &handle03, tmp_ctx, 2, 2556);
575 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
576 ctx->printer_info[0].info2.printername, 0);
577 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
579 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
580 ctx->printer_info[0].info2.printername, 0);
581 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
583 ret &= test_GetPrinter(tctx, p, &handle03, tmp_ctx, 7, 0);
585 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
586 ctx->printer_info[0].info2.printername, 0);
587 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
589 ret &= test_ClosePrinter(tctx, p, &handle03);
591 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
592 ctx->printer_info[0].info2.printername, 0);
593 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
595 ret &= test_OpenPrinterEx(tctx, p, &handle03, server_name, 0);
596 ret &= test_GetPrinterData(tctx, p, &handle03, "W3SvcInstalled",
598 ret &= test_ClosePrinter(tctx, p, &handle03);
600 ret &= test_ClosePrinter(tctx, p, &unused_handle1);
601 ret &= test_ClosePrinter(tctx, p, &handle02);
603 ret &= test_OpenPrinterEx(tctx, p, &handle02,
604 ctx->printer_info[0].info2.sharename, 0);
605 ret &= test_GetPrinter(tctx, p, &handle02, tmp_ctx, 2, 0);
606 ret &= test_ClosePrinter(tctx, p, &handle02);
609 ret &= test_ClosePrinter(tctx, p, &handle01);
611 talloc_free(tmp_ctx);
616 struct torture_suite *torture_rpc_spoolss_win(TALLOC_CTX *mem_ctx)
618 struct torture_suite *suite = torture_suite_create(mem_ctx, "SPOOLSS-WIN");
620 struct torture_rpc_tcase *tcase = torture_suite_add_rpc_iface_tcase(suite,
621 "win", &ndr_table_spoolss);
623 torture_rpc_tcase_add_test(tcase, "testWinXP", test_WinXP);