auth.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811
  1. /*
  2. * linux/net/sunrpc/auth.c
  3. *
  4. * Generic RPC client authentication API.
  5. *
  6. * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
  7. */
  8. #include <linux/types.h>
  9. #include <linux/sched.h>
  10. #include <linux/module.h>
  11. #include <linux/slab.h>
  12. #include <linux/errno.h>
  13. #include <linux/hash.h>
  14. #include <linux/sunrpc/clnt.h>
  15. #include <linux/sunrpc/gss_api.h>
  16. #include <linux/spinlock.h>
  17. #ifdef RPC_DEBUG
  18. # define RPCDBG_FACILITY RPCDBG_AUTH
  19. #endif
  20. #define RPC_CREDCACHE_DEFAULT_HASHBITS (4)
  21. struct rpc_cred_cache {
  22. struct hlist_head *hashtable;
  23. unsigned int hashbits;
  24. spinlock_t lock;
  25. };
  26. static unsigned int auth_hashbits = RPC_CREDCACHE_DEFAULT_HASHBITS;
  27. static DEFINE_SPINLOCK(rpc_authflavor_lock);
  28. static const struct rpc_authops *auth_flavors[RPC_AUTH_MAXFLAVOR] = {
  29. &authnull_ops, /* AUTH_NULL */
  30. &authunix_ops, /* AUTH_UNIX */
  31. NULL, /* others can be loadable modules */
  32. };
  33. static LIST_HEAD(cred_unused);
  34. static unsigned long number_cred_unused;
  35. #define MAX_HASHTABLE_BITS (14)
  36. static int param_set_hashtbl_sz(const char *val, const struct kernel_param *kp)
  37. {
  38. unsigned long num;
  39. unsigned int nbits;
  40. int ret;
  41. if (!val)
  42. goto out_inval;
  43. ret = strict_strtoul(val, 0, &num);
  44. if (ret == -EINVAL)
  45. goto out_inval;
  46. nbits = fls(num);
  47. if (num > (1U << nbits))
  48. nbits++;
  49. if (nbits > MAX_HASHTABLE_BITS || nbits < 2)
  50. goto out_inval;
  51. *(unsigned int *)kp->arg = nbits;
  52. return 0;
  53. out_inval:
  54. return -EINVAL;
  55. }
  56. static int param_get_hashtbl_sz(char *buffer, const struct kernel_param *kp)
  57. {
  58. unsigned int nbits;
  59. nbits = *(unsigned int *)kp->arg;
  60. return sprintf(buffer, "%u", 1U << nbits);
  61. }
  62. #define param_check_hashtbl_sz(name, p) __param_check(name, p, unsigned int);
  63. static struct kernel_param_ops param_ops_hashtbl_sz = {
  64. .set = param_set_hashtbl_sz,
  65. .get = param_get_hashtbl_sz,
  66. };
  67. module_param_named(auth_hashtable_size, auth_hashbits, hashtbl_sz, 0644);
  68. MODULE_PARM_DESC(auth_hashtable_size, "RPC credential cache hashtable size");
  69. static u32
  70. pseudoflavor_to_flavor(u32 flavor) {
  71. if (flavor >= RPC_AUTH_MAXFLAVOR)
  72. return RPC_AUTH_GSS;
  73. return flavor;
  74. }
  75. int
  76. rpcauth_register(const struct rpc_authops *ops)
  77. {
  78. rpc_authflavor_t flavor;
  79. int ret = -EPERM;
  80. if ((flavor = ops->au_flavor) >= RPC_AUTH_MAXFLAVOR)
  81. return -EINVAL;
  82. spin_lock(&rpc_authflavor_lock);
  83. if (auth_flavors[flavor] == NULL) {
  84. auth_flavors[flavor] = ops;
  85. ret = 0;
  86. }
  87. spin_unlock(&rpc_authflavor_lock);
  88. return ret;
  89. }
  90. EXPORT_SYMBOL_GPL(rpcauth_register);
  91. int
  92. rpcauth_unregister(const struct rpc_authops *ops)
  93. {
  94. rpc_authflavor_t flavor;
  95. int ret = -EPERM;
  96. if ((flavor = ops->au_flavor) >= RPC_AUTH_MAXFLAVOR)
  97. return -EINVAL;
  98. spin_lock(&rpc_authflavor_lock);
  99. if (auth_flavors[flavor] == ops) {
  100. auth_flavors[flavor] = NULL;
  101. ret = 0;
  102. }
  103. spin_unlock(&rpc_authflavor_lock);
  104. return ret;
  105. }
  106. EXPORT_SYMBOL_GPL(rpcauth_unregister);
  107. /**
  108. * rpcauth_get_pseudoflavor - check if security flavor is supported
  109. * @flavor: a security flavor
  110. * @info: a GSS mech OID, quality of protection, and service value
  111. *
  112. * Verifies that an appropriate kernel module is available or already loaded.
  113. * Returns an equivalent pseudoflavor, or RPC_AUTH_MAXFLAVOR if "flavor" is
  114. * not supported locally.
  115. */
  116. rpc_authflavor_t
  117. rpcauth_get_pseudoflavor(rpc_authflavor_t flavor, struct rpcsec_gss_info *info)
  118. {
  119. const struct rpc_authops *ops;
  120. rpc_authflavor_t pseudoflavor;
  121. ops = auth_flavors[flavor];
  122. if (ops == NULL)
  123. request_module("rpc-auth-%u", flavor);
  124. spin_lock(&rpc_authflavor_lock);
  125. ops = auth_flavors[flavor];
  126. if (ops == NULL || !try_module_get(ops->owner)) {
  127. spin_unlock(&rpc_authflavor_lock);
  128. return RPC_AUTH_MAXFLAVOR;
  129. }
  130. spin_unlock(&rpc_authflavor_lock);
  131. pseudoflavor = flavor;
  132. if (ops->info2flavor != NULL)
  133. pseudoflavor = ops->info2flavor(info);
  134. module_put(ops->owner);
  135. return pseudoflavor;
  136. }
  137. EXPORT_SYMBOL_GPL(rpcauth_get_pseudoflavor);
  138. /**
  139. * rpcauth_get_gssinfo - find GSS tuple matching a GSS pseudoflavor
  140. * @pseudoflavor: GSS pseudoflavor to match
  141. * @info: rpcsec_gss_info structure to fill in
  142. *
  143. * Returns zero and fills in "info" if pseudoflavor matches a
  144. * supported mechanism.
  145. */
  146. int
  147. rpcauth_get_gssinfo(rpc_authflavor_t pseudoflavor, struct rpcsec_gss_info *info)
  148. {
  149. rpc_authflavor_t flavor = pseudoflavor_to_flavor(pseudoflavor);
  150. const struct rpc_authops *ops;
  151. int result;
  152. ops = auth_flavors[flavor];
  153. if (ops == NULL)
  154. request_module("rpc-auth-%u", flavor);
  155. spin_lock(&rpc_authflavor_lock);
  156. ops = auth_flavors[flavor];
  157. if (ops == NULL || !try_module_get(ops->owner)) {
  158. spin_unlock(&rpc_authflavor_lock);
  159. return -ENOENT;
  160. }
  161. spin_unlock(&rpc_authflavor_lock);
  162. result = -ENOENT;
  163. if (ops->flavor2info != NULL)
  164. result = ops->flavor2info(pseudoflavor, info);
  165. module_put(ops->owner);
  166. return result;
  167. }
  168. EXPORT_SYMBOL_GPL(rpcauth_get_gssinfo);
  169. /**
  170. * rpcauth_list_flavors - discover registered flavors and pseudoflavors
  171. * @array: array to fill in
  172. * @size: size of "array"
  173. *
  174. * Returns the number of array items filled in, or a negative errno.
  175. *
  176. * The returned array is not sorted by any policy. Callers should not
  177. * rely on the order of the items in the returned array.
  178. */
  179. int
  180. rpcauth_list_flavors(rpc_authflavor_t *array, int size)
  181. {
  182. rpc_authflavor_t flavor;
  183. int result = 0;
  184. spin_lock(&rpc_authflavor_lock);
  185. for (flavor = 0; flavor < RPC_AUTH_MAXFLAVOR; flavor++) {
  186. const struct rpc_authops *ops = auth_flavors[flavor];
  187. rpc_authflavor_t pseudos[4];
  188. int i, len;
  189. if (result >= size) {
  190. result = -ENOMEM;
  191. break;
  192. }
  193. if (ops == NULL)
  194. continue;
  195. if (ops->list_pseudoflavors == NULL) {
  196. array[result++] = ops->au_flavor;
  197. continue;
  198. }
  199. len = ops->list_pseudoflavors(pseudos, ARRAY_SIZE(pseudos));
  200. if (len < 0) {
  201. result = len;
  202. break;
  203. }
  204. for (i = 0; i < len; i++) {
  205. if (result >= size) {
  206. result = -ENOMEM;
  207. break;
  208. }
  209. array[result++] = pseudos[i];
  210. }
  211. }
  212. spin_unlock(&rpc_authflavor_lock);
  213. dprintk("RPC: %s returns %d\n", __func__, result);
  214. return result;
  215. }
  216. EXPORT_SYMBOL_GPL(rpcauth_list_flavors);
  217. struct rpc_auth *
  218. rpcauth_create(rpc_authflavor_t pseudoflavor, struct rpc_clnt *clnt)
  219. {
  220. struct rpc_auth *auth;
  221. const struct rpc_authops *ops;
  222. u32 flavor = pseudoflavor_to_flavor(pseudoflavor);
  223. auth = ERR_PTR(-EINVAL);
  224. if (flavor >= RPC_AUTH_MAXFLAVOR)
  225. goto out;
  226. if ((ops = auth_flavors[flavor]) == NULL)
  227. request_module("rpc-auth-%u", flavor);
  228. spin_lock(&rpc_authflavor_lock);
  229. ops = auth_flavors[flavor];
  230. if (ops == NULL || !try_module_get(ops->owner)) {
  231. spin_unlock(&rpc_authflavor_lock);
  232. goto out;
  233. }
  234. spin_unlock(&rpc_authflavor_lock);
  235. auth = ops->create(clnt, pseudoflavor);
  236. module_put(ops->owner);
  237. if (IS_ERR(auth))
  238. return auth;
  239. if (clnt->cl_auth)
  240. rpcauth_release(clnt->cl_auth);
  241. clnt->cl_auth = auth;
  242. out:
  243. return auth;
  244. }
  245. EXPORT_SYMBOL_GPL(rpcauth_create);
  246. void
  247. rpcauth_release(struct rpc_auth *auth)
  248. {
  249. if (!atomic_dec_and_test(&auth->au_count))
  250. return;
  251. auth->au_ops->destroy(auth);
  252. }
  253. static DEFINE_SPINLOCK(rpc_credcache_lock);
  254. static void
  255. rpcauth_unhash_cred_locked(struct rpc_cred *cred)
  256. {
  257. hlist_del_rcu(&cred->cr_hash);
  258. smp_mb__before_clear_bit();
  259. clear_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags);
  260. }
  261. static int
  262. rpcauth_unhash_cred(struct rpc_cred *cred)
  263. {
  264. spinlock_t *cache_lock;
  265. int ret;
  266. cache_lock = &cred->cr_auth->au_credcache->lock;
  267. spin_lock(cache_lock);
  268. ret = atomic_read(&cred->cr_count) == 0;
  269. if (ret)
  270. rpcauth_unhash_cred_locked(cred);
  271. spin_unlock(cache_lock);
  272. return ret;
  273. }
  274. /*
  275. * Initialize RPC credential cache
  276. */
  277. int
  278. rpcauth_init_credcache(struct rpc_auth *auth)
  279. {
  280. struct rpc_cred_cache *new;
  281. unsigned int hashsize;
  282. new = kmalloc(sizeof(*new), GFP_KERNEL);
  283. if (!new)
  284. goto out_nocache;
  285. new->hashbits = auth_hashbits;
  286. hashsize = 1U << new->hashbits;
  287. new->hashtable = kcalloc(hashsize, sizeof(new->hashtable[0]), GFP_KERNEL);
  288. if (!new->hashtable)
  289. goto out_nohashtbl;
  290. spin_lock_init(&new->lock);
  291. auth->au_credcache = new;
  292. return 0;
  293. out_nohashtbl:
  294. kfree(new);
  295. out_nocache:
  296. return -ENOMEM;
  297. }
  298. EXPORT_SYMBOL_GPL(rpcauth_init_credcache);
  299. /*
  300. * Destroy a list of credentials
  301. */
  302. static inline
  303. void rpcauth_destroy_credlist(struct list_head *head)
  304. {
  305. struct rpc_cred *cred;
  306. while (!list_empty(head)) {
  307. cred = list_entry(head->next, struct rpc_cred, cr_lru);
  308. list_del_init(&cred->cr_lru);
  309. put_rpccred(cred);
  310. }
  311. }
  312. /*
  313. * Clear the RPC credential cache, and delete those credentials
  314. * that are not referenced.
  315. */
  316. void
  317. rpcauth_clear_credcache(struct rpc_cred_cache *cache)
  318. {
  319. LIST_HEAD(free);
  320. struct hlist_head *head;
  321. struct rpc_cred *cred;
  322. unsigned int hashsize = 1U << cache->hashbits;
  323. int i;
  324. spin_lock(&rpc_credcache_lock);
  325. spin_lock(&cache->lock);
  326. for (i = 0; i < hashsize; i++) {
  327. head = &cache->hashtable[i];
  328. while (!hlist_empty(head)) {
  329. cred = hlist_entry(head->first, struct rpc_cred, cr_hash);
  330. get_rpccred(cred);
  331. if (!list_empty(&cred->cr_lru)) {
  332. list_del(&cred->cr_lru);
  333. number_cred_unused--;
  334. }
  335. list_add_tail(&cred->cr_lru, &free);
  336. rpcauth_unhash_cred_locked(cred);
  337. }
  338. }
  339. spin_unlock(&cache->lock);
  340. spin_unlock(&rpc_credcache_lock);
  341. rpcauth_destroy_credlist(&free);
  342. }
  343. /*
  344. * Destroy the RPC credential cache
  345. */
  346. void
  347. rpcauth_destroy_credcache(struct rpc_auth *auth)
  348. {
  349. struct rpc_cred_cache *cache = auth->au_credcache;
  350. if (cache) {
  351. auth->au_credcache = NULL;
  352. rpcauth_clear_credcache(cache);
  353. kfree(cache->hashtable);
  354. kfree(cache);
  355. }
  356. }
  357. EXPORT_SYMBOL_GPL(rpcauth_destroy_credcache);
  358. #define RPC_AUTH_EXPIRY_MORATORIUM (60 * HZ)
  359. /*
  360. * Remove stale credentials. Avoid sleeping inside the loop.
  361. */
  362. static int
  363. rpcauth_prune_expired(struct list_head *free, int nr_to_scan)
  364. {
  365. spinlock_t *cache_lock;
  366. struct rpc_cred *cred, *next;
  367. unsigned long expired = jiffies - RPC_AUTH_EXPIRY_MORATORIUM;
  368. list_for_each_entry_safe(cred, next, &cred_unused, cr_lru) {
  369. if (nr_to_scan-- == 0)
  370. break;
  371. /*
  372. * Enforce a 60 second garbage collection moratorium
  373. * Note that the cred_unused list must be time-ordered.
  374. */
  375. if (time_in_range(cred->cr_expire, expired, jiffies) &&
  376. test_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags) != 0)
  377. return 0;
  378. list_del_init(&cred->cr_lru);
  379. number_cred_unused--;
  380. if (atomic_read(&cred->cr_count) != 0)
  381. continue;
  382. cache_lock = &cred->cr_auth->au_credcache->lock;
  383. spin_lock(cache_lock);
  384. if (atomic_read(&cred->cr_count) == 0) {
  385. get_rpccred(cred);
  386. list_add_tail(&cred->cr_lru, free);
  387. rpcauth_unhash_cred_locked(cred);
  388. }
  389. spin_unlock(cache_lock);
  390. }
  391. return (number_cred_unused / 100) * sysctl_vfs_cache_pressure;
  392. }
  393. /*
  394. * Run memory cache shrinker.
  395. */
  396. static int
  397. rpcauth_cache_shrinker(struct shrinker *shrink, struct shrink_control *sc)
  398. {
  399. LIST_HEAD(free);
  400. int res;
  401. int nr_to_scan = sc->nr_to_scan;
  402. gfp_t gfp_mask = sc->gfp_mask;
  403. if ((gfp_mask & GFP_KERNEL) != GFP_KERNEL)
  404. return (nr_to_scan == 0) ? 0 : -1;
  405. if (list_empty(&cred_unused))
  406. return 0;
  407. spin_lock(&rpc_credcache_lock);
  408. res = rpcauth_prune_expired(&free, nr_to_scan);
  409. spin_unlock(&rpc_credcache_lock);
  410. rpcauth_destroy_credlist(&free);
  411. return res;
  412. }
  413. /*
  414. * Look up a process' credentials in the authentication cache
  415. */
  416. struct rpc_cred *
  417. rpcauth_lookup_credcache(struct rpc_auth *auth, struct auth_cred * acred,
  418. int flags)
  419. {
  420. LIST_HEAD(free);
  421. struct rpc_cred_cache *cache = auth->au_credcache;
  422. struct rpc_cred *cred = NULL,
  423. *entry, *new;
  424. unsigned int nr;
  425. nr = hash_long(from_kuid(&init_user_ns, acred->uid), cache->hashbits);
  426. rcu_read_lock();
  427. hlist_for_each_entry_rcu(entry, &cache->hashtable[nr], cr_hash) {
  428. if (!entry->cr_ops->crmatch(acred, entry, flags))
  429. continue;
  430. spin_lock(&cache->lock);
  431. if (test_bit(RPCAUTH_CRED_HASHED, &entry->cr_flags) == 0) {
  432. spin_unlock(&cache->lock);
  433. continue;
  434. }
  435. cred = get_rpccred(entry);
  436. spin_unlock(&cache->lock);
  437. break;
  438. }
  439. rcu_read_unlock();
  440. if (cred != NULL)
  441. goto found;
  442. new = auth->au_ops->crcreate(auth, acred, flags);
  443. if (IS_ERR(new)) {
  444. cred = new;
  445. goto out;
  446. }
  447. spin_lock(&cache->lock);
  448. hlist_for_each_entry(entry, &cache->hashtable[nr], cr_hash) {
  449. if (!entry->cr_ops->crmatch(acred, entry, flags))
  450. continue;
  451. cred = get_rpccred(entry);
  452. break;
  453. }
  454. if (cred == NULL) {
  455. cred = new;
  456. set_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags);
  457. hlist_add_head_rcu(&cred->cr_hash, &cache->hashtable[nr]);
  458. } else
  459. list_add_tail(&new->cr_lru, &free);
  460. spin_unlock(&cache->lock);
  461. found:
  462. if (test_bit(RPCAUTH_CRED_NEW, &cred->cr_flags) &&
  463. cred->cr_ops->cr_init != NULL &&
  464. !(flags & RPCAUTH_LOOKUP_NEW)) {
  465. int res = cred->cr_ops->cr_init(auth, cred);
  466. if (res < 0) {
  467. put_rpccred(cred);
  468. cred = ERR_PTR(res);
  469. }
  470. }
  471. rpcauth_destroy_credlist(&free);
  472. out:
  473. return cred;
  474. }
  475. EXPORT_SYMBOL_GPL(rpcauth_lookup_credcache);
  476. struct rpc_cred *
  477. rpcauth_lookupcred(struct rpc_auth *auth, int flags)
  478. {
  479. struct auth_cred acred;
  480. struct rpc_cred *ret;
  481. const struct cred *cred = current_cred();
  482. dprintk("RPC: looking up %s cred\n",
  483. auth->au_ops->au_name);
  484. memset(&acred, 0, sizeof(acred));
  485. acred.uid = cred->fsuid;
  486. acred.gid = cred->fsgid;
  487. acred.group_info = get_group_info(((struct cred *)cred)->group_info);
  488. ret = auth->au_ops->lookup_cred(auth, &acred, flags);
  489. put_group_info(acred.group_info);
  490. return ret;
  491. }
  492. void
  493. rpcauth_init_cred(struct rpc_cred *cred, const struct auth_cred *acred,
  494. struct rpc_auth *auth, const struct rpc_credops *ops)
  495. {
  496. INIT_HLIST_NODE(&cred->cr_hash);
  497. INIT_LIST_HEAD(&cred->cr_lru);
  498. atomic_set(&cred->cr_count, 1);
  499. cred->cr_auth = auth;
  500. cred->cr_ops = ops;
  501. cred->cr_expire = jiffies;
  502. #ifdef RPC_DEBUG
  503. cred->cr_magic = RPCAUTH_CRED_MAGIC;
  504. #endif
  505. cred->cr_uid = acred->uid;
  506. }
  507. EXPORT_SYMBOL_GPL(rpcauth_init_cred);
  508. struct rpc_cred *
  509. rpcauth_generic_bind_cred(struct rpc_task *task, struct rpc_cred *cred, int lookupflags)
  510. {
  511. dprintk("RPC: %5u holding %s cred %p\n", task->tk_pid,
  512. cred->cr_auth->au_ops->au_name, cred);
  513. return get_rpccred(cred);
  514. }
  515. EXPORT_SYMBOL_GPL(rpcauth_generic_bind_cred);
  516. static struct rpc_cred *
  517. rpcauth_bind_root_cred(struct rpc_task *task, int lookupflags)
  518. {
  519. struct rpc_auth *auth = task->tk_client->cl_auth;
  520. struct auth_cred acred = {
  521. .uid = GLOBAL_ROOT_UID,
  522. .gid = GLOBAL_ROOT_GID,
  523. };
  524. dprintk("RPC: %5u looking up %s cred\n",
  525. task->tk_pid, task->tk_client->cl_auth->au_ops->au_name);
  526. return auth->au_ops->lookup_cred(auth, &acred, lookupflags);
  527. }
  528. static struct rpc_cred *
  529. rpcauth_bind_new_cred(struct rpc_task *task, int lookupflags)
  530. {
  531. struct rpc_auth *auth = task->tk_client->cl_auth;
  532. dprintk("RPC: %5u looking up %s cred\n",
  533. task->tk_pid, auth->au_ops->au_name);
  534. return rpcauth_lookupcred(auth, lookupflags);
  535. }
  536. static int
  537. rpcauth_bindcred(struct rpc_task *task, struct rpc_cred *cred, int flags)
  538. {
  539. struct rpc_rqst *req = task->tk_rqstp;
  540. struct rpc_cred *new;
  541. int lookupflags = 0;
  542. if (flags & RPC_TASK_ASYNC)
  543. lookupflags |= RPCAUTH_LOOKUP_NEW;
  544. if (cred != NULL)
  545. new = cred->cr_ops->crbind(task, cred, lookupflags);
  546. else if (flags & RPC_TASK_ROOTCREDS)
  547. new = rpcauth_bind_root_cred(task, lookupflags);
  548. else
  549. new = rpcauth_bind_new_cred(task, lookupflags);
  550. if (IS_ERR(new))
  551. return PTR_ERR(new);
  552. if (req->rq_cred != NULL)
  553. put_rpccred(req->rq_cred);
  554. req->rq_cred = new;
  555. return 0;
  556. }
  557. void
  558. put_rpccred(struct rpc_cred *cred)
  559. {
  560. /* Fast path for unhashed credentials */
  561. if (test_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags) == 0) {
  562. if (atomic_dec_and_test(&cred->cr_count))
  563. cred->cr_ops->crdestroy(cred);
  564. return;
  565. }
  566. if (!atomic_dec_and_lock(&cred->cr_count, &rpc_credcache_lock))
  567. return;
  568. if (!list_empty(&cred->cr_lru)) {
  569. number_cred_unused--;
  570. list_del_init(&cred->cr_lru);
  571. }
  572. if (test_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags) != 0) {
  573. if (test_bit(RPCAUTH_CRED_UPTODATE, &cred->cr_flags) != 0) {
  574. cred->cr_expire = jiffies;
  575. list_add_tail(&cred->cr_lru, &cred_unused);
  576. number_cred_unused++;
  577. goto out_nodestroy;
  578. }
  579. if (!rpcauth_unhash_cred(cred)) {
  580. /* We were hashed and someone looked us up... */
  581. goto out_nodestroy;
  582. }
  583. }
  584. spin_unlock(&rpc_credcache_lock);
  585. cred->cr_ops->crdestroy(cred);
  586. return;
  587. out_nodestroy:
  588. spin_unlock(&rpc_credcache_lock);
  589. }
  590. EXPORT_SYMBOL_GPL(put_rpccred);
  591. __be32 *
  592. rpcauth_marshcred(struct rpc_task *task, __be32 *p)
  593. {
  594. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  595. dprintk("RPC: %5u marshaling %s cred %p\n",
  596. task->tk_pid, cred->cr_auth->au_ops->au_name, cred);
  597. return cred->cr_ops->crmarshal(task, p);
  598. }
  599. __be32 *
  600. rpcauth_checkverf(struct rpc_task *task, __be32 *p)
  601. {
  602. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  603. dprintk("RPC: %5u validating %s cred %p\n",
  604. task->tk_pid, cred->cr_auth->au_ops->au_name, cred);
  605. return cred->cr_ops->crvalidate(task, p);
  606. }
  607. static void rpcauth_wrap_req_encode(kxdreproc_t encode, struct rpc_rqst *rqstp,
  608. __be32 *data, void *obj)
  609. {
  610. struct xdr_stream xdr;
  611. xdr_init_encode(&xdr, &rqstp->rq_snd_buf, data);
  612. encode(rqstp, &xdr, obj);
  613. }
  614. int
  615. rpcauth_wrap_req(struct rpc_task *task, kxdreproc_t encode, void *rqstp,
  616. __be32 *data, void *obj)
  617. {
  618. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  619. dprintk("RPC: %5u using %s cred %p to wrap rpc data\n",
  620. task->tk_pid, cred->cr_ops->cr_name, cred);
  621. if (cred->cr_ops->crwrap_req)
  622. return cred->cr_ops->crwrap_req(task, encode, rqstp, data, obj);
  623. /* By default, we encode the arguments normally. */
  624. rpcauth_wrap_req_encode(encode, rqstp, data, obj);
  625. return 0;
  626. }
  627. static int
  628. rpcauth_unwrap_req_decode(kxdrdproc_t decode, struct rpc_rqst *rqstp,
  629. __be32 *data, void *obj)
  630. {
  631. struct xdr_stream xdr;
  632. xdr_init_decode(&xdr, &rqstp->rq_rcv_buf, data);
  633. return decode(rqstp, &xdr, obj);
  634. }
  635. int
  636. rpcauth_unwrap_resp(struct rpc_task *task, kxdrdproc_t decode, void *rqstp,
  637. __be32 *data, void *obj)
  638. {
  639. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  640. dprintk("RPC: %5u using %s cred %p to unwrap rpc data\n",
  641. task->tk_pid, cred->cr_ops->cr_name, cred);
  642. if (cred->cr_ops->crunwrap_resp)
  643. return cred->cr_ops->crunwrap_resp(task, decode, rqstp,
  644. data, obj);
  645. /* By default, we decode the arguments normally. */
  646. return rpcauth_unwrap_req_decode(decode, rqstp, data, obj);
  647. }
  648. int
  649. rpcauth_refreshcred(struct rpc_task *task)
  650. {
  651. struct rpc_cred *cred;
  652. int err;
  653. cred = task->tk_rqstp->rq_cred;
  654. if (cred == NULL) {
  655. err = rpcauth_bindcred(task, task->tk_msg.rpc_cred, task->tk_flags);
  656. if (err < 0)
  657. goto out;
  658. cred = task->tk_rqstp->rq_cred;
  659. }
  660. dprintk("RPC: %5u refreshing %s cred %p\n",
  661. task->tk_pid, cred->cr_auth->au_ops->au_name, cred);
  662. err = cred->cr_ops->crrefresh(task);
  663. out:
  664. if (err < 0)
  665. task->tk_status = err;
  666. return err;
  667. }
  668. void
  669. rpcauth_invalcred(struct rpc_task *task)
  670. {
  671. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  672. dprintk("RPC: %5u invalidating %s cred %p\n",
  673. task->tk_pid, cred->cr_auth->au_ops->au_name, cred);
  674. if (cred)
  675. clear_bit(RPCAUTH_CRED_UPTODATE, &cred->cr_flags);
  676. }
  677. int
  678. rpcauth_uptodatecred(struct rpc_task *task)
  679. {
  680. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  681. return cred == NULL ||
  682. test_bit(RPCAUTH_CRED_UPTODATE, &cred->cr_flags) != 0;
  683. }
  684. static struct shrinker rpc_cred_shrinker = {
  685. .shrink = rpcauth_cache_shrinker,
  686. .seeks = DEFAULT_SEEKS,
  687. };
  688. int __init rpcauth_init_module(void)
  689. {
  690. int err;
  691. err = rpc_init_authunix();
  692. if (err < 0)
  693. goto out1;
  694. err = rpc_init_generic_auth();
  695. if (err < 0)
  696. goto out2;
  697. register_shrinker(&rpc_cred_shrinker);
  698. return 0;
  699. out2:
  700. rpc_destroy_authunix();
  701. out1:
  702. return err;
  703. }
  704. void rpcauth_remove_module(void)
  705. {
  706. rpc_destroy_authunix();
  707. rpc_destroy_generic_auth();
  708. unregister_shrinker(&rpc_cred_shrinker);
  709. }