?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
s4-smbtorture: fix two valgrind warnings.
[Samba/gbeck.git] / source4 / torture / rpc / spoolss.c
blobb1889704a73901364128c2b818a23cede2e30030
1 /*
2 Unix SMB/CIFS implementation.
3 test suite for spoolss rpc operations
4
5 Copyright (C) Tim Potter 2003
6 Copyright (C) Stefan Metzmacher 2005
7 Copyright (C) Jelmer Vernooij 2007
8
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 3 of the License, or
12 (at your option) any later version.
13
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
18
19 You should have received a copy of the GNU General Public License
20 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 */
22
23 #include "includes.h"
24 #include "torture/torture.h"
25 #include "torture/rpc/rpc.h"
26 #include "librpc/gen_ndr/ndr_spoolss_c.h"
27
28 struct test_spoolss_context {
29 /* print server handle */
30 struct policy_handle server_handle;
31
32 /* for EnumPorts */
33 uint32_t port_count[3];
34 union spoolss_PortInfo *ports[3];
35
36 /* for EnumPrinterDrivers */
37 uint32_t driver_count[7];
38 union spoolss_DriverInfo *drivers[7];
39
40 /* for EnumMonitors */
41 uint32_t monitor_count[3];
42 union spoolss_MonitorInfo *monitors[3];
43
44 /* for EnumPrintProcessors */
45 uint32_t print_processor_count[2];
46 union spoolss_PrintProcessorInfo *print_processors[2];
47
48 /* for EnumPrinters */
49 uint32_t printer_count[6];
50 union spoolss_PrinterInfo *printers[6];
51 };
52
53 #define COMPARE_STRING(tctx, c,r,e) \
54 torture_assert_str_equal(tctx, c.e, r.e, "invalid value")
55
56 /* not every compiler supports __typeof__() */
57 #if (__GNUC__ >= 3)
58 #define _CHECK_FIELD_SIZE(c,r,e,type) do {\
59 if (sizeof(__typeof__(c.e)) != sizeof(type)) { \
60 torture_fail(tctx, #c "." #e "field is not " #type "\n"); \
61 }\
62 if (sizeof(__typeof__(r.e)) != sizeof(type)) { \
63 torture_fail(tctx, #r "." #e "field is not " #type "\n"); \
64 }\
65 } while(0)
66 #else
67 #define _CHECK_FIELD_SIZE(c,r,e,type) do {} while(0)
68 #endif
69
70 #define COMPARE_UINT32(tctx, c, r, e) do {\
71 _CHECK_FIELD_SIZE(c, r, e, uint32_t); \
72 torture_assert_int_equal(tctx, c.e, r.e, "invalid value"); \
73 } while(0)
74
75 #define COMPARE_STRING_ARRAY(tctx, c,r,e)
76
77 static bool test_OpenPrinter_server(struct torture_context *tctx, struct dcerpc_pipe *p, struct test_spoolss_context *ctx)
78 {
79 NTSTATUS status;
80 struct spoolss_OpenPrinter op;
81
82 op.in.printername = talloc_asprintf(ctx, "\\\\%s", dcerpc_server_name(p));
83 op.in.datatype = NULL;
84 op.in.devmode_ctr.devmode= NULL;
85 op.in.access_mask = 0;
86 op.out.handle = &ctx->server_handle;
87
88 torture_comment(tctx, "Testing OpenPrinter(%s)\n", op.in.printername);
89
90 status = dcerpc_spoolss_OpenPrinter(p, ctx, &op);
91 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_OpenPrinter failed");
92 torture_assert_werr_ok(tctx, op.out.result, "dcerpc_spoolss_OpenPrinter failed");
93
94 return true;
95 }
96
97 static bool test_EnumPorts(struct torture_context *tctx,
98 struct dcerpc_pipe *p,
99 struct test_spoolss_context *ctx)
100 {
101 NTSTATUS status;
102 struct spoolss_EnumPorts r;
103 uint16_t levels[] = { 1, 2 };
104 int i, j;
105
106 for (i=0;i<ARRAY_SIZE(levels);i++) {
107 int level = levels[i];
108 DATA_BLOB blob;
109 uint32_t needed;
110 uint32_t count;
111 union spoolss_PortInfo *info;
112
113 r.in.servername = "";
114 r.in.level = level;
115 r.in.buffer = NULL;
116 r.in.offered = 0;
117 r.out.needed = &needed;
118 r.out.count = &count;
119 r.out.info = &info;
120
121 torture_comment(tctx, "Testing EnumPorts level %u\n", r.in.level);
122
123 status = dcerpc_spoolss_EnumPorts(p, ctx, &r);
124 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPorts failed");
125 if (W_ERROR_IS_OK(r.out.result)) {
126 /* TODO: do some more checks here */
127 continue;
128 }
129 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
130 "EnumPorts unexpected return code");
131
132 blob = data_blob_talloc(ctx, NULL, needed);
133 data_blob_clear(&blob);
134 r.in.buffer = &blob;
135 r.in.offered = needed;
136
137 status = dcerpc_spoolss_EnumPorts(p, ctx, &r);
138 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPorts failed");
139
140 torture_assert_werr_ok(tctx, r.out.result, "EnumPorts failed");
141
142 torture_assert(tctx, info, "EnumPorts returned no info");
143
144 ctx->port_count[level] = count;
145 ctx->ports[level] = info;
146 }
147
148 for (i=1;i<ARRAY_SIZE(levels);i++) {
149 int level = levels[i];
150 int old_level = levels[i-1];
151 torture_assert_int_equal(tctx, ctx->port_count[level], ctx->port_count[old_level],
152 "EnumPorts invalid value");
153 }
154 /* if the array sizes are not the same we would maybe segfault in the following code */
155
156 for (i=0;i<ARRAY_SIZE(levels);i++) {
157 int level = levels[i];
158 for (j=0;j<ctx->port_count[level];j++) {
159 union spoolss_PortInfo *cur = &ctx->ports[level][j];
160 union spoolss_PortInfo *ref = &ctx->ports[2][j];
161 switch (level) {
162 case 1:
163 COMPARE_STRING(tctx, cur->info1, ref->info2, port_name);
164 break;
165 case 2:
166 /* level 2 is our reference, and it makes no sense to compare it to itself */
167 break;
168 }
169 }
170 }
171
172 return true;
173 }
174
175 static bool test_GetPrintProcessorDirectory(struct torture_context *tctx,
176 struct dcerpc_pipe *p,
177 struct test_spoolss_context *ctx)
178 {
179 NTSTATUS status;
180 struct spoolss_GetPrintProcessorDirectory r;
181 struct {
182 uint16_t level;
183 const char *server;
184 } levels[] = {{
185 .level = 1,
186 .server = NULL
187 },{
188 .level = 1,
189 .server = ""
190 },{
191 .level = 78,
192 .server = ""
193 },{
194 .level = 1,
195 .server = talloc_asprintf(ctx, "\\\\%s", dcerpc_server_name(p))
196 },{
197 .level = 1024,
198 .server = talloc_asprintf(ctx, "\\\\%s", dcerpc_server_name(p))
199 }
200 };
201 int i;
202 uint32_t needed;
203
204 for (i=0;i<ARRAY_SIZE(levels);i++) {
205 int level = levels[i].level;
206 DATA_BLOB blob;
207
208 r.in.server = levels[i].server;
209 r.in.environment = SPOOLSS_ARCHITECTURE_NT_X86;
210 r.in.level = level;
211 r.in.buffer = NULL;
212 r.in.offered = 0;
213 r.out.needed = &needed;
214
215 torture_comment(tctx, "Testing GetPrintProcessorDirectory level %u\n", r.in.level);
216
217 status = dcerpc_spoolss_GetPrintProcessorDirectory(p, ctx, &r);
218 torture_assert_ntstatus_ok(tctx, status,
219 "dcerpc_spoolss_GetPrintProcessorDirectory failed");
220 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
221 "GetPrintProcessorDirectory unexpected return code");
222
223 blob = data_blob_talloc(ctx, NULL, needed);
224 data_blob_clear(&blob);
225 r.in.buffer = &blob;
226 r.in.offered = needed;
227
228 status = dcerpc_spoolss_GetPrintProcessorDirectory(p, ctx, &r);
229 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_GetPrintProcessorDirectory failed");
230
231 torture_assert_werr_ok(tctx, r.out.result, "GetPrintProcessorDirectory failed");
232 }
233
234 return true;
235 }
236
237
238 static bool test_GetPrinterDriverDirectory(struct torture_context *tctx,
239 struct dcerpc_pipe *p,
240 struct test_spoolss_context *ctx)
241 {
242 NTSTATUS status;
243 struct spoolss_GetPrinterDriverDirectory r;
244 struct {
245 uint16_t level;
246 const char *server;
247 } levels[] = {{
248 .level = 1,
249 .server = NULL
250 },{
251 .level = 1,
252 .server = ""
253 },{
254 .level = 78,
255 .server = ""
256 },{
257 .level = 1,
258 .server = talloc_asprintf(ctx, "\\\\%s", dcerpc_server_name(p))
259 },{
260 .level = 1024,
261 .server = talloc_asprintf(ctx, "\\\\%s", dcerpc_server_name(p))
262 }
263 };
264 int i;
265 uint32_t needed;
266
267 for (i=0;i<ARRAY_SIZE(levels);i++) {
268 int level = levels[i].level;
269 DATA_BLOB blob;
270
271 r.in.server = levels[i].server;
272 r.in.environment = SPOOLSS_ARCHITECTURE_NT_X86;
273 r.in.level = level;
274 r.in.buffer = NULL;
275 r.in.offered = 0;
276 r.out.needed = &needed;
277
278 torture_comment(tctx, "Testing GetPrinterDriverDirectory level %u\n", r.in.level);
279
280 status = dcerpc_spoolss_GetPrinterDriverDirectory(p, ctx, &r);
281 torture_assert_ntstatus_ok(tctx, status,
282 "dcerpc_spoolss_GetPrinterDriverDirectory failed");
283 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
284 "GetPrinterDriverDirectory unexpected return code");
285
286 blob = data_blob_talloc(ctx, NULL, needed);
287 data_blob_clear(&blob);
288 r.in.buffer = &blob;
289 r.in.offered = needed;
290
291 status = dcerpc_spoolss_GetPrinterDriverDirectory(p, ctx, &r);
292 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_GetPrinterDriverDirectory failed");
293
294 torture_assert_werr_ok(tctx, r.out.result, "GetPrinterDriverDirectory failed");
295 }
296
297 return true;
298 }
299
300 static bool test_EnumPrinterDrivers(struct torture_context *tctx,
301 struct dcerpc_pipe *p,
302 struct test_spoolss_context *ctx)
303 {
304 NTSTATUS status;
305 struct spoolss_EnumPrinterDrivers r;
306 uint16_t levels[] = { 1, 2, 3, 4, 5, 6 };
307 int i, j;
308
309 for (i=0;i<ARRAY_SIZE(levels);i++) {
310 int level = levels[i];
311 DATA_BLOB blob;
312 uint32_t needed;
313 uint32_t count;
314 union spoolss_DriverInfo *info;
315
316 r.in.server = "";
317 r.in.environment = SPOOLSS_ARCHITECTURE_NT_X86;
318 r.in.level = level;
319 r.in.buffer = NULL;
320 r.in.offered = 0;
321 r.out.needed = &needed;
322 r.out.count = &count;
323 r.out.info = &info;
324
325 torture_comment(tctx, "Testing EnumPrinterDrivers level %u\n", r.in.level);
326
327 status = dcerpc_spoolss_EnumPrinterDrivers(p, ctx, &r);
328 torture_assert_ntstatus_ok(tctx, status,
329 "dcerpc_spoolss_EnumPrinterDrivers failed");
330 if (W_ERROR_IS_OK(r.out.result)) {
331 /* TODO: do some more checks here */
332 continue;
333 }
334 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
335 "EnumPrinterDrivers failed");
336
337 blob = data_blob_talloc(ctx, NULL, needed);
338 data_blob_clear(&blob);
339 r.in.buffer = &blob;
340 r.in.offered = needed;
341
342 status = dcerpc_spoolss_EnumPrinterDrivers(p, ctx, &r);
343 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPrinterDrivers failed");
344
345 torture_assert_werr_ok(tctx, r.out.result, "EnumPrinterDrivers failed");
346
347 ctx->driver_count[level] = count;
348 ctx->drivers[level] = info;
349 }
350
351 for (i=1;i<ARRAY_SIZE(levels);i++) {
352 int level = levels[i];
353 int old_level = levels[i-1];
354 torture_assert_int_equal(tctx, ctx->driver_count[level], ctx->driver_count[old_level],
355 "EnumPrinterDrivers invalid value");
356 }
357
358 for (i=0;i<ARRAY_SIZE(levels);i++) {
359 int level = levels[i];
360 for (j=0;j<ctx->driver_count[level];j++) {
361 union spoolss_DriverInfo *cur = &ctx->drivers[level][j];
362 union spoolss_DriverInfo *ref = &ctx->drivers[6][j];
363 switch (level) {
364 case 1:
365 COMPARE_STRING(tctx, cur->info1, ref->info6, driver_name);
366 break;
367 case 2:
368 COMPARE_UINT32(tctx, cur->info2, ref->info6, version);
369 COMPARE_STRING(tctx, cur->info2, ref->info6, driver_name);
370 COMPARE_STRING(tctx, cur->info2, ref->info6, architecture);
371 COMPARE_STRING(tctx, cur->info2, ref->info6, driver_path);
372 COMPARE_STRING(tctx, cur->info2, ref->info6, data_file);
373 COMPARE_STRING(tctx, cur->info2, ref->info6, config_file);
374 break;
375 case 3:
376 COMPARE_UINT32(tctx, cur->info3, ref->info6, version);
377 COMPARE_STRING(tctx, cur->info3, ref->info6, driver_name);
378 COMPARE_STRING(tctx, cur->info3, ref->info6, architecture);
379 COMPARE_STRING(tctx, cur->info3, ref->info6, driver_path);
380 COMPARE_STRING(tctx, cur->info3, ref->info6, data_file);
381 COMPARE_STRING(tctx, cur->info3, ref->info6, config_file);
382 COMPARE_STRING(tctx, cur->info3, ref->info6, help_file);
383 COMPARE_STRING_ARRAY(tctx, cur->info3, ref->info6, dependent_files);
384 COMPARE_STRING(tctx, cur->info3, ref->info6, monitor_name);
385 COMPARE_STRING(tctx, cur->info3, ref->info6, default_datatype);
386 break;
387 case 4:
388 COMPARE_UINT32(tctx, cur->info4, ref->info6, version);
389 COMPARE_STRING(tctx, cur->info4, ref->info6, driver_name);
390 COMPARE_STRING(tctx, cur->info4, ref->info6, architecture);
391 COMPARE_STRING(tctx, cur->info4, ref->info6, driver_path);
392 COMPARE_STRING(tctx, cur->info4, ref->info6, data_file);
393 COMPARE_STRING(tctx, cur->info4, ref->info6, config_file);
394 COMPARE_STRING(tctx, cur->info4, ref->info6, help_file);
395 COMPARE_STRING_ARRAY(tctx, cur->info4, ref->info6, dependent_files);
396 COMPARE_STRING(tctx, cur->info4, ref->info6, monitor_name);
397 COMPARE_STRING(tctx, cur->info4, ref->info6, default_datatype);
398 COMPARE_STRING_ARRAY(tctx, cur->info4, ref->info6, previous_names);
399 break;
400 case 5:
401 COMPARE_UINT32(tctx, cur->info5, ref->info6, version);
402 COMPARE_STRING(tctx, cur->info5, ref->info6, driver_name);
403 COMPARE_STRING(tctx, cur->info5, ref->info6, architecture);
404 COMPARE_STRING(tctx, cur->info5, ref->info6, driver_path);
405 COMPARE_STRING(tctx, cur->info5, ref->info6, data_file);
406 COMPARE_STRING(tctx, cur->info5, ref->info6, config_file);
407 /*COMPARE_UINT32(tctx, cur->info5, ref->info6, driver_attributes);*/
408 /*COMPARE_UINT32(tctx, cur->info5, ref->info6, config_version);*/
409 /*TODO: ! COMPARE_UINT32(tctx, cur->info5, ref->info6, driver_version); */
410 break;
411 case 6:
412 /* level 6 is our reference, and it makes no sense to compare it to itself */
413 break;
414 }
415 }
416 }
417
418 return true;
419 }
420
421 static bool test_EnumMonitors(struct torture_context *tctx,
422 struct dcerpc_pipe *p,
423 struct test_spoolss_context *ctx)
424 {
425 NTSTATUS status;
426 struct spoolss_EnumMonitors r;
427 uint16_t levels[] = { 1, 2 };
428 int i, j;
429
430 for (i=0;i<ARRAY_SIZE(levels);i++) {
431 int level = levels[i];
432 DATA_BLOB blob;
433 uint32_t needed;
434 uint32_t count;
435 union spoolss_MonitorInfo *info;
436
437 r.in.servername = "";
438 r.in.level = level;
439 r.in.buffer = NULL;
440 r.in.offered = 0;
441 r.out.needed = &needed;
442 r.out.count = &count;
443 r.out.info = &info;
444
445 torture_comment(tctx, "Testing EnumMonitors level %u\n", r.in.level);
446
447 status = dcerpc_spoolss_EnumMonitors(p, ctx, &r);
448 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumMonitors failed");
449 if (W_ERROR_IS_OK(r.out.result)) {
450 /* TODO: do some more checks here */
451 continue;
452 }
453 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
454 "EnumMonitors failed");
455
456 blob = data_blob_talloc(ctx, NULL, needed);
457 data_blob_clear(&blob);
458 r.in.buffer = &blob;
459 r.in.offered = needed;
460
461 status = dcerpc_spoolss_EnumMonitors(p, ctx, &r);
462 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumMonitors failed");
463
464 torture_assert_werr_ok(tctx, r.out.result, "EnumMonitors failed");
465
466 ctx->monitor_count[level] = count;
467 ctx->monitors[level] = info;
468 }
469
470 for (i=1;i<ARRAY_SIZE(levels);i++) {
471 int level = levels[i];
472 int old_level = levels[i-1];
473 torture_assert_int_equal(tctx, ctx->monitor_count[level], ctx->monitor_count[old_level],
474 "EnumMonitors invalid value");
475 }
476
477 for (i=0;i<ARRAY_SIZE(levels);i++) {
478 int level = levels[i];
479 for (j=0;j<ctx->monitor_count[level];j++) {
480 union spoolss_MonitorInfo *cur = &ctx->monitors[level][j];
481 union spoolss_MonitorInfo *ref = &ctx->monitors[2][j];
482 switch (level) {
483 case 1:
484 COMPARE_STRING(tctx, cur->info1, ref->info2, monitor_name);
485 break;
486 case 2:
487 /* level 2 is our reference, and it makes no sense to compare it to itself */
488 break;
489 }
490 }
491 }
492
493 return true;
494 }
495
496 static bool test_EnumPrintProcessors(struct torture_context *tctx,
497 struct dcerpc_pipe *p,
498 struct test_spoolss_context *ctx)
499 {
500 NTSTATUS status;
501 struct spoolss_EnumPrintProcessors r;
502 uint16_t levels[] = { 1 };
503 int i, j;
504
505 for (i=0;i<ARRAY_SIZE(levels);i++) {
506 int level = levels[i];
507 DATA_BLOB blob;
508 uint32_t needed;
509 uint32_t count;
510 union spoolss_PrintProcessorInfo *info;
511
512 r.in.servername = "";
513 r.in.environment = "Windows NT x86";
514 r.in.level = level;
515 r.in.buffer = NULL;
516 r.in.offered = 0;
517 r.out.needed = &needed;
518 r.out.count = &count;
519 r.out.info = &info;
520
521 torture_comment(tctx, "Testing EnumPrintProcessors level %u\n", r.in.level);
522
523 status = dcerpc_spoolss_EnumPrintProcessors(p, ctx, &r);
524 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPrintProcessors failed");
525 if (W_ERROR_IS_OK(r.out.result)) {
526 /* TODO: do some more checks here */
527 continue;
528 }
529 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
530 "EnumPrintProcessors unexpected return code");
531
532 blob = data_blob_talloc(ctx, NULL, needed);
533 data_blob_clear(&blob);
534 r.in.buffer = &blob;
535 r.in.offered = needed;
536
537 status = dcerpc_spoolss_EnumPrintProcessors(p, ctx, &r);
538 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPrintProcessors failed");
539
540 torture_assert_werr_ok(tctx, r.out.result, "EnumPrintProcessors failed");
541
542 ctx->print_processor_count[level] = count;
543 ctx->print_processors[level] = info;
544 }
545
546 for (i=1;i<ARRAY_SIZE(levels);i++) {
547 int level = levels[i];
548 int old_level = levels[i-1];
549 torture_assert_int_equal(tctx, ctx->print_processor_count[level], ctx->print_processor_count[old_level],
550 "EnumPrintProcessors failed");
551 }
552
553 for (i=0;i<ARRAY_SIZE(levels);i++) {
554 int level = levels[i];
555 for (j=0;j<ctx->print_processor_count[level];j++) {
556 #if 0
557 union spoolss_PrintProcessorInfo *cur = &ctx->print_processors[level][j];
558 union spoolss_PrintProcessorInfo *ref = &ctx->print_processors[1][j];
559 #endif
560 switch (level) {
561 case 1:
562 /* level 1 is our reference, and it makes no sense to compare it to itself */
563 break;
564 }
565 }
566 }
567
568 return true;
569 }
570
571 static bool test_EnumPrintProcDataTypes(struct torture_context *tctx,
572 struct dcerpc_pipe *p,
573 struct test_spoolss_context *ctx)
574 {
575 NTSTATUS status;
576 struct spoolss_EnumPrintProcDataTypes r;
577 uint16_t levels[] = { 1 };
578 int i;
579
580 for (i=0;i<ARRAY_SIZE(levels);i++) {
581 int level = levels[i];
582 DATA_BLOB blob;
583 uint32_t needed;
584 uint32_t count;
585 union spoolss_PrintProcDataTypesInfo *info;
586
587 r.in.servername = "";
588 r.in.print_processor_name = "winprint";
589 r.in.level = level;
590 r.in.buffer = NULL;
591 r.in.offered = 0;
592 r.out.needed = &needed;
593 r.out.count = &count;
594 r.out.info = &info;
595
596 torture_comment(tctx, "Testing EnumPrintProcDataTypes level %u\n", r.in.level);
597
598 status = dcerpc_spoolss_EnumPrintProcDataTypes(p, ctx, &r);
599 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPrintProcDataType failed");
600 if (W_ERROR_IS_OK(r.out.result)) {
601 /* TODO: do some more checks here */
602 continue;
603 }
604 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
605 "EnumPrintProcDataTypes unexpected return code");
606
607 blob = data_blob_talloc(ctx, NULL, needed);
608 data_blob_clear(&blob);
609 r.in.buffer = &blob;
610 r.in.offered = needed;
611
612 status = dcerpc_spoolss_EnumPrintProcDataTypes(p, ctx, &r);
613 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPrintProcDataTypes failed");
614
615 torture_assert_werr_ok(tctx, r.out.result, "EnumPrintProcDataTypes failed");
616 }
617
618 return true;
619 }
620
621
622 static bool test_EnumPrinters(struct torture_context *tctx,
623 struct dcerpc_pipe *p,
624 struct test_spoolss_context *ctx)
625 {
626 struct spoolss_EnumPrinters r;
627 NTSTATUS status;
628 uint16_t levels[] = { 0, 1, 2, 4, 5 };
629 int i, j;
630
631 for (i=0;i<ARRAY_SIZE(levels);i++) {
632 int level = levels[i];
633 DATA_BLOB blob;
634 uint32_t needed;
635 uint32_t count;
636 union spoolss_PrinterInfo *info;
637
638 r.in.flags = PRINTER_ENUM_LOCAL;
639 r.in.server = "";
640 r.in.level = level;
641 r.in.buffer = NULL;
642 r.in.offered = 0;
643 r.out.needed = &needed;
644 r.out.count = &count;
645 r.out.info = &info;
646
647 torture_comment(tctx, "Testing EnumPrinters level %u\n", r.in.level);
648
649 status = dcerpc_spoolss_EnumPrinters(p, ctx, &r);
650 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPrinters failed");
651 if (W_ERROR_IS_OK(r.out.result)) {
652 /* TODO: do some more checks here */
653 continue;
654 }
655 torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
656 "EnumPrinters unexpected return code");
657
658 blob = data_blob_talloc(ctx, NULL, needed);
659 data_blob_clear(&blob);
660 r.in.buffer = &blob;
661 r.in.offered = needed;
662
663 status = dcerpc_spoolss_EnumPrinters(p, ctx, &r);
664 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EnumPrinters failed");
665
666 torture_assert_werr_ok(tctx, r.out.result, "EnumPrinters failed");
667
668 ctx->printer_count[level] = count;
669 ctx->printers[level] = info;
670 }
671
672 for (i=1;i<ARRAY_SIZE(levels);i++) {
673 int level = levels[i];
674 int old_level = levels[i-1];
675 torture_assert_int_equal(tctx, ctx->printer_count[level], ctx->printer_count[old_level],
676 "EnumPrinters invalid value");
677 }
678
679 for (i=0;i<ARRAY_SIZE(levels);i++) {
680 int level = levels[i];
681 for (j=0;j<ctx->printer_count[level];j++) {
682 union spoolss_PrinterInfo *cur = &ctx->printers[level][j];
683 union spoolss_PrinterInfo *ref = &ctx->printers[2][j];
684 switch (level) {
685 case 0:
686 COMPARE_STRING(tctx, cur->info0, ref->info2, printername);
687 COMPARE_STRING(tctx, cur->info0, ref->info2, servername);
688 COMPARE_UINT32(tctx, cur->info0, ref->info2, cjobs);
689 /*COMPARE_UINT32(tctx, cur->info0, ref->info2, total_jobs);
690 COMPARE_UINT32(tctx, cur->info0, ref->info2, total_bytes);
691 COMPARE_SPOOLSS_TIME(cur->info0, ref->info2, spoolss_Time time);
692 COMPARE_UINT32(tctx, cur->info0, ref->info2, global_counter);
693 COMPARE_UINT32(tctx, cur->info0, ref->info2, total_pages);
694 COMPARE_UINT32(tctx, cur->info0, ref->info2, version);
695 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown10);
696 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown11);
697 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown12);
698 COMPARE_UINT32(tctx, cur->info0, ref->info2, session_counter);
699 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown14);
700 COMPARE_UINT32(tctx, cur->info0, ref->info2, printer_errors);
701 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown16);
702 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown17);
703 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown18);
704 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown19);
705 COMPARE_UINT32(tctx, cur->info0, ref->info2, change_id);
706 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown21);*/
707 COMPARE_UINT32(tctx, cur->info0, ref->info2, status);
708 /*COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown23);
709 COMPARE_UINT32(tctx, cur->info0, ref->info2, c_setprinter);
710 COMPARE_UINT16(cur->info0, ref->info2, unknown25);
711 COMPARE_UINT16(cur->info0, ref->info2, unknown26);
712 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown27);
713 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown28);
714 COMPARE_UINT32(tctx, cur->info0, ref->info2, unknown29);*/
715 break;
716 case 1:
717 /*COMPARE_UINT32(tctx, cur->info1, ref->info2, flags);*/
718 /*COMPARE_STRING(tctx, cur->info1, ref->info2, name);*/
719 /*COMPARE_STRING(tctx, cur->info1, ref->info2, description);*/
720 COMPARE_STRING(tctx, cur->info1, ref->info2, comment);
721 break;
722 case 2:
723 /* level 2 is our reference, and it makes no sense to compare it to itself */
724 break;
725 case 4:
726 COMPARE_STRING(tctx, cur->info4, ref->info2, printername);
727 COMPARE_STRING(tctx, cur->info4, ref->info2, servername);
728 COMPARE_UINT32(tctx, cur->info4, ref->info2, attributes);
729 break;
730 case 5:
731 COMPARE_STRING(tctx, cur->info5, ref->info2, printername);
732 COMPARE_STRING(tctx, cur->info5, ref->info2, portname);
733 COMPARE_UINT32(tctx, cur->info5, ref->info2, attributes);
734 /*COMPARE_UINT32(tctx, cur->info5, ref->info2, device_not_selected_timeout);
735 COMPARE_UINT32(tctx, cur->info5, ref->info2, transmission_retry_timeout);*/
736 break;
737 }
738 }
739 }
740
741 /* TODO:
742 * - verify that the port of a printer was in the list returned by EnumPorts
743 */
744
745 return true;
746 }
747
748 static bool test_GetPrinter(struct torture_context *tctx,
749 struct dcerpc_pipe *p,
750 struct policy_handle *handle)
751 {
752 NTSTATUS status;
753 struct spoolss_GetPrinter r;
754 uint16_t levels[] = {0, 1, 2, 3, 4, 5, 6, 7, 8};
755 int i;
756 uint32_t needed;
757
758 for (i=0;i<ARRAY_SIZE(levels);i++) {
759 r.in.handle = handle;
760 r.in.level = levels[i];
761 r.in.buffer = NULL;
762 r.in.offered = 0;
763 r.out.needed = &needed;
764
765 torture_comment(tctx, "Testing GetPrinter level %u\n", r.in.level);
766
767 status = dcerpc_spoolss_GetPrinter(p, tctx, &r);
768 torture_assert_ntstatus_ok(tctx, status, "GetPrinter failed");
769
770 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
771 DATA_BLOB blob = data_blob_talloc(tctx, NULL, needed);
772 data_blob_clear(&blob);
773 r.in.buffer = &blob;
774 r.in.offered = needed;
775 status = dcerpc_spoolss_GetPrinter(p, tctx, &r);
776 }
777
778 torture_assert_ntstatus_ok(tctx, status, "GetPrinter failed");
779
780 torture_assert_werr_ok(tctx, r.out.result, "GetPrinter failed");
781 }
782
783 return true;
784 }
785
786
787 static bool test_ClosePrinter(struct torture_context *tctx,
788 struct dcerpc_pipe *p,
789 struct policy_handle *handle)
790 {
791 NTSTATUS status;
792 struct spoolss_ClosePrinter r;
793
794 r.in.handle = handle;
795 r.out.handle = handle;
796
797 torture_comment(tctx, "Testing ClosePrinter\n");
798
799 status = dcerpc_spoolss_ClosePrinter(p, tctx, &r);
800 torture_assert_ntstatus_ok(tctx, status, "ClosePrinter failed");
801
802 return true;
803 }
804
805 static bool test_GetForm(struct torture_context *tctx,
806 struct dcerpc_pipe *p,
807 struct policy_handle *handle,
808 const char *form_name,
809 uint32_t level)
810 {
811 NTSTATUS status;
812 struct spoolss_GetForm r;
813 uint32_t needed;
814
815 r.in.handle = handle;
816 r.in.form_name = form_name;
817 r.in.level = level;
818 r.in.buffer = NULL;
819 r.in.offered = 0;
820 r.out.needed = &needed;
821
822 torture_comment(tctx, "Testing GetForm level %d\n", r.in.level);
823
824 status = dcerpc_spoolss_GetForm(p, tctx, &r);
825 torture_assert_ntstatus_ok(tctx, status, "GetForm failed");
826
827 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
828 DATA_BLOB blob = data_blob_talloc(tctx, NULL, needed);
829 data_blob_clear(&blob);
830 r.in.buffer = &blob;
831 r.in.offered = needed;
832 status = dcerpc_spoolss_GetForm(p, tctx, &r);
833 torture_assert_ntstatus_ok(tctx, status, "GetForm failed");
834
835 torture_assert_werr_ok(tctx, r.out.result, "GetForm failed");
836
837 torture_assert(tctx, r.out.info, "No form info returned");
838 }
839
840 torture_assert_werr_ok(tctx, r.out.result, "GetForm failed");
841
842 return true;
843 }
844
845 static bool test_EnumForms(struct torture_context *tctx,
846 struct dcerpc_pipe *p,
847 struct policy_handle *handle, bool print_server)
848 {
849 NTSTATUS status;
850 struct spoolss_EnumForms r;
851 bool ret = true;
852 uint32_t needed;
853 uint32_t count;
854 uint32_t levels[] = { 1, 2 };
855 int i;
856
857 for (i=0; i<ARRAY_SIZE(levels); i++) {
858
859 union spoolss_FormInfo *info;
860
861 r.in.handle = handle;
862 r.in.level = levels[i];
863 r.in.buffer = NULL;
864 r.in.offered = 0;
865 r.out.needed = &needed;
866 r.out.count = &count;
867 r.out.info = &info;
868
869 torture_comment(tctx, "Testing EnumForms level %d\n", levels[i]);
870
871 status = dcerpc_spoolss_EnumForms(p, tctx, &r);
872 torture_assert_ntstatus_ok(tctx, status, "EnumForms failed");
873
874 if ((r.in.level == 2) && (W_ERROR_EQUAL(r.out.result, WERR_UNKNOWN_LEVEL))) {
875 break;
876 }
877
878 if (print_server && W_ERROR_EQUAL(r.out.result, WERR_BADFID))
879 torture_fail(tctx, "EnumForms on the PrintServer isn't supported by test server (NT4)");
880
881 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
882 int j;
883 DATA_BLOB blob = data_blob_talloc(tctx, NULL, needed);
884 data_blob_clear(&blob);
885 r.in.buffer = &blob;
886 r.in.offered = needed;
887
888 status = dcerpc_spoolss_EnumForms(p, tctx, &r);
889
890 torture_assert(tctx, info, "No forms returned");
891
892 for (j = 0; j < count; j++) {
893 if (!print_server)
894 ret &= test_GetForm(tctx, p, handle, info[j].info1.form_name, levels[i]);
895 }
896 }
897
898 torture_assert_ntstatus_ok(tctx, status, "EnumForms failed");
899
900 torture_assert_werr_ok(tctx, r.out.result, "EnumForms failed");
901 }
902
903 return true;
904 }
905
906 static bool test_DeleteForm(struct torture_context *tctx,
907 struct dcerpc_pipe *p,
908 struct policy_handle *handle,
909 const char *form_name)
910 {
911 NTSTATUS status;
912 struct spoolss_DeleteForm r;
913
914 r.in.handle = handle;
915 r.in.form_name = form_name;
916
917 status = dcerpc_spoolss_DeleteForm(p, tctx, &r);
918
919 torture_assert_ntstatus_ok(tctx, status, "DeleteForm failed");
920
921 torture_assert_werr_ok(tctx, r.out.result, "DeleteForm failed");
922
923 return true;
924 }
925
926 static bool test_AddForm(struct torture_context *tctx,
927 struct dcerpc_pipe *p,
928 struct policy_handle *handle, bool print_server)
929 {
930 struct spoolss_AddForm r;
931 struct spoolss_AddFormInfo1 addform;
932 const char *form_name = "testform3";
933 NTSTATUS status;
934 bool ret = true;
935
936 r.in.handle = handle;
937 r.in.level = 1;
938 r.in.info.info1 = &addform;
939 addform.flags = SPOOLSS_FORM_USER;
940 addform.form_name = form_name;
941 addform.size.width = 50;
942 addform.size.height = 25;
943 addform.area.left = 5;
944 addform.area.top = 10;
945 addform.area.right = 45;
946 addform.area.bottom = 15;
947
948 status = dcerpc_spoolss_AddForm(p, tctx, &r);
949
950 torture_assert_ntstatus_ok(tctx, status, "AddForm failed");
951
952 torture_assert_werr_ok(tctx, r.out.result, "AddForm failed");
953
954 if (!print_server) ret &= test_GetForm(tctx, p, handle, form_name, 1);
955
956 {
957 struct spoolss_SetForm sf;
958 struct spoolss_AddFormInfo1 setform;
959
960 sf.in.handle = handle;
961 sf.in.form_name = form_name;
962 sf.in.level = 1;
963 sf.in.info.info1= &setform;
964 setform.flags = addform.flags;
965 setform.form_name = addform.form_name;
966 setform.size = addform.size;
967 setform.area = addform.area;
968
969 setform.size.width = 1234;
970
971 status = dcerpc_spoolss_SetForm(p, tctx, &sf);
972
973 torture_assert_ntstatus_ok(tctx, status, "SetForm failed");
974
975 torture_assert_werr_ok(tctx, r.out.result, "SetForm failed");
976 }
977
978 if (!print_server) ret &= test_GetForm(tctx, p, handle, form_name, 1);
979
980 if (!test_DeleteForm(tctx, p, handle, form_name)) {
981 ret = false;
982 }
983
984 return ret;
985 }
986
987 static bool test_EnumPorts_old(struct torture_context *tctx,
988 struct dcerpc_pipe *p)
989 {
990 NTSTATUS status;
991 struct spoolss_EnumPorts r;
992 uint32_t needed;
993 uint32_t count;
994 union spoolss_PortInfo *info;
995
996 r.in.servername = talloc_asprintf(tctx, "\\\\%s",
997 dcerpc_server_name(p));
998 r.in.level = 2;
999 r.in.buffer = NULL;
1000 r.in.offered = 0;
1001 r.out.needed = &needed;
1002 r.out.count = &count;
1003 r.out.info = &info;
1004
1005 torture_comment(tctx, "Testing EnumPorts\n");
1006
1007 status = dcerpc_spoolss_EnumPorts(p, tctx, &r);
1008
1009 torture_assert_ntstatus_ok(tctx, status, "EnumPorts failed");
1010
1011 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
1012 DATA_BLOB blob = data_blob_talloc(tctx, NULL, needed);
1013 data_blob_clear(&blob);
1014 r.in.buffer = &blob;
1015 r.in.offered = needed;
1016
1017 status = dcerpc_spoolss_EnumPorts(p, tctx, &r);
1018 torture_assert_ntstatus_ok(tctx, status, "EnumPorts failed");
1019
1020 torture_assert(tctx, info, "No ports returned");
1021 }
1022
1023 return true;
1024 }
1025
1026 static bool test_AddPort(struct torture_context *tctx,
1027 struct dcerpc_pipe *p)
1028 {
1029 NTSTATUS status;
1030 struct spoolss_AddPort r;
1031
1032 r.in.server_name = talloc_asprintf(tctx, "\\\\%s",
1033 dcerpc_server_name(p));
1034 r.in.unknown = 0;
1035 r.in.monitor_name = "foo";
1036
1037 torture_comment(tctx, "Testing AddPort\n");
1038
1039 status = dcerpc_spoolss_AddPort(p, tctx, &r);
1040
1041 torture_assert_ntstatus_ok(tctx, status, "AddPort failed");
1042
1043 /* win2k3 returns WERR_NOT_SUPPORTED */
1044
1045 #if 0
1046
1047 if (!W_ERROR_IS_OK(r.out.result)) {
1048 printf("AddPort failed - %s\n", win_errstr(r.out.result));
1049 return false;
1050 }
1051
1052 #endif
1053
1054 return true;
1055 }
1056
1057 static bool test_GetJob(struct torture_context *tctx,
1058 struct dcerpc_pipe *p,
1059 struct policy_handle *handle, uint32_t job_id)
1060 {
1061 NTSTATUS status;
1062 struct spoolss_GetJob r;
1063 uint32_t needed;
1064
1065 r.in.handle = handle;
1066 r.in.job_id = job_id;
1067 r.in.level = 1;
1068 r.in.buffer = NULL;
1069 r.in.offered = 0;
1070 r.out.needed = &needed;
1071
1072 torture_comment(tctx, "Testing GetJob\n");
1073
1074 status = dcerpc_spoolss_GetJob(p, tctx, &r);
1075 torture_assert_ntstatus_ok(tctx, status, "GetJob failed");
1076
1077 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
1078 DATA_BLOB blob = data_blob_talloc(tctx, NULL, needed);
1079 data_blob_clear(&blob);
1080 r.in.buffer = &blob;
1081 r.in.offered = needed;
1082
1083 status = dcerpc_spoolss_GetJob(p, tctx, &r);
1084
1085 torture_assert(tctx, r.out.info, "No job info returned");
1086 }
1087
1088 return true;
1089 }
1090
1091 static bool test_SetJob(struct torture_context *tctx,
1092 struct dcerpc_pipe *p,
1093 struct policy_handle *handle, uint32_t job_id,
1094 enum spoolss_JobControl command)
1095 {
1096 NTSTATUS status;
1097 struct spoolss_SetJob r;
1098
1099 r.in.handle = handle;
1100 r.in.job_id = job_id;
1101 r.in.ctr = NULL;
1102 r.in.command = command;
1103
1104 torture_comment(tctx, "Testing SetJob\n");
1105
1106 status = dcerpc_spoolss_SetJob(p, tctx, &r);
1107 torture_assert_ntstatus_ok(tctx, status, "SetJob failed");
1108 torture_assert_werr_ok(tctx, r.out.result, "SetJob failed");
1109
1110 return true;
1111 }
1112
1113 static bool test_AddJob(struct torture_context *tctx,
1114 struct dcerpc_pipe *p,
1115 struct policy_handle *handle)
1116 {
1117 NTSTATUS status;
1118 struct spoolss_AddJob r;
1119 uint32_t needed;
1120
1121 r.in.level = 0;
1122 r.in.handle = handle;
1123 r.in.offered = 0;
1124 r.out.needed = &needed;
1125 r.in.buffer = r.out.buffer = NULL;
1126
1127 torture_comment(tctx, "Testing AddJob\n");
1128
1129 status = dcerpc_spoolss_AddJob(p, tctx, &r);
1130 torture_assert_werr_equal(tctx, r.out.result, WERR_UNKNOWN_LEVEL, "AddJob failed");
1131
1132 r.in.level = 1;
1133
1134 status = dcerpc_spoolss_AddJob(p, tctx, &r);
1135 torture_assert_werr_equal(tctx, r.out.result, WERR_INVALID_PARAM, "AddJob failed");
1136
1137 return true;
1138 }
1139
1140
1141 static bool test_EnumJobs(struct torture_context *tctx,
1142 struct dcerpc_pipe *p,
1143 struct policy_handle *handle)
1144 {
1145 NTSTATUS status;
1146 struct spoolss_EnumJobs r;
1147 uint32_t needed;
1148 uint32_t count;
1149 union spoolss_JobInfo *info;
1150
1151 r.in.handle = handle;
1152 r.in.firstjob = 0;
1153 r.in.numjobs = 0xffffffff;
1154 r.in.level = 1;
1155 r.in.buffer = NULL;
1156 r.in.offered = 0;
1157 r.out.needed = &needed;
1158 r.out.count = &count;
1159 r.out.info = &info;
1160
1161 torture_comment(tctx, "Testing EnumJobs\n");
1162
1163 status = dcerpc_spoolss_EnumJobs(p, tctx, &r);
1164
1165 torture_assert_ntstatus_ok(tctx, status, "EnumJobs failed");
1166
1167 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
1168 int j;
1169 DATA_BLOB blob = data_blob_talloc(tctx, NULL, needed);
1170 data_blob_clear(&blob);
1171 r.in.buffer = &blob;
1172 r.in.offered = needed;
1173
1174 status = dcerpc_spoolss_EnumJobs(p, tctx, &r);
1175
1176 torture_assert(tctx, info, "No jobs returned");
1177
1178 for (j = 0; j < count; j++) {
1179
1180 test_GetJob(tctx, p, handle, info[j].info1.job_id);
1181 test_SetJob(tctx, p, handle, info[j].info1.job_id, SPOOLSS_JOB_CONTROL_PAUSE);
1182 test_SetJob(tctx, p, handle, info[j].info1.job_id, SPOOLSS_JOB_CONTROL_RESUME);
1183 }
1184
1185 } else {
1186 torture_assert_werr_ok(tctx, r.out.result, "EnumJobs failed");
1187 }
1188
1189 return true;
1190 }
1191
1192 static bool test_DoPrintTest(struct torture_context *tctx,
1193 struct dcerpc_pipe *p,
1194 struct policy_handle *handle)
1195 {
1196 bool ret = true;
1197 NTSTATUS status;
1198 struct spoolss_StartDocPrinter s;
1199 struct spoolss_DocumentInfo1 info1;
1200 struct spoolss_StartPagePrinter sp;
1201 struct spoolss_WritePrinter w;
1202 struct spoolss_EndPagePrinter ep;
1203 struct spoolss_EndDocPrinter e;
1204 int i;
1205 uint32_t job_id;
1206 uint32_t num_written;
1207
1208 torture_comment(tctx, "Testing StartDocPrinter\n");
1209
1210 s.in.handle = handle;
1211 s.in.level = 1;
1212 s.in.info.info1 = &info1;
1213 s.out.job_id = &job_id;
1214 info1.document_name = "TorturePrintJob";
1215 info1.output_file = NULL;
1216 info1.datatype = "RAW";
1217
1218 status = dcerpc_spoolss_StartDocPrinter(p, tctx, &s);
1219 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_StartDocPrinter failed");
1220 torture_assert_werr_ok(tctx, s.out.result, "StartDocPrinter failed");
1221
1222 for (i=1; i < 4; i++) {
1223 torture_comment(tctx, "Testing StartPagePrinter: Page[%d]\n", i);
1224
1225 sp.in.handle = handle;
1226
1227 status = dcerpc_spoolss_StartPagePrinter(p, tctx, &sp);
1228 torture_assert_ntstatus_ok(tctx, status,
1229 "dcerpc_spoolss_StartPagePrinter failed");
1230 torture_assert_werr_ok(tctx, sp.out.result, "StartPagePrinter failed");
1231
1232 torture_comment(tctx, "Testing WritePrinter: Page[%d]\n", i);
1233
1234 w.in.handle = handle;
1235 w.in.data = data_blob_string_const(talloc_asprintf(tctx,"TortureTestPage: %d\nData\n",i));
1236 w.out.num_written = &num_written;
1237
1238 status = dcerpc_spoolss_WritePrinter(p, tctx, &w);
1239 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_WritePrinter failed");
1240 torture_assert_werr_ok(tctx, w.out.result, "WritePrinter failed");
1241
1242 torture_comment(tctx, "Testing EndPagePrinter: Page[%d]\n", i);
1243
1244 ep.in.handle = handle;
1245
1246 status = dcerpc_spoolss_EndPagePrinter(p, tctx, &ep);
1247 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EndPagePrinter failed");
1248 torture_assert_werr_ok(tctx, ep.out.result, "EndPagePrinter failed");
1249 }
1250
1251 torture_comment(tctx, "Testing EndDocPrinter\n");
1252
1253 e.in.handle = handle;
1254
1255 status = dcerpc_spoolss_EndDocPrinter(p, tctx, &e);
1256 torture_assert_ntstatus_ok(tctx, status, "dcerpc_spoolss_EndDocPrinter failed");
1257 torture_assert_werr_ok(tctx, e.out.result, "EndDocPrinter failed");
1258
1259 ret &= test_AddJob(tctx, p, handle);
1260 ret &= test_EnumJobs(tctx, p, handle);
1261
1262 ret &= test_SetJob(tctx, p, handle, job_id, SPOOLSS_JOB_CONTROL_DELETE);
1263
1264 return ret;
1265 }
1266
1267 static bool test_PausePrinter(struct torture_context *tctx,
1268 struct dcerpc_pipe *p,
1269 struct policy_handle *handle)
1270 {
1271 NTSTATUS status;
1272 struct spoolss_SetPrinter r;
1273 struct spoolss_SetPrinterInfoCtr info_ctr;
1274 struct spoolss_DevmodeContainer devmode_ctr;
1275 struct sec_desc_buf secdesc_ctr;
1276
1277 info_ctr.level = 0;
1278 info_ctr.info.info0 = NULL;
1279
1280 ZERO_STRUCT(devmode_ctr);
1281 ZERO_STRUCT(secdesc_ctr);
1282
1283 r.in.handle = handle;
1284 r.in.info_ctr = &info_ctr;
1285 r.in.devmode_ctr = &devmode_ctr;
1286 r.in.secdesc_ctr = &secdesc_ctr;
1287 r.in.command = SPOOLSS_PRINTER_CONTROL_PAUSE;
1288
1289 torture_comment(tctx, "Testing SetPrinter: SPOOLSS_PRINTER_CONTROL_PAUSE\n");
1290
1291 status = dcerpc_spoolss_SetPrinter(p, tctx, &r);
1292
1293 torture_assert_ntstatus_ok(tctx, status, "SetPrinter failed");
1294
1295 torture_assert_werr_ok(tctx, r.out.result, "SetPrinter failed");
1296
1297 return true;
1298 }
1299
1300 static bool test_ResumePrinter(struct torture_context *tctx,
1301 struct dcerpc_pipe *p,
1302 struct policy_handle *handle)
1303 {
1304 NTSTATUS status;
1305 struct spoolss_SetPrinter r;
1306 struct spoolss_SetPrinterInfoCtr info_ctr;
1307 struct spoolss_DevmodeContainer devmode_ctr;
1308 struct sec_desc_buf secdesc_ctr;
1309
1310 info_ctr.level = 0;
1311 info_ctr.info.info0 = NULL;
1312
1313 ZERO_STRUCT(devmode_ctr);
1314 ZERO_STRUCT(secdesc_ctr);
1315
1316 r.in.handle = handle;
1317 r.in.info_ctr = &info_ctr;
1318 r.in.devmode_ctr = &devmode_ctr;
1319 r.in.secdesc_ctr = &secdesc_ctr;
1320 r.in.command = SPOOLSS_PRINTER_CONTROL_RESUME;
1321
1322 torture_comment(tctx, "Testing SetPrinter: SPOOLSS_PRINTER_CONTROL_RESUME\n");
1323
1324 status = dcerpc_spoolss_SetPrinter(p, tctx, &r);
1325
1326 torture_assert_ntstatus_ok(tctx, status, "SetPrinter failed");
1327
1328 torture_assert_werr_ok(tctx, r.out.result, "SetPrinter failed");
1329
1330 return true;
1331 }
1332
1333 static bool test_GetPrinterData(struct torture_context *tctx,
1334 struct dcerpc_pipe *p,
1335 struct policy_handle *handle,
1336 const char *value_name)
1337 {
1338 NTSTATUS status;
1339 struct spoolss_GetPrinterData r;
1340 uint32_t needed;
1341 enum winreg_Type type;
1342 union spoolss_PrinterData data;
1343
1344 r.in.handle = handle;
1345 r.in.value_name = value_name;
1346 r.in.offered = 0;
1347 r.out.needed = &needed;
1348 r.out.type = &type;
1349 r.out.data = &data;
1350
1351 torture_comment(tctx, "Testing GetPrinterData\n");
1352
1353 status = dcerpc_spoolss_GetPrinterData(p, tctx, &r);
1354 torture_assert_ntstatus_ok(tctx, status, "GetPrinterData failed");
1355
1356 if (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA)) {
1357 r.in.offered = needed;
1358
1359 status = dcerpc_spoolss_GetPrinterData(p, tctx, &r);
1360 torture_assert_ntstatus_ok(tctx, status, "GetPrinterData failed");
1361
1362 torture_assert_werr_ok(tctx, r.out.result, "GetPrinterData failed");
1363 }
1364
1365 return true;
1366 }
1367
1368 static bool test_GetPrinterDataEx(struct torture_context *tctx,
1369 struct dcerpc_pipe *p,
1370 struct policy_handle *handle,
1371 const char *key_name,
1372 const char *value_name)
1373 {
1374 NTSTATUS status;
1375 struct spoolss_GetPrinterDataEx r;
1376 enum winreg_Type type;
1377 uint32_t needed;
1378
1379 r.in.handle = handle;
1380 r.in.key_name = key_name;
1381 r.in.value_name = value_name;
1382 r.in.offered = 0;
1383 r.out.type = &type;
1384 r.out.needed = &needed;
1385 r.out.buffer = NULL;
1386
1387 torture_comment(tctx, "Testing GetPrinterDataEx\n");
1388
1389 status = dcerpc_spoolss_GetPrinterDataEx(p, tctx, &r);
1390 if (!NT_STATUS_IS_OK(status)) {
1391 if (NT_STATUS_EQUAL(status,NT_STATUS_NET_WRITE_FAULT) &&
1392 p->last_fault_code == DCERPC_FAULT_OP_RNG_ERROR) {
1393 torture_skip(tctx, "GetPrinterDataEx not supported by server\n");
1394 }
1395 torture_assert_ntstatus_ok(tctx, status, "GetPrinterDataEx failed");
1396 }
1397
1398 if (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA)) {
1399 r.in.offered = needed;
1400 r.out.buffer = talloc_array(tctx, uint8_t, needed);
1401
1402 status = dcerpc_spoolss_GetPrinterDataEx(p, tctx, &r);
1403 torture_assert_ntstatus_ok(tctx, status, "GetPrinterDataEx failed");
1404
1405 torture_assert_werr_ok(tctx, r.out.result, "GetPrinterDataEx failed");
1406 }
1407
1408 return true;
1409 }
1410
1411 static bool test_EnumPrinterData(struct torture_context *tctx, struct dcerpc_pipe *p,
1412 struct policy_handle *handle)
1413 {
1414 NTSTATUS status;
1415 struct spoolss_EnumPrinterData r;
1416
1417 ZERO_STRUCT(r);
1418 r.in.handle = handle;
1419 r.in.enum_index = 0;
1420
1421 do {
1422 uint32_t value_size = 0;
1423 uint32_t data_size = 0;
1424 enum winreg_Type type = 0;
1425
1426 r.in.value_offered = value_size;
1427 r.out.value_needed = &value_size;
1428 r.in.data_offered = data_size;
1429 r.out.data_needed = &data_size;
1430
1431 r.out.type = &type;
1432 r.out.data = talloc_zero_array(tctx, uint8_t, 0);
1433
1434 torture_comment(tctx, "Testing EnumPrinterData\n");
1435
1436 status = dcerpc_spoolss_EnumPrinterData(p, tctx, &r);
1437
1438 torture_assert_ntstatus_ok(tctx, status, "EnumPrinterData failed");
1439
1440 r.in.value_offered = value_size;
1441 r.out.value_name = talloc_zero_array(tctx, const char, value_size);
1442 r.in.data_offered = data_size;
1443 r.out.data = talloc_zero_array(tctx, uint8_t, data_size);
1444
1445 status = dcerpc_spoolss_EnumPrinterData(p, tctx, &r);
1446
1447 torture_assert_ntstatus_ok(tctx, status, "EnumPrinterData failed");
1448
1449 test_GetPrinterData(tctx, p, handle, r.out.value_name);
1450
1451 test_GetPrinterDataEx(tctx,
1452 p, handle, "PrinterDriverData",
1453 r.out.value_name);
1454
1455 r.in.enum_index++;
1456
1457 } while (W_ERROR_IS_OK(r.out.result));
1458
1459 return true;
1460 }
1461
1462 static bool test_EnumPrinterDataEx(struct torture_context *tctx,
1463 struct dcerpc_pipe *p,
1464 struct policy_handle *handle)
1465 {
1466 NTSTATUS status;
1467 struct spoolss_EnumPrinterDataEx r;
1468 struct spoolss_PrinterEnumValues *info;
1469 uint32_t needed;
1470 uint32_t count;
1471
1472 r.in.handle = handle;
1473 r.in.key_name = "PrinterDriverData";
1474 r.in.offered = 0;
1475 r.out.needed = &needed;
1476 r.out.count = &count;
1477 r.out.info = &info;
1478
1479 torture_comment(tctx, "Testing EnumPrinterDataEx\n");
1480
1481 status = dcerpc_spoolss_EnumPrinterDataEx(p, tctx, &r);
1482 torture_assert_ntstatus_ok(tctx, status, "EnumPrinterDataEx failed");
1483
1484 r.in.offered = needed;
1485
1486 status = dcerpc_spoolss_EnumPrinterDataEx(p, tctx, &r);
1487
1488 torture_assert_ntstatus_ok(tctx, status, "EnumPrinterDataEx failed");
1489
1490 return true;
1491 }
1492
1493
1494 static bool test_DeletePrinterData(struct torture_context *tctx,
1495 struct dcerpc_pipe *p,
1496 struct policy_handle *handle,
1497 const char *value_name)
1498 {
1499 NTSTATUS status;
1500 struct spoolss_DeletePrinterData r;
1501
1502 r.in.handle = handle;
1503 r.in.value_name = value_name;
1504
1505 torture_comment(tctx, "Testing DeletePrinterData\n");
1506
1507 status = dcerpc_spoolss_DeletePrinterData(p, tctx, &r);
1508
1509 torture_assert_ntstatus_ok(tctx, status, "DeletePrinterData failed");
1510
1511 return true;
1512 }
1513
1514 static bool test_SetPrinterData(struct torture_context *tctx,
1515 struct dcerpc_pipe *p,
1516 struct policy_handle *handle)
1517 {
1518 NTSTATUS status;
1519 struct spoolss_SetPrinterData r;
1520 const char *value_name = "spottyfoot";
1521
1522 r.in.handle = handle;
1523 r.in.value_name = value_name;
1524 r.in.type = REG_SZ;
1525 r.in.data.string = "dog";
1526
1527 torture_comment(tctx, "Testing SetPrinterData\n");
1528
1529 status = dcerpc_spoolss_SetPrinterData(p, tctx, &r);
1530
1531 torture_assert_ntstatus_ok(tctx, status, "SetPrinterData failed");
1532
1533 if (!test_GetPrinterData(tctx, p, handle, value_name)) {
1534 return false;
1535 }
1536
1537 if (!test_DeletePrinterData(tctx, p, handle, value_name)) {
1538 return false;
1539 }
1540
1541 return true;
1542 }
1543
1544 static bool test_SecondaryClosePrinter(struct torture_context *tctx,
1545 struct dcerpc_pipe *p,
1546 struct policy_handle *handle)
1547 {
1548 NTSTATUS status;
1549 struct dcerpc_binding *b;
1550 struct dcerpc_pipe *p2;
1551 struct spoolss_ClosePrinter cp;
1552
1553 /* only makes sense on SMB */
1554 if (p->conn->transport.transport != NCACN_NP) {
1555 return true;
1556 }
1557
1558 torture_comment(tctx, "testing close on secondary pipe\n");
1559
1560 status = dcerpc_parse_binding(tctx, p->conn->binding_string, &b);
1561 torture_assert_ntstatus_ok(tctx, status, "Failed to parse dcerpc binding");
1562
1563 status = dcerpc_secondary_connection(p, &p2, b);
1564 torture_assert_ntstatus_ok(tctx, status, "Failed to create secondary connection");
1565
1566 status = dcerpc_bind_auth_none(p2, &ndr_table_spoolss);
1567 torture_assert_ntstatus_ok(tctx, status, "Failed to create bind on secondary connection");
1568
1569 cp.in.handle = handle;
1570 cp.out.handle = handle;
1571
1572 status = dcerpc_spoolss_ClosePrinter(p2, tctx, &cp);
1573 torture_assert_ntstatus_equal(tctx, status, NT_STATUS_NET_WRITE_FAULT,
1574 "ERROR: Allowed close on secondary connection");
1575
1576 torture_assert_int_equal(tctx, p2->last_fault_code, DCERPC_FAULT_CONTEXT_MISMATCH,
1577 "Unexpected fault code");
1578
1579 talloc_free(p2);
1580
1581 return true;
1582 }
1583
1584 static bool test_OpenPrinter_badname(struct torture_context *tctx,
1585 struct dcerpc_pipe *p, const char *name)
1586 {
1587 NTSTATUS status;
1588 struct spoolss_OpenPrinter op;
1589 struct spoolss_OpenPrinterEx opEx;
1590 struct policy_handle handle;
1591 bool ret = true;
1592
1593 op.in.printername = name;
1594 op.in.datatype = NULL;
1595 op.in.devmode_ctr.devmode= NULL;
1596 op.in.access_mask = 0;
1597 op.out.handle = &handle;
1598
1599 torture_comment(tctx, "\nTesting OpenPrinter(%s) with bad name\n", op.in.printername);
1600
1601 status = dcerpc_spoolss_OpenPrinter(p, tctx, &op);
1602 torture_assert_ntstatus_ok(tctx, status, "OpenPrinter failed");
1603 if (!W_ERROR_EQUAL(WERR_INVALID_PRINTER_NAME,op.out.result)) {
1604 torture_comment(tctx, "OpenPrinter(%s) unexpected result[%s] should be WERR_INVALID_PRINTER_NAME\n",
1605 name, win_errstr(op.out.result));
1606 }
1607
1608 if (W_ERROR_IS_OK(op.out.result)) {
1609 ret &=test_ClosePrinter(tctx, p, &handle);
1610 }
1611
1612 opEx.in.printername = name;
1613 opEx.in.datatype = NULL;
1614 opEx.in.devmode_ctr.devmode = NULL;
1615 opEx.in.access_mask = 0;
1616 opEx.in.level = 1;
1617 opEx.in.userlevel.level1 = NULL;
1618 opEx.out.handle = &handle;
1619
1620 torture_comment(tctx, "Testing OpenPrinterEx(%s) with bad name\n", opEx.in.printername);
1621
1622 status = dcerpc_spoolss_OpenPrinterEx(p, tctx, &opEx);
1623 torture_assert_ntstatus_ok(tctx, status, "OpenPrinterEx failed");
1624 if (!W_ERROR_EQUAL(WERR_INVALID_PARAM,opEx.out.result)) {
1625 torture_comment(tctx, "OpenPrinterEx(%s) unexpected result[%s] should be WERR_INVALID_PARAM\n",
1626 name, win_errstr(opEx.out.result));
1627 }
1628
1629 if (W_ERROR_IS_OK(opEx.out.result)) {
1630 ret &=test_ClosePrinter(tctx, p, &handle);
1631 }
1632
1633 return ret;
1634 }
1635
1636 static bool test_OpenPrinter(struct torture_context *tctx,
1637 struct dcerpc_pipe *p,
1638 const char *name)
1639 {
1640 NTSTATUS status;
1641 struct spoolss_OpenPrinter r;
1642 struct policy_handle handle;
1643 bool ret = true;
1644
1645 r.in.printername = talloc_asprintf(tctx, "\\\\%s\\%s", dcerpc_server_name(p), name);
1646 r.in.datatype = NULL;
1647 r.in.devmode_ctr.devmode= NULL;
1648 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1649 r.out.handle = &handle;
1650
1651 torture_comment(tctx, "Testing OpenPrinter(%s)\n", r.in.printername);
1652
1653 status = dcerpc_spoolss_OpenPrinter(p, tctx, &r);
1654
1655 torture_assert_ntstatus_ok(tctx, status, "OpenPrinter failed");
1656
1657 torture_assert_werr_ok(tctx, r.out.result, "OpenPrinter failed");
1658
1659 if (!test_GetPrinter(tctx, p, &handle)) {
1660 ret = false;
1661 }
1662
1663 if (!torture_setting_bool(tctx, "samba3", false)) {
1664 if (!test_SecondaryClosePrinter(tctx, p, &handle)) {
1665 ret = false;
1666 }
1667 }
1668
1669 if (!test_ClosePrinter(tctx, p, &handle)) {
1670 ret = false;
1671 }
1672
1673 return ret;
1674 }
1675
1676 static bool call_OpenPrinterEx(struct torture_context *tctx,
1677 struct dcerpc_pipe *p,
1678 const char *name, struct policy_handle *handle)
1679 {
1680 struct spoolss_OpenPrinterEx r;
1681 struct spoolss_UserLevel1 userlevel1;
1682 NTSTATUS status;
1683
1684 if (name && name[0]) {
1685 r.in.printername = talloc_asprintf(tctx, "\\\\%s\\%s",
1686 dcerpc_server_name(p), name);
1687 } else {
1688 r.in.printername = talloc_asprintf(tctx, "\\\\%s",
1689 dcerpc_server_name(p));
1690 }
1691
1692 r.in.datatype = NULL;
1693 r.in.devmode_ctr.devmode= NULL;
1694 r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1695 r.in.level = 1;
1696 r.in.userlevel.level1 = &userlevel1;
1697 r.out.handle = handle;
1698
1699 userlevel1.size = 1234;
1700 userlevel1.client = "hello";
1701 userlevel1.user = "spottyfoot!";
1702 userlevel1.build = 1;
1703 userlevel1.major = 2;
1704 userlevel1.minor = 3;
1705 userlevel1.processor = 4;
1706
1707 torture_comment(tctx, "Testing OpenPrinterEx(%s)\n", r.in.printername);
1708
1709 status = dcerpc_spoolss_OpenPrinterEx(p, tctx, &r);
1710
1711 torture_assert_ntstatus_ok(tctx, status, "OpenPrinterEx failed");
1712
1713 torture_assert_werr_ok(tctx, r.out.result, "OpenPrinterEx failed");
1714
1715 return true;
1716 }
1717
1718 static bool test_OpenPrinterEx(struct torture_context *tctx,
1719 struct dcerpc_pipe *p,
1720 const char *name)
1721 {
1722 struct policy_handle handle;
1723 bool ret = true;
1724
1725 if (!call_OpenPrinterEx(tctx, p, name, &handle)) {
1726 return false;
1727 }
1728
1729 if (!test_GetPrinter(tctx, p, &handle)) {
1730 ret = false;
1731 }
1732
1733 if (!test_EnumForms(tctx, p, &handle, false)) {
1734 ret = false;
1735 }
1736
1737 if (!test_AddForm(tctx, p, &handle, false)) {
1738 ret = false;
1739 }
1740
1741 if (!test_EnumPrinterData(tctx, p, &handle)) {
1742 ret = false;
1743 }
1744
1745 if (!test_EnumPrinterDataEx(tctx, p, &handle)) {
1746 ret = false;
1747 }
1748
1749 if (!test_PausePrinter(tctx, p, &handle)) {
1750 ret = false;
1751 }
1752
1753 if (!test_DoPrintTest(tctx, p, &handle)) {
1754 ret = false;
1755 }
1756
1757 if (!test_ResumePrinter(tctx, p, &handle)) {
1758 ret = false;
1759 }
1760
1761 if (!test_SetPrinterData(tctx, p, &handle)) {
1762 ret = false;
1763 }
1764
1765 if (!torture_setting_bool(tctx, "samba3", false)) {
1766 if (!test_SecondaryClosePrinter(tctx, p, &handle)) {
1767 ret = false;
1768 }
1769 }
1770
1771 if (!test_ClosePrinter(tctx, p, &handle)) {
1772 ret = false;
1773 }
1774
1775 return ret;
1776 }
1777
1778 static bool test_EnumPrinters_old(struct torture_context *tctx, struct dcerpc_pipe *p)
1779 {
1780 struct spoolss_EnumPrinters r;
1781 NTSTATUS status;
1782 uint16_t levels[] = {1, 2, 4, 5};
1783 int i;
1784 bool ret = true;
1785
1786 for (i=0;i<ARRAY_SIZE(levels);i++) {
1787 union spoolss_PrinterInfo *info;
1788 int j;
1789 uint32_t needed;
1790 uint32_t count;
1791
1792 r.in.flags = PRINTER_ENUM_LOCAL;
1793 r.in.server = "";
1794 r.in.level = levels[i];
1795 r.in.buffer = NULL;
1796 r.in.offered = 0;
1797 r.out.needed = &needed;
1798 r.out.count = &count;
1799 r.out.info = &info;
1800
1801 torture_comment(tctx, "Testing EnumPrinters level %u\n", r.in.level);
1802
1803 status = dcerpc_spoolss_EnumPrinters(p, tctx, &r);
1804 torture_assert_ntstatus_ok(tctx, status, "EnumPrinters failed");
1805
1806 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
1807 DATA_BLOB blob = data_blob_talloc(tctx, NULL, needed);
1808 data_blob_clear(&blob);
1809 r.in.buffer = &blob;
1810 r.in.offered = needed;
1811 status = dcerpc_spoolss_EnumPrinters(p, tctx, &r);
1812 }
1813
1814 torture_assert_ntstatus_ok(tctx, status, "EnumPrinters failed");
1815
1816 torture_assert_werr_ok(tctx, r.out.result, "EnumPrinters failed");
1817
1818 if (!info) {
1819 torture_comment(tctx, "No printers returned\n");
1820 return true;
1821 }
1822
1823 for (j=0;j<count;j++) {
1824 if (r.in.level == 1) {
1825 char *unc = talloc_strdup(tctx, info[j].info1.name);
1826 char *slash, *name;
1827 name = unc;
1828 if (unc[0] == '\\' && unc[1] == '\\') {
1829 unc +=2;
1830 }
1831 slash = strchr(unc, '\\');
1832 if (slash) {
1833 slash++;
1834 name = slash;
1835 }
1836 if (!test_OpenPrinter(tctx, p, name)) {
1837 ret = false;
1838 }
1839 if (!test_OpenPrinterEx(tctx, p, name)) {
1840 ret = false;
1841 }
1842 }
1843 }
1844 }
1845
1846 return ret;
1847 }
1848
1849 #if 0
1850 static bool test_GetPrinterDriver2(struct dcerpc_pipe *p,
1851 struct policy_handle *handle,
1852 const char *driver_name)
1853 {
1854 NTSTATUS status;
1855 struct spoolss_GetPrinterDriver2 r;
1856 uint32_t needed;
1857 uint32_t server_major_version;
1858 uint32_t server_minor_version;
1859
1860 r.in.handle = handle;
1861 r.in.architecture = "W32X86";
1862 r.in.level = 1;
1863 r.in.buffer = NULL;
1864 r.in.offered = 0;
1865 r.in.client_major_version = 0;
1866 r.in.client_minor_version = 0;
1867 r.out.needed = &needed;
1868 r.out.server_major_version = &server_major_version;
1869 r.out.server_minor_version = &server_minor_version;
1870
1871 printf("Testing GetPrinterDriver2\n");
1872
1873 status = dcerpc_spoolss_GetPrinterDriver2(p, tctx, &r);
1874 if (!NT_STATUS_IS_OK(status)) {
1875 printf("GetPrinterDriver2 failed - %s\n", nt_errstr(status));
1876 return false;
1877 }
1878
1879 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
1880 r.in.offered = needed;
1881 status = dcerpc_spoolss_GetPrinterDriver2(p, tctx, &r);
1882 }
1883
1884 if (!NT_STATUS_IS_OK(status)) {
1885 printf("GetPrinterDriver2 failed - %s\n",
1886 nt_errstr(status));
1887 return false;
1888 }
1889
1890 if (!W_ERROR_IS_OK(r.out.result)) {
1891 printf("GetPrinterDriver2 failed - %s\n",
1892 win_errstr(r.out.result));
1893 return false;
1894 }
1895
1896 return true;
1897 }
1898 #endif
1899
1900 static bool test_EnumPrinterDrivers_old(struct torture_context *tctx,
1901 struct dcerpc_pipe *p)
1902 {
1903 struct spoolss_EnumPrinterDrivers r;
1904 NTSTATUS status;
1905 uint16_t levels[] = {1, 2, 3, 4, 5, 6};
1906 int i;
1907
1908 for (i=0;i<ARRAY_SIZE(levels);i++) {
1909
1910 uint32_t needed;
1911 uint32_t count;
1912 union spoolss_DriverInfo *info;
1913
1914 r.in.server = talloc_asprintf(tctx, "\\\\%s", dcerpc_server_name(p));
1915 r.in.environment = "Windows NT x86";
1916 r.in.level = levels[i];
1917 r.in.buffer = NULL;
1918 r.in.offered = 0;
1919 r.out.needed = &needed;
1920 r.out.count = &count;
1921 r.out.info = &info;
1922
1923 torture_comment(tctx, "Testing EnumPrinterDrivers level %u\n", r.in.level);
1924
1925 status = dcerpc_spoolss_EnumPrinterDrivers(p, tctx, &r);
1926
1927 torture_assert_ntstatus_ok(tctx, status, "EnumPrinterDrivers failed");
1928
1929 if (W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
1930 DATA_BLOB blob = data_blob_talloc(tctx, NULL, needed);
1931 data_blob_clear(&blob);
1932 r.in.buffer = &blob;
1933 r.in.offered = needed;
1934 status = dcerpc_spoolss_EnumPrinterDrivers(p, tctx, &r);
1935 }
1936
1937 torture_assert_ntstatus_ok(tctx, status, "EnumPrinterDrivers failed");
1938
1939 torture_assert_werr_ok(tctx, r.out.result, "EnumPrinterDrivers failed");
1940
1941 if (!info) {
1942 torture_comment(tctx, "No printer drivers returned\n");
1943 break;
1944 }
1945 }
1946
1947 return true;
1948 }
1949
1950 bool torture_rpc_spoolss(struct torture_context *torture)
1951 {
1952 NTSTATUS status;
1953 struct dcerpc_pipe *p;
1954 bool ret = true;
1955 struct test_spoolss_context *ctx;
1956
1957 status = torture_rpc_connection(torture, &p, &ndr_table_spoolss);
1958 if (!NT_STATUS_IS_OK(status)) {
1959 return false;
1960 }
1961
1962 ctx = talloc_zero(torture, struct test_spoolss_context);
1963
1964 ret &= test_OpenPrinter_server(torture, p, ctx);
1965
1966 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "W3SvcInstalled");
1967 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "BeepEnabled");
1968 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "EventLog");
1969 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "NetPopup");
1970 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "NetPopupToComputer");
1971 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "MajorVersion");
1972 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "MinorVersion");
1973 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "DefaultSpoolDirectory");
1974 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "Architecture");
1975 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "DsPresent");
1976 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "OSVersion");
1977 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "OSVersionEx");
1978 ret &= test_GetPrinterData(torture, p, &ctx->server_handle, "DNSMachineName");
1979 ret &= test_EnumForms(torture, p, &ctx->server_handle, true);
1980 ret &= test_AddForm(torture, p, &ctx->server_handle, true);
1981 ret &= test_EnumPorts(torture, p, ctx);
1982 ret &= test_GetPrinterDriverDirectory(torture, p, ctx);
1983 ret &= test_GetPrintProcessorDirectory(torture, p, ctx);
1984 ret &= test_EnumPrinterDrivers(torture, p, ctx);
1985 ret &= test_EnumMonitors(torture, p, ctx);
1986 ret &= test_EnumPrintProcessors(torture, p, ctx);
1987 ret &= test_EnumPrintProcDataTypes(torture, p, ctx);
1988 ret &= test_EnumPrinters(torture, p, ctx);
1989 ret &= test_OpenPrinter_badname(torture, p, "__INVALID_PRINTER__");
1990 ret &= test_OpenPrinter_badname(torture, p, "\\\\__INVALID_HOST__");
1991 ret &= test_OpenPrinter_badname(torture, p, "");
1992 ret &= test_OpenPrinter_badname(torture, p, "\\\\\\");
1993 ret &= test_OpenPrinter_badname(torture, p, "\\\\\\__INVALID_PRINTER__");
1994 ret &= test_OpenPrinter_badname(torture, p, talloc_asprintf(torture, "\\\\%s\\", dcerpc_server_name(p)));
1995 ret &= test_OpenPrinter_badname(torture, p,
1996 talloc_asprintf(torture, "\\\\%s\\__INVALID_PRINTER__", dcerpc_server_name(p)));
1997
1998
1999 ret &= test_AddPort(torture, p);
2000 ret &= test_EnumPorts_old(torture, p);
2001 ret &= test_EnumPrinters_old(torture, p);
2002 ret &= test_EnumPrinterDrivers_old(torture, p);
2003
2004 return ret;
2005 }
WIP