Leaked source code of windows server 2003
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.
 
 
 
 
 
 

144 lines
4.3 KiB

/***
*mbsnset.c - Sets first n charcaters of string to given character (MBCS)
*
* Copyright (c) 1985-2001, Microsoft Corporation. All rights reserved.
*
*Purpose:
* Sets first n charcaters of string to given character (MBCS)
*
*Revision History:
* 11-19-92 KRS Ported from 16-bit sources.
* 08-20-93 CFW Change short params to int for 32-bit tree.
* 10-05-93 GJF Replaced _CRTAPI1 with __cdecl.
* 04-15-93 CFW Add _MB_CP_LOCK.
* 05-09-94 CFW Optimize for SBCS.
* 05-19-94 CFW Enable non-Win32.
* 09-11-97 GJF Replaced __mbcodepage == 0 with _ISNOTMBCP.
* 04-17-98 GJF Revised multithread support based on threadmbcinfo
* structs
*
*******************************************************************************/
#ifdef _MBCS
#include <mtdll.h>
#include <cruntime.h>
#include <string.h>
#include <mbdata.h>
#include <mbctype.h>
#include <mbstring.h>
/***
* _mbsnset - Sets first n charcaters of string to given character (MBCS)
*
*Purpose:
* Sets the first n characters of string to the supplied
* character value. If the length of string is less than n,
* the length of string is used in place of n. Handles
* MBCS chars correctly.
*
* There are several factors that make this routine complicated:
* (1) The fill value may be 1 or 2 bytes long.
* (2) The fill operation may end by hitting the count value
* or by hitting the end of the string.
* (3) A null terminating char is NOT placed at the end of
* the string.
*
* Cases to be careful of (both of these can occur at once):
* (1) Leaving an "orphaned" trail byte in the string (e.g.,
* overwriting a lead byte but not the corresponding trail byte).
* (2) Writing only the 1st byte of a 2-byte fill value because the
* end of string was encountered.
*
*Entry:
* unsigned char *string = string to modify
* unsigned int val = value to fill string with
* size_t count = count of characters to fill
*
*
*Exit:
* Returns string = now filled with char val
*
*Uses:
*
*Exceptions:
*
*******************************************************************************/
unsigned char * __cdecl _mbsnset(
unsigned char *string,
unsigned int val,
size_t count
)
{
unsigned char *start = string;
unsigned int leadbyte = 0;
unsigned char highval, lowval;
#ifdef _MT
pthreadmbcinfo ptmbci = _getptd()->ptmbcinfo;
if ( ptmbci != __ptmbcinfo )
ptmbci = __updatetmbcinfo();
#endif
/*
* leadbyte flag indicates if the last byte we overwrote was
* a lead byte or not.
*/
#ifdef _MT
if ( _ISNOTMBCP_MT(ptmbci) )
#else
if ( _ISNOTMBCP )
#endif
return _strnset(string, val, count);
if (highval = (unsigned char)(val>>8)) {
/* double byte value */
lowval = (unsigned char)(val & 0x00ff);
while (count-- && *string) {
#ifdef _MT
leadbyte = __ismbbtruelead_mt(ptmbci, leadbyte, *string);
#else
leadbyte = _ismbbtruelead(leadbyte, *string);
#endif
*string++ = highval;
if (*string) {
#ifdef _MT
leadbyte = __ismbbtruelead_mt(ptmbci, leadbyte, *string);
#else
leadbyte = _ismbbtruelead(leadbyte, *string);
#endif
*string++ = lowval;
}
else
/* overwrite orphaned highval byte */
*(string-1) = ' ';
}
}
else {
/* single byte value */
while (count-- && *string) {
#ifdef _MT
leadbyte = __ismbbtruelead_mt(ptmbci, leadbyte, *string);
#else
leadbyte = _ismbbtruelead(leadbyte, *string);
#endif
*string++ = (unsigned char)val;
}
}
/* overwrite orphaned trailing byte, if necessary */
if(leadbyte && *string)
*string = ' ';
return( start );
}
#endif /* _MBCS */