windows-server-2003/admin/cmdline/bootini/bootcfg.h

#ifndef __BOOTINI_H
#define __BOOTINI_H
#endif // __BOOTINI_H

//
// constants / defines / enumerations
//

#define ADD_OFFSET(_p,_o) (PVOID)((PUCHAR)(_p) + (_p)->_o)

//registry path
#define REGISTRY_PATH       L"Software\\Microsoft\\Windows\\CurrentVersion\\Setup"

// Error constants
#define ERROR_CONNECT_SERVERNOTFOUND        0xFFFF0001
#define ERROR_CONNECT_LOGINFAIL             0xFFFF0002
#define ERROR_CONNECT_UNKNOWNERROR          0xFFFF0003
#define ERROR_FMT_INVALID_OPTIONVALUE       _T( "ERROR: Invalid value specified for the option '%s'.\n" )
#define ERROR_FMT_INVALIDOSENTRY            _T( "ERROR: Invalid OS entry line number specified.\nThere are only '%s' OS keys.\n" )

// Exit values
#define EXIT_SUCCESS                        0
#define EXIT_FAILURE                        1

#define STRING20  20
#define FRIENDLY_NAME_LENGTH                  68

// Options
#define CMDOPTION_QUERY                     _T( "query" )
#define CMDOPTION_COPY                      _T( "copy" )
#define CMDOPTION_DELETE                    _T( "delete" )
#define CMDOPTION_USAGE                     _T( "?|h" )
#define CMDOPTION_DEFAULT                   _T( "" )
#define CMDOPTION_RAW                       _T("raw")
#define CMDOPTION_DEFAULTOS                 _T("default")
#define CMDOPTION_TIMEOUT                   _T("timeout")
#define CMDOPTION_DEBUG                     _T("debug")
#define CMDOPTION_EMS                       _T("ems")
#define CMDOPTION_DBG1394                       _T("dbg1394")
#define CMDOPTION_ADDSW                     _T("addsw")
#define CMDOPTION_RMSW                      _T("rmsw")
#define CMDOPTION_APPEND                    _T("a")
#define CMDOPTION_MIRROR                    _T("mirror")
#define CMDOPTION_CHANNEL                   _T("ch")
#define CMDOPTION_LIST                      _T("list")
#define CMDOPTION_ADD                       _T("add")
#define CMDOPTION_UPDATE                    _T("update")

#define OPTION_RAW                          _T("/raw")
#define OPTION_RAW1                         _T("-raw")
#define CMDOPTION_CLONE                     _T("clone")


#define CMDOTHEROPTIONS                     _T( "s|server|u|user|p|password|description|descrip|defaultos|basevideo|baudrate|debugport|maxmem|nodebug|crashdebug|noserialmice|sos|redirect|on|off|edit" )

#define BAUD_RATE_VALUES_EMS          _T("9600|19200|57600|115200")
#define BAUD_RATE_VALUES_DEBUG        _T("9600|19200|38400|57600|115200")

#define COM_PORT_RANGE            _T("COM1|COM2|COM3|COM4")


#define EMS_PORT_VALUES           _T("COM1|COM2|COM3|COM4|BIOSSET")

#define CMDOPTION_DEBUG_VALUES    _T("ON|OFF|EDIT")
#define CMDOPTION_EMS_VALUES    _T("ON|OFF|EDIT")

#define CMDOPTION_EMS_VALUES_IA64    _T("ON|OFF")


#define DEBUGPORT_1394      _T("/debugport=1394")

#define DEBUGPORT           _T("/debugport=")
#define BAUD_TOKEN          _T("/baudrate=")

#define TOKEN_CHANNEL       _T("/channel")

#define TOKEN_BACKSLASH4  _T("\\\\")
#define TOKEN_BACKSLASH6  _T("\\\\\\")


// Other switches or sub-options
#define SWITCH_SERVER                       _T( "s" )
#define SWITCH_USER             _T( "u" )
#define SWITCH_PASSWORD             _T( "p" )
#define SWITCH_TIMEOUT              _T( "to" )
#define SWITCH_DEFAULTOS            _T( "do" )
#define SWITCH_BASEVIDEO            _T( "bv" )
#define SWITCH_DEBUG                _T( "dbg" )
#define SWITCH_BAUDRATE             _T( "br" )
#define SWITCH_DEBUGPORT            _T( "dp" )
#define SWITCH_MAXMEM               _T( "mm" )
#define SWITCH_NODEBUG              _T( "nd" )
#define SWITCH_CRASHDEBUG           _T( "cd" )
#define SWITCH_NOSERIALMICE         _T( "ns" )
#define SWITCH_SOS              _T( "so" )
#define SWITCH_REDIRECT             _T( "re" )
#define SWITCH_DESCRIPTION          _T( "d" )
#define SWITCH_RAWSTRING            _T("raw")
#define TIMEOUT_SWITCH              _T("timeout")
#define SWITCH_PORT                 _T("port")
#define SWITCH_BAUD               _T("baud")
#define SWITCH_NOGUIBOOT          _T("ng")
#define SWITCH_UPTDRV             _T("upddrv")
#define SWITCH_ID               _T("id")
#define SWITCH_EDIT             _T("edit")
#define SWITCH_DISK    		      _T("disk")
#define SWITCH_SOURCEGUID          _T("sg")
#define SWITCH_TARGETGUID          _T("tg")


// Strings
#define OFF_STRING                          _T( "OFF" )
#define ON_STRING                           _T( "ON" )
#define COM_STRING                          _T( "COM" )
#define INIPATH                             _T( "c:\\boottest.ini" )
#define OPERATINGSYSTEMSECTION              _T( "operating systems" )
#define BOOTLOADERSECTION           _T( "boot loader" )

#define  COL_FRIENDLYNAME                       GetResString(IDS_COL_FRIENDLYNAME)
#define  COL_BOOTID                             GetResString(IDS_COL_BOOTID)
#define  COL_BOOTOPTION                             GetResString(IDS_COL_COL_BOOTOPTION)
#define  COL_ARCPATH                        GetResString(IDS_COL_ARCPATH)
#define  OS_HEADER                          GetResString(IDS_OS_HEADER)
#define  DASHES_OS                          GetResString(IDS_DASHES_OS)
#define  BOOT_HEADER                        GetResString(IDS_BOOT_HEADER)
#define  DASHES_BOOTOS                      GetResString(IDS_DASHES_BOOTOS)
#define TIMEOUT_VAL                     GetResString(IDS_TIMEOUT)
#define DEFAULT_OS                      GetResString(IDS_DEFAULT_OS)
#define DEFAULT_ARC                     GetResString(IDS_DEFAULT_ARC)

#define COL_FRIENDLYNAME_WIDTH          AsLong(GetResString(IDS_COL_FRIENDLYNAME_SIZE),10)
#define COL_BOOTID_WIDTH            AsLong(GetResString(IDS_COL_BOOTID_SIZE),10)
#define COL_BOOTOPTION_WIDTH            AsLong(GetResString(IDS_COL_BOOTOPTION_SIZE),10)
#define COL_ARC_WIDTH               AsLong(GetResString(IDS_COL_ARCPATH_SIZE),10)

#define TIMEOUT_WIDTH               AsLong(GetResString(IDS_TIMEOUT_SIZE),10)
#define DEFAULT_ARC_WIDTH           AsLong(GetResString(IDS_DEFAULT_ARC_SIZE),10)


#define KEY_DEFAULT                         _T("default")
#define ONOFFVALUES                         _T(" on off ")
#define KEY_REDIRECT                        _T("redirect")




// Typedefs of standard string sizes
typedef TCHAR STRING256 [ 256 ];

#define ID_DEL_HELP_BEGIN           IDS_DELETE_HELP_BEGIN
#define ID_DEL_HELP_END             IDS_DELETE_HELP_END

#define ID_CHANGE_HELP_BEGIN        IDS_CHANGE_HELP_BEGIN
#define ID_CHANGE_HELP_END          IDS_CHANGE_HELP_END

#define ID_QUERY_HELP_BEGIN         IDS_QUERY_HELP_BEGIN
#define ID_QUERY_HELP_END           IDS_QUERY_HELP_END

#define ID_COPY_HELP_BEGIN          IDS_COPY_HELP_BEGIN
#define ID_COPY_HELP_END            IDS_COPY_HELP_END

#define ID_MAIN_HELP_BEGIN          IDS_MAIN_HELP_BEGIN
#define ID_MAIN_HELP_END            IDS_MAIN_HELP_END


#define RAW_HELP_BEGIN              IDS_RAW_HELP_BEGIN
#define RAW_HELP_END                IDS_RAW_HELP_END

#define TIMEOUT_HELP_BEGIN          IDS_TIMOUTHELP_BEGIN
#define TIMEOUT_HELP_END            IDS_TIMOUTHELP_END

#define DEFAULT_BEGIN               IDS_DEFAULT_BEGIN
#define DEFAULT_END                 IDS_DEFAULT_END

#define ID_MAIN_HELP_BEGIN1         IDS_MAIN_HELP_BEGIN1
#define ID_MAIN_HELP_END1           IDS_MAIN_HELP_END1

#define ID_MAIN_HELP_IA64_BEGIN     IDS_MAIN_HELP_IA64_BEGIN
#define ID_MAIN_HELP_IA64_END       IDS_MAIN_HELP_IA64_END
#define RAW_HELP_IA64_BEGIN         IDS_RAW_HELP_IA64_BEGIN
#define RAW_HELP_IA64_END           IDS_RAW_HELP_IA64_END
#define ID_QUERY_HELP64_BEGIN       IDS_QUERY_HELP64_BEGIN
#define ID_QUERY_HELP64_END         IDS_QUERY_HELP64_END
#define ID_DEL_HELP_IA64_BEGIN      IDS_DELETE_HELP_IA64_BEGIN
#define ID_DEL_HELP_IA64_END        IDS_DELETE_HELP_IA64_END
#define ID_COPY_HELP_IA64_BEGIN     IDS_COPY_HELP_IA64_BEGIN
#define ID_COPY_HELP_IA64_END       IDS_COPY_HELP_IA64_END
#define TIMEOUT_HELP_IA64_BEGIN     IDS_TIMOUTHELP_IA64_BEGIN
#define TIMEOUT_HELP_IA64_END       IDS_TIMOUTHELP_IA64_END
#define DEFAULT_IA64_BEGIN          IDS_DEFAULT_IA64_BEGIN
#define DEFAULT_IA64_END            IDS_DEFAULT_IA64_END



#define SAFEFREE(pVal) \
    FreeMemory((LPVOID *)&pVal); 

#define SIZE_OF_ARRAY_IN_CHARS(x) \
            GetBufferSize(x)/sizeof(WCHAR)

#define SAFECLOSE(stream) \
if(stream != NULL)  \
{ \
    fclose(stream);\
    stream = NULL ; \
}



// function prototypes

// Main functions
DWORD ChangeBootIniSettings(DWORD argc, LPCTSTR argv[]);
DWORD CopyBootIniSettings(DWORD argc, LPCTSTR argv[]);
DWORD DeleteBootIniSettings(DWORD argc, LPCTSTR argv[]);
DWORD QueryBootIniSettings(DWORD argc, LPCTSTR argv[]);

DWORD ChangeTimeOut(DWORD argc,LPCTSTR argv[]);
DWORD ChangeDefaultOs(DWORD argc,LPCTSTR argv[]);


// Function used to get all the keys of a specified section in
// the specified INI file
TARRAY getKeysOfINISection( LPTSTR szinifile, LPTSTR sziniSection );

// Function used to get all the key-value pairs of a specified section in
// the specified INI file
TARRAY getKeyValueOfINISection( LPTSTR szinifile, LPTSTR sziniSection );

// Function used to delete a key from a specifed section of the
// specified ini file
BOOL deleteKeyFromINISection( LPTSTR szkey, LPTSTR szinifile, LPTSTR sziniSection );

// Function used to build the INI string containing all the key-value pairs.
LPTSTR stringFromDynamicArray( TARRAY arrKeyValuePairs );

// Function used to remove a sub-string from a given string
DWORD removeSubString( LPTSTR szString, LPCTSTR szSubString );

// Function used to connect to the specified server with the given credentials
// and return the file pointer of the boot.ini file
BOOL openConnection(LPTSTR server, LPTSTR user,
                     LPTSTR password, LPTSTR filepath,BOOL bNeedPwd,FILE *stream,PBOOL pbConnFlag);

VOID FormHeader1(BOOL bHeader,TCOLUMNS *ResultHeader);

// Exit function
VOID properExit( DWORD dwExitCode, LPTSTR szFilePath );

// Usage functions
VOID displayChangeUsage();
VOID displayDeleteUsage();
VOID displayQueryUsage();
VOID displayRawUsage_X86();
VOID displayRawUsage_IA64();
DWORD displayMainUsage_X86();
VOID displayMainUsage_IA64();
VOID displayChangeOSUsage_X86();
VOID displayDefaultEntryUsage_IA64();
VOID displayRmSwUsage_X86();
VOID displayAddSwUsage_X86();

VOID displayQueryUsage_IA64();
VOID displayQueryUsage_X86();
VOID displayCopyUsage_IA64();
VOID displayCopyUsage_X86();
VOID displayChangeUsage_IA64();
VOID displayChangeUsage_X86();
VOID displayDeleteUsage_IA64();
VOID displayDeleteUsage_X86();
VOID displayTimeOutUsage_IA64();

VOID displayTimeOutUsage_X86();
VOID displayEmsUsage_X86();
VOID displayDebugUsage_X86();
VOID displayDebugUsage_IA64();
VOID displayEmsUsage_IA64();
VOID displayRmSwUsage_IA64();
VOID displayAddSwUsage_IA64();

VOID displayDbg1394Usage_X86();
VOID displayDbg1394Usage_IA64();
VOID displayMirrorUsage_IA64() ;
//VOID displayListUsage_IA64();
//VOID displayUpdateUsage_IA64();

// Function used to process the main options

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,
                        PBOOL pbList ,
                        PBOOL pbUpdate,
                        PBOOL pbClone
                        );


BOOL resetFileAttrib( LPTSTR szFilePath );

BOOL stringFromDynamicArray1( TARRAY arrKeyValuePairs ,LPTSTR szFinalStr );
DWORD stringFromDynamicArray2( TARRAY arrKeyValuePairs ,LPTSTR *szFinalStr );

BOOL EnumerateOsEntries(PBOOT_ENTRY_LIST *ntBootEntries,PULONG ulLength );

VOID FormHeader(BOOL bHeader,TCOLUMNS *ResultHeader,BOOL bVerbose);

DWORD AppendRawString(  DWORD argc, LPCTSTR argv[] );

DWORD ProcessDebugSwitch(  DWORD argc, LPCTSTR argv[] );

DWORD ProcessEmsSwitch(  DWORD argc, LPCTSTR argv[] );

VOID GetComPortType(LPTSTR  szString,LPTSTR szTemp );

VOID GetBaudRateVal(LPTSTR  szString, LPTSTR szTemp);

DWORD getKeysOfSpecifiedINISection( LPTSTR sziniFile, LPTSTR sziniSection,LPCWSTR szKeyName ,LPTSTR szValue );

DWORD ValidateSwitches(PBOOT_ENTRY bootEntry, LPTSTR szNewFriendlyName ,LPTSTR szRawString);

DWORD ProcessAddSwSwitch(  DWORD argc, LPCTSTR argv[] );
DWORD ProcessRmSwSwitch(  DWORD argc, LPCTSTR argv[] );

DWORD GetSubString(LPTSTR szString,LPTSTR szPartString,LPTSTR szFullString);

DWORD ProcessDbg1394Switch(  DWORD argc, LPCTSTR argv[] );
DWORD ProcessListSwitch_IA64(  DWORD argc, LPCTSTR argv[] );

BOOL IsWin64(void) ;

DWORD GetCPUInfo(LPTSTR szComputerName);
DWORD CheckSystemType(LPTSTR szServer);
VOID SafeCloseConnection(LPTSTR szServer,BOOL bFlag) ;
LPTSTR DivideToken( LPTSTR szString,LPTSTR szFinalString );
//LPWSTR DivideToken( LPWSTR szString );


#define ERROR_TAG               GetResString(IDS_ERROR_TAG)
#define ERROR_LOAD_DLL              GetResString(IDS_ERROR_LOAD)
#define ERROR_NO_NVRAM              GetResString(IDS_ERROR_NO_NVRAM)
#define ERROR_UNEXPECTED            GetResString(IDS_UNEXPECTED_ERROR)
#define ERROR_NO_PRIVILAGE          GetResString(IDS_NO_PRIVILAGE)
#define DELETE_SUCCESS              GetResString(IDS_DELETE_SUCCESS)
#define DELETE_FAILURE              GetResString(IDS_DELETE_FAILURE)


#define IDENTIFIER_VALUE  _T("Identifier")
#define SUBKEY _T("HARDWARE\\DESCRIPTION\\SYSTEM\\CENTRALPROCESSOR\\0")

#define IDENTIFIER_VALUE2 _T("SystemPartition")
#define IDENTIFIER_VALUE3 _T("OsLoaderPath")


#define X86_MACHINE _T("x86")

#define SYSTEM_64_BIT 2
#define SYSTEM_32_BIT 3
#define ERROR_RETREIVE_REGISTRY 4



#define NODEBUG_SWITCH _T("/nodebug")
#define BASEVIDEO_SWITCH _T("/basevideo")
#define DEBUG_SWITCH _T("/debug")
#define DEBUG_SWITCH1 _T("/DEBUG")

#define COM_SWITCH _T("COM")
#define CRASHDEBUG_SWITCH _T("/crashdebug")
#define SOS_SWITCH _T("/sos")
#define REDIRECT_SWITCH _T("/redirect")
#define OS_FIELD _T( "operating systems" )

#define SOS_VALUE _T("/sos")
#define NOGUI_VALUE  _T("/noguiboot")
#define BASEVIDEO_VALUE _T("/basevideo")

#define MAXMEM_VALUE _T("/MAXMEM")
#define MAXMEM_VALUE1 _T("/maxmem")

#define NTDLL_FIELD _T("\\ntdll.dll")

#define TOKEN_EQUAL _T("=")
#define TOKEN_BACKSLASH _T("\"")
#define TOKEN_BRACKET _T('[')
#define TOKEN_DELIM _T('\0')
#define TOKEN_EMPTYSPACE _T(" ")

#define TOKEN_SINGLEQUOTE _T("\"")

#define TOKEN_BACKSLASH4 _T("\\\\")
#define TOKEN_BACKSLASH2 _T("\\")
#define TOKEN_FWDSLASH1 _T("/")
#define TOKEN_C_DOLLAR _T("C$")
#define TOKEN_STAR    _T("*")

#define TOKEN_BOOTINI_PATH _T("boot.ini")
#define TOKEN_DOLLAR _T('$')
#define TOKEN_PATH _T("c:\\boot.ini")
#define TOKEN_COLON _T(':')
#define TOKEN_COLONSYMBOL _T(":")

#define TOKEN_50SPACES _T("                                                 ")
#define TOKEN_1394   _T("1394")

#define TOKEN_NA _T("N/A")
#define TOKEN_NEXTLINE _T("\n")
#define MAX_COLUMNS 4

#define TIMEOUT_MAX 999
#define TIMEOUT_MIN 0

#define READ_MODE  _T("r")
#define READWRITE_MODE _T("r+")

#define BOOT_COLUMNS 2

#define COL0        0
#define COL1        1
#define COL2        2
#define COL3        3

#define TABLE_FORMAT                        GetResString(IDS_FORMAT_TABLE)
#define LIST_FORMAT                     GetResString(IDS_FORMAT_LIST)
#define CSV_FORMAT                      GetResString(IDS_FORMAT_CSV)

#define MAX_INI_LENGTH  2500
#define MAX_STRING_LENGTH1 5000


#define ON_OFF_EDIT _T("ON|OFF|EDIT")

#define TOKEN_DEBUGPORT _T("/debugport")


#define PORT_COM1  _T("/debugport=com1")
#define PORT_COM2  _T("/debugport=com2")
#define PORT_COM3  _T("/debugport=com3")
#define PORT_COM4  _T("/debugport=com4")
#define PORT_1394  _T("/debugport=1394")


#define BAUD_VAL6  _T("/baudrate=19200")
#define BAUD_VAL7  _T("/baudrate=38400")
#define BAUD_VAL8  _T("/baudrate=57600")
#define BAUD_VAL9  _T("/baudrate=115200")
#define BAUD_VAL10  _T("/baudrate=9600")

#define BAUD_RATE       _T("/baudrate")
#define REDIRECT    _T("/redirect")

#define VALUE_OFF _T("off")
#define VALUE_ON _T("on")

#define EDIT_STRING _T("EDIT")

#define KEY_BAUDRATE  _T("redirectbaudrate")

#define USEBIOSSET _T("biosset")

#define USEBIOSSETTINGS _T("USEBIOSSETTINGS")

#define ERROR_PROFILE_STRING _T("ERROR:")

#define ERROR_PROFILE_STRING1 _T("")

#define REDIRECT_STRING _T("redirect")

#define BAUDRATE_STRING _T("redirectbaudrate")

#define MAX_BOOTID_VAL  20

#define STRING255 255

#define STRING10 10


#define DRIVE_C 'C'
#define DRIVE_D 'D'
#define DRIVE_E 'E'
#define DRIVE_F 'F'
#define DRIVE_G 'G'
#define DRIVE_H 'H'
#define DRIVE_I 'I'
#define DRIVE_J 'J'
#define DRIVE_K 'K'
#define DRIVE_L 'L'
#define DRIVE_M 'M'
#define DRIVE_N 'N'
#define DRIVE_O 'O'
#define DRIVE_P 'P'
#define DRIVE_Q 'Q'
#define DRIVE_R 'R'
#define DRIVE_S 'S'
#define DRIVE_T 'T'
#define DRIVE_U 'U'
#define DRIVE_V 'V'
#define DRIVE_W 'W'
#define DRIVE_X 'X'
#define DRIVE_Y 'Y'
#define DRIVE_Z 'Z'


#define OI_SERVER 1
#define OI_USER 2
#define OI_PASSWORD 3

//global variable to store attributes
DWORD g_dwAttributes;

DWORD IsAdminGroup( BOOL *bAdmin );
BOOL GetTokenHandle(OUT PHANDLE hTokenHandle);
BOOL CheckAdminAccess( LPCTSTR pszMachine );