pycredentials: Add set_named_ccache()

Signed-off-by: Andrew Bartlett <abartlet@samba.org>
Reviewed-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>

diff --git a/auth/credentials/pycredentials.c b/auth/credentials/pycredentials.c
index a55529141b58b6f0f2459b458f9b2f04618efb17..638ae8de2ed16053b868855ca53acb87b1a337fd 100644 (file)
--- a/auth/credentials/pycredentials.c
+++ b/auth/credentials/pycredentials.c
@@ -571,6 +571,48 @@ static PyObject *py_creds_get_named_ccache(PyObject *self, PyObject *args)
        return NULL;
 }
 
+static PyObject *py_creds_set_named_ccache(PyObject *self, PyObject *args)
+{
+       struct loadparm_context *lp_ctx = NULL;
+       enum credentials_obtained obt = CRED_SPECIFIED;
+       const char *error_string = NULL;
+       TALLOC_CTX *mem_ctx = NULL;
+       char *newval = NULL;
+       PyObject *py_lp_ctx = Py_None;
+       int _obt = obt;
+       int ret;
+
+       if (!PyArg_ParseTuple(args, "s|iO", &newval, &_obt, &py_lp_ctx))
+               return NULL;
+
+       mem_ctx = talloc_new(NULL);
+       if (mem_ctx == NULL) {
+               PyErr_NoMemory();
+               return NULL;
+       }
+
+       lp_ctx = lpcfg_from_py_object(mem_ctx, py_lp_ctx);
+       if (lp_ctx == NULL) {
+               talloc_free(mem_ctx);
+               return NULL;
+       }
+
+       ret = cli_credentials_set_ccache(PyCredentials_AsCliCredentials(self),
+                                        lp_ctx,
+                                        newval, CRED_SPECIFIED,
+                                        &error_string);
+
+       if (ret != 0) {
+               PyErr_SetString(PyExc_RuntimeError,
+                               error_string != NULL ? error_string : "NULL");
+               talloc_free(mem_ctx);
+               return NULL;
+       }
+
+       talloc_free(mem_ctx);
+       Py_RETURN_NONE;
+}
+
 static PyObject *py_creds_set_gensec_features(PyObject *self, PyObject *args)
 {
        unsigned int gensec_features;
@@ -756,6 +798,9 @@ static PyMethodDef py_creds_methods[] = {
        { "guess", py_creds_guess, METH_VARARGS, NULL },
        { "set_machine_account", py_creds_set_machine_account, METH_VARARGS, NULL },
        { "get_named_ccache", py_creds_get_named_ccache, METH_VARARGS, NULL },
+       { "set_named_ccache", py_creds_set_named_ccache, METH_VARARGS,
+               "S.set_named_ccache(krb5_ccache_name, obtained, lp) -> None\n"
+               "Set credentials to KRB5 Credentials Cache (by name)." },
        { "set_gensec_features", py_creds_set_gensec_features, METH_VARARGS, NULL },
        { "get_gensec_features", py_creds_get_gensec_features, METH_NOARGS, NULL },
        { "get_forced_sasl_mech", py_creds_get_forced_sasl_mech, METH_NOARGS,

diff --git a/python/samba/tests/krb5_credentials.py b/python/samba/tests/krb5_credentials.py
index a3eb034e2c7ef856ff8f1df15d6a0e1cf15001a6..cad19da249dbc81650d8ce878ea13a4b24ccb245 100644 (file)
--- a/python/samba/tests/krb5_credentials.py
+++ b/python/samba/tests/krb5_credentials.py
@@ -74,6 +74,15 @@ class PyKrb5CredentialsTests(TestCase):
         ccache = self.machine_creds.get_named_ccache(self.lp)
         self.assertIsNotNone(ccache.get_name())
 
+    def test_set_named_ccache(self):
+        ccache = self.machine_creds.get_named_ccache(self.lp)
+
+        creds = Credentials()
+        creds.set_named_ccache(ccache.get_name())
+
+        ccache2 = creds.get_named_ccache(self.lp)
+        self.assertEqual(ccache.get_name(), ccache2.get_name())
+
     #
     # Create the machine account
     def create_machine_account(self):