mirror of https://github.com/lianthony/NT4.0
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
448 lines
14 KiB
448 lines
14 KiB
/***
|
|
*aw_map.c - A & W versions of LCMapString.
|
|
*
|
|
* Copyright (c) 1993-1994, Microsoft Corporation. All rights reserved.
|
|
*
|
|
*Purpose:
|
|
* Use either LCMapStringA or LCMapStringW depending on which is available
|
|
*
|
|
*Revision History:
|
|
* 09-14-93 CFW Module created.
|
|
* 09-17-93 CFW Use unsigned chars.
|
|
* 09-23-93 CFW Correct NLS API params and comments about same.
|
|
* 10-07-93 CFW Optimize WideCharToMultiByte, use NULL default char.
|
|
* 10-22-93 CFW Test for invalid MB chars using global preset flag.
|
|
* 11-09-93 CFW Allow user to pass in code page.
|
|
* 11-18-93 CFW Test for entry point function stubs.
|
|
* 02-23-94 CFW Use W flavor whenever possible.
|
|
* 03-31-94 CFW Include awint.h.
|
|
* 07-26-94 CFW Bug fix #14730, LCMapString goes past NULLs.
|
|
* 12-21-94 CFW Remove invalid MB chars NT 3.1 hack.
|
|
* 12-27-94 CFW Call direct, all OS's have stubs.
|
|
* 01-10-95 CFW Debug CRT allocs.
|
|
*
|
|
*******************************************************************************/
|
|
|
|
#include <cruntime.h>
|
|
#include <internal.h>
|
|
#include <stdlib.h>
|
|
#include <setlocal.h>
|
|
#include <awint.h>
|
|
#include <dbgint.h>
|
|
|
|
#define USE_W 1
|
|
#define USE_A 2
|
|
|
|
/***
|
|
*int __cdecl strncnt - count characters in a string, up to n.
|
|
*
|
|
*Purpose:
|
|
* Internal local support function. Counts characters in string before NULL.
|
|
* If NULL not found in n chars, then return n.
|
|
*
|
|
*Entry:
|
|
* const char *string - start of string
|
|
* int n - byte count
|
|
*
|
|
*Exit:
|
|
* returns number of bytes from start of string to
|
|
* NULL (exclusive), up to n.
|
|
*
|
|
*Exceptions:
|
|
*
|
|
*******************************************************************************/
|
|
|
|
static int __cdecl strncnt (
|
|
const char *string,
|
|
int cnt
|
|
)
|
|
{
|
|
int n = cnt;
|
|
char *cp = (char *)string;
|
|
|
|
while (n-- && *cp)
|
|
cp++;
|
|
|
|
if (!*cp)
|
|
return cp - string;
|
|
return cnt;
|
|
}
|
|
|
|
/***
|
|
*int __cdecl wcsncnt - count wide characters in a string, up to n.
|
|
*
|
|
*Purpose:
|
|
* Internal local support function. Counts characters in string before NULL.
|
|
* If NULL not found in n chars, then return n.
|
|
*
|
|
*Entry:
|
|
* const wchar_t *string - start of string
|
|
* int n - byte count
|
|
*
|
|
*Exit:
|
|
* returns number of wide characaters from start of string to
|
|
* NULL (exclusive), up to n.
|
|
*
|
|
*Exceptions:
|
|
*
|
|
*******************************************************************************/
|
|
|
|
static int __cdecl wcsncnt (
|
|
const wchar_t *string,
|
|
int cnt
|
|
)
|
|
{
|
|
int n = cnt;
|
|
wchar_t *cp = (wchar_t *)string;
|
|
|
|
while (n-- && *cp)
|
|
cp++;
|
|
|
|
if (!*cp)
|
|
return cp - string;
|
|
return cnt;
|
|
}
|
|
|
|
/***
|
|
*int __cdecl __crtLCMapStringW - Get type information about a wide string.
|
|
*
|
|
*Purpose:
|
|
* Internal support function. Assumes info in wide string format. Tries
|
|
* to use NLS API call LCMapStringW if available and uses LCMapStringA
|
|
* if it must. If neither are available it fails and returns 0.
|
|
*
|
|
*Entry:
|
|
* LCID Locale - locale context for the comparison.
|
|
* DWORD dwMapFlags - see NT\Chicago docs
|
|
* LPCWSTR lpSrcStr - pointer to string to be mapped
|
|
* int cchSrc - wide char (word) count of input string (including NULL if any)
|
|
* (-1 if NULL terminated)
|
|
* LPWSTR lpDestStr - pointer to memory to store mapping
|
|
* int cchDest - wide char (word) count of buffer (including NULL)
|
|
* int code_page - for MB/WC conversion. If 0, use __lc_codepage
|
|
*
|
|
* NOTE: if LCMAP_SORTKEY is specified, then cchDest refers to number of BYTES,
|
|
* not number of wide chars. The return string will be a series of bytes
|
|
* with a NULL byte terminator.
|
|
*
|
|
*Exit:
|
|
* Success: if LCMAP_SORKEY:
|
|
* number of bytes written to lpDestStr (including NULL byte terminator)
|
|
* else
|
|
* number of wide characters written to lpDestStr (including NULL)
|
|
* Failure: 0
|
|
*
|
|
*Exceptions:
|
|
*
|
|
*******************************************************************************/
|
|
|
|
int __cdecl __crtLCMapStringW(
|
|
LCID Locale,
|
|
DWORD dwMapFlags,
|
|
LPCWSTR lpSrcStr,
|
|
int cchSrc,
|
|
LPWSTR lpDestStr,
|
|
int cchDest,
|
|
int code_page
|
|
)
|
|
{
|
|
static int f_use = 0;
|
|
|
|
/*
|
|
* Look for unstubbed 'preferred' flavor. Otherwise use available flavor.
|
|
* Must actually call the function to ensure it's not a stub.
|
|
*/
|
|
|
|
if (0 == f_use)
|
|
{
|
|
if (0 != LCMapStringW(0, LCMAP_LOWERCASE, L"\0", 1, NULL, 0))
|
|
f_use = USE_W;
|
|
|
|
else if (0 != LCMapStringA(0, LCMAP_LOWERCASE, "\0", 1, NULL, 0))
|
|
f_use = USE_A;
|
|
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* LCMapString will map past NULL. Must find NULL if in string
|
|
* before cchSrc wide characters.
|
|
*/
|
|
if (cchSrc > 0)
|
|
cchSrc = wcsncnt(lpSrcStr, cchSrc);
|
|
|
|
/* Use "W" version */
|
|
|
|
if (USE_W == f_use)
|
|
{
|
|
return LCMapStringW(Locale, dwMapFlags, lpSrcStr, cchSrc, lpDestStr, cchDest);
|
|
}
|
|
|
|
/* Use "A" version */
|
|
|
|
if (USE_A == f_use)
|
|
{
|
|
int retval;
|
|
int inbuff_size;
|
|
int outbuff_size;
|
|
unsigned char *inbuffer = NULL;
|
|
unsigned char *outbuffer = NULL;
|
|
|
|
/*
|
|
* Convert string and return the requested information. Note that
|
|
* we are converting to a multibyte string so there is not a
|
|
* one-to-one correspondence between number of wide chars in the
|
|
* input string and the number of *bytes* in the buffer. However,
|
|
* there had *better be* a one-to-one correspondence between the
|
|
* number of wide characters and the number of multibyte characters
|
|
* (enforced by WC_SEPCHARS) in the buffer or the resulting mapped
|
|
* string will be worthless to the user.
|
|
*
|
|
*/
|
|
|
|
/*
|
|
* Use __lc_codepage for conversion if code_page not specified
|
|
*/
|
|
|
|
if (0 == code_page)
|
|
code_page = __lc_codepage;
|
|
|
|
/* find out how big a buffer we need (includes NULL if any) */
|
|
if (0 == (inbuff_size = WideCharToMultiByte(code_page,
|
|
WC_COMPOSITECHECK | WC_SEPCHARS, lpSrcStr, cchSrc,
|
|
NULL, 0, NULL, NULL)))
|
|
return 0;
|
|
|
|
/* allocate enough space for chars */
|
|
if (NULL == (inbuffer = (unsigned char *)
|
|
_malloc_crt(inbuff_size * sizeof(char))))
|
|
return 0;
|
|
|
|
/* do the conversion */
|
|
if (0 == WideCharToMultiByte(code_page,
|
|
WC_COMPOSITECHECK | WC_SEPCHARS, lpSrcStr, cchSrc,
|
|
inbuffer, inbuff_size, NULL, NULL))
|
|
goto error_cleanup;
|
|
|
|
/* get size required for string mapping */
|
|
if (0 == (outbuff_size = LCMapStringA(Locale, dwMapFlags, inbuffer, inbuff_size, NULL, 0)))
|
|
goto error_cleanup;
|
|
|
|
/* allocate enough space for chars and NULL */
|
|
if (NULL == (outbuffer = (unsigned char *)
|
|
_malloc_crt(outbuff_size * sizeof(char))))
|
|
goto error_cleanup;
|
|
|
|
/* do string mapping */
|
|
if (0 == LCMapStringA(Locale, dwMapFlags, inbuffer, inbuff_size, outbuffer, outbuff_size))
|
|
goto error_cleanup;
|
|
|
|
if (dwMapFlags & LCMAP_SORTKEY)
|
|
{
|
|
/* outbuff_size > cchDest is allowed */
|
|
retval = outbuff_size;
|
|
|
|
if (0 != cchDest)
|
|
/* SORTKEY returns BYTES, just copy */
|
|
strncpy((char *)lpDestStr, (char *)outbuffer, cchDest < outbuff_size ? cchDest : outbuff_size);
|
|
}
|
|
else {
|
|
if (0 == cchDest)
|
|
{
|
|
/* get size required */
|
|
if (0 == (retval = MultiByteToWideChar(code_page,
|
|
MB_PRECOMPOSED, outbuffer, outbuff_size, NULL, 0)))
|
|
goto error_cleanup;
|
|
} else {
|
|
/* convert mapping */
|
|
if (0 == (retval = MultiByteToWideChar(code_page,
|
|
MB_PRECOMPOSED, outbuffer, outbuff_size, lpDestStr, cchDest)))
|
|
goto error_cleanup;
|
|
}
|
|
}
|
|
|
|
_free_crt(inbuffer);
|
|
_free_crt(outbuffer);
|
|
return retval;
|
|
|
|
error_cleanup:
|
|
_free_crt(inbuffer);
|
|
_free_crt(outbuffer);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
/***
|
|
*int __cdecl __crtLCMapStringA - Get type information about an ANSI string.
|
|
*
|
|
*Purpose:
|
|
* Internal support function. Assumes info in ANSI string format. Tries
|
|
* to use NLS API call LCMapStringA if available and uses LCMapStringW
|
|
* if it must. If neither are available it fails and returns 0.
|
|
*
|
|
*Entry:
|
|
* LCID Locale - locale context for the comparison.
|
|
* DWORD dwMapFlags - see NT\Chicago docs
|
|
* LPCSTR lpSrcStr - pointer to string to be mapped
|
|
* int cchSrc - wide char (word) count of input string (including NULL if any)
|
|
* (-1 if NULL terminated)
|
|
* LPSTR lpDestStr - pointer to memory to store mapping
|
|
* int cchDest - char (byte) count of buffer (including NULL)
|
|
* int code_page - for MB/WC conversion. If 0, use __lc_codepage
|
|
*
|
|
*Exit:
|
|
* Success: number of chars written to lpDestStr (including NULL)
|
|
* Failure: 0
|
|
*
|
|
*Exceptions:
|
|
*
|
|
*******************************************************************************/
|
|
|
|
int __cdecl __crtLCMapStringA(
|
|
LCID Locale,
|
|
DWORD dwMapFlags,
|
|
LPCSTR lpSrcStr,
|
|
int cchSrc,
|
|
LPSTR lpDestStr,
|
|
int cchDest,
|
|
int code_page
|
|
)
|
|
{
|
|
static int f_use = 0;
|
|
|
|
/*
|
|
* Look for unstubbed 'preferred' flavor. Otherwise use available flavor.
|
|
* Must actually call the function to ensure it's not a stub.
|
|
*/
|
|
|
|
if (0 == f_use)
|
|
{
|
|
if (0 != LCMapStringA(0, LCMAP_LOWERCASE, "\0", 1, NULL, 0))
|
|
f_use = USE_A;
|
|
|
|
else if (0 != LCMapStringW(0, LCMAP_LOWERCASE, L"\0", 1, NULL, 0))
|
|
f_use = USE_W;
|
|
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* LCMapString will map past NULL. Must find NULL if in string
|
|
* before cchSrc characters.
|
|
*/
|
|
if (cchSrc > 0)
|
|
cchSrc = strncnt(lpSrcStr, cchSrc);
|
|
|
|
/* Use "A" version */
|
|
|
|
if (USE_A == f_use)
|
|
{
|
|
return LCMapStringA(Locale, dwMapFlags, lpSrcStr, cchSrc, lpDestStr, cchDest);
|
|
}
|
|
|
|
/* Use "W" version */
|
|
|
|
if (USE_W == f_use)
|
|
{
|
|
int retval;
|
|
int inbuff_size;
|
|
int outbuff_size;
|
|
wchar_t *inwbuffer = NULL;
|
|
wchar_t *outwbuffer = NULL;
|
|
|
|
/*
|
|
* Convert string and return the requested information. Note that
|
|
* we are converting to a wide string so there is not a
|
|
* one-to-one correspondence between number of wide chars in the
|
|
* input string and the number of *bytes* in the buffer. However,
|
|
* there had *better be* a one-to-one correspondence between the
|
|
* number of wide characters and the number of multibyte characters
|
|
* or the resulting mapped string will be worthless to the user.
|
|
*
|
|
*/
|
|
|
|
/*
|
|
* Use __lc_codepage for conversion if code_page not specified
|
|
*/
|
|
|
|
if (0 == code_page)
|
|
code_page = __lc_codepage;
|
|
|
|
/* find out how big a buffer we need (includes NULL if any) */
|
|
if (0 == (inbuff_size = MultiByteToWideChar(code_page,
|
|
MB_PRECOMPOSED|MB_ERR_INVALID_CHARS, lpSrcStr, cchSrc, NULL, 0)))
|
|
return 0;
|
|
|
|
/* allocate enough space for wide chars */
|
|
if (NULL == (inwbuffer = (wchar_t *)
|
|
_malloc_crt(inbuff_size * sizeof(wchar_t))))
|
|
return 0;
|
|
|
|
/* do the conversion */
|
|
if (0 == MultiByteToWideChar(code_page,
|
|
MB_PRECOMPOSED, lpSrcStr, cchSrc, inwbuffer, inbuff_size))
|
|
goto error_cleanup;
|
|
|
|
/* get size required for string mapping */
|
|
if (0 == (retval = LCMapStringW(Locale, dwMapFlags,
|
|
inwbuffer, inbuff_size, NULL, 0)))
|
|
goto error_cleanup;
|
|
|
|
|
|
if (dwMapFlags & LCMAP_SORTKEY)
|
|
{
|
|
/* retval is size in BYTES */
|
|
|
|
if (0 != cchDest)
|
|
{
|
|
if (retval > cchDest)
|
|
goto error_cleanup;
|
|
|
|
/* do string mapping */
|
|
if (0 == LCMapStringW(Locale, dwMapFlags,
|
|
inwbuffer, inbuff_size, (LPWSTR)lpDestStr, cchDest))
|
|
goto error_cleanup;
|
|
}
|
|
}
|
|
else {
|
|
/* retval is size in wide chars */
|
|
|
|
outbuff_size = retval;
|
|
|
|
/* allocate enough space for wide chars (includes NULL if any) */
|
|
if (NULL == (outwbuffer = (wchar_t *)
|
|
_malloc_crt(outbuff_size * sizeof(wchar_t))))
|
|
goto error_cleanup;
|
|
|
|
/* do string mapping */
|
|
if (0 == LCMapStringW(Locale, dwMapFlags, inwbuffer, inbuff_size, outwbuffer, outbuff_size))
|
|
goto error_cleanup;
|
|
|
|
if (0 == cchDest)
|
|
{
|
|
/* get size required */
|
|
if (0 == (retval = WideCharToMultiByte(code_page,
|
|
WC_COMPOSITECHECK | WC_SEPCHARS, outwbuffer, outbuff_size,
|
|
NULL, 0, NULL, NULL)))
|
|
goto error_cleanup;
|
|
} else {
|
|
/* convert mapping */
|
|
if (0 == (retval = WideCharToMultiByte(code_page,
|
|
WC_COMPOSITECHECK | WC_SEPCHARS, outwbuffer, outbuff_size,
|
|
lpDestStr, cchDest, NULL, NULL)))
|
|
goto error_cleanup;
|
|
}
|
|
}
|
|
|
|
_free_crt(inwbuffer);
|
|
_free_crt(outwbuffer);
|
|
return retval;
|
|
|
|
error_cleanup:
|
|
_free_crt(inwbuffer);
|
|
_free_crt(outwbuffer);
|
|
return 0;
|
|
}
|
|
}
|