contrib/isc-dhcp/minires/res_mkquery.c

   1 /*
   2  * Copyright (c) 1985, 1993
   3  *    The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by the University of
  16  *      California, Berkeley and its contributors.
  17  * 4. Neither the name of the University nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 /*
  35  * Portions Copyright (c) 1993 by Digital Equipment Corporation.
  36  *
  37  * Permission to use, copy, modify, and distribute this software for any
  38  * purpose with or without fee is hereby granted, provided that the above
  39  * copyright notice and this permission notice appear in all copies, and that
  40  * the name of Digital Equipment Corporation not be used in advertising or
  41  * publicity pertaining to distribution of the document or software without
  42  * specific, written prior permission.
  43  *
  44  * THE SOFTWARE IS PROVIDED "AS IS" AND DIGITAL EQUIPMENT CORP. DISCLAIMS ALL
  45  * WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES
  46  * OF MERCHANTABILITY AND FITNESS.   IN NO EVENT SHALL DIGITAL EQUIPMENT
  47  * CORPORATION BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
  48  * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
  49  * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
  50  * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
  51  * SOFTWARE.
  52  */
  53
  54 /*
  55  * Portions Copyright (c) 1996-1999 by Internet Software Consortium.
  56  *
  57  * Permission to use, copy, modify, and distribute this software for any
  58  * purpose with or without fee is hereby granted, provided that the above
  59  * copyright notice and this permission notice appear in all copies.
  60  *
  61  * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS
  62  * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
  63  * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE
  64  * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
  65  * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
  66  * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
  67  * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
  68  * SOFTWARE.
  69  */
  70
  71 #if defined(LIBC_SCCS) && !defined(lint)
  72 static const char sccsid[] = "@(#)res_mkquery.c 8.1 (Berkeley) 6/4/93";
  73 static const char rcsid[] = "$Id: res_mkquery.c,v 1.4 2001/01/16 22:33:14 mellon Exp $";
  74 #endif /* LIBC_SCCS and not lint */
  75
  76 #include <sys/types.h>
  77 #include <sys/param.h>
  78 #include <netinet/in.h>
  79 #include <netdb.h>
  80 #include <stdio.h>
  81 #include <string.h>
  82 #include <sys/socket.h>
  83
  84 #include "minires/minires.h"
  85 #include "arpa/nameser.h"
  86
  87 extern const char *_res_opcodes[];
  88
  89 /*
  90  * Form all types of queries.
  91  * Returns the size of the result or -1.
  92  */
  93 isc_result_t
  94 res_nmkquery(res_state statp,
  95              int op,                    /* opcode of query */
  96              const char *dname,         /* domain name */
  97              ns_class class, ns_type type,      /* class and type of query */
  98              const u_char *data,        /* resource record data */
  99              unsigned datalen,          /* length of data */
 100              const u_char *newrr_in,    /* new rr for modify or append */
 101              double *buf,               /* buffer to put query */
 102              unsigned buflen,           /* size of buffer */
 103              unsigned *rbuflen)         /* returned size of buffer */
 104 {
 105         register HEADER *hp;
 106         register u_char *cp;
 107         register int n;
 108         u_char *dnptrs[20], **dpp, **lastdnptr;
 109
 110         /*
 111          * Initialize header fields.
 112          */
 113         if ((buf == NULL) || (buflen < HFIXEDSZ))
 114                 return ISC_R_INVALIDARG;
 115         memset(buf, 0, HFIXEDSZ);
 116         hp = (HEADER *) buf;
 117         hp->id = htons(++statp->id);
 118         hp->opcode = op;
 119         hp->rd = (statp->options & RES_RECURSE) != 0;
 120         hp->rcode = NOERROR;
 121         cp = ((u_char *)buf) + HFIXEDSZ;
 122         buflen -= HFIXEDSZ;
 123         dpp = dnptrs;
 124         *dpp++ = (u_char *)buf;
 125         *dpp++ = NULL;
 126         lastdnptr = dnptrs + sizeof dnptrs / sizeof dnptrs[0];
 127         /*
 128          * perform opcode specific processing
 129          */
 130         switch (op) {
 131         case QUERY:     /*FALLTHROUGH*/
 132         case NS_NOTIFY_OP:
 133                 if ((buflen -= QFIXEDSZ) < 0)
 134                         return ISC_R_NOSPACE;
 135                 if ((n = dn_comp(dname, cp, buflen, dnptrs, lastdnptr)) < 0)
 136                         return ISC_R_NOSPACE;
 137                 cp += n;
 138                 buflen -= n;
 139                 putUShort(cp, type);
 140                 cp += INT16SZ;
 141                 putUShort(cp, class);
 142                 cp += INT16SZ;
 143                 hp->qdcount = htons(1);
 144                 if (op == QUERY || data == NULL)
 145                         break;
 146                 /*
 147                  * Make an additional record for completion domain.
 148                  */
 149                 buflen -= RRFIXEDSZ;
 150                 n = dn_comp((const char *)data, cp, buflen, dnptrs, lastdnptr);
 151                 if (n < 0)
 152                         return ISC_R_NOSPACE;
 153                 cp += n;
 154                 buflen -= n;
 155                 putUShort(cp, T_NULL);
 156                 cp += INT16SZ;
 157                 putUShort(cp, class);
 158                 cp += INT16SZ;
 159                 putULong(cp, 0);
 160                 cp += INT32SZ;
 161                 putUShort(cp, 0);
 162                 cp += INT16SZ;
 163                 hp->arcount = htons(1);
 164                 break;
 165
 166         case IQUERY:
 167                 /*
 168                  * Initialize answer section
 169                  */
 170                 if (buflen < 1 + RRFIXEDSZ + datalen)
 171                         return ISC_R_NOSPACE;
 172                 *cp++ = '\0';   /* no domain name */
 173                 putUShort(cp, type);
 174                 cp += INT16SZ;
 175                 putUShort(cp, class);
 176                 cp += INT16SZ;
 177                 putULong(cp, 0);
 178                 cp += INT32SZ;
 179                 putUShort(cp, datalen);
 180                 cp += INT16SZ;
 181                 if (datalen) {
 182                         memcpy(cp, data, datalen);
 183                         cp += datalen;
 184                 }
 185                 hp->ancount = htons(1);
 186                 break;
 187
 188         default:
 189                 return ISC_R_NOTIMPLEMENTED;
 190         }
 191         *rbuflen = cp - ((u_char *)buf);
 192         return ISC_R_SUCCESS;
 193 }