2 Unix SMB/CIFS implementation.
3 test suite for spoolss rpc operations
5 Copyright (C) Tim Potter 2003
6 Copyright (C) Stefan Metzmacher 2005
7 Copyright (C) Jelmer Vernooij 2007
8 Copyright (C) Guenther Deschner 2009-2011,2013
10 This program is free software; you can redistribute it and/or modify
11 it under the terms of the GNU General Public License as published by
12 the Free Software Foundation; either version 3 of the License, or
13 (at your option) any later version.
15 This program is distributed in the hope that it will be useful,
16 but WITHOUT ANY WARRANTY; without even the implied warranty of
17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 GNU General Public License for more details.
20 You should have received a copy of the GNU General Public License
21 along with this program. If not, see <http://www.gnu.org/licenses/>.
25 #include "torture/torture.h"
26 #include "librpc/gen_ndr/ndr_misc.h"
27 #include "librpc/gen_ndr/ndr_spoolss.h"
28 #include "librpc/gen_ndr/ndr_spoolss_c.h"
29 #include "librpc/gen_ndr/ndr_winreg_c.h"
30 #include "librpc/gen_ndr/ndr_security.h"
31 #include "libcli/security/security.h"
32 #include "torture/rpc/torture_rpc.h"
33 #include "param/param.h"
34 #include "lib/registry/registry.h"
35 #include "libcli/libcli.h"
36 #include "libcli/raw/raw_proto.h"
37 #include "libcli/resolve/resolve.h"
38 #include "libcli/smb2/smb2.h"
39 #include "libcli/smb2/smb2_calls.h"
40 #include "lib/cmdline/popt_common.h"
41 #include "system/filesys.h"
42 #include "torture/ndr/ndr.h"
43 #include "torture/smb2/proto.h"
45 #define TORTURE_WELLKNOWN_PRINTER "torture_wkn_printer"
46 #define TORTURE_PRINTER "torture_printer"
47 #define TORTURE_WELLKNOWN_PRINTER_EX "torture_wkn_printer_ex"
48 #define TORTURE_PRINTER_EX "torture_printer_ex"
49 #define TORTURE_DRIVER "torture_driver"
50 #define TORTURE_DRIVER_EX "torture_driver_ex"
51 #define TORTURE_DRIVER_ADOBE "torture_driver_adobe"
52 #define TORTURE_DRIVER_EX_ADOBE "torture_driver_ex_adobe"
53 #define TORTURE_DRIVER_ADOBE_CUPSADDSMB "torture_driver_adobe_cupsaddsmb"
54 #define TORTURE_DRIVER_TIMESTAMPS "torture_driver_timestamps"
55 #define TORTURE_DRIVER_DELETER "torture_driver_deleter"
56 #define TORTURE_DRIVER_DELETERIN "torture_driver_deleterin"
57 #define TORTURE_PRINTER_STATIC1 "print1"
59 #define TOP_LEVEL_PRINT_KEY "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Print"
60 #define TOP_LEVEL_PRINT_PRINTERS_KEY TOP_LEVEL_PRINT_KEY "\\Printers"
61 #define TOP_LEVEL_CONTROL_KEY "SYSTEM\\CurrentControlSet\\Control\\Print"
62 #define TOP_LEVEL_CONTROL_FORMS_KEY TOP_LEVEL_CONTROL_KEY "\\Forms"
63 #define TOP_LEVEL_CONTROL_PRINTERS_KEY TOP_LEVEL_CONTROL_KEY "\\Printers"
64 #define TOP_LEVEL_CONTROL_ENVIRONMENTS_KEY TOP_LEVEL_CONTROL_KEY "\\Environments"
66 struct test_spoolss_context {
67 struct dcerpc_pipe *spoolss_pipe;
69 /* server environment */
70 const char *environment;
72 /* print server handle */
73 struct policy_handle server_handle;
76 uint32_t port_count[3];
77 union spoolss_PortInfo *ports[3];
79 /* for EnumPrinterDrivers */
80 uint32_t driver_count[9];
81 union spoolss_DriverInfo *drivers[9];
83 /* for EnumMonitors */
84 uint32_t monitor_count[3];
85 union spoolss_MonitorInfo *monitors[3];
87 /* for EnumPrintProcessors */
88 uint32_t print_processor_count[2];
89 union spoolss_PrintProcessorInfo *print_processors[2];
91 /* for EnumPrinters */
92 uint32_t printer_count[6];
93 union spoolss_PrinterInfo *printers[6];
96 struct torture_driver_context {
98 const char *driver_directory;
99 const char *environment;
102 const char *driver_directory;
103 const char *environment;
105 struct spoolss_AddDriverInfo8 info8;
109 struct torture_printer_context {
110 struct dcerpc_pipe *spoolss_pipe;
111 struct spoolss_SetPrinterInfo2 info2;
112 struct torture_driver_context driver;
117 struct spoolss_DeviceMode *devmode;
118 struct policy_handle handle;
121 static bool upload_printer_driver(struct torture_context *tctx,
122 const char *server_name,
123 struct torture_driver_context *d);
124 static bool remove_printer_driver(struct torture_context *tctx,
125 const char *server_name,
126 struct torture_driver_context *d);
127 static bool fillup_printserver_info(struct torture_context *tctx,
128 struct dcerpc_pipe *p,
129 struct torture_driver_context *d);
130 static bool test_AddPrinterDriver_args_level_3(struct torture_context *tctx,
131 struct dcerpc_binding_handle *b,
132 const char *server_name,
133 struct spoolss_AddDriverInfo8 *r,
136 const char *remote_driver_dir);
138 #define COMPARE_STRING(tctx, c,r,e) \
139 torture_assert_str_equal(tctx, c.e, r.e, "invalid value")
141 /* not every compiler supports __typeof__() */
143 #define _CHECK_FIELD_SIZE(c,r,e,type) do {\
144 if (sizeof(__typeof__(c.e)) != sizeof(type)) { \
145 torture_fail(tctx, #c "." #e "field is not " #type "\n"); \
147 if (sizeof(__typeof__(r.e)) != sizeof(type)) { \
148 torture_fail(tctx, #r "." #e "field is not " #type "\n"); \
152 #define _CHECK_FIELD_SIZE(c,r,e,type) do {} while(0)
155 #define COMPARE_UINT32(tctx, c, r, e) do {\
156 _CHECK_FIELD_SIZE(c, r, e, uint32_t); \
157 torture_assert_int_equal(tctx, c.e, r.e, "invalid value"); \
160 #define COMPARE_UINT64(tctx, c, r, e) do {\
161 _CHECK_FIELD_SIZE(c, r, e, uint64_t); \
162 torture_assert_int_equal(tctx, c.e, r.e, "invalid value"); \
166 #define COMPARE_NTTIME(tctx, c, r, e) do {\
167 _CHECK_FIELD_SIZE(c, r, e, NTTIME); \
168 torture_assert_int_equal(tctx, c.e, r.e, "invalid value"); \
171 #define COMPARE_STRING_ARRAY(tctx, c,r,e) do {\
173 if (!c.e && !r.e) { \
177 torture_fail(tctx, #r "." #e " field is NULL and " #c "." #e " is not\n"); \
180 torture_fail(tctx, #c "." #e " field is NULL and " #r "." #e " is not\n"); \
182 for (__i=0;c.e[__i] != NULL; __i++) { \
183 torture_assert_str_equal(tctx, c.e[__i], r.e[__i], "invalid value"); \
187 #define CHECK_ALIGN(size, n) do {\
189 torture_warning(tctx, "%d is *NOT* %d byte aligned, should be %d",\
190 size, n, size + n - (size % n));\
194 #define DO_ROUND(size, n) (((size)+((n)-1)) & ~((n)-1))
196 #define CHECK_NEEDED_SIZE_ENUM_LEVEL(fn, info, level, count, needed, align) do { \
197 if (torture_setting_bool(tctx, "spoolss_check_size", false)) {\
198 uint32_t size = ndr_size_##fn##_info(tctx, level, count, info);\
199 uint32_t round_size = DO_ROUND(size, align);\
200 if (round_size != needed) {\
201 torture_warning(tctx, __location__": "#fn" level %d (count: %d) got unexpected needed size: %d, we calculated: %d", level, count, needed, round_size);\
202 CHECK_ALIGN(size, align);\
207 #define CHECK_NEEDED_SIZE_ENUM(fn, info, count, needed, align) do { \
208 if (torture_setting_bool(tctx, "spoolss_check_size", false)) {\
209 uint32_t size = ndr_size_##fn##_info(tctx, count, info);\
210 uint32_t round_size = DO_ROUND(size, align);\
211 if (round_size != needed) {\
212 torture_warning(tctx, __location__": "#fn" (count: %d) got unexpected needed size: %d, we calculated: %d", count, needed, round_size);\
213 CHECK_ALIGN(size, align);\
218 #define CHECK_NEEDED_SIZE_LEVEL(fn, info, level, needed, align) do { \
219 if (torture_setting_bool(tctx, "spoolss_check_size", false)) {\
220 uint32_t size = ndr_size_##fn(info, level, 0);\
221 uint32_t round_size = DO_ROUND(size, align);\
222 if (round_size != needed) {\
223 torture_warning(tctx, __location__": "#fn" level %d got unexpected needed size: %d, we calculated: %d", level, needed, round_size);\
224 CHECK_ALIGN(size, align);\
229 static bool PrinterInfo_to_SetPrinterInfo(struct torture_context *tctx,
230 const union spoolss_PrinterInfo *i,
232 union spoolss_SetPrinterInfo *s)
236 s->info0 = talloc(tctx, struct spoolss_SetPrinterInfo0);
239 s->info2 = talloc(tctx, struct spoolss_SetPrinterInfo2);
240 s->info2->servername = i->info2.servername;
241 s->info2->printername = i->info2.printername;
242 s->info2->sharename = i->info2.sharename;
243 s->info2->portname = i->info2.portname;
244 s->info2->drivername = i->info2.drivername;
245 s->info2->comment = i->info2.comment;
246 s->info2->location = i->info2.location;
247 s->info2->devmode_ptr = NULL;
248 s->info2->sepfile = i->info2.sepfile;
249 s->info2->printprocessor = i->info2.printprocessor;
250 s->info2->datatype = i->info2.datatype;
251 s->info2->parameters = i->info2.parameters;
252 s->info2->secdesc_ptr = NULL;
253 s->info2->attributes = i->info2.attributes;
254 s->info2->priority = i->info2.priority;
255 s->info2->defaultpriority = i->info2.defaultpriority;
256 s->info2->starttime = i->info2.starttime;
257 s->info2->untiltime = i->info2.untiltime;
258 s->info2->status = i->info2.status;
259 s->info2->cjobs = i->info2.cjobs;
260 s->info2->averageppm = i->info2.averageppm;
276 static bool test_OpenPrinter_server(struct torture_context *tctx,
277 struct dcerpc_pipe *p,
278 struct policy_handle *server_handle)
281 struct spoolss_OpenPrinter op;
282 struct dcerpc_binding_handle *b = p->binding_handle;
284 op.in.printername = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
285 op.in.datatype = NULL;
286 op.in.devmode_ctr.devmode= NULL;
287 op.in.access_mask = 0;
288 op.out.handle = server_handle;
290 torture_comment(tctx, "Testing OpenPrinter(%s)\n", op.in.printername);
292 status = dcerpc_spoolss_OpenPrinter_r(b, tctx, &op);
293 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_OpenPrinter failed");
294 torture_assert_werr_ok(tctx, op.out.result, "dcerpc_spoolss_OpenPrinter failed");
299 static bool test_EnumPorts(struct torture_context *tctx,
302 struct test_spoolss_context *ctx =
303 talloc_get_type_abort(private_data, struct test_spoolss_context);
304 struct dcerpc_pipe *p = ctx->spoolss_pipe;
305 struct dcerpc_binding_handle *b = p->binding_handle;
307 struct spoolss_EnumPorts r;
308 uint16_t levels[] = { 1, 2 };
311 for (i=0;i<ARRAY_SIZE(levels);i++) {
312 int level = levels[i];
316 union spoolss_PortInfo *info;
318 r.in.servername = "";
322 r.out.needed = &needed;
323 r.out.count = &count;
326 torture_comment(tctx, "Testing EnumPorts level %u\n", r.in.level);
328 status = dcerpc_spoolss_EnumPorts_r(b, ctx, &r);
329 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPorts failed");
330 if (W_ERROR_IS_OK(r.out.result)) {
331 /* TODO: do some more checks here */
334 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
335 "EnumPorts unexpected return code");
337 blob = data_blob_talloc_zero(ctx, needed);
339 r.in.offered = needed;
341 status = dcerpc_spoolss_EnumPorts_r(b, ctx, &r);
342 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPorts failed");
344 torture_assert_werr_ok(tctx, r.out.result, "EnumPorts failed");
346 torture_assert(tctx, info, "EnumPorts returned no info");
348 CHECK_NEEDED_SIZE_ENUM_LEVEL(spoolss_EnumPorts, info, r.in.level, count, needed, 4);
350 ctx->port_count[level] = count;
351 ctx->ports[level] = info;
354 for (i=1;i<ARRAY_SIZE(levels);i++) {
355 int level = levels[i];
356 int old_level = levels[i-1];
357 torture_assert_int_equal(tctx, ctx->port_count[level], ctx->port_count[old_level],
358 "EnumPorts invalid value");
360 /* if the array sizes are not the same we would maybe segfault in the following code */
362 for (i=0;i<ARRAY_SIZE(levels);i++) {
363 int level = levels[i];
364 for (j=0;j<ctx->port_count[level];j++) {
365 union spoolss_PortInfo *cur = &ctx->ports[level][j];
366 union spoolss_PortInfo *ref = &ctx->ports[2][j];
369 COMPARE_STRING(tctx, cur->info1, ref->info2, port_name);
372 /* level 2 is our reference, and it makes no sense to compare it to itself */
381 static bool test_GetPrintProcessorDirectory(struct torture_context *tctx,
384 struct test_spoolss_context *ctx =
385 talloc_get_type_abort(private_data, struct test_spoolss_context);
388 struct dcerpc_pipe *p = ctx->spoolss_pipe;
389 struct dcerpc_binding_handle *b = p->binding_handle;
390 struct spoolss_GetPrintProcessorDirectory r;
405 .server = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p))
408 .server = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p))
414 for (i=0;i<ARRAY_SIZE(levels);i++) {
415 int level = levels[i].level;
418 r.in.server = levels[i].server;
419 r.in.environment = ctx->environment;
423 r.out.needed = &needed;
425 torture_comment(tctx, "Testing GetPrintProcessorDirectory level %u\n", r.in.level);
427 status = dcerpc_spoolss_GetPrintProcessorDirectory_r(b, tctx, &r);
428 torture_assert_ntstatus_ok(tctx, status,
429 "dcerpc_spoolss_GetPrintProcessorDirectory failed");
430 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
431 "GetPrintProcessorDirectory unexpected return code");
433 blob = data_blob_talloc_zero(tctx, needed);
435 r.in.offered = needed;
437 status = dcerpc_spoolss_GetPrintProcessorDirectory_r(b, tctx, &r);
438 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_GetPrintProcessorDirectory failed");
440 torture_assert_werr_ok(tctx, r.out.result, "GetPrintProcessorDirectory failed");
442 CHECK_NEEDED_SIZE_LEVEL(spoolss_PrintProcessorDirectoryInfo, r.out.info, r.in.level, needed, 2);
449 static bool test_GetPrinterDriverDirectory(struct torture_context *tctx,
452 struct test_spoolss_context *ctx =
453 talloc_get_type_abort(private_data, struct test_spoolss_context);
456 struct dcerpc_pipe *p = ctx->spoolss_pipe;
457 struct dcerpc_binding_handle *b = p->binding_handle;
458 struct spoolss_GetPrinterDriverDirectory r;
473 .server = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p))
476 .server = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p))
482 for (i=0;i<ARRAY_SIZE(levels);i++) {
483 int level = levels[i].level;
486 r.in.server = levels[i].server;
487 r.in.environment = ctx->environment;
491 r.out.needed = &needed;
493 torture_comment(tctx, "Testing GetPrinterDriverDirectory level %u\n", r.in.level);
495 status = dcerpc_spoolss_GetPrinterDriverDirectory_r(b, tctx, &r);
496 torture_assert_ntstatus_ok(tctx, status,
497 "dcerpc_spoolss_GetPrinterDriverDirectory failed");
498 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
499 "GetPrinterDriverDirectory unexpected return code");
501 blob = data_blob_talloc_zero(tctx, needed);
503 r.in.offered = needed;
505 status = dcerpc_spoolss_GetPrinterDriverDirectory_r(b, tctx, &r);
506 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_GetPrinterDriverDirectory failed");
508 torture_assert_werr_ok(tctx, r.out.result, "GetPrinterDriverDirectory failed");
510 CHECK_NEEDED_SIZE_LEVEL(spoolss_DriverDirectoryInfo, r.out.info, r.in.level, needed, 2);
516 static bool test_EnumPrinterDrivers_buffers(struct torture_context *tctx,
517 struct dcerpc_binding_handle *b,
518 const char *server_name,
519 const char *environment,
523 union spoolss_DriverInfo **info_p)
525 struct spoolss_EnumPrinterDrivers r;
528 union spoolss_DriverInfo *info;
532 buffer = data_blob_talloc_zero(tctx, offered);
535 r.in.server = server_name;
536 r.in.environment = environment;
538 r.in.buffer = offered ? &buffer : NULL;
539 r.in.offered = offered;
540 r.out.needed = &needed;
541 r.out.count = &count;
544 torture_comment(tctx, "Testing EnumPrinterDrivers(%s) level %u, offered: %u\n",
545 r.in.environment, r.in.level, r.in.offered);
547 torture_assert_ntstatus_ok(tctx,
548 dcerpc_spoolss_EnumPrinterDrivers_r(b, tctx, &r),
549 "EnumPrinterDrivers failed");
550 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
551 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
553 r.in.offered = needed;
555 torture_assert_ntstatus_ok(tctx,
556 dcerpc_spoolss_EnumPrinterDrivers_r(b, tctx, &r),
557 "EnumPrinterDrivers failed");
560 torture_assert_werr_ok(tctx, r.out.result,
561 "EnumPrinterDrivers failed");
570 CHECK_NEEDED_SIZE_ENUM_LEVEL(spoolss_EnumPrinterDrivers, info, r.in.level, count, needed, 4);
577 static bool test_EnumPrinterDrivers_args(struct torture_context *tctx,
578 struct dcerpc_binding_handle *b,
579 const char *server_name,
580 const char *environment,
583 union spoolss_DriverInfo **info_p)
585 return test_EnumPrinterDrivers_buffers(tctx, b, server_name,
586 environment, level, 0,
590 static bool test_EnumPrinterDrivers_findone(struct torture_context *tctx,
591 struct dcerpc_binding_handle *b,
592 const char *server_name,
593 const char *environment,
595 const char *driver_name,
596 union spoolss_DriverInfo *info_p)
599 union spoolss_DriverInfo *info;
601 const char *environment_ret = NULL;
604 test_EnumPrinterDrivers_args(tctx, b, server_name, environment, level, &count, &info),
605 "failed to enumerate printer drivers");
607 for (i=0; i < count; i++) {
608 const char *driver_name_ret = "";
611 driver_name_ret = info[i].info1.driver_name;
614 driver_name_ret = info[i].info2.driver_name;
615 environment_ret = info[i].info2.architecture;
618 driver_name_ret = info[i].info3.driver_name;
619 environment_ret = info[i].info3.architecture;
622 driver_name_ret = info[i].info4.driver_name;
623 environment_ret = info[i].info4.architecture;
626 driver_name_ret = info[i].info5.driver_name;
627 environment_ret = info[i].info5.architecture;
630 driver_name_ret = info[i].info6.driver_name;
631 environment_ret = info[i].info6.architecture;
634 driver_name_ret = info[i].info7.driver_name;
637 driver_name_ret = info[i].info8.driver_name;
638 environment_ret = info[i].info8.architecture;
643 if (environment_ret) {
644 torture_assert_str_equal(tctx, environment, environment_ret, "architecture mismatch");
646 if (strequal(driver_name, driver_name_ret)) {
657 static bool test_EnumPrinterDrivers(struct torture_context *tctx,
660 struct test_spoolss_context *ctx =
661 talloc_get_type_abort(private_data, struct test_spoolss_context);
662 struct dcerpc_pipe *p = ctx->spoolss_pipe;
663 struct dcerpc_binding_handle *b = p->binding_handle;
664 uint16_t levels[] = { 1, 2, 3, 4, 5, 6, 8 };
665 uint16_t buffer_sizes[] = { 0, 1024, 6040, 0xffff };
668 /* FIXME: gd, come back and fix "" as server, and handle
669 * priority of returned error codes in torture test and samba 3
671 const char *server_name = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
672 const char *environments[2];
674 environments[0] = SPOOLSS_ARCHITECTURE_ALL;
675 environments[1] = ctx->environment;
677 for (a=0;a<ARRAY_SIZE(environments);a++) {
679 for (i=0;i<ARRAY_SIZE(buffer_sizes);i++) {
681 test_EnumPrinterDrivers_buffers(tctx, b, server_name,
685 "failed to enumerate drivers");
688 for (i=0;i<ARRAY_SIZE(levels);i++) {
689 int level = levels[i];
691 union spoolss_DriverInfo *info;
694 test_EnumPrinterDrivers_args(tctx, b, server_name, environments[a], level, &count, &info),
695 "failed to enumerate drivers");
697 ctx->driver_count[level] = count;
698 ctx->drivers[level] = info;
701 for (i=1;i<ARRAY_SIZE(levels);i++) {
702 int level = levels[i];
703 int old_level = levels[i-1];
705 torture_assert_int_equal(tctx, ctx->driver_count[level], ctx->driver_count[old_level],
706 "EnumPrinterDrivers invalid value");
709 for (i=0;i<ARRAY_SIZE(levels);i++) {
710 int level = levels[i];
712 for (j=0;j<ctx->driver_count[level - 1];j++) {
713 union spoolss_DriverInfo *cur = &ctx->drivers[level - 1][j];
714 union spoolss_DriverInfo *ref = &ctx->drivers[7][j];
718 COMPARE_STRING(tctx, cur->info1, ref->info8, driver_name);
721 COMPARE_UINT32(tctx, cur->info2, ref->info8, version);
722 COMPARE_STRING(tctx, cur->info2, ref->info8, driver_name);
723 COMPARE_STRING(tctx, cur->info2, ref->info8, architecture);
724 COMPARE_STRING(tctx, cur->info2, ref->info8, driver_path);
725 COMPARE_STRING(tctx, cur->info2, ref->info8, data_file);
726 COMPARE_STRING(tctx, cur->info2, ref->info8, config_file);
729 COMPARE_UINT32(tctx, cur->info3, ref->info8, version);
730 COMPARE_STRING(tctx, cur->info3, ref->info8, driver_name);
731 COMPARE_STRING(tctx, cur->info3, ref->info8, architecture);
732 COMPARE_STRING(tctx, cur->info3, ref->info8, driver_path);
733 COMPARE_STRING(tctx, cur->info3, ref->info8, data_file);
734 COMPARE_STRING(tctx, cur->info3, ref->info8, config_file);
735 COMPARE_STRING(tctx, cur->info3, ref->info8, help_file);
736 COMPARE_STRING_ARRAY(tctx, cur->info3, ref->info8, dependent_files);
737 COMPARE_STRING(tctx, cur->info3, ref->info8, monitor_name);
738 COMPARE_STRING(tctx, cur->info3, ref->info8, default_datatype);
741 COMPARE_UINT32(tctx, cur->info4, ref->info8, version);
742 COMPARE_STRING(tctx, cur->info4, ref->info8, driver_name);
743 COMPARE_STRING(tctx, cur->info4, ref->info8, architecture);
744 COMPARE_STRING(tctx, cur->info4, ref->info8, driver_path);
745 COMPARE_STRING(tctx, cur->info4, ref->info8, data_file);
746 COMPARE_STRING(tctx, cur->info4, ref->info8, config_file);
747 COMPARE_STRING(tctx, cur->info4, ref->info8, help_file);
748 COMPARE_STRING_ARRAY(tctx, cur->info4, ref->info8, dependent_files);
749 COMPARE_STRING(tctx, cur->info4, ref->info8, monitor_name);
750 COMPARE_STRING(tctx, cur->info4, ref->info8, default_datatype);
751 COMPARE_STRING_ARRAY(tctx, cur->info4, ref->info8, previous_names);
754 COMPARE_UINT32(tctx, cur->info5, ref->info8, version);
755 COMPARE_STRING(tctx, cur->info5, ref->info8, driver_name);
756 COMPARE_STRING(tctx, cur->info5, ref->info8, architecture);
757 COMPARE_STRING(tctx, cur->info5, ref->info8, driver_path);
758 COMPARE_STRING(tctx, cur->info5, ref->info8, data_file);
759 COMPARE_STRING(tctx, cur->info5, ref->info8, config_file);
760 /*COMPARE_UINT32(tctx, cur->info5, ref->info8, driver_attributes);*/
761 /*COMPARE_UINT32(tctx, cur->info5, ref->info8, config_version);*/
762 /*TODO: ! COMPARE_UINT32(tctx, cur->info5, ref->info8, driver_version); */
765 COMPARE_UINT32(tctx, cur->info6, ref->info8, version);
766 COMPARE_STRING(tctx, cur->info6, ref->info8, driver_name);
767 COMPARE_STRING(tctx, cur->info6, ref->info8, architecture);
768 COMPARE_STRING(tctx, cur->info6, ref->info8, driver_path);
769 COMPARE_STRING(tctx, cur->info6, ref->info8, data_file);
770 COMPARE_STRING(tctx, cur->info6, ref->info8, config_file);
771 COMPARE_STRING(tctx, cur->info6, ref->info8, help_file);
772 COMPARE_STRING_ARRAY(tctx, cur->info6, ref->info8, dependent_files);
773 COMPARE_STRING(tctx, cur->info6, ref->info8, monitor_name);
774 COMPARE_STRING(tctx, cur->info6, ref->info8, default_datatype);
775 COMPARE_STRING_ARRAY(tctx, cur->info6, ref->info8, previous_names);
776 COMPARE_NTTIME(tctx, cur->info6, ref->info8, driver_date);
777 COMPARE_UINT64(tctx, cur->info6, ref->info8, driver_version);
778 COMPARE_STRING(tctx, cur->info6, ref->info8, manufacturer_name);
779 COMPARE_STRING(tctx, cur->info6, ref->info8, manufacturer_url);
780 COMPARE_STRING(tctx, cur->info6, ref->info8, hardware_id);
781 COMPARE_STRING(tctx, cur->info6, ref->info8, provider);
784 /* level 8 is our reference, and it makes no sense to compare it to itself */
794 static bool test_EnumMonitors(struct torture_context *tctx,
797 struct test_spoolss_context *ctx =
798 talloc_get_type_abort(private_data, struct test_spoolss_context);
799 struct dcerpc_pipe *p = ctx->spoolss_pipe;
800 struct dcerpc_binding_handle *b = p->binding_handle;
802 struct spoolss_EnumMonitors r;
803 uint16_t levels[] = { 1, 2 };
806 for (i=0;i<ARRAY_SIZE(levels);i++) {
807 int level = levels[i];
811 union spoolss_MonitorInfo *info;
813 r.in.servername = "";
817 r.out.needed = &needed;
818 r.out.count = &count;
821 torture_comment(tctx, "Testing EnumMonitors level %u\n", r.in.level);
823 status = dcerpc_spoolss_EnumMonitors_r(b, ctx, &r);
824 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumMonitors failed");
825 if (W_ERROR_IS_OK(r.out.result)) {
826 /* TODO: do some more checks here */
829 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
830 "EnumMonitors failed");
832 blob = data_blob_talloc_zero(ctx, needed);
834 r.in.offered = needed;
836 status = dcerpc_spoolss_EnumMonitors_r(b, ctx, &r);
837 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumMonitors failed");
839 torture_assert_werr_ok(tctx, r.out.result, "EnumMonitors failed");
841 CHECK_NEEDED_SIZE_ENUM_LEVEL(spoolss_EnumMonitors, info, r.in.level, count, needed, 4);
843 ctx->monitor_count[level] = count;
844 ctx->monitors[level] = info;
847 for (i=1;i<ARRAY_SIZE(levels);i++) {
848 int level = levels[i];
849 int old_level = levels[i-1];
850 torture_assert_int_equal(tctx, ctx->monitor_count[level], ctx->monitor_count[old_level],
851 "EnumMonitors invalid value");
854 for (i=0;i<ARRAY_SIZE(levels);i++) {
855 int level = levels[i];
856 for (j=0;j<ctx->monitor_count[level];j++) {
857 union spoolss_MonitorInfo *cur = &ctx->monitors[level][j];
858 union spoolss_MonitorInfo *ref = &ctx->monitors[2][j];
861 COMPARE_STRING(tctx, cur->info1, ref->info2, monitor_name);
864 /* level 2 is our reference, and it makes no sense to compare it to itself */
873 static bool test_EnumPrintProcessors_level(struct torture_context *tctx,
874 struct dcerpc_binding_handle *b,
875 const char *environment,
878 union spoolss_PrintProcessorInfo **info_p,
879 WERROR expected_result)
881 struct spoolss_EnumPrintProcessors r;
885 union spoolss_PrintProcessorInfo *info;
887 r.in.servername = "";
888 r.in.environment = environment;
892 r.out.needed = &needed;
893 r.out.count = &count;
896 torture_comment(tctx, "Testing EnumPrintProcessors(%s) level %u\n",
897 r.in.environment, r.in.level);
899 torture_assert_ntstatus_ok(tctx,
900 dcerpc_spoolss_EnumPrintProcessors_r(b, tctx, &r),
901 "EnumPrintProcessors failed");
902 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
903 blob = data_blob_talloc_zero(tctx, needed);
905 r.in.offered = needed;
906 torture_assert_ntstatus_ok(tctx,
907 dcerpc_spoolss_EnumPrintProcessors_r(b, tctx, &r),
908 "EnumPrintProcessors failed");
910 torture_assert_werr_equal(tctx, r.out.result, expected_result,
911 "EnumPrintProcessors failed");
913 CHECK_NEEDED_SIZE_ENUM_LEVEL(spoolss_EnumPrintProcessors, info, level, count, needed, 4);
925 static bool test_EnumPrintProcessors(struct torture_context *tctx,
928 struct test_spoolss_context *ctx =
929 talloc_get_type_abort(private_data, struct test_spoolss_context);
931 uint16_t levels[] = {0, 1, 2, 3, 32, 256 };
932 uint16_t ok[] = {0, 1, 0, 0, 0, 0 };
934 struct dcerpc_pipe *p = ctx->spoolss_pipe;
935 struct dcerpc_binding_handle *b = p->binding_handle;
938 test_EnumPrintProcessors_level(tctx, b, "phantasy", 1, NULL, NULL, WERR_INVALID_ENVIRONMENT),
939 "test_EnumPrintProcessors_level failed");
941 for (i=0;i<ARRAY_SIZE(levels);i++) {
942 union spoolss_PrintProcessorInfo *info;
944 WERROR expected_result = ok[i] ? WERR_OK : WERR_INVALID_LEVEL;
947 test_EnumPrintProcessors_level(tctx, b, ctx->environment, levels[i], &count, &info, expected_result),
948 "test_EnumPrintProcessors_level failed");
954 static bool test_EnumPrintProcessorDataTypes_level(struct torture_context *tctx,
955 struct dcerpc_binding_handle *b,
956 const char *print_processor_name,
959 union spoolss_PrintProcDataTypesInfo **info_p,
960 WERROR expected_result)
962 struct spoolss_EnumPrintProcessorDataTypes r;
966 union spoolss_PrintProcDataTypesInfo *info;
968 r.in.servername = "";
969 r.in.print_processor_name = print_processor_name;
973 r.out.needed = &needed;
974 r.out.count = &count;
977 torture_comment(tctx, "Testing EnumPrintProcessorDataTypes(%s) level %u\n",
978 r.in.print_processor_name, r.in.level);
980 torture_assert_ntstatus_ok(tctx,
981 dcerpc_spoolss_EnumPrintProcessorDataTypes_r(b, tctx, &r),
982 "EnumPrintProcessorDataTypes failed");
983 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
984 blob = data_blob_talloc_zero(tctx, needed);
986 r.in.offered = needed;
987 torture_assert_ntstatus_ok(tctx,
988 dcerpc_spoolss_EnumPrintProcessorDataTypes_r(b, tctx, &r),
989 "EnumPrintProcessorDataTypes failed");
991 torture_assert_werr_equal(tctx, r.out.result, expected_result,
992 "EnumPrintProcessorDataTypes failed");
994 CHECK_NEEDED_SIZE_ENUM_LEVEL(spoolss_EnumPrintProcessorDataTypes, info, level, count, needed, 4);
1006 static bool test_EnumPrintProcessorDataTypes(struct torture_context *tctx,
1009 struct test_spoolss_context *ctx =
1010 talloc_get_type_abort(private_data, struct test_spoolss_context);
1012 uint16_t levels[] = {0, 1, 2, 3, 32, 256 };
1013 uint16_t ok[] = {0, 1, 0, 0, 0, 0 };
1015 struct dcerpc_pipe *p = ctx->spoolss_pipe;
1016 struct dcerpc_binding_handle *b = p->binding_handle;
1018 torture_assert(tctx,
1019 test_EnumPrintProcessorDataTypes_level(tctx, b, NULL, 1, NULL, NULL, WERR_UNKNOWN_PRINTPROCESSOR),
1020 "test_EnumPrintProcessorDataTypes_level failed");
1022 torture_assert(tctx,
1023 test_EnumPrintProcessorDataTypes_level(tctx, b, "nonexisting", 1, NULL, NULL, WERR_UNKNOWN_PRINTPROCESSOR),
1024 "test_EnumPrintProcessorDataTypes_level failed");
1026 for (i=0;i<ARRAY_SIZE(levels);i++) {
1027 int level = levels[i];
1029 union spoolss_PrintProcDataTypesInfo *info;
1030 WERROR expected_result = ok[i] ? WERR_OK : WERR_INVALID_LEVEL;
1032 torture_assert(tctx,
1033 test_EnumPrintProcessorDataTypes_level(tctx, b, "winprint", level, &count, &info, expected_result),
1034 "test_EnumPrintProcessorDataTypes_level failed");
1038 union spoolss_PrintProcessorInfo *info;
1041 torture_assert(tctx,
1042 test_EnumPrintProcessors_level(tctx, b, ctx->environment, 1, &count, &info, WERR_OK),
1043 "test_EnumPrintProcessors_level failed");
1045 for (i=0; i < count; i++) {
1046 torture_assert(tctx,
1047 test_EnumPrintProcessorDataTypes_level(tctx, b, info[i].info1.print_processor_name, 1, NULL, NULL, WERR_OK),
1048 "test_EnumPrintProcessorDataTypes_level failed");
1056 static bool test_EnumPrinters(struct torture_context *tctx,
1059 struct test_spoolss_context *ctx =
1060 talloc_get_type_abort(private_data, struct test_spoolss_context);
1061 struct dcerpc_pipe *p = ctx->spoolss_pipe;
1062 struct dcerpc_binding_handle *b = p->binding_handle;
1063 struct spoolss_EnumPrinters r;
1065 uint16_t levels[] = { 0, 1, 2, 4, 5 };
1068 for (i=0;i<ARRAY_SIZE(levels);i++) {
1069 int level = levels[i];
1073 union spoolss_PrinterInfo *info;
1075 r.in.flags = PRINTER_ENUM_LOCAL;
1080 r.out.needed = &needed;
1081 r.out.count = &count;
1084 torture_comment(tctx, "Testing EnumPrinters level %u\n", r.in.level);
1086 status = dcerpc_spoolss_EnumPrinters_r(b, ctx, &r);
1087 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPrinters failed");
1088 if (W_ERROR_IS_OK(r.out.result)) {
1089 /* TODO: do some more checks here */
1092 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
1093 "EnumPrinters unexpected return code");
1095 blob = data_blob_talloc_zero(ctx, needed);
1096 r.in.buffer = &blob;
1097 r.in.offered = needed;
1099 status = dcerpc_spoolss_EnumPrinters_r(b, ctx, &r);
1100 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPrinters failed");
1102 torture_assert_werr_ok(tctx, r.out.result, "EnumPrinters failed");
1104 CHECK_NEEDED_SIZE_ENUM_LEVEL(spoolss_EnumPrinters, info, r.in.level, count, needed, 4);
1106 ctx->printer_count[level] = count;
1107 ctx->printers[level] = info;
1110 for (i=1;i<ARRAY_SIZE(levels);i++) {
1111 int level = levels[i];
1112 int old_level = levels[i-1];
1113 torture_assert_int_equal(tctx, ctx->printer_count[level], ctx->printer_count[old_level],
1114 "EnumPrinters invalid value");
1117 for (i=0;i<ARRAY_SIZE(levels);i++) {
1118 int level = levels[i];
1119 for (j=0;j<ctx->printer_count[level];j++) {
1120 union spoolss_PrinterInfo *cur = &ctx->printers[level][j];
1121 union spoolss_PrinterInfo *ref = &ctx->printers[2][j];
1124 COMPARE_STRING(tctx, cur->info0, ref->info2, printername);
1125 COMPARE_STRING(tctx, cur->info0, ref->info2, servername);
1126 COMPARE_UINT32(tctx, cur->info0, ref->info2, cjobs);
1127 /*COMPARE_UINT32(tctx, cur->info0, ref->info2, total_jobs);
1128 COMPARE_UINT32(tctx, cur->info0, ref->info2, total_bytes);
1129 COMPARE_SPOOLSS_TIME(cur->info0, ref->info2, spoolss_Time time);
1130 COMPARE_UINT32(tctx, cur->info0, ref->info2, global_counter);
1131 COMPARE_UINT32(tctx, cur->info0, ref->info2, total_pages);
1132 COMPARE_UINT32(tctx, cur->info0, ref->info2, version);
1133 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown10);
1134 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown11);
1135 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown12);
1136 COMPARE_UINT32(tctx, cur->info0, ref->info2, session_counter);
1137 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown14);
1138 COMPARE_UINT32(tctx, cur->info0, ref->info2, printer_errors);
1139 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown16);
1140 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown17);
1141 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown18);
1142 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown19);
1143 COMPARE_UINT32(tctx, cur->info0, ref->info2, change_id);
1144 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown21);*/
1145 COMPARE_UINT32(tctx, cur->info0, ref->info2, status);
1146 /*COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown23);
1147 COMPARE_UINT32(tctx, cur->info0, ref->info2, c_setprinter);
1148 COMPARE_UINT16(cur->info0, ref->info2, unknown25);
1149 COMPARE_UINT16(cur->info0, ref->info2, unknown26);
1150 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown27);
1151 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown28);
1152 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown29);*/
1155 /*COMPARE_UINT32(tctx, cur->info1, ref->info2, flags);*/
1156 /*COMPARE_STRING(tctx, cur->info1, ref->info2, name);*/
1157 /*COMPARE_STRING(tctx, cur->info1, ref->info2, description);*/
1158 COMPARE_STRING(tctx, cur->info1, ref->info2, comment);
1161 /* level 2 is our reference, and it makes no sense to compare it to itself */
1164 COMPARE_STRING(tctx, cur->info4, ref->info2, printername);
1165 COMPARE_STRING(tctx, cur->info4, ref->info2, servername);
1166 COMPARE_UINT32(tctx, cur->info4, ref->info2, attributes);
1169 COMPARE_STRING(tctx, cur->info5, ref->info2, printername);
1170 COMPARE_STRING(tctx, cur->info5, ref->info2, portname);
1171 COMPARE_UINT32(tctx, cur->info5, ref->info2, attributes);
1172 /*COMPARE_UINT32(tctx, cur->info5, ref->info2, device_not_selected_timeout);
1173 COMPARE_UINT32(tctx, cur->info5, ref->info2, transmission_retry_timeout);*/
1180 * - verify that the port of a printer was in the list returned by EnumPorts
1186 static bool test_GetPrinterDriver2(struct torture_context *tctx,
1187 struct dcerpc_binding_handle *b,
1188 struct policy_handle *handle,
1189 const char *driver_name,
1190 const char *environment);
1192 bool test_GetPrinter_level(struct torture_context *tctx,
1193 struct dcerpc_binding_handle *b,
1194 struct policy_handle *handle,
1196 union spoolss_PrinterInfo *info)
1198 struct spoolss_GetPrinter r;
1201 r.in.handle = handle;
1205 r.out.needed = &needed;
1207 torture_comment(tctx, "Testing GetPrinter level %u\n", r.in.level);
1209 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_GetPrinter_r(b, tctx, &r),
1210 "GetPrinter failed");
1212 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
1213 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
1214 r.in.buffer = &blob;
1215 r.in.offered = needed;
1217 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_GetPrinter_r(b, tctx, &r),
1218 "GetPrinter failed");
1221 torture_assert_werr_ok(tctx, r.out.result, "GetPrinter failed");
1223 CHECK_NEEDED_SIZE_LEVEL(spoolss_PrinterInfo, r.out.info, r.in.level, needed, 4);
1225 if (info && r.out.info) {
1226 *info = *r.out.info;
1233 static bool test_GetPrinter(struct torture_context *tctx,
1234 struct dcerpc_binding_handle *b,
1235 struct policy_handle *handle,
1236 const char *environment)
1238 uint32_t levels[] = {0, 1, 2, 3, 4, 5, 6, 7, 8};
1241 for (i=0;i<ARRAY_SIZE(levels);i++) {
1243 union spoolss_PrinterInfo info;
1247 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, levels[i], &info),
1248 "failed to call GetPrinter");
1250 if ((levels[i] == 2) && info.info2.drivername && strlen(info.info2.drivername)) {
1251 torture_assert(tctx,
1252 test_GetPrinterDriver2(tctx, b, handle, info.info2.drivername, environment),
1253 "failed to call test_GetPrinterDriver2");
1260 static bool test_SetPrinter(struct torture_context *tctx,
1261 struct dcerpc_binding_handle *b,
1262 struct policy_handle *handle,
1263 struct spoolss_SetPrinterInfoCtr *info_ctr,
1264 struct spoolss_DevmodeContainer *devmode_ctr,
1265 struct sec_desc_buf *secdesc_ctr,
1266 enum spoolss_PrinterControl command)
1268 struct spoolss_SetPrinter r;
1270 r.in.handle = handle;
1271 r.in.info_ctr = info_ctr;
1272 r.in.devmode_ctr = devmode_ctr;
1273 r.in.secdesc_ctr = secdesc_ctr;
1274 r.in.command = command;
1276 torture_comment(tctx, "Testing SetPrinter level %d\n", r.in.info_ctr->level);
1278 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_SetPrinter_r(b, tctx, &r),
1279 "failed to call SetPrinter");
1280 torture_assert(tctx, (W_ERROR_EQUAL(r.out.result, WERR_OK)
1281 || W_ERROR_EQUAL(r.out.result, WERR_IO_PENDING)),
1282 "SetPrinter failed");
1287 static bool test_SetPrinter_errors(struct torture_context *tctx,
1288 struct dcerpc_binding_handle *b,
1289 struct policy_handle *handle)
1291 struct spoolss_SetPrinter r;
1292 uint16_t levels[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
1295 struct spoolss_SetPrinterInfoCtr info_ctr;
1296 struct spoolss_DevmodeContainer devmode_ctr;
1297 struct sec_desc_buf secdesc_ctr;
1300 info_ctr.info.info0 = NULL;
1302 ZERO_STRUCT(devmode_ctr);
1303 ZERO_STRUCT(secdesc_ctr);
1305 r.in.handle = handle;
1306 r.in.info_ctr = &info_ctr;
1307 r.in.devmode_ctr = &devmode_ctr;
1308 r.in.secdesc_ctr = &secdesc_ctr;
1311 torture_comment(tctx, "Testing SetPrinter all zero\n");
1313 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_SetPrinter_r(b, tctx, &r),
1314 "failed to call SetPrinter");
1315 torture_assert_werr_equal(tctx, r.out.result, WERR_INVALID_PARAM,
1316 "failed to call SetPrinter");
1319 for (i=0; i < ARRAY_SIZE(levels); i++) {
1321 struct spoolss_SetPrinterInfo0 info0;
1322 struct spoolss_SetPrinterInfo1 info1;
1323 struct spoolss_SetPrinterInfo2 info2;
1324 struct spoolss_SetPrinterInfo3 info3;
1325 struct spoolss_SetPrinterInfo4 info4;
1326 struct spoolss_SetPrinterInfo5 info5;
1327 struct spoolss_SetPrinterInfo6 info6;
1328 struct spoolss_SetPrinterInfo7 info7;
1329 struct spoolss_SetPrinterInfo8 info8;
1330 struct spoolss_SetPrinterInfo9 info9;
1333 info_ctr.level = levels[i];
1334 switch (levels[i]) {
1337 info_ctr.info.info0 = &info0;
1341 info_ctr.info.info1 = &info1;
1345 info_ctr.info.info2 = &info2;
1349 info_ctr.info.info3 = &info3;
1353 info_ctr.info.info4 = &info4;
1357 info_ctr.info.info5 = &info5;
1361 info_ctr.info.info6 = &info6;
1365 info_ctr.info.info7 = &info7;
1369 info_ctr.info.info8 = &info8;
1373 info_ctr.info.info9 = &info9;
1377 torture_comment(tctx, "Testing SetPrinter level %d, command %d\n",
1378 info_ctr.level, r.in.command);
1380 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_SetPrinter_r(b, tctx, &r),
1381 "failed to call SetPrinter");
1383 switch (r.in.command) {
1384 case SPOOLSS_PRINTER_CONTROL_UNPAUSE: /* 0 */
1385 /* is ignored for all levels other then 0 */
1386 if (info_ctr.level > 0) {
1390 case SPOOLSS_PRINTER_CONTROL_PAUSE: /* 1 */
1391 case SPOOLSS_PRINTER_CONTROL_RESUME: /* 2 */
1392 case SPOOLSS_PRINTER_CONTROL_PURGE: /* 3 */
1393 if (info_ctr.level > 0) {
1394 /* is invalid for all levels other then 0 */
1395 torture_assert_werr_equal(tctx, r.out.result, WERR_INVALID_PRINTER_COMMAND,
1396 "unexpected error code returned");
1399 torture_assert_werr_ok(tctx, r.out.result,
1400 "failed to call SetPrinter with non 0 command");
1405 case SPOOLSS_PRINTER_CONTROL_SET_STATUS: /* 4 */
1406 /* FIXME: gd needs further investigation */
1408 torture_assert_werr_equal(tctx, r.out.result, WERR_INVALID_PRINTER_COMMAND,
1409 "unexpected error code returned");
1413 switch (info_ctr.level) {
1415 torture_assert_werr_equal(tctx, r.out.result, WERR_UNKNOWN_LEVEL,
1416 "unexpected error code returned");
1419 torture_assert_werr_equal(tctx, r.out.result, WERR_UNKNOWN_PRINTER_DRIVER,
1420 "unexpected error code returned");
1426 torture_assert_werr_equal(tctx, r.out.result, WERR_INVALID_PARAM,
1427 "unexpected error code returned");
1430 torture_assert_werr_equal(tctx, r.out.result, WERR_NOT_SUPPORTED,
1431 "unexpected error code returned");
1434 torture_assert_werr_ok(tctx, r.out.result,
1435 "failed to call SetPrinter");
1440 if (r.in.command < 5) {
1448 static void clear_info2(struct spoolss_SetPrinterInfoCtr *r)
1450 if ((r->level == 2) && (r->info.info2)) {
1451 r->info.info2->secdesc_ptr = NULL;
1452 r->info.info2->devmode_ptr = NULL;
1456 static bool test_PrinterInfo(struct torture_context *tctx,
1457 struct dcerpc_binding_handle *b,
1458 struct policy_handle *handle)
1461 struct spoolss_SetPrinter s;
1462 struct spoolss_GetPrinter q;
1463 struct spoolss_GetPrinter q0;
1464 struct spoolss_SetPrinterInfoCtr info_ctr;
1465 union spoolss_PrinterInfo info;
1466 struct spoolss_DevmodeContainer devmode_ctr;
1467 struct sec_desc_buf secdesc_ctr;
1472 uint32_t status_list[] = {
1473 /* these do not stick
1474 PRINTER_STATUS_PAUSED,
1475 PRINTER_STATUS_ERROR,
1476 PRINTER_STATUS_PENDING_DELETION, */
1477 PRINTER_STATUS_PAPER_JAM,
1478 PRINTER_STATUS_PAPER_OUT,
1479 PRINTER_STATUS_MANUAL_FEED,
1480 PRINTER_STATUS_PAPER_PROBLEM,
1481 PRINTER_STATUS_OFFLINE,
1482 PRINTER_STATUS_IO_ACTIVE,
1483 PRINTER_STATUS_BUSY,
1484 PRINTER_STATUS_PRINTING,
1485 PRINTER_STATUS_OUTPUT_BIN_FULL,
1486 PRINTER_STATUS_NOT_AVAILABLE,
1487 PRINTER_STATUS_WAITING,
1488 PRINTER_STATUS_PROCESSING,
1489 PRINTER_STATUS_INITIALIZING,
1490 PRINTER_STATUS_WARMING_UP,
1491 PRINTER_STATUS_TONER_LOW,
1492 PRINTER_STATUS_NO_TONER,
1493 PRINTER_STATUS_PAGE_PUNT,
1494 PRINTER_STATUS_USER_INTERVENTION,
1495 PRINTER_STATUS_OUT_OF_MEMORY,
1496 PRINTER_STATUS_DOOR_OPEN,
1497 PRINTER_STATUS_SERVER_UNKNOWN,
1498 PRINTER_STATUS_POWER_SAVE,
1499 /* these do not stick
1508 uint32_t default_attribute = PRINTER_ATTRIBUTE_LOCAL;
1509 uint32_t attribute_list[] = {
1510 PRINTER_ATTRIBUTE_QUEUED,
1511 /* fails with WERR_INVALID_DATATYPE:
1512 PRINTER_ATTRIBUTE_DIRECT, */
1514 PRINTER_ATTRIBUTE_DEFAULT, */
1515 PRINTER_ATTRIBUTE_SHARED,
1517 PRINTER_ATTRIBUTE_NETWORK, */
1518 PRINTER_ATTRIBUTE_HIDDEN,
1519 PRINTER_ATTRIBUTE_LOCAL,
1520 PRINTER_ATTRIBUTE_ENABLE_DEVQ,
1521 PRINTER_ATTRIBUTE_KEEPPRINTEDJOBS,
1522 PRINTER_ATTRIBUTE_DO_COMPLETE_FIRST,
1523 PRINTER_ATTRIBUTE_WORK_OFFLINE,
1525 PRINTER_ATTRIBUTE_ENABLE_BIDI, */
1526 /* fails with WERR_INVALID_DATATYPE:
1527 PRINTER_ATTRIBUTE_RAW_ONLY, */
1528 /* these do not stick
1529 PRINTER_ATTRIBUTE_PUBLISHED,
1530 PRINTER_ATTRIBUTE_FAX,
1531 PRINTER_ATTRIBUTE_TS,
1550 torture_skip(tctx, "Printer Info test is currently broken, skipping");
1553 ZERO_STRUCT(devmode_ctr);
1554 ZERO_STRUCT(secdesc_ctr);
1556 s.in.handle = handle;
1558 s.in.info_ctr = &info_ctr;
1559 s.in.devmode_ctr = &devmode_ctr;
1560 s.in.secdesc_ctr = &secdesc_ctr;
1562 q.in.handle = handle;
1566 #define TESTGETCALL(call, r) \
1567 r.in.buffer = NULL; \
1569 r.out.needed = &needed; \
1570 status = dcerpc_spoolss_ ##call## _r(b, tctx, &r); \
1571 if (!NT_STATUS_IS_OK(status)) { \
1572 torture_comment(tctx, #call " level %u failed - %s (%s)\n", \
1573 r.in.level, nt_errstr(status), __location__); \
1577 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {\
1578 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed); \
1579 r.in.buffer = &blob; \
1580 r.in.offered = needed; \
1582 status = dcerpc_spoolss_ ##call## _r(b, tctx, &r); \
1583 if (!NT_STATUS_IS_OK(status)) { \
1584 torture_comment(tctx, #call " level %u failed - %s (%s)\n", \
1585 r.in.level, nt_errstr(status), __location__); \
1589 if (!W_ERROR_IS_OK(r.out.result)) { \
1590 torture_comment(tctx, #call " level %u failed - %s (%s)\n", \
1591 r.in.level, win_errstr(r.out.result), __location__); \
1597 #define TESTSETCALL_EXP(call, r, err) \
1598 clear_info2(&info_ctr);\
1599 status = dcerpc_spoolss_ ##call## _r(b, tctx, &r); \
1600 if (!NT_STATUS_IS_OK(status)) { \
1601 torture_comment(tctx, #call " level %u failed - %s (%s)\n", \
1602 r.in.info_ctr->level, nt_errstr(status), __location__); \
1606 if (!W_ERROR_IS_OK(err)) { \
1607 if (!W_ERROR_EQUAL(err, r.out.result)) { \
1608 torture_comment(tctx, #call " level %u failed - %s, expected %s (%s)\n", \
1609 r.in.info_ctr->level, win_errstr(r.out.result), win_errstr(err), __location__); \
1614 if (!W_ERROR_IS_OK(r.out.result)) { \
1615 torture_comment(tctx, #call " level %u failed - %s (%s)\n", \
1616 r.in.info_ctr->level, win_errstr(r.out.result), __location__); \
1621 #define TESTSETCALL(call, r) \
1622 TESTSETCALL_EXP(call, r, WERR_OK)
1624 #define STRING_EQUAL(s1, s2, field) \
1625 if ((s1 && !s2) || (s2 && !s1) || strcmp(s1, s2)) { \
1626 torture_comment(tctx, "Failed to set %s to '%s' (%s)\n", \
1627 #field, s2, __location__); \
1632 #define MEM_EQUAL(s1, s2, length, field) \
1633 if ((s1 && !s2) || (s2 && !s1) || memcmp(s1, s2, length)) { \
1634 torture_comment(tctx, "Failed to set %s to '%s' (%s)\n", \
1635 #field, (const char *)s2, __location__); \
1640 #define INT_EQUAL(i1, i2, field) \
1642 torture_comment(tctx, "Failed to set %s to 0x%llx - got 0x%llx (%s)\n", \
1643 #field, (unsigned long long)i2, (unsigned long long)i1, __location__); \
1648 #define SD_EQUAL(sd1, sd2, field) \
1649 if (!security_descriptor_equal(sd1, sd2)) { \
1650 torture_comment(tctx, "Failed to set %s (%s)\n", \
1651 #field, __location__); \
1656 #define TEST_PRINTERINFO_STRING_EXP_ERR(lvl1, field1, lvl2, field2, value, err) do { \
1658 torture_comment(tctx, "field test %d/%s vs %d/%s\n", lvl1, #field1, lvl2, #field2); \
1659 q.in.level = lvl1; \
1660 TESTGETCALL(GetPrinter, q) \
1661 info_ctr.level = lvl1; \
1662 p = (void *)&q.out.info->info ## lvl1; \
1663 info_ctr.info.info ## lvl1 = (struct spoolss_SetPrinterInfo ## lvl1 *)p; \
1664 info_ctr.info.info ## lvl1->field1 = value;\
1665 TESTSETCALL_EXP(SetPrinter, s, err) \
1666 info_ctr.info.info ## lvl1->field1 = ""; \
1667 TESTGETCALL(GetPrinter, q) \
1668 info_ctr.info.info ## lvl1->field1 = value; \
1669 STRING_EQUAL(info_ctr.info.info ## lvl1->field1, value, field1); \
1670 q.in.level = lvl2; \
1671 TESTGETCALL(GetPrinter, q) \
1672 p = (void *)&q.out.info->info ## lvl2; \
1673 info_ctr.info.info ## lvl2 = (struct spoolss_SetPrinterInfo ## lvl2 *)p; \
1674 STRING_EQUAL(info_ctr.info.info ## lvl2->field2, value, field2); \
1677 #define TEST_PRINTERINFO_STRING(lvl1, field1, lvl2, field2, value) do { \
1678 TEST_PRINTERINFO_STRING_EXP_ERR(lvl1, field1, lvl2, field2, value, WERR_OK); \
1681 #define TEST_PRINTERINFO_INT_EXP(lvl1, field1, lvl2, field2, value, exp_value) do { \
1683 torture_comment(tctx, "field test %d/%s vs %d/%s\n", lvl1, #field1, lvl2, #field2); \
1684 q.in.level = lvl1; \
1685 TESTGETCALL(GetPrinter, q) \
1686 info_ctr.level = lvl1; \
1687 p = (void *)&q.out.info->info ## lvl1; \
1688 info_ctr.info.info ## lvl1 = (struct spoolss_SetPrinterInfo ## lvl1 *)p; \
1689 info_ctr.info.info ## lvl1->field1 = value; \
1690 TESTSETCALL(SetPrinter, s) \
1691 info_ctr.info.info ## lvl1->field1 = 0; \
1692 TESTGETCALL(GetPrinter, q) \
1693 p = (void *)&q.out.info->info ## lvl1; \
1694 info_ctr.info.info ## lvl1 = (struct spoolss_SetPrinterInfo ## lvl1 *)p; \
1695 INT_EQUAL(info_ctr.info.info ## lvl1->field1, exp_value, field1); \
1696 q.in.level = lvl2; \
1697 TESTGETCALL(GetPrinter, q) \
1698 p = (void *)&q.out.info->info ## lvl2; \
1699 info_ctr.info.info ## lvl2 = (struct spoolss_SetPrinterInfo ## lvl2 *)p; \
1700 INT_EQUAL(info_ctr.info.info ## lvl2->field2, exp_value, field1); \
1703 #define TEST_PRINTERINFO_INT(lvl1, field1, lvl2, field2, value) do { \
1704 TEST_PRINTERINFO_INT_EXP(lvl1, field1, lvl2, field2, value, value); \
1708 do { TESTGETCALL(GetPrinter, q0) } while (0);
1710 TEST_PRINTERINFO_STRING(2, comment, 1, comment, "xx2-1 comment");
1711 TEST_PRINTERINFO_STRING(2, comment, 2, comment, "xx2-2 comment");
1713 /* level 0 printername does not stick */
1714 /* TEST_PRINTERINFO_STRING(2, printername, 0, printername, "xx2-0 printer"); */
1715 TEST_PRINTERINFO_STRING(2, printername, 1, name, "xx2-1 printer");
1716 TEST_PRINTERINFO_STRING(2, printername, 2, printername, "xx2-2 printer");
1717 TEST_PRINTERINFO_STRING(2, printername, 4, printername, "xx2-4 printer");
1718 TEST_PRINTERINFO_STRING(2, printername, 5, printername, "xx2-5 printer");
1719 /* TEST_PRINTERINFO_STRING(4, printername, 0, printername, "xx4-0 printer"); */
1720 TEST_PRINTERINFO_STRING(4, printername, 1, name, "xx4-1 printer");
1721 TEST_PRINTERINFO_STRING(4, printername, 2, printername, "xx4-2 printer");
1722 TEST_PRINTERINFO_STRING(4, printername, 4, printername, "xx4-4 printer");
1723 TEST_PRINTERINFO_STRING(4, printername, 5, printername, "xx4-5 printer");
1724 /* TEST_PRINTERINFO_STRING(5, printername, 0, printername, "xx5-0 printer"); */
1725 TEST_PRINTERINFO_STRING(5, printername, 1, name, "xx5-1 printer");
1726 TEST_PRINTERINFO_STRING(5, printername, 2, printername, "xx5-2 printer");
1727 TEST_PRINTERINFO_STRING(5, printername, 4, printername, "xx5-4 printer");
1728 TEST_PRINTERINFO_STRING(5, printername, 5, printername, "xx5-5 printer");
1730 /* servername can be set but does not stick
1731 TEST_PRINTERINFO_STRING(2, servername, 0, servername, "xx2-0 servername");
1732 TEST_PRINTERINFO_STRING(2, servername, 2, servername, "xx2-2 servername");
1733 TEST_PRINTERINFO_STRING(2, servername, 4, servername, "xx2-4 servername");
1736 /* passing an invalid port will result in WERR_UNKNOWN_PORT */
1737 TEST_PRINTERINFO_STRING_EXP_ERR(2, portname, 2, portname, "xx2-2 portname", WERR_UNKNOWN_PORT);
1738 TEST_PRINTERINFO_STRING_EXP_ERR(2, portname, 5, portname, "xx2-5 portname", WERR_UNKNOWN_PORT);
1739 TEST_PRINTERINFO_STRING_EXP_ERR(5, portname, 2, portname, "xx5-2 portname", WERR_UNKNOWN_PORT);
1740 TEST_PRINTERINFO_STRING_EXP_ERR(5, portname, 5, portname, "xx5-5 portname", WERR_UNKNOWN_PORT);
1742 TEST_PRINTERINFO_STRING(2, sharename, 2, sharename, "xx2-2 sharename");
1743 /* passing an invalid driver will result in WERR_UNKNOWN_PRINTER_DRIVER */
1744 TEST_PRINTERINFO_STRING_EXP_ERR(2, drivername, 2, drivername, "xx2-2 drivername", WERR_UNKNOWN_PRINTER_DRIVER);
1745 TEST_PRINTERINFO_STRING(2, location, 2, location, "xx2-2 location");
1746 /* passing an invalid sepfile will result in WERR_INVALID_SEPARATOR_FILE */
1747 TEST_PRINTERINFO_STRING_EXP_ERR(2, sepfile, 2, sepfile, "xx2-2 sepfile", WERR_INVALID_SEPARATOR_FILE);
1748 /* passing an invalid printprocessor will result in WERR_UNKNOWN_PRINTPROCESSOR */
1749 TEST_PRINTERINFO_STRING_EXP_ERR(2, printprocessor, 2, printprocessor, "xx2-2 printprocessor", WERR_UNKNOWN_PRINTPROCESSOR);
1750 TEST_PRINTERINFO_STRING(2, datatype, 2, datatype, "xx2-2 datatype");
1751 TEST_PRINTERINFO_STRING(2, parameters, 2, parameters, "xx2-2 parameters");
1753 for (i=0; i < ARRAY_SIZE(attribute_list); i++) {
1754 /* TEST_PRINTERINFO_INT_EXP(2, attributes, 1, flags,
1756 (attribute_list[i] | default_attribute)
1758 TEST_PRINTERINFO_INT_EXP(2, attributes, 2, attributes,
1760 (attribute_list[i] | default_attribute)
1762 TEST_PRINTERINFO_INT_EXP(2, attributes, 4, attributes,
1764 (attribute_list[i] | default_attribute)
1766 TEST_PRINTERINFO_INT_EXP(2, attributes, 5, attributes,
1768 (attribute_list[i] | default_attribute)
1770 /* TEST_PRINTERINFO_INT_EXP(4, attributes, 1, flags,
1772 (attribute_list[i] | default_attribute)
1774 TEST_PRINTERINFO_INT_EXP(4, attributes, 2, attributes,
1776 (attribute_list[i] | default_attribute)
1778 TEST_PRINTERINFO_INT_EXP(4, attributes, 4, attributes,
1780 (attribute_list[i] | default_attribute)
1782 TEST_PRINTERINFO_INT_EXP(4, attributes, 5, attributes,
1784 (attribute_list[i] | default_attribute)
1786 /* TEST_PRINTERINFO_INT_EXP(5, attributes, 1, flags,
1788 (attribute_list[i] | default_attribute)
1790 TEST_PRINTERINFO_INT_EXP(5, attributes, 2, attributes,
1792 (attribute_list[i] | default_attribute)
1794 TEST_PRINTERINFO_INT_EXP(5, attributes, 4, attributes,
1796 (attribute_list[i] | default_attribute)
1798 TEST_PRINTERINFO_INT_EXP(5, attributes, 5, attributes,
1800 (attribute_list[i] | default_attribute)
1804 for (i=0; i < ARRAY_SIZE(status_list); i++) {
1805 /* level 2 sets do not stick
1806 TEST_PRINTERINFO_INT(2, status, 0, status, status_list[i]);
1807 TEST_PRINTERINFO_INT(2, status, 2, status, status_list[i]);
1808 TEST_PRINTERINFO_INT(2, status, 6, status, status_list[i]); */
1809 TEST_PRINTERINFO_INT(6, status, 0, status, status_list[i]);
1810 TEST_PRINTERINFO_INT(6, status, 2, status, status_list[i]);
1811 TEST_PRINTERINFO_INT(6, status, 6, status, status_list[i]);
1814 /* priorities need to be between 0 and 99
1815 passing an invalid priority will result in WERR_INVALID_PRIORITY */
1816 TEST_PRINTERINFO_INT(2, priority, 2, priority, 0);
1817 TEST_PRINTERINFO_INT(2, priority, 2, priority, 1);
1818 TEST_PRINTERINFO_INT(2, priority, 2, priority, 99);
1819 /* TEST_PRINTERINFO_INT(2, priority, 2, priority, 100); */
1820 TEST_PRINTERINFO_INT(2, defaultpriority,2, defaultpriority, 0);
1821 TEST_PRINTERINFO_INT(2, defaultpriority,2, defaultpriority, 1);
1822 TEST_PRINTERINFO_INT(2, defaultpriority,2, defaultpriority, 99);
1823 /* TEST_PRINTERINFO_INT(2, defaultpriority,2, defaultpriority, 100); */
1825 TEST_PRINTERINFO_INT(2, starttime, 2, starttime, __LINE__);
1826 TEST_PRINTERINFO_INT(2, untiltime, 2, untiltime, __LINE__);
1829 TEST_PRINTERINFO_INT(2, cjobs, 2, cjobs, __LINE__);
1830 TEST_PRINTERINFO_INT(2, averageppm, 2, averageppm, __LINE__); */
1833 TEST_PRINTERINFO_INT(5, device_not_selected_timeout, 5, device_not_selected_timeout, __LINE__);
1834 TEST_PRINTERINFO_INT(5, transmission_retry_timeout, 5, transmission_retry_timeout, __LINE__); */
1836 /* FIXME: gd also test devmode and secdesc behavior */
1839 /* verify composition of level 1 description field */
1840 const char *description;
1844 do { TESTGETCALL(GetPrinter, q0) } while (0);
1846 description = talloc_strdup(tctx, q0.out.info->info1.description);
1849 do { TESTGETCALL(GetPrinter, q0) } while (0);
1851 tmp = talloc_asprintf(tctx, "%s,%s,%s",
1852 q0.out.info->info2.printername,
1853 q0.out.info->info2.drivername,
1854 q0.out.info->info2.location);
1856 do { STRING_EQUAL(description, tmp, "description")} while (0);
1862 static bool test_security_descriptor_equal(struct torture_context *tctx,
1863 const struct security_descriptor *sd1,
1864 const struct security_descriptor *sd2)
1871 torture_comment(tctx, "%s\n", __location__);
1875 torture_assert_int_equal(tctx, sd1->revision, sd2->revision, "revision mismatch");
1876 torture_assert_int_equal(tctx, sd1->type, sd2->type, "type mismatch");
1878 torture_assert_sid_equal(tctx, sd1->owner_sid, sd2->owner_sid, "owner mismatch");
1879 torture_assert_sid_equal(tctx, sd1->group_sid, sd2->group_sid, "group mismatch");
1881 if (!security_acl_equal(sd1->sacl, sd2->sacl)) {
1882 torture_comment(tctx, "%s: sacl mismatch\n", __location__);
1883 NDR_PRINT_DEBUG(security_acl, sd1->sacl);
1884 NDR_PRINT_DEBUG(security_acl, sd2->sacl);
1887 if (!security_acl_equal(sd1->dacl, sd2->dacl)) {
1888 torture_comment(tctx, "%s: dacl mismatch\n", __location__);
1889 NDR_PRINT_DEBUG(security_acl, sd1->dacl);
1890 NDR_PRINT_DEBUG(security_acl, sd2->dacl);
1897 static bool test_sd_set_level(struct torture_context *tctx,
1898 struct dcerpc_binding_handle *b,
1899 struct policy_handle *handle,
1901 struct security_descriptor *sd)
1903 struct spoolss_SetPrinterInfoCtr info_ctr;
1904 struct spoolss_DevmodeContainer devmode_ctr;
1905 struct sec_desc_buf secdesc_ctr;
1906 union spoolss_SetPrinterInfo sinfo;
1907 union spoolss_PrinterInfo info;
1908 struct spoolss_SetPrinterInfo3 info3;
1910 ZERO_STRUCT(devmode_ctr);
1911 ZERO_STRUCT(secdesc_ctr);
1915 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info), "");
1916 torture_assert(tctx, PrinterInfo_to_SetPrinterInfo(tctx, &info, 2, &sinfo), "");
1919 info_ctr.info = sinfo;
1925 info3.sec_desc_ptr = NULL;
1928 info_ctr.info.info3 = &info3;
1936 secdesc_ctr.sd = sd;
1938 torture_assert(tctx,
1939 test_SetPrinter(tctx, b, handle, &info_ctr, &devmode_ctr, &secdesc_ctr, 0), "");
1944 static bool test_PrinterInfo_SDs(struct torture_context *tctx,
1945 struct dcerpc_binding_handle *b,
1946 struct policy_handle *handle)
1948 union spoolss_PrinterInfo info;
1949 struct security_descriptor *sd1, *sd2;
1952 /* just compare level 2 and level 3 */
1954 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info), "");
1956 sd1 = info.info2.secdesc;
1958 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 3, &info), "");
1960 sd2 = info.info3.secdesc;
1962 torture_assert(tctx, test_security_descriptor_equal(tctx, sd1, sd2),
1963 "SD level 2 != SD level 3");
1966 /* query level 2, set level 2, query level 2 */
1968 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info), "");
1970 sd1 = info.info2.secdesc;
1972 torture_assert(tctx, test_sd_set_level(tctx, b, handle, 2, sd1), "");
1974 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info), "");
1976 sd2 = info.info2.secdesc;
1977 if (sd1->type & SEC_DESC_DACL_DEFAULTED) {
1978 torture_comment(tctx, "removing SEC_DESC_DACL_DEFAULTED\n");
1979 sd1->type &= ~SEC_DESC_DACL_DEFAULTED;
1982 torture_assert(tctx, test_security_descriptor_equal(tctx, sd1, sd2),
1983 "SD level 2 != SD level 2 after SD has been set via level 2");
1986 /* query level 2, set level 3, query level 2 */
1988 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info), "");
1990 sd1 = info.info2.secdesc;
1992 torture_assert(tctx, test_sd_set_level(tctx, b, handle, 3, sd1), "");
1994 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info), "");
1996 sd2 = info.info2.secdesc;
1998 torture_assert(tctx, test_security_descriptor_equal(tctx, sd1, sd2),
1999 "SD level 2 != SD level 2 after SD has been set via level 3");
2001 /* set modified sd level 3, query level 2 */
2003 for (i=0; i < 93; i++) {
2004 struct security_ace a;
2005 const char *sid_string = talloc_asprintf(tctx, "S-1-5-32-9999%i", i);
2006 a.type = SEC_ACE_TYPE_ACCESS_ALLOWED;
2008 a.size = 0; /* autogenerated */
2010 a.trustee = *dom_sid_parse_talloc(tctx, sid_string);
2011 torture_assert_ntstatus_ok(tctx, security_descriptor_dacl_add(sd1, &a), "");
2014 torture_assert(tctx, test_sd_set_level(tctx, b, handle, 3, sd1), "");
2016 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info), "");
2017 sd2 = info.info2.secdesc;
2019 if (sd1->type & SEC_DESC_DACL_DEFAULTED) {
2020 torture_comment(tctx, "removing SEC_DESC_DACL_DEFAULTED\n");
2021 sd1->type &= ~SEC_DESC_DACL_DEFAULTED;
2024 torture_assert(tctx, test_security_descriptor_equal(tctx, sd1, sd2),
2025 "modified SD level 2 != SD level 2 after SD has been set via level 3");
2032 * wrapper call that saves original sd, runs tests, and restores sd
2035 static bool test_PrinterInfo_SD(struct torture_context *tctx,
2036 struct dcerpc_binding_handle *b,
2037 struct policy_handle *handle)
2039 union spoolss_PrinterInfo info;
2040 struct security_descriptor *sd;
2043 torture_comment(tctx, "Testing Printer Security Descriptors\n");
2045 /* save original sd */
2047 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info),
2048 "failed to get initial security descriptor");
2050 sd = security_descriptor_copy(tctx, info.info2.secdesc);
2054 ret = test_PrinterInfo_SDs(tctx, b, handle);
2056 /* restore original sd */
2058 torture_assert(tctx, test_sd_set_level(tctx, b, handle, 3, sd),
2059 "failed to restore initial security descriptor");
2061 torture_comment(tctx, "Printer Security Descriptors test %s\n\n",
2062 ret ? "succeeded" : "failed");
2068 static bool test_devmode_set_level(struct torture_context *tctx,
2069 struct dcerpc_binding_handle *b,
2070 struct policy_handle *handle,
2072 struct spoolss_DeviceMode *devmode)
2074 struct spoolss_SetPrinterInfoCtr info_ctr;
2075 struct spoolss_DevmodeContainer devmode_ctr;
2076 struct sec_desc_buf secdesc_ctr;
2077 union spoolss_SetPrinterInfo sinfo;
2079 ZERO_STRUCT(devmode_ctr);
2080 ZERO_STRUCT(secdesc_ctr);
2084 union spoolss_PrinterInfo info;
2085 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info), "");
2086 torture_assert(tctx, PrinterInfo_to_SetPrinterInfo(tctx, &info, 2, &sinfo), "");
2089 info_ctr.info = sinfo;
2094 struct spoolss_SetPrinterInfo8 info8;
2096 info8.devmode_ptr = NULL;
2099 info_ctr.info.info8 = &info8;
2107 devmode_ctr.devmode = devmode;
2109 torture_assert(tctx,
2110 test_SetPrinter(tctx, b, handle, &info_ctr, &devmode_ctr, &secdesc_ctr, 0), "");
2116 static bool test_devicemode_equal(struct torture_context *tctx,
2117 const struct spoolss_DeviceMode *d1,
2118 const struct spoolss_DeviceMode *d2)
2125 torture_comment(tctx, "%s\n", __location__);
2128 torture_assert_str_equal(tctx, d1->devicename, d2->devicename, "devicename mismatch");
2129 torture_assert_int_equal(tctx, d1->specversion, d2->specversion, "specversion mismatch");
2130 torture_assert_int_equal(tctx, d1->driverversion, d2->driverversion, "driverversion mismatch");
2131 torture_assert_int_equal(tctx, d1->size, d2->size, "size mismatch");
2132 torture_assert_int_equal(tctx, d1->__driverextra_length, d2->__driverextra_length, "__driverextra_length mismatch");
2133 torture_assert_int_equal(tctx, d1->fields, d2->fields, "fields mismatch");
2134 torture_assert_int_equal(tctx, d1->orientation, d2->orientation, "orientation mismatch");
2135 torture_assert_int_equal(tctx, d1->papersize, d2->papersize, "papersize mismatch");
2136 torture_assert_int_equal(tctx, d1->paperlength, d2->paperlength, "paperlength mismatch");
2137 torture_assert_int_equal(tctx, d1->paperwidth, d2->paperwidth, "paperwidth mismatch");
2138 torture_assert_int_equal(tctx, d1->scale, d2->scale, "scale mismatch");
2139 torture_assert_int_equal(tctx, d1->copies, d2->copies, "copies mismatch");
2140 torture_assert_int_equal(tctx, d1->defaultsource, d2->defaultsource, "defaultsource mismatch");
2141 torture_assert_int_equal(tctx, d1->printquality, d2->printquality, "printquality mismatch");
2142 torture_assert_int_equal(tctx, d1->color, d2->color, "color mismatch");
2143 torture_assert_int_equal(tctx, d1->duplex, d2->duplex, "duplex mismatch");
2144 torture_assert_int_equal(tctx, d1->yresolution, d2->yresolution, "yresolution mismatch");
2145 torture_assert_int_equal(tctx, d1->ttoption, d2->ttoption, "ttoption mismatch");
2146 torture_assert_int_equal(tctx, d1->collate, d2->collate, "collate mismatch");
2147 torture_assert_str_equal(tctx, d1->formname, d2->formname, "formname mismatch");
2148 torture_assert_int_equal(tctx, d1->logpixels, d2->logpixels, "logpixels mismatch");
2149 torture_assert_int_equal(tctx, d1->bitsperpel, d2->bitsperpel, "bitsperpel mismatch");
2150 torture_assert_int_equal(tctx, d1->pelswidth, d2->pelswidth, "pelswidth mismatch");
2151 torture_assert_int_equal(tctx, d1->pelsheight, d2->pelsheight, "pelsheight mismatch");
2152 torture_assert_int_equal(tctx, d1->displayflags, d2->displayflags, "displayflags mismatch");
2153 torture_assert_int_equal(tctx, d1->displayfrequency, d2->displayfrequency, "displayfrequency mismatch");
2154 torture_assert_int_equal(tctx, d1->icmmethod, d2->icmmethod, "icmmethod mismatch");
2155 torture_assert_int_equal(tctx, d1->icmintent, d2->icmintent, "icmintent mismatch");
2156 torture_assert_int_equal(tctx, d1->mediatype, d2->mediatype, "mediatype mismatch");
2157 torture_assert_int_equal(tctx, d1->dithertype, d2->dithertype, "dithertype mismatch");
2158 torture_assert_int_equal(tctx, d1->reserved1, d2->reserved1, "reserved1 mismatch");
2159 torture_assert_int_equal(tctx, d1->reserved2, d2->reserved2, "reserved2 mismatch");
2160 torture_assert_int_equal(tctx, d1->panningwidth, d2->panningwidth, "panningwidth mismatch");
2161 torture_assert_int_equal(tctx, d1->panningheight, d2->panningheight, "panningheight mismatch");
2162 torture_assert_data_blob_equal(tctx, d1->driverextra_data, d2->driverextra_data, "driverextra_data mismatch");
2167 static bool test_devicemode_full(struct torture_context *tctx,
2168 struct dcerpc_binding_handle *b,
2169 struct policy_handle *handle)
2171 struct spoolss_SetPrinter s;
2172 struct spoolss_GetPrinter q;
2173 struct spoolss_SetPrinterInfoCtr info_ctr;
2174 struct spoolss_SetPrinterInfo8 info8;
2175 union spoolss_PrinterInfo info;
2176 struct spoolss_DevmodeContainer devmode_ctr;
2177 struct sec_desc_buf secdesc_ctr;
2182 #define TEST_DEVMODE_INT_EXP_RESULT(lvl1, field1, lvl2, field2, value, exp_value, expected_result) do { \
2183 torture_comment(tctx, "field test %d/%s vs %d/%s\n", lvl1, #field1, lvl2, #field2); \
2184 q.in.level = lvl1; \
2185 TESTGETCALL(GetPrinter, q) \
2186 info_ctr.level = lvl1; \
2188 void *p = (void *)&q.out.info->info ## lvl1; \
2189 info_ctr.info.info ## lvl1 = (struct spoolss_SetPrinterInfo ## lvl1 *)p; \
2190 } else if (lvl1 == 8) {\
2191 info_ctr.info.info ## lvl1 = &info8; \
2193 devmode_ctr.devmode = q.out.info->info ## lvl1.devmode; \
2194 devmode_ctr.devmode->field1 = value; \
2195 TESTSETCALL_EXP(SetPrinter, s, expected_result) \
2196 if (W_ERROR_IS_OK(expected_result)) { \
2197 TESTGETCALL(GetPrinter, q) \
2198 INT_EQUAL(q.out.info->info ## lvl1.devmode->field1, exp_value, field1); \
2199 q.in.level = lvl2; \
2200 TESTGETCALL(GetPrinter, q) \
2201 INT_EQUAL(q.out.info->info ## lvl2.devmode->field2, exp_value, field1); \
2205 #define TEST_DEVMODE_INT_EXP(lvl1, field1, lvl2, field2, value, expected_result) do { \
2206 TEST_DEVMODE_INT_EXP_RESULT(lvl1, field1, lvl2, field2, value, value, expected_result); \
2209 #define TEST_DEVMODE_INT(lvl1, field1, lvl2, field2, value) do { \
2210 TEST_DEVMODE_INT_EXP_RESULT(lvl1, field1, lvl2, field2, value, value, WERR_OK); \
2213 ZERO_STRUCT(devmode_ctr);
2214 ZERO_STRUCT(secdesc_ctr);
2217 s.in.handle = handle;
2219 s.in.info_ctr = &info_ctr;
2220 s.in.devmode_ctr = &devmode_ctr;
2221 s.in.secdesc_ctr = &secdesc_ctr;
2223 q.in.handle = handle;
2227 const char *devicename;/* [charset(UTF16)] */
2228 enum spoolss_DeviceModeSpecVersion specversion;
2229 uint16_t driverversion;
2230 uint16_t __driverextra_length;/* [value(r->driverextra_data.length)] */
2233 TEST_DEVMODE_INT_EXP(8, size, 8, size, __LINE__, WERR_INVALID_PARAM);
2234 TEST_DEVMODE_INT_EXP(8, size, 8, size, 0, WERR_INVALID_PARAM);
2235 TEST_DEVMODE_INT_EXP(8, size, 8, size, 0xffff, WERR_INVALID_PARAM);
2236 TEST_DEVMODE_INT_EXP(8, size, 8, size, ndr_size_spoolss_DeviceMode(devmode_ctr.devmode, 0), (devmode_ctr.devmode->__driverextra_length > 0 ) ? WERR_INVALID_PARAM : WERR_OK);
2237 TEST_DEVMODE_INT(8, size, 8, size, ndr_size_spoolss_DeviceMode(devmode_ctr.devmode, 0) - devmode_ctr.devmode->__driverextra_length);
2239 devmode_ctr.devmode->driverextra_data = data_blob_string_const("foobar");
2240 torture_assert(tctx,
2241 test_devmode_set_level(tctx, b, handle, 8, devmode_ctr.devmode),
2242 "failed to set devmode");
2244 TEST_DEVMODE_INT_EXP(8, size, 8, size, ndr_size_spoolss_DeviceMode(devmode_ctr.devmode, 0), (devmode_ctr.devmode->__driverextra_length > 0 ) ? WERR_INVALID_PARAM : WERR_OK);
2245 TEST_DEVMODE_INT(8, size, 8, size, ndr_size_spoolss_DeviceMode(devmode_ctr.devmode, 0) - devmode_ctr.devmode->__driverextra_length);
2247 TEST_DEVMODE_INT(8, orientation, 8, orientation, __LINE__);
2248 TEST_DEVMODE_INT(8, papersize, 8, papersize, __LINE__);
2249 TEST_DEVMODE_INT(8, paperlength, 8, paperlength, __LINE__);
2250 TEST_DEVMODE_INT(8, paperwidth, 8, paperwidth, __LINE__);
2251 TEST_DEVMODE_INT(8, scale, 8, scale, __LINE__);
2252 TEST_DEVMODE_INT(8, copies, 8, copies, __LINE__);
2253 TEST_DEVMODE_INT(8, defaultsource, 8, defaultsource, __LINE__);
2254 TEST_DEVMODE_INT(8, printquality, 8, printquality, __LINE__);
2255 TEST_DEVMODE_INT(8, color, 8, color, __LINE__);
2256 TEST_DEVMODE_INT(8, duplex, 8, duplex, __LINE__);
2257 TEST_DEVMODE_INT(8, yresolution, 8, yresolution, __LINE__);
2258 TEST_DEVMODE_INT(8, ttoption, 8, ttoption, __LINE__);
2259 TEST_DEVMODE_INT(8, collate, 8, collate, __LINE__);
2261 const char *formname;/* [charset(UTF16)] */
2263 TEST_DEVMODE_INT(8, logpixels, 8, logpixels, __LINE__);
2264 TEST_DEVMODE_INT(8, bitsperpel, 8, bitsperpel, __LINE__);
2265 TEST_DEVMODE_INT(8, pelswidth, 8, pelswidth, __LINE__);
2266 TEST_DEVMODE_INT(8, pelsheight, 8, pelsheight, __LINE__);
2267 TEST_DEVMODE_INT(8, displayflags, 8, displayflags, __LINE__);
2268 TEST_DEVMODE_INT(8, displayfrequency, 8, displayfrequency, __LINE__);
2269 TEST_DEVMODE_INT(8, icmmethod, 8, icmmethod, __LINE__);
2270 TEST_DEVMODE_INT(8, icmintent, 8, icmintent, __LINE__);
2271 TEST_DEVMODE_INT(8, mediatype, 8, mediatype, __LINE__);
2272 TEST_DEVMODE_INT(8, dithertype, 8, dithertype, __LINE__);
2273 TEST_DEVMODE_INT(8, reserved1, 8, reserved1, __LINE__);
2274 TEST_DEVMODE_INT(8, reserved2, 8, reserved2, __LINE__);
2275 TEST_DEVMODE_INT(8, panningwidth, 8, panningwidth, __LINE__);
2276 TEST_DEVMODE_INT(8, panningheight, 8, panningheight, __LINE__);
2281 static bool call_OpenPrinterEx(struct torture_context *tctx,
2282 struct dcerpc_pipe *p,
2284 struct spoolss_DeviceMode *devmode,
2285 struct policy_handle *handle);
2287 static bool test_PrinterInfo_DevModes(struct torture_context *tctx,
2288 struct dcerpc_pipe *p,
2289 struct policy_handle *handle,
2292 union spoolss_PrinterInfo info;
2293 struct spoolss_DeviceMode *devmode;
2294 struct spoolss_DeviceMode *devmode2;
2295 struct policy_handle handle_devmode;
2296 struct dcerpc_binding_handle *b = p->binding_handle;
2298 /* simply compare level8 and level2 devmode */
2300 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 8, &info), "");
2302 devmode = info.info8.devmode;
2304 if (devmode && devmode->size == 0) {
2306 "devmode of zero size!");
2309 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info), "");
2311 devmode2 = info.info2.devmode;
2313 if (devmode2 && devmode2->size == 0) {
2315 "devmode of zero size!");
2318 torture_assert(tctx, test_devicemode_equal(tctx, devmode, devmode2),
2319 "DM level 8 != DM level 2");
2322 /* set devicemode level 8 and see if it persists */
2324 devmode->copies = 93;
2325 devmode->formname = talloc_strdup(tctx, "Legal");
2327 torture_assert(tctx, test_devmode_set_level(tctx, b, handle, 8, devmode), "");
2329 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 8, &info), "");
2331 devmode2 = info.info8.devmode;
2333 torture_assert(tctx, test_devicemode_equal(tctx, devmode, devmode2),
2334 "modified DM level 8 != DM level 8 after DM has been set via level 8");
2336 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info), "");
2338 devmode2 = info.info2.devmode;
2340 torture_assert(tctx, test_devicemode_equal(tctx, devmode, devmode2),
2341 "modified DM level 8 != DM level 2");
2344 /* set devicemode level 2 and see if it persists */
2346 devmode->copies = 39;
2347 devmode->formname = talloc_strdup(tctx, "Executive");
2349 torture_assert(tctx, test_devmode_set_level(tctx, b, handle, 2, devmode), "");
2351 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 8, &info), "");
2353 devmode2 = info.info8.devmode;
2355 torture_assert(tctx, test_devicemode_equal(tctx, devmode, devmode2),
2356 "modified DM level 8 != DM level 8 after DM has been set via level 2");
2358 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info), "");
2360 devmode2 = info.info2.devmode;
2362 torture_assert(tctx, test_devicemode_equal(tctx, devmode, devmode2),
2363 "modified DM level 8 != DM level 2");
2366 /* check every single bit in public part of devicemode */
2368 torture_assert(tctx, test_devicemode_full(tctx, b, handle),
2369 "failed to set every single devicemode component");
2372 /* change formname upon open and see if it persists in getprinter calls */
2374 devmode->formname = talloc_strdup(tctx, "A4");
2375 devmode->copies = 42;
2377 torture_assert(tctx, call_OpenPrinterEx(tctx, p, name, devmode, &handle_devmode),
2378 "failed to open printer handle");
2380 torture_assert(tctx, test_GetPrinter_level(tctx, b, &handle_devmode, 8, &info), "");
2382 devmode2 = info.info8.devmode;
2384 if (strequal(devmode->devicename, devmode2->devicename)) {
2385 torture_warning(tctx, "devicenames are the same\n");
2387 torture_comment(tctx, "devicename passed in for open: %s\n", devmode->devicename);
2388 torture_comment(tctx, "devicename after level 8 get: %s\n", devmode2->devicename);
2391 if (strequal(devmode->formname, devmode2->formname)) {
2392 torture_warning(tctx, "formname are the same\n");
2394 torture_comment(tctx, "formname passed in for open: %s\n", devmode->formname);
2395 torture_comment(tctx, "formname after level 8 get: %s\n", devmode2->formname);
2398 if (devmode->copies == devmode2->copies) {
2399 torture_warning(tctx, "copies are the same\n");
2401 torture_comment(tctx, "copies passed in for open: %d\n", devmode->copies);
2402 torture_comment(tctx, "copies after level 8 get: %d\n", devmode2->copies);
2405 torture_assert(tctx, test_GetPrinter_level(tctx, b, &handle_devmode, 2, &info), "");
2407 devmode2 = info.info2.devmode;
2409 if (strequal(devmode->devicename, devmode2->devicename)) {
2410 torture_warning(tctx, "devicenames are the same\n");
2412 torture_comment(tctx, "devicename passed in for open: %s\n", devmode->devicename);
2413 torture_comment(tctx, "devicename after level 2 get: %s\n", devmode2->devicename);
2416 if (strequal(devmode->formname, devmode2->formname)) {
2417 torture_warning(tctx, "formname is the same\n");
2419 torture_comment(tctx, "formname passed in for open: %s\n", devmode->formname);
2420 torture_comment(tctx, "formname after level 2 get: %s\n", devmode2->formname);
2423 if (devmode->copies == devmode2->copies) {
2424 torture_warning(tctx, "copies are the same\n");
2426 torture_comment(tctx, "copies passed in for open: %d\n", devmode->copies);
2427 torture_comment(tctx, "copies after level 2 get: %d\n", devmode2->copies);
2430 test_ClosePrinter(tctx, b, &handle_devmode);
2436 * wrapper call that saves original devmode, runs tests, and restores devmode
2439 static bool test_PrinterInfo_DevMode(struct torture_context *tctx,
2440 struct dcerpc_pipe *p,
2441 struct policy_handle *handle,
2443 struct spoolss_DeviceMode *addprinter_devmode)
2445 union spoolss_PrinterInfo info;
2446 struct spoolss_DeviceMode *devmode;
2448 struct dcerpc_binding_handle *b = p->binding_handle;
2450 torture_comment(tctx, "Testing Printer Devicemodes\n");
2452 /* save original devmode */
2454 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 8, &info),
2455 "failed to get initial global devicemode");
2457 devmode = info.info8.devmode;
2459 if (devmode && devmode->size == 0) {
2461 "devmode of zero size!");
2464 if (addprinter_devmode) {
2465 if (!test_devicemode_equal(tctx, devmode, addprinter_devmode)) {
2466 torture_warning(tctx, "current global DM is != DM provided in addprinter");
2472 ret = test_PrinterInfo_DevModes(tctx, p, handle, name);
2474 /* restore original devmode */
2476 torture_assert(tctx, test_devmode_set_level(tctx, b, handle, 8, devmode),
2477 "failed to restore initial global device mode");
2479 torture_comment(tctx, "Printer Devicemodes test %s\n\n",
2480 ret ? "succeeded" : "failed");
2486 bool test_ClosePrinter(struct torture_context *tctx,
2487 struct dcerpc_binding_handle *b,
2488 struct policy_handle *handle)
2491 struct spoolss_ClosePrinter r;
2493 r.in.handle = handle;
2494 r.out.handle = handle;
2496 torture_comment(tctx, "Testing ClosePrinter\n");
2498 status = dcerpc_spoolss_ClosePrinter_r(b, tctx, &r);
2499 torture_assert_ntstatus_ok(tctx, status, "ClosePrinter failed");
2500 torture_assert_werr_ok(tctx, r.out.result, "ClosePrinter failed");
2505 static bool test_GetForm_args(struct torture_context *tctx,
2506 struct dcerpc_binding_handle *b,
2507 struct policy_handle *handle,
2508 const char *form_name,
2510 union spoolss_FormInfo *info_p)
2513 struct spoolss_GetForm r;
2516 r.in.handle = handle;
2517 r.in.form_name = form_name;
2521 r.out.needed = &needed;
2523 torture_comment(tctx, "Testing GetForm(%s) level %d\n", form_name, r.in.level);
2525 status = dcerpc_spoolss_GetForm_r(b, tctx, &r);
2526 torture_assert_ntstatus_ok(tctx, status, "GetForm failed");
2528 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
2529 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
2530 r.in.buffer = &blob;
2531 r.in.offered = needed;
2532 status = dcerpc_spoolss_GetForm_r(b, tctx, &r);
2533 torture_assert_ntstatus_ok(tctx, status, "GetForm failed");
2535 torture_assert_werr_ok(tctx, r.out.result, "GetForm failed");
2537 torture_assert(tctx, r.out.info, "No form info returned");
2540 torture_assert_werr_ok(tctx, r.out.result, "GetForm failed");
2542 CHECK_NEEDED_SIZE_LEVEL(spoolss_FormInfo, r.out.info, r.in.level, needed, 4);
2545 *info_p = *r.out.info;
2551 static bool test_GetForm(struct torture_context *tctx,
2552 struct dcerpc_binding_handle *b,
2553 struct policy_handle *handle,
2554 const char *form_name,
2557 return test_GetForm_args(tctx, b, handle, form_name, level, NULL);
2560 static bool test_EnumForms(struct torture_context *tctx,
2561 struct dcerpc_binding_handle *b,
2562 struct policy_handle *handle,
2566 union spoolss_FormInfo **info_p)
2568 struct spoolss_EnumForms r;
2571 union spoolss_FormInfo *info;
2573 r.in.handle = handle;
2577 r.out.needed = &needed;
2578 r.out.count = &count;
2581 torture_comment(tctx, "Testing EnumForms level %d\n", r.in.level);
2583 torture_assert_ntstatus_ok(tctx,
2584 dcerpc_spoolss_EnumForms_r(b, tctx, &r),
2585 "EnumForms failed");
2587 if ((r.in.level == 2) && (W_ERROR_EQUAL(r.out.result, WERR_UNKNOWN_LEVEL))) {
2588 torture_skip(tctx, "EnumForms level 2 not supported");
2591 if (print_server && W_ERROR_EQUAL(r.out.result, WERR_BADFID)) {
2592 torture_fail(tctx, "EnumForms on the PrintServer isn't supported by test server (NT4)");
2595 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
2596 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
2597 r.in.buffer = &blob;
2598 r.in.offered = needed;
2600 torture_assert_ntstatus_ok(tctx,
2601 dcerpc_spoolss_EnumForms_r(b, tctx, &r),
2602 "EnumForms failed");
2604 torture_assert(tctx, info, "No forms returned");
2607 torture_assert_werr_ok(tctx, r.out.result, "EnumForms failed");
2609 CHECK_NEEDED_SIZE_ENUM_LEVEL(spoolss_EnumForms, info, r.in.level, count, needed, 4);
2621 static bool test_EnumForms_all(struct torture_context *tctx,
2622 struct dcerpc_binding_handle *b,
2623 struct policy_handle *handle,
2626 uint32_t levels[] = { 1, 2 };
2629 for (i=0; i<ARRAY_SIZE(levels); i++) {
2632 union spoolss_FormInfo *info = NULL;
2634 torture_assert(tctx,
2635 test_EnumForms(tctx, b, handle, print_server, levels[i], &count, &info),
2636 "failed to enum forms");
2638 for (j = 0; j < count; j++) {
2639 if (!print_server) {
2640 torture_assert(tctx,
2641 test_GetForm(tctx, b, handle, info[j].info1.form_name, levels[i]),
2642 "failed to get form");
2650 static bool test_EnumForms_find_one(struct torture_context *tctx,
2651 struct dcerpc_binding_handle *b,
2652 struct policy_handle *handle,
2654 const char *form_name)
2656 union spoolss_FormInfo *info;
2661 torture_assert(tctx,
2662 test_EnumForms(tctx, b, handle, print_server, 1, &count, &info),
2663 "failed to enumerate forms");
2665 for (i=0; i<count; i++) {
2666 if (strequal(form_name, info[i].info1.form_name)) {
2675 static bool test_DeleteForm(struct torture_context *tctx,
2676 struct dcerpc_binding_handle *b,
2677 struct policy_handle *handle,
2678 const char *form_name,
2679 WERROR expected_result)
2681 struct spoolss_DeleteForm r;
2683 r.in.handle = handle;
2684 r.in.form_name = form_name;
2686 torture_comment(tctx, "Testing DeleteForm(%s)\n", form_name);
2688 torture_assert_ntstatus_ok(tctx,
2689 dcerpc_spoolss_DeleteForm_r(b, tctx, &r),
2690 "DeleteForm failed");
2691 torture_assert_werr_equal(tctx, r.out.result, expected_result,
2692 "DeleteForm gave unexpected result");
2693 if (W_ERROR_IS_OK(r.out.result)) {
2694 torture_assert_ntstatus_ok(tctx,
2695 dcerpc_spoolss_DeleteForm_r(b, tctx, &r),
2696 "2nd DeleteForm failed");
2697 torture_assert_werr_equal(tctx, r.out.result, WERR_INVALID_FORM_NAME,
2698 "2nd DeleteForm failed");
2704 static bool test_AddForm(struct torture_context *tctx,
2705 struct dcerpc_binding_handle *b,
2706 struct policy_handle *handle,
2708 union spoolss_AddFormInfo *info,
2709 WERROR expected_result)
2711 struct spoolss_AddForm r;
2712 struct spoolss_AddFormInfoCtr info_ctr;
2714 info_ctr.level = level;
2715 info_ctr.info = *info;
2718 torture_skip(tctx, "only level 1 supported");
2721 r.in.handle = handle;
2722 r.in.info_ctr = &info_ctr;
2724 torture_comment(tctx, "Testing AddForm(%s) level %d, type %d\n",
2725 r.in.info_ctr->info.info1->form_name, level,
2726 r.in.info_ctr->info.info1->flags);
2728 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_AddForm_r(b, tctx, &r),
2730 torture_assert_werr_equal(tctx, r.out.result, expected_result,
2731 "AddForm gave unexpected result");
2733 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_AddForm_r(b, tctx, &r),
2734 "2nd AddForm failed");
2735 if (W_ERROR_EQUAL(expected_result, WERR_INVALID_PARAM)) {
2736 torture_assert_werr_equal(tctx, r.out.result, WERR_INVALID_PARAM,
2737 "2nd AddForm gave unexpected result");
2739 torture_assert_werr_equal(tctx, r.out.result, WERR_FILE_EXISTS,
2740 "2nd AddForm gave unexpected result");
2746 static bool test_SetForm(struct torture_context *tctx,
2747 struct dcerpc_binding_handle *b,
2748 struct policy_handle *handle,
2749 const char *form_name,
2751 union spoolss_AddFormInfo *info)
2753 struct spoolss_SetForm r;
2754 struct spoolss_AddFormInfoCtr info_ctr;
2756 info_ctr.level = level;
2757 info_ctr.info = *info;
2759 r.in.handle = handle;
2760 r.in.form_name = form_name;
2761 r.in.info_ctr = &info_ctr;
2763 torture_comment(tctx, "Testing SetForm(%s) level %d\n",
2766 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_SetForm_r(b, tctx, &r),
2769 torture_assert_werr_ok(tctx, r.out.result,
2775 static bool test_GetForm_winreg(struct torture_context *tctx,
2776 struct dcerpc_binding_handle *b,
2777 struct policy_handle *handle,
2778 const char *key_name,
2779 const char *form_name,
2780 enum winreg_Type *w_type,
2785 static bool test_Forms_args(struct torture_context *tctx,
2786 struct dcerpc_binding_handle *b,
2787 struct policy_handle *handle,
2789 const char *printer_name,
2790 struct dcerpc_binding_handle *winreg_handle,
2791 struct policy_handle *hive_handle,
2792 const char *form_name,
2793 struct spoolss_AddFormInfo1 *info1,
2794 WERROR expected_add_result,
2795 WERROR expected_delete_result)
2797 union spoolss_FormInfo info;
2798 union spoolss_AddFormInfo add_info;
2800 enum winreg_Type w_type;
2805 add_info.info1 = info1;
2807 torture_assert(tctx,
2808 test_AddForm(tctx, b, handle, 1, &add_info, expected_add_result),
2809 "failed to add form");
2811 if (winreg_handle && hive_handle && W_ERROR_IS_OK(expected_add_result)) {
2813 struct spoolss_FormInfo1 i1;
2815 torture_assert(tctx,
2816 test_GetForm_winreg(tctx, winreg_handle, hive_handle, TOP_LEVEL_CONTROL_FORMS_KEY, form_name, &w_type, &w_size, &w_length, &w_data),
2817 "failed to get form via winreg");
2819 i1.size.width = IVAL(w_data, 0);
2820 i1.size.height = IVAL(w_data, 4);
2821 i1.area.left = IVAL(w_data, 8);
2822 i1.area.top = IVAL(w_data, 12);
2823 i1.area.right = IVAL(w_data, 16);
2824 i1.area.bottom = IVAL(w_data, 20);
2825 /* skip index here */
2826 i1.flags = IVAL(w_data, 28);
2828 torture_assert_int_equal(tctx, w_type, REG_BINARY, "unexpected type");
2829 torture_assert_int_equal(tctx, w_size, 0x20, "unexpected size");
2830 torture_assert_int_equal(tctx, w_length, 0x20, "unexpected length");
2831 torture_assert_int_equal(tctx, i1.size.width, add_info.info1->size.width, "width mismatch");
2832 torture_assert_int_equal(tctx, i1.size.height, add_info.info1->size.height, "height mismatch");
2833 torture_assert_int_equal(tctx, i1.area.left, add_info.info1->area.left, "left mismatch");
2834 torture_assert_int_equal(tctx, i1.area.top, add_info.info1->area.top, "top mismatch");
2835 torture_assert_int_equal(tctx, i1.area.right, add_info.info1->area.right, "right mismatch");
2836 torture_assert_int_equal(tctx, i1.area.bottom, add_info.info1->area.bottom, "bottom mismatch");
2837 torture_assert_int_equal(tctx, i1.flags, add_info.info1->flags, "flags mismatch");
2840 if (!print_server && W_ERROR_IS_OK(expected_add_result)) {
2841 torture_assert(tctx,
2842 test_GetForm_args(tctx, b, handle, form_name, 1, &info),
2843 "failed to get added form");
2845 torture_assert_int_equal(tctx, info.info1.size.width, add_info.info1->size.width, "width mismatch");
2846 torture_assert_int_equal(tctx, info.info1.size.height, add_info.info1->size.height, "height mismatch");
2847 torture_assert_int_equal(tctx, info.info1.area.left, add_info.info1->area.left, "left mismatch");
2848 torture_assert_int_equal(tctx, info.info1.area.top, add_info.info1->area.top, "top mismatch");
2849 torture_assert_int_equal(tctx, info.info1.area.right, add_info.info1->area.right, "right mismatch");
2850 torture_assert_int_equal(tctx, info.info1.area.bottom, add_info.info1->area.bottom, "bottom mismatch");
2851 torture_assert_int_equal(tctx, info.info1.flags, add_info.info1->flags, "flags mismatch");
2853 if (winreg_handle && hive_handle) {
2855 struct spoolss_FormInfo1 i1;
2857 i1.size.width = IVAL(w_data, 0);
2858 i1.size.height = IVAL(w_data, 4);
2859 i1.area.left = IVAL(w_data, 8);
2860 i1.area.top = IVAL(w_data, 12);
2861 i1.area.right = IVAL(w_data, 16);
2862 i1.area.bottom = IVAL(w_data, 20);
2863 /* skip index here */
2864 i1.flags = IVAL(w_data, 28);
2866 torture_assert_int_equal(tctx, i1.size.width, info.info1.size.width, "width mismatch");
2867 torture_assert_int_equal(tctx, i1.size.height, info.info1.size.height, "height mismatch");
2868 torture_assert_int_equal(tctx, i1.area.left, info.info1.area.left, "left mismatch");
2869 torture_assert_int_equal(tctx, i1.area.top, info.info1.area.top, "top mismatch");
2870 torture_assert_int_equal(tctx, i1.area.right, info.info1.area.right, "right mismatch");
2871 torture_assert_int_equal(tctx, i1.area.bottom, info.info1.area.bottom, "bottom mismatch");
2872 torture_assert_int_equal(tctx, i1.flags, info.info1.flags, "flags mismatch");
2875 add_info.info1->size.width = 1234;
2877 torture_assert(tctx,
2878 test_SetForm(tctx, b, handle, form_name, 1, &add_info),
2879 "failed to set form");
2880 torture_assert(tctx,
2881 test_GetForm_args(tctx, b, handle, form_name, 1, &info),
2882 "failed to get setted form");
2884 torture_assert_int_equal(tctx, info.info1.size.width, add_info.info1->size.width, "width mismatch");
2887 if (!W_ERROR_EQUAL(expected_add_result, WERR_INVALID_PARAM)) {
2888 torture_assert(tctx,
2889 test_EnumForms_find_one(tctx, b, handle, print_server, form_name),
2890 "Newly added form not found in enum call");
2893 torture_assert(tctx,
2894 test_DeleteForm(tctx, b, handle, form_name, expected_delete_result),
2895 "failed to delete form");
2900 static bool test_Forms(struct torture_context *tctx,
2901 struct dcerpc_binding_handle *b,
2902 struct policy_handle *handle,
2904 const char *printer_name,
2905 struct dcerpc_binding_handle *winreg_handle,
2906 struct policy_handle *hive_handle)
2908 const struct spoolss_FormSize size = {
2912 const struct spoolss_FormArea area = {
2921 struct spoolss_AddFormInfo1 info1;
2922 WERROR expected_add_result;
2923 WERROR expected_delete_result;
2927 .flags = SPOOLSS_FORM_USER,
2928 .form_name = "testform_user",
2932 .expected_add_result = WERR_OK,
2933 .expected_delete_result = WERR_OK
2936 weird, we can add a builtin form but we can never remove it
2941 .flags = SPOOLSS_FORM_BUILTIN,
2942 .form_name = "testform_builtin",
2946 .expected_add_result = WERR_OK,
2947 .expected_delete_result = WERR_INVALID_PARAM,
2952 .flags = SPOOLSS_FORM_PRINTER,
2953 .form_name = "testform_printer",
2957 .expected_add_result = WERR_OK,
2958 .expected_delete_result = WERR_OK
2962 .flags = SPOOLSS_FORM_USER,
2963 .form_name = "Letter",
2967 .expected_add_result = WERR_FILE_EXISTS,
2968 .expected_delete_result = WERR_INVALID_PARAM
2972 .flags = SPOOLSS_FORM_BUILTIN,
2973 .form_name = "Letter",
2977 .expected_add_result = WERR_FILE_EXISTS,
2978 .expected_delete_result = WERR_INVALID_PARAM
2982 .flags = SPOOLSS_FORM_PRINTER,
2983 .form_name = "Letter",
2987 .expected_add_result = WERR_FILE_EXISTS,
2988 .expected_delete_result = WERR_INVALID_PARAM
2993 .form_name = "invalid_flags",
2997 .expected_add_result = WERR_INVALID_PARAM,
2998 .expected_delete_result = WERR_INVALID_FORM_NAME
3003 for (i=0; i < ARRAY_SIZE(forms); i++) {
3004 torture_assert(tctx,
3005 test_Forms_args(tctx, b, handle, print_server, printer_name,
3006 winreg_handle, hive_handle,
3007 forms[i].info1.form_name,
3009 forms[i].expected_add_result,
3010 forms[i].expected_delete_result),
3011 talloc_asprintf(tctx, "failed to test form '%s'", forms[i].info1.form_name));
3017 static bool test_EnumPorts_old(struct torture_context *tctx,
3020 struct test_spoolss_context *ctx =
3021 talloc_get_type_abort(private_data, struct test_spoolss_context);
3024 struct spoolss_EnumPorts r;
3027 union spoolss_PortInfo *info;
3028 struct dcerpc_pipe *p = ctx->spoolss_pipe;
3029 struct dcerpc_binding_handle *b = p->binding_handle;
3031 r.in.servername = talloc_asprintf(tctx, "\\\\%s",
3032 dcerpc_server_name(p));
3036 r.out.needed = &needed;
3037 r.out.count = &count;
3040 torture_comment(tctx, "Testing EnumPorts\n");
3042 status = dcerpc_spoolss_EnumPorts_r(b, tctx, &r);
3044 torture_assert_ntstatus_ok(tctx, status, "EnumPorts failed");
3046 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
3047 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
3048 r.in.buffer = &blob;
3049 r.in.offered = needed;
3051 status = dcerpc_spoolss_EnumPorts_r(b, tctx, &r);
3052 torture_assert_ntstatus_ok(tctx, status, "EnumPorts failed");
3053 torture_assert_werr_ok(tctx, r.out.result, "EnumPorts failed");
3055 torture_assert(tctx, info, "No ports returned");
3058 torture_assert_werr_ok(tctx, r.out.result, "EnumPorts failed");
3060 CHECK_NEEDED_SIZE_ENUM_LEVEL(spoolss_EnumPorts, info, 2, count, needed, 4);
3065 static bool test_AddPort(struct torture_context *tctx,
3068 struct test_spoolss_context *ctx =
3069 talloc_get_type_abort(private_data, struct test_spoolss_context);
3072 struct spoolss_AddPort r;
3073 struct dcerpc_pipe *p = ctx->spoolss_pipe;
3074 struct dcerpc_binding_handle *b = p->binding_handle;
3076 r.in.server_name = talloc_asprintf(tctx, "\\\\%s",
3077 dcerpc_server_name(p));
3079 r.in.monitor_name = "foo";
3081 torture_comment(tctx, "Testing AddPort\n");
3083 status = dcerpc_spoolss_AddPort_r(b, tctx, &r);
3085 torture_assert_ntstatus_ok(tctx, status, "AddPort failed");
3087 /* win2k3 returns WERR_NOT_SUPPORTED */
3091 if (!W_ERROR_IS_OK(r.out.result)) {
3092 printf("AddPort failed - %s\n", win_errstr(r.out.result));
3101 static bool test_GetJob_args(struct torture_context *tctx,
3102 struct dcerpc_binding_handle *b,
3103 struct policy_handle *handle,
3106 union spoolss_JobInfo *info_p)
3109 struct spoolss_GetJob r;
3110 union spoolss_JobInfo info;
3113 r.in.handle = handle;
3114 r.in.job_id = job_id;
3118 r.out.needed = &needed;
3121 torture_comment(tctx, "Testing GetJob(%d), level %d\n", job_id, r.in.level);
3123 status = dcerpc_spoolss_GetJob_r(b, tctx, &r);
3124 torture_assert_ntstatus_ok(tctx, status, "GetJob failed");
3126 torture_assert_werr_equal(tctx, r.out.result, WERR_UNKNOWN_LEVEL, "Unexpected return code");
3129 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
3130 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
3131 r.in.buffer = &blob;
3132 r.in.offered = needed;
3134 status = dcerpc_spoolss_GetJob_r(b, tctx, &r);
3135 torture_assert_ntstatus_ok(tctx, status, "GetJob failed");
3138 torture_assert_werr_ok(tctx, r.out.result, "GetJob failed");
3139 torture_assert(tctx, r.out.info, "No job info returned");
3141 CHECK_NEEDED_SIZE_LEVEL(spoolss_JobInfo, r.out.info, r.in.level, needed, 4);
3144 *info_p = *r.out.info;
3151 static bool test_GetJob(struct torture_context *tctx,
3152 struct dcerpc_binding_handle *b,
3153 struct policy_handle *handle,
3156 uint32_t levels[] = {0, 1, 2 /* 3, 4 */};
3159 for (i=0; i < ARRAY_SIZE(levels); i++) {
3160 torture_assert(tctx,
3161 test_GetJob_args(tctx, b, handle, job_id, levels[i], NULL),
3169 static bool test_SetJob(struct torture_context *tctx,
3170 struct dcerpc_binding_handle *b,
3171 struct policy_handle *handle,
3173 struct spoolss_JobInfoContainer *ctr,
3174 enum spoolss_JobControl command)
3177 struct spoolss_SetJob r;
3179 r.in.handle = handle;
3180 r.in.job_id = job_id;
3182 r.in.command = command;
3185 case SPOOLSS_JOB_CONTROL_PAUSE:
3186 torture_comment(tctx, "Testing SetJob(%d), SPOOLSS_JOB_CONTROL_PAUSE\n", job_id);
3188 case SPOOLSS_JOB_CONTROL_RESUME:
3189 torture_comment(tctx, "Testing SetJob(%d), SPOOLSS_JOB_CONTROL_RESUME\n", job_id);
3191 case SPOOLSS_JOB_CONTROL_CANCEL:
3192 torture_comment(tctx, "Testing SetJob(%d), SPOOLSS_JOB_CONTROL_CANCEL\n", job_id);
3194 case SPOOLSS_JOB_CONTROL_RESTART:
3195 torture_comment(tctx, "Testing SetJob(%d), SPOOLSS_JOB_CONTROL_RESTART\n", job_id);
3197 case SPOOLSS_JOB_CONTROL_DELETE:
3198 torture_comment(tctx, "Testing SetJob(%d), SPOOLSS_JOB_CONTROL_DELETE\n", job_id);
3200 case SPOOLSS_JOB_CONTROL_SEND_TO_PRINTER:
3201 torture_comment(tctx, "Testing SetJob(%d), SPOOLSS_JOB_CONTROL_SEND_TO_PRINTER\n", job_id);
3203 case SPOOLSS_JOB_CONTROL_LAST_PAGE_EJECTED:
3204 torture_comment(tctx, "Testing SetJob(%d), SPOOLSS_JOB_CONTROL_LAST_PAGE_EJECTED\n", job_id);
3206 case SPOOLSS_JOB_CONTROL_RETAIN:
3207 torture_comment(tctx, "Testing SetJob(%d), SPOOLSS_JOB_CONTROL_RETAIN\n", job_id);
3209 case SPOOLSS_JOB_CONTROL_RELEASE:
3210 torture_comment(tctx, "Testing SetJob(%d), SPOOLSS_JOB_CONTROL_RELEASE\n", job_id);
3213 torture_comment(tctx, "Testing SetJob(%d)\n", job_id);
3217 status = dcerpc_spoolss_SetJob_r(b, tctx, &r);
3218 torture_assert_ntstatus_ok(tctx, status, "SetJob failed");
3219 torture_assert_werr_ok(tctx, r.out.result, "SetJob failed");
3224 static bool test_AddJob(struct torture_context *tctx,
3225 struct dcerpc_binding_handle *b,
3226 struct policy_handle *handle)
3229 struct spoolss_AddJob r;
3233 r.in.handle = handle;
3235 r.out.needed = &needed;
3236 r.in.buffer = r.out.buffer = NULL;
3238 torture_comment(tctx, "Testing AddJob\n");
3240 status = dcerpc_spoolss_AddJob_r(b, tctx, &r);
3241 torture_assert_ntstatus_ok(tctx, status, "AddJob failed");
3242 torture_assert_werr_equal(tctx, r.out.result, WERR_UNKNOWN_LEVEL, "AddJob failed");
3246 status = dcerpc_spoolss_AddJob_r(b, tctx, &r);
3247 torture_assert_ntstatus_ok(tctx, status, "AddJob failed");
3248 torture_assert_werr_equal(tctx, r.out.result, WERR_INVALID_PARAM, "AddJob failed");
3254 static bool test_EnumJobs_args(struct torture_context *tctx,
3255 struct dcerpc_binding_handle *b,
3256 struct policy_handle *handle,
3258 WERROR werr_expected,
3260 union spoolss_JobInfo **info_p)
3263 struct spoolss_EnumJobs r;
3266 union spoolss_JobInfo *info;
3268 r.in.handle = handle;
3270 r.in.numjobs = 0xffffffff;
3274 r.out.needed = &needed;
3275 r.out.count = &count;
3278 torture_comment(tctx, "Testing EnumJobs level %d\n", level);
3280 status = dcerpc_spoolss_EnumJobs_r(b, tctx, &r);
3282 torture_assert_ntstatus_ok(tctx, status, "EnumJobs failed");
3284 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
3285 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
3286 r.in.buffer = &blob;
3287 r.in.offered = needed;
3289 status = dcerpc_spoolss_EnumJobs_r(b, tctx, &r);
3291 torture_assert_ntstatus_ok(tctx, status, "EnumJobs failed");
3292 torture_assert_werr_equal(tctx, r.out.result, werr_expected,
3294 torture_assert(tctx, info, "No jobs returned");
3296 CHECK_NEEDED_SIZE_ENUM_LEVEL(spoolss_EnumJobs, *r.out.info, r.in.level, count, needed, 4);
3299 torture_assert_werr_equal(tctx, r.out.result, werr_expected,
3313 static bool test_JobPropertiesEnum(struct torture_context *tctx,
3314 struct dcerpc_binding_handle *b,
3315 struct policy_handle *handle,
3318 struct spoolss_EnumJobNamedProperties r;
3319 uint32_t pcProperties = 0;
3320 struct spoolss_PrintNamedProperty *ppProperties = NULL;
3322 r.in.hPrinter = handle;
3323 r.in.JobId = job_id;
3324 r.out.pcProperties = &pcProperties;
3325 r.out.ppProperties = &ppProperties;
3327 torture_comment(tctx, "Testing EnumJobNamedProperties(%d)\n", job_id);
3329 torture_assert_ntstatus_ok(tctx,
3330 dcerpc_spoolss_EnumJobNamedProperties_r(b, tctx, &r),
3331 "spoolss_EnumJobNamedProperties failed");
3332 torture_assert_werr_ok(tctx, r.out.result,
3333 "spoolss_EnumJobNamedProperties failed");
3338 static bool test_JobPropertySet(struct torture_context *tctx,
3339 struct dcerpc_binding_handle *b,
3340 struct policy_handle *handle,
3342 struct spoolss_PrintNamedProperty *property)
3344 struct spoolss_SetJobNamedProperty r;
3346 r.in.hPrinter = handle;
3347 r.in.JobId = job_id;
3348 r.in.pProperty = property;
3350 torture_comment(tctx, "Testing SetJobNamedProperty(%d) %s - %d\n",
3351 job_id, property->propertyName,
3352 property->propertyValue.ePropertyType);
3354 torture_assert_ntstatus_ok(tctx,
3355 dcerpc_spoolss_SetJobNamedProperty_r(b, tctx, &r),
3356 "spoolss_SetJobNamedProperty failed");
3357 torture_assert_werr_ok(tctx, r.out.result,
3358 "spoolss_SetJobNamedProperty failed");
3363 static bool test_JobPropertyGetValue(struct torture_context *tctx,
3364 struct dcerpc_binding_handle *b,
3365 struct policy_handle *handle,
3367 const char *property_name,
3368 struct spoolss_PrintPropertyValue *value)
3370 struct spoolss_GetJobNamedPropertyValue r;
3372 r.in.hPrinter = handle;
3373 r.in.JobId = job_id;
3374 r.in.pszName = property_name;
3375 r.out.pValue = value;
3377 torture_comment(tctx, "Testing GetJobNamedPropertyValue(%d) %s\n",
3378 job_id, property_name);
3380 torture_assert_ntstatus_ok(tctx,
3381 dcerpc_spoolss_GetJobNamedPropertyValue_r(b, tctx, &r),
3382 "spoolss_GetJobNamedPropertyValue failed");
3383 torture_assert_werr_ok(tctx, r.out.result,
3384 "spoolss_GetJobNamedPropertyValue failed");
3389 static bool test_JobPropertyDelete(struct torture_context *tctx,
3390 struct dcerpc_binding_handle *b,
3391 struct policy_handle *handle,
3393 const char *property_name)
3395 struct spoolss_DeleteJobNamedProperty r;
3397 r.in.hPrinter = handle;
3398 r.in.JobId = job_id;
3399 r.in.pszName = property_name;
3401 torture_comment(tctx, "Testing DeleteJobNamedProperty(%d) %s\n",
3402 job_id, property_name);
3404 torture_assert_ntstatus_ok(tctx,
3405 dcerpc_spoolss_DeleteJobNamedProperty_r(b, tctx, &r),
3406 "spoolss_DeleteJobNamedProperty failed");
3407 torture_assert_werr_ok(tctx, r.out.result,
3408 "spoolss_DeleteJobNamedProperty failed");
3413 static bool test_DoPrintTest_add_one_job_common(struct torture_context *tctx,
3414 struct dcerpc_binding_handle *b,
3415 struct policy_handle *handle,
3416 const char *document_name,
3417 const char *datatype,
3421 struct spoolss_StartDocPrinter s;
3422 struct spoolss_DocumentInfoCtr info_ctr;
3423 struct spoolss_DocumentInfo1 info1;
3424 struct spoolss_StartPagePrinter sp;
3425 struct spoolss_WritePrinter w;
3426 struct spoolss_EndPagePrinter ep;
3427 struct spoolss_EndDocPrinter e;
3429 uint32_t num_written;
3431 torture_comment(tctx, "Testing StartDocPrinter\n");
3433 s.in.handle = handle;
3434 s.in.info_ctr = &info_ctr;
3435 s.out.job_id = job_id;
3437 info1.document_name = document_name;
3438 info1.output_file = NULL;
3439 info1.datatype = datatype;
3442 info_ctr.info.info1 = &info1;
3444 status = dcerpc_spoolss_StartDocPrinter_r(b, tctx, &s);
3445 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_StartDocPrinter failed");
3446 torture_assert_werr_ok(tctx, s.out.result, "StartDocPrinter failed");
3448 for (i=1; i < 4; i++) {
3449 union spoolss_JobInfo ginfo;
3452 torture_comment(tctx, "Testing StartPagePrinter: Page[%d], JobId[%d]\n", i, *job_id);
3454 sp.in.handle = handle;
3456 status = dcerpc_spoolss_StartPagePrinter_r(b, tctx, &sp);
3457 torture_assert_ntstatus_ok(tctx, status,
3458 "dcerpc_spoolss_StartPagePrinter failed");
3459 torture_assert_werr_ok(tctx, sp.out.result, "StartPagePrinter failed");
3461 ok = test_GetJob_args(tctx, b, handle, *job_id, 1, &ginfo);
3463 torture_comment(tctx, "test_GetJob failed for JobId[%d]\n", *job_id);
3466 torture_comment(tctx, "Testing WritePrinter: Page[%d], JobId[%d]\n", i, *job_id);
3468 w.in.handle = handle;
3469 w.in.data = data_blob_string_const(talloc_asprintf(tctx,"TortureTestPage: %d\nData\n",i));
3470 w.out.num_written = &num_written;
3472 status = dcerpc_spoolss_WritePrinter_r(b, tctx, &w);
3473 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_WritePrinter failed");
3474 torture_assert_werr_ok(tctx, w.out.result, "WritePrinter failed");
3476 torture_comment(tctx, "Testing EndPagePrinter: Page[%d], JobId[%d]\n", i, *job_id);
3478 ep.in.handle = handle;
3480 status = dcerpc_spoolss_EndPagePrinter_r(b, tctx, &ep);
3481 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EndPagePrinter failed");
3482 torture_assert_werr_ok(tctx, ep.out.result, "EndPagePrinter failed");
3485 torture_comment(tctx, "Testing EndDocPrinter: JobId[%d]\n", *job_id);
3487 e.in.handle = handle;
3489 status = dcerpc_spoolss_EndDocPrinter_r(b, tctx, &e);
3490 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EndDocPrinter failed");
3491 torture_assert_werr_ok(tctx, e.out.result, "EndDocPrinter failed");
3496 static bool test_DoPrintTest_add_one_job(struct torture_context *tctx,
3497 struct dcerpc_binding_handle *b,
3498 struct policy_handle *handle,
3499 const char *document_name,
3502 test_DoPrintTest_add_one_job_common(tctx, b, handle, document_name, "RAW", job_id);
3507 static bool test_DoPrintTest_add_one_job_v4(struct torture_context *tctx,
3508 struct dcerpc_binding_handle *b,
3509 struct policy_handle *handle,
3510 const char *document_name,
3513 test_DoPrintTest_add_one_job_common(tctx, b, handle, document_name, "XPS_PASS", job_id);
3519 static bool test_DoPrintTest_check_jobs(struct torture_context *tctx,
3520 struct dcerpc_binding_handle *b,
3521 struct policy_handle *handle,
3526 union spoolss_JobInfo *info = NULL;
3529 torture_assert(tctx,
3530 test_AddJob(tctx, b, handle),
3533 torture_assert(tctx,
3534 test_EnumJobs_args(tctx, b, handle, 1, WERR_OK, &count, &info),
3535 "EnumJobs level 1 failed");
3537 torture_assert_int_equal(tctx, count, num_jobs, "unexpected number of jobs in queue");
3539 for (i=0; i < num_jobs; i++) {
3540 union spoolss_JobInfo ginfo;
3541 const char *document_name;
3542 const char *new_document_name = "any_other_docname";
3543 struct spoolss_JobInfoContainer ctr;
3544 struct spoolss_SetJobInfo1 info1;
3546 torture_assert_int_equal(tctx, info[i].info1.job_id, job_ids[i], "job id mismatch");
3548 torture_assert(tctx,
3549 test_GetJob_args(tctx, b, handle, info[i].info1.job_id, 1, &ginfo),
3550 "failed to call test_GetJob");
3552 torture_assert_int_equal(tctx, ginfo.info1.job_id, info[i].info1.job_id, "job id mismatch");
3554 document_name = ginfo.info1.document_name;
3556 info1.job_id = ginfo.info1.job_id;
3557 info1.printer_name = ginfo.info1.printer_name;
3558 info1.server_name = ginfo.info1.server_name;
3559 info1.user_name = ginfo.info1.user_name;
3560 info1.document_name = new_document_name;
3561 info1.data_type = ginfo.info1.data_type;
3562 info1.text_status = ginfo.info1.text_status;
3563 info1.status = ginfo.info1.status;
3564 info1.priority = ginfo.info1.priority;
3565 info1.position = ginfo.info1.position;
3566 info1.total_pages = ginfo.info1.total_pages;
3567 info1.pages_printed = ginfo.info1.pages_printed;
3568 info1.submitted = ginfo.info1.submitted;
3571 ctr.info.info1 = &info1;
3573 torture_assert(tctx,
3574 test_SetJob(tctx, b, handle, info[i].info1.job_id, &ctr, 0),
3575 "failed to call test_SetJob level 1");
3577 torture_assert(tctx,
3578 test_GetJob_args(tctx, b, handle, info[i].info1.job_id, 1, &ginfo),
3579 "failed to call test_GetJob");
3581 if (strequal(ginfo.info1.document_name, document_name)) {
3582 torture_warning(tctx,
3583 "document_name did *NOT* change from '%s' to '%s'\n",
3584 document_name, new_document_name);
3588 for (i=0; i < num_jobs; i++) {
3589 if (!test_SetJob(tctx, b, handle, info[i].info1.job_id, NULL, SPOOLSS_JOB_CONTROL_PAUSE)) {
3590 torture_warning(tctx, "failed to pause printjob\n");
3592 if (!test_SetJob(tctx, b, handle, info[i].info1.job_id, NULL, SPOOLSS_JOB_CONTROL_RESUME)) {
3593 torture_warning(tctx, "failed to resume printjob\n");
3600 static bool test_DoPrintTest(struct torture_context *tctx,
3601 struct dcerpc_binding_handle *b,
3602 struct policy_handle *handle)
3605 uint32_t num_jobs = 8;
3609 torture_comment(tctx, "Testing real print operations\n");
3611 job_ids = talloc_zero_array(tctx, uint32_t, num_jobs);
3613 for (i=0; i < num_jobs; i++) {
3614 ret &= test_DoPrintTest_add_one_job(tctx, b, handle, "TorturePrintJob", &job_ids[i]);
3617 for (i=0; i < num_jobs; i++) {
3618 ret &= test_SetJob(tctx, b, handle, job_ids[i], NULL, SPOOLSS_JOB_CONTROL_DELETE);
3621 for (i=0; i < num_jobs; i++) {
3622 ret &= test_DoPrintTest_add_one_job_v4(tctx, b, handle, "TorturePrintJob v4", &job_ids[i]);
3625 for (i=0; i < num_jobs; i++) {
3626 ret &= test_SetJob(tctx, b, handle, job_ids[i], NULL, SPOOLSS_JOB_CONTROL_DELETE);
3630 torture_comment(tctx, "real print operations test succeeded\n\n");
3636 static bool test_DoPrintTest_extended(struct torture_context *tctx,
3637 struct dcerpc_binding_handle *b,
3638 struct policy_handle *handle)
3641 uint32_t num_jobs = 8;
3644 torture_comment(tctx, "Testing real print operations (extended)\n");
3646 job_ids = talloc_zero_array(tctx, uint32_t, num_jobs);
3648 for (i=0; i < num_jobs; i++) {
3649 ret &= test_DoPrintTest_add_one_job(tctx, b, handle, "TorturePrintJob", &job_ids[i]);
3652 ret &= test_DoPrintTest_check_jobs(tctx, b, handle, num_jobs, job_ids);
3654 for (i=0; i < num_jobs; i++) {
3655 ret &= test_SetJob(tctx, b, handle, job_ids[i], NULL, SPOOLSS_JOB_CONTROL_DELETE);
3659 torture_comment(tctx, "real print operations (extended) test succeeded\n\n");
3665 static bool test_JobPrintProperties_equal(struct torture_context *tctx,
3666 struct spoolss_PrintPropertyValue *got,
3667 struct spoolss_PrintNamedProperty *exp)
3669 torture_assert_int_equal(tctx,
3671 exp->propertyValue.ePropertyType,
3674 switch (exp->propertyValue.ePropertyType) {
3675 case kRpcPropertyTypeString:
3676 torture_assert_str_equal(tctx,
3677 got->value.propertyString,
3678 exp->propertyValue.value.propertyString,
3681 case kRpcPropertyTypeInt32:
3682 torture_assert_int_equal(tctx,
3683 got->value.propertyInt32,
3684 exp->propertyValue.value.propertyInt32,
3687 case kRpcPropertyTypeInt64:
3688 torture_assert_u64_equal(tctx,
3689 got->value.propertyInt64,
3690 exp->propertyValue.value.propertyInt64,
3693 case kRpcPropertyTypeByte:
3694 torture_assert_int_equal(tctx,
3695 got->value.propertyByte,
3696 exp->propertyValue.value.propertyByte,
3699 case kRpcPropertyTypeBuffer:
3700 torture_assert_int_equal(tctx,
3701 got->value.propertyBlob.cbBuf,
3702 exp->propertyValue.value.propertyBlob.cbBuf,
3703 "propertyBlob.cbBuf");
3704 torture_assert_mem_equal(tctx,
3705 got->value.propertyBlob.pBuf,
3706 exp->propertyValue.value.propertyBlob.pBuf,
3707 exp->propertyValue.value.propertyBlob.cbBuf,
3708 "propertyBlob.pBuf");
3717 static bool test_JobPrintProperties(struct torture_context *tctx,
3718 struct dcerpc_binding_handle *b,
3719 struct policy_handle *handle,
3722 struct spoolss_PrintNamedProperty in;
3723 struct spoolss_PrintPropertyValue out;
3725 DATA_BLOB blob = data_blob_string_const("blob");
3727 const char *property_name;
3728 enum spoolss_EPrintPropertyType type;
3729 union spoolss_PrintPropertyValueUnion value;
3730 WERROR expected_result;
3733 .property_name = "torture_property_string",
3734 .type = kRpcPropertyTypeString,
3735 .value.propertyString = "torture_property_value_string",
3737 .property_name = "torture_property_int32",
3738 .type = kRpcPropertyTypeInt32,
3739 .value.propertyInt32 = 42,
3741 .property_name = "torture_property_int64",
3742 .type = kRpcPropertyTypeInt64,
3743 .value.propertyInt64 = 0xaffe,
3745 .property_name = "torture_property_byte",
3746 .type = kRpcPropertyTypeByte,
3747 .value.propertyByte = 0xab,
3749 .property_name = "torture_property_buffer",
3750 .type = kRpcPropertyTypeBuffer,
3751 .value.propertyBlob.cbBuf = blob.length,
3752 .value.propertyBlob.pBuf = blob.data,
3756 torture_assert(tctx,
3757 test_JobPropertiesEnum(tctx, b, handle, job_id),
3758 "failed to enum properties");
3760 for (i=0; i <ARRAY_SIZE(tests); i++) {
3762 in.propertyName = tests[i].property_name;
3763 in.propertyValue.ePropertyType = tests[i].type;
3764 in.propertyValue.value = tests[i].value;
3766 torture_assert(tctx,
3767 test_JobPropertySet(tctx, b, handle, job_id, &in),
3768 "failed to set property");
3770 torture_assert(tctx,
3771 test_JobPropertyGetValue(tctx, b, handle, job_id, in.propertyName, &out),
3772 "failed to get property");
3774 torture_assert(tctx,
3775 test_JobPrintProperties_equal(tctx, &out, &in),
3776 "property unequal");
3778 torture_assert(tctx,
3779 test_JobPropertiesEnum(tctx, b, handle, job_id),
3780 "failed to enum properties");
3782 torture_assert(tctx,
3783 test_JobPropertyDelete(tctx, b, handle, job_id, in.propertyName),
3784 "failed to delete job property");
3787 torture_assert(tctx,
3788 test_JobPropertiesEnum(tctx, b, handle, job_id),
3789 "failed to enum properties");
3794 static bool test_DoPrintTest_properties(struct torture_context *tctx,
3795 struct dcerpc_binding_handle *b,
3796 struct policy_handle *handle)
3798 uint32_t num_jobs = 8;
3801 torture_comment(tctx, "Testing real print operations (properties)\n");
3803 job_ids = talloc_zero_array(tctx, uint32_t, num_jobs);
3805 for (i=0; i < num_jobs; i++) {
3806 torture_assert(tctx,
3807 test_DoPrintTest_add_one_job(tctx, b, handle, "TorturePrintJob", &job_ids[i]),
3808 "failed to create print job");
3811 for (i=0; i < num_jobs; i++) {
3812 torture_assert(tctx,
3813 test_JobPrintProperties(tctx, b, handle, job_ids[i]),
3814 "failed to test job properties");
3818 for (i=0; i < num_jobs; i++) {
3819 torture_assert(tctx,
3820 test_SetJob(tctx, b, handle, job_ids[i], NULL, SPOOLSS_JOB_CONTROL_DELETE),
3821 "failed to delete printjob");
3824 torture_comment(tctx, "real print operations (properties) test succeeded\n\n");
3829 static bool test_PausePrinter(struct torture_context *tctx,
3830 struct dcerpc_binding_handle *b,
3831 struct policy_handle *handle)
3834 struct spoolss_SetPrinter r;
3835 struct spoolss_SetPrinterInfoCtr info_ctr;
3836 struct spoolss_DevmodeContainer devmode_ctr;
3837 struct sec_desc_buf secdesc_ctr;
3840 info_ctr.info.info0 = NULL;
3842 ZERO_STRUCT(devmode_ctr);
3843 ZERO_STRUCT(secdesc_ctr);
3845 r.in.handle = handle;
3846 r.in.info_ctr = &info_ctr;
3847 r.in.devmode_ctr = &devmode_ctr;
3848 r.in.secdesc_ctr = &secdesc_ctr;
3849 r.in.command = SPOOLSS_PRINTER_CONTROL_PAUSE;
3851 torture_comment(tctx, "Testing SetPrinter: SPOOLSS_PRINTER_CONTROL_PAUSE\n");
3853 status = dcerpc_spoolss_SetPrinter_r(b, tctx, &r);
3855 torture_assert_ntstatus_ok(tctx, status, "SetPrinter failed");
3857 torture_assert_werr_ok(tctx, r.out.result, "SetPrinter failed");
3862 static bool test_ResumePrinter(struct torture_context *tctx,
3863 struct dcerpc_binding_handle *b,
3864 struct policy_handle *handle)
3867 struct spoolss_SetPrinter r;
3868 struct spoolss_SetPrinterInfoCtr info_ctr;
3869 struct spoolss_DevmodeContainer devmode_ctr;
3870 struct sec_desc_buf secdesc_ctr;
3873 info_ctr.info.info0 = NULL;
3875 ZERO_STRUCT(devmode_ctr);
3876 ZERO_STRUCT(secdesc_ctr);
3878 r.in.handle = handle;
3879 r.in.info_ctr = &info_ctr;
3880 r.in.devmode_ctr = &devmode_ctr;
3881 r.in.secdesc_ctr = &secdesc_ctr;
3882 r.in.command = SPOOLSS_PRINTER_CONTROL_RESUME;
3884 torture_comment(tctx, "Testing SetPrinter: SPOOLSS_PRINTER_CONTROL_RESUME\n");
3886 status = dcerpc_spoolss_SetPrinter_r(b, tctx, &r);
3888 torture_assert_ntstatus_ok(tctx, status, "SetPrinter failed");
3890 torture_assert_werr_ok(tctx, r.out.result, "SetPrinter failed");
3895 static bool test_printer_purge(struct torture_context *tctx,
3896 struct dcerpc_binding_handle *b,
3897 struct policy_handle *handle)
3900 struct spoolss_SetPrinter r;
3901 struct spoolss_SetPrinterInfoCtr info_ctr;
3902 struct spoolss_DevmodeContainer devmode_ctr;
3903 struct sec_desc_buf secdesc_ctr;
3906 info_ctr.info.info0 = NULL;
3908 ZERO_STRUCT(devmode_ctr);
3909 ZERO_STRUCT(secdesc_ctr);
3911 r.in.handle = handle;
3912 r.in.info_ctr = &info_ctr;
3913 r.in.devmode_ctr = &devmode_ctr;
3914 r.in.secdesc_ctr = &secdesc_ctr;
3915 r.in.command = SPOOLSS_PRINTER_CONTROL_PURGE;
3917 torture_comment(tctx, "Testing SetPrinter: SPOOLSS_PRINTER_CONTROL_PURGE\n");
3919 status = dcerpc_spoolss_SetPrinter_r(b, tctx, &r);
3920 torture_assert_ntstatus_ok(tctx, status, "SetPrinter failed");
3921 torture_assert_werr_ok(tctx, r.out.result, "SetPrinter failed");
3926 static bool test_GetPrinterData_checktype(struct torture_context *tctx,
3927 struct dcerpc_binding_handle *b,
3928 struct policy_handle *handle,
3929 const char *value_name,
3930 enum winreg_Type *expected_type,
3931 enum winreg_Type *type_p,
3936 struct spoolss_GetPrinterData r;
3938 enum winreg_Type type;
3939 union spoolss_PrinterData data;
3941 r.in.handle = handle;
3942 r.in.value_name = value_name;
3944 r.out.needed = &needed;
3946 r.out.data = talloc_zero_array(tctx, uint8_t, r.in.offered);
3948 torture_comment(tctx, "Testing GetPrinterData(%s)\n", r.in.value_name);
3950 status = dcerpc_spoolss_GetPrinterData_r(b, tctx, &r);
3951 torture_assert_ntstatus_ok(tctx, status, "GetPrinterData failed");
3953 if (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA)) {
3954 if (expected_type) {
3955 torture_assert_int_equal(tctx, type, *expected_type, "unexpected type");
3957 r.in.offered = needed;
3958 r.out.data = talloc_zero_array(tctx, uint8_t, r.in.offered);
3959 status = dcerpc_spoolss_GetPrinterData_r(b, tctx, &r);
3960 torture_assert_ntstatus_ok(tctx, status, "GetPrinterData failed");
3963 torture_assert_werr_ok(tctx, r.out.result,
3964 talloc_asprintf(tctx, "GetPrinterData(%s) failed", r.in.value_name));
3966 CHECK_NEEDED_SIZE_LEVEL(spoolss_PrinterData, &data, type, needed, 1);
3973 *data_p = r.out.data;
3983 static bool test_GetPrinterData(struct torture_context *tctx,
3984 struct dcerpc_binding_handle *b,
3985 struct policy_handle *handle,
3986 const char *value_name,
3987 enum winreg_Type *type_p,
3991 return test_GetPrinterData_checktype(tctx, b, handle, value_name,
3992 NULL, type_p, data_p, needed_p);
3995 static bool test_GetPrinterDataEx_checktype(struct torture_context *tctx,
3996 struct dcerpc_pipe *p,
3997 struct policy_handle *handle,
3998 const char *key_name,
3999 const char *value_name,
4000 enum winreg_Type *expected_type,
4001 enum winreg_Type *type_p,
4006 struct spoolss_GetPrinterDataEx r;
4007 enum winreg_Type type;
4009 union spoolss_PrinterData data;
4010 struct dcerpc_binding_handle *b = p->binding_handle;
4012 r.in.handle = handle;
4013 r.in.key_name = key_name;
4014 r.in.value_name = value_name;
4017 r.out.needed = &needed;
4018 r.out.data = talloc_zero_array(tctx, uint8_t, r.in.offered);
4020 torture_comment(tctx, "Testing GetPrinterDataEx(%s - %s)\n",
4021 r.in.key_name, r.in.value_name);
4023 status = dcerpc_spoolss_GetPrinterDataEx_r(b, tctx, &r);
4024 if (!NT_STATUS_IS_OK(status)) {
4025 if (NT_STATUS_EQUAL(status, NT_STATUS_RPC_PROCNUM_OUT_OF_RANGE)) {
4026 torture_skip(tctx, "GetPrinterDataEx not supported by server\n");
4028 torture_assert_ntstatus_ok(tctx, status, "GetPrinterDataEx failed");
4031 if (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA)) {
4032 if (expected_type) {
4033 torture_assert_int_equal(tctx, type, *expected_type, "unexpected type");
4035 r.in.offered = needed;
4036 r.out.data = talloc_zero_array(tctx, uint8_t, r.in.offered);
4037 status = dcerpc_spoolss_GetPrinterDataEx_r(b, tctx, &r);
4038 torture_assert_ntstatus_ok(tctx, status, "GetPrinterDataEx failed");
4041 torture_assert_werr_ok(tctx, r.out.result,
4042 talloc_asprintf(tctx, "GetPrinterDataEx(%s - %s) failed", r.in.key_name, r.in.value_name));
4044 CHECK_NEEDED_SIZE_LEVEL(spoolss_PrinterData, &data, type, needed, 1);
4051 *data_p = r.out.data;
4061 static bool test_GetPrinterDataEx(struct torture_context *tctx,
4062 struct dcerpc_pipe *p,
4063 struct policy_handle *handle,
4064 const char *key_name,
4065 const char *value_name,
4066 enum winreg_Type *type_p,
4070 return test_GetPrinterDataEx_checktype(tctx, p, handle, key_name, value_name,
4071 NULL, type_p, data_p, needed_p);
4074 static bool test_get_environment(struct torture_context *tctx,
4075 struct dcerpc_binding_handle *b,
4076 struct policy_handle *handle,
4077 const char **architecture)
4080 enum winreg_Type type;
4084 torture_assert(tctx,
4085 test_GetPrinterData(tctx, b, handle, "Architecture", &type, &data, &needed),
4086 "failed to get Architecture");
4088 torture_assert_int_equal(tctx, type, REG_SZ, "unexpected type");
4090 blob = data_blob_const(data, needed);
4091 *architecture = reg_val_data_string(tctx, REG_SZ, blob);
4096 static bool test_GetPrinterData_list(struct torture_context *tctx,
4099 struct test_spoolss_context *ctx =
4100 talloc_get_type_abort(private_data, struct test_spoolss_context);
4101 struct dcerpc_pipe *p = ctx->spoolss_pipe;
4102 struct dcerpc_binding_handle *b = p->binding_handle;
4103 const char *list[] = {
4107 /* "NetPopup", not on w2k8 */
4108 /* "NetPopupToComputer", not on w2k8 */
4111 "DefaultSpoolDirectory",
4115 /* "OSVersionEx", not on s3 */
4120 for (i=0; i < ARRAY_SIZE(list); i++) {
4121 enum winreg_Type type = REG_NONE;
4122 enum winreg_Type type_ex1 = REG_NONE;
4123 enum winreg_Type type_ex2 = REG_NONE;
4125 uint8_t *data_ex1 = NULL;
4126 uint8_t *data_ex2 = NULL;
4128 uint32_t needed_ex1 = 0;
4129 uint32_t needed_ex2 = 0;
4131 torture_assert(tctx, test_GetPrinterData(tctx, b, &ctx->server_handle, list[i], &type, &data, &needed),
4132 talloc_asprintf(tctx, "GetPrinterData failed on %s\n", list[i]));
4133 torture_assert(tctx, test_GetPrinterDataEx(tctx, p, &ctx->server_handle, "random_string", list[i], &type_ex1, &data_ex1, &needed_ex1),
4134 talloc_asprintf(tctx, "GetPrinterDataEx failed on %s\n", list[i]));
4135 torture_assert(tctx, test_GetPrinterDataEx(tctx, p, &ctx->server_handle, "", list[i], &type_ex2, &data_ex2, &needed_ex2),
4136 talloc_asprintf(tctx, "GetPrinterDataEx failed on %s\n", list[i]));
4137 torture_assert_int_equal(tctx, type, type_ex1, "type mismatch");
4138 torture_assert_int_equal(tctx, type, type_ex2, "type mismatch");
4139 torture_assert_int_equal(tctx, needed, needed_ex1, "needed mismatch");
4140 torture_assert_int_equal(tctx, needed, needed_ex2, "needed mismatch");
4141 torture_assert_mem_equal(tctx, data, data_ex1, needed, "data mismatch");
4142 torture_assert_mem_equal(tctx, data, data_ex2, needed, "data mismatch");
4148 static bool test_EnumPrinterData(struct torture_context *tctx,
4149 struct dcerpc_pipe *p,
4150 struct policy_handle *handle,
4151 uint32_t enum_index,
4152 uint32_t value_offered,
4153 uint32_t data_offered,
4154 enum winreg_Type *type_p,
4155 uint32_t *value_needed_p,
4156 uint32_t *data_needed_p,
4157 const char **value_name_p,
4161 struct spoolss_EnumPrinterData r;
4162 uint32_t data_needed;
4163 uint32_t value_needed;
4164 enum winreg_Type type;
4165 struct dcerpc_binding_handle *b = p->binding_handle;
4167 r.in.handle = handle;
4168 r.in.enum_index = enum_index;
4169 r.in.value_offered = value_offered;
4170 r.in.data_offered = data_offered;
4171 r.out.data_needed = &data_needed;
4172 r.out.value_needed = &value_needed;
4174 r.out.data = talloc_zero_array(tctx, uint8_t, r.in.data_offered);
4175 r.out.value_name = talloc_zero_array(tctx, const char, r.in.value_offered);
4177 torture_comment(tctx, "Testing EnumPrinterData(%d)\n", enum_index);
4179 torture_assert_ntstatus_ok(tctx,
4180 dcerpc_spoolss_EnumPrinterData_r(b, tctx, &r),
4181 "EnumPrinterData failed");
4186 if (value_needed_p) {
4187 *value_needed_p = value_needed;
4189 if (data_needed_p) {
4190 *data_needed_p = data_needed;
4193 *value_name_p = r.out.value_name;
4196 *data_p = r.out.data;
4199 *result_p = r.out.result;
4206 static bool test_EnumPrinterData_all(struct torture_context *tctx,
4207 struct dcerpc_pipe *p,
4208 struct policy_handle *handle)
4210 uint32_t enum_index = 0;
4211 enum winreg_Type type;
4212 uint32_t value_needed;
4213 uint32_t data_needed;
4215 const char *value_name;
4218 torture_comment(tctx, "Testing EnumPrinterData\n");
4221 torture_assert(tctx,
4222 test_EnumPrinterData(tctx, p, handle, enum_index, 0, 0,
4223 &type, &value_needed, &data_needed,
4224 &value_name, &data, &result),
4225 "EnumPrinterData failed");
4227 if (W_ERROR_EQUAL(result, WERR_NO_MORE_ITEMS)) {
4231 torture_assert(tctx,
4232 test_EnumPrinterData(tctx, p, handle, enum_index, value_needed, data_needed,
4233 &type, &value_needed, &data_needed,
4234 &value_name, &data, &result),
4235 "EnumPrinterData failed");
4237 if (W_ERROR_EQUAL(result, WERR_NO_MORE_ITEMS)) {
4243 } while (W_ERROR_IS_OK(result));
4245 torture_comment(tctx, "EnumPrinterData test succeeded\n");
4250 static bool test_EnumPrinterDataEx(struct torture_context *tctx,
4251 struct dcerpc_binding_handle *b,
4252 struct policy_handle *handle,
4253 const char *key_name,
4255 struct spoolss_PrinterEnumValues **info_p)
4257 struct spoolss_EnumPrinterDataEx r;
4258 struct spoolss_PrinterEnumValues *info;
4262 r.in.handle = handle;
4263 r.in.key_name = key_name;
4265 r.out.needed = &needed;
4266 r.out.count = &count;
4269 torture_comment(tctx, "Testing EnumPrinterDataEx(%s)\n", key_name);
4271 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_EnumPrinterDataEx_r(b, tctx, &r),
4272 "EnumPrinterDataEx failed");
4273 if (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA)) {
4274 r.in.offered = needed;
4275 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_EnumPrinterDataEx_r(b, tctx, &r),
4276 "EnumPrinterDataEx failed");
4279 torture_assert_werr_ok(tctx, r.out.result, "EnumPrinterDataEx failed");
4281 CHECK_NEEDED_SIZE_ENUM(spoolss_EnumPrinterDataEx, info, count, needed, 1);
4293 static bool test_SetPrinterData(struct torture_context *tctx,
4294 struct dcerpc_binding_handle *b,
4295 struct policy_handle *handle,
4296 const char *value_name,
4297 enum winreg_Type type,
4300 static bool test_DeletePrinterData(struct torture_context *tctx,
4301 struct dcerpc_binding_handle *b,
4302 struct policy_handle *handle,
4303 const char *value_name);
4305 static bool test_EnumPrinterData_consistency(struct torture_context *tctx,
4306 struct dcerpc_pipe *p,
4307 struct policy_handle *handle)
4310 struct spoolss_PrinterEnumValues *info;
4312 uint32_t value_needed, data_needed;
4313 uint32_t value_offered, data_offered;
4315 struct dcerpc_binding_handle *b = p->binding_handle;
4317 enum winreg_Type type;
4320 torture_comment(tctx, "Testing EnumPrinterData vs EnumPrinterDataEx consistency\n");
4322 torture_assert(tctx, push_reg_sz(tctx, &blob, "torture_data1"), "");
4325 torture_assert(tctx,
4326 test_SetPrinterData(tctx, b, handle, "torture_value1", type, blob.data, blob.length),
4327 "SetPrinterData failed");
4329 blob = data_blob_string_const("torture_data2");
4331 torture_assert(tctx,
4332 test_SetPrinterData(tctx, b, handle, "torture_value2", REG_BINARY, blob.data, blob.length),
4333 "SetPrinterData failed");
4335 blob = data_blob_talloc(tctx, NULL, 4);
4336 SIVAL(blob.data, 0, 0x11223344);
4338 torture_assert(tctx,
4339 test_SetPrinterData(tctx, b, handle, "torture_value3", type, blob.data, blob.length),
4340 "SetPrinterData failed");
4342 torture_assert(tctx,
4343 test_EnumPrinterDataEx(tctx, b, handle, "PrinterDriverData", &count, &info),
4344 "failed to call EnumPrinterDataEx");
4346 /* get the max sizes for value and data */
4348 torture_assert(tctx,
4349 test_EnumPrinterData(tctx, p, handle, 0, 0, 0,
4350 NULL, &value_needed, &data_needed,
4351 NULL, NULL, &result),
4352 "EnumPrinterData failed");
4353 torture_assert_werr_ok(tctx, result, "unexpected result");
4355 /* check if the reply from the EnumPrinterData really matches max values */
4357 for (i=0; i < count; i++) {
4358 if (info[i].value_name_len > value_needed) {
4360 talloc_asprintf(tctx,
4361 "EnumPrinterDataEx gave a reply with value length %d which is larger then expected max value length %d from EnumPrinterData",
4362 info[i].value_name_len, value_needed));
4364 if (info[i].data_length > data_needed) {
4366 talloc_asprintf(tctx,
4367 "EnumPrinterDataEx gave a reply with data length %d which is larger then expected max data length %d from EnumPrinterData",
4368 info[i].data_length, data_needed));
4372 /* assuming that both EnumPrinterData and EnumPrinterDataEx do either
4373 * sort or not sort the replies by value name, we should be able to do
4374 * the following entry comparison */
4376 data_offered = data_needed;
4377 value_offered = value_needed;
4379 for (i=0; i < count; i++) {
4381 const char *value_name;
4384 torture_assert(tctx,
4385 test_EnumPrinterData(tctx, p, handle, i, value_offered, data_offered,
4386 &type, &value_needed, &data_needed,
4387 &value_name, &data, &result),
4388 "EnumPrinterData failed");
4390 if (i -1 == count) {
4391 torture_assert_werr_equal(tctx, result, WERR_NO_MORE_ITEMS,
4392 "unexpected result");
4395 torture_assert_werr_ok(tctx, result, "unexpected result");
4398 torture_assert_int_equal(tctx, type, info[i].type, "type mismatch");
4399 torture_assert_int_equal(tctx, value_needed, info[i].value_name_len, "value name length mismatch");
4400 torture_assert_str_equal(tctx, value_name, info[i].value_name, "value name mismatch");
4401 torture_assert_int_equal(tctx, data_needed, info[i].data_length, "data length mismatch");
4402 torture_assert_mem_equal(tctx, data, info[i].data->data, info[i].data_length, "data mismatch");
4405 torture_assert(tctx,
4406 test_DeletePrinterData(tctx, b, handle, "torture_value1"),
4407 "DeletePrinterData failed");
4408 torture_assert(tctx,
4409 test_DeletePrinterData(tctx, b, handle, "torture_value2"),
4410 "DeletePrinterData failed");
4411 torture_assert(tctx,
4412 test_DeletePrinterData(tctx, b, handle, "torture_value3"),
4413 "DeletePrinterData failed");
4415 torture_comment(tctx, "EnumPrinterData vs EnumPrinterDataEx consistency test succeeded\n\n");
4420 static bool test_DeletePrinterData(struct torture_context *tctx,
4421 struct dcerpc_binding_handle *b,
4422 struct policy_handle *handle,
4423 const char *value_name)
4426 struct spoolss_DeletePrinterData r;
4428 r.in.handle = handle;
4429 r.in.value_name = value_name;
4431 torture_comment(tctx, "Testing DeletePrinterData(%s)\n",
4434 status = dcerpc_spoolss_DeletePrinterData_r(b, tctx, &r);
4436 torture_assert_ntstatus_ok(tctx, status, "DeletePrinterData failed");
4437 torture_assert_werr_ok(tctx, r.out.result, "DeletePrinterData failed");
4442 static bool test_DeletePrinterDataEx(struct torture_context *tctx,
4443 struct dcerpc_binding_handle *b,
4444 struct policy_handle *handle,
4445 const char *key_name,
4446 const char *value_name)
4448 struct spoolss_DeletePrinterDataEx r;
4450 r.in.handle = handle;
4451 r.in.key_name = key_name;
4452 r.in.value_name = value_name;
4454 torture_comment(tctx, "Testing DeletePrinterDataEx(%s - %s)\n",
4455 r.in.key_name, r.in.value_name);
4457 torture_assert_ntstatus_ok(tctx,
4458 dcerpc_spoolss_DeletePrinterDataEx_r(b, tctx, &r),
4459 "DeletePrinterDataEx failed");
4460 torture_assert_werr_ok(tctx, r.out.result,
4461 "DeletePrinterDataEx failed");
4466 static bool test_DeletePrinterKey(struct torture_context *tctx,
4467 struct dcerpc_binding_handle *b,
4468 struct policy_handle *handle,
4469 const char *key_name)
4471 struct spoolss_DeletePrinterKey r;
4473 r.in.handle = handle;
4474 r.in.key_name = key_name;
4476 torture_comment(tctx, "Testing DeletePrinterKey(%s)\n", r.in.key_name);
4478 if (strequal(key_name, "") && !torture_setting_bool(tctx, "dangerous", false)) {
4479 torture_skip(tctx, "not wiping out printer registry - enable dangerous tests to use\n");
4483 torture_assert_ntstatus_ok(tctx,
4484 dcerpc_spoolss_DeletePrinterKey_r(b, tctx, &r),
4485 "DeletePrinterKey failed");
4486 torture_assert_werr_ok(tctx, r.out.result,
4487 "DeletePrinterKey failed");
4492 static bool test_winreg_OpenHKLM(struct torture_context *tctx,
4493 struct dcerpc_binding_handle *b,
4494 struct policy_handle *handle)
4496 struct winreg_OpenHKLM r;
4498 r.in.system_name = NULL;
4499 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
4500 r.out.handle = handle;
4502 torture_comment(tctx, "Testing winreg_OpenHKLM\n");
4504 torture_assert_ntstatus_ok(tctx, dcerpc_winreg_OpenHKLM_r(b, tctx, &r), "OpenHKLM failed");
4505 torture_assert_werr_ok(tctx, r.out.result, "OpenHKLM failed");
4510 static void init_winreg_String(struct winreg_String *name, const char *s)
4514 name->name_len = 2 * (strlen_m(s) + 1);
4515 name->name_size = name->name_len;
4518 name->name_size = 0;
4522 static bool test_winreg_OpenKey_opts(struct torture_context *tctx,
4523 struct dcerpc_binding_handle *b,
4524 struct policy_handle *hive_handle,
4525 const char *keyname,
4527 struct policy_handle *key_handle)
4529 struct winreg_OpenKey r;
4531 r.in.parent_handle = hive_handle;
4532 init_winreg_String(&r.in.keyname, keyname);
4533 r.in.options = options;
4534 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
4535 r.out.handle = key_handle;
4537 torture_comment(tctx, "Testing winreg_OpenKey(%s)\n", keyname);
4539 torture_assert_ntstatus_ok(tctx, dcerpc_winreg_OpenKey_r(b, tctx, &r), "OpenKey failed");
4540 torture_assert_werr_ok(tctx, r.out.result, "OpenKey failed");
4545 static bool test_winreg_OpenKey(struct torture_context *tctx,
4546 struct dcerpc_binding_handle *b,
4547 struct policy_handle *hive_handle,
4548 const char *keyname,
4549 struct policy_handle *key_handle)
4551 return test_winreg_OpenKey_opts(tctx, b, hive_handle, keyname,
4552 REG_OPTION_NON_VOLATILE, key_handle);
4555 static bool test_winreg_CloseKey(struct torture_context *tctx,
4556 struct dcerpc_binding_handle *b,
4557 struct policy_handle *handle)
4559 struct winreg_CloseKey r;
4561 r.in.handle = handle;
4562 r.out.handle = handle;
4564 torture_comment(tctx, "Testing winreg_CloseKey\n");
4566 torture_assert_ntstatus_ok(tctx, dcerpc_winreg_CloseKey_r(b, tctx, &r), "CloseKey failed");
4567 torture_assert_werr_ok(tctx, r.out.result, "CloseKey failed");
4572 bool test_winreg_QueryValue(struct torture_context *tctx,
4573 struct dcerpc_binding_handle *b,
4574 struct policy_handle *handle,
4575 const char *value_name,
4576 enum winreg_Type *type_p,
4577 uint32_t *data_size_p,
4578 uint32_t *data_length_p,
4581 struct winreg_QueryValue r;
4582 enum winreg_Type type = REG_NONE;
4583 uint32_t data_size = 0;
4584 uint32_t data_length = 0;
4585 struct winreg_String valuename;
4586 uint8_t *data = NULL;
4588 init_winreg_String(&valuename, value_name);
4590 data = talloc_zero_array(tctx, uint8_t, 0);
4592 r.in.handle = handle;
4593 r.in.value_name = &valuename;
4595 r.in.data_size = &data_size;
4596 r.in.data_length = &data_length;
4600 r.out.data_size = &data_size;
4601 r.out.data_length = &data_length;
4603 torture_comment(tctx, "Testing winreg_QueryValue(%s)\n", value_name);
4605 torture_assert_ntstatus_ok(tctx, dcerpc_winreg_QueryValue_r(b, tctx, &r), "QueryValue failed");
4606 if (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA)) {
4607 *r.in.data_size = *r.out.data_size;
4608 data = talloc_zero_array(tctx, uint8_t, *r.in.data_size);
4611 torture_assert_ntstatus_ok(tctx, dcerpc_winreg_QueryValue_r(b, tctx, &r), "QueryValue failed");
4613 torture_assert_werr_ok(tctx, r.out.result, "QueryValue failed");
4616 *type_p = *r.out.type;
4619 *data_size_p = *r.out.data_size;
4621 if (data_length_p) {
4622 *data_length_p = *r.out.data_length;
4625 *data_p = r.out.data;
4631 static bool test_winreg_query_printerdata(struct torture_context *tctx,
4632 struct dcerpc_binding_handle *b,
4633 struct policy_handle *handle,
4634 const char *printer_name,
4635 const char *key_name,
4636 const char *value_name,
4637 enum winreg_Type *w_type,
4642 const char *printer_key;
4643 struct policy_handle key_handle;
4645 printer_key = talloc_asprintf(tctx, "%s\\%s\\%s",
4646 TOP_LEVEL_PRINT_PRINTERS_KEY, printer_name, key_name);
4648 torture_assert(tctx,
4649 test_winreg_OpenKey(tctx, b, handle, printer_key, &key_handle), "");
4651 torture_assert(tctx,
4652 test_winreg_QueryValue(tctx, b, &key_handle, value_name, w_type, w_size, w_length, w_data), "");
4654 torture_assert(tctx,
4655 test_winreg_CloseKey(tctx, b, &key_handle), "");
4660 static bool test_GetForm_winreg(struct torture_context *tctx,
4661 struct dcerpc_binding_handle *b,
4662 struct policy_handle *handle,
4663 const char *key_name,
4664 const char *form_name,
4665 enum winreg_Type *w_type,
4670 struct policy_handle key_handle;
4672 torture_assert(tctx,
4673 test_winreg_OpenKey(tctx, b, handle, key_name, &key_handle), "");
4675 torture_assert(tctx,
4676 test_winreg_QueryValue(tctx, b, &key_handle, form_name, w_type, w_size, w_length, w_data), "");
4678 torture_assert(tctx,
4679 test_winreg_CloseKey(tctx, b, &key_handle), "");
4684 static bool test_winreg_symbolic_link(struct torture_context *tctx,
4685 struct dcerpc_binding_handle *b,
4686 struct policy_handle *handle,
4687 const char *symlink_keyname,
4688 const char *symlink_destination)
4690 /* check if the first key is a symlink to the second key */
4692 enum winreg_Type w_type;
4696 struct policy_handle key_handle;
4700 if (torture_setting_bool(tctx, "samba3", false)) {
4701 torture_skip(tctx, "skip winreg symlink test against samba");
4704 torture_assert(tctx,
4705 test_winreg_OpenKey_opts(tctx, b, handle, symlink_keyname, REG_OPTION_OPEN_LINK, &key_handle),
4706 "failed to open key link");
4708 torture_assert(tctx,
4709 test_winreg_QueryValue(tctx, b, &key_handle,
4710 "SymbolicLinkValue",
4711 &w_type, &w_size, &w_length, &w_data),
4712 "failed to query for 'SymbolicLinkValue' attribute");
4714 torture_assert_int_equal(tctx, w_type, REG_LINK, "unexpected type");
4716 blob = data_blob(w_data, w_size);
4717 str = reg_val_data_string(tctx, REG_SZ, blob);
4719 torture_assert_str_equal(tctx, str, symlink_destination, "unexpected symlink target string");
4721 torture_assert(tctx,
4722 test_winreg_CloseKey(tctx, b, &key_handle),
4723 "failed to close key link");
4728 static const char *strip_unc(const char *unc)
4736 if (unc[0] == '\\' && unc[1] == '\\') {
4740 name = strchr(unc, '\\');
4748 static bool test_GetPrinterInfo_winreg(struct torture_context *tctx,
4749 struct dcerpc_binding_handle *b,
4750 struct policy_handle *handle,
4751 const char *printer_name,
4752 struct dcerpc_binding_handle *winreg_handle,
4753 struct policy_handle *hive_handle)
4755 union spoolss_PrinterInfo info;
4756 const char *keys[] = {
4757 TOP_LEVEL_CONTROL_PRINTERS_KEY,
4758 TOP_LEVEL_PRINT_PRINTERS_KEY
4761 const char *printername, *sharename;
4763 torture_comment(tctx, "Testing Printer Info and winreg consistency\n");
4765 torture_assert(tctx,
4766 test_GetPrinter_level(tctx, b, handle, 2, &info),
4767 "failed to get printer info level 2");
4769 printername = strip_unc(info.info2.printername);
4770 sharename = strip_unc(info.info2.sharename);
4772 #define test_sz(wname, iname) \
4776 enum winreg_Type w_type;\
4780 torture_assert(tctx,\
4781 test_winreg_QueryValue(tctx, winreg_handle, &key_handle, wname,\
4782 &w_type, &w_size, &w_length, &w_data),\
4783 "failed to query winreg");\
4784 torture_assert_int_equal(tctx, w_type, REG_SZ, "unexpected type");\
4785 blob = data_blob(w_data, w_size);\
4786 str = reg_val_data_string(tctx, REG_SZ, blob);\
4787 if (w_size == 2 && iname == NULL) {\
4788 /*torture_comment(tctx, "%s: \"\", %s: (null)\n", #wname, #iname);\ */\
4790 torture_assert_str_equal(tctx, str, iname,\
4791 talloc_asprintf(tctx, "%s - %s mismatch", #wname, #iname));\
4795 #define test_dword(wname, iname) \
4798 enum winreg_Type w_type;\
4802 torture_assert(tctx,\
4803 test_winreg_QueryValue(tctx, winreg_handle, &key_handle, wname,\
4804 &w_type, &w_size, &w_length, &w_data),\
4805 "failed to query winreg");\
4806 torture_assert_int_equal(tctx, w_type, REG_DWORD, "unexpected type");\
4807 torture_assert_int_equal(tctx, w_size, 4, "unexpected size");\
4808 torture_assert_int_equal(tctx, w_length, 4, "unexpected length");\
4809 value = IVAL(w_data, 0);\
4810 torture_assert_int_equal(tctx, value, iname,\
4811 talloc_asprintf(tctx, "%s - %s mismatch", #wname, #iname));\
4814 #define test_binary(wname, iname) \
4816 enum winreg_Type w_type;\
4820 torture_assert(tctx,\
4821 test_winreg_QueryValue(tctx, winreg_handle, &key_handle, wname,\
4822 &w_type, &w_size, &w_length, &w_data),\
4823 "failed to query winreg");\
4824 torture_assert_int_equal(tctx, w_type, REG_BINARY, "unexpected type");\
4825 torture_assert_int_equal(tctx, w_size, iname.length, "unexpected length");\
4826 torture_assert_mem_equal(tctx, w_data, iname.data, w_size, \
4831 #define test_dm(wname, iname) \
4834 struct spoolss_DeviceMode dm;\
4835 enum ndr_err_code ndr_err;\
4836 enum winreg_Type w_type;\
4840 torture_assert(tctx,\
4841 test_winreg_QueryValue(tctx, winreg_handle, &key_handle, wname,\
4842 &w_type, &w_size, &w_length, &w_data),\
4843 "failed to query winreg");\
4844 torture_assert_int_equal(tctx, w_type, REG_BINARY, "unexpected type");\
4845 blob = data_blob(w_data, w_size);\
4846 ndr_err = ndr_pull_struct_blob(&blob, tctx, &dm,\
4847 (ndr_pull_flags_fn_t)ndr_pull_spoolss_DeviceMode);\
4848 torture_assert_ndr_success(tctx, ndr_err, "failed to unmarshall dm");\
4849 torture_assert(tctx, test_devicemode_equal(tctx, &dm, iname),\
4853 #define test_sd(wname, iname) \
4856 struct security_descriptor sd;\
4857 enum ndr_err_code ndr_err;\
4858 enum winreg_Type w_type;\
4862 torture_assert(tctx,\
4863 test_winreg_QueryValue(tctx, winreg_handle, &key_handle, wname,\
4864 &w_type, &w_size, &w_length, &w_data),\
4865 "failed to query winreg");\
4866 torture_assert_int_equal(tctx, w_type, REG_BINARY, "unexpected type");\
4867 blob = data_blob(w_data, w_size);\
4868 ndr_err = ndr_pull_struct_blob(&blob, tctx, &sd,\
4869 (ndr_pull_flags_fn_t)ndr_pull_security_descriptor);\
4870 torture_assert_ndr_success(tctx, ndr_err, "failed to unmarshall sd");\
4871 torture_assert(tctx, test_security_descriptor_equal(tctx, &sd, iname),\
4875 #define test_multi_sz(wname, iname) \
4878 const char **array;\
4879 enum winreg_Type w_type;\
4884 torture_assert(tctx,\
4885 test_winreg_QueryValue(tctx, winreg_handle, &key_handle, wname,\
4886 &w_type, &w_size, &w_length, &w_data),\
4887 "failed to query winreg");\
4888 torture_assert_int_equal(tctx, w_type, REG_MULTI_SZ, "unexpected type");\
4889 blob = data_blob(w_data, w_size);\
4890 torture_assert(tctx, \
4891 pull_reg_multi_sz(tctx, &blob, &array),\
4892 "failed to pull multi sz");\
4893 for (i=0; array[i] != NULL; i++) {\
4894 torture_assert_str_equal(tctx, array[i], iname[i],\
4895 talloc_asprintf(tctx, "%s - %s mismatch", #wname, iname[i]));\
4899 if (!test_winreg_symbolic_link(tctx, winreg_handle, hive_handle,
4900 TOP_LEVEL_CONTROL_PRINTERS_KEY,
4901 "\\Registry\\Machine\\Software\\Microsoft\\Windows NT\\CurrentVersion\\Print\\Printers"))
4903 torture_warning(tctx, "failed to check for winreg symlink");
4906 for (i=0; i < ARRAY_SIZE(keys); i++) {
4908 const char *printer_key;
4909 struct policy_handle key_handle;
4911 printer_key = talloc_asprintf(tctx, "%s\\%s",
4912 keys[i], printer_name);
4914 torture_assert(tctx,
4915 test_winreg_OpenKey(tctx, winreg_handle, hive_handle, printer_key, &key_handle), "");
4917 test_sz("Name", printername);
4918 test_sz("Share Name", sharename);
4919 test_sz("Port", info.info2.portname);
4920 test_sz("Printer Driver", info.info2.drivername);
4921 test_sz("Description", info.info2.comment);
4922 test_sz("Location", info.info2.location);
4923 test_sz("Separator File", info.info2.sepfile);
4924 test_sz("Print Processor", info.info2.printprocessor);
4925 test_sz("Datatype", info.info2.datatype);
4926 test_sz("Parameters", info.info2.parameters);
4927 /* winreg: 0, spoolss not */
4928 /* test_dword("Attributes", info.info2.attributes); */
4929 test_dword("Priority", info.info2.priority);
4930 test_dword("Default Priority", info.info2.defaultpriority);
4931 /* winreg: 60, spoolss: 0 */
4932 /* test_dword("StartTime", info.info2.starttime); */
4933 /* test_dword("UntilTime", info.info2.untiltime); */
4934 /* winreg != spoolss */
4935 /* test_dword("Status", info.info2.status); */
4936 test_dm("Default DevMode", info.info2.devmode);
4937 test_sd("Security", info.info2.secdesc);
4939 torture_assert(tctx,
4940 test_winreg_CloseKey(tctx, winreg_handle, &key_handle), "");
4946 torture_comment(tctx, "Printer Info and winreg consistency test succeeded\n\n");
4951 static bool test_PrintProcessors(struct torture_context *tctx,
4952 struct dcerpc_binding_handle *b,
4953 const char *environment,
4954 struct dcerpc_binding_handle *winreg_handle,
4955 struct policy_handle *hive_handle)
4957 union spoolss_PrintProcessorInfo *info;
4961 torture_comment(tctx, "Testing Print Processor Info and winreg consistency\n");
4963 torture_assert(tctx,
4964 test_EnumPrintProcessors_level(tctx, b, environment, 1, &count, &info, WERR_OK),
4965 "failed to enum print processors level 1");
4967 for (i=0; i < count; i++) {
4969 const char *processor_key;
4970 struct policy_handle key_handle;
4972 processor_key = talloc_asprintf(tctx, "%s\\%s\\Print Processors\\%s",
4973 TOP_LEVEL_CONTROL_ENVIRONMENTS_KEY,
4975 info[i].info1.print_processor_name);
4977 torture_assert(tctx,
4978 test_winreg_OpenKey(tctx, winreg_handle, hive_handle, processor_key, &key_handle), "");
4980 /* nothing to check in there so far */
4982 torture_assert(tctx,
4983 test_winreg_CloseKey(tctx, winreg_handle, &key_handle), "");
4986 torture_comment(tctx, "Print Processor Info and winreg consistency test succeeded\n\n");
4991 static bool test_GetPrinterDriver2_level(struct torture_context *tctx,
4992 struct dcerpc_binding_handle *b,
4993 struct policy_handle *handle,
4994 const char *driver_name,
4995 const char *architecture,
4997 uint32_t client_major_version,
4998 uint32_t client_minor_version,
4999 union spoolss_DriverInfo *info_p,
5002 static const char *strip_path(const char *path)
5010 p = strrchr(path, '\\');
5018 static const char **strip_paths(const char **path_array)
5022 if (path_array == NULL) {
5026 for (i=0; path_array[i] != NULL; i++) {
5027 path_array[i] = strip_path(path_array[i]);
5033 static const char *driver_winreg_date(TALLOC_CTX *mem_ctx, NTTIME nt)
5039 return talloc_strdup(mem_ctx, "01/01/1601");
5042 t = nt_time_to_unix(nt);
5045 return talloc_asprintf(mem_ctx, "%02d/%02d/%04d",
5046 tm->tm_mon + 1, tm->tm_mday, tm->tm_year + 1900);
5049 static const char *driver_winreg_version(TALLOC_CTX *mem_ctx, uint64_t v)
5051 return talloc_asprintf(mem_ctx, "%u.%u.%u.%u",
5052 (unsigned)((v >> 48) & 0xFFFF),
5053 (unsigned)((v >> 32) & 0xFFFF),
5054 (unsigned)((v >> 16) & 0xFFFF),
5055 (unsigned)(v & 0xFFFF));
5058 static bool test_GetDriverInfo_winreg(struct torture_context *tctx,
5059 struct dcerpc_binding_handle *b,
5060 struct policy_handle *handle,
5061 const char *printer_name,
5062 const char *driver_name,
5063 const char *environment,
5064 enum spoolss_DriverOSVersion version,
5065 struct dcerpc_binding_handle *winreg_handle,
5066 struct policy_handle *hive_handle,
5067 const char *server_name_slash)
5069 WERROR result = WERR_OK;
5070 union spoolss_DriverInfo info;
5071 const char *driver_key;
5072 struct policy_handle key_handle;
5074 const char *driver_path;
5075 const char *data_file;
5076 const char *config_file;
5077 const char *help_file;
5078 const char **dependent_files;
5080 const char *driver_date;
5081 const char *inbox_driver_date;
5083 const char *driver_version;
5084 const char *inbox_driver_version;
5086 ZERO_STRUCT(key_handle);
5088 torture_comment(tctx, "Testing Driver Info and winreg consistency\n");
5090 driver_key = talloc_asprintf(tctx, "%s\\%s\\Drivers\\Version-%d\\%s",
5091 TOP_LEVEL_CONTROL_ENVIRONMENTS_KEY,
5096 torture_assert(tctx,
5097 test_winreg_OpenKey(tctx, winreg_handle, hive_handle, driver_key, &key_handle),
5098 "failed to open driver key");
5100 if (torture_setting_bool(tctx, "samba3", false) ||
5101 torture_setting_bool(tctx, "w2k3", false)) {
5106 torture_assert(tctx,
5107 test_GetPrinterDriver2_level(tctx, b, handle, driver_name, environment, 8, version, 0, &info, &result),
5108 "failed to get driver info level 8");
5110 torture_assert(tctx,
5111 test_EnumPrinterDrivers_findone(tctx, b, server_name_slash, environment, 8, driver_name, &info),
5112 "failed to get driver info level 8");
5115 if (W_ERROR_EQUAL(result, WERR_INVALID_LEVEL)) {
5119 driver_path = strip_path(info.info8.driver_path);
5120 data_file = strip_path(info.info8.data_file);
5121 config_file = strip_path(info.info8.config_file);
5122 help_file = strip_path(info.info8.help_file);
5123 dependent_files = strip_paths(info.info8.dependent_files);
5125 driver_date = driver_winreg_date(tctx, info.info8.driver_date);
5126 inbox_driver_date = driver_winreg_date(tctx, info.info8.min_inbox_driver_ver_date);
5128 driver_version = driver_winreg_version(tctx, info.info8.driver_version);
5129 inbox_driver_version = driver_winreg_version(tctx, info.info8.min_inbox_driver_ver_version);
5131 test_sz("Configuration File", config_file);
5132 test_sz("Data File", data_file);
5133 test_sz("Datatype", info.info8.default_datatype);
5134 test_sz("Driver", driver_path);
5135 test_sz("DriverDate", driver_date);
5136 test_sz("DriverVersion", driver_version);
5137 test_sz("HardwareID", info.info8.hardware_id);
5138 test_sz("Help File", help_file);
5139 test_sz("InfPath", info.info8.inf_path);
5140 test_sz("Manufacturer", info.info8.manufacturer_name);
5141 test_sz("MinInboxDriverVerDate", inbox_driver_date);
5142 test_sz("MinInboxDriverVerVersion", inbox_driver_version);
5143 test_sz("Monitor", info.info8.monitor_name);
5144 test_sz("OEM URL", info.info8.manufacturer_url);
5145 test_sz("Print Processor", info.info8.print_processor);
5146 test_sz("Provider", info.info8.provider);
5147 test_sz("VendorSetup", info.info8.vendor_setup);
5148 test_multi_sz("ColorProfiles", info.info8.color_profiles);
5149 test_multi_sz("Dependent Files", dependent_files);
5150 test_multi_sz("CoreDependencies", info.info8.core_driver_dependencies);
5151 test_multi_sz("Previous Names", info.info8.previous_names);
5152 /* test_dword("Attributes", ?); */
5153 test_dword("PrinterDriverAttributes", info.info8.printer_driver_attributes);
5154 test_dword("Version", info.info8.version);
5155 /* test_dword("TempDir", ?); */
5160 torture_assert(tctx,
5161 test_GetPrinterDriver2_level(tctx, b, handle, driver_name, environment, 6, version, 0, &info, &result),
5162 "failed to get driver info level 6");
5164 torture_assert(tctx,
5165 test_EnumPrinterDrivers_findone(tctx, b, server_name_slash, environment, 6, driver_name, &info),
5166 "failed to get driver info level 6");
5169 driver_path = strip_path(info.info6.driver_path);
5170 data_file = strip_path(info.info6.data_file);
5171 config_file = strip_path(info.info6.config_file);
5172 help_file = strip_path(info.info6.help_file);
5173 dependent_files = strip_paths(info.info6.dependent_files);
5175 driver_date = driver_winreg_date(tctx, info.info6.driver_date);
5177 driver_version = driver_winreg_version(tctx, info.info6.driver_version);
5179 test_sz("Configuration File", config_file);
5180 test_sz("Data File", data_file);
5181 test_sz("Datatype", info.info6.default_datatype);
5182 test_sz("Driver", driver_path);
5183 if (torture_setting_bool(tctx, "w2k3", false)) {
5184 DATA_BLOB blob = data_blob_talloc_zero(tctx, 8);
5185 push_nttime(blob.data, 0, info.info6.driver_date);
5186 test_binary("DriverDate", blob);
5187 SBVAL(blob.data, 0, info.info6.driver_version);
5188 test_binary("DriverVersion", blob);
5190 test_sz("DriverDate", driver_date);
5191 test_sz("DriverVersion", driver_version);
5193 test_sz("HardwareID", info.info6.hardware_id);
5194 test_sz("Help File", help_file);
5195 test_sz("Manufacturer", info.info6.manufacturer_name);
5196 test_sz("Monitor", info.info6.monitor_name);
5197 test_sz("OEM URL", info.info6.manufacturer_url);
5198 test_sz("Provider", info.info6.provider);
5199 test_multi_sz("Dependent Files", dependent_files);
5200 test_multi_sz("Previous Names", info.info6.previous_names);
5201 /* test_dword("Attributes", ?); */
5202 test_dword("Version", info.info6.version);
5203 /* test_dword("TempDir", ?); */
5206 torture_assert(tctx,
5207 test_GetPrinterDriver2_level(tctx, b, handle, driver_name, environment, 3, version, 0, &info, &result),
5208 "failed to get driver info level 3");
5210 torture_assert(tctx,
5211 test_EnumPrinterDrivers_findone(tctx, b, server_name_slash, environment, 3, driver_name, &info),
5212 "failed to get driver info level 3");
5215 driver_path = strip_path(info.info3.driver_path);
5216 data_file = strip_path(info.info3.data_file);
5217 config_file = strip_path(info.info3.config_file);
5218 help_file = strip_path(info.info3.help_file);
5219 dependent_files = strip_paths(info.info3.dependent_files);
5221 test_sz("Configuration File", config_file);
5222 test_sz("Data File", data_file);
5223 test_sz("Datatype", info.info3.default_datatype);
5224 test_sz("Driver", driver_path);
5225 test_sz("Help File", help_file);
5226 test_sz("Monitor", info.info3.monitor_name);
5227 test_multi_sz("Dependent Files", dependent_files);
5228 /* test_dword("Attributes", ?); */
5229 test_dword("Version", info.info3.version);
5230 /* test_dword("TempDir", ?); */
5233 torture_assert(tctx,
5234 test_winreg_CloseKey(tctx, winreg_handle, &key_handle), "");
5236 torture_comment(tctx, "Driver Info and winreg consistency test succeeded\n\n");
5244 static bool test_SetPrinterData(struct torture_context *tctx,
5245 struct dcerpc_binding_handle *b,
5246 struct policy_handle *handle,
5247 const char *value_name,
5248 enum winreg_Type type,
5252 struct spoolss_SetPrinterData r;
5254 r.in.handle = handle;
5255 r.in.value_name = value_name;
5258 r.in.offered = offered;
5260 torture_comment(tctx, "Testing SetPrinterData(%s)\n",
5263 torture_assert_ntstatus_ok(tctx,
5264 dcerpc_spoolss_SetPrinterData_r(b, tctx, &r),
5265 "SetPrinterData failed");
5266 torture_assert_werr_ok(tctx, r.out.result,
5267 "SetPrinterData failed");
5272 static bool test_SetPrinterData_matrix(struct torture_context *tctx,
5273 struct dcerpc_binding_handle *b,
5274 struct policy_handle *handle,
5275 const char *printer_name,
5276 struct dcerpc_binding_handle *winreg_handle,
5277 struct policy_handle *hive_handle)
5279 const char *values[] = {
5283 /* FIXME: not working with s3 atm. */
5289 /* FIXME: not working with s3 atm. */
5296 for (i=0; i < ARRAY_SIZE(values); i++) {
5298 enum winreg_Type type, expected_type = REG_SZ;
5303 torture_assert(tctx, push_reg_sz(tctx, &blob, "dog"), "");
5306 torture_assert(tctx,
5307 test_SetPrinterData(tctx, b, handle, values[i], REG_SZ, blob.data, blob.length),
5308 "SetPrinterData failed");
5310 torture_assert(tctx,
5311 test_GetPrinterData_checktype(tctx, b, handle, values[i], &expected_type, &type, &data, &needed),
5312 "GetPrinterData failed");
5314 torture_assert_int_equal(tctx, type, REG_SZ, "type mismatch");
5315 torture_assert_int_equal(tctx, needed, blob.length, "size mismatch");
5316 torture_assert_mem_equal(tctx, data, blob.data, blob.length, "buffer mismatch");
5318 if (winreg_handle && hive_handle) {
5320 enum winreg_Type w_type;
5325 torture_assert(tctx,
5326 test_winreg_query_printerdata(tctx, winreg_handle, hive_handle,
5327 printer_name, "PrinterDriverData", values[i],
5328 &w_type, &w_size, &w_length, &w_data), "");
5330 torture_assert_int_equal(tctx, w_type, REG_SZ, "winreg type mismatch");
5331 torture_assert_int_equal(tctx, w_size, blob.length, "winreg size mismatch");
5332 torture_assert_int_equal(tctx, w_length, blob.length, "winreg length mismatch");
5333 torture_assert_mem_equal(tctx, w_data, blob.data, blob.length, "winreg buffer mismatch");
5336 torture_assert(tctx,
5337 test_DeletePrinterData(tctx, b, handle, values[i]),
5338 "DeletePrinterData failed");
5345 static bool test_EnumPrinterKey(struct torture_context *tctx,
5346 struct dcerpc_binding_handle *b,
5347 struct policy_handle *handle,
5348 const char *key_name,
5349 const char ***array);
5351 static bool test_SetPrinterDataEx(struct torture_context *tctx,
5352 struct dcerpc_binding_handle *b,
5353 struct policy_handle *handle,
5354 const char *key_name,
5355 const char *value_name,
5356 enum winreg_Type type,
5361 struct spoolss_SetPrinterDataEx r;
5363 r.in.handle = handle;
5364 r.in.key_name = key_name;
5365 r.in.value_name = value_name;
5368 r.in.offered = offered;
5370 torture_comment(tctx, "Testing SetPrinterDataEx(%s - %s) type: %s, offered: 0x%08x\n",
5371 r.in.key_name, r.in.value_name, str_regtype(r.in.type), r.in.offered);
5373 status = dcerpc_spoolss_SetPrinterDataEx_r(b, tctx, &r);
5375 torture_assert_ntstatus_ok(tctx, status, "SetPrinterDataEx failed");
5376 torture_assert_werr_ok(tctx, r.out.result, "SetPrinterDataEx failed");
5381 static bool test_SetPrinterDataEx_keys(struct torture_context *tctx,
5382 struct dcerpc_pipe *p,
5383 struct policy_handle *handle)
5385 struct dcerpc_binding_handle *b = p->binding_handle;
5386 const char *value_name = "dog";
5387 const char *keys[] = {
5390 "torturedataex_with_subkey\\subkey",
5391 "torturedataex_with_subkey\\subkey:0",
5392 "torturedataex_with_subkey\\subkey:1",
5393 "torturedataex_with_subkey\\subkey\\subsubkey",
5394 "torturedataex_with_subkey\\subkey\\subsubkey:0",
5395 "torturedataex_with_subkey\\subkey\\subsubkey:1",
5402 "torture/data ex/sub",
5405 "torture//data ex/sub",
5406 "torture//data ex//sub",
5410 for (i=0; i < ARRAY_SIZE(keys); i++) {
5414 enum winreg_Type type;
5415 DATA_BLOB blob_in, blob_out;
5416 const char **subkeys;
5418 struct spoolss_PrinterEnumValues *einfo;
5421 blob_in = data_blob_talloc(tctx, NULL, 42);
5423 generate_random_buffer(blob_in.data, blob_in.length);
5425 torture_assert(tctx,
5426 test_SetPrinterDataEx(tctx, b, handle, keys[i], value_name, REG_BINARY, blob_in.data, blob_in.length),
5427 "failed to call SetPrinterDataEx");
5429 torture_assert(tctx,
5430 test_GetPrinterDataEx(tctx, p, handle, keys[i], value_name, &type, &blob_out.data, &needed),
5431 "failed to call GetPrinterDataEx");
5433 blob_out.length = needed;
5434 torture_assert(tctx,
5435 test_EnumPrinterDataEx(tctx, b, handle, keys[i], &ecount, &einfo),
5436 "failed to call EnumPrinterDataEx");
5438 torture_assert_int_equal(tctx, type, REG_BINARY, "type mismatch");
5439 torture_assert_int_equal(tctx, blob_out.length, blob_in.length, "size mismatch");
5440 torture_assert_mem_equal(tctx, blob_out.data, blob_in.data, blob_in.length, "buffer mismatch");
5442 torture_assert_int_equal(tctx, ecount, 1, "unexpected enum count");
5443 torture_assert_str_equal(tctx, einfo[0].value_name, value_name, "value_name mismatch");
5444 torture_assert_int_equal(tctx, einfo[0].value_name_len, strlen_m_term(value_name)*2, "unexpected value_name_len");
5445 torture_assert_int_equal(tctx, einfo[0].type, REG_BINARY, "type mismatch");
5446 torture_assert_int_equal(tctx, einfo[0].data_length, blob_in.length, "size mismatch");
5447 if (einfo[0].data_length > 0) {
5448 torture_assert_mem_equal(tctx, einfo[0].data->data, blob_in.data, blob_in.length, "buffer mismatch");
5451 key = talloc_strdup(tctx, keys[i]);
5453 if (!test_DeletePrinterDataEx(tctx, b, handle, keys[i], value_name)) {
5457 c = strchr(key, '\\');
5461 /* we have subkeys */
5465 if (!test_EnumPrinterKey(tctx, b, handle, key, &subkeys)) {
5469 for (k=0; subkeys && subkeys[k]; k++) {
5471 const char *current_key = talloc_asprintf(tctx, "%s\\%s", key, subkeys[k]);
5473 if (!test_DeletePrinterKey(tctx, b, handle, current_key)) {
5478 if (!test_DeletePrinterKey(tctx, b, handle, key)) {
5483 if (!test_DeletePrinterKey(tctx, b, handle, key)) {
5492 static bool test_SetPrinterDataEx_values(struct torture_context *tctx,
5493 struct dcerpc_pipe *p,
5494 struct policy_handle *handle)
5496 struct dcerpc_binding_handle *b = p->binding_handle;
5497 const char *key = "torturedataex";
5498 const char *values[] = {
5508 for (i=0; i < ARRAY_SIZE(values); i++) {
5510 enum winreg_Type type;
5511 DATA_BLOB blob_in, blob_out;
5513 struct spoolss_PrinterEnumValues *einfo;
5516 if (torture_setting_bool(tctx, "samba3", false)) {
5518 q = strrchr(values[i], ',');
5520 torture_comment(tctx, "skipping valuename '%s' including ',' character against Samba3\n",
5526 blob_in = data_blob_talloc(tctx, NULL, 42);
5528 generate_random_buffer(blob_in.data, blob_in.length);
5530 torture_assert(tctx,
5531 test_SetPrinterDataEx(tctx, b, handle, key, values[i], REG_BINARY, blob_in.data, blob_in.length),
5532 "failed to call SetPrinterDataEx");
5534 torture_assert(tctx,
5535 test_GetPrinterDataEx(tctx, p, handle, key, values[i], &type, &blob_out.data, &needed),
5536 "failed to call GetPrinterDataEx");
5538 blob_out.length = needed;
5539 torture_assert(tctx,
5540 test_EnumPrinterDataEx(tctx, b, handle, key, &ecount, &einfo),
5541 "failed to call EnumPrinterDataEx");
5543 torture_assert_int_equal(tctx, type, REG_BINARY, "type mismatch");
5544 torture_assert_int_equal(tctx, blob_out.length, blob_in.length, "size mismatch");
5545 torture_assert_mem_equal(tctx, blob_out.data, blob_in.data, blob_in.length, "buffer mismatch");
5547 torture_assert_int_equal(tctx, ecount, 1, "unexpected enum count");
5548 torture_assert_str_equal(tctx, einfo[0].value_name, values[i], "value_name mismatch");
5549 torture_assert_int_equal(tctx, einfo[0].value_name_len, strlen_m_term(values[i])*2, "unexpected value_name_len");
5550 torture_assert_int_equal(tctx, einfo[0].type, REG_BINARY, "type mismatch");
5551 torture_assert_int_equal(tctx, einfo[0].data_length, blob_in.length, "size mismatch");
5552 if (einfo[0].data_length > 0) {
5553 torture_assert_mem_equal(tctx, einfo[0].data->data, blob_in.data, blob_in.length, "buffer mismatch");
5556 torture_assert(tctx,
5557 test_DeletePrinterDataEx(tctx, b, handle, key, values[i]),
5558 "failed to call DeletePrinterDataEx");
5565 static bool test_SetPrinterDataEx_matrix(struct torture_context *tctx,
5566 struct dcerpc_pipe *p,
5567 struct policy_handle *handle,
5568 const char *printername,
5569 struct dcerpc_binding_handle *winreg_handle,
5570 struct policy_handle *hive_handle)
5572 struct dcerpc_binding_handle *b = p->binding_handle;
5573 const char *value_name = "dog";
5574 const char *key_name = "torturedataex";
5575 enum winreg_Type types[] = {
5581 const char *str = "abcdefghi";
5584 for (t=0; t < ARRAY_SIZE(types); t++) {
5585 for (s=0; s < strlen(str); s++) {
5587 enum winreg_Type type;
5588 const char *string = talloc_strndup(tctx, str, s);
5589 const char *array[2];
5590 DATA_BLOB blob = data_blob_string_const(string);
5593 uint32_t needed, offered = 0;
5595 struct spoolss_PrinterEnumValues *einfo;
5597 array[0] = talloc_strdup(tctx, string);
5600 if (types[t] == REG_DWORD) {
5607 offered = blob.length;
5610 data = data_blob_talloc(tctx, NULL, 4);
5611 SIVAL(data.data, 0, 0x12345678);
5615 torture_assert(tctx, push_reg_sz(tctx, &data, string), "");
5617 offered = data.length;
5618 /*strlen_m_term(data.string)*2;*/
5621 torture_assert(tctx, push_reg_multi_sz(tctx, &data, array), "");
5622 type = REG_MULTI_SZ;
5623 offered = data.length;
5626 torture_fail(tctx, talloc_asprintf(tctx, "type %d untested\n", types[t]));
5629 torture_assert(tctx,
5630 test_SetPrinterDataEx(tctx, b, handle, key_name, value_name, types[t], data.data, offered),
5631 "failed to call SetPrinterDataEx");
5633 torture_assert(tctx,
5634 test_GetPrinterDataEx_checktype(tctx, p, handle, key_name, value_name, &types[t], &type, &data_out, &needed),
5635 "failed to call GetPrinterDataEx");
5637 torture_assert(tctx,
5638 test_EnumPrinterDataEx(tctx, b, handle, key_name, &ecount, &einfo),
5639 "failed to call EnumPrinterDataEx");
5641 torture_assert_int_equal(tctx, types[t], type, "type mismatch");
5642 torture_assert_int_equal(tctx, needed, offered, "size mismatch");
5643 torture_assert_mem_equal(tctx, data_out, data.data, offered, "buffer mismatch");
5645 torture_assert_int_equal(tctx, ecount, 1, "unexpected enum count");
5646 torture_assert_str_equal(tctx, einfo[0].value_name, value_name, "value_name mismatch");
5647 torture_assert_int_equal(tctx, einfo[0].value_name_len, strlen_m_term(value_name)*2, "unexpected value_name_len");
5648 torture_assert_int_equal(tctx, einfo[0].type, types[t], "type mismatch");
5649 torture_assert_int_equal(tctx, einfo[0].data_length, offered, "size mismatch");
5650 if (einfo[0].data_length > 0) {
5651 torture_assert_mem_equal(tctx, einfo[0].data->data, data.data, offered, "buffer mismatch");
5654 if (winreg_handle && hive_handle) {
5655 enum winreg_Type w_type;
5660 torture_assert(tctx,
5661 test_winreg_query_printerdata(tctx, winreg_handle, hive_handle,
5662 printername, key_name, value_name,
5663 &w_type, &w_size, &w_length, &w_data), "");
5665 torture_assert_int_equal(tctx, w_type, types[t], "winreg type mismatch");
5666 torture_assert_int_equal(tctx, w_size, offered, "winreg size mismatch");
5667 torture_assert_int_equal(tctx, w_length, offered, "winreg length mismatch");
5668 torture_assert_mem_equal(tctx, w_data, data.data, offered, "winreg buffer mismatch");
5671 torture_assert(tctx,
5672 test_DeletePrinterDataEx(tctx, b, handle, key_name, value_name),
5673 "failed to call DeletePrinterDataEx");
5680 static bool test_PrinterData_winreg(struct torture_context *tctx,
5681 struct dcerpc_pipe *p,
5682 struct policy_handle *handle,
5683 const char *printer_name)
5685 struct dcerpc_binding_handle *b = p->binding_handle;
5686 struct dcerpc_pipe *p2;
5688 struct policy_handle hive_handle;
5689 struct dcerpc_binding_handle *b2;
5691 torture_assert_ntstatus_ok(tctx,
5692 torture_rpc_connection(tctx, &p2, &ndr_table_winreg),
5693 "could not open winreg pipe");
5694 b2 = p2->binding_handle;
5696 torture_assert(tctx, test_winreg_OpenHKLM(tctx, b2, &hive_handle), "");
5698 ret &= test_SetPrinterData_matrix(tctx, b, handle, printer_name, b2, &hive_handle);
5699 ret &= test_SetPrinterDataEx_matrix(tctx, p, handle, printer_name, b2, &hive_handle);
5701 test_winreg_CloseKey(tctx, b2, &hive_handle);
5708 static bool test_Forms_winreg(struct torture_context *tctx,
5709 struct dcerpc_binding_handle *b,
5710 struct policy_handle *handle,
5712 const char *printer_name)
5714 struct dcerpc_pipe *p2;
5716 struct policy_handle hive_handle;
5717 struct dcerpc_binding_handle *b2;
5719 torture_assert_ntstatus_ok(tctx,
5720 torture_rpc_connection(tctx, &p2, &ndr_table_winreg),
5721 "could not open winreg pipe");
5722 b2 = p2->binding_handle;
5724 torture_assert(tctx, test_winreg_OpenHKLM(tctx, b2, &hive_handle), "");
5726 ret = test_Forms(tctx, b, handle, print_server, printer_name, b2, &hive_handle);
5728 test_winreg_CloseKey(tctx, b2, &hive_handle);
5735 static bool test_PrinterInfo_winreg(struct torture_context *tctx,
5736 struct dcerpc_pipe *p,
5737 struct policy_handle *handle,
5738 const char *printer_name)
5740 struct dcerpc_binding_handle *b = p->binding_handle;
5741 struct dcerpc_pipe *p2;
5743 struct policy_handle hive_handle;
5744 struct dcerpc_binding_handle *b2;
5746 torture_assert_ntstatus_ok(tctx,
5747 torture_rpc_connection(tctx, &p2, &ndr_table_winreg),
5748 "could not open winreg pipe");
5749 b2 = p2->binding_handle;
5751 torture_assert(tctx, test_winreg_OpenHKLM(tctx, b2, &hive_handle), "");
5753 ret = test_GetPrinterInfo_winreg(tctx, b, handle, printer_name, b2, &hive_handle);
5755 test_winreg_CloseKey(tctx, b2, &hive_handle);
5762 static bool test_DriverInfo_winreg(struct torture_context *tctx,
5763 struct dcerpc_pipe *p,
5764 struct policy_handle *handle,
5765 const char *printer_name,
5766 const char *driver_name,
5767 const char *environment,
5768 enum spoolss_DriverOSVersion version)
5770 struct dcerpc_binding_handle *b = p->binding_handle;
5771 struct dcerpc_pipe *p2;
5773 struct policy_handle hive_handle;
5774 struct dcerpc_binding_handle *b2;
5776 torture_assert_ntstatus_ok(tctx,
5777 torture_rpc_connection(tctx, &p2, &ndr_table_winreg),
5778 "could not open winreg pipe");
5779 b2 = p2->binding_handle;
5781 torture_assert(tctx, test_winreg_OpenHKLM(tctx, b2, &hive_handle), "");
5783 ret = test_GetDriverInfo_winreg(tctx, b, handle, printer_name, driver_name, environment, version, b2, &hive_handle, NULL);
5785 test_winreg_CloseKey(tctx, b2, &hive_handle);
5792 static bool test_PrintProcessors_winreg(struct torture_context *tctx,
5793 struct dcerpc_binding_handle *b,
5794 const char *environment)
5796 struct dcerpc_pipe *p2;
5798 struct policy_handle hive_handle;
5799 struct dcerpc_binding_handle *b2;
5801 torture_assert_ntstatus_ok(tctx,
5802 torture_rpc_connection(tctx, &p2, &ndr_table_winreg),
5803 "could not open winreg pipe");
5804 b2 = p2->binding_handle;
5806 torture_assert(tctx, test_winreg_OpenHKLM(tctx, b2, &hive_handle), "");
5808 ret = test_PrintProcessors(tctx, b, environment, b2, &hive_handle);
5810 test_winreg_CloseKey(tctx, b2, &hive_handle);
5817 static bool test_PrinterData_DsSpooler(struct torture_context *tctx,
5818 struct dcerpc_pipe *p,
5819 struct policy_handle *handle,
5820 const char *printer_name)
5822 struct spoolss_SetPrinterInfoCtr info_ctr;
5823 struct spoolss_DevmodeContainer devmode_ctr;
5824 struct sec_desc_buf secdesc_ctr;
5825 union spoolss_SetPrinterInfo sinfo;
5826 union spoolss_PrinterInfo info;
5827 struct dcerpc_binding_handle *b = p->binding_handle;
5830 ZERO_STRUCT(info_ctr);
5831 ZERO_STRUCT(devmode_ctr);
5832 ZERO_STRUCT(secdesc_ctr);
5834 torture_comment(tctx, "Testing DsSpooler <-> SetPrinter relations\n");
5836 torture_assert(tctx,
5837 test_GetPrinter_level(tctx, b, handle, 2, &info),
5838 "failed to query Printer level 2");
5840 torture_assert(tctx,
5841 PrinterInfo_to_SetPrinterInfo(tctx, &info, 2, &sinfo),
5842 "failed to convert");
5845 info_ctr.info = sinfo;
5847 #define TEST_SZ(wname, iname) \
5849 enum winreg_Type type;\
5854 torture_assert(tctx,\
5855 test_GetPrinterDataEx(tctx, p, handle, "DsSpooler", wname, &type, &data, &needed),\
5856 "failed to query");\
5857 torture_assert_int_equal(tctx, type, REG_SZ, "unexpected type");\
5858 blob = data_blob_const(data, needed);\
5859 torture_assert(tctx,\
5860 pull_reg_sz(tctx, &blob, &str),\
5861 "failed to pull REG_SZ");\
5862 torture_assert_str_equal(tctx, str, iname, "unexpected result");\
5866 #define TEST_SET_SZ(wname, iname, val) \
5868 enum winreg_Type type;\
5873 sinfo.info2->iname = val;\
5874 torture_assert(tctx,\
5875 test_SetPrinter(tctx, b, handle, &info_ctr, &devmode_ctr, &secdesc_ctr, 0),\
5876 "failed to call SetPrinter");\
5877 torture_assert(tctx,\
5878 test_GetPrinterDataEx(tctx, p, handle, "DsSpooler", wname, &type, &data, &needed),\
5879 "failed to query");\
5880 torture_assert_int_equal(tctx, type, REG_SZ, "unexpected type");\
5881 blob = data_blob_const(data, needed);\
5882 torture_assert(tctx,\
5883 pull_reg_sz(tctx, &blob, &str),\
5884 "failed to pull REG_SZ");\
5885 torture_assert_str_equal(tctx, str, val, "unexpected result");\
5888 #define TEST_SET_DWORD(wname, iname, val) \
5890 enum winreg_Type type;\
5894 sinfo.info2->iname = val;\
5895 torture_assert(tctx,\
5896 test_SetPrinter(tctx, b, handle, &info_ctr, &devmode_ctr, &secdesc_ctr, 0),\
5897 "failed to call SetPrinter");\
5898 torture_assert(tctx,\
5899 test_GetPrinterDataEx(tctx, p, handle, "DsSpooler", wname, &type, &data, &needed),\
5900 "failed to query");\
5901 torture_assert_int_equal(tctx, type, REG_DWORD, "unexpected type");\
5902 torture_assert_int_equal(tctx, needed, 4, "unexpected length");\
5903 value = IVAL(data, 0); \
5904 torture_assert_int_equal(tctx, value, val, "unexpected result");\
5907 TEST_SET_SZ("description", comment, "newval");
5908 TEST_SET_SZ("location", location, "newval");
5909 TEST_SET_SZ("driverName", drivername, "newval");
5910 /* TEST_SET_DWORD("priority", priority, 25); */
5912 torture_assert(tctx,
5913 test_GetPrinter_level(tctx, b, handle, 2, &info),
5914 "failed to query Printer level 2");
5916 TEST_SZ("description", info.info2.comment);
5917 TEST_SZ("driverName", info.info2.drivername);
5918 TEST_SZ("location", info.info2.location);
5920 pname = strrchr(info.info2.printername, '\\');
5921 if (pname == NULL) {
5922 pname = info.info2.printername;
5926 TEST_SZ("printerName", pname);
5927 /* TEST_SZ("printSeparatorFile", info.info2.sepfile); */
5928 /* TEST_SZ("printShareName", info.info2.sharename); */
5930 /* FIXME gd: complete the list */
5936 torture_comment(tctx, "DsSpooler <-> SetPrinter relations test succeeded\n\n");
5941 static bool test_print_processors_winreg(struct torture_context *tctx,
5944 struct test_spoolss_context *ctx =
5945 talloc_get_type_abort(private_data, struct test_spoolss_context);
5946 struct dcerpc_pipe *p = ctx->spoolss_pipe;
5947 struct dcerpc_binding_handle *b = p->binding_handle;
5949 return test_PrintProcessors_winreg(tctx, b, ctx->environment);
5952 static bool test_AddPrintProcessor(struct torture_context *tctx,
5953 struct dcerpc_binding_handle *b,
5954 const char *environment,
5955 const char *path_name,
5956 const char *print_processor_name,
5957 WERROR expected_error)
5959 struct spoolss_AddPrintProcessor r;
5962 r.in.architecture = environment;
5963 r.in.path_name = path_name;
5964 r.in.print_processor_name = print_processor_name;
5966 torture_comment(tctx, "Testing AddPrintProcessor(%s)\n",
5967 print_processor_name);
5969 torture_assert_ntstatus_ok(tctx,
5970 dcerpc_spoolss_AddPrintProcessor_r(b, tctx, &r),
5971 "spoolss_AddPrintProcessor failed");
5972 torture_assert_werr_equal(tctx, r.out.result, expected_error,
5973 "spoolss_AddPrintProcessor failed");
5978 static bool test_DeletePrintProcessor(struct torture_context *tctx,
5979 struct dcerpc_binding_handle *b,
5980 const char *environment,
5981 const char *print_processor_name,
5982 WERROR expected_error)
5984 struct spoolss_DeletePrintProcessor r;
5987 r.in.architecture = environment;
5988 r.in.print_processor_name = print_processor_name;
5990 torture_comment(tctx, "Testing DeletePrintProcessor(%s)\n",
5991 print_processor_name);
5993 torture_assert_ntstatus_ok(tctx,
5994 dcerpc_spoolss_DeletePrintProcessor_r(b, tctx, &r),
5995 "spoolss_DeletePrintProcessor failed");
5996 torture_assert_werr_equal(tctx, r.out.result, expected_error,
5997 "spoolss_DeletePrintProcessor failed");
6002 static bool test_add_print_processor(struct torture_context *tctx,
6005 struct test_spoolss_context *ctx =
6006 talloc_get_type_abort(private_data, struct test_spoolss_context);
6007 struct dcerpc_pipe *p = ctx->spoolss_pipe;
6008 struct dcerpc_binding_handle *b = p->binding_handle;
6012 const char *environment;
6013 const char *path_name;
6014 const char *print_processor_name;
6015 WERROR expected_add_result;
6016 WERROR expected_del_result;
6019 .environment = ctx->environment,
6021 .print_processor_name = "winprint",
6022 .expected_add_result = WERR_PRINT_PROCESSOR_ALREADY_INSTALLED,
6023 .expected_del_result = WERR_CAN_NOT_COMPLETE
6025 .environment = ctx->environment,
6027 .print_processor_name = "unknown",
6028 .expected_add_result = WERR_MOD_NOT_FOUND,
6029 .expected_del_result = WERR_UNKNOWN_PRINTPROCESSOR
6033 for (i=0; i < ARRAY_SIZE(tests); i++) {
6034 torture_assert(tctx,
6035 test_AddPrintProcessor(tctx, b,
6036 tests[i].environment,
6038 tests[i].print_processor_name,
6039 tests[i].expected_add_result),
6040 "add print processor failed");
6041 torture_assert(tctx,
6042 test_DeletePrintProcessor(tctx, b,
6043 tests[i].environment,
6044 tests[i].print_processor_name,
6045 tests[i].expected_del_result),
6046 "delete print processor failed");
6052 static bool test_GetChangeID_PrinterData(struct torture_context *tctx,
6053 struct dcerpc_binding_handle *b,
6054 struct policy_handle *handle,
6055 uint32_t *change_id)
6057 enum winreg_Type type;
6061 torture_assert(tctx,
6062 test_GetPrinterData(tctx, b, handle, "ChangeID", &type, &data, &needed),
6063 "failed to call GetPrinterData");
6065 torture_assert(tctx, type == REG_DWORD, "unexpected type");
6066 torture_assert_int_equal(tctx, needed, 4, "unexpected size");
6068 *change_id = IVAL(data, 0);
6073 static bool test_GetChangeID_PrinterDataEx(struct torture_context *tctx,
6074 struct dcerpc_pipe *p,
6075 struct policy_handle *handle,
6076 uint32_t *change_id)
6078 enum winreg_Type type = REG_NONE;
6079 uint8_t *data = NULL;
6080 uint32_t needed = 0;
6082 torture_assert(tctx,
6083 test_GetPrinterDataEx(tctx, p, handle, "PrinterDriverData", "ChangeID", &type, &data, &needed),
6084 "failed to call GetPrinterData");
6086 torture_assert(tctx, type == REG_DWORD, "unexpected type");
6087 torture_assert_int_equal(tctx, needed, 4, "unexpected size");
6089 *change_id = IVAL(data, 0);
6094 static bool test_GetChangeID_PrinterInfo(struct torture_context *tctx,
6095 struct dcerpc_binding_handle *b,
6096 struct policy_handle *handle,
6097 uint32_t *change_id)
6099 union spoolss_PrinterInfo info;
6101 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 0, &info),
6102 "failed to query Printer level 0");
6104 *change_id = info.info0.change_id;
6109 static bool test_ChangeID(struct torture_context *tctx,
6110 struct dcerpc_pipe *p,
6111 struct policy_handle *handle)
6113 uint32_t change_id, change_id_ex, change_id_info;
6114 uint32_t change_id2, change_id_ex2, change_id_info2;
6115 union spoolss_PrinterInfo info;
6116 const char *comment;
6117 struct dcerpc_binding_handle *b = p->binding_handle;
6119 torture_comment(tctx, "Testing ChangeID: id change test #1\n");
6121 torture_assert(tctx, test_GetChangeID_PrinterData(tctx, b, handle, &change_id),
6122 "failed to query for ChangeID");
6123 torture_assert(tctx, test_GetChangeID_PrinterDataEx(tctx, p, handle, &change_id_ex),
6124 "failed to query for ChangeID");
6125 torture_assert(tctx, test_GetChangeID_PrinterInfo(tctx, b, handle, &change_id_info),
6126 "failed to query for ChangeID");
6128 torture_assert_int_equal(tctx, change_id, change_id_ex,
6129 "change_ids should all be equal");
6130 torture_assert_int_equal(tctx, change_id_ex, change_id_info,
6131 "change_ids should all be equal");
6134 torture_comment(tctx, "Testing ChangeID: id change test #2\n");
6136 torture_assert(tctx, test_GetChangeID_PrinterData(tctx, b, handle, &change_id),
6137 "failed to query for ChangeID");
6138 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info),
6139 "failed to query Printer level 2");
6140 torture_assert(tctx, test_GetChangeID_PrinterDataEx(tctx, p, handle, &change_id_ex),
6141 "failed to query for ChangeID");
6142 torture_assert(tctx, test_GetChangeID_PrinterInfo(tctx, b, handle, &change_id_info),
6143 "failed to query for ChangeID");
6144 torture_assert_int_equal(tctx, change_id, change_id_ex,
6145 "change_id should not have changed");
6146 torture_assert_int_equal(tctx, change_id_ex, change_id_info,
6147 "change_id should not have changed");
6150 torture_comment(tctx, "Testing ChangeID: id change test #3\n");
6152 torture_assert(tctx, test_GetChangeID_PrinterData(tctx, b, handle, &change_id),
6153 "failed to query for ChangeID");
6154 torture_assert(tctx, test_GetChangeID_PrinterDataEx(tctx, p, handle, &change_id_ex),
6155 "failed to query for ChangeID");
6156 torture_assert(tctx, test_GetChangeID_PrinterInfo(tctx, b, handle, &change_id_info),
6157 "failed to query for ChangeID");
6158 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info),
6159 "failed to query Printer level 2");
6160 comment = talloc_strdup(tctx, info.info2.comment);
6163 struct spoolss_SetPrinterInfoCtr info_ctr;
6164 struct spoolss_DevmodeContainer devmode_ctr;
6165 struct sec_desc_buf secdesc_ctr;
6166 union spoolss_SetPrinterInfo sinfo;
6168 ZERO_STRUCT(info_ctr);
6169 ZERO_STRUCT(devmode_ctr);
6170 ZERO_STRUCT(secdesc_ctr);
6173 torture_assert(tctx, PrinterInfo_to_SetPrinterInfo(tctx, &info, 2, &sinfo), "");
6174 sinfo.info2->comment = "torture_comment";
6177 info_ctr.info = sinfo;
6179 torture_assert(tctx, test_SetPrinter(tctx, b, handle, &info_ctr, &devmode_ctr, &secdesc_ctr, 0),
6180 "failed to call SetPrinter");
6182 sinfo.info2->comment = comment;
6184 torture_assert(tctx, test_SetPrinter(tctx, b, handle, &info_ctr, &devmode_ctr, &secdesc_ctr, 0),
6185 "failed to call SetPrinter");
6189 torture_assert(tctx, test_GetChangeID_PrinterData(tctx, b, handle, &change_id2),
6190 "failed to query for ChangeID");
6191 torture_assert(tctx, test_GetChangeID_PrinterDataEx(tctx, p, handle, &change_id_ex2),
6192 "failed to query for ChangeID");
6193 torture_assert(tctx, test_GetChangeID_PrinterInfo(tctx, b, handle, &change_id_info2),
6194 "failed to query for ChangeID");
6196 torture_assert_int_equal(tctx, change_id2, change_id_ex2,
6197 "change_ids should all be equal");
6198 torture_assert_int_equal(tctx, change_id_ex2, change_id_info2,
6199 "change_ids should all be equal");
6201 torture_assert(tctx, (change_id < change_id2),
6202 talloc_asprintf(tctx, "change_id %d needs to be larger than change_id %d",
6203 change_id2, change_id));
6204 torture_assert(tctx, (change_id_ex < change_id_ex2),
6205 talloc_asprintf(tctx, "change_id %d needs to be larger than change_id %d",
6206 change_id_ex2, change_id_ex));
6207 torture_assert(tctx, (change_id_info < change_id_info2),
6208 talloc_asprintf(tctx, "change_id %d needs to be larger than change_id %d",
6209 change_id_info2, change_id_info));
6211 torture_comment(tctx, "ChangeID tests succeeded\n\n");
6216 static bool test_SecondaryClosePrinter(struct torture_context *tctx,
6217 struct dcerpc_pipe *p,
6218 struct policy_handle *handle)
6221 struct cli_credentials *anon_creds;
6222 const struct dcerpc_binding *binding2;
6223 struct dcerpc_pipe *p2;
6224 struct spoolss_ClosePrinter cp;
6226 /* only makes sense on SMB */
6227 if (p->conn->transport.transport != NCACN_NP) {
6231 torture_comment(tctx, "Testing close on secondary pipe\n");
6233 anon_creds = cli_credentials_init_anon(tctx);
6234 torture_assert(tctx, anon_creds != NULL, "cli_credentials_init_anon failed");
6236 binding2 = p->binding;
6237 status = dcerpc_secondary_auth_connection(p, binding2, &ndr_table_spoolss,
6238 anon_creds, tctx->lp_ctx,
6240 torture_assert_ntstatus_ok(tctx, status, "Failed to create secondary connection");
6242 cp.in.handle = handle;
6243 cp.out.handle = handle;
6245 status = dcerpc_spoolss_ClosePrinter_r(p2->binding_handle, tctx, &cp);
6246 torture_assert_ntstatus_equal(tctx, status, NT_STATUS_RPC_SS_CONTEXT_MISMATCH,
6247 "ERROR: Allowed close on secondary connection");
6254 static bool test_OpenPrinter_badname(struct torture_context *tctx,
6255 struct dcerpc_binding_handle *b, const char *name)
6258 struct spoolss_OpenPrinter op;
6259 struct spoolss_OpenPrinterEx opEx;
6260 struct policy_handle handle;
6263 op.in.printername = name;
6264 op.in.datatype = NULL;
6265 op.in.devmode_ctr.devmode= NULL;
6266 op.in.access_mask = 0;
6267 op.out.handle = &handle;
6269 torture_comment(tctx, "Testing OpenPrinter(%s) with bad name\n", op.in.printername);
6271 status = dcerpc_spoolss_OpenPrinter_r(b, tctx, &op);
6272 torture_assert_ntstatus_ok(tctx, status, "OpenPrinter failed");
6273 torture_assert_werr_equal(tctx, op.out.result, WERR_INVALID_PRINTER_NAME,
6274 "unexpected result");
6276 if (W_ERROR_IS_OK(op.out.result)) {
6277 ret &=test_ClosePrinter(tctx, b, &handle);
6280 opEx.in.printername = name;
6281 opEx.in.datatype = NULL;
6282 opEx.in.devmode_ctr.devmode = NULL;
6283 opEx.in.access_mask = 0;
6284 opEx.in.userlevel_ctr.level = 1;
6285 opEx.in.userlevel_ctr.user_info.level1 = NULL;
6286 opEx.out.handle = &handle;
6288 torture_comment(tctx, "Testing OpenPrinterEx(%s) with bad name\n", opEx.in.printername);
6290 status = dcerpc_spoolss_OpenPrinterEx_r(b, tctx, &opEx);
6291 torture_assert_ntstatus_ok(tctx, status, "OpenPrinterEx failed");
6292 torture_assert_werr_equal(tctx, opEx.out.result, WERR_INVALID_PARAM,
6293 "unexpected result");
6295 if (W_ERROR_IS_OK(opEx.out.result)) {
6296 ret &=test_ClosePrinter(tctx, b, &handle);
6302 static bool test_OpenPrinter_badname_list(struct torture_context *tctx,
6305 struct test_spoolss_context *ctx =
6306 talloc_get_type_abort(private_data, struct test_spoolss_context);
6308 const char *badnames[] = {
6309 "__INVALID_PRINTER__",
6310 "\\\\__INVALID_HOST__",
6313 "\\\\\\__INVALID_PRINTER__"
6315 const char *badname;
6316 struct dcerpc_pipe *p = ctx->spoolss_pipe;
6317 const char *server_name = dcerpc_server_name(p);
6318 struct dcerpc_binding_handle *b = p->binding_handle;
6321 for (i=0; i < ARRAY_SIZE(badnames); i++) {
6322 torture_assert(tctx,
6323 test_OpenPrinter_badname(tctx, b, badnames[i]),
6327 badname = talloc_asprintf(tctx, "\\\\%s\\", server_name);
6328 torture_assert(tctx,
6329 test_OpenPrinter_badname(tctx, b, badname),
6332 badname = talloc_asprintf(tctx, "\\\\%s\\__INVALID_PRINTER__", server_name);
6333 torture_assert(tctx,
6334 test_OpenPrinter_badname(tctx, b, badname),
6340 static bool test_OpenPrinter(struct torture_context *tctx,
6341 struct dcerpc_pipe *p,
6343 const char *environment,
6347 struct spoolss_OpenPrinter r;
6348 struct policy_handle handle;
6350 struct dcerpc_binding_handle *b = p->binding_handle;
6352 r.in.printername = name;
6353 r.in.datatype = NULL;
6354 r.in.devmode_ctr.devmode= NULL;
6355 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
6356 r.out.handle = &handle;
6358 torture_comment(tctx, "Testing OpenPrinter(%s)\n", r.in.printername);
6360 status = dcerpc_spoolss_OpenPrinter_r(b, tctx, &r);
6362 torture_assert_ntstatus_ok(tctx, status, "OpenPrinter failed");
6364 torture_assert_werr_ok(tctx, r.out.result, "OpenPrinter failed");
6370 if (!test_GetPrinter(tctx, b, &handle, environment)) {
6374 if (!torture_setting_bool(tctx, "samba3", false)) {
6375 if (!test_SecondaryClosePrinter(tctx, p, &handle)) {
6381 if (!test_ClosePrinter(tctx, b, &handle)) {
6388 static bool test_OpenPrinterEx(struct torture_context *tctx,
6389 struct dcerpc_binding_handle *b,
6390 const char *printername,
6391 const char *datatype,
6392 struct spoolss_DeviceMode *devmode,
6393 uint32_t access_mask,
6394 struct spoolss_UserLevelCtr *userlevel_ctr,
6395 struct policy_handle *handle,
6396 WERROR expected_result)
6398 struct spoolss_OpenPrinterEx r;
6400 r.in.printername = printername;
6401 r.in.datatype = datatype;
6402 r.in.devmode_ctr.devmode= devmode;
6403 r.in.access_mask = access_mask;
6404 r.in.userlevel_ctr = *userlevel_ctr;
6405 r.out.handle = handle;
6407 torture_comment(tctx, "Testing OpenPrinterEx(%s)\n", r.in.printername);
6409 torture_assert_ntstatus_ok(tctx,
6410 dcerpc_spoolss_OpenPrinterEx_r(b, tctx, &r),
6411 "OpenPrinterEx failed");
6413 torture_assert_werr_equal(tctx, r.out.result, expected_result,
6414 "OpenPrinterEx failed");
6419 static bool call_OpenPrinterEx(struct torture_context *tctx,
6420 struct dcerpc_pipe *p,
6422 struct spoolss_DeviceMode *devmode,
6423 struct policy_handle *handle)
6425 struct spoolss_UserLevelCtr userlevel_ctr;
6426 struct spoolss_UserLevel1 userlevel1;
6427 struct dcerpc_binding_handle *b = p->binding_handle;
6429 userlevel1.size = 1234;
6430 userlevel1.client = "hello";
6431 userlevel1.user = "spottyfoot!";
6432 userlevel1.build = 1;
6433 userlevel1.major = 2;
6434 userlevel1.minor = 3;
6435 userlevel1.processor = 4;
6437 userlevel_ctr.level = 1;
6438 userlevel_ctr.user_info.level1 = &userlevel1;
6440 return test_OpenPrinterEx(tctx, b, name, NULL, devmode,
6441 SEC_FLAG_MAXIMUM_ALLOWED,
6447 static bool test_printer_rename(struct torture_context *tctx,
6450 struct torture_printer_context *t =
6451 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
6452 struct dcerpc_pipe *p = t->spoolss_pipe;
6455 union spoolss_PrinterInfo info;
6456 union spoolss_SetPrinterInfo sinfo;
6457 struct spoolss_SetPrinterInfoCtr info_ctr;
6458 struct spoolss_DevmodeContainer devmode_ctr;
6459 struct sec_desc_buf secdesc_ctr;
6460 const char *printer_name;
6461 const char *printer_name_orig;
6462 const char *printer_name_new = "SAMBA smbtorture Test Printer (Copy 2)";
6463 struct policy_handle new_handle;
6465 struct dcerpc_binding_handle *b = p->binding_handle;
6467 ZERO_STRUCT(devmode_ctr);
6468 ZERO_STRUCT(secdesc_ctr);
6470 torture_comment(tctx, "Testing Printer rename operations\n");
6472 torture_assert(tctx,
6473 test_GetPrinter_level(tctx, b, &t->handle, 2, &info),
6474 "failed to call GetPrinter level 2");
6476 printer_name_orig = talloc_strdup(tctx, info.info2.printername);
6478 q = strrchr(info.info2.printername, '\\');
6480 torture_warning(tctx,
6481 "server returns printername %s incl. servername although we did not set servername", info.info2.printername);
6484 torture_assert(tctx,
6485 PrinterInfo_to_SetPrinterInfo(tctx, &info, 2, &sinfo), "");
6487 sinfo.info2->printername = printer_name_new;
6490 info_ctr.info = sinfo;
6492 torture_assert(tctx,
6493 test_SetPrinter(tctx, b, &t->handle, &info_ctr, &devmode_ctr, &secdesc_ctr, 0),
6494 "failed to call SetPrinter level 2");
6496 torture_assert(tctx,
6497 test_GetPrinter_level(tctx, b, &t->handle, 2, &info),
6498 "failed to call GetPrinter level 2");
6500 printer_name = talloc_strdup(tctx, info.info2.printername);
6502 q = strrchr(info.info2.printername, '\\');
6504 torture_warning(tctx,
6505 "server returns printername %s incl. servername although we did not set servername", info.info2.printername);
6510 torture_assert_str_equal(tctx, printer_name, printer_name_new,
6511 "new printer name was not set");
6513 /* samba currently cannot fully rename printers */
6514 if (!torture_setting_bool(tctx, "samba3", false)) {
6515 torture_assert(tctx,
6516 test_OpenPrinter_badname(tctx, b, printer_name_orig),
6517 "still can open printer with oldname after rename");
6519 torture_warning(tctx, "*not* checking for open with oldname after rename for samba3");
6522 torture_assert(tctx,
6523 call_OpenPrinterEx(tctx, p, printer_name_new, NULL, &new_handle),
6524 "failed to open printer with new name");
6526 torture_assert(tctx,
6527 test_GetPrinter_level(tctx, b, &new_handle, 2, &info),
6528 "failed to call GetPrinter level 2");
6530 torture_assert_str_equal(tctx, info.info2.printername, printer_name_new,
6531 "new printer name was not set");
6533 torture_assert(tctx,
6534 test_ClosePrinter(tctx, b, &new_handle),
6535 "failed to close printer");
6537 torture_comment(tctx, "Printer rename operations test succeeded\n\n");
6542 static bool test_openprinter(struct torture_context *tctx,
6543 struct dcerpc_binding_handle *b,
6544 const char *real_printername)
6546 struct spoolss_UserLevelCtr userlevel_ctr;
6547 struct policy_handle handle;
6548 struct spoolss_UserLevel1 userlevel1;
6549 const char *printername = NULL;
6554 WERROR expected_result;
6557 .suffix = "rubbish",
6558 .expected_result = WERR_INVALID_PRINTER_NAME
6560 .suffix = ", LocalOnl",
6561 .expected_result = WERR_INVALID_PRINTER_NAME
6563 .suffix = ", localOnly",
6564 .expected_result = WERR_INVALID_PRINTER_NAME
6566 .suffix = ", localonl",
6567 .expected_result = WERR_INVALID_PRINTER_NAME
6569 .suffix = ",LocalOnl",
6570 .expected_result = WERR_INVALID_PRINTER_NAME
6572 .suffix = ",localOnl2",
6573 .expected_result = WERR_INVALID_PRINTER_NAME
6575 .suffix = ", DrvConver2t",
6576 .expected_result = WERR_INVALID_PRINTER_NAME
6578 .suffix = ", drvconvert",
6579 .expected_result = WERR_INVALID_PRINTER_NAME
6581 .suffix = ",drvconvert",
6582 .expected_result = WERR_INVALID_PRINTER_NAME
6584 .suffix = ", DrvConvert",
6585 .expected_result = WERR_OK
6587 .suffix = " , DrvConvert",
6588 .expected_result = WERR_INVALID_PRINTER_NAME
6590 .suffix = ",DrvConvert",
6591 .expected_result = WERR_OK
6593 .suffix = ", DrvConvertsadfasdf",
6594 .expected_result = WERR_OK
6596 .suffix = ",DrvConvertasdfasd",
6597 .expected_result = WERR_OK
6599 .suffix = ", LocalOnly",
6600 .expected_result = WERR_OK
6602 .suffix = " , LocalOnly",
6603 .expected_result = WERR_INVALID_PRINTER_NAME
6605 .suffix = ",LocalOnly",
6606 .expected_result = WERR_OK
6608 .suffix = ", LocalOnlysagi4gjfkd",
6609 .expected_result = WERR_OK
6611 .suffix = ",LocalOnlysagi4gjfkd",
6612 .expected_result = WERR_OK
6616 userlevel1.size = 1234;
6617 userlevel1.client = "hello";
6618 userlevel1.user = "spottyfoot!";
6619 userlevel1.build = 1;
6620 userlevel1.major = 2;
6621 userlevel1.minor = 3;
6622 userlevel1.processor = 4;
6624 userlevel_ctr.level = 1;
6625 userlevel_ctr.user_info.level1 = &userlevel1;
6627 torture_comment(tctx, "Testing openprinterex printername pattern\n");
6629 torture_assert(tctx,
6630 test_OpenPrinterEx(tctx, b, real_printername, NULL, NULL, 0,
6631 &userlevel_ctr, &handle,
6633 "OpenPrinterEx failed");
6634 test_ClosePrinter(tctx, b, &handle);
6636 for (i=0; i < ARRAY_SIZE(tests); i++) {
6638 printername = talloc_asprintf(tctx, "%s%s",
6642 torture_assert(tctx,
6643 test_OpenPrinterEx(tctx, b, printername, NULL, NULL, 0,
6644 &userlevel_ctr, &handle,
6645 tests[i].expected_result),
6646 "OpenPrinterEx failed");
6647 if (W_ERROR_IS_OK(tests[i].expected_result)) {
6648 test_ClosePrinter(tctx, b, &handle);
6656 static bool test_existing_printer_openprinterex(struct torture_context *tctx,
6657 struct dcerpc_pipe *p,
6659 const char *environment)
6661 struct policy_handle handle;
6663 struct dcerpc_binding_handle *b = p->binding_handle;
6665 if (!test_openprinter(tctx, b, name)) {
6669 if (!call_OpenPrinterEx(tctx, p, name, NULL, &handle)) {
6673 if (!test_PrinterInfo_SD(tctx, b, &handle)) {
6677 if (!test_GetPrinter(tctx, b, &handle, environment)) {
6681 if (!test_EnumForms_all(tctx, b, &handle, false)) {
6685 if (!test_Forms(tctx, b, &handle, false, name, NULL, NULL)) {
6689 if (!test_Forms_winreg(tctx, b, &handle, false, name)) {
6693 if (!test_EnumPrinterData_all(tctx, p, &handle)) {
6697 if (!test_EnumPrinterDataEx(tctx, b, &handle, "PrinterDriverData", NULL, NULL)) {
6701 if (!test_EnumPrinterData_consistency(tctx, p, &handle)) {
6705 if (!test_printer_all_keys(tctx, b, &handle)) {
6709 if (!test_PausePrinter(tctx, b, &handle)) {
6713 if (!test_DoPrintTest(tctx, b, &handle)) {
6717 if (!test_ResumePrinter(tctx, b, &handle)) {
6721 if (!test_SetPrinterData_matrix(tctx, b, &handle, name, NULL, NULL)) {
6725 if (!test_SetPrinterDataEx_matrix(tctx, p, &handle, name, NULL, NULL)) {
6729 if (!torture_setting_bool(tctx, "samba3", false)) {
6730 if (!test_SecondaryClosePrinter(tctx, p, &handle)) {
6735 if (!test_ClosePrinter(tctx, b, &handle)) {
6742 static bool test_EnumPrinters_old(struct torture_context *tctx,
6745 struct test_spoolss_context *ctx =
6746 talloc_get_type_abort(private_data, struct test_spoolss_context);
6747 struct spoolss_EnumPrinters r;
6749 uint16_t levels[] = {1, 2, 4, 5};
6752 struct dcerpc_pipe *p = ctx->spoolss_pipe;
6753 struct dcerpc_binding_handle *b = p->binding_handle;
6755 for (i=0;i<ARRAY_SIZE(levels);i++) {
6756 union spoolss_PrinterInfo *info;
6761 r.in.flags = PRINTER_ENUM_LOCAL;
6763 r.in.level = levels[i];
6766 r.out.needed = &needed;
6767 r.out.count = &count;
6770 torture_comment(tctx, "Testing EnumPrinters level %u\n", r.in.level);
6772 status = dcerpc_spoolss_EnumPrinters_r(b, tctx, &r);
6773 torture_assert_ntstatus_ok(tctx, status, "EnumPrinters failed");
6775 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
6776 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
6777 r.in.buffer = &blob;
6778 r.in.offered = needed;
6779 status = dcerpc_spoolss_EnumPrinters_r(b, tctx, &r);
6782 torture_assert_ntstatus_ok(tctx, status, "EnumPrinters failed");
6784 torture_assert_werr_ok(tctx, r.out.result, "EnumPrinters failed");
6786 CHECK_NEEDED_SIZE_ENUM_LEVEL(spoolss_EnumPrinters, info, r.in.level, count, needed, 4);
6789 torture_comment(tctx, "No printers returned\n");
6793 for (j=0;j<count;j++) {
6794 if (r.in.level == 1) {
6795 char *unc = talloc_strdup(tctx, info[j].info1.name);
6796 char *slash, *name, *full_name;
6798 if (unc[0] == '\\' && unc[1] == '\\') {
6801 slash = strchr(unc, '\\');
6806 full_name = talloc_asprintf(tctx, "\\\\%s\\%s",
6807 dcerpc_server_name(p), name);
6808 if (!test_OpenPrinter(tctx, p, name, ctx->environment, true)) {
6811 if (!test_OpenPrinter(tctx, p, full_name, ctx->environment, true)) {
6814 if (!test_OpenPrinter(tctx, p, name, ctx->environment, false)) {
6817 if (!test_existing_printer_openprinterex(tctx, p, name, ctx->environment)) {
6827 static bool test_EnumPrinters_level(struct torture_context *tctx,
6828 struct dcerpc_binding_handle *b,
6830 const char *servername,
6833 union spoolss_PrinterInfo **info_p)
6835 struct spoolss_EnumPrinters r;
6836 union spoolss_PrinterInfo *info;
6841 r.in.server = servername;
6845 r.out.needed = &needed;
6846 r.out.count = &count;
6849 torture_comment(tctx, "Testing EnumPrinters(%s) level %u\n",
6850 r.in.server, r.in.level);
6852 torture_assert_ntstatus_ok(tctx,
6853 dcerpc_spoolss_EnumPrinters_r(b, tctx, &r),
6854 "EnumPrinters failed");
6855 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
6856 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
6857 r.in.buffer = &blob;
6858 r.in.offered = needed;
6859 torture_assert_ntstatus_ok(tctx,
6860 dcerpc_spoolss_EnumPrinters_r(b, tctx, &r),
6861 "EnumPrinters failed");
6864 torture_assert_werr_ok(tctx, r.out.result, "EnumPrinters failed");
6866 CHECK_NEEDED_SIZE_ENUM_LEVEL(spoolss_EnumPrinters, info, r.in.level, count, needed, 4);
6878 static const char *get_short_printername(struct torture_context *tctx,
6881 const char *short_name;
6883 if (name[0] == '\\' && name[1] == '\\') {
6885 short_name = strchr(name, '\\');
6887 return talloc_strdup(tctx, short_name+1);
6894 static const char *get_full_printername(struct torture_context *tctx,
6897 const char *full_name = talloc_strdup(tctx, name);
6900 if (name && name[0] == '\\' && name[1] == '\\') {
6902 p = strchr(name, '\\');
6911 static bool test_OnePrinter_servername(struct torture_context *tctx,
6912 struct dcerpc_pipe *p,
6913 struct dcerpc_binding_handle *b,
6914 const char *servername,
6915 const char *printername)
6917 union spoolss_PrinterInfo info;
6918 const char *short_name = get_short_printername(tctx, printername);
6919 const char *full_name = get_full_printername(tctx, printername);
6922 struct policy_handle handle;
6923 torture_assert(tctx,
6924 call_OpenPrinterEx(tctx, p, short_name, NULL, &handle),
6925 "failed to open printer");
6927 torture_assert(tctx,
6928 test_GetPrinter_level(tctx, b, &handle, 2, &info),
6929 "failed to get printer info");
6931 torture_assert_casestr_equal(tctx, info.info2.servername, NULL,
6932 "unexpected servername");
6933 torture_assert_casestr_equal(tctx, info.info2.printername, short_name,
6934 "unexpected printername");
6936 if (info.info2.devmode) {
6937 const char *expected_devicename;
6938 expected_devicename = talloc_strndup(tctx, short_name, MIN(strlen(short_name), 31));
6939 torture_assert_casestr_equal(tctx, info.info2.devmode->devicename, expected_devicename,
6940 "unexpected devicemode devicename");
6943 torture_assert(tctx,
6944 test_ClosePrinter(tctx, b, &handle),
6945 "failed to close printer");
6949 struct policy_handle handle;
6951 torture_assert(tctx,
6952 call_OpenPrinterEx(tctx, p, full_name, NULL, &handle),
6953 "failed to open printer");
6955 torture_assert(tctx,
6956 test_GetPrinter_level(tctx, b, &handle, 2, &info),
6957 "failed to get printer info");
6959 torture_assert_casestr_equal(tctx, info.info2.servername, servername,
6960 "unexpected servername");
6961 torture_assert_casestr_equal(tctx, info.info2.printername, full_name,
6962 "unexpected printername");
6964 if (info.info2.devmode) {
6965 const char *expected_devicename;
6966 expected_devicename = talloc_strndup(tctx, full_name, MIN(strlen(full_name), 31));
6967 torture_assert_casestr_equal(tctx, info.info2.devmode->devicename, expected_devicename,
6968 "unexpected devicemode devicename");
6971 torture_assert(tctx,
6972 test_ClosePrinter(tctx, b, &handle),
6973 "failed to close printer");
6979 static bool test_EnumPrinters_servername(struct torture_context *tctx,
6982 struct test_spoolss_context *ctx =
6983 talloc_get_type_abort(private_data, struct test_spoolss_context);
6985 struct dcerpc_pipe *p = ctx->spoolss_pipe;
6986 struct dcerpc_binding_handle *b = p->binding_handle;
6988 union spoolss_PrinterInfo *info;
6989 const char *servername;
6990 uint32_t flags = PRINTER_ENUM_NAME|PRINTER_ENUM_LOCAL;
6992 torture_comment(tctx, "Testing servername behaviour in EnumPrinters and GetPrinters\n");
6994 servername = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
6996 torture_assert(tctx,
6997 test_EnumPrinters_level(tctx, b, flags, servername, 2, &count, &info),
6998 "failed to enumerate printers");
7000 for (i=0; i < count; i++) {
7002 torture_assert_casestr_equal(tctx, info[i].info2.servername, servername,
7003 "unexpected servername");
7005 torture_assert(tctx,
7006 test_OnePrinter_servername(tctx, p, b, servername, info[i].info2.printername),
7007 "failed to check printer");
7012 torture_assert(tctx,
7013 test_EnumPrinters_level(tctx, b, flags, servername, 2, &count, &info),
7014 "failed to enumerate printers");
7016 for (i=0; i < count; i++) {
7018 torture_assert_casestr_equal(tctx, info[i].info2.servername, NULL,
7019 "unexpected servername");
7021 torture_assert(tctx,
7022 test_OnePrinter_servername(tctx, p, b, servername, info[i].info2.printername),
7023 "failed to check printer");
7031 static bool test_GetPrinterDriver(struct torture_context *tctx,
7032 struct dcerpc_binding_handle *b,
7033 struct policy_handle *handle,
7034 const char *driver_name)
7036 struct spoolss_GetPrinterDriver r;
7039 r.in.handle = handle;
7040 r.in.architecture = "W32X86";
7044 r.out.needed = &needed;
7046 torture_comment(tctx, "Testing GetPrinterDriver level %d\n", r.in.level);
7048 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_GetPrinterDriver_r(b, tctx, &r),
7049 "failed to call GetPrinterDriver");
7050 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
7051 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
7052 r.in.buffer = &blob;
7053 r.in.offered = needed;
7054 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_GetPrinterDriver_r(b, tctx, &r),
7055 "failed to call GetPrinterDriver");
7058 torture_assert_werr_ok(tctx, r.out.result,
7059 "failed to call GetPrinterDriver");
7061 CHECK_NEEDED_SIZE_LEVEL(spoolss_DriverInfo, r.out.info, r.in.level, needed, 4);
7067 static bool test_GetPrinterDriver2_level(struct torture_context *tctx,
7068 struct dcerpc_binding_handle *b,
7069 struct policy_handle *handle,
7070 const char *driver_name,
7071 const char *architecture,
7073 uint32_t client_major_version,
7074 uint32_t client_minor_version,
7075 union spoolss_DriverInfo *info_p,
7079 struct spoolss_GetPrinterDriver2 r;
7081 uint32_t server_major_version;
7082 uint32_t server_minor_version;
7084 r.in.handle = handle;
7085 r.in.architecture = architecture;
7086 r.in.client_major_version = client_major_version;
7087 r.in.client_minor_version = client_minor_version;
7091 r.out.needed = &needed;
7092 r.out.server_major_version = &server_major_version;
7093 r.out.server_minor_version = &server_minor_version;
7095 torture_comment(tctx, "Testing GetPrinterDriver2(%s) level %d\n",
7096 driver_name, r.in.level);
7098 torture_assert_ntstatus_ok(tctx,
7099 dcerpc_spoolss_GetPrinterDriver2_r(b, tctx, &r),
7100 "failed to call GetPrinterDriver2");
7101 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
7102 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
7103 r.in.buffer = &blob;
7104 r.in.offered = needed;
7105 torture_assert_ntstatus_ok(tctx,
7106 dcerpc_spoolss_GetPrinterDriver2_r(b, tctx, &r),
7107 "failed to call GetPrinterDriver2");
7111 *result_p = r.out.result;
7114 if (W_ERROR_EQUAL(r.out.result, WERR_INVALID_LEVEL)) {
7115 switch (r.in.level) {
7118 torture_comment(tctx,
7119 "level %d not implemented, not considering as an error\n",
7127 torture_assert_werr_ok(tctx, r.out.result,
7128 "failed to call GetPrinterDriver2");
7130 CHECK_NEEDED_SIZE_LEVEL(spoolss_DriverInfo, r.out.info, r.in.level, needed, 4);
7133 *info_p = *r.out.info;
7139 static bool test_GetPrinterDriver2(struct torture_context *tctx,
7140 struct dcerpc_binding_handle *b,
7141 struct policy_handle *handle,
7142 const char *driver_name,
7143 const char *architecture)
7145 uint16_t levels[] = {1, 2, 3, 4, 5, 6, 8, 101 };
7149 for (i=0;i<ARRAY_SIZE(levels);i++) {
7151 torture_assert(tctx,
7152 test_GetPrinterDriver2_level(tctx, b, handle, driver_name, architecture, levels[i], 3, 0, NULL, NULL),
7159 static bool test_EnumPrinterDrivers_old(struct torture_context *tctx,
7162 struct test_spoolss_context *ctx =
7163 talloc_get_type_abort(private_data, struct test_spoolss_context);
7164 uint16_t levels[] = {1, 2, 3, 4, 5, 6};
7166 struct dcerpc_pipe *p = ctx->spoolss_pipe;
7167 struct dcerpc_binding_handle *b = p->binding_handle;
7168 const char *server_name = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
7170 for (i=0;i<ARRAY_SIZE(levels);i++) {
7173 union spoolss_DriverInfo *info;
7175 torture_assert(tctx,
7176 test_EnumPrinterDrivers_args(tctx, b, server_name, ctx->environment, levels[i], &count, &info),
7177 "failed to enumerate drivers");
7180 torture_comment(tctx, "No printer drivers returned\n");
7188 static bool test_DeletePrinter(struct torture_context *tctx,
7189 struct dcerpc_binding_handle *b,
7190 struct policy_handle *handle)
7192 struct spoolss_DeletePrinter r;
7194 torture_comment(tctx, "Testing DeletePrinter\n");
7196 r.in.handle = handle;
7198 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_DeletePrinter_r(b, tctx, &r),
7199 "failed to delete printer");
7200 torture_assert_werr_ok(tctx, r.out.result,
7201 "failed to delete printer");
7206 static bool test_EnumPrinters_findname(struct torture_context *tctx,
7207 struct dcerpc_binding_handle *b,
7213 struct spoolss_EnumPrinters e;
7215 union spoolss_PrinterInfo *info;
7226 e.out.count = &count;
7228 e.out.needed = &needed;
7230 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_EnumPrinters_r(b, tctx, &e),
7231 "failed to enum printers");
7233 if (W_ERROR_EQUAL(e.out.result, WERR_INSUFFICIENT_BUFFER)) {
7234 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
7235 e.in.buffer = &blob;
7236 e.in.offered = needed;
7238 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_EnumPrinters_r(b, tctx, &e),
7239 "failed to enum printers");
7242 torture_assert_werr_ok(tctx, e.out.result,
7243 "failed to enum printers");
7245 for (i=0; i < count; i++) {
7247 const char *current = NULL;
7252 current = info[i].info1.name;
7256 if (strequal(current, name)) {
7261 q = strrchr(current, '\\');
7264 torture_warning(tctx,
7265 "server returns printername %s incl. servername although we did not set servername", current);
7268 if (strequal(q, name)) {
7278 static bool test_AddPrinter_wellknown(struct torture_context *tctx,
7279 struct dcerpc_pipe *p,
7280 const char *printername,
7284 struct spoolss_AddPrinter r;
7285 struct spoolss_AddPrinterEx rex;
7286 struct spoolss_SetPrinterInfoCtr info_ctr;
7287 struct spoolss_SetPrinterInfo1 info1;
7288 struct spoolss_DevmodeContainer devmode_ctr;
7289 struct sec_desc_buf secdesc_ctr;
7290 struct spoolss_UserLevelCtr userlevel_ctr;
7291 struct policy_handle handle;
7293 struct dcerpc_binding_handle *b = p->binding_handle;
7295 ZERO_STRUCT(devmode_ctr);
7296 ZERO_STRUCT(secdesc_ctr);
7297 ZERO_STRUCT(userlevel_ctr);
7300 torture_comment(tctx, "Testing AddPrinter%s(%s) level 1\n",
7301 ex ? "Ex":"", printername);
7303 /* try to add printer to wellknown printer list (level 1) */
7305 userlevel_ctr.level = 1;
7307 info_ctr.info.info1 = &info1;
7310 rex.in.server = NULL;
7311 rex.in.info_ctr = &info_ctr;
7312 rex.in.devmode_ctr = &devmode_ctr;
7313 rex.in.secdesc_ctr = &secdesc_ctr;
7314 rex.in.userlevel_ctr = &userlevel_ctr;
7315 rex.out.handle = &handle;
7318 r.in.info_ctr = &info_ctr;
7319 r.in.devmode_ctr = &devmode_ctr;
7320 r.in.secdesc_ctr = &secdesc_ctr;
7321 r.out.handle = &handle;
7323 torture_assert_ntstatus_ok(tctx, ex ? dcerpc_spoolss_AddPrinterEx_r(b, tctx, &rex) :
7324 dcerpc_spoolss_AddPrinter_r(b, tctx, &r),
7325 "failed to add printer");
7326 result = ex ? rex.out.result : r.out.result;
7327 torture_assert_werr_equal(tctx, result, WERR_INVALID_PRINTER_NAME,
7328 "unexpected result code");
7330 info1.name = printername;
7331 info1.flags = PRINTER_ATTRIBUTE_SHARED;
7333 torture_assert_ntstatus_ok(tctx, ex ? dcerpc_spoolss_AddPrinterEx_r(b, tctx, &rex) :
7334 dcerpc_spoolss_AddPrinter_r(b, tctx, &r),
7335 "failed to add printer");
7336 result = ex ? rex.out.result : r.out.result;
7337 torture_assert_werr_equal(tctx, result, WERR_PRINTER_ALREADY_EXISTS,
7338 "unexpected result code");
7340 /* bizarre protocol, WERR_PRINTER_ALREADY_EXISTS means success here,
7341 better do a real check to see the printer is really there */
7343 torture_assert(tctx, test_EnumPrinters_findname(tctx, b,
7344 PRINTER_ENUM_NETWORK, 1,
7347 "failed to enum printers");
7349 torture_assert(tctx, found, "failed to find newly added printer");
7353 torture_assert_ntstatus_ok(tctx, ex ? dcerpc_spoolss_AddPrinterEx_r(b, tctx, &rex) :
7354 dcerpc_spoolss_AddPrinter_r(b, tctx, &r),
7355 "failed to add printer");
7356 result = ex ? rex.out.result : r.out.result;
7357 torture_assert_werr_equal(tctx, result, WERR_PRINTER_ALREADY_EXISTS,
7358 "unexpected result code");
7360 /* bizarre protocol, WERR_PRINTER_ALREADY_EXISTS means success here,
7361 better do a real check to see the printer has really been removed
7362 from the well known printer list */
7366 torture_assert(tctx, test_EnumPrinters_findname(tctx, b,
7367 PRINTER_ENUM_NETWORK, 1,
7370 "failed to enum printers");
7372 torture_assert(tctx, !found, "printer still in well known printer list");
7377 static bool test_AddPrinter_normal(struct torture_context *tctx,
7378 struct dcerpc_pipe *p,
7379 struct policy_handle *handle_p,
7380 const char *printername,
7381 const char *drivername,
7382 const char *portname,
7383 struct spoolss_DeviceMode *devmode,
7387 struct spoolss_AddPrinter r;
7388 struct spoolss_AddPrinterEx rex;
7389 struct spoolss_SetPrinterInfoCtr info_ctr;
7390 struct spoolss_SetPrinterInfo2 info2;
7391 struct spoolss_DevmodeContainer devmode_ctr;
7392 struct sec_desc_buf secdesc_ctr;
7393 struct spoolss_UserLevelCtr userlevel_ctr;
7394 struct policy_handle handle;
7396 bool existing_printer_deleted = false;
7397 struct dcerpc_binding_handle *b = p->binding_handle;
7399 ZERO_STRUCT(devmode_ctr);
7400 ZERO_STRUCT(secdesc_ctr);
7401 ZERO_STRUCT(userlevel_ctr);
7403 torture_comment(tctx, "Testing AddPrinter%s(%s) level 2\n",
7404 ex ? "Ex":"", printername);
7406 devmode_ctr.devmode = devmode;
7408 userlevel_ctr.level = 1;
7410 rex.in.server = NULL;
7411 rex.in.info_ctr = &info_ctr;
7412 rex.in.devmode_ctr = &devmode_ctr;
7413 rex.in.secdesc_ctr = &secdesc_ctr;
7414 rex.in.userlevel_ctr = &userlevel_ctr;
7415 rex.out.handle = &handle;
7418 r.in.info_ctr = &info_ctr;
7419 r.in.devmode_ctr = &devmode_ctr;
7420 r.in.secdesc_ctr = &secdesc_ctr;
7421 r.out.handle = &handle;
7425 /* try to add printer to printer list (level 2) */
7429 info_ctr.info.info2 = &info2;
7432 torture_assert_ntstatus_ok(tctx, ex ? dcerpc_spoolss_AddPrinterEx_r(b, tctx, &rex) :
7433 dcerpc_spoolss_AddPrinter_r(b, tctx, &r),
7434 "failed to add printer");
7435 result = ex ? rex.out.result : r.out.result;
7436 torture_assert_werr_equal(tctx, result, WERR_INVALID_PRINTER_NAME,
7437 "unexpected result code");
7439 info2.printername = printername;
7441 torture_assert_ntstatus_ok(tctx, ex ? dcerpc_spoolss_AddPrinterEx_r(b, tctx, &rex) :
7442 dcerpc_spoolss_AddPrinter_r(b, tctx, &r),
7443 "failed to add printer");
7444 result = ex ? rex.out.result : r.out.result;
7446 if (W_ERROR_EQUAL(result, WERR_PRINTER_ALREADY_EXISTS)) {
7447 struct policy_handle printer_handle;
7449 if (existing_printer_deleted) {
7450 torture_fail(tctx, "already deleted printer still existing?");
7453 torture_assert(tctx, call_OpenPrinterEx(tctx, p, printername, NULL, &printer_handle),
7454 "failed to open printer handle");
7456 torture_assert(tctx, test_DeletePrinter(tctx, b, &printer_handle),
7457 "failed to delete printer");
7459 torture_assert(tctx, test_ClosePrinter(tctx, b, &printer_handle),
7460 "failed to close server handle");
7462 existing_printer_deleted = true;
7467 torture_assert_werr_equal(tctx, result, WERR_UNKNOWN_PORT,
7468 "unexpected result code");
7470 info2.portname = portname;
7472 torture_assert_ntstatus_ok(tctx, ex ? dcerpc_spoolss_AddPrinterEx_r(b, tctx, &rex) :
7473 dcerpc_spoolss_AddPrinter_r(b, tctx, &r),
7474 "failed to add printer");
7475 result = ex ? rex.out.result : r.out.result;
7476 torture_assert_werr_equal(tctx, result, WERR_UNKNOWN_PRINTER_DRIVER,
7477 "unexpected result code");
7479 info2.drivername = drivername;
7481 torture_assert_ntstatus_ok(tctx, ex ? dcerpc_spoolss_AddPrinterEx_r(b, tctx, &rex) :
7482 dcerpc_spoolss_AddPrinter_r(b, tctx, &r),
7483 "failed to add printer");
7484 result = ex ? rex.out.result : r.out.result;
7486 /* w2k8r2 allows one to add printer w/o defining printprocessor */
7488 if (!W_ERROR_IS_OK(result)) {
7489 torture_assert_werr_equal(tctx, result, WERR_UNKNOWN_PRINTPROCESSOR,
7490 "unexpected result code");
7492 info2.printprocessor = "winprint";
7494 torture_assert_ntstatus_ok(tctx, ex ? dcerpc_spoolss_AddPrinterEx_r(b, tctx, &rex) :
7495 dcerpc_spoolss_AddPrinter_r(b, tctx, &r),
7496 "failed to add printer");
7497 result = ex ? rex.out.result : r.out.result;
7498 torture_assert_werr_ok(tctx, result,
7499 "failed to add printer");
7504 /* we are paranoid, really check if the printer is there now */
7506 torture_assert(tctx, test_EnumPrinters_findname(tctx, b,
7507 PRINTER_ENUM_LOCAL, 1,
7510 "failed to enum printers");
7511 torture_assert(tctx, found, "failed to find newly added printer");
7513 torture_assert_ntstatus_ok(tctx, ex ? dcerpc_spoolss_AddPrinterEx_r(b, tctx, &rex) :
7514 dcerpc_spoolss_AddPrinter_r(b, tctx, &r),
7515 "failed to add printer");
7516 result = ex ? rex.out.result : r.out.result;
7517 torture_assert_werr_equal(tctx, result, WERR_PRINTER_ALREADY_EXISTS,
7518 "unexpected result code");
7523 static bool test_printer_info(struct torture_context *tctx,
7526 struct torture_printer_context *t =
7527 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
7528 struct dcerpc_pipe *p = t->spoolss_pipe;
7529 struct dcerpc_binding_handle *b = p->binding_handle;
7533 if (torture_setting_bool(tctx, "samba3", false)) {
7534 torture_skip(tctx, "skipping printer info cross tests against samba 3");
7537 if (!test_PrinterInfo(tctx, b, &t->handle)) {
7541 if (!test_SetPrinter_errors(tctx, b, &t->handle)) {
7548 static bool test_EnumPrinterKey(struct torture_context *tctx,
7549 struct dcerpc_binding_handle *b,
7550 struct policy_handle *handle,
7551 const char *key_name,
7552 const char ***array)
7554 struct spoolss_EnumPrinterKey r;
7555 uint32_t needed = 0;
7556 union spoolss_KeyNames key_buffer;
7557 int32_t offered[] = { 0, 1, 2, 3, 4, 5, -1, -2, -3, -4, -5, 256, 512, 1024, 2048 };
7561 r.in.handle = handle;
7562 r.in.key_name = key_name;
7563 r.out.key_buffer = &key_buffer;
7564 r.out.needed = &needed;
7565 r.out._ndr_size = &_ndr_size;
7567 for (i=0; i < ARRAY_SIZE(offered); i++) {
7569 if (offered[i] < 0 && needed) {
7573 r.in.offered = needed + offered[i];
7575 r.in.offered = offered[i];
7578 ZERO_STRUCT(key_buffer);
7580 torture_comment(tctx, "Testing EnumPrinterKey(%s) with %d offered\n", r.in.key_name, r.in.offered);
7582 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_EnumPrinterKey_r(b, tctx, &r),
7583 "failed to call EnumPrinterKey");
7584 if (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA)) {
7586 torture_assert(tctx, (_ndr_size == r.in.offered/2),
7587 talloc_asprintf(tctx, "EnumPrinterKey size mismatch, _ndr_size %d (expected %d)",
7588 _ndr_size, r.in.offered/2));
7590 r.in.offered = needed;
7591 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_EnumPrinterKey_r(b, tctx, &r),
7592 "failed to call EnumPrinterKey");
7595 if (offered[i] > 0) {
7596 torture_assert_werr_ok(tctx, r.out.result,
7597 "failed to call EnumPrinterKey");
7600 torture_assert(tctx, (_ndr_size == r.in.offered/2),
7601 talloc_asprintf(tctx, "EnumPrinterKey size mismatch, _ndr_size %d (expected %d)",
7602 _ndr_size, r.in.offered/2));
7604 torture_assert(tctx, (*r.out.needed <= r.in.offered),
7605 talloc_asprintf(tctx, "EnumPrinterKey size mismatch: needed %d is not <= offered %d", *r.out.needed, r.in.offered));
7607 torture_assert(tctx, (*r.out.needed <= _ndr_size * 2),
7608 talloc_asprintf(tctx, "EnumPrinterKey size mismatch: needed %d is not <= _ndr_size %d * 2", *r.out.needed, _ndr_size));
7610 if (key_buffer.string_array) {
7611 uint32_t calc_needed = 0;
7613 for (s=0; key_buffer.string_array[s]; s++) {
7614 calc_needed += strlen_m_term(key_buffer.string_array[s])*2;
7616 if (!key_buffer.string_array[0]) {
7621 torture_assert_int_equal(tctx, *r.out.needed, calc_needed,
7622 "EnumPrinterKey unexpected size");
7627 *array = key_buffer.string_array;
7633 bool test_printer_all_keys(struct torture_context *tctx,
7634 struct dcerpc_binding_handle *b,
7635 struct policy_handle *handle)
7637 const char **key_array = NULL;
7640 torture_comment(tctx, "Testing Printer Keys\n");
7642 torture_assert(tctx, test_EnumPrinterKey(tctx, b, handle, "", &key_array),
7643 "failed to call test_EnumPrinterKey");
7645 for (i=0; key_array && key_array[i]; i++) {
7646 torture_assert(tctx, test_EnumPrinterKey(tctx, b, handle, key_array[i], NULL),
7647 "failed to call test_EnumPrinterKey");
7649 for (i=0; key_array && key_array[i]; i++) {
7650 torture_assert(tctx, test_EnumPrinterDataEx(tctx, b, handle, key_array[i], NULL, NULL),
7651 "failed to call test_EnumPrinterDataEx");
7654 torture_comment(tctx, "Printer Keys test succeeded\n\n");
7659 static bool test_openprinter_wrap(struct torture_context *tctx,
7662 struct torture_printer_context *t =
7663 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
7664 struct dcerpc_pipe *p = t->spoolss_pipe;
7665 struct dcerpc_binding_handle *b = p->binding_handle;
7666 const char *printername = t->info2.printername;
7668 return test_openprinter(tctx, b, printername);
7671 static bool test_csetprinter(struct torture_context *tctx,
7674 struct torture_printer_context *t =
7675 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
7676 struct dcerpc_pipe *p = t->spoolss_pipe;
7678 const char *printername = talloc_asprintf(tctx, "%s2", t->info2.printername);
7679 const char *drivername = t->added_driver ? t->driver.info8.driver_name : t->info2.drivername;
7680 const char *portname = t->info2.portname;
7682 union spoolss_PrinterInfo info;
7683 struct policy_handle new_handle, new_handle2;
7684 struct dcerpc_binding_handle *b = p->binding_handle;
7686 torture_comment(tctx, "Testing c_setprinter\n");
7688 torture_assert(tctx,
7689 test_GetPrinter_level(tctx, b, &t->handle, 0, &info),
7690 "failed to get level 0 printer info");
7691 torture_comment(tctx, "csetprinter on initial printer handle: %d\n",
7692 info.info0.c_setprinter);
7694 /* check if c_setprinter on 1st handle increases after a printer has
7697 torture_assert(tctx,
7698 test_AddPrinter_normal(tctx, p, &new_handle, printername, drivername, portname, NULL, false),
7699 "failed to add new printer");
7700 torture_assert(tctx,
7701 test_GetPrinter_level(tctx, b, &t->handle, 0, &info),
7702 "failed to get level 0 printer info");
7703 torture_comment(tctx, "csetprinter on initial printer handle (after add): %d\n",
7704 info.info0.c_setprinter);
7706 /* check if c_setprinter on new handle increases after a printer has
7709 torture_assert(tctx,
7710 test_GetPrinter_level(tctx, b, &new_handle, 0, &info),
7711 "failed to get level 0 printer info");
7712 torture_comment(tctx, "csetprinter on created handle: %d\n",
7713 info.info0.c_setprinter);
7715 /* open the new printer and check if c_setprinter increases */
7717 torture_assert(tctx,
7718 call_OpenPrinterEx(tctx, p, printername, NULL, &new_handle2),
7719 "failed to open created printer");
7720 torture_assert(tctx,
7721 test_GetPrinter_level(tctx, b, &new_handle2, 0, &info),
7722 "failed to get level 0 printer info");
7723 torture_comment(tctx, "csetprinter on new handle (after openprinter): %d\n",
7724 info.info0.c_setprinter);
7728 torture_assert(tctx,
7729 test_ClosePrinter(tctx, b, &new_handle2),
7730 "failed to close printer");
7731 torture_assert(tctx,
7732 test_DeletePrinter(tctx, b, &new_handle),
7733 "failed to delete new printer");
7738 static bool compose_local_driver_directory(struct torture_context *tctx,
7739 const char *environment,
7740 const char *local_dir,
7745 p = strrchr(local_dir, '/');
7751 if (strequal(environment, "Windows x64")) {
7752 if (!strequal(p, "x64")) {
7753 *path = talloc_asprintf(tctx, "%s/x64", local_dir);
7755 } else if (strequal(environment, "Windows NT x86")) {
7756 if (!strequal(p, "i386")) {
7757 *path = talloc_asprintf(tctx, "%s/i386", local_dir);
7760 torture_assert(tctx, "unknown environment: '%s'\n", environment);
7767 static struct spoolss_DeviceMode *torture_devicemode(TALLOC_CTX *mem_ctx,
7768 const char *devicename)
7770 struct spoolss_DeviceMode *r;
7772 r = talloc_zero(mem_ctx, struct spoolss_DeviceMode);
7777 r->devicename = talloc_strdup(r, devicename);
7778 r->specversion = DMSPEC_NT4_AND_ABOVE;
7779 r->driverversion = 0x0600;
7781 r->__driverextra_length = 0;
7782 r->fields = DEVMODE_FORMNAME |
7784 DEVMODE_PRINTQUALITY |
7785 DEVMODE_DEFAULTSOURCE |
7789 DEVMODE_ORIENTATION;
7790 r->orientation = DMORIENT_PORTRAIT;
7791 r->papersize = DMPAPER_LETTER;
7796 r->defaultsource = DMBIN_FORMSOURCE;
7797 r->printquality = DMRES_HIGH;
7798 r->color = DMRES_MONOCHROME;
7799 r->duplex = DMDUP_SIMPLEX;
7801 r->ttoption = DMTT_SUBDEV;
7802 r->collate = DMCOLLATE_FALSE;
7803 r->formname = talloc_strdup(r, "Letter");
7809 static bool test_architecture_buffer(struct torture_context *tctx,
7812 struct test_spoolss_context *ctx =
7813 talloc_get_type_abort(private_data, struct test_spoolss_context);
7815 struct spoolss_OpenPrinterEx r;
7816 struct spoolss_UserLevel1 u1;
7817 struct policy_handle handle;
7818 uint32_t architectures[] = {
7819 PROCESSOR_ARCHITECTURE_INTEL,
7820 PROCESSOR_ARCHITECTURE_IA64,
7821 PROCESSOR_ARCHITECTURE_AMD64
7825 struct dcerpc_pipe *p = ctx->spoolss_pipe;
7826 struct dcerpc_binding_handle *b = p->binding_handle;
7828 for (i=0; i < ARRAY_SIZE(architectures); i++) {
7830 torture_comment(tctx, "Testing OpenPrinterEx with architecture %d\n", architectures[i]);
7838 u1.processor = architectures[i];
7840 r.in.printername = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
7841 r.in.datatype = NULL;
7842 r.in.devmode_ctr.devmode= NULL;
7843 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
7844 r.in.userlevel_ctr.level = 1;
7845 r.in.userlevel_ctr.user_info.level1 = &u1;
7846 r.out.handle = &handle;
7848 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_OpenPrinterEx_r(b, tctx, &r), "");
7849 torture_assert_werr_ok(tctx, r.out.result, "");
7852 struct spoolss_EnumPrinters e;
7854 union spoolss_PrinterInfo *info;
7856 e.in.flags = PRINTER_ENUM_LOCAL;
7861 e.out.count = &count;
7863 e.out.needed = &needed[i];
7865 torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_EnumPrinters_r(b, tctx, &e), "");
7867 torture_comment(tctx, "needed was %d\n", needed[i]);
7871 torture_assert(tctx, test_ClosePrinter(tctx, b, &handle), "");
7874 for (i=1; i < ARRAY_SIZE(architectures); i++) {
7875 if (needed[i-1] != needed[i]) {
7877 talloc_asprintf(tctx, "needed size %d for architecture %d != needed size %d for architecture %d\n",
7878 needed[i-1], architectures[i-1], needed[i], architectures[i]));
7885 static bool test_get_core_printer_drivers_arch_guid(struct torture_context *tctx,
7886 struct dcerpc_pipe *p,
7887 const char *architecture,
7888 const char *guid_str,
7889 const char **package_id)
7891 struct spoolss_GetCorePrinterDrivers r;
7892 struct spoolss_CorePrinterDriver core_printer_drivers;
7895 struct dcerpc_binding_handle *b = p->binding_handle;
7898 s = talloc_zero_array(tctx, const char *, 2);
7900 r.in.servername = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
7901 r.in.architecture = "foobar";
7902 r.in.core_driver_size = 0;
7903 r.in.core_driver_dependencies = "";
7904 r.in.core_printer_driver_count = 0;
7905 r.out.core_printer_drivers = &core_printer_drivers;
7907 torture_assert_ntstatus_ok(tctx,
7908 dcerpc_spoolss_GetCorePrinterDrivers_r(b, tctx, &r),
7909 "spoolss_GetCorePrinterDrivers failed");
7910 torture_assert_hresult_equal(tctx, r.out.result, HRES_E_INVALIDARG,
7911 "spoolss_GetCorePrinterDrivers failed");
7913 guid = GUID_random();
7914 s[0] = GUID_string2(tctx, &guid);
7916 torture_assert(tctx,
7917 push_reg_multi_sz(tctx, &blob, s),
7918 "push_reg_multi_sz failed");
7920 r.in.core_driver_size = blob.length;
7921 r.in.core_driver_dependencies = s[0];
7922 r.in.core_printer_driver_count = 1;
7923 r.out.core_printer_drivers = talloc_zero_array(tctx, struct spoolss_CorePrinterDriver, r.in.core_printer_driver_count);
7925 torture_assert_ntstatus_ok(tctx,
7926 dcerpc_spoolss_GetCorePrinterDrivers_r(b, tctx, &r),
7927 "spoolss_GetCorePrinterDrivers failed");
7928 torture_assert_werr_equal(tctx,
7929 W_ERROR(WIN32_FROM_HRESULT(r.out.result)), WERR_INVALID_ENVIRONMENT,
7930 "spoolss_GetCorePrinterDrivers failed");
7932 r.in.architecture = architecture;
7934 torture_assert_ntstatus_ok(tctx,
7935 dcerpc_spoolss_GetCorePrinterDrivers_r(b, tctx, &r),
7936 "spoolss_GetCorePrinterDrivers failed");
7937 torture_assert_werr_equal(tctx,
7938 W_ERROR(WIN32_FROM_HRESULT(r.out.result)), WERR_NOT_FOUND,
7939 "spoolss_GetCorePrinterDrivers failed");
7941 s[0] = talloc_strdup(s, guid_str);
7943 torture_assert(tctx,
7944 push_reg_multi_sz(tctx, &blob, s),
7945 "push_reg_multi_sz failed");
7947 r.in.core_driver_size = blob.length;
7948 r.in.core_driver_dependencies = s[0];
7949 r.in.core_printer_driver_count = 1;
7950 r.out.core_printer_drivers = talloc_zero_array(tctx, struct spoolss_CorePrinterDriver, r.in.core_printer_driver_count);
7952 torture_assert_ntstatus_ok(tctx,
7953 dcerpc_spoolss_GetCorePrinterDrivers_r(b, tctx, &r),
7954 "spoolss_GetCorePrinterDrivers failed");
7955 torture_assert_hresult_ok(tctx, r.out.result,
7956 "spoolss_GetCorePrinterDrivers failed");
7959 *package_id = r.out.core_printer_drivers[0].szPackageID;
7965 static bool test_get_core_printer_drivers(struct torture_context *tctx,
7968 struct test_spoolss_context *ctx =
7969 talloc_get_type_abort(private_data, struct test_spoolss_context);
7971 const char *architectures[] = {
7972 SPOOLSS_ARCHITECTURE_NT_X86,
7973 SPOOLSS_ARCHITECTURE_x64
7976 struct dcerpc_pipe *p = ctx->spoolss_pipe;
7978 for (i=0; i < ARRAY_SIZE(architectures); i++) {
7980 torture_comment(tctx, "Testing GetCorePrinterDrivers(\"%s\",\"%s\")\n",
7982 SPOOLSS_CORE_PRINT_PACKAGE_FILES_XPSDRV);
7984 torture_assert(tctx,
7985 test_get_core_printer_drivers_arch_guid(tctx, p,
7987 SPOOLSS_CORE_PRINT_PACKAGE_FILES_XPSDRV,
7995 static bool test_get_printer_driver_package_path(struct torture_context *tctx,
7998 struct test_spoolss_context *ctx =
7999 talloc_get_type_abort(private_data, struct test_spoolss_context);
8001 const char *architectures[] = {
8002 SPOOLSS_ARCHITECTURE_NT_X86,
8003 SPOOLSS_ARCHITECTURE_x64
8006 struct dcerpc_pipe *p = ctx->spoolss_pipe;
8007 struct dcerpc_binding_handle *b = p->binding_handle;
8009 for (i=0; i < ARRAY_SIZE(architectures); i++) {
8010 struct spoolss_GetPrinterDriverPackagePath r;
8011 uint32_t required = 0;
8012 const char *package_id = NULL;
8014 test_get_core_printer_drivers_arch_guid(tctx, p,
8016 SPOOLSS_CORE_PRINT_PACKAGE_FILES_XPSDRV,
8019 torture_comment(tctx, "Testing GetPrinterDriverPackagePath(\"%s\",\"%s\")\n",
8020 architectures[i], package_id);
8022 r.in.servername = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
8023 r.in.architecture = "foobar";
8024 r.in.language = NULL;
8025 r.in.package_id = "";
8026 r.in.driver_package_cab_size = 0;
8027 r.in.driver_package_cab = NULL;
8029 r.out.required = &required;
8030 r.out.driver_package_cab = NULL;
8032 torture_assert_ntstatus_ok(tctx,
8033 dcerpc_spoolss_GetPrinterDriverPackagePath_r(b, tctx, &r),
8034 "spoolss_GetPrinterDriverPackagePath failed");
8035 torture_assert_werr_equal(tctx,
8036 W_ERROR(WIN32_FROM_HRESULT(r.out.result)), WERR_INVALID_ENVIRONMENT,
8037 "spoolss_GetPrinterDriverPackagePath failed");
8039 r.in.architecture = architectures[i];
8041 torture_assert_ntstatus_ok(tctx,
8042 dcerpc_spoolss_GetPrinterDriverPackagePath_r(b, tctx, &r),
8043 "spoolss_GetPrinterDriverPackagePath failed");
8044 torture_assert_werr_equal(tctx,
8045 W_ERROR(WIN32_FROM_HRESULT(r.out.result)), WERR_FILE_NOT_FOUND,
8046 "spoolss_GetPrinterDriverPackagePath failed");
8048 r.in.package_id = package_id;
8050 torture_assert_ntstatus_ok(tctx,
8051 dcerpc_spoolss_GetPrinterDriverPackagePath_r(b, tctx, &r),
8052 "spoolss_GetPrinterDriverPackagePath failed");
8053 torture_assert_hresult_ok(tctx, r.out.result,
8054 "spoolss_GetPrinterDriverPackagePath failed");
8056 r.in.driver_package_cab_size = required;
8057 r.in.driver_package_cab = talloc_zero_array(tctx, char, required);
8058 r.out.driver_package_cab = talloc_zero_array(tctx, char, required);
8060 torture_assert_ntstatus_ok(tctx,
8061 dcerpc_spoolss_GetPrinterDriverPackagePath_r(b, tctx, &r),
8062 "spoolss_GetPrinterDriverPackagePath failed");
8063 torture_assert_hresult_ok(tctx, r.out.result,
8064 "spoolss_GetPrinterDriverPackagePath failed");
8066 r.in.servername = NULL;
8068 torture_assert_ntstatus_ok(tctx,
8069 dcerpc_spoolss_GetPrinterDriverPackagePath_r(b, tctx, &r),
8070 "spoolss_GetPrinterDriverPackagePath failed");
8071 torture_assert_werr_equal(tctx,
8072 W_ERROR(WIN32_FROM_HRESULT(r.out.result)), WERR_INSUFFICIENT_BUFFER,
8073 "spoolss_GetPrinterDriverPackagePath failed");
8075 r.in.driver_package_cab_size = required;
8076 r.in.driver_package_cab = talloc_zero_array(tctx, char, required);
8077 r.out.driver_package_cab = talloc_zero_array(tctx, char, required);
8079 torture_assert_ntstatus_ok(tctx,
8080 dcerpc_spoolss_GetPrinterDriverPackagePath_r(b, tctx, &r),
8081 "spoolss_GetPrinterDriverPackagePath failed");
8082 torture_assert_hresult_ok(tctx, r.out.result,
8083 "spoolss_GetPrinterDriverPackagePath failed");
8090 static bool test_PrintServer_Forms_Winreg(struct torture_context *tctx,
8093 struct test_spoolss_context *ctx =
8094 talloc_get_type_abort(private_data, struct test_spoolss_context);
8095 struct dcerpc_pipe *p = ctx->spoolss_pipe;
8096 struct dcerpc_binding_handle *b = p->binding_handle;
8098 return test_Forms_winreg(tctx, b, &ctx->server_handle, true, NULL);
8101 static bool test_PrintServer_Forms(struct torture_context *tctx,
8104 struct test_spoolss_context *ctx =
8105 talloc_get_type_abort(private_data, struct test_spoolss_context);
8106 struct dcerpc_pipe *p = ctx->spoolss_pipe;
8107 struct dcerpc_binding_handle *b = p->binding_handle;
8109 return test_Forms(tctx, b, &ctx->server_handle, true, NULL, NULL, NULL);
8112 static bool test_PrintServer_EnumForms(struct torture_context *tctx,
8115 struct test_spoolss_context *ctx =
8116 talloc_get_type_abort(private_data, struct test_spoolss_context);
8117 struct dcerpc_pipe *p = ctx->spoolss_pipe;
8118 struct dcerpc_binding_handle *b = p->binding_handle;
8120 return test_EnumForms_all(tctx, b, &ctx->server_handle, true);
8123 static bool torture_rpc_spoolss_setup_common(struct torture_context *tctx, struct test_spoolss_context *t)
8127 status = torture_rpc_connection(tctx, &t->spoolss_pipe, &ndr_table_spoolss);
8129 torture_assert_ntstatus_ok(tctx, status, "Error connecting to server");
8131 torture_assert(tctx,
8132 test_OpenPrinter_server(tctx, t->spoolss_pipe, &t->server_handle),
8133 "failed to open printserver");
8134 torture_assert(tctx,
8135 test_get_environment(tctx, t->spoolss_pipe->binding_handle, &t->server_handle, &t->environment),
8136 "failed to get environment");
8141 static bool torture_rpc_spoolss_setup(struct torture_context *tctx, void **data)
8143 struct test_spoolss_context *t;
8145 *data = t = talloc_zero(tctx, struct test_spoolss_context);
8147 return torture_rpc_spoolss_setup_common(tctx, t);
8150 static bool torture_rpc_spoolss_teardown_common(struct torture_context *tctx, struct test_spoolss_context *t)
8152 test_ClosePrinter(tctx, t->spoolss_pipe->binding_handle, &t->server_handle);
8157 static bool torture_rpc_spoolss_teardown(struct torture_context *tctx, void *data)
8159 struct test_spoolss_context *t = talloc_get_type(data, struct test_spoolss_context);
8162 ret = torture_rpc_spoolss_teardown_common(tctx, t);
8168 static bool torture_rpc_spoolss_printer_setup_common(struct torture_context *tctx, struct torture_printer_context *t)
8170 struct dcerpc_pipe *p;
8171 struct dcerpc_binding_handle *b;
8172 const char *server_name_slash;
8173 const char *driver_name;
8174 const char *printer_name;
8175 const char *port_name;
8177 torture_assert_ntstatus_ok(tctx,
8178 torture_rpc_connection(tctx, &t->spoolss_pipe, &ndr_table_spoolss),
8179 "Error connecting to server");
8181 p = t->spoolss_pipe;
8182 b = p->binding_handle;
8183 server_name_slash = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
8185 t->driver.info8.version = SPOOLSS_DRIVER_VERSION_200X;
8186 t->driver.info8.driver_name = TORTURE_DRIVER;
8187 t->driver.info8.driver_path = "pscript5.dll";
8188 t->driver.info8.data_file = "cups6.ppd";
8189 t->driver.info8.config_file = "ps5ui.dll";
8190 t->driver.info8.help_file = "pscript.hlp";
8191 t->driver.info8.default_datatype = "RAW";
8192 t->driver.info8.dependent_files = talloc_zero(t, struct spoolss_StringArray);
8193 t->driver.info8.dependent_files->string = talloc_zero_array(t, const char *, 8 + 1);
8194 t->driver.info8.dependent_files->string[0] = "pscript5.dll";
8195 t->driver.info8.dependent_files->string[1] = "cups6.ppd";
8196 t->driver.info8.dependent_files->string[2] = "ps5ui.dll";
8197 t->driver.info8.dependent_files->string[3] = "pscript.hlp";
8198 t->driver.info8.dependent_files->string[4] = "pscript.ntf";
8199 t->driver.info8.dependent_files->string[5] = "cups6.ini";
8200 t->driver.info8.dependent_files->string[6] = "cupsps6.dll";
8201 t->driver.info8.dependent_files->string[7] = "cupsui6.dll";
8203 t->driver.local.driver_directory= "/usr/share/cups/drivers";
8205 t->info2.portname = "LPT1:";
8207 printer_name = t->info2.printername;
8208 port_name = t->info2.portname;
8210 torture_assert(tctx,
8211 fillup_printserver_info(tctx, p, &t->driver),
8212 "failed to fillup printserver info");
8214 t->driver.info8.architecture = talloc_strdup(t, t->driver.remote.environment);
8216 torture_assert(tctx,
8217 compose_local_driver_directory(tctx, t->driver.remote.environment,
8218 t->driver.local.driver_directory,
8219 &t->driver.local.driver_directory),
8220 "failed to compose local driver directory");
8222 t->info2.drivername = "Microsoft XPS Document Writer";
8224 if (test_EnumPrinterDrivers_findone(tctx, b, server_name_slash, t->driver.remote.environment, 3, t->info2.drivername, NULL)) {
8225 torture_comment(tctx, "driver '%s' (architecture: %s, version: 3) is present on server\n",
8226 t->info2.drivername, t->driver.remote.environment);
8227 t->have_driver = true;
8231 torture_comment(tctx, "driver '%s' (architecture: %s, version: 3) does not exist on the server\n",
8232 t->info2.drivername, t->driver.remote.environment);
8234 t->info2.drivername = "Microsoft XPS Document Writer v4";
8236 if (test_EnumPrinterDrivers_findone(tctx, b, server_name_slash, t->driver.remote.environment, 3, t->info2.drivername, NULL)) {
8237 torture_comment(tctx, "driver '%s' (architecture: %s, version: 4) is present on server\n",
8238 t->info2.drivername, t->driver.remote.environment);
8239 t->have_driver = true;
8243 torture_comment(tctx, "trying to upload own driver\n");
8245 if (!directory_exist(t->driver.local.driver_directory)) {
8246 torture_warning(tctx, "no local driver is available!");
8247 t->have_driver = false;
8251 torture_assert(tctx,
8252 upload_printer_driver(tctx, dcerpc_server_name(p), &t->driver),
8253 "failed to upload printer driver");
8255 torture_assert(tctx,
8256 test_AddPrinterDriver_args_level_3(tctx, b, server_name_slash, &t->driver.info8, 0, false, NULL),
8257 "failed to add driver");
8259 t->added_driver = true;
8260 t->have_driver = true;
8263 driver_name = t->added_driver ? t->driver.info8.driver_name : t->info2.drivername;
8266 torture_assert(tctx,
8267 test_AddPrinter_wellknown(tctx, p, printer_name, t->ex),
8268 "failed to add wellknown printer");
8270 torture_assert(tctx,
8271 test_AddPrinter_normal(tctx, p, &t->handle, printer_name, driver_name, port_name, t->devmode, t->ex),
8272 "failed to add printer");
8278 static bool torture_rpc_spoolss_printer_setup(struct torture_context *tctx, void **data)
8280 struct torture_printer_context *t;
8282 *data = t = talloc_zero(tctx, struct torture_printer_context);
8285 t->wellknown = false;
8286 t->info2.printername = TORTURE_PRINTER;
8289 return torture_rpc_spoolss_printer_setup_common(tctx, t);
8292 static bool torture_rpc_spoolss_printerex_setup(struct torture_context *tctx, void **data)
8294 struct torture_printer_context *t;
8296 *data = t = talloc_zero(tctx, struct torture_printer_context);
8299 t->wellknown = false;
8300 t->info2.printername = TORTURE_PRINTER_EX;
8303 return torture_rpc_spoolss_printer_setup_common(tctx, t);
8306 static bool torture_rpc_spoolss_printerwkn_setup(struct torture_context *tctx, void **data)
8308 struct torture_printer_context *t;
8310 *data = t = talloc_zero(tctx, struct torture_printer_context);
8313 t->wellknown = true;
8314 t->info2.printername = TORTURE_WELLKNOWN_PRINTER;
8319 torture_skip(tctx, "skipping AddPrinter level 1");
8322 return torture_rpc_spoolss_printer_setup_common(tctx, t);
8325 static bool torture_rpc_spoolss_printerexwkn_setup(struct torture_context *tctx, void **data)
8327 struct torture_printer_context *t;
8329 *data = t = talloc_zero(tctx, struct torture_printer_context);
8332 t->wellknown = true;
8333 t->info2.printername = TORTURE_WELLKNOWN_PRINTER_EX;
8338 torture_skip(tctx, "skipping AddPrinterEx level 1");
8341 return torture_rpc_spoolss_printer_setup_common(tctx, t);
8345 static bool torture_rpc_spoolss_printerdm_setup(struct torture_context *tctx, void **data)
8347 struct torture_printer_context *t;
8349 *data = t = talloc_zero(tctx, struct torture_printer_context);
8352 t->wellknown = false;
8353 t->info2.printername = TORTURE_PRINTER_EX;
8354 t->devmode = torture_devicemode(t, TORTURE_PRINTER_EX);
8356 return torture_rpc_spoolss_printer_setup_common(tctx, t);
8360 static bool torture_rpc_spoolss_printer_teardown_common(struct torture_context *tctx, struct torture_printer_context *t)
8363 struct dcerpc_pipe *p = t->spoolss_pipe;
8364 struct dcerpc_binding_handle *b;
8365 const char *printer_name = t->info2.printername;
8367 if (t->added_driver) {
8368 torture_assert(tctx,
8369 remove_printer_driver(tctx, dcerpc_server_name(t->spoolss_pipe), &t->driver),
8370 "failed to remove printer driver");
8373 if (p && !t->wellknown) {
8374 b = p->binding_handle;
8376 torture_assert(tctx,
8377 test_DeletePrinter(tctx, b, &t->handle),
8378 "failed to delete printer");
8380 torture_assert(tctx,
8381 test_EnumPrinters_findname(tctx, b, PRINTER_ENUM_LOCAL, 1,
8382 printer_name, &found),
8383 "failed to enumerate printers");
8385 torture_assert(tctx, !found, "deleted printer still there");
8391 static bool torture_rpc_spoolss_printer_teardown(struct torture_context *tctx, void *data)
8393 struct torture_printer_context *t = talloc_get_type(data, struct torture_printer_context);
8396 ret = torture_rpc_spoolss_printer_teardown_common(tctx, t);
8402 static bool test_print_test(struct torture_context *tctx,
8405 struct torture_printer_context *t =
8406 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8407 struct dcerpc_pipe *p = t->spoolss_pipe;
8408 struct dcerpc_binding_handle *b = p->binding_handle;
8410 torture_assert(tctx,
8411 test_PausePrinter(tctx, b, &t->handle),
8412 "failed to pause printer");
8414 torture_assert(tctx,
8415 test_DoPrintTest(tctx, b, &t->handle),
8416 "failed to do print test");
8418 torture_assert(tctx,
8419 test_ResumePrinter(tctx, b, &t->handle),
8420 "failed to resume printer");
8425 static bool test_print_test_extended(struct torture_context *tctx,
8428 struct torture_printer_context *t =
8429 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8430 struct dcerpc_pipe *p = t->spoolss_pipe;
8431 struct dcerpc_binding_handle *b = p->binding_handle;
8434 torture_assert(tctx,
8435 test_PausePrinter(tctx, b, &t->handle),
8436 "failed to pause printer");
8438 ret = test_DoPrintTest_extended(tctx, b, &t->handle);
8440 torture_comment(tctx, "WARNING! failed to do extended print test\n");
8441 if (torture_setting_bool(tctx, "samba3", false)) {
8442 torture_comment(tctx, "non-critical for samba3\n");
8444 tctx->last_result = TORTURE_SKIP;
8448 torture_assert(tctx,
8449 test_ResumePrinter(tctx, b, &t->handle),
8450 "failed to resume printer");
8455 static bool test_print_test_properties(struct torture_context *tctx,
8458 struct torture_printer_context *t =
8459 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8460 struct dcerpc_pipe *p = t->spoolss_pipe;
8461 struct dcerpc_binding_handle *b = p->binding_handle;
8463 if (torture_setting_bool(tctx, "samba3", false)) {
8464 torture_skip(tctx, "skip printer job property tests against samba");
8467 torture_assert(tctx,
8468 test_PausePrinter(tctx, b, &t->handle),
8469 "failed to pause printer");
8471 torture_assert(tctx,
8472 test_DoPrintTest_properties(tctx, b, &t->handle),
8473 "failed to test print job properties");
8475 torture_assert(tctx,
8476 test_ResumePrinter(tctx, b, &t->handle),
8477 "failed to resume printer");
8482 /* use smbd file IO to spool a print job */
8483 static bool test_print_test_smbd(struct torture_context *tctx,
8486 struct torture_printer_context *t =
8487 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8488 struct dcerpc_pipe *p = t->spoolss_pipe;
8489 struct dcerpc_binding_handle *b = p->binding_handle;
8492 union spoolss_JobInfo *info = NULL;
8495 struct smb2_tree *tree;
8496 struct smb2_handle job_h;
8497 struct cli_credentials *credentials = cmdline_credentials;
8498 struct smbcli_options options;
8499 TALLOC_CTX *mem_ctx = talloc_new(tctx);
8501 * Do not test against the dynamically added printers, printing via
8502 * smbd means that a different spoolss process may handle the
8503 * OpenPrinter request to the one that handled the AddPrinter request.
8504 * This currently leads to an ugly race condition where one process
8505 * sees the new printer and one doesn't.
8507 const char *share = TORTURE_PRINTER_STATIC1;
8509 torture_comment(tctx, "Testing smbd job spooling\n");
8510 lpcfg_smbcli_options(tctx->lp_ctx, &options);
8512 status = smb2_connect(mem_ctx,
8513 torture_setting_string(tctx, "host", NULL),
8514 lpcfg_smb_ports(tctx->lp_ctx),
8516 lpcfg_resolve_context(tctx->lp_ctx),
8521 lpcfg_socket_options(tctx->lp_ctx),
8522 lpcfg_gensec_settings(tctx, tctx->lp_ctx));
8523 if (!NT_STATUS_IS_OK(status)) {
8524 printf("Failed to connect to SMB2 printer %s - %s\n",
8525 share, nt_errstr(status));
8529 status = torture_smb2_testfile(tree, "smbd_spooler_job", &job_h);
8530 torture_assert_ntstatus_ok(tctx, status, "smbd spool job create");
8532 status = smb2_util_write(tree, job_h, "exciting print job data", 0,
8533 sizeof("exciting print job data"));
8534 torture_assert_ntstatus_ok(tctx, status, "smbd spool job write");
8536 /* check back end spoolss job was created */
8537 torture_assert(tctx,
8538 test_EnumJobs_args(tctx, b, &t->handle, 1, WERR_OK,
8540 "EnumJobs level 1 failed");
8542 for (i = 0; i < count; i++) {
8543 if (!strcmp(info[i].info1.document_name, "smbd_spooler_job")) {
8547 torture_assert(tctx, (i != count), "smbd_spooler_job not found");
8549 status = smb2_util_close(tree, job_h);
8550 torture_assert_ntstatus_ok(tctx, status, "smbd spool job close");
8552 /* disconnect from printer share */
8553 talloc_free(mem_ctx);
8558 static bool test_print_test_purge(struct torture_context *tctx,
8561 struct torture_printer_context *t =
8562 (struct torture_printer_context *)talloc_get_type_abort(private_data,
8563 struct torture_printer_context);
8564 struct dcerpc_pipe *p = t->spoolss_pipe;
8565 struct dcerpc_binding_handle *b = p->binding_handle;
8566 uint32_t num_jobs = 8;
8571 union spoolss_JobInfo *info;
8573 torture_assert(tctx,
8574 test_PausePrinter(tctx, b, &t->handle),
8575 "failed to pause printer");
8577 job_ids = talloc_zero_array(tctx, uint32_t, num_jobs);
8578 for (i=0; i < num_jobs; i++) {
8579 ret = test_DoPrintTest_add_one_job(tctx, b, &t->handle,
8582 torture_assert(tctx, ret, "failed to add print job");
8585 torture_assert(tctx,
8586 test_EnumJobs_args(tctx, b, &t->handle, 1, WERR_OK,
8588 "EnumJobs level 1 failed");
8590 torture_assert_int_equal(tctx, count, num_jobs,
8591 "unexpected number of jobs in queue");
8593 torture_assert(tctx,
8594 test_printer_purge(tctx, b, &t->handle),
8595 "failed to purge printer");
8597 torture_assert(tctx,
8598 test_EnumJobs_args(tctx, b, &t->handle, 1, WERR_OK,
8600 "EnumJobs level 1 failed");
8602 torture_assert_int_equal(tctx, count, 0,
8603 "unexpected number of jobs in queue");
8605 torture_assert(tctx,
8606 test_ResumePrinter(tctx, b, &t->handle),
8607 "failed to resume printer");
8612 static bool test_printer_sd(struct torture_context *tctx,
8615 struct torture_printer_context *t =
8616 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8617 struct dcerpc_pipe *p = t->spoolss_pipe;
8618 struct dcerpc_binding_handle *b = p->binding_handle;
8620 torture_assert(tctx,
8621 test_PrinterInfo_SD(tctx, b, &t->handle),
8622 "failed to test security descriptors");
8627 static bool test_printer_dm(struct torture_context *tctx,
8630 struct torture_printer_context *t =
8631 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8632 struct dcerpc_pipe *p = t->spoolss_pipe;
8634 torture_assert(tctx,
8635 test_PrinterInfo_DevMode(tctx, p, &t->handle, t->info2.printername, t->devmode),
8636 "failed to test devicemodes");
8641 static bool test_printer_info_winreg(struct torture_context *tctx,
8644 struct torture_printer_context *t =
8645 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8646 struct dcerpc_pipe *p = t->spoolss_pipe;
8648 torture_assert(tctx,
8649 test_PrinterInfo_winreg(tctx, p, &t->handle, t->info2.printername),
8650 "failed to test printer info winreg");
8655 static bool test_printer_change_id(struct torture_context *tctx,
8658 struct torture_printer_context *t =
8659 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8660 struct dcerpc_pipe *p = t->spoolss_pipe;
8662 torture_assert(tctx,
8663 test_ChangeID(tctx, p, &t->handle),
8664 "failed to test change id");
8669 static bool test_printer_keys(struct torture_context *tctx,
8672 struct torture_printer_context *t =
8673 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8674 struct dcerpc_pipe *p = t->spoolss_pipe;
8675 struct dcerpc_binding_handle *b = p->binding_handle;
8677 torture_assert(tctx,
8678 test_printer_all_keys(tctx, b, &t->handle),
8679 "failed to test printer keys");
8684 static bool test_printer_data_consistency(struct torture_context *tctx,
8687 struct torture_printer_context *t =
8688 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8689 struct dcerpc_pipe *p = t->spoolss_pipe;
8691 torture_assert(tctx,
8692 test_EnumPrinterData_consistency(tctx, p, &t->handle),
8693 "failed to test printer data consistency");
8698 static bool test_printer_data_keys(struct torture_context *tctx,
8701 struct torture_printer_context *t =
8702 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8703 struct dcerpc_pipe *p = t->spoolss_pipe;
8705 torture_assert(tctx,
8706 test_SetPrinterDataEx_keys(tctx, p, &t->handle),
8707 "failed to test printer data keys");
8712 static bool test_printer_data_values(struct torture_context *tctx,
8715 struct torture_printer_context *t =
8716 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8717 struct dcerpc_pipe *p = t->spoolss_pipe;
8719 torture_assert(tctx,
8720 test_SetPrinterDataEx_values(tctx, p, &t->handle),
8721 "failed to test printer data values");
8726 static bool test_printer_data_set(struct torture_context *tctx,
8729 struct torture_printer_context *t =
8730 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8731 struct dcerpc_pipe *p = t->spoolss_pipe;
8733 torture_assert(tctx,
8734 test_SetPrinterDataEx_matrix(tctx, p, &t->handle, t->info2.printername, NULL, NULL),
8735 "failed to test printer data set");
8740 static bool test_printer_data_winreg(struct torture_context *tctx,
8743 struct torture_printer_context *t =
8744 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8745 struct dcerpc_pipe *p = t->spoolss_pipe;
8747 torture_assert(tctx,
8748 test_PrinterData_winreg(tctx, p, &t->handle, t->info2.printername),
8749 "failed to test printer data winreg");
8754 static bool test_printer_data_dsspooler(struct torture_context *tctx,
8757 struct torture_printer_context *t =
8758 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
8759 struct dcerpc_pipe *p = t->spoolss_pipe;
8761 torture_assert(tctx,
8762 test_PrinterData_DsSpooler(tctx, p, &t->handle, t->info2.printername),
8763 "failed to test printer data winreg dsspooler");
8768 static bool test_printer_ic(struct torture_context *tctx,
8771 struct torture_printer_context *t =
8772 talloc_get_type_abort(private_data,
8773 struct torture_printer_context);
8774 struct dcerpc_pipe *p = t->spoolss_pipe;
8775 struct dcerpc_binding_handle *b = p->binding_handle;
8776 struct policy_handle gdi_handle;
8778 if (torture_setting_bool(tctx, "samba3", false)) {
8779 torture_skip(tctx, "skip printer information context tests against samba");
8783 struct spoolss_CreatePrinterIC r;
8784 struct spoolss_DevmodeContainer devmode_ctr;
8786 ZERO_STRUCT(devmode_ctr);
8788 r.in.handle = &t->handle;
8789 r.in.devmode_ctr = &devmode_ctr;
8790 r.out.gdi_handle = &gdi_handle;
8792 torture_assert_ntstatus_ok(tctx,
8793 dcerpc_spoolss_CreatePrinterIC_r(b, tctx, &r),
8794 "CreatePrinterIC failed");
8795 torture_assert_werr_ok(tctx, r.out.result,
8796 "CreatePrinterIC failed");
8800 struct spoolss_PlayGDIScriptOnPrinterIC r;
8803 uint32_t num_fonts = 0;
8805 in = data_blob_string_const("");
8807 r.in.gdi_handle = &gdi_handle;
8809 r.in.cIn = in.length;
8812 for (i = 0; i < 4; i++) {
8814 out = data_blob_talloc_zero(tctx, i);
8816 r.in.cOut = out.length;
8817 r.out.pOut = out.data;
8819 torture_assert_ntstatus_ok(tctx,
8820 dcerpc_spoolss_PlayGDIScriptOnPrinterIC_r(b, tctx, &r),
8821 "PlayGDIScriptOnPrinterIC failed");
8822 torture_assert_werr_equal(tctx, r.out.result, WERR_NOMEM,
8823 "PlayGDIScriptOnPrinterIC failed");
8826 out = data_blob_talloc_zero(tctx, 4);
8828 r.in.cOut = out.length;
8829 r.out.pOut = out.data;
8831 torture_assert_ntstatus_ok(tctx,
8832 dcerpc_spoolss_PlayGDIScriptOnPrinterIC_r(b, tctx, &r),
8833 "PlayGDIScriptOnPrinterIC failed");
8834 torture_assert_werr_equal(tctx, r.out.result, WERR_OK,
8835 "PlayGDIScriptOnPrinterIC failed");
8837 /* now we should have the required length, so retry with a
8838 * buffer which is large enough to carry all font ids */
8840 num_fonts = IVAL(r.out.pOut, 0);
8842 torture_comment(tctx, "PlayGDIScriptOnPrinterIC gave font count of %d\n", num_fonts);
8844 out = data_blob_talloc_zero(tctx,
8845 num_fonts * sizeof(struct UNIVERSAL_FONT_ID) + 4);
8847 r.in.cOut = out.length;
8848 r.out.pOut = out.data;
8850 torture_assert_ntstatus_ok(tctx,
8851 dcerpc_spoolss_PlayGDIScriptOnPrinterIC_r(b, tctx, &r),
8852 "PlayGDIScriptOnPrinterIC failed");
8853 torture_assert_werr_equal(tctx, r.out.result, WERR_OK,
8854 "PlayGDIScriptOnPrinterIC failed");
8859 struct spoolss_DeletePrinterIC r;
8861 r.in.gdi_handle = &gdi_handle;
8862 r.out.gdi_handle = &gdi_handle;
8864 torture_assert_ntstatus_ok(tctx,
8865 dcerpc_spoolss_DeletePrinterIC_r(b, tctx, &r),
8866 "DeletePrinterIC failed");
8867 torture_assert_werr_ok(tctx, r.out.result,
8868 "DeletePrinterIC failed");
8875 static bool test_printer_bidi(struct torture_context *tctx,
8878 struct torture_printer_context *t =
8879 talloc_get_type_abort(private_data,
8880 struct torture_printer_context);
8881 struct dcerpc_pipe *p = t->spoolss_pipe;
8882 struct dcerpc_binding_handle *b = p->binding_handle;
8883 struct spoolss_SendRecvBidiData r;
8884 struct RPC_BIDI_REQUEST_CONTAINER bidi_req;
8885 struct RPC_BIDI_RESPONSE_CONTAINER *bidi_rep = NULL;
8887 if (torture_setting_bool(tctx, "samba3", false)) {
8888 torture_skip(tctx, "skip printer bidirectional tests against samba");
8891 ZERO_STRUCT(bidi_req);
8893 r.in.hPrinter = t->handle;
8894 r.in.pAction = "foobar";
8895 r.in.pReqData = &bidi_req;
8896 r.out.ppRespData = &bidi_rep;
8898 torture_assert_ntstatus_ok(tctx,
8899 dcerpc_spoolss_SendRecvBidiData_r(b, tctx, &r),
8900 "SendRecvBidiData failed");
8901 torture_assert_werr_equal(tctx, r.out.result, WERR_NOT_SUPPORTED,
8902 "SendRecvBidiData failed");
8904 if (!(t->info2.attributes & PRINTER_ATTRIBUTE_ENABLE_BIDI)) {
8905 torture_skip(tctx, "skipping further tests as printer is not BIDI enabled");
8908 r.in.pAction = BIDI_ACTION_ENUM_SCHEMA;
8910 torture_assert_ntstatus_ok(tctx,
8911 dcerpc_spoolss_SendRecvBidiData_r(b, tctx, &r),
8912 "SendRecvBidiData failed");
8913 torture_assert_werr_ok(tctx, r.out.result,
8914 "SendRecvBidiData failed");
8919 static bool test_printer_set_publish(struct torture_context *tctx,
8920 struct dcerpc_binding_handle *b,
8921 struct policy_handle *handle)
8923 union spoolss_PrinterInfo info;
8924 struct spoolss_SetPrinterInfo7 info7;
8925 struct spoolss_SetPrinterInfoCtr info_ctr;
8926 struct spoolss_DevmodeContainer devmode_ctr;
8927 struct sec_desc_buf secdesc_ctr;
8930 info7.action = DSPRINT_PUBLISH;
8932 ZERO_STRUCT(info_ctr);
8933 ZERO_STRUCT(devmode_ctr);
8934 ZERO_STRUCT(secdesc_ctr);
8936 info_ctr.info.info7 = &info7;
8938 torture_assert(tctx,
8939 test_SetPrinter(tctx, b, handle, &info_ctr,
8940 &devmode_ctr, &secdesc_ctr, 0), "");
8942 torture_assert(tctx,
8943 test_GetPrinter_level(tctx, b, handle, 2, &info),
8945 torture_assert(tctx,
8946 (info.info2.attributes & PRINTER_ATTRIBUTE_PUBLISHED),
8947 "info2 publish flag not set");
8948 torture_assert(tctx,
8949 test_GetPrinter_level(tctx, b, handle, 7, &info),
8951 if (info.info7.action & DSPRINT_PENDING) {
8952 torture_comment(tctx, "publish is pending\n");
8953 torture_assert_int_equal(tctx,
8955 (DSPRINT_PENDING | DSPRINT_PUBLISH),
8956 "info7 publish flag not set");
8959 torture_assert_int_equal(tctx,
8962 "info7 publish flag not set");
8963 torture_assert_ntstatus_ok(tctx,
8964 GUID_from_string(info.info7.guid,
8966 "invalid published printer GUID");
8972 static bool test_printer_set_unpublish(struct torture_context *tctx,
8973 struct dcerpc_binding_handle *b,
8974 struct policy_handle *handle)
8976 union spoolss_PrinterInfo info;
8977 struct spoolss_SetPrinterInfo7 info7;
8978 struct spoolss_SetPrinterInfoCtr info_ctr;
8979 struct spoolss_DevmodeContainer devmode_ctr;
8980 struct sec_desc_buf secdesc_ctr;
8982 info7.action = DSPRINT_UNPUBLISH;
8985 ZERO_STRUCT(info_ctr);
8986 ZERO_STRUCT(devmode_ctr);
8987 ZERO_STRUCT(secdesc_ctr);
8989 info_ctr.info.info7 = &info7;
8991 torture_assert(tctx,
8992 test_SetPrinter(tctx, b, handle, &info_ctr,
8993 &devmode_ctr, &secdesc_ctr, 0), "");
8995 torture_assert(tctx,
8996 test_GetPrinter_level(tctx, b, handle, 2, &info),
8998 torture_assert(tctx,
8999 !(info.info2.attributes & PRINTER_ATTRIBUTE_PUBLISHED),
9000 "info2 publish flag still set");
9001 torture_assert(tctx,
9002 test_GetPrinter_level(tctx, b, handle, 7, &info),
9005 if (info.info7.action & DSPRINT_PENDING) {
9007 torture_comment(tctx, "unpublish is pending\n");
9008 torture_assert_int_equal(tctx,
9010 (DSPRINT_PENDING | DSPRINT_UNPUBLISH),
9011 "info7 unpublish flag not set");
9012 torture_assert_ntstatus_ok(tctx,
9013 GUID_from_string(info.info7.guid,
9015 "invalid printer GUID");
9017 torture_assert_int_equal(tctx,
9018 info.info7.action, DSPRINT_UNPUBLISH,
9019 "info7 unpublish flag not set");
9025 static bool test_printer_publish_toggle(struct torture_context *tctx,
9028 struct torture_printer_context *t =
9029 talloc_get_type_abort(private_data,
9030 struct torture_printer_context);
9031 struct dcerpc_pipe *p = t->spoolss_pipe;
9032 struct dcerpc_binding_handle *b = p->binding_handle;
9033 struct policy_handle *handle = &t->handle;
9034 union spoolss_PrinterInfo info7;
9035 union spoolss_PrinterInfo info2;
9037 /* check publish status via level 7 and level 2 */
9038 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 7, &info7),
9040 torture_assert(tctx, test_GetPrinter_level(tctx, b, handle, 2, &info2),
9043 if (info2.info2.attributes & PRINTER_ATTRIBUTE_PUBLISHED) {
9044 torture_assert_int_equal(tctx,
9045 info7.info7.action, DSPRINT_PUBLISH,
9046 "info7 publish flag not set");
9047 torture_assert(tctx, test_printer_set_unpublish(tctx, b, handle), "");
9048 torture_assert(tctx, test_printer_set_publish(tctx, b, handle), "");
9050 torture_assert_int_equal(tctx,
9051 info7.info7.action, DSPRINT_UNPUBLISH,
9052 "info7 unpublish flag not set");
9053 torture_assert(tctx, test_printer_set_publish(tctx, b, handle), "");
9054 torture_assert(tctx, test_printer_set_unpublish(tctx, b, handle), "");
9060 static bool test_driver_info_winreg(struct torture_context *tctx,
9063 struct torture_printer_context *t =
9064 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
9065 struct dcerpc_pipe *p = t->spoolss_pipe;
9066 const char *driver_name = t->added_driver ? t->driver.info8.driver_name : t->info2.drivername;
9068 if (!t->have_driver) {
9069 torture_skip(tctx, "skipping driver info winreg test as we don't have a driver");
9072 torture_assert(tctx,
9073 test_DriverInfo_winreg(tctx, p, &t->handle, t->info2.printername, driver_name, t->driver.remote.environment, 3),
9074 "failed to test driver info winreg");
9079 static bool test_print_job_enum(struct torture_context *tctx,
9082 struct torture_printer_context *t =
9083 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
9084 struct dcerpc_pipe *p = t->spoolss_pipe;
9085 struct dcerpc_binding_handle *b = p->binding_handle;
9087 uint32_t num_jobs = 8;
9090 union spoolss_JobInfo *info = NULL;
9093 torture_assert(tctx,
9094 test_PausePrinter(tctx, b, &t->handle),
9095 "failed to pause printer");
9097 /* purge in case of any jobs from previous tests */
9098 torture_assert(tctx,
9099 test_printer_purge(tctx, b, &t->handle),
9100 "failed to purge printer");
9102 /* enum before jobs, valid level */
9103 torture_assert(tctx,
9104 test_EnumJobs_args(tctx, b, &t->handle, 1, WERR_OK,
9106 "EnumJobs with valid level");
9107 torture_assert_int_equal(tctx, count, 0, "EnumJobs count");
9108 torture_assert(tctx,
9109 test_EnumJobs_args(tctx, b, &t->handle, 2, WERR_OK,
9111 "EnumJobs with valid level");
9112 torture_assert_int_equal(tctx, count, 0, "EnumJobs count");
9113 /* enum before jobs, invalid level - expect failure */
9114 torture_assert(tctx,
9115 test_EnumJobs_args(tctx, b, &t->handle, 100,
9118 "EnumJobs with invalid level");
9120 job_ids = talloc_zero_array(tctx, uint32_t, num_jobs);
9122 for (i = 0; i < num_jobs; i++) {
9123 ret = test_DoPrintTest_add_one_job(tctx, b, &t->handle,
9126 torture_assert(tctx, ret, "failed to add print job");
9129 /* enum after jobs, valid level */
9130 torture_assert(tctx,
9131 test_EnumJobs_args(tctx, b, &t->handle, 1, WERR_OK,
9133 "EnumJobs with valid level");
9134 torture_assert_int_equal(tctx, count, num_jobs, "EnumJobs count");
9135 torture_assert(tctx,
9136 test_EnumJobs_args(tctx, b, &t->handle, 2, WERR_OK,
9138 "EnumJobs with valid level");
9139 torture_assert_int_equal(tctx, count, num_jobs, "EnumJobs count");
9140 /* enum after jobs, invalid level - expect failure */
9141 torture_assert(tctx,
9142 test_EnumJobs_args(tctx, b, &t->handle, 100,
9145 "EnumJobs with invalid level");
9147 for (i = 0; i < num_jobs; i++) {
9148 test_SetJob(tctx, b, &t->handle, job_ids[i], NULL,
9149 SPOOLSS_JOB_CONTROL_DELETE);
9152 torture_assert(tctx,
9153 test_ResumePrinter(tctx, b, &t->handle),
9154 "failed to resume printer");
9159 static bool test_printer_log_jobinfo(struct torture_context *tctx,
9162 struct torture_printer_context *t =
9163 (struct torture_printer_context *)talloc_get_type_abort(private_data, struct torture_printer_context);
9164 struct dcerpc_pipe *p = t->spoolss_pipe;
9165 struct dcerpc_binding_handle *b = p->binding_handle;
9166 struct spoolss_BranchOfficeJobDataContainer info;
9169 struct spoolss_LogJobInfoForBranchOffice r;
9171 torture_comment(tctx, "Testing LogJobInfoForBranchOffice\n");
9173 info.cJobDataEntries = 0;
9174 info.JobData = NULL;
9176 r.in.hPrinter = &t->handle;
9177 r.in.pBranchOfficeJobDataContainer = &info;
9179 torture_assert_ntstatus_ok(tctx,
9180 dcerpc_spoolss_LogJobInfoForBranchOffice_r(b, tctx, &r),
9181 "LogJobInfoForBranchOffice failed");
9182 torture_assert_werr_equal(tctx, r.out.result, WERR_INVALID_PARAMETER,
9183 "LogJobInfoForBranchOffice failed");
9185 info.cJobDataEntries = 1;
9186 info.JobData = talloc_zero_array(tctx, struct spoolss_BranchOfficeJobData, info.cJobDataEntries);
9188 info.JobData[0].eEventType = kLogOfflineFileFull;
9189 info.JobData[0].JobId = 42;
9190 info.JobData[0].JobInfo.LogOfflineFileFull.pMachineName = talloc_strdup(tctx, "mthelena");
9192 torture_assert_ntstatus_ok(tctx,
9193 dcerpc_spoolss_LogJobInfoForBranchOffice_r(b, tctx, &r),
9194 "LogJobInfoForBranchOffice failed");
9195 torture_assert_werr_equal(tctx, r.out.result, WERR_OK,
9196 "LogJobInfoForBranchOffice failed");
9198 info.cJobDataEntries = 42;
9199 info.JobData = talloc_zero_array(tctx, struct spoolss_BranchOfficeJobData, info.cJobDataEntries);
9201 for (i=0; i < info.cJobDataEntries; i++) {
9202 info.JobData[i].eEventType = kLogOfflineFileFull;
9203 info.JobData[i].JobId = i;
9204 info.JobData[i].JobInfo.LogOfflineFileFull.pMachineName = talloc_asprintf(tctx, "torture_%d", i);
9207 torture_assert_ntstatus_ok(tctx,
9208 dcerpc_spoolss_LogJobInfoForBranchOffice_r(b, tctx, &r),
9209 "LogJobInfoForBranchOffice failed");
9210 torture_assert_werr_equal(tctx, r.out.result, WERR_OK,
9211 "LogJobInfoForBranchOffice failed");
9216 void torture_tcase_printer(struct torture_tcase *tcase)
9218 torture_tcase_add_simple_test(tcase, "openprinter", test_openprinter_wrap);
9219 torture_tcase_add_simple_test(tcase, "csetprinter", test_csetprinter);
9220 torture_tcase_add_simple_test(tcase, "print_test", test_print_test);
9221 torture_tcase_add_simple_test(tcase, "print_test_extended", test_print_test_extended);
9222 torture_tcase_add_simple_test(tcase, "print_test_smbd", test_print_test_smbd);
9223 torture_tcase_add_simple_test(tcase, "print_test_properties", test_print_test_properties);
9224 torture_tcase_add_simple_test(tcase, "print_test_purge", test_print_test_purge);
9225 torture_tcase_add_simple_test(tcase, "printer_info", test_printer_info);
9226 torture_tcase_add_simple_test(tcase, "sd", test_printer_sd);
9227 torture_tcase_add_simple_test(tcase, "dm", test_printer_dm);
9228 torture_tcase_add_simple_test(tcase, "printer_info_winreg", test_printer_info_winreg);
9229 torture_tcase_add_simple_test(tcase, "change_id", test_printer_change_id);
9230 torture_tcase_add_simple_test(tcase, "keys", test_printer_keys);
9231 torture_tcase_add_simple_test(tcase, "printerdata_consistency", test_printer_data_consistency);
9232 torture_tcase_add_simple_test(tcase, "printerdata_keys", test_printer_data_keys);
9233 torture_tcase_add_simple_test(tcase, "printerdata_values", test_printer_data_values);
9234 torture_tcase_add_simple_test(tcase, "printerdata_set", test_printer_data_set);
9235 torture_tcase_add_simple_test(tcase, "printerdata_winreg", test_printer_data_winreg);
9236 torture_tcase_add_simple_test(tcase, "printerdata_dsspooler", test_printer_data_dsspooler);
9237 torture_tcase_add_simple_test(tcase, "driver_info_winreg", test_driver_info_winreg);
9238 torture_tcase_add_simple_test(tcase, "printer_rename", test_printer_rename);
9239 torture_tcase_add_simple_test(tcase, "printer_ic", test_printer_ic);
9240 torture_tcase_add_simple_test(tcase, "bidi", test_printer_bidi);
9241 torture_tcase_add_simple_test(tcase, "publish_toggle",
9242 test_printer_publish_toggle);
9243 torture_tcase_add_simple_test(tcase, "print_job_enum", test_print_job_enum);
9244 torture_tcase_add_simple_test(tcase, "log_jobinfo", test_printer_log_jobinfo);
9247 struct torture_suite *torture_rpc_spoolss_printer(TALLOC_CTX *mem_ctx)
9249 struct torture_suite *suite = torture_suite_create(mem_ctx, "printer");
9250 struct torture_tcase *tcase;
9252 tcase = torture_suite_add_tcase(suite, "addprinter");
9254 torture_tcase_set_fixture(tcase,
9255 torture_rpc_spoolss_printer_setup,
9256 torture_rpc_spoolss_printer_teardown);
9258 torture_tcase_printer(tcase);
9260 tcase = torture_suite_add_tcase(suite, "addprinterex");
9262 torture_tcase_set_fixture(tcase,
9263 torture_rpc_spoolss_printerex_setup,
9264 torture_rpc_spoolss_printer_teardown);
9266 torture_tcase_printer(tcase);
9268 tcase = torture_suite_add_tcase(suite, "addprinterwkn");
9270 torture_tcase_set_fixture(tcase,
9271 torture_rpc_spoolss_printerwkn_setup,
9272 torture_rpc_spoolss_printer_teardown);
9274 tcase = torture_suite_add_tcase(suite, "addprinterexwkn");
9276 torture_tcase_set_fixture(tcase,
9277 torture_rpc_spoolss_printerexwkn_setup,
9278 torture_rpc_spoolss_printer_teardown);
9281 /* test is not correct */
9282 tcase = torture_suite_add_tcase(suite, "addprinterdm");
9284 torture_tcase_set_fixture(tcase,
9285 torture_rpc_spoolss_printerdm_setup,
9286 torture_rpc_spoolss_printer_teardown);
9288 torture_tcase_printer(tcase);
9293 struct torture_suite *torture_rpc_spoolss(TALLOC_CTX *mem_ctx)
9295 struct torture_suite *suite = torture_suite_create(mem_ctx, "spoolss");
9296 struct torture_tcase *tcase = torture_suite_add_tcase(suite, "printserver");
9298 torture_tcase_set_fixture(tcase,
9299 torture_rpc_spoolss_setup,
9300 torture_rpc_spoolss_teardown);
9302 torture_tcase_add_simple_test(tcase, "openprinter_badnamelist", test_OpenPrinter_badname_list);
9303 torture_tcase_add_simple_test(tcase, "printer_data_list", test_GetPrinterData_list);
9304 torture_tcase_add_simple_test(tcase, "enum_forms", test_PrintServer_EnumForms);
9305 torture_tcase_add_simple_test(tcase, "forms", test_PrintServer_Forms);
9306 torture_tcase_add_simple_test(tcase, "forms_winreg", test_PrintServer_Forms_Winreg);
9307 torture_tcase_add_simple_test(tcase, "enum_ports", test_EnumPorts);
9308 torture_tcase_add_simple_test(tcase, "add_port", test_AddPort);
9309 torture_tcase_add_simple_test(tcase, "get_printer_driver_directory", test_GetPrinterDriverDirectory);
9310 torture_tcase_add_simple_test(tcase, "get_print_processor_directory", test_GetPrintProcessorDirectory);
9311 torture_tcase_add_simple_test(tcase, "enum_printer_drivers", test_EnumPrinterDrivers);
9312 torture_tcase_add_simple_test(tcase, "enum_monitors", test_EnumMonitors);
9313 torture_tcase_add_simple_test(tcase, "enum_print_processors", test_EnumPrintProcessors);
9314 torture_tcase_add_simple_test(tcase, "print_processors_winreg", test_print_processors_winreg);
9315 torture_tcase_add_simple_test(tcase, "add_processor", test_add_print_processor);
9316 torture_tcase_add_simple_test(tcase, "enum_printprocdata", test_EnumPrintProcessorDataTypes);
9317 torture_tcase_add_simple_test(tcase, "enum_printers", test_EnumPrinters);
9318 torture_tcase_add_simple_test(tcase, "enum_ports_old", test_EnumPorts_old);
9319 torture_tcase_add_simple_test(tcase, "enum_printers_old", test_EnumPrinters_old);
9320 torture_tcase_add_simple_test(tcase, "enum_printers_servername", test_EnumPrinters_servername);
9321 torture_tcase_add_simple_test(tcase, "enum_printer_drivers_old", test_EnumPrinterDrivers_old);
9322 torture_tcase_add_simple_test(tcase, "architecture_buffer", test_architecture_buffer);
9323 torture_tcase_add_simple_test(tcase, "get_core_printer_drivers", test_get_core_printer_drivers);
9324 torture_tcase_add_simple_test(tcase, "get_printer_driver_package_path", test_get_printer_driver_package_path);
9326 torture_suite_add_suite(suite, torture_rpc_spoolss_printer(suite));
9331 static bool test_GetPrinterDriverDirectory_getdir(struct torture_context *tctx,
9332 struct dcerpc_binding_handle *b,
9334 const char *environment,
9337 struct spoolss_GetPrinterDriverDirectory r;
9340 r.in.server = server;
9341 r.in.environment = environment;
9345 r.out.needed = &needed;
9347 torture_assert_ntstatus_ok(tctx,
9348 dcerpc_spoolss_GetPrinterDriverDirectory_r(b, tctx, &r),
9349 "failed to query driver directory");
9351 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
9352 DATA_BLOB blob = data_blob_talloc_zero(tctx, needed);
9353 r.in.buffer = &blob;
9354 r.in.offered = needed;
9356 torture_assert_ntstatus_ok(tctx,
9357 dcerpc_spoolss_GetPrinterDriverDirectory_r(b, tctx, &r),
9358 "failed to query driver directory");
9361 torture_assert_werr_ok(tctx, r.out.result,
9362 "failed to query driver directory");
9365 *dir_p = r.out.info->info1.directory_name;
9371 static const char *get_driver_from_info(struct spoolss_AddDriverInfoCtr *info_ctr)
9373 if (info_ctr == NULL) {
9377 switch (info_ctr->level) {
9379 return info_ctr->info.info1->driver_name;
9381 return info_ctr->info.info2->driver_name;
9383 return info_ctr->info.info3->driver_name;
9385 return info_ctr->info.info4->driver_name;
9387 return info_ctr->info.info6->driver_name;
9389 return info_ctr->info.info8->driver_name;
9395 static const char *get_environment_from_info(struct spoolss_AddDriverInfoCtr *info_ctr)
9397 if (info_ctr == NULL) {
9401 switch (info_ctr->level) {
9403 return info_ctr->info.info2->architecture;
9405 return info_ctr->info.info3->architecture;
9407 return info_ctr->info.info4->architecture;
9409 return info_ctr->info.info6->architecture;
9411 return info_ctr->info.info8->architecture;
9418 static bool test_AddPrinterDriver_exp(struct torture_context *tctx,
9419 struct dcerpc_binding_handle *b,
9420 const char *servername,
9421 struct spoolss_AddDriverInfoCtr *info_ctr,
9422 WERROR expected_result)
9424 struct spoolss_AddPrinterDriver r;
9425 const char *drivername = get_driver_from_info(info_ctr);
9426 const char *environment = get_environment_from_info(info_ctr);
9428 r.in.servername = servername;
9429 r.in.info_ctr = info_ctr;
9431 torture_comment(tctx, "Testing AddPrinterDriver(%s) level: %d, environment: '%s'\n",
9432 drivername, info_ctr->level, environment);
9434 torture_assert_ntstatus_ok(tctx,
9435 dcerpc_spoolss_AddPrinterDriver_r(b, tctx, &r),
9436 "spoolss_AddPrinterDriver failed");
9437 torture_assert_werr_equal(tctx, r.out.result, expected_result,
9438 "spoolss_AddPrinterDriver failed with unexpected result");
9444 static bool test_AddPrinterDriverEx_exp(struct torture_context *tctx,
9445 struct dcerpc_binding_handle *b,
9446 const char *servername,
9447 struct spoolss_AddDriverInfoCtr *info_ctr,
9449 WERROR expected_result)
9451 struct spoolss_AddPrinterDriverEx r;
9452 const char *drivername = get_driver_from_info(info_ctr);
9453 const char *environment = get_environment_from_info(info_ctr);
9455 r.in.servername = servername;
9456 r.in.info_ctr = info_ctr;
9459 torture_comment(tctx, "Testing AddPrinterDriverEx(%s) level: %d, environment: '%s'\n",
9460 drivername, info_ctr->level, environment);
9462 torture_assert_ntstatus_ok(tctx,
9463 dcerpc_spoolss_AddPrinterDriverEx_r(b, tctx, &r),
9464 "AddPrinterDriverEx failed");
9465 torture_assert_werr_equal(tctx, r.out.result, expected_result,
9466 "AddPrinterDriverEx failed with unexpected result");
9471 #define ASSERT_DRIVER_PATH(tctx, path, driver_dir, cmt) \
9472 if (path && strlen(path)) {\
9473 torture_assert_strn_equal(tctx, path, driver_dir, strlen(driver_dir), cmt); \
9476 static bool test_AddPrinterDriver_args_level_1(struct torture_context *tctx,
9477 struct dcerpc_binding_handle *b,
9478 const char *server_name,
9479 struct spoolss_AddDriverInfo8 *r,
9482 const char *remote_driver_dir)
9484 struct spoolss_AddDriverInfoCtr info_ctr;
9485 struct spoolss_AddDriverInfo1 info1;
9490 info_ctr.info.info1 = &info1;
9493 torture_assert(tctx,
9494 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_UNKNOWN_LEVEL),
9495 "failed to test AddPrinterDriverEx level 1");
9497 torture_assert(tctx,
9498 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_UNKNOWN_LEVEL),
9499 "failed to test AddPrinterDriver level 1");
9502 info1.driver_name = r->driver_name;
9505 torture_assert(tctx,
9506 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_UNKNOWN_LEVEL),
9507 "failed to test AddPrinterDriverEx level 1");
9509 torture_assert(tctx,
9510 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_UNKNOWN_LEVEL),
9511 "failed to test AddPrinterDriver level 1");
9517 static bool test_AddPrinterDriver_args_level_2(struct torture_context *tctx,
9518 struct dcerpc_binding_handle *b,
9519 const char *server_name,
9520 struct spoolss_AddDriverInfo8 *r,
9523 const char *remote_driver_dir)
9525 struct spoolss_AddDriverInfoCtr info_ctr;
9526 struct spoolss_AddDriverInfo2 info2;
9527 union spoolss_DriverInfo info;
9532 info_ctr.info.info2 = &info2;
9535 torture_assert(tctx,
9536 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_INVALID_PARAM),
9537 "failed to test AddPrinterDriverEx level 2");
9539 torture_assert(tctx,
9540 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_INVALID_PARAM),
9541 "failed to test AddPrinterDriver level 2");
9544 info2.driver_name = r->driver_name;
9547 torture_assert(tctx,
9548 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_INVALID_PARAM),
9549 "failed to test AddPrinterDriverEx level 2");
9551 torture_assert(tctx,
9552 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_INVALID_PARAM),
9553 "failed to test AddPrinterDriver level 2");
9556 info2.version = r->version;
9559 torture_assert(tctx,
9560 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_INVALID_PARAM),
9561 "failed to test AddPrinterDriverEx level 2");
9563 torture_assert(tctx,
9564 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_INVALID_PARAM),
9565 "failed to test AddPrinterDriver level 2");
9568 info2.architecture = r->architecture;
9571 torture_assert(tctx,
9572 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_INVALID_PARAM),
9573 "failed to test AddPrinterDriverEx level 2");
9575 torture_assert(tctx,
9576 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_INVALID_PARAM),
9577 "failed to test AddPrinterDriver level 2");
9580 info2.driver_path = r->driver_path;
9583 torture_assert(tctx,
9584 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_INVALID_PARAM),
9585 "failed to test AddPrinterDriverEx level 2");
9587 torture_assert(tctx,
9588 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_INVALID_PARAM),
9589 "failed to test AddPrinterDriver level 2");
9592 info2.data_file = r->data_file;
9595 torture_assert(tctx,
9596 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_INVALID_PARAM),
9597 "failed to test AddPrinterDriverEx level 2");
9599 torture_assert(tctx,
9600 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_INVALID_PARAM),
9601 "failed to test AddPrinterDriver level 2");
9604 info2.config_file = r->config_file;
9607 torture_assert(tctx,
9608 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, 0, WERR_INVALID_PARAM),
9609 "failed to test AddPrinterDriverEx");
9613 torture_assert(tctx,
9614 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_OK),
9615 "failed to test AddPrinterDriverEx level 2");
9617 torture_assert(tctx,
9618 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_OK),
9619 "failed to test AddPrinterDriver level 2");
9622 torture_assert(tctx,
9623 test_EnumPrinterDrivers_findone(tctx, b, server_name, r->architecture, 2, r->driver_name, &info),
9624 "failed to find added printer driver");
9626 if (remote_driver_dir) {
9627 ASSERT_DRIVER_PATH(tctx, info.info2.driver_path, remote_driver_dir, "unexpected path");
9628 ASSERT_DRIVER_PATH(tctx, info.info2.data_file, remote_driver_dir, "unexpected path");
9629 ASSERT_DRIVER_PATH(tctx, info.info2.config_file, remote_driver_dir, "unexpected path");
9635 static bool test_AddPrinterDriver_args_level_3(struct torture_context *tctx,
9636 struct dcerpc_binding_handle *b,
9637 const char *server_name,
9638 struct spoolss_AddDriverInfo8 *r,
9641 const char *remote_driver_dir)
9643 struct spoolss_AddDriverInfoCtr info_ctr;
9644 struct spoolss_AddDriverInfo3 info3;
9645 union spoolss_DriverInfo info;
9647 info3.driver_name = r->driver_name;
9648 info3.version = r->version;
9649 info3.architecture = r->architecture;
9650 info3.driver_path = r->driver_path;
9651 info3.data_file = r->data_file;
9652 info3.config_file = r->config_file;
9653 info3.help_file = r->help_file;
9654 info3.monitor_name = r->monitor_name;
9655 info3.default_datatype = r->default_datatype;
9656 info3._ndr_size_dependent_files = r->_ndr_size_dependent_files;
9657 info3.dependent_files = r->dependent_files;
9660 info_ctr.info.info3 = &info3;
9663 torture_assert(tctx,
9664 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_OK),
9665 "failed to test AddPrinterDriverEx level 3");
9667 torture_assert(tctx,
9668 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_OK),
9669 "failed to test AddPrinterDriver level 3");
9672 torture_assert(tctx,
9673 test_EnumPrinterDrivers_findone(tctx, b, server_name, r->architecture, 3, r->driver_name, &info),
9674 "failed to find added printer driver");
9676 if (remote_driver_dir) {
9678 ASSERT_DRIVER_PATH(tctx, info.info3.driver_path, remote_driver_dir, "unexpected path");
9679 ASSERT_DRIVER_PATH(tctx, info.info3.data_file, remote_driver_dir, "unexpected path");
9680 ASSERT_DRIVER_PATH(tctx, info.info3.config_file, remote_driver_dir, "unexpected path");
9681 ASSERT_DRIVER_PATH(tctx, info.info3.help_file, remote_driver_dir, "unexpected path");
9682 for (i=0; info.info3.dependent_files && info.info3.dependent_files[i] != NULL; i++) {
9683 ASSERT_DRIVER_PATH(tctx, info.info3.dependent_files[i], remote_driver_dir, "unexpected path");
9690 static bool test_AddPrinterDriver_args_level_4(struct torture_context *tctx,
9691 struct dcerpc_binding_handle *b,
9692 const char *server_name,
9693 struct spoolss_AddDriverInfo8 *r,
9696 const char *remote_driver_dir)
9698 struct spoolss_AddDriverInfoCtr info_ctr;
9699 struct spoolss_AddDriverInfo4 info4;
9700 union spoolss_DriverInfo info;
9702 info4.version = r->version;
9703 info4.driver_name = r->driver_name;
9704 info4.architecture = r->architecture;
9705 info4.driver_path = r->driver_path;
9706 info4.data_file = r->data_file;
9707 info4.config_file = r->config_file;
9708 info4.help_file = r->help_file;
9709 info4.monitor_name = r->monitor_name;
9710 info4.default_datatype = r->default_datatype;
9711 info4._ndr_size_dependent_files = r->_ndr_size_dependent_files;
9712 info4.dependent_files = r->dependent_files;
9713 info4._ndr_size_previous_names = r->_ndr_size_previous_names;
9714 info4.previous_names = r->previous_names;
9717 info_ctr.info.info4 = &info4;
9720 torture_assert(tctx,
9721 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_OK),
9722 "failed to test AddPrinterDriverEx level 4");
9724 torture_assert(tctx,
9725 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_OK),
9726 "failed to test AddPrinterDriver level 4");
9729 torture_assert(tctx,
9730 test_EnumPrinterDrivers_findone(tctx, b, server_name, r->architecture, 4, r->driver_name, &info),
9731 "failed to find added printer driver");
9733 if (remote_driver_dir) {
9735 ASSERT_DRIVER_PATH(tctx, info.info4.driver_path, remote_driver_dir, "unexpected path");
9736 ASSERT_DRIVER_PATH(tctx, info.info4.data_file, remote_driver_dir, "unexpected path");
9737 ASSERT_DRIVER_PATH(tctx, info.info4.config_file, remote_driver_dir, "unexpected path");
9738 ASSERT_DRIVER_PATH(tctx, info.info4.help_file, remote_driver_dir, "unexpected path");
9739 for (i=0; info.info4.dependent_files && info.info4.dependent_files[i] != NULL; i++) {
9740 ASSERT_DRIVER_PATH(tctx, info.info4.dependent_files[i], remote_driver_dir, "unexpected path");
9747 static bool test_AddPrinterDriver_args_level_6(struct torture_context *tctx,
9748 struct dcerpc_binding_handle *b,
9749 const char *server_name,
9750 struct spoolss_AddDriverInfo8 *r,
9753 const char *remote_driver_dir)
9755 struct spoolss_AddDriverInfoCtr info_ctr;
9756 struct spoolss_AddDriverInfo6 info6;
9757 union spoolss_DriverInfo info;
9759 info6.version = r->version;
9760 info6.driver_name = r->driver_name;
9761 info6.architecture = r->architecture;
9762 info6.driver_path = r->driver_path;
9763 info6.data_file = r->data_file;
9764 info6.config_file = r->config_file;
9765 info6.help_file = r->help_file;
9766 info6.monitor_name = r->monitor_name;
9767 info6.default_datatype = r->default_datatype;
9768 info6._ndr_size_dependent_files = r->_ndr_size_dependent_files;
9769 info6.dependent_files = r->dependent_files;
9770 info6._ndr_size_previous_names = r->_ndr_size_previous_names;
9771 info6.previous_names = r->previous_names;
9772 info6.driver_date = r->driver_date;
9773 info6.driver_version = r->driver_version;
9774 info6.manufacturer_name = r->manufacturer_name;
9775 info6.manufacturer_url = r->manufacturer_url;
9776 info6.hardware_id = r->hardware_id;
9777 info6.provider = r->provider;
9780 info_ctr.info.info6 = &info6;
9783 torture_assert(tctx,
9784 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_OK),
9785 "failed to test AddPrinterDriverEx level 6");
9787 torture_assert(tctx,
9788 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_UNKNOWN_LEVEL),
9789 "failed to test AddPrinterDriver level 6");
9792 /* spoolss_AddPrinterDriver does not deal with level 6 or 8 - gd */
9798 torture_assert(tctx,
9799 test_EnumPrinterDrivers_findone(tctx, b, server_name, r->architecture, 6, r->driver_name, &info),
9800 "failed to find added printer driver");
9802 if (remote_driver_dir) {
9804 ASSERT_DRIVER_PATH(tctx, info.info6.driver_path, remote_driver_dir, "unexpected path");
9805 ASSERT_DRIVER_PATH(tctx, info.info6.data_file, remote_driver_dir, "unexpected path");
9806 ASSERT_DRIVER_PATH(tctx, info.info6.config_file, remote_driver_dir, "unexpected path");
9807 ASSERT_DRIVER_PATH(tctx, info.info6.help_file, remote_driver_dir, "unexpected path");
9808 for (i=0; info.info6.dependent_files && info.info6.dependent_files[i] != NULL; i++) {
9809 ASSERT_DRIVER_PATH(tctx, info.info6.dependent_files[i], remote_driver_dir, "unexpected path");
9813 torture_assert_nttime_equal(tctx, info.info6.driver_date, info6.driver_date, "driverdate mismatch");
9814 torture_assert_u64_equal(tctx, info.info6.driver_version, info6.driver_version, "driverversion mismatch");
9819 static bool test_AddPrinterDriver_args_level_8(struct torture_context *tctx,
9820 struct dcerpc_binding_handle *b,
9821 const char *server_name,
9822 struct spoolss_AddDriverInfo8 *r,
9825 const char *remote_driver_dir)
9827 struct spoolss_AddDriverInfoCtr info_ctr;
9828 union spoolss_DriverInfo info;
9831 info_ctr.info.info8 = r;
9834 torture_assert(tctx,
9835 test_AddPrinterDriverEx_exp(tctx, b, server_name, &info_ctr, flags, WERR_OK),
9836 "failed to test AddPrinterDriverEx level 8");
9838 torture_assert(tctx,
9839 test_AddPrinterDriver_exp(tctx, b, server_name, &info_ctr, WERR_UNKNOWN_LEVEL),
9840 "failed to test AddPrinterDriver level 8");
9843 /* spoolss_AddPrinterDriver does not deal with level 6 or 8 - gd */
9849 torture_assert(tctx,
9850 test_EnumPrinterDrivers_findone(tctx, b, server_name, r->architecture, 8, r->driver_name, &info),
9851 "failed to find added printer driver");
9853 if (remote_driver_dir) {
9855 ASSERT_DRIVER_PATH(tctx, info.info8.driver_path, remote_driver_dir, "unexpected path");
9856 ASSERT_DRIVER_PATH(tctx, info.info8.data_file, remote_driver_dir, "unexpected path");
9857 ASSERT_DRIVER_PATH(tctx, info.info8.config_file, remote_driver_dir, "unexpected path");
9858 ASSERT_DRIVER_PATH(tctx, info.info8.help_file, remote_driver_dir, "unexpected path");
9859 for (i=0; info.info8.dependent_files && info.info8.dependent_files[i] != NULL; i++) {
9860 ASSERT_DRIVER_PATH(tctx, info.info8.dependent_files[i], remote_driver_dir, "unexpected path");
9864 torture_assert_nttime_equal(tctx, info.info8.driver_date, r->driver_date, "driverdate mismatch");
9865 torture_assert_u64_equal(tctx, info.info8.driver_version, r->driver_version, "driverversion mismatch");
9870 #undef ASSERT_DRIVER_PATH
9872 static bool test_DeletePrinterDriver_exp(struct torture_context *tctx,
9873 struct dcerpc_binding_handle *b,
9876 const char *environment,
9877 WERROR expected_result)
9879 struct spoolss_DeletePrinterDriver r;
9881 r.in.server = server;
9882 r.in.architecture = environment;
9883 r.in.driver = driver;
9885 torture_comment(tctx, "Testing DeletePrinterDriver(%s)\n", driver);
9887 torture_assert_ntstatus_ok(tctx,
9888 dcerpc_spoolss_DeletePrinterDriver_r(b, tctx, &r),
9889 "DeletePrinterDriver failed");
9890 torture_assert_werr_equal(tctx, r.out.result, expected_result,
9891 "DeletePrinterDriver failed with unexpected result");
9896 static bool test_DeletePrinterDriverEx_exp(struct torture_context *tctx,
9897 struct dcerpc_binding_handle *b,
9900 const char *environment,
9901 uint32_t delete_flags,
9903 WERROR expected_result)
9905 struct spoolss_DeletePrinterDriverEx r;
9907 r.in.server = server;
9908 r.in.architecture = environment;
9909 r.in.driver = driver;
9910 r.in.delete_flags = delete_flags;
9911 r.in.version = version;
9913 torture_comment(tctx, "Testing DeletePrinterDriverEx(%s)\n", driver);
9915 torture_assert_ntstatus_ok(tctx,
9916 dcerpc_spoolss_DeletePrinterDriverEx_r(b, tctx, &r),
9917 "DeletePrinterDriverEx failed");
9918 torture_assert_werr_equal(tctx, r.out.result, expected_result,
9919 "DeletePrinterDriverEx failed with unexpected result");
9924 static bool test_DeletePrinterDriver(struct torture_context *tctx,
9925 struct dcerpc_binding_handle *b,
9926 const char *server_name,
9928 const char *environment)
9930 torture_assert(tctx,
9931 test_DeletePrinterDriver_exp(tctx, b, server_name, driver, "FOOBAR", WERR_INVALID_ENVIRONMENT),
9932 "failed to delete driver");
9934 torture_assert(tctx,
9935 test_DeletePrinterDriver_exp(tctx, b, server_name, driver, environment, WERR_OK),
9936 "failed to delete driver");
9938 if (test_EnumPrinterDrivers_findone(tctx, b, server_name, environment, 1, driver, NULL)) {
9939 torture_fail(tctx, "deleted driver still enumerated");
9942 torture_assert(tctx,
9943 test_DeletePrinterDriver_exp(tctx, b, server_name, driver, environment, WERR_UNKNOWN_PRINTER_DRIVER),
9944 "2nd delete failed");
9949 static bool test_DeletePrinterDriverEx(struct torture_context *tctx,
9950 struct dcerpc_binding_handle *b,
9951 const char *server_name,
9953 const char *environment,
9954 uint32_t delete_flags,
9957 torture_assert(tctx,
9958 test_DeletePrinterDriverEx_exp(tctx, b, server_name, driver, "FOOBAR", delete_flags, version, WERR_INVALID_ENVIRONMENT),
9959 "failed to delete driver");
9961 torture_assert(tctx,
9962 test_DeletePrinterDriverEx_exp(tctx, b, server_name, driver, environment, delete_flags, version, WERR_OK),
9963 "failed to delete driver");
9965 if (test_EnumPrinterDrivers_findone(tctx, b, server_name, environment, 1, driver, NULL)) {
9966 torture_fail(tctx, "deleted driver still enumerated");
9969 torture_assert(tctx,
9970 test_DeletePrinterDriverEx_exp(tctx, b, server_name, driver, environment, delete_flags, version, WERR_UNKNOWN_PRINTER_DRIVER),
9971 "2nd delete failed");
9976 static bool test_PrinterDriver_args(struct torture_context *tctx,
9977 struct dcerpc_binding_handle *b,
9978 const char *server_name,
9980 struct spoolss_AddDriverInfo8 *r,
9982 uint32_t delete_flags,
9983 uint32_t delete_version,
9985 const char *remote_driver_dir)
9991 ret = test_AddPrinterDriver_args_level_1(tctx, b, server_name, r, add_flags, ex, remote_driver_dir);
9994 ret = test_AddPrinterDriver_args_level_2(tctx, b, server_name, r, add_flags, ex, remote_driver_dir);
9997 ret = test_AddPrinterDriver_args_level_3(tctx, b, server_name, r, add_flags, ex, remote_driver_dir);
10000 ret = test_AddPrinterDriver_args_level_4(tctx, b, server_name, r, add_flags, ex, remote_driver_dir);
10003 ret = test_AddPrinterDriver_args_level_6(tctx, b, server_name, r, add_flags, ex, remote_driver_dir);
10006 ret = test_AddPrinterDriver_args_level_8(tctx, b, server_name, r, add_flags, ex, remote_driver_dir);
10012 if (ret == false) {
10020 /* spoolss_AddPrinterDriver does not deal with level 6 or 8 - gd */
10022 if (!ex && (level == 6 || level == 8)) {
10027 struct dcerpc_pipe *p2;
10028 struct policy_handle hive_handle;
10029 struct dcerpc_binding_handle *b2;
10031 torture_assert_ntstatus_ok(tctx,
10032 torture_rpc_connection(tctx, &p2, &ndr_table_winreg),
10033 "could not open winreg pipe");
10034 b2 = p2->binding_handle;
10036 torture_assert(tctx, test_winreg_OpenHKLM(tctx, b2, &hive_handle), "");
10038 ret = test_GetDriverInfo_winreg(tctx, b, NULL, NULL, r->driver_name, r->architecture, r->version, b2, &hive_handle, server_name);
10040 test_winreg_CloseKey(tctx, b2, &hive_handle);
10046 return test_DeletePrinterDriverEx(tctx, b, server_name, r->driver_name, r->architecture, delete_flags, r->version);
10048 return test_DeletePrinterDriver(tctx, b, server_name, r->driver_name, r->architecture);
10052 static bool fillup_printserver_info(struct torture_context *tctx,
10053 struct dcerpc_pipe *p,
10054 struct torture_driver_context *d)
10056 struct policy_handle server_handle;
10057 struct dcerpc_binding_handle *b = p->binding_handle;
10058 const char *server_name_slash = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
10060 torture_assert(tctx,
10061 test_OpenPrinter_server(tctx, p, &server_handle),
10062 "failed to open printserver");
10063 torture_assert(tctx,
10064 test_get_environment(tctx, b, &server_handle, &d->remote.environment),
10065 "failed to get environment");
10066 torture_assert(tctx,
10067 test_ClosePrinter(tctx, b, &server_handle),
10068 "failed to close printserver");
10070 torture_assert(tctx,
10071 test_GetPrinterDriverDirectory_getdir(tctx, b, server_name_slash,
10072 d->local.environment ? d->local.environment : d->remote.environment,
10073 &d->remote.driver_directory),
10074 "failed to get driver directory");
10079 static const char *driver_directory_dir(const char *driver_directory)
10083 p = strrchr(driver_directory, '\\');
10091 static const char *driver_directory_share(struct torture_context *tctx,
10092 const char *driver_directory)
10097 if (driver_directory[0] == '\\' && driver_directory[1] == '\\') {
10098 driver_directory += 2;
10101 p = talloc_strdup(tctx, driver_directory);
10103 torture_assert(tctx,
10104 next_token_talloc(tctx, &p, &tok, "\\"),
10105 "cannot explode uri");
10106 torture_assert(tctx,
10107 next_token_talloc(tctx, &p, &tok, "\\"),
10108 "cannot explode uri");
10113 static bool upload_printer_driver_file(struct torture_context *tctx,
10114 struct smbcli_state *cli,
10115 struct torture_driver_context *d,
10116 const char *file_name)
10121 int maxwrite = 64512;
10124 const char *remote_dir = driver_directory_dir(d->remote.driver_directory);
10125 const char *local_name = talloc_asprintf(tctx, "%s/%s", d->local.driver_directory, file_name);
10126 const char *remote_name = talloc_asprintf(tctx, "%s\\%s", remote_dir, file_name);
10128 if (!file_name || strlen(file_name) == 0) {
10132 torture_comment(tctx, "Uploading %s to %s\n", local_name, remote_name);
10134 fnum = smbcli_open(cli->tree, remote_name, O_RDWR|O_CREAT|O_TRUNC, DENY_NONE);
10136 torture_fail(tctx, talloc_asprintf(tctx, "failed to open remote file: %s\n", remote_name));
10139 f = x_fopen(local_name, O_RDONLY, 0);
10141 torture_fail(tctx, talloc_asprintf(tctx, "failed to open local file: %s\n", local_name));
10144 buf = talloc_array(tctx, uint8_t, maxwrite);
10150 while (!x_feof(f)) {
10154 if ((n = x_fread(buf, 1, n, f)) < 1) {
10155 if((n == 0) && x_feof(f))
10156 break; /* Empty local file. */
10158 torture_warning(tctx,
10159 "failed to read file: %s\n", strerror(errno));
10163 ret = smbcli_write(cli->tree, fnum, 0, buf, nread + start, n);
10166 torture_warning(tctx,
10167 "failed to write file: %s\n", smbcli_errstr(cli->tree));
10176 torture_assert_ntstatus_ok(tctx,
10177 smbcli_close(cli->tree, fnum),
10178 "failed to close file");
10183 static bool connect_printer_driver_share(struct torture_context *tctx,
10184 const char *server_name,
10185 const char *share_name,
10186 struct smbcli_state **cli)
10188 struct smbcli_options smb_options;
10189 struct smbcli_session_options smb_session_options;
10191 torture_comment(tctx, "Connecting printer driver share '%s' on '%s'\n",
10192 share_name, server_name);
10194 lpcfg_smbcli_options(tctx->lp_ctx, &smb_options);
10195 lpcfg_smbcli_session_options(tctx->lp_ctx, &smb_session_options);
10197 torture_assert_ntstatus_ok(tctx,
10198 smbcli_full_connection(tctx, cli, server_name,
10199 lpcfg_smb_ports(tctx->lp_ctx),
10201 lpcfg_socket_options(tctx->lp_ctx),
10202 cmdline_credentials,
10203 lpcfg_resolve_context(tctx->lp_ctx),
10206 &smb_session_options,
10207 lpcfg_gensec_settings(tctx, tctx->lp_ctx)),
10208 "failed to open driver share");
10213 static bool upload_printer_driver(struct torture_context *tctx,
10214 const char *server_name,
10215 struct torture_driver_context *d)
10217 struct smbcli_state *cli;
10218 const char *share_name = driver_directory_share(tctx, d->remote.driver_directory);
10221 torture_assert(tctx,
10222 connect_printer_driver_share(tctx, server_name, share_name, &cli),
10223 "failed to connect to driver share");
10225 torture_comment(tctx, "Uploading printer driver files to \\\\%s\\%s\n",
10226 server_name, share_name);
10228 torture_assert(tctx,
10229 upload_printer_driver_file(tctx, cli, d, d->info8.driver_path),
10230 "failed to upload driver_path");
10231 torture_assert(tctx,
10232 upload_printer_driver_file(tctx, cli, d, d->info8.data_file),
10233 "failed to upload data_file");
10234 torture_assert(tctx,
10235 upload_printer_driver_file(tctx, cli, d, d->info8.config_file),
10236 "failed to upload config_file");
10237 torture_assert(tctx,
10238 upload_printer_driver_file(tctx, cli, d, d->info8.help_file),
10239 "failed to upload help_file");
10240 if (d->info8.dependent_files) {
10241 for (i=0; d->info8.dependent_files->string && d->info8.dependent_files->string[i] != NULL; i++) {
10242 torture_assert(tctx,
10243 upload_printer_driver_file(tctx, cli, d, d->info8.dependent_files->string[i]),
10244 "failed to upload dependent_files");
10253 static bool check_printer_driver_file(struct torture_context *tctx,
10254 struct smbcli_state *cli,
10255 struct torture_driver_context *d,
10256 const char *file_name)
10258 const char *remote_arch_dir = driver_directory_dir(d->remote.driver_directory);
10259 const char *remote_name = talloc_asprintf(tctx, "%s\\%d\\%s",
10265 torture_assert(tctx, (file_name && strlen(file_name) != 0), "invalid filename");
10267 torture_comment(tctx, "checking for driver file at %s\n", remote_name);
10269 fnum = smbcli_open(cli->tree, remote_name, O_RDONLY, DENY_NONE);
10274 torture_assert_ntstatus_ok(tctx,
10275 smbcli_close(cli->tree, fnum),
10276 "failed to close driver file");
10281 static bool check_printer_driver_files(struct torture_context *tctx,
10282 const char *server_name,
10283 struct torture_driver_context *d,
10286 struct smbcli_state *cli;
10287 const char *share_name = driver_directory_share(tctx, d->remote.driver_directory);
10290 torture_assert(tctx,
10291 connect_printer_driver_share(tctx, server_name, share_name, &cli),
10292 "failed to connect to driver share");
10294 torture_comment(tctx, "checking %sexistent driver files at \\\\%s\\%s\n",
10295 (expect_exist ? "": "non-"),
10296 server_name, share_name);
10298 if (d->info8.driver_path && d->info8.driver_path[0]) {
10299 torture_assert(tctx,
10300 check_printer_driver_file(tctx, cli, d, d->info8.driver_path) == expect_exist,
10301 "failed driver_path check");
10303 if (d->info8.data_file && d->info8.data_file[0]) {
10304 torture_assert(tctx,
10305 check_printer_driver_file(tctx, cli, d, d->info8.data_file) == expect_exist,
10306 "failed data_file check");
10308 if (d->info8.config_file && d->info8.config_file[0]) {
10309 torture_assert(tctx,
10310 check_printer_driver_file(tctx, cli, d, d->info8.config_file) == expect_exist,
10311 "failed config_file check");
10313 if (d->info8.help_file && d->info8.help_file[0]) {
10314 torture_assert(tctx,
10315 check_printer_driver_file(tctx, cli, d, d->info8.help_file) == expect_exist,
10316 "failed help_file check");
10318 if (d->info8.dependent_files) {
10319 for (i=0; d->info8.dependent_files->string && d->info8.dependent_files->string[i] != NULL; i++) {
10320 torture_assert(tctx,
10321 check_printer_driver_file(tctx, cli, d, d->info8.dependent_files->string[i]) == expect_exist,
10322 "failed dependent_files check");
10331 static bool remove_printer_driver_file(struct torture_context *tctx,
10332 struct smbcli_state *cli,
10333 struct torture_driver_context *d,
10334 const char *file_name)
10336 const char *remote_name;
10337 const char *remote_dir = driver_directory_dir(d->remote.driver_directory);
10339 if (!file_name || strlen(file_name) == 0) {
10343 remote_name = talloc_asprintf(tctx, "%s\\%s", remote_dir, file_name);
10345 torture_comment(tctx, "Removing %s\n", remote_name);
10347 torture_assert_ntstatus_ok(tctx,
10348 smbcli_unlink(cli->tree, remote_name),
10349 "failed to unlink");
10354 static bool remove_printer_driver(struct torture_context *tctx,
10355 const char *server_name,
10356 struct torture_driver_context *d)
10358 struct smbcli_state *cli;
10359 const char *share_name = driver_directory_share(tctx, d->remote.driver_directory);
10362 torture_assert(tctx,
10363 connect_printer_driver_share(tctx, server_name, share_name, &cli),
10364 "failed to connect to driver share");
10366 torture_comment(tctx, "Removing printer driver files from \\\\%s\\%s\n",
10367 server_name, share_name);
10369 torture_assert(tctx,
10370 remove_printer_driver_file(tctx, cli, d, d->info8.driver_path),
10371 "failed to remove driver_path");
10372 torture_assert(tctx,
10373 remove_printer_driver_file(tctx, cli, d, d->info8.data_file),
10374 "failed to remove data_file");
10375 if (!strequal(d->info8.config_file, d->info8.driver_path)) {
10376 torture_assert(tctx,
10377 remove_printer_driver_file(tctx, cli, d, d->info8.config_file),
10378 "failed to remove config_file");
10380 torture_assert(tctx,
10381 remove_printer_driver_file(tctx, cli, d, d->info8.help_file),
10382 "failed to remove help_file");
10383 if (d->info8.dependent_files) {
10384 for (i=0; d->info8.dependent_files->string && d->info8.dependent_files->string[i] != NULL; i++) {
10385 if (strequal(d->info8.dependent_files->string[i], d->info8.driver_path) ||
10386 strequal(d->info8.dependent_files->string[i], d->info8.data_file) ||
10387 strequal(d->info8.dependent_files->string[i], d->info8.config_file) ||
10388 strequal(d->info8.dependent_files->string[i], d->info8.help_file)) {
10391 torture_assert(tctx,
10392 remove_printer_driver_file(tctx, cli, d, d->info8.dependent_files->string[i]),
10393 "failed to remove dependent_files");
10403 static bool test_add_driver_arg(struct torture_context *tctx,
10404 struct dcerpc_pipe *p,
10405 struct torture_driver_context *d)
10408 struct dcerpc_binding_handle *b = p->binding_handle;
10409 const char *server_name_slash = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
10410 uint32_t levels[] = { 1, 2, 3, 4, 6, 8 };
10412 struct spoolss_AddDriverInfo8 info8;
10413 uint32_t add_flags = APD_COPY_NEW_FILES;
10414 uint32_t delete_flags = 0;
10416 ZERO_STRUCT(info8);
10418 torture_comment(tctx, "Testing PrinterDriver%s '%s' for environment '%s'\n",
10419 d->ex ? "Ex" : "", d->info8.driver_name, d->local.environment);
10421 torture_assert(tctx,
10422 fillup_printserver_info(tctx, p, d),
10423 "failed to fillup printserver info");
10425 if (!directory_exist(d->local.driver_directory)) {
10426 torture_skip(tctx, "Skipping Printer Driver test as no local driver is available");
10429 torture_assert(tctx,
10430 upload_printer_driver(tctx, dcerpc_server_name(p), d),
10431 "failed to upload printer driver");
10434 if (d->info8.dependent_files) {
10435 info8.dependent_files = talloc_zero(tctx, struct spoolss_StringArray);
10436 if (d->info8.dependent_files->string) {
10437 for (i=0; d->info8.dependent_files->string[i] != NULL; i++) {
10439 info8.dependent_files->string = talloc_zero_array(info8.dependent_files, const char *, i+1);
10440 for (i=0; d->info8.dependent_files->string[i] != NULL; i++) {
10441 info8.dependent_files->string[i] = talloc_strdup(info8.dependent_files->string, d->info8.dependent_files->string[i]);
10445 info8.architecture = d->local.environment;
10447 for (i=0; i < ARRAY_SIZE(levels); i++) {
10449 if (torture_setting_bool(tctx, "samba3", false)) {
10450 switch (levels[i]) {
10453 torture_comment(tctx, "skipping level %d against samba\n", levels[i]);
10459 if (torture_setting_bool(tctx, "w2k3", false)) {
10460 switch (levels[i]) {
10462 torture_comment(tctx, "skipping level %d against w2k3\n", levels[i]);
10469 torture_comment(tctx,
10470 "Testing PrinterDriver%s '%s' add & delete level %d\n",
10471 d->ex ? "Ex" : "", info8.driver_name, levels[i]);
10473 ret &= test_PrinterDriver_args(tctx, b, server_name_slash, levels[i], &info8, add_flags, delete_flags, d->info8.version, d->ex, d->remote.driver_directory);
10476 info8.driver_path = talloc_asprintf(tctx, "%s\\%s", d->remote.driver_directory, d->info8.driver_path);
10477 info8.data_file = talloc_asprintf(tctx, "%s\\%s", d->remote.driver_directory, d->info8.data_file);
10478 if (d->info8.config_file) {
10479 info8.config_file = talloc_asprintf(tctx, "%s\\%s", d->remote.driver_directory, d->info8.config_file);
10481 if (d->info8.help_file) {
10482 info8.help_file = talloc_asprintf(tctx, "%s\\%s", d->remote.driver_directory, d->info8.help_file);
10484 if (d->info8.dependent_files && d->info8.dependent_files->string) {
10485 for (i=0; d->info8.dependent_files->string[i] != NULL; i++) {
10486 info8.dependent_files->string[i] = talloc_asprintf(tctx, "%s\\%s", d->remote.driver_directory, d->info8.dependent_files->string[i]);
10490 for (i=0; i < ARRAY_SIZE(levels); i++) {
10492 if (torture_setting_bool(tctx, "samba3", false)) {
10493 switch (levels[i]) {
10501 if (torture_setting_bool(tctx, "w2k3", false)) {
10502 switch (levels[i]) {
10504 torture_comment(tctx, "skipping level %d against w2k3\n", levels[i]);
10511 torture_comment(tctx,
10512 "Testing PrinterDriver%s '%s' add & delete level %d (full unc paths)\n",
10513 d->ex ? "Ex" : "", info8.driver_name, levels[i]);
10515 ret &= test_PrinterDriver_args(tctx, b, server_name_slash, levels[i], &info8, add_flags, delete_flags, d->info8.version, d->ex, d->remote.driver_directory);
10518 torture_assert(tctx,
10519 remove_printer_driver(tctx, dcerpc_server_name(p), d),
10520 "failed to remove printer driver");
10522 torture_comment(tctx, "\n");
10527 static bool test_add_driver_ex_64(struct torture_context *tctx,
10528 struct dcerpc_pipe *p)
10530 struct torture_driver_context *d;
10532 d = talloc_zero(tctx, struct torture_driver_context);
10534 d->info8.version = SPOOLSS_DRIVER_VERSION_200X;
10535 d->info8.driver_name = TORTURE_DRIVER_EX;
10536 d->info8.architecture = NULL;
10537 d->info8.driver_path = talloc_strdup(d, "pscript5.dll");
10538 d->info8.data_file = talloc_strdup(d, "cups6.ppd");
10539 d->info8.config_file = talloc_strdup(d, "cupsui6.dll");
10540 d->local.environment = talloc_strdup(d, "Windows x64");
10541 d->local.driver_directory = talloc_strdup(d, "/usr/share/cups/drivers/x64");
10544 return test_add_driver_arg(tctx, p, d);
10547 static bool test_add_driver_ex_32(struct torture_context *tctx,
10548 struct dcerpc_pipe *p)
10550 struct torture_driver_context *d;
10552 d = talloc_zero(tctx, struct torture_driver_context);
10554 d->info8.version = SPOOLSS_DRIVER_VERSION_200X;
10555 d->info8.driver_name = TORTURE_DRIVER_EX;
10556 d->info8.architecture = NULL;
10557 d->info8.driver_path = talloc_strdup(d, "pscript5.dll");
10558 d->info8.data_file = talloc_strdup(d, "cups6.ppd");
10559 d->info8.config_file = talloc_strdup(d, "cupsui6.dll");
10560 d->local.environment = talloc_strdup(d, "Windows NT x86");
10561 d->local.driver_directory = talloc_strdup(d, "/usr/share/cups/drivers/i386");
10564 return test_add_driver_arg(tctx, p, d);
10567 static bool test_add_driver_64(struct torture_context *tctx,
10568 struct dcerpc_pipe *p)
10570 struct torture_driver_context *d;
10572 d = talloc_zero(tctx, struct torture_driver_context);
10574 d->info8.version = SPOOLSS_DRIVER_VERSION_200X;
10575 d->info8.driver_name = TORTURE_DRIVER;
10576 d->info8.architecture = NULL;
10577 d->info8.driver_path = talloc_strdup(d, "pscript5.dll");
10578 d->info8.data_file = talloc_strdup(d, "cups6.ppd");
10579 d->info8.config_file = talloc_strdup(d, "cupsui6.dll");
10580 d->local.environment = talloc_strdup(d, "Windows x64");
10581 d->local.driver_directory = talloc_strdup(d, "/usr/share/cups/drivers/x64");
10584 return test_add_driver_arg(tctx, p, d);
10587 static bool test_add_driver_32(struct torture_context *tctx,
10588 struct dcerpc_pipe *p)
10590 struct torture_driver_context *d;
10592 d = talloc_zero(tctx, struct torture_driver_context);
10594 d->info8.version = SPOOLSS_DRIVER_VERSION_200X;
10595 d->info8.driver_name = TORTURE_DRIVER;
10596 d->info8.architecture = NULL;
10597 d->info8.driver_path = talloc_strdup(d, "pscript5.dll");
10598 d->info8.data_file = talloc_strdup(d, "cups6.ppd");
10599 d->info8.config_file = talloc_strdup(d, "cupsui6.dll");
10600 d->local.environment = talloc_strdup(d, "Windows NT x86");
10601 d->local.driver_directory = talloc_strdup(d, "/usr/share/cups/drivers/i386");
10604 return test_add_driver_arg(tctx, p, d);
10607 static bool test_add_driver_adobe(struct torture_context *tctx,
10608 struct dcerpc_pipe *p)
10610 struct torture_driver_context *d;
10612 if (!torture_setting_bool(tctx, "samba3", false)) {
10613 torture_skip(tctx, "skipping adobe test which only works against samba3");
10616 d = talloc_zero(tctx, struct torture_driver_context);
10618 d->info8.version = SPOOLSS_DRIVER_VERSION_9X;
10619 d->info8.driver_name = TORTURE_DRIVER_ADOBE;
10620 d->info8.architecture = NULL;
10621 d->info8.driver_path = talloc_strdup(d, "ADOBEPS4.DRV");
10622 d->info8.data_file = talloc_strdup(d, "DEFPRTR2.PPD");
10623 d->info8.config_file = talloc_strdup(d, "ADOBEPS4.DRV");
10625 d->info8.help_file = talloc_strdup(d, "ADOBEPS4.HLP");
10626 d->info8.monitor_name = talloc_strdup(d, "PSMON.DLL");
10628 d->local.environment = talloc_strdup(d, "Windows 4.0");
10629 d->local.driver_directory = talloc_strdup(d, "/usr/share/cups/drivers/adobe/");
10632 return test_add_driver_arg(tctx, p, d);
10635 static bool test_add_driver_adobe_cupsaddsmb(struct torture_context *tctx,
10636 struct dcerpc_pipe *p)
10638 struct torture_driver_context *d;
10639 struct spoolss_StringArray *a;
10641 if (!torture_setting_bool(tctx, "samba3", false)) {
10642 torture_skip(tctx, "skipping cupsaddsmb test which only works against samba3");
10645 d = talloc_zero(tctx, struct torture_driver_context);
10647 d->info8.version = SPOOLSS_DRIVER_VERSION_9X;
10648 d->info8.driver_name = TORTURE_DRIVER_ADOBE_CUPSADDSMB;
10649 d->info8.architecture = NULL;
10650 d->info8.driver_path = talloc_strdup(d, "ADOBEPS4.DRV");
10651 d->info8.data_file = talloc_strdup(d, "DEFPRTR2.PPD");
10652 d->info8.config_file = NULL;
10653 d->info8.help_file = talloc_strdup(d, "ADOBEPS4.HLP");
10654 d->info8.monitor_name = talloc_strdup(d, "PSMON.DLL");
10655 d->info8.default_datatype = talloc_strdup(d, "RAW");
10657 a = talloc_zero(d, struct spoolss_StringArray);
10658 a->string = talloc_zero_array(a, const char *, 7);
10659 a->string[0] = talloc_strdup(a->string, "ADOBEPS4.DRV");
10660 a->string[1] = talloc_strdup(a->string, "DEFPRTR2.PPD");
10661 a->string[2] = talloc_strdup(a->string, "ADOBEPS4.HLP");
10662 a->string[3] = talloc_strdup(a->string, "PSMON.DLL");
10663 a->string[4] = talloc_strdup(a->string, "ADFONTS.MFM");
10664 a->string[5] = talloc_strdup(a->string, "ICONLIB.DLL");
10666 d->info8.dependent_files = a;
10667 d->local.environment = talloc_strdup(d, "Windows 4.0");
10668 d->local.driver_directory = talloc_strdup(d, "/usr/share/cups/drivers/adobe/");
10671 return test_add_driver_arg(tctx, p, d);
10674 static bool test_add_driver_timestamps(struct torture_context *tctx,
10675 struct dcerpc_pipe *p)
10677 struct torture_driver_context *d;
10678 struct timeval t = timeval_current();
10680 d = talloc_zero(tctx, struct torture_driver_context);
10682 d->info8.version = SPOOLSS_DRIVER_VERSION_200X;
10683 d->info8.driver_name = TORTURE_DRIVER_TIMESTAMPS;
10684 d->info8.architecture = NULL;
10685 d->info8.driver_path = talloc_strdup(d, "pscript5.dll");
10686 d->info8.data_file = talloc_strdup(d, "cups6.ppd");
10687 d->info8.config_file = talloc_strdup(d, "cupsui6.dll");
10688 d->info8.driver_date = timeval_to_nttime(&t);
10689 d->local.environment = talloc_strdup(d, "Windows NT x86");
10690 d->local.driver_directory = talloc_strdup(d, "/usr/share/cups/drivers/i386");
10693 torture_assert(tctx,
10694 test_add_driver_arg(tctx, p, d),
10697 unix_to_nt_time(&d->info8.driver_date, 1);
10699 torture_assert(tctx,
10700 test_add_driver_arg(tctx, p, d),
10706 static bool test_multiple_drivers(struct torture_context *tctx,
10707 struct dcerpc_pipe *p)
10709 struct torture_driver_context *d;
10710 struct dcerpc_binding_handle *b = p->binding_handle;
10711 const char *server_name_slash = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
10713 struct spoolss_AddDriverInfo8 info8;
10714 uint32_t add_flags = APD_COPY_NEW_FILES;
10715 uint32_t delete_flags = 0;
10717 d = talloc_zero(tctx, struct torture_driver_context);
10719 d->info8.version = SPOOLSS_DRIVER_VERSION_200X;
10720 d->info8.driver_path = talloc_strdup(d, "pscript5.dll");
10721 d->info8.data_file = talloc_strdup(d, "cups6.ppd");
10722 d->info8.config_file = talloc_strdup(d, "cupsui6.dll");
10723 d->local.environment = talloc_strdup(d, "Windows NT x86");
10724 d->local.driver_directory = talloc_strdup(d, "/usr/share/cups/drivers/i386");
10727 torture_assert(tctx,
10728 fillup_printserver_info(tctx, p, d),
10729 "failed to fillup printserver info");
10731 if (!directory_exist(d->local.driver_directory)) {
10732 torture_skip(tctx, "Skipping Printer Driver test as no local driver is available");
10735 torture_assert(tctx,
10736 upload_printer_driver(tctx, dcerpc_server_name(p), d),
10737 "failed to upload printer driver");
10740 info8.architecture = d->local.environment;
10742 for (i=0; i < 3; i++) {
10743 info8.driver_name = talloc_asprintf(d, "torture_test_driver_%d", i);
10745 torture_assert(tctx,
10746 test_AddPrinterDriver_args_level_3(tctx, b, server_name_slash, &info8, add_flags, true, NULL),
10747 "failed to add driver");
10750 torture_assert(tctx,
10751 test_DeletePrinterDriverEx(tctx, b, server_name_slash, "torture_test_driver_0", info8.architecture, delete_flags, info8.version),
10752 "failed to delete driver");
10754 torture_assert(tctx,
10755 test_EnumPrinterDrivers_findone(tctx, b, server_name_slash, info8.architecture, 3, "torture_test_driver_1", NULL),
10756 "torture_test_driver_1 no longer on the server");
10758 torture_assert(tctx,
10759 test_EnumPrinterDrivers_findone(tctx, b, server_name_slash, info8.architecture, 3, "torture_test_driver_2", NULL),
10760 "torture_test_driver_2 no longer on the server");
10762 torture_assert(tctx,
10763 test_DeletePrinterDriverEx(tctx, b, server_name_slash, "torture_test_driver_1", info8.architecture, delete_flags, info8.version),
10764 "failed to delete driver");
10766 torture_assert(tctx,
10767 test_EnumPrinterDrivers_findone(tctx, b, server_name_slash, info8.architecture, 3, "torture_test_driver_2", NULL),
10768 "torture_test_driver_2 no longer on the server");
10770 torture_assert(tctx,
10771 test_DeletePrinterDriverEx(tctx, b, server_name_slash, "torture_test_driver_2", info8.architecture, delete_flags, info8.version),
10772 "failed to delete driver");
10774 torture_assert(tctx,
10775 remove_printer_driver(tctx, dcerpc_server_name(p), d),
10776 "failed to remove printer driver");
10781 static bool test_del_driver_all_files(struct torture_context *tctx,
10782 struct dcerpc_pipe *p)
10784 struct torture_driver_context *d;
10785 struct spoolss_StringArray *a;
10786 uint32_t add_flags = APD_COPY_NEW_FILES;
10787 uint32_t delete_flags = DPD_DELETE_ALL_FILES;
10788 struct dcerpc_binding_handle *b = p->binding_handle;
10789 const char *server_name_slash = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
10791 d = talloc_zero(tctx, struct torture_driver_context);
10794 d->info8.version = SPOOLSS_DRIVER_VERSION_200X;
10795 d->info8.driver_name = TORTURE_DRIVER_DELETER;
10796 d->info8.architecture = NULL;
10797 d->info8.driver_path = talloc_strdup(d, "pscript5.dll");
10798 d->info8.data_file = talloc_strdup(d, "cups6.ppd");
10799 d->info8.config_file = talloc_strdup(d, "cupsui6.dll");
10800 d->info8.help_file = talloc_strdup(d, "pscript.hlp");
10801 d->local.environment = talloc_strdup(d, SPOOLSS_ARCHITECTURE_x64);
10802 d->local.driver_directory = talloc_strdup(d, "/usr/share/cups/drivers/x64");
10804 a = talloc_zero(d, struct spoolss_StringArray);
10805 a->string = talloc_zero_array(a, const char *, 3);
10806 a->string[0] = talloc_strdup(a->string, "cups6.inf");
10807 a->string[1] = talloc_strdup(a->string, "cups6.ini");
10809 d->info8.dependent_files = a;
10810 d->info8.architecture = d->local.environment;
10812 torture_assert(tctx,
10813 fillup_printserver_info(tctx, p, d),
10814 "failed to fillup printserver info");
10816 if (!directory_exist(d->local.driver_directory)) {
10817 torture_skip(tctx, "Skipping Printer Driver test as no local driver is available");
10820 torture_assert(tctx,
10821 upload_printer_driver(tctx, dcerpc_server_name(p), d),
10822 "failed to upload printer driver");
10824 torture_assert(tctx,
10825 test_AddPrinterDriver_args_level_3(tctx, b, server_name_slash, &d->info8, add_flags, true, NULL),
10826 "failed to add driver");
10828 torture_assert(tctx,
10829 test_DeletePrinterDriverEx(tctx, b, server_name_slash,
10830 d->info8.driver_name,
10831 d->local.environment,
10834 "failed to delete driver");
10836 torture_assert(tctx,
10837 check_printer_driver_files(tctx, dcerpc_server_name(p), d, false),
10838 "printer driver file check failed");
10844 static bool test_del_driver_unused_files(struct torture_context *tctx,
10845 struct dcerpc_pipe *p)
10847 struct torture_driver_context *d1;
10848 struct torture_driver_context *d2;
10849 uint32_t add_flags = APD_COPY_NEW_FILES;
10850 struct dcerpc_binding_handle *b = p->binding_handle;
10851 const char *server_name_slash = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
10853 d1 = talloc_zero(tctx, struct torture_driver_context);
10855 d1->info8.version = SPOOLSS_DRIVER_VERSION_200X;
10856 d1->info8.driver_name = TORTURE_DRIVER_DELETER;
10857 d1->info8.architecture = NULL;
10858 d1->info8.driver_path = talloc_strdup(d1, "pscript5.dll");
10859 d1->info8.data_file = talloc_strdup(d1, "cups6.ppd");
10860 d1->info8.config_file = talloc_strdup(d1, "cupsui6.dll");
10861 d1->info8.help_file = talloc_strdup(d1, "pscript.hlp");
10862 d1->local.environment = talloc_strdup(d1, SPOOLSS_ARCHITECTURE_x64);
10863 d1->local.driver_directory = talloc_strdup(d1, "/usr/share/cups/drivers/x64");
10864 d1->info8.architecture = d1->local.environment;
10866 d2 = talloc_zero(tctx, struct torture_driver_context);
10868 d2->info8.version = SPOOLSS_DRIVER_VERSION_200X;
10869 d2->info8.driver_name = TORTURE_DRIVER_DELETERIN;
10870 d2->info8.architecture = NULL;
10871 d2->info8.driver_path = talloc_strdup(d2, "pscript5.dll"); /* overlapping */
10872 d2->info8.data_file = talloc_strdup(d2, "cupsps6.dll");
10873 d2->info8.config_file = talloc_strdup(d2, "cups6.ini");
10874 d2->info8.help_file = talloc_strdup(d2, "pscript.hlp"); /* overlapping */
10875 d2->local.environment = talloc_strdup(d2, SPOOLSS_ARCHITECTURE_x64);
10876 d2->local.driver_directory = talloc_strdup(d2, "/usr/share/cups/drivers/x64");
10877 d2->info8.architecture = d2->local.environment;
10879 torture_assert(tctx,
10880 fillup_printserver_info(tctx, p, d1),
10881 "failed to fillup printserver info");
10882 torture_assert(tctx,
10883 fillup_printserver_info(tctx, p, d2),
10884 "failed to fillup printserver info");
10886 if (!directory_exist(d1->local.driver_directory)) {
10887 torture_skip(tctx, "Skipping Printer Driver test as no local driver is available");
10890 torture_assert(tctx,
10891 upload_printer_driver(tctx, dcerpc_server_name(p), d1),
10892 "failed to upload printer driver");
10893 torture_assert(tctx,
10894 test_AddPrinterDriver_args_level_3(tctx, b, server_name_slash, &d1->info8, add_flags, true, NULL),
10895 "failed to add driver");
10897 torture_assert(tctx,
10898 upload_printer_driver(tctx, dcerpc_server_name(p), d2),
10899 "failed to upload printer driver");
10900 torture_assert(tctx,
10901 test_AddPrinterDriver_args_level_3(tctx, b, server_name_slash, &d2->info8, add_flags, true, NULL),
10902 "failed to add driver");
10904 /* some files are in use by a separate driver, should fail */
10905 torture_assert(tctx,
10906 test_DeletePrinterDriverEx_exp(tctx, b, server_name_slash,
10907 d1->info8.driver_name,
10908 d1->local.environment,
10909 DPD_DELETE_ALL_FILES,
10911 WERR_PRINTER_DRIVER_IN_USE),
10912 "invalid delete driver response");
10914 /* should only delete files not in use by other driver */
10915 torture_assert(tctx,
10916 test_DeletePrinterDriverEx_exp(tctx, b, server_name_slash,
10917 d1->info8.driver_name,
10918 d1->local.environment,
10919 DPD_DELETE_UNUSED_FILES,
10922 "failed to delete driver (unused files)");
10924 /* check non-overlapping were deleted */
10925 d1->info8.driver_path = NULL;
10926 d1->info8.help_file = NULL;
10927 torture_assert(tctx,
10928 check_printer_driver_files(tctx, dcerpc_server_name(p), d1, false),
10929 "printer driver file check failed");
10930 /* d2 files should be uneffected */
10931 torture_assert(tctx,
10932 check_printer_driver_files(tctx, dcerpc_server_name(p), d2, true),
10933 "printer driver file check failed");
10935 torture_assert(tctx,
10936 test_DeletePrinterDriverEx_exp(tctx, b, server_name_slash,
10937 d2->info8.driver_name,
10938 d2->local.environment,
10939 DPD_DELETE_ALL_FILES,
10942 "failed to delete driver");
10944 torture_assert(tctx,
10945 check_printer_driver_files(tctx, dcerpc_server_name(p), d2, false),
10946 "printer driver file check failed");
10953 struct torture_suite *torture_rpc_spoolss_driver(TALLOC_CTX *mem_ctx)
10955 struct torture_suite *suite = torture_suite_create(mem_ctx, "spoolss.driver");
10957 struct torture_rpc_tcase *tcase = torture_suite_add_rpc_iface_tcase(suite,
10958 "driver", &ndr_table_spoolss);
10959 torture_rpc_tcase_add_test(tcase, "add_driver_64", test_add_driver_64);
10960 torture_rpc_tcase_add_test(tcase, "add_driver_ex_64", test_add_driver_ex_64);
10962 torture_rpc_tcase_add_test(tcase, "add_driver_32", test_add_driver_32);
10963 torture_rpc_tcase_add_test(tcase, "add_driver_ex_32", test_add_driver_ex_32);
10965 torture_rpc_tcase_add_test(tcase, "add_driver_adobe", test_add_driver_adobe);
10967 torture_rpc_tcase_add_test(tcase, "add_driver_adobe_cupsaddsmb", test_add_driver_adobe_cupsaddsmb);
10969 torture_rpc_tcase_add_test(tcase, "add_driver_timestamps", test_add_driver_timestamps);
10971 torture_rpc_tcase_add_test(tcase, "multiple_drivers", test_multiple_drivers);
10973 torture_rpc_tcase_add_test(tcase, "del_driver_all_files", test_del_driver_all_files);
10975 torture_rpc_tcase_add_test(tcase, "del_driver_unused_files", test_del_driver_unused_files);