|
|
/***
*a_map.c - A version of LCMapString.** Copyright (c) 1993-2001, 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.* 02-15-97 RDK For narrow mapping, try W version first so Windows NT* can process nonANSI codepage correctly.* 03-16-97 RDK Added error flag to __crtLCMapStringA.* 05-09-97 GJF Renamed and moved __crtLCMapStringW into a separate * file. Revised to use _alloca instead of malloc. Also,* reformatted.* 05-27-98 GJF Changed strncnt() so that it will never examine the* (cnt + 1)-th byte of the string.* 08-18-98 GJF Use _malloc_crt if _alloca fails.* 04-28-99 GJF Changed dwFlags arg value to 0 in WideCharToMultiByte* calls to avoid problems with codepage 1258 on NT 5.0.* 12-10-99 GB Added support for recovery from stack overflow around * _alloca().* 05-17-00 GB Use ERROR_CALL_NOT_IMPLEMENTED for existance of W API* 08-23-00 GB Fixed bug with non Ansi CP on Win9x.* 07-07-01 BWT Fix error path (set ret=FALSE before branching, not after)********************************************************************************/
#include <cruntime.h>
#include <internal.h>
#include <stdlib.h>
#include <setlocal.h>
#include <locale.h>
#include <awint.h>
#include <dbgint.h>
#include <malloc.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 is 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++;
return cnt - n - 1;}
/***
*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* BOOL bError - TRUE if MB_ERR_INVALID_CHARS set on call to* MultiByteToWideChar when GetStringTypeW used.**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, BOOL bError ){ 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. * (Always try wide version first so WinNT can process codepage correctly.) */ if (0 == f_use) { if (0 != LCMapStringW(0, LCMAP_LOWERCASE, L"\0", 1, NULL, 0)) f_use = USE_W; else if (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) f_use = USE_A; }
/*
* 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 || f_use == 0) { char *cbuffer = NULL; char *cbuffer1 = NULL; int ret; int malloc_flag1 = 0; int AnsiCP; int buff_size;
if (0 == Locale) Locale = __lc_handle[LC_CTYPE]; if (0 == code_page) code_page = __lc_codepage;
if ( -1 == (AnsiCP = __ansicp(Locale))) return FALSE; /* LCMapStringA uses ANSI code page to map the string. Check if
* codepage is ansi, if not convert the input string to ansi * codepage then map to a temporary string and then convert temp * string to DestStr. */ if ( AnsiCP != code_page) { cbuffer = __convertcp(code_page, AnsiCP, lpSrcStr, &cchSrc, NULL, 0); if (cbuffer == NULL) return FALSE; lpSrcStr = cbuffer; if (0 == (buff_size = LCMapStringA( Locale, dwMapFlags, lpSrcStr, cchSrc, 0, 0 ))) { ret = FALSE; goto cleanupA; } __try{ cbuffer1 = (char *)_alloca(sizeof(char) * buff_size); memset(cbuffer1, 0, sizeof(char) * buff_size); } __except (EXCEPTION_EXECUTE_HANDLER) { _resetstkoflw(); cbuffer1 = NULL; } if ( cbuffer1 == NULL ) { if ( (cbuffer1 = (char *)_malloc_crt(sizeof(char) * buff_size)) == NULL) { ret = FALSE; goto cleanupA; } (void)memset( cbuffer1, 0, sizeof(char) * buff_size); malloc_flag1++; } if (0 == (buff_size = LCMapStringA( Locale, dwMapFlags, lpSrcStr, cchSrc, cbuffer1, buff_size))) { ret = FALSE; } else { if ( NULL == __convertcp(AnsiCP, code_page, cbuffer1, &buff_size, lpDestStr, cchDest)) ret = FALSE; else ret = TRUE; }cleanupA: if(malloc_flag1) _free_crt(cbuffer1); } else { ret = LCMapStringA( Locale, dwMapFlags, lpSrcStr, cchSrc, lpDestStr, cchDest ); } if ( cbuffer != NULL) _free_crt(cbuffer); return ret; }
/* Use "W" version */
if (USE_W == f_use) { int retval = 0; int inbuff_size; int outbuff_size; wchar_t *inwbuffer; wchar_t *outwbuffer; int malloc_flag1 = 0; int malloc_flag2 = 0;
/*
* 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, bError ? MB_PRECOMPOSED | MB_ERR_INVALID_CHARS : MB_PRECOMPOSED, lpSrcStr, cchSrc, NULL, 0 )) ) return 0;
/* allocate enough space for wide chars */ __try { inwbuffer = (wchar_t *)_alloca( inbuff_size * sizeof(wchar_t) ); } __except(EXCEPTION_EXECUTE_HANDLER) { _resetstkoflw(); inwbuffer = NULL; }
if ( inwbuffer == NULL ) { if ( (inwbuffer = (wchar_t *)_malloc_crt(inbuff_size * sizeof(wchar_t))) == NULL ) return 0; malloc_flag1++; }
/* 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) */ __try { outwbuffer = (wchar_t *)_alloca( outbuff_size * sizeof(wchar_t) ); } __except(EXCEPTION_EXECUTE_HANDLER) { _resetstkoflw(); outwbuffer = NULL; }
if ( outwbuffer == NULL ) { if ( (outwbuffer = (wchar_t *)_malloc_crt(outbuff_size * sizeof(wchar_t))) == NULL ) goto error_cleanup; malloc_flag2++; }
/* 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, 0, outwbuffer, outbuff_size, NULL, 0, NULL, NULL )) ) goto error_cleanup; } else { /* convert mapping */ if ( 0 == (retval = WideCharToMultiByte( code_page, 0, outwbuffer, outbuff_size, lpDestStr, cchDest, NULL, NULL )) ) goto error_cleanup; } }
error_cleanup: if ( malloc_flag2 ) _free_crt(outwbuffer);
if ( malloc_flag1 ) _free_crt(inwbuffer);
return retval; } else /* f_use is neither USE_A nor USE_W */ return 0;}
|
