sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-02-24 13:57:43 +03:00
Code

librpc:pyrpc: Allow new authenticated rpc connection on the same transport as the basis_connection

Browse Source 
BUG: https://bugzilla.samba.org/show_bug.cgi?id=15680

Signed-off-by: Stefan Metzmacher <metze@samba.org>
Reviewed-by: Andreas Schneider <asn@samba.org>
This commit is contained in:
Stefan Metzmacher 2024-07-17 18:11:49 +02:00 committed by Andreas Schneider
parent a094a29e42
commit 2c171fb1b8
1 changed files with 70 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
source4/librpc/rpc/pyrpc_util.c
View File

@ -181,6 +181,8 @@ PyObject *py_dcerpc_interface_init_helper(PyTypeObject *type, PyObject *args, Py
struct dcerpc_pipe *base_pipe;
PyObject *py_base;
PyTypeObject *ClientConnection_Type;
struct loadparm_context *lp_ctx = NULL;
struct cli_credentials *credentials = NULL;
py_base = PyImport_ImportModule("samba.dcerpc.base");
if (py_base == NULL) {
@ -225,16 +227,76 @@ PyObject *py_dcerpc_interface_init_helper(PyTypeObject *type, PyObject *args, Py
return NULL;
}
status = dcerpc_secondary_context(base_pipe, &ret->pipe, table);
if (!NT_STATUS_IS_OK(status)) {
PyErr_SetNTSTATUS(status);
Py_DECREF(ret);
Py_DECREF(py_base);
Py_DECREF(ClientConnection_Type);
return NULL;
if (py_lp_ctx != Py_None) {
lp_ctx = lpcfg_from_py_object(ret->ev, py_lp_ctx);
if (lp_ctx == NULL) {
PyErr_SetString(PyExc_TypeError, "Expected loadparm context");
Py_DECREF(ret);
return NULL;
}
}
if (py_credentials != Py_None) {
credentials = cli_credentials_from_py_object(py_credentials);
if (credentials == NULL) {
PyErr_SetString(PyExc_TypeError, "Expected credentials");
Py_DECREF(ret);
return NULL;
}
}
if (credentials != NULL) {
struct dcerpc_binding *binding = NULL;
if (lp_ctx == NULL) {
PyErr_SetString(
PyExc_TypeError,
"Expected a loadparm context together "
"with provided credentials");
Py_DECREF(ret);
Py_DECREF(py_base);
Py_DECREF(ClientConnection_Type);
return NULL;
}
status = dcerpc_parse_binding(ret->mem_ctx,
binding_string,
&binding);
if (!NT_STATUS_IS_OK(status)) {
PyErr_SetNTSTATUS(status);
Py_DECREF(ret);
Py_DECREF(py_base);
Py_DECREF(ClientConnection_Type);
return NULL;
}
status = dcerpc_secondary_auth_connection(base_pipe,
binding,
table,
credentials,
lp_ctx,
ret->mem_ctx,
&ret->pipe);
TALLOC_FREE(binding);
if (!NT_STATUS_IS_OK(status)) {
PyErr_SetNTSTATUS(status);
Py_DECREF(ret);
Py_DECREF(py_base);
Py_DECREF(ClientConnection_Type);
return NULL;
}
} else {
status = dcerpc_secondary_context(base_pipe, &ret->pipe, table);
if (!NT_STATUS_IS_OK(status)) {
PyErr_SetNTSTATUS(status);
Py_DECREF(ret);
Py_DECREF(py_base);
Py_DECREF(ClientConnection_Type);
return NULL;
}
ret->pipe = talloc_steal(ret->mem_ctx, ret->pipe);
}
ret->pipe = talloc_steal(ret->mem_ctx, ret->pipe);
Py_XDECREF(ClientConnection_Type);
Py_XDECREF(py_base);
} else {