Better defaults for heap and stack sizes.
[wine/wine64.git] / dlls / gdi / printdrv.c
blobfdd054e9033cdc9e6750fec26ebb3607982b0540
1 /*
2 * Implementation of some printer driver bits
4 * Copyright 1996 John Harvey
5 * Copyright 1998 Huw Davies
6 * Copyright 1998 Andreas Mohr
7 * Copyright 1999 Klaas van Gend
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.
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.
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
24 #include "config.h"
25 #include "wine/port.h"
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.h"
52 #include "heap.h"
53 #include "file.h"
55 WINE_DEFAULT_DEBUG_CHANNEL(print);
57 static char PrinterModel[] = "Printer Model";
58 static char DefaultDevMode[] = "Default DevMode";
59 static char PrinterDriverData[] = "PrinterDriverData";
60 static char Printers[] = "System\\CurrentControlSet\\Control\\Print\\Printers\\";
63 /******************************************************************
64 * StartDocA [GDI32.@]
66 * StartDoc calls the STARTDOC Escape with the input data pointing to DocName
67 * and the output data (which is used as a second input parameter).pointing at
68 * the whole docinfo structure. This seems to be an undocumented feature of
69 * the STARTDOC Escape.
71 * Note: we now do it the other way, with the STARTDOC Escape calling StartDoc.
73 INT WINAPI StartDocA(HDC hdc, const DOCINFOA* doc)
75 INT ret = 0;
76 DC *dc = DC_GetDCPtr( hdc );
78 TRACE("DocName = '%s' Output = '%s' Datatype = '%s'\n",
79 doc->lpszDocName, doc->lpszOutput, doc->lpszDatatype);
81 if(!dc) return SP_ERROR;
83 if (dc->funcs->pStartDoc) ret = dc->funcs->pStartDoc( dc->physDev, doc );
84 GDI_ReleaseObj( hdc );
85 return ret;
88 /*************************************************************************
89 * StartDocW [GDI32.@]
92 INT WINAPI StartDocW(HDC hdc, const DOCINFOW* doc)
94 DOCINFOA docA;
95 INT ret;
97 docA.cbSize = doc->cbSize;
98 docA.lpszDocName = doc->lpszDocName ?
99 HEAP_strdupWtoA( GetProcessHeap(), 0, doc->lpszDocName ) : NULL;
100 docA.lpszOutput = doc->lpszOutput ?
101 HEAP_strdupWtoA( GetProcessHeap(), 0, doc->lpszOutput ) : NULL;
102 docA.lpszDatatype = doc->lpszDatatype ?
103 HEAP_strdupWtoA( GetProcessHeap(), 0, doc->lpszDatatype ) : NULL;
104 docA.fwType = doc->fwType;
106 ret = StartDocA(hdc, &docA);
108 if(docA.lpszDocName)
109 HeapFree( GetProcessHeap(), 0, (LPSTR)docA.lpszDocName );
110 if(docA.lpszOutput)
111 HeapFree( GetProcessHeap(), 0, (LPSTR)docA.lpszOutput );
112 if(docA.lpszDatatype)
113 HeapFree( GetProcessHeap(), 0, (LPSTR)docA.lpszDatatype );
115 return ret;
119 /******************************************************************
120 * EndDoc [GDI32.@]
123 INT WINAPI EndDoc(HDC hdc)
125 INT ret = 0;
126 DC *dc = DC_GetDCPtr( hdc );
127 if(!dc) return SP_ERROR;
129 if (dc->funcs->pEndDoc) ret = dc->funcs->pEndDoc( dc->physDev );
130 GDI_ReleaseObj( hdc );
131 return ret;
135 /******************************************************************
136 * StartPage [GDI32.@]
139 INT WINAPI StartPage(HDC hdc)
141 INT ret = 1;
142 DC *dc = DC_GetDCPtr( hdc );
143 if(!dc) return SP_ERROR;
145 if(dc->funcs->pStartPage)
146 ret = dc->funcs->pStartPage( dc->physDev );
147 else
148 FIXME("stub\n");
149 GDI_ReleaseObj( hdc );
150 return ret;
154 /******************************************************************
155 * EndPage [GDI32.@]
158 INT WINAPI EndPage(HDC hdc)
160 ABORTPROC abort_proc;
161 INT ret = 0;
162 DC *dc = DC_GetDCPtr( hdc );
163 if(!dc) return SP_ERROR;
165 if (dc->funcs->pEndPage) ret = dc->funcs->pEndPage( dc->physDev );
166 abort_proc = dc->pAbortProc;
167 GDI_ReleaseObj( hdc );
168 if (abort_proc && !abort_proc( hdc, 0 ))
170 EndDoc( hdc );
171 ret = 0;
173 return ret;
177 /******************************************************************************
178 * AbortDoc [GDI32.@]
180 INT WINAPI AbortDoc(HDC hdc)
182 INT ret = 0;
183 DC *dc = DC_GetDCPtr( hdc );
184 if(!dc) return SP_ERROR;
186 if (dc->funcs->pAbortDoc) ret = dc->funcs->pAbortDoc( dc->physDev );
187 GDI_ReleaseObj( hdc );
188 return ret;
191 /**********************************************************************
192 * QueryAbort (GDI.155)
194 * Calls the app's AbortProc function if avail.
196 * RETURNS
197 * TRUE if no AbortProc avail or AbortProc wants to continue printing.
198 * FALSE if AbortProc wants to abort printing.
200 BOOL16 WINAPI QueryAbort16(HDC16 hdc16, INT16 reserved)
202 BOOL ret = TRUE;
203 HDC hdc = HDC_32( hdc16 );
204 DC *dc = DC_GetDCPtr( hdc );
205 ABORTPROC abproc;
207 if(!dc) {
208 ERR("Invalid hdc %p\n", hdc);
209 return FALSE;
212 abproc = dc->pAbortProc;
213 GDI_ReleaseObj( hdc );
215 if (abproc)
216 ret = abproc(hdc, 0);
217 return ret;
221 /**********************************************************************
222 * call_abort_proc16
224 static BOOL CALLBACK call_abort_proc16( HDC hdc, INT code )
226 ABORTPROC16 proc16;
227 DC *dc = DC_GetDCPtr( hdc );
229 if (!dc) return FALSE;
230 proc16 = dc->pAbortProc16;
231 GDI_ReleaseObj( hdc );
232 if (proc16)
234 WORD args[2];
235 DWORD ret;
237 args[1] = HDC_16(hdc);
238 args[0] = code;
239 WOWCallback16Ex( (DWORD)proc16, WCB16_PASCAL, sizeof(args), args, &ret );
240 return LOWORD(ret);
242 return TRUE;
246 /**********************************************************************
247 * SetAbortProc (GDI.381)
249 INT16 WINAPI SetAbortProc16(HDC16 hdc16, ABORTPROC16 abrtprc)
251 HDC hdc = HDC_32( hdc16 );
252 DC *dc = DC_GetDCPtr( hdc );
254 if (!dc) return FALSE;
255 dc->pAbortProc16 = abrtprc;
256 GDI_ReleaseObj( hdc );
257 return SetAbortProc( hdc, call_abort_proc16 );
260 /**********************************************************************
261 * SetAbortProc (GDI32.@)
264 INT WINAPI SetAbortProc(HDC hdc, ABORTPROC abrtprc)
266 DC *dc = DC_GetDCPtr( hdc );
268 if (!dc) return FALSE;
269 dc->pAbortProc = abrtprc;
270 GDI_ReleaseObj( hdc );
271 return TRUE;
275 /****************** misc. printer related functions */
278 * The following function should implement a queing system
280 struct hpq
282 struct hpq *next;
283 int tag;
284 int key;
287 static struct hpq *hpqueue;
289 /**********************************************************************
290 * CreatePQ (GDI.230)
293 HPQ16 WINAPI CreatePQ16(INT16 size)
295 #if 0
296 HGLOBAL16 hpq = 0;
297 WORD tmp_size;
298 LPWORD pPQ;
300 tmp_size = size << 2;
301 if (!(hpq = GlobalAlloc16(GMEM_SHARE|GMEM_MOVEABLE, tmp_size + 8)))
302 return 0xffff;
303 pPQ = GlobalLock16(hpq);
304 *pPQ++ = 0;
305 *pPQ++ = tmp_size;
306 *pPQ++ = 0;
307 *pPQ++ = 0;
308 GlobalUnlock16(hpq);
310 return (HPQ16)hpq;
311 #else
312 FIXME("(%d): stub\n",size);
313 return 1;
314 #endif
317 /**********************************************************************
318 * DeletePQ (GDI.235)
321 INT16 WINAPI DeletePQ16(HPQ16 hPQ)
323 return GlobalFree16((HGLOBAL16)hPQ);
326 /**********************************************************************
327 * ExtractPQ (GDI.232)
330 INT16 WINAPI ExtractPQ16(HPQ16 hPQ)
332 struct hpq *queue, *prev, *current, *currentPrev;
333 int key = 0, tag = -1;
334 currentPrev = prev = NULL;
335 queue = current = hpqueue;
336 if (current)
337 key = current->key;
339 while (current)
341 currentPrev = current;
342 current = current->next;
343 if (current)
345 if (current->key < key)
347 queue = current;
348 prev = currentPrev;
352 if (queue)
354 tag = queue->tag;
356 if (prev)
357 prev->next = queue->next;
358 else
359 hpqueue = queue->next;
360 HeapFree(GetProcessHeap(), 0, queue);
363 TRACE("%x got tag %d key %d\n", hPQ, tag, key);
365 return tag;
368 /**********************************************************************
369 * InsertPQ (GDI.233)
372 INT16 WINAPI InsertPQ16(HPQ16 hPQ, INT16 tag, INT16 key)
374 struct hpq *queueItem = HeapAlloc(GetProcessHeap(), 0, sizeof(struct hpq));
375 if(queueItem == NULL) {
376 ERR("Memory exausted!\n");
377 return FALSE;
379 queueItem->next = hpqueue;
380 hpqueue = queueItem;
381 queueItem->key = key;
382 queueItem->tag = tag;
384 FIXME("(%x %d %d): stub???\n", hPQ, tag, key);
385 return TRUE;
388 /**********************************************************************
389 * MinPQ (GDI.231)
392 INT16 WINAPI MinPQ16(HPQ16 hPQ)
394 FIXME("(%x): stub\n", hPQ);
395 return 0;
398 /**********************************************************************
399 * SizePQ (GDI.234)
402 INT16 WINAPI SizePQ16(HPQ16 hPQ, INT16 sizechange)
404 FIXME("(%x %d): stub\n", hPQ, sizechange);
405 return -1;
411 * The following functions implement part of the spooling process to
412 * print manager. I would like to see wine have a version of print managers
413 * that used LPR/LPD. For simplicity print jobs will be sent to a file for
414 * now.
416 typedef struct PRINTJOB
418 char *pszOutput;
419 char *pszTitle;
420 HDC16 hDC;
421 HANDLE16 hHandle;
422 int nIndex;
423 int fd;
424 } PRINTJOB, *PPRINTJOB;
426 #define MAX_PRINT_JOBS 1
427 #define SP_OK 1
429 PPRINTJOB gPrintJobsTable[MAX_PRINT_JOBS];
432 static PPRINTJOB FindPrintJobFromHandle(HANDLE16 hHandle)
434 return gPrintJobsTable[0];
437 static int CreateSpoolFile(LPCSTR pszOutput)
439 int fd=-1;
440 char psCmd[1024];
441 char *psCmdP = psCmd;
442 HKEY hkey;
444 /* TTD convert the 'output device' into a spool file name */
446 if (pszOutput == NULL || *pszOutput == '\0')
447 return -1;
449 psCmd[0] = 0;
450 if (!strncmp("LPR:",pszOutput,4))
452 if(!RegOpenKeyA(HKEY_LOCAL_MACHINE, "Software\\Wine\\Wine\\Config\\spooler", &hkey))
454 DWORD type, count = sizeof(psCmd);
455 RegQueryValueExA(hkey, pszOutput, 0, &type, psCmd, &count);
456 RegCloseKey(hkey);
458 if(psCmd[0] == 0)
459 sprintf(psCmd,"|lpr -P%s",pszOutput+4);
461 else
463 if(!RegOpenKeyA(HKEY_LOCAL_MACHINE, "Software\\Wine\\Wine\\Config\\spooler", &hkey))
465 DWORD type, count = sizeof(psCmd);
466 RegQueryValueExA(hkey, pszOutput, 0, &type, psCmd, &count);
467 RegCloseKey(hkey);
470 TRACE("Got printerSpoolCommand '%s' for output device '%s'\n",
471 psCmd, pszOutput);
472 if (!*psCmd)
473 psCmdP = (char *)pszOutput;
474 else
476 while (*psCmdP && isspace(*psCmdP))
478 psCmdP++;
480 if (!*psCmdP)
481 return -1;
483 TRACE("command: '%s'\n", psCmdP);
484 if (*psCmdP == '|')
486 int fds[2];
487 if (pipe(fds)) {
488 ERR("pipe() failed!\n");
489 return -1;
491 if (fork() == 0)
493 psCmdP++;
495 TRACE("In child need to exec %s\n",psCmdP);
496 close(0);
497 dup2(fds[0],0);
498 close (fds[1]);
500 /* reset signals that we previously set to SIG_IGN */
501 signal( SIGPIPE, SIG_DFL );
502 signal( SIGCHLD, SIG_DFL );
504 system(psCmdP);
505 exit(0);
508 close (fds[0]);
509 fd = fds[1];
510 TRACE("Need to execute a cmnd and pipe the output to it\n");
512 else
514 char buffer[MAX_PATH];
516 TRACE("Just assume it's a file\n");
519 * The file name can be dos based, we have to find its
520 * Unix correspondant file name
522 wine_get_unix_file_name(psCmdP, buffer, sizeof(buffer));
524 if ((fd = open(buffer, O_CREAT | O_TRUNC | O_WRONLY , 0600)) < 0)
526 ERR("Failed to create spool file '%s' ('%s'). (error %s)\n",
527 buffer, psCmdP, strerror(errno));
530 return fd;
533 static int FreePrintJob(HANDLE16 hJob)
535 int nRet = SP_ERROR;
536 PPRINTJOB pPrintJob;
538 pPrintJob = FindPrintJobFromHandle(hJob);
539 if (pPrintJob != NULL)
541 gPrintJobsTable[pPrintJob->nIndex] = NULL;
542 HeapFree(GetProcessHeap(), 0, pPrintJob->pszOutput);
543 HeapFree(GetProcessHeap(), 0, pPrintJob->pszTitle);
544 if (pPrintJob->fd >= 0) close(pPrintJob->fd);
545 HeapFree(GetProcessHeap(), 0, pPrintJob);
546 nRet = SP_OK;
548 return nRet;
551 /**********************************************************************
552 * OpenJob (GDI.240)
555 HPJOB16 WINAPI OpenJob16(LPCSTR lpOutput, LPCSTR lpTitle, HDC16 hDC)
557 HPJOB16 hHandle = (HPJOB16)SP_ERROR;
558 PPRINTJOB pPrintJob;
560 TRACE("'%s' '%s' %04x\n", lpOutput, lpTitle, hDC);
562 pPrintJob = gPrintJobsTable[0];
563 if (pPrintJob == NULL)
565 int fd;
567 /* Try and create a spool file */
568 fd = CreateSpoolFile(lpOutput);
569 if (fd >= 0)
571 pPrintJob = HeapAlloc(GetProcessHeap(), 0, sizeof(PRINTJOB));
572 if(pPrintJob == NULL) {
573 WARN("Memory exausted!\n");
574 return hHandle;
577 hHandle = 1;
579 pPrintJob->pszOutput = HeapAlloc(GetProcessHeap(), 0, strlen(lpOutput)+1);
580 strcpy( pPrintJob->pszOutput, lpOutput );
581 if(lpTitle)
583 pPrintJob->pszTitle = HeapAlloc(GetProcessHeap(), 0, strlen(lpTitle)+1);
584 strcpy( pPrintJob->pszTitle, lpTitle );
586 pPrintJob->hDC = hDC;
587 pPrintJob->fd = fd;
588 pPrintJob->nIndex = 0;
589 pPrintJob->hHandle = hHandle;
590 gPrintJobsTable[pPrintJob->nIndex] = pPrintJob;
593 TRACE("return %04x\n", hHandle);
594 return hHandle;
597 /**********************************************************************
598 * CloseJob (GDI.243)
601 INT16 WINAPI CloseJob16(HPJOB16 hJob)
603 int nRet = SP_ERROR;
604 PPRINTJOB pPrintJob = NULL;
606 TRACE("%04x\n", hJob);
608 pPrintJob = FindPrintJobFromHandle(hJob);
609 if (pPrintJob != NULL)
611 /* Close the spool file */
612 close(pPrintJob->fd);
613 FreePrintJob(hJob);
614 nRet = 1;
616 return nRet;
619 /**********************************************************************
620 * WriteSpool (GDI.241)
623 INT16 WINAPI WriteSpool16(HPJOB16 hJob, LPSTR lpData, INT16 cch)
625 int nRet = SP_ERROR;
626 PPRINTJOB pPrintJob = NULL;
628 TRACE("%04x %08lx %04x\n", hJob, (DWORD)lpData, cch);
630 pPrintJob = FindPrintJobFromHandle(hJob);
631 if (pPrintJob != NULL && pPrintJob->fd >= 0 && cch)
633 if (write(pPrintJob->fd, lpData, cch) != cch)
634 nRet = SP_OUTOFDISK;
635 else
636 nRet = cch;
637 #if 0
638 /* FIXME: We just cannot call 16 bit functions from here, since we
639 * have acquired several locks (DC). And we do not really need to.
641 if (pPrintJob->hDC == 0) {
642 TRACE("hDC == 0 so no QueryAbort\n");
644 else if (!(QueryAbort16(pPrintJob->hDC, (nRet == SP_OUTOFDISK) ? nRet : 0 )))
646 CloseJob16(hJob); /* printing aborted */
647 nRet = SP_APPABORT;
649 #endif
651 return nRet;
654 typedef INT (WINAPI *MSGBOX_PROC)( HWND, LPCSTR, LPCSTR, UINT );
656 /**********************************************************************
657 * WriteDialog (GDI.242)
660 INT16 WINAPI WriteDialog16(HPJOB16 hJob, LPSTR lpMsg, INT16 cchMsg)
662 HMODULE mod;
663 MSGBOX_PROC pMessageBoxA;
664 INT16 ret = 0;
666 TRACE("%04x %04x '%s'\n", hJob, cchMsg, lpMsg);
668 if ((mod = GetModuleHandleA("user32.dll")))
670 if ((pMessageBoxA = (MSGBOX_PROC)GetProcAddress( mod, "MessageBoxA" )))
671 ret = pMessageBoxA(0, lpMsg, "Printing Error", MB_OKCANCEL);
673 return ret;
677 /**********************************************************************
678 * DeleteJob (GDI.244)
681 INT16 WINAPI DeleteJob16(HPJOB16 hJob, INT16 nNotUsed)
683 int nRet;
685 TRACE("%04x\n", hJob);
687 nRet = FreePrintJob(hJob);
688 return nRet;
692 * The following two function would allow a page to be sent to the printer
693 * when it has been processed. For simplicity they havn't been implemented.
694 * This means a whole job has to be processed before it is sent to the printer.
697 /**********************************************************************
698 * StartSpoolPage (GDI.246)
701 INT16 WINAPI StartSpoolPage16(HPJOB16 hJob)
703 FIXME("StartSpoolPage GDI.246 unimplemented\n");
704 return 1;
709 /**********************************************************************
710 * EndSpoolPage (GDI.247)
713 INT16 WINAPI EndSpoolPage16(HPJOB16 hJob)
715 FIXME("EndSpoolPage GDI.247 unimplemented\n");
716 return 1;
720 /**********************************************************************
721 * GetSpoolJob (GDI.245)
724 DWORD WINAPI GetSpoolJob16(int nOption, LONG param)
726 DWORD retval = 0;
727 TRACE("In GetSpoolJob param 0x%lx noption %d\n",param, nOption);
728 return retval;
732 /******************************************************************
733 * DrvGetPrinterDataInternal
735 * Helper for DrvGetPrinterData
737 static DWORD DrvGetPrinterDataInternal(LPSTR RegStr_Printer,
738 LPBYTE lpPrinterData, int cbData, int what)
740 DWORD res = -1;
741 HKEY hkey;
742 DWORD dwType, cbQueryData;
744 if (!(RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey))) {
745 if (what == INT_PD_DEFAULT_DEVMODE) { /* "Default DevMode" */
746 if (!(RegQueryValueExA(hkey, DefaultDevMode, 0, &dwType, 0, &cbQueryData))) {
747 if (!lpPrinterData)
748 res = cbQueryData;
749 else if ((cbQueryData) && (cbQueryData <= cbData)) {
750 cbQueryData = cbData;
751 if (RegQueryValueExA(hkey, DefaultDevMode, 0,
752 &dwType, lpPrinterData, &cbQueryData))
753 res = cbQueryData;
756 } else { /* "Printer Driver" */
757 cbQueryData = 32;
758 RegQueryValueExA(hkey, "Printer Driver", 0,
759 &dwType, lpPrinterData, &cbQueryData);
760 res = cbQueryData;
763 if (hkey) RegCloseKey(hkey);
764 return res;
767 /******************************************************************
768 * DrvGetPrinterData (GDI.282)
771 DWORD WINAPI DrvGetPrinterData16(LPSTR lpPrinter, LPSTR lpProfile,
772 LPDWORD lpType, LPBYTE lpPrinterData,
773 int cbData, LPDWORD lpNeeded)
775 LPSTR RegStr_Printer;
776 HKEY hkey = 0, hkey2 = 0;
777 DWORD res = 0;
778 DWORD dwType, PrinterAttr, cbPrinterAttr, SetData, size;
780 if (HIWORD(lpPrinter))
781 TRACE("printer %s\n",lpPrinter);
782 else
783 TRACE("printer %p\n",lpPrinter);
784 if (HIWORD(lpProfile))
785 TRACE("profile %s\n",lpProfile);
786 else
787 TRACE("profile %p\n",lpProfile);
788 TRACE("lpType %p\n",lpType);
790 if ((!lpPrinter) || (!lpProfile) || (!lpNeeded))
791 return ERROR_INVALID_PARAMETER;
793 RegStr_Printer = HeapAlloc(GetProcessHeap(), 0,
794 strlen(Printers) + strlen(lpPrinter) + 2);
795 strcpy(RegStr_Printer, Printers);
796 strcat(RegStr_Printer, lpPrinter);
798 if (((DWORD)lpProfile == INT_PD_DEFAULT_DEVMODE) || (HIWORD(lpProfile) &&
799 (!strcmp(lpProfile, DefaultDevMode)))) {
800 size = DrvGetPrinterDataInternal(RegStr_Printer, lpPrinterData, cbData,
801 INT_PD_DEFAULT_DEVMODE);
802 if (size+1) {
803 *lpNeeded = size;
804 if ((lpPrinterData) && (*lpNeeded > cbData))
805 res = ERROR_MORE_DATA;
807 else res = ERROR_INVALID_PRINTER_NAME;
809 else
810 if (((DWORD)lpProfile == INT_PD_DEFAULT_MODEL) || (HIWORD(lpProfile) &&
811 (!strcmp(lpProfile, PrinterModel)))) {
812 *lpNeeded = 32;
813 if (!lpPrinterData) goto failed;
814 if (cbData < 32) {
815 res = ERROR_MORE_DATA;
816 goto failed;
818 size = DrvGetPrinterDataInternal(RegStr_Printer, lpPrinterData, cbData,
819 INT_PD_DEFAULT_MODEL);
820 if ((size+1) && (lpType))
821 *lpType = REG_SZ;
822 else
823 res = ERROR_INVALID_PRINTER_NAME;
825 else
827 if ((res = RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey)))
828 goto failed;
829 cbPrinterAttr = 4;
830 if ((res = RegQueryValueExA(hkey, "Attributes", 0,
831 &dwType, (LPBYTE)&PrinterAttr, &cbPrinterAttr)))
832 goto failed;
833 if ((res = RegOpenKeyA(hkey, PrinterDriverData, &hkey2)))
834 goto failed;
835 *lpNeeded = cbData;
836 res = RegQueryValueExA(hkey2, lpProfile, 0,
837 lpType, lpPrinterData, lpNeeded);
838 if ((res != ERROR_CANTREAD) &&
839 ((PrinterAttr &
840 (PRINTER_ATTRIBUTE_ENABLE_BIDI|PRINTER_ATTRIBUTE_NETWORK))
841 == PRINTER_ATTRIBUTE_NETWORK))
843 if (!(res) && (*lpType == REG_DWORD) && (*(LPDWORD)lpPrinterData == -1))
844 res = ERROR_INVALID_DATA;
846 else
848 SetData = -1;
849 RegSetValueExA(hkey2, lpProfile, 0, REG_DWORD, (LPBYTE)&SetData, 4); /* no result returned */
853 failed:
854 if (hkey2) RegCloseKey(hkey2);
855 if (hkey) RegCloseKey(hkey);
856 HeapFree(GetProcessHeap(), 0, RegStr_Printer);
857 return res;
861 /******************************************************************
862 * DrvSetPrinterData (GDI.281)
865 DWORD WINAPI DrvSetPrinterData16(LPSTR lpPrinter, LPSTR lpProfile,
866 DWORD lpType, LPBYTE lpPrinterData,
867 DWORD dwSize)
869 LPSTR RegStr_Printer;
870 HKEY hkey = 0;
871 DWORD res = 0;
873 if (HIWORD(lpPrinter))
874 TRACE("printer %s\n",lpPrinter);
875 else
876 TRACE("printer %p\n",lpPrinter);
877 if (HIWORD(lpProfile))
878 TRACE("profile %s\n",lpProfile);
879 else
880 TRACE("profile %p\n",lpProfile);
881 TRACE("lpType %08lx\n",lpType);
883 if ((!lpPrinter) || (!lpProfile) ||
884 ((DWORD)lpProfile == INT_PD_DEFAULT_MODEL) || (HIWORD(lpProfile) &&
885 (!strcmp(lpProfile, PrinterModel))))
886 return ERROR_INVALID_PARAMETER;
888 RegStr_Printer = HeapAlloc(GetProcessHeap(), 0,
889 strlen(Printers) + strlen(lpPrinter) + 2);
890 strcpy(RegStr_Printer, Printers);
891 strcat(RegStr_Printer, lpPrinter);
893 if (((DWORD)lpProfile == INT_PD_DEFAULT_DEVMODE) || (HIWORD(lpProfile) &&
894 (!strcmp(lpProfile, DefaultDevMode)))) {
895 if ( RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey)
896 != ERROR_SUCCESS ||
897 RegSetValueExA(hkey, DefaultDevMode, 0, REG_BINARY,
898 lpPrinterData, dwSize) != ERROR_SUCCESS )
899 res = ERROR_INVALID_PRINTER_NAME;
901 else
903 strcat(RegStr_Printer, "\\");
905 if( (res = RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey)) ==
906 ERROR_SUCCESS ) {
908 if (!lpPrinterData)
909 res = RegDeleteValueA(hkey, lpProfile);
910 else
911 res = RegSetValueExA(hkey, lpProfile, 0, lpType,
912 lpPrinterData, dwSize);
916 if (hkey) RegCloseKey(hkey);
917 HeapFree(GetProcessHeap(), 0, RegStr_Printer);
918 return res;