]> git.ipfire.org Git - thirdparty/squid.git/blame - src/ip/Address.cc
SourceFormat Enforcement
[thirdparty/squid.git] / src / ip / Address.cc
CommitLineData
41d93087 1/*
2cd0bda2 2 * Copyright (C) 1996-2017 The Squid Software Foundation and contributors
bbc27441
AJ
3 *
4 * Squid software is distributed under GPLv2+ license and includes
5 * contributions from numerous individuals and organizations.
6 * Please see the COPYING and CONTRIBUTORS files for details.
41d93087 7 */
bbc27441
AJ
8
9/* DEBUG: section 14 IP Storage and Handling */
10
f7f3304a 11#include "squid.h"
82b7abe3 12#include "Debug.h"
96d89ea0 13#include "ip/Address.h"
055421ee 14#include "ip/tools.h"
1ef0b9ce
AJ
15#include "util.h"
16
074d6a40
AJ
17#include <cassert>
18#include <cstring>
41d93087 19#if HAVE_ARPA_INET_H
489520a9
AJ
20/* for inet_ntoa() */
21#include <arpa/inet.h>
41d93087 22#endif
760da772
AJ
23#if HAVE_WS2TCPIP_H
24// Windows IPv6 definitions
25#include <ws2tcpip.h>
26#endif
27
28// some OS (ie WIndows) define IN6_ADDR_EQUAL instead
29#if !defined(IN6_ARE_ADDR_EQUAL) && _SQUID_WINDOWS_
30#define IN6_ARE_ADDR_EQUAL IN6_ADDR_EQUAL
31#endif
41d93087 32
41d93087 33/* Debugging only. Dump the address content when a fatal assert is encountered. */
41d93087 34#define IASSERT(a,b) \
f53969cc
SM
35 if(!(b)){ printf("assert \"%s\" at line %d\n", a, __LINE__); \
36 printf("Ip::Address invalid? with isIPv4()=%c, isIPv6()=%c\n",(isIPv4()?'T':'F'),(isIPv6()?'T':'F')); \
37 printf("ADDRESS:"); \
38 for(unsigned int i = 0; i < sizeof(mSocketAddr_.sin6_addr); ++i) { \
39 printf(" %x", mSocketAddr_.sin6_addr.s6_addr[i]); \
40 } printf("\n"); assert(b); \
41 }
41d93087 42
41d93087 43int
4dd643d5 44Ip::Address::cidr() const
41d93087 45{
760da772 46 uint8_t shift,ipbyte;
41d93087 47 uint8_t bit,caught;
48 int len = 0;
4dd643d5 49 const uint8_t *ptr= mSocketAddr_.sin6_addr.s6_addr;
41d93087 50
51 /* Let's scan all the bits from Most Significant to Least */
52 /* Until we find an "0" bit. Then, we return */
53 shift=0;
54
41d93087 55 /* return IPv4 CIDR for any Mapped address */
56 /* Thus only check the mapped bit */
57
4dd643d5 58 if ( !isIPv6() ) {
41d93087 59 shift = 12;
60 }
61
4dd643d5 62 for (; shift<sizeof(mSocketAddr_.sin6_addr) ; ++shift) {
760da772 63 ipbyte= *(ptr+shift);
41d93087 64
760da772 65 if (ipbyte == 0xFF) {
41d93087 66 len += 8;
67 continue ; /* A short-cut */
68 }
69
5e263176 70 for (caught = 0 , bit= 7 ; !caught && (bit <= 7); --bit) {
760da772 71 caught = ((ipbyte & 0x80) == 0x00); /* Found a '0' at 'bit' ? */
41d93087 72
73 if (!caught)
d7ae3534 74 ++len;
41d93087 75
760da772 76 ipbyte <<= 1;
41d93087 77 }
78
79 if (caught)
80 break; /* We have found the most significant "0" bit. */
81 }
82
83 return len;
84}
85
881c4733 86int
4dd643d5 87Ip::Address::applyMask(Ip::Address const &mask_addr)
41d93087 88{
4dd643d5
AJ
89 uint32_t *p1 = (uint32_t*)(&mSocketAddr_.sin6_addr);
90 uint32_t const *p2 = (uint32_t const *)(&mask_addr.mSocketAddr_.sin6_addr);
91 unsigned int blen = sizeof(mSocketAddr_.sin6_addr)/sizeof(uint32_t);
41d93087 92 unsigned int changes = 0;
93
d7ae3534 94 for (unsigned int i = 0; i < blen; ++i) {
26ac0430 95 if ((p1[i] & p2[i]) != p1[i])
d7ae3534 96 ++changes;
41d93087 97
98 p1[i] &= p2[i];
99 }
100
41d93087 101 return changes;
102}
103
b7ac5457 104bool
4dd643d5 105Ip::Address::applyMask(const unsigned int cidrMask, int mtype)
41d93087 106{
107 uint8_t clearbits = 0;
108 uint8_t* p = NULL;
109
41d93087 110 // validation and short-cuts.
4dd643d5 111 if (cidrMask > 128)
41d93087 112 return false;
113
4dd643d5 114 if (cidrMask > 32 && mtype == AF_INET)
41d93087 115 return false;
116
4dd643d5 117 if (cidrMask == 0) {
2c1b9590 118 /* CIDR /0 is NoAddr regardless of the IPv4/IPv6 protocol */
4dd643d5 119 setNoAddr();
2c1b9590
AJ
120 return true;
121 }
122
4dd643d5 123 clearbits = (uint8_t)( (mtype==AF_INET6?128:32) - cidrMask);
41d93087 124
125 // short-cut
26ac0430 126 if (clearbits == 0)
41d93087 127 return true;
128
4dd643d5 129 p = (uint8_t*)(&mSocketAddr_.sin6_addr) + 15;
41d93087 130
4dd643d5 131 for (; clearbits>0 && p >= (uint8_t*)&mSocketAddr_.sin6_addr ; --p ) {
26ac0430 132 if (clearbits < 8) {
41d93087 133 *p &= ((0xFF << clearbits) & 0xFF);
134 clearbits = 0;
135 } else {
136 *p &= 0x00;
137 clearbits -= 8;
138 }
139 }
140
141 return true;
142}
143
b7ac5457 144bool
4dd643d5 145Ip::Address::isSockAddr() const
41d93087 146{
4dd643d5 147 return (mSocketAddr_.sin6_port != 0);
41d93087 148}
149
b7ac5457 150bool
4dd643d5 151Ip::Address::isIPv4() const
41d93087 152{
4dd643d5 153 return IN6_IS_ADDR_V4MAPPED( &mSocketAddr_.sin6_addr );
41d93087 154}
155
b7ac5457 156bool
4dd643d5 157Ip::Address::isIPv6() const
41d93087 158{
4dd643d5 159 return !isIPv4();
41d93087 160}
161
b7ac5457 162bool
4dd643d5 163Ip::Address::isAnyAddr() const
41d93087 164{
4dd643d5 165 return IN6_IS_ADDR_UNSPECIFIED(&mSocketAddr_.sin6_addr) || IN6_ARE_ADDR_EQUAL(&mSocketAddr_.sin6_addr, &v4_anyaddr);
41d93087 166}
167
168/// NOTE: Does NOT clear the Port stored. Ony the Address and Type.
b7ac5457 169void
4dd643d5 170Ip::Address::setAnyAddr()
41d93087 171{
4dd643d5 172 memset(&mSocketAddr_.sin6_addr, 0, sizeof(struct in6_addr) );
41d93087 173}
174
b7ac5457
AJ
175/// NOTE: completely empties the Ip::Address structure. Address, Port, Type, everything.
176void
4dd643d5 177Ip::Address::setEmpty()
41d93087 178{
4dd643d5 179 memset(&mSocketAddr_, 0, sizeof(mSocketAddr_) );
41d93087 180}
181
349a322f 182#if _SQUID_AIX_
d48a52bf
A
183// Bug 2885 comment 78 explains.
184// In short AIX has a different netinet/in.h union definition
349a322f
AJ
185const struct in6_addr Ip::Address::v4_localhost = {{{ 0x00000000, 0x00000000, 0x0000ffff, 0x7f000001 }}};
186const struct in6_addr Ip::Address::v4_anyaddr = {{{ 0x00000000, 0x00000000, 0x0000ffff, 0x00000000 }}};
187const struct in6_addr Ip::Address::v4_noaddr = {{{ 0x00000000, 0x00000000, 0x0000ffff, 0xffffffff }}};
188const struct in6_addr Ip::Address::v6_noaddr = {{{ 0xffffffff, 0xffffffff, 0xffffffff, 0xffffffff }}};
189#else
f53969cc
SM
190const struct in6_addr Ip::Address::v4_localhost = {{{
191 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
192 0x00, 0x00, 0xff, 0xff, 0x7f, 0x00, 0x00, 0x01
193 }
194 }
c0248e14 195};
f53969cc
SM
196const struct in6_addr Ip::Address::v4_anyaddr = {{{
197 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
198 0x00, 0x00, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00
199 }
200 }
c0248e14 201};
f53969cc
SM
202const struct in6_addr Ip::Address::v4_noaddr = {{{
203 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
204 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff
205 }
206 }
0eb08770 207};
f53969cc
SM
208const struct in6_addr Ip::Address::v6_noaddr = {{{
209 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
210 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff
211 }
212 }
c0248e14 213};
349a322f 214#endif
41d93087 215
b7ac5457 216bool
4dd643d5 217Ip::Address::setIPv4()
c0248e14 218{
4dd643d5
AJ
219 if ( isLocalhost() ) {
220 mSocketAddr_.sin6_addr = v4_localhost;
41d93087 221 return true;
222 }
41d93087 223
4dd643d5
AJ
224 if ( isAnyAddr() ) {
225 mSocketAddr_.sin6_addr = v4_anyaddr;
41d93087 226 return true;
227 }
228
4dd643d5
AJ
229 if ( isNoAddr() ) {
230 mSocketAddr_.sin6_addr = v4_noaddr;
0906f01d
CT
231 return true;
232 }
233
4dd643d5 234 if ( isIPv4())
41d93087 235 return true;
236
237 // anything non-IPv4 and non-convertable is BAD.
238 return false;
41d93087 239}
240
b7ac5457 241bool
4dd643d5 242Ip::Address::isLocalhost() const
41d93087 243{
4dd643d5 244 return IN6_IS_ADDR_LOOPBACK( &mSocketAddr_.sin6_addr ) || IN6_ARE_ADDR_EQUAL( &mSocketAddr_.sin6_addr, &v4_localhost );
41d93087 245}
246
b7ac5457 247void
4dd643d5 248Ip::Address::setLocalhost()
41d93087 249{
055421ee 250 if (Ip::EnableIpv6) {
4dd643d5
AJ
251 mSocketAddr_.sin6_addr = in6addr_loopback;
252 mSocketAddr_.sin6_family = AF_INET6;
055421ee 253 } else {
4dd643d5
AJ
254 mSocketAddr_.sin6_addr = v4_localhost;
255 mSocketAddr_.sin6_family = AF_INET;
055421ee 256 }
41d93087 257}
258
b7ac5457 259bool
4dd643d5 260Ip::Address::isSiteLocal6() const
a98c2da5 261{
dd19b634
AJ
262 // RFC 4193 the site-local allocated range is fc00::/7
263 // with fd00::/8 as the only currently allocated range (so we test it first).
264 // BUG: as of 2010-02 Linux and BSD define IN6_IS_ADDR_SITELOCAL() to check for fec::/10
4dd643d5
AJ
265 return mSocketAddr_.sin6_addr.s6_addr[0] == static_cast<uint8_t>(0xfd) ||
266 mSocketAddr_.sin6_addr.s6_addr[0] == static_cast<uint8_t>(0xfc);
a98c2da5
AJ
267}
268
b7ac5457 269bool
4dd643d5 270Ip::Address::isSiteLocalAuto() const
a98c2da5 271{
b6a91265
AJ
272 return mSocketAddr_.sin6_addr.s6_addr[11] == static_cast<uint8_t>(0xff) &&
273 mSocketAddr_.sin6_addr.s6_addr[12] == static_cast<uint8_t>(0xfe);
a98c2da5
AJ
274}
275
b7ac5457 276bool
4dd643d5 277Ip::Address::isNoAddr() const
41d93087 278{
279 // IFF the address == 0xff..ff (all ones)
4dd643d5
AJ
280 return IN6_ARE_ADDR_EQUAL( &mSocketAddr_.sin6_addr, &v6_noaddr )
281 || IN6_ARE_ADDR_EQUAL( &mSocketAddr_.sin6_addr, &v4_noaddr );
41d93087 282}
283
b7ac5457 284void
4dd643d5 285Ip::Address::setNoAddr()
41d93087 286{
4dd643d5
AJ
287 memset(&mSocketAddr_.sin6_addr, 0xFF, sizeof(struct in6_addr) );
288 mSocketAddr_.sin6_family = AF_INET6;
41d93087 289}
290
b7ac5457 291bool
4dd643d5 292Ip::Address::getReverseString6(char buf[MAX_IPSTRLEN], const struct in6_addr &dat) const
41d93087 293{
294 char *p = buf;
295 unsigned char const *r = dat.s6_addr;
296
297 /* RFC1886 says: */
298 /* 4321:0:1:2:3:4:567:89ab */
299 /* must be sent */
300 /* b.a.9.8.7.6.5.0.4.0.0.0.3.0.0.0.2.0.0.0.1.0.0.0.0.0.0.0.1.2.3.4.ip6.int. */
301
302 /* Work from the binary field. Anything else may have representation changes. */
303 /* The sin6_port and sin6_addr members shall be in network byte order. */
304
305 /* Compile Err: 'Too many arguments for format. */
306
5e263176 307 for (int i = 15; i >= 0; --i, p+=4) {
41d93087 308 snprintf(p, 5, "%x.%x.", ((r[i])&0xf), (((r[i])>>4)&0xf) );
309 }
310
311 /* RFC3152 says: */
312 /* ip6.int is now deprecated TLD, use ip6.arpa instead. */
313 snprintf(p,10,"ip6.arpa.");
314
315 return true;
316}
41d93087 317
b7ac5457 318bool
4dd643d5 319Ip::Address::getReverseString4(char buf[MAX_IPSTRLEN], const struct in_addr &dat) const
41d93087 320{
321 unsigned int i = (unsigned int) ntohl(dat.s_addr);
322 snprintf(buf, 32, "%u.%u.%u.%u.in-addr.arpa.",
323 i & 255,
324 (i >> 8) & 255,
325 (i >> 16) & 255,
326 (i >> 24) & 255);
327 return true;
328}
329
b7ac5457 330bool
4dd643d5 331Ip::Address::getReverseString(char buf[MAX_IPSTRLEN], int show_type) const
41d93087 332{
333
26ac0430 334 if (show_type == AF_UNSPEC) {
4dd643d5 335 show_type = isIPv6() ? AF_INET6 : AF_INET ;
41d93087 336 }
337
4dd643d5
AJ
338 if (show_type == AF_INET && isIPv4()) {
339 struct in_addr* tmp = (struct in_addr*)&mSocketAddr_.sin6_addr.s6_addr[12];
340 return getReverseString4(buf, *tmp);
341 } else if ( show_type == AF_INET6 && isIPv6() ) {
342 return getReverseString6(buf, mSocketAddr_.sin6_addr);
41d93087 343 }
344
4dd643d5 345 debugs(14, DBG_CRITICAL, "Unable to convert '" << toStr(buf,MAX_IPSTRLEN) << "' to the rDNS type requested.");
41d93087 346
347 buf[0] = '\0';
348
349 return false;
350}
351
b7ac5457
AJ
352Ip::Address&
353Ip::Address::operator =(const Ip::Address &s)
41d93087 354{
b7ac5457 355 memcpy(this, &s, sizeof(Ip::Address));
41d93087 356 return *this;
357};
358
b7ac5457 359Ip::Address::Address(const char*s)
41d93087 360{
4dd643d5
AJ
361 setEmpty();
362 lookupHostIP(s, true);
41d93087 363}
364
b7ac5457
AJ
365bool
366Ip::Address::operator =(const char* s)
41d93087 367{
4dd643d5 368 return lookupHostIP(s, true);
41d93087 369}
370
b7ac5457
AJ
371bool
372Ip::Address::GetHostByName(const char* s)
41d93087 373{
4dd643d5 374 return lookupHostIP(s, false);
41d93087 375}
376
b7ac5457 377bool
4dd643d5 378Ip::Address::lookupHostIP(const char *s, bool nodns)
41d93087 379{
41d93087 380 struct addrinfo want;
41d93087 381 memset(&want, 0, sizeof(struct addrinfo));
26ac0430 382 if (nodns) {
41d93087 383 want.ai_flags = AI_NUMERICHOST; // prevent actual DNS lookups!
384 }
41d93087 385
4dd643d5
AJ
386 int err = 0;
387 struct addrinfo *res = NULL;
27bc2077 388 if ( (err = getaddrinfo(s, NULL, &want, &res)) != 0) {
380de14e 389 debugs(14,3, HERE << "Given Non-IP '" << s << "': " << gai_strerror(err) );
27bc2077 390 /* free the memory getaddrinfo() dynamically allocated. */
4dd643d5 391 if (res)
27bc2077 392 freeaddrinfo(res);
41d93087 393 return false;
394 }
395
8b96fe86 396 struct addrinfo *resHead = res; // we need to free the whole list later
430b5d15
AJ
397 if (!Ip::EnableIpv6) {
398 // if we are IPv6-disabled, use first-IPv4 instead of first-IP.
399 struct addrinfo *maybeIpv4 = res;
400 while (maybeIpv4) {
401 if (maybeIpv4->ai_family == AF_INET)
402 break;
403 maybeIpv4 = maybeIpv4->ai_next;
404 }
405 if (maybeIpv4 != NULL)
406 res = maybeIpv4;
407 // else IPv6-only host, let the caller deal with first-IP anyway.
408 }
409
41d93087 410 /*
411 * NP: =(sockaddr_*) may alter the port. we don't want that.
412 * all we have been given as input was an IPA.
413 */
4dd643d5 414 short portSaved = port();
41d93087 415 operator=(*res);
4dd643d5 416 port(portSaved);
41d93087 417
27bc2077 418 /* free the memory getaddrinfo() dynamically allocated. */
8b96fe86 419 freeaddrinfo(resHead);
41d93087 420 return true;
421}
422
b7ac5457 423Ip::Address::Address(struct sockaddr_in const &s)
41d93087 424{
4dd643d5 425 setEmpty();
41d93087 426 operator=(s);
427};
428
b7ac5457
AJ
429Ip::Address &
430Ip::Address::operator =(struct sockaddr_in const &s)
41d93087 431{
4dd643d5
AJ
432 map4to6((const in_addr)s.sin_addr, mSocketAddr_.sin6_addr);
433 mSocketAddr_.sin6_port = s.sin_port;
434 mSocketAddr_.sin6_family = AF_INET6;
41d93087 435 return *this;
436};
437
b7ac5457
AJ
438Ip::Address &
439Ip::Address::operator =(const struct sockaddr_storage &s)
0e1a47f0 440{
0e1a47f0 441 /* some AF_* magic to tell socket types apart and what we need to do */
26ac0430 442 if (s.ss_family == AF_INET6) {
7da22087 443 memcpy(&mSocketAddr_, &s, sizeof(struct sockaddr_in6));
26ac0430 444 } else { // convert it to our storage mapping.
0e1a47f0 445 struct sockaddr_in *sin = (struct sockaddr_in*)&s;
4dd643d5
AJ
446 mSocketAddr_.sin6_port = sin->sin_port;
447 map4to6( sin->sin_addr, mSocketAddr_.sin6_addr);
0e1a47f0 448 }
0e1a47f0
AJ
449 return *this;
450};
451
b7ac5457 452Ip::Address::Address(struct sockaddr_in6 const &s)
41d93087 453{
4dd643d5 454 setEmpty();
41d93087 455 operator=(s);
456};
457
b7ac5457
AJ
458Ip::Address &
459Ip::Address::operator =(struct sockaddr_in6 const &s)
41d93087 460{
4dd643d5 461 memcpy(&mSocketAddr_, &s, sizeof(struct sockaddr_in6));
b5587d15 462
41d93087 463 return *this;
464};
41d93087 465
b7ac5457 466Ip::Address::Address(struct in_addr const &s)
41d93087 467{
4dd643d5 468 setEmpty();
41d93087 469 operator=(s);
470};
471
b7ac5457
AJ
472Ip::Address &
473Ip::Address::operator =(struct in_addr const &s)
41d93087 474{
4dd643d5
AJ
475 map4to6((const in_addr)s, mSocketAddr_.sin6_addr);
476 mSocketAddr_.sin6_family = AF_INET6;
41d93087 477 return *this;
478};
479
b7ac5457 480Ip::Address::Address(struct in6_addr const &s)
41d93087 481{
4dd643d5 482 setEmpty();
41d93087 483 operator=(s);
484};
485
b7ac5457
AJ
486Ip::Address &
487Ip::Address::operator =(struct in6_addr const &s)
41d93087 488{
b5587d15 489
4dd643d5
AJ
490 memcpy(&mSocketAddr_.sin6_addr, &s, sizeof(struct in6_addr));
491 mSocketAddr_.sin6_family = AF_INET6;
b5587d15 492
41d93087 493 return *this;
494};
41d93087 495
b7ac5457 496Ip::Address::Address(const Ip::Address &s)
41d93087 497{
4dd643d5 498 setEmpty();
41d93087 499 operator=(s);
500}
501
b7ac5457 502Ip::Address::Address(const struct hostent &s)
41d93087 503{
4dd643d5 504 setEmpty();
41d93087 505 operator=(s);
506}
507
b7ac5457
AJ
508bool
509Ip::Address::operator =(const struct hostent &s)
41d93087 510{
511
512 struct in_addr* ipv4 = NULL;
513
514 struct in6_addr* ipv6 = NULL;
515
516 //struct hostent {
517 // char *h_name; /* official name of host */
518 // char **h_aliases; /* alias list */
519 // int h_addrtype; /* host address type */
520 // int h_length; /* length of address */
521 // char **h_addr_list; /* list of addresses */
522 //}
523
26ac0430 524 switch (s.h_addrtype) {
41d93087 525
526 case AF_INET:
527 ipv4 = (in_addr*)(s.h_addr_list[0]);
528 /* this */
529 operator=(*ipv4);
530 break;
531
532 case AF_INET6:
533 ipv6 = (in6_addr*)(s.h_addr_list[0]);
41d93087 534 /* this */
535 operator=(*ipv6);
41d93087 536 break;
537
538 default:
26ac0430
AJ
539 IASSERT("false",false);
540 return false;
41d93087 541 }
542
543 return true;
544}
545
b7ac5457 546Ip::Address::Address(const struct addrinfo &s)
41d93087 547{
4dd643d5 548 setEmpty();
41d93087 549 operator=(s);
550}
551
b7ac5457
AJ
552bool
553Ip::Address::operator =(const struct addrinfo &s)
41d93087 554{
555
556 struct sockaddr_in* ipv4 = NULL;
557
558 struct sockaddr_in6* ipv6 = NULL;
559
560 //struct addrinfo {
561 // int ai_flags; /* input flags */
562 // int ai_family; /* protocol family for socket */
563 // int ai_socktype; /* socket type */
564 // int ai_protocol; /* protocol for socket */
565 // socklen_t ai_addrlen; /* length of socket-address */
566 // struct sockaddr *ai_addr; /* socket-address for socket */
567 // char *ai_canonname; /* canonical name for service location */
568 // struct addrinfo *ai_next; /* pointer to next in list */
569 //}
570
26ac0430 571 switch (s.ai_family) {
41d93087 572
573 case AF_INET:
574 ipv4 = (sockaddr_in*)(s.ai_addr);
575 /* this */
576 assert(ipv4);
577 operator=(*ipv4);
578 break;
579
580 case AF_INET6:
581 ipv6 = (sockaddr_in6*)(s.ai_addr);
41d93087 582 /* this */
583 assert(ipv6);
584 operator=(*ipv6);
41d93087 585 break;
586
587 case AF_UNSPEC:
588 default:
589 // attempt to handle partially initialised addrinfo.
590 // such as those where data only comes from getsockopt()
26ac0430 591 if (s.ai_addr != NULL) {
26ac0430 592 if (s.ai_addrlen == sizeof(struct sockaddr_in6)) {
41d93087 593 operator=(*((struct sockaddr_in6*)s.ai_addr));
594 return true;
055421ee
AJ
595 } else if (s.ai_addrlen == sizeof(struct sockaddr_in)) {
596 operator=(*((struct sockaddr_in*)s.ai_addr));
597 return true;
598 }
41d93087 599 }
600 return false;
601 }
602
603 return true;
604}
605
b7ac5457 606void
4dd643d5 607Ip::Address::getAddrInfo(struct addrinfo *&dst, int force) const
41d93087 608{
26ac0430 609 if (dst == NULL) {
41d93087 610 dst = new addrinfo;
611 }
612
613 memset(dst, 0, sizeof(struct addrinfo));
614
615 // set defaults
3db830d3
AJ
616 // Mac OS X does not emit a flag indicating the output is numeric (IP address)
617#if _SQUID_APPLE_
618 dst->ai_flags = 0;
619#else
41d93087 620 dst->ai_flags = AI_NUMERICHOST;
3db830d3 621#endif
41d93087 622
26ac0430 623 if (dst->ai_socktype == 0)
41d93087 624 dst->ai_socktype = SOCK_STREAM;
625
26ac0430 626 if (dst->ai_socktype == SOCK_STREAM // implies TCP
41d93087 627 && dst->ai_protocol == 0)
628 dst->ai_protocol = IPPROTO_TCP;
629
26ac0430 630 if (dst->ai_socktype == SOCK_DGRAM // implies UDP
41d93087 631 && dst->ai_protocol == 0)
632 dst->ai_protocol = IPPROTO_UDP;
633
4dd643d5 634 if (force == AF_INET6 || (force == AF_UNSPEC && Ip::EnableIpv6 && isIPv6()) ) {
41d93087 635 dst->ai_addr = (struct sockaddr*)new sockaddr_in6;
636
637 memset(dst->ai_addr,0,sizeof(struct sockaddr_in6));
638
4dd643d5 639 getSockAddr(*((struct sockaddr_in6*)dst->ai_addr));
41d93087 640
641 dst->ai_addrlen = sizeof(struct sockaddr_in6);
642
643 dst->ai_family = ((struct sockaddr_in6*)dst->ai_addr)->sin6_family;
9d92af86
AJ
644
645#if 0
646 /**
647 * Enable only if you must and please report to squid-dev if you find a need for this.
648 *
649 * Vista may need this to cope with dual-stack (unsetting IP6_V6ONLY).
650 * http://msdn.microsoft.com/en-us/library/ms738574(VS.85).aspx
651 * Linux appears to only do some things when its present.
652 * (93) Bad Protocol
653 * FreeBSD dies horribly when using dual-stack with it set.
654 * (43) Protocol not supported
655 */
41d93087 656 dst->ai_protocol = IPPROTO_IPV6;
9d92af86
AJ
657#endif
658
4dd643d5 659 } else if ( force == AF_INET || (force == AF_UNSPEC && isIPv4()) ) {
41d93087 660
055421ee 661 dst->ai_addr = (struct sockaddr*)new sockaddr_in;
41d93087 662
055421ee 663 memset(dst->ai_addr,0,sizeof(struct sockaddr_in));
41d93087 664
4dd643d5 665 getSockAddr(*((struct sockaddr_in*)dst->ai_addr));
41d93087 666
055421ee 667 dst->ai_addrlen = sizeof(struct sockaddr_in);
41d93087 668
055421ee
AJ
669 dst->ai_family = ((struct sockaddr_in*)dst->ai_addr)->sin_family;
670 } else {
671 IASSERT("false",false);
672 }
41d93087 673}
674
b7ac5457 675void
851614a8 676Ip::Address::InitAddr(struct addrinfo *&ai)
41d93087 677{
26ac0430 678 if (ai == NULL) {
41d93087 679 ai = new addrinfo;
680 memset(ai,0,sizeof(struct addrinfo));
681 }
682
683 // remove any existing data.
26ac0430 684 if (ai->ai_addr) delete ai->ai_addr;
41d93087 685
686 ai->ai_addr = (struct sockaddr*)new sockaddr_in6;
687 memset(ai->ai_addr, 0, sizeof(struct sockaddr_in6));
688
689 ai->ai_addrlen = sizeof(struct sockaddr_in6);
690
691}
692
b7ac5457 693void
851614a8 694Ip::Address::FreeAddr(struct addrinfo *&ai)
41d93087 695{
26ac0430 696 if (ai == NULL) return;
41d93087 697
26ac0430 698 if (ai->ai_addr) delete ai->ai_addr;
41d93087 699
700 ai->ai_addr = NULL;
701
702 ai->ai_addrlen = 0;
703
704 // NP: name fields are NOT allocated at present.
705 delete ai;
706
707 ai = NULL;
708}
709
b7ac5457
AJ
710int
711Ip::Address::matchIPAddr(const Ip::Address &rhs) const
41d93087 712{
4dd643d5
AJ
713 uint8_t *l = (uint8_t*)mSocketAddr_.sin6_addr.s6_addr;
714 uint8_t *r = (uint8_t*)rhs.mSocketAddr_.sin6_addr.s6_addr;
41d93087 715
716 // loop a byte-wise compare
717 // NP: match MUST be R-to-L : L-to-R produces inconsistent gt/lt results at varying CIDR
718 // expected difference on CIDR is gt/eq or lt/eq ONLY.
4dd643d5 719 for (unsigned int i = 0 ; i < sizeof(mSocketAddr_.sin6_addr) ; ++i) {
41d93087 720
26ac0430 721 if (l[i] < r[i])
41d93087 722 return -1;
723
26ac0430 724 if (l[i] > r[i])
41d93087 725 return 1;
726 }
727
728 return 0;
729}
730
a67d2b2e
AR
731int
732Ip::Address::compareWhole(const Ip::Address &rhs) const
607a7bd4
AR
733{
734 return memcmp(this, &rhs, sizeof(*this));
735}
736
b7ac5457
AJ
737bool
738Ip::Address::operator ==(const Ip::Address &s) const
41d93087 739{
740 return (0 == matchIPAddr(s));
741}
742
b7ac5457
AJ
743bool
744Ip::Address::operator !=(const Ip::Address &s) const
41d93087 745{
746 return ! ( operator==(s) );
747}
748
b7ac5457
AJ
749bool
750Ip::Address::operator <=(const Ip::Address &rhs) const
41d93087 751{
4dd643d5 752 if (isAnyAddr() && !rhs.isAnyAddr())
41d93087 753 return true;
754
755 return (matchIPAddr(rhs) <= 0);
756}
757
b7ac5457
AJ
758bool
759Ip::Address::operator >=(const Ip::Address &rhs) const
41d93087 760{
4dd643d5 761 if (isNoAddr() && !rhs.isNoAddr())
41d93087 762 return true;
763
764 return ( matchIPAddr(rhs) >= 0);
765}
766
b7ac5457
AJ
767bool
768Ip::Address::operator >(const Ip::Address &rhs) const
41d93087 769{
4dd643d5 770 if (isNoAddr() && !rhs.isNoAddr())
41d93087 771 return true;
772
773 return ( matchIPAddr(rhs) > 0);
774}
775
b7ac5457
AJ
776bool
777Ip::Address::operator <(const Ip::Address &rhs) const
41d93087 778{
4dd643d5 779 if (isAnyAddr() && !rhs.isAnyAddr())
41d93087 780 return true;
781
782 return ( matchIPAddr(rhs) < 0);
783}
784
f45dd259 785unsigned short
4dd643d5 786Ip::Address::port() const
41d93087 787{
4dd643d5 788 return ntohs( mSocketAddr_.sin6_port );
41d93087 789}
790
f45dd259 791unsigned short
4dd643d5 792Ip::Address::port(unsigned short prt)
41d93087 793{
4dd643d5 794 mSocketAddr_.sin6_port = htons(prt);
41d93087 795
796 return prt;
797}
798
799/**
4dd643d5 800 * toStr Given a buffer writes a readable ascii version of the IPA and/or port stored
41d93087 801 *
802 * Buffer must be of a size large enough to hold the converted address.
803 * This size is provided in the form of a global defined variable MAX_IPSTRLEN
804 * Should a buffer shorter be provided the string result will be truncated
805 * at the length of the available buffer.
806 *
807 * A copy of the buffer is also returned for simple immediate display.
808 */
b7ac5457 809char *
4dd643d5 810Ip::Address::toStr(char* buf, const unsigned int blen, int force) const
41d93087 811{
812 // Ensure we have a buffer.
26ac0430 813 if (buf == NULL) {
41d93087 814 return NULL;
815 }
816
817 /* some external code may have blindly memset a parent. */
818 /* thats okay, our default is known */
4dd643d5
AJ
819 if ( isAnyAddr() ) {
820 if (isIPv6())
8ea75675 821 memcpy(buf,"::\0", min(static_cast<unsigned int>(3),blen));
4dd643d5 822 else if (isIPv4())
8ea75675 823 memcpy(buf,"0.0.0.0\0", min(static_cast<unsigned int>(8),blen));
41d93087 824 return buf;
825 }
826
827 memset(buf,0,blen); // clear buffer before write
828
829 /* Pure-IPv6 CANNOT be displayed in IPv4 format. */
830 /* However IPv4 CAN. */
4dd643d5
AJ
831 if ( force == AF_INET && !isIPv4() ) {
832 if ( isIPv6() ) {
1dc746da 833 memcpy(buf, "{!IPv4}\0", min(static_cast<unsigned int>(8),blen));
41d93087 834 }
835 return buf;
836 }
837
4dd643d5 838 if ( force == AF_INET6 || (force == AF_UNSPEC && isIPv6()) ) {
41d93087 839
4dd643d5 840 inet_ntop(AF_INET6, &mSocketAddr_.sin6_addr, buf, blen);
41d93087 841
4dd643d5 842 } else if ( force == AF_INET || (force == AF_UNSPEC && isIPv4()) ) {
41d93087 843
844 struct in_addr tmp;
4dd643d5 845 getInAddr(tmp);
27bc2077 846 inet_ntop(AF_INET, &tmp, buf, blen);
41d93087 847 } else {
fa84c01d 848 debugs(14, DBG_CRITICAL, "WARNING: Corrupt IP Address details OR required to display in unknown format (" <<
26ac0430 849 force << "). accepted={" << AF_UNSPEC << "," << AF_INET << "," << AF_INET6 << "}");
41d93087 850 fprintf(stderr,"WARNING: Corrupt IP Address details OR required to display in unknown format (%d). accepted={%d,%d,%d} ",
851 force, AF_UNSPEC, AF_INET, AF_INET6);
1dc746da 852 memcpy(buf,"dead:beef::\0", min(static_cast<unsigned int>(13),blen));
41d93087 853 assert(false);
854 }
855
856 return buf;
857}
858
b7ac5457 859unsigned int
4dd643d5 860Ip::Address::toHostStr(char *buf, const unsigned int blen) const
e1381638 861{
41d93087 862 char *p = buf;
863
4dd643d5 864 if (isIPv6() && blen > 0) {
41d93087 865 *p = '[';
d7ae3534 866 ++p;
41d93087 867 }
868
188a640f 869 /* 8 being space for [ ] : and port digits */
4dd643d5
AJ
870 if ( isIPv6() )
871 toStr(p, blen-8, AF_INET6);
41d93087 872 else
4dd643d5 873 toStr(p, blen-8, AF_INET);
41d93087 874
875 // find the end of the new string
26ac0430 876 while (*p != '\0' && p < buf+blen)
d7ae3534 877 ++p;
41d93087 878
4dd643d5 879 if (isIPv6() && p < (buf+blen-1) ) {
41d93087 880 *p = ']';
d7ae3534 881 ++p;
41d93087 882 }
883
884 /* terminate just in case. */
885 *p = '\0';
886
887 /* return size of buffer now used */
888 return (p - buf);
889}
890
b7ac5457 891char *
4dd643d5 892Ip::Address::toUrl(char* buf, unsigned int blen) const
e1381638 893{
41d93087 894 char *p = buf;
895
896 // Ensure we have a buffer.
897
26ac0430 898 if (buf == NULL) {
41d93087 899 return NULL;
900 }
901
4dd643d5 902 p += toHostStr(p, blen);
41d93087 903
4dd643d5 904 if (mSocketAddr_.sin6_port > 0 && p <= (buf+blen-7) ) {
842fe4cc 905 // ':port' (short int) needs at most 6 bytes plus 1 for 0-terminator
4dd643d5 906 snprintf(p, 7, ":%d", port() );
41d93087 907 }
908
909 // force a null-terminated string
303bfd76 910 buf[blen-1] = '\0';
41d93087 911
912 return buf;
913}
914
b7ac5457 915void
4dd643d5 916Ip::Address::getSockAddr(struct sockaddr_storage &addr, const int family) const
e1381638 917{
52b694c2
AJ
918 struct sockaddr_in *sin = NULL;
919
4dd643d5 920 if ( family == AF_INET && !isIPv4()) {
0e1a47f0 921 // FIXME INET6: caller using the wrong socket type!
4dd643d5 922 debugs(14, DBG_CRITICAL, HERE << "Ip::Address::getSockAddr : Cannot convert non-IPv4 to IPv4. from " << *this);
0e1a47f0
AJ
923 assert(false);
924 }
925
4dd643d5 926 if ( family == AF_INET6 || (family == AF_UNSPEC && isIPv6()) ) {
1ef0b9ce 927 struct sockaddr_in6 *ss6 = (struct sockaddr_in6*)&addr;
4dd643d5
AJ
928 getSockAddr(*ss6);
929 } else if ( family == AF_INET || (family == AF_UNSPEC && isIPv4()) ) {
52b694c2 930 sin = (struct sockaddr_in*)&addr;
4dd643d5 931 getSockAddr(*sin);
52b694c2
AJ
932 } else {
933 IASSERT("false",false);
0e1a47f0 934 }
0e1a47f0
AJ
935}
936
b7ac5457 937void
4dd643d5 938Ip::Address::getSockAddr(struct sockaddr_in &buf) const
e1381638 939{
4dd643d5 940 if ( isIPv4() ) {
41d93087 941 buf.sin_family = AF_INET;
4dd643d5
AJ
942 buf.sin_port = mSocketAddr_.sin6_port;
943 map6to4( mSocketAddr_.sin6_addr, buf.sin_addr);
26ac0430 944 } else {
4dd643d5 945 debugs(14, DBG_CRITICAL, HERE << "Ip::Address::getSockAddr : Cannot convert non-IPv4 to IPv4. from " << *this );
41d93087 946
947 memset(&buf,0xFFFFFFFF,sizeof(struct sockaddr_in));
948 assert(false);
949 }
950
12f45551
AJ
951#if HAVE_SIN_LEN_IN_SAI
952 /* not all OS have this field, BUT when they do it can be a problem if set wrong */
1ef0b9ce 953 buf.sin_len = sizeof(struct sockaddr_in);
12f45551 954#endif
41d93087 955}
956
b7ac5457 957void
4dd643d5 958Ip::Address::getSockAddr(struct sockaddr_in6 &buf) const
e1381638 959{
4dd643d5 960 memcpy(&buf, &mSocketAddr_, sizeof(struct sockaddr_in6));
41d93087 961 /* maintain address family. It may have changed inside us. */
962 buf.sin6_family = AF_INET6;
12f45551
AJ
963
964#if HAVE_SIN6_LEN_IN_SAI
965 /* not all OS have this field, BUT when they do it can be a problem if set wrong */
1ef0b9ce 966 buf.sin6_len = sizeof(struct sockaddr_in6);
12f45551 967#endif
41d93087 968}
41d93087 969
b7ac5457 970void
4dd643d5 971Ip::Address::map4to6(const struct in_addr &in, struct in6_addr &out) const
e1381638 972{
41d93087 973 /* check for special cases */
974
26ac0430 975 if ( in.s_addr == 0x00000000) {
41d93087 976 /* ANYADDR */
0eb08770 977 out = v4_anyaddr;
26ac0430 978 } else if ( in.s_addr == 0xFFFFFFFF) {
41d93087 979 /* NOADDR */
0eb08770 980 out = v4_noaddr;
26ac0430 981 } else {
41d93087 982 /* general */
0eb08770 983 out = v4_anyaddr;
2c8fff41
AJ
984 out.s6_addr[12] = ((uint8_t *)&in.s_addr)[0];
985 out.s6_addr[13] = ((uint8_t *)&in.s_addr)[1];
986 out.s6_addr[14] = ((uint8_t *)&in.s_addr)[2];
987 out.s6_addr[15] = ((uint8_t *)&in.s_addr)[3];
41d93087 988 }
989}
990
b7ac5457 991void
4dd643d5 992Ip::Address::map6to4(const struct in6_addr &in, struct in_addr &out) const
e1381638 993{
41d93087 994 /* ANYADDR */
995 /* NOADDR */
996 /* general */
997
998 memset(&out, 0, sizeof(struct in_addr));
2c8fff41
AJ
999 ((uint8_t *)&out.s_addr)[0] = in.s6_addr[12];
1000 ((uint8_t *)&out.s_addr)[1] = in.s6_addr[13];
1001 ((uint8_t *)&out.s_addr)[2] = in.s6_addr[14];
1002 ((uint8_t *)&out.s_addr)[3] = in.s6_addr[15];
41d93087 1003}
1004
b7ac5457 1005void
68fb74b0 1006Ip::Address::getInAddr(struct in6_addr &buf) const
e1381638 1007{
4dd643d5 1008 memcpy(&buf, &mSocketAddr_.sin6_addr, sizeof(struct in6_addr));
41d93087 1009}
41d93087 1010
b7ac5457 1011bool
4dd643d5 1012Ip::Address::getInAddr(struct in_addr &buf) const
e1381638 1013{
4dd643d5 1014 if ( isIPv4() ) {
68fb74b0 1015 map6to4(mSocketAddr_.sin6_addr, buf);
41d93087 1016 return true;
1017 }
41d93087 1018
1019 // default:
1020 // non-compatible IPv6 Pure Address
1021
4dd643d5 1022 debugs(14, DBG_IMPORTANT, HERE << "Ip::Address::getInAddr : Cannot convert non-IPv4 to IPv4. IPA=" << *this);
41d93087 1023 memset(&buf,0xFFFFFFFF,sizeof(struct in_addr));
1024 assert(false);
1025 return false;
1026}
f53969cc 1027