lib: modules: Change XXX_init interface from XXX_init(void) to XXX_init(TALLOC_CTX *)
[amitay/samba.git] / source3 / smbd / smb2_ioctl.c
index 56c075e1dbe914a4ddcc97c9597558e0d9a42b05..993682ffdbbfc081e7002a648af0fe8c86ce075b 100644 (file)
@@ -23,9 +23,9 @@
 #include "smbd/globals.h"
 #include "../libcli/smb/smb_common.h"
 #include "../lib/util/tevent_ntstatus.h"
-#include "rpc_server/srv_pipe_hnd.h"
 #include "include/ntioctl.h"
-#include "../librpc/ndr/libndr.h"
+#include "smb2_ioctl_private.h"
+#include "librpc/gen_ndr/ioctl.h"
 
 static struct tevent_req *smbd_smb2_ioctl_send(TALLOC_CTX *mem_ctx,
                                               struct tevent_context *ev,
@@ -45,7 +45,6 @@ NTSTATUS smbd_smb2_request_process_ioctl(struct smbd_smb2_request *req)
 {
        NTSTATUS status;
        const uint8_t *inbody;
-       int i = req->current_idx;
        uint32_t min_buffer_offset;
        uint32_t max_buffer_offset;
        uint32_t min_output_offset;
@@ -74,7 +73,7 @@ NTSTATUS smbd_smb2_request_process_ioctl(struct smbd_smb2_request *req)
        if (!NT_STATUS_IS_OK(status)) {
                return smbd_smb2_request_error(req, status);
        }
-       inbody = (const uint8_t *)req->in.vector[i+1].iov_base;
+       inbody = SMBD_SMB2_IN_BODY_PTR(req);
 
        in_ctl_code             = IVAL(inbody, 0x04);
        in_file_id_persistent   = BVAL(inbody, 0x08);
@@ -87,8 +86,8 @@ NTSTATUS smbd_smb2_request_process_ioctl(struct smbd_smb2_request *req)
        in_max_output_length    = IVAL(inbody, 0x2C);
        in_flags                = IVAL(inbody, 0x30);
 
-       min_buffer_offset = SMB2_HDR_BODY + req->in.vector[i+1].iov_len;
-       max_buffer_offset = min_buffer_offset + req->in.vector[i+2].iov_len;
+       min_buffer_offset = SMB2_HDR_BODY + SMBD_SMB2_IN_BODY_LEN(req);
+       max_buffer_offset = min_buffer_offset + SMBD_SMB2_IN_DYN_LEN(req);
        min_output_offset = min_buffer_offset;
 
        /*
@@ -114,7 +113,7 @@ NTSTATUS smbd_smb2_request_process_ioctl(struct smbd_smb2_request *req)
                allowed_length_in = max_buffer_offset - in_input_offset;
 
                tmp_ofs = in_input_offset - min_buffer_offset;
-               in_input_buffer.data = (uint8_t *)req->in.vector[i+2].iov_base;
+               in_input_buffer.data = SMBD_SMB2_IN_DYN_PTR(req);
                in_input_buffer.data += tmp_ofs;
                in_input_buffer.length = in_input_length;
                min_output_offset += tmp_ofs;
@@ -151,7 +150,7 @@ NTSTATUS smbd_smb2_request_process_ioctl(struct smbd_smb2_request *req)
                }
 
                tmp_ofs = in_output_offset - min_buffer_offset;
-               in_output_buffer.data = (uint8_t *)req->in.vector[i+2].iov_base;
+               in_output_buffer.data = SMBD_SMB2_IN_DYN_PTR(req);
                in_output_buffer.data += tmp_ofs;
                in_output_buffer.length = in_output_length;
        }
@@ -229,12 +228,47 @@ NTSTATUS smbd_smb2_request_process_ioctl(struct smbd_smb2_request *req)
        return smbd_smb2_request_pending_queue(req, subreq, 1000);
 }
 
+/*
+ * 3.3.4.4 Sending an Error Response
+ * An error code other than one of the following indicates a failure:
+ */
+static bool smbd_smb2_ioctl_is_failure(uint32_t ctl_code, NTSTATUS status,
+                                      size_t data_size)
+{
+       if (NT_STATUS_IS_OK(status)) {
+               return false;
+       }
+
+       /*
+        * STATUS_BUFFER_OVERFLOW in a FSCTL_PIPE_TRANSCEIVE, FSCTL_PIPE_PEEK or
+        * FSCTL_DFS_GET_REFERRALS Response specified in section 2.2.32.<153>
+        */
+       if (NT_STATUS_EQUAL(status, STATUS_BUFFER_OVERFLOW)
+        && ((ctl_code == FSCTL_PIPE_TRANSCEIVE)
+         || (ctl_code == FSCTL_PIPE_PEEK)
+         || (ctl_code == FSCTL_DFS_GET_REFERRALS))) {
+               return false;
+       }
+
+       /*
+        * Any status other than STATUS_SUCCESS in a FSCTL_SRV_COPYCHUNK or
+        * FSCTL_SRV_COPYCHUNK_WRITE Response, when returning an
+        * SRV_COPYCHUNK_RESPONSE as described in section 2.2.32.1.
+        */
+       if (((ctl_code == FSCTL_SRV_COPYCHUNK)
+                               || (ctl_code == FSCTL_SRV_COPYCHUNK_WRITE))
+        && (data_size == sizeof(struct srv_copychunk_rsp))) {
+               return false;
+       }
+
+       return true;
+}
+
 static void smbd_smb2_request_ioctl_done(struct tevent_req *subreq)
 {
        struct smbd_smb2_request *req = tevent_req_callback_data(subreq,
                                        struct smbd_smb2_request);
        const uint8_t *inbody;
-       int i = req->current_idx;
        DATA_BLOB outbody;
        DATA_BLOB outdyn;
        uint32_t in_ctl_code;
@@ -259,17 +293,22 @@ static void smbd_smb2_request_ioctl_done(struct tevent_req *subreq)
        TALLOC_FREE(subreq);
        if (disconnect) {
                error = status;
-               smbd_server_connection_terminate(req->sconn,
+               smbd_server_connection_terminate(req->xconn,
                                                 nt_errstr(error));
                return;
        }
 
-       if (NT_STATUS_EQUAL(status, STATUS_BUFFER_OVERFLOW)) {
-               /* also ok */
-       } else if (!NT_STATUS_IS_OK(status)) {
+       inbody = SMBD_SMB2_IN_BODY_PTR(req);
+
+       in_ctl_code             = IVAL(inbody, 0x04);
+       in_file_id_persistent   = BVAL(inbody, 0x08);
+       in_file_id_volatile     = BVAL(inbody, 0x10);
+
+       if (smbd_smb2_ioctl_is_failure(in_ctl_code, status,
+                                      out_output_buffer.length)) {
                error = smbd_smb2_request_error(req, status);
                if (!NT_STATUS_IS_OK(error)) {
-                       smbd_server_connection_terminate(req->sconn,
+                       smbd_server_connection_terminate(req->xconn,
                                                         nt_errstr(error));
                        return;
                }
@@ -279,17 +318,11 @@ static void smbd_smb2_request_ioctl_done(struct tevent_req *subreq)
        out_input_offset = SMB2_HDR_BODY + 0x30;
        out_output_offset = SMB2_HDR_BODY + 0x30;
 
-       inbody = (const uint8_t *)req->in.vector[i+1].iov_base;
-
-       in_ctl_code             = IVAL(inbody, 0x04);
-       in_file_id_persistent   = BVAL(inbody, 0x08);
-       in_file_id_volatile     = BVAL(inbody, 0x10);
-
-       outbody = data_blob_talloc(req->out.vector, NULL, 0x30);
+       outbody = smbd_smb2_generate_outbody(req, 0x30);
        if (outbody.data == NULL) {
                error = smbd_smb2_request_error(req, NT_STATUS_NO_MEMORY);
                if (!NT_STATUS_IS_OK(error)) {
-                       smbd_server_connection_terminate(req->sconn,
+                       smbd_server_connection_terminate(req->xconn,
                                                         nt_errstr(error));
                        return;
                }
@@ -324,25 +357,12 @@ static void smbd_smb2_request_ioctl_done(struct tevent_req *subreq)
        error = smbd_smb2_request_done_ex(req, status, outbody, &outdyn,
                                          __location__);
        if (!NT_STATUS_IS_OK(error)) {
-               smbd_server_connection_terminate(req->sconn,
+               smbd_server_connection_terminate(req->xconn,
                                                 nt_errstr(error));
                return;
        }
 }
 
-struct smbd_smb2_ioctl_state {
-       struct smbd_smb2_request *smb2req;
-       struct smb_request *smbreq;
-       files_struct *fsp;
-       DATA_BLOB in_input;
-       uint32_t in_max_output;
-       DATA_BLOB out_output;
-       bool disconnect;
-};
-
-static void smbd_smb2_ioctl_pipe_write_done(struct tevent_req *subreq);
-static void smbd_smb2_ioctl_pipe_read_done(struct tevent_req *subreq);
-
 static struct tevent_req *smbd_smb2_ioctl_send(TALLOC_CTX *mem_ctx,
                                               struct tevent_context *ev,
                                               struct smbd_smb2_request *smb2req,
@@ -355,7 +375,6 @@ static struct tevent_req *smbd_smb2_ioctl_send(TALLOC_CTX *mem_ctx,
        struct tevent_req *req;
        struct smbd_smb2_ioctl_state *state;
        struct smb_request *smbreq;
-       struct tevent_req *subreq;
 
        req = tevent_req_create(mem_ctx, &state,
                                struct smbd_smb2_ioctl_state);
@@ -380,335 +399,34 @@ static struct tevent_req *smbd_smb2_ioctl_send(TALLOC_CTX *mem_ctx,
        }
        state->smbreq = smbreq;
 
-       switch (in_ctl_code) {
-       case 0x00060194: /* FSCTL_DFS_GET_REFERRALS */
-       {
-               uint16_t in_max_referral_level;
-               DATA_BLOB in_file_name_buffer;
-               char *in_file_name_string;
-               size_t in_file_name_string_size;
-               bool ok;
-               bool overflow = false;
-               NTSTATUS status;
-               int dfs_size;
-               char *dfs_data = NULL;
-
-               if (!IS_IPC(smbreq->conn)) {
-                       tevent_req_nterror(req, NT_STATUS_INVALID_DEVICE_REQUEST);
-                       return tevent_req_post(req, ev);
-               }
-
-               if (!lp_host_msdfs()) {
+       switch (in_ctl_code & IOCTL_DEV_TYPE_MASK) {
+       case FSCTL_DFS:
+               return smb2_ioctl_dfs(in_ctl_code, ev, req, state);
+               break;
+       case FSCTL_FILESYSTEM:
+               return smb2_ioctl_filesys(in_ctl_code, ev, req, state);
+               break;
+       case FSCTL_NAMED_PIPE:
+               return smb2_ioctl_named_pipe(in_ctl_code, ev, req, state);
+               break;
+       case FSCTL_NETWORK_FILESYSTEM:
+               return smb2_ioctl_network_fs(in_ctl_code, ev, req, state);
+               break;
+       default:
+               if (IS_IPC(smbreq->conn)) {
                        tevent_req_nterror(req, NT_STATUS_FS_DRIVER_REQUIRED);
-                       return tevent_req_post(req, ev);
-               }
-
-               if (in_input.length < (2 + 2)) {
-                       tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
-                       return tevent_req_post(req, ev);
-               }
-
-               in_max_referral_level = SVAL(in_input.data, 0);
-               in_file_name_buffer.data = in_input.data + 2;
-               in_file_name_buffer.length = in_input.length - 2;
-
-               ok = convert_string_talloc(state, CH_UTF16, CH_UNIX,
-                                          in_file_name_buffer.data,
-                                          in_file_name_buffer.length,
-                                          &in_file_name_string,
-                                          &in_file_name_string_size);
-               if (!ok) {
-                       tevent_req_nterror(req, NT_STATUS_ILLEGAL_CHARACTER);
-                       return tevent_req_post(req, ev);
-               }
-
-               dfs_size = setup_dfs_referral(smbreq->conn,
-                                             in_file_name_string,
-                                             in_max_referral_level,
-                                             &dfs_data, &status);
-               if (dfs_size < 0) {
-                       tevent_req_nterror(req, status);
-                       return tevent_req_post(req, ev);
-               }
-
-               if (dfs_size > in_max_output) {
-                       /*
-                        * TODO: we need a testsuite for this
-                        */
-                       overflow = true;
-                       dfs_size = in_max_output;
-               }
-
-               state->out_output = data_blob_talloc(state,
-                                                    (uint8_t *)dfs_data,
-                                                    dfs_size);
-               SAFE_FREE(dfs_data);
-               if (dfs_size > 0 &&
-                   tevent_req_nomem(state->out_output.data, req)) {
-                       return tevent_req_post(req, ev);
-               }
-
-               if (overflow) {
-                       tevent_req_nterror(req, STATUS_BUFFER_OVERFLOW);
                } else {
-                       tevent_req_done(req);
-               }
-               return tevent_req_post(req, ev);
-       }
-       case 0x0011C017: /* FSCTL_PIPE_TRANSCEIVE */
-
-               if (!IS_IPC(smbreq->conn)) {
-                       tevent_req_nterror(req, NT_STATUS_NOT_SUPPORTED);
-                       return tevent_req_post(req, ev);
-               }
-
-               if (fsp == NULL) {
-                       tevent_req_nterror(req, NT_STATUS_FILE_CLOSED);
-                       return tevent_req_post(req, ev);
-               }
-
-               if (!fsp_is_np(fsp)) {
-                       tevent_req_nterror(req, NT_STATUS_FILE_CLOSED);
-                       return tevent_req_post(req, ev);
-               }
-
-               DEBUG(10,("smbd_smb2_ioctl_send: np_write_send of size %u\n",
-                       (unsigned int)in_input.length ));
-
-               subreq = np_write_send(state, ev,
-                                      fsp->fake_file_handle,
-                                      in_input.data,
-                                      in_input.length);
-               if (tevent_req_nomem(subreq, req)) {
-                       return tevent_req_post(req, ev);
-               }
-               tevent_req_set_callback(subreq,
-                                       smbd_smb2_ioctl_pipe_write_done,
-                                       req);
-               return req;
-
-       case FSCTL_VALIDATE_NEGOTIATE_INFO:
-       {
-               struct smbXsrv_connection *conn = smbreq->sconn->conn;
-               uint32_t in_capabilities;
-               DATA_BLOB in_guid_blob;
-               struct GUID in_guid;
-               uint16_t in_security_mode;
-               uint16_t in_num_dialects;
-               uint16_t i;
-               DATA_BLOB out_guid_blob;
-               NTSTATUS status;
-
-               if (in_input.length < 0x18) {
-                       tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
-                       return tevent_req_post(req, ev);
-               }
-
-               in_capabilities = IVAL(in_input.data, 0x00);
-               in_guid_blob = data_blob_const(in_input.data + 0x04, 16);
-               in_security_mode = SVAL(in_input.data, 0x14);
-               in_num_dialects = SVAL(in_input.data, 0x16);
-
-               if (in_input.length != (0x18 + in_num_dialects*2)) {
-                       tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
-                       return tevent_req_post(req, ev);
-               }
-
-               if (in_max_output < 0x18) {
-                       tevent_req_nterror(req, NT_STATUS_BUFFER_TOO_SMALL);
-                       return tevent_req_post(req, ev);
-               }
-
-               status = GUID_from_ndr_blob(&in_guid_blob, &in_guid);
-               if (tevent_req_nterror(req, status)) {
-                       return tevent_req_post(req, ev);
-               }
-
-               if (in_num_dialects != conn->smb2.client.num_dialects) {
-                       state->disconnect = true;
-                       tevent_req_nterror(req, NT_STATUS_ACCESS_DENIED);
-                       return tevent_req_post(req, ev);
-               }
-
-               for (i=0; i < in_num_dialects; i++) {
-                       uint16_t v = SVAL(in_input.data, 0x18 + i*2);
-
-                       if (conn->smb2.client.dialects[i] != v) {
-                               state->disconnect = true;
-                               tevent_req_nterror(req, NT_STATUS_ACCESS_DENIED);
-                               return tevent_req_post(req, ev);
-                       }
-               }
-
-               if (!GUID_compare(&in_guid, &conn->smb2.client.guid)) {
-                       state->disconnect = true;
-                       tevent_req_nterror(req, NT_STATUS_ACCESS_DENIED);
-                       return tevent_req_post(req, ev);
-               }
-
-               if (in_security_mode != conn->smb2.client.security_mode) {
-                       state->disconnect = true;
-                       tevent_req_nterror(req, NT_STATUS_ACCESS_DENIED);
-                       return tevent_req_post(req, ev);
-               }
-
-               if (in_capabilities != conn->smb2.client.capabilities) {
-                       state->disconnect = true;
-                       tevent_req_nterror(req, NT_STATUS_ACCESS_DENIED);
-                       return tevent_req_post(req, ev);
-               }
-
-               status = GUID_to_ndr_blob(&conn->smb2.server.guid, state,
-                                         &out_guid_blob);
-               if (tevent_req_nterror(req, status)) {
-                       return tevent_req_post(req, ev);
-               }
-
-               state->out_output = data_blob_talloc(state, NULL, 0x18);
-               if (tevent_req_nomem(state->out_output.data, req)) {
-                       return tevent_req_post(req, ev);
-               }
-
-               SIVAL(state->out_output.data, 0x00, conn->smb2.server.capabilities);
-               memcpy(state->out_output.data+0x04, out_guid_blob.data, 16);
-               SIVAL(state->out_output.data, 0x14, conn->smb2.server.security_mode);
-               SIVAL(state->out_output.data, 0x16, conn->smb2.server.dialect);
-
-               tevent_req_done(req);
-               return tevent_req_post(req, ev);
-       }
-
-       default: {
-               uint8_t *out_data = NULL;
-               uint32_t out_data_len = 0;
-               NTSTATUS status;
-
-               if (fsp == NULL) {
-                       tevent_req_nterror(req, NT_STATUS_FILE_CLOSED);
-                       return tevent_req_post(req, ev);
-               }
-
-               status = SMB_VFS_FSCTL(fsp,
-                                      state,
-                                      in_ctl_code,
-                                      smbreq->flags2,
-                                      in_input.data,
-                                      in_input.length,
-                                      &out_data,
-                                      in_max_output,
-                                      &out_data_len);
-               state->out_output = data_blob_const(out_data, out_data_len);
-               if (NT_STATUS_IS_OK(status)) {
-                       tevent_req_done(req);
-                       return tevent_req_post(req, ev);
-               }
-
-               if (NT_STATUS_EQUAL(status, NT_STATUS_NOT_SUPPORTED)) {
-                       if (IS_IPC(smbreq->conn)) {
-                               status = NT_STATUS_FS_DRIVER_REQUIRED;
-                       } else {
-                               status = NT_STATUS_INVALID_DEVICE_REQUEST;
-                       }
+                       tevent_req_nterror(req, NT_STATUS_INVALID_DEVICE_REQUEST);
                }
 
-               tevent_req_nterror(req, status);
                return tevent_req_post(req, ev);
-       }
+               break;
        }
 
        tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR);
        return tevent_req_post(req, ev);
 }
 
-static void smbd_smb2_ioctl_pipe_write_done(struct tevent_req *subreq)
-{
-       struct tevent_req *req = tevent_req_callback_data(subreq,
-                                struct tevent_req);
-       struct smbd_smb2_ioctl_state *state = tevent_req_data(req,
-                                             struct smbd_smb2_ioctl_state);
-       NTSTATUS status;
-       ssize_t nwritten = -1;
-
-       status = np_write_recv(subreq, &nwritten);
-
-       DEBUG(10,("smbd_smb2_ioctl_pipe_write_done: received %ld\n",
-               (long int)nwritten ));
-
-       TALLOC_FREE(subreq);
-       if (!NT_STATUS_IS_OK(status)) {
-               NTSTATUS old = status;
-               status = nt_status_np_pipe(old);
-               tevent_req_nterror(req, status);
-               return;
-       }
-
-       if (nwritten != state->in_input.length) {
-               tevent_req_nterror(req, NT_STATUS_PIPE_NOT_AVAILABLE);
-               return;
-       }
-
-       state->out_output = data_blob_talloc(state, NULL, state->in_max_output);
-       if (state->in_max_output > 0 &&
-           tevent_req_nomem(state->out_output.data, req)) {
-               return;
-       }
-
-       DEBUG(10,("smbd_smb2_ioctl_pipe_write_done: issuing np_read_send "
-               "of size %u\n",
-               (unsigned int)state->out_output.length ));
-
-       TALLOC_FREE(subreq);
-       subreq = np_read_send(state->smbreq->conn,
-                             state->smb2req->sconn->ev_ctx,
-                             state->fsp->fake_file_handle,
-                             state->out_output.data,
-                             state->out_output.length);
-       if (tevent_req_nomem(subreq, req)) {
-               return;
-       }
-       tevent_req_set_callback(subreq, smbd_smb2_ioctl_pipe_read_done, req);
-}
-
-static void smbd_smb2_ioctl_pipe_read_done(struct tevent_req *subreq)
-{
-       struct tevent_req *req = tevent_req_callback_data(subreq,
-                                struct tevent_req);
-       struct smbd_smb2_ioctl_state *state = tevent_req_data(req,
-                                             struct smbd_smb2_ioctl_state);
-       NTSTATUS status;
-       NTSTATUS old;
-       ssize_t nread = -1;
-       bool is_data_outstanding = false;
-
-       status = np_read_recv(subreq, &nread, &is_data_outstanding);
-       TALLOC_FREE(subreq);
-
-       old = status;
-       status = nt_status_np_pipe(old);
-
-       DEBUG(10,("smbd_smb2_ioctl_pipe_read_done: np_read_recv nread = %d "
-                "is_data_outstanding = %d, status = %s%s%s\n",
-               (int)nread,
-               (int)is_data_outstanding,
-               nt_errstr(old),
-               NT_STATUS_EQUAL(old, status)?"":" => ",
-               NT_STATUS_EQUAL(old, status)?"":nt_errstr(status)));
-
-       if (!NT_STATUS_IS_OK(status)) {
-               tevent_req_nterror(req, status);
-               return;
-       }
-
-       state->out_output.length = nread;
-
-       if (is_data_outstanding) {
-               tevent_req_nterror(req, STATUS_BUFFER_OVERFLOW);
-               return;
-       }
-
-       tevent_req_done(req);
-}
-
 static NTSTATUS smbd_smb2_ioctl_recv(struct tevent_req *req,
                                     TALLOC_CTX *mem_ctx,
                                     DATA_BLOB *out_output,
@@ -717,19 +435,23 @@ static NTSTATUS smbd_smb2_ioctl_recv(struct tevent_req *req,
        NTSTATUS status = NT_STATUS_OK;
        struct smbd_smb2_ioctl_state *state = tevent_req_data(req,
                                              struct smbd_smb2_ioctl_state);
+       enum tevent_req_state req_state;
+       uint64_t err;
 
        *disconnect = state->disconnect;
 
-       if (tevent_req_is_nterror(req, &status)) {
-               if (!NT_STATUS_EQUAL(status, STATUS_BUFFER_OVERFLOW)) {
-                       tevent_req_received(req);
-                       return status;
-               }
+       if ((tevent_req_is_error(req, &req_state, &err) == false)
+        || (req_state == TEVENT_REQ_USER_ERROR)) {
+               /*
+                * Return output buffer to caller if the ioctl was successfully
+                * processed, even if a user error occurred. Some ioctls return
+                * data on failure.
+                */
+               *out_output = state->out_output;
+               talloc_steal(mem_ctx, out_output->data);
        }
 
-       *out_output = state->out_output;
-       talloc_steal(mem_ctx, out_output->data);
-
+       tevent_req_is_nterror(req, &status);
        tevent_req_received(req);
        return status;
 }