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";
180 static char *kwlist
[] = {"level", "arch", NULL
};
182 /* Parse parameters */
184 if (!PyArg_ParseTupleAndKeywords(
185 args
, kw
, "|is", kwlist
, &level
, &arch
))
188 /* Call rpc function */
190 werror
= cli_spoolss_getprinterdriver(
191 hnd
->cli
, hnd
->mem_ctx
, 0, &needed
, &hnd
->pol
, level
,
194 if (W_ERROR_V(werror
) == ERRinsufficientbuffer
)
195 werror
= cli_spoolss_getprinterdriver(
196 hnd
->cli
, hnd
->mem_ctx
, needed
, NULL
, &hnd
->pol
,
199 if (!W_ERROR_IS_OK(werror
)) {
200 PyErr_SetObject(spoolss_werror
, py_werror_tuple(werror
));
208 py_from_DRIVER_INFO_1(&result
, ctr
.info1
);
211 py_from_DRIVER_INFO_2(&result
, ctr
.info2
);
214 py_from_DRIVER_INFO_3(&result
, ctr
.info3
);
217 py_from_DRIVER_INFO_6(&result
, ctr
.info6
);
220 PyErr_SetString(spoolss_error
, "unsupported info level");
228 /* Fetch printer driver directory */
230 PyObject
*spoolss_getprinterdriverdir(PyObject
*self
, PyObject
*args
,
234 PyObject
*result
= NULL
, *creds
= NULL
;
235 DRIVER_DIRECTORY_CTR ctr
;
236 uint32 needed
, level
= 1;
237 char *arch
= "Windows NT x86", *server
, *errstr
;
238 static char *kwlist
[] = {"server", "level", "arch", "creds", NULL
};
239 struct cli_state
*cli
= NULL
;
240 TALLOC_CTX
*mem_ctx
= NULL
;
242 /* Parse parameters */
244 if (!PyArg_ParseTupleAndKeywords(
245 args
, kw
, "s|isO", kwlist
, &server
, &level
,
249 if (server
[0] != '\\' || server
[1] != '\\') {
250 PyErr_SetString(PyExc_ValueError
, "UNC name required");
256 if (creds
&& creds
!= Py_None
&& !PyDict_Check(creds
)) {
257 PyErr_SetString(PyExc_TypeError
,
258 "credentials must be dictionary or None");
262 /* Call rpc function */
264 if (!(cli
= open_pipe_creds(server
, creds
, PI_SPOOLSS
, &errstr
))) {
265 PyErr_SetString(spoolss_error
, errstr
);
270 if (!(mem_ctx
= talloc_init("spoolss_getprinterdriverdir"))) {
272 spoolss_error
, "unable to init talloc context\n");
276 werror
= cli_spoolss_getprinterdriverdir(
277 cli
, mem_ctx
, 0, &needed
, level
, arch
, &ctr
);
279 if (W_ERROR_V(werror
) == ERRinsufficientbuffer
)
280 werror
= cli_spoolss_getprinterdriverdir(
281 cli
, mem_ctx
, needed
, NULL
, level
, arch
, &ctr
);
283 if (!W_ERROR_IS_OK(werror
)) {
284 PyErr_SetObject(spoolss_werror
, py_werror_tuple(werror
));
292 py_from_DRIVER_DIRECTORY_1(&result
, ctr
.info1
);
295 PyErr_SetString(spoolss_error
, "unknown info level");
304 talloc_destroy(mem_ctx
);
309 PyObject
*spoolss_addprinterdriver(PyObject
*self
, PyObject
*args
,
312 static char *kwlist
[] = { "server", "info", "creds", NULL
};
313 char *server
, *errstr
;
315 PyObject
*info
, *result
= NULL
, *creds
= NULL
;
317 TALLOC_CTX
*mem_ctx
= NULL
;
318 struct cli_state
*cli
= NULL
;
319 PRINTER_DRIVER_CTR ctr
;
321 DRIVER_INFO_3 driver_3
;
324 if (!PyArg_ParseTupleAndKeywords(
325 args
, kw
, "sO!|O", kwlist
, &server
, &PyDict_Type
,
329 if (server
[0] == '\\' || server
[1] == '\\')
332 if (creds
&& creds
!= Py_None
&& !PyDict_Check(creds
)) {
333 PyErr_SetString(PyExc_TypeError
,
334 "credentials must be dictionary or None");
338 if (!(mem_ctx
= talloc_init("spoolss_addprinterdriver"))) {
340 spoolss_error
, "unable to init talloc context\n");
344 if (!(cli
= open_pipe_creds(server
, creds
, PI_SPOOLSS
, &errstr
))) {
345 PyErr_SetString(spoolss_error
, errstr
);
350 if (!get_level_value(info
, &level
)) {
351 PyErr_SetString(spoolss_error
, "invalid info level");
356 PyErr_SetString(spoolss_error
, "unsupported info level");
365 ctr
.info3
= &dinfo
.driver_3
;
367 if (!py_to_DRIVER_INFO_3(&dinfo
.driver_3
, info
)) {
368 PyErr_SetString(spoolss_error
,
369 "error converting to driver info 3");
375 PyErr_SetString(spoolss_error
, "unsupported info level");
379 werror
= cli_spoolss_addprinterdriver(cli
, mem_ctx
, level
, &ctr
);
381 if (!W_ERROR_IS_OK(werror
)) {
382 PyErr_SetObject(spoolss_werror
, py_werror_tuple(werror
));
394 talloc_destroy(mem_ctx
);
400 PyObject
*spoolss_addprinterdriverex(PyObject
*self
, PyObject
*args
,
403 /* Not supported by Samba server */
405 PyErr_SetString(spoolss_error
, "Not implemented");
409 PyObject
*spoolss_deleteprinterdriver(PyObject
*self
, PyObject
*args
,
412 PyErr_SetString(spoolss_error
, "Not implemented");
416 PyObject
*spoolss_deleteprinterdriverex(PyObject
*self
, PyObject
*args
,
419 PyErr_SetString(spoolss_error
, "Not implemented");