SUDO Integration - in-memory cache in responder

New sudo responder option: cache_timeout

https://fedorahosted.org/sssd/ticket/1111

5 files changed, 415 insertions, 9 deletions

diff --git a/src/responder/sudo/sudosrv.c b/src/responder/sudo/sudosrv.c
index 841be43e..6b7eae07 100644
--- a/src/responder/sudo/sudosrv.c
+++ b/src/responder/sudo/sudosrv.c
@@ -129,6 +129,27 @@ int sudo_process_init(TALLOC_CTX *mem_ctx,
                             sudo_dp_reconnect_init, iter);
     }
 
+    /* Get responder options */
+
+    /* Get cache_timeout option */
+    ret = confdb_get_int(sudo_ctx->rctx->cdb, sudo_ctx,
+                         CONFDB_SUDO_CONF_ENTRY, CONFDB_SUDO_CACHE_TIMEOUT,
+                         CONFDB_DEFAULT_SUDO_CACHE_TIMEOUT,
+                         &sudo_ctx->cache_timeout);
+    if (ret != EOK) {
+        DEBUG(SSSDBG_FATAL_FAILURE, ("Error reading from confdb (%d) [%s]\n",
+              ret, strerror(ret)));
+        return ret;
+    }
+
+    /* Initialize in-memory cache */
+    ret = sudosrv_cache_init(sudo_ctx, 10, &sudo_ctx->cache);
+    if (ret != EOK) {
+        DEBUG(SSSDBG_FATAL_FAILURE,
+              ("Could not create hash table: [%s]", strerror(ret)));
+        return ret;
+    }
+
     DEBUG(SSSDBG_TRACE_FUNC, ("SUDO Initialization complete\n"));
 
     return EOK;
diff --git a/src/responder/sudo/sudosrv_cache.c b/src/responder/sudo/sudosrv_cache.c
new file mode 100644
index 00000000..a3a13cad
--- /dev/null
+++ b/src/responder/sudo/sudosrv_cache.c
@@ -0,0 +1,299 @@
+/*
+    Authors:
+        Pavel Březina <pbrezina@redhat.com>
+
+    Copyright (C) 2011 Red Hat
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 3 of the License, or
+    (at your option) any later version.
+
+    This program 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 General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <talloc.h>
+#include <dhash.h>
+#include <time.h>
+
+#include "util/util.h"
+#include "confdb/confdb.h"
+#include "db/sysdb.h"
+#include "responder/sudo/sudosrv_private.h"
+
+static void sudosrv_cache_remove(struct tevent_context *ev,
+                                 struct tevent_timer *te,
+                                 struct timeval tv,
+                                 void *pvt);
+
+struct sudo_cache_entry {
+    hash_table_t *table;
+    hash_key_t *key;
+    size_t num_rules;
+    struct sysdb_attrs **rules;
+
+    struct sudo_ctx *sudo_ctx;
+};
+
+errno_t sudosrv_cache_init(TALLOC_CTX *mem_ctx,
+                           unsigned long count,
+                           hash_table_t **table)
+{
+    return sss_hash_create(mem_ctx, count, table);
+}
+
+static errno_t
+sudosrv_cache_reinit(struct sudo_ctx *sudo_ctx)
+{
+    errno_t ret;
+
+    talloc_free(sudo_ctx->cache);
+
+    ret = sudosrv_cache_init(sudo_ctx, 10, &sudo_ctx->cache);
+    if (ret != EOK) {
+        DEBUG(SSSDBG_FATAL_FAILURE,
+                ("Could not re-initialize hash table: [%s]", strerror(ret)));
+    }
+    return ret;
+}
+
+static hash_key_t *sudosrv_cache_create_key(TALLOC_CTX *mem_ctx,
+                                            struct sss_domain_info *domain,
+                                            const char *username)
+{
+    hash_key_t *key = talloc_zero(NULL, hash_key_t);
+    if (key == NULL) {
+        return NULL;
+    }
+
+    key->type = HASH_KEY_STRING;
+    if (username == NULL) {
+        key->str = talloc_strdup(key, domain->name);
+    } else {
+        key->str = talloc_asprintf(key, "%s:%s", domain->name, username);
+    }
+
+    if (key->str == NULL) {
+        talloc_free(key);
+        return NULL;
+    }
+
+    return talloc_steal(mem_ctx, key);
+}
+
+errno_t sudosrv_cache_set_entry(struct tevent_context *ev,
+                                struct sudo_ctx *sudo_ctx,
+                                hash_table_t *table,
+                                struct sss_domain_info *domain,
+                                const char *username,
+                                size_t num_rules,
+                                struct sysdb_attrs **rules,
+                                time_t timeout)
+{
+    struct sudo_cache_entry *cache_entry = NULL;
+    hash_key_t *key = NULL;
+    hash_value_t value;
+    TALLOC_CTX *tmp_ctx = NULL;
+    struct tevent_timer *timer = NULL;
+    struct timeval tv;
+    errno_t ret;
+    int hret;
+
+    tmp_ctx = talloc_new(NULL);
+    if (tmp_ctx == NULL) {
+        goto done;
+    }
+
+    /* create key */
+    key = sudosrv_cache_create_key(tmp_ctx, domain, username);
+    if (key == NULL) {
+        DEBUG(SSSDBG_CRIT_FAILURE, ("Unable to create hash key.\n"));
+        ret = ENOMEM;
+        goto done;
+    }
+
+    /* create value */
+    cache_entry = talloc_zero(tmp_ctx, struct sudo_cache_entry);
+    if (cache_entry == NULL) {
+        DEBUG(SSSDBG_CRIT_FAILURE, ("Unable to create hash value.\n"));
+        ret = ENOMEM;
+        goto done;
+    }
+    cache_entry->table = table;
+    cache_entry->key = key;
+    cache_entry->num_rules = num_rules;
+    cache_entry->rules = rules;
+    cache_entry->sudo_ctx = sudo_ctx;
+
+    value.type = HASH_VALUE_PTR;
+    value.ptr = cache_entry;
+
+    /* insert value */
+    hret = hash_enter(table, key, &value);
+    if (hret != EOK) {
+        DEBUG(SSSDBG_CRIT_FAILURE,
+              ("Unable to add [%s] to SUDO cache", key->str));
+        DEBUG(SSSDBG_TRACE_LIBS,
+              ("Hash error [%d][%s]", hret, hash_error_string(hret)));
+        ret = EIO;
+        goto done;
+    }
+
+    /* Create a timer event to remove the entry from the cache */
+    tv = tevent_timeval_current_ofs(timeout, 0);
+    timer = tevent_add_timer(ev, cache_entry, tv,
+                             sudosrv_cache_remove,
+                             cache_entry);
+    if (timer == NULL) {
+        ret = ENOMEM;
+        goto done;
+    }
+
+    /* everythig is ok, steal the pointers */
+    talloc_steal(cache_entry, key);
+    talloc_steal(cache_entry, rules);
+    talloc_steal(table, cache_entry);
+
+    ret = EOK;
+
+done:
+    talloc_free(tmp_ctx);
+    return ret;
+}
+
+static void free_cache_entry_cb(struct tevent_context *ev,
+                                struct tevent_immediate *imm,
+                                void *pvt)
+{
+    struct sudo_cache_entry *cache_entry =
+        talloc_get_type(pvt, struct sudo_cache_entry);
+    talloc_free(cache_entry);
+}
+
+static void sudosrv_cache_remove(struct tevent_context *ev,
+                                 struct tevent_timer *te,
+                                 struct timeval tv,
+                                 void *pvt)
+{
+    int hret;
+    hash_key_t *key;
+    struct sudo_cache_entry *cache_entry;
+    struct tevent_immediate *imm;
+
+    cache_entry = talloc_get_type(pvt, struct sudo_cache_entry);
+    key = cache_entry->key;
+
+    hret = hash_delete(cache_entry->table, key);
+    if (hret != HASH_SUCCESS && hret != HASH_ERROR_KEY_NOT_FOUND
+        && hret != HASH_ERROR_BAD_KEY_TYPE) {
+        DEBUG(SSSDBG_MINOR_FAILURE,
+              ("Could not clear [%s] from SUDO cache.\n", key->str));
+        DEBUG(SSSDBG_TRACE_LIBS,
+               ("Hash error [%d][%s]", hret, hash_error_string(hret)));
+
+        /* corrupted memory, re-initialize table */
+        sudosrv_cache_reinit(cache_entry->sudo_ctx);
+    } else {
+        DEBUG(SSSDBG_TRACE_INTERNAL,
+              ("[%s] removed from SUDO cache\n", key->str));
+
+        imm = tevent_create_immediate(ev);
+        if (imm == NULL) {
+            DEBUG(SSSDBG_CRIT_FAILURE, ("Out of memory\n"));
+            return;
+        }
+        tevent_schedule_immediate(imm, ev, free_cache_entry_cb, cache_entry);
+    }
+}
+
+static errno_t sudosrv_cache_lookup_internal(hash_table_t *table,
+                                             struct sss_domain_info *domain,
+                                             const char *username,
+                                             size_t *num_rules,
+                                             struct sysdb_attrs ***rules)
+{
+    struct sudo_cache_entry *cache_entry = NULL;
+    hash_key_t *key = NULL;
+    hash_value_t value;
+    TALLOC_CTX *tmp_ctx = NULL;
+    errno_t ret;
+    int hret;
+
+    tmp_ctx = talloc_new(NULL);
+    if (tmp_ctx == NULL) {
+        goto done;
+    }
+
+    /* create key */
+    key = sudosrv_cache_create_key(tmp_ctx, domain, username);
+    if (key == NULL) {
+        DEBUG(SSSDBG_CRIT_FAILURE, ("Unable to create hash key.\n"));
+        ret = ENOMEM;
+        goto done;
+    }
+
+    hret = hash_lookup(table, key, &value);
+    if (hret == HASH_SUCCESS) {
+        /* cache hit */
+        cache_entry = value.ptr;
+        *num_rules = cache_entry->num_rules;
+        *rules = cache_entry->rules;
+        ret = EOK;
+    } else if (hret == HASH_ERROR_KEY_NOT_FOUND) {
+        /* cache miss */
+        ret = ENOENT;
+    } else {
+        /* error */
+        ret = EIO;
+    }
+
+done:
+    talloc_free(tmp_ctx);
+    return ret;
+}
+
+errno_t sudosrv_cache_lookup(hash_table_t *table,
+                             struct sudo_dom_ctx *dctx,
+                             bool check_next,
+                             const char *username,
+                             size_t *num_rules,
+                             struct sysdb_attrs ***rules)
+{
+    struct sss_domain_info *domain = dctx->domain;
+    errno_t ret;
+
+    if (!check_next) {
+        return sudosrv_cache_lookup_internal(table, dctx->domain, username,
+                                             num_rules, rules);
+    }
+
+    while (domain != NULL) {
+        if (domain->fqnames) {
+            domain = domain->next;
+            continue;
+        }
+
+        ret = sudosrv_cache_lookup_internal(table, domain, username,
+                                            num_rules, rules);
+        if (ret == EOK) {
+            /* user is in this domain */
+            dctx->domain = domain;
+            return ret;
+        } else if (ret != ENOENT) {
+            /* error */
+            return ret;
+        }
+
+        /* user is not in this domain cache, check next */
+        domain = domain->next;
+    }
+
+    /* user is not in cache */
+    return ENOENT;
+}
diff --git a/src/responder/sudo/sudosrv_cmd.c b/src/responder/sudo/sudosrv_cmd.c
index 3550e8ba..cef245fe 100644
--- a/src/responder/sudo/sudosrv_cmd.c
+++ b/src/responder/sudo/sudosrv_cmd.c
@@ -151,6 +151,15 @@ static int sudosrv_cmd_get_sudorules(struct cli_ctx *cli_ctx)
     cmd_ctx->cli_ctx = cli_ctx;
     cmd_ctx->type = SSS_DP_SUDO_USER;
 
+    /* get responder ctx */
+    cmd_ctx->sudo_ctx = talloc_get_type(cli_ctx->rctx->pvt_ctx, struct sudo_ctx);
+    if (!cmd_ctx->sudo_ctx) {
+        DEBUG(SSSDBG_FATAL_FAILURE, ("sudo_ctx not set\n"));
+        ret = EFAULT;
+        goto done;
+    }
+
+    /* create domain ctx */
     dctx = talloc_zero(cmd_ctx, struct sudo_dom_ctx);
     if (!dctx) {
         ret = ENOMEM;
@@ -201,8 +210,18 @@ static int sudosrv_cmd_get_sudorules(struct cli_ctx *cli_ctx)
         cmd_ctx->check_next = true;
     }
 
-    /* ok, find it ! */
-    ret = sudosrv_get_sudorules(dctx);
+    /* try to find rules in in-memory cache */
+    ret = sudosrv_cache_lookup(cmd_ctx->sudo_ctx->cache, dctx,
+                               cmd_ctx->check_next, cmd_ctx->username,
+                               &dctx->res_count, &dctx->res);
+    if (ret == EOK) {
+        /* cache hit */
+        DEBUG(SSSDBG_FUNC_DATA, ("Returning rules for [%s@%s] "
+              "from in-memory cache\n", cmd_ctx->username, dctx->domain->name));
+    } else if (ret == ENOENT) {
+        /* cache expired or missed */
+        ret = sudosrv_get_sudorules(dctx);
+    } /* else error */
 
 done:
     return sudosrv_cmd_done(dctx, ret);
@@ -224,6 +243,15 @@ static int sudosrv_cmd_get_defaults(struct cli_ctx *cli_ctx)
     cmd_ctx->username = NULL;
     cmd_ctx->check_next = false;
 
+    /* get responder ctx */
+    cmd_ctx->sudo_ctx = talloc_get_type(cli_ctx->rctx->pvt_ctx, struct sudo_ctx);
+    if (!cmd_ctx->sudo_ctx) {
+        DEBUG(SSSDBG_FATAL_FAILURE, ("sudo_ctx not set\n"));
+        ret = EFAULT;
+        goto done;
+    }
+
+    /* create domain ctx */
     dctx = talloc_zero(cmd_ctx, struct sudo_dom_ctx);
     if (!dctx) {
         ret = ENOMEM;
@@ -246,8 +274,18 @@ static int sudosrv_cmd_get_defaults(struct cli_ctx *cli_ctx)
         goto done;
     }
 
-    /* ok, find it ! */
-    ret = sudosrv_get_rules(dctx);
+    /* try to find rules in in-memory cache */
+    ret = sudosrv_cache_lookup(cmd_ctx->sudo_ctx->cache, dctx,
+                               cmd_ctx->check_next, cmd_ctx->username,
+                               &dctx->res_count, &dctx->res);
+    if (ret == EOK) {
+        /* cache hit */
+        DEBUG(SSSDBG_FUNC_DATA, ("Returning defaults settings for [%s] "
+                                 "from in-memory cache\n", dctx->domain->name));
+    } else if (ret == ENOENT) {
+        /* cache expired or missed */
+        ret = sudosrv_get_rules(dctx);
+    } /* else error */
 
 done:
     return sudosrv_cmd_done(dctx, ret);
diff --git a/src/responder/sudo/sudosrv_get_sudorules.c b/src/responder/sudo/sudosrv_get_sudorules.c
index 8cb10e9a..c53638a3 100644
--- a/src/responder/sudo/sudosrv_get_sudorules.c
+++ b/src/responder/sudo/sudosrv_get_sudorules.c
@@ -210,7 +210,6 @@ static void sudosrv_check_user_dp_callback(uint16_t err_maj, uint32_t err_min,
           ("Data Provider returned, check the cache again\n"));
     dctx->check_provider = false;
     ret = sudosrv_get_user(dctx);
-    /* FIXME - set entry into cache so that we don't perform initgroups too often */
     if (ret == EAGAIN) {
         goto done;
     } else if (ret != EOK) {
@@ -250,8 +249,9 @@ errno_t sudosrv_get_rules(struct sudo_dom_ctx *dctx)
     struct sudo_cmd_ctx *cmd_ctx = dctx->cmd_ctx;
     struct dp_callback_ctx *cb_ctx = NULL;
 
-    /* FIXME - cache logic will be here. For now, just refresh
-     * the cache unconditionally */
+    DEBUG(SSSDBG_TRACE_FUNC, ("getting rules for %s\n",
+          cmd_ctx->username ? cmd_ctx->username : "default options"));
+
     dpreq = sss_dp_get_sudoers_send(cmd_ctx->cli_ctx,
                                     cmd_ctx->cli_ctx->rctx,
                                     dctx->domain, false,
@@ -319,7 +319,6 @@ sudosrv_get_sudorules_dp_callback(uint16_t err_maj, uint32_t err_min,
                "Will try to return what we have in cache\n",
                (unsigned int)err_maj, (unsigned int)err_min, err_msg));
 
-        /* FIXME - cache or next domain? */
         /* Loop to the next domain if possible */
         if (dctx->domain->next && dctx->cmd_ctx->check_next) {
             dctx->domain = dctx->domain->next;
@@ -355,8 +354,11 @@ static errno_t sudosrv_get_sudorules_from_cache(struct sudo_dom_ctx *dctx)
     errno_t ret;
     struct sysdb_ctx *sysdb;
     struct cli_ctx *cli_ctx = dctx->cmd_ctx->cli_ctx;
+    struct sudo_ctx *sudo_ctx = dctx->cmd_ctx->sudo_ctx;
     uid_t uid;
     char **groupnames;
+    const char *safe_name = dctx->cmd_ctx->username ?
+                            dctx->cmd_ctx->username : "default rules";
 
     tmp_ctx = talloc_new(NULL);
     if (tmp_ctx == NULL) return ENOMEM;
@@ -393,8 +395,21 @@ static errno_t sudosrv_get_sudorules_from_cache(struct sudo_dom_ctx *dctx)
         goto done;
     }
 
+    /* Store result in in-memory cache */
+    ret = sudosrv_cache_set_entry(sudo_ctx->rctx->ev, sudo_ctx,
+                                  sudo_ctx->cache, dctx->domain,
+                                  dctx->cmd_ctx->username, dctx->res_count,
+                                  dctx->res, sudo_ctx->cache_timeout);
+    if (ret != EOK) {
+        DEBUG(SSSDBG_MINOR_FAILURE, ("Unable to store rules in cache for "
+              "[%s@%s]\n", safe_name, dctx->domain->name));
+    } else {
+        DEBUG(SSSDBG_FUNC_DATA, ("Rules for [%s@%s] stored in in-memory cache\n",
+                                 safe_name, dctx->domain->name));
+    }
+
     DEBUG(SSSDBG_TRACE_FUNC, ("Returning rules for [%s@%s]\n",
-          dctx->cmd_ctx->username, dctx->domain->name));
+          safe_name, dctx->domain->name));
 
     ret = EOK;
 done:
diff --git a/src/responder/sudo/sudosrv_private.h b/src/responder/sudo/sudosrv_private.h
index b59aca4a..c3feb19b 100644
--- a/src/responder/sudo/sudosrv_private.h
+++ b/src/responder/sudo/sudosrv_private.h
@@ -38,10 +38,23 @@ enum sss_dp_sudo_type {
 
 struct sudo_ctx {
     struct resp_ctx *rctx;
+
+    /*
+     * options
+     */
+    int cache_timeout;
+
+    /*
+     * Key: domain          for SSS_DP_SUDO_DEFAULTS
+     *      domain:username for SSS_DP_SUDO_USER
+     * Val: struct sudo_cache_entry *
+     */
+    hash_table_t *cache;
 };
 
 struct sudo_cmd_ctx {
     struct cli_ctx *cli_ctx;
+    struct sudo_ctx *sudo_ctx;
     enum sss_dp_sudo_type type;
     char *username;
     bool check_next;
@@ -121,4 +134,24 @@ sss_dp_get_sudoers_recv(TALLOC_CTX *mem_ctx,
                         dbus_uint32_t *err_min,
                         char **err_msg);
 
+errno_t sudosrv_cache_init(TALLOC_CTX *mem_ctx,
+                           unsigned long count,
+                           hash_table_t **table);
+
+errno_t sudosrv_cache_lookup(hash_table_t *table,
+                             struct sudo_dom_ctx *dctx,
+                             bool check_next,
+                             const char *username,
+                             size_t *res_count,
+                             struct sysdb_attrs ***res);
+
+errno_t sudosrv_cache_set_entry(struct tevent_context *ev,
+                                struct sudo_ctx *sudo_ctx,
+                                hash_table_t *table,
+                                struct sss_domain_info *domain,
+                                const char *username,
+                                size_t res_count,
+                                struct sysdb_attrs **res,
+                                time_t timeout);
+
 #endif /* _SUDOSRV_PRIVATE_H_ */