无语的CAtlTemporaryFile类

经常需要将一些内容写进临时文件,这就需要结合GetTempPath和GetTempFileName在Windows临时目录下生成一个临时文件。ATL提供了类CAtlTemporaryFile来进行这层封装,可使用此类时,才发现该类有多么地鸡肋啊.我写了一个Demo:

        CAtlTemporaryFile fileTmp;     
        fileTmp.Create();
        fileTmp.Write( COLE2CT( bstrHtml ), wcslen( bstrHtml )  );
        fileTmp.Close( );
       
       fileTmp.TempFileName();

照理说这样应该就可以了,但一直得不到我要的文件,跟踪CAtlTemporaryFile源码才发现在成员方法Close已经把生成的文件删除了.我纳闷了,这是啥意思嘛!还让不让人用啊。再一细看,发现Close方法提供了一个参数:szNewName,Close方法将生成的文件移动到szNewName指定的文件,若不指定,则直接删除生成的文件.由于Close方法移动前会删除szNewName指定的文件已确保其不存在,因此,我最后一个希望:将szNewName指定为CAtlTemporaryFile生成的文件(由TempFileName获取)也破灭了.真是悲剧了.让我重新指定一个目标文件地址,我还用CAtlTemporaryFile干嘛啊,真的简直是鸡肋.没办法,我只有定制CAtlTemporaryFile的源码了,还好该源码没有那么多的依赖关系,十分利于拷贝,哈哈.下面是我的修改版本:

#include <atlfile.h>

template< bool bDeleted >
class CAtlTemporaryFileExT
{
public:
    CAtlTemporaryFileExT() throw()
    {
    }

    ~CAtlTemporaryFileExT() throw()
    {
        // Ensure that the temporary file is closed and deleted,
        // if necessary.
        if (m_file.m_h != NULL)
        {
            Close();
        }

        //如果设为自动删除,结束时则删除文件
        if( bDeleted )
        {
            Delete();
        }
    }

    HRESULT Create(__in_opt LPCTSTR pszDir = NULL, __in DWORD dwDesiredAccess = GENERIC_WRITE) throw()
    {
        TCHAR szPath[_MAX_PATH]; 
        TCHAR tmpFileName[_MAX_PATH]; 

        ATLASSUME(m_file.m_h == NULL);

        if (pszDir == NULL)
        {
            DWORD dwRet = GetTempPath(_MAX_DIR, szPath);
            if (dwRet == 0)
            {
                // Couldn't find temporary path;
                return AtlHresultFromLastError();
            }
            else if (dwRet > _MAX_DIR)
            {
                return DISP_E_BUFFERTOOSMALL;
            }
        }
        else
        {
            if(Checked::tcsncpy_s(szPath, _countof(szPath), pszDir, _TRUNCATE)==STRUNCATE)
            {
                return DISP_E_BUFFERTOOSMALL;
            }
        }

        if (!GetTempFileName(szPath, _T("TFR"), 0, tmpFileName))
        {
            // Couldn't create temporary filename;
            return AtlHresultFromLastError();
        }
        tmpFileName[_countof(tmpFileName)-1]='\0';

        Checked::tcsncpy_s(m_szTempFileName, _countof(m_szTempFileName), tmpFileName, _TRUNCATE);
        SECURITY_ATTRIBUTES secatt;
        secatt.nLength = sizeof(secatt);
        secatt.lpSecurityDescriptor = NULL;
        secatt.bInheritHandle = TRUE;

        m_dwAccess = dwDesiredAccess;

        return m_file.Create(
            m_szTempFileName,
            m_dwAccess,
            0,
            CREATE_ALWAYS,
            FILE_ATTRIBUTE_NOT_CONTENT_INDEXED | FILE_ATTRIBUTE_TEMPORARY,
            &secatt);
    }

    HRESULT Close() throw()
    {
        ATLASSUME(m_file.m_h != NULL);

        m_file.Close();

        return S_OK;
    }

    
    HRESULT Delete()
    {
         return ::DeleteFile( m_szTempFileName ) ? S_OK : S_FALSE;
    }
    
    HRESULT HandsOff() throw()
    {
        m_file.Flush();
        m_file.Close();

        return S_OK;
    }

    HRESULT HandsOn() throw()
    {
        HRESULT hr = m_file.Create(
            m_szTempFileName,
            m_dwAccess,
            0,
            OPEN_EXISTING);
        if (FAILED(hr))
            return hr;

        return m_file.Seek(0, FILE_END);
    }

    HRESULT Read(
        __out_bcount(nBufSize) LPVOID pBuffer,
        __in DWORD nBufSize,
        __out DWORD& nBytesRead) throw()
    {
        return m_file.Read(pBuffer, nBufSize, nBytesRead);
    }

    HRESULT Write(
        __in_bcount(nBufSize) LPCVOID pBuffer,
        __in DWORD nBufSize,
        __out_opt DWORD* pnBytesWritten = NULL) throw()
    {
        return m_file.Write(pBuffer, nBufSize, pnBytesWritten);
    }

    HRESULT Seek(__in LONGLONG nOffset, __in DWORD dwFrom = FILE_CURRENT) throw()
    {
        return m_file.Seek(nOffset, dwFrom);
    }

    HRESULT GetPosition(__out ULONGLONG& nPos) const throw()
    {
        return m_file.GetPosition(nPos);
    }

    HRESULT Flush() throw()
    {
        return m_file.Flush();
    }

    HRESULT LockRange(__in ULONGLONG nPos, __in ULONGLONG nCount) throw()
    {
        return m_file.LockRange(nPos, nCount);
    }

    HRESULT UnlockRange(__in ULONGLONG nPos, __in ULONGLONG nCount) throw()
    {
        return m_file.UnlockRange(nPos, nCount);
    }

    HRESULT SetSize(__in ULONGLONG nNewLen) throw()
    {
        return m_file.SetSize(nNewLen);
    }

    HRESULT GetSize(__out ULONGLONG& nLen) const throw()
    {
        return m_file.GetSize(nLen);
    }

    operator HANDLE() throw()
    {
        return m_file;
    }

    LPCTSTR TempFileName() throw()
    {
        return m_szTempFileName;
    }

private:
    CAtlFile m_file;
    TCHAR m_szTempFileName[_MAX_FNAME+1];
    DWORD m_dwAccess;
};

typedef CAtlTemporaryFileExT<false> CAtlTemporaryFileEx;

//自动删除版本
typedef CAtlTemporaryFileExT<true> CAtlTemporaryFileExHandle;