Proper finalization in StartServiceW.
[wine] / dlls / gdi / printdrv.c
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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.h"
52 #include "gdi_private.h"
53 #include "heap.h"
54
55 WINE_DEFAULT_DEBUG_CHANNEL(print);
56
57 static const char PrinterModel[]      = "Printer Model";
58 static const char DefaultDevMode[]    = "Default DevMode";
59 static const char PrinterDriverData[] = "PrinterDriverData";
60 static const char Printers[]          = "System\\CurrentControlSet\\Control\\Print\\Printers\\";
61
62
63 /******************************************************************
64  *                  StartDocA  [GDI32.@]
65  *
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.
70  *
71  * Note: we now do it the other way, with the STARTDOC Escape calling StartDoc.
72  */
73 INT WINAPI StartDocA(HDC hdc, const DOCINFOA* doc)
74 {
75     INT ret = 0;
76     DC *dc = DC_GetDCPtr( hdc );
77
78     TRACE("DocName = '%s' Output = '%s' Datatype = '%s'\n",
79           doc->lpszDocName, doc->lpszOutput, doc->lpszDatatype);
80
81     if(!dc) return SP_ERROR;
82
83     if (dc->funcs->pStartDoc) ret = dc->funcs->pStartDoc( dc->physDev, doc );
84     GDI_ReleaseObj( hdc );
85     return ret;
86 }
87
88 /*************************************************************************
89  *                  StartDocW [GDI32.@]
90  *
91  */
92 INT WINAPI StartDocW(HDC hdc, const DOCINFOW* doc)
93 {
94     DOCINFOA docA;
95     INT ret;
96
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;
105
106     ret = StartDocA(hdc, &docA);
107
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 );
114
115     return ret;
116 }
117
118
119 /******************************************************************
120  *                  EndDoc  [GDI32.@]
121  *
122  */
123 INT WINAPI EndDoc(HDC hdc)
124 {
125     INT ret = 0;
126     DC *dc = DC_GetDCPtr( hdc );
127     if(!dc) return SP_ERROR;
128
129     if (dc->funcs->pEndDoc) ret = dc->funcs->pEndDoc( dc->physDev );
130     GDI_ReleaseObj( hdc );
131     return ret;
132 }
133
134
135 /******************************************************************
136  *                  StartPage  [GDI32.@]
137  *
138  */
139 INT WINAPI StartPage(HDC hdc)
140 {
141     INT ret = 1;
142     DC *dc = DC_GetDCPtr( hdc );
143     if(!dc) return SP_ERROR;
144
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;
151 }
152
153
154 /******************************************************************
155  *                  EndPage  [GDI32.@]
156  *
157  */
158 INT WINAPI EndPage(HDC hdc)
159 {
160     ABORTPROC abort_proc;
161     INT ret = 0;
162     DC *dc = DC_GetDCPtr( hdc );
163     if(!dc) return SP_ERROR;
164
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 ))
169     {
170         EndDoc( hdc );
171         ret = 0;
172     }
173     return ret;
174 }
175
176
177 /******************************************************************************
178  *                 AbortDoc  [GDI32.@]
179  */
180 INT WINAPI AbortDoc(HDC hdc)
181 {
182     INT ret = 0;
183     DC *dc = DC_GetDCPtr( hdc );
184     if(!dc) return SP_ERROR;
185
186     if (dc->funcs->pAbortDoc) ret = dc->funcs->pAbortDoc( dc->physDev );
187     GDI_ReleaseObj( hdc );
188     return ret;
189 }
190
191 /**********************************************************************
192  *           QueryAbort   (GDI.155)
193  *
194  *  Calls the app's AbortProc function if avail.
195  *
196  * RETURNS
197  * TRUE if no AbortProc avail or AbortProc wants to continue printing.
198  * FALSE if AbortProc wants to abort printing.
199  */
200 BOOL16 WINAPI QueryAbort16(HDC16 hdc16, INT16 reserved)
201 {
202     BOOL ret = TRUE;
203     HDC hdc = HDC_32( hdc16 );
204     DC *dc = DC_GetDCPtr( hdc );
205     ABORTPROC abproc;
206
207     if(!dc) {
208         ERR("Invalid hdc %p\n", hdc);
209         return FALSE;
210     }
211
212     abproc = dc->pAbortProc;
213     GDI_ReleaseObj( hdc );
214
215     if (abproc)
216         ret = abproc(hdc, 0);
217     return ret;
218 }
219
220
221 /**********************************************************************
222  *           call_abort_proc16
223  */
224 static BOOL CALLBACK call_abort_proc16( HDC hdc, INT code )
225 {
226     ABORTPROC16 proc16;
227     DC *dc = DC_GetDCPtr( hdc );
228
229     if (!dc) return FALSE;
230     proc16 = dc->pAbortProc16;
231     GDI_ReleaseObj( hdc );
232     if (proc16)
233     {
234         WORD args[2];
235         DWORD ret;
236
237         args[1] = HDC_16(hdc);
238         args[0] = code;
239         WOWCallback16Ex( (DWORD)proc16, WCB16_PASCAL, sizeof(args), args, &ret );
240         return LOWORD(ret);
241     }
242     return TRUE;
243 }
244
245
246 /**********************************************************************
247  *           SetAbortProc   (GDI.381)
248  */
249 INT16 WINAPI SetAbortProc16(HDC16 hdc16, ABORTPROC16 abrtprc)
250 {
251     HDC hdc = HDC_32( hdc16 );
252     DC *dc = DC_GetDCPtr( hdc );
253
254     if (!dc) return FALSE;
255     dc->pAbortProc16 = abrtprc;
256     GDI_ReleaseObj( hdc );
257     return SetAbortProc( hdc, call_abort_proc16 );
258 }
259
260 /**********************************************************************
261  *           SetAbortProc   (GDI32.@)
262  *
263  */
264 INT WINAPI SetAbortProc(HDC hdc, ABORTPROC abrtprc)
265 {
266     DC *dc = DC_GetDCPtr( hdc );
267
268     if (!dc) return FALSE;
269     dc->pAbortProc = abrtprc;
270     GDI_ReleaseObj( hdc );
271     return TRUE;
272 }
273
274
275 /****************** misc. printer related functions */
276
277 /*
278  * The following function should implement a queing system
279  */
280 struct hpq
281 {
282     struct hpq  *next;
283     int          tag;
284     int          key;
285 };
286
287 static struct hpq *hpqueue;
288
289 /**********************************************************************
290  *           CreatePQ   (GDI.230)
291  *
292  */
293 HPQ16 WINAPI CreatePQ16(INT16 size)
294 {
295 #if 0
296     HGLOBAL16 hpq = 0;
297     WORD tmp_size;
298     LPWORD pPQ;
299
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);
309
310     return (HPQ16)hpq;
311 #else
312     FIXME("(%d): stub\n",size);
313     return 1;
314 #endif
315 }
316
317 /**********************************************************************
318  *           DeletePQ   (GDI.235)
319  *
320  */
321 INT16 WINAPI DeletePQ16(HPQ16 hPQ)
322 {
323     return GlobalFree16((HGLOBAL16)hPQ);
324 }
325
326 /**********************************************************************
327  *           ExtractPQ   (GDI.232)
328  *
329  */
330 INT16 WINAPI ExtractPQ16(HPQ16 hPQ)
331 {
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;
338
339     while (current)
340     {
341         currentPrev = current;
342         current = current->next;
343         if (current)
344         {
345             if (current->key < key)
346             {
347                 queue = current;
348                 prev = currentPrev;
349             }
350         }
351     }
352     if (queue)
353     {
354         tag = queue->tag;
355
356         if (prev)
357             prev->next = queue->next;
358         else
359             hpqueue = queue->next;
360         HeapFree(GetProcessHeap(), 0, queue);
361     }
362
363     TRACE("%x got tag %d key %d\n", hPQ, tag, key);
364
365     return tag;
366 }
367
368 /**********************************************************************
369  *           InsertPQ   (GDI.233)
370  *
371  */
372 INT16 WINAPI InsertPQ16(HPQ16 hPQ, INT16 tag, INT16 key)
373 {
374     struct hpq *queueItem = HeapAlloc(GetProcessHeap(), 0, sizeof(struct hpq));
375     if(queueItem == NULL) {
376         ERR("Memory exausted!\n");
377         return FALSE;
378     }
379     queueItem->next = hpqueue;
380     hpqueue = queueItem;
381     queueItem->key = key;
382     queueItem->tag = tag;
383
384     FIXME("(%x %d %d): stub???\n", hPQ, tag, key);
385     return TRUE;
386 }
387
388 /**********************************************************************
389  *           MinPQ   (GDI.231)
390  *
391  */
392 INT16 WINAPI MinPQ16(HPQ16 hPQ)
393 {
394     FIXME("(%x): stub\n", hPQ);
395     return 0;
396 }
397
398 /**********************************************************************
399  *           SizePQ   (GDI.234)
400  *
401  */
402 INT16 WINAPI SizePQ16(HPQ16 hPQ, INT16 sizechange)
403 {
404     FIXME("(%x %d): stub\n", hPQ, sizechange);
405     return -1;
406 }
407
408
409
410 /*
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.
415  */
416 typedef struct PRINTJOB
417 {
418     char        *pszOutput;
419     char        *pszTitle;
420     HDC16       hDC;
421     HANDLE16    hHandle;
422     int         nIndex;
423     int         fd;
424 } PRINTJOB, *PPRINTJOB;
425
426 #define MAX_PRINT_JOBS 1
427 #define SP_OK 1
428
429 PPRINTJOB gPrintJobsTable[MAX_PRINT_JOBS];
430
431
432 static PPRINTJOB FindPrintJobFromHandle(HANDLE16 hHandle)
433 {
434     return gPrintJobsTable[0];
435 }
436
437 static int CreateSpoolFile(LPCSTR pszOutput)
438 {
439     int fd=-1;
440     char psCmd[1024];
441     char *psCmdP = psCmd;
442     HKEY hkey;
443
444     /* TTD convert the 'output device' into a spool file name */
445
446     if (pszOutput == NULL || *pszOutput == '\0')
447       return -1;
448
449     psCmd[0] = 0;
450     if (!strncmp("LPR:",pszOutput,4))
451     {
452         if(!RegOpenKeyA(HKEY_LOCAL_MACHINE, "Software\\Wine\\Wine\\Config\\spooler", &hkey))
453         {
454             DWORD type, count = sizeof(psCmd);
455             RegQueryValueExA(hkey, pszOutput, 0, &type, psCmd, &count);
456             RegCloseKey(hkey);
457         }
458         if(psCmd[0] == 0)
459             sprintf(psCmd,"|lpr -P%s",pszOutput+4);
460     }
461     else
462     {
463         if(!RegOpenKeyA(HKEY_LOCAL_MACHINE, "Software\\Wine\\Wine\\Config\\spooler", &hkey))
464         {
465             DWORD type, count = sizeof(psCmd);
466             RegQueryValueExA(hkey, pszOutput, 0, &type, psCmd, &count);
467             RegCloseKey(hkey);
468         }
469     }
470     TRACE("Got printerSpoolCommand '%s' for output device '%s'\n",
471           psCmd, pszOutput);
472     if (!*psCmd)
473         psCmdP = (char *)pszOutput;
474     else
475     {
476         while (*psCmdP && isspace(*psCmdP))
477         {
478             psCmdP++;
479         }
480         if (!*psCmdP)
481             return -1;
482     }
483     TRACE("command: '%s'\n", psCmdP);
484     if (*psCmdP == '|')
485     {
486         int fds[2];
487         if (pipe(fds)) {
488             ERR("pipe() failed!\n"); 
489             return -1;
490         }
491         if (fork() == 0)
492         {
493             psCmdP++;
494
495             TRACE("In child need to exec %s\n",psCmdP);
496             close(0);
497             dup2(fds[0],0);
498             close (fds[1]);
499
500             /* reset signals that we previously set to SIG_IGN */
501             signal( SIGPIPE, SIG_DFL );
502             signal( SIGCHLD, SIG_DFL );
503
504             system(psCmdP);
505             exit(0);
506
507         }
508         close (fds[0]);
509         fd = fds[1];
510         TRACE("Need to execute a cmnd and pipe the output to it\n");
511     }
512     else
513     {
514         char *buffer;
515         WCHAR psCmdPW[MAX_PATH];
516
517         TRACE("Just assume it's a file\n");
518
519         /**
520          * The file name can be dos based, we have to find its
521          * Unix correspondant file name
522          */
523         MultiByteToWideChar(CP_ACP, 0, psCmdP, -1, psCmdPW, MAX_PATH);
524         if ((buffer = wine_get_unix_file_name(psCmdPW)))
525         {
526             if ((fd = open(buffer, O_CREAT | O_TRUNC | O_WRONLY , 0600)) < 0)
527             {
528                 ERR("Failed to create spool file '%s' ('%s'). (error %s)\n",
529                     buffer, psCmdP, strerror(errno));
530             }
531             HeapFree(GetProcessHeap(), 0, buffer);
532         }
533     }
534     return fd;
535 }
536
537 static int FreePrintJob(HANDLE16 hJob)
538 {
539     int nRet = SP_ERROR;
540     PPRINTJOB pPrintJob;
541
542     pPrintJob = FindPrintJobFromHandle(hJob);
543     if (pPrintJob != NULL)
544     {
545         gPrintJobsTable[pPrintJob->nIndex] = NULL;
546         HeapFree(GetProcessHeap(), 0, pPrintJob->pszOutput);
547         HeapFree(GetProcessHeap(), 0, pPrintJob->pszTitle);
548         if (pPrintJob->fd >= 0) close(pPrintJob->fd);
549         HeapFree(GetProcessHeap(), 0, pPrintJob);
550         nRet = SP_OK;
551     }
552     return nRet;
553 }
554
555 /**********************************************************************
556  *           OpenJob   (GDI.240)
557  *
558  */
559 HPJOB16 WINAPI OpenJob16(LPCSTR lpOutput, LPCSTR lpTitle, HDC16 hDC)
560 {
561     HPJOB16 hHandle = (HPJOB16)SP_ERROR;
562     PPRINTJOB pPrintJob;
563
564     TRACE("'%s' '%s' %04x\n", lpOutput, lpTitle, hDC);
565
566     pPrintJob = gPrintJobsTable[0];
567     if (pPrintJob == NULL)
568     {
569         int fd;
570
571         /* Try and create a spool file */
572         fd = CreateSpoolFile(lpOutput);
573         if (fd >= 0)
574         {
575             pPrintJob = HeapAlloc(GetProcessHeap(), 0, sizeof(PRINTJOB));
576             if(pPrintJob == NULL) {
577                 WARN("Memory exausted!\n");
578                 return hHandle;
579             }
580
581             hHandle = 1;
582
583             pPrintJob->pszOutput = HeapAlloc(GetProcessHeap(), 0, strlen(lpOutput)+1);
584             strcpy( pPrintJob->pszOutput, lpOutput );
585             if(lpTitle)
586             {
587                 pPrintJob->pszTitle = HeapAlloc(GetProcessHeap(), 0, strlen(lpTitle)+1);
588                 strcpy( pPrintJob->pszTitle, lpTitle );
589             }
590             pPrintJob->hDC = hDC;
591             pPrintJob->fd = fd;
592             pPrintJob->nIndex = 0;
593             pPrintJob->hHandle = hHandle;
594             gPrintJobsTable[pPrintJob->nIndex] = pPrintJob;
595         }
596     }
597     TRACE("return %04x\n", hHandle);
598     return hHandle;
599 }
600
601 /**********************************************************************
602  *           CloseJob   (GDI.243)
603  *
604  */
605 INT16 WINAPI CloseJob16(HPJOB16 hJob)
606 {
607     int nRet = SP_ERROR;
608     PPRINTJOB pPrintJob = NULL;
609
610     TRACE("%04x\n", hJob);
611
612     pPrintJob = FindPrintJobFromHandle(hJob);
613     if (pPrintJob != NULL)
614     {
615         /* Close the spool file */
616         close(pPrintJob->fd);
617         FreePrintJob(hJob);
618         nRet  = 1;
619     }
620     return nRet;
621 }
622
623 /**********************************************************************
624  *           WriteSpool   (GDI.241)
625  *
626  */
627 INT16 WINAPI WriteSpool16(HPJOB16 hJob, LPSTR lpData, INT16 cch)
628 {
629     int nRet = SP_ERROR;
630     PPRINTJOB pPrintJob = NULL;
631
632     TRACE("%04x %08lx %04x\n", hJob, (DWORD)lpData, cch);
633
634     pPrintJob = FindPrintJobFromHandle(hJob);
635     if (pPrintJob != NULL && pPrintJob->fd >= 0 && cch)
636     {
637         if (write(pPrintJob->fd, lpData, cch) != cch)
638           nRet = SP_OUTOFDISK;
639         else
640           nRet = cch;
641 #if 0
642         /* FIXME: We just cannot call 16 bit functions from here, since we
643          * have acquired several locks (DC). And we do not really need to.
644          */
645         if (pPrintJob->hDC == 0) {
646             TRACE("hDC == 0 so no QueryAbort\n");
647         }
648         else if (!(QueryAbort16(pPrintJob->hDC, (nRet == SP_OUTOFDISK) ? nRet : 0 )))
649         {
650             CloseJob16(hJob); /* printing aborted */
651             nRet = SP_APPABORT;
652         }
653 #endif
654     }
655     return nRet;
656 }
657
658 typedef INT (WINAPI *MSGBOX_PROC)( HWND, LPCSTR, LPCSTR, UINT );
659
660 /**********************************************************************
661  *           WriteDialog   (GDI.242)
662  *
663  */
664 INT16 WINAPI WriteDialog16(HPJOB16 hJob, LPSTR lpMsg, INT16 cchMsg)
665 {
666     HMODULE mod;
667     MSGBOX_PROC pMessageBoxA;
668     INT16 ret = 0;
669
670     TRACE("%04x %04x '%s'\n", hJob,  cchMsg, lpMsg);
671
672     if ((mod = GetModuleHandleA("user32.dll")))
673     {
674         if ((pMessageBoxA = (MSGBOX_PROC)GetProcAddress( mod, "MessageBoxA" )))
675             ret = pMessageBoxA(0, lpMsg, "Printing Error", MB_OKCANCEL);
676     }
677     return ret;
678 }
679
680
681 /**********************************************************************
682  *           DeleteJob  (GDI.244)
683  *
684  */
685 INT16 WINAPI DeleteJob16(HPJOB16 hJob, INT16 nNotUsed)
686 {
687     int nRet;
688
689     TRACE("%04x\n", hJob);
690
691     nRet = FreePrintJob(hJob);
692     return nRet;
693 }
694
695 /*
696  * The following two function would allow a page to be sent to the printer
697  * when it has been processed.  For simplicity they havn't been implemented.
698  * This means a whole job has to be processed before it is sent to the printer.
699  */
700
701 /**********************************************************************
702  *           StartSpoolPage   (GDI.246)
703  *
704  */
705 INT16 WINAPI StartSpoolPage16(HPJOB16 hJob)
706 {
707     FIXME("StartSpoolPage GDI.246 unimplemented\n");
708     return 1;
709
710 }
711
712
713 /**********************************************************************
714  *           EndSpoolPage   (GDI.247)
715  *
716  */
717 INT16 WINAPI EndSpoolPage16(HPJOB16 hJob)
718 {
719     FIXME("EndSpoolPage GDI.247 unimplemented\n");
720     return 1;
721 }
722
723
724 /**********************************************************************
725  *           GetSpoolJob   (GDI.245)
726  *
727  */
728 DWORD WINAPI GetSpoolJob16(int nOption, LONG param)
729 {
730     DWORD retval = 0;
731     TRACE("In GetSpoolJob param 0x%lx noption %d\n",param, nOption);
732     return retval;
733 }
734
735
736 /******************************************************************
737  *                  DrvGetPrinterDataInternal
738  *
739  * Helper for DrvGetPrinterData
740  */
741 static DWORD DrvGetPrinterDataInternal(LPSTR RegStr_Printer,
742 LPBYTE lpPrinterData, int cbData, int what)
743 {
744     DWORD res = -1;
745     HKEY hkey;
746     DWORD dwType, cbQueryData;
747
748     if (!(RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey))) {
749         if (what == INT_PD_DEFAULT_DEVMODE) { /* "Default DevMode" */
750             if (!(RegQueryValueExA(hkey, DefaultDevMode, 0, &dwType, 0, &cbQueryData))) {
751                 if (!lpPrinterData)
752                     res = cbQueryData;
753                 else if ((cbQueryData) && (cbQueryData <= cbData)) {
754                     cbQueryData = cbData;
755                     if (RegQueryValueExA(hkey, DefaultDevMode, 0,
756                                 &dwType, lpPrinterData, &cbQueryData))
757                         res = cbQueryData;
758                 }
759             }
760         } else { /* "Printer Driver" */
761             cbQueryData = 32;
762             RegQueryValueExA(hkey, "Printer Driver", 0,
763                         &dwType, lpPrinterData, &cbQueryData);
764             res = cbQueryData;
765         }
766     }
767     if (hkey) RegCloseKey(hkey);
768     return res;
769 }
770
771 /******************************************************************
772  *                DrvGetPrinterData     (GDI.282)
773  *
774  */
775 DWORD WINAPI DrvGetPrinterData16(LPSTR lpPrinter, LPSTR lpProfile,
776                                LPDWORD lpType, LPBYTE lpPrinterData,
777                                int cbData, LPDWORD lpNeeded)
778 {
779     LPSTR RegStr_Printer;
780     HKEY hkey = 0, hkey2 = 0;
781     DWORD res = 0;
782     DWORD dwType, PrinterAttr, cbPrinterAttr, SetData, size;
783
784     if (HIWORD(lpPrinter))
785             TRACE("printer %s\n",lpPrinter);
786     else
787             TRACE("printer %p\n",lpPrinter);
788     if (HIWORD(lpProfile))
789             TRACE("profile %s\n",lpProfile);
790     else
791             TRACE("profile %p\n",lpProfile);
792     TRACE("lpType %p\n",lpType);
793
794     if ((!lpPrinter) || (!lpProfile) || (!lpNeeded))
795         return ERROR_INVALID_PARAMETER;
796
797     RegStr_Printer = HeapAlloc(GetProcessHeap(), 0,
798                                strlen(Printers) + strlen(lpPrinter) + 2);
799     strcpy(RegStr_Printer, Printers);
800     strcat(RegStr_Printer, lpPrinter);
801
802     if (((DWORD)lpProfile == INT_PD_DEFAULT_DEVMODE) || (HIWORD(lpProfile) &&
803     (!strcmp(lpProfile, DefaultDevMode)))) {
804         size = DrvGetPrinterDataInternal(RegStr_Printer, lpPrinterData, cbData,
805                                          INT_PD_DEFAULT_DEVMODE);
806         if (size+1) {
807             *lpNeeded = size;
808             if ((lpPrinterData) && (*lpNeeded > cbData))
809                 res = ERROR_MORE_DATA;
810         }
811         else res = ERROR_INVALID_PRINTER_NAME;
812     }
813     else
814     if (((DWORD)lpProfile == INT_PD_DEFAULT_MODEL) || (HIWORD(lpProfile) &&
815     (!strcmp(lpProfile, PrinterModel)))) {
816         *lpNeeded = 32;
817         if (!lpPrinterData) goto failed;
818         if (cbData < 32) {
819             res = ERROR_MORE_DATA;
820             goto failed;
821         }
822         size = DrvGetPrinterDataInternal(RegStr_Printer, lpPrinterData, cbData,
823                                          INT_PD_DEFAULT_MODEL);
824         if ((size+1) && (lpType))
825             *lpType = REG_SZ;
826         else
827             res = ERROR_INVALID_PRINTER_NAME;
828     }
829     else
830     {
831         if ((res = RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey)))
832             goto failed;
833         cbPrinterAttr = 4;
834         if ((res = RegQueryValueExA(hkey, "Attributes", 0,
835                         &dwType, (LPBYTE)&PrinterAttr, &cbPrinterAttr)))
836             goto failed;
837         if ((res = RegOpenKeyA(hkey, PrinterDriverData, &hkey2)))
838             goto failed;
839         *lpNeeded = cbData;
840         res = RegQueryValueExA(hkey2, lpProfile, 0,
841                 lpType, lpPrinterData, lpNeeded);
842         if ((res != ERROR_CANTREAD) &&
843          ((PrinterAttr &
844         (PRINTER_ATTRIBUTE_ENABLE_BIDI|PRINTER_ATTRIBUTE_NETWORK))
845         == PRINTER_ATTRIBUTE_NETWORK))
846         {
847             if (!(res) && (*lpType == REG_DWORD) && (*(LPDWORD)lpPrinterData == -1))
848                 res = ERROR_INVALID_DATA;
849         }
850         else
851         {
852             SetData = -1;
853             RegSetValueExA(hkey2, lpProfile, 0, REG_DWORD, (LPBYTE)&SetData, 4); /* no result returned */
854         }
855     }
856
857 failed:
858     if (hkey2) RegCloseKey(hkey2);
859     if (hkey) RegCloseKey(hkey);
860     HeapFree(GetProcessHeap(), 0, RegStr_Printer);
861     return res;
862 }
863
864
865 /******************************************************************
866  *                 DrvSetPrinterData     (GDI.281)
867  *
868  */
869 DWORD WINAPI DrvSetPrinterData16(LPSTR lpPrinter, LPSTR lpProfile,
870                                DWORD lpType, LPBYTE lpPrinterData,
871                                DWORD dwSize)
872 {
873     LPSTR RegStr_Printer;
874     HKEY hkey = 0;
875     DWORD res = 0;
876
877     if (HIWORD(lpPrinter))
878             TRACE("printer %s\n",lpPrinter);
879     else
880             TRACE("printer %p\n",lpPrinter);
881     if (HIWORD(lpProfile))
882             TRACE("profile %s\n",lpProfile);
883     else
884             TRACE("profile %p\n",lpProfile);
885     TRACE("lpType %08lx\n",lpType);
886
887     if ((!lpPrinter) || (!lpProfile) ||
888     ((DWORD)lpProfile == INT_PD_DEFAULT_MODEL) || (HIWORD(lpProfile) &&
889     (!strcmp(lpProfile, PrinterModel))))
890         return ERROR_INVALID_PARAMETER;
891
892     RegStr_Printer = HeapAlloc(GetProcessHeap(), 0,
893                         strlen(Printers) + strlen(lpPrinter) + 2);
894     strcpy(RegStr_Printer, Printers);
895     strcat(RegStr_Printer, lpPrinter);
896
897     if (((DWORD)lpProfile == INT_PD_DEFAULT_DEVMODE) || (HIWORD(lpProfile) &&
898     (!strcmp(lpProfile, DefaultDevMode)))) {
899         if ( RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey)
900              != ERROR_SUCCESS ||
901              RegSetValueExA(hkey, DefaultDevMode, 0, REG_BINARY,
902                               lpPrinterData, dwSize) != ERROR_SUCCESS )
903                 res = ERROR_INVALID_PRINTER_NAME;
904     }
905     else
906     {
907         strcat(RegStr_Printer, "\\");
908
909         if( (res = RegOpenKeyA(HKEY_LOCAL_MACHINE, RegStr_Printer, &hkey)) ==
910             ERROR_SUCCESS ) {
911
912             if (!lpPrinterData)
913                 res = RegDeleteValueA(hkey, lpProfile);
914             else
915                 res = RegSetValueExA(hkey, lpProfile, 0, lpType,
916                                        lpPrinterData, dwSize);
917         }
918     }
919
920     if (hkey) RegCloseKey(hkey);
921     HeapFree(GetProcessHeap(), 0, RegStr_Printer);
922     return res;
923 }