libcli/smb: pass smbXcli_tcon to smb2cli_flush*()
[kai/samba.git] / source3 / torture / test_notify_online.c
1 /*
2    Unix SMB/CIFS implementation.
3    Make sure that for offline files pread and pwrite trigger a notify
4    Copyright (C) Volker Lendecke 2011
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 3 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, see <http://www.gnu.org/licenses/>.
18 */
19
20 #include "includes.h"
21 #include "torture/proto.h"
22 #include "libcli/security/security.h"
23 #include "lib/util/tevent_ntstatus.h"
24 #include "libsmb/libsmb.h"
25
26 extern char *test_filename;
27
28 struct notify_online_state {
29         struct tevent_context *ev;
30         struct cli_state *cli;
31         uint16_t dnum;
32         const char *fname;
33         uint16_t fnum;
34         bool got_notify;
35 };
36
37 static void notify_online_opened_dir(struct tevent_req *subreq);
38 static void notify_online_notify_callback(struct tevent_req *subreq);
39 static void notify_online_opened_file(struct tevent_req *subreq);
40 static void notify_online_sent_read(struct tevent_req *subreq);
41 static void notify_online_sent_closefile(struct tevent_req *subreq);
42 static void notify_online_waited(struct tevent_req *subreq);
43 static void notify_online_sent_closedir(struct tevent_req *subreq);
44
45 static struct tevent_req *notify_online_send(
46         TALLOC_CTX *mem_ctx, struct tevent_context *ev,
47         struct cli_state *cli, const char *dname, const char *fname)
48 {
49         struct tevent_req *req, *subreq;
50         struct notify_online_state *state;
51
52         req = tevent_req_create(mem_ctx, &state, struct notify_online_state);
53         if (req == NULL) {
54                 return NULL;
55         }
56         state->ev = ev;
57         state->cli = cli;
58         state->fname = fname;
59
60         subreq = cli_ntcreate_send(
61                 state, ev, cli, dname, EXTENDED_RESPONSE_REQUIRED,
62                 SEC_FILE_READ_DATA, 0,
63                 FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
64                 FILE_OPEN, 0, 0);
65         if (tevent_req_nomem(subreq, req)) {
66                 return tevent_req_post(req, ev);
67         }
68         tevent_req_set_callback(subreq, notify_online_opened_dir, req);
69         return req;
70 }
71
72 static void notify_online_opened_dir(struct tevent_req *subreq)
73 {
74         struct tevent_req *req = tevent_req_callback_data(
75                 subreq, struct tevent_req);
76         struct notify_online_state *state = tevent_req_data(
77                 req, struct notify_online_state);
78         NTSTATUS status;
79
80         status = cli_ntcreate_recv(subreq, &state->dnum);
81         TALLOC_FREE(subreq);
82         if (tevent_req_nterror(req, status)) {
83                 return;
84         }
85         subreq = cli_notify_send(state, state->ev, state->cli, state->dnum,
86                                  128, FILE_NOTIFY_CHANGE_ATTRIBUTES, false);
87         if (tevent_req_nomem(subreq, req)) {
88                 return;
89         }
90         tevent_req_set_callback(subreq, notify_online_notify_callback, req);
91
92         subreq = cli_ntcreate_send(
93                 state, state->ev, state->cli, state->fname, 0,
94                 GENERIC_READ_ACCESS, FILE_ATTRIBUTE_NORMAL,
95                 FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE,
96                 FILE_OPEN, FILE_NON_DIRECTORY_FILE, 0);
97         if (tevent_req_nomem(subreq, req)) {
98                 return;
99         }
100         tevent_req_set_callback(subreq, notify_online_opened_file, req);
101 }
102
103 static void notify_online_notify_callback(struct tevent_req *subreq)
104 {
105         struct tevent_req *req = tevent_req_callback_data(
106                 subreq, struct tevent_req);
107         struct notify_online_state *state = tevent_req_data(
108                 req, struct notify_online_state);
109         NTSTATUS status;
110         uint32_t num_changes;
111         struct notify_change *changes;
112
113         status = cli_notify_recv(subreq, state, &num_changes, &changes);
114         TALLOC_FREE(subreq);
115         if (tevent_req_nterror(req, status)) {
116                 return;
117         }
118         if ((num_changes == 1)
119             && (changes[0].action == NOTIFY_ACTION_MODIFIED)
120             && (strcmp(changes[0].name, state->fname) == 0)) {
121                 state->got_notify = true;
122         }
123         tevent_req_done(req);
124 }
125
126 static void notify_online_opened_file(struct tevent_req *subreq)
127 {
128         struct tevent_req *req = tevent_req_callback_data(
129                 subreq, struct tevent_req);
130         struct notify_online_state *state = tevent_req_data(
131                 req, struct notify_online_state);
132         NTSTATUS status;
133
134         status = cli_ntcreate_recv(subreq, &state->fnum);
135         TALLOC_FREE(subreq);
136         if (tevent_req_nterror(req, status)) {
137                 return;
138         }
139         subreq = cli_read_andx_send(
140                 state, state->ev, state->cli, state->fnum, 0, 1);
141         if (tevent_req_nomem(subreq, req)) {
142                 return;
143         }
144         tevent_req_set_callback(subreq, notify_online_sent_read, req);
145 }
146
147 static void notify_online_sent_read(struct tevent_req *subreq)
148 {
149         struct tevent_req *req = tevent_req_callback_data(
150                 subreq, struct tevent_req);
151         struct notify_online_state *state = tevent_req_data(
152                 req, struct notify_online_state);
153         NTSTATUS status;
154         ssize_t received;
155         uint8_t *buf;
156
157         status = cli_read_andx_recv(subreq, &received, &buf);
158         TALLOC_FREE(subreq);
159         if (tevent_req_nterror(req, status)) {
160                 return;
161         }
162         subreq = cli_close_send(
163                 state, state->ev, state->cli, state->fnum);
164         if (tevent_req_nomem(subreq, req)) {
165                 return;
166         }
167         tevent_req_set_callback(subreq, notify_online_sent_closefile, req);
168 }
169
170 static void notify_online_sent_closefile(struct tevent_req *subreq)
171 {
172         struct tevent_req *req = tevent_req_callback_data(
173                 subreq, struct tevent_req);
174         struct notify_online_state *state = tevent_req_data(
175                 req, struct notify_online_state);
176         NTSTATUS status;
177
178         status = cli_close_recv(subreq);
179         TALLOC_FREE(subreq);
180         if (tevent_req_nterror(req, status)) {
181                 return;
182         }
183         subreq = tevent_wakeup_send(
184                 state, state->ev, timeval_current_ofs(10, 0));
185         if (tevent_req_nomem(subreq, req)) {
186                 return;
187         }
188         tevent_req_set_callback(subreq, notify_online_waited, req);
189 }
190
191 static void notify_online_waited(struct tevent_req *subreq)
192 {
193         struct tevent_req *req = tevent_req_callback_data(
194                 subreq, struct tevent_req);
195         struct notify_online_state *state = tevent_req_data(
196                 req, struct notify_online_state);
197
198         tevent_wakeup_recv(subreq);
199         TALLOC_FREE(subreq);
200         subreq = cli_close_send(
201                 state, state->ev, state->cli, state->dnum);
202         if (tevent_req_nomem(subreq, req)) {
203                 return;
204         }
205         tevent_req_set_callback(subreq, notify_online_sent_closedir, req);
206 }
207
208 static void notify_online_sent_closedir(struct tevent_req *subreq)
209 {
210         struct tevent_req *req = tevent_req_callback_data(
211                 subreq, struct tevent_req);
212         NTSTATUS status;
213
214         status = cli_close_recv(subreq);
215         TALLOC_FREE(subreq);
216         if (tevent_req_nterror(req, status)) {
217                 return;
218         }
219 }
220
221 static NTSTATUS notify_online_recv(struct tevent_req *req, bool *got_notify)
222 {
223         struct notify_online_state *state = tevent_req_data(
224                 req, struct notify_online_state);
225         NTSTATUS status;
226
227         if (tevent_req_is_nterror(req, &status)) {
228                 return status;
229         }
230         *got_notify = state->got_notify;
231         return NT_STATUS_OK;
232 }
233
234 static NTSTATUS notify_online(struct cli_state *cli,
235                               const char *dirname, const char *filename,
236                               bool *got_notify)
237 {
238         TALLOC_CTX *frame = talloc_stackframe();
239         struct event_context *ev;
240         struct tevent_req *req;
241         NTSTATUS status = NT_STATUS_NO_MEMORY;
242
243         ev = event_context_init(frame);
244         if (ev == NULL) {
245                 goto fail;
246         }
247         req = notify_online_send(frame, ev, cli, dirname, filename);
248         if (req == NULL) {
249                 goto fail;
250         }
251         if (!tevent_req_poll_ntstatus(req, ev, &status)) {
252                 goto fail;
253         }
254         status = notify_online_recv(req, got_notify);
255  fail:
256         TALLOC_FREE(frame);
257         return status;
258 }
259
260 bool run_notify_online(int dummy)
261 {
262         struct cli_state *cli;
263         NTSTATUS status;
264         char *p;
265         const char *dir;
266         const char *file;
267         bool got_notify = false;
268
269         printf("Starting NOTIFY_ONLINE\n");
270
271         if (test_filename == NULL) {
272                 fprintf(stderr, "<-f filename> missing\n");
273                 return false;
274         }
275
276         if (!torture_open_connection(&cli, 0)) {
277                 return false;
278         }
279
280         p = strrchr(test_filename, '/');
281         if (p != NULL) {
282                 dir = SMB_STRNDUP(test_filename, p-test_filename);
283                 file = SMB_STRDUP(p+1);
284         } else {
285                 dir = "";
286                 file = test_filename;
287         }
288
289         status = notify_online(cli, dir, file, &got_notify);
290         d_printf("notify_online returned %s (%d)\n", nt_errstr(status),
291                  (int)got_notify);
292         torture_close_connection(cli);
293         return NT_STATUS_IS_OK(status) && got_notify;
294 }