2 * Functions for looking up the remote name or addr of a socket.
4 * Copyright (C) 1992-2001 Andrew Tridgell <tridge@samba.org>
5 * Copyright (C) 2001, 2002 Martin Pool <mbp@samba.org>
6 * Copyright (C) 2002-2022 Wayne Davison
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 3 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, visit the http://fsf.org website.
23 * This file is now converted to use the new-style getaddrinfo()
24 * interface, which supports IPv6 but is also supported on recent
25 * IPv4-only machines. On systems that don't have that interface, we
26 * emulate it using the KAME implementation.
34 static const char default_name
[] = "UNKNOWN";
35 static const char proxyv2sig
[] = "\r\n\r\n\0\r\nQUIT\n";
37 static char ipaddr_buf
[100];
39 #define PROXY_V2_SIG_SIZE ((int)sizeof proxyv2sig - 1)
40 #define PROXY_V2_HEADER_SIZE (PROXY_V2_SIG_SIZE + 1 + 1 + 2)
45 #define PROXY_FAM_TCPv4 0x11
46 #define PROXY_FAM_TCPv6 0x21
48 #define GET_SOCKADDR_FAMILY(ss) ((struct sockaddr*)ss)->sa_family
50 static void client_sockaddr(int fd
, struct sockaddr_storage
*ss
, socklen_t
*ss_len
);
51 static int check_name(const char *ipaddr
, const struct sockaddr_storage
*ss
, char *name_buf
, size_t name_buf_size
);
52 static int valid_ipaddr(const char *s
, int allow_scope
);
54 /* Return the IP addr of the client as a string. */
55 char *client_addr(int fd
)
57 struct sockaddr_storage ss
;
58 socklen_t length
= sizeof ss
;
63 if (am_daemon
< 0) { /* daemon over --rsh mode */
65 strlcpy(ipaddr_buf
, "0.0.0.0", sizeof ipaddr_buf
);
66 if ((env_str
= getenv("REMOTE_HOST")) != NULL
67 || (env_str
= getenv("SSH_CONNECTION")) != NULL
68 || (env_str
= getenv("SSH_CLIENT")) != NULL
69 || (env_str
= getenv("SSH2_CLIENT")) != NULL
) {
71 strlcpy(ipaddr_buf
, env_str
, sizeof ipaddr_buf
);
72 /* Truncate the value to just the IP address. */
73 if ((p
= strchr(ipaddr_buf
, ' ')) != NULL
)
76 if (valid_ipaddr(ipaddr_buf
, True
))
80 client_sockaddr(fd
, &ss
, &length
);
81 getnameinfo((struct sockaddr
*)&ss
, length
, ipaddr_buf
, sizeof ipaddr_buf
, NULL
, 0, NI_NUMERICHOST
);
88 * Return the DNS name of the client.
90 * The name is statically cached so that repeated lookups are quick,
91 * so there is a limit of one lookup per customer.
93 * If anything goes wrong, including the name->addr->name check, then
94 * we just use "UNKNOWN", so you can use that value in hosts allow
97 * After translation from sockaddr to name we do a forward lookup to
98 * make sure nobody is spoofing PTR records.
100 char *client_name(const char *ipaddr
)
102 static char name_buf
[100];
104 struct sockaddr_storage ss
;
106 struct addrinfo hint
, *answer
;
112 strlcpy(name_buf
, default_name
, sizeof name_buf
);
114 if (strcmp(ipaddr
, "0.0.0.0") == 0)
117 memset(&ss
, 0, sizeof ss
);
118 memset(&hint
, 0, sizeof hint
);
120 #ifdef AI_NUMERICHOST
121 hint
.ai_flags
= AI_NUMERICHOST
;
123 hint
.ai_socktype
= SOCK_STREAM
;
125 if ((err
= getaddrinfo(ipaddr
, NULL
, &hint
, &answer
)) != 0) {
126 rprintf(FLOG
, "malformed address %s: %s\n", ipaddr
, gai_strerror(err
));
130 switch (answer
->ai_family
) {
132 ss_len
= sizeof (struct sockaddr_in
);
133 memcpy(&ss
, answer
->ai_addr
, ss_len
);
137 ss_len
= sizeof (struct sockaddr_in6
);
138 memcpy(&ss
, answer
->ai_addr
, ss_len
);
142 NOISY_DEATH("Unknown ai_family value");
144 freeaddrinfo(answer
);
147 err
= getnameinfo((struct sockaddr
*)&ss
, ss_len
, name_buf
, sizeof name_buf
,
148 port_buf
, sizeof port_buf
, NI_NAMEREQD
| NI_NUMERICSERV
);
150 strlcpy(name_buf
, default_name
, sizeof name_buf
);
151 rprintf(FLOG
, "name lookup failed for %s: %s\n", ipaddr
, gai_strerror(err
));
153 check_name(ipaddr
, &ss
, name_buf
, sizeof name_buf
);
159 /* Try to read a proxy protocol header (V1 or V2). Returns 1 on success or 0 on failure. */
160 int read_proxy_protocol_header(int fd
)
167 char sig
[PROXY_V2_SIG_SIZE
];
192 read_buf(fd
, (char*)&hdr
, PROXY_V2_SIG_SIZE
);
194 if (memcmp(hdr
.v2
.sig
, proxyv2sig
, PROXY_V2_SIG_SIZE
) == 0) { /* Proxy V2 */
197 read_buf(fd
, (char*)&hdr
+ PROXY_V2_SIG_SIZE
, PROXY_V2_HEADER_SIZE
- PROXY_V2_SIG_SIZE
);
199 ver
= (hdr
.v2
.ver_cmd
& 0xf0) >> 4;
200 cmd
= (hdr
.v2
.ver_cmd
& 0x0f);
201 size
= (hdr
.v2
.len
[0] << 8) + hdr
.v2
.len
[1];
203 if (ver
!= 2 || size
+ PROXY_V2_HEADER_SIZE
> (int)sizeof hdr
)
206 /* Grab all the remaining data in the binary request. */
207 read_buf(fd
, (char*)&hdr
+ PROXY_V2_HEADER_SIZE
, size
);
211 switch (hdr
.v2
.fam
) {
212 case PROXY_FAM_TCPv4
:
213 if (size
!= sizeof hdr
.v2
.addr
.ip4
)
215 inet_ntop(AF_INET
, hdr
.v2
.addr
.ip4
.src_addr
, ipaddr_buf
, sizeof ipaddr_buf
);
216 return valid_ipaddr(ipaddr_buf
, False
);
218 case PROXY_FAM_TCPv6
:
219 if (size
!= sizeof hdr
.v2
.addr
.ip6
)
221 inet_ntop(AF_INET6
, hdr
.v2
.addr
.ip6
.src_addr
, ipaddr_buf
, sizeof ipaddr_buf
);
222 return valid_ipaddr(ipaddr_buf
, False
);
227 /* For an unsupported protocol we'll ignore the proxy data (leaving ipaddr_buf unset)
228 * and accept the connection, which will get handled as a normal socket addr. */
239 if (memcmp(hdr
.v1
.line
, "PROXY", 5) == 0) { /* Proxy V1 */
240 char *endc
, *sp
, *p
= hdr
.v1
.line
+ PROXY_V2_SIG_SIZE
;
244 if (!strchr(hdr
.v1
.line
, '\n')) {
249 if (p
- hdr
.v1
.line
>= (int)sizeof hdr
.v1
.line
- 1)
255 endc
= strchr(hdr
.v1
.line
, '\r');
256 if (!endc
|| endc
[1] != '\n' || endc
[2])
264 if (strncmp(p
, "TCP4", 4) == 0)
266 else if (strncmp(p
, "TCP6", 4) == 0)
268 else if (strncmp(p
, "UNKNOWN", 7) == 0)
276 if ((sp
= strchr(p
, ' ')) == NULL
)
279 if (!valid_ipaddr(p
, False
))
281 strlcpy(ipaddr_buf
, p
, sizeof ipaddr_buf
); /* It will always fit when valid. */
284 if ((sp
= strchr(p
, ' ')) == NULL
)
287 if (!valid_ipaddr(p
, False
))
289 /* Ignore destination address. */
292 if ((sp
= strchr(p
, ' ')) == NULL
)
295 port_chk
= strtol(p
, &endc
, 10);
296 if (*endc
|| port_chk
== 0)
298 /* Ignore source port. */
301 port_chk
= strtol(p
, &endc
, 10);
302 if (*endc
|| port_chk
== 0)
304 /* Ignore destination port. */
314 * Get the sockaddr for the client.
316 * If it comes in as an ipv4 address mapped into IPv6 format then we
317 * convert it back to a regular IPv4.
319 static void client_sockaddr(int fd
, struct sockaddr_storage
*ss
, socklen_t
*ss_len
)
321 memset(ss
, 0, sizeof *ss
);
323 if (getpeername(fd
, (struct sockaddr
*) ss
, ss_len
)) {
324 /* FIXME: Can we really not continue? */
325 rsyserr(FLOG
, errno
, "getpeername on fd%d failed", fd
);
326 exit_cleanup(RERR_SOCKETIO
);
330 if (GET_SOCKADDR_FAMILY(ss
) == AF_INET6
331 && IN6_IS_ADDR_V4MAPPED(&((struct sockaddr_in6
*)ss
)->sin6_addr
)) {
332 /* OK, so ss is in the IPv6 family, but it is really
333 * an IPv4 address: something like
334 * "::ffff:10.130.1.2". If we use it as-is, then the
335 * reverse lookup might fail or perhaps something else
336 * bad might happen. So instead we convert it to an
337 * equivalent address in the IPv4 address family. */
338 struct sockaddr_in6 sin6
;
339 struct sockaddr_in
*sin
;
341 memcpy(&sin6
, ss
, sizeof sin6
);
342 sin
= (struct sockaddr_in
*)ss
;
343 memset(sin
, 0, sizeof *sin
);
344 sin
->sin_family
= AF_INET
;
345 *ss_len
= sizeof (struct sockaddr_in
);
346 #ifdef HAVE_SOCKADDR_IN_LEN
347 sin
->sin_len
= *ss_len
;
349 sin
->sin_port
= sin6
.sin6_port
;
351 /* There is a macro to extract the mapped part
352 * (IN6_V4MAPPED_TO_SINADDR ?), but it does not seem
353 * to be present in the Linux headers. */
354 memcpy(&sin
->sin_addr
, &sin6
.sin6_addr
.s6_addr
[12], sizeof sin
->sin_addr
);
361 * Compare an addrinfo from the resolver to a sockinfo.
363 * Like strcmp, returns 0 for identical.
365 static int compare_addrinfo_sockaddr(const struct addrinfo
*ai
, const struct sockaddr_storage
*ss
)
367 int ss_family
= GET_SOCKADDR_FAMILY(ss
);
368 const char fn
[] = "compare_addrinfo_sockaddr";
370 if (ai
->ai_family
!= ss_family
) {
371 rprintf(FLOG
, "%s: response family %d != %d\n",
372 fn
, ai
->ai_family
, ss_family
);
376 /* The comparison method depends on the particular AF. */
377 if (ss_family
== AF_INET
) {
378 const struct sockaddr_in
*sin1
, *sin2
;
380 sin1
= (const struct sockaddr_in
*) ss
;
381 sin2
= (const struct sockaddr_in
*) ai
->ai_addr
;
383 return memcmp(&sin1
->sin_addr
, &sin2
->sin_addr
, sizeof sin1
->sin_addr
);
387 if (ss_family
== AF_INET6
) {
388 const struct sockaddr_in6
*sin1
, *sin2
;
390 sin1
= (const struct sockaddr_in6
*) ss
;
391 sin2
= (const struct sockaddr_in6
*) ai
->ai_addr
;
393 if (ai
->ai_addrlen
< (int)sizeof (struct sockaddr_in6
)) {
394 rprintf(FLOG
, "%s: too short sockaddr_in6; length=%d\n",
395 fn
, (int)ai
->ai_addrlen
);
399 if (memcmp(&sin1
->sin6_addr
, &sin2
->sin6_addr
, sizeof sin1
->sin6_addr
))
402 #ifdef HAVE_SOCKADDR_IN6_SCOPE_ID
403 if (sin1
->sin6_scope_id
!= sin2
->sin6_scope_id
)
416 * Do a forward lookup on @p name_buf and make sure it corresponds to
417 * @p ss -- otherwise we may be being spoofed. If we suspect we are,
418 * then we don't abort the connection but just emit a warning, and
419 * change @p name_buf to be "UNKNOWN".
421 * We don't do anything with the service when checking the name,
422 * because it doesn't seem that it could be spoofed in any way, and
423 * getaddrinfo on random service names seems to cause problems on AIX.
425 static int check_name(const char *ipaddr
, const struct sockaddr_storage
*ss
, char *name_buf
, size_t name_buf_size
)
427 struct addrinfo hints
, *res
, *res0
;
429 int ss_family
= GET_SOCKADDR_FAMILY(ss
);
431 memset(&hints
, 0, sizeof hints
);
432 hints
.ai_family
= ss_family
;
433 hints
.ai_flags
= AI_CANONNAME
;
434 hints
.ai_socktype
= SOCK_STREAM
;
435 error
= getaddrinfo(name_buf
, NULL
, &hints
, &res0
);
437 rprintf(FLOG
, "forward name lookup for %s failed: %s\n",
438 name_buf
, gai_strerror(error
));
439 strlcpy(name_buf
, default_name
, name_buf_size
);
443 /* Given all these results, we expect that one of them will be
444 * the same as ss. The comparison is a bit complicated. */
445 for (res
= res0
; res
; res
= res
->ai_next
) {
446 if (!compare_addrinfo_sockaddr(res
, ss
))
447 break; /* OK, identical */
451 /* We hit the end of the list without finding an
452 * address that was the same as ss. */
453 rprintf(FLOG
, "no known address for \"%s\": "
454 "spoofed address?\n", name_buf
);
455 strlcpy(name_buf
, default_name
, name_buf_size
);
456 } else if (res
== NULL
) {
457 /* We hit the end of the list without finding an
458 * address that was the same as ss. */
459 rprintf(FLOG
, "%s is not a known address for \"%s\": "
460 "spoofed address?\n", ipaddr
, name_buf
);
461 strlcpy(name_buf
, default_name
, name_buf_size
);
468 /* Returns 1 for a valid IPv4 or IPv6 addr, or 0 for a bad one. */
469 static int valid_ipaddr(const char *s
, int allow_scope
)
473 if (strchr(s
, ':') != NULL
) { /* Only IPv6 has a colon. */
474 int count
, saw_double_colon
= 0;
477 if (*s
== ':') { /* A colon at the start must be a :: */
480 saw_double_colon
= 1;
484 for (count
= 0; count
< 8; count
++) {
486 return saw_double_colon
;
487 if (allow_scope
&& *s
== '%') {
488 if (saw_double_colon
)
493 if (strchr(s
, ':') == NULL
&& strchr(s
, '.') != NULL
) {
494 if ((!saw_double_colon
&& count
!= 6) || (saw_double_colon
&& count
> 6))
500 if (!isHexDigit(s
++)) /* Need 1-4 hex digits */
502 if (isHexDigit(s
) && isHexDigit(++s
) && isHexDigit(++s
) && isHexDigit(++s
))
509 if (saw_double_colon
)
511 saw_double_colon
= 1;
518 if (allow_scope
&& *s
== '%')
519 for (s
++; isAlNum(s
); s
++) { }
520 return !*s
&& s
[-1] != '%';
525 for (i
= 0; i
< 4; i
++) {
529 if (i
&& *s
++ != '.')
531 n
= strtol(s
, &end
, 10);
532 if (n
> 255 || n
< 0 || end
<= s
|| end
> s
+3)