smbd: Don't pass lease_idx down to set_share_mode()
[gd/samba-autobuild/.git] / source3 / locking / locking.c
index f96887e683239140954741f2d56666bb72cd3407..ddae920398683f6bb7135e16476723a9b1bda494 100644 (file)
@@ -37,6 +37,7 @@
 
 #include "includes.h"
 #include "system/filesys.h"
+#include "lib/util/server_id.h"
 #include "locking/proto.h"
 #include "smbd/globals.h"
 #include "dbwrap/dbwrap.h"
@@ -46,6 +47,7 @@
 #include "messages.h"
 #include "util_tdb.h"
 #include "../librpc/gen_ndr/ndr_open_files.h"
+#include "librpc/gen_ndr/ndr_file_id.h"
 #include "locking/leases_db.h"
 
 #undef DBGC_CLASS
@@ -102,7 +104,7 @@ void init_strict_lock_struct(files_struct *fsp,
         plock->lock_flav = lp_posix_cifsu_locktype(fsp);
 }
 
-bool strict_lock_default(files_struct *fsp, struct lock_struct *plock)
+bool strict_lock_check_default(files_struct *fsp, struct lock_struct *plock)
 {
        struct byte_range_lock *br_lck;
        int strict_locking = lp_strict_locking(fsp->conn->params);
@@ -117,17 +119,21 @@ bool strict_lock_default(files_struct *fsp, struct lock_struct *plock)
        }
 
        if (strict_locking == Auto) {
-               if  (EXCLUSIVE_OPLOCK_TYPE(fsp->oplock_type) &&
-                    (plock->lock_type == READ_LOCK ||
-                     plock->lock_type == WRITE_LOCK)) {
-                       DEBUG(10, ("is_locked: optimisation - exclusive oplock "
-                                  "on file %s\n", fsp_str_dbg(fsp)));
+               uint32_t lease_type = fsp_lease_type(fsp);
+
+               if ((lease_type & SMB2_LEASE_READ) &&
+                    (plock->lock_type == READ_LOCK))
+               {
+                       DBG_DEBUG("optimisation - read lease on file %s\n",
+                                 fsp_str_dbg(fsp));
                        return true;
                }
-               if ((fsp->oplock_type == LEVEL_II_OPLOCK) &&
-                   (plock->lock_type == READ_LOCK)) {
-                       DEBUG(10, ("is_locked: optimisation - level II oplock "
-                                  "on file %s\n", fsp_str_dbg(fsp)));
+
+               if ((lease_type & SMB2_LEASE_WRITE) &&
+                    (plock->lock_type == WRITE_LOCK))
+               {
+                       DBG_DEBUG("optimisation - write lease on file %s\n",
+                                 fsp_str_dbg(fsp));
                        return true;
                }
        }
@@ -144,6 +150,9 @@ bool strict_lock_default(files_struct *fsp, struct lock_struct *plock)
                 * autocleanup. This is the slow path anyway.
                 */
                br_lck = brl_get_locks(talloc_tos(), fsp);
+               if (br_lck == NULL) {
+                       return true;
+               }
                ret = brl_locktest(br_lck, plock);
                TALLOC_FREE(br_lck);
        }
@@ -158,10 +167,6 @@ bool strict_lock_default(files_struct *fsp, struct lock_struct *plock)
        return ret;
 }
 
-void strict_unlock_default(files_struct *fsp, struct lock_struct *plock)
-{
-}
-
 /****************************************************************************
  Find out if a lock could be granted - return who is blocking us if we can't.
 ****************************************************************************/
@@ -344,7 +349,7 @@ NTSTATUS do_unlock(struct messaging_context *msg_ctx,
 ****************************************************************************/
 
 NTSTATUS do_lock_cancel(files_struct *fsp,
-                       uint64 smblctx,
+                       uint64_t smblctx,
                        uint64_t count,
                        uint64_t offset,
                        enum brl_flavour lock_flav)
@@ -423,19 +428,23 @@ void locking_close_file(struct messaging_context *msg_ctx,
  Print out a share mode.
 ********************************************************************/
 
-char *share_mode_str(TALLOC_CTX *ctx, int num, const struct share_mode_entry *e)
+char *share_mode_str(TALLOC_CTX *ctx, int num,
+                    const struct file_id *id,
+                    const struct share_mode_entry *e)
 {
+       struct server_id_buf tmp;
+
        return talloc_asprintf(ctx, "share_mode_entry[%d]: "
                 "pid = %s, share_access = 0x%x, private_options = 0x%x, "
                 "access_mask = 0x%x, mid = 0x%llx, type= 0x%x, gen_id = %llu, "
                 "uid = %u, flags = %u, file_id %s, name_hash = 0x%x",
                 num,
-                procid_str_static(&e->pid),
+                server_id_str_buf(e->pid, &tmp),
                 e->share_access, e->private_options,
                 e->access_mask, (unsigned long long)e->op_mid,
                 e->op_type, (unsigned long long)e->share_file_id,
                 (unsigned int)e->uid, (unsigned int)e->flags,
-                file_id_string_tos(&e->id),
+                file_id_string_tos(id),
                 (unsigned int)e->name_hash);
 }
 
@@ -471,7 +480,7 @@ bool rename_share_filename(struct messaging_context *msg_ctx,
        size_t sn_len;
        size_t msg_len;
        char *frm = NULL;
-       int i;
+       uint32_t i;
        bool strip_two_chars = false;
        bool has_stream = smb_fname_dst->stream_name != NULL;
        struct server_id self_pid = messaging_server_id(msg_ctx);
@@ -532,6 +541,8 @@ bool rename_share_filename(struct messaging_context *msg_ctx,
        /* Send the messages. */
        for (i=0; i<d->num_share_modes; i++) {
                struct share_mode_entry *se = &d->share_modes[i];
+               struct server_id_buf tmp;
+
                if (!is_valid_share_mode_entry(se)) {
                        continue;
                }
@@ -556,13 +567,37 @@ bool rename_share_filename(struct messaging_context *msg_ctx,
                DEBUG(10,("rename_share_filename: sending rename message to "
                          "pid %s file_id %s sharepath %s base_name %s "
                          "stream_name %s\n",
-                         procid_str_static(&se->pid),
+                         server_id_str_buf(se->pid, &tmp),
                          file_id_string_tos(&id),
                          d->servicepath, d->base_name,
                        has_stream ? d->stream_name : ""));
 
                messaging_send_buf(msg_ctx, se->pid, MSG_SMB_FILE_RENAME,
-                                  (uint8 *)frm, msg_len);
+                                  (uint8_t *)frm, msg_len);
+       }
+
+       for (i=0; i<d->num_leases; i++) {
+               /* Update the filename in leases_db. */
+               NTSTATUS status;
+               struct share_mode_lease *l;
+
+               l = &d->leases[i];
+
+               status = leases_db_rename(&l->client_guid,
+                                       &l->lease_key,
+                                       &id,
+                                       d->servicepath,
+                                       d->base_name,
+                                       d->stream_name);
+               if (!NT_STATUS_IS_OK(status)) {
+                       /* Any error recovery possible here ? */
+                       DEBUG(1,("Failed to rename lease key for "
+                               "renamed file %s:%s. %s\n",
+                               d->base_name,
+                               d->stream_name,
+                               nt_errstr(status)));
+                       continue;
+               }
        }
 
        return True;
@@ -625,8 +660,6 @@ bool is_valid_share_mode_entry(const struct share_mode_entry *e)
 static void remove_share_mode_lease(struct share_mode_data *d,
                                    struct share_mode_entry *e)
 {
-       struct GUID client_guid;
-       struct smb2_lease_key lease_key;
        uint16_t op_type;
        uint32_t lease_idx;
        uint32_t i;
@@ -649,14 +682,19 @@ static void remove_share_mode_lease(struct share_mode_data *d,
        e->lease_idx = UINT32_MAX;
 
        for (i=0; i<d->num_share_modes; i++) {
-               if (d->share_modes[i].stale) {
+               struct share_mode_entry *e2 = &d->share_modes[i];
+
+               if (e2->stale) {
                        continue;
                }
-               if (e == &d->share_modes[i]) {
+               if (e == e2) {
                        /* Not ourselves. */
                        continue;
                }
-               if (d->share_modes[i].lease_idx == lease_idx) {
+               if (smb2_lease_equal(&e->client_guid,
+                                    &e->lease_key,
+                                    &e2->client_guid,
+                                    &e2->lease_key)) {
                        break;
                }
        }
@@ -667,11 +705,6 @@ static void remove_share_mode_lease(struct share_mode_data *d,
                return;
        }
 
-       memcpy(&client_guid,
-               &d->leases[lease_idx].client_guid,
-               sizeof(client_guid));
-       lease_key = d->leases[lease_idx].lease_key;
-
        d->num_leases -= 1;
        d->leases[lease_idx] = d->leases[d->num_leases];
 
@@ -681,16 +714,15 @@ static void remove_share_mode_lease(struct share_mode_data *d,
        for (i=0; i<d->num_share_modes; i++) {
                if (d->share_modes[i].lease_idx == d->num_leases) {
                        d->share_modes[i].lease_idx = lease_idx;
-                       d->share_modes[i].lease = &d->leases[lease_idx];
                }
        }
 
        {
                NTSTATUS status;
 
-               status = leases_db_del(&client_guid,
-                                       &lease_key,
-                                       &e->id);
+               status = leases_db_del(&e->client_guid,
+                                      &e->lease_key,
+                                      &d->id);
 
                DEBUG(10, ("%s: leases_db_del returned %s\n", __func__,
                           nt_errstr(status)));
@@ -704,11 +736,14 @@ static void remove_share_mode_lease(struct share_mode_data *d,
  */
 bool share_mode_stale_pid(struct share_mode_data *d, uint32_t idx)
 {
+       struct server_id_buf tmp;
        struct share_mode_entry *e;
 
        if (idx > d->num_share_modes) {
-               DEBUG(1, ("Asking for index %u, only %u around\n",
-                         idx, (unsigned)d->num_share_modes));
+               DBG_WARNING("Asking for index %"PRIu32", "
+                           "only %"PRIu32" around\n",
+                           idx,
+                           d->num_share_modes);
                return false;
        }
        e = &d->share_modes[idx];
@@ -719,36 +754,36 @@ bool share_mode_stale_pid(struct share_mode_data *d, uint32_t idx)
                return true;
        }
        if (serverid_exists(&e->pid)) {
-               DEBUG(10, ("PID %s (index %u out of %u) still exists\n",
-                          procid_str_static(&e->pid), idx,
-                          (unsigned)d->num_share_modes));
+               DBG_DEBUG("PID %s (index %"PRIu32" out of %"PRIu32") "
+                         "still exists\n",
+                         server_id_str_buf(e->pid, &tmp),
+                         idx,
+                         d->num_share_modes);
                return false;
        }
-       DEBUG(10, ("PID %s (index %u out of %u) does not exist anymore\n",
-                  procid_str_static(&e->pid), idx,
-                  (unsigned)d->num_share_modes));
+       DBG_DEBUG("PID %s (index %"PRIu32" out of %"PRIu32") "
+                 "does not exist anymore\n",
+                 server_id_str_buf(e->pid, &tmp),
+                 idx,
+                 d->num_share_modes);
 
        e->stale = true;
 
        if (d->num_delete_tokens != 0) {
-               uint32_t i, num_stale;
-
-               /*
-                * We cannot have any delete tokens
-                * if there are no valid share modes.
-                */
-
-               num_stale = 0;
+               uint32_t i;
 
                for (i=0; i<d->num_share_modes; i++) {
-                       if (d->share_modes[i].stale) {
-                               num_stale += 1;
+                       bool valid = !d->share_modes[i].stale;
+                       if (valid) {
+                               break;
                        }
                }
 
-               if (num_stale == d->num_share_modes) {
+               if (i == d->num_share_modes) {
                        /*
-                        * No non-stale share mode found
+                        * No valid (non-stale) share mode found, all
+                        * who might have set the delete token are
+                        * gone.
                         */
                        TALLOC_FREE(d->delete_tokens);
                        d->num_delete_tokens = 0;
@@ -771,27 +806,22 @@ void remove_stale_share_mode_entries(struct share_mode_data *d)
                        struct share_mode_entry *m = d->share_modes;
                        m[i] = m[d->num_share_modes-1];
                        d->num_share_modes -= 1;
-               } else {
-                       i += 1;
+                       continue;
                }
+               i += 1;
        }
 }
 
-bool set_share_mode(struct share_mode_lock *lck, struct files_struct *fsp,
-                   uid_t uid, uint64_t mid, uint16_t op_type,
-                   uint32_t lease_idx)
+bool set_share_mode(struct share_mode_lock *lck,
+                   struct files_struct *fsp,
+                   uid_t uid,
+                   uint64_t mid,
+                   uint16_t op_type,
+                   const struct GUID *client_guid,
+                   const struct smb2_lease_key *lease_key)
 {
        struct share_mode_data *d = lck->data;
        struct share_mode_entry *tmp, *e;
-       struct share_mode_lease *lease = NULL;
-
-       if (lease_idx == UINT32_MAX) {
-               lease = NULL;
-       } else if (lease_idx >= d->num_leases) {
-               return false;
-       } else {
-               lease = &d->leases[lease_idx];
-       }
 
        tmp = talloc_realloc(d, d->share_modes, struct share_mode_entry,
                             d->num_share_modes+1);
@@ -810,25 +840,58 @@ bool set_share_mode(struct share_mode_lock *lck, struct files_struct *fsp,
        e->access_mask = fsp->access_mask;
        e->op_mid = mid;
        e->op_type = op_type;
-       e->lease_idx = lease_idx;
-       e->lease = lease;
+
+       if (op_type == LEASE_OPLOCK) {
+               uint32_t i;
+
+               e->client_guid = *client_guid;
+               e->lease_key = *lease_key;
+
+               /*
+                * Need to set lease_idx. This is essentially
+                * find_share_mode_lease(), but that will go away
+                * soon. So don't add the dependency here.
+                */
+
+               for (i=0; i<d->num_leases; i++) {
+                       struct share_mode_lease *l = &d->leases[i];
+
+                       if (smb2_lease_equal(client_guid,
+                                            lease_key,
+                                            &l->client_guid,
+                                            &l->lease_key)) {
+                               break;
+                       }
+               }
+
+               if (i == d->num_leases) {
+                       DBG_WARNING("lease not found\n");
+                       d->num_share_modes -= 1;
+                       return false;
+               }
+
+               e->lease_idx = i;
+       } else {
+               e->lease_idx = UINT32_MAX;
+       }
+
        e->time.tv_sec = fsp->open_time.tv_sec;
        e->time.tv_usec = fsp->open_time.tv_usec;
-       e->id = fsp->file_id;
        e->share_file_id = fsp->fh->gen_id;
-       e->uid = (uint32)uid;
-       e->flags = fsp->posix_open ? SHARE_MODE_FLAG_POSIX_OPEN : 0;
+       e->uid = (uint32_t)uid;
+       e->flags = (fsp->posix_flags & FSP_POSIX_FLAGS_OPEN) ?
+               SHARE_MODE_FLAG_POSIX_OPEN : 0;
        e->name_hash = fsp->name_hash;
 
        return true;
 }
 
-static struct share_mode_entry *find_share_mode_entry(
+struct share_mode_entry *find_share_mode_entry(
        struct share_mode_lock *lck, files_struct *fsp)
 {
        struct share_mode_data *d = lck->data;
        struct server_id pid;
-       int i;
+       uint32_t i;
 
        pid = messaging_server_id(fsp->conn->sconn->msg_ctx);
 
@@ -841,9 +904,6 @@ static struct share_mode_entry *find_share_mode_entry(
                if (!serverid_equal(&pid, &e->pid)) {
                        continue;
                }
-               if (!file_id_equal(&fsp->file_id, &e->id)) {
-                       continue;
-               }
                if (fsp->fh->gen_id != e->share_file_id) {
                        continue;
                }
@@ -853,8 +913,7 @@ static struct share_mode_entry *find_share_mode_entry(
 }
 
 /*******************************************************************
- Del the share mode of a file for this process. Return the number of
- entries left.
+ Del the share mode of a file for this process.
 ********************************************************************/
 
 bool del_share_mode(struct share_mode_lock *lck, files_struct *fsp)
@@ -977,6 +1036,27 @@ static bool add_delete_on_close_token(struct share_mode_data *d,
        return true;
 }
 
+void reset_delete_on_close_lck(files_struct *fsp,
+                              struct share_mode_lock *lck)
+{
+       struct share_mode_data *d = lck->data;
+       uint32_t i;
+
+       for (i=0; i<d->num_delete_tokens; i++) {
+               struct delete_token *dt = &d->delete_tokens[i];
+
+               if (dt->name_hash == fsp->name_hash) {
+                       d->modified = true;
+
+                       /* Delete this entry. */
+                       TALLOC_FREE(dt->delete_nt_token);
+                       TALLOC_FREE(dt->delete_token);
+                       *dt = d->delete_tokens[d->num_delete_tokens-1];
+                       d->num_delete_tokens -= 1;
+               }
+       }
+}
+
 /****************************************************************************
  Sets the delete on close flag over all share modes on this file.
  Modify the share mode entry for all files open
@@ -990,54 +1070,63 @@ static bool add_delete_on_close_token(struct share_mode_data *d,
 
 void set_delete_on_close_lck(files_struct *fsp,
                        struct share_mode_lock *lck,
-                       bool delete_on_close,
                        const struct security_token *nt_tok,
                        const struct security_unix_token *tok)
 {
+       struct messaging_context *msg_ctx = fsp->conn->sconn->msg_ctx;
        struct share_mode_data *d = lck->data;
-       int i;
+       uint32_t i;
        bool ret;
+       DATA_BLOB fid_blob = {};
+       enum ndr_err_code ndr_err;
 
-       if (delete_on_close) {
-               SMB_ASSERT(nt_tok != NULL);
-               SMB_ASSERT(tok != NULL);
-       } else {
-               SMB_ASSERT(nt_tok == NULL);
-               SMB_ASSERT(tok == NULL);
-       }
+       SMB_ASSERT(nt_tok != NULL);
+       SMB_ASSERT(tok != NULL);
 
        for (i=0; i<d->num_delete_tokens; i++) {
                struct delete_token *dt = &d->delete_tokens[i];
                if (dt->name_hash == fsp->name_hash) {
                        d->modified = true;
-                       if (delete_on_close == false) {
-                               /* Delete this entry. */
-                               TALLOC_FREE(dt->delete_nt_token);
-                               TALLOC_FREE(dt->delete_token);
-                               *dt = d->delete_tokens[
-                                       d->num_delete_tokens-1];
-                               d->num_delete_tokens -= 1;
-                       } else {
-                               /* Replace this token with the
-                                  given tok. */
-                               TALLOC_FREE(dt->delete_nt_token);
-                               dt->delete_nt_token = dup_nt_token(dt, nt_tok);
-                               SMB_ASSERT(dt->delete_nt_token != NULL);
-                               TALLOC_FREE(dt->delete_token);
-                               dt->delete_token = copy_unix_token(dt, tok);
-                               SMB_ASSERT(dt->delete_token != NULL);
-                       }
+
+                       /* Replace this token with the given tok. */
+                       TALLOC_FREE(dt->delete_nt_token);
+                       dt->delete_nt_token = dup_nt_token(dt, nt_tok);
+                       SMB_ASSERT(dt->delete_nt_token != NULL);
+                       TALLOC_FREE(dt->delete_token);
+                       dt->delete_token = copy_unix_token(dt, tok);
+                       SMB_ASSERT(dt->delete_token != NULL);
+
                        return;
                }
        }
 
-       if (!delete_on_close) {
-               /* Nothing to delete - not found. */
-               return;
-       }
-
        ret = add_delete_on_close_token(lck->data, fsp->name_hash, nt_tok, tok);
        SMB_ASSERT(ret);
+
+       ndr_err = ndr_push_struct_blob(&fid_blob, talloc_tos(), &fsp->file_id,
+                                      (ndr_push_flags_fn_t)ndr_push_file_id);
+       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+               DEBUG(10, ("ndr_push_file_id failed: %s\n",
+                          ndr_errstr(ndr_err)));
+       }
+
+       for (i=0; i<d->num_share_modes; i++) {
+               struct share_mode_entry *e = &d->share_modes[i];
+               NTSTATUS status;
+
+               status = messaging_send(
+                       msg_ctx, e->pid, MSG_SMB_NOTIFY_CANCEL_DELETED,
+                       &fid_blob);
+
+               if (!NT_STATUS_IS_OK(status)) {
+                       struct server_id_buf tmp;
+                       DEBUG(10, ("%s: messaging_send to %s returned %s\n",
+                                  __func__, server_id_str_buf(e->pid, &tmp),
+                                  nt_errstr(status)));
+               }
+       }
+
+       TALLOC_FREE(fid_blob.data);
 }
 
 bool set_delete_on_close(files_struct *fsp, bool delete_on_close,
@@ -1057,13 +1146,9 @@ bool set_delete_on_close(files_struct *fsp, bool delete_on_close,
        }
 
        if (delete_on_close) {
-               set_delete_on_close_lck(fsp, lck, true,
-                       nt_tok,
-                       tok);
+               set_delete_on_close_lck(fsp, lck, nt_tok, tok);
        } else {
-               set_delete_on_close_lck(fsp, lck, false,
-                       NULL,
-                       NULL);
+               reset_delete_on_close_lck(fsp, lck);
        }
 
        if (fsp->is_directory) {
@@ -1180,3 +1265,34 @@ struct timespec get_share_mode_write_time(struct share_mode_lock *lck)
        }
        return d->old_write_time;
 }
+
+bool file_has_open_streams(files_struct *fsp)
+{
+       struct share_mode_lock *lock = NULL;
+       struct share_mode_data *d = NULL;
+       uint32_t i;
+
+       lock = get_existing_share_mode_lock(talloc_tos(), fsp->file_id);
+       if (lock == NULL) {
+               return false;
+       }
+       d = lock->data;
+
+       for (i = 0; i < d->num_share_modes; i++) {
+               struct share_mode_entry *e = &d->share_modes[i];
+
+               if (share_mode_stale_pid(d, i)) {
+                       continue;
+               }
+
+               if (e->private_options &
+                   NTCREATEX_OPTIONS_PRIVATE_STREAM_BASEOPEN)
+               {
+                       TALLOC_FREE(lock);
+                       return true;
+               }
+       }
+
+       TALLOC_FREE(lock);
+       return false;
+}