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.
153 lines
4.5 KiB
153 lines
4.5 KiB
/*++
|
|
|
|
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;
|
|
}
|
|
|
|
|
|
|