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