s3: include smbd/smbd.h where needed.
[amitay/samba.git] / source3 / smbd / blocking.c
index 4e0d5289f88e78ae3c0cd0e8f9d08995bb3cbcda..548a3f0845124ca85a77b76ea90f3e8770a2296a 100644 (file)
@@ -2,95 +2,75 @@
    Unix SMB/CIFS implementation.
    Blocking Locking functions
    Copyright (C) Jeremy Allison 1998-2003
-   
+
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
-   
+
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
-   
+
    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
+#include "smbd/smbd.h"
+#include "smbd/globals.h"
+#include "librpc/gen_ndr/messaging.h"
+
 #undef DBGC_CLASS
 #define DBGC_CLASS DBGC_LOCKING
 
-/****************************************************************************
- This is the structure to queue to implement blocking locks.
- notify. It consists of the requesting SMB and the expiry time.
-*****************************************************************************/
-
-typedef struct _blocking_lock_record {
-       struct _blocking_lock_record *next;
-       struct _blocking_lock_record *prev;
-       int com_type;
-       files_struct *fsp;
-       struct timeval expire_time;
-       int lock_num;
-       SMB_BIG_UINT offset;
-       SMB_BIG_UINT count;
-       uint32 lock_pid;
-       uint32 blocking_pid; /* PID that blocks us. */
-       enum brl_flavour lock_flav;
-       enum brl_type lock_type;
-       char *inbuf;
-       int length;
-       bool encrypted;
-} blocking_lock_record;
-
-/* dlink list we store pending lock records on. */
-static blocking_lock_record *blocking_lock_queue;
-
-/* dlink list we move cancelled lock records onto. */
-static blocking_lock_record *blocking_lock_cancelled_queue;
-
-/* The event that makes us process our blocking lock queue */
-static struct timed_event *brl_timeout;
-
-/****************************************************************************
- Destructor for the above structure.
-****************************************************************************/
-
-static void free_blocking_lock_record(blocking_lock_record *blr)
-{
-       SAFE_FREE(blr->inbuf);
-       SAFE_FREE(blr);
-}
-
 /****************************************************************************
  Determine if this is a secondary element of a chained SMB.
   **************************************************************************/
 
-static bool in_chained_smb(void)
-{
-       return (chain_size != 0);
-}
-
 static void received_unlock_msg(struct messaging_context *msg,
                                void *private_data,
                                uint32_t msg_type,
                                struct server_id server_id,
                                DATA_BLOB *data);
-static void process_blocking_lock_queue(void);
 
-static void brl_timeout_fn(struct event_context *event_ctx,
+void brl_timeout_fn(struct event_context *event_ctx,
                           struct timed_event *te,
-                          const struct timeval *now,
+                          struct timeval now,
                           void *private_data)
 {
-       SMB_ASSERT(brl_timeout == te);
-       TALLOC_FREE(brl_timeout);
+       struct smbd_server_connection *sconn = talloc_get_type_abort(
+               private_data, struct smbd_server_connection);
+
+       if (sconn->using_smb2) {
+               SMB_ASSERT(sconn->smb2.locks.brl_timeout == te);
+               TALLOC_FREE(sconn->smb2.locks.brl_timeout);
+       } else {
+               SMB_ASSERT(sconn->smb1.locks.brl_timeout == te);
+               TALLOC_FREE(sconn->smb1.locks.brl_timeout);
+       }
 
        change_to_root_user();  /* TODO: Possibly run all timed events as
                                 * root */
 
-       process_blocking_lock_queue();
+       process_blocking_lock_queue(sconn);
+}
+
+/****************************************************************************
+ We need a version of timeval_min that treats zero timval as infinite.
+****************************************************************************/
+
+struct timeval timeval_brl_min(const struct timeval *tv1,
+                                       const struct timeval *tv2)
+{
+       if (timeval_is_zero(tv1)) {
+               return *tv2;
+       }
+       if (timeval_is_zero(tv2)) {
+               return *tv1;
+       }
+       return timeval_min(tv1, tv2);
 }
 
 /****************************************************************************
@@ -98,46 +78,71 @@ static void brl_timeout_fn(struct event_context *event_ctx,
  next processing.
 ****************************************************************************/
 
-static bool recalc_brl_timeout(void)
+static bool recalc_brl_timeout(struct smbd_server_connection *sconn)
 {
-       blocking_lock_record *brl;
+       struct blocking_lock_record *blr;
        struct timeval next_timeout;
+       int max_brl_timeout = lp_parm_int(-1, "brl", "recalctime", 5);
 
-       TALLOC_FREE(brl_timeout);
+       TALLOC_FREE(sconn->smb1.locks.brl_timeout);
 
-       next_timeout = timeval_zero();  
+       next_timeout = timeval_zero();
 
-       for (brl = blocking_lock_queue; brl; brl = brl->next) {
-               if (timeval_is_zero(&brl->expire_time)) {
+       for (blr = sconn->smb1.locks.blocking_lock_queue; blr; blr = blr->next) {
+               if (timeval_is_zero(&blr->expire_time)) {
                        /*
-                        * If we're blocked on pid 0xFFFFFFFF this is
+                        * If we're blocked on pid 0xFFFFFFFFFFFFFFFFLL this is
                         * a POSIX lock, so calculate a timeout of
                         * 10 seconds into the future.
                         */
-                        if (brl->blocking_pid == 0xFFFFFFFF) {
+                        if (blr->blocking_smblctx == 0xFFFFFFFFFFFFFFFFLL) {
                                struct timeval psx_to = timeval_current_ofs(10, 0);
-                               next_timeout = timeval_min(&next_timeout, &psx_to);
+                               next_timeout = timeval_brl_min(&next_timeout, &psx_to);
                         }
 
                        continue;
                }
 
-               if (timeval_is_zero(&next_timeout)) {
-                       next_timeout = brl->expire_time;
-               }
-               else {
-                       next_timeout = timeval_min(&next_timeout,
-                                                  &brl->expire_time);
-               }
+               next_timeout = timeval_brl_min(&next_timeout, &blr->expire_time);
        }
 
        if (timeval_is_zero(&next_timeout)) {
+               DEBUG(10, ("Next timeout = Infinite.\n"));
                return True;
        }
 
-       if (!(brl_timeout = event_add_timed(smbd_event_context(), NULL,
-                                           next_timeout, "brl_timeout",
-                                           brl_timeout_fn, NULL))) {
+       /* 
+        to account for unclean shutdowns by clients we need a
+        maximum timeout that we use for checking pending locks. If
+        we have any pending locks at all, then check if the pending
+        lock can continue at least every brl:recalctime seconds
+        (default 5 seconds).
+
+        This saves us needing to do a message_send_all() in the
+        SIGCHLD handler in the parent daemon. That
+        message_send_all() caused O(n^2) work to be done when IP
+        failovers happened in clustered Samba, which could make the
+        entire system unusable for many minutes.
+       */
+
+       if (max_brl_timeout > 0) {
+               struct timeval min_to = timeval_current_ofs(max_brl_timeout, 0);
+               next_timeout = timeval_min(&next_timeout, &min_to);
+       }
+
+       if (DEBUGLVL(10)) {
+               struct timeval cur, from_now;
+
+               cur = timeval_current();
+               from_now = timeval_until(&cur, &next_timeout);
+               DEBUG(10, ("Next timeout = %d.%d seconds from now.\n",
+                   (int)from_now.tv_sec, (int)from_now.tv_usec));
+       }
+
+       sconn->smb1.locks.brl_timeout = event_add_timed(smbd_event_context(),
+                                                       NULL, next_timeout,
+                                                       brl_timeout_fn, sconn);
+       if (sconn->smb1.locks.brl_timeout == NULL) {
                return False;
        }
 
@@ -150,23 +155,36 @@ static bool recalc_brl_timeout(void)
 ****************************************************************************/
 
 bool push_blocking_lock_request( struct byte_range_lock *br_lck,
-               const struct smb_request *req,
+               struct smb_request *req,
                files_struct *fsp,
                int lock_timeout,
                int lock_num,
-               uint32 lock_pid,
+               uint64_t smblctx,
                enum brl_type lock_type,
                enum brl_flavour lock_flav,
-               SMB_BIG_UINT offset,
-               SMB_BIG_UINT count,
-               uint32 blocking_pid)
+               uint64_t offset,
+               uint64_t count,
+               uint64_t blocking_smblctx)
 {
-       static bool set_lock_msg;
-       size_t length = smb_len(req->inbuf)+4;
-       blocking_lock_record *blr;
+       struct smbd_server_connection *sconn = req->sconn;
+       struct blocking_lock_record *blr;
        NTSTATUS status;
 
-       if(in_chained_smb() ) {
+       if (req->smb2req) {
+               return push_blocking_lock_request_smb2(br_lck,
+                               req,
+                               fsp,
+                               lock_timeout,
+                               lock_num,
+                               smblctx,
+                               lock_type,
+                               lock_flav,
+                               offset,
+                               count,
+                               blocking_smblctx);
+       }
+
+       if(req_is_in_chain(req)) {
                DEBUG(0,("push_blocking_lock_request: cannot queue a chained request (currently).\n"));
                return False;
        }
@@ -176,7 +194,8 @@ bool push_blocking_lock_request( struct byte_range_lock *br_lck,
         * the expiration time here.
         */
 
-       if((blr = SMB_MALLOC_P(blocking_lock_record)) == NULL) {
+       blr = talloc(NULL, struct blocking_lock_record);
+       if (blr == NULL) {
                DEBUG(0,("push_blocking_lock_request: Malloc fail !\n" ));
                return False;
        }
@@ -184,13 +203,6 @@ bool push_blocking_lock_request( struct byte_range_lock *br_lck,
        blr->next = NULL;
        blr->prev = NULL;
 
-       if((blr->inbuf = (char *)SMB_MALLOC(length)) == NULL) {
-               DEBUG(0,("push_blocking_lock_request: Malloc fail (2)!\n" ));
-               SAFE_FREE(blr);
-               return False;
-       }
-
-       blr->com_type = CVAL(req->inbuf,smb_com);
        blr->fsp = fsp;
        if (lock_timeout == -1) {
                blr->expire_time.tv_sec = 0;
@@ -200,52 +212,53 @@ bool push_blocking_lock_request( struct byte_range_lock *br_lck,
                                        (lock_timeout % 1000) * 1000);
        }
        blr->lock_num = lock_num;
-       blr->lock_pid = lock_pid;
-       blr->blocking_pid = blocking_pid;
+       blr->smblctx = smblctx;
+       blr->blocking_smblctx = blocking_smblctx;
        blr->lock_flav = lock_flav;
        blr->lock_type = lock_type;
        blr->offset = offset;
        blr->count = count;
-       memcpy(blr->inbuf, req->inbuf, length);
-       blr->length = length;
-       blr->encrypted = req->encrypted;
+      
+       /* Specific brl_lock() implementations can fill this in. */
+       blr->blr_private = NULL;
 
        /* Add a pending lock record for this. */
-       status = brl_lock(smbd_messaging_context(), br_lck,
-                       lock_pid,
-                       procid_self(),
+       status = brl_lock(req->sconn->msg_ctx,
+                       br_lck,
+                       smblctx,
+                       sconn_server_id(req->sconn),
                        offset,
                        count,
                        lock_type == READ_LOCK ? PENDING_READ_LOCK : PENDING_WRITE_LOCK,
                        blr->lock_flav,
-                       lock_timeout ? True : False, /* blocking_lock. */
-                       NULL);
+                       True,
+                       NULL,
+                       blr);
 
        if (!NT_STATUS_IS_OK(status)) {
                DEBUG(0,("push_blocking_lock_request: failed to add PENDING_LOCK record.\n"));
-               DLIST_REMOVE(blocking_lock_queue, blr);
-               free_blocking_lock_record(blr);
+               TALLOC_FREE(blr);
                return False;
        }
 
-       DLIST_ADD_END(blocking_lock_queue, blr, blocking_lock_record *);
-       recalc_brl_timeout();
+       SMB_PERFCOUNT_DEFER_OP(&req->pcd, &req->pcd);
+       blr->req = talloc_move(blr, &req);
+
+       DLIST_ADD_END(sconn->smb1.locks.blocking_lock_queue, blr, struct blocking_lock_record *);
+       recalc_brl_timeout(sconn);
 
        /* Ensure we'll receive messages when this is unlocked. */
-       if (!set_lock_msg) {
-               messaging_register(smbd_messaging_context(), NULL,
+       if (!sconn->smb1.locks.blocking_lock_unlock_state) {
+               messaging_register(sconn->msg_ctx, NULL,
                                   MSG_SMB_UNLOCK, received_unlock_msg);
-               set_lock_msg = True;
+               sconn->smb1.locks.blocking_lock_unlock_state = true;
        }
 
-       DEBUG(3,("push_blocking_lock_request: lock request length=%u blocked with "
+       DEBUG(3,("push_blocking_lock_request: lock request blocked with "
                "expiry time (%u sec. %u usec) (+%d msec) for fnum = %d, name = %s\n",
-               (unsigned int)length, (unsigned int)blr->expire_time.tv_sec,
+               (unsigned int)blr->expire_time.tv_sec,
                (unsigned int)blr->expire_time.tv_usec, lock_timeout,
-               blr->fsp->fnum, blr->fsp->fsp_name ));
-
-       /* Push the MID of this packet on the signing queue. */
-       srv_defer_sign_response(SVAL(req->inbuf,smb_mid));
+               blr->fsp->fnum, fsp_str_dbg(blr->fsp)));
 
        return True;
 }
@@ -254,16 +267,9 @@ bool push_blocking_lock_request( struct byte_range_lock *br_lck,
  Return a lockingX success SMB.
 *****************************************************************************/
 
-static void reply_lockingX_success(blocking_lock_record *blr)
+static void reply_lockingX_success(struct blocking_lock_record *blr)
 {
-       struct smb_request *req;
-
-       if (!(req = talloc(talloc_tos(), struct smb_request))) {
-               smb_panic("Could not allocate smb_request");
-       }
-
-       init_smb_request(req, (uint8 *)blr->inbuf, 0, blr->encrypted);
-       reply_outbuf(req, 2, 0);
+       reply_outbuf(blr->req, 2, 0);
 
        /*
         * As this message is a lockingX call we must handle
@@ -273,26 +279,16 @@ static void reply_lockingX_success(blocking_lock_record *blr)
         * that here and must set up the chain info manually.
         */
 
-       chain_reply(req);
-
-       if (!srv_send_smb(smbd_server_fd(),
-                       (char *)req->outbuf,
-                       IS_CONN_ENCRYPTED(blr->fsp->conn))) {
-               exit_server_cleanly("send_blocking_reply: srv_send_smb failed.");
-       }
+       chain_reply(blr->req);
+       TALLOC_FREE(blr->req->outbuf);
 }
 
 /****************************************************************************
  Return a generic lock fail error blocking call.
 *****************************************************************************/
 
-static void generic_blocking_lock_error(blocking_lock_record *blr, NTSTATUS status)
+static void generic_blocking_lock_error(struct blocking_lock_record *blr, NTSTATUS status)
 {
-       char outbuf[smb_size];
-       char *inbuf = blr->inbuf;
-
-       construct_reply_common(inbuf, outbuf);
-
        /* whenever a timeout is given w2k maps LOCK_NOT_GRANTED to
           FILE_LOCK_CONFLICT! (tridge) */
        if (NT_STATUS_EQUAL(status, NT_STATUS_LOCK_NOT_GRANTED)) {
@@ -304,9 +300,10 @@ static void generic_blocking_lock_error(blocking_lock_record *blr, NTSTATUS stat
                files_struct *fsp = blr->fsp;
 
                if (fsp) {
-                       fsp->last_lock_failure.context.smbpid = blr->lock_pid;
+                       fsp->last_lock_failure.context.smblctx = blr->smblctx;
                        fsp->last_lock_failure.context.tid = fsp->conn->cnum;
-                       fsp->last_lock_failure.context.pid = procid_self();
+                       fsp->last_lock_failure.context.pid =
+                               sconn_server_id(fsp->conn->sconn);
                        fsp->last_lock_failure.start = blr->offset;
                        fsp->last_lock_failure.size = blr->count;
                        fsp->last_lock_failure.fnum = fsp->fnum;
@@ -315,10 +312,13 @@ static void generic_blocking_lock_error(blocking_lock_record *blr, NTSTATUS stat
                }
        }
 
-       ERROR_NT(status);
-       if (!srv_send_smb(smbd_server_fd(),outbuf, blr->encrypted)) {
+       reply_nterror(blr->req, status);
+       if (!srv_send_smb(blr->req->sconn, (char *)blr->req->outbuf,
+                         true, blr->req->seqnum+1,
+                         blr->req->encrypted, NULL)) {
                exit_server_cleanly("generic_blocking_lock_error: srv_send_smb failed.");
        }
+       TALLOC_FREE(blr->req->outbuf);
 }
 
 /****************************************************************************
@@ -326,20 +326,20 @@ static void generic_blocking_lock_error(blocking_lock_record *blr, NTSTATUS stat
  obtained first.
 *****************************************************************************/
 
-static void reply_lockingX_error(blocking_lock_record *blr, NTSTATUS status)
+static void reply_lockingX_error(struct blocking_lock_record *blr, NTSTATUS status)
 {
-       char *inbuf = blr->inbuf;
        files_struct *fsp = blr->fsp;
-       uint16 num_ulocks = SVAL(inbuf,smb_vwv6);
-       SMB_BIG_UINT count = (SMB_BIG_UINT)0, offset = (SMB_BIG_UINT) 0;
-       uint32 lock_pid;
-       unsigned char locktype = CVAL(inbuf,smb_vwv3);
+       uint16 num_ulocks = SVAL(blr->req->vwv+6, 0);
+       uint64_t count = (uint64_t)0, offset = (uint64_t) 0;
+       uint64_t smblctx;
+       unsigned char locktype = CVAL(blr->req->vwv+3, 0);
        bool large_file_format = (locktype & LOCKING_ANDX_LARGE_FILES);
-       char *data;
+       uint8_t *data;
        int i;
 
-       data = smb_buf(inbuf) + ((large_file_format ? 20 : 10)*num_ulocks);
-       
+       data = (uint8_t *)blr->req->buf
+               + ((large_file_format ? 20 : 10)*num_ulocks);
+
        /* 
         * Data now points at the beginning of the list
         * of smb_lkrng structs.
@@ -350,27 +350,27 @@ static void reply_lockingX_error(blocking_lock_record *blr, NTSTATUS status)
         * as under POSIX rules, if we have a lock already there, we
         * will delete it (and we shouldn't) .....
         */
-       
+
        for(i = blr->lock_num - 1; i >= 0; i--) {
                bool err;
-               
-               lock_pid = get_lock_pid( data, i, large_file_format);
+
+               smblctx = get_lock_pid( data, i, large_file_format);
                count = get_lock_count( data, i, large_file_format);
                offset = get_lock_offset( data, i, large_file_format, &err);
-               
+
                /*
                 * We know err cannot be set as if it was the lock
                 * request would never have been queued. JRA.
                 */
-               
-               do_unlock(smbd_messaging_context(),
+
+               do_unlock(fsp->conn->sconn->msg_ctx,
                        fsp,
-                       lock_pid,
+                       smblctx,
                        count,
                        offset,
                        WINDOWS_LOCK);
        }
-       
+
        generic_blocking_lock_error(blr, status);
 }
 
@@ -378,30 +378,34 @@ static void reply_lockingX_error(blocking_lock_record *blr, NTSTATUS status)
  Return a lock fail error.
 *****************************************************************************/
 
-static void blocking_lock_reply_error(blocking_lock_record *blr, NTSTATUS status)
+static void blocking_lock_reply_error(struct blocking_lock_record *blr, NTSTATUS status)
 {
-       switch(blr->com_type) {
+       DEBUG(10, ("Replying with error=%s. BLR = %p\n", nt_errstr(status), blr));
+
+       switch(blr->req->cmd) {
        case SMBlockingX:
                reply_lockingX_error(blr, status);
                break;
        case SMBtrans2:
        case SMBtranss2:
-               {
-                       char outbuf[smb_size];
-                       char *inbuf = blr->inbuf;
-                       construct_reply_common(inbuf, outbuf);
-                       /* construct_reply_common has done us the favor to pre-fill the
-                        * command field with SMBtranss2 which is wrong :-)
-                        */
-                       SCVAL(outbuf,smb_com,SMBtrans2);
-                       ERROR_NT(status);
-                       if (!srv_send_smb(smbd_server_fd(),
-                                       outbuf,
-                                       IS_CONN_ENCRYPTED(blr->fsp->conn))) {
-                               exit_server_cleanly("blocking_lock_reply_error: srv_send_smb failed.");
-                       }
-                       break;
+               reply_nterror(blr->req, status);
+
+               /*
+                * construct_reply_common has done us the favor to pre-fill
+                * the command field with SMBtranss2 which is wrong :-)
+                */
+               SCVAL(blr->req->outbuf,smb_com,SMBtrans2);
+
+               if (!srv_send_smb(blr->req->sconn,
+                                 (char *)blr->req->outbuf,
+                                 true, blr->req->seqnum+1,
+                                 IS_CONN_ENCRYPTED(blr->fsp->conn),
+                                 NULL)) {
+                       exit_server_cleanly("blocking_lock_reply_error: "
+                                           "srv_send_smb failed.");
                }
+               TALLOC_FREE(blr->req->outbuf);
+               break;
        default:
                DEBUG(0,("blocking_lock_reply_error: PANIC - unknown type on blocking lock queue - exiting.!\n"));
                exit_server("PANIC - unknown type on blocking lock queue");
@@ -413,20 +417,20 @@ static void blocking_lock_reply_error(blocking_lock_record *blr, NTSTATUS status
  Returns True if we want to be removed from the list.
 *****************************************************************************/
 
-static bool process_lockingX(blocking_lock_record *blr)
+static bool process_lockingX(struct blocking_lock_record *blr)
 {
-       char *inbuf = blr->inbuf;
-       unsigned char locktype = CVAL(inbuf,smb_vwv3);
+       unsigned char locktype = CVAL(blr->req->vwv+3, 0);
        files_struct *fsp = blr->fsp;
-       uint16 num_ulocks = SVAL(inbuf,smb_vwv6);
-       uint16 num_locks = SVAL(inbuf,smb_vwv7);
-       SMB_BIG_UINT count = (SMB_BIG_UINT)0, offset = (SMB_BIG_UINT)0;
-       uint32 lock_pid;
+       uint16 num_ulocks = SVAL(blr->req->vwv+6, 0);
+       uint16 num_locks = SVAL(blr->req->vwv+7, 0);
+       uint64_t count = (uint64_t)0, offset = (uint64_t)0;
+       uint64_t smblctx;
        bool large_file_format = (locktype & LOCKING_ANDX_LARGE_FILES);
-       char *data;
+       uint8_t *data;
        NTSTATUS status = NT_STATUS_OK;
 
-       data = smb_buf(inbuf) + ((large_file_format ? 20 : 10)*num_ulocks);
+       data = (uint8_t *)blr->req->buf
+               + ((large_file_format ? 20 : 10)*num_ulocks);
 
        /* 
         * Data now points at the beginning of the list
@@ -437,26 +441,27 @@ static bool process_lockingX(blocking_lock_record *blr)
                struct byte_range_lock *br_lck = NULL;
                bool err;
 
-               lock_pid = get_lock_pid( data, blr->lock_num, large_file_format);
+               smblctx = get_lock_pid( data, blr->lock_num, large_file_format);
                count = get_lock_count( data, blr->lock_num, large_file_format);
                offset = get_lock_offset( data, blr->lock_num, large_file_format, &err);
-               
+
                /*
                 * We know err cannot be set as if it was the lock
                 * request would never have been queued. JRA.
                 */
                errno = 0;
-               br_lck = do_lock(smbd_messaging_context(),
+               br_lck = do_lock(fsp->conn->sconn->msg_ctx,
                                fsp,
-                               lock_pid,
+                               smblctx,
                                count,
-                               offset, 
+                               offset,
                                ((locktype & LOCKING_ANDX_SHARED_LOCK) ?
                                        READ_LOCK : WRITE_LOCK),
                                WINDOWS_LOCK,
                                True,
                                &status,
-                               &blr->blocking_pid);
+                               &blr->blocking_smblctx,
+                               blr);
 
                TALLOC_FREE(br_lck);
 
@@ -469,20 +474,22 @@ static bool process_lockingX(blocking_lock_record *blr)
                /*
                 * Success - we got all the locks.
                 */
-               
-               DEBUG(3,("process_lockingX file = %s, fnum=%d type=%d num_locks=%d\n",
-                        fsp->fsp_name, fsp->fnum, (unsigned int)locktype, num_locks) );
+
+               DEBUG(3,("process_lockingX file = %s, fnum=%d type=%d "
+                        "num_locks=%d\n", fsp_str_dbg(fsp), fsp->fnum,
+                        (unsigned int)locktype, num_locks));
 
                reply_lockingX_success(blr);
                return True;
-       } else if (!NT_STATUS_EQUAL(status,NT_STATUS_LOCK_NOT_GRANTED) &&
-                       !NT_STATUS_EQUAL(status,NT_STATUS_FILE_LOCK_CONFLICT)) {
-                       /*
-                        * We have other than a "can't get lock"
-                        * error. Free any locks we had and return an error.
-                        * Return True so we get dequeued.
-                        */
-               
+       }
+
+       if (!NT_STATUS_EQUAL(status,NT_STATUS_LOCK_NOT_GRANTED) &&
+           !NT_STATUS_EQUAL(status,NT_STATUS_FILE_LOCK_CONFLICT)) {
+               /*
+                * We have other than a "can't get lock"
+                * error. Free any locks we had and return an error.
+                * Return True so we get dequeued.
+                */
                blocking_lock_reply_error(blr, status);
                return True;
        }
@@ -490,11 +497,11 @@ static bool process_lockingX(blocking_lock_record *blr)
        /*
         * Still can't get all the locks - keep waiting.
         */
-       
+
        DEBUG(10,("process_lockingX: only got %d locks of %d needed for file %s, fnum = %d. \
 Waiting....\n", 
-                 blr->lock_num, num_locks, fsp->fsp_name, fsp->fnum));
-       
+                blr->lock_num, num_locks, fsp_str_dbg(fsp), fsp->fnum));
+
        return False;
 }
 
@@ -503,21 +510,22 @@ Waiting....\n",
  Returns True if we want to be removed from the list.
 *****************************************************************************/
 
-static bool process_trans2(blocking_lock_record *blr)
+static bool process_trans2(struct blocking_lock_record *blr)
 {
-       struct smb_request *req;
        char params[2];
        NTSTATUS status;
-       struct byte_range_lock *br_lck = do_lock(smbd_messaging_context(),
+       struct byte_range_lock *br_lck = do_lock(
+                                               blr->fsp->conn->sconn->msg_ctx,
                                                blr->fsp,
-                                               blr->lock_pid,
+                                               blr->smblctx,
                                                blr->count,
                                                blr->offset,
                                                blr->lock_type,
                                                blr->lock_flav,
                                                True,
                                                &status,
-                                               &blr->blocking_pid);
+                                               &blr->blocking_smblctx,
+                                               blr);
        TALLOC_FREE(br_lck);
 
        if (!NT_STATUS_IS_OK(status)) {
@@ -535,17 +543,9 @@ static bool process_trans2(blocking_lock_record *blr)
 
        /* We finally got the lock, return success. */
 
-       if (!(req = talloc(talloc_tos(), struct smb_request))) {
-               blocking_lock_reply_error(blr, NT_STATUS_NO_MEMORY);
-               return True;
-       }
-
-       init_smb_request(req, (uint8 *)blr->inbuf, 0, blr->encrypted);
-
-       SCVAL(req->inbuf, smb_com, SMBtrans2);
        SSVAL(params,0,0);
        /* Fake up max_data_bytes here - we know it fits. */
-       send_trans2_replies(blr->fsp->conn, req, params, 2, NULL, 0, 0xffff);
+       send_trans2_replies(blr->fsp->conn, blr->req, params, 2, NULL, 0, 0xffff);
        return True;
 }
 
@@ -555,9 +555,9 @@ static bool process_trans2(blocking_lock_record *blr)
  Returns True if we want to be removed from the list.
 *****************************************************************************/
 
-static bool blocking_lock_record_process(blocking_lock_record *blr)
+static bool blocking_lock_record_process(struct blocking_lock_record *blr)
 {
-       switch(blr->com_type) {
+       switch(blr->req->cmd) {
                case SMBlockingX:
                        return process_lockingX(blr);
                case SMBtrans2:
@@ -572,92 +572,121 @@ static bool blocking_lock_record_process(blocking_lock_record *blr)
 
 /****************************************************************************
  Cancel entries by fnum from the blocking lock pending queue.
+ Called when a file is closed.
 *****************************************************************************/
 
-void cancel_pending_lock_requests_by_fid(files_struct *fsp, struct byte_range_lock *br_lck)
+void cancel_pending_lock_requests_by_fid(files_struct *fsp,
+                       struct byte_range_lock *br_lck,
+                       enum file_close_type close_type)
 {
-       blocking_lock_record *blr, *next = NULL;
-
-       for(blr = blocking_lock_queue; blr; blr = next) {
-               next = blr->next;
-               if(blr->fsp->fnum == fsp->fnum) {
-                       unsigned char locktype = 0;
-
-                       if (blr->com_type == SMBlockingX) {
-                               locktype = CVAL(blr->inbuf,smb_vwv3);
-                       }
+       struct smbd_server_connection *sconn = fsp->conn->sconn;
+       struct blocking_lock_record *blr, *blr_cancelled, *next = NULL;
+
+       if (sconn->using_smb2) {
+               cancel_pending_lock_requests_by_fid_smb2(fsp,
+                                       br_lck,
+                                       close_type);
+               return;
+       }
 
-                       if (br_lck) {
-                               DEBUG(10,("remove_pending_lock_requests_by_fid - removing request type %d for \
-file %s fnum = %d\n", blr->com_type, fsp->fsp_name, fsp->fnum ));
+       for(blr = sconn->smb1.locks.blocking_lock_queue; blr; blr = next) {
+               unsigned char locktype = 0;
 
-                               brl_lock_cancel(br_lck,
-                                       blr->lock_pid,
-                                       procid_self(),
-                                       blr->offset,
-                                       blr->count,
-                                       blr->lock_flav);
+               next = blr->next;
+               if (blr->fsp->fnum != fsp->fnum) {
+                       continue;
+               }
 
-                               blocking_lock_cancel(fsp,
-                                       blr->lock_pid,
-                                       blr->offset,
-                                       blr->count,
-                                       blr->lock_flav,
-                                       locktype,
-                                       NT_STATUS_RANGE_NOT_LOCKED);
-                       }
-                       /* We're closing the file fsp here, so ensure
-                        * we don't have a dangling pointer. */
-                       blr->fsp = NULL;
+               if (blr->req->cmd == SMBlockingX) {
+                       locktype = CVAL(blr->req->vwv+3, 0);
                }
+
+               DEBUG(10, ("remove_pending_lock_requests_by_fid - removing "
+                          "request type %d for file %s fnum = %d\n",
+                          blr->req->cmd, fsp_str_dbg(fsp), fsp->fnum));
+
+               blr_cancelled = blocking_lock_cancel_smb1(fsp,
+                                    blr->smblctx,
+                                    blr->offset,
+                                    blr->count,
+                                    blr->lock_flav,
+                                    locktype,
+                                    NT_STATUS_RANGE_NOT_LOCKED);
+
+               SMB_ASSERT(blr_cancelled == blr);
+
+               brl_lock_cancel(br_lck,
+                               blr->smblctx,
+                               sconn_server_id(sconn),
+                               blr->offset,
+                               blr->count,
+                               blr->lock_flav,
+                               blr);
+
+               /* We're closing the file fsp here, so ensure
+                * we don't have a dangling pointer. */
+               blr->fsp = NULL;
        }
 }
 
 /****************************************************************************
  Delete entries by mid from the blocking lock pending queue. Always send reply.
+ Only called from the SMB1 cancel code.
 *****************************************************************************/
 
-void remove_pending_lock_requests_by_mid(int mid)
+void remove_pending_lock_requests_by_mid_smb1(
+       struct smbd_server_connection *sconn, uint64_t mid)
 {
-       blocking_lock_record *blr, *next = NULL;
+       struct blocking_lock_record *blr, *next = NULL;
+
+       for(blr = sconn->smb1.locks.blocking_lock_queue; blr; blr = next) {
+               files_struct *fsp;
+               struct byte_range_lock *br_lck;
 
-       for(blr = blocking_lock_queue; blr; blr = next) {
                next = blr->next;
-               if(SVAL(blr->inbuf,smb_mid) == mid) {
-                       files_struct *fsp = blr->fsp;
-                       struct byte_range_lock *br_lck = brl_get_locks(NULL, fsp);
 
-                       if (br_lck) {
-                               DEBUG(10,("remove_pending_lock_requests_by_mid - removing request type %d for \
-file %s fnum = %d\n", blr->com_type, fsp->fsp_name, fsp->fnum ));
+               if (blr->req->mid != mid) {
+                       continue;
+               }
 
-                               brl_lock_cancel(br_lck,
-                                       blr->lock_pid,
-                                       procid_self(),
+               fsp = blr->fsp;
+               br_lck = brl_get_locks(talloc_tos(), fsp);
+
+               if (br_lck) {
+                       DEBUG(10, ("remove_pending_lock_requests_by_mid_smb1 - "
+                                  "removing request type %d for file %s fnum "
+                                  "= %d\n", blr->req->cmd, fsp_str_dbg(fsp),
+                                  fsp->fnum ));
+
+                       brl_lock_cancel(br_lck,
+                                       blr->smblctx,
+                                       sconn_server_id(sconn),
                                        blr->offset,
                                        blr->count,
-                                       blr->lock_flav);
-                               TALLOC_FREE(br_lck);
-                       }
-
-                       blocking_lock_reply_error(blr,NT_STATUS_FILE_LOCK_CONFLICT);
-                       DLIST_REMOVE(blocking_lock_queue, blr);
-                       free_blocking_lock_record(blr);
+                                       blr->lock_flav,
+                                       blr);
+                       TALLOC_FREE(br_lck);
                }
+
+               blocking_lock_reply_error(blr,NT_STATUS_FILE_LOCK_CONFLICT);
+               DLIST_REMOVE(sconn->smb1.locks.blocking_lock_queue, blr);
+               TALLOC_FREE(blr);
        }
 }
 
 /****************************************************************************
  Is this mid a blocking lock request on the queue ?
+ Currently only called from the SMB1 unix extensions POSIX lock code.
 *****************************************************************************/
 
-bool blocking_lock_was_deferred(int mid)
+bool blocking_lock_was_deferred_smb1(
+       struct smbd_server_connection *sconn, uint64_t mid)
 {
-       blocking_lock_record *blr, *next = NULL;
+       struct blocking_lock_record *blr, *next = NULL;
 
-       for(blr = blocking_lock_queue; blr; blr = next) {
+       for(blr = sconn->smb1.locks.blocking_lock_queue; blr; blr = next) {
                next = blr->next;
-               if(SVAL(blr->inbuf,smb_mid) == mid) {
+               if(blr->req->mid == mid) {
                        return True;
                }
        }
@@ -674,133 +703,89 @@ static void received_unlock_msg(struct messaging_context *msg,
                                struct server_id server_id,
                                DATA_BLOB *data)
 {
+       struct smbd_server_connection *sconn;
+
+       sconn = msg_ctx_to_sconn(msg);
+       if (sconn == NULL) {
+               DEBUG(1, ("could not find sconn\n"));
+               return;
+       }
+
        DEBUG(10,("received_unlock_msg\n"));
-       process_blocking_lock_queue();
+       process_blocking_lock_queue(sconn);
 }
 
 /****************************************************************************
  Process the blocking lock queue. Note that this is only called as root.
 *****************************************************************************/
 
-static void process_blocking_lock_queue(void)
+void process_blocking_lock_queue(struct smbd_server_connection *sconn)
 {
        struct timeval tv_curr = timeval_current();
-       blocking_lock_record *blr, *next = NULL;
-       bool recalc_timeout = False;
+       struct blocking_lock_record *blr, *next = NULL;
+
+       if (sconn->using_smb2) {
+               process_blocking_lock_queue_smb2(sconn, tv_curr);
+               return;
+       }
 
        /*
         * Go through the queue and see if we can get any of the locks.
         */
 
-       for (blr = blocking_lock_queue; blr; blr = next) {
-               connection_struct *conn = NULL;
-               uint16 vuid;
-               files_struct *fsp = NULL;
+       for (blr = sconn->smb1.locks.blocking_lock_queue; blr; blr = next) {
 
                next = blr->next;
 
-               /*
-                * Ensure we don't have any old chain_fsp values
-                * sitting around....
-                */
-               chain_size = 0;
-               file_chain_reset();
-               fsp = blr->fsp;
-
-               conn = conn_find(SVAL(blr->inbuf,smb_tid));
-               vuid = (lp_security() == SEC_SHARE) ? UID_FIELD_INVALID :
-                               SVAL(blr->inbuf,smb_uid);
-
-               DEBUG(5,("process_blocking_lock_queue: examining pending lock fnum = %d for file %s\n",
-                       fsp->fnum, fsp->fsp_name ));
-
-               if(!change_to_user(conn,vuid)) {
-                       struct byte_range_lock *br_lck = brl_get_locks(NULL, fsp);
-
-                       /*
-                        * Remove the entry and return an error to the client.
-                        */
-
-                       if (br_lck) {
-                               brl_lock_cancel(br_lck,
-                                       blr->lock_pid,
-                                       procid_self(),
-                                       blr->offset,
-                                       blr->count,
-                                       blr->lock_flav);
-                               TALLOC_FREE(br_lck);
-                       }
-
-                       DEBUG(0,("process_blocking_lock_queue: Unable to become user vuid=%d.\n",
-                               vuid ));
-                       blocking_lock_reply_error(blr,NT_STATUS_ACCESS_DENIED);
-                       DLIST_REMOVE(blocking_lock_queue, blr);
-                       free_blocking_lock_record(blr);
-                       recalc_timeout = True;
-                       continue;
-               }
-
-               if(!set_current_service(conn,SVAL(blr->inbuf,smb_flg),True)) {
-                       struct byte_range_lock *br_lck = brl_get_locks(NULL, fsp);
-
-                       /*
-                        * Remove the entry and return an error to the client.
-                        */
-
-                       if (br_lck) {
-                               brl_lock_cancel(br_lck,
-                                       blr->lock_pid,
-                                       procid_self(),
-                                       blr->offset,
-                                       blr->count,
-                                       blr->lock_flav);
-                               TALLOC_FREE(br_lck);
-                       }
-
-                       DEBUG(0,("process_blocking_lock_queue: Unable to become service Error was %s.\n", strerror(errno) ));
-                       blocking_lock_reply_error(blr,NT_STATUS_ACCESS_DENIED);
-                       DLIST_REMOVE(blocking_lock_queue, blr);
-                       free_blocking_lock_record(blr);
-                       recalc_timeout = True;
-                       change_to_root_user();
-                       continue;
-               }
-
                /*
                 * Go through the remaining locks and try and obtain them.
                 * The call returns True if all locks were obtained successfully
                 * and False if we still need to wait.
                 */
 
+               DEBUG(10, ("Processing BLR = %p\n", blr));
+
+               /* We use set_current_service so connections with
+                * pending locks are not marked as idle.
+                */
+
+               set_current_service(blr->fsp->conn,
+                               SVAL(blr->req->inbuf,smb_flg),
+                               false);
+
                if(blocking_lock_record_process(blr)) {
-                       struct byte_range_lock *br_lck = brl_get_locks(NULL, fsp);
+                       struct byte_range_lock *br_lck = brl_get_locks(
+                               talloc_tos(), blr->fsp);
+
+                       DEBUG(10, ("BLR_process returned true: cancelling and "
+                           "removing lock. BLR = %p\n", blr));
 
                        if (br_lck) {
                                brl_lock_cancel(br_lck,
-                                       blr->lock_pid,
-                                       procid_self(),
+                                       blr->smblctx,
+                                       sconn_server_id(sconn),
                                        blr->offset,
                                        blr->count,
-                                       blr->lock_flav);
+                                       blr->lock_flav,
+                                       blr);
                                TALLOC_FREE(br_lck);
                        }
 
-                       DLIST_REMOVE(blocking_lock_queue, blr);
-                       free_blocking_lock_record(blr);
-                       recalc_timeout = True;
-                       change_to_root_user();
+                       DLIST_REMOVE(sconn->smb1.locks.blocking_lock_queue, blr);
+                       TALLOC_FREE(blr);
                        continue;
                }
 
-               change_to_root_user();
-
                /*
                 * We couldn't get the locks for this record on the list.
                 * If the time has expired, return a lock error.
                 */
 
                if (!timeval_is_zero(&blr->expire_time) && timeval_compare(&blr->expire_time, &tv_curr) <= 0) {
-                       struct byte_range_lock *br_lck = brl_get_locks(NULL, fsp);
+                       struct byte_range_lock *br_lck = brl_get_locks(
+                               talloc_tos(), blr->fsp);
+
+                       DEBUG(10, ("Lock timed out! BLR = %p\n", blr));
 
                        /*
                         * Lock expired - throw away all previously
@@ -808,35 +793,35 @@ static void process_blocking_lock_queue(void)
                         */
 
                        if (br_lck) {
-                               DEBUG(5,("process_blocking_lock_queue: pending lock fnum = %d for file %s timed out.\n",
-                                       fsp->fnum, fsp->fsp_name ));
+                               DEBUG(5,("process_blocking_lock_queue: "
+                                        "pending lock fnum = %d for file %s "
+                                        "timed out.\n", blr->fsp->fnum,
+                                        fsp_str_dbg(blr->fsp)));
 
                                brl_lock_cancel(br_lck,
-                                       blr->lock_pid,
-                                       procid_self(),
+                                       blr->smblctx,
+                                       sconn_server_id(sconn),
                                        blr->offset,
                                        blr->count,
-                                       blr->lock_flav);
+                                       blr->lock_flav,
+                                       blr);
                                TALLOC_FREE(br_lck);
                        }
 
                        blocking_lock_reply_error(blr,NT_STATUS_FILE_LOCK_CONFLICT);
-                       DLIST_REMOVE(blocking_lock_queue, blr);
-                       free_blocking_lock_record(blr);
-                       recalc_timeout = True;
+                       DLIST_REMOVE(sconn->smb1.locks.blocking_lock_queue, blr);
+                       TALLOC_FREE(blr);
                }
        }
 
-       if (recalc_timeout) {
-               recalc_brl_timeout();
-       }
+       recalc_brl_timeout(sconn);
 }
 
 /****************************************************************************
  Handle a cancel message. Lock already moved onto the cancel queue.
 *****************************************************************************/
 
-#define MSG_BLOCKING_LOCK_CANCEL_SIZE (sizeof(blocking_lock_record *) + sizeof(NTSTATUS))
+#define MSG_BLOCKING_LOCK_CANCEL_SIZE (sizeof(struct blocking_lock_record *) + sizeof(NTSTATUS))
 
 static void process_blocking_lock_cancel_message(struct messaging_context *ctx,
                                                 void *private_data,
@@ -844,9 +829,10 @@ static void process_blocking_lock_cancel_message(struct messaging_context *ctx,
                                                 struct server_id server_id,
                                                 DATA_BLOB *data)
 {
+       struct smbd_server_connection *sconn;
        NTSTATUS err;
        const char *msg = (const char *)data->data;
-       blocking_lock_record *blr;
+       struct blocking_lock_record *blr;
 
        if (data->data == NULL) {
                smb_panic("process_blocking_lock_cancel_message: null msg");
@@ -858,6 +844,12 @@ static void process_blocking_lock_cancel_message(struct messaging_context *ctx,
                smb_panic("process_blocking_lock_cancel_message: bad msg");
         }
 
+       sconn = msg_ctx_to_sconn(ctx);
+       if (sconn == NULL) {
+               DEBUG(1, ("could not find sconn\n"));
+               return;
+       }
+
        memcpy(&blr, msg, sizeof(blr));
        memcpy(&err, &msg[sizeof(blr)], sizeof(NTSTATUS));
 
@@ -865,38 +857,40 @@ static void process_blocking_lock_cancel_message(struct messaging_context *ctx,
                nt_errstr(err) ));
 
        blocking_lock_reply_error(blr, err);
-       DLIST_REMOVE(blocking_lock_cancelled_queue, blr);
-       free_blocking_lock_record(blr);
+       DLIST_REMOVE(sconn->smb1.locks.blocking_lock_cancelled_queue, blr);
+       TALLOC_FREE(blr);
 }
 
 /****************************************************************************
  Send ourselves a blocking lock cancelled message. Handled asynchronously above.
+ Returns the blocking_lock_record that is being cancelled.
+ Only called from the SMB1 code.
 *****************************************************************************/
 
-bool blocking_lock_cancel(files_struct *fsp,
-                       uint32 lock_pid,
-                       SMB_BIG_UINT offset,
-                       SMB_BIG_UINT count,
+struct blocking_lock_record *blocking_lock_cancel_smb1(files_struct *fsp,
+                       uint64_t smblctx,
+                       uint64_t offset,
+                       uint64_t count,
                        enum brl_flavour lock_flav,
                        unsigned char locktype,
                         NTSTATUS err)
 {
-       static bool initialized;
+       struct smbd_server_connection *sconn = fsp->conn->sconn;
        char msg[MSG_BLOCKING_LOCK_CANCEL_SIZE];
-       blocking_lock_record *blr;
+       struct blocking_lock_record *blr;
 
-       if (!initialized) {
+       if (!sconn->smb1.locks.blocking_lock_cancel_state) {
                /* Register our message. */
-               messaging_register(smbd_messaging_context(), NULL,
+               messaging_register(sconn->msg_ctx, NULL,
                                   MSG_SMB_BLOCKING_LOCK_CANCEL,
                                   process_blocking_lock_cancel_message);
 
-               initialized = True;
+               sconn->smb1.locks.blocking_lock_cancel_state = True;
        }
 
-       for (blr = blocking_lock_queue; blr; blr = blr->next) {
+       for (blr = sconn->smb1.locks.blocking_lock_queue; blr; blr = blr->next) {
                if (fsp == blr->fsp &&
-                               lock_pid == blr->lock_pid &&
+                               smblctx == blr->smblctx &&
                                offset == blr->offset &&
                                count == blr->count &&
                                lock_flav == blr->lock_flav) {
@@ -905,27 +899,27 @@ bool blocking_lock_cancel(files_struct *fsp,
        }
 
        if (!blr) {
-               return False;
+               return NULL;
        }
 
        /* Check the flags are right. */
-       if (blr->com_type == SMBlockingX &&
+       if (blr->req->cmd == SMBlockingX &&
                (locktype & LOCKING_ANDX_LARGE_FILES) !=
-                       (CVAL(blr->inbuf,smb_vwv3) & LOCKING_ANDX_LARGE_FILES)) {
-               return False;
+                       (CVAL(blr->req->vwv+3, 0) & LOCKING_ANDX_LARGE_FILES)) {
+               return NULL;
        }
 
        /* Move to cancelled queue. */
-       DLIST_REMOVE(blocking_lock_queue, blr);
-       DLIST_ADD(blocking_lock_cancelled_queue, blr);
+       DLIST_REMOVE(sconn->smb1.locks.blocking_lock_queue, blr);
+       DLIST_ADD(sconn->smb1.locks.blocking_lock_cancelled_queue, blr);
 
        /* Create the message. */
        memcpy(msg, &blr, sizeof(blr));
        memcpy(&msg[sizeof(blr)], &err, sizeof(NTSTATUS));
 
-       messaging_send_buf(smbd_messaging_context(), procid_self(),
+       messaging_send_buf(sconn->msg_ctx, sconn_server_id(sconn),
                           MSG_SMB_BLOCKING_LOCK_CANCEL,
                           (uint8 *)&msg, sizeof(msg));
 
-       return True;
+       return blr;
 }