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; 1089554d493SNamjae Jeon lease->epoch = le16_to_cpu(lctx->epoch) + 1; 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)) { 549f5ef78c4SNamjae Jeon lease->epoch++; 55038c8a9a5SSteve French lease->state |= lctx->req_state; 55138c8a9a5SSteve French if (lctx->req_state & 55238c8a9a5SSteve French SMB2_LEASE_WRITE_CACHING_LE) 55338c8a9a5SSteve French lease_read_to_write(opinfo); 5543c1e602aSNamjae Jeon 55538c8a9a5SSteve French } 55638c8a9a5SSteve French } else if ((atomic_read(&ci->op_count) + 55738c8a9a5SSteve French atomic_read(&ci->sop_count)) > 1) { 55838c8a9a5SSteve French if (lctx->req_state == 55938c8a9a5SSteve French (SMB2_LEASE_READ_CACHING_LE | 560f5ef78c4SNamjae Jeon SMB2_LEASE_HANDLE_CACHING_LE)) { 561f5ef78c4SNamjae Jeon lease->epoch++; 56238c8a9a5SSteve French lease->state = lctx->req_state; 56338c8a9a5SSteve French } 564f5ef78c4SNamjae Jeon } 56538c8a9a5SSteve French 56638c8a9a5SSteve French if (lctx->req_state && lease->state == 567f5ef78c4SNamjae Jeon SMB2_LEASE_NONE_LE) { 568f5ef78c4SNamjae Jeon lease->epoch++; 56938c8a9a5SSteve French lease_none_upgrade(opinfo, lctx->req_state); 57038c8a9a5SSteve French } 571f5ef78c4SNamjae Jeon } 57238c8a9a5SSteve French read_lock(&ci->m_lock); 57338c8a9a5SSteve French } 57438c8a9a5SSteve French read_unlock(&ci->m_lock); 57538c8a9a5SSteve French 57638c8a9a5SSteve French return m_opinfo; 57738c8a9a5SSteve French } 57838c8a9a5SSteve French 57938c8a9a5SSteve French static void wait_for_break_ack(struct oplock_info *opinfo) 58038c8a9a5SSteve French { 58138c8a9a5SSteve French int rc = 0; 58238c8a9a5SSteve French 58338c8a9a5SSteve French rc = wait_event_interruptible_timeout(opinfo->oplock_q, 58438c8a9a5SSteve French opinfo->op_state == OPLOCK_STATE_NONE || 58538c8a9a5SSteve French opinfo->op_state == OPLOCK_CLOSING, 58638c8a9a5SSteve French OPLOCK_WAIT_TIME); 58738c8a9a5SSteve French 58838c8a9a5SSteve French /* is this a timeout ? */ 58938c8a9a5SSteve French if (!rc) { 59038c8a9a5SSteve French if (opinfo->is_lease) 59138c8a9a5SSteve French opinfo->o_lease->state = SMB2_LEASE_NONE_LE; 59238c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_NONE; 59338c8a9a5SSteve French opinfo->op_state = OPLOCK_STATE_NONE; 59438c8a9a5SSteve French } 59538c8a9a5SSteve French } 59638c8a9a5SSteve French 59738c8a9a5SSteve French static void wake_up_oplock_break(struct oplock_info *opinfo) 59838c8a9a5SSteve French { 59938c8a9a5SSteve French clear_bit_unlock(0, &opinfo->pending_break); 60038c8a9a5SSteve French /* memory barrier is needed for wake_up_bit() */ 60138c8a9a5SSteve French smp_mb__after_atomic(); 60238c8a9a5SSteve French wake_up_bit(&opinfo->pending_break, 0); 60338c8a9a5SSteve French } 60438c8a9a5SSteve French 60538c8a9a5SSteve French static int oplock_break_pending(struct oplock_info *opinfo, int req_op_level) 60638c8a9a5SSteve French { 60738c8a9a5SSteve French while (test_and_set_bit(0, &opinfo->pending_break)) { 60838c8a9a5SSteve French wait_on_bit(&opinfo->pending_break, 0, TASK_UNINTERRUPTIBLE); 60938c8a9a5SSteve French 61038c8a9a5SSteve French /* Not immediately break to none. */ 61138c8a9a5SSteve French opinfo->open_trunc = 0; 61238c8a9a5SSteve French 61338c8a9a5SSteve French if (opinfo->op_state == OPLOCK_CLOSING) 61438c8a9a5SSteve French return -ENOENT; 61532908169SNamjae Jeon else if (opinfo->level <= req_op_level) { 61632908169SNamjae Jeon if (opinfo->is_lease && 61732908169SNamjae Jeon opinfo->o_lease->state != 61832908169SNamjae Jeon (SMB2_LEASE_HANDLE_CACHING_LE | 61932908169SNamjae Jeon SMB2_LEASE_READ_CACHING_LE)) 62038c8a9a5SSteve French return 1; 62138c8a9a5SSteve French } 62232908169SNamjae Jeon } 62338c8a9a5SSteve French 62432908169SNamjae Jeon if (opinfo->level <= req_op_level) { 62532908169SNamjae Jeon if (opinfo->is_lease && 62632908169SNamjae Jeon opinfo->o_lease->state != 62732908169SNamjae Jeon (SMB2_LEASE_HANDLE_CACHING_LE | 62832908169SNamjae Jeon SMB2_LEASE_READ_CACHING_LE)) { 62938c8a9a5SSteve French wake_up_oplock_break(opinfo); 63038c8a9a5SSteve French return 1; 63138c8a9a5SSteve French } 63232908169SNamjae Jeon } 63338c8a9a5SSteve French return 0; 63438c8a9a5SSteve French } 63538c8a9a5SSteve French 63638c8a9a5SSteve French /** 63738c8a9a5SSteve French * __smb2_oplock_break_noti() - send smb2 oplock break cmd from conn 63838c8a9a5SSteve French * to client 63938c8a9a5SSteve French * @wk: smb work object 64038c8a9a5SSteve French * 64138c8a9a5SSteve French * There are two ways this function can be called. 1- while file open we break 64238c8a9a5SSteve French * from exclusive/batch lock to levelII oplock and 2- while file write/truncate 64338c8a9a5SSteve French * we break from levelII oplock no oplock. 64438c8a9a5SSteve French * work->request_buf contains oplock_info. 64538c8a9a5SSteve French */ 64638c8a9a5SSteve French static void __smb2_oplock_break_noti(struct work_struct *wk) 64738c8a9a5SSteve French { 64838c8a9a5SSteve French struct smb2_oplock_break *rsp = NULL; 64938c8a9a5SSteve French struct ksmbd_work *work = container_of(wk, struct ksmbd_work, work); 65038c8a9a5SSteve French struct oplock_break_info *br_info = work->request_buf; 65138c8a9a5SSteve French struct smb2_hdr *rsp_hdr; 65238c8a9a5SSteve French struct ksmbd_file *fp; 65338c8a9a5SSteve French 65438c8a9a5SSteve French fp = ksmbd_lookup_durable_fd(br_info->fid); 65538c8a9a5SSteve French if (!fp) 65638c8a9a5SSteve French goto out; 65738c8a9a5SSteve French 658041bba44SNamjae Jeon if (allocate_interim_rsp_buf(work)) { 65938c8a9a5SSteve French pr_err("smb2_allocate_rsp_buf failed! "); 66038c8a9a5SSteve French ksmbd_fd_put(work, fp); 66138c8a9a5SSteve French goto out; 66238c8a9a5SSteve French } 66338c8a9a5SSteve French 66438c8a9a5SSteve French rsp_hdr = smb2_get_msg(work->response_buf); 66538c8a9a5SSteve French memset(rsp_hdr, 0, sizeof(struct smb2_hdr) + 2); 66638c8a9a5SSteve French rsp_hdr->ProtocolId = SMB2_PROTO_NUMBER; 66738c8a9a5SSteve French rsp_hdr->StructureSize = SMB2_HEADER_STRUCTURE_SIZE; 66838c8a9a5SSteve French rsp_hdr->CreditRequest = cpu_to_le16(0); 66938c8a9a5SSteve French rsp_hdr->Command = SMB2_OPLOCK_BREAK; 67038c8a9a5SSteve French rsp_hdr->Flags = (SMB2_FLAGS_SERVER_TO_REDIR); 67138c8a9a5SSteve French rsp_hdr->NextCommand = 0; 67238c8a9a5SSteve French rsp_hdr->MessageId = cpu_to_le64(-1); 67338c8a9a5SSteve French rsp_hdr->Id.SyncId.ProcessId = 0; 67438c8a9a5SSteve French rsp_hdr->Id.SyncId.TreeId = 0; 67538c8a9a5SSteve French rsp_hdr->SessionId = 0; 67638c8a9a5SSteve French memset(rsp_hdr->Signature, 0, 16); 67738c8a9a5SSteve French 67838c8a9a5SSteve French rsp = smb2_get_msg(work->response_buf); 67938c8a9a5SSteve French 68038c8a9a5SSteve French rsp->StructureSize = cpu_to_le16(24); 68138c8a9a5SSteve French if (!br_info->open_trunc && 68238c8a9a5SSteve French (br_info->level == SMB2_OPLOCK_LEVEL_BATCH || 68338c8a9a5SSteve French br_info->level == SMB2_OPLOCK_LEVEL_EXCLUSIVE)) 68438c8a9a5SSteve French rsp->OplockLevel = SMB2_OPLOCK_LEVEL_II; 68538c8a9a5SSteve French else 68638c8a9a5SSteve French rsp->OplockLevel = SMB2_OPLOCK_LEVEL_NONE; 68738c8a9a5SSteve French rsp->Reserved = 0; 68838c8a9a5SSteve French rsp->Reserved2 = 0; 68938c8a9a5SSteve French rsp->PersistentFid = fp->persistent_id; 69038c8a9a5SSteve French rsp->VolatileFid = fp->volatile_id; 69138c8a9a5SSteve French 692e2b76ab8SNamjae Jeon ksmbd_fd_put(work, fp); 693e2b76ab8SNamjae Jeon if (ksmbd_iov_pin_rsp(work, (void *)rsp, 694e2b76ab8SNamjae Jeon sizeof(struct smb2_oplock_break))) 695e2b76ab8SNamjae Jeon goto out; 69638c8a9a5SSteve French 69738c8a9a5SSteve French ksmbd_debug(OPLOCK, 69838c8a9a5SSteve French "sending oplock break v_id %llu p_id = %llu lock level = %d\n", 69938c8a9a5SSteve French rsp->VolatileFid, rsp->PersistentFid, rsp->OplockLevel); 70038c8a9a5SSteve French 70138c8a9a5SSteve French ksmbd_conn_write(work); 70238c8a9a5SSteve French 70338c8a9a5SSteve French out: 70438c8a9a5SSteve French ksmbd_free_work_struct(work); 70538c8a9a5SSteve French } 70638c8a9a5SSteve French 70738c8a9a5SSteve French /** 70838c8a9a5SSteve French * smb2_oplock_break_noti() - send smb2 exclusive/batch to level2 oplock 70938c8a9a5SSteve French * break command from server to client 71038c8a9a5SSteve French * @opinfo: oplock info object 71138c8a9a5SSteve French * 71238c8a9a5SSteve French * Return: 0 on success, otherwise error 71338c8a9a5SSteve French */ 71438c8a9a5SSteve French static int smb2_oplock_break_noti(struct oplock_info *opinfo) 71538c8a9a5SSteve French { 71638c8a9a5SSteve French struct ksmbd_conn *conn = opinfo->conn; 71738c8a9a5SSteve French struct oplock_break_info *br_info; 71838c8a9a5SSteve French int ret = 0; 71938c8a9a5SSteve French struct ksmbd_work *work = ksmbd_alloc_work_struct(); 72038c8a9a5SSteve French 72138c8a9a5SSteve French if (!work) 72238c8a9a5SSteve French return -ENOMEM; 72338c8a9a5SSteve French 72438c8a9a5SSteve French br_info = kmalloc(sizeof(struct oplock_break_info), GFP_KERNEL); 72538c8a9a5SSteve French if (!br_info) { 72638c8a9a5SSteve French ksmbd_free_work_struct(work); 72738c8a9a5SSteve French return -ENOMEM; 72838c8a9a5SSteve French } 72938c8a9a5SSteve French 73038c8a9a5SSteve French br_info->level = opinfo->level; 73138c8a9a5SSteve French br_info->fid = opinfo->fid; 73238c8a9a5SSteve French br_info->open_trunc = opinfo->open_trunc; 73338c8a9a5SSteve French 73438c8a9a5SSteve French work->request_buf = (char *)br_info; 73538c8a9a5SSteve French work->conn = conn; 73638c8a9a5SSteve French work->sess = opinfo->sess; 73738c8a9a5SSteve French 73838c8a9a5SSteve French if (opinfo->op_state == OPLOCK_ACK_WAIT) { 73938c8a9a5SSteve French INIT_WORK(&work->work, __smb2_oplock_break_noti); 74038c8a9a5SSteve French ksmbd_queue_work(work); 74138c8a9a5SSteve French 74238c8a9a5SSteve French wait_for_break_ack(opinfo); 74338c8a9a5SSteve French } else { 74438c8a9a5SSteve French __smb2_oplock_break_noti(&work->work); 74538c8a9a5SSteve French if (opinfo->level == SMB2_OPLOCK_LEVEL_II) 74638c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_NONE; 74738c8a9a5SSteve French } 74838c8a9a5SSteve French return ret; 74938c8a9a5SSteve French } 75038c8a9a5SSteve French 75138c8a9a5SSteve French /** 75238c8a9a5SSteve French * __smb2_lease_break_noti() - send lease break command from server 75338c8a9a5SSteve French * to client 75438c8a9a5SSteve French * @wk: smb work object 75538c8a9a5SSteve French */ 75638c8a9a5SSteve French static void __smb2_lease_break_noti(struct work_struct *wk) 75738c8a9a5SSteve French { 75838c8a9a5SSteve French struct smb2_lease_break *rsp = NULL; 75938c8a9a5SSteve French struct ksmbd_work *work = container_of(wk, struct ksmbd_work, work); 76038c8a9a5SSteve French struct lease_break_info *br_info = work->request_buf; 76138c8a9a5SSteve French struct smb2_hdr *rsp_hdr; 76238c8a9a5SSteve French 763041bba44SNamjae Jeon if (allocate_interim_rsp_buf(work)) { 76438c8a9a5SSteve French ksmbd_debug(OPLOCK, "smb2_allocate_rsp_buf failed! "); 76538c8a9a5SSteve French goto out; 76638c8a9a5SSteve French } 76738c8a9a5SSteve French 76838c8a9a5SSteve French rsp_hdr = smb2_get_msg(work->response_buf); 76938c8a9a5SSteve French memset(rsp_hdr, 0, sizeof(struct smb2_hdr) + 2); 77038c8a9a5SSteve French rsp_hdr->ProtocolId = SMB2_PROTO_NUMBER; 77138c8a9a5SSteve French rsp_hdr->StructureSize = SMB2_HEADER_STRUCTURE_SIZE; 77238c8a9a5SSteve French rsp_hdr->CreditRequest = cpu_to_le16(0); 77338c8a9a5SSteve French rsp_hdr->Command = SMB2_OPLOCK_BREAK; 77438c8a9a5SSteve French rsp_hdr->Flags = (SMB2_FLAGS_SERVER_TO_REDIR); 77538c8a9a5SSteve French rsp_hdr->NextCommand = 0; 77638c8a9a5SSteve French rsp_hdr->MessageId = cpu_to_le64(-1); 77738c8a9a5SSteve French rsp_hdr->Id.SyncId.ProcessId = 0; 77838c8a9a5SSteve French rsp_hdr->Id.SyncId.TreeId = 0; 77938c8a9a5SSteve French rsp_hdr->SessionId = 0; 78038c8a9a5SSteve French memset(rsp_hdr->Signature, 0, 16); 78138c8a9a5SSteve French 78238c8a9a5SSteve French rsp = smb2_get_msg(work->response_buf); 78338c8a9a5SSteve French rsp->StructureSize = cpu_to_le16(44); 78438c8a9a5SSteve French rsp->Epoch = br_info->epoch; 78538c8a9a5SSteve French rsp->Flags = 0; 78638c8a9a5SSteve French 78738c8a9a5SSteve French if (br_info->curr_state & (SMB2_LEASE_WRITE_CACHING_LE | 78838c8a9a5SSteve French SMB2_LEASE_HANDLE_CACHING_LE)) 78938c8a9a5SSteve French rsp->Flags = SMB2_NOTIFY_BREAK_LEASE_FLAG_ACK_REQUIRED; 79038c8a9a5SSteve French 79138c8a9a5SSteve French memcpy(rsp->LeaseKey, br_info->lease_key, SMB2_LEASE_KEY_SIZE); 79238c8a9a5SSteve French rsp->CurrentLeaseState = br_info->curr_state; 79338c8a9a5SSteve French rsp->NewLeaseState = br_info->new_state; 79438c8a9a5SSteve French rsp->BreakReason = 0; 79538c8a9a5SSteve French rsp->AccessMaskHint = 0; 79638c8a9a5SSteve French rsp->ShareMaskHint = 0; 79738c8a9a5SSteve French 798e2b76ab8SNamjae Jeon if (ksmbd_iov_pin_rsp(work, (void *)rsp, 799e2b76ab8SNamjae Jeon sizeof(struct smb2_lease_break))) 800e2b76ab8SNamjae Jeon goto out; 80138c8a9a5SSteve French 80238c8a9a5SSteve French ksmbd_conn_write(work); 80338c8a9a5SSteve French 80438c8a9a5SSteve French out: 80538c8a9a5SSteve French ksmbd_free_work_struct(work); 80638c8a9a5SSteve French } 80738c8a9a5SSteve French 80838c8a9a5SSteve French /** 80938c8a9a5SSteve French * smb2_lease_break_noti() - break lease when a new client request 81038c8a9a5SSteve French * write lease 81138c8a9a5SSteve French * @opinfo: conains lease state information 81238c8a9a5SSteve French * 81338c8a9a5SSteve French * Return: 0 on success, otherwise error 81438c8a9a5SSteve French */ 81538c8a9a5SSteve French static int smb2_lease_break_noti(struct oplock_info *opinfo) 81638c8a9a5SSteve French { 81738c8a9a5SSteve French struct ksmbd_conn *conn = opinfo->conn; 81838c8a9a5SSteve French struct list_head *tmp, *t; 81938c8a9a5SSteve French struct ksmbd_work *work; 82038c8a9a5SSteve French struct lease_break_info *br_info; 82138c8a9a5SSteve French struct lease *lease = opinfo->o_lease; 82238c8a9a5SSteve French 82338c8a9a5SSteve French work = ksmbd_alloc_work_struct(); 82438c8a9a5SSteve French if (!work) 82538c8a9a5SSteve French return -ENOMEM; 82638c8a9a5SSteve French 82738c8a9a5SSteve French br_info = kmalloc(sizeof(struct lease_break_info), GFP_KERNEL); 82838c8a9a5SSteve French if (!br_info) { 82938c8a9a5SSteve French ksmbd_free_work_struct(work); 83038c8a9a5SSteve French return -ENOMEM; 83138c8a9a5SSteve French } 83238c8a9a5SSteve French 83338c8a9a5SSteve French br_info->curr_state = lease->state; 83438c8a9a5SSteve French br_info->new_state = lease->new_state; 83538c8a9a5SSteve French if (lease->version == 2) 83638c8a9a5SSteve French br_info->epoch = cpu_to_le16(++lease->epoch); 83738c8a9a5SSteve French else 83838c8a9a5SSteve French br_info->epoch = 0; 83938c8a9a5SSteve French memcpy(br_info->lease_key, lease->lease_key, SMB2_LEASE_KEY_SIZE); 84038c8a9a5SSteve French 84138c8a9a5SSteve French work->request_buf = (char *)br_info; 84238c8a9a5SSteve French work->conn = conn; 84338c8a9a5SSteve French work->sess = opinfo->sess; 84438c8a9a5SSteve French 84538c8a9a5SSteve French if (opinfo->op_state == OPLOCK_ACK_WAIT) { 84638c8a9a5SSteve French list_for_each_safe(tmp, t, &opinfo->interim_list) { 84738c8a9a5SSteve French struct ksmbd_work *in_work; 84838c8a9a5SSteve French 84938c8a9a5SSteve French in_work = list_entry(tmp, struct ksmbd_work, 85038c8a9a5SSteve French interim_entry); 85138c8a9a5SSteve French setup_async_work(in_work, NULL, NULL); 85238c8a9a5SSteve French smb2_send_interim_resp(in_work, STATUS_PENDING); 853fa86141fSNamjae Jeon list_del_init(&in_work->interim_entry); 854fa86141fSNamjae Jeon release_async_work(in_work); 85538c8a9a5SSteve French } 85638c8a9a5SSteve French INIT_WORK(&work->work, __smb2_lease_break_noti); 85738c8a9a5SSteve French ksmbd_queue_work(work); 85838c8a9a5SSteve French wait_for_break_ack(opinfo); 85938c8a9a5SSteve French } else { 86038c8a9a5SSteve French __smb2_lease_break_noti(&work->work); 86138c8a9a5SSteve French if (opinfo->o_lease->new_state == SMB2_LEASE_NONE_LE) { 86238c8a9a5SSteve French opinfo->level = SMB2_OPLOCK_LEVEL_NONE; 86338c8a9a5SSteve French opinfo->o_lease->state = SMB2_LEASE_NONE_LE; 86438c8a9a5SSteve French } 86538c8a9a5SSteve French } 86638c8a9a5SSteve French return 0; 86738c8a9a5SSteve French } 86838c8a9a5SSteve French 86938c8a9a5SSteve French static void wait_lease_breaking(struct oplock_info *opinfo) 87038c8a9a5SSteve French { 87138c8a9a5SSteve French if (!opinfo->is_lease) 87238c8a9a5SSteve French return; 87338c8a9a5SSteve French 87438c8a9a5SSteve French wake_up_interruptible_all(&opinfo->oplock_brk); 87538c8a9a5SSteve French if (atomic_read(&opinfo->breaking_cnt)) { 87638c8a9a5SSteve French int ret = 0; 87738c8a9a5SSteve French 87838c8a9a5SSteve French ret = wait_event_interruptible_timeout(opinfo->oplock_brk, 87938c8a9a5SSteve French atomic_read(&opinfo->breaking_cnt) == 0, 88038c8a9a5SSteve French HZ); 88138c8a9a5SSteve French if (!ret) 88238c8a9a5SSteve French atomic_set(&opinfo->breaking_cnt, 0); 88338c8a9a5SSteve French } 88438c8a9a5SSteve French } 88538c8a9a5SSteve French 88638c8a9a5SSteve French static int oplock_break(struct oplock_info *brk_opinfo, int req_op_level) 88738c8a9a5SSteve French { 88838c8a9a5SSteve French int err = 0; 88938c8a9a5SSteve French 89038c8a9a5SSteve French /* Need to break exclusive/batch oplock, write lease or overwrite_if */ 89138c8a9a5SSteve French ksmbd_debug(OPLOCK, 89238c8a9a5SSteve French "request to send oplock(level : 0x%x) break notification\n", 89338c8a9a5SSteve French brk_opinfo->level); 89438c8a9a5SSteve French 89538c8a9a5SSteve French if (brk_opinfo->is_lease) { 89638c8a9a5SSteve French struct lease *lease = brk_opinfo->o_lease; 89738c8a9a5SSteve French 89838c8a9a5SSteve French atomic_inc(&brk_opinfo->breaking_cnt); 89938c8a9a5SSteve French err = oplock_break_pending(brk_opinfo, req_op_level); 90038c8a9a5SSteve French if (err) 90138c8a9a5SSteve French return err < 0 ? err : 0; 90238c8a9a5SSteve French 90338c8a9a5SSteve French if (brk_opinfo->open_trunc) { 90438c8a9a5SSteve French /* 90538c8a9a5SSteve French * Create overwrite break trigger the lease break to 90638c8a9a5SSteve French * none. 90738c8a9a5SSteve French */ 90838c8a9a5SSteve French lease->new_state = SMB2_LEASE_NONE_LE; 90938c8a9a5SSteve French } else { 91038c8a9a5SSteve French if (lease->state & SMB2_LEASE_WRITE_CACHING_LE) { 91138c8a9a5SSteve French if (lease->state & SMB2_LEASE_HANDLE_CACHING_LE) 91238c8a9a5SSteve French lease->new_state = 91338c8a9a5SSteve French SMB2_LEASE_READ_CACHING_LE | 91438c8a9a5SSteve French SMB2_LEASE_HANDLE_CACHING_LE; 91538c8a9a5SSteve French else 91638c8a9a5SSteve French lease->new_state = 91738c8a9a5SSteve French SMB2_LEASE_READ_CACHING_LE; 91838c8a9a5SSteve French } else { 9193c1e602aSNamjae Jeon if (lease->state & SMB2_LEASE_HANDLE_CACHING_LE && 9203c1e602aSNamjae Jeon !lease->is_dir) 92138c8a9a5SSteve French lease->new_state = 92238c8a9a5SSteve French SMB2_LEASE_READ_CACHING_LE; 92338c8a9a5SSteve French else 92438c8a9a5SSteve French lease->new_state = SMB2_LEASE_NONE_LE; 92538c8a9a5SSteve French } 92638c8a9a5SSteve French } 92738c8a9a5SSteve French 92838c8a9a5SSteve French if (lease->state & (SMB2_LEASE_WRITE_CACHING_LE | 92938c8a9a5SSteve French SMB2_LEASE_HANDLE_CACHING_LE)) 93038c8a9a5SSteve French brk_opinfo->op_state = OPLOCK_ACK_WAIT; 93138c8a9a5SSteve French else 93238c8a9a5SSteve French atomic_dec(&brk_opinfo->breaking_cnt); 93338c8a9a5SSteve French } else { 93438c8a9a5SSteve French err = oplock_break_pending(brk_opinfo, req_op_level); 93538c8a9a5SSteve French if (err) 93638c8a9a5SSteve French return err < 0 ? err : 0; 93738c8a9a5SSteve French 93838c8a9a5SSteve French if (brk_opinfo->level == SMB2_OPLOCK_LEVEL_BATCH || 93938c8a9a5SSteve French brk_opinfo->level == SMB2_OPLOCK_LEVEL_EXCLUSIVE) 94038c8a9a5SSteve French brk_opinfo->op_state = OPLOCK_ACK_WAIT; 94138c8a9a5SSteve French } 94238c8a9a5SSteve French 94338c8a9a5SSteve French if (brk_opinfo->is_lease) 94438c8a9a5SSteve French err = smb2_lease_break_noti(brk_opinfo); 94538c8a9a5SSteve French else 94638c8a9a5SSteve French err = smb2_oplock_break_noti(brk_opinfo); 94738c8a9a5SSteve French 94838c8a9a5SSteve French ksmbd_debug(OPLOCK, "oplock granted = %d\n", brk_opinfo->level); 94938c8a9a5SSteve French if (brk_opinfo->op_state == OPLOCK_CLOSING) 95038c8a9a5SSteve French err = -ENOENT; 95138c8a9a5SSteve French wake_up_oplock_break(brk_opinfo); 95238c8a9a5SSteve French 95338c8a9a5SSteve French wait_lease_breaking(brk_opinfo); 95438c8a9a5SSteve French 95538c8a9a5SSteve French return err; 95638c8a9a5SSteve French } 95738c8a9a5SSteve French 95838c8a9a5SSteve French void destroy_lease_table(struct ksmbd_conn *conn) 95938c8a9a5SSteve French { 96038c8a9a5SSteve French struct lease_table *lb, *lbtmp; 96138c8a9a5SSteve French struct oplock_info *opinfo; 96238c8a9a5SSteve French 96338c8a9a5SSteve French write_lock(&lease_list_lock); 96438c8a9a5SSteve French if (list_empty(&lease_table_list)) { 96538c8a9a5SSteve French write_unlock(&lease_list_lock); 96638c8a9a5SSteve French return; 96738c8a9a5SSteve French } 96838c8a9a5SSteve French 96938c8a9a5SSteve French list_for_each_entry_safe(lb, lbtmp, &lease_table_list, l_entry) { 97038c8a9a5SSteve French if (conn && memcmp(lb->client_guid, conn->ClientGUID, 97138c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE)) 97238c8a9a5SSteve French continue; 97338c8a9a5SSteve French again: 97438c8a9a5SSteve French rcu_read_lock(); 97538c8a9a5SSteve French list_for_each_entry_rcu(opinfo, &lb->lease_list, 97638c8a9a5SSteve French lease_entry) { 97738c8a9a5SSteve French rcu_read_unlock(); 97838c8a9a5SSteve French lease_del_list(opinfo); 97938c8a9a5SSteve French goto again; 98038c8a9a5SSteve French } 98138c8a9a5SSteve French rcu_read_unlock(); 98238c8a9a5SSteve French list_del(&lb->l_entry); 98338c8a9a5SSteve French kfree(lb); 98438c8a9a5SSteve French } 98538c8a9a5SSteve French write_unlock(&lease_list_lock); 98638c8a9a5SSteve French } 98738c8a9a5SSteve French 98838c8a9a5SSteve French int find_same_lease_key(struct ksmbd_session *sess, struct ksmbd_inode *ci, 98938c8a9a5SSteve French struct lease_ctx_info *lctx) 99038c8a9a5SSteve French { 99138c8a9a5SSteve French struct oplock_info *opinfo; 99238c8a9a5SSteve French int err = 0; 99338c8a9a5SSteve French struct lease_table *lb; 99438c8a9a5SSteve French 99538c8a9a5SSteve French if (!lctx) 99638c8a9a5SSteve French return err; 99738c8a9a5SSteve French 99838c8a9a5SSteve French read_lock(&lease_list_lock); 99938c8a9a5SSteve French if (list_empty(&lease_table_list)) { 100038c8a9a5SSteve French read_unlock(&lease_list_lock); 100138c8a9a5SSteve French return 0; 100238c8a9a5SSteve French } 100338c8a9a5SSteve French 100438c8a9a5SSteve French list_for_each_entry(lb, &lease_table_list, l_entry) { 100538c8a9a5SSteve French if (!memcmp(lb->client_guid, sess->ClientGUID, 100638c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE)) 100738c8a9a5SSteve French goto found; 100838c8a9a5SSteve French } 100938c8a9a5SSteve French read_unlock(&lease_list_lock); 101038c8a9a5SSteve French 101138c8a9a5SSteve French return 0; 101238c8a9a5SSteve French 101338c8a9a5SSteve French found: 101438c8a9a5SSteve French rcu_read_lock(); 101538c8a9a5SSteve French list_for_each_entry_rcu(opinfo, &lb->lease_list, lease_entry) { 101638c8a9a5SSteve French if (!atomic_inc_not_zero(&opinfo->refcount)) 101738c8a9a5SSteve French continue; 101838c8a9a5SSteve French rcu_read_unlock(); 101938c8a9a5SSteve French if (opinfo->o_fp->f_ci == ci) 102038c8a9a5SSteve French goto op_next; 102138c8a9a5SSteve French err = compare_guid_key(opinfo, sess->ClientGUID, 102238c8a9a5SSteve French lctx->lease_key); 102338c8a9a5SSteve French if (err) { 102438c8a9a5SSteve French err = -EINVAL; 102538c8a9a5SSteve French ksmbd_debug(OPLOCK, 102638c8a9a5SSteve French "found same lease key is already used in other files\n"); 102738c8a9a5SSteve French opinfo_put(opinfo); 102838c8a9a5SSteve French goto out; 102938c8a9a5SSteve French } 103038c8a9a5SSteve French op_next: 103138c8a9a5SSteve French opinfo_put(opinfo); 103238c8a9a5SSteve French rcu_read_lock(); 103338c8a9a5SSteve French } 103438c8a9a5SSteve French rcu_read_unlock(); 103538c8a9a5SSteve French 103638c8a9a5SSteve French out: 103738c8a9a5SSteve French read_unlock(&lease_list_lock); 103838c8a9a5SSteve French return err; 103938c8a9a5SSteve French } 104038c8a9a5SSteve French 104138c8a9a5SSteve French static void copy_lease(struct oplock_info *op1, struct oplock_info *op2) 104238c8a9a5SSteve French { 104338c8a9a5SSteve French struct lease *lease1 = op1->o_lease; 104438c8a9a5SSteve French struct lease *lease2 = op2->o_lease; 104538c8a9a5SSteve French 104638c8a9a5SSteve French op2->level = op1->level; 104738c8a9a5SSteve French lease2->state = lease1->state; 104838c8a9a5SSteve French memcpy(lease2->lease_key, lease1->lease_key, 104938c8a9a5SSteve French SMB2_LEASE_KEY_SIZE); 105038c8a9a5SSteve French lease2->duration = lease1->duration; 105138c8a9a5SSteve French lease2->flags = lease1->flags; 10529554d493SNamjae Jeon lease2->epoch = lease1->epoch; 1053fc0db77bSNamjae Jeon lease2->version = lease1->version; 105438c8a9a5SSteve French } 105538c8a9a5SSteve French 105638c8a9a5SSteve French static int add_lease_global_list(struct oplock_info *opinfo) 105738c8a9a5SSteve French { 105838c8a9a5SSteve French struct lease_table *lb; 105938c8a9a5SSteve French 106038c8a9a5SSteve French read_lock(&lease_list_lock); 106138c8a9a5SSteve French list_for_each_entry(lb, &lease_table_list, l_entry) { 106238c8a9a5SSteve French if (!memcmp(lb->client_guid, opinfo->conn->ClientGUID, 106338c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE)) { 106438c8a9a5SSteve French opinfo->o_lease->l_lb = lb; 106538c8a9a5SSteve French lease_add_list(opinfo); 106638c8a9a5SSteve French read_unlock(&lease_list_lock); 106738c8a9a5SSteve French return 0; 106838c8a9a5SSteve French } 106938c8a9a5SSteve French } 107038c8a9a5SSteve French read_unlock(&lease_list_lock); 107138c8a9a5SSteve French 107238c8a9a5SSteve French lb = kmalloc(sizeof(struct lease_table), GFP_KERNEL); 107338c8a9a5SSteve French if (!lb) 107438c8a9a5SSteve French return -ENOMEM; 107538c8a9a5SSteve French 107638c8a9a5SSteve French memcpy(lb->client_guid, opinfo->conn->ClientGUID, 107738c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE); 107838c8a9a5SSteve French INIT_LIST_HEAD(&lb->lease_list); 107938c8a9a5SSteve French spin_lock_init(&lb->lb_lock); 108038c8a9a5SSteve French opinfo->o_lease->l_lb = lb; 108138c8a9a5SSteve French lease_add_list(opinfo); 108238c8a9a5SSteve French lb_add(lb); 108338c8a9a5SSteve French return 0; 108438c8a9a5SSteve French } 108538c8a9a5SSteve French 108638c8a9a5SSteve French static void set_oplock_level(struct oplock_info *opinfo, int level, 108738c8a9a5SSteve French struct lease_ctx_info *lctx) 108838c8a9a5SSteve French { 108938c8a9a5SSteve French switch (level) { 109038c8a9a5SSteve French case SMB2_OPLOCK_LEVEL_BATCH: 109138c8a9a5SSteve French case SMB2_OPLOCK_LEVEL_EXCLUSIVE: 109238c8a9a5SSteve French grant_write_oplock(opinfo, level, lctx); 109338c8a9a5SSteve French break; 109438c8a9a5SSteve French case SMB2_OPLOCK_LEVEL_II: 109538c8a9a5SSteve French grant_read_oplock(opinfo, lctx); 109638c8a9a5SSteve French break; 109738c8a9a5SSteve French default: 109838c8a9a5SSteve French grant_none_oplock(opinfo, lctx); 109938c8a9a5SSteve French break; 110038c8a9a5SSteve French } 110138c8a9a5SSteve French } 110238c8a9a5SSteve French 11033c1e602aSNamjae Jeon void smb_send_parent_lease_break_noti(struct ksmbd_file *fp, 11043c1e602aSNamjae Jeon struct lease_ctx_info *lctx) 11053c1e602aSNamjae Jeon { 11063c1e602aSNamjae Jeon struct oplock_info *opinfo; 11073c1e602aSNamjae Jeon struct ksmbd_inode *p_ci = NULL; 11083c1e602aSNamjae Jeon 11093c1e602aSNamjae Jeon if (lctx->version != 2) 11103c1e602aSNamjae Jeon return; 11113c1e602aSNamjae Jeon 11123c1e602aSNamjae Jeon p_ci = ksmbd_inode_lookup_lock(fp->filp->f_path.dentry->d_parent); 11133c1e602aSNamjae Jeon if (!p_ci) 11143c1e602aSNamjae Jeon return; 11153c1e602aSNamjae Jeon 11163c1e602aSNamjae Jeon read_lock(&p_ci->m_lock); 11173c1e602aSNamjae Jeon list_for_each_entry(opinfo, &p_ci->m_op_list, op_entry) { 11183c1e602aSNamjae Jeon if (!opinfo->is_lease) 11193c1e602aSNamjae Jeon continue; 11203c1e602aSNamjae Jeon 11213c1e602aSNamjae Jeon if (opinfo->o_lease->state != SMB2_OPLOCK_LEVEL_NONE && 11223c1e602aSNamjae Jeon (!(lctx->flags & SMB2_LEASE_FLAG_PARENT_LEASE_KEY_SET_LE) || 11233c1e602aSNamjae Jeon !compare_guid_key(opinfo, fp->conn->ClientGUID, 11243c1e602aSNamjae Jeon lctx->parent_lease_key))) { 11253c1e602aSNamjae Jeon if (!atomic_inc_not_zero(&opinfo->refcount)) 11263c1e602aSNamjae Jeon continue; 11273c1e602aSNamjae Jeon 11283c1e602aSNamjae Jeon atomic_inc(&opinfo->conn->r_count); 11293c1e602aSNamjae Jeon if (ksmbd_conn_releasing(opinfo->conn)) { 11303c1e602aSNamjae Jeon atomic_dec(&opinfo->conn->r_count); 11313c1e602aSNamjae Jeon continue; 11323c1e602aSNamjae Jeon } 11333c1e602aSNamjae Jeon 11343c1e602aSNamjae Jeon read_unlock(&p_ci->m_lock); 11353c1e602aSNamjae Jeon oplock_break(opinfo, SMB2_OPLOCK_LEVEL_NONE); 11363c1e602aSNamjae Jeon opinfo_conn_put(opinfo); 11373c1e602aSNamjae Jeon read_lock(&p_ci->m_lock); 11383c1e602aSNamjae Jeon } 11393c1e602aSNamjae Jeon } 11403c1e602aSNamjae Jeon read_unlock(&p_ci->m_lock); 11413c1e602aSNamjae Jeon 11423c1e602aSNamjae Jeon ksmbd_inode_put(p_ci); 11433c1e602aSNamjae Jeon } 11443c1e602aSNamjae Jeon 1145ab5a0a1cSNamjae Jeon void smb_lazy_parent_lease_break_close(struct ksmbd_file *fp) 1146ab5a0a1cSNamjae Jeon { 1147ab5a0a1cSNamjae Jeon struct oplock_info *opinfo; 1148ab5a0a1cSNamjae Jeon struct ksmbd_inode *p_ci = NULL; 1149ab5a0a1cSNamjae Jeon 1150ab5a0a1cSNamjae Jeon rcu_read_lock(); 1151ab5a0a1cSNamjae Jeon opinfo = rcu_dereference(fp->f_opinfo); 1152ab5a0a1cSNamjae Jeon rcu_read_unlock(); 1153ab5a0a1cSNamjae Jeon 1154ab5a0a1cSNamjae Jeon if (!opinfo->is_lease || opinfo->o_lease->version != 2) 1155ab5a0a1cSNamjae Jeon return; 1156ab5a0a1cSNamjae Jeon 1157ab5a0a1cSNamjae Jeon p_ci = ksmbd_inode_lookup_lock(fp->filp->f_path.dentry->d_parent); 1158ab5a0a1cSNamjae Jeon if (!p_ci) 1159ab5a0a1cSNamjae Jeon return; 1160ab5a0a1cSNamjae Jeon 1161ab5a0a1cSNamjae Jeon read_lock(&p_ci->m_lock); 1162ab5a0a1cSNamjae Jeon list_for_each_entry(opinfo, &p_ci->m_op_list, op_entry) { 1163ab5a0a1cSNamjae Jeon if (!opinfo->is_lease) 1164ab5a0a1cSNamjae Jeon continue; 1165ab5a0a1cSNamjae Jeon 1166ab5a0a1cSNamjae Jeon if (opinfo->o_lease->state != SMB2_OPLOCK_LEVEL_NONE) { 1167ab5a0a1cSNamjae Jeon if (!atomic_inc_not_zero(&opinfo->refcount)) 1168ab5a0a1cSNamjae Jeon continue; 1169ab5a0a1cSNamjae Jeon 1170ab5a0a1cSNamjae Jeon atomic_inc(&opinfo->conn->r_count); 1171ab5a0a1cSNamjae Jeon if (ksmbd_conn_releasing(opinfo->conn)) { 1172ab5a0a1cSNamjae Jeon atomic_dec(&opinfo->conn->r_count); 1173ab5a0a1cSNamjae Jeon continue; 1174ab5a0a1cSNamjae Jeon } 1175ab5a0a1cSNamjae Jeon read_unlock(&p_ci->m_lock); 1176ab5a0a1cSNamjae Jeon oplock_break(opinfo, SMB2_OPLOCK_LEVEL_NONE); 1177ab5a0a1cSNamjae Jeon opinfo_conn_put(opinfo); 1178ab5a0a1cSNamjae Jeon read_lock(&p_ci->m_lock); 1179ab5a0a1cSNamjae Jeon } 1180ab5a0a1cSNamjae Jeon } 1181ab5a0a1cSNamjae Jeon read_unlock(&p_ci->m_lock); 1182ab5a0a1cSNamjae Jeon 1183ab5a0a1cSNamjae Jeon ksmbd_inode_put(p_ci); 1184ab5a0a1cSNamjae Jeon } 1185ab5a0a1cSNamjae Jeon 118638c8a9a5SSteve French /** 118738c8a9a5SSteve French * smb_grant_oplock() - handle oplock/lease request on file open 118838c8a9a5SSteve French * @work: smb work 118938c8a9a5SSteve French * @req_op_level: oplock level 119038c8a9a5SSteve French * @pid: id of open file 119138c8a9a5SSteve French * @fp: ksmbd file pointer 119238c8a9a5SSteve French * @tid: Tree id of connection 119338c8a9a5SSteve French * @lctx: lease context information on file open 119438c8a9a5SSteve French * @share_ret: share mode 119538c8a9a5SSteve French * 119638c8a9a5SSteve French * Return: 0 on success, otherwise error 119738c8a9a5SSteve French */ 119838c8a9a5SSteve French int smb_grant_oplock(struct ksmbd_work *work, int req_op_level, u64 pid, 119938c8a9a5SSteve French struct ksmbd_file *fp, __u16 tid, 120038c8a9a5SSteve French struct lease_ctx_info *lctx, int share_ret) 120138c8a9a5SSteve French { 120238c8a9a5SSteve French struct ksmbd_session *sess = work->sess; 120338c8a9a5SSteve French int err = 0; 120438c8a9a5SSteve French struct oplock_info *opinfo = NULL, *prev_opinfo = NULL; 120538c8a9a5SSteve French struct ksmbd_inode *ci = fp->f_ci; 120638c8a9a5SSteve French bool prev_op_has_lease; 120738c8a9a5SSteve French __le32 prev_op_state = 0; 120838c8a9a5SSteve French 120908e6c823SNamjae Jeon /* Only v2 leases handle the directory */ 121008e6c823SNamjae Jeon if (S_ISDIR(file_inode(fp->filp)->i_mode)) { 121108e6c823SNamjae Jeon if (!lctx || lctx->version != 2) 121208e6c823SNamjae Jeon return 0; 121308e6c823SNamjae Jeon } 121408e6c823SNamjae Jeon 121538c8a9a5SSteve French opinfo = alloc_opinfo(work, pid, tid); 121638c8a9a5SSteve French if (!opinfo) 121738c8a9a5SSteve French return -ENOMEM; 121838c8a9a5SSteve French 121938c8a9a5SSteve French if (lctx) { 122038c8a9a5SSteve French err = alloc_lease(opinfo, lctx); 122138c8a9a5SSteve French if (err) 122238c8a9a5SSteve French goto err_out; 122338c8a9a5SSteve French opinfo->is_lease = 1; 122438c8a9a5SSteve French } 122538c8a9a5SSteve French 122638c8a9a5SSteve French /* ci does not have any oplock */ 122738c8a9a5SSteve French if (!opinfo_count(fp)) 122838c8a9a5SSteve French goto set_lev; 122938c8a9a5SSteve French 123038c8a9a5SSteve French /* grant none-oplock if second open is trunc */ 123138c8a9a5SSteve French if (fp->attrib_only && fp->cdoption != FILE_OVERWRITE_IF_LE && 123238c8a9a5SSteve French fp->cdoption != FILE_OVERWRITE_LE && 123338c8a9a5SSteve French fp->cdoption != FILE_SUPERSEDE_LE) { 123438c8a9a5SSteve French req_op_level = SMB2_OPLOCK_LEVEL_NONE; 123538c8a9a5SSteve French goto set_lev; 123638c8a9a5SSteve French } 123738c8a9a5SSteve French 123838c8a9a5SSteve French if (lctx) { 123938c8a9a5SSteve French struct oplock_info *m_opinfo; 124038c8a9a5SSteve French 124138c8a9a5SSteve French /* is lease already granted ? */ 124238c8a9a5SSteve French m_opinfo = same_client_has_lease(ci, sess->ClientGUID, 124338c8a9a5SSteve French lctx); 124438c8a9a5SSteve French if (m_opinfo) { 124538c8a9a5SSteve French copy_lease(m_opinfo, opinfo); 124638c8a9a5SSteve French if (atomic_read(&m_opinfo->breaking_cnt)) 124738c8a9a5SSteve French opinfo->o_lease->flags = 124838c8a9a5SSteve French SMB2_LEASE_FLAG_BREAK_IN_PROGRESS_LE; 124938c8a9a5SSteve French goto out; 125038c8a9a5SSteve French } 125138c8a9a5SSteve French } 125238c8a9a5SSteve French prev_opinfo = opinfo_get_list(ci); 125338c8a9a5SSteve French if (!prev_opinfo || 125436322523SNamjae Jeon (prev_opinfo->level == SMB2_OPLOCK_LEVEL_NONE && lctx)) { 125536322523SNamjae Jeon opinfo_conn_put(prev_opinfo); 125638c8a9a5SSteve French goto set_lev; 125736322523SNamjae Jeon } 125838c8a9a5SSteve French prev_op_has_lease = prev_opinfo->is_lease; 125938c8a9a5SSteve French if (prev_op_has_lease) 126038c8a9a5SSteve French prev_op_state = prev_opinfo->o_lease->state; 126138c8a9a5SSteve French 126238c8a9a5SSteve French if (share_ret < 0 && 126338c8a9a5SSteve French prev_opinfo->level == SMB2_OPLOCK_LEVEL_EXCLUSIVE) { 126438c8a9a5SSteve French err = share_ret; 126536322523SNamjae Jeon opinfo_conn_put(prev_opinfo); 126638c8a9a5SSteve French goto err_out; 126738c8a9a5SSteve French } 126838c8a9a5SSteve French 126938c8a9a5SSteve French if (prev_opinfo->level != SMB2_OPLOCK_LEVEL_BATCH && 127038c8a9a5SSteve French prev_opinfo->level != SMB2_OPLOCK_LEVEL_EXCLUSIVE) { 127136322523SNamjae Jeon opinfo_conn_put(prev_opinfo); 127238c8a9a5SSteve French goto op_break_not_needed; 127338c8a9a5SSteve French } 127438c8a9a5SSteve French 127538c8a9a5SSteve French list_add(&work->interim_entry, &prev_opinfo->interim_list); 127638c8a9a5SSteve French err = oplock_break(prev_opinfo, SMB2_OPLOCK_LEVEL_II); 127736322523SNamjae Jeon opinfo_conn_put(prev_opinfo); 127838c8a9a5SSteve French if (err == -ENOENT) 127938c8a9a5SSteve French goto set_lev; 128038c8a9a5SSteve French /* Check all oplock was freed by close */ 128138c8a9a5SSteve French else if (err < 0) 128238c8a9a5SSteve French goto err_out; 128338c8a9a5SSteve French 128438c8a9a5SSteve French op_break_not_needed: 128538c8a9a5SSteve French if (share_ret < 0) { 128638c8a9a5SSteve French err = share_ret; 128738c8a9a5SSteve French goto err_out; 128838c8a9a5SSteve French } 128938c8a9a5SSteve French 129038c8a9a5SSteve French if (req_op_level != SMB2_OPLOCK_LEVEL_NONE) 129138c8a9a5SSteve French req_op_level = SMB2_OPLOCK_LEVEL_II; 129238c8a9a5SSteve French 129338c8a9a5SSteve French /* grant fixed oplock on stacked locking between lease and oplock */ 129438c8a9a5SSteve French if (prev_op_has_lease && !lctx) 129538c8a9a5SSteve French if (prev_op_state & SMB2_LEASE_HANDLE_CACHING_LE) 129638c8a9a5SSteve French req_op_level = SMB2_OPLOCK_LEVEL_NONE; 129738c8a9a5SSteve French 129838c8a9a5SSteve French if (!prev_op_has_lease && lctx) { 129938c8a9a5SSteve French req_op_level = SMB2_OPLOCK_LEVEL_II; 130038c8a9a5SSteve French lctx->req_state = SMB2_LEASE_READ_CACHING_LE; 130138c8a9a5SSteve French } 130238c8a9a5SSteve French 130338c8a9a5SSteve French set_lev: 130438c8a9a5SSteve French set_oplock_level(opinfo, req_op_level, lctx); 130538c8a9a5SSteve French 130638c8a9a5SSteve French out: 130738c8a9a5SSteve French rcu_assign_pointer(fp->f_opinfo, opinfo); 130838c8a9a5SSteve French opinfo->o_fp = fp; 130938c8a9a5SSteve French 131038c8a9a5SSteve French opinfo_count_inc(fp); 131138c8a9a5SSteve French opinfo_add(opinfo); 131238c8a9a5SSteve French if (opinfo->is_lease) { 131338c8a9a5SSteve French err = add_lease_global_list(opinfo); 131438c8a9a5SSteve French if (err) 131538c8a9a5SSteve French goto err_out; 131638c8a9a5SSteve French } 131738c8a9a5SSteve French 131838c8a9a5SSteve French return 0; 131938c8a9a5SSteve French err_out: 132038c8a9a5SSteve French free_opinfo(opinfo); 132138c8a9a5SSteve French return err; 132238c8a9a5SSteve French } 132338c8a9a5SSteve French 132438c8a9a5SSteve French /** 132538c8a9a5SSteve French * smb_break_all_write_oplock() - break batch/exclusive oplock to level2 132638c8a9a5SSteve French * @work: smb work 132738c8a9a5SSteve French * @fp: ksmbd file pointer 132838c8a9a5SSteve French * @is_trunc: truncate on open 132938c8a9a5SSteve French */ 133038c8a9a5SSteve French static void smb_break_all_write_oplock(struct ksmbd_work *work, 133138c8a9a5SSteve French struct ksmbd_file *fp, int is_trunc) 133238c8a9a5SSteve French { 133338c8a9a5SSteve French struct oplock_info *brk_opinfo; 133438c8a9a5SSteve French 133538c8a9a5SSteve French brk_opinfo = opinfo_get_list(fp->f_ci); 133638c8a9a5SSteve French if (!brk_opinfo) 133738c8a9a5SSteve French return; 133838c8a9a5SSteve French if (brk_opinfo->level != SMB2_OPLOCK_LEVEL_BATCH && 133938c8a9a5SSteve French brk_opinfo->level != SMB2_OPLOCK_LEVEL_EXCLUSIVE) { 134036322523SNamjae Jeon opinfo_conn_put(brk_opinfo); 134138c8a9a5SSteve French return; 134238c8a9a5SSteve French } 134338c8a9a5SSteve French 134438c8a9a5SSteve French brk_opinfo->open_trunc = is_trunc; 134538c8a9a5SSteve French list_add(&work->interim_entry, &brk_opinfo->interim_list); 134638c8a9a5SSteve French oplock_break(brk_opinfo, SMB2_OPLOCK_LEVEL_II); 134736322523SNamjae Jeon opinfo_conn_put(brk_opinfo); 134838c8a9a5SSteve French } 134938c8a9a5SSteve French 135038c8a9a5SSteve French /** 135138c8a9a5SSteve French * smb_break_all_levII_oplock() - send level2 oplock or read lease break command 135238c8a9a5SSteve French * from server to client 135338c8a9a5SSteve French * @work: smb work 135438c8a9a5SSteve French * @fp: ksmbd file pointer 135538c8a9a5SSteve French * @is_trunc: truncate on open 135638c8a9a5SSteve French */ 135738c8a9a5SSteve French void smb_break_all_levII_oplock(struct ksmbd_work *work, struct ksmbd_file *fp, 135838c8a9a5SSteve French int is_trunc) 135938c8a9a5SSteve French { 136038c8a9a5SSteve French struct oplock_info *op, *brk_op; 136138c8a9a5SSteve French struct ksmbd_inode *ci; 136238c8a9a5SSteve French struct ksmbd_conn *conn = work->conn; 136338c8a9a5SSteve French 136438c8a9a5SSteve French if (!test_share_config_flag(work->tcon->share_conf, 136538c8a9a5SSteve French KSMBD_SHARE_FLAG_OPLOCKS)) 136638c8a9a5SSteve French return; 136738c8a9a5SSteve French 136838c8a9a5SSteve French ci = fp->f_ci; 136938c8a9a5SSteve French op = opinfo_get(fp); 137038c8a9a5SSteve French 137138c8a9a5SSteve French rcu_read_lock(); 137238c8a9a5SSteve French list_for_each_entry_rcu(brk_op, &ci->m_op_list, op_entry) { 137338c8a9a5SSteve French if (!atomic_inc_not_zero(&brk_op->refcount)) 137438c8a9a5SSteve French continue; 137536322523SNamjae Jeon 137636322523SNamjae Jeon atomic_inc(&brk_op->conn->r_count); 137736322523SNamjae Jeon if (ksmbd_conn_releasing(brk_op->conn)) { 137836322523SNamjae Jeon atomic_dec(&brk_op->conn->r_count); 137936322523SNamjae Jeon continue; 138036322523SNamjae Jeon } 138136322523SNamjae Jeon 138238c8a9a5SSteve French rcu_read_unlock(); 138338c8a9a5SSteve French if (brk_op->is_lease && (brk_op->o_lease->state & 138438c8a9a5SSteve French (~(SMB2_LEASE_READ_CACHING_LE | 138538c8a9a5SSteve French SMB2_LEASE_HANDLE_CACHING_LE)))) { 138638c8a9a5SSteve French ksmbd_debug(OPLOCK, "unexpected lease state(0x%x)\n", 138738c8a9a5SSteve French brk_op->o_lease->state); 138838c8a9a5SSteve French goto next; 138938c8a9a5SSteve French } else if (brk_op->level != 139038c8a9a5SSteve French SMB2_OPLOCK_LEVEL_II) { 139138c8a9a5SSteve French ksmbd_debug(OPLOCK, "unexpected oplock(0x%x)\n", 139238c8a9a5SSteve French brk_op->level); 139338c8a9a5SSteve French goto next; 139438c8a9a5SSteve French } 139538c8a9a5SSteve French 139638c8a9a5SSteve French /* Skip oplock being break to none */ 139738c8a9a5SSteve French if (brk_op->is_lease && 139838c8a9a5SSteve French brk_op->o_lease->new_state == SMB2_LEASE_NONE_LE && 139938c8a9a5SSteve French atomic_read(&brk_op->breaking_cnt)) 140038c8a9a5SSteve French goto next; 140138c8a9a5SSteve French 140238c8a9a5SSteve French if (op && op->is_lease && brk_op->is_lease && 140338c8a9a5SSteve French !memcmp(conn->ClientGUID, brk_op->conn->ClientGUID, 140438c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE) && 140538c8a9a5SSteve French !memcmp(op->o_lease->lease_key, brk_op->o_lease->lease_key, 140638c8a9a5SSteve French SMB2_LEASE_KEY_SIZE)) 140738c8a9a5SSteve French goto next; 140838c8a9a5SSteve French brk_op->open_trunc = is_trunc; 140938c8a9a5SSteve French oplock_break(brk_op, SMB2_OPLOCK_LEVEL_NONE); 141038c8a9a5SSteve French next: 141136322523SNamjae Jeon opinfo_conn_put(brk_op); 141238c8a9a5SSteve French rcu_read_lock(); 141338c8a9a5SSteve French } 141438c8a9a5SSteve French rcu_read_unlock(); 141538c8a9a5SSteve French 141638c8a9a5SSteve French if (op) 141738c8a9a5SSteve French opinfo_put(op); 141838c8a9a5SSteve French } 141938c8a9a5SSteve French 142038c8a9a5SSteve French /** 142138c8a9a5SSteve French * smb_break_all_oplock() - break both batch/exclusive and level2 oplock 142238c8a9a5SSteve French * @work: smb work 142338c8a9a5SSteve French * @fp: ksmbd file pointer 142438c8a9a5SSteve French */ 142538c8a9a5SSteve French void smb_break_all_oplock(struct ksmbd_work *work, struct ksmbd_file *fp) 142638c8a9a5SSteve French { 142738c8a9a5SSteve French if (!test_share_config_flag(work->tcon->share_conf, 142838c8a9a5SSteve French KSMBD_SHARE_FLAG_OPLOCKS)) 142938c8a9a5SSteve French return; 143038c8a9a5SSteve French 143138c8a9a5SSteve French smb_break_all_write_oplock(work, fp, 1); 143238c8a9a5SSteve French smb_break_all_levII_oplock(work, fp, 1); 143338c8a9a5SSteve French } 143438c8a9a5SSteve French 143538c8a9a5SSteve French /** 143638c8a9a5SSteve French * smb2_map_lease_to_oplock() - map lease state to corresponding oplock type 143738c8a9a5SSteve French * @lease_state: lease type 143838c8a9a5SSteve French * 143938c8a9a5SSteve French * Return: 0 if no mapping, otherwise corresponding oplock type 144038c8a9a5SSteve French */ 144138c8a9a5SSteve French __u8 smb2_map_lease_to_oplock(__le32 lease_state) 144238c8a9a5SSteve French { 144338c8a9a5SSteve French if (lease_state == (SMB2_LEASE_HANDLE_CACHING_LE | 144438c8a9a5SSteve French SMB2_LEASE_READ_CACHING_LE | 144538c8a9a5SSteve French SMB2_LEASE_WRITE_CACHING_LE)) { 144638c8a9a5SSteve French return SMB2_OPLOCK_LEVEL_BATCH; 144738c8a9a5SSteve French } else if (lease_state != SMB2_LEASE_WRITE_CACHING_LE && 144838c8a9a5SSteve French lease_state & SMB2_LEASE_WRITE_CACHING_LE) { 144938c8a9a5SSteve French if (!(lease_state & SMB2_LEASE_HANDLE_CACHING_LE)) 145038c8a9a5SSteve French return SMB2_OPLOCK_LEVEL_EXCLUSIVE; 145138c8a9a5SSteve French } else if (lease_state & SMB2_LEASE_READ_CACHING_LE) { 145238c8a9a5SSteve French return SMB2_OPLOCK_LEVEL_II; 145338c8a9a5SSteve French } 145438c8a9a5SSteve French return 0; 145538c8a9a5SSteve French } 145638c8a9a5SSteve French 145738c8a9a5SSteve French /** 145838c8a9a5SSteve French * create_lease_buf() - create lease context for open cmd response 145938c8a9a5SSteve French * @rbuf: buffer to create lease context response 146038c8a9a5SSteve French * @lease: buffer to stored parsed lease state information 146138c8a9a5SSteve French */ 146238c8a9a5SSteve French void create_lease_buf(u8 *rbuf, struct lease *lease) 146338c8a9a5SSteve French { 146438c8a9a5SSteve French if (lease->version == 2) { 146538c8a9a5SSteve French struct create_lease_v2 *buf = (struct create_lease_v2 *)rbuf; 146638c8a9a5SSteve French 146738c8a9a5SSteve French memset(buf, 0, sizeof(struct create_lease_v2)); 146838c8a9a5SSteve French memcpy(buf->lcontext.LeaseKey, lease->lease_key, 146938c8a9a5SSteve French SMB2_LEASE_KEY_SIZE); 147038c8a9a5SSteve French buf->lcontext.LeaseFlags = lease->flags; 14719554d493SNamjae Jeon buf->lcontext.Epoch = cpu_to_le16(lease->epoch); 147238c8a9a5SSteve French buf->lcontext.LeaseState = lease->state; 147338c8a9a5SSteve French memcpy(buf->lcontext.ParentLeaseKey, lease->parent_lease_key, 147438c8a9a5SSteve French SMB2_LEASE_KEY_SIZE); 147538c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 147638c8a9a5SSteve French (struct create_lease_v2, lcontext)); 147738c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context_v2)); 147838c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 147938c8a9a5SSteve French (struct create_lease_v2, Name)); 148038c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 148138c8a9a5SSteve French buf->Name[0] = 'R'; 148238c8a9a5SSteve French buf->Name[1] = 'q'; 148338c8a9a5SSteve French buf->Name[2] = 'L'; 148438c8a9a5SSteve French buf->Name[3] = 's'; 148538c8a9a5SSteve French } else { 148638c8a9a5SSteve French struct create_lease *buf = (struct create_lease *)rbuf; 148738c8a9a5SSteve French 148838c8a9a5SSteve French memset(buf, 0, sizeof(struct create_lease)); 148938c8a9a5SSteve French memcpy(buf->lcontext.LeaseKey, lease->lease_key, SMB2_LEASE_KEY_SIZE); 149038c8a9a5SSteve French buf->lcontext.LeaseFlags = lease->flags; 149138c8a9a5SSteve French buf->lcontext.LeaseState = lease->state; 149238c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 149338c8a9a5SSteve French (struct create_lease, lcontext)); 149438c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context)); 149538c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 149638c8a9a5SSteve French (struct create_lease, Name)); 149738c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 149838c8a9a5SSteve French buf->Name[0] = 'R'; 149938c8a9a5SSteve French buf->Name[1] = 'q'; 150038c8a9a5SSteve French buf->Name[2] = 'L'; 150138c8a9a5SSteve French buf->Name[3] = 's'; 150238c8a9a5SSteve French } 150338c8a9a5SSteve French } 150438c8a9a5SSteve French 150538c8a9a5SSteve French /** 150638c8a9a5SSteve French * parse_lease_state() - parse lease context containted in file open request 150738c8a9a5SSteve French * @open_req: buffer containing smb2 file open(create) request 1508572388ffSNamjae Jeon * @is_dir: whether leasing file is directory 150938c8a9a5SSteve French * 151038c8a9a5SSteve French * Return: oplock state, -ENOENT if create lease context not found 151138c8a9a5SSteve French */ 1512572388ffSNamjae Jeon struct lease_ctx_info *parse_lease_state(void *open_req, bool is_dir) 151338c8a9a5SSteve French { 151438c8a9a5SSteve French struct create_context *cc; 151538c8a9a5SSteve French struct smb2_create_req *req = (struct smb2_create_req *)open_req; 1516fc6c6a3cSNamjae Jeon struct lease_ctx_info *lreq; 1517fc6c6a3cSNamjae Jeon 1518fc6c6a3cSNamjae Jeon cc = smb2_find_context_vals(req, SMB2_CREATE_REQUEST_LEASE, 4); 1519fc6c6a3cSNamjae Jeon if (IS_ERR_OR_NULL(cc)) 1520fc6c6a3cSNamjae Jeon return NULL; 1521fc6c6a3cSNamjae Jeon 1522fc6c6a3cSNamjae Jeon lreq = kzalloc(sizeof(struct lease_ctx_info), GFP_KERNEL); 152338c8a9a5SSteve French if (!lreq) 152438c8a9a5SSteve French return NULL; 152538c8a9a5SSteve French 152638c8a9a5SSteve French if (sizeof(struct lease_context_v2) == le32_to_cpu(cc->DataLength)) { 152738c8a9a5SSteve French struct create_lease_v2 *lc = (struct create_lease_v2 *)cc; 152838c8a9a5SSteve French 152938c8a9a5SSteve French memcpy(lreq->lease_key, lc->lcontext.LeaseKey, SMB2_LEASE_KEY_SIZE); 15303c1e602aSNamjae Jeon if (is_dir) { 1531572388ffSNamjae Jeon lreq->req_state = lc->lcontext.LeaseState & 1532572388ffSNamjae Jeon ~SMB2_LEASE_WRITE_CACHING_LE; 15333c1e602aSNamjae Jeon lreq->is_dir = true; 15343c1e602aSNamjae Jeon } else 153538c8a9a5SSteve French lreq->req_state = lc->lcontext.LeaseState; 153638c8a9a5SSteve French lreq->flags = lc->lcontext.LeaseFlags; 1537bc025d49SNamjae Jeon lreq->epoch = lc->lcontext.Epoch; 153838c8a9a5SSteve French lreq->duration = lc->lcontext.LeaseDuration; 153938c8a9a5SSteve French memcpy(lreq->parent_lease_key, lc->lcontext.ParentLeaseKey, 154038c8a9a5SSteve French SMB2_LEASE_KEY_SIZE); 154138c8a9a5SSteve French lreq->version = 2; 154238c8a9a5SSteve French } else { 154338c8a9a5SSteve French struct create_lease *lc = (struct create_lease *)cc; 154438c8a9a5SSteve French 154538c8a9a5SSteve French memcpy(lreq->lease_key, lc->lcontext.LeaseKey, SMB2_LEASE_KEY_SIZE); 154638c8a9a5SSteve French lreq->req_state = lc->lcontext.LeaseState; 154738c8a9a5SSteve French lreq->flags = lc->lcontext.LeaseFlags; 154838c8a9a5SSteve French lreq->duration = lc->lcontext.LeaseDuration; 154938c8a9a5SSteve French lreq->version = 1; 155038c8a9a5SSteve French } 155138c8a9a5SSteve French return lreq; 155238c8a9a5SSteve French } 155338c8a9a5SSteve French 155438c8a9a5SSteve French /** 155538c8a9a5SSteve French * smb2_find_context_vals() - find a particular context info in open request 155638c8a9a5SSteve French * @open_req: buffer containing smb2 file open(create) request 155738c8a9a5SSteve French * @tag: context name to search for 155838c8a9a5SSteve French * @tag_len: the length of tag 155938c8a9a5SSteve French * 156038c8a9a5SSteve French * Return: pointer to requested context, NULL if @str context not found 156138c8a9a5SSteve French * or error pointer if name length is invalid. 156238c8a9a5SSteve French */ 156338c8a9a5SSteve French struct create_context *smb2_find_context_vals(void *open_req, const char *tag, int tag_len) 156438c8a9a5SSteve French { 156538c8a9a5SSteve French struct create_context *cc; 156638c8a9a5SSteve French unsigned int next = 0; 156738c8a9a5SSteve French char *name; 156838c8a9a5SSteve French struct smb2_create_req *req = (struct smb2_create_req *)open_req; 156938c8a9a5SSteve French unsigned int remain_len, name_off, name_len, value_off, value_len, 157038c8a9a5SSteve French cc_len; 157138c8a9a5SSteve French 157238c8a9a5SSteve French /* 157338c8a9a5SSteve French * CreateContextsOffset and CreateContextsLength are guaranteed to 157438c8a9a5SSteve French * be valid because of ksmbd_smb2_check_message(). 157538c8a9a5SSteve French */ 157638c8a9a5SSteve French cc = (struct create_context *)((char *)req + 157738c8a9a5SSteve French le32_to_cpu(req->CreateContextsOffset)); 157838c8a9a5SSteve French remain_len = le32_to_cpu(req->CreateContextsLength); 157938c8a9a5SSteve French do { 158038c8a9a5SSteve French cc = (struct create_context *)((char *)cc + next); 158138c8a9a5SSteve French if (remain_len < offsetof(struct create_context, Buffer)) 158238c8a9a5SSteve French return ERR_PTR(-EINVAL); 158338c8a9a5SSteve French 158438c8a9a5SSteve French next = le32_to_cpu(cc->Next); 158538c8a9a5SSteve French name_off = le16_to_cpu(cc->NameOffset); 158638c8a9a5SSteve French name_len = le16_to_cpu(cc->NameLength); 158738c8a9a5SSteve French value_off = le16_to_cpu(cc->DataOffset); 158838c8a9a5SSteve French value_len = le32_to_cpu(cc->DataLength); 158938c8a9a5SSteve French cc_len = next ? next : remain_len; 159038c8a9a5SSteve French 159138c8a9a5SSteve French if ((next & 0x7) != 0 || 159238c8a9a5SSteve French next > remain_len || 159338c8a9a5SSteve French name_off != offsetof(struct create_context, Buffer) || 159438c8a9a5SSteve French name_len < 4 || 159538c8a9a5SSteve French name_off + name_len > cc_len || 159638c8a9a5SSteve French (value_off & 0x7) != 0 || 159717d5b135SNamjae Jeon (value_len && value_off < name_off + (name_len < 8 ? 8 : name_len)) || 159838c8a9a5SSteve French ((u64)value_off + value_len > cc_len)) 159938c8a9a5SSteve French return ERR_PTR(-EINVAL); 160038c8a9a5SSteve French 160138c8a9a5SSteve French name = (char *)cc + name_off; 160238c8a9a5SSteve French if (name_len == tag_len && !memcmp(name, tag, name_len)) 160338c8a9a5SSteve French return cc; 160438c8a9a5SSteve French 160538c8a9a5SSteve French remain_len -= next; 160638c8a9a5SSteve French } while (next != 0); 160738c8a9a5SSteve French 160838c8a9a5SSteve French return NULL; 160938c8a9a5SSteve French } 161038c8a9a5SSteve French 161138c8a9a5SSteve French /** 161238c8a9a5SSteve French * create_durable_rsp_buf() - create durable handle context 161338c8a9a5SSteve French * @cc: buffer to create durable context response 161438c8a9a5SSteve French */ 161538c8a9a5SSteve French void create_durable_rsp_buf(char *cc) 161638c8a9a5SSteve French { 161738c8a9a5SSteve French struct create_durable_rsp *buf; 161838c8a9a5SSteve French 161938c8a9a5SSteve French buf = (struct create_durable_rsp *)cc; 162038c8a9a5SSteve French memset(buf, 0, sizeof(struct create_durable_rsp)); 162138c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 162238c8a9a5SSteve French (struct create_durable_rsp, Data)); 162338c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(8); 162438c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 162538c8a9a5SSteve French (struct create_durable_rsp, Name)); 162638c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 162738c8a9a5SSteve French /* SMB2_CREATE_DURABLE_HANDLE_RESPONSE is "DHnQ" */ 162838c8a9a5SSteve French buf->Name[0] = 'D'; 162938c8a9a5SSteve French buf->Name[1] = 'H'; 163038c8a9a5SSteve French buf->Name[2] = 'n'; 163138c8a9a5SSteve French buf->Name[3] = 'Q'; 163238c8a9a5SSteve French } 163338c8a9a5SSteve French 163438c8a9a5SSteve French /** 163538c8a9a5SSteve French * create_durable_v2_rsp_buf() - create durable handle v2 context 163638c8a9a5SSteve French * @cc: buffer to create durable context response 163738c8a9a5SSteve French * @fp: ksmbd file pointer 163838c8a9a5SSteve French */ 163938c8a9a5SSteve French void create_durable_v2_rsp_buf(char *cc, struct ksmbd_file *fp) 164038c8a9a5SSteve French { 164138c8a9a5SSteve French struct create_durable_v2_rsp *buf; 164238c8a9a5SSteve French 164338c8a9a5SSteve French buf = (struct create_durable_v2_rsp *)cc; 164438c8a9a5SSteve French memset(buf, 0, sizeof(struct create_durable_rsp)); 164538c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 164638c8a9a5SSteve French (struct create_durable_rsp, Data)); 164738c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(8); 164838c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 164938c8a9a5SSteve French (struct create_durable_rsp, Name)); 165038c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 165138c8a9a5SSteve French /* SMB2_CREATE_DURABLE_HANDLE_RESPONSE_V2 is "DH2Q" */ 165238c8a9a5SSteve French buf->Name[0] = 'D'; 165338c8a9a5SSteve French buf->Name[1] = 'H'; 165438c8a9a5SSteve French buf->Name[2] = '2'; 165538c8a9a5SSteve French buf->Name[3] = 'Q'; 165638c8a9a5SSteve French 165738c8a9a5SSteve French buf->Timeout = cpu_to_le32(fp->durable_timeout); 165838c8a9a5SSteve French } 165938c8a9a5SSteve French 166038c8a9a5SSteve French /** 166138c8a9a5SSteve French * create_mxac_rsp_buf() - create query maximal access context 166238c8a9a5SSteve French * @cc: buffer to create maximal access context response 166338c8a9a5SSteve French * @maximal_access: maximal access 166438c8a9a5SSteve French */ 166538c8a9a5SSteve French void create_mxac_rsp_buf(char *cc, int maximal_access) 166638c8a9a5SSteve French { 166738c8a9a5SSteve French struct create_mxac_rsp *buf; 166838c8a9a5SSteve French 166938c8a9a5SSteve French buf = (struct create_mxac_rsp *)cc; 167038c8a9a5SSteve French memset(buf, 0, sizeof(struct create_mxac_rsp)); 167138c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 167238c8a9a5SSteve French (struct create_mxac_rsp, QueryStatus)); 167338c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(8); 167438c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 167538c8a9a5SSteve French (struct create_mxac_rsp, Name)); 167638c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 167738c8a9a5SSteve French /* SMB2_CREATE_QUERY_MAXIMAL_ACCESS_RESPONSE is "MxAc" */ 167838c8a9a5SSteve French buf->Name[0] = 'M'; 167938c8a9a5SSteve French buf->Name[1] = 'x'; 168038c8a9a5SSteve French buf->Name[2] = 'A'; 168138c8a9a5SSteve French buf->Name[3] = 'c'; 168238c8a9a5SSteve French 168338c8a9a5SSteve French buf->QueryStatus = STATUS_SUCCESS; 168438c8a9a5SSteve French buf->MaximalAccess = cpu_to_le32(maximal_access); 168538c8a9a5SSteve French } 168638c8a9a5SSteve French 168738c8a9a5SSteve French void create_disk_id_rsp_buf(char *cc, __u64 file_id, __u64 vol_id) 168838c8a9a5SSteve French { 168938c8a9a5SSteve French struct create_disk_id_rsp *buf; 169038c8a9a5SSteve French 169138c8a9a5SSteve French buf = (struct create_disk_id_rsp *)cc; 169238c8a9a5SSteve French memset(buf, 0, sizeof(struct create_disk_id_rsp)); 169338c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 169438c8a9a5SSteve French (struct create_disk_id_rsp, DiskFileId)); 169538c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(32); 169638c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 169738c8a9a5SSteve French (struct create_mxac_rsp, Name)); 169838c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(4); 169938c8a9a5SSteve French /* SMB2_CREATE_QUERY_ON_DISK_ID_RESPONSE is "QFid" */ 170038c8a9a5SSteve French buf->Name[0] = 'Q'; 170138c8a9a5SSteve French buf->Name[1] = 'F'; 170238c8a9a5SSteve French buf->Name[2] = 'i'; 170338c8a9a5SSteve French buf->Name[3] = 'd'; 170438c8a9a5SSteve French 170538c8a9a5SSteve French buf->DiskFileId = cpu_to_le64(file_id); 170638c8a9a5SSteve French buf->VolumeId = cpu_to_le64(vol_id); 170738c8a9a5SSteve French } 170838c8a9a5SSteve French 170938c8a9a5SSteve French /** 171038c8a9a5SSteve French * create_posix_rsp_buf() - create posix extension context 171138c8a9a5SSteve French * @cc: buffer to create posix on posix response 171238c8a9a5SSteve French * @fp: ksmbd file pointer 171338c8a9a5SSteve French */ 171438c8a9a5SSteve French void create_posix_rsp_buf(char *cc, struct ksmbd_file *fp) 171538c8a9a5SSteve French { 171638c8a9a5SSteve French struct create_posix_rsp *buf; 171738c8a9a5SSteve French struct inode *inode = file_inode(fp->filp); 171838c8a9a5SSteve French struct mnt_idmap *idmap = file_mnt_idmap(fp->filp); 171938c8a9a5SSteve French vfsuid_t vfsuid = i_uid_into_vfsuid(idmap, inode); 172038c8a9a5SSteve French vfsgid_t vfsgid = i_gid_into_vfsgid(idmap, inode); 172138c8a9a5SSteve French 172238c8a9a5SSteve French buf = (struct create_posix_rsp *)cc; 172338c8a9a5SSteve French memset(buf, 0, sizeof(struct create_posix_rsp)); 172438c8a9a5SSteve French buf->ccontext.DataOffset = cpu_to_le16(offsetof 172538c8a9a5SSteve French (struct create_posix_rsp, nlink)); 172638c8a9a5SSteve French /* 172738c8a9a5SSteve French * DataLength = nlink(4) + reparse_tag(4) + mode(4) + 172838c8a9a5SSteve French * domain sid(28) + unix group sid(16). 172938c8a9a5SSteve French */ 173038c8a9a5SSteve French buf->ccontext.DataLength = cpu_to_le32(56); 173138c8a9a5SSteve French buf->ccontext.NameOffset = cpu_to_le16(offsetof 173238c8a9a5SSteve French (struct create_posix_rsp, Name)); 173338c8a9a5SSteve French buf->ccontext.NameLength = cpu_to_le16(POSIX_CTXT_DATA_LEN); 173438c8a9a5SSteve French /* SMB2_CREATE_TAG_POSIX is "0x93AD25509CB411E7B42383DE968BCD7C" */ 173538c8a9a5SSteve French buf->Name[0] = 0x93; 173638c8a9a5SSteve French buf->Name[1] = 0xAD; 173738c8a9a5SSteve French buf->Name[2] = 0x25; 173838c8a9a5SSteve French buf->Name[3] = 0x50; 173938c8a9a5SSteve French buf->Name[4] = 0x9C; 174038c8a9a5SSteve French buf->Name[5] = 0xB4; 174138c8a9a5SSteve French buf->Name[6] = 0x11; 174238c8a9a5SSteve French buf->Name[7] = 0xE7; 174338c8a9a5SSteve French buf->Name[8] = 0xB4; 174438c8a9a5SSteve French buf->Name[9] = 0x23; 174538c8a9a5SSteve French buf->Name[10] = 0x83; 174638c8a9a5SSteve French buf->Name[11] = 0xDE; 174738c8a9a5SSteve French buf->Name[12] = 0x96; 174838c8a9a5SSteve French buf->Name[13] = 0x8B; 174938c8a9a5SSteve French buf->Name[14] = 0xCD; 175038c8a9a5SSteve French buf->Name[15] = 0x7C; 175138c8a9a5SSteve French 175238c8a9a5SSteve French buf->nlink = cpu_to_le32(inode->i_nlink); 175338c8a9a5SSteve French buf->reparse_tag = cpu_to_le32(fp->volatile_id); 175438c8a9a5SSteve French buf->mode = cpu_to_le32(inode->i_mode & 0777); 175538c8a9a5SSteve French /* 175638c8a9a5SSteve French * SidBuffer(44) contain two sids(Domain sid(28), UNIX group sid(16)). 175738c8a9a5SSteve French * Domain sid(28) = revision(1) + num_subauth(1) + authority(6) + 175838c8a9a5SSteve French * sub_auth(4 * 4(num_subauth)) + RID(4). 175938c8a9a5SSteve French * UNIX group id(16) = revision(1) + num_subauth(1) + authority(6) + 176038c8a9a5SSteve French * sub_auth(4 * 1(num_subauth)) + RID(4). 176138c8a9a5SSteve French */ 176238c8a9a5SSteve French id_to_sid(from_kuid_munged(&init_user_ns, vfsuid_into_kuid(vfsuid)), 176338c8a9a5SSteve French SIDOWNER, (struct smb_sid *)&buf->SidBuffer[0]); 176438c8a9a5SSteve French id_to_sid(from_kgid_munged(&init_user_ns, vfsgid_into_kgid(vfsgid)), 176538c8a9a5SSteve French SIDUNIX_GROUP, (struct smb_sid *)&buf->SidBuffer[28]); 176638c8a9a5SSteve French } 176738c8a9a5SSteve French 176838c8a9a5SSteve French /* 176938c8a9a5SSteve French * Find lease object(opinfo) for given lease key/fid from lease 177038c8a9a5SSteve French * break/file close path. 177138c8a9a5SSteve French */ 177238c8a9a5SSteve French /** 177338c8a9a5SSteve French * lookup_lease_in_table() - find a matching lease info object 177438c8a9a5SSteve French * @conn: connection instance 177538c8a9a5SSteve French * @lease_key: lease key to be searched for 177638c8a9a5SSteve French * 177738c8a9a5SSteve French * Return: opinfo if found matching opinfo, otherwise NULL 177838c8a9a5SSteve French */ 177938c8a9a5SSteve French struct oplock_info *lookup_lease_in_table(struct ksmbd_conn *conn, 178038c8a9a5SSteve French char *lease_key) 178138c8a9a5SSteve French { 178238c8a9a5SSteve French struct oplock_info *opinfo = NULL, *ret_op = NULL; 178338c8a9a5SSteve French struct lease_table *lt; 178438c8a9a5SSteve French int ret; 178538c8a9a5SSteve French 178638c8a9a5SSteve French read_lock(&lease_list_lock); 178738c8a9a5SSteve French list_for_each_entry(lt, &lease_table_list, l_entry) { 178838c8a9a5SSteve French if (!memcmp(lt->client_guid, conn->ClientGUID, 178938c8a9a5SSteve French SMB2_CLIENT_GUID_SIZE)) 179038c8a9a5SSteve French goto found; 179138c8a9a5SSteve French } 179238c8a9a5SSteve French 179338c8a9a5SSteve French read_unlock(&lease_list_lock); 179438c8a9a5SSteve French return NULL; 179538c8a9a5SSteve French 179638c8a9a5SSteve French found: 179738c8a9a5SSteve French rcu_read_lock(); 179838c8a9a5SSteve French list_for_each_entry_rcu(opinfo, <->lease_list, lease_entry) { 179938c8a9a5SSteve French if (!atomic_inc_not_zero(&opinfo->refcount)) 180038c8a9a5SSteve French continue; 180138c8a9a5SSteve French rcu_read_unlock(); 180238c8a9a5SSteve French if (!opinfo->op_state || opinfo->op_state == OPLOCK_CLOSING) 180338c8a9a5SSteve French goto op_next; 180438c8a9a5SSteve French if (!(opinfo->o_lease->state & 180538c8a9a5SSteve French (SMB2_LEASE_HANDLE_CACHING_LE | 180638c8a9a5SSteve French SMB2_LEASE_WRITE_CACHING_LE))) 180738c8a9a5SSteve French goto op_next; 180838c8a9a5SSteve French ret = compare_guid_key(opinfo, conn->ClientGUID, 180938c8a9a5SSteve French lease_key); 181038c8a9a5SSteve French if (ret) { 181138c8a9a5SSteve French ksmbd_debug(OPLOCK, "found opinfo\n"); 181238c8a9a5SSteve French ret_op = opinfo; 181338c8a9a5SSteve French goto out; 181438c8a9a5SSteve French } 181538c8a9a5SSteve French op_next: 181638c8a9a5SSteve French opinfo_put(opinfo); 181738c8a9a5SSteve French rcu_read_lock(); 181838c8a9a5SSteve French } 181938c8a9a5SSteve French rcu_read_unlock(); 182038c8a9a5SSteve French 182138c8a9a5SSteve French out: 182238c8a9a5SSteve French read_unlock(&lease_list_lock); 182338c8a9a5SSteve French return ret_op; 182438c8a9a5SSteve French } 1825