src/common/sckaddr.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/common/sckaddr.cpp
   3 // Purpose:     Network address manager
   4 // Author:      Guilhem Lavaux
   5 // Created:     26/04/97
   6 // Modified by: Vadim Zeitlin to use wxSockAddressImpl on 2008-12-28
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) 1997, 1998 Guilhem Lavaux
   9 //              (c) 2008 Vadim Zeitlin
  10 // Licence:     wxWindows licence
  11 /////////////////////////////////////////////////////////////////////////////
  12
  13 // ============================================================================
  14 // declarations
  15 // ============================================================================
  16
  17 // ----------------------------------------------------------------------------
  18 // headers
  19 // ----------------------------------------------------------------------------
  20
  21 // For compilers that support precompilation, includes "wx.h".
  22 #include "wx/wxprec.h"
  23
  24 #ifdef __BORLANDC__
  25     #pragma hdrstop
  26 #endif
  27
  28 #if wxUSE_SOCKETS
  29
  30 #ifndef WX_PRECOMP
  31     #include "wx/object.h"
  32     #include "wx/log.h"
  33     #include "wx/intl.h"
  34     #include "wx/thread.h"
  35
  36     #include <stdio.h>
  37     #include <stdlib.h>
  38     #include <ctype.h>
  39
  40     #include <memory.h>
  41 #endif // !WX_PRECOMP
  42
  43 #include "wx/socket.h"
  44 #include "wx/sckaddr.h"
  45 #include "wx/private/socket.h"
  46 #include "wx/private/sckaddr.h"
  47
  48 #include <errno.h>
  49
  50 #if defined(__UNIX__) && !defined(__CYGWIN__)
  51     #include <netdb.h>
  52     #include <arpa/inet.h>
  53 #endif // __UNIX__
  54
  55 #ifndef INADDR_NONE
  56     #define INADDR_NONE INADDR_ANY
  57 #endif
  58
  59 // ----------------------------------------------------------------------------
  60 // wxRTTI macros
  61 // ----------------------------------------------------------------------------
  62
  63 IMPLEMENT_ABSTRACT_CLASS(wxSockAddress, wxObject)
  64 IMPLEMENT_ABSTRACT_CLASS(wxIPaddress, wxSockAddress)
  65 IMPLEMENT_DYNAMIC_CLASS(wxIPV4address, wxIPaddress)
  66 #if wxUSE_IPV6
  67 IMPLEMENT_DYNAMIC_CLASS(wxIPV6address, wxIPaddress)
  68 #endif
  69 #if defined(__UNIX__) && !defined(__WINDOWS__) && !defined(__WINE__)
  70 IMPLEMENT_DYNAMIC_CLASS(wxUNIXaddress, wxSockAddress)
  71 #endif
  72
  73 // ============================================================================
  74 // implementation of thread-safe/reentrant functions if they're missing
  75 // ============================================================================
  76
  77 // TODO: use POSIX getaddrinfo() (also available in Winsock 2) for simplicity
  78 //       and to use the same code for IPv4 and IPv6 support
  79
  80 #ifdef __WINDOWS__
  81     #define HAVE_INET_ADDR
  82
  83     #ifndef HAVE_GETHOSTBYNAME
  84     #define HAVE_GETHOSTBYNAME
  85     #endif
  86     #ifndef HAVE_GETSERVBYNAME
  87     #define HAVE_GETSERVBYNAME
  88     #endif
  89
  90     // under MSW getxxxbyname() functions are MT-safe (but not reentrant) so
  91     // we don't need to serialize calls to them
  92     #define wxHAS_MT_SAFE_GETBY_FUNCS
  93
  94     #if wxUSE_IPV6
  95         // this header does dynamic dispatching of getaddrinfo/freeaddrinfo()
  96         // by implementing them in its own code if the system versions are not
  97         // available (as is the case for anything < XP)
  98         //
  99         // NB: if this is not available for the other compilers (so far tested
 100         //      with MSVC only) we should just use wxDynamicLibrary "manually"
 101         #ifdef __VISUALC__
 102             // disable a warning occurring in Microsoft own version of this file
 103             #pragma warning(disable:4706)
 104         #endif
 105         #include <wspiapi.h>
 106         #ifdef __VISUALC__
 107             #pragma warning(default:4706)
 108         #endif
 109     #endif
 110 #endif // __WINDOWS__
 111
 112 // we assume that we have gethostbyaddr_r() if and only if we have
 113 // gethostbyname_r() and that it uses the similar conventions to it (see
 114 // comment in configure)
 115 #define HAVE_GETHOSTBYADDR HAVE_GETHOSTBYNAME
 116 #ifdef HAVE_FUNC_GETHOSTBYNAME_R_3
 117     #define HAVE_FUNC_GETHOSTBYADDR_R_3
 118 #endif
 119 #ifdef HAVE_FUNC_GETHOSTBYNAME_R_5
 120     #define HAVE_FUNC_GETHOSTBYADDR_R_5
 121 #endif
 122 #ifdef HAVE_FUNC_GETHOSTBYNAME_R_6
 123     #define HAVE_FUNC_GETHOSTBYADDR_R_6
 124 #endif
 125
 126 // the _r functions need the extra buffer parameter but unfortunately its type
 127 // differs between different systems and for the systems which use opaque
 128 // structs for it (at least AIX and OpenBSD) it must be zero-filled before
 129 // being passed to the system functions
 130 #ifdef HAVE_FUNC_GETHOSTBYNAME_R_3
 131     struct wxGethostBuf : hostent_data
 132     {
 133         wxGethostBuf()
 134         {
 135             memset(this, 0, sizeof(hostent_data));
 136         }
 137     };
 138 #else
 139     typedef char wxGethostBuf[1024];
 140 #endif
 141
 142 #ifdef HAVE_FUNC_GETSERVBYNAME_R_4
 143     struct wxGetservBuf : servent_data
 144     {
 145         wxGetservBuf()
 146         {
 147             memset(this, 0, sizeof(servent_data));
 148         }
 149     };
 150 #else
 151     typedef char wxGetservBuf[1024];
 152 #endif
 153
 154 #if defined(wxHAS_MT_SAFE_GETBY_FUNCS) || !wxUSE_THREADS
 155     #define wxLOCK_GETBY_MUTEX(name)
 156 #else // may need mutexes to protect getxxxbyxxx() calls
 157     #if defined(HAVE_GETHOSTBYNAME) || \
 158         defined(HAVE_GETHOSTBYADDR) || \
 159         defined(HAVE_GETSERVBYNAME)
 160         #include "wx/thread.h"
 161
 162         namespace
 163         {
 164             // these mutexes are used to serialize
 165             wxMutex nameLock,   // gethostbyname()
 166                     addrLock,   // gethostbyaddr()
 167                     servLock;   // getservbyname()
 168         }
 169
 170         #define wxLOCK_GETBY_MUTEX(name) wxMutexLocker locker(name ## Lock)
 171     #endif // we don't have _r functions
 172 #endif // wxUSE_THREADS
 173
 174 namespace
 175 {
 176
 177 #if defined(HAVE_GETHOSTBYNAME)
 178 hostent *deepCopyHostent(hostent *h,
 179                          const hostent *he,
 180                          char *buffer,
 181                          int size,
 182                          int *err)
 183 {
 184     /* copy old structure */
 185     memcpy(h, he, sizeof(hostent));
 186
 187     /* copy name */
 188     int len = strlen(h->h_name);
 189     if (len > size)
 190     {
 191         *err = ENOMEM;
 192         return NULL;
 193     }
 194     memcpy(buffer, h->h_name, len);
 195     buffer[len] = '\0';
 196     h->h_name = buffer;
 197
 198     /* track position in the buffer */
 199     int pos = len + 1;
 200
 201     /* reuse len to store address length */
 202     len = h->h_length;
 203
 204     /* ensure pointer alignment */
 205     unsigned int misalign = sizeof(char *) - pos%sizeof(char *);
 206     if(misalign < sizeof(char *))
 207         pos += misalign;
 208
 209     /* leave space for pointer list */
 210     char **p = h->h_addr_list, **q;
 211     char **h_addr_list = (char **)(buffer + pos);
 212     while(*(p++) != 0)
 213         pos += sizeof(char *);
 214
 215     /* copy addresses and fill new pointer list */
 216     for (p = h->h_addr_list, q = h_addr_list; *p != 0; p++, q++)
 217     {
 218         if (size < pos + len)
 219         {
 220             *err = ENOMEM;
 221             return NULL;
 222         }
 223         memcpy(buffer + pos, *p, len); /* copy content */
 224         *q = buffer + pos; /* set copied pointer to copied content */
 225         pos += len;
 226     }
 227     *++q = 0; /* null terminate the pointer list */
 228     h->h_addr_list = h_addr_list; /* copy pointer to pointers */
 229
 230     /* ensure word alignment of pointers */
 231     misalign = sizeof(char *) - pos%sizeof(char *);
 232     if(misalign < sizeof(char *))
 233         pos += misalign;
 234
 235     /* leave space for pointer list */
 236     p = h->h_aliases;
 237     char **h_aliases = (char **)(buffer + pos);
 238     while(*(p++) != 0)
 239         pos += sizeof(char *);
 240
 241     /* copy aliases and fill new pointer list */
 242     for (p = h->h_aliases, q = h_aliases; *p != 0; p++, q++)
 243     {
 244         len = strlen(*p);
 245         if (size <= pos + len)
 246         {
 247             *err = ENOMEM;
 248             return NULL;
 249         }
 250         memcpy(buffer + pos, *p, len); /* copy content */
 251         buffer[pos + len] = '\0';
 252         *q = buffer + pos; /* set copied pointer to copied content */
 253         pos += len + 1;
 254     }
 255     *++q = 0; /* null terminate the pointer list */
 256     h->h_aliases = h_aliases; /* copy pointer to pointers */
 257
 258     return h;
 259 }
 260 #endif // HAVE_GETHOSTBYNAME
 261
 262 hostent *wxGethostbyname_r(const char *hostname,
 263                            hostent *h,
 264                            wxGethostBuf buffer,
 265                            int size,
 266                            int *err)
 267 {
 268     hostent *he;
 269 #if defined(HAVE_FUNC_GETHOSTBYNAME_R_6)
 270     gethostbyname_r(hostname, h, buffer, size, &he, err);
 271 #elif defined(HAVE_FUNC_GETHOSTBYNAME_R_5)
 272     he = gethostbyname_r(hostname, h, buffer, size, err);
 273 #elif defined(HAVE_FUNC_GETHOSTBYNAME_R_3)
 274     wxUnusedVar(var);
 275     *err = gethostbyname_r(hostname, h,  &buffer);
 276     he = h;
 277 #elif defined(HAVE_GETHOSTBYNAME)
 278     wxLOCK_GETBY_MUTEX(name);
 279
 280     he = gethostbyname(hostname);
 281     *err = h_errno;
 282
 283     if ( he )
 284         he = deepCopyHostent(h, he, buffer, size, err);
 285 #else
 286     #error "No gethostbyname[_r]()"
 287 #endif
 288
 289     return he;
 290 }
 291
 292 hostent *wxGethostbyaddr_r(const char *addr_buf,
 293                            int buf_size,
 294                            int proto,
 295                            hostent *h,
 296                            wxGethostBuf buffer,
 297                            int size,
 298                            int *err)
 299 {
 300     hostent *he;
 301 #if defined(HAVE_FUNC_GETHOSTBYADDR_R_6)
 302     gethostbyaddr_r(addr_buf, buf_size, proto, h, buffer, size, &he, err);
 303 #elif defined(HAVE_FUNC_GETHOSTBYADDR_R_5)
 304     he = gethostbyaddr_r(addr_buf, buf_size, proto, h, buffer, size, err);
 305 #elif defined(HAVE_FUNC_GETHOSTBYADDR_R_3)
 306     wxUnusedVar(size);
 307     *err = gethostbyaddr_r(addr_buf, buf_size, proto, h, &buffer);
 308     he = h;
 309 #elif defined(HAVE_GETHOSTBYADDR)
 310     wxLOCK_GETBY_MUTEX(addr);
 311
 312     he = gethostbyaddr(addr_buf, buf_size, proto);
 313     *err = h_errno;
 314
 315     if ( he )
 316         he = deepCopyHostent(h, he, buffer, size, err);
 317 #else
 318     #error "No gethostbyaddr[_r]()"
 319 #endif
 320
 321     return he;
 322 }
 323
 324 #if defined(HAVE_GETSERVBYNAME)
 325 servent *deepCopyServent(servent *s,
 326                          servent *se,
 327                          char *buffer,
 328                          int size)
 329 {
 330     /* copy plain old structure */
 331     memcpy(s, se, sizeof(servent));
 332
 333     /* copy name */
 334     int len = strlen(s->s_name);
 335     if (len >= size)
 336     {
 337         return NULL;
 338     }
 339     memcpy(buffer, s->s_name, len);
 340     buffer[len] = '\0';
 341     s->s_name = buffer;
 342
 343     /* track position in the buffer */
 344     int pos = len + 1;
 345
 346     /* copy protocol */
 347     len = strlen(s->s_proto);
 348     if (pos + len >= size)
 349     {
 350         return NULL;
 351     }
 352     memcpy(buffer + pos, s->s_proto, len);
 353     buffer[pos + len] = '\0';
 354     s->s_proto = buffer + pos;
 355
 356     /* track position in the buffer */
 357     pos += len + 1;
 358
 359     /* ensure pointer alignment */
 360     unsigned int misalign = sizeof(char *) - pos%sizeof(char *);
 361     if(misalign < sizeof(char *))
 362         pos += misalign;
 363
 364     /* leave space for pointer list */
 365     char **p = s->s_aliases, **q;
 366     char **s_aliases = (char **)(buffer + pos);
 367     while(*(p++) != 0)
 368         pos += sizeof(char *);
 369
 370     /* copy addresses and fill new pointer list */
 371     for (p = s->s_aliases, q = s_aliases; *p != 0; p++, q++){
 372         len = strlen(*p);
 373         if (size <= pos + len)
 374         {
 375             return NULL;
 376         }
 377         memcpy(buffer + pos, *p, len); /* copy content */
 378         buffer[pos + len] = '\0';
 379         *q = buffer + pos; /* set copied pointer to copied content */
 380         pos += len + 1;
 381     }
 382     *++q = 0; /* null terminate the pointer list */
 383     s->s_aliases = s_aliases; /* copy pointer to pointers */
 384     return s;
 385 }
 386 #endif // HAVE_GETSERVBYNAME
 387
 388 servent *wxGetservbyname_r(const char *port,
 389                            const char *protocol,
 390                            servent *serv,
 391                            wxGetservBuf& buffer,
 392                            int size)
 393 {
 394     servent *se;
 395 #if defined(HAVE_FUNC_GETSERVBYNAME_R_6)
 396     getservbyname_r(port, protocol, serv, buffer, size, &se);
 397 #elif defined(HAVE_FUNC_GETSERVBYNAME_R_5)
 398     se = getservbyname_r(port, protocol, serv, buffer, size);
 399 #elif defined(HAVE_FUNC_GETSERVBYNAME_R_4)
 400     wxUnusedVar(size);
 401     if ( getservbyname_r(port, protocol, serv, &buffer) != 0 )
 402         return NULL;
 403 #elif defined(HAVE_GETSERVBYNAME)
 404     wxLOCK_GETBY_MUTEX(serv);
 405
 406     se = getservbyname(port, protocol);
 407     if ( se )
 408         se = deepCopyServent(serv, se, buffer, size);
 409 #else
 410     #error "No getservbyname[_r]()"
 411 #endif
 412     return se;
 413 }
 414
 415 } // anonymous namespace
 416
 417 // ============================================================================
 418 // wxSockAddressImpl implementation
 419 // ============================================================================
 420
 421 // FIXME-VC6: helper macros to call Alloc/Get() hiding the ugly dummy argument
 422 #define ALLOC(T) Alloc(static_cast<T *>(NULL))
 423 #define GET(T) Get(static_cast<T *>(NULL))
 424
 425 // ----------------------------------------------------------------------------
 426 // INET or INET6 address family
 427 // ----------------------------------------------------------------------------
 428
 429 wxString wxSockAddressImpl::GetHostName() const
 430 {
 431     const void *addrbuf;
 432     int addrbuflen;
 433
 434 #if wxUSE_IPV6
 435     if ( m_family == FAMILY_INET6 )
 436     {
 437         sockaddr_in6 * const addr6 = GET(sockaddr_in6);
 438         addrbuf = &addr6->sin6_addr;
 439         addrbuflen = sizeof(addr6->sin6_addr);
 440     }
 441     else
 442 #endif // wxUSE_IPV6
 443     {
 444         sockaddr_in * const addr = GET(sockaddr_in);
 445         if ( !addr )
 446             return wxString();
 447
 448         addrbuf = &addr->sin_addr;
 449         addrbuflen = sizeof(addr->sin_addr);
 450     }
 451
 452     hostent he;
 453     wxGethostBuf buffer;
 454     int err;
 455     if ( !wxGethostbyaddr_r
 456           (
 457             static_cast<const char *>(addrbuf),
 458             addrbuflen,
 459             m_family,
 460             &he,
 461             buffer,
 462             sizeof(buffer),
 463             &err
 464           ) )
 465     {
 466         return wxString();
 467     }
 468
 469     return wxString::FromUTF8(he.h_name);
 470 }
 471
 472 bool wxSockAddressImpl::SetPortName(const wxString& name, const char *protocol)
 473 {
 474     // test whether it's a number first
 475     unsigned long port;
 476     if ( name.ToULong(&port) )
 477     {
 478         if ( port > 65535 )
 479             return false;
 480     }
 481     else // it's a service name
 482     {
 483         wxGetservBuf buffer;
 484         servent se;
 485         if ( !wxGetservbyname_r(name.utf8_str(), protocol, &se,
 486                                 buffer, sizeof(buffer)) )
 487             return false;
 488
 489         // s_port is in network byte order and SetPort() uses the host byte
 490         // order and we prefer to reuse it from here instead of assigning to
 491         // sin_port directly
 492         port = ntohs(se.s_port);
 493     }
 494
 495     return SetPort(port);
 496 }
 497
 498 // ----------------------------------------------------------------------------
 499 // INET address family
 500 // ----------------------------------------------------------------------------
 501
 502 void wxSockAddressImpl::CreateINET()
 503 {
 504     wxASSERT_MSG( Is(FAMILY_UNSPEC), "recreating address as different type?" );
 505
 506     m_family = FAMILY_INET;
 507     sockaddr_in * const addr = ALLOC(sockaddr_in);
 508     addr->sin_family = FAMILY_INET;
 509 }
 510
 511 bool wxSockAddressImpl::SetHostName4(const wxString& name)
 512 {
 513     sockaddr_in * const addr = GET(sockaddr_in);
 514     if ( !addr )
 515         return false;
 516
 517     const wxScopedCharBuffer namebuf(name.utf8_str());
 518
 519     // first check if this is an address in quad dotted notation
 520 #if defined(HAVE_INET_ATON)
 521     if ( inet_aton(namebuf, &addr->sin_addr) )
 522         return true;
 523 #elif defined(HAVE_INET_ADDR)
 524     addr->sin_addr.s_addr = inet_addr(namebuf);
 525     if ( addr->sin_addr.s_addr != INADDR_NONE )
 526         return true;
 527 #else
 528     #error "Neither inet_aton() nor inet_addr() is available?"
 529 #endif
 530
 531     // it's a host name, resolve it
 532     hostent he;
 533     wxGethostBuf buffer;
 534     int err;
 535     if ( !wxGethostbyname_r(namebuf, &he, buffer, sizeof(buffer), &err) )
 536         return false;
 537
 538     addr->sin_addr.s_addr = ((in_addr *)he.h_addr)->s_addr;
 539     return true;
 540 }
 541
 542 bool wxSockAddressImpl::GetHostAddress(wxUint32 *address) const
 543 {
 544     sockaddr_in * const addr = GET(sockaddr_in);
 545     if ( !addr )
 546         return false;
 547
 548     *address = ntohl(addr->sin_addr.s_addr);
 549
 550     return true;
 551 }
 552
 553 bool wxSockAddressImpl::SetHostAddress(wxUint32 address)
 554 {
 555     sockaddr_in * const addr = GET(sockaddr_in);
 556     if ( !addr )
 557         return false;
 558
 559     addr->sin_addr.s_addr = htonl(address);
 560
 561     return true;
 562 }
 563
 564 wxUint16 wxSockAddressImpl::GetPort4() const
 565 {
 566     sockaddr_in * const addr = GET(sockaddr_in);
 567     if ( !addr )
 568         return 0;
 569
 570     return ntohs(addr->sin_port);
 571 }
 572
 573 bool wxSockAddressImpl::SetPort4(wxUint16 port)
 574 {
 575     sockaddr_in * const addr = GET(sockaddr_in);
 576     if ( !addr )
 577         return false;
 578
 579     addr->sin_port = htons(port);
 580
 581     return true;
 582 }
 583
 584 #if wxUSE_IPV6
 585
 586 // ----------------------------------------------------------------------------
 587 // INET6 address family
 588 // ----------------------------------------------------------------------------
 589
 590 void wxSockAddressImpl::CreateINET6()
 591 {
 592     wxASSERT_MSG( Is(FAMILY_UNSPEC), "recreating address as different type?" );
 593
 594     m_family = FAMILY_INET6;
 595     sockaddr_in6 * const addr = ALLOC(sockaddr_in6);
 596     addr->sin6_family = FAMILY_INET6;
 597 }
 598
 599 bool wxSockAddressImpl::SetHostName6(const wxString& hostname)
 600 {
 601     sockaddr_in6 * const addr = GET(sockaddr_in6);
 602     if ( !addr )
 603         return false;
 604
 605     addrinfo hints;
 606     memset(&hints, 0, sizeof(hints));
 607     hints.ai_family = AF_INET6;
 608
 609     addrinfo *info = NULL;
 610     int rc = getaddrinfo(hostname.utf8_str(), NULL, &hints, &info);
 611     if ( rc )
 612     {
 613         // use gai_strerror()?
 614         return false;
 615     }
 616
 617     wxCHECK_MSG( info, false, "should have info on success" );
 618
 619     wxASSERT_MSG( int(info->ai_addrlen) == m_len, "unexpected address length" );
 620
 621     memcpy(addr, info->ai_addr, info->ai_addrlen);
 622     freeaddrinfo(info);
 623
 624     return true;
 625 }
 626
 627 bool wxSockAddressImpl::GetHostAddress(in6_addr *address) const
 628 {
 629     sockaddr_in6 * const addr = GET(sockaddr_in6);
 630     if ( !addr )
 631         return false;
 632
 633     *address = addr->sin6_addr;
 634
 635     return true;
 636 }
 637
 638 bool wxSockAddressImpl::SetHostAddress(const in6_addr& address)
 639 {
 640     sockaddr_in6 * const addr = GET(sockaddr_in6);
 641     if ( !addr )
 642         return false;
 643
 644     addr->sin6_addr = address;
 645
 646     return true;
 647 }
 648
 649 wxUint16 wxSockAddressImpl::GetPort6() const
 650 {
 651     sockaddr_in6 * const addr = GET(sockaddr_in6);
 652     if ( !addr )
 653         return 0;
 654
 655     return ntohs(addr->sin6_port);
 656 }
 657
 658 bool wxSockAddressImpl::SetPort6(wxUint16 port)
 659 {
 660     sockaddr_in6 * const addr = GET(sockaddr_in6);
 661     if ( !addr )
 662         return false;
 663
 664     addr->sin6_port = htons(port);
 665
 666     return true;
 667 }
 668
 669 bool wxSockAddressImpl::SetToAnyAddress6()
 670 {
 671     static const in6_addr any = IN6ADDR_ANY_INIT;
 672
 673     return SetHostAddress(any);
 674 }
 675
 676 #endif // wxUSE_IPV6
 677
 678 #ifdef wxHAS_UNIX_DOMAIN_SOCKETS
 679
 680 // ----------------------------------------------------------------------------
 681 // Unix address family
 682 // ----------------------------------------------------------------------------
 683
 684 #ifndef UNIX_PATH_MAX
 685     #define UNIX_PATH_MAX (WXSIZEOF(((sockaddr_un *)NULL)->sun_path))
 686 #endif
 687
 688 void wxSockAddressImpl::CreateUnix()
 689 {
 690     wxASSERT_MSG( Is(FAMILY_UNSPEC), "recreating address as different type?" );
 691
 692     m_family = FAMILY_UNIX;
 693     sockaddr_un * const addr = ALLOC(sockaddr_un);
 694     addr->sun_family = FAMILY_UNIX;
 695     addr->sun_path[0] = '\0';
 696 }
 697
 698 bool wxSockAddressImpl::SetPath(const wxString& path)
 699 {
 700     sockaddr_un * const addr = GET(sockaddr_un);
 701     if ( !addr )
 702         return false;
 703
 704     const wxScopedCharBuffer buf(path.utf8_str());
 705     if ( strlen(buf) >= UNIX_PATH_MAX )
 706         return false;
 707
 708     wxStrlcpy(addr->sun_path, buf, UNIX_PATH_MAX);
 709
 710     return true;
 711 }
 712
 713 wxString wxSockAddressImpl::GetPath() const
 714 {
 715     sockaddr_un * const addr = GET(sockaddr_un);
 716     if ( !addr )
 717         return wxString();
 718
 719     return wxString::FromUTF8(addr->sun_path);
 720 }
 721
 722 #endif // wxHAS_UNIX_DOMAIN_SOCKETS
 723
 724 #undef GET
 725 #undef ALLOC
 726
 727 // ----------------------------------------------------------------------------
 728 // wxSockAddress
 729 // ----------------------------------------------------------------------------
 730
 731 const sockaddr *wxSockAddress::GetAddressData() const
 732 {
 733     return GetAddress().GetAddr();
 734 }
 735
 736 int wxSockAddress::GetAddressDataLen() const
 737 {
 738     return GetAddress().GetLen();
 739 }
 740
 741 void wxSockAddress::Init()
 742 {
 743     if ( wxIsMainThread() && !wxSocketBase::IsInitialized() )
 744     {
 745         // we must do it before using any socket functions
 746         (void)wxSocketBase::Initialize();
 747     }
 748 }
 749
 750 wxSockAddress::wxSockAddress()
 751 {
 752     Init();
 753
 754     m_impl = new wxSockAddressImpl();
 755 }
 756
 757 wxSockAddress::wxSockAddress(const wxSockAddress& other)
 758     : wxObject()
 759 {
 760     Init();
 761
 762     m_impl = new wxSockAddressImpl(*other.m_impl);
 763 }
 764
 765 wxSockAddress::~wxSockAddress()
 766 {
 767     delete m_impl;
 768 }
 769
 770 void wxSockAddress::SetAddress(const wxSockAddressImpl& address)
 771 {
 772     if ( &address != m_impl )
 773     {
 774         delete m_impl;
 775         m_impl = new wxSockAddressImpl(address);
 776     }
 777 }
 778
 779 wxSockAddress& wxSockAddress::operator=(const wxSockAddress& addr)
 780 {
 781     SetAddress(addr.GetAddress());
 782
 783     return *this;
 784 }
 785
 786 void wxSockAddress::Clear()
 787 {
 788     m_impl->Clear();
 789 }
 790
 791 // ----------------------------------------------------------------------------
 792 // wxIPaddress
 793 // ----------------------------------------------------------------------------
 794
 795 wxSockAddressImpl& wxIPaddress::GetImpl()
 796 {
 797     if ( m_impl->GetFamily() == wxSockAddressImpl::FAMILY_UNSPEC )
 798         m_impl->CreateINET();
 799
 800     return *m_impl;
 801 }
 802
 803 bool wxIPaddress::Hostname(const wxString& name)
 804 {
 805     wxCHECK_MSG( !name.empty(), false, "empty host name is invalid" );
 806
 807     m_origHostname = name;
 808
 809     return GetImpl().SetHostName(name);
 810 }
 811
 812 bool wxIPaddress::Service(const wxString& name)
 813 {
 814     return GetImpl().SetPortName(name, "tcp");
 815 }
 816
 817 bool wxIPaddress::Service(unsigned short port)
 818 {
 819     return GetImpl().SetPort(port);
 820 }
 821
 822 bool wxIPaddress::LocalHost()
 823 {
 824     return Hostname("localhost");
 825 }
 826
 827 wxString wxIPaddress::Hostname() const
 828 {
 829     return GetImpl().GetHostName();
 830 }
 831
 832 unsigned short wxIPaddress::Service() const
 833 {
 834     return GetImpl().GetPort();
 835 }
 836
 837 bool wxIPaddress::operator==(const wxIPaddress& addr) const
 838 {
 839     return Hostname().Cmp(addr.Hostname()) == 0 &&
 840            Service() == addr.Service();
 841 }
 842
 843 bool wxIPaddress::AnyAddress()
 844 {
 845     return GetImpl().SetToAnyAddress();
 846 }
 847
 848 // ----------------------------------------------------------------------------
 849 // wxIPV4address
 850 // ----------------------------------------------------------------------------
 851
 852 void wxIPV4address::DoInitImpl()
 853 {
 854     m_impl->CreateINET();
 855 }
 856
 857 bool wxIPV4address::Hostname(unsigned long addr)
 858 {
 859     if ( !GetImpl().SetHostAddress(addr) )
 860     {
 861         m_origHostname.clear();
 862         return false;
 863     }
 864
 865     m_origHostname = Hostname();
 866     return true;
 867 }
 868
 869 bool wxIPV4address::IsLocalHost() const
 870 {
 871     return Hostname() == "localhost" || IPAddress() == "127.0.0.1";
 872 }
 873
 874 wxString wxIPV4address::IPAddress() const
 875 {
 876     wxUint32 addr;
 877     if ( !GetImpl().GetHostAddress(&addr) )
 878         return wxString();
 879
 880     return wxString::Format
 881            (
 882             "%u.%u.%u.%u",
 883             (addr >> 24) & 0xff,
 884             (addr >> 16) & 0xff,
 885             (addr >> 8) & 0xff,
 886             addr & 0xff
 887            );
 888 }
 889
 890 bool wxIPV4address::BroadcastAddress()
 891 {
 892     return GetImpl().SetToBroadcastAddress();
 893 }
 894
 895 #if wxUSE_IPV6
 896
 897 // ---------------------------------------------------------------------------
 898 // wxIPV6address
 899 // ---------------------------------------------------------------------------
 900
 901 void wxIPV6address::DoInitImpl()
 902 {
 903     m_impl->CreateINET6();
 904 }
 905
 906 bool wxIPV6address::Hostname(unsigned char addr[16])
 907 {
 908     unsigned short wk[8];
 909     for ( int i = 0; i < 8; ++i )
 910     {
 911         wk[i] = addr[2*i];
 912         wk[i] <<= 8;
 913         wk[i] |= addr[2*i+1];
 914     }
 915
 916     return Hostname
 917            (
 918                 wxString::Format
 919                 (
 920                  "%x:%x:%x:%x:%x:%x:%x:%x",
 921                  wk[0], wk[1], wk[2], wk[3], wk[4], wk[5], wk[6], wk[7]
 922                 )
 923            );
 924 }
 925
 926 bool wxIPV6address::IsLocalHost() const
 927 {
 928     if ( Hostname() == "localhost" )
 929         return true;
 930
 931     wxString addr = IPAddress();
 932     return addr == wxT("::1") ||
 933                 addr == wxT("0:0:0:0:0:0:0:1") ||
 934                     addr == wxT("::ffff:127.0.0.1");
 935 }
 936
 937 wxString wxIPV6address::IPAddress() const
 938 {
 939     union
 940     {
 941         in6_addr addr6;
 942         wxUint8 bytes[16];
 943     } u;
 944
 945     if ( !GetImpl().GetHostAddress(&u.addr6) )
 946         return wxString();
 947
 948     const wxUint8 * const addr = u.bytes;
 949
 950     wxUint16 words[8];
 951     int i,
 952         prefix_zero_count = 0;
 953     for ( i = 0; i < 8; ++i )
 954     {
 955         words[i] = addr[i*2];
 956         words[i] <<= 8;
 957         words[i] |= addr[i*2+1];
 958         if ( i == prefix_zero_count && words[i] == 0 )
 959             ++prefix_zero_count;
 960     }
 961
 962     wxString result;
 963     if ( prefix_zero_count == 8 )
 964     {
 965         result = wxT( "::" );
 966     }
 967     else if ( prefix_zero_count == 6 && words[5] == 0xFFFF )
 968     {
 969         // IPv4 mapped
 970         result.Printf("::ffff:%d.%d.%d.%d",
 971                       addr[12], addr[13], addr[14], addr[15]);
 972     }
 973     else // general case
 974     {
 975         result = ":";
 976         for ( i = prefix_zero_count; i < 8; ++i )
 977         {
 978             result += wxString::Format(":%x", words[i]);
 979         }
 980     }
 981
 982     return result;
 983 }
 984
 985 #endif // wxUSE_IPV6
 986
 987 #ifdef wxHAS_UNIX_DOMAIN_SOCKETS
 988
 989 // ---------------------------------------------------------------------------
 990 // wxUNIXaddress
 991 // ---------------------------------------------------------------------------
 992
 993 wxSockAddressImpl& wxUNIXaddress::GetUNIX()
 994 {
 995     if ( m_impl->GetFamily() == wxSockAddressImpl::FAMILY_UNSPEC )
 996         m_impl->CreateUnix();
 997
 998     return *m_impl;
 999 }
1000
1001 void wxUNIXaddress::Filename(const wxString& fname)
1002 {
1003     GetUNIX().SetPath(fname);
1004 }
1005
1006 wxString wxUNIXaddress::Filename() const
1007 {
1008     return GetUNIX().GetPath();
1009 }
1010
1011 #endif // wxHAS_UNIX_DOMAIN_SOCKETS
1012
1013 #endif // wxUSE_SOCKETS