smb2_ioctl: track copychunk response output state
[obnox/samba/samba-obnox.git] / source3 / smbd / smb2_ioctl_network_fs.c
1 /*
2    Unix SMB/CIFS implementation.
3    Core SMB2 server
4
5    Copyright (C) Stefan Metzmacher 2009
6    Copyright (C) David Disseldorp 2012
7
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.
12
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.
17
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/>.
20 */
21
22 #include "includes.h"
23 #include "smbd/smbd.h"
24 #include "smbd/globals.h"
25 #include "../libcli/smb/smb_common.h"
26 #include "../libcli/security/security.h"
27 #include "../lib/util/tevent_ntstatus.h"
28 #include "../lib/ccan/build_assert/build_assert.h"
29 #include "include/ntioctl.h"
30 #include "../librpc/ndr/libndr.h"
31 #include "librpc/gen_ndr/ndr_ioctl.h"
32 #include "smb2_ioctl_private.h"
33
34 #define COPYCHUNK_MAX_CHUNKS    256             /* 2k8r2 & win8 = 256 */
35 #define COPYCHUNK_MAX_CHUNK_LEN 1048576         /* 2k8r2 & win8 = 1048576 */
36 #define COPYCHUNK_MAX_TOTAL_LEN 16777216        /* 2k8r2 & win8 = 16777216 */
37 static void copychunk_pack_limits(struct srv_copychunk_rsp *cc_rsp)
38 {
39         cc_rsp->chunks_written = COPYCHUNK_MAX_CHUNKS;
40         cc_rsp->chunk_bytes_written = COPYCHUNK_MAX_CHUNK_LEN;
41         cc_rsp->total_bytes_written = COPYCHUNK_MAX_TOTAL_LEN;
42 }
43
44 static NTSTATUS copychunk_check_limits(struct srv_copychunk_copy *cc_copy)
45 {
46         uint32_t i;
47         uint32_t total_len = 0;
48
49         if (cc_copy->chunk_count > COPYCHUNK_MAX_CHUNKS) {
50                 return NT_STATUS_INVALID_PARAMETER;
51         }
52
53         for (i = 0; i < cc_copy->chunk_count; i++) {
54                 if (cc_copy->chunks[i].length > COPYCHUNK_MAX_CHUNK_LEN) {
55                         return NT_STATUS_INVALID_PARAMETER;
56                 }
57                 total_len += cc_copy->chunks[i].length;
58         }
59         if (total_len > COPYCHUNK_MAX_TOTAL_LEN) {
60                 return NT_STATUS_INVALID_PARAMETER;
61         }
62
63         return NT_STATUS_OK;
64 }
65
66 static void copychunk_unlock_all(struct files_struct *src_fsp,
67                                  struct files_struct *dst_fsp,
68                                  struct lock_struct *rd_locks,
69                                  struct lock_struct *wr_locks,
70                                  uint32_t num_locks)
71 {
72
73         uint32_t i;
74
75         for (i = 0; i < num_locks; i++) {
76                 SMB_VFS_STRICT_UNLOCK(src_fsp->conn, src_fsp, &rd_locks[i]);
77                 SMB_VFS_STRICT_UNLOCK(dst_fsp->conn, dst_fsp, &wr_locks[i]);
78         }
79 }
80
81 /* request read and write locks for each chunk */
82 static NTSTATUS copychunk_lock_all(TALLOC_CTX *mem_ctx,
83                                    struct srv_copychunk_copy *cc_copy,
84                                    struct files_struct *src_fsp,
85                                    struct files_struct *dst_fsp,
86                                    struct lock_struct **rd_locks,
87                                    struct lock_struct **wr_locks,
88                                    uint32_t *num_locks)
89 {
90         NTSTATUS status;
91         uint32_t i;
92         struct lock_struct *rlocks;
93         struct lock_struct *wlocks;
94
95         rlocks = talloc_array(mem_ctx, struct lock_struct,
96                               cc_copy->chunk_count);
97         if (rlocks == NULL) {
98                 status = NT_STATUS_NO_MEMORY;
99                 goto err_out;
100         }
101
102         wlocks = talloc_array(mem_ctx, struct lock_struct,
103                               cc_copy->chunk_count);
104         if (wlocks == NULL) {
105                 status = NT_STATUS_NO_MEMORY;
106                 goto err_rlocks_free;
107         }
108
109         for (i = 0; i < cc_copy->chunk_count; i++) {
110                 init_strict_lock_struct(src_fsp,
111                                         src_fsp->op->global->open_persistent_id,
112                                         cc_copy->chunks[i].source_off,
113                                         cc_copy->chunks[i].length,
114                                         READ_LOCK,
115                                         &rlocks[i]);
116                 init_strict_lock_struct(dst_fsp,
117                                         dst_fsp->op->global->open_persistent_id,
118                                         cc_copy->chunks[i].target_off,
119                                         cc_copy->chunks[i].length,
120                                         WRITE_LOCK,
121                                         &wlocks[i]);
122
123                 if (!SMB_VFS_STRICT_LOCK(src_fsp->conn, src_fsp, &rlocks[i])) {
124                         status = NT_STATUS_FILE_LOCK_CONFLICT;
125                         goto err_unlock;
126                 }
127                 if (!SMB_VFS_STRICT_LOCK(dst_fsp->conn, dst_fsp, &wlocks[i])) {
128                         /* unlock last rlock, otherwise missed by cleanup */
129                         SMB_VFS_STRICT_UNLOCK(src_fsp->conn, src_fsp,
130                                               &rlocks[i]);
131                         status = NT_STATUS_FILE_LOCK_CONFLICT;
132                         goto err_unlock;
133                 }
134         }
135
136         *rd_locks = rlocks;
137         *wr_locks = wlocks;
138         *num_locks = i;
139
140         return NT_STATUS_OK;
141
142 err_unlock:
143         if (i > 0) {
144                 /* cleanup all locks successfully issued so far */
145                 copychunk_unlock_all(src_fsp, dst_fsp, rlocks, wlocks, i);
146         }
147         talloc_free(wlocks);
148 err_rlocks_free:
149         talloc_free(rlocks);
150 err_out:
151         return status;
152 }
153
154 struct fsctl_srv_copychunk_state {
155         struct connection_struct *conn;
156         uint32_t dispatch_count;
157         uint32_t recv_count;
158         uint32_t bad_recv_count;
159         NTSTATUS status;
160         off_t total_written;
161         struct files_struct *src_fsp;
162         struct files_struct *dst_fsp;
163         struct lock_struct *wlocks;
164         struct lock_struct *rlocks;
165         uint32_t num_locks;
166         enum {
167                 COPYCHUNK_OUT_EMPTY = 0,
168                 COPYCHUNK_OUT_LIMITS,
169                 COPYCHUNK_OUT_RSP,
170         } out_data;
171 };
172 static void fsctl_srv_copychunk_vfs_done(struct tevent_req *subreq);
173
174 static struct tevent_req *fsctl_srv_copychunk_send(TALLOC_CTX *mem_ctx,
175                                                    struct tevent_context *ev,
176                                                    struct files_struct *dst_fsp,
177                                                    DATA_BLOB *in_input,
178                                                    struct smbd_smb2_request *smb2req)
179 {
180         struct tevent_req *req;
181         struct srv_copychunk_copy cc_copy;
182         enum ndr_err_code ndr_ret;
183         uint64_t src_persistent_h;
184         uint64_t src_volatile_h;
185         int i;
186         struct srv_copychunk *chunk;
187         struct fsctl_srv_copychunk_state *state;
188
189         req = tevent_req_create(mem_ctx, &state,
190                                 struct fsctl_srv_copychunk_state);
191         if (req == NULL) {
192                 return NULL;
193         }
194         state->conn = dst_fsp->conn;
195         ndr_ret = ndr_pull_struct_blob(in_input, mem_ctx, &cc_copy,
196                         (ndr_pull_flags_fn_t)ndr_pull_srv_copychunk_copy);
197         if (ndr_ret != NDR_ERR_SUCCESS) {
198                 DEBUG(0, ("failed to unmarshall copy chunk req\n"));
199                 state->status = NT_STATUS_INVALID_PARAMETER;
200                 tevent_req_nterror(req, state->status);
201                 return tevent_req_post(req, ev);
202         }
203
204         /* persistent/volatile keys sent as the resume key */
205         src_persistent_h = BVAL(cc_copy.source_key, 0);
206         src_volatile_h = BVAL(cc_copy.source_key, 8);
207         state->src_fsp = file_fsp_get(smb2req, src_persistent_h, src_volatile_h);
208         if (state->src_fsp == NULL) {
209                 DEBUG(3, ("invalid resume key in copy chunk req\n"));
210                 state->status = NT_STATUS_OBJECT_NAME_NOT_FOUND;
211                 tevent_req_nterror(req, state->status);
212                 return tevent_req_post(req, ev);
213         }
214
215         state->dst_fsp = dst_fsp;
216         /*
217          * [MS-SMB2] 3.3.5.15.6 Handling a Server-Side Data Copy Request
218          * If Open.GrantedAccess of the destination file does not
219          * include FILE_WRITE_DATA, then the request MUST be failed with
220          * STATUS_ACCESS_DENIED. If Open.GrantedAccess of the
221          * destination file does not include FILE_READ_DATA access and
222          * the CtlCode is FSCTL_SRV_COPYCHUNK, then the request MUST be
223          * failed with STATUS_ACCESS_DENIED.
224          */
225         if (!CHECK_WRITE(state->dst_fsp)) {
226                 state->status = NT_STATUS_ACCESS_DENIED;
227                 tevent_req_nterror(req, state->status);
228                 return tevent_req_post(req, ev);
229         }
230         if (!CHECK_READ(state->dst_fsp, smb2req->smb1req)) {
231                 state->status = NT_STATUS_ACCESS_DENIED;
232                 tevent_req_nterror(req, state->status);
233                 return tevent_req_post(req, ev);
234         }
235         if (!CHECK_READ(state->src_fsp, smb2req->smb1req)) {
236                 state->status = NT_STATUS_ACCESS_DENIED;
237                 tevent_req_nterror(req, state->status);
238                 return tevent_req_post(req, ev);
239         }
240
241         state->status = copychunk_check_limits(&cc_copy);
242         if (tevent_req_nterror(req, state->status)) {
243                 DEBUG(3, ("copy chunk req exceeds limits\n"));
244                 state->out_data = COPYCHUNK_OUT_LIMITS;
245                 return tevent_req_post(req, ev);
246         }
247
248         /* any errors from here onwards should carry copychunk response data */
249         state->out_data = COPYCHUNK_OUT_RSP;
250
251         state->status = copychunk_lock_all(state,
252                                            &cc_copy,
253                                            state->src_fsp,
254                                            state->dst_fsp,
255                                            &state->rlocks,
256                                            &state->wlocks,
257                                            &state->num_locks);
258         if (tevent_req_nterror(req, state->status)) {
259                 return tevent_req_post(req, ev);
260         }
261
262         for (i = 0; i < cc_copy.chunk_count; i++) {
263                 struct tevent_req *vfs_subreq;
264                 chunk = &cc_copy.chunks[i];
265                 vfs_subreq = SMB_VFS_COPY_CHUNK_SEND(dst_fsp->conn,
266                                                      state, ev,
267                                                      state->src_fsp,
268                                                      chunk->source_off,
269                                                      state->dst_fsp,
270                                                      chunk->target_off,
271                                                      chunk->length);
272                 if (vfs_subreq == NULL) {
273                         DEBUG(0, ("VFS copy chunk send failed\n"));
274                         state->status = NT_STATUS_NO_MEMORY;
275                         if (state->dispatch_count == 0) {
276                                 /* nothing dispatched, return immediately */
277                                 copychunk_unlock_all(state->src_fsp,
278                                                      state->dst_fsp,
279                                                      state->rlocks,
280                                                      state->wlocks,
281                                                      state->num_locks);
282                                 tevent_req_nterror(req, state->status);
283                                 return tevent_req_post(req, ev);
284                         } else {
285                                 /*
286                                  * wait for dispatched to complete before
287                                  * returning error, locks held.
288                                  */
289                                 break;
290                         }
291                 }
292                 tevent_req_set_callback(vfs_subreq,
293                                         fsctl_srv_copychunk_vfs_done, req);
294                 state->dispatch_count++;
295         }
296
297         /* hold locks until all dispatched requests are completed */
298         return req;
299 }
300
301 static void fsctl_srv_copychunk_vfs_done(struct tevent_req *subreq)
302 {
303         struct tevent_req *req = tevent_req_callback_data(
304                 subreq, struct tevent_req);
305         struct fsctl_srv_copychunk_state *state = tevent_req_data(req,
306                                         struct fsctl_srv_copychunk_state);
307         off_t chunk_nwritten;
308         NTSTATUS status;
309
310         state->recv_count++;
311         status = SMB_VFS_COPY_CHUNK_RECV(state->conn, subreq,
312                                          &chunk_nwritten);
313         TALLOC_FREE(subreq);
314         if (NT_STATUS_IS_OK(status)) {
315                 DEBUG(10, ("good copy chunk recv %d of %d\n",
316                            state->recv_count,
317                            state->dispatch_count));
318                 state->total_written += chunk_nwritten;
319         } else {
320                 DEBUG(0, ("bad status in copy chunk recv %d of %d: %s\n",
321                           state->recv_count,
322                           state->dispatch_count,
323                           nt_errstr(status)));
324                 state->bad_recv_count++;
325                 /* may overwrite previous failed status */
326                 state->status = status;
327         }
328
329         if (state->recv_count != state->dispatch_count) {
330                 /*
331                  * Wait for all VFS copy_chunk requests to complete, even
332                  * if an error is received for a specific chunk.
333                  */
334                 return;
335         }
336
337         /* all VFS copy_chunk requests done */
338         copychunk_unlock_all(state->src_fsp,
339                              state->dst_fsp,
340                              state->rlocks,
341                              state->wlocks,
342                              state->num_locks);
343
344         if (!tevent_req_nterror(req, state->status)) {
345                 tevent_req_done(req);
346         }
347 }
348
349 static NTSTATUS fsctl_srv_copychunk_recv(struct tevent_req *req,
350                                          struct srv_copychunk_rsp *cc_rsp,
351                                          bool *pack_rsp)
352 {
353         struct fsctl_srv_copychunk_state *state = tevent_req_data(req,
354                                         struct fsctl_srv_copychunk_state);
355         NTSTATUS status;
356
357         switch (state->out_data) {
358         case COPYCHUNK_OUT_EMPTY:
359                 *pack_rsp = false;
360                 break;
361         case COPYCHUNK_OUT_LIMITS:
362                 /* 2.2.32.1 - send back our maximum transfer size limits */
363                 copychunk_pack_limits(cc_rsp);
364                 *pack_rsp = true;
365                 break;
366         case COPYCHUNK_OUT_RSP:
367                 cc_rsp->chunks_written = state->recv_count - state->bad_recv_count;
368                 cc_rsp->chunk_bytes_written = 0;
369                 cc_rsp->total_bytes_written = state->total_written;
370                 *pack_rsp = true;
371                 break;
372         default:        /* not reached */
373                 assert(1);
374                 break;
375         }
376         status = state->status;
377         tevent_req_received(req);
378
379         return status;
380 }
381
382 static NTSTATUS fsctl_validate_neg_info(TALLOC_CTX *mem_ctx,
383                                         struct tevent_context *ev,
384                                         struct smbXsrv_connection *conn,
385                                         DATA_BLOB *in_input,
386                                         uint32_t in_max_output,
387                                         DATA_BLOB *out_output,
388                                         bool *disconnect)
389 {
390         uint32_t in_capabilities;
391         DATA_BLOB in_guid_blob;
392         struct GUID in_guid;
393         uint16_t in_security_mode;
394         uint16_t in_num_dialects;
395         uint16_t i;
396         DATA_BLOB out_guid_blob;
397         NTSTATUS status;
398
399         if (in_input->length < 0x18) {
400                 return NT_STATUS_INVALID_PARAMETER;
401         }
402
403         in_capabilities = IVAL(in_input->data, 0x00);
404         in_guid_blob = data_blob_const(in_input->data + 0x04, 16);
405         in_security_mode = SVAL(in_input->data, 0x14);
406         in_num_dialects = SVAL(in_input->data, 0x16);
407
408         if (in_input->length < (0x18 + in_num_dialects*2)) {
409                 return NT_STATUS_INVALID_PARAMETER;
410         }
411
412         if (in_max_output < 0x18) {
413                 return NT_STATUS_BUFFER_TOO_SMALL;
414         }
415
416         status = GUID_from_ndr_blob(&in_guid_blob, &in_guid);
417         if (!NT_STATUS_IS_OK(status)) {
418                 return status;
419         }
420
421         if (in_num_dialects != conn->smb2.client.num_dialects) {
422                 *disconnect = true;
423                 return NT_STATUS_ACCESS_DENIED;
424         }
425
426         for (i=0; i < in_num_dialects; i++) {
427                 uint16_t v = SVAL(in_input->data, 0x18 + i*2);
428
429                 if (conn->smb2.client.dialects[i] != v) {
430                         *disconnect = true;
431                         return NT_STATUS_ACCESS_DENIED;
432                 }
433         }
434
435         if (GUID_compare(&in_guid, &conn->smb2.client.guid) != 0) {
436                 *disconnect = true;
437                 return NT_STATUS_ACCESS_DENIED;
438         }
439
440         if (in_security_mode != conn->smb2.client.security_mode) {
441                 *disconnect = true;
442                 return NT_STATUS_ACCESS_DENIED;
443         }
444
445         if (in_capabilities != conn->smb2.client.capabilities) {
446                 *disconnect = true;
447                 return NT_STATUS_ACCESS_DENIED;
448         }
449
450         status = GUID_to_ndr_blob(&conn->smb2.server.guid, mem_ctx,
451                                   &out_guid_blob);
452         if (!NT_STATUS_IS_OK(status)) {
453                 return status;
454         }
455
456         *out_output = data_blob_talloc(mem_ctx, NULL, 0x18);
457         if (out_output->data == NULL) {
458                 return NT_STATUS_NO_MEMORY;
459         }
460
461         SIVAL(out_output->data, 0x00, conn->smb2.server.capabilities);
462         memcpy(out_output->data+0x04, out_guid_blob.data, 16);
463         SIVAL(out_output->data, 0x14, conn->smb2.server.security_mode);
464         SIVAL(out_output->data, 0x16, conn->smb2.server.dialect);
465
466         return NT_STATUS_OK;
467 }
468
469 static NTSTATUS fsctl_srv_req_resume_key(TALLOC_CTX *mem_ctx,
470                                          struct tevent_context *ev,
471                                          struct files_struct *fsp,
472                                          uint32_t in_max_output,
473                                          DATA_BLOB *out_output)
474 {
475         struct req_resume_key_rsp rkey_rsp;
476         enum ndr_err_code ndr_ret;
477         DATA_BLOB output;
478
479         if (fsp == NULL) {
480                 return NT_STATUS_FILE_CLOSED;
481         }
482
483         ZERO_STRUCT(rkey_rsp);
484         /* combine persistent and volatile handles for the resume key */
485         SBVAL(rkey_rsp.resume_key, 0, fsp->op->global->open_persistent_id);
486         SBVAL(rkey_rsp.resume_key, 8, fsp->op->global->open_volatile_id);
487
488         ndr_ret = ndr_push_struct_blob(&output, mem_ctx, &rkey_rsp,
489                         (ndr_push_flags_fn_t)ndr_push_req_resume_key_rsp);
490         if (ndr_ret != NDR_ERR_SUCCESS) {
491                 return NT_STATUS_INTERNAL_ERROR;
492         }
493
494         if (in_max_output < output.length) {
495                 DEBUG(1, ("max output %u too small for resume key rsp %ld\n",
496                           in_max_output, (long int)output.length));
497                 return NT_STATUS_INVALID_PARAMETER;
498         }
499         *out_output = output;
500
501         return NT_STATUS_OK;
502 }
503
504 static void smb2_ioctl_network_fs_copychunk_done(struct tevent_req *subreq);
505
506 struct tevent_req *smb2_ioctl_network_fs(uint32_t ctl_code,
507                                          struct tevent_context *ev,
508                                          struct tevent_req *req,
509                                          struct smbd_smb2_ioctl_state *state)
510 {
511         struct tevent_req *subreq;
512         NTSTATUS status;
513
514         switch (ctl_code) {
515         case FSCTL_SRV_COPYCHUNK:
516                 subreq = fsctl_srv_copychunk_send(state, ev, state->fsp,
517                                                   &state->in_input,
518                                                   state->smb2req);
519                 if (tevent_req_nomem(subreq, req)) {
520                         return tevent_req_post(req, ev);
521                 }
522                 tevent_req_set_callback(subreq,
523                                         smb2_ioctl_network_fs_copychunk_done,
524                                         req);
525                 return req;
526                 break;
527         case FSCTL_VALIDATE_NEGOTIATE_INFO:
528                 status = fsctl_validate_neg_info(state, ev,
529                                                  state->smbreq->sconn->conn,
530                                                  &state->in_input,
531                                                  state->in_max_output,
532                                                  &state->out_output,
533                                                  &state->disconnect);
534                 if (!tevent_req_nterror(req, status)) {
535                         tevent_req_done(req);
536                 }
537                 return tevent_req_post(req, ev);
538                 break;
539         case FSCTL_SRV_REQUEST_RESUME_KEY:
540                 status = fsctl_srv_req_resume_key(state, ev, state->fsp,
541                                                   state->in_max_output,
542                                                   &state->out_output);
543                 if (!tevent_req_nterror(req, status)) {
544                         tevent_req_done(req);
545                 }
546                 return tevent_req_post(req, ev);
547                 break;
548         default: {
549                 uint8_t *out_data = NULL;
550                 uint32_t out_data_len = 0;
551
552                 if (state->fsp == NULL) {
553                         status = NT_STATUS_NOT_SUPPORTED;
554                 } else {
555                         status = SMB_VFS_FSCTL(state->fsp,
556                                                state,
557                                                ctl_code,
558                                                state->smbreq->flags2,
559                                                state->in_input.data,
560                                                state->in_input.length,
561                                                &out_data,
562                                                state->in_max_output,
563                                                &out_data_len);
564                         state->out_output = data_blob_const(out_data, out_data_len);
565                         if (NT_STATUS_IS_OK(status)) {
566                                 tevent_req_done(req);
567                                 return tevent_req_post(req, ev);
568                         }
569                 }
570
571                 if (NT_STATUS_EQUAL(status, NT_STATUS_NOT_SUPPORTED)) {
572                         if (IS_IPC(state->smbreq->conn)) {
573                                 status = NT_STATUS_FS_DRIVER_REQUIRED;
574                         } else {
575                                 status = NT_STATUS_INVALID_DEVICE_REQUEST;
576                         }
577                 }
578
579                 tevent_req_nterror(req, status);
580                 return tevent_req_post(req, ev);
581                 break;
582         }
583         }
584
585         tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR);
586         return tevent_req_post(req, ev);
587 }
588
589 static void smb2_ioctl_network_fs_copychunk_done(struct tevent_req *subreq)
590 {
591         struct tevent_req *req = tevent_req_callback_data(subreq,
592                                                           struct tevent_req);
593         struct smbd_smb2_ioctl_state *ioctl_state = tevent_req_data(req,
594                                                 struct smbd_smb2_ioctl_state);
595         struct srv_copychunk_rsp cc_rsp;
596         NTSTATUS status;
597         bool pack_rsp = false;
598
599         ZERO_STRUCT(cc_rsp);
600         status = fsctl_srv_copychunk_recv(subreq, &cc_rsp, &pack_rsp);
601         TALLOC_FREE(subreq);
602         if (pack_rsp == true) {
603                 enum ndr_err_code ndr_ret;
604                 ndr_ret = ndr_push_struct_blob(&ioctl_state->out_output,
605                                                ioctl_state,
606                                                &cc_rsp,
607                                 (ndr_push_flags_fn_t)ndr_push_srv_copychunk_rsp);
608                 if (ndr_ret != NDR_ERR_SUCCESS) {
609                         status = NT_STATUS_INTERNAL_ERROR;
610                 }
611         }
612
613         if (!tevent_req_nterror(req, status)) {
614                 tevent_req_done(req);
615         }
616 }