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;
39 /* This is a convenience function for all OpenPrinterEx calls */
40 static bool test_OpenPrinterEx(struct torture_context *tctx,
41 struct dcerpc_pipe *p,
42 struct policy_handle *handle,
43 const char *printer_name,
47 struct spoolss_OpenPrinterEx op;
48 struct spoolss_UserLevel1 ul_1;
50 torture_comment(tctx, "Opening printer '%s'\n", printer_name);
52 op.in.printername = talloc_strdup(tctx, printer_name);
53 op.in.datatype = NULL;
54 op.in.devmode_ctr.devmode = NULL;
55 op.in.access_mask = access_mask;
57 op.in.userlevel.level1 = &ul_1;
58 op.out.handle = handle;
61 ul_1.client = "\\clientname";
62 ul_1.user = "username";
66 ul_1.processor = 4567;
68 status = dcerpc_spoolss_OpenPrinterEx(p, tctx, &op);
69 torture_assert_ntstatus_ok(tctx, status, "OpenPrinterEx failed");
70 torture_assert_werr_ok(tctx, op.out.result, "OpenPrinterEx failed");
75 static bool test_OpenPrinterAsAdmin(struct torture_context *tctx,
76 struct dcerpc_pipe *p,
77 const char *printername)
80 struct spoolss_OpenPrinterEx op;
81 struct spoolss_ClosePrinter cp;
82 struct spoolss_UserLevel1 ul_1;
83 struct policy_handle handle;
86 ul_1.client = "\\clientname";
87 ul_1.user = "username";
91 ul_1.processor = 4567;
93 op.in.printername = talloc_strdup(tctx, printername);
94 op.in.datatype = NULL;
95 op.in.devmode_ctr.devmode = NULL;
96 op.in.access_mask = SERVER_ALL_ACCESS;
98 op.in.userlevel.level1 = &ul_1;
99 op.out.handle = &handle;
101 cp.in.handle = &handle;
102 cp.out.handle = &handle;
104 torture_comment(tctx, "Testing OpenPrinterEx(%s) with admin rights\n",
107 status = dcerpc_spoolss_OpenPrinterEx(p, tctx, &op);
109 if (NT_STATUS_IS_OK(status) && W_ERROR_IS_OK(op.out.result)) {
110 status = dcerpc_spoolss_ClosePrinter(p, tctx, &cp);
111 torture_assert_ntstatus_ok(tctx, status, "ClosePrinter failed");
118 static bool test_ClosePrinter(struct torture_context *tctx,
119 struct dcerpc_pipe *p,
120 struct policy_handle *handle);
122 /* This replicates the opening sequence of OpenPrinterEx calls XP does */
123 static bool test_OpenPrinterSequence(struct torture_context *tctx,
124 struct dcerpc_pipe *p,
125 struct policy_handle *handle)
128 char *printername = talloc_asprintf(tctx, "\\\\%s",
129 dcerpc_server_name(p));
131 /* First, see if we can open the printer read_only */
132 ret = test_OpenPrinterEx(tctx, p, handle, printername, 0);
133 torture_assert(tctx, ret == true, "OpenPrinterEx failed.");
135 ret = test_ClosePrinter(tctx, p, handle);
136 torture_assert(tctx, ret, "ClosePrinter failed");
138 /* Now let's see if we have admin rights to it. */
139 ret = test_OpenPrinterAsAdmin(tctx, p, printername);
140 torture_assert(tctx, ret == true,
141 "OpenPrinterEx as admin failed unexpectedly.");
143 ret = test_OpenPrinterEx(tctx, p, handle, printername, SERVER_EXECUTE);
144 torture_assert(tctx, ret == true, "OpenPrinterEx failed.");
149 static bool test_GetPrinterData(struct torture_context *tctx,
150 struct dcerpc_pipe *p,
151 struct policy_handle *handle,
152 const char *value_name,
153 WERROR expected_werr,
154 uint32_t expected_value)
157 struct spoolss_GetPrinterData gpd;
159 enum winreg_Type type;
160 union spoolss_PrinterData data;
162 torture_comment(tctx, "Testing GetPrinterData(%s).\n", value_name);
163 gpd.in.handle = handle;
164 gpd.in.value_name = value_name;
166 gpd.out.needed = &needed;
167 gpd.out.type = &type;
168 gpd.out.data = &data;
170 status = dcerpc_spoolss_GetPrinterData(p, tctx, &gpd);
171 torture_assert_ntstatus_ok(tctx, status, "GetPrinterData failed.");
172 torture_assert_werr_equal(tctx, gpd.out.result, expected_werr,
173 "GetPrinterData did not return expected error value.");
175 if (W_ERROR_IS_OK(expected_werr)) {
176 torture_assert_int_equal(tctx, data.value,
178 "GetPrinterData did not return expected value.");
183 static bool test_EnumPrinters(struct torture_context *tctx,
184 struct dcerpc_pipe *p,
185 struct test_spoolss_win_context *ctx,
186 uint32_t initial_blob_size)
189 struct spoolss_EnumPrinters ep;
190 DATA_BLOB blob = data_blob_talloc_zero(ctx, initial_blob_size);
193 union spoolss_PrinterInfo *info;
195 ep.in.flags = PRINTER_ENUM_NAME;
196 ep.in.server = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
198 ep.in.buffer = &blob;
199 ep.in.offered = initial_blob_size;
200 ep.out.needed = &needed;
201 ep.out.count = &count;
204 status = dcerpc_spoolss_EnumPrinters(p, ctx, &ep);
205 torture_assert_ntstatus_ok(tctx, status, "EnumPrinters failed.");
207 if (W_ERROR_EQUAL(ep.out.result, WERR_INSUFFICIENT_BUFFER)) {
208 blob = data_blob_talloc_zero(ctx, needed);
209 ep.in.buffer = &blob;
210 ep.in.offered = needed;
211 status = dcerpc_spoolss_EnumPrinters(p, ctx, &ep);
212 torture_assert_ntstatus_ok(tctx, status,"EnumPrinters failed.");
215 torture_assert_werr_ok(tctx, ep.out.result, "EnumPrinters failed.");
217 ctx->printer_count = count;
218 ctx->printer_info = info;
220 torture_comment(tctx, "Found %d printer(s).\n", ctx->printer_count);
225 static bool test_GetPrinter(struct torture_context *tctx,
226 struct dcerpc_pipe *p,
227 struct policy_handle *handle,
228 struct test_spoolss_win_context *ctx,
230 uint32_t initial_blob_size)
233 struct spoolss_GetPrinter gp;
234 DATA_BLOB blob = data_blob_talloc_zero(ctx, initial_blob_size);
237 torture_comment(tctx, "Test GetPrinter level %d\n", level);
239 gp.in.handle = handle;
241 gp.in.buffer = (initial_blob_size == 0)?NULL:&blob;
242 gp.in.offered = initial_blob_size;
243 gp.out.needed = &needed;
245 status = dcerpc_spoolss_GetPrinter(p, tctx, &gp);
246 torture_assert_ntstatus_ok(tctx, status, "GetPrinter failed");
248 if (W_ERROR_EQUAL(gp.out.result, WERR_INSUFFICIENT_BUFFER)) {
249 blob = data_blob_talloc_zero(ctx, needed);
250 gp.in.buffer = &blob;
251 gp.in.offered = needed;
252 status = dcerpc_spoolss_GetPrinter(p, tctx, &gp);
253 torture_assert_ntstatus_ok(tctx, status, "GetPrinter failed");
256 torture_assert_werr_ok(tctx, gp.out.result, "GetPrinter failed");
258 ctx->current_info = gp.out.info;
262 static bool test_EnumJobs(struct torture_context *tctx,
263 struct dcerpc_pipe *p,
264 struct policy_handle *handle)
267 struct spoolss_EnumJobs ej;
268 DATA_BLOB blob = data_blob_talloc_zero(tctx, 1024);
271 union spoolss_JobInfo *info;
273 torture_comment(tctx, "Test EnumJobs\n");
275 ej.in.handle = handle;
277 ej.in.buffer = &blob;
278 ej.in.offered = 1024;
279 ej.out.needed = &needed;
280 ej.out.count = &count;
283 status = dcerpc_spoolss_EnumJobs(p, tctx, &ej);
284 torture_assert_ntstatus_ok(tctx, status, "EnumJobs failed");
285 torture_assert_werr_ok(tctx, ej.out.result, "EnumJobs failed");
290 static bool test_GetPrinterDriver2(struct torture_context *tctx,
291 struct dcerpc_pipe *p,
292 struct policy_handle *handle)
295 struct spoolss_GetPrinterDriver2 gpd2;
296 DATA_BLOB blob = data_blob_talloc_zero(tctx, 87424);
298 uint32_t server_major_version;
299 uint32_t server_minor_version;
301 torture_comment(tctx, "Testing GetPrinterDriver2\n");
303 gpd2.in.handle = handle;
304 gpd2.in.architecture = "Windows NT x86";
306 gpd2.in.buffer = &blob;
307 gpd2.in.offered = 87424;
308 gpd2.in.client_major_version = 3;
309 gpd2.in.client_minor_version = 0;
310 gpd2.out.needed = &needed;
311 gpd2.out.server_major_version = &server_major_version;
312 gpd2.out.server_minor_version = &server_minor_version;
314 status = dcerpc_spoolss_GetPrinterDriver2(p, tctx, &gpd2);
315 torture_assert_ntstatus_ok(tctx, status, "GetPrinterDriver2 failed");
316 torture_assert_werr_ok(tctx, gpd2.out.result,
317 "GetPrinterDriver2 failed.");
322 static bool test_EnumForms(struct torture_context *tctx,
323 struct dcerpc_pipe *p,
324 struct policy_handle *handle,
325 uint32_t initial_blob_size)
328 struct spoolss_EnumForms ef;
329 DATA_BLOB blob = data_blob_talloc_zero(tctx, initial_blob_size);
332 union spoolss_FormInfo *info;
334 torture_comment(tctx, "Testing EnumForms\n");
336 ef.in.handle = handle;
338 ef.in.buffer = (initial_blob_size == 0)?NULL:&blob;
339 ef.in.offered = initial_blob_size;
340 ef.out.needed = &needed;
341 ef.out.count = &count;
344 status = dcerpc_spoolss_EnumForms(p, tctx, &ef);
345 torture_assert_ntstatus_ok(tctx, status, "EnumForms failed");
347 if (W_ERROR_EQUAL(ef.out.result, WERR_INSUFFICIENT_BUFFER)) {
348 blob = data_blob_talloc_zero(tctx, needed);
349 ef.in.buffer = &blob;
350 ef.in.offered = needed;
351 status = dcerpc_spoolss_EnumForms(p, tctx, &ef);
352 torture_assert_ntstatus_ok(tctx, status, "EnumForms failed");
355 torture_assert_werr_ok(tctx, ef.out.result, "EnumForms failed");
360 static bool test_EnumPrinterKey(struct torture_context *tctx,
361 struct dcerpc_pipe *p,
362 struct policy_handle *handle,
364 struct test_spoolss_win_context *ctx)
367 struct spoolss_EnumPrinterKey epk;
369 const char **key_buffer = NULL;
371 torture_comment(tctx, "Testing EnumPrinterKey(%s)\n", key);
373 epk.in.handle = handle;
374 epk.in.key_name = talloc_strdup(tctx, key);
376 epk.out.needed = &needed;
377 epk.out.key_buffer = &key_buffer;
379 status = dcerpc_spoolss_EnumPrinterKey(p, tctx, &epk);
380 torture_assert_ntstatus_ok(tctx, status, "EnumPrinterKey failed");
383 if (W_ERROR_EQUAL(epk.out.result, WERR_MORE_DATA)) {
384 epk.in.offered = needed;
385 status = dcerpc_spoolss_EnumPrinterKey(p, tctx, &epk);
386 torture_assert_ntstatus_ok(tctx, status,
387 "EnumPrinterKey failed");
390 torture_assert_werr_ok(tctx, epk.out.result, "EnumPrinterKey failed");
392 ctx->printer_keys = key_buffer;
397 static bool test_EnumPrinterDataEx(struct torture_context *tctx,
398 struct dcerpc_pipe *p,
399 struct policy_handle *handle,
401 uint32_t initial_blob_size,
402 WERROR expected_error)
405 struct spoolss_EnumPrinterDataEx epde;
406 struct spoolss_PrinterEnumValues *info;
410 torture_comment(tctx, "Testing EnumPrinterDataEx(%s)\n", key);
412 epde.in.handle = handle;
413 epde.in.key_name = talloc_strdup(tctx, key);
415 epde.out.needed = &needed;
416 epde.out.count = &count;
417 epde.out.info = &info;
419 status = dcerpc_spoolss_EnumPrinterDataEx(p, tctx, &epde);
420 torture_assert_ntstatus_ok(tctx, status, "EnumPrinterDataEx failed.");
421 if (W_ERROR_EQUAL(epde.out.result, WERR_MORE_DATA)) {
422 epde.in.offered = needed;
423 status = dcerpc_spoolss_EnumPrinterDataEx(p, tctx, &epde);
424 torture_assert_ntstatus_ok(tctx, status,
425 "EnumPrinterDataEx failed.");
428 torture_assert_werr_equal(tctx, epde.out.result, expected_error,
429 "EnumPrinterDataEx failed.");
434 static bool test_ClosePrinter(struct torture_context *tctx,
435 struct dcerpc_pipe *p,
436 struct policy_handle *handle)
439 struct spoolss_ClosePrinter cp;
441 cp.in.handle = handle;
442 cp.out.handle = handle;
444 status = dcerpc_spoolss_ClosePrinter(p, tctx, &cp);
445 torture_assert_ntstatus_ok(tctx, status, "ClosePrinter failed");
450 static bool test_WinXP(struct torture_context *tctx, struct dcerpc_pipe *p)
453 struct test_spoolss_win_context *ctx, *tmp_ctx;
454 struct policy_handle handle01, handle02, handle03, handle04;
455 /* Sometimes a handle stays unused. In order to make this clear in the
456 * code, the unused_handle structures are used for that. */
457 struct policy_handle unused_handle1, unused_handle2;
461 ntvfs_init(tctx->lp_ctx);
463 ctx = talloc_zero(tctx, struct test_spoolss_win_context);
464 tmp_ctx = talloc_zero(tctx, struct test_spoolss_win_context);
466 ret &= test_OpenPrinterSequence(tctx, p, &handle01);
467 ret &= test_GetPrinterData(tctx, p, &handle01,"UISingleJobStatusString",
468 WERR_INVALID_PARAM, 0);
469 torture_comment(tctx, "Skip RemoteFindNextPrinterChangeNotifyEx test\n");
471 server_name = talloc_asprintf(ctx, "\\\\%s", dcerpc_server_name(p));
472 ret &= test_OpenPrinterEx(tctx, p, &unused_handle1, server_name, 0);
474 ret &= test_EnumPrinters(tctx, p, ctx, 1024);
476 ret &= test_OpenPrinterEx(tctx, p, &handle02, server_name, 0);
477 ret &= test_GetPrinterData(tctx, p, &handle02, "MajorVersion", WERR_OK,
479 ret &= test_ClosePrinter(tctx, p, &handle02);
481 /* If no printers were found, skip all tests that need a printer */
482 if (ctx->printer_count == 0) {
486 ret &= test_OpenPrinterEx(tctx, p, &handle02,
487 ctx->printer_info[0].info2.printername,
489 ret &= test_GetPrinter(tctx, p, &handle02, ctx, 2, 0);
491 torture_assert_str_equal(tctx, ctx->current_info->info2.printername,
492 ctx->printer_info[0].info2.printername,
493 "GetPrinter returned unexpected printername");
494 /*FIXME: Test more components of the PrinterInfo2 struct */
496 ret &= test_OpenPrinterEx(tctx, p, &handle03,
497 ctx->printer_info[0].info2.printername, 0);
498 ret &= test_GetPrinter(tctx, p, &handle03, ctx, 0, 1164);
499 ret &= test_GetPrinter(tctx, p, &handle03, ctx, 2, 0);
501 ret &= test_OpenPrinterEx(tctx, p, &handle04,
502 ctx->printer_info[0].info2.printername, 0);
503 ret &= test_GetPrinter(tctx, p, &handle04, ctx, 2, 0);
504 ret &= test_ClosePrinter(tctx, p, &handle04);
506 ret &= test_OpenPrinterEx(tctx, p, &handle04,
507 ctx->printer_info[0].info2.printername, 0);
508 ret &= test_GetPrinter(tctx, p, &handle04, ctx, 2, 4096);
509 ret &= test_ClosePrinter(tctx, p, &handle04);
511 ret &= test_OpenPrinterAsAdmin(tctx, p,
512 ctx->printer_info[0].info2.printername);
514 ret &= test_OpenPrinterEx(tctx, p, &handle04,
515 ctx->printer_info[0].info2.printername, PRINTER_READ);
516 ret &= test_GetPrinterData(tctx, p, &handle04,"UISingleJobStatusString",
518 torture_comment(tctx, "Skip RemoteFindNextPrinterChangeNotifyEx test\n");
520 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
521 ctx->printer_info[0].info2.printername, 0);
523 ret &= test_EnumJobs(tctx, p, &handle04);
524 ret &= test_GetPrinter(tctx, p, &handle04, ctx, 2, 4096);
526 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
527 ret &= test_ClosePrinter(tctx, p, &handle04);
529 ret &= test_EnumPrinters(tctx, p, ctx, 1556);
530 ret &= test_GetPrinterDriver2(tctx, p, &handle03);
531 ret &= test_EnumForms(tctx, p, &handle03, 0);
533 ret &= test_EnumPrinterKey(tctx, p, &handle03, "", ctx);
535 for (i=0; ctx->printer_keys[i] != NULL; i++) {
537 ret &= test_EnumPrinterKey(tctx, p, &handle03,
538 ctx->printer_keys[i],
540 ret &= test_EnumPrinterDataEx(tctx, p, &handle03,
541 ctx->printer_keys[i], 0,
545 ret &= test_EnumPrinterDataEx(tctx, p, &handle03, "", 0,
548 ret &= test_GetPrinter(tctx, p, &handle03, tmp_ctx, 2, 0);
550 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
551 ctx->printer_info[0].info2.printername, 0);
552 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
554 ret &= test_GetPrinter(tctx, p, &handle03, tmp_ctx, 2, 2556);
556 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
557 ctx->printer_info[0].info2.printername, 0);
558 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
560 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
561 ctx->printer_info[0].info2.printername, 0);
562 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
564 ret &= test_GetPrinter(tctx, p, &handle03, tmp_ctx, 7, 0);
566 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
567 ctx->printer_info[0].info2.printername, 0);
568 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
570 ret &= test_ClosePrinter(tctx, p, &handle03);
572 ret &= test_OpenPrinterEx(tctx, p, &unused_handle2,
573 ctx->printer_info[0].info2.printername, 0);
574 ret &= test_ClosePrinter(tctx, p, &unused_handle2);
576 ret &= test_OpenPrinterEx(tctx, p, &handle03, server_name, 0);
577 ret &= test_GetPrinterData(tctx, p, &handle03, "W3SvcInstalled",
579 ret &= test_ClosePrinter(tctx, p, &handle03);
581 ret &= test_ClosePrinter(tctx, p, &unused_handle1);
582 ret &= test_ClosePrinter(tctx, p, &handle02);
584 ret &= test_OpenPrinterEx(tctx, p, &handle02,
585 ctx->printer_info[0].info2.sharename, 0);
586 ret &= test_GetPrinter(tctx, p, &handle02, tmp_ctx, 2, 0);
587 ret &= test_ClosePrinter(tctx, p, &handle02);
590 ret &= test_ClosePrinter(tctx, p, &handle01);
592 talloc_free(tmp_ctx);
597 struct torture_suite *torture_rpc_spoolss_win(TALLOC_CTX *mem_ctx)
599 struct torture_suite *suite = torture_suite_create(mem_ctx, "SPOOLSS-WIN");
601 struct torture_rpc_tcase *tcase = torture_suite_add_rpc_iface_tcase(suite,
602 "win", &ndr_table_spoolss);
604 torture_rpc_tcase_add_test(tcase, "testWinXP", test_WinXP);