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