567 lines
20 KiB
C
567 lines
20 KiB
C
/*
|
|
* 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
|
|
*
|
|
* 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) = 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) = 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
|