mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
971 lines
28 KiB
971 lines
28 KiB
/*--
|
|
|
|
Copyright (c) 1997-1997 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
dssetdc.c
|
|
|
|
Abstract:
|
|
|
|
Command line tool for promoting/demoting servers into and out of the Ds
|
|
|
|
Author:
|
|
|
|
1-Apr-1997 Mac McLain (macm) Created
|
|
|
|
Environment:
|
|
|
|
User mode only.
|
|
Requires ANSI C extensions: slash-slash comments, long external names.
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
#include <nt.h>
|
|
#include <ntrtl.h>
|
|
#include <nturtl.h>
|
|
#include <windows.h>
|
|
#include <ntlsa.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <dsrole.h>
|
|
#include <dsrolep.h>
|
|
|
|
#define TAG_DNS "dns"
|
|
#define TAG_FLAT "flat"
|
|
#define TAG_SITE "site"
|
|
#define TAG_DB "db"
|
|
#define TAG_LOG "log"
|
|
#define TAG_SYSVOL "sysvol"
|
|
#define TAG_PARENT "parent"
|
|
#define TAG_REPLICA "replica"
|
|
#define TAG_USER "user"
|
|
#define TAG_PASSWORD "password"
|
|
#define TAG_SERVER "server"
|
|
#define TAG_OPTIONS "options"
|
|
#define TAG_LEVEL "level"
|
|
#define TAG_ROLE "role"
|
|
#define TAG_LASTDC "lastdc"
|
|
#define TAG_ADMINPWD "adminpwd"
|
|
#define TAG_WAIT "wait"
|
|
#define TAG_FIXDC "fixdc"
|
|
|
|
//
|
|
// Macro to help command line parsing...
|
|
//
|
|
#define PARSE_CMD_VALUE_AND_CONTINUE( tag, value, buff, ptr ) \
|
|
if ( !_strnicmp( value, tag, sizeof( tag ) - 1 ) ) { \
|
|
value += sizeof(tag) - 1; \
|
|
if ( *value == ':' ) { \
|
|
value++; \
|
|
mbstowcs(buff, value, strlen(value) + 1); \
|
|
ptr = buff; \
|
|
continue; \
|
|
} \
|
|
}
|
|
|
|
VOID
|
|
Usage (
|
|
PWSTR DefaultDns,
|
|
PWSTR DefaultFlat,
|
|
PWSTR DefaultSite,
|
|
PWSTR DefaultPath
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Displays the expected usage
|
|
|
|
Arguments:
|
|
|
|
Return Value:
|
|
|
|
VOID
|
|
|
|
--*/
|
|
{
|
|
printf("dssetdc -promote <parameters>\n");
|
|
printf(" -info <parameters>\n");
|
|
printf(" -demote <parameters>\n");
|
|
printf(" -save\n");
|
|
printf(" -abort <parameters>\n");
|
|
printf(" -upgrade <parameters>\n");
|
|
printf(" -fixdc <parameters>\n");
|
|
printf(" where:\n");
|
|
printf(" promote parameters are:\n");
|
|
printf(" -%s:dns domain name of the new domain/domain to install as replica "
|
|
"of. Defaults to %ws\n", TAG_DNS, DefaultDns);
|
|
printf(" -%s:NetBIOS domain name. Defaults to %ws\n", TAG_FLAT, DefaultFlat);
|
|
printf(" -%s:site name. Defaults to %ws\n", TAG_SITE, DefaultSite);
|
|
printf(" -%s:db path Defaults to %ws.\n", TAG_DB, DefaultPath);
|
|
printf(" -%s:log path Defaults to %ws.\n", TAG_LOG, DefaultPath);
|
|
printf(" -%s:sysvol path. Defaults to %ws. [Must be NT5 NTFS]\n", TAG_SYSVOL, DefaultPath);
|
|
printf(" [-%s:parent dns domain name if this is a child domain]\n", TAG_PARENT);
|
|
printf(" [-%s:replica partner]\n", TAG_REPLICA);
|
|
printf(" [-%s:account]\n", TAG_USER);
|
|
printf(" [-%s:password]\n", TAG_PASSWORD);
|
|
printf(" [-%s:options]\n", TAG_OPTIONS);
|
|
printf(" [-%s:if 1, block until the call has completed\n",TAG_WAIT);
|
|
printf(" info parameters are:\n");
|
|
printf(" -%s:Remote server to obtain the information for\n", TAG_SERVER);
|
|
printf(" -%s:info level\n", TAG_LEVEL);
|
|
printf(" Valid info levels are:\n");
|
|
printf(" 1 - PrimaryDomainInformation\n");
|
|
printf(" 2 - Upgrade State information\n");
|
|
printf(" demote parameters are:\n");
|
|
printf(" -%s:server role\n", TAG_ROLE);
|
|
printf(" Valid server roles are:\n");
|
|
printf(" 2 - Member server\n");
|
|
printf(" 3 - Standalone server\n");
|
|
printf(" [-%s:account]\n", TAG_USER);
|
|
printf(" [-%s:password]\n", TAG_PASSWORD);
|
|
printf(" [-%s:options]\n", TAG_OPTIONS);
|
|
printf(" [-%s:Whether this is the last dc in the domain]\n", TAG_LASTDC);
|
|
printf(" Valid options are:\n");
|
|
printf(" 0 - Not the last dc in the domain\n");
|
|
printf(" 1 - Last DC in the domain\n");
|
|
printf(" [-%s:if 1, block until the call has completed\n",TAG_WAIT);
|
|
printf(" -abort parameters are:\n");
|
|
printf(" -%s:new administrator password. Defaults to NULL\n", TAG_ADMINPWD );
|
|
printf(" -upgrade parameters are:\n");
|
|
printf(" -%s:dns domain name of the new domain. Defaults to %ws\n", TAG_DNS, DefaultDns);
|
|
printf(" -%s:site name. Defaults to %ws\n", TAG_SITE, DefaultSite);
|
|
printf(" -%s:db path Defaults to %ws.\n", TAG_DB, DefaultPath);
|
|
printf(" -%s:log path Defaults to %ws.\n", TAG_LOG, DefaultPath);
|
|
printf(" -%s:sysvol path. Defaults to %ws. [Must be NT5 NTFS]\n", TAG_SYSVOL, DefaultPath);
|
|
printf(" [-%s:parent dns domain]\n", TAG_PARENT);
|
|
printf(" [-%s:account]\n", TAG_USER);
|
|
printf(" [-%s:password]\n", TAG_PASSWORD);
|
|
printf(" [-%s:options]\n", TAG_OPTIONS);
|
|
printf(" -fixdc parameters are:\n");
|
|
printf(" [-%s:server]Remote server to obtain the information for\n", TAG_SERVER);
|
|
printf(" [-%s:server]Remote server to sync with\n", TAG_REPLICA);
|
|
printf(" [-%s:account]\n", TAG_USER);
|
|
printf(" [-%s:password]\n", TAG_PASSWORD);
|
|
printf(" [-%s:options]\n", TAG_OPTIONS);
|
|
printf(" Valid options (in HEX) are:\n");
|
|
printf(" 0x00000001 - Create the machine account if necessary\n");
|
|
printf(" 0x00000002 - Sync the machine password\n");
|
|
printf(" 0x00000004 - Change the account time\n");
|
|
printf(" 0x00000008 - Re-init the time service\n");
|
|
printf(" 0x00000010 - Reconfigure the default services\n");
|
|
printf(" 0x00000020 - Force a DS replication cycle\n");
|
|
printf(" 0x00000040 - Fixup NTFRS\n");
|
|
}
|
|
|
|
|
|
DWORD
|
|
GetAndDumpInfo(
|
|
IN PWSTR Server,
|
|
IN ULONG Level
|
|
)
|
|
{
|
|
DWORD Win32Error = ERROR_SUCCESS;
|
|
PDSROLE_UPGRADE_STATUS_INFO UpgradeInfo = NULL;
|
|
PDSROLE_PRIMARY_DOMAIN_INFO_BASIC DomainInfo = NULL;
|
|
PBYTE Buffer = NULL;
|
|
PWSTR Roles[ ] = {
|
|
L"DsRole_RoleStandaloneWorkstation",
|
|
L"DsRole_RoleMemberWorkstation",
|
|
L"DsRole_RoleStandaloneServer",
|
|
L"DsRole_RoleMemberServer",
|
|
L"DsRole_RoleBackupDomainController",
|
|
L"DsRole_RolePrimaryDomainController"
|
|
};
|
|
PWSTR ServerRoles[ ] = {
|
|
L"Unknown",
|
|
L"Primary",
|
|
L"Backup"
|
|
};
|
|
|
|
|
|
Win32Error = DsRoleGetPrimaryDomainInformation( Server,
|
|
Level,
|
|
( PBYTE * )&Buffer );
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
switch ( Level ) {
|
|
case DsRolePrimaryDomainInfoBasic:
|
|
|
|
DomainInfo = ( PDSROLE_PRIMARY_DOMAIN_INFO_BASIC )Buffer;
|
|
printf( "Machine Role: %lu ( %ws )\n", DomainInfo->MachineRole,
|
|
Roles[ DomainInfo->MachineRole ] );
|
|
printf( "Flags: 0x%lx\n", DomainInfo->Flags );
|
|
printf( "Flat name: %ws\n", DomainInfo->DomainNameFlat );
|
|
printf( "Dns Domain name: %ws\n", DomainInfo->DomainNameDns );
|
|
printf( "Dns Forest name: %ws\n", DomainInfo->DomainForestName );
|
|
break;
|
|
|
|
case DsRoleUpgradeStatus:
|
|
UpgradeInfo = ( PDSROLE_UPGRADE_STATUS_INFO )Buffer;
|
|
|
|
printf( "Upgrade: %s\n",
|
|
( UpgradeInfo->OperationState & DSROLE_UPGRADE_IN_PROGRESS ) ?
|
|
"TRUE" : "FALSE" );
|
|
printf( "Previous server role: %ws\n", ServerRoles[ UpgradeInfo->PreviousServerState ] );
|
|
|
|
}
|
|
|
|
DsRoleFreeMemory( DomainInfo );
|
|
|
|
} else {
|
|
|
|
printf( "DsRoleGetPrimaryDomainInformation failed with %lu\n", Win32Error );
|
|
}
|
|
|
|
|
|
return( Win32Error );
|
|
}
|
|
|
|
|
|
|
|
|
|
DWORD
|
|
BuildDefaults(
|
|
IN PWSTR Dns,
|
|
IN PWSTR Flat,
|
|
IN PWSTR Site,
|
|
IN PWSTR Path
|
|
)
|
|
{
|
|
DWORD Win32Error = ERROR_SUCCESS;
|
|
PWSTR Scratch;
|
|
ULONG Options;
|
|
|
|
//
|
|
// First, the easy ones
|
|
//
|
|
wcscpy( Site, L"First Site" );
|
|
|
|
ExpandEnvironmentStrings( L"%systemroot%\\ntds", Path, MAX_PATH );
|
|
|
|
return( Win32Error );
|
|
}
|
|
|
|
|
|
DWORD
|
|
CopyDsDitFiles(
|
|
IN LPWSTR DsPath
|
|
)
|
|
{
|
|
DWORD Win32Error = ERROR_SUCCESS;
|
|
WCHAR Source[MAX_PATH + 1];
|
|
WCHAR Dest[MAX_PATH + 1];
|
|
ULONG SrcLen, DestLen = 0;
|
|
PWSTR DsDitFiles[] = {
|
|
L"ntds.dit",
|
|
L"schema.ini"
|
|
};
|
|
PWSTR CleanupFiles[] = {
|
|
L"edb.chk",
|
|
L"edb.log",
|
|
L"hierarch.dat",
|
|
L"ntds.dit",
|
|
L"res1.log",
|
|
L"res2.log",
|
|
L"schema.ini",
|
|
L"edb00001.log",
|
|
L"edb00002.log",
|
|
L"edb00003.log"
|
|
L"edb00004.log"
|
|
L"edb00005.log"
|
|
L"edb00006.log"
|
|
};
|
|
PWSTR Current;
|
|
ULONG i;
|
|
|
|
|
|
|
|
if( ExpandEnvironmentStrings( L"%WINDIR%\\system32\\", Source, MAX_PATH ) == FALSE ) {
|
|
|
|
Win32Error = GetLastError();
|
|
|
|
} else {
|
|
|
|
SrcLen = wcslen( Source );
|
|
wcscpy( Dest, DsPath );
|
|
|
|
if ( *(Dest + (wcslen( DsPath ) - 1 )) != L'\\' ) {
|
|
|
|
wcscat( Dest, L"\\" );
|
|
}
|
|
|
|
DestLen = wcslen( Dest );
|
|
|
|
}
|
|
|
|
|
|
//
|
|
// See if the source directory exists...
|
|
//
|
|
if ( Win32Error == ERROR_SUCCESS && GetFileAttributes( DsPath ) == 0x10 ) {
|
|
|
|
for ( i = 0; i < sizeof( CleanupFiles) / sizeof( PWSTR ); i++ ) {
|
|
|
|
wcscpy( Dest + DestLen, CleanupFiles[i] );
|
|
|
|
if ( DeleteFile( Dest ) == FALSE ) {
|
|
|
|
Win32Error = GetLastError();
|
|
|
|
if ( Win32Error == ERROR_FILE_NOT_FOUND ) {
|
|
|
|
Win32Error = ERROR_SUCCESS;
|
|
|
|
} else {
|
|
|
|
printf("Failed to remove %ws: %lu\n", Dest, Win32Error );
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
//
|
|
// Then, create the destination directory
|
|
//
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
Current = wcschr( DsPath + 4, L'\\' );
|
|
|
|
while ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
if ( Current != NULL ) {
|
|
|
|
*Current = UNICODE_NULL;
|
|
|
|
}
|
|
|
|
if ( CreateDirectory( DsPath, NULL ) == FALSE ) {
|
|
|
|
Win32Error = GetLastError();
|
|
|
|
if ( Win32Error == ERROR_ALREADY_EXISTS) {
|
|
|
|
Win32Error = ERROR_SUCCESS;
|
|
}
|
|
}
|
|
|
|
if ( Current != NULL ) {
|
|
|
|
*Current = L'\\';
|
|
|
|
Current = wcschr( Current + 1, L'\\' );
|
|
|
|
} else {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
return( Win32Error );
|
|
}
|
|
|
|
|
|
DWORD
|
|
Promote( LPWSTR Dns, LPWSTR Flat, LPWSTR Site, LPWSTR Db, LPWSTR Log,
|
|
LPWSTR SysVol, LPWSTR Parent, LPWSTR Replica, LPWSTR User,
|
|
LPWSTR Password, LPWSTR Server, ULONG Options, BOOL Block
|
|
)
|
|
{
|
|
DWORD Win32Error = ERROR_SUCCESS;
|
|
DSROLE_SERVEROP_HANDLE Handle;
|
|
PDSROLE_SERVEROP_STATUS Status;
|
|
PDSROLE_SERVEROP_RESULTS Results;
|
|
|
|
if ( !Server ) {
|
|
Win32Error = CopyDsDitFiles( Db );
|
|
|
|
if ( Win32Error != ERROR_SUCCESS ) {
|
|
|
|
return( Win32Error );
|
|
}
|
|
}
|
|
|
|
//
|
|
// Now, do the install
|
|
//
|
|
if ( Replica != NULL ) {
|
|
|
|
Win32Error = DsRoleDcAsReplica( Server,
|
|
Dns,
|
|
Replica,
|
|
Site,
|
|
Db,
|
|
Log,
|
|
NULL,
|
|
SysVol,
|
|
NULL,
|
|
User,
|
|
Password,
|
|
NULL,
|
|
Options,
|
|
&Handle );
|
|
|
|
} else {
|
|
|
|
Win32Error = DsRoleDcAsDc( Server,
|
|
Dns,
|
|
Flat,
|
|
NULL,
|
|
Site,
|
|
Db,
|
|
Log,
|
|
SysVol,
|
|
Parent,
|
|
NULL,
|
|
User,
|
|
Password,
|
|
NULL,
|
|
Options,
|
|
&Handle );
|
|
}
|
|
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
if ( !Block ) {
|
|
do {
|
|
|
|
Sleep( 6000 );
|
|
|
|
Win32Error = DsRoleGetDcOperationProgress( NULL,
|
|
Handle,
|
|
&Status );
|
|
|
|
if ( Win32Error == ERROR_SUCCESS || Win32Error == ERROR_IO_PENDING ) {
|
|
|
|
printf("%ws\n", Status->CurrentOperationDisplayString );
|
|
DsRoleFreeMemory( Status );
|
|
}
|
|
|
|
|
|
} while( Win32Error == ERROR_IO_PENDING);
|
|
|
|
if ( Win32Error != ERROR_SUCCESS ) {
|
|
|
|
printf("Failed determining the operation progress: %lu\n", Win32Error );
|
|
}
|
|
|
|
} else {
|
|
|
|
printf( "Block on DsRoleGetDcOperationResutls call\n" );
|
|
}
|
|
|
|
} else {
|
|
|
|
printf( "Failed to install as a Dc: %lu\n", Win32Error );
|
|
}
|
|
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
Win32Error = DsRoleGetDcOperationResults( NULL,
|
|
Handle,
|
|
&Results );
|
|
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
Win32Error = Results->OperationStatus;;
|
|
printf( "OperationResults->OperationStatusDisplayString: %ws\n",
|
|
Results->OperationStatusDisplayString );
|
|
|
|
printf( "OperationResults->ServerInstalledSite: %ws\n",
|
|
Results->ServerInstalledSite );
|
|
|
|
DsRoleFreeMemory( Results );
|
|
|
|
} else {
|
|
|
|
printf( "Failed to determine the operation results: %lu\n", Win32Error );
|
|
}
|
|
}
|
|
|
|
return( Win32Error );
|
|
}
|
|
|
|
DWORD
|
|
Demote( LPWSTR User, LPWSTR Password, ULONG Role, ULONG Options, BOOL LastDc )
|
|
{
|
|
DWORD Win32Error = ERROR_SUCCESS;
|
|
DSROLE_SERVEROP_HANDLE Handle;
|
|
PDSROLE_SERVEROP_STATUS Status;
|
|
PDSROLE_SERVEROP_RESULTS Results;
|
|
|
|
Win32Error = DsRoleDemoteDc( NULL, NULL, Role, User, Password, Options,
|
|
LastDc, NULL, &Handle );
|
|
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
do {
|
|
|
|
Sleep( 6000 );
|
|
|
|
Win32Error = DsRoleGetDcOperationProgress( NULL,
|
|
Handle,
|
|
&Status );
|
|
|
|
if ( Win32Error == ERROR_SUCCESS || Win32Error == ERROR_IO_PENDING ) {
|
|
|
|
printf("%ws\n", Status->CurrentOperationDisplayString );
|
|
DsRoleFreeMemory( Status );
|
|
}
|
|
|
|
|
|
} while( Win32Error == ERROR_IO_PENDING);
|
|
|
|
if ( Win32Error != ERROR_SUCCESS ) {
|
|
|
|
printf("Failed determining the operation progress: %lu\n", Win32Error );
|
|
}
|
|
|
|
} else {
|
|
|
|
printf( "Failed to demote a Dc: %lu\n", Win32Error );
|
|
}
|
|
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
Win32Error = DsRoleGetDcOperationResults( NULL,
|
|
Handle,
|
|
&Results );
|
|
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
Win32Error = Results->OperationStatus;;
|
|
printf( "OperationResults->OperationStatusDisplayString: %ws\n",
|
|
Results->OperationStatusDisplayString );
|
|
|
|
printf( "OperationResults->ServerInstalledSite: %ws\n",
|
|
Results->ServerInstalledSite );
|
|
|
|
DsRoleFreeMemory( Results );
|
|
|
|
} else {
|
|
|
|
printf( "Failed to determine the operation results: %lu\n", Win32Error );
|
|
}
|
|
}
|
|
|
|
return( Win32Error );
|
|
}
|
|
|
|
DWORD
|
|
Save( VOID )
|
|
{
|
|
DWORD Win32Error = ERROR_SUCCESS;
|
|
|
|
Win32Error = DsRoleServerSaveStateForUpgrade( NULL );
|
|
|
|
if ( Win32Error != ERROR_SUCCESS ) {
|
|
|
|
printf("DsRoleServerSaveStateForUpgrade failed with %lu\n", Win32Error );
|
|
}
|
|
|
|
return( Win32Error );
|
|
}
|
|
|
|
DWORD
|
|
Abort(
|
|
PWSTR AdminPwd
|
|
)
|
|
{
|
|
DWORD Win32Error = ERROR_SUCCESS;
|
|
|
|
Win32Error = DsRoleAbortDownlevelServerUpgrade( AdminPwd, NULL, NULL, 0 );
|
|
|
|
if ( Win32Error != ERROR_SUCCESS ) {
|
|
|
|
printf("DsRoleAbortDownlevelServerUpgrade failed with %lu\n", Win32Error );
|
|
}
|
|
|
|
|
|
return( Win32Error );
|
|
}
|
|
|
|
DWORD
|
|
Upgrade( LPWSTR Dns, LPWSTR Site, LPWSTR Db, LPWSTR Log, LPWSTR SysVol,
|
|
LPWSTR Parent, LPWSTR User, LPWSTR Password, ULONG Options )
|
|
{
|
|
DWORD Win32Error = ERROR_SUCCESS;
|
|
DSROLE_SERVEROP_HANDLE Handle;
|
|
PDSROLE_SERVEROP_STATUS Status;
|
|
PDSROLE_SERVEROP_RESULTS Results;
|
|
|
|
Win32Error = DsRoleUpgradeDownlevelServer( ( LPCWSTR )Dns, ( LPCWSTR )Site, ( LPCWSTR )Db,
|
|
( LPCWSTR )Log, ( LPCWSTR )SysVol,
|
|
( LPCWSTR )Parent, NULL, ( LPCWSTR )User,
|
|
( LPCWSTR )Password, NULL, Options, &Handle );
|
|
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
do {
|
|
|
|
Sleep( 6000 );
|
|
|
|
Win32Error = DsRoleGetDcOperationProgress( NULL,
|
|
Handle,
|
|
&Status );
|
|
|
|
if ( Win32Error == ERROR_SUCCESS || Win32Error == ERROR_IO_PENDING ) {
|
|
|
|
printf("%ws\n", Status->CurrentOperationDisplayString );
|
|
DsRoleFreeMemory( Status );
|
|
}
|
|
|
|
|
|
} while( Win32Error == ERROR_IO_PENDING);
|
|
|
|
if ( Win32Error != ERROR_SUCCESS ) {
|
|
|
|
printf("Failed determining the operation progress: %lu\n", Win32Error );
|
|
}
|
|
|
|
} else {
|
|
|
|
printf( "Failed to install as a Dc: %lu\n", Win32Error );
|
|
}
|
|
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
Win32Error = DsRoleGetDcOperationResults( NULL,
|
|
Handle,
|
|
&Results );
|
|
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
Win32Error = Results->OperationStatus;;
|
|
printf( "OperationResults->OperationStatusDisplayString: %ws\n",
|
|
Results->OperationStatusDisplayString );
|
|
|
|
printf( "OperationResults->ServerInstalledSite: %ws\n",
|
|
Results->ServerInstalledSite );
|
|
|
|
DsRoleFreeMemory( Results );
|
|
|
|
} else {
|
|
|
|
printf( "Failed to determine the operation results: %lu\n", Win32Error );
|
|
}
|
|
}
|
|
|
|
|
|
return( Win32Error );
|
|
}
|
|
|
|
INT
|
|
__cdecl main (
|
|
int argc,
|
|
char *argv[])
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
The main the for this executable
|
|
|
|
Arguments:
|
|
|
|
argc - Count of arguments
|
|
argv - List of arguments
|
|
|
|
Return Value:
|
|
|
|
VOID
|
|
|
|
--*/
|
|
{
|
|
DWORD Win32Error = ERROR_SUCCESS, OpErr;
|
|
WCHAR DnsBuff[MAX_PATH + 1], SiteBuff[MAX_PATH + 1];
|
|
WCHAR DbBuff[MAX_PATH + 1], LogBuff[MAX_PATH + 1], ParentBuff[MAX_PATH + 1];
|
|
WCHAR ReplicaBuff[MAX_PATH + 1], UserBuff[MAX_PATH + 1], PasswordBuff[MAX_PATH + 1];
|
|
WCHAR ScratchBuff[MAX_PATH], ServerBuff[MAX_PATH], FlatBuff[MAX_PATH + 1];
|
|
WCHAR SysVolBuff[MAX_PATH + 1], RoleBuff[ 10 ], LastDCBuff[ 10 ], WaitBuff[ 10 ];
|
|
WCHAR AdminPwdBuff[ MAX_PATH + 1 ];
|
|
ULONG Options = 0, Level = 0, Role = 0, FailedOperation, CompletedOperations;
|
|
PWSTR Parent = NULL, Replica = NULL, User = NULL, Password = NULL, Dns = NULL, Flat = NULL;
|
|
PWSTR Site = NULL, Db = NULL, Log = NULL, Scratch = NULL, Server = NULL;
|
|
PWSTR SysVol = NULL, RoleScratch = NULL, LastDCScratch = NULL, AdminPwd = NULL, Wait = NULL;
|
|
INT i = 1;
|
|
BOOL LastDC = FALSE, BuildDefaultsFailed = FALSE, Block = FALSE;
|
|
PSTR Current;
|
|
|
|
Win32Error = BuildDefaults( DnsBuff, FlatBuff, SiteBuff, DbBuff );
|
|
|
|
if ( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
Dns = DnsBuff;
|
|
Site = SiteBuff;
|
|
Db = DbBuff;
|
|
Log = DbBuff;
|
|
Flat = FlatBuff;
|
|
|
|
} else {
|
|
|
|
BuildDefaultsFailed = TRUE;
|
|
|
|
}
|
|
|
|
|
|
if (argc > 1 && (_stricmp( argv[1], "-?") == 0 || _stricmp( argv[1], "/?") == 0 ) ) {
|
|
|
|
if ( BuildDefaultsFailed ) {
|
|
|
|
printf( "Failed to get the defaults: %lu\n", Win32Error );
|
|
}
|
|
Usage( Dns, Flat, Site, Db );
|
|
goto Done;
|
|
|
|
|
|
} else if (argc > 1 && ( _stricmp( argv[ 1 ], "-info" ) == 0 || _stricmp( argv[ 1 ], "/info" ) == 0 ) ) {
|
|
|
|
for (i = 2; i < argc; i++ ) {
|
|
|
|
Current = argv[i];
|
|
|
|
if ( !( *Current == '-' || *Current == '/' ) ) {
|
|
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
break;
|
|
}
|
|
|
|
Current++;
|
|
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_SERVER, Current, ServerBuff, Server );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_LEVEL, Current, ScratchBuff, Scratch );
|
|
}
|
|
|
|
if ( Win32Error == ERROR_SUCCESS && Scratch ) {
|
|
|
|
Level = wcstol( Scratch, &Scratch, 10 );
|
|
}
|
|
|
|
Win32Error = GetAndDumpInfo( Server, Level );
|
|
goto Done;
|
|
|
|
} else if (argc > 1 && ( _stricmp( argv[ 1 ], "-promote" ) == 0 || _stricmp( argv[ 1 ], "/promote" ) == 0) ) {
|
|
|
|
if ( BuildDefaultsFailed ) {
|
|
|
|
printf( "Failed to get the defaults: %lu\n", Win32Error );
|
|
goto Done;
|
|
|
|
}
|
|
|
|
for (i = 2; i < argc; i++ ) {
|
|
|
|
Current = argv[i];
|
|
|
|
if ( !( *Current == '-' || *Current == '/' ) ) {
|
|
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
break;
|
|
}
|
|
|
|
Current++;
|
|
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_DNS, Current, DnsBuff, Dns );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_FLAT, Current, FlatBuff, Flat );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_SITE, Current, SiteBuff, Site );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_DB, Current, DbBuff, Db );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_LOG, Current, LogBuff, Log );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_SYSVOL, Current, SysVolBuff, SysVol );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_PARENT, Current, ParentBuff, Parent );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_REPLICA, Current, ReplicaBuff, Replica );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_USER, Current, UserBuff, User );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_PASSWORD, Current, PasswordBuff, Password );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_SERVER, Current, ServerBuff, Server );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_OPTIONS, Current, ScratchBuff, Scratch );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_WAIT, Current, WaitBuff, Wait );
|
|
|
|
printf("Unexpected command line value %s\n\n", argv[i] );
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
|
|
break;
|
|
}
|
|
|
|
if ( Scratch != NULL ) {
|
|
|
|
Options = wcstoul( Scratch, &Scratch, 0 );
|
|
}
|
|
|
|
//
|
|
// Validate the parameters
|
|
//
|
|
if ( Dns == NULL || Db == NULL || Log == NULL || Flat == NULL ) {
|
|
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
if( Win32Error == ERROR_INVALID_PARAMETER ) {
|
|
|
|
Usage( Dns, Flat, Site, Db );
|
|
goto Done;
|
|
}
|
|
|
|
if ( Wait && wcscmp( Wait, L"1" ) ) {
|
|
|
|
Block = TRUE;
|
|
}
|
|
|
|
Win32Error = Promote( Dns, Flat, Site, Db, Log, SysVol, Parent,
|
|
Replica, User, Password, Server, Options, Block );
|
|
} else if (argc > 1 && ( _stricmp( argv[ 1 ], "-demote" ) == 0 || _stricmp( argv[ 1 ], "/demote" ) == 0 ) ) {
|
|
|
|
for (i = 2; i < argc; i++ ) {
|
|
|
|
Current = argv[i];
|
|
|
|
if ( !( *Current == '-' || *Current == '/' ) ) {
|
|
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
break;
|
|
}
|
|
|
|
Current++;
|
|
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_USER, Current, UserBuff, User );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_PASSWORD, Current, PasswordBuff, Password );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_ROLE, Current, RoleBuff, RoleScratch );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_OPTIONS, Current, ScratchBuff, Scratch );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_LASTDC, Current, LastDCBuff, LastDCScratch );
|
|
|
|
printf("Unexpected command line value %s\n\n", argv[i] );
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
|
|
break;
|
|
}
|
|
|
|
if ( Scratch != NULL ) {
|
|
|
|
Options = wcstoul( Scratch, &Scratch, 0 );
|
|
}
|
|
|
|
if ( RoleScratch == NULL ) {
|
|
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
|
|
} else {
|
|
|
|
Role = wcstoul( RoleScratch, &RoleScratch, 0);
|
|
}
|
|
|
|
if ( LastDCScratch != NULL ) {
|
|
|
|
LastDC = ( BOOL )wcstoul( LastDCScratch, &LastDCScratch, 0);
|
|
}
|
|
|
|
if( Win32Error == ERROR_INVALID_PARAMETER ) {
|
|
|
|
Usage( Dns, Flat, Site, Db );
|
|
goto Done;
|
|
}
|
|
|
|
Win32Error = Demote( User, Password, Role, Options, LastDC );
|
|
|
|
} else if (argc > 1 && ( _stricmp( argv[ 1 ], "-save" ) == 0 || _stricmp( argv[ 1 ], "/save" ) == 0 ) ) {
|
|
|
|
Win32Error = Save();
|
|
|
|
} else if (argc > 1 && ( _stricmp( argv[ 1 ], "-abort" ) == 0 || _stricmp( argv[ 1 ], "/abort" ) == 0 ) ) {
|
|
|
|
for (i = 2; i < argc; i++ ) {
|
|
|
|
Current = argv[i];
|
|
|
|
if ( !( *Current == '-' || *Current == '/' ) ) {
|
|
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
break;
|
|
}
|
|
|
|
Current++;
|
|
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_ADMINPWD, Current, AdminPwdBuff, AdminPwd );
|
|
|
|
printf("Unexpected command line value %s\n\n", argv[i] );
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
|
|
break;
|
|
}
|
|
|
|
|
|
Win32Error = Abort( AdminPwd );
|
|
|
|
} else if (argc > 1 && ( _stricmp( argv[ 1 ], "-upgrade" ) == 0 || _stricmp( argv[ 1 ], "/upgrade" ) == 0) ) {
|
|
|
|
if ( BuildDefaultsFailed ) {
|
|
|
|
printf( "Failed to get the defaults: %lu\n", Win32Error );
|
|
goto Done;
|
|
|
|
}
|
|
|
|
for (i = 2; i < argc; i++ ) {
|
|
|
|
Current = argv[i];
|
|
|
|
if ( !( *Current == '-' || *Current == '/' ) ) {
|
|
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
break;
|
|
}
|
|
|
|
Current++;
|
|
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_DNS, Current, DnsBuff, Dns );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_SITE, Current, SiteBuff, Site );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_DB, Current, DbBuff, Db );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_LOG, Current, LogBuff, Log );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_SYSVOL, Current, SysVolBuff, SysVol );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_PARENT, Current, ParentBuff, Parent );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_USER, Current, UserBuff, User );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_PASSWORD, Current, PasswordBuff, Password );
|
|
PARSE_CMD_VALUE_AND_CONTINUE( TAG_OPTIONS, Current, ScratchBuff, Scratch );
|
|
|
|
printf("Unexpected command line value %s\n\n", argv[i] );
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
|
|
break;
|
|
}
|
|
|
|
if ( Scratch != NULL ) {
|
|
|
|
Options = wcstoul( Scratch, &Scratch, 0 );
|
|
}
|
|
|
|
//
|
|
// Validate the parameters
|
|
//
|
|
if ( Dns == NULL || Db == NULL || Log == NULL ) {
|
|
|
|
Win32Error = ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
if( Win32Error == ERROR_INVALID_PARAMETER ) {
|
|
|
|
Usage( Dns, Flat, Site, Db );
|
|
goto Done;
|
|
}
|
|
|
|
Win32Error = Upgrade( Dns, Site, Db, Log, SysVol, Parent, User,
|
|
Password, Options );
|
|
}
|
|
|
|
|
|
Done:
|
|
if( Win32Error == ERROR_SUCCESS ) {
|
|
|
|
printf("The command completed successfully\n");
|
|
|
|
} else {
|
|
|
|
printf("The command failed with an error %lu\n", Win32Error );
|
|
|
|
}
|
|
|
|
return( Win32Error );
|
|
}
|