413 lines
10 KiB
C
413 lines
10 KiB
C
|
/*++
|
||
|
|
|
||
|
|
Copyright (c) 1991 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
message.c
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
This module provides support routines to map DosxxxMessage APIs to
|
||
|
|
the FormatMessage syntax and semantics.
|
||
|
|
|
||
|
|
Author:
|
||
|
|
|
||
|
|
Dan Hinsley (DanHi) 24-Sept-1991
|
||
|
|
|
||
|
|
Environment:
|
||
|
|
|
||
|
|
Contains NT specific code.
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
#define ERROR_MR_MSG_TOO_LONG 316
|
||
|
|
#define ERROR_MR_UN_ACC_MSGF 318
|
||
|
|
#define ERROR_MR_INV_IVCOUNT 320
|
||
|
|
|
||
|
|
#include <nt.h>
|
||
|
|
#include <ntrtl.h>
|
||
|
|
#include <nturtl.h>
|
||
|
|
#define NOMINMAX // Avoid windows vs. stdlib.h conflicts.
|
||
|
|
#include <windef.h>
|
||
|
|
#include <winbase.h>
|
||
|
|
#include <winnls.h>
|
||
|
|
#include <lmcons.h>
|
||
|
|
#include <lmerr.h>
|
||
|
|
#include <netdebug.h> // NetpKdPrint
|
||
|
|
#include <netlib.h> // NetpMemory*
|
||
|
|
#include <netlibnt.h> // NetpNtStatusToApiStatus
|
||
|
|
#include <string.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h> // itoa
|
||
|
|
#include <tstring.h>
|
||
|
|
|
||
|
|
//
|
||
|
|
// forward declare
|
||
|
|
//
|
||
|
|
DWORD MyAllocUnicode( LPSTR pszAscii, LPWSTR * ppwszUnicode ) ;
|
||
|
|
|
||
|
|
DWORD MyAllocUnicodeVector( LPSTR * ppszAscii,
|
||
|
|
LPWSTR* ppwszUnicode,
|
||
|
|
UINT cpwszUnicode ) ;
|
||
|
|
|
||
|
|
VOID MyFreeUnicode( LPWSTR pwszUnicode ) ;
|
||
|
|
|
||
|
|
VOID MyFreeUnicodeVector( LPWSTR * ppwsz, UINT cpwsz ) ;
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// 100 is plenty since FormatMessage only take 99 & old DosGetMessage 9.
|
||
|
|
//
|
||
|
|
#define MAX_INSERT_STRINGS (100)
|
||
|
|
|
||
|
|
WORD
|
||
|
|
DosGetMessage(
|
||
|
|
IN LPSTR * InsertionStrings,
|
||
|
|
IN WORD NumberofStrings,
|
||
|
|
OUT LPBYTE Buffer,
|
||
|
|
IN WORD BufferLength,
|
||
|
|
IN WORD MessageId,
|
||
|
|
IN LPTSTR FileName,
|
||
|
|
OUT PWORD pMessageLength
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
This maps the OS/2 DosGetMessage API to the NT FormatMessage API.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
InsertionStrings - Pointer to an array of strings that will be used
|
||
|
|
to replace the %n's in the message.
|
||
|
|
|
||
|
|
NumberofStrings - The number of insertion strings.
|
||
|
|
|
||
|
|
Buffer - The buffer to put the message into.
|
||
|
|
|
||
|
|
BufferLength - The length of the supplied buffer.
|
||
|
|
|
||
|
|
MessageId - The message number to retrieve.
|
||
|
|
|
||
|
|
FileName - The name of the message file to get the message from.
|
||
|
|
|
||
|
|
pMessageLength - A pointer to return the length of the returned message.
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
NERR_Success
|
||
|
|
ERROR_MR_MSG_TOO_LONG
|
||
|
|
ERROR_MR_INV_IVCOUNT
|
||
|
|
ERROR_MR_UN_ACC_MSGF
|
||
|
|
ERROR_MR_MID_NOT_FOUND
|
||
|
|
ERROR_INVALID_PARAMETER
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
|
||
|
|
DWORD dwFlags = FORMAT_MESSAGE_ARGUMENT_ARRAY;
|
||
|
|
DWORD Status, i;
|
||
|
|
LPWSTR UnicodeIStrings[MAX_INSERT_STRINGS] ;
|
||
|
|
LPWSTR UnicodeBuffer = NULL;
|
||
|
|
LPWSTR UnicodeNumberString = NULL ;
|
||
|
|
CHAR NumberString [18];
|
||
|
|
|
||
|
|
static HANDLE lpSource = NULL ;
|
||
|
|
static TCHAR CurrentMsgFile[MAX_PATH] = {0,} ;
|
||
|
|
|
||
|
|
//
|
||
|
|
// init clear the output string
|
||
|
|
//
|
||
|
|
Status = NERR_Success;
|
||
|
|
if (BufferLength)
|
||
|
|
Buffer[0] = '\0' ;
|
||
|
|
if (pMessageLength)
|
||
|
|
*pMessageLength = 0;
|
||
|
|
|
||
|
|
//
|
||
|
|
// make sure we are not over loaded & allocate
|
||
|
|
// memory for the Unicode buffer
|
||
|
|
//
|
||
|
|
if (NumberofStrings > MAX_INSERT_STRINGS)
|
||
|
|
return ERROR_INVALID_PARAMETER ;
|
||
|
|
if (!(UnicodeBuffer = NetpMemoryAllocate(BufferLength * sizeof(WCHAR))))
|
||
|
|
return ERROR_NOT_ENOUGH_MEMORY ;
|
||
|
|
|
||
|
|
//
|
||
|
|
// init the string table & map the strings to unicode
|
||
|
|
//
|
||
|
|
for (i = 0; i < MAX_INSERT_STRINGS; i++)
|
||
|
|
UnicodeIStrings[i] = NULL ;
|
||
|
|
Status = MyAllocUnicodeVector(InsertionStrings,
|
||
|
|
UnicodeIStrings,
|
||
|
|
NumberofStrings) ;
|
||
|
|
if (Status)
|
||
|
|
goto ExitPoint ;
|
||
|
|
|
||
|
|
//
|
||
|
|
// See if they want to get the message from the system message file.
|
||
|
|
//
|
||
|
|
|
||
|
|
if (! STRCMP(FileName, OS2MSG_FILENAME)) {
|
||
|
|
dwFlags |= FORMAT_MESSAGE_FROM_SYSTEM;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// They want it from a separate message file. Get a handle to DLL
|
||
|
|
// If its for the same file as before, dont reload.
|
||
|
|
//
|
||
|
|
if (!(lpSource && !STRCMP(CurrentMsgFile,FileName)))
|
||
|
|
{
|
||
|
|
if (lpSource)
|
||
|
|
{
|
||
|
|
FreeLibrary(lpSource) ;
|
||
|
|
}
|
||
|
|
STRCPY(CurrentMsgFile, FileName) ;
|
||
|
|
lpSource = LoadLibraryEx(FileName, NULL, LOAD_LIBRARY_AS_DATAFILE);
|
||
|
|
if (!lpSource)
|
||
|
|
{
|
||
|
|
Status = ERROR_MR_UN_ACC_MSGF;
|
||
|
|
goto ExitPoint ;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
dwFlags |= FORMAT_MESSAGE_FROM_HMODULE;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// If they just want to get the message back for later formatting,
|
||
|
|
// ignore the insert strings.
|
||
|
|
//
|
||
|
|
if (NumberofStrings == 0)
|
||
|
|
{
|
||
|
|
dwFlags |= FORMAT_MESSAGE_IGNORE_INSERTS;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// call the Unicode version
|
||
|
|
//
|
||
|
|
*pMessageLength = (WORD) FormatMessageW(dwFlags,
|
||
|
|
(LPVOID) lpSource,
|
||
|
|
(DWORD) MessageId,
|
||
|
|
0, // LanguageId defaulted
|
||
|
|
UnicodeBuffer,
|
||
|
|
(DWORD)BufferLength,
|
||
|
|
(va_list *)UnicodeIStrings);
|
||
|
|
|
||
|
|
//
|
||
|
|
// If it failed get the return code and map it to an OS/2 equivalent
|
||
|
|
//
|
||
|
|
|
||
|
|
if (*pMessageLength == 0)
|
||
|
|
{
|
||
|
|
UnicodeBuffer[0] = 0 ;
|
||
|
|
Status = GetLastError();
|
||
|
|
if (Status == ERROR_MR_MID_NOT_FOUND)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// get the message number in Unicode
|
||
|
|
//
|
||
|
|
_itoa(MessageId, NumberString, 16);
|
||
|
|
Status = MyAllocUnicode(NumberString, &UnicodeNumberString) ;
|
||
|
|
if (Status)
|
||
|
|
goto ExitPoint ;
|
||
|
|
|
||
|
|
//
|
||
|
|
// re-setup to get it from the system. use the not found message
|
||
|
|
//
|
||
|
|
dwFlags = FORMAT_MESSAGE_ARGUMENT_ARRAY |
|
||
|
|
FORMAT_MESSAGE_FROM_SYSTEM;
|
||
|
|
MessageId = ERROR_MR_MID_NOT_FOUND ;
|
||
|
|
|
||
|
|
//
|
||
|
|
// setup insert strings
|
||
|
|
//
|
||
|
|
MyFreeUnicodeVector(UnicodeIStrings, NumberofStrings) ;
|
||
|
|
UnicodeIStrings[0] = UnicodeNumberString ;
|
||
|
|
UnicodeIStrings[1] = FileName ;
|
||
|
|
|
||
|
|
//
|
||
|
|
// recall the API
|
||
|
|
//
|
||
|
|
*pMessageLength = (WORD) FormatMessageW(dwFlags,
|
||
|
|
(LPVOID) lpSource,
|
||
|
|
(DWORD) MessageId,
|
||
|
|
0, // LanguageId defaulted
|
||
|
|
UnicodeBuffer,
|
||
|
|
(DWORD)BufferLength,
|
||
|
|
(va_list *)UnicodeIStrings);
|
||
|
|
UnicodeIStrings[1] = NULL ;
|
||
|
|
|
||
|
|
//
|
||
|
|
// revert to original error
|
||
|
|
//
|
||
|
|
Status = ERROR_MR_MID_NOT_FOUND ;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (UnicodeBuffer[0])
|
||
|
|
{
|
||
|
|
BOOL fUsedDefault;
|
||
|
|
|
||
|
|
*pMessageLength = (WORD)WideCharToMultiByte(CP_OEMCP,
|
||
|
|
0,
|
||
|
|
UnicodeBuffer,
|
||
|
|
-1,
|
||
|
|
Buffer,
|
||
|
|
BufferLength,
|
||
|
|
NULL, // use system default char
|
||
|
|
&fUsedDefault );
|
||
|
|
if (*pMessageLength == 0)
|
||
|
|
{
|
||
|
|
Status = GetLastError() ;
|
||
|
|
goto ExitPoint ;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
ExitPoint:
|
||
|
|
//
|
||
|
|
// note: UnicodeNumberString dont need to be freed
|
||
|
|
// since if used, they would be in the UnicodeIStrings which is whacked
|
||
|
|
//
|
||
|
|
if (UnicodeBuffer) NetpMemoryFree(UnicodeBuffer) ;
|
||
|
|
MyFreeUnicodeVector(UnicodeIStrings, NumberofStrings) ;
|
||
|
|
return (WORD)(Status);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/************** misc unicode helper routines *************************/
|
||
|
|
|
||
|
|
|
||
|
|
/*
|
||
|
|
* MyAllocUnicode
|
||
|
|
* Given a MBCS string, allocate a new Unicode translation of that string
|
||
|
|
*
|
||
|
|
* IN
|
||
|
|
* pszAscii - pointer to original MBCS string
|
||
|
|
* ppwszUnicode - pointer to cell to hold new Unicode string addr
|
||
|
|
* OUT
|
||
|
|
* ppwszUnicode - contains new Unicode string
|
||
|
|
*
|
||
|
|
* RETURNS
|
||
|
|
* Error code, 0 if successful.
|
||
|
|
*
|
||
|
|
* The client must free the allocated string with MyFreeUnicode.
|
||
|
|
*/
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
MyAllocUnicode(
|
||
|
|
LPSTR pszAscii,
|
||
|
|
LPWSTR * ppwszUnicode )
|
||
|
|
{
|
||
|
|
UINT count;
|
||
|
|
BYTE * pbAlloc;
|
||
|
|
INT cbAscii;
|
||
|
|
|
||
|
|
if (pszAscii == NULL)
|
||
|
|
{
|
||
|
|
*ppwszUnicode = NULL;
|
||
|
|
return NERR_Success;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Calculate size of Unicode string.
|
||
|
|
cbAscii = strlen(pszAscii)+1;
|
||
|
|
pbAlloc = (BYTE *) NetpMemoryAllocate(sizeof(WCHAR) * cbAscii) ;
|
||
|
|
if (!pbAlloc)
|
||
|
|
return ERROR_NOT_ENOUGH_MEMORY ;
|
||
|
|
|
||
|
|
*ppwszUnicode = (LPWSTR)pbAlloc;
|
||
|
|
|
||
|
|
count = MultiByteToWideChar(CP_OEMCP,
|
||
|
|
MB_PRECOMPOSED,
|
||
|
|
pszAscii,
|
||
|
|
cbAscii,
|
||
|
|
*ppwszUnicode,
|
||
|
|
cbAscii);
|
||
|
|
if (count == 0)
|
||
|
|
{
|
||
|
|
*ppwszUnicode = NULL;
|
||
|
|
NetpMemoryFree(pbAlloc);
|
||
|
|
return ( GetLastError() );
|
||
|
|
}
|
||
|
|
|
||
|
|
return NERR_Success;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* MyAllocUnicode
|
||
|
|
* Given an array of MBCS strings, allocate a new array of Unicode strings
|
||
|
|
*
|
||
|
|
* IN
|
||
|
|
* ppszAscii - array of MBCS strings
|
||
|
|
* cpwszUnicode - number of elements to translate
|
||
|
|
* OUT
|
||
|
|
* ppwszUnicode - output array of UnicodeStrings
|
||
|
|
*
|
||
|
|
* RETURNS
|
||
|
|
* Error code, 0 if successful.
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
DWORD
|
||
|
|
MyAllocUnicodeVector(
|
||
|
|
LPSTR * ppszAscii,
|
||
|
|
LPWSTR* ppwszUnicode,
|
||
|
|
UINT cpwszUnicode )
|
||
|
|
{
|
||
|
|
DWORD err;
|
||
|
|
UINT i;
|
||
|
|
|
||
|
|
for (i = 0; i < cpwszUnicode; ++i)
|
||
|
|
{
|
||
|
|
err = MyAllocUnicode(ppszAscii[i], ppwszUnicode+i);
|
||
|
|
if (err)
|
||
|
|
{
|
||
|
|
MyFreeUnicodeVector(ppwszUnicode,i);
|
||
|
|
return err;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return NERR_Success;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/*
|
||
|
|
* MyFreeUnicode
|
||
|
|
* Deallocates a Unicode string alloc'd by MyAllocUnicode (q.v.).
|
||
|
|
*
|
||
|
|
* IN
|
||
|
|
* pwszUnicode - pointer to the alloc'd string
|
||
|
|
*/
|
||
|
|
|
||
|
|
VOID
|
||
|
|
MyFreeUnicode( LPWSTR pwszUnicode )
|
||
|
|
{
|
||
|
|
if (pwszUnicode != NULL)
|
||
|
|
NetpMemoryFree((LPBYTE)pwszUnicode);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/*
|
||
|
|
* MyFreeUnicodeVector
|
||
|
|
* Deallocates an array of Unicodes string alloc'd by MyAllocUnicodeVector.
|
||
|
|
*
|
||
|
|
* IN
|
||
|
|
* pp - pointer to the array of strings
|
||
|
|
*/
|
||
|
|
|
||
|
|
VOID
|
||
|
|
MyFreeUnicodeVector(
|
||
|
|
LPWSTR * ppwsz,
|
||
|
|
UINT cpwsz )
|
||
|
|
{
|
||
|
|
while (cpwsz-- > 0)
|
||
|
|
{
|
||
|
|
MyFreeUnicode(*ppwsz);
|
||
|
|
*ppwsz = NULL ;
|
||
|
|
ppwsz++ ;
|
||
|
|
}
|
||
|
|
}
|