xref: /freebsd/crypto/openssh/canohost.c (revision 333ee03933b2892718b59734cd7dd3608b6caf1a)
1333ee039SDag-Erling Smørgrav /* $OpenBSD: canohost.c,v 1.61 2006/08/03 03:34:41 deraadt Exp $ */
2511b41d2SMark Murray /*
3511b41d2SMark Murray  * Author: Tatu Ylonen <ylo@cs.hut.fi>
4511b41d2SMark Murray  * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
5511b41d2SMark Murray  *                    All rights reserved
6511b41d2SMark Murray  * Functions for returning the canonical host name of the remote site.
7511b41d2SMark Murray  *
8c2d3a559SKris Kennaway  * As far as I am concerned, the code I have written for this software
9c2d3a559SKris Kennaway  * can be used freely for any purpose.  Any derived versions of this
10c2d3a559SKris Kennaway  * software must be clearly marked as such, and if the derived work is
11c2d3a559SKris Kennaway  * incompatible with the protocol description in the RFC file, it must be
12c2d3a559SKris Kennaway  * called by a name other than "ssh" or "Secure Shell".
13511b41d2SMark Murray  */
14511b41d2SMark Murray 
15511b41d2SMark Murray #include "includes.h"
16511b41d2SMark Murray 
17333ee039SDag-Erling Smørgrav #include <sys/types.h>
18333ee039SDag-Erling Smørgrav #include <sys/socket.h>
19333ee039SDag-Erling Smørgrav 
20333ee039SDag-Erling Smørgrav #include <netinet/in.h>
21333ee039SDag-Erling Smørgrav #include <arpa/inet.h>
22333ee039SDag-Erling Smørgrav 
23333ee039SDag-Erling Smørgrav #include <ctype.h>
24333ee039SDag-Erling Smørgrav #include <errno.h>
25333ee039SDag-Erling Smørgrav #include <netdb.h>
26333ee039SDag-Erling Smørgrav #include <stdio.h>
27333ee039SDag-Erling Smørgrav #include <stdlib.h>
28333ee039SDag-Erling Smørgrav #include <string.h>
29333ee039SDag-Erling Smørgrav #include <stdarg.h>
30333ee039SDag-Erling Smørgrav 
31511b41d2SMark Murray #include "xmalloc.h"
32333ee039SDag-Erling Smørgrav #include "packet.h"
33ca3176e7SBrian Feldman #include "log.h"
34ca3176e7SBrian Feldman #include "canohost.h"
35ca3176e7SBrian Feldman 
36af12a3e7SDag-Erling Smørgrav static void check_ip_options(int, char *);
37511b41d2SMark Murray 
38511b41d2SMark Murray /*
39511b41d2SMark Murray  * Return the canonical name of the host at the other end of the socket. The
40511b41d2SMark Murray  * caller should free the returned string with xfree.
41511b41d2SMark Murray  */
42511b41d2SMark Murray 
43af12a3e7SDag-Erling Smørgrav static char *
4421e764dfSDag-Erling Smørgrav get_remote_hostname(int sock, int use_dns)
45511b41d2SMark Murray {
46511b41d2SMark Murray 	struct sockaddr_storage from;
47511b41d2SMark Murray 	int i;
48511b41d2SMark Murray 	socklen_t fromlen;
49511b41d2SMark Murray 	struct addrinfo hints, *ai, *aitop;
50ca3176e7SBrian Feldman 	char name[NI_MAXHOST], ntop[NI_MAXHOST], ntop2[NI_MAXHOST];
51511b41d2SMark Murray 
52511b41d2SMark Murray 	/* Get IP address of client. */
53511b41d2SMark Murray 	fromlen = sizeof(from);
54511b41d2SMark Murray 	memset(&from, 0, sizeof(from));
5521e764dfSDag-Erling Smørgrav 	if (getpeername(sock, (struct sockaddr *)&from, &fromlen) < 0) {
56511b41d2SMark Murray 		debug("getpeername failed: %.100s", strerror(errno));
571ec0d754SDag-Erling Smørgrav 		cleanup_exit(255);
58511b41d2SMark Murray 	}
59989dd127SDag-Erling Smørgrav 
60333ee039SDag-Erling Smørgrav 	if (from.ss_family == AF_INET)
61333ee039SDag-Erling Smørgrav 		check_ip_options(sock, ntop);
62333ee039SDag-Erling Smørgrav 
631ec0d754SDag-Erling Smørgrav 	ipv64_normalise_mapped(&from, &fromlen);
64989dd127SDag-Erling Smørgrav 
65e73e9afaSDag-Erling Smørgrav 	if (from.ss_family == AF_INET6)
66e73e9afaSDag-Erling Smørgrav 		fromlen = sizeof(struct sockaddr_in6);
67ca3176e7SBrian Feldman 
68511b41d2SMark Murray 	if (getnameinfo((struct sockaddr *)&from, fromlen, ntop, sizeof(ntop),
69511b41d2SMark Murray 	    NULL, 0, NI_NUMERICHOST) != 0)
70511b41d2SMark Murray 		fatal("get_remote_hostname: getnameinfo NI_NUMERICHOST failed");
71511b41d2SMark Murray 
72cf2b5f3bSDag-Erling Smørgrav 	if (!use_dns)
73cf2b5f3bSDag-Erling Smørgrav 		return xstrdup(ntop);
74cf2b5f3bSDag-Erling Smørgrav 
75ca3176e7SBrian Feldman 	debug3("Trying to reverse map address %.100s.", ntop);
76511b41d2SMark Murray 	/* Map the IP address to a host name. */
77511b41d2SMark Murray 	if (getnameinfo((struct sockaddr *)&from, fromlen, name, sizeof(name),
78ca3176e7SBrian Feldman 	    NULL, 0, NI_NAMEREQD) != 0) {
79ca3176e7SBrian Feldman 		/* Host name not found.  Use ip address. */
80ca3176e7SBrian Feldman 		return xstrdup(ntop);
81ca3176e7SBrian Feldman 	}
82ca3176e7SBrian Feldman 
83cf2b5f3bSDag-Erling Smørgrav 	/*
84cf2b5f3bSDag-Erling Smørgrav 	 * if reverse lookup result looks like a numeric hostname,
85cf2b5f3bSDag-Erling Smørgrav 	 * someone is trying to trick us by PTR record like following:
86cf2b5f3bSDag-Erling Smørgrav 	 *	1.1.1.10.in-addr.arpa.	IN PTR	2.3.4.5
87cf2b5f3bSDag-Erling Smørgrav 	 */
88cf2b5f3bSDag-Erling Smørgrav 	memset(&hints, 0, sizeof(hints));
89cf2b5f3bSDag-Erling Smørgrav 	hints.ai_socktype = SOCK_DGRAM;	/*dummy*/
90cf2b5f3bSDag-Erling Smørgrav 	hints.ai_flags = AI_NUMERICHOST;
91cf2b5f3bSDag-Erling Smørgrav 	if (getaddrinfo(name, "0", &hints, &ai) == 0) {
92cf2b5f3bSDag-Erling Smørgrav 		logit("Nasty PTR record \"%s\" is set up for %s, ignoring",
93cf2b5f3bSDag-Erling Smørgrav 		    name, ntop);
94cf2b5f3bSDag-Erling Smørgrav 		freeaddrinfo(ai);
95cf2b5f3bSDag-Erling Smørgrav 		return xstrdup(ntop);
96cf2b5f3bSDag-Erling Smørgrav 	}
97cf2b5f3bSDag-Erling Smørgrav 
98511b41d2SMark Murray 	/*
99511b41d2SMark Murray 	 * Convert it to all lowercase (which is expected by the rest
100511b41d2SMark Murray 	 * of this software).
101511b41d2SMark Murray 	 */
102511b41d2SMark Murray 	for (i = 0; name[i]; i++)
103511b41d2SMark Murray 		if (isupper(name[i]))
104333ee039SDag-Erling Smørgrav 			name[i] = (char)tolower(name[i]);
105511b41d2SMark Murray 	/*
106511b41d2SMark Murray 	 * Map it back to an IP address and check that the given
107511b41d2SMark Murray 	 * address actually is an address of this host.  This is
108511b41d2SMark Murray 	 * necessary because anyone with access to a name server can
109511b41d2SMark Murray 	 * define arbitrary names for an IP address. Mapping from
110511b41d2SMark Murray 	 * name to IP address can be trusted better (but can still be
111511b41d2SMark Murray 	 * fooled if the intruder has access to the name server of
112511b41d2SMark Murray 	 * the domain).
113511b41d2SMark Murray 	 */
114511b41d2SMark Murray 	memset(&hints, 0, sizeof(hints));
115511b41d2SMark Murray 	hints.ai_family = from.ss_family;
116511b41d2SMark Murray 	hints.ai_socktype = SOCK_STREAM;
117511b41d2SMark Murray 	if (getaddrinfo(name, NULL, &hints, &aitop) != 0) {
118cf2b5f3bSDag-Erling Smørgrav 		logit("reverse mapping checking getaddrinfo for %.700s "
119333ee039SDag-Erling Smørgrav 		    "[%s] failed - POSSIBLE BREAK-IN ATTEMPT!", name, ntop);
120ca3176e7SBrian Feldman 		return xstrdup(ntop);
121511b41d2SMark Murray 	}
122511b41d2SMark Murray 	/* Look for the address from the list of addresses. */
123511b41d2SMark Murray 	for (ai = aitop; ai; ai = ai->ai_next) {
124511b41d2SMark Murray 		if (getnameinfo(ai->ai_addr, ai->ai_addrlen, ntop2,
125511b41d2SMark Murray 		    sizeof(ntop2), NULL, 0, NI_NUMERICHOST) == 0 &&
126511b41d2SMark Murray 		    (strcmp(ntop, ntop2) == 0))
127511b41d2SMark Murray 				break;
128511b41d2SMark Murray 	}
129511b41d2SMark Murray 	freeaddrinfo(aitop);
130511b41d2SMark Murray 	/* If we reached the end of the list, the address was not there. */
131511b41d2SMark Murray 	if (!ai) {
132511b41d2SMark Murray 		/* Address not found for the host name. */
133cf2b5f3bSDag-Erling Smørgrav 		logit("Address %.100s maps to %.600s, but this does not "
134b74df5b2SDag-Erling Smørgrav 		    "map back to the address - POSSIBLE BREAK-IN ATTEMPT!",
135511b41d2SMark Murray 		    ntop, name);
136ca3176e7SBrian Feldman 		return xstrdup(ntop);
137511b41d2SMark Murray 	}
138ca3176e7SBrian Feldman 	return xstrdup(name);
139511b41d2SMark Murray }
140511b41d2SMark Murray 
141511b41d2SMark Murray /*
142511b41d2SMark Murray  * If IP options are supported, make sure there are none (log and
143511b41d2SMark Murray  * disconnect them if any are found).  Basically we are worried about
144511b41d2SMark Murray  * source routing; it can be used to pretend you are somebody
145511b41d2SMark Murray  * (ip-address) you are not. That itself may be "almost acceptable"
146511b41d2SMark Murray  * under certain circumstances, but rhosts autentication is useless
147511b41d2SMark Murray  * if source routing is accepted. Notice also that if we just dropped
148511b41d2SMark Murray  * source routing here, the other side could use IP spoofing to do
149511b41d2SMark Murray  * rest of the interaction and could still bypass security.  So we
150511b41d2SMark Murray  * exit here if we detect any IP options.
151511b41d2SMark Murray  */
152ca3176e7SBrian Feldman /* IPv4 only */
153af12a3e7SDag-Erling Smørgrav static void
15421e764dfSDag-Erling Smørgrav check_ip_options(int sock, char *ipaddr)
155ca3176e7SBrian Feldman {
156cf2b5f3bSDag-Erling Smørgrav #ifdef IP_OPTIONS
157ca3176e7SBrian Feldman 	u_char options[200];
158ca3176e7SBrian Feldman 	char text[sizeof(options) * 3 + 1];
159511b41d2SMark Murray 	socklen_t option_size;
160d4ecd108SDag-Erling Smørgrav 	u_int i;
161d4ecd108SDag-Erling Smørgrav 	int ipproto;
162511b41d2SMark Murray 	struct protoent *ip;
163511b41d2SMark Murray 
164511b41d2SMark Murray 	if ((ip = getprotobyname("ip")) != NULL)
165511b41d2SMark Murray 		ipproto = ip->p_proto;
166511b41d2SMark Murray 	else
167511b41d2SMark Murray 		ipproto = IPPROTO_IP;
168511b41d2SMark Murray 	option_size = sizeof(options);
16921e764dfSDag-Erling Smørgrav 	if (getsockopt(sock, ipproto, IP_OPTIONS, options,
170511b41d2SMark Murray 	    &option_size) >= 0 && option_size != 0) {
171ca3176e7SBrian Feldman 		text[0] = '\0';
172ca3176e7SBrian Feldman 		for (i = 0; i < option_size; i++)
173ca3176e7SBrian Feldman 			snprintf(text + i*3, sizeof(text) - i*3,
174ca3176e7SBrian Feldman 			    " %2.2x", options[i]);
175b74df5b2SDag-Erling Smørgrav 		fatal("Connection from %.100s with IP options:%.800s",
176ca3176e7SBrian Feldman 		    ipaddr, text);
177511b41d2SMark Murray 	}
178cf2b5f3bSDag-Erling Smørgrav #endif /* IP_OPTIONS */
179511b41d2SMark Murray }
180511b41d2SMark Murray 
181aa49c926SDag-Erling Smørgrav void
1821ec0d754SDag-Erling Smørgrav ipv64_normalise_mapped(struct sockaddr_storage *addr, socklen_t *len)
1831ec0d754SDag-Erling Smørgrav {
1841ec0d754SDag-Erling Smørgrav 	struct sockaddr_in6 *a6 = (struct sockaddr_in6 *)addr;
1851ec0d754SDag-Erling Smørgrav 	struct sockaddr_in *a4 = (struct sockaddr_in *)addr;
1861ec0d754SDag-Erling Smørgrav 	struct in_addr inaddr;
1871ec0d754SDag-Erling Smørgrav 	u_int16_t port;
1881ec0d754SDag-Erling Smørgrav 
1891ec0d754SDag-Erling Smørgrav 	if (addr->ss_family != AF_INET6 ||
1901ec0d754SDag-Erling Smørgrav 	    !IN6_IS_ADDR_V4MAPPED(&a6->sin6_addr))
1911ec0d754SDag-Erling Smørgrav 		return;
1921ec0d754SDag-Erling Smørgrav 
1931ec0d754SDag-Erling Smørgrav 	debug3("Normalising mapped IPv4 in IPv6 address");
1941ec0d754SDag-Erling Smørgrav 
1951ec0d754SDag-Erling Smørgrav 	memcpy(&inaddr, ((char *)&a6->sin6_addr) + 12, sizeof(inaddr));
1961ec0d754SDag-Erling Smørgrav 	port = a6->sin6_port;
1971ec0d754SDag-Erling Smørgrav 
1981ec0d754SDag-Erling Smørgrav 	memset(addr, 0, sizeof(*a4));
1991ec0d754SDag-Erling Smørgrav 
2001ec0d754SDag-Erling Smørgrav 	a4->sin_family = AF_INET;
2011ec0d754SDag-Erling Smørgrav 	*len = sizeof(*a4);
2021ec0d754SDag-Erling Smørgrav 	memcpy(&a4->sin_addr, &inaddr, sizeof(inaddr));
2031ec0d754SDag-Erling Smørgrav 	a4->sin_port = port;
2041ec0d754SDag-Erling Smørgrav }
2051ec0d754SDag-Erling Smørgrav 
206511b41d2SMark Murray /*
207511b41d2SMark Murray  * Return the canonical name of the host in the other side of the current
208511b41d2SMark Murray  * connection.  The host name is cached, so it is efficient to call this
209511b41d2SMark Murray  * several times.
210511b41d2SMark Murray  */
211511b41d2SMark Murray 
212511b41d2SMark Murray const char *
213cf2b5f3bSDag-Erling Smørgrav get_canonical_hostname(int use_dns)
214511b41d2SMark Murray {
215b74df5b2SDag-Erling Smørgrav 	char *host;
216511b41d2SMark Murray 	static char *canonical_host_name = NULL;
217b74df5b2SDag-Erling Smørgrav 	static char *remote_ip = NULL;
218511b41d2SMark Murray 
219ca3176e7SBrian Feldman 	/* Check if we have previously retrieved name with same option. */
220b74df5b2SDag-Erling Smørgrav 	if (use_dns && canonical_host_name != NULL)
221511b41d2SMark Murray 		return canonical_host_name;
222b74df5b2SDag-Erling Smørgrav 	if (!use_dns && remote_ip != NULL)
223b74df5b2SDag-Erling Smørgrav 		return remote_ip;
224511b41d2SMark Murray 
225511b41d2SMark Murray 	/* Get the real hostname if socket; otherwise return UNKNOWN. */
226511b41d2SMark Murray 	if (packet_connection_is_on_socket())
227b74df5b2SDag-Erling Smørgrav 		host = get_remote_hostname(packet_get_connection_in(), use_dns);
228511b41d2SMark Murray 	else
229b74df5b2SDag-Erling Smørgrav 		host = "UNKNOWN";
230511b41d2SMark Murray 
231b74df5b2SDag-Erling Smørgrav 	if (use_dns)
232b74df5b2SDag-Erling Smørgrav 		canonical_host_name = host;
233b74df5b2SDag-Erling Smørgrav 	else
234b74df5b2SDag-Erling Smørgrav 		remote_ip = host;
235b74df5b2SDag-Erling Smørgrav 	return host;
236511b41d2SMark Murray }
237511b41d2SMark Murray 
238511b41d2SMark Murray /*
239e73e9afaSDag-Erling Smørgrav  * Returns the local/remote IP-address/hostname of socket as a string.
240e73e9afaSDag-Erling Smørgrav  * The returned string must be freed.
241ca3176e7SBrian Feldman  */
242af12a3e7SDag-Erling Smørgrav static char *
24321e764dfSDag-Erling Smørgrav get_socket_address(int sock, int remote, int flags)
244ca3176e7SBrian Feldman {
245ca3176e7SBrian Feldman 	struct sockaddr_storage addr;
246ca3176e7SBrian Feldman 	socklen_t addrlen;
247ca3176e7SBrian Feldman 	char ntop[NI_MAXHOST];
248aa49c926SDag-Erling Smørgrav 	int r;
249ca3176e7SBrian Feldman 
250ca3176e7SBrian Feldman 	/* Get IP address of client. */
251ca3176e7SBrian Feldman 	addrlen = sizeof(addr);
252ca3176e7SBrian Feldman 	memset(&addr, 0, sizeof(addr));
253ca3176e7SBrian Feldman 
254ca3176e7SBrian Feldman 	if (remote) {
25521e764dfSDag-Erling Smørgrav 		if (getpeername(sock, (struct sockaddr *)&addr, &addrlen)
256f388f5efSDag-Erling Smørgrav 		    < 0)
257ca3176e7SBrian Feldman 			return NULL;
258ca3176e7SBrian Feldman 	} else {
25921e764dfSDag-Erling Smørgrav 		if (getsockname(sock, (struct sockaddr *)&addr, &addrlen)
260f388f5efSDag-Erling Smørgrav 		    < 0)
261ca3176e7SBrian Feldman 			return NULL;
262ca3176e7SBrian Feldman 	}
263e73e9afaSDag-Erling Smørgrav 
264e73e9afaSDag-Erling Smørgrav 	/* Work around Linux IPv6 weirdness */
265e73e9afaSDag-Erling Smørgrav 	if (addr.ss_family == AF_INET6)
266e73e9afaSDag-Erling Smørgrav 		addrlen = sizeof(struct sockaddr_in6);
267e73e9afaSDag-Erling Smørgrav 
268aa49c926SDag-Erling Smørgrav 	ipv64_normalise_mapped(&addr, &addrlen);
269aa49c926SDag-Erling Smørgrav 
270ca3176e7SBrian Feldman 	/* Get the address in ascii. */
271aa49c926SDag-Erling Smørgrav 	if ((r = getnameinfo((struct sockaddr *)&addr, addrlen, ntop,
272aa49c926SDag-Erling Smørgrav 	    sizeof(ntop), NULL, 0, flags)) != 0) {
273aa49c926SDag-Erling Smørgrav 		error("get_socket_address: getnameinfo %d failed: %s", flags,
274aa49c926SDag-Erling Smørgrav 		    r == EAI_SYSTEM ? strerror(errno) : gai_strerror(r));
275ca3176e7SBrian Feldman 		return NULL;
276ca3176e7SBrian Feldman 	}
277ca3176e7SBrian Feldman 	return xstrdup(ntop);
278ca3176e7SBrian Feldman }
279ca3176e7SBrian Feldman 
280ca3176e7SBrian Feldman char *
28121e764dfSDag-Erling Smørgrav get_peer_ipaddr(int sock)
282ca3176e7SBrian Feldman {
283f388f5efSDag-Erling Smørgrav 	char *p;
284f388f5efSDag-Erling Smørgrav 
28521e764dfSDag-Erling Smørgrav 	if ((p = get_socket_address(sock, 1, NI_NUMERICHOST)) != NULL)
286f388f5efSDag-Erling Smørgrav 		return p;
287f388f5efSDag-Erling Smørgrav 	return xstrdup("UNKNOWN");
288ca3176e7SBrian Feldman }
289ca3176e7SBrian Feldman 
290ca3176e7SBrian Feldman char *
29121e764dfSDag-Erling Smørgrav get_local_ipaddr(int sock)
292ca3176e7SBrian Feldman {
293f388f5efSDag-Erling Smørgrav 	char *p;
294f388f5efSDag-Erling Smørgrav 
29521e764dfSDag-Erling Smørgrav 	if ((p = get_socket_address(sock, 0, NI_NUMERICHOST)) != NULL)
296f388f5efSDag-Erling Smørgrav 		return p;
297f388f5efSDag-Erling Smørgrav 	return xstrdup("UNKNOWN");
298ca3176e7SBrian Feldman }
299ca3176e7SBrian Feldman 
300ca3176e7SBrian Feldman char *
30121e764dfSDag-Erling Smørgrav get_local_name(int sock)
302ca3176e7SBrian Feldman {
30321e764dfSDag-Erling Smørgrav 	return get_socket_address(sock, 0, NI_NAMEREQD);
304ca3176e7SBrian Feldman }
305ca3176e7SBrian Feldman 
306ca3176e7SBrian Feldman /*
307511b41d2SMark Murray  * Returns the IP-address of the remote host as a string.  The returned
308511b41d2SMark Murray  * string must not be freed.
309511b41d2SMark Murray  */
310511b41d2SMark Murray 
311511b41d2SMark Murray const char *
312af12a3e7SDag-Erling Smørgrav get_remote_ipaddr(void)
313511b41d2SMark Murray {
314511b41d2SMark Murray 	static char *canonical_host_ip = NULL;
315511b41d2SMark Murray 
316ca3176e7SBrian Feldman 	/* Check whether we have cached the ipaddr. */
317ca3176e7SBrian Feldman 	if (canonical_host_ip == NULL) {
318ca3176e7SBrian Feldman 		if (packet_connection_is_on_socket()) {
319ca3176e7SBrian Feldman 			canonical_host_ip =
320ca3176e7SBrian Feldman 			    get_peer_ipaddr(packet_get_connection_in());
321ca3176e7SBrian Feldman 			if (canonical_host_ip == NULL)
3221ec0d754SDag-Erling Smørgrav 				cleanup_exit(255);
323ca3176e7SBrian Feldman 		} else {
324ca3176e7SBrian Feldman 			/* If not on socket, return UNKNOWN. */
325ca3176e7SBrian Feldman 			canonical_host_ip = xstrdup("UNKNOWN");
326511b41d2SMark Murray 		}
327ca3176e7SBrian Feldman 	}
328511b41d2SMark Murray 	return canonical_host_ip;
329511b41d2SMark Murray }
330511b41d2SMark Murray 
3317e03cf33SBrian Feldman const char *
332cf2b5f3bSDag-Erling Smørgrav get_remote_name_or_ip(u_int utmp_len, int use_dns)
3337e03cf33SBrian Feldman {
334ca3176e7SBrian Feldman 	static const char *remote = "";
335ca3176e7SBrian Feldman 	if (utmp_len > 0)
336cf2b5f3bSDag-Erling Smørgrav 		remote = get_canonical_hostname(use_dns);
337ca3176e7SBrian Feldman 	if (utmp_len == 0 || strlen(remote) > utmp_len)
338ca3176e7SBrian Feldman 		remote = get_remote_ipaddr();
339ca3176e7SBrian Feldman 	return remote;
3407e03cf33SBrian Feldman }
3417e03cf33SBrian Feldman 
342511b41d2SMark Murray /* Returns the local/remote port for the socket. */
343511b41d2SMark Murray 
344af12a3e7SDag-Erling Smørgrav static int
345511b41d2SMark Murray get_sock_port(int sock, int local)
346511b41d2SMark Murray {
347511b41d2SMark Murray 	struct sockaddr_storage from;
348511b41d2SMark Murray 	socklen_t fromlen;
349511b41d2SMark Murray 	char strport[NI_MAXSERV];
350aa49c926SDag-Erling Smørgrav 	int r;
351511b41d2SMark Murray 
352511b41d2SMark Murray 	/* Get IP address of client. */
353511b41d2SMark Murray 	fromlen = sizeof(from);
354511b41d2SMark Murray 	memset(&from, 0, sizeof(from));
355511b41d2SMark Murray 	if (local) {
356511b41d2SMark Murray 		if (getsockname(sock, (struct sockaddr *)&from, &fromlen) < 0) {
357511b41d2SMark Murray 			error("getsockname failed: %.100s", strerror(errno));
358511b41d2SMark Murray 			return 0;
359511b41d2SMark Murray 		}
360511b41d2SMark Murray 	} else {
361511b41d2SMark Murray 		if (getpeername(sock, (struct sockaddr *)&from, &fromlen) < 0) {
362511b41d2SMark Murray 			debug("getpeername failed: %.100s", strerror(errno));
363d4ecd108SDag-Erling Smørgrav 			return -1;
364511b41d2SMark Murray 		}
365511b41d2SMark Murray 	}
366e73e9afaSDag-Erling Smørgrav 
367e73e9afaSDag-Erling Smørgrav 	/* Work around Linux IPv6 weirdness */
368e73e9afaSDag-Erling Smørgrav 	if (from.ss_family == AF_INET6)
369e73e9afaSDag-Erling Smørgrav 		fromlen = sizeof(struct sockaddr_in6);
370e73e9afaSDag-Erling Smørgrav 
371511b41d2SMark Murray 	/* Return port number. */
372aa49c926SDag-Erling Smørgrav 	if ((r = getnameinfo((struct sockaddr *)&from, fromlen, NULL, 0,
373aa49c926SDag-Erling Smørgrav 	    strport, sizeof(strport), NI_NUMERICSERV)) != 0)
374aa49c926SDag-Erling Smørgrav 		fatal("get_sock_port: getnameinfo NI_NUMERICSERV failed: %s",
375aa49c926SDag-Erling Smørgrav 		    r == EAI_SYSTEM ? strerror(errno) : gai_strerror(r));
376511b41d2SMark Murray 	return atoi(strport);
377511b41d2SMark Murray }
378511b41d2SMark Murray 
379511b41d2SMark Murray /* Returns remote/local port number for the current connection. */
380511b41d2SMark Murray 
381af12a3e7SDag-Erling Smørgrav static int
382511b41d2SMark Murray get_port(int local)
383511b41d2SMark Murray {
384511b41d2SMark Murray 	/*
385511b41d2SMark Murray 	 * If the connection is not a socket, return 65535.  This is
386511b41d2SMark Murray 	 * intentionally chosen to be an unprivileged port number.
387511b41d2SMark Murray 	 */
388511b41d2SMark Murray 	if (!packet_connection_is_on_socket())
389511b41d2SMark Murray 		return 65535;
390511b41d2SMark Murray 
391511b41d2SMark Murray 	/* Get socket and return the port number. */
392511b41d2SMark Murray 	return get_sock_port(packet_get_connection_in(), local);
393511b41d2SMark Murray }
394511b41d2SMark Murray 
395511b41d2SMark Murray int
396511b41d2SMark Murray get_peer_port(int sock)
397511b41d2SMark Murray {
398511b41d2SMark Murray 	return get_sock_port(sock, 0);
399511b41d2SMark Murray }
400511b41d2SMark Murray 
401511b41d2SMark Murray int
402af12a3e7SDag-Erling Smørgrav get_remote_port(void)
403511b41d2SMark Murray {
40421e764dfSDag-Erling Smørgrav 	static int port = -1;
40521e764dfSDag-Erling Smørgrav 
40621e764dfSDag-Erling Smørgrav 	/* Cache to avoid getpeername() on a dead connection */
40721e764dfSDag-Erling Smørgrav 	if (port == -1)
40821e764dfSDag-Erling Smørgrav 		port = get_port(0);
40921e764dfSDag-Erling Smørgrav 
41021e764dfSDag-Erling Smørgrav 	return port;
411511b41d2SMark Murray }
412511b41d2SMark Murray 
413511b41d2SMark Murray int
414af12a3e7SDag-Erling Smørgrav get_local_port(void)
415511b41d2SMark Murray {
416511b41d2SMark Murray 	return get_port(1);
417511b41d2SMark Murray }
418