1/*
2 * Copyright (C) 1995, 1996, 1997, and 1998 WIDE Project.
3 * 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. Neither the name of the project nor the names of its contributors
14 *    may be used to endorse or promote products derived from this software
15 *    without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 */
29
30/*
31 * Issues to be discussed:
32 * - Thread safe-ness must be checked
33 * - Return values.  There seems to be no standard for return value (RFC2133)
34 *   but INRIA implementation returns EAI_xxx defined for getaddrinfo().
35 */
36
37#include "ruby/config.h"
38#ifdef RUBY_EXTCONF_H
39#include RUBY_EXTCONF_H
40#endif
41#include <stdio.h>
42#include <sys/types.h>
43#ifndef _WIN32
44#if defined(__BEOS__) && !defined(__HAIKU__) && !defined(BONE)
45# include <net/socket.h>
46#else
47# include <sys/socket.h>
48#endif
49#include <netinet/in.h>
50#if defined(HAVE_ARPA_INET_H)
51#include <arpa/inet.h>
52#endif
53#if defined(HAVE_ARPA_NAMESER_H)
54#include <arpa/nameser.h>
55#endif
56#include <netdb.h>
57#if defined(HAVE_RESOLV_H)
58#include <resolv.h>
59#endif
60#endif
61#ifdef _WIN32
62#include <winsock2.h>
63#include <ws2tcpip.h>
64#define snprintf _snprintf
65#endif
66
67#include <string.h>
68#include <stddef.h>
69
70#ifdef SOCKS5
71#include <socks.h>
72#endif
73
74#include "addrinfo.h"
75#include "sockport.h"
76
77#define SUCCESS 0
78#define ANY 0
79#define YES 1
80#define NO  0
81
82struct sockinet {
83	u_char	si_len;
84	u_char	si_family;
85	u_short	si_port;
86};
87
88static struct afd {
89	int a_af;
90	int a_addrlen;
91	int a_socklen;
92	int a_off;
93} afdl [] = {
94#ifdef INET6
95#define N_INET6 0
96	{PF_INET6, sizeof(struct in6_addr),
97	 sizeof(struct sockaddr_in6),
98	 offsetof(struct sockaddr_in6, sin6_addr)},
99#define N_INET  1
100#else
101#define N_INET  0
102#endif
103	{PF_INET, sizeof(struct in_addr),
104	 sizeof(struct sockaddr_in),
105	 offsetof(struct sockaddr_in, sin_addr)},
106	{0, 0, 0, 0},
107};
108
109#define ENI_NOSOCKET 	0
110#define ENI_NOSERVNAME	1
111#define ENI_NOHOSTNAME	2
112#define ENI_MEMORY	3
113#define ENI_SYSTEM	4
114#define ENI_FAMILY	5
115#define ENI_SALEN	6
116
117#ifndef HAVE_INET_NTOP
118static const char *
119inet_ntop(int af, const void *addr, char *numaddr, size_t numaddr_len)
120{
121#ifdef HAVE_INET_NTOA
122	struct in_addr in;
123	memcpy(&in.s_addr, addr, sizeof(in.s_addr));
124	snprintf(numaddr, numaddr_len, "%s", inet_ntoa(in));
125#else
126	unsigned long x = ntohl(*(unsigned long*)addr);
127	snprintf(numaddr, numaddr_len, "%d.%d.%d.%d",
128		 (int) (x>>24) & 0xff, (int) (x>>16) & 0xff,
129		 (int) (x>> 8) & 0xff, (int) (x>> 0) & 0xff);
130#endif
131	return numaddr;
132}
133#endif
134
135int
136getnameinfo(const struct sockaddr *sa, socklen_t salen, char *host, socklen_t hostlen, char *serv, socklen_t servlen, int flags)
137{
138	struct afd *afd;
139	struct hostent *hp;
140	u_short port;
141	int family, len, i;
142	char *addr, *p;
143	u_long v4a;
144#ifdef INET6
145	u_char pfx;
146#endif
147	int h_error;
148	char numserv[512];
149	char numaddr[512];
150
151	if (sa == NULL)
152		return ENI_NOSOCKET;
153
154	len = SA_LEN(sa);
155	if (len != salen) return ENI_SALEN;
156
157	family = sa->sa_family;
158	for (i = 0; afdl[i].a_af; i++)
159		if (afdl[i].a_af == family) {
160			afd = &afdl[i];
161			goto found;
162		}
163	return ENI_FAMILY;
164
165 found:
166	if (len != afd->a_socklen) return ENI_SALEN;
167
168	port = ((struct sockinet *)sa)->si_port; /* network byte order */
169	addr = (char *)sa + afd->a_off;
170
171	if (serv == NULL || servlen == 0) {
172		/* what we should do? */
173	} else if (flags & NI_NUMERICSERV) {
174		snprintf(numserv, sizeof(numserv), "%d", ntohs(port));
175		if (strlen(numserv) + 1 > servlen)
176			return ENI_MEMORY;
177		strcpy(serv, numserv);
178	} else {
179#if defined(HAVE_GETSERVBYPORT)
180		struct servent *sp = getservbyport(port, (flags & NI_DGRAM) ? "udp" : "tcp");
181		if (sp) {
182			if (strlen(sp->s_name) + 1 > servlen)
183				return ENI_MEMORY;
184			strcpy(serv, sp->s_name);
185		} else
186			return ENI_NOSERVNAME;
187#else
188		return ENI_NOSERVNAME;
189#endif
190	}
191
192	switch (sa->sa_family) {
193	case AF_INET:
194		v4a = ntohl(((struct sockaddr_in *)sa)->sin_addr.s_addr);
195		if (IN_MULTICAST(v4a) || IN_EXPERIMENTAL(v4a))
196			flags |= NI_NUMERICHOST;
197		v4a >>= IN_CLASSA_NSHIFT;
198		if (v4a == 0)
199			flags |= NI_NUMERICHOST;
200		break;
201#ifdef INET6
202	case AF_INET6:
203#ifdef HAVE_ADDR8
204		pfx = ((struct sockaddr_in6 *)sa)->sin6_addr.s6_addr8[0];
205#else
206		pfx = ((struct sockaddr_in6 *)sa)->sin6_addr.s6_addr[0];
207#endif
208		if (pfx == 0 || pfx == 0xfe || pfx == 0xff)
209			flags |= NI_NUMERICHOST;
210		break;
211#endif
212	}
213	if (host == NULL || hostlen == 0) {
214		/* what should we do? */
215	} else if (flags & NI_NUMERICHOST) {
216		if (inet_ntop(afd->a_af, addr, numaddr, sizeof(numaddr))
217		    == NULL)
218			return ENI_SYSTEM;
219		if (strlen(numaddr) > hostlen)
220			return ENI_MEMORY;
221		strcpy(host, numaddr);
222	} else {
223#ifdef INET6
224		hp = getipnodebyaddr(addr, afd->a_addrlen, afd->a_af, &h_error);
225#else
226		hp = gethostbyaddr(addr, afd->a_addrlen, afd->a_af);
227		h_error = h_errno;
228#endif
229
230		if (hp) {
231			if (flags & NI_NOFQDN) {
232				p = strchr(hp->h_name, '.');
233				if (p) *p = '\0';
234			}
235			if (strlen(hp->h_name) + 1 > hostlen) {
236#ifdef INET6
237				freehostent(hp);
238#endif
239				return ENI_MEMORY;
240			}
241			strcpy(host, hp->h_name);
242#ifdef INET6
243			freehostent(hp);
244#endif
245		} else {
246			if (flags & NI_NAMEREQD)
247				return ENI_NOHOSTNAME;
248			if (inet_ntop(afd->a_af, addr, numaddr, sizeof(numaddr))
249			    == NULL)
250				return ENI_NOHOSTNAME;
251			if (strlen(numaddr) > hostlen)
252				return ENI_MEMORY;
253			strcpy(host, numaddr);
254		}
255	}
256	return SUCCESS;
257}
258