5fa30de
From 6f0aa79c3e8dd93e723f29bf46e1b8b14403254f Mon Sep 17 00:00:00 2001
5fa30de
From: Sumit Bose <sbose@redhat.com>
5fa30de
Date: Mon, 5 Dec 2016 18:25:44 +0100
5fa30de
Subject: [PATCH] Kerberos: fall back to tcp SRV lookup
5fa30de
5fa30de
---
5fa30de
 service/realm-kerberos-provider.c | 48 +++++++++++++++++++++++++++++++--------
5fa30de
 1 file changed, 39 insertions(+), 9 deletions(-)
5fa30de
5fa30de
diff --git a/service/realm-kerberos-provider.c b/service/realm-kerberos-provider.c
5fa30de
index 2b3a0f8..1477ae8 100644
5fa30de
--- a/service/realm-kerberos-provider.c
5fa30de
+++ b/service/realm-kerberos-provider.c
5fa30de
@@ -19,6 +19,7 @@
5fa30de
 #include "realm-kerberos-provider.h"
5fa30de
 
5fa30de
 #include <errno.h>
5fa30de
+#include <string.h>
5fa30de
 
5fa30de
 struct _RealmKerberosProvider {
5fa30de
 	RealmProvider parent;
5fa30de
@@ -38,28 +39,54 @@ realm_kerberos_provider_init (RealmKerberosProvider *self)
5fa30de
 
5fa30de
 }
5fa30de
 
5fa30de
+typedef struct {
5fa30de
+	gchar *name;
5fa30de
+	const char *prot;
5fa30de
+} NameProtPair;
5fa30de
+
5fa30de
+static void
5fa30de
+name_prot_pair_free (gpointer data)
5fa30de
+{
5fa30de
+	NameProtPair *name_prot_pair = data;
5fa30de
+	g_free (name_prot_pair->name);
5fa30de
+	g_free (name_prot_pair);
5fa30de
+}
5fa30de
+
5fa30de
 static void
5fa30de
 on_kerberos_discover (GObject *source,
5fa30de
                       GAsyncResult *result,
5fa30de
                       gpointer user_data)
5fa30de
 {
5fa30de
 	GTask *task = G_TASK (user_data);
5fa30de
-	const gchar *domain = g_task_get_task_data (task);
5fa30de
+	NameProtPair *name_prot_pair = g_task_get_task_data (task);
5fa30de
 	GError *error = NULL;
5fa30de
 	RealmDisco *disco;
5fa30de
 	GList *targets;
5fa30de
+	GResolver *resolver;
5fa30de
 
5fa30de
 	targets = g_resolver_lookup_service_finish (G_RESOLVER (source), result, &error);
5fa30de
 	if (targets) {
5fa30de
 		g_list_free_full (targets, (GDestroyNotify)g_srv_target_free);
5fa30de
-		disco = realm_disco_new (domain);
5fa30de
-		disco->kerberos_realm = g_ascii_strup (domain, -1);
5fa30de
+		disco = realm_disco_new (name_prot_pair->name);
5fa30de
+		disco->kerberos_realm = g_ascii_strup (name_prot_pair->name, -1);
5fa30de
 		g_task_return_pointer (task, disco, realm_disco_unref);
5fa30de
 
5fa30de
 	} else if (error) {
5fa30de
-		g_debug ("Resolving %s failed: %s", domain, error->message);
5fa30de
+		g_debug ("Resolving %s failed: %s", name_prot_pair->name, error->message);
5fa30de
 		g_error_free (error);
5fa30de
-		g_task_return_pointer (task, NULL, NULL);
5fa30de
+
5fa30de
+		if (strcmp (name_prot_pair->prot, "tcp") == 0) {
5fa30de
+			g_task_return_pointer (task, NULL, NULL);
5fa30de
+		} else {
5fa30de
+			/* Try tcp */
5fa30de
+			name_prot_pair->prot = "tcp";
5fa30de
+			resolver = g_resolver_get_default ();
5fa30de
+			g_resolver_lookup_service_async (resolver, "kerberos", name_prot_pair->prot,
5fa30de
+			                                 name_prot_pair->name,
5fa30de
+			                                 g_task_get_cancellable (task),
5fa30de
+			                                 on_kerberos_discover, g_object_ref (task));
5fa30de
+			g_object_unref (resolver);
5fa30de
+		}
5fa30de
 	}
5fa30de
 
5fa30de
 	g_object_unref (task);
5fa30de
@@ -76,7 +103,7 @@ realm_kerberos_provider_discover_async (RealmProvider *provider,
5fa30de
 	GTask *task;
5fa30de
 	const gchar *software;
5fa30de
 	GResolver *resolver;
5fa30de
-	gchar *name;
5fa30de
+	NameProtPair *name_prot_pair;
5fa30de
 
5fa30de
 	task = g_task_new (provider, NULL, callback, user_data);
5fa30de
 
5fa30de
@@ -86,12 +113,15 @@ realm_kerberos_provider_discover_async (RealmProvider *provider,
5fa30de
 		g_task_return_pointer (task, NULL, NULL);
5fa30de
 
5fa30de
 	} else {
5fa30de
-		name = g_hostname_to_ascii (string);
5fa30de
+		name_prot_pair = g_new0 (NameProtPair, 1);
5fa30de
+		name_prot_pair->name = g_hostname_to_ascii (string);
5fa30de
+		name_prot_pair->prot = "udp";
5fa30de
 		resolver = g_resolver_get_default ();
5fa30de
-		g_resolver_lookup_service_async (resolver, "kerberos", "udp", name,
5fa30de
+		g_resolver_lookup_service_async (resolver, "kerberos", name_prot_pair->prot,
5fa30de
+		                                 name_prot_pair->name,
5fa30de
 		                                 realm_invocation_get_cancellable (invocation),
5fa30de
 		                                 on_kerberos_discover, g_object_ref (task));
5fa30de
-		g_task_set_task_data (task, name, g_free);
5fa30de
+		g_task_set_task_data (task, name_prot_pair, name_prot_pair_free);
5fa30de
 		g_object_unref (resolver);
5fa30de
 	}
5fa30de
 
5fa30de
-- 
5fa30de
2.9.3
5fa30de