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