source4/torture/rap/rap.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    test suite for various RAP operations
   4    Copyright (C) Volker Lendecke 2004
   5    Copyright (C) Tim Potter 2005
   6    Copyright (C) Jelmer Vernooij 2007
   7    Copyright (C) Guenther Deschner 2010
   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 "libcli/libcli.h"
  25 #include "torture/smbtorture.h"
  26 #include "torture/util.h"
  27 #include "../librpc/gen_ndr/ndr_rap.h"
  28 #include "librpc/ndr/libndr.h"
  29 #include "param/param.h"
  30 #include "torture/rap/proto.h"
  31
  32 #define RAP_GOTO(call) do { \
  33         NTSTATUS _status; \
  34         _status = call; \
  35         if (!NT_STATUS_IS_OK(_status)) { \
  36                 result = _status; \
  37                 goto done; \
  38         } \
  39 } while (0)
  40
  41 #define RAP_RETURN(call) do { \
  42         NTSTATUS _status; \
  43         _status = call; \
  44         if (!NT_STATUS_IS_OK(_status)) { \
  45                 return _status; \
  46         } \
  47 } while (0)
  48
  49
  50 #define NDR_GOTO(call) do { \
  51         enum ndr_err_code _ndr_err; \
  52         _ndr_err = call; \
  53         if (!NDR_ERR_CODE_IS_SUCCESS(_ndr_err)) { \
  54                 result = ndr_map_error2ntstatus(_ndr_err); \
  55                 goto done; \
  56         } \
  57 } while (0)
  58
  59 #define NDR_RETURN(call) do { \
  60         enum ndr_err_code _ndr_err; \
  61         _ndr_err = call; \
  62         if (!NDR_ERR_CODE_IS_SUCCESS(_ndr_err)) { \
  63                 return ndr_map_error2ntstatus(_ndr_err); \
  64         } \
  65 } while (0)
  66
  67 struct rap_call {
  68         uint16_t callno;
  69         char *paramdesc;
  70         const char *datadesc;
  71         const char *auxdatadesc;
  72
  73         uint16_t status;
  74         uint16_t convert;
  75
  76         uint16_t rcv_paramlen, rcv_datalen;
  77
  78         struct ndr_push *ndr_push_param;
  79         struct ndr_push *ndr_push_data;
  80         struct ndr_pull *ndr_pull_param;
  81         struct ndr_pull *ndr_pull_data;
  82 };
  83
  84 #define RAPNDR_FLAGS (LIBNDR_FLAG_NOALIGN|LIBNDR_FLAG_STR_ASCII|LIBNDR_FLAG_STR_NULLTERM);
  85
  86 static struct rap_call *new_rap_cli_call(TALLOC_CTX *mem_ctx, uint16_t callno)
  87 {
  88         struct rap_call *call;
  89
  90         call = talloc(mem_ctx, struct rap_call);
  91
  92         if (call == NULL)
  93                 return NULL;
  94
  95         call->callno = callno;
  96         call->rcv_paramlen = 4;
  97
  98         call->paramdesc = NULL;
  99         call->datadesc = NULL;
 100         call->auxdatadesc = NULL;
 101
 102         call->ndr_push_param = ndr_push_init_ctx(mem_ctx);
 103         call->ndr_push_param->flags = RAPNDR_FLAGS;
 104
 105         call->ndr_push_data = ndr_push_init_ctx(mem_ctx);
 106         call->ndr_push_data->flags = RAPNDR_FLAGS;
 107
 108         return call;
 109 }
 110
 111 static void rap_cli_push_paramdesc(struct rap_call *call, char desc)
 112 {
 113         int len = 0;
 114
 115         if (call->paramdesc != NULL)
 116                 len = strlen(call->paramdesc);
 117
 118         call->paramdesc = talloc_realloc(call,
 119                                          call->paramdesc,
 120                                          char,
 121                                          len+2);
 122
 123         call->paramdesc[len] = desc;
 124         call->paramdesc[len+1] = '\0';
 125 }
 126
 127 static void rap_cli_push_word(struct rap_call *call, uint16_t val)
 128 {
 129         rap_cli_push_paramdesc(call, 'W');
 130         ndr_push_uint16(call->ndr_push_param, NDR_SCALARS, val);
 131 }
 132
 133 static void rap_cli_push_dword(struct rap_call *call, uint32_t val)
 134 {
 135         rap_cli_push_paramdesc(call, 'D');
 136         ndr_push_uint32(call->ndr_push_param, NDR_SCALARS, val);
 137 }
 138
 139 static void rap_cli_push_rcvbuf(struct rap_call *call, int len)
 140 {
 141         rap_cli_push_paramdesc(call, 'r');
 142         rap_cli_push_paramdesc(call, 'L');
 143         ndr_push_uint16(call->ndr_push_param, NDR_SCALARS, len);
 144         call->rcv_datalen = len;
 145 }
 146
 147 static void rap_cli_push_sendbuf(struct rap_call *call, int len)
 148 {
 149         rap_cli_push_paramdesc(call, 's');
 150         rap_cli_push_paramdesc(call, 'T');
 151         ndr_push_uint16(call->ndr_push_param, NDR_SCALARS, len);
 152 }
 153
 154 static void rap_cli_push_param(struct rap_call *call, uint16_t val)
 155 {
 156         rap_cli_push_paramdesc(call, 'P');
 157         ndr_push_uint16(call->ndr_push_param, NDR_SCALARS, val);
 158 }
 159
 160 static void rap_cli_expect_multiple_entries(struct rap_call *call)
 161 {
 162         rap_cli_push_paramdesc(call, 'e');
 163         rap_cli_push_paramdesc(call, 'h');
 164         call->rcv_paramlen += 4; /* uint16_t entry count, uint16_t total */
 165 }
 166
 167 static void rap_cli_expect_word(struct rap_call *call)
 168 {
 169         rap_cli_push_paramdesc(call, 'h');
 170         call->rcv_paramlen += 2;
 171 }
 172
 173 static void rap_cli_push_string(struct rap_call *call, const char *str)
 174 {
 175         if (str == NULL) {
 176                 rap_cli_push_paramdesc(call, 'O');
 177                 return;
 178         }
 179         rap_cli_push_paramdesc(call, 'z');
 180         ndr_push_string(call->ndr_push_param, NDR_SCALARS, str);
 181 }
 182
 183 static void rap_cli_expect_format(struct rap_call *call, const char *format)
 184 {
 185         call->datadesc = format;
 186 }
 187
 188 static void rap_cli_expect_extra_format(struct rap_call *call, const char *format)
 189 {
 190         call->auxdatadesc = format;
 191 }
 192
 193 static NTSTATUS rap_pull_string(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr,
 194                                 uint16_t convert, const char **dest)
 195 {
 196         uint16_t string_offset;
 197         uint16_t ignore;
 198         const char *p;
 199         size_t len;
 200
 201         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &string_offset));
 202         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &ignore));
 203
 204         string_offset -= convert;
 205
 206         if (string_offset+1 > ndr->data_size)
 207                 return NT_STATUS_INVALID_PARAMETER;
 208
 209         p = (const char *)(ndr->data + string_offset);
 210         len = strnlen(p, ndr->data_size-string_offset);
 211
 212         if ( string_offset + len + 1 >  ndr->data_size )
 213                 return NT_STATUS_INVALID_PARAMETER;
 214
 215         *dest = talloc_zero_array(mem_ctx, char, len+1);
 216         pull_string(discard_const_p(char, *dest), p, len+1, len, STR_ASCII);
 217
 218         return NT_STATUS_OK;
 219 }
 220
 221 static NTSTATUS rap_cli_do_call(struct smbcli_tree *tree,
 222                                 struct rap_call *call)
 223 {
 224         NTSTATUS result;
 225         DATA_BLOB param_blob;
 226         DATA_BLOB data_blob;
 227         struct ndr_push *params;
 228         struct ndr_push *data;
 229         struct smb_trans2 trans;
 230
 231         params = ndr_push_init_ctx(call);
 232
 233         if (params == NULL)
 234                 return NT_STATUS_NO_MEMORY;
 235
 236         params->flags = RAPNDR_FLAGS;
 237
 238         data = ndr_push_init_ctx(call);
 239
 240         if (data == NULL)
 241                 return NT_STATUS_NO_MEMORY;
 242
 243         data->flags = RAPNDR_FLAGS;
 244
 245         trans.in.max_param = call->rcv_paramlen;
 246         trans.in.max_data = call->rcv_datalen;
 247         trans.in.max_setup = 0;
 248         trans.in.flags = 0;
 249         trans.in.timeout = 0;
 250         trans.in.setup_count = 0;
 251         trans.in.setup = NULL;
 252         trans.in.trans_name = "\\PIPE\\LANMAN";
 253
 254         NDR_RETURN(ndr_push_uint16(params, NDR_SCALARS, call->callno));
 255         if (call->paramdesc)
 256                 NDR_RETURN(ndr_push_string(params, NDR_SCALARS, call->paramdesc));
 257         if (call->datadesc)
 258                 NDR_RETURN(ndr_push_string(params, NDR_SCALARS, call->datadesc));
 259
 260         param_blob = ndr_push_blob(call->ndr_push_param);
 261         NDR_RETURN(ndr_push_bytes(params, param_blob.data,
 262                                  param_blob.length));
 263
 264         data_blob = ndr_push_blob(call->ndr_push_data);
 265         NDR_RETURN(ndr_push_bytes(data, data_blob.data,
 266                                  data_blob.length));
 267
 268         if (call->auxdatadesc)
 269                 NDR_RETURN(ndr_push_string(params, NDR_SCALARS, call->auxdatadesc));
 270
 271         trans.in.params = ndr_push_blob(params);
 272         trans.in.data = ndr_push_blob(data);
 273
 274         result = smb_raw_trans(tree, call, &trans);
 275
 276         if (!NT_STATUS_IS_OK(result))
 277                 return result;
 278
 279         call->ndr_pull_param = ndr_pull_init_blob(&trans.out.params, call);
 280         call->ndr_pull_param->flags = RAPNDR_FLAGS;
 281
 282         call->ndr_pull_data = ndr_pull_init_blob(&trans.out.data, call);
 283         call->ndr_pull_data->flags = RAPNDR_FLAGS;
 284
 285         return result;
 286 }
 287
 288
 289 static NTSTATUS smbcli_rap_netshareenum(struct smbcli_tree *tree,
 290                                         TALLOC_CTX *mem_ctx,
 291                                         struct rap_NetShareEnum *r)
 292 {
 293         struct rap_call *call;
 294         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 295         int i;
 296
 297         call = new_rap_cli_call(tree, RAP_WshareEnum);
 298
 299         if (call == NULL)
 300                 return NT_STATUS_NO_MEMORY;
 301
 302         rap_cli_push_word(call, r->in.level); /* Level */
 303         rap_cli_push_rcvbuf(call, r->in.bufsize);
 304         rap_cli_expect_multiple_entries(call);
 305
 306         switch(r->in.level) {
 307         case 0:
 308                 rap_cli_expect_format(call, "B13");
 309                 break;
 310         case 1:
 311                 rap_cli_expect_format(call, "B13BWz");
 312                 break;
 313         }
 314
 315         if (DEBUGLEVEL >= 10) {
 316                 NDR_PRINT_IN_DEBUG(rap_NetShareEnum, r);
 317         }
 318
 319         result = rap_cli_do_call(tree, call);
 320
 321         if (!NT_STATUS_IS_OK(result))
 322                 goto done;
 323
 324         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 325         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 326         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
 327         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
 328
 329         r->out.info = talloc_array(mem_ctx, union rap_share_info, r->out.count);
 330
 331         if (r->out.info == NULL) {
 332                 result = NT_STATUS_NO_MEMORY;
 333                 goto done;
 334         }
 335
 336         for (i=0; i<r->out.count; i++) {
 337                 switch(r->in.level) {
 338                 case 0:
 339                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 340                                                 r->out.info[i].info0.share_name, 13));
 341                         break;
 342                 case 1:
 343                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 344                                                 r->out.info[i].info1.share_name, 13));
 345                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 346                                                 &r->out.info[i].info1.reserved1, 1));
 347                         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_data,
 348                                                NDR_SCALARS, &r->out.info[i].info1.share_type));
 349                         RAP_GOTO(rap_pull_string(mem_ctx, call->ndr_pull_data,
 350                                                r->out.convert,
 351                                                &r->out.info[i].info1.comment));
 352                         break;
 353                 }
 354         }
 355
 356         if (DEBUGLEVEL >= 10) {
 357                 NDR_PRINT_OUT_DEBUG(rap_NetShareEnum, r);
 358         }
 359         result = NT_STATUS_OK;
 360
 361  done:
 362         talloc_free(call);
 363         return result;
 364 }
 365
 366 static bool test_netshareenum(struct torture_context *tctx,
 367                               struct smbcli_state *cli)
 368 {
 369         struct rap_NetShareEnum r;
 370         int i;
 371
 372         r.in.level = 1;
 373         r.in.bufsize = 8192;
 374
 375         torture_assert_ntstatus_ok(tctx,
 376                 smbcli_rap_netshareenum(cli->tree, tctx, &r), "");
 377
 378         for (i=0; i<r.out.count; i++) {
 379                 printf("%s %d %s\n", r.out.info[i].info1.share_name,
 380                        r.out.info[i].info1.share_type,
 381                        r.out.info[i].info1.comment);
 382         }
 383
 384         return true;
 385 }
 386
 387 static NTSTATUS smbcli_rap_netserverenum2(struct smbcli_tree *tree,
 388                                           TALLOC_CTX *mem_ctx,
 389                                           struct rap_NetServerEnum2 *r)
 390 {
 391         struct rap_call *call;
 392         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 393         int i;
 394
 395         call = new_rap_cli_call(mem_ctx, RAP_NetServerEnum2);
 396
 397         if (call == NULL)
 398                 return NT_STATUS_NO_MEMORY;
 399
 400         rap_cli_push_word(call, r->in.level);
 401         rap_cli_push_rcvbuf(call, r->in.bufsize);
 402         rap_cli_expect_multiple_entries(call);
 403         rap_cli_push_dword(call, r->in.servertype);
 404         rap_cli_push_string(call, r->in.domain);
 405
 406         switch(r->in.level) {
 407         case 0:
 408                 rap_cli_expect_format(call, "B16");
 409                 break;
 410         case 1:
 411                 rap_cli_expect_format(call, "B16BBDz");
 412                 break;
 413         }
 414
 415         if (DEBUGLEVEL >= 10) {
 416                 NDR_PRINT_IN_DEBUG(rap_NetServerEnum2, r);
 417         }
 418
 419         result = rap_cli_do_call(tree, call);
 420
 421         if (!NT_STATUS_IS_OK(result))
 422                 goto done;
 423
 424         result = NT_STATUS_INVALID_PARAMETER;
 425
 426         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 427         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 428         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
 429         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
 430
 431         r->out.info = talloc_array(mem_ctx, union rap_server_info, r->out.count);
 432
 433         if (r->out.info == NULL) {
 434                 result = NT_STATUS_NO_MEMORY;
 435                 goto done;
 436         }
 437
 438         for (i=0; i<r->out.count; i++) {
 439                 switch(r->in.level) {
 440                 case 0:
 441                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 442                                                 r->out.info[i].info0.name, 16));
 443                         break;
 444                 case 1:
 445                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 446                                                 r->out.info[i].info1.name, 16));
 447                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 448                                               &r->out.info[i].info1.version_major, 1));
 449                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 450                                               &r->out.info[i].info1.version_minor, 1));
 451                         NDR_GOTO(ndr_pull_uint32(call->ndr_pull_data,
 452                                                NDR_SCALARS, &r->out.info[i].info1.servertype));
 453                         RAP_GOTO(rap_pull_string(mem_ctx, call->ndr_pull_data,
 454                                                r->out.convert,
 455                                                &r->out.info[i].info1.comment));
 456                 }
 457         }
 458
 459         if (DEBUGLEVEL >= 10) {
 460                 NDR_PRINT_OUT_DEBUG(rap_NetServerEnum2, r);
 461         }
 462
 463         result = NT_STATUS_OK;
 464
 465  done:
 466         talloc_free(call);
 467         return result;
 468 }
 469
 470 static bool test_netserverenum(struct torture_context *tctx,
 471                                struct smbcli_state *cli)
 472 {
 473         struct rap_NetServerEnum2 r;
 474         int i;
 475
 476         r.in.level = 0;
 477         r.in.bufsize = 8192;
 478         r.in.servertype = 0xffffffff;
 479         r.in.servertype = 0x80000000;
 480         r.in.domain = NULL;
 481
 482         torture_assert_ntstatus_ok(tctx,
 483                    smbcli_rap_netserverenum2(cli->tree, tctx, &r), "");
 484
 485         for (i=0; i<r.out.count; i++) {
 486                 switch (r.in.level) {
 487                 case 0:
 488                         printf("%s\n", r.out.info[i].info0.name);
 489                         break;
 490                 case 1:
 491                         printf("%s %x %s\n", r.out.info[i].info1.name,
 492                                r.out.info[i].info1.servertype,
 493                                r.out.info[i].info1.comment);
 494                         break;
 495                 }
 496         }
 497
 498         return true;
 499 }
 500
 501 NTSTATUS smbcli_rap_netservergetinfo(struct smbcli_tree *tree,
 502                                      TALLOC_CTX *mem_ctx,
 503                                      struct rap_WserverGetInfo *r)
 504 {
 505         struct rap_call *call;
 506         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 507
 508         if (!(call = new_rap_cli_call(mem_ctx, RAP_WserverGetInfo))) {
 509                 return NT_STATUS_NO_MEMORY;
 510         }
 511
 512         rap_cli_push_word(call, r->in.level);
 513         rap_cli_push_rcvbuf(call, r->in.bufsize);
 514         rap_cli_expect_word(call);
 515
 516         switch(r->in.level) {
 517         case 0:
 518                 rap_cli_expect_format(call, "B16");
 519                 break;
 520         case 1:
 521                 rap_cli_expect_format(call, "B16BBDz");
 522                 break;
 523         default:
 524                 result = NT_STATUS_INVALID_PARAMETER;
 525                 goto done;
 526         }
 527
 528         if (DEBUGLEVEL >= 10) {
 529                 NDR_PRINT_IN_DEBUG(rap_WserverGetInfo, r);
 530         }
 531
 532         result = rap_cli_do_call(tree, call);
 533
 534         if (!NT_STATUS_IS_OK(result))
 535                 goto done;
 536
 537         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 538         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 539         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
 540
 541         switch(r->in.level) {
 542         case 0:
 543                 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 544                                         r->out.info.info0.name, 16));
 545                 break;
 546         case 1:
 547                 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 548                                         r->out.info.info1.name, 16));
 549                 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 550                                       &r->out.info.info1.version_major, 1));
 551                 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 552                                       &r->out.info.info1.version_minor, 1));
 553                 NDR_GOTO(ndr_pull_uint32(call->ndr_pull_data,
 554                                        NDR_SCALARS, &r->out.info.info1.servertype));
 555                 RAP_GOTO(rap_pull_string(mem_ctx, call->ndr_pull_data,
 556                                        r->out.convert,
 557                                        &r->out.info.info1.comment));
 558         }
 559
 560         if (DEBUGLEVEL >= 10) {
 561                 NDR_PRINT_OUT_DEBUG(rap_WserverGetInfo, r);
 562         }
 563  done:
 564         talloc_free(call);
 565         return result;
 566 }
 567
 568 static enum ndr_err_code ndr_pull_rap_NetPrintQEnum_data(struct ndr_pull *ndr, struct rap_NetPrintQEnum *r)
 569 {
 570         uint32_t cntr_info_0;
 571         TALLOC_CTX *_mem_save_info_0;
 572
 573         NDR_PULL_ALLOC_N(ndr, r->out.info, r->out.count);
 574         _mem_save_info_0 = NDR_PULL_GET_MEM_CTX(ndr);
 575         NDR_PULL_SET_MEM_CTX(ndr, r->out.info, 0);
 576         for (cntr_info_0 = 0; cntr_info_0 < r->out.count; cntr_info_0++) {
 577                 NDR_CHECK(ndr_pull_set_switch_value(ndr, &r->out.info[cntr_info_0], r->in.level));
 578                 NDR_CHECK(ndr_pull_rap_printq_info(ndr, NDR_SCALARS, &r->out.info[cntr_info_0]));
 579         }
 580         for (cntr_info_0 = 0; cntr_info_0 < r->out.count; cntr_info_0++) {
 581                 NDR_CHECK(ndr_pull_rap_printq_info(ndr, NDR_BUFFERS, &r->out.info[cntr_info_0]));
 582         }
 583         NDR_PULL_SET_MEM_CTX(ndr, _mem_save_info_0, 0);
 584
 585         return NDR_ERR_SUCCESS;
 586 }
 587
 588 NTSTATUS smbcli_rap_netprintqenum(struct smbcli_tree *tree,
 589                                   TALLOC_CTX *mem_ctx,
 590                                   struct rap_NetPrintQEnum *r)
 591 {
 592         struct rap_call *call;
 593         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 594
 595         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintQEnum))) {
 596                 return NT_STATUS_NO_MEMORY;
 597         }
 598
 599         rap_cli_push_word(call, r->in.level);
 600         rap_cli_push_rcvbuf(call, r->in.bufsize);
 601         rap_cli_expect_multiple_entries(call);
 602
 603         switch(r->in.level) {
 604         case 0:
 605                 rap_cli_expect_format(call, "B13");
 606                 break;
 607         case 1:
 608                 rap_cli_expect_format(call, "B13BWWWzzzzzWW");
 609                 break;
 610         case 2:
 611                 rap_cli_expect_format(call, "B13BWWWzzzzzWN");
 612                 rap_cli_expect_extra_format(call, "WB21BB16B10zWWzDDz");
 613                 break;
 614         case 3:
 615                 rap_cli_expect_format(call, "zWWWWzzzzWWzzl");
 616                 break;
 617         case 4:
 618                 rap_cli_expect_format(call, "zWWWWzzzzWNzzl");
 619                 rap_cli_expect_extra_format(call, "WWzWWDDzz");
 620                 /* no mention of extra format in MS-RAP */
 621                 break;
 622         case 5:
 623                 rap_cli_expect_format(call, "z");
 624                 break;
 625         default:
 626                 result = NT_STATUS_INVALID_PARAMETER;
 627                 goto done;
 628         }
 629
 630         if (DEBUGLEVEL >= 10) {
 631                 NDR_PRINT_IN_DEBUG(rap_NetPrintQEnum, r);
 632         }
 633
 634         result = rap_cli_do_call(tree, call);
 635
 636         if (!NT_STATUS_IS_OK(result))
 637                 goto done;
 638
 639         result = NT_STATUS_INVALID_PARAMETER;
 640
 641         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 642         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 643         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
 644         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
 645
 646         call->ndr_pull_data->relative_rap_convert = r->out.convert;
 647
 648         NDR_GOTO(ndr_pull_rap_NetPrintQEnum_data(call->ndr_pull_data, r));
 649
 650         r->out.info = talloc_steal(mem_ctx, r->out.info);
 651
 652         if (DEBUGLEVEL >= 10) {
 653                 NDR_PRINT_OUT_DEBUG(rap_NetPrintQEnum, r);
 654         }
 655
 656         result = NT_STATUS_OK;
 657
 658  done:
 659         talloc_free(call);
 660         return result;
 661 }
 662
 663 NTSTATUS smbcli_rap_netprintqgetinfo(struct smbcli_tree *tree,
 664                                      TALLOC_CTX *mem_ctx,
 665                                      struct rap_NetPrintQGetInfo *r)
 666 {
 667         struct rap_call *call;
 668         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 669
 670         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintQGetInfo))) {
 671                 return NT_STATUS_NO_MEMORY;
 672         }
 673
 674         rap_cli_push_string(call, r->in.PrintQueueName);
 675         rap_cli_push_word(call, r->in.level);
 676         rap_cli_push_rcvbuf(call, r->in.bufsize);
 677         rap_cli_expect_word(call);
 678
 679         switch(r->in.level) {
 680         case 0:
 681                 rap_cli_expect_format(call, "B13");
 682                 break;
 683         case 1:
 684                 rap_cli_expect_format(call, "B13BWWWzzzzzWW");
 685                 break;
 686         case 2:
 687                 rap_cli_expect_format(call, "B13BWWWzzzzzWN");
 688                 rap_cli_expect_extra_format(call, "WB21BB16B10zWWzDDz");
 689                 break;
 690         case 3:
 691                 rap_cli_expect_format(call, "zWWWWzzzzWWzzl");
 692                 break;
 693         case 4:
 694                 rap_cli_expect_format(call, "zWWWWzzzzWNzzl");
 695                 rap_cli_expect_extra_format(call, "WWzWWDDzz");
 696                 /* no mention of extra format in MS-RAP */
 697                 break;
 698         case 5:
 699                 rap_cli_expect_format(call, "z");
 700                 break;
 701         default:
 702                 result = NT_STATUS_INVALID_PARAMETER;
 703                 goto done;
 704         }
 705
 706         if (DEBUGLEVEL >= 10) {
 707                 NDR_PRINT_IN_DEBUG(rap_NetPrintQGetInfo, r);
 708         }
 709
 710         result = rap_cli_do_call(tree, call);
 711
 712         if (!NT_STATUS_IS_OK(result))
 713                 goto done;
 714
 715         result = NT_STATUS_INVALID_PARAMETER;
 716
 717         ZERO_STRUCT(r->out);
 718
 719         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 720         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 721         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
 722
 723         if (r->out.status == 0) {
 724                 call->ndr_pull_data->relative_rap_convert = r->out.convert;
 725
 726                 NDR_GOTO(ndr_pull_set_switch_value(call->ndr_pull_data, &r->out.info, r->in.level));
 727                 NDR_GOTO(ndr_pull_rap_printq_info(call->ndr_pull_data, NDR_SCALARS|NDR_BUFFERS, &r->out.info));
 728         }
 729
 730         if (DEBUGLEVEL >= 10) {
 731                 NDR_PRINT_OUT_DEBUG(rap_NetPrintQGetInfo, r);
 732         }
 733
 734         result = NT_STATUS_OK;
 735  done:
 736         talloc_free(call);
 737         return result;
 738 }
 739
 740 NTSTATUS smbcli_rap_netprintjobpause(struct smbcli_tree *tree,
 741                                      TALLOC_CTX *mem_ctx,
 742                                      struct rap_NetPrintJobPause *r)
 743 {
 744         struct rap_call *call;
 745         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 746
 747         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintJobPause))) {
 748                 return NT_STATUS_NO_MEMORY;
 749         }
 750
 751         rap_cli_push_word(call, r->in.JobID);
 752
 753         rap_cli_expect_format(call, "W");
 754
 755         if (DEBUGLEVEL >= 10) {
 756                 NDR_PRINT_IN_DEBUG(rap_NetPrintJobPause, r);
 757         }
 758
 759         result = rap_cli_do_call(tree, call);
 760
 761         if (!NT_STATUS_IS_OK(result))
 762                 goto done;
 763
 764         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 765         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 766
 767         if (DEBUGLEVEL >= 10) {
 768                 NDR_PRINT_OUT_DEBUG(rap_NetPrintJobPause, r);
 769         }
 770
 771  done:
 772         talloc_free(call);
 773         return result;
 774 }
 775
 776 NTSTATUS smbcli_rap_netprintjobcontinue(struct smbcli_tree *tree,
 777                                         TALLOC_CTX *mem_ctx,
 778                                         struct rap_NetPrintJobContinue *r)
 779 {
 780         struct rap_call *call;
 781         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 782
 783         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintJobContinue))) {
 784                 return NT_STATUS_NO_MEMORY;
 785         }
 786
 787         rap_cli_push_word(call, r->in.JobID);
 788
 789         rap_cli_expect_format(call, "W");
 790
 791         if (DEBUGLEVEL >= 10) {
 792                 NDR_PRINT_IN_DEBUG(rap_NetPrintJobContinue, r);
 793         }
 794
 795         result = rap_cli_do_call(tree, call);
 796
 797         if (!NT_STATUS_IS_OK(result))
 798                 goto done;
 799
 800         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 801         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 802
 803         if (DEBUGLEVEL >= 10) {
 804                 NDR_PRINT_OUT_DEBUG(rap_NetPrintJobContinue, r);
 805         }
 806
 807  done:
 808         talloc_free(call);
 809         return result;
 810 }
 811
 812 NTSTATUS smbcli_rap_netprintjobdelete(struct smbcli_tree *tree,
 813                                       TALLOC_CTX *mem_ctx,
 814                                       struct rap_NetPrintJobDelete *r)
 815 {
 816         struct rap_call *call;
 817         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 818
 819         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintJobDel))) {
 820                 return NT_STATUS_NO_MEMORY;
 821         }
 822
 823         rap_cli_push_word(call, r->in.JobID);
 824
 825         rap_cli_expect_format(call, "W");
 826
 827         if (DEBUGLEVEL >= 10) {
 828                 NDR_PRINT_IN_DEBUG(rap_NetPrintJobDelete, r);
 829         }
 830
 831         result = rap_cli_do_call(tree, call);
 832
 833         if (!NT_STATUS_IS_OK(result))
 834                 goto done;
 835
 836         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 837         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 838
 839         if (DEBUGLEVEL >= 10) {
 840                 NDR_PRINT_OUT_DEBUG(rap_NetPrintJobDelete, r);
 841         }
 842
 843  done:
 844         talloc_free(call);
 845         return result;
 846 }
 847
 848 NTSTATUS smbcli_rap_netprintqueuepause(struct smbcli_tree *tree,
 849                                        TALLOC_CTX *mem_ctx,
 850                                        struct rap_NetPrintQueuePause *r)
 851 {
 852         struct rap_call *call;
 853         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 854
 855         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintQPause))) {
 856                 return NT_STATUS_NO_MEMORY;
 857         }
 858
 859         rap_cli_push_string(call, r->in.PrintQueueName);
 860
 861         rap_cli_expect_format(call, "");
 862
 863         if (DEBUGLEVEL >= 10) {
 864                 NDR_PRINT_IN_DEBUG(rap_NetPrintQueuePause, r);
 865         }
 866
 867         result = rap_cli_do_call(tree, call);
 868
 869         if (!NT_STATUS_IS_OK(result))
 870                 goto done;
 871
 872         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 873         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 874
 875         if (DEBUGLEVEL >= 10) {
 876                 NDR_PRINT_OUT_DEBUG(rap_NetPrintQueuePause, r);
 877         }
 878
 879  done:
 880         talloc_free(call);
 881         return result;
 882 }
 883
 884 NTSTATUS smbcli_rap_netprintqueueresume(struct smbcli_tree *tree,
 885                                         TALLOC_CTX *mem_ctx,
 886                                         struct rap_NetPrintQueueResume *r)
 887 {
 888         struct rap_call *call;
 889         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 890
 891         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintQContinue))) {
 892                 return NT_STATUS_NO_MEMORY;
 893         }
 894
 895         rap_cli_push_string(call, r->in.PrintQueueName);
 896
 897         rap_cli_expect_format(call, "");
 898
 899         if (DEBUGLEVEL >= 10) {
 900                 NDR_PRINT_IN_DEBUG(rap_NetPrintQueueResume, r);
 901         }
 902
 903         result = rap_cli_do_call(tree, call);
 904
 905         if (!NT_STATUS_IS_OK(result))
 906                 goto done;
 907
 908         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 909         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 910
 911         if (DEBUGLEVEL >= 10) {
 912                 NDR_PRINT_OUT_DEBUG(rap_NetPrintQueueResume, r);
 913         }
 914
 915  done:
 916         talloc_free(call);
 917         return result;
 918 }
 919
 920 NTSTATUS smbcli_rap_netprintqueuepurge(struct smbcli_tree *tree,
 921                                        TALLOC_CTX *mem_ctx,
 922                                        struct rap_NetPrintQueuePurge *r)
 923 {
 924         struct rap_call *call;
 925         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 926
 927         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintQPurge))) {
 928                 return NT_STATUS_NO_MEMORY;
 929         }
 930
 931         rap_cli_push_string(call, r->in.PrintQueueName);
 932
 933         rap_cli_expect_format(call, "");
 934
 935         if (DEBUGLEVEL >= 10) {
 936                 NDR_PRINT_IN_DEBUG(rap_NetPrintQueuePurge, r);
 937         }
 938
 939         result = rap_cli_do_call(tree, call);
 940
 941         if (!NT_STATUS_IS_OK(result))
 942                 goto done;
 943
 944         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 945         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 946
 947         if (DEBUGLEVEL >= 10) {
 948                 NDR_PRINT_OUT_DEBUG(rap_NetPrintQueuePurge, r);
 949         }
 950
 951  done:
 952         talloc_free(call);
 953         return result;
 954 }
 955
 956 static enum ndr_err_code ndr_pull_rap_NetPrintJobEnum_data(struct ndr_pull *ndr, struct rap_NetPrintJobEnum *r)
 957 {
 958         uint32_t cntr_info_0;
 959         TALLOC_CTX *_mem_save_info_0;
 960
 961         NDR_PULL_ALLOC_N(ndr, r->out.info, r->out.count);
 962         _mem_save_info_0 = NDR_PULL_GET_MEM_CTX(ndr);
 963         NDR_PULL_SET_MEM_CTX(ndr, r->out.info, 0);
 964         for (cntr_info_0 = 0; cntr_info_0 < r->out.count; cntr_info_0++) {
 965                 NDR_CHECK(ndr_pull_set_switch_value(ndr, &r->out.info[cntr_info_0], r->in.level));
 966                 NDR_CHECK(ndr_pull_rap_printj_info(ndr, NDR_SCALARS, &r->out.info[cntr_info_0]));
 967         }
 968         for (cntr_info_0 = 0; cntr_info_0 < r->out.count; cntr_info_0++) {
 969                 NDR_CHECK(ndr_pull_rap_printj_info(ndr, NDR_BUFFERS, &r->out.info[cntr_info_0]));
 970         }
 971         NDR_PULL_SET_MEM_CTX(ndr, _mem_save_info_0, 0);
 972
 973         return NDR_ERR_SUCCESS;
 974 }
 975
 976 NTSTATUS smbcli_rap_netprintjobenum(struct smbcli_tree *tree,
 977                                     TALLOC_CTX *mem_ctx,
 978                                     struct rap_NetPrintJobEnum *r)
 979 {
 980         struct rap_call *call;
 981         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 982
 983         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintJobEnum))) {
 984                 return NT_STATUS_NO_MEMORY;
 985         }
 986
 987         rap_cli_push_string(call, r->in.PrintQueueName);
 988         rap_cli_push_word(call, r->in.level);
 989         rap_cli_push_rcvbuf(call, r->in.bufsize);
 990         rap_cli_expect_multiple_entries(call);
 991
 992         switch(r->in.level) {
 993         case 0:
 994                 rap_cli_expect_format(call, "W");
 995                 break;
 996         case 1:
 997                 rap_cli_expect_format(call, "WB21BB16B10zWWzDDz");
 998                 break;
 999         case 2:
1000                 rap_cli_expect_format(call, "WWzWWDDzz");
1001                 break;
1002         case 3:
1003                 rap_cli_expect_format(call, "WWzWWDDzzzzzzzzzzlz");
1004                 break;
1005         case 4:
1006                 rap_cli_expect_format(call, "WWzWWDDzzzzzDDDDDDD");
1007                 break;
1008         default:
1009                 result = NT_STATUS_INVALID_PARAMETER;
1010                 goto done;
1011         }
1012
1013         if (DEBUGLEVEL >= 10) {
1014                 NDR_PRINT_IN_DEBUG(rap_NetPrintJobEnum, r);
1015         }
1016
1017         result = rap_cli_do_call(tree, call);
1018
1019         if (!NT_STATUS_IS_OK(result))
1020                 goto done;
1021
1022         result = NT_STATUS_INVALID_PARAMETER;
1023
1024         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
1025         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
1026         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
1027         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
1028
1029         call->ndr_pull_data->relative_rap_convert = r->out.convert;
1030
1031         NDR_GOTO(ndr_pull_rap_NetPrintJobEnum_data(call->ndr_pull_data, r));
1032
1033         if (DEBUGLEVEL >= 10) {
1034                 NDR_PRINT_OUT_DEBUG(rap_NetPrintJobEnum, r);
1035         }
1036
1037         r->out.info = talloc_steal(mem_ctx, r->out.info);
1038
1039         result = NT_STATUS_OK;
1040
1041  done:
1042         talloc_free(call);
1043         return result;
1044 }
1045
1046 NTSTATUS smbcli_rap_netprintjobgetinfo(struct smbcli_tree *tree,
1047                                        TALLOC_CTX *mem_ctx,
1048                                        struct rap_NetPrintJobGetInfo *r)
1049 {
1050         struct rap_call *call;
1051         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1052
1053         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintJobGetInfo))) {
1054                 return NT_STATUS_NO_MEMORY;
1055         }
1056
1057         rap_cli_push_word(call, r->in.JobID);
1058         rap_cli_push_word(call, r->in.level);
1059         rap_cli_push_rcvbuf(call, r->in.bufsize);
1060         rap_cli_expect_word(call);
1061
1062         switch(r->in.level) {
1063         case 0:
1064                 rap_cli_expect_format(call, "W");
1065                 break;
1066         case 1:
1067                 rap_cli_expect_format(call, "WB21BB16B10zWWzDDz");
1068                 break;
1069         case 2:
1070                 rap_cli_expect_format(call, "WWzWWDDzz");
1071                 break;
1072         case 3:
1073                 rap_cli_expect_format(call, "WWzWWDDzzzzzzzzzzlz");
1074                 break;
1075         case 4:
1076                 rap_cli_expect_format(call, "WWzWWDDzzzzzDDDDDDD");
1077                 break;
1078         default:
1079                 result = NT_STATUS_INVALID_PARAMETER;
1080                 goto done;
1081         }
1082
1083         if (DEBUGLEVEL >= 10) {
1084                 NDR_PRINT_IN_DEBUG(rap_NetPrintJobGetInfo, r);
1085         }
1086
1087         result = rap_cli_do_call(tree, call);
1088
1089         if (!NT_STATUS_IS_OK(result))
1090                 goto done;
1091
1092         result = NT_STATUS_INVALID_PARAMETER;
1093
1094         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
1095         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
1096         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
1097
1098         call->ndr_pull_data->relative_rap_convert = r->out.convert;
1099
1100         NDR_GOTO(ndr_pull_set_switch_value(call->ndr_pull_data, &r->out.info, r->in.level));
1101         NDR_GOTO(ndr_pull_rap_printj_info(call->ndr_pull_data, NDR_SCALARS|NDR_BUFFERS, &r->out.info));
1102
1103         if (DEBUGLEVEL >= 10) {
1104                 NDR_PRINT_OUT_DEBUG(rap_NetPrintJobGetInfo, r);
1105         }
1106
1107         result = NT_STATUS_OK;
1108
1109  done:
1110         talloc_free(call);
1111         return result;
1112 }
1113
1114 NTSTATUS smbcli_rap_netprintjobsetinfo(struct smbcli_tree *tree,
1115                                        TALLOC_CTX *mem_ctx,
1116                                        struct rap_NetPrintJobSetInfo *r)
1117 {
1118         struct rap_call *call;
1119         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1120
1121         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintJobSetInfo))) {
1122                 return NT_STATUS_NO_MEMORY;
1123         }
1124
1125         rap_cli_push_word(call, r->in.JobID);
1126         rap_cli_push_word(call, r->in.level);
1127         rap_cli_push_sendbuf(call, r->in.bufsize);
1128         rap_cli_push_param(call, r->in.ParamNum);
1129
1130         switch (r->in.ParamNum) {
1131         case RAP_PARAM_JOBNUM:
1132         case RAP_PARAM_JOBPOSITION:
1133         case RAP_PARAM_JOBSTATUS:
1134                 NDR_GOTO(ndr_push_uint16(call->ndr_push_param, NDR_SCALARS, r->in.Param.value));
1135                 break;
1136         case RAP_PARAM_USERNAME:
1137         case RAP_PARAM_NOTIFYNAME:
1138         case RAP_PARAM_DATATYPE:
1139         case RAP_PARAM_PARAMETERS_STRING:
1140         case RAP_PARAM_JOBSTATUSSTR:
1141         case RAP_PARAM_JOBCOMMENT:
1142                 NDR_GOTO(ndr_push_string(call->ndr_push_param, NDR_SCALARS, r->in.Param.string));
1143                 break;
1144         case RAP_PARAM_TIMESUBMITTED:
1145         case RAP_PARAM_JOBSIZE:
1146                 NDR_GOTO(ndr_push_uint32(call->ndr_push_param, NDR_SCALARS, r->in.Param.value4));
1147                 break;
1148         default:
1149                 result = NT_STATUS_INVALID_PARAMETER;
1150                 break;
1151         }
1152
1153         /* not really sure if this is correct */
1154         rap_cli_expect_format(call, "WB21BB16B10zWWzDDz");
1155
1156         if (DEBUGLEVEL >= 10) {
1157                 NDR_PRINT_IN_DEBUG(rap_NetPrintJobSetInfo, r);
1158         }
1159
1160         result = rap_cli_do_call(tree, call);
1161
1162         if (!NT_STATUS_IS_OK(result))
1163                 goto done;
1164
1165         result = NT_STATUS_INVALID_PARAMETER;
1166
1167         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
1168         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
1169
1170         result = NT_STATUS_OK;
1171
1172         if (!NT_STATUS_IS_OK(result)) {
1173                 goto done;
1174         }
1175
1176         if (DEBUGLEVEL >= 10) {
1177                 NDR_PRINT_OUT_DEBUG(rap_NetPrintJobSetInfo, r);
1178         }
1179
1180  done:
1181         talloc_free(call);
1182         return result;
1183 }
1184
1185 static enum ndr_err_code ndr_pull_rap_NetPrintDestEnum_data(struct ndr_pull *ndr, struct rap_NetPrintDestEnum *r)
1186 {
1187         uint32_t cntr_info_0;
1188         TALLOC_CTX *_mem_save_info_0;
1189
1190         NDR_PULL_ALLOC_N(ndr, r->out.info, r->out.count);
1191         _mem_save_info_0 = NDR_PULL_GET_MEM_CTX(ndr);
1192         NDR_PULL_SET_MEM_CTX(ndr, r->out.info, 0);
1193         for (cntr_info_0 = 0; cntr_info_0 < r->out.count; cntr_info_0++) {
1194                 NDR_CHECK(ndr_pull_set_switch_value(ndr, &r->out.info[cntr_info_0], r->in.level));
1195                 NDR_CHECK(ndr_pull_rap_printdest_info(ndr, NDR_SCALARS, &r->out.info[cntr_info_0]));
1196         }
1197         for (cntr_info_0 = 0; cntr_info_0 < r->out.count; cntr_info_0++) {
1198                 NDR_CHECK(ndr_pull_rap_printdest_info(ndr, NDR_BUFFERS, &r->out.info[cntr_info_0]));
1199         }
1200         NDR_PULL_SET_MEM_CTX(ndr, _mem_save_info_0, 0);
1201
1202         return NDR_ERR_SUCCESS;
1203 }
1204
1205
1206 NTSTATUS smbcli_rap_netprintdestenum(struct smbcli_tree *tree,
1207                                      TALLOC_CTX *mem_ctx,
1208                                      struct rap_NetPrintDestEnum *r)
1209 {
1210         struct rap_call *call;
1211         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1212
1213         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintDestEnum))) {
1214                 return NT_STATUS_NO_MEMORY;
1215         }
1216
1217         rap_cli_push_word(call, r->in.level);
1218         rap_cli_push_rcvbuf(call, r->in.bufsize);
1219         rap_cli_expect_multiple_entries(call);
1220
1221         switch(r->in.level) {
1222         case 0:
1223                 rap_cli_expect_format(call, "B9");
1224                 break;
1225         case 1:
1226                 rap_cli_expect_format(call, "B9B21WWzW");
1227                 break;
1228         case 2:
1229                 rap_cli_expect_format(call, "z");
1230                 break;
1231         case 3:
1232                 rap_cli_expect_format(call, "zzzWWzzzWW");
1233                 break;
1234         default:
1235                 result = NT_STATUS_INVALID_PARAMETER;
1236                 goto done;
1237         }
1238
1239         if (DEBUGLEVEL >= 10) {
1240                 NDR_PRINT_IN_DEBUG(rap_NetPrintDestEnum, r);
1241         }
1242
1243         result = rap_cli_do_call(tree, call);
1244
1245         if (!NT_STATUS_IS_OK(result))
1246                 goto done;
1247
1248         result = NT_STATUS_INVALID_PARAMETER;
1249
1250         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
1251         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
1252         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
1253         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
1254
1255         call->ndr_pull_data->relative_rap_convert = r->out.convert;
1256
1257         NDR_GOTO(ndr_pull_rap_NetPrintDestEnum_data(call->ndr_pull_data, r));
1258
1259         r->out.info = talloc_steal(mem_ctx, r->out.info);
1260
1261         if (DEBUGLEVEL >= 10) {
1262                 NDR_PRINT_OUT_DEBUG(rap_NetPrintDestEnum, r);
1263         }
1264
1265         result = NT_STATUS_OK;
1266
1267  done:
1268         talloc_free(call);
1269         return result;
1270 }
1271
1272 NTSTATUS smbcli_rap_netprintdestgetinfo(struct smbcli_tree *tree,
1273                                         TALLOC_CTX *mem_ctx,
1274                                         struct rap_NetPrintDestGetInfo *r)
1275 {
1276         struct rap_call *call;
1277         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1278
1279         if (!(call = new_rap_cli_call(mem_ctx, RAP_WPrintDestGetInfo))) {
1280                 return NT_STATUS_NO_MEMORY;
1281         }
1282
1283         rap_cli_push_string(call, r->in.PrintDestName);
1284         rap_cli_push_word(call, r->in.level);
1285         rap_cli_push_rcvbuf(call, r->in.bufsize);
1286         rap_cli_expect_word(call);
1287
1288         switch(r->in.level) {
1289         case 0:
1290                 rap_cli_expect_format(call, "B9");
1291                 break;
1292         case 1:
1293                 rap_cli_expect_format(call, "B9B21WWzW");
1294                 break;
1295         case 2:
1296                 rap_cli_expect_format(call, "z");
1297                 break;
1298         case 3:
1299                 rap_cli_expect_format(call, "zzzWWzzzWW");
1300                 break;
1301         default:
1302                 result = NT_STATUS_INVALID_PARAMETER;
1303                 goto done;
1304         }
1305
1306         if (DEBUGLEVEL >= 10) {
1307                 NDR_PRINT_IN_DEBUG(rap_NetPrintDestGetInfo, r);
1308         }
1309
1310         result = rap_cli_do_call(tree, call);
1311
1312         if (!NT_STATUS_IS_OK(result))
1313                 goto done;
1314
1315         result = NT_STATUS_INVALID_PARAMETER;
1316
1317         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
1318         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
1319         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
1320
1321         call->ndr_pull_data->relative_rap_convert = r->out.convert;
1322
1323         NDR_GOTO(ndr_pull_set_switch_value(call->ndr_pull_data, &r->out.info, r->in.level));
1324         NDR_GOTO(ndr_pull_rap_printdest_info(call->ndr_pull_data, NDR_SCALARS|NDR_BUFFERS, &r->out.info));
1325
1326         if (DEBUGLEVEL >= 10) {
1327                 NDR_PRINT_OUT_DEBUG(rap_NetPrintDestGetInfo, r);
1328         }
1329
1330         result = NT_STATUS_OK;
1331
1332  done:
1333         talloc_free(call);
1334         return result;
1335 }
1336
1337 NTSTATUS smbcli_rap_netuserpasswordset2(struct smbcli_tree *tree,
1338                                         struct smb_iconv_convenience *iconv_convenience,
1339                                         TALLOC_CTX *mem_ctx,
1340                                         struct rap_NetUserPasswordSet2 *r)
1341 {
1342         struct rap_call *call;
1343         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1344
1345         if (!(call = new_rap_cli_call(mem_ctx, RAP_WUserPasswordSet2))) {
1346                 return NT_STATUS_NO_MEMORY;
1347         }
1348
1349         rap_cli_push_string(call, r->in.UserName);
1350         rap_cli_push_paramdesc(call, 'b');
1351         rap_cli_push_paramdesc(call, '1');
1352         rap_cli_push_paramdesc(call, '6');
1353         ndr_push_array_uint8(call->ndr_push_param, NDR_SCALARS, r->in.OldPassword, 16);
1354         rap_cli_push_paramdesc(call, 'b');
1355         rap_cli_push_paramdesc(call, '1');
1356         rap_cli_push_paramdesc(call, '6');
1357         ndr_push_array_uint8(call->ndr_push_param, NDR_SCALARS, r->in.NewPassword, 16);
1358         rap_cli_push_word(call, r->in.EncryptedPassword);
1359         rap_cli_push_word(call, r->in.RealPasswordLength);
1360
1361         rap_cli_expect_format(call, "");
1362
1363         if (DEBUGLEVEL >= 10) {
1364                 NDR_PRINT_IN_DEBUG(rap_NetUserPasswordSet2, r);
1365         }
1366
1367         result = rap_cli_do_call(tree, call);
1368
1369         if (!NT_STATUS_IS_OK(result))
1370                 goto done;
1371
1372         result = NT_STATUS_INVALID_PARAMETER;
1373
1374         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
1375         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
1376
1377         result = NT_STATUS_OK;
1378
1379         if (!NT_STATUS_IS_OK(result)) {
1380                 goto done;
1381         }
1382
1383         if (DEBUGLEVEL >= 10) {
1384                 NDR_PRINT_OUT_DEBUG(rap_NetUserPasswordSet2, r);
1385         }
1386
1387  done:
1388         talloc_free(call);
1389         return result;
1390 }
1391
1392 NTSTATUS smbcli_rap_netoemchangepassword(struct smbcli_tree *tree,
1393                                          struct smb_iconv_convenience *iconv_convenience,
1394                                          TALLOC_CTX *mem_ctx,
1395                                          struct rap_NetOEMChangePassword *r)
1396 {
1397         struct rap_call *call;
1398         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1399
1400         if (!(call = new_rap_cli_call(mem_ctx, RAP_SamOEMChgPasswordUser2_P))) {
1401                 return NT_STATUS_NO_MEMORY;
1402         }
1403
1404         rap_cli_push_string(call, r->in.UserName);
1405         rap_cli_push_sendbuf(call, 532);
1406         ndr_push_array_uint8(call->ndr_push_data, NDR_SCALARS, r->in.crypt_password, 516);
1407         ndr_push_array_uint8(call->ndr_push_data, NDR_SCALARS, r->in.password_hash, 16);
1408
1409         rap_cli_expect_format(call, "B516B16");
1410
1411         if (DEBUGLEVEL >= 10) {
1412                 NDR_PRINT_IN_DEBUG(rap_NetOEMChangePassword, r);
1413         }
1414
1415         result = rap_cli_do_call(tree, call);
1416
1417         if (!NT_STATUS_IS_OK(result))
1418                 goto done;
1419
1420         result = NT_STATUS_INVALID_PARAMETER;
1421
1422         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
1423         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
1424
1425         result = NT_STATUS_OK;
1426
1427         if (!NT_STATUS_IS_OK(result)) {
1428                 goto done;
1429         }
1430
1431         if (DEBUGLEVEL >= 10) {
1432                 NDR_PRINT_OUT_DEBUG(rap_NetOEMChangePassword, r);
1433         }
1434
1435  done:
1436         talloc_free(call);
1437         return result;
1438 }
1439
1440 NTSTATUS smbcli_rap_netusergetinfo(struct smbcli_tree *tree,
1441                                    TALLOC_CTX *mem_ctx,
1442                                    struct rap_NetUserGetInfo *r)
1443 {
1444         struct rap_call *call;
1445         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
1446
1447         if (!(call = new_rap_cli_call(mem_ctx, RAP_WUserGetInfo))) {
1448                 return NT_STATUS_NO_MEMORY;
1449         }
1450
1451         rap_cli_push_string(call, r->in.UserName);
1452         rap_cli_push_word(call, r->in.level);
1453         rap_cli_push_rcvbuf(call, r->in.bufsize);
1454         rap_cli_expect_word(call);
1455
1456         switch(r->in.level) {
1457         case 0:
1458                 rap_cli_expect_format(call, "B21");
1459                 break;
1460         case 1:
1461                 rap_cli_expect_format(call, "B21BB16DWzzWz");
1462                 break;
1463         case 2:
1464                 rap_cli_expect_format(call, "B21BB16DWzzWzDzzzzDDDDWb21WWzWW");
1465                 break;
1466         case 10:
1467                 rap_cli_expect_format(call, "B21Bzzz");
1468                 break;
1469         case 11:
1470                 rap_cli_expect_format(call, "B21BzzzWDDzzDDWWzWzDWb21W");
1471                 break;
1472         default:
1473                 result = NT_STATUS_INVALID_PARAMETER;
1474                 goto done;
1475         }
1476
1477         if (DEBUGLEVEL >= 10) {
1478                 NDR_PRINT_IN_DEBUG(rap_NetUserGetInfo, r);
1479         }
1480
1481         result = rap_cli_do_call(tree, call);
1482
1483         if (!NT_STATUS_IS_OK(result))
1484                 goto done;
1485
1486         NDR_GOTO(ndr_pull_rap_status(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
1487         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
1488         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
1489
1490         call->ndr_pull_data->relative_rap_convert = r->out.convert;
1491
1492         NDR_GOTO(ndr_pull_set_switch_value(call->ndr_pull_data, &r->out.info, r->in.level));
1493         NDR_GOTO(ndr_pull_rap_netuser_info(call->ndr_pull_data, NDR_SCALARS|NDR_BUFFERS, &r->out.info));
1494
1495         if (DEBUGLEVEL >= 10) {
1496                 NDR_PRINT_OUT_DEBUG(rap_NetUserGetInfo, r);
1497         }
1498
1499         result = NT_STATUS_OK;
1500
1501  done:
1502         talloc_free(call);
1503         return result;
1504 }
1505
1506
1507 static bool test_netservergetinfo(struct torture_context *tctx,
1508                                   struct smbcli_state *cli)
1509 {
1510         struct rap_WserverGetInfo r;
1511         bool res = true;
1512
1513         r.in.bufsize = 0xffff;
1514
1515         r.in.level = 0;
1516         torture_assert_ntstatus_ok(tctx,
1517                 smbcli_rap_netservergetinfo(cli->tree, tctx, &r),
1518                 "rap_netservergetinfo level 0 failed");
1519         torture_assert_werr_ok(tctx, W_ERROR(r.out.status),
1520                 "rap_netservergetinfo level 0 failed");
1521
1522         r.in.level = 1;
1523         torture_assert_ntstatus_ok(tctx,
1524                 smbcli_rap_netservergetinfo(cli->tree, tctx, &r),
1525                 "rap_netservergetinfo level 1 failed");
1526         torture_assert_werr_ok(tctx, W_ERROR(r.out.status),
1527                 "rap_netservergetinfo level 1 failed");
1528
1529         return res;
1530 }
1531
1532 bool torture_rap_scan(struct torture_context *torture, struct smbcli_state *cli)
1533 {
1534         int callno;
1535
1536         for (callno = 0; callno < 0xffff; callno++) {
1537                 struct rap_call *call = new_rap_cli_call(torture, callno);
1538                 NTSTATUS result;
1539
1540                 result = rap_cli_do_call(cli->tree, call);
1541
1542                 if (!NT_STATUS_EQUAL(result, NT_STATUS_INVALID_PARAMETER))
1543                         continue;
1544
1545                 printf("callno %d is RAP call\n", callno);
1546         }
1547
1548         return true;
1549 }
1550
1551 NTSTATUS torture_rap_init(void)
1552 {
1553         struct torture_suite *suite = torture_suite_create(talloc_autofree_context(), "rap");
1554         struct torture_suite *suite_basic = torture_suite_create(suite, "basic");
1555
1556         torture_suite_add_suite(suite, suite_basic);
1557         torture_suite_add_suite(suite, torture_rap_rpc(suite));
1558         torture_suite_add_suite(suite, torture_rap_printing(suite));
1559         torture_suite_add_suite(suite, torture_rap_sam(suite));
1560
1561         torture_suite_add_1smb_test(suite_basic, "netserverenum",
1562                                     test_netserverenum);
1563         torture_suite_add_1smb_test(suite_basic, "netshareenum",
1564                                     test_netshareenum);
1565         torture_suite_add_1smb_test(suite_basic, "netservergetinfo",
1566                                     test_netservergetinfo);
1567
1568         torture_suite_add_1smb_test(suite, "scan", torture_rap_scan);
1569
1570         suite->description = talloc_strdup(suite,
1571                                                 "Remote Administration Protocol tests");
1572
1573         torture_register_suite(suite);
1574
1575         return NT_STATUS_OK;
1576 }