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.
 
 
 
 
 
 

446 lines
7.4 KiB

/*++
Copyright (c) 1990-2000 Microsoft Corporation
Module Name:
Chcp
Abstract:
Chcpo is a DOS5-Compatible codepage utility
Author:
Ramon Juan San Andres (ramonsa) 01-May-1991
Revision History:
--*/
#include "ulib.hxx"
#include "arg.hxx"
#include "stream.hxx"
#include "smsg.hxx"
#include "wstring.hxx"
#include "rtmsg.h"
#include "chcp.hxx"
#include <winnlsp.h>
VOID __cdecl
main (
)
/*++
Routine Description:
Main function of the Chcp utility
Arguments:
None.
Return Value:
None.
Notes:
--*/
{
SetThreadUILanguage(0);
//
// Initialize stuff
//
DEFINE_CLASS_DESCRIPTOR( CHCP );
{
CHCP Chcp;
if ( Chcp.Initialize() ) {
Chcp.Chcp();
}
}
}
DEFINE_CONSTRUCTOR( CHCP, PROGRAM );
CHCP::~CHCP (
)
/*++
Routine Description:
Destructs a CHCP object
Arguments:
None.
Return Value:
None.
Notes:
--*/
{
}
BOOLEAN
CHCP::Initialize (
)
/*++
Routine Description:
Initializes a CHCP object
Arguments:
None.
Return Value:
TRUE if initialized.
Notes:
--*/
{
//
// Initialize program object
//
if ( PROGRAM::Initialize( MSG_CHCP_USAGE ) &&
_Screen.Initialize( )
) {
_SetCodePage = FALSE;
_CodePage = 0;
return TRUE;
}
return FALSE;
}
BOOLEAN
CHCP::Chcp (
)
/*++
Routine Description:
Does the Chcp thing.
Arguments:
None.
Return Value:
TRUE.
Notes:
--*/
{
ValidateVersion();
if ( ParseArguments() ) {
if ( _SetCodePage ) {
//
// Set the code page
//
if ( !SetCodePage() ) {
ExitProgram( EXIT_ERROR );
}
} else {
//
// Display current code page
//
if ( !DisplayCodePage() ) {
ExitProgram( EXIT_ERROR );
}
}
ExitProgram( EXIT_NORMAL );
} else {
ExitProgram( EXIT_ERROR );
}
return TRUE;
}
BOOLEAN
CHCP::DisplayCodePage (
)
/*++
Routine Description:
Displays the active code page
Arguments:
None.
Return Value:
TRUE if success, FALSE if syntax error.
Notes:
--*/
{
DisplayMessage(
MSG_CHCP_ACTIVE_CODEPAGE,
NORMAL_MESSAGE, "%d",
_Screen.QueryCodePage( )
);
return TRUE;
}
BOOLEAN
CHCP::ParseArguments (
)
/*++
Routine Description:
Parses arguments
Arguments:
None.
Return Value:
TRUE if success, FALSE if syntax error.
Notes:
--*/
{
ARGUMENT_LEXEMIZER ArgLex;
ARRAY LexArray;
ARRAY ArgArray;
STRING_ARGUMENT ProgramNameArgument;
LONG_ARGUMENT CodePageArgument;
FLAG_ARGUMENT UsageArgument;
if ( !ArgArray.Initialize() ||
!LexArray.Initialize() ||
!ArgLex.Initialize( &LexArray )
) {
DisplayMessage( MSG_CHCP_INTERNAL_ERROR, ERROR_MESSAGE );
ExitProgram( EXIT_ERROR );
}
if ( !ProgramNameArgument.Initialize( "*" ) ||
!UsageArgument.Initialize( "/?" ) ||
!CodePageArgument.Initialize( "*" )
) {
DisplayMessage( MSG_CHCP_INTERNAL_ERROR, ERROR_MESSAGE );
ExitProgram( EXIT_ERROR );
}
if ( !ArgArray.Put( &ProgramNameArgument ) ||
!ArgArray.Put( &UsageArgument ) ||
!ArgArray.Put( &CodePageArgument )
) {
DisplayMessage( MSG_CHCP_INTERNAL_ERROR, ERROR_MESSAGE );
ExitProgram( EXIT_ERROR );
}
//
// Set up the defaults
//
ArgLex.PutSwitches( "/" );
ArgLex.SetCaseSensitive( FALSE );
if ( !ArgLex.PrepareToParse() ) {
DisplayMessage( MSG_CHCP_INTERNAL_ERROR, ERROR_MESSAGE );
ExitProgram( EXIT_ERROR );
}
if ( !ArgLex.DoParsing( &ArgArray ) ) {
DisplayMessage( MSG_CHCP_INVALID_PARAMETER, ERROR_MESSAGE, "%W", ArgLex.QueryInvalidArgument() );
ExitProgram( EXIT_ERROR );
}
//
// Display Help if requested
//
if ( UsageArgument.IsValueSet() ) {
DisplayMessage( MSG_CHCP_USAGE, NORMAL_MESSAGE );
ExitProgram( EXIT_NORMAL );
}
if ( CodePageArgument.IsValueSet() ) {
_SetCodePage = TRUE;
_CodePage = (DWORD)CodePageArgument.QueryLong();
} else {
_SetCodePage = FALSE;
}
return TRUE;
}
inline bool IsFarEastCodePage(UINT cp)
{
return cp == 932 || cp == 949 || cp == 936 || cp == 950;
}
inline bool IsFarEastLocale(LANGID langid)
{
// Accepts primay langid only
return langid == LANG_JAPANESE || langid == LANG_KOREAN || langid == LANG_CHINESE;
}
inline bool IsFarEastSystemLocale()
{
return IsFarEastLocale(PRIMARYLANGID(GetSystemDefaultLangID()));
}
BOOLEAN
CHCP::SetCodePage (
)
/*++
Routine Description:
Sets the active code page
Arguments:
None.
Return Value:
TRUE if success, FALSE if syntax error.
Notes:
--*/
{
UINT OldCP = _Screen.QueryCodePage( );
if (IsFarEastSystemLocale() &&
(IsFarEastCodePage(_CodePage) || IsFarEastCodePage(OldCP)) &&
_CodePage != OldCP) {
/*
* This CLS function is needed only if it's FE.
*/
_Screen.MoveCursorTo(0, 0);
_Screen.EraseScreenAndResetAttribute();
}
if ( _Screen.SetOutputCodePage( _CodePage ) ) {
if (_Screen.SetCodePage( _CodePage ) ) {
/*
// Comment from NT4J:
// Since FormatMessage checks the current TEB's locale, and the Locale for
// CHCP is initialized when the message class is initialized, the TEB has to
// be updated after the code page is changed successfully. All other code
// pages other than JP and US are ignored.
if (IsFarEastSystemLocale()) {
LANGID LangId;
switch (_CodePage) {
case 932:
LangId = MAKELANGID(LANG_JAPANESE, SUBLANG_DEFAULT);
break;
case 949:
LangId = MAKELANGID(LANG_KOREAN, SUBLANG_KOREAN);
break;
case 936:
LangId = MAKELANGID(LANG_CHINESE, SUBLANG_CHINESE_SIMPLIFIED);
break;
case 950:
LangId = MAKELANGID(LANG_CHINESE, SUBLANG_CHINESE_TRADITIONAL);
break;
default:
LangId = MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US);
break;
}
SetThreadLocale(MAKELCID(LangId, SORT_DEFAULT));
}
*/
SetThreadUILanguage(0);
return DisplayCodePage( );
} else {
// SetOutputCodePage failed.
// Restore the privous input code page
_Screen.SetOutputCodePage( OldCP );
}
}
// Was unable to set the given code page
DisplayMessage( MSG_CHCP_INVALID_CODEPAGE, ERROR_MESSAGE );
return FALSE;
}