| /* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*- */ |
| |
| /* GIO - GLib Input, Output and Streaming Library |
| * |
| * Copyright (C) 2008 Red Hat, Inc. |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General |
| * Public License along with this library; if not, write to the |
| * Free Software Foundation, Inc., 59 Temple Place, Suite 330, |
| * Boston, MA 02111-1307, USA. |
| */ |
| |
| #include "config.h" |
| #include <glib.h> |
| #include "glibintl.h" |
| |
| #include "gnetworkservice.h" |
| #include "gcancellable.h" |
| #include "ginetaddress.h" |
| #include "ginetsocketaddress.h" |
| #include "gresolver.h" |
| #include "gsimpleasyncresult.h" |
| #include "gsocketaddressenumerator.h" |
| #include "gsocketconnectable.h" |
| #include "gsrvtarget.h" |
| |
| #include <stdlib.h> |
| #include <string.h> |
| |
| #include "gioalias.h" |
| |
| /** |
| * SECTION:gnetworkservice |
| * @short_description: A GSocketConnectable for resolving SRV records |
| * @include: gio/gio.h |
| * |
| * Like #GNetworkAddress does with hostnames, #GNetworkService |
| * provides an easy way to resolve a SRV record, and then attempt to |
| * connect to one of the hosts that implements that service, handling |
| * service priority/weighting, multiple IP addresses, and multiple |
| * address families. |
| * |
| * See #GSrvTarget for more information about SRV records, and see |
| * #GSocketConnectable for and example of using the connectable |
| * interface. |
| */ |
| |
| /** |
| * GNetworkService: |
| * |
| * A #GSocketConnectable for resolving a SRV record and connecting to |
| * that service. |
| */ |
| |
| struct _GNetworkServicePrivate |
| { |
| gchar *service, *protocol, *domain; |
| GList *targets; |
| }; |
| |
| enum { |
| PROP_0, |
| PROP_SERVICE, |
| PROP_PROTOCOL, |
| PROP_DOMAIN, |
| }; |
| |
| static void g_network_service_set_property (GObject *object, |
| guint prop_id, |
| const GValue *value, |
| GParamSpec *pspec); |
| static void g_network_service_get_property (GObject *object, |
| guint prop_id, |
| GValue *value, |
| GParamSpec *pspec); |
| |
| static void g_network_service_connectable_iface_init (GSocketConnectableIface *iface); |
| static GSocketAddressEnumerator *g_network_service_connectable_enumerate (GSocketConnectable *connectable); |
| |
| G_DEFINE_TYPE_WITH_CODE (GNetworkService, g_network_service, G_TYPE_OBJECT, |
| G_IMPLEMENT_INTERFACE (G_TYPE_SOCKET_CONNECTABLE, |
| g_network_service_connectable_iface_init)) |
| |
| static void |
| g_network_service_finalize (GObject *object) |
| { |
| GNetworkService *srv = G_NETWORK_SERVICE (object); |
| |
| g_free (srv->priv->service); |
| g_free (srv->priv->protocol); |
| g_free (srv->priv->domain); |
| |
| if (srv->priv->targets) |
| g_resolver_free_targets (srv->priv->targets); |
| |
| G_OBJECT_CLASS (g_network_service_parent_class)->finalize (object); |
| } |
| |
| static void |
| g_network_service_class_init (GNetworkServiceClass *klass) |
| { |
| GObjectClass *gobject_class = G_OBJECT_CLASS (klass); |
| |
| g_type_class_add_private (klass, sizeof (GNetworkServicePrivate)); |
| |
| gobject_class->set_property = g_network_service_set_property; |
| gobject_class->get_property = g_network_service_get_property; |
| gobject_class->finalize = g_network_service_finalize; |
| |
| g_object_class_install_property (gobject_class, PROP_SERVICE, |
| g_param_spec_string ("service", |
| P_("Service"), |
| P_("Service name, eg \"ldap\""), |
| NULL, |
| G_PARAM_READWRITE | |
| G_PARAM_CONSTRUCT_ONLY | |
| G_PARAM_STATIC_STRINGS)); |
| g_object_class_install_property (gobject_class, PROP_PROTOCOL, |
| g_param_spec_string ("protocol", |
| P_("Protocol"), |
| P_("Network protocol, eg \"tcp\""), |
| NULL, |
| G_PARAM_READWRITE | |
| G_PARAM_CONSTRUCT_ONLY | |
| G_PARAM_STATIC_STRINGS)); |
| g_object_class_install_property (gobject_class, PROP_DOMAIN, |
| g_param_spec_string ("domain", |
| P_("Domain"), |
| P_("Network domain, eg, \"example.com\""), |
| NULL, |
| G_PARAM_READWRITE | |
| G_PARAM_CONSTRUCT_ONLY | |
| G_PARAM_STATIC_STRINGS)); |
| } |
| |
| static void |
| g_network_service_connectable_iface_init (GSocketConnectableIface *connectable_iface) |
| { |
| connectable_iface->enumerate = g_network_service_connectable_enumerate; |
| } |
| |
| static void |
| g_network_service_init (GNetworkService *srv) |
| { |
| srv->priv = G_TYPE_INSTANCE_GET_PRIVATE (srv, G_TYPE_NETWORK_SERVICE, |
| GNetworkServicePrivate); |
| } |
| |
| static void |
| g_network_service_set_property (GObject *object, |
| guint prop_id, |
| const GValue *value, |
| GParamSpec *pspec) |
| { |
| GNetworkService *srv = G_NETWORK_SERVICE (object); |
| |
| switch (prop_id) |
| { |
| case PROP_SERVICE: |
| srv->priv->service = g_value_dup_string (value); |
| break; |
| |
| case PROP_PROTOCOL: |
| srv->priv->protocol = g_value_dup_string (value); |
| break; |
| |
| case PROP_DOMAIN: |
| srv->priv->domain = g_value_dup_string (value); |
| break; |
| |
| default: |
| G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); |
| break; |
| } |
| } |
| |
| static void |
| g_network_service_get_property (GObject *object, |
| guint prop_id, |
| GValue *value, |
| GParamSpec *pspec) |
| { |
| GNetworkService *srv = G_NETWORK_SERVICE (object); |
| |
| switch (prop_id) |
| { |
| case PROP_SERVICE: |
| g_value_set_string (value, g_network_service_get_service (srv)); |
| break; |
| |
| case PROP_PROTOCOL: |
| g_value_set_string (value, g_network_service_get_protocol (srv)); |
| break; |
| |
| case PROP_DOMAIN: |
| g_value_set_string (value, g_network_service_get_domain (srv)); |
| break; |
| |
| default: |
| G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); |
| break; |
| } |
| } |
| |
| /** |
| * g_network_service_new: |
| * @service: the service type to look up (eg, "ldap") |
| * @protocol: the networking protocol to use for @service (eg, "tcp") |
| * @domain: the DNS domain to look up the service in |
| * |
| * Creates a new #GNetworkService representing the given @service, |
| * @protocol, and @domain. This will initially be unresolved; use the |
| * #GSocketConnectable interface to resolve it. |
| * |
| * Return value: a new #GNetworkService |
| * |
| * Since: 2.22 |
| */ |
| GSocketConnectable * |
| g_network_service_new (const gchar *service, |
| const gchar *protocol, |
| const gchar *domain) |
| { |
| return g_object_new (G_TYPE_NETWORK_SERVICE, |
| "service", service, |
| "protocol", protocol, |
| "domain", domain, |
| NULL); |
| } |
| |
| /** |
| * g_network_service_get_service: |
| * @srv: a #GNetworkService |
| * |
| * Gets @srv's service name (eg, "ldap"). |
| * |
| * Return value: @srv's service name |
| * |
| * Since: 2.22 |
| */ |
| const gchar * |
| g_network_service_get_service (GNetworkService *srv) |
| { |
| g_return_val_if_fail (G_IS_NETWORK_SERVICE (srv), NULL); |
| |
| return srv->priv->service; |
| } |
| |
| /** |
| * g_network_service_get_protocol: |
| * @srv: a #GNetworkService |
| * |
| * Gets @srv's protocol name (eg, "tcp"). |
| * |
| * Return value: @srv's protocol name |
| * |
| * Since: 2.22 |
| */ |
| const gchar * |
| g_network_service_get_protocol (GNetworkService *srv) |
| { |
| g_return_val_if_fail (G_IS_NETWORK_SERVICE (srv), NULL); |
| |
| return srv->priv->protocol; |
| } |
| |
| /** |
| * g_network_service_get_domain: |
| * @srv: a #GNetworkService |
| * |
| * Gets the domain that @srv serves. This might be either UTF-8 or |
| * ASCII-encoded, depending on what @srv was created with. |
| * |
| * Return value: @srv's domain name |
| * |
| * Since: 2.22 |
| */ |
| const gchar * |
| g_network_service_get_domain (GNetworkService *srv) |
| { |
| g_return_val_if_fail (G_IS_NETWORK_SERVICE (srv), NULL); |
| |
| return srv->priv->domain; |
| } |
| |
| #define G_TYPE_NETWORK_SERVICE_ADDRESS_ENUMERATOR (_g_network_service_address_enumerator_get_type ()) |
| #define G_NETWORK_SERVICE_ADDRESS_ENUMERATOR(obj) (G_TYPE_CHECK_INSTANCE_CAST ((obj), G_TYPE_NETWORK_SERVICE_ADDRESS_ENUMERATOR, GNetworkServiceAddressEnumerator)) |
| |
| typedef struct { |
| GSocketAddressEnumerator parent_instance; |
| |
| GResolver *resolver; |
| GNetworkService *srv; |
| GList *addrs, *a, *t; |
| |
| GError *error; |
| |
| /* For async operation */ |
| GCancellable *cancellable; |
| GSimpleAsyncResult *result; |
| } GNetworkServiceAddressEnumerator; |
| |
| typedef struct { |
| GSocketAddressEnumeratorClass parent_class; |
| |
| } GNetworkServiceAddressEnumeratorClass; |
| |
| G_DEFINE_TYPE (GNetworkServiceAddressEnumerator, _g_network_service_address_enumerator, G_TYPE_SOCKET_ADDRESS_ENUMERATOR) |
| |
| static void |
| g_network_service_address_enumerator_finalize (GObject *object) |
| { |
| GNetworkServiceAddressEnumerator *srv_enum = |
| G_NETWORK_SERVICE_ADDRESS_ENUMERATOR (object); |
| |
| g_object_unref (srv_enum->srv); |
| if (srv_enum->addrs) |
| { |
| while (srv_enum->a) |
| { |
| g_object_unref (srv_enum->a->data); |
| srv_enum->a = srv_enum->a->next; |
| } |
| g_list_free (srv_enum->addrs); |
| } |
| g_object_unref (srv_enum->resolver); |
| if (srv_enum->error) |
| g_error_free (srv_enum->error); |
| |
| G_OBJECT_CLASS (_g_network_service_address_enumerator_parent_class)->finalize (object); |
| } |
| |
| static GSocketAddress * |
| g_network_service_address_enumerator_next (GSocketAddressEnumerator *enumerator, |
| GCancellable *cancellable, |
| GError **error) |
| { |
| GNetworkServiceAddressEnumerator *srv_enum = |
| G_NETWORK_SERVICE_ADDRESS_ENUMERATOR (enumerator); |
| GSrvTarget *target; |
| GSocketAddress *sockaddr; |
| |
| /* If we haven't yet resolved srv, do that */ |
| if (!srv_enum->srv->priv->targets) |
| { |
| GList *targets; |
| |
| targets = g_resolver_lookup_service (srv_enum->resolver, |
| srv_enum->srv->priv->service, |
| srv_enum->srv->priv->protocol, |
| srv_enum->srv->priv->domain, |
| cancellable, error); |
| if (!targets) |
| return NULL; |
| |
| if (!srv_enum->srv->priv->targets) |
| srv_enum->srv->priv->targets = targets; |
| srv_enum->t = srv_enum->srv->priv->targets; |
| } |
| |
| /* Make sure we have a set of resolved addresses for the current |
| * target. When resolving the first target, we save the GError, if |
| * any. If any later target succeeds, we'll free the earlier error, |
| * but if we get to the last target without any of them resolving, |
| * we return that initial error. |
| */ |
| do |
| { |
| /* Return if we're out of targets. */ |
| if (!srv_enum->t) |
| { |
| if (srv_enum->error) |
| { |
| g_propagate_error (error, srv_enum->error); |
| srv_enum->error = NULL; |
| } |
| return NULL; |
| } |
| target = srv_enum->t->data; |
| |
| /* If we haven't resolved the addrs for the current target, do that */ |
| if (!srv_enum->addrs) |
| { |
| GError **error_p; |
| |
| error_p = (srv_enum->t == srv_enum->srv->priv->targets) ? &srv_enum->error : NULL; |
| srv_enum->addrs = g_resolver_lookup_by_name (srv_enum->resolver, |
| g_srv_target_get_hostname (target), |
| cancellable, error_p); |
| if (g_cancellable_set_error_if_cancelled (cancellable, error)) |
| return NULL; |
| |
| if (srv_enum->addrs) |
| { |
| srv_enum->a = srv_enum->addrs; |
| if (srv_enum->error) |
| { |
| g_error_free (srv_enum->error); |
| srv_enum->error = NULL; |
| } |
| } |
| else |
| { |
| /* Try the next target */ |
| srv_enum->t = srv_enum->t->next; |
| } |
| } |
| } |
| while (!srv_enum->addrs); |
| |
| /* Return the next address for this target. If it's the last one, |
| * advance the target counter. |
| */ |
| sockaddr = g_inet_socket_address_new (srv_enum->a->data, |
| g_srv_target_get_port (target)); |
| g_object_unref (srv_enum->a->data); |
| srv_enum->a = srv_enum->a->next; |
| |
| if (!srv_enum->a) |
| { |
| g_list_free (srv_enum->addrs); |
| srv_enum->addrs = NULL; |
| srv_enum->t = srv_enum->t->next; |
| } |
| |
| return sockaddr; |
| } |
| |
| static void next_async_resolved_targets (GObject *source_object, |
| GAsyncResult *result, |
| gpointer user_data); |
| static void next_async_have_targets (GNetworkServiceAddressEnumerator *srv_enum); |
| static void next_async_resolved_addresses (GObject *source_object, |
| GAsyncResult *result, |
| gpointer user_data); |
| static void next_async_have_addresses (GNetworkServiceAddressEnumerator *srv_enum); |
| |
| /* The async version is basically the same as the sync, except we have |
| * to split it into multiple functions. |
| */ |
| static void |
| g_network_service_address_enumerator_next_async (GSocketAddressEnumerator *enumerator, |
| GCancellable *cancellable, |
| GAsyncReadyCallback callback, |
| gpointer user_data) |
| { |
| GNetworkServiceAddressEnumerator *srv_enum = |
| G_NETWORK_SERVICE_ADDRESS_ENUMERATOR (enumerator); |
| |
| g_return_if_fail (srv_enum->result == NULL); |
| |
| srv_enum->result = g_simple_async_result_new (G_OBJECT (enumerator), |
| callback, user_data, |
| g_network_service_address_enumerator_next_async); |
| srv_enum->cancellable = cancellable; |
| |
| /* If we haven't yet resolved srv, do that */ |
| if (!srv_enum->srv->priv->targets) |
| { |
| g_resolver_lookup_service_async (srv_enum->resolver, |
| srv_enum->srv->priv->service, |
| srv_enum->srv->priv->protocol, |
| srv_enum->srv->priv->domain, |
| cancellable, |
| next_async_resolved_targets, |
| srv_enum); |
| } |
| else |
| next_async_have_targets (srv_enum); |
| } |
| |
| static void |
| next_async_resolved_targets (GObject *source_object, |
| GAsyncResult *result, |
| gpointer user_data) |
| { |
| GNetworkServiceAddressEnumerator *srv_enum = user_data; |
| GList *targets; |
| GError *error = NULL; |
| |
| targets = g_resolver_lookup_service_finish (srv_enum->resolver, result, &error); |
| if (!srv_enum->srv->priv->targets) |
| { |
| if (error) |
| { |
| GSimpleAsyncResult *simple = srv_enum->result; |
| |
| srv_enum->result = NULL; |
| g_simple_async_result_set_from_error (simple, error); |
| g_error_free (error); |
| g_simple_async_result_complete (simple); |
| g_object_unref (simple); |
| return; |
| } |
| |
| srv_enum->srv->priv->targets = targets; |
| srv_enum->t = srv_enum->srv->priv->targets; |
| } |
| |
| next_async_have_targets (srv_enum); |
| } |
| |
| static void |
| next_async_have_targets (GNetworkServiceAddressEnumerator *srv_enum) |
| { |
| GSrvTarget *target; |
| |
| /* Get the current target, check if we're already done. */ |
| if (!srv_enum->t) |
| { |
| if (srv_enum->error) |
| { |
| g_simple_async_result_set_from_error (srv_enum->result, srv_enum->error); |
| g_error_free (srv_enum->error); |
| srv_enum->error = NULL; |
| } |
| g_simple_async_result_complete_in_idle (srv_enum->result); |
| g_object_unref (srv_enum->result); |
| srv_enum->result = NULL; |
| return; |
| } |
| target = srv_enum->t->data; |
| |
| /* If we haven't resolved the addrs for the current target, do that */ |
| if (!srv_enum->addrs) |
| { |
| g_resolver_lookup_by_name_async (srv_enum->resolver, |
| g_srv_target_get_hostname (target), |
| srv_enum->cancellable, |
| next_async_resolved_addresses, |
| srv_enum); |
| } |
| else |
| next_async_have_addresses (srv_enum); |
| } |
| |
| static void |
| next_async_resolved_addresses (GObject *source_object, |
| GAsyncResult *result, |
| gpointer user_data) |
| { |
| GNetworkServiceAddressEnumerator *srv_enum = user_data; |
| GError *error = NULL; |
| |
| srv_enum->addrs = g_resolver_lookup_by_name_finish (srv_enum->resolver, result, &error); |
| if (srv_enum->addrs) |
| { |
| srv_enum->a = srv_enum->addrs; |
| if (srv_enum->error) |
| { |
| g_error_free (srv_enum->error); |
| srv_enum->error = NULL; |
| } |
| next_async_have_addresses (srv_enum); |
| } |
| else |
| { |
| if (g_cancellable_is_cancelled (srv_enum->cancellable)) |
| { |
| GSimpleAsyncResult *simple = srv_enum->result; |
| |
| srv_enum->result = NULL; |
| g_simple_async_result_set_from_error (srv_enum->result, error); |
| g_error_free (error); |
| g_simple_async_result_complete (simple); |
| g_object_unref (simple); |
| } |
| else |
| { |
| if (srv_enum->t == srv_enum->srv->priv->targets) |
| srv_enum->error = error; |
| else |
| g_error_free (error); |
| |
| /* Try the next target */ |
| srv_enum->t = srv_enum->t->next; |
| next_async_have_targets (srv_enum); |
| } |
| } |
| } |
| |
| static void |
| next_async_have_addresses (GNetworkServiceAddressEnumerator *srv_enum) |
| { |
| GSocketAddress *sockaddr; |
| GSimpleAsyncResult *simple = srv_enum->result; |
| |
| /* Return the next address for this target. If it's the last one, |
| * advance the target counter. |
| */ |
| sockaddr = g_inet_socket_address_new (srv_enum->a->data, |
| g_srv_target_get_port (srv_enum->t->data)); |
| g_object_unref (srv_enum->a->data); |
| |
| srv_enum->a = srv_enum->a->next; |
| if (!srv_enum->a) |
| { |
| g_list_free (srv_enum->addrs); |
| srv_enum->addrs = NULL; |
| srv_enum->t = srv_enum->t->next; |
| } |
| |
| srv_enum->result = NULL; |
| g_simple_async_result_set_op_res_gpointer (simple, sockaddr, NULL); |
| g_simple_async_result_complete_in_idle (simple); |
| g_object_unref (simple); |
| } |
| |
| static GSocketAddress * |
| g_network_service_address_enumerator_next_finish (GSocketAddressEnumerator *enumerator, |
| GAsyncResult *result, |
| GError **error) |
| { |
| GSimpleAsyncResult *simple = G_SIMPLE_ASYNC_RESULT (result); |
| GSocketAddress *sockaddr; |
| |
| if (g_simple_async_result_propagate_error (simple, error)) |
| return NULL; |
| |
| sockaddr = g_simple_async_result_get_op_res_gpointer (simple); |
| return sockaddr ? g_object_ref (sockaddr) : NULL; |
| } |
| |
| static void |
| _g_network_service_address_enumerator_init (GNetworkServiceAddressEnumerator *enumerator) |
| { |
| } |
| |
| static void |
| _g_network_service_address_enumerator_class_init (GNetworkServiceAddressEnumeratorClass *addrenum_class) |
| { |
| GObjectClass *object_class = G_OBJECT_CLASS (addrenum_class); |
| GSocketAddressEnumeratorClass *enumerator_class = |
| G_SOCKET_ADDRESS_ENUMERATOR_CLASS (addrenum_class); |
| |
| enumerator_class->next = g_network_service_address_enumerator_next; |
| enumerator_class->next_async = g_network_service_address_enumerator_next_async; |
| enumerator_class->next_finish = g_network_service_address_enumerator_next_finish; |
| object_class->finalize = g_network_service_address_enumerator_finalize; |
| } |
| |
| static GSocketAddressEnumerator * |
| g_network_service_connectable_enumerate (GSocketConnectable *connectable) |
| { |
| GNetworkServiceAddressEnumerator *srv_enum; |
| |
| srv_enum = g_object_new (G_TYPE_NETWORK_SERVICE_ADDRESS_ENUMERATOR, NULL); |
| srv_enum->srv = g_object_ref (connectable); |
| srv_enum->resolver = g_resolver_get_default (); |
| |
| return (GSocketAddressEnumerator *)srv_enum; |
| } |
| |
| #define __G_NETWORK_SERVICE_C__ |
| #include "gioaliasdef.c" |