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