gio/gnetworkaddress.c

   1 /* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*- */
   2
   3 /* GIO - GLib Input, Output and Streaming Library
   4  *
   5  * Copyright (C) 2008 Red Hat, Inc.
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General
  18  * Public License along with this library; if not, write to the
  19  * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
  20  * Boston, MA 02111-1307, USA.
  21  */
  22
  23 #include "config.h"
  24 #include <glib.h>
  25 #include "glibintl.h"
  26
  27 #include <stdlib.h>
  28 #include "gnetworkaddress.h"
  29 #include "gasyncresult.h"
  30 #include "ginetaddress.h"
  31 #include "ginetsocketaddress.h"
  32 #include "gnetworkingprivate.h"
  33 #include "gproxyaddressenumerator.h"
  34 #include "gresolver.h"
  35 #include "gtask.h"
  36 #include "gsocketaddressenumerator.h"
  37 #include "gioerror.h"
  38 #include "gsocketconnectable.h"
  39
  40 #include <string.h>
  41
  42
  43 /**
  44  * SECTION:gnetworkaddress
  45  * @short_description: A GSocketConnectable for resolving hostnames
  46  * @include: gio/gio.h
  47  *
  48  * #GNetworkAddress provides an easy way to resolve a hostname and
  49  * then attempt to connect to that host, handling the possibility of
  50  * multiple IP addresses and multiple address families.
  51  *
  52  * See #GSocketConnectable for and example of using the connectable
  53  * interface.
  54  */
  55
  56 /**
  57  * GNetworkAddress:
  58  *
  59  * A #GSocketConnectable for resolving a hostname and connecting to
  60  * that host.
  61  */
  62
  63 struct _GNetworkAddressPrivate {
  64   gchar *hostname;
  65   guint16 port;
  66   GList *sockaddrs;
  67   gchar *scheme;
  68
  69   gint64 resolver_serial;
  70 };
  71
  72 enum {
  73   PROP_0,
  74   PROP_HOSTNAME,
  75   PROP_PORT,
  76   PROP_SCHEME,
  77 };
  78
  79 static void g_network_address_set_property (GObject      *object,
  80                                             guint         prop_id,
  81                                             const GValue *value,
  82                                             GParamSpec   *pspec);
  83 static void g_network_address_get_property (GObject      *object,
  84                                             guint         prop_id,
  85                                             GValue       *value,
  86                                             GParamSpec   *pspec);
  87
  88 static void                      g_network_address_connectable_iface_init       (GSocketConnectableIface *iface);
  89 static GSocketAddressEnumerator *g_network_address_connectable_enumerate        (GSocketConnectable      *connectable);
  90 static GSocketAddressEnumerator *g_network_address_connectable_proxy_enumerate  (GSocketConnectable      *connectable);
  91
  92 G_DEFINE_TYPE_WITH_CODE (GNetworkAddress, g_network_address, G_TYPE_OBJECT,
  93                          G_ADD_PRIVATE (GNetworkAddress)
  94                          G_IMPLEMENT_INTERFACE (G_TYPE_SOCKET_CONNECTABLE,
  95                                                 g_network_address_connectable_iface_init))
  96
  97 static void
  98 g_network_address_finalize (GObject *object)
  99 {
 100   GNetworkAddress *addr = G_NETWORK_ADDRESS (object);
 101
 102   g_free (addr->priv->hostname);
 103   g_free (addr->priv->scheme);
 104
 105   if (addr->priv->sockaddrs)
 106     {
 107       GList *a;
 108
 109       for (a = addr->priv->sockaddrs; a; a = a->next)
 110         g_object_unref (a->data);
 111       g_list_free (addr->priv->sockaddrs);
 112     }
 113
 114   G_OBJECT_CLASS (g_network_address_parent_class)->finalize (object);
 115 }
 116
 117 static void
 118 g_network_address_class_init (GNetworkAddressClass *klass)
 119 {
 120   GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
 121
 122   gobject_class->set_property = g_network_address_set_property;
 123   gobject_class->get_property = g_network_address_get_property;
 124   gobject_class->finalize = g_network_address_finalize;
 125
 126   g_object_class_install_property (gobject_class, PROP_HOSTNAME,
 127                                    g_param_spec_string ("hostname",
 128                                                         P_("Hostname"),
 129                                                         P_("Hostname to resolve"),
 130                                                         NULL,
 131                                                         G_PARAM_READWRITE |
 132                                                         G_PARAM_CONSTRUCT_ONLY |
 133                                                         G_PARAM_STATIC_STRINGS));
 134   g_object_class_install_property (gobject_class, PROP_PORT,
 135                                    g_param_spec_uint ("port",
 136                                                       P_("Port"),
 137                                                       P_("Network port"),
 138                                                       0, 65535, 0,
 139                                                       G_PARAM_READWRITE |
 140                                                       G_PARAM_CONSTRUCT_ONLY |
 141                                                       G_PARAM_STATIC_STRINGS));
 142
 143   g_object_class_install_property (gobject_class, PROP_SCHEME,
 144                                    g_param_spec_string ("scheme",
 145                                                         P_("Scheme"),
 146                                                         P_("URI Scheme"),
 147                                                         NULL,
 148                                                         G_PARAM_READWRITE |
 149                                                         G_PARAM_CONSTRUCT_ONLY |
 150                                                         G_PARAM_STATIC_STRINGS));
 151 }
 152
 153 static void
 154 g_network_address_connectable_iface_init (GSocketConnectableIface *connectable_iface)
 155 {
 156   connectable_iface->enumerate  = g_network_address_connectable_enumerate;
 157   connectable_iface->proxy_enumerate = g_network_address_connectable_proxy_enumerate;
 158 }
 159
 160 static void
 161 g_network_address_init (GNetworkAddress *addr)
 162 {
 163   addr->priv = g_network_address_get_instance_private (addr);
 164 }
 165
 166 static void
 167 g_network_address_set_property (GObject      *object,
 168                                 guint         prop_id,
 169                                 const GValue *value,
 170                                 GParamSpec   *pspec)
 171 {
 172   GNetworkAddress *addr = G_NETWORK_ADDRESS (object);
 173
 174   switch (prop_id)
 175     {
 176     case PROP_HOSTNAME:
 177       g_free (addr->priv->hostname);
 178       addr->priv->hostname = g_value_dup_string (value);
 179       break;
 180
 181     case PROP_PORT:
 182       addr->priv->port = g_value_get_uint (value);
 183       break;
 184
 185     case PROP_SCHEME:
 186       if (addr->priv->scheme)
 187         g_free (addr->priv->scheme);
 188       addr->priv->scheme = g_value_dup_string (value);
 189       break;
 190
 191     default:
 192       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 193       break;
 194     }
 195
 196 }
 197
 198 static void
 199 g_network_address_get_property (GObject    *object,
 200                                 guint       prop_id,
 201                                 GValue     *value,
 202                                 GParamSpec *pspec)
 203 {
 204   GNetworkAddress *addr = G_NETWORK_ADDRESS (object);
 205
 206   switch (prop_id)
 207     {
 208     case PROP_HOSTNAME:
 209       g_value_set_string (value, addr->priv->hostname);
 210       break;
 211
 212     case PROP_PORT:
 213       g_value_set_uint (value, addr->priv->port);
 214       break;
 215
 216     case PROP_SCHEME:
 217       g_value_set_string (value, addr->priv->scheme);
 218       break;
 219
 220     default:
 221       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 222       break;
 223     }
 224
 225 }
 226
 227 static void
 228 g_network_address_set_addresses (GNetworkAddress *addr,
 229                                  GList           *addresses,
 230                                  guint64          resolver_serial)
 231 {
 232   GList *a;
 233   GSocketAddress *sockaddr;
 234
 235   g_return_if_fail (addresses != NULL && addr->priv->sockaddrs == NULL);
 236
 237   for (a = addresses; a; a = a->next)
 238     {
 239       sockaddr = g_inet_socket_address_new (a->data, addr->priv->port);
 240       addr->priv->sockaddrs = g_list_prepend (addr->priv->sockaddrs, sockaddr);
 241       g_object_unref (a->data);
 242     }
 243   g_list_free (addresses);
 244   addr->priv->sockaddrs = g_list_reverse (addr->priv->sockaddrs);
 245
 246   addr->priv->resolver_serial = resolver_serial;
 247 }
 248
 249 static gboolean
 250 g_network_address_parse_sockaddr (GNetworkAddress *addr)
 251 {
 252   struct addrinfo hints, *res = NULL;
 253   GSocketAddress *sockaddr;
 254   gchar port[32];
 255
 256   memset (&hints, 0, sizeof (hints));
 257   hints.ai_flags = AI_NUMERICHOST
 258 #ifdef AI_NUMERICSERV
 259     | AI_NUMERICSERV
 260 #endif
 261     ;
 262   g_snprintf (port, sizeof (port), "%u", addr->priv->port);
 263
 264   if (getaddrinfo (addr->priv->hostname, port, &hints, &res) != 0)
 265     return FALSE;
 266
 267   sockaddr = g_socket_address_new_from_native (res->ai_addr, res->ai_addrlen);
 268   freeaddrinfo (res);
 269   if (!sockaddr || !G_IS_INET_SOCKET_ADDRESS (sockaddr))
 270     return FALSE;
 271
 272   addr->priv->sockaddrs = g_list_prepend (addr->priv->sockaddrs, sockaddr);
 273   return TRUE;
 274 }
 275
 276 /**
 277  * g_network_address_new:
 278  * @hostname: the hostname
 279  * @port: the port
 280  *
 281  * Creates a new #GSocketConnectable for connecting to the given
 282  * @hostname and @port.
 283  *
 284  * Return value: (transfer full) (type GNetworkAddress): the new #GNetworkAddress
 285  *
 286  * Since: 2.22
 287  */
 288 GSocketConnectable *
 289 g_network_address_new (const gchar *hostname,
 290                        guint16      port)
 291 {
 292   return g_object_new (G_TYPE_NETWORK_ADDRESS,
 293                        "hostname", hostname,
 294                        "port", port,
 295                        NULL);
 296 }
 297
 298 /**
 299  * g_network_address_parse:
 300  * @host_and_port: the hostname and optionally a port
 301  * @default_port: the default port if not in @host_and_port
 302  * @error: a pointer to a #GError, or %NULL
 303  *
 304  * Creates a new #GSocketConnectable for connecting to the given
 305  * @hostname and @port. May fail and return %NULL in case
 306  * parsing @host_and_port fails.
 307  *
 308  * @host_and_port may be in any of a number of recognised formats; an IPv6
 309  * address, an IPv4 address, or a domain name (in which case a DNS
 310  * lookup is performed). Quoting with [] is supported for all address
 311  * types. A port override may be specified in the usual way with a
 312  * colon.
 313  *
 314  * If no port is specified in @host_and_port then @default_port will be
 315  * used as the port number to connect to.
 316  *
 317  * In general, @host_and_port is expected to be provided by the user
 318  * (allowing them to give the hostname, and a port overide if necessary)
 319  * and @default_port is expected to be provided by the application.
 320  *
 321  * (The port component of @host_and_port can also be specified as a
 322  * service name rather than as a numeric port, but this functionality
 323  * is deprecated, because it depends on the contents of /etc/services,
 324  * which is generally quite sparse on platforms other than Linux.)
 325  *
 326  * Return value: (transfer full): the new #GNetworkAddress, or %NULL on error
 327  *
 328  * Since: 2.22
 329  */
 330 GSocketConnectable *
 331 g_network_address_parse (const gchar  *host_and_port,
 332                          guint16       default_port,
 333                          GError      **error)
 334 {
 335   GSocketConnectable *connectable;
 336   const gchar *port;
 337   guint16 portnum;
 338   gchar *name;
 339
 340   g_return_val_if_fail (host_and_port != NULL, NULL);
 341
 342   port = NULL;
 343   if (host_and_port[0] == '[')
 344     /* escaped host part (to allow, eg. "[2001:db8::1]:888") */
 345     {
 346       const gchar *end;
 347
 348       end = strchr (host_and_port, ']');
 349       if (end == NULL)
 350         {
 351           g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
 352                        _("Hostname '%s' contains '[' but not ']'"), host_and_port);
 353           return NULL;
 354         }
 355
 356       if (end[1] == '\0')
 357         port = NULL;
 358       else if (end[1] == ':')
 359         port = &end[2];
 360       else
 361         {
 362           g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
 363                        "The ']' character (in hostname '%s') must come at the"
 364                        " end or be immediately followed by ':' and a port",
 365                        host_and_port);
 366           return NULL;
 367         }
 368
 369       name = g_strndup (host_and_port + 1, end - host_and_port - 1);
 370     }
 371
 372   else if ((port = strchr (host_and_port, ':')))
 373     /* string has a ':' in it */
 374     {
 375       /* skip ':' */
 376       port++;
 377
 378       if (strchr (port, ':'))
 379         /* more than one ':' in string */
 380         {
 381           /* this is actually an unescaped IPv6 address */
 382           name = g_strdup (host_and_port);
 383           port = NULL;
 384         }
 385       else
 386         name = g_strndup (host_and_port, port - host_and_port - 1);
 387     }
 388
 389   else
 390     /* plain hostname, no port */
 391     name = g_strdup (host_and_port);
 392
 393   if (port != NULL)
 394     {
 395       if (port[0] == '\0')
 396         {
 397           g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
 398                        "If a ':' character is given, it must be followed by a "
 399                        "port (in hostname '%s').", host_and_port);
 400           g_free (name);
 401           return NULL;
 402         }
 403
 404       else if ('0' <= port[0] && port[0] <= '9')
 405         {
 406           char *end;
 407           long value;
 408
 409           value = strtol (port, &end, 10);
 410           if (*end != '\0' || value < 0 || value > G_MAXUINT16)
 411             {
 412               g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
 413                            "Invalid numeric port '%s' specified in hostname '%s'",
 414                            port, host_and_port);
 415               g_free (name);
 416               return NULL;
 417             }
 418
 419           portnum = value;
 420         }
 421
 422       else
 423         {
 424           struct servent *entry;
 425
 426           entry = getservbyname (port, "tcp");
 427           if (entry == NULL)
 428             {
 429               g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
 430                            "Unknown service '%s' specified in hostname '%s'",
 431                            port, host_and_port);
 432 #ifdef HAVE_ENDSERVENT
 433               endservent ();
 434 #endif
 435               g_free (name);
 436               return NULL;
 437             }
 438
 439           portnum = g_ntohs (entry->s_port);
 440
 441 #ifdef HAVE_ENDSERVENT
 442           endservent ();
 443 #endif
 444         }
 445     }
 446   else
 447     {
 448       /* No port in host_and_port */
 449       portnum = default_port;
 450     }
 451
 452   connectable = g_network_address_new (name, portnum);
 453   g_free (name);
 454
 455   return connectable;
 456 }
 457
 458 /* Allowed characters outside alphanumeric for unreserved. */
 459 #define G_URI_OTHER_UNRESERVED "-._~"
 460
 461 /* This or something equivalent will eventually go into glib/guri.h */
 462 gboolean
 463 _g_uri_parse_authority (const char  *uri,
 464                         char       **host,
 465                         guint16     *port,
 466                         char       **userinfo)
 467 {
 468   char *tmp_str;
 469   const char *start, *p;
 470   char c;
 471
 472   g_return_val_if_fail (uri != NULL, FALSE);
 473
 474   if (host)
 475     *host = NULL;
 476
 477   if (port)
 478     *port = 0;
 479
 480   if (userinfo)
 481     *userinfo = NULL;
 482
 483   /* From RFC 3986 Decodes:
 484    * URI          = scheme ":" hier-part [ "?" query ] [ "#" fragment ]
 485    * hier-part    = "//" authority path-abempty
 486    * path-abempty = *( "/" segment )
 487    * authority    = [ userinfo "@" ] host [ ":" port ]
 488    */
 489
 490   /* Check we have a valid scheme */
 491   tmp_str = g_uri_parse_scheme (uri);
 492
 493   if (tmp_str == NULL)
 494     return FALSE;
 495
 496   g_free (tmp_str);
 497
 498   /* Decode hier-part:
 499    *  hier-part   = "//" authority path-abempty
 500    */
 501   p = uri;
 502   start = strstr (p, "//");
 503
 504   if (start == NULL)
 505     return FALSE;
 506
 507   start += 2;
 508
 509   if (strchr (start, '@') != NULL)
 510     {
 511       /* Decode userinfo:
 512        * userinfo      = *( unreserved / pct-encoded / sub-delims / ":" )
 513        * unreserved    = ALPHA / DIGIT / "-" / "." / "_" / "~"
 514        * pct-encoded   = "%" HEXDIG HEXDIG
 515        */
 516       p = start;
 517       while (1)
 518         {
 519           c = *p++;
 520
 521           if (c == '@')
 522             break;
 523
 524           /* pct-encoded */
 525           if (c == '%')
 526             {
 527               if (!(g_ascii_isxdigit (p[0]) ||
 528                     g_ascii_isxdigit (p[1])))
 529                 return FALSE;
 530
 531               p++;
 532
 533               continue;
 534             }
 535
 536           /* unreserved /  sub-delims / : */
 537           if (!(g_ascii_isalnum (c) ||
 538                 strchr (G_URI_OTHER_UNRESERVED, c) ||
 539                 strchr (G_URI_RESERVED_CHARS_SUBCOMPONENT_DELIMITERS, c) ||
 540                 c == ':'))
 541             return FALSE;
 542         }
 543
 544       if (userinfo)
 545         *userinfo = g_strndup (start, p - start - 1);
 546
 547       start = p;
 548     }
 549   else
 550     {
 551       p = start;
 552     }
 553
 554
 555   /* decode host:
 556    * host          = IP-literal / IPv4address / reg-name
 557    * reg-name      = *( unreserved / pct-encoded / sub-delims )
 558    */
 559
 560   /* If IPv6 or IPvFuture */
 561   if (*p == '[')
 562     {
 563       start++;
 564       p++;
 565       while (1)
 566         {
 567           c = *p++;
 568
 569           if (c == ']')
 570             break;
 571
 572           /* unreserved /  sub-delims */
 573           if (!(g_ascii_isalnum (c) ||
 574                 strchr (G_URI_OTHER_UNRESERVED, c) ||
 575                 strchr (G_URI_RESERVED_CHARS_SUBCOMPONENT_DELIMITERS, c) ||
 576                 c == ':' ||
 577                 c == '.'))
 578             goto error;
 579         }
 580     }
 581   else
 582     {
 583       while (1)
 584         {
 585           c = *p++;
 586
 587           if (c == ':' ||
 588               c == '/' ||
 589               c == '?' ||
 590               c == '#' ||
 591               c == '\0')
 592             break;
 593
 594           /* pct-encoded */
 595           if (c == '%')
 596             {
 597               if (!(g_ascii_isxdigit (p[0]) ||
 598                     g_ascii_isxdigit (p[1])))
 599                 goto error;
 600
 601               p++;
 602
 603               continue;
 604             }
 605
 606           /* unreserved /  sub-delims */
 607           if (!(g_ascii_isalnum (c) ||
 608                 strchr (G_URI_OTHER_UNRESERVED, c) ||
 609                 strchr (G_URI_RESERVED_CHARS_SUBCOMPONENT_DELIMITERS, c)))
 610             goto error;
 611         }
 612     }
 613
 614   if (host)
 615     *host = g_uri_unescape_segment (start, p - 1, NULL);
 616
 617   if (c == ':')
 618     {
 619       /* Decode pot:
 620        *  port          = *DIGIT
 621        */
 622       guint tmp = 0;
 623
 624       while (1)
 625         {
 626           c = *p++;
 627
 628           if (c == '/' ||
 629               c == '?' ||
 630               c == '#' ||
 631               c == '\0')
 632             break;
 633
 634           if (!g_ascii_isdigit (c))
 635             goto error;
 636
 637           tmp = (tmp * 10) + (c - '0');
 638
 639           if (tmp > 65535)
 640             goto error;
 641         }
 642       if (port)
 643         *port = (guint16) tmp;
 644     }
 645
 646   return TRUE;
 647
 648 error:
 649   if (host && *host)
 650     {
 651       g_free (*host);
 652       *host = NULL;
 653     }
 654
 655   if (userinfo && *userinfo)
 656     {
 657       g_free (*userinfo);
 658       *userinfo = NULL;
 659     }
 660
 661   return FALSE;
 662 }
 663
 664 gchar *
 665 _g_uri_from_authority (const gchar *protocol,
 666                        const gchar *host,
 667                        guint        port,
 668                        const gchar *userinfo)
 669 {
 670   GString *uri;
 671
 672   uri = g_string_new (protocol);
 673   g_string_append (uri, "://");
 674
 675   if (userinfo)
 676     {
 677       g_string_append_uri_escaped (uri, userinfo, G_URI_RESERVED_CHARS_ALLOWED_IN_USERINFO, FALSE);
 678       g_string_append_c (uri, '@');
 679     }
 680
 681   if (g_hostname_is_non_ascii (host))
 682     {
 683       gchar *ace_encoded = g_hostname_to_ascii (host);
 684
 685       if (!ace_encoded)
 686         {
 687           g_string_free (uri, TRUE);
 688           return NULL;
 689         }
 690       g_string_append (uri, ace_encoded);
 691       g_free (ace_encoded);
 692     }
 693   else if (strchr (host, ':'))
 694     g_string_append_printf (uri, "[%s]", host);
 695   else
 696     g_string_append (uri, host);
 697
 698   if (port != 0)
 699     g_string_append_printf (uri, ":%u", port);
 700
 701   return g_string_free (uri, FALSE);
 702 }
 703
 704 /**
 705  * g_network_address_parse_uri:
 706  * @uri: the hostname and optionally a port
 707  * @default_port: The default port if none is found in the URI
 708  * @error: a pointer to a #GError, or %NULL
 709  *
 710  * Creates a new #GSocketConnectable for connecting to the given
 711  * @uri. May fail and return %NULL in case parsing @uri fails.
 712  *
 713  * Using this rather than g_network_address_new() or
 714  * g_network_address_parse() allows #GSocketClient to determine
 715  * when to use application-specific proxy protocols.
 716  *
 717  * Return value: (transfer full): the new #GNetworkAddress, or %NULL on error
 718  *
 719  * Since: 2.26
 720  */
 721 GSocketConnectable *
 722 g_network_address_parse_uri (const gchar  *uri,
 723                              guint16       default_port,
 724                              GError      **error)
 725 {
 726   GSocketConnectable *conn;
 727   gchar *scheme;
 728   gchar *hostname;
 729   guint16 port;
 730
 731   if (!_g_uri_parse_authority (uri, &hostname, &port, NULL))
 732     {
 733       g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
 734                    "Invalid URI '%s'",
 735                    uri);
 736       return NULL;
 737     }
 738
 739   if (port == 0)
 740     port = default_port;
 741
 742   scheme = g_uri_parse_scheme (uri);
 743
 744   conn = g_object_new (G_TYPE_NETWORK_ADDRESS,
 745                        "hostname", hostname,
 746                        "port", port,
 747                        "scheme", scheme,
 748                        NULL);
 749
 750   g_free (scheme);
 751   g_free (hostname);
 752
 753   return conn;
 754 }
 755
 756 /**
 757  * g_network_address_get_hostname:
 758  * @addr: a #GNetworkAddress
 759  *
 760  * Gets @addr's hostname. This might be either UTF-8 or ASCII-encoded,
 761  * depending on what @addr was created with.
 762  *
 763  * Return value: @addr's hostname
 764  *
 765  * Since: 2.22
 766  */
 767 const gchar *
 768 g_network_address_get_hostname (GNetworkAddress *addr)
 769 {
 770   g_return_val_if_fail (G_IS_NETWORK_ADDRESS (addr), NULL);
 771
 772   return addr->priv->hostname;
 773 }
 774
 775 /**
 776  * g_network_address_get_port:
 777  * @addr: a #GNetworkAddress
 778  *
 779  * Gets @addr's port number
 780  *
 781  * Return value: @addr's port (which may be 0)
 782  *
 783  * Since: 2.22
 784  */
 785 guint16
 786 g_network_address_get_port (GNetworkAddress *addr)
 787 {
 788   g_return_val_if_fail (G_IS_NETWORK_ADDRESS (addr), 0);
 789
 790   return addr->priv->port;
 791 }
 792
 793 /**
 794  * g_network_address_get_scheme:
 795  * @addr: a #GNetworkAddress
 796  *
 797  * Gets @addr's scheme
 798  *
 799  * Return value: @addr's scheme (%NULL if not built from URI)
 800  *
 801  * Since: 2.26
 802  */
 803 const gchar *
 804 g_network_address_get_scheme (GNetworkAddress *addr)
 805 {
 806   g_return_val_if_fail (G_IS_NETWORK_ADDRESS (addr), NULL);
 807
 808   return addr->priv->scheme;
 809 }
 810
 811 #define G_TYPE_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (_g_network_address_address_enumerator_get_type ())
 812 #define G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR(obj) (G_TYPE_CHECK_INSTANCE_CAST ((obj), G_TYPE_NETWORK_ADDRESS_ADDRESS_ENUMERATOR, GNetworkAddressAddressEnumerator))
 813
 814 typedef struct {
 815   GSocketAddressEnumerator parent_instance;
 816
 817   GNetworkAddress *addr;
 818   GList *addresses;
 819   GList *next;
 820 } GNetworkAddressAddressEnumerator;
 821
 822 typedef struct {
 823   GSocketAddressEnumeratorClass parent_class;
 824
 825 } GNetworkAddressAddressEnumeratorClass;
 826
 827 static GType _g_network_address_address_enumerator_get_type (void);
 828 G_DEFINE_TYPE (GNetworkAddressAddressEnumerator, _g_network_address_address_enumerator, G_TYPE_SOCKET_ADDRESS_ENUMERATOR)
 829
 830 static void
 831 g_network_address_address_enumerator_finalize (GObject *object)
 832 {
 833   GNetworkAddressAddressEnumerator *addr_enum =
 834     G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (object);
 835
 836   g_object_unref (addr_enum->addr);
 837
 838   G_OBJECT_CLASS (_g_network_address_address_enumerator_parent_class)->finalize (object);
 839 }
 840
 841 static GSocketAddress *
 842 g_network_address_address_enumerator_next (GSocketAddressEnumerator  *enumerator,
 843                                            GCancellable              *cancellable,
 844                                            GError                   **error)
 845 {
 846   GNetworkAddressAddressEnumerator *addr_enum =
 847     G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (enumerator);
 848   GSocketAddress *sockaddr;
 849
 850   if (addr_enum->addresses == NULL)
 851     {
 852       GNetworkAddress *addr = addr_enum->addr;
 853       GResolver *resolver = g_resolver_get_default ();
 854       gint64 serial = g_resolver_get_serial (resolver);
 855
 856       if (addr->priv->resolver_serial != 0 &&
 857           addr->priv->resolver_serial != serial)
 858         {
 859           /* Resolver has reloaded, discard cached addresses */
 860           g_list_free_full (addr->priv->sockaddrs, g_object_unref);
 861           addr->priv->sockaddrs = NULL;
 862         }
 863
 864       if (!addr->priv->sockaddrs)
 865         g_network_address_parse_sockaddr (addr);
 866       if (!addr->priv->sockaddrs)
 867         {
 868           GList *addresses;
 869
 870           addresses = g_resolver_lookup_by_name (resolver,
 871                                                  addr->priv->hostname,
 872                                                  cancellable, error);
 873           if (!addresses)
 874             {
 875               g_object_unref (resolver);
 876               return NULL;
 877             }
 878
 879           g_network_address_set_addresses (addr, addresses, serial);
 880         }
 881
 882       addr_enum->addresses = addr->priv->sockaddrs;
 883       addr_enum->next = addr_enum->addresses;
 884       g_object_unref (resolver);
 885     }
 886
 887   if (addr_enum->next == NULL)
 888     return NULL;
 889
 890   sockaddr = addr_enum->next->data;
 891   addr_enum->next = addr_enum->next->next;
 892   return g_object_ref (sockaddr);
 893 }
 894
 895 static void
 896 have_addresses (GNetworkAddressAddressEnumerator *addr_enum,
 897                 GTask *task, GError *error)
 898 {
 899   GSocketAddress *sockaddr;
 900
 901   addr_enum->addresses = addr_enum->addr->priv->sockaddrs;
 902   addr_enum->next = addr_enum->addresses;
 903
 904   if (addr_enum->next)
 905     {
 906       sockaddr = g_object_ref (addr_enum->next->data);
 907       addr_enum->next = addr_enum->next->next;
 908     }
 909   else
 910     sockaddr = NULL;
 911
 912   if (error)
 913     g_task_return_error (task, error);
 914   else
 915     g_task_return_pointer (task, sockaddr, g_object_unref);
 916   g_object_unref (task);
 917 }
 918
 919 static void
 920 got_addresses (GObject      *source_object,
 921                GAsyncResult *result,
 922                gpointer      user_data)
 923 {
 924   GTask *task = user_data;
 925   GNetworkAddressAddressEnumerator *addr_enum = g_task_get_source_object (task);
 926   GResolver *resolver = G_RESOLVER (source_object);
 927   GList *addresses;
 928   GError *error = NULL;
 929
 930   if (!addr_enum->addr->priv->sockaddrs)
 931     {
 932       addresses = g_resolver_lookup_by_name_finish (resolver, result, &error);
 933
 934       if (!error)
 935         {
 936           g_network_address_set_addresses (addr_enum->addr, addresses,
 937                                            g_resolver_get_serial (resolver));
 938         }
 939     }
 940   have_addresses (addr_enum, task, error);
 941 }
 942
 943 static void
 944 g_network_address_address_enumerator_next_async (GSocketAddressEnumerator  *enumerator,
 945                                                  GCancellable              *cancellable,
 946                                                  GAsyncReadyCallback        callback,
 947                                                  gpointer                   user_data)
 948 {
 949   GNetworkAddressAddressEnumerator *addr_enum =
 950     G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (enumerator);
 951   GSocketAddress *sockaddr;
 952   GTask *task;
 953
 954   task = g_task_new (addr_enum, cancellable, callback, user_data);
 955
 956   if (addr_enum->addresses == NULL)
 957     {
 958       GNetworkAddress *addr = addr_enum->addr;
 959       GResolver *resolver = g_resolver_get_default ();
 960       gint64 serial = g_resolver_get_serial (resolver);
 961
 962       if (addr->priv->resolver_serial != 0 &&
 963           addr->priv->resolver_serial != serial)
 964         {
 965           /* Resolver has reloaded, discard cached addresses */
 966           g_list_free_full (addr->priv->sockaddrs, g_object_unref);
 967           addr->priv->sockaddrs = NULL;
 968         }
 969
 970       if (!addr->priv->sockaddrs)
 971         {
 972           if (g_network_address_parse_sockaddr (addr))
 973             have_addresses (addr_enum, task, NULL);
 974           else
 975             {
 976               g_resolver_lookup_by_name_async (resolver,
 977                                                addr->priv->hostname,
 978                                                cancellable,
 979                                                got_addresses, task);
 980             }
 981           g_object_unref (resolver);
 982           return;
 983         }
 984
 985       addr_enum->addresses = addr->priv->sockaddrs;
 986       addr_enum->next = addr_enum->addresses;
 987       g_object_unref (resolver);
 988     }
 989
 990   if (addr_enum->next)
 991     {
 992       sockaddr = g_object_ref (addr_enum->next->data);
 993       addr_enum->next = addr_enum->next->next;
 994     }
 995   else
 996     sockaddr = NULL;
 997
 998   g_task_return_pointer (task, sockaddr, g_object_unref);
 999   g_object_unref (task);
1000 }
1001
1002 static GSocketAddress *
1003 g_network_address_address_enumerator_next_finish (GSocketAddressEnumerator  *enumerator,
1004                                                   GAsyncResult              *result,
1005                                                   GError                   **error)
1006 {
1007   g_return_val_if_fail (g_task_is_valid (result, enumerator), NULL);
1008
1009   return g_task_propagate_pointer (G_TASK (result), error);
1010 }
1011
1012 static void
1013 _g_network_address_address_enumerator_init (GNetworkAddressAddressEnumerator *enumerator)
1014 {
1015 }
1016
1017 static void
1018 _g_network_address_address_enumerator_class_init (GNetworkAddressAddressEnumeratorClass *addrenum_class)
1019 {
1020   GObjectClass *object_class = G_OBJECT_CLASS (addrenum_class);
1021   GSocketAddressEnumeratorClass *enumerator_class =
1022     G_SOCKET_ADDRESS_ENUMERATOR_CLASS (addrenum_class);
1023
1024   enumerator_class->next = g_network_address_address_enumerator_next;
1025   enumerator_class->next_async = g_network_address_address_enumerator_next_async;
1026   enumerator_class->next_finish = g_network_address_address_enumerator_next_finish;
1027   object_class->finalize = g_network_address_address_enumerator_finalize;
1028 }
1029
1030 static GSocketAddressEnumerator *
1031 g_network_address_connectable_enumerate (GSocketConnectable *connectable)
1032 {
1033   GNetworkAddressAddressEnumerator *addr_enum;
1034
1035   addr_enum = g_object_new (G_TYPE_NETWORK_ADDRESS_ADDRESS_ENUMERATOR, NULL);
1036   addr_enum->addr = g_object_ref (connectable);
1037
1038   return (GSocketAddressEnumerator *)addr_enum;
1039 }
1040
1041 static GSocketAddressEnumerator *
1042 g_network_address_connectable_proxy_enumerate (GSocketConnectable *connectable)
1043 {
1044   GNetworkAddress *self = G_NETWORK_ADDRESS (connectable);
1045   GSocketAddressEnumerator *proxy_enum;
1046   gchar *uri;
1047
1048   uri = _g_uri_from_authority (self->priv->scheme ? self->priv->scheme : "none",
1049                                self->priv->hostname,
1050                                self->priv->port,
1051                                NULL);
1052
1053   proxy_enum = g_object_new (G_TYPE_PROXY_ADDRESS_ENUMERATOR,
1054                              "connectable", connectable,
1055                              "uri", uri,
1056                              NULL);
1057
1058   g_free (uri);
1059
1060   return proxy_enum;
1061 }