gnetworkaddress: allow IPv6 scope ids in URIs
[platform/upstream/glib.git] / 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_socktype = SOCK_STREAM;
258   hints.ai_flags = AI_NUMERICHOST
259 #ifdef AI_NUMERICSERV
260     | AI_NUMERICSERV
261 #endif
262     ;
263   g_snprintf (port, sizeof (port), "%u", addr->priv->port);
264
265   if (getaddrinfo (addr->priv->hostname, port, &hints, &res) != 0)
266     return FALSE;
267
268   sockaddr = g_socket_address_new_from_native (res->ai_addr, res->ai_addrlen);
269   freeaddrinfo (res);
270   if (!sockaddr || !G_IS_INET_SOCKET_ADDRESS (sockaddr))
271     return FALSE;
272
273   addr->priv->sockaddrs = g_list_prepend (addr->priv->sockaddrs, sockaddr);
274   return TRUE;
275 }
276
277 /**
278  * g_network_address_new:
279  * @hostname: the hostname
280  * @port: the port
281  *
282  * Creates a new #GSocketConnectable for connecting to the given
283  * @hostname and @port.
284  *
285  * Return value: (transfer full) (type GNetworkAddress): the new #GNetworkAddress
286  *
287  * Since: 2.22
288  */
289 GSocketConnectable *
290 g_network_address_new (const gchar *hostname,
291                        guint16      port)
292 {
293   return g_object_new (G_TYPE_NETWORK_ADDRESS,
294                        "hostname", hostname,
295                        "port", port,
296                        NULL);
297 }
298
299 /**
300  * g_network_address_parse:
301  * @host_and_port: the hostname and optionally a port
302  * @default_port: the default port if not in @host_and_port
303  * @error: a pointer to a #GError, or %NULL
304  *
305  * Creates a new #GSocketConnectable for connecting to the given
306  * @hostname and @port. May fail and return %NULL in case
307  * parsing @host_and_port fails.
308  *
309  * @host_and_port may be in any of a number of recognised formats; an IPv6
310  * address, an IPv4 address, or a domain name (in which case a DNS
311  * lookup is performed). Quoting with [] is supported for all address
312  * types. A port override may be specified in the usual way with a
313  * colon.
314  *
315  * If no port is specified in @host_and_port then @default_port will be
316  * used as the port number to connect to.
317  *
318  * In general, @host_and_port is expected to be provided by the user
319  * (allowing them to give the hostname, and a port overide if necessary)
320  * and @default_port is expected to be provided by the application.
321  *
322  * (The port component of @host_and_port can also be specified as a
323  * service name rather than as a numeric port, but this functionality
324  * is deprecated, because it depends on the contents of /etc/services,
325  * which is generally quite sparse on platforms other than Linux.)
326  *
327  * Return value: (transfer full): the new #GNetworkAddress, or %NULL on error
328  *
329  * Since: 2.22
330  */
331 GSocketConnectable *
332 g_network_address_parse (const gchar  *host_and_port,
333                          guint16       default_port,
334                          GError      **error)
335 {
336   GSocketConnectable *connectable;
337   const gchar *port;
338   guint16 portnum;
339   gchar *name;
340
341   g_return_val_if_fail (host_and_port != NULL, NULL);
342
343   port = NULL;
344   if (host_and_port[0] == '[')
345     /* escaped host part (to allow, eg. "[2001:db8::1]:888") */
346     {
347       const gchar *end;
348
349       end = strchr (host_and_port, ']');
350       if (end == NULL)
351         {
352           g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
353                        _("Hostname '%s' contains '[' but not ']'"), host_and_port);
354           return NULL;
355         }
356
357       if (end[1] == '\0')
358         port = NULL;
359       else if (end[1] == ':')
360         port = &end[2];
361       else
362         {
363           g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
364                        "The ']' character (in hostname '%s') must come at the"
365                        " end or be immediately followed by ':' and a port",
366                        host_and_port);
367           return NULL;
368         }
369
370       name = g_strndup (host_and_port + 1, end - host_and_port - 1);
371     }
372
373   else if ((port = strchr (host_and_port, ':')))
374     /* string has a ':' in it */
375     {
376       /* skip ':' */
377       port++;
378
379       if (strchr (port, ':'))
380         /* more than one ':' in string */
381         {
382           /* this is actually an unescaped IPv6 address */
383           name = g_strdup (host_and_port);
384           port = NULL;
385         }
386       else
387         name = g_strndup (host_and_port, port - host_and_port - 1);
388     }
389
390   else
391     /* plain hostname, no port */
392     name = g_strdup (host_and_port);
393
394   if (port != NULL)
395     {
396       if (port[0] == '\0')
397         {
398           g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
399                        "If a ':' character is given, it must be followed by a "
400                        "port (in hostname '%s').", host_and_port);
401           g_free (name);
402           return NULL;
403         }
404
405       else if ('0' <= port[0] && port[0] <= '9')
406         {
407           char *end;
408           long value;
409
410           value = strtol (port, &end, 10);
411           if (*end != '\0' || value < 0 || value > G_MAXUINT16)
412             {
413               g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
414                            "Invalid numeric port '%s' specified in hostname '%s'",
415                            port, host_and_port);
416               g_free (name);
417               return NULL;
418             }
419
420           portnum = value;
421         }
422
423       else
424         {
425           struct servent *entry;
426
427           entry = getservbyname (port, "tcp");
428           if (entry == NULL)
429             {
430               g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
431                            "Unknown service '%s' specified in hostname '%s'",
432                            port, host_and_port);
433 #ifdef HAVE_ENDSERVENT
434               endservent ();
435 #endif
436               g_free (name);
437               return NULL;
438             }
439
440           portnum = g_ntohs (entry->s_port);
441
442 #ifdef HAVE_ENDSERVENT
443           endservent ();
444 #endif
445         }
446     }
447   else
448     {
449       /* No port in host_and_port */
450       portnum = default_port;
451     }
452
453   connectable = g_network_address_new (name, portnum);
454   g_free (name);
455
456   return connectable;
457 }
458
459 /* Allowed characters outside alphanumeric for unreserved. */
460 #define G_URI_OTHER_UNRESERVED "-._~"
461
462 /* This or something equivalent will eventually go into glib/guri.h */
463 gboolean
464 _g_uri_parse_authority (const char  *uri,
465                         char       **host,
466                         guint16     *port,
467                         char       **userinfo)
468 {
469   char *tmp_str;
470   const char *start, *p;
471   char c;
472
473   g_return_val_if_fail (uri != NULL, FALSE);
474
475   if (host)
476     *host = NULL;
477
478   if (port)
479     *port = 0;
480
481   if (userinfo)
482     *userinfo = NULL;
483
484   /* From RFC 3986 Decodes:
485    * URI          = scheme ":" hier-part [ "?" query ] [ "#" fragment ]
486    * hier-part    = "//" authority path-abempty
487    * path-abempty = *( "/" segment )
488    * authority    = [ userinfo "@" ] host [ ":" port ]
489    */
490
491   /* Check we have a valid scheme */
492   tmp_str = g_uri_parse_scheme (uri);
493
494   if (tmp_str == NULL)
495     return FALSE;
496
497   g_free (tmp_str);
498
499   /* Decode hier-part:
500    *  hier-part   = "//" authority path-abempty
501    */
502   p = uri;
503   start = strstr (p, "//");
504
505   if (start == NULL)
506     return FALSE;
507
508   start += 2;
509
510   if (strchr (start, '@') != NULL)
511     {
512       /* Decode userinfo:
513        * userinfo      = *( unreserved / pct-encoded / sub-delims / ":" )
514        * unreserved    = ALPHA / DIGIT / "-" / "." / "_" / "~"
515        * pct-encoded   = "%" HEXDIG HEXDIG
516        */
517       p = start;
518       while (1)
519         {
520           c = *p++;
521
522           if (c == '@')
523             break;
524
525           /* pct-encoded */
526           if (c == '%')
527             {
528               if (!(g_ascii_isxdigit (p[0]) ||
529                     g_ascii_isxdigit (p[1])))
530                 return FALSE;
531
532               p++;
533
534               continue;
535             }
536
537           /* unreserved /  sub-delims / : */
538           if (!(g_ascii_isalnum (c) ||
539                 strchr (G_URI_OTHER_UNRESERVED, c) ||
540                 strchr (G_URI_RESERVED_CHARS_SUBCOMPONENT_DELIMITERS, c) ||
541                 c == ':'))
542             return FALSE;
543         }
544
545       if (userinfo)
546         *userinfo = g_strndup (start, p - start - 1);
547
548       start = p;
549     }
550   else
551     {
552       p = start;
553     }
554
555
556   /* decode host:
557    * host          = IP-literal / IPv4address / reg-name
558    * reg-name      = *( unreserved / pct-encoded / sub-delims )
559    */
560
561   /* If IPv6 or IPvFuture */
562   if (*p == '[')
563     {
564       gboolean has_scope_id = FALSE, has_bad_scope_id = FALSE;
565
566       start++;
567       p++;
568       while (1)
569         {
570           c = *p++;
571
572           if (c == ']')
573             break;
574
575           if (c == '%' && !has_scope_id)
576             {
577               has_scope_id = TRUE;
578               if (p[0] != '2' || p[1] != '5')
579                 has_bad_scope_id = TRUE;
580               continue;
581             }
582
583           /* unreserved /  sub-delims */
584           if (!(g_ascii_isalnum (c) ||
585                 strchr (G_URI_OTHER_UNRESERVED, c) ||
586                 strchr (G_URI_RESERVED_CHARS_SUBCOMPONENT_DELIMITERS, c) ||
587                 c == ':' ||
588                 c == '.'))
589             goto error;
590         }
591
592       if (host)
593         {
594           if (has_bad_scope_id)
595             *host = g_strndup (start, p - start - 1);
596           else
597             *host = g_uri_unescape_segment (start, p - 1, NULL);
598         }
599
600       c = *p++;
601     }
602   else
603     {
604       while (1)
605         {
606           c = *p++;
607
608           if (c == ':' ||
609               c == '/' ||
610               c == '?' ||
611               c == '#' ||
612               c == '\0')
613             break;
614
615           /* pct-encoded */
616           if (c == '%')
617             {
618               if (!(g_ascii_isxdigit (p[0]) ||
619                     g_ascii_isxdigit (p[1])))
620                 goto error;
621
622               p++;
623
624               continue;
625             }
626
627           /* unreserved /  sub-delims */
628           if (!(g_ascii_isalnum (c) ||
629                 strchr (G_URI_OTHER_UNRESERVED, c) ||
630                 strchr (G_URI_RESERVED_CHARS_SUBCOMPONENT_DELIMITERS, c)))
631             goto error;
632         }
633
634       if (host)
635         *host = g_uri_unescape_segment (start, p - 1, NULL);
636     }
637
638   if (c == ':')
639     {
640       /* Decode port:
641        *  port          = *DIGIT
642        */
643       guint tmp = 0;
644
645       while (1)
646         {
647           c = *p++;
648
649           if (c == '/' ||
650               c == '?' ||
651               c == '#' ||
652               c == '\0')
653             break;
654
655           if (!g_ascii_isdigit (c))
656             goto error;
657
658           tmp = (tmp * 10) + (c - '0');
659
660           if (tmp > 65535)
661             goto error;
662         }
663       if (port)
664         *port = (guint16) tmp;
665     }
666
667   return TRUE;
668
669 error:
670   if (host && *host)
671     {
672       g_free (*host);
673       *host = NULL;
674     }
675
676   if (userinfo && *userinfo)
677     {
678       g_free (*userinfo);
679       *userinfo = NULL;
680     }
681
682   return FALSE;
683 }
684
685 gchar *
686 _g_uri_from_authority (const gchar *protocol,
687                        const gchar *host,
688                        guint        port,
689                        const gchar *userinfo)
690 {
691   GString *uri;
692
693   uri = g_string_new (protocol);
694   g_string_append (uri, "://");
695
696   if (userinfo)
697     {
698       g_string_append_uri_escaped (uri, userinfo, G_URI_RESERVED_CHARS_ALLOWED_IN_USERINFO, FALSE);
699       g_string_append_c (uri, '@');
700     }
701
702   if (g_hostname_is_non_ascii (host))
703     {
704       gchar *ace_encoded = g_hostname_to_ascii (host);
705
706       if (!ace_encoded)
707         {
708           g_string_free (uri, TRUE);
709           return NULL;
710         }
711       g_string_append (uri, ace_encoded);
712       g_free (ace_encoded);
713     }
714   else if (strchr (host, ':'))
715     g_string_append_printf (uri, "[%s]", host);
716   else
717     g_string_append (uri, host);
718
719   if (port != 0)
720     g_string_append_printf (uri, ":%u", port);
721
722   return g_string_free (uri, FALSE);
723 }
724
725 /**
726  * g_network_address_parse_uri:
727  * @uri: the hostname and optionally a port
728  * @default_port: The default port if none is found in the URI
729  * @error: a pointer to a #GError, or %NULL
730  *
731  * Creates a new #GSocketConnectable for connecting to the given
732  * @uri. May fail and return %NULL in case parsing @uri fails.
733  *
734  * Using this rather than g_network_address_new() or
735  * g_network_address_parse() allows #GSocketClient to determine
736  * when to use application-specific proxy protocols.
737  *
738  * Return value: (transfer full): the new #GNetworkAddress, or %NULL on error
739  *
740  * Since: 2.26
741  */
742 GSocketConnectable *
743 g_network_address_parse_uri (const gchar  *uri,
744                              guint16       default_port,
745                              GError      **error)
746 {
747   GSocketConnectable *conn;
748   gchar *scheme;
749   gchar *hostname;
750   guint16 port;
751
752   if (!_g_uri_parse_authority (uri, &hostname, &port, NULL))
753     {
754       g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
755                    "Invalid URI '%s'",
756                    uri);
757       return NULL;
758     }
759
760   if (port == 0)
761     port = default_port;
762
763   scheme = g_uri_parse_scheme (uri);
764
765   conn = g_object_new (G_TYPE_NETWORK_ADDRESS,
766                        "hostname", hostname,
767                        "port", port,
768                        "scheme", scheme,
769                        NULL);
770
771   g_free (scheme);
772   g_free (hostname);
773
774   return conn;
775 }
776
777 /**
778  * g_network_address_get_hostname:
779  * @addr: a #GNetworkAddress
780  *
781  * Gets @addr's hostname. This might be either UTF-8 or ASCII-encoded,
782  * depending on what @addr was created with.
783  *
784  * Return value: @addr's hostname
785  *
786  * Since: 2.22
787  */
788 const gchar *
789 g_network_address_get_hostname (GNetworkAddress *addr)
790 {
791   g_return_val_if_fail (G_IS_NETWORK_ADDRESS (addr), NULL);
792
793   return addr->priv->hostname;
794 }
795
796 /**
797  * g_network_address_get_port:
798  * @addr: a #GNetworkAddress
799  *
800  * Gets @addr's port number
801  *
802  * Return value: @addr's port (which may be 0)
803  *
804  * Since: 2.22
805  */
806 guint16
807 g_network_address_get_port (GNetworkAddress *addr)
808 {
809   g_return_val_if_fail (G_IS_NETWORK_ADDRESS (addr), 0);
810
811   return addr->priv->port;
812 }
813
814 /**
815  * g_network_address_get_scheme:
816  * @addr: a #GNetworkAddress
817  *
818  * Gets @addr's scheme
819  *
820  * Return value: @addr's scheme (%NULL if not built from URI)
821  *
822  * Since: 2.26
823  */
824 const gchar *
825 g_network_address_get_scheme (GNetworkAddress *addr)
826 {
827   g_return_val_if_fail (G_IS_NETWORK_ADDRESS (addr), NULL);
828
829   return addr->priv->scheme;
830 }
831
832 #define G_TYPE_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (_g_network_address_address_enumerator_get_type ())
833 #define G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR(obj) (G_TYPE_CHECK_INSTANCE_CAST ((obj), G_TYPE_NETWORK_ADDRESS_ADDRESS_ENUMERATOR, GNetworkAddressAddressEnumerator))
834
835 typedef struct {
836   GSocketAddressEnumerator parent_instance;
837
838   GNetworkAddress *addr;
839   GList *addresses;
840   GList *next;
841 } GNetworkAddressAddressEnumerator;
842
843 typedef struct {
844   GSocketAddressEnumeratorClass parent_class;
845
846 } GNetworkAddressAddressEnumeratorClass;
847
848 static GType _g_network_address_address_enumerator_get_type (void);
849 G_DEFINE_TYPE (GNetworkAddressAddressEnumerator, _g_network_address_address_enumerator, G_TYPE_SOCKET_ADDRESS_ENUMERATOR)
850
851 static void
852 g_network_address_address_enumerator_finalize (GObject *object)
853 {
854   GNetworkAddressAddressEnumerator *addr_enum =
855     G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (object);
856
857   g_object_unref (addr_enum->addr);
858
859   G_OBJECT_CLASS (_g_network_address_address_enumerator_parent_class)->finalize (object);
860 }
861
862 static GSocketAddress *
863 g_network_address_address_enumerator_next (GSocketAddressEnumerator  *enumerator,
864                                            GCancellable              *cancellable,
865                                            GError                   **error)
866 {
867   GNetworkAddressAddressEnumerator *addr_enum =
868     G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (enumerator);
869   GSocketAddress *sockaddr;
870
871   if (addr_enum->addresses == NULL)
872     {
873       GNetworkAddress *addr = addr_enum->addr;
874       GResolver *resolver = g_resolver_get_default ();
875       gint64 serial = g_resolver_get_serial (resolver);
876
877       if (addr->priv->resolver_serial != 0 &&
878           addr->priv->resolver_serial != serial)
879         {
880           /* Resolver has reloaded, discard cached addresses */
881           g_list_free_full (addr->priv->sockaddrs, g_object_unref);
882           addr->priv->sockaddrs = NULL;
883         }
884
885       if (!addr->priv->sockaddrs)
886         g_network_address_parse_sockaddr (addr);
887       if (!addr->priv->sockaddrs)
888         {
889           GList *addresses;
890
891           addresses = g_resolver_lookup_by_name (resolver,
892                                                  addr->priv->hostname,
893                                                  cancellable, error);
894           if (!addresses)
895             {
896               g_object_unref (resolver);
897               return NULL;
898             }
899
900           g_network_address_set_addresses (addr, addresses, serial);
901         }
902           
903       addr_enum->addresses = addr->priv->sockaddrs;
904       addr_enum->next = addr_enum->addresses;
905       g_object_unref (resolver);
906     }
907
908   if (addr_enum->next == NULL)
909     return NULL;
910
911   sockaddr = addr_enum->next->data;
912   addr_enum->next = addr_enum->next->next;
913   return g_object_ref (sockaddr);
914 }
915
916 static void
917 have_addresses (GNetworkAddressAddressEnumerator *addr_enum,
918                 GTask *task, GError *error)
919 {
920   GSocketAddress *sockaddr;
921
922   addr_enum->addresses = addr_enum->addr->priv->sockaddrs;
923   addr_enum->next = addr_enum->addresses;
924
925   if (addr_enum->next)
926     {
927       sockaddr = g_object_ref (addr_enum->next->data);
928       addr_enum->next = addr_enum->next->next;
929     }
930   else
931     sockaddr = NULL;
932
933   if (error)
934     g_task_return_error (task, error);
935   else
936     g_task_return_pointer (task, sockaddr, g_object_unref);
937   g_object_unref (task);
938 }
939
940 static void
941 got_addresses (GObject      *source_object,
942                GAsyncResult *result,
943                gpointer      user_data)
944 {
945   GTask *task = user_data;
946   GNetworkAddressAddressEnumerator *addr_enum = g_task_get_source_object (task);
947   GResolver *resolver = G_RESOLVER (source_object);
948   GList *addresses;
949   GError *error = NULL;
950
951   if (!addr_enum->addr->priv->sockaddrs)
952     {
953       addresses = g_resolver_lookup_by_name_finish (resolver, result, &error);
954
955       if (!error)
956         {
957           g_network_address_set_addresses (addr_enum->addr, addresses,
958                                            g_resolver_get_serial (resolver));
959         }
960     }
961   have_addresses (addr_enum, task, error);
962 }
963
964 static void
965 g_network_address_address_enumerator_next_async (GSocketAddressEnumerator  *enumerator,
966                                                  GCancellable              *cancellable,
967                                                  GAsyncReadyCallback        callback,
968                                                  gpointer                   user_data)
969 {
970   GNetworkAddressAddressEnumerator *addr_enum =
971     G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (enumerator);
972   GSocketAddress *sockaddr;
973   GTask *task;
974
975   task = g_task_new (addr_enum, cancellable, callback, user_data);
976
977   if (addr_enum->addresses == NULL)
978     {
979       GNetworkAddress *addr = addr_enum->addr;
980       GResolver *resolver = g_resolver_get_default ();
981       gint64 serial = g_resolver_get_serial (resolver);
982
983       if (addr->priv->resolver_serial != 0 &&
984           addr->priv->resolver_serial != serial)
985         {
986           /* Resolver has reloaded, discard cached addresses */
987           g_list_free_full (addr->priv->sockaddrs, g_object_unref);
988           addr->priv->sockaddrs = NULL;
989         }
990
991       if (!addr->priv->sockaddrs)
992         {
993           if (g_network_address_parse_sockaddr (addr))
994             have_addresses (addr_enum, task, NULL);
995           else
996             {
997               g_resolver_lookup_by_name_async (resolver,
998                                                addr->priv->hostname,
999                                                cancellable,
1000                                                got_addresses, task);
1001             }
1002           g_object_unref (resolver);
1003           return;
1004         }
1005
1006       addr_enum->addresses = addr->priv->sockaddrs;
1007       addr_enum->next = addr_enum->addresses;
1008       g_object_unref (resolver);
1009     }
1010
1011   if (addr_enum->next)
1012     {
1013       sockaddr = g_object_ref (addr_enum->next->data);
1014       addr_enum->next = addr_enum->next->next;
1015     }
1016   else
1017     sockaddr = NULL;
1018
1019   g_task_return_pointer (task, sockaddr, g_object_unref);
1020   g_object_unref (task);
1021 }
1022
1023 static GSocketAddress *
1024 g_network_address_address_enumerator_next_finish (GSocketAddressEnumerator  *enumerator,
1025                                                   GAsyncResult              *result,
1026                                                   GError                   **error)
1027 {
1028   g_return_val_if_fail (g_task_is_valid (result, enumerator), NULL);
1029
1030   return g_task_propagate_pointer (G_TASK (result), error);
1031 }
1032
1033 static void
1034 _g_network_address_address_enumerator_init (GNetworkAddressAddressEnumerator *enumerator)
1035 {
1036 }
1037
1038 static void
1039 _g_network_address_address_enumerator_class_init (GNetworkAddressAddressEnumeratorClass *addrenum_class)
1040 {
1041   GObjectClass *object_class = G_OBJECT_CLASS (addrenum_class);
1042   GSocketAddressEnumeratorClass *enumerator_class =
1043     G_SOCKET_ADDRESS_ENUMERATOR_CLASS (addrenum_class);
1044
1045   enumerator_class->next = g_network_address_address_enumerator_next;
1046   enumerator_class->next_async = g_network_address_address_enumerator_next_async;
1047   enumerator_class->next_finish = g_network_address_address_enumerator_next_finish;
1048   object_class->finalize = g_network_address_address_enumerator_finalize;
1049 }
1050
1051 static GSocketAddressEnumerator *
1052 g_network_address_connectable_enumerate (GSocketConnectable *connectable)
1053 {
1054   GNetworkAddressAddressEnumerator *addr_enum;
1055
1056   addr_enum = g_object_new (G_TYPE_NETWORK_ADDRESS_ADDRESS_ENUMERATOR, NULL);
1057   addr_enum->addr = g_object_ref (connectable);
1058
1059   return (GSocketAddressEnumerator *)addr_enum;
1060 }
1061
1062 static GSocketAddressEnumerator *
1063 g_network_address_connectable_proxy_enumerate (GSocketConnectable *connectable)
1064 {
1065   GNetworkAddress *self = G_NETWORK_ADDRESS (connectable);
1066   GSocketAddressEnumerator *proxy_enum;
1067   gchar *uri;
1068
1069   uri = _g_uri_from_authority (self->priv->scheme ? self->priv->scheme : "none",
1070                                self->priv->hostname,
1071                                self->priv->port,
1072                                NULL);
1073
1074   proxy_enum = g_object_new (G_TYPE_PROXY_ADDRESS_ENUMERATOR,
1075                              "connectable", connectable,
1076                              "uri", uri,
1077                              NULL);
1078
1079   g_free (uri);
1080
1081   return proxy_enum;
1082 }