From 9396e620134760e7b562b5452f34ec80dc6f2af7 Mon Sep 17 00:00:00 2001
From: Stephen Gallagher <sgallagh@redhat.com>
Date: Thu, 6 Nov 2008 15:45:24 -0500
Subject: Store all domains served by the SSSD to a binary-tree map for fast
 NSS lookup.

Changed the "section" feature of confdb.c to use '/'
as a delimiter instead of '.', because this conflicted
with the ability to use dots in domain names.
---
 server/nss/nsssrv.c     | 58 +++++++++++++++++++++++++++++++++++++++++++++++--
 server/nss/nsssrv.h     |  1 +
 server/nss/nsssrv_ldb.c |  2 +-
 3 files changed, 58 insertions(+), 3 deletions(-)

(limited to 'server/nss')

diff --git a/server/nss/nsssrv.c b/server/nss/nsssrv.c
index bd0f761b..93f8d223 100644
--- a/server/nss/nsssrv.c
+++ b/server/nss/nsssrv.c
@@ -38,9 +38,12 @@
 #include "dbus/dbus.h"
 #include "sbus/sssd_dbus.h"
 #include "sbus_interfaces.h"
+#include "util/btreemap.h"
 
 static int provide_identity(DBusMessage *message, void *data, DBusMessage **r);
 static int reply_ping(DBusMessage *message, void *data, DBusMessage **r);
+static int nss_init_domains(struct nss_ctx *nctx);
+static int _domain_comparator(void *key1, void *key2);
 
 struct sbus_method nss_sbus_methods[] = {
     {SERVICE_METHOD_IDENTITY, provide_identity},
@@ -250,7 +253,7 @@ static int nss_sbus_init(struct nss_ctx *nctx)
     int ret;
 
     ret = confdb_get_string(nctx->cdb, nctx,
-                            "config.services.monitor", "sbusAddress",
+                            "config/services/monitor", "sbusAddress",
                             DEFAULT_SBUS_ADDRESS, &sbus_address);
     if (ret != EOK) {
         return ret;
@@ -313,7 +316,7 @@ static int set_unix_socket(struct nss_ctx *nctx)
     int ret;
 
     ret = confdb_get_string(nctx->cdb, nctx,
-                            "config.services.nss", "unixSocket",
+                            "config/services/nss", "unixSocket",
                             SSS_NSS_SOCKET_NAME, &nctx->sock_name);
     if (ret != EOK) {
         return ret;
@@ -363,6 +366,51 @@ failed:
     return EIO;
 }
 
+static int _domain_comparator(void *key1, void *key2)
+{
+    return strcmp((char *)key1, (char *)key2);
+}
+
+static int nss_init_domains(struct nss_ctx *nctx)
+{
+    char **domains;
+    char *basedn;
+    TALLOC_CTX *tmp_ctx;
+    int ret, i;
+    int retval;
+
+    tmp_ctx = talloc_new(nctx);
+    ret = confdb_get_domains(nctx->cdb, tmp_ctx, &domains);
+    if (ret != EOK) {
+        retval = ret;
+        goto done;
+    }
+
+    i = 0;
+    while (domains[i] != NULL) {
+        DEBUG(3, ("Adding domain %s to the map\n", domains[i]));
+        /* Look up the appropriate basedn for this domain */
+        ret = confdb_get_domain_basedn(nctx->cdb, tmp_ctx, domains[i], &basedn);
+        DEBUG(3, ("BaseDN: %s\n", basedn));
+        btreemap_set_value(&nctx->domain_map, domains[i], basedn, _domain_comparator);
+        i++;
+    }
+    if (i == 0) {
+        /* No domains configured!
+         * Note: this should never happen, since LOCAL should
+         * always be configured */
+        DEBUG(0, ("No domains configured on this client!\n"));
+        retval = EINVAL;
+        goto done;
+    }
+
+    retval = EOK;
+
+done:
+    talloc_free(tmp_ctx);
+    return retval;
+}
+
 void nss_task_init(struct task_server *task)
 {
     struct nss_ctx *nctx;
@@ -384,6 +432,12 @@ void nss_task_init(struct task_server *task)
         return;
     }
 
+    ret = nss_init_domains(nctx);
+    if (ret != EOK) {
+        task_server_terminate(task, "fatal error setting up domain map\n");
+        return;
+    }
+
     ret = nss_sbus_init(nctx);
     if (ret != EOK) {
         task_server_terminate(task, "fatal error setting up message bus\n");
diff --git a/server/nss/nsssrv.h b/server/nss/nsssrv.h
index 5e264937..5d49e79f 100644
--- a/server/nss/nsssrv.h
+++ b/server/nss/nsssrv.h
@@ -50,6 +50,7 @@ struct nss_ctx {
     struct confdb_ctx *cdb;
     char *sock_name;
     struct nss_sbus_ctx *ns_ctx;
+    struct btreemap *domain_map;
 };
 
 struct cli_ctx {
diff --git a/server/nss/nsssrv_ldb.c b/server/nss/nsssrv_ldb.c
index 56cdb5d7..418b16e2 100644
--- a/server/nss/nsssrv_ldb.c
+++ b/server/nss/nsssrv_ldb.c
@@ -600,7 +600,7 @@ int nss_ldb_initgroups(TALLOC_CTX *mem_ctx,
     return LDB_SUCCESS;
 }
 
-#define NSS_LDB_CONF_SECTION "config.services.nss"
+#define NSS_LDB_CONF_SECTION "config/services/nss"
 
 static int nss_ldb_read_var(TALLOC_CTX *tmp_ctx,
                             struct confdb_ctx *cdb,
-- 
cgit