gdi32: Add more parameter checks in GdiGradientFill.
[wine] / dlls / gdi32 / enhmfdrv / objects.c
1 /*
2  * Enhanced MetaFile objects
3  *
4  * Copyright 1999 Huw D M Davies
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library 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 GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include <stdlib.h>
22 #include <stdio.h>
23 #include <string.h>
24
25 #include "enhmfdrv/enhmetafiledrv.h"
26 #include "gdi_private.h"
27 #include "wine/debug.h"
28
29 WINE_DEFAULT_DEBUG_CHANNEL(enhmetafile);
30
31
32 /******************************************************************
33  *         EMFDRV_AddHandle
34  */
35 static UINT EMFDRV_AddHandle( PHYSDEV dev, HGDIOBJ obj )
36 {
37     EMFDRV_PDEVICE *physDev = (EMFDRV_PDEVICE *)dev;
38     UINT index;
39
40     for(index = 0; index < physDev->handles_size; index++)
41         if(physDev->handles[index] == 0) break;
42     if(index == physDev->handles_size) {
43         physDev->handles_size += HANDLE_LIST_INC;
44         physDev->handles = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
45                                        physDev->handles,
46                                        physDev->handles_size * sizeof(physDev->handles[0]));
47     }
48     physDev->handles[index] = obj;
49
50     physDev->cur_handles++;
51     if(physDev->cur_handles > physDev->emh->nHandles)
52         physDev->emh->nHandles++;
53
54     return index + 1; /* index 0 is reserved for the hmf, so we increment everything by 1 */
55 }
56
57 /******************************************************************
58  *         EMFDRV_FindObject
59  */
60 static UINT EMFDRV_FindObject( PHYSDEV dev, HGDIOBJ obj )
61 {
62     EMFDRV_PDEVICE *physDev = (EMFDRV_PDEVICE*) dev;
63     UINT index;
64
65     for(index = 0; index < physDev->handles_size; index++)
66         if(physDev->handles[index] == obj) break;
67
68     if(index == physDev->handles_size) return 0;
69
70     return index + 1;
71 }
72
73
74 /******************************************************************
75  *         EMFDRV_DeleteObject
76  */
77 BOOL EMFDRV_DeleteObject( PHYSDEV dev, HGDIOBJ obj )
78 {
79     EMRDELETEOBJECT emr;
80     EMFDRV_PDEVICE *physDev = (EMFDRV_PDEVICE*) dev;
81     UINT index;
82     BOOL ret = TRUE;
83
84     if(!(index = EMFDRV_FindObject(dev, obj))) return 0;
85
86     emr.emr.iType = EMR_DELETEOBJECT;
87     emr.emr.nSize = sizeof(emr);
88     emr.ihObject = index;
89
90     if(!EMFDRV_WriteRecord( dev, &emr.emr ))
91         ret = FALSE;
92
93     physDev->handles[index - 1] = 0;
94     physDev->cur_handles--;
95     return ret;
96 }
97
98
99 /***********************************************************************
100  *           EMFDRV_SelectBitmap
101  */
102 HBITMAP EMFDRV_SelectBitmap( PHYSDEV dev, HBITMAP hbitmap )
103 {
104     return 0;
105 }
106
107
108 /***********************************************************************
109  *           EMFDRV_CreateBrushIndirect
110  */
111 DWORD EMFDRV_CreateBrushIndirect( PHYSDEV dev, HBRUSH hBrush )
112 {
113     DWORD index = 0;
114     LOGBRUSH logbrush;
115
116     if (!GetObjectA( hBrush, sizeof(logbrush), &logbrush )) return 0;
117
118     switch (logbrush.lbStyle) {
119     case BS_SOLID:
120     case BS_HATCHED:
121     case BS_NULL:
122       {
123         EMRCREATEBRUSHINDIRECT emr;
124         emr.emr.iType = EMR_CREATEBRUSHINDIRECT;
125         emr.emr.nSize = sizeof(emr);
126         emr.ihBrush = index = EMFDRV_AddHandle( dev, hBrush );
127         emr.lb.lbStyle = logbrush.lbStyle;
128         emr.lb.lbColor = logbrush.lbColor;
129         emr.lb.lbHatch = logbrush.lbHatch;
130
131         if(!EMFDRV_WriteRecord( dev, &emr.emr ))
132             index = 0;
133       }
134       break;
135     case BS_PATTERN:
136     case BS_DIBPATTERN:
137       {
138         EMRCREATEDIBPATTERNBRUSHPT *emr;
139         char buffer[FIELD_OFFSET( BITMAPINFO, bmiColors[256] )];
140         BITMAPINFO *info = (BITMAPINFO *)buffer;
141         DWORD info_size, image_size;
142         void *bits;
143         UINT usage;
144
145         if (!get_brush_bitmap_info( hBrush, info, &bits, &usage )) break;
146         info_size = bitmap_info_size( info, usage );
147         image_size = get_dib_image_size( info );
148
149         emr = HeapAlloc( GetProcessHeap(), 0, sizeof(EMRCREATEDIBPATTERNBRUSHPT)+info_size+image_size );
150         if(!emr) break;
151
152         if (logbrush.lbStyle == BS_PATTERN && info->bmiHeader.biBitCount == 1)
153         {
154             /* Presumably to reduce the size of the written EMF, MS supports an
155              * undocumented iUsage value of 2, indicating a mono bitmap without the
156              * 8 byte 2 entry black/white palette. Stupidly, they could have saved
157              * over 20 bytes more by also ignoring the BITMAPINFO fields that are
158              * irrelevant/constant for monochrome bitmaps.
159              * FIXME: It may be that the DIB functions themselves accept this value.
160              */
161             emr->emr.iType = EMR_CREATEMONOBRUSH;
162             usage = DIB_PAL_MONO;
163             /* FIXME: There is an extra DWORD written by native before the BMI.
164              *        Not sure what its meant to contain.
165              */
166             emr->offBmi = sizeof( EMRCREATEDIBPATTERNBRUSHPT ) + sizeof(DWORD);
167             emr->cbBmi = sizeof( BITMAPINFOHEADER );
168         }
169         else
170         {
171             emr->emr.iType = EMR_CREATEDIBPATTERNBRUSHPT;
172             emr->offBmi = sizeof( EMRCREATEDIBPATTERNBRUSHPT );
173             emr->cbBmi = info_size;
174         }
175         emr->emr.nSize = emr->offBits + emr->cbBits;
176         emr->ihBrush = index = EMFDRV_AddHandle( dev, hBrush );
177         emr->iUsage = usage;
178         emr->offBits = emr->offBmi + emr->cbBmi;
179         emr->cbBits = image_size;
180
181         memcpy( (BYTE *)emr + emr->offBmi, info, emr->cbBmi );
182         memcpy( (BYTE *)emr + emr->offBits, bits, emr->cbBits );
183
184         if(!EMFDRV_WriteRecord( dev, &emr->emr ))
185             index = 0;
186         HeapFree( GetProcessHeap(), 0, emr );
187       }
188       break;
189
190     default:
191         FIXME("Unknown style %x\n", logbrush.lbStyle);
192         break;
193     }
194     return index;
195 }
196
197
198 /***********************************************************************
199  *           EMFDRV_SelectBrush
200  */
201 HBRUSH EMFDRV_SelectBrush( PHYSDEV dev, HBRUSH hBrush, HBITMAP bitmap,
202                            const BITMAPINFO *info, void *bits, UINT usage )
203 {
204     EMFDRV_PDEVICE *physDev = (EMFDRV_PDEVICE*)dev;
205     EMRSELECTOBJECT emr;
206     DWORD index;
207     int i;
208
209     if (physDev->restoring) return hBrush;  /* don't output SelectObject records during RestoreDC */
210
211     /* If the object is a stock brush object, do not need to create it.
212      * See definitions in  wingdi.h for range of stock brushes.
213      * We do however have to handle setting the higher order bit to
214      * designate that this is a stock object.
215      */
216     for (i = WHITE_BRUSH; i <= DC_BRUSH; i++)
217     {
218         if (hBrush == GetStockObject(i))
219         {
220             index = i | 0x80000000;
221             goto found;
222         }
223     }
224     if((index = EMFDRV_FindObject(dev, hBrush)) != 0)
225         goto found;
226
227     if (!(index = EMFDRV_CreateBrushIndirect(dev, hBrush ))) return 0;
228     GDI_hdc_using_object(hBrush, dev->hdc);
229
230  found:
231     emr.emr.iType = EMR_SELECTOBJECT;
232     emr.emr.nSize = sizeof(emr);
233     emr.ihObject = index;
234     return EMFDRV_WriteRecord( dev, &emr.emr ) ? hBrush : 0;
235 }
236
237
238 /******************************************************************
239  *         EMFDRV_CreateFontIndirect
240  */
241 static BOOL EMFDRV_CreateFontIndirect(PHYSDEV dev, HFONT hFont )
242 {
243     DWORD index = 0;
244     EMREXTCREATEFONTINDIRECTW emr;
245     int i;
246
247     if (!GetObjectW( hFont, sizeof(emr.elfw.elfLogFont), &emr.elfw.elfLogFont )) return 0;
248
249     emr.emr.iType = EMR_EXTCREATEFONTINDIRECTW;
250     emr.emr.nSize = (sizeof(emr) + 3) / 4 * 4;
251     emr.ihFont = index = EMFDRV_AddHandle( dev, hFont );
252     emr.elfw.elfFullName[0] = '\0';
253     emr.elfw.elfStyle[0]    = '\0';
254     emr.elfw.elfVersion     = 0;
255     emr.elfw.elfStyleSize   = 0;
256     emr.elfw.elfMatch       = 0;
257     emr.elfw.elfReserved    = 0;
258     for(i = 0; i < ELF_VENDOR_SIZE; i++)
259         emr.elfw.elfVendorId[i] = 0;
260     emr.elfw.elfCulture                 = PAN_CULTURE_LATIN;
261     emr.elfw.elfPanose.bFamilyType      = PAN_NO_FIT;
262     emr.elfw.elfPanose.bSerifStyle      = PAN_NO_FIT;
263     emr.elfw.elfPanose.bWeight          = PAN_NO_FIT;
264     emr.elfw.elfPanose.bProportion      = PAN_NO_FIT;
265     emr.elfw.elfPanose.bContrast        = PAN_NO_FIT;
266     emr.elfw.elfPanose.bStrokeVariation = PAN_NO_FIT;
267     emr.elfw.elfPanose.bArmStyle        = PAN_NO_FIT;
268     emr.elfw.elfPanose.bLetterform      = PAN_NO_FIT;
269     emr.elfw.elfPanose.bMidline         = PAN_NO_FIT;
270     emr.elfw.elfPanose.bXHeight         = PAN_NO_FIT;
271
272     if(!EMFDRV_WriteRecord( dev, &emr.emr ))
273         index = 0;
274     return index;
275 }
276
277
278 /***********************************************************************
279  *           EMFDRV_SelectFont
280  */
281 HFONT EMFDRV_SelectFont( PHYSDEV dev, HFONT hFont )
282 {
283     EMFDRV_PDEVICE *physDev = (EMFDRV_PDEVICE*)dev;
284     EMRSELECTOBJECT emr;
285     DWORD index;
286     int i;
287
288     if (physDev->restoring) goto done;  /* don't output SelectObject records during RestoreDC */
289
290     /* If the object is a stock font object, do not need to create it.
291      * See definitions in  wingdi.h for range of stock fonts.
292      * We do however have to handle setting the higher order bit to
293      * designate that this is a stock object.
294      */
295
296     for (i = OEM_FIXED_FONT; i <= DEFAULT_GUI_FONT; i++)
297     {
298         if (i != DEFAULT_PALETTE && hFont == GetStockObject(i))
299         {
300             index = i | 0x80000000;
301             goto found;
302         }
303     }
304
305     if((index = EMFDRV_FindObject(dev, hFont)) != 0)
306         goto found;
307
308     if (!(index = EMFDRV_CreateFontIndirect(dev, hFont ))) return 0;
309     GDI_hdc_using_object(hFont, dev->hdc);
310
311  found:
312     emr.emr.iType = EMR_SELECTOBJECT;
313     emr.emr.nSize = sizeof(emr);
314     emr.ihObject = index;
315     if(!EMFDRV_WriteRecord( dev, &emr.emr ))
316         return 0;
317 done:
318     dev = GET_NEXT_PHYSDEV( dev, pSelectFont );
319     dev->funcs->pSelectFont( dev, hFont );
320     return hFont;
321 }
322
323
324
325 /******************************************************************
326  *         EMFDRV_CreatePenIndirect
327  */
328 static DWORD EMFDRV_CreatePenIndirect(PHYSDEV dev, HPEN hPen)
329 {
330     EMRCREATEPEN emr;
331     DWORD index = 0;
332
333     if (!GetObjectW( hPen, sizeof(emr.lopn), &emr.lopn ))
334     {
335         /* must be an extended pen */
336         EXTLOGPEN *elp;
337         INT size = GetObjectW( hPen, 0, NULL );
338
339         if (!size) return 0;
340
341         elp = HeapAlloc( GetProcessHeap(), 0, size );
342
343         GetObjectW( hPen, size, elp );
344         /* FIXME: add support for user style pens */
345         emr.lopn.lopnStyle = elp->elpPenStyle;
346         emr.lopn.lopnWidth.x = elp->elpWidth;
347         emr.lopn.lopnWidth.y = 0;
348         emr.lopn.lopnColor = elp->elpColor;
349
350         HeapFree( GetProcessHeap(), 0, elp );
351     }
352
353     emr.emr.iType = EMR_CREATEPEN;
354     emr.emr.nSize = sizeof(emr);
355     emr.ihPen = index = EMFDRV_AddHandle( dev, hPen );
356
357     if(!EMFDRV_WriteRecord( dev, &emr.emr ))
358         index = 0;
359     return index;
360 }
361
362 /******************************************************************
363  *         EMFDRV_SelectPen
364  */
365 HPEN EMFDRV_SelectPen(PHYSDEV dev, HPEN hPen )
366 {
367     EMFDRV_PDEVICE *physDev = (EMFDRV_PDEVICE*)dev;
368     EMRSELECTOBJECT emr;
369     DWORD index;
370     int i;
371
372     if (physDev->restoring) return hPen;  /* don't output SelectObject records during RestoreDC */
373
374     /* If the object is a stock pen object, do not need to create it.
375      * See definitions in  wingdi.h for range of stock pens.
376      * We do however have to handle setting the higher order bit to
377      * designate that this is a stock object.
378      */
379
380     for (i = WHITE_PEN; i <= DC_PEN; i++)
381     {
382         if (hPen == GetStockObject(i))
383         {
384             index = i | 0x80000000;
385             goto found;
386         }
387     }
388     if((index = EMFDRV_FindObject(dev, hPen)) != 0)
389         goto found;
390
391     if (!(index = EMFDRV_CreatePenIndirect(dev, hPen))) return 0;
392     GDI_hdc_using_object(hPen, dev->hdc);
393
394  found:
395     emr.emr.iType = EMR_SELECTOBJECT;
396     emr.emr.nSize = sizeof(emr);
397     emr.ihObject = index;
398     return EMFDRV_WriteRecord( dev, &emr.emr ) ? hPen : 0;
399 }
400
401
402 /******************************************************************
403  *         EMFDRV_CreatePalette
404  */
405 static DWORD EMFDRV_CreatePalette(PHYSDEV dev, HPALETTE hPal)
406 {
407     WORD i;
408     struct {
409         EMRCREATEPALETTE hdr;
410         PALETTEENTRY entry[255];
411     } pal;
412
413     memset( &pal, 0, sizeof(pal) );
414
415     if (!GetObjectW( hPal, sizeof(pal.hdr.lgpl) + sizeof(pal.entry), &pal.hdr.lgpl ))
416         return 0;
417
418     for (i = 0; i < pal.hdr.lgpl.palNumEntries; i++)
419         pal.hdr.lgpl.palPalEntry[i].peFlags = 0;
420
421     pal.hdr.emr.iType = EMR_CREATEPALETTE;
422     pal.hdr.emr.nSize = sizeof(pal.hdr) + pal.hdr.lgpl.palNumEntries * sizeof(PALETTEENTRY);
423     pal.hdr.ihPal = EMFDRV_AddHandle( dev, hPal );
424
425     if (!EMFDRV_WriteRecord( dev, &pal.hdr.emr ))
426         pal.hdr.ihPal = 0;
427     return pal.hdr.ihPal;
428 }
429
430 /******************************************************************
431  *         EMFDRV_SelectPalette
432  */
433 HPALETTE EMFDRV_SelectPalette( PHYSDEV dev, HPALETTE hPal, BOOL force )
434 {
435     EMFDRV_PDEVICE *physDev = (EMFDRV_PDEVICE*)dev;
436     EMRSELECTPALETTE emr;
437     DWORD index;
438
439     if (physDev->restoring) return hPal;  /* don't output SelectObject records during RestoreDC */
440
441     if (hPal == GetStockObject( DEFAULT_PALETTE ))
442     {
443         index = DEFAULT_PALETTE | 0x80000000;
444         goto found;
445     }
446
447     if ((index = EMFDRV_FindObject( dev, hPal )) != 0)
448         goto found;
449
450     if (!(index = EMFDRV_CreatePalette( dev, hPal ))) return 0;
451     GDI_hdc_using_object( hPal, dev->hdc );
452
453 found:
454     emr.emr.iType = EMR_SELECTPALETTE;
455     emr.emr.nSize = sizeof(emr);
456     emr.ihPal = index;
457     return EMFDRV_WriteRecord( dev, &emr.emr ) ? hPal : 0;
458 }
459
460 /******************************************************************
461  *         EMFDRV_SetDCBrushColor
462  */
463 COLORREF EMFDRV_SetDCBrushColor( PHYSDEV dev, COLORREF color )
464 {
465     EMFDRV_PDEVICE *physDev = (EMFDRV_PDEVICE*)dev;
466     EMRSELECTOBJECT emr;
467     DWORD index;
468
469     if (GetCurrentObject( dev->hdc, OBJ_BRUSH ) != GetStockObject( DC_BRUSH )) return color;
470
471     if (physDev->dc_brush) DeleteObject( physDev->dc_brush );
472     if (!(physDev->dc_brush = CreateSolidBrush( color ))) return CLR_INVALID;
473     if (!(index = EMFDRV_CreateBrushIndirect(dev, physDev->dc_brush ))) return CLR_INVALID;
474     GDI_hdc_using_object( physDev->dc_brush, dev->hdc );
475     emr.emr.iType = EMR_SELECTOBJECT;
476     emr.emr.nSize = sizeof(emr);
477     emr.ihObject = index;
478     return EMFDRV_WriteRecord( dev, &emr.emr ) ? color : CLR_INVALID;
479 }
480
481 /******************************************************************
482  *         EMFDRV_SetDCPenColor
483  */
484 COLORREF EMFDRV_SetDCPenColor( PHYSDEV dev, COLORREF color )
485 {
486     EMFDRV_PDEVICE *physDev = (EMFDRV_PDEVICE*)dev;
487     EMRSELECTOBJECT emr;
488     DWORD index;
489     LOGPEN logpen = { PS_SOLID, { 0, 0 }, color };
490
491     if (GetCurrentObject( dev->hdc, OBJ_PEN ) != GetStockObject( DC_PEN )) return color;
492
493     if (physDev->dc_pen) DeleteObject( physDev->dc_pen );
494     if (!(physDev->dc_pen = CreatePenIndirect( &logpen ))) return CLR_INVALID;
495     if (!(index = EMFDRV_CreatePenIndirect(dev, physDev->dc_pen))) return CLR_INVALID;
496     GDI_hdc_using_object( physDev->dc_pen, dev->hdc );
497     emr.emr.iType = EMR_SELECTOBJECT;
498     emr.emr.nSize = sizeof(emr);
499     emr.ihObject = index;
500     return EMFDRV_WriteRecord( dev, &emr.emr ) ? color : CLR_INVALID;
501 }
502
503 /******************************************************************
504  *         EMFDRV_GdiComment
505  */
506 BOOL EMFDRV_GdiComment(PHYSDEV dev, UINT bytes, CONST BYTE *buffer)
507 {
508     EMRGDICOMMENT *emr;
509     UINT total, rounded_size;
510     BOOL ret;
511
512     rounded_size = (bytes+3) & ~3;
513     total = offsetof(EMRGDICOMMENT,Data) + rounded_size;
514
515     emr = HeapAlloc(GetProcessHeap(), 0, total);
516     emr->emr.iType = EMR_GDICOMMENT;
517     emr->emr.nSize = total;
518     emr->cbData = bytes;
519     memset(&emr->Data[bytes], 0, rounded_size - bytes);
520     memcpy(&emr->Data[0], buffer, bytes);
521
522     ret = EMFDRV_WriteRecord( dev, &emr->emr );
523
524     HeapFree(GetProcessHeap(), 0, emr);
525
526     return ret;
527 }