/******************************************************************************
Copyright(c) Microsoft Corporation
Module Name:
BootCfg.cpp
Abstract:
This file is intended to have the functionality for
configuring, displaying, changing and deleting boot.ini
settings for the local host or a remote system.
Author:
J.S.Vasu 17/1/2001
Revision History:
J.S.Vasu 17/1/2001 Localisation,function headers
SanthoshM.B 10/2/2001 Added 64 bit functionality Code.
J.S.Vasu 15/2/2001 Added the functionality of 32 bit and 64 bit acc to the DCR's.
******************************************************************************/
// Include files
#include "pch.h"
#include "resource.h"
#include "BootCfg.h"
#include "BootCfg64.h"
// ***************************************************************************
//
// Routine description : Main function which calls all the other main functions depending on
// the option specified by the user.
//
// Arguments:
// [in] argc : argument count specified at the command prompt.
// [in] argv : arguments specified at the command prompt.
//
// Return Value : DWORD
// 0 : If the utility successfully performs the specified operation.
// 1 : If the utility is unsuccessful in performing the specified operation.
// ***************************************************************************
// Main function of the program
DWORD _cdecl _tmain( DWORD argc, LPCTSTR argv[] )
{
// Declaring the main option switches as boolean values
BOOL bUsage = FALSE ;
BOOL bCopy = FALSE ;
BOOL bChange = FALSE ;
BOOL bQuery = FALSE ;
BOOL bDelete = FALSE ;
BOOL bRawString = FALSE ;
BOOL bResult = FALSE ;
BOOL bNeedPwd = FALSE ;
DWORD dwExitcode = ERROR_SUCCESS;
BOOL bProcess = FALSE ;
BOOL bTimeOut = FALSE ;
BOOL bDefault = FALSE ;
BOOL bDebug = FALSE ;
BOOL bEms = FALSE ;
BOOL bAddSw = FALSE ;
BOOL bRmSw = FALSE ;
BOOL bDbg1394 = FALSE ;
BOOL bMirror = FALSE ;
TCHAR szServer[MAX_RES_STRING] = NULL_STRING ;
DWORD dwRetVal = 0 ;
#ifdef _WIN64
dwExitcode = InitializeEFI();
#endif
if(argc == 1)
{
dwExitcode = QueryBootIniSettings( argc, argv );
return dwExitcode;
}
// Call the preProcessOptions function to find out the option selected by the user
dwExitcode = preProcessOptions( argc, argv, &bUsage, &bCopy, &bQuery, &bDelete,&bRawString,&bDefault,&bTimeOut,&bDebug,&bEms,&bAddSw,&bRmSw,&bDbg1394,&bMirror);
if(dwExitcode == EXIT_FAILURE)
{
ReleaseGlobals();
return dwExitcode;
}
/*#ifdef _WIN64
dwExitcode = InitializeEFI();
#endif
*/
if(bUsage && bTimeOut)
{
#ifndef _WIN64
//check the remote systemtype and display error
//if it is a 64 bit
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_SUCCESS )
{
displayTimeOutUsage_X86() ;
}
else
{
return EXIT_FAILURE ;
}
#else
displayTimeOutUsage_IA64();
#endif
ReleaseGlobals();
return EXIT_SUCCESS ;
}
if(bUsage && bDefault)
{
#ifndef _WIN64
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_SUCCESS )
{
displayChangeOSUsage_X86();
}
else
{
return EXIT_FAILURE ;
}
#else
displayDefaultEntryUsage_IA64();
#endif
ReleaseGlobals();
return EXIT_SUCCESS ;
}
// If BootIni.exe /?
if( ( bUsage ==TRUE)&& ( bCopy==FALSE )&& (bQuery==FALSE)&&(bDelete==FALSE)&&(bRawString ==FALSE)
&&(bDefault==FALSE)&&(bTimeOut==FALSE) && (bDebug==FALSE)&& (bEms==FALSE)&&(bAddSw==FALSE)
&&(bRmSw==FALSE)&&( bDbg1394==FALSE )&&(bMirror== FALSE) )
{
#ifndef _WIN64
dwExitcode = displayMainUsage_X86();
#else
displayMainUsage_IA64();
return EXIT_SUCCESS ;
#endif
}
if(bRawString)
{
#ifndef _WIN64
dwExitcode = AppendRawString(argc,argv);
#else
dwExitcode = RawStringOsOptions_IA64(argc,argv);
#endif
ReleaseGlobals();
return dwExitcode;
}
// If BootIni.exe -copy option is selected
if( bCopy )
{
#ifndef _WIN64
dwExitcode = CopyBootIniSettings( argc, argv );
#else
dwExitcode = CopyBootIniSettings_IA64( argc, argv);
#endif
}
// If BootIni.exe -delete option is selected
if( bDelete )
{
#ifndef _WIN64
dwExitcode = DeleteBootIniSettings( argc, argv );
#else
dwExitcode = DeleteBootIniSettings_IA64( argc, argv );
#endif
}
// If BootIni.exe -query option is selected
if( bQuery )
{
dwExitcode = QueryBootIniSettings( argc, argv );
}
if(bTimeOut)
{
#ifndef _WIN64
dwExitcode = ChangeTimeOut(argc,argv);
#else
dwExitcode = ChangeTimeOut_IA64(argc,argv);
#endif
}
if(bDefault)
{
#ifndef _WIN64
dwExitcode = ChangeDefaultOs(argc,argv);
#else
dwExitcode = ChangeDefaultBootEntry_IA64(argc,argv);
#endif
}
if(bDebug )
{
#ifndef _WIN64
dwExitcode = ProcessDebugSwitch( argc, argv );
#else
dwExitcode = ProcessDebugSwitch_IA64(argc,argv);
#endif
}
if(bEms )
{
#ifndef _WIN64
dwExitcode = ProcessEmsSwitch( argc, argv );
#else
dwExitcode = ProcessEmsSwitch_IA64(argc,argv);
#endif
}
if(bAddSw )
{
#ifndef _WIN64
dwExitcode = ProcessAddSwSwitch( argc, argv );
#else
dwExitcode = ProcessAddSwSwitch_IA64(argc,argv);
#endif
}
if(bRmSw )
{
#ifndef _WIN64
dwExitcode = ProcessRmSwSwitch( argc, argv );
#else
dwExitcode = ProcessRmSwSwitch_IA64( argc, argv );
#endif
}
if (bDbg1394 )
{
#ifndef _WIN64
dwExitcode = ProcessDbg1394Switch(argc,argv);
#else
dwExitcode = ProcessDbg1394Switch_IA64(argc,argv);
#endif
}
if(bMirror)
{
#ifdef _WIN64
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_SYNTAX));
//dwExitcode = ProcessMirrorSwitch_IA64(argc,argv);
#else
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_SYNTAX));
#endif
}
// exit with the appropriate return value if there is no problem
ReleaseGlobals();
return dwExitcode;
}
// ***************************************************************************
//
//
// Routine Description : Function used to process the main options
//
// Arguments:
// [ in ] argc : Number of command line arguments
// [ in ] argv : Array containing command line arguments
// [ out ] pbUsage : Pointer to boolean variable which will indicate
// whether usage option is specified by the user.
// [ out ] pbCopy : Pointer to boolean variable which will indicate
// whether copy option is specified by the user.
// [ out ] pbQuery : Pointer to boolean variable which will indicate
// whether query option is specified by the user.
// [ out ] pbChange : Pointer to boolean variable which will indicate
// whether change option is specified by the user.
// [ out ] pbDelete : Pointer to boolean variable which will indicate
// whether delete option is specified by the user.
// [ out ] pbRawString : Pointer to the boolean indicating whether raw option
// is specified by the user.
// [ out ] pbDefault : Pointer to the boolean indicating whether default option
// is specified by the user.
// [ out ] pbTimeOut : Pointer to the boolean indicating whether timeout option
// is specified by the user.
// [ out ] pbDebug : Pointer to the boolean indicating whether debug option
// is specified by the user.
// [ out ] pbEms : Pointer to the boolean indicating whether ems option
// is specified by the user.
// [ out ] pbAddSw : Pointer to the boolean indicating whether Addsw option
// is specified by the user.
// [ out ] pbRmSw : Pointer to the boolean indicating whether rmsw option
// is specified by the user.
// [ out ] pbDbg1394 : Pointer to the boolean indicating whether dbg1394 option
// is specified by the user.
// [ out ] pbMirror : Pointer to the boolean indicating whether mirror option
// is specified by the user.
//
// Return Type : Bool
// A Bool value indicating EXIT_SUCCESS on success else
// EXIT_FAILURE on failure
//
// ***************************************************************************
DWORD preProcessOptions( DWORD argc, LPCTSTR argv[],
PBOOL pbUsage,
PBOOL pbCopy,
PBOOL pbQuery,
PBOOL pbDelete,
PBOOL pbRawString,
PBOOL pbDefault,
PBOOL pbTimeOut,
PBOOL pbDebug,
PBOOL pbEms,
PBOOL pbAddSw,
PBOOL pbRmSw,
PBOOL pbDbg1394 ,
PBOOL pbMirror
)
{
// Initialise a boolean variable bOthers to find out whether switches other
// than the main swithces are selected by the user
BOOL bOthers = FALSE;
DWORD dwCount = 0;
DWORD dwi = 0;
BOOL bMainUsage = FALSE ;
TCHAR szServer[MAX_RES_STRING] = NULL_STRING ;
DWORD dwRetVal = 0;
BOOL bConnFlag = FALSE ;
// Populate the TCMDPARSER structure and pass the structure to the DoParseParam
// function. DoParseParam function populates the corresponding variables depending
// upon the command line input.
TCMDPARSER cmdOptions[] = {
{ CMDOPTION_COPY, 0, 1, 0, pbCopy, NULL_STRING, NULL, NULL },
{ CMDOPTION_QUERY, 0, 1, 0, pbQuery, NULL_STRING, NULL, NULL },
{ CMDOPTION_DELETE, 0, 1, 0, pbDelete, NULL_STRING, NULL, NULL },
{ CMDOPTION_USAGE, CP_USAGE, 1, 0, pbUsage, NULL_STRING, NULL, NULL },
{ CMDOTHEROPTIONS, CP_DEFAULT, 0, 0, &bOthers, NULL_STRING, NULL, NULL },
{ CMDOPTION_RAW, 0,1, 0, pbRawString, NULL_STRING, NULL, NULL },
{ CMDOPTION_DEFAULTOS, 0,1,0,pbDefault,NULL_STRING,NULL,NULL},
{ CMDOPTION_TIMEOUT, 0,1,0,pbTimeOut,NULL_STRING,NULL,NULL},
{ CMDOPTION_DEBUG, 0,1,0,pbDebug,NULL_STRING,NULL,NULL},
{ CMDOPTION_EMS, 0,1,0,pbEms,NULL_STRING,NULL,NULL},
{ CMDOPTION_ADDSW, 0,1,0,pbAddSw,NULL_STRING,NULL,NULL},
{ CMDOPTION_RMSW, 0,1,0,pbRmSw,NULL_STRING,NULL,NULL},
{ CMDOPTION_DBG1394, 0,1,0,pbDbg1394,NULL_STRING,NULL,NULL},
{ CMDOPTION_MIRROR, 0,1,0,pbMirror,NULL_STRING,NULL,NULL}
};
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
// If there is an error while parsing, display "Invalid Syntax"
// If more than one main option is selected, then display error message
// If usage is specified for sub-options
// If none of the options are specified
if ( ! DoParseParam( argc, argv, SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
DISPLAY_MESSAGE(stderr,GetReason());
return EXIT_FAILURE ;
}
//checking if the user has entered more than 1 option.
if (*pbCopy)
{
dwCount++ ;
}
if (*pbQuery)
{
dwCount++ ;
}
if (*pbDelete)
{
dwCount++ ;
}
if (*pbRawString)
{
dwCount++ ;
// Check if any of the other valid switches have been
// given as an input to the raw string
if( *pbTimeOut || *pbDebug || *pbAddSw
||*pbRmSw || *pbDbg1394 || *pbEms
||*pbDelete || *pbCopy || *pbQuery
||*pbDefault || *pbMirror)
{
// Check wether the usage switch has been entered
if( *pbUsage )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_MAIN_USAGE));
return ( EXIT_FAILURE );
}
// Check if the other option is specified after the
// 'raw' option
for( dwi = 0; dwi < argc; dwi++ )
{
if( lstrcmpi( argv[ dwi ], _T("-raw") ) == 0
|| lstrcmpi( argv[ dwi ], _T("/raw") ) == 0 )
{
if( (dwi+1) == argc )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_MAIN_USAGE));
return ( EXIT_FAILURE );
}
else if( _tcschr( argv[ dwi + 1 ], _T( '\"' ) ) != 0 )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_MAIN_USAGE));
return ( EXIT_FAILURE );
}
}
}
dwCount--;
}
}
if (*pbDefault)
{
dwCount++ ;
}
if (*pbTimeOut)
{
dwCount++ ;
}
if (*pbDebug)
{
dwCount++ ;
}
if(*pbAddSw)
{
dwCount++ ;
}
if(*pbRmSw)
{
dwCount++ ;
}
if(*pbDbg1394)
{
dwCount++ ;
}
if(*pbEms)
{
dwCount++ ;
}
if(*pbMirror)
{
dwCount++ ;
}
//display an error message if the user enters more than 1 main option
if( ( ( dwCount > 1 ) ) ||
//display an error message if the user enters 1 main option along with other junk
( (*pbUsage) && bOthers ) ||
//display an error message if the user does not enter any main option
( !(*pbCopy) && !(*pbQuery) && !(*pbDelete) && !(*pbUsage) && !(*pbRawString)&& !(*pbDefault)&&!(*pbTimeOut)&&!(*pbDebug)&& !( *pbEms)&& !(*pbAddSw)&& !(*pbRmSw)&& !(*pbDbg1394)&& !(*pbMirror) ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_MAIN_USAGE));
return ( EXIT_FAILURE );
}
return ( EXIT_SUCCESS );
}
/*****************************************************************************
Routine Description:
This routine is to make another OS instance copy for which you
can add switches.
Arguments:
[in] : argc Number of command line arguments
[in] : argv Array containing command line arguments
Return Value :
DWORD
******************************************************************************/
DWORD CopyBootIniSettings( DWORD argc, LPCTSTR argv[] )
{
// Declaring the main option switch and the usage switch as
// boolean values and initialising them
BOOL bCopy = FALSE ;
BOOL bUsage = FALSE;
// size of the buffer and the key buffer storing the keys
// Length of the buffer will be incremented dynamically based on
// the number of keys present in the boot.ini file. First the length
// is initialised to 100 TCHAR bytes.
LPTSTR szBuf = NULL_STRING ;
// File pointer pointing to the boot.ini file
FILE *stream = NULL;
// dynamic array containing all the keys of the boot.ini file
TARRAY arr = NULL;
// The key for which a new OS instance is to be built
TCHAR key[MAX_RES_STRING] = NULL_STRING;
BOOL bRes = FALSE ;
// Variable storing the path of boot.ini file
LPTSTR szPath = NULL_STRING ;
TCHAR szTmpPath[MAX_RES_STRING] = NULL_STRING ;
// Variable which stores the new key-value pair
TCHAR newInstance[500] = NULL_STRING ;
// Variable to keep track the OS entry specified by the user
DWORD dwDefault = 0;
// It contains the return length of GetPrivateProfileSection API
DWORD dwReturnLen = 0;
// Number of keys present in the boot.ini file
DWORD dwNumKeys = 0;
BOOL bNeedPwd =FALSE;
DWORD dwId = 0;
TCHAR szMesgBuffer[MAX_RES_STRING] = NULL_STRING;
// Initialising the variables that are passed to TCMDPARSER structure
STRING256 szServer = NULL_STRING;
STRING256 szUser = NULL_STRING;
STRING256 szPassword = NULL_STRING;
STRING256 szDescription = NULL_STRING;
BOOL bFlag = FALSE ;
BOOL bMemFlag = FALSE ;
DWORD dwLength = MAX_STRING_LENGTH1 ;
LPCTSTR szToken = NULL ;
DWORD dwRetVal = 0 ;
BOOL bConnFlag = FALSE ;
STRING256 szCopyStr = NULL_STRING ;
LPCTSTR szToken1 = NULL ;
// Builiding the TCMDPARSER structure
TCMDPARSER cmdOptions[] = {
{ CMDOPTION_COPY, CP_MAIN_OPTION, 1, 0, &bCopy, NULL_STRING, NULL, NULL },
{ SWITCH_SERVER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szServer, NULL_STRING, NULL, NULL },
{ SWITCH_USER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szUser, NULL_STRING, NULL, NULL },
{ SWITCH_PASSWORD, CP_TYPE_TEXT | CP_VALUE_OPTIONAL, 1, 0, &szPassword, NULL_STRING, NULL, NULL },
{ SWITCH_DESCRIPTION, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szDescription, NULL_STRING, NULL, NULL },
{ CMDOPTION_USAGE, CP_USAGE, 0, 0, &bUsage, 0, 0 },
{ SWITCH_ID, CP_TYPE_UNUMERIC | CP_VALUE_MANDATORY | CP_MANDATORY, 1, 0, &dwDefault, NULL_STRING, NULL, NULL }
};
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
szBuf = ( LPTSTR ) malloc( dwLength*sizeof( TCHAR ) );
if(szBuf == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
SAFEFREE(szBuf);
return (EXIT_FAILURE);
}
// Variable storing the path of boot.ini file
szPath = (TCHAR*)malloc(MAX_RES_STRING* sizeof(TCHAR));
if(szPath == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
SAFEFREE(szBuf);
SAFEFREE(szPath);
return (EXIT_FAILURE);
}
//copy the Asterix token which is required for password prompting.
_tcscpy(cmdOptions[3].szValues,TOKEN_ASTERIX) ;
_tcscpy(szPassword,TOKEN_ASTERIX);
// Parsing the copy option switches
if ( ! DoParseParam( argc, argv, SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
ShowMessage(stderr,GetReason());
SAFEFREE(szBuf);
SAFEFREE(szPath);
return (EXIT_FAILURE);
}
// Displaying copy usage if user specified -? with -copy option
if( bUsage )
{
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_SUCCESS )
{
SAFEFREE(szBuf);
SAFEFREE(szPath);
displayCopyUsage_X86();
return (EXIT_SUCCESS);
}
else
{
SAFEFREE(szBuf);
SAFEFREE(szPath);
return (EXIT_FAILURE);
}
}
//display an error message saying that Friendly name
// must be restricted ot 67 characters.
if(lstrlen(szDescription) >= 67)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_FRIENDLY_NAME));
SAFEFREE(szBuf);
SAFEFREE(szPath);
return (EXIT_FAILURE);
}
//display error message if the username is entered with out a machine name
if( (cmdOptions[1].dwActuals == 0)&&(cmdOptions[2].dwActuals != 0))
{
SetReason(GetResString(IDS_USER_BUT_NOMACHINE));
ShowMessage(stderr,GetReason());
SAFEFREE(szBuf);
SAFEFREE(szPath);
return EXIT_FAILURE ;
}
//display error message if the user enters password without entering username
if( (cmdOptions[2].dwActuals == 0)&&(cmdOptions[3].dwActuals != 0))
{
SetReason(GetResString(IDS_PASSWD_BUT_NOUSER));
ShowMessage(stderr,GetReason());
SAFEFREE(szBuf);
SAFEFREE(szPath);
return EXIT_FAILURE ;
}
// for prompting the password if the user enters
// * after -p option.
if( ( _tcscmp(szPassword,TOKEN_ASTERIX )==0 ) && (IsLocalSystem(szServer)==FALSE ))
{
bNeedPwd = TRUE ;
}
//set the bneedpassword to true if the server name is specified and password is not specified.
if((cmdOptions[1].dwActuals!=0)&&(cmdOptions[3].dwActuals==0))
{
if( (lstrlen( szServer ) != 0) && (IsLocalSystem(szServer)==FALSE) )
{
bNeedPwd = TRUE ;
}
else
{
bNeedPwd = FALSE ;
}
if(_tcslen(szPassword)!= 0 )
{
_tcscpy(szPassword,NULL_STRING);
}
}
//display an error message if the server is empty.
if( (cmdOptions[1].dwActuals!=0)&&(lstrlen(szServer)==0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_SERVER));
SAFEFREE(szBuf);
SAFEFREE(szPath);
return EXIT_FAILURE ;
}
//display an error message if the user is empty.
if((cmdOptions[2].dwActuals!=0)&&(lstrlen(szUser)==0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_USER));
SAFEFREE(szBuf);
SAFEFREE(szPath);
return EXIT_FAILURE ;
}
// Establishing connection to the specified machine and getting the file pointer
// of the boot.ini file if there is no error while establishing connection
lstrcpy(szPath, PATH_BOOTINI );
if(StrCmpN(szServer,TOKEN_BACKSLASH4,2)==0)
{
if(!StrCmpN(szServer,TOKEN_BACKSLASH6,3)==0)
{
szToken = _tcstok(szServer,TOKEN_BACKSLASH4);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
return (EXIT_FAILURE);
}
lstrcpy(szServer,szToken);
}
}
if( (IsLocalSystem(szServer)==TRUE)&&(lstrlen(szUser)!=0))
{
DISPLAY_MESSAGE(stdout,GetResString(WARN_LOCALCREDENTIALS));
_tcscpy(szServer,_T(""));
}
bFlag = openConnection( szServer, szUser, szPassword, szPath,bNeedPwd,stream,&bConnFlag);
if(bFlag == EXIT_FAILURE)
{
SAFECLOSE(stream);
SAFEFREE(szBuf);
SAFEFREE(szPath);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting the keys of the Operating system section in the boot.ini file
arr = getKeyValueOfINISection( szPath, OS_FIELD );
if(arr == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
SAFEFREE(szBuf);
SAFEFREE(szPath);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
lstrcpy(szTmpPath,szPath);
// Getting the total number of keys in the operating systems section
dwNumKeys = DynArrayGetCount(arr);
if((dwNumKeys >= MAX_BOOTID_VAL) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_MAX_BOOTID));
resetFileAttrib(szPath);
SAFEFREE(szBuf);
SAFEFREE(szPath);
DestroyDynamicArray(&arr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Displaying error message if the number of keys is less than the OS entry
// line number specified by the user
if( ( dwDefault <= 0 ) || ( dwDefault > dwNumKeys ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_BOOTID));
resetFileAttrib(szPath);
SAFEFREE(szBuf);
SAFEFREE(szPath);
DestroyDynamicArray(&arr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting the key of the OS entry specified by the user
if(arr != NULL)
{
LPCWSTR pwsz = NULL ;
pwsz = DynArrayItemAsString( arr, dwDefault - 1 ) ;
if(pwsz != NULL)
{
_tcscpy( key,pwsz);
}
else
{
resetFileAttrib(szPath);
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
SAFEFREE(szBuf);
SAFEFREE(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arr);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
}
else
{
resetFileAttrib(szPath);
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
SAFEFREE(szBuf);
SAFEFREE(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arr);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Copying the key to a newInstance variable which will hold the new key-value pair.
lstrcpy(newInstance, key);
szToken = _tcstok(newInstance,TOKEN_EQUAL);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
resetFileAttrib(szPath);
SAFEFREE(szBuf);
SAFEFREE(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arr);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Building the newInstance.
lstrcat(newInstance, TOKEN_EQUAL );
lstrcat(newInstance, TOKEN_SINGLEQUOTE);
// Copying the description specified by the user as the value of the new key.
lstrcat(newInstance, szDescription);
lstrcat(newInstance, TOKEN_SINGLEQUOTE);
//
//concatenating the star so that the boundschecker does not
//give invalid argument exception
//
lstrcat(key,TOKEN_STAR);
szToken = _tcstok(key,TOKEN_FWDSLASH1);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
resetFileAttrib(szPath);
SAFEFREE(szBuf);
SAFEFREE(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arr);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// get the Os Load options into the szToken1 string.
szToken1 = _tcstok(NULL,TOKEN_STAR);
lstrcpy(szCopyStr,szToken1);
lstrcat(newInstance,TOKEN_EMPTYSPACE);
lstrcat(newInstance,TOKEN_FWDSLASH1) ;
lstrcat(newInstance,szCopyStr);
newInstance[lstrlen(newInstance) + 1] = _T('\0');
// Reallocating the buffer length depending on the return value and of the
// GetPrivateProfileSection API. Its return a value 2 less than the buffer size
// if the buffer size is not enough to hold all the key-value pairs.
while( 1 )
{
dwReturnLen = GetPrivateProfileSection(OS_FIELD, szBuf, dwLength, szTmpPath);
// If buffer length is sufficient break
if ( dwLength - 2 != dwReturnLen )
break;
// Increasing the buffer length and reallocate the size
dwLength += 100;
szBuf = ( LPTSTR ) realloc( szBuf, dwLength * sizeof( TCHAR ) );
if (szBuf == NULL)
{
bMemFlag = TRUE ;
break ;
}
}
//display error message if there is no enough memory
if (bMemFlag == TRUE)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
resetFileAttrib(szTmpPath);
SAFEFREE(szBuf);
SAFEFREE(szPath);
DestroyDynamicArray(&arr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Adding the new key-value pair
lstrcat(szBuf + dwReturnLen, newInstance);
// Appending the null character at the end
*(szBuf + dwReturnLen + 1 + lstrlen(newInstance)) = '\0';
// Writing to the profile section with new key-value pair
if( WritePrivateProfileSection(OS_FIELD, szBuf, szTmpPath) != 0 )
{
_stprintf(szMesgBuffer,GetResString(IDS_COPY_SUCCESS),dwDefault);
DISPLAY_MESSAGE(stdout,szMesgBuffer);
bRes = resetFileAttrib(szPath);
bRes = resetFileAttrib(szTmpPath);
SAFEFREE(szBuf);
SAFEFREE(szPath);
DestroyDynamicArray(&arr);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return(bRes);
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_COPY_OS));
resetFileAttrib(szPath);
resetFileAttrib(szTmpPath);
SAFEFREE(szBuf);
SAFEFREE(szPath);
DestroyDynamicArray(&arr);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return (EXIT_FAILURE);
}
// Closing the opened boot.ini file handl
SAFECLOSE(stream);
bRes = resetFileAttrib(szPath);
bRes = resetFileAttrib(szTmpPath);
SAFEFREE(szBuf);
SAFEFREE(szPath);
DestroyDynamicArray(&arr);
SafeCloseConnection(szServer,bConnFlag);
return (bRes);
}
/*****************************************************************************
Routine Description:
This routine is to delete an OS entry from the Operating systems
section of Boot.ini file in the specified machine.
Arguments:
[in] : argc Number of command line arguments
[in] : argv Array containing command line arguments
Return Value :
DWORD
******************************************************************************/
DWORD DeleteBootIniSettings( DWORD argc, LPCTSTR argv[] )
{
// Declaring the main option switch and the usage switch as
// boolean values and initialsing them
BOOL bDelete = FALSE ;
BOOL bUsage = FALSE;
BOOL bRes = FALSE ;
// dwDefault variable stores the OS entry specified by the user
DWORD dwDefault = 0;
// Variables containing the count of key-value pairs
DWORD dwInitialCount = 0;
// Variable which will store the final key-value pairs
LPTSTR szFinalStr = NULL_STRING;
// array storing all the keys of the operating systems section
TARRAY arrKeyValue;
// Variable storing the path of boot.ini file
LPTSTR szPath = NULL_STRING ;
// File pointer to the boot.ini file
FILE *stream = NULL;
BOOL bNeedPwd = FALSE ;
BOOL bFlag = FALSE ;
TCHAR szMesgBuffer[MAX_RES_STRING] = NULL_STRING;
// Initialising the variables that are passed to TCMDPARSER structure
STRING256 szServer = NULL_STRING;
STRING256 szUser = NULL_STRING;
STRING256 szPassword = NULL_STRING;
STRING100 szDescription = NULL_STRING;
LPCTSTR szToken = NULL ;
DWORD dwRetVal = 0 ;
BOOL bConnFlag = FALSE ;
// Builiding the TCMDPARSER structure
TCMDPARSER cmdOptions[] = {
{ CMDOPTION_DELETE, CP_MAIN_OPTION, 1, 0, &bDelete, NULL_STRING, NULL, NULL },
{ SWITCH_SERVER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szServer, NULL_STRING, NULL, NULL },
{ SWITCH_USER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szUser, NULL_STRING, NULL, NULL },
{ SWITCH_PASSWORD, CP_TYPE_TEXT | CP_VALUE_OPTIONAL, 1, 0, &szPassword, NULL_STRING, NULL, NULL },
{ CMDOPTION_USAGE, CP_USAGE, 0, 0, &bUsage, NULL_STRING, NULL, NULL },
{ SWITCH_ID, CP_TYPE_NUMERIC | CP_VALUE_MANDATORY | CP_MANDATORY, 1, 0, &dwDefault, NULL_STRING, NULL, NULL }
};
//
//check if the remote system is 64 bit and if so
// display an error.
//
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
szFinalStr =(TCHAR*)malloc(MAX_STRING_LENGTH1 *sizeof(TCHAR));
if(szFinalStr == NULL )
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
return EXIT_FAILURE ;
}
// Variable storing the path of boot.ini file
szPath = (TCHAR*)malloc(MAX_RES_STRING * sizeof(TCHAR));
if(szPath == NULL)
{
SAFEFREE(szFinalStr);
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
return EXIT_FAILURE ;
}
//copy the Asterix token which is required for password prompting.
_tcscpy(cmdOptions[3].szValues,TOKEN_ASTERIX) ;
_tcscpy(szPassword,TOKEN_ASTERIX);
// Parsing the delete option switches
if ( ! DoParseParam( argc, argv, SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
ShowMessage(stderr,GetReason());
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
return (EXIT_FAILURE);
}
// Displaying delete usage if user specified -? with -delete option
if( bUsage )
{
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_SUCCESS )
{
displayDeleteUsage_X86();
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
return (EXIT_SUCCESS);
}else
{
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
return (EXIT_FAILURE);
}
}
//display error message if the username is entered with out a machine name
if( (cmdOptions[1].dwActuals == 0)&&(cmdOptions[2].dwActuals != 0))
{
SetReason(GetResString(IDS_USER_BUT_NOMACHINE));
ShowMessage(stderr,GetReason());
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
return EXIT_FAILURE ;
}
if( (cmdOptions[2].dwActuals == 0)&&(cmdOptions[3].dwActuals != 0))
{
SetReason(GetResString(IDS_PASSWD_BUT_NOUSER));
ShowMessage(stderr,GetReason());
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
return EXIT_FAILURE ;
}
//
//for setting the bNeedPwd
if( ( _tcscmp(szPassword,TOKEN_ASTERIX )==0 ) && (IsLocalSystem(szServer)==FALSE ))
{
bNeedPwd = TRUE ;
}
//set the bneedpassword to true if the server name is specified and password is not specified.
if((cmdOptions[1].dwActuals!=0)&&(cmdOptions[3].dwActuals==0))
{
if( (lstrlen( szServer ) != 0) && (IsLocalSystem(szServer)==FALSE) )
{
bNeedPwd = TRUE ;
}
else
{
bNeedPwd = FALSE ;
}
if(_tcslen(szPassword)!= 0 )
{
_tcscpy(szPassword,NULL_STRING);
}
}
//display an error message if the server is empty.
if( (cmdOptions[1].dwActuals!=0)&&(lstrlen(szServer)==0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_SERVER));
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
return EXIT_FAILURE ;
}
//display an error message if the user is empty.
if((cmdOptions[2].dwActuals!=0)&&(lstrlen(szUser)==0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_USER));
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
return EXIT_FAILURE ;
}
// Searching the path of the boot.ini file in the specified system
// In general the boot.ini file resides in c:\.
// Search Path function can be used to trace out the path of boot.ini file
// SearchPath( "\\", "boottest.ini", ".ini", 80, (LPTSTR)filepath, (LPTSTR*)&filepart );
lstrcpy(szPath, PATH_BOOTINI );
if(StrCmpN(szServer,TOKEN_BACKSLASH4,2)==0)
{
if(!StrCmpN(szServer,TOKEN_BACKSLASH6,3)==0)
{
szToken = _tcstok(szServer,TOKEN_BACKSLASH4);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
return (EXIT_FAILURE);
}
else
{
lstrcpy(szServer,szToken);
}
}
}
if( (IsLocalSystem(szServer)==TRUE)&&(lstrlen(szUser)!=0))
{
DISPLAY_MESSAGE(stdout,GetResString(WARN_LOCALCREDENTIALS));
_tcscpy(szServer,_T(""));
}
// Establishing connection to the specified machine and getting the file pointer
// of the boot.ini file if there is no error while establishing connection
bFlag = openConnection( szServer, szUser, szPassword, szPath,bNeedPwd,stream,&bConnFlag );
if(bFlag == EXIT_FAILURE)
{
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting all the key-value pairs of the operating system into a dynamic
// array for manipulation.
arrKeyValue = getKeyValueOfINISection( szPath, OS_FIELD);
if(arrKeyValue == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting the total no: of key-value pairs in the operating system section.
dwInitialCount = DynArrayGetCount(arrKeyValue);
// Checking whether the given OS entry is valid or not. If the OS entry given
// is greater than the number of keys present, then display an error message
if( ( dwDefault <= 0 ) || ( dwDefault > dwInitialCount ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_BOOTID));
resetFileAttrib(szPath);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
DestroyDynamicArray(&arrKeyValue);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// If only one OS entry is present and if the user tries to delete the OS entry, then
// display an error message
if( DynArrayGetCount(arrKeyValue) == 1)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ONLY_ONE_OS));
resetFileAttrib(szPath);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
DestroyDynamicArray(&arrKeyValue);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Remove the OS entry specified by the user from the dynamic array
DynArrayRemove(arrKeyValue, dwDefault - 1);
// Setting the buffer to 0, to avoid any junk value
memset(szFinalStr, 0, MAX_STRING_LENGTH1);
// Forming the final string from all the key-value pairs
if (stringFromDynamicArray1( arrKeyValue,szFinalStr ) == EXIT_FAILURE)
{
resetFileAttrib(szPath);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
DestroyDynamicArray(&arrKeyValue);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Writing to the profile section with new key-value pair
// If the return value is non-zero, then there is an error.
if( WritePrivateProfileSection(OS_FIELD, szFinalStr, szPath ) != 0 )
{
_stprintf(szMesgBuffer,GetResString(IDS_DEL_SUCCESS),dwDefault);
DISPLAY_MESSAGE(stdout,szMesgBuffer);
bRes = resetFileAttrib(szPath);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
DestroyDynamicArray(&arrKeyValue);
SafeCloseConnection(szServer,bConnFlag);
return (bRes);
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_DELETE_OS));
resetFileAttrib(szPath);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
DestroyDynamicArray(&arrKeyValue);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Closing the boot.ini stream
SAFECLOSE(stream);
bRes = resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
SAFEFREE(szPath);
DestroyDynamicArray(&arrKeyValue);
SafeCloseConnection(szServer,bConnFlag);
return (bRes);
}
/*****************************************************************************
Routine Description:
This routine is to display the current boot.ini file settings for
the specified system.
Arguments:
[in] : argc Number of command line arguments
[in] : argv Array containing command line arguments
Return Value :
DWORD
******************************************************************************/
DWORD QueryBootIniSettings( DWORD argc, LPCTSTR argv[] )
{
// File pointer pointing to the boot.ini file
FILE *stream = NULL;
// Declaring the main option switch and the usage switch as
// boolean values and initialsing them
BOOL bQuery = FALSE ;
BOOL bUsage = FALSE;
BOOL bExitVal = TRUE;
BOOL bNeedPwd = FALSE ;
BOOL bVerbose = TRUE ;
TCOLUMNS ResultHeader[ MAX_COLUMNS ];
TARRAY arrResults = NULL ;
TARRAY arrKeyValuePairs = NULL;
TARRAY arrBootLoader = NULL;
DWORD dwFormatType = 0;
BOOL bHeader = TRUE ;
DWORD dwLength = 0 ;
DWORD dwCnt = 0;
TCHAR szValue[MAX_RES_STRING] = NULL_STRING ;
TCHAR szFriendlyName[MAX_CMD_LENGTH] = TOKEN_NA ;
TCHAR szBootOptions[MAX_RES_STRING] = TOKEN_NA ;
TCHAR szBootEntry[MAX_RES_STRING] = TOKEN_NA ;
TCHAR szArcPath[MAX_RES_STRING] = TOKEN_NA ;
TCHAR szTmpString[MAX_RES_STRING] = TOKEN_NA ;
PTCHAR psztok = NULL ;
DWORD dwRow = 0;
DWORD dwCount = 0;
BOOL bRes = FALSE ;
BOOL bFlag = FALSE ;
DWORD dwIndex = 0 ;
DWORD dwLength1 = 0 ;
DWORD dwFinalLength = 0 ;
// Initialising the variables that are passed to TCMDPARSER structure
STRING256 szServer = NULL_STRING;
STRING256 szUser = NULL_STRING;
STRING256 szPassword = NULL_STRING;
STRING100 szPath = NULL_STRING;
LPCWSTR szKeyName[MAX_RES_STRING] ;
TCHAR szResults[MAX_RES_STRING][MAX_RES_STRING] ;
TCHAR szDisplay[MAX_RES_STRING] = NULL_STRING ;
DWORD dwSectionFlag = 0 ;
LPCTSTR szToken = NULL ;
DWORD dwRetVal= 0 ;
BOOL bConnFlag = FALSE ;
BOOL bTokenFlag = FALSE ;
BOOL bPasswdFlag = FALSE ;
// Builiding the TCMDPARSER structure
TCMDPARSER cmdOptions[] = {
{ CMDOPTION_QUERY, CP_MAIN_OPTION, 1, 0, &bQuery, NULL_STRING, NULL, NULL },
{ SWITCH_SERVER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szServer, NULL_STRING, NULL, NULL },
{ SWITCH_USER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szUser, NULL_STRING, NULL, NULL },
{ SWITCH_PASSWORD, CP_TYPE_TEXT | CP_VALUE_OPTIONAL, 1, 0, &szPassword, NULL_STRING, NULL, NULL },
{ CMDOPTION_USAGE, CP_USAGE, 0, 0, &bUsage, NULL_STRING, NULL, NULL }
};
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
_tcscpy(cmdOptions[3].szValues,TOKEN_ASTERIX) ;
_tcscpy(szPassword,TOKEN_ASTERIX);
// Parsing all the switches specified with -query option
if ( ! DoParseParam( argc, argv,SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
ShowMessage(stderr,GetReason());
return (EXIT_FAILURE);
}
if(StrCmpN(szServer,TOKEN_BACKSLASH4,2)==0)
{
if(!StrCmpN(szServer,TOKEN_BACKSLASH6,3)==0)
{
szToken = _tcstok(szServer,TOKEN_BACKSLASH4);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
return (EXIT_FAILURE);
}
lstrcpy(szServer,szToken);
}
}
// Displaying query usage if user specified -? with -query option
if( bUsage )
{
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_SUCCESS )
{
displayQueryUsage();
return (EXIT_SUCCESS);
}
else
{
return (EXIT_FAILURE);
}
}
#ifdef _WIN64
bExitVal = QueryBootIniSettings_IA64();
return bExitVal;
#endif
//display error message if the username is entered with out a machine name
if( (cmdOptions[1].dwActuals == 0)&&(cmdOptions[2].dwActuals != 0))
{
SetReason(GetResString(IDS_USER_BUT_NOMACHINE));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
if( (cmdOptions[2].dwActuals == 0)&&(cmdOptions[3].dwActuals != 0))
{
SetReason(GetResString(IDS_PASSWD_BUT_NOUSER));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
if( ( _tcscmp(szPassword,TOKEN_ASTERIX )==0 ) && (IsLocalSystem(szServer)==FALSE ))
{
bNeedPwd = TRUE ;
}
//set the bneedpassword to true if the server name is specified and password is not specified.
if((cmdOptions[1].dwActuals!=0)&&(cmdOptions[3].dwActuals==0))
{
if( (lstrlen( szServer ) != 0) && (IsLocalSystem(szServer)==FALSE) )
{
bNeedPwd = TRUE ;
}
else
{
bNeedPwd = FALSE ;
}
if(_tcslen(szPassword)!= 0 )
{
_tcscpy(szPassword,NULL_STRING);
}
}
if((cmdOptions[1].dwActuals!=0)&&(lstrlen(szServer)==0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_SERVER));
return EXIT_FAILURE ;
}
if( (cmdOptions[2].dwActuals!=0)&&(lstrlen(szUser)==0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_USER));
return EXIT_FAILURE ;
}
if( (IsLocalSystem(szServer)==TRUE)&&(lstrlen(szUser)!=0))
{
DISPLAY_MESSAGE(stdout,GetResString(WARN_LOCALCREDENTIALS));
_tcscpy(szServer,_T(""));
}
//set the default format as list
dwFormatType = SR_FORMAT_LIST;
//forms the header for the OS options
FormHeader(bHeader,ResultHeader,bVerbose);
//create dynamic array to hold the results for the BootOptions
arrResults = CreateDynamicArray();
//create dynamic array to hold the results for the BootLoader section
arrBootLoader = CreateDynamicArray();
if(arrResults == NULL || arrBootLoader == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
if (arrResults!=NULL)
DestroyDynamicArray(&arrResults);
return (EXIT_FAILURE);
}
lstrcpy(szPath, PATH_BOOTINI );
bFlag = openConnection( szServer, szUser, szPassword, szPath,bNeedPwd,stream,&bConnFlag);
if(bFlag == EXIT_FAILURE)
{
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootLoader);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
//to store entries corresponding to Operating Systems sections
arrKeyValuePairs = getKeyValueOfINISection( szPath, OS_FIELD );
//to store entries corresponding to BootLoader section
arrBootLoader = getKeysOfINISection(szPath,BOOTLOADERSECTION);
if( (arrBootLoader == NULL)||(arrKeyValuePairs == NULL))
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootLoader);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
dwCount = DynArrayGetCount(arrBootLoader);
//to display the Header Column
DISPLAY_MESSAGE(stdout,TOKEN_NEXTLINE);
DISPLAY_MESSAGE(stdout,BOOT_HEADER);
DISPLAY_MESSAGE(stdout,DASHES_BOOTOS);
// this loop is for calculating the maximum width of the the keys which will be displayed.
for(dwIndex=0;dwIndex < dwCount;dwIndex++)
{
szKeyName[dwIndex] = DynArrayItemAsString(arrBootLoader,dwIndex);
//the value correspondin to the key is obtained.
dwSectionFlag = getKeysOfSpecifiedINISection(szPath ,BOOTLOADERSECTION,szKeyName[dwIndex],szResults[dwIndex]);
if (dwSectionFlag == MALLOC_FAILURE)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootLoader);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
dwLength1 = lstrlen(szKeyName[dwIndex]);
if (dwLength1 > dwFinalLength)
{
dwFinalLength = dwLength1;
}
}
// display the results of the bootloader section.
for(dwIndex=0;dwIndex < dwCount;dwIndex++)
{
dwLength1 = dwFinalLength - lstrlen(szKeyName[dwIndex]) + 1;
DISPLAY_MESSAGE(stdout,szKeyName[dwIndex]);
_tcscpy(szDisplay,TOKEN_COLONSYMBOL);
_tcsncat(szDisplay,TOKEN_50SPACES,dwLength1);
DISPLAY_MESSAGE(stdout,szDisplay);
DISPLAY_MESSAGE(stdout,szResults[dwIndex]);
DISPLAY_MESSAGE(stdout,TOKEN_NEXTLINE);
}
DISPLAY_MESSAGE(stdout,TOKEN_NEXTLINE);
psztok = NULL ;
//getting the count of the number of boot entries
dwLength = DynArrayGetCount(arrKeyValuePairs);
for(dwCnt=0;dwCnt < dwLength;dwCnt++ )
{
dwRow = DynArrayAppendRow(arrResults,MAX_COLUMNS) ;
lstrcpy(szFriendlyName,NULL_STRING);
lstrcpy(szBootOptions,NULL_STRING);
lstrcpy(szTmpString,NULL_STRING);
if(arrKeyValuePairs != NULL)
{
LPCWSTR pwsz = NULL;
pwsz = DynArrayItemAsString( arrKeyValuePairs,dwCnt );
if(pwsz != NULL)
{
lstrcpy(szValue,pwsz);
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
SAFECLOSE(stream);
resetFileAttrib(szPath);
DestroyDynamicArray(&arrBootLoader);
DestroyDynamicArray(&arrKeyValuePairs);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
SAFECLOSE(stream);
resetFileAttrib(szPath);
DestroyDynamicArray(&arrBootLoader);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
psztok = _tcstok(szValue,TOKEN_EQUAL);
if(psztok != NULL)
{
lstrcpy(szArcPath,psztok);
psztok = _tcstok(NULL,TOKEN_FWDSLASH1 );
if(psztok != NULL)
{
lstrcpy(szFriendlyName,psztok);
lstrcat(szValue,TOKEN_STAR);
psztok = _tcstok(NULL,TOKEN_STAR);
if(psztok != NULL)
{
lstrcpy(szTmpString,psztok);
}
else
{
bTokenFlag = TRUE ;
}
}else
{
bTokenFlag = TRUE ;
}
}
else
{
bTokenFlag = TRUE ;
}
_ltow(dwCnt+1,szBootEntry,10);
DynArraySetString2( arrResults,dwRow ,COL0,szBootEntry,0 );
if(lstrlen(szFriendlyName)==0)
{
lstrcpy(szFriendlyName,TOKEN_NA);
}
DynArraySetString2( arrResults,dwRow ,COL1,szFriendlyName,0 );
DynArraySetString2(arrResults,dwRow,COL2,szArcPath,0);
if(lstrlen(szTmpString) != 0)
{
lstrcat(szBootOptions,TOKEN_FWDSLASH1);
lstrcat(szBootOptions,szTmpString);
}
else
{
lstrcpy(szBootOptions,TOKEN_NA);
}
DynArraySetString2( arrResults,dwRow ,COL3,szBootOptions,0 );
}
DISPLAY_MESSAGE(stdout,OS_HEADER);
DISPLAY_MESSAGE(stdout,DASHES_OS);
ShowResults(MAX_COLUMNS, ResultHeader, dwFormatType,arrResults ) ;
// Closing the boot.ini stream and destroying the dynamic arrays.
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootLoader);
DestroyDynamicArray(&arrKeyValuePairs);
SAFECLOSE(stream);
bRes = resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return (bRes);
}
/*****************************************************************************
Routine Description:
This function gets all the keys present in the specified section of
an .ini file and then returns the dynamic array containing all the
keys
Arguments:
[in] sziniFile : Name of the ini file.
[in] szinisection : Name of the section in the boot.ini.
Return Value :
TARRAY ( pointer to the dynamic array )
******************************************************************************/
TARRAY getKeysOfINISection( LPTSTR sziniFile, LPTSTR sziniSection )
{
// Dynamic array which will hold all the keys
TARRAY arrKeys = NULL;
// Number of characters returned by the GetPrivateProfileString function
DWORD len = 0 ;
// Variables used in the loop
DWORD i = 0 ;
DWORD j = 0 ;
// Buffer which will be populated by the GetPrivateProfileString function
LPTSTR inBuf = NULL ;
DWORD dwLength = MAX_STRING_LENGTH1;
// Temporary variable which will contain the individual keys
LPTSTR szTemp = NULL ;
inBuf = (LPTSTR)malloc(dwLength*sizeof(TCHAR));
if(inBuf==NULL)
{
return NULL ;
}
szTemp = (LPTSTR)malloc(dwLength*sizeof(TCHAR));
if((szTemp == NULL))
{
SAFEFREE(inBuf);
return NULL ;
}
memset(inBuf,0,dwLength);
memset(szTemp,0,dwLength);
while(1)
{
// Getting all the keys from the boot.ini file
len = GetPrivateProfileString (sziniSection,
NULL,
ERROR_PROFILE_STRING,
inBuf,
dwLength,
sziniFile);
//if the size of the string is not sufficient then increment the size.
if(len == dwLength-2)
{
dwLength +=100 ;
inBuf = (LPTSTR)realloc(inBuf,dwLength*sizeof(TCHAR));
if(inBuf == NULL)
{
SAFEFREE(inBuf);
SAFEFREE(szTemp);
return NULL ;
}
szTemp = (LPTSTR)realloc(szTemp,dwLength*sizeof(TCHAR));
if(szTemp == NULL)
{
SAFEFREE(inBuf);
SAFEFREE(szTemp);
return NULL ;
}
}
else
break ;
}
// Creating a dynamic array by using the function in the DynArray.c module.
// This dynamic array will contain all the keys.
arrKeys = CreateDynamicArray();
if(arrKeys == NULL)
{
SAFEFREE(inBuf);
SAFEFREE(szTemp);
return NULL ;
}
// Looping through the characters returned by the above function
while(i<len)
{
// Each individual key will be got in arrTest array
szTemp[ j++ ] = inBuf[ i ];
if( inBuf[ i ] == TOKEN_DELIM )
{
// Setting j to 0 to start the next key.
j = 0;
// Appending each key to the dynamic array
DynArrayAppendString( arrKeys, szTemp, 0 );
if(lstrlen(szTemp)==0)
{
SAFEFREE(inBuf);
SAFEFREE(szTemp);
return NULL ;
}
}
// Incrementing loop variable
i++;
}
SAFEFREE(inBuf);
SAFEFREE(szTemp);
// returning the dynamic array containing all the keys
return arrKeys;
}
/*****************************************************************************
Routine Description:
This function gets all the key-value pairs of the [operating systems]
section and returns a dynamic array containing all the key-value pairs
Arguments:
[in] sziniFile : Name of the ini file.
[in] szinisection : Name of the section in the boot.ini.
Return Value :
TARRAY ( pointer to the dynamic array )
******************************************************************************/
TARRAY getKeyValueOfINISection( LPTSTR iniFile, LPTSTR sziniSection )
{
// Dynamic array which will hold all the key-value pairs
TARRAY arrKeyValue = NULL;
// Number of characters returned by the GetPrivateProfileSection function
DWORD len = 0;
// Variables used in the loop
DWORD i = 0 ;
DWORD j = 0 ;
LPTSTR inbuf = NULL;
// Buffer which will be populated by the GetPrivateProfileSection function
// Temporary variable which will contain the individual keys
LPTSTR szTemp = NULL ;
DWORD dwLength = MAX_STRING_LENGTH1 ;
// Initialising loop variables
i = 0;
j = 0;
//return NULL if failed to allocate memory.
inbuf = (LPTSTR)malloc(dwLength*sizeof(TCHAR));
if(inbuf==NULL)
{
return NULL ;
}
//return NULL if failed to allocate memory
szTemp = (LPTSTR)malloc(dwLength*sizeof(TCHAR));
if(szTemp == NULL)
{
SAFEFREE(inbuf);
return NULL ;
}
memset(inbuf,0,dwLength);
while(1)
{
// Getting all the key-value pairs from the boot.ini file
len = GetPrivateProfileSection (sziniSection, inbuf,dwLength, iniFile);
if(len == dwLength -2)
{
dwLength +=100 ;
inbuf = (LPTSTR)realloc(inbuf,dwLength*sizeof(TCHAR));
szTemp = (LPTSTR)realloc(szTemp,dwLength*sizeof(TCHAR));
if((inbuf== NULL)||(szTemp==NULL))
{
SAFEFREE(inbuf);
SAFEFREE(szTemp);
return NULL ;
}
}
else
break ;
}
inbuf[lstrlen(inbuf)] = '\0';
// Creating a dynamic array by using the function in the DynArray.c module.
// This dynamic array will contain all the key-value pairs.
arrKeyValue = CreateDynamicArray();
if(arrKeyValue == NULL)
{
SAFEFREE(inbuf);
SAFEFREE(szTemp);
return NULL ;
}
// Looping through the characters returned by the above function
while(i<len)
{
// Each individual key will be got in arrTest array
szTemp[ j++ ] = inbuf[ i ];
if( inbuf[ i ] == TOKEN_DELIM)
{
szTemp[j+1] = '\0';
// Setting j to 0 to start the next key.
j = 0;
// Appending each key-value to the dynamic array
DynArrayAppendString( arrKeyValue, szTemp, 0 );
if(lstrlen(szTemp)==0)
{
SAFEFREE(inbuf);
SAFEFREE(szTemp);
return NULL ;
}
}
// Incrementing loop variable
i++;
}
// returning the dynamic array containing all the key-value pairs
SAFEFREE(inbuf);
SAFEFREE(szTemp);
return arrKeyValue;
}
/*****************************************************************************
Routine Description:
This function deletes a key from an ini section of an ini file
Arguments:
[in] szKey : Name of the key which has to be deleted
from the given section present in the
given ini file
[in] sziniFile : Name of the ini file.
[in] szinisection : Name of the section in the boot.ini.
Return Value :
BOOL (TRUE if there is no error, else the value is FALSE)
******************************************************************************/
BOOL deleteKeyFromINISection( LPTSTR szKey, LPTSTR sziniFile, LPTSTR sziniSection )
{
// If the third parameter (default value) is NULL, the key pointed to by
// the key parameter is deleted from the specified section of the specified
// INI file
if( WritePrivateProfileString( sziniSection, szKey, NULL, sziniFile ) == 0 )
{
// If there is an error while writing then return false
return FALSE;
}
// If there is no error, then return true
return TRUE;
}
/*****************************************************************************
Routine Description:
This function removes a sub-string from a string
Arguments:
[in] szString : Main string
[in] szSubString : Sub-string
Return Value :
VOID
******************************************************************************/
VOID removeSubString( LPTSTR szString, LPCTSTR szSubString )
{
TCHAR szFinalStr[MAX_STRING_LENGTH1] = NULL_STRING ;
DWORD dwSize = 1;
TCHAR sep[] = TOKEN_EMPTYSPACE;
PTCHAR pszToken = NULL_STRING;
// Character space is used for tokenising
lstrcpy( sep, _T(" ") );
// Getting the first token
pszToken = _tcstok( szString, sep );
while( pszToken != NULL )
{
// If the token is equal to the sub-string, then the token
// is not added to the final string. The final string contains
// all the tokens except the sub-string specified.
if(lstrcmpi( pszToken, szSubString ) != 0 )
{
lstrcpy( szFinalStr + dwSize - 1, TOKEN_EMPTYSPACE);
lstrcpy( szFinalStr + dwSize, pszToken );
dwSize = dwSize + lstrlen(pszToken) + 1;
}
// Getting the next token
pszToken = _tcstok( NULL, sep );
}
lstrcpy(szString,szFinalStr);
}
/*****************************************************************************
Routine Description:
This function establishes a connection to the specified system with
the given credentials.
Arguments:
[in] szServer : server name to coonect to
[in] szUser : User Name
[in] szPassword : password
[in] bNeedPwd : Boolean for asking the password.
[in] szPath : path of the ini file .
Return Value :
BOOL (EXIT_SUCCESS if there is no error, else the value is EXIT_FAILURE)
******************************************************************************/
BOOL openConnection( STRING256 szServer, STRING256 szUser,STRING256 szPassword,
STRING100 szPath,BOOL bNeedPwd,FILE *stream,PBOOL pbConnFlag)
{
// Declaring the file path string which will hold the path of boot.ini file
TCHAR filePath[MAX_RES_STRING] = NULL_STRING ;
// Loop variable
DWORD i = 0;
// Position of the character we are searching for
#ifndef _WIN64
DWORD dwPos = 0;
#else
__int64 dwPos = 0;
#endif
// Boolean variable which will keep trach whether boot.ini file is present in
// any drive of a system or not
BOOL bFound = FALSE;
DWORD dwRetVal = 0 ;
// All the possible drive letters in a system
// A and B are removed, since they are assigned for floppy disk drives.
TCHAR szDrives[] = { DRIVE_C, DRIVE_D, DRIVE_E, DRIVE_F, DRIVE_G, DRIVE_H, DRIVE_I, DRIVE_J,DRIVE_K, DRIVE_L,
DRIVE_M, DRIVE_N, DRIVE_O, DRIVE_P,DRIVE_Q,DRIVE_R,DRIVE_S,DRIVE_T,DRIVE_U,
DRIVE_V,DRIVE_W,DRIVE_X,DRIVE_Y,DRIVE_Z};
// Pointer to the first occurence of the character '$'
TCHAR *pdest = NULL;
BOOL bResult = FALSE;
INT nRetVal = 0;
*pbConnFlag = TRUE ;
if( lstrcmpi(szServer, NULL_STRING) != 0 )
{
//bResult = EstablishConnection(szServer,szUser,SIZE_OF_ARRAY(szUser),szPassword,SIZE_OF_ARRAY(szPassword),bNeedPwd);
bResult = EstablishConnection(szServer,szUser,256,szPassword,SIZE_OF_ARRAY(szPassword),bNeedPwd);
if (bResult == FALSE)
{
DISPLAY_MESSAGE( stderr,ERROR_TAG );
DISPLAY_MESSAGE( stderr, GetReason());
return EXIT_FAILURE ;
}
else
{
switch( GetLastError() )
{
case I_NO_CLOSE_CONNECTION:
*pbConnFlag = FALSE ;
break;
case E_LOCAL_CREDENTIALS:
case ERROR_SESSION_CREDENTIAL_CONFLICT:
{
*pbConnFlag = FALSE ;
break;
}
}
}
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
// Building the file path of boot.ini file
// File Path is in \\server\C$\boot.ini format.
// Right now assuming that boot.ini file is in C Drive
lstrcpy(filePath, TOKEN_BACKSLASH4);
lstrcat(filePath, szServer);
lstrcat(filePath, TOKEN_BACKSLASH2);
lstrcat(filePath, TOKEN_C_DOLLAR);
lstrcat(filePath, TOKEN_BACKSLASH2);
lstrcat(filePath, TOKEN_BOOTINI_PATH);
lstrcpy(szPath, filePath);
// Finding the drive containing the boot.ini file
// For remote computer the path will be in the format
// \\MACHINE\DRIVELETTER$\boot.ini
while( i < 25 )
{
pdest = _tcschr(filePath, TOKEN_DOLLAR);
if (pdest==NULL)
{
return FALSE ;
}
dwPos = pdest - filePath ;
filePath[dwPos - 1] = szDrives[i];
stream = _tfopen(filePath,READ_MODE);
// If the boot.ini is found
if( stream != NULL )
{
bFound = TRUE;
fclose(stream);
break;
}
i++;
}
}
else
{
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
lstrcpy(filePath, TOKEN_PATH);
// Finding the drive containing the boot.ini file
// For local computer the path will be in the format
// DRIVELETTER:\boot.ini
while( i < 25 )
{
pdest = _tcschr(filePath,_T(':'));
if (pdest==NULL)
{
return FALSE ;
}
dwPos = pdest - filePath + 1;
filePath[dwPos - 2] = szDrives[i];
stream = _tfopen(filePath, READ_MODE);
// If the boot.ini is found
if(stream != NULL )
{
fclose(stream);
bFound = TRUE;
break;
}
i++;
}
}
// If boot.ini is not found in any drives, then display error message
if( bFound == FALSE )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_BOOTINI));
return EXIT_FAILURE ;
}
nRetVal = _tchmod(filePath, _S_IREAD | _S_IWRITE);
// Changing the file permissions of the boot.ini file
if( nRetVal != 0 )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_READWRITE_BOOTINI));
return EXIT_FAILURE ;
}
else if (nRetVal == -1)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_FILE));
return EXIT_FAILURE ;
}
// Open the Boot.ini file with both read and write mode
// If there is no error in opening the file, then return success
// of the boot.ini file to the calling function
return EXIT_SUCCESS ;
}
// ***************************************************************************
// Routine Description:
// This function fetches 64 bit Delete Usage information from resource file and displays it
//
// Arguments:
// None
//
// Return Value:
// void
// ***************************************************************************
void displayDeleteUsage_IA64()
{
DWORD dwIndex = ID_DEL_HELP_IA64_BEGIN;
for(;dwIndex <= ID_DEL_HELP_IA64_END;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description:
// This function fetches 32 bit Delete Usage information from resource file and displays it
//
// Arguments:
// None
//
// Return Value:
// void
// ***************************************************************************
void displayDeleteUsage_X86()
{
DWORD dwIndex = ID_DEL_HELP_BEGIN;
for(;dwIndex <= ID_DEL_HELP_END;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description:
// This function fetches 64 bit Copy Usage information from resource file and displays it
//
// Arguments:
// None
//
// Return Value:
// void
// ***************************************************************************
VOID displayCopyUsage_IA64()
{
DWORD dwIndex = ID_COPY_HELP_IA64_BEGIN;
for(;dwIndex <=ID_COPY_HELP_IA64_END;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description:
// This function fetches 32 bit Copy Usage information from resource file and displays it
//
// Arguments:
// None
//
// Return Value:
// void
// ***************************************************************************
VOID displayCopyUsage_X86()
{
DWORD dwIndex = ID_COPY_HELP_BEGIN;
for(;dwIndex <=ID_COPY_HELP_END;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description:
// This function fetches Query Usage information from resource file and displays it
//
// Arguments:
// None
//
// Return Value:
// void
// ***************************************************************************
VOID displayQueryUsage()
{
#ifdef _WIN64
displayQueryUsage_IA64();
#else
displayQueryUsage_X86();
#endif
}
// ***************************************************************************
// Routine Description:
// This function fetches Query Usage information from resource file and displays it
//
// Arguments:
// None
//
// Return Value:
// void
// ***************************************************************************
VOID displayQueryUsage_IA64()
{
DWORD dwIndex = ID_QUERY_HELP64_BEGIN ;
for(;dwIndex <= ID_QUERY_HELP64_END ;dwIndex++ )
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description:
// This function fetches Query Usage information from resource file and displays it
//
// Arguments:
// None
//
// Return Value:
// void
// ***************************************************************************
VOID displayQueryUsage_X86()
{
DWORD dwIndex = ID_QUERY_HELP_BEGIN ;
for(;dwIndex <= ID_QUERY_HELP_END;dwIndex++ )
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description:
// This function fetches Main Usage information from resource file and displays it
//
// Arguments:
// None
//
// Return Value:
// void
// ***************************************************************************
DWORD displayMainUsage_X86()
{
TCHAR szServer[MAX_RES_STRING] = NULL_STRING ;
DWORD dwRetVal = 0;
DWORD dwIndex = ID_MAIN_HELP_BEGIN1 ;
//display the error message if the target system is a 64 bit system or if error occured in
//retreiving the information
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return (EXIT_FAILURE);
}
for(;dwIndex <= ID_MAIN_HELP_END1 ;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
return EXIT_SUCCESS ;
}
// ***************************************************************************
// Routine Description:
// This function fetches Usage information for the 64 bit system
//
// Arguments:
// None
//
// Return Value:
// void
// ***************************************************************************
VOID displayMainUsage_IA64()
{
DWORD dwIndex = ID_MAIN_HELP_IA64_BEGIN ;
for(;dwIndex <= ID_MAIN_HELP_IA64_END ;dwIndex++)
{
if( (dwIndex == IDS_MAIN_HELP23_IA64 ) || (dwIndex == IDS_MAIN_HELP24_IA64) || (dwIndex == IDS_MAIN_HELP38_IA64) )
{
continue;
}
else
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
}
/*****************************************************************************
Routine Description:
This function resets the permissions with the original set of
permissions ( -readonly -hidden -system )
and then exits with the given exit code.
Arguments
[in] szFilePath : File Path of the boot.ini file
Return Value :
BOOL (EXIT_SUCCESS if there is no error, else the value is EXIT_FAILURE)
******************************************************************************/
BOOL resetFileAttrib( LPTSTR szFilePath )
{
if(szFilePath==NULL)
{
return FALSE ;
}
// Resetting the file permission of the boot.ini file to its original
// permission list( -r, -h, -s )
if( _tchmod(szFilePath, _S_IREAD != 0 ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_RESET_ERROR));
return EXIT_FAILURE ;
}
return EXIT_SUCCESS ;
}
/*****************************************************************************
Routine Description:
This function returns a string from a dynamic array .
Arguments
[in] arrKeyValuePairs : Dynamic array which contains all the
key-value pairs.
[out] szFiinalStr : String which is formed from all the key-value pairs
Return Value :
BOOL (EXIT_SUCCESS if there is no error, else the value is EXIT_FAILURE)
******************************************************************************/
BOOL stringFromDynamicArray1( TARRAY arrKeyValuePairs ,LPTSTR szFinalStr )
{
// Total number of elements in the array
DWORD dwKeyValueCount = 0;
// Variable used to keep track the current position while appending strings.
DWORD dwStrSize = 0;
// Loop variable
DWORD i = 0;
// Initialsing size and loop variables to 0
dwStrSize = 0;
i = 0;
if( (arrKeyValuePairs ==NULL)||(szFinalStr==NULL))
{
return EXIT_FAILURE ;
}
// Getting the total number of key-value pairs
if(arrKeyValuePairs != NULL)
{
dwKeyValueCount = DynArrayGetCount(arrKeyValuePairs);
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
return (EXIT_FAILURE) ;
}
// Looping through all the key-value pairs and building the final string
// containing all the key value pairs. This string has to be passed to
// WriteProfileSection
while( (i <= dwKeyValueCount - 1 )&& (arrKeyValuePairs != NULL) )
{
// Building the final string, by getting each key-value pair present in the
// dynamic array
if(arrKeyValuePairs != NULL)
{
LPCWSTR pwsz = NULL;
pwsz = DynArrayItemAsString( arrKeyValuePairs, i ) ;
if(pwsz != NULL)
{
lstrcpy(szFinalStr + dwStrSize, pwsz );
dwStrSize = dwStrSize + lstrlen(pwsz) + 1;
i++;
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
return EXIT_FAILURE ;
}
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
return EXIT_FAILURE ;
}
}
return EXIT_SUCCESS ;
}
// ***************************************************************************
// Routine Description:
// This function is used to build the header and also display the
// result in the required format as specified by the user.
//
// Arguments:
// [ in ] arrResults : argument(s) count specified at the command prompt
// [ in ] dwFormatType : format flags
// [ in ] bHeader : Boolean for specifying if the header is required or not.
//
// Return Value:
// none
//
// ***************************************************************************
VOID FormHeader(BOOL bHeader,TCOLUMNS *ResultHeader,BOOL bVerbose)
{
//OS Entry
ResultHeader[COL0].dwWidth = COL_BOOTOPTION_WIDTH ;
ResultHeader[COL0].dwFlags = SR_TYPE_STRING;
ResultHeader[COL0].pFunction = NULL;
ResultHeader[COL0].pFunctionData = NULL;
lstrcpy( ResultHeader[COL0].szFormat, NULL_STRING );
lstrcpy( ResultHeader[COL0].szColumn,COL_BOOTOPTION );
ResultHeader[COL1].dwWidth = COL_FRIENDLYNAME_WIDTH;
ResultHeader[COL1].dwFlags = SR_TYPE_STRING;
ResultHeader[COL1].pFunction = NULL;
ResultHeader[COL1].pFunctionData = NULL;
lstrcpy( ResultHeader[COL1].szFormat, NULL_STRING );
lstrcpy( ResultHeader[COL1].szColumn,COL_FRIENDLYNAME );
ResultHeader[COL2].dwWidth = COL_ARC_WIDTH;
ResultHeader[COL2].dwFlags = SR_TYPE_STRING;
ResultHeader[COL2].pFunction = NULL;
ResultHeader[COL2].pFunctionData = NULL;
lstrcpy( ResultHeader[COL2].szFormat, NULL_STRING );
lstrcpy( ResultHeader[COL2].szColumn,COL_ARCPATH );
ResultHeader[COL3].dwWidth = COL_BOOTID_WIDTH;
ResultHeader[COL3].dwFlags = SR_TYPE_STRING;
ResultHeader[COL3].pFunction = NULL;
ResultHeader[COL3].pFunctionData = NULL;
lstrcpy( ResultHeader[COL3].szFormat, NULL_STRING );
lstrcpy( ResultHeader[COL3].szColumn,COL_BOOTID );
}
// ***************************************************************************
// Routine Description:
// This routine is to display the current boot.ini file settings for
// the specified system.
//
// Arguments:
// [ in ] argc : Number of command line arguments
// [ in ] argv : Array containing command line arguments
// Return Value:
// DWORD
//
// ***************************************************************************
DWORD AppendRawString( DWORD argc, LPCTSTR argv[] )
{
BOOL bUsage = FALSE ;
BOOL bNeedPwd = FALSE ;
BOOL bRaw = FALSE ;
DWORD dwId = 0;
DWORD dwDefault = 0;
TARRAY arr ;
TCHAR szkey[MAX_RES_STRING] = NULL_STRING;
FILE *stream = NULL;
// Initialising the variables that are passed to TCMDPARSER structure
STRING256 szServer = NULL_STRING;
STRING256 szUser = NULL_STRING;
STRING256 szPassword = NULL_STRING;
STRING100 szPath = NULL_STRING;
STRING256 szRawString = NULL_STRING ;
DWORD dwNumKeys = 0;
BOOL bRes = FALSE ;
PTCHAR pToken = NULL ;
LPTSTR szFinalStr = NULL ;
BOOL bFlag = FALSE ;
TCHAR szBuffer[MAX_RES_STRING] = NULL_STRING ;
LPCTSTR szToken = NULL ;
DWORD dwRetVal = 0 ;
BOOL bConnFlag = FALSE ;
BOOL bAppendFlag = FALSE ;
TCHAR szErrorMsg[MAX_RES_STRING] = NULL_STRING ;
// Building the TCMDPARSER structure
TCMDPARSER cmdOptions[] =
{
{ CMDOPTION_RAW, CP_MAIN_OPTION, 1, 0,&bRaw, NULL_STRING, NULL, NULL },
{ SWITCH_SERVER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szServer, NULL_STRING, NULL, NULL },
{ SWITCH_USER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szUser, NULL_STRING, NULL, NULL },
{ SWITCH_PASSWORD, CP_TYPE_TEXT | CP_VALUE_OPTIONAL, 1, 0, &szPassword, NULL_STRING, NULL, NULL },
{ CMDOPTION_USAGE, CP_USAGE, 1, 0, &bUsage, NULL_STRING, NULL, NULL },
{ SWITCH_ID, CP_TYPE_NUMERIC | CP_VALUE_MANDATORY | CP_MANDATORY, 1, 0, &dwDefault, NULL_STRING, NULL, NULL },
{ CMDOPTION_DEFAULT, CP_DEFAULT | CP_TYPE_TEXT | CP_MANDATORY, 1, 0, &szRawString,NULL_STRING, NULL, NULL },
{ CMDOPTION_APPEND , 0, 1, 0, &bAppendFlag,NULL_STRING, NULL, NULL }
};
//
//check if the remote system is 64 bit and if so
// display an error.
//
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
szFinalStr = (TCHAR*) malloc(MAX_STRING_LENGTH1* sizeof(TCHAR) );
if (szFinalStr== NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_TAG));
ShowLastError(stderr);
return (EXIT_FAILURE);
}
_tcscpy(cmdOptions[3].szValues,TOKEN_ASTERIX) ;
_tcscpy(szPassword,TOKEN_ASTERIX);
// Parsing the copy option switches
if ( ! DoParseParam( argc, argv, SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
ShowMessage(stderr,GetReason());
SAFEFREE(szFinalStr);
return (EXIT_FAILURE);
}
// Displaying query usage if user specified -? with -query option
if( bUsage )
{
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_SUCCESS )
{
displayRawUsage_X86();
SAFEFREE(szFinalStr);
return (EXIT_SUCCESS);
}
else
{
SAFEFREE(szFinalStr);
return (EXIT_FAILURE);
}
}
// error checking in case the
// raw string does not start with a "/" .
if(*szRawString != _T('/'))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_FWDSLASH));
SAFEFREE(szFinalStr);
return (EXIT_FAILURE);
}
//display error message if the username is entered with out a machine name
if( (cmdOptions[1].dwActuals == 0)&&(cmdOptions[2].dwActuals != 0))
{
SetReason(GetResString(IDS_USER_BUT_NOMACHINE));
ShowMessage(stderr,GetReason());
SAFEFREE(szFinalStr);
return EXIT_FAILURE ;
}
if( (cmdOptions[2].dwActuals == 0)&&(cmdOptions[3].dwActuals != 0))
{
SetReason(GetResString(IDS_PASSWD_BUT_NOUSER));
ShowMessage(stderr,GetReason());
SAFEFREE(szFinalStr);
return EXIT_FAILURE ;
}
//for setting the bNeedPwd
if( ( _tcscmp(szPassword,TOKEN_ASTERIX )==0 ) && (IsLocalSystem(szServer)==FALSE ))
{
bNeedPwd = TRUE ;
}
//set the bneedpassword to true if the server name is specified and password is not specified.
if((cmdOptions[1].dwActuals!=0)&&(cmdOptions[3].dwActuals==0))
{
if( (lstrlen( szServer ) != 0) && (IsLocalSystem(szServer)==FALSE) )
{
bNeedPwd = TRUE ;
}
else
{
bNeedPwd = FALSE ;
}
if(_tcslen(szPassword)!= 0 )
{
_tcscpy(szPassword,NULL_STRING);
}
}
//display an error message if the server is empty.
if((cmdOptions[1].dwActuals!=0)&&(lstrlen(szServer)==0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_SERVER));
SAFEFREE(szFinalStr);
return EXIT_FAILURE ;
}
//display an error message if the user is empty.
if((cmdOptions[2].dwActuals!=0)&&(lstrlen(szUser)==0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_USER));
SAFEFREE(szFinalStr);
return EXIT_FAILURE ;
}
// Establishing connection to the specified machine and getting the file pointer
// of the boot.ini file if there is no error while establishing connection
lstrcpy(szPath, PATH_BOOTINI );
if(StrCmpN(szServer,TOKEN_BACKSLASH4,2)==0)
{
if(!StrCmpN(szServer,TOKEN_BACKSLASH6,3)==0)
{
szToken = _tcstok(szServer,TOKEN_BACKSLASH4);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
return (EXIT_FAILURE);
}
lstrcpy(szServer,szToken);
}
}
if( (IsLocalSystem(szServer)==TRUE)&&(lstrlen(szUser)!=0))
{
DISPLAY_MESSAGE(stdout,GetResString(WARN_LOCALCREDENTIALS));
_tcscpy(szServer,_T(""));
}
bFlag = openConnection( szServer, szUser, szPassword, szPath,bNeedPwd,stream,&bConnFlag );
if(bFlag == EXIT_FAILURE)
{
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting the keys of the Operating system section in the boot.ini file
arr = getKeyValueOfINISection( szPath, OS_FIELD );
if(arr == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
// Getting the total number of keys in the operating systems section
dwNumKeys = DynArrayGetCount(arr);
//
if((dwNumKeys >= MAX_BOOTID_VAL)&&(dwDefault >= MAX_BOOTID_VAL ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_MAX_BOOTID));
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
DestroyDynamicArray(&arr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Displaying error message if the number of keys is less than the OS entry
// line number specified by the user
if( ( dwDefault <= 0 ) || ( dwDefault > dwNumKeys ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_BOOTID));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting the key of the OS entry specified by the user
if (arr != NULL)
{
LPCWSTR pwsz = NULL;
pwsz = DynArrayItemAsString( arr, dwDefault - 1 ) ;
if(pwsz != NULL)
{
_tcscpy( szkey,pwsz);
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
if(bAppendFlag == FALSE)
{
pToken = _tcstok(szkey ,TOKEN_FWDSLASH1);
if(pToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
}
lstrcat(szkey , TOKEN_EMPTYSPACE);
CharLower(szRawString);
lstrcat(szkey ,szRawString);
if( _tcslen(szkey) >= MAX_RES_STRING)
{
_stprintf(szErrorMsg,GetResString(IDS_ERROR_STRING_LENGTH),MAX_RES_STRING);
DISPLAY_MESSAGE( stderr,szErrorMsg);
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
DynArrayRemove(arr, dwDefault - 1 );
DynArrayInsertString(arr, dwDefault - 1, szkey, MAX_RES_STRING);
// Setting the buffer to 0, to avoid any junk value
memset(szFinalStr, 0, MAX_STRING_LENGTH1);
// Forming the final string from all the key-value pairs
if (stringFromDynamicArray1( arr,szFinalStr) == EXIT_FAILURE)
{
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
// Writing to the profile section with new key-value pair
// If the return value is non-zero, then there is an error.
if( WritePrivateProfileSection(OS_FIELD, szFinalStr, szPath ) != 0 )
{
_stprintf(szBuffer,GetResString(IDS_SWITCH_ADD), dwDefault );
DISPLAY_MESSAGE(stdout,szBuffer);
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_ADD_SWITCHES));
DestroyDynamicArray(&arr);
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
//reset the file attributes and free the memory and close the connection to the server.
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_SUCCESS);
}
// ***************************************************************************
// Routine Description:
// This routine is to display the current boot.ini file settings for
// the specified system.
//
// Arguments:
// none
// Return Value:
// VOID
//
// ***************************************************************************
VOID displayRawUsage_X86()
{
DWORD dwIndex = RAW_HELP_BEGIN;
for(;dwIndex <= RAW_HELP_END;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description:
// Display the help for the 64 bit raw option.
//
// Arguments:
// none
// Return Value:
// VOID
//
// ***************************************************************************
VOID displayRawUsage_IA64()
{
DWORD dwIndex = RAW_HELP_IA64_BEGIN;
for(;dwIndex <= RAW_HELP_IA64_END;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description:
// This routine is to change the timout of the boot.ini file settings for
// the specified system.
// Arguments:
// [in ] argc : Number of command line arguments
// [in ] argv : Array containing command line arguments
//
// Return Value:
// DWORD
//
// ***************************************************************************
DWORD ChangeTimeOut(DWORD argc,LPCTSTR argv[])
{
STRING256 szServer = NULL_STRING;
STRING256 szUser = NULL_STRING;
STRING256 szPassword = NULL_STRING;
STRING100 szPath = NULL_STRING;
FILE *stream = NULL;
BOOL bNeedPwd = FALSE ;
TARRAY arrResults;
BOOL bRes= FALSE ;
DWORD dwCount = 0;
DWORD dwTimeOut = 0 ;
BOOL bFlag = 0 ;
TCHAR timeOutstr[STRING20] = NULL_STRING;
LPCTSTR szToken = NULL ;
DWORD dwRetVal = 0 ;
BOOL bConnFlag = FALSE ;
TCMDPARSER cmdOptions[] =
{
{ CMDOPTION_TIMEOUT, CP_MAIN_OPTION | CP_TYPE_UNUMERIC | CP_VALUE_OPTIONAL|CP_VALUE_MANDATORY, 1, 0,&dwTimeOut,NULL_STRING, NULL, NULL},
{ SWITCH_SERVER,CP_TYPE_TEXT|CP_VALUE_MANDATORY,1,0,&szServer,NULL_STRING,NULL,NULL},
{ SWITCH_USER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szUser, NULL_STRING, NULL, NULL },
{ SWITCH_PASSWORD, CP_TYPE_TEXT | CP_VALUE_OPTIONAL, 1, 0, &szPassword, NULL_STRING, NULL, NULL },
};
//
//check if the remote system is 64 bit and if so
// display an error.
//
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
//copy the Asterix token which is required for password prompting.
_tcscpy(cmdOptions[3].szValues,TOKEN_ASTERIX) ;
_tcscpy(szPassword,TOKEN_ASTERIX);
if( ! DoParseParam( argc, argv, SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
//display error message if the username is entered with out a machine name
if( (cmdOptions[1].dwActuals == 0)&&(cmdOptions[2].dwActuals != 0))
{
SetReason(GetResString(IDS_USER_BUT_NOMACHINE));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
if( (cmdOptions[2].dwActuals == 0)&&(cmdOptions[3].dwActuals != 0))
{
SetReason(GetResString(IDS_PASSWD_BUT_NOUSER));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
//for setting the bNeedPwd
if( ( _tcscmp(szPassword,TOKEN_ASTERIX )==0 ) && (IsLocalSystem(szServer)==FALSE ))
{
bNeedPwd = TRUE ;
}
//set the bneedpassword to true if the server name is specified and password is not specified.
if((cmdOptions[1].dwActuals!=0)&&(cmdOptions[3].dwActuals==0))
{
if( (lstrlen( szServer ) != 0) && (IsLocalSystem(szServer)==FALSE) )
{
bNeedPwd = TRUE ;
}
else
{
bNeedPwd = FALSE ;
}
if(_tcslen(szPassword)!= 0 )
{
_tcscpy(szPassword,NULL_STRING);
}
}
//display an error message if the server is empty.
if((cmdOptions[1].dwActuals!=0)&&(lstrlen(szServer)==0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_SERVER));
return EXIT_FAILURE ;
}
//display an error message if the user is empty.
if( (cmdOptions[2].dwActuals!=0)&&(lstrlen(szUser)==0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_USER));
return EXIT_FAILURE ;
}
// Establishing connection to the specified machine and getting the file pointer
// of the boot.ini file if there is no error while establishing connection
lstrcpy(szPath, PATH_BOOTINI );
if(StrCmpN(szServer,TOKEN_BACKSLASH4,2)==0)
{
if(!StrCmpN(szServer,TOKEN_BACKSLASH6,3)==0)
{
szToken = _tcstok(szServer,TOKEN_BACKSLASH4);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
return (EXIT_FAILURE);
}
lstrcpy(szServer,szToken);
}
}
//display a warning message if it is a local system and set the server name to empty.
if( (IsLocalSystem(szServer)==TRUE)&&(lstrlen(szUser)!=0))
{
DISPLAY_MESSAGE(stdout,GetResString(WARN_LOCALCREDENTIALS));
_tcscpy(szServer,_T(""));
}
bFlag = openConnection( szServer, szUser, szPassword, szPath,bNeedPwd,stream,&bConnFlag);
if(bFlag == EXIT_FAILURE)
{
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
arrResults = CreateDynamicArray();
if(arrResults == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
resetFileAttrib(szPath);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return (EXIT_FAILURE);
}
dwCount = DynArrayGetCount(arrResults);
if(dwTimeOut > TIMEOUT_MAX)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_TIMEOUT_RANGE));
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return (EXIT_FAILURE);
}
// Converting the numeric value to string because the WritePrivateProfileString
// takes only string value as the value for a particular key
_itot( dwTimeOut, timeOutstr, 10 );
// Changing the timeout value
if( WritePrivateProfileString( BOOTLOADERSECTION,TIMEOUT_SWITCH,
timeOutstr, szPath ) != 0 )
{
DestroyDynamicArray(&arrResults);
DISPLAY_MESSAGE(stdout,GetResString(IDS_TIMEOUT_CHANGE));
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return EXIT_SUCCESS ;
}
// DISPLAY Error message and exit with Error code of 1.
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_TIMEOUT));
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
// ***************************************************************************
// Routine Description:
// Display the help for the timeout option.
// Arguments:
// NONE.
// Return Value:
// VOID
//
// ***************************************************************************
VOID displayTimeOutUsage_X86()
{
DWORD dwIndex = TIMEOUT_HELP_BEGIN;
for(;dwIndex <= TIMEOUT_HELP_END;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description:
// Display the help for the 64 BIT timeout option.
// Arguments:
// NONE.
// Return Value:
// VOID
//
// ***************************************************************************
VOID displayTimeOutUsage_IA64()
{
DWORD dwIndex = TIMEOUT_HELP_IA64_BEGIN;
for(;dwIndex <= TIMEOUT_HELP_IA64_END;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description:
// This routine is to change the Default OS boot.ini file settings for
// the specified system.
// Arguments:
// [IN] argc Number of command line arguments
// [IN] argv Array containing command line arguments
//
// Return Value:
// DWORD (EXIT_SUCCESS for success and EXIT_FAILURE for Failure.
//
// ***************************************************************************
DWORD ChangeDefaultOs(DWORD argc,LPCTSTR argv[])
{
STRING256 szServer = NULL_STRING ;
STRING256 szUser = NULL_STRING ;
STRING256 szPassword = NULL_STRING ;
DWORD dwId = 0;
BOOL bDefaultOs = FALSE ;
STRING100 szPath ;
FILE *stream = NULL;
BOOL bNeedPwd = FALSE ;
TARRAY arrResults ;
DWORD dwCount = 0;
BOOL bFlag = FALSE ;
PTCHAR psztok = NULL ;
TCHAR szTmp[MAX_RES_STRING] = NULL_STRING ;
TCHAR szTmpBootId[MAX_RES_STRING] = NULL_STRING ;
TCHAR szDefaultId[MAX_RES_STRING] = NULL_STRING ;
DWORD dwValue = 0 ;
BOOL bExitFlag = FALSE ;
LPCWSTR pwsz[MAX_RES_STRING] ;
LPCWSTR pwszBootId [MAX_RES_STRING] ;
LPTSTR szFinalStr = NULL ;
LPCTSTR szToken = NULL ;
DWORD dwRetVal = 0 ;
BOOL bConnFlag = FALSE ;
BOOL bRes = FALSE;
TCMDPARSER cmdOptions[] =
{
{ CMDOPTION_DEFAULTOS, CP_MAIN_OPTION, 1, 0,&bDefaultOs, NULL_STRING, NULL, NULL },
{ SWITCH_SERVER,CP_TYPE_TEXT|CP_VALUE_MANDATORY,1,0,&szServer,NULL_STRING,NULL,NULL},
{ SWITCH_USER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szUser, NULL_STRING, NULL, NULL },
{ SWITCH_PASSWORD, CP_TYPE_TEXT | CP_VALUE_OPTIONAL, 1, 0, &szPassword, NULL_STRING, NULL, NULL },
{ SWITCH_ID,CP_TYPE_NUMERIC | CP_VALUE_MANDATORY| CP_MANDATORY, 1, 0, &dwId, NULL_STRING, NULL, NULL }
};
//
//check if the remote system is 64 bit and if so
// display an error.
//
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
//copy the Asterix token which is required for password prompting.
_tcscpy(cmdOptions[3].szValues,TOKEN_ASTERIX) ;
_tcscpy(szPassword,TOKEN_ASTERIX);
if( ! DoParseParam( argc, argv, SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
if(dwId <= 0)
{
SetReason(GetResString( IDS_INVALID_OSID));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
//display error message if the username is entered with out a machine name
if( (cmdOptions[1].dwActuals == 0)&&(cmdOptions[2].dwActuals != 0))
{
SetReason(GetResString(IDS_USER_BUT_NOMACHINE));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
if( (cmdOptions[2].dwActuals == 0)&&(cmdOptions[3].dwActuals != 0))
{
SetReason(GetResString(IDS_PASSWD_BUT_NOUSER));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
//display an error message if the server is empty.
if((cmdOptions[1].dwActuals!=0)&&(lstrlen(szServer)==0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_SERVER));
return EXIT_FAILURE ;
}
//display an error message if the user is empty.
if((cmdOptions[2].dwActuals!=0)&&(lstrlen(szUser)==0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_USER));
return EXIT_FAILURE ;
}
//for setting the bNeedPwd
if( ( _tcscmp(szPassword,TOKEN_ASTERIX )==0 ) && (IsLocalSystem(szServer)==FALSE ))
{
bNeedPwd = TRUE ;
}
//set the bneedpassword to true if the server name is specified and password is not specified.
if((cmdOptions[1].dwActuals!=0)&&(cmdOptions[3].dwActuals==0))
{
if( (lstrlen( szServer ) != 0) && (IsLocalSystem(szServer)==FALSE) )
{
bNeedPwd = TRUE ;
}
else
{
bNeedPwd = FALSE ;
}
if(_tcslen(szPassword)!= 0 )
{
_tcscpy(szPassword,NULL_STRING);
}
}
// Establishing connection to the specified machine and getting the file pointer
// of the boot.ini file if there is no error while establishing connection
lstrcpy(szPath, PATH_BOOTINI );
if(StrCmpN(szServer,TOKEN_BACKSLASH4,2)==0)
{
if(!StrCmpN(szServer,TOKEN_BACKSLASH6,3)==0)
{
szToken = _tcstok(szServer,TOKEN_BACKSLASH4);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
return (EXIT_FAILURE);
}
lstrcpy(szServer,szToken);
}
}
if( (IsLocalSystem(szServer)==TRUE)&&(lstrlen(szUser)!=0))
{
DISPLAY_MESSAGE(stdout,GetResString(WARN_LOCALCREDENTIALS));
_tcscpy(szServer,_T(""));
}
bFlag = openConnection( szServer, szUser, szPassword, szPath,bNeedPwd,stream,&bConnFlag);
if(bFlag == EXIT_FAILURE)
{
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
arrResults = CreateDynamicArray();
//return failure if failed to allocate memory
if(arrResults == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return (EXIT_FAILURE);
}
arrResults = getKeyValueOfINISection( szPath, OS_FIELD );
if(arrResults == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
dwCount = DynArrayGetCount(arrResults);
if(dwId<=0 || dwId > dwCount )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_OSID));
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
if(arrResults !=NULL)
{
pwsz[0] = DynArrayItemAsString(arrResults, dwId - 1);
pwszBootId[0] = DynArrayItemAsString(arrResults, 0);
if( (pwsz != NULL) || (pwszBootId != NULL) )
{
lstrcpy(szTmp,pwsz[0]) ;
lstrcpy(szDefaultId,pwsz[0]);
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
szFinalStr = (LPTSTR)malloc(MAX_STRING_LENGTH1*sizeof(TCHAR));
if(szFinalStr == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
//loop through all the Boot entries and
for(dwValue = dwId -1 ; dwValue > 0 ; dwValue-- )
{
pwszBootId[dwValue] = DynArrayItemAsString(arrResults,dwValue );
pwsz[dwValue] = DynArrayItemAsString(arrResults, dwValue - 1);
DynArrayRemove(arrResults, dwValue );
if (pwsz[dwValue] != NULL)
{
lstrcpy(szTmpBootId,pwsz[dwValue]) ;
}
else
{
bExitFlag = TRUE ;
break ;
}
DynArrayInsertString(arrResults, dwValue, szTmpBootId, MAX_RES_STRING);
}
if (bExitFlag == TRUE )
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
return EXIT_FAILURE ;
}
DynArrayRemove(arrResults, 0 );
DynArrayInsertString(arrResults, 0, szDefaultId, MAX_RES_STRING);
// Setting the buffer to 0, to avoid any junk value
memset(szFinalStr, 0, MAX_STRING_LENGTH1);
if(stringFromDynamicArray1( arrResults,szFinalStr) == EXIT_FAILURE )
{
bExitFlag = TRUE ;
}
// Writing to the profile section with new key-value pair
// If the return value is non-zero, then there is an error.
if( ( WritePrivateProfileSection(OS_FIELD, szFinalStr, szPath ) == 0 ) || (bExitFlag == TRUE ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERR_CHANGE));
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
return EXIT_FAILURE ;
}
//to strip of the unwanted string from the string and save the required part in the Boot Loader section.
szToken = _tcstok(szTmp,TOKEN_EQUAL);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
return (EXIT_FAILURE);
}
if( WritePrivateProfileString( BOOTLOADERSECTION, KEY_DEFAULT, szTmp,
szPath ) != 0 )
{
DISPLAY_MESSAGE(stdout,GetResString(IDS_DEF_CHANGE));
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERR_CHANGE));
DestroyDynamicArray(&arrResults);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
return EXIT_FAILURE ;
}
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
return bRes ;
}
// ***************************************************************************
// Routine Description : Display the help for the default entry option (x86).
//
// Parameters : none
//
// Return Type : VOID
//
// ***************************************************************************
VOID displayChangeOSUsage_X86()
{
DWORD dwIndex = DEFAULT_BEGIN;
for(;dwIndex <=DEFAULT_END;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
//
// Routine Description : Display the help for the default entry option (IA64).
//
// Parameters : none
//
// Return Type : VOID
//
// ***************************************************************************
VOID displayDefaultEntryUsage_IA64()
{
DWORD dwIndex = DEFAULT_IA64_BEGIN;
for(;dwIndex <=DEFAULT_IA64_END;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description:
// Implement the Debug switch.
// Arguments:
// [IN] argc Number of command line arguments
// [IN] argv Array containing command line arguments
//
// Return Value:
// DWORD (EXIT_SUCCESS for success and EXIT_FAILURE for Failure.)
//
// ***************************************************************************
DWORD ProcessDebugSwitch( DWORD argc, LPCTSTR argv[] )
{
BOOL bUsage = FALSE ;
BOOL bNeedPwd = FALSE ;
BOOL bDebug = FALSE ;
DWORD dwId = 0;
TARRAY arrResults ;
FILE *stream = NULL;
// Initialising the variables that are passed to TCMDPARSER structure
STRING256 szServer = NULL_STRING;
STRING256 szUser = NULL_STRING;
STRING256 szPassword = NULL_STRING;
STRING100 szPath = NULL_STRING;
TCHAR szDebug[MAX_RES_STRING] = NULL_STRING ;
TCHAR szPort[MAX_RES_STRING] = NULL_STRING ;
TCHAR szBoot[MAX_RES_STRING] = NULL_STRING ;
BOOL bRes = FALSE ;
LPTSTR szFinalStr = NULL ;
BOOL bFlag = FALSE ;
DWORD dwBaudRate = 0 ;
DWORD dwCount = 0 ;
DWORD dwSectionFlag = 0 ;
TCHAR szBuffer[MAX_RES_STRING] = NULL_STRING ;
TCHAR szTmpBuffer[MAX_RES_STRING] = NULL_STRING ;
TCHAR szBaudRate[MAX_RES_STRING] = NULL_STRING ;
TCHAR szString[MAX_STRING_LENGTH1] = NULL_STRING ;
TCHAR szTemp[MAX_RES_STRING] = NULL_STRING ;
_TCHAR *szValues[2] = {NULL};
TCHAR szMessage[MAX_STRING_LENGTH] = NULL_STRING;
TCHAR szErrorMsg[MAX_RES_STRING] = NULL_STRING ;
DWORD dwCode = 0 ;
LPCTSTR szToken = NULL ;
DWORD dwRetVal = 0 ;
BOOL bConnFlag = FALSE ;
// Building the TCMDPARSER structure
TCMDPARSER cmdOptions[] =
{
{ CMDOPTION_DEBUG, CP_MAIN_OPTION, 1, 0,&bDebug, NULL_STRING, NULL, NULL },
{ SWITCH_SERVER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szServer, NULL_STRING, NULL, NULL },
{ SWITCH_USER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szUser, NULL_STRING, NULL, NULL },
{ SWITCH_PASSWORD, CP_TYPE_TEXT | CP_VALUE_OPTIONAL, 1, 0, &szPassword, NULL_STRING, NULL, NULL },
{ CMDOPTION_USAGE, CP_USAGE, 1, 0, &bUsage, NULL_STRING, NULL, NULL },
{ SWITCH_ID, CP_TYPE_NUMERIC | CP_VALUE_MANDATORY | CP_MANDATORY, 1, 0, &dwId, NULL_STRING, NULL, NULL },
{ SWITCH_PORT, CP_TYPE_TEXT | CP_VALUE_MANDATORY|CP_MODE_VALUES,1,0,&szPort,COM_PORT_RANGE,NULL,NULL},
{ SWITCH_BAUD, CP_TYPE_TEXT | CP_VALUE_MANDATORY |CP_MODE_VALUES,1,0,&szBaudRate,BAUD_RATE_VALUES_DEBUG,NULL,NULL},
{ CMDOPTION_DEFAULT, CP_DEFAULT | CP_TYPE_TEXT | CP_MANDATORY , 1, 0, &szDebug,NULL_STRING, NULL, NULL }
};
//
//check if the remote system is 64 bit and if so
// display an error.
//
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
//copy the Asterix token which is required for password prompting.
_tcscpy(cmdOptions[3].szValues,TOKEN_ASTERIX) ;
_tcscpy(szPassword,TOKEN_ASTERIX);
// Parsing the copy option switches
if ( ! DoParseParam( argc, argv, SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
ShowMessage(stderr,GetReason());
return (EXIT_FAILURE);
}
if(bUsage)
{
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_SUCCESS )
{
displayDebugUsage_X86();
return (EXIT_SUCCESS);
}else
{
return (EXIT_FAILURE);
}
}
if(dwId <= 0)
{
SetReason(GetResString( IDS_INVALID_OSID));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
//display error message if the username is entered with out a machine name
if( (cmdOptions[1].dwActuals == 0)&&(cmdOptions[2].dwActuals != 0))
{
SetReason(GetResString(IDS_USER_BUT_NOMACHINE));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
if( (cmdOptions[2].dwActuals == 0)&&(cmdOptions[3].dwActuals != 0))
{
SetReason(GetResString(IDS_PASSWD_BUT_NOUSER));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
//for setting the bNeedPwd
if( ( _tcscmp(szPassword,TOKEN_ASTERIX )==0 ) && (IsLocalSystem(szServer)==FALSE ))
{
bNeedPwd = TRUE ;
}
//set the bneedpassword to true if the server name is specified and password is not specified.
if((cmdOptions[1].dwActuals!=0)&&(cmdOptions[3].dwActuals==0))
{
if( (lstrlen( szServer ) != 0) && (IsLocalSystem(szServer)==FALSE) )
{
bNeedPwd = TRUE ;
}
else
{
bNeedPwd = FALSE ;
}
if(_tcslen(szPassword)!= 0 )
{
_tcscpy(szPassword,NULL_STRING);
}
}
//display an error message if the server is empty.
if( (cmdOptions[1].dwActuals!=0)&&(lstrlen(szServer)==0) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_SERVER));
return EXIT_FAILURE ;
}
//display an error message if the user is empty.
if( (cmdOptions[2].dwActuals!=0) && (lstrlen(szUser)==0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_USER));
return EXIT_FAILURE ;
}
//display an error message if the user specifies any string other than on,off,edit.
if( !( ( lstrcmpi(szDebug,VALUE_ON)== 0)|| (lstrcmpi(szDebug,VALUE_OFF)== 0) ||(lstrcmpi(szDebug,EDIT_STRING)== 0) ))
{
szValues[0]= (_TCHAR *)szDebug ;
szValues[1]= (_TCHAR *)CMDOPTION_DEBUG ;
FormatMessage(FORMAT_MESSAGE_FROM_STRING|FORMAT_MESSAGE_ARGUMENT_ARRAY,
GetResString(IDS_OPTION_SNTAX_ERROR),0,MAKELANGID(LANG_NEUTRAL,
SUBLANG_DEFAULT),szMessage, MAX_STRING_LENGTH,(va_list*)szValues);
DISPLAY_MESSAGE(stderr,szMessage);
return EXIT_FAILURE;
}
if( (lstrcmpi(szDebug,EDIT_STRING)== 0)&& (lstrlen(szPort)==0) && (lstrlen(szBaudRate)==0) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_EDIT_SYNTAX));
return EXIT_FAILURE;
}
// Establishing connection to the specified machine and getting the file pointer
// of the boot.ini file if there is no error while establishing connection
arrResults = CreateDynamicArray();
//return failure if failed to allocate memory
if(arrResults == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
return (EXIT_FAILURE);
}
lstrcpy(szPath, PATH_BOOTINI );
if(StrCmpN(szServer,TOKEN_BACKSLASH4,2)==0)
{
if(!StrCmpN(szServer,TOKEN_BACKSLASH6,3)==0)
{
szToken = _tcstok(szServer,TOKEN_BACKSLASH4);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
return (EXIT_FAILURE);
}
lstrcpy(szServer,szToken);
}
}
// display a warning message if it is a local system and set the
// server name to empty.
if( (IsLocalSystem(szServer)==TRUE)&&(lstrlen(szUser)!=0))
{
DISPLAY_MESSAGE(stdout,GetResString(WARN_LOCALCREDENTIALS));
lstrcpy(szServer,_T(""));
}
bFlag = openConnection( szServer, szUser, szPassword, szPath,bNeedPwd,stream,&bConnFlag);
if(bFlag == EXIT_FAILURE)
{
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
arrResults = getKeyValueOfINISection( szPath, OS_FIELD );
if(arrResults == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
//getting the number of boot entries
dwCount = DynArrayGetCount(arrResults);
if(dwId<=0 || dwId > dwCount )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_OSID));
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
lstrcpy(szString ,DynArrayItemAsString(arrResults, dwId - 1 ));
if(lstrlen(szString) == 0)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
// check if the user entered the value of debug as on and do accordingly
if( lstrcmpi(szDebug,VALUE_ON)== 0)
{
//display error message if the user specifies Baud rate with out specifying COM port.
if((lstrlen(szPort)== 0) && (lstrlen(szBaudRate)!= 0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_SYNTAX_DEBUG));
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
//check if the debug switch is already present and if so display a error message.
if( (_tcsstr(szString,DEBUG_SWITCH) != NULL ) && ( (lstrlen(szPort)== 0)&&(lstrlen(szBaudRate)== 0) ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_DUPL_DEBUG));
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
else
{
if(_tcsstr(szString,DEBUG_SWITCH) == NULL )
{
lstrcat(szString,TOKEN_EMPTYSPACE);
lstrcat(szString,DEBUG_SWITCH);
}
}
// get the type of the Com port present
dwCode = GetSubString(szString,TOKEN_DEBUGPORT,szTemp);
if((lstrlen(szTemp )!= 0)&& (lstrlen(szPort)!= 0))
{
_stprintf(szBuffer,GetResString(IDS_ERROR_DUPLICATE_COM_PORT), dwId );
DISPLAY_MESSAGE(stderr,szBuffer);
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
// if the debug port is specified by the user.
if(lstrlen(szPort)!= 0)
{
dwSectionFlag = getKeysOfSpecifiedINISection(szPath ,BOOTLOADERSECTION,REDIRECT_STRING,szBoot);
if (dwSectionFlag == MALLOC_FAILURE)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
if(lstrlen(szBoot)!= 0)
{
if (lstrcmpi(szBoot,szPort)==0)
{
DISPLAY_MESSAGE( stderr, GetResString(IDS_ERROR_REDIRECT_PORT));
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
}
lstrcat(szTmpBuffer,TOKEN_EMPTYSPACE);
lstrcat(szTmpBuffer,TOKEN_DEBUGPORT) ;
lstrcat(szTmpBuffer,TOKEN_EQUAL) ;
CharLower(szPort);
lstrcat(szTmpBuffer,szPort);
lstrcat(szString,szTmpBuffer);
}
lstrcpy(szTemp,NULL_STRING);
//to add the Baud rate value specified by the user.
GetBaudRateVal(szString,szTemp) ;
if(lstrlen(szBaudRate)!=0)
{
if(lstrlen(szTemp )!= 0)
{
_stprintf(szBuffer,GetResString(IDS_ERROR_DUPLICATE_BAUD_VAL), dwId );
DISPLAY_MESSAGE(stderr,szBuffer);
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}else
{
//forming the string to be concatenated to the BootEntry string
lstrcpy(szTemp,BAUD_RATE);
lstrcat(szTemp,TOKEN_EQUAL);
lstrcat(szTemp,szBaudRate);
//append the string containing the modified port value to the string
lstrcat(szString,TOKEN_EMPTYSPACE);
lstrcat(szString,szTemp);
}
}
}
else if( lstrcmpi(szDebug,VALUE_OFF)== 0)
{
if((lstrlen(szPort)!= 0) || (lstrlen(szBaudRate)!= 0))
{
DestroyDynamicArray(&arrResults);
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_SYNTAX_DEBUG));
resetFileAttrib(szPath);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
if (_tcsstr(szString,DEBUG_SWITCH) == 0 )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_DEBUG_ABSENT));
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
else
{
// remove the /debug switch.
removeSubString(szString,DEBUG_SWITCH);
lstrcpy(szTemp,NULL_STRING);
// get the type of the Com port present
dwCode = GetSubString(szString,TOKEN_DEBUGPORT,szTemp);
if(lstrcmpi(szTemp,PORT_1394)==0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_1394_REMOVE));
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
// remove the /debugport=comport switch if it is present from the Boot Entry
if (lstrlen(szTemp )!= 0)
{
removeSubString(szString,szTemp);
}
lstrcpy(szTemp , NULL_STRING );
//remove the baud rate switch if it is present.
GetBaudRateVal(szString,szTemp) ;
if (lstrlen(szTemp )!= 0)
{
removeSubString(szString,szTemp);
}
}
}
// if the user enters the EDIT option
else if(lstrcmpi(szDebug,SWITCH_EDIT)== 0)
{
//display error message if the /debugport=1394 switch is present already.
if(_tcsstr(szString,DEBUGPORT_1394)!=0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_EDIT_1394_SWITCH));
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
if (_tcsstr(szString,DEBUG_SWITCH) == 0 )
{
_stprintf(szBuffer,GetResString(IDS_ERROR_NO_DBG_SWITCH), dwId );
DISPLAY_MESSAGE(stderr,szBuffer);
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
lstrcpy(szTemp,NULL_STRING);
dwCode = GetSubString(szString,TOKEN_DEBUGPORT,szTemp);
//display an error message if user is trying to add baudrate value
// when there is no COM port present in the boot options.
if( (lstrlen(szTemp)==0)&&(lstrlen(szPort)== 0)&&(lstrlen(szBaudRate)!= 0) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_COM_PORT));
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
else
{
// chk if the port has been spec by the user
if((lstrlen(szPort)== 0)&&(lstrlen(szBaudRate)== 0))
{
DestroyDynamicArray(&arrResults);
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_SYNTAX_DEBUG));
resetFileAttrib(szPath);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
if(lstrlen(szPort)!= 0)
{
lstrcpy(szTemp , NULL_STRING );
// get the type of the Com port present
lstrcpy(szTemp,NULL_STRING);
dwCode = GetSubString(szString,TOKEN_DEBUGPORT,szTemp);
//display error message if there is no COM port found at all in the OS option
//changed for displaying error
if(lstrlen(szTemp )== 0 )
{
_stprintf(szBuffer,GetResString(IDS_ERROR_NO_COM_PORT), dwId );
DISPLAY_MESSAGE(stderr,szBuffer);
bRes = resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
// remove the /debugport=comport switch if it is present from the Boot Entry
removeSubString(szString,szTemp);
lstrcpy(szTemp,TOKEN_DEBUGPORT) ;
lstrcat(szTemp,TOKEN_EQUAL);
CharUpper(szPort) ;
lstrcat(szTemp,szPort);
dwSectionFlag = getKeysOfSpecifiedINISection(szPath ,BOOTLOADERSECTION,REDIRECT_STRING,szBoot);
if (dwSectionFlag == MALLOC_FAILURE)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
if(lstrlen(szBoot)!= 0)
{
if (lstrcmpi(szBoot,szPort)==0)
{
DISPLAY_MESSAGE( stderr, GetResString(IDS_ERROR_REDIRECT_PORT));
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
}
//append the string containing the modified port value to the string
lstrcat(szString,TOKEN_EMPTYSPACE);
lstrcat(szString,szTemp);
}
//to edit the baud rate value
if(lstrlen(szBaudRate)!= 0)
{
lstrcpy(szTemp , NULL_STRING );
//remove the baud rate switch if it is present.
GetBaudRateVal(szString,szTemp) ;
// remove the swithc to be changed.
removeSubString(szString,szTemp);
//forming the string to be concatenated to the BootEntry string
lstrcpy(szTemp,BAUD_RATE);
lstrcat(szTemp,TOKEN_EQUAL);
lstrcat(szTemp,szBaudRate);
//append the string containing the modified port value to the string
lstrcat(szString,TOKEN_EMPTYSPACE);
lstrcat(szString,szTemp);
}
}
}
//display an error message if the Os Load Options string is more than
// 254 characters in length.
if( _tcslen(szString) >= MAX_RES_STRING)
{
_stprintf(szErrorMsg,GetResString(IDS_ERROR_STRING_LENGTH),MAX_RES_STRING);
DISPLAY_MESSAGE( stderr,szErrorMsg);
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
DynArrayRemove(arrResults, dwId - 1 );
DynArrayInsertString(arrResults, dwId - 1, szString, MAX_STRING_LENGTH1);
// Variable storing the path of boot.ini file
szFinalStr = (TCHAR*)malloc(MAX_STRING_LENGTH1* sizeof(TCHAR));
if (szFinalStr == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
ShowLastError(stderr);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Setting the buffer to 0, to avoid any junk value
memset(szFinalStr, 0, MAX_STRING_LENGTH1);
// Forming the final string from all the key-value pairs
if (stringFromDynamicArray1( arrResults,szFinalStr) == EXIT_FAILURE)
{
DestroyDynamicArray(&arrResults);
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
// Writing to the profile section with new key-value pair
// If the return value is non-zero, then there is an error.
if( WritePrivateProfileSection(OS_FIELD, szFinalStr, szPath ) != 0 )
{
_stprintf(szBuffer,GetResString(IDS_SWITCH_CHANGE), dwId );
DISPLAY_MESSAGE(stdout,szBuffer);
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_ADD_SWITCHES));
DestroyDynamicArray(&arrResults);
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
bRes = resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
return (bRes) ;
}
// ***************************************************************************
//
// Routine Description : Get the Type of Baud Rate present in Boot Entry
//
// Parameters : szString : The String which is to be searched.
// szTemp : String which will get the com port type
// Return Type : VOID
//
// ***************************************************************************
VOID GetBaudRateVal(LPTSTR szString, LPTSTR szTemp)
{
if(_tcsstr(szString,BAUD_VAL6)!=0)
{
lstrcpy(szTemp,BAUD_VAL6);
}
else if(_tcsstr(szString,BAUD_VAL7)!=0)
{
lstrcpy(szTemp,BAUD_VAL7);
}
else if(_tcsstr(szString,BAUD_VAL8)!=0)
{
lstrcpy(szTemp,BAUD_VAL8);
}
else if(_tcsstr(szString,BAUD_VAL9)!=0)
{
lstrcpy(szTemp,BAUD_VAL9);
}
else if(_tcsstr(szString,BAUD_VAL10)!=0)
{
lstrcpy(szTemp,BAUD_VAL10);
}
}
// ***************************************************************************
// Routine Description:
// Implement the ProcessEmsSwitch switch.
// Arguments:
// [IN] argc Number of command line arguments
// [IN] argv Array containing command line arguments
//
// Return Value:
// DWORD (EXIT_SUCCESS for success and EXIT_FAILURE for Failure.)
//
// ***************************************************************************
DWORD ProcessEmsSwitch( DWORD argc, LPCTSTR argv[] )
{
BOOL bUsage = FALSE ;
BOOL bNeedPwd = FALSE ;
BOOL bEms = FALSE ;
DWORD dwId = 0;
TARRAY arrResults ;
TARRAY arrBootIni ;
FILE *stream = NULL;
// Initialising the variables that are passed to TCMDPARSER structure
STRING256 szServer = NULL_STRING;
STRING256 szUser = NULL_STRING;
STRING256 szPassword = NULL_STRING;
STRING100 szPath = NULL_STRING;
TCHAR szPort[MAX_RES_STRING] = NULL_STRING ;
BOOL bRes = FALSE ;
BOOL bFlag = FALSE ;
DWORD dwCount = 0 ;
TCHAR szBuffer[MAX_RES_STRING] = NULL_STRING ;
TCHAR szDefault[MAX_RES_STRING] = NULL_STRING ;
TCHAR szString[MAX_STRING_LENGTH1] = NULL_STRING ;
TCHAR szBaudRate[MAX_RES_STRING] = NULL_STRING ;
TCHAR szBoot[MAX_RES_STRING] = NULL_STRING ;
LPTSTR szFinalStr = NULL ;
BOOL bRedirectFlag = FALSE ;
TCHAR szRedirectBaudrate[MAX_RES_STRING] = NULL_STRING ;
BOOL bRedirectBaudFlag = FALSE ;
TCHAR szErrorMsg[MAX_RES_STRING] = NULL_STRING ;
DWORD dwSectionFlag = FALSE ;
TCHAR szDebugPort[MAX_RES_STRING] = NULL_STRING ;
TCHAR szBootString[MAX_RES_STRING] = NULL_STRING ;
DWORD dwI = 0 ;
BOOL bMemFlag = FALSE ;
BOOL bDefault = FALSE ;
LPCTSTR szToken = NULL ;
DWORD dwRetVal = 0 ;
BOOL bConnFlag = FALSE ;
TCMDPARSER cmdOptions[] =
{
{ CMDOPTION_EMS, CP_MAIN_OPTION, 1, 0,&bDefault, NULL_STRING, NULL, NULL },
{ SWITCH_SERVER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szServer, NULL_STRING, NULL, NULL },
{ SWITCH_USER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szUser, NULL_STRING, NULL, NULL },
{ SWITCH_PASSWORD, CP_TYPE_TEXT | CP_VALUE_OPTIONAL, 1, 0, &szPassword, NULL_STRING, NULL, NULL },
{ CMDOPTION_USAGE, CP_USAGE, 1, 0, &bUsage, NULL_STRING, NULL, NULL },
{ SWITCH_ID, CP_TYPE_NUMERIC | CP_VALUE_MANDATORY , 1, 0, &dwId, NULL_STRING, NULL, NULL },
{ SWITCH_PORT, CP_TYPE_TEXT | CP_VALUE_MANDATORY|CP_MODE_VALUES,1,0,&szPort,EMS_PORT_VALUES,NULL,NULL},
{ SWITCH_BAUD, CP_TYPE_TEXT | CP_VALUE_MANDATORY |CP_MODE_VALUES,1,0,&szBaudRate,BAUD_RATE_VALUES_EMS,NULL,NULL},
{ CMDOPTION_DEFAULT, CP_DEFAULT | CP_TYPE_TEXT | CP_MANDATORY , 1, 0, &szDefault,NULL_STRING, NULL, NULL }
};
//
//check if the remote system is 64 bit and if so
// display an error.
//
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
//copy the Asterix token which is required for password prompting.
_tcscpy(cmdOptions[3].szValues,TOKEN_ASTERIX) ;
_tcscpy(szPassword,TOKEN_ASTERIX);
// Parsing the copy option switches
if ( ! DoParseParam( argc, argv, SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
ShowMessage(stderr,GetReason());
return (EXIT_FAILURE);
}
if(bUsage)
{
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_SUCCESS )
{
displayEmsUsage_X86() ;
return (EXIT_SUCCESS) ;
}else
{
return (EXIT_FAILURE);
}
}
//display error message if the username is entered with out a machine name
if( (cmdOptions[1].dwActuals == 0)&&(cmdOptions[2].dwActuals != 0))
{
SetReason(GetResString(IDS_USER_BUT_NOMACHINE));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
if( (cmdOptions[2].dwActuals == 0)&&(cmdOptions[3].dwActuals != 0))
{
SetReason(GetResString(IDS_PASSWD_BUT_NOUSER));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
//for setting the bNeedPwd
if( ( _tcscmp(szPassword,TOKEN_ASTERIX )==0 ) && (IsLocalSystem(szServer)==FALSE ))
{
bNeedPwd = TRUE ;
}
//set the bneedpassword to true if the server name is specified and password is not specified.
if((cmdOptions[1].dwActuals!=0)&&(cmdOptions[3].dwActuals==0))
{
if( (lstrlen( szServer ) != 0) && (IsLocalSystem(szServer)==FALSE) )
{
bNeedPwd = TRUE ;
}
else
{
bNeedPwd = FALSE ;
}
if(_tcslen(szPassword)!= 0 )
{
_tcscpy(szPassword,NULL_STRING);
}
}
//display an error message if the server is empty.
if( (cmdOptions[1].dwActuals!=0)&&(lstrlen(szServer)==0) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_SERVER));
return EXIT_FAILURE ;
}
//display an error message if the user is empty.
if((cmdOptions[2].dwActuals!=0)&&(lstrlen(szUser)==0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_USER));
return EXIT_FAILURE ;
}
//display error message if the user enters any invalid string.
if( !( ( lstrcmpi(szDefault,VALUE_ON)== 0) || (lstrcmpi(szDefault,VALUE_OFF)== 0 ) ||(lstrcmpi(szDefault,EDIT_STRING)== 0) ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_SYNTAX_EMS));
return EXIT_FAILURE;
}
if( (lstrcmpi(szDefault,EDIT_STRING)== 0)&& (lstrlen(szPort)==0) && (lstrlen(szBaudRate)==0) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_EDIT_SYNTAX));
return EXIT_FAILURE;
}
if( ( (lstrcmpi(szDefault,ON_STRING)== 0) || (lstrcmpi(szDefault,OFF_STRING)== 0) )&& (cmdOptions[5].dwActuals==0) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_ID_MISSING));
DISPLAY_MESSAGE(stderr,GetResString(IDS_EMS_HELP));
return EXIT_FAILURE;
}
// Establishing connection to the specified machine and getting the file pointer
// of the boot.ini file if there is no error while establishing connection
lstrcpy(szPath, PATH_BOOTINI );
if(StrCmpN(szServer,TOKEN_BACKSLASH4,2)==0)
{
if(!StrCmpN(szServer,TOKEN_BACKSLASH6,3)==0)
{
szToken = _tcstok(szServer,TOKEN_BACKSLASH4);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
return (EXIT_FAILURE);
}
lstrcpy(szServer,szToken);
}
}
if( (IsLocalSystem(szServer)==TRUE)&&(lstrlen(szUser)!=0))
{
DISPLAY_MESSAGE(stdout,GetResString(WARN_LOCALCREDENTIALS));
_tcscpy(szServer,_T(""));
}
bFlag = openConnection( szServer, szUser, szPassword, szPath,bNeedPwd,stream,&bConnFlag);
if(bFlag == EXIT_FAILURE)
{
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
arrResults = CreateDynamicArray();
//return failure if failed to allocate memory
if(arrResults == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return (EXIT_FAILURE);
}
arrResults = getKeyValueOfINISection( szPath, OS_FIELD );
if(arrResults != NULL)
{
lstrcpy(szString ,DynArrayItemAsString(arrResults, dwId - 1 ));
}
else
{
bMemFlag = TRUE ;
}
if((szString == NULL)||(bMemFlag == TRUE ))
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
//getting the number of boot entries
dwCount = DynArrayGetCount(arrResults);
if((dwId<=0 || dwId > dwCount ) && (lstrcmpi(szDefault,SWITCH_EDIT)!= 0) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_OSID));
DestroyDynamicArray(&arrResults);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
// common code till here . from here process acc to the ON/OFF/EDIT flag.
if(lstrcmpi(szDefault,ON_STRING)==0)
{
if((_tcsstr(szString,REDIRECT) != 0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_DUPL_REDIRECT_SWITCH));
DestroyDynamicArray(&arrResults);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
//Display an error message if there is no redirect port present in the
// bootloader section and the user also does not specify the COM port.
if ((lstrlen(szPort)== 0))
{
dwSectionFlag = getKeysOfSpecifiedINISection(szPath ,BOOTLOADERSECTION,REDIRECT_STRING,szBoot);
if (dwSectionFlag == MALLOC_FAILURE)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
DestroyDynamicArray(&arrResults);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
if(lstrlen(szBoot)== 0 )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NO_PORT));
DestroyDynamicArray(&arrResults);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
}
if(lstrlen(szPort)!= 0)
{
dwSectionFlag = getKeysOfSpecifiedINISection(szPath ,BOOTLOADERSECTION,BAUDRATE_STRING,szRedirectBaudrate);
if (dwSectionFlag == MALLOC_FAILURE)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
DestroyDynamicArray(&arrResults);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
dwSectionFlag = getKeysOfSpecifiedINISection(szPath ,BOOTLOADERSECTION,REDIRECT_STRING,szBoot);
if (dwSectionFlag == MALLOC_FAILURE)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
DestroyDynamicArray(&arrResults);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
//display warning message if the redirect=COMX entry is already present in the BootLoader section.
if(lstrlen(szBoot)!= 0 )
{
DISPLAY_MESSAGE(stdout,GetResString(IDS_WARN_REDIRECT));
bRedirectFlag = TRUE ;
}
if( (lstrlen(szRedirectBaudrate)!=0)&&(lstrlen(szBaudRate)!= 0 ))
{
DISPLAY_MESSAGE(stdout,GetResString(IDS_WARN_REDIRECTBAUD));
bRedirectBaudFlag = TRUE ;
}
// if the Boot loader section does not
// contain any port for redirection.
if(!bRedirectFlag)
{
if (lstrcmpi(szPort,USEBIOSSET)== 0)
{
lstrcpy(szPort,USEBIOSSETTINGS);
}
// scan the entire BOOT.INI and check if the specified Port
lstrcpy(szDebugPort,DEBUGPORT);
lstrcat(szDebugPort,szPort);
arrBootIni = getKeyValueOfINISection( szPath, OS_FIELD );
if(arrBootIni == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
//
//loop through all the OS entries and check if any of the
//
for(dwI = 0 ;dwI < dwCount-1 ; dwI++ )
{
lstrcpy(szBootString ,DynArrayItemAsString(arrBootIni,dwI));
CharLower(szDebugPort);
if(_tcsstr(szBootString,szDebugPort)!= 0)
{
DISPLAY_MESSAGE( stderr, GetResString(IDS_ERROR_DEBUG_PORT));
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootIni);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
}
//convert the com port value specified by user to upper case for storing into the ini file.
CharUpper(szPort);
if( WritePrivateProfileString( BOOTLOADERSECTION,KEY_REDIRECT,szPort, szPath ) != 0 )
{
DISPLAY_MESSAGE(stdout,GetResString(IDS_EMS_CHANGE_BOOTLOADER));
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_EMS_CHANGE_ERROR_BLOADER));
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootIni);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
}
}
if(!bRedirectBaudFlag)
{
// to add the baudrate to the BOOTLOADER section.
if(lstrlen(szBaudRate) != 0 )
{
if( WritePrivateProfileString( BOOTLOADERSECTION,KEY_BAUDRATE,szBaudRate, szPath ) != 0 )
{
DISPLAY_MESSAGE(stdout,GetResString(IDS_EMS_CHANGE_BAUDRATE));
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_EMS_CHANGE_ERROR_BAUDRATE));
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootIni);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
}
}
//add the /redirect into the OS options.
lstrcat(szString,TOKEN_EMPTYSPACE);
lstrcat(szString,REDIRECT);
//display an error message if the Os Load Options string is more than
// 254 characters in length.
if( _tcslen(szString) >= MAX_RES_STRING)
{
_stprintf(szErrorMsg,GetResString(IDS_ERROR_STRING_LENGTH),MAX_RES_STRING);
DISPLAY_MESSAGE( stderr,szErrorMsg);
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootIni);
SAFECLOSE(stream);
return (EXIT_FAILURE);
}
DynArrayRemove(arrResults, dwId - 1 );
DynArrayInsertString(arrResults, dwId - 1, szString, MAX_STRING_LENGTH1);
szFinalStr = (TCHAR*)malloc(MAX_STRING_LENGTH1* sizeof(TCHAR));
if (szFinalStr == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootIni);
SAFECLOSE(stream);
return (EXIT_FAILURE);
}
// Setting the buffer to 0, to avoid any junk value
memset(szFinalStr, 0, MAX_STRING_LENGTH1);
// Forming the final string from all the key-value pairs
if (stringFromDynamicArray1( arrResults,szFinalStr) == EXIT_FAILURE)
{
DestroyDynamicArray(&arrResults);
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootIni);
SAFECLOSE(stream);
return EXIT_FAILURE;
}
// Writing to the profile section with new key-value pair
// If the return value is non-zero, then there is an error.
if( WritePrivateProfileSection(OS_FIELD, szFinalStr, szPath ) != 0 )
{
_stprintf(szBuffer,GetResString(IDS_SWITCH_CHANGE), dwId );
DISPLAY_MESSAGE(stdout,szBuffer);
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_ADD_SWITCHES));
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootIni);
SAFECLOSE(stream);
return (EXIT_FAILURE);
}
}
if(lstrcmpi(szDefault,EDIT_STRING)==0)
{
//display error message if user enters a id for the edit option.
if(dwId!=0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_SYNTAX_EMS));
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
return EXIT_FAILURE ;
}
if (lstrcmpi(szPort,USEBIOSSET)== 0)
{
lstrcpy(szPort,USEBIOSSETTINGS);
}
//get the keys of the specified ini section.
dwSectionFlag = getKeysOfSpecifiedINISection(szPath ,BOOTLOADERSECTION,BAUDRATE_STRING,szRedirectBaudrate);
if (dwSectionFlag == MALLOC_FAILURE)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
DestroyDynamicArray(&arrResults);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
//get the keys of the specified ini section.
dwSectionFlag = getKeysOfSpecifiedINISection(szPath ,BOOTLOADERSECTION,REDIRECT_STRING,szBoot);
if (dwSectionFlag == MALLOC_FAILURE)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
DestroyDynamicArray(&arrResults);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
if( (lstrlen(szBoot) == 0 ) && ((cmdOptions[6].dwActuals!=0)) )
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_COM_PORT));
DestroyDynamicArray(&arrResults);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
if( (lstrlen(szRedirectBaudrate) == 0 ) && ((cmdOptions[7].dwActuals!=0)) )
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_ERROR_BAUDRATE_HELP));
DestroyDynamicArray(&arrResults);
SAFECLOSE(stream);
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
lstrcpy(szDebugPort,DEBUGPORT);
lstrcat(szDebugPort,szPort);
arrBootIni = getKeyValueOfINISection( szPath, OS_FIELD );
if(arrBootIni == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
//
//loop through all the OS entries and check if any of the
//
for(dwI = 0 ;dwI < dwCount-1 ; dwI++ )
{
lstrcpy(szBootString ,DynArrayItemAsString(arrBootIni,dwI));
CharLower(szDebugPort);
if(_tcsstr(szBootString,szDebugPort)!= 0)
{
DISPLAY_MESSAGE( stderr, GetResString(IDS_ERROR_DEBUG_PORT));
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootIni);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
}
// edit the Boot loader section with the redirect values entered by the user.
CharUpper(szPort);
if(lstrlen(szPort)!= 0)
{
if( WritePrivateProfileString( BOOTLOADERSECTION,KEY_REDIRECT,
szPort, szPath ) != 0 )
{
DISPLAY_MESSAGE(stdout,GetResString(IDS_EMS_CHANGE_BOOTLOADER));
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_EMS_CHANGE_ERROR_BLOADER));
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootIni);
SafeCloseConnection(szServer,bConnFlag);
SAFECLOSE(stream);
return EXIT_FAILURE ;
}
}
// edit the Boot loader section with the baudrate values entered by the user.
if(lstrlen(szBaudRate)!= 0)
{
if( WritePrivateProfileString( BOOTLOADERSECTION,KEY_BAUDRATE,
szBaudRate, szPath ) != 0 )
{
DISPLAY_MESSAGE(stdout,GetResString(IDS_EMS_CHANGE_BAUDRATE));
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_EMS_CHANGE_ERROR_BAUDRATE));
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
DestroyDynamicArray(&arrResults);
DestroyDynamicArray(&arrBootIni);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
}
}
// if the option value is off.
if(lstrcmpi(szDefault,VALUE_OFF)==0)
{
//display an error message if either the com port or baud rate is typed in the command line
if((lstrlen(szBaudRate)!=0)||(lstrlen(szPort)!=0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_SYNTAX_EMS));
DestroyDynamicArray(&arrResults);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
// display error message if the /redirect switch is not present in the Boot.ini
if((_tcsstr(szString,REDIRECT) == 0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_REDIRECT_SWITCH));
DestroyDynamicArray(&arrResults);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
//remove the /redirect switch from the OS entry specified .
removeSubString(szString,REDIRECT);
//display an error message if the Os Load options string is more than
// 255 characters in length.
if( _tcslen(szString) >= MAX_RES_STRING)
{
_stprintf(szErrorMsg,GetResString(IDS_ERROR_STRING_LENGTH),MAX_RES_STRING);
DISPLAY_MESSAGE( stderr,szErrorMsg);
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
DynArrayRemove(arrResults, dwId - 1 );
DynArrayInsertString(arrResults, dwId - 1, szString, MAX_STRING_LENGTH1);
szFinalStr = (TCHAR*)malloc(MAX_STRING_LENGTH1* sizeof(TCHAR));
if (szFinalStr == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Setting the buffer to 0, to avoid any junk value
memset(szFinalStr, 0, MAX_STRING_LENGTH1);
// Forming the final string from all the key-value pairs
if (stringFromDynamicArray1( arrResults,szFinalStr) == EXIT_FAILURE)
{
DestroyDynamicArray(&arrResults);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
// Writing to the profile section with new key-value pair
// If the return value is non-zero, then there is an error.
if( WritePrivateProfileSection(OS_FIELD, szFinalStr, szPath ) != 0 )
{
_stprintf(szBuffer,GetResString(IDS_SWITCH_CHANGE), dwId );
DISPLAY_MESSAGE(stdout,szBuffer);
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_ADD_SWITCHES));
DestroyDynamicArray(&arrResults);
SAFEFREE(szFinalStr);
resetFileAttrib(szPath);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
}
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arrResults);
SafeCloseConnection(szServer,bConnFlag);
return (bRes) ;
}
// ***************************************************************************
// Routine Description : Display the help for the Ems entry option (X86).
//
// Parameters : none
//
// Return Type : VOID
//
// ***************************************************************************
VOID displayEmsUsage_X86()
{
DWORD dwIndex = IDS_EMS_BEGIN_X86 ;
for(;dwIndex <=IDS_EMS_END_X86;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description : Display the help for the Debug entry option (X86).
//
// Parameters : none
//
// Return Type : VOID
// ***************************************************************************
VOID displayDebugUsage_X86()
{
DWORD dwIndex = IDS_DEBUG_BEGIN_X86 ;
for(;dwIndex <=IDS_DEBUG_END_X86;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description : Display the help for the Ems entry option (IA64).
//
// Parameters : none
//
// Return Type : VOID
//
// ***************************************************************************
VOID displayEmsUsage_IA64()
{
DWORD dwIndex = IDS_EMS_BEGIN_IA64 ;
for(;dwIndex <=IDS_EMS_END_IA64;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description : Display the help for the Debug entry option (IA64).
//
// Parameters : none
//
// Return Type : VOID
// ***************************************************************************
VOID displayDebugUsage_IA64()
{
DWORD dwIndex = IDS_DEBUG_BEGIN_IA64 ;
for(;dwIndex <= IDS_DEBUG_END_IA64 ;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description : This function gets all the keys present in the specified section of
// an .ini file and then returns the dynamic array containing all the
// keys
//
// Parameters : LPTSTR sziniFile (in) - Name of the ini file.
// LPTSTR szinisection (in) - Name of the section in the boot.ini.
//
// Return Type : TARRAY ( pointer to the dynamic array )
//
// ***************************************************************************
DWORD getKeysOfSpecifiedINISection( LPTSTR sziniFile, LPTSTR sziniSection,LPCWSTR szKeyName ,LPTSTR szValue)
{
// Number of characters returned by the GetPrivateProfileString function
DWORD len = 0;
DWORD dwLength = MAX_STRING_LENGTH1 ;
// Buffer which will be populated by the GetPrivateProfileString function
LPTSTR inBuf = NULL ;
inBuf = (LPTSTR)malloc(dwLength*sizeof(TCHAR));
if(inBuf == NULL)
{
return MALLOC_FAILURE ;
}
while(1)
{
// Getting all the keys from the boot.ini file
len = GetPrivateProfileString
(sziniSection,
szKeyName,
ERROR_PROFILE_STRING1,
inBuf,
dwLength,
sziniFile);
//if the size of the string is not sufficient then increment the size.
if(len == dwLength-2)
{
dwLength +=100 ;
inBuf = (LPTSTR)realloc(inBuf,dwLength*sizeof(TCHAR));
if(inBuf == NULL)
{
SAFEFREE(inBuf);
return MALLOC_FAILURE;
}
}
else
break ;
}
//copy the value into the destination buffer only if
// the size is less than 255 else return FAILURE.
//
if(lstrlen(inBuf) <= MAX_RES_STRING)
{
lstrcpy(szValue,inBuf);
}
else
{
SAFEFREE(inBuf);
return MALLOC_FAILURE;
}
SAFEFREE(inBuf);
return EXIT_SUCCESS ;
}
// ***************************************************************************
// Routine Description:
// Implement the Add Switch switch.
// Arguments:
// [IN] argc Number of command line arguments
// [IN] argv Array containing command line arguments
//
// Return Value:
// DWORD (EXIT_SUCCESS for success and EXIT_FAILURE for Failure.)
//
// ***************************************************************************
DWORD ProcessAddSwSwitch( DWORD argc, LPCTSTR argv[] )
{
BOOL bUsage = FALSE ;
BOOL bNeedPwd = FALSE ;
BOOL bAddSw = FALSE ;
DWORD dwDefault = 0;
TARRAY arr ;
TCHAR szkey[MAX_STRING_LENGTH1] = NULL_STRING;
FILE *stream = NULL;
// Initialising the variables that are passed to TCMDPARSER structure
STRING256 szServer = NULL_STRING;
STRING256 szUser = NULL_STRING;
STRING256 szPassword = NULL_STRING;
STRING100 szPath = NULL_STRING;
DWORD dwNumKeys = 0;
BOOL bRes = FALSE ;
LPTSTR szFinalStr = NULL ;
BOOL bFlag = FALSE ;
TCHAR szMaxmem[10] = NULL_STRING ;
TCHAR szBuffer[MAX_RES_STRING] = NULL_STRING ;
TCHAR szErrorMsg[MAX_RES_STRING] = NULL_STRING ;
BOOL bBaseVideo = FALSE ;
BOOL bSos = FALSE ;
BOOL bNoGui = FALSE ;
DWORD dwMaxmem = 0 ;
BOOL bErrorFlag = FALSE ;
LPCTSTR szToken = NULL ;
DWORD dwRetVal = 0 ;
BOOL bConnFlag = FALSE ;
TCMDPARSER cmdOptions[] =
{
{ CMDOPTION_ADDSW, CP_MAIN_OPTION, 1, 0,&bAddSw, NULL_STRING, NULL, NULL },
{ SWITCH_SERVER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szServer, NULL_STRING, NULL, NULL },
{ SWITCH_USER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szUser, NULL_STRING, NULL, NULL },
{ SWITCH_PASSWORD, CP_TYPE_TEXT | CP_VALUE_OPTIONAL, 1, 0, &szPassword, NULL_STRING, NULL, NULL },
{ CMDOPTION_USAGE, CP_USAGE, 1, 0, &bUsage, NULL_STRING, NULL, NULL },
{ SWITCH_ID, CP_TYPE_NUMERIC | CP_VALUE_MANDATORY|CP_MANDATORY , 1, 0, &dwDefault, NULL_STRING, NULL, NULL },
{ SWITCH_MAXMEM, CP_TYPE_UNUMERIC | CP_VALUE_MANDATORY,1,0,&dwMaxmem,NULL_STRING,NULL,NULL},
{ SWITCH_BASEVIDEO, 0,1,0,&bBaseVideo,NULL_STRING,NULL,NULL},
{ SWITCH_NOGUIBOOT, 0,1,0,&bNoGui,NULL_STRING,NULL,NULL},
{ SWITCH_SOS, 0,1,0,&bSos,NULL_STRING,NULL,NULL},
};
//
//check if the remote system is 64 bit and if so
// display an error.
//
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
//copy the Asterix token which is required for password prompting.
_tcscpy(cmdOptions[3].szValues,TOKEN_ASTERIX) ;
_tcscpy(szPassword,TOKEN_ASTERIX);
// Parsing the copy option switches
if ( ! DoParseParam( argc, argv, SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
ShowMessage(stderr,GetReason());
return (EXIT_FAILURE);
}
// Displaying query usage if user specified -? with -query option
if( bUsage )
{
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_SUCCESS )
{
displayAddSwUsage_X86();
return (EXIT_SUCCESS);
}else
{
return (EXIT_FAILURE);
}
}
if( (cmdOptions[6].dwActuals!=0) && (dwMaxmem < 32 ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_MAXMEM_VALUES));
return EXIT_FAILURE ;
}
//display an error message if the user does not enter even one of
if((dwMaxmem==0)&& (!bBaseVideo)&& (!bNoGui)&&(!bSos) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_SYNTAX_ADDSW));
return EXIT_FAILURE ;
}
//display error message if the username is entered with out a machine name
if( (cmdOptions[1].dwActuals == 0)&&(cmdOptions[2].dwActuals != 0))
{
SetReason(GetResString(IDS_USER_BUT_NOMACHINE));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
if( (cmdOptions[2].dwActuals == 0)&&(cmdOptions[3].dwActuals != 0))
{
SetReason(GetResString(IDS_PASSWD_BUT_NOUSER));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
//for setting the bNeedPwd
if( ( _tcscmp(szPassword,TOKEN_ASTERIX )==0 ) && (IsLocalSystem(szServer)==FALSE ))
{
bNeedPwd = TRUE ;
}
//set the bneedpassword to true if the server name is specified and password is not specified.
if((cmdOptions[1].dwActuals!=0)&&(cmdOptions[3].dwActuals==0))
{
if( (lstrlen( szServer ) != 0) && (IsLocalSystem(szServer)==FALSE) )
{
bNeedPwd = TRUE ;
}
else
{
bNeedPwd = FALSE ;
}
if(_tcslen(szPassword)!= 0 )
{
_tcscpy(szPassword,NULL_STRING);
}
}
//display an error message if the server is empty.
if((cmdOptions[1].dwActuals!=0)&&(lstrlen(szServer)==0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_SERVER));
return EXIT_FAILURE ;
}
//display an error message if the user is empty.
if((cmdOptions[2].dwActuals!=0)&&(lstrlen(szUser)==0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_USER));
return EXIT_FAILURE ;
}
szFinalStr = (TCHAR*) malloc(MAX_STRING_LENGTH1* sizeof(TCHAR) );
if (szFinalStr== NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_TAG));
ShowLastError(stderr);
return (EXIT_FAILURE);
}
// Establishing connection to the specified machine and getting the file pointer
// of the boot.ini file if there is no error while establishing connection
lstrcpy(szPath, PATH_BOOTINI );
if(StrCmpN(szServer,TOKEN_BACKSLASH4,2)==0)
{
if(!StrCmpN(szServer,TOKEN_BACKSLASH6,3)==0)
{
szToken = _tcstok(szServer,TOKEN_BACKSLASH4);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
SAFEFREE(szFinalStr);
return (EXIT_FAILURE);
}
lstrcpy(szServer,szToken);
}
}
if( (IsLocalSystem(szServer)==TRUE)&&(lstrlen(szUser)!=0))
{
DISPLAY_MESSAGE(stdout,GetResString(WARN_LOCALCREDENTIALS));
_tcscpy(szServer,_T(""));
}
bFlag = openConnection( szServer, szUser, szPassword, szPath,bNeedPwd,stream,&bConnFlag);
if(bFlag == EXIT_FAILURE)
{
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting the keys of the Operating system section in the boot.ini file
arr = getKeyValueOfINISection( szPath, OS_FIELD );
if(arr == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
// Getting the total number of keys in the operating systems section
dwNumKeys = DynArrayGetCount(arr);
if((dwNumKeys >= MAX_BOOTID_VAL)&& (dwDefault >= MAX_BOOTID_VAL ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_MAX_BOOTID));
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
DestroyDynamicArray(&arr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Displaying error message if the number of keys is less than the OS entry
// line number specified by the user
if( ( dwDefault <= 0 ) || ( dwDefault > dwNumKeys ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_BOOTID));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting the key of the OS entry specified by the user
if (arr != NULL)
{
LPCWSTR pwsz = NULL;
pwsz = DynArrayItemAsString( arr, dwDefault - 1 ) ;
if(pwsz != NULL)
{
_tcscpy( szkey,pwsz);
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
//if the max mem switch is specified by the user.
if(dwMaxmem != 0)
{
if(_tcsstr(szkey,MAXMEM_VALUE1) != 0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_DUPL_MAXMEM_SWITCH));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
else
{
lstrcat(szkey , TOKEN_EMPTYSPACE);
lstrcat(szkey ,MAXMEM_VALUE1);
lstrcat(szkey,TOKEN_EQUAL);
_ltow(dwMaxmem,szMaxmem,10);
lstrcat(szkey,szMaxmem);
}
}
// if the base video is specified by the user.
if (bBaseVideo)
{
if(_tcsstr(szkey,BASEVIDEO_VALUE) != 0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_DUPL_BASEVIDEO_SWITCH));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
else
{
lstrcat(szkey , TOKEN_EMPTYSPACE);
lstrcat(szkey ,BASEVIDEO_SWITCH);
}
}
// if the SOS is specified by the user.
if(bSos)
{
if(_tcsstr(szkey,SOS_VALUE) != 0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_DUPL_SOS_SWITCH ) );
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
else
{
lstrcat(szkey , TOKEN_EMPTYSPACE);
lstrcat(szkey ,SOS_SWITCH);
}
}
// if the noguiboot is specified by the user.
if(bNoGui)
{
if(_tcsstr(szkey,NOGUI_VALUE) != 0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_DUPL_NOGUI_SWITCH));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
else
{
lstrcat(szkey , TOKEN_EMPTYSPACE);
lstrcat(szkey ,NOGUI_VALUE );
}
}
if( _tcslen(szkey) >= MAX_RES_STRING)
{
_stprintf(szErrorMsg,GetResString(IDS_ERROR_STRING_LENGTH),MAX_RES_STRING);
DISPLAY_MESSAGE( stderr,szErrorMsg);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
DynArrayRemove(arr, dwDefault - 1 );
DynArrayInsertString(arr, dwDefault - 1, szkey, MAX_STRING_LENGTH1);
// Setting the buffer to 0, to avoid any junk value
memset(szFinalStr, 0, MAX_STRING_LENGTH1);
// Forming the final string from all the key-value pairs
if (stringFromDynamicArray1( arr,szFinalStr) == EXIT_FAILURE)
{
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
// Writing to the profile section with new key-value pair
// If the return value is non-zero, then there is an error.
if( WritePrivateProfileSection(OS_FIELD, szFinalStr, szPath ) != 0 )
{
_stprintf(szBuffer,GetResString(IDS_SWITCH_ADD), dwDefault );
DISPLAY_MESSAGE(stdout,szBuffer);
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_ADD_SWITCHES));
DestroyDynamicArray(&arr);
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
//reset the file attributes and free the memory and close the connection to the server.
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (bRes);
}
// ***************************************************************************
// Routine Description:
// This routine is to remove the switches to the boot.ini file settings for
// the specified system.
// Arguments:
// [IN] argc Number of command line arguments
// [IN] argv Array containing command line arguments
//
// Return Value:
// DWORD (EXIT_SUCCESS for success and EXIT_FAILURE for Failure.)
//
// ***************************************************************************
DWORD ProcessRmSwSwitch( DWORD argc, LPCTSTR argv[] )
{
BOOL bUsage = FALSE ;
BOOL bNeedPwd = FALSE ;
BOOL bRmSw = FALSE ;
DWORD dwDefault = 0;
TARRAY arr = NULL ;
TCHAR szkey[MAX_STRING_LENGTH1] = NULL_STRING;
FILE *stream = NULL;
// Initialising the variables that are passed to TCMDPARSER structure
STRING256 szServer = NULL_STRING;
STRING256 szUser = NULL_STRING;
STRING256 szPassword = NULL_STRING;
STRING100 szPath = NULL_STRING;
TCHAR szTmp[MAX_RES_STRING] = NULL_STRING ;
TCHAR szChangeKeyValue[MAX_RES_STRING] = NULL_STRING ;
DWORD dwNumKeys = 0;
BOOL bRes = FALSE ;
PTCHAR pToken = NULL ;
LPTSTR szFinalStr = NULL ;
BOOL bFlag = FALSE ;
TCHAR szMaxmem[STRING10] = NULL_STRING ;
TCHAR szBuffer[MAX_RES_STRING] = NULL_STRING ;
TCHAR szDefault[MAX_RES_STRING] = NULL_STRING ;
BOOL bBaseVideo = FALSE ;
BOOL bSos = FALSE ;
BOOL bNoGui = FALSE ;
DWORD bMaxmem = FALSE ;
BOOL bErrorFlag = FALSE ;
TCHAR szTemp[MAX_RES_STRING] = NULL_STRING ;
TCHAR szString[MAX_RES_STRING] = NULL_STRING ;
LPTSTR szSubString = NULL ;
DWORD dwCode = 0;
LPCTSTR szToken = NULL ;
DWORD dwRetVal = 0;
BOOL bConnFlag = FALSE ;
TCMDPARSER cmdOptions[] =
{
{ CMDOPTION_RMSW, CP_MAIN_OPTION, 1, 0,&bRmSw, NULL_STRING, NULL, NULL },
{ SWITCH_SERVER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szServer, NULL_STRING, NULL, NULL },
{ SWITCH_USER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szUser, NULL_STRING, NULL, NULL },
{ SWITCH_PASSWORD, CP_TYPE_TEXT | CP_VALUE_OPTIONAL, 1, 0, &szPassword, NULL_STRING, NULL, NULL },
{ CMDOPTION_USAGE, CP_USAGE, 1, 0, &bUsage, NULL_STRING, NULL, NULL },
{ SWITCH_ID, CP_TYPE_NUMERIC | CP_VALUE_MANDATORY|CP_MANDATORY , 1, 0, &dwDefault, NULL_STRING, NULL, NULL },
{ SWITCH_MAXMEM, 0,1,0,&bMaxmem,NULL_STRING,NULL,NULL},
{ SWITCH_BASEVIDEO, 0,1,0,&bBaseVideo,NULL_STRING,NULL,NULL},
{ SWITCH_NOGUIBOOT, 0,1,0,&bNoGui,NULL_STRING,NULL,NULL},
{ SWITCH_SOS, 0,1,0,&bSos,NULL_STRING,NULL,NULL},
};
//
//check if the remote system is 64 bit and if so
// display an error.
//
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
//copy the Asterix token which is required for password prompting.
_tcscpy(cmdOptions[3].szValues,TOKEN_ASTERIX) ;
_tcscpy(szPassword,TOKEN_ASTERIX);
// Parsing the copy option switches
if ( ! DoParseParam( argc, argv, SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
ShowMessage(stderr,GetReason());
return (EXIT_FAILURE);
}
// Displaying query usage if user specified -? with -query option
if( bUsage )
{
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_SUCCESS )
{
displayRmSwUsage_X86();
return (EXIT_SUCCESS);
}else
{
return (EXIT_FAILURE);
}
}
//display error message if the username is entered with out a machine name
if( (cmdOptions[1].dwActuals == 0)&&(cmdOptions[2].dwActuals != 0))
{
SetReason(GetResString(IDS_USER_BUT_NOMACHINE));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
if( (cmdOptions[2].dwActuals == 0)&&(cmdOptions[3].dwActuals != 0))
{
SetReason(GetResString(IDS_PASSWD_BUT_NOUSER));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
//for setting the bNeedPwd
if( ( _tcscmp(szPassword,TOKEN_ASTERIX )==0 ) && (IsLocalSystem(szServer)==FALSE ))
{
bNeedPwd = TRUE ;
}
//set the bneedpassword to true if the server name is specified and password is not specified.
if((cmdOptions[1].dwActuals!=0)&&(cmdOptions[3].dwActuals==0))
{
if( (lstrlen( szServer ) != 0) && (IsLocalSystem(szServer)==FALSE) )
{
bNeedPwd = TRUE ;
}
else
{
bNeedPwd = FALSE ;
}
if(_tcslen(szPassword)!= 0 )
{
_tcscpy(szPassword,NULL_STRING);
}
}
//display an error message if the server is empty.
if((cmdOptions[1].dwActuals!=0)&&(lstrlen(szServer)==0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_SERVER));
return EXIT_FAILURE ;
}
//display an error message if the user is empty.
if((cmdOptions[2].dwActuals!=0)&&(lstrlen(szUser)==0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_USER));
return EXIT_FAILURE ;
}
//display an error mesage if none of the options are specified.
if((!bSos)&&(!bBaseVideo)&&(!bNoGui)&&(!bMaxmem))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_SYNTAX_RMSW));
return EXIT_FAILURE ;
}
//display a warning message if the user specifies local system name
// with -s.
szFinalStr = (TCHAR*) malloc(MAX_STRING_LENGTH1* sizeof(TCHAR) );
if (szFinalStr== NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_TAG));
ShowLastError(stderr);
return (EXIT_FAILURE);
}
// Establishing connection to the specified machine and getting the file pointer
// of the boot.ini file if there is no error while establishing connection
lstrcpy(szPath, PATH_BOOTINI );
if(StrCmpN(szServer,TOKEN_BACKSLASH4,2)==0)
{
if(!StrCmpN(szServer,TOKEN_BACKSLASH6,3)==0)
{
szToken = _tcstok(szServer,TOKEN_BACKSLASH4);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
SAFEFREE(szFinalStr);
return (EXIT_FAILURE);
}
lstrcpy(szServer,szToken);
}
}
if( (IsLocalSystem(szServer)==TRUE)&&(lstrlen(szUser)!=0))
{
DISPLAY_MESSAGE(stdout,GetResString(WARN_LOCALCREDENTIALS));
_tcscpy(szServer,_T(""));
}
bFlag = openConnection( szServer, szUser, szPassword, szPath,bNeedPwd,stream,&bConnFlag);
if(bFlag == EXIT_FAILURE)
{
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting the keys of the Operating system section in the boot.ini file
arr = getKeyValueOfINISection( szPath, OS_FIELD );
if(arr == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
// Getting the total number of keys in the operating systems section
dwNumKeys = DynArrayGetCount(arr);
if( (dwNumKeys >= MAX_BOOTID_VAL)&&(dwDefault >= MAX_BOOTID_VAL ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_MAX_BOOTID));
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
DestroyDynamicArray(&arr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Displaying error message if the number of keys is less than the OS entry
// line number specified by the user
if( ( dwDefault <= 0 ) || ( dwDefault > dwNumKeys ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_BOOTID));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting the key of the OS entry specified by the user
if (arr != NULL)
{
LPCWSTR pwsz = NULL;
pwsz = DynArrayItemAsString( arr, dwDefault - 1 ) ;
if(pwsz != NULL)
{
_tcscpy( szkey,pwsz);
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
//if the max mem switch is specified by the user.
if(bMaxmem==TRUE)
{
if(_tcsstr(szkey,MAXMEM_VALUE1) == 0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_MAXMEM_SWITCH));
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
else
{
szSubString = ( LPTSTR ) malloc( MAX_RES_STRING*sizeof( TCHAR ) );
if(szSubString == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_TAG));
ShowLastError(stderr);
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
lstrcpy(szTemp,NULL_STRING);
dwCode = GetSubString(szkey,MAXMEM_VALUE1,szSubString);
//remove the substring specified.
if(dwCode == EXIT_SUCCESS)
{
removeSubString(szkey,szSubString);
}
}
}
// if the base video is specified by the user.
if (bBaseVideo==TRUE)
{
if(_tcsstr(szkey,BASEVIDEO_VALUE) == 0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_BV_SWITCH));
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
else
{
removeSubString(szkey,BASEVIDEO_VALUE);
}
}
// if the SOS is specified by the user.
if(bSos==TRUE)
{
if(_tcsstr(szkey,SOS_VALUE) == 0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_SOS_SWITCH ) );
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
else
{
removeSubString(szkey,SOS_VALUE);
}
}
// if the noguiboot is specified by the user.
if(bNoGui==TRUE)
{
if(_tcsstr(szkey,NOGUI_VALUE) == 0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_NOGUI_SWITCH));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
else
{
removeSubString(szkey,NOGUI_VALUE);
}
}
DynArrayRemove(arr, dwDefault - 1 );
DynArrayInsertString(arr, dwDefault - 1, szkey, MAX_STRING_LENGTH1);
// Setting the buffer to 0, to avoid any junk value
memset(szFinalStr, 0, MAX_STRING_LENGTH1);
// Forming the final string from all the key-value pairs
if (stringFromDynamicArray1( arr,szFinalStr) == EXIT_FAILURE)
{
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
// Writing to the profile section with new key-value pair
// If the return value is non-zero, then there is an error.
if( WritePrivateProfileSection(OS_FIELD, szFinalStr, szPath ) != 0 )
{
_stprintf(szBuffer,GetResString(IDS_SWITCH_DELETE), dwDefault );
DISPLAY_MESSAGE(stdout,szBuffer);
}
else
{
_stprintf(szBuffer,GetResString(IDS_NO_SWITCH_DELETE), dwDefault );
DISPLAY_MESSAGE(stderr,szBuffer);
DestroyDynamicArray(&arr);
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
//reset the file attributes and free the memory and close the connection to the server.
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_SUCCESS);
}
// ***************************************************************************
// Routine Description : Display the help for the AddSw entry option (X86).
//
// Parameters : none
//
// Return Type : VOID
//
// ***************************************************************************
VOID displayAddSwUsage_X86()
{
DWORD dwIndex = IDS_ADDSW_BEGIN_X86 ;
for(;dwIndex <=IDS_ADDSW_END_X86;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
// Routine Description : Display the help for the AddSw entry option (IA64).
//
// Arguments : none
//
// Return Type : VOID
//
// ***************************************************************************
VOID displayAddSwUsage_IA64()
{
DWORD dwIndex = IDS_ADDSW_BEGIN_IA64 ;
for(;dwIndex <=IDS_ADDSW_END_IA64;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
//
// Routine Description : Display the help for the RmSw entry option (IA64).
//
// Arguments : none
//
// Return Type : VOID
//
// ***************************************************************************
VOID displayRmSwUsage_IA64()
{
DWORD dwIndex = IDS_RMSW_BEGIN_IA64 ;
for(;dwIndex <=IDS_RMSW_END_IA64;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
//
// Routine Description : Display the help for the RmSw entry option (X86).
//
// Arguments : none
//
// Return Type : VOID
//
// ***************************************************************************
VOID displayRmSwUsage_X86()
{
DWORD dwIndex = IDS_RMSW_BEGIN_X86 ;
for(;dwIndex <=IDS_RMSW_END_X86;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
//
// Routine Description : This function retreives a part of the string.
//
// Parameters :
// LPTSTR szString (in) - String in which substring is to be found.
// LPTSTR szPartString (in) - Part String whose remaining substring is to be found.
// LPTSTR pszFullString (out) - String in which substring is to be found.
//
//
// Return Type : DWORD
//
//
// ***************************************************************************
DWORD GetSubString(LPTSTR szString,LPTSTR szPartString,LPTSTR pszFullString)
{
TCHAR szTemp[MAX_RES_STRING]= NULL_STRING ;
PTCHAR pszMemValue = NULL ;
PTCHAR pszdest = NULL ;
#ifndef _WIN64
DWORD dwPos = 0;
#else
INT64 dwPos = 0;
#endif
pszMemValue = _tcsstr(szString,szPartString);
if(pszMemValue == NULL)
{
return EXIT_FAILURE ;
}
//copy the remaining part of the string into a buffer
lstrcpy(szTemp,pszMemValue);
//search for the empty space.
pszdest = _tcschr(szTemp,_T(' '));
if (pszdest==NULL)
{
//the api returns NULL if it is not able to find the
// character . This means that the required switch is at the end
//of the string . so we are copying it fully
lstrcpy(pszFullString,szTemp);
return EXIT_SUCCESS ;
}
dwPos = pszdest - szTemp ;
szTemp[dwPos] = _T('\0');
lstrcpy(pszFullString,szTemp);
return EXIT_SUCCESS ;
}
// ***************************************************************************
// Routine Description:
// This routine is to add/remove the /debugport=1394
// switches to the boot.ini file settings for the specified system.
// Arguments:
// [IN] argc Number of command line arguments
// [IN] argv Array containing command line arguments
//
// Return Value:
// DWORD (EXIT_SUCCESS for success and EXIT_FAILURE for Failure.)
//
// ***************************************************************************
DWORD ProcessDbg1394Switch( DWORD argc, LPCTSTR argv[] )
{
BOOL bUsage = FALSE ;
BOOL bNeedPwd = FALSE ;
BOOL bDbg1394 = FALSE ;
DWORD dwDefault = 0;
TARRAY arr ;
TCHAR szkey[MAX_RES_STRING] = NULL_STRING;
FILE *stream = NULL;
// Initialising the variables that are passed to TCMDPARSER structure
STRING256 szServer = NULL_STRING;
STRING256 szUser = NULL_STRING;
STRING256 szPassword = NULL_STRING;
STRING100 szPath = NULL_STRING;
DWORD dwNumKeys = 0;
BOOL bRes = FALSE ;
LPTSTR szFinalStr = NULL ;
BOOL bFlag = FALSE ;
TCHAR szMaxmem[STRING10] = NULL_STRING ;
TCHAR szBuffer[MAX_RES_STRING] = NULL_STRING ;
TCHAR szDefault[MAX_RES_STRING] = NULL_STRING ;
TCHAR szTemp[MAX_RES_STRING] = NULL_STRING ;
TCHAR szErrorMsg[MAX_RES_STRING] = NULL_STRING ;
LPTSTR szSubString = NULL ;
DWORD dwCode = 0;
DWORD dwChannel = 0;
TCHAR szChannel[MAX_RES_STRING] = NULL_STRING ;
LPCTSTR szToken = NULL ;
DWORD dwRetVal = 0 ;
BOOL bConnFlag = FALSE ;
TCMDPARSER cmdOptions[] =
{
{ CMDOPTION_DBG1394, CP_MAIN_OPTION, 1, 0,&bDbg1394,NULL_STRING , NULL, NULL },
{ SWITCH_SERVER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szServer, NULL_STRING, NULL, NULL },
{ SWITCH_USER, CP_TYPE_TEXT | CP_VALUE_MANDATORY, 1, 0, &szUser, NULL_STRING, NULL, NULL },
{ SWITCH_PASSWORD, CP_TYPE_TEXT | CP_VALUE_OPTIONAL, 1, 0, &szPassword, NULL_STRING, NULL, NULL },
{ CMDOPTION_USAGE, CP_USAGE, 1, 0, &bUsage, NULL_STRING, NULL, NULL },
{ SWITCH_ID, CP_TYPE_NUMERIC | CP_VALUE_MANDATORY| CP_MANDATORY , 1, 0, &dwDefault, NULL_STRING, NULL, NULL },
{ CMDOPTION_CHANNEL, CP_TYPE_NUMERIC | CP_VALUE_MANDATORY,1,0,&dwChannel,NULL_STRING,NULL,NULL},
{ CMDOPTION_DEFAULT, CP_DEFAULT|CP_TYPE_TEXT | CP_MANDATORY, 1, 0, &szDefault,NULL_STRING, NULL, NULL }
};
//
//check if the remote system is 64 bit and if so
// display an error.
//
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_FAILURE )
{
return EXIT_FAILURE ;
}
//copy the Asterix token which is required for password prompting.
_tcscpy(cmdOptions[3].szValues,TOKEN_ASTERIX) ;
_tcscpy(szPassword,TOKEN_ASTERIX);
// Parsing the copy option switches
if ( ! DoParseParam( argc, argv, SIZE_OF_ARRAY(cmdOptions ), cmdOptions ) )
{
DISPLAY_MESSAGE(stderr,ERROR_TAG);
ShowMessage(stderr,GetReason());
return (EXIT_FAILURE);
}
// Displaying query usage if user specified -? with -query option
if( bUsage )
{
dwRetVal = CheckSystemType( szServer);
if(dwRetVal==EXIT_SUCCESS )
{
displayDbg1394Usage_X86();
return (EXIT_SUCCESS);
}else
{
return (EXIT_FAILURE);
}
}
//
//display error message if user enters a value
// other than on or off
//
if( ( lstrcmpi(szDefault,OFF_STRING)!=0 ) && (lstrcmpi(szDefault,ON_STRING)!=0 ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_DEFAULT_MISSING));
DISPLAY_MESSAGE(stderr,GetResString(IDS_1394_HELP));
return (EXIT_FAILURE);
}
if((cmdOptions[5].dwActuals == 0) &&(dwDefault == 0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_ID_MISSING));
DISPLAY_MESSAGE(stderr,GetResString(IDS_1394_HELP));
return (EXIT_FAILURE);
}
if(( lstrcmpi(szDefault,OFF_STRING)==0 ) &&(dwChannel != 0) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_SYNTAX_DBG1394));
return (EXIT_FAILURE);
}
if( ( lstrcmpi(szDefault,ON_STRING)==0 ) && (cmdOptions[6].dwActuals != 0) && ( (dwChannel < 1) ||(dwChannel > 64 )) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_CH_RANGE));
return (EXIT_FAILURE);
}
//display error message if the username is entered with out a machine name
if( (cmdOptions[1].dwActuals == 0)&&(cmdOptions[2].dwActuals != 0))
{
SetReason(GetResString(IDS_USER_BUT_NOMACHINE));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
if( (cmdOptions[2].dwActuals == 0)&&(cmdOptions[3].dwActuals != 0))
{
SetReason(GetResString(IDS_PASSWD_BUT_NOUSER));
ShowMessage(stderr,GetReason());
return EXIT_FAILURE ;
}
//for setting the bNeedPwd
if( ( _tcscmp(szPassword,TOKEN_ASTERIX )==0 ) && (IsLocalSystem(szServer)==FALSE ))
{
bNeedPwd = TRUE ;
}
//set the bneedpassword to true if the server name is specified and password is not specified.
if((cmdOptions[1].dwActuals!=0)&&(cmdOptions[3].dwActuals==0))
{
if( (lstrlen( szServer ) != 0) && (IsLocalSystem(szServer)==FALSE) )
{
bNeedPwd = TRUE ;
}
else
{
bNeedPwd = FALSE ;
}
if(_tcslen(szPassword)!= 0 )
{
_tcscpy(szPassword,NULL_STRING);
}
}
//display an error message if the server is empty.
if((cmdOptions[1].dwActuals!=0)&&(lstrlen(szServer)==0))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_SERVER));
return EXIT_FAILURE ;
}
//display an error message if the user is empty.
if((cmdOptions[2].dwActuals!=0)&&(lstrlen(szUser)==0 ))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_NULL_USER));
return EXIT_FAILURE ;
}
if(StrCmpN(szServer,TOKEN_BACKSLASH4,2)==0)
{
if(!StrCmpN(szServer,TOKEN_BACKSLASH6,3)==0)
{
szToken = _tcstok(szServer,TOKEN_BACKSLASH4);
if(szToken == NULL)
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
return (EXIT_FAILURE);
}
lstrcpy(szServer,szToken);
}
}
szFinalStr = (TCHAR*) malloc(MAX_STRING_LENGTH1* sizeof(TCHAR) );
if (szFinalStr== NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_TAG));
ShowLastError(stderr);
return (EXIT_FAILURE);
}
// Establishing connection to the specified machine and getting the file pointer
// of the boot.ini file if there is no error while establishing connection
lstrcpy(szPath, PATH_BOOTINI );
//display a warning message if the user specifies local system name
// with -s.
if( (IsLocalSystem(szServer)==TRUE)&&(lstrlen(szUser)!=0))
{
DISPLAY_MESSAGE(stdout,GetResString(WARN_LOCALCREDENTIALS));
_tcscpy(szServer,_T(""));
}
bFlag = openConnection( szServer, szUser, szPassword, szPath,bNeedPwd,stream,&bConnFlag);
if(bFlag == EXIT_FAILURE)
{
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting the keys of the Operating system section in the boot.ini file
arr = getKeyValueOfINISection( szPath, OS_FIELD );
if(arr == NULL)
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
// Getting the total number of keys in the operating systems section
dwNumKeys = DynArrayGetCount(arr);
if( (dwNumKeys >= MAX_BOOTID_VAL) && (dwDefault >= MAX_BOOTID_VAL ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_MAX_BOOTID));
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
DestroyDynamicArray(&arr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Displaying error message if the number of keys is less than the OS entry
// line number specified by the user
if( ( dwDefault <= 0 ) || ( dwDefault > dwNumKeys ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_BOOTID));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
// Getting the key of the OS entry specified by the user
if (arr != NULL)
{
LPCWSTR pwsz = NULL;
pwsz = DynArrayItemAsString( arr, dwDefault - 1 ) ;
if(pwsz != NULL)
{
_tcscpy( szkey,pwsz);
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
}
else
{
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
DISPLAY_MESSAGE( stderr, ERROR_TAG);
ShowLastError(stderr);
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
if(lstrcmpi(szDefault,ON_STRING)==0 )
{
if(_tcsstr(szkey,DEBUGPORT) != 0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_DUPLICATE_ENTRY));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
if((_tcsstr(szkey,DEBUG_SWITCH) == 0))
{
lstrcat(szkey,TOKEN_EMPTYSPACE);
lstrcat(szkey,DEBUG_SWITCH);
}
lstrcat(szkey,TOKEN_EMPTYSPACE);
lstrcat(szkey,DEBUGPORT_1394) ;
if(dwChannel!=0)
{
//frame the string and concatenate to the Os Load options.
lstrcat(szkey,TOKEN_EMPTYSPACE);
lstrcat(szkey,TOKEN_CHANNEL);
lstrcat(szkey,TOKEN_EQUAL);
_ltow(dwChannel,szChannel,10);
lstrcat(szkey,szChannel);
}
}
if(lstrcmpi(szDefault,OFF_STRING)==0 )
{
if(_tcsstr(szkey,DEBUGPORT_1394) == 0)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_1394_SWITCH));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
removeSubString(szkey,DEBUGPORT_1394);
removeSubString(szkey,DEBUG_SWITCH);
if(_tcsstr(szkey,TOKEN_CHANNEL)!=0)
{
lstrcpy(szTemp,NULL_STRING);
dwCode = GetSubString(szkey,TOKEN_CHANNEL,szTemp);
if(dwCode == EXIT_FAILURE )
{
DISPLAY_MESSAGE( stderr,GetResString(IDS_NO_TOKENS));
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
if(lstrlen(szTemp)!=0)
{
removeSubString(szkey,szTemp);
}
}
}
if( _tcslen(szkey) >= MAX_RES_STRING)
{
_stprintf(szErrorMsg,GetResString(IDS_ERROR_STRING_LENGTH),MAX_RES_STRING);
DISPLAY_MESSAGE( stderr,szErrorMsg);
resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE ;
}
DynArrayRemove(arr, dwDefault - 1 );
DynArrayInsertString(arr, dwDefault - 1, szkey, MAX_RES_STRING);
// Setting the buffer to 0, to avoid any junk value
memset(szFinalStr, 0, MAX_STRING_LENGTH1);
// Forming the final string from all the key-value pairs
if (stringFromDynamicArray1( arr,szFinalStr) == EXIT_FAILURE)
{
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
resetFileAttrib(szPath);
SafeCloseConnection(szServer,bConnFlag);
return EXIT_FAILURE;
}
// Writing to the profile section with new key-value pair
// If the return value is non-zero, then there is an error.
if( WritePrivateProfileSection(OS_FIELD, szFinalStr, szPath ) != 0 )
{
_stprintf(szBuffer,GetResString(IDS_SUCCESS_CHANGE_OSOPTIONS), dwDefault );
DISPLAY_MESSAGE(stdout,szBuffer);
}
else
{
_stprintf(szBuffer,GetResString(IDS_ERROR_LOAD_OSOPTIONS), dwDefault );
DISPLAY_MESSAGE(stderr,szBuffer);
DestroyDynamicArray(&arr);
resetFileAttrib(szPath);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (EXIT_FAILURE);
}
//reset the file attributes and free the memory and close the connection to the server.
bRes = resetFileAttrib(szPath);
DestroyDynamicArray(&arr);
SAFEFREE(szFinalStr);
SAFECLOSE(stream);
SafeCloseConnection(szServer,bConnFlag);
return (bRes);
}
// ***************************************************************************
//
// Routine Description : Display the help for the Dbg1394 entry option (X86).
//
// Arguments : none
//
// Return Type : VOID
//
// ***************************************************************************
VOID displayDbg1394Usage_X86()
{
DWORD dwIndex = IDS_DBG1394_BEGIN_X86 ;
for(;dwIndex <=IDS_DBG1394_END_X86;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
//
// Routine Description : Display the help for the Dbg1394 entry option (IA64).
//
// Arguments : none
//
// Return Type : VOID
//
// ***************************************************************************
VOID displayDbg1394Usage_IA64()
{
DWORD dwIndex = IDS_DBG1394_BEGIN_IA64 ;
for(;dwIndex <=IDS_DBG1394_END_IA64 ;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}
// ***************************************************************************
//
// Routine Description : determines if the computer is 32 bit system or 64 bit
// Arguments :
// [ in ] szComputerName : System name
// Return Type : DWORD
// TRUE : if the system is a 32 bit system
// FALSE : if the system is a 64 bit system
// ***************************************************************************
DWORD GetCPUInfo(LPTSTR szComputerName)
{
HKEY hKey1 = 0;
HKEY hRemoteKey = 0;
TCHAR szPath[MAX_STRING_LENGTH + 1] = SUBKEY ;
DWORD dwValueSize = MAX_STRING_LENGTH + 1;
DWORD dwRetCode = ERROR_SUCCESS;
DWORD dwError = 0;
TCHAR szTmpCompName[MAX_STRING_LENGTH+1] = NULL_STRING;
TCHAR szTemp[MAX_RES_STRING+1] = NULL_STRING ;
DWORD len = lstrlen(szTemp);
TCHAR szVal[MAX_RES_STRING+1] = NULL_STRING ;
DWORD dwLength = MAX_STRING_LENGTH ;
LPTSTR szReturnValue = NULL ;
DWORD dwCode = 0 ;
szReturnValue = ( LPTSTR ) malloc( dwLength*sizeof( TCHAR ) );
if(szReturnValue == NULL)
{
return ERROR_RETREIVE_REGISTRY ;
}
if(lstrlen(szComputerName)!= 0 )
{
lstrcpy(szTmpCompName,TOKEN_BACKSLASH4);
lstrcat(szTmpCompName,szComputerName);
}
else
{
lstrcpy(szTmpCompName,szComputerName);
}
// Get Remote computer local machine key
dwError = RegConnectRegistry(szTmpCompName,HKEY_LOCAL_MACHINE,&hRemoteKey);
if (dwError == ERROR_SUCCESS)
{
dwError = RegOpenKeyEx(hRemoteKey,szPath,0,KEY_READ,&hKey1);
if (dwError == ERROR_SUCCESS)
{
dwRetCode = RegQueryValueEx(hKey1, IDENTIFIER_VALUE, NULL, NULL,(LPBYTE) szReturnValue, &dwValueSize);
if (dwRetCode == ERROR_MORE_DATA)
{
szReturnValue = ( LPTSTR ) realloc( szReturnValue , dwValueSize * sizeof( TCHAR ) );
dwRetCode = RegQueryValueEx(hKey1, IDENTIFIER_VALUE, NULL, NULL,(LPBYTE) szReturnValue, &dwValueSize);
}
if(dwRetCode != ERROR_SUCCESS)
{
RegCloseKey(hKey1);
RegCloseKey(hRemoteKey);
SAFEFREE(szReturnValue);
return ERROR_RETREIVE_REGISTRY ;
}
}
else
{
RegCloseKey(hRemoteKey);
SAFEFREE(szReturnValue);
return ERROR_RETREIVE_REGISTRY ;
}
RegCloseKey(hKey1);
}
else
{
RegCloseKey(hRemoteKey);
SAFEFREE(szReturnValue);
return ERROR_RETREIVE_REGISTRY ;
}
RegCloseKey(hRemoteKey);
lstrcpy(szVal,X86_MACHINE);
//check if the specified system contains the words x86 (belongs to the 32 )
// set the flag to true if the specified system is 64 bit .
if( !_tcsstr(szReturnValue,szVal))
{
dwCode = SYSTEM_64_BIT ;
}
else
{
dwCode = SYSTEM_32_BIT ;
}
SAFEFREE(szReturnValue);
return dwCode ;
}//GetCPUInfo
// ***************************************************************************
//
// Routine Description : determines if the computer is 32 bit system or 64 bit
// Arguments :
// [ in ] szServer : System name
// Return Type : DWORD
// EXIT_FAILURE : if the system is a 32 bit system
// EXIT_SUCCESS : if the system is a 64 bit system
// ***************************************************************************
DWORD CheckSystemType(LPTSTR szServer)
{
DWORD dwSystemType = 0 ;
#ifndef _WIN64
//display the error message if the target system is a 64 bit system or if error occured in
//retreiving the information
dwSystemType = GetCPUInfo(szServer);
if(dwSystemType == ERROR_RETREIVE_REGISTRY)
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_SYSTEM_INFO));
return (EXIT_FAILURE);
}
if(dwSystemType == SYSTEM_64_BIT)
{
if(lstrlen(szServer)== 0 )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_ERROR_VERSION_MISMATCH));
}
else
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_REMOTE_NOT_SUPPORTED));
}
return (EXIT_FAILURE);
}
#endif
return EXIT_SUCCESS ;
}
// ***************************************************************************
//
// Routine Description : determines if the computer is 32 bit system or 64 bit
// Arguments :
// [ in ] szServer : System name
// [ in ] bFlag : Flag
// Return Type : VOID
//
//
// ***************************************************************************
VOID SafeCloseConnection(LPTSTR szServer,BOOL bFlag)
{
if (bFlag )
{
CloseConnection(szServer);
}
}
// ***************************************************************************
//
// Routine Description : Display the help for the mirror option (IA64).
// Arguments :
// : NONE
//
// Return Type : VOID
//
//
// ***************************************************************************
VOID displayMirrorUsage_IA64()
{
DWORD dwIndex = IDS_MIRROR_BEGIN_IA64 ;
for(;dwIndex <=IDS_MIRROR_END_IA64 ;dwIndex++)
{
DISPLAY_MESSAGE(stdout,GetResString(dwIndex));
}
}