11da177e4SLinus Torvalds /* 21da177e4SLinus Torvalds * linux/net/sunrpc/auth.c 31da177e4SLinus Torvalds * 41da177e4SLinus Torvalds * Generic RPC client authentication API. 51da177e4SLinus Torvalds * 61da177e4SLinus Torvalds * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de> 71da177e4SLinus Torvalds */ 81da177e4SLinus Torvalds 91da177e4SLinus Torvalds #include <linux/types.h> 101da177e4SLinus Torvalds #include <linux/sched.h> 111da177e4SLinus Torvalds #include <linux/module.h> 121da177e4SLinus Torvalds #include <linux/slab.h> 131da177e4SLinus Torvalds #include <linux/errno.h> 1425337fdcSTrond Myklebust #include <linux/hash.h> 151da177e4SLinus Torvalds #include <linux/sunrpc/clnt.h> 161da177e4SLinus Torvalds #include <linux/spinlock.h> 171da177e4SLinus Torvalds 181da177e4SLinus Torvalds #ifdef RPC_DEBUG 191da177e4SLinus Torvalds # define RPCDBG_FACILITY RPCDBG_AUTH 201da177e4SLinus Torvalds #endif 211da177e4SLinus Torvalds 22fc1b356fSTrond Myklebust static DEFINE_SPINLOCK(rpc_authflavor_lock); 23f1c0a861STrond Myklebust static const struct rpc_authops *auth_flavors[RPC_AUTH_MAXFLAVOR] = { 241da177e4SLinus Torvalds &authnull_ops, /* AUTH_NULL */ 251da177e4SLinus Torvalds &authunix_ops, /* AUTH_UNIX */ 261da177e4SLinus Torvalds NULL, /* others can be loadable modules */ 271da177e4SLinus Torvalds }; 281da177e4SLinus Torvalds 29e092bdcdSTrond Myklebust static LIST_HEAD(cred_unused); 30f5c2187cSTrond Myklebust static unsigned long number_cred_unused; 31e092bdcdSTrond Myklebust 321da177e4SLinus Torvalds static u32 331da177e4SLinus Torvalds pseudoflavor_to_flavor(u32 flavor) { 341da177e4SLinus Torvalds if (flavor >= RPC_AUTH_MAXFLAVOR) 351da177e4SLinus Torvalds return RPC_AUTH_GSS; 361da177e4SLinus Torvalds return flavor; 371da177e4SLinus Torvalds } 381da177e4SLinus Torvalds 391da177e4SLinus Torvalds int 40f1c0a861STrond Myklebust rpcauth_register(const struct rpc_authops *ops) 411da177e4SLinus Torvalds { 421da177e4SLinus Torvalds rpc_authflavor_t flavor; 43fc1b356fSTrond Myklebust int ret = -EPERM; 441da177e4SLinus Torvalds 451da177e4SLinus Torvalds if ((flavor = ops->au_flavor) >= RPC_AUTH_MAXFLAVOR) 461da177e4SLinus Torvalds return -EINVAL; 47fc1b356fSTrond Myklebust spin_lock(&rpc_authflavor_lock); 48fc1b356fSTrond Myklebust if (auth_flavors[flavor] == NULL) { 491da177e4SLinus Torvalds auth_flavors[flavor] = ops; 50fc1b356fSTrond Myklebust ret = 0; 51fc1b356fSTrond Myklebust } 52fc1b356fSTrond Myklebust spin_unlock(&rpc_authflavor_lock); 53fc1b356fSTrond Myklebust return ret; 541da177e4SLinus Torvalds } 55e8914c65STrond Myklebust EXPORT_SYMBOL_GPL(rpcauth_register); 561da177e4SLinus Torvalds 571da177e4SLinus Torvalds int 58f1c0a861STrond Myklebust rpcauth_unregister(const struct rpc_authops *ops) 591da177e4SLinus Torvalds { 601da177e4SLinus Torvalds rpc_authflavor_t flavor; 61fc1b356fSTrond Myklebust int ret = -EPERM; 621da177e4SLinus Torvalds 631da177e4SLinus Torvalds if ((flavor = ops->au_flavor) >= RPC_AUTH_MAXFLAVOR) 641da177e4SLinus Torvalds return -EINVAL; 65fc1b356fSTrond Myklebust spin_lock(&rpc_authflavor_lock); 66fc1b356fSTrond Myklebust if (auth_flavors[flavor] == ops) { 671da177e4SLinus Torvalds auth_flavors[flavor] = NULL; 68fc1b356fSTrond Myklebust ret = 0; 69fc1b356fSTrond Myklebust } 70fc1b356fSTrond Myklebust spin_unlock(&rpc_authflavor_lock); 71fc1b356fSTrond Myklebust return ret; 721da177e4SLinus Torvalds } 73e8914c65STrond Myklebust EXPORT_SYMBOL_GPL(rpcauth_unregister); 741da177e4SLinus Torvalds 751da177e4SLinus Torvalds struct rpc_auth * 761da177e4SLinus Torvalds rpcauth_create(rpc_authflavor_t pseudoflavor, struct rpc_clnt *clnt) 771da177e4SLinus Torvalds { 781da177e4SLinus Torvalds struct rpc_auth *auth; 79f1c0a861STrond Myklebust const struct rpc_authops *ops; 801da177e4SLinus Torvalds u32 flavor = pseudoflavor_to_flavor(pseudoflavor); 811da177e4SLinus Torvalds 82f344f6dfSOlaf Kirch auth = ERR_PTR(-EINVAL); 83f344f6dfSOlaf Kirch if (flavor >= RPC_AUTH_MAXFLAVOR) 84f344f6dfSOlaf Kirch goto out; 85f344f6dfSOlaf Kirch 86f344f6dfSOlaf Kirch #ifdef CONFIG_KMOD 87f344f6dfSOlaf Kirch if ((ops = auth_flavors[flavor]) == NULL) 88f344f6dfSOlaf Kirch request_module("rpc-auth-%u", flavor); 89f344f6dfSOlaf Kirch #endif 90fc1b356fSTrond Myklebust spin_lock(&rpc_authflavor_lock); 91fc1b356fSTrond Myklebust ops = auth_flavors[flavor]; 92fc1b356fSTrond Myklebust if (ops == NULL || !try_module_get(ops->owner)) { 93fc1b356fSTrond Myklebust spin_unlock(&rpc_authflavor_lock); 94f344f6dfSOlaf Kirch goto out; 95fc1b356fSTrond Myklebust } 96fc1b356fSTrond Myklebust spin_unlock(&rpc_authflavor_lock); 971da177e4SLinus Torvalds auth = ops->create(clnt, pseudoflavor); 98fc1b356fSTrond Myklebust module_put(ops->owner); 996a19275aSJ. Bruce Fields if (IS_ERR(auth)) 1006a19275aSJ. Bruce Fields return auth; 1011da177e4SLinus Torvalds if (clnt->cl_auth) 102de7a8ce3STrond Myklebust rpcauth_release(clnt->cl_auth); 1031da177e4SLinus Torvalds clnt->cl_auth = auth; 104f344f6dfSOlaf Kirch 105f344f6dfSOlaf Kirch out: 1061da177e4SLinus Torvalds return auth; 1071da177e4SLinus Torvalds } 108e8914c65STrond Myklebust EXPORT_SYMBOL_GPL(rpcauth_create); 1091da177e4SLinus Torvalds 1101da177e4SLinus Torvalds void 111de7a8ce3STrond Myklebust rpcauth_release(struct rpc_auth *auth) 1121da177e4SLinus Torvalds { 1131da177e4SLinus Torvalds if (!atomic_dec_and_test(&auth->au_count)) 1141da177e4SLinus Torvalds return; 1151da177e4SLinus Torvalds auth->au_ops->destroy(auth); 1161da177e4SLinus Torvalds } 1171da177e4SLinus Torvalds 1181da177e4SLinus Torvalds static DEFINE_SPINLOCK(rpc_credcache_lock); 1191da177e4SLinus Torvalds 12031be5bf1STrond Myklebust static void 12131be5bf1STrond Myklebust rpcauth_unhash_cred_locked(struct rpc_cred *cred) 12231be5bf1STrond Myklebust { 12331be5bf1STrond Myklebust hlist_del_rcu(&cred->cr_hash); 12431be5bf1STrond Myklebust smp_mb__before_clear_bit(); 12531be5bf1STrond Myklebust clear_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags); 12631be5bf1STrond Myklebust } 12731be5bf1STrond Myklebust 1289499b434STrond Myklebust static void 1299499b434STrond Myklebust rpcauth_unhash_cred(struct rpc_cred *cred) 1309499b434STrond Myklebust { 1319499b434STrond Myklebust spinlock_t *cache_lock; 1329499b434STrond Myklebust 1339499b434STrond Myklebust cache_lock = &cred->cr_auth->au_credcache->lock; 1349499b434STrond Myklebust spin_lock(cache_lock); 1359499b434STrond Myklebust if (atomic_read(&cred->cr_count) == 0) 1369499b434STrond Myklebust rpcauth_unhash_cred_locked(cred); 1379499b434STrond Myklebust spin_unlock(cache_lock); 1389499b434STrond Myklebust } 1399499b434STrond Myklebust 1401da177e4SLinus Torvalds /* 1411da177e4SLinus Torvalds * Initialize RPC credential cache 1421da177e4SLinus Torvalds */ 1431da177e4SLinus Torvalds int 144f5c2187cSTrond Myklebust rpcauth_init_credcache(struct rpc_auth *auth) 1451da177e4SLinus Torvalds { 1461da177e4SLinus Torvalds struct rpc_cred_cache *new; 1471da177e4SLinus Torvalds int i; 1481da177e4SLinus Torvalds 1498b3a7005SKris Katterjohn new = kmalloc(sizeof(*new), GFP_KERNEL); 1501da177e4SLinus Torvalds if (!new) 1511da177e4SLinus Torvalds return -ENOMEM; 1521da177e4SLinus Torvalds for (i = 0; i < RPC_CREDCACHE_NR; i++) 1531da177e4SLinus Torvalds INIT_HLIST_HEAD(&new->hashtable[i]); 1549499b434STrond Myklebust spin_lock_init(&new->lock); 1551da177e4SLinus Torvalds auth->au_credcache = new; 1561da177e4SLinus Torvalds return 0; 1571da177e4SLinus Torvalds } 158e8914c65STrond Myklebust EXPORT_SYMBOL_GPL(rpcauth_init_credcache); 1591da177e4SLinus Torvalds 1601da177e4SLinus Torvalds /* 1611da177e4SLinus Torvalds * Destroy a list of credentials 1621da177e4SLinus Torvalds */ 1631da177e4SLinus Torvalds static inline 164e092bdcdSTrond Myklebust void rpcauth_destroy_credlist(struct list_head *head) 1651da177e4SLinus Torvalds { 1661da177e4SLinus Torvalds struct rpc_cred *cred; 1671da177e4SLinus Torvalds 168e092bdcdSTrond Myklebust while (!list_empty(head)) { 169e092bdcdSTrond Myklebust cred = list_entry(head->next, struct rpc_cred, cr_lru); 170e092bdcdSTrond Myklebust list_del_init(&cred->cr_lru); 1711da177e4SLinus Torvalds put_rpccred(cred); 1721da177e4SLinus Torvalds } 1731da177e4SLinus Torvalds } 1741da177e4SLinus Torvalds 1751da177e4SLinus Torvalds /* 1761da177e4SLinus Torvalds * Clear the RPC credential cache, and delete those credentials 1771da177e4SLinus Torvalds * that are not referenced. 1781da177e4SLinus Torvalds */ 1791da177e4SLinus Torvalds void 1803ab9bb72STrond Myklebust rpcauth_clear_credcache(struct rpc_cred_cache *cache) 1811da177e4SLinus Torvalds { 182e092bdcdSTrond Myklebust LIST_HEAD(free); 183e092bdcdSTrond Myklebust struct hlist_head *head; 1841da177e4SLinus Torvalds struct rpc_cred *cred; 1851da177e4SLinus Torvalds int i; 1861da177e4SLinus Torvalds 1871da177e4SLinus Torvalds spin_lock(&rpc_credcache_lock); 1889499b434STrond Myklebust spin_lock(&cache->lock); 1891da177e4SLinus Torvalds for (i = 0; i < RPC_CREDCACHE_NR; i++) { 190e092bdcdSTrond Myklebust head = &cache->hashtable[i]; 191e092bdcdSTrond Myklebust while (!hlist_empty(head)) { 192e092bdcdSTrond Myklebust cred = hlist_entry(head->first, struct rpc_cred, cr_hash); 193e092bdcdSTrond Myklebust get_rpccred(cred); 194f5c2187cSTrond Myklebust if (!list_empty(&cred->cr_lru)) { 195f5c2187cSTrond Myklebust list_del(&cred->cr_lru); 196f5c2187cSTrond Myklebust number_cred_unused--; 197f5c2187cSTrond Myklebust } 198f5c2187cSTrond Myklebust list_add_tail(&cred->cr_lru, &free); 19931be5bf1STrond Myklebust rpcauth_unhash_cred_locked(cred); 2001da177e4SLinus Torvalds } 2011da177e4SLinus Torvalds } 2029499b434STrond Myklebust spin_unlock(&cache->lock); 2031da177e4SLinus Torvalds spin_unlock(&rpc_credcache_lock); 2041da177e4SLinus Torvalds rpcauth_destroy_credlist(&free); 2051da177e4SLinus Torvalds } 2061da177e4SLinus Torvalds 2073ab9bb72STrond Myklebust /* 2083ab9bb72STrond Myklebust * Destroy the RPC credential cache 2093ab9bb72STrond Myklebust */ 2103ab9bb72STrond Myklebust void 2113ab9bb72STrond Myklebust rpcauth_destroy_credcache(struct rpc_auth *auth) 2123ab9bb72STrond Myklebust { 2133ab9bb72STrond Myklebust struct rpc_cred_cache *cache = auth->au_credcache; 2143ab9bb72STrond Myklebust 2153ab9bb72STrond Myklebust if (cache) { 2163ab9bb72STrond Myklebust auth->au_credcache = NULL; 2173ab9bb72STrond Myklebust rpcauth_clear_credcache(cache); 2183ab9bb72STrond Myklebust kfree(cache); 2193ab9bb72STrond Myklebust } 2203ab9bb72STrond Myklebust } 221e8914c65STrond Myklebust EXPORT_SYMBOL_GPL(rpcauth_destroy_credcache); 2223ab9bb72STrond Myklebust 2231da177e4SLinus Torvalds /* 2241da177e4SLinus Torvalds * Remove stale credentials. Avoid sleeping inside the loop. 2251da177e4SLinus Torvalds */ 226f5c2187cSTrond Myklebust static int 227f5c2187cSTrond Myklebust rpcauth_prune_expired(struct list_head *free, int nr_to_scan) 2281da177e4SLinus Torvalds { 2299499b434STrond Myklebust spinlock_t *cache_lock; 2301da177e4SLinus Torvalds struct rpc_cred *cred; 2311da177e4SLinus Torvalds 232e092bdcdSTrond Myklebust while (!list_empty(&cred_unused)) { 233e092bdcdSTrond Myklebust cred = list_entry(cred_unused.next, struct rpc_cred, cr_lru); 234e092bdcdSTrond Myklebust list_del_init(&cred->cr_lru); 235f5c2187cSTrond Myklebust number_cred_unused--; 236e092bdcdSTrond Myklebust if (atomic_read(&cred->cr_count) != 0) 237e092bdcdSTrond Myklebust continue; 2389499b434STrond Myklebust cache_lock = &cred->cr_auth->au_credcache->lock; 2399499b434STrond Myklebust spin_lock(cache_lock); 2409499b434STrond Myklebust if (atomic_read(&cred->cr_count) == 0) { 241e092bdcdSTrond Myklebust get_rpccred(cred); 242e092bdcdSTrond Myklebust list_add_tail(&cred->cr_lru, free); 24331be5bf1STrond Myklebust rpcauth_unhash_cred_locked(cred); 244f5c2187cSTrond Myklebust nr_to_scan--; 2451da177e4SLinus Torvalds } 2469499b434STrond Myklebust spin_unlock(cache_lock); 247f5c2187cSTrond Myklebust if (nr_to_scan == 0) 248f5c2187cSTrond Myklebust break; 2499499b434STrond Myklebust } 250f5c2187cSTrond Myklebust return nr_to_scan; 2511da177e4SLinus Torvalds } 252e092bdcdSTrond Myklebust 253e092bdcdSTrond Myklebust /* 254f5c2187cSTrond Myklebust * Run memory cache shrinker. 255e092bdcdSTrond Myklebust */ 256f5c2187cSTrond Myklebust static int 257f5c2187cSTrond Myklebust rpcauth_cache_shrinker(int nr_to_scan, gfp_t gfp_mask) 258e092bdcdSTrond Myklebust { 259f5c2187cSTrond Myklebust LIST_HEAD(free); 260f5c2187cSTrond Myklebust int res; 261f5c2187cSTrond Myklebust 262f5c2187cSTrond Myklebust if (list_empty(&cred_unused)) 263f5c2187cSTrond Myklebust return 0; 26431be5bf1STrond Myklebust spin_lock(&rpc_credcache_lock); 265f5c2187cSTrond Myklebust nr_to_scan = rpcauth_prune_expired(&free, nr_to_scan); 266f5c2187cSTrond Myklebust res = (number_cred_unused / 100) * sysctl_vfs_cache_pressure; 26731be5bf1STrond Myklebust spin_unlock(&rpc_credcache_lock); 268f5c2187cSTrond Myklebust rpcauth_destroy_credlist(&free); 269f5c2187cSTrond Myklebust return res; 2701da177e4SLinus Torvalds } 2711da177e4SLinus Torvalds 2721da177e4SLinus Torvalds /* 2731da177e4SLinus Torvalds * Look up a process' credentials in the authentication cache 2741da177e4SLinus Torvalds */ 2751da177e4SLinus Torvalds struct rpc_cred * 2761da177e4SLinus Torvalds rpcauth_lookup_credcache(struct rpc_auth *auth, struct auth_cred * acred, 2778a317760STrond Myklebust int flags) 2781da177e4SLinus Torvalds { 279e092bdcdSTrond Myklebust LIST_HEAD(free); 2801da177e4SLinus Torvalds struct rpc_cred_cache *cache = auth->au_credcache; 281e092bdcdSTrond Myklebust struct hlist_node *pos; 28231be5bf1STrond Myklebust struct rpc_cred *cred = NULL, 28331be5bf1STrond Myklebust *entry, *new; 28425337fdcSTrond Myklebust unsigned int nr; 28525337fdcSTrond Myklebust 28625337fdcSTrond Myklebust nr = hash_long(acred->uid, RPC_CREDCACHE_HASHBITS); 2871da177e4SLinus Torvalds 28831be5bf1STrond Myklebust rcu_read_lock(); 28931be5bf1STrond Myklebust hlist_for_each_entry_rcu(entry, pos, &cache->hashtable[nr], cr_hash) { 29031be5bf1STrond Myklebust if (!entry->cr_ops->crmatch(acred, entry, flags)) 29131be5bf1STrond Myklebust continue; 2929499b434STrond Myklebust spin_lock(&cache->lock); 29331be5bf1STrond Myklebust if (test_bit(RPCAUTH_CRED_HASHED, &entry->cr_flags) == 0) { 2949499b434STrond Myklebust spin_unlock(&cache->lock); 29531be5bf1STrond Myklebust continue; 29631be5bf1STrond Myklebust } 29731be5bf1STrond Myklebust cred = get_rpccred(entry); 2989499b434STrond Myklebust spin_unlock(&cache->lock); 29931be5bf1STrond Myklebust break; 30031be5bf1STrond Myklebust } 30131be5bf1STrond Myklebust rcu_read_unlock(); 30231be5bf1STrond Myklebust 3039499b434STrond Myklebust if (cred != NULL) 30431be5bf1STrond Myklebust goto found; 30531be5bf1STrond Myklebust 30631be5bf1STrond Myklebust new = auth->au_ops->crcreate(auth, acred, flags); 30731be5bf1STrond Myklebust if (IS_ERR(new)) { 30831be5bf1STrond Myklebust cred = new; 30931be5bf1STrond Myklebust goto out; 31031be5bf1STrond Myklebust } 31131be5bf1STrond Myklebust 3129499b434STrond Myklebust spin_lock(&cache->lock); 313e092bdcdSTrond Myklebust hlist_for_each_entry(entry, pos, &cache->hashtable[nr], cr_hash) { 314e092bdcdSTrond Myklebust if (!entry->cr_ops->crmatch(acred, entry, flags)) 315e092bdcdSTrond Myklebust continue; 316e092bdcdSTrond Myklebust cred = get_rpccred(entry); 3171da177e4SLinus Torvalds break; 3181da177e4SLinus Torvalds } 31931be5bf1STrond Myklebust if (cred == NULL) { 32031be5bf1STrond Myklebust cred = new; 32131be5bf1STrond Myklebust set_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags); 32231be5bf1STrond Myklebust hlist_add_head_rcu(&cred->cr_hash, &cache->hashtable[nr]); 32331be5bf1STrond Myklebust } else 324e092bdcdSTrond Myklebust list_add_tail(&new->cr_lru, &free); 3259499b434STrond Myklebust spin_unlock(&cache->lock); 32631be5bf1STrond Myklebust found: 32731be5bf1STrond Myklebust if (test_bit(RPCAUTH_CRED_NEW, &cred->cr_flags) 328fba3bad4STrond Myklebust && cred->cr_ops->cr_init != NULL 329fba3bad4STrond Myklebust && !(flags & RPCAUTH_LOOKUP_NEW)) { 330fba3bad4STrond Myklebust int res = cred->cr_ops->cr_init(auth, cred); 331fba3bad4STrond Myklebust if (res < 0) { 332fba3bad4STrond Myklebust put_rpccred(cred); 333fba3bad4STrond Myklebust cred = ERR_PTR(res); 334fba3bad4STrond Myklebust } 3351da177e4SLinus Torvalds } 33631be5bf1STrond Myklebust rpcauth_destroy_credlist(&free); 33731be5bf1STrond Myklebust out: 33831be5bf1STrond Myklebust return cred; 3391da177e4SLinus Torvalds } 340e8914c65STrond Myklebust EXPORT_SYMBOL_GPL(rpcauth_lookup_credcache); 3411da177e4SLinus Torvalds 3421da177e4SLinus Torvalds struct rpc_cred * 3438a317760STrond Myklebust rpcauth_lookupcred(struct rpc_auth *auth, int flags) 3441da177e4SLinus Torvalds { 3451da177e4SLinus Torvalds struct auth_cred acred = { 3461da177e4SLinus Torvalds .uid = current->fsuid, 3471da177e4SLinus Torvalds .gid = current->fsgid, 3481da177e4SLinus Torvalds .group_info = current->group_info, 3491da177e4SLinus Torvalds }; 3501da177e4SLinus Torvalds struct rpc_cred *ret; 3511da177e4SLinus Torvalds 3521da177e4SLinus Torvalds dprintk("RPC: looking up %s cred\n", 3531da177e4SLinus Torvalds auth->au_ops->au_name); 3541da177e4SLinus Torvalds get_group_info(acred.group_info); 3558a317760STrond Myklebust ret = auth->au_ops->lookup_cred(auth, &acred, flags); 3561da177e4SLinus Torvalds put_group_info(acred.group_info); 3571da177e4SLinus Torvalds return ret; 3581da177e4SLinus Torvalds } 359e8914c65STrond Myklebust EXPORT_SYMBOL_GPL(rpcauth_lookupcred); 3601da177e4SLinus Torvalds 3615fe4755eSTrond Myklebust void 3625fe4755eSTrond Myklebust rpcauth_init_cred(struct rpc_cred *cred, const struct auth_cred *acred, 3635fe4755eSTrond Myklebust struct rpc_auth *auth, const struct rpc_credops *ops) 3645fe4755eSTrond Myklebust { 3655fe4755eSTrond Myklebust INIT_HLIST_NODE(&cred->cr_hash); 366e092bdcdSTrond Myklebust INIT_LIST_HEAD(&cred->cr_lru); 3675fe4755eSTrond Myklebust atomic_set(&cred->cr_count, 1); 3685fe4755eSTrond Myklebust cred->cr_auth = auth; 3695fe4755eSTrond Myklebust cred->cr_ops = ops; 3705fe4755eSTrond Myklebust cred->cr_expire = jiffies; 3715fe4755eSTrond Myklebust #ifdef RPC_DEBUG 3725fe4755eSTrond Myklebust cred->cr_magic = RPCAUTH_CRED_MAGIC; 3735fe4755eSTrond Myklebust #endif 3745fe4755eSTrond Myklebust cred->cr_uid = acred->uid; 3755fe4755eSTrond Myklebust } 376e8914c65STrond Myklebust EXPORT_SYMBOL_GPL(rpcauth_init_cred); 3775fe4755eSTrond Myklebust 3785c691044STrond Myklebust void 3794ccda2cdSTrond Myklebust rpcauth_generic_bind_cred(struct rpc_task *task, struct rpc_cred *cred) 3804ccda2cdSTrond Myklebust { 3814ccda2cdSTrond Myklebust task->tk_msg.rpc_cred = get_rpccred(cred); 3824ccda2cdSTrond Myklebust dprintk("RPC: %5u holding %s cred %p\n", task->tk_pid, 3834ccda2cdSTrond Myklebust cred->cr_auth->au_ops->au_name, cred); 3844ccda2cdSTrond Myklebust } 3855c691044STrond Myklebust EXPORT_SYMBOL_GPL(rpcauth_generic_bind_cred); 3864ccda2cdSTrond Myklebust 3874ccda2cdSTrond Myklebust static void 388af093835STrond Myklebust rpcauth_bind_root_cred(struct rpc_task *task) 3891da177e4SLinus Torvalds { 3901be27f36STrond Myklebust struct rpc_auth *auth = task->tk_client->cl_auth; 3911da177e4SLinus Torvalds struct auth_cred acred = { 392af093835STrond Myklebust .uid = 0, 393af093835STrond Myklebust .gid = 0, 3941da177e4SLinus Torvalds }; 3951da177e4SLinus Torvalds struct rpc_cred *ret; 3961da177e4SLinus Torvalds 39746121cf7SChuck Lever dprintk("RPC: %5u looking up %s cred\n", 3981be27f36STrond Myklebust task->tk_pid, task->tk_client->cl_auth->au_ops->au_name); 399af093835STrond Myklebust ret = auth->au_ops->lookup_cred(auth, &acred, 0); 4001da177e4SLinus Torvalds if (!IS_ERR(ret)) 4011da177e4SLinus Torvalds task->tk_msg.rpc_cred = ret; 4021da177e4SLinus Torvalds else 4031da177e4SLinus Torvalds task->tk_status = PTR_ERR(ret); 404af093835STrond Myklebust } 405af093835STrond Myklebust 4064ccda2cdSTrond Myklebust static void 4074ccda2cdSTrond Myklebust rpcauth_bind_new_cred(struct rpc_task *task) 408af093835STrond Myklebust { 409af093835STrond Myklebust struct rpc_auth *auth = task->tk_client->cl_auth; 410af093835STrond Myklebust struct rpc_cred *ret; 411af093835STrond Myklebust 412af093835STrond Myklebust dprintk("RPC: %5u looking up %s cred\n", 413af093835STrond Myklebust task->tk_pid, auth->au_ops->au_name); 414af093835STrond Myklebust ret = rpcauth_lookupcred(auth, 0); 415af093835STrond Myklebust if (!IS_ERR(ret)) 416af093835STrond Myklebust task->tk_msg.rpc_cred = ret; 417af093835STrond Myklebust else 418af093835STrond Myklebust task->tk_status = PTR_ERR(ret); 4191da177e4SLinus Torvalds } 4201da177e4SLinus Torvalds 4211da177e4SLinus Torvalds void 4224ccda2cdSTrond Myklebust rpcauth_bindcred(struct rpc_task *task, struct rpc_cred *cred, int flags) 4231da177e4SLinus Torvalds { 4244ccda2cdSTrond Myklebust if (cred != NULL) 4255c691044STrond Myklebust cred->cr_ops->crbind(task, cred); 4264ccda2cdSTrond Myklebust else if (flags & RPC_TASK_ROOTCREDS) 4274ccda2cdSTrond Myklebust rpcauth_bind_root_cred(task); 4284ccda2cdSTrond Myklebust else 4294ccda2cdSTrond Myklebust rpcauth_bind_new_cred(task); 4301da177e4SLinus Torvalds } 4311da177e4SLinus Torvalds 4321da177e4SLinus Torvalds void 4331da177e4SLinus Torvalds put_rpccred(struct rpc_cred *cred) 4341da177e4SLinus Torvalds { 435e092bdcdSTrond Myklebust /* Fast path for unhashed credentials */ 43631be5bf1STrond Myklebust if (test_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags) != 0) 437e092bdcdSTrond Myklebust goto need_lock; 438e092bdcdSTrond Myklebust 4391da177e4SLinus Torvalds if (!atomic_dec_and_test(&cred->cr_count)) 4401da177e4SLinus Torvalds return; 441e092bdcdSTrond Myklebust goto out_destroy; 442e092bdcdSTrond Myklebust need_lock: 443e092bdcdSTrond Myklebust if (!atomic_dec_and_lock(&cred->cr_count, &rpc_credcache_lock)) 444e092bdcdSTrond Myklebust return; 445f5c2187cSTrond Myklebust if (!list_empty(&cred->cr_lru)) { 446f5c2187cSTrond Myklebust number_cred_unused--; 447e092bdcdSTrond Myklebust list_del_init(&cred->cr_lru); 448f5c2187cSTrond Myklebust } 449e092bdcdSTrond Myklebust if (test_bit(RPCAUTH_CRED_UPTODATE, &cred->cr_flags) == 0) 4509499b434STrond Myklebust rpcauth_unhash_cred(cred); 45131be5bf1STrond Myklebust else if (test_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags) != 0) { 452e092bdcdSTrond Myklebust cred->cr_expire = jiffies; 453e092bdcdSTrond Myklebust list_add_tail(&cred->cr_lru, &cred_unused); 454f5c2187cSTrond Myklebust number_cred_unused++; 455e092bdcdSTrond Myklebust spin_unlock(&rpc_credcache_lock); 456e092bdcdSTrond Myklebust return; 457e092bdcdSTrond Myklebust } 458e092bdcdSTrond Myklebust spin_unlock(&rpc_credcache_lock); 459e092bdcdSTrond Myklebust out_destroy: 4601da177e4SLinus Torvalds cred->cr_ops->crdestroy(cred); 4611da177e4SLinus Torvalds } 462e8914c65STrond Myklebust EXPORT_SYMBOL_GPL(put_rpccred); 4631da177e4SLinus Torvalds 4641da177e4SLinus Torvalds void 4651da177e4SLinus Torvalds rpcauth_unbindcred(struct rpc_task *task) 4661da177e4SLinus Torvalds { 4671da177e4SLinus Torvalds struct rpc_cred *cred = task->tk_msg.rpc_cred; 4681da177e4SLinus Torvalds 46946121cf7SChuck Lever dprintk("RPC: %5u releasing %s cred %p\n", 4701be27f36STrond Myklebust task->tk_pid, cred->cr_auth->au_ops->au_name, cred); 4711da177e4SLinus Torvalds 4721da177e4SLinus Torvalds put_rpccred(cred); 4731da177e4SLinus Torvalds task->tk_msg.rpc_cred = NULL; 4741da177e4SLinus Torvalds } 4751da177e4SLinus Torvalds 476d8ed029dSAlexey Dobriyan __be32 * 477d8ed029dSAlexey Dobriyan rpcauth_marshcred(struct rpc_task *task, __be32 *p) 4781da177e4SLinus Torvalds { 4791da177e4SLinus Torvalds struct rpc_cred *cred = task->tk_msg.rpc_cred; 4801da177e4SLinus Torvalds 48146121cf7SChuck Lever dprintk("RPC: %5u marshaling %s cred %p\n", 4821be27f36STrond Myklebust task->tk_pid, cred->cr_auth->au_ops->au_name, cred); 4830bbacc40SChuck Lever 4841da177e4SLinus Torvalds return cred->cr_ops->crmarshal(task, p); 4851da177e4SLinus Torvalds } 4861da177e4SLinus Torvalds 487d8ed029dSAlexey Dobriyan __be32 * 488d8ed029dSAlexey Dobriyan rpcauth_checkverf(struct rpc_task *task, __be32 *p) 4891da177e4SLinus Torvalds { 4901da177e4SLinus Torvalds struct rpc_cred *cred = task->tk_msg.rpc_cred; 4911da177e4SLinus Torvalds 49246121cf7SChuck Lever dprintk("RPC: %5u validating %s cred %p\n", 4931be27f36STrond Myklebust task->tk_pid, cred->cr_auth->au_ops->au_name, cred); 4940bbacc40SChuck Lever 4951da177e4SLinus Torvalds return cred->cr_ops->crvalidate(task, p); 4961da177e4SLinus Torvalds } 4971da177e4SLinus Torvalds 4981da177e4SLinus Torvalds int 4991da177e4SLinus Torvalds rpcauth_wrap_req(struct rpc_task *task, kxdrproc_t encode, void *rqstp, 500d8ed029dSAlexey Dobriyan __be32 *data, void *obj) 5011da177e4SLinus Torvalds { 5021da177e4SLinus Torvalds struct rpc_cred *cred = task->tk_msg.rpc_cred; 5031da177e4SLinus Torvalds 50446121cf7SChuck Lever dprintk("RPC: %5u using %s cred %p to wrap rpc data\n", 5051da177e4SLinus Torvalds task->tk_pid, cred->cr_ops->cr_name, cred); 5061da177e4SLinus Torvalds if (cred->cr_ops->crwrap_req) 5071da177e4SLinus Torvalds return cred->cr_ops->crwrap_req(task, encode, rqstp, data, obj); 5081da177e4SLinus Torvalds /* By default, we encode the arguments normally. */ 509be879c4eSJ. Bruce Fields return rpc_call_xdrproc(encode, rqstp, data, obj); 5101da177e4SLinus Torvalds } 5111da177e4SLinus Torvalds 5121da177e4SLinus Torvalds int 5131da177e4SLinus Torvalds rpcauth_unwrap_resp(struct rpc_task *task, kxdrproc_t decode, void *rqstp, 514d8ed029dSAlexey Dobriyan __be32 *data, void *obj) 5151da177e4SLinus Torvalds { 5161da177e4SLinus Torvalds struct rpc_cred *cred = task->tk_msg.rpc_cred; 5171da177e4SLinus Torvalds 51846121cf7SChuck Lever dprintk("RPC: %5u using %s cred %p to unwrap rpc data\n", 5191da177e4SLinus Torvalds task->tk_pid, cred->cr_ops->cr_name, cred); 5201da177e4SLinus Torvalds if (cred->cr_ops->crunwrap_resp) 5211da177e4SLinus Torvalds return cred->cr_ops->crunwrap_resp(task, decode, rqstp, 5221da177e4SLinus Torvalds data, obj); 5231da177e4SLinus Torvalds /* By default, we decode the arguments normally. */ 524be879c4eSJ. Bruce Fields return rpc_call_xdrproc(decode, rqstp, data, obj); 5251da177e4SLinus Torvalds } 5261da177e4SLinus Torvalds 5271da177e4SLinus Torvalds int 5281da177e4SLinus Torvalds rpcauth_refreshcred(struct rpc_task *task) 5291da177e4SLinus Torvalds { 5301da177e4SLinus Torvalds struct rpc_cred *cred = task->tk_msg.rpc_cred; 5311da177e4SLinus Torvalds int err; 5321da177e4SLinus Torvalds 53346121cf7SChuck Lever dprintk("RPC: %5u refreshing %s cred %p\n", 5341be27f36STrond Myklebust task->tk_pid, cred->cr_auth->au_ops->au_name, cred); 5350bbacc40SChuck Lever 5361da177e4SLinus Torvalds err = cred->cr_ops->crrefresh(task); 5371da177e4SLinus Torvalds if (err < 0) 5381da177e4SLinus Torvalds task->tk_status = err; 5391da177e4SLinus Torvalds return err; 5401da177e4SLinus Torvalds } 5411da177e4SLinus Torvalds 5421da177e4SLinus Torvalds void 5431da177e4SLinus Torvalds rpcauth_invalcred(struct rpc_task *task) 5441da177e4SLinus Torvalds { 545fc432dd9STrond Myklebust struct rpc_cred *cred = task->tk_msg.rpc_cred; 546fc432dd9STrond Myklebust 54746121cf7SChuck Lever dprintk("RPC: %5u invalidating %s cred %p\n", 5481be27f36STrond Myklebust task->tk_pid, cred->cr_auth->au_ops->au_name, cred); 549fc432dd9STrond Myklebust if (cred) 550fc432dd9STrond Myklebust clear_bit(RPCAUTH_CRED_UPTODATE, &cred->cr_flags); 5511da177e4SLinus Torvalds } 5521da177e4SLinus Torvalds 5531da177e4SLinus Torvalds int 5541da177e4SLinus Torvalds rpcauth_uptodatecred(struct rpc_task *task) 5551da177e4SLinus Torvalds { 556fc432dd9STrond Myklebust struct rpc_cred *cred = task->tk_msg.rpc_cred; 557fc432dd9STrond Myklebust 558fc432dd9STrond Myklebust return cred == NULL || 559fc432dd9STrond Myklebust test_bit(RPCAUTH_CRED_UPTODATE, &cred->cr_flags) != 0; 5601da177e4SLinus Torvalds } 561f5c2187cSTrond Myklebust 5628e1f936bSRusty Russell static struct shrinker rpc_cred_shrinker = { 5638e1f936bSRusty Russell .shrink = rpcauth_cache_shrinker, 5648e1f936bSRusty Russell .seeks = DEFAULT_SEEKS, 5658e1f936bSRusty Russell }; 566f5c2187cSTrond Myklebust 567f5c2187cSTrond Myklebust void __init rpcauth_init_module(void) 568f5c2187cSTrond Myklebust { 569f5c2187cSTrond Myklebust rpc_init_authunix(); 5709a559efdSTrond Myklebust rpc_init_generic_auth(); 5718e1f936bSRusty Russell register_shrinker(&rpc_cred_shrinker); 572f5c2187cSTrond Myklebust } 573f5c2187cSTrond Myklebust 574f5c2187cSTrond Myklebust void __exit rpcauth_remove_module(void) 575f5c2187cSTrond Myklebust { 5768e1f936bSRusty Russell unregister_shrinker(&rpc_cred_shrinker); 577f5c2187cSTrond Myklebust } 578