2 * linux/fs/nfs/dns_resolve.c
4 * Copyright (c) 2009 Trond Myklebust <Trond.Myklebust@netapp.com>
6 * Resolves DNS hostnames into valid ip addresses
9 #ifdef CONFIG_NFS_USE_KERNEL_DNS
11 #include <linux/sunrpc/clnt.h>
12 #include <linux/dns_resolver.h>
14 ssize_t
nfs_dns_resolve_name(struct net
*net
, char *name
, size_t namelen
,
15 struct sockaddr
*sa
, size_t salen
)
21 ip_len
= dns_query(NULL
, name
, namelen
, NULL
, &ip_addr
, NULL
);
23 ret
= rpc_pton(&init_net
, ip_addr
, ip_len
, sa
, salen
);
32 #include <linux/hash.h>
33 #include <linux/string.h>
34 #include <linux/kmod.h>
35 #include <linux/slab.h>
36 #include <linux/module.h>
37 #include <linux/socket.h>
38 #include <linux/seq_file.h>
39 #include <linux/inet.h>
40 #include <linux/sunrpc/clnt.h>
41 #include <linux/sunrpc/cache.h>
42 #include <linux/sunrpc/svcauth.h>
43 #include <linux/sunrpc/rpc_pipe_fs.h>
45 #include "dns_resolve.h"
46 #include "cache_lib.h"
49 #define NFS_DNS_HASHBITS 4
50 #define NFS_DNS_HASHTBL_SIZE (1 << NFS_DNS_HASHBITS)
58 struct sockaddr_storage addr
;
63 static void nfs_dns_ent_update(struct cache_head
*cnew
,
64 struct cache_head
*ckey
)
66 struct nfs_dns_ent
*new;
67 struct nfs_dns_ent
*key
;
69 new = container_of(cnew
, struct nfs_dns_ent
, h
);
70 key
= container_of(ckey
, struct nfs_dns_ent
, h
);
72 memcpy(&new->addr
, &key
->addr
, key
->addrlen
);
73 new->addrlen
= key
->addrlen
;
76 static void nfs_dns_ent_init(struct cache_head
*cnew
,
77 struct cache_head
*ckey
)
79 struct nfs_dns_ent
*new;
80 struct nfs_dns_ent
*key
;
82 new = container_of(cnew
, struct nfs_dns_ent
, h
);
83 key
= container_of(ckey
, struct nfs_dns_ent
, h
);
86 new->hostname
= kstrndup(key
->hostname
, key
->namelen
, GFP_KERNEL
);
88 new->namelen
= key
->namelen
;
89 nfs_dns_ent_update(cnew
, ckey
);
96 static void nfs_dns_ent_put(struct kref
*ref
)
98 struct nfs_dns_ent
*item
;
100 item
= container_of(ref
, struct nfs_dns_ent
, h
.ref
);
101 kfree(item
->hostname
);
105 static struct cache_head
*nfs_dns_ent_alloc(void)
107 struct nfs_dns_ent
*item
= kmalloc(sizeof(*item
), GFP_KERNEL
);
110 item
->hostname
= NULL
;
118 static unsigned int nfs_dns_hash(const struct nfs_dns_ent
*key
)
120 return hash_str(key
->hostname
, NFS_DNS_HASHBITS
);
123 static void nfs_dns_request(struct cache_detail
*cd
,
124 struct cache_head
*ch
,
125 char **bpp
, int *blen
)
127 struct nfs_dns_ent
*key
= container_of(ch
, struct nfs_dns_ent
, h
);
129 qword_add(bpp
, blen
, key
->hostname
);
133 static int nfs_dns_upcall(struct cache_detail
*cd
,
134 struct cache_head
*ch
)
136 struct nfs_dns_ent
*key
= container_of(ch
, struct nfs_dns_ent
, h
);
139 ret
= nfs_cache_upcall(cd
, key
->hostname
);
141 ret
= sunrpc_cache_pipe_upcall(cd
, ch
, nfs_dns_request
);
145 static int nfs_dns_match(struct cache_head
*ca
,
146 struct cache_head
*cb
)
148 struct nfs_dns_ent
*a
;
149 struct nfs_dns_ent
*b
;
151 a
= container_of(ca
, struct nfs_dns_ent
, h
);
152 b
= container_of(cb
, struct nfs_dns_ent
, h
);
154 if (a
->namelen
== 0 || a
->namelen
!= b
->namelen
)
156 return memcmp(a
->hostname
, b
->hostname
, a
->namelen
) == 0;
159 static int nfs_dns_show(struct seq_file
*m
, struct cache_detail
*cd
,
160 struct cache_head
*h
)
162 struct nfs_dns_ent
*item
;
166 seq_puts(m
, "# ip address hostname ttl\n");
169 item
= container_of(h
, struct nfs_dns_ent
, h
);
170 ttl
= item
->h
.expiry_time
- seconds_since_boot();
174 if (!test_bit(CACHE_NEGATIVE
, &h
->flags
)) {
175 char buf
[INET6_ADDRSTRLEN
+IPV6_SCOPE_ID_LEN
+1];
177 rpc_ntop((struct sockaddr
*)&item
->addr
, buf
, sizeof(buf
));
178 seq_printf(m
, "%15s ", buf
);
180 seq_puts(m
, "<none> ");
181 seq_printf(m
, "%15s %ld\n", item
->hostname
, ttl
);
185 static struct nfs_dns_ent
*nfs_dns_lookup(struct cache_detail
*cd
,
186 struct nfs_dns_ent
*key
)
188 struct cache_head
*ch
;
190 ch
= sunrpc_cache_lookup(cd
,
195 return container_of(ch
, struct nfs_dns_ent
, h
);
198 static struct nfs_dns_ent
*nfs_dns_update(struct cache_detail
*cd
,
199 struct nfs_dns_ent
*new,
200 struct nfs_dns_ent
*key
)
202 struct cache_head
*ch
;
204 ch
= sunrpc_cache_update(cd
,
209 return container_of(ch
, struct nfs_dns_ent
, h
);
212 static int nfs_dns_parse(struct cache_detail
*cd
, char *buf
, int buflen
)
214 char buf1
[NFS_DNS_HOSTNAME_MAXLEN
+1];
215 struct nfs_dns_ent key
, *item
;
220 if (buf
[buflen
-1] != '\n')
222 buf
[buflen
-1] = '\0';
224 len
= qword_get(&buf
, buf1
, sizeof(buf1
));
227 key
.addrlen
= rpc_pton(&init_net
, buf1
, len
,
228 (struct sockaddr
*)&key
.addr
,
231 len
= qword_get(&buf
, buf1
, sizeof(buf1
));
237 memset(&key
.h
, 0, sizeof(key
.h
));
239 ttl
= get_expiry(&buf
);
242 key
.h
.expiry_time
= ttl
+ seconds_since_boot();
245 item
= nfs_dns_lookup(cd
, &key
);
249 if (key
.addrlen
== 0)
250 set_bit(CACHE_NEGATIVE
, &key
.h
.flags
);
252 item
= nfs_dns_update(cd
, &key
, item
);
257 cache_put(&item
->h
, cd
);
262 static int do_cache_lookup(struct cache_detail
*cd
,
263 struct nfs_dns_ent
*key
,
264 struct nfs_dns_ent
**item
,
265 struct nfs_cache_defer_req
*dreq
)
269 *item
= nfs_dns_lookup(cd
, key
);
271 ret
= cache_check(cd
, &(*item
)->h
, &dreq
->req
);
278 static int do_cache_lookup_nowait(struct cache_detail
*cd
,
279 struct nfs_dns_ent
*key
,
280 struct nfs_dns_ent
**item
)
284 *item
= nfs_dns_lookup(cd
, key
);
288 if (!test_bit(CACHE_VALID
, &(*item
)->h
.flags
)
289 || (*item
)->h
.expiry_time
< seconds_since_boot()
290 || cd
->flush_time
> (*item
)->h
.last_refresh
)
293 if (test_bit(CACHE_NEGATIVE
, &(*item
)->h
.flags
))
297 cache_put(&(*item
)->h
, cd
);
303 static int do_cache_lookup_wait(struct cache_detail
*cd
,
304 struct nfs_dns_ent
*key
,
305 struct nfs_dns_ent
**item
)
307 struct nfs_cache_defer_req
*dreq
;
310 dreq
= nfs_cache_defer_req_alloc();
313 ret
= do_cache_lookup(cd
, key
, item
, dreq
);
314 if (ret
== -EAGAIN
) {
315 ret
= nfs_cache_wait_for_upcall(dreq
);
317 ret
= do_cache_lookup_nowait(cd
, key
, item
);
319 nfs_cache_defer_req_put(dreq
);
324 ssize_t
nfs_dns_resolve_name(struct net
*net
, char *name
,
325 size_t namelen
, struct sockaddr
*sa
, size_t salen
)
327 struct nfs_dns_ent key
= {
331 struct nfs_dns_ent
*item
= NULL
;
333 struct nfs_net
*nn
= net_generic(net
, nfs_net_id
);
335 ret
= do_cache_lookup_wait(nn
->nfs_dns_resolve
, &key
, &item
);
337 if (salen
>= item
->addrlen
) {
338 memcpy(sa
, &item
->addr
, item
->addrlen
);
342 cache_put(&item
->h
, nn
->nfs_dns_resolve
);
343 } else if (ret
== -ENOENT
)
348 int nfs_dns_resolver_cache_init(struct net
*net
)
351 struct nfs_net
*nn
= net_generic(net
, nfs_net_id
);
352 struct cache_detail
*cd
;
353 struct cache_head
**tbl
;
355 cd
= kzalloc(sizeof(struct cache_detail
), GFP_KERNEL
);
359 tbl
= kzalloc(NFS_DNS_HASHTBL_SIZE
* sizeof(struct cache_head
*),
364 cd
->owner
= THIS_MODULE
,
365 cd
->hash_size
= NFS_DNS_HASHTBL_SIZE
,
366 cd
->hash_table
= tbl
,
367 cd
->name
= "dns_resolve",
368 cd
->cache_put
= nfs_dns_ent_put
,
369 cd
->cache_upcall
= nfs_dns_upcall
,
370 cd
->cache_parse
= nfs_dns_parse
,
371 cd
->cache_show
= nfs_dns_show
,
372 cd
->match
= nfs_dns_match
,
373 cd
->init
= nfs_dns_ent_init
,
374 cd
->update
= nfs_dns_ent_update
,
375 cd
->alloc
= nfs_dns_ent_alloc
,
378 err
= nfs_cache_register_net(net
, cd
);
381 nn
->nfs_dns_resolve
= cd
;
385 nfs_cache_destroy(cd
);
386 kfree(cd
->hash_table
);
393 void nfs_dns_resolver_cache_destroy(struct net
*net
)
395 struct nfs_net
*nn
= net_generic(net
, nfs_net_id
);
396 struct cache_detail
*cd
= nn
->nfs_dns_resolve
;
398 nfs_cache_unregister_net(net
, cd
);
399 nfs_cache_destroy(cd
);
400 kfree(cd
->hash_table
);
404 static int rpc_pipefs_event(struct notifier_block
*nb
, unsigned long event
,
407 struct super_block
*sb
= ptr
;
408 struct net
*net
= sb
->s_fs_info
;
409 struct nfs_net
*nn
= net_generic(net
, nfs_net_id
);
410 struct cache_detail
*cd
= nn
->nfs_dns_resolve
;
416 if (!try_module_get(THIS_MODULE
))
420 case RPC_PIPEFS_MOUNT
:
421 ret
= nfs_cache_register_sb(sb
, cd
);
423 case RPC_PIPEFS_UMOUNT
:
424 nfs_cache_unregister_sb(sb
, cd
);
430 module_put(THIS_MODULE
);
434 static struct notifier_block nfs_dns_resolver_block
= {
435 .notifier_call
= rpc_pipefs_event
,
438 int nfs_dns_resolver_init(void)
440 return rpc_pipefs_notifier_register(&nfs_dns_resolver_block
);
443 void nfs_dns_resolver_destroy(void)
445 rpc_pipefs_notifier_unregister(&nfs_dns_resolver_block
);