| /* |
| * fs/nfs/nfs4session.c |
| * |
| * Copyright (c) 2012 Trond Myklebust <Trond.Myklebust@netapp.com> |
| * |
| */ |
| #include <linux/kernel.h> |
| #include <linux/errno.h> |
| #include <linux/string.h> |
| #include <linux/printk.h> |
| #include <linux/slab.h> |
| #include <linux/sunrpc/sched.h> |
| #include <linux/sunrpc/bc_xprt.h> |
| #include <linux/nfs.h> |
| #include <linux/nfs4.h> |
| #include <linux/nfs_fs.h> |
| #include <linux/module.h> |
| |
| #include "nfs4_fs.h" |
| #include "internal.h" |
| #include "nfs4session.h" |
| #include "callback.h" |
| |
| #define NFSDBG_FACILITY NFSDBG_STATE |
| |
| static void nfs4_init_slot_table(struct nfs4_slot_table *tbl, const char *queue) |
| { |
| tbl->highest_used_slotid = NFS4_NO_SLOT; |
| spin_lock_init(&tbl->slot_tbl_lock); |
| rpc_init_priority_wait_queue(&tbl->slot_tbl_waitq, queue); |
| init_waitqueue_head(&tbl->slot_waitq); |
| init_completion(&tbl->complete); |
| } |
| |
| /* |
| * nfs4_shrink_slot_table - free retired slots from the slot table |
| */ |
| static void nfs4_shrink_slot_table(struct nfs4_slot_table *tbl, u32 newsize) |
| { |
| struct nfs4_slot **p; |
| if (newsize >= tbl->max_slots) |
| return; |
| |
| p = &tbl->slots; |
| while (newsize--) |
| p = &(*p)->next; |
| while (*p) { |
| struct nfs4_slot *slot = *p; |
| |
| *p = slot->next; |
| kfree(slot); |
| tbl->max_slots--; |
| } |
| } |
| |
| /** |
| * nfs4_slot_tbl_drain_complete - wake waiters when drain is complete |
| * @tbl - controlling slot table |
| * |
| */ |
| void nfs4_slot_tbl_drain_complete(struct nfs4_slot_table *tbl) |
| { |
| if (nfs4_slot_tbl_draining(tbl)) |
| complete(&tbl->complete); |
| } |
| |
| /* |
| * nfs4_free_slot - free a slot and efficiently update slot table. |
| * |
| * freeing a slot is trivially done by clearing its respective bit |
| * in the bitmap. |
| * If the freed slotid equals highest_used_slotid we want to update it |
| * so that the server would be able to size down the slot table if needed, |
| * otherwise we know that the highest_used_slotid is still in use. |
| * When updating highest_used_slotid there may be "holes" in the bitmap |
| * so we need to scan down from highest_used_slotid to 0 looking for the now |
| * highest slotid in use. |
| * If none found, highest_used_slotid is set to NFS4_NO_SLOT. |
| * |
| * Must be called while holding tbl->slot_tbl_lock |
| */ |
| void nfs4_free_slot(struct nfs4_slot_table *tbl, struct nfs4_slot *slot) |
| { |
| u32 slotid = slot->slot_nr; |
| |
| /* clear used bit in bitmap */ |
| __clear_bit(slotid, tbl->used_slots); |
| |
| /* update highest_used_slotid when it is freed */ |
| if (slotid == tbl->highest_used_slotid) { |
| u32 new_max = find_last_bit(tbl->used_slots, slotid); |
| if (new_max < slotid) |
| tbl->highest_used_slotid = new_max; |
| else { |
| tbl->highest_used_slotid = NFS4_NO_SLOT; |
| nfs4_slot_tbl_drain_complete(tbl); |
| } |
| } |
| dprintk("%s: slotid %u highest_used_slotid %u\n", __func__, |
| slotid, tbl->highest_used_slotid); |
| } |
| |
| static struct nfs4_slot *nfs4_new_slot(struct nfs4_slot_table *tbl, |
| u32 slotid, u32 seq_init, gfp_t gfp_mask) |
| { |
| struct nfs4_slot *slot; |
| |
| slot = kzalloc(sizeof(*slot), gfp_mask); |
| if (slot) { |
| slot->table = tbl; |
| slot->slot_nr = slotid; |
| slot->seq_nr = seq_init; |
| } |
| return slot; |
| } |
| |
| static struct nfs4_slot *nfs4_find_or_create_slot(struct nfs4_slot_table *tbl, |
| u32 slotid, u32 seq_init, gfp_t gfp_mask) |
| { |
| struct nfs4_slot **p, *slot; |
| |
| p = &tbl->slots; |
| for (;;) { |
| if (*p == NULL) { |
| *p = nfs4_new_slot(tbl, tbl->max_slots, |
| seq_init, gfp_mask); |
| if (*p == NULL) |
| break; |
| tbl->max_slots++; |
| } |
| slot = *p; |
| if (slot->slot_nr == slotid) |
| return slot; |
| p = &slot->next; |
| } |
| return ERR_PTR(-ENOMEM); |
| } |
| |
| static void nfs4_lock_slot(struct nfs4_slot_table *tbl, |
| struct nfs4_slot *slot) |
| { |
| u32 slotid = slot->slot_nr; |
| |
| __set_bit(slotid, tbl->used_slots); |
| if (slotid > tbl->highest_used_slotid || |
| tbl->highest_used_slotid == NFS4_NO_SLOT) |
| tbl->highest_used_slotid = slotid; |
| slot->generation = tbl->generation; |
| } |
| |
| /* |
| * nfs4_try_to_lock_slot - Given a slot try to allocate it |
| * |
| * Note: must be called with the slot_tbl_lock held. |
| */ |
| bool nfs4_try_to_lock_slot(struct nfs4_slot_table *tbl, struct nfs4_slot *slot) |
| { |
| if (nfs4_test_locked_slot(tbl, slot->slot_nr)) |
| return false; |
| nfs4_lock_slot(tbl, slot); |
| return true; |
| } |
| |
| /* |
| * nfs4_lookup_slot - Find a slot but don't allocate it |
| * |
| * Note: must be called with the slot_tbl_lock held. |
| */ |
| struct nfs4_slot *nfs4_lookup_slot(struct nfs4_slot_table *tbl, u32 slotid) |
| { |
| if (slotid <= tbl->max_slotid) |
| return nfs4_find_or_create_slot(tbl, slotid, 0, GFP_NOWAIT); |
| return ERR_PTR(-E2BIG); |
| } |
| |
| static int nfs4_slot_get_seqid(struct nfs4_slot_table *tbl, u32 slotid, |
| u32 *seq_nr) |
| __must_hold(&tbl->slot_tbl_lock) |
| { |
| struct nfs4_slot *slot; |
| int ret; |
| |
| slot = nfs4_lookup_slot(tbl, slotid); |
| ret = PTR_ERR_OR_ZERO(slot); |
| if (!ret) |
| *seq_nr = slot->seq_nr; |
| |
| return ret; |
| } |
| |
| /* |
| * nfs4_slot_seqid_in_use - test if a slot sequence id is still in use |
| * |
| * Given a slot table, slot id and sequence number, determine if the |
| * RPC call in question is still in flight. This function is mainly |
| * intended for use by the callback channel. |
| */ |
| static bool nfs4_slot_seqid_in_use(struct nfs4_slot_table *tbl, |
| u32 slotid, u32 seq_nr) |
| { |
| u32 cur_seq = 0; |
| bool ret = false; |
| |
| spin_lock(&tbl->slot_tbl_lock); |
| if (nfs4_slot_get_seqid(tbl, slotid, &cur_seq) == 0 && |
| cur_seq == seq_nr && test_bit(slotid, tbl->used_slots)) |
| ret = true; |
| spin_unlock(&tbl->slot_tbl_lock); |
| return ret; |
| } |
| |
| /* |
| * nfs4_slot_wait_on_seqid - wait until a slot sequence id is complete |
| * |
| * Given a slot table, slot id and sequence number, wait until the |
| * corresponding RPC call completes. This function is mainly |
| * intended for use by the callback channel. |
| */ |
| int nfs4_slot_wait_on_seqid(struct nfs4_slot_table *tbl, |
| u32 slotid, u32 seq_nr, |
| unsigned long timeout) |
| { |
| if (wait_event_timeout(tbl->slot_waitq, |
| !nfs4_slot_seqid_in_use(tbl, slotid, seq_nr), |
| timeout) == 0) |
| return -ETIMEDOUT; |
| return 0; |
| } |
| |
| /* |
| * nfs4_alloc_slot - efficiently look for a free slot |
| * |
| * nfs4_alloc_slot looks for an unset bit in the used_slots bitmap. |
| * If found, we mark the slot as used, update the highest_used_slotid, |
| * and respectively set up the sequence operation args. |
| * |
| * Note: must be called with under the slot_tbl_lock. |
| */ |
| struct nfs4_slot *nfs4_alloc_slot(struct nfs4_slot_table *tbl) |
| { |
| struct nfs4_slot *ret = ERR_PTR(-EBUSY); |
| u32 slotid; |
| |
| dprintk("--> %s used_slots=%04lx highest_used=%u max_slots=%u\n", |
| __func__, tbl->used_slots[0], tbl->highest_used_slotid, |
| tbl->max_slotid + 1); |
| slotid = find_first_zero_bit(tbl->used_slots, tbl->max_slotid + 1); |
| if (slotid <= tbl->max_slotid) { |
| ret = nfs4_find_or_create_slot(tbl, slotid, 1, GFP_NOWAIT); |
| if (!IS_ERR(ret)) |
| nfs4_lock_slot(tbl, ret); |
| } |
| dprintk("<-- %s used_slots=%04lx highest_used=%u slotid=%u\n", |
| __func__, tbl->used_slots[0], tbl->highest_used_slotid, |
| !IS_ERR(ret) ? ret->slot_nr : NFS4_NO_SLOT); |
| return ret; |
| } |
| |
| static int nfs4_grow_slot_table(struct nfs4_slot_table *tbl, |
| u32 max_reqs, u32 ivalue) |
| { |
| if (max_reqs <= tbl->max_slots) |
| return 0; |
| if (!IS_ERR(nfs4_find_or_create_slot(tbl, max_reqs - 1, ivalue, GFP_NOFS))) |
| return 0; |
| return -ENOMEM; |
| } |
| |
| static void nfs4_reset_slot_table(struct nfs4_slot_table *tbl, |
| u32 server_highest_slotid, |
| u32 ivalue) |
| { |
| struct nfs4_slot **p; |
| |
| nfs4_shrink_slot_table(tbl, server_highest_slotid + 1); |
| p = &tbl->slots; |
| while (*p) { |
| (*p)->seq_nr = ivalue; |
| (*p)->interrupted = 0; |
| p = &(*p)->next; |
| } |
| tbl->highest_used_slotid = NFS4_NO_SLOT; |
| tbl->target_highest_slotid = server_highest_slotid; |
| tbl->server_highest_slotid = server_highest_slotid; |
| tbl->d_target_highest_slotid = 0; |
| tbl->d2_target_highest_slotid = 0; |
| tbl->max_slotid = server_highest_slotid; |
| } |
| |
| /* |
| * (re)Initialise a slot table |
| */ |
| static int nfs4_realloc_slot_table(struct nfs4_slot_table *tbl, |
| u32 max_reqs, u32 ivalue) |
| { |
| int ret; |
| |
| dprintk("--> %s: max_reqs=%u, tbl->max_slots %u\n", __func__, |
| max_reqs, tbl->max_slots); |
| |
| if (max_reqs > NFS4_MAX_SLOT_TABLE) |
| max_reqs = NFS4_MAX_SLOT_TABLE; |
| |
| ret = nfs4_grow_slot_table(tbl, max_reqs, ivalue); |
| if (ret) |
| goto out; |
| |
| spin_lock(&tbl->slot_tbl_lock); |
| nfs4_reset_slot_table(tbl, max_reqs - 1, ivalue); |
| spin_unlock(&tbl->slot_tbl_lock); |
| |
| dprintk("%s: tbl=%p slots=%p max_slots=%u\n", __func__, |
| tbl, tbl->slots, tbl->max_slots); |
| out: |
| dprintk("<-- %s: return %d\n", __func__, ret); |
| return ret; |
| } |
| |
| /* |
| * nfs4_release_slot_table - release all slot table entries |
| */ |
| static void nfs4_release_slot_table(struct nfs4_slot_table *tbl) |
| { |
| nfs4_shrink_slot_table(tbl, 0); |
| } |
| |
| /** |
| * nfs4_shutdown_slot_table - release resources attached to a slot table |
| * @tbl: slot table to shut down |
| * |
| */ |
| void nfs4_shutdown_slot_table(struct nfs4_slot_table *tbl) |
| { |
| nfs4_release_slot_table(tbl); |
| rpc_destroy_wait_queue(&tbl->slot_tbl_waitq); |
| } |
| |
| /** |
| * nfs4_setup_slot_table - prepare a stand-alone slot table for use |
| * @tbl: slot table to set up |
| * @max_reqs: maximum number of requests allowed |
| * @queue: name to give RPC wait queue |
| * |
| * Returns zero on success, or a negative errno. |
| */ |
| int nfs4_setup_slot_table(struct nfs4_slot_table *tbl, unsigned int max_reqs, |
| const char *queue) |
| { |
| nfs4_init_slot_table(tbl, queue); |
| return nfs4_realloc_slot_table(tbl, max_reqs, 0); |
| } |
| |
| static bool nfs41_assign_slot(struct rpc_task *task, void *pslot) |
| { |
| struct nfs4_sequence_args *args = task->tk_msg.rpc_argp; |
| struct nfs4_sequence_res *res = task->tk_msg.rpc_resp; |
| struct nfs4_slot *slot = pslot; |
| struct nfs4_slot_table *tbl = slot->table; |
| |
| if (nfs4_slot_tbl_draining(tbl) && !args->sa_privileged) |
| return false; |
| slot->generation = tbl->generation; |
| args->sa_slot = slot; |
| res->sr_timestamp = jiffies; |
| res->sr_slot = slot; |
| res->sr_status_flags = 0; |
| res->sr_status = 1; |
| return true; |
| } |
| |
| static bool __nfs41_wake_and_assign_slot(struct nfs4_slot_table *tbl, |
| struct nfs4_slot *slot) |
| { |
| if (rpc_wake_up_first(&tbl->slot_tbl_waitq, nfs41_assign_slot, slot)) |
| return true; |
| return false; |
| } |
| |
| bool nfs41_wake_and_assign_slot(struct nfs4_slot_table *tbl, |
| struct nfs4_slot *slot) |
| { |
| if (slot->slot_nr > tbl->max_slotid) |
| return false; |
| return __nfs41_wake_and_assign_slot(tbl, slot); |
| } |
| |
| static bool nfs41_try_wake_next_slot_table_entry(struct nfs4_slot_table *tbl) |
| { |
| struct nfs4_slot *slot = nfs4_alloc_slot(tbl); |
| if (!IS_ERR(slot)) { |
| bool ret = __nfs41_wake_and_assign_slot(tbl, slot); |
| if (ret) |
| return ret; |
| nfs4_free_slot(tbl, slot); |
| } |
| return false; |
| } |
| |
| void nfs41_wake_slot_table(struct nfs4_slot_table *tbl) |
| { |
| for (;;) { |
| if (!nfs41_try_wake_next_slot_table_entry(tbl)) |
| break; |
| } |
| } |
| |
| #if defined(CONFIG_NFS_V4_1) |
| |
| static void nfs41_set_max_slotid_locked(struct nfs4_slot_table *tbl, |
| u32 target_highest_slotid) |
| { |
| u32 max_slotid; |
| |
| max_slotid = min(NFS4_MAX_SLOT_TABLE - 1, target_highest_slotid); |
| if (max_slotid > tbl->server_highest_slotid) |
| max_slotid = tbl->server_highest_slotid; |
| if (max_slotid > tbl->target_highest_slotid) |
| max_slotid = tbl->target_highest_slotid; |
| tbl->max_slotid = max_slotid; |
| nfs41_wake_slot_table(tbl); |
| } |
| |
| /* Update the client's idea of target_highest_slotid */ |
| static void nfs41_set_target_slotid_locked(struct nfs4_slot_table *tbl, |
| u32 target_highest_slotid) |
| { |
| if (tbl->target_highest_slotid == target_highest_slotid) |
| return; |
| tbl->target_highest_slotid = target_highest_slotid; |
| tbl->generation++; |
| } |
| |
| void nfs41_set_target_slotid(struct nfs4_slot_table *tbl, |
| u32 target_highest_slotid) |
| { |
| spin_lock(&tbl->slot_tbl_lock); |
| nfs41_set_target_slotid_locked(tbl, target_highest_slotid); |
| tbl->d_target_highest_slotid = 0; |
| tbl->d2_target_highest_slotid = 0; |
| nfs41_set_max_slotid_locked(tbl, target_highest_slotid); |
| spin_unlock(&tbl->slot_tbl_lock); |
| } |
| |
| static void nfs41_set_server_slotid_locked(struct nfs4_slot_table *tbl, |
| u32 highest_slotid) |
| { |
| if (tbl->server_highest_slotid == highest_slotid) |
| return; |
| if (tbl->highest_used_slotid > highest_slotid) |
| return; |
| /* Deallocate slots */ |
| nfs4_shrink_slot_table(tbl, highest_slotid + 1); |
| tbl->server_highest_slotid = highest_slotid; |
| } |
| |
| static s32 nfs41_derivative_target_slotid(s32 s1, s32 s2) |
| { |
| s1 -= s2; |
| if (s1 == 0) |
| return 0; |
| if (s1 < 0) |
| return (s1 - 1) >> 1; |
| return (s1 + 1) >> 1; |
| } |
| |
| static int nfs41_sign_s32(s32 s1) |
| { |
| if (s1 > 0) |
| return 1; |
| if (s1 < 0) |
| return -1; |
| return 0; |
| } |
| |
| static bool nfs41_same_sign_or_zero_s32(s32 s1, s32 s2) |
| { |
| if (!s1 || !s2) |
| return true; |
| return nfs41_sign_s32(s1) == nfs41_sign_s32(s2); |
| } |
| |
| /* Try to eliminate outliers by checking for sharp changes in the |
| * derivatives and second derivatives |
| */ |
| static bool nfs41_is_outlier_target_slotid(struct nfs4_slot_table *tbl, |
| u32 new_target) |
| { |
| s32 d_target, d2_target; |
| bool ret = true; |
| |
| d_target = nfs41_derivative_target_slotid(new_target, |
| tbl->target_highest_slotid); |
| d2_target = nfs41_derivative_target_slotid(d_target, |
| tbl->d_target_highest_slotid); |
| /* Is first derivative same sign? */ |
| if (nfs41_same_sign_or_zero_s32(d_target, tbl->d_target_highest_slotid)) |
| ret = false; |
| /* Is second derivative same sign? */ |
| if (nfs41_same_sign_or_zero_s32(d2_target, tbl->d2_target_highest_slotid)) |
| ret = false; |
| tbl->d_target_highest_slotid = d_target; |
| tbl->d2_target_highest_slotid = d2_target; |
| return ret; |
| } |
| |
| void nfs41_update_target_slotid(struct nfs4_slot_table *tbl, |
| struct nfs4_slot *slot, |
| struct nfs4_sequence_res *res) |
| { |
| spin_lock(&tbl->slot_tbl_lock); |
| if (!nfs41_is_outlier_target_slotid(tbl, res->sr_target_highest_slotid)) |
| nfs41_set_target_slotid_locked(tbl, res->sr_target_highest_slotid); |
| if (tbl->generation == slot->generation) |
| nfs41_set_server_slotid_locked(tbl, res->sr_highest_slotid); |
| nfs41_set_max_slotid_locked(tbl, res->sr_target_highest_slotid); |
| spin_unlock(&tbl->slot_tbl_lock); |
| } |
| |
| static void nfs4_release_session_slot_tables(struct nfs4_session *session) |
| { |
| nfs4_release_slot_table(&session->fc_slot_table); |
| nfs4_release_slot_table(&session->bc_slot_table); |
| } |
| |
| /* |
| * Initialize or reset the forechannel and backchannel tables |
| */ |
| int nfs4_setup_session_slot_tables(struct nfs4_session *ses) |
| { |
| struct nfs4_slot_table *tbl; |
| int status; |
| |
| dprintk("--> %s\n", __func__); |
| /* Fore channel */ |
| tbl = &ses->fc_slot_table; |
| tbl->session = ses; |
| status = nfs4_realloc_slot_table(tbl, ses->fc_attrs.max_reqs, 1); |
| if (status || !(ses->flags & SESSION4_BACK_CHAN)) /* -ENOMEM */ |
| return status; |
| /* Back channel */ |
| tbl = &ses->bc_slot_table; |
| tbl->session = ses; |
| status = nfs4_realloc_slot_table(tbl, ses->bc_attrs.max_reqs, 0); |
| if (status && tbl->slots == NULL) |
| /* Fore and back channel share a connection so get |
| * both slot tables or neither */ |
| nfs4_release_session_slot_tables(ses); |
| return status; |
| } |
| |
| struct nfs4_session *nfs4_alloc_session(struct nfs_client *clp) |
| { |
| struct nfs4_session *session; |
| |
| session = kzalloc(sizeof(struct nfs4_session), GFP_NOFS); |
| if (!session) |
| return NULL; |
| |
| nfs4_init_slot_table(&session->fc_slot_table, "ForeChannel Slot table"); |
| nfs4_init_slot_table(&session->bc_slot_table, "BackChannel Slot table"); |
| session->session_state = 1<<NFS4_SESSION_INITING; |
| |
| session->clp = clp; |
| return session; |
| } |
| |
| static void nfs4_destroy_session_slot_tables(struct nfs4_session *session) |
| { |
| nfs4_shutdown_slot_table(&session->fc_slot_table); |
| nfs4_shutdown_slot_table(&session->bc_slot_table); |
| } |
| |
| void nfs4_destroy_session(struct nfs4_session *session) |
| { |
| struct rpc_xprt *xprt; |
| const struct cred *cred; |
| |
| cred = nfs4_get_clid_cred(session->clp); |
| nfs4_proc_destroy_session(session, cred); |
| put_cred(cred); |
| |
| rcu_read_lock(); |
| xprt = rcu_dereference(session->clp->cl_rpcclient->cl_xprt); |
| rcu_read_unlock(); |
| dprintk("%s Destroy backchannel for xprt %p\n", |
| __func__, xprt); |
| xprt_destroy_backchannel(xprt, NFS41_BC_MIN_CALLBACKS); |
| nfs4_destroy_session_slot_tables(session); |
| kfree(session); |
| } |
| |
| /* |
| * With sessions, the client is not marked ready until after a |
| * successful EXCHANGE_ID and CREATE_SESSION. |
| * |
| * Map errors cl_cons_state errors to EPROTONOSUPPORT to indicate |
| * other versions of NFS can be tried. |
| */ |
| static int nfs41_check_session_ready(struct nfs_client *clp) |
| { |
| int ret; |
| |
| if (clp->cl_cons_state == NFS_CS_SESSION_INITING) { |
| ret = nfs4_client_recover_expired_lease(clp); |
| if (ret) |
| return ret; |
| } |
| if (clp->cl_cons_state < NFS_CS_READY) |
| return -EPROTONOSUPPORT; |
| smp_rmb(); |
| return 0; |
| } |
| |
| int nfs4_init_session(struct nfs_client *clp) |
| { |
| if (!nfs4_has_session(clp)) |
| return 0; |
| |
| clear_bit(NFS4_SESSION_INITING, &clp->cl_session->session_state); |
| return nfs41_check_session_ready(clp); |
| } |
| |
| int nfs4_init_ds_session(struct nfs_client *clp, unsigned long lease_time) |
| { |
| struct nfs4_session *session = clp->cl_session; |
| int ret; |
| |
| spin_lock(&clp->cl_lock); |
| if (test_and_clear_bit(NFS4_SESSION_INITING, &session->session_state)) { |
| /* |
| * Do not set NFS_CS_CHECK_LEASE_TIME instead set the |
| * DS lease to be equal to the MDS lease. |
| */ |
| clp->cl_lease_time = lease_time; |
| clp->cl_last_renewal = jiffies; |
| } |
| spin_unlock(&clp->cl_lock); |
| |
| ret = nfs41_check_session_ready(clp); |
| if (ret) |
| return ret; |
| /* Test for the DS role */ |
| if (!is_ds_client(clp)) |
| return -ENODEV; |
| return 0; |
| } |
| EXPORT_SYMBOL_GPL(nfs4_init_ds_session); |
| |
| #endif /* defined(CONFIG_NFS_V4_1) */ |