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
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 3 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #include "includes.h"
  23 #include "libcli/libcli.h"
  24 #include "torture/smbtorture.h"
  25 #include "torture/util.h"
  26 #include "../librpc/gen_ndr/ndr_rap.h"
  27 #include "librpc/ndr/libndr.h"
  28 #include "param/param.h"
  29 #include "torture/rap/proto.h"
  30
  31 #define RAP_GOTO(call) do { \
  32         NTSTATUS _status; \
  33         _status = call; \
  34         if (!NT_STATUS_IS_OK(_status)) { \
  35                 result = _status; \
  36                 goto done; \
  37         } \
  38 } while (0)
  39
  40 #define RAP_RETURN(call) do { \
  41         NTSTATUS _status; \
  42         _status = call; \
  43         if (!NT_STATUS_IS_OK(_status)) { \
  44                 return _status; \
  45         } \
  46 } while (0)
  47
  48
  49 #define NDR_GOTO(call) do { \
  50         enum ndr_err_code _ndr_err; \
  51         _ndr_err = call; \
  52         if (!NDR_ERR_CODE_IS_SUCCESS(_ndr_err)) { \
  53                 result = ndr_map_error2ntstatus(_ndr_err); \
  54                 goto done; \
  55         } \
  56 } while (0)
  57
  58 #define NDR_RETURN(call) do { \
  59         enum ndr_err_code _ndr_err; \
  60         _ndr_err = call; \
  61         if (!NDR_ERR_CODE_IS_SUCCESS(_ndr_err)) { \
  62                 return ndr_map_error2ntstatus(_ndr_err); \
  63         } \
  64 } while (0)
  65
  66 struct rap_call {
  67         uint16_t callno;
  68         char *paramdesc;
  69         const char *datadesc;
  70
  71         uint16_t status;
  72         uint16_t convert;
  73
  74         uint16_t rcv_paramlen, rcv_datalen;
  75
  76         struct ndr_push *ndr_push_param;
  77         struct ndr_push *ndr_push_data;
  78         struct ndr_pull *ndr_pull_param;
  79         struct ndr_pull *ndr_pull_data;
  80 };
  81
  82 #define RAPNDR_FLAGS (LIBNDR_FLAG_NOALIGN|LIBNDR_FLAG_STR_ASCII|LIBNDR_FLAG_STR_NULLTERM);
  83
  84 static struct rap_call *new_rap_cli_call(TALLOC_CTX *mem_ctx, struct smb_iconv_convenience *iconv_convenience, uint16_t callno)
  85 {
  86         struct rap_call *call;
  87
  88         call = talloc(mem_ctx, struct rap_call);
  89
  90         if (call == NULL)
  91                 return NULL;
  92
  93         call->callno = callno;
  94         call->rcv_paramlen = 4;
  95
  96         call->paramdesc = NULL;
  97         call->datadesc = NULL;
  98
  99         call->ndr_push_param = ndr_push_init_ctx(mem_ctx, iconv_convenience);
 100         call->ndr_push_param->flags = RAPNDR_FLAGS;
 101
 102         call->ndr_push_data = ndr_push_init_ctx(mem_ctx, iconv_convenience);
 103         call->ndr_push_data->flags = RAPNDR_FLAGS;
 104
 105         return call;
 106 }
 107
 108 static void rap_cli_push_paramdesc(struct rap_call *call, char desc)
 109 {
 110         int len = 0;
 111
 112         if (call->paramdesc != NULL)
 113                 len = strlen(call->paramdesc);
 114
 115         call->paramdesc = talloc_realloc(call,
 116                                          call->paramdesc,
 117                                          char,
 118                                          len+2);
 119
 120         call->paramdesc[len] = desc;
 121         call->paramdesc[len+1] = '\0';
 122 }
 123
 124 static void rap_cli_push_word(struct rap_call *call, uint16_t val)
 125 {
 126         rap_cli_push_paramdesc(call, 'W');
 127         ndr_push_uint16(call->ndr_push_param, NDR_SCALARS, val);
 128 }
 129
 130 static void rap_cli_push_dword(struct rap_call *call, uint32_t val)
 131 {
 132         rap_cli_push_paramdesc(call, 'D');
 133         ndr_push_uint32(call->ndr_push_param, NDR_SCALARS, val);
 134 }
 135
 136 static void rap_cli_push_rcvbuf(struct rap_call *call, int len)
 137 {
 138         rap_cli_push_paramdesc(call, 'r');
 139         rap_cli_push_paramdesc(call, 'L');
 140         ndr_push_uint16(call->ndr_push_param, NDR_SCALARS, len);
 141         call->rcv_datalen = len;
 142 }
 143
 144 static void rap_cli_expect_multiple_entries(struct rap_call *call)
 145 {
 146         rap_cli_push_paramdesc(call, 'e');
 147         rap_cli_push_paramdesc(call, 'h');
 148         call->rcv_paramlen += 4; /* uint16_t entry count, uint16_t total */
 149 }
 150
 151 static void rap_cli_expect_word(struct rap_call *call)
 152 {
 153         rap_cli_push_paramdesc(call, 'h');
 154         call->rcv_paramlen += 2;
 155 }
 156
 157 static void rap_cli_push_string(struct rap_call *call, const char *str)
 158 {
 159         if (str == NULL) {
 160                 rap_cli_push_paramdesc(call, 'O');
 161                 return;
 162         }
 163         rap_cli_push_paramdesc(call, 'z');
 164         ndr_push_string(call->ndr_push_param, NDR_SCALARS, str);
 165 }
 166
 167 static void rap_cli_expect_format(struct rap_call *call, const char *format)
 168 {
 169         call->datadesc = format;
 170 }
 171
 172 static NTSTATUS rap_pull_string(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr,
 173                                 uint16_t convert, const char **dest)
 174 {
 175         uint16_t string_offset;
 176         uint16_t ignore;
 177         const char *p;
 178         size_t len;
 179
 180         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &string_offset));
 181         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &ignore));
 182
 183         string_offset -= convert;
 184
 185         if (string_offset+1 > ndr->data_size)
 186                 return NT_STATUS_INVALID_PARAMETER;
 187
 188         p = (const char *)(ndr->data + string_offset);
 189         len = strnlen(p, ndr->data_size-string_offset);
 190
 191         if ( string_offset + len + 1 >  ndr->data_size )
 192                 return NT_STATUS_INVALID_PARAMETER;
 193
 194         *dest = talloc_zero_array(mem_ctx, char, len+1);
 195         pull_string((char *)*dest, p, len+1, len, STR_ASCII);
 196
 197         return NT_STATUS_OK;
 198 }
 199
 200 static NTSTATUS rap_cli_do_call(struct smbcli_tree *tree,
 201                                 struct smb_iconv_convenience *iconv_convenience,
 202                                 struct rap_call *call)
 203 {
 204         NTSTATUS result;
 205         DATA_BLOB param_blob;
 206         struct ndr_push *params;
 207         struct smb_trans2 trans;
 208
 209         params = ndr_push_init_ctx(call, iconv_convenience);
 210
 211         if (params == NULL)
 212                 return NT_STATUS_NO_MEMORY;
 213
 214         params->flags = RAPNDR_FLAGS;
 215
 216         trans.in.max_param = call->rcv_paramlen;
 217         trans.in.max_data = call->rcv_datalen;
 218         trans.in.max_setup = 0;
 219         trans.in.flags = 0;
 220         trans.in.timeout = 0;
 221         trans.in.setup_count = 0;
 222         trans.in.setup = NULL;
 223         trans.in.trans_name = "\\PIPE\\LANMAN";
 224
 225         NDR_RETURN(ndr_push_uint16(params, NDR_SCALARS, call->callno));
 226         if (call->paramdesc)
 227                 NDR_RETURN(ndr_push_string(params, NDR_SCALARS, call->paramdesc));
 228         if (call->datadesc)
 229                 NDR_RETURN(ndr_push_string(params, NDR_SCALARS, call->datadesc));
 230
 231         param_blob = ndr_push_blob(call->ndr_push_param);
 232         NDR_RETURN(ndr_push_bytes(params, param_blob.data,
 233                                  param_blob.length));
 234
 235         trans.in.params = ndr_push_blob(params);
 236         trans.in.data = data_blob(NULL, 0);
 237
 238         result = smb_raw_trans(tree, call, &trans);
 239
 240         if (!NT_STATUS_IS_OK(result))
 241                 return result;
 242
 243         call->ndr_pull_param = ndr_pull_init_blob(&trans.out.params, call,
 244                                                   iconv_convenience);
 245         call->ndr_pull_param->flags = RAPNDR_FLAGS;
 246
 247         call->ndr_pull_data = ndr_pull_init_blob(&trans.out.data, call,
 248                                                  iconv_convenience);
 249         call->ndr_pull_data->flags = RAPNDR_FLAGS;
 250
 251         return result;
 252 }
 253
 254
 255 static NTSTATUS smbcli_rap_netshareenum(struct smbcli_tree *tree,
 256                                         struct smb_iconv_convenience *iconv_convenience,
 257                                         TALLOC_CTX *mem_ctx,
 258                                         struct rap_NetShareEnum *r)
 259 {
 260         struct rap_call *call;
 261         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 262         int i;
 263
 264         call = new_rap_cli_call(tree, iconv_convenience, RAP_WshareEnum);
 265
 266         if (call == NULL)
 267                 return NT_STATUS_NO_MEMORY;
 268
 269         rap_cli_push_word(call, r->in.level); /* Level */
 270         rap_cli_push_rcvbuf(call, r->in.bufsize);
 271         rap_cli_expect_multiple_entries(call);
 272
 273         switch(r->in.level) {
 274         case 0:
 275                 rap_cli_expect_format(call, "B13");
 276                 break;
 277         case 1:
 278                 rap_cli_expect_format(call, "B13BWz");
 279                 break;
 280         }
 281
 282         if (DEBUGLEVEL >= 10) {
 283                 NDR_PRINT_IN_DEBUG(rap_NetShareEnum, r);
 284         }
 285
 286         result = rap_cli_do_call(tree, iconv_convenience, call);
 287
 288         if (!NT_STATUS_IS_OK(result))
 289                 goto done;
 290
 291         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 292         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 293         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
 294         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
 295
 296         r->out.info = talloc_array(mem_ctx, union rap_share_info, r->out.count);
 297
 298         if (r->out.info == NULL) {
 299                 result = NT_STATUS_NO_MEMORY;
 300                 goto done;
 301         }
 302
 303         for (i=0; i<r->out.count; i++) {
 304                 switch(r->in.level) {
 305                 case 0:
 306                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 307                                                 r->out.info[i].info0.share_name, 13));
 308                         break;
 309                 case 1:
 310                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 311                                                 r->out.info[i].info1.share_name, 13));
 312                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 313                                                 &r->out.info[i].info1.reserved1, 1));
 314                         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_data,
 315                                                NDR_SCALARS, &r->out.info[i].info1.share_type));
 316                         RAP_GOTO(rap_pull_string(mem_ctx, call->ndr_pull_data,
 317                                                r->out.convert,
 318                                                &r->out.info[i].info1.comment));
 319                         break;
 320                 }
 321         }
 322
 323         if (DEBUGLEVEL >= 10) {
 324                 NDR_PRINT_OUT_DEBUG(rap_NetShareEnum, r);
 325         }
 326         result = NT_STATUS_OK;
 327
 328  done:
 329         talloc_free(call);
 330         return result;
 331 }
 332
 333 static bool test_netshareenum(struct torture_context *tctx,
 334                               struct smbcli_state *cli)
 335 {
 336         struct rap_NetShareEnum r;
 337         int i;
 338
 339         r.in.level = 1;
 340         r.in.bufsize = 8192;
 341
 342         torture_assert_ntstatus_ok(tctx,
 343                 smbcli_rap_netshareenum(cli->tree, lp_iconv_convenience(tctx->lp_ctx), tctx, &r), "");
 344
 345         for (i=0; i<r.out.count; i++) {
 346                 printf("%s %d %s\n", r.out.info[i].info1.share_name,
 347                        r.out.info[i].info1.share_type,
 348                        r.out.info[i].info1.comment);
 349         }
 350
 351         return true;
 352 }
 353
 354 static NTSTATUS smbcli_rap_netserverenum2(struct smbcli_tree *tree,
 355                                           struct smb_iconv_convenience *iconv_convenience,
 356                                           TALLOC_CTX *mem_ctx,
 357                                           struct rap_NetServerEnum2 *r)
 358 {
 359         struct rap_call *call;
 360         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 361         int i;
 362
 363         call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_NetServerEnum2);
 364
 365         if (call == NULL)
 366                 return NT_STATUS_NO_MEMORY;
 367
 368         rap_cli_push_word(call, r->in.level);
 369         rap_cli_push_rcvbuf(call, r->in.bufsize);
 370         rap_cli_expect_multiple_entries(call);
 371         rap_cli_push_dword(call, r->in.servertype);
 372         rap_cli_push_string(call, r->in.domain);
 373
 374         switch(r->in.level) {
 375         case 0:
 376                 rap_cli_expect_format(call, "B16");
 377                 break;
 378         case 1:
 379                 rap_cli_expect_format(call, "B16BBDz");
 380                 break;
 381         }
 382
 383         if (DEBUGLEVEL >= 10) {
 384                 NDR_PRINT_IN_DEBUG(rap_NetServerEnum2, r);
 385         }
 386
 387         result = rap_cli_do_call(tree, iconv_convenience, call);
 388
 389         if (!NT_STATUS_IS_OK(result))
 390                 goto done;
 391
 392         result = NT_STATUS_INVALID_PARAMETER;
 393
 394         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 395         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 396         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
 397         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
 398
 399         r->out.info = talloc_array(mem_ctx, union rap_server_info, r->out.count);
 400
 401         if (r->out.info == NULL) {
 402                 result = NT_STATUS_NO_MEMORY;
 403                 goto done;
 404         }
 405
 406         for (i=0; i<r->out.count; i++) {
 407                 switch(r->in.level) {
 408                 case 0:
 409                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 410                                                 r->out.info[i].info0.name, 16));
 411                         break;
 412                 case 1:
 413                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 414                                                 r->out.info[i].info1.name, 16));
 415                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 416                                               &r->out.info[i].info1.version_major, 1));
 417                         NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 418                                               &r->out.info[i].info1.version_minor, 1));
 419                         NDR_GOTO(ndr_pull_uint32(call->ndr_pull_data,
 420                                                NDR_SCALARS, &r->out.info[i].info1.servertype));
 421                         RAP_GOTO(rap_pull_string(mem_ctx, call->ndr_pull_data,
 422                                                r->out.convert,
 423                                                &r->out.info[i].info1.comment));
 424                 }
 425         }
 426
 427         if (DEBUGLEVEL >= 10) {
 428                 NDR_PRINT_OUT_DEBUG(rap_NetServerEnum2, r);
 429         }
 430
 431         result = NT_STATUS_OK;
 432
 433  done:
 434         talloc_free(call);
 435         return result;
 436 }
 437
 438 static bool test_netserverenum(struct torture_context *tctx,
 439                                struct smbcli_state *cli)
 440 {
 441         struct rap_NetServerEnum2 r;
 442         int i;
 443
 444         r.in.level = 0;
 445         r.in.bufsize = 8192;
 446         r.in.servertype = 0xffffffff;
 447         r.in.servertype = 0x80000000;
 448         r.in.domain = NULL;
 449
 450         torture_assert_ntstatus_ok(tctx,
 451                    smbcli_rap_netserverenum2(cli->tree, lp_iconv_convenience(tctx->lp_ctx), tctx, &r), "");
 452
 453         for (i=0; i<r.out.count; i++) {
 454                 switch (r.in.level) {
 455                 case 0:
 456                         printf("%s\n", r.out.info[i].info0.name);
 457                         break;
 458                 case 1:
 459                         printf("%s %x %s\n", r.out.info[i].info1.name,
 460                                r.out.info[i].info1.servertype,
 461                                r.out.info[i].info1.comment);
 462                         break;
 463                 }
 464         }
 465
 466         return true;
 467 }
 468
 469 NTSTATUS smbcli_rap_netservergetinfo(struct smbcli_tree *tree,
 470                                               struct smb_iconv_convenience *iconv_convenience,
 471                                      TALLOC_CTX *mem_ctx,
 472                                      struct rap_WserverGetInfo *r)
 473 {
 474         struct rap_call *call;
 475         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 476
 477         if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WserverGetInfo))) {
 478                 return NT_STATUS_NO_MEMORY;
 479         }
 480
 481         rap_cli_push_word(call, r->in.level);
 482         rap_cli_push_rcvbuf(call, r->in.bufsize);
 483         rap_cli_expect_word(call);
 484
 485         switch(r->in.level) {
 486         case 0:
 487                 rap_cli_expect_format(call, "B16");
 488                 break;
 489         case 1:
 490                 rap_cli_expect_format(call, "B16BBDz");
 491                 break;
 492         default:
 493                 result = NT_STATUS_INVALID_PARAMETER;
 494                 goto done;
 495         }
 496
 497         if (DEBUGLEVEL >= 10) {
 498                 NDR_PRINT_IN_DEBUG(rap_WserverGetInfo, r);
 499         }
 500
 501         result = rap_cli_do_call(tree, iconv_convenience, call);
 502
 503         if (!NT_STATUS_IS_OK(result))
 504                 goto done;
 505
 506         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 507         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 508         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
 509
 510         switch(r->in.level) {
 511         case 0:
 512                 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 513                                         r->out.info.info0.name, 16));
 514                 break;
 515         case 1:
 516                 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 517                                         r->out.info.info1.name, 16));
 518                 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 519                                       &r->out.info.info1.version_major, 1));
 520                 NDR_GOTO(ndr_pull_bytes(call->ndr_pull_data,
 521                                       &r->out.info.info1.version_minor, 1));
 522                 NDR_GOTO(ndr_pull_uint32(call->ndr_pull_data,
 523                                        NDR_SCALARS, &r->out.info.info1.servertype));
 524                 RAP_GOTO(rap_pull_string(mem_ctx, call->ndr_pull_data,
 525                                        r->out.convert,
 526                                        &r->out.info.info1.comment));
 527         }
 528
 529         if (DEBUGLEVEL >= 10) {
 530                 NDR_PRINT_OUT_DEBUG(rap_WserverGetInfo, r);
 531         }
 532  done:
 533         talloc_free(call);
 534         return result;
 535 }
 536
 537 static NTSTATUS rap_pull_rap_PrintQueue0(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue0 *r)
 538 {
 539         NDR_RETURN(ndr_pull_bytes(ndr, r->PrintQName, 13));
 540         return NT_STATUS_OK;
 541 }
 542
 543 static NTSTATUS rap_pull_rap_PrintQueue1(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue1 *r)
 544 {
 545         NDR_RETURN(ndr_pull_bytes(ndr, r->PrintQName, 13));
 546         NDR_RETURN(ndr_pull_uint8(ndr, NDR_SCALARS, &r->Pad1));
 547         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->Priority));
 548         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->StartTime));
 549         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->UntilTime));
 550         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->SeparatorPageFilename));
 551         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintProcessorDllName));
 552         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintDestinationsName));
 553         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintParameterString));
 554         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->CommentString));
 555         NDR_RETURN(ndr_pull_rap_PrintQStatusCode(ndr, NDR_SCALARS, &r->PrintQStatus));
 556         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->PrintJobCount));
 557
 558         return NT_STATUS_OK;
 559 }
 560
 561 static NTSTATUS rap_pull_rap_PrintQueue3(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue3 *r)
 562 {
 563         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintQueueName));
 564         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->Priority));
 565         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->StartTime));
 566         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->UntilTime));
 567         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->Pad));
 568         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->SeparatorPageFilename));
 569         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintProcessorDllName));
 570         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintParameterString));
 571         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->CommentString));
 572         NDR_RETURN(ndr_pull_rap_PrintQStatusCode(ndr, NDR_SCALARS, &r->PrintQStatus));
 573         NDR_RETURN(ndr_pull_uint16(ndr, NDR_SCALARS, &r->PrintJobCount));
 574         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->Printers));
 575         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->DriverName));
 576         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintDriverData));
 577
 578         return NT_STATUS_OK;
 579 }
 580
 581 static NTSTATUS rap_pull_rap_PrintQueue5(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr, uint16_t convert, struct rap_PrintQueue5 *r)
 582 {
 583         RAP_RETURN(rap_pull_string(mem_ctx, ndr, convert, &r->PrintQueueName));
 584
 585         return NT_STATUS_OK;
 586 }
 587
 588 NTSTATUS smbcli_rap_netprintqenum(struct smbcli_tree *tree,
 589                                   struct smb_iconv_convenience *iconv_convenience,
 590                                   TALLOC_CTX *mem_ctx,
 591                                   struct rap_NetPrintQEnum *r)
 592 {
 593         struct rap_call *call;
 594         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 595         int i;
 596
 597         if (!(call = new_rap_cli_call(mem_ctx, iconv_convenience, RAP_WPrintQEnum))) {
 598                 return NT_STATUS_NO_MEMORY;
 599         }
 600
 601         rap_cli_push_word(call, r->in.level);
 602         rap_cli_push_rcvbuf(call, r->in.bufsize);
 603         rap_cli_expect_multiple_entries(call);
 604
 605         switch(r->in.level) {
 606         case 0:
 607                 rap_cli_expect_format(call, "B13");
 608                 break;
 609         case 1:
 610                 rap_cli_expect_format(call, "B13BWWWzzzzzWW");
 611                 break;
 612         case 2:
 613                 rap_cli_expect_format(call, "B13BWWWzzzzzWN");
 614                 break;
 615         case 3:
 616                 rap_cli_expect_format(call, "zWWWWzzzzWWzzl");
 617                 break;
 618         case 4:
 619                 rap_cli_expect_format(call, "zWWWWzzzzWNzzl");
 620                 break;
 621         case 5:
 622                 rap_cli_expect_format(call, "z");
 623                 break;
 624         default:
 625                 result = NT_STATUS_INVALID_PARAMETER;
 626                 goto done;
 627         }
 628
 629         if (DEBUGLEVEL >= 10) {
 630                 NDR_PRINT_IN_DEBUG(rap_NetPrintQEnum, r);
 631         }
 632
 633         result = rap_cli_do_call(tree, iconv_convenience, call);
 634
 635         if (!NT_STATUS_IS_OK(result))
 636                 goto done;
 637
 638         result = NT_STATUS_INVALID_PARAMETER;
 639
 640         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
 641         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
 642         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
 643         NDR_GOTO(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
 644
 645         r->out.info = talloc_zero_array(mem_ctx, union rap_printq_info, r->out.count);
 646
 647         if (r->out.info == NULL) {
 648                 result = NT_STATUS_NO_MEMORY;
 649                 goto done;
 650         }
 651
 652         for (i=0; i<r->out.count; i++) {
 653                 switch(r->in.level) {
 654                 case 0:
 655                         result = rap_pull_rap_PrintQueue0(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info0);
 656                         break;
 657                 case 1:
 658                         result = rap_pull_rap_PrintQueue1(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info1);
 659                         break;
 660                 case 3:
 661                         result = rap_pull_rap_PrintQueue3(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info3);
 662                         break;
 663                 case 5:
 664                         result = rap_pull_rap_PrintQueue5(mem_ctx, call->ndr_pull_data, r->out.convert, &r->out.info[i].info5);
 665                         break;
 666                 }
 667         }
 668
 669         if (!NT_STATUS_IS_OK(result)) {
 670                 goto done;
 671         }
 672
 673         if (DEBUGLEVEL >= 10) {
 674                 NDR_PRINT_OUT_DEBUG(rap_NetPrintQEnum, r);
 675         }
 676
 677         result = NT_STATUS_OK;
 678
 679  done:
 680         talloc_free(call);
 681         return result;
 682 }
 683
 684 static bool test_netservergetinfo(struct torture_context *tctx,
 685                                   struct smbcli_state *cli)
 686 {
 687         struct rap_WserverGetInfo r;
 688         bool res = true;
 689
 690         r.in.bufsize = 0xffff;
 691
 692         r.in.level = 0;
 693         torture_assert_ntstatus_ok(tctx,
 694                 smbcli_rap_netservergetinfo(cli->tree, lp_iconv_convenience(tctx->lp_ctx), tctx, &r),
 695                 "rap_netservergetinfo level 0 failed");
 696
 697         if (torture_setting_bool(tctx, "samba3", false)) {
 698                 torture_skip(tctx, "skipping netservergetinfo level 1 against samba3");
 699         }
 700
 701         r.in.level = 1;
 702         torture_assert_ntstatus_ok(tctx,
 703                 smbcli_rap_netservergetinfo(cli->tree, lp_iconv_convenience(tctx->lp_ctx), tctx, &r),
 704                 "rap_netservergetinfo level 1 failed");
 705
 706         return res;
 707 }
 708
 709 bool torture_rap_scan(struct torture_context *torture, struct smbcli_state *cli)
 710 {
 711         int callno;
 712
 713         for (callno = 0; callno < 0xffff; callno++) {
 714                 struct rap_call *call = new_rap_cli_call(torture, lp_iconv_convenience(torture->lp_ctx), callno);
 715                 NTSTATUS result;
 716
 717                 result = rap_cli_do_call(cli->tree, lp_iconv_convenience(torture->lp_ctx), call);
 718
 719                 if (!NT_STATUS_EQUAL(result, NT_STATUS_INVALID_PARAMETER))
 720                         continue;
 721
 722                 printf("callno %d is RAP call\n", callno);
 723         }
 724
 725         return true;
 726 }
 727
 728 NTSTATUS torture_rap_init(void)
 729 {
 730         struct torture_suite *suite = torture_suite_create(talloc_autofree_context(), "RAP");
 731         struct torture_suite *suite_basic = torture_suite_create(suite, "BASIC");
 732
 733         torture_suite_add_suite(suite, suite_basic);
 734         torture_suite_add_suite(suite, torture_rap_rpc(suite));
 735         torture_suite_add_suite(suite, torture_rap_printing(suite));
 736
 737         torture_suite_add_1smb_test(suite_basic, "netserverenum",
 738                                     test_netserverenum);
 739         torture_suite_add_1smb_test(suite_basic, "netshareenum",
 740                                     test_netshareenum);
 741         torture_suite_add_1smb_test(suite_basic, "netservergetinfo",
 742                                     test_netservergetinfo);
 743
 744         torture_suite_add_1smb_test(suite, "SCAN", torture_rap_scan);
 745
 746         suite->description = talloc_strdup(suite,
 747                                                 "Remote Administration Protocol tests");
 748
 749         torture_register_suite(suite);
 750
 751         return NT_STATUS_OK;
 752 }