source: rtems-libbsd/freebsd/lib/libc/resolv/res_mkquery.c @ bceabc9

4.1155-freebsd-126-freebsd-12freebsd-9.3
Last change on this file since bceabc9 was bceabc9, checked in by Sebastian Huber <sebastian.huber@…>, on 10/09/13 at 20:42:09

Move files to match FreeBSD layout
  • Property mode set to 100644
File size: 8.6 KB
Line 
1#include "port_before.h"
2
3/*
4 * Copyright (c) 1985, 1993
5 *    The Regents of the University of California.  All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 * 4. Neither the name of the University nor the names of its contributors
16 *    may be used to endorse or promote products derived from this software
17 *    without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32/*
33 * Portions Copyright (c) 1993 by Digital Equipment Corporation.
34 *
35 * Permission to use, copy, modify, and distribute this software for any
36 * purpose with or without fee is hereby granted, provided that the above
37 * copyright notice and this permission notice appear in all copies, and that
38 * the name of Digital Equipment Corporation not be used in advertising or
39 * publicity pertaining to distribution of the document or software without
40 * specific, written prior permission.
41 *
42 * THE SOFTWARE IS PROVIDED "AS IS" AND DIGITAL EQUIPMENT CORP. DISCLAIMS ALL
43 * WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES
44 * OF MERCHANTABILITY AND FITNESS.   IN NO EVENT SHALL DIGITAL EQUIPMENT
45 * CORPORATION BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
46 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
47 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
48 * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
49 * SOFTWARE.
50 */
51
52/*
53 * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
54 * Portions Copyright (c) 1996-1999 by Internet Software Consortium.
55 *
56 * Permission to use, copy, modify, and distribute this software for any
57 * purpose with or without fee is hereby granted, provided that the above
58 * copyright notice and this permission notice appear in all copies.
59 *
60 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
61 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
62 * MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR
63 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
64 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
65 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
66 * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
67 */
68
69#if defined(LIBC_SCCS) && !defined(lint)
70static const char sccsid[] = "@(#)res_mkquery.c 8.1 (Berkeley) 6/4/93";
71static const char rcsid[] = "$Id: res_mkquery.c,v 1.5.18.2 2008/04/03 23:15:15 marka Exp $";
72#endif /* LIBC_SCCS and not lint */
73#include <sys/cdefs.h>
74__FBSDID("$FreeBSD$");
75
76#include "port_before.h"
77#include <sys/types.h>
78#include <sys/param.h>
79#include <netinet/in.h>
80#include <arpa/nameser.h>
81#include <netdb.h>
82#include <resolv.h>
83#include <stdio.h>
84#include <string.h>
85#include "port_after.h"
86
87/* Options.  Leave them on. */
88#define DEBUG
89
90extern const char *_res_opcodes[];
91
92/*%
93 * Form all types of queries.
94 * Returns the size of the result or -1.
95 */
96int
97res_nmkquery(res_state statp,
98             int op,                    /*!< opcode of query  */
99             const char *dname,         /*!< domain name  */
100             int class, int type,       /*!< class and type of query  */
101             const u_char *data,        /*!< resource record data  */
102             int datalen,               /*!< length of data  */
103             const u_char *newrr_in,    /*!< new rr for modify or append  */
104             u_char *buf,               /*!< buffer to put query  */
105             int buflen)                /*!< size of buffer  */
106{
107        HEADER *hp;
108        u_char *cp, *ep;
109        int n;
110        u_char *dnptrs[20], **dpp, **lastdnptr;
111
112        UNUSED(newrr_in);
113
114#ifdef DEBUG
115        if (statp->options & RES_DEBUG)
116                printf(";; res_nmkquery(%s, %s, %s, %s)\n",
117                       _res_opcodes[op], dname, p_class(class), p_type(type));
118#endif
119        /*
120         * Initialize header fields.
121         */
122        if ((buf == NULL) || (buflen < HFIXEDSZ))
123                return (-1);
124        memset(buf, 0, HFIXEDSZ);
125        hp = (HEADER *) buf;
126        hp->id = htons(++statp->id);
127        hp->opcode = op;
128        hp->rd = (statp->options & RES_RECURSE) != 0U;
129        hp->rcode = NOERROR;
130        cp = buf + HFIXEDSZ;
131        ep = buf + buflen;
132        dpp = dnptrs;
133        *dpp++ = buf;
134        *dpp++ = NULL;
135        lastdnptr = dnptrs + sizeof dnptrs / sizeof dnptrs[0];
136        /*
137         * perform opcode specific processing
138         */
139        switch (op) {
140        case QUERY:     /*FALLTHROUGH*/
141        case NS_NOTIFY_OP:
142                if (ep - cp < QFIXEDSZ)
143                        return (-1);
144                if ((n = dn_comp(dname, cp, ep - cp - QFIXEDSZ, dnptrs,
145                    lastdnptr)) < 0)
146                        return (-1);
147                cp += n;
148                ns_put16(type, cp);
149                cp += INT16SZ;
150                ns_put16(class, cp);
151                cp += INT16SZ;
152                hp->qdcount = htons(1);
153                if (op == QUERY || data == NULL)
154                        break;
155                /*
156                 * Make an additional record for completion domain.
157                 */
158                if ((ep - cp) < RRFIXEDSZ)
159                        return (-1);
160                n = dn_comp((const char *)data, cp, ep - cp - RRFIXEDSZ,
161                            dnptrs, lastdnptr);
162                if (n < 0)
163                        return (-1);
164                cp += n;
165                ns_put16(T_NULL, cp);
166                cp += INT16SZ;
167                ns_put16(class, cp);
168                cp += INT16SZ;
169                ns_put32(0, cp);
170                cp += INT32SZ;
171                ns_put16(0, cp);
172                cp += INT16SZ;
173                hp->arcount = htons(1);
174                break;
175
176        case IQUERY:
177                /*
178                 * Initialize answer section
179                 */
180                if (ep - cp < 1 + RRFIXEDSZ + datalen)
181                        return (-1);
182                *cp++ = '\0';   /*%< no domain name */
183                ns_put16(type, cp);
184                cp += INT16SZ;
185                ns_put16(class, cp);
186                cp += INT16SZ;
187                ns_put32(0, cp);
188                cp += INT32SZ;
189                ns_put16(datalen, cp);
190                cp += INT16SZ;
191                if (datalen) {
192                        memcpy(cp, data, datalen);
193                        cp += datalen;
194                }
195                hp->ancount = htons(1);
196                break;
197
198        default:
199                return (-1);
200        }
201        return (cp - buf);
202}
203
204#ifdef RES_USE_EDNS0
205/* attach OPT pseudo-RR, as documented in RFC2671 (EDNS0). */
206
207int
208res_nopt(res_state statp,
209         int n0,                /*%< current offset in buffer */
210         u_char *buf,           /*%< buffer to put query */
211         int buflen,            /*%< size of buffer */
212         int anslen)            /*%< UDP answer buffer size */
213{
214        HEADER *hp;
215        u_char *cp, *ep;
216        u_int16_t flags = 0;
217
218#ifdef DEBUG
219        if ((statp->options & RES_DEBUG) != 0U)
220                printf(";; res_nopt()\n");
221#endif
222
223        hp = (HEADER *) buf;
224        cp = buf + n0;
225        ep = buf + buflen;
226
227        if ((ep - cp) < 1 + RRFIXEDSZ)
228                return (-1);
229
230        *cp++ = 0;                              /*%< "." */
231        ns_put16(ns_t_opt, cp);                 /*%< TYPE */
232        cp += INT16SZ;
233        if (anslen > 0xffff)
234                anslen = 0xffff;                /* limit to 16bit value */
235        ns_put16(anslen & 0xffff, cp);          /*%< CLASS = UDP payload size */
236        cp += INT16SZ;
237        *cp++ = NOERROR;                        /*%< extended RCODE */
238        *cp++ = 0;                              /*%< EDNS version */
239
240        if (statp->options & RES_USE_DNSSEC) {
241#ifdef DEBUG
242                if (statp->options & RES_DEBUG)
243                        printf(";; res_opt()... ENDS0 DNSSEC\n");
244#endif
245                flags |= NS_OPT_DNSSEC_OK;
246        }
247        ns_put16(flags, cp);
248        cp += INT16SZ;
249
250        ns_put16(0U, cp);                       /*%< RDLEN */
251        cp += INT16SZ;
252
253        hp->arcount = htons(ntohs(hp->arcount) + 1);
254
255        return (cp - buf);
256}
257
258/*
259 * Construct variable data (RDATA) block for OPT psuedo-RR, append it
260 * to the buffer, then update the RDLEN field (previously set to zero by
261 * res_nopt()) with the new RDATA length.
262 */
263int
264res_nopt_rdata(res_state statp,
265          int n0,               /*%< current offset in buffer */
266          u_char *buf,          /*%< buffer to put query */
267          int buflen,           /*%< size of buffer */
268          u_char *rdata,        /*%< ptr to start of opt rdata */
269          u_short code,         /*%< OPTION-CODE */
270          u_short len,          /*%< OPTION-LENGTH */
271          u_char *data)         /*%< OPTION_DATA */
272{
273        register u_char *cp, *ep;
274
275#ifdef DEBUG
276        if ((statp->options & RES_DEBUG) != 0U)
277                printf(";; res_nopt_rdata()\n");
278#endif
279
280        cp = buf + n0;
281        ep = buf + buflen;
282
283        if ((ep - cp) < (4 + len))
284                return (-1);
285
286        if (rdata < (buf + 2) || rdata >= ep)
287                return (-1);
288
289        ns_put16(code, cp);
290        cp += INT16SZ;
291
292        ns_put16(len, cp);
293        cp += INT16SZ;
294
295        memcpy(cp, data, len);
296        cp += len;
297
298        len = cp - rdata;
299        ns_put16(len, rdata - 2);       /* Update RDLEN field */
300
301        return (cp - buf);
302}
303#endif
304
305/*! \file */
Note: See TracBrowser for help on using the repository browser.