Commit | Line | Data |
---|---|---|
de75fe0a RS |
1 | /* |
2 | * Implementation of some printer driver bits | |
3 | * | |
4 | * Copyright 1996 John Harvey | |
5 | * Copyright 1998 Huw Davies | |
6 | * Copyright 1998 Andreas Mohr | |
7 | * Copyright 1999 Klaas van Gend | |
8 | * | |
9 | * This library is free software; you can redistribute it and/or | |
10 | * modify it under the terms of the GNU Lesser General Public | |
11 | * License as published by the Free Software Foundation; either | |
12 | * version 2.1 of the License, or (at your option) any later version. | |
13 | * | |
14 | * This library is distributed in the hope that it will be useful, | |
15 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
16 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
17 | * Lesser General Public License for more details. | |
18 | * | |
19 | * You should have received a copy of the GNU Lesser General Public | |
20 | * License along with this library; if not, write to the Free Software | |
21 | * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA | |
22 | */ | |
23 | ||
24 | #include "config.h" | |
25 | #include "wine/port.h" | |
26 | ||
27 | #include <stdarg.h> | |
28 | #include <stdio.h> | |
29 | #include <signal.h> | |
30 | #include <stdlib.h> | |
31 | #include <string.h> | |
32 | #include <ctype.h> | |
33 | #include <errno.h> | |
34 | #ifdef HAVE_IO_H | |
35 | # include <io.h> | |
36 | #endif | |
37 | #ifdef HAVE_UNISTD_H | |
38 | # include <unistd.h> | |
39 | #endif | |
40 | #include <fcntl.h> | |
41 | #include "windef.h" | |
42 | #include "winbase.h" | |
43 | #include "winuser.h" | |
44 | #include "wine/winbase16.h" | |
45 | #include "wine/wingdi16.h" | |
46 | #include "winspool.h" | |
47 | #include "winerror.h" | |
48 | #include "winreg.h" | |
49 | #include "wownt32.h" | |
50 | #include "wine/debug.h" | |
51 | #include "gdi_private.h" | |
52 | ||
53 | WINE_DEFAULT_DEBUG_CHANNEL(print); | |
54 | ||
55 | static const char PrinterModel[] = "Printer Model"; | |
56 | static const char DefaultDevMode[] = "Default DevMode"; | |
57 | static const char PrinterDriverData[] = "PrinterDriverData"; | |
58 | static const char Printers[] = "System\\CurrentControlSet\\Control\\Print\\Printers\\"; | |
59 | ||
60 | /********************************************************************** | |
61 | * QueryAbort (GDI.155) | |
62 | * | |
63 | * Calls the app's AbortProc function if avail. | |
64 | * | |
65 | * RETURNS | |
66 | * TRUE if no AbortProc avail or AbortProc wants to continue printing. | |
67 | * FALSE if AbortProc wants to abort printing. | |
68 | */ | |
69 | BOOL16 WINAPI QueryAbort16(HDC16 hdc16, INT16 reserved) | |
70 | { | |
71 | BOOL ret = TRUE; | |
72 | HDC hdc = HDC_32( hdc16 ); | |
73 | DC *dc = get_dc_ptr( hdc ); | |
74 | ||
75 | if(!dc) { | |
76 | ERR("Invalid hdc %p\n", hdc); | |
77 | return FALSE; | |
78 | } | |
79 | if (dc->pAbortProc) ret = dc->pAbortProc(hdc, 0); | |
80 | release_dc_ptr( dc ); | |
81 | return ret; | |
82 | } | |
83 | ||
84 | ||
85 | /********************************************************************** | |
86 | * call_abort_proc16 | |
87 | */ | |
88 | static BOOL CALLBACK call_abort_proc16( HDC hdc, INT code ) | |
89 | { | |
90 | ABORTPROC16 proc16; | |
91 | DC *dc = get_dc_ptr( hdc ); | |
92 | ||
93 | if (!dc) return FALSE; | |
94 | proc16 = dc->pAbortProc16; | |
95 | release_dc_ptr( dc ); | |
96 | if (proc16) | |
97 | { | |
98 | WORD args[2]; | |
99 | DWORD ret; | |
100 | ||
101 | args[1] = HDC_16(hdc); | |
102 | args[0] = code; | |
103 | WOWCallback16Ex( (DWORD)proc16, WCB16_PASCAL, sizeof(args), args, &ret ); | |
104 | return LOWORD(ret); | |
105 | } | |
106 | return TRUE; | |
107 | } | |
108 | ||
109 | ||
110 | /********************************************************************** | |
111 | * SetAbortProc (GDI.381) | |
112 | */ | |
113 | INT16 WINAPI SetAbortProc16(HDC16 hdc16, ABORTPROC16 abrtprc) | |
114 | { | |
115 | HDC hdc = HDC_32( hdc16 ); | |
116 | DC *dc = get_dc_ptr( hdc ); | |
117 | ||
118 | if (!dc) return FALSE; | |
119 | dc->pAbortProc16 = abrtprc; | |
120 | dc->pAbortProc = call_abort_proc16; | |
121 | release_dc_ptr( dc ); | |
122 | return TRUE; | |
123 | } | |
124 | ||
125 | /********************************************************************** | |
126 | * SetAbortProc (GDI32.@) | |
127 | * | |
128 | */ | |
129 | INT WINAPI SetAbortProc(HDC hdc, ABORTPROC abrtprc) | |
130 | { | |
131 | DC *dc = get_dc_ptr( hdc ); | |
132 | ||
133 | if (!dc) return FALSE; | |
134 | dc->pAbortProc = abrtprc; | |
135 | release_dc_ptr( dc ); | |
136 | return TRUE; | |
137 | } | |
138 | ||
139 | ||
140 | /****************** misc. printer related functions */ | |
141 | ||
142 | /* | |
143 | * The following function should implement a queing system | |
144 | */ | |
145 | struct hpq | |
146 | { | |
147 | struct hpq *next; | |
148 | int tag; | |
149 | int key; | |
150 | }; | |
151 | ||
152 | static struct hpq *hpqueue; | |
153 | ||
154 | /********************************************************************** | |
155 | * CreatePQ (GDI.230) | |
156 | * | |
157 | */ | |
158 | HPQ16 WINAPI CreatePQ16(INT16 size) | |
159 | { | |
160 | #if 0 | |
161 | HGLOBAL16 hpq = 0; | |
162 | WORD tmp_size; | |
163 | LPWORD pPQ; | |
164 | ||
165 | tmp_size = size << 2; | |
166 | if (!(hpq = GlobalAlloc16(GMEM_SHARE|GMEM_MOVEABLE, tmp_size + 8))) | |
167 | return 0xffff; | |
168 | pPQ = GlobalLock16(hpq); | |
169 | *pPQ++ = 0; | |
170 | *pPQ++ = tmp_size; | |
171 | *pPQ++ = 0; | |
172 | *pPQ++ = 0; | |
173 | GlobalUnlock16(hpq); | |
174 | ||
175 | return (HPQ16)hpq; | |
176 | #else | |
177 | FIXME("(%d): stub\n",size); | |
178 | return 1; | |
179 | #endif | |
180 | } | |
181 | ||
182 | /********************************************************************** | |
183 | * DeletePQ (GDI.235) | |
184 | * | |
185 | */ | |
186 | INT16 WINAPI DeletePQ16(HPQ16 hPQ) | |
187 | { | |
188 | return GlobalFree16(hPQ); | |
189 | } | |
190 | ||
191 | /********************************************************************** | |
192 | * ExtractPQ (GDI.232) | |
193 | * | |
194 | */ | |
195 | INT16 WINAPI ExtractPQ16(HPQ16 hPQ) | |
196 | { | |
197 | struct hpq *queue, *prev, *current, *currentPrev; | |
198 | int key = 0, tag = -1; | |
199 | currentPrev = prev = NULL; | |
200 | queue = current = hpqueue; | |
201 | if (current) | |
202 | key = current->key; | |
203 | ||
204 | while (current) | |
205 | { | |
206 | currentPrev = current; | |
207 | current = current->next; | |
208 | if (current) | |
209 | { | |
210 | if (current->key < key) | |
211 | { | |
212 | queue = current; | |
213 | prev = currentPrev; | |
214 | } | |
215 | } | |
216 | } | |
217 | if (queue) | |
218 | { | |
219 | tag = queue->tag; | |
220 | ||
221 | if (prev) | |
222 | prev->next = queue->next; | |
223 | else | |
224 | hpqueue = queue->next; | |
225 | HeapFree(GetProcessHeap(), 0, queue); | |
226 | } | |
227 | ||
228 | TRACE("%x got tag %d key %d\n", hPQ, tag, key); | |
229 | ||
230 | return tag; | |
231 | } | |
232 | ||
233 | /********************************************************************** | |
234 | * InsertPQ (GDI.233) | |
235 | * | |
236 | */ | |
237 | INT16 WINAPI InsertPQ16(HPQ16 hPQ, INT16 tag, INT16 key) | |
238 | { | |
239 | struct hpq *queueItem = HeapAlloc(GetProcessHeap(), 0, sizeof(struct hpq)); | |
240 | if(queueItem == NULL) { | |
241 | ERR("Memory exausted!\n"); | |
242 | return FALSE; | |
243 | } | |
244 | queueItem->next = hpqueue; | |
245 | hpqueue = queueItem; | |
246 | queueItem->key = key; | |
247 | queueItem->tag = tag; | |
248 | ||
249 | FIXME("(%x %d %d): stub???\n", hPQ, tag, key); | |
250 | return TRUE; | |
251 | } | |
252 | ||
253 | /********************************************************************** | |
254 | * MinPQ (GDI.231) | |
255 | * | |
256 | */ | |
257 | INT16 WINAPI MinPQ16(HPQ16 hPQ) | |
258 | { | |
259 | FIXME("(%x): stub\n", hPQ); | |
260 | return 0; | |
261 | } | |
262 | ||
263 | /********************************************************************** | |
264 | * SizePQ (GDI.234) | |
265 | * | |
266 | */ | |
267 | INT16 WINAPI SizePQ16(HPQ16 hPQ, INT16 sizechange) | |
268 | { | |
269 | FIXME("(%x %d): stub\n", hPQ, sizechange); | |
270 | return -1; | |
271 | } | |
272 | ||
273 | ||
274 | ||
275 | /* | |
276 | * The following functions implement part of the spooling process to | |
277 | * print manager. I would like to see wine have a version of print managers | |
278 | * that used LPR/LPD. For simplicity print jobs will be sent to a file for | |
279 | * now. | |
280 | */ | |
281 | typedef struct PRINTJOB | |
282 | { | |
283 | char *pszOutput; | |
284 | char *pszTitle; | |
285 | HDC16 hDC; | |
286 | HANDLE16 hHandle; | |
287 | int nIndex; | |
288 | int fd; | |
289 | } PRINTJOB, *PPRINTJOB; | |
290 | ||
291 | #define MAX_PRINT_JOBS 1 | |
292 | #define SP_OK 1 | |
293 | ||
294 | static PPRINTJOB gPrintJobsTable[MAX_PRINT_JOBS]; | |
295 | ||
296 | ||
297 | static PPRINTJOB FindPrintJobFromHandle(HANDLE16 hHandle) | |
298 | { | |
299 | return gPrintJobsTable[0]; | |
300 | } | |
301 | ||
302 | static int CreateSpoolFile(LPCSTR pszOutput) | |
303 | { | |
304 | int fd=-1; | |
305 | char psCmd[1024]; | |
306 | const char *psCmdP = psCmd; | |
307 | HKEY hkey; | |
308 | ||
309 | /* TTD convert the 'output device' into a spool file name */ | |
310 | ||
311 | if (pszOutput == NULL || *pszOutput == '\0') | |
312 | return -1; | |
313 | ||
314 | psCmd[0] = 0; | |
315 | /* @@ Wine registry key: HKCU\Software\Wine\Printing\Spooler */ | |
316 | if(!RegOpenKeyA(HKEY_CURRENT_USER, "Software\\Wine\\Printing\\Spooler", &hkey)) | |
317 | { | |
318 | DWORD type, count = sizeof(psCmd); | |
319 | RegQueryValueExA(hkey, pszOutput, 0, &type, (LPBYTE)psCmd, &count); | |
320 | RegCloseKey(hkey); | |
321 | } | |
322 | if (!psCmd[0] && !strncmp("LPR:",pszOutput,4)) | |
323 | sprintf(psCmd,"|lpr -P'%s'",pszOutput+4); | |
324 | ||
325 | TRACE("Got printerSpoolCommand '%s' for output device '%s'\n", | |
326 | psCmd, pszOutput); | |
327 | if (!*psCmd) | |
328 | psCmdP = pszOutput; | |
329 | else | |
330 | { | |
331 | while (*psCmdP && isspace(*psCmdP)) | |
332 | { | |
333 | psCmdP++; | |
334 | } | |
335 | if (!*psCmdP) | |
336 | return -1; | |
337 | } | |
338 | TRACE("command: '%s'\n", psCmdP); | |
339 | #ifdef HAVE_FORK | |
340 | if (*psCmdP == '|') | |
341 | { | |
342 | int fds[2]; | |
343 | if (pipe(fds)) { | |
344 | ERR("pipe() failed!\n"); | |
345 | return -1; | |
346 | } | |
347 | if (fork() == 0) | |
348 | { | |
349 | psCmdP++; | |
350 | ||
351 | TRACE("In child need to exec %s\n",psCmdP); | |
352 | close(0); | |
353 | dup2(fds[0],0); | |
354 | close (fds[1]); | |
355 | ||
356 | /* reset signals that we previously set to SIG_IGN */ | |
357 | signal( SIGPIPE, SIG_DFL ); | |
358 | signal( SIGCHLD, SIG_DFL ); | |
359 | ||
ec1c9492 | 360 | execl("/bin/sh", "/bin/sh", "-c", psCmdP, NULL); |
de75fe0a RS |
361 | _exit(1); |
362 | ||
363 | } | |
364 | close (fds[0]); | |
365 | fd = fds[1]; | |
366 | TRACE("Need to execute a cmnd and pipe the output to it\n"); | |
367 | } | |
368 | else | |
369 | #endif | |
370 | { | |
371 | char *buffer; | |
372 | WCHAR psCmdPW[MAX_PATH]; | |
373 | ||
374 | TRACE("Just assume it's a file\n"); | |
375 | ||
376 | /** | |
377 | * The file name can be dos based, we have to find its | |
45b944ed | 378 | * corresponding Unix file name. |
de75fe0a RS |
379 | */ |
380 | MultiByteToWideChar(CP_ACP, 0, psCmdP, -1, psCmdPW, MAX_PATH); | |
381 | if ((buffer = wine_get_unix_file_name(psCmdPW))) | |
382 | { | |
383 | if ((fd = open(buffer, O_CREAT | O_TRUNC | O_WRONLY, 0666)) < 0) | |
384 | { | |
385 | ERR("Failed to create spool file '%s' ('%s'). (error %s)\n", | |
386 | buffer, psCmdP, strerror(errno)); | |
387 | } | |
388 | HeapFree(GetProcessHeap(), 0, buffer); | |
389 | } | |
390 | } | |
391 | return fd; | |
392 | } | |
393 | ||
394 | static int FreePrintJob(HANDLE16 hJob) | |
395 | { | |
396 | int nRet = SP_ERROR; | |
397 | PPRINTJOB pPrintJob; | |
398 | ||
399 | pPrintJob = FindPrintJobFromHandle(hJob); | |
400 | if (pPrintJob != NULL) | |
401 | { | |
402 | gPrintJobsTable[pPrintJob->nIndex] = NULL; | |
403 | HeapFree(GetProcessHeap(), 0, pPrintJob->pszOutput); | |
404 | HeapFree(GetProcessHeap(), 0, pPrintJob->pszTitle); | |
405 | if (pPrintJob->fd >= 0) close(pPrintJob->fd); | |
406 | HeapFree(GetProcessHeap(), 0, pPrintJob); | |
407 | nRet = SP_OK; | |
408 | } | |
409 | return nRet; | |
410 | } | |
411 | ||
412 | /********************************************************************** | |
413 | * OpenJob (GDI.240) | |
414 | * | |
415 | */ | |
416 | HPJOB16 WINAPI OpenJob16(LPCSTR lpOutput, LPCSTR lpTitle, HDC16 hDC) | |
417 | { | |
418 | HPJOB16 hHandle = (HPJOB16)SP_ERROR; | |
419 | PPRINTJOB pPrintJob; | |
420 | ||
421 | TRACE("'%s' '%s' %04x\n", lpOutput, lpTitle, hDC); | |
422 | ||
423 | pPrintJob = gPrintJobsTable[0]; | |
424 | if (pPrintJob == NULL) | |
425 | { | |
426 | int fd; | |
427 | ||
428 | /* Try and create a spool file */ | |
429 | fd = CreateSpoolFile(lpOutput); | |
430 | if (fd >= 0) | |
431 | { | |
432 | pPrintJob = HeapAlloc(GetProcessHeap(), 0, sizeof(PRINTJOB)); | |
433 | if(pPrintJob == NULL) { | |
434 | WARN("Memory exausted!\n"); | |
435 | return hHandle; | |
436 | } | |
437 | ||
438 | hHandle = 1; | |
439 | ||
440 | pPrintJob->pszOutput = HeapAlloc(GetProcessHeap(), 0, strlen(lpOutput)+1); | |
441 | strcpy( pPrintJob->pszOutput, lpOutput ); | |
442 | if(lpTitle) | |
443 | { | |
444 | pPrintJob->pszTitle = HeapAlloc(GetProcessHeap(), 0, strlen(lpTitle)+1); | |
445 | strcpy( pPrintJob->pszTitle, lpTitle ); | |
446 | } | |
447 | pPrintJob->hDC = hDC; | |
448 | pPrintJob->fd = fd; | |
449 | pPrintJob->nIndex = 0; | |
450 | pPrintJob->hHandle = hHandle; | |
451 | gPrintJobsTable[pPrintJob->nIndex] = pPrintJob; | |
452 | } | |
453 | } | |
454 | TRACE("return %04x\n", hHandle); | |
455 | return hHandle; | |
456 | } | |
457 | ||
458 | /********************************************************************** | |
459 | * CloseJob (GDI.243) | |
460 | * | |
461 | */ | |
462 | INT16 WINAPI CloseJob16(HPJOB16 hJob) | |
463 | { | |
464 | int nRet = SP_ERROR; | |
465 | PPRINTJOB pPrintJob = NULL; | |
466 | ||
467 | TRACE("%04x\n", hJob); | |
468 | ||
469 | pPrintJob = FindPrintJobFromHandle(hJob); | |
470 | if (pPrintJob != NULL) | |
471 | { | |
472 | /* Close the spool file */ | |
473 | close(pPrintJob->fd); | |
474 | FreePrintJob(hJob); | |
475 | nRet = 1; | |
476 | } | |
477 | return nRet; | |
478 | } | |
479 | ||
480 | /********************************************************************** | |
481 | * WriteSpool (GDI.241) | |
482 | * | |
483 | */ | |
484 | INT16 WINAPI WriteSpool16(HPJOB16 hJob, LPSTR lpData, INT16 cch) | |
485 | { | |
486 | int nRet = SP_ERROR; | |
487 | PPRINTJOB pPrintJob = NULL; | |
488 | ||
489 | TRACE("%04x %p %04x\n", hJob, lpData, cch); | |
490 | ||
491 | pPrintJob = FindPrintJobFromHandle(hJob); | |
492 | if (pPrintJob != NULL && pPrintJob->fd >= 0 && cch) | |
493 | { | |
494 | if (write(pPrintJob->fd, lpData, cch) != cch) | |
495 | nRet = SP_OUTOFDISK; | |
496 | else | |
497 | nRet = cch; | |
498 | #if 0 | |
499 | /* FIXME: We just cannot call 16 bit functions from here, since we | |
500 | * have acquired several locks (DC). And we do not really need to. | |
501 | */ | |
502 | if (pPrintJob->hDC == 0) { | |
503 | TRACE("hDC == 0 so no QueryAbort\n"); | |
504 | } | |
505 | else if (!(QueryAbort16(pPrintJob->hDC, (nRet == SP_OUTOFDISK) ? nRet : 0 ))) | |
506 | { | |
507 | CloseJob16(hJob); /* printing aborted */ | |
508 | nRet = SP_APPABORT; | |
509 | } | |
510 | #endif | |
511 | } | |
512 | return nRet; | |
513 | } | |
514 | ||
515 | typedef INT (WINAPI *MSGBOX_PROC)( HWND, LPCSTR, LPCSTR, UINT ); | |
516 | ||
517 | /********************************************************************** | |
518 | * WriteDialog (GDI.242) | |
519 | * | |
520 | */ | |
521 | INT16 WINAPI WriteDialog16(HPJOB16 hJob, LPSTR lpMsg, INT16 cchMsg) | |
522 | { | |
523 | HMODULE mod; | |
524 | MSGBOX_PROC pMessageBoxA; | |
525 | INT16 ret = 0; | |
526 | ||
527 | TRACE("%04x %04x '%s'\n", hJob, cchMsg, lpMsg); | |
528 | ||
529 | if ((mod = GetModuleHandleA("user32.dll"))) | |
530 | { | |
531 | if ((pMessageBoxA = (MSGBOX_PROC)GetProcAddress( mod, "MessageBoxA" ))) | |
532 | ret = pMessageBoxA(0, lpMsg, "Printing Error", MB_OKCANCEL); | |
533 | } | |
534 | return ret; | |
535 | } | |
536 | ||
537 | ||
538 | /********************************************************************** | |
539 | * DeleteJob (GDI.244) | |
540 | * | |
541 | */ | |
542 | INT16 WINAPI DeleteJob16(HPJOB16 hJob, INT16 nNotUsed) | |
543 | { | |
544 | int nRet; | |
545 | ||
546 | TRACE("%04x\n", hJob); | |
547 | ||
548 | nRet = FreePrintJob(hJob); | |
549 | return nRet; | |
550 | } | |
551 | ||
552 | /* | |
553 | * The following two function would allow a page to be sent to the printer | |
554 | * when it has been processed. For simplicity they haven't been implemented. | |
555 | * This means a whole job has to be processed before it is sent to the printer. | |
556 | */ | |
557 | ||
558 | /********************************************************************** | |
559 | * StartSpoolPage (GDI.246) | |
560 | * | |
561 | */ | |
562 | INT16 WINAPI StartSpoolPage16(HPJOB16 hJob) | |
563 | { | |
564 | FIXME("StartSpoolPage GDI.246 unimplemented\n"); | |
565 | return 1; | |
566 | ||
567 | } | |
568 | ||
569 | ||
570 | /********************************************************************** | |
571 | * EndSpoolPage (GDI.247) | |
572 | * | |
573 | */ | |
574 | INT16 WINAPI EndSpoolPage16(HPJOB16 hJob) | |
575 | { | |
576 | FIXME("EndSpoolPage GDI.247 unimplemented\n"); | |
577 | return 1; | |
578 | } | |
579 | ||
580 | ||
581 | /********************************************************************** | |
582 | * GetSpoolJob (GDI.245) | |
583 | * | |
584 | */ | |
585 | DWORD WINAPI GetSpoolJob16(int nOption, LONG param) | |
586 | { | |
587 | DWORD retval = 0; | |
588 | TRACE("In GetSpoolJob param 0x%x noption %d\n",param, nOption); | |
589 | return retval; | |
590 | } | |
591 | ||
592 | ||
593 | /****************************************************************** | |
594 | * DrvGetPrinterDataInternal | |
595 | * | |
596 | * Helper for DrvGetPrinterData | |
597 | */ | |
598 | static DWORD DrvGetPrinterDataInternal(LPSTR RegStr_Printer, | |
599 | LPBYTE lpPrinterData, int cbData, int what) | |
600 | { | |
601 | DWORD res = -1; | |
602 | HKEY hkey; | |
603 | DWORD dwType, cbQueryData; | |
604 | ||
605 | if (!(RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey))) { | |
606 | if (what == INT_PD_DEFAULT_DEVMODE) { /* "Default DevMode" */ | |
607 | if (!(RegQueryValueExA(hkey, DefaultDevMode, 0, &dwType, 0, &cbQueryData))) { | |
608 | if (!lpPrinterData) | |
609 | res = cbQueryData; | |
610 | else if ((cbQueryData) && (cbQueryData <= cbData)) { | |
611 | cbQueryData = cbData; | |
612 | if (RegQueryValueExA(hkey, DefaultDevMode, 0, | |
613 | &dwType, lpPrinterData, &cbQueryData)) | |
614 | res = cbQueryData; | |
615 | } | |
616 | } | |
617 | } else { /* "Printer Driver" */ | |
618 | cbQueryData = 32; | |
619 | RegQueryValueExA(hkey, "Printer Driver", 0, | |
620 | &dwType, lpPrinterData, &cbQueryData); | |
621 | res = cbQueryData; | |
622 | } | |
623 | } | |
624 | if (hkey) RegCloseKey(hkey); | |
625 | return res; | |
626 | } | |
627 | ||
628 | /****************************************************************** | |
629 | * DrvGetPrinterData (GDI.282) | |
630 | * | |
631 | */ | |
632 | DWORD WINAPI DrvGetPrinterData16(LPSTR lpPrinter, LPSTR lpProfile, | |
633 | LPDWORD lpType, LPBYTE lpPrinterData, | |
634 | int cbData, LPDWORD lpNeeded) | |
635 | { | |
636 | LPSTR RegStr_Printer; | |
637 | HKEY hkey = 0, hkey2 = 0; | |
638 | DWORD res = 0; | |
639 | DWORD dwType, PrinterAttr, cbPrinterAttr, SetData, size; | |
640 | ||
641 | if (HIWORD(lpPrinter)) | |
642 | TRACE("printer %s\n",lpPrinter); | |
643 | else | |
644 | TRACE("printer %p\n",lpPrinter); | |
645 | if (HIWORD(lpProfile)) | |
646 | TRACE("profile %s\n",lpProfile); | |
647 | else | |
648 | TRACE("profile %p\n",lpProfile); | |
649 | TRACE("lpType %p\n",lpType); | |
650 | ||
651 | if ((!lpPrinter) || (!lpProfile) || (!lpNeeded)) | |
652 | return ERROR_INVALID_PARAMETER; | |
653 | ||
654 | RegStr_Printer = HeapAlloc(GetProcessHeap(), 0, | |
655 | strlen(Printers) + strlen(lpPrinter) + 2); | |
656 | strcpy(RegStr_Printer, Printers); | |
657 | strcat(RegStr_Printer, lpPrinter); | |
658 | ||
659 | if ((PtrToUlong(lpProfile) == INT_PD_DEFAULT_DEVMODE) || (HIWORD(lpProfile) && | |
660 | (!strcmp(lpProfile, DefaultDevMode)))) { | |
661 | size = DrvGetPrinterDataInternal(RegStr_Printer, lpPrinterData, cbData, | |
662 | INT_PD_DEFAULT_DEVMODE); | |
663 | if (size+1) { | |
664 | *lpNeeded = size; | |
665 | if ((lpPrinterData) && (*lpNeeded > cbData)) | |
666 | res = ERROR_MORE_DATA; | |
667 | } | |
668 | else res = ERROR_INVALID_PRINTER_NAME; | |
669 | } | |
670 | else | |
671 | if ((PtrToUlong(lpProfile) == INT_PD_DEFAULT_MODEL) || (HIWORD(lpProfile) && | |
672 | (!strcmp(lpProfile, PrinterModel)))) { | |
673 | *lpNeeded = 32; | |
674 | if (!lpPrinterData) goto failed; | |
675 | if (cbData < 32) { | |
676 | res = ERROR_MORE_DATA; | |
677 | goto failed; | |
678 | } | |
679 | size = DrvGetPrinterDataInternal(RegStr_Printer, lpPrinterData, cbData, | |
680 | INT_PD_DEFAULT_MODEL); | |
681 | if ((size+1) && (lpType)) | |
682 | *lpType = REG_SZ; | |
683 | else | |
684 | res = ERROR_INVALID_PRINTER_NAME; | |
685 | } | |
686 | else | |
687 | { | |
688 | if ((res = RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey))) | |
689 | goto failed; | |
690 | cbPrinterAttr = 4; | |
691 | if ((res = RegQueryValueExA(hkey, "Attributes", 0, | |
692 | &dwType, (LPBYTE)&PrinterAttr, &cbPrinterAttr))) | |
693 | goto failed; | |
694 | if ((res = RegOpenKeyA(hkey, PrinterDriverData, &hkey2))) | |
695 | goto failed; | |
696 | *lpNeeded = cbData; | |
697 | res = RegQueryValueExA(hkey2, lpProfile, 0, | |
698 | lpType, lpPrinterData, lpNeeded); | |
699 | if ((res != ERROR_CANTREAD) && | |
700 | ((PrinterAttr & | |
701 | (PRINTER_ATTRIBUTE_ENABLE_BIDI|PRINTER_ATTRIBUTE_NETWORK)) | |
702 | == PRINTER_ATTRIBUTE_NETWORK)) | |
703 | { | |
704 | if (!(res) && (*lpType == REG_DWORD) && (*(LPDWORD)lpPrinterData == -1)) | |
705 | res = ERROR_INVALID_DATA; | |
706 | } | |
707 | else | |
708 | { | |
709 | SetData = -1; | |
710 | RegSetValueExA(hkey2, lpProfile, 0, REG_DWORD, (LPBYTE)&SetData, 4); /* no result returned */ | |
711 | } | |
712 | } | |
713 | ||
714 | failed: | |
715 | if (hkey2) RegCloseKey(hkey2); | |
716 | if (hkey) RegCloseKey(hkey); | |
717 | HeapFree(GetProcessHeap(), 0, RegStr_Printer); | |
718 | return res; | |
719 | } | |
720 | ||
721 | ||
722 | /****************************************************************** | |
723 | * DrvSetPrinterData (GDI.281) | |
724 | * | |
725 | */ | |
726 | DWORD WINAPI DrvSetPrinterData16(LPSTR lpPrinter, LPSTR lpProfile, | |
727 | DWORD lpType, LPBYTE lpPrinterData, | |
728 | DWORD dwSize) | |
729 | { | |
730 | LPSTR RegStr_Printer; | |
731 | HKEY hkey = 0; | |
732 | DWORD res = 0; | |
733 | ||
734 | if (HIWORD(lpPrinter)) | |
735 | TRACE("printer %s\n",lpPrinter); | |
736 | else | |
737 | TRACE("printer %p\n",lpPrinter); | |
738 | if (HIWORD(lpProfile)) | |
739 | TRACE("profile %s\n",lpProfile); | |
740 | else | |
741 | TRACE("profile %p\n",lpProfile); | |
742 | TRACE("lpType %08x\n",lpType); | |
743 | ||
744 | if ((!lpPrinter) || (!lpProfile) || | |
745 | (PtrToUlong(lpProfile) == INT_PD_DEFAULT_MODEL) || (HIWORD(lpProfile) && | |
746 | (!strcmp(lpProfile, PrinterModel)))) | |
747 | return ERROR_INVALID_PARAMETER; | |
748 | ||
749 | RegStr_Printer = HeapAlloc(GetProcessHeap(), 0, | |
750 | strlen(Printers) + strlen(lpPrinter) + 2); | |
751 | strcpy(RegStr_Printer, Printers); | |
752 | strcat(RegStr_Printer, lpPrinter); | |
753 | ||
754 | if ((PtrToUlong(lpProfile) == INT_PD_DEFAULT_DEVMODE) || (HIWORD(lpProfile) && | |
755 | (!strcmp(lpProfile, DefaultDevMode)))) { | |
756 | if ( RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey) | |
757 | != ERROR_SUCCESS || | |
758 | RegSetValueExA(hkey, DefaultDevMode, 0, REG_BINARY, | |
759 | lpPrinterData, dwSize) != ERROR_SUCCESS ) | |
760 | res = ERROR_INVALID_PRINTER_NAME; | |
761 | } | |
762 | else | |
763 | { | |
764 | strcat(RegStr_Printer, "\\"); | |
765 | ||
766 | if( (res = RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey)) == | |
767 | ERROR_SUCCESS ) { | |
768 | ||
769 | if (!lpPrinterData) | |
770 | res = RegDeleteValueA(hkey, lpProfile); | |
771 | else | |
772 | res = RegSetValueExA(hkey, lpProfile, 0, lpType, | |
773 | lpPrinterData, dwSize); | |
774 | } | |
775 | } | |
776 | ||
777 | if (hkey) RegCloseKey(hkey); | |
778 | HeapFree(GetProcessHeap(), 0, RegStr_Printer); | |
779 | return res; | |
780 | } |