sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-08-04 08:22:08 +03:00
Code Activity

auth:creds: Add python bindings for (get|set)_smb_ipc_signing

Browse Source 
Signed-off-by: Andreas Schneider <asn@samba.org>
Reviewed-by: Stefan Metzmacher <metze@samba.org>
This commit is contained in:
Andreas Schneider
2020-06-03 12:32:46 +02:00
committed by Andreas Schneider
parent 71d65278e1
commit ef12caea07
2 changed files with 62 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
auth/credentials/pycredentials.c
View File

@ -976,6 +976,52 @@ static PyObject *py_creds_set_smb_signing(PyObject *self, PyObject *args)
Py_RETURN_NONE; Py_RETURN_NONE;
} }
static PyObject *py_creds_get_smb_ipc_signing(PyObject *self, PyObject *unused)
{
enum smb_signing_setting signing_state;
struct cli_credentials *creds = NULL;
creds = PyCredentials_AsCliCredentials(self);
if (creds == NULL) {
PyErr_Format(PyExc_TypeError, "Credentials expected");
return NULL;
}
signing_state = cli_credentials_get_smb_ipc_signing(creds);
return PyLong_FromLong(signing_state);
}
static PyObject *py_creds_set_smb_ipc_signing(PyObject *self, PyObject *args)
{
enum smb_signing_setting signing_state;
struct cli_credentials *creds = NULL;
enum credentials_obtained obt = CRED_SPECIFIED;
creds = PyCredentials_AsCliCredentials(self);
if (creds == NULL) {
PyErr_Format(PyExc_TypeError, "Credentials expected");
return NULL;
}
if (!PyArg_ParseTuple(args, "i|i", &signing_state, &obt)) {
return NULL;
}
switch (signing_state) {
case SMB_SIGNING_DEFAULT:
case SMB_SIGNING_OFF:
case SMB_SIGNING_IF_REQUIRED:
case SMB_SIGNING_DESIRED:
case SMB_SIGNING_REQUIRED:
break;
default:
PyErr_Format(PyExc_TypeError, "Invalid signing state value");
return NULL;
}
cli_credentials_set_smb_ipc_signing(creds, signing_state, obt);
Py_RETURN_NONE;
}
static PyMethodDef py_creds_methods[] = { static PyMethodDef py_creds_methods[] = {
{ {
.ml_name = "get_username", .ml_name = "get_username",
@ -1266,6 +1312,16 @@ static PyMethodDef py_creds_methods[] = {
.ml_meth = py_creds_set_smb_signing, .ml_meth = py_creds_set_smb_signing,
.ml_flags = METH_VARARGS, .ml_flags = METH_VARARGS,
}, },
{
.ml_name = "get_smb_ipc_signing",
.ml_meth = py_creds_get_smb_ipc_signing,
.ml_flags = METH_NOARGS,
},
{
.ml_name = "set_smb_ipc_signing",
.ml_meth = py_creds_set_smb_ipc_signing,
.ml_flags = METH_VARARGS,
},
{ .ml_name = NULL } { .ml_name = NULL }
}; };

6
python/samba/tests/credentials.py
View File

@ -454,3 +454,9 @@ class CredentialsTests(samba.tests.TestCaseInTempDir):
self.assertEqual(creds.get_smb_signing(), credentials.SMB_SIGNING_DEFAULT) self.assertEqual(creds.get_smb_signing(), credentials.SMB_SIGNING_DEFAULT)
creds.set_smb_signing(credentials.SMB_SIGNING_REQUIRED) creds.set_smb_signing(credentials.SMB_SIGNING_REQUIRED)
self.assertEqual(creds.get_smb_signing(), credentials.SMB_SIGNING_REQUIRED) self.assertEqual(creds.get_smb_signing(), credentials.SMB_SIGNING_REQUIRED)
def test_smb_ipc_signing(self):
creds = credentials.Credentials()
self.assertEqual(creds.get_smb_ipc_signing(), credentials.SMB_SIGNING_REQUIRED)
creds.set_smb_ipc_signing(credentials.SMB_SIGNING_OFF)
self.assertEqual(creds.get_smb_ipc_signing(), credentials.SMB_SIGNING_OFF)