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.
 
 
 
 
 
 

328 lines
8.4 KiB

/*++
Copyright (c) 1991-1992 Microsoft Corporation
Module Name:
CopyStr.c
Abstract:
This module contains the main two functions for copying and converting
strings between the default codepage for the LAN and wide characters
(i.e. UNICODE).
Author:
John Rogers (JohnRo) 24-Sep-1991
Environment:
Only runs under NT, although the interface is portable (Win/32).
Requires ANSI C extensions: slash-slash comments, long external names.
Note:
These functions assume that the machine's default codepage is the same
as the LAN's default codepage.
Revision History:
24-Sep-1991 JohnRo
Created.
04-Oct-1991 JohnRo
Fixed buffer length bug in CopyStrToWStr().
24-Oct-1991 JohnRo
Corrected Environment comments above.
21-Nov-1991 JohnRo
Added some alignment assertions.
26-Nov-1991 JohnRo
Added NetpNCopy routines (used by new net config helpers).
03-Jan-1992 JohnRo
Added NetpCopyStringToTStr() for FAKE_PER_PROCESS_RW_CONFIG handling.
29-Apr-1992 JohnRo
Fixed NetpNCopyStrToWStr() under UNICODE. Ditto NetpNCopyWStrToStr.
Made a few changes suggested by PC-LINT.
21-May-1992 JohnRo
Removed bogus assert in NetpNCopyStrToWStr().
--*/
// These must be included first:
#include <nt.h> // Must be first. IN, VOID, etc.
#include <windef.h> // LPWSTR.
#include <lmcons.h> // (Needed by NetLibNt.h)
// These can be in any order:
#include <align.h> // ROUND_UP_POINTER(), ALIGN_WCHAR.
#include <lmapibuf.h> // NetApiBufferFree().
#include <netdebug.h> // NetpAssert(), etc.
#include <ntrtl.h> // RtlUnicodeStringToOemString(), etc.
#include <string.h> // strlen().
#include <tstring.h> // Most of my prototypes.
#include <stdlib.h> // wcslen(), wcsncpy().
#include <winerror.h> // NO_ERROR, ERROR_NOT_ENOUGH_MEMORY.
VOID
NetpCopyWStrToStr(
OUT LPSTR Dest,
IN LPWSTR Src
)
/*++
Routine Description:
NetpCopyWStrToStr copies characters from a source string
to a destination, converting as it copies them.
Arguments:
Dest - is an LPSTR indicating where the converted characters are to go.
This string will be in the default codepage for the LAN.
Src - is in LPWSTR indicating the source string.
Return Value:
None.
--*/
{
OEM_STRING DestAnsi;
NTSTATUS NtStatus;
UNICODE_STRING SrcUnicode;
NetpAssert( Dest != NULL );
NetpAssert( Src != NULL );
NetpAssert( ((LPVOID)Dest) != ((LPVOID)Src) );
NetpAssert( ROUND_UP_POINTER( Src, ALIGN_WCHAR ) == Src );
*Dest = '\0';
NetpInitOemString(
& DestAnsi, // output: struct
Dest); // input: null terminated
//
// Tell RTL routines there's enough memory out there.
// Note that using a max length in characters is by
// design as this is what this routine's callers
// expect (the max length should normally be in bytes).
//
DestAnsi.MaximumLength = (USHORT) (wcslen(Src)+1);
RtlInitUnicodeString(
& SrcUnicode, // output: struct
Src); // input: null terminated
NtStatus = RtlUnicodeStringToOemString(
& DestAnsi, // output: struct
& SrcUnicode, // input: struct
(BOOLEAN) FALSE); // don't allocate string.
NetpAssert( NT_SUCCESS(NtStatus) );
} // NetpCopyWStrToStr
VOID
NetpCopyStrToWStr(
OUT LPWSTR Dest,
IN LPSTR Src
)
/*++
Routine Description:
NetpCopyStrToWStr copies characters from a source string
to a destination, converting as it copies them.
Arguments:
Dest - is an LPWSTR indicating where the converted characters are to go.
Src - is in LPSTR indicating the source string. This must be a string in
the default codepage of the LAN.
Return Value:
None.
--*/
{
UNICODE_STRING DestUnicode;
NTSTATUS NtStatus;
OEM_STRING SrcAnsi;
NetpAssert( Dest != NULL );
NetpAssert( Src != NULL );
NetpAssert( ((LPVOID)Dest) != ((LPVOID)Src) );
NetpAssert( ROUND_UP_POINTER( Dest, ALIGN_WCHAR ) == Dest );
*Dest = L'\0';
NetpInitOemString(
& SrcAnsi, // output: struct
Src); // input: null terminated
RtlInitUnicodeString(
& DestUnicode, // output: struct
Dest); // input: null terminated
// Tell RTL routines there's enough memory out there.
DestUnicode.MaximumLength = (USHORT)
((USHORT) (strlen(Src)+1)) * (USHORT) sizeof(wchar_t);
NtStatus = RtlOemStringToUnicodeString(
& DestUnicode, // output: struct
& SrcAnsi, // input: struct
(BOOLEAN) FALSE); // don't allocate string.
NetpAssert( NT_SUCCESS(NtStatus) );
} // NetpCopyStrToWStr
NET_API_STATUS
NetpNCopyStrToWStr(
OUT LPWSTR Dest,
IN LPSTR Src, // string in default LAN codepage
IN DWORD CharCount
)
{
LPWSTR TempW;
NetpAssert( Dest != NULL );
NetpAssert( Src != NULL );
NetpAssert( ((LPVOID)Dest) != ((LPVOID)Src) );
NetpAssert( ROUND_UP_POINTER( Dest, ALIGN_WCHAR ) == Dest );
// Allocate a copy of the full string, and convert to UNICODE.
TempW = NetpAllocWStrFromStr( Src );
if (TempW == NULL) {
// Out of memory!
return (ERROR_NOT_ENOUGH_MEMORY);
}
// Copy portion of array. Append with nulls if necessary.
// (Thank god for the C runtime library! --JR)
(VOID) wcsncpy( Dest, TempW, CharCount );
// Free our temporary string.
(VOID) NetApiBufferFree( TempW );
return (NO_ERROR);
} // NetpNCopyStrToWStr
NET_API_STATUS
NetpNCopyWStrToStr(
OUT LPSTR Dest, // string in default LAN codepage
IN LPWSTR Src,
IN DWORD CharCount
)
{
LPSTR TempStr;
NetpAssert( Dest != NULL );
NetpAssert( Src != NULL );
NetpAssert( ((LPVOID)Dest) != ((LPVOID)Src) );
NetpAssert( ROUND_UP_POINTER( Src, ALIGN_WCHAR ) == Src );
// Allocate a copy of the full string, and convert to UNICODE.
TempStr = NetpAllocStrFromWStr( Src );
if (TempStr == NULL) {
return (ERROR_NOT_ENOUGH_MEMORY);
}
// Copy N chars, pad with nulls, etc.
(VOID) strncpy( Dest, TempStr, CharCount );
// Free our temporary data.
(VOID) NetApiBufferFree( TempStr );
return (NO_ERROR);
} // NetpNCopyWStrToStr
// NetpCopyStrToStr that assumes Dest <= wcslen(SRC)+1 * sizeof(WCHAR)
// This will be called whereever we have made sure that the proper
// buffer size has been allocated. Eventually we can replace
// NetpCopStrToStr with this entirely once all calling functions
// have been fixed up.
VOID NetpCopyWStrToStrDBCS(
OUT LPSTR Dest,
IN LPWSTR Src
)
/*++
Routine Description:
NetpCopyWStrToStr copies characters from a source string
to a destination, converting as it copies them.
Arguments:
Dest - is an LPSTR indicating where the converted characters are to go.
This string will be in the default codepage for the LAN.
Src - is in LPWSTR indicating the source string.
Return Value:
None.
--*/
{
NTSTATUS NtStatus;
LONG Index;
ULONG DestLen = NetpUnicodeToDBCSLen( Src )+1;
NetpAssert( Dest != NULL );
NetpAssert( Src != NULL );
NetpAssert( ((LPVOID)Dest) != ((LPVOID)Src) );
NetpAssert( ROUND_UP_POINTER( Src, ALIGN_WCHAR ) == Src );
NtStatus = RtlUnicodeToOemN(
Dest, // Destination string
DestLen, // Destination string length
&Index, // Last char in translated string
Src, // Source string
wcslen(Src)*sizeof(WCHAR) // Length of source string
);
Dest[Index] = '\0';
NetpAssert( NT_SUCCESS(NtStatus) );
} // NetpCopyWStrToStr
ULONG
NetpUnicodeToDBCSLen(
IN LPWSTR Src
)
{
UNICODE_STRING SrcUnicode;
RtlInitUnicodeString(
& SrcUnicode, // output: struct
Src); // input: null terminated
return( RtlUnicodeStringToOemSize( &SrcUnicode ) -1 );
}