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