/*++
Copyright (c) 1991  Microsoft Corporation

Module Name:
    rtreplac.c

Abstract:
    NT level registry api test program, basic non-error paths.

    Perform an NtReplaceKey call to replace a registry hive.

    rtreplac  <KeyPath> <NewHiveName> <BackupOldHiveName>

    Example:

        rtreplac \registry\machine\user newuser olduser

Author:
    John Vert (jvert) 8-May-1992
--*/

#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>

#include "cmp.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define WORK_SIZE   1024

void __cdecl main(int, char *);
void processargs();

UNICODE_STRING  KeyPath;
WCHAR           KeyPathBuffer[WORK_SIZE];

UNICODE_STRING  NewName;
WCHAR           NewNameBuffer[WORK_SIZE];
UNICODE_STRING  OldName;
WCHAR           OldNameBuffer[WORK_SIZE];


void __cdecl main(int argc, char *argv[])
{
    NTSTATUS status;
    OBJECT_ATTRIBUTES NewAttributes;
    OBJECT_ATTRIBUTES OldAttributes;
    OBJECT_ATTRIBUTES ObjectAttributes;
    IO_STATUS_BLOCK  IoStatus;
    HANDLE  FileHandle;
    HANDLE  KeyHandle;
    BOOLEAN WasEnabled;

    // Process args

    KeyPath.MaximumLength = WORK_SIZE;
    KeyPath.Length = 0L;
    KeyPath.Buffer = &(KeyPathBuffer[0]);

    NewName.MaximumLength = WORK_SIZE;
    NewName.Length = 0L;
    NewName.Buffer = &(NewNameBuffer[0]);

    OldName.MaximumLength = WORK_SIZE;
    OldName.Length = 0L;
    OldName.Buffer = &(OldNameBuffer[0]);

    processargs(argc, argv);

    // Set up and open FileName

    printf("rtreplac: starting\n");

    InitializeObjectAttributes(&NewAttributes, &NewName, OBJ_CASE_INSENSITIVE, (HANDLE)NULL, NULL);
    InitializeObjectAttributes(&OldAttributes, &OldName, OBJ_CASE_INSENSITIVE, (HANDLE)NULL, NULL);
    InitializeObjectAttributes(&ObjectAttributes, &KeyPath, OBJ_CASE_INSENSITIVE, (HANDLE)NULL, NULL);
    status = NtOpenKey(&KeyHandle, MAXIMUM_ALLOWED, &ObjectAttributes);
    if (!NT_SUCCESS(status)) {
        printf("rtreplac: key open failed status = %08lx\n", status);
        exit(1);
    }

    RtlAdjustPrivilege(SE_RESTORE_PRIVILEGE, TRUE, FALSE, &WasEnabled);
    status = NtReplaceKey(&NewAttributes, KeyHandle, &OldAttributes);
    RtlAdjustPrivilege(SE_RESTORE_PRIVILEGE, WasEnabled, FALSE, &WasEnabled);
    if (!NT_SUCCESS(status)) {
        printf("rtreplac: NtReplaceKey failed status = %08lx\n", status);
        exit(1);
    }

    printf("rtreplac: success\n");
    exit(0);
}


void processargs(int argc, char *argv[])
{
    ANSI_STRING temp;
    UNICODE_STRING DosFileName;

    if ( (argc != 4) )
    {
        printf("Usage: %s <KeyName> <NewFileName> <OldFileName>\n", argv[0]);
        exit(1);
    }

    RtlInitAnsiString(&temp, argv[1]);
    RtlAnsiStringToUnicodeString(&KeyPath, &temp, TRUE);
    RtlInitAnsiString(&temp, argv[2]);
    RtlAnsiStringToUnicodeString(&DosFileName, &temp, TRUE);
    RtlDosPathNameToNtPathName_U( DosFileName.Buffer, &NewName, NULL, NULL );
    RtlInitAnsiString(&temp, argv[3]);
    RtlAnsiStringToUnicodeString(&DosFileName, &temp, TRUE);
    RtlDosPathNameToNtPathName_U( DosFileName.Buffer, &OldName, NULL, NULL );
}