2 Unix SMB/CIFS implementation.
3 Make sure that for offline files pread and pwrite trigger a notify
4 Copyright (C) Volker Lendecke 2011
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.
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.
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/>.
21 #include "torture/proto.h"
22 #include "libcli/security/security.h"
23 #include "lib/util/tevent_ntstatus.h"
24 #include "libsmb/libsmb.h"
26 extern char *test_filename;
28 struct notify_online_state {
29 struct tevent_context *ev;
30 struct cli_state *cli;
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);
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)
49 struct tevent_req *req, *subreq;
50 struct notify_online_state *state;
52 req = tevent_req_create(mem_ctx, &state, struct notify_online_state);
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, SMB2_IMPERSONATION_IMPERSONATION, 0);
65 if (tevent_req_nomem(subreq, req)) {
66 return tevent_req_post(req, ev);
68 tevent_req_set_callback(subreq, notify_online_opened_dir, req);
72 static void notify_online_opened_dir(struct tevent_req *subreq)
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);
80 status = cli_ntcreate_recv(subreq, &state->dnum, NULL);
82 if (tevent_req_nterror(req, status)) {
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)) {
90 tevent_req_set_callback(subreq, notify_online_notify_callback, req);
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,
97 SMB2_IMPERSONATION_IMPERSONATION, 0);
98 if (tevent_req_nomem(subreq, req)) {
101 tevent_req_set_callback(subreq, notify_online_opened_file, req);
104 static void notify_online_notify_callback(struct tevent_req *subreq)
106 struct tevent_req *req = tevent_req_callback_data(
107 subreq, struct tevent_req);
108 struct notify_online_state *state = tevent_req_data(
109 req, struct notify_online_state);
111 uint32_t num_changes;
112 struct notify_change *changes;
114 status = cli_notify_recv(subreq, state, &num_changes, &changes);
116 if (tevent_req_nterror(req, status)) {
119 if ((num_changes == 1)
120 && (changes[0].action == NOTIFY_ACTION_MODIFIED)
121 && (strcmp(changes[0].name, state->fname) == 0)) {
122 state->got_notify = true;
124 tevent_req_done(req);
127 static void notify_online_opened_file(struct tevent_req *subreq)
129 struct tevent_req *req = tevent_req_callback_data(
130 subreq, struct tevent_req);
131 struct notify_online_state *state = tevent_req_data(
132 req, struct notify_online_state);
135 status = cli_ntcreate_recv(subreq, &state->fnum, NULL);
137 if (tevent_req_nterror(req, status)) {
140 subreq = cli_read_andx_send(
141 state, state->ev, state->cli, state->fnum, 0, 1);
142 if (tevent_req_nomem(subreq, req)) {
145 tevent_req_set_callback(subreq, notify_online_sent_read, req);
148 static void notify_online_sent_read(struct tevent_req *subreq)
150 struct tevent_req *req = tevent_req_callback_data(
151 subreq, struct tevent_req);
152 struct notify_online_state *state = tevent_req_data(
153 req, struct notify_online_state);
158 status = cli_read_andx_recv(subreq, &received, &buf);
160 if (tevent_req_nterror(req, status)) {
163 subreq = cli_close_send(state, state->ev, state->cli, state->fnum, 0);
164 if (tevent_req_nomem(subreq, req)) {
167 tevent_req_set_callback(subreq, notify_online_sent_closefile, req);
170 static void notify_online_sent_closefile(struct tevent_req *subreq)
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);
178 status = cli_close_recv(subreq);
180 if (tevent_req_nterror(req, status)) {
183 subreq = tevent_wakeup_send(
184 state, state->ev, timeval_current_ofs(10, 0));
185 if (tevent_req_nomem(subreq, req)) {
188 tevent_req_set_callback(subreq, notify_online_waited, req);
191 static void notify_online_waited(struct tevent_req *subreq)
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);
198 tevent_wakeup_recv(subreq);
200 subreq = cli_close_send(state, state->ev, state->cli, state->dnum, 0);
201 if (tevent_req_nomem(subreq, req)) {
204 tevent_req_set_callback(subreq, notify_online_sent_closedir, req);
207 static void notify_online_sent_closedir(struct tevent_req *subreq)
209 struct tevent_req *req = tevent_req_callback_data(
210 subreq, struct tevent_req);
213 status = cli_close_recv(subreq);
215 if (tevent_req_nterror(req, status)) {
220 static NTSTATUS notify_online_recv(struct tevent_req *req, bool *got_notify)
222 struct notify_online_state *state = tevent_req_data(
223 req, struct notify_online_state);
226 if (tevent_req_is_nterror(req, &status)) {
229 *got_notify = state->got_notify;
233 static NTSTATUS notify_online(struct cli_state *cli,
234 const char *dirname, const char *filename,
237 TALLOC_CTX *frame = talloc_stackframe();
238 struct tevent_context *ev;
239 struct tevent_req *req;
240 NTSTATUS status = NT_STATUS_NO_MEMORY;
242 ev = samba_tevent_context_init(frame);
246 req = notify_online_send(frame, ev, cli, dirname, filename);
250 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
253 status = notify_online_recv(req, got_notify);
259 bool run_notify_online(int dummy)
261 struct cli_state *cli;
266 bool got_notify = false;
268 printf("Starting NOTIFY_ONLINE\n");
270 if (test_filename == NULL) {
271 fprintf(stderr, "<-f filename> missing\n");
275 if (!torture_open_connection(&cli, 0)) {
279 p = strrchr(test_filename, '/');
281 dir = SMB_STRNDUP(test_filename, p-test_filename);
282 file = SMB_STRDUP(p+1);
285 file = test_filename;
288 status = notify_online(cli, dir, file, &got_notify);
289 d_printf("notify_online returned %s (%d)\n", nt_errstr(status),
291 torture_close_connection(cli);
292 return NT_STATUS_IS_OK(status) && got_notify;