auth.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814
  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. if (flavor >= RPC_AUTH_MAXFLAVOR)
  153. return -EINVAL;
  154. ops = auth_flavors[flavor];
  155. if (ops == NULL)
  156. request_module("rpc-auth-%u", flavor);
  157. spin_lock(&rpc_authflavor_lock);
  158. ops = auth_flavors[flavor];
  159. if (ops == NULL || !try_module_get(ops->owner)) {
  160. spin_unlock(&rpc_authflavor_lock);
  161. return -ENOENT;
  162. }
  163. spin_unlock(&rpc_authflavor_lock);
  164. result = -ENOENT;
  165. if (ops->flavor2info != NULL)
  166. result = ops->flavor2info(pseudoflavor, info);
  167. module_put(ops->owner);
  168. return result;
  169. }
  170. EXPORT_SYMBOL_GPL(rpcauth_get_gssinfo);
  171. /**
  172. * rpcauth_list_flavors - discover registered flavors and pseudoflavors
  173. * @array: array to fill in
  174. * @size: size of "array"
  175. *
  176. * Returns the number of array items filled in, or a negative errno.
  177. *
  178. * The returned array is not sorted by any policy. Callers should not
  179. * rely on the order of the items in the returned array.
  180. */
  181. int
  182. rpcauth_list_flavors(rpc_authflavor_t *array, int size)
  183. {
  184. rpc_authflavor_t flavor;
  185. int result = 0;
  186. spin_lock(&rpc_authflavor_lock);
  187. for (flavor = 0; flavor < RPC_AUTH_MAXFLAVOR; flavor++) {
  188. const struct rpc_authops *ops = auth_flavors[flavor];
  189. rpc_authflavor_t pseudos[4];
  190. int i, len;
  191. if (result >= size) {
  192. result = -ENOMEM;
  193. break;
  194. }
  195. if (ops == NULL)
  196. continue;
  197. if (ops->list_pseudoflavors == NULL) {
  198. array[result++] = ops->au_flavor;
  199. continue;
  200. }
  201. len = ops->list_pseudoflavors(pseudos, ARRAY_SIZE(pseudos));
  202. if (len < 0) {
  203. result = len;
  204. break;
  205. }
  206. for (i = 0; i < len; i++) {
  207. if (result >= size) {
  208. result = -ENOMEM;
  209. break;
  210. }
  211. array[result++] = pseudos[i];
  212. }
  213. }
  214. spin_unlock(&rpc_authflavor_lock);
  215. dprintk("RPC: %s returns %d\n", __func__, result);
  216. return result;
  217. }
  218. EXPORT_SYMBOL_GPL(rpcauth_list_flavors);
  219. struct rpc_auth *
  220. rpcauth_create(rpc_authflavor_t pseudoflavor, struct rpc_clnt *clnt)
  221. {
  222. struct rpc_auth *auth;
  223. const struct rpc_authops *ops;
  224. u32 flavor = pseudoflavor_to_flavor(pseudoflavor);
  225. auth = ERR_PTR(-EINVAL);
  226. if (flavor >= RPC_AUTH_MAXFLAVOR)
  227. goto out;
  228. if ((ops = auth_flavors[flavor]) == NULL)
  229. request_module("rpc-auth-%u", flavor);
  230. spin_lock(&rpc_authflavor_lock);
  231. ops = auth_flavors[flavor];
  232. if (ops == NULL || !try_module_get(ops->owner)) {
  233. spin_unlock(&rpc_authflavor_lock);
  234. goto out;
  235. }
  236. spin_unlock(&rpc_authflavor_lock);
  237. auth = ops->create(clnt, pseudoflavor);
  238. module_put(ops->owner);
  239. if (IS_ERR(auth))
  240. return auth;
  241. if (clnt->cl_auth)
  242. rpcauth_release(clnt->cl_auth);
  243. clnt->cl_auth = auth;
  244. out:
  245. return auth;
  246. }
  247. EXPORT_SYMBOL_GPL(rpcauth_create);
  248. void
  249. rpcauth_release(struct rpc_auth *auth)
  250. {
  251. if (!atomic_dec_and_test(&auth->au_count))
  252. return;
  253. auth->au_ops->destroy(auth);
  254. }
  255. static DEFINE_SPINLOCK(rpc_credcache_lock);
  256. static void
  257. rpcauth_unhash_cred_locked(struct rpc_cred *cred)
  258. {
  259. hlist_del_rcu(&cred->cr_hash);
  260. smp_mb__before_clear_bit();
  261. clear_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags);
  262. }
  263. static int
  264. rpcauth_unhash_cred(struct rpc_cred *cred)
  265. {
  266. spinlock_t *cache_lock;
  267. int ret;
  268. cache_lock = &cred->cr_auth->au_credcache->lock;
  269. spin_lock(cache_lock);
  270. ret = atomic_read(&cred->cr_count) == 0;
  271. if (ret)
  272. rpcauth_unhash_cred_locked(cred);
  273. spin_unlock(cache_lock);
  274. return ret;
  275. }
  276. /*
  277. * Initialize RPC credential cache
  278. */
  279. int
  280. rpcauth_init_credcache(struct rpc_auth *auth)
  281. {
  282. struct rpc_cred_cache *new;
  283. unsigned int hashsize;
  284. new = kmalloc(sizeof(*new), GFP_KERNEL);
  285. if (!new)
  286. goto out_nocache;
  287. new->hashbits = auth_hashbits;
  288. hashsize = 1U << new->hashbits;
  289. new->hashtable = kcalloc(hashsize, sizeof(new->hashtable[0]), GFP_KERNEL);
  290. if (!new->hashtable)
  291. goto out_nohashtbl;
  292. spin_lock_init(&new->lock);
  293. auth->au_credcache = new;
  294. return 0;
  295. out_nohashtbl:
  296. kfree(new);
  297. out_nocache:
  298. return -ENOMEM;
  299. }
  300. EXPORT_SYMBOL_GPL(rpcauth_init_credcache);
  301. /*
  302. * Destroy a list of credentials
  303. */
  304. static inline
  305. void rpcauth_destroy_credlist(struct list_head *head)
  306. {
  307. struct rpc_cred *cred;
  308. while (!list_empty(head)) {
  309. cred = list_entry(head->next, struct rpc_cred, cr_lru);
  310. list_del_init(&cred->cr_lru);
  311. put_rpccred(cred);
  312. }
  313. }
  314. /*
  315. * Clear the RPC credential cache, and delete those credentials
  316. * that are not referenced.
  317. */
  318. void
  319. rpcauth_clear_credcache(struct rpc_cred_cache *cache)
  320. {
  321. LIST_HEAD(free);
  322. struct hlist_head *head;
  323. struct rpc_cred *cred;
  324. unsigned int hashsize = 1U << cache->hashbits;
  325. int i;
  326. spin_lock(&rpc_credcache_lock);
  327. spin_lock(&cache->lock);
  328. for (i = 0; i < hashsize; i++) {
  329. head = &cache->hashtable[i];
  330. while (!hlist_empty(head)) {
  331. cred = hlist_entry(head->first, struct rpc_cred, cr_hash);
  332. get_rpccred(cred);
  333. if (!list_empty(&cred->cr_lru)) {
  334. list_del(&cred->cr_lru);
  335. number_cred_unused--;
  336. }
  337. list_add_tail(&cred->cr_lru, &free);
  338. rpcauth_unhash_cred_locked(cred);
  339. }
  340. }
  341. spin_unlock(&cache->lock);
  342. spin_unlock(&rpc_credcache_lock);
  343. rpcauth_destroy_credlist(&free);
  344. }
  345. /*
  346. * Destroy the RPC credential cache
  347. */
  348. void
  349. rpcauth_destroy_credcache(struct rpc_auth *auth)
  350. {
  351. struct rpc_cred_cache *cache = auth->au_credcache;
  352. if (cache) {
  353. auth->au_credcache = NULL;
  354. rpcauth_clear_credcache(cache);
  355. kfree(cache->hashtable);
  356. kfree(cache);
  357. }
  358. }
  359. EXPORT_SYMBOL_GPL(rpcauth_destroy_credcache);
  360. #define RPC_AUTH_EXPIRY_MORATORIUM (60 * HZ)
  361. /*
  362. * Remove stale credentials. Avoid sleeping inside the loop.
  363. */
  364. static int
  365. rpcauth_prune_expired(struct list_head *free, int nr_to_scan)
  366. {
  367. spinlock_t *cache_lock;
  368. struct rpc_cred *cred, *next;
  369. unsigned long expired = jiffies - RPC_AUTH_EXPIRY_MORATORIUM;
  370. list_for_each_entry_safe(cred, next, &cred_unused, cr_lru) {
  371. if (nr_to_scan-- == 0)
  372. break;
  373. /*
  374. * Enforce a 60 second garbage collection moratorium
  375. * Note that the cred_unused list must be time-ordered.
  376. */
  377. if (time_in_range(cred->cr_expire, expired, jiffies) &&
  378. test_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags) != 0)
  379. return 0;
  380. list_del_init(&cred->cr_lru);
  381. number_cred_unused--;
  382. if (atomic_read(&cred->cr_count) != 0)
  383. continue;
  384. cache_lock = &cred->cr_auth->au_credcache->lock;
  385. spin_lock(cache_lock);
  386. if (atomic_read(&cred->cr_count) == 0) {
  387. get_rpccred(cred);
  388. list_add_tail(&cred->cr_lru, free);
  389. rpcauth_unhash_cred_locked(cred);
  390. }
  391. spin_unlock(cache_lock);
  392. }
  393. return (number_cred_unused / 100) * sysctl_vfs_cache_pressure;
  394. }
  395. /*
  396. * Run memory cache shrinker.
  397. */
  398. static int
  399. rpcauth_cache_shrinker(struct shrinker *shrink, struct shrink_control *sc)
  400. {
  401. LIST_HEAD(free);
  402. int res;
  403. int nr_to_scan = sc->nr_to_scan;
  404. gfp_t gfp_mask = sc->gfp_mask;
  405. if ((gfp_mask & GFP_KERNEL) != GFP_KERNEL)
  406. return (nr_to_scan == 0) ? 0 : -1;
  407. if (list_empty(&cred_unused))
  408. return 0;
  409. spin_lock(&rpc_credcache_lock);
  410. res = rpcauth_prune_expired(&free, nr_to_scan);
  411. spin_unlock(&rpc_credcache_lock);
  412. rpcauth_destroy_credlist(&free);
  413. return res;
  414. }
  415. /*
  416. * Look up a process' credentials in the authentication cache
  417. */
  418. struct rpc_cred *
  419. rpcauth_lookup_credcache(struct rpc_auth *auth, struct auth_cred * acred,
  420. int flags)
  421. {
  422. LIST_HEAD(free);
  423. struct rpc_cred_cache *cache = auth->au_credcache;
  424. struct rpc_cred *cred = NULL,
  425. *entry, *new;
  426. unsigned int nr;
  427. nr = hash_long(from_kuid(&init_user_ns, acred->uid), cache->hashbits);
  428. rcu_read_lock();
  429. hlist_for_each_entry_rcu(entry, &cache->hashtable[nr], cr_hash) {
  430. if (!entry->cr_ops->crmatch(acred, entry, flags))
  431. continue;
  432. spin_lock(&cache->lock);
  433. if (test_bit(RPCAUTH_CRED_HASHED, &entry->cr_flags) == 0) {
  434. spin_unlock(&cache->lock);
  435. continue;
  436. }
  437. cred = get_rpccred(entry);
  438. spin_unlock(&cache->lock);
  439. break;
  440. }
  441. rcu_read_unlock();
  442. if (cred != NULL)
  443. goto found;
  444. new = auth->au_ops->crcreate(auth, acred, flags);
  445. if (IS_ERR(new)) {
  446. cred = new;
  447. goto out;
  448. }
  449. spin_lock(&cache->lock);
  450. hlist_for_each_entry(entry, &cache->hashtable[nr], cr_hash) {
  451. if (!entry->cr_ops->crmatch(acred, entry, flags))
  452. continue;
  453. cred = get_rpccred(entry);
  454. break;
  455. }
  456. if (cred == NULL) {
  457. cred = new;
  458. set_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags);
  459. hlist_add_head_rcu(&cred->cr_hash, &cache->hashtable[nr]);
  460. } else
  461. list_add_tail(&new->cr_lru, &free);
  462. spin_unlock(&cache->lock);
  463. found:
  464. if (test_bit(RPCAUTH_CRED_NEW, &cred->cr_flags) &&
  465. cred->cr_ops->cr_init != NULL &&
  466. !(flags & RPCAUTH_LOOKUP_NEW)) {
  467. int res = cred->cr_ops->cr_init(auth, cred);
  468. if (res < 0) {
  469. put_rpccred(cred);
  470. cred = ERR_PTR(res);
  471. }
  472. }
  473. rpcauth_destroy_credlist(&free);
  474. out:
  475. return cred;
  476. }
  477. EXPORT_SYMBOL_GPL(rpcauth_lookup_credcache);
  478. struct rpc_cred *
  479. rpcauth_lookupcred(struct rpc_auth *auth, int flags)
  480. {
  481. struct auth_cred acred;
  482. struct rpc_cred *ret;
  483. const struct cred *cred = current_cred();
  484. dprintk("RPC: looking up %s cred\n",
  485. auth->au_ops->au_name);
  486. memset(&acred, 0, sizeof(acred));
  487. acred.uid = cred->fsuid;
  488. acred.gid = cred->fsgid;
  489. acred.group_info = get_group_info(((struct cred *)cred)->group_info);
  490. ret = auth->au_ops->lookup_cred(auth, &acred, flags);
  491. put_group_info(acred.group_info);
  492. return ret;
  493. }
  494. void
  495. rpcauth_init_cred(struct rpc_cred *cred, const struct auth_cred *acred,
  496. struct rpc_auth *auth, const struct rpc_credops *ops)
  497. {
  498. INIT_HLIST_NODE(&cred->cr_hash);
  499. INIT_LIST_HEAD(&cred->cr_lru);
  500. atomic_set(&cred->cr_count, 1);
  501. cred->cr_auth = auth;
  502. cred->cr_ops = ops;
  503. cred->cr_expire = jiffies;
  504. #ifdef RPC_DEBUG
  505. cred->cr_magic = RPCAUTH_CRED_MAGIC;
  506. #endif
  507. cred->cr_uid = acred->uid;
  508. }
  509. EXPORT_SYMBOL_GPL(rpcauth_init_cred);
  510. struct rpc_cred *
  511. rpcauth_generic_bind_cred(struct rpc_task *task, struct rpc_cred *cred, int lookupflags)
  512. {
  513. dprintk("RPC: %5u holding %s cred %p\n", task->tk_pid,
  514. cred->cr_auth->au_ops->au_name, cred);
  515. return get_rpccred(cred);
  516. }
  517. EXPORT_SYMBOL_GPL(rpcauth_generic_bind_cred);
  518. static struct rpc_cred *
  519. rpcauth_bind_root_cred(struct rpc_task *task, int lookupflags)
  520. {
  521. struct rpc_auth *auth = task->tk_client->cl_auth;
  522. struct auth_cred acred = {
  523. .uid = GLOBAL_ROOT_UID,
  524. .gid = GLOBAL_ROOT_GID,
  525. };
  526. dprintk("RPC: %5u looking up %s cred\n",
  527. task->tk_pid, task->tk_client->cl_auth->au_ops->au_name);
  528. return auth->au_ops->lookup_cred(auth, &acred, lookupflags);
  529. }
  530. static struct rpc_cred *
  531. rpcauth_bind_new_cred(struct rpc_task *task, int lookupflags)
  532. {
  533. struct rpc_auth *auth = task->tk_client->cl_auth;
  534. dprintk("RPC: %5u looking up %s cred\n",
  535. task->tk_pid, auth->au_ops->au_name);
  536. return rpcauth_lookupcred(auth, lookupflags);
  537. }
  538. static int
  539. rpcauth_bindcred(struct rpc_task *task, struct rpc_cred *cred, int flags)
  540. {
  541. struct rpc_rqst *req = task->tk_rqstp;
  542. struct rpc_cred *new;
  543. int lookupflags = 0;
  544. if (flags & RPC_TASK_ASYNC)
  545. lookupflags |= RPCAUTH_LOOKUP_NEW;
  546. if (cred != NULL)
  547. new = cred->cr_ops->crbind(task, cred, lookupflags);
  548. else if (flags & RPC_TASK_ROOTCREDS)
  549. new = rpcauth_bind_root_cred(task, lookupflags);
  550. else
  551. new = rpcauth_bind_new_cred(task, lookupflags);
  552. if (IS_ERR(new))
  553. return PTR_ERR(new);
  554. if (req->rq_cred != NULL)
  555. put_rpccred(req->rq_cred);
  556. req->rq_cred = new;
  557. return 0;
  558. }
  559. void
  560. put_rpccred(struct rpc_cred *cred)
  561. {
  562. /* Fast path for unhashed credentials */
  563. if (test_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags) == 0) {
  564. if (atomic_dec_and_test(&cred->cr_count))
  565. cred->cr_ops->crdestroy(cred);
  566. return;
  567. }
  568. if (!atomic_dec_and_lock(&cred->cr_count, &rpc_credcache_lock))
  569. return;
  570. if (!list_empty(&cred->cr_lru)) {
  571. number_cred_unused--;
  572. list_del_init(&cred->cr_lru);
  573. }
  574. if (test_bit(RPCAUTH_CRED_HASHED, &cred->cr_flags) != 0) {
  575. if (test_bit(RPCAUTH_CRED_UPTODATE, &cred->cr_flags) != 0) {
  576. cred->cr_expire = jiffies;
  577. list_add_tail(&cred->cr_lru, &cred_unused);
  578. number_cred_unused++;
  579. goto out_nodestroy;
  580. }
  581. if (!rpcauth_unhash_cred(cred)) {
  582. /* We were hashed and someone looked us up... */
  583. goto out_nodestroy;
  584. }
  585. }
  586. spin_unlock(&rpc_credcache_lock);
  587. cred->cr_ops->crdestroy(cred);
  588. return;
  589. out_nodestroy:
  590. spin_unlock(&rpc_credcache_lock);
  591. }
  592. EXPORT_SYMBOL_GPL(put_rpccred);
  593. __be32 *
  594. rpcauth_marshcred(struct rpc_task *task, __be32 *p)
  595. {
  596. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  597. dprintk("RPC: %5u marshaling %s cred %p\n",
  598. task->tk_pid, cred->cr_auth->au_ops->au_name, cred);
  599. return cred->cr_ops->crmarshal(task, p);
  600. }
  601. __be32 *
  602. rpcauth_checkverf(struct rpc_task *task, __be32 *p)
  603. {
  604. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  605. dprintk("RPC: %5u validating %s cred %p\n",
  606. task->tk_pid, cred->cr_auth->au_ops->au_name, cred);
  607. return cred->cr_ops->crvalidate(task, p);
  608. }
  609. static void rpcauth_wrap_req_encode(kxdreproc_t encode, struct rpc_rqst *rqstp,
  610. __be32 *data, void *obj)
  611. {
  612. struct xdr_stream xdr;
  613. xdr_init_encode(&xdr, &rqstp->rq_snd_buf, data);
  614. encode(rqstp, &xdr, obj);
  615. }
  616. int
  617. rpcauth_wrap_req(struct rpc_task *task, kxdreproc_t encode, void *rqstp,
  618. __be32 *data, void *obj)
  619. {
  620. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  621. dprintk("RPC: %5u using %s cred %p to wrap rpc data\n",
  622. task->tk_pid, cred->cr_ops->cr_name, cred);
  623. if (cred->cr_ops->crwrap_req)
  624. return cred->cr_ops->crwrap_req(task, encode, rqstp, data, obj);
  625. /* By default, we encode the arguments normally. */
  626. rpcauth_wrap_req_encode(encode, rqstp, data, obj);
  627. return 0;
  628. }
  629. static int
  630. rpcauth_unwrap_req_decode(kxdrdproc_t decode, struct rpc_rqst *rqstp,
  631. __be32 *data, void *obj)
  632. {
  633. struct xdr_stream xdr;
  634. xdr_init_decode(&xdr, &rqstp->rq_rcv_buf, data);
  635. return decode(rqstp, &xdr, obj);
  636. }
  637. int
  638. rpcauth_unwrap_resp(struct rpc_task *task, kxdrdproc_t decode, void *rqstp,
  639. __be32 *data, void *obj)
  640. {
  641. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  642. dprintk("RPC: %5u using %s cred %p to unwrap rpc data\n",
  643. task->tk_pid, cred->cr_ops->cr_name, cred);
  644. if (cred->cr_ops->crunwrap_resp)
  645. return cred->cr_ops->crunwrap_resp(task, decode, rqstp,
  646. data, obj);
  647. /* By default, we decode the arguments normally. */
  648. return rpcauth_unwrap_req_decode(decode, rqstp, data, obj);
  649. }
  650. int
  651. rpcauth_refreshcred(struct rpc_task *task)
  652. {
  653. struct rpc_cred *cred;
  654. int err;
  655. cred = task->tk_rqstp->rq_cred;
  656. if (cred == NULL) {
  657. err = rpcauth_bindcred(task, task->tk_msg.rpc_cred, task->tk_flags);
  658. if (err < 0)
  659. goto out;
  660. cred = task->tk_rqstp->rq_cred;
  661. }
  662. dprintk("RPC: %5u refreshing %s cred %p\n",
  663. task->tk_pid, cred->cr_auth->au_ops->au_name, cred);
  664. err = cred->cr_ops->crrefresh(task);
  665. out:
  666. if (err < 0)
  667. task->tk_status = err;
  668. return err;
  669. }
  670. void
  671. rpcauth_invalcred(struct rpc_task *task)
  672. {
  673. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  674. dprintk("RPC: %5u invalidating %s cred %p\n",
  675. task->tk_pid, cred->cr_auth->au_ops->au_name, cred);
  676. if (cred)
  677. clear_bit(RPCAUTH_CRED_UPTODATE, &cred->cr_flags);
  678. }
  679. int
  680. rpcauth_uptodatecred(struct rpc_task *task)
  681. {
  682. struct rpc_cred *cred = task->tk_rqstp->rq_cred;
  683. return cred == NULL ||
  684. test_bit(RPCAUTH_CRED_UPTODATE, &cred->cr_flags) != 0;
  685. }
  686. static struct shrinker rpc_cred_shrinker = {
  687. .shrink = rpcauth_cache_shrinker,
  688. .seeks = DEFAULT_SEEKS,
  689. };
  690. int __init rpcauth_init_module(void)
  691. {
  692. int err;
  693. err = rpc_init_authunix();
  694. if (err < 0)
  695. goto out1;
  696. err = rpc_init_generic_auth();
  697. if (err < 0)
  698. goto out2;
  699. register_shrinker(&rpc_cred_shrinker);
  700. return 0;
  701. out2:
  702. rpc_destroy_authunix();
  703. out1:
  704. return err;
  705. }
  706. void rpcauth_remove_module(void)
  707. {
  708. rpc_destroy_authunix();
  709. rpc_destroy_generic_auth();
  710. unregister_shrinker(&rpc_cred_shrinker);
  711. }