gnetworkaddress: fix parsing of URIs with IPv6 literal and port
[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       start++;
565       p++;
566       while (1)
567         {
568           c = *p++;
569
570           if (c == ']')
571             break;
572
573           /* unreserved /  sub-delims */
574           if (!(g_ascii_isalnum (c) ||
575                 strchr (G_URI_OTHER_UNRESERVED, c) ||
576                 strchr (G_URI_RESERVED_CHARS_SUBCOMPONENT_DELIMITERS, c) ||
577                 c == ':' ||
578                 c == '.'))
579             goto error;
580         }
581     }
582   else
583     {
584       while (1)
585         {
586           c = *p++;
587
588           if (c == ':' ||
589               c == '/' ||
590               c == '?' ||
591               c == '#' ||
592               c == '\0')
593             break;
594
595           /* pct-encoded */
596           if (c == '%')
597             {
598               if (!(g_ascii_isxdigit (p[0]) ||
599                     g_ascii_isxdigit (p[1])))
600                 goto error;
601
602               p++;
603
604               continue;
605             }
606
607           /* unreserved /  sub-delims */
608           if (!(g_ascii_isalnum (c) ||
609                 strchr (G_URI_OTHER_UNRESERVED, c) ||
610                 strchr (G_URI_RESERVED_CHARS_SUBCOMPONENT_DELIMITERS, c)))
611             goto error;
612         }
613     }
614
615   if (host)
616     *host = g_uri_unescape_segment (start, p - 1, NULL);
617
618   if (c == ']')
619     c = *p++;
620
621   if (c == ':')
622     {
623       /* Decode pot:
624        *  port          = *DIGIT
625        */
626       guint tmp = 0;
627
628       while (1)
629         {
630           c = *p++;
631
632           if (c == '/' ||
633               c == '?' ||
634               c == '#' ||
635               c == '\0')
636             break;
637
638           if (!g_ascii_isdigit (c))
639             goto error;
640
641           tmp = (tmp * 10) + (c - '0');
642
643           if (tmp > 65535)
644             goto error;
645         }
646       if (port)
647         *port = (guint16) tmp;
648     }
649
650   return TRUE;
651
652 error:
653   if (host && *host)
654     {
655       g_free (*host);
656       *host = NULL;
657     }
658
659   if (userinfo && *userinfo)
660     {
661       g_free (*userinfo);
662       *userinfo = NULL;
663     }
664
665   return FALSE;
666 }
667
668 gchar *
669 _g_uri_from_authority (const gchar *protocol,
670                        const gchar *host,
671                        guint        port,
672                        const gchar *userinfo)
673 {
674   GString *uri;
675
676   uri = g_string_new (protocol);
677   g_string_append (uri, "://");
678
679   if (userinfo)
680     {
681       g_string_append_uri_escaped (uri, userinfo, G_URI_RESERVED_CHARS_ALLOWED_IN_USERINFO, FALSE);
682       g_string_append_c (uri, '@');
683     }
684
685   if (g_hostname_is_non_ascii (host))
686     {
687       gchar *ace_encoded = g_hostname_to_ascii (host);
688
689       if (!ace_encoded)
690         {
691           g_string_free (uri, TRUE);
692           return NULL;
693         }
694       g_string_append (uri, ace_encoded);
695       g_free (ace_encoded);
696     }
697   else if (strchr (host, ':'))
698     g_string_append_printf (uri, "[%s]", host);
699   else
700     g_string_append (uri, host);
701
702   if (port != 0)
703     g_string_append_printf (uri, ":%u", port);
704
705   return g_string_free (uri, FALSE);
706 }
707
708 /**
709  * g_network_address_parse_uri:
710  * @uri: the hostname and optionally a port
711  * @default_port: The default port if none is found in the URI
712  * @error: a pointer to a #GError, or %NULL
713  *
714  * Creates a new #GSocketConnectable for connecting to the given
715  * @uri. May fail and return %NULL in case parsing @uri fails.
716  *
717  * Using this rather than g_network_address_new() or
718  * g_network_address_parse() allows #GSocketClient to determine
719  * when to use application-specific proxy protocols.
720  *
721  * Return value: (transfer full): the new #GNetworkAddress, or %NULL on error
722  *
723  * Since: 2.26
724  */
725 GSocketConnectable *
726 g_network_address_parse_uri (const gchar  *uri,
727                              guint16       default_port,
728                              GError      **error)
729 {
730   GSocketConnectable *conn;
731   gchar *scheme;
732   gchar *hostname;
733   guint16 port;
734
735   if (!_g_uri_parse_authority (uri, &hostname, &port, NULL))
736     {
737       g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
738                    "Invalid URI '%s'",
739                    uri);
740       return NULL;
741     }
742
743   if (port == 0)
744     port = default_port;
745
746   scheme = g_uri_parse_scheme (uri);
747
748   conn = g_object_new (G_TYPE_NETWORK_ADDRESS,
749                        "hostname", hostname,
750                        "port", port,
751                        "scheme", scheme,
752                        NULL);
753
754   g_free (scheme);
755   g_free (hostname);
756
757   return conn;
758 }
759
760 /**
761  * g_network_address_get_hostname:
762  * @addr: a #GNetworkAddress
763  *
764  * Gets @addr's hostname. This might be either UTF-8 or ASCII-encoded,
765  * depending on what @addr was created with.
766  *
767  * Return value: @addr's hostname
768  *
769  * Since: 2.22
770  */
771 const gchar *
772 g_network_address_get_hostname (GNetworkAddress *addr)
773 {
774   g_return_val_if_fail (G_IS_NETWORK_ADDRESS (addr), NULL);
775
776   return addr->priv->hostname;
777 }
778
779 /**
780  * g_network_address_get_port:
781  * @addr: a #GNetworkAddress
782  *
783  * Gets @addr's port number
784  *
785  * Return value: @addr's port (which may be 0)
786  *
787  * Since: 2.22
788  */
789 guint16
790 g_network_address_get_port (GNetworkAddress *addr)
791 {
792   g_return_val_if_fail (G_IS_NETWORK_ADDRESS (addr), 0);
793
794   return addr->priv->port;
795 }
796
797 /**
798  * g_network_address_get_scheme:
799  * @addr: a #GNetworkAddress
800  *
801  * Gets @addr's scheme
802  *
803  * Return value: @addr's scheme (%NULL if not built from URI)
804  *
805  * Since: 2.26
806  */
807 const gchar *
808 g_network_address_get_scheme (GNetworkAddress *addr)
809 {
810   g_return_val_if_fail (G_IS_NETWORK_ADDRESS (addr), NULL);
811
812   return addr->priv->scheme;
813 }
814
815 #define G_TYPE_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (_g_network_address_address_enumerator_get_type ())
816 #define G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR(obj) (G_TYPE_CHECK_INSTANCE_CAST ((obj), G_TYPE_NETWORK_ADDRESS_ADDRESS_ENUMERATOR, GNetworkAddressAddressEnumerator))
817
818 typedef struct {
819   GSocketAddressEnumerator parent_instance;
820
821   GNetworkAddress *addr;
822   GList *addresses;
823   GList *next;
824 } GNetworkAddressAddressEnumerator;
825
826 typedef struct {
827   GSocketAddressEnumeratorClass parent_class;
828
829 } GNetworkAddressAddressEnumeratorClass;
830
831 static GType _g_network_address_address_enumerator_get_type (void);
832 G_DEFINE_TYPE (GNetworkAddressAddressEnumerator, _g_network_address_address_enumerator, G_TYPE_SOCKET_ADDRESS_ENUMERATOR)
833
834 static void
835 g_network_address_address_enumerator_finalize (GObject *object)
836 {
837   GNetworkAddressAddressEnumerator *addr_enum =
838     G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (object);
839
840   g_object_unref (addr_enum->addr);
841
842   G_OBJECT_CLASS (_g_network_address_address_enumerator_parent_class)->finalize (object);
843 }
844
845 static GSocketAddress *
846 g_network_address_address_enumerator_next (GSocketAddressEnumerator  *enumerator,
847                                            GCancellable              *cancellable,
848                                            GError                   **error)
849 {
850   GNetworkAddressAddressEnumerator *addr_enum =
851     G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (enumerator);
852   GSocketAddress *sockaddr;
853
854   if (addr_enum->addresses == NULL)
855     {
856       GNetworkAddress *addr = addr_enum->addr;
857       GResolver *resolver = g_resolver_get_default ();
858       gint64 serial = g_resolver_get_serial (resolver);
859
860       if (addr->priv->resolver_serial != 0 &&
861           addr->priv->resolver_serial != serial)
862         {
863           /* Resolver has reloaded, discard cached addresses */
864           g_list_free_full (addr->priv->sockaddrs, g_object_unref);
865           addr->priv->sockaddrs = NULL;
866         }
867
868       if (!addr->priv->sockaddrs)
869         g_network_address_parse_sockaddr (addr);
870       if (!addr->priv->sockaddrs)
871         {
872           GList *addresses;
873
874           addresses = g_resolver_lookup_by_name (resolver,
875                                                  addr->priv->hostname,
876                                                  cancellable, error);
877           if (!addresses)
878             {
879               g_object_unref (resolver);
880               return NULL;
881             }
882
883           g_network_address_set_addresses (addr, addresses, serial);
884         }
885           
886       addr_enum->addresses = addr->priv->sockaddrs;
887       addr_enum->next = addr_enum->addresses;
888       g_object_unref (resolver);
889     }
890
891   if (addr_enum->next == NULL)
892     return NULL;
893
894   sockaddr = addr_enum->next->data;
895   addr_enum->next = addr_enum->next->next;
896   return g_object_ref (sockaddr);
897 }
898
899 static void
900 have_addresses (GNetworkAddressAddressEnumerator *addr_enum,
901                 GTask *task, GError *error)
902 {
903   GSocketAddress *sockaddr;
904
905   addr_enum->addresses = addr_enum->addr->priv->sockaddrs;
906   addr_enum->next = addr_enum->addresses;
907
908   if (addr_enum->next)
909     {
910       sockaddr = g_object_ref (addr_enum->next->data);
911       addr_enum->next = addr_enum->next->next;
912     }
913   else
914     sockaddr = NULL;
915
916   if (error)
917     g_task_return_error (task, error);
918   else
919     g_task_return_pointer (task, sockaddr, g_object_unref);
920   g_object_unref (task);
921 }
922
923 static void
924 got_addresses (GObject      *source_object,
925                GAsyncResult *result,
926                gpointer      user_data)
927 {
928   GTask *task = user_data;
929   GNetworkAddressAddressEnumerator *addr_enum = g_task_get_source_object (task);
930   GResolver *resolver = G_RESOLVER (source_object);
931   GList *addresses;
932   GError *error = NULL;
933
934   if (!addr_enum->addr->priv->sockaddrs)
935     {
936       addresses = g_resolver_lookup_by_name_finish (resolver, result, &error);
937
938       if (!error)
939         {
940           g_network_address_set_addresses (addr_enum->addr, addresses,
941                                            g_resolver_get_serial (resolver));
942         }
943     }
944   have_addresses (addr_enum, task, error);
945 }
946
947 static void
948 g_network_address_address_enumerator_next_async (GSocketAddressEnumerator  *enumerator,
949                                                  GCancellable              *cancellable,
950                                                  GAsyncReadyCallback        callback,
951                                                  gpointer                   user_data)
952 {
953   GNetworkAddressAddressEnumerator *addr_enum =
954     G_NETWORK_ADDRESS_ADDRESS_ENUMERATOR (enumerator);
955   GSocketAddress *sockaddr;
956   GTask *task;
957
958   task = g_task_new (addr_enum, cancellable, callback, user_data);
959
960   if (addr_enum->addresses == NULL)
961     {
962       GNetworkAddress *addr = addr_enum->addr;
963       GResolver *resolver = g_resolver_get_default ();
964       gint64 serial = g_resolver_get_serial (resolver);
965
966       if (addr->priv->resolver_serial != 0 &&
967           addr->priv->resolver_serial != serial)
968         {
969           /* Resolver has reloaded, discard cached addresses */
970           g_list_free_full (addr->priv->sockaddrs, g_object_unref);
971           addr->priv->sockaddrs = NULL;
972         }
973
974       if (!addr->priv->sockaddrs)
975         {
976           if (g_network_address_parse_sockaddr (addr))
977             have_addresses (addr_enum, task, NULL);
978           else
979             {
980               g_resolver_lookup_by_name_async (resolver,
981                                                addr->priv->hostname,
982                                                cancellable,
983                                                got_addresses, task);
984             }
985           g_object_unref (resolver);
986           return;
987         }
988
989       addr_enum->addresses = addr->priv->sockaddrs;
990       addr_enum->next = addr_enum->addresses;
991       g_object_unref (resolver);
992     }
993
994   if (addr_enum->next)
995     {
996       sockaddr = g_object_ref (addr_enum->next->data);
997       addr_enum->next = addr_enum->next->next;
998     }
999   else
1000     sockaddr = NULL;
1001
1002   g_task_return_pointer (task, sockaddr, g_object_unref);
1003   g_object_unref (task);
1004 }
1005
1006 static GSocketAddress *
1007 g_network_address_address_enumerator_next_finish (GSocketAddressEnumerator  *enumerator,
1008                                                   GAsyncResult              *result,
1009                                                   GError                   **error)
1010 {
1011   g_return_val_if_fail (g_task_is_valid (result, enumerator), NULL);
1012
1013   return g_task_propagate_pointer (G_TASK (result), error);
1014 }
1015
1016 static void
1017 _g_network_address_address_enumerator_init (GNetworkAddressAddressEnumerator *enumerator)
1018 {
1019 }
1020
1021 static void
1022 _g_network_address_address_enumerator_class_init (GNetworkAddressAddressEnumeratorClass *addrenum_class)
1023 {
1024   GObjectClass *object_class = G_OBJECT_CLASS (addrenum_class);
1025   GSocketAddressEnumeratorClass *enumerator_class =
1026     G_SOCKET_ADDRESS_ENUMERATOR_CLASS (addrenum_class);
1027
1028   enumerator_class->next = g_network_address_address_enumerator_next;
1029   enumerator_class->next_async = g_network_address_address_enumerator_next_async;
1030   enumerator_class->next_finish = g_network_address_address_enumerator_next_finish;
1031   object_class->finalize = g_network_address_address_enumerator_finalize;
1032 }
1033
1034 static GSocketAddressEnumerator *
1035 g_network_address_connectable_enumerate (GSocketConnectable *connectable)
1036 {
1037   GNetworkAddressAddressEnumerator *addr_enum;
1038
1039   addr_enum = g_object_new (G_TYPE_NETWORK_ADDRESS_ADDRESS_ENUMERATOR, NULL);
1040   addr_enum->addr = g_object_ref (connectable);
1041
1042   return (GSocketAddressEnumerator *)addr_enum;
1043 }
1044
1045 static GSocketAddressEnumerator *
1046 g_network_address_connectable_proxy_enumerate (GSocketConnectable *connectable)
1047 {
1048   GNetworkAddress *self = G_NETWORK_ADDRESS (connectable);
1049   GSocketAddressEnumerator *proxy_enum;
1050   gchar *uri;
1051
1052   uri = _g_uri_from_authority (self->priv->scheme ? self->priv->scheme : "none",
1053                                self->priv->hostname,
1054                                self->priv->port,
1055                                NULL);
1056
1057   proxy_enum = g_object_new (G_TYPE_PROXY_ADDRESS_ENUMERATOR,
1058                              "connectable", connectable,
1059                              "uri", uri,
1060                              NULL);
1061
1062   g_free (uri);
1063
1064   return proxy_enum;
1065 }