2 Python wrappers for DCERPC/SMB client routines.
4 Copyright (C) Tim Potter, 2002
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.
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.
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.
21 #include "python/py_spoolss.h"
23 /* Enumerate printer drivers */
25 PyObject
*spoolss_enumprinterdrivers(PyObject
*self
, PyObject
*args
,
29 PyObject
*result
= NULL
, *creds
= NULL
;
30 PRINTER_DRIVER_CTR ctr
;
32 uint32 needed
, num_drivers
;
33 char *arch
= "Windows NT x86", *server
, *errstr
;
34 static char *kwlist
[] = {"server", "level", "creds", "arch", NULL
};
35 struct cli_state
*cli
= NULL
;
36 TALLOC_CTX
*mem_ctx
= NULL
;
38 /* Parse parameters */
40 if (!PyArg_ParseTupleAndKeywords(
41 args
, kw
, "s|iOs", kwlist
, &server
, &level
, &creds
,
45 if (server
[0] != '\\' || server
[1] != '\\') {
46 PyErr_SetString(PyExc_ValueError
, "UNC name required");
52 if (creds
&& creds
!= Py_None
&& !PyDict_Check(creds
)) {
53 PyErr_SetString(PyExc_TypeError
,
54 "credentials must be dictionary or None");
58 /* Call rpc function */
60 if (!(cli
= open_pipe_creds(server
, creds
, PI_SPOOLSS
, &errstr
))) {
61 PyErr_SetString(spoolss_error
, errstr
);
66 if (!(mem_ctx
= talloc_init("spoolss_enumprinterdrivers"))) {
68 spoolss_error
, "unable to init talloc context\n");
72 werror
= cli_spoolss_enumprinterdrivers(
73 cli
, mem_ctx
, 0, &needed
, level
, arch
,
76 if (W_ERROR_V(werror
) == ERRinsufficientbuffer
)
77 werror
= cli_spoolss_enumprinterdrivers(
78 cli
, mem_ctx
, needed
, NULL
, level
, arch
,
81 if (!W_ERROR_IS_OK(werror
)) {
82 PyErr_SetObject(spoolss_werror
, py_werror_tuple(werror
));
90 result
= PyDict_New();
92 for (i
= 0; i
< num_drivers
; i
++) {
96 rpcstr_pull(name
, ctr
.info1
[i
].name
.buffer
,
97 sizeof(fstring
), -1, STR_TERMINATE
);
99 py_from_DRIVER_INFO_1(&value
, &ctr
.info1
[i
]);
101 PyDict_SetItemString(result
, name
, value
);
106 result
= PyDict_New();
108 for(i
= 0; i
< num_drivers
; i
++) {
112 rpcstr_pull(name
, ctr
.info2
[i
].name
.buffer
,
113 sizeof(fstring
), -1, STR_TERMINATE
);
115 py_from_DRIVER_INFO_2(&value
, &ctr
.info2
[i
]);
117 PyDict_SetItemString(result
, name
, value
);
122 result
= PyDict_New();
124 for(i
= 0; i
< num_drivers
; i
++) {
128 rpcstr_pull(name
, ctr
.info3
[i
].name
.buffer
,
129 sizeof(fstring
), -1, STR_TERMINATE
);
131 py_from_DRIVER_INFO_3(&value
, &ctr
.info3
[i
]);
133 PyDict_SetItemString(result
, name
, value
);
138 result
= PyDict_New();
140 for(i
= 0; i
< num_drivers
; i
++) {
144 rpcstr_pull(name
, ctr
.info6
[i
].name
.buffer
,
145 sizeof(fstring
), -1, STR_TERMINATE
);
147 py_from_DRIVER_INFO_6(&value
, &ctr
.info6
[i
]);
149 PyList_SetItem(result
, i
, value
);
154 PyErr_SetString(spoolss_error
, "unknown info level");
163 talloc_destroy(mem_ctx
);
168 /* Fetch printer driver */
170 PyObject
*spoolss_hnd_getprinterdriver(PyObject
*self
, PyObject
*args
,
173 spoolss_policy_hnd_object
*hnd
= (spoolss_policy_hnd_object
*)self
;
175 PyObject
*result
= Py_None
;
176 PRINTER_DRIVER_CTR ctr
;
179 char *arch
= "Windows NT x86";
181 static char *kwlist
[] = {"level", "arch", NULL
};
183 /* Parse parameters */
185 if (!PyArg_ParseTupleAndKeywords(
186 args
, kw
, "|is", kwlist
, &level
, &arch
))
189 /* Call rpc function */
191 werror
= cli_spoolss_getprinterdriver(
192 hnd
->cli
, hnd
->mem_ctx
, 0, &needed
, &hnd
->pol
, level
,
193 version
, arch
, &ctr
);
195 if (W_ERROR_V(werror
) == ERRinsufficientbuffer
)
196 werror
= cli_spoolss_getprinterdriver(
197 hnd
->cli
, hnd
->mem_ctx
, needed
, NULL
, &hnd
->pol
,
198 level
, version
, arch
, &ctr
);
200 if (!W_ERROR_IS_OK(werror
)) {
201 PyErr_SetObject(spoolss_werror
, py_werror_tuple(werror
));
209 py_from_DRIVER_INFO_1(&result
, ctr
.info1
);
212 py_from_DRIVER_INFO_2(&result
, ctr
.info2
);
215 py_from_DRIVER_INFO_3(&result
, ctr
.info3
);
218 py_from_DRIVER_INFO_6(&result
, ctr
.info6
);
221 PyErr_SetString(spoolss_error
, "unsupported info level");
229 /* Fetch printer driver directory */
231 PyObject
*spoolss_getprinterdriverdir(PyObject
*self
, PyObject
*args
,
235 PyObject
*result
= NULL
, *creds
= NULL
;
236 DRIVER_DIRECTORY_CTR ctr
;
237 uint32 needed
, level
= 1;
238 char *arch
= "Windows NT x86", *server
, *errstr
;
239 static char *kwlist
[] = {"server", "level", "arch", "creds", NULL
};
240 struct cli_state
*cli
= NULL
;
241 TALLOC_CTX
*mem_ctx
= NULL
;
243 /* Parse parameters */
245 if (!PyArg_ParseTupleAndKeywords(
246 args
, kw
, "s|isO", kwlist
, &server
, &level
,
250 if (server
[0] != '\\' || server
[1] != '\\') {
251 PyErr_SetString(PyExc_ValueError
, "UNC name required");
257 if (creds
&& creds
!= Py_None
&& !PyDict_Check(creds
)) {
258 PyErr_SetString(PyExc_TypeError
,
259 "credentials must be dictionary or None");
263 /* Call rpc function */
265 if (!(cli
= open_pipe_creds(server
, creds
, PI_SPOOLSS
, &errstr
))) {
266 PyErr_SetString(spoolss_error
, errstr
);
271 if (!(mem_ctx
= talloc_init("spoolss_getprinterdriverdir"))) {
273 spoolss_error
, "unable to init talloc context\n");
277 werror
= cli_spoolss_getprinterdriverdir(
278 cli
, mem_ctx
, 0, &needed
, level
, arch
, &ctr
);
280 if (W_ERROR_V(werror
) == ERRinsufficientbuffer
)
281 werror
= cli_spoolss_getprinterdriverdir(
282 cli
, mem_ctx
, needed
, NULL
, level
, arch
, &ctr
);
284 if (!W_ERROR_IS_OK(werror
)) {
285 PyErr_SetObject(spoolss_werror
, py_werror_tuple(werror
));
293 py_from_DRIVER_DIRECTORY_1(&result
, ctr
.info1
);
296 PyErr_SetString(spoolss_error
, "unknown info level");
305 talloc_destroy(mem_ctx
);
310 PyObject
*spoolss_addprinterdriver(PyObject
*self
, PyObject
*args
,
313 static char *kwlist
[] = { "server", "info", "creds", NULL
};
314 char *server
, *errstr
;
316 PyObject
*info
, *result
= NULL
, *creds
= NULL
;
318 TALLOC_CTX
*mem_ctx
= NULL
;
319 struct cli_state
*cli
= NULL
;
320 PRINTER_DRIVER_CTR ctr
;
322 DRIVER_INFO_3 driver_3
;
325 if (!PyArg_ParseTupleAndKeywords(
326 args
, kw
, "sO!|O", kwlist
, &server
, &PyDict_Type
,
330 if (server
[0] == '\\' || server
[1] == '\\')
333 if (creds
&& creds
!= Py_None
&& !PyDict_Check(creds
)) {
334 PyErr_SetString(PyExc_TypeError
,
335 "credentials must be dictionary or None");
339 if (!(mem_ctx
= talloc_init("spoolss_addprinterdriver"))) {
341 spoolss_error
, "unable to init talloc context\n");
345 if (!(cli
= open_pipe_creds(server
, creds
, PI_SPOOLSS
, &errstr
))) {
346 PyErr_SetString(spoolss_error
, errstr
);
351 if (!get_level_value(info
, &level
)) {
352 PyErr_SetString(spoolss_error
, "invalid info level");
357 PyErr_SetString(spoolss_error
, "unsupported info level");
366 ctr
.info3
= &dinfo
.driver_3
;
368 if (!py_to_DRIVER_INFO_3(&dinfo
.driver_3
, info
)) {
369 PyErr_SetString(spoolss_error
,
370 "error converting to driver info 3");
376 PyErr_SetString(spoolss_error
, "unsupported info level");
380 werror
= cli_spoolss_addprinterdriver(cli
, mem_ctx
, level
, &ctr
);
382 if (!W_ERROR_IS_OK(werror
)) {
383 PyErr_SetObject(spoolss_werror
, py_werror_tuple(werror
));
395 talloc_destroy(mem_ctx
);
401 PyObject
*spoolss_addprinterdriverex(PyObject
*self
, PyObject
*args
,
404 /* Not supported by Samba server */
406 PyErr_SetString(spoolss_error
, "Not implemented");
410 PyObject
*spoolss_deleteprinterdriver(PyObject
*self
, PyObject
*args
,
413 PyErr_SetString(spoolss_error
, "Not implemented");
417 PyObject
*spoolss_deleteprinterdriverex(PyObject
*self
, PyObject
*args
,
420 PyErr_SetString(spoolss_error
, "Not implemented");