Ticket #21967: testkeep.cpp

#include <windows.h>
#include <stdio.h>
#include <tchar.h>
#include <exception>

void PrintLastError(int line) {
    DWORD errorMessageID = ::GetLastError();
    if(errorMessageID == 0) {
        _tprintf(_T("Operation successful.\n"));
        return;
    }

    LPTSTR messageBuffer = nullptr;
    size_t size = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
                                NULL, errorMessageID, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&messageBuffer, 0, NULL);

    _tprintf(_T("Line: %d Error code %lu: %s\n"), line, errorMessageID, messageBuffer);
    LocalFree(messageBuffer);
}

int main() {
    try {
        TCHAR tempFilePath[MAX_PATH] = _T("Z:\\test.exe.tmp");
        wchar_t finalFilePath[MAX_PATH] = L"Z:\\test.exe";

        // Create a file
        HANDLE hFile = CreateFile(tempFilePath, GENERIC_READ | GENERIC_WRITE | DELETE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
        if (hFile == INVALID_HANDLE_VALUE) {
            PrintLastError(__LINE__);
            return 1;
        }

        // Create a file mapping object
        HANDLE hMapFile = CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 1024, NULL);
        if (hMapFile == NULL) {
            PrintLastError(__LINE__);
            CloseHandle(hFile);
            return 1;
        }

        // Map a view of the file into the address space
        LPVOID pMapView = MapViewOfFile(hMapFile, FILE_MAP_ALL_ACCESS, 0, 0, 1024);
        if (pMapView == NULL) {
            PrintLastError(__LINE__);
            CloseHandle(hMapFile);
            CloseHandle(hFile);
            return 1;
        }
        // Close the mapping handle; the mapping is still kept open.
        CloseHandle(hMapFile);

        // Write to the memory area
        const char data[] = "Sample Data";
        memcpy(pMapView, data, sizeof(data));

        // Unmap the file
        UnmapViewOfFile(pMapView);
    //    FlushFileBuffers(hFile);

        char buf[sizeof(FILE_RENAME_INFO) + sizeof(wchar_t)*100];
        FILE_RENAME_INFO *rename = (FILE_RENAME_INFO*)buf;
        rename->ReplaceIfExists = 1;
        rename->RootDirectory = 0;
        rename->FileNameLength = wcslen(finalFilePath) * sizeof(wchar_t);
        memcpy(rename->FileName, finalFilePath, wcslen(finalFilePath) * sizeof(wchar_t));

        // Rename with SetFileInformationByHandle(FileRenameInfo)
        if (!SetFileInformationByHandle(hFile, FileRenameInfo, buf, sizeof(buf))) {
            PrintLastError(__LINE__);
            CloseHandle(hFile);
            return 1;
        }
        CloseHandle(hFile);

        _tprintf(_T("File operation completed successfully.") );
        return 0;
    }
    catch (std::exception& e) {
        _tprintf(_T("%s\n"), e.what() );
        return 1;
    }
}