2 Python wrappers for DCERPC/SMB client routines.
4 Copyright (C) Tim Potter, 2002
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software
18 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 #include "python/py_lsa.h"
23 PyObject
*new_lsa_policy_hnd_object(struct cli_state
*cli
, TALLOC_CTX
*mem_ctx
,
26 lsa_policy_hnd_object
*o
;
28 o
= PyObject_New(lsa_policy_hnd_object
, &lsa_policy_hnd_type
);
32 memcpy(&o
->pol
, pol
, sizeof(POLICY_HND
));
38 * Exceptions raised by this module
41 PyObject
*lsa_error
; /* This indicates a non-RPC related error
42 such as name lookup failure */
44 PyObject
*lsa_ntstatus
; /* This exception is raised when a RPC call
45 returns a status code other than
49 * Open/close lsa handles
52 static PyObject
*lsa_open_policy(PyObject
*self
, PyObject
*args
,
55 static char *kwlist
[] = { "servername", "creds", "access", NULL
};
56 char *server
, *errstr
;
57 PyObject
*creds
= NULL
, *result
= NULL
;
58 uint32 desired_access
= MAXIMUM_ALLOWED_ACCESS
;
59 struct cli_state
*cli
= NULL
;
61 TALLOC_CTX
*mem_ctx
= NULL
;
64 if (!PyArg_ParseTupleAndKeywords(
65 args
, kw
, "s|Oi", kwlist
, &server
, &creds
, &desired_access
))
68 if (creds
&& creds
!= Py_None
&& !PyDict_Check(creds
)) {
69 PyErr_SetString(PyExc_TypeError
,
70 "credentials must be dictionary or None");
74 if (server
[0] != '\\' || server
[1] != '\\') {
75 PyErr_SetString(PyExc_ValueError
, "UNC name required");
81 if (!(cli
= open_pipe_creds(server
, creds
, PIPE_LSARPC
, &errstr
))) {
82 PyErr_SetString(lsa_error
, errstr
);
87 if (!(mem_ctx
= talloc_init())) {
88 PyErr_SetString(lsa_error
, "unable to init talloc context\n");
92 ntstatus
= cli_lsa_open_policy(cli
, mem_ctx
, True
,
93 SEC_RIGHTS_MAXIMUM_ALLOWED
, &hnd
);
95 if (!NT_STATUS_IS_OK(ntstatus
)) {
96 PyErr_SetObject(lsa_ntstatus
, py_ntstatus_tuple(ntstatus
));
100 result
= new_lsa_policy_hnd_object(cli
, mem_ctx
, &hnd
);
108 talloc_destroy(mem_ctx
);
114 static PyObject
*lsa_close(PyObject
*self
, PyObject
*args
, PyObject
*kw
)
117 lsa_policy_hnd_object
*hnd
;
120 /* Parse parameters */
122 if (!PyArg_ParseTuple(args
, "O!", &lsa_policy_hnd_type
, &po
))
125 hnd
= (lsa_policy_hnd_object
*)po
;
127 /* Call rpc function */
129 result
= cli_lsa_close(hnd
->cli
, hnd
->mem_ctx
, &hnd
->pol
);
131 /* Cleanup samba stuff */
133 cli_shutdown(hnd
->cli
);
134 talloc_destroy(hnd
->mem_ctx
);
142 static PyObject
*lsa_lookup_names(PyObject
*self
, PyObject
*args
)
144 PyObject
*py_names
, *result
;
146 lsa_policy_hnd_object
*hnd
= (lsa_policy_hnd_object
*)self
;
152 if (!PyArg_ParseTuple(args
, "O", &py_names
))
155 if (!PyList_Check(py_names
) && !PyString_Check(py_names
)) {
156 PyErr_SetString(PyExc_TypeError
, "must be list or string");
160 if (PyList_Check(py_names
)) {
162 /* Convert list to char ** array */
164 num_names
= PyList_Size(py_names
);
165 names
= (const char **)talloc(
166 hnd
->mem_ctx
, num_names
* sizeof(char *));
168 for (i
= 0; i
< num_names
; i
++) {
169 PyObject
*obj
= PyList_GetItem(py_names
, i
);
171 names
[i
] = talloc_strdup(hnd
->mem_ctx
, PyString_AsString(obj
));
176 /* Just a single element */
179 names
= (const char **)talloc(hnd
->mem_ctx
, sizeof(char *));
181 names
[0] = PyString_AsString(py_names
);
184 ntstatus
= cli_lsa_lookup_names(hnd
->cli
, hnd
->mem_ctx
, &hnd
->pol
,
185 num_names
, names
, &sids
, &name_types
);
187 if (!NT_STATUS_IS_OK(ntstatus
) && NT_STATUS_V(ntstatus
) != 0x107) {
188 PyErr_SetObject(lsa_ntstatus
, py_ntstatus_tuple(ntstatus
));
192 result
= PyList_New(num_names
);
194 for (i
= 0; i
< num_names
; i
++) {
195 PyObject
*sid_obj
, *obj
;
197 py_from_SID(&sid_obj
, &sids
[i
]);
199 obj
= Py_BuildValue("(Oi)", sid_obj
, name_types
[i
]);
201 PyList_SetItem(result
, i
, obj
);
207 static PyObject
*lsa_lookup_sids(PyObject
*self
, PyObject
*args
,
210 PyObject
*py_sids
, *result
;
213 char **domains
, **names
;
215 lsa_policy_hnd_object
*hnd
= (lsa_policy_hnd_object
*)self
;
218 if (!PyArg_ParseTuple(args
, "O", &py_sids
))
221 if (!PyList_Check(py_sids
) && !PyString_Check(py_sids
)) {
222 PyErr_SetString(PyExc_TypeError
, "must be list or string");
226 if (PyList_Check(py_sids
)) {
228 /* Convert dictionary to char ** array */
230 num_sids
= PyList_Size(py_sids
);
231 sids
= (DOM_SID
*)talloc(hnd
->mem_ctx
, num_sids
* sizeof(DOM_SID
));
233 memset(sids
, 0, num_sids
* sizeof(DOM_SID
));
235 for (i
= 0; i
< num_sids
; i
++) {
236 PyObject
*obj
= PyList_GetItem(py_sids
, i
);
238 string_to_sid(&sids
[i
], PyString_AsString(obj
));
243 /* Just a single element */
246 sids
= (DOM_SID
*)talloc(hnd
->mem_ctx
, sizeof(DOM_SID
));
248 string_to_sid(&sids
[0], PyString_AsString(py_sids
));
251 ntstatus
= cli_lsa_lookup_sids(hnd
->cli
, hnd
->mem_ctx
, &hnd
->pol
,
252 num_sids
, sids
, &domains
, &names
,
255 if (!NT_STATUS_IS_OK(ntstatus
)) {
256 PyErr_SetObject(lsa_ntstatus
, py_ntstatus_tuple(ntstatus
));
260 result
= PyList_New(num_sids
);
262 for (i
= 0; i
< num_sids
; i
++) {
265 obj
= Py_BuildValue("{sssssi}", "username", names
[i
],
266 "domain", domains
[i
], "name_type",
269 PyList_SetItem(result
, i
, obj
);
275 static PyObject
*lsa_enum_trust_dom(PyObject
*self
, PyObject
*args
)
277 lsa_policy_hnd_object
*hnd
= (lsa_policy_hnd_object
*)self
;
279 uint32 enum_ctx
= 0, num_domains
, i
, pref_num_domains
= 0;
281 DOM_SID
*domain_sids
;
284 if (!PyArg_ParseTuple(args
, ""))
287 ntstatus
= cli_lsa_enum_trust_dom(
288 hnd
->cli
, hnd
->mem_ctx
, &hnd
->pol
, &enum_ctx
,
289 &pref_num_domains
, &num_domains
, &domain_names
, &domain_sids
);
291 if (!NT_STATUS_IS_OK(ntstatus
)) {
292 PyErr_SetObject(lsa_ntstatus
, py_ntstatus_tuple(ntstatus
));
296 result
= PyList_New(num_domains
);
298 for (i
= 0; i
< num_domains
; i
++) {
301 sid_to_string(sid_str
, &domain_sids
[i
]);
304 Py_BuildValue("(ss)", domain_names
[i
], sid_str
));
311 * Method dispatch tables
314 static PyMethodDef lsa_hnd_methods
[] = {
318 { "lookup_sids", (PyCFunction
)lsa_lookup_sids
,
319 METH_VARARGS
| METH_KEYWORDS
,
320 "Convert sids to names." },
322 { "lookup_names", (PyCFunction
)lsa_lookup_names
,
323 METH_VARARGS
| METH_KEYWORDS
,
324 "Convert names to sids." },
326 /* Trusted domains */
328 { "enum_trusted_domains", (PyCFunction
)lsa_enum_trust_dom
,
330 "Enumerate trusted domains." },
335 static void py_lsa_policy_hnd_dealloc(PyObject
* self
)
340 static PyObject
*py_lsa_policy_hnd_getattr(PyObject
*self
, char *attrname
)
342 return Py_FindMethod(lsa_hnd_methods
, self
, attrname
);
345 PyTypeObject lsa_policy_hnd_type
= {
346 PyObject_HEAD_INIT(NULL
)
349 sizeof(lsa_policy_hnd_object
),
351 py_lsa_policy_hnd_dealloc
, /*tp_dealloc*/
353 py_lsa_policy_hnd_getattr
, /*tp_getattr*/
358 0, /*tp_as_sequence*/
363 static PyMethodDef lsa_methods
[] = {
365 /* Open/close lsa handles */
367 { "open_policy", (PyCFunction
)lsa_open_policy
,
368 METH_VARARGS
| METH_KEYWORDS
,
369 "Open a policy handle" },
371 { "close", (PyCFunction
)lsa_close
,
373 "Close a policy handle" },
375 /* Other stuff - this should really go into a samba config module
376 but for the moment let's leave it here. */
378 { "setup_logging", (PyCFunction
)py_setup_logging
,
379 METH_VARARGS
| METH_KEYWORDS
,
380 "Set up debug logging.
382 Initialises Samba's debug logging system. One argument is expected which
383 is a boolean specifying whether debugging is interactive and sent to stdout
388 >>> spoolss.setup_logging(interactive = 1)" },
390 { "get_debuglevel", (PyCFunction
)get_debuglevel
,
392 "Set the current debug level.
396 >>> spoolss.get_debuglevel()
399 { "set_debuglevel", (PyCFunction
)set_debuglevel
,
401 "Get the current debug level.
405 >>> spoolss.set_debuglevel(10)" },
410 static struct const_vals
{
413 } module_const_vals
[] = {
417 static void const_init(PyObject
*dict
)
419 struct const_vals
*tmp
;
422 for (tmp
= module_const_vals
; tmp
->name
; tmp
++) {
423 obj
= PyInt_FromLong(tmp
->value
);
424 PyDict_SetItemString(dict
, tmp
->name
, obj
);
430 * Module initialisation
435 PyObject
*module
, *dict
;
437 /* Initialise module */
439 module
= Py_InitModule("lsa", lsa_methods
);
440 dict
= PyModule_GetDict(module
);
442 lsa_error
= PyErr_NewException("lsa.error", NULL
, NULL
);
443 PyDict_SetItemString(dict
, "error", lsa_error
);
445 lsa_ntstatus
= PyErr_NewException("lsa.ntstatus", NULL
, NULL
);
446 PyDict_SetItemString(dict
, "ntstatus", lsa_ntstatus
);
448 /* Initialise policy handle object */
450 lsa_policy_hnd_type
.ob_type
= &PyType_Type
;
452 /* Initialise constants */
456 /* Do samba initialisation */
460 setup_logging("lsa", True
);