import HEAD into svn+ssh://svn.samba.org/home/svn/samba/trunk
[metze/old/v3-2-winbind-ndr.git] / 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 /* Enumerate ports */
24
25 PyObject *spoolss_enumports(PyObject *self, PyObject *args, PyObject *kw)
26 {
27         WERROR werror;
28         PyObject *result = NULL, *creds = NULL;
29         uint32 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, *errstr;
35         PORT_INFO_CTR ctr;
36
37         /* Parse parameters */
38
39         if (!PyArg_ParseTupleAndKeywords(
40                     args, kw, "s|iO", kwlist, &server, &level, &creds))
41                 return NULL;
42         
43         if (server[0] != '\\' || server[1] != '\\') {
44                 PyErr_SetString(PyExc_ValueError, "UNC name required");
45                 return NULL;
46         }
47
48         server += 2;
49
50         if (creds && creds != Py_None && !PyDict_Check(creds)) {
51                 PyErr_SetString(PyExc_TypeError, 
52                                 "credentials must be dictionary or None");
53                 return NULL;
54         }
55
56         if (!(cli = open_pipe_creds(server, creds, PI_SPOOLSS, &errstr))) {
57                 PyErr_SetString(spoolss_error, errstr);
58                 free(errstr);
59                 goto done;
60         }
61
62         if (!(mem_ctx = talloc_init("spoolss_enumports"))) {
63                 PyErr_SetString(
64                         spoolss_error, "unable to init talloc context\n");
65                 goto done;
66         }
67
68         /* Call rpc function */
69         
70         werror = cli_spoolss_enum_ports(
71                 cli, mem_ctx, 0, &needed, level, &num_ports, &ctr);
72
73         if (W_ERROR_V(werror) == ERRinsufficientbuffer)
74                 werror = cli_spoolss_enum_ports(
75                         cli, mem_ctx, needed, NULL, level,
76                         &num_ports, &ctr);
77
78         if (!W_ERROR_IS_OK(werror)) {
79                 PyErr_SetObject(spoolss_werror, py_werror_tuple(werror));
80                 goto done;
81         }
82
83         /* Return value */
84         
85         switch (level) {
86         case 1: 
87                 result = PyDict_New();
88
89                 for (i = 0; i < num_ports; i++) {
90                         PyObject *value;
91                         fstring name;
92
93                         rpcstr_pull(name, ctr.port.info_1[i].port_name.buffer,
94                                     sizeof(fstring), -1, STR_TERMINATE);
95
96                         py_from_PORT_INFO_1(&value, &ctr.port.info_1[i]);
97
98                         PyDict_SetItemString(
99                                 value, "level", PyInt_FromLong(1));
100
101                         PyDict_SetItemString(result, name, value);
102                 }
103
104                 break;
105         case 2:
106                 result = PyDict_New();
107
108                 for(i = 0; i < num_ports; i++) {
109                         PyObject *value;
110                         fstring name;
111
112                         rpcstr_pull(name, ctr.port.info_2[i].port_name.buffer,
113                                     sizeof(fstring), -1, STR_TERMINATE);
114
115                         py_from_PORT_INFO_2(&value, &ctr.port.info_2[i]);
116
117                         PyDict_SetItemString(
118                                 value, "level", PyInt_FromLong(2));
119
120                         PyDict_SetItemString(result, name, value);
121                 }
122                 
123                 break;
124         default:
125                 PyErr_SetString(spoolss_error, "unknown info level");
126                 goto done;
127         }
128
129  done:
130         if (cli)
131                 cli_shutdown(cli);
132         
133         if (mem_ctx)
134                 talloc_destroy(mem_ctx);
135
136         return result;
137 }