Blame SOURCES/0023-krb5-Lookahead-resolving-of-host-names.patch

5cd47f
From 385c8c7e0e6f184dd61953745bfe04a5a79a951a Mon Sep 17 00:00:00 2001
5cd47f
From: Tomas Halman <thalman@redhat.com>
5cd47f
Date: Fri, 15 Mar 2019 10:27:50 +0100
5cd47f
Subject: [PATCH 23/23] krb5: Lookahead resolving of host names
5cd47f
5cd47f
The caller that initializes
5cd47f
the fail over service (maybe with be_fo_add_service) should provide
5cd47f
a hint with the value of the lookahead option. Then, if a request for
5cd47f
server resolution is triggered, the fail over code would resolve a server
5cd47f
and afterwards check if enough fo_server entries with a valid hostname
5cd47f
in the struct server_common structure. If not, the request would
5cd47f
check if any of the fo_server structures represents a SRV query and
5cd47f
try to resolve the query to receive more host names.
5cd47f
5cd47f
Resolves:
5cd47f
https://pagure.io/SSSD/sssd/issue/3975
5cd47f
5cd47f
Reviewed-by: Jakub Hrozek <jhrozek@redhat.com>
5cd47f
(cherry picked from commit e8d806d9bbb1ba288ed6a83158113f4d8f8a8929)
5cd47f
---
5cd47f
 Makefile.am                               |  1 +
5cd47f
 src/man/sssd-krb5.5.xml                   | 34 +++++++++++
5cd47f
 src/providers/ad/ad_common.c              | 10 +++-
5cd47f
 src/providers/ad/ad_common.h              |  2 +
5cd47f
 src/providers/ad/ad_init.c                |  2 +
5cd47f
 src/providers/ad/ad_opts.c                |  2 +
5cd47f
 src/providers/ad/ad_subdomains.c          | 12 +++-
5cd47f
 src/providers/ipa/ipa_common.c            | 14 +++--
5cd47f
 src/providers/ipa/ipa_opts.c              |  2 +
5cd47f
 src/providers/ipa/ipa_subdomains.c        |  4 +-
5cd47f
 src/providers/ipa/ipa_subdomains_server.c |  7 +++
5cd47f
 src/providers/krb5/krb5_common.c          | 71 ++++++++++++++++++++++-
5cd47f
 src/providers/krb5/krb5_common.h          | 13 ++++-
5cd47f
 src/providers/krb5/krb5_init.c            | 19 +++++-
5cd47f
 src/providers/krb5/krb5_opts.c            |  1 +
5cd47f
 src/providers/ldap/ldap_common.c          |  9 +++
5cd47f
 src/providers/ldap/ldap_opts.c            |  1 +
5cd47f
 src/providers/ldap/sdap.h                 |  1 +
5cd47f
 18 files changed, 193 insertions(+), 12 deletions(-)
5cd47f
5cd47f
diff --git a/Makefile.am b/Makefile.am
5cd47f
index e22423071..0c24ae664 100644
5cd47f
--- a/Makefile.am
5cd47f
+++ b/Makefile.am
5cd47f
@@ -3339,6 +3339,7 @@ test_ad_subdom_LDADD = \
5cd47f
     libsss_idmap.la \
5cd47f
     libsss_test_common.la \
5cd47f
     libdlopen_test_providers.la \
5cd47f
+    libsss_krb5_common.la \
5cd47f
     $(NULL)
5cd47f
 
5cd47f
 test_ipa_subdom_util_SOURCES = \
5cd47f
diff --git a/src/man/sssd-krb5.5.xml b/src/man/sssd-krb5.5.xml
5cd47f
index 60b7dfb50..5a0bb5e9c 100644
5cd47f
--- a/src/man/sssd-krb5.5.xml
5cd47f
+++ b/src/man/sssd-krb5.5.xml
5cd47f
@@ -501,6 +501,40 @@
5cd47f
                     </listitem>
5cd47f
                 </varlistentry>
5cd47f
 
5cd47f
+                <varlistentry>
5cd47f
+                    <term>krb5_kdcinfo_lookahead (string)</term>
5cd47f
+                    <listitem>
5cd47f
+                        <para>
5cd47f
+                            When krb5_use_kdcinfo is set to true, you can limit the amount
5cd47f
+                            of servers handed to
5cd47f
+                            <citerefentry>
5cd47f
+                                <refentrytitle>sssd_krb5_locator_plugin</refentrytitle>
5cd47f
+                                <manvolnum>8</manvolnum>
5cd47f
+                            </citerefentry>.
5cd47f
+                            This might be helpful when there are too many servers
5cd47f
+                            discovered using SRV record.
5cd47f
+                        </para>
5cd47f
+                        <para>
5cd47f
+                            The krb5_kdcinfo_lookahead option contains two
5cd47f
+                            numbers seperated by a colon. The first number represents
5cd47f
+                            number of primary servers used and the second number
5cd47f
+                            specifies the number of backup servers.
5cd47f
+                        </para>
5cd47f
+                        <para>
5cd47f
+                            For example <emphasis>10:0</emphasis> means that up to
5cd47f
+                            10 primary servers will be handed to
5cd47f
+                            <citerefentry>
5cd47f
+                                <refentrytitle>sssd_krb5_locator_plugin</refentrytitle>
5cd47f
+                                <manvolnum>8</manvolnum>
5cd47f
+                            </citerefentry>.
5cd47f
+                            but no backup servers.
5cd47f
+                        </para>
5cd47f
+                        <para>
5cd47f
+                            Default: 3:1
5cd47f
+                        </para>
5cd47f
+                    </listitem>
5cd47f
+                </varlistentry>
5cd47f
+
5cd47f
                 <varlistentry>
5cd47f
                     <term>krb5_use_enterprise_principal (boolean)</term>
5cd47f
                     <listitem>
5cd47f
diff --git a/src/providers/ad/ad_common.c b/src/providers/ad/ad_common.c
5cd47f
index b7f34daa9..4d1800806 100644
5cd47f
--- a/src/providers/ad/ad_common.c
5cd47f
+++ b/src/providers/ad/ad_common.c
5cd47f
@@ -729,6 +729,8 @@ ad_failover_init(TALLOC_CTX *mem_ctx, struct be_ctx *bectx,
5cd47f
                  const char *ad_gc_service,
5cd47f
                  const char *ad_domain,
5cd47f
                  bool use_kdcinfo,
5cd47f
+                 size_t n_lookahead_primary,
5cd47f
+                 size_t n_lookahead_backup,
5cd47f
                  struct ad_service **_service)
5cd47f
 {
5cd47f
     errno_t ret;
5cd47f
@@ -760,7 +762,9 @@ ad_failover_init(TALLOC_CTX *mem_ctx, struct be_ctx *bectx,
5cd47f
 
5cd47f
     service->krb5_service = krb5_service_new(service, bectx,
5cd47f
                                              ad_service, krb5_realm,
5cd47f
-                                             use_kdcinfo);
5cd47f
+                                             use_kdcinfo,
5cd47f
+                                             n_lookahead_primary,
5cd47f
+                                             n_lookahead_backup);
5cd47f
     if (!service->krb5_service) {
5cd47f
         ret = ENOMEM;
5cd47f
         goto done;
5cd47f
@@ -1292,6 +1296,10 @@ ad_get_auth_options(TALLOC_CTX *mem_ctx,
5cd47f
     DEBUG(SSSDBG_CONF_SETTINGS, "Option %s set to %s\n",
5cd47f
           krb5_options[KRB5_USE_KDCINFO].opt_name,
5cd47f
           ad_opts->service->krb5_service->write_kdcinfo ? "true" : "false");
5cd47f
+    sss_krb5_parse_lookahead(
5cd47f
+        dp_opt_get_string(krb5_options, KRB5_KDCINFO_LOOKAHEAD),
5cd47f
+        &ad_opts->service->krb5_service->lookahead_primary,
5cd47f
+        &ad_opts->service->krb5_service->lookahead_backup);
5cd47f
 
5cd47f
     *_opts = talloc_steal(mem_ctx, krb5_options);
5cd47f
 
5cd47f
diff --git a/src/providers/ad/ad_common.h b/src/providers/ad/ad_common.h
5cd47f
index 529753a8a..638465958 100644
5cd47f
--- a/src/providers/ad/ad_common.h
5cd47f
+++ b/src/providers/ad/ad_common.h
5cd47f
@@ -147,6 +147,8 @@ ad_failover_init(TALLOC_CTX *mem_ctx, struct be_ctx *ctx,
5cd47f
                  const char *ad_gc_service,
5cd47f
                  const char *ad_domain,
5cd47f
                  bool use_kdcinfo,
5cd47f
+                 size_t n_lookahead_primary,
5cd47f
+                 size_t n_lookahead_backup,
5cd47f
                  struct ad_service **_service);
5cd47f
 
5cd47f
 errno_t
5cd47f
diff --git a/src/providers/ad/ad_init.c b/src/providers/ad/ad_init.c
5cd47f
index 637efb761..612d4587e 100644
5cd47f
--- a/src/providers/ad/ad_init.c
5cd47f
+++ b/src/providers/ad/ad_init.c
5cd47f
@@ -160,6 +160,8 @@ static errno_t ad_init_options(TALLOC_CTX *mem_ctx,
5cd47f
                            ad_realm, AD_SERVICE_NAME, AD_GC_SERVICE_NAME,
5cd47f
                            dp_opt_get_string(ad_options->basic, AD_DOMAIN),
5cd47f
                            false, /* will be set in ad_get_auth_options() */
5cd47f
+                           (size_t) -1,
5cd47f
+                           (size_t) -1,
5cd47f
                            &ad_options->service);
5cd47f
     if (ret != EOK) {
5cd47f
         DEBUG(SSSDBG_FATAL_FAILURE, "Failed to init AD failover service: "
5cd47f
diff --git a/src/providers/ad/ad_opts.c b/src/providers/ad/ad_opts.c
5cd47f
index c408295f3..652d8bb27 100644
5cd47f
--- a/src/providers/ad/ad_opts.c
5cd47f
+++ b/src/providers/ad/ad_opts.c
5cd47f
@@ -111,6 +111,7 @@ struct dp_option ad_def_ldap_opts[] = {
5cd47f
     { "krb5_realm", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     { "krb5_canonicalize", DP_OPT_BOOL, BOOL_FALSE, BOOL_FALSE },
5cd47f
     { "krb5_use_kdcinfo", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
+    { "krb5_kdcinfo_lookahead", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     { "ldap_pwd_policy", DP_OPT_STRING, { "none" }, NULL_STRING },
5cd47f
     { "ldap_referrals", DP_OPT_BOOL, BOOL_FALSE, BOOL_FALSE },
5cd47f
     { "account_cache_expiration", DP_OPT_NUMBER, { .number = 0 }, NULL_NUMBER },
5cd47f
@@ -175,6 +176,7 @@ struct dp_option ad_def_krb5_opts[] = {
5cd47f
     { "krb5_canonicalize", DP_OPT_BOOL, BOOL_FALSE, BOOL_FALSE },
5cd47f
     { "krb5_use_enterprise_principal", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
     { "krb5_use_kdcinfo", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
+    { "krb5_kdcinfo_lookahead", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     { "krb5_map_user", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     DP_OPTION_TERMINATOR
5cd47f
 };
5cd47f
diff --git a/src/providers/ad/ad_subdomains.c b/src/providers/ad/ad_subdomains.c
5cd47f
index 4fc4be094..b4ad347e4 100644
5cd47f
--- a/src/providers/ad/ad_subdomains.c
5cd47f
+++ b/src/providers/ad/ad_subdomains.c
5cd47f
@@ -280,6 +280,8 @@ ad_subdom_ad_ctx_new(struct be_ctx *be_ctx,
5cd47f
     const char *keytab;
5cd47f
     char *subdom_conf_path;
5cd47f
     bool use_kdcinfo = false;
5cd47f
+    size_t n_lookahead_primary = SSS_KRB5_LOOKAHEAD_PRIMARY_DEFAULT;
5cd47f
+    size_t n_lookahead_backup = SSS_KRB5_LOOKAHEAD_BACKUP_DEFAULT;
5cd47f
 
5cd47f
     realm = dp_opt_get_cstring(id_ctx->ad_options->basic, AD_KRB5_REALM);
5cd47f
     hostname = dp_opt_get_cstring(id_ctx->ad_options->basic, AD_HOSTNAME);
5cd47f
@@ -331,6 +333,11 @@ ad_subdom_ad_ctx_new(struct be_ctx *be_ctx,
5cd47f
             && id_ctx->ad_options->auth_ctx->opts != NULL) {
5cd47f
         use_kdcinfo = dp_opt_get_bool(id_ctx->ad_options->auth_ctx->opts,
5cd47f
                                       KRB5_USE_KDCINFO);
5cd47f
+        sss_krb5_parse_lookahead(
5cd47f
+            dp_opt_get_string(id_ctx->ad_options->auth_ctx->opts,
5cd47f
+                              KRB5_KDCINFO_LOOKAHEAD),
5cd47f
+            &n_lookahead_primary,
5cd47f
+            &n_lookahead_backup);
5cd47f
     }
5cd47f
 
5cd47f
     DEBUG(SSSDBG_TRACE_ALL,
5cd47f
@@ -339,7 +346,10 @@ ad_subdom_ad_ctx_new(struct be_ctx *be_ctx,
5cd47f
 
5cd47f
     ret = ad_failover_init(ad_options, be_ctx, servers, backup_servers,
5cd47f
                            subdom->realm, service_name, gc_service_name,
5cd47f
-                           subdom->name, use_kdcinfo, &ad_options->service);
5cd47f
+                           subdom->name, use_kdcinfo,
5cd47f
+                           n_lookahead_primary,
5cd47f
+                           n_lookahead_backup,
5cd47f
+                           &ad_options->service);
5cd47f
     if (ret != EOK) {
5cd47f
         DEBUG(SSSDBG_OP_FAILURE, "Cannot initialize AD failover\n");
5cd47f
         talloc_free(ad_options);
5cd47f
diff --git a/src/providers/ipa/ipa_common.c b/src/providers/ipa/ipa_common.c
5cd47f
index 1ed2e2203..871fb9bbc 100644
5cd47f
--- a/src/providers/ipa/ipa_common.c
5cd47f
+++ b/src/providers/ipa/ipa_common.c
5cd47f
@@ -801,6 +801,12 @@ int ipa_get_auth_options(struct ipa_options *ipa_opts,
5cd47f
     DEBUG(SSSDBG_CONF_SETTINGS, "Option %s set to %s\n",
5cd47f
           ipa_opts->auth[KRB5_USE_KDCINFO].opt_name,
5cd47f
           ipa_opts->service->krb5_service->write_kdcinfo ? "true" : "false");
5cd47f
+    if (ipa_opts->service->krb5_service->write_kdcinfo) {
5cd47f
+        sss_krb5_parse_lookahead(
5cd47f
+            dp_opt_get_string(ipa_opts->auth, KRB5_KDCINFO_LOOKAHEAD),
5cd47f
+            &ipa_opts->service->krb5_service->lookahead_primary,
5cd47f
+            &ipa_opts->service->krb5_service->lookahead_backup);
5cd47f
+    }
5cd47f
 
5cd47f
     *_opts = ipa_opts->auth;
5cd47f
     ret = EOK;
5cd47f
@@ -1022,10 +1028,10 @@ int ipa_service_init(TALLOC_CTX *memctx, struct be_ctx *ctx,
5cd47f
 
5cd47f
     service->krb5_service = krb5_service_new(service, ctx,
5cd47f
                                              "IPA", realm,
5cd47f
-                                             true); /* The configured value
5cd47f
-                                                     * will be set later when
5cd47f
-                                                     * the auth provider is set up
5cd47f
-                                                     */
5cd47f
+                                             true,   /* The configured value */
5cd47f
+                                             0,      /* will be set later when */
5cd47f
+                                             0);     /* the auth provider is set up */
5cd47f
+
5cd47f
     if (!service->krb5_service) {
5cd47f
         ret = ENOMEM;
5cd47f
         goto done;
5cd47f
diff --git a/src/providers/ipa/ipa_opts.c b/src/providers/ipa/ipa_opts.c
5cd47f
index 373dc0e53..600b9ec4b 100644
5cd47f
--- a/src/providers/ipa/ipa_opts.c
5cd47f
+++ b/src/providers/ipa/ipa_opts.c
5cd47f
@@ -121,6 +121,7 @@ struct dp_option ipa_def_ldap_opts[] = {
5cd47f
     { "krb5_realm", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     { "krb5_canonicalize", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
     { "krb5_use_kdcinfo", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
+    { "krb5_kdcinfo_lookahead", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     { "ldap_pwd_policy", DP_OPT_STRING, { "none" } , NULL_STRING },
5cd47f
     { "ldap_referrals", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
     { "account_cache_expiration", DP_OPT_NUMBER, { .number = 0 }, NULL_NUMBER },
5cd47f
@@ -320,6 +321,7 @@ struct dp_option ipa_def_krb5_opts[] = {
5cd47f
     { "krb5_canonicalize", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
     { "krb5_use_enterprise_principal", DP_OPT_BOOL, BOOL_FALSE, BOOL_FALSE },
5cd47f
     { "krb5_use_kdcinfo", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
+    { "krb5_kdcinfo_lookahead", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     { "krb5_map_user", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     DP_OPTION_TERMINATOR
5cd47f
 };
5cd47f
diff --git a/src/providers/ipa/ipa_subdomains.c b/src/providers/ipa/ipa_subdomains.c
5cd47f
index da1279e3e..94365aaca 100644
5cd47f
--- a/src/providers/ipa/ipa_subdomains.c
5cd47f
+++ b/src/providers/ipa/ipa_subdomains.c
5cd47f
@@ -692,7 +692,9 @@ ipa_subdom_get_k5_svc(struct ipa_subdomains_ctx *ctx,
5cd47f
                                         ctx->be_ctx,
5cd47f
                                         "IPA",
5cd47f
                                         dom->realm,
5cd47f
-                                        use_kdcinfo);
5cd47f
+                                        use_kdcinfo,
5cd47f
+                                        (size_t) -1,
5cd47f
+                                        (size_t) -1);
5cd47f
     if (k5svc_ent->k5svc == NULL) {
5cd47f
         talloc_free(k5svc_ent);
5cd47f
         return NULL;
5cd47f
diff --git a/src/providers/ipa/ipa_subdomains_server.c b/src/providers/ipa/ipa_subdomains_server.c
5cd47f
index 43a3053cb..dd0933642 100644
5cd47f
--- a/src/providers/ipa/ipa_subdomains_server.c
5cd47f
+++ b/src/providers/ipa/ipa_subdomains_server.c
5cd47f
@@ -225,6 +225,8 @@ ipa_ad_ctx_new(struct be_ctx *be_ctx,
5cd47f
     errno_t ret;
5cd47f
     const char *extra_attrs;
5cd47f
     bool use_kdcinfo = false;
5cd47f
+    size_t n_lookahead_primary = (size_t)-1;
5cd47f
+    size_t n_lookahead_backup = (size_t)-1;
5cd47f
 
5cd47f
     ad_domain = subdom->name;
5cd47f
     DEBUG(SSSDBG_TRACE_LIBS, "Setting up AD subdomain %s\n", subdom->name);
5cd47f
@@ -284,6 +286,10 @@ ipa_ad_ctx_new(struct be_ctx *be_ctx,
5cd47f
     if (id_ctx->ipa_options != NULL && id_ctx->ipa_options->auth != NULL) {
5cd47f
         use_kdcinfo = dp_opt_get_bool(id_ctx->ipa_options->auth,
5cd47f
                                       KRB5_USE_KDCINFO);
5cd47f
+        sss_krb5_parse_lookahead(
5cd47f
+            dp_opt_get_string(id_ctx->ipa_options->auth, KRB5_KDCINFO_LOOKAHEAD),
5cd47f
+            &n_lookahead_primary,
5cd47f
+            &n_lookahead_backup);
5cd47f
     }
5cd47f
 
5cd47f
     DEBUG(SSSDBG_TRACE_ALL,
5cd47f
@@ -297,6 +303,7 @@ ipa_ad_ctx_new(struct be_ctx *be_ctx,
5cd47f
                            subdom->realm,
5cd47f
                            service_name, gc_service_name,
5cd47f
                            subdom->name, use_kdcinfo,
5cd47f
+                           n_lookahead_primary, n_lookahead_backup,
5cd47f
                            &ad_options->service);
5cd47f
     if (ret != EOK) {
5cd47f
         DEBUG(SSSDBG_OP_FAILURE, "Cannot initialize AD failover\n");
5cd47f
diff --git a/src/providers/krb5/krb5_common.c b/src/providers/krb5/krb5_common.c
5cd47f
index 1e33fc0f5..e56820b8d 100644
5cd47f
--- a/src/providers/krb5/krb5_common.c
5cd47f
+++ b/src/providers/krb5/krb5_common.c
5cd47f
@@ -390,6 +390,39 @@ done:
5cd47f
     return ret;
5cd47f
 }
5cd47f
 
5cd47f
+void sss_krb5_parse_lookahead(const char *param, size_t *primary, size_t *backup)
5cd47f
+{
5cd47f
+    int ret;
5cd47f
+
5cd47f
+    if (primary == NULL || backup == NULL) {
5cd47f
+        return;
5cd47f
+    }
5cd47f
+
5cd47f
+    *primary = SSS_KRB5_LOOKAHEAD_PRIMARY_DEFAULT;
5cd47f
+    *backup = SSS_KRB5_LOOKAHEAD_BACKUP_DEFAULT;
5cd47f
+
5cd47f
+    if (param == NULL) {
5cd47f
+        return;
5cd47f
+    }
5cd47f
+
5cd47f
+    if (strchr(param, ':')) {
5cd47f
+        ret = sscanf(param, "%zu:%zu", primary, backup);
5cd47f
+        if (ret != 2) {
5cd47f
+            DEBUG(SSSDBG_MINOR_FAILURE, "Could not parse krb5_kdcinfo_lookahead!\n");
5cd47f
+        }
5cd47f
+    } else {
5cd47f
+        ret = sscanf(param, "%zu", primary);
5cd47f
+        if (ret != 1) {
5cd47f
+            DEBUG(SSSDBG_MINOR_FAILURE, "Could not parse krb5_kdcinfo_lookahead!\n");
5cd47f
+        }
5cd47f
+    }
5cd47f
+
5cd47f
+    DEBUG(SSSDBG_CONF_SETTINGS,
5cd47f
+          "Option krb5_kdcinfo_lookahead set to %zu:%zu",
5cd47f
+          *primary, *backup);
5cd47f
+}
5cd47f
+
5cd47f
+
5cd47f
 static int remove_info_files_destructor(void *p)
5cd47f
 {
5cd47f
     int ret;
5cd47f
@@ -668,6 +701,13 @@ errno_t write_krb5info_file_from_fo_server(struct krb5_service *krb5_service,
5cd47f
     int primary;
5cd47f
     const char *address;
5cd47f
     errno_t ret;
5cd47f
+    size_t n_lookahead_primary;
5cd47f
+    size_t n_lookahead_backup;
5cd47f
+
5cd47f
+    if (krb5_service == NULL) {
5cd47f
+        DEBUG(SSSDBG_CRIT_FAILURE, "The krb5_service must not be NULL!\n");
5cd47f
+        return EINVAL;
5cd47f
+    }
5cd47f
 
5cd47f
     tmp_ctx = talloc_new(NULL);
5cd47f
     if (tmp_ctx == NULL) {
5cd47f
@@ -675,6 +715,9 @@ errno_t write_krb5info_file_from_fo_server(struct krb5_service *krb5_service,
5cd47f
         return ENOMEM;
5cd47f
     }
5cd47f
 
5cd47f
+    n_lookahead_primary = krb5_service->lookahead_primary;
5cd47f
+    n_lookahead_backup = krb5_service->lookahead_backup;
5cd47f
+
5cd47f
     server_idx = 0;
5cd47f
     server_list = talloc_zero_array(tmp_ctx,
5cd47f
                                     const char *,
5cd47f
@@ -689,6 +732,15 @@ errno_t write_krb5info_file_from_fo_server(struct krb5_service *krb5_service,
5cd47f
         address = fo_server_address_or_name(tmp_ctx, server);
5cd47f
         if (address) {
5cd47f
             server_list[server_idx++] = address;
5cd47f
+            if (fo_is_server_primary(server)) {
5cd47f
+                if (n_lookahead_primary > 0) {
5cd47f
+                    n_lookahead_primary--;
5cd47f
+                }
5cd47f
+            } else {
5cd47f
+                if (n_lookahead_backup > 0) {
5cd47f
+                    n_lookahead_backup--;
5cd47f
+                }
5cd47f
+            }
5cd47f
         } else {
5cd47f
             DEBUG(SSSDBG_CRIT_FAILURE,
5cd47f
                   "Server without name and address found in list.\n");
5cd47f
@@ -700,6 +752,8 @@ errno_t write_krb5info_file_from_fo_server(struct krb5_service *krb5_service,
5cd47f
              item != server;
5cd47f
              item = fo_server_next(item) ? fo_server_next(item) : fo_server_first(item)) {
5cd47f
 
5cd47f
+            if (primary && n_lookahead_primary == 0) break;
5cd47f
+            if (!primary && n_lookahead_backup == 0) break;
5cd47f
             if (primary && !fo_is_server_primary(item)) continue;
5cd47f
             if (!primary && fo_is_server_primary(item)) continue;
5cd47f
             if (filter != NULL && filter(item)) continue;
5cd47f
@@ -712,6 +766,11 @@ errno_t write_krb5info_file_from_fo_server(struct krb5_service *krb5_service,
5cd47f
             }
5cd47f
 
5cd47f
             server_list[server_idx++] = address;
5cd47f
+            if (primary) {
5cd47f
+                n_lookahead_primary--;
5cd47f
+            } else {
5cd47f
+                n_lookahead_backup--;
5cd47f
+            }
5cd47f
         }
5cd47f
     }
5cd47f
     if (server_list[0] == NULL) {
5cd47f
@@ -901,7 +960,9 @@ struct krb5_service *krb5_service_new(TALLOC_CTX *mem_ctx,
5cd47f
                                       struct be_ctx *be_ctx,
5cd47f
                                       const char *service_name,
5cd47f
                                       const char *realm,
5cd47f
-                                      bool use_kdcinfo)
5cd47f
+                                      bool use_kdcinfo,
5cd47f
+                                      size_t n_lookahead_primary,
5cd47f
+                                      size_t n_lookahead_backup)
5cd47f
 {
5cd47f
     struct krb5_service *service;
5cd47f
 
5cd47f
@@ -927,6 +988,9 @@ struct krb5_service *krb5_service_new(TALLOC_CTX *mem_ctx,
5cd47f
           realm,
5cd47f
           use_kdcinfo ? "true" : "false");
5cd47f
     service->write_kdcinfo = use_kdcinfo;
5cd47f
+    service->lookahead_primary = n_lookahead_primary;
5cd47f
+    service->lookahead_backup = n_lookahead_backup;
5cd47f
+
5cd47f
     service->be_ctx = be_ctx;
5cd47f
     return service;
5cd47f
 }
5cd47f
@@ -937,6 +1001,8 @@ int krb5_service_init(TALLOC_CTX *memctx, struct be_ctx *ctx,
5cd47f
                       const char *backup_servers,
5cd47f
                       const char *realm,
5cd47f
                       bool use_kdcinfo,
5cd47f
+                      size_t n_lookahead_primary,
5cd47f
+                      size_t n_lookahead_backup,
5cd47f
                       struct krb5_service **_service)
5cd47f
 {
5cd47f
     TALLOC_CTX *tmp_ctx;
5cd47f
@@ -948,7 +1014,8 @@ int krb5_service_init(TALLOC_CTX *memctx, struct be_ctx *ctx,
5cd47f
         return ENOMEM;
5cd47f
     }
5cd47f
 
5cd47f
-    service = krb5_service_new(tmp_ctx, ctx, service_name, realm, use_kdcinfo);
5cd47f
+    service = krb5_service_new(tmp_ctx, ctx, service_name, realm, use_kdcinfo,
5cd47f
+                               n_lookahead_primary, n_lookahead_backup);
5cd47f
     if (!service) {
5cd47f
         ret = ENOMEM;
5cd47f
         goto done;
5cd47f
diff --git a/src/providers/krb5/krb5_common.h b/src/providers/krb5/krb5_common.h
5cd47f
index be541626b..441c52b34 100644
5cd47f
--- a/src/providers/krb5/krb5_common.h
5cd47f
+++ b/src/providers/krb5/krb5_common.h
5cd47f
@@ -38,6 +38,8 @@
5cd47f
 
5cd47f
 #define SSS_KRB5KDC_FO_SRV "KERBEROS"
5cd47f
 #define SSS_KRB5KPASSWD_FO_SRV "KPASSWD"
5cd47f
+#define SSS_KRB5_LOOKAHEAD_PRIMARY_DEFAULT 3
5cd47f
+#define SSS_KRB5_LOOKAHEAD_BACKUP_DEFAULT 1
5cd47f
 
5cd47f
 enum krb5_opts {
5cd47f
     KRB5_KDC = 0,
5cd47f
@@ -59,6 +61,7 @@ enum krb5_opts {
5cd47f
     KRB5_CANONICALIZE,
5cd47f
     KRB5_USE_ENTERPRISE_PRINCIPAL,
5cd47f
     KRB5_USE_KDCINFO,
5cd47f
+    KRB5_KDCINFO_LOOKAHEAD,
5cd47f
     KRB5_MAP_USER,
5cd47f
 
5cd47f
     KRB5_OPTS
5cd47f
@@ -71,6 +74,8 @@ struct krb5_service {
5cd47f
     char *name;
5cd47f
     char *realm;
5cd47f
     bool write_kdcinfo;
5cd47f
+    size_t lookahead_primary;
5cd47f
+    size_t lookahead_backup;
5cd47f
     bool removal_callback_available;
5cd47f
 };
5cd47f
 
5cd47f
@@ -160,6 +165,8 @@ errno_t krb5_try_kdcip(struct confdb_ctx *cdb, const char *conf_path,
5cd47f
 errno_t sss_krb5_get_options(TALLOC_CTX *memctx, struct confdb_ctx *cdb,
5cd47f
                              const char *conf_path, struct dp_option **_opts);
5cd47f
 
5cd47f
+void sss_krb5_parse_lookahead(const char *param, size_t *primary, size_t *backup);
5cd47f
+
5cd47f
 errno_t write_krb5info_file(struct krb5_service *krb5_service,
5cd47f
                             const char **server_list,
5cd47f
                             const char *service);
5cd47f
@@ -173,7 +180,9 @@ struct krb5_service *krb5_service_new(TALLOC_CTX *mem_ctx,
5cd47f
                                       struct be_ctx *be_ctx,
5cd47f
                                       const char *service_name,
5cd47f
                                       const char *realm,
5cd47f
-                                      bool use_kdcinfo);
5cd47f
+                                      bool use_kdcinfo,
5cd47f
+                                      size_t n_lookahead_primary,
5cd47f
+                                      size_t n_lookahead_backup);
5cd47f
 
5cd47f
 int krb5_service_init(TALLOC_CTX *memctx, struct be_ctx *ctx,
5cd47f
                       const char *service_name,
5cd47f
@@ -181,6 +190,8 @@ int krb5_service_init(TALLOC_CTX *memctx, struct be_ctx *ctx,
5cd47f
                       const char *backup_servers,
5cd47f
                       const char *realm,
5cd47f
                       bool use_kdcinfo,
5cd47f
+                      size_t n_lookahead_primary,
5cd47f
+                      size_t n_lookahead_backup,
5cd47f
                       struct krb5_service **_service);
5cd47f
 
5cd47f
 void remove_krb5_info_files_callback(void *pvt);
5cd47f
diff --git a/src/providers/krb5/krb5_init.c b/src/providers/krb5/krb5_init.c
5cd47f
index 66ae68fb4..3f4c1b361 100644
5cd47f
--- a/src/providers/krb5/krb5_init.c
5cd47f
+++ b/src/providers/krb5/krb5_init.c
5cd47f
@@ -40,6 +40,8 @@ static errno_t krb5_init_kpasswd(struct krb5_ctx *ctx,
5cd47f
     const char *backup_servers;
5cd47f
     const char *kdc_servers;
5cd47f
     bool use_kdcinfo;
5cd47f
+    size_t n_lookahead_primary;
5cd47f
+    size_t n_lookahead_backup;
5cd47f
     errno_t ret;
5cd47f
 
5cd47f
     realm = dp_opt_get_string(ctx->opts, KRB5_REALM);
5cd47f
@@ -52,6 +54,9 @@ static errno_t krb5_init_kpasswd(struct krb5_ctx *ctx,
5cd47f
     primary_servers = dp_opt_get_string(ctx->opts, KRB5_KPASSWD);
5cd47f
     backup_servers = dp_opt_get_string(ctx->opts, KRB5_BACKUP_KPASSWD);
5cd47f
     use_kdcinfo = dp_opt_get_bool(ctx->opts, KRB5_USE_KDCINFO);
5cd47f
+    sss_krb5_parse_lookahead(dp_opt_get_string(ctx->opts, KRB5_KDCINFO_LOOKAHEAD),
5cd47f
+                             &n_lookahead_primary, &n_lookahead_backup);
5cd47f
+
5cd47f
 
5cd47f
     if (primary_servers == NULL && backup_servers != NULL) {
5cd47f
         DEBUG(SSSDBG_CONF_SETTINGS, "kpasswd server wasn't specified but "
5cd47f
@@ -67,7 +72,10 @@ static errno_t krb5_init_kpasswd(struct krb5_ctx *ctx,
5cd47f
     } else {
5cd47f
         ret = krb5_service_init(ctx, be_ctx, SSS_KRB5KPASSWD_FO_SRV,
5cd47f
                                 primary_servers, backup_servers, realm,
5cd47f
-                                use_kdcinfo, &ctx->kpasswd_service);
5cd47f
+                                use_kdcinfo,
5cd47f
+                                n_lookahead_primary,
5cd47f
+                                n_lookahead_backup,
5cd47f
+                                &ctx->kpasswd_service);
5cd47f
         if (ret != EOK) {
5cd47f
             DEBUG(SSSDBG_FATAL_FAILURE,
5cd47f
                   "Failed to init KRB5KPASSWD failover service!\n");
5cd47f
@@ -84,6 +92,8 @@ static errno_t krb5_init_kdc(struct krb5_ctx *ctx, struct be_ctx *be_ctx)
5cd47f
     const char *backup_servers;
5cd47f
     const char *realm;
5cd47f
     bool use_kdcinfo;
5cd47f
+    size_t n_lookahead_primary;
5cd47f
+    size_t n_lookahead_backup;
5cd47f
     errno_t ret;
5cd47f
 
5cd47f
     realm = dp_opt_get_string(ctx->opts, KRB5_REALM);
5cd47f
@@ -96,10 +106,15 @@ static errno_t krb5_init_kdc(struct krb5_ctx *ctx, struct be_ctx *be_ctx)
5cd47f
     backup_servers = dp_opt_get_string(ctx->opts, KRB5_BACKUP_KDC);
5cd47f
 
5cd47f
     use_kdcinfo = dp_opt_get_bool(ctx->opts, KRB5_USE_KDCINFO);
5cd47f
+    sss_krb5_parse_lookahead(dp_opt_get_string(ctx->opts, KRB5_KDCINFO_LOOKAHEAD),
5cd47f
+                             &n_lookahead_primary, &n_lookahead_backup);
5cd47f
 
5cd47f
     ret = krb5_service_init(ctx, be_ctx, SSS_KRB5KDC_FO_SRV,
5cd47f
                             primary_servers, backup_servers, realm,
5cd47f
-                            use_kdcinfo, &ctx->service);
5cd47f
+                            use_kdcinfo,
5cd47f
+                            n_lookahead_primary,
5cd47f
+                            n_lookahead_backup,
5cd47f
+                            &ctx->service);
5cd47f
     if (ret != EOK) {
5cd47f
         DEBUG(SSSDBG_FATAL_FAILURE, "Failed to init KRB5 failover service!\n");
5cd47f
         return ret;
5cd47f
diff --git a/src/providers/krb5/krb5_opts.c b/src/providers/krb5/krb5_opts.c
5cd47f
index 6bec52767..05395e0f4 100644
5cd47f
--- a/src/providers/krb5/krb5_opts.c
5cd47f
+++ b/src/providers/krb5/krb5_opts.c
5cd47f
@@ -42,6 +42,7 @@ struct dp_option default_krb5_opts[] = {
5cd47f
     { "krb5_canonicalize", DP_OPT_BOOL, BOOL_FALSE, BOOL_FALSE },
5cd47f
     { "krb5_use_enterprise_principal", DP_OPT_BOOL, BOOL_FALSE, BOOL_FALSE },
5cd47f
     { "krb5_use_kdcinfo", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
+    { "krb5_kdcinfo_lookahead", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     { "krb5_map_user", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     DP_OPTION_TERMINATOR
5cd47f
 };
5cd47f
diff --git a/src/providers/ldap/ldap_common.c b/src/providers/ldap/ldap_common.c
5cd47f
index 9cd8ec09c..4c49f08c5 100644
5cd47f
--- a/src/providers/ldap/ldap_common.c
5cd47f
+++ b/src/providers/ldap/ldap_common.c
5cd47f
@@ -335,6 +335,8 @@ int sdap_gssapi_init(TALLOC_CTX *mem_ctx,
5cd47f
     const char *krb5_opt_realm;
5cd47f
     struct krb5_service *service = NULL;
5cd47f
     TALLOC_CTX *tmp_ctx;
5cd47f
+    size_t n_lookahead_primary;
5cd47f
+    size_t n_lookahead_backup;
5cd47f
 
5cd47f
     tmp_ctx = talloc_new(NULL);
5cd47f
     if (tmp_ctx == NULL) return ENOMEM;
5cd47f
@@ -361,11 +363,18 @@ int sdap_gssapi_init(TALLOC_CTX *mem_ctx,
5cd47f
         }
5cd47f
     }
5cd47f
 
5cd47f
+    sss_krb5_parse_lookahead(
5cd47f
+        dp_opt_get_string(opts, SDAP_KRB5_KDCINFO_LOOKAHEAD),
5cd47f
+        &n_lookahead_primary,
5cd47f
+        &n_lookahead_backup);
5cd47f
+
5cd47f
     ret = krb5_service_init(mem_ctx, bectx,
5cd47f
                             SSS_KRB5KDC_FO_SRV, krb5_servers,
5cd47f
                             krb5_backup_servers, krb5_realm,
5cd47f
                             dp_opt_get_bool(opts,
5cd47f
                                             SDAP_KRB5_USE_KDCINFO),
5cd47f
+                            n_lookahead_primary,
5cd47f
+                            n_lookahead_backup,
5cd47f
                             &service);
5cd47f
     if (ret != EOK) {
5cd47f
         DEBUG(SSSDBG_FATAL_FAILURE, "Failed to init KRB5 failover service!\n");
5cd47f
diff --git a/src/providers/ldap/ldap_opts.c b/src/providers/ldap/ldap_opts.c
5cd47f
index f7574fac2..613fe7463 100644
5cd47f
--- a/src/providers/ldap/ldap_opts.c
5cd47f
+++ b/src/providers/ldap/ldap_opts.c
5cd47f
@@ -82,6 +82,7 @@ struct dp_option default_basic_opts[] = {
5cd47f
     { "krb5_realm", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     { "krb5_canonicalize", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
     { "krb5_use_kdcinfo", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
+    { "krb5_kdcinfo_lookahead", DP_OPT_STRING, NULL_STRING, NULL_STRING },
5cd47f
     { "ldap_pwd_policy", DP_OPT_STRING, { "none" }, NULL_STRING },
5cd47f
     { "ldap_referrals", DP_OPT_BOOL, BOOL_TRUE, BOOL_TRUE },
5cd47f
     { "account_cache_expiration", DP_OPT_NUMBER, { .number = 0 }, NULL_NUMBER },
5cd47f
diff --git a/src/providers/ldap/sdap.h b/src/providers/ldap/sdap.h
5cd47f
index 0f79ae9de..48061d389 100644
5cd47f
--- a/src/providers/ldap/sdap.h
5cd47f
+++ b/src/providers/ldap/sdap.h
5cd47f
@@ -199,6 +199,7 @@ enum sdap_basic_opt {
5cd47f
     SDAP_KRB5_REALM,
5cd47f
     SDAP_KRB5_CANONICALIZE,
5cd47f
     SDAP_KRB5_USE_KDCINFO,
5cd47f
+    SDAP_KRB5_KDCINFO_LOOKAHEAD,
5cd47f
     SDAP_PWD_POLICY,
5cd47f
     SDAP_REFERRALS,
5cd47f
     SDAP_ACCOUNT_CACHE_EXPIRATION,
5cd47f
-- 
5cd47f
2.19.1
5cd47f