rtrest2.c File Reference

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

Go to the source code of this file.

Defines
#define	WORK_SIZE   1024
Functions
void __cdecl	main (int, char *)
void	processargs ()
void __cdecl	main (int argc, char *argv[])
void	processargs (int argc, char *argv[])
Variables
UNICODE_STRING	KeyPath
WCHAR	KeyPathBuffer [WORK_SIZE]
UNICODE_STRING	FileName
WCHAR	FileNameBuffer [WORK_SIZE]

---

Define Documentation

#define WORK_SIZE   1024

Definition at line 37 of file rtrest2.c.

---

Function Documentation

void __cdecl main (  int  argc, char *  argv[] )

Definition at line 49 of file rtrest2.c. References exit, FileName, FileNameBuffer, KeyPath, KeyPathBuffer, L, NT_SUCCESS, NtCreateFile(), NtOpenKey(), NtRestoreKey(), NTSTATUS(), NULL, ObjectAttributes, processargs(), and WORK_SIZE. { NTSTATUS status; OBJECT_ATTRIBUTES ObjectAttributes; IO_STATUS_BLOCK IoStatus; HANDLE FileHandle; HANDLE KeyHandle; // // Process args // KeyPath.MaximumLength = WORK_SIZE; KeyPath.Length = 0L; KeyPath.Buffer = &(KeyPathBuffer[0]); FileName.MaximumLength = WORK_SIZE; FileName.Length = 0L; FileName.Buffer = &(FileNameBuffer[0]); processargs(argc, argv); // // Set up and open FileName // printf("rtrestor: starting\n"); InitializeObjectAttributes( &ObjectAttributes, &FileName, 0, (HANDLE)NULL, NULL ); ObjectAttributes.Attributes |= OBJ_CASE_INSENSITIVE; status = NtCreateFile( &FileHandle, GENERIC_READ | SYNCHRONIZE, &ObjectAttributes, &IoStatus, 0, // AllocationSize FILE_ATTRIBUTE_NORMAL, 0, // ShareAccess FILE_OPEN_IF, FILE_SYNCHRONOUS_IO_NONALERT, NULL, // EaBuffer 0 // EaLength ); if (!NT_SUCCESS(status)) { printf("rtsave: file open failed status = %08lx\n", status); exit(1); } InitializeObjectAttributes( &ObjectAttributes, &KeyPath, 0, (HANDLE)NULL, NULL ); ObjectAttributes.Attributes |= OBJ_CASE_INSENSITIVE; status = NtOpenKey( &KeyHandle, MAXIMUM_ALLOWED, &ObjectAttributes ); if (!NT_SUCCESS(status)) { printf("rtsave: key open failed status = %08lx\n", status); exit(1); } status = NtRestoreKey(KeyHandle, FileHandle); if (!NT_SUCCESS(status)) { printf("rtrestor: NtRestorKey failed status = %08lx\n", status); exit(1); } printf("rtsave: success\n"); exit(0); }

void __cdecl main (  int  , char *  )

void processargs (  int  argc, char *  argv[] )

Definition at line 140 of file rtrest2.c. References exit, FileName, KeyPath, RtlAnsiStringToUnicodeString(), RtlInitAnsiString(), and TRUE. { ANSI_STRING temp; if ( (argc != 3) ) { printf("Usage: %s <KeyName> <FileName>\n", argv[0]); exit(1); } RtlInitAnsiString( &temp, argv[1] ); RtlAnsiStringToUnicodeString( &KeyPath, &temp, TRUE ); RtlInitAnsiString( &temp, argv[2] ); RtlAnsiStringToUnicodeString( &FileName, &temp, TRUE ); return; }

void processargs (   )

---

Variable Documentation

UNICODE_STRING FileName

Definition at line 45 of file rtrest2.c.

WCHAR FileNameBuffer[WORK_SIZE]

Definition at line 46 of file rtrest2.c.

UNICODE_STRING KeyPath

Definition at line 42 of file rtrest2.c.

WCHAR KeyPathBuffer[WORK_SIZE]

Definition at line 43 of file rtrest2.c.

---