1/* Copyright (C) 1997-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/*
19 * Copyright (c) 1996,1999 by Internet Software Consortium.
20 *
21 * Permission to use, copy, modify, and distribute this software for any
22 * purpose with or without fee is hereby granted, provided that the above
23 * copyright notice and this permission notice appear in all copies.
24 *
25 * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS
26 * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
27 * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE
28 * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
29 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
30 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
31 * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
32 * SOFTWARE.
33 */
34
35/*
36 * This file is primarily maintained by <tytso@mit.edu> and <ghudson@mit.edu>.
37 */
38
39/*
40 * hesiod.c --- the core portion of the hesiod resolver.
41 *
42 * This file is derived from the hesiod library from Project Athena;
43 * It has been extensively rewritten by Theodore Ts'o to have a more
44 * thread-safe interface.
45 */
46
47/* Imports */
48
49#include <sys/types.h>
50#include <netinet/in.h>
51#include <arpa/nameser.h>
52
53#include <errno.h>
54#include <netdb.h>
55#include <resolv.h>
56#include <stdio.h>
57#include <stdlib.h>
58#include <string.h>
59
60#include "hesiod.h"
61#include "hesiod_p.h"
62
63#define _PATH_HESIOD_CONF "/etc/hesiod.conf"
64
65/* Forward */
66
67static int parse_config_file(struct hesiod_p *ctx, const char *filename);
68static char ** get_txt_records(struct hesiod_p *ctx, int class,
69 const char *name);
70
71/* Public */
72
73/*
74 * This function is called to initialize a hesiod_p.
75 */
76int
77hesiod_init(void **context) {
78 struct hesiod_p *ctx;
79 const char *configname;
80 char *cp;
81
82 ctx = malloc(sizeof(struct hesiod_p));
83 if (ctx == 0)
84 return (-1);
85
86 ctx->LHS = NULL;
87 ctx->RHS = NULL;
88 /* Set default query classes. */
89 ctx->classes[0] = C_IN;
90 ctx->classes[1] = C_HS;
91
92 configname = __libc_secure_getenv("HESIOD_CONFIG");
93 if (!configname)
94 configname = _PATH_HESIOD_CONF;
95 if (parse_config_file(ctx, configname) < 0) {
96 goto cleanup;
97 }
98 /*
99 * The default RHS can be overridden by an environment
100 * variable.
101 */
102 if ((cp = __libc_secure_getenv("HES_DOMAIN")) != NULL) {
103 free(ctx->RHS);
104 ctx->RHS = malloc(strlen(cp)+2);
105 if (!ctx->RHS)
106 goto cleanup;
107 if (cp[0] == '.')
108 strcpy(ctx->RHS, cp);
109 else {
110 ctx->RHS[0] = '.';
111 strcpy(ctx->RHS + 1, cp);
112 }
113 }
114
115 /*
116 * If there is no default hesiod realm set, we return an
117 * error.
118 */
119 if (!ctx->RHS) {
120 __set_errno(ENOEXEC);
121 goto cleanup;
122 }
123
124 *context = ctx;
125 return (0);
126
127 cleanup:
128 hesiod_end(ctx);
129 return (-1);
130}
131
132/*
133 * This function deallocates the hesiod_p
134 */
135void
136hesiod_end(void *context) {
137 struct hesiod_p *ctx = (struct hesiod_p *) context;
138 int save_errno = errno;
139
140 free(ctx->RHS);
141 free(ctx->LHS);
142 free(ctx);
143 __set_errno(save_errno);
144}
145
146/*
147 * This function takes a hesiod (name, type) and returns a DNS
148 * name which is to be resolved.
149 */
150char *
151hesiod_to_bind(void *context, const char *name, const char *type) {
152 struct hesiod_p *ctx = (struct hesiod_p *) context;
153 char *bindname;
154 char **rhs_list = NULL;
155 const char *RHS, *cp;
156 char *endp;
157
158 /* Decide what our RHS is, and set cp to the end of the actual name. */
159 if ((cp = strchr(name, '@')) != NULL) {
160 if (strchr(cp + 1, '.'))
161 RHS = cp + 1;
162 else if ((rhs_list = hesiod_resolve(context, cp + 1,
163 "rhs-extension")) != NULL)
164 RHS = *rhs_list;
165 else {
166 __set_errno(ENOENT);
167 return (NULL);
168 }
169 } else {
170 RHS = ctx->RHS;
171 cp = name + strlen(name);
172 }
173
174 /*
175 * Allocate the space we need, including up to three periods and
176 * the terminating NUL.
177 */
178 if ((bindname = malloc((cp - name) + strlen(type) + strlen(RHS) +
179 (ctx->LHS ? strlen(ctx->LHS) : 0) + 4)) == NULL) {
180 if (rhs_list)
181 hesiod_free_list(context, rhs_list);
182 return NULL;
183 }
184
185 /* Now put together the DNS name. */
186 endp = (char *) __mempcpy (bindname, name, cp - name);
187 *endp++ = '.';
188 endp = (char *) __stpcpy (endp, type);
189 if (ctx->LHS) {
190 if (ctx->LHS[0] != '.')
191 *endp++ = '.';
192 endp = __stpcpy (endp, ctx->LHS);
193 }
194 if (RHS[0] != '.')
195 *endp++ = '.';
196 strcpy (endp, RHS);
197
198 if (rhs_list)
199 hesiod_free_list(context, rhs_list);
200
201 return (bindname);
202}
203
204/*
205 * This is the core function. Given a hesiod (name, type), it
206 * returns an array of strings returned by the resolver.
207 */
208char **
209hesiod_resolve(void *context, const char *name, const char *type) {
210 struct hesiod_p *ctx = (struct hesiod_p *) context;
211 char *bindname = hesiod_to_bind(context, name, type);
212 char **retvec;
213
214 if (bindname == NULL)
215 return (NULL);
216
217 retvec = get_txt_records(ctx, ctx->classes[0], bindname);
218
219 if (retvec == NULL && (errno == ENOENT || errno == ECONNREFUSED) && ctx->classes[1])
220 retvec = get_txt_records(ctx, ctx->classes[1], bindname);
221
222
223 free(bindname);
224 return (retvec);
225}
226
227void
228hesiod_free_list(void *context, char **list) {
229 char **p;
230
231 for (p = list; *p; p++)
232 free(*p);
233 free(list);
234}
235
236/*
237 * This function parses the /etc/hesiod.conf file
238 */
239static int
240parse_config_file(struct hesiod_p *ctx, const char *filename) {
241 char buf[MAXDNAME+7];
242 FILE *fp;
243
244 /*
245 * Clear the existing configuration variable, just in case
246 * they're set.
247 */
248 free(ctx->RHS);
249 free(ctx->LHS);
250 ctx->RHS = ctx->LHS = 0;
251 /* Set default query classes. */
252 ctx->classes[0] = C_IN;
253 ctx->classes[1] = C_HS;
254
255 /*
256 * Now open and parse the file...
257 */
258 if (!(fp = fopen(filename, "rce")))
259 return (-1);
260
261 while (fgets(buf, sizeof(buf), fp) != NULL) {
262 char *key, *data, *cp, **cpp;
263
264 cp = buf;
265 if (*cp == '#' || *cp == '\n' || *cp == '\r')
266 continue;
267 while(*cp == ' ' || *cp == '\t')
268 cp++;
269 key = cp;
270 while(*cp != ' ' && *cp != '\t' && *cp != '=')
271 cp++;
272 *cp++ = '\0';
273
274 while(*cp == ' ' || *cp == '\t' || *cp == '=')
275 cp++;
276 data = cp;
277 while(*cp != ' ' && *cp != '\n' && *cp != '\r')
278 cp++;
279 *cp++ = '\0';
280
281 cpp = NULL;
282 if (strcasecmp(key, "lhs") == 0)
283 cpp = &ctx->LHS;
284 else if (strcasecmp(key, "rhs") == 0)
285 cpp = &ctx->RHS;
286 if (cpp) {
287 *cpp = strdup(data);
288 if (!*cpp)
289 goto cleanup;
290 } else if (strcasecmp(key, "classes") == 0) {
291 int n = 0;
292 while (*data && n < 2) {
293 cp = strchrnul(data, ',');
294 if (*cp != '\0')
295 *cp++ = '\0';
296 if (strcasecmp(data, "IN") == 0)
297 ctx->classes[n++] = C_IN;
298 else if (strcasecmp(data, "HS") == 0)
299 ctx->classes[n++] = C_HS;
300 data = cp;
301 }
302 if (n == 0) {
303 /* Restore the default. Better than
304 nother at all. */
305 ctx->classes[0] = C_IN;
306 ctx->classes[1] = C_HS;
307 } else if (n == 1
308 || ctx->classes[0] == ctx->classes[1])
309 ctx->classes[1] = 0;
310 }
311 }
312 fclose(fp);
313 return (0);
314
315 cleanup:
316 fclose(fp);
317 free(ctx->RHS);
318 free(ctx->LHS);
319 ctx->RHS = ctx->LHS = 0;
320 return (-1);
321}
322
323/*
324 * Given a DNS class and a DNS name, do a lookup for TXT records, and
325 * return a list of them.
326 */
327static char **
328get_txt_records(struct hesiod_p *ctx, int class, const char *name) {
329 struct {
330 int type; /* RR type */
331 int class; /* RR class */
332 int dlen; /* len of data section */
333 u_char *data; /* pointer to data */
334 } rr;
335 HEADER *hp;
336 u_char qbuf[MAX_HESRESP], abuf[MAX_HESRESP];
337 u_char *cp, *erdata, *eom;
338 char *dst, *edst, **list;
339 int ancount, qdcount;
340 int i, j, n, skip;
341
342 /*
343 * Construct the query and send it.
344 */
345 n = res_mkquery(QUERY, name, class, T_TXT, NULL, 0,
346 NULL, qbuf, MAX_HESRESP);
347 if (n < 0) {
348 __set_errno(EMSGSIZE);
349 return (NULL);
350 }
351 n = res_send(qbuf, n, abuf, MAX_HESRESP);
352 if (n < 0) {
353 __set_errno(ECONNREFUSED);
354 return (NULL);
355 }
356 if (n < HFIXEDSZ) {
357 __set_errno(EMSGSIZE);
358 return (NULL);
359 }
360
361 /*
362 * OK, parse the result.
363 */
364 hp = (HEADER *) abuf;
365 ancount = ntohs(hp->ancount);
366 qdcount = ntohs(hp->qdcount);
367 cp = abuf + sizeof(HEADER);
368 eom = abuf + n;
369
370 /* Skip query, trying to get to the answer section which follows. */
371 for (i = 0; i < qdcount; i++) {
372 skip = dn_skipname(cp, eom);
373 if (skip < 0 || cp + skip + QFIXEDSZ > eom) {
374 __set_errno(EMSGSIZE);
375 return (NULL);
376 }
377 cp += skip + QFIXEDSZ;
378 }
379
380 list = malloc((ancount + 1) * sizeof(char *));
381 if (!list)
382 return (NULL);
383 j = 0;
384 for (i = 0; i < ancount; i++) {
385 skip = dn_skipname(cp, eom);
386 if (skip < 0) {
387 __set_errno(EMSGSIZE);
388 goto cleanup;
389 }
390 cp += skip;
391 if (cp + 3 * INT16SZ + INT32SZ > eom) {
392 __set_errno(EMSGSIZE);
393 goto cleanup;
394 }
395 rr.type = ns_get16(cp);
396 cp += INT16SZ;
397 rr.class = ns_get16(cp);
398 cp += INT16SZ + INT32SZ; /* skip the ttl, too */
399 rr.dlen = ns_get16(cp);
400 cp += INT16SZ;
401 if (rr.dlen == 0 || cp + rr.dlen > eom) {
402 __set_errno(EMSGSIZE);
403 goto cleanup;
404 }
405 rr.data = cp;
406 cp += rr.dlen;
407 if (rr.class != class || rr.type != T_TXT)
408 continue;
409 if (!(list[j] = malloc(rr.dlen)))
410 goto cleanup;
411 dst = list[j++];
412 edst = dst + rr.dlen;
413 erdata = rr.data + rr.dlen;
414 cp = rr.data;
415 while (cp < erdata) {
416 n = (unsigned char) *cp++;
417 if (cp + n > eom || dst + n > edst) {
418 __set_errno(EMSGSIZE);
419 goto cleanup;
420 }
421 memcpy(dst, cp, n);
422 cp += n;
423 dst += n;
424 }
425 if (cp != erdata) {
426 __set_errno(EMSGSIZE);
427 goto cleanup;
428 }
429 *dst = '\0';
430 }
431 list[j] = NULL;
432 if (j == 0) {
433 __set_errno(ENOENT);
434 goto cleanup;
435 }
436 return (list);
437
438 cleanup:
439 for (i = 0; i < j; i++)
440 free(list[i]);
441 free(list);
442 return (NULL);
443}
444