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