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, SMB2_IMPERSONATION_IMPERSONATION, 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, NULL);
  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,
  97                 SMB2_IMPERSONATION_IMPERSONATION, 0);
  98         if (tevent_req_nomem(subreq, req)) {
  99                 return;
 100         }
 101         tevent_req_set_callback(subreq, notify_online_opened_file, req);
 102 }
 103
 104 static void notify_online_notify_callback(struct tevent_req *subreq)
 105 {
 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);
 110         NTSTATUS status;
 111         uint32_t num_changes;
 112         struct notify_change *changes;
 113
 114         status = cli_notify_recv(subreq, state, &num_changes, &changes);
 115         TALLOC_FREE(subreq);
 116         if (tevent_req_nterror(req, status)) {
 117                 return;
 118         }
 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;
 123         }
 124         tevent_req_done(req);
 125 }
 126
 127 static void notify_online_opened_file(struct tevent_req *subreq)
 128 {
 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);
 133         NTSTATUS status;
 134
 135         status = cli_ntcreate_recv(subreq, &state->fnum, NULL);
 136         TALLOC_FREE(subreq);
 137         if (tevent_req_nterror(req, status)) {
 138                 return;
 139         }
 140         subreq = cli_read_andx_send(
 141                 state, state->ev, state->cli, state->fnum, 0, 1);
 142         if (tevent_req_nomem(subreq, req)) {
 143                 return;
 144         }
 145         tevent_req_set_callback(subreq, notify_online_sent_read, req);
 146 }
 147
 148 static void notify_online_sent_read(struct tevent_req *subreq)
 149 {
 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);
 154         NTSTATUS status;
 155         ssize_t received;
 156         uint8_t *buf;
 157
 158         status = cli_read_andx_recv(subreq, &received, &buf);
 159         TALLOC_FREE(subreq);
 160         if (tevent_req_nterror(req, status)) {
 161                 return;
 162         }
 163         subreq = cli_close_send(
 164                 state, state->ev, state->cli, state->fnum);
 165         if (tevent_req_nomem(subreq, req)) {
 166                 return;
 167         }
 168         tevent_req_set_callback(subreq, notify_online_sent_closefile, req);
 169 }
 170
 171 static void notify_online_sent_closefile(struct tevent_req *subreq)
 172 {
 173         struct tevent_req *req = tevent_req_callback_data(
 174                 subreq, struct tevent_req);
 175         struct notify_online_state *state = tevent_req_data(
 176                 req, struct notify_online_state);
 177         NTSTATUS status;
 178
 179         status = cli_close_recv(subreq);
 180         TALLOC_FREE(subreq);
 181         if (tevent_req_nterror(req, status)) {
 182                 return;
 183         }
 184         subreq = tevent_wakeup_send(
 185                 state, state->ev, timeval_current_ofs(10, 0));
 186         if (tevent_req_nomem(subreq, req)) {
 187                 return;
 188         }
 189         tevent_req_set_callback(subreq, notify_online_waited, req);
 190 }
 191
 192 static void notify_online_waited(struct tevent_req *subreq)
 193 {
 194         struct tevent_req *req = tevent_req_callback_data(
 195                 subreq, struct tevent_req);
 196         struct notify_online_state *state = tevent_req_data(
 197                 req, struct notify_online_state);
 198
 199         tevent_wakeup_recv(subreq);
 200         TALLOC_FREE(subreq);
 201         subreq = cli_close_send(
 202                 state, state->ev, state->cli, state->dnum);
 203         if (tevent_req_nomem(subreq, req)) {
 204                 return;
 205         }
 206         tevent_req_set_callback(subreq, notify_online_sent_closedir, req);
 207 }
 208
 209 static void notify_online_sent_closedir(struct tevent_req *subreq)
 210 {
 211         struct tevent_req *req = tevent_req_callback_data(
 212                 subreq, struct tevent_req);
 213         NTSTATUS status;
 214
 215         status = cli_close_recv(subreq);
 216         TALLOC_FREE(subreq);
 217         if (tevent_req_nterror(req, status)) {
 218                 return;
 219         }
 220 }
 221
 222 static NTSTATUS notify_online_recv(struct tevent_req *req, bool *got_notify)
 223 {
 224         struct notify_online_state *state = tevent_req_data(
 225                 req, struct notify_online_state);
 226         NTSTATUS status;
 227
 228         if (tevent_req_is_nterror(req, &status)) {
 229                 return status;
 230         }
 231         *got_notify = state->got_notify;
 232         return NT_STATUS_OK;
 233 }
 234
 235 static NTSTATUS notify_online(struct cli_state *cli,
 236                               const char *dirname, const char *filename,
 237                               bool *got_notify)
 238 {
 239         TALLOC_CTX *frame = talloc_stackframe();
 240         struct tevent_context *ev;
 241         struct tevent_req *req;
 242         NTSTATUS status = NT_STATUS_NO_MEMORY;
 243
 244         ev = samba_tevent_context_init(frame);
 245         if (ev == NULL) {
 246                 goto fail;
 247         }
 248         req = notify_online_send(frame, ev, cli, dirname, filename);
 249         if (req == NULL) {
 250                 goto fail;
 251         }
 252         if (!tevent_req_poll_ntstatus(req, ev, &status)) {
 253                 goto fail;
 254         }
 255         status = notify_online_recv(req, got_notify);
 256  fail:
 257         TALLOC_FREE(frame);
 258         return status;
 259 }
 260
 261 bool run_notify_online(int dummy)
 262 {
 263         struct cli_state *cli;
 264         NTSTATUS status;
 265         char *p;
 266         const char *dir;
 267         const char *file;
 268         bool got_notify = false;
 269
 270         printf("Starting NOTIFY_ONLINE\n");
 271
 272         if (test_filename == NULL) {
 273                 fprintf(stderr, "<-f filename> missing\n");
 274                 return false;
 275         }
 276
 277         if (!torture_open_connection(&cli, 0)) {
 278                 return false;
 279         }
 280
 281         p = strrchr(test_filename, '/');
 282         if (p != NULL) {
 283                 dir = SMB_STRNDUP(test_filename, p-test_filename);
 284                 file = SMB_STRDUP(p+1);
 285         } else {
 286                 dir = "";
 287                 file = test_filename;
 288         }
 289
 290         status = notify_online(cli, dir, file, &got_notify);
 291         d_printf("notify_online returned %s (%d)\n", nt_errstr(status),
 292                  (int)got_notify);
 293         torture_close_connection(cli);
 294         return NT_STATUS_IS_OK(status) && got_notify;
 295 }