gdi32: Fix invalid parameter handling in EnumICMProfiles and SetICMProfile.
[wine] / dlls / gdi32 / icm.c
1 /*
2  * Image Color Management
3  *
4  * Copyright 2004 Marcus Meissner
5  * Copyright 2008 Hans Leidekker
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  */
21
22 #include "config.h"
23
24 #include <stdarg.h>
25 #include <string.h>
26
27 #include "windef.h"
28 #include "winbase.h"
29 #include "wingdi.h"
30 #include "winnls.h"
31 #include "winreg.h"
32
33 #include "gdi_private.h"
34
35 #include "wine/debug.h"
36 #include "wine/unicode.h"
37
38 WINE_DEFAULT_DEBUG_CHANNEL(icm);
39
40
41 struct enum_profiles
42 {
43     BOOL unicode;
44     union
45     {
46         ICMENUMPROCA funcA;
47         ICMENUMPROCW funcW;
48     } callback;
49     LPARAM data;
50 };
51
52 INT CALLBACK enum_profiles_callback( LPWSTR filename, LPARAM lparam )
53 {
54     int len, ret = -1;
55     struct enum_profiles *ep = (struct enum_profiles *)lparam;
56     char *filenameA;
57
58     if (ep->unicode)
59         return ep->callback.funcW( filename, ep->data );
60
61     len = WideCharToMultiByte( CP_ACP, 0, filename, -1, NULL, 0, NULL, NULL );
62     filenameA = HeapAlloc( GetProcessHeap(), 0, len );
63     if (filenameA)
64     {
65         WideCharToMultiByte( CP_ACP, 0, filename, -1, filenameA, len, NULL, NULL );
66         ret = ep->callback.funcA( filenameA, ep->data );
67         HeapFree( GetProcessHeap(), 0, filenameA );
68     }
69     return ret;
70 }
71
72 /***********************************************************************
73  *           EnumICMProfilesA    (GDI32.@)
74  */
75 INT WINAPI EnumICMProfilesA(HDC hdc, ICMENUMPROCA func, LPARAM lparam)
76 {
77     DC *dc;
78     INT ret = -1;
79
80     TRACE("%p, %p, 0x%08lx\n", hdc, func, lparam);
81
82     if (!func) return -1;
83     if ((dc = get_dc_ptr(hdc)))
84     {
85         if (dc->funcs->pEnumICMProfiles)
86         {
87             struct enum_profiles ep;
88
89             ep.unicode        = FALSE;
90             ep.callback.funcA = func;
91             ep.data           = lparam;
92             ret = dc->funcs->pEnumICMProfiles(dc->physDev, enum_profiles_callback, (LPARAM)&ep);
93         }
94         release_dc_ptr(dc);
95     }
96     return ret;
97 }
98
99 /***********************************************************************
100  *           EnumICMProfilesW    (GDI32.@)
101  */
102 INT WINAPI EnumICMProfilesW(HDC hdc, ICMENUMPROCW func, LPARAM lparam)
103 {
104     DC *dc;
105     INT ret = -1;
106
107     TRACE("%p, %p, 0x%08lx\n", hdc, func, lparam);
108
109     if (!func) return -1;
110     if ((dc = get_dc_ptr(hdc)))
111     {
112         if (dc->funcs->pEnumICMProfiles)
113         {
114             struct enum_profiles ep;
115
116             ep.unicode        = TRUE;
117             ep.callback.funcW = func;
118             ep.data           = lparam;
119             ret = dc->funcs->pEnumICMProfiles(dc->physDev, enum_profiles_callback, (LPARAM)&ep);
120         }
121         release_dc_ptr(dc);
122     }
123     return ret;
124 }
125
126 /**********************************************************************
127  *           GetICMProfileA   (GDI32.@)
128  *
129  * Returns the filename of the specified device context's color
130  * management profile, even if color management is not enabled
131  * for that DC.
132  *
133  * RETURNS
134  *    TRUE if filename is copied successfully.
135  *    FALSE if the buffer length pointed to by size is too small.
136  *
137  * FIXME
138  *    How does Windows assign these? Some registry key?
139  */
140 BOOL WINAPI GetICMProfileA(HDC hdc, LPDWORD size, LPSTR filename)
141 {
142     WCHAR filenameW[MAX_PATH];
143     DWORD buflen = MAX_PATH;
144     BOOL ret = FALSE;
145
146     TRACE("%p, %p, %p\n", hdc, size, filename);
147
148     if (!hdc || !size || !filename) return FALSE;
149
150     if (GetICMProfileW(hdc, &buflen, filenameW))
151     {
152         int len = WideCharToMultiByte(CP_ACP, 0, filenameW, -1, NULL, 0, NULL, NULL);
153         if (*size >= len)
154         {
155             WideCharToMultiByte(CP_ACP, 0, filenameW, -1, filename, *size, NULL, NULL);
156             ret = TRUE;
157         }
158         else SetLastError(ERROR_INSUFFICIENT_BUFFER);
159         *size = len;
160     }
161     return ret;
162 }
163
164 /**********************************************************************
165  *           GetICMProfileW     (GDI32.@)
166  */
167 BOOL WINAPI GetICMProfileW(HDC hdc, LPDWORD size, LPWSTR filename)
168 {
169     BOOL ret = FALSE;
170     DC *dc = get_dc_ptr(hdc);
171
172     TRACE("%p, %p, %p\n", hdc, size, filename);
173
174     if (dc)
175     {
176         if (dc->funcs->pGetICMProfile)
177             ret = dc->funcs->pGetICMProfile(dc->physDev, size, filename);
178         release_dc_ptr(dc);
179     }
180     return ret;
181 }
182
183 /**********************************************************************
184  *           GetLogColorSpaceA     (GDI32.@)
185  */
186 BOOL WINAPI GetLogColorSpaceA(HCOLORSPACE colorspace, LPLOGCOLORSPACEA buffer, DWORD size)
187 {
188     FIXME("%p %p 0x%08x stub\n", colorspace, buffer, size);
189     return FALSE;
190 }
191
192 /**********************************************************************
193  *           GetLogColorSpaceW      (GDI32.@)
194  */
195 BOOL WINAPI GetLogColorSpaceW(HCOLORSPACE colorspace, LPLOGCOLORSPACEW buffer, DWORD size)
196 {
197     FIXME("%p %p 0x%08x stub\n", colorspace, buffer, size);
198     return FALSE;
199 }
200
201 /**********************************************************************
202  *           SetICMProfileA         (GDI32.@)
203  */
204 BOOL WINAPI SetICMProfileA(HDC hdc, LPSTR filename)
205 {
206     FIXME("%p %s stub\n", hdc, debugstr_a(filename));
207
208     if (!filename)
209     {
210         SetLastError( ERROR_INVALID_PARAMETER );
211         return FALSE;
212     }
213     if (!hdc)
214     {
215         SetLastError( ERROR_INVALID_HANDLE );
216         return FALSE;
217     }
218     return TRUE;
219 }
220
221 /**********************************************************************
222  *           SetICMProfileW         (GDI32.@)
223  */
224 BOOL WINAPI SetICMProfileW(HDC hdc, LPWSTR filename)
225 {
226     FIXME("%p %s stub\n", hdc, debugstr_w(filename));
227
228     if (!filename)
229     {
230         SetLastError( ERROR_INVALID_PARAMETER );
231         return FALSE;
232     }
233     if (!hdc)
234     {
235         SetLastError( ERROR_INVALID_HANDLE );
236         return FALSE;
237     }
238     return TRUE;
239 }
240
241 /**********************************************************************
242  *           UpdateICMRegKeyA       (GDI32.@)
243  */
244 BOOL WINAPI UpdateICMRegKeyA(DWORD reserved, LPSTR cmid, LPSTR filename, UINT command)
245 {
246     FIXME("0x%08x, %s, %s, 0x%08x stub\n", reserved, debugstr_a(cmid), debugstr_a(filename), command);
247     return TRUE;
248 }
249
250 /**********************************************************************
251  *           UpdateICMRegKeyW       (GDI32.@)
252  */
253 BOOL WINAPI UpdateICMRegKeyW(DWORD reserved, LPWSTR cmid, LPWSTR filename, UINT command)
254 {
255     FIXME("0x%08x, %s, %s, 0x%08x stub\n", reserved, debugstr_w(cmid), debugstr_w(filename), command);
256     return TRUE;
257 }