154 lines
4.5 KiB
C++
154 lines
4.5 KiB
C++
/*++
|
||
|
||
Copyright (c) 1989-1997 Microsoft Corporation
|
||
|
||
Module Name:
|
||
|
||
bkup.c - test backup read / write
|
||
|
||
Abstract:
|
||
|
||
Driver for backup read/write
|
||
|
||
--*/
|
||
|
||
|
||
extern "C" {
|
||
#include <nt.h>
|
||
#include <ntioapi.h>
|
||
#include <ntrtl.h>
|
||
#include <nturtl.h>
|
||
}
|
||
|
||
#include <windows.h>
|
||
|
||
#include <stdio.h>
|
||
#include <string.h>
|
||
#include <tchar.h>
|
||
|
||
|
||
//
|
||
//
|
||
//
|
||
|
||
#define SHIFTBYN(c,v,n) ((c) -= (n), (v) += (n))
|
||
#define SHIFT(c,v) SHIFTBYN( (c), (v), 1 )
|
||
|
||
void Usage( void )
|
||
{
|
||
fprintf( stderr, "Usage: bkup [-r file] performs backup read, output to stdout\n" );
|
||
fprintf( stderr, " [-w backupimage newfile] performs backup write\n" );
|
||
}
|
||
|
||
int __cdecl
|
||
main (
|
||
int argc,
|
||
char **argv
|
||
)
|
||
{
|
||
SHIFT( argc, argv );
|
||
|
||
if (argc == 2 && !strcmp( *argv, "-r" )) {
|
||
SHIFT( argc, argv );
|
||
|
||
HANDLE h = CreateFile( *argv, // file name
|
||
FILE_READ_DATA | FILE_READ_ATTRIBUTES,
|
||
// desired access
|
||
FILE_SHARE_READ, // share mode
|
||
NULL, // security
|
||
OPEN_EXISTING, // disposition
|
||
0, // flags and attributes
|
||
0 ); // template file
|
||
|
||
if (h == INVALID_HANDLE_VALUE) {
|
||
fprintf( stderr, "CreateFile returned %x\n", GetLastError( ));
|
||
return 1;
|
||
}
|
||
|
||
BYTE *p = new BYTE[65536];
|
||
ULONG Length, Written;
|
||
|
||
PVOID Context = NULL;
|
||
|
||
while (TRUE) {
|
||
if (!BackupRead( h, p, 65536, &Length, FALSE, TRUE, &Context )) {
|
||
fprintf( stderr, "BackupRead returned %x\n", GetLastError( ));
|
||
return 1;
|
||
}
|
||
|
||
if (Length == 0) {
|
||
break;
|
||
}
|
||
|
||
fprintf( stderr, "Transferring %x\n", Length );
|
||
if (!WriteFile( GetStdHandle( STD_OUTPUT_HANDLE ), p, Length, &Written, NULL ))
|
||
{
|
||
fprintf( stderr, "WriteFile returned %x\n", GetLastError( ));
|
||
return 1;
|
||
}
|
||
}
|
||
} else if (argc == 3 && !strcmp( *argv, "-w" )) {
|
||
SHIFT( argc, argv );
|
||
|
||
HANDLE hOld = CreateFile( *argv, // file name
|
||
FILE_READ_DATA, // desired access
|
||
FILE_SHARE_READ, // share mode
|
||
NULL, // security
|
||
OPEN_EXISTING, // disposition
|
||
0, // flags and attributes
|
||
0 ); // template file
|
||
|
||
if (hOld == INVALID_HANDLE_VALUE) {
|
||
fprintf( stderr, "CreateFile destination returned %x\n", GetLastError( ));
|
||
return 1;
|
||
}
|
||
|
||
SHIFT( argc, argv );
|
||
|
||
HANDLE hNew = CreateFile( *argv, // file name
|
||
FILE_WRITE_DATA | FILE_WRITE_ATTRIBUTES,
|
||
// desired access
|
||
FALSE, // share mode
|
||
NULL, // security
|
||
CREATE_ALWAYS, // disposition
|
||
0, // flags and attributes
|
||
0 ); // template file
|
||
|
||
if (hNew == INVALID_HANDLE_VALUE) {
|
||
fprintf( stderr, "CreateFile destination returned %x\n", GetLastError( ));
|
||
return 1;
|
||
}
|
||
|
||
BYTE *p = new BYTE[127];
|
||
ULONG Length, Written;
|
||
|
||
PVOID Context = NULL;
|
||
|
||
while (TRUE) {
|
||
if (!ReadFile( hOld, p, 127, &Length, NULL )) {
|
||
fprintf( stderr, "ReadFile returned %x\n", GetLastError( ));
|
||
return 1;
|
||
}
|
||
|
||
if (Length == 0) {
|
||
break;
|
||
}
|
||
|
||
fprintf( stderr, "Transferring %x\n", Length );
|
||
|
||
if (!BackupWrite( hNew, p, Length, &Written, FALSE, TRUE, &Context )) {
|
||
fprintf( stderr, "BackupWrite returned %x\n", GetLastError( ));
|
||
return 1;
|
||
}
|
||
}
|
||
} else {
|
||
Usage( );
|
||
return 1;
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
|
||
|