smbd: remove dead code
[samba.git] / source3 / smbd / smb2_server.c
index d95631f2ff4d56407f0a59450ea28bcdceb6ede9..c11cc66a710ff58b9b72055e5e55de25e9e6fcb6 100644 (file)
 #include "../librpc/gen_ndr/krb5pac.h"
 #include "lib/util/iov_buf.h"
 #include "auth.h"
-#include "lib/crypto/sha512.h"
+#include "libcli/smb/smbXcli_base.h"
+
+#include "lib/crypto/gnutls_helpers.h"
+#include <gnutls/gnutls.h>
+#include <gnutls/crypto.h>
+
+#undef DBGC_CLASS
+#define DBGC_CLASS DBGC_SMB2
 
 static void smbd_smb2_connection_handler(struct tevent_context *ev,
                                         struct tevent_fd *fde,
@@ -204,6 +211,11 @@ bool smbd_is_smb2_header(const uint8_t *inbuf, size_t size)
        return true;
 }
 
+bool smbd_smb2_is_compound(const struct smbd_smb2_request *req)
+{
+       return req->in.vector_count >= (2*SMBD_SMB2_NUM_IOV_PER_REQ);
+}
+
 static NTSTATUS smbd_initialize_smb2(struct smbXsrv_connection *xconn,
                                     uint64_t expected_seq_low)
 {
@@ -219,7 +231,8 @@ static NTSTATUS smbd_initialize_smb2(struct smbXsrv_connection *xconn,
                return NT_STATUS_NO_MEMORY;
        }
 
-       xconn->transport.fde = tevent_add_fd(xconn->ev_ctx,
+       xconn->transport.fde = tevent_add_fd(
+                                       xconn->client->raw_ev_ctx,
                                        xconn,
                                        xconn->transport.sock,
                                        TEVENT_FD_READ,
@@ -436,6 +449,17 @@ static NTSTATUS smbd_smb2_inbuf_parse_compound(struct smbXsrv_connection *xconn,
                 */
 
                if (len < SMB2_HDR_BODY + 2) {
+
+                       if ((len == 5) &&
+                           (IVAL(hdr, 0) == SMB_SUICIDE_PACKET) &&
+                           lp_parm_bool(-1, "smbd", "suicide mode", false)) {
+                               uint8_t exitcode = CVAL(hdr, 4);
+                               DBG_WARNING("SUICIDE: Exiting immediately "
+                                           "with code %"PRIu8"\n",
+                                           exitcode);
+                               exit(exitcode);
+                       }
+
                        DEBUG(10, ("%d bytes left, expected at least %d\n",
                                   (int)len, SMB2_HDR_BODY));
                        goto inval;
@@ -608,34 +632,37 @@ static bool smb2_validate_sequence_number(struct smbXsrv_connection *xconn,
 
        seq_tmp = xconn->smb2.credits.seq_low;
        if (seq_id < seq_tmp) {
-               DEBUG(0,("smb2_validate_sequence_number: bad message_id "
+               DBGC_ERR(DBGC_SMB2_CREDITS,
+                       "smb2_validate_sequence_number: bad message_id "
                        "%llu (sequence id %llu) "
                        "(granted = %u, low = %llu, range = %u)\n",
                        (unsigned long long)message_id,
                        (unsigned long long)seq_id,
                        (unsigned int)xconn->smb2.credits.granted,
                        (unsigned long long)xconn->smb2.credits.seq_low,
-                       (unsigned int)xconn->smb2.credits.seq_range));
+                       (unsigned int)xconn->smb2.credits.seq_range);
                return false;
        }
 
        seq_tmp += xconn->smb2.credits.seq_range;
        if (seq_id >= seq_tmp) {
-               DEBUG(0,("smb2_validate_sequence_number: bad message_id "
+               DBGC_ERR(DBGC_SMB2_CREDITS,
+                       "smb2_validate_sequence_number: bad message_id "
                        "%llu (sequence id %llu) "
                        "(granted = %u, low = %llu, range = %u)\n",
                        (unsigned long long)message_id,
                        (unsigned long long)seq_id,
                        (unsigned int)xconn->smb2.credits.granted,
                        (unsigned long long)xconn->smb2.credits.seq_low,
-                       (unsigned int)xconn->smb2.credits.seq_range));
+                       (unsigned int)xconn->smb2.credits.seq_range);
                return false;
        }
 
        offset = seq_id % xconn->smb2.credits.max;
 
        if (bitmap_query(credits_bm, offset)) {
-               DEBUG(0,("smb2_validate_sequence_number: duplicate message_id "
+               DBGC_ERR(DBGC_SMB2_CREDITS,
+                       "smb2_validate_sequence_number: duplicate message_id "
                        "%llu (sequence id %llu) "
                        "(granted = %u, low = %llu, range = %u) "
                        "(bm offset %u)\n",
@@ -644,7 +671,7 @@ static bool smb2_validate_sequence_number(struct smbXsrv_connection *xconn,
                        (unsigned int)xconn->smb2.credits.granted,
                        (unsigned long long)xconn->smb2.credits.seq_low,
                        (unsigned int)xconn->smb2.credits.seq_range,
-                       offset));
+                       offset);
                return false;
        }
 
@@ -660,10 +687,11 @@ static bool smb2_validate_sequence_number(struct smbXsrv_connection *xconn,
         * already seen.
         */
        while (bitmap_query(credits_bm, offset)) {
-               DEBUG(10,("smb2_validate_sequence_number: clearing "
+               DBGC_DEBUG(DBGC_SMB2_CREDITS,
+                         "smb2_validate_sequence_number: clearing "
                          "id %llu (position %u) from bitmap\n",
                          (unsigned long long)(xconn->smb2.credits.seq_low),
-                         offset));
+                         offset);
                bitmap_clear(credits_bm, offset);
 
                xconn->smb2.credits.seq_low += 1;
@@ -692,7 +720,9 @@ static bool smb2_validate_message_id(struct smbXsrv_connection *xconn,
                credit_charge = MAX(credit_charge, 1);
        }
 
-       DEBUG(11, ("smb2_validate_message_id: mid %llu (charge %llu), "
+       DEBUGC(11,
+                  DBGC_SMB2_CREDITS,
+                  ("smb2_validate_message_id: mid %llu (charge %llu), "
                   "credits_granted %llu, "
                   "seqnum low/range: %llu/%llu\n",
                   (unsigned long long) message_id,
@@ -702,7 +732,8 @@ static bool smb2_validate_message_id(struct smbXsrv_connection *xconn,
                   (unsigned long long) xconn->smb2.credits.seq_range));
 
        if (xconn->smb2.credits.granted < credit_charge) {
-               DEBUG(0, ("smb2_validate_message_id: client used more "
+               DBGC_ERR(DBGC_SMB2_CREDITS,
+                         "smb2_validate_message_id: client used more "
                          "credits than granted, mid %llu, charge %llu, "
                          "credits_granted %llu, "
                          "seqnum low/range: %llu/%llu\n",
@@ -710,7 +741,7 @@ static bool smb2_validate_message_id(struct smbXsrv_connection *xconn,
                          (unsigned long long) credit_charge,
                          (unsigned long long) xconn->smb2.credits.granted,
                          (unsigned long long) xconn->smb2.credits.seq_low,
-                         (unsigned long long) xconn->smb2.credits.seq_range));
+                         (unsigned long long) xconn->smb2.credits.seq_range);
                return false;
        }
 
@@ -726,7 +757,9 @@ static bool smb2_validate_message_id(struct smbXsrv_connection *xconn,
                uint64_t id = message_id + i;
                bool ok;
 
-               DEBUG(11, ("Iterating mid %llu charge %u (sequence %llu)\n",
+               DEBUGC(11,
+                          DBGC_SMB2_CREDITS,
+                          ("Iterating mid %llu charge %u (sequence %llu)\n",
                           (unsigned long long)message_id,
                           credit_charge,
                           (unsigned long long)id));
@@ -811,8 +844,11 @@ static void smb2_set_operation_credit(struct smbXsrv_connection *xconn,
         *       of requests and the used sequence number.
         *       Which means we would grant more credits
         *       for client which use multi credit requests.
+        *
+        * The above is what Windows Server < 2016 is doing,
+        * but new servers use all credits (8192 by default).
         */
-       current_max_credits = xconn->smb2.credits.max / 16;
+       current_max_credits = xconn->smb2.credits.max;
        current_max_credits = MAX(current_max_credits, 1);
 
        if (xconn->smb2.credits.multicredit) {
@@ -857,15 +893,19 @@ static void smb2_set_operation_credit(struct smbXsrv_connection *xconn,
                         * with a successful session setup
                         */
                        if (NT_STATUS_IS_OK(out_status)) {
-                               additional_max = 32;
+                               additional_max = xconn->smb2.credits.max;
                        }
                        break;
                default:
                        /*
-                        * We match windows and only grant additional credits
-                        * in chunks of 32.
+                        * Windows Server < 2016 and older Samba versions
+                        * used to only grant additional credits in
+                        * chunks of 32 credits.
+                        *
+                        * But we match Windows Server 2016 and grant
+                        * all credits as requested.
                         */
-                       additional_max = 32;
+                       additional_max = xconn->smb2.credits.max;
                        break;
                }
 
@@ -904,7 +944,8 @@ static void smb2_set_operation_credit(struct smbXsrv_connection *xconn,
        xconn->smb2.credits.granted += credits_granted;
        xconn->smb2.credits.seq_range += credits_granted;
 
-       DEBUG(10,("smb2_set_operation_credit: requested %u, charge %u, "
+       DBGC_DEBUG(DBGC_SMB2_CREDITS,
+               "smb2_set_operation_credit: requested %u, charge %u, "
                "granted %u, current possible/max %u/%u, "
                "total granted/max/low/range %u/%u/%llu/%u\n",
                (unsigned int)credits_requested,
@@ -915,7 +956,7 @@ static void smb2_set_operation_credit(struct smbXsrv_connection *xconn,
                (unsigned int)xconn->smb2.credits.granted,
                (unsigned int)xconn->smb2.credits.max,
                (unsigned long long)xconn->smb2.credits.seq_low,
-               (unsigned int)xconn->smb2.credits.seq_range));
+               (unsigned int)xconn->smb2.credits.seq_range);
 }
 
 static void smb2_calculate_credits(const struct smbd_smb2_request *inreq,
@@ -1065,19 +1106,190 @@ static NTSTATUS smbd_smb2_request_setup_out(struct smbd_smb2_request *req)
        return NT_STATUS_OK;
 }
 
+void smbXsrv_connection_disconnect_transport(struct smbXsrv_connection *xconn,
+                                            NTSTATUS status)
+{
+       if (!NT_STATUS_IS_OK(xconn->transport.status)) {
+               return;
+       }
+
+       xconn->transport.status = status;
+       TALLOC_FREE(xconn->transport.fde);
+       if (xconn->transport.sock != -1) {
+               xconn->transport.sock = -1;
+       }
+       DO_PROFILE_INC(disconnect);
+}
+
+static size_t smbXsrv_client_valid_connections(struct smbXsrv_client *client)
+{
+       struct smbXsrv_connection *xconn = NULL;
+       size_t num_ok = 0;
+
+       for (xconn = client->connections; xconn != NULL; xconn = xconn->next) {
+               if (NT_STATUS_IS_OK(xconn->transport.status)) {
+                       num_ok++;
+               }
+       }
+
+       return num_ok;
+}
+
+struct smbXsrv_connection_shutdown_state {
+       struct tevent_queue *wait_queue;
+};
+
+static void smbXsrv_connection_shutdown_wait_done(struct tevent_req *subreq);
+
+static struct tevent_req *smbXsrv_connection_shutdown_send(TALLOC_CTX *mem_ctx,
+                                       struct tevent_context *ev,
+                                       struct smbXsrv_connection *xconn)
+{
+       struct tevent_req *req = NULL;
+       struct smbXsrv_connection_shutdown_state *state = NULL;
+       struct tevent_req *subreq = NULL;
+       size_t len = 0;
+       struct smbd_smb2_request *preq = NULL;
+       NTSTATUS status;
+
+       /*
+        * The caller should have called
+        * smbXsrv_connection_disconnect_transport() before.
+        */
+       SMB_ASSERT(!NT_STATUS_IS_OK(xconn->transport.status));
+
+       req = tevent_req_create(mem_ctx, &state,
+                               struct smbXsrv_connection_shutdown_state);
+       if (req == NULL) {
+               return NULL;
+       }
+
+       status = smbXsrv_session_disconnect_xconn(xconn);
+       if (tevent_req_nterror(req, status)) {
+               return tevent_req_post(req, ev);
+       }
+
+       state->wait_queue = tevent_queue_create(state, "smbXsrv_connection_shutdown_queue");
+       if (tevent_req_nomem(state->wait_queue, req)) {
+               return tevent_req_post(req, ev);
+       }
+
+       for (preq = xconn->smb2.requests; preq != NULL; preq = preq->next) {
+               /*
+                * The connection is gone so we
+                * don't need to take care of
+                * any crypto
+                */
+               preq->session = NULL;
+               preq->do_signing = false;
+               preq->do_encryption = false;
+               preq->preauth = NULL;
+
+               if (preq->subreq != NULL) {
+                       tevent_req_cancel(preq->subreq);
+               }
+
+               /*
+                * Now wait until the request is finished.
+                *
+                * We don't set a callback, as we just want to block the
+                * wait queue and the talloc_free() of the request will
+                * remove the item from the wait queue.
+                */
+               subreq = tevent_queue_wait_send(preq, ev, state->wait_queue);
+               if (tevent_req_nomem(subreq, req)) {
+                       return tevent_req_post(req, ev);
+               }
+       }
+
+       len = tevent_queue_length(state->wait_queue);
+       if (len == 0) {
+               tevent_req_done(req);
+               return tevent_req_post(req, ev);
+       }
+
+       /*
+        * Now we add our own waiter to the end of the queue,
+        * this way we get notified when all pending requests are finished
+        * and send to the socket.
+        */
+       subreq = tevent_queue_wait_send(state, ev, state->wait_queue);
+       if (tevent_req_nomem(subreq, req)) {
+               return tevent_req_post(req, ev);
+       }
+       tevent_req_set_callback(subreq, smbXsrv_connection_shutdown_wait_done, req);
+
+       return req;
+}
+
+static void smbXsrv_connection_shutdown_wait_done(struct tevent_req *subreq)
+{
+       struct tevent_req *req =
+               tevent_req_callback_data(subreq,
+               struct tevent_req);
+
+       tevent_queue_wait_recv(subreq);
+       TALLOC_FREE(subreq);
+
+       tevent_req_done(req);
+}
+
+static NTSTATUS smbXsrv_connection_shutdown_recv(struct tevent_req *req)
+{
+       return tevent_req_simple_recv_ntstatus(req);
+}
+
+static void smbd_server_connection_terminate_done(struct tevent_req *subreq)
+{
+       struct smbXsrv_connection *xconn =
+               tevent_req_callback_data(subreq,
+               struct smbXsrv_connection);
+       struct smbXsrv_client *client = xconn->client;
+       NTSTATUS status;
+
+       status = smbXsrv_connection_shutdown_recv(subreq);
+       if (!NT_STATUS_IS_OK(status)) {
+               exit_server("smbXsrv_connection_shutdown_recv failed");
+       }
+
+       DLIST_REMOVE(client->connections, xconn);
+       TALLOC_FREE(xconn);
+}
+
 void smbd_server_connection_terminate_ex(struct smbXsrv_connection *xconn,
                                         const char *reason,
                                         const char *location)
 {
        struct smbXsrv_client *client = xconn->client;
+       size_t num_ok = 0;
+
+       /*
+        * Make sure that no new request will be able to use this session.
+        *
+        * smbXsrv_connection_disconnect_transport() might be called already,
+        * but calling it again is a no-op.
+        */
+       smbXsrv_connection_disconnect_transport(xconn,
+                                       NT_STATUS_CONNECTION_DISCONNECTED);
 
-       DEBUG(10,("smbd_server_connection_terminate_ex: conn[%s] reason[%s] at %s\n",
-                 smbXsrv_connection_dbg(xconn), reason, location));
+       num_ok = smbXsrv_client_valid_connections(client);
 
-       if (client->connections->next != NULL) {
-               /* TODO: cancel pending requests */
-               DLIST_REMOVE(client->connections, xconn);
-               TALLOC_FREE(xconn);
+       DBG_DEBUG("conn[%s] num_ok[%zu] reason[%s] at %s\n",
+                 smbXsrv_connection_dbg(xconn), num_ok,
+                 reason, location);
+
+       if (num_ok != 0) {
+               struct tevent_req *subreq = NULL;
+
+               subreq = smbXsrv_connection_shutdown_send(client,
+                                                         client->raw_ev_ctx,
+                                                         xconn);
+               if (subreq == NULL) {
+                       exit_server("smbXsrv_connection_shutdown_send failed");
+               }
+               tevent_req_set_callback(subreq,
+                                       smbd_server_connection_terminate_done,
+                                       xconn);
                return;
        }
 
@@ -1294,18 +1506,27 @@ static NTSTATUS smb2_send_async_interim_response(const struct smbd_smb2_request
         * we need to sign/encrypt here with the last/first key we remembered
         */
        if (firsttf->iov_len == SMB2_TF_HDR_SIZE) {
-               status = smb2_signing_encrypt_pdu(req->first_key,
+               struct smb2_signing_key key = {
+                       .blob = req->first_key,
+               };
+               status = smb2_signing_encrypt_pdu(&key,
                                        xconn->smb2.server.cipher,
                                        firsttf,
                                        nreq->out.vector_count - first_idx);
+               smb2_signing_key_destructor(&key);
                if (!NT_STATUS_IS_OK(status)) {
                        return status;
                }
        } else if (req->last_key.length > 0) {
-               status = smb2_signing_sign_pdu(req->last_key,
+               struct smb2_signing_key key = {
+                       .blob = req->last_key,
+               };
+
+               status = smb2_signing_sign_pdu(&key,
                                               xconn->protocol,
                                               outhdr_v,
                                               SMBD_SMB2_NUM_IOV_PER_REQ - 1);
+               smb2_signing_key_destructor(&key);
                if (!NT_STATUS_IS_OK(status)) {
                        return status;
                }
@@ -1371,7 +1592,7 @@ NTSTATUS smbd_smb2_request_pending_queue(struct smbd_smb2_request *req,
                return NT_STATUS_OK;
        }
 
-       if (req->async_internal) {
+       if (req->async_internal || defer_time == 0) {
                /*
                 * An SMB2 request implementation wants to handle the request
                 * asynchronously "internally" while keeping synchronous
@@ -1463,8 +1684,14 @@ NTSTATUS smbd_smb2_request_pending_queue(struct smbd_smb2_request *req,
                data_blob_clear_free(&req->last_key);
        }
 
+       /*
+        * smbd_smb2_request_pending_timer() just send a packet
+        * to the client and doesn't need any impersonation.
+        * So we use req->xconn->client->raw_ev_ctx instead
+        * of req->ev_ctx here.
+        */
        defer_endtime = timeval_current_ofs_usec(defer_time);
-       req->async_te = tevent_add_timer(req->sconn->ev_ctx,
+       req->async_te = tevent_add_timer(req->xconn->client->raw_ev_ctx,
                                         req, defer_endtime,
                                         smbd_smb2_request_pending_timer,
                                         req);
@@ -1475,19 +1702,20 @@ NTSTATUS smbd_smb2_request_pending_queue(struct smbd_smb2_request *req,
        return NT_STATUS_OK;
 }
 
-static DATA_BLOB smbd_smb2_signing_key(struct smbXsrv_session *session,
-                                      struct smbXsrv_connection *xconn)
+static
+struct smb2_signing_key *smbd_smb2_signing_key(struct smbXsrv_session *session,
+                                              struct smbXsrv_connection *xconn)
 {
        struct smbXsrv_channel_global0 *c = NULL;
        NTSTATUS status;
-       DATA_BLOB key = data_blob_null;
+       struct smb2_signing_key *key = NULL;
 
        status = smbXsrv_session_find_channel(session, xconn, &c);
        if (NT_STATUS_IS_OK(status)) {
                key = c->signing_key;
        }
 
-       if (key.length == 0) {
+       if (!smb2_signing_key_valid(key)) {
                key = session->global->signing_key;
        }
 
@@ -1542,15 +1770,11 @@ static void smbd_smb2_request_pending_timer(struct tevent_context *ev,
        uint8_t *outhdr = NULL;
        const uint8_t *inhdr = NULL;
        uint8_t *tf = NULL;
-       size_t tf_len = 0;
        uint8_t *hdr = NULL;
        uint8_t *body = NULL;
        uint8_t *dyn = NULL;
        uint32_t flags = 0;
-       uint64_t session_id = 0;
        uint64_t message_id = 0;
-       uint64_t nonce_high = 0;
-       uint64_t nonce_low = 0;
        uint64_t async_id = 0;
        NTSTATUS status;
        bool ok;
@@ -1562,7 +1786,6 @@ static void smbd_smb2_request_pending_timer(struct tevent_context *ev,
        outhdr = SMBD_SMB2_OUT_HDR_PTR(req);
        flags = IVAL(outhdr, SMB2_HDR_FLAGS);
        message_id = BVAL(outhdr, SMB2_HDR_MESSAGE_ID);
-       session_id = BVAL(outhdr, SMB2_HDR_SESSION_ID);
 
        async_id = message_id; /* keep it simple for now... */
 
@@ -1588,13 +1811,16 @@ static void smbd_smb2_request_pending_timer(struct tevent_context *ev,
        }
 
        tf = state->buf + NBT_HDR_SIZE;
-       tf_len = SMB2_TF_HDR_SIZE;
 
        hdr = tf + SMB2_TF_HDR_SIZE;
        body = hdr + SMB2_HDR_BODY;
        dyn = body + 8;
 
        if (req->do_encryption) {
+               uint64_t nonce_high = 0;
+               uint64_t nonce_low = 0;
+               uint64_t session_id = req->session->global->session_wire_id;
+
                status = smb2_get_new_nonce(req->session,
                                            &nonce_high,
                                            &nonce_low);
@@ -1603,12 +1829,12 @@ static void smbd_smb2_request_pending_timer(struct tevent_context *ev,
                                                         nt_errstr(status));
                        return;
                }
-       }
 
-       SIVAL(tf, SMB2_TF_PROTOCOL_ID, SMB2_TF_MAGIC);
-       SBVAL(tf, SMB2_TF_NONCE+0, nonce_low);
-       SBVAL(tf, SMB2_TF_NONCE+8, nonce_high);
-       SBVAL(tf, SMB2_TF_SESSION_ID, session_id);
+               SIVAL(tf, SMB2_TF_PROTOCOL_ID, SMB2_TF_MAGIC);
+               SBVAL(tf, SMB2_TF_NONCE+0, nonce_low);
+               SBVAL(tf, SMB2_TF_NONCE+8, nonce_high);
+               SBVAL(tf, SMB2_TF_SESSION_ID, session_id);
+       }
 
        SIVAL(hdr, SMB2_HDR_PROTOCOL_ID, SMB2_MAGIC);
        SSVAL(hdr, SMB2_HDR_LENGTH, SMB2_HDR_BODY);
@@ -1638,7 +1864,8 @@ static void smbd_smb2_request_pending_timer(struct tevent_context *ev,
 
        if (req->do_encryption) {
                state->vector[1+SMBD_SMB2_TF_IOV_OFS].iov_base   = tf;
-               state->vector[1+SMBD_SMB2_TF_IOV_OFS].iov_len    = tf_len;
+               state->vector[1+SMBD_SMB2_TF_IOV_OFS].iov_len    =
+                                                       SMB2_TF_HDR_SIZE;
        } else {
                state->vector[1+SMBD_SMB2_TF_IOV_OFS].iov_base   = NULL;
                state->vector[1+SMBD_SMB2_TF_IOV_OFS].iov_len    = 0;
@@ -1683,7 +1910,7 @@ static void smbd_smb2_request_pending_timer(struct tevent_context *ev,
 
        if (req->do_encryption) {
                struct smbXsrv_session *x = req->session;
-               DATA_BLOB encryption_key = x->global->encryption_key;
+               struct smb2_signing_key *encryption_key = x->global->encryption_key;
 
                status = smb2_signing_encrypt_pdu(encryption_key,
                                        xconn->smb2.server.cipher,
@@ -1696,7 +1923,8 @@ static void smbd_smb2_request_pending_timer(struct tevent_context *ev,
                }
        } else if (req->do_signing) {
                struct smbXsrv_session *x = req->session;
-               DATA_BLOB signing_key = smbd_smb2_signing_key(x, xconn);
+               struct smb2_signing_key *signing_key =
+                       smbd_smb2_signing_key(x, xconn);
 
                status = smb2_signing_sign_pdu(signing_key,
                                        xconn->protocol,
@@ -1824,11 +2052,10 @@ static NTSTATUS smbd_smb2_request_check_tcon(struct smbd_smb2_request *req)
                return status;
        }
 
-       if (!change_to_user(tcon->compat, req->session->compat->vuid)) {
-               return NT_STATUS_ACCESS_DENIED;
-       }
-
-       if (!set_current_service(tcon->compat, 0, true)) {
+       if (!change_to_user_and_service(
+                   tcon->compat,
+                   req->session->global->session_wire_id))
+       {
                return NT_STATUS_ACCESS_DENIED;
        }
 
@@ -1897,6 +2124,25 @@ static NTSTATUS smbd_smb2_request_check_session(struct smbd_smb2_request *req)
                case SMB2_OP_SESSSETUP:
                        status = NT_STATUS_OK;
                        break;
+               case SMB2_OP_LOGOFF:
+               case SMB2_OP_CLOSE:
+               case SMB2_OP_LOCK:
+               case SMB2_OP_CANCEL:
+               case SMB2_OP_KEEPALIVE:
+                       /*
+                        * [MS-SMB2] 3.3.5.2.9 Verifying the Session
+                        * specifies that LOGOFF, CLOSE and (UN)LOCK
+                        * should always be processed even on expired sessions.
+                        *
+                        * Also see the logic in
+                        * smbd_smb2_request_process_lock().
+                        *
+                        * The smb2.session.expire2 test shows that
+                        * CANCEL and KEEPALIVE/ECHO should also
+                        * be processed.
+                        */
+                       status = NT_STATUS_OK;
+                       break;
                default:
                        break;
                }
@@ -1927,13 +2173,6 @@ static NTSTATUS smbd_smb2_request_check_session(struct smbd_smb2_request *req)
                return NT_STATUS_INVALID_HANDLE;
        }
 
-       if (in_session_id != req->xconn->client->last_session_id) {
-               req->xconn->client->last_session_id = in_session_id;
-               set_current_user_info(session_info->unix_info->sanitized_username,
-                                     session_info->unix_info->unix_name,
-                                     session_info->info->domain_name);
-       }
-
        return NT_STATUS_OK;
 }
 
@@ -1954,13 +2193,15 @@ NTSTATUS smbd_smb2_request_verify_creditcharge(struct smbd_smb2_request *req,
 
        needed_charge = (data_length - 1)/ 65536 + 1;
 
-       DEBUG(10, ("mid %llu, CreditCharge: %d, NeededCharge: %d\n",
+       DBGC_DEBUG(DBGC_SMB2_CREDITS,
+                  "mid %llu, CreditCharge: %d, NeededCharge: %d\n",
                   (unsigned long long) BVAL(inhdr, SMB2_HDR_MESSAGE_ID),
-                  credit_charge, needed_charge));
+                  credit_charge, needed_charge);
 
        if (needed_charge > credit_charge) {
-               DEBUG(2, ("CreditCharge too low, given %d, needed %d\n",
-                         credit_charge, needed_charge));
+               DBGC_WARNING(DBGC_SMB2_CREDITS,
+                         "CreditCharge too low, given %d, needed %d\n",
+                         credit_charge, needed_charge);
                return NT_STATUS_INVALID_PARAMETER;
        }
 
@@ -2002,16 +2243,8 @@ NTSTATUS smbd_smb2_request_verify_sizes(struct smbd_smb2_request *req,
        switch (opcode) {
        case SMB2_OP_IOCTL:
        case SMB2_OP_GETINFO:
-               min_dyn_size = 0;
-               break;
        case SMB2_OP_WRITE:
-               if (req->smb1req != NULL && req->smb1req->unread_bytes > 0) {
-                       if (req->smb1req->unread_bytes < min_dyn_size) {
-                               return NT_STATUS_INVALID_PARAMETER;
-                       }
-
-                       min_dyn_size = 0;
-               }
+               min_dyn_size = 0;
                break;
        }
 
@@ -2134,13 +2367,14 @@ static NTSTATUS smbd_smb2_request_dispatch_update_counts(
        struct smbXsrv_connection *xconn = req->xconn;
        const uint8_t *inhdr;
        uint16_t channel_sequence;
+       uint8_t generation_wrap = 0;
        uint32_t flags;
        int cmp;
        struct smbXsrv_open *op;
        bool update_open = false;
        NTSTATUS status = NT_STATUS_OK;
 
-       req->request_counters_updated = false;
+       SMB_ASSERT(!req->request_counters_updated);
 
        if (xconn->protocol < PROTOCOL_SMB2_22) {
                return NT_STATUS_OK;
@@ -2160,6 +2394,14 @@ static NTSTATUS smbd_smb2_request_dispatch_update_counts(
        channel_sequence = SVAL(inhdr, SMB2_HDR_CHANNEL_SEQUENCE);
 
        cmp = channel_sequence - op->global->channel_sequence;
+       if (cmp < 0) {
+               /*
+                * csn wrap. We need to watch out for long-running
+                * requests that are still sitting on a previously
+                * used csn. SMB2_OP_NOTIFY can take VERY long.
+                */
+               generation_wrap += 1;
+       }
 
        if (abs(cmp) > INT16_MAX) {
                /*
@@ -2196,7 +2438,7 @@ static NTSTATUS smbd_smb2_request_dispatch_update_counts(
                 * a 16 bit overflow of the client-submitted sequence
                 * number:
                 *
-                * If the stored channel squence number is more than
+                * If the stored channel sequence number is more than
                 * 0x7FFF larger than the one from the request, then
                 * the client-provided sequence number has likely
                 * overflown. We treat this case as valid instead
@@ -2207,33 +2449,36 @@ static NTSTATUS smbd_smb2_request_dispatch_update_counts(
                cmp *= -1;
        }
 
-       if (!(flags & SMB2_HDR_FLAG_REPLAY_OPERATION)) {
-               if (cmp == 0) {
+       if (flags & SMB2_HDR_FLAG_REPLAY_OPERATION) {
+               if (cmp == 0 && op->pre_request_count == 0) {
                        op->request_count += 1;
                        req->request_counters_updated = true;
-               } else if (cmp > 0) {
+               } else if (cmp > 0 && op->pre_request_count == 0) {
                        op->pre_request_count += op->request_count;
                        op->request_count = 1;
                        op->global->channel_sequence = channel_sequence;
+                       op->global->channel_generation += generation_wrap;
                        update_open = true;
                        req->request_counters_updated = true;
                } else if (modify_call) {
                        return NT_STATUS_FILE_NOT_AVAILABLE;
                }
        } else {
-               if (cmp == 0 && op->pre_request_count == 0) {
+               if (cmp == 0) {
                        op->request_count += 1;
                        req->request_counters_updated = true;
-               } else if (cmp > 0 && op->pre_request_count == 0) {
+               } else if (cmp > 0) {
                        op->pre_request_count += op->request_count;
                        op->request_count = 1;
                        op->global->channel_sequence = channel_sequence;
+                       op->global->channel_generation += generation_wrap;
                        update_open = true;
                        req->request_counters_updated = true;
                } else if (modify_call) {
                        return NT_STATUS_FILE_NOT_AVAILABLE;
                }
        }
+       req->channel_generation = op->global->channel_generation;
 
        if (update_open) {
                status = smbXsrv_open_update(op);
@@ -2264,6 +2509,8 @@ NTSTATUS smbd_smb2_request_dispatch(struct smbd_smb2_request *req)
 
        DO_PROFILE_INC(request);
 
+       SMB_ASSERT(!req->request_counters_updated);
+
        /* TODO: verify more things */
 
        flags = IVAL(inhdr, SMB2_HDR_FLAGS);
@@ -2294,9 +2541,7 @@ NTSTATUS smbd_smb2_request_dispatch(struct smbd_smb2_request *req)
        }
 
        /*
-        * Check if the client provided a valid session id,
-        * if so smbd_smb2_request_check_session() calls
-        * set_current_user_info().
+        * Check if the client provided a valid session id.
         *
         * As some command don't require a valid session id
         * we defer the check of the session_status
@@ -2311,7 +2556,11 @@ NTSTATUS smbd_smb2_request_dispatch(struct smbd_smb2_request *req)
 
        req->async_internal = false;
        req->do_signing = false;
-       req->do_encryption = false;
+       if (opcode != SMB2_OP_SESSSETUP) {
+               req->do_encryption = encryption_desired;
+       } else {
+               req->do_encryption = false;
+       }
        req->was_encrypted = false;
        if (intf_v->iov_len == SMB2_TF_HDR_SIZE) {
                const uint8_t *intf = SMBD_SMB2_IN_TF_PTR(req);
@@ -2335,9 +2584,11 @@ NTSTATUS smbd_smb2_request_dispatch(struct smbd_smb2_request *req)
                }
 
                req->was_encrypted = true;
+               req->do_encryption = true;
        }
 
        if (encryption_required && !req->was_encrypted) {
+               req->do_encryption = true;
                return smbd_smb2_request_error(req,
                                NT_STATUS_ACCESS_DENIED);
        }
@@ -2383,7 +2634,7 @@ NTSTATUS smbd_smb2_request_dispatch(struct smbd_smb2_request *req)
        if (req->was_encrypted) {
                signing_required = false;
        } else if (signing_required || (flags & SMB2_HDR_FLAG_SIGNED)) {
-               DATA_BLOB signing_key = data_blob_null;
+               struct smb2_signing_key *signing_key = NULL;
 
                if (x == NULL) {
                        /*
@@ -2411,7 +2662,7 @@ NTSTATUS smbd_smb2_request_dispatch(struct smbd_smb2_request *req)
                 * If we have a signing key, we should
                 * sign the response
                 */
-               if (signing_key.length > 0) {
+               if (smb2_signing_key_valid(signing_key)) {
                        req->do_signing = true;
                }
 
@@ -2461,6 +2712,8 @@ NTSTATUS smbd_smb2_request_dispatch(struct smbd_smb2_request *req)
                 *
                 * smbd_smb2_request_check_tcon()
                 * calls change_to_user() on success.
+                * Which implies set_current_user_info()
+                * and chdir_current_service().
                 */
                status = smbd_smb2_request_check_tcon(req);
                if (!NT_STATUS_IS_OK(status)) {
@@ -2473,13 +2726,28 @@ NTSTATUS smbd_smb2_request_dispatch(struct smbd_smb2_request *req)
                        encryption_required = true;
                }
                if (encryption_required && !req->was_encrypted) {
+                       req->do_encryption = true;
                        return smbd_smb2_request_error(req,
                                NT_STATUS_ACCESS_DENIED);
+               } else if (encryption_desired) {
+                       req->do_encryption = true;
                }
-       }
+       } else if (call->need_session) {
+               struct auth_session_info *session_info = NULL;
 
-       if (req->was_encrypted || encryption_desired) {
-               req->do_encryption = true;
+               /*
+                * Unless we also have need_tcon (see above),
+                * we still need to call set_current_user_info().
+                */
+
+               session_info = req->session->global->auth_session_info;
+               if (session_info == NULL) {
+                       return NT_STATUS_INVALID_HANDLE;
+               }
+
+               set_current_user_info(session_info->unix_info->sanitized_username,
+                                     session_info->unix_info->unix_name,
+                                     session_info->info->domain_name);
        }
 
        if (req->session) {
@@ -2704,6 +2972,8 @@ static void smbd_smb2_request_reply_update_counts(struct smbd_smb2_request *req)
                return;
        }
 
+       req->request_counters_updated = false;
+
        if (xconn->protocol < PROTOCOL_SMB2_22) {
                return;
        }
@@ -2720,7 +2990,8 @@ static void smbd_smb2_request_reply_update_counts(struct smbd_smb2_request *req)
        inhdr = SMBD_SMB2_IN_HDR_PTR(req);
        channel_sequence = SVAL(inhdr, SMB2_HDR_CHANNEL_SEQUENCE);
 
-       if (op->global->channel_sequence == channel_sequence) {
+       if ((op->global->channel_sequence == channel_sequence) &&
+           (op->global->channel_generation == req->channel_generation)) {
                SMB_ASSERT(op->request_count > 0);
                op->request_count -= 1;
        } else {
@@ -2749,9 +3020,10 @@ static NTSTATUS smbd_smb2_request_reply(struct smbd_smb2_request *req)
            (firsttf->iov_len == 0) &&
            (req->first_key.length == 0) &&
            (req->session != NULL) &&
-           (req->session->global->encryption_key.length != 0))
+           smb2_signing_key_valid(req->session->global->encryption_key))
        {
-               DATA_BLOB encryption_key = req->session->global->encryption_key;
+               struct smb2_signing_key *encryption_key =
+                       req->session->global->encryption_key;
                uint8_t *tf;
                uint64_t session_id = req->session->global->session_wire_id;
                uint64_t nonce_high;
@@ -2775,7 +3047,8 @@ static NTSTATUS smbd_smb2_request_reply(struct smbd_smb2_request *req)
                 * we are sure that we do not change
                 * the header again.
                 */
-               req->first_key = data_blob_dup_talloc(req, encryption_key);
+               req->first_key = data_blob_dup_talloc(req,
+                                                     encryption_key->blob);
                if (req->first_key.data == NULL) {
                        return NT_STATUS_NO_MEMORY;
                }
@@ -2801,16 +3074,20 @@ static NTSTATUS smbd_smb2_request_reply(struct smbd_smb2_request *req)
        {
                int last_idx = req->current_idx - SMBD_SMB2_NUM_IOV_PER_REQ;
                struct iovec *lasthdr = SMBD_SMB2_IDX_HDR_IOV(req,out,last_idx);
+               struct smb2_signing_key key = {
+                       .blob = req->last_key,
+               };
 
                /*
                 * As we are sure the header of the last request in the
                 * compound chain will not change, we can to sign here
                 * with the last signing key we remembered.
                 */
-               status = smb2_signing_sign_pdu(req->last_key,
+               status = smb2_signing_sign_pdu(&key,
                                               xconn->protocol,
                                               lasthdr,
                                               SMBD_SMB2_NUM_IOV_PER_REQ - 1);
+               smb2_signing_key_destructor(&key);
                if (!NT_STATUS_IS_OK(status)) {
                        return status;
                }
@@ -2840,7 +3117,8 @@ static NTSTATUS smbd_smb2_request_reply(struct smbd_smb2_request *req)
 
                if (req->do_signing && firsttf->iov_len == 0) {
                        struct smbXsrv_session *x = req->session;
-                       DATA_BLOB signing_key = smbd_smb2_signing_key(x, xconn);
+                       struct smb2_signing_key *signing_key =
+                               smbd_smb2_signing_key(x, xconn);
 
                        /*
                         * we need to remember the signing key
@@ -2848,14 +3126,20 @@ static NTSTATUS smbd_smb2_request_reply(struct smbd_smb2_request *req)
                         * we are sure that we do not change
                         * the header again.
                         */
-                       req->last_key = data_blob_dup_talloc(req, signing_key);
+                       req->last_key = data_blob_dup_talloc(req,
+                                                            signing_key->blob);
                        if (req->last_key.data == NULL) {
                                return NT_STATUS_NO_MEMORY;
                        }
                }
 
+               /*
+                * smbd_smb2_request_dispatch() will redo the impersonation.
+                * So we use req->xconn->client->raw_ev_ctx instead
+                * of req->ev_ctx here.
+                */
                tevent_schedule_immediate(im,
-                                       req->sconn->ev_ctx,
+                                       req->xconn->client->raw_ev_ctx,
                                        smbd_smb2_request_dispatch_immediate,
                                        req);
                return NT_STATUS_OK;
@@ -2878,16 +3162,21 @@ static NTSTATUS smbd_smb2_request_reply(struct smbd_smb2_request *req)
         * now check if we need to sign the current response
         */
        if (firsttf->iov_len == SMB2_TF_HDR_SIZE) {
-               status = smb2_signing_encrypt_pdu(req->first_key,
+               struct smb2_signing_key key = {
+                       .blob = req->first_key,
+               };
+               status = smb2_signing_encrypt_pdu(&key,
                                        xconn->smb2.server.cipher,
                                        firsttf,
                                        req->out.vector_count - first_idx);
+               smb2_signing_key_destructor(&key);
                if (!NT_STATUS_IS_OK(status)) {
                        return status;
                }
        } else if (req->do_signing) {
                struct smbXsrv_session *x = req->session;
-               DATA_BLOB signing_key = smbd_smb2_signing_key(x, xconn);
+               struct smb2_signing_key *signing_key =
+                       smbd_smb2_signing_key(x, xconn);
 
                status = smb2_signing_sign_pdu(signing_key,
                                               xconn->protocol,
@@ -2902,28 +3191,54 @@ static NTSTATUS smbd_smb2_request_reply(struct smbd_smb2_request *req)
        }
 
        if (req->preauth != NULL) {
-               struct hc_sha512state sctx;
-               int i;
+               gnutls_hash_hd_t hash_hnd = NULL;
+               size_t i;
+               int rc;
 
-               samba_SHA512_Init(&sctx);
-               samba_SHA512_Update(&sctx, req->preauth->sha512_value,
-                                   sizeof(req->preauth->sha512_value));
+               rc = gnutls_hash_init(&hash_hnd, GNUTLS_DIG_SHA512);
+               if (rc < 0) {
+                       return gnutls_error_to_ntstatus(rc, NT_STATUS_HASH_NOT_SUPPORTED);
+               }
+               rc = gnutls_hash(hash_hnd,
+                           req->preauth->sha512_value,
+                           sizeof(req->preauth->sha512_value));
+               if (rc < 0) {
+                       gnutls_hash_deinit(hash_hnd, NULL);
+                       return gnutls_error_to_ntstatus(rc, NT_STATUS_HASH_NOT_SUPPORTED);
+               }
                for (i = 1; i < req->in.vector_count; i++) {
-                       samba_SHA512_Update(&sctx,
-                                           req->in.vector[i].iov_base,
-                                           req->in.vector[i].iov_len);
+                       rc = gnutls_hash(hash_hnd,
+                                        req->in.vector[i].iov_base,
+                                        req->in.vector[i].iov_len);
+                       if (rc < 0) {
+                               gnutls_hash_deinit(hash_hnd, NULL);
+                               return gnutls_error_to_ntstatus(rc, NT_STATUS_HASH_NOT_SUPPORTED);
+                       }
+               }
+               if (rc < 0) {
+                       gnutls_hash_deinit(hash_hnd, NULL);
+                       return gnutls_error_to_ntstatus(rc, NT_STATUS_HASH_NOT_SUPPORTED);
+               }
+               gnutls_hash_output(hash_hnd, req->preauth->sha512_value);
+
+               rc = gnutls_hash(hash_hnd,
+                                req->preauth->sha512_value,
+                                sizeof(req->preauth->sha512_value));
+               if (rc < 0) {
+                       gnutls_hash_deinit(hash_hnd, NULL);
+                       return gnutls_error_to_ntstatus(rc, NT_STATUS_HASH_NOT_SUPPORTED);
                }
-               samba_SHA512_Final(req->preauth->sha512_value, &sctx);
-
-               samba_SHA512_Init(&sctx);
-               samba_SHA512_Update(&sctx, req->preauth->sha512_value,
-                                   sizeof(req->preauth->sha512_value));
                for (i = 1; i < req->out.vector_count; i++) {
-                       samba_SHA512_Update(&sctx,
-                                           req->out.vector[i].iov_base,
-                                           req->out.vector[i].iov_len);
+                       rc = gnutls_hash(hash_hnd,
+                                        req->out.vector[i].iov_base,
+                                        req->out.vector[i].iov_len);
+                       if (rc < 0) {
+                               gnutls_hash_deinit(hash_hnd, NULL);
+                               return gnutls_error_to_ntstatus(rc, NT_STATUS_HASH_NOT_SUPPORTED);
+                       }
                }
-               samba_SHA512_Final(req->preauth->sha512_value, &sctx);
+
+               gnutls_hash_deinit(hash_hnd, req->preauth->sha512_value);
 
                req->preauth = NULL;
        }
@@ -2997,13 +3312,20 @@ NTSTATUS smbd_smb2_request_done_ex(struct smbd_smb2_request *req,
        struct iovec *outbody_v;
        struct iovec *outdyn_v;
        uint32_t next_command_ofs;
+       uint64_t mid;
 
-       DEBUG(10,("smbd_smb2_request_done_ex: "
-                 "idx[%d] status[%s] body[%u] dyn[%s:%u] at %s\n",
-                 req->current_idx, nt_errstr(status), (unsigned int)body.length,
-                 dyn ? "yes": "no",
+       outhdr = SMBD_SMB2_OUT_HDR_PTR(req);
+       mid = BVAL(outhdr, SMB2_HDR_MESSAGE_ID);
+
+       DBG_DEBUG("mid [%"PRIu64"] idx[%d] status[%s] "
+                 "body[%u] dyn[%s:%u] at %s\n",
+                 mid,
+                 req->current_idx,
+                 nt_errstr(status),
+                 (unsigned int)body.length,
+                 dyn ? "yes" : "no",
                  (unsigned int)(dyn ? dyn->length : 0),
-                 location));
+                 location);
 
        if (body.length < 2) {
                return smbd_smb2_request_error(req, NT_STATUS_INTERNAL_ERROR);
@@ -3013,7 +3335,6 @@ NTSTATUS smbd_smb2_request_done_ex(struct smbd_smb2_request *req,
                return smbd_smb2_request_error(req, NT_STATUS_INTERNAL_ERROR);
        }
 
-       outhdr = SMBD_SMB2_OUT_HDR_PTR(req);
        outbody_v = SMBD_SMB2_OUT_BODY_IOV(req);
        outdyn_v = SMBD_SMB2_OUT_DYN_IOV(req);
 
@@ -3273,7 +3594,8 @@ static NTSTATUS smbd_smb2_send_break(struct smbXsrv_connection *xconn,
        }
 
        if (do_encryption) {
-               DATA_BLOB encryption_key = session->global->encryption_key;
+               struct smb2_signing_key *encryption_key =
+                       session->global->encryption_key;
 
                status = smb2_signing_encrypt_pdu(encryption_key,
                                        xconn->smb2.server.cipher,
@@ -3413,6 +3735,9 @@ static bool is_smb2_recvfile_write(struct smbd_smb2_request_read_state *state)
        if (IS_PRINT(fsp->conn)) {
                return false;
        }
+       if (fsp->base_fsp != NULL) {
+               return false;
+       }
 
        DEBUG(10,("Doing recvfile write len = %u\n",
                (unsigned int)(state->pktfull - state->pktlen)));
@@ -3469,7 +3794,7 @@ static NTSTATUS smbd_smb2_request_next_incoming(struct smbXsrv_connection *xconn
        return NT_STATUS_OK;
 }
 
-void smbd_smb2_process_negprot(struct smbXsrv_connection *xconn,
+NTSTATUS smbd_smb2_process_negprot(struct smbXsrv_connection *xconn,
                               uint64_t expected_seq_low,
                               const uint8_t *inpdu, size_t size)
 {
@@ -3483,25 +3808,25 @@ void smbd_smb2_process_negprot(struct smbXsrv_connection *xconn,
        status = smbd_initialize_smb2(xconn, expected_seq_low);
        if (!NT_STATUS_IS_OK(status)) {
                smbd_server_connection_terminate(xconn, nt_errstr(status));
-               return;
+               return status;
        }
 
        status = smbd_smb2_request_create(xconn, inpdu, size, &req);
        if (!NT_STATUS_IS_OK(status)) {
                smbd_server_connection_terminate(xconn, nt_errstr(status));
-               return;
+               return status;
        }
 
        status = smbd_smb2_request_validate(req);
        if (!NT_STATUS_IS_OK(status)) {
                smbd_server_connection_terminate(xconn, nt_errstr(status));
-               return;
+               return status;
        }
 
        status = smbd_smb2_request_setup_out(req);
        if (!NT_STATUS_IS_OK(status)) {
                smbd_server_connection_terminate(xconn, nt_errstr(status));
-               return;
+               return status;
        }
 
 #ifdef WITH_PROFILE
@@ -3516,16 +3841,17 @@ void smbd_smb2_process_negprot(struct smbXsrv_connection *xconn,
        status = smbd_smb2_request_dispatch(req);
        if (!NT_STATUS_IS_OK(status)) {
                smbd_server_connection_terminate(xconn, nt_errstr(status));
-               return;
+               return status;
        }
 
        status = smbd_smb2_request_next_incoming(xconn);
        if (!NT_STATUS_IS_OK(status)) {
                smbd_server_connection_terminate(xconn, nt_errstr(status));
-               return;
+               return status;
        }
 
        sconn->num_requests++;
+       return NT_STATUS_OK;
 }
 
 static int socket_error_from_errno(int ret,
@@ -3594,6 +3920,7 @@ static NTSTATUS smbd_smb2_flush_send_queue(struct smbXsrv_connection *xconn)
        while (xconn->smb2.send_queue != NULL) {
                struct smbd_smb2_send_queue *e = xconn->smb2.send_queue;
                bool ok;
+               struct msghdr msg;
 
                if (e->sendfile_header != NULL) {
                        size_t size = 0;
@@ -3637,12 +3964,19 @@ static NTSTATUS smbd_smb2_flush_send_queue(struct smbXsrv_connection *xconn)
                        talloc_free(e->mem_ctx);
 
                        if (!NT_STATUS_IS_OK(status)) {
+                               smbXsrv_connection_disconnect_transport(xconn,
+                                                                       status);
                                return status;
                        }
                        continue;
                }
 
-               ret = writev(xconn->transport.sock, e->vector, e->count);
+               msg = (struct msghdr) {
+                       .msg_iov = e->vector,
+                       .msg_iovlen = e->count,
+               };
+
+               ret = sendmsg(xconn->transport.sock, &msg, 0);
                if (ret == 0) {
                        /* propagate end of file */
                        return NT_STATUS_INTERNAL_ERROR;
@@ -3654,7 +3988,10 @@ static NTSTATUS smbd_smb2_flush_send_queue(struct smbXsrv_connection *xconn)
                        return NT_STATUS_OK;
                }
                if (err != 0) {
-                       return map_nt_error_from_unix_common(err);
+                       status = map_nt_error_from_unix_common(err);
+                       smbXsrv_connection_disconnect_transport(xconn,
+                                                               status);
+                       return status;
                }
 
                ok = iov_advance(&e->vector, &e->count, ret);
@@ -3698,6 +4035,7 @@ static NTSTATUS smbd_smb2_io_handler(struct smbXsrv_connection *xconn,
        bool retry;
        NTSTATUS status;
        NTTIME now;
+       struct msghdr msg;
 
        if (!NT_STATUS_IS_OK(xconn->transport.status)) {
                /*
@@ -3732,10 +4070,18 @@ again:
                state->vector.iov_len = NBT_HDR_SIZE;
        }
 
-       ret = readv(xconn->transport.sock, &state->vector, 1);
+       msg = (struct msghdr) {
+               .msg_iov = &state->vector,
+               .msg_iovlen = 1,
+       };
+
+       ret = recvmsg(xconn->transport.sock, &msg, 0);
        if (ret == 0) {
                /* propagate end of file */
-               return NT_STATUS_END_OF_FILE;
+               status = NT_STATUS_END_OF_FILE;
+               smbXsrv_connection_disconnect_transport(xconn,
+                                                       status);
+               return status;
        }
        err = socket_error_from_errno(ret, errno, &retry);
        if (retry) {
@@ -3744,7 +4090,10 @@ again:
                return NT_STATUS_OK;
        }
        if (err != 0) {
-               return map_nt_error_from_unix_common(err);
+               status = map_nt_error_from_unix_common(err);
+               smbXsrv_connection_disconnect_transport(xconn,
+                                                       status);
+               return status;
        }
 
        if (ret < state->vector.iov_len) {