HP2500C driver writes devmode with private data that ends
[samba.git] / 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 }