?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Do not check for non NULL pointer before HeapFree'ing it. It's
[wine.git] / dlls / kernel / format_msg.c
blobd1144ea9f3bcdf4afd9e315bd3189e4a3d38ecae
1 /*
2 * FormatMessage implementation
3 *
4 * Copyright 1996 Marcus Meissner
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 */
20
21 #include "config.h"
22
23 #include <stdarg.h>
24 #include <stdio.h>
25 #include <string.h>
26
27 #include "ntstatus.h"
28 #include "windef.h"
29 #include "winbase.h"
30 #include "winerror.h"
31 #include "winreg.h"
32 #include "winternl.h"
33 #include "winuser.h"
34 #include "winnls.h"
35 #include "wine/unicode.h"
36 #include "kernel_private.h"
37 #include "wine/debug.h"
38
39 WINE_DEFAULT_DEBUG_CHANNEL(resource);
40
41
42 /* Messages...used by FormatMessage (KERNEL32.something)
43 *
44 * They can be specified either directly or using a message ID and
45 * loading them from the resource.
46 *
47 * The resourcedata has following format:
48 * start:
49 * 0: DWORD nrofentries
50 * nrofentries * subentry:
51 * 0: DWORD firstentry
52 * 4: DWORD lastentry
53 * 8: DWORD offset from start to the stringentries
54 *
55 * (lastentry-firstentry) * stringentry:
56 * 0: WORD len (0 marks end) [ includes the 4 byte header length ]
57 * 2: WORD flags
58 * 4: CHAR[len-4]
59 * (stringentry i of a subentry refers to the ID 'firstentry+i')
60 *
61 * Yes, ANSI strings in win32 resources. Go figure.
62 */
63
64 static const WCHAR PCNTFMTWSTR[] = { '%','%','%','s',0 };
65 static const WCHAR FMTWSTR[] = { '%','s',0 };
66
67 /**********************************************************************
68 * load_messageW (internal)
69 */
70 static LPWSTR load_messageW( HMODULE module, UINT id, WORD lang )
71 {
72 const MESSAGE_RESOURCE_ENTRY *mre;
73 WCHAR *buffer;
74
75 TRACE("module = %p, id = %08x\n", module, id );
76
77 if (!module) module = GetModuleHandleW( NULL );
78 if (RtlFindMessage( module, RT_MESSAGETABLE, lang, id, &mre ) != STATUS_SUCCESS) return NULL;
79
80 if (mre->Flags & MESSAGE_RESOURCE_UNICODE)
81 {
82 int len = (strlenW( (const WCHAR *)mre->Text ) + 1) * sizeof(WCHAR);
83 if (!(buffer = HeapAlloc( GetProcessHeap(), 0, len ))) return NULL;
84 memcpy( buffer, mre->Text, len );
85 }
86 else
87 {
88 int len = MultiByteToWideChar( CP_ACP, 0, mre->Text, -1, NULL, 0 );
89 if (!(buffer = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) ))) return NULL;
90 MultiByteToWideChar( CP_ACP, 0, mre->Text, -1, buffer, len );
91 }
92 TRACE("returning %s\n", wine_dbgstr_w(buffer));
93 return buffer;
94 }
95
96
97 /**********************************************************************
98 * load_messageA (internal)
99 */
100 static LPSTR load_messageA( HMODULE module, UINT id, WORD lang )
101 {
102 const MESSAGE_RESOURCE_ENTRY *mre;
103 char *buffer;
104
105 TRACE("module = %p, id = %08x\n", module, id );
106
107 if (!module) module = GetModuleHandleW( NULL );
108 if (RtlFindMessage( module, RT_MESSAGETABLE, lang, id, &mre ) != STATUS_SUCCESS) return NULL;
109
110 if (mre->Flags & MESSAGE_RESOURCE_UNICODE)
111 {
112 int len = WideCharToMultiByte( CP_ACP, 0, (const WCHAR *)mre->Text, -1, NULL, 0, NULL, NULL );
113 if (!(buffer = HeapAlloc( GetProcessHeap(), 0, len ))) return NULL;
114 WideCharToMultiByte( CP_ACP, 0, (const WCHAR *)mre->Text, -1, buffer, len, NULL, NULL );
115 }
116 else
117 {
118 int len = strlen(mre->Text) + 1;
119 if (!(buffer = HeapAlloc( GetProcessHeap(), 0, len ))) return NULL;
120 memcpy( buffer, mre->Text, len );
121 }
122 TRACE("returning %s\n", wine_dbgstr_a(buffer));
123 return buffer;
124 }
125
126
127 /***********************************************************************
128 * FormatMessageA (KERNEL32.@)
129 * FIXME: missing wrap,
130 */
131 DWORD WINAPI FormatMessageA(
132 DWORD dwFlags,
133 LPCVOID lpSource,
134 DWORD dwMessageId,
135 DWORD dwLanguageId,
136 LPSTR lpBuffer,
137 DWORD nSize,
138 va_list* _args )
139 {
140 LPDWORD args=(LPDWORD)_args;
141 #if defined(__i386__) || defined(__sparc__)
142 /* This implementation is completely dependent on the format of the va_list on x86 CPUs */
143 LPSTR target,t;
144 DWORD talloced;
145 LPSTR from,f;
146 DWORD width = dwFlags & FORMAT_MESSAGE_MAX_WIDTH_MASK;
147 BOOL eos = FALSE;
148 CHAR ch;
149
150 TRACE("(0x%lx,%p,%ld,0x%lx,%p,%ld,%p)\n",
151 dwFlags,lpSource,dwMessageId,dwLanguageId,lpBuffer,nSize,args);
152 if ((dwFlags & FORMAT_MESSAGE_FROM_STRING)
153 &&((dwFlags & FORMAT_MESSAGE_FROM_SYSTEM)
154 || (dwFlags & FORMAT_MESSAGE_FROM_HMODULE))) return 0;
155
156 if (width && width != FORMAT_MESSAGE_MAX_WIDTH_MASK)
157 FIXME("line wrapping (%lu) not supported.\n", width);
158 from = NULL;
159 if (dwFlags & FORMAT_MESSAGE_FROM_STRING)
160 {
161 from = HeapAlloc( GetProcessHeap(), 0, strlen((LPCSTR)lpSource)+1 );
162 strcpy( from, (LPCSTR)lpSource );
163 }
164 else {
165 from = NULL;
166 if (dwFlags & FORMAT_MESSAGE_FROM_HMODULE)
167 from = load_messageA( (HMODULE)lpSource, dwMessageId, dwLanguageId );
168 if (!from && (dwFlags & FORMAT_MESSAGE_FROM_SYSTEM))
169 from = load_messageA( kernel32_handle, dwMessageId, dwLanguageId );
170
171 if (!from)
172 {
173 SetLastError (ERROR_RESOURCE_LANG_NOT_FOUND);
174 return 0;
175 }
176 }
177 target = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, 100);
178 t = target;
179 talloced= 100;
180
181 #define ADD_TO_T(c) do { \
182 *t++=c;\
183 if (t-target == talloced) {\
184 target = (char*)HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,target,talloced*2);\
185 t = target+talloced;\
186 talloced*=2;\
187 }\
188 } while (0)
189
190 if (from) {
191 f=from;
192 if (dwFlags & FORMAT_MESSAGE_IGNORE_INSERTS) {
193 while (*f && !eos)
194 ADD_TO_T(*f++);
195 }
196 else {
197 while (*f && !eos) {
198 if (*f=='%') {
199 int insertnr;
200 char *fmtstr,*x,*lastf;
201 DWORD *argliststart;
202
203 fmtstr = NULL;
204 lastf = f;
205 f++;
206 if (!*f) {
207 ADD_TO_T('%');
208 continue;
209 }
210 switch (*f) {
211 case '1':case '2':case '3':case '4':case '5':
212 case '6':case '7':case '8':case '9':
213 insertnr=*f-'0';
214 switch (f[1]) {
215 case '0':case '1':case '2':case '3':
216 case '4':case '5':case '6':case '7':
217 case '8':case '9':
218 f++;
219 insertnr=insertnr*10+*f-'0';
220 f++;
221 break;
222 default:
223 f++;
224 break;
225 }
226 if (*f=='!') {
227 f++;
228 if (NULL!=(x=strchr(f,'!'))) {
229 *x='\0';
230 fmtstr=HeapAlloc(GetProcessHeap(),0,strlen(f)+2);
231 sprintf(fmtstr,"%%%s",f);
232 f=x+1;
233 } else {
234 fmtstr=HeapAlloc(GetProcessHeap(),0,strlen(f)+2);
235 sprintf(fmtstr,"%%%s",f);
236 f+=strlen(f); /*at \0*/
237 }
238 } else {
239 if(!args) break;
240 fmtstr = HeapAlloc(GetProcessHeap(),0,3);
241 strcpy( fmtstr, "%s" );
242 }
243 if (args) {
244 int sz;
245 LPSTR b;
246
247 if (dwFlags & FORMAT_MESSAGE_ARGUMENT_ARRAY)
248 argliststart=args+insertnr-1;
249 else
250 argliststart=(*(DWORD**)args)+insertnr-1;
251
252 /* FIXME: precision and width components are not handled correctly */
253 if ( (strcmp(fmtstr, "%ls") == 0) || (strcmp(fmtstr,"%S") == 0) ) {
254 sz = WideCharToMultiByte( CP_ACP, 0, *(WCHAR**)argliststart, -1, NULL, 0, NULL, NULL);
255 b = HeapAlloc(GetProcessHeap(), 0, sz);
256 WideCharToMultiByte( CP_ACP, 0, *(WCHAR**)argliststart, -1, b, sz, NULL, NULL);
257 } else {
258 b = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sz = 1000);
259 /* CMF - This makes a BIG assumption about va_list */
260 TRACE("A BIG assumption\n");
261 vsnprintf(b, sz, fmtstr, (va_list) argliststart);
262 }
263 for (x=b; *x; x++) ADD_TO_T(*x);
264
265 HeapFree(GetProcessHeap(),0,b);
266 } else {
267 /* NULL args - copy formatstr
268 * (probably wrong)
269 */
270 while ((lastf<f)&&(*lastf)) {
271 ADD_TO_T(*lastf++);
272 }
273 }
274 HeapFree(GetProcessHeap(),0,fmtstr);
275 break;
276 case 'n':
277 ADD_TO_T('\r');
278 ADD_TO_T('\n');
279 f++;
280 break;
281 case '0':
282 eos = TRUE;
283 f++;
284 break;
285 default:
286 ADD_TO_T(*f++);
287 break;
288 }
289 } else {
290 ch = *f;
291 f++;
292 if (ch == '\r') {
293 if (*f == '\n')
294 f++;
295 if(width)
296 ADD_TO_T(' ');
297 else
298 {
299 ADD_TO_T('\r');
300 ADD_TO_T('\n');
301 }
302 } else {
303 if (ch == '\n')
304 {
305 if(width)
306 ADD_TO_T(' ');
307 else
308 {
309 ADD_TO_T('\r');
310 ADD_TO_T('\n');
311 }
312 }
313 else
314 ADD_TO_T(ch);
315 }
316 }
317 }
318 }
319 *t='\0';
320 }
321 talloced = strlen(target)+1;
322 if (nSize && talloced<nSize) {
323 target = (char*)HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,target,nSize);
324 }
325 TRACE("-- %s\n",debugstr_a(target));
326 if (dwFlags & FORMAT_MESSAGE_ALLOCATE_BUFFER) {
327 *((LPVOID*)lpBuffer) = (LPVOID)LocalAlloc(GMEM_ZEROINIT,max(nSize, talloced));
328 memcpy(*(LPSTR*)lpBuffer,target,talloced);
329 } else {
330 lstrcpynA(lpBuffer,target,nSize);
331 }
332 HeapFree(GetProcessHeap(),0,target);
333 HeapFree(GetProcessHeap(),0,from);
334 TRACE("-- returning %d\n", (dwFlags & FORMAT_MESSAGE_ALLOCATE_BUFFER) ? strlen(*(LPSTR*)lpBuffer):strlen(lpBuffer));
335 return (dwFlags & FORMAT_MESSAGE_ALLOCATE_BUFFER) ?
336 strlen(*(LPSTR*)lpBuffer):
337 strlen(lpBuffer);
338 #else
339 return 0;
340 #endif /* __i386__ */
341 }
342 #undef ADD_TO_T
343
344
345 /***********************************************************************
346 * FormatMessageW (KERNEL32.@)
347 */
348 DWORD WINAPI FormatMessageW(
349 DWORD dwFlags,
350 LPCVOID lpSource,
351 DWORD dwMessageId,
352 DWORD dwLanguageId,
353 LPWSTR lpBuffer,
354 DWORD nSize,
355 va_list* _args )
356 {
357 LPDWORD args=(LPDWORD)_args;
358 #if defined(__i386__) || defined(__sparc__)
359 /* This implementation is completely dependent on the format of the va_list on x86 CPUs */
360 LPWSTR target,t;
361 DWORD talloced;
362 LPWSTR from,f;
363 DWORD width = dwFlags & FORMAT_MESSAGE_MAX_WIDTH_MASK;
364 BOOL eos = FALSE;
365 CHAR ch;
366
367 TRACE("(0x%lx,%p,%ld,0x%lx,%p,%ld,%p)\n",
368 dwFlags,lpSource,dwMessageId,dwLanguageId,lpBuffer,nSize,args);
369 if ((dwFlags & FORMAT_MESSAGE_FROM_STRING)
370 &&((dwFlags & FORMAT_MESSAGE_FROM_SYSTEM)
371 || (dwFlags & FORMAT_MESSAGE_FROM_HMODULE))) return 0;
372
373 if (width && width != FORMAT_MESSAGE_MAX_WIDTH_MASK)
374 FIXME("line wrapping not supported.\n");
375 from = NULL;
376 if (dwFlags & FORMAT_MESSAGE_FROM_STRING) {
377 from = HeapAlloc( GetProcessHeap(), 0, (strlenW((LPCWSTR)lpSource) + 1) *
378 sizeof(WCHAR) );
379 strcpyW( from, (LPCWSTR)lpSource );
380 }
381 else {
382 from = NULL;
383 if (dwFlags & FORMAT_MESSAGE_FROM_HMODULE)
384 from = load_messageW( (HMODULE)lpSource, dwMessageId, dwLanguageId );
385 if (!from && (dwFlags & FORMAT_MESSAGE_FROM_SYSTEM))
386 from = load_messageW( kernel32_handle, dwMessageId, dwLanguageId );
387
388 if (!from)
389 {
390 SetLastError (ERROR_RESOURCE_LANG_NOT_FOUND);
391 return 0;
392 }
393 }
394 target = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, 100 * sizeof(WCHAR) );
395 t = target;
396 talloced= 100;
397
398 #define ADD_TO_T(c) do {\
399 *t++=c;\
400 if (t-target == talloced) {\
401 target = (WCHAR*)HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,target,talloced*2*sizeof(WCHAR));\
402 t = target+talloced;\
403 talloced*=2;\
404 } \
405 } while (0)
406
407 if (from) {
408 f=from;
409 if (dwFlags & FORMAT_MESSAGE_IGNORE_INSERTS) {
410 while (*f && !eos)
411 ADD_TO_T(*f++);
412 }
413 else {
414 while (*f && !eos) {
415 if (*f=='%') {
416 int insertnr;
417 WCHAR *fmtstr,*sprintfbuf,*x;
418 DWORD *argliststart;
419
420 fmtstr = NULL;
421 f++;
422 if (!*f) {
423 ADD_TO_T('%');
424 continue;
425 }
426
427 switch (*f) {
428 case '1':case '2':case '3':case '4':case '5':
429 case '6':case '7':case '8':case '9':
430 insertnr=*f-'0';
431 switch (f[1]) {
432 case '0':case '1':case '2':case '3':
433 case '4':case '5':case '6':case '7':
434 case '8':case '9':
435 f++;
436 insertnr=insertnr*10+*f-'0';
437 f++;
438 break;
439 default:
440 f++;
441 break;
442 }
443 if (*f=='!') {
444 f++;
445 if (NULL!=(x=strchrW(f,'!'))) {
446 *x='\0';
447 fmtstr=HeapAlloc( GetProcessHeap(), 0,(strlenW(f)+2)*sizeof(WCHAR));
448 sprintfW(fmtstr,PCNTFMTWSTR,f);
449 f=x+1;
450 } else {
451 fmtstr=HeapAlloc(GetProcessHeap(),0,(strlenW(f)+2)*sizeof(WCHAR));
452 sprintfW(fmtstr,PCNTFMTWSTR,f);
453 f+=strlenW(f); /*at \0*/
454 }
455 } else {
456 if(!args) break;
457 fmtstr = HeapAlloc( GetProcessHeap(),0,3*sizeof(WCHAR));
458 strcpyW( fmtstr, FMTWSTR );
459 }
460 if (dwFlags & FORMAT_MESSAGE_ARGUMENT_ARRAY)
461 argliststart=args+insertnr-1;
462 else
463 argliststart=(*(DWORD**)args)+insertnr-1;
464
465 if (fmtstr[strlenW(fmtstr)-1]=='s' && argliststart[0]) {
466 DWORD xarr[3];
467
468 xarr[0]=*(argliststart+0);
469 /* possible invalid pointers */
470 xarr[1]=*(argliststart+1);
471 xarr[2]=*(argliststart+2);
472 sprintfbuf=HeapAlloc(GetProcessHeap(),0,(strlenW((LPWSTR)argliststart[0])*2+1)*sizeof(WCHAR));
473
474 /* CMF - This makes a BIG assumption about va_list */
475 vsprintfW(sprintfbuf, fmtstr, (va_list) xarr);
476 } else {
477 sprintfbuf=HeapAlloc(GetProcessHeap(),0,100);
478
479 /* CMF - This makes a BIG assumption about va_list */
480 vsprintfW(sprintfbuf, fmtstr, (va_list) argliststart);
481 }
482 x=sprintfbuf;
483 while (*x) {
484 ADD_TO_T(*x++);
485 }
486 HeapFree(GetProcessHeap(),0,sprintfbuf);
487 HeapFree(GetProcessHeap(),0,fmtstr);
488 break;
489 case 'n':
490 ADD_TO_T('\r');
491 ADD_TO_T('\n');
492 f++;
493 break;
494 case '0':
495 eos = TRUE;
496 f++;
497 break;
498 default:
499 ADD_TO_T(*f++);
500 break;
501 }
502 } else {
503 ch = *f;
504 f++;
505 if (ch == '\r') {
506 if (*f == '\n')
507 f++;
508 if(width)
509 ADD_TO_T(' ');
510 else
511 {
512 ADD_TO_T('\r');
513 ADD_TO_T('\n');
514 }
515 } else {
516 if (ch == '\n')
517 {
518 if(width)
519 ADD_TO_T(' ');
520 else
521 {
522 ADD_TO_T('\r');
523 ADD_TO_T('\n');
524 }
525 }
526 else
527 ADD_TO_T(ch);
528 }
529 }
530 }
531 }
532 *t='\0';
533 }
534 talloced = strlenW(target)+1;
535 if (nSize && talloced<nSize)
536 target = (WCHAR*)HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,target,nSize*sizeof(WCHAR));
537 if (dwFlags & FORMAT_MESSAGE_ALLOCATE_BUFFER) {
538 /* nSize is the MINIMUM size */
539 DWORD len = strlenW(target) + 1;
540 *((LPVOID*)lpBuffer) = (LPVOID)LocalAlloc(GMEM_ZEROINIT,len*sizeof(WCHAR));
541 strcpyW(*(LPWSTR*)lpBuffer, target);
542 }
543 else lstrcpynW(lpBuffer, target, nSize);
544
545 HeapFree(GetProcessHeap(),0,target);
546 HeapFree(GetProcessHeap(),0,from);
547 TRACE("ret=%s\n", wine_dbgstr_w((dwFlags & FORMAT_MESSAGE_ALLOCATE_BUFFER) ?
548 *(LPWSTR*)lpBuffer : lpBuffer));
549 return (dwFlags & FORMAT_MESSAGE_ALLOCATE_BUFFER) ?
550 strlenW(*(LPWSTR*)lpBuffer):
551 strlenW(lpBuffer);
552 #else
553 return 0;
554 #endif /* __i386__ */
555 }
556 #undef ADD_TO_T
Windows API implementation