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