mirror of
https://github.com/reactos/reactos.git
synced 2024-07-04 03:34:16 +00:00
304 lines
8.5 KiB
C
304 lines
8.5 KiB
C
/*
|
|
* CopyFile, MoveFile and related routines test
|
|
*/
|
|
|
|
#include <ctype.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <tchar.h>
|
|
#include <windows.h>
|
|
|
|
static TCHAR
|
|
FindOtherDrive()
|
|
{
|
|
DWORD drives = GetLogicalDrives();
|
|
BOOL found = FALSE;
|
|
TCHAR drive;
|
|
TCHAR rootdir[] = _T( "?:\\" );
|
|
TCHAR currentdir[MAX_PATH + 1];
|
|
|
|
if (0 != GetCurrentDirectory(MAX_PATH + 1, currentdir)) {
|
|
for (drive = _T('A'); ! found && drive <= _T('Z'); drive++) {
|
|
if (0 != (drives & (1 << (drive - _T('A'))))&&
|
|
drive != _totupper(currentdir[0])) {
|
|
rootdir[0] = drive;
|
|
found = (DRIVE_FIXED == GetDriveType(rootdir));
|
|
}
|
|
}
|
|
}
|
|
|
|
return found ? drive - 1 : _T( ' ' );
|
|
}
|
|
|
|
static void
|
|
DeleteTestFile(LPCTSTR filename)
|
|
{
|
|
SetFileAttributes(filename, FILE_ATTRIBUTE_NORMAL);
|
|
DeleteFile(filename);
|
|
}
|
|
|
|
static void
|
|
CreateTestFile(LPCTSTR filename, DWORD attributes)
|
|
{
|
|
HANDLE file;
|
|
char buffer[4096];
|
|
DWORD wrote;
|
|
int c;
|
|
|
|
DeleteTestFile(filename);
|
|
file = CreateFile(filename,
|
|
GENERIC_READ | GENERIC_WRITE,
|
|
0,
|
|
NULL,
|
|
CREATE_ALWAYS,
|
|
0,
|
|
0);
|
|
|
|
if (INVALID_HANDLE_VALUE == file) {
|
|
fprintf(stderr, "CreateFile failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
for(c = 0; c < sizeof(buffer); c++) {
|
|
buffer[c] = (char) c;
|
|
}
|
|
if (! WriteFile(file, buffer, sizeof(buffer), &wrote, NULL)) {
|
|
fprintf(stderr, "WriteFile failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
CloseHandle(file);
|
|
|
|
if (! SetFileAttributes(filename, attributes)) {
|
|
fprintf(stderr, "SetFileAttributes failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
static void
|
|
DeleteTestDir(LPCTSTR dirname)
|
|
{
|
|
RemoveDirectory(dirname);
|
|
}
|
|
|
|
static void
|
|
CreateTestDir(LPCTSTR dirname)
|
|
{
|
|
if (! CreateDirectory(dirname, NULL)) {
|
|
fprintf(stderr, "CreateDirectory failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
static void
|
|
CheckTestFile(LPCTSTR filename, DWORD attributes)
|
|
{
|
|
HANDLE file;
|
|
char buffer[4096];
|
|
DWORD read;
|
|
int c;
|
|
DWORD diskattr;
|
|
|
|
file = CreateFile(filename,
|
|
GENERIC_READ,
|
|
0,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
0,
|
|
0);
|
|
|
|
if (INVALID_HANDLE_VALUE == file) {
|
|
fprintf(stderr, "CreateFile failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
|
|
if (! ReadFile(file, buffer, sizeof(buffer), &read, NULL)) {
|
|
fprintf(stderr, "ReadFile failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
if (read != sizeof(buffer)) {
|
|
fprintf(stderr, "Trying to read %u bytes but got %lu bytes\n", sizeof(buffer), read);
|
|
exit(1);
|
|
}
|
|
for(c = 0; c < sizeof(buffer); c++) {
|
|
if (buffer[c] != (char) c) {
|
|
fprintf(stderr, "File contents changed at position %u\n", c);
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
CloseHandle(file);
|
|
|
|
diskattr = GetFileAttributes(filename);
|
|
if (INVALID_FILE_ATTRIBUTES == diskattr) {
|
|
fprintf(stderr, "GetFileAttributes failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
if (diskattr != attributes) {
|
|
fprintf(stderr, "Attribute mismatch, expected 0x%08lx found 0x%08lx\n", attributes, diskattr);
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
TCHAR otherdrive;
|
|
TCHAR otherfile[ ] = _T("?:\\other.dat");
|
|
|
|
otherdrive = FindOtherDrive();
|
|
|
|
printf("Testing simple move\n");
|
|
CreateTestFile(_T("begin.dat"), FILE_ATTRIBUTE_ARCHIVE);
|
|
DeleteTestFile(_T("end.dat"));
|
|
if (! MoveFile(_T("begin.dat"), _T("end.dat"))) {
|
|
fprintf(stderr, "MoveFile failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
CheckTestFile(_T("end.dat"), FILE_ATTRIBUTE_ARCHIVE);
|
|
DeleteTestFile(_T("end.dat"));
|
|
|
|
printf("Testing move of non-existing file\n");
|
|
DeleteTestFile(_T("begin.dat"));
|
|
DeleteTestFile(_T("end.dat"));
|
|
if (MoveFile(_T("begin.dat"), _T("end.dat"))) {
|
|
fprintf(stderr, "MoveFile succeeded but shouldn't have\n");
|
|
exit(1);
|
|
} else if (ERROR_FILE_NOT_FOUND != GetLastError()) {
|
|
fprintf(stderr, "MoveFile failed with unexpected code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
DeleteTestFile(_T("end.dat"));
|
|
|
|
/* Not correctly implemented in ros, destination file is kept open after this */
|
|
#if 0
|
|
printf("Testing move to existing file\n");
|
|
CreateTestFile(_T("begin.dat"), FILE_ATTRIBUTE_ARCHIVE);
|
|
CreateTestFile(_T("end.dat"), FILE_ATTRIBUTE_ARCHIVE);
|
|
if (MoveFile(_T("begin.dat"), _T("end.dat"))) {
|
|
fprintf(stderr, "MoveFile succeeded but shouldn't have\n");
|
|
exit(1);
|
|
} else if (ERROR_ALREADY_EXISTS != GetLastError()) {
|
|
fprintf(stderr, "MoveFile failed with unexpected code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
DeleteTestFile(_T("begin.dat"));
|
|
DeleteTestFile(_T("end.dat"));
|
|
#endif
|
|
|
|
/* Not implemented yet in ros */
|
|
#if 0
|
|
printf("Testing directory move\n");
|
|
CreateTestDir(_T("begin"));
|
|
CreateTestFile(_T("begin\\file.dat"), FILE_ATTRIBUTE_NORMAL);
|
|
DeleteTestDir(_T("end"));
|
|
if (! MoveFile(_T("begin"), _T("end"))) {
|
|
fprintf(stderr, "MoveFile failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
CheckTestFile(_T("end\\file.dat"), FILE_ATTRIBUTE_NORMAL);
|
|
DeleteTestFile(_T("end\\file.dat"));
|
|
DeleteTestDir(_T("end"));
|
|
#endif
|
|
|
|
printf("Testing file move to different directory\n");
|
|
CreateTestFile(_T("file.dat"), FILE_ATTRIBUTE_NORMAL);
|
|
CreateTestDir(_T("end"));
|
|
if (! MoveFile(_T("file.dat"), _T("end\\file.dat"))) {
|
|
fprintf(stderr, "MoveFile failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
CheckTestFile(_T("end\\file.dat"), FILE_ATTRIBUTE_ARCHIVE);
|
|
DeleteTestFile(_T("end\\file.dat"));
|
|
DeleteTestDir(_T("end"));
|
|
|
|
printf("Testing move of read-only file\n");
|
|
CreateTestFile(_T("begin.dat"), FILE_ATTRIBUTE_READONLY);
|
|
DeleteTestFile(_T("end.dat"));
|
|
if (! MoveFile(_T("begin.dat"), _T("end.dat"))) {
|
|
fprintf(stderr, "MoveFile failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
CheckTestFile(_T("end.dat"), FILE_ATTRIBUTE_ARCHIVE | FILE_ATTRIBUTE_READONLY);
|
|
DeleteTestFile(_T("end.dat"));
|
|
|
|
printf("Testing move to different drive\n");
|
|
if (_T(' ') != otherdrive) {
|
|
otherfile[0] = otherdrive;
|
|
CreateTestFile(_T("begin.dat"), FILE_ATTRIBUTE_ARCHIVE);
|
|
DeleteTestFile(otherfile);
|
|
if (! MoveFile(_T("begin.dat"), otherfile)) {
|
|
fprintf(stderr, "MoveFile failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
CheckTestFile(otherfile, FILE_ATTRIBUTE_ARCHIVE);
|
|
DeleteTestFile(otherfile);
|
|
} else {
|
|
printf(" Test skipped, no other drive available\n");
|
|
}
|
|
|
|
printf("Testing move/overwrite of existing file\n");
|
|
CreateTestFile(_T("begin.dat"), FILE_ATTRIBUTE_ARCHIVE);
|
|
CreateTestFile(_T("end.dat"), FILE_ATTRIBUTE_ARCHIVE);
|
|
if (! MoveFileEx(_T("begin.dat"), _T("end.dat"), MOVEFILE_REPLACE_EXISTING)) {
|
|
fprintf(stderr, "MoveFileEx failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
DeleteTestFile(_T("begin.dat"));
|
|
DeleteTestFile(_T("end.dat"));
|
|
|
|
/* Not (correctly) implemented in ros yet */
|
|
#if 0
|
|
printf("Testing move/overwrite of existing readonly file\n");
|
|
CreateTestFile(_T("begin.dat"), FILE_ATTRIBUTE_ARCHIVE);
|
|
CreateTestFile(_T("end.dat"), FILE_ATTRIBUTE_READONLY);
|
|
if (MoveFileEx(_T("begin.dat"), _T("end.dat"), MOVEFILE_REPLACE_EXISTING)) {
|
|
fprintf(stderr, "MoveFileEx succeeded but shouldn't have\n");
|
|
exit(1);
|
|
} else if (ERROR_ALREADY_EXISTS != GetLastError() &&
|
|
ERROR_ACCESS_DENIED != GetLastError()) {
|
|
fprintf(stderr, "MoveFileEx failed with unexpected code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
DeleteTestFile(_T("begin.dat"));
|
|
DeleteTestFile(_T("end.dat"));
|
|
#endif
|
|
|
|
/* Not implemented in ros yet */
|
|
#if 0
|
|
printf("Testing move to different drive without COPY_ALLOWED\n");
|
|
if (_T(' ') != otherdrive) {
|
|
otherfile[0] = otherdrive;
|
|
CreateTestFile(_T("begin.dat"), FILE_ATTRIBUTE_ARCHIVE);
|
|
DeleteTestFile(otherfile);
|
|
if (MoveFileEx(_T("begin.dat"), otherfile, 0)) {
|
|
fprintf(stderr, "MoveFileEx succeeded but shouldn't have\n");
|
|
exit(1);
|
|
} else if (ERROR_NOT_SAME_DEVICE != GetLastError()) {
|
|
fprintf(stderr, "MoveFileEx failed with unexpected code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
DeleteTestFile(otherfile);
|
|
} else {
|
|
printf(" Test skipped, no other drive available\n");
|
|
}
|
|
#endif
|
|
|
|
printf("Testing move to different drive with COPY_ALLOWED\n");
|
|
if (_T(' ') != otherdrive) {
|
|
otherfile[0] = otherdrive;
|
|
CreateTestFile(_T("begin.dat"), FILE_ATTRIBUTE_ARCHIVE);
|
|
DeleteTestFile(otherfile);
|
|
if (! MoveFileEx(_T("begin.dat"), otherfile, MOVEFILE_COPY_ALLOWED)) {
|
|
fprintf(stderr, "MoveFileEx failed with code %lu\n", GetLastError());
|
|
exit(1);
|
|
}
|
|
CheckTestFile(otherfile, FILE_ATTRIBUTE_ARCHIVE);
|
|
DeleteTestFile(otherfile);
|
|
} else {
|
|
printf(" Test skipped, no other drive available\n");
|
|
}
|
|
|
|
printf("All tests successfully completed\n");
|
|
|
|
return 0;
|
|
}
|