TnbRecoveryFiles.h

[詳解]

#pragma once
#include "TnbFileName.h"
#include "TnbStrVector.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CRecoveryFiles
{
public:
 
    CRecoveryFiles(bool isDefaultRecoveryMode = false) : m_isDefaultRecoveryMode(isDefaultRecoveryMode)
    {
    }
 
    ~CRecoveryFiles(void)
    {
        if ( m_isDefaultRecoveryMode )
        {
            RecoverAll();
        }
        else
        {
            UnregisterAll();
        }
    }
 
    bool Register(LPCTSTR lpszFile)
    {
        CFileName fn(lpszFile);
        if ( fn.IsExist() )
        {
            CStr bak;
            loop ( i, 0x10000 )
            {
                bak = fn.GetFullName() + CStr::Fmt(_T(".%04X.bak"), i);
                if ( ! m_Exist(bak) )
                {
                    break;
                }
            }
            if ( fn.CopyTo(bak) )
            {
                TParam p(fn.GetFullName(), bak);
                p.backHandle = ::CreateFile(bak, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
                m_names.Add(p);
                return true;
            }
        }
        return false;
    }
 
    bool UnregisterAll(void)
    {
        bool r = true;
        loop ( i, m_names.GetSize() )
        {
            m_names[i].backHandle.Null();
            r &= !! ::DeleteFile(m_names[i].backName);
        }
        m_names.RemoveAll();
        return r;
    }
 
    bool RecoverAll(void)
    {
        bool r = true;
        //== 先にtmpファイル作成
        loop ( i, m_names.GetSize() )
        {
            CStr reg = m_names[i].registName;
            if ( m_Exist(reg) )
            {
                CStr tmp;
                loop ( ii, 0x10000 )
                {
                    tmp = reg + CStr::Fmt(_T(".%04X.tmp"), ii);
                    if ( ! m_Exist(tmp) )
                    {
                        break;
                    }
                }
                if ( ! ::MoveFile(reg, tmp) )
                {
                    _GetLastError("MoveFile");
                    m_names[i].tempName.Empty();
                    r = false;
                }
                else
                {
                    m_names[i].tempName = tmp;
                }
            }
        }
        //== tmpファイルが出来たものはbakをオリジナルにリネームし、tmpを削除。
        loop ( i, m_names.GetSize() )
        {
            m_names[i].backHandle.Null();
            if ( ::MoveFile(m_names[i].backName, m_names[i].registName) )
            {
                if ( ! m_names[i].tempName.IsEmpty() )
                {
                    VERIFY( ::DeleteFile(m_names[i].tempName) );
                }
            }
            else
            {
                // 意図的に bak ファイルを消した場合など.
                r = false;
            }
        }
        m_names.RemoveAll();
        return r;
    }
 
private:
    bool m_Exist(LPCTSTR lpszName) const
    {
        WIN32_FIND_DATA t;
        HANDLE h = ::FindFirstFile(lpszName, &t);
        if ( h != INVALID_HANDLE_VALUE )
        {
            ::FindClose(h);
            return true;
        }
        return false;
    }
    struct TParam
    {
        CStr            registName; 
        CStr            backName;   
        CHandleHandle   backHandle; 
        CStr            tempName;   
        TParam(LPCTSTR reg = NULL, LPCTSTR tmp = NULL) : registName(reg), backName(tmp)
        {
        }
    };
    CVectorT<TParam>    m_names;                    
    bool                m_isDefaultRecoveryMode;    
    friend class CRecoveryFilesTest;
};
 
 
 
}; // TNB