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