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#include "python/py_conv.h"
23
24struct pyconv py_PRINTER_INFO_0[] = {
25	{ "name", PY_UNISTR, offsetof(PRINTER_INFO_0, printername) },
26	{ "server_name", PY_UNISTR, offsetof(PRINTER_INFO_0, servername) },
27
28	{ "cjobs", PY_UINT32, offsetof(PRINTER_INFO_0, cjobs) },
29	{ "total_jobs", PY_UINT32, offsetof(PRINTER_INFO_0, total_jobs) },
30	{ "total_bytes", PY_UINT32, offsetof(PRINTER_INFO_0, total_bytes) },
31
32	{ "year", PY_UINT16, offsetof(PRINTER_INFO_0, year) },
33	{ "month", PY_UINT16, offsetof(PRINTER_INFO_0, month) },
34	{ "day_of_week", PY_UINT16, offsetof(PRINTER_INFO_0, dayofweek) },
35	{ "day", PY_UINT16, offsetof(PRINTER_INFO_0, day) },
36	{ "hour", PY_UINT16, offsetof(PRINTER_INFO_0, hour) },
37	{ "minute", PY_UINT16, offsetof(PRINTER_INFO_0, minute) },
38	{ "second", PY_UINT16, offsetof(PRINTER_INFO_0, second) },
39	{ "milliseconds", PY_UINT16, offsetof(PRINTER_INFO_0, milliseconds) },
40
41	{ "global_counter", PY_UINT32, offsetof(PRINTER_INFO_0, global_counter) },
42	{ "total_pages", PY_UINT32, offsetof(PRINTER_INFO_0, total_pages) },
43
44	{ "major_version", PY_UINT16, offsetof(PRINTER_INFO_0, major_version) },
45	{ "build_version", PY_UINT16, offsetof(PRINTER_INFO_0, build_version) },
46
47	{ "unknown7", PY_UINT32, offsetof(PRINTER_INFO_0, unknown7) },
48	{ "unknown8", PY_UINT32, offsetof(PRINTER_INFO_0, unknown8) },
49	{ "unknown9", PY_UINT32, offsetof(PRINTER_INFO_0, unknown9) },
50	{ "session_counter", PY_UINT32, offsetof(PRINTER_INFO_0, session_counter)},
51	{ "unknown11", PY_UINT32, offsetof(PRINTER_INFO_0, unknown11) },
52	{ "printer_errors", PY_UINT32, offsetof(PRINTER_INFO_0, printer_errors) },
53	{ "unknown13", PY_UINT32, offsetof(PRINTER_INFO_0, unknown13) },
54	{ "unknown14", PY_UINT32, offsetof(PRINTER_INFO_0, unknown14) },
55	{ "unknown15", PY_UINT32, offsetof(PRINTER_INFO_0, unknown15) },
56	{ "unknown16", PY_UINT32, offsetof(PRINTER_INFO_0, unknown16) },
57	{ "change_id", PY_UINT32, offsetof(PRINTER_INFO_0, change_id) },
58	{ "unknown18", PY_UINT32, offsetof(PRINTER_INFO_0, unknown18) },
59	{ "status", PY_UINT32, offsetof(PRINTER_INFO_0, status) },
60	{ "unknown20", PY_UINT32, offsetof(PRINTER_INFO_0, unknown20) },
61	{ "c_setprinter", PY_UINT32, offsetof(PRINTER_INFO_0, c_setprinter) },
62	{ "unknown22", PY_UINT32, offsetof(PRINTER_INFO_0, unknown22) },
63	{ "unknown23", PY_UINT32, offsetof(PRINTER_INFO_0, unknown23) },
64	{ "unknown24", PY_UINT32, offsetof(PRINTER_INFO_0, unknown24) },
65	{ "unknown25", PY_UINT32, offsetof(PRINTER_INFO_0, unknown25) },
66	{ "unknown26", PY_UINT32, offsetof(PRINTER_INFO_0, unknown26) },
67	{ "unknown27", PY_UINT32, offsetof(PRINTER_INFO_0, unknown27) },
68	{ "unknown28", PY_UINT32, offsetof(PRINTER_INFO_0, unknown28) },
69	{ "unknown29", PY_UINT32, offsetof(PRINTER_INFO_0, unknown29) },
70
71	{ NULL }
72};
73
74struct pyconv py_PRINTER_INFO_1[] = {
75	{ "name", PY_UNISTR, offsetof(PRINTER_INFO_1, name) },
76	{ "description", PY_UNISTR, offsetof(PRINTER_INFO_1, description) },
77	{ "comment", PY_UNISTR, offsetof(PRINTER_INFO_1, comment) },
78	{ "flags", PY_UINT32, offsetof(PRINTER_INFO_1, flags) },
79	{ NULL }
80};
81
82struct pyconv py_PRINTER_INFO_2[] = {
83	{ "server_name", PY_UNISTR, offsetof(PRINTER_INFO_2, servername) },
84	{ "name", PY_UNISTR, offsetof(PRINTER_INFO_2, printername) },
85	{ "share_name", PY_UNISTR, offsetof(PRINTER_INFO_2, sharename) },
86	{ "port_name", PY_UNISTR, offsetof(PRINTER_INFO_2, portname) },
87	{ "driver_name", PY_UNISTR, offsetof(PRINTER_INFO_2, drivername) },
88	{ "comment", PY_UNISTR, offsetof(PRINTER_INFO_2, comment) },
89	{ "location", PY_UNISTR, offsetof(PRINTER_INFO_2, location) },
90	{ "datatype", PY_UNISTR, offsetof(PRINTER_INFO_2, datatype) },
91	{ "sepfile", PY_UNISTR, offsetof(PRINTER_INFO_2, sepfile) },
92	{ "print_processor", PY_UNISTR, offsetof(PRINTER_INFO_2, printprocessor) },
93	{ "parameters", PY_UNISTR, offsetof(PRINTER_INFO_2, parameters) },
94	{ "attributes", PY_UINT32, offsetof(PRINTER_INFO_2, attributes) },
95	{ "default_priority", PY_UINT32, offsetof(PRINTER_INFO_2, defaultpriority) },
96	{ "priority", PY_UINT32, offsetof(PRINTER_INFO_2, priority) },
97	{ "start_time", PY_UINT32, offsetof(PRINTER_INFO_2, starttime) },
98	{ "until_time", PY_UINT32, offsetof(PRINTER_INFO_2, untiltime) },
99	{ "status", PY_UINT32, offsetof(PRINTER_INFO_2, status) },
100	{ "cjobs", PY_UINT32, offsetof(PRINTER_INFO_2, cjobs) },
101	{ "average_ppm", PY_UINT32, offsetof(PRINTER_INFO_2, averageppm) },
102	{ NULL }
103};
104
105struct pyconv py_PRINTER_INFO_3[] = {
106	{ "flags", PY_UINT32, offsetof(PRINTER_INFO_3, flags) },
107	{ NULL }
108};
109
110struct pyconv py_DEVICEMODE[] = {
111	{ "device_name", PY_UNISTR, offsetof(DEVICEMODE, devicename) },
112	{ "spec_version", PY_UINT16, offsetof(DEVICEMODE, specversion) },
113	{ "driver_version", PY_UINT16, offsetof(DEVICEMODE, driverversion) },
114	{ "size", PY_UINT16, offsetof(DEVICEMODE, size) },
115	{ "fields", PY_UINT16, offsetof(DEVICEMODE, fields) },
116	{ "orientation", PY_UINT16, offsetof(DEVICEMODE, orientation) },
117	{ "paper_size", PY_UINT16, offsetof(DEVICEMODE, papersize) },
118	{ "paper_width", PY_UINT16, offsetof(DEVICEMODE, paperwidth) },
119	{ "paper_length", PY_UINT16, offsetof(DEVICEMODE, paperlength) },
120	{ "scale", PY_UINT16, offsetof(DEVICEMODE, scale) },
121	{ "copies", PY_UINT16, offsetof(DEVICEMODE, copies) },
122	{ "default_source", PY_UINT16, offsetof(DEVICEMODE, defaultsource) },
123	{ "print_quality", PY_UINT16, offsetof(DEVICEMODE, printquality) },
124	{ "color", PY_UINT16, offsetof(DEVICEMODE, color) },
125	{ "duplex", PY_UINT16, offsetof(DEVICEMODE, duplex) },
126	{ "y_resolution", PY_UINT16, offsetof(DEVICEMODE, yresolution) },
127	{ "tt_option", PY_UINT16, offsetof(DEVICEMODE, ttoption) },
128	{ "collate", PY_UINT16, offsetof(DEVICEMODE, collate) },
129	{ "form_name", PY_UNISTR, offsetof(DEVICEMODE, formname) },
130	{ "log_pixels", PY_UINT16, offsetof(DEVICEMODE, logpixels) },
131	{ "bits_per_pel", PY_UINT32, offsetof(DEVICEMODE, bitsperpel) },
132	{ "pels_width", PY_UINT32, offsetof(DEVICEMODE, pelswidth) },
133	{ "pels_height", PY_UINT32, offsetof(DEVICEMODE, pelsheight) },
134	{ "display_flags", PY_UINT32, offsetof(DEVICEMODE, displayflags) },
135	{ "display_frequency", PY_UINT32, offsetof(DEVICEMODE, displayfrequency) },
136	{ "icm_method", PY_UINT32, offsetof(DEVICEMODE, icmmethod) },
137	{ "icm_intent", PY_UINT32, offsetof(DEVICEMODE, icmintent) },
138	{ "media_type", PY_UINT32, offsetof(DEVICEMODE, mediatype) },
139	{ "dither_type", PY_UINT32, offsetof(DEVICEMODE, dithertype) },
140	{ "reserved1", PY_UINT32, offsetof(DEVICEMODE, reserved1) },
141	{ "reserved2", PY_UINT32, offsetof(DEVICEMODE, reserved2) },
142	{ "panning_width", PY_UINT32, offsetof(DEVICEMODE, panningwidth) },
143	{ "panning_height", PY_UINT32, offsetof(DEVICEMODE, panningheight) },
144	{ NULL }
145};
146
147/*
148 * Convert between DEVICEMODE and Python
149 */
150
151BOOL py_from_DEVICEMODE(PyObject **dict, DEVICEMODE *devmode)
152{
153	*dict = from_struct(devmode, py_DEVICEMODE);
154
155	PyDict_SetItemString(*dict, "private",
156			     PyString_FromStringAndSize(
157				     devmode->private, devmode->driverextra));
158
159	return True;
160}
161
162BOOL py_to_DEVICEMODE(DEVICEMODE *devmode, PyObject *dict)
163{
164	PyObject *obj, *dict_copy = PyDict_Copy(dict);
165	BOOL result = False;
166
167	if (!(obj = PyDict_GetItemString(dict_copy, "private")))
168		goto done;
169
170	if (!PyString_Check(obj))
171		goto done;
172
173	devmode->private = PyString_AsString(obj);
174	devmode->driverextra = PyString_Size(obj);
175
176	PyDict_DelItemString(dict_copy, "private");
177
178	if (!to_struct(devmode, dict_copy, py_DEVICEMODE))
179		goto done;
180
181	result = True;
182
183done:
184	Py_DECREF(dict_copy);
185	return result;
186}
187
188/*
189 * Convert between PRINTER_INFO_0 and Python
190 */
191
192BOOL py_from_PRINTER_INFO_0(PyObject **dict, PRINTER_INFO_0 *info)
193{
194	*dict = from_struct(info, py_PRINTER_INFO_0);
195	PyDict_SetItemString(*dict, "level", PyInt_FromLong(0));
196	return True;
197}
198
199BOOL py_to_PRINTER_INFO_0(PRINTER_INFO_0 *info, PyObject *dict)
200{
201	return False;
202}
203
204/*
205 * Convert between PRINTER_INFO_1 and Python
206 */
207
208BOOL py_from_PRINTER_INFO_1(PyObject **dict, PRINTER_INFO_1 *info)
209{
210	*dict = from_struct(info, py_PRINTER_INFO_1);
211	PyDict_SetItemString(*dict, "level", PyInt_FromLong(1));
212	return True;
213}
214
215BOOL py_to_PRINTER_INFO_1(PRINTER_INFO_1 *info, PyObject *dict)
216{
217	PyObject *obj, *dict_copy = PyDict_Copy(dict);
218	BOOL result = False;
219
220	if (!(obj = PyDict_GetItemString(dict_copy, "level")) ||
221	    !PyInt_Check(obj))
222		goto done;
223
224	PyDict_DelItemString(dict_copy, "level");
225
226	if (!to_struct(info, dict_copy, py_PRINTER_INFO_1))
227		goto done;
228
229	result = True;
230
231done:
232	Py_DECREF(dict_copy);
233	return result;
234}
235
236/*
237 * Convert between PRINTER_INFO_2 and Python
238 */
239
240BOOL py_from_PRINTER_INFO_2(PyObject **dict, PRINTER_INFO_2 *info)
241{
242	PyObject *obj;
243
244	*dict = from_struct(info, py_PRINTER_INFO_2);
245
246	/* The security descriptor could be NULL */
247
248	if (info->secdesc) {
249		if (py_from_SECDESC(&obj, info->secdesc))
250			PyDict_SetItemString(*dict, "security_descriptor", obj);
251	}
252
253	/* Bong!  The devmode could be NULL */
254
255	if (info->devmode)
256		py_from_DEVICEMODE(&obj, info->devmode);
257	else
258		obj = PyDict_New();
259
260	PyDict_SetItemString(*dict, "device_mode", obj);
261
262	PyDict_SetItemString(*dict, "level", PyInt_FromLong(2));
263
264	return True;
265}
266
267BOOL py_to_PRINTER_INFO_2(PRINTER_INFO_2 *info, PyObject *dict,
268			  TALLOC_CTX *mem_ctx)
269{
270	PyObject *obj, *dict_copy = PyDict_Copy(dict);
271	BOOL result = False;
272
273	/* Convert security descriptor - may be NULL */
274
275	info->secdesc = NULL;
276
277	if ((obj = PyDict_GetItemString(dict_copy, "security_descriptor"))) {
278
279		if (!PyDict_Check(obj))
280			goto done;
281
282		if (!py_to_SECDESC(&info->secdesc, obj, mem_ctx))
283			goto done;
284
285		PyDict_DelItemString(dict_copy, "security_descriptor");
286	}
287
288	/* Convert device mode */
289
290	if (!(obj = PyDict_GetItemString(dict_copy, "device_mode"))
291	    || !PyDict_Check(obj))
292		goto done;
293
294	info->devmode = talloc(mem_ctx, sizeof(DEVICEMODE));
295
296	if (!py_to_DEVICEMODE(info->devmode, obj))
297		goto done;
298
299	PyDict_DelItemString(dict_copy, "device_mode");
300
301	/* Check info level */
302
303	if (!(obj = PyDict_GetItemString(dict_copy, "level")) ||
304	    !PyInt_Check(obj))
305		goto done;
306
307	PyDict_DelItemString(dict_copy, "level");
308
309	/* Convert remaining elements of dictionary */
310
311	if (!to_struct(info, dict_copy, py_PRINTER_INFO_2))
312		goto done;
313
314	result = True;
315
316done:
317	Py_DECREF(dict_copy);
318	return result;
319}
320
321/*
322 * Convert between PRINTER_INFO_1 and Python
323 */
324
325BOOL py_from_PRINTER_INFO_3(PyObject **dict, PRINTER_INFO_3 *info)
326{
327	PyObject *obj;
328
329	*dict = from_struct(info, py_PRINTER_INFO_3);
330
331	if (py_from_SECDESC(&obj, info->secdesc))
332		PyDict_SetItemString(*dict, "security_descriptor", obj);
333
334	PyDict_SetItemString(*dict, "level", PyInt_FromLong(3));
335
336	return True;
337}
338
339BOOL py_to_PRINTER_INFO_3(PRINTER_INFO_3 *info, PyObject *dict,
340			  TALLOC_CTX *mem_ctx)
341{
342	PyObject *obj;
343
344	if (!to_struct(info, dict, py_PRINTER_INFO_3))
345		return False;
346
347	if (!(obj = PyDict_GetItemString(dict, "security_descriptor")))
348		return False;
349
350	if (!py_to_SECDESC(&info->secdesc, obj, mem_ctx))
351		return False;
352
353	return True;
354}
355