source4/torture/rpc/spoolss.c

   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 }