(getanswer_r): Correctly track usage of user-provided buffer.
[kopensolaris-gnu/glibc.git] / resolv / getnetnamadr.c
1 /* Copyright (c) 1993 Carlos Leandro and Rui Salgueiro
2  *      Dep. Matematica Universidade de Coimbra, Portugal, Europe
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  */
8 /*
9  * Copyright (c) 1983, 1993
10  *      The Regents of the University of California.  All rights reserved.
11  *
12  * Redistribution and use in source and binary forms, with or without
13  * modification, are permitted provided that the following conditions
14  * are met:
15  * 1. Redistributions of source code must retain the above copyright
16  *    notice, this list of conditions and the following disclaimer.
17  * 2. Redistributions in binary form must reproduce the above copyright
18  *    notice, this list of conditions and the following disclaimer in the
19  *    documentation and/or other materials provided with the distribution.
20  * 4. Neither the name of the University nor the names of its contributors
21  *    may be used to endorse or promote products derived from this software
22  *    without specific prior written permission.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34  * SUCH DAMAGE.
35  */
36
37 #if defined(LIBC_SCCS) && !defined(lint)
38 static char sccsid[] = "@(#)getnetbyaddr.c      8.1 (Berkeley) 6/4/93";
39 static char sccsid_[] = "from getnetnamadr.c    1.4 (Coimbra) 93/06/03";
40 static char rcsid[] = "$Id$";
41 #endif /* LIBC_SCCS and not lint */
42
43 #include <sys/types.h>
44 #include <sys/param.h>
45 #include <sys/socket.h>
46 #include <netinet/in.h>
47 #include <arpa/inet.h>
48 #include <arpa/nameser.h>
49
50 #include <stdio.h>
51 #include <netdb.h>
52 #include <resolv.h>
53 #include <ctype.h>
54 #include <errno.h>
55 #include <string.h>
56
57 #ifndef h_errno
58 extern int h_errno;
59 #endif
60
61 #if defined(mips) && defined(SYSTYPE_BSD43) && !defined(errno)
62 extern int errno;
63 #endif
64
65 struct netent *_getnetbyaddr __P((long net, int type));
66 struct netent *_getnetbyname __P((const char *name));
67
68 #define BYADDR 0
69 #define BYNAME 1
70 #define MAXALIASES      35
71
72 #if PACKETSZ > 1024
73 #define MAXPACKET       PACKETSZ
74 #else
75 #define MAXPACKET       1024
76 #endif
77
78 typedef union {
79         HEADER  hdr;
80         u_char  buf[MAXPACKET];
81 } querybuf;
82
83 typedef union {
84         long    al;
85         char    ac;
86 } align;
87
88 static struct netent *
89 getnetanswer(answer, anslen, net_i)
90         querybuf *answer;
91         int anslen;
92         int net_i;
93 {
94
95         register HEADER *hp;
96         register u_char *cp;
97         register int n;
98         u_char *eom;
99         int type, class, buflen, ancount, qdcount, haveanswer, i, nchar;
100         char aux1[30], aux2[30], ans[30], *in, *st, *pauxt, *bp, **ap,
101                 *paux1 = &aux1[0], *paux2 = &aux2[0], flag = 0;
102 static  struct netent net_entry;
103 static  char *net_aliases[MAXALIASES], netbuf[BUFSIZ+1];
104
105         /*
106          * find first satisfactory answer
107          *
108          *      answer --> +------------+  ( MESSAGE )
109          *                 |   Header   |
110          *                 +------------+
111          *                 |  Question  | the question for the name server
112          *                 +------------+
113          *                 |   Answer   | RRs answering the question
114          *                 +------------+
115          *                 | Authority  | RRs pointing toward an authority
116          *                 | Additional | RRs holding additional information
117          *                 +------------+
118          */
119         eom = answer->buf + anslen;
120         hp = &answer->hdr;
121         ancount = ntohs(hp->ancount); /* #/records in the answer section */
122         qdcount = ntohs(hp->qdcount); /* #/entries in the question section */
123         bp = netbuf;
124         buflen = sizeof(netbuf);
125         cp = answer->buf + HFIXEDSZ;
126         if (!qdcount) {
127                 if (hp->aa)
128                         __set_h_errno (HOST_NOT_FOUND);
129                 else
130                         __set_h_errno (TRY_AGAIN);
131                 return (NULL);
132         }
133         while (qdcount-- > 0)
134                 cp += __dn_skipname(cp, eom) + QFIXEDSZ;
135         ap = net_aliases;
136         *ap = NULL;
137         net_entry.n_aliases = net_aliases;
138         haveanswer = 0;
139         while (--ancount >= 0 && cp < eom) {
140                 n = dn_expand(answer->buf, eom, cp, bp, buflen);
141                 if ((n < 0) || !res_dnok(bp))
142                         break;
143                 cp += n;
144                 ans[0] = '\0';
145                 (void)strcpy(&ans[0], bp);
146                 GETSHORT(type, cp);
147                 GETSHORT(class, cp);
148                 cp += INT32SZ;          /* TTL */
149                 GETSHORT(n, cp);
150                 if (class == C_IN && type == T_PTR) {
151                         n = dn_expand(answer->buf, eom, cp, bp, buflen);
152                         if ((n < 0) || !res_hnok(bp)) {
153                                 cp += n;
154                                 return (NULL);
155                         }
156                         cp += n;
157                         *ap++ = bp;
158                         bp += strlen(bp) + 1;
159                         net_entry.n_addrtype =
160                                 (class == C_IN) ? AF_INET : AF_UNSPEC;
161                         haveanswer++;
162                 }
163         }
164         if (haveanswer) {
165                 *ap = NULL;
166                 switch (net_i) {
167                 case BYADDR:
168                         net_entry.n_name = *net_entry.n_aliases;
169                         net_entry.n_net = 0L;
170                         break;
171                 case BYNAME:
172                         in = *net_entry.n_aliases;
173                         net_entry.n_name = &ans[0];
174                         aux2[0] = '\0';
175                         for (i = 0; i < 4; i++) {
176                                 for (st = in, nchar = 0;
177                                      *st != '.';
178                                      st++, nchar++)
179                                         ;
180                                 if (nchar != 1 || *in != '0' || flag) {
181                                         flag = 1;
182                                         (void)strncpy(paux1,
183                                                       (i==0) ? in : in-1,
184                                                       (i==0) ?nchar : nchar+1);
185                                         paux1[(i==0) ? nchar : nchar+1] = '\0';
186                                         pauxt = paux2;
187                                         paux2 = strcat(paux1, paux2);
188                                         paux1 = pauxt;
189                                 }
190                                 in = ++st;
191                         }
192                         net_entry.n_net = inet_network(paux2);
193                         break;
194                 }
195                 net_entry.n_aliases++;
196                 return (&net_entry);
197         }
198         __set_h_errno (TRY_AGAIN);
199         return (NULL);
200 }
201
202 struct netent *
203 getnetbyaddr(net, net_type)
204         register u_long net;
205         register int net_type;
206 {
207         unsigned int netbr[4];
208         int nn, anslen;
209         querybuf buf;
210         char qbuf[MAXDNAME];
211         u_int32_t net2;         /* Changed from unsigned long --roland */
212         struct netent *net_entry;
213
214         if (net_type != AF_INET)
215                 return (_getnetbyaddr(net, net_type));
216
217         for (nn = 4, net2 = net; net2; net2 >>= 8)
218                 netbr[--nn] = net2 & 0xff;
219         switch (nn) {
220         case 3:         /* Class A */
221                 sprintf(qbuf, "0.0.0.%u.in-addr.arpa", netbr[3]);
222                 break;
223         case 2:         /* Class B */
224                 sprintf(qbuf, "0.0.%u.%u.in-addr.arpa", netbr[3], netbr[2]);
225                 break;
226         case 1:         /* Class C */
227                 sprintf(qbuf, "0.%u.%u.%u.in-addr.arpa", netbr[3], netbr[2],
228                     netbr[1]);
229                 break;
230         case 0:         /* Class D - E */
231                 sprintf(qbuf, "%u.%u.%u.%u.in-addr.arpa", netbr[3], netbr[2],
232                     netbr[1], netbr[0]);
233                 break;
234         }
235         anslen = res_query(qbuf, C_IN, T_PTR, (u_char *)&buf, sizeof(buf));
236         if (anslen < 0) {
237 #ifdef DEBUG
238                 if (_res.options & RES_DEBUG)
239                         printf("res_query failed\n");
240 #endif
241                 if (errno == ECONNREFUSED)
242                         return (_getnetbyaddr(net, net_type));
243                 return (NULL);
244         }
245         net_entry = getnetanswer(&buf, anslen, BYADDR);
246         if (net_entry) {
247                 unsigned u_net = net;   /* maybe net should be unsigned ? */
248
249                 /* Strip trailing zeros */
250                 while ((u_net & 0xff) == 0 && u_net != 0)
251                         u_net >>= 8;
252                 net_entry->n_net = u_net;
253                 return (net_entry);
254         }
255         return (_getnetbyaddr(net, net_type));
256 }
257
258 struct netent *
259 getnetbyname(net)
260         register const char *net;
261 {
262         int anslen;
263         querybuf buf;
264         char qbuf[MAXDNAME];
265         struct netent *net_entry;
266
267         if ((_res.options & RES_INIT) == 0 && res_init() == -1) {
268                 __set_h_errno (NETDB_INTERNAL);
269                 return (NULL);
270         }
271         strcpy(&qbuf[0], net);
272         anslen = res_search(qbuf, C_IN, T_PTR, (u_char *)&buf, sizeof(buf));
273         if (anslen < 0) {
274 #ifdef DEBUG
275                 if (_res.options & RES_DEBUG)
276                         printf("res_query failed\n");
277 #endif
278                 if (errno == ECONNREFUSED)
279                         return (_getnetbyname(net));
280                 return (_getnetbyname(net));
281         }
282         net_entry = getnetanswer(&buf, anslen, BYNAME);
283         if (net_entry)
284                 return (net_entry);
285         return (_getnetbyname(net));
286 }