2 Unix SMB/CIFS implementation.
4 test server handling of unexpected client disconnects
6 Copyright (C) Andrew Tridgell 2004
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
23 #include "system/filesys.h"
24 #include "libcli/raw/libcliraw.h"
25 #include "libcli/raw/raw_proto.h"
26 #include "libcli/libcli.h"
27 #include "torture/util.h"
28 #include "torture/basic/proto.h"
30 #define BASEDIR "\\test_disconnect"
32 #define CHECK_STATUS(status, correct) do { \
33 if (!NT_STATUS_EQUAL(status, correct)) { \
34 printf("(%s) Incorrect status %s - should be %s\n", \
35 __location__, nt_errstr(status), nt_errstr(correct)); \
41 test disconnect after async open
43 static bool test_disconnect_open(struct smbcli_state
*cli
, TALLOC_CTX
*mem_ctx
)
46 struct smbcli_request
*req1
, *req2
;
49 printf("trying open/disconnect\n");
51 io
.generic
.level
= RAW_OPEN_NTCREATEX
;
52 io
.ntcreatex
.in
.root_fid
.fnum
= 0;
53 io
.ntcreatex
.in
.flags
= 0;
54 io
.ntcreatex
.in
.access_mask
= SEC_FILE_READ_DATA
;
55 io
.ntcreatex
.in
.create_options
= 0;
56 io
.ntcreatex
.in
.file_attr
= FILE_ATTRIBUTE_NORMAL
;
57 io
.ntcreatex
.in
.share_access
= NTCREATEX_SHARE_ACCESS_READ
;
58 io
.ntcreatex
.in
.alloc_size
= 0;
59 io
.ntcreatex
.in
.open_disposition
= NTCREATEX_DISP_OPEN_IF
;
60 io
.ntcreatex
.in
.impersonation
= NTCREATEX_IMPERSONATION_ANONYMOUS
;
61 io
.ntcreatex
.in
.security_flags
= 0;
62 io
.ntcreatex
.in
.fname
= BASEDIR
"\\open.dat";
63 status
= smb_raw_open(cli
->tree
, mem_ctx
, &io
);
64 CHECK_STATUS(status
, NT_STATUS_OK
);
66 io
.ntcreatex
.in
.share_access
= 0;
67 req1
= smb_raw_open_send(cli
->tree
, &io
);
68 req2
= smb_raw_open_send(cli
->tree
, &io
);
70 printf("test_disconnect_open: smb_raw_open_send() "
75 status
= smbcli_chkpath(cli
->tree
, "\\");
76 CHECK_STATUS(status
, NT_STATUS_OK
);
85 test disconnect with timed lock
87 static bool test_disconnect_lock(struct smbcli_state
*cli
, TALLOC_CTX
*mem_ctx
)
92 struct smbcli_request
*req
;
93 struct smb_lock_entry lock
[1];
95 printf("trying disconnect with async lock\n");
97 fnum
= smbcli_open(cli
->tree
, BASEDIR
"\\write.dat",
98 O_RDWR
| O_CREAT
, DENY_NONE
);
100 printf("open failed in mux_write - %s\n", smbcli_errstr(cli
->tree
));
104 io
.lockx
.level
= RAW_LOCK_LOCKX
;
105 io
.lockx
.in
.file
.fnum
= fnum
;
106 io
.lockx
.in
.mode
= 0;
107 io
.lockx
.in
.timeout
= 0;
108 io
.lockx
.in
.lock_cnt
= 1;
109 io
.lockx
.in
.ulock_cnt
= 0;
113 io
.lockx
.in
.locks
= &lock
[0];
115 status
= smb_raw_lock(cli
->tree
, &io
);
116 CHECK_STATUS(status
, NT_STATUS_OK
);
119 io
.lockx
.in
.timeout
= 3000;
120 req
= smb_raw_lock_send(cli
->tree
, &io
);
122 printf("test_disconnect_lock: smb_raw_lock_send() "
127 status
= smbcli_chkpath(cli
->tree
, "\\");
128 CHECK_STATUS(status
, NT_STATUS_OK
);
138 basic testing of disconnects
140 bool torture_disconnect(struct torture_context
*torture
)
145 extern int torture_numops
;
146 struct smbcli_state
*cli
;
148 mem_ctx
= talloc_init("torture_raw_mux");
150 if (!torture_open_connection(&cli
, torture
, 0)) {
154 torture_assert(torture
, torture_setup_dir(cli
, BASEDIR
), "Failed to setup up test directory: " BASEDIR
);
156 for (i
=0;i
<torture_numops
;i
++) {
157 ret
&= test_disconnect_lock(cli
, mem_ctx
);
158 if (!torture_open_connection(&cli
, torture
, 0)) {
162 ret
&= test_disconnect_open(cli
, mem_ctx
);
163 if (!torture_open_connection(&cli
, torture
, 0)) {
167 if (torture_setting_bool(torture
, "samba3", false)) {
169 * In Samba3 it might happen that the old smbd from
170 * test_disconnect_lock is not scheduled before the
171 * new process comes in. Try to get rid of the random
172 * failures in the build farm.
178 smb_raw_exit(cli
->session
);
179 smbcli_deltree(cli
->tree
, BASEDIR
);
180 talloc_free(mem_ctx
);