Added SRVSVC pipe support. I've started implementing NetServerGetInfo()

diff --git a/source/python/py_srvsvc.c b/source/python/py_srvsvc.c
new file mode 100644 (file)
index 0000000..40827cc
--- /dev/null
+++ b/source/python/py_srvsvc.c
@@ -0,0 +1,178 @@
+/* 
+   Python wrappers for DCERPC/SMB client routines.
+
+   Copyright (C) Tim Potter, 2003
+   
+   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 2 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, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#include "python/py_srvsvc.h"
+
+/* Exceptions this module can raise */
+
+PyObject *srvsvc_error, *srvsvc_werror;
+
+static struct const_vals {
+       char *name;
+       uint32 value;
+} module_const_vals[] = {
+       { NULL },
+};
+
+static void const_init(PyObject *dict)
+{
+       struct const_vals *tmp;
+       PyObject *obj;
+
+       for (tmp = module_const_vals; tmp->name; tmp++) {
+               obj = PyInt_FromLong(tmp->value);
+               PyDict_SetItemString(dict, tmp->name, obj);
+               Py_DECREF(obj);
+       }
+}
+
+/* NetServerGetInfo */
+
+PyObject *srvsvc_netservergetinfo(PyObject *self, PyObject *args,
+                                 PyObject *kw)
+{
+       static char *kwlist[] = { "server", "level", "creds", NULL };
+       char *unc_name, *server, *errstr;
+       PyObject *creds = NULL, *result = NULL;
+       struct cli_state *cli;
+       TALLOC_CTX *mem_ctx = NULL;
+       uint32 level;
+       SRV_INFO_CTR ctr;
+       WERROR status;
+
+       if (!PyArg_ParseTupleAndKeywords(
+                   args, kw, "si|O", kwlist, &unc_name, &level, &creds))
+               return NULL;
+
+       if (unc_name[0] != '\\' || unc_name[1] != '\\') {
+               PyErr_SetString(PyExc_ValueError, "UNC name required");
+               return NULL;
+       }
+
+       server = strdup(unc_name + 2);
+
+       if (strchr(server, '\\')) {
+               char *c = strchr(server, '\\');
+               *c = 0;
+       }
+
+       if (creds && creds != Py_None && !PyDict_Check(creds)) {
+               PyErr_SetString(PyExc_TypeError, 
+                               "credentials must be dictionary or None");
+               return NULL;
+       }
+
+       if (!(cli = open_pipe_creds(server, creds, PI_SRVSVC, &errstr))) {
+               PyErr_SetString(srvsvc_error, errstr);
+               free(errstr);
+               goto done;
+       }
+
+       if (!(mem_ctx = talloc_init("srvsvc_netservergetinfo"))) {
+               PyErr_SetString(srvsvc_error, 
+                               "unable to init talloc context\n");
+               goto done;
+       }
+
+       ZERO_STRUCT(ctr);
+
+       status = cli_srvsvc_net_srv_get_info(cli, mem_ctx, level, &ctr);
+
+       if (!NT_STATUS_IS_OK(status)) {
+               PyErr_SetObject(srvsvc_error, py_werror_tuple(status));
+               goto done;
+       }
+
+       result = Py_None;
+       Py_INCREF(Py_None);
+
+done:
+       if (mem_ctx)
+               talloc_destroy(mem_ctx);
+
+       return result;
+}
+
+/*
+ * Module initialisation 
+ */
+
+static PyMethodDef srvsvc_methods[] = {
+       { "netservergetinfo", (PyCFunction)srvsvc_netservergetinfo,
+         METH_VARARGS | METH_KEYWORDS,
+         "Retrieve information about a particular server." },
+
+       { "setup_logging", (PyCFunction)py_setup_logging, 
+         METH_VARARGS | METH_KEYWORDS, 
+         "Set up debug logging.
+
+Initialises Samba's debug logging system.  One argument is expected which
+is a boolean specifying whether debugging is interactive and sent to stdout
+or logged to a file.
+
+Example:
+
+>>> srvsvc.setup_logging(interactive = 1)" },
+
+       { "get_debuglevel", (PyCFunction)get_debuglevel, 
+         METH_VARARGS, 
+         "Set the current debug level.
+
+Example:
+
+>>> srvsvc.get_debuglevel()
+0" },
+
+       { "set_debuglevel", (PyCFunction)set_debuglevel, 
+         METH_VARARGS, 
+         "Get the current debug level.
+
+Example:
+
+>>> srvsvc.set_debuglevel(10)" },
+
+       { NULL }
+};
+
+void initsrvsvc(void)
+{
+       PyObject *module, *dict;
+
+       /* Initialise module */
+
+       module = Py_InitModule("srvsvc", srvsvc_methods);
+       dict = PyModule_GetDict(module);
+
+       /* Exceptions we can raise */
+
+       srvsvc_error = PyErr_NewException("srvsvc.error", NULL, NULL);
+       PyDict_SetItemString(dict, "error", srvsvc_error);
+
+       srvsvc_werror = PyErr_NewException("srvsvc.werror", NULL, NULL);
+       PyDict_SetItemString(dict, "werror", srvsvc_werror);
+
+       /* Initialise constants */
+
+       const_init(dict);
+
+       /* Do samba initialisation */
+
+       py_samba_init();
+}

diff --git a/source/python/py_srvsvc.h b/source/python/py_srvsvc.h
new file mode 100644 (file)
index 0000000..17b8eeb
--- /dev/null
+++ b/source/python/py_srvsvc.h
@@ -0,0 +1,29 @@
+/* 
+   Python wrappers for DCERPC/SMB client routines.
+
+   Copyright (C) Tim Potter, 2003
+   
+   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 2 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, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#ifndef _PY_SRVSVC_H
+#define _PY_SRVSVC_H
+
+#include "includes.h"
+#include "Python.h"
+
+#include "python/py_common_proto.h"
+
+#endif /* _PY_SRVSVC_H */

diff --git a/source/python/setup.py b/source/python/setup.py
index b15f906bc2f8b0f6d0b4a65adecccef27e8ee866..7d5df71b1040b113c4de8166817484e1cc6ba5c6 100755 (executable)
--- a/source/python/setup.py
+++ b/source/python/setup.py
@@ -144,6 +144,16 @@ setup(
               extra_compile_args = flags_list,
               extra_objects = obj_list),
 
+    # SRVSVC pipe module
+
+    Extension(name = "srvsvc",
+              sources = [samba_srcdir + "python/py_srvsvc.c",
+                         samba_srcdir + "python/py_common.c"],
+              libraries = lib_list,
+              library_dirs = ["/usr/kerberos/lib"],
+              extra_compile_args = flags_list,
+              extra_objects = obj_list),
+
     # tdb module
 
     Extension(name = "tdb",