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