movefile.cpp File Reference

#include "windows.h"
#include <stdio.h>
#include <tchar.h>
#include "stdlib.h"
#include "string.h"

Include dependency graph for movefile.cpp:

Go to the source code of this file.

Macros
#define	WIN32_LEAN_AND_MEAN

Functions
void	Usage ()
int	ShowRegValue ()
int	DeleteValue ()
int	Generate ()
int	Check ()
int	_tmain (int argc, _TCHAR *argv[])

Macro Definition Documentation

WIN32_LEAN_AND_MEAN

#define WIN32_LEAN_AND_MEAN

Definition at line 11 of file movefile.cpp.

Function Documentation

_tmain()

int _tmain	(	int	argc,
		_TCHAR *	argv[]
	)

Definition at line 201 of file movefile.cpp.

{
    if( argc<2 ) {
        Usage();
        return 1;
    }

    if( 0 == strncmp(argv[1], "-g", 2)  )
    {
        // generate test files and registry values
        return Generate();
    }
    else if( 0 == strncmp(argv[1], "-c", 2) )
    {
        // check generated files
        return Check();
    }
    else if( 0 == strncmp(argv[1], "-s", 2) )
    {
        //
        return ShowRegValue();
    }
    else if( 0 == strncmp(argv[1], "-d", 2) )
    {
        return DeleteValue();
    }
    else
    {
        Usage();
        return 1;
    }

    return 0;
}

Check()

int Check

(

)

Definition at line 196 of file movefile.cpp.

{
    return 0;
}

Referenced by _tmain(), START_TEST(), UITOOLS95_DFC_ButtonCheckRadio(), and UpdateStatus().

DeleteValue()

int DeleteValue

(

)

Definition at line 60 of file movefile.cpp.

{
    HKEY hKey;
    LONG retValue;
    // test registry entry
    retValue = RegOpenKeyEx(HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Control\\Session Manager", 0, KEY_SET_VALUE, &hKey);
    if( ERROR_SUCCESS != retValue ) {
        printf("RegOpenKeyEx err=%ld \n", retValue);
        return 1;
    }

    retValue = RegDeleteValue(hKey, "PendingFileRenameOperations");
    if( ERROR_SUCCESS != retValue ) {
        printf("RegDeleteValue err=%ld \n", retValue);
    }

    RegCloseKey(hKey);

    return 0;
}

Referenced by _tmain().

Generate()

int Generate

(

)

Definition at line 81 of file movefile.cpp.

{
    char sBuf[255];
    DWORD dwSize;
    HANDLE hFile = NULL;
    BOOL fReturnValue;

    const char szxReplacedFile[] = "c:\\testFileIsReplaced";
    const char szxMovedFileWithRepl[] = "c:\\testFileShouldBeMovedW";
    const char szxMovedFile[] = "c:\\testFileShouldBeMoved";
    const char szxNewMovedFile[] = "c:\\testFileIsMoved";
    const char szxDeletedFile[] = "c:\\testFileShouldBeDeleted";

    memset(sBuf, 0xaa, sizeof(sBuf));

    // create the first file for moving
    hFile = CreateFile(
        szxMovedFile,
        FILE_ALL_ACCESS,
        0,
        NULL,
        CREATE_ALWAYS,
        FILE_ATTRIBUTE_NORMAL,
        NULL);
    if(NULL == hFile) {
        printf("Can't create the %s file, err=%ld \n", szxMovedFile, GetLastError());
        return 1;
    }
    WriteFile(hFile, sBuf, sizeof(sBuf), &dwSize, NULL);
    CloseHandle(hFile);

    // create the second file for removing
    hFile = CreateFile(
        szxDeletedFile,
        FILE_ALL_ACCESS,
        0,
        NULL,
        CREATE_ALWAYS,
        FILE_ATTRIBUTE_NORMAL,
        NULL);
    if(NULL == hFile) {
        printf("Can't create the %s file, err=%ld \n", szxDeletedFile, GetLastError());
        return 1;
    }
    WriteFile(hFile, sBuf, sizeof(sBuf), &dwSize, NULL);
    CloseHandle(hFile);

    hFile = CreateFile(
        szxReplacedFile,
        FILE_ALL_ACCESS,
        0,
        NULL,
        CREATE_ALWAYS,
        FILE_ATTRIBUTE_NORMAL,
        NULL);
    if(NULL == hFile) {
        printf("Can't create the %s file, err=%ld \n", szxReplacedFile, GetLastError());
        return 1;
    }
    WriteFile(hFile, sBuf, sizeof(sBuf), &dwSize, NULL);
    CloseHandle(hFile);


    hFile = CreateFile(
        szxMovedFileWithRepl,
        FILE_ALL_ACCESS,
        0,
        NULL,
        CREATE_ALWAYS,
        FILE_ATTRIBUTE_NORMAL,
        NULL);
    if(NULL == hFile) {
        printf("Can't create the %s file, err=%ld \n", szxMovedFileWithRepl, GetLastError());
        return 1;
    }
    WriteFile(hFile, sBuf, sizeof(sBuf), &dwSize, NULL);
    CloseHandle(hFile);


    fReturnValue = MoveFileEx(
        szxDeletedFile,
        NULL,
        MOVEFILE_DELAY_UNTIL_REBOOT);
    if( !fReturnValue ) {
        printf("Can't move the %s file, err=%ld \n", szxDeletedFile, GetLastError());
        return 1;
    }

    ShowRegValue();

    fReturnValue = MoveFileEx(
        szxMovedFile,
        szxNewMovedFile,
        MOVEFILE_DELAY_UNTIL_REBOOT);
    if( !fReturnValue ) {
        printf("Can't move the %s file, err=%ld \n", szxMovedFile, GetLastError());
        return 1;
    }

    ShowRegValue();

    fReturnValue = MoveFileEx(
        szxMovedFileWithRepl,
        szxReplacedFile,
        MOVEFILE_DELAY_UNTIL_REBOOT|MOVEFILE_REPLACE_EXISTING);
    if( !fReturnValue ) {
        printf("Can't move the %s file, err=%ld \n", szxMovedFileWithRepl, GetLastError());
        return 1;
    }

    ShowRegValue();

    return 0;
}

Referenced by _tmain().

ShowRegValue()

int ShowRegValue

(

)

Definition at line 28 of file movefile.cpp.

{
    BYTE lpBuff[255];
    memset(lpBuff, 0, sizeof(lpBuff));

    DWORD lSize = sizeof(lpBuff);
    HKEY hKey;
    LONG retValue;
    // test registry entry
    retValue = RegOpenKeyEx(HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Control\\Session Manager", 0, KEY_QUERY_VALUE, &hKey);
    if( ERROR_SUCCESS != retValue ) {
        printf("RegOpenKeyEx err=%ld \n", retValue);
        return 1;
    }

    retValue = RegQueryValueEx(hKey, "PendingFileRenameOperations", NULL, NULL, lpBuff, &lSize);
    if( ERROR_SUCCESS != retValue ) {
        printf("RegQueryValueEx err=%ld \n", retValue);
        lSize = 0;
    }

    printf("reg data: \n");
    for(UINT i=0; i<lSize; i++) {
        printf("%c", lpBuff[i]);
    }
    printf("\n");

    RegCloseKey(hKey);

    return 0;
}

Referenced by _tmain(), and Generate().

Usage()

void Usage

(

void

)

Definition at line 19 of file movefile.cpp.

{
    printf(" Usage: smssTest.exe -g|c|s|d \n \
            g - generate test files \n \
            c - check files after reboot \n \
            s - show registry entry \n \
            d - delete registry value \n");
}

Referenced by _tmain().