Fix some argument checking bugs found by testsuite.
[tprouty/samba.git] / source / python / py_spoolss_printers_conv.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 #include "python/py_conv.h"
23
24 struct 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
74 struct 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
82 struct 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
105 struct pyconv py_PRINTER_INFO_3[] = {
106         { "flags", PY_UINT32, offsetof(PRINTER_INFO_3, flags) },
107         { NULL }
108 };      
109
110 struct 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
151 BOOL 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
162 BOOL 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
183 done:
184         Py_DECREF(dict_copy);
185         return result;
186 }
187
188 /*
189  * Convert between PRINTER_INFO_0 and Python 
190  */
191
192 BOOL 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
199 BOOL 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
208 BOOL 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
215 BOOL 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
231 done:
232         Py_DECREF(dict_copy);
233         return result;
234 }
235
236 /*
237  * Convert between PRINTER_INFO_2 and Python 
238  */
239
240 BOOL 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
267 BOOL 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
316 done:
317         Py_DECREF(dict_copy);
318         return result;
319 }
320
321 /*
322  * Convert between PRINTER_INFO_1 and Python 
323  */
324
325 BOOL 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
339 BOOL 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 }