138c8a9a5SSteve French // SPDX-License-Identifier: GPL-2.0-or-later 238c8a9a5SSteve French /* 338c8a9a5SSteve French * Copyright (C) 2016 Namjae Jeon <linkinjeon@kernel.org> 438c8a9a5SSteve French * Copyright (C) 2018 Samsung Electronics Co., Ltd. 538c8a9a5SSteve French */ 638c8a9a5SSteve French 738c8a9a5SSteve French #include <linux/moduleparam.h> 838c8a9a5SSteve French 938c8a9a5SSteve French #include "glob.h" 1038c8a9a5SSteve French #include "oplock.h" 1138c8a9a5SSteve French 1238c8a9a5SSteve French #include "smb_common.h" 1338c8a9a5SSteve French #include "smbstatus.h" 1438c8a9a5SSteve French #include "connection.h" 1538c8a9a5SSteve French #include "mgmt/user_session.h" 1638c8a9a5SSteve French #include "mgmt/share_config.h" 1738c8a9a5SSteve French #include "mgmt/tree_connect.h" 1838c8a9a5SSteve French 1938c8a9a5SSteve French static LIST_HEAD(lease_table_list); 2038c8a9a5SSteve French static DEFINE_RWLOCK(lease_list_lock); 2138c8a9a5SSteve French 2238c8a9a5SSteve French /** 2338c8a9a5SSteve French * alloc_opinfo() - allocate a new opinfo object for oplock info 2438c8a9a5SSteve French * @work: smb work 2538c8a9a5SSteve French * @id: fid of open file 2638c8a9a5SSteve French * @Tid: tree id of connection 2738c8a9a5SSteve French * 2838c8a9a5SSteve French * Return: allocated opinfo object on success, otherwise NULL 2938c8a9a5SSteve French */ 3038c8a9a5SSteve French static struct oplock_info *alloc_opinfo(struct ksmbd_work *work, 3138c8a9a5SSteve French u64 id, __u16 Tid) 3238c8a9a5SSteve French { 3338c8a9a5SSteve French struct ksmbd_conn *conn = work->conn; 3438c8a9a5SSteve French struct ksmbd_session *sess = work->sess; 3538c8a9a5SSteve French struct oplock_info *opinfo; 3638c8a9a5SSteve French 3738c8a9a5SSteve French opinfo = kzalloc(sizeof(struct oplock_info), GFP_KERNEL); 3838c8a9a5SSteve French if (!opinfo) 3938c8a9a5SSteve French return NULL; 4038c8a9a5SSteve French 4138c8a9a5SSteve French opinfo->sess = sess; 4238c8a9a5SSteve French opinfo->conn = conn; 4338c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_NONE; 4438c8a9a5SSteve French opinfo->op_state = OPLOCK_STATE_NONE; 4538c8a9a5SSteve French opinfo->pending_break = 0; 4638c8a9a5SSteve French opinfo->fid = id; 4738c8a9a5SSteve French opinfo->Tid = Tid; 4838c8a9a5SSteve French INIT_LIST_HEAD(&opinfo->op_entry); 4938c8a9a5SSteve French INIT_LIST_HEAD(&opinfo->interim_list); 5038c8a9a5SSteve French init_waitqueue_head(&opinfo->oplock_q); 5138c8a9a5SSteve French init_waitqueue_head(&opinfo->oplock_brk); 5238c8a9a5SSteve French atomic_set(&opinfo->refcount, 1); 5338c8a9a5SSteve French atomic_set(&opinfo->breaking_cnt, 0); 5438c8a9a5SSteve French 5538c8a9a5SSteve French return opinfo; 5638c8a9a5SSteve French } 5738c8a9a5SSteve French 5838c8a9a5SSteve French static void lease_add_list(struct oplock_info *opinfo) 5938c8a9a5SSteve French { 6038c8a9a5SSteve French struct lease_table *lb = opinfo->o_lease->l_lb; 6138c8a9a5SSteve French 6238c8a9a5SSteve French spin_lock(&lb->lb_lock); 6338c8a9a5SSteve French list_add_rcu(&opinfo->lease_entry, &lb->lease_list); 6438c8a9a5SSteve French spin_unlock(&lb->lb_lock); 6538c8a9a5SSteve French } 6638c8a9a5SSteve French 6738c8a9a5SSteve French static void lease_del_list(struct oplock_info *opinfo) 6838c8a9a5SSteve French { 6938c8a9a5SSteve French struct lease_table *lb = opinfo->o_lease->l_lb; 7038c8a9a5SSteve French 7138c8a9a5SSteve French if (!lb) 7238c8a9a5SSteve French return; 7338c8a9a5SSteve French 7438c8a9a5SSteve French spin_lock(&lb->lb_lock); 7538c8a9a5SSteve French if (list_empty(&opinfo->lease_entry)) { 7638c8a9a5SSteve French spin_unlock(&lb->lb_lock); 7738c8a9a5SSteve French return; 7838c8a9a5SSteve French } 7938c8a9a5SSteve French 8038c8a9a5SSteve French list_del_init(&opinfo->lease_entry); 8138c8a9a5SSteve French opinfo->o_lease->l_lb = NULL; 8238c8a9a5SSteve French spin_unlock(&lb->lb_lock); 8338c8a9a5SSteve French } 8438c8a9a5SSteve French 8538c8a9a5SSteve French static void lb_add(struct lease_table *lb) 8638c8a9a5SSteve French { 8738c8a9a5SSteve French write_lock(&lease_list_lock); 8838c8a9a5SSteve French list_add(&lb->l_entry, &lease_table_list); 8938c8a9a5SSteve French write_unlock(&lease_list_lock); 9038c8a9a5SSteve French } 9138c8a9a5SSteve French 9238c8a9a5SSteve French static int alloc_lease(struct oplock_info *opinfo, struct lease_ctx_info *lctx) 9338c8a9a5SSteve French { 9438c8a9a5SSteve French struct lease *lease; 9538c8a9a5SSteve French 9638c8a9a5SSteve French lease = kmalloc(sizeof(struct lease), GFP_KERNEL); 9738c8a9a5SSteve French if (!lease) 9838c8a9a5SSteve French return -ENOMEM; 9938c8a9a5SSteve French 10038c8a9a5SSteve French memcpy(lease->lease_key, lctx->lease_key, SMB2_LEASE_KEY_SIZE); 10138c8a9a5SSteve French lease->state = lctx->req_state; 10238c8a9a5SSteve French lease->new_state = 0; 10338c8a9a5SSteve French lease->flags = lctx->flags; 10438c8a9a5SSteve French lease->duration = lctx->duration; 1053c1e602aSNamjae Jeon lease->is_dir = lctx->is_dir; 10638c8a9a5SSteve French memcpy(lease->parent_lease_key, lctx->parent_lease_key, SMB2_LEASE_KEY_SIZE); 10738c8a9a5SSteve French lease->version = lctx->version; 108bc025d49SNamjae Jeon lease->epoch = le16_to_cpu(lctx->epoch); 10938c8a9a5SSteve French INIT_LIST_HEAD(&opinfo->lease_entry); 11038c8a9a5SSteve French opinfo->o_lease = lease; 11138c8a9a5SSteve French 11238c8a9a5SSteve French return 0; 11338c8a9a5SSteve French } 11438c8a9a5SSteve French 11538c8a9a5SSteve French static void free_lease(struct oplock_info *opinfo) 11638c8a9a5SSteve French { 11738c8a9a5SSteve French struct lease *lease; 11838c8a9a5SSteve French 11938c8a9a5SSteve French lease = opinfo->o_lease; 12038c8a9a5SSteve French kfree(lease); 12138c8a9a5SSteve French } 12238c8a9a5SSteve French 12338c8a9a5SSteve French static void free_opinfo(struct oplock_info *opinfo) 12438c8a9a5SSteve French { 12538c8a9a5SSteve French if (opinfo->is_lease) 12638c8a9a5SSteve French free_lease(opinfo); 12738c8a9a5SSteve French kfree(opinfo); 12838c8a9a5SSteve French } 12938c8a9a5SSteve French 13038c8a9a5SSteve French static inline void opinfo_free_rcu(struct rcu_head *rcu_head) 13138c8a9a5SSteve French { 13238c8a9a5SSteve French struct oplock_info *opinfo; 13338c8a9a5SSteve French 13438c8a9a5SSteve French opinfo = container_of(rcu_head, struct oplock_info, rcu_head); 13538c8a9a5SSteve French free_opinfo(opinfo); 13638c8a9a5SSteve French } 13738c8a9a5SSteve French 13838c8a9a5SSteve French struct oplock_info *opinfo_get(struct ksmbd_file *fp) 13938c8a9a5SSteve French { 14038c8a9a5SSteve French struct oplock_info *opinfo; 14138c8a9a5SSteve French 14238c8a9a5SSteve French rcu_read_lock(); 14338c8a9a5SSteve French opinfo = rcu_dereference(fp->f_opinfo); 14438c8a9a5SSteve French if (opinfo && !atomic_inc_not_zero(&opinfo->refcount)) 14538c8a9a5SSteve French opinfo = NULL; 14638c8a9a5SSteve French rcu_read_unlock(); 14738c8a9a5SSteve French 14838c8a9a5SSteve French return opinfo; 14938c8a9a5SSteve French } 15038c8a9a5SSteve French 15138c8a9a5SSteve French static struct oplock_info *opinfo_get_list(struct ksmbd_inode *ci) 15238c8a9a5SSteve French { 15338c8a9a5SSteve French struct oplock_info *opinfo; 15438c8a9a5SSteve French 15538c8a9a5SSteve French if (list_empty(&ci->m_op_list)) 15638c8a9a5SSteve French return NULL; 15738c8a9a5SSteve French 15838c8a9a5SSteve French rcu_read_lock(); 15938c8a9a5SSteve French opinfo = list_first_or_null_rcu(&ci->m_op_list, struct oplock_info, 16038c8a9a5SSteve French op_entry); 16136322523SNamjae Jeon if (opinfo) { 16236322523SNamjae Jeon if (!atomic_inc_not_zero(&opinfo->refcount)) 16338c8a9a5SSteve French opinfo = NULL; 16436322523SNamjae Jeon else { 16536322523SNamjae Jeon atomic_inc(&opinfo->conn->r_count); 16636322523SNamjae Jeon if (ksmbd_conn_releasing(opinfo->conn)) { 16736322523SNamjae Jeon atomic_dec(&opinfo->conn->r_count); 16836322523SNamjae Jeon atomic_dec(&opinfo->refcount); 16936322523SNamjae Jeon opinfo = NULL; 17036322523SNamjae Jeon } 17136322523SNamjae Jeon } 17236322523SNamjae Jeon } 17336322523SNamjae Jeon 17438c8a9a5SSteve French rcu_read_unlock(); 17538c8a9a5SSteve French 17638c8a9a5SSteve French return opinfo; 17738c8a9a5SSteve French } 17838c8a9a5SSteve French 17936322523SNamjae Jeon static void opinfo_conn_put(struct oplock_info *opinfo) 18036322523SNamjae Jeon { 18136322523SNamjae Jeon struct ksmbd_conn *conn; 18236322523SNamjae Jeon 18336322523SNamjae Jeon if (!opinfo) 18436322523SNamjae Jeon return; 18536322523SNamjae Jeon 18636322523SNamjae Jeon conn = opinfo->conn; 18736322523SNamjae Jeon /* 18836322523SNamjae Jeon * Checking waitqueue to dropping pending requests on 18936322523SNamjae Jeon * disconnection. waitqueue_active is safe because it 19036322523SNamjae Jeon * uses atomic operation for condition. 19136322523SNamjae Jeon */ 19236322523SNamjae Jeon if (!atomic_dec_return(&conn->r_count) && waitqueue_active(&conn->r_count_q)) 19336322523SNamjae Jeon wake_up(&conn->r_count_q); 19436322523SNamjae Jeon opinfo_put(opinfo); 19536322523SNamjae Jeon } 19636322523SNamjae Jeon 19738c8a9a5SSteve French void opinfo_put(struct oplock_info *opinfo) 19838c8a9a5SSteve French { 19938c8a9a5SSteve French if (!atomic_dec_and_test(&opinfo->refcount)) 20038c8a9a5SSteve French return; 20138c8a9a5SSteve French 20238c8a9a5SSteve French call_rcu(&opinfo->rcu_head, opinfo_free_rcu); 20338c8a9a5SSteve French } 20438c8a9a5SSteve French 20538c8a9a5SSteve French static void opinfo_add(struct oplock_info *opinfo) 20638c8a9a5SSteve French { 20738c8a9a5SSteve French struct ksmbd_inode *ci = opinfo->o_fp->f_ci; 20838c8a9a5SSteve French 20938c8a9a5SSteve French write_lock(&ci->m_lock); 21038c8a9a5SSteve French list_add_rcu(&opinfo->op_entry, &ci->m_op_list); 21138c8a9a5SSteve French write_unlock(&ci->m_lock); 21238c8a9a5SSteve French } 21338c8a9a5SSteve French 21438c8a9a5SSteve French static void opinfo_del(struct oplock_info *opinfo) 21538c8a9a5SSteve French { 21638c8a9a5SSteve French struct ksmbd_inode *ci = opinfo->o_fp->f_ci; 21738c8a9a5SSteve French 21838c8a9a5SSteve French if (opinfo->is_lease) { 21938c8a9a5SSteve French write_lock(&lease_list_lock); 22038c8a9a5SSteve French lease_del_list(opinfo); 22138c8a9a5SSteve French write_unlock(&lease_list_lock); 22238c8a9a5SSteve French } 22338c8a9a5SSteve French write_lock(&ci->m_lock); 22438c8a9a5SSteve French list_del_rcu(&opinfo->op_entry); 22538c8a9a5SSteve French write_unlock(&ci->m_lock); 22638c8a9a5SSteve French } 22738c8a9a5SSteve French 22838c8a9a5SSteve French static unsigned long opinfo_count(struct ksmbd_file *fp) 22938c8a9a5SSteve French { 23038c8a9a5SSteve French if (ksmbd_stream_fd(fp)) 23138c8a9a5SSteve French return atomic_read(&fp->f_ci->sop_count); 23238c8a9a5SSteve French else 23338c8a9a5SSteve French return atomic_read(&fp->f_ci->op_count); 23438c8a9a5SSteve French } 23538c8a9a5SSteve French 23638c8a9a5SSteve French static void opinfo_count_inc(struct ksmbd_file *fp) 23738c8a9a5SSteve French { 23838c8a9a5SSteve French if (ksmbd_stream_fd(fp)) 23938c8a9a5SSteve French return atomic_inc(&fp->f_ci->sop_count); 24038c8a9a5SSteve French else 24138c8a9a5SSteve French return atomic_inc(&fp->f_ci->op_count); 24238c8a9a5SSteve French } 24338c8a9a5SSteve French 24438c8a9a5SSteve French static void opinfo_count_dec(struct ksmbd_file *fp) 24538c8a9a5SSteve French { 24638c8a9a5SSteve French if (ksmbd_stream_fd(fp)) 24738c8a9a5SSteve French return atomic_dec(&fp->f_ci->sop_count); 24838c8a9a5SSteve French else 24938c8a9a5SSteve French return atomic_dec(&fp->f_ci->op_count); 25038c8a9a5SSteve French } 25138c8a9a5SSteve French 25238c8a9a5SSteve French /** 25338c8a9a5SSteve French * opinfo_write_to_read() - convert a write oplock to read oplock 25438c8a9a5SSteve French * @opinfo: current oplock info 25538c8a9a5SSteve French * 25638c8a9a5SSteve French * Return: 0 on success, otherwise -EINVAL 25738c8a9a5SSteve French */ 25838c8a9a5SSteve French int opinfo_write_to_read(struct oplock_info *opinfo) 25938c8a9a5SSteve French { 26038c8a9a5SSteve French struct lease *lease = opinfo->o_lease; 26138c8a9a5SSteve French 26238c8a9a5SSteve French if (!(opinfo->level == SMB2_OPLOCK_LEVEL_BATCH || 26338c8a9a5SSteve French opinfo->level == SMB2_OPLOCK_LEVEL_EXCLUSIVE)) { 26438c8a9a5SSteve French pr_err("bad oplock(0x%x)\n", opinfo->level); 26538c8a9a5SSteve French if (opinfo->is_lease) 26638c8a9a5SSteve French pr_err("lease state(0x%x)\n", lease->state); 26738c8a9a5SSteve French return -EINVAL; 26838c8a9a5SSteve French } 26938c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_II; 27038c8a9a5SSteve French 27138c8a9a5SSteve French if (opinfo->is_lease) 27238c8a9a5SSteve French lease->state = lease->new_state; 27338c8a9a5SSteve French return 0; 27438c8a9a5SSteve French } 27538c8a9a5SSteve French 27638c8a9a5SSteve French /** 27738c8a9a5SSteve French * opinfo_read_handle_to_read() - convert a read/handle oplock to read oplock 27838c8a9a5SSteve French * @opinfo: current oplock info 27938c8a9a5SSteve French * 28038c8a9a5SSteve French * Return: 0 on success, otherwise -EINVAL 28138c8a9a5SSteve French */ 28238c8a9a5SSteve French int opinfo_read_handle_to_read(struct oplock_info *opinfo) 28338c8a9a5SSteve French { 28438c8a9a5SSteve French struct lease *lease = opinfo->o_lease; 28538c8a9a5SSteve French 28638c8a9a5SSteve French lease->state = lease->new_state; 28738c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_II; 28838c8a9a5SSteve French return 0; 28938c8a9a5SSteve French } 29038c8a9a5SSteve French 29138c8a9a5SSteve French /** 29238c8a9a5SSteve French * opinfo_write_to_none() - convert a write oplock to none 29338c8a9a5SSteve French * @opinfo: current oplock info 29438c8a9a5SSteve French * 29538c8a9a5SSteve French * Return: 0 on success, otherwise -EINVAL 29638c8a9a5SSteve French */ 29738c8a9a5SSteve French int opinfo_write_to_none(struct oplock_info *opinfo) 29838c8a9a5SSteve French { 29938c8a9a5SSteve French struct lease *lease = opinfo->o_lease; 30038c8a9a5SSteve French 30138c8a9a5SSteve French if (!(opinfo->level == SMB2_OPLOCK_LEVEL_BATCH || 30238c8a9a5SSteve French opinfo->level == SMB2_OPLOCK_LEVEL_EXCLUSIVE)) { 30338c8a9a5SSteve French pr_err("bad oplock(0x%x)\n", opinfo->level); 30438c8a9a5SSteve French if (opinfo->is_lease) 30538c8a9a5SSteve French pr_err("lease state(0x%x)\n", lease->state); 30638c8a9a5SSteve French return -EINVAL; 30738c8a9a5SSteve French } 30838c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_NONE; 30938c8a9a5SSteve French if (opinfo->is_lease) 31038c8a9a5SSteve French lease->state = lease->new_state; 31138c8a9a5SSteve French return 0; 31238c8a9a5SSteve French } 31338c8a9a5SSteve French 31438c8a9a5SSteve French /** 31538c8a9a5SSteve French * opinfo_read_to_none() - convert a write read to none 31638c8a9a5SSteve French * @opinfo: current oplock info 31738c8a9a5SSteve French * 31838c8a9a5SSteve French * Return: 0 on success, otherwise -EINVAL 31938c8a9a5SSteve French */ 32038c8a9a5SSteve French int opinfo_read_to_none(struct oplock_info *opinfo) 32138c8a9a5SSteve French { 32238c8a9a5SSteve French struct lease *lease = opinfo->o_lease; 32338c8a9a5SSteve French 32438c8a9a5SSteve French if (opinfo->level != SMB2_OPLOCK_LEVEL_II) { 32538c8a9a5SSteve French pr_err("bad oplock(0x%x)\n", opinfo->level); 32638c8a9a5SSteve French if (opinfo->is_lease) 32738c8a9a5SSteve French pr_err("lease state(0x%x)\n", lease->state); 32838c8a9a5SSteve French return -EINVAL; 32938c8a9a5SSteve French } 33038c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_NONE; 33138c8a9a5SSteve French if (opinfo->is_lease) 33238c8a9a5SSteve French lease->state = lease->new_state; 33338c8a9a5SSteve French return 0; 33438c8a9a5SSteve French } 33538c8a9a5SSteve French 33638c8a9a5SSteve French /** 33738c8a9a5SSteve French * lease_read_to_write() - upgrade lease state from read to write 33838c8a9a5SSteve French * @opinfo: current lease info 33938c8a9a5SSteve French * 34038c8a9a5SSteve French * Return: 0 on success, otherwise -EINVAL 34138c8a9a5SSteve French */ 34238c8a9a5SSteve French int lease_read_to_write(struct oplock_info *opinfo) 34338c8a9a5SSteve French { 34438c8a9a5SSteve French struct lease *lease = opinfo->o_lease; 34538c8a9a5SSteve French 34638c8a9a5SSteve French if (!(lease->state & SMB2_LEASE_READ_CACHING_LE)) { 34738c8a9a5SSteve French ksmbd_debug(OPLOCK, "bad lease state(0x%x)\n", lease->state); 34838c8a9a5SSteve French return -EINVAL; 34938c8a9a5SSteve French } 35038c8a9a5SSteve French 35138c8a9a5SSteve French lease->new_state = SMB2_LEASE_NONE_LE; 35238c8a9a5SSteve French lease->state |= SMB2_LEASE_WRITE_CACHING_LE; 35338c8a9a5SSteve French if (lease->state & SMB2_LEASE_HANDLE_CACHING_LE) 35438c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_BATCH; 35538c8a9a5SSteve French else 35638c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_EXCLUSIVE; 35738c8a9a5SSteve French return 0; 35838c8a9a5SSteve French } 35938c8a9a5SSteve French 36038c8a9a5SSteve French /** 36138c8a9a5SSteve French * lease_none_upgrade() - upgrade lease state from none 36238c8a9a5SSteve French * @opinfo: current lease info 36338c8a9a5SSteve French * @new_state: new lease state 36438c8a9a5SSteve French * 36538c8a9a5SSteve French * Return: 0 on success, otherwise -EINVAL 36638c8a9a5SSteve French */ 36738c8a9a5SSteve French static int lease_none_upgrade(struct oplock_info *opinfo, __le32 new_state) 36838c8a9a5SSteve French { 36938c8a9a5SSteve French struct lease *lease = opinfo->o_lease; 37038c8a9a5SSteve French 37138c8a9a5SSteve French if (!(lease->state == SMB2_LEASE_NONE_LE)) { 37238c8a9a5SSteve French ksmbd_debug(OPLOCK, "bad lease state(0x%x)\n", lease->state); 37338c8a9a5SSteve French return -EINVAL; 37438c8a9a5SSteve French } 37538c8a9a5SSteve French 37638c8a9a5SSteve French lease->new_state = SMB2_LEASE_NONE_LE; 37738c8a9a5SSteve French lease->state = new_state; 37838c8a9a5SSteve French if (lease->state & SMB2_LEASE_HANDLE_CACHING_LE) 37938c8a9a5SSteve French if (lease->state & SMB2_LEASE_WRITE_CACHING_LE) 38038c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_BATCH; 38138c8a9a5SSteve French else 38238c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_II; 38338c8a9a5SSteve French else if (lease->state & SMB2_LEASE_WRITE_CACHING_LE) 38438c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_EXCLUSIVE; 38538c8a9a5SSteve French else if (lease->state & SMB2_LEASE_READ_CACHING_LE) 38638c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_II; 38738c8a9a5SSteve French 38838c8a9a5SSteve French return 0; 38938c8a9a5SSteve French } 39038c8a9a5SSteve French 39138c8a9a5SSteve French /** 39238c8a9a5SSteve French * close_id_del_oplock() - release oplock object at file close time 39338c8a9a5SSteve French * @fp: ksmbd file pointer 39438c8a9a5SSteve French */ 39538c8a9a5SSteve French void close_id_del_oplock(struct ksmbd_file *fp) 39638c8a9a5SSteve French { 39738c8a9a5SSteve French struct oplock_info *opinfo; 39838c8a9a5SSteve French 399ab5a0a1cSNamjae Jeon if (fp->reserve_lease_break) 400ab5a0a1cSNamjae Jeon smb_lazy_parent_lease_break_close(fp); 40138c8a9a5SSteve French 40238c8a9a5SSteve French opinfo = opinfo_get(fp); 40338c8a9a5SSteve French if (!opinfo) 40438c8a9a5SSteve French return; 40538c8a9a5SSteve French 40638c8a9a5SSteve French opinfo_del(opinfo); 40738c8a9a5SSteve French 40838c8a9a5SSteve French rcu_assign_pointer(fp->f_opinfo, NULL); 40938c8a9a5SSteve French if (opinfo->op_state == OPLOCK_ACK_WAIT) { 41038c8a9a5SSteve French opinfo->op_state = OPLOCK_CLOSING; 41138c8a9a5SSteve French wake_up_interruptible_all(&opinfo->oplock_q); 41238c8a9a5SSteve French if (opinfo->is_lease) { 41338c8a9a5SSteve French atomic_set(&opinfo->breaking_cnt, 0); 41438c8a9a5SSteve French wake_up_interruptible_all(&opinfo->oplock_brk); 41538c8a9a5SSteve French } 41638c8a9a5SSteve French } 41738c8a9a5SSteve French 41838c8a9a5SSteve French opinfo_count_dec(fp); 41938c8a9a5SSteve French atomic_dec(&opinfo->refcount); 42038c8a9a5SSteve French opinfo_put(opinfo); 42138c8a9a5SSteve French } 42238c8a9a5SSteve French 42338c8a9a5SSteve French /** 42438c8a9a5SSteve French * grant_write_oplock() - grant exclusive/batch oplock or write lease 42538c8a9a5SSteve French * @opinfo_new: new oplock info object 42638c8a9a5SSteve French * @req_oplock: request oplock 42738c8a9a5SSteve French * @lctx: lease context information 42838c8a9a5SSteve French * 42938c8a9a5SSteve French * Return: 0 43038c8a9a5SSteve French */ 43138c8a9a5SSteve French static void grant_write_oplock(struct oplock_info *opinfo_new, int req_oplock, 43238c8a9a5SSteve French struct lease_ctx_info *lctx) 43338c8a9a5SSteve French { 43438c8a9a5SSteve French struct lease *lease = opinfo_new->o_lease; 43538c8a9a5SSteve French 43638c8a9a5SSteve French if (req_oplock == SMB2_OPLOCK_LEVEL_BATCH) 43738c8a9a5SSteve French opinfo_new->level = SMB2_OPLOCK_LEVEL_BATCH; 43838c8a9a5SSteve French else 43938c8a9a5SSteve French opinfo_new->level = SMB2_OPLOCK_LEVEL_EXCLUSIVE; 44038c8a9a5SSteve French 44138c8a9a5SSteve French if (lctx) { 44238c8a9a5SSteve French lease->state = lctx->req_state; 44338c8a9a5SSteve French memcpy(lease->lease_key, lctx->lease_key, SMB2_LEASE_KEY_SIZE); 44438c8a9a5SSteve French } 44538c8a9a5SSteve French } 44638c8a9a5SSteve French 44738c8a9a5SSteve French /** 44838c8a9a5SSteve French * grant_read_oplock() - grant level2 oplock or read lease 44938c8a9a5SSteve French * @opinfo_new: new oplock info object 45038c8a9a5SSteve French * @lctx: lease context information 45138c8a9a5SSteve French * 45238c8a9a5SSteve French * Return: 0 45338c8a9a5SSteve French */ 45438c8a9a5SSteve French static void grant_read_oplock(struct oplock_info *opinfo_new, 45538c8a9a5SSteve French struct lease_ctx_info *lctx) 45638c8a9a5SSteve French { 45738c8a9a5SSteve French struct lease *lease = opinfo_new->o_lease; 45838c8a9a5SSteve French 45938c8a9a5SSteve French opinfo_new->level = SMB2_OPLOCK_LEVEL_II; 46038c8a9a5SSteve French 46138c8a9a5SSteve French if (lctx) { 46238c8a9a5SSteve French lease->state = SMB2_LEASE_READ_CACHING_LE; 46338c8a9a5SSteve French if (lctx->req_state & SMB2_LEASE_HANDLE_CACHING_LE) 46438c8a9a5SSteve French lease->state |= SMB2_LEASE_HANDLE_CACHING_LE; 46538c8a9a5SSteve French memcpy(lease->lease_key, lctx->lease_key, SMB2_LEASE_KEY_SIZE); 46638c8a9a5SSteve French } 46738c8a9a5SSteve French } 46838c8a9a5SSteve French 46938c8a9a5SSteve French /** 47038c8a9a5SSteve French * grant_none_oplock() - grant none oplock or none lease 47138c8a9a5SSteve French * @opinfo_new: new oplock info object 47238c8a9a5SSteve French * @lctx: lease context information 47338c8a9a5SSteve French * 47438c8a9a5SSteve French * Return: 0 47538c8a9a5SSteve French */ 47638c8a9a5SSteve French static void grant_none_oplock(struct oplock_info *opinfo_new, 47738c8a9a5SSteve French struct lease_ctx_info *lctx) 47838c8a9a5SSteve French { 47938c8a9a5SSteve French struct lease *lease = opinfo_new->o_lease; 48038c8a9a5SSteve French 48138c8a9a5SSteve French opinfo_new->level = SMB2_OPLOCK_LEVEL_NONE; 48238c8a9a5SSteve French 48338c8a9a5SSteve French if (lctx) { 48438c8a9a5SSteve French lease->state = 0; 48538c8a9a5SSteve French memcpy(lease->lease_key, lctx->lease_key, SMB2_LEASE_KEY_SIZE); 48638c8a9a5SSteve French } 48738c8a9a5SSteve French } 48838c8a9a5SSteve French 48938c8a9a5SSteve French static inline int compare_guid_key(struct oplock_info *opinfo, 49038c8a9a5SSteve French const char *guid1, const char *key1) 49138c8a9a5SSteve French { 49238c8a9a5SSteve French const char *guid2, *key2; 49338c8a9a5SSteve French 49438c8a9a5SSteve French guid2 = opinfo->conn->ClientGUID; 49538c8a9a5SSteve French key2 = opinfo->o_lease->lease_key; 49638c8a9a5SSteve French if (!memcmp(guid1, guid2, SMB2_CLIENT_GUID_SIZE) && 49738c8a9a5SSteve French !memcmp(key1, key2, SMB2_LEASE_KEY_SIZE)) 49838c8a9a5SSteve French return 1; 49938c8a9a5SSteve French 50038c8a9a5SSteve French return 0; 50138c8a9a5SSteve French } 50238c8a9a5SSteve French 50338c8a9a5SSteve French /** 50438c8a9a5SSteve French * same_client_has_lease() - check whether current lease request is 50538c8a9a5SSteve French * from lease owner of file 50638c8a9a5SSteve French * @ci: master file pointer 50738c8a9a5SSteve French * @client_guid: Client GUID 50838c8a9a5SSteve French * @lctx: lease context information 50938c8a9a5SSteve French * 51038c8a9a5SSteve French * Return: oplock(lease) object on success, otherwise NULL 51138c8a9a5SSteve French */ 51238c8a9a5SSteve French static struct oplock_info *same_client_has_lease(struct ksmbd_inode *ci, 51338c8a9a5SSteve French char *client_guid, 51438c8a9a5SSteve French struct lease_ctx_info *lctx) 51538c8a9a5SSteve French { 51638c8a9a5SSteve French int ret; 51738c8a9a5SSteve French struct lease *lease; 51838c8a9a5SSteve French struct oplock_info *opinfo; 51938c8a9a5SSteve French struct oplock_info *m_opinfo = NULL; 52038c8a9a5SSteve French 52138c8a9a5SSteve French if (!lctx) 52238c8a9a5SSteve French return NULL; 52338c8a9a5SSteve French 52438c8a9a5SSteve French /* 52538c8a9a5SSteve French * Compare lease key and client_guid to know request from same owner 52638c8a9a5SSteve French * of same client 52738c8a9a5SSteve French */ 52838c8a9a5SSteve French read_lock(&ci->m_lock); 52938c8a9a5SSteve French list_for_each_entry(opinfo, &ci->m_op_list, op_entry) { 53038c8a9a5SSteve French if (!opinfo->is_lease) 53138c8a9a5SSteve French continue; 53238c8a9a5SSteve French read_unlock(&ci->m_lock); 53338c8a9a5SSteve French lease = opinfo->o_lease; 53438c8a9a5SSteve French 53538c8a9a5SSteve French ret = compare_guid_key(opinfo, client_guid, lctx->lease_key); 53638c8a9a5SSteve French if (ret) { 53738c8a9a5SSteve French m_opinfo = opinfo; 53838c8a9a5SSteve French /* skip upgrading lease about breaking lease */ 53938c8a9a5SSteve French if (atomic_read(&opinfo->breaking_cnt)) { 54038c8a9a5SSteve French read_lock(&ci->m_lock); 54138c8a9a5SSteve French continue; 54238c8a9a5SSteve French } 54338c8a9a5SSteve French 54438c8a9a5SSteve French /* upgrading lease */ 54538c8a9a5SSteve French if ((atomic_read(&ci->op_count) + 54638c8a9a5SSteve French atomic_read(&ci->sop_count)) == 1) { 5473c1e602aSNamjae Jeon if (lease->state != SMB2_LEASE_NONE_LE && 5483c1e602aSNamjae Jeon lease->state == (lctx->req_state & lease->state)) { 54938c8a9a5SSteve French lease->state |= lctx->req_state; 55038c8a9a5SSteve French if (lctx->req_state & 55138c8a9a5SSteve French SMB2_LEASE_WRITE_CACHING_LE) 55238c8a9a5SSteve French lease_read_to_write(opinfo); 5533c1e602aSNamjae Jeon 55438c8a9a5SSteve French } 55538c8a9a5SSteve French } else if ((atomic_read(&ci->op_count) + 55638c8a9a5SSteve French atomic_read(&ci->sop_count)) > 1) { 55738c8a9a5SSteve French if (lctx->req_state == 55838c8a9a5SSteve French (SMB2_LEASE_READ_CACHING_LE | 55938c8a9a5SSteve French SMB2_LEASE_HANDLE_CACHING_LE)) 56038c8a9a5SSteve French lease->state = lctx->req_state; 56138c8a9a5SSteve French } 56238c8a9a5SSteve French 56338c8a9a5SSteve French if (lctx->req_state && lease->state == 56438c8a9a5SSteve French SMB2_LEASE_NONE_LE) 56538c8a9a5SSteve French lease_none_upgrade(opinfo, lctx->req_state); 56638c8a9a5SSteve French } 56738c8a9a5SSteve French read_lock(&ci->m_lock); 56838c8a9a5SSteve French } 56938c8a9a5SSteve French read_unlock(&ci->m_lock); 57038c8a9a5SSteve French 57138c8a9a5SSteve French return m_opinfo; 57238c8a9a5SSteve French } 57338c8a9a5SSteve French 57438c8a9a5SSteve French static void wait_for_break_ack(struct oplock_info *opinfo) 57538c8a9a5SSteve French { 57638c8a9a5SSteve French int rc = 0; 57738c8a9a5SSteve French 57838c8a9a5SSteve French rc = wait_event_interruptible_timeout(opinfo->oplock_q, 57938c8a9a5SSteve French opinfo->op_state == OPLOCK_STATE_NONE || 58038c8a9a5SSteve French opinfo->op_state == OPLOCK_CLOSING, 58138c8a9a5SSteve French OPLOCK_WAIT_TIME); 58238c8a9a5SSteve French 58338c8a9a5SSteve French /* is this a timeout ? */ 58438c8a9a5SSteve French if (!rc) { 58538c8a9a5SSteve French if (opinfo->is_lease) 58638c8a9a5SSteve French opinfo->o_lease->state = SMB2_LEASE_NONE_LE; 58738c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_NONE; 58838c8a9a5SSteve French opinfo->op_state = OPLOCK_STATE_NONE; 58938c8a9a5SSteve French } 59038c8a9a5SSteve French } 59138c8a9a5SSteve French 59238c8a9a5SSteve French static void wake_up_oplock_break(struct oplock_info *opinfo) 59338c8a9a5SSteve French { 59438c8a9a5SSteve French clear_bit_unlock(0, &opinfo->pending_break); 59538c8a9a5SSteve French /* memory barrier is needed for wake_up_bit() */ 59638c8a9a5SSteve French smp_mb__after_atomic(); 59738c8a9a5SSteve French wake_up_bit(&opinfo->pending_break, 0); 59838c8a9a5SSteve French } 59938c8a9a5SSteve French 60038c8a9a5SSteve French static int oplock_break_pending(struct oplock_info *opinfo, int req_op_level) 60138c8a9a5SSteve French { 60238c8a9a5SSteve French while (test_and_set_bit(0, &opinfo->pending_break)) { 60338c8a9a5SSteve French wait_on_bit(&opinfo->pending_break, 0, TASK_UNINTERRUPTIBLE); 60438c8a9a5SSteve French 60538c8a9a5SSteve French /* Not immediately break to none. */ 60638c8a9a5SSteve French opinfo->open_trunc = 0; 60738c8a9a5SSteve French 60838c8a9a5SSteve French if (opinfo->op_state == OPLOCK_CLOSING) 60938c8a9a5SSteve French return -ENOENT; 61038c8a9a5SSteve French else if (!opinfo->is_lease && opinfo->level <= req_op_level) 61138c8a9a5SSteve French return 1; 61238c8a9a5SSteve French } 61338c8a9a5SSteve French 61438c8a9a5SSteve French if (!opinfo->is_lease && opinfo->level <= req_op_level) { 61538c8a9a5SSteve French wake_up_oplock_break(opinfo); 61638c8a9a5SSteve French return 1; 61738c8a9a5SSteve French } 61838c8a9a5SSteve French return 0; 61938c8a9a5SSteve French } 62038c8a9a5SSteve French 62138c8a9a5SSteve French /** 62238c8a9a5SSteve French * __smb2_oplock_break_noti() - send smb2 oplock break cmd from conn 62338c8a9a5SSteve French * to client 62438c8a9a5SSteve French * @wk: smb work object 62538c8a9a5SSteve French * 62638c8a9a5SSteve French * There are two ways this function can be called. 1- while file open we break 62738c8a9a5SSteve French * from exclusive/batch lock to levelII oplock and 2- while file write/truncate 62838c8a9a5SSteve French * we break from levelII oplock no oplock. 62938c8a9a5SSteve French * work->request_buf contains oplock_info. 63038c8a9a5SSteve French */ 63138c8a9a5SSteve French static void __smb2_oplock_break_noti(struct work_struct *wk) 63238c8a9a5SSteve French { 63338c8a9a5SSteve French struct smb2_oplock_break *rsp = NULL; 63438c8a9a5SSteve French struct ksmbd_work *work = container_of(wk, struct ksmbd_work, work); 63538c8a9a5SSteve French struct oplock_break_info *br_info = work->request_buf; 63638c8a9a5SSteve French struct smb2_hdr *rsp_hdr; 63738c8a9a5SSteve French struct ksmbd_file *fp; 63838c8a9a5SSteve French 63938c8a9a5SSteve French fp = ksmbd_lookup_durable_fd(br_info->fid); 64038c8a9a5SSteve French if (!fp) 64138c8a9a5SSteve French goto out; 64238c8a9a5SSteve French 643041bba44SNamjae Jeon if (allocate_interim_rsp_buf(work)) { 64438c8a9a5SSteve French pr_err("smb2_allocate_rsp_buf failed! "); 64538c8a9a5SSteve French ksmbd_fd_put(work, fp); 64638c8a9a5SSteve French goto out; 64738c8a9a5SSteve French } 64838c8a9a5SSteve French 64938c8a9a5SSteve French rsp_hdr = smb2_get_msg(work->response_buf); 65038c8a9a5SSteve French memset(rsp_hdr, 0, sizeof(struct smb2_hdr) + 2); 65138c8a9a5SSteve French rsp_hdr->ProtocolId = SMB2_PROTO_NUMBER; 65238c8a9a5SSteve French rsp_hdr->StructureSize = SMB2_HEADER_STRUCTURE_SIZE; 65338c8a9a5SSteve French rsp_hdr->CreditRequest = cpu_to_le16(0); 65438c8a9a5SSteve French rsp_hdr->Command = SMB2_OPLOCK_BREAK; 65538c8a9a5SSteve French rsp_hdr->Flags = (SMB2_FLAGS_SERVER_TO_REDIR); 65638c8a9a5SSteve French rsp_hdr->NextCommand = 0; 65738c8a9a5SSteve French rsp_hdr->MessageId = cpu_to_le64(-1); 65838c8a9a5SSteve French rsp_hdr->Id.SyncId.ProcessId = 0; 65938c8a9a5SSteve French rsp_hdr->Id.SyncId.TreeId = 0; 66038c8a9a5SSteve French rsp_hdr->SessionId = 0; 66138c8a9a5SSteve French memset(rsp_hdr->Signature, 0, 16); 66238c8a9a5SSteve French 66338c8a9a5SSteve French rsp = smb2_get_msg(work->response_buf); 66438c8a9a5SSteve French 66538c8a9a5SSteve French rsp->StructureSize = cpu_to_le16(24); 66638c8a9a5SSteve French if (!br_info->open_trunc && 66738c8a9a5SSteve French (br_info->level == SMB2_OPLOCK_LEVEL_BATCH || 66838c8a9a5SSteve French br_info->level == SMB2_OPLOCK_LEVEL_EXCLUSIVE)) 66938c8a9a5SSteve French rsp->OplockLevel = SMB2_OPLOCK_LEVEL_II; 67038c8a9a5SSteve French else 67138c8a9a5SSteve French rsp->OplockLevel = SMB2_OPLOCK_LEVEL_NONE; 67238c8a9a5SSteve French rsp->Reserved = 0; 67338c8a9a5SSteve French rsp->Reserved2 = 0; 67438c8a9a5SSteve French rsp->PersistentFid = fp->persistent_id; 67538c8a9a5SSteve French rsp->VolatileFid = fp->volatile_id; 67638c8a9a5SSteve French 677e2b76ab8SNamjae Jeon ksmbd_fd_put(work, fp); 678e2b76ab8SNamjae Jeon if (ksmbd_iov_pin_rsp(work, (void *)rsp, 679e2b76ab8SNamjae Jeon sizeof(struct smb2_oplock_break))) 680e2b76ab8SNamjae Jeon goto out; 68138c8a9a5SSteve French 68238c8a9a5SSteve French ksmbd_debug(OPLOCK, 68338c8a9a5SSteve French "sending oplock break v_id %llu p_id = %llu lock level = %d\n", 68438c8a9a5SSteve French rsp->VolatileFid, rsp->PersistentFid, rsp->OplockLevel); 68538c8a9a5SSteve French 68638c8a9a5SSteve French ksmbd_conn_write(work); 68738c8a9a5SSteve French 68838c8a9a5SSteve French out: 68938c8a9a5SSteve French ksmbd_free_work_struct(work); 69038c8a9a5SSteve French } 69138c8a9a5SSteve French 69238c8a9a5SSteve French /** 69338c8a9a5SSteve French * smb2_oplock_break_noti() - send smb2 exclusive/batch to level2 oplock 69438c8a9a5SSteve French * break command from server to client 69538c8a9a5SSteve French * @opinfo: oplock info object 69638c8a9a5SSteve French * 69738c8a9a5SSteve French * Return: 0 on success, otherwise error 69838c8a9a5SSteve French */ 69938c8a9a5SSteve French static int smb2_oplock_break_noti(struct oplock_info *opinfo) 70038c8a9a5SSteve French { 70138c8a9a5SSteve French struct ksmbd_conn *conn = opinfo->conn; 70238c8a9a5SSteve French struct oplock_break_info *br_info; 70338c8a9a5SSteve French int ret = 0; 70438c8a9a5SSteve French struct ksmbd_work *work = ksmbd_alloc_work_struct(); 70538c8a9a5SSteve French 70638c8a9a5SSteve French if (!work) 70738c8a9a5SSteve French return -ENOMEM; 70838c8a9a5SSteve French 70938c8a9a5SSteve French br_info = kmalloc(sizeof(struct oplock_break_info), GFP_KERNEL); 71038c8a9a5SSteve French if (!br_info) { 71138c8a9a5SSteve French ksmbd_free_work_struct(work); 71238c8a9a5SSteve French return -ENOMEM; 71338c8a9a5SSteve French } 71438c8a9a5SSteve French 71538c8a9a5SSteve French br_info->level = opinfo->level; 71638c8a9a5SSteve French br_info->fid = opinfo->fid; 71738c8a9a5SSteve French br_info->open_trunc = opinfo->open_trunc; 71838c8a9a5SSteve French 71938c8a9a5SSteve French work->request_buf = (char *)br_info; 72038c8a9a5SSteve French work->conn = conn; 72138c8a9a5SSteve French work->sess = opinfo->sess; 72238c8a9a5SSteve French 72338c8a9a5SSteve French if (opinfo->op_state == OPLOCK_ACK_WAIT) { 72438c8a9a5SSteve French INIT_WORK(&work->work, __smb2_oplock_break_noti); 72538c8a9a5SSteve French ksmbd_queue_work(work); 72638c8a9a5SSteve French 72738c8a9a5SSteve French wait_for_break_ack(opinfo); 72838c8a9a5SSteve French } else { 72938c8a9a5SSteve French __smb2_oplock_break_noti(&work->work); 73038c8a9a5SSteve French if (opinfo->level == SMB2_OPLOCK_LEVEL_II) 73138c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_NONE; 73238c8a9a5SSteve French } 73338c8a9a5SSteve French return ret; 73438c8a9a5SSteve French } 73538c8a9a5SSteve French 73638c8a9a5SSteve French /** 73738c8a9a5SSteve French * __smb2_lease_break_noti() - send lease break command from server 73838c8a9a5SSteve French * to client 73938c8a9a5SSteve French * @wk: smb work object 74038c8a9a5SSteve French */ 74138c8a9a5SSteve French static void __smb2_lease_break_noti(struct work_struct *wk) 74238c8a9a5SSteve French { 74338c8a9a5SSteve French struct smb2_lease_break *rsp = NULL; 74438c8a9a5SSteve French struct ksmbd_work *work = container_of(wk, struct ksmbd_work, work); 74538c8a9a5SSteve French struct lease_break_info *br_info = work->request_buf; 74638c8a9a5SSteve French struct smb2_hdr *rsp_hdr; 74738c8a9a5SSteve French 748041bba44SNamjae Jeon if (allocate_interim_rsp_buf(work)) { 74938c8a9a5SSteve French ksmbd_debug(OPLOCK, "smb2_allocate_rsp_buf failed! "); 75038c8a9a5SSteve French goto out; 75138c8a9a5SSteve French } 75238c8a9a5SSteve French 75338c8a9a5SSteve French rsp_hdr = smb2_get_msg(work->response_buf); 75438c8a9a5SSteve French memset(rsp_hdr, 0, sizeof(struct smb2_hdr) + 2); 75538c8a9a5SSteve French rsp_hdr->ProtocolId = SMB2_PROTO_NUMBER; 75638c8a9a5SSteve French rsp_hdr->StructureSize = SMB2_HEADER_STRUCTURE_SIZE; 75738c8a9a5SSteve French rsp_hdr->CreditRequest = cpu_to_le16(0); 75838c8a9a5SSteve French rsp_hdr->Command = SMB2_OPLOCK_BREAK; 75938c8a9a5SSteve French rsp_hdr->Flags = (SMB2_FLAGS_SERVER_TO_REDIR); 76038c8a9a5SSteve French rsp_hdr->NextCommand = 0; 76138c8a9a5SSteve French rsp_hdr->MessageId = cpu_to_le64(-1); 76238c8a9a5SSteve French rsp_hdr->Id.SyncId.ProcessId = 0; 76338c8a9a5SSteve French rsp_hdr->Id.SyncId.TreeId = 0; 76438c8a9a5SSteve French rsp_hdr->SessionId = 0; 76538c8a9a5SSteve French memset(rsp_hdr->Signature, 0, 16); 76638c8a9a5SSteve French 76738c8a9a5SSteve French rsp = smb2_get_msg(work->response_buf); 76838c8a9a5SSteve French rsp->StructureSize = cpu_to_le16(44); 76938c8a9a5SSteve French rsp->Epoch = br_info->epoch; 77038c8a9a5SSteve French rsp->Flags = 0; 77138c8a9a5SSteve French 77238c8a9a5SSteve French if (br_info->curr_state & (SMB2_LEASE_WRITE_CACHING_LE | 77338c8a9a5SSteve French SMB2_LEASE_HANDLE_CACHING_LE)) 77438c8a9a5SSteve French rsp->Flags = SMB2_NOTIFY_BREAK_LEASE_FLAG_ACK_REQUIRED; 77538c8a9a5SSteve French 77638c8a9a5SSteve French memcpy(rsp->LeaseKey, br_info->lease_key, SMB2_LEASE_KEY_SIZE); 77738c8a9a5SSteve French rsp->CurrentLeaseState = br_info->curr_state; 77838c8a9a5SSteve French rsp->NewLeaseState = br_info->new_state; 77938c8a9a5SSteve French rsp->BreakReason = 0; 78038c8a9a5SSteve French rsp->AccessMaskHint = 0; 78138c8a9a5SSteve French rsp->ShareMaskHint = 0; 78238c8a9a5SSteve French 783e2b76ab8SNamjae Jeon if (ksmbd_iov_pin_rsp(work, (void *)rsp, 784e2b76ab8SNamjae Jeon sizeof(struct smb2_lease_break))) 785e2b76ab8SNamjae Jeon goto out; 78638c8a9a5SSteve French 78738c8a9a5SSteve French ksmbd_conn_write(work); 78838c8a9a5SSteve French 78938c8a9a5SSteve French out: 79038c8a9a5SSteve French ksmbd_free_work_struct(work); 79138c8a9a5SSteve French } 79238c8a9a5SSteve French 79338c8a9a5SSteve French /** 79438c8a9a5SSteve French * smb2_lease_break_noti() - break lease when a new client request 79538c8a9a5SSteve French * write lease 79638c8a9a5SSteve French * @opinfo: conains lease state information 79738c8a9a5SSteve French * 79838c8a9a5SSteve French * Return: 0 on success, otherwise error 79938c8a9a5SSteve French */ 80038c8a9a5SSteve French static int smb2_lease_break_noti(struct oplock_info *opinfo) 80138c8a9a5SSteve French { 80238c8a9a5SSteve French struct ksmbd_conn *conn = opinfo->conn; 80338c8a9a5SSteve French struct list_head *tmp, *t; 80438c8a9a5SSteve French struct ksmbd_work *work; 80538c8a9a5SSteve French struct lease_break_info *br_info; 80638c8a9a5SSteve French struct lease *lease = opinfo->o_lease; 80738c8a9a5SSteve French 80838c8a9a5SSteve French work = ksmbd_alloc_work_struct(); 80938c8a9a5SSteve French if (!work) 81038c8a9a5SSteve French return -ENOMEM; 81138c8a9a5SSteve French 81238c8a9a5SSteve French br_info = kmalloc(sizeof(struct lease_break_info), GFP_KERNEL); 81338c8a9a5SSteve French if (!br_info) { 81438c8a9a5SSteve French ksmbd_free_work_struct(work); 81538c8a9a5SSteve French return -ENOMEM; 81638c8a9a5SSteve French } 81738c8a9a5SSteve French 81838c8a9a5SSteve French br_info->curr_state = lease->state; 81938c8a9a5SSteve French br_info->new_state = lease->new_state; 82038c8a9a5SSteve French if (lease->version == 2) 82138c8a9a5SSteve French br_info->epoch = cpu_to_le16(++lease->epoch); 82238c8a9a5SSteve French else 82338c8a9a5SSteve French br_info->epoch = 0; 82438c8a9a5SSteve French memcpy(br_info->lease_key, lease->lease_key, SMB2_LEASE_KEY_SIZE); 82538c8a9a5SSteve French 82638c8a9a5SSteve French work->request_buf = (char *)br_info; 82738c8a9a5SSteve French work->conn = conn; 82838c8a9a5SSteve French work->sess = opinfo->sess; 82938c8a9a5SSteve French 83038c8a9a5SSteve French if (opinfo->op_state == OPLOCK_ACK_WAIT) { 83138c8a9a5SSteve French list_for_each_safe(tmp, t, &opinfo->interim_list) { 83238c8a9a5SSteve French struct ksmbd_work *in_work; 83338c8a9a5SSteve French 83438c8a9a5SSteve French in_work = list_entry(tmp, struct ksmbd_work, 83538c8a9a5SSteve French interim_entry); 83638c8a9a5SSteve French setup_async_work(in_work, NULL, NULL); 83738c8a9a5SSteve French smb2_send_interim_resp(in_work, STATUS_PENDING); 838fa86141fSNamjae Jeon list_del_init(&in_work->interim_entry); 839fa86141fSNamjae Jeon release_async_work(in_work); 84038c8a9a5SSteve French } 84138c8a9a5SSteve French INIT_WORK(&work->work, __smb2_lease_break_noti); 84238c8a9a5SSteve French ksmbd_queue_work(work); 84338c8a9a5SSteve French wait_for_break_ack(opinfo); 84438c8a9a5SSteve French } else { 84538c8a9a5SSteve French __smb2_lease_break_noti(&work->work); 84638c8a9a5SSteve French if (opinfo->o_lease->new_state == SMB2_LEASE_NONE_LE) { 84738c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_NONE; 84838c8a9a5SSteve French opinfo->o_lease->state = SMB2_LEASE_NONE_LE; 84938c8a9a5SSteve French } 85038c8a9a5SSteve French } 85138c8a9a5SSteve French return 0; 85238c8a9a5SSteve French } 85338c8a9a5SSteve French 85438c8a9a5SSteve French static void wait_lease_breaking(struct oplock_info *opinfo) 85538c8a9a5SSteve French { 85638c8a9a5SSteve French if (!opinfo->is_lease) 85738c8a9a5SSteve French return; 85838c8a9a5SSteve French 85938c8a9a5SSteve French wake_up_interruptible_all(&opinfo->oplock_brk); 86038c8a9a5SSteve French if (atomic_read(&opinfo->breaking_cnt)) { 86138c8a9a5SSteve French int ret = 0; 86238c8a9a5SSteve French 86338c8a9a5SSteve French ret = wait_event_interruptible_timeout(opinfo->oplock_brk, 86438c8a9a5SSteve French atomic_read(&opinfo->breaking_cnt) == 0, 86538c8a9a5SSteve French HZ); 86638c8a9a5SSteve French if (!ret) 86738c8a9a5SSteve French atomic_set(&opinfo->breaking_cnt, 0); 86838c8a9a5SSteve French } 86938c8a9a5SSteve French } 87038c8a9a5SSteve French 87138c8a9a5SSteve French static int oplock_break(struct oplock_info *brk_opinfo, int req_op_level) 87238c8a9a5SSteve French { 87338c8a9a5SSteve French int err = 0; 87438c8a9a5SSteve French 87538c8a9a5SSteve French /* Need to break exclusive/batch oplock, write lease or overwrite_if */ 87638c8a9a5SSteve French ksmbd_debug(OPLOCK, 87738c8a9a5SSteve French "request to send oplock(level : 0x%x) break notification\n", 87838c8a9a5SSteve French brk_opinfo->level); 87938c8a9a5SSteve French 88038c8a9a5SSteve French if (brk_opinfo->is_lease) { 88138c8a9a5SSteve French struct lease *lease = brk_opinfo->o_lease; 88238c8a9a5SSteve French 88338c8a9a5SSteve French atomic_inc(&brk_opinfo->breaking_cnt); 88438c8a9a5SSteve French 88538c8a9a5SSteve French err = oplock_break_pending(brk_opinfo, req_op_level); 88638c8a9a5SSteve French if (err) 88738c8a9a5SSteve French return err < 0 ? err : 0; 88838c8a9a5SSteve French 88938c8a9a5SSteve French if (brk_opinfo->open_trunc) { 89038c8a9a5SSteve French /* 89138c8a9a5SSteve French * Create overwrite break trigger the lease break to 89238c8a9a5SSteve French * none. 89338c8a9a5SSteve French */ 89438c8a9a5SSteve French lease->new_state = SMB2_LEASE_NONE_LE; 89538c8a9a5SSteve French } else { 89638c8a9a5SSteve French if (lease->state & SMB2_LEASE_WRITE_CACHING_LE) { 89738c8a9a5SSteve French if (lease->state & SMB2_LEASE_HANDLE_CACHING_LE) 89838c8a9a5SSteve French lease->new_state = 89938c8a9a5SSteve French SMB2_LEASE_READ_CACHING_LE | 90038c8a9a5SSteve French SMB2_LEASE_HANDLE_CACHING_LE; 90138c8a9a5SSteve French else 90238c8a9a5SSteve French lease->new_state = 90338c8a9a5SSteve French SMB2_LEASE_READ_CACHING_LE; 90438c8a9a5SSteve French } else { 9053c1e602aSNamjae Jeon if (lease->state & SMB2_LEASE_HANDLE_CACHING_LE && 9063c1e602aSNamjae Jeon !lease->is_dir) 90738c8a9a5SSteve French lease->new_state = 90838c8a9a5SSteve French SMB2_LEASE_READ_CACHING_LE; 90938c8a9a5SSteve French else 91038c8a9a5SSteve French lease->new_state = SMB2_LEASE_NONE_LE; 91138c8a9a5SSteve French } 91238c8a9a5SSteve French } 91338c8a9a5SSteve French 91438c8a9a5SSteve French if (lease->state & (SMB2_LEASE_WRITE_CACHING_LE | 91538c8a9a5SSteve French SMB2_LEASE_HANDLE_CACHING_LE)) 91638c8a9a5SSteve French brk_opinfo->op_state = OPLOCK_ACK_WAIT; 91738c8a9a5SSteve French else 91838c8a9a5SSteve French atomic_dec(&brk_opinfo->breaking_cnt); 91938c8a9a5SSteve French } else { 92038c8a9a5SSteve French err = oplock_break_pending(brk_opinfo, req_op_level); 92138c8a9a5SSteve French if (err) 92238c8a9a5SSteve French return err < 0 ? err : 0; 92338c8a9a5SSteve French 92438c8a9a5SSteve French if (brk_opinfo->level == SMB2_OPLOCK_LEVEL_BATCH || 92538c8a9a5SSteve French brk_opinfo->level == SMB2_OPLOCK_LEVEL_EXCLUSIVE) 92638c8a9a5SSteve French brk_opinfo->op_state = OPLOCK_ACK_WAIT; 92738c8a9a5SSteve French } 92838c8a9a5SSteve French 92938c8a9a5SSteve French if (brk_opinfo->is_lease) 93038c8a9a5SSteve French err = smb2_lease_break_noti(brk_opinfo); 93138c8a9a5SSteve French else 93238c8a9a5SSteve French err = smb2_oplock_break_noti(brk_opinfo); 93338c8a9a5SSteve French 93438c8a9a5SSteve French ksmbd_debug(OPLOCK, "oplock granted = %d\n", brk_opinfo->level); 93538c8a9a5SSteve French if (brk_opinfo->op_state == OPLOCK_CLOSING) 93638c8a9a5SSteve French err = -ENOENT; 93738c8a9a5SSteve French wake_up_oplock_break(brk_opinfo); 93838c8a9a5SSteve French 93938c8a9a5SSteve French wait_lease_breaking(brk_opinfo); 94038c8a9a5SSteve French 94138c8a9a5SSteve French return err; 94238c8a9a5SSteve French } 94338c8a9a5SSteve French 94438c8a9a5SSteve French void destroy_lease_table(struct ksmbd_conn *conn) 94538c8a9a5SSteve French { 94638c8a9a5SSteve French struct lease_table *lb, *lbtmp; 94738c8a9a5SSteve French struct oplock_info *opinfo; 94838c8a9a5SSteve French 94938c8a9a5SSteve French write_lock(&lease_list_lock); 95038c8a9a5SSteve French if (list_empty(&lease_table_list)) { 95138c8a9a5SSteve French write_unlock(&lease_list_lock); 95238c8a9a5SSteve French return; 95338c8a9a5SSteve French } 95438c8a9a5SSteve French 95538c8a9a5SSteve French list_for_each_entry_safe(lb, lbtmp, &lease_table_list, l_entry) { 95638c8a9a5SSteve French if (conn && memcmp(lb->client_guid, conn->ClientGUID, 95738c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE)) 95838c8a9a5SSteve French continue; 95938c8a9a5SSteve French again: 96038c8a9a5SSteve French rcu_read_lock(); 96138c8a9a5SSteve French list_for_each_entry_rcu(opinfo, &lb->lease_list, 96238c8a9a5SSteve French lease_entry) { 96338c8a9a5SSteve French rcu_read_unlock(); 96438c8a9a5SSteve French lease_del_list(opinfo); 96538c8a9a5SSteve French goto again; 96638c8a9a5SSteve French } 96738c8a9a5SSteve French rcu_read_unlock(); 96838c8a9a5SSteve French list_del(&lb->l_entry); 96938c8a9a5SSteve French kfree(lb); 97038c8a9a5SSteve French } 97138c8a9a5SSteve French write_unlock(&lease_list_lock); 97238c8a9a5SSteve French } 97338c8a9a5SSteve French 97438c8a9a5SSteve French int find_same_lease_key(struct ksmbd_session *sess, struct ksmbd_inode *ci, 97538c8a9a5SSteve French struct lease_ctx_info *lctx) 97638c8a9a5SSteve French { 97738c8a9a5SSteve French struct oplock_info *opinfo; 97838c8a9a5SSteve French int err = 0; 97938c8a9a5SSteve French struct lease_table *lb; 98038c8a9a5SSteve French 98138c8a9a5SSteve French if (!lctx) 98238c8a9a5SSteve French return err; 98338c8a9a5SSteve French 98438c8a9a5SSteve French read_lock(&lease_list_lock); 98538c8a9a5SSteve French if (list_empty(&lease_table_list)) { 98638c8a9a5SSteve French read_unlock(&lease_list_lock); 98738c8a9a5SSteve French return 0; 98838c8a9a5SSteve French } 98938c8a9a5SSteve French 99038c8a9a5SSteve French list_for_each_entry(lb, &lease_table_list, l_entry) { 99138c8a9a5SSteve French if (!memcmp(lb->client_guid, sess->ClientGUID, 99238c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE)) 99338c8a9a5SSteve French goto found; 99438c8a9a5SSteve French } 99538c8a9a5SSteve French read_unlock(&lease_list_lock); 99638c8a9a5SSteve French 99738c8a9a5SSteve French return 0; 99838c8a9a5SSteve French 99938c8a9a5SSteve French found: 100038c8a9a5SSteve French rcu_read_lock(); 100138c8a9a5SSteve French list_for_each_entry_rcu(opinfo, &lb->lease_list, lease_entry) { 100238c8a9a5SSteve French if (!atomic_inc_not_zero(&opinfo->refcount)) 100338c8a9a5SSteve French continue; 100438c8a9a5SSteve French rcu_read_unlock(); 100538c8a9a5SSteve French if (opinfo->o_fp->f_ci == ci) 100638c8a9a5SSteve French goto op_next; 100738c8a9a5SSteve French err = compare_guid_key(opinfo, sess->ClientGUID, 100838c8a9a5SSteve French lctx->lease_key); 100938c8a9a5SSteve French if (err) { 101038c8a9a5SSteve French err = -EINVAL; 101138c8a9a5SSteve French ksmbd_debug(OPLOCK, 101238c8a9a5SSteve French "found same lease key is already used in other files\n"); 101338c8a9a5SSteve French opinfo_put(opinfo); 101438c8a9a5SSteve French goto out; 101538c8a9a5SSteve French } 101638c8a9a5SSteve French op_next: 101738c8a9a5SSteve French opinfo_put(opinfo); 101838c8a9a5SSteve French rcu_read_lock(); 101938c8a9a5SSteve French } 102038c8a9a5SSteve French rcu_read_unlock(); 102138c8a9a5SSteve French 102238c8a9a5SSteve French out: 102338c8a9a5SSteve French read_unlock(&lease_list_lock); 102438c8a9a5SSteve French return err; 102538c8a9a5SSteve French } 102638c8a9a5SSteve French 102738c8a9a5SSteve French static void copy_lease(struct oplock_info *op1, struct oplock_info *op2) 102838c8a9a5SSteve French { 102938c8a9a5SSteve French struct lease *lease1 = op1->o_lease; 103038c8a9a5SSteve French struct lease *lease2 = op2->o_lease; 103138c8a9a5SSteve French 103238c8a9a5SSteve French op2->level = op1->level; 103338c8a9a5SSteve French lease2->state = lease1->state; 103438c8a9a5SSteve French memcpy(lease2->lease_key, lease1->lease_key, 103538c8a9a5SSteve French SMB2_LEASE_KEY_SIZE); 103638c8a9a5SSteve French lease2->duration = lease1->duration; 103738c8a9a5SSteve French lease2->flags = lease1->flags; 1038bc025d49SNamjae Jeon lease2->epoch = lease1->epoch++; 1039*fc0db77bSNamjae Jeon lease2->version = lease1->version; 104038c8a9a5SSteve French } 104138c8a9a5SSteve French 104238c8a9a5SSteve French static int add_lease_global_list(struct oplock_info *opinfo) 104338c8a9a5SSteve French { 104438c8a9a5SSteve French struct lease_table *lb; 104538c8a9a5SSteve French 104638c8a9a5SSteve French read_lock(&lease_list_lock); 104738c8a9a5SSteve French list_for_each_entry(lb, &lease_table_list, l_entry) { 104838c8a9a5SSteve French if (!memcmp(lb->client_guid, opinfo->conn->ClientGUID, 104938c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE)) { 105038c8a9a5SSteve French opinfo->o_lease->l_lb = lb; 105138c8a9a5SSteve French lease_add_list(opinfo); 105238c8a9a5SSteve French read_unlock(&lease_list_lock); 105338c8a9a5SSteve French return 0; 105438c8a9a5SSteve French } 105538c8a9a5SSteve French } 105638c8a9a5SSteve French read_unlock(&lease_list_lock); 105738c8a9a5SSteve French 105838c8a9a5SSteve French lb = kmalloc(sizeof(struct lease_table), GFP_KERNEL); 105938c8a9a5SSteve French if (!lb) 106038c8a9a5SSteve French return -ENOMEM; 106138c8a9a5SSteve French 106238c8a9a5SSteve French memcpy(lb->client_guid, opinfo->conn->ClientGUID, 106338c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE); 106438c8a9a5SSteve French INIT_LIST_HEAD(&lb->lease_list); 106538c8a9a5SSteve French spin_lock_init(&lb->lb_lock); 106638c8a9a5SSteve French opinfo->o_lease->l_lb = lb; 106738c8a9a5SSteve French lease_add_list(opinfo); 106838c8a9a5SSteve French lb_add(lb); 106938c8a9a5SSteve French return 0; 107038c8a9a5SSteve French } 107138c8a9a5SSteve French 107238c8a9a5SSteve French static void set_oplock_level(struct oplock_info *opinfo, int level, 107338c8a9a5SSteve French struct lease_ctx_info *lctx) 107438c8a9a5SSteve French { 107538c8a9a5SSteve French switch (level) { 107638c8a9a5SSteve French case SMB2_OPLOCK_LEVEL_BATCH: 107738c8a9a5SSteve French case SMB2_OPLOCK_LEVEL_EXCLUSIVE: 107838c8a9a5SSteve French grant_write_oplock(opinfo, level, lctx); 107938c8a9a5SSteve French break; 108038c8a9a5SSteve French case SMB2_OPLOCK_LEVEL_II: 108138c8a9a5SSteve French grant_read_oplock(opinfo, lctx); 108238c8a9a5SSteve French break; 108338c8a9a5SSteve French default: 108438c8a9a5SSteve French grant_none_oplock(opinfo, lctx); 108538c8a9a5SSteve French break; 108638c8a9a5SSteve French } 108738c8a9a5SSteve French } 108838c8a9a5SSteve French 10893c1e602aSNamjae Jeon void smb_send_parent_lease_break_noti(struct ksmbd_file *fp, 10903c1e602aSNamjae Jeon struct lease_ctx_info *lctx) 10913c1e602aSNamjae Jeon { 10923c1e602aSNamjae Jeon struct oplock_info *opinfo; 10933c1e602aSNamjae Jeon struct ksmbd_inode *p_ci = NULL; 10943c1e602aSNamjae Jeon 10953c1e602aSNamjae Jeon if (lctx->version != 2) 10963c1e602aSNamjae Jeon return; 10973c1e602aSNamjae Jeon 10983c1e602aSNamjae Jeon p_ci = ksmbd_inode_lookup_lock(fp->filp->f_path.dentry->d_parent); 10993c1e602aSNamjae Jeon if (!p_ci) 11003c1e602aSNamjae Jeon return; 11013c1e602aSNamjae Jeon 11023c1e602aSNamjae Jeon read_lock(&p_ci->m_lock); 11033c1e602aSNamjae Jeon list_for_each_entry(opinfo, &p_ci->m_op_list, op_entry) { 11043c1e602aSNamjae Jeon if (!opinfo->is_lease) 11053c1e602aSNamjae Jeon continue; 11063c1e602aSNamjae Jeon 11073c1e602aSNamjae Jeon if (opinfo->o_lease->state != SMB2_OPLOCK_LEVEL_NONE && 11083c1e602aSNamjae Jeon (!(lctx->flags & SMB2_LEASE_FLAG_PARENT_LEASE_KEY_SET_LE) || 11093c1e602aSNamjae Jeon !compare_guid_key(opinfo, fp->conn->ClientGUID, 11103c1e602aSNamjae Jeon lctx->parent_lease_key))) { 11113c1e602aSNamjae Jeon if (!atomic_inc_not_zero(&opinfo->refcount)) 11123c1e602aSNamjae Jeon continue; 11133c1e602aSNamjae Jeon 11143c1e602aSNamjae Jeon atomic_inc(&opinfo->conn->r_count); 11153c1e602aSNamjae Jeon if (ksmbd_conn_releasing(opinfo->conn)) { 11163c1e602aSNamjae Jeon atomic_dec(&opinfo->conn->r_count); 11173c1e602aSNamjae Jeon continue; 11183c1e602aSNamjae Jeon } 11193c1e602aSNamjae Jeon 11203c1e602aSNamjae Jeon read_unlock(&p_ci->m_lock); 11213c1e602aSNamjae Jeon oplock_break(opinfo, SMB2_OPLOCK_LEVEL_NONE); 11223c1e602aSNamjae Jeon opinfo_conn_put(opinfo); 11233c1e602aSNamjae Jeon read_lock(&p_ci->m_lock); 11243c1e602aSNamjae Jeon } 11253c1e602aSNamjae Jeon } 11263c1e602aSNamjae Jeon read_unlock(&p_ci->m_lock); 11273c1e602aSNamjae Jeon 11283c1e602aSNamjae Jeon ksmbd_inode_put(p_ci); 11293c1e602aSNamjae Jeon } 11303c1e602aSNamjae Jeon 1131ab5a0a1cSNamjae Jeon void smb_lazy_parent_lease_break_close(struct ksmbd_file *fp) 1132ab5a0a1cSNamjae Jeon { 1133ab5a0a1cSNamjae Jeon struct oplock_info *opinfo; 1134ab5a0a1cSNamjae Jeon struct ksmbd_inode *p_ci = NULL; 1135ab5a0a1cSNamjae Jeon 1136ab5a0a1cSNamjae Jeon rcu_read_lock(); 1137ab5a0a1cSNamjae Jeon opinfo = rcu_dereference(fp->f_opinfo); 1138ab5a0a1cSNamjae Jeon rcu_read_unlock(); 1139ab5a0a1cSNamjae Jeon 1140ab5a0a1cSNamjae Jeon if (!opinfo->is_lease || opinfo->o_lease->version != 2) 1141ab5a0a1cSNamjae Jeon return; 1142ab5a0a1cSNamjae Jeon 1143ab5a0a1cSNamjae Jeon p_ci = ksmbd_inode_lookup_lock(fp->filp->f_path.dentry->d_parent); 1144ab5a0a1cSNamjae Jeon if (!p_ci) 1145ab5a0a1cSNamjae Jeon return; 1146ab5a0a1cSNamjae Jeon 1147ab5a0a1cSNamjae Jeon read_lock(&p_ci->m_lock); 1148ab5a0a1cSNamjae Jeon list_for_each_entry(opinfo, &p_ci->m_op_list, op_entry) { 1149ab5a0a1cSNamjae Jeon if (!opinfo->is_lease) 1150ab5a0a1cSNamjae Jeon continue; 1151ab5a0a1cSNamjae Jeon 1152ab5a0a1cSNamjae Jeon if (opinfo->o_lease->state != SMB2_OPLOCK_LEVEL_NONE) { 1153ab5a0a1cSNamjae Jeon if (!atomic_inc_not_zero(&opinfo->refcount)) 1154ab5a0a1cSNamjae Jeon continue; 1155ab5a0a1cSNamjae Jeon 1156ab5a0a1cSNamjae Jeon atomic_inc(&opinfo->conn->r_count); 1157ab5a0a1cSNamjae Jeon if (ksmbd_conn_releasing(opinfo->conn)) { 1158ab5a0a1cSNamjae Jeon atomic_dec(&opinfo->conn->r_count); 1159ab5a0a1cSNamjae Jeon continue; 1160ab5a0a1cSNamjae Jeon } 1161ab5a0a1cSNamjae Jeon read_unlock(&p_ci->m_lock); 1162ab5a0a1cSNamjae Jeon oplock_break(opinfo, SMB2_OPLOCK_LEVEL_NONE); 1163ab5a0a1cSNamjae Jeon opinfo_conn_put(opinfo); 1164ab5a0a1cSNamjae Jeon read_lock(&p_ci->m_lock); 1165ab5a0a1cSNamjae Jeon } 1166ab5a0a1cSNamjae Jeon } 1167ab5a0a1cSNamjae Jeon read_unlock(&p_ci->m_lock); 1168ab5a0a1cSNamjae Jeon 1169ab5a0a1cSNamjae Jeon ksmbd_inode_put(p_ci); 1170ab5a0a1cSNamjae Jeon } 1171ab5a0a1cSNamjae Jeon 117238c8a9a5SSteve French /** 117338c8a9a5SSteve French * smb_grant_oplock() - handle oplock/lease request on file open 117438c8a9a5SSteve French * @work: smb work 117538c8a9a5SSteve French * @req_op_level: oplock level 117638c8a9a5SSteve French * @pid: id of open file 117738c8a9a5SSteve French * @fp: ksmbd file pointer 117838c8a9a5SSteve French * @tid: Tree id of connection 117938c8a9a5SSteve French * @lctx: lease context information on file open 118038c8a9a5SSteve French * @share_ret: share mode 118138c8a9a5SSteve French * 118238c8a9a5SSteve French * Return: 0 on success, otherwise error 118338c8a9a5SSteve French */ 118438c8a9a5SSteve French int smb_grant_oplock(struct ksmbd_work *work, int req_op_level, u64 pid, 118538c8a9a5SSteve French struct ksmbd_file *fp, __u16 tid, 118638c8a9a5SSteve French struct lease_ctx_info *lctx, int share_ret) 118738c8a9a5SSteve French { 118838c8a9a5SSteve French struct ksmbd_session *sess = work->sess; 118938c8a9a5SSteve French int err = 0; 119038c8a9a5SSteve French struct oplock_info *opinfo = NULL, *prev_opinfo = NULL; 119138c8a9a5SSteve French struct ksmbd_inode *ci = fp->f_ci; 119238c8a9a5SSteve French bool prev_op_has_lease; 119338c8a9a5SSteve French __le32 prev_op_state = 0; 119438c8a9a5SSteve French 119508e6c823SNamjae Jeon /* Only v2 leases handle the directory */ 119608e6c823SNamjae Jeon if (S_ISDIR(file_inode(fp->filp)->i_mode)) { 119708e6c823SNamjae Jeon if (!lctx || lctx->version != 2) 119808e6c823SNamjae Jeon return 0; 119908e6c823SNamjae Jeon } 120008e6c823SNamjae Jeon 120138c8a9a5SSteve French opinfo = alloc_opinfo(work, pid, tid); 120238c8a9a5SSteve French if (!opinfo) 120338c8a9a5SSteve French return -ENOMEM; 120438c8a9a5SSteve French 120538c8a9a5SSteve French if (lctx) { 120638c8a9a5SSteve French err = alloc_lease(opinfo, lctx); 120738c8a9a5SSteve French if (err) 120838c8a9a5SSteve French goto err_out; 120938c8a9a5SSteve French opinfo->is_lease = 1; 121038c8a9a5SSteve French } 121138c8a9a5SSteve French 121238c8a9a5SSteve French /* ci does not have any oplock */ 121338c8a9a5SSteve French if (!opinfo_count(fp)) 121438c8a9a5SSteve French goto set_lev; 121538c8a9a5SSteve French 121638c8a9a5SSteve French /* grant none-oplock if second open is trunc */ 121738c8a9a5SSteve French if (fp->attrib_only && fp->cdoption != FILE_OVERWRITE_IF_LE && 121838c8a9a5SSteve French fp->cdoption != FILE_OVERWRITE_LE && 121938c8a9a5SSteve French fp->cdoption != FILE_SUPERSEDE_LE) { 122038c8a9a5SSteve French req_op_level = SMB2_OPLOCK_LEVEL_NONE; 122138c8a9a5SSteve French goto set_lev; 122238c8a9a5SSteve French } 122338c8a9a5SSteve French 122438c8a9a5SSteve French if (lctx) { 122538c8a9a5SSteve French struct oplock_info *m_opinfo; 122638c8a9a5SSteve French 122738c8a9a5SSteve French /* is lease already granted ? */ 122838c8a9a5SSteve French m_opinfo = same_client_has_lease(ci, sess->ClientGUID, 122938c8a9a5SSteve French lctx); 123038c8a9a5SSteve French if (m_opinfo) { 123138c8a9a5SSteve French copy_lease(m_opinfo, opinfo); 123238c8a9a5SSteve French if (atomic_read(&m_opinfo->breaking_cnt)) 123338c8a9a5SSteve French opinfo->o_lease->flags = 123438c8a9a5SSteve French SMB2_LEASE_FLAG_BREAK_IN_PROGRESS_LE; 123538c8a9a5SSteve French goto out; 123638c8a9a5SSteve French } 123738c8a9a5SSteve French } 123838c8a9a5SSteve French prev_opinfo = opinfo_get_list(ci); 123938c8a9a5SSteve French if (!prev_opinfo || 124036322523SNamjae Jeon (prev_opinfo->level == SMB2_OPLOCK_LEVEL_NONE && lctx)) { 124136322523SNamjae Jeon opinfo_conn_put(prev_opinfo); 124238c8a9a5SSteve French goto set_lev; 124336322523SNamjae Jeon } 124438c8a9a5SSteve French prev_op_has_lease = prev_opinfo->is_lease; 124538c8a9a5SSteve French if (prev_op_has_lease) 124638c8a9a5SSteve French prev_op_state = prev_opinfo->o_lease->state; 124738c8a9a5SSteve French 124838c8a9a5SSteve French if (share_ret < 0 && 124938c8a9a5SSteve French prev_opinfo->level == SMB2_OPLOCK_LEVEL_EXCLUSIVE) { 125038c8a9a5SSteve French err = share_ret; 125136322523SNamjae Jeon opinfo_conn_put(prev_opinfo); 125238c8a9a5SSteve French goto err_out; 125338c8a9a5SSteve French } 125438c8a9a5SSteve French 125538c8a9a5SSteve French if (prev_opinfo->level != SMB2_OPLOCK_LEVEL_BATCH && 125638c8a9a5SSteve French prev_opinfo->level != SMB2_OPLOCK_LEVEL_EXCLUSIVE) { 125736322523SNamjae Jeon opinfo_conn_put(prev_opinfo); 125838c8a9a5SSteve French goto op_break_not_needed; 125938c8a9a5SSteve French } 126038c8a9a5SSteve French 126138c8a9a5SSteve French list_add(&work->interim_entry, &prev_opinfo->interim_list); 126238c8a9a5SSteve French err = oplock_break(prev_opinfo, SMB2_OPLOCK_LEVEL_II); 126336322523SNamjae Jeon opinfo_conn_put(prev_opinfo); 126438c8a9a5SSteve French if (err == -ENOENT) 126538c8a9a5SSteve French goto set_lev; 126638c8a9a5SSteve French /* Check all oplock was freed by close */ 126738c8a9a5SSteve French else if (err < 0) 126838c8a9a5SSteve French goto err_out; 126938c8a9a5SSteve French 127038c8a9a5SSteve French op_break_not_needed: 127138c8a9a5SSteve French if (share_ret < 0) { 127238c8a9a5SSteve French err = share_ret; 127338c8a9a5SSteve French goto err_out; 127438c8a9a5SSteve French } 127538c8a9a5SSteve French 127638c8a9a5SSteve French if (req_op_level != SMB2_OPLOCK_LEVEL_NONE) 127738c8a9a5SSteve French req_op_level = SMB2_OPLOCK_LEVEL_II; 127838c8a9a5SSteve French 127938c8a9a5SSteve French /* grant fixed oplock on stacked locking between lease and oplock */ 128038c8a9a5SSteve French if (prev_op_has_lease && !lctx) 128138c8a9a5SSteve French if (prev_op_state & SMB2_LEASE_HANDLE_CACHING_LE) 128238c8a9a5SSteve French req_op_level = SMB2_OPLOCK_LEVEL_NONE; 128338c8a9a5SSteve French 128438c8a9a5SSteve French if (!prev_op_has_lease && lctx) { 128538c8a9a5SSteve French req_op_level = SMB2_OPLOCK_LEVEL_II; 128638c8a9a5SSteve French lctx->req_state = SMB2_LEASE_READ_CACHING_LE; 128738c8a9a5SSteve French } 128838c8a9a5SSteve French 128938c8a9a5SSteve French set_lev: 129038c8a9a5SSteve French set_oplock_level(opinfo, req_op_level, lctx); 129138c8a9a5SSteve French 129238c8a9a5SSteve French out: 129338c8a9a5SSteve French rcu_assign_pointer(fp->f_opinfo, opinfo); 129438c8a9a5SSteve French opinfo->o_fp = fp; 129538c8a9a5SSteve French 129638c8a9a5SSteve French opinfo_count_inc(fp); 129738c8a9a5SSteve French opinfo_add(opinfo); 129838c8a9a5SSteve French if (opinfo->is_lease) { 129938c8a9a5SSteve French err = add_lease_global_list(opinfo); 130038c8a9a5SSteve French if (err) 130138c8a9a5SSteve French goto err_out; 130238c8a9a5SSteve French } 130338c8a9a5SSteve French 130438c8a9a5SSteve French return 0; 130538c8a9a5SSteve French err_out: 130638c8a9a5SSteve French free_opinfo(opinfo); 130738c8a9a5SSteve French return err; 130838c8a9a5SSteve French } 130938c8a9a5SSteve French 131038c8a9a5SSteve French /** 131138c8a9a5SSteve French * smb_break_all_write_oplock() - break batch/exclusive oplock to level2 131238c8a9a5SSteve French * @work: smb work 131338c8a9a5SSteve French * @fp: ksmbd file pointer 131438c8a9a5SSteve French * @is_trunc: truncate on open 131538c8a9a5SSteve French */ 131638c8a9a5SSteve French static void smb_break_all_write_oplock(struct ksmbd_work *work, 131738c8a9a5SSteve French struct ksmbd_file *fp, int is_trunc) 131838c8a9a5SSteve French { 131938c8a9a5SSteve French struct oplock_info *brk_opinfo; 132038c8a9a5SSteve French 132138c8a9a5SSteve French brk_opinfo = opinfo_get_list(fp->f_ci); 132238c8a9a5SSteve French if (!brk_opinfo) 132338c8a9a5SSteve French return; 132438c8a9a5SSteve French if (brk_opinfo->level != SMB2_OPLOCK_LEVEL_BATCH && 132538c8a9a5SSteve French brk_opinfo->level != SMB2_OPLOCK_LEVEL_EXCLUSIVE) { 132636322523SNamjae Jeon opinfo_conn_put(brk_opinfo); 132738c8a9a5SSteve French return; 132838c8a9a5SSteve French } 132938c8a9a5SSteve French 133038c8a9a5SSteve French brk_opinfo->open_trunc = is_trunc; 133138c8a9a5SSteve French list_add(&work->interim_entry, &brk_opinfo->interim_list); 133238c8a9a5SSteve French oplock_break(brk_opinfo, SMB2_OPLOCK_LEVEL_II); 133336322523SNamjae Jeon opinfo_conn_put(brk_opinfo); 133438c8a9a5SSteve French } 133538c8a9a5SSteve French 133638c8a9a5SSteve French /** 133738c8a9a5SSteve French * smb_break_all_levII_oplock() - send level2 oplock or read lease break command 133838c8a9a5SSteve French * from server to client 133938c8a9a5SSteve French * @work: smb work 134038c8a9a5SSteve French * @fp: ksmbd file pointer 134138c8a9a5SSteve French * @is_trunc: truncate on open 134238c8a9a5SSteve French */ 134338c8a9a5SSteve French void smb_break_all_levII_oplock(struct ksmbd_work *work, struct ksmbd_file *fp, 134438c8a9a5SSteve French int is_trunc) 134538c8a9a5SSteve French { 134638c8a9a5SSteve French struct oplock_info *op, *brk_op; 134738c8a9a5SSteve French struct ksmbd_inode *ci; 134838c8a9a5SSteve French struct ksmbd_conn *conn = work->conn; 134938c8a9a5SSteve French 135038c8a9a5SSteve French if (!test_share_config_flag(work->tcon->share_conf, 135138c8a9a5SSteve French KSMBD_SHARE_FLAG_OPLOCKS)) 135238c8a9a5SSteve French return; 135338c8a9a5SSteve French 135438c8a9a5SSteve French ci = fp->f_ci; 135538c8a9a5SSteve French op = opinfo_get(fp); 135638c8a9a5SSteve French 135738c8a9a5SSteve French rcu_read_lock(); 135838c8a9a5SSteve French list_for_each_entry_rcu(brk_op, &ci->m_op_list, op_entry) { 135938c8a9a5SSteve French if (!atomic_inc_not_zero(&brk_op->refcount)) 136038c8a9a5SSteve French continue; 136136322523SNamjae Jeon 136236322523SNamjae Jeon atomic_inc(&brk_op->conn->r_count); 136336322523SNamjae Jeon if (ksmbd_conn_releasing(brk_op->conn)) { 136436322523SNamjae Jeon atomic_dec(&brk_op->conn->r_count); 136536322523SNamjae Jeon continue; 136636322523SNamjae Jeon } 136736322523SNamjae Jeon 136838c8a9a5SSteve French rcu_read_unlock(); 136938c8a9a5SSteve French if (brk_op->is_lease && (brk_op->o_lease->state & 137038c8a9a5SSteve French (~(SMB2_LEASE_READ_CACHING_LE | 137138c8a9a5SSteve French SMB2_LEASE_HANDLE_CACHING_LE)))) { 137238c8a9a5SSteve French ksmbd_debug(OPLOCK, "unexpected lease state(0x%x)\n", 137338c8a9a5SSteve French brk_op->o_lease->state); 137438c8a9a5SSteve French goto next; 137538c8a9a5SSteve French } else if (brk_op->level != 137638c8a9a5SSteve French SMB2_OPLOCK_LEVEL_II) { 137738c8a9a5SSteve French ksmbd_debug(OPLOCK, "unexpected oplock(0x%x)\n", 137838c8a9a5SSteve French brk_op->level); 137938c8a9a5SSteve French goto next; 138038c8a9a5SSteve French } 138138c8a9a5SSteve French 138238c8a9a5SSteve French /* Skip oplock being break to none */ 138338c8a9a5SSteve French if (brk_op->is_lease && 138438c8a9a5SSteve French brk_op->o_lease->new_state == SMB2_LEASE_NONE_LE && 138538c8a9a5SSteve French atomic_read(&brk_op->breaking_cnt)) 138638c8a9a5SSteve French goto next; 138738c8a9a5SSteve French 138838c8a9a5SSteve French if (op && op->is_lease && brk_op->is_lease && 138938c8a9a5SSteve French !memcmp(conn->ClientGUID, brk_op->conn->ClientGUID, 139038c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE) && 139138c8a9a5SSteve French !memcmp(op->o_lease->lease_key, brk_op->o_lease->lease_key, 139238c8a9a5SSteve French SMB2_LEASE_KEY_SIZE)) 139338c8a9a5SSteve French goto next; 139438c8a9a5SSteve French brk_op->open_trunc = is_trunc; 139538c8a9a5SSteve French oplock_break(brk_op, SMB2_OPLOCK_LEVEL_NONE); 139638c8a9a5SSteve French next: 139736322523SNamjae Jeon opinfo_conn_put(brk_op); 139838c8a9a5SSteve French rcu_read_lock(); 139938c8a9a5SSteve French } 140038c8a9a5SSteve French rcu_read_unlock(); 140138c8a9a5SSteve French 140238c8a9a5SSteve French if (op) 140338c8a9a5SSteve French opinfo_put(op); 140438c8a9a5SSteve French } 140538c8a9a5SSteve French 140638c8a9a5SSteve French /** 140738c8a9a5SSteve French * smb_break_all_oplock() - break both batch/exclusive and level2 oplock 140838c8a9a5SSteve French * @work: smb work 140938c8a9a5SSteve French * @fp: ksmbd file pointer 141038c8a9a5SSteve French */ 141138c8a9a5SSteve French void smb_break_all_oplock(struct ksmbd_work *work, struct ksmbd_file *fp) 141238c8a9a5SSteve French { 141338c8a9a5SSteve French if (!test_share_config_flag(work->tcon->share_conf, 141438c8a9a5SSteve French KSMBD_SHARE_FLAG_OPLOCKS)) 141538c8a9a5SSteve French return; 141638c8a9a5SSteve French 141738c8a9a5SSteve French smb_break_all_write_oplock(work, fp, 1); 141838c8a9a5SSteve French smb_break_all_levII_oplock(work, fp, 1); 141938c8a9a5SSteve French } 142038c8a9a5SSteve French 142138c8a9a5SSteve French /** 142238c8a9a5SSteve French * smb2_map_lease_to_oplock() - map lease state to corresponding oplock type 142338c8a9a5SSteve French * @lease_state: lease type 142438c8a9a5SSteve French * 142538c8a9a5SSteve French * Return: 0 if no mapping, otherwise corresponding oplock type 142638c8a9a5SSteve French */ 142738c8a9a5SSteve French __u8 smb2_map_lease_to_oplock(__le32 lease_state) 142838c8a9a5SSteve French { 142938c8a9a5SSteve French if (lease_state == (SMB2_LEASE_HANDLE_CACHING_LE | 143038c8a9a5SSteve French SMB2_LEASE_READ_CACHING_LE | 143138c8a9a5SSteve French SMB2_LEASE_WRITE_CACHING_LE)) { 143238c8a9a5SSteve French return SMB2_OPLOCK_LEVEL_BATCH; 143338c8a9a5SSteve French } else if (lease_state != SMB2_LEASE_WRITE_CACHING_LE && 143438c8a9a5SSteve French lease_state & SMB2_LEASE_WRITE_CACHING_LE) { 143538c8a9a5SSteve French if (!(lease_state & SMB2_LEASE_HANDLE_CACHING_LE)) 143638c8a9a5SSteve French return SMB2_OPLOCK_LEVEL_EXCLUSIVE; 143738c8a9a5SSteve French } else if (lease_state & SMB2_LEASE_READ_CACHING_LE) { 143838c8a9a5SSteve French return SMB2_OPLOCK_LEVEL_II; 143938c8a9a5SSteve French } 144038c8a9a5SSteve French return 0; 144138c8a9a5SSteve French } 144238c8a9a5SSteve French 144338c8a9a5SSteve French /** 144438c8a9a5SSteve French * create_lease_buf() - create lease context for open cmd response 144538c8a9a5SSteve French * @rbuf: buffer to create lease context response 144638c8a9a5SSteve French * @lease: buffer to stored parsed lease state information 144738c8a9a5SSteve French */ 144838c8a9a5SSteve French void create_lease_buf(u8 *rbuf, struct lease *lease) 144938c8a9a5SSteve French { 145038c8a9a5SSteve French if (lease->version == 2) { 145138c8a9a5SSteve French struct create_lease_v2 *buf = (struct create_lease_v2 *)rbuf; 145238c8a9a5SSteve French 145338c8a9a5SSteve French memset(buf, 0, sizeof(struct create_lease_v2)); 145438c8a9a5SSteve French memcpy(buf->lcontext.LeaseKey, lease->lease_key, 145538c8a9a5SSteve French SMB2_LEASE_KEY_SIZE); 145638c8a9a5SSteve French buf->lcontext.LeaseFlags = lease->flags; 1457bc025d49SNamjae Jeon buf->lcontext.Epoch = cpu_to_le16(++lease->epoch); 145838c8a9a5SSteve French buf->lcontext.LeaseState = lease->state; 145938c8a9a5SSteve French memcpy(buf->lcontext.ParentLeaseKey, lease->parent_lease_key, 146038c8a9a5SSteve French SMB2_LEASE_KEY_SIZE); 146138c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 146238c8a9a5SSteve French (struct create_lease_v2, lcontext)); 146338c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context_v2)); 146438c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 146538c8a9a5SSteve French (struct create_lease_v2, Name)); 146638c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 146738c8a9a5SSteve French buf->Name[0] = 'R'; 146838c8a9a5SSteve French buf->Name[1] = 'q'; 146938c8a9a5SSteve French buf->Name[2] = 'L'; 147038c8a9a5SSteve French buf->Name[3] = 's'; 147138c8a9a5SSteve French } else { 147238c8a9a5SSteve French struct create_lease *buf = (struct create_lease *)rbuf; 147338c8a9a5SSteve French 147438c8a9a5SSteve French memset(buf, 0, sizeof(struct create_lease)); 147538c8a9a5SSteve French memcpy(buf->lcontext.LeaseKey, lease->lease_key, SMB2_LEASE_KEY_SIZE); 147638c8a9a5SSteve French buf->lcontext.LeaseFlags = lease->flags; 147738c8a9a5SSteve French buf->lcontext.LeaseState = lease->state; 147838c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 147938c8a9a5SSteve French (struct create_lease, lcontext)); 148038c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context)); 148138c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 148238c8a9a5SSteve French (struct create_lease, Name)); 148338c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 148438c8a9a5SSteve French buf->Name[0] = 'R'; 148538c8a9a5SSteve French buf->Name[1] = 'q'; 148638c8a9a5SSteve French buf->Name[2] = 'L'; 148738c8a9a5SSteve French buf->Name[3] = 's'; 148838c8a9a5SSteve French } 148938c8a9a5SSteve French } 149038c8a9a5SSteve French 149138c8a9a5SSteve French /** 149238c8a9a5SSteve French * parse_lease_state() - parse lease context containted in file open request 149338c8a9a5SSteve French * @open_req: buffer containing smb2 file open(create) request 1494572388ffSNamjae Jeon * @is_dir: whether leasing file is directory 149538c8a9a5SSteve French * 149638c8a9a5SSteve French * Return: oplock state, -ENOENT if create lease context not found 149738c8a9a5SSteve French */ 1498572388ffSNamjae Jeon struct lease_ctx_info *parse_lease_state(void *open_req, bool is_dir) 149938c8a9a5SSteve French { 150038c8a9a5SSteve French struct create_context *cc; 150138c8a9a5SSteve French struct smb2_create_req *req = (struct smb2_create_req *)open_req; 1502fc6c6a3cSNamjae Jeon struct lease_ctx_info *lreq; 1503fc6c6a3cSNamjae Jeon 1504fc6c6a3cSNamjae Jeon cc = smb2_find_context_vals(req, SMB2_CREATE_REQUEST_LEASE, 4); 1505fc6c6a3cSNamjae Jeon if (IS_ERR_OR_NULL(cc)) 1506fc6c6a3cSNamjae Jeon return NULL; 1507fc6c6a3cSNamjae Jeon 1508fc6c6a3cSNamjae Jeon lreq = kzalloc(sizeof(struct lease_ctx_info), GFP_KERNEL); 150938c8a9a5SSteve French if (!lreq) 151038c8a9a5SSteve French return NULL; 151138c8a9a5SSteve French 151238c8a9a5SSteve French if (sizeof(struct lease_context_v2) == le32_to_cpu(cc->DataLength)) { 151338c8a9a5SSteve French struct create_lease_v2 *lc = (struct create_lease_v2 *)cc; 151438c8a9a5SSteve French 151538c8a9a5SSteve French memcpy(lreq->lease_key, lc->lcontext.LeaseKey, SMB2_LEASE_KEY_SIZE); 15163c1e602aSNamjae Jeon if (is_dir) { 1517572388ffSNamjae Jeon lreq->req_state = lc->lcontext.LeaseState & 1518572388ffSNamjae Jeon ~SMB2_LEASE_WRITE_CACHING_LE; 15193c1e602aSNamjae Jeon lreq->is_dir = true; 15203c1e602aSNamjae Jeon } else 152138c8a9a5SSteve French lreq->req_state = lc->lcontext.LeaseState; 152238c8a9a5SSteve French lreq->flags = lc->lcontext.LeaseFlags; 1523bc025d49SNamjae Jeon lreq->epoch = lc->lcontext.Epoch; 152438c8a9a5SSteve French lreq->duration = lc->lcontext.LeaseDuration; 152538c8a9a5SSteve French memcpy(lreq->parent_lease_key, lc->lcontext.ParentLeaseKey, 152638c8a9a5SSteve French SMB2_LEASE_KEY_SIZE); 152738c8a9a5SSteve French lreq->version = 2; 152838c8a9a5SSteve French } else { 152938c8a9a5SSteve French struct create_lease *lc = (struct create_lease *)cc; 153038c8a9a5SSteve French 153138c8a9a5SSteve French memcpy(lreq->lease_key, lc->lcontext.LeaseKey, SMB2_LEASE_KEY_SIZE); 153238c8a9a5SSteve French lreq->req_state = lc->lcontext.LeaseState; 153338c8a9a5SSteve French lreq->flags = lc->lcontext.LeaseFlags; 153438c8a9a5SSteve French lreq->duration = lc->lcontext.LeaseDuration; 153538c8a9a5SSteve French lreq->version = 1; 153638c8a9a5SSteve French } 153738c8a9a5SSteve French return lreq; 153838c8a9a5SSteve French } 153938c8a9a5SSteve French 154038c8a9a5SSteve French /** 154138c8a9a5SSteve French * smb2_find_context_vals() - find a particular context info in open request 154238c8a9a5SSteve French * @open_req: buffer containing smb2 file open(create) request 154338c8a9a5SSteve French * @tag: context name to search for 154438c8a9a5SSteve French * @tag_len: the length of tag 154538c8a9a5SSteve French * 154638c8a9a5SSteve French * Return: pointer to requested context, NULL if @str context not found 154738c8a9a5SSteve French * or error pointer if name length is invalid. 154838c8a9a5SSteve French */ 154938c8a9a5SSteve French struct create_context *smb2_find_context_vals(void *open_req, const char *tag, int tag_len) 155038c8a9a5SSteve French { 155138c8a9a5SSteve French struct create_context *cc; 155238c8a9a5SSteve French unsigned int next = 0; 155338c8a9a5SSteve French char *name; 155438c8a9a5SSteve French struct smb2_create_req *req = (struct smb2_create_req *)open_req; 155538c8a9a5SSteve French unsigned int remain_len, name_off, name_len, value_off, value_len, 155638c8a9a5SSteve French cc_len; 155738c8a9a5SSteve French 155838c8a9a5SSteve French /* 155938c8a9a5SSteve French * CreateContextsOffset and CreateContextsLength are guaranteed to 156038c8a9a5SSteve French * be valid because of ksmbd_smb2_check_message(). 156138c8a9a5SSteve French */ 156238c8a9a5SSteve French cc = (struct create_context *)((char *)req + 156338c8a9a5SSteve French le32_to_cpu(req->CreateContextsOffset)); 156438c8a9a5SSteve French remain_len = le32_to_cpu(req->CreateContextsLength); 156538c8a9a5SSteve French do { 156638c8a9a5SSteve French cc = (struct create_context *)((char *)cc + next); 156738c8a9a5SSteve French if (remain_len < offsetof(struct create_context, Buffer)) 156838c8a9a5SSteve French return ERR_PTR(-EINVAL); 156938c8a9a5SSteve French 157038c8a9a5SSteve French next = le32_to_cpu(cc->Next); 157138c8a9a5SSteve French name_off = le16_to_cpu(cc->NameOffset); 157238c8a9a5SSteve French name_len = le16_to_cpu(cc->NameLength); 157338c8a9a5SSteve French value_off = le16_to_cpu(cc->DataOffset); 157438c8a9a5SSteve French value_len = le32_to_cpu(cc->DataLength); 157538c8a9a5SSteve French cc_len = next ? next : remain_len; 157638c8a9a5SSteve French 157738c8a9a5SSteve French if ((next & 0x7) != 0 || 157838c8a9a5SSteve French next > remain_len || 157938c8a9a5SSteve French name_off != offsetof(struct create_context, Buffer) || 158038c8a9a5SSteve French name_len < 4 || 158138c8a9a5SSteve French name_off + name_len > cc_len || 158238c8a9a5SSteve French (value_off & 0x7) != 0 || 158317d5b135SNamjae Jeon (value_len && value_off < name_off + (name_len < 8 ? 8 : name_len)) || 158438c8a9a5SSteve French ((u64)value_off + value_len > cc_len)) 158538c8a9a5SSteve French return ERR_PTR(-EINVAL); 158638c8a9a5SSteve French 158738c8a9a5SSteve French name = (char *)cc + name_off; 158838c8a9a5SSteve French if (name_len == tag_len && !memcmp(name, tag, name_len)) 158938c8a9a5SSteve French return cc; 159038c8a9a5SSteve French 159138c8a9a5SSteve French remain_len -= next; 159238c8a9a5SSteve French } while (next != 0); 159338c8a9a5SSteve French 159438c8a9a5SSteve French return NULL; 159538c8a9a5SSteve French } 159638c8a9a5SSteve French 159738c8a9a5SSteve French /** 159838c8a9a5SSteve French * create_durable_rsp_buf() - create durable handle context 159938c8a9a5SSteve French * @cc: buffer to create durable context response 160038c8a9a5SSteve French */ 160138c8a9a5SSteve French void create_durable_rsp_buf(char *cc) 160238c8a9a5SSteve French { 160338c8a9a5SSteve French struct create_durable_rsp *buf; 160438c8a9a5SSteve French 160538c8a9a5SSteve French buf = (struct create_durable_rsp *)cc; 160638c8a9a5SSteve French memset(buf, 0, sizeof(struct create_durable_rsp)); 160738c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 160838c8a9a5SSteve French (struct create_durable_rsp, Data)); 160938c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(8); 161038c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 161138c8a9a5SSteve French (struct create_durable_rsp, Name)); 161238c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 161338c8a9a5SSteve French /* SMB2_CREATE_DURABLE_HANDLE_RESPONSE is "DHnQ" */ 161438c8a9a5SSteve French buf->Name[0] = 'D'; 161538c8a9a5SSteve French buf->Name[1] = 'H'; 161638c8a9a5SSteve French buf->Name[2] = 'n'; 161738c8a9a5SSteve French buf->Name[3] = 'Q'; 161838c8a9a5SSteve French } 161938c8a9a5SSteve French 162038c8a9a5SSteve French /** 162138c8a9a5SSteve French * create_durable_v2_rsp_buf() - create durable handle v2 context 162238c8a9a5SSteve French * @cc: buffer to create durable context response 162338c8a9a5SSteve French * @fp: ksmbd file pointer 162438c8a9a5SSteve French */ 162538c8a9a5SSteve French void create_durable_v2_rsp_buf(char *cc, struct ksmbd_file *fp) 162638c8a9a5SSteve French { 162738c8a9a5SSteve French struct create_durable_v2_rsp *buf; 162838c8a9a5SSteve French 162938c8a9a5SSteve French buf = (struct create_durable_v2_rsp *)cc; 163038c8a9a5SSteve French memset(buf, 0, sizeof(struct create_durable_rsp)); 163138c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 163238c8a9a5SSteve French (struct create_durable_rsp, Data)); 163338c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(8); 163438c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 163538c8a9a5SSteve French (struct create_durable_rsp, Name)); 163638c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 163738c8a9a5SSteve French /* SMB2_CREATE_DURABLE_HANDLE_RESPONSE_V2 is "DH2Q" */ 163838c8a9a5SSteve French buf->Name[0] = 'D'; 163938c8a9a5SSteve French buf->Name[1] = 'H'; 164038c8a9a5SSteve French buf->Name[2] = '2'; 164138c8a9a5SSteve French buf->Name[3] = 'Q'; 164238c8a9a5SSteve French 164338c8a9a5SSteve French buf->Timeout = cpu_to_le32(fp->durable_timeout); 164438c8a9a5SSteve French } 164538c8a9a5SSteve French 164638c8a9a5SSteve French /** 164738c8a9a5SSteve French * create_mxac_rsp_buf() - create query maximal access context 164838c8a9a5SSteve French * @cc: buffer to create maximal access context response 164938c8a9a5SSteve French * @maximal_access: maximal access 165038c8a9a5SSteve French */ 165138c8a9a5SSteve French void create_mxac_rsp_buf(char *cc, int maximal_access) 165238c8a9a5SSteve French { 165338c8a9a5SSteve French struct create_mxac_rsp *buf; 165438c8a9a5SSteve French 165538c8a9a5SSteve French buf = (struct create_mxac_rsp *)cc; 165638c8a9a5SSteve French memset(buf, 0, sizeof(struct create_mxac_rsp)); 165738c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 165838c8a9a5SSteve French (struct create_mxac_rsp, QueryStatus)); 165938c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(8); 166038c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 166138c8a9a5SSteve French (struct create_mxac_rsp, Name)); 166238c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 166338c8a9a5SSteve French /* SMB2_CREATE_QUERY_MAXIMAL_ACCESS_RESPONSE is "MxAc" */ 166438c8a9a5SSteve French buf->Name[0] = 'M'; 166538c8a9a5SSteve French buf->Name[1] = 'x'; 166638c8a9a5SSteve French buf->Name[2] = 'A'; 166738c8a9a5SSteve French buf->Name[3] = 'c'; 166838c8a9a5SSteve French 166938c8a9a5SSteve French buf->QueryStatus = STATUS_SUCCESS; 167038c8a9a5SSteve French buf->MaximalAccess = cpu_to_le32(maximal_access); 167138c8a9a5SSteve French } 167238c8a9a5SSteve French 167338c8a9a5SSteve French void create_disk_id_rsp_buf(char *cc, __u64 file_id, __u64 vol_id) 167438c8a9a5SSteve French { 167538c8a9a5SSteve French struct create_disk_id_rsp *buf; 167638c8a9a5SSteve French 167738c8a9a5SSteve French buf = (struct create_disk_id_rsp *)cc; 167838c8a9a5SSteve French memset(buf, 0, sizeof(struct create_disk_id_rsp)); 167938c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 168038c8a9a5SSteve French (struct create_disk_id_rsp, DiskFileId)); 168138c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(32); 168238c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 168338c8a9a5SSteve French (struct create_mxac_rsp, Name)); 168438c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 168538c8a9a5SSteve French /* SMB2_CREATE_QUERY_ON_DISK_ID_RESPONSE is "QFid" */ 168638c8a9a5SSteve French buf->Name[0] = 'Q'; 168738c8a9a5SSteve French buf->Name[1] = 'F'; 168838c8a9a5SSteve French buf->Name[2] = 'i'; 168938c8a9a5SSteve French buf->Name[3] = 'd'; 169038c8a9a5SSteve French 169138c8a9a5SSteve French buf->DiskFileId = cpu_to_le64(file_id); 169238c8a9a5SSteve French buf->VolumeId = cpu_to_le64(vol_id); 169338c8a9a5SSteve French } 169438c8a9a5SSteve French 169538c8a9a5SSteve French /** 169638c8a9a5SSteve French * create_posix_rsp_buf() - create posix extension context 169738c8a9a5SSteve French * @cc: buffer to create posix on posix response 169838c8a9a5SSteve French * @fp: ksmbd file pointer 169938c8a9a5SSteve French */ 170038c8a9a5SSteve French void create_posix_rsp_buf(char *cc, struct ksmbd_file *fp) 170138c8a9a5SSteve French { 170238c8a9a5SSteve French struct create_posix_rsp *buf; 170338c8a9a5SSteve French struct inode *inode = file_inode(fp->filp); 170438c8a9a5SSteve French struct mnt_idmap *idmap = file_mnt_idmap(fp->filp); 170538c8a9a5SSteve French vfsuid_t vfsuid = i_uid_into_vfsuid(idmap, inode); 170638c8a9a5SSteve French vfsgid_t vfsgid = i_gid_into_vfsgid(idmap, inode); 170738c8a9a5SSteve French 170838c8a9a5SSteve French buf = (struct create_posix_rsp *)cc; 170938c8a9a5SSteve French memset(buf, 0, sizeof(struct create_posix_rsp)); 171038c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 171138c8a9a5SSteve French (struct create_posix_rsp, nlink)); 171238c8a9a5SSteve French /* 171338c8a9a5SSteve French * DataLength = nlink(4) + reparse_tag(4) + mode(4) + 171438c8a9a5SSteve French * domain sid(28) + unix group sid(16). 171538c8a9a5SSteve French */ 171638c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(56); 171738c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 171838c8a9a5SSteve French (struct create_posix_rsp, Name)); 171938c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(POSIX_CTXT_DATA_LEN); 172038c8a9a5SSteve French /* SMB2_CREATE_TAG_POSIX is "0x93AD25509CB411E7B42383DE968BCD7C" */ 172138c8a9a5SSteve French buf->Name[0] = 0x93; 172238c8a9a5SSteve French buf->Name[1] = 0xAD; 172338c8a9a5SSteve French buf->Name[2] = 0x25; 172438c8a9a5SSteve French buf->Name[3] = 0x50; 172538c8a9a5SSteve French buf->Name[4] = 0x9C; 172638c8a9a5SSteve French buf->Name[5] = 0xB4; 172738c8a9a5SSteve French buf->Name[6] = 0x11; 172838c8a9a5SSteve French buf->Name[7] = 0xE7; 172938c8a9a5SSteve French buf->Name[8] = 0xB4; 173038c8a9a5SSteve French buf->Name[9] = 0x23; 173138c8a9a5SSteve French buf->Name[10] = 0x83; 173238c8a9a5SSteve French buf->Name[11] = 0xDE; 173338c8a9a5SSteve French buf->Name[12] = 0x96; 173438c8a9a5SSteve French buf->Name[13] = 0x8B; 173538c8a9a5SSteve French buf->Name[14] = 0xCD; 173638c8a9a5SSteve French buf->Name[15] = 0x7C; 173738c8a9a5SSteve French 173838c8a9a5SSteve French buf->nlink = cpu_to_le32(inode->i_nlink); 173938c8a9a5SSteve French buf->reparse_tag = cpu_to_le32(fp->volatile_id); 174038c8a9a5SSteve French buf->mode = cpu_to_le32(inode->i_mode & 0777); 174138c8a9a5SSteve French /* 174238c8a9a5SSteve French * SidBuffer(44) contain two sids(Domain sid(28), UNIX group sid(16)). 174338c8a9a5SSteve French * Domain sid(28) = revision(1) + num_subauth(1) + authority(6) + 174438c8a9a5SSteve French * sub_auth(4 * 4(num_subauth)) + RID(4). 174538c8a9a5SSteve French * UNIX group id(16) = revision(1) + num_subauth(1) + authority(6) + 174638c8a9a5SSteve French * sub_auth(4 * 1(num_subauth)) + RID(4). 174738c8a9a5SSteve French */ 174838c8a9a5SSteve French id_to_sid(from_kuid_munged(&init_user_ns, vfsuid_into_kuid(vfsuid)), 174938c8a9a5SSteve French SIDOWNER, (struct smb_sid *)&buf->SidBuffer[0]); 175038c8a9a5SSteve French id_to_sid(from_kgid_munged(&init_user_ns, vfsgid_into_kgid(vfsgid)), 175138c8a9a5SSteve French SIDUNIX_GROUP, (struct smb_sid *)&buf->SidBuffer[28]); 175238c8a9a5SSteve French } 175338c8a9a5SSteve French 175438c8a9a5SSteve French /* 175538c8a9a5SSteve French * Find lease object(opinfo) for given lease key/fid from lease 175638c8a9a5SSteve French * break/file close path. 175738c8a9a5SSteve French */ 175838c8a9a5SSteve French /** 175938c8a9a5SSteve French * lookup_lease_in_table() - find a matching lease info object 176038c8a9a5SSteve French * @conn: connection instance 176138c8a9a5SSteve French * @lease_key: lease key to be searched for 176238c8a9a5SSteve French * 176338c8a9a5SSteve French * Return: opinfo if found matching opinfo, otherwise NULL 176438c8a9a5SSteve French */ 176538c8a9a5SSteve French struct oplock_info *lookup_lease_in_table(struct ksmbd_conn *conn, 176638c8a9a5SSteve French char *lease_key) 176738c8a9a5SSteve French { 176838c8a9a5SSteve French struct oplock_info *opinfo = NULL, *ret_op = NULL; 176938c8a9a5SSteve French struct lease_table *lt; 177038c8a9a5SSteve French int ret; 177138c8a9a5SSteve French 177238c8a9a5SSteve French read_lock(&lease_list_lock); 177338c8a9a5SSteve French list_for_each_entry(lt, &lease_table_list, l_entry) { 177438c8a9a5SSteve French if (!memcmp(lt->client_guid, conn->ClientGUID, 177538c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE)) 177638c8a9a5SSteve French goto found; 177738c8a9a5SSteve French } 177838c8a9a5SSteve French 177938c8a9a5SSteve French read_unlock(&lease_list_lock); 178038c8a9a5SSteve French return NULL; 178138c8a9a5SSteve French 178238c8a9a5SSteve French found: 178338c8a9a5SSteve French rcu_read_lock(); 178438c8a9a5SSteve French list_for_each_entry_rcu(opinfo, <->lease_list, lease_entry) { 178538c8a9a5SSteve French if (!atomic_inc_not_zero(&opinfo->refcount)) 178638c8a9a5SSteve French continue; 178738c8a9a5SSteve French rcu_read_unlock(); 178838c8a9a5SSteve French if (!opinfo->op_state || opinfo->op_state == OPLOCK_CLOSING) 178938c8a9a5SSteve French goto op_next; 179038c8a9a5SSteve French if (!(opinfo->o_lease->state & 179138c8a9a5SSteve French (SMB2_LEASE_HANDLE_CACHING_LE | 179238c8a9a5SSteve French SMB2_LEASE_WRITE_CACHING_LE))) 179338c8a9a5SSteve French goto op_next; 179438c8a9a5SSteve French ret = compare_guid_key(opinfo, conn->ClientGUID, 179538c8a9a5SSteve French lease_key); 179638c8a9a5SSteve French if (ret) { 179738c8a9a5SSteve French ksmbd_debug(OPLOCK, "found opinfo\n"); 179838c8a9a5SSteve French ret_op = opinfo; 179938c8a9a5SSteve French goto out; 180038c8a9a5SSteve French } 180138c8a9a5SSteve French op_next: 180238c8a9a5SSteve French opinfo_put(opinfo); 180338c8a9a5SSteve French rcu_read_lock(); 180438c8a9a5SSteve French } 180538c8a9a5SSteve French rcu_read_unlock(); 180638c8a9a5SSteve French 180738c8a9a5SSteve French out: 180838c8a9a5SSteve French read_unlock(&lease_list_lock); 180938c8a9a5SSteve French return ret_op; 181038c8a9a5SSteve French } 1811