s3: Refactor smbd_smb2_request_process_negprot
[metze/samba/wip.git] / source3 / smbd / smb2_negprot.c
index 56a30d0d71ea229cbe8a5ec0f4c10e70ebe14395..5fa1fbbe008bdbf90e5f746e97925ea246e7e9ca 100644 (file)
 #include "smbd/smbd.h"
 #include "smbd/globals.h"
 #include "../libcli/smb/smb_common.h"
+#include "../lib/tsocket/tsocket.h"
+#include "../librpc/ndr/libndr.h"
+
+extern fstring remote_proto;
 
 /*
  * this is the entry point if SMB2 is selected via
@@ -78,86 +82,153 @@ void reply_smb20ff(struct smb_request *req, uint16_t choice)
        reply_smb20xx(req, SMB2_DIALECT_REVISION_2FF);
 }
 
-NTSTATUS smbd_smb2_request_process_negprot(struct smbd_smb2_request *req)
+enum protocol_types smbd_smb2_protocol_dialect_match(const uint8_t *indyn,
+                               const int dialect_count,
+                               uint16_t *dialect)
 {
-       const uint8_t *inbody;
-       const uint8_t *indyn = NULL;
-       int i = req->current_idx;
-       DATA_BLOB outbody;
-       DATA_BLOB outdyn;
-       DATA_BLOB negprot_spnego_blob;
-       uint16_t security_offset;
-       DATA_BLOB security_buffer;
-       size_t expected_body_size = 0x24;
-       size_t body_size;
-       size_t expected_dyn_size = 0;
-       size_t c;
-       uint16_t security_mode;
-       uint16_t dialect_count;
-       uint16_t dialect = 0;
-       uint32_t capabilities;
+       size_t c = 0;
        enum protocol_types protocol = PROTOCOL_NONE;
-       uint32_t max_limit;
-       uint32_t max_trans = lp_smb2_max_trans();
-       uint32_t max_read = lp_smb2_max_read();
-       uint32_t max_write = lp_smb2_max_write();
 
-/* TODO: drop the connection with INVALID_PARAMETER */
+       for (c=0; protocol == PROTOCOL_NONE && c < dialect_count; c++) {
+               if (lp_server_max_protocol() < PROTOCOL_SMB3_00) {
+                       break;
+               }
+               if (lp_server_min_protocol() > PROTOCOL_SMB3_00) {
+                       break;
+               }
 
-       if (req->in.vector[i+1].iov_len != (expected_body_size & 0xFFFFFFFE)) {
-               return smbd_smb2_request_error(req, NT_STATUS_INVALID_PARAMETER);
+               *dialect = SVAL(indyn, c*2);
+               if (*dialect == SMB3_DIALECT_REVISION_300) {
+                       protocol = PROTOCOL_SMB3_00;
+                       break;
+               }
        }
 
-       inbody = (const uint8_t *)req->in.vector[i+1].iov_base;
+       for (c=0; protocol == PROTOCOL_NONE && c < dialect_count; c++) {
+               if (lp_server_max_protocol() < PROTOCOL_SMB2_24) {
+                       break;
+               }
+               if (lp_server_min_protocol() > PROTOCOL_SMB2_24) {
+                       break;
+               }
 
-       body_size = SVAL(inbody, 0x00);
-       if (body_size != expected_body_size) {
-               return smbd_smb2_request_error(req, NT_STATUS_INVALID_PARAMETER);
+               *dialect = SVAL(indyn, c*2);
+               if (*dialect == SMB2_DIALECT_REVISION_224) {
+                       protocol = PROTOCOL_SMB2_24;
+                       break;
+               }
        }
 
-       dialect_count = SVAL(inbody, 0x02);
-       if (dialect_count == 0) {
-               return smbd_smb2_request_error(req, NT_STATUS_INVALID_PARAMETER);
-       }
+       for (c=0; protocol == PROTOCOL_NONE && c < dialect_count; c++) {
+               if (lp_server_max_protocol() < PROTOCOL_SMB2_22) {
+                       break;
+               }
+               if (lp_server_min_protocol() > PROTOCOL_SMB2_22) {
+                       break;
+               }
 
-       expected_dyn_size = dialect_count * 2;
-       if (req->in.vector[i+2].iov_len < expected_dyn_size) {
-               return smbd_smb2_request_error(req, NT_STATUS_INVALID_PARAMETER);
+               *dialect = SVAL(indyn, c*2);
+               if (*dialect == SMB2_DIALECT_REVISION_222) {
+                       protocol = PROTOCOL_SMB2_22;
+                       break;
+               }
        }
-       indyn = (const uint8_t *)req->in.vector[i+2].iov_base;
 
        for (c=0; protocol == PROTOCOL_NONE && c < dialect_count; c++) {
-               if (lp_maxprotocol() < PROTOCOL_SMB2_10) {
+               if (lp_server_max_protocol() < PROTOCOL_SMB2_10) {
                        break;
                }
-               if (lp_minprotocol() > PROTOCOL_SMB2_10) {
+               if (lp_server_min_protocol() > PROTOCOL_SMB2_10) {
                        break;
                }
 
-               dialect = SVAL(indyn, c*2);
-               if (dialect == SMB2_DIALECT_REVISION_210) {
+               *dialect = SVAL(indyn, c*2);
+               if (*dialect == SMB2_DIALECT_REVISION_210) {
                        protocol = PROTOCOL_SMB2_10;
                        break;
                }
        }
 
        for (c=0; protocol == PROTOCOL_NONE && c < dialect_count; c++) {
-               if (lp_maxprotocol() < PROTOCOL_SMB2_02) {
+               if (lp_server_max_protocol() < PROTOCOL_SMB2_02) {
                        break;
                }
-               if (lp_minprotocol() > PROTOCOL_SMB2_02) {
+               if (lp_server_min_protocol() > PROTOCOL_SMB2_02) {
                        break;
                }
 
-               dialect = SVAL(indyn, c*2);
-               if (dialect == SMB2_DIALECT_REVISION_202) {
+               *dialect = SVAL(indyn, c*2);
+               if (*dialect == SMB2_DIALECT_REVISION_202) {
                        protocol = PROTOCOL_SMB2_02;
                        break;
                }
        }
 
+       return protocol;
+}
+
+NTSTATUS smbd_smb2_request_process_negprot(struct smbd_smb2_request *req)
+{
+       NTSTATUS status;
+       const uint8_t *inbody;
+       const uint8_t *indyn = NULL;
+       DATA_BLOB outbody;
+       DATA_BLOB outdyn;
+       DATA_BLOB negprot_spnego_blob;
+       uint16_t security_offset;
+       DATA_BLOB security_buffer;
+       size_t expected_dyn_size = 0;
+       size_t c;
+       uint16_t security_mode;
+       uint16_t dialect_count;
+       uint16_t in_security_mode;
+       uint32_t in_capabilities;
+       DATA_BLOB in_guid_blob;
+       struct GUID in_guid;
+       uint16_t dialect = 0;
+       uint32_t capabilities;
+       DATA_BLOB out_guid_blob;
+       struct GUID out_guid;
+       enum protocol_types protocol = PROTOCOL_NONE;
+       uint32_t max_limit;
+       uint32_t max_trans = lp_smb2_max_trans();
+       uint32_t max_read = lp_smb2_max_read();
+       uint32_t max_write = lp_smb2_max_write();
+       NTTIME now = timeval_to_nttime(&req->request_time);
+
+       status = smbd_smb2_request_verify_sizes(req, 0x24);
+       if (!NT_STATUS_IS_OK(status)) {
+               return smbd_smb2_request_error(req, status);
+       }
+       inbody = SMBD_SMB2_IN_BODY_PTR(req);
+
+       dialect_count = SVAL(inbody, 0x02);
+
+       in_security_mode = SVAL(inbody, 0x04);
+       in_capabilities = IVAL(inbody, 0x08);
+       in_guid_blob = data_blob_const(inbody + 0x0C, 16);
+
+       if (dialect_count == 0) {
+               return smbd_smb2_request_error(req, NT_STATUS_INVALID_PARAMETER);
+       }
+
+       status = GUID_from_ndr_blob(&in_guid_blob, &in_guid);
+       if (!NT_STATUS_IS_OK(status)) {
+               return smbd_smb2_request_error(req, status);
+       }
+
+       expected_dyn_size = dialect_count * 2;
+       if (SMBD_SMB2_IN_DYN_LEN(req) < expected_dyn_size) {
+               return smbd_smb2_request_error(req, NT_STATUS_INVALID_PARAMETER);
+       }
+       indyn = SMBD_SMB2_IN_DYN_PTR(req);
+
+       protocol = smbd_smb2_protocol_dialect_match(indyn,
+                                       dialect_count,
+                                       &dialect);
+
        for (c=0; protocol == PROTOCOL_NONE && c < dialect_count; c++) {
-               if (lp_maxprotocol() < PROTOCOL_SMB2_10) {
+               if (lp_server_max_protocol() < PROTOCOL_SMB2_10) {
                        break;
                }
 
@@ -175,14 +246,16 @@ NTSTATUS smbd_smb2_request_process_negprot(struct smbd_smb2_request *req)
                return smbd_smb2_request_error(req, NT_STATUS_NOT_SUPPORTED);
        }
 
-       if (dialect != SMB2_DIALECT_REVISION_2FF) {
-               set_Protocol(protocol);
-       }
-
        if (get_remote_arch() != RA_SAMBA) {
                set_remote_arch(RA_VISTA);
        }
 
+       fstr_sprintf(remote_proto, "SMB%X_%02X",
+                    (dialect >> 8) & 0xFF, dialect & 0xFF);
+
+       reload_services(req->sconn, conn_snum_used, true);
+       DEBUG(3,("Selected protocol %s\n", remote_proto));
+
        /* negprot_spnego() returns a the server guid in the first 16 bytes */
        negprot_spnego_blob = negprot_spnego(req, req->sconn);
        if (negprot_spnego_blob.data == NULL) {
@@ -194,7 +267,7 @@ NTSTATUS smbd_smb2_request_process_negprot(struct smbd_smb2_request *req)
        }
 
        security_mode = SMB2_NEGOTIATE_SIGNING_ENABLED;
-       if (lp_server_signing() == Required) {
+       if (lp_server_signing() == SMB_SIGNING_REQUIRED) {
                security_mode |= SMB2_NEGOTIATE_SIGNING_REQUIRED;
        }
 
@@ -203,15 +276,46 @@ NTSTATUS smbd_smb2_request_process_negprot(struct smbd_smb2_request *req)
                capabilities |= SMB2_CAP_DFS;
        }
 
+       if ((protocol >= PROTOCOL_SMB2_24) &&
+           (lp_smb_encrypt(-1) != SMB_SIGNING_OFF) &&
+           (in_capabilities & SMB2_CAP_ENCRYPTION)) {
+               capabilities |= SMB2_CAP_ENCRYPTION;
+       }
+
        /*
-        * Unless we implement SMB2_CAP_LARGE_MTU,
         * 0x10000 (65536) is the maximum allowed message size
+        * for SMB 2.0
         */
        max_limit = 0x10000;
 
-       max_trans = MIN(max_limit, max_trans);
-       max_read  = MIN(max_limit, max_read);
-       max_write = MIN(max_limit, max_write);
+       if (protocol >= PROTOCOL_SMB2_10) {
+               int p = 0;
+
+               if (tsocket_address_is_inet(req->sconn->local_address, "ip")) {
+                       p = tsocket_address_inet_port(req->sconn->local_address);
+               }
+
+               /* largeMTU is not supported over NBT (tcp port 139) */
+               if (p != NBT_SMB_PORT) {
+                       capabilities |= SMB2_CAP_LARGE_MTU;
+                       req->sconn->smb2.supports_multicredit = true;
+
+                       /* SMB >= 2.1 has 1 MB of allowed size */
+                       max_limit = 0x100000; /* 1MB */
+               }
+       }
+
+       /*
+        * the defaults are 1MB, but we'll limit this to max_limit based on
+        * the dialect (64kb for SMB2.0, 1MB for SMB2.1 with LargeMTU)
+        *
+        * user configured values exceeding the limits will be overwritten,
+        * only smaller values will be accepted
+        */
+
+       max_trans = MIN(max_limit, lp_smb2_max_trans());
+       max_read = MIN(max_limit, lp_smb2_max_read());
+       max_write = MIN(max_limit, lp_smb2_max_write());
 
        security_offset = SMB2_HDR_BODY + 0x40;
 
@@ -224,7 +328,13 @@ NTSTATUS smbd_smb2_request_process_negprot(struct smbd_smb2_request *req)
        security_buffer = data_blob_const(NULL, 0);
 #endif
 
-       outbody = data_blob_talloc(req->out.vector, NULL, 0x40);
+       out_guid_blob = data_blob_const(negprot_spnego_blob.data, 16);
+       status = GUID_from_ndr_blob(&out_guid_blob, &out_guid);
+       if (!NT_STATUS_IS_OK(status)) {
+               return smbd_smb2_request_error(req, status);
+       }
+
+       outbody = smbd_smb2_generate_outbody(req, 0x40);
        if (outbody.data == NULL) {
                return smbd_smb2_request_error(req, NT_STATUS_NO_MEMORY);
        }
@@ -235,13 +345,13 @@ NTSTATUS smbd_smb2_request_process_negprot(struct smbd_smb2_request *req)
        SSVAL(outbody.data, 0x04, dialect);     /* dialect revision */
        SSVAL(outbody.data, 0x06, 0);           /* reserved */
        memcpy(outbody.data + 0x08,
-              negprot_spnego_blob.data, 16);   /* server guid */
+              out_guid_blob.data, 16); /* server guid */
        SIVAL(outbody.data, 0x18,
              capabilities);                    /* capabilities */
        SIVAL(outbody.data, 0x1C, max_trans);   /* max transact size */
-       SIVAL(outbody.data, 0x20, max_trans);   /* max read size */
-       SIVAL(outbody.data, 0x24, max_trans);   /* max write size */
-       SBVAL(outbody.data, 0x28, 0);           /* system time */
+       SIVAL(outbody.data, 0x20, max_read);    /* max read size */
+       SIVAL(outbody.data, 0x24, max_write);   /* max write size */
+       SBVAL(outbody.data, 0x28, now);         /* system time */
        SBVAL(outbody.data, 0x30, 0);           /* server start time */
        SSVAL(outbody.data, 0x38,
              security_offset);                 /* security buffer offset */
@@ -253,5 +363,40 @@ NTSTATUS smbd_smb2_request_process_negprot(struct smbd_smb2_request *req)
 
        req->sconn->using_smb2 = true;
 
+       if (dialect != SMB2_DIALECT_REVISION_2FF) {
+               struct smbXsrv_connection *conn = req->sconn->conn;
+
+               status = smbXsrv_connection_init_tables(conn, protocol);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return smbd_smb2_request_error(req, status);
+               }
+
+               conn->smb2.client.capabilities = in_capabilities;
+               conn->smb2.client.security_mode = in_security_mode;
+               conn->smb2.client.guid = in_guid;
+               conn->smb2.client.num_dialects = dialect_count;
+               conn->smb2.client.dialects = talloc_array(conn,
+                                                         uint16_t,
+                                                         dialect_count);
+               if (conn->smb2.client.dialects == NULL) {
+                       return smbd_smb2_request_error(req, NT_STATUS_NO_MEMORY);
+               }
+               for (c=0; c < dialect_count; c++) {
+                       conn->smb2.client.dialects[c] = SVAL(indyn, c*2);
+               }
+
+               conn->smb2.server.capabilities = capabilities;
+               conn->smb2.server.security_mode = security_mode;
+               conn->smb2.server.guid = out_guid;
+               conn->smb2.server.dialect = dialect;
+               conn->smb2.server.max_trans = max_trans;
+               conn->smb2.server.max_read  = max_read;
+               conn->smb2.server.max_write = max_write;
+
+               req->sconn->smb2.max_trans = max_trans;
+               req->sconn->smb2.max_read  = max_read;
+               req->sconn->smb2.max_write = max_write;
+       }
+
        return smbd_smb2_request_done(req, outbody, &outdyn);
 }