1 /* Copyright (C) 1998-2005, 2006, 2007 Free Software Foundation, Inc.
2 This file is part of the GNU C Library.
3 Contributed by Ulrich Drepper <drepper@cygnus.com>, 1998.
5 The GNU C Library is free software; you can redistribute it and/or
6 modify it under the terms of the GNU Lesser General Public
7 License as published by the Free Software Foundation; either
8 version 2.1 of the License, or (at your option) any later version.
10 The GNU C Library is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 Lesser General Public License for more details.
15 You should have received a copy of the GNU Lesser General Public
16 License along with the GNU C Library; if not, write to the Free
17 Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
29 #include <arpa/nameser.h>
30 #include <not-cancel.h>
32 #include "nscd-client.h"
33 #include "nscd_proto.h"
35 int __nss_not_use_nscd_hosts
;
37 static int nscd_gethst_r (const char *key
, size_t keylen
, request_type type
,
38 struct hostent
*resultbuf
, char *buffer
,
39 size_t buflen
, struct hostent
**result
,
40 int *h_errnop
) internal_function
;
44 __nscd_gethostbyname_r (const char *name
, struct hostent
*resultbuf
,
45 char *buffer
, size_t buflen
, struct hostent
**result
,
50 reqtype
= (_res
.options
& RES_USE_INET6
) ? GETHOSTBYNAMEv6
: GETHOSTBYNAME
;
52 return nscd_gethst_r (name
, strlen (name
) + 1, reqtype
, resultbuf
,
53 buffer
, buflen
, result
, h_errnop
);
58 __nscd_gethostbyname2_r (const char *name
, int af
, struct hostent
*resultbuf
,
59 char *buffer
, size_t buflen
, struct hostent
**result
,
64 reqtype
= af
== AF_INET6
? GETHOSTBYNAMEv6
: GETHOSTBYNAME
;
66 return nscd_gethst_r (name
, strlen (name
) + 1, reqtype
, resultbuf
,
67 buffer
, buflen
, result
, h_errnop
);
72 __nscd_gethostbyaddr_r (const void *addr
, socklen_t len
, int type
,
73 struct hostent
*resultbuf
, char *buffer
, size_t buflen
,
74 struct hostent
**result
, int *h_errnop
)
78 if (!((len
== INADDRSZ
&& type
== AF_INET
)
79 || (len
== IN6ADDRSZ
&& type
== AF_INET6
)))
80 /* LEN and TYPE do not match. */
83 reqtype
= type
== AF_INET6
? GETHOSTBYADDRv6
: GETHOSTBYADDR
;
85 return nscd_gethst_r (addr
, len
, reqtype
, resultbuf
, buffer
, buflen
, result
,
90 libc_locked_map_ptr (, __hst_map_handle
) attribute_hidden
;
91 /* Note that we only free the structure if necessary. The memory
92 mapping is not removed since it is not visible to the malloc
94 libc_freeres_fn (hst_map_free
)
96 if (__hst_map_handle
.mapped
!= NO_MAPPING
)
98 void *p
= __hst_map_handle
.mapped
;
99 __hst_map_handle
.mapped
= NO_MAPPING
;
107 nscd_gethst_r (const char *key
, size_t keylen
, request_type type
,
108 struct hostent
*resultbuf
, char *buffer
, size_t buflen
,
109 struct hostent
**result
, int *h_errnop
)
114 /* If the mapping is available, try to search there instead of
115 communicating with the nscd. */
116 struct mapped_database
*mapped
;
117 mapped
= __nscd_get_map_ref (GETFDHST
, "hosts", &__hst_map_handle
,
121 const char *h_name
= NULL
;
122 const uint32_t *aliases_len
= NULL
;
123 const char *addr_list
= NULL
;
124 size_t addr_list_len
= 0;
126 const char *recend
= (const char *) ~UINTMAX_C (0);
128 hst_response_header hst_resp
;
129 if (mapped
!= NO_MAPPING
)
131 /* No const qualifier, as it can change during garbage collection. */
132 struct datahead
*found
= __nscd_cache_search (type
, key
, keylen
, mapped
);
135 h_name
= (char *) (&found
->data
[0].hstdata
+ 1);
136 hst_resp
= found
->data
[0].hstdata
;
137 aliases_len
= (uint32_t *) (h_name
+ hst_resp
.h_name_len
);
138 addr_list
= ((char *) aliases_len
139 + hst_resp
.h_aliases_cnt
* sizeof (uint32_t));
140 addr_list_len
= hst_resp
.h_addr_list_cnt
* INADDRSZ
;
141 recend
= (const char *) found
->data
+ found
->recsize
;
142 /* Now check if we can trust hst_resp fields. If GC is
143 in progress, it can contain anything. */
144 if (mapped
->head
->gc_cycle
!= gc_cycle
)
150 #ifndef _STRING_ARCH_unaligned
151 /* The aliases_len array in the mapped database might very
152 well be unaligned. We will access it word-wise so on
153 platforms which do not tolerate unaligned accesses we
154 need to make an aligned copy. */
155 if (((uintptr_t) aliases_len
& (__alignof__ (*aliases_len
) - 1))
158 uint32_t *tmp
= alloca (hst_resp
.h_aliases_cnt
159 * sizeof (uint32_t));
160 aliases_len
= memcpy (tmp
, aliases_len
,
161 hst_resp
.h_aliases_cnt
162 * sizeof (uint32_t));
165 if (type
!= GETHOSTBYADDR
&& type
!= GETHOSTBYNAME
)
167 if (hst_resp
.h_length
== INADDRSZ
)
168 addr_list
+= addr_list_len
;
169 addr_list_len
= hst_resp
.h_addr_list_cnt
* IN6ADDRSZ
;
171 if (__builtin_expect ((const char *) addr_list
+ addr_list_len
179 sock
= __nscd_open_socket (key
, keylen
, type
, &hst_resp
,
183 __nss_not_use_nscd_hosts
= 1;
188 /* No value found so far. */
191 if (__builtin_expect (hst_resp
.found
== -1, 0))
193 /* The daemon does not cache this database. */
194 __nss_not_use_nscd_hosts
= 1;
198 if (hst_resp
.found
== 1)
209 /* A first check whether the buffer is sufficiently large is possible. */
210 /* Now allocate the buffer the array for the group members. We must
211 align the pointer and the base of the h_addr_list pointers. */
212 align1
= ((__alignof__ (char *) - (cp
- ((char *) 0)))
213 & (__alignof__ (char *) - 1));
214 align2
= ((__alignof__ (char *) - ((cp
+ align1
+ hst_resp
.h_name_len
)
216 & (__alignof__ (char *) - 1));
217 if (buflen
< (align1
+ hst_resp
.h_name_len
+ align2
218 + ((hst_resp
.h_aliases_cnt
+ hst_resp
.h_addr_list_cnt
221 + hst_resp
.h_addr_list_cnt
* (type
== AF_INET
222 ? INADDRSZ
: IN6ADDRSZ
)))
225 *h_errnop
= NETDB_INTERNAL
;
226 __set_errno (ERANGE
);
232 /* Prepare the result as far as we can. */
233 resultbuf
->h_aliases
= (char **) cp
;
234 cp
+= (hst_resp
.h_aliases_cnt
+ 1) * sizeof (char *);
235 resultbuf
->h_addr_list
= (char **) cp
;
236 cp
+= (hst_resp
.h_addr_list_cnt
+ 1) * sizeof (char *);
238 resultbuf
->h_name
= cp
;
239 cp
+= hst_resp
.h_name_len
+ align2
;
241 if (type
== GETHOSTBYADDR
|| type
== GETHOSTBYNAME
)
243 resultbuf
->h_addrtype
= AF_INET
;
244 resultbuf
->h_length
= INADDRSZ
;
248 resultbuf
->h_addrtype
= AF_INET6
;
249 resultbuf
->h_length
= IN6ADDRSZ
;
251 for (cnt
= 0; cnt
< hst_resp
.h_addr_list_cnt
; ++cnt
)
253 resultbuf
->h_addr_list
[cnt
] = cp
;
254 cp
+= resultbuf
->h_length
;
256 resultbuf
->h_addr_list
[cnt
] = NULL
;
260 vec
[0].iov_base
= resultbuf
->h_name
;
261 vec
[0].iov_len
= hst_resp
.h_name_len
;
262 total_len
= hst_resp
.h_name_len
;
265 if (hst_resp
.h_aliases_cnt
> 0)
267 aliases_len
= alloca (hst_resp
.h_aliases_cnt
268 * sizeof (uint32_t));
269 vec
[n
].iov_base
= (void *) aliases_len
;
270 vec
[n
].iov_len
= hst_resp
.h_aliases_cnt
* sizeof (uint32_t);
272 total_len
+= hst_resp
.h_aliases_cnt
* sizeof (uint32_t);
276 if (type
== GETHOSTBYADDR
|| type
== GETHOSTBYNAME
)
278 vec
[n
].iov_base
= resultbuf
->h_addr_list
[0];
279 vec
[n
].iov_len
= hst_resp
.h_addr_list_cnt
* INADDRSZ
;
281 total_len
+= hst_resp
.h_addr_list_cnt
* INADDRSZ
;
287 if (hst_resp
.h_length
== INADDRSZ
)
289 ignore
= alloca (hst_resp
.h_addr_list_cnt
* INADDRSZ
);
290 vec
[n
].iov_base
= ignore
;
291 vec
[n
].iov_len
= hst_resp
.h_addr_list_cnt
* INADDRSZ
;
293 total_len
+= hst_resp
.h_addr_list_cnt
* INADDRSZ
;
298 vec
[n
].iov_base
= resultbuf
->h_addr_list
[0];
299 vec
[n
].iov_len
= hst_resp
.h_addr_list_cnt
* IN6ADDRSZ
;
301 total_len
+= hst_resp
.h_addr_list_cnt
* IN6ADDRSZ
;
306 if ((size_t) __readvall (sock
, vec
, n
) != total_len
)
311 memcpy (resultbuf
->h_name
, h_name
, hst_resp
.h_name_len
);
312 memcpy (resultbuf
->h_addr_list
[0], addr_list
, addr_list_len
);
315 /* Now we also can read the aliases. */
317 for (cnt
= 0; cnt
< hst_resp
.h_aliases_cnt
; ++cnt
)
319 resultbuf
->h_aliases
[cnt
] = cp
;
320 cp
+= aliases_len
[cnt
];
321 total_len
+= aliases_len
[cnt
];
323 resultbuf
->h_aliases
[cnt
] = NULL
;
325 if (__builtin_expect ((const char *) addr_list
+ addr_list_len
326 + total_len
> recend
, 0))
328 /* aliases_len array might contain garbage during nscd GC cycle,
329 retry rather than fail in that case. */
330 if (addr_list
!= NULL
&& mapped
->head
->gc_cycle
!= gc_cycle
)
334 /* See whether this would exceed the buffer capacity. */
335 if (__builtin_expect (cp
> buffer
+ buflen
, 0))
337 /* aliases_len array might contain garbage during nscd GC cycle,
338 retry rather than fail in that case. */
339 if (addr_list
!= NULL
&& mapped
->head
->gc_cycle
!= gc_cycle
)
347 /* And finally read the aliases. */
348 if (addr_list
== NULL
)
351 || ((size_t) __readall (sock
, resultbuf
->h_aliases
[0], total_len
)
360 memcpy (resultbuf
->h_aliases
[0],
361 (const char *) addr_list
+ addr_list_len
, total_len
);
363 /* Try to detect corrupt databases. */
364 if (resultbuf
->h_name
[hst_resp
.h_name_len
- 1] != '\0'
365 || ({for (cnt
= 0; cnt
< hst_resp
.h_aliases_cnt
; ++cnt
)
366 if (resultbuf
->h_aliases
[cnt
][aliases_len
[cnt
] - 1]
369 cnt
< hst_resp
.h_aliases_cnt
; }))
371 /* We cannot use the database. */
372 if (mapped
->head
->gc_cycle
!= gc_cycle
)
383 /* Store the error number. */
384 *h_errnop
= hst_resp
.error
;
386 /* The `errno' to some value != ERANGE. */
387 __set_errno (ENOENT
);
388 /* Even though we have not found anything, the result is zero. */
394 close_not_cancel_no_status (sock
);
396 if (__nscd_drop_map_ref (mapped
, &gc_cycle
) != 0)
398 /* When we come here this means there has been a GC cycle while we
399 were looking for the data. This means the data might have been
400 inconsistent. Retry if possible. */
401 if ((gc_cycle
& 1) != 0 || ++nretries
== 5 || retval
== -1)
403 /* nscd is just running gc now. Disable using the mapping. */
404 if (atomic_decrement_val (&mapped
->counter
) == 0)
405 __nscd_unmap (mapped
);