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
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 2 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program; if not, write to the Free Software
  18    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19 */
  20
  21 #include "includes.h"
  22
  23 struct rap_call {
  24         TALLOC_CTX *mem_ctx;
  25         uint16 callno;
  26         char *paramdesc;
  27         const char *datadesc;
  28
  29         struct smb_trans2 trans;
  30
  31         uint16 status;
  32         uint16 convert;
  33
  34         struct ndr_push *ndr_push_param;
  35         struct ndr_push *ndr_push_data;
  36         struct ndr_pull *ndr_pull_param;
  37         struct ndr_pull *ndr_pull_data;
  38 };
  39
  40 #define RAPNDR_FLAGS (LIBNDR_FLAG_NOALIGN|LIBNDR_FLAG_STR_ASCII|LIBNDR_FLAG_STR_NULLTERM);
  41
  42 static struct rap_call *new_rap_cli_call(uint16 callno)
  43 {
  44         struct rap_call *call;
  45         TALLOC_CTX *mem_ctx = talloc_init("rap_call");
  46
  47         if (mem_ctx == NULL)
  48                 return NULL;
  49
  50         call = talloc_p(mem_ctx, struct rap_call);
  51
  52         if (call == NULL)
  53                 return NULL;
  54
  55         ZERO_STRUCTP(call);
  56
  57         call->callno = callno;
  58         call->trans.in.max_param = 4;   /* uint16 error, uint16 "convert" */
  59         call->mem_ctx = mem_ctx;
  60
  61         call->ndr_push_param = ndr_push_init_ctx(mem_ctx);
  62         call->ndr_push_param->flags = RAPNDR_FLAGS;
  63
  64         call->ndr_push_data = ndr_push_init_ctx(mem_ctx);
  65         call->ndr_push_data->flags = RAPNDR_FLAGS;
  66
  67         return call;
  68 }
  69
  70 static void destroy_rap_call(struct rap_call *call)
  71 {
  72         talloc_destroy(call->mem_ctx);
  73 }
  74
  75 static void rap_cli_push_paramdesc(struct rap_call *call, char desc)
  76 {
  77         int len = 0;
  78
  79         if (call->paramdesc != NULL)
  80                 len = strlen(call->paramdesc);
  81
  82         call->paramdesc = talloc_realloc(call->mem_ctx, call->paramdesc,
  83                                          len+2);
  84         call->paramdesc[len] = desc;
  85         call->paramdesc[len+1] = '\0';
  86 }
  87
  88 static void rap_cli_push_word(struct rap_call *call, uint16 val)
  89 {
  90         rap_cli_push_paramdesc(call, 'W');
  91         ndr_push_uint16(call->ndr_push_param, val);
  92 }
  93
  94 static void rap_cli_push_dword(struct rap_call *call, uint32 val)
  95 {
  96         rap_cli_push_paramdesc(call, 'D');
  97         ndr_push_uint32(call->ndr_push_param, val);
  98 }
  99
 100 static void rap_cli_push_rcvbuf(struct rap_call *call, int len)
 101 {
 102         rap_cli_push_paramdesc(call, 'r');
 103         rap_cli_push_paramdesc(call, 'L');
 104         ndr_push_uint16(call->ndr_push_param, len);
 105         call->trans.in.max_data = len;
 106 }
 107
 108 static void rap_cli_expect_multiple_entries(struct rap_call *call)
 109 {
 110         rap_cli_push_paramdesc(call, 'e');
 111         rap_cli_push_paramdesc(call, 'h');
 112         call->trans.in.max_param += 4;  /* uint16 entry count, uint16 total */
 113 }
 114
 115 static void rap_cli_push_string(struct rap_call *call, const char *str)
 116 {
 117         if (str == NULL) {
 118                 rap_cli_push_paramdesc(call, 'O');
 119                 return;
 120         }
 121         rap_cli_push_paramdesc(call, 'z');
 122         ndr_push_string(call->ndr_push_param, NDR_SCALARS, str);
 123 }
 124
 125 static void rap_cli_expect_format(struct rap_call *call, const char *format)
 126 {
 127         call->datadesc = format;
 128 }
 129
 130 static NTSTATUS rap_pull_string(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr,
 131                                 uint16 convert, char **dest)
 132 {
 133         uint16 string_offset;
 134         uint16 ignore;
 135         char *p;
 136         size_t len;
 137
 138         NDR_CHECK(ndr_pull_uint16(ndr, &string_offset));
 139         NDR_CHECK(ndr_pull_uint16(ndr, &ignore));
 140
 141         string_offset -= convert;
 142
 143         if (string_offset+1 > ndr->data_size)
 144                 return NT_STATUS_INVALID_PARAMETER;
 145
 146         p = ndr->data + string_offset;
 147         len = strnlen(p, ndr->data_size-string_offset);
 148
 149         if ( string_offset + len + 1 >  ndr->data_size )
 150                 return NT_STATUS_INVALID_PARAMETER;
 151
 152         *dest = talloc_zero(mem_ctx, len+1);
 153         pull_ascii(*dest, p, len+1, len, 0);
 154
 155         return NT_STATUS_OK;
 156 }
 157
 158 static NTSTATUS rap_cli_do_call(struct cli_state *cli, TALLOC_CTX *mem_ctx,
 159                                 struct rap_call *call)
 160 {
 161         NTSTATUS result;
 162         DATA_BLOB param_blob;
 163         struct ndr_push *params;
 164
 165         params = ndr_push_init_ctx(mem_ctx);
 166
 167         if (params == NULL)
 168                 return NT_STATUS_NO_MEMORY;
 169
 170         params->flags = RAPNDR_FLAGS;
 171
 172         call->trans.in.max_setup = 0;
 173         call->trans.in.flags = 0;
 174         call->trans.in.timeout = 0;
 175         call->trans.in.setup_count = 0;
 176         call->trans.in.setup = NULL;
 177         call->trans.in.trans_name = "\\PIPE\\LANMAN";
 178
 179         NDR_CHECK(ndr_push_uint16(params, call->callno));
 180         NDR_CHECK(ndr_push_string(params, NDR_SCALARS, call->paramdesc));
 181         NDR_CHECK(ndr_push_string(params, NDR_SCALARS, call->datadesc));
 182
 183         param_blob = ndr_push_blob(call->ndr_push_param);
 184         NDR_CHECK(ndr_push_bytes(params, param_blob.data,
 185                                  param_blob.length));
 186
 187         call->trans.in.params = ndr_push_blob(params);
 188         call->trans.in.data = data_blob(NULL, 0);
 189
 190         result = smb_raw_trans(cli->tree, call->mem_ctx, &call->trans);
 191
 192         if (!NT_STATUS_IS_OK(result))
 193                 return result;
 194
 195         call->ndr_pull_param = ndr_pull_init_blob(&call->trans.out.params,
 196                                                   call->mem_ctx);
 197         call->ndr_pull_param->flags = RAPNDR_FLAGS;
 198
 199         call->ndr_pull_data = ndr_pull_init_blob(&call->trans.out.data,
 200                                                  call->mem_ctx);
 201         call->ndr_pull_data->flags = RAPNDR_FLAGS;
 202
 203         return result;
 204 }
 205
 206 #define NDR_OK(call) do { NTSTATUS _status; \
 207                              _status = call; \
 208                              if (!NT_STATUS_IS_OK(_status)) \
 209                                 goto done; \
 210                         } while (0)
 211
 212 static NTSTATUS cli_rap_netshareenum(struct cli_state *cli,
 213                                      TALLOC_CTX *mem_ctx,
 214                                      struct rap_NetShareEnum *r)
 215 {
 216         struct rap_call *call;
 217         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 218         int i;
 219
 220         call = new_rap_cli_call(0);
 221
 222         if (call == NULL)
 223                 return NT_STATUS_NO_MEMORY;
 224
 225         rap_cli_push_word(call, r->in.level); /* Level */
 226         rap_cli_push_rcvbuf(call, r->in.bufsize);
 227         rap_cli_expect_multiple_entries(call);
 228
 229         switch(r->in.level) {
 230         case 0:
 231                 rap_cli_expect_format(call, "B13");
 232                 break;
 233         case 1:
 234                 rap_cli_expect_format(call, "B13BWz");
 235                 break;
 236         }
 237
 238         result = rap_cli_do_call(cli, mem_ctx, call);
 239
 240         if (!NT_STATUS_IS_OK(result))
 241                 goto done;
 242
 243         NDR_OK(ndr_pull_uint16(call->ndr_pull_param, &r->out.status));
 244         NDR_OK(ndr_pull_uint16(call->ndr_pull_param, &r->out.convert));
 245         NDR_OK(ndr_pull_uint16(call->ndr_pull_param, &r->out.count));
 246         NDR_OK(ndr_pull_uint16(call->ndr_pull_param, &r->out.available));
 247
 248         r->out.info = talloc_array_p(mem_ctx, union rap_shareenum_info,
 249                                      r->out.count);
 250
 251         if (r->out.info == NULL)
 252                 return NT_STATUS_NO_MEMORY;
 253
 254         for (i=0; i<r->out.count; i++) {
 255                 switch(r->in.level) {
 256                 case 0:
 257                         NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
 258                                               r->out.info[i].info0.name, 13));
 259                         break;
 260                 case 1:
 261                         NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
 262                                               r->out.info[i].info1.name, 13));
 263                         NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
 264                                               &r->out.info[i].info1.pad, 1));
 265                         NDR_OK(ndr_pull_uint16(call->ndr_pull_data,
 266                                                &r->out.info[i].info1.type));
 267                         NDR_OK(rap_pull_string(mem_ctx, call->ndr_pull_data,
 268                                                r->out.convert,
 269                                                &r->out.info[i].info1.comment));
 270                         break;
 271                 }
 272         }
 273
 274         result = NT_STATUS_OK;
 275
 276  done:
 277         destroy_rap_call(call);
 278
 279         return result;
 280 }
 281
 282 static BOOL test_netshareenum(struct cli_state *cli, TALLOC_CTX *mem_ctx)
 283 {
 284         struct rap_NetShareEnum r;
 285         int i;
 286
 287         r.in.level = 1;
 288         r.in.bufsize = 8192;
 289
 290         if (!NT_STATUS_IS_OK(cli_rap_netshareenum(cli, mem_ctx, &r)))
 291                 return False;
 292
 293         for (i=0; i<r.out.count; i++) {
 294                 printf("%s %d %s\n", r.out.info[i].info1.name,
 295                        r.out.info[i].info1.type,
 296                        r.out.info[i].info1.comment);
 297         }
 298
 299         return True;
 300 }
 301
 302 static NTSTATUS cli_rap_netserverenum2(struct cli_state *cli,
 303                                        TALLOC_CTX *mem_ctx,
 304                                        struct rap_NetServerEnum2 *r)
 305 {
 306         struct rap_call *call;
 307         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
 308         int i;
 309
 310         call = new_rap_cli_call(104);
 311
 312         if (call == NULL)
 313                 return NT_STATUS_NO_MEMORY;
 314
 315         rap_cli_push_word(call, r->in.level);
 316         rap_cli_push_rcvbuf(call, r->in.bufsize);
 317         rap_cli_expect_multiple_entries(call);
 318         rap_cli_push_dword(call, r->in.servertype);
 319         rap_cli_push_string(call, r->in.domain);
 320
 321         switch(r->in.level) {
 322         case 0:
 323                 rap_cli_expect_format(call, "B16");
 324                 break;
 325         case 1:
 326                 rap_cli_expect_format(call, "B16BBDz");
 327                 break;
 328         }
 329
 330         result = rap_cli_do_call(cli, mem_ctx, call);
 331
 332         if (!NT_STATUS_IS_OK(result))
 333                 goto done;
 334
 335         result = NT_STATUS_INVALID_PARAMETER;
 336
 337         NDR_OK(ndr_pull_uint16(call->ndr_pull_param, &r->out.status));
 338         NDR_OK(ndr_pull_uint16(call->ndr_pull_param, &r->out.convert));
 339         NDR_OK(ndr_pull_uint16(call->ndr_pull_param, &r->out.count));
 340         NDR_OK(ndr_pull_uint16(call->ndr_pull_param, &r->out.available));
 341
 342         r->out.info = talloc_array_p(mem_ctx, union rap_server_info,
 343                                      r->out.count);
 344
 345         if (r->out.info == NULL)
 346                 return NT_STATUS_NO_MEMORY;
 347
 348         for (i=0; i<r->out.count; i++) {
 349                 switch(r->in.level) {
 350                 case 0:
 351                         NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
 352                                               r->out.info[i].info0.name, 16));
 353                         break;
 354                 case 1:
 355                         NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
 356                                               r->out.info[i].info1.name, 16));
 357                         NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
 358                                               &r->out.info[i].info1.version_major, 1));
 359                         NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
 360                                               &r->out.info[i].info1.version_minor, 1));
 361                         NDR_OK(ndr_pull_uint32(call->ndr_pull_data,
 362                                                &r->out.info[i].info1.servertype));
 363                         NDR_OK(rap_pull_string(mem_ctx, call->ndr_pull_data,
 364                                                r->out.convert,
 365                                                &r->out.info[i].info1.comment));
 366                 }
 367         }
 368
 369         result = NT_STATUS_OK;
 370
 371  done:
 372         destroy_rap_call(call);
 373
 374         return result;
 375 }
 376
 377 static BOOL test_netserverenum(struct cli_state *cli, TALLOC_CTX *mem_ctx)
 378 {
 379         struct rap_NetServerEnum2 r;
 380         int i;
 381
 382         r.in.level = 0;
 383         r.in.bufsize = 8192;
 384         r.in.servertype = 0xffffffff;
 385         r.in.servertype = 0x80000000;
 386         r.in.domain = NULL;
 387
 388         if (!NT_STATUS_IS_OK(cli_rap_netserverenum2(cli, mem_ctx, &r)))
 389                 return False;
 390
 391         for (i=0; i<r.out.count; i++) {
 392                 switch (r.in.level) {
 393                 case 0:
 394                         printf("%s\n", r.out.info[i].info0.name);
 395                         break;
 396                 case 1:
 397                         printf("%s %x %s\n", r.out.info[i].info1.name,
 398                                r.out.info[i].info1.servertype,
 399                                r.out.info[i].info1.comment);
 400                         break;
 401                 }
 402         }
 403
 404         return True;
 405 }
 406
 407
 408
 409 static BOOL test_rap(struct cli_state *cli, TALLOC_CTX *mem_ctx)
 410 {
 411         BOOL res = True;
 412
 413         if (!test_netserverenum(cli, mem_ctx))
 414                 res = False;
 415
 416         if (!test_netshareenum(cli, mem_ctx))
 417                 res = False;
 418
 419         return res;
 420 }
 421
 422 BOOL torture_raw_rap(int dummy)
 423 {
 424         struct cli_state *cli;
 425         BOOL ret = True;
 426         TALLOC_CTX *mem_ctx;
 427
 428         if (!torture_open_connection(&cli)) {
 429                 return False;
 430         }
 431
 432         mem_ctx = talloc_init("torture_raw_rap");
 433
 434         if (!test_rap(cli, mem_ctx)) {
 435                 ret = False;
 436         }
 437
 438         torture_close_connection(cli);
 439         talloc_destroy(mem_ctx);
 440         return ret;
 441 }