source/python/py_common.c

   1 /*
   2    Python wrappers for DCERPC/SMB client routines.
   3
   4    Copyright (C) Tim Potter, 2002
   5
   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.
  10
  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.
  15
  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.
  19 */
  20
  21 #include "includes.h"
  22 #include "Python.h"
  23
  24 #include "python/py_common_proto.h"
  25
  26 /* Return a tuple of (error code, error string) from a WERROR */
  27
  28 PyObject *py_werror_tuple(WERROR werror)
  29 {
  30         return Py_BuildValue("[is]", W_ERROR_V(werror),
  31                              dos_errstr(werror));
  32 }
  33
  34 /* Return a tuple of (error code, error string) from a WERROR */
  35
  36 PyObject *py_ntstatus_tuple(NTSTATUS ntstatus)
  37 {
  38         return Py_BuildValue("[is]", NT_STATUS_V(ntstatus),
  39                              nt_errstr(ntstatus));
  40 }
  41
  42 /* Initialise samba client routines */
  43
  44 static BOOL initialised;
  45
  46 void py_samba_init(void)
  47 {
  48         extern pstring global_myname;
  49         char *p;
  50
  51         if (initialised)
  52                 return;
  53
  54         /* Load configuration file */
  55
  56         if (!lp_load(dyn_CONFIGFILE, True, False, False))
  57                 fprintf(stderr, "Can't load %s\n", dyn_CONFIGFILE);
  58
  59         /* Misc other stuff */
  60
  61         load_interfaces();
  62
  63         fstrcpy(global_myname, myhostname());
  64         p = strchr(global_myname, '.');
  65         if (p)
  66                 *p = 0;
  67
  68         initialised = True;
  69 }
  70
  71 /* Debuglevel routines */
  72
  73 PyObject *get_debuglevel(PyObject *self, PyObject *args)
  74 {
  75         PyObject *debuglevel;
  76
  77         if (!PyArg_ParseTuple(args, ""))
  78                 return NULL;
  79
  80         debuglevel = PyInt_FromLong(DEBUGLEVEL);
  81
  82         return debuglevel;
  83 }
  84
  85 PyObject *set_debuglevel(PyObject *self, PyObject *args)
  86 {
  87         int debuglevel;
  88
  89         if (!PyArg_ParseTuple(args, "i", &debuglevel))
  90                 return NULL;
  91
  92         DEBUGLEVEL = debuglevel;
  93
  94         Py_INCREF(Py_None);
  95         return Py_None;
  96 }
  97
  98 /* Initialise logging */
  99
 100 PyObject *py_setup_logging(PyObject *self, PyObject *args, PyObject *kw)
 101 {
 102         BOOL interactive = False;
 103         char *logfilename = NULL;
 104         static char *kwlist[] = {"interactive", "logfilename", NULL};
 105
 106         if (!PyArg_ParseTupleAndKeywords(
 107                     args, kw, "|is", kwlist, &interactive, &logfilename))
 108                 return NULL;
 109
 110         if (interactive && logfilename) {
 111                 PyErr_SetString(PyExc_RuntimeError,
 112                                 "can't be interactive and set log file name");
 113                 return NULL;
 114         }
 115
 116         if (interactive)
 117                 setup_logging("spoolss", True);
 118
 119         if (logfilename) {
 120                 lp_set_logfile(logfilename);
 121                 setup_logging(logfilename, False);
 122                 reopen_logs();
 123         }
 124
 125         Py_INCREF(Py_None);
 126         return Py_None;
 127 }
 128
 129 /* Return a cli_state to a RPC pipe on the given server.  Use the
 130    credentials passed if not NULL.  If an error occurs errstr is set to a
 131    string describing the error and NULL is returned.  If set, errstr must
 132    be freed by calling free(). */
 133
 134 struct cli_state *open_pipe_creds(char *server, PyObject *creds,
 135                                   char *pipe_name, char **errstr)
 136 {
 137         char *username = "", *password = "", *domain = "";
 138         struct cli_state *cli;
 139         NTSTATUS result;
 140         struct in_addr server_ip;
 141         extern pstring global_myname;
 142
 143         /* Extract credentials from the python dictionary */
 144
 145         if (creds && PyDict_Size(creds) > 0) {
 146                 PyObject *username_obj, *password_obj, *domain_obj;
 147
 148                 /* Check credentials passed are valid.  This means the
 149                    username, domain and password keys must exist and be
 150                    string objects. */
 151
 152                 username_obj = PyDict_GetItemString(creds, "username");
 153                 domain_obj = PyDict_GetItemString(creds, "domain");
 154                 password_obj = PyDict_GetItemString(creds, "password");
 155
 156                 if (!username_obj || !domain_obj || !password_obj) {
 157                 creds_error:
 158                         *errstr = strdup("invalid credentials");
 159                         return NULL;
 160                 }
 161
 162                 if (!PyString_Check(username_obj) ||
 163                     !PyString_Check(domain_obj) ||
 164                     !PyString_Check(password_obj))
 165                         goto creds_error;
 166
 167                 username = PyString_AsString(username_obj);
 168                 domain = PyString_AsString(domain_obj);
 169                 password = PyString_AsString(password_obj);
 170
 171                 if (!username || !domain || !password)
 172                         goto creds_error;
 173         }
 174
 175         /* Now try to connect */
 176
 177         if (!resolve_name(server, &server_ip, 0x20))  {
 178                 asprintf(errstr, "unable to resolve %s", server);
 179                 return NULL;
 180         }
 181
 182         result = cli_full_connection(
 183                 &cli, global_myname, server, &server_ip, 0, "IPC$", "IPC",
 184                 username, domain, password, strlen(password));
 185
 186         if (!NT_STATUS_IS_OK(result) || !cli_nt_session_open(cli, pipe_name)) {
 187                 cli_shutdown(cli);
 188                 free(cli);
 189                 *errstr = strdup("pipe not available");
 190                 return NULL;
 191         }
 192
 193         *errstr = NULL;
 194
 195         return cli;
 196 }
 197
 198 /* Return true if a dictionary contains a "level" key with an integer
 199    value.  Set the value if so. */
 200
 201 BOOL get_level_value(PyObject *dict, uint32 *level)
 202 {
 203         PyObject *obj;
 204
 205         if (!(obj = PyDict_GetItemString(dict, "level")) ||
 206             !PyInt_Check(obj))
 207                 return False;
 208
 209         if (level)
 210                 *level = PyInt_AsLong(obj);
 211
 212         return True;
 213 }