source3/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 /* Enumerate ports */
  24
  25 PyObject *spoolss_enumports(PyObject *self, PyObject *args, PyObject *kw)
  26 {
  27         WERROR werror;
  28         PyObject *result, *creds = NULL;
  29         int level = 1;
  30         uint32 i, needed, num_ports;
  31         static char *kwlist[] = {"server", "level", "creds", NULL};
  32         TALLOC_CTX *mem_ctx = NULL;
  33         struct cli_state *cli = NULL;
  34         char *server;
  35         PORT_INFO_CTR ctr;
  36
  37         /* Parse parameters */
  38
  39         if (!PyArg_ParseTupleAndKeywords(args, kw, "s|iO!", kwlist,
  40                                          &server, &creds, &level,
  41                                          &PyDict_Type))
  42                 return NULL;
  43
  44         if (server[0] == '\\' && server[1] == '\\')
  45                 server += 2;
  46
  47         mem_ctx = talloc_init();
  48         cli = open_pipe_creds(server, creds, cli_spoolss_initialise, NULL);
  49
  50         /* Call rpc function */
  51
  52         werror = cli_spoolss_enum_ports(
  53                 cli, mem_ctx, 0, &needed, level, &num_ports, &ctr);
  54
  55         if (W_ERROR_V(werror) == ERRinsufficientbuffer)
  56                 werror = cli_spoolss_enum_ports(
  57                         cli, mem_ctx, needed, NULL, level,
  58                         &num_ports, &ctr);
  59
  60         /* Return value */
  61
  62         result = Py_None;
  63
  64         if (!W_ERROR_IS_OK(werror))
  65                 goto done;
  66
  67         result = PyList_New(num_ports);
  68
  69         switch (level) {
  70         case 1:
  71                 for (i = 0; i < num_ports; i++) {
  72                         PyObject *value;
  73
  74                         py_from_PORT_INFO_1(&value, &ctr.port.info_1[i]);
  75
  76                         PyList_SetItem(result, i, value);
  77                 }
  78
  79                 break;
  80         case 2:
  81                 for(i = 0; i < num_ports; i++) {
  82                         PyObject *value;
  83
  84                         py_from_PORT_INFO_2(&value, &ctr.port.info_2[i]);
  85
  86                         PyList_SetItem(result, i, value);
  87                 }
  88
  89                 break;
  90         }
  91
  92  done:
  93         Py_INCREF(result);
  94         return result;
  95 }