able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/base/efiutil/efilib/efisrc/wstring.cxx

1563 lines
30 KiB
C++
Raw Normal View History

Add source files
2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1992-2000 Microsoft Corporation
Module Name:
wstring.cxx
--*/
#include <pch.cxx>
#define _NTAPI_ULIB_
#define _ULIB_MEMBER_
#include "ulib.hxx"
#include "wstring.hxx"
#if !defined( _EFICHECK_ )
#include <stdio.h>
#include <wchar.h>
#endif
BOOLEAN WSTRING::_UseAnsiConversions = FALSE;
BOOLEAN WSTRING::_UseConsoleConversions = FALSE;
#if defined FE_SB
BOOLEAN WSTRING::_UseAnsiConversionsPrev = FALSE;
BOOLEAN WSTRING::_UseConsoleConversionsPrev = FALSE;
#endif
// Helper functions for OEM/Unicode conversion. Note that these
// are abstracted to private functions to make it easier to set
// them up for various environments.
//
INLINE
BOOLEAN
WSTRING::ConvertOemToUnicodeN(
PWSTR UnicodeString,
ULONG MaxBytesInUnicodeString,
PULONG BytesInUnicodeString,
PCHAR OemString,
ULONG BytesInOemString
)
{
UINT32 i;
// BUGBUG this is a big hack.
if( MaxBytesInUnicodeString == 0 ) {
*BytesInUnicodeString = 2*BytesInOemString;
return TRUE;
}
if( MaxBytesInUnicodeString < 2*BytesInOemString ) {
*BytesInUnicodeString = 2*BytesInOemString;
return FALSE;
}
memset(UnicodeString,0,MaxBytesInUnicodeString);
for (i=0; i<BytesInOemString;i++) {
UnicodeString[i] = (WCHAR)(OemString[i]);
}
*BytesInUnicodeString = BytesInOemString*2;
return TRUE;
}
INLINE
BOOLEAN
WSTRING::ConvertUnicodeToOemN(
PCHAR OemString,
ULONG MaxBytesInOemString,
PULONG BytesInOemString,
PWSTR UnicodeString,
ULONG BytesInUnicodeString
)
{
INT32 len = (INT32)StrLen(UnicodeString);
INT32 i;
UCHAR a;
// BUGBUG this is a big hack. I just discard any non-ANSI unicode character.
if( MaxBytesInOemString == 0 ) {
*BytesInOemString = len;
return TRUE;
}
if( MaxBytesInOemString < (ULONG)len ) {
*BytesInOemString = len;
return FALSE;
}
for( i=0; i<len; i++ ) {
if( HIBYTE(UnicodeString[i] != 0 )){
a = '?';
} else {
a = LOBYTE(UnicodeString[i]);
}
OemString[i] = a;
}
*BytesInOemString = len;
return TRUE;
}
INLINE
VOID
WSTRING::Construct(
)
{
_s = NULL;
_l = 0;
}
DEFINE_CONSTRUCTOR( WSTRING, OBJECT );
BOOLEAN
WSTRING::Initialize(
IN PCWSTRING InitialString,
IN CHNUM Position,
IN CHNUM Length
)
/*++
Routine Description:
This routine initializes the current string by copying the contents
of the given string.
Arguments:
InitialString - Supplies the initial string.
Position - Supplies the position in the given string to start at.
Length - Supplies the length of the string.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
DebugAssert(Position <= InitialString->_l);
Length = min(Length, InitialString->_l - Position);
if (!NewBuf(Length)) {
return FALSE;
}
memcpy(_s, InitialString->_s + Position, (UINT) Length*sizeof(WCHAR));
return TRUE;
}
BOOLEAN
WSTRING::Initialize(
IN PCWSTR InitialString,
IN CHNUM StringLength
)
/*++
Routine Description:
This routine initializes the current string by copying the contents
of the given string.
Arguments:
InitialString - Supplies the initial string.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
if (StringLength == TO_END) {
StringLength = wcslen(InitialString);
}
if (!NewBuf(StringLength)) {
return FALSE;
}
memcpy(_s, InitialString, (UINT) StringLength*sizeof(WCHAR));
return TRUE;
}
BOOLEAN
WSTRING::Initialize(
IN PCSTR InitialString,
IN CHNUM StringLength
)
/*++
Routine Description:
This routine initializes the current string by copying the contents
of the given string.
Arguments:
InitialString - Supplies the initial string.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
CHNUM length=0;
BOOLEAN status;
if (StringLength == TO_END) {
StringLength = strlen(InitialString);
}
if (!StringLength) {
return Resize(0);
}
// We want to avoid making two calls to RtlOemToUnicodeN so
// try to guess an adequate size for the buffer.
if (!NewBuf(StringLength)) {
return FALSE;
}
status = ConvertOemToUnicodeN(_s, _l*sizeof(WCHAR),
&length, (PSTR) InitialString,
StringLength);
length /= sizeof(WCHAR);
if (status) {
return Resize(length);
}
// We didn't manage to make in one try so ask exactly how much
// we need and then make the call.
status = ConvertOemToUnicodeN(NULL, 0, &length, (PSTR) InitialString,
StringLength);
length /= sizeof(WCHAR);
if (!status || !NewBuf(length)) {
return FALSE;
}
status = ConvertOemToUnicodeN(_s, _l*sizeof(WCHAR),
&length, (PSTR) InitialString, StringLength);
if (!status) {
return FALSE;
}
DebugAssert(length == _l*sizeof(WCHAR));
return TRUE;
}
BOOLEAN
WSTRING::Initialize(
IN LONG Number
)
/*++
Routine Description:
This routine initializes the current string by copying the contents
of the given string.
Arguments:
Number - Supplies the number to initialize the string to.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
WCHAR tmp[64];
SPrint(tmp, 64, TEXT("%d"), Number);
return Initialize(tmp);
}
NONVIRTUAL
PWSTRING
WSTRING::QueryString(
IN CHNUM Position,
IN CHNUM Length
) CONST
/*++
Routine Description:
This routine returns a copy of this string from the specified
coordinates.
Arguments:
Position - Supplies the initialize position of the string.
Length - Supplies the length of the string.
Return Value:
A pointer to a string or NULL.
--*/
{
PWSTRING p;
if (!(p = NEW DSTRING) ||
!p->Initialize(this, Position, Length)) {
DELETE(p);
}
return p;
}
BOOLEAN
WSTRING::QueryNumber(
OUT PLONG Number,
IN CHNUM Position,
IN CHNUM Length
) CONST
/*++
Routine Description:
This routine queries a number from the string.
Arguments:
Number - Returns the number parsed out of the string.
Position - Supplies the position of the number.
Length - Supplies the length of the number.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
FSTRING String;
PSTR p;
CHNUM spn;
if (Position >= _l) {
return FALSE;
}
Length = min(Length, _l - Position);
//
// Note that 123+123 will be a number!
//
String.Initialize((PWSTR) L"1234567890+-");
spn = Strspn(&String, Position);
if ((spn == INVALID_CHNUM || spn >= Position + Length) &&
(p = QuerySTR(Position, Length))) {
*Number = atol(p);
DELETE(p);
return TRUE;
}
return FALSE;
}
VOID
WSTRING::DeleteChAt(
IN CHNUM Position,
IN CHNUM Length
)
/*++
Routine Description:
This routine removes the character at the given position.
Arguments:
Position - Supplies the position of the character to remove.
Length - Supplies the number of characters to remove.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
DebugAssert(Position <= _l);
Length = min(Length, _l - Position);
memmove(_s + Position, _s + Position + Length,
(UINT) (_l - Position - Length)*sizeof(WCHAR));
Resize(_l - Length);
}
NONVIRTUAL
BOOLEAN
WSTRING::InsertString(
IN CHNUM AtPosition,
IN PCWSTRING String,
IN CHNUM FromPosition,
IN CHNUM FromLength
)
/*++
Routine Description:
This routine inserts the given string at the given position in
this string.
Arguments:
AtPosition - Supplies the position at which to insert the string.
String - Supplies the string to insert.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
CHNUM old_length;
DebugAssert(AtPosition <= _l);
DebugAssert(FromPosition <= String->_l);
FromLength = min(FromLength, String->_l - FromPosition);
old_length = _l;
if (!Resize(_l + FromLength)) {
return FALSE;
}
memmove(_s + AtPosition + FromLength, _s + AtPosition,
(UINT) (old_length - AtPosition)*sizeof(WCHAR));
memcpy(_s + AtPosition, String->_s + FromPosition,
(UINT) FromLength*sizeof(WCHAR));
return TRUE;
}
NONVIRTUAL
BOOLEAN
WSTRING::Replace(
IN CHNUM AtPosition,
IN CHNUM AtLength,
IN PCWSTRING String,
IN CHNUM FromPosition,
IN CHNUM FromLength
)
/*++
Routine Description:
This routine replaces the contents of this string from
'Position' to 'Length' with the contents of 'String2'
from 'Position2' to 'Length2'.
Arguments:
AtPosition - Supplies the position to replace at.
AtLength - Supplies the length to replace at.
String - Supplies the string to replace with.
FromPosition - Supplies the position to replace from in String2.
FromLength - Supplies the position to replace from in String2.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
CHNUM old_length;
DebugAssert(AtPosition <= _l);
DebugAssert(FromPosition <= String->_l);
AtLength = min(AtLength, _l - AtPosition);
FromLength = min(FromLength, String->_l - FromPosition);
// Make sure up front that we have the room but don't disturb
// the string.
if (FromLength > AtLength) {
old_length = _l;
if (!Resize(_l + FromLength - AtLength)) {
return FALSE;
}
Resize(old_length);
}
DeleteChAt(AtPosition, AtLength);
if (!InsertString(AtPosition, String, FromPosition, FromLength)) {
DebugAbort("This absolutely can never happen\n");
return FALSE;
}
return TRUE;
}
NONVIRTUAL
BOOLEAN
WSTRING::ReplaceWithChars(
IN CHNUM AtPosition,
IN CHNUM AtLength,
IN WCHAR Character,
IN CHNUM FromLength
)
/*++
Routine Description:
This routine replaces the contents of this string from
AtPosition of AtLength with the string formed by Character
of FromLength.
Arguments:
AtPosition - Supplies the position to replace at.
AtLength - Supplies the length to replace at.
Character - Supplies the character to replace with.
FromLength - Supplies the total number of new characters to replace the old one with.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
CHNUM old_length;
PWCHAR currptr, endptr;
DebugAssert(AtPosition <= _l);
AtLength = min(AtLength, _l - AtPosition);
// Make sure up front that we have the room but don't disturb
// the string.
if (FromLength > AtLength) {
old_length = _l;
if (!Resize(_l + FromLength - AtLength)) {
return FALSE;
}
Resize(old_length);
}
DeleteChAt(AtPosition, AtLength);
old_length = _l;
if (!Resize(_l + FromLength)) {
DebugPrint("This should not fail\n");
return FALSE;
}
memmove(_s + AtPosition + FromLength, _s + AtPosition,
(UINT) (old_length - AtPosition)*sizeof(WCHAR));
for (currptr = _s + AtPosition, endptr = currptr + FromLength;
currptr < endptr;
currptr++) {
*currptr = Character;
}
return TRUE;
}
PWSTR
WSTRING::QueryWSTR(
IN CHNUM Position,
IN CHNUM Length,
OUT PWSTR Buffer,
IN CHNUM BufferLength,
IN BOOLEAN ForceNull
) CONST
/*++
Routine Description:
This routine makes a copy of this string into the provided
buffer. If this string is not provided then a buffer is
allocated on the heap.
Arguments:
Position - Supplies the position within this string.
Length - Supplies the length of this string to take.
Buffer - Supplies the buffer to copy to.
BufferLength - Supplies the number of characters in the buffer.
ForceNull - Specifies whether or not to force the final character
of the buffer to be NULL in the case when there
isn't enough room for the whole string including
the NULL.
Return Value:
A pointer to a NULL terminated string.
--*/
{
DebugAssert(Position <= _l);
Length = min(Length, _l - Position);
if (!Buffer) {
BufferLength = Length + 1;
if (!(Buffer = (PWCHAR) MALLOC(BufferLength*sizeof(WCHAR)))) {
return NULL;
}
}
if (BufferLength > Length) {
memcpy(Buffer, _s + Position, (UINT) Length*sizeof(WCHAR));
Buffer[Length] = 0;
} else {
memcpy(Buffer, _s + Position, (UINT) BufferLength*sizeof(WCHAR));
if (ForceNull) {
Buffer[BufferLength - 1] = 0;
}
}
return Buffer;
}
PSTR
WSTRING::QuerySTR(
IN CHNUM Position,
IN CHNUM Length,
OUT PSTR Buffer,
IN CHNUM BufferLength,
IN BOOLEAN ForceNull
) CONST
/*++
Routine Description:
This routine computes a multi-byte version of the current
unicode string. If the buffer is not supplied then it
will be allocated by this routine.
Arguments:
Position - Supplies the position within this string.
Length - Supplies the length of this string to take.
Buffer - Supplies the buffer to convert into.
BufferLength - Supplies the number of characters in this buffer.
ForceNull - Specifies whether or not to force a NULL even
when the buffer is too small for the string.
Return Value:
A pointer to a NULL terminated multi byte string.
--*/
{
ULONG ansi_length;
DebugAssert(Position <= _l);
Length = min(Length, _l - Position);
// First special case the empty result.
if (!Length) {
if (!Buffer) {
if (!(Buffer = (PSTR) MALLOC(1))) {
return NULL;
}
} else if (!BufferLength) {
return NULL;
}
Buffer[0] = 0;
return Buffer;
}
// Next case is that the buffer is not provided and thus
// we have to figure out what size it should be.
if (!Buffer) {
// We want to avoid too many calls to RtlUnicodeToOemN
// so we'll estimate a correct size for the buffer and
// hope that that works.
BufferLength = 2*Length + 1;
if (!(Buffer = (PSTR) MALLOC(BufferLength))) {
return NULL;
}
if (ConvertUnicodeToOemN(Buffer, BufferLength - 1,
&ansi_length, _s + Position,
Length*sizeof(WCHAR))) {
Buffer[ansi_length] = 0;
return Buffer;
}
// We failed to estimate the necessary size of the buffer.
// So ask the correct size and try again.
FREE(Buffer);
if (!ConvertUnicodeToOemN(NULL, 0, &ansi_length,
_s + Position, Length*sizeof(WCHAR))) {
return NULL;
}
BufferLength = ansi_length + 1;
if (!(Buffer = (PSTR) MALLOC(BufferLength))) {
return NULL;
}
}
if (!ConvertUnicodeToOemN(Buffer, BufferLength, &ansi_length,
_s + Position, Length*sizeof(WCHAR))) {
return NULL;
}
if (BufferLength > ansi_length) {
Buffer[ansi_length] = 0;
} else {
if (ForceNull) {
Buffer[BufferLength - 1] = 0;
}
}
return Buffer;
}
BOOLEAN
WSTRING::Strcat(
IN PCWSTRING String
)
/*++
Routine Description:
This routine concatenates the given string onto this one.
Arguments:
String - Supplies the string to concatenate to this one.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
CHNUM old_length;
old_length = _l;
if (!Resize(_l + String->_l)) {
return FALSE;
}
memcpy(_s + old_length, String->_s, (UINT) String->_l*sizeof(WCHAR));
return TRUE;
}
NONVIRTUAL
PWSTRING
WSTRING::Strupr(
IN CHNUM StartPosition,
IN CHNUM Length
)
/*++
Routine Description:
This routine upcases a portion of this string.
Arguments:
StartPosition - Supplies the start position of the substring to upcase.
Length - Supplies the length of the substring to upscase.
Return Value:
A pointer to this string.
--*/
{
WCHAR c;
DebugAssert(StartPosition <= _l);
Length = min(Length, _l - StartPosition);
c = _s[StartPosition + Length];
_s[StartPosition + Length] = 0;
// BUGBUG don't have an upcase function in EFI.
// _wcsupr(_s + StartPosition);
_s[StartPosition + Length] = c;
return this;
}
NONVIRTUAL
PWSTRING
WSTRING::Strlwr(
IN CHNUM StartPosition,
IN CHNUM Length
)
/*++
Routine Description:
This routine lowercases a portion of this string.
Arguments:
StartPosition - Supplies the start position of the substring to lowercase.
Length - Supplies the length of the substring to lowercase.
Return Value:
A pointer to this string.
--*/
{
WCHAR c;
DebugAssert(StartPosition <= _l);
Length = min(Length, _l - StartPosition);
c = _s[StartPosition + Length];
_s[StartPosition + Length] = 0;
// BUGBUG we don't have a lowercase function in EFI
// _wcslwr(_s + StartPosition);
_s[StartPosition + Length] = c;
return this;
}
NONVIRTUAL
LONG
WSTRING::Strcmp(
IN PCWSTRING String,
IN CHNUM LeftPosition,
IN CHNUM LeftLength,
IN CHNUM RightPosition,
IN CHNUM RightLength
) CONST
/*++
Routine Description:
This routine compares two substrings.
Arguments:
String - Supplies the string to compare this one to.
LeftPosition - Supplies the postion for the left substring.
LeftLength - Supplies the length of the left substring.
LeftPosition - Supplies the postion for the left substring.
LeftLength - Supplies the length of the left substring.
Return Value:
<0 - Left substring is less than right substring.
0 - Left and Right substrings are equal
>0 - Left substring is greater than right substring.
--*/
{
WCHAR c, d;
LONG r;
DebugAssert(LeftPosition <= _l);
DebugAssert(RightPosition <= String->_l);
LeftLength = min(LeftLength, _l - LeftPosition);
RightLength = min(RightLength, String->_l - RightPosition);
c = _s[LeftPosition + LeftLength];
d = String->_s[RightPosition + RightLength];
_s[LeftPosition + LeftLength] = 0;
String->_s[RightPosition + RightLength] = 0;
r = wcscmp(_s + LeftPosition, String->_s + RightPosition);
_s[LeftPosition + LeftLength] = c;
String->_s[RightPosition + RightLength] = d;
return r;
}
NONVIRTUAL
LONG
WSTRING::Stricmp(
IN PCWSTRING String,
IN CHNUM LeftPosition,
IN CHNUM LeftLength,
IN CHNUM RightPosition,
IN CHNUM RightLength
) CONST
/*++
Routine Description:
This routine compares two substrings insensitive of case.
Arguments:
String - Supplies the string to compare this one to.
LeftPosition - Supplies the postion for the left substring.
LeftLength - Supplies the length of the left substring.
LeftPosition - Supplies the postion for the left substring.
LeftLength - Supplies the length of the left substring.
Return Value:
<0 - Left substring is less than right substring.
0 - Left and Right substrings are equal
>0 - Left substring is greater than right substring.
--*/
{
WCHAR c, d;
LONG r;
DebugAssert(LeftPosition <= _l);
DebugAssert(RightPosition <= String->_l);
LeftLength = min(LeftLength, _l - LeftPosition);
RightLength = min(RightLength, String->_l - RightPosition);
c = _s[LeftPosition + LeftLength];
d = String->_s[RightPosition + RightLength];
_s[LeftPosition + LeftLength] = 0;
String->_s[RightPosition + RightLength] = 0;
#if !defined _AUTOCHECK_ && !defined _EFICHECK_
// This works around a bug in the libc version of wcsicoll, where
// it doesn't specify STRINGSORT to CompareString(). To reproduce the
// bug, try sorting 1 and -1. (-1 should sort before 1.)
//
r = CompareString(GetUserDefaultLCID(),
NORM_IGNORECASE | SORT_STRINGSORT,
_s + LeftPosition,
-1,
String->_s + RightPosition,
-1
);
if (r >= 1) {
//
// return codes 1, 2, and 3 map to -1, 0, and 1.
//
_s[LeftPosition + LeftLength] = c;
String->_s[RightPosition + RightLength] = d;
return r - 2;
}
// If 'r' is 0, this indicates failure and we'll fall through and
// call wcsicoll.
//
#endif // _AUTOCHECK_
r = _wcsicmp(_s + LeftPosition, String->_s + RightPosition);
_s[LeftPosition + LeftLength] = c;
String->_s[RightPosition + RightLength] = d;
return r;
}
PWSTR
WSTRING::SkipWhite(
IN PWSTR p
)
{
#ifdef FE_SB
while (*p) {
if (iswspace(*p))
p++;
else if ( *p == 0x3000 )
{
*p++ = TEXT(' ');
}
else
break;
}
#else
while (iswspace(*p)) {
p++;
}
#endif
return p;
}
/**************************************************************************/
/* Compare two strings, ignoring white space, case is significant, return */
/* 0 if identical, <>0 otherwise. Leading and trailing white space is */
/* ignored, internal white space is treated as single characters. */
/**************************************************************************/
INT
WSTRING::Strcmps (
IN PWSTR p1,
IN PWSTR p2
)
{
WCHAR *q;
p1 = WSTRING::SkipWhite(p1); /* skip any leading white space */
p2 = WSTRING::SkipWhite(p2);
while (TRUE)
{
if (*p1 == *p2)
{
if (*p1++ == 0) /* quit if at the end */
return (0);
else
p2++;
#ifdef FE_SB
if (CheckSpace(p1))
#else
if (iswspace(*p1)) /* compress multiple spaces */
#endif
{
q = WSTRING::SkipWhite(p1);
p1 = (*q == 0) ? q : q - 1;
}
#ifdef FE_SB
if (CheckSpace(p2))
#else
if (iswspace(*p2))
#endif
{
q = WSTRING::SkipWhite(p2);
p2 = (*q == 0) ? q : q - 1;
}
}
else
return *p1-*p2;
}
}
/**************************************************************************/
/* Compare two strings, ignoring white space, case is not significant, */
/* return 0 if identical, <>0 otherwise. Leading and trailing white */
/* space is ignored, internal white space is treated as single characters.*/
/**************************************************************************/
INT
WSTRING::Strcmpis (
IN PWSTR p1,
IN PWSTR p2
)
{
WCHAR *q;
#ifdef FE_SB
WCHAR c1,c2;
#endif
p1 = WSTRING::SkipWhite(p1); /* skip any leading white space */
p2 = WSTRING::SkipWhite(p2);
while (TRUE)
{
if (towupper(*p1) == towupper(*p2))
{
if (*p1++ == 0) /* quit if at the end */
return (0);
else
p2++;
#ifdef FE_SB
if (CheckSpace(p1))
#else
if (iswspace(*p1)) /* compress multiple spaces */
#endif
{
q = SkipWhite(p1);
p1 = (*q == 0) ? q : q - 1;
}
#ifdef FE_SB
if (CheckSpace(p2))
#else
if (iswspace(*p2))
#endif
{
q = WSTRING::SkipWhite(p2);
p2 = (*q == 0) ? q : q - 1;
}
}
else
return *p1-*p2;
}
}
#ifdef FE_SB
/**************************************************************************/
/* Routine: CheckSpace */
/* Arguments: an arbitrary string */
/* Function: Determine whether there is a space in the string. */
/* Side effects: none */
/**************************************************************************/
INT
WSTRING::CheckSpace(
IN PWSTR s
)
{
if (iswspace(*s) || *s == 0x3000 )
return (TRUE);
else
return (FALSE);
}
#endif
#define DUMMY_ULIB_EXPORT
DEFINE_EXPORTED_CONSTRUCTOR( FSTRING, WSTRING, DUMMY_ULIB_EXPORT );
BOOLEAN
FSTRING::Resize(
IN CHNUM NewStringLength
)
/*++
Routine Description:
This routine implements the WSTRING Resize routine by using
the buffer supplied at initialization time.
Arguments:
NewStringLength - Supplies the new length of the string.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
return NewBuf(NewStringLength);
}
BOOLEAN
FSTRING::NewBuf(
IN CHNUM NewStringLength
)
/*++
Routine Description:
This routine implements the WSTRING NewBuf routine by using
the buffer supplied at initialization time.
Arguments:
NewStringLength - Supplies the new length of the string.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
if (NewStringLength >= _buffer_length) {
return FALSE;
}
PutString((PWSTR) GetWSTR(), NewStringLength);
return TRUE;
}
INLINE
VOID
DSTRING::Construct(
)
/*++
Routine Description:
This routine initializes the string to a valid initial state.
Arguments:
None.
Return Value:
None.
--*/
{
_buf = NULL;
_length = 0;
}
DEFINE_EXPORTED_CONSTRUCTOR( DSTRING, WSTRING, DUMMY_ULIB_EXPORT );
DSTRING::~DSTRING(
)
/*++
Routine Description:
Destructor for DSTRING.
Arguments:
None.
Return Value:
None.
--*/
{
FREE(_buf);
}
BOOLEAN
DSTRING::Resize(
IN CHNUM NewStringLength
)
/*++
Routine Description:
This routine resizes this string to the specified new size.
Arguments:
NewStringLength - Supplies the new length of the string.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
PWSTR new_buf;
if (NewStringLength >= _length) {
if (_buf) {
#if !defined( _EFICHECK_ )
if (!(new_buf = (PWSTR)
REALLOC(_buf, (NewStringLength + 1)*sizeof(WCHAR)))) {
return FALSE;
}
#else
new_buf = (PWSTR) ReallocatePool( _buf,
_length * sizeof(WCHAR),
(NewStringLength + 1)*sizeof(WCHAR) );
if ( new_buf == NULL ) {
return FALSE;
}
#endif
} else {
if (!(new_buf = (PWSTR)
MALLOC((NewStringLength + 1)*sizeof(WCHAR)))) {
return FALSE;
}
}
_buf = new_buf;
_length = NewStringLength + 1;
}
PutString(_buf, NewStringLength);
return TRUE;
}
BOOLEAN
DSTRING::NewBuf(
IN CHNUM NewStringLength
)
/*++
Routine Description:
This routine resizes this string to the specified new size.
Arguments:
NewStringLength - Supplies the new length of the string.
Return Value:
FALSE - Failure.
TRUE - Success.
--*/
{
PWSTR new_buf;
if (NewStringLength >= _length) {
if (!(new_buf = (PWSTR)
MALLOC((NewStringLength + 1)*sizeof(WCHAR)))) {
return FALSE;
}
if (_buf) {
FREE(_buf);
}
_buf = new_buf;
_length = NewStringLength + 1;
}
PutString(_buf, NewStringLength);
return TRUE;
}
#if defined FE_SB
VOID
WSTRING::ResetConversions(
)
{
_UseAnsiConversions = _UseAnsiConversionsPrev;
_UseConsoleConversions = _UseConsoleConversionsPrev;
}
#endif
VOID
WSTRING::SetAnsiConversions(
)
/*++
Routine Description:
This routine declares that all conversions from multi byte
to unicode will take place using the ANSI code page. Note
that this is a STATIC method. Therefore this switch affects
*all* WSTRINGs.
Arguments:
None.
Return Value:
None.
--*/
{
#if defined FE_SB
_UseAnsiConversionsPrev = _UseAnsiConversions;
_UseConsoleConversionsPrev = _UseConsoleConversions;
#endif
_UseAnsiConversions = TRUE;
_UseConsoleConversions = FALSE;
}
VOID
WSTRING::SetOemConversions(
)
/*++
Routine Description:
This routine declares that all conversions from multi byte
to unicode will take place using the OEM code page. Note
that this is a STATIC method. Therefore this switch affects
*all* WSTRINGs.
This is the default if neither this nor the above function is
called.
Arguments:
None.
Return Value:
None.
--*/
{
#if defined FE_SB
_UseAnsiConversionsPrev = _UseAnsiConversions;
_UseConsoleConversionsPrev = _UseConsoleConversions;
#endif
_UseAnsiConversions = FALSE;
_UseConsoleConversions = FALSE;
}
VOID
WSTRING::SetConsoleConversions(
)
/*++
Routine Description:
This routine declares that all conversions from multi byte
to unicode will take place using the current console code page.
Note that this is a STATIC method. Therefore this switch
affects *all* WSTRINGs.
Arguments:
None.
Return Value:
None.
--*/
{
#if defined FE_SB
_UseAnsiConversionsPrev = _UseAnsiConversions;
_UseConsoleConversionsPrev = _UseConsoleConversions;
#endif
_UseAnsiConversions = FALSE;
_UseConsoleConversions = TRUE;
}
#if defined FE_SB
CHNUM
WSTRING::QueryByteCount(
) CONST
/*++
Routine Description:
This routine returns the number of ANSI bytes the UNICODE string
consists of.
Arguments:
None.
Return Value:
Number of ANSI bytes the UNICODE string is made from, or INVALID_CHNUM
on error.
--*/
{
ULONG ansi_length;
ULONG BufferLen = _l * sizeof(WCHAR) + 1;
PSTR Buffer;
BOOLEAN success;
if ( !_l ) {
return( (CHNUM)0 );
}
if (NULL == (Buffer = (PSTR)MALLOC( BufferLen ))) {
return( INVALID_CHNUM );
}
success = ConvertUnicodeToOemN( Buffer, BufferLen - 1, &ansi_length,
_s, BufferLen - 1 );
FREE( Buffer );
if (!success) {
return INVALID_CHNUM;
}
return ansi_length;
}
#endif
Reference in a new issue Copy permalink