source/python/py_spoolss_ports.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 "python/py_spoolss.h"
  22
  23 struct pyconv py_PORT_INFO_1[] = {
  24         { "name", PY_UNISTR, offsetof(PORT_INFO_1, port_name) },
  25         { NULL }
  26 };
  27
  28 struct pyconv py_PORT_INFO_2[] = {
  29         { "name", PY_UNISTR, offsetof(PORT_INFO_2, port_name) },
  30         { "monitor_name", PY_UNISTR, offsetof(PORT_INFO_2, monitor_name) },
  31         { "description", PY_UNISTR, offsetof(PORT_INFO_2, description) },
  32         { "reserved", PY_UINT32, offsetof(PORT_INFO_2, reserved) },
  33         { "type", PY_UINT32, offsetof(PORT_INFO_2, port_type) },
  34         { NULL }
  35 };
  36
  37 /* Enumerate ports */
  38
  39 PyObject *spoolss_enumports(PyObject *self, PyObject *args, PyObject *kw)
  40 {
  41         WERROR werror;
  42         PyObject *result, *creds = NULL;
  43         int level = 1;
  44         uint32 i, needed, num_ports;
  45         static char *kwlist[] = {"server", "level", "creds", NULL};
  46         TALLOC_CTX *mem_ctx = NULL;
  47         struct cli_state *cli = NULL;
  48         char *server;
  49         PORT_INFO_CTR ctr;
  50
  51         /* Parse parameters */
  52
  53         if (!PyArg_ParseTupleAndKeywords(args, kw, "s|iO!", kwlist,
  54                                          &server, &creds, &level,
  55                                          &PyDict_Type))
  56                 return NULL;
  57
  58         if (server[0] == '\\' && server[1] == '\\')
  59                 server += 2;
  60
  61         mem_ctx = talloc_init();
  62         cli = open_pipe_creds(server, creds, cli_spoolss_initialise, NULL);
  63
  64         /* Call rpc function */
  65
  66         werror = cli_spoolss_enum_ports(
  67                 cli, mem_ctx, 0, &needed, level, &num_ports, &ctr);
  68
  69         if (W_ERROR_V(werror) == ERRinsufficientbuffer)
  70                 werror = cli_spoolss_enum_ports(
  71                         cli, mem_ctx, needed, NULL, level,
  72                         &num_ports, &ctr);
  73
  74         /* Return value */
  75
  76         result = Py_None;
  77
  78         if (!W_ERROR_IS_OK(werror))
  79                 goto done;
  80
  81         result = PyList_New(num_ports);
  82
  83         switch (level) {
  84         case 1:
  85                 for (i = 0; i < num_ports; i++) {
  86                         PyObject *value;
  87
  88                         value = from_struct (
  89                                 &ctr.port.info_1[i], py_PORT_INFO_1);
  90
  91                         PyList_SetItem(result, i, value);
  92                 }
  93
  94                 break;
  95         case 2:
  96                 for(i = 0; i < num_ports; i++) {
  97                         PyObject *value;
  98
  99                         value = from_struct(
 100                                 &ctr.port.info_2[i], py_PORT_INFO_2);
 101
 102                         PyList_SetItem(result, i, value);
 103                 }
 104
 105                 break;
 106         }
 107
 108  done:
 109         Py_INCREF(result);
 110         return result;
 111 }