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