s4-smbtorture: add tests for spoolss_EnumPrinterKey to RPC-SPOOLSS-PRINTER.

Guenther

diff --git a/source4/torture/rpc/spoolss.c b/source4/torture/rpc/spoolss.c
index bfe667240cde51935f2d1985109bc9069109c597..895f6b94fce3e0ba5a4e75e64362b5c16aa8d627 100644 (file)
--- a/source4/torture/rpc/spoolss.c
+++ b/source4/torture/rpc/spoolss.c
@@ -23,7 +23,9 @@
 #include "includes.h"
 #include "torture/torture.h"
 #include "torture/rpc/rpc.h"
+#include "librpc/gen_ndr/ndr_misc.h"
 #include "librpc/gen_ndr/ndr_spoolss_c.h"
+#include "param/param.h"
 
 struct test_spoolss_context {
        /* print server handle */
@@ -2000,6 +2002,77 @@ static bool test_EnumPrinterDrivers_old(struct torture_context *tctx,
        return true;
 }
 
+bool test_printer_keys(struct torture_context *tctx,
+                      struct dcerpc_pipe *p,
+                      struct policy_handle *handle)
+{
+       DATA_BLOB blob;
+       const char **key_array = NULL;
+       int i;
+
+       {
+               struct spoolss_EnumPrinterKey r;
+               uint32_t needed;
+               uint16_t *key_buffer = talloc_zero_array(tctx, uint16_t, 0);
+
+               r.in.handle = handle;
+               r.in.key_name = "";
+               r.in.offered = 0;
+               r.out.key_buffer = key_buffer;
+               r.out.needed = &needed;
+
+               torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_EnumPrinterKey(p, tctx, &r),
+                       "failed to call EnumPrinterKey");
+               if (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA)) {
+                       r.in.offered = needed;
+                       key_buffer = talloc_zero_array(tctx, uint16_t, needed/2);
+                       r.out.key_buffer = key_buffer;
+                       torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_EnumPrinterKey(p, tctx, &r),
+                               "failed to call EnumPrinterKey");
+               }
+               torture_assert_werr_ok(tctx, r.out.result,
+                       "failed to call EnumPrinterKey");
+
+               blob = data_blob_const(key_buffer, needed);
+       }
+
+       {
+               union winreg_Data data;
+               enum ndr_err_code ndr_err;
+               ndr_err = ndr_pull_union_blob(&blob, tctx, lp_iconv_convenience(tctx->lp_ctx),
+                                       &data, REG_MULTI_SZ,
+                                       (ndr_pull_flags_fn_t)ndr_pull_winreg_Data);
+               torture_assert_ndr_success(tctx, ndr_err, "failed to pull REG_MULTI_SZ");
+               key_array = data.string_array;
+       }
+
+       for (i=0; key_array[i]; i++) {
+               struct spoolss_EnumPrinterDataEx r;
+               uint32_t count;
+               struct spoolss_PrinterEnumValues *info;
+               uint32_t needed;
+
+               r.in.handle = handle;
+               r.in.key_name = key_array[i];
+               r.in.offered = 0;
+               r.out.count = &count;
+               r.out.info = &info;
+               r.out.needed = &needed;
+
+               torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_EnumPrinterDataEx(p, tctx, &r),
+                       "failed to call EnumPrinterDataEx");
+               if (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA)) {
+                       r.in.offered = needed;
+                       torture_assert_ntstatus_ok(tctx, dcerpc_spoolss_EnumPrinterDataEx(p, tctx, &r),
+                               "failed to call EnumPrinterDataEx");
+               }
+               torture_assert_werr_ok(tctx, r.out.result,
+                       "failed to call EnumPrinterDataEx");
+       }
+
+       return true;
+}
+
 bool torture_rpc_spoolss(struct torture_context *torture)
 {
        NTSTATUS status;