Windows编程_Lesson004_项目预备_同步IO操作

封装File类

我们在进行文件操作时，会有两个版本，一个带_s的版本，一个不带_s的版本，带_s的版本多传递了一个size_t的参数，这个参数有两个功能：

1. 进行文件操作的时候不会越界；
2. 不会被别人使用shellcode进行注入，造成被黑客攻击。
   所以在Windows下编程，建议使用_s版本。

先看原始的例子

#include <iostream>
#include <Windows.h>
#include <tchar.h>



class WindowException
{
public:
    WindowException(DWORD dwErrorCode) : m_dwErrorCode(dwErrorCode)
    {
        FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
            nullptr, m_dwErrorCode, 0, reinterpret_cast<LPWSTR>(&m_strErrorMsg), 0, nullptr);
    }
    ~WindowException()
    {
        LocalFree(m_strErrorMsg);
    }
    const TCHAR *What() const
    {
        return m_strErrorMsg;
    }

private:
    DWORD m_dwErrorCode;
    TCHAR *m_strErrorMsg;
};

class MyFile
{
public:
    MyFile(const TCHAR *strFilePath = _TEXT("")) : m_hFile(INVALID_HANDLE_VALUE)
    {
        SetPath(strFilePath);
    }
    ~MyFile()
    {
        CloseFile();
    }
    BOOL OpenFile(DWORD dwDesiredAccess = GENERIC_READ|GENERIC_WRITE,
        DWORD                 dwShareMode = FILE_SHARE_READ,
        DWORD                 dwCreationDisposition = OPEN_ALWAYS,
        DWORD                 dwFlagsAndAttributes = FILE_ATTRIBUTE_NORMAL,
        LPSECURITY_ATTRIBUTES lpSecurityAttributes = nullptr,
        HANDLE                hTemplateFile = nullptr
    )
    {
        BOOL bRet = TRUE;

        m_hFile = CreateFile(m_strFilePath, dwDesiredAccess, dwShareMode, lpSecurityAttributes, dwCreationDisposition, dwFlagsAndAttributes, hTemplateFile);

        if (m_hFile == INVALID_HANDLE_VALUE)
        {
            bRet = FALSE;
            throw WindowException(GetLastError());
        }

        return bRet;
    }

    VOID CloseFile()
    {
        if (m_hFile != INVALID_HANDLE_VALUE)
        {
            CloseHandle(m_hFile);
            m_hFile = INVALID_HANDLE_VALUE;
        }
    }

    VOID SetPath(const TCHAR *strFilePath)
    {
        SIZE_T nstrLen = _tcslen(strFilePath) + sizeof(TCHAR);
        m_strFilePath = new TCHAR[nstrLen];
        _tcscpy_s(m_strFilePath, nstrLen, strFilePath);
    }
    const TCHAR *GetPath() const
    {
        return m_strFilePath;
    }

    const HANDLE GetFileHandle() const
    {
        return m_hFile;
    }

private:
    HANDLE m_hFile;
    TCHAR *m_strFilePath;
};


int main()
{
    try
    {
        MyFile my_file;
        my_file.SetPath(TEXT("Demo.txt"));
        my_file.OpenFile();


        // 1 取文件大小

        // 1.1 物理大小
        LARGE_INTEGER file_size = { 0 };
        GetFileSizeEx(my_file.GetFileHandle(), &file_size);
        std::cout << "file size:" << file_size.QuadPart << std::endl;

        // 1.2 磁盘上占用的大小
        file_size.LowPart = GetCompressedFileSize(my_file.GetPath(), reinterpret_cast<LPDWORD>(&(file_size.HighPart)));
        std::cout << "compress file size:" << file_size.QuadPart << std::endl;

        // 2. 同步IO
        // 2.1 读
        BYTE bBuf[MAXBYTE] = { 0 };
        DWORD dwNumberofBytesRead = 0;

        // 用64位的值来记录当前文件操作的偏移量
        LARGE_INTEGER largeBegin = { 0 };
        file_size.LowPart = 100;

        SetFilePointerEx(my_file.GetFileHandle(), file_size, nullptr, FILE_BEGIN);

        ReadFile(my_file.GetFileHandle(), bBuf, MAXBYTE, &dwNumberofBytesRead, nullptr);
        bBuf[MAXBYTE - 1] = '\0';
        std::cout << reinterpret_cast<char *>(bBuf) << "写入的大小为：" << dwNumberofBytesRead << std::endl;

        // 3. 写
        DWORD dwNumberofBytesWritten = 0;
        WriteFile(my_file.GetFileHandle(), bBuf, MAXBYTE, &dwNumberofBytesWritten, nullptr);
        std::cout << "写入成功！" << std::endl;


        // 文件本身大小为22 我想设置为1024
        // 设置文件尾
        file_size.QuadPart = 1024;
        SetFilePointerEx(my_file.GetFileHandle(), file_size, nullptr, FILE_BEGIN);

        // 上面的文件操作是在做BT下载的时候用的比较多，把里面的文件分成一块一块的部分，将文件大小设置为目标文件大小
        SetEndOfFile(my_file.GetFileHandle());

        my_file.CloseFile();
    }
    catch(WindowException &exception)
    {
        MessageBox(nullptr, exception.What(), L"提示", MB_OK);
    }

    return 0;
}
运行结果：