| /* |
| * FormatMessage implementation |
| * |
| * Copyright 1996 Marcus Meissner |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| */ |
| |
| #include "config.h" |
| |
| #include <stdarg.h> |
| #include <stdio.h> |
| #include <string.h> |
| |
| #include "ntstatus.h" |
| #define WIN32_NO_STATUS |
| #include "windef.h" |
| #include "winbase.h" |
| #include "winerror.h" |
| #include "winternl.h" |
| #include "winuser.h" |
| #include "winnls.h" |
| #include "wine/unicode.h" |
| #include "kernel_private.h" |
| #include "wine/debug.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(resource); |
| |
| |
| /* Messages...used by FormatMessage (KERNEL32.something) |
| * |
| * They can be specified either directly or using a message ID and |
| * loading them from the resource. |
| * |
| * The resourcedata has following format: |
| * start: |
| * 0: DWORD nrofentries |
| * nrofentries * subentry: |
| * 0: DWORD firstentry |
| * 4: DWORD lastentry |
| * 8: DWORD offset from start to the stringentries |
| * |
| * (lastentry-firstentry) * stringentry: |
| * 0: WORD len (0 marks end) [ includes the 4 byte header length ] |
| * 2: WORD flags |
| * 4: CHAR[len-4] |
| * (stringentry i of a subentry refers to the ID 'firstentry+i') |
| * |
| * Yes, ANSI strings in win32 resources. Go figure. |
| */ |
| |
| static const WCHAR PCNTFMTWSTR[] = { '%','%','%','s',0 }; |
| static const WCHAR FMTWSTR[] = { '%','s',0 }; |
| |
| /********************************************************************** |
| * load_messageW (internal) |
| */ |
| static LPWSTR load_messageW( HMODULE module, UINT id, WORD lang ) |
| { |
| const MESSAGE_RESOURCE_ENTRY *mre; |
| WCHAR *buffer; |
| |
| TRACE("module = %p, id = %08x\n", module, id ); |
| |
| if (!module) module = GetModuleHandleW( NULL ); |
| if (RtlFindMessage( module, RT_MESSAGETABLE, lang, id, &mre ) != STATUS_SUCCESS) return NULL; |
| |
| if (mre->Flags & MESSAGE_RESOURCE_UNICODE) |
| { |
| int len = (strlenW( (const WCHAR *)mre->Text ) + 1) * sizeof(WCHAR); |
| if (!(buffer = HeapAlloc( GetProcessHeap(), 0, len ))) return NULL; |
| memcpy( buffer, mre->Text, len ); |
| } |
| else |
| { |
| int len = MultiByteToWideChar( CP_ACP, 0, (const char *)mre->Text, -1, NULL, 0 ); |
| if (!(buffer = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) ))) return NULL; |
| MultiByteToWideChar( CP_ACP, 0, (const char*)mre->Text, -1, buffer, len ); |
| } |
| TRACE("returning %s\n", wine_dbgstr_w(buffer)); |
| return buffer; |
| } |
| |
| |
| /********************************************************************** |
| * load_messageA (internal) |
| */ |
| static LPSTR load_messageA( HMODULE module, UINT id, WORD lang ) |
| { |
| const MESSAGE_RESOURCE_ENTRY *mre; |
| char *buffer; |
| |
| TRACE("module = %p, id = %08x\n", module, id ); |
| |
| if (!module) module = GetModuleHandleW( NULL ); |
| if (RtlFindMessage( module, RT_MESSAGETABLE, lang, id, &mre ) != STATUS_SUCCESS) return NULL; |
| |
| if (mre->Flags & MESSAGE_RESOURCE_UNICODE) |
| { |
| int len = WideCharToMultiByte( CP_ACP, 0, (const WCHAR *)mre->Text, -1, NULL, 0, NULL, NULL ); |
| if (!(buffer = HeapAlloc( GetProcessHeap(), 0, len ))) return NULL; |
| WideCharToMultiByte( CP_ACP, 0, (const WCHAR *)mre->Text, -1, buffer, len, NULL, NULL ); |
| } |
| else |
| { |
| int len = strlen((const char*)mre->Text) + 1; |
| if (!(buffer = HeapAlloc( GetProcessHeap(), 0, len ))) return NULL; |
| memcpy( buffer, mre->Text, len ); |
| } |
| TRACE("returning %s\n", wine_dbgstr_a(buffer)); |
| return buffer; |
| } |
| |
| |
| /*********************************************************************** |
| * FormatMessageA (KERNEL32.@) |
| * FIXME: missing wrap, |
| */ |
| DWORD WINAPI FormatMessageA( |
| DWORD dwFlags, |
| LPCVOID lpSource, |
| DWORD dwMessageId, |
| DWORD dwLanguageId, |
| LPSTR lpBuffer, |
| DWORD nSize, |
| va_list* _args ) |
| { |
| LPDWORD args=(LPDWORD)_args; |
| DWORD ret = 0; |
| #if defined(__i386__) || defined(__sparc__) |
| /* This implementation is completely dependent on the format of the va_list on x86 CPUs */ |
| LPSTR target,t; |
| DWORD talloced; |
| LPSTR from,f; |
| DWORD width = dwFlags & FORMAT_MESSAGE_MAX_WIDTH_MASK; |
| BOOL eos = FALSE; |
| CHAR ch; |
| |
| TRACE("(0x%x,%p,%d,0x%x,%p,%d,%p)\n", |
| dwFlags,lpSource,dwMessageId,dwLanguageId,lpBuffer,nSize,args); |
| if ((dwFlags & FORMAT_MESSAGE_FROM_STRING) |
| &&((dwFlags & FORMAT_MESSAGE_FROM_SYSTEM) |
| || (dwFlags & FORMAT_MESSAGE_FROM_HMODULE))) return 0; |
| |
| if (!lpBuffer) |
| { |
| SetLastError(ERROR_NOT_ENOUGH_MEMORY); |
| return 0; |
| } |
| |
| if (width && width != FORMAT_MESSAGE_MAX_WIDTH_MASK) |
| FIXME("line wrapping (%u) not supported.\n", width); |
| from = NULL; |
| if (dwFlags & FORMAT_MESSAGE_FROM_STRING) |
| { |
| from = HeapAlloc( GetProcessHeap(), 0, strlen((LPCSTR)lpSource)+1 ); |
| strcpy( from, (LPCSTR)lpSource ); |
| } |
| else { |
| from = NULL; |
| if (dwFlags & FORMAT_MESSAGE_FROM_HMODULE) |
| from = load_messageA( (HMODULE)lpSource, dwMessageId, dwLanguageId ); |
| if (!from && (dwFlags & FORMAT_MESSAGE_FROM_SYSTEM)) |
| from = load_messageA( kernel32_handle, dwMessageId, dwLanguageId ); |
| |
| if (!from) |
| { |
| SetLastError (ERROR_RESOURCE_LANG_NOT_FOUND); |
| return 0; |
| } |
| } |
| target = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, 100); |
| t = target; |
| talloced= 100; |
| |
| #define ADD_TO_T(c) do { \ |
| *t++=c;\ |
| if ((DWORD)(t-target) == talloced) {\ |
| target = HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,target,talloced*2);\ |
| t = target+talloced;\ |
| talloced*=2;\ |
| }\ |
| } while (0) |
| |
| if (from) { |
| f=from; |
| if (dwFlags & FORMAT_MESSAGE_IGNORE_INSERTS) { |
| while (*f && !eos) |
| ADD_TO_T(*f++); |
| } |
| else { |
| while (*f && !eos) { |
| if (*f=='%') { |
| int insertnr; |
| char *fmtstr,*x,*lastf; |
| DWORD *argliststart; |
| |
| fmtstr = NULL; |
| lastf = f; |
| f++; |
| if (!*f) { |
| ADD_TO_T('%'); |
| continue; |
| } |
| switch (*f) { |
| case '1':case '2':case '3':case '4':case '5': |
| case '6':case '7':case '8':case '9': |
| insertnr=*f-'0'; |
| switch (f[1]) { |
| case '0':case '1':case '2':case '3': |
| case '4':case '5':case '6':case '7': |
| case '8':case '9': |
| f++; |
| insertnr=insertnr*10+*f-'0'; |
| f++; |
| break; |
| default: |
| f++; |
| break; |
| } |
| if (*f=='!') { |
| f++; |
| if (NULL!=(x=strchr(f,'!'))) { |
| *x='\0'; |
| fmtstr=HeapAlloc(GetProcessHeap(),0,strlen(f)+2); |
| sprintf(fmtstr,"%%%s",f); |
| f=x+1; |
| } else { |
| fmtstr=HeapAlloc(GetProcessHeap(),0,strlen(f)+2); |
| sprintf(fmtstr,"%%%s",f); |
| f+=strlen(f); /*at \0*/ |
| } |
| } else { |
| if(!args) break; |
| fmtstr = HeapAlloc(GetProcessHeap(),0,3); |
| strcpy( fmtstr, "%s" ); |
| } |
| if (args) { |
| int sz; |
| LPSTR b; |
| |
| if (dwFlags & FORMAT_MESSAGE_ARGUMENT_ARRAY) |
| argliststart=args+insertnr-1; |
| else |
| argliststart=(*(DWORD**)args)+insertnr-1; |
| |
| /* FIXME: precision and width components are not handled correctly */ |
| if ( (strcmp(fmtstr, "%ls") == 0) || (strcmp(fmtstr,"%S") == 0) ) { |
| sz = WideCharToMultiByte( CP_ACP, 0, *(WCHAR**)argliststart, -1, NULL, 0, NULL, NULL); |
| b = HeapAlloc(GetProcessHeap(), 0, sz); |
| WideCharToMultiByte( CP_ACP, 0, *(WCHAR**)argliststart, -1, b, sz, NULL, NULL); |
| } else { |
| b = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sz = 1000); |
| /* CMF - This makes a BIG assumption about va_list */ |
| TRACE("A BIG assumption\n"); |
| vsnprintf(b, sz, fmtstr, (va_list) argliststart); |
| } |
| for (x=b; *x; x++) ADD_TO_T(*x); |
| |
| HeapFree(GetProcessHeap(),0,b); |
| } else { |
| /* NULL args - copy formatstr |
| * (probably wrong) |
| */ |
| while ((lastf<f)&&(*lastf)) { |
| ADD_TO_T(*lastf++); |
| } |
| } |
| HeapFree(GetProcessHeap(),0,fmtstr); |
| break; |
| case 'n': |
| ADD_TO_T('\r'); |
| ADD_TO_T('\n'); |
| f++; |
| break; |
| case '0': |
| eos = TRUE; |
| f++; |
| break; |
| default: |
| ADD_TO_T(*f++); |
| break; |
| } |
| } else { |
| ch = *f; |
| f++; |
| if (ch == '\r') { |
| if (*f == '\n') |
| f++; |
| if(width) |
| ADD_TO_T(' '); |
| else |
| { |
| ADD_TO_T('\r'); |
| ADD_TO_T('\n'); |
| } |
| } else { |
| if (ch == '\n') |
| { |
| if(width) |
| ADD_TO_T(' '); |
| else |
| { |
| ADD_TO_T('\r'); |
| ADD_TO_T('\n'); |
| } |
| } |
| else |
| ADD_TO_T(ch); |
| } |
| } |
| } |
| } |
| *t='\0'; |
| } |
| talloced = strlen(target)+1; |
| if (nSize && talloced<nSize) { |
| target = HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,target,nSize); |
| } |
| TRACE("-- %s\n",debugstr_a(target)); |
| if (dwFlags & FORMAT_MESSAGE_ALLOCATE_BUFFER) { |
| *((LPVOID*)lpBuffer) = (LPVOID)LocalAlloc(LMEM_ZEROINIT,max(nSize, talloced)); |
| memcpy(*(LPSTR*)lpBuffer,target,talloced); |
| } else { |
| lstrcpynA(lpBuffer,target,nSize); |
| } |
| HeapFree(GetProcessHeap(),0,target); |
| HeapFree(GetProcessHeap(),0,from); |
| ret = (dwFlags & FORMAT_MESSAGE_ALLOCATE_BUFFER) ? strlen(*(LPSTR*)lpBuffer) : strlen(lpBuffer); |
| #endif /* __i386__ */ |
| TRACE("-- returning %d\n", ret); |
| return ret; |
| } |
| #undef ADD_TO_T |
| |
| |
| /*********************************************************************** |
| * FormatMessageW (KERNEL32.@) |
| */ |
| DWORD WINAPI FormatMessageW( |
| DWORD dwFlags, |
| LPCVOID lpSource, |
| DWORD dwMessageId, |
| DWORD dwLanguageId, |
| LPWSTR lpBuffer, |
| DWORD nSize, |
| va_list* _args ) |
| { |
| LPDWORD args=(LPDWORD)_args; |
| #if defined(__i386__) || defined(__sparc__) |
| /* This implementation is completely dependent on the format of the va_list on x86 CPUs */ |
| LPWSTR target,t; |
| DWORD talloced; |
| LPWSTR from,f; |
| DWORD width = dwFlags & FORMAT_MESSAGE_MAX_WIDTH_MASK; |
| BOOL eos = FALSE; |
| WCHAR ch; |
| |
| TRACE("(0x%x,%p,%d,0x%x,%p,%d,%p)\n", |
| dwFlags,lpSource,dwMessageId,dwLanguageId,lpBuffer,nSize,args); |
| if ((dwFlags & FORMAT_MESSAGE_FROM_STRING) |
| &&((dwFlags & FORMAT_MESSAGE_FROM_SYSTEM) |
| || (dwFlags & FORMAT_MESSAGE_FROM_HMODULE))) return 0; |
| |
| if (!lpBuffer) |
| { |
| SetLastError(ERROR_INVALID_PARAMETER); |
| return 0; |
| } |
| |
| if (width && width != FORMAT_MESSAGE_MAX_WIDTH_MASK) |
| FIXME("line wrapping not supported.\n"); |
| from = NULL; |
| if (dwFlags & FORMAT_MESSAGE_FROM_STRING) { |
| from = HeapAlloc( GetProcessHeap(), 0, (strlenW((LPCWSTR)lpSource) + 1) * |
| sizeof(WCHAR) ); |
| strcpyW( from, (LPCWSTR)lpSource ); |
| } |
| else { |
| from = NULL; |
| if (dwFlags & FORMAT_MESSAGE_FROM_HMODULE) |
| from = load_messageW( (HMODULE)lpSource, dwMessageId, dwLanguageId ); |
| if (!from && (dwFlags & FORMAT_MESSAGE_FROM_SYSTEM)) |
| from = load_messageW( kernel32_handle, dwMessageId, dwLanguageId ); |
| |
| if (!from) |
| { |
| SetLastError (ERROR_RESOURCE_LANG_NOT_FOUND); |
| return 0; |
| } |
| } |
| target = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, 100 * sizeof(WCHAR) ); |
| t = target; |
| talloced= 100; |
| |
| #define ADD_TO_T(c) do {\ |
| *t++=c;\ |
| if ((DWORD)(t-target) == talloced) {\ |
| target = HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,target,talloced*2*sizeof(WCHAR));\ |
| t = target+talloced;\ |
| talloced*=2;\ |
| } \ |
| } while (0) |
| |
| if (from) { |
| f=from; |
| if (dwFlags & FORMAT_MESSAGE_IGNORE_INSERTS) { |
| while (*f && !eos) |
| ADD_TO_T(*f++); |
| } |
| else { |
| while (*f && !eos) { |
| if (*f=='%') { |
| int insertnr; |
| WCHAR *fmtstr,*sprintfbuf,*x; |
| DWORD *argliststart; |
| |
| fmtstr = NULL; |
| f++; |
| if (!*f) { |
| ADD_TO_T('%'); |
| continue; |
| } |
| |
| switch (*f) { |
| case '1':case '2':case '3':case '4':case '5': |
| case '6':case '7':case '8':case '9': |
| insertnr=*f-'0'; |
| switch (f[1]) { |
| case '0':case '1':case '2':case '3': |
| case '4':case '5':case '6':case '7': |
| case '8':case '9': |
| f++; |
| insertnr=insertnr*10+*f-'0'; |
| f++; |
| break; |
| default: |
| f++; |
| break; |
| } |
| if (*f=='!') { |
| f++; |
| if (NULL!=(x=strchrW(f,'!'))) { |
| *x='\0'; |
| fmtstr=HeapAlloc( GetProcessHeap(), 0,(strlenW(f)+2)*sizeof(WCHAR)); |
| sprintfW(fmtstr,PCNTFMTWSTR,f); |
| f=x+1; |
| } else { |
| fmtstr=HeapAlloc(GetProcessHeap(),0,(strlenW(f)+2)*sizeof(WCHAR)); |
| sprintfW(fmtstr,PCNTFMTWSTR,f); |
| f+=strlenW(f); /*at \0*/ |
| } |
| } else { |
| if(!args) break; |
| fmtstr = HeapAlloc( GetProcessHeap(),0,3*sizeof(WCHAR)); |
| strcpyW( fmtstr, FMTWSTR ); |
| } |
| if (dwFlags & FORMAT_MESSAGE_ARGUMENT_ARRAY) |
| argliststart=args+insertnr-1; |
| else |
| argliststart=(*(DWORD**)args)+insertnr-1; |
| |
| if (fmtstr[strlenW(fmtstr)-1]=='s' && argliststart[0]) { |
| DWORD xarr[3]; |
| |
| xarr[0]=*(argliststart+0); |
| /* possible invalid pointers */ |
| xarr[1]=*(argliststart+1); |
| xarr[2]=*(argliststart+2); |
| sprintfbuf=HeapAlloc(GetProcessHeap(),0,(strlenW((LPWSTR)argliststart[0])*2+1)*sizeof(WCHAR)); |
| |
| /* CMF - This makes a BIG assumption about va_list */ |
| vsprintfW(sprintfbuf, fmtstr, (va_list) xarr); |
| } else { |
| sprintfbuf=HeapAlloc(GetProcessHeap(),0,100); |
| |
| /* CMF - This makes a BIG assumption about va_list */ |
| vsprintfW(sprintfbuf, fmtstr, (va_list) argliststart); |
| } |
| x=sprintfbuf; |
| while (*x) { |
| ADD_TO_T(*x++); |
| } |
| HeapFree(GetProcessHeap(),0,sprintfbuf); |
| HeapFree(GetProcessHeap(),0,fmtstr); |
| break; |
| case 'n': |
| ADD_TO_T('\r'); |
| ADD_TO_T('\n'); |
| f++; |
| break; |
| case '0': |
| eos = TRUE; |
| f++; |
| break; |
| default: |
| ADD_TO_T(*f++); |
| break; |
| } |
| } else { |
| ch = *f; |
| f++; |
| if (ch == '\r') { |
| if (*f == '\n') |
| f++; |
| if(width) |
| ADD_TO_T(' '); |
| else |
| { |
| ADD_TO_T('\r'); |
| ADD_TO_T('\n'); |
| } |
| } else { |
| if (ch == '\n') |
| { |
| if(width) |
| ADD_TO_T(' '); |
| else |
| { |
| ADD_TO_T('\r'); |
| ADD_TO_T('\n'); |
| } |
| } |
| else |
| ADD_TO_T(ch); |
| } |
| } |
| } |
| } |
| *t='\0'; |
| } |
| talloced = strlenW(target)+1; |
| if (nSize && talloced<nSize) |
| target = HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,target,nSize*sizeof(WCHAR)); |
| if (dwFlags & FORMAT_MESSAGE_ALLOCATE_BUFFER) { |
| /* nSize is the MINIMUM size */ |
| DWORD len = strlenW(target) + 1; |
| *((LPVOID*)lpBuffer) = (LPVOID)LocalAlloc(LMEM_ZEROINIT,len*sizeof(WCHAR)); |
| strcpyW(*(LPWSTR*)lpBuffer, target); |
| } |
| else lstrcpynW(lpBuffer, target, nSize); |
| |
| HeapFree(GetProcessHeap(),0,target); |
| HeapFree(GetProcessHeap(),0,from); |
| TRACE("ret=%s\n", wine_dbgstr_w((dwFlags & FORMAT_MESSAGE_ALLOCATE_BUFFER) ? |
| *(LPWSTR*)lpBuffer : lpBuffer)); |
| return (dwFlags & FORMAT_MESSAGE_ALLOCATE_BUFFER) ? |
| strlenW(*(LPWSTR*)lpBuffer): |
| strlenW(lpBuffer); |
| #else |
| return 0; |
| #endif /* __i386__ */ |
| } |
| #undef ADD_TO_T |
