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