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