krb5: add copy_ccache_into_memory()

Reviewed-by: Jakub Hrozek <jhrozek@redhat.com>

4 files changed, 383 insertions, 0 deletions

diff --git a/Makefile.am b/Makefile.am
index b9c25f906..54527223a 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -215,6 +215,7 @@ if HAVE_CMOCKA
         sdap-tests \
         test_sysdb_views \
         test_be_ptask \
+        test_copy_ccache \
         $(NULL)
 
 if BUILD_IFP
@@ -2096,6 +2097,23 @@ test_be_ptask_LDADD = \
     libsss_test_common.la \
     $(NULL)
 
+test_copy_ccache_SOURCES = \
+    src/tests/cmocka/test_copy_ccache.c \
+    src/providers/krb5/krb5_ccache.c \
+    src/util/sss_krb5.c \
+    $(NULL)
+test_copy_ccache_CFLAGS = \
+    $(AM_CFLAGS) \
+    $(NULL)
+test_copy_ccache_LDADD = \
+    $(CMOCKA_LIBS) \
+    $(POPT_LIBS) \
+    $(TALLOC_LIBS) \
+    $(KRB5_LIBS) \
+    $(SSSD_INTERNAL_LTLIBS) \
+    libsss_test_common.la \
+    $(NULL)
+
 endif # HAVE_CMOCKA
 
 noinst_PROGRAMS = pam_test_client
diff --git a/src/providers/krb5/krb5_ccache.c b/src/providers/krb5/krb5_ccache.c
index 8da562b50..e24179c96 100644
--- a/src/providers/krb5/krb5_ccache.c
+++ b/src/providers/krb5/krb5_ccache.c
@@ -672,3 +672,113 @@ errno_t safe_remove_old_ccache_file(const char *old_ccache,
 
     return sss_krb5_cc_destroy(old_ccache, uid, gid);
 }
+
+krb5_error_code copy_ccache_into_memory(TALLOC_CTX *mem_ctx, krb5_context kctx,
+                                        const char *ccache_file,
+                                        char **_mem_name)
+{
+    krb5_error_code kerr;
+    krb5_ccache ccache;
+    krb5_ccache mem_ccache = NULL;
+    char *ccache_name = NULL;
+    krb5_principal princ = NULL;
+    char *mem_name = NULL;
+    char *sep;
+
+    kerr = krb5_cc_resolve(kctx, ccache_file, &ccache);
+    if (kerr != 0) {
+        DEBUG(SSSDBG_CRIT_FAILURE, "error resolving ccache [%s].\n",
+                                    ccache_file);
+        return kerr;
+    }
+
+    kerr = krb5_cc_get_full_name(kctx, ccache, &ccache_name);
+    if (kerr != 0) {
+        DEBUG(SSSDBG_CRIT_FAILURE, "Failed to read name for ccache [%s].\n",
+                                    ccache_file);
+        goto done;
+    }
+
+    sep = strchr(ccache_name, ':');
+    if (sep == NULL || sep[1] == '\0') {
+        DEBUG(SSSDBG_CRIT_FAILURE,
+              "Ccache name [%s] does not have delimiter[:] .\n", ccache_name);
+        kerr = KRB5KRB_ERR_GENERIC;
+        goto done;
+    }
+
+    if (strncmp(ccache_name, "MEMORY:", sizeof("MEMORY:") -1) == 0) {
+        DEBUG(SSSDBG_TRACE_FUNC, "Ccache [%s] is already memory ccache.\n",
+                                 ccache_name);
+        *_mem_name = talloc_strdup(mem_ctx, ccache_name);
+        if(*_mem_name == NULL) {
+            DEBUG(SSSDBG_OP_FAILURE, "talloc_strdup failed.\n");
+            kerr = KRB5KRB_ERR_GENERIC;
+            goto done;
+        }
+        kerr = 0;
+        goto done;
+    }
+    if (strncmp(ccache_name, "FILE:", sizeof("FILE:") -1) == 0) {
+        mem_name = talloc_asprintf(mem_ctx, "MEMORY:%s", sep + 1);
+        if (mem_name == NULL) {
+            DEBUG(SSSDBG_OP_FAILURE, "talloc_asprintf failed.\n");
+            kerr = KRB5KRB_ERR_GENERIC;
+            goto done;
+        }
+    } else {
+        DEBUG(SSSDBG_MINOR_FAILURE, "Unexpected ccache type for ccache [%s], " \
+                                    "currently only FILE is supported.\n",
+                                    ccache_name);
+        kerr = KRB5KRB_ERR_GENERIC;
+        goto done;
+    }
+
+    kerr = krb5_cc_resolve(kctx, mem_name, &mem_ccache);
+    if (kerr != 0) {
+        DEBUG(SSSDBG_CRIT_FAILURE, "error resolving ccache [%s].\n", mem_name);
+        goto done;
+    }
+
+    kerr = krb5_cc_get_principal(kctx, ccache, &princ);
+    if (kerr != 0) {
+        DEBUG(SSSDBG_CRIT_FAILURE,
+              "error reading principal from ccache [%s].\n", ccache_name);
+        goto done;
+    }
+
+    kerr = krb5_cc_initialize(kctx, mem_ccache, princ);
+    if (kerr != 0) {
+        DEBUG(SSSDBG_CRIT_FAILURE,
+              "Failed to initialize ccache [%s].\n", mem_name);
+        goto done;
+    }
+
+    kerr = krb5_cc_copy_creds(kctx, ccache, mem_ccache);
+    if (kerr != 0) {
+        DEBUG(SSSDBG_CRIT_FAILURE,
+              "Failed to copy ccache [%s] to [%s].\n", ccache_name, mem_name);
+        goto done;
+    }
+
+    *_mem_name = mem_name;
+    kerr = 0;
+
+done:
+    if (kerr != 0) {
+        talloc_free(mem_name);
+    }
+
+    free(ccache_name);
+    krb5_free_principal(kctx, princ);
+
+    if (krb5_cc_close(kctx, ccache) != 0) {
+        DEBUG(SSSDBG_OP_FAILURE, "krb5_cc_close failed.\n");
+    }
+
+    if (krb5_cc_close(kctx, mem_ccache) != 0) {
+        DEBUG(SSSDBG_OP_FAILURE, "krb5_cc_close failed.\n");
+    }
+
+    return  kerr;
+}
diff --git a/src/providers/krb5/krb5_ccache.h b/src/providers/krb5/krb5_ccache.h
index e47df3665..f3928e644 100644
--- a/src/providers/krb5/krb5_ccache.h
+++ b/src/providers/krb5/krb5_ccache.h
@@ -53,4 +53,21 @@ errno_t safe_remove_old_ccache_file(const char *old_ccache,
                                     const char *new_ccache,
                                     uid_t uid, gid_t gid);
 
+/**
+ * @brief Copy given ccache into a MEMORY ccache
+ *
+ * @param[in] mem_ctx Talloc memory context the new ccache name should be
+ *                    allocated on
+ * @param[in] kctx Kerberos context
+ * @param[in] ccache_file Name of existing ccache
+ * @param[out] _mem_name Name of the new MEMORY ccache
+ *
+ * In contrast to MEMORY keytabs MEMORY ccaches can and must be removed
+ * explicitly with krb5_cc_destroy() from the memory. Just calling
+ * krb5_cc_close() will keep the MEMORY ccache in memory even if there are no
+ * open handles for the given MEMORY ccache.
+ */
+krb5_error_code copy_ccache_into_memory(TALLOC_CTX *mem_ctx, krb5_context kctx,
+                                        const char *ccache_file,
+                                        char **_mem_name);
 #endif /* __KRB5_CCACHE_H__ */
diff --git a/src/tests/cmocka/test_copy_ccache.c b/src/tests/cmocka/test_copy_ccache.c
new file mode 100644
index 000000000..c7a5573b8
--- /dev/null
+++ b/src/tests/cmocka/test_copy_ccache.c
@@ -0,0 +1,238 @@
+/*
+    Authors:
+        Sumit Bose <sbose@redhat.com>
+
+    Copyright (C) 2014 Red Hat
+
+    SSSD tests: Tests ccache utilities
+
+    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 <stdio.h>
+#include <popt.h>
+
+#include "util/sss_krb5.h"
+#include "providers/krb5/krb5_common.h"
+#include "providers/krb5/krb5_ccache.h"
+#include "tests/cmocka/common_mock.h"
+
+#define CCACHE_TEST_CLIENT_PRINC "test/client@TEST.CCACHE"
+#define CCACHE_TEST_SERVER_PRINC "test/server@TEST.CCACHE"
+#define CCACHE_PATH TEST_DIR "/ccache_test.ccache"
+
+struct ccache_test_ctx {
+    krb5_context kctx;
+    const char *ccache_file_name;
+    krb5_principal client_principal;
+    krb5_principal server_principal;
+};
+
+void setup_ccache(void **state)
+{
+    struct ccache_test_ctx *test_ctx;
+    krb5_error_code kerr;
+    krb5_ccache ccache;
+    krb5_creds test_creds;
+    static krb5_address addr;
+    int add=0x12345;
+    krb5_authdata *a;
+
+    static krb5_address *addrs[] = {
+        &addr,
+        NULL,
+    };
+
+    assert_true(leak_check_setup());
+
+
+    test_ctx = talloc_zero(global_talloc_context, struct ccache_test_ctx);
+    assert_non_null(test_ctx);
+
+    kerr = krb5_init_context(&test_ctx->kctx);
+    assert_int_equal(kerr, 0);
+
+    addr.magic = KV5M_ADDRESS;
+    addr.addrtype = ADDRTYPE_INET;
+    addr.length = 4;
+    addr.contents = (krb5_octet *) &add;
+
+    memset(&test_creds, 0, sizeof(test_creds));
+    test_creds.magic = KV5M_CREDS;
+    kerr = krb5_parse_name(test_ctx->kctx, CCACHE_TEST_CLIENT_PRINC,
+                           &test_ctx->client_principal);
+    assert_int_equal(kerr, 0);
+    test_creds.client = test_ctx->client_principal;
+    kerr = krb5_parse_name(test_ctx->kctx, CCACHE_TEST_SERVER_PRINC,
+                           &test_ctx->server_principal);
+    assert_int_equal(kerr, 0);
+    test_creds.server = test_ctx->server_principal;
+
+    test_creds.keyblock.magic = KV5M_KEYBLOCK;
+    test_creds.keyblock.contents = 0;
+    test_creds.keyblock.enctype = 1;
+    test_creds.keyblock.length = 1;
+    test_creds.keyblock.contents = (unsigned char *) discard_const("1");
+    test_creds.times.authtime = 1111;
+    test_creds.times.starttime = 2222;
+    test_creds.times.endtime = 3333;
+    test_creds.times.renew_till = 4444;
+    test_creds.is_skey = 1;
+    test_creds.ticket_flags = 5555;
+    test_creds.addresses = addrs;
+
+    test_creds.ticket.magic = KV5M_DATA;
+    test_creds.ticket.length = sizeof("Ticket");
+    test_creds.ticket.data = discard_const("Ticket");
+
+    test_creds.authdata = malloc (2 * sizeof(krb5_authdata *));
+    assert_non_null(test_creds.authdata);
+
+    a = (krb5_authdata *) malloc(sizeof(krb5_authdata));
+    assert_non_null(a);
+
+    a->magic = KV5M_AUTHDATA;
+    a->ad_type = KRB5_AUTHDATA_IF_RELEVANT;
+    a->contents = (krb5_octet * ) malloc(1);
+    assert_non_null(a->contents);
+    a->contents[0]=5;
+    a->length = 1;
+    test_creds.authdata[0] = a;
+    test_creds.authdata[1] = NULL;
+
+
+    test_ctx->ccache_file_name = "FILE:" CCACHE_PATH;
+
+    kerr = krb5_cc_resolve(test_ctx->kctx, test_ctx->ccache_file_name,
+                               &ccache);
+    assert_int_equal(kerr, 0);
+
+    kerr = krb5_cc_initialize(test_ctx->kctx, ccache, test_creds.client);
+    assert_int_equal(kerr, 0);
+
+    kerr = krb5_cc_store_cred(test_ctx->kctx, ccache, &test_creds);
+    assert_int_equal(kerr, 0);
+
+    kerr = krb5_cc_close(test_ctx->kctx, ccache);
+    assert_int_equal(kerr, 0);
+
+    check_leaks_push(test_ctx);
+    *state = test_ctx;
+
+    krb5_free_authdata(test_ctx->kctx, test_creds.authdata);
+}
+
+void teardown_ccache(void **state)
+{
+    int ret;
+    struct ccache_test_ctx *test_ctx = talloc_get_type(*state,
+                                                        struct ccache_test_ctx);
+    assert_non_null(test_ctx);
+
+    krb5_free_principal(test_ctx->kctx, test_ctx->client_principal);
+    krb5_free_principal(test_ctx->kctx, test_ctx->server_principal);
+    krb5_free_context(test_ctx->kctx);
+
+    ret = unlink(CCACHE_PATH);
+    assert_int_equal(ret, 0);
+
+    assert_true(check_leaks_pop(test_ctx) == true);
+    talloc_free(test_ctx);
+    assert_true(leak_check_teardown());
+}
+
+void test_copy_ccache(void **state)
+{
+    krb5_error_code kerr;
+    char *mem_ccache_name;
+    krb5_ccache ccache;
+    krb5_creds mcreds;
+    krb5_creds creds;
+    krb5_principal mem_principal;
+    struct ccache_test_ctx *test_ctx = talloc_get_type(*state,
+                                                        struct ccache_test_ctx);
+    assert_non_null(test_ctx);
+
+    kerr = copy_ccache_into_memory(test_ctx, test_ctx->kctx,
+                                   test_ctx->ccache_file_name,
+                                   &mem_ccache_name);
+    assert_int_equal(kerr, 0);
+    assert_non_null(mem_ccache_name);
+
+    kerr = krb5_cc_resolve(test_ctx->kctx, mem_ccache_name, &ccache);
+    assert_int_equal(kerr, 0);
+
+    talloc_free(mem_ccache_name);
+
+    kerr = krb5_cc_get_principal(test_ctx->kctx, ccache, &mem_principal);
+    assert_int_equal(kerr, 0);
+    assert_non_null(mem_principal);
+
+    assert_true(krb5_principal_compare(test_ctx->kctx, mem_principal,
+                                       test_ctx->client_principal));
+    krb5_free_principal(test_ctx->kctx, mem_principal);
+
+    memset(&mcreds, 0, sizeof(mcreds));
+    memset(&creds, 0, sizeof(mcreds));
+    mcreds.client = test_ctx->client_principal;
+    mcreds.server = test_ctx->server_principal;
+    kerr = krb5_cc_retrieve_cred(test_ctx->kctx, ccache, 0, &mcreds, &creds);
+    assert_int_equal(kerr, 0);
+    krb5_free_cred_contents(test_ctx->kctx, &creds);
+
+    kerr = krb5_cc_destroy(test_ctx->kctx, ccache);
+    assert_int_equal(kerr, 0);
+}
+
+int main(int argc, const char *argv[])
+{
+    poptContext pc;
+    int opt;
+    int rv;
+    struct poptOption long_options[] = {
+        POPT_AUTOHELP
+        SSSD_DEBUG_OPTS
+        POPT_TABLEEND
+    };
+
+    const UnitTest tests[] = {
+        unit_test_setup_teardown(test_copy_ccache,
+                                 setup_ccache, teardown_ccache),
+    };
+
+    /* Set debug level to invalid value so we can deside if -d 0 was used. */
+    debug_level = SSSDBG_INVALID;
+
+    pc = poptGetContext(argv[0], argc, argv, long_options, 0);
+    while((opt = poptGetNextOpt(pc)) != -1) {
+        switch(opt) {
+        default:
+            fprintf(stderr, "\nInvalid option %s: %s\n\n",
+                    poptBadOption(pc, 0), poptStrerror(opt));
+            poptPrintUsage(pc, stderr, 0);
+            return 1;
+        }
+    }
+    poptFreeContext(pc);
+
+    DEBUG_CLI_INIT(debug_level);
+
+    /* Even though normally the tests should clean up after themselves
+     * they might not after a failed run. Remove the old db to be sure */
+    tests_set_cwd();
+
+    rv = run_tests(tests);
+
+    return rv;
+}