source4/torture/torture.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    SMB torture tester
   4    Copyright (C) Andrew Tridgell 1997-2003
   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 #include "dynconfig.h"
  23 #include "clilist.h"
  24 #include "lib/cmdline/popt_common.h"
  25 #include "libcli/raw/libcliraw.h"
  26 #include "system/time.h"
  27 #include "system/wait.h"
  28 #include "ioctl.h"
  29 #include "librpc/gen_ndr/ndr_security.h"
  30
  31 int torture_nprocs=4;
  32 int torture_numops=100;
  33 int torture_entries=1000;
  34 int torture_failures=1;
  35 int torture_seed=0;
  36 static int procnum; /* records process count number when forking */
  37 static struct smbcli_state *current_cli;
  38 static BOOL use_oplocks;
  39 static BOOL use_level_II_oplocks;
  40
  41 BOOL torture_showall = False;
  42
  43 #define CHECK_MAX_FAILURES(label) do { if (++failures >= torture_failures) goto label; } while (0)
  44
  45 static struct smbcli_state *open_nbt_connection(void)
  46 {
  47         struct nmb_name called, calling;
  48         struct smbcli_state *cli;
  49         const char *host = lp_parm_string(-1, "torture", "host");
  50
  51         make_nmb_name(&calling, lp_netbios_name(), 0x0);
  52         choose_called_name(&called, host, 0x20);
  53
  54         cli = smbcli_state_init(NULL);
  55         if (!cli) {
  56                 printf("Failed initialize smbcli_struct to connect with %s\n", host);
  57                 return NULL;
  58         }
  59
  60         if (!smbcli_socket_connect(cli, host)) {
  61                 printf("Failed to connect with %s\n", host);
  62                 return cli;
  63         }
  64
  65         if (!smbcli_transport_establish(cli, &calling, &called)) {
  66                 /*
  67                  * Well, that failed, try *SMBSERVER ...
  68                  * However, we must reconnect as well ...
  69                  */
  70                 if (!smbcli_socket_connect(cli, host)) {
  71                         printf("Failed to connect with %s\n", host);
  72                         return False;
  73                 }
  74
  75                 make_nmb_name(&called, "*SMBSERVER", 0x20);
  76                 if (!smbcli_transport_establish(cli, &calling, &called)) {
  77                         printf("%s rejected the session\n",host);
  78                         printf("We tried with a called name of %s & %s\n",
  79                                 host, "*SMBSERVER");
  80                         smbcli_shutdown(cli);
  81                         return NULL;
  82                 }
  83         }
  84
  85         return cli;
  86 }
  87
  88 BOOL torture_open_connection_share(struct smbcli_state **c,
  89                                    const char *hostname,
  90                                    const char *sharename)
  91 {
  92         NTSTATUS status;
  93         const char *username = lp_parm_string(-1, "torture", "username");
  94         const char *userdomain = lp_parm_string(-1, "torture", "userdomain");
  95         const char *password = lp_parm_string(-1, "torture", "password");
  96
  97         status = smbcli_full_connection(NULL,
  98                                         c, lp_netbios_name(),
  99                                         hostname,
 100                                         sharename, NULL,
 101                                         username, username[0]?userdomain:"",
 102                                         password);
 103         if (!NT_STATUS_IS_OK(status)) {
 104                 printf("Failed to open connection - %s\n", nt_errstr(status));
 105                 return False;
 106         }
 107
 108         (*c)->transport->options.use_oplocks = use_oplocks;
 109         (*c)->transport->options.use_level2_oplocks = use_level_II_oplocks;
 110
 111         return True;
 112 }
 113
 114 BOOL torture_open_connection(struct smbcli_state **c)
 115 {
 116         const char *host = lp_parm_string(-1, "torture", "host");
 117         const char *share = lp_parm_string(-1, "torture", "share");
 118
 119         return torture_open_connection_share(c, host, share);
 120 }
 121
 122
 123
 124 BOOL torture_close_connection(struct smbcli_state *c)
 125 {
 126         BOOL ret = True;
 127         if (!c) return True;
 128         if (NT_STATUS_IS_ERR(smbcli_tdis(c))) {
 129                 printf("tdis failed (%s)\n", smbcli_errstr(c->tree));
 130                 ret = False;
 131         }
 132         smbcli_shutdown(c);
 133         return ret;
 134 }
 135
 136 /* initialise a DCOM context */
 137 NTSTATUS torture_dcom_init(struct dcom_context **ctx)
 138 {
 139         dcom_init(ctx, lp_parm_string(-1, "torture", "userdomain"),
 140                                         lp_parm_string(-1, "torture", "username"),
 141                                         lp_parm_string(-1, "torture", "password"));
 142
 143         return NT_STATUS_OK;
 144 }
 145
 146 /* open a rpc connection to the chosen binding string */
 147 NTSTATUS torture_rpc_connection(struct dcerpc_pipe **p,
 148                                 const char *pipe_name,
 149                                 const char *pipe_uuid,
 150                                 uint32_t pipe_version)
 151 {
 152         NTSTATUS status;
 153         const char *binding = lp_parm_string(-1, "torture", "binding");
 154
 155         if (!binding) {
 156                 printf("You must specify a ncacn binding string\n");
 157                 return NT_STATUS_INVALID_PARAMETER;
 158         }
 159
 160         status = dcerpc_pipe_connect(p, binding, pipe_uuid, pipe_version,
 161                                      lp_parm_string(-1, "torture", "userdomain"),
 162                                      lp_parm_string(-1, "torture", "username"),
 163                                      lp_parm_string(-1, "torture", "password"));
 164
 165         return status;
 166 }
 167
 168 /* open a rpc connection to a specific transport */
 169 NTSTATUS torture_rpc_connection_transport(struct dcerpc_pipe **p,
 170                                           const char *pipe_name,
 171                                           const char *pipe_uuid,
 172                                           uint32_t pipe_version,
 173                                           enum dcerpc_transport_t transport)
 174 {
 175         NTSTATUS status;
 176         const char *binding = lp_parm_string(-1, "torture", "binding");
 177         struct dcerpc_binding b;
 178         TALLOC_CTX *mem_ctx = talloc_init("torture_rpc_connection_smb");
 179
 180         if (!binding) {
 181                 printf("You must specify a ncacn binding string\n");
 182                 return NT_STATUS_INVALID_PARAMETER;
 183         }
 184
 185         status = dcerpc_parse_binding(mem_ctx, binding, &b);
 186         if (!NT_STATUS_IS_OK(status)) {
 187                 DEBUG(0,("Failed to parse dcerpc binding '%s'\n", binding));
 188                 talloc_destroy(mem_ctx);
 189                 return status;
 190         }
 191
 192         b.transport = transport;
 193
 194         status = dcerpc_pipe_connect_b(p, &b, pipe_uuid, pipe_version,
 195                                        lp_parm_string(-1, "torture", "userdomain"),
 196                                        lp_parm_string(-1, "torture", "username"),
 197                                        lp_parm_string(-1, "torture", "password"));
 198
 199         return status;
 200 }
 201
 202 /* close a rpc connection to a named pipe */
 203 NTSTATUS torture_rpc_close(struct dcerpc_pipe *p)
 204 {
 205         dcerpc_pipe_close(p);
 206         return NT_STATUS_OK;
 207 }
 208
 209
 210 /* check if the server produced the expected error code */
 211 BOOL check_error(const char *location, struct smbcli_state *c,
 212                  uint8_t eclass, uint32_t ecode, NTSTATUS nterr)
 213 {
 214         if (smbcli_is_dos_error(c->tree)) {
 215                 uint8_t class;
 216                 uint32_t num;
 217
 218                 /* Check DOS error */
 219
 220                 smbcli_dos_error(c, &class, &num);
 221
 222                 if (eclass != class || ecode != num) {
 223                         printf("unexpected error code class=%d code=%d\n",
 224                                (int)class, (int)num);
 225                         printf(" expected %d/%d %s (at %s)\n",
 226                                (int)eclass, (int)ecode, nt_errstr(nterr), location);
 227                         return False;
 228                 }
 229
 230         } else {
 231                 NTSTATUS status;
 232
 233                 /* Check NT error */
 234
 235                 status = smbcli_nt_error(c->tree);
 236
 237                 if (NT_STATUS_V(nterr) != NT_STATUS_V(status)) {
 238                         printf("unexpected error code %s\n", nt_errstr(status));
 239                         printf(" expected %s (at %s)\n", nt_errstr(nterr), location);
 240                         return False;
 241                 }
 242         }
 243
 244         return True;
 245 }
 246
 247
 248 static BOOL wait_lock(struct smbcli_state *c, int fnum, uint32_t offset, uint32_t len)
 249 {
 250         while (NT_STATUS_IS_ERR(smbcli_lock(c->tree, fnum, offset, len, -1, WRITE_LOCK))) {
 251                 if (!check_error(__location__, c, ERRDOS, ERRlock, NT_STATUS_LOCK_NOT_GRANTED)) return False;
 252         }
 253         return True;
 254 }
 255
 256
 257 static BOOL rw_torture(struct smbcli_state *c)
 258 {
 259         const char *lockfname = "\\torture.lck";
 260         char *fname;
 261         int fnum;
 262         int fnum2;
 263         pid_t pid2, pid = getpid();
 264         int i, j;
 265         uint8_t buf[1024];
 266         BOOL correct = True;
 267
 268         fnum2 = smbcli_open(c->tree, lockfname, O_RDWR | O_CREAT | O_EXCL,
 269                          DENY_NONE);
 270         if (fnum2 == -1)
 271                 fnum2 = smbcli_open(c->tree, lockfname, O_RDWR, DENY_NONE);
 272         if (fnum2 == -1) {
 273                 printf("open of %s failed (%s)\n", lockfname, smbcli_errstr(c->tree));
 274                 return False;
 275         }
 276
 277
 278         for (i=0;i<torture_numops;i++) {
 279                 uint_t n = (uint_t)sys_random()%10;
 280                 if (i % 10 == 0) {
 281                         printf("%d\r", i); fflush(stdout);
 282                 }
 283                 asprintf(&fname, "\\torture.%u", n);
 284
 285                 if (!wait_lock(c, fnum2, n*sizeof(int), sizeof(int))) {
 286                         return False;
 287                 }
 288
 289                 fnum = smbcli_open(c->tree, fname, O_RDWR | O_CREAT | O_TRUNC, DENY_ALL);
 290                 if (fnum == -1) {
 291                         printf("open failed (%s)\n", smbcli_errstr(c->tree));
 292                         correct = False;
 293                         break;
 294                 }
 295
 296                 if (smbcli_write(c->tree, fnum, 0, &pid, 0, sizeof(pid)) != sizeof(pid)) {
 297                         printf("write failed (%s)\n", smbcli_errstr(c->tree));
 298                         correct = False;
 299                 }
 300
 301                 for (j=0;j<50;j++) {
 302                         if (smbcli_write(c->tree, fnum, 0, buf,
 303                                       sizeof(pid)+(j*sizeof(buf)),
 304                                       sizeof(buf)) != sizeof(buf)) {
 305                                 printf("write failed (%s)\n", smbcli_errstr(c->tree));
 306                                 correct = False;
 307                         }
 308                 }
 309
 310                 pid2 = 0;
 311
 312                 if (smbcli_read(c->tree, fnum, &pid2, 0, sizeof(pid)) != sizeof(pid)) {
 313                         printf("read failed (%s)\n", smbcli_errstr(c->tree));
 314                         correct = False;
 315                 }
 316
 317                 if (pid2 != pid) {
 318                         printf("data corruption!\n");
 319                         correct = False;
 320                 }
 321
 322                 if (NT_STATUS_IS_ERR(smbcli_close(c->tree, fnum))) {
 323                         printf("close failed (%s)\n", smbcli_errstr(c->tree));
 324                         correct = False;
 325                 }
 326
 327                 if (NT_STATUS_IS_ERR(smbcli_unlink(c->tree, fname))) {
 328                         printf("unlink failed (%s)\n", smbcli_errstr(c->tree));
 329                         correct = False;
 330                 }
 331
 332                 if (NT_STATUS_IS_ERR(smbcli_unlock(c->tree, fnum2, n*sizeof(int), sizeof(int)))) {
 333                         printf("unlock failed (%s)\n", smbcli_errstr(c->tree));
 334                         correct = False;
 335                 }
 336                 free(fname);
 337         }
 338
 339         smbcli_close(c->tree, fnum2);
 340         smbcli_unlink(c->tree, lockfname);
 341
 342         printf("%d\n", i);
 343
 344         return correct;
 345 }
 346
 347 static BOOL run_torture(struct smbcli_state *cli, int dummy)
 348 {
 349         BOOL ret;
 350
 351         ret = rw_torture(cli);
 352
 353         if (!torture_close_connection(cli)) {
 354                 ret = False;
 355         }
 356
 357         return ret;
 358 }
 359
 360 static BOOL rw_torture3(struct smbcli_state *c, const char *lockfname)
 361 {
 362         int fnum = -1;
 363         uint_t i = 0;
 364         uint8_t buf[131072];
 365         uint8_t buf_rd[131072];
 366         uint_t count;
 367         uint_t countprev = 0;
 368         ssize_t sent = 0;
 369         BOOL correct = True;
 370
 371         for (i = 0; i < sizeof(buf); i += sizeof(uint32_t))
 372         {
 373                 SIVAL(buf, i, sys_random());
 374         }
 375
 376         if (procnum == 0)
 377         {
 378                 fnum = smbcli_open(c->tree, lockfname, O_RDWR | O_CREAT | O_EXCL,
 379                                 DENY_NONE);
 380                 if (fnum == -1) {
 381                         printf("first open read/write of %s failed (%s)\n",
 382                                         lockfname, smbcli_errstr(c->tree));
 383                         return False;
 384                 }
 385         }
 386         else
 387         {
 388                 for (i = 0; i < 500 && fnum == -1; i++)
 389                 {
 390                         fnum = smbcli_open(c->tree, lockfname, O_RDONLY,
 391                                         DENY_NONE);
 392                         msleep(10);
 393                 }
 394                 if (fnum == -1) {
 395                         printf("second open read-only of %s failed (%s)\n",
 396                                         lockfname, smbcli_errstr(c->tree));
 397                         return False;
 398                 }
 399         }
 400
 401         i = 0;
 402         for (count = 0; count < sizeof(buf); count += sent)
 403         {
 404                 if (count >= countprev) {
 405                         printf("%d %8d\r", i, count);
 406                         fflush(stdout);
 407                         i++;
 408                         countprev += (sizeof(buf) / 20);
 409                 }
 410
 411                 if (procnum == 0)
 412                 {
 413                         sent = ((uint_t)sys_random()%(20))+ 1;
 414                         if (sent > sizeof(buf) - count)
 415                         {
 416                                 sent = sizeof(buf) - count;
 417                         }
 418
 419                         if (smbcli_write(c->tree, fnum, 0, buf+count, count, (size_t)sent) != sent) {
 420                                 printf("write failed (%s)\n", smbcli_errstr(c->tree));
 421                                 correct = False;
 422                         }
 423                 }
 424                 else
 425                 {
 426                         sent = smbcli_read(c->tree, fnum, buf_rd+count, count,
 427                                         sizeof(buf)-count);
 428                         if (sent < 0)
 429                         {
 430                                 printf("read failed offset:%d size:%d (%s)\n",
 431                                                 count, sizeof(buf)-count,
 432                                                 smbcli_errstr(c->tree));
 433                                 correct = False;
 434                                 sent = 0;
 435                         }
 436                         if (sent > 0)
 437                         {
 438                                 if (memcmp(buf_rd+count, buf+count, sent) != 0)
 439                                 {
 440                                         printf("read/write compare failed\n");
 441                                         printf("offset: %d req %d recvd %d\n",
 442                                                 count, sizeof(buf)-count, sent);
 443                                         correct = False;
 444                                         break;
 445                                 }
 446                         }
 447                 }
 448
 449         }
 450
 451         if (NT_STATUS_IS_ERR(smbcli_close(c->tree, fnum))) {
 452                 printf("close failed (%s)\n", smbcli_errstr(c->tree));
 453                 correct = False;
 454         }
 455
 456         return correct;
 457 }
 458
 459 static BOOL rw_torture2(struct smbcli_state *c1, struct smbcli_state *c2)
 460 {
 461         const char *lockfname = "\\torture2.lck";
 462         int fnum1;
 463         int fnum2;
 464         int i;
 465         uint8_t buf[131072];
 466         uint8_t buf_rd[131072];
 467         BOOL correct = True;
 468         ssize_t bytes_read, bytes_written;
 469
 470         if (smbcli_deltree(c1->tree, lockfname) == -1) {
 471                 printf("unlink failed (%s)\n", smbcli_errstr(c1->tree));
 472         }
 473
 474         fnum1 = smbcli_open(c1->tree, lockfname, O_RDWR | O_CREAT | O_EXCL,
 475                          DENY_NONE);
 476         if (fnum1 == -1) {
 477                 printf("first open read/write of %s failed (%s)\n",
 478                                 lockfname, smbcli_errstr(c1->tree));
 479                 return False;
 480         }
 481         fnum2 = smbcli_open(c2->tree, lockfname, O_RDONLY,
 482                          DENY_NONE);
 483         if (fnum2 == -1) {
 484                 printf("second open read-only of %s failed (%s)\n",
 485                                 lockfname, smbcli_errstr(c2->tree));
 486                 smbcli_close(c1->tree, fnum1);
 487                 return False;
 488         }
 489
 490         printf("Checking data integrity over %d ops\n", torture_numops);
 491
 492         for (i=0;i<torture_numops;i++)
 493         {
 494                 size_t buf_size = ((uint_t)sys_random()%(sizeof(buf)-1))+ 1;
 495                 if (i % 10 == 0) {
 496                         printf("%d\r", i); fflush(stdout);
 497                 }
 498
 499                 generate_random_buffer(buf, buf_size);
 500
 501                 if ((bytes_written = smbcli_write(c1->tree, fnum1, 0, buf, 0, buf_size)) != buf_size) {
 502                         printf("write failed (%s)\n", smbcli_errstr(c1->tree));
 503                         printf("wrote %d, expected %d\n", bytes_written, buf_size);
 504                         correct = False;
 505                         break;
 506                 }
 507
 508                 if ((bytes_read = smbcli_read(c2->tree, fnum2, buf_rd, 0, buf_size)) != buf_size) {
 509                         printf("read failed (%s)\n", smbcli_errstr(c2->tree));
 510                         printf("read %d, expected %d\n", bytes_read, buf_size);
 511                         correct = False;
 512                         break;
 513                 }
 514
 515                 if (memcmp(buf_rd, buf, buf_size) != 0)
 516                 {
 517                         printf("read/write compare failed\n");
 518                         correct = False;
 519                         break;
 520                 }
 521         }
 522
 523         if (NT_STATUS_IS_ERR(smbcli_close(c2->tree, fnum2))) {
 524                 printf("close failed (%s)\n", smbcli_errstr(c2->tree));
 525                 correct = False;
 526         }
 527         if (NT_STATUS_IS_ERR(smbcli_close(c1->tree, fnum1))) {
 528                 printf("close failed (%s)\n", smbcli_errstr(c1->tree));
 529                 correct = False;
 530         }
 531
 532         if (NT_STATUS_IS_ERR(smbcli_unlink(c1->tree, lockfname))) {
 533                 printf("unlink failed (%s)\n", smbcli_errstr(c1->tree));
 534                 correct = False;
 535         }
 536
 537         return correct;
 538 }
 539
 540 static BOOL run_readwritetest(void)
 541 {
 542         struct smbcli_state *cli1, *cli2;
 543         BOOL test1, test2 = True;
 544
 545         if (!torture_open_connection(&cli1) || !torture_open_connection(&cli2)) {
 546                 return False;
 547         }
 548
 549         printf("starting readwritetest\n");
 550
 551         test1 = rw_torture2(cli1, cli2);
 552         printf("Passed readwritetest v1: %s\n", BOOLSTR(test1));
 553
 554         if (test1) {
 555                 test2 = rw_torture2(cli1, cli1);
 556                 printf("Passed readwritetest v2: %s\n", BOOLSTR(test2));
 557         }
 558
 559         if (!torture_close_connection(cli1)) {
 560                 test1 = False;
 561         }
 562
 563         if (!torture_close_connection(cli2)) {
 564                 test2 = False;
 565         }
 566
 567         return (test1 && test2);
 568 }
 569
 570 static BOOL run_readwritemulti(struct smbcli_state *cli, int dummy)
 571 {
 572         BOOL test;
 573
 574         test = rw_torture3(cli, "\\multitest.txt");
 575
 576         if (!torture_close_connection(cli)) {
 577                 test = False;
 578         }
 579
 580         return test;
 581 }
 582
 583
 584 /*
 585   this checks to see if a secondary tconx can use open files from an
 586   earlier tconx
 587  */
 588 static BOOL run_tcon_test(void)
 589 {
 590         struct smbcli_state *cli;
 591         const char *fname = "\\tcontest.tmp";
 592         int fnum1;
 593         uint16_t cnum1, cnum2, cnum3;
 594         uint16_t vuid1, vuid2;
 595         uint8_t buf[4];
 596         BOOL ret = True;
 597         struct smbcli_tree *tree1;
 598         const char *host = lp_parm_string(-1, "torture", "host");
 599         const char *share = lp_parm_string(-1, "torture", "share");
 600         const char *password = lp_parm_string(-1, "torture", "password");
 601
 602         if (!torture_open_connection(&cli)) {
 603                 return False;
 604         }
 605
 606         printf("starting tcontest\n");
 607
 608         if (smbcli_deltree(cli->tree, fname) == -1) {
 609                 printf("unlink of %s failed (%s)\n", fname, smbcli_errstr(cli->tree));
 610         }
 611
 612         fnum1 = smbcli_open(cli->tree, fname, O_RDWR|O_CREAT|O_EXCL, DENY_NONE);
 613         if (fnum1 == -1) {
 614                 printf("open of %s failed (%s)\n", fname, smbcli_errstr(cli->tree));
 615                 return False;
 616         }
 617
 618         cnum1 = cli->tree->tid;
 619         vuid1 = cli->session->vuid;
 620
 621         memset(&buf, 0, 4); /* init buf so valgrind won't complain */
 622         if (smbcli_write(cli->tree, fnum1, 0, buf, 130, 4) != 4) {
 623                 printf("initial write failed (%s)\n", smbcli_errstr(cli->tree));
 624                 return False;
 625         }
 626
 627         tree1 = cli->tree;      /* save old tree connection */
 628         if (NT_STATUS_IS_ERR(smbcli_send_tconX(cli, share, "?????", password))) {
 629                 printf("%s refused 2nd tree connect (%s)\n", host,
 630                            smbcli_errstr(cli->tree));
 631                 smbcli_shutdown(cli);
 632                 return False;
 633         }
 634
 635         cnum2 = cli->tree->tid;
 636         cnum3 = MAX(cnum1, cnum2) + 1; /* any invalid number */
 637         vuid2 = cli->session->vuid + 1;
 638
 639         /* try a write with the wrong tid */
 640         cli->tree->tid = cnum2;
 641
 642         if (smbcli_write(cli->tree, fnum1, 0, buf, 130, 4) == 4) {
 643                 printf("* server allows write with wrong TID\n");
 644                 ret = False;
 645         } else {
 646                 printf("server fails write with wrong TID : %s\n", smbcli_errstr(cli->tree));
 647         }
 648
 649
 650         /* try a write with an invalid tid */
 651         cli->tree->tid = cnum3;
 652
 653         if (smbcli_write(cli->tree, fnum1, 0, buf, 130, 4) == 4) {
 654                 printf("* server allows write with invalid TID\n");
 655                 ret = False;
 656         } else {
 657                 printf("server fails write with invalid TID : %s\n", smbcli_errstr(cli->tree));
 658         }
 659
 660         /* try a write with an invalid vuid */
 661         cli->session->vuid = vuid2;
 662         cli->tree->tid = cnum1;
 663
 664         if (smbcli_write(cli->tree, fnum1, 0, buf, 130, 4) == 4) {
 665                 printf("* server allows write with invalid VUID\n");
 666                 ret = False;
 667         } else {
 668                 printf("server fails write with invalid VUID : %s\n", smbcli_errstr(cli->tree));
 669         }
 670
 671         cli->session->vuid = vuid1;
 672         cli->tree->tid = cnum1;
 673
 674         if (NT_STATUS_IS_ERR(smbcli_close(cli->tree, fnum1))) {
 675                 printf("close failed (%s)\n", smbcli_errstr(cli->tree));
 676                 return False;
 677         }
 678
 679         cli->tree->tid = cnum2;
 680
 681         if (NT_STATUS_IS_ERR(smbcli_tdis(cli))) {
 682                 printf("secondary tdis failed (%s)\n", smbcli_errstr(cli->tree));
 683                 return False;
 684         }
 685
 686         cli->tree = tree1;  /* restore initial tree */
 687         cli->tree->tid = cnum1;
 688
 689         if (!torture_close_connection(cli)) {
 690                 return False;
 691         }
 692
 693         return ret;
 694 }
 695
 696
 697
 698 static BOOL tcon_devtest(struct smbcli_state *cli,
 699                          const char *myshare, const char *devtype,
 700                          NTSTATUS expected_error)
 701 {
 702         BOOL status;
 703         BOOL ret;
 704         const char *password = lp_parm_string(-1, "torture", "password");
 705
 706         status = NT_STATUS_IS_OK(smbcli_send_tconX(cli, myshare, devtype,
 707                                                 password));
 708
 709         printf("Trying share %s with devtype %s\n", myshare, devtype);
 710
 711         if (NT_STATUS_IS_OK(expected_error)) {
 712                 if (status) {
 713                         ret = True;
 714                 } else {
 715                         printf("tconX to share %s with type %s "
 716                                "should have succeeded but failed\n",
 717                                myshare, devtype);
 718                         ret = False;
 719                 }
 720                 smbcli_tdis(cli);
 721         } else {
 722                 if (status) {
 723                         printf("tconx to share %s with type %s "
 724                                "should have failed but succeeded\n",
 725                                myshare, devtype);
 726                         ret = False;
 727                 } else {
 728                         if (NT_STATUS_EQUAL(smbcli_nt_error(cli->tree),
 729                                             expected_error)) {
 730                                 ret = True;
 731                         } else {
 732                                 printf("Returned unexpected error\n");
 733                                 ret = False;
 734                         }
 735                 }
 736         }
 737         return ret;
 738 }
 739
 740 /*
 741  checks for correct tconX support
 742  */
 743 static BOOL run_tcon_devtype_test(void)
 744 {
 745         struct smbcli_state *cli1 = NULL;
 746         NTSTATUS status;
 747         BOOL ret = True;
 748         const char *host = lp_parm_string(-1, "torture", "host");
 749         const char *share = lp_parm_string(-1, "torture", "share");
 750         const char *username = lp_parm_string(-1, "torture", "username");
 751         const char *userdomain = lp_parm_string(-1, "torture", "userdomain");
 752         const char *password = lp_parm_string(-1, "torture", "password");
 753
 754         status = smbcli_full_connection(NULL,
 755                                         &cli1, lp_netbios_name(),
 756                                         host,
 757                                         share, NULL,
 758                                         username, userdomain,
 759                                         password);
 760
 761         if (!NT_STATUS_IS_OK(status)) {
 762                 printf("could not open connection\n");
 763                 return False;
 764         }
 765
 766         if (!tcon_devtest(cli1, "IPC$", "A:", NT_STATUS_BAD_DEVICE_TYPE))
 767                 ret = False;
 768
 769         if (!tcon_devtest(cli1, "IPC$", "?????", NT_STATUS_OK))
 770                 ret = False;
 771
 772         if (!tcon_devtest(cli1, "IPC$", "LPT:", NT_STATUS_BAD_DEVICE_TYPE))
 773                 ret = False;
 774
 775         if (!tcon_devtest(cli1, "IPC$", "IPC", NT_STATUS_OK))
 776                 ret = False;
 777
 778         if (!tcon_devtest(cli1, "IPC$", "FOOBA", NT_STATUS_BAD_DEVICE_TYPE))
 779                 ret = False;
 780
 781         if (!tcon_devtest(cli1, share, "A:", NT_STATUS_OK))
 782                 ret = False;
 783
 784         if (!tcon_devtest(cli1, share, "?????", NT_STATUS_OK))
 785                 ret = False;
 786
 787         if (!tcon_devtest(cli1, share, "LPT:", NT_STATUS_BAD_DEVICE_TYPE))
 788                 ret = False;
 789
 790         if (!tcon_devtest(cli1, share, "IPC", NT_STATUS_BAD_DEVICE_TYPE))
 791                 ret = False;
 792
 793         if (!tcon_devtest(cli1, share, "FOOBA", NT_STATUS_BAD_DEVICE_TYPE))
 794                 ret = False;
 795
 796         smbcli_shutdown(cli1);
 797
 798         if (ret)
 799                 printf("Passed tcondevtest\n");
 800
 801         return ret;
 802 }
 803
 804
 805 /*
 806 test whether fnums and tids open on one VC are available on another (a major
 807 security hole)
 808 */
 809 static BOOL run_fdpasstest(void)
 810 {
 811         struct smbcli_state *cli1, *cli2;
 812         const char *fname = "\\fdpass.tst";
 813         int fnum1, oldtid;
 814         uint8_t buf[1024];
 815
 816         if (!torture_open_connection(&cli1) || !torture_open_connection(&cli2)) {
 817                 return False;
 818         }
 819
 820         printf("starting fdpasstest\n");
 821
 822         smbcli_unlink(cli1->tree, fname);
 823
 824         printf("Opening a file on connection 1\n");
 825
 826         fnum1 = smbcli_open(cli1->tree, fname, O_RDWR|O_CREAT|O_EXCL, DENY_NONE);
 827         if (fnum1 == -1) {
 828                 printf("open of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
 829                 return False;
 830         }
 831
 832         printf("writing to file on connection 1\n");
 833
 834         if (smbcli_write(cli1->tree, fnum1, 0, "hello world\n", 0, 13) != 13) {
 835                 printf("write failed (%s)\n", smbcli_errstr(cli1->tree));
 836                 return False;
 837         }
 838
 839         oldtid = cli2->tree->tid;
 840         cli2->session->vuid = cli1->session->vuid;
 841         cli2->tree->tid = cli1->tree->tid;
 842         cli2->session->pid = cli1->session->pid;
 843
 844         printf("reading from file on connection 2\n");
 845
 846         if (smbcli_read(cli2->tree, fnum1, buf, 0, 13) == 13) {
 847                 printf("read succeeded! nasty security hole [%s]\n",
 848                        buf);
 849                 return False;
 850         }
 851
 852         smbcli_close(cli1->tree, fnum1);
 853         smbcli_unlink(cli1->tree, fname);
 854
 855         cli2->tree->tid = oldtid;
 856
 857         torture_close_connection(cli1);
 858         torture_close_connection(cli2);
 859
 860         printf("finished fdpasstest\n");
 861         return True;
 862 }
 863
 864
 865 /*
 866 test the timing of deferred open requests
 867 */
 868 static BOOL run_deferopen(struct smbcli_state *cli, int dummy)
 869 {
 870         const char *fname = "\\defer_open_test.dat";
 871         int retries=4;
 872         int i = 0;
 873         BOOL correct = True;
 874
 875         if (retries <= 0) {
 876                 printf("failed to connect\n");
 877                 return False;
 878         }
 879
 880         printf("Testing deferred open requests.\n");
 881
 882         while (i < 4) {
 883                 int fnum = -1;
 884
 885                 do {
 886                         struct timeval tv;
 887                         tv = timeval_current();
 888                         fnum = smbcli_nt_create_full(cli->tree, fname, 0,
 889                                                      SEC_RIGHTS_FILE_ALL,
 890                                                      FILE_ATTRIBUTE_NORMAL,
 891                                                      NTCREATEX_SHARE_ACCESS_NONE,
 892                                                      NTCREATEX_DISP_OPEN_IF, 0, 0);
 893                         if (fnum != -1) {
 894                                 break;
 895                         }
 896                         if (NT_STATUS_EQUAL(smbcli_nt_error(cli->tree),NT_STATUS_SHARING_VIOLATION)) {
 897                                 double e = timeval_elapsed(&tv);
 898                                 if (e < 0.5 || e > 1.5) {
 899                                         fprintf(stderr,"Timing incorrect %.2f violation\n",
 900                                                 e);
 901                                 }
 902                         }
 903                 } while (NT_STATUS_EQUAL(smbcli_nt_error(cli->tree),NT_STATUS_SHARING_VIOLATION));
 904
 905                 if (fnum == -1) {
 906                         fprintf(stderr,"Failed to open %s, error=%s\n", fname, smbcli_errstr(cli->tree));
 907                         return False;
 908                 }
 909
 910                 printf("pid %u open %d\n", getpid(), i);
 911
 912                 sleep(10);
 913                 i++;
 914                 if (NT_STATUS_IS_ERR(smbcli_close(cli->tree, fnum))) {
 915                         fprintf(stderr,"Failed to close %s, error=%s\n", fname, smbcli_errstr(cli->tree));
 916                         return False;
 917                 }
 918                 sleep(2);
 919         }
 920
 921         if (NT_STATUS_IS_ERR(smbcli_unlink(cli->tree, fname))) {
 922                 /* All until the last unlink will fail with sharing violation. */
 923                 if (!NT_STATUS_EQUAL(smbcli_nt_error(cli->tree),NT_STATUS_SHARING_VIOLATION)) {
 924                         printf("unlink of %s failed (%s)\n", fname, smbcli_errstr(cli->tree));
 925                         correct = False;
 926                 }
 927         }
 928
 929         printf("deferred test finished\n");
 930         if (!torture_close_connection(cli)) {
 931                 correct = False;
 932         }
 933         return correct;
 934 }
 935
 936 /*
 937 test how many open files this server supports on the one socket
 938 */
 939 static BOOL run_maxfidtest(struct smbcli_state *cli, int dummy)
 940 {
 941 #define MAXFID_TEMPLATE "\\maxfid\\fid%d\\maxfid.%d.%d"
 942         char *fname;
 943         int fnums[0x11000], i;
 944         int retries=4, maxfid;
 945         BOOL correct = True;
 946
 947         if (retries <= 0) {
 948                 printf("failed to connect\n");
 949                 return False;
 950         }
 951
 952         if (smbcli_deltree(cli->tree, "\\maxfid") == -1) {
 953                 printf("Failed to deltree \\maxfid - %s\n",
 954                        smbcli_errstr(cli->tree));
 955                 return False;
 956         }
 957         if (NT_STATUS_IS_ERR(smbcli_mkdir(cli->tree, "\\maxfid"))) {
 958                 printf("Failed to mkdir \\maxfid, error=%s\n",
 959                        smbcli_errstr(cli->tree));
 960                 return False;
 961         }
 962
 963         printf("Testing maximum number of open files\n");
 964
 965         for (i=0; i<0x11000; i++) {
 966                 if (i % 1000 == 0) {
 967                         asprintf(&fname, "\\maxfid\\fid%d", i/1000);
 968                         if (NT_STATUS_IS_ERR(smbcli_mkdir(cli->tree, fname))) {
 969                                 printf("Failed to mkdir %s, error=%s\n",
 970                                        fname, smbcli_errstr(cli->tree));
 971                                 return False;
 972                         }
 973                         free(fname);
 974                 }
 975                 asprintf(&fname, MAXFID_TEMPLATE, i/1000, i,(int)getpid());
 976                 if ((fnums[i] = smbcli_open(cli->tree, fname,
 977                                         O_RDWR|O_CREAT|O_TRUNC, DENY_NONE)) ==
 978                     -1) {
 979                         printf("open of %s failed (%s)\n",
 980                                fname, smbcli_errstr(cli->tree));
 981                         printf("maximum fnum is %d\n", i);
 982                         break;
 983                 }
 984                 free(fname);
 985                 printf("%6d\r", i);
 986         }
 987         printf("%6d\n", i);
 988         i--;
 989
 990         maxfid = i;
 991
 992         printf("cleaning up\n");
 993         for (i=0;i<maxfid/2;i++) {
 994                 asprintf(&fname, MAXFID_TEMPLATE, i/1000, i,(int)getpid());
 995                 if (NT_STATUS_IS_ERR(smbcli_close(cli->tree, fnums[i]))) {
 996                         printf("Close of fnum %d failed - %s\n", fnums[i], smbcli_errstr(cli->tree));
 997                 }
 998                 if (NT_STATUS_IS_ERR(smbcli_unlink(cli->tree, fname))) {
 999                         printf("unlink of %s failed (%s)\n",
1000                                fname, smbcli_errstr(cli->tree));
1001                         correct = False;
1002                 }
1003                 free(fname);
1004
1005                 asprintf(&fname, MAXFID_TEMPLATE, (maxfid-i)/1000, maxfid-i,(int)getpid());
1006                 if (NT_STATUS_IS_ERR(smbcli_close(cli->tree, fnums[maxfid-i]))) {
1007                         printf("Close of fnum %d failed - %s\n", fnums[maxfid-i], smbcli_errstr(cli->tree));
1008                 }
1009                 if (NT_STATUS_IS_ERR(smbcli_unlink(cli->tree, fname))) {
1010                         printf("unlink of %s failed (%s)\n",
1011                                fname, smbcli_errstr(cli->tree));
1012                         correct = False;
1013                 }
1014                 free(fname);
1015
1016                 printf("%6d %6d\r", i, maxfid-i);
1017         }
1018         printf("%6d\n", 0);
1019
1020         if (smbcli_deltree(cli->tree, "\\maxfid") == -1) {
1021                 printf("Failed to deltree \\maxfid - %s\n",
1022                        smbcli_errstr(cli->tree));
1023                 return False;
1024         }
1025
1026         printf("maxfid test finished\n");
1027         if (!torture_close_connection(cli)) {
1028                 correct = False;
1029         }
1030         return correct;
1031 #undef MAXFID_TEMPLATE
1032 }
1033
1034 /* send smb negprot commands, not reading the response */
1035 static BOOL run_negprot_nowait(void)
1036 {
1037         int i;
1038         struct smbcli_state *cli, *cli2;
1039         BOOL correct = True;
1040
1041         printf("starting negprot nowait test\n");
1042
1043         cli = open_nbt_connection();
1044         if (!cli) {
1045                 return False;
1046         }
1047
1048         printf("Filling send buffer\n");
1049
1050         for (i=0;i<10000;i++) {
1051                 struct smbcli_request *req;
1052                 time_t t1 = time(NULL);
1053                 req = smb_raw_negotiate_send(cli->transport, PROTOCOL_NT1);
1054                 while (req->state == SMBCLI_REQUEST_SEND && time(NULL) < t1+5) {
1055                         smbcli_transport_process(cli->transport);
1056                 }
1057                 if (req->state == SMBCLI_REQUEST_ERROR) {
1058                         printf("Failed to fill pipe - %s\n", nt_errstr(req->status));
1059                         torture_close_connection(cli);
1060                         return correct;
1061                 }
1062                 if (req->state == SMBCLI_REQUEST_SEND) {
1063                         break;
1064                 }
1065         }
1066
1067         if (i == 10000) {
1068                 printf("send buffer failed to fill\n");
1069                 if (!torture_close_connection(cli)) {
1070                         correct = False;
1071                 }
1072                 return correct;
1073         }
1074
1075         printf("send buffer filled after %d requests\n", i);
1076
1077         printf("Opening secondary connection\n");
1078         if (!torture_open_connection(&cli2)) {
1079                 return False;
1080         }
1081
1082         if (!torture_close_connection(cli)) {
1083                 correct = False;
1084         }
1085
1086         if (!torture_close_connection(cli2)) {
1087                 correct = False;
1088         }
1089
1090         printf("finished negprot nowait test\n");
1091
1092         return correct;
1093 }
1094
1095
1096 /*
1097   This checks how the getatr calls works
1098 */
1099 static BOOL run_attrtest(void)
1100 {
1101         struct smbcli_state *cli;
1102         int fnum;
1103         time_t t, t2;
1104         const char *fname = "\\attrib123456789.tst";
1105         BOOL correct = True;
1106
1107         printf("starting attrib test\n");
1108
1109         if (!torture_open_connection(&cli)) {
1110                 return False;
1111         }
1112
1113         smbcli_unlink(cli->tree, fname);
1114         fnum = smbcli_open(cli->tree, fname,
1115                         O_RDWR | O_CREAT | O_TRUNC, DENY_NONE);
1116         smbcli_close(cli->tree, fnum);
1117
1118         if (NT_STATUS_IS_ERR(smbcli_getatr(cli->tree, fname, NULL, NULL, &t))) {
1119                 printf("getatr failed (%s)\n", smbcli_errstr(cli->tree));
1120                 correct = False;
1121         }
1122
1123         printf("New file time is %s", ctime(&t));
1124
1125         if (abs(t - time(NULL)) > 60*60*24*10) {
1126                 printf("ERROR: SMBgetatr bug. time is %s",
1127                        ctime(&t));
1128                 t = time(NULL);
1129                 correct = False;
1130         }
1131
1132         t2 = t-60*60*24; /* 1 day ago */
1133
1134         printf("Setting file time to %s", ctime(&t2));
1135
1136         if (NT_STATUS_IS_ERR(smbcli_setatr(cli->tree, fname, 0, t2))) {
1137                 printf("setatr failed (%s)\n", smbcli_errstr(cli->tree));
1138                 correct = True;
1139         }
1140
1141         if (NT_STATUS_IS_ERR(smbcli_getatr(cli->tree, fname, NULL, NULL, &t))) {
1142                 printf("getatr failed (%s)\n", smbcli_errstr(cli->tree));
1143                 correct = True;
1144         }
1145
1146         printf("Retrieved file time as %s", ctime(&t));
1147
1148         if (t != t2) {
1149                 printf("ERROR: getatr/setatr bug. times are\n%s",
1150                        ctime(&t));
1151                 printf("%s", ctime(&t2));
1152                 correct = True;
1153         }
1154
1155         smbcli_unlink(cli->tree, fname);
1156
1157         if (!torture_close_connection(cli)) {
1158                 correct = False;
1159         }
1160
1161         printf("attrib test finished\n");
1162
1163         return correct;
1164 }
1165
1166
1167 /*
1168   This checks a couple of trans2 calls
1169 */
1170 static BOOL run_trans2test(void)
1171 {
1172         struct smbcli_state *cli;
1173         int fnum;
1174         size_t size;
1175         time_t c_time, a_time, m_time, w_time, m_time2;
1176         const char *fname = "\\trans2.tst";
1177         const char *dname = "\\trans2";
1178         const char *fname2 = "\\trans2\\trans2.tst";
1179         const char *pname;
1180         BOOL correct = True;
1181
1182         printf("starting trans2 test\n");
1183
1184         if (!torture_open_connection(&cli)) {
1185                 return False;
1186         }
1187
1188         smbcli_unlink(cli->tree, fname);
1189
1190         printf("Testing qfileinfo\n");
1191
1192         fnum = smbcli_open(cli->tree, fname,
1193                         O_RDWR | O_CREAT | O_TRUNC, DENY_NONE);
1194         if (NT_STATUS_IS_ERR(smbcli_qfileinfo(cli->tree, fnum, NULL, &size, &c_time, &a_time, &m_time,
1195                            NULL, NULL))) {
1196                 printf("ERROR: qfileinfo failed (%s)\n", smbcli_errstr(cli->tree));
1197                 correct = False;
1198         }
1199
1200         printf("Testing NAME_INFO\n");
1201
1202         if (NT_STATUS_IS_ERR(smbcli_qfilename(cli->tree, fnum, &pname))) {
1203                 printf("ERROR: qfilename failed (%s)\n", smbcli_errstr(cli->tree));
1204                 correct = False;
1205         }
1206
1207         if (!pname || strcmp(pname, fname)) {
1208                 printf("qfilename gave different name? [%s] [%s]\n",
1209                        fname, pname);
1210                 correct = False;
1211         }
1212
1213         smbcli_close(cli->tree, fnum);
1214         smbcli_unlink(cli->tree, fname);
1215
1216         fnum = smbcli_open(cli->tree, fname,
1217                         O_RDWR | O_CREAT | O_TRUNC, DENY_NONE);
1218         if (fnum == -1) {
1219                 printf("open of %s failed (%s)\n", fname, smbcli_errstr(cli->tree));
1220                 return False;
1221         }
1222         smbcli_close(cli->tree, fnum);
1223
1224         printf("Checking for sticky create times\n");
1225
1226         if (NT_STATUS_IS_ERR(smbcli_qpathinfo(cli->tree, fname, &c_time, &a_time, &m_time, &size, NULL))) {
1227                 printf("ERROR: qpathinfo failed (%s)\n", smbcli_errstr(cli->tree));
1228                 correct = False;
1229         } else {
1230                 if (c_time != m_time) {
1231                         printf("create time=%s", ctime(&c_time));
1232                         printf("modify time=%s", ctime(&m_time));
1233                         printf("This system appears to have sticky create times\n");
1234                 }
1235                 if (a_time % (60*60) == 0) {
1236                         printf("access time=%s", ctime(&a_time));
1237                         printf("This system appears to set a midnight access time\n");
1238                         correct = False;
1239                 }
1240
1241                 if (abs(m_time - time(NULL)) > 60*60*24*7) {
1242                         printf("ERROR: totally incorrect times - maybe word reversed? mtime=%s", ctime(&m_time));
1243                         correct = False;
1244                 }
1245         }
1246
1247
1248         smbcli_unlink(cli->tree, fname);
1249         fnum = smbcli_open(cli->tree, fname,
1250                         O_RDWR | O_CREAT | O_TRUNC, DENY_NONE);
1251         smbcli_close(cli->tree, fnum);
1252         if (NT_STATUS_IS_ERR(smbcli_qpathinfo2(cli->tree, fname, &c_time, &a_time, &m_time, &w_time, &size, NULL, NULL))) {
1253                 printf("ERROR: qpathinfo2 failed (%s)\n", smbcli_errstr(cli->tree));
1254                 correct = False;
1255         } else {
1256                 if (w_time < 60*60*24*2) {
1257                         printf("write time=%s", ctime(&w_time));
1258                         printf("This system appears to set a initial 0 write time\n");
1259                         correct = False;
1260                 }
1261         }
1262
1263         smbcli_unlink(cli->tree, fname);
1264
1265
1266         /* check if the server updates the directory modification time
1267            when creating a new file */
1268         if (NT_STATUS_IS_ERR(smbcli_mkdir(cli->tree, dname))) {
1269                 printf("ERROR: mkdir failed (%s)\n", smbcli_errstr(cli->tree));
1270                 correct = False;
1271         }
1272         sleep(3);
1273         if (NT_STATUS_IS_ERR(smbcli_qpathinfo2(cli->tree, "\\trans2\\", &c_time, &a_time, &m_time, &w_time, &size, NULL, NULL))) {
1274                 printf("ERROR: qpathinfo2 failed (%s)\n", smbcli_errstr(cli->tree));
1275                 correct = False;
1276         }
1277
1278         fnum = smbcli_open(cli->tree, fname2,
1279                         O_RDWR | O_CREAT | O_TRUNC, DENY_NONE);
1280         smbcli_write(cli->tree, fnum,  0, &fnum, 0, sizeof(fnum));
1281         smbcli_close(cli->tree, fnum);
1282         if (NT_STATUS_IS_ERR(smbcli_qpathinfo2(cli->tree, "\\trans2\\", &c_time, &a_time, &m_time2, &w_time, &size, NULL, NULL))) {
1283                 printf("ERROR: qpathinfo2 failed (%s)\n", smbcli_errstr(cli->tree));
1284                 correct = False;
1285         } else {
1286                 if (m_time2 == m_time) {
1287                         printf("This system does not update directory modification times\n");
1288                         correct = False;
1289                 }
1290         }
1291         smbcli_unlink(cli->tree, fname2);
1292         smbcli_rmdir(cli->tree, dname);
1293
1294         if (!torture_close_connection(cli)) {
1295                 correct = False;
1296         }
1297
1298         printf("trans2 test finished\n");
1299
1300         return correct;
1301 }
1302
1303
1304
1305 /* FIRST_DESIRED_ACCESS   0xf019f */
1306 #define FIRST_DESIRED_ACCESS   SEC_FILE_READ_DATA|SEC_FILE_WRITE_DATA|SEC_FILE_APPEND_DATA|\
1307                                SEC_FILE_READ_EA|                           /* 0xf */ \
1308                                SEC_FILE_WRITE_EA|SEC_FILE_READ_ATTRIBUTE|     /* 0x90 */ \
1309                                SEC_FILE_WRITE_ATTRIBUTE|                  /* 0x100 */ \
1310                                SEC_STD_DELETE|SEC_STD_READ_CONTROL|\
1311                                SEC_STD_WRITE_DAC|SEC_STD_WRITE_OWNER     /* 0xf0000 */
1312 /* SECOND_DESIRED_ACCESS  0xe0080 */
1313 #define SECOND_DESIRED_ACCESS  SEC_FILE_READ_ATTRIBUTE|                   /* 0x80 */ \
1314                                SEC_STD_READ_CONTROL|SEC_STD_WRITE_DAC|\
1315                                SEC_STD_WRITE_OWNER                      /* 0xe0000 */
1316
1317 #if 0
1318 #define THIRD_DESIRED_ACCESS   FILE_READ_ATTRIBUTE|                   /* 0x80 */ \
1319                                READ_CONTROL|WRITE_DAC|\
1320                                SEC_FILE_READ_DATA|\
1321                                WRITE_OWNER                      /* */
1322 #endif
1323
1324 /*
1325   Test ntcreate calls made by xcopy
1326  */
1327 static BOOL run_xcopy(void)
1328 {
1329         struct smbcli_state *cli1;
1330         const char *fname = "\\test.txt";
1331         BOOL correct = True;
1332         int fnum1, fnum2;
1333
1334         printf("starting xcopy test\n");
1335
1336         if (!torture_open_connection(&cli1)) {
1337                 return False;
1338         }
1339
1340         fnum1 = smbcli_nt_create_full(cli1->tree, fname, 0,
1341                                       FIRST_DESIRED_ACCESS,
1342                                       FILE_ATTRIBUTE_ARCHIVE,
1343                                       NTCREATEX_SHARE_ACCESS_NONE,
1344                                       NTCREATEX_DISP_OVERWRITE_IF,
1345                                       0x4044, 0);
1346
1347         if (fnum1 == -1) {
1348                 printf("First open failed - %s\n", smbcli_errstr(cli1->tree));
1349                 return False;
1350         }
1351
1352         fnum2 = smbcli_nt_create_full(cli1->tree, fname, 0,
1353                                    SECOND_DESIRED_ACCESS, 0,
1354                                    NTCREATEX_SHARE_ACCESS_READ|NTCREATEX_SHARE_ACCESS_WRITE|NTCREATEX_SHARE_ACCESS_DELETE, NTCREATEX_DISP_OPEN,
1355                                    0x200000, 0);
1356         if (fnum2 == -1) {
1357                 printf("second open failed - %s\n", smbcli_errstr(cli1->tree));
1358                 return False;
1359         }
1360
1361         if (!torture_close_connection(cli1)) {
1362                 correct = False;
1363         }
1364
1365         return correct;
1366 }
1367
1368
1369 /*
1370   see how many RPC pipes we can open at once
1371 */
1372 static BOOL run_pipe_number(void)
1373 {
1374         struct smbcli_state *cli1;
1375         const char *pipe_name = "\\WKSSVC";
1376         int fnum;
1377         int num_pipes = 0;
1378
1379         printf("starting pipenumber test\n");
1380         if (!torture_open_connection(&cli1)) {
1381                 return False;
1382         }
1383
1384         while(1) {
1385                 fnum = smbcli_nt_create_full(cli1->tree, pipe_name, 0, SEC_FILE_READ_DATA, FILE_ATTRIBUTE_NORMAL,
1386                                    NTCREATEX_SHARE_ACCESS_READ|NTCREATEX_SHARE_ACCESS_WRITE, NTCREATEX_DISP_OPEN_IF, 0, 0);
1387
1388                 if (fnum == -1) {
1389                         printf("Open of pipe %s failed with error (%s)\n", pipe_name, smbcli_errstr(cli1->tree));
1390                         break;
1391                 }
1392                 num_pipes++;
1393                 printf("%d\r", num_pipes);
1394                 fflush(stdout);
1395         }
1396
1397         printf("pipe_number test - we can open %d %s pipes.\n", num_pipes, pipe_name );
1398         torture_close_connection(cli1);
1399         return True;
1400 }
1401
1402
1403
1404
1405 /*
1406   open N connections to the server and just hold them open
1407   used for testing performance when there are N idle users
1408   already connected
1409  */
1410  static BOOL torture_holdcon(void)
1411 {
1412         int i;
1413         struct smbcli_state **cli;
1414         int num_dead = 0;
1415
1416         printf("Opening %d connections\n", torture_numops);
1417
1418         cli = malloc_array_p(struct smbcli_state *, torture_numops);
1419
1420         for (i=0;i<torture_numops;i++) {
1421                 if (!torture_open_connection(&cli[i])) {
1422                         return False;
1423                 }
1424                 printf("opened %d connections\r", i);
1425                 fflush(stdout);
1426         }
1427
1428         printf("\nStarting pings\n");
1429
1430         while (1) {
1431                 for (i=0;i<torture_numops;i++) {
1432                         NTSTATUS status;
1433                         if (cli[i]) {
1434                                 status = smbcli_chkpath(cli[i]->tree, "\\");
1435                                 if (!NT_STATUS_IS_OK(status)) {
1436                                         printf("Connection %d is dead\n", i);
1437                                         cli[i] = NULL;
1438                                         num_dead++;
1439                                 }
1440                                 usleep(100);
1441                         }
1442                 }
1443
1444                 if (num_dead == torture_numops) {
1445                         printf("All connections dead - finishing\n");
1446                         break;
1447                 }
1448
1449                 printf(".");
1450                 fflush(stdout);
1451         }
1452
1453         return True;
1454 }
1455
1456 /*
1457   Try with a wrong vuid and check error message.
1458  */
1459
1460 static BOOL run_vuidtest(void)
1461 {
1462         struct smbcli_state *cli;
1463         const char *fname = "\\vuid.tst";
1464         int fnum;
1465         size_t size;
1466         time_t c_time, a_time, m_time;
1467         BOOL correct = True;
1468
1469         uint16_t orig_vuid;
1470         NTSTATUS result;
1471
1472         printf("starting vuid test\n");
1473
1474         if (!torture_open_connection(&cli)) {
1475                 return False;
1476         }
1477
1478         smbcli_unlink(cli->tree, fname);
1479
1480         fnum = smbcli_open(cli->tree, fname,
1481                         O_RDWR | O_CREAT | O_TRUNC, DENY_NONE);
1482
1483         orig_vuid = cli->session->vuid;
1484
1485         cli->session->vuid += 1234;
1486
1487         printf("Testing qfileinfo with wrong vuid\n");
1488
1489         if (NT_STATUS_IS_OK(result = smbcli_qfileinfo(cli->tree, fnum, NULL,
1490                                                    &size, &c_time, &a_time,
1491                                                    &m_time, NULL, NULL))) {
1492                 printf("ERROR: qfileinfo passed with wrong vuid\n");
1493                 correct = False;
1494         }
1495
1496         if ( (cli->transport->error.etype != ETYPE_DOS) ||
1497              (cli->transport->error.e.dos.eclass != ERRSRV) ||
1498              (cli->transport->error.e.dos.ecode != ERRbaduid) ) {
1499                 printf("ERROR: qfileinfo should have returned DOS error "
1500                        "ERRSRV:ERRbaduid\n  but returned %s\n",
1501                        smbcli_errstr(cli->tree));
1502                 correct = False;
1503         }
1504
1505         cli->session->vuid -= 1234;
1506
1507         if (NT_STATUS_IS_ERR(smbcli_close(cli->tree, fnum))) {
1508                 printf("close failed (%s)\n", smbcli_errstr(cli->tree));
1509                 correct = False;
1510         }
1511
1512         smbcli_unlink(cli->tree, fname);
1513
1514         if (!torture_close_connection(cli)) {
1515                 correct = False;
1516         }
1517
1518         printf("vuid test finished\n");
1519
1520         return correct;
1521 }
1522
1523 /*
1524   Test open mode returns on read-only files.
1525  */
1526  static BOOL run_opentest(void)
1527 {
1528         static struct smbcli_state *cli1;
1529         static struct smbcli_state *cli2;
1530         const char *fname = "\\readonly.file";
1531         int fnum1, fnum2;
1532         uint8_t buf[20];
1533         size_t fsize;
1534         BOOL correct = True;
1535         char *tmp_path;
1536         int failures = 0;
1537
1538         printf("starting open test\n");
1539
1540         if (!torture_open_connection(&cli1)) {
1541                 return False;
1542         }
1543
1544         smbcli_setatr(cli1->tree, fname, 0, 0);
1545         smbcli_unlink(cli1->tree, fname);
1546
1547         fnum1 = smbcli_open(cli1->tree, fname, O_RDWR|O_CREAT|O_EXCL, DENY_NONE);
1548         if (fnum1 == -1) {
1549                 printf("open of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1550                 return False;
1551         }
1552
1553         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1554                 printf("close2 failed (%s)\n", smbcli_errstr(cli1->tree));
1555                 return False;
1556         }
1557
1558         if (NT_STATUS_IS_ERR(smbcli_setatr(cli1->tree, fname, FILE_ATTRIBUTE_READONLY, 0))) {
1559                 printf("smbcli_setatr failed (%s)\n", smbcli_errstr(cli1->tree));
1560                 CHECK_MAX_FAILURES(error_test1);
1561                 return False;
1562         }
1563
1564         fnum1 = smbcli_open(cli1->tree, fname, O_RDONLY, DENY_WRITE);
1565         if (fnum1 == -1) {
1566                 printf("open of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1567                 CHECK_MAX_FAILURES(error_test1);
1568                 return False;
1569         }
1570
1571         /* This will fail - but the error should be ERRnoaccess, not ERRbadshare. */
1572         fnum2 = smbcli_open(cli1->tree, fname, O_RDWR, DENY_ALL);
1573
1574         if (check_error(__location__, cli1, ERRDOS, ERRnoaccess,
1575                         NT_STATUS_ACCESS_DENIED)) {
1576                 printf("correct error code ERRDOS/ERRnoaccess returned\n");
1577         }
1578
1579         printf("finished open test 1\n");
1580 error_test1:
1581         smbcli_close(cli1->tree, fnum1);
1582
1583         /* Now try not readonly and ensure ERRbadshare is returned. */
1584
1585         smbcli_setatr(cli1->tree, fname, 0, 0);
1586
1587         fnum1 = smbcli_open(cli1->tree, fname, O_RDONLY, DENY_WRITE);
1588         if (fnum1 == -1) {
1589                 printf("open of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1590                 return False;
1591         }
1592
1593         /* This will fail - but the error should be ERRshare. */
1594         fnum2 = smbcli_open(cli1->tree, fname, O_RDWR, DENY_ALL);
1595
1596         if (check_error(__location__, cli1, ERRDOS, ERRbadshare,
1597                         NT_STATUS_SHARING_VIOLATION)) {
1598                 printf("correct error code ERRDOS/ERRbadshare returned\n");
1599         }
1600
1601         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1602                 printf("close2 failed (%s)\n", smbcli_errstr(cli1->tree));
1603                 return False;
1604         }
1605
1606         smbcli_unlink(cli1->tree, fname);
1607
1608         printf("finished open test 2\n");
1609
1610         /* Test truncate open disposition on file opened for read. */
1611
1612         fnum1 = smbcli_open(cli1->tree, fname, O_RDWR|O_CREAT|O_EXCL, DENY_NONE);
1613         if (fnum1 == -1) {
1614                 printf("(3) open (1) of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1615                 return False;
1616         }
1617
1618         /* write 20 bytes. */
1619
1620         memset(buf, '\0', 20);
1621
1622         if (smbcli_write(cli1->tree, fnum1, 0, buf, 0, 20) != 20) {
1623                 printf("write failed (%s)\n", smbcli_errstr(cli1->tree));
1624                 correct = False;
1625         }
1626
1627         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1628                 printf("(3) close1 failed (%s)\n", smbcli_errstr(cli1->tree));
1629                 return False;
1630         }
1631
1632         /* Ensure size == 20. */
1633         if (NT_STATUS_IS_ERR(smbcli_getatr(cli1->tree, fname, NULL, &fsize, NULL))) {
1634                 printf("(3) getatr failed (%s)\n", smbcli_errstr(cli1->tree));
1635                 CHECK_MAX_FAILURES(error_test3);
1636                 return False;
1637         }
1638
1639         if (fsize != 20) {
1640                 printf("(3) file size != 20\n");
1641                 CHECK_MAX_FAILURES(error_test3);
1642                 return False;
1643         }
1644
1645         /* Now test if we can truncate a file opened for readonly. */
1646
1647         fnum1 = smbcli_open(cli1->tree, fname, O_RDONLY|O_TRUNC, DENY_NONE);
1648         if (fnum1 == -1) {
1649                 printf("(3) open (2) of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1650                 CHECK_MAX_FAILURES(error_test3);
1651                 return False;
1652         }
1653
1654         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1655                 printf("close2 failed (%s)\n", smbcli_errstr(cli1->tree));
1656                 return False;
1657         }
1658
1659         /* Ensure size == 0. */
1660         if (NT_STATUS_IS_ERR(smbcli_getatr(cli1->tree, fname, NULL, &fsize, NULL))) {
1661                 printf("(3) getatr failed (%s)\n", smbcli_errstr(cli1->tree));
1662                 CHECK_MAX_FAILURES(error_test3);
1663                 return False;
1664         }
1665
1666         if (fsize != 0) {
1667                 printf("(3) file size != 0\n");
1668                 CHECK_MAX_FAILURES(error_test3);
1669                 return False;
1670         }
1671         printf("finished open test 3\n");
1672 error_test3:
1673         smbcli_unlink(cli1->tree, fname);
1674
1675
1676         printf("testing ctemp\n");
1677         fnum1 = smbcli_ctemp(cli1->tree, "\\", &tmp_path);
1678         if (fnum1 == -1) {
1679                 printf("ctemp failed (%s)\n", smbcli_errstr(cli1->tree));
1680                 CHECK_MAX_FAILURES(error_test4);
1681                 return False;
1682         }
1683         printf("ctemp gave path %s\n", tmp_path);
1684         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1685                 printf("close of temp failed (%s)\n", smbcli_errstr(cli1->tree));
1686         }
1687         if (NT_STATUS_IS_ERR(smbcli_unlink(cli1->tree, tmp_path))) {
1688                 printf("unlink of temp failed (%s)\n", smbcli_errstr(cli1->tree));
1689         }
1690 error_test4:
1691         /* Test the non-io opens... */
1692
1693         if (!torture_open_connection(&cli2)) {
1694                 return False;
1695         }
1696
1697         smbcli_setatr(cli2->tree, fname, 0, 0);
1698         smbcli_unlink(cli2->tree, fname);
1699
1700         printf("TEST #1 testing 2 non-io opens (no delete)\n");
1701
1702         fnum1 = smbcli_nt_create_full(cli1->tree, fname, 0, SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1703                                    NTCREATEX_SHARE_ACCESS_NONE, NTCREATEX_DISP_OVERWRITE_IF, 0, 0);
1704
1705         if (fnum1 == -1) {
1706                 printf("test 1 open 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1707                 CHECK_MAX_FAILURES(error_test10);
1708                 return False;
1709         }
1710
1711         fnum2 = smbcli_nt_create_full(cli2->tree, fname, 0, SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1712                                    NTCREATEX_SHARE_ACCESS_NONE, NTCREATEX_DISP_OPEN_IF, 0, 0);
1713         if (fnum2 == -1) {
1714                 printf("test 1 open 2 of %s failed (%s)\n", fname, smbcli_errstr(cli2->tree));
1715                 CHECK_MAX_FAILURES(error_test10);
1716                 return False;
1717         }
1718
1719         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1720                 printf("test 1 close 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1721                 return False;
1722         }
1723         if (NT_STATUS_IS_ERR(smbcli_close(cli2->tree, fnum2))) {
1724                 printf("test 1 close 2 of %s failed (%s)\n", fname, smbcli_errstr(cli2->tree));
1725                 return False;
1726         }
1727
1728         printf("non-io open test #1 passed.\n");
1729 error_test10:
1730         smbcli_unlink(cli1->tree, fname);
1731
1732         printf("TEST #2 testing 2 non-io opens (first with delete)\n");
1733
1734         fnum1 = smbcli_nt_create_full(cli1->tree, fname, 0, SEC_STD_DELETE|SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1735                                    NTCREATEX_SHARE_ACCESS_NONE, NTCREATEX_DISP_OVERWRITE_IF, 0, 0);
1736
1737         if (fnum1 == -1) {
1738                 printf("test 2 open 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1739                 CHECK_MAX_FAILURES(error_test20);
1740                 return False;
1741         }
1742
1743         fnum2 = smbcli_nt_create_full(cli2->tree, fname, 0, SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1744                                    NTCREATEX_SHARE_ACCESS_NONE, NTCREATEX_DISP_OPEN_IF, 0, 0);
1745
1746         if (fnum2 == -1) {
1747                 printf("test 2 open 2 of %s failed (%s)\n", fname, smbcli_errstr(cli2->tree));
1748                 CHECK_MAX_FAILURES(error_test20);
1749                 return False;
1750         }
1751
1752         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1753                 printf("test 1 close 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1754                 return False;
1755         }
1756         if (NT_STATUS_IS_ERR(smbcli_close(cli2->tree, fnum2))) {
1757                 printf("test 1 close 2 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1758                 return False;
1759         }
1760
1761         printf("non-io open test #2 passed.\n");
1762 error_test20:
1763         smbcli_unlink(cli1->tree, fname);
1764
1765         printf("TEST #3 testing 2 non-io opens (second with delete)\n");
1766
1767         fnum1 = smbcli_nt_create_full(cli1->tree, fname, 0, SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1768                                    NTCREATEX_SHARE_ACCESS_NONE, NTCREATEX_DISP_OVERWRITE_IF, 0, 0);
1769
1770         if (fnum1 == -1) {
1771                 printf("test 3 open 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1772                 CHECK_MAX_FAILURES(error_test30);
1773                 return False;
1774         }
1775
1776         fnum2 = smbcli_nt_create_full(cli2->tree, fname, 0, SEC_STD_DELETE|SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1777                                    NTCREATEX_SHARE_ACCESS_NONE, NTCREATEX_DISP_OPEN_IF, 0, 0);
1778
1779         if (fnum2 == -1) {
1780                 printf("test 3 open 2 of %s failed (%s)\n", fname, smbcli_errstr(cli2->tree));
1781                 CHECK_MAX_FAILURES(error_test30);
1782                 return False;
1783         }
1784
1785         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1786                 printf("test 3 close 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1787                 return False;
1788         }
1789         if (NT_STATUS_IS_ERR(smbcli_close(cli2->tree, fnum2))) {
1790                 printf("test 3 close 2 of %s failed (%s)\n", fname, smbcli_errstr(cli2->tree));
1791                 return False;
1792         }
1793
1794         printf("non-io open test #3 passed.\n");
1795 error_test30:
1796         smbcli_unlink(cli1->tree, fname);
1797
1798         printf("TEST #4 testing 2 non-io opens (both with delete)\n");
1799
1800         fnum1 = smbcli_nt_create_full(cli1->tree, fname, 0, SEC_STD_DELETE|SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1801                                    NTCREATEX_SHARE_ACCESS_NONE, NTCREATEX_DISP_OVERWRITE_IF, 0, 0);
1802
1803         if (fnum1 == -1) {
1804                 printf("test 4 open 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1805                 CHECK_MAX_FAILURES(error_test40);
1806                 return False;
1807         }
1808
1809         fnum2 = smbcli_nt_create_full(cli2->tree, fname, 0, SEC_STD_DELETE|SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1810                                    NTCREATEX_SHARE_ACCESS_NONE, NTCREATEX_DISP_OPEN_IF, 0, 0);
1811
1812         if (fnum2 != -1) {
1813                 printf("test 4 open 2 of %s SUCCEEDED - should have failed (%s)\n", fname, smbcli_errstr(cli2->tree));
1814                 CHECK_MAX_FAILURES(error_test40);
1815                 return False;
1816         }
1817
1818         printf("test 4 open 2 of %s gave %s (correct error should be %s)\n", fname, smbcli_errstr(cli2->tree), "sharing violation");
1819
1820         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1821                 printf("test 4 close 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1822                 return False;
1823         }
1824
1825         printf("non-io open test #4 passed.\n");
1826 error_test40:
1827         smbcli_unlink(cli1->tree, fname);
1828
1829         printf("TEST #5 testing 2 non-io opens (both with delete - both with file share delete)\n");
1830
1831         fnum1 = smbcli_nt_create_full(cli1->tree, fname, 0, SEC_STD_DELETE|SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1832                                    NTCREATEX_SHARE_ACCESS_DELETE, NTCREATEX_DISP_OVERWRITE_IF, 0, 0);
1833
1834         if (fnum1 == -1) {
1835                 printf("test 5 open 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1836                 CHECK_MAX_FAILURES(error_test50);
1837                 return False;
1838         }
1839
1840         fnum2 = smbcli_nt_create_full(cli2->tree, fname, 0, SEC_STD_DELETE|SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1841                                    NTCREATEX_SHARE_ACCESS_DELETE, NTCREATEX_DISP_OPEN_IF, 0, 0);
1842
1843         if (fnum2 == -1) {
1844                 printf("test 5 open 2 of %s failed (%s)\n", fname, smbcli_errstr(cli2->tree));
1845                 CHECK_MAX_FAILURES(error_test50);
1846                 return False;
1847         }
1848
1849         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1850                 printf("test 5 close 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1851                 return False;
1852         }
1853
1854         if (NT_STATUS_IS_ERR(smbcli_close(cli2->tree, fnum2))) {
1855                 printf("test 5 close 2 of %s failed (%s)\n", fname, smbcli_errstr(cli2->tree));
1856                 return False;
1857         }
1858
1859         printf("non-io open test #5 passed.\n");
1860 error_test50:
1861         printf("TEST #6 testing 1 non-io open, one io open\n");
1862
1863         smbcli_unlink(cli1->tree, fname);
1864
1865         fnum1 = smbcli_nt_create_full(cli1->tree, fname, 0, SEC_FILE_READ_DATA, FILE_ATTRIBUTE_NORMAL,
1866                                    NTCREATEX_SHARE_ACCESS_NONE, NTCREATEX_DISP_OVERWRITE_IF, 0, 0);
1867
1868         if (fnum1 == -1) {
1869                 printf("test 6 open 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1870                 CHECK_MAX_FAILURES(error_test60);
1871                 return False;
1872         }
1873
1874         fnum2 = smbcli_nt_create_full(cli2->tree, fname, 0, SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1875                                    NTCREATEX_SHARE_ACCESS_READ, NTCREATEX_DISP_OPEN_IF, 0, 0);
1876
1877         if (fnum2 == -1) {
1878                 printf("test 6 open 2 of %s failed (%s)\n", fname, smbcli_errstr(cli2->tree));
1879                 CHECK_MAX_FAILURES(error_test60);
1880                 return False;
1881         }
1882
1883         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1884                 printf("test 6 close 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1885                 return False;
1886         }
1887
1888         if (NT_STATUS_IS_ERR(smbcli_close(cli2->tree, fnum2))) {
1889                 printf("test 6 close 2 of %s failed (%s)\n", fname, smbcli_errstr(cli2->tree));
1890                 return False;
1891         }
1892
1893         printf("non-io open test #6 passed.\n");
1894 error_test60:
1895         printf("TEST #7 testing 1 non-io open, one io open with delete\n");
1896
1897         smbcli_unlink(cli1->tree, fname);
1898
1899         fnum1 = smbcli_nt_create_full(cli1->tree, fname, 0, SEC_FILE_READ_DATA, FILE_ATTRIBUTE_NORMAL,
1900                                    NTCREATEX_SHARE_ACCESS_NONE, NTCREATEX_DISP_OVERWRITE_IF, 0, 0);
1901
1902         if (fnum1 == -1) {
1903                 printf("test 7 open 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1904                 CHECK_MAX_FAILURES(error_test70);
1905                 return False;
1906         }
1907
1908         fnum2 = smbcli_nt_create_full(cli2->tree, fname, 0, SEC_STD_DELETE|SEC_FILE_READ_ATTRIBUTE, FILE_ATTRIBUTE_NORMAL,
1909                                    NTCREATEX_SHARE_ACCESS_READ|NTCREATEX_SHARE_ACCESS_DELETE, NTCREATEX_DISP_OPEN_IF, 0, 0);
1910
1911         if (fnum2 != -1) {
1912                 printf("test 7 open 2 of %s SUCCEEDED - should have failed (%s)\n", fname, smbcli_errstr(cli2->tree));
1913                 CHECK_MAX_FAILURES(error_test70);
1914                 return False;
1915         }
1916
1917         printf("test 7 open 2 of %s gave %s (correct error should be %s)\n", fname, smbcli_errstr(cli2->tree), "sharing violation");
1918
1919         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1920                 printf("test 7 close 1 of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1921                 return False;
1922         }
1923
1924         printf("non-io open test #7 passed.\n");
1925
1926 error_test70:
1927
1928         printf("TEST #8 testing one normal open, followed by lock, followed by open with truncate\n");
1929
1930         smbcli_unlink(cli1->tree, fname);
1931
1932         fnum1 = smbcli_open(cli1->tree, fname, O_RDWR|O_CREAT, DENY_NONE);
1933         if (fnum1 == -1) {
1934                 printf("(8) open (1) of %s failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1935                 return False;
1936         }
1937
1938         /* write 20 bytes. */
1939
1940         memset(buf, '\0', 20);
1941
1942         if (smbcli_write(cli1->tree, fnum1, 0, buf, 0, 20) != 20) {
1943                 printf("(8) write failed (%s)\n", smbcli_errstr(cli1->tree));
1944                 correct = False;
1945         }
1946
1947         /* Ensure size == 20. */
1948         if (NT_STATUS_IS_ERR(smbcli_getatr(cli1->tree, fname, NULL, &fsize, NULL))) {
1949                 printf("(8) getatr (1) failed (%s)\n", smbcli_errstr(cli1->tree));
1950                 CHECK_MAX_FAILURES(error_test80);
1951                 return False;
1952         }
1953
1954         if (fsize != 20) {
1955                 printf("(8) file size != 20\n");
1956                 CHECK_MAX_FAILURES(error_test80);
1957                 return False;
1958         }
1959
1960         /* Get an exclusive lock on the open file. */
1961         if (NT_STATUS_IS_ERR(smbcli_lock(cli1->tree, fnum1, 0, 4, 0, WRITE_LOCK))) {
1962                 printf("(8) lock1 failed (%s)\n", smbcli_errstr(cli1->tree));
1963                 CHECK_MAX_FAILURES(error_test80);
1964                 return False;
1965         }
1966
1967         fnum2 = smbcli_open(cli1->tree, fname, O_RDWR|O_TRUNC, DENY_NONE);
1968         if (fnum1 == -1) {
1969                 printf("(8) open (2) of %s with truncate failed (%s)\n", fname, smbcli_errstr(cli1->tree));
1970                 return False;
1971         }
1972
1973         /* Ensure size == 0. */
1974         if (NT_STATUS_IS_ERR(smbcli_getatr(cli1->tree, fname, NULL, &fsize, NULL))) {
1975                 printf("(8) getatr (2) failed (%s)\n", smbcli_errstr(cli1->tree));
1976                 CHECK_MAX_FAILURES(error_test80);
1977                 return False;
1978         }
1979
1980         if (fsize != 0) {
1981                 printf("(8) file size != 0\n");
1982                 CHECK_MAX_FAILURES(error_test80);
1983                 return False;
1984         }
1985
1986         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum1))) {
1987                 printf("(8) close1 failed (%s)\n", smbcli_errstr(cli1->tree));
1988                 return False;
1989         }
1990
1991         if (NT_STATUS_IS_ERR(smbcli_close(cli1->tree, fnum2))) {
1992                 printf("(8) close1 failed (%s)\n", smbcli_errstr(cli1->tree));
1993                 return False;
1994         }
1995
1996 error_test80:
1997
1998         printf("open test #8 passed.\n");
1999
2000         smbcli_unlink(cli1->tree, fname);
2001
2002         if (!torture_close_connection(cli1)) {
2003                 correct = False;
2004         }
2005         if (!torture_close_connection(cli2)) {
2006                 correct = False;
2007         }
2008
2009         return correct;
2010 }
2011
2012
2013 /*
2014   sees what IOCTLs are supported
2015  */
2016 BOOL torture_ioctl_test(void)
2017 {
2018         struct smbcli_state *cli;
2019         uint16_t device, function;
2020         int fnum;
2021         const char *fname = "\\ioctl.dat";
2022         NTSTATUS status;
2023         union smb_ioctl parms;
2024         TALLOC_CTX *mem_ctx;
2025
2026         if (!torture_open_connection(&cli)) {
2027                 return False;
2028         }
2029
2030         mem_ctx = talloc_init("ioctl_test");
2031
2032         printf("starting ioctl test\n");
2033
2034         smbcli_unlink(cli->tree, fname);
2035
2036         fnum = smbcli_open(cli->tree, fname, O_RDWR|O_CREAT|O_EXCL, DENY_NONE);
2037         if (fnum == -1) {
2038                 printf("open of %s failed (%s)\n", fname, smbcli_errstr(cli->tree));
2039                 return False;
2040         }
2041
2042         parms.ioctl.level = RAW_IOCTL_IOCTL;
2043         parms.ioctl.in.fnum = fnum;
2044         parms.ioctl.in.request = IOCTL_QUERY_JOB_INFO;
2045         status = smb_raw_ioctl(cli->tree, mem_ctx, &parms);
2046         printf("ioctl job info: %s\n", smbcli_errstr(cli->tree));
2047
2048         for (device=0;device<0x100;device++) {
2049                 printf("testing device=0x%x\n", device);
2050                 for (function=0;function<0x100;function++) {
2051                         parms.ioctl.in.request = (device << 16) | function;
2052                         status = smb_raw_ioctl(cli->tree, mem_ctx, &parms);
2053
2054                         if (NT_STATUS_IS_OK(status)) {
2055                                 printf("ioctl device=0x%x function=0x%x OK : %d bytes\n",
2056                                         device, function, parms.ioctl.out.blob.length);
2057                         }
2058                 }
2059         }
2060
2061         if (!torture_close_connection(cli)) {
2062                 return False;
2063         }
2064
2065         return True;
2066 }
2067
2068
2069 /*
2070   tries variants of chkpath
2071  */
2072 BOOL torture_chkpath_test(void)
2073 {
2074         struct smbcli_state *cli;
2075         int fnum;
2076         BOOL ret;
2077
2078         if (!torture_open_connection(&cli)) {
2079                 return False;
2080         }
2081
2082         printf("starting chkpath test\n");
2083
2084         printf("Testing valid and invalid paths\n");
2085
2086         /* cleanup from an old run */
2087         smbcli_rmdir(cli->tree, "\\chkpath.dir\\dir2");
2088         smbcli_unlink(cli->tree, "\\chkpath.dir\\*");
2089         smbcli_rmdir(cli->tree, "\\chkpath.dir");
2090
2091         if (NT_STATUS_IS_ERR(smbcli_mkdir(cli->tree, "\\chkpath.dir"))) {
2092                 printf("mkdir1 failed : %s\n", smbcli_errstr(cli->tree));
2093                 return False;
2094         }
2095
2096         if (NT_STATUS_IS_ERR(smbcli_mkdir(cli->tree, "\\chkpath.dir\\dir2"))) {
2097                 printf("mkdir2 failed : %s\n", smbcli_errstr(cli->tree));
2098                 return False;
2099         }
2100
2101         fnum = smbcli_open(cli->tree, "\\chkpath.dir\\foo.txt", O_RDWR|O_CREAT|O_EXCL, DENY_NONE);
2102         if (fnum == -1) {
2103                 printf("open1 failed (%s)\n", smbcli_errstr(cli->tree));
2104                 return False;
2105         }
2106         smbcli_close(cli->tree, fnum);
2107
2108         if (NT_STATUS_IS_ERR(smbcli_chkpath(cli->tree, "\\chkpath.dir"))) {
2109                 printf("chkpath1 failed: %s\n", smbcli_errstr(cli->tree));
2110                 ret = False;
2111         }
2112
2113         if (NT_STATUS_IS_ERR(smbcli_chkpath(cli->tree, "\\chkpath.dir\\dir2"))) {
2114                 printf("chkpath2 failed: %s\n", smbcli_errstr(cli->tree));
2115                 ret = False;
2116         }
2117
2118         if (NT_STATUS_IS_ERR(smbcli_chkpath(cli->tree, "\\chkpath.dir\\foo.txt"))) {
2119                 ret = check_error(__location__, cli, ERRDOS, ERRbadpath,
2120                                   NT_STATUS_NOT_A_DIRECTORY);
2121         } else {
2122                 printf("* chkpath on a file should fail\n");
2123                 ret = False;
2124         }
2125
2126         if (NT_STATUS_IS_ERR(smbcli_chkpath(cli->tree, "\\chkpath.dir\\bar.txt"))) {
2127                 ret = check_error(__location__, cli, ERRDOS, ERRbadfile,
2128                                   NT_STATUS_OBJECT_NAME_NOT_FOUND);
2129         } else {
2130                 printf("* chkpath on a non existent file should fail\n");
2131                 ret = False;
2132         }
2133
2134         if (NT_STATUS_IS_ERR(smbcli_chkpath(cli->tree, "\\chkpath.dir\\dirxx\\bar.txt"))) {
2135                 ret = check_error(__location__, cli, ERRDOS, ERRbadpath,
2136                                   NT_STATUS_OBJECT_PATH_NOT_FOUND);
2137         } else {
2138                 printf("* chkpath on a non existent component should fail\n");
2139                 ret = False;
2140         }
2141
2142         smbcli_rmdir(cli->tree, "\\chkpath.dir\\dir2");
2143         smbcli_unlink(cli->tree, "\\chkpath.dir\\*");
2144         smbcli_rmdir(cli->tree, "\\chkpath.dir");
2145
2146         if (!torture_close_connection(cli)) {
2147                 return False;
2148         }
2149
2150         return ret;
2151 }
2152
2153
2154 static void sigcont(int sig)
2155 {
2156 }
2157
2158 double torture_create_procs(BOOL (*fn)(struct smbcli_state *, int), BOOL *result)
2159 {
2160         int i, status;
2161         volatile pid_t *child_status;
2162         volatile BOOL *child_status_out;
2163         int synccount;
2164         int tries = 8;
2165         double start_time_limit = 10 + (torture_nprocs * 1.5);
2166         char **unc_list = NULL;
2167         const char *p;
2168         int num_unc_names = 0;
2169         struct timeval tv;
2170
2171         *result = True;
2172
2173         synccount = 0;
2174
2175         signal(SIGCONT, sigcont);
2176
2177         child_status = (volatile pid_t *)shm_setup(sizeof(pid_t)*torture_nprocs);
2178         if (!child_status) {
2179                 printf("Failed to setup shared memory\n");
2180                 return -1;
2181         }
2182
2183         child_status_out = (volatile BOOL *)shm_setup(sizeof(BOOL)*torture_nprocs);
2184         if (!child_status_out) {
2185                 printf("Failed to setup result status shared memory\n");
2186                 return -1;
2187         }
2188
2189         p = lp_parm_string(-1, "torture", "unclist");
2190         if (p) {
2191                 unc_list = file_lines_load(p, &num_unc_names);
2192                 if (!unc_list || num_unc_names <= 0) {
2193                         printf("Failed to load unc names list from '%s'\n", p);
2194                         exit(1);
2195                 }
2196         }
2197
2198         for (i = 0; i < torture_nprocs; i++) {
2199                 child_status[i] = 0;
2200                 child_status_out[i] = True;
2201         }
2202
2203         tv = timeval_current();
2204
2205         for (i=0;i<torture_nprocs;i++) {
2206                 procnum = i;
2207                 if (fork() == 0) {
2208                         char *myname;
2209                         const char *hostname=NULL, *sharename;
2210
2211                         pid_t mypid = getpid();
2212                         sys_srandom(((int)mypid) ^ ((int)time(NULL)));
2213
2214                         asprintf(&myname, "CLIENT%d", i);
2215                         lp_set_cmdline("netbios name", myname);
2216                         free(myname);
2217
2218
2219                         if (unc_list) {
2220                                 if (!smbcli_parse_unc(unc_list[i % num_unc_names],
2221                                                       NULL, &hostname, &sharename)) {
2222                                         printf("Failed to parse UNC name %s\n",
2223                                                unc_list[i % num_unc_names]);
2224                                         exit(1);
2225                                 }
2226                         }
2227
2228                         while (1) {
2229                                 if (hostname) {
2230                                         if (torture_open_connection_share(&current_cli,
2231                                                                           hostname,
2232                                                                           sharename)) {
2233                                                 break;
2234                                         }
2235                                 } else if (torture_open_connection(&current_cli)) {
2236                                                 break;
2237                                 }
2238                                 if (tries-- == 0) {
2239                                         printf("pid %d failed to start\n", (int)getpid());
2240                                         _exit(1);
2241                                 }
2242                                 msleep(100);
2243                         }
2244
2245                         child_status[i] = getpid();
2246
2247                         pause();
2248
2249                         if (child_status[i]) {
2250                                 printf("Child %d failed to start!\n", i);
2251                                 child_status_out[i] = 1;
2252                                 _exit(1);
2253                         }
2254
2255                         child_status_out[i] = fn(current_cli, i);
2256                         _exit(0);
2257                 }
2258         }
2259
2260         do {
2261                 synccount = 0;
2262                 for (i=0;i<torture_nprocs;i++) {
2263                         if (child_status[i]) synccount++;
2264                 }
2265                 if (synccount == torture_nprocs) break;
2266                 msleep(100);
2267         } while (timeval_elapsed(&tv) < start_time_limit);
2268
2269         if (synccount != torture_nprocs) {
2270                 printf("FAILED TO START %d CLIENTS (started %d)\n", torture_nprocs, synccount);
2271                 *result = False;
2272                 return timeval_elapsed(&tv);
2273         }
2274
2275         printf("Starting %d clients\n", torture_nprocs);
2276
2277         /* start the client load */
2278         tv = timeval_current();
2279         for (i=0;i<torture_nprocs;i++) {
2280                 child_status[i] = 0;
2281         }
2282
2283         printf("%d clients started\n", torture_nprocs);
2284
2285         kill(0, SIGCONT);
2286
2287         for (i=0;i<torture_nprocs;i++) {
2288                 int ret;
2289                 while ((ret=sys_waitpid(0, &status, 0)) == -1 && errno == EINTR) /* noop */ ;
2290                 if (ret == -1 || WEXITSTATUS(status) != 0) {
2291                         *result = False;
2292                 }
2293         }
2294
2295         printf("\n");
2296
2297         for (i=0;i<torture_nprocs;i++) {
2298                 if (!child_status_out[i]) {
2299                         *result = False;
2300                 }
2301         }
2302         return timeval_elapsed(&tv);
2303 }
2304
2305 #define FLAG_MULTIPROC 1
2306
2307 static struct {
2308         const char *name;
2309         BOOL (*fn)(void);
2310         BOOL (*multi_fn)(struct smbcli_state *, int );
2311 } torture_ops[] = {
2312         /* base tests */
2313         {"BASE-FDPASS", run_fdpasstest, 0},
2314         {"BASE-LOCK1",  torture_locktest1,  0},
2315         {"BASE-LOCK2",  torture_locktest2,  0},
2316         {"BASE-LOCK3",  torture_locktest3,  0},
2317         {"BASE-LOCK4",  torture_locktest4,  0},
2318         {"BASE-LOCK5",  torture_locktest5,  0},
2319         {"BASE-LOCK6",  torture_locktest6,  0},
2320         {"BASE-LOCK7",  torture_locktest7,  0},
2321         {"BASE-UNLINK", torture_unlinktest, 0},
2322         {"BASE-ATTR",   run_attrtest,   0},
2323         {"BASE-TRANS2", run_trans2test, 0},
2324         {"BASE-NEGNOWAIT", run_negprot_nowait, 0},
2325         {"BASE-DIR1",  torture_dirtest1, 0},
2326         {"BASE-DIR2",  torture_dirtest2, 0},
2327         {"BASE-DENY1",  torture_denytest1, 0},
2328         {"BASE-DENY2",  torture_denytest2, 0},
2329         {"BASE-DENY3",  torture_denytest3, 0},
2330         {"BASE-DENYDOS",  torture_denydos_sharing, 0},
2331         {"BASE-NTDENY1",  NULL, torture_ntdenytest1},
2332         {"BASE-NTDENY2",  torture_ntdenytest2, 0},
2333         {"BASE-TCON",  run_tcon_test, 0},
2334         {"BASE-TCONDEV",  run_tcon_devtype_test, 0},
2335         {"BASE-VUID", run_vuidtest, 0},
2336         {"BASE-RW1",  run_readwritetest, 0},
2337         {"BASE-RW2",  NULL, run_readwritemulti},
2338         {"BASE-OPEN", run_opentest, 0},
2339         {"BASE-DEFER_OPEN", NULL, run_deferopen},
2340         {"BASE-XCOPY", run_xcopy, 0},
2341         {"BASE-RENAME", torture_test_rename, 0},
2342         {"BASE-DELETE", torture_test_delete, 0},
2343         {"BASE-PROPERTIES", torture_test_properties, 0},
2344         {"BASE-MANGLE", torture_mangle, 0},
2345         {"BASE-OPENATTR", torture_openattrtest, 0},
2346         {"BASE-CHARSET", torture_charset, 0},
2347         {"BASE-CHKPATH",  torture_chkpath_test, 0},
2348         {"BASE-SECLEAK",  torture_sec_leak, 0},
2349         {"BASE-DISCONNECT",  torture_disconnect, 0},
2350         {"BASE-DELAYWRITE", torture_delay_write, 0},
2351
2352         /* benchmarking tests */
2353         {"BENCH-HOLDCON",  torture_holdcon, 0},
2354         {"BENCH-NBENCH",  torture_nbench, 0},
2355         {"BENCH-TORTURE", NULL, run_torture},
2356
2357         /* RAW smb tests */
2358         {"RAW-QFSINFO", torture_raw_qfsinfo, 0},
2359         {"RAW-QFILEINFO", torture_raw_qfileinfo, 0},
2360         {"RAW-SFILEINFO", torture_raw_sfileinfo, 0},
2361         {"RAW-SFILEINFO-BUG", torture_raw_sfileinfo_bug, 0},
2362         {"RAW-SEARCH", torture_raw_search, 0},
2363         {"RAW-CLOSE", torture_raw_close, 0},
2364         {"RAW-OPEN", torture_raw_open, 0},
2365         {"RAW-MKDIR", torture_raw_mkdir, 0},
2366         {"RAW-OPLOCK", torture_raw_oplock, 0},
2367         {"RAW-NOTIFY", torture_raw_notify, 0},
2368         {"RAW-MUX", torture_raw_mux, 0},
2369         {"RAW-IOCTL", torture_raw_ioctl, 0},
2370         {"RAW-CHKPATH", torture_raw_chkpath, 0},
2371         {"RAW-UNLINK", torture_raw_unlink, 0},
2372         {"RAW-READ", torture_raw_read, 0},
2373         {"RAW-WRITE", torture_raw_write, 0},
2374         {"RAW-LOCK", torture_raw_lock, 0},
2375         {"RAW-CONTEXT", torture_raw_context, 0},
2376         {"RAW-RENAME", torture_raw_rename, 0},
2377         {"RAW-SEEK", torture_raw_seek, 0},
2378         {"RAW-EAS", torture_raw_eas, 0},
2379         {"RAW-STREAMS", torture_raw_streams, 0},
2380         {"RAW-ACLS", torture_raw_acls, 0},
2381         {"RAW-RAP", torture_raw_rap, 0},
2382         {"RAW-COMPOSITE", torture_raw_composite, 0},
2383
2384         /* protocol scanners */
2385         {"SCAN-TRANS2", torture_trans2_scan, 0},
2386         {"SCAN-NTTRANS", torture_nttrans_scan, 0},
2387         {"SCAN-ALIASES", torture_trans2_aliases, 0},
2388         {"SCAN-SMB", torture_smb_scan, 0},
2389         {"SCAN-MAXFID", NULL, run_maxfidtest},
2390         {"SCAN-UTABLE", torture_utable, 0},
2391         {"SCAN-CASETABLE", torture_casetable, 0},
2392         {"SCAN-PIPE_NUMBER", run_pipe_number, 0},
2393         {"SCAN-IOCTL",  torture_ioctl_test, 0},
2394
2395         /* rpc testers */
2396         {"RPC-LSA", torture_rpc_lsa, 0},
2397         {"RPC-ECHO", torture_rpc_echo, 0},
2398         {"RPC-DFS", torture_rpc_dfs, 0},
2399         {"RPC-SPOOLSS", torture_rpc_spoolss, 0},
2400         {"RPC-SAMR", torture_rpc_samr, 0},
2401         {"RPC-NETLOGON", torture_rpc_netlogon, 0},
2402         {"RPC-SAMLOGON", torture_rpc_samlogon, 0},
2403         {"RPC-SAMSYNC", torture_rpc_samsync, 0},
2404         {"RPC-SCHANNEL", torture_rpc_schannel, 0},
2405         {"RPC-WKSSVC", torture_rpc_wkssvc, 0},
2406         {"RPC-SRVSVC", torture_rpc_srvsvc, 0},
2407         {"RPC-SVCCTL", torture_rpc_svcctl, 0},
2408         {"RPC-ATSVC", torture_rpc_atsvc, 0},
2409         {"RPC-EVENTLOG", torture_rpc_eventlog, 0},
2410         {"RPC-EPMAPPER", torture_rpc_epmapper, 0},
2411         {"RPC-WINREG", torture_rpc_winreg, 0},
2412         {"RPC-OXIDRESOLVE", torture_rpc_oxidresolve, 0},
2413         {"RPC-REMACT", torture_rpc_remact, 0},
2414         {"RPC-MGMT", torture_rpc_mgmt, 0},
2415         {"RPC-SCANNER", torture_rpc_scanner, 0},
2416         {"RPC-AUTOIDL", torture_rpc_autoidl, 0},
2417         {"RPC-COUNTCALLS", torture_rpc_countcalls, 0},
2418         {"RPC-MULTIBIND", torture_multi_bind, 0},
2419         {"RPC-DRSUAPI", torture_rpc_drsuapi, 0},
2420         {"RPC-LOGIN", torture_rpc_login, 0},
2421         {"RPC-ROT", torture_rpc_rot, 0},
2422         {"RPC-DSSETUP", torture_rpc_dssetup, 0},
2423         {"RPC-ALTERCONTEXT", torture_rpc_alter_context, 0},
2424
2425         /* Distributed COM testers */
2426         {"DCOM-SIMPLE", torture_dcom_simple, 0},
2427
2428         /* local (no server) testers */
2429         {"LOCAL-NTLMSSP", torture_ntlmssp_self_check, 0},
2430         {"LOCAL-ICONV", torture_local_iconv, 0},
2431         {"LOCAL-TALLOC", torture_local_talloc, 0},
2432         {"LOCAL-MESSAGING", torture_local_messaging, 0},
2433         {"LOCAL-BINDING", torture_local_binding_string, 0},
2434         {"LOCAL-IDTREE", torture_local_idtree, 0},
2435
2436         /* ldap testers */
2437         {"LDAP-BASIC", torture_ldap_basic, 0},
2438
2439         {NULL, NULL, 0}};
2440
2441
2442
2443 /****************************************************************************
2444 run a specified test or "ALL"
2445 ****************************************************************************/
2446 static BOOL run_test(const char *name)
2447 {
2448         BOOL ret = True;
2449         int i;
2450         BOOL matched = False;
2451
2452         if (strequal(name,"ALL")) {
2453                 for (i=0;torture_ops[i].name;i++) {
2454                         if (!run_test(torture_ops[i].name)) {
2455                                 ret = False;
2456                         }
2457                 }
2458                 return ret;
2459         }
2460
2461         for (i=0;torture_ops[i].name;i++) {
2462                 if (gen_fnmatch(name, torture_ops[i].name) == 0) {
2463                         double t;
2464                         matched = True;
2465                         init_iconv();
2466                         printf("Running %s\n", torture_ops[i].name);
2467                         if (torture_ops[i].multi_fn) {
2468                                 BOOL result = False;
2469                                 t = torture_create_procs(torture_ops[i].multi_fn,
2470                                                          &result);
2471                                 if (!result) {
2472                                         ret = False;
2473                                         printf("TEST %s FAILED!\n", torture_ops[i].name);
2474                                 }
2475
2476                         } else {
2477                                 struct timeval tv = timeval_current();
2478                                 if (!torture_ops[i].fn()) {
2479                                         ret = False;
2480                                         printf("TEST %s FAILED!\n", torture_ops[i].name);
2481                                 }
2482                                 t = timeval_elapsed(&tv);
2483                         }
2484                         printf("%s took %g secs\n\n", torture_ops[i].name, t);
2485                 }
2486         }
2487
2488         if (!matched) {
2489                 printf("Unknown torture operation '%s'\n", name);
2490         }
2491
2492         return ret;
2493 }
2494
2495
2496 static void parse_dns(const char *dns)
2497 {
2498         char *userdn, *basedn, *secret;
2499         char *p, *d;
2500
2501         /* retrievieng the userdn */
2502         p = strchr_m(dns, '#');
2503         if (!p) {
2504                 lp_set_cmdline("torture:ldap_userdn", "");
2505                 lp_set_cmdline("torture:ldap_basedn", "");
2506                 lp_set_cmdline("torture:ldap_secret", "");
2507                 return;
2508         }
2509         userdn = strndup(dns, p - dns);
2510         lp_set_cmdline("torture:ldap_userdn", userdn);
2511
2512         /* retrieve the basedn */
2513         d = p + 1;
2514         p = strchr_m(d, '#');
2515         if (!p) {
2516                 lp_set_cmdline("torture:ldap_basedn", "");
2517                 lp_set_cmdline("torture:ldap_secret", "");
2518                 return;
2519         }
2520         basedn = strndup(d, p - d);
2521         lp_set_cmdline("torture:ldap_basedn", basedn);
2522
2523         /* retrieve the secret */
2524         p = p + 1;
2525         if (!p) {
2526                 lp_set_cmdline("torture:ldap_secret", "");
2527                 return;
2528         }
2529         secret = strdup(p);
2530         lp_set_cmdline("torture:ldap_secret", secret);
2531
2532         printf ("%s - %s - %s\n", userdn, basedn, secret);
2533
2534 }
2535
2536 static void usage(poptContext pc)
2537 {
2538         int i;
2539         int perline = 5;
2540
2541         poptPrintUsage(pc, stdout, 0);
2542         printf("\n");
2543
2544         printf("The binding format is:\n\n");
2545
2546         printf("  TRANSPORT:host[flags]\n\n");
2547
2548         printf("  where TRANSPORT is either ncacn_np for SMB or ncacn_ip_tcp for RPC/TCP\n\n");
2549
2550         printf("  'host' is an IP or hostname or netbios name. If the binding string\n");
2551         printf("  identifies the server side of an endpoint, 'host' may be an empty\n");
2552         printf("  string.\n\n");
2553
2554         printf("  'flags' can include a SMB pipe name if using the ncacn_np transport or\n");
2555         printf("  a TCP port number if using the ncacn_ip_tcp transport, otherwise they\n");
2556         printf("  will be auto-determined.\n\n");
2557
2558         printf("  other recognised flags are:\n\n");
2559
2560         printf("    sign : enable ntlmssp signing\n");
2561         printf("    seal : enable ntlmssp sealing\n");
2562         printf("    connect : enable rpc connect level auth (auth, but no sign or seal)\n");
2563         printf("    validate: enable the NDR validator\n");
2564         printf("    print: enable debugging of the packets\n");
2565         printf("    bigendian: use bigendian RPC\n");
2566         printf("    padcheck: check reply data for non-zero pad bytes\n\n");
2567
2568         printf("  For example, these all connect to the samr pipe:\n\n");
2569
2570         printf("    ncacn_np:myserver\n");
2571         printf("    ncacn_np:myserver[samr]\n");
2572         printf("    ncacn_np:myserver[\\pipe\\samr]\n");
2573         printf("    ncacn_np:myserver[/pipe/samr]\n");
2574         printf("    ncacn_np:myserver[samr,sign,print]\n");
2575         printf("    ncacn_np:myserver[\\pipe\\samr,sign,seal,bigendian]\n");
2576         printf("    ncacn_np:myserver[/pipe/samr,seal,validate]\n");
2577         printf("    ncacn_np:\n");
2578         printf("    ncacn_np:[/pipe/samr]\n\n");
2579
2580         printf("    ncacn_ip_tcp:myserver\n");
2581         printf("    ncacn_ip_tcp:myserver[1024]\n");
2582         printf("    ncacn_ip_tcp:myserver[1024,sign,seal]\n\n");
2583
2584         printf("The unc format is:\n\n");
2585
2586         printf("    //server/share\n\n");
2587
2588         printf("tests are:");
2589         for (i=0;torture_ops[i].name;i++) {
2590                 if ((i%perline)==0) {
2591                         printf("\n");
2592                 }
2593                 printf("%s ", torture_ops[i].name);
2594         }
2595         printf("\n\n");
2596
2597         printf("default test is ALL\n");
2598
2599         exit(1);
2600 }
2601
2602 static BOOL is_binding_string(const char *binding_string)
2603 {
2604         TALLOC_CTX *mem_ctx = talloc_init("is_binding_string");
2605         struct dcerpc_binding binding_struct;
2606         NTSTATUS status;
2607
2608         status = dcerpc_parse_binding(mem_ctx, binding_string, &binding_struct);
2609
2610         talloc_destroy(mem_ctx);
2611         return NT_STATUS_IS_OK(status);
2612 }
2613
2614 /****************************************************************************
2615   main program
2616 ****************************************************************************/
2617  int main(int argc,char *argv[])
2618 {
2619         int opt, i;
2620         char *p;
2621         BOOL correct = True;
2622         int argc_new;
2623         char **argv_new;
2624         poptContext pc;
2625         enum {OPT_LOADFILE=1000,OPT_UNCLIST,OPT_TIMELIMIT,OPT_DNS,OPT_DANGEROUS};
2626         struct poptOption long_options[] = {
2627                 POPT_AUTOHELP
2628                 {"smb-ports",   'p', POPT_ARG_STRING, NULL,             0,      "SMB ports",    NULL},
2629                 {"seed",          0, POPT_ARG_INT,  &torture_seed,      0,      "seed",         NULL},
2630                 {"num-progs",     0, POPT_ARG_INT,  &torture_nprocs,    0,      "num progs",    NULL},
2631                 {"num-ops",       0, POPT_ARG_INT,  &torture_numops,    0,      "num ops",      NULL},
2632                 {"entries",       0, POPT_ARG_INT,  &torture_entries,   0,      "entries",      NULL},
2633                 {"use-oplocks", 'L', POPT_ARG_NONE, &use_oplocks,       0,      "use oplocks",  NULL},
2634                 {"show-all",      0, POPT_ARG_NONE, &torture_showall,   0,      "show all",     NULL},
2635                 {"loadfile",      0, POPT_ARG_STRING,   NULL,   OPT_LOADFILE,   "loadfile",     NULL},
2636                 {"unclist",       0, POPT_ARG_STRING,   NULL,   OPT_UNCLIST,    "unclist",      NULL},
2637                 {"timelimit",   't', POPT_ARG_STRING,   NULL,   OPT_TIMELIMIT,  "timelimit",    NULL},
2638                 {"failures",    'f', POPT_ARG_INT,  &torture_failures,  0,      "failures",     NULL},
2639                 {"parse-dns",   'D', POPT_ARG_STRING,   NULL,   OPT_DNS,        "parse-dns",    NULL},
2640                 {"dangerous",   'X', POPT_ARG_NONE,     NULL,   OPT_DANGEROUS,  "dangerous",    NULL},
2641                 POPT_COMMON_SAMBA
2642                 POPT_COMMON_CONNECTION
2643                 POPT_COMMON_CREDENTIALS
2644                 POPT_COMMON_VERSION
2645                 POPT_TABLEEND
2646         };
2647
2648         setup_logging("smbtorture", DEBUG_STDOUT);
2649
2650 #ifdef HAVE_SETBUFFER
2651         setbuffer(stdout, NULL, 0);
2652 #endif
2653
2654         pc = poptGetContext("smbtorture", argc, (const char **) argv, long_options,
2655                                 POPT_CONTEXT_KEEP_FIRST);
2656
2657         poptSetOtherOptionHelp(pc, "<binding>|<unc> TEST1 TEST2 ...");
2658
2659         while((opt = poptGetNextOpt(pc)) != -1) {
2660                 switch (opt) {
2661                 case OPT_LOADFILE:
2662                         lp_set_cmdline("torture:loadfile", poptGetOptArg(pc));
2663                         break;
2664                 case OPT_UNCLIST:
2665                         lp_set_cmdline("torture:unclist", poptGetOptArg(pc));
2666                         break;
2667                 case OPT_TIMELIMIT:
2668                         lp_set_cmdline("torture:timelimit", poptGetOptArg(pc));
2669                         break;
2670                 case OPT_DNS:
2671                         parse_dns(poptGetOptArg(pc));
2672                         break;
2673                 case OPT_DANGEROUS:
2674                         lp_set_cmdline("torture:dangerous", "Yes");
2675                         break;
2676                 default:
2677                         d_printf("Invalid option %s: %s\n",
2678                                  poptBadOption(pc, 0), poptStrerror(opt));
2679                         usage(pc);
2680                         exit(1);
2681                 }
2682         }
2683
2684         lp_load(dyn_CONFIGFILE,True,False,False);
2685         load_interfaces();
2686
2687         smbtorture_init_subsystems;
2688
2689
2690         if (torture_seed == 0) {
2691                 torture_seed = time(NULL);
2692         }
2693         printf("Using seed %d\n", torture_seed);
2694         srandom(torture_seed);
2695
2696         argv_new = discard_const_p(char *, poptGetArgs(pc));
2697
2698         argc_new = argc;
2699         for (i=0; i<argc; i++) {
2700                 if (argv_new[i] == NULL) {
2701                         argc_new = i;
2702                         break;
2703                 }
2704         }
2705
2706         if (argc_new < 3) {
2707                 usage(pc);
2708                 exit(1);
2709         }
2710
2711         for(p = argv_new[1]; *p; p++) {
2712                 if(*p == '\\')
2713                         *p = '/';
2714         }
2715
2716         /* see if its a RPC transport specifier */
2717         if (is_binding_string(argv_new[1])) {
2718                 lp_set_cmdline("torture:binding", argv_new[1]);
2719         } else {
2720                 char *binding = NULL;
2721                 const char *host = NULL, *share = NULL;
2722
2723                 if (!smbcli_parse_unc(argv_new[1], NULL, &host, &share)) {
2724                         d_printf("Invalid option: %s is not a valid torture target (share or binding string)\n\n", argv_new[1]);
2725                         usage(pc);
2726                 }
2727
2728                 lp_set_cmdline("torture:host", host);
2729                 lp_set_cmdline("torture:share", share);
2730                 asprintf(&binding, "ncacn_np:%s", host);
2731                 lp_set_cmdline("torture:binding", binding);
2732         }
2733
2734         if (!lp_parm_string(-1,"torture","username")) {
2735                 lp_set_cmdline("torture:username", cmdline_get_username());
2736         }
2737         if (!lp_parm_string(-1,"torture","userdomain")) {
2738                 /*
2739                  * backward compatibility
2740                  * maybe we should remove this to make this consistent
2741                  * for all cmdline tools
2742                  * --metze
2743                  */
2744                 if (strequal(lp_netbios_name(),cmdline_get_userdomain())) {
2745                         cmdline_set_userdomain(lp_workgroup());
2746                 }
2747                 lp_set_cmdline("torture:userdomain", cmdline_get_userdomain());
2748         }
2749         if (!lp_parm_string(-1,"torture","password")) {
2750                 lp_set_cmdline("torture:password", cmdline_get_userpassword());
2751         }
2752
2753         if (argc_new == 0) {
2754                 printf("You must specify a test to run, or 'ALL'\n");
2755         } else {
2756                 for (i=2;i<argc_new;i++) {
2757                         if (!run_test(argv_new[i])) {
2758                                 correct = False;
2759                         }
2760                 }
2761         }
2762
2763         if (correct) {
2764                 return(0);
2765         } else {
2766                 return(1);
2767         }
2768 }