s4-smbtorture: rework smbcli_rap_netprintqenum a bit.
[amitay/samba.git] / source4 / torture / rap / rap.c
index bddfacef3297f1491ae573f92356fea6b52f6002..e553f9687d31b2c5908a59330b9eb6529a007f46 100644 (file)
@@ -23,7 +23,7 @@
 #include "libcli/libcli.h"
 #include "torture/smbtorture.h"
 #include "torture/util.h"
-#include "../librpc/gen_ndr/rap.h"
+#include "../librpc/gen_ndr/ndr_rap.h"
 #include "librpc/ndr/libndr.h"
 #include "param/param.h"
 #include "torture/rap/proto.h"
        } \
 } while (0)
 
+#define RAP_RETURN(call) do { \
+       NTSTATUS _status; \
+       _status = call; \
+       if (!NT_STATUS_IS_OK(_status)) { \
+               return _status; \
+       } \
+} while (0)
+
+
 #define NDR_GOTO(call) do { \
        enum ndr_err_code _ndr_err; \
        _ndr_err = call; \
@@ -270,6 +279,10 @@ static NTSTATUS smbcli_rap_netshareenum(struct smbcli_tree *tree,
                break;
        }
 
+       if (DEBUGLEVEL >= 10) {
+               NDR_PRINT_IN_DEBUG(rap_NetShareEnum, r);
+       }
+
        result = rap_cli_do_call(tree, iconv_convenience, call);
 
        if (!NT_STATUS_IS_OK(result))
@@ -307,6 +320,9 @@ static NTSTATUS smbcli_rap_netshareenum(struct smbcli_tree *tree,
                }
        }
 
+       if (DEBUGLEVEL >= 10) {
+               NDR_PRINT_OUT_DEBUG(rap_NetShareEnum, r);
+       }
        result = NT_STATUS_OK;
 
  done:
@@ -364,6 +380,10 @@ static NTSTATUS smbcli_rap_netserverenum2(struct smbcli_tree *tree,
                break;
        }
 
+       if (DEBUGLEVEL >= 10) {
+               NDR_PRINT_IN_DEBUG(rap_NetServerEnum2, r);
+       }
+
        result = rap_cli_do_call(tree, iconv_convenience, call);
 
        if (!NT_STATUS_IS_OK(result))
@@ -404,6 +424,10 @@ static NTSTATUS smbcli_rap_netserverenum2(struct smbcli_tree *tree,
                }
        }
 
+       if (DEBUGLEVEL >= 10) {
+               NDR_PRINT_OUT_DEBUG(rap_NetServerEnum2, r);
+       }
+
        result = NT_STATUS_OK;
 
  done:
@@ -470,6 +494,10 @@ NTSTATUS smbcli_rap_netservergetinfo(struct smbcli_tree *tree,
                goto done;
        }
 
+       if (DEBUGLEVEL >= 10) {
+               NDR_PRINT_IN_DEBUG(rap_WserverGetInfo, r);
+       }
+
        result = rap_cli_do_call(tree, iconv_convenience, call);
 
        if (!NT_STATUS_IS_OK(result))
@@ -497,6 +525,157 @@ NTSTATUS smbcli_rap_netservergetinfo(struct smbcli_tree *tree,
                                       r->out.convert,
                                       &r->out.info.info1.comment));
        }
+
+       if (DEBUGLEVEL >= 10) {
+               NDR_PRINT_OUT_DEBUG(rap_WserverGetInfo, r);
+       }
+ done:
+       talloc_free(call);
+       return result;
+}
+
+static NTSTATUS rap_pull_rap_PrintQueue0(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue0 *r)
+{
+       NDR_RETURN(ndr_pull_bytes(ndr, r->PrintQName, 13));
+       return NT_STATUS_OK;
+}
+
+static NTSTATUS rap_pull_rap_PrintQueue1(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue1 *r)
+{
+       NDR_RETURN(ndr_pull_bytes(ndr, r->PrintQName, 13));
+       NDR_RETURN(ndr_pull_uint8(ndr, NDR_SCALARS, &r->Pad1));
+       NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->Priority));
+       NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->StartTime));
+       NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->UntilTime));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->SeparatorPageFilename));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintProcessorDllName));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintDestinationsName));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintParameterString));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->CommentString));
+       NDR_RETURN(ndr_pull_rap_PrintQStatusCode(ndr, NDR_SCALARS, &r->PrintQStatus));
+       NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->PrintJobCount));
+
+       return NT_STATUS_OK;
+}
+
+static NTSTATUS rap_pull_rap_PrintQueue3(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue3 *r)
+{
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintQueueName));
+       NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->Priority));
+       NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->StartTime));
+       NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->UntilTime));
+       NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->Pad));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->SeparatorPageFilename));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintProcessorDllName));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintParameterString));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->CommentString));
+       NDR_RETURN(ndr_pull_rap_PrintQStatusCode(ndr, NDR_SCALARS, &r->PrintQStatus));
+       NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->PrintJobCount));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->Printers));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->DriverName));
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintDriverData));
+
+       return NT_STATUS_OK;
+}
+
+static NTSTATUS rap_pull_rap_PrintQueue5(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue5 *r)
+{
+       RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintQueueName));
+
+       return NT_STATUS_OK;
+}
+
+NTSTATUS smbcli_rap_netprintqenum(struct smbcli_tree *tree,
+                                 struct smb_iconv_convenience *iconv_convenience,
+                                 TALLOC_CTX *mem_ctx,
+                                 struct rap_NetPrintQEnum *r)
+{
+       struct rap_call *call;
+       NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
+       int i;
+
+       if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WPrintQEnum))) {
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       rap_cli_push_word(call, r->in.level);
+       rap_cli_push_rcvbuf(call, r->in.bufsize);
+       rap_cli_expect_multiple_entries(call);
+
+       switch(r->in.level) {
+       case 0:
+               rap_cli_expect_format(call, "B13");
+               break;
+       case 1:
+               rap_cli_expect_format(call, "B13BWWWzzzzzWW");
+               break;
+       case 2:
+               rap_cli_expect_format(call, "B13BWWWzzzzzWN");
+               break;
+       case 3:
+               rap_cli_expect_format(call, "zWWWWzzzzWWzzl");
+               break;
+       case 4:
+               rap_cli_expect_format(call, "zWWWWzzzzWNzzl");
+               break;
+       case 5:
+               rap_cli_expect_format(call, "z");
+               break;
+       default:
+               result = NT_STATUS_INVALID_PARAMETER;
+               goto done;
+       }
+
+       if (DEBUGLEVEL >= 10) {
+               NDR_PRINT_IN_DEBUG(rap_NetPrintQEnum, r);
+       }
+
+       result = rap_cli_do_call(tree, iconv_convenience, call);
+
+       if (!NT_STATUS_IS_OK(result))
+               goto done;
+
+       result = NT_STATUS_INVALID_PARAMETER;
+
+       NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
+       NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
+       NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
+       NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
+
+       r->out.info = talloc_zero_array(mem_ctx, union rap_printq_info, r->out.count);
+
+       if (r->out.info == NULL) {
+               result = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
+
+       for (i=0; i<r->out.count; i++) {
+               switch(r->in.level) {
+               case 0:
+                       result = rap_pull_rap_PrintQueue0(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info0);
+                       break;
+               case 1:
+                       result = rap_pull_rap_PrintQueue1(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info1);
+                       break;
+               case 3:
+                       result = rap_pull_rap_PrintQueue3(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info3);
+                       break;
+               case 5:
+                       result = rap_pull_rap_PrintQueue5(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info5);
+                       break;
+               }
+       }
+
+       if (!NT_STATUS_IS_OK(result)) {
+               goto done;
+       }
+
+       if (DEBUGLEVEL >= 10) {
+               NDR_PRINT_OUT_DEBUG(rap_NetPrintQEnum, r);
+       }
+
+       result = NT_STATUS_OK;
+
  done:
        talloc_free(call);
        return result;
@@ -553,6 +732,7 @@ NTSTATUS torture_rap_init(void)
 
        torture_suite_add_suite(suite, suite_basic);
        torture_suite_add_suite(suite, torture_rap_rpc(suite));
+       torture_suite_add_suite(suite, torture_rap_printing(suite));
 
        torture_suite_add_1smb_test(suite_basic, "netserverenum", 
                                    test_netserverenum);