1/* Copyright (C) 1996-2017 Free Software Foundation, Inc.
2 This file is part of the GNU C Library.
3
4 The GNU C Library is free software; you can redistribute it and/or
5 modify it under the terms of the GNU Lesser General Public
6 License as published by the Free Software Foundation; either
7 version 2.1 of the License, or (at your option) any later version.
8
9 The GNU C Library is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 Lesser General Public License for more details.
13
14 You should have received a copy of the GNU Lesser General Public
15 License along with the GNU C Library; if not, see
16 <http://www.gnu.org/licenses/>. */
17
18#include <assert.h>
19#include <errno.h>
20#include <libc-lock.h>
21#include <stdlib.h>
22#include <resolv.h>
23
24#include "nsswitch.h"
25
26/*******************************************************************\
27|* Here we assume several symbols to be defined: *|
28|* *|
29|* LOOKUP_TYPE - the return type of the function *|
30|* *|
31|* FUNCTION_NAME - name of the non-reentrant function *|
32|* *|
33|* DATABASE_NAME - name of the database the function accesses *|
34|* (e.g., host, services, ...) *|
35|* *|
36|* ADD_PARAMS - additional parameter, can vary in number *|
37|* *|
38|* ADD_VARIABLES - names of additional parameter *|
39|* *|
40|* BUFLEN - length of buffer allocated for the non *|
41|* reentrant version *|
42|* *|
43|* Optionally the following vars can be defined: *|
44|* *|
45|* NEED_H_ERRNO - an extra parameter will be passed to point to *|
46|* the global `h_errno' variable. *|
47|* *|
48\*******************************************************************/
49
50
51#ifdef HANDLE_DIGITS_DOTS
52# include <resolv/resolv_context.h>
53#endif
54
55/* To make the real sources a bit prettier. */
56#define REENTRANT_NAME APPEND_R (FUNCTION_NAME)
57#define APPEND_R(name) APPEND_R1 (name)
58#define APPEND_R1(name) name##_r
59#define INTERNAL(name) INTERNAL1 (name)
60#define INTERNAL1(name) __##name
61
62/* Sometimes we need to store error codes in the `h_errno' variable. */
63#ifdef NEED_H_ERRNO
64# define H_ERRNO_PARM , int *h_errnop
65# define H_ERRNO_VAR , &h_errno_tmp
66# define H_ERRNO_VAR_P &h_errno_tmp
67#else
68# define H_ERRNO_PARM
69# define H_ERRNO_VAR
70# define H_ERRNO_VAR_P NULL
71#endif
72
73#ifdef HAVE_AF
74# define AF_VAL af
75#else
76# define AF_VAL AF_INET
77#endif
78
79/* Prototype for reentrant version we use here. */
80extern int INTERNAL (REENTRANT_NAME) (ADD_PARAMS, LOOKUP_TYPE *resbuf,
81 char *buffer, size_t buflen,
82 LOOKUP_TYPE **result H_ERRNO_PARM);
83
84/* We need to protect the dynamic buffer handling. */
85__libc_lock_define_initialized (static, lock);
86
87/* This points to the static buffer used. */
88libc_freeres_ptr (static char *buffer);
89
90
91LOOKUP_TYPE *
92FUNCTION_NAME (ADD_PARAMS)
93{
94 static size_t buffer_size;
95 static LOOKUP_TYPE resbuf;
96 LOOKUP_TYPE *result;
97#ifdef NEED_H_ERRNO
98 int h_errno_tmp = 0;
99#endif
100
101#ifdef HANDLE_DIGITS_DOTS
102 /* Wrap both __nss_hostname_digits_dots and the actual lookup
103 function call in the same context. */
104 struct resolv_context *res_ctx = __resolv_context_get ();
105 if (res_ctx == NULL)
106 {
107# if NEED_H_ERRNO
108 __set_h_errno (NETDB_INTERNAL);
109# endif
110 return NULL;
111 }
112#endif
113
114 /* Get lock. */
115 __libc_lock_lock (lock);
116
117 if (buffer == NULL)
118 {
119 buffer_size = BUFLEN;
120 buffer = (char *) malloc (buffer_size);
121 }
122
123#ifdef HANDLE_DIGITS_DOTS
124 if (buffer != NULL)
125 {
126 if (__nss_hostname_digits_dots_context
127 (res_ctx, name, &resbuf, &buffer, &buffer_size, 0, &result, NULL,
128 AF_VAL, H_ERRNO_VAR_P))
129 goto done;
130 }
131#endif
132
133 while (buffer != NULL
134 && (INTERNAL (REENTRANT_NAME) (ADD_VARIABLES, &resbuf, buffer,
135 buffer_size, &result H_ERRNO_VAR)
136 == ERANGE)
137#ifdef NEED_H_ERRNO
138 && h_errno_tmp == NETDB_INTERNAL
139#endif
140 )
141 {
142 char *new_buf;
143 buffer_size *= 2;
144 new_buf = (char *) realloc (buffer, buffer_size);
145 if (new_buf == NULL)
146 {
147 /* We are out of memory. Free the current buffer so that the
148 process gets a chance for a normal termination. */
149 free (buffer);
150 __set_errno (ENOMEM);
151 }
152 buffer = new_buf;
153 }
154
155 if (buffer == NULL)
156 result = NULL;
157
158#ifdef HANDLE_DIGITS_DOTS
159done:
160#endif
161 /* Release lock. */
162 __libc_lock_unlock (lock);
163
164#ifdef HANDLE_DIGITS_DOTS
165 __resolv_context_put (res_ctx);
166#endif
167
168#ifdef NEED_H_ERRNO
169 if (h_errno_tmp != 0)
170 __set_h_errno (h_errno_tmp);
171#endif
172
173 return result;
174}
175
176nss_interface_function (FUNCTION_NAME)
177