auth.c 18 KB

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